• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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