No public description

PiperOrigin-RevId: 634690620

Author: jscott1989

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

@@ -247,6 +247,13 @@ public void run() {
                 ordinalParam(String.class, "restriction"),
                 ordinalParam(boolean.class, "enabled"))
             .setDescription("Set the given user restriction."));
+    flags.addCommand(
+        command(
+                "set-user-restriction-on-parent",
+                this::setUserRestrictionOnParent,
+                ordinalParam(String.class, "restriction"),
+                ordinalParam(boolean.class, "enabled"))
+            .setDescription("Set the given user restriction on the parent user."));
     flags.addCommand(
         command("lock-now", this::lockNow, optional(namedParam(int.class, "flags")))
             .setDescription("Lock the device (now! :-)."));
@@ -789,6 +796,19 @@ private void setUserRestriction(String userRestriction, boolean enabled) {
         (e) -> onError(e, "Error setting user restriction '%s' to %b", userRestriction, enabled));
   }
 
+  private void setUserRestrictionOnParent(String userRestriction, boolean enabled) {
+    Log.i(TAG, "setUserRestrictionOnParent(" + userRestriction + ", " + enabled + ")");
+    DevicePolicyManagerGateway parentDpmGateway =
+        DevicePolicyManagerGatewayImpl.forParentProfile(mContext);
+    parentDpmGateway.setUserRestriction(
+        userRestriction,
+        enabled,
+        (v) -> onSuccess("User restriction '%s' set to %b", userRestriction, enabled),
+        (e) ->
+            onError(
+                e, "Error setting parent user restriction '%s' to %b", userRestriction, enabled));
+  }
+
   private void lockNow(Integer flags) {
     if (flags == null) {
       Log.i(TAG, "lockNow()");

src/main/java/com/afwsamples/testdpc/policy/UserRestriction.java

@@ -10,6 +10,7 @@
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_LOCK_CREDENTIALS;
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_MOBILE_NETWORK;
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_MODIFY_USERS;
+import static android.Manifest.permission.MANAGE_DEVICE_POLICY_PROFILES;
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_RESTRICT_PRIVATE_DNS;
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_SAFE_BOOT;
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_SMS;
@@ -19,6 +20,7 @@
 import static android.Manifest.permission.MANAGE_DEVICE_POLICY_WIFI;
 import static android.os.UserManager.ALLOW_PARENT_PROFILE_APP_LINKING;
 import static android.os.UserManager.DISALLOW_ADD_MANAGED_PROFILE;
+import static android.os.UserManager.DISALLOW_ADD_PRIVATE_PROFILE;
 import static android.os.UserManager.DISALLOW_ADD_USER;
 import static android.os.UserManager.DISALLOW_ADD_WIFI_CONFIG;
 import static android.os.UserManager.DISALLOW_ADJUST_VOLUME;
@@ -221,6 +223,10 @@ public UserRestriction(String key, int titleResId, String permission) {
     new UserRestriction(DISALLOW_ULTRA_WIDEBAND_RADIO, R.string.disallow_ultra_wideband_radio),
     new UserRestriction(DISALLOW_ASSIST_CONTENT, R.string.disallow_assist_content),
     new UserRestriction(DISALLOW_SIM_GLOBALLY, R.string.disallow_sim_globally),
+    new UserRestriction(
+        DISALLOW_ADD_PRIVATE_PROFILE,
+        R.string.disallow_add_private_profile,
+        MANAGE_DEVICE_POLICY_PROFILES),
   };
 
   /**
@@ -259,6 +265,7 @@ public UserRestriction(String key, int titleResId, String permission) {
     DISALLOW_ULTRA_WIDEBAND_RADIO,
     DISALLOW_CONFIG_BRIGHTNESS,
     DISALLOW_CONFIG_SCREEN_TIMEOUT,
+    DISALLOW_ADD_PRIVATE_PROFILE,
   };
 
   /** Setting these user restrictions only have effect on primary users. */
@@ -318,7 +325,8 @@ public UserRestriction(String key, int titleResId, String permission) {
     DISALLOW_AIRPLANE_MODE,
     DISALLOW_CONFIG_SCREEN_TIMEOUT,
     DISALLOW_CONFIG_BRIGHTNESS,
-    DISALLOW_AMBIENT_DISPLAY
+    DISALLOW_AMBIENT_DISPLAY,
+    DISALLOW_ADD_PRIVATE_PROFILE,
   };
 
   /** These user restrictions are added in MNC. */
@@ -376,7 +384,10 @@ public UserRestriction(String key, int titleResId, String permission) {
   public static String[] VIC_PLUS_RESTRICTIONS = {DISALLOW_SIM_GLOBALLY};
 
   public static String[] VIC_PLUS_PARENT_RESTRICTIONS = {
-    DISALLOW_CONFIG_BRIGHTNESS, DISALLOW_CONFIG_SCREEN_TIMEOUT, DISALLOW_ASSIST_CONTENT
+    DISALLOW_CONFIG_BRIGHTNESS,
+    DISALLOW_CONFIG_SCREEN_TIMEOUT,
+    DISALLOW_ASSIST_CONTENT,
+    DISALLOW_ADD_PRIVATE_PROFILE
   };
 
   public static UserRestriction getRestriction(String restrictionKey) {

src/main/res/values/strings.xml

@@ -547,6 +547,7 @@
     <string name="disallow_ultra_wideband_radio">Disallow ultra wideband radio</string>
     <string name="disallow_assist_content">Disallow assist content</string>
     <string name="disallow_sim_globally">Disallow sim globally</string>
+    <string name="disallow_add_private_profile">Disallow add private profile</string>
     <string name="user_restriction_error_msg">Operation not allowed</string>
 
     <!-- Strings for select app -->