1 //===- llvm/unittest/CodeGen/DIETest.cpp ----------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "llvm/CodeGen/DIE.h"
10 #include "TestAsmPrinter.h"
11 #include "llvm/CodeGen/AsmPrinter.h"
12 #include "llvm/MC/MCContext.h"
13 #include "llvm/MC/MCExpr.h"
14 #include "llvm/Testing/Support/Error.h"
15
16 using namespace llvm;
17 using testing::_;
18 using testing::SaveArg;
19
20 namespace {
21
22 using DIETestParams =
23 std::tuple<unsigned, dwarf::DwarfFormat, dwarf::Form, unsigned>;
24
25 class DIEFixtureBase : public testing::TestWithParam<DIETestParams> {
26 protected:
SetUp()27 void SetUp() override {
28 unsigned Version;
29 dwarf::DwarfFormat Format;
30 std::tie(Version, Format, Form, Size) = GetParam();
31 auto ExpectedTestPrinter =
32 TestAsmPrinter::create("x86_64-pc-linux", Version, Format);
33 ASSERT_THAT_EXPECTED(ExpectedTestPrinter, Succeeded());
34 TestPrinter = std::move(ExpectedTestPrinter.get());
35 if (!TestPrinter)
36 GTEST_SKIP();
37 }
38
39 dwarf::Form Form;
40 unsigned Size;
41 std::unique_ptr<TestAsmPrinter> TestPrinter;
42 };
43
44 struct DIEExprFixture : public DIEFixtureBase {
SetUp__anon6d8051950111::DIEExprFixture45 void SetUp() override {
46 DIEFixtureBase::SetUp();
47 if (!TestPrinter)
48 return;
49
50 Val = MCConstantExpr::create(42, TestPrinter->getCtx());
51 }
52
53 const MCExpr *Val = nullptr;
54 };
55
TEST_P(DIEExprFixture,SizeOf)56 TEST_P(DIEExprFixture, SizeOf) {
57 DIEExpr Tst(Val);
58 EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
59 }
60
TEST_P(DIEExprFixture,EmitValue)61 TEST_P(DIEExprFixture, EmitValue) {
62 DIEExpr Tst(Val);
63 EXPECT_CALL(TestPrinter->getMS(), emitValueImpl(Val, Size, _));
64 Tst.emitValue(TestPrinter->getAP(), Form);
65 }
66
67 INSTANTIATE_TEST_SUITE_P(
68 DIETestParams, DIEExprFixture,
69 testing::Values(
70 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data4, 4u},
71 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data8, 8u},
72 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_sec_offset, 4u},
73 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data4, 4u},
74 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data8, 8u},
75 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_sec_offset, 8u}));
76
77 struct DIELabelFixture : public DIEFixtureBase {
SetUp__anon6d8051950111::DIELabelFixture78 void SetUp() override {
79 DIEFixtureBase::SetUp();
80 if (!TestPrinter)
81 return;
82
83 Val = TestPrinter->getCtx().createTempSymbol();
84 }
85
86 const MCSymbol *Val = nullptr;
87 };
88
TEST_P(DIELabelFixture,SizeOf)89 TEST_P(DIELabelFixture, SizeOf) {
90 DIELabel Tst(Val);
91 EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
92 }
93
TEST_P(DIELabelFixture,EmitValue)94 TEST_P(DIELabelFixture, EmitValue) {
95 DIELabel Tst(Val);
96
97 const MCExpr *Arg0 = nullptr;
98 EXPECT_CALL(TestPrinter->getMS(), emitValueImpl(_, Size, _))
99 .WillOnce(SaveArg<0>(&Arg0));
100 Tst.emitValue(TestPrinter->getAP(), Form);
101
102 const MCSymbolRefExpr *ActualArg0 = dyn_cast_or_null<MCSymbolRefExpr>(Arg0);
103 ASSERT_NE(ActualArg0, nullptr);
104 EXPECT_EQ(&(ActualArg0->getSymbol()), Val);
105 }
106
107 INSTANTIATE_TEST_SUITE_P(
108 DIETestParams, DIELabelFixture,
109 testing::Values(
110 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data4, 4u},
111 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data8, 8u},
112 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_sec_offset, 4u},
113 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_strp, 4u},
114 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_addr, 8u},
115 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data4, 4u},
116 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data8, 8u},
117 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_sec_offset, 8u},
118 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_strp, 8u},
119 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_addr, 8u}));
120
121 struct DIEDeltaFixture : public DIEFixtureBase {
SetUp__anon6d8051950111::DIEDeltaFixture122 void SetUp() override {
123 DIEFixtureBase::SetUp();
124 if (!TestPrinter)
125 return;
126
127 Hi = TestPrinter->getCtx().createTempSymbol();
128 Lo = TestPrinter->getCtx().createTempSymbol();
129 }
130
131 const MCSymbol *Hi = nullptr;
132 const MCSymbol *Lo = nullptr;
133 };
134
TEST_P(DIEDeltaFixture,SizeOf)135 TEST_P(DIEDeltaFixture, SizeOf) {
136 DIEDelta Tst(Hi, Lo);
137 EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
138 }
139
TEST_P(DIEDeltaFixture,EmitValue)140 TEST_P(DIEDeltaFixture, EmitValue) {
141 DIEDelta Tst(Hi, Lo);
142 EXPECT_CALL(TestPrinter->getMS(), emitAbsoluteSymbolDiff(Hi, Lo, Size));
143 Tst.emitValue(TestPrinter->getAP(), Form);
144 }
145
146 INSTANTIATE_TEST_SUITE_P(
147 DIETestParams, DIEDeltaFixture,
148 testing::Values(
149 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data4, 4u},
150 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data8, 8u},
151 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_sec_offset, 4u},
152 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data4, 4u},
153 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data8, 8u},
154 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_sec_offset, 8u}));
155
156 struct DIELocListFixture : public DIEFixtureBase {
SetUp__anon6d8051950111::DIELocListFixture157 void SetUp() override { DIEFixtureBase::SetUp(); }
158 };
159
TEST_P(DIELocListFixture,SizeOf)160 TEST_P(DIELocListFixture, SizeOf) {
161 DIELocList Tst(999);
162 EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
163 }
164
165 INSTANTIATE_TEST_SUITE_P(
166 DIETestParams, DIELocListFixture,
167 testing::Values(
168 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_loclistx, 2u},
169 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_data4, 4u},
170 DIETestParams{4, dwarf::DWARF32, dwarf::DW_FORM_sec_offset, 4u},
171 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_loclistx, 2u},
172 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_data8, 8u},
173 DIETestParams{4, dwarf::DWARF64, dwarf::DW_FORM_sec_offset, 8u}));
174
175 } // end namespace
176