Edit

kc3-lang/angle/src/compiler/fuzz

Branch :

  • Show log

    Commit

  • Author : Olli Etuaho
    Date : 2016-11-22 15:42:37
    Hash : 28cb0368
    Message : Check for misconfiguration of shader built-ins Fail compiler initialization if the built-in resources are invalid. This avoids creating zero-sized arrays out of built-ins into the symbol table, which could later lead to asserts when these built-ins were indexed by constants. BUG=chromium:667468 TEST=angle_unittests Change-Id: I9553c7c91ea355abb35b9cc6088ee14b40b0922b Reviewed-on: https://chromium-review.googlesource.com/413037 Commit-Queue: Olli Etuaho <oetuaho@nvidia.com> Reviewed-by: Jamie Madill <jmadill@chromium.org> Reviewed-by: Geoff Lang <geofflang@chromium.org> Reviewed-by: Corentin Wallez <cwallez@chromium.org>

  • translator_fuzzer.cpp
  • //
    // Copyright (c) 2016 The ANGLE Project Authors. All rights reserved.
    // Use of this source code is governed by a BSD-style license that can be
    // found in the LICENSE file.
    //
    
    // translator_fuzzer.cpp: A libfuzzer fuzzer for the shader translator.
    
    #include <stddef.h>
    #include <stdint.h>
    #include <unordered_map>
    #include <iostream>
    
    #include "compiler/translator/Compiler.h"
    #include "angle_gl.h"
    
    using namespace sh;
    
    struct TranslatorCacheKey
    {
        bool operator==(const TranslatorCacheKey &other) const
        {
            return type == other.type && spec == other.spec && output == other.output;
        }
    
        uint32_t type   = 0;
        uint32_t spec   = 0;
        uint32_t output = 0;
    };
    
    namespace std
    {
    
    template <>
    struct hash<TranslatorCacheKey>
    {
        std::size_t operator()(const TranslatorCacheKey &k) const
        {
            return (hash<uint32_t>()(k.type) << 1) ^ (hash<uint32_t>()(k.spec) >> 1) ^
                   hash<uint32_t>()(k.output);
        }
    };
    }  // namespace std
    
    static std::unordered_map<TranslatorCacheKey, TCompiler *> translators;
    
    extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
    {
        // Reserve some size for future compile options
        const size_t kHeaderSize = 128;
    
        if (size <= kHeaderSize)
        {
            return 0;
        }
    
        // Make sure the rest of data will be a valid C string so that we don't have to copy it.
        if (data[size - 1] != 0)
        {
            return 0;
        }
    
        uint32_t type    = *reinterpret_cast<const uint32_t *>(data);
        uint32_t spec    = *reinterpret_cast<const uint32_t *>(data + 4);
        uint32_t output  = *reinterpret_cast<const uint32_t *>(data + 8);
        uint64_t options = *reinterpret_cast<const uint64_t *>(data + 12);
    
        if (type != GL_FRAGMENT_SHADER && type != GL_VERTEX_SHADER)
        {
            return 0;
        }
    
        if (spec != SH_GLES2_SPEC && type != SH_WEBGL_SPEC && spec != SH_GLES3_SPEC &&
            spec != SH_WEBGL2_SPEC)
        {
            return 0;
        }
    
        std::vector<uint32_t> validOutputs;
        validOutputs.push_back(SH_ESSL_OUTPUT);
        validOutputs.push_back(SH_GLSL_COMPATIBILITY_OUTPUT);
        validOutputs.push_back(SH_GLSL_130_OUTPUT);
        validOutputs.push_back(SH_GLSL_140_OUTPUT);
        validOutputs.push_back(SH_GLSL_150_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_330_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_400_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_410_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_420_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_430_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_440_CORE_OUTPUT);
        validOutputs.push_back(SH_GLSL_450_CORE_OUTPUT);
        validOutputs.push_back(SH_HLSL_3_0_OUTPUT);
        validOutputs.push_back(SH_HLSL_4_1_OUTPUT);
        validOutputs.push_back(SH_HLSL_4_0_FL9_3_OUTPUT);
        bool found = false;
        for (auto valid : validOutputs)
        {
            found = found || (valid == output);
        }
        if (!found)
        {
            return 0;
        }
    
        size -= kHeaderSize;
        data += kHeaderSize;
    
        if (!ShInitialize())
        {
            return 0;
        }
    
        TranslatorCacheKey key;
        key.type   = type;
        key.spec   = spec;
        key.output = output;
    
        if (translators.find(key) == translators.end())
        {
            TCompiler *translator = ConstructCompiler(type, static_cast<ShShaderSpec>(spec),
                                                      static_cast<ShShaderOutput>(output));
    
            if (!translator)
            {
                return 0;
            }
    
            ShBuiltInResources resources;
            ShInitBuiltInResources(&resources);
    
            // Enable all the extensions to have more coverage
            resources.OES_standard_derivatives        = 1;
            resources.OES_EGL_image_external          = 1;
            resources.OES_EGL_image_external_essl3    = 1;
            resources.NV_EGL_stream_consumer_external = 1;
            resources.ARB_texture_rectangle           = 1;
            resources.EXT_blend_func_extended         = 1;
            resources.EXT_draw_buffers                = 1;
            resources.EXT_frag_depth                  = 1;
            resources.EXT_shader_texture_lod          = 1;
            resources.WEBGL_debug_shader_precision    = 1;
            resources.EXT_shader_framebuffer_fetch    = 1;
            resources.NV_shader_framebuffer_fetch     = 1;
            resources.ARM_shader_framebuffer_fetch    = 1;
            resources.MaxDualSourceDrawBuffers        = 1;
    
            if (!translator->Init(resources))
            {
                DeleteCompiler(translator);
                return 0;
            }
    
            translators[key] = translator;
        }
    
        TCompiler *translator = translators[key];
    
        const char *shaderStrings[] = {reinterpret_cast<const char *>(data)};
        translator->compile(shaderStrings, 1, options);
    
        return 0;
    }