xref: /freebsd/contrib/llvm-project/llvm/lib/DebugInfo/DWARF/DWARFDebugFrame.cpp (revision bdd1243df58e60e85101c09001d9812a789b6bc4)
10b57cec5SDimitry Andric //===- DWARFDebugFrame.h - Parsing of .debug_frame ------------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric 
90b57cec5SDimitry Andric #include "llvm/DebugInfo/DWARF/DWARFDebugFrame.h"
100b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h"
110b57cec5SDimitry Andric #include "llvm/ADT/StringExtras.h"
120b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h"
130b57cec5SDimitry Andric #include "llvm/BinaryFormat/Dwarf.h"
1481ad6265SDimitry Andric #include "llvm/DebugInfo/DIContext.h"
1581ad6265SDimitry Andric #include "llvm/DebugInfo/DWARF/DWARFDataExtractor.h"
160b57cec5SDimitry Andric #include "llvm/Support/Compiler.h"
170b57cec5SDimitry Andric #include "llvm/Support/DataExtractor.h"
180b57cec5SDimitry Andric #include "llvm/Support/Errc.h"
190b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
200b57cec5SDimitry Andric #include "llvm/Support/Format.h"
210b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
220b57cec5SDimitry Andric #include <algorithm>
230b57cec5SDimitry Andric #include <cassert>
240b57cec5SDimitry Andric #include <cinttypes>
250b57cec5SDimitry Andric #include <cstdint>
26*bdd1243dSDimitry Andric #include <optional>
270b57cec5SDimitry Andric 
280b57cec5SDimitry Andric using namespace llvm;
290b57cec5SDimitry Andric using namespace dwarf;
300b57cec5SDimitry Andric 
31*bdd1243dSDimitry Andric static void printRegister(raw_ostream &OS, DIDumpOptions DumpOpts,
32e8d8bef9SDimitry Andric                           unsigned RegNum) {
33*bdd1243dSDimitry Andric   if (DumpOpts.GetNameForDWARFReg) {
34*bdd1243dSDimitry Andric     auto RegName = DumpOpts.GetNameForDWARFReg(RegNum, DumpOpts.IsEH);
35*bdd1243dSDimitry Andric     if (!RegName.empty()) {
36e8d8bef9SDimitry Andric       OS << RegName;
37e8d8bef9SDimitry Andric       return;
38e8d8bef9SDimitry Andric     }
39e8d8bef9SDimitry Andric   }
40e8d8bef9SDimitry Andric   OS << "reg" << RegNum;
41e8d8bef9SDimitry Andric }
420b57cec5SDimitry Andric 
43fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createUnspecified() { return {Unspecified}; }
44fe6060f1SDimitry Andric 
45fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createUndefined() { return {Undefined}; }
46fe6060f1SDimitry Andric 
47fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createSame() { return {Same}; }
48fe6060f1SDimitry Andric 
49fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createIsConstant(int32_t Value) {
50*bdd1243dSDimitry Andric   return {Constant, InvalidRegisterNumber, Value, std::nullopt, false};
51fe6060f1SDimitry Andric }
52fe6060f1SDimitry Andric 
53fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createIsCFAPlusOffset(int32_t Offset) {
54*bdd1243dSDimitry Andric   return {CFAPlusOffset, InvalidRegisterNumber, Offset, std::nullopt, false};
55fe6060f1SDimitry Andric }
56fe6060f1SDimitry Andric 
57fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createAtCFAPlusOffset(int32_t Offset) {
58*bdd1243dSDimitry Andric   return {CFAPlusOffset, InvalidRegisterNumber, Offset, std::nullopt, true};
59fe6060f1SDimitry Andric }
60fe6060f1SDimitry Andric 
61fe6060f1SDimitry Andric UnwindLocation
62fe6060f1SDimitry Andric UnwindLocation::createIsRegisterPlusOffset(uint32_t RegNum, int32_t Offset,
63*bdd1243dSDimitry Andric                                            std::optional<uint32_t> AddrSpace) {
64fe6060f1SDimitry Andric   return {RegPlusOffset, RegNum, Offset, AddrSpace, false};
65fe6060f1SDimitry Andric }
66fe6060f1SDimitry Andric 
67fe6060f1SDimitry Andric UnwindLocation
68fe6060f1SDimitry Andric UnwindLocation::createAtRegisterPlusOffset(uint32_t RegNum, int32_t Offset,
69*bdd1243dSDimitry Andric                                            std::optional<uint32_t> AddrSpace) {
70fe6060f1SDimitry Andric   return {RegPlusOffset, RegNum, Offset, AddrSpace, true};
71fe6060f1SDimitry Andric }
72fe6060f1SDimitry Andric 
73fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createIsDWARFExpression(DWARFExpression Expr) {
74fe6060f1SDimitry Andric   return {Expr, false};
75fe6060f1SDimitry Andric }
76fe6060f1SDimitry Andric 
77fe6060f1SDimitry Andric UnwindLocation UnwindLocation::createAtDWARFExpression(DWARFExpression Expr) {
78fe6060f1SDimitry Andric   return {Expr, true};
79fe6060f1SDimitry Andric }
80fe6060f1SDimitry Andric 
81*bdd1243dSDimitry Andric void UnwindLocation::dump(raw_ostream &OS, DIDumpOptions DumpOpts) const {
82fe6060f1SDimitry Andric   if (Dereference)
83fe6060f1SDimitry Andric     OS << '[';
84fe6060f1SDimitry Andric   switch (Kind) {
85fe6060f1SDimitry Andric   case Unspecified:
86fe6060f1SDimitry Andric     OS << "unspecified";
87fe6060f1SDimitry Andric     break;
88fe6060f1SDimitry Andric   case Undefined:
89fe6060f1SDimitry Andric     OS << "undefined";
90fe6060f1SDimitry Andric     break;
91fe6060f1SDimitry Andric   case Same:
92fe6060f1SDimitry Andric     OS << "same";
93fe6060f1SDimitry Andric     break;
94fe6060f1SDimitry Andric   case CFAPlusOffset:
95fe6060f1SDimitry Andric     OS << "CFA";
96fe6060f1SDimitry Andric     if (Offset == 0)
97fe6060f1SDimitry Andric       break;
98fe6060f1SDimitry Andric     if (Offset > 0)
99fe6060f1SDimitry Andric       OS << "+";
100fe6060f1SDimitry Andric     OS << Offset;
101fe6060f1SDimitry Andric     break;
102fe6060f1SDimitry Andric   case RegPlusOffset:
103*bdd1243dSDimitry Andric     printRegister(OS, DumpOpts, RegNum);
104fe6060f1SDimitry Andric     if (Offset == 0 && !AddrSpace)
105fe6060f1SDimitry Andric       break;
106fe6060f1SDimitry Andric     if (Offset >= 0)
107fe6060f1SDimitry Andric       OS << "+";
108fe6060f1SDimitry Andric     OS << Offset;
109fe6060f1SDimitry Andric     if (AddrSpace)
110fe6060f1SDimitry Andric       OS << " in addrspace" << *AddrSpace;
111fe6060f1SDimitry Andric     break;
112*bdd1243dSDimitry Andric   case DWARFExpr: {
113*bdd1243dSDimitry Andric     Expr->print(OS, DumpOpts, nullptr);
114fe6060f1SDimitry Andric     break;
115*bdd1243dSDimitry Andric   }
116fe6060f1SDimitry Andric   case Constant:
117fe6060f1SDimitry Andric     OS << Offset;
118fe6060f1SDimitry Andric     break;
119fe6060f1SDimitry Andric   }
120fe6060f1SDimitry Andric   if (Dereference)
121fe6060f1SDimitry Andric     OS << ']';
122fe6060f1SDimitry Andric }
123fe6060f1SDimitry Andric 
124fe6060f1SDimitry Andric raw_ostream &llvm::dwarf::operator<<(raw_ostream &OS,
125fe6060f1SDimitry Andric                                      const UnwindLocation &UL) {
126*bdd1243dSDimitry Andric   auto DumpOpts = DIDumpOptions();
127*bdd1243dSDimitry Andric   UL.dump(OS, DumpOpts);
128fe6060f1SDimitry Andric   return OS;
129fe6060f1SDimitry Andric }
130fe6060f1SDimitry Andric 
131fe6060f1SDimitry Andric bool UnwindLocation::operator==(const UnwindLocation &RHS) const {
132fe6060f1SDimitry Andric   if (Kind != RHS.Kind)
133fe6060f1SDimitry Andric     return false;
134fe6060f1SDimitry Andric   switch (Kind) {
135fe6060f1SDimitry Andric   case Unspecified:
136fe6060f1SDimitry Andric   case Undefined:
137fe6060f1SDimitry Andric   case Same:
138fe6060f1SDimitry Andric     return true;
139fe6060f1SDimitry Andric   case CFAPlusOffset:
140fe6060f1SDimitry Andric     return Offset == RHS.Offset && Dereference == RHS.Dereference;
141fe6060f1SDimitry Andric   case RegPlusOffset:
142fe6060f1SDimitry Andric     return RegNum == RHS.RegNum && Offset == RHS.Offset &&
143fe6060f1SDimitry Andric            Dereference == RHS.Dereference;
144fe6060f1SDimitry Andric   case DWARFExpr:
145fe6060f1SDimitry Andric     return *Expr == *RHS.Expr && Dereference == RHS.Dereference;
146fe6060f1SDimitry Andric   case Constant:
147fe6060f1SDimitry Andric     return Offset == RHS.Offset;
148fe6060f1SDimitry Andric   }
149fe6060f1SDimitry Andric   return false;
150fe6060f1SDimitry Andric }
151fe6060f1SDimitry Andric 
152*bdd1243dSDimitry Andric void RegisterLocations::dump(raw_ostream &OS, DIDumpOptions DumpOpts) const {
153fe6060f1SDimitry Andric   bool First = true;
154fe6060f1SDimitry Andric   for (const auto &RegLocPair : Locations) {
155fe6060f1SDimitry Andric     if (First)
156fe6060f1SDimitry Andric       First = false;
157fe6060f1SDimitry Andric     else
158fe6060f1SDimitry Andric       OS << ", ";
159*bdd1243dSDimitry Andric     printRegister(OS, DumpOpts, RegLocPair.first);
160fe6060f1SDimitry Andric     OS << '=';
161*bdd1243dSDimitry Andric     RegLocPair.second.dump(OS, DumpOpts);
162fe6060f1SDimitry Andric   }
163fe6060f1SDimitry Andric }
164fe6060f1SDimitry Andric 
165fe6060f1SDimitry Andric raw_ostream &llvm::dwarf::operator<<(raw_ostream &OS,
166fe6060f1SDimitry Andric                                      const RegisterLocations &RL) {
167*bdd1243dSDimitry Andric   auto DumpOpts = DIDumpOptions();
168*bdd1243dSDimitry Andric   RL.dump(OS, DumpOpts);
169fe6060f1SDimitry Andric   return OS;
170fe6060f1SDimitry Andric }
171fe6060f1SDimitry Andric 
172*bdd1243dSDimitry Andric void UnwindRow::dump(raw_ostream &OS, DIDumpOptions DumpOpts,
173fe6060f1SDimitry Andric                      unsigned IndentLevel) const {
174fe6060f1SDimitry Andric   OS.indent(2 * IndentLevel);
175fe6060f1SDimitry Andric   if (hasAddress())
176fe6060f1SDimitry Andric     OS << format("0x%" PRIx64 ": ", *Address);
177fe6060f1SDimitry Andric   OS << "CFA=";
178*bdd1243dSDimitry Andric   CFAValue.dump(OS, DumpOpts);
179fe6060f1SDimitry Andric   if (RegLocs.hasLocations()) {
180fe6060f1SDimitry Andric     OS << ": ";
181*bdd1243dSDimitry Andric     RegLocs.dump(OS, DumpOpts);
182fe6060f1SDimitry Andric   }
183fe6060f1SDimitry Andric   OS << "\n";
184fe6060f1SDimitry Andric }
185fe6060f1SDimitry Andric 
186fe6060f1SDimitry Andric raw_ostream &llvm::dwarf::operator<<(raw_ostream &OS, const UnwindRow &Row) {
187*bdd1243dSDimitry Andric   auto DumpOpts = DIDumpOptions();
188*bdd1243dSDimitry Andric   Row.dump(OS, DumpOpts, 0);
189fe6060f1SDimitry Andric   return OS;
190fe6060f1SDimitry Andric }
191fe6060f1SDimitry Andric 
192*bdd1243dSDimitry Andric void UnwindTable::dump(raw_ostream &OS, DIDumpOptions DumpOpts,
193fe6060f1SDimitry Andric                        unsigned IndentLevel) const {
194fe6060f1SDimitry Andric   for (const UnwindRow &Row : Rows)
195*bdd1243dSDimitry Andric     Row.dump(OS, DumpOpts, IndentLevel);
196fe6060f1SDimitry Andric }
197fe6060f1SDimitry Andric 
198fe6060f1SDimitry Andric raw_ostream &llvm::dwarf::operator<<(raw_ostream &OS, const UnwindTable &Rows) {
199*bdd1243dSDimitry Andric   auto DumpOpts = DIDumpOptions();
200*bdd1243dSDimitry Andric   Rows.dump(OS, DumpOpts, 0);
201fe6060f1SDimitry Andric   return OS;
202fe6060f1SDimitry Andric }
203fe6060f1SDimitry Andric 
204fe6060f1SDimitry Andric Expected<UnwindTable> UnwindTable::create(const FDE *Fde) {
205fe6060f1SDimitry Andric   const CIE *Cie = Fde->getLinkedCIE();
206fe6060f1SDimitry Andric   if (Cie == nullptr)
207fe6060f1SDimitry Andric     return createStringError(errc::invalid_argument,
208fe6060f1SDimitry Andric                              "unable to get CIE for FDE at offset 0x%" PRIx64,
209fe6060f1SDimitry Andric                              Fde->getOffset());
210fe6060f1SDimitry Andric 
211fe6060f1SDimitry Andric   // Rows will be empty if there are no CFI instructions.
212fe6060f1SDimitry Andric   if (Cie->cfis().empty() && Fde->cfis().empty())
213fe6060f1SDimitry Andric     return UnwindTable();
214fe6060f1SDimitry Andric 
215fe6060f1SDimitry Andric   UnwindTable UT;
216fe6060f1SDimitry Andric   UnwindRow Row;
217fe6060f1SDimitry Andric   Row.setAddress(Fde->getInitialLocation());
218fe6060f1SDimitry Andric   UT.EndAddress = Fde->getInitialLocation() + Fde->getAddressRange();
219fe6060f1SDimitry Andric   if (Error CieError = UT.parseRows(Cie->cfis(), Row, nullptr))
220fe6060f1SDimitry Andric     return std::move(CieError);
221fe6060f1SDimitry Andric   // We need to save the initial locations of registers from the CIE parsing
222fe6060f1SDimitry Andric   // in case we run into DW_CFA_restore or DW_CFA_restore_extended opcodes.
223fe6060f1SDimitry Andric   const RegisterLocations InitialLocs = Row.getRegisterLocations();
224fe6060f1SDimitry Andric   if (Error FdeError = UT.parseRows(Fde->cfis(), Row, &InitialLocs))
225fe6060f1SDimitry Andric     return std::move(FdeError);
226fe6060f1SDimitry Andric   // May be all the CFI instructions were DW_CFA_nop amd Row becomes empty.
227fe6060f1SDimitry Andric   // Do not add that to the unwind table.
228fe6060f1SDimitry Andric   if (Row.getRegisterLocations().hasLocations() ||
229fe6060f1SDimitry Andric       Row.getCFAValue().getLocation() != UnwindLocation::Unspecified)
230fe6060f1SDimitry Andric     UT.Rows.push_back(Row);
231fe6060f1SDimitry Andric   return UT;
232fe6060f1SDimitry Andric }
233fe6060f1SDimitry Andric 
234fe6060f1SDimitry Andric Expected<UnwindTable> UnwindTable::create(const CIE *Cie) {
235fe6060f1SDimitry Andric   // Rows will be empty if there are no CFI instructions.
236fe6060f1SDimitry Andric   if (Cie->cfis().empty())
237fe6060f1SDimitry Andric     return UnwindTable();
238fe6060f1SDimitry Andric 
239fe6060f1SDimitry Andric   UnwindTable UT;
240fe6060f1SDimitry Andric   UnwindRow Row;
241fe6060f1SDimitry Andric   if (Error CieError = UT.parseRows(Cie->cfis(), Row, nullptr))
242fe6060f1SDimitry Andric     return std::move(CieError);
243fe6060f1SDimitry Andric   // May be all the CFI instructions were DW_CFA_nop amd Row becomes empty.
244fe6060f1SDimitry Andric   // Do not add that to the unwind table.
245fe6060f1SDimitry Andric   if (Row.getRegisterLocations().hasLocations() ||
246fe6060f1SDimitry Andric       Row.getCFAValue().getLocation() != UnwindLocation::Unspecified)
247fe6060f1SDimitry Andric     UT.Rows.push_back(Row);
248fe6060f1SDimitry Andric   return UT;
249fe6060f1SDimitry Andric }
250fe6060f1SDimitry Andric 
2510b57cec5SDimitry Andric // See DWARF standard v3, section 7.23
2520b57cec5SDimitry Andric const uint8_t DWARF_CFI_PRIMARY_OPCODE_MASK = 0xc0;
2530b57cec5SDimitry Andric const uint8_t DWARF_CFI_PRIMARY_OPERAND_MASK = 0x3f;
2540b57cec5SDimitry Andric 
2558bcb0991SDimitry Andric Error CFIProgram::parse(DWARFDataExtractor Data, uint64_t *Offset,
2568bcb0991SDimitry Andric                         uint64_t EndOffset) {
2575ffd83dbSDimitry Andric   DataExtractor::Cursor C(*Offset);
2585ffd83dbSDimitry Andric   while (C && C.tell() < EndOffset) {
2595ffd83dbSDimitry Andric     uint8_t Opcode = Data.getRelocatedValue(C, 1);
2605ffd83dbSDimitry Andric     if (!C)
2615ffd83dbSDimitry Andric       break;
2620b57cec5SDimitry Andric 
2635ffd83dbSDimitry Andric     // Some instructions have a primary opcode encoded in the top bits.
2645ffd83dbSDimitry Andric     if (uint8_t Primary = Opcode & DWARF_CFI_PRIMARY_OPCODE_MASK) {
2650b57cec5SDimitry Andric       // If it's a primary opcode, the first operand is encoded in the bottom
2660b57cec5SDimitry Andric       // bits of the opcode itself.
2670b57cec5SDimitry Andric       uint64_t Op1 = Opcode & DWARF_CFI_PRIMARY_OPERAND_MASK;
2680b57cec5SDimitry Andric       switch (Primary) {
2690b57cec5SDimitry Andric       case DW_CFA_advance_loc:
2700b57cec5SDimitry Andric       case DW_CFA_restore:
2710b57cec5SDimitry Andric         addInstruction(Primary, Op1);
2720b57cec5SDimitry Andric         break;
2730b57cec5SDimitry Andric       case DW_CFA_offset:
2745ffd83dbSDimitry Andric         addInstruction(Primary, Op1, Data.getULEB128(C));
2750b57cec5SDimitry Andric         break;
2765ffd83dbSDimitry Andric       default:
2775ffd83dbSDimitry Andric         llvm_unreachable("invalid primary CFI opcode");
2780b57cec5SDimitry Andric       }
2795ffd83dbSDimitry Andric       continue;
2805ffd83dbSDimitry Andric     }
2815ffd83dbSDimitry Andric 
2820b57cec5SDimitry Andric     // Extended opcode - its value is Opcode itself.
2830b57cec5SDimitry Andric     switch (Opcode) {
2840b57cec5SDimitry Andric     default:
2850b57cec5SDimitry Andric       return createStringError(errc::illegal_byte_sequence,
2865ffd83dbSDimitry Andric                                "invalid extended CFI opcode 0x%" PRIx8, Opcode);
2870b57cec5SDimitry Andric     case DW_CFA_nop:
2880b57cec5SDimitry Andric     case DW_CFA_remember_state:
2890b57cec5SDimitry Andric     case DW_CFA_restore_state:
2900b57cec5SDimitry Andric     case DW_CFA_GNU_window_save:
2910b57cec5SDimitry Andric       // No operands
2920b57cec5SDimitry Andric       addInstruction(Opcode);
2930b57cec5SDimitry Andric       break;
2940b57cec5SDimitry Andric     case DW_CFA_set_loc:
2950b57cec5SDimitry Andric       // Operands: Address
2965ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getRelocatedAddress(C));
2970b57cec5SDimitry Andric       break;
2980b57cec5SDimitry Andric     case DW_CFA_advance_loc1:
2990b57cec5SDimitry Andric       // Operands: 1-byte delta
3005ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getRelocatedValue(C, 1));
3010b57cec5SDimitry Andric       break;
3020b57cec5SDimitry Andric     case DW_CFA_advance_loc2:
3030b57cec5SDimitry Andric       // Operands: 2-byte delta
3045ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getRelocatedValue(C, 2));
3050b57cec5SDimitry Andric       break;
3060b57cec5SDimitry Andric     case DW_CFA_advance_loc4:
3070b57cec5SDimitry Andric       // Operands: 4-byte delta
3085ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getRelocatedValue(C, 4));
3090b57cec5SDimitry Andric       break;
3100b57cec5SDimitry Andric     case DW_CFA_restore_extended:
3110b57cec5SDimitry Andric     case DW_CFA_undefined:
3120b57cec5SDimitry Andric     case DW_CFA_same_value:
3130b57cec5SDimitry Andric     case DW_CFA_def_cfa_register:
3140b57cec5SDimitry Andric     case DW_CFA_def_cfa_offset:
3150b57cec5SDimitry Andric     case DW_CFA_GNU_args_size:
3160b57cec5SDimitry Andric       // Operands: ULEB128
3175ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getULEB128(C));
3180b57cec5SDimitry Andric       break;
3190b57cec5SDimitry Andric     case DW_CFA_def_cfa_offset_sf:
3200b57cec5SDimitry Andric       // Operands: SLEB128
3215ffd83dbSDimitry Andric       addInstruction(Opcode, Data.getSLEB128(C));
3220b57cec5SDimitry Andric       break;
323fe6060f1SDimitry Andric     case DW_CFA_LLVM_def_aspace_cfa:
324fe6060f1SDimitry Andric     case DW_CFA_LLVM_def_aspace_cfa_sf: {
325fe6060f1SDimitry Andric       auto RegNum = Data.getULEB128(C);
326fe6060f1SDimitry Andric       auto CfaOffset = Opcode == DW_CFA_LLVM_def_aspace_cfa
327fe6060f1SDimitry Andric                            ? Data.getULEB128(C)
328fe6060f1SDimitry Andric                            : Data.getSLEB128(C);
329fe6060f1SDimitry Andric       auto AddressSpace = Data.getULEB128(C);
330fe6060f1SDimitry Andric       addInstruction(Opcode, RegNum, CfaOffset, AddressSpace);
331fe6060f1SDimitry Andric       break;
332fe6060f1SDimitry Andric     }
3330b57cec5SDimitry Andric     case DW_CFA_offset_extended:
3340b57cec5SDimitry Andric     case DW_CFA_register:
3350b57cec5SDimitry Andric     case DW_CFA_def_cfa:
3360b57cec5SDimitry Andric     case DW_CFA_val_offset: {
3370b57cec5SDimitry Andric       // Operands: ULEB128, ULEB128
3380b57cec5SDimitry Andric       // Note: We can not embed getULEB128 directly into function
3390b57cec5SDimitry Andric       // argument list. getULEB128 changes Offset and order of evaluation
3400b57cec5SDimitry Andric       // for arguments is unspecified.
3415ffd83dbSDimitry Andric       uint64_t op1 = Data.getULEB128(C);
3425ffd83dbSDimitry Andric       uint64_t op2 = Data.getULEB128(C);
3430b57cec5SDimitry Andric       addInstruction(Opcode, op1, op2);
3440b57cec5SDimitry Andric       break;
3450b57cec5SDimitry Andric     }
3460b57cec5SDimitry Andric     case DW_CFA_offset_extended_sf:
3470b57cec5SDimitry Andric     case DW_CFA_def_cfa_sf:
3480b57cec5SDimitry Andric     case DW_CFA_val_offset_sf: {
3490b57cec5SDimitry Andric       // Operands: ULEB128, SLEB128
3500b57cec5SDimitry Andric       // Note: see comment for the previous case
3515ffd83dbSDimitry Andric       uint64_t op1 = Data.getULEB128(C);
3525ffd83dbSDimitry Andric       uint64_t op2 = (uint64_t)Data.getSLEB128(C);
3530b57cec5SDimitry Andric       addInstruction(Opcode, op1, op2);
3540b57cec5SDimitry Andric       break;
3550b57cec5SDimitry Andric     }
3560b57cec5SDimitry Andric     case DW_CFA_def_cfa_expression: {
3575ffd83dbSDimitry Andric       uint64_t ExprLength = Data.getULEB128(C);
3580b57cec5SDimitry Andric       addInstruction(Opcode, 0);
3595ffd83dbSDimitry Andric       StringRef Expression = Data.getBytes(C, ExprLength);
3605ffd83dbSDimitry Andric 
3615ffd83dbSDimitry Andric       DataExtractor Extractor(Expression, Data.isLittleEndian(),
3625ffd83dbSDimitry Andric                               Data.getAddressSize());
3635ffd83dbSDimitry Andric       // Note. We do not pass the DWARF format to DWARFExpression, because
3645ffd83dbSDimitry Andric       // DW_OP_call_ref, the only operation which depends on the format, is
3655ffd83dbSDimitry Andric       // prohibited in call frame instructions, see sec. 6.4.2 in DWARFv5.
3665ffd83dbSDimitry Andric       Instructions.back().Expression =
3675ffd83dbSDimitry Andric           DWARFExpression(Extractor, Data.getAddressSize());
3680b57cec5SDimitry Andric       break;
3690b57cec5SDimitry Andric     }
3700b57cec5SDimitry Andric     case DW_CFA_expression:
3710b57cec5SDimitry Andric     case DW_CFA_val_expression: {
3725ffd83dbSDimitry Andric       uint64_t RegNum = Data.getULEB128(C);
3730b57cec5SDimitry Andric       addInstruction(Opcode, RegNum, 0);
3745ffd83dbSDimitry Andric 
3755ffd83dbSDimitry Andric       uint64_t BlockLength = Data.getULEB128(C);
3765ffd83dbSDimitry Andric       StringRef Expression = Data.getBytes(C, BlockLength);
3775ffd83dbSDimitry Andric       DataExtractor Extractor(Expression, Data.isLittleEndian(),
3785ffd83dbSDimitry Andric                               Data.getAddressSize());
3795ffd83dbSDimitry Andric       // Note. We do not pass the DWARF format to DWARFExpression, because
3805ffd83dbSDimitry Andric       // DW_OP_call_ref, the only operation which depends on the format, is
3815ffd83dbSDimitry Andric       // prohibited in call frame instructions, see sec. 6.4.2 in DWARFv5.
3825ffd83dbSDimitry Andric       Instructions.back().Expression =
3835ffd83dbSDimitry Andric           DWARFExpression(Extractor, Data.getAddressSize());
3840b57cec5SDimitry Andric       break;
3850b57cec5SDimitry Andric     }
3860b57cec5SDimitry Andric     }
3870b57cec5SDimitry Andric   }
3880b57cec5SDimitry Andric 
3895ffd83dbSDimitry Andric   *Offset = C.tell();
3905ffd83dbSDimitry Andric   return C.takeError();
3910b57cec5SDimitry Andric }
3920b57cec5SDimitry Andric 
393fe6060f1SDimitry Andric StringRef CFIProgram::callFrameString(unsigned Opcode) const {
394fe6060f1SDimitry Andric   return dwarf::CallFrameString(Opcode, Arch);
395fe6060f1SDimitry Andric }
3960b57cec5SDimitry Andric 
397fe6060f1SDimitry Andric const char *CFIProgram::operandTypeString(CFIProgram::OperandType OT) {
398fe6060f1SDimitry Andric #define ENUM_TO_CSTR(e)                                                        \
399fe6060f1SDimitry Andric   case e:                                                                      \
400fe6060f1SDimitry Andric     return #e;
401fe6060f1SDimitry Andric   switch (OT) {
402fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_Unset);
403fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_None);
404fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_Address);
405fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_Offset);
406fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_FactoredCodeOffset);
407fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_SignedFactDataOffset);
408fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_UnsignedFactDataOffset);
409fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_Register);
410fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_AddressSpace);
411fe6060f1SDimitry Andric     ENUM_TO_CSTR(OT_Expression);
412fe6060f1SDimitry Andric   }
413fe6060f1SDimitry Andric   return "<unknown CFIProgram::OperandType>";
414fe6060f1SDimitry Andric }
4150b57cec5SDimitry Andric 
416fe6060f1SDimitry Andric llvm::Expected<uint64_t>
417fe6060f1SDimitry Andric CFIProgram::Instruction::getOperandAsUnsigned(const CFIProgram &CFIP,
418fe6060f1SDimitry Andric                                               uint32_t OperandIdx) const {
419fe6060f1SDimitry Andric   if (OperandIdx >= MaxOperands)
420fe6060f1SDimitry Andric     return createStringError(errc::invalid_argument,
421fe6060f1SDimitry Andric                              "operand index %" PRIu32 " is not valid",
422fe6060f1SDimitry Andric                              OperandIdx);
423fe6060f1SDimitry Andric   OperandType Type = CFIP.getOperandTypes()[Opcode][OperandIdx];
424fe6060f1SDimitry Andric   uint64_t Operand = Ops[OperandIdx];
425fe6060f1SDimitry Andric   switch (Type) {
426fe6060f1SDimitry Andric   case OT_Unset:
427fe6060f1SDimitry Andric   case OT_None:
428fe6060f1SDimitry Andric   case OT_Expression:
429fe6060f1SDimitry Andric     return createStringError(errc::invalid_argument,
430fe6060f1SDimitry Andric                              "op[%" PRIu32 "] has type %s which has no value",
431fe6060f1SDimitry Andric                              OperandIdx, CFIProgram::operandTypeString(Type));
4320b57cec5SDimitry Andric 
433fe6060f1SDimitry Andric   case OT_Offset:
434fe6060f1SDimitry Andric   case OT_SignedFactDataOffset:
435fe6060f1SDimitry Andric   case OT_UnsignedFactDataOffset:
436fe6060f1SDimitry Andric     return createStringError(
437fe6060f1SDimitry Andric         errc::invalid_argument,
438fe6060f1SDimitry Andric         "op[%" PRIu32 "] has OperandType OT_Offset which produces a signed "
439fe6060f1SDimitry Andric         "result, call getOperandAsSigned instead",
440fe6060f1SDimitry Andric         OperandIdx);
441fe6060f1SDimitry Andric 
442fe6060f1SDimitry Andric   case OT_Address:
443fe6060f1SDimitry Andric   case OT_Register:
444fe6060f1SDimitry Andric   case OT_AddressSpace:
445fe6060f1SDimitry Andric     return Operand;
446fe6060f1SDimitry Andric 
447fe6060f1SDimitry Andric   case OT_FactoredCodeOffset: {
448fe6060f1SDimitry Andric     const uint64_t CodeAlignmentFactor = CFIP.codeAlign();
449fe6060f1SDimitry Andric     if (CodeAlignmentFactor == 0)
450fe6060f1SDimitry Andric       return createStringError(
451fe6060f1SDimitry Andric           errc::invalid_argument,
452fe6060f1SDimitry Andric           "op[%" PRIu32 "] has type OT_FactoredCodeOffset but code alignment "
453fe6060f1SDimitry Andric           "is zero",
454fe6060f1SDimitry Andric           OperandIdx);
455fe6060f1SDimitry Andric     return Operand * CodeAlignmentFactor;
456fe6060f1SDimitry Andric   }
457fe6060f1SDimitry Andric   }
458fe6060f1SDimitry Andric   llvm_unreachable("invalid operand type");
459fe6060f1SDimitry Andric }
460fe6060f1SDimitry Andric 
461fe6060f1SDimitry Andric llvm::Expected<int64_t>
462fe6060f1SDimitry Andric CFIProgram::Instruction::getOperandAsSigned(const CFIProgram &CFIP,
463fe6060f1SDimitry Andric                                             uint32_t OperandIdx) const {
464fe6060f1SDimitry Andric   if (OperandIdx >= MaxOperands)
465fe6060f1SDimitry Andric     return createStringError(errc::invalid_argument,
466fe6060f1SDimitry Andric                              "operand index %" PRIu32 " is not valid",
467fe6060f1SDimitry Andric                              OperandIdx);
468fe6060f1SDimitry Andric   OperandType Type = CFIP.getOperandTypes()[Opcode][OperandIdx];
469fe6060f1SDimitry Andric   uint64_t Operand = Ops[OperandIdx];
470fe6060f1SDimitry Andric   switch (Type) {
471fe6060f1SDimitry Andric   case OT_Unset:
472fe6060f1SDimitry Andric   case OT_None:
473fe6060f1SDimitry Andric   case OT_Expression:
474fe6060f1SDimitry Andric     return createStringError(errc::invalid_argument,
475fe6060f1SDimitry Andric                              "op[%" PRIu32 "] has type %s which has no value",
476fe6060f1SDimitry Andric                              OperandIdx, CFIProgram::operandTypeString(Type));
477fe6060f1SDimitry Andric 
478fe6060f1SDimitry Andric   case OT_Address:
479fe6060f1SDimitry Andric   case OT_Register:
480fe6060f1SDimitry Andric   case OT_AddressSpace:
481fe6060f1SDimitry Andric     return createStringError(
482fe6060f1SDimitry Andric         errc::invalid_argument,
483fe6060f1SDimitry Andric         "op[%" PRIu32 "] has OperandType %s which produces an unsigned result, "
484fe6060f1SDimitry Andric         "call getOperandAsUnsigned instead",
485fe6060f1SDimitry Andric         OperandIdx, CFIProgram::operandTypeString(Type));
486fe6060f1SDimitry Andric 
487fe6060f1SDimitry Andric   case OT_Offset:
488fe6060f1SDimitry Andric     return (int64_t)Operand;
489fe6060f1SDimitry Andric 
490fe6060f1SDimitry Andric   case OT_FactoredCodeOffset:
491fe6060f1SDimitry Andric   case OT_SignedFactDataOffset: {
492fe6060f1SDimitry Andric     const int64_t DataAlignmentFactor = CFIP.dataAlign();
493fe6060f1SDimitry Andric     if (DataAlignmentFactor == 0)
494fe6060f1SDimitry Andric       return createStringError(errc::invalid_argument,
495fe6060f1SDimitry Andric                                "op[%" PRIu32 "] has type %s but data "
496fe6060f1SDimitry Andric                                "alignment is zero",
497fe6060f1SDimitry Andric                                OperandIdx, CFIProgram::operandTypeString(Type));
498fe6060f1SDimitry Andric     return int64_t(Operand) * DataAlignmentFactor;
499fe6060f1SDimitry Andric   }
500fe6060f1SDimitry Andric 
501fe6060f1SDimitry Andric   case OT_UnsignedFactDataOffset: {
502fe6060f1SDimitry Andric     const int64_t DataAlignmentFactor = CFIP.dataAlign();
503fe6060f1SDimitry Andric     if (DataAlignmentFactor == 0)
504fe6060f1SDimitry Andric       return createStringError(errc::invalid_argument,
505fe6060f1SDimitry Andric                                "op[%" PRIu32
506fe6060f1SDimitry Andric                                "] has type OT_UnsignedFactDataOffset but data "
507fe6060f1SDimitry Andric                                "alignment is zero",
508fe6060f1SDimitry Andric                                OperandIdx);
509fe6060f1SDimitry Andric     return Operand * DataAlignmentFactor;
510fe6060f1SDimitry Andric   }
511fe6060f1SDimitry Andric   }
512fe6060f1SDimitry Andric   llvm_unreachable("invalid operand type");
513fe6060f1SDimitry Andric }
514fe6060f1SDimitry Andric 
515fe6060f1SDimitry Andric Error UnwindTable::parseRows(const CFIProgram &CFIP, UnwindRow &Row,
516fe6060f1SDimitry Andric                              const RegisterLocations *InitialLocs) {
517*bdd1243dSDimitry Andric   // State consists of CFA value and register locations.
518*bdd1243dSDimitry Andric   std::vector<std::pair<UnwindLocation, RegisterLocations>> States;
519fe6060f1SDimitry Andric   for (const CFIProgram::Instruction &Inst : CFIP) {
520fe6060f1SDimitry Andric     switch (Inst.Opcode) {
521fe6060f1SDimitry Andric     case dwarf::DW_CFA_set_loc: {
522fe6060f1SDimitry Andric       // The DW_CFA_set_loc instruction takes a single operand that
523fe6060f1SDimitry Andric       // represents a target address. The required action is to create a new
524fe6060f1SDimitry Andric       // table row using the specified address as the location. All other
525fe6060f1SDimitry Andric       // values in the new row are initially identical to the current row.
526fe6060f1SDimitry Andric       // The new location value is always greater than the current one. If
527fe6060f1SDimitry Andric       // the segment_size field of this FDE's CIE is non- zero, the initial
528fe6060f1SDimitry Andric       // location is preceded by a segment selector of the given length
529fe6060f1SDimitry Andric       llvm::Expected<uint64_t> NewAddress = Inst.getOperandAsUnsigned(CFIP, 0);
530fe6060f1SDimitry Andric       if (!NewAddress)
531fe6060f1SDimitry Andric         return NewAddress.takeError();
532fe6060f1SDimitry Andric       if (*NewAddress <= Row.getAddress())
533fe6060f1SDimitry Andric         return createStringError(
534fe6060f1SDimitry Andric             errc::invalid_argument,
535fe6060f1SDimitry Andric             "%s with adrress 0x%" PRIx64 " which must be greater than the "
536fe6060f1SDimitry Andric             "current row address 0x%" PRIx64,
537fe6060f1SDimitry Andric             CFIP.callFrameString(Inst.Opcode).str().c_str(), *NewAddress,
538fe6060f1SDimitry Andric             Row.getAddress());
539fe6060f1SDimitry Andric       Rows.push_back(Row);
540fe6060f1SDimitry Andric       Row.setAddress(*NewAddress);
541fe6060f1SDimitry Andric       break;
542fe6060f1SDimitry Andric     }
543fe6060f1SDimitry Andric 
544fe6060f1SDimitry Andric     case dwarf::DW_CFA_advance_loc:
545fe6060f1SDimitry Andric     case dwarf::DW_CFA_advance_loc1:
546fe6060f1SDimitry Andric     case dwarf::DW_CFA_advance_loc2:
547fe6060f1SDimitry Andric     case dwarf::DW_CFA_advance_loc4: {
548fe6060f1SDimitry Andric       // The DW_CFA_advance instruction takes a single operand that
549fe6060f1SDimitry Andric       // represents a constant delta. The required action is to create a new
550fe6060f1SDimitry Andric       // table row with a location value that is computed by taking the
551fe6060f1SDimitry Andric       // current entry’s location value and adding the value of delta *
552fe6060f1SDimitry Andric       // code_alignment_factor. All other values in the new row are initially
553fe6060f1SDimitry Andric       // identical to the current row.
554fe6060f1SDimitry Andric       Rows.push_back(Row);
555fe6060f1SDimitry Andric       llvm::Expected<uint64_t> Offset = Inst.getOperandAsUnsigned(CFIP, 0);
556fe6060f1SDimitry Andric       if (!Offset)
557fe6060f1SDimitry Andric         return Offset.takeError();
558fe6060f1SDimitry Andric       Row.slideAddress(*Offset);
559fe6060f1SDimitry Andric       break;
560fe6060f1SDimitry Andric     }
561fe6060f1SDimitry Andric 
562fe6060f1SDimitry Andric     case dwarf::DW_CFA_restore:
563fe6060f1SDimitry Andric     case dwarf::DW_CFA_restore_extended: {
564fe6060f1SDimitry Andric       // The DW_CFA_restore instruction takes a single operand (encoded with
565fe6060f1SDimitry Andric       // the opcode) that represents a register number. The required action
566fe6060f1SDimitry Andric       // is to change the rule for the indicated register to the rule
567fe6060f1SDimitry Andric       // assigned it by the initial_instructions in the CIE.
568fe6060f1SDimitry Andric       if (InitialLocs == nullptr)
569fe6060f1SDimitry Andric         return createStringError(
570fe6060f1SDimitry Andric             errc::invalid_argument, "%s encountered while parsing a CIE",
571fe6060f1SDimitry Andric             CFIP.callFrameString(Inst.Opcode).str().c_str());
572fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
573fe6060f1SDimitry Andric       if (!RegNum)
574fe6060f1SDimitry Andric         return RegNum.takeError();
575*bdd1243dSDimitry Andric       if (std::optional<UnwindLocation> O =
576fe6060f1SDimitry Andric               InitialLocs->getRegisterLocation(*RegNum))
577fe6060f1SDimitry Andric         Row.getRegisterLocations().setRegisterLocation(*RegNum, *O);
578fe6060f1SDimitry Andric       else
579fe6060f1SDimitry Andric         Row.getRegisterLocations().removeRegisterLocation(*RegNum);
580fe6060f1SDimitry Andric       break;
581fe6060f1SDimitry Andric     }
582fe6060f1SDimitry Andric 
583fe6060f1SDimitry Andric     case dwarf::DW_CFA_offset:
584fe6060f1SDimitry Andric     case dwarf::DW_CFA_offset_extended:
585fe6060f1SDimitry Andric     case dwarf::DW_CFA_offset_extended_sf: {
586fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
587fe6060f1SDimitry Andric       if (!RegNum)
588fe6060f1SDimitry Andric         return RegNum.takeError();
589fe6060f1SDimitry Andric       llvm::Expected<int64_t> Offset = Inst.getOperandAsSigned(CFIP, 1);
590fe6060f1SDimitry Andric       if (!Offset)
591fe6060f1SDimitry Andric         return Offset.takeError();
592fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
593fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createAtCFAPlusOffset(*Offset));
594fe6060f1SDimitry Andric       break;
595fe6060f1SDimitry Andric     }
596fe6060f1SDimitry Andric 
597fe6060f1SDimitry Andric     case dwarf::DW_CFA_nop:
598fe6060f1SDimitry Andric       break;
599fe6060f1SDimitry Andric 
600fe6060f1SDimitry Andric     case dwarf::DW_CFA_remember_state:
601*bdd1243dSDimitry Andric       States.push_back(
602*bdd1243dSDimitry Andric           std::make_pair(Row.getCFAValue(), Row.getRegisterLocations()));
603fe6060f1SDimitry Andric       break;
604fe6060f1SDimitry Andric 
605fe6060f1SDimitry Andric     case dwarf::DW_CFA_restore_state:
606*bdd1243dSDimitry Andric       if (States.empty())
607fe6060f1SDimitry Andric         return createStringError(errc::invalid_argument,
608fe6060f1SDimitry Andric                                  "DW_CFA_restore_state without a matching "
609fe6060f1SDimitry Andric                                  "previous DW_CFA_remember_state");
610*bdd1243dSDimitry Andric       Row.getCFAValue() = States.back().first;
611*bdd1243dSDimitry Andric       Row.getRegisterLocations() = States.back().second;
612*bdd1243dSDimitry Andric       States.pop_back();
613fe6060f1SDimitry Andric       break;
614fe6060f1SDimitry Andric 
615fe6060f1SDimitry Andric     case dwarf::DW_CFA_GNU_window_save:
616fe6060f1SDimitry Andric       switch (CFIP.triple()) {
617fe6060f1SDimitry Andric       case Triple::aarch64:
618fe6060f1SDimitry Andric       case Triple::aarch64_be:
619fe6060f1SDimitry Andric       case Triple::aarch64_32: {
620fe6060f1SDimitry Andric         // DW_CFA_GNU_window_save is used for different things on different
621fe6060f1SDimitry Andric         // architectures. For aarch64 it is known as
622fe6060f1SDimitry Andric         // DW_CFA_AARCH64_negate_ra_state. The action is to toggle the
623fe6060f1SDimitry Andric         // value of the return address state between 1 and 0. If there is
624fe6060f1SDimitry Andric         // no rule for the AARCH64_DWARF_PAUTH_RA_STATE register, then it
625fe6060f1SDimitry Andric         // should be initially set to 1.
626fe6060f1SDimitry Andric         constexpr uint32_t AArch64DWARFPAuthRaState = 34;
627fe6060f1SDimitry Andric         auto LRLoc = Row.getRegisterLocations().getRegisterLocation(
628fe6060f1SDimitry Andric             AArch64DWARFPAuthRaState);
629fe6060f1SDimitry Andric         if (LRLoc) {
630fe6060f1SDimitry Andric           if (LRLoc->getLocation() == UnwindLocation::Constant) {
631fe6060f1SDimitry Andric             // Toggle the constant value from 0 to 1 or 1 to 0.
632fe6060f1SDimitry Andric             LRLoc->setConstant(LRLoc->getConstant() ^ 1);
633fe6060f1SDimitry Andric           } else {
634fe6060f1SDimitry Andric             return createStringError(
635fe6060f1SDimitry Andric                 errc::invalid_argument,
636fe6060f1SDimitry Andric                 "%s encountered when existing rule for this register is not "
637fe6060f1SDimitry Andric                 "a constant",
638fe6060f1SDimitry Andric                 CFIP.callFrameString(Inst.Opcode).str().c_str());
639fe6060f1SDimitry Andric           }
640fe6060f1SDimitry Andric         } else {
641fe6060f1SDimitry Andric           Row.getRegisterLocations().setRegisterLocation(
642fe6060f1SDimitry Andric               AArch64DWARFPAuthRaState, UnwindLocation::createIsConstant(1));
643fe6060f1SDimitry Andric         }
644fe6060f1SDimitry Andric         break;
645fe6060f1SDimitry Andric       }
646fe6060f1SDimitry Andric 
647fe6060f1SDimitry Andric       case Triple::sparc:
648fe6060f1SDimitry Andric       case Triple::sparcv9:
649fe6060f1SDimitry Andric       case Triple::sparcel:
650fe6060f1SDimitry Andric         for (uint32_t RegNum = 16; RegNum < 32; ++RegNum) {
651fe6060f1SDimitry Andric           Row.getRegisterLocations().setRegisterLocation(
652fe6060f1SDimitry Andric               RegNum, UnwindLocation::createAtCFAPlusOffset((RegNum - 16) * 8));
653fe6060f1SDimitry Andric         }
654fe6060f1SDimitry Andric         break;
655fe6060f1SDimitry Andric 
656fe6060f1SDimitry Andric       default: {
657fe6060f1SDimitry Andric         return createStringError(
658fe6060f1SDimitry Andric             errc::not_supported,
659fe6060f1SDimitry Andric             "DW_CFA opcode %#x is not supported for architecture %s",
660fe6060f1SDimitry Andric             Inst.Opcode, Triple::getArchTypeName(CFIP.triple()).str().c_str());
661fe6060f1SDimitry Andric 
662fe6060f1SDimitry Andric         break;
663fe6060f1SDimitry Andric       }
664fe6060f1SDimitry Andric       }
665fe6060f1SDimitry Andric       break;
666fe6060f1SDimitry Andric 
667fe6060f1SDimitry Andric     case dwarf::DW_CFA_undefined: {
668fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
669fe6060f1SDimitry Andric       if (!RegNum)
670fe6060f1SDimitry Andric         return RegNum.takeError();
671fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
672fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createUndefined());
673fe6060f1SDimitry Andric       break;
674fe6060f1SDimitry Andric     }
675fe6060f1SDimitry Andric 
676fe6060f1SDimitry Andric     case dwarf::DW_CFA_same_value: {
677fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
678fe6060f1SDimitry Andric       if (!RegNum)
679fe6060f1SDimitry Andric         return RegNum.takeError();
680fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
681fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createSame());
682fe6060f1SDimitry Andric       break;
683fe6060f1SDimitry Andric     }
684fe6060f1SDimitry Andric 
685fe6060f1SDimitry Andric     case dwarf::DW_CFA_GNU_args_size:
686fe6060f1SDimitry Andric       break;
687fe6060f1SDimitry Andric 
688fe6060f1SDimitry Andric     case dwarf::DW_CFA_register: {
689fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
690fe6060f1SDimitry Andric       if (!RegNum)
691fe6060f1SDimitry Andric         return RegNum.takeError();
692fe6060f1SDimitry Andric       llvm::Expected<uint64_t> NewRegNum = Inst.getOperandAsUnsigned(CFIP, 1);
693fe6060f1SDimitry Andric       if (!NewRegNum)
694fe6060f1SDimitry Andric         return NewRegNum.takeError();
695fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
696fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createIsRegisterPlusOffset(*NewRegNum, 0));
697fe6060f1SDimitry Andric       break;
698fe6060f1SDimitry Andric     }
699fe6060f1SDimitry Andric 
700fe6060f1SDimitry Andric     case dwarf::DW_CFA_val_offset:
701fe6060f1SDimitry Andric     case dwarf::DW_CFA_val_offset_sf: {
702fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
703fe6060f1SDimitry Andric       if (!RegNum)
704fe6060f1SDimitry Andric         return RegNum.takeError();
705fe6060f1SDimitry Andric       llvm::Expected<int64_t> Offset = Inst.getOperandAsSigned(CFIP, 1);
706fe6060f1SDimitry Andric       if (!Offset)
707fe6060f1SDimitry Andric         return Offset.takeError();
708fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
709fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createIsCFAPlusOffset(*Offset));
710fe6060f1SDimitry Andric       break;
711fe6060f1SDimitry Andric     }
712fe6060f1SDimitry Andric 
713fe6060f1SDimitry Andric     case dwarf::DW_CFA_expression: {
714fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
715fe6060f1SDimitry Andric       if (!RegNum)
716fe6060f1SDimitry Andric         return RegNum.takeError();
717fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
718fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createAtDWARFExpression(*Inst.Expression));
719fe6060f1SDimitry Andric       break;
720fe6060f1SDimitry Andric     }
721fe6060f1SDimitry Andric 
722fe6060f1SDimitry Andric     case dwarf::DW_CFA_val_expression: {
723fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
724fe6060f1SDimitry Andric       if (!RegNum)
725fe6060f1SDimitry Andric         return RegNum.takeError();
726fe6060f1SDimitry Andric       Row.getRegisterLocations().setRegisterLocation(
727fe6060f1SDimitry Andric           *RegNum, UnwindLocation::createIsDWARFExpression(*Inst.Expression));
728fe6060f1SDimitry Andric       break;
729fe6060f1SDimitry Andric     }
730fe6060f1SDimitry Andric 
731fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa_register: {
732fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
733fe6060f1SDimitry Andric       if (!RegNum)
734fe6060f1SDimitry Andric         return RegNum.takeError();
735fe6060f1SDimitry Andric       if (Row.getCFAValue().getLocation() != UnwindLocation::RegPlusOffset)
736fe6060f1SDimitry Andric         Row.getCFAValue() =
737fe6060f1SDimitry Andric             UnwindLocation::createIsRegisterPlusOffset(*RegNum, 0);
738fe6060f1SDimitry Andric       else
739fe6060f1SDimitry Andric         Row.getCFAValue().setRegister(*RegNum);
740fe6060f1SDimitry Andric       break;
741fe6060f1SDimitry Andric     }
742fe6060f1SDimitry Andric 
743fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa_offset:
744fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa_offset_sf: {
745fe6060f1SDimitry Andric       llvm::Expected<int64_t> Offset = Inst.getOperandAsSigned(CFIP, 0);
746fe6060f1SDimitry Andric       if (!Offset)
747fe6060f1SDimitry Andric         return Offset.takeError();
748fe6060f1SDimitry Andric       if (Row.getCFAValue().getLocation() != UnwindLocation::RegPlusOffset) {
749fe6060f1SDimitry Andric         return createStringError(
750fe6060f1SDimitry Andric             errc::invalid_argument,
751fe6060f1SDimitry Andric             "%s found when CFA rule was not RegPlusOffset",
752fe6060f1SDimitry Andric             CFIP.callFrameString(Inst.Opcode).str().c_str());
753fe6060f1SDimitry Andric       }
754fe6060f1SDimitry Andric       Row.getCFAValue().setOffset(*Offset);
755fe6060f1SDimitry Andric       break;
756fe6060f1SDimitry Andric     }
757fe6060f1SDimitry Andric 
758fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa:
759fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa_sf: {
760fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
761fe6060f1SDimitry Andric       if (!RegNum)
762fe6060f1SDimitry Andric         return RegNum.takeError();
763fe6060f1SDimitry Andric       llvm::Expected<int64_t> Offset = Inst.getOperandAsSigned(CFIP, 1);
764fe6060f1SDimitry Andric       if (!Offset)
765fe6060f1SDimitry Andric         return Offset.takeError();
766fe6060f1SDimitry Andric       Row.getCFAValue() =
767fe6060f1SDimitry Andric           UnwindLocation::createIsRegisterPlusOffset(*RegNum, *Offset);
768fe6060f1SDimitry Andric       break;
769fe6060f1SDimitry Andric     }
770fe6060f1SDimitry Andric 
771fe6060f1SDimitry Andric     case dwarf::DW_CFA_LLVM_def_aspace_cfa:
772fe6060f1SDimitry Andric     case dwarf::DW_CFA_LLVM_def_aspace_cfa_sf: {
773fe6060f1SDimitry Andric       llvm::Expected<uint64_t> RegNum = Inst.getOperandAsUnsigned(CFIP, 0);
774fe6060f1SDimitry Andric       if (!RegNum)
775fe6060f1SDimitry Andric         return RegNum.takeError();
776fe6060f1SDimitry Andric       llvm::Expected<int64_t> Offset = Inst.getOperandAsSigned(CFIP, 1);
777fe6060f1SDimitry Andric       if (!Offset)
778fe6060f1SDimitry Andric         return Offset.takeError();
779fe6060f1SDimitry Andric       llvm::Expected<uint32_t> CFAAddrSpace =
780fe6060f1SDimitry Andric           Inst.getOperandAsUnsigned(CFIP, 2);
781fe6060f1SDimitry Andric       if (!CFAAddrSpace)
782fe6060f1SDimitry Andric         return CFAAddrSpace.takeError();
783fe6060f1SDimitry Andric       Row.getCFAValue() = UnwindLocation::createIsRegisterPlusOffset(
784fe6060f1SDimitry Andric           *RegNum, *Offset, *CFAAddrSpace);
785fe6060f1SDimitry Andric       break;
786fe6060f1SDimitry Andric     }
787fe6060f1SDimitry Andric 
788fe6060f1SDimitry Andric     case dwarf::DW_CFA_def_cfa_expression:
789fe6060f1SDimitry Andric       Row.getCFAValue() =
790fe6060f1SDimitry Andric           UnwindLocation::createIsDWARFExpression(*Inst.Expression);
791fe6060f1SDimitry Andric       break;
792fe6060f1SDimitry Andric     }
793fe6060f1SDimitry Andric   }
794fe6060f1SDimitry Andric   return Error::success();
795fe6060f1SDimitry Andric }
796fe6060f1SDimitry Andric 
797fe6060f1SDimitry Andric ArrayRef<CFIProgram::OperandType[CFIProgram::MaxOperands]>
798fe6060f1SDimitry Andric CFIProgram::getOperandTypes() {
799fe6060f1SDimitry Andric   static OperandType OpTypes[DW_CFA_restore + 1][MaxOperands];
8000b57cec5SDimitry Andric   static bool Initialized = false;
8010b57cec5SDimitry Andric   if (Initialized) {
802fe6060f1SDimitry Andric     return ArrayRef<OperandType[MaxOperands]>(&OpTypes[0], DW_CFA_restore + 1);
8030b57cec5SDimitry Andric   }
8040b57cec5SDimitry Andric   Initialized = true;
8050b57cec5SDimitry Andric 
806fe6060f1SDimitry Andric #define DECLARE_OP3(OP, OPTYPE0, OPTYPE1, OPTYPE2)                             \
8070b57cec5SDimitry Andric   do {                                                                         \
8080b57cec5SDimitry Andric     OpTypes[OP][0] = OPTYPE0;                                                  \
8090b57cec5SDimitry Andric     OpTypes[OP][1] = OPTYPE1;                                                  \
810fe6060f1SDimitry Andric     OpTypes[OP][2] = OPTYPE2;                                                  \
8110b57cec5SDimitry Andric   } while (false)
812fe6060f1SDimitry Andric #define DECLARE_OP2(OP, OPTYPE0, OPTYPE1)                                      \
813fe6060f1SDimitry Andric   DECLARE_OP3(OP, OPTYPE0, OPTYPE1, OT_None)
8140b57cec5SDimitry Andric #define DECLARE_OP1(OP, OPTYPE0) DECLARE_OP2(OP, OPTYPE0, OT_None)
8150b57cec5SDimitry Andric #define DECLARE_OP0(OP) DECLARE_OP1(OP, OT_None)
8160b57cec5SDimitry Andric 
8170b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_set_loc, OT_Address);
8180b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_advance_loc, OT_FactoredCodeOffset);
8190b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_advance_loc1, OT_FactoredCodeOffset);
8200b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_advance_loc2, OT_FactoredCodeOffset);
8210b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_advance_loc4, OT_FactoredCodeOffset);
8220b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_MIPS_advance_loc8, OT_FactoredCodeOffset);
8230b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_def_cfa, OT_Register, OT_Offset);
8240b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_def_cfa_sf, OT_Register, OT_SignedFactDataOffset);
8250b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_def_cfa_register, OT_Register);
826fe6060f1SDimitry Andric   DECLARE_OP3(DW_CFA_LLVM_def_aspace_cfa, OT_Register, OT_Offset,
827fe6060f1SDimitry Andric               OT_AddressSpace);
828fe6060f1SDimitry Andric   DECLARE_OP3(DW_CFA_LLVM_def_aspace_cfa_sf, OT_Register,
829fe6060f1SDimitry Andric               OT_SignedFactDataOffset, OT_AddressSpace);
8300b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_def_cfa_offset, OT_Offset);
8310b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_def_cfa_offset_sf, OT_SignedFactDataOffset);
8320b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_def_cfa_expression, OT_Expression);
8330b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_undefined, OT_Register);
8340b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_same_value, OT_Register);
8350b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_offset, OT_Register, OT_UnsignedFactDataOffset);
8360b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_offset_extended, OT_Register, OT_UnsignedFactDataOffset);
8370b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_offset_extended_sf, OT_Register, OT_SignedFactDataOffset);
8380b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_val_offset, OT_Register, OT_UnsignedFactDataOffset);
8390b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_val_offset_sf, OT_Register, OT_SignedFactDataOffset);
8400b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_register, OT_Register, OT_Register);
8410b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_expression, OT_Register, OT_Expression);
8420b57cec5SDimitry Andric   DECLARE_OP2(DW_CFA_val_expression, OT_Register, OT_Expression);
8430b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_restore, OT_Register);
8440b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_restore_extended, OT_Register);
8450b57cec5SDimitry Andric   DECLARE_OP0(DW_CFA_remember_state);
8460b57cec5SDimitry Andric   DECLARE_OP0(DW_CFA_restore_state);
8470b57cec5SDimitry Andric   DECLARE_OP0(DW_CFA_GNU_window_save);
8480b57cec5SDimitry Andric   DECLARE_OP1(DW_CFA_GNU_args_size, OT_Offset);
8490b57cec5SDimitry Andric   DECLARE_OP0(DW_CFA_nop);
8500b57cec5SDimitry Andric 
8510b57cec5SDimitry Andric #undef DECLARE_OP0
8520b57cec5SDimitry Andric #undef DECLARE_OP1
8530b57cec5SDimitry Andric #undef DECLARE_OP2
8540b57cec5SDimitry Andric 
855fe6060f1SDimitry Andric   return ArrayRef<OperandType[MaxOperands]>(&OpTypes[0], DW_CFA_restore + 1);
8560b57cec5SDimitry Andric }
8570b57cec5SDimitry Andric 
8580b57cec5SDimitry Andric /// Print \p Opcode's operand number \p OperandIdx which has value \p Operand.
859e8d8bef9SDimitry Andric void CFIProgram::printOperand(raw_ostream &OS, DIDumpOptions DumpOpts,
860e8d8bef9SDimitry Andric                               const Instruction &Instr, unsigned OperandIdx,
861e8d8bef9SDimitry Andric                               uint64_t Operand) const {
862fe6060f1SDimitry Andric   assert(OperandIdx < MaxOperands);
8630b57cec5SDimitry Andric   uint8_t Opcode = Instr.Opcode;
8640b57cec5SDimitry Andric   OperandType Type = getOperandTypes()[Opcode][OperandIdx];
8650b57cec5SDimitry Andric 
8660b57cec5SDimitry Andric   switch (Type) {
8670b57cec5SDimitry Andric   case OT_Unset: {
8680b57cec5SDimitry Andric     OS << " Unsupported " << (OperandIdx ? "second" : "first") << " operand to";
869fe6060f1SDimitry Andric     auto OpcodeName = callFrameString(Opcode);
8700b57cec5SDimitry Andric     if (!OpcodeName.empty())
8710b57cec5SDimitry Andric       OS << " " << OpcodeName;
8720b57cec5SDimitry Andric     else
8730b57cec5SDimitry Andric       OS << format(" Opcode %x",  Opcode);
8740b57cec5SDimitry Andric     break;
8750b57cec5SDimitry Andric   }
8760b57cec5SDimitry Andric   case OT_None:
8770b57cec5SDimitry Andric     break;
8780b57cec5SDimitry Andric   case OT_Address:
8790b57cec5SDimitry Andric     OS << format(" %" PRIx64, Operand);
8800b57cec5SDimitry Andric     break;
8810b57cec5SDimitry Andric   case OT_Offset:
8820b57cec5SDimitry Andric     // The offsets are all encoded in a unsigned form, but in practice
8830b57cec5SDimitry Andric     // consumers use them signed. It's most certainly legacy due to
8840b57cec5SDimitry Andric     // the lack of signed variants in the first Dwarf standards.
8850b57cec5SDimitry Andric     OS << format(" %+" PRId64, int64_t(Operand));
8860b57cec5SDimitry Andric     break;
8870b57cec5SDimitry Andric   case OT_FactoredCodeOffset: // Always Unsigned
8880b57cec5SDimitry Andric     if (CodeAlignmentFactor)
8890b57cec5SDimitry Andric       OS << format(" %" PRId64, Operand * CodeAlignmentFactor);
8900b57cec5SDimitry Andric     else
8910b57cec5SDimitry Andric       OS << format(" %" PRId64 "*code_alignment_factor" , Operand);
8920b57cec5SDimitry Andric     break;
8930b57cec5SDimitry Andric   case OT_SignedFactDataOffset:
8940b57cec5SDimitry Andric     if (DataAlignmentFactor)
8950b57cec5SDimitry Andric       OS << format(" %" PRId64, int64_t(Operand) * DataAlignmentFactor);
8960b57cec5SDimitry Andric     else
8970b57cec5SDimitry Andric       OS << format(" %" PRId64 "*data_alignment_factor" , int64_t(Operand));
8980b57cec5SDimitry Andric     break;
8990b57cec5SDimitry Andric   case OT_UnsignedFactDataOffset:
9000b57cec5SDimitry Andric     if (DataAlignmentFactor)
9010b57cec5SDimitry Andric       OS << format(" %" PRId64, Operand * DataAlignmentFactor);
9020b57cec5SDimitry Andric     else
9030b57cec5SDimitry Andric       OS << format(" %" PRId64 "*data_alignment_factor" , Operand);
9040b57cec5SDimitry Andric     break;
9050b57cec5SDimitry Andric   case OT_Register:
906e8d8bef9SDimitry Andric     OS << ' ';
907*bdd1243dSDimitry Andric     printRegister(OS, DumpOpts, Operand);
9080b57cec5SDimitry Andric     break;
909fe6060f1SDimitry Andric   case OT_AddressSpace:
910fe6060f1SDimitry Andric     OS << format(" in addrspace%" PRId64, Operand);
911fe6060f1SDimitry Andric     break;
9120b57cec5SDimitry Andric   case OT_Expression:
9130b57cec5SDimitry Andric     assert(Instr.Expression && "missing DWARFExpression object");
9140b57cec5SDimitry Andric     OS << " ";
915*bdd1243dSDimitry Andric     Instr.Expression->print(OS, DumpOpts, nullptr);
9160b57cec5SDimitry Andric     break;
9170b57cec5SDimitry Andric   }
9180b57cec5SDimitry Andric }
9190b57cec5SDimitry Andric 
920e8d8bef9SDimitry Andric void CFIProgram::dump(raw_ostream &OS, DIDumpOptions DumpOpts,
9210b57cec5SDimitry Andric                       unsigned IndentLevel) const {
9220b57cec5SDimitry Andric   for (const auto &Instr : Instructions) {
9230b57cec5SDimitry Andric     uint8_t Opcode = Instr.Opcode;
9240b57cec5SDimitry Andric     OS.indent(2 * IndentLevel);
925fe6060f1SDimitry Andric     OS << callFrameString(Opcode) << ":";
9260b57cec5SDimitry Andric     for (unsigned i = 0; i < Instr.Ops.size(); ++i)
927*bdd1243dSDimitry Andric       printOperand(OS, DumpOpts, Instr, i, Instr.Ops[i]);
9280b57cec5SDimitry Andric     OS << '\n';
9290b57cec5SDimitry Andric   }
9300b57cec5SDimitry Andric }
9310b57cec5SDimitry Andric 
9325ffd83dbSDimitry Andric // Returns the CIE identifier to be used by the requested format.
9335ffd83dbSDimitry Andric // CIE ids for .debug_frame sections are defined in Section 7.24 of DWARFv5.
9345ffd83dbSDimitry Andric // For CIE ID in .eh_frame sections see
9355ffd83dbSDimitry Andric // https://refspecs.linuxfoundation.org/LSB_5.0.0/LSB-Core-generic/LSB-Core-generic/ehframechpt.html
9365ffd83dbSDimitry Andric constexpr uint64_t getCIEId(bool IsDWARF64, bool IsEH) {
9375ffd83dbSDimitry Andric   if (IsEH)
9385ffd83dbSDimitry Andric     return 0;
9395ffd83dbSDimitry Andric   if (IsDWARF64)
9405ffd83dbSDimitry Andric     return DW64_CIE_ID;
9415ffd83dbSDimitry Andric   return DW_CIE_ID;
9425ffd83dbSDimitry Andric }
9435ffd83dbSDimitry Andric 
944*bdd1243dSDimitry Andric void CIE::dump(raw_ostream &OS, DIDumpOptions DumpOpts) const {
9455ffd83dbSDimitry Andric   // A CIE with a zero length is a terminator entry in the .eh_frame section.
946*bdd1243dSDimitry Andric   if (DumpOpts.IsEH && Length == 0) {
9475ffd83dbSDimitry Andric     OS << format("%08" PRIx64, Offset) << " ZERO terminator\n";
9485ffd83dbSDimitry Andric     return;
9495ffd83dbSDimitry Andric   }
9505ffd83dbSDimitry Andric 
9515ffd83dbSDimitry Andric   OS << format("%08" PRIx64, Offset)
9525ffd83dbSDimitry Andric      << format(" %0*" PRIx64, IsDWARF64 ? 16 : 8, Length)
953*bdd1243dSDimitry Andric      << format(" %0*" PRIx64, IsDWARF64 && !DumpOpts.IsEH ? 16 : 8,
954*bdd1243dSDimitry Andric                getCIEId(IsDWARF64, DumpOpts.IsEH))
9555ffd83dbSDimitry Andric      << " CIE\n"
956fe6060f1SDimitry Andric      << "  Format:                " << FormatString(IsDWARF64) << "\n";
957*bdd1243dSDimitry Andric   if (DumpOpts.IsEH && Version != 1)
958fe6060f1SDimitry Andric     OS << "WARNING: unsupported CIE version\n";
959fe6060f1SDimitry Andric   OS << format("  Version:               %d\n", Version)
9605ffd83dbSDimitry Andric      << "  Augmentation:          \"" << Augmentation << "\"\n";
9610b57cec5SDimitry Andric   if (Version >= 4) {
9620b57cec5SDimitry Andric     OS << format("  Address size:          %u\n", (uint32_t)AddressSize);
9630b57cec5SDimitry Andric     OS << format("  Segment desc size:     %u\n",
9640b57cec5SDimitry Andric                  (uint32_t)SegmentDescriptorSize);
9650b57cec5SDimitry Andric   }
9660b57cec5SDimitry Andric   OS << format("  Code alignment factor: %u\n", (uint32_t)CodeAlignmentFactor);
9670b57cec5SDimitry Andric   OS << format("  Data alignment factor: %d\n", (int32_t)DataAlignmentFactor);
9680b57cec5SDimitry Andric   OS << format("  Return address column: %d\n", (int32_t)ReturnAddressRegister);
9690b57cec5SDimitry Andric   if (Personality)
9700b57cec5SDimitry Andric     OS << format("  Personality Address: %016" PRIx64 "\n", *Personality);
9710b57cec5SDimitry Andric   if (!AugmentationData.empty()) {
9720b57cec5SDimitry Andric     OS << "  Augmentation data:    ";
9730b57cec5SDimitry Andric     for (uint8_t Byte : AugmentationData)
9740b57cec5SDimitry Andric       OS << ' ' << hexdigit(Byte >> 4) << hexdigit(Byte & 0xf);
9750b57cec5SDimitry Andric     OS << "\n";
9760b57cec5SDimitry Andric   }
9770b57cec5SDimitry Andric   OS << "\n";
978*bdd1243dSDimitry Andric   CFIs.dump(OS, DumpOpts);
9790b57cec5SDimitry Andric   OS << "\n";
980fe6060f1SDimitry Andric 
981fe6060f1SDimitry Andric   if (Expected<UnwindTable> RowsOrErr = UnwindTable::create(this))
982*bdd1243dSDimitry Andric     RowsOrErr->dump(OS, DumpOpts, 1);
983fe6060f1SDimitry Andric   else {
984fe6060f1SDimitry Andric     DumpOpts.RecoverableErrorHandler(joinErrors(
985fe6060f1SDimitry Andric         createStringError(errc::invalid_argument,
986fe6060f1SDimitry Andric                           "decoding the CIE opcodes into rows failed"),
987fe6060f1SDimitry Andric         RowsOrErr.takeError()));
988fe6060f1SDimitry Andric   }
989fe6060f1SDimitry Andric   OS << "\n";
9900b57cec5SDimitry Andric }
9910b57cec5SDimitry Andric 
992*bdd1243dSDimitry Andric void FDE::dump(raw_ostream &OS, DIDumpOptions DumpOpts) const {
9935ffd83dbSDimitry Andric   OS << format("%08" PRIx64, Offset)
9945ffd83dbSDimitry Andric      << format(" %0*" PRIx64, IsDWARF64 ? 16 : 8, Length)
995*bdd1243dSDimitry Andric      << format(" %0*" PRIx64, IsDWARF64 && !DumpOpts.IsEH ? 16 : 8, CIEPointer)
9965ffd83dbSDimitry Andric      << " FDE cie=";
9975ffd83dbSDimitry Andric   if (LinkedCIE)
9985ffd83dbSDimitry Andric     OS << format("%08" PRIx64, LinkedCIE->getOffset());
9995ffd83dbSDimitry Andric   else
10005ffd83dbSDimitry Andric     OS << "<invalid offset>";
10015ffd83dbSDimitry Andric   OS << format(" pc=%08" PRIx64 "...%08" PRIx64 "\n", InitialLocation,
10025ffd83dbSDimitry Andric                InitialLocation + AddressRange);
10035ffd83dbSDimitry Andric   OS << "  Format:       " << FormatString(IsDWARF64) << "\n";
10040b57cec5SDimitry Andric   if (LSDAAddress)
10050b57cec5SDimitry Andric     OS << format("  LSDA Address: %016" PRIx64 "\n", *LSDAAddress);
1006*bdd1243dSDimitry Andric   CFIs.dump(OS, DumpOpts);
10070b57cec5SDimitry Andric   OS << "\n";
1008fe6060f1SDimitry Andric 
1009fe6060f1SDimitry Andric   if (Expected<UnwindTable> RowsOrErr = UnwindTable::create(this))
1010*bdd1243dSDimitry Andric     RowsOrErr->dump(OS, DumpOpts, 1);
1011fe6060f1SDimitry Andric   else {
1012fe6060f1SDimitry Andric     DumpOpts.RecoverableErrorHandler(joinErrors(
1013fe6060f1SDimitry Andric         createStringError(errc::invalid_argument,
1014fe6060f1SDimitry Andric                           "decoding the FDE opcodes into rows failed"),
1015fe6060f1SDimitry Andric         RowsOrErr.takeError()));
1016fe6060f1SDimitry Andric   }
1017fe6060f1SDimitry Andric   OS << "\n";
10180b57cec5SDimitry Andric }
10190b57cec5SDimitry Andric 
10200b57cec5SDimitry Andric DWARFDebugFrame::DWARFDebugFrame(Triple::ArchType Arch,
10210b57cec5SDimitry Andric     bool IsEH, uint64_t EHFrameAddress)
10220b57cec5SDimitry Andric     : Arch(Arch), IsEH(IsEH), EHFrameAddress(EHFrameAddress) {}
10230b57cec5SDimitry Andric 
10240b57cec5SDimitry Andric DWARFDebugFrame::~DWARFDebugFrame() = default;
10250b57cec5SDimitry Andric 
10260b57cec5SDimitry Andric static void LLVM_ATTRIBUTE_UNUSED dumpDataAux(DataExtractor Data,
10278bcb0991SDimitry Andric                                               uint64_t Offset, int Length) {
10280b57cec5SDimitry Andric   errs() << "DUMP: ";
10290b57cec5SDimitry Andric   for (int i = 0; i < Length; ++i) {
10300b57cec5SDimitry Andric     uint8_t c = Data.getU8(&Offset);
10310b57cec5SDimitry Andric     errs().write_hex(c); errs() << " ";
10320b57cec5SDimitry Andric   }
10330b57cec5SDimitry Andric   errs() << "\n";
10340b57cec5SDimitry Andric }
10350b57cec5SDimitry Andric 
10365ffd83dbSDimitry Andric Error DWARFDebugFrame::parse(DWARFDataExtractor Data) {
10378bcb0991SDimitry Andric   uint64_t Offset = 0;
10388bcb0991SDimitry Andric   DenseMap<uint64_t, CIE *> CIEs;
10390b57cec5SDimitry Andric 
10400b57cec5SDimitry Andric   while (Data.isValidOffset(Offset)) {
10418bcb0991SDimitry Andric     uint64_t StartOffset = Offset;
10420b57cec5SDimitry Andric 
10435ffd83dbSDimitry Andric     uint64_t Length;
10445ffd83dbSDimitry Andric     DwarfFormat Format;
10455ffd83dbSDimitry Andric     std::tie(Length, Format) = Data.getInitialLength(&Offset);
10465ffd83dbSDimitry Andric     bool IsDWARF64 = Format == DWARF64;
10470b57cec5SDimitry Andric 
10485ffd83dbSDimitry Andric     // If the Length is 0, then this CIE is a terminator. We add it because some
10495ffd83dbSDimitry Andric     // dumper tools might need it to print something special for such entries
10505ffd83dbSDimitry Andric     // (e.g. llvm-objdump --dwarf=frames prints "ZERO terminator").
10515ffd83dbSDimitry Andric     if (Length == 0) {
10525ffd83dbSDimitry Andric       auto Cie = std::make_unique<CIE>(
10535ffd83dbSDimitry Andric           IsDWARF64, StartOffset, 0, 0, SmallString<8>(), 0, 0, 0, 0, 0,
1054*bdd1243dSDimitry Andric           SmallString<8>(), 0, 0, std::nullopt, std::nullopt, Arch);
10555ffd83dbSDimitry Andric       CIEs[StartOffset] = Cie.get();
10565ffd83dbSDimitry Andric       Entries.push_back(std::move(Cie));
10575ffd83dbSDimitry Andric       break;
10580b57cec5SDimitry Andric     }
10590b57cec5SDimitry Andric 
10600b57cec5SDimitry Andric     // At this point, Offset points to the next field after Length.
10610b57cec5SDimitry Andric     // Length is the structure size excluding itself. Compute an offset one
10620b57cec5SDimitry Andric     // past the end of the structure (needed to know how many instructions to
10630b57cec5SDimitry Andric     // read).
10648bcb0991SDimitry Andric     uint64_t StartStructureOffset = Offset;
10658bcb0991SDimitry Andric     uint64_t EndStructureOffset = Offset + Length;
10660b57cec5SDimitry Andric 
10670b57cec5SDimitry Andric     // The Id field's size depends on the DWARF format
10685ffd83dbSDimitry Andric     Error Err = Error::success();
10695ffd83dbSDimitry Andric     uint64_t Id = Data.getRelocatedValue((IsDWARF64 && !IsEH) ? 8 : 4, &Offset,
10705ffd83dbSDimitry Andric                                          /*SectionIndex=*/nullptr, &Err);
10715ffd83dbSDimitry Andric     if (Err)
10725ffd83dbSDimitry Andric       return Err;
10730b57cec5SDimitry Andric 
10745ffd83dbSDimitry Andric     if (Id == getCIEId(IsDWARF64, IsEH)) {
10750b57cec5SDimitry Andric       uint8_t Version = Data.getU8(&Offset);
10760b57cec5SDimitry Andric       const char *Augmentation = Data.getCStr(&Offset);
10770b57cec5SDimitry Andric       StringRef AugmentationString(Augmentation ? Augmentation : "");
10780b57cec5SDimitry Andric       uint8_t AddressSize = Version < 4 ? Data.getAddressSize() :
10790b57cec5SDimitry Andric                                           Data.getU8(&Offset);
10800b57cec5SDimitry Andric       Data.setAddressSize(AddressSize);
10810b57cec5SDimitry Andric       uint8_t SegmentDescriptorSize = Version < 4 ? 0 : Data.getU8(&Offset);
10820b57cec5SDimitry Andric       uint64_t CodeAlignmentFactor = Data.getULEB128(&Offset);
10830b57cec5SDimitry Andric       int64_t DataAlignmentFactor = Data.getSLEB128(&Offset);
10840b57cec5SDimitry Andric       uint64_t ReturnAddressRegister =
10850b57cec5SDimitry Andric           Version == 1 ? Data.getU8(&Offset) : Data.getULEB128(&Offset);
10860b57cec5SDimitry Andric 
10870b57cec5SDimitry Andric       // Parse the augmentation data for EH CIEs
10880b57cec5SDimitry Andric       StringRef AugmentationData("");
10890b57cec5SDimitry Andric       uint32_t FDEPointerEncoding = DW_EH_PE_absptr;
10900b57cec5SDimitry Andric       uint32_t LSDAPointerEncoding = DW_EH_PE_omit;
1091*bdd1243dSDimitry Andric       std::optional<uint64_t> Personality;
1092*bdd1243dSDimitry Andric       std::optional<uint32_t> PersonalityEncoding;
10930b57cec5SDimitry Andric       if (IsEH) {
1094*bdd1243dSDimitry Andric         std::optional<uint64_t> AugmentationLength;
10958bcb0991SDimitry Andric         uint64_t StartAugmentationOffset;
10968bcb0991SDimitry Andric         uint64_t EndAugmentationOffset;
10970b57cec5SDimitry Andric 
10980b57cec5SDimitry Andric         // Walk the augmentation string to get all the augmentation data.
10990b57cec5SDimitry Andric         for (unsigned i = 0, e = AugmentationString.size(); i != e; ++i) {
11000b57cec5SDimitry Andric           switch (AugmentationString[i]) {
11010b57cec5SDimitry Andric           default:
11025ffd83dbSDimitry Andric             return createStringError(
11035ffd83dbSDimitry Andric                 errc::invalid_argument,
110481ad6265SDimitry Andric                 "unknown augmentation character %c in entry at 0x%" PRIx64,
110581ad6265SDimitry Andric                 AugmentationString[i], StartOffset);
11060b57cec5SDimitry Andric           case 'L':
11070b57cec5SDimitry Andric             LSDAPointerEncoding = Data.getU8(&Offset);
11080b57cec5SDimitry Andric             break;
11090b57cec5SDimitry Andric           case 'P': {
11100b57cec5SDimitry Andric             if (Personality)
11115ffd83dbSDimitry Andric               return createStringError(
11125ffd83dbSDimitry Andric                   errc::invalid_argument,
11135ffd83dbSDimitry Andric                   "duplicate personality in entry at 0x%" PRIx64, StartOffset);
11140b57cec5SDimitry Andric             PersonalityEncoding = Data.getU8(&Offset);
11150b57cec5SDimitry Andric             Personality = Data.getEncodedPointer(
11160b57cec5SDimitry Andric                 &Offset, *PersonalityEncoding,
11170b57cec5SDimitry Andric                 EHFrameAddress ? EHFrameAddress + Offset : 0);
11180b57cec5SDimitry Andric             break;
11190b57cec5SDimitry Andric           }
11200b57cec5SDimitry Andric           case 'R':
11210b57cec5SDimitry Andric             FDEPointerEncoding = Data.getU8(&Offset);
11220b57cec5SDimitry Andric             break;
11230b57cec5SDimitry Andric           case 'S':
11240b57cec5SDimitry Andric             // Current frame is a signal trampoline.
11250b57cec5SDimitry Andric             break;
11260b57cec5SDimitry Andric           case 'z':
11270b57cec5SDimitry Andric             if (i)
11285ffd83dbSDimitry Andric               return createStringError(
11295ffd83dbSDimitry Andric                   errc::invalid_argument,
11305ffd83dbSDimitry Andric                   "'z' must be the first character at 0x%" PRIx64, StartOffset);
11310b57cec5SDimitry Andric             // Parse the augmentation length first.  We only parse it if
11320b57cec5SDimitry Andric             // the string contains a 'z'.
11330b57cec5SDimitry Andric             AugmentationLength = Data.getULEB128(&Offset);
11340b57cec5SDimitry Andric             StartAugmentationOffset = Offset;
11358bcb0991SDimitry Andric             EndAugmentationOffset = Offset + *AugmentationLength;
11360b57cec5SDimitry Andric             break;
11370b57cec5SDimitry Andric           case 'B':
11380b57cec5SDimitry Andric             // B-Key is used for signing functions associated with this
11390b57cec5SDimitry Andric             // augmentation string
11400b57cec5SDimitry Andric             break;
114181ad6265SDimitry Andric             // This stack frame contains MTE tagged data, so needs to be
114281ad6265SDimitry Andric             // untagged on unwind.
114381ad6265SDimitry Andric           case 'G':
114481ad6265SDimitry Andric             break;
11450b57cec5SDimitry Andric           }
11460b57cec5SDimitry Andric         }
11470b57cec5SDimitry Andric 
114881ad6265SDimitry Andric         if (AugmentationLength) {
11490b57cec5SDimitry Andric           if (Offset != EndAugmentationOffset)
11505ffd83dbSDimitry Andric             return createStringError(errc::invalid_argument,
11515ffd83dbSDimitry Andric                                      "parsing augmentation data at 0x%" PRIx64
11525ffd83dbSDimitry Andric                                      " failed",
11535ffd83dbSDimitry Andric                                      StartOffset);
11540b57cec5SDimitry Andric           AugmentationData = Data.getData().slice(StartAugmentationOffset,
11550b57cec5SDimitry Andric                                                   EndAugmentationOffset);
11560b57cec5SDimitry Andric         }
11570b57cec5SDimitry Andric       }
11580b57cec5SDimitry Andric 
11598bcb0991SDimitry Andric       auto Cie = std::make_unique<CIE>(
11605ffd83dbSDimitry Andric           IsDWARF64, StartOffset, Length, Version, AugmentationString,
11615ffd83dbSDimitry Andric           AddressSize, SegmentDescriptorSize, CodeAlignmentFactor,
11625ffd83dbSDimitry Andric           DataAlignmentFactor, ReturnAddressRegister, AugmentationData,
11635ffd83dbSDimitry Andric           FDEPointerEncoding, LSDAPointerEncoding, Personality,
11645ffd83dbSDimitry Andric           PersonalityEncoding, Arch);
11650b57cec5SDimitry Andric       CIEs[StartOffset] = Cie.get();
11660b57cec5SDimitry Andric       Entries.emplace_back(std::move(Cie));
11670b57cec5SDimitry Andric     } else {
11680b57cec5SDimitry Andric       // FDE
11690b57cec5SDimitry Andric       uint64_t CIEPointer = Id;
11700b57cec5SDimitry Andric       uint64_t InitialLocation = 0;
11710b57cec5SDimitry Andric       uint64_t AddressRange = 0;
1172*bdd1243dSDimitry Andric       std::optional<uint64_t> LSDAAddress;
11730b57cec5SDimitry Andric       CIE *Cie = CIEs[IsEH ? (StartStructureOffset - CIEPointer) : CIEPointer];
11740b57cec5SDimitry Andric 
11750b57cec5SDimitry Andric       if (IsEH) {
11760b57cec5SDimitry Andric         // The address size is encoded in the CIE we reference.
11770b57cec5SDimitry Andric         if (!Cie)
11785ffd83dbSDimitry Andric           return createStringError(errc::invalid_argument,
11795ffd83dbSDimitry Andric                                    "parsing FDE data at 0x%" PRIx64
11805ffd83dbSDimitry Andric                                    " failed due to missing CIE",
11815ffd83dbSDimitry Andric                                    StartOffset);
1182e8d8bef9SDimitry Andric         if (auto Val =
1183e8d8bef9SDimitry Andric                 Data.getEncodedPointer(&Offset, Cie->getFDEPointerEncoding(),
1184e8d8bef9SDimitry Andric                                        EHFrameAddress + Offset)) {
11850b57cec5SDimitry Andric           InitialLocation = *Val;
11860b57cec5SDimitry Andric         }
11870b57cec5SDimitry Andric         if (auto Val = Data.getEncodedPointer(
11880b57cec5SDimitry Andric                 &Offset, Cie->getFDEPointerEncoding(), 0)) {
11890b57cec5SDimitry Andric           AddressRange = *Val;
11900b57cec5SDimitry Andric         }
11910b57cec5SDimitry Andric 
11920b57cec5SDimitry Andric         StringRef AugmentationString = Cie->getAugmentationString();
11930b57cec5SDimitry Andric         if (!AugmentationString.empty()) {
11940b57cec5SDimitry Andric           // Parse the augmentation length and data for this FDE.
11950b57cec5SDimitry Andric           uint64_t AugmentationLength = Data.getULEB128(&Offset);
11960b57cec5SDimitry Andric 
11978bcb0991SDimitry Andric           uint64_t EndAugmentationOffset = Offset + AugmentationLength;
11980b57cec5SDimitry Andric 
11990b57cec5SDimitry Andric           // Decode the LSDA if the CIE augmentation string said we should.
12000b57cec5SDimitry Andric           if (Cie->getLSDAPointerEncoding() != DW_EH_PE_omit) {
12010b57cec5SDimitry Andric             LSDAAddress = Data.getEncodedPointer(
12020b57cec5SDimitry Andric                 &Offset, Cie->getLSDAPointerEncoding(),
12030b57cec5SDimitry Andric                 EHFrameAddress ? Offset + EHFrameAddress : 0);
12040b57cec5SDimitry Andric           }
12050b57cec5SDimitry Andric 
12060b57cec5SDimitry Andric           if (Offset != EndAugmentationOffset)
12075ffd83dbSDimitry Andric             return createStringError(errc::invalid_argument,
12085ffd83dbSDimitry Andric                                      "parsing augmentation data at 0x%" PRIx64
12095ffd83dbSDimitry Andric                                      " failed",
12105ffd83dbSDimitry Andric                                      StartOffset);
12110b57cec5SDimitry Andric         }
12120b57cec5SDimitry Andric       } else {
12130b57cec5SDimitry Andric         InitialLocation = Data.getRelocatedAddress(&Offset);
12140b57cec5SDimitry Andric         AddressRange = Data.getRelocatedAddress(&Offset);
12150b57cec5SDimitry Andric       }
12160b57cec5SDimitry Andric 
12175ffd83dbSDimitry Andric       Entries.emplace_back(new FDE(IsDWARF64, StartOffset, Length, CIEPointer,
12185ffd83dbSDimitry Andric                                    InitialLocation, AddressRange, Cie,
12195ffd83dbSDimitry Andric                                    LSDAAddress, Arch));
12200b57cec5SDimitry Andric     }
12210b57cec5SDimitry Andric 
12220b57cec5SDimitry Andric     if (Error E =
12235ffd83dbSDimitry Andric             Entries.back()->cfis().parse(Data, &Offset, EndStructureOffset))
12245ffd83dbSDimitry Andric       return E;
12250b57cec5SDimitry Andric 
12260b57cec5SDimitry Andric     if (Offset != EndStructureOffset)
12275ffd83dbSDimitry Andric       return createStringError(
12285ffd83dbSDimitry Andric           errc::invalid_argument,
12295ffd83dbSDimitry Andric           "parsing entry instructions at 0x%" PRIx64 " failed", StartOffset);
12300b57cec5SDimitry Andric   }
12315ffd83dbSDimitry Andric 
12325ffd83dbSDimitry Andric   return Error::success();
12330b57cec5SDimitry Andric }
12340b57cec5SDimitry Andric 
12350b57cec5SDimitry Andric FrameEntry *DWARFDebugFrame::getEntryAtOffset(uint64_t Offset) const {
12360b57cec5SDimitry Andric   auto It = partition_point(Entries, [=](const std::unique_ptr<FrameEntry> &E) {
12370b57cec5SDimitry Andric     return E->getOffset() < Offset;
12380b57cec5SDimitry Andric   });
12390b57cec5SDimitry Andric   if (It != Entries.end() && (*It)->getOffset() == Offset)
12400b57cec5SDimitry Andric     return It->get();
12410b57cec5SDimitry Andric   return nullptr;
12420b57cec5SDimitry Andric }
12430b57cec5SDimitry Andric 
1244e8d8bef9SDimitry Andric void DWARFDebugFrame::dump(raw_ostream &OS, DIDumpOptions DumpOpts,
1245*bdd1243dSDimitry Andric                            std::optional<uint64_t> Offset) const {
1246*bdd1243dSDimitry Andric   DumpOpts.IsEH = IsEH;
12470b57cec5SDimitry Andric   if (Offset) {
12480b57cec5SDimitry Andric     if (auto *Entry = getEntryAtOffset(*Offset))
1249*bdd1243dSDimitry Andric       Entry->dump(OS, DumpOpts);
12500b57cec5SDimitry Andric     return;
12510b57cec5SDimitry Andric   }
12520b57cec5SDimitry Andric 
12530b57cec5SDimitry Andric   OS << "\n";
12540b57cec5SDimitry Andric   for (const auto &Entry : Entries)
1255*bdd1243dSDimitry Andric     Entry->dump(OS, DumpOpts);
12560b57cec5SDimitry Andric }
1257