From 1a38056930b449c088335f554d1677fe1ba58dfe Mon Sep 17 00:00:00 2001 From: Jouni Malinen Date: Tue, 16 Jan 2024 21:12:43 +0200 Subject: [PATCH] tests: Encrypted Key Data field in EAPOL-Key msg 2/4 and 4/4 Signed-off-by: Jouni Malinen --- tests/hwsim/test_ap_psk.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/tests/hwsim/test_ap_psk.py b/tests/hwsim/test_ap_psk.py index 7675c3b27..3e6de6172 100644 --- a/tests/hwsim/test_ap_psk.py +++ b/tests/hwsim/test_ap_psk.py @@ -3788,3 +3788,35 @@ def test_rsn_eapol_m4_extra(dev, apdev): raise Exception("Failed to add test elements") dev[0].connect(ssid, psk=passphrase, scan_freq="2412") hapd.wait_sta() + +def test_rsn_eapol_m2_encrypt(dev, apdev): + """Encrypted Key Data field in EAPOL-Key msg 2/4""" + ssid = "test-rsn" + passphrase = 'qwertyuiop' + params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase) + hapd = hostapd.add_ap(apdev[0], params) + + # Add a reserved element and KDE into EAPOL-Key msg 2/4 and request the + # Key Data field to be encrypted. + elems = '02051122334455' + 'dd05000facff11' + if "OK" not in dev[0].request("TEST_EAPOL_M2_ELEMS " + elems): + raise Exception("Failed to add test elements") + dev[0].set("encrypt_eapol_m2", "1") + dev[0].connect(ssid, psk=passphrase, scan_freq="2412") + hapd.wait_sta() + +def test_rsn_eapol_m4_encrypt(dev, apdev): + """Encrypted Key Data field in EAPOL-Key msg 4/4""" + ssid = "test-rsn" + passphrase = 'qwertyuiop' + params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase) + hapd = hostapd.add_ap(apdev[0], params) + + # Add a reserved element and KDE into EAPOL-Key msg 4/4 and request the + # Key Data field to be encrypted. + elems = '02051122334455' + 'dd05000facff11' + if "OK" not in dev[0].request("TEST_EAPOL_M4_ELEMS " + elems): + raise Exception("Failed to add test elements") + dev[0].set("encrypt_eapol_m4", "1") + dev[0].connect(ssid, psk=passphrase, scan_freq="2412") + hapd.wait_sta()