Fix START_WPS null exception

Fixed an uncaught null being passed in through the startWps API.

Bug: 36491634
Test: Added a unit test
Change-Id: Ic8f5cec9d43332b4c8701582455e662a6f885f20

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
index 5c3504906..f3cf0b888 100644
--- a/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/WifiStateMachineTest.java
@@ -1512,4 +1512,20 @@ public class WifiStateMachineTest {
         addNetworkAndVerifySuccess(false);
         verify(mWifiConnectivityManager, never()).setUserConnectChoice(eq(0));
     }
+
+    /**
+     * Test START_WPS with a null wpsInfo object fails gracefully (No NPE)
+     */
+    @Test
+    public void testStartWps_nullWpsInfo() throws Exception {
+        loadComponentsInStaMode();
+        mWsm.setOperationalMode(WifiStateMachine.CONNECT_MODE);
+        assertEquals(WifiStateMachine.CONNECT_MODE, mWsm.getOperationalModeForTest());
+        assertEquals("DisconnectedState", getCurrentState().getName());
+        mLooper.startAutoDispatch();
+        Message reply = mWsmAsyncChannel.sendMessageSynchronously(WifiManager.START_WPS, 0, 0,
+                null);
+        mLooper.stopAutoDispatch();
+        assertEquals(WifiManager.WPS_FAILED, reply.what);
+    }
 }