166702629SJames Henderson //===- DWARFDebugLineTest.cpp ---------------------------------------------===//
266702629SJames Henderson //
3*2946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*2946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
5*2946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
666702629SJames Henderson //
766702629SJames Henderson //===----------------------------------------------------------------------===//
866702629SJames Henderson 
966702629SJames Henderson #include "DwarfGenerator.h"
1066702629SJames Henderson #include "DwarfUtils.h"
1166702629SJames Henderson #include "llvm/DebugInfo/DWARF/DWARFContext.h"
1266702629SJames Henderson #include "llvm/DebugInfo/DWARF/DWARFDebugLine.h"
1366702629SJames Henderson #include "llvm/Object/ObjectFile.h"
1466702629SJames Henderson #include "llvm/Testing/Support/Error.h"
1566702629SJames Henderson #include "gtest/gtest.h"
1666702629SJames Henderson 
1766702629SJames Henderson using namespace llvm;
1866702629SJames Henderson using namespace dwarf;
19a3acf99eSJames Henderson using namespace dwarfgen;
2066702629SJames Henderson using namespace object;
2166702629SJames Henderson using namespace utils;
22a3acf99eSJames Henderson using namespace testing;
2366702629SJames Henderson 
2466702629SJames Henderson namespace {
25a3acf99eSJames Henderson struct CommonFixture {
26a3acf99eSJames Henderson   CommonFixture()
27004b729eSJames Henderson       : LineData("", true, 0), Recoverable(Error::success()),
28004b729eSJames Henderson         RecordRecoverable(std::bind(&CommonFixture::recordRecoverable, this,
29a3acf99eSJames Henderson                                     std::placeholders::_1)),
30004b729eSJames Henderson         Unrecoverable(Error::success()),
31004b729eSJames Henderson         RecordUnrecoverable(std::bind(&CommonFixture::recordUnrecoverable, this,
32a3acf99eSJames Henderson                                       std::placeholders::_1)){};
3366702629SJames Henderson 
34004b729eSJames Henderson   ~CommonFixture() {
35004b729eSJames Henderson     EXPECT_FALSE(Recoverable);
36004b729eSJames Henderson     EXPECT_FALSE(Unrecoverable);
37004b729eSJames Henderson   }
38a3acf99eSJames Henderson 
39a3acf99eSJames Henderson   bool setupGenerator(uint16_t Version = 4) {
4066702629SJames Henderson     Triple T = getHostTripleForAddrSize(8);
4166702629SJames Henderson     if (!isConfigurationSupported(T))
4266702629SJames Henderson       return false;
43a3acf99eSJames Henderson     auto ExpectedGenerator = Generator::create(T, Version);
4466702629SJames Henderson     if (ExpectedGenerator)
45a3acf99eSJames Henderson       Gen.reset(ExpectedGenerator->release());
4666702629SJames Henderson     return true;
4766702629SJames Henderson   }
4866702629SJames Henderson 
49a3acf99eSJames Henderson   void generate() {
50a3acf99eSJames Henderson     Context = createContext();
51a3acf99eSJames Henderson     assert(Context != nullptr && "test state is not valid");
52a3acf99eSJames Henderson     const DWARFObject &Obj = Context->getDWARFObj();
53198a87c8SJames Henderson     LineData = DWARFDataExtractor(Obj, Obj.getLineSection(),
54198a87c8SJames Henderson                                   sys::IsLittleEndianHost, 8);
55a3acf99eSJames Henderson   }
56a3acf99eSJames Henderson 
5766702629SJames Henderson   std::unique_ptr<DWARFContext> createContext() {
58a3acf99eSJames Henderson     if (!Gen)
5966702629SJames Henderson       return nullptr;
60a3acf99eSJames Henderson     StringRef FileBytes = Gen->generate();
6166702629SJames Henderson     MemoryBufferRef FileBuffer(FileBytes, "dwarf");
6266702629SJames Henderson     auto Obj = object::ObjectFile::createObjectFile(FileBuffer);
6366702629SJames Henderson     if (Obj)
6466702629SJames Henderson       return DWARFContext::create(**Obj);
6566702629SJames Henderson     return nullptr;
6666702629SJames Henderson   }
6766702629SJames Henderson 
68a3acf99eSJames Henderson   DWARFDebugLine::SectionParser setupParser() {
69a3acf99eSJames Henderson     LineTable &LT = Gen->addLineTable(DWARF32);
70a3acf99eSJames Henderson     LT.addExtendedOpcode(9, DW_LNE_set_address, {{0xadd4e55, LineTable::Quad}});
71a3acf99eSJames Henderson     LT.addStandardOpcode(DW_LNS_copy, {});
72a3acf99eSJames Henderson     LT.addByte(0xaa);
73a3acf99eSJames Henderson     LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
74a3acf99eSJames Henderson 
75a3acf99eSJames Henderson     LineTable &LT2 = Gen->addLineTable(DWARF64);
76a3acf99eSJames Henderson     LT2.addExtendedOpcode(9, DW_LNE_set_address,
77a3acf99eSJames Henderson                           {{0x11223344, LineTable::Quad}});
78a3acf99eSJames Henderson     LT2.addStandardOpcode(DW_LNS_copy, {});
79a3acf99eSJames Henderson     LT2.addByte(0xbb);
80a3acf99eSJames Henderson     LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
81a3acf99eSJames Henderson 
82a3acf99eSJames Henderson     generate();
83a3acf99eSJames Henderson 
84a3acf99eSJames Henderson     return DWARFDebugLine::SectionParser(LineData, *Context, CUs, TUs);
85a3acf99eSJames Henderson   }
86a3acf99eSJames Henderson 
87004b729eSJames Henderson   void recordRecoverable(Error Err) {
88004b729eSJames Henderson     Recoverable = joinErrors(std::move(Recoverable), std::move(Err));
89004b729eSJames Henderson   }
90004b729eSJames Henderson   void recordUnrecoverable(Error Err) {
91004b729eSJames Henderson     Unrecoverable = joinErrors(std::move(Unrecoverable), std::move(Err));
92a3acf99eSJames Henderson   }
93a3acf99eSJames Henderson 
94a3acf99eSJames Henderson   void checkError(ArrayRef<StringRef> ExpectedMsgs, Error Err) {
95a3acf99eSJames Henderson     ASSERT_TRUE(Err.operator bool());
96a3acf99eSJames Henderson     size_t WhichMsg = 0;
97a3acf99eSJames Henderson     Error Remaining =
98a3acf99eSJames Henderson         handleErrors(std::move(Err), [&](const ErrorInfoBase &Actual) {
99a3acf99eSJames Henderson           ASSERT_LT(WhichMsg, ExpectedMsgs.size());
100a3acf99eSJames Henderson           // Use .str(), because googletest doesn't visualise a StringRef
101a3acf99eSJames Henderson           // properly.
102a3acf99eSJames Henderson           EXPECT_EQ(Actual.message(), ExpectedMsgs[WhichMsg++].str());
103a3acf99eSJames Henderson         });
104a3acf99eSJames Henderson     EXPECT_EQ(WhichMsg, ExpectedMsgs.size());
105a3acf99eSJames Henderson     EXPECT_FALSE(Remaining);
106a3acf99eSJames Henderson   }
107a3acf99eSJames Henderson 
108a3acf99eSJames Henderson   void checkError(StringRef ExpectedMsg, Error Err) {
109a3acf99eSJames Henderson     checkError(ArrayRef<StringRef>{ExpectedMsg}, std::move(Err));
110a3acf99eSJames Henderson   }
111a3acf99eSJames Henderson 
112a3acf99eSJames Henderson   void checkGetOrParseLineTableEmitsError(StringRef ExpectedMsg,
113a3acf99eSJames Henderson                                           uint64_t Offset = 0) {
114a3acf99eSJames Henderson     auto ExpectedLineTable = Line.getOrParseLineTable(
115004b729eSJames Henderson         LineData, Offset, *Context, nullptr, RecordRecoverable);
116a3acf99eSJames Henderson     EXPECT_FALSE(ExpectedLineTable);
117004b729eSJames Henderson     EXPECT_FALSE(Recoverable);
118a3acf99eSJames Henderson 
119a3acf99eSJames Henderson     checkError(ExpectedMsg, ExpectedLineTable.takeError());
120a3acf99eSJames Henderson   }
121a3acf99eSJames Henderson 
122a3acf99eSJames Henderson   std::unique_ptr<Generator> Gen;
123a3acf99eSJames Henderson   std::unique_ptr<DWARFContext> Context;
124a3acf99eSJames Henderson   DWARFDataExtractor LineData;
125a3acf99eSJames Henderson   DWARFDebugLine Line;
126004b729eSJames Henderson   Error Recoverable;
127004b729eSJames Henderson   std::function<void(Error)> RecordRecoverable;
128004b729eSJames Henderson   Error Unrecoverable;
129004b729eSJames Henderson   std::function<void(Error)> RecordUnrecoverable;
130a3acf99eSJames Henderson 
131143eaeabSPaul Robinson   SmallVector<std::unique_ptr<DWARFUnit>, 2> CUs;
1327f330944SPaul Robinson   SmallVector<std::unique_ptr<DWARFUnit>, 2> TUs;
13366702629SJames Henderson };
13466702629SJames Henderson 
135a3acf99eSJames Henderson // Fixtures must derive from "Test", but parameterised fixtures from
136a3acf99eSJames Henderson // "TestWithParam". It does not seem possible to inherit from both, so we share
137a3acf99eSJames Henderson // the common state in a separate class, inherited by the two fixture classes.
138a3acf99eSJames Henderson struct DebugLineBasicFixture : public Test, public CommonFixture {};
139a3acf99eSJames Henderson 
140a3acf99eSJames Henderson struct DebugLineParameterisedFixture
141a3acf99eSJames Henderson     : public TestWithParam<std::pair<uint16_t, DwarfFormat>>,
142a3acf99eSJames Henderson       public CommonFixture {
143a3acf99eSJames Henderson   void SetUp() { std::tie(Version, Format) = GetParam(); }
144a3acf99eSJames Henderson 
145a3acf99eSJames Henderson   uint16_t Version;
146a3acf99eSJames Henderson   DwarfFormat Format;
147a3acf99eSJames Henderson };
148a3acf99eSJames Henderson 
149a3acf99eSJames Henderson void checkDefaultPrologue(uint16_t Version, DwarfFormat Format,
150a3acf99eSJames Henderson                           DWARFDebugLine::Prologue Prologue,
151a3acf99eSJames Henderson                           uint64_t BodyLength) {
152a3acf99eSJames Henderson   // Check version specific fields and values.
153a3acf99eSJames Henderson   uint64_t UnitLength;
154a3acf99eSJames Henderson   uint64_t PrologueLength;
155a3acf99eSJames Henderson   switch (Version) {
156a3acf99eSJames Henderson   case 4:
157a3acf99eSJames Henderson     PrologueLength = 36;
158a3acf99eSJames Henderson     UnitLength = PrologueLength + 2;
15911a9de74SJames Henderson     EXPECT_EQ(Prologue.MaxOpsPerInst, 1u);
160a3acf99eSJames Henderson     break;
161a3acf99eSJames Henderson   case 2:
162a3acf99eSJames Henderson   case 3:
163a3acf99eSJames Henderson     PrologueLength = 35;
164a3acf99eSJames Henderson     UnitLength = PrologueLength + 2;
165a3acf99eSJames Henderson     break;
166a3acf99eSJames Henderson   case 5:
167a3acf99eSJames Henderson     PrologueLength = 39;
168a3acf99eSJames Henderson     UnitLength = PrologueLength + 4;
16911a9de74SJames Henderson     EXPECT_EQ(Prologue.getAddressSize(), 8u);
17011a9de74SJames Henderson     EXPECT_EQ(Prologue.SegSelectorSize, 0u);
171a3acf99eSJames Henderson     break;
172a3acf99eSJames Henderson   default:
173a3acf99eSJames Henderson     llvm_unreachable("unsupported DWARF version");
174a3acf99eSJames Henderson   }
175a3acf99eSJames Henderson   UnitLength += BodyLength + (Format == DWARF32 ? 4 : 8);
176a3acf99eSJames Henderson 
177a3acf99eSJames Henderson   EXPECT_EQ(Prologue.TotalLength, UnitLength);
178a3acf99eSJames Henderson   EXPECT_EQ(Prologue.PrologueLength, PrologueLength);
17911a9de74SJames Henderson   EXPECT_EQ(Prologue.MinInstLength, 1u);
18011a9de74SJames Henderson   EXPECT_EQ(Prologue.DefaultIsStmt, 1u);
181a3acf99eSJames Henderson   EXPECT_EQ(Prologue.LineBase, -5);
18211a9de74SJames Henderson   EXPECT_EQ(Prologue.LineRange, 14u);
18311a9de74SJames Henderson   EXPECT_EQ(Prologue.OpcodeBase, 13u);
184a3acf99eSJames Henderson   std::vector<uint8_t> ExpectedLengths = {0, 1, 1, 1, 1, 0, 0, 0, 1, 0, 0, 1};
185a3acf99eSJames Henderson   EXPECT_EQ(Prologue.StandardOpcodeLengths, ExpectedLengths);
18611a9de74SJames Henderson   ASSERT_EQ(Prologue.IncludeDirectories.size(), 1u);
187a3acf99eSJames Henderson   ASSERT_EQ(Prologue.IncludeDirectories[0].getForm(), DW_FORM_string);
188a3acf99eSJames Henderson   EXPECT_STREQ(*Prologue.IncludeDirectories[0].getAsCString(), "a dir");
18911a9de74SJames Henderson   ASSERT_EQ(Prologue.FileNames.size(), 1u);
190a3acf99eSJames Henderson   ASSERT_EQ(Prologue.FileNames[0].Name.getForm(), DW_FORM_string);
191a3acf99eSJames Henderson   EXPECT_STREQ(*Prologue.FileNames[0].Name.getAsCString(), "a file");
192a3acf99eSJames Henderson }
193a3acf99eSJames Henderson 
194a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, GetOrParseLineTableAtInvalidOffset) {
195a3acf99eSJames Henderson   if (!setupGenerator())
196a3acf99eSJames Henderson     return;
197a3acf99eSJames Henderson   generate();
198a3acf99eSJames Henderson 
199a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
200a3acf99eSJames Henderson       "offset 0x00000000 is not a valid debug line section offset", 0);
201a3acf99eSJames Henderson   // Repeat to show that an error is reported each time.
202a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
203a3acf99eSJames Henderson       "offset 0x00000000 is not a valid debug line section offset", 0);
204a3acf99eSJames Henderson   // Show that an error is reported for later offsets too.
205a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
206a3acf99eSJames Henderson       "offset 0x00000001 is not a valid debug line section offset", 1);
207a3acf99eSJames Henderson }
208a3acf99eSJames Henderson 
209a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, GetOrParseLineTableAtInvalidOffsetAfterData) {
210a3acf99eSJames Henderson   if (!setupGenerator())
21166702629SJames Henderson     return;
21266702629SJames Henderson 
213a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
214a3acf99eSJames Henderson   LT.setCustomPrologue({{0, LineTable::Byte}});
21566702629SJames Henderson 
216a3acf99eSJames Henderson   generate();
217a3acf99eSJames Henderson 
218a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
219a3acf99eSJames Henderson       "offset 0x00000001 is not a valid debug line section offset", 1);
220a3acf99eSJames Henderson }
221a3acf99eSJames Henderson 
222a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, GetOrParseLineTableValidTable) {
223a3acf99eSJames Henderson   if (!setupGenerator(Version))
224a3acf99eSJames Henderson     return;
225a3acf99eSJames Henderson 
226a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
227a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
228a3acf99eSJames Henderson 
229a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
230a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address, {{0xadd4e55, LineTable::Quad}});
231a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
232a3acf99eSJames Henderson   LT.addByte(0xaa);
233a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
234a3acf99eSJames Henderson 
235a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(Format);
236a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address, {{0x11223344, LineTable::Quad}});
237a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
238a3acf99eSJames Henderson   LT2.addByte(0xbb);
239a3acf99eSJames Henderson   LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
240a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address, {{0x55667788, LineTable::Quad}});
241a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
242a3acf99eSJames Henderson   LT2.addByte(0xcc);
243a3acf99eSJames Henderson   LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
244a3acf99eSJames Henderson 
245a3acf99eSJames Henderson   generate();
246a3acf99eSJames Henderson 
247004b729eSJames Henderson   auto ExpectedLineTable = Line.getOrParseLineTable(LineData, 0, *Context,
248004b729eSJames Henderson                                                     nullptr, RecordRecoverable);
249a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable.operator bool());
250004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
251a3acf99eSJames Henderson   const DWARFDebugLine::LineTable *Expected = *ExpectedLineTable;
252a3acf99eSJames Henderson   checkDefaultPrologue(Version, Format, Expected->Prologue, 16);
253a68aee07SRoman Lebedev   EXPECT_EQ(Expected->Sequences.size(), 1u);
254a3acf99eSJames Henderson 
255a3acf99eSJames Henderson   uint64_t SecondOffset =
256a3acf99eSJames Henderson       Expected->Prologue.sizeofTotalLength() + Expected->Prologue.TotalLength;
257004b729eSJames Henderson   Recoverable = Error::success();
258a3acf99eSJames Henderson   auto ExpectedLineTable2 = Line.getOrParseLineTable(
259004b729eSJames Henderson       LineData, SecondOffset, *Context, nullptr, RecordRecoverable);
260a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable2.operator bool());
261004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
262a3acf99eSJames Henderson   const DWARFDebugLine::LineTable *Expected2 = *ExpectedLineTable2;
263a3acf99eSJames Henderson   checkDefaultPrologue(Version, Format, Expected2->Prologue, 32);
26411a9de74SJames Henderson   EXPECT_EQ(Expected2->Sequences.size(), 2u);
265a3acf99eSJames Henderson 
266a3acf99eSJames Henderson   EXPECT_NE(Expected, Expected2);
267a3acf99eSJames Henderson 
268a3acf99eSJames Henderson   // Check that if the same offset is requested, the exact same pointer is
269a3acf99eSJames Henderson   // returned.
270004b729eSJames Henderson   Recoverable = Error::success();
271004b729eSJames Henderson   auto ExpectedLineTable3 = Line.getOrParseLineTable(
272004b729eSJames Henderson       LineData, 0, *Context, nullptr, RecordRecoverable);
273a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable3.operator bool());
274004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
275a3acf99eSJames Henderson   EXPECT_EQ(Expected, *ExpectedLineTable3);
276a3acf99eSJames Henderson 
277004b729eSJames Henderson   Recoverable = Error::success();
278a3acf99eSJames Henderson   auto ExpectedLineTable4 = Line.getOrParseLineTable(
279004b729eSJames Henderson       LineData, SecondOffset, *Context, nullptr, RecordRecoverable);
280a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable4.operator bool());
281004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
282a3acf99eSJames Henderson   EXPECT_EQ(Expected2, *ExpectedLineTable4);
283a3acf99eSJames Henderson 
284a3acf99eSJames Henderson   // TODO: Add tests that show that the body of the programs have been read
285a3acf99eSJames Henderson   // correctly.
286a3acf99eSJames Henderson }
287a3acf99eSJames Henderson 
288a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForReservedLength) {
289a3acf99eSJames Henderson   if (!setupGenerator())
290a3acf99eSJames Henderson     return;
291a3acf99eSJames Henderson 
292a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
293a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
294a3acf99eSJames Henderson 
295a3acf99eSJames Henderson   generate();
296a3acf99eSJames Henderson 
297a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
298a3acf99eSJames Henderson       "parsing line table prologue at offset 0x00000000 unsupported reserved "
299a3acf99eSJames Henderson       "unit length found of value 0xffffff00");
300a3acf99eSJames Henderson }
301a3acf99eSJames Henderson 
302a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForLowVersion) {
303a3acf99eSJames Henderson   if (!setupGenerator())
304a3acf99eSJames Henderson     return;
305a3acf99eSJames Henderson 
306a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
307a3acf99eSJames Henderson   LT.setCustomPrologue(
308a3acf99eSJames Henderson       {{LineTable::Half, LineTable::Long}, {1, LineTable::Half}});
309a3acf99eSJames Henderson 
310a3acf99eSJames Henderson   generate();
311a3acf99eSJames Henderson 
312a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError("parsing line table prologue at offset "
313a3acf99eSJames Henderson                                      "0x00000000 found unsupported version "
314a3acf99eSJames Henderson                                      "0x01");
315a3acf99eSJames Henderson }
316a3acf99eSJames Henderson 
317a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForInvalidV5IncludeDirTable) {
318a3acf99eSJames Henderson   if (!setupGenerator(5))
319a3acf99eSJames Henderson     return;
320a3acf99eSJames Henderson 
321a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
322a3acf99eSJames Henderson   LT.setCustomPrologue({
323a3acf99eSJames Henderson       {19, LineTable::Long}, // unit length
324a3acf99eSJames Henderson       {5, LineTable::Half},  // version
325a3acf99eSJames Henderson       {8, LineTable::Byte},  // addr size
326a3acf99eSJames Henderson       {0, LineTable::Byte},  // segment selector size
327a3acf99eSJames Henderson       {11, LineTable::Long}, // prologue length
328a3acf99eSJames Henderson       {1, LineTable::Byte},  // min instruction length
329a3acf99eSJames Henderson       {1, LineTable::Byte},  // max ops per instruction
330a3acf99eSJames Henderson       {1, LineTable::Byte},  // default is_stmt
331a3acf99eSJames Henderson       {0, LineTable::Byte},  // line base
332a3acf99eSJames Henderson       {14, LineTable::Byte}, // line range
333a3acf99eSJames Henderson       {2, LineTable::Byte},  // opcode base (small to reduce the amount of
334a3acf99eSJames Henderson                              // setup required).
335a3acf99eSJames Henderson       {0, LineTable::Byte},  // standard opcode lengths
336a3acf99eSJames Henderson       {0, LineTable::Byte},  // directory entry format count (should not be
337a3acf99eSJames Henderson                              // zero).
338a3acf99eSJames Henderson       {0, LineTable::ULEB},  // directories count
339a3acf99eSJames Henderson       {0, LineTable::Byte},  // file name entry format count
340a3acf99eSJames Henderson       {0, LineTable::ULEB}   // file name entry count
341a3acf99eSJames Henderson   });
342a3acf99eSJames Henderson 
343a3acf99eSJames Henderson   generate();
344a3acf99eSJames Henderson 
345a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
346a3acf99eSJames Henderson       "parsing line table prologue at 0x00000000 found an invalid directory or "
347a3acf99eSJames Henderson       "file table description at 0x00000014");
348a3acf99eSJames Henderson }
349a3acf99eSJames Henderson 
350a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, ErrorForTooLargePrologueLength) {
351a3acf99eSJames Henderson   if (!setupGenerator(Version))
352a3acf99eSJames Henderson     return;
353a3acf99eSJames Henderson 
354a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
355a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
356a3acf99eSJames Henderson 
357a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
358a3acf99eSJames Henderson   DWARFDebugLine::Prologue Prologue = LT.createBasicPrologue();
359a3acf99eSJames Henderson   ++Prologue.PrologueLength;
360a3acf99eSJames Henderson   LT.setPrologue(Prologue);
361a3acf99eSJames Henderson 
362a3acf99eSJames Henderson   generate();
363a3acf99eSJames Henderson 
364a3acf99eSJames Henderson   uint64_t ExpectedEnd =
365a3acf99eSJames Henderson       Prologue.TotalLength + 1 + Prologue.sizeofTotalLength();
366a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
367a3acf99eSJames Henderson       (Twine("parsing line table prologue at 0x00000000 should have ended at "
368a3acf99eSJames Henderson              "0x000000") +
369a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd) + " but it ended at 0x000000" +
370a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd - 1))
371a3acf99eSJames Henderson           .str());
372a3acf99eSJames Henderson }
373a3acf99eSJames Henderson 
374a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, ErrorForTooShortPrologueLength) {
375a3acf99eSJames Henderson   if (!setupGenerator(Version))
376a3acf99eSJames Henderson     return;
377a3acf99eSJames Henderson 
378a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
379a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
380a3acf99eSJames Henderson 
381a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
382a3acf99eSJames Henderson   DWARFDebugLine::Prologue Prologue = LT.createBasicPrologue();
383a3acf99eSJames Henderson   // FIXME: Ideally, we'd test for 1 less than expected, but the code does not
384a3acf99eSJames Henderson   // currently fail if missing only the terminator of a v2-4 file table.
385a3acf99eSJames Henderson   if (Version < 5)
386a3acf99eSJames Henderson     Prologue.PrologueLength -= 2;
387a3acf99eSJames Henderson   else
388a3acf99eSJames Henderson     Prologue.PrologueLength -= 1;
389a3acf99eSJames Henderson   LT.setPrologue(Prologue);
390a3acf99eSJames Henderson 
391a3acf99eSJames Henderson   generate();
392a3acf99eSJames Henderson 
393a3acf99eSJames Henderson   uint64_t ExpectedEnd =
394a3acf99eSJames Henderson       Prologue.TotalLength - 1 + Prologue.sizeofTotalLength();
395a3acf99eSJames Henderson   if (Version < 5)
396a3acf99eSJames Henderson     --ExpectedEnd;
397a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
398a3acf99eSJames Henderson       (Twine("parsing line table prologue at 0x00000000 should have ended at "
399a3acf99eSJames Henderson              "0x000000") +
400a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd) + " but it ended at 0x000000" +
401a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd + 1))
402a3acf99eSJames Henderson           .str());
403a3acf99eSJames Henderson }
404a3acf99eSJames Henderson 
405a3acf99eSJames Henderson INSTANTIATE_TEST_CASE_P(
406a3acf99eSJames Henderson     LineTableTestParams, DebugLineParameterisedFixture,
407a3acf99eSJames Henderson     Values(std::make_pair(
408a3acf99eSJames Henderson                2, DWARF32), // Test lower-bound of v2-3 fields and DWARF32.
409a3acf99eSJames Henderson            std::make_pair(3, DWARF32), // Test upper-bound of v2-3 fields.
410a3acf99eSJames Henderson            std::make_pair(4, DWARF64), // Test v4 fields and DWARF64.
411a3acf99eSJames Henderson            std::make_pair(5, DWARF32), std::make_pair(5, DWARF64)), );
412a3acf99eSJames Henderson 
413a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForInvalidExtendedOpcodeLength) {
414a3acf99eSJames Henderson   if (!setupGenerator())
415a3acf99eSJames Henderson     return;
416a3acf99eSJames Henderson 
417a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
418a3acf99eSJames Henderson   // The Length should be 1 for an end sequence opcode.
419a3acf99eSJames Henderson   LT.addExtendedOpcode(2, DW_LNE_end_sequence, {});
420a3acf99eSJames Henderson 
421a3acf99eSJames Henderson   generate();
422a3acf99eSJames Henderson 
423a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError("unexpected line op length at offset "
424a3acf99eSJames Henderson                                      "0x00000030 expected 0x02 found 0x01");
425a3acf99eSJames Henderson }
426a3acf99eSJames Henderson 
427a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForMismatchedAddressSize) {
428a3acf99eSJames Henderson   if (!setupGenerator())
429a3acf99eSJames Henderson     return;
430a3acf99eSJames Henderson 
431a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
432a3acf99eSJames Henderson   // The line data extractor expects size 8 (Quad) addresses.
433a3acf99eSJames Henderson   LT.addExtendedOpcode(5, DW_LNE_set_address, {{0x11223344, LineTable::Long}});
434a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
435a3acf99eSJames Henderson   LT.addByte(0xaa);
436a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
437a3acf99eSJames Henderson 
438a3acf99eSJames Henderson   generate();
439a3acf99eSJames Henderson 
440a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
441a3acf99eSJames Henderson       "mismatching address size at offset 0x00000030 expected 0x08 found 0x04");
442a3acf99eSJames Henderson }
443a3acf99eSJames Henderson 
444a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, CallbackUsedForUnterminatedSequence) {
445a3acf99eSJames Henderson   if (!setupGenerator())
446a3acf99eSJames Henderson     return;
447a3acf99eSJames Henderson 
448a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
449a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address,
450a3acf99eSJames Henderson                        {{0x1122334455667788, LineTable::Quad}});
451a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
452a3acf99eSJames Henderson   LT.addByte(0xaa);
453a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
454a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address,
455a3acf99eSJames Henderson                        {{0x99aabbccddeeff00, LineTable::Quad}});
456a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
457a3acf99eSJames Henderson   LT.addByte(0xbb);
458a3acf99eSJames Henderson   LT.addByte(0xcc);
459a3acf99eSJames Henderson 
460a3acf99eSJames Henderson   generate();
461a3acf99eSJames Henderson 
462004b729eSJames Henderson   auto ExpectedLineTable = Line.getOrParseLineTable(LineData, 0, *Context,
463004b729eSJames Henderson                                                     nullptr, RecordRecoverable);
464004b729eSJames Henderson   checkError("last sequence in debug line table is not terminated!",
465004b729eSJames Henderson              std::move(Recoverable));
466a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable.operator bool());
467a68aee07SRoman Lebedev   EXPECT_EQ((*ExpectedLineTable)->Rows.size(), 6u);
468a3acf99eSJames Henderson   // The unterminated sequence is not added to the sequence list.
469a68aee07SRoman Lebedev   EXPECT_EQ((*ExpectedLineTable)->Sequences.size(), 1u);
470a3acf99eSJames Henderson }
471a3acf99eSJames Henderson 
472a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserParsesCorrectly) {
473a3acf99eSJames Henderson   if (!setupGenerator())
474a3acf99eSJames Henderson     return;
475a3acf99eSJames Henderson 
476a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser = setupParser();
477a3acf99eSJames Henderson 
47811a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 0u);
479a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
480a3acf99eSJames Henderson 
481004b729eSJames Henderson   DWARFDebugLine::LineTable Parsed =
482004b729eSJames Henderson       Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
483a3acf99eSJames Henderson   checkDefaultPrologue(4, DWARF32, Parsed.Prologue, 16);
48411a9de74SJames Henderson   EXPECT_EQ(Parsed.Sequences.size(), 1u);
48511a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 62u);
486a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
487a3acf99eSJames Henderson 
488a3acf99eSJames Henderson   DWARFDebugLine::LineTable Parsed2 =
489004b729eSJames Henderson       Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
490a3acf99eSJames Henderson   checkDefaultPrologue(4, DWARF64, Parsed2.Prologue, 16);
49111a9de74SJames Henderson   EXPECT_EQ(Parsed2.Sequences.size(), 1u);
49211a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 136u);
493a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
494a3acf99eSJames Henderson 
495004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
496004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
497a3acf99eSJames Henderson }
498a3acf99eSJames Henderson 
499a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserSkipsCorrectly) {
500a3acf99eSJames Henderson   if (!setupGenerator())
501a3acf99eSJames Henderson     return;
502a3acf99eSJames Henderson 
503a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser = setupParser();
504a3acf99eSJames Henderson 
50511a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 0u);
506a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
507a3acf99eSJames Henderson 
508004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
50911a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 62u);
510a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
511a3acf99eSJames Henderson 
512004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
51311a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 136u);
514a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
515a3acf99eSJames Henderson 
516004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
517a3acf99eSJames Henderson }
518a3acf99eSJames Henderson 
519a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserAlwaysDoneForEmptySection) {
520a3acf99eSJames Henderson   if (!setupGenerator())
521a3acf99eSJames Henderson     return;
522a3acf99eSJames Henderson 
523a3acf99eSJames Henderson   generate();
524a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
525a3acf99eSJames Henderson 
526a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
527a3acf99eSJames Henderson }
528a3acf99eSJames Henderson 
529a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserMovesToEndForBadLengthWhenParsing) {
530a3acf99eSJames Henderson   if (!setupGenerator())
531a3acf99eSJames Henderson     return;
532a3acf99eSJames Henderson 
533a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
534a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
535a3acf99eSJames Henderson   Gen->addLineTable();
536a3acf99eSJames Henderson   generate();
537a3acf99eSJames Henderson 
538a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
539004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
540a3acf99eSJames Henderson 
54111a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 4u);
542a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
543004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
544a3acf99eSJames Henderson 
545a3acf99eSJames Henderson   checkError("parsing line table prologue at offset 0x00000000 unsupported "
546a3acf99eSJames Henderson              "reserved unit length found of value 0xffffff00",
547004b729eSJames Henderson              std::move(Unrecoverable));
548a3acf99eSJames Henderson }
549a3acf99eSJames Henderson 
550a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserMovesToEndForBadLengthWhenSkipping) {
551a3acf99eSJames Henderson   if (!setupGenerator())
552a3acf99eSJames Henderson     return;
553a3acf99eSJames Henderson 
554a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
555a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
556a3acf99eSJames Henderson   Gen->addLineTable();
557a3acf99eSJames Henderson   generate();
558a3acf99eSJames Henderson 
559a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
560004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
561a3acf99eSJames Henderson 
56211a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 4u);
563a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
564a3acf99eSJames Henderson 
565a3acf99eSJames Henderson   checkError("parsing line table prologue at offset 0x00000000 unsupported "
566a3acf99eSJames Henderson              "reserved unit length found of value 0xffffff00",
567004b729eSJames Henderson              std::move(Unrecoverable));
568a3acf99eSJames Henderson }
569a3acf99eSJames Henderson 
570a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserReportsFirstErrorInEachTableWhenParsing) {
571a3acf99eSJames Henderson   if (!setupGenerator())
572a3acf99eSJames Henderson     return;
573a3acf99eSJames Henderson 
574a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
575a3acf99eSJames Henderson   LT.setCustomPrologue({{2, LineTable::Long}, {0, LineTable::Half}});
576a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
577a3acf99eSJames Henderson   LT2.setCustomPrologue({{2, LineTable::Long}, {1, LineTable::Half}});
578a3acf99eSJames Henderson   generate();
579a3acf99eSJames Henderson 
580a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
581004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
582a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
583004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
584a3acf99eSJames Henderson 
585a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
586004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
587a3acf99eSJames Henderson 
588a3acf99eSJames Henderson   checkError({"parsing line table prologue at offset 0x00000000 found "
589a3acf99eSJames Henderson               "unsupported version 0x00",
590a3acf99eSJames Henderson               "parsing line table prologue at offset 0x00000006 found "
591a3acf99eSJames Henderson               "unsupported version 0x01"},
592004b729eSJames Henderson              std::move(Unrecoverable));
593a3acf99eSJames Henderson }
594a3acf99eSJames Henderson 
595a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserReportsNonPrologueProblemsWhenParsing) {
596a3acf99eSJames Henderson   if (!setupGenerator())
597a3acf99eSJames Henderson     return;
598a3acf99eSJames Henderson 
599a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
600a3acf99eSJames Henderson   LT.addExtendedOpcode(0x42, DW_LNE_end_sequence, {});
601a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
602a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address,
603a3acf99eSJames Henderson                         {{0x1234567890abcdef, LineTable::Quad}});
604a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
605a3acf99eSJames Henderson   LT2.addByte(0xbb);
606a3acf99eSJames Henderson   generate();
607a3acf99eSJames Henderson 
608a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
609004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
610004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
611a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
612a3acf99eSJames Henderson   checkError(
613a3acf99eSJames Henderson       "unexpected line op length at offset 0x00000030 expected 0x42 found 0x01",
614004b729eSJames Henderson       std::move(Unrecoverable));
615a3acf99eSJames Henderson 
616a3acf99eSJames Henderson   // Reset the error state so that it does not confuse the next set of checks.
617004b729eSJames Henderson   Unrecoverable = Error::success();
618004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
619a3acf99eSJames Henderson 
620a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
621004b729eSJames Henderson   checkError("last sequence in debug line table is not terminated!",
622004b729eSJames Henderson              std::move(Recoverable));
623004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
624a3acf99eSJames Henderson }
625a3acf99eSJames Henderson 
626a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture,
627a3acf99eSJames Henderson        ParserReportsPrologueErrorsInEachTableWhenSkipping) {
628a3acf99eSJames Henderson   if (!setupGenerator())
629a3acf99eSJames Henderson     return;
630a3acf99eSJames Henderson 
631a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
632a3acf99eSJames Henderson   LT.setCustomPrologue({{2, LineTable::Long}, {0, LineTable::Half}});
633a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
634a3acf99eSJames Henderson   LT2.setCustomPrologue({{2, LineTable::Long}, {1, LineTable::Half}});
635a3acf99eSJames Henderson   generate();
636a3acf99eSJames Henderson 
637a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
638004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
639a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
640004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
641a3acf99eSJames Henderson 
642a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
643a3acf99eSJames Henderson 
644a3acf99eSJames Henderson   checkError({"parsing line table prologue at offset 0x00000000 found "
645a3acf99eSJames Henderson               "unsupported version 0x00",
646a3acf99eSJames Henderson               "parsing line table prologue at offset 0x00000006 found "
647a3acf99eSJames Henderson               "unsupported version 0x01"},
648004b729eSJames Henderson              std::move(Unrecoverable));
649a3acf99eSJames Henderson }
650a3acf99eSJames Henderson 
651a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserIgnoresNonPrologueErrorsWhenSkipping) {
652a3acf99eSJames Henderson   if (!setupGenerator())
653a3acf99eSJames Henderson     return;
654a3acf99eSJames Henderson 
655a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
656a3acf99eSJames Henderson   LT.addExtendedOpcode(42, DW_LNE_end_sequence, {});
657a3acf99eSJames Henderson   generate();
658a3acf99eSJames Henderson 
659a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
660004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
661a3acf99eSJames Henderson 
662a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
663004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
66466702629SJames Henderson }
66566702629SJames Henderson 
66666702629SJames Henderson } // end anonymous namespace
667