mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-25 08:48:31 -05:00
263 lines
12 KiB
Python
263 lines
12 KiB
Python
# hostapd control interface
|
|
# Copyright (c) 2014, Qualcomm Atheros, Inc.
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
import hostapd
|
|
|
|
def test_hapd_ctrl_status(dev, apdev):
|
|
"""hostapd ctrl_iface STATUS commands"""
|
|
ssid = "hapd-ctrl"
|
|
bssid = apdev[0]['bssid']
|
|
params = hostapd.wpa2_params(ssid=ssid, passphrase="12345678")
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
status = hapd.get_status()
|
|
driver = hapd.get_driver_status()
|
|
|
|
if status['bss[0]'] != apdev[0]['ifname']:
|
|
raise Exception("Unexpected bss[0]")
|
|
if status['ssid[0]'] != ssid:
|
|
raise Exception("Unexpected ssid[0]")
|
|
if status['bssid[0]'] != bssid:
|
|
raise Exception("Unexpected bssid[0]")
|
|
if status['freq'] != "2412":
|
|
raise Exception("Unexpected freq")
|
|
|
|
if driver['beacon_set'] != "1":
|
|
raise Exception("Unexpected beacon_set")
|
|
if driver['addr'] != bssid:
|
|
raise Exception("Unexpected addr")
|
|
|
|
def test_hapd_ctrl_p2p_manager(dev, apdev):
|
|
"""hostapd as P2P Device manager"""
|
|
ssid = "hapd-p2p-mgr"
|
|
passphrase = "12345678"
|
|
params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase)
|
|
params['manage_p2p'] = '1'
|
|
params['allow_cross_connection'] = '0'
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
dev[0].connect(ssid, psk=passphrase, scan_freq="2412")
|
|
addr = dev[0].p2p_dev_addr()
|
|
if "OK" not in hapd.request("DEAUTHENTICATE " + addr + " p2p=2"):
|
|
raise Exception("DEAUTHENTICATE command failed")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
|
|
if ev is None:
|
|
raise Exception("Disconnection event timed out")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Re-connection timed out")
|
|
|
|
if "OK" not in hapd.request("DISASSOCIATE " + addr + " p2p=2"):
|
|
raise Exception("DISASSOCIATE command failed")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
|
|
if ev is None:
|
|
raise Exception("Disconnection event timed out")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Re-connection timed out")
|
|
|
|
def test_hapd_ctrl_sta(dev, apdev):
|
|
"""hostapd and STA ctrl_iface commands"""
|
|
ssid = "hapd-ctrl-sta"
|
|
passphrase = "12345678"
|
|
params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase)
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
dev[0].connect(ssid, psk=passphrase, scan_freq="2412")
|
|
addr = dev[0].p2p_dev_addr()
|
|
if "FAIL" in hapd.request("STA " + addr):
|
|
raise Exception("Unexpected STA failure")
|
|
if "FAIL" not in hapd.request("STA " + addr + " eapol"):
|
|
raise Exception("Unexpected STA-eapol success")
|
|
if "FAIL" not in hapd.request("STA 00:11:22:33:44"):
|
|
raise Exception("Unexpected STA success")
|
|
if "FAIL" not in hapd.request("STA 00:11:22:33:44:55"):
|
|
raise Exception("Unexpected STA success")
|
|
|
|
if len(hapd.request("STA-NEXT " + addr).splitlines()) > 0:
|
|
raise Exception("Unexpected STA-NEXT result")
|
|
if "FAIL" not in hapd.request("STA-NEXT 00:11:22:33:44"):
|
|
raise Exception("Unexpected STA-NEXT success")
|
|
|
|
def test_hapd_ctrl_disconnect(dev, apdev):
|
|
"""hostapd and disconnection ctrl_iface commands"""
|
|
ssid = "hapd-ctrl"
|
|
passphrase = "12345678"
|
|
params = hostapd.wpa2_params(ssid=ssid, passphrase=passphrase)
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
dev[0].connect(ssid, psk=passphrase, scan_freq="2412")
|
|
addr = dev[0].p2p_dev_addr()
|
|
|
|
if "FAIL" not in hapd.request("DEAUTHENTICATE 00:11:22:33:44"):
|
|
raise Exception("Unexpected DEAUTHENTICATE success")
|
|
|
|
if "OK" not in hapd.request("DEAUTHENTICATE ff:ff:ff:ff:ff:ff"):
|
|
raise Exception("Unexpected DEAUTHENTICATE failure")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
|
|
if ev is None:
|
|
raise Exception("Disconnection event timed out")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Re-connection timed out")
|
|
|
|
if "FAIL" not in hapd.request("DISASSOCIATE 00:11:22:33:44"):
|
|
raise Exception("Unexpected DISASSOCIATE success")
|
|
|
|
if "OK" not in hapd.request("DISASSOCIATE ff:ff:ff:ff:ff:ff"):
|
|
raise Exception("Unexpected DISASSOCIATE failure")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=5)
|
|
if ev is None:
|
|
raise Exception("Disconnection event timed out")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-CONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Re-connection timed out")
|
|
|
|
def test_hapd_ctrl_chan_switch(dev, apdev):
|
|
"""hostapd and CHAN_SWITCH ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("CHAN_SWITCH "):
|
|
raise Exception("Unexpected CHAN_SWITCH success")
|
|
if "FAIL" not in hapd.request("CHAN_SWITCH qwerty 2422"):
|
|
raise Exception("Unexpected CHAN_SWITCH success")
|
|
if "FAIL" not in hapd.request("CHAN_SWITCH 5 qwerty"):
|
|
raise Exception("Unexpected CHAN_SWITCH success")
|
|
if "FAIL" not in hapd.request("CHAN_SWITCH 0 2432 center_freq1=123 center_freq2=234 bandwidth=1000 sec_channel_offset=20 ht vht"):
|
|
raise Exception("Unexpected CHAN_SWITCH success")
|
|
|
|
def test_hapd_ctrl_level(dev, apdev):
|
|
"""hostapd and LEVEL ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("LEVEL 0"):
|
|
raise Exception("Unexpected LEVEL success on non-monitor interface")
|
|
|
|
def test_hapd_ctrl_new_sta(dev, apdev):
|
|
"""hostapd and NEW_STA ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("NEW_STA 00:11:22:33:44"):
|
|
raise Exception("Unexpected NEW_STA success")
|
|
if "OK" not in hapd.request("NEW_STA 00:11:22:33:44:55"):
|
|
raise Exception("Unexpected NEW_STA failure")
|
|
if "AUTHORIZED" not in hapd.request("STA 00:11:22:33:44:55"):
|
|
raise Esception("Unexpected NEW_STA STA status")
|
|
|
|
def test_hapd_ctrl_get(dev, apdev):
|
|
"""hostapd and GET ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("GET foo"):
|
|
raise Exception("Unexpected GET success")
|
|
if "FAIL" in hapd.request("GET version"):
|
|
raise Exception("Unexpected GET version failure")
|
|
|
|
def test_hapd_ctrl_unknown(dev, apdev):
|
|
"""hostapd and unknown ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "UNKNOWN COMMAND" not in hapd.request("FOO"):
|
|
raise Exception("Unexpected response")
|
|
|
|
def test_hapd_ctrl_hs20_wnm_notif(dev, apdev):
|
|
"""hostapd and HS20_WNM_NOTIF ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("HS20_WNM_NOTIF 00:11:22:33:44 http://example.com/"):
|
|
raise Exception("Unexpected HS20_WNM_NOTIF success")
|
|
if "FAIL" not in hapd.request("HS20_WNM_NOTIF 00:11:22:33:44:55http://example.com/"):
|
|
raise Exception("Unexpected HS20_WNM_NOTIF success")
|
|
|
|
def test_hapd_ctrl_hs20_deauth_req(dev, apdev):
|
|
"""hostapd and HS20_DEAUTH_REQ ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("HS20_DEAUTH_REQ 00:11:22:33:44 1 120 http://example.com/"):
|
|
raise Exception("Unexpected HS20_DEAUTH_REQ success")
|
|
if "FAIL" not in hapd.request("HS20_DEAUTH_REQ 00:11:22:33:44:55"):
|
|
raise Exception("Unexpected HS20_DEAUTH_REQ success")
|
|
if "FAIL" not in hapd.request("HS20_DEAUTH_REQ 00:11:22:33:44:55 1"):
|
|
raise Exception("Unexpected HS20_DEAUTH_REQ success")
|
|
|
|
def test_hapd_ctrl_disassoc_imminent(dev, apdev):
|
|
"""hostapd and DISASSOC_IMMINENT ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("DISASSOC_IMMINENT 00:11:22:33:44"):
|
|
raise Exception("Unexpected DISASSOC_IMMINENT success")
|
|
if "FAIL" not in hapd.request("DISASSOC_IMMINENT 00:11:22:33:44:55"):
|
|
raise Exception("Unexpected DISASSOC_IMMINENT success")
|
|
if "FAIL" not in hapd.request("DISASSOC_IMMINENT 00:11:22:33:44:55 2"):
|
|
raise Exception("Unexpected DISASSOC_IMMINENT success")
|
|
dev[0].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
addr = dev[0].p2p_interface_addr()
|
|
if "OK" not in hapd.request("DISASSOC_IMMINENT " + addr + " 2"):
|
|
raise Exception("Unexpected DISASSOC_IMMINENT failure")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-SCAN-RESULTS"], 15)
|
|
if ev is None:
|
|
raise Exception("Scan timed out")
|
|
|
|
def test_hapd_ctrl_ess_disassoc(dev, apdev):
|
|
"""hostapd and ESS_DISASSOC ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
if "FAIL" not in hapd.request("ESS_DISASSOC 00:11:22:33:44"):
|
|
raise Exception("Unexpected ESS_DISASSOCT success")
|
|
if "FAIL" not in hapd.request("ESS_DISASSOC 00:11:22:33:44:55"):
|
|
raise Exception("Unexpected ESS_DISASSOC success")
|
|
dev[0].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
addr = dev[0].p2p_interface_addr()
|
|
if "FAIL" not in hapd.request("ESS_DISASSOC " + addr):
|
|
raise Exception("Unexpected ESS_DISASSOC success")
|
|
if "FAIL" not in hapd.request("ESS_DISASSOC " + addr + " -1"):
|
|
raise Exception("Unexpected ESS_DISASSOC success")
|
|
if "FAIL" not in hapd.request("ESS_DISASSOC " + addr + " 1"):
|
|
raise Exception("Unexpected ESS_DISASSOC success")
|
|
if "OK" not in hapd.request("ESS_DISASSOC " + addr + " 20 http://example.com/"):
|
|
raise Exception("Unexpected ESS_DISASSOC failure")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-SCAN-RESULTS"], 15)
|
|
if ev is None:
|
|
raise Exception("Scan timed out")
|
|
|
|
def test_hapd_ctrl_set_deny_mac_file(dev, apdev):
|
|
"""hostapd and SET deny_mac_file ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
dev[0].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
dev[1].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
if "OK" not in hapd.request("SET deny_mac_file hostapd.macaddr"):
|
|
raise Exception("Unexpected SET failure")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], 15)
|
|
if ev is None:
|
|
raise Exception("Disconnection timeout")
|
|
ev = dev[1].wait_event(["CTRL-EVENT-DISCONNECTED"], 1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected disconnection")
|
|
|
|
def test_hapd_ctrl_set_accept_mac_file(dev, apdev):
|
|
"""hostapd and SET accept_mac_file ctrl_iface command"""
|
|
ssid = "hapd-ctrl"
|
|
params = { "ssid": ssid }
|
|
hapd = hostapd.add_ap(apdev[0]['ifname'], params)
|
|
dev[0].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
dev[1].connect(ssid, key_mgmt="NONE", scan_freq="2412")
|
|
hapd.request("SET macaddr_acl 1")
|
|
if "OK" not in hapd.request("SET accept_mac_file hostapd.macaddr"):
|
|
raise Exception("Unexpected SET failure")
|
|
ev = dev[1].wait_event(["CTRL-EVENT-DISCONNECTED"], 15)
|
|
if ev is None:
|
|
raise Exception("Disconnection timeout")
|
|
ev = dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], 1)
|
|
if ev is not None:
|
|
raise Exception("Unexpected disconnection")
|