hostsidetests/devicepolicy/src/com/android/cts/devicepolicy/BaseDeviceAdminServiceTest.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2017 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package com.android.cts.devicepolicy;

 import com.android.tradefed.log.LogUtil.CLog;

 import java.util.concurrent.TimeUnit;

 public abstract class BaseDeviceAdminServiceTest extends BaseDevicePolicyTest {
     protected static final String OWNER_PKG = "com.android.cts.deviceadminservice";
     protected static final String OWNER_PKG_B = "com.android.cts.deviceadminserviceb";

     protected static final String OWNER_APK_1 = "CtsDeviceAdminService1.apk";
     protected static final String OWNER_APK_2 = "CtsDeviceAdminService2.apk";
     protected static final String OWNER_APK_3 = "CtsDeviceAdminService3.apk";
     protected static final String OWNER_APK_4 = "CtsDeviceAdminService4.apk";

     protected static final String OWNER_APK_B = "CtsDeviceAdminServiceB.apk";

     protected static final String ADMIN_RECEIVER_TEST_CLASS = ".MyOwner";

     protected static final String OWNER_COMPONENT = OWNER_PKG + "/" + ADMIN_RECEIVER_TEST_CLASS;
     protected static final String OWNER_SERVICE = OWNER_PKG + "/.MyService";
     protected static final String OWNER_SERVICE2 = OWNER_PKG + "/.MyService2";

     protected static final String OWNER_COMPONENT_B = OWNER_PKG_B + "/"
             + OWNER_PKG + ADMIN_RECEIVER_TEST_CLASS;
     protected static final String OWNER_SERVICE_B = OWNER_PKG_B + "/"
             + OWNER_PKG + ".MyService";

     private static final int TIMEOUT_SECONDS = 3 * 60;

     private boolean mMultiUserSupported;

     @Override
     protected void setUp() throws Exception {
         super.setUp();

         mMultiUserSupported = getMaxNumberOfUsersSupported() > 1 && getDevice().getApiLevel() >= 21;
     }

     @Override
     protected void tearDown() throws Exception {
         if (isTestEnabled()) {
             removeAdmin(OWNER_COMPONENT, getUserId());
             removeAdmin(OWNER_COMPONENT_B, getUserId());
             getDevice().uninstallPackage(OWNER_PKG);
             getDevice().uninstallPackage(OWNER_PKG_B);
         }
         super.tearDown();
     }

     protected abstract int getUserId();

     protected abstract boolean isTestEnabled();

     protected void executeDeviceTestMethod(String className, String testName) throws Exception {
         runDeviceTestsAsUser(OWNER_PKG, className, testName, getUserId());
     }

     protected interface RunnableWithThrowable {
         void run() throws Throwable;
     }

     protected void withRetry(RunnableWithThrowable test) throws Throwable {
         final long until = System.nanoTime() + TimeUnit.SECONDS.toNanos(TIMEOUT_SECONDS);

         Thread.sleep(500);

         Throwable lastThrowable = null;
         while (System.nanoTime() < until) {
             try {
                 test.run();
                 // Pass, return.
                 return;
             } catch (Throwable th) {
                 lastThrowable = th;
             }
             Thread.sleep(3000);
         }
         if (lastThrowable != null) {
             throw lastThrowable;
         }
         fail("Internal error: test didn't run, exception not thrown.");
     }

     protected abstract void setAsOwnerOrFail(String component) throws Exception;

     public void testAll() throws Throwable {
         if (!isTestEnabled()) {
             return;
         }

         // Install
         installAppAsUser(OWNER_APK_1, getUserId());
         setAsOwnerOrFail(OWNER_COMPONENT);

         withRetry(() -> assertServiceBound(OWNER_SERVICE));

         // Remove admin.
         removeAdmin(OWNER_COMPONENT, getUserId());
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

         // Overwrite -> update.
         installAppAsUser(OWNER_APK_1, getUserId());
         setAsOwnerOrFail(OWNER_COMPONENT);
         withRetry(() -> assertServiceBound(OWNER_SERVICE));

         installAppAsUser(OWNER_APK_2, getUserId());
         withRetry(() -> assertServiceBound(OWNER_SERVICE)); // Should still be bound.

         // Service exported -> not bound.
         installAppAsUser(OWNER_APK_3, getUserId());
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

         // Recover.
         installAppAsUser(OWNER_APK_2, getUserId());
         withRetry(() -> assertServiceBound(OWNER_SERVICE));

         // Multiple service found -> not bound.
         installAppAsUser(OWNER_APK_4, getUserId());
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

         // Disable service1 -> now there's only one service, so should be bound.
         executeDeviceTestMethod(".ComponentController", "testDisableService1");
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
         withRetry(() -> assertServiceBound(OWNER_SERVICE2));

         executeDeviceTestMethod(".ComponentController", "testDisableService2");
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

         executeDeviceTestMethod(".ComponentController", "testEnableService1");
         withRetry(() -> assertServiceBound(OWNER_SERVICE));
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

         executeDeviceTestMethod(".ComponentController", "testEnableService2");
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

         // Remove admin.
         removeAdmin(OWNER_COMPONENT, getUserId());
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

         // Retry with package 1 and remove admin.
         installAppAsUser(OWNER_APK_1, getUserId());
         setAsOwnerOrFail(OWNER_COMPONENT);
         withRetry(() -> assertServiceBound(OWNER_SERVICE));

         removeAdmin(OWNER_COMPONENT, getUserId());
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

         // Now install package B and make it the owner.  OWNER_APK_1 still exists, but it shouldn't
         // interfere.
         installAppAsUser(OWNER_APK_B, getUserId());
         setAsOwnerOrFail(OWNER_COMPONENT_B);
         withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
         withRetry(() -> assertServiceBound(OWNER_SERVICE_B));
     }

     private String rumpDumpSysService(String component) throws Exception {
         final String command = "dumpsys activity services " + component;
         final String commandOutput = getDevice().executeShellCommand(command);
         CLog.d("Output for command " + command + ":\n" + commandOutput);
         return commandOutput;
     }

     private void assertServiceBound(String component) throws Exception {
         final String commandOutput = rumpDumpSysService(component);
         for (String line : commandOutput.split("\r*\n")) {
             if (line.contains("ConnectionRecord") && line.contains(component)) {
                 return;
             }
         }
         fail("Service " + OWNER_SERVICE + " not bound.  Output was:\n" + commandOutput);
     }

     private void assertServiceNotBound(String component) throws Exception {
         final String commandOutput = rumpDumpSysService(component);
         for (String line : commandOutput.split("\r*\n")) {
             if (line.contains("ConnectionRecord") && line.contains(component)) {
                 fail("Service " + OWNER_SERVICE + " is bound.  Output was:\n" + commandOutput);
             }
         }
     }

 /* When the service is bound, "dumpsys activity services" shows something like this:
   * ServiceRecord{1525afe u0 com.android.cts.deviceadminservice/.MyService}
     intent={cmp=com.android.cts.deviceadminservice/.MyService}
     packageName=com.android.cts.deviceadminservice
     processName=com.android.cts.deviceadminservice
     baseDir=/data/app/com.android.cts.deviceadminservice-kXKTlCILmDfib2P76FI75A==/base.apk
     dataDir=/data/user/0/com.android.cts.deviceadminservice
     app=ProcessRecord{205d686 22751:com.android.cts.deviceadminservice/u0a143}
     createTime=-3s957ms startingBgTimeout=--
     lastActivity=-3s927ms restartTime=-3s927ms createdFromFg=true
     Bindings:
     * IntentBindRecord{57d5b47 CREATE}:
       intent={cmp=com.android.cts.deviceadminservice/.MyService}
       binder=android.os.BinderProxy@819af74
       requested=true received=true hasBound=true doRebind=false
       * Client AppBindRecord{e1d3c9d ProcessRecord{a8162c0 757:system/1000}}
         Per-process Connections:
           ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}
     All Connections:
       ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}

   * ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}
     binding=AppBindRecord{e1d3c9d com.android.cts.deviceadminservice/.MyService:system}
     conn=android.app.LoadedApk$ServiceDispatcher$InnerConnection@51e9080 flags=0x4000001
  */
 }
	/*
	* Copyright (C) 2017 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package com.android.cts.devicepolicy;

	import com.android.tradefed.log.LogUtil.CLog;

	import java.util.concurrent.TimeUnit;

	public abstract class BaseDeviceAdminServiceTest extends BaseDevicePolicyTest {
	protected static final String OWNER_PKG = "com.android.cts.deviceadminservice";
	protected static final String OWNER_PKG_B = "com.android.cts.deviceadminserviceb";

	protected static final String OWNER_APK_1 = "CtsDeviceAdminService1.apk";
	protected static final String OWNER_APK_2 = "CtsDeviceAdminService2.apk";
	protected static final String OWNER_APK_3 = "CtsDeviceAdminService3.apk";
	protected static final String OWNER_APK_4 = "CtsDeviceAdminService4.apk";

	protected static final String OWNER_APK_B = "CtsDeviceAdminServiceB.apk";

	protected static final String ADMIN_RECEIVER_TEST_CLASS = ".MyOwner";

	protected static final String OWNER_COMPONENT = OWNER_PKG + "/" + ADMIN_RECEIVER_TEST_CLASS;
	protected static final String OWNER_SERVICE = OWNER_PKG + "/.MyService";
	protected static final String OWNER_SERVICE2 = OWNER_PKG + "/.MyService2";

	protected static final String OWNER_COMPONENT_B = OWNER_PKG_B + "/"
	+ OWNER_PKG + ADMIN_RECEIVER_TEST_CLASS;
	protected static final String OWNER_SERVICE_B = OWNER_PKG_B + "/"
	+ OWNER_PKG + ".MyService";

	private static final int TIMEOUT_SECONDS = 3 * 60;

	private boolean mMultiUserSupported;

	@Override
	protected void setUp() throws Exception {
	super.setUp();

	mMultiUserSupported = getMaxNumberOfUsersSupported() > 1 && getDevice().getApiLevel() >= 21;
	}

	@Override
	protected void tearDown() throws Exception {
	if (isTestEnabled()) {
	removeAdmin(OWNER_COMPONENT, getUserId());
	removeAdmin(OWNER_COMPONENT_B, getUserId());
	getDevice().uninstallPackage(OWNER_PKG);
	getDevice().uninstallPackage(OWNER_PKG_B);
	}
	super.tearDown();
	}

	protected abstract int getUserId();

	protected abstract boolean isTestEnabled();

	protected void executeDeviceTestMethod(String className, String testName) throws Exception {
	runDeviceTestsAsUser(OWNER_PKG, className, testName, getUserId());
	}

	protected interface RunnableWithThrowable {
	void run() throws Throwable;
	}

	protected void withRetry(RunnableWithThrowable test) throws Throwable {
	final long until = System.nanoTime() + TimeUnit.SECONDS.toNanos(TIMEOUT_SECONDS);

	Thread.sleep(500);

	Throwable lastThrowable = null;
	while (System.nanoTime() < until) {
	try {
	test.run();
	// Pass, return.
	return;
	} catch (Throwable th) {
	lastThrowable = th;
	}
	Thread.sleep(3000);
	}
	if (lastThrowable != null) {
	throw lastThrowable;
	}
	fail("Internal error: test didn't run, exception not thrown.");
	}

	protected abstract void setAsOwnerOrFail(String component) throws Exception;

	public void testAll() throws Throwable {
	if (!isTestEnabled()) {
	return;
	}

	// Install
	installAppAsUser(OWNER_APK_1, getUserId());
	setAsOwnerOrFail(OWNER_COMPONENT);

	withRetry(() -> assertServiceBound(OWNER_SERVICE));

	// Remove admin.
	removeAdmin(OWNER_COMPONENT, getUserId());
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

	// Overwrite -> update.
	installAppAsUser(OWNER_APK_1, getUserId());
	setAsOwnerOrFail(OWNER_COMPONENT);
	withRetry(() -> assertServiceBound(OWNER_SERVICE));

	installAppAsUser(OWNER_APK_2, getUserId());
	withRetry(() -> assertServiceBound(OWNER_SERVICE)); // Should still be bound.

	// Service exported -> not bound.
	installAppAsUser(OWNER_APK_3, getUserId());
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

	// Recover.
	installAppAsUser(OWNER_APK_2, getUserId());
	withRetry(() -> assertServiceBound(OWNER_SERVICE));

	// Multiple service found -> not bound.
	installAppAsUser(OWNER_APK_4, getUserId());
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

	// Disable service1 -> now there's only one service, so should be bound.
	executeDeviceTestMethod(".ComponentController", "testDisableService1");
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
	withRetry(() -> assertServiceBound(OWNER_SERVICE2));

	executeDeviceTestMethod(".ComponentController", "testDisableService2");
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

	executeDeviceTestMethod(".ComponentController", "testEnableService1");
	withRetry(() -> assertServiceBound(OWNER_SERVICE));
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

	executeDeviceTestMethod(".ComponentController", "testEnableService2");
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE2));

	// Remove admin.
	removeAdmin(OWNER_COMPONENT, getUserId());
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

	// Retry with package 1 and remove admin.
	installAppAsUser(OWNER_APK_1, getUserId());
	setAsOwnerOrFail(OWNER_COMPONENT);
	withRetry(() -> assertServiceBound(OWNER_SERVICE));

	removeAdmin(OWNER_COMPONENT, getUserId());
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));

	// Now install package B and make it the owner. OWNER_APK_1 still exists, but it shouldn't
	// interfere.
	installAppAsUser(OWNER_APK_B, getUserId());
	setAsOwnerOrFail(OWNER_COMPONENT_B);
	withRetry(() -> assertServiceNotBound(OWNER_SERVICE));
	withRetry(() -> assertServiceBound(OWNER_SERVICE_B));
	}

	private String rumpDumpSysService(String component) throws Exception {
	final String command = "dumpsys activity services " + component;
	final String commandOutput = getDevice().executeShellCommand(command);
	CLog.d("Output for command " + command + ":\n" + commandOutput);
	return commandOutput;
	}

	private void assertServiceBound(String component) throws Exception {
	final String commandOutput = rumpDumpSysService(component);
	for (String line : commandOutput.split("\r*\n")) {
	if (line.contains("ConnectionRecord") && line.contains(component)) {
	return;
	}
	}
	fail("Service " + OWNER_SERVICE + " not bound. Output was:\n" + commandOutput);
	}

	private void assertServiceNotBound(String component) throws Exception {
	final String commandOutput = rumpDumpSysService(component);
	for (String line : commandOutput.split("\r*\n")) {
	if (line.contains("ConnectionRecord") && line.contains(component)) {
	fail("Service " + OWNER_SERVICE + " is bound. Output was:\n" + commandOutput);
	}
	}
	}

	/* When the service is bound, "dumpsys activity services" shows something like this:
	* ServiceRecord{1525afe u0 com.android.cts.deviceadminservice/.MyService}
	intent={cmp=com.android.cts.deviceadminservice/.MyService}
	packageName=com.android.cts.deviceadminservice
	processName=com.android.cts.deviceadminservice
	baseDir=/data/app/com.android.cts.deviceadminservice-kXKTlCILmDfib2P76FI75A==/base.apk
	dataDir=/data/user/0/com.android.cts.deviceadminservice
	app=ProcessRecord{205d686 22751:com.android.cts.deviceadminservice/u0a143}
	createTime=-3s957ms startingBgTimeout=--
	lastActivity=-3s927ms restartTime=-3s927ms createdFromFg=true
	Bindings:
	* IntentBindRecord{57d5b47 CREATE}:
	intent={cmp=com.android.cts.deviceadminservice/.MyService}
	binder=android.os.BinderProxy@819af74
	requested=true received=true hasBound=true doRebind=false
	* Client AppBindRecord{e1d3c9d ProcessRecord{a8162c0 757:system/1000}}
	Per-process Connections:
	ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}
	All Connections:
	ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}

	* ConnectionRecord{10ab6b9 u0 CR FGS com.android.cts.deviceadminservice/.MyService:@51e9080}
	binding=AppBindRecord{e1d3c9d com.android.cts.deviceadminservice/.MyService:system}
	conn=android.app.LoadedApk$ServiceDispatcher$InnerConnection@51e9080 flags=0x4000001
	*/
	}