diff --git a/tests/hwsim/test_sae.py b/tests/hwsim/test_sae.py index 29e2ebbb2..dbf630f48 100644 --- a/tests/hwsim/test_sae.py +++ b/tests/hwsim/test_sae.py @@ -2162,3 +2162,36 @@ def test_sae_h2e_rsnxe_mismatch_assoc(dev, apdev): finally: dev[0].set("sae_groups", "") dev[0].set("sae_pwe", "0") + +def test_sae_h2e_rsnxe_mismatch_ap(dev, apdev): + """SAE H2E and RSNXE mismatch in EAPOL-Key msg 3/4""" + run_sae_h2e_rsnxe_mismatch_ap(dev, apdev, "F40100") + +def test_sae_h2e_rsnxe_mismatch_ap2(dev, apdev): + """SAE H2E and RSNXE mismatch in EAPOL-Key msg 3/4""" + run_sae_h2e_rsnxe_mismatch_ap(dev, apdev, "F400") + +def run_sae_h2e_rsnxe_mismatch_ap(dev, apdev, rsnxe): + params = hostapd.wpa2_params(ssid="sae-pwe", passphrase="12345678") + params['wpa_key_mgmt'] = 'SAE' + params['sae_pwe'] = "1" + params['rsnxe_override_eapol'] = rsnxe + hapd = hostapd.add_ap(apdev[0], params) + try: + dev[0].set("sae_groups", "19") + dev[0].set("sae_pwe", "1") + dev[0].connect("sae-pwe", psk="12345678", key_mgmt="SAE", + scan_freq="2412", wait_connect=False) + ev = dev[0].wait_event(["Associated with"], timeout=10) + if ev is None: + raise Exception("No indication of association seen") + ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED", + "CTRL-EVENT-DISCONNECTED"], timeout=5) + dev[0].request("REMOVE_NETWORK all") + if ev is None: + raise Exception("No disconnection seen") + if "CTRL-EVENT-DISCONNECTED" not in ev: + raise Exception("Unexpected connection") + finally: + dev[0].set("sae_groups", "") + dev[0].set("sae_pwe", "0")