Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2018 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.role; |
| 18 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 19 | import android.annotation.NonNull; |
| 20 | import android.annotation.Nullable; |
| 21 | import android.annotation.UserIdInt; |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 22 | import android.annotation.WorkerThread; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 23 | import android.app.role.IRoleManagerCallback; |
| 24 | import android.app.role.RoleManagerCallback; |
| 25 | import android.content.ComponentName; |
| 26 | import android.content.Context; |
| 27 | import android.content.Intent; |
| 28 | import android.content.ServiceConnection; |
| 29 | import android.os.Handler; |
| 30 | import android.os.IBinder; |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 31 | import android.os.RemoteException; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 32 | import android.os.UserHandle; |
| 33 | import android.rolecontrollerservice.IRoleControllerService; |
| 34 | import android.rolecontrollerservice.RoleControllerService; |
| 35 | import android.util.Slog; |
| 36 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 37 | import com.android.internal.util.function.pooled.PooledLambda; |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 38 | import com.android.server.FgThread; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 39 | |
| 40 | import java.util.ArrayDeque; |
| 41 | import java.util.Queue; |
| 42 | |
| 43 | /** |
| 44 | * Handles connection with {@link RoleControllerService}. |
| 45 | */ |
| 46 | public class RemoteRoleControllerService { |
| 47 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 48 | static final boolean DEBUG = false; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 49 | private static final String LOG_TAG = RemoteRoleControllerService.class.getSimpleName(); |
| 50 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 51 | @NonNull |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 52 | private final Connection mConnection; |
| 53 | |
| 54 | public RemoteRoleControllerService(@UserIdInt int userId, @NonNull Context context) { |
| 55 | mConnection = new Connection(userId, context); |
| 56 | } |
| 57 | |
| 58 | /** |
| 59 | * Add a specific application to the holders of a role. If the role is exclusive, the previous |
| 60 | * holder will be replaced. |
| 61 | * |
| 62 | * @see RoleControllerService#onAddRoleHolder(String, String, RoleManagerCallback) |
| 63 | */ |
| 64 | public void onAddRoleHolder(@NonNull String roleName, @NonNull String packageName, |
| 65 | @NonNull IRoleManagerCallback callback) { |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 66 | mConnection.enqueueCall(new Connection.Call((service, callbackDelegate) -> |
| 67 | service.onAddRoleHolder(roleName, packageName, callbackDelegate), callback)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 68 | } |
| 69 | |
| 70 | /** |
| 71 | * Remove a specific application from the holders of a role. |
| 72 | * |
| 73 | * @see RoleControllerService#onRemoveRoleHolder(String, String, RoleManagerCallback) |
| 74 | */ |
| 75 | public void onRemoveRoleHolder(@NonNull String roleName, @NonNull String packageName, |
| 76 | @NonNull IRoleManagerCallback callback) { |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 77 | mConnection.enqueueCall(new Connection.Call((service, callbackDelegate) -> |
| 78 | service.onRemoveRoleHolder(roleName, packageName, callbackDelegate), callback)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 79 | } |
| 80 | |
| 81 | /** |
| 82 | * Remove all holders of a role. |
| 83 | * |
| 84 | * @see RoleControllerService#onClearRoleHolders(String, RoleManagerCallback) |
| 85 | */ |
| 86 | public void onClearRoleHolders(@NonNull String roleName, |
| 87 | @NonNull IRoleManagerCallback callback) { |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 88 | mConnection.enqueueCall(new Connection.Call((service, callbackDelegate) -> |
| 89 | service.onClearRoleHolders(roleName, callbackDelegate), callback)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 90 | } |
| 91 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 92 | /** |
| 93 | * Performs granting of default roles and permissions and appops |
| 94 | * |
| 95 | * @see RoleControllerService#onGrantDefaultRoles(RoleManagerCallback) |
| 96 | */ |
| 97 | public void onGrantDefaultRoles(@NonNull IRoleManagerCallback callback) { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 98 | mConnection.enqueueCall(new Connection.Call(IRoleControllerService::onGrantDefaultRoles, |
| 99 | callback)); |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 100 | } |
| 101 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 102 | private static final class Connection implements ServiceConnection { |
| 103 | |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 104 | private static final long UNBIND_DELAY_MILLIS = 15 * 1000; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 105 | |
| 106 | @UserIdInt |
| 107 | private final int mUserId; |
| 108 | |
| 109 | @NonNull |
| 110 | private final Context mContext; |
| 111 | |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 112 | @NonNull |
| 113 | private final Handler mHandler = FgThread.getHandler(); |
| 114 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 115 | private boolean mBound; |
| 116 | |
| 117 | @Nullable |
| 118 | private IRoleControllerService mService; |
| 119 | |
| 120 | @NonNull |
| 121 | private final Queue<Call> mPendingCalls = new ArrayDeque<>(); |
| 122 | |
| 123 | @NonNull |
| 124 | private final Runnable mUnbindRunnable = this::unbind; |
| 125 | |
| 126 | Connection(@UserIdInt int userId, @NonNull Context context) { |
| 127 | mUserId = userId; |
| 128 | mContext = context; |
| 129 | } |
| 130 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 131 | @Override |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 132 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 133 | public void onServiceConnected(@NonNull ComponentName name, @NonNull IBinder service) { |
| 134 | mService = IRoleControllerService.Stub.asInterface(service); |
| 135 | executePendingCalls(); |
| 136 | } |
| 137 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 138 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 139 | private void executePendingCalls() { |
| 140 | while (!mPendingCalls.isEmpty()) { |
| 141 | Call call = mPendingCalls.poll(); |
| 142 | call.execute(mService); |
| 143 | } |
| 144 | scheduleUnbind(); |
| 145 | } |
| 146 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 147 | @Override |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 148 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 149 | public void onServiceDisconnected(@NonNull ComponentName name) { |
| 150 | mService = null; |
| 151 | } |
| 152 | |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 153 | @Override |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 154 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 155 | public void onBindingDied(@NonNull ComponentName name) { |
| 156 | unbind(); |
| 157 | } |
| 158 | |
| 159 | public void enqueueCall(@NonNull Call call) { |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 160 | if (DEBUG) { |
| 161 | Slog.i(LOG_TAG, "Enqueue " + call); |
| 162 | } |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 163 | mHandler.executeOrSendMessage(PooledLambda.obtainMessage(Connection::executeCall, this, |
| 164 | call)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 165 | } |
| 166 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 167 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 168 | private void executeCall(@NonNull Call call) { |
| 169 | ensureBound(); |
| 170 | if (mService == null) { |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 171 | if (DEBUG) { |
| 172 | Slog.i(LOG_TAG, "Delaying until service connected: " + call); |
| 173 | } |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 174 | mPendingCalls.offer(call); |
| 175 | return; |
| 176 | } |
| 177 | call.execute(mService); |
| 178 | scheduleUnbind(); |
| 179 | } |
| 180 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 181 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 182 | private void ensureBound() { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 183 | mHandler.removeCallbacks(mUnbindRunnable); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 184 | if (!mBound) { |
| 185 | Intent intent = new Intent(RoleControllerService.SERVICE_INTERFACE); |
| 186 | intent.setPackage(mContext.getPackageManager() |
| 187 | .getPermissionControllerPackageName()); |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 188 | // Use direct handler to ensure onServiceConnected callback happens in the same |
| 189 | // call frame, as required by onGrantDefaultRoles |
| 190 | // |
| 191 | // Note that as a result, onServiceConnected may happen not on main thread! |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 192 | mBound = mContext.bindServiceAsUser(intent, this, Context.BIND_AUTO_CREATE, |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 193 | mHandler, UserHandle.of(mUserId)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 194 | } |
| 195 | } |
| 196 | |
| 197 | private void scheduleUnbind() { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 198 | mHandler.removeCallbacks(mUnbindRunnable); |
| 199 | mHandler.postDelayed(mUnbindRunnable, UNBIND_DELAY_MILLIS); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 200 | } |
| 201 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 202 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 203 | private void unbind() { |
| 204 | if (mBound) { |
| 205 | mService = null; |
| 206 | mContext.unbindService(this); |
| 207 | mBound = false; |
| 208 | } |
| 209 | } |
| 210 | |
| 211 | public static class Call { |
| 212 | |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 213 | private static final int TIMEOUT_MILLIS = 15 * 1000; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 214 | |
| 215 | @NonNull |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 216 | private final CallExecutor mCallExecutor; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 217 | |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 218 | @NonNull |
| 219 | private final IRoleManagerCallback mCallback; |
| 220 | |
| 221 | @NonNull |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 222 | private final Handler mHandler = FgThread.getHandler(); |
| 223 | |
| 224 | @NonNull |
Hai Zhang | b295ac4 | 2018-11-16 16:08:18 -0800 | [diff] [blame] | 225 | private final Runnable mTimeoutRunnable = this::notifyTimeout; |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 226 | |
| 227 | private boolean mCallbackNotified; |
| 228 | |
| 229 | private Call(@NonNull CallExecutor callExecutor, |
| 230 | @NonNull IRoleManagerCallback callback) { |
| 231 | mCallExecutor = callExecutor; |
| 232 | mCallback = callback; |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 233 | } |
| 234 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 235 | @WorkerThread |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 236 | public void execute(IRoleControllerService service) { |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 237 | if (DEBUG) { |
| 238 | Slog.i(LOG_TAG, "Executing " + this); |
| 239 | } |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 240 | try { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 241 | mHandler.postDelayed(mTimeoutRunnable, TIMEOUT_MILLIS); |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 242 | mCallExecutor.execute(service, new CallbackDelegate()); |
| 243 | } catch (RemoteException e) { |
| 244 | Slog.e(LOG_TAG, "Error calling RoleControllerService", e); |
| 245 | notifyCallback(false); |
| 246 | } |
| 247 | } |
| 248 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 249 | @WorkerThread |
Hai Zhang | b295ac4 | 2018-11-16 16:08:18 -0800 | [diff] [blame] | 250 | private void notifyTimeout() { |
| 251 | Slog.e(LOG_TAG, "Call timed out, calling onFailure()"); |
| 252 | notifyCallback(false); |
| 253 | } |
| 254 | |
| 255 | @WorkerThread |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 256 | private void notifyCallback(boolean success) { |
| 257 | if (mCallbackNotified) { |
| 258 | return; |
| 259 | } |
| 260 | mCallbackNotified = true; |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 261 | mHandler.removeCallbacks(mTimeoutRunnable); |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 262 | try { |
| 263 | if (success) { |
| 264 | mCallback.onSuccess(); |
| 265 | } else { |
| 266 | mCallback.onFailure(); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 267 | } |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 268 | } catch (RemoteException e) { |
| 269 | Slog.e(LOG_TAG, "Error calling " + (success ? "onSuccess()" : "onFailure()") |
| 270 | + " callback", e); |
| 271 | } |
| 272 | } |
| 273 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 274 | @Override |
| 275 | public String toString() { |
| 276 | return "Call with callback: " + mCallback; |
| 277 | } |
| 278 | |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 279 | @FunctionalInterface |
| 280 | public interface CallExecutor { |
| 281 | |
Eugene Susla | a4200f82 | 2018-11-09 18:06:43 -0800 | [diff] [blame] | 282 | @WorkerThread |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 283 | void execute(IRoleControllerService service, IRoleManagerCallback callbackDelegate) |
| 284 | throws RemoteException; |
| 285 | } |
| 286 | |
| 287 | private class CallbackDelegate extends IRoleManagerCallback.Stub { |
| 288 | |
| 289 | @Override |
| 290 | public void onSuccess() throws RemoteException { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 291 | mHandler.sendMessage(PooledLambda.obtainMessage(Call::notifyCallback, Call.this, |
| 292 | true)); |
Hai Zhang | 527dd51 | 2018-10-19 11:41:33 -0700 | [diff] [blame] | 293 | } |
| 294 | |
| 295 | @Override |
| 296 | public void onFailure() throws RemoteException { |
Hai Zhang | fcbc338 | 2018-12-06 15:34:05 -0800 | [diff] [blame] | 297 | mHandler.sendMessage(PooledLambda.obtainMessage(Call::notifyCallback, Call.this, |
| 298 | false)); |
Hai Zhang | b777668 | 2018-09-25 15:10:57 -0700 | [diff] [blame] | 299 | } |
| 300 | } |
| 301 | } |
| 302 | } |
| 303 | } |