Added commands to get / set application restrictions.

Example:

$ alias testdpc='adb shell dumpsys activity service --user 0 com.afwsamples.testdpc'

$ testdpc |grep app-restriction
    	set-app-restrictions <PKG> <K1 V1> [Kn Vn] - sets the key/value (as String) application restrictions for the given application
    	get-app-restrictions [PKG1] [PKGNn] - get the application restrictions for the given apps, or for TestDPC itself (using UserManager) when PKG is not passed

$ testdpc get-app-restrictions
SERVICE com.afwsamples.testdpc/.DeviceAdminService affcbf pid=29012 user=0
  Client:
    No app restrictions for com.afwsamples.testdpc

$ testdpc set-app-restrictions com.afwsamples.testdpc dude sweet
SERVICE com.afwsamples.testdpc/.DeviceAdminService affcbf pid=29012 user=0
  Client:
    Set 1 app restrictions for com.afwsamples.testdpc

$ testdpc get-app-restrictions
SERVICE com.afwsamples.testdpc/.DeviceAdminService affcbf pid=29012 user=0
  Client:
    1 app restrictions for com.afwsamples.testdpc
      dude = sweet

$ testdpc get-app-restrictions com.afwsamples.testdpc foo bar
SERVICE com.afwsamples.testdpc/.DeviceAdminService affcbf pid=29012 user=0
  Client:
    1 app restrictions for com.afwsamples.testdpc
      dude = sweet
    No app restrictions for foo
    No app restrictions for bar

Test: see above
Bug: 171350084

Change-Id: I90d9be54d4bf3bffdd83369275c15fc1a267140c
(cherry picked from commit 992a3d51951cbf277943d0be33abb51445a144f1)

Author: the-felipeal

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

@@ -19,6 +19,7 @@
 import android.content.ComponentName;
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.graphics.Bitmap;
+import android.os.Bundle;
 import android.os.PersistableBundle;
 import android.os.UserHandle;
 import android.os.UserManager;
@@ -299,6 +300,22 @@ void setLockTaskFeatures(int flags, @NonNull Consumer<Void> onSuccess,
      */
     boolean isLockTaskPermitted(String packageName);
 
+    /**
+     * See {@link android.app.admin.DevicePolicyManager#setApplicationRestrictions(ComponentName, String, Bundle)}.
+     */
+    void setApplicationRestrictions(String packageName, Bundle settings,
+            @NonNull Consumer<Void> onSuccess, @NonNull Consumer<Exception> onError);
+
+    /**
+     * See {@link android.app.admin.DevicePolicyManager#getApplicationRestrictions(ComponentName, String)}.
+     */
+    Bundle getApplicationRestrictions(String packageName);
+
+    /**
+     * Gets this app's own restrictions using {@link UserManager#getApplicationRestrictions(String)}.
+     */
+    Bundle getSelfRestrictions();
+
     /**
      * Used on error callbacks to indicate a {@link android.app.admin.DevicePolicyManager} method
      * call failed.

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

@@ -552,6 +552,29 @@ public boolean isLockTaskPermitted(String packageName) {
         return mDevicePolicyManager.isLockTaskPermitted(packageName);
     }
 
+    @Override
+    public Bundle getApplicationRestrictions(String packageName) {
+        return mDevicePolicyManager.getApplicationRestrictions(mAdminComponentName, packageName);
+    }
+
+    @Override
+    public Bundle getSelfRestrictions() {
+        return mUserManager.getApplicationRestrictions(mAdminComponentName.getPackageName());
+    }
+
+    @Override
+    public void setApplicationRestrictions(String packageName, Bundle settings,
+            Consumer<Void> onSuccess, Consumer<Exception> onError) {
+        Log.d(TAG, "setApplicationRestrictions(" + packageName + ")");
+        try {
+            mDevicePolicyManager.setApplicationRestrictions(mAdminComponentName, packageName,
+                    settings);
+            onSuccess.accept(null);
+        } catch (Exception e) {
+            onError.accept(e);
+        }
+    }
+
     @Override
     public String toString() {
         return "DevicePolicyManagerGatewayImpl[" + mAdminComponentName + "]";

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

@@ -20,6 +20,8 @@
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.graphics.Bitmap;
 import android.graphics.BitmapFactory;
+import android.os.Build;
+import android.os.Bundle;
 import android.os.Environment;
 import android.os.UserHandle;
 import android.util.Log;
@@ -89,6 +91,8 @@ final class ShellCommand {
     private static final String CMD_IS_LOCK_TASK_PERMITTED = "is-lock-task-permitted";
     private static final String CMD_SET_LOCK_TASK_FEATURES = "set-lock-task-features";
     private static final String CMD_GET_LOCK_TASK_FEATURES = "get-lock-task-features";
+    private static final String CMD_SET_APP_RESTRICTIONS = "set-app-restrictions";
+    private static final String CMD_GET_APP_RESTRICTIONS = "get-app-restrictions";
 
     private static final String ARG_FLAGS = "--flags";
 
@@ -231,6 +235,12 @@ public void run() {
             case CMD_GET_LOCK_TASK_FEATURES:
                 execute(() -> getLockTaskFeatures());
                 break;
+            case CMD_SET_APP_RESTRICTIONS:
+                execute(() -> setAppRestrictions());
+                break;
+            case CMD_GET_APP_RESTRICTIONS:
+                execute(() -> getAppRestrictions());
+                break;
             default:
                 mWriter.printf("Invalid command: %s\n\n", cmd);
                 showUsage();
@@ -317,6 +327,12 @@ private void showUsage() {
                 + "have tasks locked\n", CMD_IS_LOCK_TASK_PERMITTED);
         mWriter.printf("\t%s <FLAGS> - set the lock task features\n", CMD_SET_LOCK_TASK_FEATURES);
         mWriter.printf("\t%s - get the lock task features\n", CMD_GET_LOCK_TASK_FEATURES);
+        mWriter.printf("\t%s <PKG> [K1 V1] [Kn Vn] - sets the key/value (as String) application "
+                + "restrictions for the given app (or resets when no key/value is passed)\n",
+                CMD_SET_APP_RESTRICTIONS);
+        mWriter.printf("\t%s [PKG1] [PKGNn] - get the application restrictions for the given apps, "
+                + "or for TestDPC itself (using UserManager) when PKG is not passed\n",
+                CMD_GET_APP_RESTRICTIONS);
     }
 
     private void createUser() {
@@ -692,11 +708,51 @@ private void setLockTaskFeatures() {
 
     private void getLockTaskFeatures() {
         int flags = mDevicePolicyManagerGateway.getLockTaskFeatures();
-        String features= Util.lockTaskFeaturesToString(flags);
+        String features = Util.lockTaskFeaturesToString(flags);
 
         mWriter.printf("%s (%d)\n", features, flags);
     }
 
+    private void setAppRestrictions() {
+        // TODO(b/171350084): check args size
+        String packageName = mArgs[1];
+        Bundle settings = new Bundle();
+        for (int i = 2; i < mArgs.length; i++) {
+            String key = mArgs[i];
+            String value = mArgs[++i];
+            settings.putString(key, value);
+        }
+        mDevicePolicyManagerGateway.setApplicationRestrictions(packageName, settings,
+                (v) -> onSuccess("Set %d app restrictions for %s", settings.size(), packageName),
+                (e) -> onError(e, "Error setting app restrictions for %s", packageName));
+    }
+
+    private void getAppRestrictions() {
+        if (mArgs.length == 1) {
+            printAppRestrictions(mContext.getPackageName(),
+                    mDevicePolicyManagerGateway.getSelfRestrictions());
+            return;
+        }
+
+        for (int i = 1; i < mArgs.length; i++) {
+            String packageName = mArgs[i];
+            Bundle settings = mDevicePolicyManagerGateway.getApplicationRestrictions(packageName);
+            printAppRestrictions(packageName, settings);
+        }
+    }
+
+    private void printAppRestrictions(String packageName, Bundle settings) {
+        if (settings == null || settings.isEmpty()) {
+            mWriter.printf("No app restrictions for %s\n", packageName);
+            return;
+        }
+        mWriter.printf("%d app restrictions for %s\n", settings.size(), packageName);
+        for (String key : settings.keySet()) {
+            Object value = settings.get(key);
+            mWriter.printf("  %s = %s\n", key, value);
+        }
+    }
+
     private static String permittedToString(boolean permitted) {
         return permitted ? "PERMITTED" : "NOT PERMITTED";
     }