mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-25 08:48:31 -05:00
115 lines
3.9 KiB
Python
115 lines
3.9 KiB
Python
#!/usr/bin/python
|
|
#
|
|
# P2P autonomous GO test cases
|
|
# Copyright (c) 2013, 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 subprocess
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
import hwsim_utils
|
|
|
|
def autogo(go):
|
|
logger.info("Start autonomous GO " + go.ifname)
|
|
res = go.p2p_start_go()
|
|
logger.debug("res: " + str(res))
|
|
|
|
def connect_cli(go, client):
|
|
logger.info("Try to connect the client to the GO")
|
|
pin = client.wps_read_pin()
|
|
go.p2p_go_authorize_client(pin)
|
|
client.p2p_connect_group(go.p2p_dev_addr(), pin, timeout=60)
|
|
logger.info("Client connected")
|
|
hwsim_utils.test_connectivity_p2p(go, client)
|
|
|
|
def test_autogo(dev):
|
|
"""P2P autonomous GO and client joining group"""
|
|
autogo(dev[0])
|
|
connect_cli(dev[0], dev[1])
|
|
dev[0].remove_group()
|
|
try:
|
|
dev[1].remove_group()
|
|
except:
|
|
pass
|
|
|
|
def test_autogo_2cli(dev):
|
|
"""P2P autonomous GO and two clients joining group"""
|
|
autogo(dev[0])
|
|
connect_cli(dev[0], dev[1])
|
|
connect_cli(dev[0], dev[2])
|
|
hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
|
|
dev[2].remove_group()
|
|
dev[1].remove_group()
|
|
dev[0].remove_group()
|
|
|
|
def wlantest_tdls(field, bssid, addr1, addr2):
|
|
res = subprocess.check_output(["../../wlantest/wlantest_cli",
|
|
"get_tdls_counter", field, bssid, addr1,
|
|
addr2]);
|
|
if "FAIL" in res:
|
|
raise Exception("wlantest_cli command failed")
|
|
return int(res)
|
|
|
|
def wlantest_tdls_clear(bssid, addr1, addr2):
|
|
subprocess.call(["../../wlantest/wlantest_cli",
|
|
"clear_tdls_counters", bssid, addr1, addr2]);
|
|
|
|
def test_autogo_tdls(dev):
|
|
"""P2P autonomous GO and two clients using TDLS"""
|
|
go = dev[0]
|
|
logger.info("Start autonomous GO with fixed parameters " + go.ifname)
|
|
id = go.add_network()
|
|
go.set_network_quoted(id, "ssid", "DIRECT-tdls")
|
|
go.set_network_quoted(id, "psk", "12345678")
|
|
go.set_network(id, "mode", "3")
|
|
go.set_network(id, "disabled", "2")
|
|
res = go.p2p_start_go(persistent=id)
|
|
logger.debug("res: " + str(res))
|
|
subprocess.call(["../../wlantest/wlantest_cli", "flush"]);
|
|
subprocess.call(["../../wlantest/wlantest_cli", "add_passphrase",
|
|
"12345678"]);
|
|
connect_cli(go, dev[1])
|
|
connect_cli(go, dev[2])
|
|
hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
|
|
bssid = dev[0].p2p_interface_addr()
|
|
addr1 = dev[1].p2p_interface_addr()
|
|
addr2 = dev[2].p2p_interface_addr()
|
|
dev[1].tdls_setup(addr2)
|
|
time.sleep(1)
|
|
hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
|
|
conf = wlantest_tdls("setup_conf_ok", bssid, addr1, addr2);
|
|
if conf == 0:
|
|
raise Exception("No TDLS Setup Confirm (success) seen")
|
|
dl = wlantest_tdls("valid_direct_link", bssid, addr1, addr2);
|
|
if dl == 0:
|
|
raise Exception("No valid frames through direct link")
|
|
wlantest_tdls_clear(bssid, addr1, addr2);
|
|
dev[1].tdls_teardown(addr2)
|
|
time.sleep(1)
|
|
teardown = wlantest_tdls("teardown", bssid, addr1, addr2);
|
|
if conf == teardown:
|
|
raise Exception("No TDLS Setup Teardown seen")
|
|
wlantest_tdls_clear(bssid, addr1, addr2);
|
|
hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
|
|
ap_path = wlantest_tdls("valid_ap_path", bssid, addr1, addr2);
|
|
if ap_path == 0:
|
|
raise Exception("No valid frames via AP path")
|
|
direct_link = wlantest_tdls("valid_direct_link", bssid, addr1, addr2);
|
|
if direct_link > 0:
|
|
raise Exception("Unexpected frames through direct link")
|
|
idirect_link = wlantest_tdls("invalid_direct_link", bssid, addr1, addr2);
|
|
if idirect_link > 0:
|
|
raise Exception("Unexpected frames through direct link (invalid)")
|
|
dev[2].remove_group()
|
|
dev[1].remove_group()
|
|
dev[0].remove_group()
|
|
|
|
def add_tests(tests):
|
|
tests.append(test_autogo)
|
|
tests.append(test_autogo_2cli)
|
|
tests.append(test_autogo_tdls)
|