1 // Copyright 2021 Code Intelligence GmbH
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "java_reproducer.h"
16
17 #include "fuzzed_data_provider.h"
18 #include "jvm_tooling.h"
19
20 namespace {
21 const char kRecordingFuzzedDataProviderClass[] =
22 "com/code_intelligence/jazzer/runtime/RecordingFuzzedDataProvider";
23 }
24
25 namespace jazzer {
GetFuzzedDataProviderJavaObject(const JVM & jvm)26 jobject GetFuzzedDataProviderJavaObject(const JVM &jvm) {
27 static jobject java_object = nullptr;
28 if (java_object == nullptr) {
29 jclass java_class = jvm.FindClass(kFuzzedDataProviderImplClass);
30 jmethodID java_constructor = jvm.GetMethodID(java_class, "<init>", "()V");
31 jobject local_ref = jvm.GetEnv().NewObject(java_class, java_constructor);
32 // We leak a global reference here as it will be used until JVM exit.
33 java_object = jvm.GetEnv().NewGlobalRef(local_ref);
34 }
35 return java_object;
36 }
37
GetRecordingFuzzedDataProviderJavaObject(const JVM & jvm)38 jobject GetRecordingFuzzedDataProviderJavaObject(const JVM &jvm) {
39 auto &env = jvm.GetEnv();
40 jclass java_class = jvm.FindClass(kRecordingFuzzedDataProviderClass);
41 jmethodID java_make_proxy = jvm.GetStaticMethodID(
42 java_class, "makeFuzzedDataProviderProxy",
43 "()Lcom/code_intelligence/jazzer/api/FuzzedDataProvider;", true);
44 jobject local_ref = env.CallStaticObjectMethod(java_class, java_make_proxy);
45 if (env.ExceptionCheck()) {
46 env.ExceptionDescribe();
47 exit(1);
48 }
49 // This global reference is deleted in SerializeRecordingFuzzedDataProvider.
50 jobject global_ref = env.NewGlobalRef(local_ref);
51 env.DeleteLocalRef(local_ref);
52 return global_ref;
53 }
54
SerializeRecordingFuzzedDataProvider(const JVM & jvm,jobject recorder)55 std::string SerializeRecordingFuzzedDataProvider(const JVM &jvm,
56 jobject recorder) {
57 auto &env = jvm.GetEnv();
58 jclass java_class = jvm.FindClass(kRecordingFuzzedDataProviderClass);
59 jmethodID java_serialize =
60 jvm.GetStaticMethodID(java_class, "serializeFuzzedDataProviderProxy",
61 "(Lcom/code_intelligence/jazzer/api/"
62 "FuzzedDataProvider;)Ljava/lang/String;",
63 true);
64 auto serialized_recorder =
65 (jstring)env.CallStaticObjectMethod(java_class, java_serialize, recorder);
66 env.DeleteLocalRef(java_class);
67 env.DeleteGlobalRef(recorder);
68 if (env.ExceptionCheck()) {
69 env.ExceptionDescribe();
70 exit(1);
71 }
72 const char *serialized_recorder_cstr =
73 env.GetStringUTFChars(serialized_recorder, nullptr);
74 std::string out(serialized_recorder_cstr);
75 env.ReleaseStringUTFChars(serialized_recorder, serialized_recorder_cstr);
76 env.DeleteLocalRef(serialized_recorder);
77 return out;
78 }
79 } // namespace jazzer
80