166702629SJames Henderson //===- DWARFDebugLineTest.cpp ---------------------------------------------===//
266702629SJames Henderson //
366702629SJames Henderson //                     The LLVM Compiler Infrastructure
466702629SJames Henderson //
566702629SJames Henderson // This file is distributed under the University of Illinois Open Source
666702629SJames Henderson // License. See LICENSE.TXT for details.
766702629SJames Henderson //
866702629SJames Henderson //===----------------------------------------------------------------------===//
966702629SJames Henderson 
1066702629SJames Henderson #include "DwarfGenerator.h"
1166702629SJames Henderson #include "DwarfUtils.h"
1266702629SJames Henderson #include "llvm/DebugInfo/DWARF/DWARFContext.h"
1366702629SJames Henderson #include "llvm/DebugInfo/DWARF/DWARFDebugLine.h"
1466702629SJames Henderson #include "llvm/Object/ObjectFile.h"
1566702629SJames Henderson #include "llvm/Testing/Support/Error.h"
1666702629SJames Henderson #include "gtest/gtest.h"
1766702629SJames Henderson 
1866702629SJames Henderson using namespace llvm;
1966702629SJames Henderson using namespace dwarf;
20a3acf99eSJames Henderson using namespace dwarfgen;
2166702629SJames Henderson using namespace object;
2266702629SJames Henderson using namespace utils;
23a3acf99eSJames Henderson using namespace testing;
2466702629SJames Henderson 
2566702629SJames Henderson namespace {
26a3acf99eSJames Henderson struct CommonFixture {
27a3acf99eSJames Henderson   CommonFixture()
28004b729eSJames Henderson       : LineData("", true, 0), Recoverable(Error::success()),
29004b729eSJames Henderson         RecordRecoverable(std::bind(&CommonFixture::recordRecoverable, this,
30a3acf99eSJames Henderson                                     std::placeholders::_1)),
31004b729eSJames Henderson         Unrecoverable(Error::success()),
32004b729eSJames Henderson         RecordUnrecoverable(std::bind(&CommonFixture::recordUnrecoverable, this,
33a3acf99eSJames Henderson                                       std::placeholders::_1)){};
3466702629SJames Henderson 
35004b729eSJames Henderson   ~CommonFixture() {
36004b729eSJames Henderson     EXPECT_FALSE(Recoverable);
37004b729eSJames Henderson     EXPECT_FALSE(Unrecoverable);
38004b729eSJames Henderson   }
39a3acf99eSJames Henderson 
40a3acf99eSJames Henderson   bool setupGenerator(uint16_t Version = 4) {
4166702629SJames Henderson     Triple T = getHostTripleForAddrSize(8);
4266702629SJames Henderson     if (!isConfigurationSupported(T))
4366702629SJames Henderson       return false;
44a3acf99eSJames Henderson     auto ExpectedGenerator = Generator::create(T, Version);
4566702629SJames Henderson     if (ExpectedGenerator)
46a3acf99eSJames Henderson       Gen.reset(ExpectedGenerator->release());
4766702629SJames Henderson     return true;
4866702629SJames Henderson   }
4966702629SJames Henderson 
50a3acf99eSJames Henderson   void generate() {
51a3acf99eSJames Henderson     Context = createContext();
52a3acf99eSJames Henderson     assert(Context != nullptr && "test state is not valid");
53a3acf99eSJames Henderson     const DWARFObject &Obj = Context->getDWARFObj();
54198a87c8SJames Henderson     LineData = DWARFDataExtractor(Obj, Obj.getLineSection(),
55198a87c8SJames Henderson                                   sys::IsLittleEndianHost, 8);
56a3acf99eSJames Henderson   }
57a3acf99eSJames Henderson 
5866702629SJames Henderson   std::unique_ptr<DWARFContext> createContext() {
59a3acf99eSJames Henderson     if (!Gen)
6066702629SJames Henderson       return nullptr;
61a3acf99eSJames Henderson     StringRef FileBytes = Gen->generate();
6266702629SJames Henderson     MemoryBufferRef FileBuffer(FileBytes, "dwarf");
6366702629SJames Henderson     auto Obj = object::ObjectFile::createObjectFile(FileBuffer);
6466702629SJames Henderson     if (Obj)
6566702629SJames Henderson       return DWARFContext::create(**Obj);
6666702629SJames Henderson     return nullptr;
6766702629SJames Henderson   }
6866702629SJames Henderson 
69a3acf99eSJames Henderson   DWARFDebugLine::SectionParser setupParser() {
70a3acf99eSJames Henderson     LineTable &LT = Gen->addLineTable(DWARF32);
71a3acf99eSJames Henderson     LT.addExtendedOpcode(9, DW_LNE_set_address, {{0xadd4e55, LineTable::Quad}});
72a3acf99eSJames Henderson     LT.addStandardOpcode(DW_LNS_copy, {});
73a3acf99eSJames Henderson     LT.addByte(0xaa);
74a3acf99eSJames Henderson     LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
75a3acf99eSJames Henderson 
76a3acf99eSJames Henderson     LineTable &LT2 = Gen->addLineTable(DWARF64);
77a3acf99eSJames Henderson     LT2.addExtendedOpcode(9, DW_LNE_set_address,
78a3acf99eSJames Henderson                           {{0x11223344, LineTable::Quad}});
79a3acf99eSJames Henderson     LT2.addStandardOpcode(DW_LNS_copy, {});
80a3acf99eSJames Henderson     LT2.addByte(0xbb);
81a3acf99eSJames Henderson     LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
82a3acf99eSJames Henderson 
83a3acf99eSJames Henderson     generate();
84a3acf99eSJames Henderson 
85a3acf99eSJames Henderson     return DWARFDebugLine::SectionParser(LineData, *Context, CUs, TUs);
86a3acf99eSJames Henderson   }
87a3acf99eSJames Henderson 
88004b729eSJames Henderson   void recordRecoverable(Error Err) {
89004b729eSJames Henderson     Recoverable = joinErrors(std::move(Recoverable), std::move(Err));
90004b729eSJames Henderson   }
91004b729eSJames Henderson   void recordUnrecoverable(Error Err) {
92004b729eSJames Henderson     Unrecoverable = joinErrors(std::move(Unrecoverable), std::move(Err));
93a3acf99eSJames Henderson   }
94a3acf99eSJames Henderson 
95a3acf99eSJames Henderson   void checkError(ArrayRef<StringRef> ExpectedMsgs, Error Err) {
96a3acf99eSJames Henderson     ASSERT_TRUE(Err.operator bool());
97a3acf99eSJames Henderson     size_t WhichMsg = 0;
98a3acf99eSJames Henderson     Error Remaining =
99a3acf99eSJames Henderson         handleErrors(std::move(Err), [&](const ErrorInfoBase &Actual) {
100a3acf99eSJames Henderson           ASSERT_LT(WhichMsg, ExpectedMsgs.size());
101a3acf99eSJames Henderson           // Use .str(), because googletest doesn't visualise a StringRef
102a3acf99eSJames Henderson           // properly.
103a3acf99eSJames Henderson           EXPECT_EQ(Actual.message(), ExpectedMsgs[WhichMsg++].str());
104a3acf99eSJames Henderson         });
105a3acf99eSJames Henderson     EXPECT_EQ(WhichMsg, ExpectedMsgs.size());
106a3acf99eSJames Henderson     EXPECT_FALSE(Remaining);
107a3acf99eSJames Henderson   }
108a3acf99eSJames Henderson 
109a3acf99eSJames Henderson   void checkError(StringRef ExpectedMsg, Error Err) {
110a3acf99eSJames Henderson     checkError(ArrayRef<StringRef>{ExpectedMsg}, std::move(Err));
111a3acf99eSJames Henderson   }
112a3acf99eSJames Henderson 
113a3acf99eSJames Henderson   void checkGetOrParseLineTableEmitsError(StringRef ExpectedMsg,
114a3acf99eSJames Henderson                                           uint64_t Offset = 0) {
115a3acf99eSJames Henderson     auto ExpectedLineTable = Line.getOrParseLineTable(
116004b729eSJames Henderson         LineData, Offset, *Context, nullptr, RecordRecoverable);
117a3acf99eSJames Henderson     EXPECT_FALSE(ExpectedLineTable);
118004b729eSJames Henderson     EXPECT_FALSE(Recoverable);
119a3acf99eSJames Henderson 
120a3acf99eSJames Henderson     checkError(ExpectedMsg, ExpectedLineTable.takeError());
121a3acf99eSJames Henderson   }
122a3acf99eSJames Henderson 
123a3acf99eSJames Henderson   std::unique_ptr<Generator> Gen;
124a3acf99eSJames Henderson   std::unique_ptr<DWARFContext> Context;
125a3acf99eSJames Henderson   DWARFDataExtractor LineData;
126a3acf99eSJames Henderson   DWARFDebugLine Line;
127004b729eSJames Henderson   Error Recoverable;
128004b729eSJames Henderson   std::function<void(Error)> RecordRecoverable;
129004b729eSJames Henderson   Error Unrecoverable;
130004b729eSJames Henderson   std::function<void(Error)> RecordUnrecoverable;
131a3acf99eSJames Henderson 
132*143eaeabSPaul Robinson   SmallVector<std::unique_ptr<DWARFUnit>, 2> CUs;
133*143eaeabSPaul Robinson   std::deque<DWARFUnitSection> TUs;
13466702629SJames Henderson };
13566702629SJames Henderson 
136a3acf99eSJames Henderson // Fixtures must derive from "Test", but parameterised fixtures from
137a3acf99eSJames Henderson // "TestWithParam". It does not seem possible to inherit from both, so we share
138a3acf99eSJames Henderson // the common state in a separate class, inherited by the two fixture classes.
139a3acf99eSJames Henderson struct DebugLineBasicFixture : public Test, public CommonFixture {};
140a3acf99eSJames Henderson 
141a3acf99eSJames Henderson struct DebugLineParameterisedFixture
142a3acf99eSJames Henderson     : public TestWithParam<std::pair<uint16_t, DwarfFormat>>,
143a3acf99eSJames Henderson       public CommonFixture {
144a3acf99eSJames Henderson   void SetUp() { std::tie(Version, Format) = GetParam(); }
145a3acf99eSJames Henderson 
146a3acf99eSJames Henderson   uint16_t Version;
147a3acf99eSJames Henderson   DwarfFormat Format;
148a3acf99eSJames Henderson };
149a3acf99eSJames Henderson 
150a3acf99eSJames Henderson void checkDefaultPrologue(uint16_t Version, DwarfFormat Format,
151a3acf99eSJames Henderson                           DWARFDebugLine::Prologue Prologue,
152a3acf99eSJames Henderson                           uint64_t BodyLength) {
153a3acf99eSJames Henderson   // Check version specific fields and values.
154a3acf99eSJames Henderson   uint64_t UnitLength;
155a3acf99eSJames Henderson   uint64_t PrologueLength;
156a3acf99eSJames Henderson   switch (Version) {
157a3acf99eSJames Henderson   case 4:
158a3acf99eSJames Henderson     PrologueLength = 36;
159a3acf99eSJames Henderson     UnitLength = PrologueLength + 2;
16011a9de74SJames Henderson     EXPECT_EQ(Prologue.MaxOpsPerInst, 1u);
161a3acf99eSJames Henderson     break;
162a3acf99eSJames Henderson   case 2:
163a3acf99eSJames Henderson   case 3:
164a3acf99eSJames Henderson     PrologueLength = 35;
165a3acf99eSJames Henderson     UnitLength = PrologueLength + 2;
166a3acf99eSJames Henderson     break;
167a3acf99eSJames Henderson   case 5:
168a3acf99eSJames Henderson     PrologueLength = 39;
169a3acf99eSJames Henderson     UnitLength = PrologueLength + 4;
17011a9de74SJames Henderson     EXPECT_EQ(Prologue.getAddressSize(), 8u);
17111a9de74SJames Henderson     EXPECT_EQ(Prologue.SegSelectorSize, 0u);
172a3acf99eSJames Henderson     break;
173a3acf99eSJames Henderson   default:
174a3acf99eSJames Henderson     llvm_unreachable("unsupported DWARF version");
175a3acf99eSJames Henderson   }
176a3acf99eSJames Henderson   UnitLength += BodyLength + (Format == DWARF32 ? 4 : 8);
177a3acf99eSJames Henderson 
178a3acf99eSJames Henderson   EXPECT_EQ(Prologue.TotalLength, UnitLength);
179a3acf99eSJames Henderson   EXPECT_EQ(Prologue.PrologueLength, PrologueLength);
18011a9de74SJames Henderson   EXPECT_EQ(Prologue.MinInstLength, 1u);
18111a9de74SJames Henderson   EXPECT_EQ(Prologue.DefaultIsStmt, 1u);
182a3acf99eSJames Henderson   EXPECT_EQ(Prologue.LineBase, -5);
18311a9de74SJames Henderson   EXPECT_EQ(Prologue.LineRange, 14u);
18411a9de74SJames Henderson   EXPECT_EQ(Prologue.OpcodeBase, 13u);
185a3acf99eSJames Henderson   std::vector<uint8_t> ExpectedLengths = {0, 1, 1, 1, 1, 0, 0, 0, 1, 0, 0, 1};
186a3acf99eSJames Henderson   EXPECT_EQ(Prologue.StandardOpcodeLengths, ExpectedLengths);
18711a9de74SJames Henderson   ASSERT_EQ(Prologue.IncludeDirectories.size(), 1u);
188a3acf99eSJames Henderson   ASSERT_EQ(Prologue.IncludeDirectories[0].getForm(), DW_FORM_string);
189a3acf99eSJames Henderson   EXPECT_STREQ(*Prologue.IncludeDirectories[0].getAsCString(), "a dir");
19011a9de74SJames Henderson   ASSERT_EQ(Prologue.FileNames.size(), 1u);
191a3acf99eSJames Henderson   ASSERT_EQ(Prologue.FileNames[0].Name.getForm(), DW_FORM_string);
192a3acf99eSJames Henderson   EXPECT_STREQ(*Prologue.FileNames[0].Name.getAsCString(), "a file");
193a3acf99eSJames Henderson }
194a3acf99eSJames Henderson 
195a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, GetOrParseLineTableAtInvalidOffset) {
196a3acf99eSJames Henderson   if (!setupGenerator())
197a3acf99eSJames Henderson     return;
198a3acf99eSJames Henderson   generate();
199a3acf99eSJames Henderson 
200a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
201a3acf99eSJames Henderson       "offset 0x00000000 is not a valid debug line section offset", 0);
202a3acf99eSJames Henderson   // Repeat to show that an error is reported each time.
203a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
204a3acf99eSJames Henderson       "offset 0x00000000 is not a valid debug line section offset", 0);
205a3acf99eSJames Henderson   // Show that an error is reported for later offsets too.
206a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
207a3acf99eSJames Henderson       "offset 0x00000001 is not a valid debug line section offset", 1);
208a3acf99eSJames Henderson }
209a3acf99eSJames Henderson 
210a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, GetOrParseLineTableAtInvalidOffsetAfterData) {
211a3acf99eSJames Henderson   if (!setupGenerator())
21266702629SJames Henderson     return;
21366702629SJames Henderson 
214a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
215a3acf99eSJames Henderson   LT.setCustomPrologue({{0, LineTable::Byte}});
21666702629SJames Henderson 
217a3acf99eSJames Henderson   generate();
218a3acf99eSJames Henderson 
219a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
220a3acf99eSJames Henderson       "offset 0x00000001 is not a valid debug line section offset", 1);
221a3acf99eSJames Henderson }
222a3acf99eSJames Henderson 
223a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, GetOrParseLineTableValidTable) {
224a3acf99eSJames Henderson   if (!setupGenerator(Version))
225a3acf99eSJames Henderson     return;
226a3acf99eSJames Henderson 
227a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
228a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
229a3acf99eSJames Henderson 
230a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
231a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address, {{0xadd4e55, LineTable::Quad}});
232a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
233a3acf99eSJames Henderson   LT.addByte(0xaa);
234a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
235a3acf99eSJames Henderson 
236a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(Format);
237a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address, {{0x11223344, LineTable::Quad}});
238a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
239a3acf99eSJames Henderson   LT2.addByte(0xbb);
240a3acf99eSJames Henderson   LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
241a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address, {{0x55667788, LineTable::Quad}});
242a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
243a3acf99eSJames Henderson   LT2.addByte(0xcc);
244a3acf99eSJames Henderson   LT2.addExtendedOpcode(1, DW_LNE_end_sequence, {});
245a3acf99eSJames Henderson 
246a3acf99eSJames Henderson   generate();
247a3acf99eSJames Henderson 
248004b729eSJames Henderson   auto ExpectedLineTable = Line.getOrParseLineTable(LineData, 0, *Context,
249004b729eSJames Henderson                                                     nullptr, RecordRecoverable);
250a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable.operator bool());
251004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
252a3acf99eSJames Henderson   const DWARFDebugLine::LineTable *Expected = *ExpectedLineTable;
253a3acf99eSJames Henderson   checkDefaultPrologue(Version, Format, Expected->Prologue, 16);
254a68aee07SRoman Lebedev   EXPECT_EQ(Expected->Sequences.size(), 1u);
255a3acf99eSJames Henderson 
256a3acf99eSJames Henderson   uint64_t SecondOffset =
257a3acf99eSJames Henderson       Expected->Prologue.sizeofTotalLength() + Expected->Prologue.TotalLength;
258004b729eSJames Henderson   Recoverable = Error::success();
259a3acf99eSJames Henderson   auto ExpectedLineTable2 = Line.getOrParseLineTable(
260004b729eSJames Henderson       LineData, SecondOffset, *Context, nullptr, RecordRecoverable);
261a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable2.operator bool());
262004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
263a3acf99eSJames Henderson   const DWARFDebugLine::LineTable *Expected2 = *ExpectedLineTable2;
264a3acf99eSJames Henderson   checkDefaultPrologue(Version, Format, Expected2->Prologue, 32);
26511a9de74SJames Henderson   EXPECT_EQ(Expected2->Sequences.size(), 2u);
266a3acf99eSJames Henderson 
267a3acf99eSJames Henderson   EXPECT_NE(Expected, Expected2);
268a3acf99eSJames Henderson 
269a3acf99eSJames Henderson   // Check that if the same offset is requested, the exact same pointer is
270a3acf99eSJames Henderson   // returned.
271004b729eSJames Henderson   Recoverable = Error::success();
272004b729eSJames Henderson   auto ExpectedLineTable3 = Line.getOrParseLineTable(
273004b729eSJames Henderson       LineData, 0, *Context, nullptr, RecordRecoverable);
274a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable3.operator bool());
275004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
276a3acf99eSJames Henderson   EXPECT_EQ(Expected, *ExpectedLineTable3);
277a3acf99eSJames Henderson 
278004b729eSJames Henderson   Recoverable = Error::success();
279a3acf99eSJames Henderson   auto ExpectedLineTable4 = Line.getOrParseLineTable(
280004b729eSJames Henderson       LineData, SecondOffset, *Context, nullptr, RecordRecoverable);
281a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable4.operator bool());
282004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
283a3acf99eSJames Henderson   EXPECT_EQ(Expected2, *ExpectedLineTable4);
284a3acf99eSJames Henderson 
285a3acf99eSJames Henderson   // TODO: Add tests that show that the body of the programs have been read
286a3acf99eSJames Henderson   // correctly.
287a3acf99eSJames Henderson }
288a3acf99eSJames Henderson 
289a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForReservedLength) {
290a3acf99eSJames Henderson   if (!setupGenerator())
291a3acf99eSJames Henderson     return;
292a3acf99eSJames Henderson 
293a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
294a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
295a3acf99eSJames Henderson 
296a3acf99eSJames Henderson   generate();
297a3acf99eSJames Henderson 
298a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
299a3acf99eSJames Henderson       "parsing line table prologue at offset 0x00000000 unsupported reserved "
300a3acf99eSJames Henderson       "unit length found of value 0xffffff00");
301a3acf99eSJames Henderson }
302a3acf99eSJames Henderson 
303a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForLowVersion) {
304a3acf99eSJames Henderson   if (!setupGenerator())
305a3acf99eSJames Henderson     return;
306a3acf99eSJames Henderson 
307a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
308a3acf99eSJames Henderson   LT.setCustomPrologue(
309a3acf99eSJames Henderson       {{LineTable::Half, LineTable::Long}, {1, LineTable::Half}});
310a3acf99eSJames Henderson 
311a3acf99eSJames Henderson   generate();
312a3acf99eSJames Henderson 
313a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError("parsing line table prologue at offset "
314a3acf99eSJames Henderson                                      "0x00000000 found unsupported version "
315a3acf99eSJames Henderson                                      "0x01");
316a3acf99eSJames Henderson }
317a3acf99eSJames Henderson 
318a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForInvalidV5IncludeDirTable) {
319a3acf99eSJames Henderson   if (!setupGenerator(5))
320a3acf99eSJames Henderson     return;
321a3acf99eSJames Henderson 
322a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
323a3acf99eSJames Henderson   LT.setCustomPrologue({
324a3acf99eSJames Henderson       {19, LineTable::Long}, // unit length
325a3acf99eSJames Henderson       {5, LineTable::Half},  // version
326a3acf99eSJames Henderson       {8, LineTable::Byte},  // addr size
327a3acf99eSJames Henderson       {0, LineTable::Byte},  // segment selector size
328a3acf99eSJames Henderson       {11, LineTable::Long}, // prologue length
329a3acf99eSJames Henderson       {1, LineTable::Byte},  // min instruction length
330a3acf99eSJames Henderson       {1, LineTable::Byte},  // max ops per instruction
331a3acf99eSJames Henderson       {1, LineTable::Byte},  // default is_stmt
332a3acf99eSJames Henderson       {0, LineTable::Byte},  // line base
333a3acf99eSJames Henderson       {14, LineTable::Byte}, // line range
334a3acf99eSJames Henderson       {2, LineTable::Byte},  // opcode base (small to reduce the amount of
335a3acf99eSJames Henderson                              // setup required).
336a3acf99eSJames Henderson       {0, LineTable::Byte},  // standard opcode lengths
337a3acf99eSJames Henderson       {0, LineTable::Byte},  // directory entry format count (should not be
338a3acf99eSJames Henderson                              // zero).
339a3acf99eSJames Henderson       {0, LineTable::ULEB},  // directories count
340a3acf99eSJames Henderson       {0, LineTable::Byte},  // file name entry format count
341a3acf99eSJames Henderson       {0, LineTable::ULEB}   // file name entry count
342a3acf99eSJames Henderson   });
343a3acf99eSJames Henderson 
344a3acf99eSJames Henderson   generate();
345a3acf99eSJames Henderson 
346a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
347a3acf99eSJames Henderson       "parsing line table prologue at 0x00000000 found an invalid directory or "
348a3acf99eSJames Henderson       "file table description at 0x00000014");
349a3acf99eSJames Henderson }
350a3acf99eSJames Henderson 
351a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, ErrorForTooLargePrologueLength) {
352a3acf99eSJames Henderson   if (!setupGenerator(Version))
353a3acf99eSJames Henderson     return;
354a3acf99eSJames Henderson 
355a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
356a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
357a3acf99eSJames Henderson 
358a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
359a3acf99eSJames Henderson   DWARFDebugLine::Prologue Prologue = LT.createBasicPrologue();
360a3acf99eSJames Henderson   ++Prologue.PrologueLength;
361a3acf99eSJames Henderson   LT.setPrologue(Prologue);
362a3acf99eSJames Henderson 
363a3acf99eSJames Henderson   generate();
364a3acf99eSJames Henderson 
365a3acf99eSJames Henderson   uint64_t ExpectedEnd =
366a3acf99eSJames Henderson       Prologue.TotalLength + 1 + Prologue.sizeofTotalLength();
367a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
368a3acf99eSJames Henderson       (Twine("parsing line table prologue at 0x00000000 should have ended at "
369a3acf99eSJames Henderson              "0x000000") +
370a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd) + " but it ended at 0x000000" +
371a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd - 1))
372a3acf99eSJames Henderson           .str());
373a3acf99eSJames Henderson }
374a3acf99eSJames Henderson 
375a3acf99eSJames Henderson TEST_P(DebugLineParameterisedFixture, ErrorForTooShortPrologueLength) {
376a3acf99eSJames Henderson   if (!setupGenerator(Version))
377a3acf99eSJames Henderson     return;
378a3acf99eSJames Henderson 
379a3acf99eSJames Henderson   SCOPED_TRACE("Checking Version " + std::to_string(Version) + ", Format " +
380a3acf99eSJames Henderson                (Format == DWARF64 ? "DWARF64" : "DWARF32"));
381a3acf99eSJames Henderson 
382a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(Format);
383a3acf99eSJames Henderson   DWARFDebugLine::Prologue Prologue = LT.createBasicPrologue();
384a3acf99eSJames Henderson   // FIXME: Ideally, we'd test for 1 less than expected, but the code does not
385a3acf99eSJames Henderson   // currently fail if missing only the terminator of a v2-4 file table.
386a3acf99eSJames Henderson   if (Version < 5)
387a3acf99eSJames Henderson     Prologue.PrologueLength -= 2;
388a3acf99eSJames Henderson   else
389a3acf99eSJames Henderson     Prologue.PrologueLength -= 1;
390a3acf99eSJames Henderson   LT.setPrologue(Prologue);
391a3acf99eSJames Henderson 
392a3acf99eSJames Henderson   generate();
393a3acf99eSJames Henderson 
394a3acf99eSJames Henderson   uint64_t ExpectedEnd =
395a3acf99eSJames Henderson       Prologue.TotalLength - 1 + Prologue.sizeofTotalLength();
396a3acf99eSJames Henderson   if (Version < 5)
397a3acf99eSJames Henderson     --ExpectedEnd;
398a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
399a3acf99eSJames Henderson       (Twine("parsing line table prologue at 0x00000000 should have ended at "
400a3acf99eSJames Henderson              "0x000000") +
401a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd) + " but it ended at 0x000000" +
402a3acf99eSJames Henderson        Twine::utohexstr(ExpectedEnd + 1))
403a3acf99eSJames Henderson           .str());
404a3acf99eSJames Henderson }
405a3acf99eSJames Henderson 
406a3acf99eSJames Henderson INSTANTIATE_TEST_CASE_P(
407a3acf99eSJames Henderson     LineTableTestParams, DebugLineParameterisedFixture,
408a3acf99eSJames Henderson     Values(std::make_pair(
409a3acf99eSJames Henderson                2, DWARF32), // Test lower-bound of v2-3 fields and DWARF32.
410a3acf99eSJames Henderson            std::make_pair(3, DWARF32), // Test upper-bound of v2-3 fields.
411a3acf99eSJames Henderson            std::make_pair(4, DWARF64), // Test v4 fields and DWARF64.
412a3acf99eSJames Henderson            std::make_pair(5, DWARF32), std::make_pair(5, DWARF64)), );
413a3acf99eSJames Henderson 
414a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForInvalidExtendedOpcodeLength) {
415a3acf99eSJames Henderson   if (!setupGenerator())
416a3acf99eSJames Henderson     return;
417a3acf99eSJames Henderson 
418a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
419a3acf99eSJames Henderson   // The Length should be 1 for an end sequence opcode.
420a3acf99eSJames Henderson   LT.addExtendedOpcode(2, DW_LNE_end_sequence, {});
421a3acf99eSJames Henderson 
422a3acf99eSJames Henderson   generate();
423a3acf99eSJames Henderson 
424a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError("unexpected line op length at offset "
425a3acf99eSJames Henderson                                      "0x00000030 expected 0x02 found 0x01");
426a3acf99eSJames Henderson }
427a3acf99eSJames Henderson 
428a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ErrorForMismatchedAddressSize) {
429a3acf99eSJames Henderson   if (!setupGenerator())
430a3acf99eSJames Henderson     return;
431a3acf99eSJames Henderson 
432a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
433a3acf99eSJames Henderson   // The line data extractor expects size 8 (Quad) addresses.
434a3acf99eSJames Henderson   LT.addExtendedOpcode(5, DW_LNE_set_address, {{0x11223344, LineTable::Long}});
435a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
436a3acf99eSJames Henderson   LT.addByte(0xaa);
437a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
438a3acf99eSJames Henderson 
439a3acf99eSJames Henderson   generate();
440a3acf99eSJames Henderson 
441a3acf99eSJames Henderson   checkGetOrParseLineTableEmitsError(
442a3acf99eSJames Henderson       "mismatching address size at offset 0x00000030 expected 0x08 found 0x04");
443a3acf99eSJames Henderson }
444a3acf99eSJames Henderson 
445a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, CallbackUsedForUnterminatedSequence) {
446a3acf99eSJames Henderson   if (!setupGenerator())
447a3acf99eSJames Henderson     return;
448a3acf99eSJames Henderson 
449a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
450a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address,
451a3acf99eSJames Henderson                        {{0x1122334455667788, LineTable::Quad}});
452a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
453a3acf99eSJames Henderson   LT.addByte(0xaa);
454a3acf99eSJames Henderson   LT.addExtendedOpcode(1, DW_LNE_end_sequence, {});
455a3acf99eSJames Henderson   LT.addExtendedOpcode(9, DW_LNE_set_address,
456a3acf99eSJames Henderson                        {{0x99aabbccddeeff00, LineTable::Quad}});
457a3acf99eSJames Henderson   LT.addStandardOpcode(DW_LNS_copy, {});
458a3acf99eSJames Henderson   LT.addByte(0xbb);
459a3acf99eSJames Henderson   LT.addByte(0xcc);
460a3acf99eSJames Henderson 
461a3acf99eSJames Henderson   generate();
462a3acf99eSJames Henderson 
463004b729eSJames Henderson   auto ExpectedLineTable = Line.getOrParseLineTable(LineData, 0, *Context,
464004b729eSJames Henderson                                                     nullptr, RecordRecoverable);
465004b729eSJames Henderson   checkError("last sequence in debug line table is not terminated!",
466004b729eSJames Henderson              std::move(Recoverable));
467a3acf99eSJames Henderson   ASSERT_TRUE(ExpectedLineTable.operator bool());
468a68aee07SRoman Lebedev   EXPECT_EQ((*ExpectedLineTable)->Rows.size(), 6u);
469a3acf99eSJames Henderson   // The unterminated sequence is not added to the sequence list.
470a68aee07SRoman Lebedev   EXPECT_EQ((*ExpectedLineTable)->Sequences.size(), 1u);
471a3acf99eSJames Henderson }
472a3acf99eSJames Henderson 
473a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserParsesCorrectly) {
474a3acf99eSJames Henderson   if (!setupGenerator())
475a3acf99eSJames Henderson     return;
476a3acf99eSJames Henderson 
477a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser = setupParser();
478a3acf99eSJames Henderson 
47911a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 0u);
480a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
481a3acf99eSJames Henderson 
482004b729eSJames Henderson   DWARFDebugLine::LineTable Parsed =
483004b729eSJames Henderson       Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
484a3acf99eSJames Henderson   checkDefaultPrologue(4, DWARF32, Parsed.Prologue, 16);
48511a9de74SJames Henderson   EXPECT_EQ(Parsed.Sequences.size(), 1u);
48611a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 62u);
487a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
488a3acf99eSJames Henderson 
489a3acf99eSJames Henderson   DWARFDebugLine::LineTable Parsed2 =
490004b729eSJames Henderson       Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
491a3acf99eSJames Henderson   checkDefaultPrologue(4, DWARF64, Parsed2.Prologue, 16);
49211a9de74SJames Henderson   EXPECT_EQ(Parsed2.Sequences.size(), 1u);
49311a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 136u);
494a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
495a3acf99eSJames Henderson 
496004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
497004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
498a3acf99eSJames Henderson }
499a3acf99eSJames Henderson 
500a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserSkipsCorrectly) {
501a3acf99eSJames Henderson   if (!setupGenerator())
502a3acf99eSJames Henderson     return;
503a3acf99eSJames Henderson 
504a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser = setupParser();
505a3acf99eSJames Henderson 
50611a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 0u);
507a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
508a3acf99eSJames Henderson 
509004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
51011a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 62u);
511a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
512a3acf99eSJames Henderson 
513004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
51411a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 136u);
515a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
516a3acf99eSJames Henderson 
517004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
518a3acf99eSJames Henderson }
519a3acf99eSJames Henderson 
520a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserAlwaysDoneForEmptySection) {
521a3acf99eSJames Henderson   if (!setupGenerator())
522a3acf99eSJames Henderson     return;
523a3acf99eSJames Henderson 
524a3acf99eSJames Henderson   generate();
525a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
526a3acf99eSJames Henderson 
527a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
528a3acf99eSJames Henderson }
529a3acf99eSJames Henderson 
530a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserMovesToEndForBadLengthWhenParsing) {
531a3acf99eSJames Henderson   if (!setupGenerator())
532a3acf99eSJames Henderson     return;
533a3acf99eSJames Henderson 
534a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
535a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
536a3acf99eSJames Henderson   Gen->addLineTable();
537a3acf99eSJames Henderson   generate();
538a3acf99eSJames Henderson 
539a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
540004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
541a3acf99eSJames Henderson 
54211a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 4u);
543a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
544004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
545a3acf99eSJames Henderson 
546a3acf99eSJames Henderson   checkError("parsing line table prologue at offset 0x00000000 unsupported "
547a3acf99eSJames Henderson              "reserved unit length found of value 0xffffff00",
548004b729eSJames Henderson              std::move(Unrecoverable));
549a3acf99eSJames Henderson }
550a3acf99eSJames Henderson 
551a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserMovesToEndForBadLengthWhenSkipping) {
552a3acf99eSJames Henderson   if (!setupGenerator())
553a3acf99eSJames Henderson     return;
554a3acf99eSJames Henderson 
555a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable();
556a3acf99eSJames Henderson   LT.setCustomPrologue({{0xffffff00, LineTable::Long}});
557a3acf99eSJames Henderson   Gen->addLineTable();
558a3acf99eSJames Henderson   generate();
559a3acf99eSJames Henderson 
560a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
561004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
562a3acf99eSJames Henderson 
56311a9de74SJames Henderson   EXPECT_EQ(Parser.getOffset(), 4u);
564a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
565a3acf99eSJames Henderson 
566a3acf99eSJames Henderson   checkError("parsing line table prologue at offset 0x00000000 unsupported "
567a3acf99eSJames Henderson              "reserved unit length found of value 0xffffff00",
568004b729eSJames Henderson              std::move(Unrecoverable));
569a3acf99eSJames Henderson }
570a3acf99eSJames Henderson 
571a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserReportsFirstErrorInEachTableWhenParsing) {
572a3acf99eSJames Henderson   if (!setupGenerator())
573a3acf99eSJames Henderson     return;
574a3acf99eSJames Henderson 
575a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
576a3acf99eSJames Henderson   LT.setCustomPrologue({{2, LineTable::Long}, {0, LineTable::Half}});
577a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
578a3acf99eSJames Henderson   LT2.setCustomPrologue({{2, LineTable::Long}, {1, LineTable::Half}});
579a3acf99eSJames Henderson   generate();
580a3acf99eSJames Henderson 
581a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
582004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
583a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
584004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
585a3acf99eSJames Henderson 
586a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
587004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
588a3acf99eSJames Henderson 
589a3acf99eSJames Henderson   checkError({"parsing line table prologue at offset 0x00000000 found "
590a3acf99eSJames Henderson               "unsupported version 0x00",
591a3acf99eSJames Henderson               "parsing line table prologue at offset 0x00000006 found "
592a3acf99eSJames Henderson               "unsupported version 0x01"},
593004b729eSJames Henderson              std::move(Unrecoverable));
594a3acf99eSJames Henderson }
595a3acf99eSJames Henderson 
596a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserReportsNonPrologueProblemsWhenParsing) {
597a3acf99eSJames Henderson   if (!setupGenerator())
598a3acf99eSJames Henderson     return;
599a3acf99eSJames Henderson 
600a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
601a3acf99eSJames Henderson   LT.addExtendedOpcode(0x42, DW_LNE_end_sequence, {});
602a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
603a3acf99eSJames Henderson   LT2.addExtendedOpcode(9, DW_LNE_set_address,
604a3acf99eSJames Henderson                         {{0x1234567890abcdef, LineTable::Quad}});
605a3acf99eSJames Henderson   LT2.addStandardOpcode(DW_LNS_copy, {});
606a3acf99eSJames Henderson   LT2.addByte(0xbb);
607a3acf99eSJames Henderson   generate();
608a3acf99eSJames Henderson 
609a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
610004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
611004b729eSJames Henderson   EXPECT_FALSE(Recoverable);
612a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
613a3acf99eSJames Henderson   checkError(
614a3acf99eSJames Henderson       "unexpected line op length at offset 0x00000030 expected 0x42 found 0x01",
615004b729eSJames Henderson       std::move(Unrecoverable));
616a3acf99eSJames Henderson 
617a3acf99eSJames Henderson   // Reset the error state so that it does not confuse the next set of checks.
618004b729eSJames Henderson   Unrecoverable = Error::success();
619004b729eSJames Henderson   Parser.parseNext(RecordRecoverable, RecordUnrecoverable);
620a3acf99eSJames Henderson 
621a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
622004b729eSJames Henderson   checkError("last sequence in debug line table is not terminated!",
623004b729eSJames Henderson              std::move(Recoverable));
624004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
625a3acf99eSJames Henderson }
626a3acf99eSJames Henderson 
627a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture,
628a3acf99eSJames Henderson        ParserReportsPrologueErrorsInEachTableWhenSkipping) {
629a3acf99eSJames Henderson   if (!setupGenerator())
630a3acf99eSJames Henderson     return;
631a3acf99eSJames Henderson 
632a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
633a3acf99eSJames Henderson   LT.setCustomPrologue({{2, LineTable::Long}, {0, LineTable::Half}});
634a3acf99eSJames Henderson   LineTable &LT2 = Gen->addLineTable(DWARF32);
635a3acf99eSJames Henderson   LT2.setCustomPrologue({{2, LineTable::Long}, {1, LineTable::Half}});
636a3acf99eSJames Henderson   generate();
637a3acf99eSJames Henderson 
638a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
639004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
640a3acf99eSJames Henderson   ASSERT_FALSE(Parser.done());
641004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
642a3acf99eSJames Henderson 
643a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
644a3acf99eSJames Henderson 
645a3acf99eSJames Henderson   checkError({"parsing line table prologue at offset 0x00000000 found "
646a3acf99eSJames Henderson               "unsupported version 0x00",
647a3acf99eSJames Henderson               "parsing line table prologue at offset 0x00000006 found "
648a3acf99eSJames Henderson               "unsupported version 0x01"},
649004b729eSJames Henderson              std::move(Unrecoverable));
650a3acf99eSJames Henderson }
651a3acf99eSJames Henderson 
652a3acf99eSJames Henderson TEST_F(DebugLineBasicFixture, ParserIgnoresNonPrologueErrorsWhenSkipping) {
653a3acf99eSJames Henderson   if (!setupGenerator())
654a3acf99eSJames Henderson     return;
655a3acf99eSJames Henderson 
656a3acf99eSJames Henderson   LineTable &LT = Gen->addLineTable(DWARF32);
657a3acf99eSJames Henderson   LT.addExtendedOpcode(42, DW_LNE_end_sequence, {});
658a3acf99eSJames Henderson   generate();
659a3acf99eSJames Henderson 
660a3acf99eSJames Henderson   DWARFDebugLine::SectionParser Parser(LineData, *Context, CUs, TUs);
661004b729eSJames Henderson   Parser.skip(RecordUnrecoverable);
662a3acf99eSJames Henderson 
663a3acf99eSJames Henderson   EXPECT_TRUE(Parser.done());
664004b729eSJames Henderson   EXPECT_FALSE(Unrecoverable);
66566702629SJames Henderson }
66666702629SJames Henderson 
66766702629SJames Henderson } // end anonymous namespace
668