| // Copyright (c) 2020 Google LLC |
| // |
| // 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_replace_branch_from_dead_block_with_exit.h" |
| |
| #include "gtest/gtest.h" |
| #include "source/fuzz/fuzzer_util.h" |
| #include "test/fuzz/fuzz_test_util.h" |
| |
| namespace spvtools { |
| namespace fuzz { |
| namespace { |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, BasicTest) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %12 = OpTypeInt 32 1 |
| %13 = OpTypePointer Function %12 |
| %15 = OpConstant %12 1 |
| %17 = OpConstant %12 2 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %14 = OpVariable %13 Function |
| OpSelectionMerge %9 None |
| OpBranchConditional %7 %8 %21 |
| %8 = OpLabel |
| OpSelectionMerge %11 None |
| OpBranchConditional %7 %10 %16 |
| %10 = OpLabel |
| OpStore %14 %15 |
| OpBranch %20 |
| %20 = OpLabel |
| OpBranch %11 |
| %16 = OpLabel |
| OpStore %14 %17 |
| OpBranch %11 |
| %11 = OpLabel |
| OpBranch %9 |
| %21 = OpLabel |
| OpBranch %9 |
| %9 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(8); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(10); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(11); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(16); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(20); |
| |
| // Bad: 4 is not a block |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(4, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| // Bad: 200 does not exist |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(200, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| // Bad: 21 is not a dead block |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(21, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| // Bad: terminator of 8 is not OpBranch |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(8, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| // Bad: 10's successor only has 10 as a predecessor |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(10, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| #ifndef NDEBUG |
| ASSERT_DEATH( |
| TransformationReplaceBranchFromDeadBlockWithExit(20, spv::Op::OpSwitch, 0) |
| .IsApplicable(context.get(), transformation_context), |
| "Invalid early exit opcode."); |
| #endif |
| |
| auto transformation1 = |
| TransformationReplaceBranchFromDeadBlockWithExit(20, spv::Op::OpKill, 0); |
| auto transformation2 = |
| TransformationReplaceBranchFromDeadBlockWithExit(16, spv::Op::OpKill, 0); |
| ASSERT_TRUE( |
| transformation1.IsApplicable(context.get(), transformation_context)); |
| ASSERT_TRUE( |
| transformation2.IsApplicable(context.get(), transformation_context)); |
| |
| ApplyAndCheckFreshIds(transformation1, context.get(), |
| &transformation_context); |
| // Applying transformation 1 should disable transformation 2 |
| ASSERT_FALSE( |
| transformation2.IsApplicable(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 |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %12 = OpTypeInt 32 1 |
| %13 = OpTypePointer Function %12 |
| %15 = OpConstant %12 1 |
| %17 = OpConstant %12 2 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %14 = OpVariable %13 Function |
| OpSelectionMerge %9 None |
| OpBranchConditional %7 %8 %21 |
| %8 = OpLabel |
| OpSelectionMerge %11 None |
| OpBranchConditional %7 %10 %16 |
| %10 = OpLabel |
| OpStore %14 %15 |
| OpBranch %20 |
| %20 = OpLabel |
| OpKill |
| %16 = OpLabel |
| OpStore %14 %17 |
| OpBranch %11 |
| %11 = OpLabel |
| OpBranch %9 |
| %21 = OpLabel |
| OpBranch %9 |
| %9 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, |
| VertexShaderWithLoopInContinueConstruct) { |
| std::string shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %4 "main" |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeInt 32 1 |
| %7 = OpTypeFunction %6 |
| %12 = OpTypePointer Function %6 |
| %14 = OpConstant %6 0 |
| %22 = OpConstant %6 10 |
| %23 = OpTypeBool |
| %26 = OpConstant %6 1 |
| %40 = OpConstant %6 100 |
| %48 = OpConstantFalse %23 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %50 None |
| OpBranchConditional %48 %49 %50 |
| %49 = OpLabel |
| %51 = OpFunctionCall %6 %10 |
| OpBranch %50 |
| %50 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| %10 = OpFunction %6 None %7 |
| %11 = OpLabel |
| %13 = OpVariable %12 Function |
| %15 = OpVariable %12 Function |
| %33 = OpVariable %12 Function |
| OpStore %33 %14 |
| OpBranch %34 |
| %34 = OpLabel |
| OpLoopMerge %36 %37 None |
| OpBranch %38 |
| %38 = OpLabel |
| %39 = OpLoad %6 %33 |
| %41 = OpSLessThan %23 %39 %40 |
| OpBranchConditional %41 %35 %36 |
| %35 = OpLabel |
| OpSelectionMerge %202 None |
| OpBranchConditional %48 %200 %201 |
| %200 = OpLabel |
| OpBranch %202 |
| %201 = OpLabel |
| %400 = OpCopyObject %6 %14 |
| OpBranch %202 |
| %202 = OpLabel |
| OpBranch %37 |
| %37 = OpLabel |
| OpStore %13 %14 |
| OpStore %15 %14 |
| OpBranch %16 |
| %16 = OpLabel |
| OpLoopMerge %18 %19 None |
| OpBranch %20 |
| %20 = OpLabel |
| OpSelectionMerge %102 None |
| OpBranchConditional %48 %100 %101 |
| %100 = OpLabel |
| OpBranch %102 |
| %101 = OpLabel |
| OpBranch %102 |
| %102 = OpLabel |
| %21 = OpLoad %6 %15 |
| %24 = OpSLessThan %23 %21 %22 |
| OpBranchConditional %24 %17 %18 |
| %17 = OpLabel |
| OpSelectionMerge %302 None |
| OpBranchConditional %48 %300 %301 |
| %300 = OpLabel |
| OpBranch %302 |
| %301 = OpLabel |
| OpBranch %302 |
| %302 = OpLabel |
| %25 = OpLoad %6 %13 |
| %27 = OpIAdd %6 %25 %26 |
| OpStore %13 %27 |
| OpBranch %19 |
| %19 = OpLabel |
| %28 = OpLoad %6 %15 |
| %29 = OpIAdd %6 %28 %26 |
| OpStore %15 %29 |
| OpBranch %16 |
| %18 = OpLabel |
| %30 = OpLoad %6 %13 |
| %42 = OpCopyObject %6 %30 |
| %43 = OpLoad %6 %33 |
| %44 = OpIAdd %6 %43 %42 |
| OpStore %33 %44 |
| OpBranch %34 |
| %36 = OpLabel |
| %45 = OpLoad %6 %33 |
| OpReturnValue %45 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| for (auto block : {16, 17, 18, 19, 20, 34, 35, 36, 37, 38, |
| 49, 100, 101, 102, 200, 201, 202, 300, 301, 302}) { |
| transformation_context.GetFactManager()->AddFactBlockIsDead(block); |
| } |
| |
| // Bad: OpKill not allowed in vertex shader |
| ASSERT_FALSE( |
| TransformationReplaceBranchFromDeadBlockWithExit(201, spv::Op::OpKill, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: OpReturn is not allowed in function that expects a returned value. |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 200, spv::Op::OpReturn, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Return value id does not exist |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 201, spv::Op::OpReturnValue, 1000) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Return value id does not have a type |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 200, spv::Op::OpReturnValue, 6) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Return value id does not have the right type |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 201, spv::Op::OpReturnValue, 48) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Return value id is not available |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 200, spv::Op::OpReturnValue, 400) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Early exit now allowed in continue construct |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 101, spv::Op::OpUnreachable, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| // Bad: Early exit now allowed in continue construct (again) |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 300, spv::Op::OpReturnValue, 14) |
| .IsApplicable(context.get(), transformation_context)); |
| |
| auto transformation1 = TransformationReplaceBranchFromDeadBlockWithExit( |
| 200, spv::Op::OpUnreachable, 0); |
| auto transformation2 = TransformationReplaceBranchFromDeadBlockWithExit( |
| 201, spv::Op::OpReturnValue, 400); |
| |
| ASSERT_TRUE( |
| transformation1.IsApplicable(context.get(), transformation_context)); |
| ASSERT_TRUE( |
| transformation2.IsApplicable(context.get(), transformation_context)); |
| |
| ApplyAndCheckFreshIds(transformation2, context.get(), |
| &transformation_context); |
| ASSERT_FALSE( |
| transformation1.IsApplicable(context.get(), transformation_context)); |
| |
| ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options, |
| kConsoleMessageConsumer)); |
| |
| opt::Instruction* return_value_inst = |
| context->get_instr_block(201)->terminator(); |
| ASSERT_EQ(spv::Op::OpReturnValue, return_value_inst->opcode()); |
| ASSERT_EQ(SPV_OPERAND_TYPE_ID, return_value_inst->GetInOperand(0).type); |
| ASSERT_EQ(400, return_value_inst->GetSingleWordInOperand(0)); |
| |
| std::string after_transformation = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Vertex %4 "main" |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeInt 32 1 |
| %7 = OpTypeFunction %6 |
| %12 = OpTypePointer Function %6 |
| %14 = OpConstant %6 0 |
| %22 = OpConstant %6 10 |
| %23 = OpTypeBool |
| %26 = OpConstant %6 1 |
| %40 = OpConstant %6 100 |
| %48 = OpConstantFalse %23 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %50 None |
| OpBranchConditional %48 %49 %50 |
| %49 = OpLabel |
| %51 = OpFunctionCall %6 %10 |
| OpBranch %50 |
| %50 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| %10 = OpFunction %6 None %7 |
| %11 = OpLabel |
| %13 = OpVariable %12 Function |
| %15 = OpVariable %12 Function |
| %33 = OpVariable %12 Function |
| OpStore %33 %14 |
| OpBranch %34 |
| %34 = OpLabel |
| OpLoopMerge %36 %37 None |
| OpBranch %38 |
| %38 = OpLabel |
| %39 = OpLoad %6 %33 |
| %41 = OpSLessThan %23 %39 %40 |
| OpBranchConditional %41 %35 %36 |
| %35 = OpLabel |
| OpSelectionMerge %202 None |
| OpBranchConditional %48 %200 %201 |
| %200 = OpLabel |
| OpBranch %202 |
| %201 = OpLabel |
| %400 = OpCopyObject %6 %14 |
| OpReturnValue %400 |
| %202 = OpLabel |
| OpBranch %37 |
| %37 = OpLabel |
| OpStore %13 %14 |
| OpStore %15 %14 |
| OpBranch %16 |
| %16 = OpLabel |
| OpLoopMerge %18 %19 None |
| OpBranch %20 |
| %20 = OpLabel |
| OpSelectionMerge %102 None |
| OpBranchConditional %48 %100 %101 |
| %100 = OpLabel |
| OpBranch %102 |
| %101 = OpLabel |
| OpBranch %102 |
| %102 = OpLabel |
| %21 = OpLoad %6 %15 |
| %24 = OpSLessThan %23 %21 %22 |
| OpBranchConditional %24 %17 %18 |
| %17 = OpLabel |
| OpSelectionMerge %302 None |
| OpBranchConditional %48 %300 %301 |
| %300 = OpLabel |
| OpBranch %302 |
| %301 = OpLabel |
| OpBranch %302 |
| %302 = OpLabel |
| %25 = OpLoad %6 %13 |
| %27 = OpIAdd %6 %25 %26 |
| OpStore %13 %27 |
| OpBranch %19 |
| %19 = OpLabel |
| %28 = OpLoad %6 %15 |
| %29 = OpIAdd %6 %28 %26 |
| OpStore %15 %29 |
| OpBranch %16 |
| %18 = OpLabel |
| %30 = OpLoad %6 %13 |
| %42 = OpCopyObject %6 %30 |
| %43 = OpLoad %6 %33 |
| %44 = OpIAdd %6 %43 %42 |
| OpStore %33 %44 |
| OpBranch %34 |
| %36 = OpLabel |
| %45 = OpLoad %6 %33 |
| OpReturnValue %45 |
| OpFunctionEnd |
| )"; |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, OpPhi) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %12 = OpTypeInt 32 1 |
| %13 = OpTypePointer Function %12 |
| %15 = OpConstant %12 1 |
| %17 = OpConstant %12 2 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %14 = OpVariable %13 Function |
| OpSelectionMerge %9 None |
| OpBranchConditional %7 %8 %21 |
| %8 = OpLabel |
| OpSelectionMerge %11 None |
| OpBranchConditional %7 %10 %16 |
| %10 = OpLabel |
| OpStore %14 %15 |
| OpBranch %20 |
| %20 = OpLabel |
| %48 = OpCopyObject %12 %15 |
| OpBranch %11 |
| %16 = OpLabel |
| OpStore %14 %17 |
| %49 = OpCopyObject %12 %17 |
| OpBranch %11 |
| %11 = OpLabel |
| %50 = OpPhi %12 %48 %20 %49 %16 |
| OpBranch %9 |
| %21 = OpLabel |
| OpBranch %9 |
| %9 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(8); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(10); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(11); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(16); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(20); |
| |
| auto transformation1 = |
| TransformationReplaceBranchFromDeadBlockWithExit(20, spv::Op::OpKill, 0); |
| auto transformation2 = |
| TransformationReplaceBranchFromDeadBlockWithExit(16, spv::Op::OpKill, 0); |
| ASSERT_TRUE( |
| transformation1.IsApplicable(context.get(), transformation_context)); |
| ASSERT_TRUE( |
| transformation2.IsApplicable(context.get(), transformation_context)); |
| |
| ApplyAndCheckFreshIds(transformation1, context.get(), |
| &transformation_context); |
| // Applying transformation 1 should disable transformation 2 |
| ASSERT_FALSE( |
| transformation2.IsApplicable(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 |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %12 = OpTypeInt 32 1 |
| %13 = OpTypePointer Function %12 |
| %15 = OpConstant %12 1 |
| %17 = OpConstant %12 2 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %14 = OpVariable %13 Function |
| OpSelectionMerge %9 None |
| OpBranchConditional %7 %8 %21 |
| %8 = OpLabel |
| OpSelectionMerge %11 None |
| OpBranchConditional %7 %10 %16 |
| %10 = OpLabel |
| OpStore %14 %15 |
| OpBranch %20 |
| %20 = OpLabel |
| %48 = OpCopyObject %12 %15 |
| OpKill |
| %16 = OpLabel |
| OpStore %14 %17 |
| %49 = OpCopyObject %12 %17 |
| OpBranch %11 |
| %11 = OpLabel |
| %50 = OpPhi %12 %49 %16 |
| OpBranch %9 |
| %21 = OpLabel |
| OpBranch %9 |
| %9 = OpLabel |
| OpReturn |
| OpFunctionEnd |
| )"; |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, |
| DominatorAfterDeadBlockSuccessor) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpBranchConditional %7 %9 %10 |
| %9 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpCopyObject %6 %7 |
| OpBranch %8 |
| %10 = OpLabel |
| OpBranch %13 |
| %8 = OpLabel |
| OpReturn |
| %13 = OpLabel |
| OpBranch %8 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(9); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(11); |
| |
| ASSERT_FALSE(TransformationReplaceBranchFromDeadBlockWithExit( |
| 11, spv::Op::OpUnreachable, 0) |
| .IsApplicable(context.get(), transformation_context)); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, |
| UnreachableSuccessor) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpBranchConditional %7 %9 %10 |
| %9 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpCopyObject %6 %7 |
| OpBranch %8 |
| %10 = OpLabel |
| OpReturn |
| %8 = OpLabel |
| OpReturn |
| %13 = OpLabel |
| OpBranch %8 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(9); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(11); |
| |
| TransformationReplaceBranchFromDeadBlockWithExit transformation( |
| 11, spv::Op::OpUnreachable, 0); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| transformation.Apply(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 |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeBool |
| %7 = OpConstantFalse %6 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpBranchConditional %7 %9 %10 |
| %9 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpCopyObject %6 %7 |
| OpUnreachable |
| %10 = OpLabel |
| OpReturn |
| %8 = OpLabel |
| OpReturn |
| %13 = OpLabel |
| OpBranch %8 |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, |
| DeadBlockAfterItsSuccessor) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantTrue %6 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpBranchConditional %7 %9 %10 |
| %9 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpCopyObject %6 %7 |
| OpBranch %8 |
| %10 = OpLabel |
| OpBranch %13 |
| %8 = OpLabel |
| OpReturn |
| %13 = OpLabel |
| OpBranch %8 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(10); |
| transformation_context.GetFactManager()->AddFactBlockIsDead(13); |
| |
| TransformationReplaceBranchFromDeadBlockWithExit transformation( |
| 13, spv::Op::OpUnreachable, 0); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| transformation.Apply(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 |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeBool |
| %7 = OpConstantTrue %6 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpBranchConditional %7 %9 %10 |
| %9 = OpLabel |
| OpBranch %11 |
| %11 = OpLabel |
| %12 = OpCopyObject %6 %7 |
| OpBranch %8 |
| %10 = OpLabel |
| OpBranch %13 |
| %8 = OpLabel |
| OpReturn |
| %13 = OpLabel |
| OpUnreachable |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| TEST(TransformationReplaceBranchFromDeadBlockWithExitTest, |
| BranchToOuterMergeBlock) { |
| std::string 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 |
| %6 = OpTypeBool |
| %7 = OpConstantTrue %6 |
| %15 = OpTypeInt 32 0 |
| %14 = OpUndef %15 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpSwitch %14 %9 1 %8 |
| %9 = OpLabel |
| OpSelectionMerge %10 None |
| OpBranchConditional %7 %11 %10 |
| %8 = OpLabel |
| OpReturn |
| %11 = OpLabel |
| OpBranch %8 |
| %10 = OpLabel |
| OpBranch %8 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_4; |
| 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); |
| |
| transformation_context.GetFactManager()->AddFactBlockIsDead(10); |
| |
| TransformationReplaceBranchFromDeadBlockWithExit transformation( |
| 10, spv::Op::OpUnreachable, 0); |
| ASSERT_TRUE( |
| transformation.IsApplicable(context.get(), transformation_context)); |
| transformation.Apply(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 |
| OpSource ESSL 320 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeBool |
| %7 = OpConstantTrue %6 |
| %15 = OpTypeInt 32 0 |
| %14 = OpUndef %15 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| OpSelectionMerge %8 None |
| OpSwitch %14 %9 1 %8 |
| %9 = OpLabel |
| OpSelectionMerge %10 None |
| OpBranchConditional %7 %11 %10 |
| %8 = OpLabel |
| OpReturn |
| %11 = OpLabel |
| OpBranch %8 |
| %10 = OpLabel |
| OpUnreachable |
| OpFunctionEnd |
| )"; |
| |
| ASSERT_TRUE(IsEqual(env, after_transformation, context.get())); |
| } |
| |
| } // namespace |
| } // namespace fuzz |
| } // namespace spvtools |
