mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-25 08:48:31 -05:00
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
# Radio work tests
|
|
# Copyright (c) 2014, Jouni Malinen <j@w1.fi>
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
import time
|
|
import logging
|
|
logger = logging.getLogger()
|
|
|
|
import hostapd
|
|
from wpasupplicant import WpaSupplicant
|
|
|
|
def test_ext_radio_work(dev, apdev):
|
|
"""External radio work item"""
|
|
id = dev[0].request("RADIO_WORK add test-work-a")
|
|
if "FAIL" in id:
|
|
raise Exception("Failed to add radio work")
|
|
id2 = dev[0].request("RADIO_WORK add test-work-b freq=2417")
|
|
if "FAIL" in id2:
|
|
raise Exception("Failed to add radio work")
|
|
id3 = dev[0].request("RADIO_WORK add test-work-c")
|
|
if "FAIL" in id3:
|
|
raise Exception("Failed to add radio work")
|
|
|
|
ev = dev[0].wait_event(["EXT-RADIO-WORK-START"])
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to start")
|
|
if "EXT-RADIO-WORK-START " + id not in ev:
|
|
raise Exception("Unexpected radio work start id")
|
|
|
|
items = dev[0].request("RADIO_WORK show")
|
|
if "ext:test-work-a@wlan0:0:1:" not in items:
|
|
logger.info("Pending radio work items:\n" + items)
|
|
raise Exception("Radio work item(a) missing from the list")
|
|
if "ext:test-work-b@wlan0:2417:0:" not in items:
|
|
logger.info("Pending radio work items:\n" + items)
|
|
raise Exception("Radio work item(b) missing from the list")
|
|
if "ext:test-work-c@wlan0:0:0:" not in items:
|
|
logger.info("Pending radio work items:\n" + items)
|
|
raise Exception("Radio work item(c) missing from the list")
|
|
|
|
dev[0].request("RADIO_WORK done " + id2)
|
|
dev[0].request("RADIO_WORK done " + id)
|
|
|
|
ev = dev[0].wait_event(["EXT-RADIO-WORK-START"])
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to start")
|
|
if "EXT-RADIO-WORK-START " + id3 not in ev:
|
|
raise Exception("Unexpected radio work start id")
|
|
dev[0].request("RADIO_WORK done " + id3)
|
|
items = dev[0].request("RADIO_WORK show")
|
|
if "ext:" in items:
|
|
logger.info("Pending radio work items:\n" + items)
|
|
raise Exception("Unexpected remaining radio work item")
|
|
|
|
id = dev[0].request("RADIO_WORK add test-work timeout=1")
|
|
ev = dev[0].wait_event(["EXT-RADIO-WORK-START"])
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to start")
|
|
ev = dev[0].wait_event(["EXT-RADIO-WORK-TIMEOUT"], timeout=2)
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to time out")
|
|
if id not in ev:
|
|
raise Exception("Radio work id mismatch")
|
|
|
|
for i in range(5):
|
|
dev[0].request(("RADIO_WORK add test-work-%d-" % i) + 100*'a')
|
|
ev = dev[0].wait_event(["EXT-RADIO-WORK-START"])
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to start")
|
|
if "FAIL" not in dev[0].request("RADIO_WORK done 12345678"):
|
|
raise Exception("Invalid RADIO_WORK done accepted")
|
|
if "FAIL" not in dev[0].request("RADIO_WORK foo"):
|
|
raise Exception("Invalid RADIO_WORK accepted")
|
|
dev[0].request("FLUSH")
|
|
items = dev[0].request("RADIO_WORK show")
|
|
if items != "":
|
|
raise Exception("Unexpected radio work remaining after FLUSH: " + items)
|
|
|
|
def test_radio_work_cancel(dev, apdev):
|
|
"""Radio work items cancelled on interface removal"""
|
|
params = hostapd.wpa2_params(ssid="radio", passphrase="12345678")
|
|
hostapd.add_ap(apdev[0], params)
|
|
wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
|
|
wpas.interface_add("wlan5")
|
|
wpas.scan(freq="2412")
|
|
|
|
id = wpas.request("RADIO_WORK add test-work-a")
|
|
if "FAIL" in id:
|
|
raise Exception("Failed to add radio work")
|
|
ev = wpas.wait_event(["EXT-RADIO-WORK-START"])
|
|
if ev is None:
|
|
raise Exception("Timeout while waiting radio work to start")
|
|
if "EXT-RADIO-WORK-START " + id not in ev:
|
|
raise Exception("Unexpected radio work start id")
|
|
|
|
wpas.connect("radio", psk="12345678", scan_freq="2412",
|
|
wait_connect=False)
|
|
time.sleep(1)
|
|
wpas.interface_remove("wlan5")
|
|
# add to allow log file renaming
|
|
wpas.interface_add("wlan5")
|