1*795d594fSAndroid Build Coastguard Worker /*
2*795d594fSAndroid Build Coastguard Worker * Copyright (C) 2020 The Android Open Source Project
3*795d594fSAndroid Build Coastguard Worker *
4*795d594fSAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*795d594fSAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*795d594fSAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*795d594fSAndroid Build Coastguard Worker *
8*795d594fSAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*795d594fSAndroid Build Coastguard Worker *
10*795d594fSAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*795d594fSAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*795d594fSAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*795d594fSAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*795d594fSAndroid Build Coastguard Worker * limitations under the License.
15*795d594fSAndroid Build Coastguard Worker */
16*795d594fSAndroid Build Coastguard Worker
17*795d594fSAndroid Build Coastguard Worker #include <queue>
18*795d594fSAndroid Build Coastguard Worker #include <vector>
19*795d594fSAndroid Build Coastguard Worker
20*795d594fSAndroid Build Coastguard Worker #include "jvmti.h"
21*795d594fSAndroid Build Coastguard Worker
22*795d594fSAndroid Build Coastguard Worker // Test infrastructure
23*795d594fSAndroid Build Coastguard Worker #include "jvmti_helper.h"
24*795d594fSAndroid Build Coastguard Worker #include "nativehelper/scoped_local_ref.h"
25*795d594fSAndroid Build Coastguard Worker #include "nativehelper/scoped_primitive_array.h"
26*795d594fSAndroid Build Coastguard Worker #include "test_env.h"
27*795d594fSAndroid Build Coastguard Worker
28*795d594fSAndroid Build Coastguard Worker namespace art {
29*795d594fSAndroid Build Coastguard Worker namespace Test2038HiddenApiExt {
30*795d594fSAndroid Build Coastguard Worker
31*795d594fSAndroid Build Coastguard Worker template <typename T>
Dealloc(T * t)32*795d594fSAndroid Build Coastguard Worker static void Dealloc(T* t) {
33*795d594fSAndroid Build Coastguard Worker jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(t));
34*795d594fSAndroid Build Coastguard Worker }
35*795d594fSAndroid Build Coastguard Worker
36*795d594fSAndroid Build Coastguard Worker template <typename T, typename... Rest>
Dealloc(T * t,Rest...rs)37*795d594fSAndroid Build Coastguard Worker static void Dealloc(T* t, Rest... rs) {
38*795d594fSAndroid Build Coastguard Worker Dealloc(t);
39*795d594fSAndroid Build Coastguard Worker Dealloc(rs...);
40*795d594fSAndroid Build Coastguard Worker }
41*795d594fSAndroid Build Coastguard Worker
DeallocParams(jvmtiParamInfo * params,jint n_params)42*795d594fSAndroid Build Coastguard Worker static void DeallocParams(jvmtiParamInfo* params, jint n_params) {
43*795d594fSAndroid Build Coastguard Worker for (jint i = 0; i < n_params; i++) {
44*795d594fSAndroid Build Coastguard Worker Dealloc(params[i].name);
45*795d594fSAndroid Build Coastguard Worker }
46*795d594fSAndroid Build Coastguard Worker }
47*795d594fSAndroid Build Coastguard Worker
48*795d594fSAndroid Build Coastguard Worker static constexpr std::string_view kDisablePolicyName =
49*795d594fSAndroid Build Coastguard Worker "com.android.art.misc.disable_hidden_api_enforcement_policy";
50*795d594fSAndroid Build Coastguard Worker static constexpr std::string_view kGetPolicyName =
51*795d594fSAndroid Build Coastguard Worker "com.android.art.misc.get_hidden_api_enforcement_policy";
52*795d594fSAndroid Build Coastguard Worker static constexpr std::string_view kSetPolicyName =
53*795d594fSAndroid Build Coastguard Worker "com.android.art.misc.set_hidden_api_enforcement_policy";
54*795d594fSAndroid Build Coastguard Worker using GetPolicy = jvmtiError (*)(jvmtiEnv*, jint*);
55*795d594fSAndroid Build Coastguard Worker using SetPolicy = jvmtiError (*)(jvmtiEnv*, jint);
56*795d594fSAndroid Build Coastguard Worker using DisablePolicy = jvmtiError (*)(jvmtiEnv*);
57*795d594fSAndroid Build Coastguard Worker
GetExtension(JNIEnv * env,const std::string_view & name)58*795d594fSAndroid Build Coastguard Worker void* GetExtension(JNIEnv* env, const std::string_view& name) {
59*795d594fSAndroid Build Coastguard Worker // Get the extensions.
60*795d594fSAndroid Build Coastguard Worker jint n_ext = 0;
61*795d594fSAndroid Build Coastguard Worker void* result = nullptr;
62*795d594fSAndroid Build Coastguard Worker jvmtiExtensionFunctionInfo* infos = nullptr;
63*795d594fSAndroid Build Coastguard Worker if (JvmtiErrorToException(env, jvmti_env, jvmti_env->GetExtensionFunctions(&n_ext, &infos))) {
64*795d594fSAndroid Build Coastguard Worker return nullptr;
65*795d594fSAndroid Build Coastguard Worker }
66*795d594fSAndroid Build Coastguard Worker for (jint i = 0; i < n_ext; i++) {
67*795d594fSAndroid Build Coastguard Worker jvmtiExtensionFunctionInfo* cur_info = &infos[i];
68*795d594fSAndroid Build Coastguard Worker if (name == std::string_view(cur_info->id)) {
69*795d594fSAndroid Build Coastguard Worker result = reinterpret_cast<void*>(cur_info->func);
70*795d594fSAndroid Build Coastguard Worker }
71*795d594fSAndroid Build Coastguard Worker // Cleanup the cur_info
72*795d594fSAndroid Build Coastguard Worker DeallocParams(cur_info->params, cur_info->param_count);
73*795d594fSAndroid Build Coastguard Worker Dealloc(cur_info->id, cur_info->short_description, cur_info->params, cur_info->errors);
74*795d594fSAndroid Build Coastguard Worker }
75*795d594fSAndroid Build Coastguard Worker // Cleanup the array.
76*795d594fSAndroid Build Coastguard Worker Dealloc(infos);
77*795d594fSAndroid Build Coastguard Worker if (result == nullptr) {
78*795d594fSAndroid Build Coastguard Worker ScopedLocalRef<jclass> rt_exception(env, env->FindClass("java/lang/RuntimeException"));
79*795d594fSAndroid Build Coastguard Worker env->ThrowNew(rt_exception.get(), "Unable to find policy extensions.");
80*795d594fSAndroid Build Coastguard Worker return nullptr;
81*795d594fSAndroid Build Coastguard Worker }
82*795d594fSAndroid Build Coastguard Worker return result;
83*795d594fSAndroid Build Coastguard Worker }
84*795d594fSAndroid Build Coastguard Worker
Java_Main_disablePolicy(JNIEnv * env,jclass)85*795d594fSAndroid Build Coastguard Worker extern "C" JNIEXPORT jint JNICALL Java_Main_disablePolicy(JNIEnv* env, jclass) {
86*795d594fSAndroid Build Coastguard Worker jint res;
87*795d594fSAndroid Build Coastguard Worker GetPolicy get_policy = reinterpret_cast<GetPolicy>(GetExtension(env, kGetPolicyName));
88*795d594fSAndroid Build Coastguard Worker if (get_policy == nullptr) {
89*795d594fSAndroid Build Coastguard Worker return -1;
90*795d594fSAndroid Build Coastguard Worker }
91*795d594fSAndroid Build Coastguard Worker DisablePolicy disable_policy =
92*795d594fSAndroid Build Coastguard Worker reinterpret_cast<DisablePolicy>(GetExtension(env, kDisablePolicyName));
93*795d594fSAndroid Build Coastguard Worker if (disable_policy == nullptr) {
94*795d594fSAndroid Build Coastguard Worker return -1;
95*795d594fSAndroid Build Coastguard Worker }
96*795d594fSAndroid Build Coastguard Worker if (JvmtiErrorToException(env, jvmti_env, get_policy(jvmti_env, &res))) {
97*795d594fSAndroid Build Coastguard Worker return -1;
98*795d594fSAndroid Build Coastguard Worker }
99*795d594fSAndroid Build Coastguard Worker JvmtiErrorToException(env, jvmti_env, disable_policy(jvmti_env));
100*795d594fSAndroid Build Coastguard Worker return res;
101*795d594fSAndroid Build Coastguard Worker }
102*795d594fSAndroid Build Coastguard Worker
Java_Main_setPolicy(JNIEnv * env,jclass,jint pol)103*795d594fSAndroid Build Coastguard Worker extern "C" JNIEXPORT jint JNICALL Java_Main_setPolicy(JNIEnv* env, jclass, jint pol) {
104*795d594fSAndroid Build Coastguard Worker jint res;
105*795d594fSAndroid Build Coastguard Worker GetPolicy get_policy = reinterpret_cast<GetPolicy>(GetExtension(env, kGetPolicyName));
106*795d594fSAndroid Build Coastguard Worker if (get_policy == nullptr) {
107*795d594fSAndroid Build Coastguard Worker return -1;
108*795d594fSAndroid Build Coastguard Worker }
109*795d594fSAndroid Build Coastguard Worker SetPolicy set_policy = reinterpret_cast<SetPolicy>(GetExtension(env, kSetPolicyName));
110*795d594fSAndroid Build Coastguard Worker if (set_policy == nullptr) {
111*795d594fSAndroid Build Coastguard Worker return -1;
112*795d594fSAndroid Build Coastguard Worker }
113*795d594fSAndroid Build Coastguard Worker if (JvmtiErrorToException(env, jvmti_env, get_policy(jvmti_env, &res))) {
114*795d594fSAndroid Build Coastguard Worker return -1;
115*795d594fSAndroid Build Coastguard Worker }
116*795d594fSAndroid Build Coastguard Worker JvmtiErrorToException(env, jvmti_env, set_policy(jvmti_env, pol));
117*795d594fSAndroid Build Coastguard Worker return res;
118*795d594fSAndroid Build Coastguard Worker }
119*795d594fSAndroid Build Coastguard Worker
120*795d594fSAndroid Build Coastguard Worker } // namespace Test2038HiddenApiExt
121*795d594fSAndroid Build Coastguard Worker } // namespace art
122