jeltz/hostap
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

tests: Add helper functions for connection to an AP

Browse source 
Signed-hostap: Jouni Malinen <j@w1.fi>
This commit is contained in:
Jouni Malinen 2013-03-29 20:41:33 +02:00
parent e259d186cf
commit b61e418c35
2 changed files with 29 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
tests/hwsim/test_ap_tdls.py
View file

@ -22,27 +22,16 @@ def start_ap_wpa2_psk(ifname):
params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678") params = hostapd.wpa2_params(ssid="test-wpa2-psk", passphrase="12345678")
hostapd.add_ap(ifname, params) hostapd.add_ap(ifname, params)
def connect_sta(sta, ssid, psk=None, proto=None, key_mgmt=None, wep_key0=None): def connectivity(dev, ap_ifname):
logger.info("Connect STA " + sta.ifname + " to AP")
id = sta.add_network()
sta.set_network_quoted(id, "ssid", ssid)
if psk:
sta.set_network_quoted(id, "psk", psk)
if proto:
sta.set_network(id, "proto", proto)
if key_mgmt:
sta.set_network(id, "key_mgmt", key_mgmt)
if wep_key0:
sta.set_network(id, "wep_key0", wep_key0)
sta.connect_network(id)
def connect_2sta(dev, ssid):
connect_sta(dev[0], ssid, psk="12345678")
connect_sta(dev[1], ssid, psk="12345678")
hwsim_utils.test_connectivity_sta(dev[0], dev[1]) hwsim_utils.test_connectivity_sta(dev[0], dev[1])
hwsim_utils.test_connectivity(dev[0].ifname, "wlan2") hwsim_utils.test_connectivity(dev[0].ifname, "wlan2")
hwsim_utils.test_connectivity(dev[1].ifname, "wlan2") hwsim_utils.test_connectivity(dev[1].ifname, "wlan2")
def connect_2sta(dev, ssid):
dev[0].connect(ssid, psk="12345678")
dev[1].connect(ssid, psk="12345678")
connectivity(dev, "wlan2")
def connect_2sta_wpa2_psk(dev): def connect_2sta_wpa2_psk(dev):
connect_2sta(dev, "test-wpa2-psk") connect_2sta(dev, "test-wpa2-psk")
@ -50,25 +39,19 @@ def connect_2sta_wpa_psk(dev):
connect_2sta(dev, "test-wpa-psk") connect_2sta(dev, "test-wpa-psk")
def connect_2sta_wpa_psk_mixed(dev): def connect_2sta_wpa_psk_mixed(dev):
connect_sta(dev[0], "test-wpa-mixed-psk", psk="12345678", proto="WPA") dev[0].connect("test-wpa-mixed-psk", psk="12345678", proto="WPA")
connect_sta(dev[1], "test-wpa-mixed-psk", psk="12345678", proto="WPA2") dev[1].connect("test-wpa-mixed-psk", psk="12345678", proto="WPA2")
hwsim_utils.test_connectivity_sta(dev[0], dev[1]) connectivity(dev, "wlan2")
hwsim_utils.test_connectivity(dev[0].ifname, "wlan2")
hwsim_utils.test_connectivity(dev[1].ifname, "wlan2")
def connect_2sta_wep(dev): def connect_2sta_wep(dev):
connect_sta(dev[0], "test-wep", key_mgmt="NONE", wep_key0='"hello"') dev[0].connect("test-wep", key_mgmt="NONE", wep_key0='"hello"')
connect_sta(dev[1], "test-wep", key_mgmt="NONE", wep_key0='"hello"') dev[1].connect("test-wep", key_mgmt="NONE", wep_key0='"hello"')
hwsim_utils.test_connectivity_sta(dev[0], dev[1]) connectivity(dev, "wlan2")
hwsim_utils.test_connectivity(dev[0].ifname, "wlan2")
hwsim_utils.test_connectivity(dev[1].ifname, "wlan2")
def connect_2sta_open(dev): def connect_2sta_open(dev):
connect_sta(dev[0], "test-open", key_mgmt="NONE") dev[0].connect("test-open", key_mgmt="NONE")
connect_sta(dev[1], "test-open", key_mgmt="NONE") dev[1].connect("test-open", key_mgmt="NONE")
hwsim_utils.test_connectivity_sta(dev[0], dev[1]) connectivity(dev, "wlan2")
hwsim_utils.test_connectivity(dev[0].ifname, "wlan2")
hwsim_utils.test_connectivity(dev[1].ifname, "wlan2")
def wlantest_tdls(field, bssid, addr1, addr2): def wlantest_tdls(field, bssid, addr1, addr2):
res = subprocess.check_output(["../../wlantest/wlantest_cli", res = subprocess.check_output(["../../wlantest/wlantest_cli",

14
tests/hwsim/wpasupplicant.py
View file

@ -301,3 +301,17 @@ class WpaSupplicant:
if "FAIL" in self.group_request(cmd): if "FAIL" in self.group_request(cmd):
raise Exception("Failed to request TDLS teardown") raise Exception("Failed to request TDLS teardown")
return None return None
def connect(self, ssid, psk=None, proto=None, key_mgmt=None, wep_key0=None):
logger.info("Connect STA " + self.ifname + " to AP")
id = self.add_network()
self.set_network_quoted(id, "ssid", ssid)
if psk:
self.set_network_quoted(id, "psk", psk)
if proto:
self.set_network(id, "proto", proto)
if key_mgmt:
self.set_network(id, "key_mgmt", key_mgmt)
if wep_key0:
self.set_network(id, "wep_key0", wep_key0)
self.connect_network(id)