Kernel/iwd
3
0
Fork 0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-12-05 03:19:42 +01:00
Code Releases Activity
3f4a29651e
iwd/autotests/testBSSBlacklist/connection_test.py

307 lines
9.5 KiB
Python
Raw Normal View History

auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
#!/usr/bin/python3
import unittest
import sys
sys.path.append('../util')
import iwd
from iwd import IWD
from iwd import PSKAgent
from iwd import NetworkType
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
from iwd import IWD_CONFIG_DIR
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
from hostapd import HostapdCLI
from hwsim import Hwsim
import time
class Test(unittest.TestCase):
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
def test_temp_blacklist(self):
rule0 = self.rule0
rule1 = self.rule1
rule2 = self.rule2
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
bss_hostapd = self.bss_hostapd
wd = self.wd
rule0.signal = -8000
auto-t: fix blacklist test ranking The scan ranking logic was previously changed to be based off a theoretical calculated data rate rather than signal strength. For HT/VHT networks there are many data points that can be used for this calculation, but non HT/VHT networks are estimated based on a simple table mapping signal strengths to data rates. This table starts at a signal strength of -65 dBm and decreases from there, meaning any signal strengths greater than -65 dBm will end up getting the same ranking. This poses a problem for 3/4 blacklisting tests as they set signal strengths ranging from -20 to -40 dBm. IWD will then autoconnect to whatever network popped up first, which may not be the expected network. To fix this the signal strengths were changed to much lower values which ensures IWD picks the expected network.
2021-07-28 00:11:28 +02:00
rule1.signal = -7000
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
rule2.signal = -2000
psk_agent = PSKAgent("secret123")
wd.register_psk_agent(psk_agent)
dev1, dev2 = wd.list_devices(2)
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
ordered_network = dev1.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
ordered_network.network_object.connect()
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(dev1, condition)
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[2].wait_for_event('AP-STA-CONNECTED %s' % dev1.address)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
# dev1 now connected, this should max out the first AP, causing the next
# connection to fail to this AP.
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
ordered_network = dev2.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
ordered_network.network_object.connect()
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(dev2, condition)
# We should have temporarily blacklisted the first BSS, and connected
# to this one.
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[1].wait_for_event('AP-STA-CONNECTED %s' % dev2.address)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
# Now check that the first BSS is still not blacklisted. We can
# disconnect dev1, opening up the AP for more connections
dev1.disconnect()
dev2.disconnect()
ordered_network = dev2.get_ordered_network("TestBlacklist")
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
ordered_network.network_object.connect()
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(dev2, condition)
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[2].wait_for_event('AP-STA-CONNECTED %s' % dev2.address)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
wd.unregister_psk_agent(psk_agent)
def test_all_blacklisted(self):
wd = self.wd
bss_hostapd = self.bss_hostapd
rule0 = self.rule0
rule1 = self.rule1
rule2 = self.rule2
psk_agent = PSKAgent(["secret123", 'secret123'])
wd.register_psk_agent(psk_agent)
devices = wd.list_devices(1)
device = devices[0]
auto-t: use full_scan=True on testBSSBlacklist This test was missed during the test-wide change.
2022-03-30 23:17:59 +02:00
ordered_network = device.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
# Have both APs drop all packets, both should get blacklisted
rule0.drop = True
auto-t: fix testBSSBlacklist typo The property is 'enabled' not 'enable'
2022-04-01 01:16:36 +02:00
rule0.enabled = True
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
rule1.drop = True
auto-t: fix testBSSBlacklist typo The property is 'enabled' not 'enable'
2022-04-01 01:16:36 +02:00
rule1.enabled = True
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
rule2.drop = True
auto-t: fix testBSSBlacklist typo The property is 'enabled' not 'enable'
2022-04-01 01:16:36 +02:00
rule2.enabled = True
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
with self.assertRaises(iwd.FailedEx):
ordered_network.network_object.connect()
rule0.drop = False
rule1.drop = False
rule2.drop = False
# Wait for scanning (likely a quick-scan) to finish, otherwise we will
# may not have all BSS's in the list.
condition = 'not obj.scanning'
wd.wait_for_object_condition(device, condition)
# This connect should work
ordered_network.network_object.connect()
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[0].wait_for_event('AP-STA-CONNECTED %s' % device.address)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
wd.unregister_psk_agent(psk_agent)
def test_invalid_password(self):
wd = self.wd
bss_hostapd = self.bss_hostapd
psk_agent = PSKAgent("wrong_password")
wd.register_psk_agent(psk_agent)
devices = wd.list_devices(1)
device = devices[0]
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
auto-t: use full_scan=True on testBSSBlacklist This test was missed during the test-wide change.
2022-03-30 23:17:59 +02:00
ordered_network = device.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
with self.assertRaises(iwd.FailedEx):
ordered_network.network_object.connect()
wd.unregister_psk_agent(psk_agent)
psk_agent = PSKAgent("secret123")
wd.register_psk_agent(psk_agent)
ordered_network.network_object.connect()
# We failed to connect bss_hostapd[0], but with a bad password. Verify
# that this did not trigger a blacklist and that we did reconnect
# successfully to bss_hostapd[0]
self.assertIn(device.address, bss_hostapd[0].list_sta())
auto-t: test for temporary blacklist The simplest way to test this was to create a new AP, where max_num_sta=1. This only allows a single STA to connect to this AP. We connect a device to this AP, then try and connect with another. This results in hostapd failing with DENIED_NO_MORE_STAS, which will cause a temporary blacklist. We can then disconnect both devices, and reconnect the device that previously got denied. If it connects then we know the blacklist only persisted for that earlier connection.
2019-03-01 19:55:05 +01:00
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
wd.unregister_psk_agent(psk_agent)
def test_connection_success(self):
wd = self.wd
bss_hostapd = self.bss_hostapd
rule0 = self.rule0
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
psk_agent = PSKAgent("secret123")
wd.register_psk_agent(psk_agent)
auto-t: increase wait time for blacklist test This test was not reliably passing. Busy waiting is not really reliable, but in this specific case its really the only option as the blacklist must expire based on time.
2020-07-08 19:48:00 +02:00
devices = wd.list_devices(2)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
device = devices[0]
auto-t: increase wait time for blacklist test This test was not reliably passing. Busy waiting is not really reliable, but in this specific case its really the only option as the blacklist must expire based on time.
2020-07-08 19:48:00 +02:00
devices[1].disconnect()
auto-t: use full_scan=True on testBSSBlacklist This test was missed during the test-wide change.
2022-03-30 23:17:59 +02:00
ordered_network = device.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
auto-t: fix testBSSBlacklist There were a number of fixes here. The waits were changed to wait on the device state instead of the network state and hwsim rules were removed after the test as to not interfere with future tests. One of the rules was setting the signal to -10000 wich was causing the ranking to be zero.
2020-09-11 01:12:41 +02:00
rule0.drop = True
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
ordered_network.network_object.connect(wait=False)
# Have AP1 drop all packets, should result in a connection timeout
rule0.drop = True
# Note the time so later we don't sleep any longer than required
start = time.time()
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
# IWD should have attempted to connect to bss_hostapd[0], since its
# signal strength was highest. But since we dropped all packets this
# connect should fail, and the BSS should be blacklisted. Then we
# should automatically try the next BSS in the list, which is
# bss_hostapd[1]
self.assertNotIn(device.address, bss_hostapd[0].list_sta())
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[1].wait_for_event('AP-STA-CONNECTED %s' % device.address)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
rule0.drop = False
auto-t: set autoconnect=True for tests that need it After the default behavior was changed to disable autoconnect there were several tests which still needed it to be turned on.
2021-08-12 22:38:12 +02:00
device.disconnect()
device.autoconnect = True
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
auto-t: Rename setting Autoconnect to AutoConnect
2019-10-24 19:53:23 +02:00
# Now we wait... AutoConnect should take over
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
condition = 'obj.state == DeviceState.connecting'
auto-t: Increase all test timeouts Slower systems may not be able to make some timeouts that tests mandated. All timeouts were increased significantly to allow tests to pass on slow systems.
2020-09-11 01:12:37 +02:00
wd.wait_for_object_condition(device, condition)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
condition = 'obj.state == DeviceState.connected'
auto-t: Increase all test timeouts Slower systems may not be able to make some timeouts that tests mandated. All timeouts were increased significantly to allow tests to pass on slow systems.
2020-09-11 01:12:37 +02:00
wd.wait_for_object_condition(device, condition)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
# Same as before, make sure we didn't connect to the blacklisted AP.
self.assertNotIn(device.address, bss_hostapd[0].list_sta())
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[1].wait_for_event('AP-STA-CONNECTED %s' % device.address)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
# Wait for the blacklist to expire (10 seconds)
elapsed = time.time() - start
auto-t: increase wait time for blacklist test This test was not reliably passing. Busy waiting is not really reliable, but in this specific case its really the only option as the blacklist must expire based on time.
2020-07-08 19:48:00 +02:00
if elapsed < 15:
wd.wait(15 - elapsed)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
device.disconnect()
auto-t: remove direct scan() call when possible 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.
2022-03-30 22:23:11 +02:00
ordered_network = device.get_ordered_network("TestBlacklist", full_scan=True)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
self.assertEqual(ordered_network.type, NetworkType.psk)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
ordered_network.network_object.connect()
auto-t: replace waiting for network with device Many tests waited on the network object 'connected' property after issuing a Connect command. This is not correct as 'connected' is set quite early in the connection process. The correct way of doing this is waiting for the device state to change to connected. This common code was replaced, hopefully putting to rest any random failures that happen occasionally.
2020-09-14 23:02:33 +02:00
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
auto-t: use wait_for_event in testBSSBlacklist Depending on timing hostapd may not show a station with list_sta(). Instead use wait_for_event.
2022-03-28 19:28:35 +02:00
bss_hostapd[0].wait_for_event('AP-STA-CONNECTED %s' % device.address)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
self.wd.unregister_psk_agent(psk_agent)
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
def setUp(self):
self.wd = IWD(True)
def tearDown(self):
IWD.clear_storage()
self.wd = None
auto-t: fix testBSSBlacklist There were a number of fixes here. The waits were changed to wait on the device state instead of the network state and hwsim rules were removed after the test as to not interfere with future tests. One of the rules was setting the signal to -10000 wich was causing the ranking to be zero.
2020-09-11 01:12:41 +02:00
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
@classmethod
def setUpClass(cls):
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
cls.hwsim = Hwsim()
cls.bss_hostapd = [ HostapdCLI(config='ssid1.conf'),
HostapdCLI(config='ssid2.conf'),
HostapdCLI(config='ssid3.conf') ]
cls.bss_radio = [ cls.hwsim.get_radio('rad0'),
cls.hwsim.get_radio('rad1'),
cls.hwsim.get_radio('rad2') ]
cls.rule0 = cls.hwsim.rules.create()
cls.rule0.source = cls.bss_radio[0].addresses[0]
cls.rule0.bidirectional = True
cls.rule0.signal = -2000
cls.rule0.enabled = True
cls.rule1 = cls.hwsim.rules.create()
cls.rule1.source = cls.bss_radio[1].addresses[0]
cls.rule1.bidirectional = True
cls.rule1.signal = -7000
cls.rule1.enabled = True
cls.rule2 = cls.hwsim.rules.create()
cls.rule2.source = cls.bss_radio[2].addresses[0]
cls.rule2.bidirectional = True
cls.rule2.signal = -8000
cls.rule2.enabled = True
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
@classmethod
def tearDownClass(cls):
IWD.clear_storage()
auto-t: combine testBSSBlacklist tests All 4 tests were combined to reuse hwsim rule code, and improve cleanup between running tests.
2021-08-19 00:09:32 +02:00
cls.hwsim.rules.remove_all()
auto-t: Add test for BSS blacklisting
2019-01-29 21:36:13 +01:00
if __name__ == '__main__':
unittest.main(exit=True)
Copy Permalink