Correctly auto-detect 2GHz and 5GHz phy devices

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 logging, re
+from autotest_lib.client.common_lib import error
 
 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
     router implemented with Linux/mac80211.  The router must
     be pre-configured to enable ssh access and have a mac80211-based
     wireless device.  We also assume hostapd 0.7.x and iw are present
     and any necessary modules are pre-loaded.
     """
 
 
     def __init__(self, host, params, defssid):
         # Command locations.
         self.cmd_iw = "/usr/sbin/iw"
         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.wiredif = params.get('wiredev', "eth0")
         self.wlanif2 = "wlan2"
         self.wlanif5 = "wlan5"
 
-        # Default to 1st available wireless phy.
+        # Parse the output of 'iw phy' and find a device for each frequency
         if "phydev2" not in params:
             output = self.router.run("%s list" % self.cmd_iw).stdout
-            test = re.compile("Wiphy (.*)")
+            re_wiphy = re.compile("Wiphy (.*)")
+            re_mhz = re.compile("(\d+) MHz")
+            in_phy = False
+            self.phydev2 = None
+            self.phydev5 = None
             for line in output.splitlines():
-                m = test.match(line)
-                if m:
-                    self.phydev2 = m.group(1)
-                    self.phydev5 = self.phydev2
-                    break
+                match_wiphy = re_wiphy.match(line)
+                if match_wiphy:
+                    in_phy = True
+                    widevname = match_wiphy.group(1)
+                elif in_phy:
+                    if line[0] == '\t':
+                        match_mhz = re_mhz.search(line)
+                        if match_mhz:
+                            mhz = int(match_mhz.group(1))
+                            if self.phydev2 is None and \
+                                    mhz in range(2402,2472,5):
+                                self.phydev2 = widevname
+                            elif self.phydev5 is None and \
+                                    mhz in range(5100,6000,20):
+                                self.phydev5 = widevname
+                            if None not in (self.phydev2, self.phydev5):
+                                break
+                    else:
+                        in_phy = False
             else:
                 raise error.TestFail("No Wireless NIC detected on the device")
         else:
             self.phydev2 = params['phydev2']
             self.phydev5 = params.get('phydev5', self.phydev2)
 
 
         # hostapd configuration persists throughout the test, subsequent
         # 'config' commands only modify it.
         self.defssid = defssid
@@ skipping 234 matching lines @@
         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
 
 
     def get_ssid(self):
         return self.hostapd['conf']['ssid']