From ecd54bace4954b9d2a25d8fb49a8c7f1aed8004f Mon Sep 17 00:00:00 2001
From: Jouni Malinen <j@w1.fi>
Date: Fri, 3 Jan 2020 15:53:03 +0200
Subject: [PATCH] tests: DFS with RRM

Signed-off-by: Jouni Malinen <j@w1.fi>
---
 tests/hwsim/test_dfs.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/tests/hwsim/test_dfs.py b/tests/hwsim/test_dfs.py
index c33df4f76..a10d32b87 100644
--- a/tests/hwsim/test_dfs.py
+++ b/tests/hwsim/test_dfs.py
@@ -28,7 +28,7 @@ def wait_dfs_event(hapd, event, timeout):
 
 def start_dfs_ap(ap, ssid="dfs", ht=True, ht40=False,
                  ht40minus=False, vht80=False, vht20=False, chanlist=None,
-                 channel=None, country="FI"):
+                 channel=None, country="FI", rrm_beacon_report=False):
     ifname = ap['ifname']
     logger.info("Starting AP " + ifname + " on DFS channel")
     hapd = hostapd.add_ap(ap, {}, no_enable=True)
@@ -57,6 +57,8 @@ def start_dfs_ap(ap, ssid="dfs", ht=True, ht40=False,
         hapd.set("chanlist", chanlist)
     if channel:
         hapd.set("channel", str(channel))
+    if rrm_beacon_report:
+        hapd.set("rrm_beacon_report", "1")
     hapd.enable()
 
     ev = wait_dfs_event(hapd, "DFS-CAC-START", 5)
@@ -495,3 +497,32 @@ def test_dfs_cac_restart_on_enable(dev, apdev):
 
     finally:
         clear_regdom(hapd, dev)
+
+def test_dfs_rrm(dev, apdev, params):
+    """DFS with RRM [long]"""
+    if not params['long']:
+        raise HwsimSkip("Skip test case with long duration due to --long not specified")
+    try:
+        hapd = None
+        hapd = start_dfs_ap(apdev[0], country="US", rrm_beacon_report=True)
+
+        ev = wait_dfs_event(hapd, "DFS-CAC-COMPLETED", 70)
+        if "success=1" not in ev or "freq=5260" not in ev:
+            raise Exception("Unexpected DFS freq result")
+        ev = hapd.wait_event(["AP-ENABLED"], timeout=5)
+        if not ev:
+            raise Exception("AP setup timed out")
+
+        dev[0].connect("dfs", key_mgmt="NONE", scan_freq="5260")
+        dev[0].wait_regdom(country_ie=True)
+        hapd.wait_sta()
+        hwsim_utils.test_connectivity(dev[0], hapd)
+        addr = dev[0].own_addr()
+        token = hapd.request("REQ_BEACON " + addr + " " + "51000000000002ffffffffffff")
+        if "FAIL" in token:
+            raise Exception("REQ_BEACON failed")
+        ev = hapd.wait_event(["BEACON-RESP-RX"], timeout=10)
+        if ev is None:
+            raise Exception("Beacon report response not received")
+    finally:
+        clear_regdom(hapd, dev)