package/network/services/hostapd/patches/559-tests-add-WPS-tests-to-multi_ap-hwsim-tests.patch

   1 From 0729e01f5830ebf4701f0b1b7ff1bd2a2eedae40 Mon Sep 17 00:00:00 2001
   2 From: "Arnout Vandecappelle (Essensium/Mind)" <arnout@mind.be>
   3 Date: Tue, 12 Feb 2019 11:02:42 +0100
   4 Subject: [PATCH] tests: add WPS tests to multi_ap hwsim tests
   5
   6 Signed-off-by: Arnout Vandecappelle (Essensium/Mind) <arnout@mind.be>
   7 ---
   8 v4: new patch
   9 ---
  10  tests/hwsim/test_multi_ap.py | 164 +++++++++++++++++++++++++++++++++++
  11  1 file changed, 164 insertions(+)
  12
  13 --- a/tests/hwsim/test_multi_ap.py
  14 +++ b/tests/hwsim/test_multi_ap.py
  15 @@ -61,3 +61,167 @@ def test_multi_ap_fronthaul_on_ap(dev, a
  16          raise Exception("Connection result not reported")
  17      if "CTRL-EVENT-DISCONNECTED" not in ev:
  18          raise Exception("Unexpected connection result")
  19 +
  20 +def run_multi_ap_wps(dev, apdev, params, multi_ap_bssid = None):
  21 +    """Helper for running Multi-AP WPS tests
  22 +
  23 +    dev[0] does multi_ap WPS, dev[1] does normal WPS. apdev[0] is the fronthaul
  24 +    BSS. If there is a separate backhaul BSS, it must have been set up by the
  25 +    caller. params are the normal SSID parameters, they will be extended with
  26 +    the WPS parameters. multi_ap_bssid must be given if it is not equal to the
  27 +    fronthaul BSSID."""
  28 +
  29 +    if multi_ap_bssid is None:
  30 +        multi_ap_bssid = apdev[0]['bssid']
  31 +    params.update({"wps_state": "2", "eap_server": "1"})
  32 +
  33 +    # WPS with multi-ap station dev[0]
  34 +    hapd = hostapd.add_ap(apdev[0], params)
  35 +    hapd.request("WPS_PBC")
  36 +    if "PBC Status: Active" not in hapd.request("WPS_GET_STATUS"):
  37 +        raise Exception("PBC status not shown correctly")
  38 +
  39 +    dev[0].request("WPS_PBC multi_ap=1")
  40 +    dev[0].wait_connected(timeout=20)
  41 +    status = dev[0].get_status()
  42 +    if status['wpa_state'] != 'COMPLETED' or status['bssid'] != multi_ap_bssid:
  43 +        raise Exception("Not fully connected")
  44 +    if status['ssid'] != params['multi_ap_backhaul_ssid'].strip('"'):
  45 +        raise Exception("Unexpected SSID %s != %s" % (status['ssid'], params["multi_ap_backhaul_ssid"]))
  46 +    if status['pairwise_cipher'] != 'CCMP':
  47 +        raise Exception("Unexpected encryption configuration %s" % status['pairwise_cipher'])
  48 +    if status['key_mgmt'] != 'WPA2-PSK':
  49 +        raise Exception("Unexpected key_mgmt")
  50 +
  51 +    status = hapd.request("WPS_GET_STATUS")
  52 +    if "PBC Status: Disabled" not in status:
  53 +        raise Exception("PBC status not shown correctly")
  54 +    if "Last WPS result: Success" not in status:
  55 +        raise Exception("Last WPS result not shown correctly")
  56 +    if "Peer Address: " + dev[0].p2p_interface_addr() not in status:
  57 +        raise Exception("Peer address not shown correctly")
  58 +
  59 +    if len(dev[0].list_networks()) != 1:
  60 +        raise Exception("Unexpected number of network blocks")
  61 +
  62 +    # WPS with non-multi-ap station dev[1]
  63 +    hapd.request("WPS_PBC")
  64 +    if "PBC Status: Active" not in hapd.request("WPS_GET_STATUS"):
  65 +        raise Exception("PBC status not shown correctly")
  66 +
  67 +    dev[1].request("WPS_PBC")
  68 +    dev[1].wait_connected(timeout=20)
  69 +    status = dev[1].get_status()
  70 +    if status['wpa_state'] != 'COMPLETED' or status['bssid'] != apdev[0]['bssid']:
  71 +        raise Exception("Not fully connected")
  72 +    if status['ssid'] != params["ssid"]:
  73 +        raise Exception("Unexpected SSID")
  74 +    # Fronthaul may be something else than WPA2-PSK so don't test it.
  75 +
  76 +    status = hapd.request("WPS_GET_STATUS")
  77 +    if "PBC Status: Disabled" not in status:
  78 +        raise Exception("PBC status not shown correctly")
  79 +    if "Last WPS result: Success" not in status:
  80 +        raise Exception("Last WPS result not shown correctly")
  81 +    if "Peer Address: " + dev[1].p2p_interface_addr() not in status:
  82 +        raise Exception("Peer address not shown correctly")
  83 +
  84 +    if len(dev[1].list_networks()) != 1:
  85 +        raise Exception("Unexpected number of network blocks")
  86 +
  87 +
  88 +def test_multi_ap_wps_shared(dev, apdev):
  89 +    """WPS on shared fronthaul/backhaul AP"""
  90 +    ssid = "multi-ap-wps"
  91 +    passphrase = "12345678"
  92 +    params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase)
  93 +    params.update({"multi_ap": "3",
  94 +                   "multi_ap_backhaul_ssid": '"%s"' % ssid,
  95 +                   "multi_ap_backhaul_wpa_passphrase": passphrase})
  96 +    run_multi_ap_wps(dev, apdev, params)
  97 +
  98 +def test_multi_ap_wps_shared_psk(dev, apdev):
  99 +    """WPS on shared fronthaul/backhaul AP using PSK"""
 100 +    ssid = "multi-ap-wps"
 101 +    psk = "1234567890abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
 102 +    params = hostapd.wpa2_params(ssid=ssid)
 103 +    params.update({"wpa_psk": psk, "multi_ap": "3",
 104 +                   "multi_ap_backhaul_ssid": '"%s"' % ssid,
 105 +                   "multi_ap_backhaul_wpa_psk": psk})
 106 +    run_multi_ap_wps(dev, apdev, params)
 107 +
 108 +def test_multi_ap_wps_split(dev, apdev):
 109 +    """WPS on split fronthaul and backhaul AP"""
 110 +    backhaul_ssid = "multi-ap-backhaul-wps"
 111 +    backhaul_passphrase = "87654321"
 112 +    params = hostapd.wpa2_params(ssid="multi-ap-fronthaul-wps", passphrase="12345678")
 113 +    params.update({"multi_ap": "2",
 114 +                   "multi_ap_backhaul_ssid": '"%s"' % backhaul_ssid,
 115 +                   "multi_ap_backhaul_wpa_passphrase": backhaul_passphrase})
 116 +    params_backhaul = hostapd.wpa2_params(ssid=backhaul_ssid, passphrase=backhaul_passphrase)
 117 +    params_backhaul.update({"multi_ap": "1"})
 118 +    hapd_backhaul = hostapd.add_ap(apdev[1], params_backhaul)
 119 +
 120 +    run_multi_ap_wps(dev, apdev, params, hapd_backhaul.own_addr())
 121 +
 122 +def test_multi_ap_wps_split_psk(dev, apdev):
 123 +    """WPS on split fronthaul and backhaul AP"""
 124 +    backhaul_ssid = "multi-ap-backhaul-wps"
 125 +    backhaul_psk = "1234567890abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
 126 +    params = hostapd.wpa2_params(ssid="multi-ap-fronthaul-wps", passphrase="12345678")
 127 +    params.update({"multi_ap": "2",
 128 +                   "multi_ap_backhaul_ssid": '"%s"' % backhaul_ssid,
 129 +                   "multi_ap_backhaul_wpa_psk": backhaul_psk})
 130 +    params_backhaul = hostapd.wpa2_params(ssid=backhaul_ssid)
 131 +    params_backhaul.update({"multi_ap": "1", "wpa_psk": backhaul_psk})
 132 +    hapd_backhaul = hostapd.add_ap(apdev[1], params_backhaul)
 133 +
 134 +    run_multi_ap_wps(dev, apdev, params, hapd_backhaul.own_addr())
 135 +
 136 +def test_multi_ap_wps_split_mixed(dev, apdev):
 137 +    """WPS on split fronthaul and backhaul AP with mixed-mode fronthaul"""
 138 +    backhaul_ssid = "multi-ap-backhaul-wps"
 139 +    backhaul_passphrase = "87654321"
 140 +    params = hostapd.wpa_mixed_params(ssid="multi-ap-fronthaul-wps", passphrase="12345678")
 141 +    params.update({"multi_ap": "2",
 142 +                   "multi_ap_backhaul_ssid": '"%s"' % backhaul_ssid,
 143 +                   "multi_ap_backhaul_wpa_passphrase": backhaul_passphrase})
 144 +    params_backhaul = hostapd.wpa2_params(ssid=backhaul_ssid, passphrase=backhaul_passphrase)
 145 +    params_backhaul.update({"multi_ap": "1"})
 146 +    hapd_backhaul = hostapd.add_ap(apdev[1], params_backhaul)
 147 +
 148 +    run_multi_ap_wps(dev, apdev, params, hapd_backhaul.own_addr())
 149 +
 150 +def test_multi_ap_wps_split_open(dev, apdev):
 151 +    """WPS on split fronthaul and backhaul AP with open fronthaul"""
 152 +    backhaul_ssid = "multi-ap-backhaul-wps"
 153 +    backhaul_passphrase = "87654321"
 154 +    params = {"ssid": "multi-ap-wps-fronthaul", "multi_ap": "2",
 155 +              "multi_ap_backhaul_ssid": '"%s"' % backhaul_ssid,
 156 +              "multi_ap_backhaul_wpa_passphrase": backhaul_passphrase}
 157 +    params_backhaul = hostapd.wpa2_params(ssid=backhaul_ssid, passphrase=backhaul_passphrase)
 158 +    params_backhaul.update({"multi_ap": "1"})
 159 +    hapd_backhaul = hostapd.add_ap(apdev[1], params_backhaul)
 160 +
 161 +    run_multi_ap_wps(dev, apdev, params, hapd_backhaul.own_addr())
 162 +
 163 +def test_multi_ap_wps_fail_non_multi_ap(dev, apdev):
 164 +    """Multi-AP WPS on non-WPS AP fails"""
 165 +
 166 +    params = hostapd.wpa2_params(ssid="non-multi-ap-wps", passphrase="12345678")
 167 +    params.update({"wps_state": "2", "eap_server": "1"})
 168 +
 169 +    hapd = hostapd.add_ap(apdev[0], params)
 170 +    hapd.request("WPS_PBC")
 171 +    if "PBC Status: Active" not in hapd.request("WPS_GET_STATUS"):
 172 +        raise Exception("PBC status not shown correctly")
 173 +
 174 +    dev[0].request("WPS_PBC multi_ap=1")
 175 +    # Since we will fail to associate and WPS doesn't even get started, there
 176 +    # isn't much we can do except wait for timeout. For PBC, it is not possible
 177 +    # to change the timeout from 2 minutes. Instead of waiting for the timeout,
 178 +    # just check that WPS doesn't finish within reasonable time.
 179 +    ev = dev[0].wait_event(["WPS-SUCCESS", "WPS-FAIL"], timeout=20)
 180 +    if ev:
 181 +        raise Exception("WPS operation completed: " + ev)
 182 +    dev[0].request("WPS_CANCEL")