| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import android_commands | 5 import android_commands |
| 6 import constants | 6 import constants |
| 7 import logging | 7 import logging |
| 8 import os | 8 import os |
| 9 import subprocess | 9 import subprocess |
| 10 import time | 10 import time |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 self._fake_dns = None | 25 self._fake_dns = None |
| 26 self._original_dns = None | 26 self._original_dns = None |
| 27 | 27 |
| 28 def _PushAndStartFakeDns(self): | 28 def _PushAndStartFakeDns(self): |
| 29 """Starts the fake_dns server that replies all name queries 127.0.0.1. | 29 """Starts the fake_dns server that replies all name queries 127.0.0.1. |
| 30 | 30 |
| 31 Returns: | 31 Returns: |
| 32 subprocess instance connected to the fake_dns process on the device. | 32 subprocess instance connected to the fake_dns process on the device. |
| 33 """ | 33 """ |
| 34 self._adb.PushIfNeeded( | 34 self._adb.PushIfNeeded( |
| 35 os.path.join(constants.CHROME_DIR, 'out', self._build_type, 'fake_dns'), | 35 os.path.join(constants.DIR_SOURCE_ROOT, 'out', self._build_type, |
| 36 'fake_dns'), |
| 36 FakeDns._FAKE_DNS_PATH) | 37 FakeDns._FAKE_DNS_PATH) |
| 37 return subprocess.Popen( | 38 return subprocess.Popen( |
| 38 ['adb', '-s', self._adb._adb.GetSerialNumber(), | 39 ['adb', '-s', self._adb._adb.GetSerialNumber(), |
| 39 'shell', '%s -D' % FakeDns._FAKE_DNS_PATH]) | 40 'shell', '%s -D' % FakeDns._FAKE_DNS_PATH]) |
| 40 | 41 |
| 41 def SetUp(self): | 42 def SetUp(self): |
| 42 """Configures the system to point to a DNS server that replies 127.0.0.1. | 43 """Configures the system to point to a DNS server that replies 127.0.0.1. |
| 43 | 44 |
| 44 This can be used in combination with the forwarder to forward all web | 45 This can be used in combination with the forwarder to forward all web |
| 45 traffic to a replay server. | 46 traffic to a replay server. |
| 46 | 47 |
| 47 The TearDown() method will perform all cleanup. | 48 The TearDown() method will perform all cleanup. |
| 48 """ | 49 """ |
| 49 self._adb.RunShellCommand('ip route add 8.8.8.0/24 via 127.0.0.1 dev lo') | 50 self._adb.RunShellCommand('ip route add 8.8.8.0/24 via 127.0.0.1 dev lo') |
| 50 self._fake_dns = self._PushAndStartFakeDns() | 51 self._fake_dns = self._PushAndStartFakeDns() |
| 51 self._original_dns = self._adb.RunShellCommand('getprop net.dns1')[0] | 52 self._original_dns = self._adb.RunShellCommand('getprop net.dns1')[0] |
| 52 self._adb.RunShellCommand('setprop net.dns1 127.0.0.1') | 53 self._adb.RunShellCommand('setprop net.dns1 127.0.0.1') |
| 53 time.sleep(2) # Time for server to start and the setprop to take effect. | 54 time.sleep(2) # Time for server to start and the setprop to take effect. |
| 54 | 55 |
| 55 def TearDown(self): | 56 def TearDown(self): |
| 56 """Shuts down the fake_dns.""" | 57 """Shuts down the fake_dns.""" |
| 57 if self._fake_dns: | 58 if self._fake_dns: |
| 58 if not self._original_dns or self._original_dns == '127.0.0.1': | 59 if not self._original_dns or self._original_dns == '127.0.0.1': |
| 59 logging.warning('Bad original DNS, falling back to Google DNS.') | 60 logging.warning('Bad original DNS, falling back to Google DNS.') |
| 60 self._original_dns = '8.8.8.8' | 61 self._original_dns = '8.8.8.8' |
| 61 self._adb.RunShellCommand('setprop net.dns1 %s' % self._original_dns) | 62 self._adb.RunShellCommand('setprop net.dns1 %s' % self._original_dns) |
| 62 self._fake_dns.kill() | 63 self._fake_dns.kill() |
| 63 self._adb.RunShellCommand('ip route del 8.8.8.0/24 via 127.0.0.1 dev lo') | 64 self._adb.RunShellCommand('ip route del 8.8.8.0/24 via 127.0.0.1 dev lo') |
| OLD | NEW |