OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # | 2 # |
3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | 3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
6 | 6 |
7 """Command line tool for forwarding ports from a device to the host. | 7 """Command line tool for forwarding ports from a device to the host. |
8 | 8 |
9 Allows an Android device to connect to services running on the host machine, | 9 Allows an Android device to connect to services running on the host machine, |
10 i.e., "adb forward" in reverse. Requires |host_forwarder| and |device_forwarder| | 10 i.e., "adb forward" in reverse. Requires |host_forwarder| and |device_forwarder| |
11 to be built. | 11 to be built. |
12 """ | 12 """ |
13 | 13 |
14 import logging | 14 import logging |
15 import optparse | 15 import optparse |
16 import sys | 16 import sys |
17 import time | 17 import time |
18 | 18 |
19 from pylib import constants | 19 from pylib import constants |
20 from pylib import forwarder | 20 from pylib import forwarder |
21 from pylib.device import adb_wrapper | 21 from pylib.device import adb_wrapper |
22 from pylib.device import device_blacklist | |
23 from pylib.device import device_errors | 22 from pylib.device import device_errors |
24 from pylib.device import device_utils | 23 from pylib.device import device_utils |
25 from pylib.utils import run_tests_helper | 24 from pylib.utils import run_tests_helper |
26 | 25 |
27 | 26 |
28 def main(argv): | 27 def main(argv): |
29 parser = optparse.OptionParser(usage='Usage: %prog [options] device_port ' | 28 parser = optparse.OptionParser(usage='Usage: %prog [options] device_port ' |
30 'host_port [device_port_2 host_port_2] ...', | 29 'host_port [device_port_2 host_port_2] ...', |
31 description=__doc__) | 30 description=__doc__) |
32 parser.add_option('-v', | 31 parser.add_option('-v', |
33 '--verbose', | 32 '--verbose', |
34 dest='verbose_count', | 33 dest='verbose_count', |
35 default=0, | 34 default=0, |
36 action='count', | 35 action='count', |
37 help='Verbose level (multiple times for more)') | 36 help='Verbose level (multiple times for more)') |
38 parser.add_option('--device', | 37 parser.add_option('--device', |
39 help='Serial number of device we should use.') | 38 help='Serial number of device we should use.') |
40 parser.add_option('--blacklist-file', help='Device blacklist JSON file.') | |
41 parser.add_option('--debug', action='store_const', const='Debug', | 39 parser.add_option('--debug', action='store_const', const='Debug', |
42 dest='build_type', default='Release', | 40 dest='build_type', default='Release', |
43 help='Use Debug build of host tools instead of Release.') | 41 help='Use Debug build of host tools instead of Release.') |
44 | 42 |
45 options, args = parser.parse_args(argv) | 43 options, args = parser.parse_args(argv) |
46 run_tests_helper.SetLogLevel(options.verbose_count) | 44 run_tests_helper.SetLogLevel(options.verbose_count) |
47 | 45 |
48 if len(args) < 2 or not len(args) % 2: | 46 if len(args) < 2 or not len(args) % 2: |
49 parser.error('Need even number of port pairs') | 47 parser.error('Need even number of port pairs') |
50 sys.exit(1) | 48 sys.exit(1) |
51 | 49 |
52 try: | 50 try: |
53 port_pairs = map(int, args[1:]) | 51 port_pairs = map(int, args[1:]) |
54 port_pairs = zip(port_pairs[::2], port_pairs[1::2]) | 52 port_pairs = zip(port_pairs[::2], port_pairs[1::2]) |
55 except ValueError: | 53 except ValueError: |
56 parser.error('Bad port number') | 54 parser.error('Bad port number') |
57 sys.exit(1) | 55 sys.exit(1) |
58 | 56 |
59 if options.blacklist_file: | 57 devices = device_utils.DeviceUtils.HealthyDevices() |
60 blacklist = device_blacklist.Blacklist(options.blacklist_file) | |
61 else: | |
62 blacklist = None | |
63 | |
64 devices = device_utils.DeviceUtils.HealthyDevices(blacklist) | |
65 | 58 |
66 if options.device: | 59 if options.device: |
67 device = next((d for d in devices if d == options.device), None) | 60 device = next((d for d in devices if d == options.device), None) |
68 if not device: | 61 if not device: |
69 raise device_errors.DeviceUnreachableError(options.device) | 62 raise device_errors.DeviceUnreachableError(options.device) |
70 elif devices: | 63 elif devices: |
71 device = devices[0] | 64 device = devices[0] |
72 logging.info('No device specified. Defaulting to %s', devices[0]) | 65 logging.info('No device specified. Defaulting to %s', devices[0]) |
73 else: | 66 else: |
74 raise device_errors.NoDevicesError() | 67 raise device_errors.NoDevicesError() |
75 | 68 |
76 constants.SetBuildType(options.build_type) | 69 constants.SetBuildType(options.build_type) |
77 try: | 70 try: |
78 forwarder.Forwarder.Map(port_pairs, device) | 71 forwarder.Forwarder.Map(port_pairs, device) |
79 while True: | 72 while True: |
80 time.sleep(60) | 73 time.sleep(60) |
81 except KeyboardInterrupt: | 74 except KeyboardInterrupt: |
82 sys.exit(0) | 75 sys.exit(0) |
83 finally: | 76 finally: |
84 forwarder.Forwarder.UnmapAllDevicePorts(device) | 77 forwarder.Forwarder.UnmapAllDevicePorts(device) |
85 | 78 |
86 if __name__ == '__main__': | 79 if __name__ == '__main__': |
87 main(sys.argv) | 80 main(sys.argv) |
OLD | NEW |