tests/hwsim/test_cfg80211.py

   1 # cfg80211 test cases
   2 # Copyright (c) 2014, 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 from remotehost import remote_compatible
   8 import logging
   9 logger = logging.getLogger()
  10 import binascii
  11 import os
  12 import time
  13
  14 import hostapd
  15 import hwsim_utils
  16 from tshark import run_tshark
  17 from nl80211 import *
  18 from wpasupplicant import WpaSupplicant
  19
  20 def nl80211_command(dev, cmd, attr):
  21     res = dev.request("VENDOR ffffffff {} {}".format(nl80211_cmd[cmd],
  22                                                      binascii.hexlify(attr).decode()))
  23     if "FAIL" in res:
  24         raise Exception("nl80211 command failed")
  25     return binascii.unhexlify(res)
  26
  27 @remote_compatible
  28 def test_cfg80211_disassociate(dev, apdev):
  29     """cfg80211 disassociation command"""
  30     hapd = hostapd.add_ap(apdev[0], {"ssid": "open"})
  31     dev[0].connect("open", key_mgmt="NONE", scan_freq="2412")
  32     ev = hapd.wait_event(["AP-STA-CONNECTED"], timeout=5)
  33     if ev is None:
  34         raise Exception("No connection event received from hostapd")
  35
  36     ifindex = int(dev[0].get_driver_status_field("ifindex"))
  37     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
  38     attrs += build_nl80211_attr_u16('REASON_CODE', 1)
  39     attrs += build_nl80211_attr_mac('MAC', apdev[0]['bssid'])
  40     nl80211_command(dev[0], 'DISASSOCIATE', attrs)
  41
  42     ev = hapd.wait_event(["AP-STA-DISCONNECTED"], timeout=5)
  43     if ev is None:
  44         raise Exception("No disconnection event received from hostapd")
  45
  46 def nl80211_frame(dev, ifindex, frame, freq=None, duration=None, offchannel_tx_ok=False):
  47     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
  48     if freq is not None:
  49         attrs += build_nl80211_attr_u32('WIPHY_FREQ', freq)
  50     if duration is not None:
  51         attrs += build_nl80211_attr_u32('DURATION', duration)
  52     if offchannel_tx_ok:
  53         attrs += build_nl80211_attr_flag('OFFCHANNEL_TX_OK')
  54     attrs += build_nl80211_attr('FRAME', frame)
  55     return parse_nl80211_attrs(nl80211_command(dev, 'FRAME', attrs))
  56
  57 def nl80211_frame_wait_cancel(dev, ifindex, cookie):
  58     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
  59     attrs += build_nl80211_attr('COOKIE', cookie)
  60     return nl80211_command(dev, 'FRAME_WAIT_CANCEL', attrs)
  61
  62 def nl80211_remain_on_channel(dev, ifindex, freq, duration):
  63     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
  64     attrs += build_nl80211_attr_u32('WIPHY_FREQ', freq)
  65     attrs += build_nl80211_attr_u32('DURATION', duration)
  66     return nl80211_command(dev, 'REMAIN_ON_CHANNEL', attrs)
  67
  68 def test_cfg80211_tx_frame(dev, apdev, params):
  69     """cfg80211 offchannel TX frame command"""
  70
  71     dev[0].p2p_start_go(freq='2412')
  72     go = WpaSupplicant(dev[0].group_ifname)
  73     frame = binascii.unhexlify("d0000000020000000100" + go.own_addr().replace(':', '') + "02000000010000000409506f9a090001dd5e506f9a0902020025080401001f0502006414060500585804510b0906000200000000000b1000585804510b0102030405060708090a0b0d1d000200000000000108000000000000000000101100084465766963652041110500585804510bdd190050f204104a0001101012000200011049000600372a000120")
  74     ifindex = int(go.get_driver_status_field("ifindex"))
  75     res = nl80211_frame(go, ifindex, frame, freq=2422, duration=500,
  76                         offchannel_tx_ok=True)
  77     time.sleep(0.1)
  78
  79     # note: Uncommenting this seems to remove the incorrect channel issue
  80     #nl80211_frame_wait_cancel(dev[0], ifindex, res[nl80211_attr['COOKIE']])
  81
  82     # note: this Action frame ends up getting sent incorrectly on 2422 MHz
  83     nl80211_frame(go, ifindex, frame, freq=2412)
  84     time.sleep(1.5)
  85     # note: also the Deauthenticate frame sent by the GO going down ends up
  86     # being transmitted incorrectly on 2422 MHz.
  87
  88     del go
  89
  90     out = run_tshark(os.path.join(params['logdir'], "hwsim0.pcapng"),
  91                      "wlan.fc.type_subtype == 13", ["radiotap.channel.freq"])
  92     if out is not None:
  93         freq = out.splitlines()
  94         if len(freq) != 2:
  95             raise Exception("Unexpected number of Action frames (%d)" % len(freq))
  96         if freq[0] != "2422":
  97             raise Exception("First Action frame on unexpected channel: %s MHz" % freq[0])
  98         if freq[1] != "2412":
  99             raise Exception("Second Action frame on unexpected channel: %s MHz" % freq[1])
 100
 101 @remote_compatible
 102 def test_cfg80211_wep_key_idx_change(dev, apdev):
 103     """WEP Shared Key authentication and key index change without deauth"""
 104     hapd = hostapd.add_ap(apdev[0],
 105                           {"ssid": "wep-shared-key",
 106                            "wep_key0": '"hello12345678"',
 107                            "wep_key1": '"other12345678"',
 108                            "auth_algs": "2"})
 109     id = dev[0].connect("wep-shared-key", key_mgmt="NONE", auth_alg="SHARED",
 110                         wep_key0='"hello12345678"',
 111                         wep_key1='"other12345678"',
 112                         wep_tx_keyidx="0",
 113                         scan_freq="2412")
 114     hwsim_utils.test_connectivity(dev[0], hapd)
 115
 116     dev[0].set_network(id, "wep_tx_keyidx", "1")
 117
 118     # clear cfg80211 auth state to allow new auth without deauth frame
 119     ifindex = int(dev[0].get_driver_status_field("ifindex"))
 120     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
 121     attrs += build_nl80211_attr_u16('REASON_CODE', 1)
 122     attrs += build_nl80211_attr_mac('MAC', apdev[0]['bssid'])
 123     attrs += build_nl80211_attr_flag('LOCAL_STATE_CHANGE')
 124     nl80211_command(dev[0], 'DEAUTHENTICATE', attrs)
 125     dev[0].wait_disconnected(timeout=5, error="Local-deauth timed out")
 126
 127     # the previous command results in deauth event followed by auto-reconnect
 128     dev[0].wait_connected(timeout=10, error="Reassociation timed out")
 129     hwsim_utils.test_connectivity(dev[0], hapd)
 130
 131 @remote_compatible
 132 def test_cfg80211_hostapd_ext_sta_remove(dev, apdev):
 133     """cfg80211 DEL_STATION issued externally to hostapd"""
 134     hapd = hostapd.add_ap(apdev[0], {"ssid": "open"})
 135     id = dev[0].connect("open", key_mgmt="NONE", scan_freq="2412")
 136
 137     ifindex = int(hapd.get_driver_status_field("ifindex"))
 138     attrs = build_nl80211_attr_u32('IFINDEX', ifindex)
 139     attrs += build_nl80211_attr_u16('REASON_CODE', 1)
 140     attrs += build_nl80211_attr_u8('MGMT_SUBTYPE', 12)
 141     attrs += build_nl80211_attr_mac('MAC', dev[0].own_addr())
 142     nl80211_command(hapd, 'DEL_STATION', attrs)
 143
 144     # Currently, hostapd ignores the NL80211_CMD_DEL_STATION event if
 145     # drv->device_ap_sme == 0 (which is the case with mac80211_hwsim), so no
 146     # further action happens here. If that event were to be used to remove the
 147     # STA entry from hostapd even in device_ap_sme == 0 case, this test case
 148     # could be extended to cover additional operations.