tests/hwsim/test_ibss.py

   1 #!/usr/bin/python
   2 #
   3 # IBSS test cases
   4 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   5 #
   6 # This software may be distributed under the terms of the BSD license.
   7 # See README for more details.
   8
   9 import logging
  10 logger = logging.getLogger()
  11 import time
  12 import re
  13
  14 import hwsim_utils
  15
  16 def connect_ibss_cmd(dev, id):
  17     dev.dump_monitor()
  18     dev.select_network(id)
  19
  20 def wait_ibss_connection(dev):
  21     logger.info(dev.ifname + " waiting for IBSS start/join to complete")
  22     ev = dev.wait_event(["CTRL-EVENT-CONNECTED"], timeout=20)
  23     if ev is None:
  24         raise Exception("Connection to the IBSS timed out")
  25     exp = r'<.>(CTRL-EVENT-CONNECTED) - Connection to ([0-9a-f:]*) completed.*'
  26     s = re.split(exp, ev)
  27     if len(s) < 3:
  28         return None
  29     return s[2]
  30
  31 def wait_4way_handshake(dev1, dev2):
  32     logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr())
  33     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr()],
  34                          timeout=20)
  35     if ev is None:
  36         raise Exception("4-way handshake in IBSS timed out")
  37
  38 def wait_4way_handshake2(dev1, dev2, dev3):
  39     logger.info(dev1.ifname + " waiting for 4-way handshake completion with " + dev2.ifname + " " + dev2.p2p_interface_addr() + " and " + dev3.p2p_interface_addr())
  40     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  41                           "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  42                          timeout=20)
  43     if ev is None:
  44         raise Exception("4-way handshake in IBSS timed out")
  45     ev = dev1.wait_event(["IBSS-RSN-COMPLETED " + dev2.p2p_interface_addr(),
  46                           "IBSS-RSN-COMPLETED " + dev3.p2p_interface_addr()],
  47                          timeout=20)
  48     if ev is None:
  49         raise Exception("4-way handshake in IBSS timed out")
  50
  51 def add_ibss(dev, ssid, psk=None, proto=None, key_mgmt=None, pairwise=None, group=None):
  52     id = dev.add_network()
  53     dev.set_network(id, "mode", "1")
  54     dev.set_network(id, "frequency", "2412")
  55     dev.set_network_quoted(id, "ssid", ssid)
  56     if psk:
  57         dev.set_network_quoted(id, "psk", psk)
  58     if proto:
  59         dev.set_network(id, "proto", proto)
  60     if key_mgmt:
  61         dev.set_network(id, "key_mgmt", key_mgmt)
  62     if pairwise:
  63         dev.set_network(id, "pairwise", pairwise)
  64     if group:
  65         dev.set_network(id, "group", group)
  66     return id
  67
  68 def add_ibss_rsn(dev, ssid):
  69     return add_ibss(dev, ssid, "12345678", "RSN", "WPA-PSK", "CCMP", "CCMP")
  70
  71 def add_ibss_wpa_none(dev, ssid):
  72     return add_ibss(dev, ssid, "12345678", "WPA", "WPA-NONE", "TKIP", "TKIP")
  73
  74 def test_ibss_rsn(dev):
  75     """IBSS RSN"""
  76     ssid="ibss-rsn"
  77
  78     logger.info("Start IBSS on the first STA")
  79     id = add_ibss_rsn(dev[0], ssid)
  80     connect_ibss_cmd(dev[0], id)
  81     bssid0 = wait_ibss_connection(dev[0])
  82
  83     logger.info("Join two STAs to the IBSS")
  84
  85     id = add_ibss_rsn(dev[1], ssid)
  86     connect_ibss_cmd(dev[1], id)
  87     bssid1 = wait_ibss_connection(dev[1])
  88     if bssid0 != bssid1:
  89         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
  90         # try to merge with a scan
  91         dev[1].scan()
  92     wait_4way_handshake(dev[0], dev[1])
  93     wait_4way_handshake(dev[1], dev[0])
  94
  95     id = add_ibss_rsn(dev[2], ssid)
  96     connect_ibss_cmd(dev[2], id)
  97     bssid2 = wait_ibss_connection(dev[2])
  98     if bssid0 != bssid2:
  99         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 100         # try to merge with a scan
 101         dev[2].scan()
 102     wait_4way_handshake(dev[0], dev[2])
 103     wait_4way_handshake2(dev[2], dev[0], dev[1])
 104
 105     # Allow some time for all peers to complete key setup
 106     time.sleep(3)
 107     hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 108     hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
 109     hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
 110
 111 def test_ibss_wpa_none(dev):
 112     """IBSS WPA-None"""
 113     ssid="ibss-wpa-none"
 114
 115     logger.info("Start IBSS on the first STA")
 116     id = add_ibss_wpa_none(dev[0], ssid)
 117     connect_ibss_cmd(dev[0], id)
 118     bssid0 = wait_ibss_connection(dev[0])
 119
 120     logger.info("Join two STAs to the IBSS")
 121
 122     id = add_ibss_wpa_none(dev[1], ssid)
 123     connect_ibss_cmd(dev[1], id)
 124     id = add_ibss_wpa_none(dev[2], ssid)
 125     connect_ibss_cmd(dev[2], id)
 126
 127     # This is a bit ugly, but no one really cares about WPA-None, so there may
 128     # not be enough justification to clean this up.. For now, wpa_supplicant
 129     # will show two connection events with mac80211_hwsim where the first one
 130     # comes with all zeros address.
 131     if bssid0 == "00:00:00:00:00:00":
 132         logger.info("Waiting for real BSSID on the first STA")
 133         bssid0 = wait_ibss_connection(dev[0])
 134
 135     bssid1 = wait_ibss_connection(dev[1])
 136     if bssid0 != bssid1:
 137         logger.info("STA0 BSSID " + bssid0 + " differs from STA1 BSSID " + bssid1)
 138         bssid1 = wait_ibss_connection(dev[1])
 139
 140     bssid2 = wait_ibss_connection(dev[2])
 141     if bssid0 != bssid2:
 142         logger.info("STA0 BSSID " + bssid0 + " differs from STA2 BSSID " + bssid2)
 143         bssid2 = wait_ibss_connection(dev[2])
 144
 145     print bssid0
 146     print bssid1
 147     print bssid2
 148
 149     # Allow some time for all peers to complete key setup
 150     time.sleep(1)
 151
 152     # This is supposed to work, but looks like WPA-None does not work with
 153     # mac80211 currently..
 154     try:
 155         hwsim_utils.test_connectivity(dev[0].ifname, dev[1].ifname)
 156     except Exception, e:
 157         logger.info("Ignoring known connectivity failure: " + str(e))
 158     try:
 159         hwsim_utils.test_connectivity(dev[0].ifname, dev[2].ifname)
 160     except Exception, e:
 161         logger.info("Ignoring known connectivity failure: " + str(e))
 162     try:
 163         hwsim_utils.test_connectivity(dev[1].ifname, dev[2].ifname)
 164     except Exception, e:
 165         logger.info("Ignoring known connectivity failure: " + str(e))