tests: Check for beacon loss when using beacon protection

This extends testing coverage to detect an issue that was fixed in
commit bf9cbb462fd9 ("Fix writing of BIGTK in FT protocol").

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

diff --git a/tests/hwsim/test_ap_ft.py b/tests/hwsim/test_ap_ft.py
index c30cb8593eb9ef0755bfa661401c5e5a0227727e..a2f1a4ead4312dedfdb422b10646e297a70bc3b8 100644 (file)
--- a/tests/hwsim/test_ap_ft.py
+++ b/tests/hwsim/test_ap_ft.py
@@ -560,6 +560,9 @@ def test_ap_ft_pmf_required_over_ds(dev, apdev):
 def test_ap_ft_pmf_beacon_prot(dev, apdev):
     """WPA2-PSK-FT AP with PMF and beacon protection"""
     run_ap_ft_pmf(dev, apdev, "1", beacon_prot=True)
+    ev = dev[0].wait_event(["CTRL-EVENT-BEACON-LOSS"], timeout=5)
+    if ev is not None:
+        raise Exception("Beacon loss detected")
 
 def run_ap_ft_pmf(dev, apdev, ieee80211w, over_ds=False, beacon_prot=False):
     ssid = "test-ft"

diff --git a/tests/hwsim/test_ap_pmf.py b/tests/hwsim/test_ap_pmf.py
index 3a9fa7a72293fa7cd78e5948fd3ce3329ee0e4e3..7b7e5a29e17319f58158654d12301e479a48c947 100644 (file)
--- a/tests/hwsim/test_ap_pmf.py
+++ b/tests/hwsim/test_ap_pmf.py
@@ -1371,6 +1371,10 @@ def run_ap_pmf_beacon_protection(dev, apdev, cipher):
     if valid_bip < 0 or invalid_bip > 0 or missing_bip > 0:
         raise Exception("Unexpected wlantest BIP counters: valid=%d invalid=%d missing=%d" % (valid_bip, invalid_bip, missing_bip))
 
+    ev = dev[0].wait_event(["CTRL-EVENT-BEACON-LOSS"], timeout=5)
+    if ev is not None:
+        raise Exception("Beacon loss detected")
+
 def test_ap_pmf_beacon_protection_mismatch(dev, apdev):
     """WPA2-PSK Beacon protection MIC mismatch"""
     run_ap_pmf_beacon_protection_mismatch(dev, apdev, False)