From f3f8ee882f86cb7987f8ba695bc9d91a92e66c1d Mon Sep 17 00:00:00 2001
From: Jouni Malinen <j@w1.fi>
Date: Sat, 9 Mar 2013 17:34:08 +0200
Subject: [PATCH] tests: Add support for separate P2P group interfaces

Signed-hostap: Jouni Malinen <j@w1.fi>
---
 tests/hwsim/hwsim_utils.py   |  4 +++-
 tests/hwsim/wpasupplicant.py | 15 +++++++++++++--
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/tests/hwsim/hwsim_utils.py b/tests/hwsim/hwsim_utils.py
index 6c5b2da28..a289e5188 100644
--- a/tests/hwsim/hwsim_utils.py
+++ b/tests/hwsim/hwsim_utils.py
@@ -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)
diff --git a/tests/hwsim/wpasupplicant.py b/tests/hwsim/wpasupplicant.py
index 082b18103..abb87c3c8 100644
--- a/tests/hwsim/wpasupplicant.py
+++ b/tests/hwsim/wpasupplicant.py
@@ -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