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_errors |
22 from pylib.device import device_utils | 23 from pylib.device import device_utils |
23 from pylib.utils import run_tests_helper | 24 from pylib.utils import run_tests_helper |
24 | 25 |
25 | 26 |
26 def main(argv): | 27 def main(argv): |
27 parser = optparse.OptionParser(usage='Usage: %prog [options] device_port ' | 28 parser = optparse.OptionParser(usage='Usage: %prog [options] device_port ' |
28 'host_port [device_port_2 host_port_2] ...', | 29 'host_port [device_port_2 host_port_2] ...', |
29 description=__doc__) | 30 description=__doc__) |
30 parser.add_option('-v', | 31 parser.add_option('-v', |
31 '--verbose', | 32 '--verbose', |
(...skipping 17 matching lines...) Expand all Loading... |
49 try: | 50 try: |
50 port_pairs = map(int, args[1:]) | 51 port_pairs = map(int, args[1:]) |
51 port_pairs = zip(port_pairs[::2], port_pairs[1::2]) | 52 port_pairs = zip(port_pairs[::2], port_pairs[1::2]) |
52 except ValueError: | 53 except ValueError: |
53 parser.error('Bad port number') | 54 parser.error('Bad port number') |
54 sys.exit(1) | 55 sys.exit(1) |
55 | 56 |
56 devices = device_utils.DeviceUtils.HealthyDevices() | 57 devices = device_utils.DeviceUtils.HealthyDevices() |
57 | 58 |
58 if options.device: | 59 if options.device: |
59 if options.device not in [str(d) for d in devices]: | 60 device = next((d for d in devices if d == options.device), None) |
60 raise Exception('Error: %s not in attached devices %s' % (options.device, | 61 if not device: |
61 ','.join(devices))) | 62 raise device_errors.DeviceUnreachableError(options.device) |
62 devices = [options.device] | 63 elif devices: |
| 64 device = devices[0] |
| 65 logging.info('No device specified. Defaulting to %s', devices[0]) |
63 else: | 66 else: |
64 if not devices: | 67 raise device_errors.NoDevicesError() |
65 raise Exception('Error: no connected devices') | |
66 logging.info('No device specified. Defaulting to %s', devices[0]) | |
67 | 68 |
68 device = devices[0] | |
69 constants.SetBuildType(options.build_type) | 69 constants.SetBuildType(options.build_type) |
70 try: | 70 try: |
71 forwarder.Forwarder.Map(port_pairs, device) | 71 forwarder.Forwarder.Map(port_pairs, device) |
72 while True: | 72 while True: |
73 time.sleep(60) | 73 time.sleep(60) |
74 except KeyboardInterrupt: | 74 except KeyboardInterrupt: |
75 sys.exit(0) | 75 sys.exit(0) |
76 finally: | 76 finally: |
77 forwarder.Forwarder.UnmapAllDevicePorts(device) | 77 forwarder.Forwarder.UnmapAllDevicePorts(device) |
78 | 78 |
79 if __name__ == '__main__': | 79 if __name__ == '__main__': |
80 main(sys.argv) | 80 main(sys.argv) |
OLD | NEW |