1 // Copyright (c) 2018 Pierre Moreau
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 <string>
16
17 #include "gmock/gmock.h"
18 #include "test/link/linker_fixture.h"
19
20 namespace spvtools {
21 namespace {
22
23 using ::testing::HasSubstr;
24 using PartialLinkage = spvtest::LinkerTest;
25
TEST_F(PartialLinkage,Allowed)26 TEST_F(PartialLinkage, Allowed) {
27 const std::string body1 = R"(
28 OpCapability Linkage
29 OpDecorate %1 LinkageAttributes "foo" Import
30 OpDecorate %2 LinkageAttributes "bar" Import
31 %3 = OpTypeFloat 32
32 %1 = OpVariable %3 Uniform
33 %2 = OpVariable %3 Uniform
34 )";
35 const std::string body2 = R"(
36 OpCapability Linkage
37 OpDecorate %1 LinkageAttributes "bar" Export
38 %2 = OpTypeFloat 32
39 %3 = OpConstant %2 3.1415
40 %1 = OpVariable %2 Uniform %3
41 )";
42
43 spvtest::Binary linked_binary;
44 LinkerOptions linker_options;
45 linker_options.SetAllowPartialLinkage(true);
46 ASSERT_EQ(SPV_SUCCESS,
47 AssembleAndLink({body1, body2}, &linked_binary, linker_options));
48
49 const std::string expected_res = R"(OpCapability Linkage
50 OpModuleProcessed "Linked by SPIR-V Tools Linker"
51 OpDecorate %1 LinkageAttributes "foo" Import
52 %2 = OpTypeFloat 32
53 %1 = OpVariable %2 Uniform
54 %3 = OpConstant %2 3.1415
55 %4 = OpVariable %2 Uniform %3
56 )";
57 std::string res_body;
58 SetDisassembleOptions(SPV_BINARY_TO_TEXT_OPTION_NO_HEADER);
59 ASSERT_EQ(SPV_SUCCESS, Disassemble(linked_binary, &res_body))
60 << GetErrorMessage();
61 EXPECT_EQ(expected_res, res_body);
62 }
63
TEST_F(PartialLinkage,Disallowed)64 TEST_F(PartialLinkage, Disallowed) {
65 const std::string body1 = R"(
66 OpCapability Linkage
67 OpDecorate %1 LinkageAttributes "foo" Import
68 OpDecorate %2 LinkageAttributes "bar" Import
69 %3 = OpTypeFloat 32
70 %1 = OpVariable %3 Uniform
71 %2 = OpVariable %3 Uniform
72 )";
73 const std::string body2 = R"(
74 OpCapability Linkage
75 OpDecorate %1 LinkageAttributes "bar" Export
76 %2 = OpTypeFloat 32
77 %3 = OpConstant %2 3.1415
78 %1 = OpVariable %2 Uniform %3
79 )";
80
81 spvtest::Binary linked_binary;
82 EXPECT_EQ(SPV_ERROR_INVALID_BINARY,
83 AssembleAndLink({body1, body2}, &linked_binary));
84 EXPECT_THAT(GetErrorMessage(),
85 HasSubstr("Unresolved external reference to \"foo\"."));
86 }
87
88 } // namespace
89 } // namespace spvtools
90