blob: a43806eecdd7814932cad6d627431533419a8f3f [file] [log] [blame]
# Copyright (c) 2011 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 re, logging
from autotest_lib.client.common_lib import error
class LinuxSystem(object):
""" This is a superclass for test machines running Linux. It provides
a common point for routines that use the cfg80211 users space tools
to manipulate the wireless stack, regardless of the role they play.
Currently the commands shared are the init, which queries for wireless
devices, along with start_capture and stop_capture. More commands
may migrate from site_linux_router as appropriate to share.
def __init__(self, host, params, role):
# Command locations.
self.cmd_iw = params.get("cmd_iw", "/usr/sbin/iw")
self.cmd_ip = params.get("cmd_ip", "/usr/sbin/ip")
self.cmd_dhcpd = params.get("cmd_dhcpd", "/usr/sbin/dhcpd")
self.cmd_tcpdump = params.get("cmd_tcpdump", "/usr/sbin/tcpdump")
self.capture_file = "/tmp/dump.pcap"
self.capture_logfile = "/tmp/dump.log"
self.capture_count = 0
self.capture_running = False
self.capture_channel = None = host
self.role = role
setattr(self, '%s_start_capture' % role, self.start_capture)
setattr(self, '%s_stop_capture' % role, self.stop_capture)
# Network interfaces.
self.phy_for_frequency = {}
# Parse the output of 'iw phy' and find a device for each frequency
output ="%s list" % self.cmd_iw).stdout
re_wiphy = re.compile("Wiphy (.*)")
re_mhz = re.compile("(\d+) MHz")
in_phy = False
for line in output.splitlines():
match_wiphy = re_wiphy.match(line)
if match_wiphy:
in_phy = True
wiphyname =
elif in_phy:
if line[0] == '\t':
match_mhz =
if match_mhz:
mhz = int(
self.phy_for_frequency[mhz] = wiphyname
in_phy = False
self.phydev2 = params.get('phydev2', None)
self.phydev5 = params.get('phydev5', None)
def _remove_interfaces(self):
self.wlanifs = {}
# Remove all wifi devices.
output ="%s dev" % self.cmd_iw).stdout
test = re.compile("[\s]*Interface (.*)")
for line in output.splitlines():
m = test.match(line)
if m:
device ="%s link set %s down" % (self.cmd_ip, device))"%s dev %s del" % (self.cmd_iw, device))
def start_capture(self, params):
if self.capture_running:
if 'channel' in params:
channel = int(params['channel'])
channel_args = "%s" % channel
for arg in ('ht20', 'ht40+', 'ht40-'):
if arg in params:
channel_args = "%s %s" % (channel_args, arg.upper())
self.capture_channel = channel
self.channel_args = channel_args
channel = self.capture_channel
channel_args = self.channel_args
self.capture_interface = self._get_wlanif(channel, "monitor")"%s dev %s set freq %s" %
(self.cmd_iw, self.capture_interface, channel_args))"%s link set %s up" % (self.cmd_ip,
self.capture_interface))"%s -i %s -w %s -s %s >%s 2>&1 &" %
params.get('snaplen', '152'),
self.capture_running = True
def stop_capture(self, params):
if not self.capture_running:
return"pkill -INT tcpdump >/dev/null 2>&1", ignore_status=True)"%s link set %s down" % (self.cmd_ip,
'debug/%s_%d.pcap' %
(self.role, self.capture_count))
self.capture_count += 1
self.capture_running = False
def _get_wlanif(self, frequency, phytype, mode = None):
""" This function is used by inherited classes, so we use the
single '_' convention rather than the '__' we usually use for
non-scriptable commands, since these cannot be inherited by
We still support the old "phydevN" parameters, but
this code is smart enough to do without it.
if mode in ('b', 'g') and self.phydev2 is not None:
phy = self.phydev2
elif mode == 'a' and self.phydev5 is not None:
phy = self.phydev5
elif frequency in self.phy_for_frequency:
phy = self.phy_for_frequency[frequency]
raise error.TestFail("Unable to find phy for frequency %d mode %s" %
(frequency, mode))
# If self.wlanifs is not defined, this is the first time we've
# allocated a wlan interface. Perform init by calling
# remove_interfaces().
if not hasattr(self, 'wlanifs'):
if phytype not in self.wlanifs:
self.wlanifs[phytype] = {}
elif phy in self.wlanifs[phytype]:
return self.wlanifs[phytype][phy]
wlanif = "%s%d" % (phytype, len(self.wlanifs[phytype].keys()))
self.wlanifs[phytype][phy] = wlanif"%s phy %s interface add %s type %s" %
(self.cmd_iw, phy, wlanif, phytype))
return wlanif