tests: wpa_supplicant AP mode with pmf=1/2

Signed-off-by: Jouni Malinen <quic_jouni@quicinc.com>

diff --git a/tests/hwsim/test_wpas_ap.py b/tests/hwsim/test_wpas_ap.py
index 68db25206ca33ecf6a2596c4de50580685263517..620cf614e9ac06960f946a8e4e8f9cf6c4b68c35 100644 (file)
--- a/tests/hwsim/test_wpas_ap.py
+++ b/tests/hwsim/test_wpas_ap.py
@@ -842,6 +842,22 @@ def test_wpas_ap_sae(dev):
     """wpa_supplicant AP mode - SAE using psk"""
     run_wpas_ap_sae(dev, False)
 
+def test_wpas_ap_sae_pmf1(dev):
+    """wpa_supplicant AP mode - SAE using psk and pmf=1"""
+    try:
+        dev[0].set("pmf", "1")
+        run_wpas_ap_sae(dev, False, pmf=2)
+    finally:
+        dev[0].set("pmf", "0")
+
+def test_wpas_ap_sae_pmf2(dev):
+    """wpa_supplicant AP mode - SAE using psk and pmf=2"""
+    try:
+        dev[0].set("pmf", "2")
+        run_wpas_ap_sae(dev, False, pmf=1)
+    finally:
+        dev[0].set("pmf", "0")
+
 def test_wpas_ap_sae_password(dev):
     """wpa_supplicant AP mode - SAE using sae_password"""
     run_wpas_ap_sae(dev, True)
@@ -856,7 +872,7 @@ def test_wpas_ap_sae_pwe_1(dev):
         dev[0].set("sae_pwe", "0")
         dev[1].set("sae_pwe", "0")
 
-def run_wpas_ap_sae(dev, sae_password, sae_password_id=False):
+def run_wpas_ap_sae(dev, sae_password, sae_password_id=False, pmf=0):
     check_sae_capab(dev[0])
     check_sae_capab(dev[1])
     dev[0].request("SET sae_groups ")
@@ -884,7 +900,7 @@ def run_wpas_ap_sae(dev, sae_password, sae_password_id=False):
 
     dev[1].request("SET sae_groups ")
     dev[1].connect("wpas-ap-sae", key_mgmt="SAE", sae_password="12345678",
-                   sae_password_id=pw_id, scan_freq="2412")
+                   sae_password_id=pw_id, scan_freq="2412", ieee80211w=str(pmf))
 
 def test_wpas_ap_scan(dev, apdev):
     """wpa_supplicant AP mode and scanning"""