xref: /aosp_15_r20/art/test/115-native-bridge/nativebridge.cc (revision 795d594fd825385562da6b089ea9b2033f3abf5a)
1*795d594fSAndroid Build Coastguard Worker /*
2*795d594fSAndroid Build Coastguard Worker  * Copyright (C) 2014 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 // A simple implementation of the native-bridge interface.
18*795d594fSAndroid Build Coastguard Worker 
19*795d594fSAndroid Build Coastguard Worker #include <dlfcn.h>
20*795d594fSAndroid Build Coastguard Worker #include <setjmp.h>
21*795d594fSAndroid Build Coastguard Worker #include <signal.h>
22*795d594fSAndroid Build Coastguard Worker #include <sys/stat.h>
23*795d594fSAndroid Build Coastguard Worker #include <unistd.h>
24*795d594fSAndroid Build Coastguard Worker 
25*795d594fSAndroid Build Coastguard Worker #include <algorithm>
26*795d594fSAndroid Build Coastguard Worker #include <cstdio>
27*795d594fSAndroid Build Coastguard Worker #include <cstdlib>
28*795d594fSAndroid Build Coastguard Worker #include <vector>
29*795d594fSAndroid Build Coastguard Worker 
30*795d594fSAndroid Build Coastguard Worker #include <jni.h>
31*795d594fSAndroid Build Coastguard Worker #include <nativebridge/native_bridge.h>
32*795d594fSAndroid Build Coastguard Worker 
33*795d594fSAndroid Build Coastguard Worker #include "base/casts.h"
34*795d594fSAndroid Build Coastguard Worker #include "base/macros.h"
35*795d594fSAndroid Build Coastguard Worker 
36*795d594fSAndroid Build Coastguard Worker struct NativeBridgeMethod {
37*795d594fSAndroid Build Coastguard Worker   const char* name;
38*795d594fSAndroid Build Coastguard Worker   const char* signature;
39*795d594fSAndroid Build Coastguard Worker   bool static_method;
40*795d594fSAndroid Build Coastguard Worker   void* fnPtr;
41*795d594fSAndroid Build Coastguard Worker   void* trampoline;
42*795d594fSAndroid Build Coastguard Worker };
43*795d594fSAndroid Build Coastguard Worker 
44*795d594fSAndroid Build Coastguard Worker static NativeBridgeMethod* find_native_bridge_method(const char *name);
45*795d594fSAndroid Build Coastguard Worker static const android::NativeBridgeRuntimeCallbacks* gNativeBridgeArtCallbacks;
46*795d594fSAndroid Build Coastguard Worker 
trampoline_JNI_OnLoad(JavaVM * vm,void * reserved)47*795d594fSAndroid Build Coastguard Worker static jint trampoline_JNI_OnLoad(JavaVM* vm, void* reserved) {
48*795d594fSAndroid Build Coastguard Worker   JNIEnv* env = nullptr;
49*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jint(*)(JavaVM*, void*);
50*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>(find_native_bridge_method("JNI_OnLoad")->fnPtr);
51*795d594fSAndroid Build Coastguard Worker 
52*795d594fSAndroid Build Coastguard Worker   vm->GetEnv(reinterpret_cast<void **>(&env), JNI_VERSION_1_6);
53*795d594fSAndroid Build Coastguard Worker   if (env == nullptr) {
54*795d594fSAndroid Build Coastguard Worker     return 0;
55*795d594fSAndroid Build Coastguard Worker   }
56*795d594fSAndroid Build Coastguard Worker 
57*795d594fSAndroid Build Coastguard Worker   jclass klass = env->FindClass("Main");
58*795d594fSAndroid Build Coastguard Worker   if (klass != nullptr) {
59*795d594fSAndroid Build Coastguard Worker     int i, count1, count2;
60*795d594fSAndroid Build Coastguard Worker     count1 = gNativeBridgeArtCallbacks->getNativeMethodCount(env, klass);
61*795d594fSAndroid Build Coastguard Worker     std::unique_ptr<JNINativeMethod[]> methods(new JNINativeMethod[count1]);
62*795d594fSAndroid Build Coastguard Worker     if (methods == nullptr) {
63*795d594fSAndroid Build Coastguard Worker       return 0;
64*795d594fSAndroid Build Coastguard Worker     }
65*795d594fSAndroid Build Coastguard Worker     count2 = gNativeBridgeArtCallbacks->getNativeMethods(env, klass, methods.get(), count1);
66*795d594fSAndroid Build Coastguard Worker     if (count1 == count2) {
67*795d594fSAndroid Build Coastguard Worker       printf("Test ART callbacks: all JNI function number is %d.\n", count1);
68*795d594fSAndroid Build Coastguard Worker     }
69*795d594fSAndroid Build Coastguard Worker 
70*795d594fSAndroid Build Coastguard Worker     for (i = 0; i < count1; i++) {
71*795d594fSAndroid Build Coastguard Worker       NativeBridgeMethod* nb_method = find_native_bridge_method(methods[i].name);
72*795d594fSAndroid Build Coastguard Worker       if (nb_method != nullptr) {
73*795d594fSAndroid Build Coastguard Worker         jmethodID mid = nullptr;
74*795d594fSAndroid Build Coastguard Worker         if (nb_method->static_method) {
75*795d594fSAndroid Build Coastguard Worker           mid = env->GetStaticMethodID(klass, methods[i].name, nb_method->signature);
76*795d594fSAndroid Build Coastguard Worker         } else {
77*795d594fSAndroid Build Coastguard Worker           mid = env->GetMethodID(klass, methods[i].name, nb_method->signature);
78*795d594fSAndroid Build Coastguard Worker         }
79*795d594fSAndroid Build Coastguard Worker         if (mid != nullptr) {
80*795d594fSAndroid Build Coastguard Worker           const char* shorty = gNativeBridgeArtCallbacks->getMethodShorty(env, mid);
81*795d594fSAndroid Build Coastguard Worker           if (strcmp(shorty, methods[i].signature) == 0) {
82*795d594fSAndroid Build Coastguard Worker             printf("    name:%s, signature:%s, shorty:%s.\n",
83*795d594fSAndroid Build Coastguard Worker                    methods[i].name, nb_method->signature, shorty);
84*795d594fSAndroid Build Coastguard Worker           }
85*795d594fSAndroid Build Coastguard Worker         }
86*795d594fSAndroid Build Coastguard Worker       }
87*795d594fSAndroid Build Coastguard Worker     }
88*795d594fSAndroid Build Coastguard Worker     methods.release();
89*795d594fSAndroid Build Coastguard Worker   }
90*795d594fSAndroid Build Coastguard Worker 
91*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
92*795d594fSAndroid Build Coastguard Worker   return fnPtr(vm, reserved);
93*795d594fSAndroid Build Coastguard Worker }
94*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testFindClassOnAttachedNativeThread(JNIEnv * env,jclass klass)95*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testFindClassOnAttachedNativeThread(JNIEnv* env, jclass klass) {
96*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = void(*)(JNIEnv*, jclass);
97*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
98*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testFindClassOnAttachedNativeThread")->fnPtr);
99*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
100*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
101*795d594fSAndroid Build Coastguard Worker }
102*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testFindFieldOnAttachedNativeThreadNative(JNIEnv * env,jclass klass)103*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testFindFieldOnAttachedNativeThreadNative(JNIEnv* env,
104*795d594fSAndroid Build Coastguard Worker                                                                            jclass klass) {
105*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = void(*)(JNIEnv*, jclass);
106*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
107*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testFindFieldOnAttachedNativeThreadNative")->fnPtr);
108*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
109*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
110*795d594fSAndroid Build Coastguard Worker }
111*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testCallStaticVoidMethodOnSubClassNative(JNIEnv * env,jclass klass)112*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testCallStaticVoidMethodOnSubClassNative(JNIEnv* env,
113*795d594fSAndroid Build Coastguard Worker                                                                           jclass klass) {
114*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = void(*)(JNIEnv*, jclass);
115*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
116*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testCallStaticVoidMethodOnSubClassNative")->fnPtr);
117*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
118*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
119*795d594fSAndroid Build Coastguard Worker }
120*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testGetMirandaMethodNative(JNIEnv * env,jclass klass)121*795d594fSAndroid Build Coastguard Worker static jobject trampoline_Java_Main_testGetMirandaMethodNative(JNIEnv* env, jclass klass) {
122*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jobject(*)(JNIEnv*, jclass);
123*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
124*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testGetMirandaMethodNative")->fnPtr);
125*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
126*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
127*795d594fSAndroid Build Coastguard Worker }
128*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testNewStringObject(JNIEnv * env,jclass klass)129*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testNewStringObject(JNIEnv* env, jclass klass) {
130*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = void(*)(JNIEnv*, jclass);
131*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
132*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testNewStringObject")->fnPtr);
133*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
134*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
135*795d594fSAndroid Build Coastguard Worker }
136*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testZeroLengthByteBuffers(JNIEnv * env,jclass klass)137*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testZeroLengthByteBuffers(JNIEnv* env, jclass klass) {
138*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = void(*)(JNIEnv*, jclass);
139*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>
140*795d594fSAndroid Build Coastguard Worker     (find_native_bridge_method("testZeroLengthByteBuffers")->fnPtr);
141*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
142*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass);
143*795d594fSAndroid Build Coastguard Worker }
144*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_byteMethod(JNIEnv * env,jclass klass,jbyte b1,jbyte b2,jbyte b3,jbyte b4,jbyte b5,jbyte b6,jbyte b7,jbyte b8,jbyte b9,jbyte b10)145*795d594fSAndroid Build Coastguard Worker static jbyte trampoline_Java_Main_byteMethod(JNIEnv* env, jclass klass, jbyte b1, jbyte b2,
146*795d594fSAndroid Build Coastguard Worker                                              jbyte b3, jbyte b4, jbyte b5, jbyte b6,
147*795d594fSAndroid Build Coastguard Worker                                              jbyte b7, jbyte b8, jbyte b9, jbyte b10) {
148*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jbyte(*)(JNIEnv*, jclass, jbyte, jbyte, jbyte, jbyte, jbyte, jbyte, jbyte, jbyte,
149*795d594fSAndroid Build Coastguard Worker                            jbyte, jbyte);
150*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>(find_native_bridge_method("byteMethod")->fnPtr);
151*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
152*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass, b1, b2, b3, b4, b5, b6, b7, b8, b9, b10);
153*795d594fSAndroid Build Coastguard Worker }
154*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_shortMethod(JNIEnv * env,jclass klass,jshort s1,jshort s2,jshort s3,jshort s4,jshort s5,jshort s6,jshort s7,jshort s8,jshort s9,jshort s10)155*795d594fSAndroid Build Coastguard Worker static jshort trampoline_Java_Main_shortMethod(JNIEnv* env, jclass klass, jshort s1, jshort s2,
156*795d594fSAndroid Build Coastguard Worker                                                jshort s3, jshort s4, jshort s5, jshort s6,
157*795d594fSAndroid Build Coastguard Worker                                                jshort s7, jshort s8, jshort s9, jshort s10) {
158*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jshort(*)(JNIEnv*, jclass, jshort, jshort, jshort, jshort, jshort, jshort, jshort,
159*795d594fSAndroid Build Coastguard Worker                             jshort, jshort, jshort);
160*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>(find_native_bridge_method("shortMethod")->fnPtr);
161*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
162*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10);
163*795d594fSAndroid Build Coastguard Worker }
164*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_booleanMethod(JNIEnv * env,jclass klass,jboolean b1,jboolean b2,jboolean b3,jboolean b4,jboolean b5,jboolean b6,jboolean b7,jboolean b8,jboolean b9,jboolean b10)165*795d594fSAndroid Build Coastguard Worker static jboolean trampoline_Java_Main_booleanMethod(JNIEnv* env, jclass klass, jboolean b1,
166*795d594fSAndroid Build Coastguard Worker                                                    jboolean b2, jboolean b3, jboolean b4,
167*795d594fSAndroid Build Coastguard Worker                                                    jboolean b5, jboolean b6, jboolean b7,
168*795d594fSAndroid Build Coastguard Worker                                                    jboolean b8, jboolean b9, jboolean b10) {
169*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jboolean(*)(JNIEnv*, jclass, jboolean, jboolean, jboolean, jboolean, jboolean,
170*795d594fSAndroid Build Coastguard Worker                               jboolean, jboolean, jboolean, jboolean, jboolean);
171*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>(find_native_bridge_method("booleanMethod")->fnPtr);
172*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
173*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass, b1, b2, b3, b4, b5, b6, b7, b8, b9, b10);
174*795d594fSAndroid Build Coastguard Worker }
175*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_charMethod(JNIEnv * env,jclass klass,jchar c1,jchar c2,jchar c3,jchar c4,jchar c5,jchar c6,jchar c7,jchar c8,jchar c9,jchar c10)176*795d594fSAndroid Build Coastguard Worker static jchar trampoline_Java_Main_charMethod(JNIEnv* env, jclass klass, jchar c1, jchar c2,
177*795d594fSAndroid Build Coastguard Worker                                              jchar c3, jchar c4, jchar c5, jchar c6,
178*795d594fSAndroid Build Coastguard Worker                                              jchar c7, jchar c8, jchar c9, jchar c10) {
179*795d594fSAndroid Build Coastguard Worker   using FnPtr_t = jchar(*)(JNIEnv*, jclass, jchar, jchar, jchar, jchar, jchar, jchar, jchar, jchar,
180*795d594fSAndroid Build Coastguard Worker                            jchar, jchar);
181*795d594fSAndroid Build Coastguard Worker   FnPtr_t fnPtr = reinterpret_cast<FnPtr_t>(find_native_bridge_method("charMethod")->fnPtr);
182*795d594fSAndroid Build Coastguard Worker   printf("%s called!\n", __FUNCTION__);
183*795d594fSAndroid Build Coastguard Worker   return fnPtr(env, klass, c1, c2, c3, c4, c5, c6, c7, c8, c9, c10);
184*795d594fSAndroid Build Coastguard Worker }
185*795d594fSAndroid Build Coastguard Worker 
186*795d594fSAndroid Build Coastguard Worker // This code is adapted from 004-SignalTest and causes a segfault.
187*795d594fSAndroid Build Coastguard Worker char *go_away_compiler = nullptr;
188*795d594fSAndroid Build Coastguard Worker 
test_sigaction_handler(int sig,siginfo_t * info,void * context)189*795d594fSAndroid Build Coastguard Worker [[ noreturn ]] static void test_sigaction_handler([[maybe_unused]] int sig,
190*795d594fSAndroid Build Coastguard Worker                                                   [[maybe_unused]] siginfo_t* info,
191*795d594fSAndroid Build Coastguard Worker                                                   [[maybe_unused]] void* context) {
192*795d594fSAndroid Build Coastguard Worker   printf("Should not reach the test sigaction handler.");
193*795d594fSAndroid Build Coastguard Worker   abort();
194*795d594fSAndroid Build Coastguard Worker }
195*795d594fSAndroid Build Coastguard Worker 
raise_sigsegv()196*795d594fSAndroid Build Coastguard Worker static void raise_sigsegv() {
197*795d594fSAndroid Build Coastguard Worker #if defined(__arm__) || defined(__i386__) || defined(__aarch64__)
198*795d594fSAndroid Build Coastguard Worker   *go_away_compiler = 'a';
199*795d594fSAndroid Build Coastguard Worker #elif defined(__riscv)
200*795d594fSAndroid Build Coastguard Worker   // Cause a SEGV using an instruction known to be 4 bytes long to account for hardcoded jump
201*795d594fSAndroid Build Coastguard Worker   // in the signal handler
202*795d594fSAndroid Build Coastguard Worker   asm volatile("ld zero, (zero);" : : :);
203*795d594fSAndroid Build Coastguard Worker #elif defined(__x86_64__)
204*795d594fSAndroid Build Coastguard Worker   // Cause a SEGV using an instruction known to be 2 bytes long to account for hardcoded jump
205*795d594fSAndroid Build Coastguard Worker   // in the signal handler
206*795d594fSAndroid Build Coastguard Worker   asm volatile("movl $0, %%eax;" "movb %%ah, (%%rax);" : : : "%eax");
207*795d594fSAndroid Build Coastguard Worker #else
208*795d594fSAndroid Build Coastguard Worker   // On other architectures we simulate SEGV.
209*795d594fSAndroid Build Coastguard Worker   kill(getpid(), SIGSEGV);
210*795d594fSAndroid Build Coastguard Worker #endif
211*795d594fSAndroid Build Coastguard Worker }
212*795d594fSAndroid Build Coastguard Worker 
trampoline_Java_Main_testSignal(JNIEnv *,jclass)213*795d594fSAndroid Build Coastguard Worker static jint trampoline_Java_Main_testSignal(JNIEnv*, jclass) {
214*795d594fSAndroid Build Coastguard Worker   // Install the sigaction handler above, which should *not* be reached as the native-bridge
215*795d594fSAndroid Build Coastguard Worker   // handler should be called first. Note: we won't chain at all, if we ever get here, we'll die.
216*795d594fSAndroid Build Coastguard Worker   struct sigaction tmp;
217*795d594fSAndroid Build Coastguard Worker   sigemptyset(&tmp.sa_mask);
218*795d594fSAndroid Build Coastguard Worker   tmp.sa_sigaction = test_sigaction_handler;
219*795d594fSAndroid Build Coastguard Worker #if !defined(__APPLE__)
220*795d594fSAndroid Build Coastguard Worker   tmp.sa_restorer = nullptr;
221*795d594fSAndroid Build Coastguard Worker #endif
222*795d594fSAndroid Build Coastguard Worker 
223*795d594fSAndroid Build Coastguard Worker   // Test segv
224*795d594fSAndroid Build Coastguard Worker   sigaction(SIGSEGV, &tmp, nullptr);
225*795d594fSAndroid Build Coastguard Worker   raise_sigsegv();
226*795d594fSAndroid Build Coastguard Worker 
227*795d594fSAndroid Build Coastguard Worker   // Test sigill
228*795d594fSAndroid Build Coastguard Worker   sigaction(SIGILL, &tmp, nullptr);
229*795d594fSAndroid Build Coastguard Worker   kill(getpid(), SIGILL);
230*795d594fSAndroid Build Coastguard Worker 
231*795d594fSAndroid Build Coastguard Worker #if defined(__BIONIC__)
232*795d594fSAndroid Build Coastguard Worker   // Do the same again, but with sigaction64.
233*795d594fSAndroid Build Coastguard Worker   struct sigaction64 tmp2;
234*795d594fSAndroid Build Coastguard Worker   sigemptyset64(&tmp2.sa_mask);
235*795d594fSAndroid Build Coastguard Worker   tmp2.sa_sigaction = test_sigaction_handler;
236*795d594fSAndroid Build Coastguard Worker #if defined(SA_RESTORER)
237*795d594fSAndroid Build Coastguard Worker   tmp2.sa_restorer = nullptr;
238*795d594fSAndroid Build Coastguard Worker #endif
239*795d594fSAndroid Build Coastguard Worker 
240*795d594fSAndroid Build Coastguard Worker   sigaction64(SIGSEGV, &tmp2, nullptr);
241*795d594fSAndroid Build Coastguard Worker   sigaction64(SIGILL, &tmp2, nullptr);
242*795d594fSAndroid Build Coastguard Worker #endif
243*795d594fSAndroid Build Coastguard Worker 
244*795d594fSAndroid Build Coastguard Worker   // Reraise SIGSEGV/SIGILL even on non-bionic, so that the expected output is
245*795d594fSAndroid Build Coastguard Worker   // the same.
246*795d594fSAndroid Build Coastguard Worker   raise_sigsegv();
247*795d594fSAndroid Build Coastguard Worker   kill(getpid(), SIGILL);
248*795d594fSAndroid Build Coastguard Worker 
249*795d594fSAndroid Build Coastguard Worker   return 1234;
250*795d594fSAndroid Build Coastguard Worker }
251*795d594fSAndroid Build Coastguard Worker 
252*795d594fSAndroid Build Coastguard Worker // Status of the tricky control path of testSignalHandlerNotReturn.
253*795d594fSAndroid Build Coastguard Worker //
254*795d594fSAndroid Build Coastguard Worker // "kNone" is the default status except testSignalHandlerNotReturn,
255*795d594fSAndroid Build Coastguard Worker // others are used by testSignalHandlerNotReturn.
256*795d594fSAndroid Build Coastguard Worker enum class TestStatus {
257*795d594fSAndroid Build Coastguard Worker   kNone,
258*795d594fSAndroid Build Coastguard Worker   kRaiseFirst,
259*795d594fSAndroid Build Coastguard Worker   kHandleFirst,
260*795d594fSAndroid Build Coastguard Worker   kRaiseSecond,
261*795d594fSAndroid Build Coastguard Worker   kHandleSecond,
262*795d594fSAndroid Build Coastguard Worker };
263*795d594fSAndroid Build Coastguard Worker 
264*795d594fSAndroid Build Coastguard Worker // State transition helper for testSignalHandlerNotReturn.
265*795d594fSAndroid Build Coastguard Worker class SignalHandlerTestStatus {
266*795d594fSAndroid Build Coastguard Worker  public:
SignalHandlerTestStatus()267*795d594fSAndroid Build Coastguard Worker   SignalHandlerTestStatus() : state_(TestStatus::kNone) {
268*795d594fSAndroid Build Coastguard Worker   }
269*795d594fSAndroid Build Coastguard Worker 
Get()270*795d594fSAndroid Build Coastguard Worker   TestStatus Get() {
271*795d594fSAndroid Build Coastguard Worker     return state_;
272*795d594fSAndroid Build Coastguard Worker   }
273*795d594fSAndroid Build Coastguard Worker 
Reset()274*795d594fSAndroid Build Coastguard Worker   void Reset() {
275*795d594fSAndroid Build Coastguard Worker     Set(TestStatus::kNone);
276*795d594fSAndroid Build Coastguard Worker   }
277*795d594fSAndroid Build Coastguard Worker 
Set(TestStatus state)278*795d594fSAndroid Build Coastguard Worker   void Set(TestStatus state) {
279*795d594fSAndroid Build Coastguard Worker     switch (state) {
280*795d594fSAndroid Build Coastguard Worker       case TestStatus::kNone:
281*795d594fSAndroid Build Coastguard Worker         AssertState(TestStatus::kHandleSecond);
282*795d594fSAndroid Build Coastguard Worker         break;
283*795d594fSAndroid Build Coastguard Worker 
284*795d594fSAndroid Build Coastguard Worker       case TestStatus::kRaiseFirst:
285*795d594fSAndroid Build Coastguard Worker         AssertState(TestStatus::kNone);
286*795d594fSAndroid Build Coastguard Worker         break;
287*795d594fSAndroid Build Coastguard Worker 
288*795d594fSAndroid Build Coastguard Worker       case TestStatus::kHandleFirst:
289*795d594fSAndroid Build Coastguard Worker         AssertState(TestStatus::kRaiseFirst);
290*795d594fSAndroid Build Coastguard Worker         break;
291*795d594fSAndroid Build Coastguard Worker 
292*795d594fSAndroid Build Coastguard Worker       case TestStatus::kRaiseSecond:
293*795d594fSAndroid Build Coastguard Worker         AssertState(TestStatus::kHandleFirst);
294*795d594fSAndroid Build Coastguard Worker         break;
295*795d594fSAndroid Build Coastguard Worker 
296*795d594fSAndroid Build Coastguard Worker       case TestStatus::kHandleSecond:
297*795d594fSAndroid Build Coastguard Worker         AssertState(TestStatus::kRaiseSecond);
298*795d594fSAndroid Build Coastguard Worker         break;
299*795d594fSAndroid Build Coastguard Worker 
300*795d594fSAndroid Build Coastguard Worker       default:
301*795d594fSAndroid Build Coastguard Worker         printf("ERROR: unknown state\n");
302*795d594fSAndroid Build Coastguard Worker         abort();
303*795d594fSAndroid Build Coastguard Worker     }
304*795d594fSAndroid Build Coastguard Worker 
305*795d594fSAndroid Build Coastguard Worker     state_ = state;
306*795d594fSAndroid Build Coastguard Worker   }
307*795d594fSAndroid Build Coastguard Worker 
308*795d594fSAndroid Build Coastguard Worker  private:
309*795d594fSAndroid Build Coastguard Worker   TestStatus state_;
310*795d594fSAndroid Build Coastguard Worker 
AssertState(TestStatus expected)311*795d594fSAndroid Build Coastguard Worker   void AssertState(TestStatus expected) {
312*795d594fSAndroid Build Coastguard Worker     if (state_ != expected) {
313*795d594fSAndroid Build Coastguard Worker       printf("ERROR: unexpected state, was %d, expected %d\n",
314*795d594fSAndroid Build Coastguard Worker              art::enum_cast<int>(state_),
315*795d594fSAndroid Build Coastguard Worker              art::enum_cast<int>(expected));
316*795d594fSAndroid Build Coastguard Worker     }
317*795d594fSAndroid Build Coastguard Worker   }
318*795d594fSAndroid Build Coastguard Worker };
319*795d594fSAndroid Build Coastguard Worker 
320*795d594fSAndroid Build Coastguard Worker static SignalHandlerTestStatus gSignalTestStatus;
321*795d594fSAndroid Build Coastguard Worker // The context is used to jump out from signal handler.
322*795d594fSAndroid Build Coastguard Worker static sigjmp_buf gSignalTestJmpBuf;
323*795d594fSAndroid Build Coastguard Worker 
324*795d594fSAndroid Build Coastguard Worker // Test whether NativeBridge can receive future signal when its handler doesn't return.
325*795d594fSAndroid Build Coastguard Worker //
326*795d594fSAndroid Build Coastguard Worker // Control path:
327*795d594fSAndroid Build Coastguard Worker //  1. Raise first SIGSEGV in test function.
328*795d594fSAndroid Build Coastguard Worker //  2. Raise another SIGSEGV in NativeBridge's signal handler which is handling
329*795d594fSAndroid Build Coastguard Worker //     the first SIGSEGV.
330*795d594fSAndroid Build Coastguard Worker //  3. Expect that NativeBridge's signal handler invokes again. And jump back
331*795d594fSAndroid Build Coastguard Worker //     to test function in when handling second SIGSEGV.
332*795d594fSAndroid Build Coastguard Worker //  4. Exit test.
333*795d594fSAndroid Build Coastguard Worker //
334*795d594fSAndroid Build Coastguard Worker // NOTE: sigchain should be aware that "special signal handler" may not return.
335*795d594fSAndroid Build Coastguard Worker //       Pay attention if this case fails.
trampoline_Java_Main_testSignalHandlerNotReturn(JNIEnv *,jclass)336*795d594fSAndroid Build Coastguard Worker static void trampoline_Java_Main_testSignalHandlerNotReturn(JNIEnv*, jclass) {
337*795d594fSAndroid Build Coastguard Worker   if (gSignalTestStatus.Get() != TestStatus::kNone) {
338*795d594fSAndroid Build Coastguard Worker     printf("ERROR: test already started?\n");
339*795d594fSAndroid Build Coastguard Worker     return;
340*795d594fSAndroid Build Coastguard Worker   }
341*795d594fSAndroid Build Coastguard Worker   printf("start testSignalHandlerNotReturn\n");
342*795d594fSAndroid Build Coastguard Worker 
343*795d594fSAndroid Build Coastguard Worker   if (sigsetjmp(gSignalTestJmpBuf, 1) == 0) {
344*795d594fSAndroid Build Coastguard Worker     gSignalTestStatus.Set(TestStatus::kRaiseFirst);
345*795d594fSAndroid Build Coastguard Worker     printf("raising first SIGSEGV\n");
346*795d594fSAndroid Build Coastguard Worker     raise_sigsegv();
347*795d594fSAndroid Build Coastguard Worker   } else {
348*795d594fSAndroid Build Coastguard Worker     // jump to here from signal handler when handling second SIGSEGV.
349*795d594fSAndroid Build Coastguard Worker     if (gSignalTestStatus.Get() != TestStatus::kHandleSecond) {
350*795d594fSAndroid Build Coastguard Worker       printf("ERROR: not jump from second SIGSEGV?\n");
351*795d594fSAndroid Build Coastguard Worker       return;
352*795d594fSAndroid Build Coastguard Worker     }
353*795d594fSAndroid Build Coastguard Worker     gSignalTestStatus.Reset();
354*795d594fSAndroid Build Coastguard Worker     printf("back to test from signal handler via siglongjmp(), and done!\n");
355*795d594fSAndroid Build Coastguard Worker   }
356*795d594fSAndroid Build Coastguard Worker }
357*795d594fSAndroid Build Coastguard Worker 
358*795d594fSAndroid Build Coastguard Worker // Signal handler for testSignalHandlerNotReturn.
359*795d594fSAndroid Build Coastguard Worker // This handler won't return.
NotReturnSignalHandler()360*795d594fSAndroid Build Coastguard Worker static bool NotReturnSignalHandler() {
361*795d594fSAndroid Build Coastguard Worker   if (gSignalTestStatus.Get() == TestStatus::kRaiseFirst) {
362*795d594fSAndroid Build Coastguard Worker     // handling first SIGSEGV
363*795d594fSAndroid Build Coastguard Worker     gSignalTestStatus.Set(TestStatus::kHandleFirst);
364*795d594fSAndroid Build Coastguard Worker     printf("handling first SIGSEGV, will raise another\n");
365*795d594fSAndroid Build Coastguard Worker     sigset_t set;
366*795d594fSAndroid Build Coastguard Worker     sigemptyset(&set);
367*795d594fSAndroid Build Coastguard Worker     sigaddset(&set, SIGSEGV);
368*795d594fSAndroid Build Coastguard Worker     printf("unblock SIGSEGV in handler\n");
369*795d594fSAndroid Build Coastguard Worker     sigprocmask(SIG_UNBLOCK, &set, nullptr);
370*795d594fSAndroid Build Coastguard Worker     gSignalTestStatus.Set(TestStatus::kRaiseSecond);
371*795d594fSAndroid Build Coastguard Worker     printf("raising second SIGSEGV\n");
372*795d594fSAndroid Build Coastguard Worker     raise_sigsegv();    // raise second SIGSEGV
373*795d594fSAndroid Build Coastguard Worker   } else if (gSignalTestStatus.Get() == TestStatus::kRaiseSecond) {
374*795d594fSAndroid Build Coastguard Worker     // handling second SIGSEGV
375*795d594fSAndroid Build Coastguard Worker     gSignalTestStatus.Set(TestStatus::kHandleSecond);
376*795d594fSAndroid Build Coastguard Worker     printf("handling second SIGSEGV, will jump back to test function\n");
377*795d594fSAndroid Build Coastguard Worker     siglongjmp(gSignalTestJmpBuf, 1);
378*795d594fSAndroid Build Coastguard Worker   }
379*795d594fSAndroid Build Coastguard Worker   printf("ERROR: should not reach here!\n");
380*795d594fSAndroid Build Coastguard Worker   return false;
381*795d594fSAndroid Build Coastguard Worker }
382*795d594fSAndroid Build Coastguard Worker 
383*795d594fSAndroid Build Coastguard Worker NativeBridgeMethod gNativeBridgeMethods[] = {
384*795d594fSAndroid Build Coastguard Worker   { "JNI_OnLoad", "", true, nullptr,
385*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_JNI_OnLoad) },
386*795d594fSAndroid Build Coastguard Worker   { "booleanMethod", "(ZZZZZZZZZZ)Z", true, nullptr,
387*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_booleanMethod) },
388*795d594fSAndroid Build Coastguard Worker   { "byteMethod", "(BBBBBBBBBB)B", true, nullptr,
389*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_byteMethod) },
390*795d594fSAndroid Build Coastguard Worker   { "charMethod", "(CCCCCCCCCC)C", true, nullptr,
391*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_charMethod) },
392*795d594fSAndroid Build Coastguard Worker   { "shortMethod", "(SSSSSSSSSS)S", true, nullptr,
393*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_shortMethod) },
394*795d594fSAndroid Build Coastguard Worker   { "testCallStaticVoidMethodOnSubClassNative", "()V", true, nullptr,
395*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testCallStaticVoidMethodOnSubClassNative) },
396*795d594fSAndroid Build Coastguard Worker   { "testFindClassOnAttachedNativeThread", "()V", true, nullptr,
397*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testFindClassOnAttachedNativeThread) },
398*795d594fSAndroid Build Coastguard Worker   { "testFindFieldOnAttachedNativeThreadNative", "()V", true, nullptr,
399*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testFindFieldOnAttachedNativeThreadNative) },
400*795d594fSAndroid Build Coastguard Worker   { "testGetMirandaMethodNative", "()Ljava/lang/reflect/Method;", true, nullptr,
401*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testGetMirandaMethodNative) },
402*795d594fSAndroid Build Coastguard Worker   { "testNewStringObject", "()V", true, nullptr,
403*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testNewStringObject) },
404*795d594fSAndroid Build Coastguard Worker   { "testZeroLengthByteBuffers", "()V", true, nullptr,
405*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testZeroLengthByteBuffers) },
406*795d594fSAndroid Build Coastguard Worker   { "testSignal", "()I", true, nullptr,
407*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testSignal) },
408*795d594fSAndroid Build Coastguard Worker   { "testSignalHandlerNotReturn", "()V", true, nullptr,
409*795d594fSAndroid Build Coastguard Worker     reinterpret_cast<void*>(trampoline_Java_Main_testSignalHandlerNotReturn) },
410*795d594fSAndroid Build Coastguard Worker };
411*795d594fSAndroid Build Coastguard Worker 
find_native_bridge_method(const char * name)412*795d594fSAndroid Build Coastguard Worker static NativeBridgeMethod* find_native_bridge_method(const char *name) {
413*795d594fSAndroid Build Coastguard Worker   const char* pname = name;
414*795d594fSAndroid Build Coastguard Worker   if (strncmp(name, "Java_Main_", 10) == 0) {
415*795d594fSAndroid Build Coastguard Worker     pname += 10;
416*795d594fSAndroid Build Coastguard Worker   }
417*795d594fSAndroid Build Coastguard Worker 
418*795d594fSAndroid Build Coastguard Worker   for (size_t i = 0; i < sizeof(gNativeBridgeMethods) / sizeof(gNativeBridgeMethods[0]); i++) {
419*795d594fSAndroid Build Coastguard Worker     if (strcmp(pname, gNativeBridgeMethods[i].name) == 0) {
420*795d594fSAndroid Build Coastguard Worker       return &gNativeBridgeMethods[i];
421*795d594fSAndroid Build Coastguard Worker     }
422*795d594fSAndroid Build Coastguard Worker   }
423*795d594fSAndroid Build Coastguard Worker   return nullptr;
424*795d594fSAndroid Build Coastguard Worker }
425*795d594fSAndroid Build Coastguard Worker 
426*795d594fSAndroid Build Coastguard Worker // NativeBridgeCallbacks implementations
native_bridge_initialize(const android::NativeBridgeRuntimeCallbacks * art_cbs,const char * app_code_cache_dir,const char * isa)427*795d594fSAndroid Build Coastguard Worker extern "C" bool native_bridge_initialize(const android::NativeBridgeRuntimeCallbacks* art_cbs,
428*795d594fSAndroid Build Coastguard Worker                                          const char* app_code_cache_dir,
429*795d594fSAndroid Build Coastguard Worker                                          [[maybe_unused]] const char* isa) {
430*795d594fSAndroid Build Coastguard Worker   struct stat st;
431*795d594fSAndroid Build Coastguard Worker   if (app_code_cache_dir != nullptr) {
432*795d594fSAndroid Build Coastguard Worker     if (stat(app_code_cache_dir, &st) == 0) {
433*795d594fSAndroid Build Coastguard Worker       if (!S_ISDIR(st.st_mode)) {
434*795d594fSAndroid Build Coastguard Worker         printf("Code cache is not a directory.\n");
435*795d594fSAndroid Build Coastguard Worker       }
436*795d594fSAndroid Build Coastguard Worker     } else {
437*795d594fSAndroid Build Coastguard Worker       perror("Error when stat-ing the code_cache:");
438*795d594fSAndroid Build Coastguard Worker     }
439*795d594fSAndroid Build Coastguard Worker   }
440*795d594fSAndroid Build Coastguard Worker 
441*795d594fSAndroid Build Coastguard Worker   if (art_cbs != nullptr) {
442*795d594fSAndroid Build Coastguard Worker     gNativeBridgeArtCallbacks = art_cbs;
443*795d594fSAndroid Build Coastguard Worker     printf("Native bridge initialized.\n");
444*795d594fSAndroid Build Coastguard Worker   }
445*795d594fSAndroid Build Coastguard Worker   return true;
446*795d594fSAndroid Build Coastguard Worker }
447*795d594fSAndroid Build Coastguard Worker 
native_bridge_loadLibrary(const char * libpath,int flag)448*795d594fSAndroid Build Coastguard Worker extern "C" void* native_bridge_loadLibrary(const char* libpath, int flag) {
449*795d594fSAndroid Build Coastguard Worker   if (strstr(libpath, "libinvalid.so") != nullptr) {
450*795d594fSAndroid Build Coastguard Worker     printf("Was to load 'libinvalid.so', force fail.\n");
451*795d594fSAndroid Build Coastguard Worker     return nullptr;
452*795d594fSAndroid Build Coastguard Worker   }
453*795d594fSAndroid Build Coastguard Worker   size_t len = strlen(libpath);
454*795d594fSAndroid Build Coastguard Worker   char* tmp = new char[len + 10];
455*795d594fSAndroid Build Coastguard Worker   strncpy(tmp, libpath, len);
456*795d594fSAndroid Build Coastguard Worker   tmp[len - 3] = '2';
457*795d594fSAndroid Build Coastguard Worker   tmp[len - 2] = '.';
458*795d594fSAndroid Build Coastguard Worker   tmp[len - 1] = 's';
459*795d594fSAndroid Build Coastguard Worker   tmp[len] = 'o';
460*795d594fSAndroid Build Coastguard Worker   tmp[len + 1] = 0;
461*795d594fSAndroid Build Coastguard Worker   void* handle = dlopen(tmp, flag);
462*795d594fSAndroid Build Coastguard Worker   delete[] tmp;
463*795d594fSAndroid Build Coastguard Worker 
464*795d594fSAndroid Build Coastguard Worker   if (handle == nullptr) {
465*795d594fSAndroid Build Coastguard Worker     printf("Handle = nullptr!\n");
466*795d594fSAndroid Build Coastguard Worker     printf("Was looking for %s.\n", libpath);
467*795d594fSAndroid Build Coastguard Worker     printf("Error = %s.\n", dlerror());
468*795d594fSAndroid Build Coastguard Worker     char cwd[1024] = {'\0'};
469*795d594fSAndroid Build Coastguard Worker     if (getcwd(cwd, sizeof(cwd)) != nullptr) {
470*795d594fSAndroid Build Coastguard Worker       printf("Current working dir: %s\n", cwd);
471*795d594fSAndroid Build Coastguard Worker     }
472*795d594fSAndroid Build Coastguard Worker   }
473*795d594fSAndroid Build Coastguard Worker   return handle;
474*795d594fSAndroid Build Coastguard Worker }
475*795d594fSAndroid Build Coastguard Worker 
native_bridge_getTrampoline(void * handle,const char * name,const char * shorty,uint32_t len)476*795d594fSAndroid Build Coastguard Worker extern "C" void* native_bridge_getTrampoline(void* handle, const char* name, const char* shorty,
477*795d594fSAndroid Build Coastguard Worker                                              [[maybe_unused]] uint32_t len) {
478*795d594fSAndroid Build Coastguard Worker   printf("Getting trampoline for %s with shorty %s.\n", name, shorty);
479*795d594fSAndroid Build Coastguard Worker 
480*795d594fSAndroid Build Coastguard Worker   // The name here is actually the JNI name, so we can directly do the lookup.
481*795d594fSAndroid Build Coastguard Worker   void* sym = dlsym(handle, name);
482*795d594fSAndroid Build Coastguard Worker   NativeBridgeMethod* method = find_native_bridge_method(name);
483*795d594fSAndroid Build Coastguard Worker   if (method == nullptr)
484*795d594fSAndroid Build Coastguard Worker     return nullptr;
485*795d594fSAndroid Build Coastguard Worker   method->fnPtr = sym;
486*795d594fSAndroid Build Coastguard Worker 
487*795d594fSAndroid Build Coastguard Worker   return method->trampoline;
488*795d594fSAndroid Build Coastguard Worker }
489*795d594fSAndroid Build Coastguard Worker 
native_bridge_isSupported(const char * libpath)490*795d594fSAndroid Build Coastguard Worker extern "C" bool native_bridge_isSupported(const char* libpath) {
491*795d594fSAndroid Build Coastguard Worker   printf("Checking for support.\n");
492*795d594fSAndroid Build Coastguard Worker 
493*795d594fSAndroid Build Coastguard Worker   if (libpath == nullptr) {
494*795d594fSAndroid Build Coastguard Worker     return false;
495*795d594fSAndroid Build Coastguard Worker   }
496*795d594fSAndroid Build Coastguard Worker   // We don't want to hijack javacore. So we should get libarttest...
497*795d594fSAndroid Build Coastguard Worker   return strcmp(libpath, "libjavacore.so") != 0;
498*795d594fSAndroid Build Coastguard Worker }
499*795d594fSAndroid Build Coastguard Worker 
500*795d594fSAndroid Build Coastguard Worker namespace android {
501*795d594fSAndroid Build Coastguard Worker 
502*795d594fSAndroid Build Coastguard Worker // Environment values required by the apps running with native bridge.
503*795d594fSAndroid Build Coastguard Worker struct NativeBridgeRuntimeValues {
504*795d594fSAndroid Build Coastguard Worker   const char* os_arch;
505*795d594fSAndroid Build Coastguard Worker   const char* cpu_abi;
506*795d594fSAndroid Build Coastguard Worker   const char* cpu_abi2;
507*795d594fSAndroid Build Coastguard Worker   const char* *supported_abis;
508*795d594fSAndroid Build Coastguard Worker   int32_t abi_count;
509*795d594fSAndroid Build Coastguard Worker };
510*795d594fSAndroid Build Coastguard Worker 
511*795d594fSAndroid Build Coastguard Worker }  // namespace android
512*795d594fSAndroid Build Coastguard Worker 
513*795d594fSAndroid Build Coastguard Worker const char* supported_abis[] = {
514*795d594fSAndroid Build Coastguard Worker     "supported1", "supported2", "supported3"
515*795d594fSAndroid Build Coastguard Worker };
516*795d594fSAndroid Build Coastguard Worker 
517*795d594fSAndroid Build Coastguard Worker const struct android::NativeBridgeRuntimeValues nb_env {
518*795d594fSAndroid Build Coastguard Worker     .os_arch = "os.arch",
519*795d594fSAndroid Build Coastguard Worker     .cpu_abi = "cpu_abi",
520*795d594fSAndroid Build Coastguard Worker     .cpu_abi2 = "cpu_abi2",
521*795d594fSAndroid Build Coastguard Worker     .supported_abis = supported_abis,
522*795d594fSAndroid Build Coastguard Worker     .abi_count = 3
523*795d594fSAndroid Build Coastguard Worker };
524*795d594fSAndroid Build Coastguard Worker 
native_bridge_getAppEnv(const char * abi)525*795d594fSAndroid Build Coastguard Worker extern "C" const struct android::NativeBridgeRuntimeValues* native_bridge_getAppEnv(
526*795d594fSAndroid Build Coastguard Worker     const char* abi) {
527*795d594fSAndroid Build Coastguard Worker   printf("Checking for getEnvValues.\n");
528*795d594fSAndroid Build Coastguard Worker 
529*795d594fSAndroid Build Coastguard Worker   if (abi == nullptr) {
530*795d594fSAndroid Build Coastguard Worker     return nullptr;
531*795d594fSAndroid Build Coastguard Worker   }
532*795d594fSAndroid Build Coastguard Worker 
533*795d594fSAndroid Build Coastguard Worker   return &nb_env;
534*795d594fSAndroid Build Coastguard Worker }
535*795d594fSAndroid Build Coastguard Worker 
536*795d594fSAndroid Build Coastguard Worker // v2 parts.
537*795d594fSAndroid Build Coastguard Worker 
native_bridge_isCompatibleWith(uint32_t bridge_version)538*795d594fSAndroid Build Coastguard Worker extern "C" bool native_bridge_isCompatibleWith(uint32_t bridge_version) {
539*795d594fSAndroid Build Coastguard Worker   return bridge_version <= 3;
540*795d594fSAndroid Build Coastguard Worker }
541*795d594fSAndroid Build Coastguard Worker 
542*795d594fSAndroid Build Coastguard Worker #if defined(__i386__) || defined(__x86_64__)
543*795d594fSAndroid Build Coastguard Worker #if defined(__APPLE__)
544*795d594fSAndroid Build Coastguard Worker #define ucontext __darwin_ucontext
545*795d594fSAndroid Build Coastguard Worker 
546*795d594fSAndroid Build Coastguard Worker #if defined(__x86_64__)
547*795d594fSAndroid Build Coastguard Worker // 64 bit mac build.
548*795d594fSAndroid Build Coastguard Worker #define CTX_EIP uc_mcontext->__ss.__rip
549*795d594fSAndroid Build Coastguard Worker #else
550*795d594fSAndroid Build Coastguard Worker // 32 bit mac build.
551*795d594fSAndroid Build Coastguard Worker #define CTX_EIP uc_mcontext->__ss.__eip
552*795d594fSAndroid Build Coastguard Worker #endif
553*795d594fSAndroid Build Coastguard Worker 
554*795d594fSAndroid Build Coastguard Worker #elif defined(__x86_64__)
555*795d594fSAndroid Build Coastguard Worker // 64 bit linux build.
556*795d594fSAndroid Build Coastguard Worker #define CTX_EIP uc_mcontext.gregs[REG_RIP]
557*795d594fSAndroid Build Coastguard Worker #else
558*795d594fSAndroid Build Coastguard Worker // 32 bit linux build.
559*795d594fSAndroid Build Coastguard Worker #define CTX_EIP uc_mcontext.gregs[REG_EIP]
560*795d594fSAndroid Build Coastguard Worker #endif
561*795d594fSAndroid Build Coastguard Worker #endif
562*795d594fSAndroid Build Coastguard Worker 
StandardSignalHandler(int sig,siginfo_t * info,void * context)563*795d594fSAndroid Build Coastguard Worker static bool StandardSignalHandler(int sig, [[maybe_unused]] siginfo_t* info, void* context) {
564*795d594fSAndroid Build Coastguard Worker   if (sig == SIGSEGV) {
565*795d594fSAndroid Build Coastguard Worker #if defined(__arm__)
566*795d594fSAndroid Build Coastguard Worker     ucontext_t* uc = reinterpret_cast<ucontext_t*>(context);
567*795d594fSAndroid Build Coastguard Worker     mcontext_t* mc = reinterpret_cast<mcontext_t*>(&uc->uc_mcontext);
568*795d594fSAndroid Build Coastguard Worker     mc->arm_pc += 2;  // Skip instruction causing segv & sigill.
569*795d594fSAndroid Build Coastguard Worker #elif defined(__aarch64__)
570*795d594fSAndroid Build Coastguard Worker     ucontext_t* uc = reinterpret_cast<ucontext_t*>(context);
571*795d594fSAndroid Build Coastguard Worker     mcontext_t* mc = reinterpret_cast<mcontext_t*>(&uc->uc_mcontext);
572*795d594fSAndroid Build Coastguard Worker     mc->pc += 4;  // Skip instruction causing segv & sigill.
573*795d594fSAndroid Build Coastguard Worker #elif defined(__riscv)
574*795d594fSAndroid Build Coastguard Worker     ucontext_t* uc = reinterpret_cast<ucontext_t*>(context);
575*795d594fSAndroid Build Coastguard Worker     mcontext_t* mc = reinterpret_cast<mcontext_t*>(&uc->uc_mcontext);
576*795d594fSAndroid Build Coastguard Worker     mc->__gregs[REG_PC] += 4;  // Skip instruction causing segv & sigill.
577*795d594fSAndroid Build Coastguard Worker #elif defined(__i386__)
578*795d594fSAndroid Build Coastguard Worker     ucontext_t* uc = reinterpret_cast<ucontext_t*>(context);
579*795d594fSAndroid Build Coastguard Worker     uc->CTX_EIP += 3;
580*795d594fSAndroid Build Coastguard Worker #elif defined(__x86_64__)
581*795d594fSAndroid Build Coastguard Worker     ucontext_t* uc = reinterpret_cast<ucontext_t*>(context);
582*795d594fSAndroid Build Coastguard Worker     uc->CTX_EIP += 2;
583*795d594fSAndroid Build Coastguard Worker #else
584*795d594fSAndroid Build Coastguard Worker     UNUSED(context);
585*795d594fSAndroid Build Coastguard Worker     UNIMPLEMENTED(FATAL) << "Unsupported architecture";
586*795d594fSAndroid Build Coastguard Worker #endif
587*795d594fSAndroid Build Coastguard Worker   }
588*795d594fSAndroid Build Coastguard Worker 
589*795d594fSAndroid Build Coastguard Worker   // We handled this...
590*795d594fSAndroid Build Coastguard Worker   return true;
591*795d594fSAndroid Build Coastguard Worker }
592*795d594fSAndroid Build Coastguard Worker 
593*795d594fSAndroid Build Coastguard Worker // A placeholder special handler, continueing after the faulting location. This code comes from
594*795d594fSAndroid Build Coastguard Worker // 004-SignalTest.
nb_signalhandler(int sig,siginfo_t * info,void * context)595*795d594fSAndroid Build Coastguard Worker static bool nb_signalhandler(int sig, siginfo_t* info, void* context) {
596*795d594fSAndroid Build Coastguard Worker   printf("NB signal handler with signal %d.\n", sig);
597*795d594fSAndroid Build Coastguard Worker 
598*795d594fSAndroid Build Coastguard Worker   if (gSignalTestStatus.Get() == TestStatus::kNone) {
599*795d594fSAndroid Build Coastguard Worker     return StandardSignalHandler(sig, info, context);
600*795d594fSAndroid Build Coastguard Worker   } else if (sig == SIGSEGV) {
601*795d594fSAndroid Build Coastguard Worker     return NotReturnSignalHandler();
602*795d594fSAndroid Build Coastguard Worker   } else {
603*795d594fSAndroid Build Coastguard Worker     printf("ERROR: should not reach here!\n");
604*795d594fSAndroid Build Coastguard Worker     return false;
605*795d594fSAndroid Build Coastguard Worker   }
606*795d594fSAndroid Build Coastguard Worker }
607*795d594fSAndroid Build Coastguard Worker 
native_bridge_getSignalHandler(int signal)608*795d594fSAndroid Build Coastguard Worker static ::android::NativeBridgeSignalHandlerFn native_bridge_getSignalHandler(int signal) {
609*795d594fSAndroid Build Coastguard Worker   // Test segv for already claimed signal, and sigill for not claimed signal
610*795d594fSAndroid Build Coastguard Worker   if ((signal == SIGSEGV) || (signal == SIGILL)) {
611*795d594fSAndroid Build Coastguard Worker     return &nb_signalhandler;
612*795d594fSAndroid Build Coastguard Worker   }
613*795d594fSAndroid Build Coastguard Worker   return nullptr;
614*795d594fSAndroid Build Coastguard Worker }
615*795d594fSAndroid Build Coastguard Worker 
native_bridge_unloadLibrary(void * handle)616*795d594fSAndroid Build Coastguard Worker extern "C" int native_bridge_unloadLibrary([[maybe_unused]] void* handle) {
617*795d594fSAndroid Build Coastguard Worker   printf("dlclose() in native bridge.\n");
618*795d594fSAndroid Build Coastguard Worker   return 0;
619*795d594fSAndroid Build Coastguard Worker }
620*795d594fSAndroid Build Coastguard Worker 
native_bridge_getError()621*795d594fSAndroid Build Coastguard Worker extern "C" const char* native_bridge_getError() {
622*795d594fSAndroid Build Coastguard Worker   printf("getError() in native bridge.\n");
623*795d594fSAndroid Build Coastguard Worker   return "";
624*795d594fSAndroid Build Coastguard Worker }
625*795d594fSAndroid Build Coastguard Worker 
native_bridge_isPathSupported(const char * library_path)626*795d594fSAndroid Build Coastguard Worker extern "C" bool native_bridge_isPathSupported([[maybe_unused]] const char* library_path) {
627*795d594fSAndroid Build Coastguard Worker   printf("Checking for path support in native bridge.\n");
628*795d594fSAndroid Build Coastguard Worker   return false;
629*795d594fSAndroid Build Coastguard Worker }
630*795d594fSAndroid Build Coastguard Worker 
631*795d594fSAndroid Build Coastguard Worker extern "C" android::native_bridge_namespace_t*
native_bridge_createNamespace(const char * name,const char * ld_library_path,const char * default_library_path,uint64_t type,const char * permitted_when_isolated_path,android::native_bridge_namespace_t * parent_ns)632*795d594fSAndroid Build Coastguard Worker native_bridge_createNamespace([[maybe_unused]] const char* name,
633*795d594fSAndroid Build Coastguard Worker                               [[maybe_unused]] const char* ld_library_path,
634*795d594fSAndroid Build Coastguard Worker                               [[maybe_unused]] const char* default_library_path,
635*795d594fSAndroid Build Coastguard Worker                               [[maybe_unused]] uint64_t type,
636*795d594fSAndroid Build Coastguard Worker                               [[maybe_unused]] const char* permitted_when_isolated_path,
637*795d594fSAndroid Build Coastguard Worker                               [[maybe_unused]] android::native_bridge_namespace_t* parent_ns) {
638*795d594fSAndroid Build Coastguard Worker   printf("Creating namespace in native bridge.\n");
639*795d594fSAndroid Build Coastguard Worker   return nullptr;
640*795d594fSAndroid Build Coastguard Worker }
641*795d594fSAndroid Build Coastguard Worker 
native_bridge_linkNamespaces(android::native_bridge_namespace_t * from,android::native_bridge_namespace_t * to,const char * shared_libs_sonames)642*795d594fSAndroid Build Coastguard Worker extern "C" bool native_bridge_linkNamespaces(
643*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] android::native_bridge_namespace_t* from,
644*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] android::native_bridge_namespace_t* to,
645*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] const char* shared_libs_sonames) {
646*795d594fSAndroid Build Coastguard Worker   printf("Linking namespaces in native bridge.\n");
647*795d594fSAndroid Build Coastguard Worker   return false;
648*795d594fSAndroid Build Coastguard Worker }
649*795d594fSAndroid Build Coastguard Worker 
native_bridge_loadLibraryExt(const char * libpath,int flag,android::native_bridge_namespace_t * ns)650*795d594fSAndroid Build Coastguard Worker extern "C" void* native_bridge_loadLibraryExt(
651*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] const char* libpath,
652*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] int flag,
653*795d594fSAndroid Build Coastguard Worker     [[maybe_unused]] android::native_bridge_namespace_t* ns) {
654*795d594fSAndroid Build Coastguard Worker   printf("Loading library with Extension in native bridge.\n");
655*795d594fSAndroid Build Coastguard Worker   return nullptr;
656*795d594fSAndroid Build Coastguard Worker }
657*795d594fSAndroid Build Coastguard Worker 
658*795d594fSAndroid Build Coastguard Worker // "NativeBridgeItf" is effectively an API (it is the name of the symbol that will be loaded
659*795d594fSAndroid Build Coastguard Worker // by the native bridge library).
660*795d594fSAndroid Build Coastguard Worker android::NativeBridgeCallbacks NativeBridgeItf{
661*795d594fSAndroid Build Coastguard Worker     // v1
662*795d594fSAndroid Build Coastguard Worker     .version = 3,
663*795d594fSAndroid Build Coastguard Worker     .initialize = &native_bridge_initialize,
664*795d594fSAndroid Build Coastguard Worker     .loadLibrary = &native_bridge_loadLibrary,
665*795d594fSAndroid Build Coastguard Worker     .getTrampoline = &native_bridge_getTrampoline,
666*795d594fSAndroid Build Coastguard Worker     .isSupported = &native_bridge_isSupported,
667*795d594fSAndroid Build Coastguard Worker     .getAppEnv = &native_bridge_getAppEnv,
668*795d594fSAndroid Build Coastguard Worker     // v2
669*795d594fSAndroid Build Coastguard Worker     .isCompatibleWith = &native_bridge_isCompatibleWith,
670*795d594fSAndroid Build Coastguard Worker     .getSignalHandler = &native_bridge_getSignalHandler,
671*795d594fSAndroid Build Coastguard Worker     // v3
672*795d594fSAndroid Build Coastguard Worker     .unloadLibrary = &native_bridge_unloadLibrary,
673*795d594fSAndroid Build Coastguard Worker     .getError = &native_bridge_getError,
674*795d594fSAndroid Build Coastguard Worker     .isPathSupported = &native_bridge_isPathSupported,
675*795d594fSAndroid Build Coastguard Worker     .unused_initAnonymousNamespace = nullptr,
676*795d594fSAndroid Build Coastguard Worker     .createNamespace = &native_bridge_createNamespace,
677*795d594fSAndroid Build Coastguard Worker     .linkNamespaces = &native_bridge_linkNamespaces,
678*795d594fSAndroid Build Coastguard Worker     .loadLibraryExt = &native_bridge_loadLibraryExt,
679*795d594fSAndroid Build Coastguard Worker };
680