Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Add background protection when work biometrics is shown" into tm-dev

Browse Source
This commit is contained in:
Alex Johnston
2022-05-17 16:35:37 +00:00
committed by Android (Google) Code Review
parent 0b28d29298 b3aa96f89f
commit ed00c662dc
16 changed files with 129 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
core/java/android/app/ITaskStackListener.aidl
View File

@@ -137,7 +137,7 @@ oneway interface ITaskStackListener {
* activities inside it belong to a managed profile user, and that user has just * activities inside it belong to a managed profile user, and that user has just
* been locked. * been locked.
*/ */
void onTaskProfileLocked(int taskId, int userId); void onTaskProfileLocked(in ActivityManager.RunningTaskInfo taskInfo);
/** /**
* Called when a task snapshot got updated. * Called when a task snapshot got updated.

2
core/java/android/app/TaskStackListener.java
View File

@@ -155,7 +155,7 @@ public abstract class TaskStackListener extends ITaskStackListener.Stub {
@Override @Override
@UnsupportedAppUsage(maxTargetSdk = Build.VERSION_CODES.R, trackingBug = 170729553) @UnsupportedAppUsage(maxTargetSdk = Build.VERSION_CODES.R, trackingBug = 170729553)
public void onTaskProfileLocked(int taskId, int userId) throws RemoteException { public void onTaskProfileLocked(RunningTaskInfo taskInfo) throws RemoteException {
} }
@Override @Override

2
libs/WindowManager/Shell/src/com/android/wm/shell/common/TaskStackListenerCallback.java
View File

@@ -38,7 +38,7 @@ public interface TaskStackListenerCallback {
default void onTaskStackChanged() { } default void onTaskStackChanged() { }
default void onTaskProfileLocked(int taskId, int userId) { } default void onTaskProfileLocked(RunningTaskInfo taskInfo) { }
default void onTaskDisplayChanged(int taskId, int newDisplayId) { } default void onTaskDisplayChanged(int taskId, int newDisplayId) { }

8
libs/WindowManager/Shell/src/com/android/wm/shell/common/TaskStackListenerImpl.java
View File

@@ -150,8 +150,8 @@ public class TaskStackListenerImpl extends TaskStackListener implements Handler.
} }
@Override @Override
public void onTaskProfileLocked(int taskId, int userId) { public void onTaskProfileLocked(ActivityManager.RunningTaskInfo taskInfo) {
mMainHandler.obtainMessage(ON_TASK_PROFILE_LOCKED, taskId, userId).sendToTarget(); mMainHandler.obtainMessage(ON_TASK_PROFILE_LOCKED, taskInfo).sendToTarget();
} }
@Override @Override
@@ -341,8 +341,10 @@ public class TaskStackListenerImpl extends TaskStackListener implements Handler.
break; break;
} }
case ON_TASK_PROFILE_LOCKED: { case ON_TASK_PROFILE_LOCKED: {
final ActivityManager.RunningTaskInfo
info = (ActivityManager.RunningTaskInfo) msg.obj;
for (int i = mTaskStackListeners.size() - 1; i >= 0; i--) { for (int i = mTaskStackListeners.size() - 1; i >= 0; i--) {
mTaskStackListeners.get(i).onTaskProfileLocked(msg.arg1, msg.arg2); mTaskStackListeners.get(i).onTaskProfileLocked(info);
} }
break; break;
} }

7
libs/WindowManager/Shell/tests/unittest/src/com/android/wm/shell/common/TaskStackListenerImplTest.java
View File

@@ -109,9 +109,10 @@ public class TaskStackListenerImplTest {
@Test @Test
public void testOnTaskProfileLocked() { public void testOnTaskProfileLocked() {
mImpl.onTaskProfileLocked(1, 2); ActivityManager.RunningTaskInfo info = mock(ActivityManager.RunningTaskInfo.class);
verify(mCallback).onTaskProfileLocked(eq(1), eq(2)); mImpl.onTaskProfileLocked(info);
verify(mOtherCallback).onTaskProfileLocked(eq(1), eq(2)); verify(mCallback).onTaskProfileLocked(eq(info));
verify(mOtherCallback).onTaskProfileLocked(eq(info));
} }
@Test @Test

2
packages/SystemUI/AndroidManifest.xml
View File

@@ -723,7 +723,7 @@
android:excludeFromRecents="true" android:excludeFromRecents="true"
android:stateNotNeeded="true" android:stateNotNeeded="true"
android:resumeWhilePausing="true" android:resumeWhilePausing="true"
android:theme="@android:style/Theme.Translucent.NoTitleBar.Fullscreen"> android:theme="@style/Theme.AppCompat.DayNight.NoActionBar">
<intent-filter> <intent-filter>
<action android:name="android.app.action.CONFIRM_DEVICE_CREDENTIAL_WITH_USER" /> <action android:name="android.app.action.CONFIRM_DEVICE_CREDENTIAL_WITH_USER" />
<category android:name="android.intent.category.DEFAULT" /> <category android:name="android.intent.category.DEFAULT" />

30
packages/SystemUI/res/layout/auth_biometric_background.xml Normal file
View File

@@ -0,0 +1,30 @@
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright (C) 2022 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.
-->
<LinearLayout
xmlns:android="http://schemas.android.com/apk/res/android"
android:layout_width="match_parent"
android:layout_height="match_parent"
android:gravity="center"
android:orientation="vertical">
<ImageView
android:id="@+id/icon"
android:layout_width="80dp"
android:layout_height="80dp"
android:layout_marginBottom="160dp"/>
</LinearLayout>

2
packages/SystemUI/shared/src/com/android/systemui/shared/system/TaskStackChangeListener.java
View File

@@ -61,7 +61,7 @@ public interface TaskStackChangeListener {
onActivityLaunchOnSecondaryDisplayRerouted(); onActivityLaunchOnSecondaryDisplayRerouted();
} }
default void onTaskProfileLocked(int taskId, int userId) { } default void onTaskProfileLocked(RunningTaskInfo taskInfo) { }
default void onTaskCreated(int taskId, ComponentName componentName) { } default void onTaskCreated(int taskId, ComponentName componentName) { }
default void onTaskRemoved(int taskId) { } default void onTaskRemoved(int taskId) { }
default void onTaskMovedToFront(int taskId) { } default void onTaskMovedToFront(int taskId) { }

7
packages/SystemUI/shared/src/com/android/systemui/shared/system/TaskStackChangeListeners.java
View File

@@ -211,8 +211,8 @@ public class TaskStackChangeListeners {
} }
@Override @Override
public void onTaskProfileLocked(int taskId, int userId) { public void onTaskProfileLocked(RunningTaskInfo taskInfo) {
mHandler.obtainMessage(ON_TASK_PROFILE_LOCKED, taskId, userId).sendToTarget(); mHandler.obtainMessage(ON_TASK_PROFILE_LOCKED, taskInfo).sendToTarget();
} }
@Override @Override
@@ -357,8 +357,9 @@ public class TaskStackChangeListeners {
break; break;
} }
case ON_TASK_PROFILE_LOCKED: { case ON_TASK_PROFILE_LOCKED: {
final RunningTaskInfo info = (RunningTaskInfo) msg.obj;
for (int i = mTaskStackListeners.size() - 1; i >= 0; i--) { for (int i = mTaskStackListeners.size() - 1; i >= 0; i--) {
mTaskStackListeners.get(i).onTaskProfileLocked(msg.arg1, msg.arg2); mTaskStackListeners.get(i).onTaskProfileLocked(info);
} }
break; break;
} }

68
packages/SystemUI/src/com/android/systemui/keyguard/WorkLockActivity.java
View File

@@ -17,23 +17,22 @@
package com.android.systemui.keyguard; package com.android.systemui.keyguard;
import static android.app.ActivityManager.TaskDescription; import static android.app.ActivityManager.TaskDescription;
import static android.app.admin.DevicePolicyResources.Strings.SystemUi.WORK_LOCK_ACCESSIBILITY;
import android.annotation.ColorInt;
import android.annotation.UserIdInt; import android.annotation.UserIdInt;
import android.app.Activity; import android.app.Activity;
import android.app.ActivityOptions; import android.app.ActivityOptions;
import android.app.KeyguardManager; import android.app.KeyguardManager;
import android.app.PendingIntent; import android.app.PendingIntent;
import android.app.admin.DevicePolicyManager;
import android.content.BroadcastReceiver; import android.content.BroadcastReceiver;
import android.content.Context; import android.content.Context;
import android.content.Intent; import android.content.Intent;
import android.content.IntentFilter; import android.content.IntentFilter;
import android.graphics.Color; import android.content.pm.PackageManager;
import android.graphics.drawable.Drawable;
import android.os.Bundle; import android.os.Bundle;
import android.os.UserHandle; import android.os.UserHandle;
import android.view.View; import android.os.UserManager;
import android.widget.ImageView;
import com.android.internal.annotations.VisibleForTesting; import com.android.internal.annotations.VisibleForTesting;
import com.android.systemui.R; import com.android.systemui.R;
@@ -52,12 +51,6 @@ import javax.inject.Inject;
public class WorkLockActivity extends Activity { public class WorkLockActivity extends Activity {
private static final String TAG = "WorkLockActivity"; private static final String TAG = "WorkLockActivity";
/**
* Contains a {@link TaskDescription} for the activity being covered.
*/
static final String EXTRA_TASK_DESCRIPTION =
"com.android.systemui.keyguard.extra.TASK_DESCRIPTION";
private static final int REQUEST_CODE_CONFIRM_CREDENTIALS = 1; private static final int REQUEST_CODE_CONFIRM_CREDENTIALS = 1;
/** /**
@@ -65,12 +58,17 @@ public class WorkLockActivity extends Activity {
* @see KeyguardManager * @see KeyguardManager
*/ */
private KeyguardManager mKgm; private KeyguardManager mKgm;
private UserManager mUserManager;
private PackageManager mPackageManager;
private final BroadcastDispatcher mBroadcastDispatcher; private final BroadcastDispatcher mBroadcastDispatcher;
@Inject @Inject
public WorkLockActivity(BroadcastDispatcher broadcastDispatcher) { public WorkLockActivity(BroadcastDispatcher broadcastDispatcher, UserManager userManager,
PackageManager packageManager) {
super(); super();
mBroadcastDispatcher = broadcastDispatcher; mBroadcastDispatcher = broadcastDispatcher;
mUserManager = userManager;
mPackageManager = packageManager;
} }
@Override @Override
@@ -91,15 +89,28 @@ public class WorkLockActivity extends Activity {
// Draw captions overlaid on the content view, so the whole window is one solid color. // Draw captions overlaid on the content view, so the whole window is one solid color.
setOverlayWithDecorCaptionEnabled(true); setOverlayWithDecorCaptionEnabled(true);
// Blank out the activity. When it is on-screen it will look like a Recents thumbnail with // Add background protection that contains a badged icon of the app being opened.
// redaction switched on. setContentView(R.layout.auth_biometric_background);
final DevicePolicyManager dpm = getSystemService(DevicePolicyManager.class); Drawable badgedIcon = getBadgedIcon();
String contentDescription = dpm.getResources().getString( if (badgedIcon != null) {
WORK_LOCK_ACCESSIBILITY, () -> getString(R.string.accessibility_desc_work_lock)); ((ImageView) findViewById(R.id.icon)).setImageDrawable(badgedIcon);
final View blankView = new View(this); }
blankView.setContentDescription(contentDescription); }
blankView.setBackgroundColor(getPrimaryColor());
setContentView(blankView); @VisibleForTesting
protected Drawable getBadgedIcon() {
String packageName = getIntent().getStringExtra(Intent.EXTRA_PACKAGE_NAME);
if (!packageName.isEmpty()) {
try {
return mUserManager.getBadgedIconForUser(mPackageManager.getApplicationIcon(
mPackageManager.getApplicationInfoAsUser(packageName,
PackageManager.ApplicationInfoFlags.of(0), getTargetUserId())),
UserHandle.of(getTargetUserId()));
} catch (PackageManager.NameNotFoundException e) {
// Unable to set the badged icon, show the background protection without an icon.
}
}
return null;
} }
/** /**
@@ -208,19 +219,4 @@ public class WorkLockActivity extends Activity {
final int getTargetUserId() { final int getTargetUserId() {
return getIntent().getIntExtra(Intent.EXTRA_USER_ID, UserHandle.myUserId()); return getIntent().getIntExtra(Intent.EXTRA_USER_ID, UserHandle.myUserId());
} }
@VisibleForTesting
@ColorInt
final int getPrimaryColor() {
final TaskDescription taskDescription = (TaskDescription)
getIntent().getExtra(EXTRA_TASK_DESCRIPTION);
if (taskDescription != null && Color.alpha(taskDescription.getPrimaryColor()) == 255) {
return taskDescription.getPrimaryColor();
} else {
// No task description. Use an organization color set by the policy controller.
final DevicePolicyManager devicePolicyManager = (DevicePolicyManager)
getSystemService(Context.DEVICE_POLICY_SERVICE);
return devicePolicyManager.getOrganizationColorForUser(getTargetUserId());
}
}
} }

23
packages/SystemUI/src/com/android/systemui/keyguard/WorkLockActivityController.java
View File

@@ -52,22 +52,17 @@ public class WorkLockActivityController {
tscl.registerTaskStackListener(mLockListener); tscl.registerTaskStackListener(mLockListener);
} }
private void startWorkChallengeInTask(int taskId, int userId) { private void startWorkChallengeInTask(ActivityManager.RunningTaskInfo info) {
ActivityManager.TaskDescription taskDescription = null; String packageName = info.baseActivity != null ? info.baseActivity.getPackageName() : "";
try {
taskDescription = mIatm.getTaskDescription(taskId);
} catch (RemoteException e) {
Log.w(TAG, "Failed to get description for task=" + taskId);
}
Intent intent = new Intent(KeyguardManager.ACTION_CONFIRM_DEVICE_CREDENTIAL_WITH_USER) Intent intent = new Intent(KeyguardManager.ACTION_CONFIRM_DEVICE_CREDENTIAL_WITH_USER)
.setComponent(new ComponentName(mContext, WorkLockActivity.class)) .setComponent(new ComponentName(mContext, WorkLockActivity.class))
.putExtra(Intent.EXTRA_USER_ID, userId) .putExtra(Intent.EXTRA_USER_ID, info.userId)
.putExtra(WorkLockActivity.EXTRA_TASK_DESCRIPTION, taskDescription) .putExtra(Intent.EXTRA_PACKAGE_NAME, packageName)
.addFlags(Intent.FLAG_ACTIVITY_REORDER_TO_FRONT .addFlags(Intent.FLAG_ACTIVITY_REORDER_TO_FRONT
| Intent.FLAG_ACTIVITY_CLEAR_TOP); | Intent.FLAG_ACTIVITY_CLEAR_TOP);
final ActivityOptions options = ActivityOptions.makeBasic(); final ActivityOptions options = ActivityOptions.makeBasic();
options.setLaunchTaskId(taskId); options.setLaunchTaskId(info.taskId);
options.setTaskOverlay(true, false /* canResume */); options.setTaskOverlay(true, false /* canResume */);
final int result = startActivityAsUser(intent, options.toBundle(), UserHandle.USER_CURRENT); final int result = startActivityAsUser(intent, options.toBundle(), UserHandle.USER_CURRENT);
@@ -77,9 +72,9 @@ public class WorkLockActivityController {
// Starting the activity inside the task failed. We can't be sure why, so to be // Starting the activity inside the task failed. We can't be sure why, so to be
// safe just remove the whole task if it still exists. // safe just remove the whole task if it still exists.
try { try {
mIatm.removeTask(taskId); mIatm.removeTask(info.taskId);
} catch (RemoteException e) { } catch (RemoteException e) {
Log.w(TAG, "Failed to get description for task=" + taskId); Log.w(TAG, "Failed to get description for task=" + info.taskId);
} }
} }
} }
@@ -112,8 +107,8 @@ public class WorkLockActivityController {
private final TaskStackChangeListener mLockListener = new TaskStackChangeListener() { private final TaskStackChangeListener mLockListener = new TaskStackChangeListener() {
@Override @Override
public void onTaskProfileLocked(int taskId, int userId) { public void onTaskProfileLocked(ActivityManager.RunningTaskInfo info) {
startWorkChallengeInTask(taskId, userId); startWorkChallengeInTask(info);
} }
}; };
} }

11
packages/SystemUI/tests/src/com/android/systemui/keyguard/WorkLockActivityControllerTest.java
View File

@@ -60,6 +60,13 @@ import org.mockito.MockitoAnnotations;
public class WorkLockActivityControllerTest extends SysuiTestCase { public class WorkLockActivityControllerTest extends SysuiTestCase {
private static final int USER_ID = 333; private static final int USER_ID = 333;
private static final int TASK_ID = 444; private static final int TASK_ID = 444;
private static final ActivityManager.RunningTaskInfo TASK_INFO =
new ActivityManager.RunningTaskInfo();
static {
TASK_INFO.userId = USER_ID;
TASK_INFO.taskId = TASK_ID;
}
private @Mock Context mContext; private @Mock Context mContext;
private @Mock TaskStackChangeListeners mTaskStackChangeListeners; private @Mock TaskStackChangeListeners mTaskStackChangeListeners;
@@ -91,7 +98,7 @@ public class WorkLockActivityControllerTest extends SysuiTestCase {
setActivityStartCode(TASK_ID, true /*taskOverlay*/, ActivityManager.START_SUCCESS); setActivityStartCode(TASK_ID, true /*taskOverlay*/, ActivityManager.START_SUCCESS);
// And the controller receives a message saying the profile is locked, // And the controller receives a message saying the profile is locked,
mTaskStackListener.onTaskProfileLocked(TASK_ID, USER_ID); mTaskStackListener.onTaskProfileLocked(TASK_INFO);
// The overlay should start and the task the activity started in should not be removed. // The overlay should start and the task the activity started in should not be removed.
verifyStartActivity(TASK_ID, true /*taskOverlay*/); verifyStartActivity(TASK_ID, true /*taskOverlay*/);
@@ -104,7 +111,7 @@ public class WorkLockActivityControllerTest extends SysuiTestCase {
setActivityStartCode(TASK_ID, true /*taskOverlay*/, ActivityManager.START_CLASS_NOT_FOUND); setActivityStartCode(TASK_ID, true /*taskOverlay*/, ActivityManager.START_CLASS_NOT_FOUND);
// And the controller receives a message saying the profile is locked, // And the controller receives a message saying the profile is locked,
mTaskStackListener.onTaskProfileLocked(TASK_ID, USER_ID); mTaskStackListener.onTaskProfileLocked(TASK_INFO);
// The task the activity started in should be removed to prevent the locked task from // The task the activity started in should be removed to prevent the locked task from
// being shown. // being shown.

66
packages/SystemUI/tests/src/com/android/systemui/keyguard/WorkLockActivityTest.java
View File

@@ -16,8 +16,6 @@
package com.android.systemui.keyguard; package com.android.systemui.keyguard;
import static android.app.ActivityManager.TaskDescription;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.eq; import static org.mockito.Mockito.eq;
@@ -25,14 +23,15 @@ import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
import android.annotation.ColorInt;
import android.annotation.UserIdInt; import android.annotation.UserIdInt;
import android.app.KeyguardManager;
import android.app.admin.DevicePolicyManager;
import android.content.Context; import android.content.Context;
import android.content.Intent; import android.content.Intent;
import android.graphics.Color; import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.graphics.drawable.Drawable;
import android.os.Looper; import android.os.Looper;
import android.os.UserHandle;
import android.os.UserManager;
import androidx.test.filters.SmallTest; import androidx.test.filters.SmallTest;
import androidx.test.runner.AndroidJUnit4; import androidx.test.runner.AndroidJUnit4;
@@ -53,18 +52,21 @@ import org.mockito.MockitoAnnotations;
@RunWith(AndroidJUnit4.class) @RunWith(AndroidJUnit4.class)
public class WorkLockActivityTest extends SysuiTestCase { public class WorkLockActivityTest extends SysuiTestCase {
private static final @UserIdInt int USER_ID = 270; private static final @UserIdInt int USER_ID = 270;
private static final String TASK_LABEL = "task label"; private static final String CALLING_PACKAGE_NAME = "com.android.test";
private @Mock DevicePolicyManager mDevicePolicyManager; private @Mock UserManager mUserManager;
private @Mock KeyguardManager mKeyguardManager; private @Mock PackageManager mPackageManager;
private @Mock Context mContext; private @Mock Context mContext;
private @Mock BroadcastDispatcher mBroadcastDispatcher; private @Mock BroadcastDispatcher mBroadcastDispatcher;
private @Mock Drawable mDrawable;
private @Mock Drawable mBadgedDrawable;
private WorkLockActivity mActivity; private WorkLockActivity mActivity;
private static class WorkLockActivityTestable extends WorkLockActivity { private static class WorkLockActivityTestable extends WorkLockActivity {
WorkLockActivityTestable(Context baseContext, BroadcastDispatcher broadcastDispatcher) { WorkLockActivityTestable(Context baseContext, BroadcastDispatcher broadcastDispatcher,
super(broadcastDispatcher); UserManager userManager, PackageManager packageManager) {
super(broadcastDispatcher, userManager, packageManager);
attachBaseContext(baseContext); attachBaseContext(baseContext);
} }
} }
@@ -73,46 +75,26 @@ public class WorkLockActivityTest extends SysuiTestCase {
public void setUp() throws Exception { public void setUp() throws Exception {
MockitoAnnotations.initMocks(this); MockitoAnnotations.initMocks(this);
when(mContext.getSystemService(eq(Context.DEVICE_POLICY_SERVICE)))
.thenReturn(mDevicePolicyManager);
when(mContext.getSystemService(eq(Context.KEYGUARD_SERVICE)))
.thenReturn(mKeyguardManager);
if (Looper.myLooper() == null) { if (Looper.myLooper() == null) {
Looper.prepare(); Looper.prepare();
} }
mActivity = new WorkLockActivityTestable(mContext, mBroadcastDispatcher); mActivity = new WorkLockActivityTestable(mContext, mBroadcastDispatcher, mUserManager,
mPackageManager);
} }
@Test @Test
public void testBackgroundAlwaysOpaque() throws Exception { public void testGetBadgedIcon() throws Exception {
final @ColorInt int orgColor = Color.rgb(250, 199, 67); ApplicationInfo info = new ApplicationInfo();
when(mDevicePolicyManager.getOrganizationColorForUser(eq(USER_ID))).thenReturn(orgColor); when(mPackageManager.getApplicationInfoAsUser(eq(CALLING_PACKAGE_NAME), any(),
eq(USER_ID))).thenReturn(info);
final @ColorInt int opaqueColor= Color.rgb(164, 198, 57); when(mPackageManager.getApplicationIcon(eq(info))).thenReturn(mDrawable);
final @ColorInt int transparentColor = Color.argb(0, 0, 0, 0); when(mUserManager.getBadgedIconForUser(any(), eq(UserHandle.of(USER_ID)))).thenReturn(
TaskDescription opaque = new TaskDescription(null, null, opaqueColor); mBadgedDrawable);
TaskDescription transparent = new TaskDescription(null, null, transparentColor);
// When a task description is provided with a suitable (opaque) primaryColor, it should be
// used as the scrim's background color.
mActivity.setIntent(new Intent() mActivity.setIntent(new Intent()
.putExtra(Intent.EXTRA_USER_ID, USER_ID) .putExtra(Intent.EXTRA_USER_ID, USER_ID)
.putExtra(WorkLockActivity.EXTRA_TASK_DESCRIPTION, opaque)); .putExtra(Intent.EXTRA_PACKAGE_NAME, CALLING_PACKAGE_NAME));
assertEquals(opaqueColor, mActivity.getPrimaryColor());
// When a task description is provided but has no primaryColor / the primaryColor is assertEquals(mBadgedDrawable, mActivity.getBadgedIcon());
// transparent, the organization color should be used instead.
mActivity.setIntent(new Intent()
.putExtra(Intent.EXTRA_USER_ID, USER_ID)
.putExtra(WorkLockActivity.EXTRA_TASK_DESCRIPTION, transparent));
assertEquals(orgColor, mActivity.getPrimaryColor());
// When no task description is provided at all, it should be treated like a transparent
// description and the organization color shown instead.
mActivity.setIntent(new Intent()
.putExtra(Intent.EXTRA_USER_ID, USER_ID));
assertEquals(orgColor, mActivity.getPrimaryColor());
} }
@Test @Test

2
services/core/java/com/android/server/wm/RootWindowContainer.java
View File

@@ -3231,7 +3231,7 @@ class RootWindowContainer extends WindowContainer<DisplayContent>
if (task.getActivity(activity -> !activity.finishing && activity.mUserId == userId) if (task.getActivity(activity -> !activity.finishing && activity.mUserId == userId)
!= null) { != null) {
mService.getTaskChangeNotificationController().notifyTaskProfileLocked( mService.getTaskChangeNotificationController().notifyTaskProfileLocked(
task.mTaskId, userId); task.getTaskInfo());
} }
}, true /* traverseTopToBottom */); }, true /* traverseTopToBottom */);
} }

8
services/core/java/com/android/server/wm/TaskChangeNotificationController.java
View File

@@ -144,7 +144,7 @@ class TaskChangeNotificationController {
}; };
private final TaskStackConsumer mNotifyTaskProfileLocked = (l, m) -> { private final TaskStackConsumer mNotifyTaskProfileLocked = (l, m) -> {
l.onTaskProfileLocked(m.arg1, m.arg2); l.onTaskProfileLocked((RunningTaskInfo) m.obj);
}; };
private final TaskStackConsumer mNotifyTaskSnapshotChanged = (l, m) -> { private final TaskStackConsumer mNotifyTaskSnapshotChanged = (l, m) -> {
@@ -467,9 +467,9 @@ class TaskChangeNotificationController {
* Notify listeners that the task has been put in a locked state because one or more of the * Notify listeners that the task has been put in a locked state because one or more of the
* activities inside it belong to a managed profile user that has been locked. * activities inside it belong to a managed profile user that has been locked.
*/ */
void notifyTaskProfileLocked(int taskId, int userId) { void notifyTaskProfileLocked(ActivityManager.RunningTaskInfo taskInfo) {
final Message msg = mHandler.obtainMessage(NOTIFY_TASK_PROFILE_LOCKED_LISTENERS_MSG, taskId, final Message msg = mHandler.obtainMessage(NOTIFY_TASK_PROFILE_LOCKED_LISTENERS_MSG,
userId); taskInfo);
forAllLocalListeners(mNotifyTaskProfileLocked, msg); forAllLocalListeners(mNotifyTaskProfileLocked, msg);
msg.sendToTarget(); msg.sendToTarget();
} }

4
services/tests/wmtests/src/com/android/server/wm/RootWindowContainerTests.java
View File

@@ -1097,7 +1097,7 @@ public class RootWindowContainerTests extends WindowTestsBase {
TaskChangeNotificationController controller = mAtm.getTaskChangeNotificationController(); TaskChangeNotificationController controller = mAtm.getTaskChangeNotificationController();
spyOn(controller); spyOn(controller);
mWm.mRoot.lockAllProfileTasks(profileUserId); mWm.mRoot.lockAllProfileTasks(profileUserId);
verify(controller).notifyTaskProfileLocked(eq(task.mTaskId), eq(profileUserId)); verify(controller).notifyTaskProfileLocked(any());
// Create the work lock activity on top of the task // Create the work lock activity on top of the task
final ActivityRecord workLockActivity = new ActivityBuilder(mAtm).setTask(task).build(); final ActivityRecord workLockActivity = new ActivityBuilder(mAtm).setTask(task).build();
@@ -1107,7 +1107,7 @@ public class RootWindowContainerTests extends WindowTestsBase {
// Make sure the listener won't be notified again. // Make sure the listener won't be notified again.
clearInvocations(controller); clearInvocations(controller);
mWm.mRoot.lockAllProfileTasks(profileUserId); mWm.mRoot.lockAllProfileTasks(profileUserId);
verify(controller, never()).notifyTaskProfileLocked(anyInt(), anyInt()); verify(controller, never()).notifyTaskProfileLocked(any());
} }
/** /**