1 /* 2 * Copyright (C) 2023 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 package com.android.adservices.shared.testing.common; 17 18 import android.app.UiAutomation; 19 20 import androidx.test.platform.app.InstrumentationRegistry; 21 22 import java.util.function.Supplier; 23 24 /** Provides helpers for Shell-related operations. */ 25 public final class ShellHelper { 26 27 // TODO(b/306522832): add unit tests 28 // NOTE: copied from ShellIdentityUtils to avoid importing compatibility-device-util-axt 29 /** 30 * Run an arbitrary piece of code while holding shell permissions. 31 * 32 * @param supplier an expression that performs the desired operation with shell permissions 33 * @param <T> the return type of the expression 34 * @return the return value of the expression 35 */ invokeWithShellPermissions(Supplier<T> supplier)36 public static <T> T invokeWithShellPermissions(Supplier<T> supplier) { 37 UiAutomation uiAutomation = InstrumentationRegistry.getInstrumentation().getUiAutomation(); 38 try { 39 uiAutomation.adoptShellPermissionIdentity(); 40 return supplier.get(); 41 } finally { 42 uiAutomation.dropShellPermissionIdentity(); 43 } 44 } 45 ShellHelper()46 private ShellHelper() { 47 throw new UnsupportedOperationException("Provides only static methods"); 48 } 49 } 50