| /* |
| * Copyright (C) 2016 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.server.pm; |
| |
| import static android.content.pm.UserInfo.FLAG_DEMO; |
| import static android.content.pm.UserInfo.FLAG_EPHEMERAL; |
| import static android.content.pm.UserInfo.FLAG_FULL; |
| import static android.content.pm.UserInfo.FLAG_GUEST; |
| import static android.content.pm.UserInfo.FLAG_MANAGED_PROFILE; |
| import static android.content.pm.UserInfo.FLAG_PROFILE; |
| import static android.content.pm.UserInfo.FLAG_RESTRICTED; |
| import static android.content.pm.UserInfo.FLAG_SYSTEM; |
| import static android.os.UserManager.USER_TYPE_FULL_DEMO; |
| import static android.os.UserManager.USER_TYPE_FULL_GUEST; |
| import static android.os.UserManager.USER_TYPE_FULL_RESTRICTED; |
| import static android.os.UserManager.USER_TYPE_FULL_SECONDARY; |
| import static android.os.UserManager.USER_TYPE_FULL_SYSTEM; |
| import static android.os.UserManager.USER_TYPE_PROFILE_MANAGED; |
| import static android.os.UserManager.USER_TYPE_SYSTEM_HEADLESS; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertTrue; |
| |
| import android.annotation.UserIdInt; |
| import android.content.pm.UserInfo; |
| import android.content.pm.UserInfo.UserInfoFlag; |
| import android.os.Looper; |
| import android.os.Parcel; |
| import android.os.UserHandle; |
| import android.os.UserManagerInternal; |
| import android.text.TextUtils; |
| |
| import androidx.test.InstrumentationRegistry; |
| import androidx.test.filters.MediumTest; |
| import androidx.test.runner.AndroidJUnit4; |
| |
| import com.android.server.LocalServices; |
| import com.android.server.pm.UserManagerService.UserData; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.DataOutputStream; |
| import java.util.List; |
| |
| /** |
| * <p>Run with:<pre> |
| * runtest -c com.android.server.pm.UserManagerServiceUserInfoTest frameworks-services |
| * </pre> |
| */ |
| @RunWith(AndroidJUnit4.class) |
| @MediumTest |
| public class UserManagerServiceUserInfoTest { |
| private UserManagerService mUserManagerService; |
| |
| @Before |
| public void setup() { |
| // Currently UserManagerService cannot be instantiated twice inside a VM without a cleanup |
| // TODO: Remove once UMS supports proper dependency injection |
| if (Looper.myLooper() == null) { |
| Looper.prepare(); |
| } |
| LocalServices.removeServiceForTest(UserManagerInternal.class); |
| mUserManagerService = new UserManagerService(InstrumentationRegistry.getContext()); |
| |
| // The tests assume that the device has one user and its the system user. |
| List<UserInfo> users = mUserManagerService.getUsers(/* excludeDying */ false); |
| assertEquals("Multiple users so this test can't run.", 1, users.size()); |
| assertEquals("Only user present isn't the system user.", |
| UserHandle.USER_SYSTEM, users.get(0).id); |
| } |
| |
| @Test |
| public void testWriteReadUserInfo() throws Exception { |
| UserData data = new UserData(); |
| data.info = createUser(); |
| |
| ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
| DataOutputStream out = new DataOutputStream(baos); |
| mUserManagerService.writeUserLP(data, out); |
| byte[] bytes = baos.toByteArray(); |
| |
| UserData read = mUserManagerService.readUserLP( |
| data.info.id, new ByteArrayInputStream(bytes)); |
| |
| assertUserInfoEquals(data.info, read.info); |
| } |
| |
| @Test |
| public void testParcelUnparcelUserInfo() throws Exception { |
| UserInfo info = createUser(); |
| |
| Parcel out = Parcel.obtain(); |
| info.writeToParcel(out, 0); |
| byte[] data = out.marshall(); |
| out.recycle(); |
| |
| Parcel in = Parcel.obtain(); |
| in.unmarshall(data, 0, data.length); |
| in.setDataPosition(0); |
| UserInfo read = UserInfo.CREATOR.createFromParcel(in); |
| in.recycle(); |
| |
| assertUserInfoEquals(info, read); |
| } |
| |
| @Test |
| public void testGetUserName() throws Exception { |
| assertFalse("System user name shouldn't be set", |
| mUserManagerService.isUserNameSet(UserHandle.USER_SYSTEM)); |
| UserInfo userInfo = mUserManagerService.getUserInfo(UserHandle.USER_SYSTEM); |
| assertFalse("A system provided name should be returned for primary user", |
| TextUtils.isEmpty(userInfo.name)); |
| |
| userInfo = createUser(); |
| userInfo.partial = false; |
| final int TEST_ID = 100; |
| userInfo.id = TEST_ID; |
| mUserManagerService.putUserInfo(userInfo); |
| assertTrue("Test user name must be set", mUserManagerService.isUserNameSet(TEST_ID)); |
| assertEquals("A Name", mUserManagerService.getUserInfo(TEST_ID).name); |
| } |
| |
| /** Test UMS.getUserTypeForUser(). */ |
| @Test |
| public void testGetUserTypeForUser() throws Exception { |
| final String typeSys = mUserManagerService.getUserTypeForUser(UserHandle.USER_SYSTEM); |
| assertTrue("System user was of invalid type " + typeSys, |
| typeSys.equals(USER_TYPE_SYSTEM_HEADLESS) || typeSys.equals(USER_TYPE_FULL_SYSTEM)); |
| |
| final int testId = 100; |
| final String typeName = "A type"; |
| UserInfo userInfo = createUser(testId, 0, typeName); |
| mUserManagerService.putUserInfo(userInfo); |
| assertEquals(typeName, mUserManagerService.getUserTypeForUser(testId)); |
| } |
| |
| /** Tests upgradeIfNecessaryLP (but without locking) for upgrading from version 8 to 9+. */ |
| @Test |
| public void testUpgradeIfNecessaryLP_9() { |
| final int versionToTest = 9; |
| |
| mUserManagerService.putUserInfo(createUser(100, FLAG_MANAGED_PROFILE, null)); |
| mUserManagerService.putUserInfo(createUser(101, |
| FLAG_GUEST | FLAG_EPHEMERAL | FLAG_FULL, null)); |
| mUserManagerService.putUserInfo(createUser(102, FLAG_RESTRICTED | FLAG_FULL, null)); |
| mUserManagerService.putUserInfo(createUser(103, FLAG_FULL, null)); |
| mUserManagerService.putUserInfo(createUser(104, FLAG_SYSTEM, null)); |
| mUserManagerService.putUserInfo(createUser(105, FLAG_SYSTEM | FLAG_FULL, null)); |
| mUserManagerService.putUserInfo(createUser(106, FLAG_DEMO | FLAG_FULL, null)); |
| |
| mUserManagerService.upgradeIfNecessaryLP(null, versionToTest - 1); |
| |
| assertEquals(USER_TYPE_PROFILE_MANAGED, mUserManagerService.getUserTypeForUser(100)); |
| assertTrue((mUserManagerService.getUserInfo(100).flags & FLAG_PROFILE) != 0); |
| |
| assertEquals(USER_TYPE_FULL_GUEST, mUserManagerService.getUserTypeForUser(101)); |
| |
| assertEquals(USER_TYPE_FULL_RESTRICTED, mUserManagerService.getUserTypeForUser(102)); |
| assertTrue((mUserManagerService.getUserInfo(102).flags & FLAG_PROFILE) == 0); |
| |
| assertEquals(USER_TYPE_FULL_SECONDARY, mUserManagerService.getUserTypeForUser(103)); |
| assertTrue((mUserManagerService.getUserInfo(103).flags & FLAG_PROFILE) == 0); |
| |
| assertEquals(USER_TYPE_SYSTEM_HEADLESS, mUserManagerService.getUserTypeForUser(104)); |
| |
| assertEquals(USER_TYPE_FULL_SYSTEM, mUserManagerService.getUserTypeForUser(105)); |
| |
| assertEquals(USER_TYPE_FULL_DEMO, mUserManagerService.getUserTypeForUser(106)); |
| } |
| |
| /** Creates a UserInfo with the given flags and userType. */ |
| private UserInfo createUser(@UserIdInt int userId, @UserInfoFlag int flags, String userType) { |
| return new UserInfo(userId, "A Name", "A path", flags, userType); |
| } |
| |
| private UserInfo createUser() { |
| UserInfo user = new UserInfo(/*id*/ 21, "A Name", "A path", /*flags*/ 0x0ff0ff, "A type"); |
| user.serialNumber = 5; |
| user.creationTime = 4L << 32; |
| user.lastLoggedInTime = 5L << 32; |
| user.lastLoggedInFingerprint = "afingerprint"; |
| user.profileGroupId = 45; |
| user.restrictedProfileParentId = 4; |
| user.profileBadge = 2; |
| user.partial = true; |
| user.guestToRemove = true; |
| user.preCreated = true; |
| return user; |
| } |
| |
| private void assertUserInfoEquals(UserInfo one, UserInfo two) { |
| assertEquals("Id not preserved", one.id, two.id); |
| assertEquals("Name not preserved", one.name, two.name); |
| assertEquals("Icon path not preserved", one.iconPath, two.iconPath); |
| assertEquals("Flags not preserved", one.flags, two.flags); |
| assertEquals("UserType not preserved", one.userType, two.userType); |
| assertEquals("profile group not preserved", one.profileGroupId, |
| two.profileGroupId); |
| assertEquals("restricted profile parent not preseved", one.restrictedProfileParentId, |
| two.restrictedProfileParentId); |
| assertEquals("profile badge not preseved", one.profileBadge, two.profileBadge); |
| assertEquals("partial not preseved", one.partial, two.partial); |
| assertEquals("guestToRemove not preseved", one.guestToRemove, two.guestToRemove); |
| assertEquals("preCreated not preseved", one.preCreated, two.preCreated); |
| } |
| } |