Add two new fuzz targets (#7386)

* Add two new fuzz targets

* Add copyright

4 files changed, 138 insertions, 6 deletions

diff --git a/projects/jsonnet/build.sh b/projects/jsonnet/build.sh
index 1b0ceed4..a1d164d9 100644
--- a/projects/jsonnet/build.sh
+++ b/projects/jsonnet/build.sh
@@ -24,9 +24,12 @@ popd
 
 INSTALL_DIR="$SRC/jsonnet"
 
-fuzzer=convert_jsonnet_fuzzer
-$CXX $CXXFLAGS -I${INSTALL_DIR}/include $LIB_FUZZING_ENGINE \
-  $fuzzer.cc -o $OUT/$fuzzer \
-  ${INSTALL_DIR}/build/libjsonnet.a \
-  ${INSTALL_DIR}/build/libmd5.a \
-  ${INSTALL_DIR}/build/libryml.a
+for fuzzer in convert_jsonnet_fuzzer_regular \
+ convert_jsonnet_fuzzer_stream \
+ convert_jsonnet_fuzzer_multi; do
+  $CXX $CXXFLAGS -I${INSTALL_DIR}/include $LIB_FUZZING_ENGINE \
+    $fuzzer.cc -o $OUT/$fuzzer \
+    ${INSTALL_DIR}/build/libjsonnet.a \
+    ${INSTALL_DIR}/build/libmd5.a \
+    ${INSTALL_DIR}/build/libryml.a
+done
diff --git a/projects/jsonnet/convert_jsonnet_fuzzer_multi.cc b/projects/jsonnet/convert_jsonnet_fuzzer_multi.cc
new file mode 100644
index 00000000..3f824799
--- /dev/null
+++ b/projects/jsonnet/convert_jsonnet_fuzzer_multi.cc
@@ -0,0 +1,57 @@
+// Copyright 2022 Google Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+////////////////////////////////////////////////////////////////////////////////
+#include <cstddef>
+#include <cstdint>
+#include <cstring>
+#include <string>
+
+extern "C" {
+#include "libjsonnet.h"
+}
+
+char* ImportCallback(void* ctx, const char* base, const char* rel,
+                     char** found_here, int* success) {
+  // Don't load file and mark it as failure.
+  *success = 0;
+  char* res = jsonnet_realloc(static_cast<struct JsonnetVm*>(ctx), nullptr, 1);
+  res[0] = 0;
+  return res;
+}
+
+std::string ConvertJsonnetToJson(const std::string& jsonnet) {
+  JsonnetVm* jvm = jsonnet_make();
+  jsonnet_import_callback(jvm, ImportCallback, jvm);
+  int error = 0;
+  char* res =
+      jsonnet_evaluate_snippet_multi(jvm, /*filename=*/"", jsonnet.c_str(), &error);
+
+  std::string json;
+  if (error == 0 && res != nullptr) {
+    json = res;
+  }
+
+  if (res) {
+    jsonnet_realloc(jvm, res, 0);
+  }
+  jsonnet_destroy(jvm);
+  return json;
+}
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
+  std::string fuzz_jsonnet(reinterpret_cast<const char*>(data), size);
+  ConvertJsonnetToJson(fuzz_jsonnet);
+  return 0;
+}
diff --git a/projects/jsonnet/convert_jsonnet_fuzzer.cc b/projects/jsonnet/convert_jsonnet_fuzzer_regular.cc
index 553095a1..034b9ed3 100644
--- a/projects/jsonnet/convert_jsonnet_fuzzer.cc
+++ b/projects/jsonnet/convert_jsonnet_fuzzer_regular.cc
@@ -1,3 +1,18 @@
+// Copyright 2022 Google Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+////////////////////////////////////////////////////////////////////////////////
 #include <cstddef>
 #include <cstdint>
 #include <cstring>
diff --git a/projects/jsonnet/convert_jsonnet_fuzzer_stream.cc b/projects/jsonnet/convert_jsonnet_fuzzer_stream.cc
new file mode 100644
index 00000000..8eb107a1
--- /dev/null
+++ b/projects/jsonnet/convert_jsonnet_fuzzer_stream.cc
@@ -0,0 +1,57 @@
+// Copyright 2022 Google Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+////////////////////////////////////////////////////////////////////////////////
+#include <cstddef>
+#include <cstdint>
+#include <cstring>
+#include <string>
+
+extern "C" {
+#include "libjsonnet.h"
+}
+
+char* ImportCallback(void* ctx, const char* base, const char* rel,
+                     char** found_here, int* success) {
+  // Don't load file and mark it as failure.
+  *success = 0;
+  char* res = jsonnet_realloc(static_cast<struct JsonnetVm*>(ctx), nullptr, 1);
+  res[0] = 0;
+  return res;
+}
+
+std::string ConvertJsonnetToJson(const std::string& jsonnet) {
+  JsonnetVm* jvm = jsonnet_make();
+  jsonnet_import_callback(jvm, ImportCallback, jvm);
+  int error = 0;
+  char* res =
+      jsonnet_evaluate_snippet_stream(jvm, /*filename=*/"", jsonnet.c_str(), &error);
+
+  std::string json;
+  if (error == 0 && res != nullptr) {
+    json = res;
+  }
+
+  if (res) {
+    jsonnet_realloc(jvm, res, 0);
+  }
+  jsonnet_destroy(jvm);
+  return json;
+}
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
+  std::string fuzz_jsonnet(reinterpret_cast<const char*>(data), size);
+  ConvertJsonnetToJson(fuzz_jsonnet);
+  return 0;
+}