skia2/fuzz/oss_fuzz/FuzzSKSL2Pipeline.cpp
John Stiles 31e4c38236 Add support for function prototypes in Pipeline stage.
Previously we did not have a Pipeline callback function for prototyping
a function, so prototypes would be discarded during translation. This
failure mode can be seen in http://review.skia.org/454741, where
FunctionPrototype.sksl is made more complex (thwarting the inliner).
This causes us to emit invalid GLSL, and dm asserts/fails in the SkSL
tests: http://screen/4PkEEWn4m4tF5e7

This CL makes the same changes to FunctionPrototype, but does not crash.

Change-Id: Ia342c7811a454f62f52677440d247e628a1bdc4f
Bug: skia:12488
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/454740
Commit-Queue: Ethan Nicholas <ethannicholas@google.com>
Reviewed-by: Ethan Nicholas <ethannicholas@google.com>
Auto-Submit: John Stiles <johnstiles@google.com>
2021-09-30 20:02:19 +00:00

69 lines
2.4 KiB
C++

/*
* Copyright 2019 Google, LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "src/gpu/GrShaderCaps.h"
#include "src/sksl/SkSLCompiler.h"
#include "src/sksl/codegen/SkSLPipelineStageCodeGenerator.h"
#include "src/sksl/ir/SkSLVarDeclarations.h"
#include "src/sksl/ir/SkSLVariable.h"
#include "fuzz/Fuzz.h"
bool FuzzSKSL2Pipeline(sk_sp<SkData> bytes) {
sk_sp<GrShaderCaps> caps = SkSL::ShaderCapsFactory::Default();
SkSL::Compiler compiler(caps.get());
SkSL::Program::Settings settings;
std::unique_ptr<SkSL::Program> program = compiler.convertProgram(
SkSL::ProgramKind::kRuntimeShader,
SkSL::String((const char*) bytes->data(),
bytes->size()),
settings);
if (!program) {
return false;
}
class Callbacks : public SkSL::PipelineStage::Callbacks {
using String = SkSL::String;
String declareUniform(const SkSL::VarDeclaration* decl) override {
return String(decl->var().name());
}
void defineFunction(const char* /*decl*/, const char* /*body*/, bool /*isMain*/) override {}
void declareFunction(const char* /*decl*/) override {}
void defineStruct(const char* /*definition*/) override {}
void declareGlobal(const char* /*declaration*/) override {}
String sampleShader(int index, String coords) override {
return "child_" + SkSL::to_string(index) + ".eval(" + coords + ")";
}
String sampleColorFilter(int index, String color) override {
return "child_" + SkSL::to_string(index) + ".eval(" + color + ")";
}
String sampleBlender(int index, String src, String dst) override {
return "child_" + SkSL::to_string(index) + ".eval(" + src + ", " + dst + ")";
}
};
Callbacks callbacks;
SkSL::PipelineStage::ConvertProgram(*program, "coords", "inColor", "half4(1)", &callbacks);
return true;
}
#if defined(SK_BUILD_FOR_LIBFUZZER)
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
if (size > 3000) {
return 0;
}
auto bytes = SkData::MakeWithoutCopy(data, size);
FuzzSKSL2Pipeline(bytes);
return 0;
}
#endif