tests: Update MBO test cases for non_pref_chan changes
[mech_eap.git] / tests / hwsim / test_mbo.py
index b3cf089..75bac40 100644 (file)
@@ -12,9 +12,121 @@ import hostapd
 import os
 import time
 
+import hostapd
 from tshark import run_tshark
 from utils import alloc_fail, fail_test
 
+def set_reg(country_code, apdev0=None, apdev1=None, dev0=None):
+    if apdev0:
+        hostapd.cmd_execute(apdev0, ['iw', 'reg', 'set', country_code])
+    if apdev1:
+        hostapd.cmd_execute(apdev1, ['iw', 'reg', 'set', country_code])
+    if dev0:
+        dev0.cmd_execute(['iw', 'reg', 'set', country_code])
+
+def run_mbo_supp_oper_classes(dev, apdev, hapd, hapd2, country):
+    """MBO and supported operating classes"""
+    addr = dev[0].own_addr()
+
+    res2 = None
+    res5 = None
+
+    dev[0].flush_scan_cache()
+    dev[0].dump_monitor()
+
+    logger.info("Country: " + country)
+    set_reg(country, apdev[0], apdev[1], dev[0])
+    for j in range(5):
+        ev = dev[0].wait_event(["CTRL-EVENT-REGDOM-CHANGE"], timeout=5)
+        if ev is None:
+            raise Exception("No regdom change event")
+        if "alpha2=" + country in ev:
+            break
+    dev[0].dump_monitor()
+    dev[1].dump_monitor()
+    dev[2].dump_monitor()
+    if hapd:
+        hapd.set("country_code", country)
+        hapd.enable()
+        dev[0].scan_for_bss(hapd.own_addr(), 5180, force_scan=True)
+        dev[0].connect("test-wnm-mbo", key_mgmt="NONE", scan_freq="5180")
+        sta = hapd.get_sta(addr)
+        res5 = sta['supp_op_classes'][2:]
+        dev[0].request("REMOVE_NETWORK all")
+        hapd.disable()
+        dev[0].wait_disconnected()
+        dev[0].dump_monitor()
+
+    hapd2.set("country_code", country)
+    hapd2.enable()
+    dev[0].scan_for_bss(hapd2.own_addr(), 2412, force_scan=True)
+    dev[0].connect("test-wnm-mbo-2", key_mgmt="NONE", scan_freq="2412")
+    sta = hapd2.get_sta(addr)
+    res2 = sta['supp_op_classes'][2:]
+    dev[0].request("REMOVE_NETWORK all")
+    hapd2.disable()
+    dev[0].wait_disconnected()
+    dev[0].dump_monitor()
+
+    return res2, res5
+
+def test_mbo_supp_oper_classes(dev, apdev):
+    """MBO and supported operating classes"""
+    params = { 'ssid': "test-wnm-mbo",
+               'mbo': '1',
+               "country_code": "US",
+               'ieee80211d': '1',
+               "ieee80211n": "1",
+               "hw_mode": "a",
+               "channel": "36" }
+    hapd = hostapd.add_ap(apdev[0], params, no_enable=True)
+
+    params = { 'ssid': "test-wnm-mbo-2",
+               'mbo': '1',
+               "country_code": "US",
+               'ieee80211d': '1',
+               "ieee80211n": "1",
+               "hw_mode": "g",
+               "channel": "1" }
+    hapd2 = hostapd.add_ap(apdev[1], params, no_enable=True)
+
+    try:
+        za2, za5 = run_mbo_supp_oper_classes(dev, apdev, hapd, hapd2, "ZA")
+        fi2, fi5 = run_mbo_supp_oper_classes(dev, apdev, hapd, hapd2, "FI")
+        us2, us5 = run_mbo_supp_oper_classes(dev, apdev, hapd, hapd2, "US")
+        jp2, jp5 = run_mbo_supp_oper_classes(dev, apdev, hapd, hapd2, "JP")
+        bd2, bd5 = run_mbo_supp_oper_classes(dev, apdev, None, hapd2, "BD")
+        kz2, kz5 = run_mbo_supp_oper_classes(dev, apdev, None, hapd2, "KZ")
+    finally:
+        dev[0].dump_monitor()
+        set_reg("00", apdev[0], apdev[1], dev[0])
+        ev = dev[0].wait_event(["CTRL-EVENT-REGDOM-CHANGE"], timeout=1)
+
+    za = "515354737475767778797a7b808182"
+    fi = "515354737475767778797a7b808182"
+    us = "515354737475767778797a7b7c7d7e7f808182"
+    jp = "51525354737475767778797a7b808182"
+    bd = "5153547c7d7e7f80"
+    kz = "515354"
+
+    tests = [ ("ZA", za, za2, za5, True),
+              ("FI", fi, fi2, fi5, True),
+              ("US", us, us2, us5, True),
+              ("JP", jp, jp2, jp5, True),
+              ("BD", bd, bd2, bd5, False),
+              ("KZ", kz, kz2, kz5, False) ]
+    for country, expected, res2, res5, inc5 in tests:
+        # For now, allow operating class 129 to be missing since not all
+        # installed regdb files include the 160 MHz channels.
+        expected2 = expected.replace('808182', '8082')
+        # For now, allow operating classes 121-123 to be missing since not all
+        # installed regdb files include the related US DFS channels.
+        expected2 = expected2.replace('78797a7b7c', '787c')
+        if res2 != expected and res2 != expected2:
+            raise Exception("Unexpected supp_op_class string (country=%s, 2.4 GHz): %s (expected: %s)" % (country, res2, expected))
+        if inc5 and res5 != expected and res5 != expected2:
+            raise Exception("Unexpected supp_op_class string (country=%s, 5 GHz): %s (expected: %s)" % (country, res5, expected))
+
 def test_mbo_assoc_disallow(dev, apdev, params):
     hapd1 = hostapd.add_ap(apdev[0], { "ssid": "MBO", "mbo": "1" })
     hapd2 = hostapd.add_ap(apdev[1], { "ssid": "MBO", "mbo": "1" })
@@ -97,8 +209,8 @@ def test_mbo_cell_capa_update_pmf(dev, apdev):
     ssid = "test-wnm-mbo"
     passphrase = "12345678"
     params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase)
-    params["wpa_key_mgmt"] = "WPA-PSK-SHA256";
-    params["ieee80211w"] = "2";
+    params["wpa_key_mgmt"] = "WPA-PSK-SHA256"
+    params["ieee80211w"] = "2"
     params['mbo'] = '1'
     hapd = hostapd.add_ap(apdev[0], params)
     bssid = apdev[0]['bssid']
@@ -155,7 +267,7 @@ def test_mbo_non_pref_chan(dev, apdev):
         raise Exception("Invalid non_pref_chan value accepted")
     if "OK" not in dev[0].request("SET non_pref_chan 81:7:200:3"):
         raise Exception("Failed to set non-preferred channel list")
-    if "OK" not in dev[0].request("SET non_pref_chan 81:7:200:1:123 81:9:100:2"):
+    if "OK" not in dev[0].request("SET non_pref_chan 81:7:200:1 81:9:100:2"):
         raise Exception("Failed to set non-preferred channel list")
 
     dev[0].connect(ssid, key_mgmt="NONE", scan_freq="2412")
@@ -165,11 +277,11 @@ def test_mbo_non_pref_chan(dev, apdev):
     logger.debug("STA: " + str(sta))
     if 'non_pref_chan[0]' not in sta:
         raise Exception("Missing non_pref_chan[0] value (assoc)")
-    if sta['non_pref_chan[0]'] != '81:200:1:123:7':
+    if sta['non_pref_chan[0]'] != '81:200:1:7':
         raise Exception("Unexpected non_pref_chan[0] value (assoc)")
     if 'non_pref_chan[1]' not in sta:
         raise Exception("Missing non_pref_chan[1] value (assoc)")
-    if sta['non_pref_chan[1]'] != '81:100:2:0:9':
+    if sta['non_pref_chan[1]'] != '81:100:2:9':
         raise Exception("Unexpected non_pref_chan[1] value (assoc)")
     if 'non_pref_chan[2]' in sta:
         raise Exception("Unexpected non_pref_chan[2] value (assoc)")
@@ -181,30 +293,26 @@ def test_mbo_non_pref_chan(dev, apdev):
     logger.debug("STA: " + str(sta))
     if 'non_pref_chan[0]' not in sta:
         raise Exception("Missing non_pref_chan[0] value (update 1)")
-    if sta['non_pref_chan[0]'] != '81:100:2:0:9':
+    if sta['non_pref_chan[0]'] != '81:100:2:9':
         raise Exception("Unexpected non_pref_chan[0] value (update 1)")
     if 'non_pref_chan[1]' in sta:
-        raise Exception("Unexpected non_pref_chan[2] value (update 1)")
+        raise Exception("Unexpected non_pref_chan[1] value (update 1)")
 
-    if "OK" not in dev[0].request("SET non_pref_chan 81:9:100:2 81:10:100:2 81:8:100:2 81:7:100:1:123 81:5:100:1:124"):
+    if "OK" not in dev[0].request("SET non_pref_chan 81:9:100:2 81:10:100:2 81:8:100:2 81:7:100:1 81:5:100:1"):
         raise Exception("Failed to update non-preferred channel list")
     time.sleep(0.1)
     sta = hapd.get_sta(addr)
     logger.debug("STA: " + str(sta))
     if 'non_pref_chan[0]' not in sta:
         raise Exception("Missing non_pref_chan[0] value (update 2)")
-    if sta['non_pref_chan[0]'] != '81:100:1:123:7':
+    if sta['non_pref_chan[0]'] != '81:100:1:7,5':
         raise Exception("Unexpected non_pref_chan[0] value (update 2)")
     if 'non_pref_chan[1]' not in sta:
         raise Exception("Missing non_pref_chan[1] value (update 2)")
-    if sta['non_pref_chan[1]'] != '81:100:1:124:5':
+    if sta['non_pref_chan[1]'] != '81:100:2:9,10,8':
         raise Exception("Unexpected non_pref_chan[1] value (update 2)")
-    if 'non_pref_chan[2]' not in sta:
-        raise Exception("Missing non_pref_chan[2] value (update 2)")
-    if sta['non_pref_chan[2]'] != '81:100:2:0:9,10,8':
+    if 'non_pref_chan[2]' in sta:
         raise Exception("Unexpected non_pref_chan[2] value (update 2)")
-    if 'non_pref_chan[3]' in sta:
-        raise Exception("Unexpected non_pref_chan[3] value (update 2)")
 
     if "OK" not in dev[0].request("SET non_pref_chan 81:5:90:2 82:14:91:2"):
         raise Exception("Failed to update non-preferred channel list")
@@ -213,11 +321,11 @@ def test_mbo_non_pref_chan(dev, apdev):
     logger.debug("STA: " + str(sta))
     if 'non_pref_chan[0]' not in sta:
         raise Exception("Missing non_pref_chan[0] value (update 3)")
-    if sta['non_pref_chan[0]'] != '81:90:2:0:5':
+    if sta['non_pref_chan[0]'] != '81:90:2:5':
         raise Exception("Unexpected non_pref_chan[0] value (update 3)")
     if 'non_pref_chan[1]' not in sta:
         raise Exception("Missing non_pref_chan[1] value (update 3)")
-    if sta['non_pref_chan[1]'] != '82:91:2:0:14':
+    if sta['non_pref_chan[1]'] != '82:91:2:14':
         raise Exception("Unexpected non_pref_chan[1] value (update 3)")
     if 'non_pref_chan[2]' in sta:
         raise Exception("Unexpected non_pref_chan[2] value (update 3)")