Added commands to set / get user location.

Test: adb shell dumpsys activity service --user 0 com.afwsamples.testdpc is-location-enabled
Test: adb shell dumpsys activity service --user 0 com.afwsamples.testdpc set-location-enabled false

Bug: 171350084
Bug: 185523465

Change-Id: I98d770a3fdeb2f9ab0d75cd0f34e0dee090e4ae4
(cherry picked from commit 6d43070c6bca8a3648f91e003c7a62efcaa6a0a7)

Author: the-felipeal

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

@@ -327,6 +327,17 @@ void setPermissionGrantState(String packageName, String permission, int grantSta
      */
     int getPermissionGrantState(String packageName, String permission);
 
+    /**
+     * See {@link android.app.admin.DevicePolicyManager#setLocationEnabled(ComponentName, boolean)}.
+     */
+    void setLocationEnabled(boolean enabled, @NonNull Consumer<Void> onSuccess,
+            @NonNull Consumer<Exception> onError);
+
+    /**
+     * See {@link android.location.LocationManager#isLocationEnabled()}.
+     */
+    boolean isLocationEnabled();
+
     /**
      * Used on error callbacks to indicate a {@link android.app.admin.DevicePolicyManager} method
      * call failed.

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

@@ -21,6 +21,8 @@
 import android.content.Context;
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.graphics.Bitmap;
+import android.location.LocationManager;
+import android.os.Build;
 import android.os.Bundle;
 import android.os.PersistableBundle;
 import android.os.UserHandle;
@@ -44,17 +46,20 @@ public final class DevicePolicyManagerGatewayImpl implements DevicePolicyManager
     private final DevicePolicyManager mDevicePolicyManager;
     private final UserManager mUserManager;
     private final ComponentName mAdminComponentName;
+    private final LocationManager mLocationManager;
 
     public DevicePolicyManagerGatewayImpl(@NonNull Context context) {
         this(context.getSystemService(DevicePolicyManager.class),
                 context.getSystemService(UserManager.class),
+                context.getSystemService(LocationManager.class),
                 DeviceAdminReceiver.getComponentName(context));
     }
 
     public DevicePolicyManagerGatewayImpl(@NonNull DevicePolicyManager dpm, @NonNull UserManager um,
-            @NonNull ComponentName admin) {
+            @NonNull LocationManager lm, @NonNull ComponentName admin) {
         mDevicePolicyManager = dpm;
         mUserManager = um;
+        mLocationManager = lm;
         mAdminComponentName = admin;
 
         Log.d(TAG, "constructor: admin=" + mAdminComponentName + ", dpm=" + dpm);
@@ -65,7 +70,8 @@ public static DevicePolicyManagerGatewayImpl forParentProfile(@NonNull Context c
         DevicePolicyManager dpm = context.getSystemService(DevicePolicyManager.class)
                 .getParentProfileInstance(admin);
         UserManager um = context.getSystemService(UserManager.class);
-        return new DevicePolicyManagerGatewayImpl(dpm, um, admin);
+        LocationManager lm = context.getSystemService(LocationManager.class);
+        return new DevicePolicyManagerGatewayImpl(dpm, um, lm, admin);
     }
 
     @Override
@@ -598,6 +604,23 @@ public int getPermissionGrantState(String packageName, String permission) {
         return grantState;
     }
 
+    @Override
+    public void setLocationEnabled(boolean enabled, Consumer<Void> onSuccess,
+            Consumer<Exception> onError) {
+        Log.d(TAG, "setLocationEnabled(" + enabled + ")");
+        try {
+            mDevicePolicyManager.setLocationEnabled(mAdminComponentName, enabled);
+            onSuccess.accept(null);
+        } catch (Exception e) {
+            onError.accept(e);
+        }
+    }
+
+    @Override
+    public boolean isLocationEnabled() {
+        return mLocationManager.isLocationEnabled();
+    }
+
     @Override
     public String toString() {
         return "DevicePolicyManagerGatewayImpl[" + mAdminComponentName + "]";

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

@@ -95,6 +95,8 @@ final class ShellCommand {
     private static final String CMD_GET_APP_RESTRICTIONS = "get-app-restrictions";
     private static final String CMD_SET_PERMISSION_GRANT_STATE = "set-permission-grant-state";
     private static final String CMD_GET_PERMISSION_GRANT_STATE = "get-permission-grant-state";
+    private static final String CMD_SET_LOCATION_ENABLED = "set-location-enabled";
+    private static final String CMD_IS_LOCATION_ENABLED = "is-location-enabled";
 
     private static final String ARG_FLAGS = "--flags";
 
@@ -249,6 +251,12 @@ public void run() {
             case CMD_GET_PERMISSION_GRANT_STATE:
                 execute(() -> getPermissionGrantState());
                 break;
+            case CMD_SET_LOCATION_ENABLED:
+                execute(() -> setLocationEnabled());
+                break;
+            case CMD_IS_LOCATION_ENABLED:
+                execute(() -> isLocationEnabled());
+                break;
             default:
                 mWriter.printf("Invalid command: %s\n\n", cmd);
                 showUsage();
@@ -346,6 +354,10 @@ private void showUsage() {
                 CMD_SET_PERMISSION_GRANT_STATE);
         mWriter.printf("\t%s <PKG> <PERMISSION> - get the grant state for the given permission / "
                 + "package\n", CMD_GET_PERMISSION_GRANT_STATE);
+        mWriter.printf("\t%s <true|false> - set location enabled for the user\n",
+                CMD_SET_LOCATION_ENABLED);
+        mWriter.printf("\t%s - get whether location is enabled for the user\n",
+                CMD_IS_LOCATION_ENABLED);
 
         // Separator for S / pre-S commands - do NOT remove line to avoid cherry-pick conflicts
     }
@@ -791,6 +803,20 @@ private void getPermissionGrantState() {
                 Util.grantStateToString(grantState));
     }
 
+    private void setLocationEnabled() {
+        // TODO(b/171350084): check args
+        boolean enabled = Boolean.parseBoolean(mArgs[1]);
+
+        mDevicePolicyManagerGateway.setLocationEnabled(enabled,
+                (v) -> onSuccess("Set location enabled to %b", enabled),
+                (e) -> onError(e, "Error setting location enabled to %b", enabled));
+    }
+
+    private void isLocationEnabled() {
+        boolean enabled = mDevicePolicyManagerGateway.isLocationEnabled();
+        mWriter.printf("Location enabled: %b\n", enabled);
+    }
+
     private static String permittedToString(boolean permitted) {
         return permitted ? "PERMITTED" : "NOT PERMITTED";
     }

app/src/main/java/com/afwsamples/testdpc/policy/PolicyManagementFragment.java

@@ -451,6 +451,7 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
     private UserManager mUserManager;
     private TelephonyManager mTelephonyManager;
     private AccountManager mAccountManager;
+    private LocationManager mLocationManager;
 
     private DpcPreference mInstallExistingPackagePreference;
 
@@ -520,8 +521,9 @@ public void onCreate(Bundle savedInstanceState) {
         mDevicePolicyManager = (DevicePolicyManager) getActivity().getSystemService(
                 Context.DEVICE_POLICY_SERVICE);
         mUserManager = (UserManager) getActivity().getSystemService(Context.USER_SERVICE);
+        mLocationManager = getActivity().getSystemService(LocationManager.class);
         mDevicePolicyManagerGateway = new DevicePolicyManagerGatewayImpl(mDevicePolicyManager,
-                mUserManager, mAdminComponentName);
+                mUserManager, mLocationManager, mAdminComponentName);
         mTelephonyManager = (TelephonyManager) getActivity()
                 .getSystemService(Context.TELEPHONY_SERVICE);
         mAccountManager = AccountManager.get(getActivity());
@@ -2440,8 +2442,7 @@ private void reloadLocationModeUi() {
 
     @TargetApi(Util.R_VERSION_CODE)
     private void reloadLocationEnabledUi() {
-        LocationManager locationManager = getActivity().getSystemService(LocationManager.class);
-        mSetLocationEnabledPreference.setChecked(locationManager.isLocationEnabled());
+        mSetLocationEnabledPreference.setChecked(mLocationManager.isLocationEnabled());
     }
 
     @TargetApi(Util.R_VERSION_CODE)