tests: Add helper functions for Hotspot 2.0 test cases

These make it easier to add new test cases for Interworking network
selection.

Signed-hostap: Jouni Malinen <jouni@qca.qualcomm.com>
This commit is contained in:
Jouni Malinen 2013-10-29 14:20:29 +02:00 committed by Jouni Malinen
parent f4defd91cd
commit bbe867672e
2 changed files with 87 additions and 92 deletions

View file

@ -44,109 +44,50 @@ def hs20_ap_params():
params['anqp_3gpp_cell_net'] = "244,91"
return params
def test_ap_hs20_select(dev, apdev):
"""Hotspot 2.0 network selection"""
bssid = apdev[0]['bssid']
params = hs20_ap_params()
params['hessid'] = bssid
hostapd.add_ap(apdev[0]['ifname'], params)
dev[0].request("SET interworking 1")
dev[0].request("SET hs20 1")
id = dev[0].add_cred()
dev[0].set_cred_quoted(id, "realm", "example.com");
dev[0].set_cred_quoted(id, "username", "test");
dev[0].set_cred_quoted(id, "password", "secret");
dev[0].set_cred_quoted(id, "domain", "example.com");
dev[0].dump_monitor()
dev[0].request("INTERWORKING_SELECT")
ev = dev[0].wait_event(["INTERWORKING-AP", "INTERWORKING-NO-MATCH"],
timeout=15)
def interworking_select(dev, bssid, type=None, no_match=False):
dev.dump_monitor()
dev.request("INTERWORKING_SELECT")
ev = dev.wait_event(["INTERWORKING-AP", "INTERWORKING-NO-MATCH"],
timeout=15)
if ev is None:
raise Exception("Network selection timed out");
if no_match:
if "INTERWORKING-NO-MATCH" not in ev:
raise Exception("Unexpected network match")
return
if "INTERWORKING-NO-MATCH" in ev:
raise Exception("Matching network not found")
if bssid not in ev:
raise Exception("Unexpected BSSID in match")
if "type=home" not in ev:
raise Exception("Home network not recognized")
if type and "type=" + type not in ev:
raise Exception("Network type not recognized correctly")
dev[0].remove_cred(id)
id = dev[0].add_cred()
dev[0].set_cred_quoted(id, "realm", "example.com")
dev[0].set_cred_quoted(id, "username", "test")
dev[0].set_cred_quoted(id, "password", "secret")
dev[0].set_cred_quoted(id, "domain", "no.match.example.com")
dev[0].dump_monitor()
dev[0].request("INTERWORKING_SELECT")
ev = dev[0].wait_event(["INTERWORKING-AP", "INTERWORKING-NO-MATCH"],
timeout=15)
if ev is None:
raise Exception("Network selection timed out");
if "INTERWORKING-NO-MATCH" in ev:
raise Exception("Matching network not found")
if bssid not in ev:
raise Exception("Unexpected BSSID in match")
if "type=roaming" not in ev:
raise Exception("Roaming network not recognized")
def check_sp_type(dev, sp_type):
type = dev.get_status_field("sp_type")
if type is None:
raise Exception("sp_type not available")
if type != sp_type:
raise Exception("sp_type did not indicate home network")
dev[0].set_cred_quoted(id, "realm", "no.match.example.com");
dev[0].dump_monitor()
dev[0].request("INTERWORKING_SELECT")
ev = dev[0].wait_event(["INTERWORKING-AP", "INTERWORKING-NO-MATCH"],
timeout=15)
if ev is None:
raise Exception("Network selection timed out");
if "INTERWORKING-NO-MATCH" not in ev:
raise Exception("Unexpected network match")
def test_ap_hs20_ext_sim(dev, apdev):
"""Hotspot 2.0 with external SIM processing"""
def hlr_auc_gw_available():
if not os.path.exists("/tmp/hlr_auc_gw.sock"):
logger.info("No hlr_auc_gw available");
return "skip"
return False
if not os.path.exists("../../hostapd/hlr_auc_gw"):
logger.info("No hlr_auc_gw available");
return "skip"
bssid = apdev[0]['bssid']
params = hs20_ap_params()
params['hessid'] = bssid
params['anqp_3gpp_cell_net'] = "232,01"
params['domain_name'] = "wlan.mnc001.mcc232.3gppnetwork.org"
hostapd.add_ap(apdev[0]['ifname'], params)
return False
return True
dev[0].request("SET interworking 1")
dev[0].request("SET hs20 1")
dev[0].request("SET external_sim 1")
def interworking_ext_sim_connect(dev, bssid, method):
dev.request("INTERWORKING_CONNECT " + bssid)
id = dev[0].add_cred()
dev[0].set_cred_quoted(id, "imsi", "23201-0000000000")
dev[0].set_cred(id, "eap", "SIM")
dev[0].dump_monitor()
dev[0].request("INTERWORKING_SELECT")
ev = dev[0].wait_event(["INTERWORKING-AP", "INTERWORKING-NO-MATCH"],
timeout=15)
if ev is None:
raise Exception("Network selection timed out")
if "INTERWORKING-NO-MATCH" in ev:
raise Exception("Matching network not found")
if bssid not in ev:
raise Exception("Unexpected BSSID in match")
if "type=home" not in ev:
raise Exception("Home network not recognized")
dev[0].request("INTERWORKING_CONNECT " + bssid)
ev = dev[0].wait_event(["CTRL-EVENT-EAP-METHOD"], timeout=15)
ev = dev.wait_event(["CTRL-EVENT-EAP-METHOD"], timeout=15)
if ev is None:
raise Exception("Network connected timed out")
if "(SIM)" not in ev:
if "(" + method + ")" not in ev:
raise Exception("Unexpected EAP method selection")
ev = dev[0].wait_event(["CTRL-REQ-SIM"], timeout=15)
ev = dev.wait_event(["CTRL-REQ-SIM"], timeout=15)
if ev is None:
raise Exception("Wait for external SIM processing request timed out")
p = ev.split(':', 2)
@ -163,13 +104,46 @@ def test_ap_hs20_ext_sim(dev, apdev):
raise Exception("Unexpected hlr_auc_gw response")
resp = res.split(' ')[2].rstrip()
dev[0].request("CTRL-RSP-SIM-" + id + ":GSM-AUTH:" + resp)
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"], timeout=15)
dev.request("CTRL-RSP-SIM-" + id + ":GSM-AUTH:" + resp)
ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=15)
if ev is None:
raise Exception("Connection timed out")
type = dev[0].get_status_field("sp_type")
if type is None:
raise Exception("sp_type not available")
if type != "home":
raise Exception("sp_type did not indicate home network")
def test_ap_hs20_select(dev, apdev):
"""Hotspot 2.0 network selection"""
bssid = apdev[0]['bssid']
params = hs20_ap_params()
params['hessid'] = bssid
hostapd.add_ap(apdev[0]['ifname'], params)
dev[0].hs20_enable()
id = dev[0].add_cred_values(realm="example.com", username="test",
password="secret", domain="example.com")
interworking_select(dev[0], bssid, "home")
dev[0].remove_cred(id)
id = dev[0].add_cred_values(realm="example.com", username="test",
password="secret",
domain="no.match.example.com")
interworking_select(dev[0], bssid, "roaming")
dev[0].set_cred_quoted(id, "realm", "no.match.example.com");
interworking_select(dev[0], bssid, no_match=True)
def test_ap_hs20_ext_sim(dev, apdev):
"""Hotspot 2.0 with external SIM processing"""
if not hlr_auc_gw_available():
return "skip"
bssid = apdev[0]['bssid']
params = hs20_ap_params()
params['hessid'] = bssid
params['anqp_3gpp_cell_net'] = "232,01"
params['domain_name'] = "wlan.mnc001.mcc232.3gppnetwork.org"
hostapd.add_ap(apdev[0]['ifname'], params)
dev[0].hs20_enable()
dev[0].request("SET external_sim 1")
dev[0].add_cred_values(imsi="23201-0000000000", eap="SIM")
interworking_select(dev[0], "home")
interworking_ext_sim_connect(dev[0], bssid, "SIM")
check_sp_type(dev[0], "home")

View file

@ -118,6 +118,10 @@ class WpaSupplicant:
raise Exception("SET_NETWORK failed")
return None
def hs20_enable(self):
self.request("SET interworking 1")
self.request("SET hs20 1")
def add_cred(self):
id = self.request("ADD_CRED")
if "FAIL" in id:
@ -142,6 +146,23 @@ class WpaSupplicant:
raise Exception("SET_CRED failed")
return None
def add_cred_values(self, realm=None, username=None, password=None,
domain=None, imsi=None, eap=None):
id = self.add_cred()
if realm:
self.set_cred_quoted(id, "realm", realm)
if username:
self.set_cred_quoted(id, "username", username)
if password:
self.set_cred_quoted(id, "password", password)
if domain:
self.set_cred_quoted(id, "domain", domain)
if imsi:
self.set_cred_quoted(id, "imsi", imsi)
if eap:
self.set_cred(id, "eap", eap)
return id;
def select_network(self, id):
id = self.request("SELECT_NETWORK " + str(id))
if "FAIL" in id: