llvm/unittests/CodeGen/DIETest.cpp

   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:
  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 {
  45   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
  56 TEST_P(DIEExprFixture, SizeOf) {
  57   DIEExpr Tst(Val);
  58   EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
  59 }
  60
  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 {
  78   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
  89 TEST_P(DIELabelFixture, SizeOf) {
  90   DIELabel Tst(Val);
  91   EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
  92 }
  93
  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 {
 122   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
 135 TEST_P(DIEDeltaFixture, SizeOf) {
 136   DIEDelta Tst(Hi, Lo);
 137   EXPECT_EQ(Size, Tst.sizeOf(TestPrinter->getAP()->getDwarfFormParams(), Form));
 138 }
 139
 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 {
 157   void SetUp() override { DIEFixtureBase::SetUp(); }
 158 };
 159
 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