mirror of
https://github.com/vanhoefm/fragattacks.git
synced 2024-11-29 02:38:22 -05:00
17ffdf3951
It was possible for some of the SAE test cases (e.g., ap_ft_sae) to fail if they were run after the sae_groups test case that left the SAE group configuration to a value that is not enabled by default. Fix this by clearing sae_groups setting in the couple of test cases that were not yet doing this. Signed-off-by: Jouni Malinen <j@w1.fi>
341 lines
11 KiB
Python
341 lines
11 KiB
Python
#!/usr/bin/python
|
|
#
|
|
# wpa_supplicant mesh mode tests
|
|
# Copyright (c) 2014, cozybit Inc.
|
|
#
|
|
# This software may be distributed under the terms of the BSD license.
|
|
# See README for more details.
|
|
|
|
|
|
def check_mesh_scan(dev, params, other_started=False):
|
|
if not other_started:
|
|
dev.dump_monitor()
|
|
id = dev.request("SCAN " + params)
|
|
if "FAIL" in id:
|
|
raise Exception("Failed to start scan")
|
|
id = int(id)
|
|
|
|
if other_started:
|
|
ev = dev.wait_event(["CTRL-EVENT-SCAN-STARTED"])
|
|
if ev is None:
|
|
raise Exception("Other scan did not start")
|
|
if "id=" + str(id) in ev:
|
|
raise Exception("Own scan id unexpectedly included in start event")
|
|
|
|
ev = dev.wait_event(["CTRL-EVENT-SCAN-RESULTS"])
|
|
if ev is None:
|
|
raise Exception("Other scan did not complete")
|
|
if "id=" + str(id) in ev:
|
|
raise Exception(
|
|
"Own scan id unexpectedly included in completed event")
|
|
|
|
ev = dev.wait_event(["CTRL-EVENT-SCAN-STARTED"])
|
|
if ev is None:
|
|
raise Exception("Scan did not start")
|
|
if "id=" + str(id) not in ev:
|
|
raise Exception("Scan id not included in start event")
|
|
|
|
ev = dev.wait_event(["CTRL-EVENT-SCAN-RESULTS"])
|
|
if ev is None:
|
|
raise Exception("Scan did not complete")
|
|
if "id=" + str(id) not in ev:
|
|
raise Exception("Scan id not included in completed event")
|
|
|
|
res = dev.request("SCAN_RESULTS")
|
|
|
|
if res.find("[MESH]") < 0:
|
|
raise Exception("Scan did not contain a MESH network")
|
|
|
|
|
|
def check_mesh_group_added(dev):
|
|
ev = dev.wait_event(["MESH-GROUP-STARTED"])
|
|
if ev is None:
|
|
raise Exception("Test exception: Couldn't join mesh")
|
|
|
|
|
|
def check_mesh_group_removed(dev):
|
|
ev = dev.wait_event(["MESH-GROUP-REMOVED"])
|
|
if ev is None:
|
|
raise Exception("Test exception: Couldn't leave mesh")
|
|
|
|
|
|
def check_mesh_peer_connected(dev):
|
|
ev = dev.wait_event(["MESH-PEER-CONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Test exception: Remote peer did not connect.")
|
|
|
|
|
|
def check_mesh_peer_disconnected(dev):
|
|
ev = dev.wait_event(["MESH-PEER-DISCONNECTED"])
|
|
if ev is None:
|
|
raise Exception("Test exception: Peer disconnect event not detected.")
|
|
|
|
|
|
def test_wpas_add_set_remove_support(dev):
|
|
"""wpa_supplicant MESH add/set/remove network support"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].remove_network(id)
|
|
|
|
|
|
def test_wpas_mesh_group_added(dev):
|
|
"""wpa_supplicant MESH group add"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
# Check for MESH-GROUP-STARTED event
|
|
check_mesh_group_added(dev[0])
|
|
|
|
|
|
def test_wpas_mesh_group_remove(dev):
|
|
"""wpa_supplicant MESH group remove"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].mesh_group_add(id)
|
|
# Check for MESH-GROUP-STARTED event
|
|
check_mesh_group_added(dev[0])
|
|
dev[0].mesh_group_remove()
|
|
# Check for MESH-GROUP-REMOVED event
|
|
check_mesh_group_removed(dev[0])
|
|
|
|
|
|
def test_wpas_mesh_peer_connected(dev):
|
|
"""wpa_supplicant MESH peer connected"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[1].set_network(id, "key_mgmt", "NONE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
|
|
def test_wpas_mesh_peer_disconnected(dev):
|
|
"""wpa_supplicant MESH peer disconnected"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[1].set_network(id, "key_mgmt", "NONE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
# Remove group on dev 1
|
|
dev[1].mesh_group_remove()
|
|
# Device 0 should get a disconnection event
|
|
check_mesh_peer_disconnected(dev[0])
|
|
|
|
|
|
def test_wpas_mesh_mode_scan(dev):
|
|
"""wpa_supplicant MESH scan support"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].set_network(id, "mesh_ht_mode", "HT40+")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[1].set_network(id, "key_mgmt", "NONE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].set_network(id, "mesh_ht_mode", "HT40+")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for Mesh scan
|
|
check_mesh_scan(dev[0], "use_id=1")
|
|
|
|
|
|
def wrap_wpas_mesh_test(test, dev, apdev):
|
|
import hwsim_utils
|
|
|
|
def _test_connectivity(dev1, dev2):
|
|
return hwsim_utils.test_connectivity(dev1, dev2)
|
|
|
|
return test(dev, apdev, _test_connectivity)
|
|
|
|
|
|
def _test_wpas_mesh_open(dev, apdev, test_connectivity):
|
|
"""wpa_supplicant open MESH network connectivity"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].set_network(id, "mesh_ht_mode", "HT40+")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[1].set_network(id, "key_mgmt", "NONE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].set_network(id, "mesh_ht_mode", "HT40+")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
# Test connectivity 0->1 and 1->0
|
|
test_connectivity(dev[0], dev[1])
|
|
test_connectivity(dev[1], dev[0])
|
|
|
|
|
|
def test_wpas_mesh_open(dev, apdev):
|
|
return wrap_wpas_mesh_test(_test_wpas_mesh_open, dev, apdev)
|
|
|
|
|
|
def _test_wpas_mesh_open_no_auto(dev, apdev, test_connectivity):
|
|
"""wpa_supplicant open MESH network connectivity"""
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[0].set_network(id, "key_mgmt", "NONE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-open")
|
|
dev[1].set_network(id, "key_mgmt", "NONE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].set_network(id, "no_auto_peer", "1")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
# Test connectivity 0->1 and 1->0
|
|
test_connectivity(dev[0], dev[1])
|
|
test_connectivity(dev[1], dev[0])
|
|
|
|
|
|
def test_wpas_mesh_open_no_auto(dev, apdev):
|
|
return wrap_wpas_mesh_test(_test_wpas_mesh_open_no_auto, dev, apdev)
|
|
|
|
|
|
def _test_wpas_mesh_secure(dev, apdev, test_connectivity):
|
|
"""wpa_supplicant secure MESH network connectivity"""
|
|
dev[0].request("SET sae_groups ")
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-sec")
|
|
dev[0].set_network(id, "key_mgmt", "SAE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].set_network_quoted(id, "psk", "thisismypassphrase!")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
dev[1].request("SET sae_groups ")
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-sec")
|
|
dev[1].set_network(id, "key_mgmt", "SAE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].set_network_quoted(id, "psk", "thisismypassphrase!")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
# Test connectivity 0->1 and 1->0
|
|
test_connectivity(dev[0], dev[1])
|
|
test_connectivity(dev[1], dev[0])
|
|
|
|
|
|
def test_wpas_mesh_secure(dev, apdev):
|
|
return wrap_wpas_mesh_test(_test_wpas_mesh_secure, dev, apdev)
|
|
|
|
|
|
def _test_wpas_mesh_secure_no_auto(dev, apdev, test_connectivity):
|
|
"""wpa_supplicant secure MESH network connectivity"""
|
|
dev[0].request("SET sae_groups ")
|
|
id = dev[0].add_network()
|
|
dev[0].set_network(id, "mode", "5")
|
|
dev[0].set_network_quoted(id, "ssid", "wpas-mesh-sec")
|
|
dev[0].set_network(id, "key_mgmt", "SAE")
|
|
dev[0].set_network(id, "frequency", "2412")
|
|
dev[0].set_network_quoted(id, "psk", "thisismypassphrase!")
|
|
dev[0].mesh_group_add(id)
|
|
|
|
dev[1].request("SET sae_groups ")
|
|
id = dev[1].add_network()
|
|
dev[1].set_network(id, "mode", "5")
|
|
dev[1].set_network_quoted(id, "ssid", "wpas-mesh-sec")
|
|
dev[1].set_network(id, "key_mgmt", "SAE")
|
|
dev[1].set_network(id, "frequency", "2412")
|
|
dev[1].set_network_quoted(id, "psk", "thisismypassphrase!")
|
|
dev[1].set_network(id, "no_auto_peer", "1")
|
|
dev[1].mesh_group_add(id)
|
|
|
|
# Check for mesh joined
|
|
check_mesh_group_added(dev[0])
|
|
check_mesh_group_added(dev[1])
|
|
|
|
# Check for peer connected
|
|
check_mesh_peer_connected(dev[0])
|
|
check_mesh_peer_connected(dev[1])
|
|
|
|
# Test connectivity 0->1 and 1->0
|
|
test_connectivity(dev[0], dev[1])
|
|
test_connectivity(dev[1], dev[0])
|
|
|
|
|
|
def test_wpas_mesh_secure_no_auto(dev, apdev):
|
|
return wrap_wpas_mesh_test(_test_wpas_mesh_secure_no_auto, dev, apdev)
|