remoting/tools/native_messaging_host_test.py

   1 #!/usr/bin/env python
   2 # Copyright 2013 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 # Interactive test script for the Chromoting host native messaging component.
   7
   8 import json
   9 import readline
  10 import struct
  11 import subprocess
  12 import sys
  13
  14
  15 def PrintMenuAndGetBuilder(messages):
  16   print
  17   for i in range(0, len(messages)):
  18     print '%d: %s' % (i + 1, messages[i][0])
  19   print 'Q: Quit'
  20   while True:
  21     choice = raw_input('Enter choice: ')
  22     if choice.lower() == 'q':
  23       return None
  24     choice = int(choice)
  25     if choice >= 1 and choice <= len(messages):
  26       return messages[choice - 1][1]
  27
  28
  29 # Message builder methods.
  30 def BuildHello():
  31   return {'type': 'hello'}
  32
  33
  34 def BuildClearAllPairedClients():
  35   return {'type': 'clearPairedClients'}
  36
  37
  38 def BuildDeletePairedClient():
  39   client_id = raw_input('Enter client id: ')
  40   return {'type': 'deletePairedClient',
  41           'clientId': client_id}
  42
  43
  44 def BuildGetHostName():
  45   return {'type': 'getHostName'}
  46
  47
  48 def BuildGetPinHash():
  49   host_id = raw_input('Enter host id: ')
  50   pin = raw_input('Enter PIN: ')
  51   return {'type': 'getPinHash',
  52           'hostId': host_id,
  53           'pin': pin}
  54
  55
  56 def BuildGenerateKeyPair():
  57   return {'type': 'generateKeyPair'}
  58
  59
  60 def BuildUpdateDaemonConfig():
  61   config_json = raw_input('Enter config JSON: ')
  62   return {'type': 'updateDaemonConfig',
  63           'config': config_json}
  64
  65
  66 def BuildGetDaemonConfig():
  67   return {'type': 'getDaemonConfig'}
  68
  69
  70 def BuildGetPairedClients():
  71   return {'type': 'getPairedClients'}
  72
  73
  74 def BuildGetUsageStatsConsent():
  75   return {'type': 'getUsageStatsConsent'}
  76
  77
  78 def BuildStartDaemon():
  79   while True:
  80     consent = raw_input('Report usage stats [y/n]? ')
  81     if consent.lower() == 'y':
  82       consent = True
  83     elif consent.lower() == 'n':
  84       consent = False
  85     else:
  86       continue
  87     break
  88   config_json = raw_input('Enter config JSON: ')
  89   return {'type': 'startDaemon',
  90           'consent': consent,
  91           'config': config_json}
  92
  93
  94 def BuildStopDaemon():
  95   return {'type': 'stopDaemon'}
  96
  97
  98 def BuildGetDaemonState():
  99   return {'type': 'getDaemonState'}
 100
 101
 102 def main():
 103   if len(sys.argv) != 2:
 104     print 'Usage: ' + sys.argv[0] + ' <path to native messaging host>'
 105     sys.exit(1)
 106
 107   native_messaging_host = sys.argv[1]
 108
 109   child = subprocess.Popen(native_messaging_host, stdin=subprocess.PIPE,
 110                            stdout=subprocess.PIPE, close_fds=True)
 111
 112   message_id = 0
 113   while True:
 114     messages = [
 115       ('Hello', BuildHello),
 116       ('Clear all paired clients', BuildClearAllPairedClients),
 117       ('Delete paired client', BuildDeletePairedClient),
 118       ('Get host name', BuildGetHostName),
 119       ('Get PIN hash', BuildGetPinHash),
 120       ('Generate key pair', BuildGenerateKeyPair),
 121       ('Update daemon config', BuildUpdateDaemonConfig),
 122       ('Get daemon config', BuildGetDaemonConfig),
 123       ('Get paired clients', BuildGetPairedClients),
 124       ('Get usage stats consent', BuildGetUsageStatsConsent),
 125       ('Start daemon', BuildStartDaemon),
 126       ('Stop daemon', BuildStopDaemon),
 127       ('Get daemon state', BuildGetDaemonState)
 128     ]
 129     builder = PrintMenuAndGetBuilder(messages)
 130     if not builder:
 131       break
 132     message_dict = builder()
 133     message_dict['id'] = message_id
 134     message = json.dumps(message_dict)
 135     message_id += 1
 136     print 'Message: ' + message
 137     child.stdin.write(struct.pack('I', len(message)))
 138     child.stdin.write(message)
 139     child.stdin.flush()
 140
 141     reply_length_bytes = child.stdout.read(4)
 142     if len(reply_length_bytes) < 4:
 143       print 'Invalid message length'
 144       break
 145     reply_length = struct.unpack('i', reply_length_bytes)[0]
 146     reply = child.stdout.read(reply_length).decode('utf-8')
 147     print 'Reply: ' + reply
 148     if len(reply) != reply_length:
 149       print 'Invalid reply length'
 150       break
 151
 152 if __name__ == '__main__':
 153   main()