| // Copyright (c) 2020 Vasyl Teliman | |
| // | |
| // 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_permute_function_parameters.h" | |
| #include "gtest/gtest.h" | |
| #include "source/fuzz/fuzzer_util.h" | |
| #include "test/fuzz/fuzz_test_util.h" | |
| namespace spvtools { | |
| namespace fuzz { | |
| namespace { | |
| TEST(TransformationPermuteFunctionParametersTest, BasicTest) { | |
| std::string shader = R"( | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %4 "main" %72 %74 | |
| OpExecutionMode %4 OriginUpperLeft | |
| OpSource ESSL 310 | |
| OpName %4 "main" | |
| OpName %12 "g(f1;f1;" | |
| OpName %10 "x" | |
| OpName %11 "y" | |
| OpName %22 "f(f1;i1;vf2;" | |
| OpName %19 "x" | |
| OpName %20 "y" | |
| OpName %21 "z" | |
| OpName %28 "cond(i1;f1;" | |
| OpName %26 "a" | |
| OpName %27 "b" | |
| OpName %53 "param" | |
| OpName %54 "param" | |
| OpName %66 "param" | |
| OpName %67 "param" | |
| OpName %72 "color" | |
| OpName %74 "gl_FragCoord" | |
| OpName %75 "param" | |
| OpName %79 "param" | |
| OpName %85 "param" | |
| OpName %86 "param" | |
| OpName %91 "param" | |
| OpName %92 "param" | |
| OpName %93 "param" | |
| OpName %99 "param" | |
| OpName %100 "param" | |
| OpName %101 "param" | |
| OpDecorate %20 RelaxedPrecision | |
| OpDecorate %26 RelaxedPrecision | |
| OpDecorate %47 RelaxedPrecision | |
| OpDecorate %58 RelaxedPrecision | |
| OpDecorate %72 Location 0 | |
| OpDecorate %74 BuiltIn FragCoord | |
| %2 = OpTypeVoid | |
| %3 = OpTypeFunction %2 | |
| %6 = OpTypeFloat 32 | |
| %7 = OpTypePointer Function %6 | |
| %8 = OpTypeVector %6 4 | |
| %9 = OpTypeFunction %8 %7 %7 | |
| %14 = OpTypeInt 32 1 | |
| %15 = OpTypePointer Function %14 | |
| %16 = OpTypeVector %6 2 | |
| %17 = OpTypePointer Function %16 | |
| %18 = OpTypeFunction %8 %7 %15 %17 | |
| %24 = OpTypeBool | |
| %25 = OpTypeFunction %24 %15 %7 | |
| %31 = OpConstant %6 255 | |
| %33 = OpConstant %6 0 | |
| %34 = OpConstant %6 1 | |
| %42 = OpTypeInt 32 0 | |
| %43 = OpConstant %42 0 | |
| %49 = OpConstant %42 1 | |
| %64 = OpConstant %14 4 | |
| %65 = OpConstant %6 5 | |
| %71 = OpTypePointer Output %8 | |
| %72 = OpVariable %71 Output | |
| %73 = OpTypePointer Input %8 | |
| %74 = OpVariable %73 Input | |
| %76 = OpTypePointer Input %6 | |
| %84 = OpConstant %14 5 | |
| %90 = OpConstant %6 3 | |
| %98 = OpConstant %6 4 | |
| %206 = OpTypeFunction %2 %14 %16 | |
| %223 = OpTypeFunction %2 %6 %8 | |
| %224 = OpTypeFunction %2 %8 %6 | |
| %233 = OpTypeFunction %2 %42 %24 | |
| %234 = OpTypeFunction %2 %24 %42 | |
| %4 = OpFunction %2 None %3 | |
| %5 = OpLabel | |
| %66 = OpVariable %15 Function | |
| %67 = OpVariable %7 Function | |
| %75 = OpVariable %7 Function | |
| %79 = OpVariable %7 Function | |
| %85 = OpVariable %15 Function | |
| %86 = OpVariable %7 Function | |
| %91 = OpVariable %7 Function | |
| %92 = OpVariable %15 Function | |
| %93 = OpVariable %17 Function | |
| %99 = OpVariable %7 Function | |
| %100 = OpVariable %15 Function | |
| %101 = OpVariable %17 Function | |
| OpStore %66 %64 | |
| OpStore %67 %65 | |
| %68 = OpFunctionCall %24 %28 %66 %67 | |
| OpSelectionMerge %70 None | |
| OpBranchConditional %68 %69 %83 | |
| %69 = OpLabel | |
| %77 = OpAccessChain %76 %74 %43 | |
| %78 = OpLoad %6 %77 | |
| OpStore %75 %78 | |
| %80 = OpAccessChain %76 %74 %49 | |
| %81 = OpLoad %6 %80 | |
| OpStore %79 %81 | |
| %82 = OpFunctionCall %8 %12 %75 %79 | |
| OpStore %72 %82 | |
| OpBranch %70 | |
| %83 = OpLabel | |
| OpStore %85 %84 | |
| OpStore %86 %65 | |
| %87 = OpFunctionCall %24 %28 %85 %86 | |
| OpSelectionMerge %89 None | |
| OpBranchConditional %87 %88 %97 | |
| %88 = OpLabel | |
| OpStore %91 %90 | |
| OpStore %92 %64 | |
| %94 = OpLoad %8 %74 | |
| %95 = OpVectorShuffle %16 %94 %94 0 1 | |
| OpStore %93 %95 | |
| %96 = OpFunctionCall %8 %22 %91 %92 %93 | |
| OpStore %72 %96 | |
| OpBranch %89 | |
| %97 = OpLabel | |
| OpStore %99 %98 | |
| OpStore %100 %84 | |
| %102 = OpLoad %8 %74 | |
| %103 = OpVectorShuffle %16 %102 %102 0 1 | |
| OpStore %101 %103 | |
| %104 = OpFunctionCall %8 %22 %99 %100 %101 | |
| OpStore %72 %104 | |
| OpBranch %89 | |
| %89 = OpLabel | |
| OpBranch %70 | |
| %70 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| ; adjust type of the function in-place | |
| %12 = OpFunction %8 None %9 | |
| %10 = OpFunctionParameter %7 | |
| %11 = OpFunctionParameter %7 | |
| %13 = OpLabel | |
| %30 = OpLoad %6 %10 | |
| %32 = OpFDiv %6 %30 %31 | |
| %35 = OpLoad %6 %11 | |
| %36 = OpFDiv %6 %35 %31 | |
| %37 = OpFSub %6 %34 %36 | |
| %38 = OpCompositeConstruct %8 %32 %33 %37 %34 | |
| OpReturnValue %38 | |
| OpFunctionEnd | |
| %22 = OpFunction %8 None %18 | |
| %19 = OpFunctionParameter %7 | |
| %20 = OpFunctionParameter %15 | |
| %21 = OpFunctionParameter %17 | |
| %23 = OpLabel | |
| %53 = OpVariable %7 Function | |
| %54 = OpVariable %7 Function | |
| %41 = OpLoad %6 %19 | |
| %44 = OpAccessChain %7 %21 %43 | |
| %45 = OpLoad %6 %44 | |
| %46 = OpFAdd %6 %41 %45 | |
| %47 = OpLoad %14 %20 | |
| %48 = OpConvertSToF %6 %47 | |
| %50 = OpAccessChain %7 %21 %49 | |
| %51 = OpLoad %6 %50 | |
| %52 = OpFAdd %6 %48 %51 | |
| OpStore %53 %46 | |
| OpStore %54 %52 | |
| %55 = OpFunctionCall %8 %12 %53 %54 | |
| OpReturnValue %55 | |
| OpFunctionEnd | |
| %28 = OpFunction %24 None %25 | |
| %26 = OpFunctionParameter %15 | |
| %27 = OpFunctionParameter %7 | |
| %29 = OpLabel | |
| %58 = OpLoad %14 %26 | |
| %59 = OpConvertSToF %6 %58 | |
| %60 = OpLoad %6 %27 | |
| %61 = OpFOrdLessThan %24 %59 %60 | |
| OpReturnValue %61 | |
| OpFunctionEnd | |
| ; create a new function type | |
| %200 = OpFunction %2 None %206 | |
| %207 = OpFunctionParameter %14 | |
| %208 = OpFunctionParameter %16 | |
| %202 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %203 = OpFunction %2 None %206 | |
| %209 = OpFunctionParameter %14 | |
| %210 = OpFunctionParameter %16 | |
| %205 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| ; reuse an existing function type | |
| %211 = OpFunction %2 None %223 | |
| %212 = OpFunctionParameter %6 | |
| %213 = OpFunctionParameter %8 | |
| %214 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %215 = OpFunction %2 None %224 | |
| %216 = OpFunctionParameter %8 | |
| %217 = OpFunctionParameter %6 | |
| %218 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %219 = OpFunction %2 None %224 | |
| %220 = OpFunctionParameter %8 | |
| %221 = OpFunctionParameter %6 | |
| %222 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| ; don't adjust the type of the function if it creates a duplicate | |
| %225 = OpFunction %2 None %233 | |
| %226 = OpFunctionParameter %42 | |
| %227 = OpFunctionParameter %24 | |
| %228 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %229 = OpFunction %2 None %234 | |
| %230 = OpFunctionParameter %24 | |
| %231 = OpFunctionParameter %42 | |
| %232 = OpLabel | |
| 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); | |
| // Can't permute main function | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(4, 105, {}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| // Can't permute invalid instruction | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(101, 105, {}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| // Permutation has too many values | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(22, 105, {2, 1, 0, 3}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| // Permutation has too few values | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(22, 105, {0, 1}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| // Permutation has invalid values 1 | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(22, 105, {3, 1, 0}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| #ifndef NDEBUG | |
| // Permutation has invalid values 2 | |
| ASSERT_DEATH(TransformationPermuteFunctionParameters(22, 105, {2, 2, 1}) | |
| .IsApplicable(context.get(), transformation_context), | |
| "Permutation has duplicates"); | |
| #endif | |
| // Result id for new function type is not fresh. | |
| ASSERT_FALSE(TransformationPermuteFunctionParameters(22, 42, {2, 1, 0}) | |
| .IsApplicable(context.get(), transformation_context)); | |
| // Successful transformations | |
| { | |
| TransformationPermuteFunctionParameters transformation(12, 105, {1, 0}); | |
| ASSERT_TRUE( | |
| transformation.IsApplicable(context.get(), transformation_context)); | |
| ApplyAndCheckFreshIds(transformation, context.get(), | |
| &transformation_context); | |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed( | |
| context.get(), validator_options, kConsoleMessageConsumer)); | |
| } | |
| { | |
| TransformationPermuteFunctionParameters transformation(28, 106, {1, 0}); | |
| ASSERT_TRUE( | |
| transformation.IsApplicable(context.get(), transformation_context)); | |
| ApplyAndCheckFreshIds(transformation, context.get(), | |
| &transformation_context); | |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed( | |
| context.get(), validator_options, kConsoleMessageConsumer)); | |
| } | |
| { | |
| TransformationPermuteFunctionParameters transformation(200, 107, {1, 0}); | |
| ASSERT_TRUE( | |
| transformation.IsApplicable(context.get(), transformation_context)); | |
| ApplyAndCheckFreshIds(transformation, context.get(), | |
| &transformation_context); | |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed( | |
| context.get(), validator_options, kConsoleMessageConsumer)); | |
| } | |
| { | |
| TransformationPermuteFunctionParameters transformation(219, 108, {1, 0}); | |
| ASSERT_TRUE( | |
| transformation.IsApplicable(context.get(), transformation_context)); | |
| ApplyAndCheckFreshIds(transformation, context.get(), | |
| &transformation_context); | |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed( | |
| context.get(), validator_options, kConsoleMessageConsumer)); | |
| } | |
| { | |
| TransformationPermuteFunctionParameters transformation(229, 109, {1, 0}); | |
| 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" %72 %74 | |
| OpExecutionMode %4 OriginUpperLeft | |
| OpSource ESSL 310 | |
| OpName %4 "main" | |
| OpName %12 "g(f1;f1;" | |
| OpName %10 "x" | |
| OpName %11 "y" | |
| OpName %22 "f(f1;i1;vf2;" | |
| OpName %19 "x" | |
| OpName %20 "y" | |
| OpName %21 "z" | |
| OpName %28 "cond(i1;f1;" | |
| OpName %26 "a" | |
| OpName %27 "b" | |
| OpName %53 "param" | |
| OpName %54 "param" | |
| OpName %66 "param" | |
| OpName %67 "param" | |
| OpName %72 "color" | |
| OpName %74 "gl_FragCoord" | |
| OpName %75 "param" | |
| OpName %79 "param" | |
| OpName %85 "param" | |
| OpName %86 "param" | |
| OpName %91 "param" | |
| OpName %92 "param" | |
| OpName %93 "param" | |
| OpName %99 "param" | |
| OpName %100 "param" | |
| OpName %101 "param" | |
| OpDecorate %20 RelaxedPrecision | |
| OpDecorate %26 RelaxedPrecision | |
| OpDecorate %47 RelaxedPrecision | |
| OpDecorate %58 RelaxedPrecision | |
| OpDecorate %72 Location 0 | |
| OpDecorate %74 BuiltIn FragCoord | |
| %2 = OpTypeVoid | |
| %3 = OpTypeFunction %2 | |
| %6 = OpTypeFloat 32 | |
| %7 = OpTypePointer Function %6 | |
| %8 = OpTypeVector %6 4 | |
| %9 = OpTypeFunction %8 %7 %7 | |
| %14 = OpTypeInt 32 1 | |
| %15 = OpTypePointer Function %14 | |
| %16 = OpTypeVector %6 2 | |
| %17 = OpTypePointer Function %16 | |
| %18 = OpTypeFunction %8 %7 %15 %17 | |
| %24 = OpTypeBool | |
| %31 = OpConstant %6 255 | |
| %33 = OpConstant %6 0 | |
| %34 = OpConstant %6 1 | |
| %42 = OpTypeInt 32 0 | |
| %43 = OpConstant %42 0 | |
| %49 = OpConstant %42 1 | |
| %64 = OpConstant %14 4 | |
| %65 = OpConstant %6 5 | |
| %71 = OpTypePointer Output %8 | |
| %72 = OpVariable %71 Output | |
| %73 = OpTypePointer Input %8 | |
| %74 = OpVariable %73 Input | |
| %76 = OpTypePointer Input %6 | |
| %84 = OpConstant %14 5 | |
| %90 = OpConstant %6 3 | |
| %98 = OpConstant %6 4 | |
| %206 = OpTypeFunction %2 %14 %16 | |
| %223 = OpTypeFunction %2 %6 %8 | |
| %224 = OpTypeFunction %2 %8 %6 | |
| %233 = OpTypeFunction %2 %42 %24 | |
| %25 = OpTypeFunction %24 %7 %15 | |
| %107 = OpTypeFunction %2 %16 %14 | |
| %4 = OpFunction %2 None %3 | |
| %5 = OpLabel | |
| %66 = OpVariable %15 Function | |
| %67 = OpVariable %7 Function | |
| %75 = OpVariable %7 Function | |
| %79 = OpVariable %7 Function | |
| %85 = OpVariable %15 Function | |
| %86 = OpVariable %7 Function | |
| %91 = OpVariable %7 Function | |
| %92 = OpVariable %15 Function | |
| %93 = OpVariable %17 Function | |
| %99 = OpVariable %7 Function | |
| %100 = OpVariable %15 Function | |
| %101 = OpVariable %17 Function | |
| OpStore %66 %64 | |
| OpStore %67 %65 | |
| %68 = OpFunctionCall %24 %28 %67 %66 | |
| OpSelectionMerge %70 None | |
| OpBranchConditional %68 %69 %83 | |
| %69 = OpLabel | |
| %77 = OpAccessChain %76 %74 %43 | |
| %78 = OpLoad %6 %77 | |
| OpStore %75 %78 | |
| %80 = OpAccessChain %76 %74 %49 | |
| %81 = OpLoad %6 %80 | |
| OpStore %79 %81 | |
| %82 = OpFunctionCall %8 %12 %79 %75 | |
| OpStore %72 %82 | |
| OpBranch %70 | |
| %83 = OpLabel | |
| OpStore %85 %84 | |
| OpStore %86 %65 | |
| %87 = OpFunctionCall %24 %28 %86 %85 | |
| OpSelectionMerge %89 None | |
| OpBranchConditional %87 %88 %97 | |
| %88 = OpLabel | |
| OpStore %91 %90 | |
| OpStore %92 %64 | |
| %94 = OpLoad %8 %74 | |
| %95 = OpVectorShuffle %16 %94 %94 0 1 | |
| OpStore %93 %95 | |
| %96 = OpFunctionCall %8 %22 %91 %92 %93 | |
| OpStore %72 %96 | |
| OpBranch %89 | |
| %97 = OpLabel | |
| OpStore %99 %98 | |
| OpStore %100 %84 | |
| %102 = OpLoad %8 %74 | |
| %103 = OpVectorShuffle %16 %102 %102 0 1 | |
| OpStore %101 %103 | |
| %104 = OpFunctionCall %8 %22 %99 %100 %101 | |
| OpStore %72 %104 | |
| OpBranch %89 | |
| %89 = OpLabel | |
| OpBranch %70 | |
| %70 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %12 = OpFunction %8 None %9 | |
| %11 = OpFunctionParameter %7 | |
| %10 = OpFunctionParameter %7 | |
| %13 = OpLabel | |
| %30 = OpLoad %6 %10 | |
| %32 = OpFDiv %6 %30 %31 | |
| %35 = OpLoad %6 %11 | |
| %36 = OpFDiv %6 %35 %31 | |
| %37 = OpFSub %6 %34 %36 | |
| %38 = OpCompositeConstruct %8 %32 %33 %37 %34 | |
| OpReturnValue %38 | |
| OpFunctionEnd | |
| %22 = OpFunction %8 None %18 | |
| %19 = OpFunctionParameter %7 | |
| %20 = OpFunctionParameter %15 | |
| %21 = OpFunctionParameter %17 | |
| %23 = OpLabel | |
| %53 = OpVariable %7 Function | |
| %54 = OpVariable %7 Function | |
| %41 = OpLoad %6 %19 | |
| %44 = OpAccessChain %7 %21 %43 | |
| %45 = OpLoad %6 %44 | |
| %46 = OpFAdd %6 %41 %45 | |
| %47 = OpLoad %14 %20 | |
| %48 = OpConvertSToF %6 %47 | |
| %50 = OpAccessChain %7 %21 %49 | |
| %51 = OpLoad %6 %50 | |
| %52 = OpFAdd %6 %48 %51 | |
| OpStore %53 %46 | |
| OpStore %54 %52 | |
| %55 = OpFunctionCall %8 %12 %54 %53 | |
| OpReturnValue %55 | |
| OpFunctionEnd | |
| %28 = OpFunction %24 None %25 | |
| %27 = OpFunctionParameter %7 | |
| %26 = OpFunctionParameter %15 | |
| %29 = OpLabel | |
| %58 = OpLoad %14 %26 | |
| %59 = OpConvertSToF %6 %58 | |
| %60 = OpLoad %6 %27 | |
| %61 = OpFOrdLessThan %24 %59 %60 | |
| OpReturnValue %61 | |
| OpFunctionEnd | |
| %200 = OpFunction %2 None %107 | |
| %208 = OpFunctionParameter %16 | |
| %207 = OpFunctionParameter %14 | |
| %202 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %203 = OpFunction %2 None %206 | |
| %209 = OpFunctionParameter %14 | |
| %210 = OpFunctionParameter %16 | |
| %205 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %211 = OpFunction %2 None %223 | |
| %212 = OpFunctionParameter %6 | |
| %213 = OpFunctionParameter %8 | |
| %214 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %215 = OpFunction %2 None %224 | |
| %216 = OpFunctionParameter %8 | |
| %217 = OpFunctionParameter %6 | |
| %218 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %219 = OpFunction %2 None %223 | |
| %221 = OpFunctionParameter %6 | |
| %220 = OpFunctionParameter %8 | |
| %222 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %225 = OpFunction %2 None %233 | |
| %226 = OpFunctionParameter %42 | |
| %227 = OpFunctionParameter %24 | |
| %228 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| %229 = OpFunction %2 None %233 | |
| %231 = OpFunctionParameter %42 | |
| %230 = OpFunctionParameter %24 | |
| %232 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); | |
| } | |
| } // namespace | |
| } // namespace fuzz | |
| } // namespace spvtools |