blob: 686ecfcd959f0edc80cf4a8b25968f1a6b32127d [file] [log] [blame]
# Copyright (c) 2012 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 collections
import ConfigParser
import logging
import os
import time
from autotest_lib.client.common_lib.cros.network import ap_constants
from autotest_lib.server.cros.ap_configurators import ap_spec
TIMEOUT = 100
def get_ap_list():
"""
Returns the list of AP's from the corresponding configuration file.
@param ap_test_type: Used to determine which type of test we're
currently running (Chaos vs Clique).
@returns a list of AP objects.
"""
aps = []
# chaos_shadow_ap_list.conf is used for testing against local APs.
for filename in ('chaos_ap_list.conf', 'chaos_shadow_ap_list.conf'):
ap_config = ConfigParser.RawConfigParser(
{AP.CONF_RPM_MANAGED: 'False'})
path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
filename)
if not os.path.exists(path):
logging.warning('Skipping missing config: "%s"', path)
continue
logging.debug('Reading config from: "%s"', path)
ap_config.read(path)
for bss in ap_config.sections():
aps.append(AP(bss, ap_config))
return aps
class APPowerException(Exception):
""" Exception raised when AP fails to power on. """
pass
class APSectionError(Exception):
""" Exception raised when AP instance does not exist in the config. """
pass
RPMUnit = collections.namedtuple('RPMUnit', 'hostname outlet')
class AP(object):
""" An instance of an ap defined in the chaos config file.
This object is a wrapper that can be used to retrieve information
about an AP in the chaos lab, and control its power.
"""
# Keys used in the config file.
CONF_SSID = 'ssid'
CONF_BRAND = 'brand'
CONF_MODEL = 'model'
CONF_WAN_MAC = 'wan mac'
CONF_WAN_HOST = 'wan_hostname'
CONF_RPM_MANAGED = 'rpm_managed'
CONF_RPM_HOSTNAME = 'rpm_hostname'
CONF_RPM_OUTLET = 'rpm_outlet'
CONF_BSS = 'bss'
CONF_BSS5 = 'bss5'
CONF_BANDWIDTH = 'bandwidth'
CONF_SECURITY = 'security'
CONF_PSK = 'psk'
CONF_FREQUENCY = 'frequency'
CONF_BAND = 'band'
CONF_CHANNEL = 'channel'
CONF_CLASS = 'class_name'
CONF_ADMIN = 'admin_url'
CONF_ADMIN_IP = 'admin_ip'
def __init__(self, bss, config):
"""
Intialize object
@param bss: string containing bssid
@param config: ConfigParser read from file
"""
if not config.has_section(bss):
raise APSectionError('BSS (%s) not defined.' % bss)
self.bss = bss
self.ap_config = config
def get_ssid(self):
"""@return string ssid for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_SSID)
def get_brand(self):
"""@return string brand for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_BRAND)
def get_model(self):
"""@return string model for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_MODEL)
def get_wan_mac(self):
"""@return string mac for WAN port of AP from config file"""
return self.ap_config.get(self.bss, self.CONF_WAN_MAC)
def get_wan_host(self):
"""@return string host for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_WAN_HOST)
def get_rpm_unit(self):
"""@return RPMUnit for this AP. None if AP is not managed via RPM."""
if not self._get_rpm_managed():
return None
return RPMUnit(
self.ap_config.get(self.bss, self.CONF_RPM_HOSTNAME),
self.ap_config.get(self.bss, self.CONF_RPM_OUTLET),
)
def get_bss(self):
"""@return string bss for AP from config file"""
try:
bss = self.ap_config.get(self.bss, self.CONF_BSS)
except ConfigParser.NoOptionError as e:
bss = 'N/A'
return bss
def get_bss5(self):
"""@return string bss5 for AP from config file"""
try:
bss5 = self.ap_config.get(self.bss, self.CONF_BSS5)
except ConfigParser.NoOptionError as e:
bss5 = 'N/A'
return bss5
def get_bandwidth(self):
"""@return string bandwidth for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_BANDWIDTH)
def get_security(self):
"""@return string security for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_SECURITY)
def get_psk(self):
"""@return string psk for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_PSK)
def get_frequency(self):
"""@return int frequency for AP from config file"""
return int(self.ap_config.get(self.bss, self.CONF_FREQUENCY))
def get_channel(self):
"""@return int channel for AP from config file"""
return ap_spec.CHANNEL_TABLE[self.get_frequency()]
def get_band(self):
"""@return string band for AP from config file"""
if self.get_frequency() < 4915:
return ap_spec.BAND_2GHZ
else:
return ap_spec.BAND_5GHZ
def get_class(self):
"""@return string class for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_CLASS)
def get_admin(self):
"""@return string admin for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_ADMIN)
def get_admin_ip(self):
"""@return admin IP for AP from config file"""
return self.ap_config.get(self.bss, self.CONF_ADMIN_IP)
def _get_rpm_managed(self):
return self.ap_config.getboolean(self.bss, self.CONF_RPM_MANAGED)
def __str__(self):
"""@return string description of AP"""
ap_info = {
'brand': self.get_brand(),
'model': self.get_model(),
'ssid' : self.get_ssid(),
'bss' : self.get_bss(),
'hostname': self.get_wan_host(),
}
return ('AP Info:\n'
' Name: %(brand)s %(model)s\n'
' SSID: %(ssid)s\n'
' BSS: %(bss)s\n'
' Hostname: %(hostname)s\n' % ap_info)