tests/hwsim/test_ibss.py

   1 # IBSS test cases
   2 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   3 #
   4 # This software may be distributed under the terms of the BSD license.
   5 # See README for more details.
   6
   7 import logging
   8 logger = logging.getLogger()
   9 import time
  10 import re
  11 import subprocess
  12
  13 import hwsim_utils
  14
  15 def connect_ibss_cmd(dev, id):
  16     dev.dump_monitor()
  17     dev.select_network(id, freq="2412")
  18
  19 def wait_ibss_connection(dev):
  20     logger.info(dev.ifname + " waiting for IBSS start/join to complete")
  21     ev = dev.wait_connected(timeout=20,
  22                             error="Connection to the IBSS timed out")
  23     exp = r'<.>(CTRL-EVENT-CONNECTED) - Connection to ([0-9a-f:]*) completed.*'
  24     s = re.split(exp, ev)
  25     if len(s) < 3:
  26         return None
  27     return s[2]
  28
  29 def wait_4way_handshake(dev1, dev2):
  30     logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr())
  31     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr()],
  32                          timeout=20)
  33     if ev is None:
  34         raise Exception("4-way handshake in IBSS timed out")
  35
  36 def wait_4way_handshake2(dev1, dev2, dev3):
  37     logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr() + " and " + dev3.p2p_interface_addr())
  38     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  39                           "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  40                          timeout=20)
  41     if ev is None:
  42         raise Exception("4-way handshake in IBSS timed out")
  43     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  44                           "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  45                          timeout=20)
  46     if ev is None:
  47         raise Exception("4-way handshake in IBSS timed out")
  48
  49 def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None,
  50              group=None, beacon_int=None, bssid=None, scan_freq=None):
  51     id = dev.add_network()
  52     dev.set_network(id, "mode", "1")
  53     dev.set_network(id, "frequency", "2412")
  54     if scan_freq:
  55         dev.set_network(id, "scan_freq", str(scan_freq))
  56     dev.set_network_quoted(id, "ssid", ssid)
  57     if psk:
  58         dev.set_network_quoted(id, "psk", psk)
  59     if proto:
  60         dev.set_network(id, "proto", proto)
  61     if key_mgmt:
  62         dev.set_network(id, "key_mgmt", key_mgmt)
  63     if pairwise:
  64         dev.set_network(id, "pairwise", pairwise)
  65     if group:
  66         dev.set_network(id, "group", group)
  67     if beacon_int:
  68         dev.set_network(id, "beacon_int", beacon_int)
  69     if bssid:
  70         dev.set_network(id, "bssid", bssid)
  71     dev.request("ENABLE_NETWORK " + str(id) + " no-connect")
  72     return id
  73
  74 def add_ibss_rsn(dev, ssid):
  75     return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP")
  76
  77 def add_ibss_wpa_none(dev, ssid):
  78     return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP")
  79
  80 def add_ibss_wpa_none_ccmp(dev, ssid):
  81     return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "CCMP", "CCMP")
  82
  83 def test_ibss_rsn(dev):
  84     """IBSS RSN"""
  85     ssid="ibss-rsn"
  86
  87     logger.info("Start IBSS on the first STA")
  88     id = add_ibss_rsn(dev[0], ssid)
  89     connect_ibss_cmd(dev[0], id)
  90     bssid0 = wait_ibss_connection(dev[0])
  91
  92     logger.info("Join two STAs to the IBSS")
  93
  94     id = add_ibss_rsn(dev[1], ssid)
  95     connect_ibss_cmd(dev[1], id)
  96     bssid1 = wait_ibss_connection(dev[1])
  97     if bssid0 != bssid1:
  98         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
  99         # try to merge with a scan
 100         dev[1].scan()
 101     wait_4way_handshake(dev[0], dev[1])
 102     wait_4way_handshake(dev[1], dev[0])
 103
 104     id = add_ibss_rsn(dev[2], ssid)
 105     connect_ibss_cmd(dev[2], id)
 106     bssid2 = wait_ibss_connection(dev[2])
 107     if bssid0 != bssid2:
 108         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 109         # try to merge with a scan
 110         dev[2].scan()
 111     wait_4way_handshake(dev[0], dev[2])
 112     wait_4way_handshake2(dev[2], dev[0], dev[1])
 113
 114     # Allow some time for all peers to complete key setup
 115     time.sleep(3)
 116     hwsim_utils.test_connectivity(dev[0], dev[1])
 117     hwsim_utils.test_connectivity(dev[0], dev[2])
 118     hwsim_utils.test_connectivity(dev[1], dev[2])
 119
 120     dev[1].request("REMOVE_NETWORK all")
 121     time.sleep(1)
 122     id = add_ibss_rsn(dev[1], ssid)
 123     connect_ibss_cmd(dev[1], id)
 124     bssid1 = wait_ibss_connection(dev[1])
 125     if bssid0 != bssid1:
 126         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 127         # try to merge with a scan
 128         dev[1].scan()
 129     wait_4way_handshake(dev[0], dev[1])
 130     wait_4way_handshake(dev[1], dev[0])
 131     time.sleep(3)
 132     hwsim_utils.test_connectivity(dev[0], dev[1])
 133
 134     if "OK" not in dev[0].request("IBSS_RSN " + dev[1].p2p_interface_addr()):
 135         raise Exception("IBSS_RSN command failed")
 136
 137 def test_ibss_wpa_none(dev):
 138     """IBSS WPA-None"""
 139     ssid="ibss-wpa-none"
 140
 141     logger.info("Start IBSS on the first STA")
 142     id = add_ibss_wpa_none(dev[0], ssid)
 143     connect_ibss_cmd(dev[0], id)
 144     bssid0 = wait_ibss_connection(dev[0])
 145
 146     # This is a bit ugly, but no one really cares about WPA-None, so there may
 147     # not be enough justification to clean this up.. For now, wpa_supplicant
 148     # will show two connection events with mac80211_hwsim where the first one
 149     # comes with all zeros address.
 150     if bssid0 == "00:00:00:00:00:00":
 151         logger.info("Waiting for real BSSID on the first STA")
 152         bssid0 = wait_ibss_connection(dev[0])
 153
 154     logger.info("Join two STAs to the IBSS")
 155
 156     id = add_ibss_wpa_none(dev[1], ssid)
 157     connect_ibss_cmd(dev[1], id)
 158     id = add_ibss_wpa_none(dev[2], ssid)
 159     connect_ibss_cmd(dev[2], id)
 160
 161     bssid1 = wait_ibss_connection(dev[1])
 162     if bssid0 != bssid1:
 163         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 164         bssid1 = wait_ibss_connection(dev[1])
 165
 166     bssid2 = wait_ibss_connection(dev[2])
 167     if bssid0 != bssid2:
 168         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 169         bssid2 = wait_ibss_connection(dev[2])
 170
 171     print bssid0
 172     print bssid1
 173     print bssid2
 174
 175     bss = dev[0].get_bss(bssid0)
 176     if not bss:
 177         bss = dev[1].get_bss(bssid1)
 178         if not bss:
 179             raise Exception("Could not find BSS entry for IBSS")
 180     if 'flags' not in bss:
 181         raise Exception("Could not get BSS flags from BSS table")
 182     if "[WPA-None-TKIP]" not in bss['flags']:
 183         raise Exception("Unexpected BSS flags: " + bss['flags'])
 184
 185     # Allow some time for all peers to complete key setup
 186     time.sleep(1)
 187
 188     # This is supposed to work, but looks like WPA-None does not work with
 189     # mac80211 currently..
 190     try:
 191         hwsim_utils.test_connectivity(dev[0], dev[1])
 192     except Exception, e:
 193         logger.info("Ignoring known connectivity failure: " + str(e))
 194     try:
 195         hwsim_utils.test_connectivity(dev[0], dev[2])
 196     except Exception, e:
 197         logger.info("Ignoring known connectivity failure: " + str(e))
 198     try:
 199         hwsim_utils.test_connectivity(dev[1], dev[2])
 200     except Exception, e:
 201         logger.info("Ignoring known connectivity failure: " + str(e))
 202
 203 def test_ibss_wpa_none_ccmp(dev):
 204     """IBSS WPA-None/CCMP"""
 205     ssid="ibss-wpa-none"
 206
 207     logger.info("Start IBSS on the first STA")
 208     id = add_ibss_wpa_none(dev[0], ssid)
 209     connect_ibss_cmd(dev[0], id)
 210     bssid0 = wait_ibss_connection(dev[0])
 211
 212     # This is a bit ugly, but no one really cares about WPA-None, so there may
 213     # not be enough justification to clean this up.. For now, wpa_supplicant
 214     # will show two connection events with mac80211_hwsim where the first one
 215     # comes with all zeros address.
 216     if bssid0 == "00:00:00:00:00:00":
 217         logger.info("Waiting for real BSSID on the first STA")
 218         bssid0 = wait_ibss_connection(dev[0])
 219
 220
 221     logger.info("Join a STA to the IBSS")
 222     id = add_ibss_wpa_none(dev[1], ssid)
 223     connect_ibss_cmd(dev[1], id)
 224
 225     bssid1 = wait_ibss_connection(dev[1])
 226     if bssid0 != bssid1:
 227         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 228         bssid1 = wait_ibss_connection(dev[1])
 229
 230     print bssid0
 231     print bssid1
 232
 233     # Allow some time for all peers to complete key setup
 234     time.sleep(1)
 235
 236     # This is supposed to work, but looks like WPA-None does not work with
 237     # mac80211 currently..
 238     try:
 239         hwsim_utils.test_connectivity(dev[0], dev[1])
 240     except Exception, e:
 241         logger.info("Ignoring known connectivity failure: " + str(e))
 242
 243 def test_ibss_open(dev):
 244     """IBSS open (no security)"""
 245     ssid="ibss"
 246     id = add_ibss(dev[0], ssid, key_mgmt="NONE", beacon_int="150")
 247     connect_ibss_cmd(dev[0], id)
 248     bssid0 = wait_ibss_connection(dev[0])
 249
 250     id = add_ibss(dev[1], ssid, key_mgmt="NONE", beacon_int="200")
 251     connect_ibss_cmd(dev[1], id)
 252     bssid1 = wait_ibss_connection(dev[1])
 253     if bssid0 != bssid1:
 254         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 255
 256     res = dev[0].request("SCAN_RESULTS")
 257     if "[IBSS]" not in res:
 258         res = dev[1].request("SCAN_RESULTS")
 259         if "[IBSS]" not in res:
 260             raise Exception("IBSS flag missing from scan results: " + res)
 261     bss = dev[0].get_bss(bssid0)
 262     if not bss:
 263         bss = dev[1].get_bss(bssid1)
 264         if not bss:
 265             raise Exception("Could not find BSS entry for IBSS")
 266     if 'flags' not in bss:
 267         raise Exception("Could not get BSS flags from BSS table")
 268     if "[IBSS]" not in bss['flags']:
 269         raise Exception("Unexpected BSS flags: " + bss['flags'])
 270
 271 def test_ibss_open_fixed_bssid(dev):
 272     """IBSS open (no security) and fixed BSSID"""
 273     ssid="ibss"
 274     bssid="02:11:22:33:44:55"
 275     try:
 276         dev[0].request("AP_SCAN 2")
 277         add_ibss(dev[0], ssid, key_mgmt="NONE", bssid=bssid, beacon_int="150")
 278         dev[0].request("REASSOCIATE")
 279
 280         dev[1].request("AP_SCAN 2")
 281         add_ibss(dev[1], ssid, key_mgmt="NONE", bssid=bssid, beacon_int="200")
 282         dev[1].request("REASSOCIATE")
 283
 284         bssid0 = wait_ibss_connection(dev[0])
 285         bssid1 = wait_ibss_connection(dev[1])
 286         if bssid0 != bssid:
 287             raise Exception("STA0 BSSID " + bssid0 + " differs from fixed BSSID " + bssid)
 288         if bssid1 != bssid:
 289             raise Exception("STA0 BSSID " + bssid0 + " differs from fixed BSSID " + bssid)
 290     finally:
 291         dev[0].request("AP_SCAN 1")
 292         dev[1].request("AP_SCAN 1")
 293
 294 def test_ibss_open_retry(dev):
 295     """IBSS open (no security) with cfg80211 retry workaround"""
 296     subprocess.check_call(['iw', 'dev', dev[0].ifname, 'set', 'type', 'adhoc'])
 297     subprocess.check_call(['iw', 'dev', dev[0].ifname, 'ibss', 'join',
 298                            'ibss-test', '2412', 'HT20', 'fixed-freq',
 299                            '02:22:33:44:55:66'])
 300     ssid="ibss"
 301     try:
 302         dev[0].request("AP_SCAN 2")
 303         id = add_ibss(dev[0], ssid, key_mgmt="NONE", beacon_int="150",
 304                       bssid="02:33:44:55:66:77", scan_freq=2412)
 305         #connect_ibss_cmd(dev[0], id)
 306         dev[0].request("REASSOCIATE")
 307         bssid0 = wait_ibss_connection(dev[0])
 308
 309         subprocess.check_call(['iw', 'dev', dev[0].ifname, 'ibss', 'leave'])
 310         time.sleep(1)
 311         dev[0].request("DISCONNECT")
 312     finally:
 313         dev[0].request("AP_SCAN 1")