1 // Copyright (c) 2019 Google LLC
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "source/fuzz/transformation_set_function_control.h"
16 #include "test/fuzz/fuzz_test_util.h"
17
18 namespace spvtools {
19 namespace fuzz {
20 namespace {
21
TEST(TransformationSetFunctionControlTest,VariousScenarios)22 TEST(TransformationSetFunctionControlTest, VariousScenarios) {
23 // This is a simple transformation; this test captures the important things
24 // to check for.
25
26 std::string shader = R"(
27 OpCapability Shader
28 %1 = OpExtInstImport "GLSL.std.450"
29 OpMemoryModel Logical GLSL450
30 OpEntryPoint Fragment %4 "main" %54
31 OpExecutionMode %4 OriginUpperLeft
32 OpSource ESSL 310
33 OpName %4 "main"
34 OpName %11 "foo(i1;i1;"
35 OpName %9 "a"
36 OpName %10 "b"
37 OpName %13 "bar("
38 OpName %17 "baz(i1;"
39 OpName %16 "x"
40 OpName %21 "boo(i1;i1;"
41 OpName %19 "a"
42 OpName %20 "b"
43 OpName %29 "g"
44 OpName %42 "param"
45 OpName %44 "param"
46 OpName %45 "param"
47 OpName %48 "param"
48 OpName %49 "param"
49 OpName %54 "color"
50 OpDecorate %54 Location 0
51 %2 = OpTypeVoid
52 %3 = OpTypeFunction %2
53 %6 = OpTypeInt 32 1
54 %7 = OpTypePointer Function %6
55 %8 = OpTypeFunction %6 %7 %7
56 %15 = OpTypeFunction %6 %7
57 %28 = OpTypePointer Private %6
58 %29 = OpVariable %28 Private
59 %30 = OpConstant %6 2
60 %31 = OpConstant %6 5
61 %51 = OpTypeFloat 32
62 %52 = OpTypeVector %51 4
63 %53 = OpTypePointer Output %52
64 %54 = OpVariable %53 Output
65 %4 = OpFunction %2 None %3
66 %5 = OpLabel
67 %42 = OpVariable %7 Function
68 %44 = OpVariable %7 Function
69 %45 = OpVariable %7 Function
70 %48 = OpVariable %7 Function
71 %49 = OpVariable %7 Function
72 %41 = OpFunctionCall %2 %13
73 OpStore %42 %30
74 %43 = OpFunctionCall %6 %17 %42
75 OpStore %44 %31
76 %46 = OpLoad %6 %29
77 OpStore %45 %46
78 %47 = OpFunctionCall %6 %21 %44 %45
79 OpStore %48 %43
80 OpStore %49 %47
81 %50 = OpFunctionCall %6 %11 %48 %49
82 OpReturn
83 OpFunctionEnd
84 %11 = OpFunction %6 Const %8
85 %9 = OpFunctionParameter %7
86 %10 = OpFunctionParameter %7
87 %12 = OpLabel
88 %23 = OpLoad %6 %9
89 %24 = OpLoad %6 %10
90 %25 = OpIAdd %6 %23 %24
91 OpReturnValue %25
92 OpFunctionEnd
93 %13 = OpFunction %2 Inline %3
94 %14 = OpLabel
95 OpStore %29 %30
96 OpReturn
97 OpFunctionEnd
98 %17 = OpFunction %6 Pure|DontInline %15
99 %16 = OpFunctionParameter %7
100 %18 = OpLabel
101 %32 = OpLoad %6 %16
102 %33 = OpIAdd %6 %31 %32
103 OpReturnValue %33
104 OpFunctionEnd
105 %21 = OpFunction %6 DontInline %8
106 %19 = OpFunctionParameter %7
107 %20 = OpFunctionParameter %7
108 %22 = OpLabel
109 %36 = OpLoad %6 %19
110 %37 = OpLoad %6 %20
111 %38 = OpIMul %6 %36 %37
112 OpReturnValue %38
113 OpFunctionEnd
114 )";
115
116 const auto env = SPV_ENV_UNIVERSAL_1_3;
117 const auto consumer = nullptr;
118 const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
119
120 FactManager fact_manager;
121 spvtools::ValidatorOptions validator_options;
122 TransformationContext transformation_context(&fact_manager,
123 validator_options);
124
125 // %36 is not a function
126 ASSERT_FALSE(TransformationSetFunctionControl(36, SpvFunctionControlMaskNone)
127 .IsApplicable(context.get(), transformation_context));
128 // Cannot add the Pure function control to %4 as it did not already have it
129 ASSERT_FALSE(TransformationSetFunctionControl(4, SpvFunctionControlPureMask)
130 .IsApplicable(context.get(), transformation_context));
131 // Cannot add the Const function control to %21 as it did not already
132 // have it
133 ASSERT_FALSE(TransformationSetFunctionControl(21, SpvFunctionControlConstMask)
134 .IsApplicable(context.get(), transformation_context));
135
136 // Set to None, removing Const
137 TransformationSetFunctionControl transformation1(11,
138 SpvFunctionControlMaskNone);
139 ASSERT_TRUE(
140 transformation1.IsApplicable(context.get(), transformation_context));
141 transformation1.Apply(context.get(), &transformation_context);
142
143 // Set to Inline; silly to do it on an entry point, but it is allowed
144 TransformationSetFunctionControl transformation2(
145 4, SpvFunctionControlInlineMask);
146 ASSERT_TRUE(
147 transformation2.IsApplicable(context.get(), transformation_context));
148 transformation2.Apply(context.get(), &transformation_context);
149
150 // Set to Pure, removing DontInline
151 TransformationSetFunctionControl transformation3(17,
152 SpvFunctionControlPureMask);
153 ASSERT_TRUE(
154 transformation3.IsApplicable(context.get(), transformation_context));
155 transformation3.Apply(context.get(), &transformation_context);
156
157 // Change from Inline to DontInline
158 TransformationSetFunctionControl transformation4(
159 13, SpvFunctionControlDontInlineMask);
160 ASSERT_TRUE(
161 transformation4.IsApplicable(context.get(), transformation_context));
162 transformation4.Apply(context.get(), &transformation_context);
163
164 std::string after_transformation = R"(
165 OpCapability Shader
166 %1 = OpExtInstImport "GLSL.std.450"
167 OpMemoryModel Logical GLSL450
168 OpEntryPoint Fragment %4 "main" %54
169 OpExecutionMode %4 OriginUpperLeft
170 OpSource ESSL 310
171 OpName %4 "main"
172 OpName %11 "foo(i1;i1;"
173 OpName %9 "a"
174 OpName %10 "b"
175 OpName %13 "bar("
176 OpName %17 "baz(i1;"
177 OpName %16 "x"
178 OpName %21 "boo(i1;i1;"
179 OpName %19 "a"
180 OpName %20 "b"
181 OpName %29 "g"
182 OpName %42 "param"
183 OpName %44 "param"
184 OpName %45 "param"
185 OpName %48 "param"
186 OpName %49 "param"
187 OpName %54 "color"
188 OpDecorate %54 Location 0
189 %2 = OpTypeVoid
190 %3 = OpTypeFunction %2
191 %6 = OpTypeInt 32 1
192 %7 = OpTypePointer Function %6
193 %8 = OpTypeFunction %6 %7 %7
194 %15 = OpTypeFunction %6 %7
195 %28 = OpTypePointer Private %6
196 %29 = OpVariable %28 Private
197 %30 = OpConstant %6 2
198 %31 = OpConstant %6 5
199 %51 = OpTypeFloat 32
200 %52 = OpTypeVector %51 4
201 %53 = OpTypePointer Output %52
202 %54 = OpVariable %53 Output
203 %4 = OpFunction %2 Inline %3
204 %5 = OpLabel
205 %42 = OpVariable %7 Function
206 %44 = OpVariable %7 Function
207 %45 = OpVariable %7 Function
208 %48 = OpVariable %7 Function
209 %49 = OpVariable %7 Function
210 %41 = OpFunctionCall %2 %13
211 OpStore %42 %30
212 %43 = OpFunctionCall %6 %17 %42
213 OpStore %44 %31
214 %46 = OpLoad %6 %29
215 OpStore %45 %46
216 %47 = OpFunctionCall %6 %21 %44 %45
217 OpStore %48 %43
218 OpStore %49 %47
219 %50 = OpFunctionCall %6 %11 %48 %49
220 OpReturn
221 OpFunctionEnd
222 %11 = OpFunction %6 None %8
223 %9 = OpFunctionParameter %7
224 %10 = OpFunctionParameter %7
225 %12 = OpLabel
226 %23 = OpLoad %6 %9
227 %24 = OpLoad %6 %10
228 %25 = OpIAdd %6 %23 %24
229 OpReturnValue %25
230 OpFunctionEnd
231 %13 = OpFunction %2 DontInline %3
232 %14 = OpLabel
233 OpStore %29 %30
234 OpReturn
235 OpFunctionEnd
236 %17 = OpFunction %6 Pure %15
237 %16 = OpFunctionParameter %7
238 %18 = OpLabel
239 %32 = OpLoad %6 %16
240 %33 = OpIAdd %6 %31 %32
241 OpReturnValue %33
242 OpFunctionEnd
243 %21 = OpFunction %6 DontInline %8
244 %19 = OpFunctionParameter %7
245 %20 = OpFunctionParameter %7
246 %22 = OpLabel
247 %36 = OpLoad %6 %19
248 %37 = OpLoad %6 %20
249 %38 = OpIMul %6 %36 %37
250 OpReturnValue %38
251 OpFunctionEnd
252 )";
253 ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
254 }
255
256 } // namespace
257 } // namespace fuzz
258 } // namespace spvtools
259