Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | package com.android.server.devicepolicy; |
| 18 | |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 19 | import static org.mockito.Mockito.mock; |
| 20 | |
Jonathan Scott | 367ebf4 | 2019-05-16 15:13:17 +0100 | [diff] [blame] | 21 | import android.annotation.Nullable; |
| 22 | import android.app.AppOpsManager; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 23 | import android.content.BroadcastReceiver; |
Makoto Onuki | a52562c | 2015-10-01 16:12:31 -0700 | [diff] [blame] | 24 | import android.content.ContentResolver; |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 25 | import android.content.Context; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 26 | import android.content.Intent; |
| 27 | import android.content.IntentFilter; |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 28 | import android.content.pm.ApplicationInfo; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 29 | import android.content.pm.PackageManager; |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 30 | import android.content.res.Resources; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 31 | import android.os.Bundle; |
| 32 | import android.os.Handler; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 33 | import android.os.UserHandle; |
Christine Franks | 361b825 | 2017-06-23 18:12:46 -0700 | [diff] [blame] | 34 | import android.os.UserManagerInternal; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 35 | import android.test.mock.MockContext; |
Robin Lee | 7f5c91c | 2017-02-08 21:27:02 +0000 | [diff] [blame] | 36 | import android.util.ArrayMap; |
yuemingw | e43cdf7 | 2017-10-12 16:52:11 +0100 | [diff] [blame] | 37 | import android.util.ExceptionUtils; |
| 38 | |
Brett Chabot | ec6bcbc | 2018-11-05 21:21:35 -0800 | [diff] [blame] | 39 | import androidx.annotation.NonNull; |
| 40 | |
yuemingw | e43cdf7 | 2017-10-12 16:52:11 +0100 | [diff] [blame] | 41 | import com.android.internal.util.FunctionalUtils; |
Tony Mak | 2f26b79 | 2016-11-28 17:54:51 +0000 | [diff] [blame] | 42 | |
Makoto Onuki | b643fb0 | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 43 | import org.junit.Assert; |
| 44 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 45 | import java.util.ArrayList; |
| 46 | import java.util.List; |
Robin Lee | 7f5c91c | 2017-02-08 21:27:02 +0000 | [diff] [blame] | 47 | import java.util.Map; |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 48 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 49 | /** |
| 50 | * Context used throughout DPMS tests. |
| 51 | */ |
| 52 | public class DpmMockContext extends MockContext { |
| 53 | /** |
| 54 | * User-id of a non-system user we use throughout unit tests. |
| 55 | */ |
| 56 | public static final int CALLER_USER_HANDLE = 20; |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 57 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 58 | /** |
Makoto Onuki | b643fb0 | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 59 | * UID corresponding to {@link #CALLER_USER_HANDLE}. |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 60 | */ |
Makoto Onuki | d932f76 | 2015-09-29 16:53:38 -0700 | [diff] [blame] | 61 | public static final int CALLER_UID = UserHandle.getUid(CALLER_USER_HANDLE, 20123); |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 62 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 63 | /** |
Eran Messeri | bb27189 | 2018-10-17 18:27:50 +0100 | [diff] [blame] | 64 | * UID corresponding to {@link #CALLER_USER_HANDLE}. |
| 65 | */ |
| 66 | public static final int CALLER_MANAGED_PROVISIONING_UID = UserHandle.getUid(CALLER_USER_HANDLE, |
| 67 | 20125); |
| 68 | |
| 69 | /** |
Makoto Onuki | b643fb0 | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 70 | * UID used when a caller is on the system user. |
| 71 | */ |
Makoto Onuki | d932f76 | 2015-09-29 16:53:38 -0700 | [diff] [blame] | 72 | public static final int CALLER_SYSTEM_USER_UID = 20321; |
Makoto Onuki | b643fb0 | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 73 | |
| 74 | /** |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 75 | * PID of the caller. |
| 76 | */ |
| 77 | public static final int CALLER_PID = 22222; |
| 78 | |
| 79 | /** |
| 80 | * UID of the system server. |
| 81 | */ |
| 82 | public static final int SYSTEM_UID = android.os.Process.SYSTEM_UID; |
| 83 | |
| 84 | /** |
| 85 | * PID of the system server. |
| 86 | */ |
| 87 | public static final int SYSTEM_PID = 11111; |
| 88 | |
Nicolas Prevot | 45d2907 | 2017-01-18 16:11:19 +0000 | [diff] [blame] | 89 | public static final String ANOTHER_PACKAGE_NAME = "com.another.package.name"; |
Nicolas Prevot | 45d2907 | 2017-01-18 16:11:19 +0000 | [diff] [blame] | 90 | public static final int ANOTHER_UID = UserHandle.getUid(UserHandle.USER_SYSTEM, 18434); |
| 91 | |
Eran Messeri | bb27189 | 2018-10-17 18:27:50 +0100 | [diff] [blame] | 92 | public static final String DELEGATE_PACKAGE_NAME = "com.delegate.package.name"; |
| 93 | public static final int DELEGATE_CERT_INSTALLER_UID = UserHandle.getUid(UserHandle.USER_SYSTEM, |
| 94 | 18437); |
| 95 | |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 96 | private final MockSystemServices mMockSystemServices; |
| 97 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 98 | public static class MockBinder { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 99 | public int callingUid = CALLER_UID; |
| 100 | public int callingPid = CALLER_PID; |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 101 | public final Map<Integer, List<String>> callingPermissions = new ArrayMap<>(); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 102 | |
| 103 | public long clearCallingIdentity() { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 104 | final long token = (((long) callingUid) << 32) | (callingPid); |
| 105 | callingUid = SYSTEM_UID; |
| 106 | callingPid = SYSTEM_PID; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 107 | return token; |
| 108 | } |
| 109 | |
| 110 | public void restoreCallingIdentity(long token) { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 111 | callingUid = (int) (token >> 32); |
| 112 | callingPid = (int) token; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 113 | } |
| 114 | |
yuemingw | e43cdf7 | 2017-10-12 16:52:11 +0100 | [diff] [blame] | 115 | public void withCleanCallingIdentity(@NonNull FunctionalUtils.ThrowingRunnable action) { |
| 116 | long callingIdentity = clearCallingIdentity(); |
| 117 | Throwable throwableToPropagate = null; |
| 118 | try { |
Eugene Susla | 2f5ee71 | 2017-06-23 17:25:24 -0700 | [diff] [blame] | 119 | action.runOrThrow(); |
yuemingw | e43cdf7 | 2017-10-12 16:52:11 +0100 | [diff] [blame] | 120 | } catch (Throwable throwable) { |
| 121 | throwableToPropagate = throwable; |
| 122 | } finally { |
| 123 | restoreCallingIdentity(callingIdentity); |
| 124 | if (throwableToPropagate != null) { |
| 125 | throw ExceptionUtils.propagate(throwableToPropagate); |
| 126 | } |
| 127 | } |
| 128 | } |
| 129 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 130 | public int getCallingUid() { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 131 | return callingUid; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 132 | } |
| 133 | |
| 134 | public int getCallingPid() { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 135 | return callingPid; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | public UserHandle getCallingUserHandle() { |
| 139 | return new UserHandle(UserHandle.getUserId(getCallingUid())); |
| 140 | } |
| 141 | |
| 142 | public boolean isCallerUidMyUid() { |
Makoto Onuki | f76b06a | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 143 | return callingUid == SYSTEM_UID; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 144 | } |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 145 | } |
| 146 | |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 147 | private final Context realTestContext; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 148 | |
| 149 | /** |
| 150 | * Use this instance to verify unimplemented methods such as {@link #sendBroadcast}. |
| 151 | * (Spying on {@code this} instance will confuse mockito somehow and I got weired "wrong number |
| 152 | * of arguments" exceptions.) |
| 153 | */ |
| 154 | public final Context spiedContext; |
| 155 | |
| 156 | public final MockBinder binder; |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 157 | public final Resources resources; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 158 | |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 159 | /** TODO: Migrate everything to use {@link #permissions} to avoid confusion. */ |
| 160 | @Deprecated |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 161 | public final List<String> callerPermissions = new ArrayList<>(); |
| 162 | |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 163 | /** Less confusing alias for {@link #callerPermissions}. */ |
| 164 | public final List<String> permissions = callerPermissions; |
| 165 | |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 166 | public String packageName = null; |
| 167 | |
| 168 | public ApplicationInfo applicationInfo = null; |
| 169 | |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 170 | public DpmMockContext(MockSystemServices mockSystemServices, Context context) { |
| 171 | mMockSystemServices = mockSystemServices; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 172 | realTestContext = context; |
Makoto Onuki | d932f76 | 2015-09-29 16:53:38 -0700 | [diff] [blame] | 173 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 174 | binder = new MockBinder(); |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 175 | resources = mock(Resources.class); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 176 | spiedContext = mock(Context.class); |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 177 | } |
| 178 | |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 179 | @Override |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 180 | public Resources getResources() { |
| 181 | return resources; |
| 182 | } |
| 183 | |
| 184 | @Override |
| 185 | public Resources.Theme getTheme() { |
| 186 | return spiedContext.getTheme(); |
| 187 | } |
| 188 | |
| 189 | @Override |
| 190 | public String getPackageName() { |
| 191 | if (packageName != null) { |
| 192 | return packageName; |
| 193 | } |
| 194 | return super.getPackageName(); |
| 195 | } |
| 196 | |
| 197 | @Override |
| 198 | public ApplicationInfo getApplicationInfo() { |
| 199 | if (applicationInfo != null) { |
| 200 | return applicationInfo; |
| 201 | } |
| 202 | return super.getApplicationInfo(); |
| 203 | } |
| 204 | |
| 205 | @Override |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 206 | public Object getSystemService(String name) { |
| 207 | switch (name) { |
Robin Lee | d2a73ed | 2016-12-19 09:07:16 +0000 | [diff] [blame] | 208 | case Context.ALARM_SERVICE: |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 209 | return mMockSystemServices.alarmManager; |
Neil Fuller | 3aedd49 | 2019-11-23 11:33:57 +0000 | [diff] [blame] | 210 | case Context.TIME_DETECTOR_SERVICE: |
| 211 | return mMockSystemServices.timeDetector; |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 212 | case Context.USER_SERVICE: |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 213 | return mMockSystemServices.userManager; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 214 | case Context.POWER_SERVICE: |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 215 | return mMockSystemServices.powerManager; |
Makoto Onuki | a31ebbc | 2015-11-23 17:15:21 -0800 | [diff] [blame] | 216 | case Context.WIFI_SERVICE: |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 217 | return mMockSystemServices.wifiManager; |
Makoto Onuki | 606da77 | 2016-12-15 14:26:55 -0800 | [diff] [blame] | 218 | case Context.ACCOUNT_SERVICE: |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 219 | return mMockSystemServices.accountManager; |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 220 | } |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 221 | throw new UnsupportedOperationException(); |
| 222 | } |
| 223 | |
| 224 | @Override |
Makoto Onuki | 72a3a0c | 2015-09-25 16:01:43 -0700 | [diff] [blame] | 225 | public String getSystemServiceName(Class<?> serviceClass) { |
| 226 | return realTestContext.getSystemServiceName(serviceClass); |
| 227 | } |
| 228 | |
| 229 | @Override |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 230 | public PackageManager getPackageManager() { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 231 | return mMockSystemServices.packageManager; |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 232 | } |
| 233 | |
Christine Franks | 361b825 | 2017-06-23 18:12:46 -0700 | [diff] [blame] | 234 | public UserManagerInternal getUserManagerInternal() { |
| 235 | return mMockSystemServices.userManagerInternal; |
| 236 | } |
| 237 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 238 | @Override |
| 239 | public void enforceCallingOrSelfPermission(String permission, String message) { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 240 | if (UserHandle.isSameApp(binder.getCallingUid(), SYSTEM_UID)) { |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 241 | return; // Assume system has all permissions. |
| 242 | } |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 243 | List<String> permissions = binder.callingPermissions.get(binder.getCallingUid()); |
| 244 | if (permissions == null) { |
| 245 | // TODO: delete the following line. to do this without breaking any tests, first it's |
| 246 | // necessary to remove all tests that set it directly. |
| 247 | permissions = callerPermissions; |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 248 | // throw new UnsupportedOperationException( |
| 249 | // "Caller UID " + binder.getCallingUid() + " doesn't exist"); |
Robin Lee | 2c68dad | 2017-03-17 12:50:24 +0000 | [diff] [blame] | 250 | } |
| 251 | if (!permissions.contains(permission)) { |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 252 | throw new SecurityException("Caller doesn't have " + permission + " : " + message); |
| 253 | } |
| 254 | } |
| 255 | |
| 256 | @Override |
| 257 | public void sendBroadcast(Intent intent) { |
| 258 | spiedContext.sendBroadcast(intent); |
| 259 | } |
| 260 | |
| 261 | @Override |
| 262 | public void sendBroadcast(Intent intent, String receiverPermission) { |
| 263 | spiedContext.sendBroadcast(intent, receiverPermission); |
| 264 | } |
| 265 | |
| 266 | @Override |
| 267 | public void sendBroadcastMultiplePermissions(Intent intent, String[] receiverPermissions) { |
| 268 | spiedContext.sendBroadcastMultiplePermissions(intent, receiverPermissions); |
| 269 | } |
| 270 | |
| 271 | @Override |
Tyler Gunn | f955e56 | 2018-04-26 14:43:31 -0700 | [diff] [blame] | 272 | public void sendBroadcastAsUserMultiplePermissions(Intent intent, UserHandle user, |
| 273 | String[] receiverPermissions) { |
| 274 | spiedContext.sendBroadcastAsUserMultiplePermissions(intent, user, receiverPermissions); |
| 275 | } |
| 276 | |
| 277 | @Override |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 278 | public void sendBroadcast(Intent intent, String receiverPermission, Bundle options) { |
| 279 | spiedContext.sendBroadcast(intent, receiverPermission, options); |
| 280 | } |
| 281 | |
| 282 | @Override |
| 283 | public void sendBroadcast(Intent intent, String receiverPermission, int appOp) { |
| 284 | spiedContext.sendBroadcast(intent, receiverPermission, appOp); |
| 285 | } |
| 286 | |
| 287 | @Override |
| 288 | public void sendOrderedBroadcast(Intent intent, String receiverPermission) { |
| 289 | spiedContext.sendOrderedBroadcast(intent, receiverPermission); |
| 290 | } |
| 291 | |
| 292 | @Override |
| 293 | public void sendOrderedBroadcast(Intent intent, String receiverPermission, |
| 294 | BroadcastReceiver resultReceiver, Handler scheduler, int initialCode, |
| 295 | String initialData, Bundle initialExtras) { |
| 296 | spiedContext.sendOrderedBroadcast(intent, receiverPermission, resultReceiver, scheduler, |
| 297 | initialCode, initialData, initialExtras); |
| 298 | } |
| 299 | |
| 300 | @Override |
| 301 | public void sendOrderedBroadcast(Intent intent, String receiverPermission, Bundle options, |
| 302 | BroadcastReceiver resultReceiver, Handler scheduler, int initialCode, |
| 303 | String initialData, Bundle initialExtras) { |
| 304 | spiedContext.sendOrderedBroadcast(intent, receiverPermission, options, resultReceiver, |
| 305 | scheduler, |
| 306 | initialCode, initialData, initialExtras); |
| 307 | } |
| 308 | |
| 309 | @Override |
| 310 | public void sendOrderedBroadcast(Intent intent, String receiverPermission, int appOp, |
| 311 | BroadcastReceiver resultReceiver, Handler scheduler, int initialCode, |
| 312 | String initialData, Bundle initialExtras) { |
| 313 | spiedContext.sendOrderedBroadcast(intent, receiverPermission, appOp, resultReceiver, |
| 314 | scheduler, |
| 315 | initialCode, initialData, initialExtras); |
| 316 | } |
| 317 | |
| 318 | @Override |
| 319 | public void sendBroadcastAsUser(Intent intent, UserHandle user) { |
Makoto Onuki | b643fb0 | 2015-09-22 15:03:44 -0700 | [diff] [blame] | 320 | if (binder.callingPid != SYSTEM_PID) { |
| 321 | // Unless called as the system process, can only call if the target user is the |
| 322 | // calling user. |
| 323 | // (The actual check is more complex; we may need to change it later.) |
| 324 | Assert.assertEquals(UserHandle.getUserId(binder.getCallingUid()), user.getIdentifier()); |
| 325 | } |
| 326 | |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 327 | spiedContext.sendBroadcastAsUser(intent, user); |
| 328 | } |
| 329 | |
| 330 | @Override |
Jonathan Scott | 367ebf4 | 2019-05-16 15:13:17 +0100 | [diff] [blame] | 331 | public void sendBroadcastAsUser(Intent intent, |
| 332 | UserHandle user, @Nullable String receiverPermission, @Nullable Bundle options) { |
| 333 | spiedContext.sendBroadcastAsUser(intent, user, receiverPermission, options); |
| 334 | } |
| 335 | |
| 336 | @Override |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 337 | public void sendBroadcastAsUser(Intent intent, UserHandle user, String receiverPermission) { |
| 338 | spiedContext.sendBroadcastAsUser(intent, user, receiverPermission); |
| 339 | } |
| 340 | |
| 341 | @Override |
| 342 | public void sendBroadcastAsUser(Intent intent, UserHandle user, String receiverPermission, |
| 343 | int appOp) { |
| 344 | spiedContext.sendBroadcastAsUser(intent, user, receiverPermission, appOp); |
| 345 | } |
| 346 | |
| 347 | @Override |
| 348 | public void sendOrderedBroadcastAsUser(Intent intent, UserHandle user, |
| 349 | String receiverPermission, BroadcastReceiver resultReceiver, Handler scheduler, |
| 350 | int initialCode, String initialData, Bundle initialExtras) { |
Jonathan Scott | 367ebf4 | 2019-05-16 15:13:17 +0100 | [diff] [blame] | 351 | sendOrderedBroadcastAsUser( |
| 352 | intent, user, receiverPermission, AppOpsManager.OP_NONE, resultReceiver, |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 353 | scheduler, initialCode, initialData, initialExtras); |
| 354 | } |
| 355 | |
| 356 | @Override |
| 357 | public void sendOrderedBroadcastAsUser(Intent intent, UserHandle user, |
| 358 | String receiverPermission, int appOp, BroadcastReceiver resultReceiver, |
| 359 | Handler scheduler, int initialCode, String initialData, Bundle initialExtras) { |
Jonathan Scott | 367ebf4 | 2019-05-16 15:13:17 +0100 | [diff] [blame] | 360 | sendOrderedBroadcastAsUser( |
| 361 | intent, user, receiverPermission, appOp, null, resultReceiver, |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 362 | scheduler, initialCode, initialData, initialExtras); |
| 363 | } |
| 364 | |
| 365 | @Override |
| 366 | public void sendOrderedBroadcastAsUser(Intent intent, UserHandle user, |
| 367 | String receiverPermission, int appOp, Bundle options, BroadcastReceiver resultReceiver, |
| 368 | Handler scheduler, int initialCode, String initialData, Bundle initialExtras) { |
| 369 | spiedContext.sendOrderedBroadcastAsUser(intent, user, receiverPermission, appOp, options, |
| 370 | resultReceiver, scheduler, initialCode, initialData, initialExtras); |
Jonathan Scott | 367ebf4 | 2019-05-16 15:13:17 +0100 | [diff] [blame] | 371 | resultReceiver.onReceive(spiedContext, intent); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 372 | } |
| 373 | |
| 374 | @Override |
| 375 | public void sendStickyBroadcast(Intent intent) { |
| 376 | spiedContext.sendStickyBroadcast(intent); |
| 377 | } |
| 378 | |
| 379 | @Override |
| 380 | public void sendStickyOrderedBroadcast(Intent intent, BroadcastReceiver resultReceiver, |
| 381 | Handler scheduler, int initialCode, String initialData, Bundle initialExtras) { |
| 382 | spiedContext.sendStickyOrderedBroadcast(intent, resultReceiver, scheduler, initialCode, |
| 383 | initialData, initialExtras); |
| 384 | } |
| 385 | |
| 386 | @Override |
| 387 | public void removeStickyBroadcast(Intent intent) { |
| 388 | spiedContext.removeStickyBroadcast(intent); |
| 389 | } |
| 390 | |
| 391 | @Override |
| 392 | public void sendStickyBroadcastAsUser(Intent intent, UserHandle user) { |
| 393 | spiedContext.sendStickyBroadcastAsUser(intent, user); |
| 394 | } |
| 395 | |
| 396 | @Override |
| 397 | public void sendStickyOrderedBroadcastAsUser(Intent intent, UserHandle user, |
| 398 | BroadcastReceiver resultReceiver, Handler scheduler, int initialCode, |
| 399 | String initialData, Bundle initialExtras) { |
| 400 | spiedContext.sendStickyOrderedBroadcastAsUser(intent, user, resultReceiver, scheduler, initialCode, |
| 401 | initialData, initialExtras); |
| 402 | } |
| 403 | |
| 404 | @Override |
| 405 | public void removeStickyBroadcastAsUser(Intent intent, UserHandle user) { |
| 406 | spiedContext.removeStickyBroadcastAsUser(intent, user); |
| 407 | } |
| 408 | |
| 409 | @Override |
| 410 | public Intent registerReceiver(BroadcastReceiver receiver, IntentFilter filter) { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 411 | mMockSystemServices.registerReceiver(receiver, filter, null); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 412 | return spiedContext.registerReceiver(receiver, filter); |
| 413 | } |
| 414 | |
| 415 | @Override |
| 416 | public Intent registerReceiver(BroadcastReceiver receiver, IntentFilter filter, |
| 417 | String broadcastPermission, Handler scheduler) { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 418 | mMockSystemServices.registerReceiver(receiver, filter, scheduler); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 419 | return spiedContext.registerReceiver(receiver, filter, broadcastPermission, scheduler); |
| 420 | } |
| 421 | |
| 422 | @Override |
| 423 | public Intent registerReceiverAsUser(BroadcastReceiver receiver, UserHandle user, |
| 424 | IntentFilter filter, String broadcastPermission, Handler scheduler) { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 425 | mMockSystemServices.registerReceiver(receiver, filter, scheduler); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 426 | return spiedContext.registerReceiverAsUser(receiver, user, filter, broadcastPermission, |
| 427 | scheduler); |
| 428 | } |
| 429 | |
| 430 | @Override |
| 431 | public void unregisterReceiver(BroadcastReceiver receiver) { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 432 | mMockSystemServices.unregisterReceiver(receiver); |
Makoto Onuki | cc4bbeb | 2015-09-17 10:28:24 -0700 | [diff] [blame] | 433 | spiedContext.unregisterReceiver(receiver); |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 434 | } |
Makoto Onuki | a52562c | 2015-10-01 16:12:31 -0700 | [diff] [blame] | 435 | |
| 436 | @Override |
Robin Lee | 7f5c91c | 2017-02-08 21:27:02 +0000 | [diff] [blame] | 437 | public Context createPackageContextAsUser(String packageName, int flags, UserHandle user) |
| 438 | throws PackageManager.NameNotFoundException { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 439 | return mMockSystemServices.createPackageContextAsUser(packageName, flags, user); |
Robin Lee | 7f5c91c | 2017-02-08 21:27:02 +0000 | [diff] [blame] | 440 | } |
| 441 | |
| 442 | @Override |
Makoto Onuki | a52562c | 2015-10-01 16:12:31 -0700 | [diff] [blame] | 443 | public ContentResolver getContentResolver() { |
Pavel Grafov | 75c0a89 | 2017-05-18 17:28:27 +0100 | [diff] [blame] | 444 | return mMockSystemServices.contentResolver; |
Makoto Onuki | a52562c | 2015-10-01 16:12:31 -0700 | [diff] [blame] | 445 | } |
Bartosz Fabianowski | b21b241 | 2016-11-17 04:53:33 +0100 | [diff] [blame] | 446 | |
| 447 | @Override |
| 448 | public int getUserId() { |
| 449 | return UserHandle.getUserId(binder.getCallingUid()); |
| 450 | } |
Eran Messeri | bb27189 | 2018-10-17 18:27:50 +0100 | [diff] [blame] | 451 | |
| 452 | @Override |
| 453 | public int checkCallingPermission(String permission) { |
| 454 | return spiedContext.checkCallingPermission(permission); |
| 455 | } |
Makoto Onuki | c9754cf | 2015-08-31 11:58:43 -0700 | [diff] [blame] | 456 | } |