| # Copyright 2016 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import os |
| |
| from autotest_lib.server.hosts import ssh_host |
| |
| RA_SCRIPT = 'sendra.py' |
| SCAPY = 'scapy-2.2.0.tar.gz' |
| SCAPY_INSTALL_COMMAND = 'sudo python setup.py install' |
| PROC_NET_SNMP6 = '/proc/net/snmp6' |
| MULTICAST_ADDR = '33:33:00:00:00:01' |
| IFACE = 'managed0' |
| LIFETIME = 180 |
| |
| |
| class IPutils(object): |
| |
| def __init__(self, host): |
| """Initializes an IP utility interface. |
| |
| @param host: Router host object. |
| |
| """ |
| self.host = host |
| self.install_path = self.host.run('mktemp -d').stdout.rstrip() |
| |
| |
| def install_scapy(self): |
| """Installs scapy on the target device. Scapy and all related files and |
| scripts will be installed in a temp directory under /tmp. |
| |
| """ |
| scapy = os.path.join(self.install_path, SCAPY) |
| ap_sshhost = ssh_host.SSHHost(hostname=self.host.hostname) |
| current_dir = os.path.dirname(os.path.realpath(__file__)) |
| send_ra_script = os.path.join(current_dir, RA_SCRIPT) |
| send_scapy = os.path.join(current_dir, SCAPY) |
| ap_sshhost.send_file(send_scapy, self.install_path) |
| ap_sshhost.send_file(send_ra_script, self.install_path) |
| |
| self.host.run('tar -xvf %s -C %s' % (scapy, self.install_path)) |
| self.host.run('cd %s; %s' % (self.install_path, SCAPY_INSTALL_COMMAND)) |
| |
| |
| def cleanup_scapy(self): |
| """Remove all scapy related files and scripts from device. |
| |
| @param host: Router host object. |
| |
| """ |
| self.host.run('rm -rf %s' % self.install_path) |
| |
| |
| def send_ra(self, mac=MULTICAST_ADDR, interval=1, count=None, iface=IFACE, |
| lifetime=LIFETIME): |
| """Invoke scapy and send RA to the device. |
| |
| @param host: Router host object. |
| @param mac: string HWAddr/MAC address to send the packets to. |
| @param interval: int Time to sleep between consecutive packets. |
| @param count: int Number of packets to be sent. |
| @param iface: string of the WiFi interface to use for sending packets. |
| @param lifetime: int original RA's router lifetime in seconds. |
| |
| """ |
| scapy_command = os.path.join(self.install_path, RA_SCRIPT) |
| options = ' -m %s -i %d -c %d -l %d -in %s' %(mac, interval, count, |
| lifetime, iface) |
| self.host.run(scapy_command + options) |
| |
| |
| def get_icmp6intype134(self, host): |
| """Read the value of Icmp6InType134 and return integer. |
| |
| @param host: DUT host object. |
| |
| @returns integer value >0 if grep is successful; 0 otherwise. |
| |
| """ |
| ra_count_str = host.run( |
| 'grep Icmp6InType134 %s || true' % PROC_NET_SNMP6).stdout |
| if ra_count_str: |
| return int(ra_count_str.split()[1]) |
| # If grep failed it means that there is no entry for Icmp6InType134 in file. |
| return 0 |
