WiFi test cleanups

server/site_bsd_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
 
+def isBSDRouter(router):
+    router_uname = router.run('uname').stdout
+    return re.search('BSD', router_uname)
+
+def find_ifnet(host, pattern):
+    list = host.run("ifconfig -l").stdout
+    for ifnet in list.split():
+        status = host.run("ifconfig %s" % ifnet).stdout
+        m = re.search(pattern, status)
+        if m:
+            return ifnet
+    return None
+
 class NotImplemented(Exception):
     def __init__(self, what):
         self.what = what
     def __str__(self):
         return repr("Test method '%s' not implemented" % self.what)
 
 
 class BSDRouter(object):
     """
     BSD-style WiFi Router support for WiFiTest class.
 
     This class implements test methods/steps that communicate with a
     router implemented with FreeBSD 8.0 and later.  The router must
     be pre-configured to enable ssh access and have a net80211-based
     wireless device.  We also assume hostapd is present for handling
     authenticator duties and any necessary modules are pre-loaded
     (e.g. wlan_ccmp, wlan_tkip, wlan_wep, wlan_xauth).
     """
 
 
     def __init__(self, host, params, defssid):
         self.router = host
-        # TODO(sleffler) default to 1st available wireless nic
-        self.phydev = params['phydev']
-        # TODO(sleffler) default to 1st available wired nic
-        self.wiredif = params['wiredev']
+        # default to 1st available wireless nic
+        if "phydev" not in params:
+            self.phydev = find_ifnet(host, ".*media:.IEEE.802.11.*")
+            if self.phydev is None:
+                raise Exception("No wireless NIC found")
+        else:
+            self.phydev = params['phydev']
+        # default to 1st available wired nic
+        if "wiredev" not in params:
+            self.wiredif = find_ifnet(host, ".*media:.Ethernet.*")
+            if self.wiredif is None:
+                raise Exception("No wired NIC found")
+        else:
+            self.wiredif = params['wiredev']
         self.defssid = defssid;
         self.wlanif = None
         self.bridgeif = None
 
         self.hostapd_keys = ("wpa", "wpa_passphrase", "wpa_key_mgmt",
             "wpa_pairwise", "wpa_group_rekey", "wpa_strict_rekey",
             "wpa_gmk_rekey", "wpa_ptk_rekey",
             "rsn_pairwise",
             "rsn_preauth", "rsh_preauth_interfaces",
             "peerkey")
         self.hostapd_conf = None
 
         # clear any previous state; this is a hack
         self.router.run("ifconfig wlan0 destroy >/dev/null 2>&1",
             ignore_status=True)
         self.router.run("ifconfig bridge0 destroy >/dev/null 2>&1",
             ignore_status=True)
         self.router.run("killall hostapd >/dev/null 2>&1", ignore_status=True)
 
-
     def create(self, params):
         """ Create a wifi device of the specified type """
 
         phydev = params.get('phydev', self.phydev)
         result = self.router.run("ifconfig wlan create wlandev %s" \
             " wlanmode %s" % (phydev, params['type']))
         self.wlanif = result.stdout[:-1]
 
         # NB: can't create+addm together 'cuz of ifconfig bug
         result = self.router.run("ifconfig bridge create")
@@ skipping 124 matching lines @@
         raise NotImplemented("check_client_event_countermeasures")
 
 
     def router_force_mic_error(self, params):
         """
         Force a Michael MIC error on the next packet.  Note this requires
         a driver that uses software crypto and a kernel with the support
         to fire oneshot MIC errors (first appeared in FreeBSD 8.1).
         """
         raise NotImplemented("force_mic_error")