tests: Add support for separate P2P group interfaces

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

View file

@ -24,4 +24,6 @@ def test_connectivity(ifname1, ifname2):
raise raise
def test_connectivity_p2p(dev1, dev2): 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)

View file

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