[tint][ir] Add roundtrip fuzzer
Change-Id: Ia0354874ed9a3c8aaf2ad6e9703107d953e8a431
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/139600
Reviewed-by: Dan Sinclair <dsinclair@chromium.org>
Commit-Queue: Ben Clayton <bclayton@google.com>
Kokoro: Kokoro <noreply+kokoro@google.com>
diff --git a/src/tint/fuzzers/CMakeLists.txt b/src/tint/fuzzers/CMakeLists.txt
index 5640f95..c53b0ec 100644
--- a/src/tint/fuzzers/CMakeLists.txt
+++ b/src/tint/fuzzers/CMakeLists.txt
@@ -56,6 +56,10 @@
add_tint_fuzzer(tint_wgsl_reader_spv_writer_fuzzer)
endif()
+if (${TINT_BUILD_WGSL_READER} AND ${TINT_BUILD_IR})
+ add_tint_fuzzer(tint_ir_roundtrip_fuzzer)
+endif()
+
if (${TINT_BUILD_WGSL_READER} AND ${TINT_BUILD_HLSL_WRITER})
add_tint_fuzzer(tint_wgsl_reader_hlsl_writer_fuzzer)
endif()
diff --git a/src/tint/fuzzers/tint_ir_roundtrip_fuzzer.cc b/src/tint/fuzzers/tint_ir_roundtrip_fuzzer.cc
new file mode 100644
index 0000000..18670d7
--- /dev/null
+++ b/src/tint/fuzzers/tint_ir_roundtrip_fuzzer.cc
@@ -0,0 +1,60 @@
+// Copyright 2023 The Tint Authors.
+//
+// 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 <iostream>
+#include <string>
+#include <unordered_set>
+
+#include "src/tint/ir/from_program.h"
+#include "src/tint/ir/to_program.h"
+#include "src/tint/reader/wgsl/parser_impl.h"
+
+[[noreturn]] void TintInternalCompilerErrorReporter(const tint::diag::List& diagnostics) {
+ auto printer = tint::diag::Printer::create(stderr, true);
+ tint::diag::Formatter{}.format(diagnostics, printer.get());
+ __builtin_trap();
+}
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
+ std::string str(reinterpret_cast<const char*>(data), size);
+
+ tint::SetInternalCompilerErrorReporter(&TintInternalCompilerErrorReporter);
+
+ tint::Source::File file("test.wgsl", str);
+
+ // Parse the wgsl, create the src program
+ tint::reader::wgsl::ParserImpl parser(&file);
+ parser.set_max_errors(1);
+ if (!parser.Parse()) {
+ return 0;
+ }
+ auto src = parser.program();
+ if (!src.IsValid()) {
+ return 0;
+ }
+
+ auto ir = tint::ir::FromProgram(&src);
+ if (!ir) {
+ std::cerr << ir.Failure() << std::endl;
+ __builtin_trap();
+ }
+
+ auto dst = tint::ir::ToProgram(ir.Get());
+ if (!dst.IsValid()) {
+ std::cerr << dst.Diagnostics() << std::endl;
+ __builtin_trap();
+ }
+
+ return 0;
+}
