| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright 2015 The Chromium Authors. All rights reserved. | |
| 3 # Use of this source code is governed by a BSD-style license that can be | |
| 4 # found in the LICENSE file. | |
| 5 | |
| 6 """The main task entrypoint.""" | |
| 7 | |
| 8 import argparse | |
| 9 import logging | |
| 10 import socket | |
| 11 import sys | |
| 12 import time | |
| 13 | |
| 14 #pylint: disable=relative-import | |
| 15 import common_lib | |
| 16 import rpc_server | |
| 17 | |
| 18 | |
| 19 def main(): | |
| 20 print ' '.join(sys.argv) | |
| 21 common_lib.InitLogging() | |
| 22 logging.info('Task starting') | |
| 23 | |
| 24 parser = argparse.ArgumentParser() | |
| 25 parser.add_argument('--otp', | |
| 26 help='One time token used to authenticate with the host') | |
| 27 parser.add_argument('--controller', | |
| 28 help='The ip address of the controller machine') | |
| 29 parser.add_argument('--idle-timeout', type=int, | |
| 30 default=common_lib.DEFAULT_TIMEOUT_SECS, | |
| 31 help='The idle timeout for the rpc server in seconds') | |
| 32 args, _ = parser.parse_known_args() | |
| 33 | |
| 34 logging.info( | |
| 35 'Registering with registration server at %s using OTP "%s"', | |
| 36 args.controller, args.otp) | |
| 37 server = common_lib.ConnectToServer(args.controller).RegisterTask( | |
| 38 args.otp, common_lib.MY_IP) | |
| 39 | |
| 40 server = rpc_server.RPCServer(args.controller, args.idle_timeout) | |
| 41 | |
| 42 server.serve_forever() | |
| 43 return 0 | |
| 44 | |
| 45 | |
| 46 if __name__ == '__main__': | |
| 47 sys.exit(main()) | |
| OLD | NEW |