Amazing-Mirror/fragattacks
1
0
Fork 0
mirror of https://github.com/vanhoefm/fragattacks.git synced 2024-11-25 00:38:24 -05:00
Code Issues Projects Releases Wiki Activity

tests: Add rfkill tests for P2P Device operations

Browse Source 
Add a couple of tests for rfkill with P2P Device operations
(with and without P2P Device interface).

Signed-off-by: Ilan Peer <ilan.peer@intel.com>
This commit is contained in:
Ilan Peer 2015-12-17 15:54:12 +02:00 committed by Jouni Malinen
parent 0e92fb8fae
commit 2e99239bd9
1 changed files with 64 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
tests/hwsim/test_rfkill.py
View File

@ -14,6 +14,7 @@ import hwsim_utils
from wpasupplicant import WpaSupplicant
from rfkill import RFKill
from utils import HwsimSkip
from hwsim import HWSimRadio
def get_rfkill(dev):
phy = dev.get_driver_status_field("phyname")
@ -116,6 +117,69 @@ def test_rfkill_autogo(dev, apdev):
rfk0.unblock()
rfk1.unblock()
def _test_rfkill_p2p_discovery(dev0, dev1):
"""rfkill block/unblock P2P Discovery"""
rfk0 = get_rfkill(dev0)
rfk1 = get_rfkill(dev1)
try:
addr0 = dev0.p2p_dev_addr()
logger.info("rfkill block 0")
rfk0.block()
logger.info("rfkill block 1")
rfk1.block()
for i in range(10):
time.sleep(0.1)
if dev0.get_status_field("wpa_state") == "INTERFACE_DISABLED" and dev1.get_status_field("wpa_state") == "INTERFACE_DISABLED":
break
if "OK" in dev0.p2p_listen():
raise Exception("P2P Listen success although in rfkill")
if "OK" in dev1.p2p_find():
raise Exception("P2P Find success although in rfkill")
dev0.dump_monitor()
dev1.dump_monitor()
logger.info("rfkill unblock 0")
rfk0.unblock()
logger.info("rfkill unblock 1")
rfk1.unblock()
for i in range(10):
time.sleep(0.1)
if dev0.get_status_field("wpa_state") != "INTERFACE_DISABLED" and dev1.get_status_field("wpa_state") != "INTERFACE_DISABLED":
break
if not "OK" in dev0.p2p_listen():
raise Exception("P2P Listen failed after unblocking rfkill")
if not dev1.discover_peer(addr0, social=True):
raise Exception("Failed to discover peer after unblocking rfkill")
finally:
rfk0.unblock()
rfk1.unblock()
dev0.p2p_stop_find()
dev1.p2p_stop_find()
dev0.dump_monitor()
dev1.dump_monitor()
def test_rfkill_p2p_discovery(dev, apdev):
"""rfkill block/unblock P2P Discovery"""
_test_rfkill_p2p_discovery(dev[0], dev[1])
def test_rfkill_p2p_discovery_p2p_dev(dev, apdev):
"""rfkill block/unblock P2P Discovery with P2P Device"""
with HWSimRadio(use_p2p_device=True) as (radio, iface):
wpas = WpaSupplicant(global_iface='/tmp/wpas-wlan5')
wpas.interface_add(iface)
_test_rfkill_p2p_discovery(dev[0], wpas)
_test_rfkill_p2p_discovery(wpas, dev[1])
def test_rfkill_hostapd(dev, apdev):
"""rfkill block/unblock during and prior to hostapd operations"""
hapd = hostapd.add_ap(apdev[0]['ifname'], { "ssid": "open" })