Connectivity: Add IP config to TDLS test

Change-Id: I70096a042faaacdaeed7d6aae16a95704202ef1d
Signed-off-by: Dmitry Shmidt <dimitrysh@google.com>

1 files changed, 14 insertions, 2 deletions

diff --git a/apps/Development/src/com/android/development/Connectivity.java b/apps/Development/src/com/android/development/Connectivity.java
index 12fa24a8e..0da1410f0 100644
--- a/apps/Development/src/com/android/development/Connectivity.java
+++ b/apps/Development/src/com/android/development/Connectivity.java
@@ -501,13 +501,25 @@ public class Connectivity extends Activity {
     private void onStartTdls() {
         mTdlsAddr = ((EditText)findViewById(R.id.sc_ip_mac)).getText().toString();
         Log.d(TAG, "TDLS: START " + mTdlsAddr);
-        mWm.enableTdlsWithMacAddress(mTdlsAddr, true);
+        InetAddress inetAddress = null;
+        try {
+            inetAddress = InetAddress.getByName(mTdlsAddr);
+            mWm.enableTdls(inetAddress, true);
+        } catch (Exception e) {
+            mWm.enableTdlsWithMacAddress(mTdlsAddr, true);
+        }
     }
 
     private void onStopTdls() {
         if (mTdlsAddr == null) return;
         Log.d(TAG, "TDLS: STOP " + mTdlsAddr);
-        mWm.enableTdlsWithMacAddress(mTdlsAddr, false);
+        InetAddress inetAddress = null;
+        try {
+            inetAddress = InetAddress.getByName(mTdlsAddr);
+            mWm.enableTdls(inetAddress, false);
+        } catch (Exception e) {
+            mWm.enableTdlsWithMacAddress(mTdlsAddr, false);
+        }
     }
 
     private void onAddDefaultRoute() {