Added options to start / stop lock-task mode using adb.

Example:

1. Set DO:

$ adb shell dpm set-device-owner com.afwsamples.testdpc/.DeviceAdminReceiver

2. Set lock-task packages

$ adb shell dumpsys activity service com.afwsamples.testdpc set-lock-task-packages com.afwsamples.testdpc

3. Launch TestDPC in lock-task mode:

$ adb shell am start -e lock-task-mode start com.afwsamples.testdpc/.PolicyManagementActivity

4. Stop lock-task mode:

$ adb shell dumpsys activity com.afwsamples.testdpc/.PolicyManagementActivity lock-task-mode stop

Test: see above
Bug: 171350084
Bug: 181238156

Change-Id: I2b0151c0fe0bb602cf1e3b7d8e9f4e796b229948
(cherry picked from commit cdf0fdddf910f800efdc72d64cee32fbde856e73)

Author: the-felipeal

app/src/main/java/com/afwsamples/testdpc/PolicyManagementActivity.java

@@ -20,6 +20,7 @@
 import android.app.Fragment;
 import android.app.FragmentManager;
 import android.os.Bundle;
+import android.util.Log;
 import android.view.Menu;
 import android.view.MenuItem;
 
@@ -38,6 +39,15 @@
 public class PolicyManagementActivity extends DumpableActivity implements
         FragmentManager.OnBackStackChangedListener {
 
+    private static final String TAG = PolicyManagementActivity.class.getSimpleName();
+
+    private static final String CMD_LOCK_TASK_MODE = "lock-task-mode";
+    private static final String LOCK_MODE_ACTION_START = "start";
+    private static final String LOCK_MODE_ACTION_STATUS = "status";
+    private static final String LOCK_MODE_ACTION_STOP = "stop";
+
+    private boolean mLockTaskMode;
+
     @Override
     public void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
@@ -73,6 +83,16 @@ public boolean onMenuItemSelected(int featureId, MenuItem item) {
         return false;
     }
 
+    @Override
+    protected void onResume() {
+        super.onResume();
+
+        String lockModeCommand = getIntent().getStringExtra(CMD_LOCK_TASK_MODE);
+        if (lockModeCommand != null) {
+            setLockTaskMode(lockModeCommand);
+        }
+    }
+
     @Override
     public void onBackStackChanged() {
         // Show the up button in actionbar if back stack has any entry.
@@ -98,4 +118,65 @@ public void onDestroy() {
         getFragmentManager().removeOnBackStackChangedListener(this);
     }
 
+    @Override
+    public void dump(String prefix, FileDescriptor fd, PrintWriter pw, String[] args) {
+        if (args != null && args.length > 0 && args[0].equals(CMD_LOCK_TASK_MODE)) {
+            String action = args.length == 1 ? LOCK_MODE_ACTION_STATUS : args[1];
+            switch (action) {
+                case LOCK_MODE_ACTION_START:
+                    pw.println("Starting lock-task mode");
+                    startLockTaskMode();
+                    break;
+                case LOCK_MODE_ACTION_STOP:
+                    pw.println("Stopping lock-task mode");
+                    stopLockTaskMode();
+                    break;
+                case LOCK_MODE_ACTION_STATUS:
+                    dumpLockModeStatus(pw);
+                    break;
+                default:
+                    pw.printf("Invalid lock-task mode action: %s\n", action);
+            }
+            return;
+        }
+        pw.print(prefix); dumpLockModeStatus(pw);
+
+        super.dump(prefix, fd, pw, args);
+    }
+
+    private void startLockTaskMode() {
+        if (mLockTaskMode) Log.w(TAG, "startLockTaskMode(): mLockTaskMode already true");
+        mLockTaskMode = true;
+
+        Log.i(TAG, "startLockTaskMode(): calling Activity.startLockTask()");
+        startLockTask();
+    }
+
+    private void stopLockTaskMode() {
+        if (!mLockTaskMode) Log.w(TAG, "startLockTaskMode(): mLockTaskMode already false");
+        mLockTaskMode = false;
+
+        Log.i(TAG, "stopLockTaskMode(): calling Activity.stopLockTask()");
+        stopLockTask();
+    }
+
+    private void dumpLockModeStatus(PrintWriter pw) {
+        pw.printf("lock-task mode: %b\n", mLockTaskMode);
+    }
+
+    private void setLockTaskMode(String action) {
+        switch (action) {
+            case LOCK_MODE_ACTION_START:
+                startLockTaskMode();
+                break;
+            case LOCK_MODE_ACTION_STOP:
+                stopLockTaskMode();
+                break;
+            case LOCK_MODE_ACTION_STATUS:
+                Log.d(TAG, "lock-task mode status: " + mLockTaskMode);
+                break;
+            default:
+                Log.e(TAG, "invalid lock-task action: " + action);
+        }
+    }
 }

app/src/main/java/com/afwsamples/testdpc/common/DumpableActivity.java

@@ -28,7 +28,7 @@
 public abstract class DumpableActivity extends Activity {
 
     @Override
-    public final void dump(String prefix, FileDescriptor fd, PrintWriter pw, String[] args) {
+    public void dump(String prefix, FileDescriptor fd, PrintWriter pw, String[] args) {
         boolean quietMode = Dumpable.isQuietMode(args);
         if (quietMode) {
             List<Fragment> fragments = getFragmentManager().getFragments();