| // Copyright (c) 2020 André Perez Maselco |
| // |
| // 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 "source/fuzz/transformation_inline_function.h" |
| |
| #include "gtest/gtest.h" |
| #include "source/fuzz/counter_overflow_id_source.h" |
| #include "source/fuzz/fuzzer_util.h" |
| #include "source/fuzz/instruction_descriptor.h" |
| #include "test/fuzz/fuzz_test_util.h" |
| |
| namespace spvtools { |
| namespace fuzz { |
| namespace { |
| |
| TEST(TransformationInlineFunctionTest, IsApplicable) { |
| std::string shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %52 "main" |
| OpExecutionMode %52 OriginUpperLeft |
| OpName %56 "function_with_void_return" |
| |
| ; Types |
| %2 = OpTypeBool |
| %3 = OpTypeFloat 32 |
| %4 = OpTypeVector %3 4 |
| %5 = OpTypePointer Function %4 |
| %6 = OpTypeVoid |
| %7 = OpTypeFunction %6 |
| %8 = OpTypeFunction %3 %5 %5 |
| |
| ; Constant scalars |
| %9 = OpConstant %3 1 |
| %10 = OpConstant %3 2 |
| %11 = OpConstant %3 3 |
| %12 = OpConstant %3 4 |
| %13 = OpConstant %3 5 |
| %14 = OpConstant %3 6 |
| %15 = OpConstant %3 7 |
| %16 = OpConstant %3 8 |
| %17 = OpConstantTrue %2 |
| |
| ; Constant vectors |
| %18 = OpConstantComposite %4 %9 %10 %11 %12 |
| %19 = OpConstantComposite %4 %13 %14 %15 %16 |
| |
| ; function with void return |
| %20 = OpFunction %6 None %7 |
| %21 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| |
| ; function with early return |
| %22 = OpFunction %6 None %7 |
| %23 = OpLabel |
| OpSelectionMerge %26 None |
| OpBranchConditional %17 %24 %25 |
| %24 = OpLabel |
| OpReturn |
| %25 = OpLabel |
| OpBranch %26 |
| %26 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| |
| ; function containing an OpKill instruction |
| %27 = OpFunction %6 None %7 |
| %28 = OpLabel |
| OpKill |
| OpFunctionEnd |
| |
| ; function containing an OpUnreachable instruction |
| %29 = OpFunction %6 None %7 |
| %30 = OpLabel |
| OpUnreachable |
| OpFunctionEnd |
| |
| ; dot product function |
| %31 = OpFunction %3 None %8 |
| %32 = OpFunctionParameter %5 |
| %33 = OpFunctionParameter %5 |
| %34 = OpLabel |
| %35 = OpLoad %4 %32 |
| %36 = OpLoad %4 %33 |
| %37 = OpCompositeExtract %3 %35 0 |
| %38 = OpCompositeExtract %3 %36 0 |
| %39 = OpFMul %3 %37 %38 |
| %40 = OpCompositeExtract %3 %35 1 |
| %41 = OpCompositeExtract %3 %36 1 |
| %42 = OpFMul %3 %40 %41 |
| %43 = OpCompositeExtract %3 %35 2 |
| %44 = OpCompositeExtract %3 %36 2 |
| %45 = OpFMul %3 %43 %44 |
| %46 = OpCompositeExtract %3 %35 3 |
| %47 = OpCompositeExtract %3 %36 3 |
| %48 = OpFMul %3 %46 %47 |
| %49 = OpFAdd %3 %39 %42 |
| %50 = OpFAdd %3 %45 %49 |
| %51 = OpFAdd %3 %48 %50 |
| OpReturnValue %51 |
| OpFunctionEnd |
| |
| ; main function |
| %52 = OpFunction %6 None %7 |
| %53 = OpLabel |
| %54 = OpVariable %5 Function |
| %55 = OpVariable %5 Function |
| %56 = OpFunctionCall %6 %20 ; function with void return |
| OpBranch %57 |
| %57 = OpLabel |
| %59 = OpFunctionCall %6 %22 ; function with early return |
| OpBranch %60 |
| %60 = OpLabel |
| %61 = OpFunctionCall %6 %27 ; function containing OpKill |
| OpBranch %62 |
| %62 = OpLabel |
| %63 = OpFunctionCall %6 %29 ; function containing OpUnreachable |
| OpBranch %64 |
| %64 = OpLabel |
| OpStore %54 %18 |
| OpStore %55 %19 |
| %65 = OpFunctionCall %3 %31 %54 %55 ; dot product function |
| OpBranch %66 |
| %66 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_5; |
| const auto consumer = nullptr; |
| const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options); |
| // Tests undefined OpFunctionCall instruction. |
| auto transformation = TransformationInlineFunction(67, {}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests false OpFunctionCall instruction. |
| transformation = TransformationInlineFunction(42, {}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests use of called function with void return. |
| transformation = TransformationInlineFunction(56, {}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests called function having an early return. |
| transformation = |
| TransformationInlineFunction(59, {{24, 67}, {25, 68}, {26, 69}}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests called function containing an OpKill instruction. |
| transformation = TransformationInlineFunction(61, {}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests called function containing an OpUnreachable instruction. |
| transformation = TransformationInlineFunction(63, {}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| // Tests applicable transformation. |
| transformation = TransformationInlineFunction(65, {{35, 67}, |
| {36, 68}, |
| {37, 69}, |
| {38, 70}, |
| {39, 71}, |
| {40, 72}, |
| {41, 73}, |
| {42, 74}, |
| {43, 75}, |
| {44, 76}, |
| {45, 77}, |
| {46, 78}, |
| {47, 79}, |
| {48, 80}, |
| {49, 81}, |
| {50, 82}, |
| {51, 83}}); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| } |
| |
| TEST(TransformationInlineFunctionTest, Apply) { |
| std::string reference_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %39 "main" |
| |
| ; Types |
| %2 = OpTypeFloat 32 |
| %3 = OpTypeVector %2 4 |
| %4 = OpTypePointer Function %3 |
| %5 = OpTypeVoid |
| %6 = OpTypeFunction %5 |
| %7 = OpTypeFunction %2 %4 %4 |
| |
| ; Constant scalars |
| %8 = OpConstant %2 1 |
| %9 = OpConstant %2 2 |
| %10 = OpConstant %2 3 |
| %11 = OpConstant %2 4 |
| %12 = OpConstant %2 5 |
| %13 = OpConstant %2 6 |
| %14 = OpConstant %2 7 |
| %15 = OpConstant %2 8 |
| |
| ; Constant vectors |
| %16 = OpConstantComposite %3 %8 %9 %10 %11 |
| %17 = OpConstantComposite %3 %12 %13 %14 %15 |
| |
| ; dot product function |
| %18 = OpFunction %2 None %7 |
| %19 = OpFunctionParameter %4 |
| %20 = OpFunctionParameter %4 |
| %21 = OpLabel |
| %22 = OpLoad %3 %19 |
| %23 = OpLoad %3 %20 |
| %24 = OpCompositeExtract %2 %22 0 |
| %25 = OpCompositeExtract %2 %23 0 |
| %26 = OpFMul %2 %24 %25 |
| %27 = OpCompositeExtract %2 %22 1 |
| %28 = OpCompositeExtract %2 %23 1 |
| %29 = OpFMul %2 %27 %28 |
| %30 = OpCompositeExtract %2 %22 2 |
| %31 = OpCompositeExtract %2 %23 2 |
| %32 = OpFMul %2 %30 %31 |
| %33 = OpCompositeExtract %2 %22 3 |
| %34 = OpCompositeExtract %2 %23 3 |
| %35 = OpFMul %2 %33 %34 |
| %36 = OpFAdd %2 %26 %29 |
| %37 = OpFAdd %2 %32 %36 |
| %38 = OpFAdd %2 %35 %37 |
| OpReturnValue %38 |
| OpFunctionEnd |
| |
| ; main function |
| %39 = OpFunction %5 None %6 |
| %40 = OpLabel |
| %41 = OpVariable %4 Function |
| %42 = OpVariable %4 Function |
| OpStore %41 %16 |
| OpStore %42 %17 |
| %43 = OpFunctionCall %2 %18 %41 %42 ; dot product function call |
| OpBranch %44 |
| %44 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_5; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, reference_shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options); |
| auto transformation = TransformationInlineFunction(43, {{22, 45}, |
| {23, 46}, |
| {24, 47}, |
| {25, 48}, |
| {26, 49}, |
| {27, 50}, |
| {28, 51}, |
| {29, 52}, |
| {30, 53}, |
| {31, 54}, |
| {32, 55}, |
| {33, 56}, |
| {34, 57}, |
| {35, 58}, |
| {36, 59}, |
| {37, 60}, |
| {38, 61}}); |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| |
| std::string variant_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %39 "main" |
| |
| ; Types |
| %2 = OpTypeFloat 32 |
| %3 = OpTypeVector %2 4 |
| %4 = OpTypePointer Function %3 |
| %5 = OpTypeVoid |
| %6 = OpTypeFunction %5 |
| %7 = OpTypeFunction %2 %4 %4 |
| |
| ; Constant scalars |
| %8 = OpConstant %2 1 |
| %9 = OpConstant %2 2 |
| %10 = OpConstant %2 3 |
| %11 = OpConstant %2 4 |
| %12 = OpConstant %2 5 |
| %13 = OpConstant %2 6 |
| %14 = OpConstant %2 7 |
| %15 = OpConstant %2 8 |
| |
| ; Constant vectors |
| %16 = OpConstantComposite %3 %8 %9 %10 %11 |
| %17 = OpConstantComposite %3 %12 %13 %14 %15 |
| |
| ; dot product function |
| %18 = OpFunction %2 None %7 |
| %19 = OpFunctionParameter %4 |
| %20 = OpFunctionParameter %4 |
| %21 = OpLabel |
| %22 = OpLoad %3 %19 |
| %23 = OpLoad %3 %20 |
| %24 = OpCompositeExtract %2 %22 0 |
| %25 = OpCompositeExtract %2 %23 0 |
| %26 = OpFMul %2 %24 %25 |
| %27 = OpCompositeExtract %2 %22 1 |
| %28 = OpCompositeExtract %2 %23 1 |
| %29 = OpFMul %2 %27 %28 |
| %30 = OpCompositeExtract %2 %22 2 |
| %31 = OpCompositeExtract %2 %23 2 |
| %32 = OpFMul %2 %30 %31 |
| %33 = OpCompositeExtract %2 %22 3 |
| %34 = OpCompositeExtract %2 %23 3 |
| %35 = OpFMul %2 %33 %34 |
| %36 = OpFAdd %2 %26 %29 |
| %37 = OpFAdd %2 %32 %36 |
| %38 = OpFAdd %2 %35 %37 |
| OpReturnValue %38 |
| OpFunctionEnd |
| |
| ; main function |
| %39 = OpFunction %5 None %6 |
| %40 = OpLabel |
| %41 = OpVariable %4 Function |
| %42 = OpVariable %4 Function |
| OpStore %41 %16 |
| OpStore %42 %17 |
| %45 = OpLoad %3 %41 |
| %46 = OpLoad %3 %42 |
| %47 = OpCompositeExtract %2 %45 0 |
| %48 = OpCompositeExtract %2 %46 0 |
| %49 = OpFMul %2 %47 %48 |
| %50 = OpCompositeExtract %2 %45 1 |
| %51 = OpCompositeExtract %2 %46 1 |
| %52 = OpFMul %2 %50 %51 |
| %53 = OpCompositeExtract %2 %45 2 |
| %54 = OpCompositeExtract %2 %46 2 |
| %55 = OpFMul %2 %53 %54 |
| %56 = OpCompositeExtract %2 %45 3 |
| %57 = OpCompositeExtract %2 %46 3 |
| %58 = OpFMul %2 %56 %57 |
| %59 = OpFAdd %2 %49 %52 |
| %60 = OpFAdd %2 %55 %59 |
| %61 = OpFAdd %2 %58 %60 |
| %43 = OpCopyObject %2 %61 |
| OpBranch %44 |
| %44 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| ASSERT_TRUE(IsEqual(env, variant_shader, context.get())); |
| } |
| |
| TEST(TransformationInlineFunctionTest, ApplyToMultipleFunctions) { |
| std::string reference_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %15 "main" |
| |
| ; Types |
| %2 = OpTypeInt 32 1 |
| %3 = OpTypeBool |
| %4 = OpTypePointer Private %2 |
| %5 = OpTypePointer Function %2 |
| %6 = OpTypeVoid |
| %7 = OpTypeFunction %6 |
| %8 = OpTypeFunction %2 %5 |
| %9 = OpTypeFunction %2 %2 |
| |
| ; Constants |
| %10 = OpConstant %2 0 |
| %11 = OpConstant %2 1 |
| %12 = OpConstant %2 2 |
| %13 = OpConstant %2 3 |
| |
| ; Global variable |
| %14 = OpVariable %4 Private |
| |
| ; main function |
| %15 = OpFunction %6 None %7 |
| %16 = OpLabel |
| %17 = OpVariable %5 Function |
| %18 = OpVariable %5 Function |
| %19 = OpVariable %5 Function |
| OpStore %17 %13 |
| %20 = OpLoad %2 %17 |
| OpStore %18 %20 |
| %21 = OpFunctionCall %2 %36 %18 |
| OpBranch %22 |
| %22 = OpLabel |
| %23 = OpFunctionCall %2 %36 %18 |
| OpStore %17 %21 |
| %24 = OpLoad %2 %17 |
| %25 = OpFunctionCall %2 %54 %24 |
| OpBranch %26 |
| %26 = OpLabel |
| %27 = OpFunctionCall %2 %54 %24 |
| %28 = OpLoad %2 %17 |
| %29 = OpIAdd %2 %28 %25 |
| OpStore %17 %29 |
| %30 = OpFunctionCall %6 %67 |
| OpBranch %31 |
| %31 = OpLabel |
| %32 = OpFunctionCall %6 %67 |
| %33 = OpLoad %2 %14 |
| %34 = OpLoad %2 %17 |
| %35 = OpIAdd %2 %34 %33 |
| OpStore %17 %35 |
| OpReturn |
| OpFunctionEnd |
| |
| ; Function %36 |
| %36 = OpFunction %2 None %8 |
| %37 = OpFunctionParameter %5 |
| %38 = OpLabel |
| %39 = OpVariable %5 Function |
| %40 = OpVariable %5 Function |
| OpStore %39 %10 |
| OpBranch %41 |
| %41 = OpLabel |
| OpLoopMerge %52 %49 None |
| OpBranch %42 |
| %42 = OpLabel |
| %43 = OpLoad %2 %39 |
| %44 = OpLoad %2 %37 |
| %45 = OpSLessThan %3 %43 %44 |
| OpBranchConditional %45 %46 %52 |
| %46 = OpLabel |
| %47 = OpLoad %2 %40 |
| %48 = OpIAdd %2 %47 %11 |
| OpStore %40 %48 |
| OpBranch %49 |
| %49 = OpLabel |
| %50 = OpLoad %2 %39 |
| %51 = OpIAdd %2 %50 %12 |
| OpStore %39 %51 |
| OpBranch %41 |
| %52 = OpLabel |
| %53 = OpLoad %2 %40 |
| OpReturnValue %53 |
| OpFunctionEnd |
| |
| ; Function %54 |
| %54 = OpFunction %2 None %9 |
| %55 = OpFunctionParameter %2 |
| %56 = OpLabel |
| %57 = OpVariable %5 Function |
| OpStore %57 %10 |
| %58 = OpSGreaterThan %3 %55 %10 |
| OpSelectionMerge %62 None |
| OpBranchConditional %58 %64 %59 |
| %59 = OpLabel |
| %60 = OpLoad %2 %57 |
| %61 = OpISub %2 %60 %12 |
| OpStore %57 %61 |
| OpBranch %62 |
| %62 = OpLabel |
| %63 = OpLoad %2 %57 |
| OpReturnValue %63 |
| %64 = OpLabel |
| %65 = OpLoad %2 %57 |
| %66 = OpIAdd %2 %65 %11 |
| OpStore %57 %66 |
| OpBranch %62 |
| OpFunctionEnd |
| |
| ; Function %67 |
| %67 = OpFunction %6 None %7 |
| %68 = OpLabel |
| OpStore %14 %12 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_3; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, reference_shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options); |
| auto transformation = TransformationInlineFunction(30, {}); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| |
| // Tests a parameter included in the id map. |
| transformation = TransformationInlineFunction(25, {{55, 69}, |
| {56, 70}, |
| {57, 71}, |
| {58, 72}, |
| {59, 73}, |
| {60, 74}, |
| {61, 75}, |
| {62, 76}, |
| {63, 77}, |
| {64, 78}, |
| {65, 79}, |
| {66, 80}}); |
| ASSERT_FALSE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| #ifndef NDEBUG |
| // Tests the id of the returned value not included in the id map. |
| transformation = TransformationInlineFunction(25, {{56, 69}, |
| {57, 70}, |
| {58, 71}, |
| {59, 72}, |
| {60, 73}, |
| {61, 74}, |
| {62, 75}, |
| {64, 76}, |
| {65, 77}, |
| {66, 78}}); |
| ASSERT_DEATH( |
| transformation.IsApplicable(context.get(), transformation_context), |
| "Bad attempt to query whether overflow ids are available."); |
| #endif |
| |
| transformation = TransformationInlineFunction(25, {{57, 69}, |
| {58, 70}, |
| {59, 71}, |
| {60, 72}, |
| {61, 73}, |
| {62, 74}, |
| {63, 75}, |
| {64, 76}, |
| {65, 77}, |
| {66, 78}}); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| |
| transformation = TransformationInlineFunction(21, {{39, 79}, |
| {40, 80}, |
| {41, 81}, |
| {42, 82}, |
| {43, 83}, |
| {44, 84}, |
| {45, 85}, |
| {46, 86}, |
| {47, 87}, |
| {48, 88}, |
| {49, 89}, |
| {50, 90}, |
| {51, 91}, |
| {52, 92}, |
| {53, 93}}); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| |
| std::string variant_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %15 "main" |
| |
| ; Types |
| %2 = OpTypeInt 32 1 |
| %3 = OpTypeBool |
| %4 = OpTypePointer Private %2 |
| %5 = OpTypePointer Function %2 |
| %6 = OpTypeVoid |
| %7 = OpTypeFunction %6 |
| %8 = OpTypeFunction %2 %5 |
| %9 = OpTypeFunction %2 %2 |
| |
| ; Constants |
| %10 = OpConstant %2 0 |
| %11 = OpConstant %2 1 |
| %12 = OpConstant %2 2 |
| %13 = OpConstant %2 3 |
| |
| ; Global variable |
| %14 = OpVariable %4 Private |
| |
| ; main function |
| %15 = OpFunction %6 None %7 |
| %16 = OpLabel |
| %80 = OpVariable %5 Function |
| %79 = OpVariable %5 Function |
| %69 = OpVariable %5 Function |
| %17 = OpVariable %5 Function |
| %18 = OpVariable %5 Function |
| %19 = OpVariable %5 Function |
| OpStore %17 %13 |
| %20 = OpLoad %2 %17 |
| OpStore %18 %20 |
| OpStore %79 %10 |
| OpBranch %81 |
| %81 = OpLabel |
| OpLoopMerge %92 %89 None |
| OpBranch %82 |
| %82 = OpLabel |
| %83 = OpLoad %2 %79 |
| %84 = OpLoad %2 %18 |
| %85 = OpSLessThan %3 %83 %84 |
| OpBranchConditional %85 %86 %92 |
| %86 = OpLabel |
| %87 = OpLoad %2 %80 |
| %88 = OpIAdd %2 %87 %11 |
| OpStore %80 %88 |
| OpBranch %89 |
| %89 = OpLabel |
| %90 = OpLoad %2 %79 |
| %91 = OpIAdd %2 %90 %12 |
| OpStore %79 %91 |
| OpBranch %81 |
| %92 = OpLabel |
| %93 = OpLoad %2 %80 |
| %21 = OpCopyObject %2 %93 |
| OpBranch %22 |
| %22 = OpLabel |
| %23 = OpFunctionCall %2 %36 %18 |
| OpStore %17 %21 |
| %24 = OpLoad %2 %17 |
| OpStore %69 %10 |
| %70 = OpSGreaterThan %3 %24 %10 |
| OpSelectionMerge %74 None |
| OpBranchConditional %70 %76 %71 |
| %71 = OpLabel |
| %72 = OpLoad %2 %69 |
| %73 = OpISub %2 %72 %12 |
| OpStore %69 %73 |
| OpBranch %74 |
| %74 = OpLabel |
| %75 = OpLoad %2 %69 |
| %25 = OpCopyObject %2 %75 |
| OpBranch %26 |
| %76 = OpLabel |
| %77 = OpLoad %2 %69 |
| %78 = OpIAdd %2 %77 %11 |
| OpStore %69 %78 |
| OpBranch %74 |
| %26 = OpLabel |
| %27 = OpFunctionCall %2 %54 %24 |
| %28 = OpLoad %2 %17 |
| %29 = OpIAdd %2 %28 %25 |
| OpStore %17 %29 |
| OpStore %14 %12 |
| OpBranch %31 |
| %31 = OpLabel |
| %32 = OpFunctionCall %6 %67 |
| %33 = OpLoad %2 %14 |
| %34 = OpLoad %2 %17 |
| %35 = OpIAdd %2 %34 %33 |
| OpStore %17 %35 |
| OpReturn |
| OpFunctionEnd |
| |
| ; Function %36 |
| %36 = OpFunction %2 None %8 |
| %37 = OpFunctionParameter %5 |
| %38 = OpLabel |
| %39 = OpVariable %5 Function |
| %40 = OpVariable %5 Function |
| OpStore %39 %10 |
| OpBranch %41 |
| %41 = OpLabel |
| OpLoopMerge %52 %49 None |
| OpBranch %42 |
| %42 = OpLabel |
| %43 = OpLoad %2 %39 |
| %44 = OpLoad %2 %37 |
| %45 = OpSLessThan %3 %43 %44 |
| OpBranchConditional %45 %46 %52 |
| %46 = OpLabel |
| %47 = OpLoad %2 %40 |
| %48 = OpIAdd %2 %47 %11 |
| OpStore %40 %48 |
| OpBranch %49 |
| %49 = OpLabel |
| %50 = OpLoad %2 %39 |
| %51 = OpIAdd %2 %50 %12 |
| OpStore %39 %51 |
| OpBranch %41 |
| %52 = OpLabel |
| %53 = OpLoad %2 %40 |
| OpReturnValue %53 |
| OpFunctionEnd |
| |
| ; Function %54 |
| %54 = OpFunction %2 None %9 |
| %55 = OpFunctionParameter %2 |
| %56 = OpLabel |
| %57 = OpVariable %5 Function |
| OpStore %57 %10 |
| %58 = OpSGreaterThan %3 %55 %10 |
| OpSelectionMerge %62 None |
| OpBranchConditional %58 %64 %59 |
| %59 = OpLabel |
| %60 = OpLoad %2 %57 |
| %61 = OpISub %2 %60 %12 |
| OpStore %57 %61 |
| OpBranch %62 |
| %62 = OpLabel |
| %63 = OpLoad %2 %57 |
| OpReturnValue %63 |
| %64 = OpLabel |
| %65 = OpLoad %2 %57 |
| %66 = OpIAdd %2 %65 %11 |
| OpStore %57 %66 |
| OpBranch %62 |
| OpFunctionEnd |
| |
| ; Function %67 |
| %67 = OpFunction %6 None %7 |
| %68 = OpLabel |
| OpStore %14 %12 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| ASSERT_TRUE(IsEqual(env, variant_shader, context.get())); |
| } |
| |
| TEST(TransformationInlineFunctionTest, HandlesOpPhisInTheSecondBlock) { |
| std::string shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" |
| OpExecutionMode %4 OriginUpperLeft |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %10 = OpTypeInt 32 0 |
| %11 = OpUndef %10 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %6 = OpFunctionCall %2 %7 |
| OpBranch %14 |
| %14 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| %7 = OpFunction %2 None %3 |
| %8 = OpLabel |
| OpBranch %13 |
| %13 = OpLabel |
| %12 = OpPhi %10 %11 %8 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_3; |
| const auto consumer = nullptr; |
| const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options); |
| TransformationInlineFunction transformation(6, |
| {{{8, 20}, {13, 21}, {12, 22}}}); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| |
| std::string after_transformation = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" |
| OpExecutionMode %4 OriginUpperLeft |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %10 = OpTypeInt 32 0 |
| %11 = OpUndef %10 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpBranch %21 |
| %21 = OpLabel |
| %22 = OpPhi %10 %11 %5 |
| OpBranch %14 |
| %14 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| %7 = OpFunction %2 None %3 |
| %8 = OpLabel |
| OpBranch %13 |
| %13 = OpLabel |
| %12 = OpPhi %10 %11 %8 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationInlineFunctionTest, OverflowIds) { |
| std::string reference_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %39 "main" |
| |
| ; Types |
| %2 = OpTypeFloat 32 |
| %3 = OpTypeVector %2 4 |
| %4 = OpTypePointer Function %3 |
| %5 = OpTypeVoid |
| %6 = OpTypeFunction %5 |
| %7 = OpTypeFunction %2 %4 %4 |
| |
| ; Constant scalars |
| %8 = OpConstant %2 1 |
| %9 = OpConstant %2 2 |
| %10 = OpConstant %2 3 |
| %11 = OpConstant %2 4 |
| %12 = OpConstant %2 5 |
| %13 = OpConstant %2 6 |
| %14 = OpConstant %2 7 |
| %15 = OpConstant %2 8 |
| |
| ; Constant vectors |
| %16 = OpConstantComposite %3 %8 %9 %10 %11 |
| %17 = OpConstantComposite %3 %12 %13 %14 %15 |
| |
| ; dot product function |
| %18 = OpFunction %2 None %7 |
| %19 = OpFunctionParameter %4 |
| %20 = OpFunctionParameter %4 |
| %21 = OpLabel |
| %22 = OpLoad %3 %19 |
| %23 = OpLoad %3 %20 |
| %24 = OpCompositeExtract %2 %22 0 |
| %25 = OpCompositeExtract %2 %23 0 |
| %26 = OpFMul %2 %24 %25 |
| %27 = OpCompositeExtract %2 %22 1 |
| %28 = OpCompositeExtract %2 %23 1 |
| %29 = OpFMul %2 %27 %28 |
| OpBranch %100 |
| %100 = OpLabel |
| %30 = OpCompositeExtract %2 %22 2 |
| %31 = OpCompositeExtract %2 %23 2 |
| %32 = OpFMul %2 %30 %31 |
| %33 = OpCompositeExtract %2 %22 3 |
| %34 = OpCompositeExtract %2 %23 3 |
| %35 = OpFMul %2 %33 %34 |
| %36 = OpFAdd %2 %26 %29 |
| %37 = OpFAdd %2 %32 %36 |
| %38 = OpFAdd %2 %35 %37 |
| OpReturnValue %38 |
| OpFunctionEnd |
| |
| ; main function |
| %39 = OpFunction %5 None %6 |
| %40 = OpLabel |
| %41 = OpVariable %4 Function |
| %42 = OpVariable %4 Function |
| OpStore %41 %16 |
| OpStore %42 %17 |
| %43 = OpFunctionCall %2 %18 %41 %42 ; dot product function call |
| OpBranch %44 |
| %44 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_5; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, reference_shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| auto overflow_ids_unique_ptr = MakeUnique<CounterOverflowIdSource>(1000); |
| auto overflow_ids_ptr = overflow_ids_unique_ptr.get(); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options, |
| std::move(overflow_ids_unique_ptr)); |
| auto transformation = TransformationInlineFunction(43, {{22, 45}, |
| {23, 46}, |
| {24, 47}, |
| {25, 48}, |
| {26, 49}, |
| {27, 50}, |
| {28, 51}, |
| {29, 52}}); |
| |
| // The following ids are left un-mapped; overflow ids will be required for |
| // them: 30, 31, 32, 33, 34, 35, 36, 37, 38, 100 |
| |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context, |
| overflow_ids_ptr->GetIssuedOverflowIds()); |
| |
| std::string variant_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %39 "main" |
| |
| ; Types |
| %2 = OpTypeFloat 32 |
| %3 = OpTypeVector %2 4 |
| %4 = OpTypePointer Function %3 |
| %5 = OpTypeVoid |
| %6 = OpTypeFunction %5 |
| %7 = OpTypeFunction %2 %4 %4 |
| |
| ; Constant scalars |
| %8 = OpConstant %2 1 |
| %9 = OpConstant %2 2 |
| %10 = OpConstant %2 3 |
| %11 = OpConstant %2 4 |
| %12 = OpConstant %2 5 |
| %13 = OpConstant %2 6 |
| %14 = OpConstant %2 7 |
| %15 = OpConstant %2 8 |
| |
| ; Constant vectors |
| %16 = OpConstantComposite %3 %8 %9 %10 %11 |
| %17 = OpConstantComposite %3 %12 %13 %14 %15 |
| |
| ; dot product function |
| %18 = OpFunction %2 None %7 |
| %19 = OpFunctionParameter %4 |
| %20 = OpFunctionParameter %4 |
| %21 = OpLabel |
| %22 = OpLoad %3 %19 |
| %23 = OpLoad %3 %20 |
| %24 = OpCompositeExtract %2 %22 0 |
| %25 = OpCompositeExtract %2 %23 0 |
| %26 = OpFMul %2 %24 %25 |
| %27 = OpCompositeExtract %2 %22 1 |
| %28 = OpCompositeExtract %2 %23 1 |
| %29 = OpFMul %2 %27 %28 |
| OpBranch %100 |
| %100 = OpLabel |
| %30 = OpCompositeExtract %2 %22 2 |
| %31 = OpCompositeExtract %2 %23 2 |
| %32 = OpFMul %2 %30 %31 |
| %33 = OpCompositeExtract %2 %22 3 |
| %34 = OpCompositeExtract %2 %23 3 |
| %35 = OpFMul %2 %33 %34 |
| %36 = OpFAdd %2 %26 %29 |
| %37 = OpFAdd %2 %32 %36 |
| %38 = OpFAdd %2 %35 %37 |
| OpReturnValue %38 |
| OpFunctionEnd |
| |
| ; main function |
| %39 = OpFunction %5 None %6 |
| %40 = OpLabel |
| %41 = OpVariable %4 Function |
| %42 = OpVariable %4 Function |
| OpStore %41 %16 |
| OpStore %42 %17 |
| %45 = OpLoad %3 %41 |
| %46 = OpLoad %3 %42 |
| %47 = OpCompositeExtract %2 %45 0 |
| %48 = OpCompositeExtract %2 %46 0 |
| %49 = OpFMul %2 %47 %48 |
| %50 = OpCompositeExtract %2 %45 1 |
| %51 = OpCompositeExtract %2 %46 1 |
| %52 = OpFMul %2 %50 %51 |
| OpBranch %1000 |
| %1000 = OpLabel |
| %1001 = OpCompositeExtract %2 %45 2 |
| %1002 = OpCompositeExtract %2 %46 2 |
| %1003 = OpFMul %2 %1001 %1002 |
| %1004 = OpCompositeExtract %2 %45 3 |
| %1005 = OpCompositeExtract %2 %46 3 |
| %1006 = OpFMul %2 %1004 %1005 |
| %1007 = OpFAdd %2 %49 %52 |
| %1008 = OpFAdd %2 %1003 %1007 |
| %1009 = OpFAdd %2 %1006 %1008 |
| %43 = OpCopyObject %2 %1009 |
| OpBranch %44 |
| %44 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| ASSERT_TRUE(IsEqual(env, variant_shader, context.get())); |
| } |
| |
| TEST(TransformationInlineFunctionTest, OpPhiInBlockFollowingCall) { |
| // This test checks that if the block after the inlined function call has an |
| // OpPhi instruction and the called function contains multiple blocks then the |
| // OpPhi instruction gets updated to refer to the return block of the inlined |
| // function, since the block containing the call will no longer be a |
| // predecessor. |
| |
| std::string reference_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" |
| OpExecutionMode %4 OriginUpperLeft |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %13 = OpTypeBool |
| %14 = OpConstantTrue %13 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpBranch %10 |
| %10 = OpLabel |
| %8 = OpFunctionCall %2 %6 |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpPhi %13 %14 %10 |
| OpReturn |
| OpFunctionEnd |
| %6 = OpFunction %2 None %3 |
| %7 = OpLabel |
| OpBranch %20 |
| %20 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_5; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, reference_shader, kFuzzAssembleOption); |
| spvtools::ValidatorOptions validator_options; |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| TransformationContext transformation_context( |
| MakeUnique<FactManager>(context.get()), validator_options); |
| auto transformation = TransformationInlineFunction(8, {{7, 100}, {20, 101}}); |
| |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| |
| ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context); |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| |
| std::string variant_shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" |
| OpExecutionMode %4 OriginUpperLeft |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %13 = OpTypeBool |
| %14 = OpConstantTrue %13 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpBranch %10 |
| %10 = OpLabel |
| OpBranch %101 |
| %101 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpPhi %13 %14 %101 |
| OpReturn |
| OpFunctionEnd |
| %6 = OpFunction %2 None %3 |
| %7 = OpLabel |
| OpBranch %20 |
| %20 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| ASSERT_TRUE(IsEqual(env, variant_shader, context.get())); |
| } |
| |
| } // namespace |
| } // namespace fuzz |
| } // namespace spvtools |
