From ae3eacf7717f0553c9434b920a799a1d61886b22 Mon Sep 17 00:00:00 2001 From: Jouni Malinen Date: Mon, 7 Sep 2015 17:46:40 +0300 Subject: [PATCH] tests: WPS ER learn OOM Signed-off-by: Jouni Malinen --- tests/hwsim/test_ap_wps.py | 50 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) diff --git a/tests/hwsim/test_ap_wps.py b/tests/hwsim/test_ap_wps.py index 4dabb22..674ab96 100644 --- a/tests/hwsim/test_ap_wps.py +++ b/tests/hwsim/test_ap_wps.py @@ -1794,6 +1794,56 @@ def _test_ap_wps_er_set_sel_reg_oom(dev, apdev): dev[0].request("WPS_ER_STOP") +def test_ap_wps_er_learn_oom(dev, apdev): + """WPS ER learn OOM""" + try: + _test_ap_wps_er_learn_oom(dev, apdev) + finally: + dev[0].request("WPS_ER_STOP") + +def _test_ap_wps_er_learn_oom(dev, apdev): + ssid = "wps-er-add-enrollee" + ap_pin = "12345670" + ap_uuid = "27ea801a-9e5c-4e73-bd82-f89cbcd10d7e" + params = { "ssid": ssid, "eap_server": "1", "wps_state": "2", + "wpa_passphrase": "12345678", "wpa": "2", + "wpa_key_mgmt": "WPA-PSK", "rsn_pairwise": "CCMP", + "device_name": "Wireless AP", "manufacturer": "Company", + "model_name": "WAP", "model_number": "123", + "serial_number": "12345", "device_type": "6-0050F204-1", + "os_version": "01020300", + "config_methods": "label push_button", + "ap_pin": ap_pin, "uuid": ap_uuid, "upnp_iface": "lo" } + hapd = hostapd.add_ap(apdev[0]['ifname'], params) + dev[0].scan_for_bss(apdev[0]['bssid'], freq=2412) + dev[0].wps_reg(apdev[0]['bssid'], ap_pin) + + dev[0].request("WPS_ER_START ifname=lo") + ev = dev[0].wait_event(["WPS-ER-AP-ADD"], timeout=10) + if ev is None: + raise Exception("AP not discovered") + + for func in [ "wps_er_http_put_message_cb", + "xml_get_base64_item;wps_er_http_put_message_cb", + "http_client_url_parse;wps_er_ap_put_message", + "wps_er_soap_hdr;wps_er_ap_put_message", + "http_client_addr;wps_er_ap_put_message" ]: + with alloc_fail(dev[0], 1, func): + dev[0].request("WPS_ER_LEARN " + ap_uuid + " " + ap_pin) + ev = dev[0].wait_event(["WPS-ER-AP-SETTINGS"], timeout=1) + if ev is not None: + raise Exception("AP learn succeeded during OOM") + + dev[0].request("WPS_ER_LEARN " + ap_uuid + " " + ap_pin) + ev = dev[0].wait_event(["WPS-ER-AP-SETTINGS"], timeout=10) + if ev is None: + raise Exception("AP learn did not succeed") + + if "FAIL" not in dev[0].request("WPS_ER_LEARN 00000000-9e5c-4e73-bd82-f89cbcd10d7e " + ap_pin): + raise Exception("WPS_ER_LEARN for unknown AP accepted") + + dev[0].request("WPS_ER_STOP") + def test_ap_wps_fragmentation(dev, apdev): """WPS with fragmentation in EAP-WSC and mixed mode WPA+WPA2""" ssid = "test-wps-fragmentation" -- 2.1.4