mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-11-28 21:19:24 +01:00
a773aa6a07
All tests which could avoid calling scan() directly have been changed to use the 'full_scan' argument to get_ordered_network. This was done because of unreliable scanning behavior on slower systems, like VMs. If we get unlucky with the scheduler some beacons are not received in time and in turn scan results are missing. Using full_scan=True works around this issue by repeatedly scanning until the SSID is found.
264 lines
9.9 KiB
Python
264 lines
9.9 KiB
Python
#!/usr/bin/python3
|
|
|
|
import unittest
|
|
import sys
|
|
|
|
sys.path.append('../util')
|
|
import iwd
|
|
from iwd import IWD
|
|
from iwd import NetworkType
|
|
from hostapd import HostapdCLI
|
|
from hwsim import Hwsim
|
|
import testutil
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
def validate(self, wd, target, autoconnect=False, connect_hidden=False):
|
|
devices = wd.list_devices(1)
|
|
device = devices[0]
|
|
device.autoconnect = autoconnect
|
|
|
|
if not autoconnect:
|
|
if not connect_hidden:
|
|
network = device.get_ordered_network('transition', full_scan=True)
|
|
|
|
network.network_object.connect()
|
|
else:
|
|
device.connect_hidden_network('owe-hidden')
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
testutil.test_iface_operstate()
|
|
|
|
if type(target) != list:
|
|
testutil.test_ifaces_connected(device.name, target.ifname)
|
|
self.assertIn(device.address, target.list_sta())
|
|
else:
|
|
found = False
|
|
for t in target:
|
|
if device.address in t.list_sta():
|
|
testutil.test_ifaces_connected(device.name, t.ifname)
|
|
found = True
|
|
|
|
self.assertTrue(found)
|
|
|
|
device.disconnect()
|
|
|
|
# Normal success case
|
|
def test_owe_transition(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_owe)
|
|
|
|
# Normal success case
|
|
def test_owe_transition_multi_network(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_open2.set_value('vendor_elements', 'dd17506f9a1c02000000f1000c6f77652d68696464656e2d32')
|
|
self.hapd_open2.set_value('ssid', 'transition-2')
|
|
self.hapd_open2.reload()
|
|
self.hapd_owe2.set_value('vendor_elements', 'dd17506f9a1c02000000f0000c7472616e736974696f6e2d32')
|
|
self.hapd_owe2.set_value('ssid', 'owe-hidden-2')
|
|
self.hapd_owe2.reload()
|
|
|
|
self.validate(self.wd, self.hapd_owe)
|
|
|
|
# Two pairs of open/OWE BSS's (OWE BSS's have different SSIDs) */
|
|
def test_owe_transition_multi_bss(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_open2.set_value('vendor_elements', 'dd17506f9a1c02000000f3000c6f77652d68696464656e2d32')
|
|
self.hapd_open2.set_value('ssid', 'transition')
|
|
self.hapd_open2.reload()
|
|
self.hapd_owe2.set_value('vendor_elements', 'dd15506f9a1c02000000f2000a7472616e736974696f6e')
|
|
self.hapd_owe2.set_value('ssid', 'owe-hidden-2')
|
|
self.hapd_owe2.reload()
|
|
|
|
self.validate(self.wd, [self.hapd_owe, self.hapd_owe2])
|
|
|
|
# Two pairs of open/OWE BSS's (OWE BSS's have same SSID) */
|
|
def test_owe_transition_multi_bss_same_ssid(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_open2.set_value('vendor_elements', 'dd15506f9a1c02000000f3000a6f77652d68696464656e')
|
|
self.hapd_open2.set_value('ssid', 'transition')
|
|
self.hapd_open2.reload()
|
|
self.hapd_owe2.set_value('vendor_elements', 'dd15506f9a1c02000000f2000a7472616e736974696f6e')
|
|
self.hapd_owe2.set_value('ssid', 'owe-hidden')
|
|
self.hapd_owe2.reload()
|
|
|
|
self.validate(self.wd, [self.hapd_owe, self.hapd_owe2])
|
|
|
|
# Normal success autoconnect case
|
|
def test_owe_transition_autoconnect(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
IWD.copy_to_storage('transition.open')
|
|
|
|
self.validate(self.wd, self.hapd_owe, autoconnect=True)
|
|
|
|
# Open BSS has invalid BSSID in OWE transition element
|
|
# Expected connection to Open BSS
|
|
def test_owe_transition_invalid_open_bssid(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000ff000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_open)
|
|
|
|
# OWE BSS has invalid BSSID in OWE transition element
|
|
# Expected connection to Open BSS
|
|
def test_owe_transition_invalid_owe_bssid(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000ff000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_open)
|
|
|
|
# No OWE hidden network exists
|
|
# Expected connection to Open BSS
|
|
def test_owe_transition_no_hidden_found(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.disable()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_open)
|
|
|
|
# Directly connect to valid OWE hidden network
|
|
# Expected connection failure
|
|
def test_owe_transition_connect_hidden_valid(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
with self.assertRaises(iwd.NotFoundEx):
|
|
self.validate(self.wd, self.hapd_owe, connect_hidden=True)
|
|
|
|
# Directly connect to invalid OWE hidden network. The SSID in the OWE BSS
|
|
# IE points to itself. And the Open BSS IE points to a non-existent SSID.
|
|
# Expected connection failure
|
|
def test_owe_transition_connect_hidden_invalid(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd18506f9a1c02000000f1000d6e6f2d6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a6f77652d68696464656e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
with self.assertRaises(iwd.NotFoundEx):
|
|
self.validate(self.wd, self.hapd_owe, connect_hidden=True)
|
|
|
|
def test_owe_transition_band_info(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd17506f9a1c02000000f1000a6f77652d68696464656e5103')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.set_value('channel', '3')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_owe)
|
|
|
|
def test_owe_transition_wrong_band_info(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd17506f9a1c02000000f1000a6f77652d68696464656e5102')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.set_value('channel', '3')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
self.hapd_open2.disable()
|
|
|
|
self.validate(self.wd, self.hapd_open)
|
|
|
|
# OWE Transition pair + additional open network with the same SSID
|
|
def test_owe_transition_extra_open(self):
|
|
self.hapd_open.set_value('vendor_elements', 'dd15506f9a1c02000000f1000a6f77652d68696464656e')
|
|
self.hapd_open.reload()
|
|
self.hapd_owe.set_value('vendor_elements', 'dd15506f9a1c02000000f0000a7472616e736974696f6e')
|
|
self.hapd_owe.reload()
|
|
|
|
self.hapd_open2.set_value('ssid', 'transition')
|
|
self.hapd_open2.reload()
|
|
|
|
self.hapd_owe2.disable()
|
|
|
|
# Set the open network signal strength very low so it gets put last on
|
|
# the network bss_list, forcing the additional open network to be
|
|
# checked first.
|
|
self.rule0 = self.hwsim.rules.create()
|
|
self.rule0.source = self.hwsim.get_radio('rad0').addresses[0]
|
|
self.rule0.signal = -4000
|
|
self.rule0.enabled = True
|
|
|
|
self.validate(self.wd, self.hapd_owe)
|
|
|
|
def setUp(self):
|
|
self.wd = IWD(True)
|
|
self.hapd_owe = HostapdCLI(config='ssidOWE.conf')
|
|
self.hapd_open = HostapdCLI(config='ssidOpen.conf')
|
|
self.hapd_owe2 = HostapdCLI(config='ssidOWE-2.conf')
|
|
self.hapd_open2 = HostapdCLI(config='ssidOpen-2.conf')
|
|
|
|
self.hwsim = Hwsim()
|
|
|
|
def tearDown(self):
|
|
IWD.clear_storage()
|
|
|
|
self.hapd_owe.set_value('channel', '1')
|
|
|
|
self.wd = None
|
|
self.hapd_open = None
|
|
self.hapd_owe = None
|
|
self.hwsim = None
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
pass
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|