testing/legion/client_controller.py

   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 client_controller code.
   7
   8 This code is the main entry point for the client machines and handles
   9 registering with the host server and running the local RPC server.
  10 """
  11
  12 import argparse
  13 import logging
  14 import socket
  15 import sys
  16 import time
  17
  18 #pylint: disable=relative-import
  19 import client_rpc_server
  20 import common_lib
  21
  22
  23 def main():
  24   print ' '.join(sys.argv)
  25   common_lib.InitLogging()
  26   logging.info('Client controller starting')
  27
  28   parser = argparse.ArgumentParser()
  29   parser.add_argument('--otp',
  30                       help='One time token used to authenticate with the host')
  31   parser.add_argument('--host',
  32                       help='The ip address of the host')
  33   parser.add_argument('--idle-timeout', type=int,
  34                       default=common_lib.DEFAULT_TIMEOUT_SECS,
  35                       help='The idle timeout for the rpc server in seconds')
  36   args, _ = parser.parse_known_args()
  37
  38   logging.info(
  39       'Registering with discovery server at %s using OTP %s', args.host,
  40       args.otp)
  41   server = common_lib.ConnectToServer(args.host).RegisterClient(
  42       args.otp, common_lib.MY_IP)
  43
  44   server = client_rpc_server.RPCServer(args.host, args.idle_timeout)
  45
  46   server.serve_forever()
  47   return 0
  48
  49
  50 if __name__ == '__main__':
  51   sys.exit(main())