Update group capability before connect

We need to get the latest group capability information before connect
now that the supplicant behavior is to do a delayed cleanup.

Bug: 6613470
Change-Id: Ie374d750950f3bd4376fd6a767bb253fd7986eb1

Author: isheriff

wifi/java/android/net/wifi/WifiNative.java

@@ -640,6 +640,27 @@ public String p2pGetDeviceAddress() {
         return "";
     }
 
+    public int getGroupCapability(String deviceAddress) {
+        int gc = 0;
+        if (TextUtils.isEmpty(deviceAddress)) return gc;
+        String peerInfo = p2pPeer(deviceAddress);
+        if (TextUtils.isEmpty(peerInfo)) return gc;
+
+        String[] tokens = peerInfo.split("\n");
+        for (String token : tokens) {
+            if (token.startsWith("group_capab=")) {
+                String[] nameValue = token.split("=");
+                if (nameValue.length != 2) break;
+                try {
+                    return Integer.decode(nameValue[1]);
+                } catch(NumberFormatException e) {
+                    return gc;
+                }
+            }
+        }
+        return gc;
+    }
+
     public String p2pPeer(String deviceAddress) {
         return doStringCommand("P2P_PEER " + deviceAddress);
     }

wifi/java/android/net/wifi/p2p/WifiP2pDeviceList.java

@@ -19,6 +19,7 @@
 import android.os.Parcelable;
 import android.os.Parcel;
 import android.net.wifi.p2p.WifiP2pDevice;
+import android.text.TextUtils;
 import android.util.Log;
 
 import java.util.ArrayList;
@@ -83,11 +84,22 @@ public void update(WifiP2pDevice device) {
         mDevices.put(device.deviceAddress, device);
     }
 
+    /** @hide */
+    public void updateGroupCapability(String deviceAddress, int groupCapab) {
+        if (TextUtils.isEmpty(deviceAddress)) return;
+        WifiP2pDevice d = mDevices.get(deviceAddress);
+        if (d != null) {
+            d.groupCapability = groupCapab;
+        }
+    }
+
     /** @hide */
     public void updateStatus(String deviceAddress, int status) {
-        if (deviceAddress == null) return;
+        if (TextUtils.isEmpty(deviceAddress)) return;
         WifiP2pDevice d = mDevices.get(deviceAddress);
-        d.status = status;
+        if (d != null) {
+            d.status = status;
+        }
     }
 
     /** @hide */

wifi/java/android/net/wifi/p2p/WifiP2pService.java

@@ -764,6 +764,10 @@ public boolean processMessage(Message message) {
                     WifiP2pConfig config = (WifiP2pConfig) message.obj;
                     mAutonomousGroup = false;
 
+                    /* Update group capability before connect */
+                    int gc = mWifiNative.getGroupCapability(config.deviceAddress);
+                    mPeers.updateGroupCapability(config.deviceAddress, gc);
+
                     if (mSavedPeerConfig != null && config.deviceAddress.equals(
                                     mSavedPeerConfig.deviceAddress)) {
                         mSavedPeerConfig = config;