Added support for per-band phys.

server/site_linux_router.py

 # Copyright (c) 2010 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
+import logging, re
 
 def isLinuxRouter(router):
     router_uname = router.run('uname').stdout
     return re.search('Linux', router_uname)
 
 class LinuxRouter(object):
     """
     Linux/mac80211-style WiFi Router support for WiFiTest class.
 
     This class implements test methods/steps that communicate with a
@@ skipping 10 matching lines @@
         self.cmd_ip = "/usr/sbin/ip"
         self.cmd_brctl = "/usr/sbin/brctl"
         self.cmd_hostapd = "/usr/sbin/hostapd"
 
         # Router host.
         self.router = host
 
         # Network interfaces.
         self.bridgeif = params.get('bridgedev', "br-lan")
         self.wiredif = params.get('wiredev', "eth1")
-        self.wlanif = "wlan0"
+        self.wlanif2 = "wlan2"
+        self.wlanif5 = "wlan5"
 
         # Default to 1st available wireless phy.
-        if "phydev" not in params:
+        if "phydev2" not in params:
             output = self.router.run("%s list" % self.cmd_iw).stdout
             test = re.compile("Wiphy (.*)")
             for line in output.splitlines():
                 m = test.match(line)
                 if m:
-                    self.phydev = m.group(1)
+                    self.phydev2 = m.group(1)
+                    self.phydev5 = self.phydev2
                     break
             else:
                 raise error.TestFail("No Wireless NIC detected on the device")
         else:
-            self.phydev = params['phydev']
+            self.phydev2 = params['phydev2']
+            self.phydev5 = params.get('phydev5', self.phydev2)
 
 
         # hostapd configuration persists throughout the test, subsequent
         # 'config' commands only modify it.
         self.hostapd = {
             'configured': False,
-            'file': "/tmp/%s.conf" % self.phydev,
+            'file': "/tmp/hostapd-test.conf",
             'driver': "nl80211",
             'conf': {
                 'ssid': defssid,
-                'interface': self.wlanif,
                 'bridge': self.bridgeif,
                 'hw_mode': 'g'
             }
         }
 
         # Kill hostapd if already running.
         self.router.run("pkill hostapd >/dev/null 2>&1", ignore_status=True)
 
         # Remove all bridges.
         output = self.router.run("%s show" % self.cmd_brctl).stdout
@@ skipping 31 matching lines @@
         phytype = {
             "sta"       : "managed",
             "monitor"   : "monitor",
             "adhoc"     : "adhoc",
             "ibss"      : "ibss",
             "ap"        : "managed",     # NB: handled by hostapd
             "hostap"    : "managed",     # NB: handled by hostapd
             "mesh"      : "mesh",
             "wds"       : "wds",
         }[params['type']]
-        phydev = params.get('phydev', self.phydev)
+
         self.router.run("%s phy %s interface add %s type %s" %
-            (self.cmd_iw, phydev, self.wlanif, phytype))
-
+            (self.cmd_iw, self.phydev2, self.wlanif2, phytype))
+        self.router.run("%s phy %s interface add %s type %s" %
+            (self.cmd_iw, self.phydev5, self.wlanif5, phytype))
 
 
     def destroy(self, params):
         """ Destroy a previously created device """
         # For linux, this is the same as deconfig.
         self.deconfig(params)
 
 
 
     def config(self, params):
@@ skipping 103 matching lines @@
                 elif k == '-ampdu':
                     pass        # TODO(sleffler) need hostapd support
                 else:
                     conf[k] = v
 
             # Aggregate ht_capab.
             if htcaps:
                 conf['ieee80211n'] = 1
                 conf['ht_capab'] = ''.join(htcaps)
 
+            # Figure out the correct interface.
+            if conf.get('hw_mode', 'b') == 'a':
+                conf['interface'] = self.wlanif5
+            else:
+                conf['interface'] = self.wlanif2
+
             # Generate hostapd.conf.
             self.router.run("cat <<EOF >%s\n%s\nEOF\n" %
                 (self.hostapd['file'], '\n'.join(
                 "%s=%s" % kv for kv in conf.iteritems())))
 
             # Run hostapd.
+            logging.info("Starting hostapd...")
             self.router.run("%s -B %s" %
                 (self.cmd_hostapd, self.hostapd['file']))
 
+
             # Set up the bridge.
+            logging.info("Setting up the bridge...")
             self.router.run("%s setfd %s %d" %
                 (self.cmd_brctl, self.bridgeif, 0))
             self.router.run("%s addif %s %s" %
                 (self.cmd_brctl, self.bridgeif, self.wiredif))
             self.router.run("%s link set %s up" %
                 (self.cmd_ip, self.wiredif))
             self.router.run("%s link set %s up" %
                 (self.cmd_ip, self.bridgeif))
 
+            logging.info("AP configured.")
+
 #        else:
 #            # use iw to manually configure interface
 
         self.hostapd['configured'] = True
 
 
     def deconfig(self, params):
         """ De-configure the AP (will also bring wlan and the bridge down) """
 
         if not self.hostapd['configured']:
             return
 
         # Taking down hostapd takes wlan0 and mon.wlan0 down.
         self.router.run("pkill hostapd >/dev/null 2>&1", ignore_status=True)
 #        self.router.run("rm -f %s" % self.hostapd['file'])
 
         # Tear down the bridge.
         self.router.run("%s link set %s down" % (self.cmd_ip, self.bridgeif),
             ignore_status=True)
         self.router.run("%s delbr %s" % (self.cmd_brctl, self.bridgeif),
             ignore_status=True)
 
         self.hostapd['configured'] = False