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

tests: Add support for separate P2P group interfaces

Browse Source 
Signed-hostap: Jouni Malinen <j@w1.fi>
This commit is contained in:
Jouni Malinen 2013-03-09 17:34:08 +02:00
parent 9e7308be40
commit f3f8ee882f
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/hwsim/hwsim_utils.py
View File

@ -24,4 +24,6 @@ def test_connectivity(ifname1, ifname2):
raise
def test_connectivity_p2p(dev1, dev2):
test_connectivity(dev1.ifname, dev2.ifname)
ifname1 = dev1.group_ifname if dev1.group_ifname else dev1.ifname
ifname2 = dev2.group_ifname if dev2.group_ifname else dev2.ifname
test_connectivity(ifname1, ifname2)

15
tests/hwsim/wpasupplicant.py
View File

@ -18,6 +18,7 @@ wpas_ctrl = '/var/run/wpa_supplicant'
class WpaSupplicant:
def __init__(self, ifname):
self.ifname = ifname
self.group_ifname = None
self.ctrl = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname))
self.mon = wpaspy.Ctrl(os.path.join(wpas_ctrl, ifname))
self.mon.attach()
@ -26,6 +27,13 @@ class WpaSupplicant:
logger.debug(self.ifname + ": CTRL: " + cmd)
return self.ctrl.request(cmd)
def group_request(self, cmd):
if self.group_ifname and self.group_ifname != self.ifname:
logger.debug(self.group_ifname + ": CTRL: " + cmd)
gctrl = wpaspy.Ctrl(os.path.join(wpas_ctrl, self.group_ifname))
return gctrl.request(cmd)
return self.request(cmd)
def ping(self):
return "PONG" in self.request("PING")
@ -35,6 +43,7 @@ class WpaSupplicant:
self.request("P2P_GROUP_REMOVE *")
self.request("REMOVE_NETWORK *")
self.request("REMOVE_CRED *")
self.group_ifname = None
def get_status(self, field):
res = self.request("STATUS")
@ -105,6 +114,7 @@ class WpaSupplicant:
res = {}
res['result'] = 'success'
res['ifname'] = s[2]
self.group_ifname = s[2]
res['role'] = s[3]
res['ssid'] = s[4]
res['freq'] = s[5]
@ -177,9 +187,10 @@ class WpaSupplicant:
def remove_group(self, ifname=None):
if ifname is None:
ifname = self.ifname
ifname = self.group_ifname if self.group_ifname else self.iname
if "OK" not in self.request("P2P_GROUP_REMOVE " + ifname):
raise Exception("Group could not be removed")
self.group_ifname = None
def p2p_start_go(self):
self.dump_monitor()
@ -194,7 +205,7 @@ class WpaSupplicant:
def p2p_go_authorize_client(self, pin):
cmd = "WPS_PIN any " + pin
if "FAIL" in self.request(cmd):
if "FAIL" in self.group_request(cmd):
raise Exception("Failed to authorize client connection on GO")
return None