mirror of https://github.com/google/oss-fuzz.git
78 lines
2.5 KiB
Diff
78 lines
2.5 KiB
Diff
diff --git a/tensorflow/security/fuzzing/cc/BUILD b/tensorflow/security/fuzzing/cc/BUILD
|
|
index 231a40bf..7330af85 100644
|
|
--- a/tensorflow/security/fuzzing/cc/BUILD
|
|
+++ b/tensorflow/security/fuzzing/cc/BUILD
|
|
@@ -55,23 +55,23 @@ tf_cc_fuzz_test(
|
|
],
|
|
)
|
|
|
|
-tf_cc_fuzz_test(
|
|
- name = "checkpoint_reader_fuzz",
|
|
- srcs = ["checkpoint_reader_fuzz.cc"],
|
|
- data = glob(["checkpoint_reader_testdata/*"]),
|
|
- tags = ["no_oss"],
|
|
- deps = [
|
|
- ":checkpoint_reader_fuzz_input_proto_cc",
|
|
- "//tensorflow/c:checkpoint_reader",
|
|
- "//tensorflow/c:tf_status_headers",
|
|
- "//tensorflow/c:tf_status_helper",
|
|
- "//tensorflow/core:framework",
|
|
- "//tensorflow/core:lib",
|
|
- "//tensorflow/core/framework:types_proto_cc",
|
|
- "//tensorflow/core/platform:resource_loader",
|
|
- "//tensorflow/core/platform:status",
|
|
- ],
|
|
-)
|
|
+#tf_cc_fuzz_test(
|
|
+# name = "checkpoint_reader_fuzz",
|
|
+# srcs = ["checkpoint_reader_fuzz.cc"],
|
|
+# data = glob(["checkpoint_reader_testdata/*"]),
|
|
+# tags = ["no_oss"],
|
|
+# deps = [
|
|
+# ":checkpoint_reader_fuzz_input_proto_cc",
|
|
+# "//tensorflow/c:checkpoint_reader",
|
|
+# "//tensorflow/c:tf_status_headers",
|
|
+# "//tensorflow/c:tf_status_helper",
|
|
+# "//tensorflow/core:framework",
|
|
+# "//tensorflow/core:lib",
|
|
+# "//tensorflow/core/framework:types_proto_cc",
|
|
+# "//tensorflow/core/platform:resource_loader",
|
|
+# "//tensorflow/core/platform:status",
|
|
+# ],
|
|
+#)
|
|
|
|
tf_proto_library(
|
|
name = "checkpoint_reader_fuzz_input_proto",
|
|
diff --git a/tensorflow/security/fuzzing/tf_fuzzing.bzl b/tensorflow/security/fuzzing/tf_fuzzing.bzl
|
|
index 9223d294..f2c795b3 100644
|
|
--- a/tensorflow/security/fuzzing/tf_fuzzing.bzl
|
|
+++ b/tensorflow/security/fuzzing/tf_fuzzing.bzl
|
|
@@ -19,6 +19,27 @@
|
|
# mutators to do structure aware fuzzing) or any other type of content that is
|
|
# not classified elsewhere.
|
|
|
|
+load(
|
|
+ "//tensorflow:tensorflow.bzl",
|
|
+ "tf_cc_test",
|
|
+)
|
|
+
|
|
+def tf_oss_fuzz_fuzztest(
|
|
+ name,
|
|
+ fuzzing_dict = [],
|
|
+ corpus = [],
|
|
+ deps = [],
|
|
+ tags = [],
|
|
+ **kwargs):
|
|
+ tf_cc_test(
|
|
+ name = name,
|
|
+ deps = deps + [
|
|
+ "@com_google_fuzztest//fuzztest",
|
|
+ "@com_google_fuzztest//fuzztest:fuzztest_gtest_main",
|
|
+ ],
|
|
+ **kwargs
|
|
+ )
|
|
+
|
|
# tf_cc_fuzz_test is a cc_test modified to include fuzzing support and dependencies for go/fuzztest.
|
|
def tf_cc_fuzz_test(
|
|
name,
|