Add managed system updates to TestDPC.

Bug: 116511569
Test: manual
Change-Id: I2888aeb57e2a393649d219ac23a8935a0baf852c

Author: Alex Kershaw

app/src/main/AndroidManifest.xml

@@ -27,6 +27,8 @@
     <uses-permission android:name="android.permission.RECEIVE_BOOT_COMPLETED" />
     <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE"/>
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE"/>
+    <uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE"/>
+    <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE"/>
 
     <application
             android:allowBackup="true"

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

@@ -33,6 +33,7 @@
 import android.app.Fragment;
 import android.app.FragmentManager;
 import android.app.admin.DevicePolicyManager;
+import android.app.admin.DevicePolicyManager.InstallUpdateCallback;
 import android.app.admin.SystemUpdateInfo;
 import android.content.ComponentName;
 import android.content.ContentResolver;
@@ -80,7 +81,6 @@
 import android.widget.RadioButton;
 import android.widget.RadioGroup;
 import android.widget.Toast;
-
 import com.afwsamples.testdpc.AddAccountActivity;
 import com.afwsamples.testdpc.BuildConfig;
 import com.afwsamples.testdpc.CrossProfileAppsFragment;
@@ -92,6 +92,7 @@
 import com.afwsamples.testdpc.common.BaseSearchablePolicyPreferenceFragment;
 import com.afwsamples.testdpc.common.CertificateUtil;
 import com.afwsamples.testdpc.common.MediaDisplayFragment;
+import com.afwsamples.testdpc.common.PackageInstallationUtils;
 import com.afwsamples.testdpc.common.UserArrayAdapter;
 import com.afwsamples.testdpc.common.Util;
 import com.afwsamples.testdpc.common.preference.CustomConstraint;
@@ -100,7 +101,6 @@
 import com.afwsamples.testdpc.common.preference.DpcPreferenceHelper;
 import com.afwsamples.testdpc.common.preference.DpcSwitchPreference;
 import com.afwsamples.testdpc.comp.BindDeviceAdminFragment;
-import com.afwsamples.testdpc.common.PackageInstallationUtils;
 import com.afwsamples.testdpc.policy.blockuninstallation.BlockUninstallationInfoArrayAdapter;
 import com.afwsamples.testdpc.policy.certificate.DelegatedCertInstallerFragment;
 import com.afwsamples.testdpc.policy.keyguard.LockScreenPolicyFragment;
@@ -126,7 +126,6 @@
 import com.afwsamples.testdpc.profilepolicy.permission.ManageAppPermissionsFragment;
 import com.afwsamples.testdpc.transferownership.PickTransferComponentFragment;
 import com.afwsamples.testdpc.util.MainThreadExecutor;
-
 import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.FileNotFoundException;
@@ -371,6 +370,8 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
 
     private static final String MANAGE_OVERRIDE_APN_KEY = "manage_override_apn";
 
+    private static final String MANAGED_SYSTEM_UPDATES_KEY = "managed_system_updates";
+
     private static final String SET_PRIVATE_DNS_MODE_KEY = "set_private_dns_mode";
 
     private static final String BATTERY_PLUGGED_ANY = Integer.toString(
@@ -621,6 +622,7 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         findPreference(SET_TIME_ZONE_KEY).setOnPreferenceClickListener(this);
 
         findPreference(MANAGE_OVERRIDE_APN_KEY).setOnPreferenceClickListener(this);
+        findPreference(MANAGED_SYSTEM_UPDATES_KEY).setOnPreferenceClickListener(this);
 
         findPreference(CROSS_PROFILE_CALENDAR_KEY).setOnPreferenceClickListener(this);
 
@@ -1061,13 +1063,46 @@ public void onPositiveButtonClicked(String[] lockTaskArray) {
             case MANAGE_OVERRIDE_APN_KEY:
                 showFragment(new OverrideApnFragment());
                 return true;
+            case MANAGED_SYSTEM_UPDATES_KEY:
+                promptInstallUpdate();
+                return true;
             case CROSS_PROFILE_CALENDAR_KEY:
                 showFragment(new CrossProfileCalendarFragment());
                 return true;
         }
         return false;
     }
 
+    @TargetApi(Build.VERSION_CODES.Q)
+    private void promptInstallUpdate() {
+        new AlertDialog.Builder(getActivity())
+                .setMessage(R.string.install_update_prompt)
+                .setTitle(R.string.install_update)
+                .setPositiveButton(
+                        R.string.install_update_prompt_yes, (dialogInterface, i) -> installUpdate())
+                .setNegativeButton(
+                        R.string.install_update_prompt_no, (dialogInterface, i) -> {})
+                .create()
+                .show();
+    }
+
+    @TargetApi(Build.VERSION_CODES.Q)
+    private void installUpdate() {
+        File file = new File(getContext().getFilesDir(), "ota.zip");
+        Uri uri = FileProvider.getUriForFile(
+                getActivity(), BuildConfig.APPLICATION_ID + ".fileprovider", file);
+        mDevicePolicyManager.installSystemUpdate(
+                mAdminComponentName,
+                uri,
+                new MainThreadExecutor(),
+                new InstallUpdateCallback() {
+                  @Override
+                  public void onInstallUpdateError(int errorCode, String errorMessage) {
+                      showToast("Error code: " + errorCode);
+                  }
+                });
+    }
+
     @RequiresApi(api = Build.VERSION_CODES.M)
     private void testKeyCanBeUsedForSigning() {
         KeyChain.choosePrivateKeyAlias(getActivity(), new KeyChainAliasCallback() {

app/src/main/res/values/strings.xml

@@ -1062,4 +1062,9 @@
     <string name="cross_profile_apps_available">TestDPC is installed in another profile, tap the icon to launch it in that profile.</string>
     <string name="cross_profile_apps_not_available">TestDPC is not installed in another profile, try installing it and come back to here.</string>
 
+    <string name="install_update">Install update from file</string>
+    <string name="install_update_prompt">This will attempt to install the update file ota.zip in the TestDPC files directory, if the file exists (i.e. if you pushed it there with adb). Do you want to continue?</string>
+    <string name="install_update_prompt_yes">Yes</string>
+    <string name="install_update_prompt_no">No</string>
+
 </resources>

app/src/main/res/xml/device_policy_header.xml

@@ -695,4 +695,10 @@
             testdpc:admin="any"
             testdpc:minSdkVersion="P" />
     </PreferenceCategory>
+
+    <com.afwsamples.testdpc.common.preference.DpcPreference
+        android:key="managed_system_updates"
+        android:title="@string/install_update"
+        testdpc:admin="deviceOwner"
+        testdpc:minSdkVersion="Q" />
 </PreferenceScreen>

app/src/main/res/xml/filepaths.xml

@@ -17,4 +17,5 @@
 
 <paths>
     <files-path path="media/" name="media" />
+    <files-path path="/" name="allfiles" />
 </paths>