mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-12-01 11:48:23 -05:00
a00a3458ed
Check whether the unexpected BSS entry is based on having received a Beacon frame instead of Probe Response frame. While this test case is using a huge beacon_int value, it is still possible for mac80211_hwsim timing to work in a way that a Beacon frame is sent. That made this test case fail in some rare cases. Fix this by ignoring the BSS entry if it is based on Beacon frame instead of Probe Response frame. Signed-off-by: Jouni Malinen <jouni@qca.qualcomm.com>
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
# Testing utilities
|
|
# Copyright (c) 2013-2015, Jouni Malinen <j@w1.fi>
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
import binascii
|
|
import os
|
|
import struct
|
|
import time
|
|
import remotehost
|
|
|
|
def get_ifnames():
|
|
ifnames = []
|
|
with open("/proc/net/dev", "r") as f:
|
|
lines = f.readlines()
|
|
for l in lines:
|
|
val = l.split(':', 1)
|
|
if len(val) == 2:
|
|
ifnames.append(val[0].strip(' '))
|
|
return ifnames
|
|
|
|
class HwsimSkip(Exception):
|
|
def __init__(self, reason):
|
|
self.reason = reason
|
|
def __str__(self):
|
|
return self.reason
|
|
|
|
class alloc_fail(object):
|
|
def __init__(self, dev, count, funcs):
|
|
self._dev = dev
|
|
self._count = count
|
|
self._funcs = funcs
|
|
def __enter__(self):
|
|
cmd = "TEST_ALLOC_FAIL %d:%s" % (self._count, self._funcs)
|
|
if "OK" not in self._dev.request(cmd):
|
|
raise HwsimSkip("TEST_ALLOC_FAIL not supported")
|
|
def __exit__(self, type, value, traceback):
|
|
if type is None:
|
|
if self._dev.request("GET_ALLOC_FAIL") != "0:%s" % self._funcs:
|
|
raise Exception("Allocation failure did not trigger")
|
|
|
|
class fail_test(object):
|
|
def __init__(self, dev, count, funcs):
|
|
self._dev = dev
|
|
self._count = count
|
|
self._funcs = funcs
|
|
def __enter__(self):
|
|
cmd = "TEST_FAIL %d:%s" % (self._count, self._funcs)
|
|
if "OK" not in self._dev.request(cmd):
|
|
raise HwsimSkip("TEST_FAIL not supported")
|
|
def __exit__(self, type, value, traceback):
|
|
if type is None:
|
|
if self._dev.request("GET_FAIL") != "0:%s" % self._funcs:
|
|
raise Exception("Test failure did not trigger")
|
|
|
|
def wait_fail_trigger(dev, cmd, note="Failure not triggered"):
|
|
for i in range(0, 40):
|
|
if dev.request(cmd).startswith("0:"):
|
|
break
|
|
if i == 39:
|
|
raise Exception(note)
|
|
time.sleep(0.05)
|
|
|
|
def require_under_vm():
|
|
with open('/proc/1/cmdline', 'r') as f:
|
|
cmd = f.read()
|
|
if "inside.sh" not in cmd:
|
|
raise HwsimSkip("Not running under VM")
|
|
|
|
def iface_is_in_bridge(bridge, ifname):
|
|
fname = "/sys/class/net/"+ifname+"/brport/bridge"
|
|
if not os.path.exists(fname):
|
|
return False
|
|
if not os.path.islink(fname):
|
|
return False
|
|
truebridge = os.path.basename(os.readlink(fname))
|
|
if bridge == truebridge:
|
|
return True
|
|
return False
|
|
|
|
def skip_with_fips(dev, reason="Not supported in FIPS mode"):
|
|
res = dev.get_capability("fips")
|
|
if res and 'FIPS' in res:
|
|
raise HwsimSkip(reason)
|
|
|
|
def get_phy(ap, ifname=None):
|
|
phy = "phy3"
|
|
try:
|
|
hostname = ap['hostname']
|
|
except:
|
|
hostname = None
|
|
host = remotehost.Host(hostname)
|
|
|
|
if ifname == None:
|
|
ifname = ap['ifname']
|
|
status, buf = host.execute(["iw", "dev", ifname, "info"])
|
|
if status != 0:
|
|
raise Exception("iw " + ifname + " info failed")
|
|
lines = buf.split("\n")
|
|
for line in lines:
|
|
if "wiphy" in line:
|
|
words = line.split()
|
|
phy = "phy" + words[1]
|
|
break
|
|
return phy
|
|
|
|
def parse_ie(buf):
|
|
ret = {}
|
|
data = binascii.unhexlify(buf)
|
|
while len(data) >= 2:
|
|
ie,elen = struct.unpack('BB', data[0:2])
|
|
data = data[2:]
|
|
if elen > len(data):
|
|
break
|
|
ret[ie] = data[0:elen]
|
|
data = data[elen:]
|
|
return ret
|