blob: 59a3d8ae7da123c0253597800ad6050a13b81502 [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2015 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""The main task entrypoint."""
import argparse
import logging
import socket
import sys
import time
# pylint: disable=relative-import
# Import common_lib first so we can setup the environment
from lib import common_lib
from legion.lib.rpc import rpc_server
def main():
print ' '.join(sys.argv)
common_lib.InitLogging()'Task starting')
parser = argparse.ArgumentParser()
help='One time token used to authenticate with the host')
help='The ip address of the controller machine')
parser.add_argument('--controller-port', type=int,
help='The port the controllers registration server is on')
parser.add_argument('--idle-timeout', type=int,
help='The idle timeout for the rpc server in seconds')
args, _ = parser.parse_known_args()
my_port = common_lib.GetUnusedPort()
'Registering with registration server at %s:%d using OTP "%s"',
args.controller, args.controller_port, args.otp)
args.controller, args.controller_port).RegisterTask(
args.otp, common_lib.MY_IP, my_port)
server = rpc_server.RpcServer(args.controller, my_port, args.idle_timeout)
server.serve_forever()'Server shutdown complete')
return 0
if __name__ == '__main__':