xref: /aosp_15_r20/art/test/2038-hiddenapi-jvmti-ext/hiddenapi_ext.cc (revision 795d594fd825385562da6b089ea9b2033f3abf5a)
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