Add toggle for Common Criteria mode

Bug: 151226692
Test: builds
Change-Id: I56a04e8052ac85a5ec16bc3c821f598c2fd1caa3

Author: Rubin Xu

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

@@ -400,6 +400,7 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
     private static final String SET_LOCATION_MODE_KEY = "set_location_mode";
     private static final String SUSPEND_PERSONAL_APPS_KEY = "suspend_personal_apps";
     private static final String PROFILE_MAX_TIME_OFF_KEY = "profile_max_time_off";
+    private static final String COMMON_CRITERIA_MODE_KEY = "common_criteria_mode";
 
     private static final String BATTERY_PLUGGED_ANY = Integer.toString(
             BatteryManager.BATTERY_PLUGGED_AC |
@@ -458,6 +459,7 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
     private DpcSwitchPreference mInstallNonMarketAppsPreference;
 
     private DpcSwitchPreference mEnableBackupServicePreference;
+    private DpcSwitchPreference mCommonCriteriaModePreference;
     private SwitchPreference mEnableSecurityLoggingPreference;
     private SwitchPreference mEnableNetworkLoggingPreference;
     private DpcSwitchPreference mSetAutoTimeRequiredPreference;
@@ -608,6 +610,8 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
             ENABLE_BACKUP_SERVICE);
         mEnableBackupServicePreference.setOnPreferenceChangeListener(this);
         mEnableBackupServicePreference.setCustomConstraint(this::validateDeviceOwnerBeforeQ);
+        mCommonCriteriaModePreference = (DpcSwitchPreference) findPreference(
+            COMMON_CRITERIA_MODE_KEY);
         findPreference(REQUEST_BUGREPORT_KEY).setOnPreferenceClickListener(this);
         mEnableSecurityLoggingPreference =
             (SwitchPreference) findPreference(ENABLE_SECURITY_LOGGING);
@@ -1431,6 +1435,9 @@ public boolean onPreferenceChange(Preference preference, Object newValue) {
                 setBackupServiceEnabled((Boolean) newValue);
                 reloadEnableBackupServiceUi();
                 return true;
+            case COMMON_CRITERIA_MODE_KEY:
+                setCommonCriteriaModeEnabled((Boolean) newValue);
+                reloadCommonCriteriaModeUi();
             case ENABLE_SECURITY_LOGGING:
                 setSecurityLoggingEnabled((Boolean) newValue);
                 reloadEnableSecurityLoggingUi();
@@ -1588,6 +1595,18 @@ private void setBackupServiceEnabled(boolean enabled) {
         mDevicePolicyManager.setBackupServiceEnabled(mAdminComponentName, enabled);
     }
 
+    //@TargetApi(VERSION_CODES.R)
+    private void setCommonCriteriaModeEnabled(boolean enabled) {
+        try {
+            ReflectionUtil.invoke(mDevicePolicyManager,
+                "setCommonCriteriaModeEnabled",
+                new Class<?>[]{ComponentName.class, boolean.class},
+                mAdminComponentName, enabled);
+        } catch (ReflectionIsTemporaryException e) {
+            Log.e(TAG, "Error invoking setCommonCriteriaModeEnabled", e);
+        }
+    }
+
     @TargetApi(VERSION_CODES.M)
     private void setKeyGuardDisabled(boolean disabled) {
         if (!mDevicePolicyManager.setKeyguardDisabled(mAdminComponentName, disabled)) {
@@ -2549,7 +2568,22 @@ private void reloadEnableSecurityLoggingUi() {
     private void reloadEnableBackupServiceUi() {
         if (mEnableBackupServicePreference.isEnabled()) {
             mEnableBackupServicePreference.setChecked(mDevicePolicyManager.isBackupServiceEnabled(
+                mAdminComponentName));
+        }
+    }
+
+    //@TargetApi(VERSION_CODES.R)
+    private void reloadCommonCriteriaModeUi() {
+        if (mCommonCriteriaModePreference.isEnabled()) {
+            try {
+                mCommonCriteriaModePreference.setChecked(
+                    (Boolean) ReflectionUtil.invoke(mDevicePolicyManager,
+                    "isCommonCriteriaModeEnabled",
+                        new Class<?>[]{ComponentName.class},
                     mAdminComponentName));
+            } catch (ReflectionIsTemporaryException e) {
+                Log.e(TAG, "Error invoking isCommonCriteriaModeEnabled", e);
+            }
         }
     }
 

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

@@ -622,6 +622,8 @@
     <!-- Backup and Restore for Device Owner -->
     <string name="enable_backup_service">Enable backup service</string>
 
+    <string name="common_criteria_mode">Enable Common Criteria mode</string>
+
     <!-- Suspension of personal apps -->
     <string name="suspend_personal_apps">Suspend personal apps</string>
 

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

@@ -723,6 +723,11 @@
             android:key="enable_backup_service"
             android:title="@string/enable_backup_service"
             testdpc:minSdkVersion="O" />
+        <com.afwsamples.testdpc.common.preference.DpcSwitchPreference
+            android:key="common_criteria_mode"
+            android:title="@string/common_criteria_mode"
+            testdpc:admin="deviceOwner|orgOwnedProfileOwner"
+            testdpc:minSdkVersion="R" />
         <com.afwsamples.testdpc.common.preference.DpcPreference
             android:key="wipe_data"
             android:title="@string/wipe_data"