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
  12 import hwsim_utils
  13
  14 def connect_ibss_cmd(dev, id):
  15     dev.dump_monitor()
  16     dev.select_network(id, freq="2412")
  17
  18 def wait_ibss_connection(dev):
  19     logger.info(dev.ifname + " waiting for IBSS start/join to complete")
  20     ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=20)
  21     if ev is None:
  22         raise Exception("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, group=None, beacon_int=None, bssid=None):
  50     id = dev.add_network()
  51     dev.set_network(id, "mode", "1")
  52     dev.set_network(id, "frequency", "2412")
  53     dev.set_network_quoted(id, "ssid", ssid)
  54     if psk:
  55         dev.set_network_quoted(id, "psk", psk)
  56     if proto:
  57         dev.set_network(id, "proto", proto)
  58     if key_mgmt:
  59         dev.set_network(id, "key_mgmt", key_mgmt)
  60     if pairwise:
  61         dev.set_network(id, "pairwise", pairwise)
  62     if group:
  63         dev.set_network(id, "group", group)
  64     if beacon_int:
  65         dev.set_network(id, "beacon_int", beacon_int)
  66     if bssid:
  67         dev.set_network(id, "bssid", bssid)
  68     dev.request("ENABLE_NETWORK " + str(id) + " no-connect")
  69     return id
  70
  71 def add_ibss_rsn(dev, ssid):
  72     return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP")
  73
  74 def add_ibss_wpa_none(dev, ssid):
  75     return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP")
  76
  77 def add_ibss_wpa_none_ccmp(dev, ssid):
  78     return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "CCMP", "CCMP")
  79
  80 def test_ibss_rsn(dev):
  81     """IBSS RSN"""
  82     ssid="ibss-rsn"
  83
  84     logger.info("Start IBSS on the first STA")
  85     id = add_ibss_rsn(dev[0], ssid)
  86     connect_ibss_cmd(dev[0], id)
  87     bssid0 = wait_ibss_connection(dev[0])
  88
  89     logger.info("Join two STAs to the IBSS")
  90
  91     id = add_ibss_rsn(dev[1], ssid)
  92     connect_ibss_cmd(dev[1], id)
  93     bssid1 = wait_ibss_connection(dev[1])
  94     if bssid0 != bssid1:
  95         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
  96         # try to merge with a scan
  97         dev[1].scan()
  98     wait_4way_handshake(dev[0], dev[1])
  99     wait_4way_handshake(dev[1], dev[0])
 100
 101     id = add_ibss_rsn(dev[2], ssid)
 102     connect_ibss_cmd(dev[2], id)
 103     bssid2 = wait_ibss_connection(dev[2])
 104     if bssid0 != bssid2:
 105         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 106         # try to merge with a scan
 107         dev[2].scan()
 108     wait_4way_handshake(dev[0], dev[2])
 109     wait_4way_handshake2(dev[2], dev[0], dev[1])
 110
 111     # Allow some time for all peers to complete key setup
 112     time.sleep(3)
 113     hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 114     hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
 115     hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
 116
 117     dev[1].request("REMOVE_NETWORK all")
 118     time.sleep(1)
 119     id = add_ibss_rsn(dev[1], ssid)
 120     connect_ibss_cmd(dev[1], id)
 121     bssid1 = wait_ibss_connection(dev[1])
 122     if bssid0 != bssid1:
 123         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 124         # try to merge with a scan
 125         dev[1].scan()
 126     wait_4way_handshake(dev[0], dev[1])
 127     wait_4way_handshake(dev[1], dev[0])
 128     time.sleep(3)
 129     hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 130
 131 def test_ibss_wpa_none(dev):
 132     """IBSS WPA-None"""
 133     ssid="ibss-wpa-none"
 134
 135     logger.info("Start IBSS on the first STA")
 136     id = add_ibss_wpa_none(dev[0], ssid)
 137     connect_ibss_cmd(dev[0], id)
 138     bssid0 = wait_ibss_connection(dev[0])
 139
 140     # This is a bit ugly, but no one really cares about WPA-None, so there may
 141     # not be enough justification to clean this up.. For now, wpa_supplicant
 142     # will show two connection events with mac80211_hwsim where the first one
 143     # comes with all zeros address.
 144     if bssid0 == "00:00:00:00:00:00":
 145         logger.info("Waiting for real BSSID on the first STA")
 146         bssid0 = wait_ibss_connection(dev[0])
 147
 148     logger.info("Join two STAs to the IBSS")
 149
 150     id = add_ibss_wpa_none(dev[1], ssid)
 151     connect_ibss_cmd(dev[1], id)
 152     id = add_ibss_wpa_none(dev[2], ssid)
 153     connect_ibss_cmd(dev[2], id)
 154
 155     bssid1 = wait_ibss_connection(dev[1])
 156     if bssid0 != bssid1:
 157         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 158         bssid1 = wait_ibss_connection(dev[1])
 159
 160     bssid2 = wait_ibss_connection(dev[2])
 161     if bssid0 != bssid2:
 162         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 163         bssid2 = wait_ibss_connection(dev[2])
 164
 165     print bssid0
 166     print bssid1
 167     print bssid2
 168
 169     # Allow some time for all peers to complete key setup
 170     time.sleep(1)
 171
 172     # This is supposed to work, but looks like WPA-None does not work with
 173     # mac80211 currently..
 174     try:
 175         hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 176     except Exception, e:
 177         logger.info("Ignoring known connectivity failure: " + str(e))
 178     try:
 179         hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
 180     except Exception, e:
 181         logger.info("Ignoring known connectivity failure: " + str(e))
 182     try:
 183         hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
 184     except Exception, e:
 185         logger.info("Ignoring known connectivity failure: " + str(e))
 186
 187 def test_ibss_wpa_none_ccmp(dev):
 188     """IBSS WPA-None/CCMP"""
 189     ssid="ibss-wpa-none"
 190
 191     logger.info("Start IBSS on the first STA")
 192     id = add_ibss_wpa_none(dev[0], ssid)
 193     connect_ibss_cmd(dev[0], id)
 194     bssid0 = wait_ibss_connection(dev[0])
 195
 196     # This is a bit ugly, but no one really cares about WPA-None, so there may
 197     # not be enough justification to clean this up.. For now, wpa_supplicant
 198     # will show two connection events with mac80211_hwsim where the first one
 199     # comes with all zeros address.
 200     if bssid0 == "00:00:00:00:00:00":
 201         logger.info("Waiting for real BSSID on the first STA")
 202         bssid0 = wait_ibss_connection(dev[0])
 203
 204
 205     logger.info("Join a STA to the IBSS")
 206     id = add_ibss_wpa_none(dev[1], ssid)
 207     connect_ibss_cmd(dev[1], id)
 208
 209     bssid1 = wait_ibss_connection(dev[1])
 210     if bssid0 != bssid1:
 211         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 212         bssid1 = wait_ibss_connection(dev[1])
 213
 214     print bssid0
 215     print bssid1
 216
 217     # Allow some time for all peers to complete key setup
 218     time.sleep(1)
 219
 220     # This is supposed to work, but looks like WPA-None does not work with
 221     # mac80211 currently..
 222     try:
 223         hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 224     except Exception, e:
 225         logger.info("Ignoring known connectivity failure: " + str(e))
 226
 227 def test_ibss_open(dev):
 228     """IBSS open (no security)"""
 229     ssid="ibss"
 230     id = add_ibss(dev[0], ssid, key_mgmt="NONE", beacon_int="150")
 231     connect_ibss_cmd(dev[0], id)
 232     bssid0 = wait_ibss_connection(dev[0])
 233
 234     id = add_ibss(dev[1], ssid, key_mgmt="NONE", beacon_int="200")
 235     connect_ibss_cmd(dev[1], id)
 236     bssid1 = wait_ibss_connection(dev[1])
 237     if bssid0 != bssid1:
 238         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 239
 240 def test_ibss_open_fixed_bssid(dev):
 241     """IBSS open (no security) and fixed BSSID"""
 242     ssid="ibss"
 243     bssid="02:11:22:33:44:55"
 244     try:
 245         dev[0].request("AP_SCAN 2")
 246         add_ibss(dev[0], ssid, key_mgmt="NONE", bssid=bssid, beacon_int="150")
 247         dev[0].request("REASSOCIATE")
 248
 249         dev[1].request("AP_SCAN 2")
 250         add_ibss(dev[1], ssid, key_mgmt="NONE", bssid=bssid, beacon_int="200")
 251         dev[1].request("REASSOCIATE")
 252
 253         bssid0 = wait_ibss_connection(dev[0])
 254         bssid1 = wait_ibss_connection(dev[1])
 255         if bssid0 != bssid:
 256             raise Exception("STA0 BSSID " + bssid0 + " differs from fixed BSSID " + bssid)
 257         if bssid1 != bssid:
 258             raise Exception("STA0 BSSID " + bssid0 + " differs from fixed BSSID " + bssid)
 259     finally:
 260         dev[0].request("AP_SCAN 1")
 261         dev[1].request("AP_SCAN 1")