blob: 0c952b59e55e4cc7f16914b52d4a19dec1ce7f1c [file] [log] [blame] [edit]
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Base class for objects to configure Linksys dual band access points
using webdriver."""
import logging
import urlparse
import dynamic_ap_configurator
import ap_spec
class LinksyseDualBandAPConfigurator(
dynamic_ap_configurator.DynamicAPConfigurator):
"""Base class for objects to configure Linksys dual band access points
using webdriver."""
def _alert_handler(self, alert):
"""Checks for any modal dialogs which popup to alert the user and
either raises a RuntimeError or ignores the alert.
Args:
alert: The modal dialog's contents.
"""
text = alert.text
# We ignore warnings that we get when we disable visibility or security
# changed to WEP, WPA Personal or WPA Enterprise.
if 'Security Mode is disabled.' in text:
alert.accept()
elif 'Your wireless security mode is not compatible with' in text:
alert.accept()
elif 'WARNING: Your Wireless-N devices will only operate ' in text:
alert.accept()
elif 'Setting the Security Mode to WEP, WPA Personal' in text:
alert.accept()
elif 'Turning off SSID Broadcast' in text:
alert.accept()
elif 'Security modes are not compatible' in text:
alert.accept()
raise RuntimeError('Security modes are not compatible. %s' % text)
elif 'security mode is not compatible with Wireless-N' in text:
alert.accept()
elif 'Please select WPA2-Personal or WPA2-Enterprise' in text:
alert.accept()
elif 'The wifi interface is current busy.' in text:
alert.accept()
self.click_button_by_xpath('//a[text()="Save Settings"]',
alert_handler=self._alert_handler)
else:
alert.accept()
raise RuntimeError('We have an unhandled alert: %s' % text)
def get_number_of_pages(self):
return 1
def get_supported_bands(self):
return [{'band': ap_spec.BAND_2GHZ,
'channels': ['Auto', 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]},
{'band': ap_spec.BAND_5GHZ,
'channels': ['Auto', 36, 40, 44, 48, 149, 153, 157, 161]}]
def get_supported_modes(self):
return [{'band': ap_spec.BAND_2GHZ,
'modes': [ap_spec.MODE_B, ap_spec.MODE_N, ap_spec.MODE_B |
ap_spec.MODE_G, ap_spec.MODE_G]},
{'band': ap_spec.BAND_5GHZ,
'modes': [ap_spec.MODE_A, ap_spec.MODE_N]}]
def is_security_mode_supported(self, security_mode):
if self.current_band == ap_spec.BAND_5GHZ:
return security_mode in (ap_spec.SECURITY_TYPE_DISABLED,
ap_spec.SECURITY_TYPE_WPAPSK,
ap_spec.SECURITY_TYPE_WPA2PSK)
return security_mode in (ap_spec.SECURITY_TYPE_DISABLED,
ap_spec.SECURITY_TYPE_WPAPSK,
ap_spec.SECURITY_TYPE_WPA2PSK,
ap_spec.SECURITY_TYPE_WEP)
def navigate_to_page(self, page_number):
if page_number == 1:
url = urlparse.urljoin(self.admin_interface_url,
'Wireless_Basic.asp')
self.get_url(url, page_title='Settings')
else:
raise RuntimeError('Invalid page number passed. Number of pages '
'%d, page value sent was %d' %
(self.get_number_of_pages(), page_number))
def save_page(self, page_number):
self.click_button_by_xpath('//a[text()="Save Settings"]',
alert_handler=self._alert_handler)
continue_xpath = '//input[@value="Continue" and @type="button"]'
self.wait_for_object_by_xpath(continue_xpath)
self.click_button_by_xpath(continue_xpath)
def set_mode(self, mode, band=None):
self.add_item_to_command_list(self._set_mode, (mode, band), 1, 800)
def _set_mode(self, mode, band=None):
mode_mapping = {ap_spec.MODE_B: 'Wireless-B Only',
ap_spec.MODE_G: 'Wireless-G Only',
ap_spec.MODE_B | ap_spec.MODE_G: 'Wireless-B/G Only',
ap_spec.MODE_N: 'Wireless-N Only',
ap_spec.MODE_A: 'Wireless-A Only'}
xpath = '//select[@name="wl0_net_mode"]'
if self.current_band == ap_spec.BAND_5GHZ or band == ap_spec.BAND_5GHZ:
self.current_band = ap_spec.BAND_5GHZ
xpath = '//select[@name="wl1_net_mode"]'
mode_name = ''
if mode in mode_mapping.keys():
mode_name = mode_mapping[mode]
if ((mode & ap_spec.MODE_A) and
(self.current_band != ap_spec.BAND_5GHZ)):
# a mode only in 5Ghz
logging.debug('Mode \'a\' is not available for 2.4Ghz band.')
return
elif ((mode & (ap_spec.MODE_B | ap_spec.MODE_G) ==
(ap_spec.MODE_B | ap_spec.MODE_G)) or
(mode & ap_spec.MODE_B == ap_spec.MODE_B) or
(mode & ap_spec.MODE_G == ap_spec.MODE_G)) and \
(self.current_band != ap_spec.BAND_2GHZ):
# b/g, b, g mode only in 2.4Ghz
logging.debug('Mode \'%s\' is not available for 5Ghz band.',
mode_name)
return
else:
raise RuntimeError('The mode selected %d is not supported by router'
' %s.', hex(mode), self.name)
self.select_item_from_popup_by_xpath(mode_name, xpath,
alert_handler=self._alert_handler)
def set_radio(self, enabled=True):
logging.debug('set_radio is not supported in Linksys dual band AP.')
return None
def set_ssid(self, ssid):
self.add_item_to_command_list(self._set_ssid, (ssid,), 1, 900)
def _set_ssid(self, ssid):
xpath = '//input[@maxlength="32" and @name="wl0_ssid"]'
if self.current_band == ap_spec.BAND_5GHZ:
xpath = '//input[@maxlength="32" and @name="wl1_ssid"]'
self.set_content_of_text_field_by_xpath(ssid, xpath)
self._ssid = ssid
def set_channel(self, channel):
self.add_item_to_command_list(self._set_channel, (channel,), 1, 900)
def _set_channel(self, channel):
position = self._get_channel_popup_position(channel)
channel_choices = ['Auto',
'1 - 2.412 GHz', '2 - 2.417 GHz', '3 - 2.422 GHz',
'4 - 2.427 GHz', '5 - 2.432 GHz', '6 - 2.437 GHz',
'7 - 2.442 GHz', '8 - 2.447 GHz', '9 - 2.452 GHz',
'10 - 2.457 GHz', '11 - 2.462 GHz']
xpath = '//select[@name="_wl0_channel"]'
if self.current_band == ap_spec.BAND_5GHZ:
xpath = '//select[@name="_wl1_channel"]'
channel_choices = ['Auto', '36 - 5.180 GHz', '40 - 5.200 GHz',
'44 - 5.220 GHz', '48 - 5.240 GHz',
'149 - 5.745 GHz', '153 - 5.765 GHz',
'157 - 5.785 GHz', '161 - 5.805 GHz']
self.select_item_from_popup_by_xpath(channel_choices[position],
xpath)
def set_band(self, band):
if band == ap_spec.BAND_5GHZ:
self.current_band = ap_spec.BAND_5GHZ
elif band == ap_spec.BAND_2GHZ:
self.current_band = ap_spec.BAND_2GHZ
else:
raise RuntimeError('Invalid band sent %s' % band)
def set_security_disabled(self):
self.add_item_to_command_list(self._set_security_disabled, (), 1, 900)
def _set_security_disabled(self):
xpath = '//select[@name="wl0_security_mode"]'
if self.current_band == ap_spec.BAND_5GHZ:
xpath = '//select[@name="wl1_security_mode"]'
self.select_item_from_popup_by_xpath('Disabled', xpath)
def set_security_wep(self, key_value, authentication):
self.add_item_to_command_list(self._set_security_wep,
(key_value, authentication), 1, 900)
def _set_security_wep(self, key_value, authentication):
popup = '//select[@name="wl0_security_mode"]'
text_field = '//input[@name="wl0_passphrase"]'
xpath = '//input[@name="wepGenerate0" and @type="button"]'
if self.current_band == ap_spec.BAND_5GHZ:
popup = '//select[@name="wl1_security_mode"]'
text_field = '//input[@name="wl1_passphrase"]'
xpath = '//input[@name="wepGenerate1" and @type="button"]'
self.wait_for_object_by_xpath(popup)
self.select_item_from_popup_by_xpath('WEP', popup,
wait_for_xpath=text_field,
alert_handler=self._alert_handler)
self.set_content_of_text_field_by_xpath(key_value, text_field,
abort_check=True)
self.click_button_by_xpath(xpath)
def set_security_wpapsk(self, security, shared_key, update_interval=None):
self.add_item_to_command_list(self._set_security_wpapsk,
(security, shared_key, update_interval),
1, 900)
def _set_security_wpapsk(self, security, shared_key, update_interval=None):
popup = '//select[@name="wl0_security_mode"]'
key_field = '//input[@name="wl0_wpa_psk"]'
if self.current_band == ap_spec.BAND_5GHZ:
popup = '//select[@name="wl1_security_mode"]'
key_field = '//input[@name="wl1_wpa_psk"]'
self.wait_for_object_by_xpath(popup)
if security == ap_spec.SECURITY_TYPE_WPAPSK:
wpa_item = 'WPA Personal'
else:
wpa_item = 'WPA2 Personal'
self.select_item_from_popup_by_xpath(wpa_item, popup,
wait_for_xpath=key_field,
alert_handler=self._alert_handler)
self.set_content_of_text_field_by_xpath(shared_key, key_field,
abort_check=True)
def is_update_interval_supported(self):
"""
Returns True if setting the PSK refresh interval is supported.
@return True is supported; False otherwise
"""
return False
def set_visibility(self, visible=True):
self.add_item_to_command_list(self._set_visibility, (visible,), 1, 900)
def _set_visibility(self, visible=True):
button = 'wl0_closed'
if self.current_band == ap_spec.BAND_5GHZ:
button = 'wl1_closed'
int_value = 0 if visible else 1
xpath = ('//input[@value="%d" and @name="%s"]' % (int_value, button))
self.click_button_by_xpath(xpath)