David Blaikie | f72dbc1 | 2016-03-01 22:29:00 +0000 | [diff] [blame] | 1 | //===-- llvm-dwp.cpp - Split DWARF merging tool for llvm ------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // A utility for merging DWARF 5 Split DWARF .dwo files into .dwp (DWARF |
| 11 | // package files). |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 14 | #include "llvm/ADT/MapVector.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/STLExtras.h" |
| 16 | #include "llvm/ADT/StringSet.h" |
| 17 | #include "llvm/CodeGen/AsmPrinter.h" |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 18 | #include "llvm/DebugInfo/DWARF/DWARFFormValue.h" |
| 19 | #include "llvm/DebugInfo/DWARF/DWARFUnitIndex.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 20 | #include "llvm/MC/MCAsmInfo.h" |
| 21 | #include "llvm/MC/MCContext.h" |
| 22 | #include "llvm/MC/MCInstrInfo.h" |
| 23 | #include "llvm/MC/MCObjectFileInfo.h" |
| 24 | #include "llvm/MC/MCRegisterInfo.h" |
| 25 | #include "llvm/MC/MCSectionELF.h" |
| 26 | #include "llvm/MC/MCStreamer.h" |
David Majnemer | 03e2cc3 | 2015-12-21 22:09:27 +0000 | [diff] [blame] | 27 | #include "llvm/MC/MCTargetOptionsCommandFlags.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 28 | #include "llvm/Object/ObjectFile.h" |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 29 | #include "llvm/Support/Compression.h" |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 30 | #include "llvm/Support/DataExtractor.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 31 | #include "llvm/Support/FileSystem.h" |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 32 | #include "llvm/Support/MathExtras.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 33 | #include "llvm/Support/MemoryBuffer.h" |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 34 | #include "llvm/Support/Options.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 35 | #include "llvm/Support/TargetRegistry.h" |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 36 | #include "llvm/Support/TargetSelect.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 37 | #include "llvm/Support/raw_ostream.h" |
| 38 | #include "llvm/Target/TargetMachine.h" |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 39 | #include <list> |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 40 | #include <iostream> |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 41 | #include <memory> |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 42 | #include <system_error> |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 43 | #include <unordered_set> |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 44 | |
| 45 | using namespace llvm; |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 46 | using namespace llvm::object; |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 47 | using namespace cl; |
| 48 | |
| 49 | OptionCategory DwpCategory("Specific Options"); |
| 50 | static list<std::string> InputFiles(Positional, OneOrMore, |
| 51 | desc("<input files>"), cat(DwpCategory)); |
| 52 | |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 53 | static opt<std::string> OutputFilename(Required, "o", |
| 54 | desc("Specify the output file."), |
| 55 | value_desc("filename"), |
| 56 | cat(DwpCategory)); |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 57 | |
| 58 | static int error(const Twine &Error, const Twine &Context) { |
| 59 | errs() << Twine("while processing ") + Context + ":\n"; |
| 60 | errs() << Twine("error: ") + Error + "\n"; |
| 61 | return 1; |
| 62 | } |
| 63 | |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 64 | static std::error_code |
| 65 | writeStringsAndOffsets(MCStreamer &Out, StringMap<uint32_t> &Strings, |
David Blaikie | bb94e44 | 2015-12-01 19:17:58 +0000 | [diff] [blame] | 66 | uint32_t &StringOffset, MCSection *StrSection, |
| 67 | MCSection *StrOffsetSection, StringRef CurStrSection, |
| 68 | StringRef CurStrOffsetSection) { |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 69 | // Could possibly produce an error or warning if one of these was non-null but |
| 70 | // the other was null. |
| 71 | if (CurStrSection.empty() || CurStrOffsetSection.empty()) |
| 72 | return std::error_code(); |
| 73 | |
| 74 | DenseMap<uint32_t, uint32_t> OffsetRemapping; |
| 75 | |
| 76 | DataExtractor Data(CurStrSection, true, 0); |
| 77 | uint32_t LocalOffset = 0; |
| 78 | uint32_t PrevOffset = 0; |
| 79 | while (const char *s = Data.getCStr(&LocalOffset)) { |
| 80 | StringRef Str(s, LocalOffset - PrevOffset - 1); |
David Blaikie | bb94e44 | 2015-12-01 19:17:58 +0000 | [diff] [blame] | 81 | auto Pair = Strings.insert(std::make_pair(Str, StringOffset)); |
| 82 | if (Pair.second) { |
| 83 | Out.SwitchSection(StrSection); |
| 84 | Out.EmitBytes( |
| 85 | StringRef(Pair.first->getKeyData(), Pair.first->getKeyLength() + 1)); |
| 86 | StringOffset += Str.size() + 1; |
| 87 | } |
| 88 | OffsetRemapping[PrevOffset] = Pair.first->second; |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 89 | PrevOffset = LocalOffset; |
| 90 | } |
| 91 | |
| 92 | Data = DataExtractor(CurStrOffsetSection, true, 0); |
| 93 | |
| 94 | Out.SwitchSection(StrOffsetSection); |
| 95 | |
| 96 | uint32_t Offset = 0; |
| 97 | uint64_t Size = CurStrOffsetSection.size(); |
| 98 | while (Offset < Size) { |
| 99 | auto OldOffset = Data.getU32(&Offset); |
| 100 | auto NewOffset = OffsetRemapping[OldOffset]; |
| 101 | Out.EmitIntValue(NewOffset, 4); |
| 102 | } |
| 103 | |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 104 | return std::error_code(); |
| 105 | } |
| 106 | |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 107 | static uint32_t getCUAbbrev(StringRef Abbrev, uint64_t AbbrCode) { |
| 108 | uint64_t CurCode; |
| 109 | uint32_t Offset = 0; |
| 110 | DataExtractor AbbrevData(Abbrev, true, 0); |
| 111 | while ((CurCode = AbbrevData.getULEB128(&Offset)) != AbbrCode) { |
| 112 | // Tag |
| 113 | AbbrevData.getULEB128(&Offset); |
| 114 | // DW_CHILDREN |
| 115 | AbbrevData.getU8(&Offset); |
| 116 | // Attributes |
| 117 | while (AbbrevData.getULEB128(&Offset) | AbbrevData.getULEB128(&Offset)) |
| 118 | ; |
| 119 | } |
| 120 | return Offset; |
| 121 | } |
| 122 | |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 123 | struct CompileUnitIdentifiers { |
| 124 | uint64_t Signature = 0; |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 125 | const char *Name = ""; |
| 126 | const char *DWOName = ""; |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 127 | }; |
| 128 | |
David Blaikie | 60fbd3b | 2016-04-05 20:16:38 +0000 | [diff] [blame] | 129 | static const char *getIndexedString(uint32_t Form, DataExtractor InfoData, |
| 130 | uint32_t &InfoOffset, StringRef StrOffsets, |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 131 | StringRef Str) { |
David Blaikie | 60fbd3b | 2016-04-05 20:16:38 +0000 | [diff] [blame] | 132 | if (Form == dwarf::DW_FORM_string) |
| 133 | return InfoData.getCStr(&InfoOffset); |
David Blaikie | ee21080 | 2016-04-13 18:38:33 +0000 | [diff] [blame^] | 134 | assert(Form == dwarf::DW_FORM_GNU_str_index && "Only string and str_index " |
| 135 | "forms are supported for DWP " |
| 136 | "string attributes"); |
David Blaikie | 60fbd3b | 2016-04-05 20:16:38 +0000 | [diff] [blame] | 137 | auto StrIndex = InfoData.getULEB128(&InfoOffset); |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 138 | DataExtractor StrOffsetsData(StrOffsets, true, 0); |
| 139 | uint32_t StrOffsetsOffset = 4 * StrIndex; |
| 140 | uint32_t StrOffset = StrOffsetsData.getU32(&StrOffsetsOffset); |
| 141 | DataExtractor StrData(Str, true, 0); |
| 142 | return StrData.getCStr(&StrOffset); |
| 143 | } |
| 144 | |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 145 | static CompileUnitIdentifiers getCUIdentifiers(StringRef Abbrev, StringRef Info, |
| 146 | StringRef StrOffsets, |
| 147 | StringRef Str) { |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 148 | uint32_t Offset = 0; |
| 149 | DataExtractor InfoData(Info, true, 0); |
| 150 | InfoData.getU32(&Offset); // Length |
| 151 | uint16_t Version = InfoData.getU16(&Offset); |
| 152 | InfoData.getU32(&Offset); // Abbrev offset (should be zero) |
| 153 | uint8_t AddrSize = InfoData.getU8(&Offset); |
| 154 | |
| 155 | uint32_t AbbrCode = InfoData.getULEB128(&Offset); |
| 156 | |
| 157 | DataExtractor AbbrevData(Abbrev, true, 0); |
| 158 | uint32_t AbbrevOffset = getCUAbbrev(Abbrev, AbbrCode); |
| 159 | uint64_t Tag = AbbrevData.getULEB128(&AbbrevOffset); |
| 160 | (void)Tag; |
| 161 | // FIXME: Real error handling |
| 162 | assert(Tag == dwarf::DW_TAG_compile_unit); |
| 163 | // DW_CHILDREN |
| 164 | AbbrevData.getU8(&AbbrevOffset); |
| 165 | uint32_t Name; |
| 166 | uint32_t Form; |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 167 | CompileUnitIdentifiers ID; |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 168 | while ((Name = AbbrevData.getULEB128(&AbbrevOffset)) | |
| 169 | (Form = AbbrevData.getULEB128(&AbbrevOffset)) && |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 170 | (Name != 0 || Form != 0)) { |
| 171 | switch (Name) { |
| 172 | case dwarf::DW_AT_name: { |
David Blaikie | 60fbd3b | 2016-04-05 20:16:38 +0000 | [diff] [blame] | 173 | ID.Name = getIndexedString(Form, InfoData, Offset, StrOffsets, Str); |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 174 | break; |
| 175 | } |
| 176 | case dwarf::DW_AT_GNU_dwo_name: { |
David Blaikie | 60fbd3b | 2016-04-05 20:16:38 +0000 | [diff] [blame] | 177 | ID.DWOName = getIndexedString(Form, InfoData, Offset, StrOffsets, Str); |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 178 | break; |
| 179 | } |
| 180 | case dwarf::DW_AT_GNU_dwo_id: |
| 181 | ID.Signature = InfoData.getU64(&Offset); |
| 182 | break; |
| 183 | default: |
| 184 | DWARFFormValue::skipValue(Form, InfoData, &Offset, Version, AddrSize); |
| 185 | } |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 186 | } |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 187 | return ID; |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 188 | } |
| 189 | |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 190 | struct UnitIndexEntry { |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 191 | DWARFUnitIndex::Entry::SectionContribution Contributions[8]; |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 192 | std::string Name; |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 193 | std::string DWOName; |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 194 | StringRef DWPName; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 195 | }; |
| 196 | |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 197 | StringRef getSubsection(StringRef Section, const DWARFUnitIndex::Entry &Entry, DWARFSectionKind Kind) { |
| 198 | const auto *Off = Entry.getOffset(Kind); |
| 199 | if (!Off) |
| 200 | return StringRef(); |
| 201 | return Section.substr(Off->Offset, Off->Length); |
| 202 | } |
| 203 | |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 204 | static void addAllTypesFromDWP( |
| 205 | MCStreamer &Out, MapVector<uint64_t, UnitIndexEntry> &TypeIndexEntries, |
| 206 | const DWARFUnitIndex &TUIndex, MCSection *OutputTypes, StringRef Types, |
| 207 | const UnitIndexEntry &TUEntry, uint32_t &TypesOffset) { |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 208 | Out.SwitchSection(OutputTypes); |
| 209 | for (const DWARFUnitIndex::Entry &E : TUIndex.getRows()) { |
| 210 | auto *I = E.getOffsets(); |
| 211 | if (!I) |
| 212 | continue; |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 213 | auto P = TypeIndexEntries.insert(std::make_pair(E.getSignature(), TUEntry)); |
| 214 | if (!P.second) |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 215 | continue; |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 216 | auto &Entry = P.first->second; |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 217 | // Zero out the debug_info contribution |
| 218 | Entry.Contributions[0] = {}; |
| 219 | for (auto Kind : TUIndex.getColumnKinds()) { |
| 220 | auto &C = Entry.Contributions[Kind - DW_SECT_INFO]; |
| 221 | C.Offset += I->Offset; |
| 222 | C.Length = I->Length; |
| 223 | ++I; |
| 224 | } |
| 225 | auto &C = Entry.Contributions[DW_SECT_TYPES - DW_SECT_INFO]; |
| 226 | Out.EmitBytes(Types.substr( |
| 227 | C.Offset - TUEntry.Contributions[DW_SECT_TYPES - DW_SECT_INFO].Offset, |
| 228 | C.Length)); |
| 229 | C.Offset = TypesOffset; |
| 230 | TypesOffset += C.Length; |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 231 | } |
| 232 | } |
| 233 | |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 234 | static void addAllTypes(MCStreamer &Out, |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 235 | MapVector<uint64_t, UnitIndexEntry> &TypeIndexEntries, |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 236 | MCSection *OutputTypes, |
| 237 | const std::vector<StringRef> &TypesSections, |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 238 | const UnitIndexEntry &CUEntry, uint32_t &TypesOffset) { |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 239 | for (StringRef Types : TypesSections) { |
| 240 | Out.SwitchSection(OutputTypes); |
| 241 | uint32_t Offset = 0; |
| 242 | DataExtractor Data(Types, true, 0); |
| 243 | while (Data.isValidOffset(Offset)) { |
| 244 | UnitIndexEntry Entry = CUEntry; |
| 245 | // Zero out the debug_info contribution |
| 246 | Entry.Contributions[0] = {}; |
| 247 | auto &C = Entry.Contributions[DW_SECT_TYPES - DW_SECT_INFO]; |
| 248 | C.Offset = TypesOffset; |
| 249 | auto PrevOffset = Offset; |
| 250 | // Length of the unit, including the 4 byte length field. |
| 251 | C.Length = Data.getU32(&Offset) + 4; |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 252 | |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 253 | Data.getU16(&Offset); // Version |
| 254 | Data.getU32(&Offset); // Abbrev offset |
| 255 | Data.getU8(&Offset); // Address size |
| 256 | auto Signature = Data.getU64(&Offset); |
| 257 | Offset = PrevOffset + C.Length; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 258 | |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 259 | auto P = TypeIndexEntries.insert(std::make_pair(Signature, Entry)); |
| 260 | if (!P.second) |
| 261 | continue; |
David Blaikie | f5cb627 | 2015-12-14 07:42:00 +0000 | [diff] [blame] | 262 | |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 263 | Out.EmitBytes(Types.substr(PrevOffset, C.Length)); |
| 264 | TypesOffset += C.Length; |
| 265 | } |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 266 | } |
| 267 | } |
| 268 | |
| 269 | static void |
| 270 | writeIndexTable(MCStreamer &Out, ArrayRef<unsigned> ContributionOffsets, |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 271 | const MapVector<uint64_t, UnitIndexEntry> &IndexEntries, |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 272 | uint32_t DWARFUnitIndex::Entry::SectionContribution::*Field) { |
| 273 | for (const auto &E : IndexEntries) |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 274 | for (size_t i = 0; i != array_lengthof(E.second.Contributions); ++i) |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 275 | if (ContributionOffsets[i]) |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 276 | Out.EmitIntValue(E.second.Contributions[i].*Field, 4); |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 277 | } |
| 278 | |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 279 | static void |
| 280 | writeIndex(MCStreamer &Out, MCSection *Section, |
| 281 | ArrayRef<unsigned> ContributionOffsets, |
| 282 | const MapVector<uint64_t, UnitIndexEntry> &IndexEntries) { |
David Blaikie | 5d6d4dc | 2016-02-26 07:04:58 +0000 | [diff] [blame] | 283 | if (IndexEntries.empty()) |
| 284 | return; |
| 285 | |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 286 | unsigned Columns = 0; |
| 287 | for (auto &C : ContributionOffsets) |
| 288 | if (C) |
| 289 | ++Columns; |
| 290 | |
| 291 | std::vector<unsigned> Buckets(NextPowerOf2(3 * IndexEntries.size() / 2)); |
| 292 | uint64_t Mask = Buckets.size() - 1; |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 293 | size_t i = 0; |
| 294 | for (const auto &P : IndexEntries) { |
| 295 | auto S = P.first; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 296 | auto H = S & Mask; |
David Blaikie | 9b49256 | 2016-04-05 17:51:40 +0000 | [diff] [blame] | 297 | auto HP = ((S >> 32) & Mask) | 1; |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 298 | while (Buckets[H]) { |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 299 | assert(S != IndexEntries.begin()[Buckets[H] - 1].first && |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 300 | "Duplicate unit"); |
David Blaikie | 9b49256 | 2016-04-05 17:51:40 +0000 | [diff] [blame] | 301 | H = (H + HP) & Mask; |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 302 | } |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 303 | Buckets[H] = i + 1; |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 304 | ++i; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 305 | } |
| 306 | |
| 307 | Out.SwitchSection(Section); |
| 308 | Out.EmitIntValue(2, 4); // Version |
| 309 | Out.EmitIntValue(Columns, 4); // Columns |
| 310 | Out.EmitIntValue(IndexEntries.size(), 4); // Num Units |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 311 | Out.EmitIntValue(Buckets.size(), 4); // Num Buckets |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 312 | |
| 313 | // Write the signatures. |
| 314 | for (const auto &I : Buckets) |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 315 | Out.EmitIntValue(I ? IndexEntries.begin()[I - 1].first : 0, 8); |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 316 | |
| 317 | // Write the indexes. |
| 318 | for (const auto &I : Buckets) |
| 319 | Out.EmitIntValue(I, 4); |
| 320 | |
| 321 | // Write the column headers (which sections will appear in the table) |
| 322 | for (size_t i = 0; i != ContributionOffsets.size(); ++i) |
| 323 | if (ContributionOffsets[i]) |
| 324 | Out.EmitIntValue(i + DW_SECT_INFO, 4); |
| 325 | |
| 326 | // Write the offsets. |
| 327 | writeIndexTable(Out, ContributionOffsets, IndexEntries, |
| 328 | &DWARFUnitIndex::Entry::SectionContribution::Offset); |
| 329 | |
| 330 | // Write the lengths. |
| 331 | writeIndexTable(Out, ContributionOffsets, IndexEntries, |
| 332 | &DWARFUnitIndex::Entry::SectionContribution::Length); |
| 333 | } |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 334 | static bool consumeCompressedDebugSectionHeader(StringRef &data, |
| 335 | uint64_t &OriginalSize) { |
| 336 | // Consume "ZLIB" prefix. |
| 337 | if (!data.startswith("ZLIB")) |
| 338 | return false; |
| 339 | data = data.substr(4); |
| 340 | // Consume uncompressed section size (big-endian 8 bytes). |
| 341 | DataExtractor extractor(data, false, 8); |
| 342 | uint32_t Offset = 0; |
| 343 | OriginalSize = extractor.getU64(&Offset); |
| 344 | if (Offset == 0) |
| 345 | return false; |
| 346 | data = data.substr(Offset); |
| 347 | return true; |
| 348 | } |
| 349 | |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 350 | void printDuplicateError(const std::pair<uint64_t, UnitIndexEntry> &PrevE, |
| 351 | const CompileUnitIdentifiers &ID, StringRef DWPName) { |
| 352 | errs() << "Duplicate DWO ID (" << PrevE.first << ") in '" << PrevE.second.Name |
| 353 | << '\''; |
| 354 | if (!PrevE.second.DWPName.empty()) { |
| 355 | errs() << " (from "; |
| 356 | if (!PrevE.second.DWOName.empty()) |
| 357 | errs() << '\'' << PrevE.second.DWOName << "' in "; |
| 358 | errs() << "'" << PrevE.second.DWPName.str() << "')"; |
| 359 | } |
| 360 | errs() << " and '" << ID.Name << '\''; |
| 361 | if (!DWPName.empty()) { |
| 362 | errs() << " (from "; |
| 363 | if (*ID.DWOName) |
| 364 | errs() << '\'' << ID.DWOName << "\' in "; |
| 365 | errs() << '\'' << DWPName << "')"; |
| 366 | } |
| 367 | errs() << '\n'; |
| 368 | } |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 369 | static std::error_code write(MCStreamer &Out, ArrayRef<std::string> Inputs) { |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 370 | const auto &MCOFI = *Out.getContext().getObjectFileInfo(); |
| 371 | MCSection *const StrSection = MCOFI.getDwarfStrDWOSection(); |
| 372 | MCSection *const StrOffsetSection = MCOFI.getDwarfStrOffDWOSection(); |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 373 | MCSection *const TypesSection = MCOFI.getDwarfTypesDWOSection(); |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 374 | MCSection *const CUIndexSection = MCOFI.getDwarfCUIndexSection(); |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 375 | MCSection *const TUIndexSection = MCOFI.getDwarfTUIndexSection(); |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 376 | const StringMap<std::pair<MCSection *, DWARFSectionKind>> KnownSections = { |
| 377 | {"debug_info.dwo", {MCOFI.getDwarfInfoDWOSection(), DW_SECT_INFO}}, |
| 378 | {"debug_types.dwo", {MCOFI.getDwarfTypesDWOSection(), DW_SECT_TYPES}}, |
| 379 | {"debug_str_offsets.dwo", {StrOffsetSection, DW_SECT_STR_OFFSETS}}, |
| 380 | {"debug_str.dwo", {StrSection, static_cast<DWARFSectionKind>(0)}}, |
| 381 | {"debug_loc.dwo", {MCOFI.getDwarfLocDWOSection(), DW_SECT_LOC}}, |
David Blaikie | b702025 | 2015-12-04 21:16:42 +0000 | [diff] [blame] | 382 | {"debug_line.dwo", {MCOFI.getDwarfLineDWOSection(), DW_SECT_LINE}}, |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 383 | {"debug_abbrev.dwo", {MCOFI.getDwarfAbbrevDWOSection(), DW_SECT_ABBREV}}, |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 384 | {"debug_cu_index", {CUIndexSection, static_cast<DWARFSectionKind>(0)}}, |
| 385 | {"debug_tu_index", {TUIndexSection, static_cast<DWARFSectionKind>(0)}}}; |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 386 | |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 387 | MapVector<uint64_t, UnitIndexEntry> IndexEntries; |
| 388 | MapVector<uint64_t, UnitIndexEntry> TypeIndexEntries; |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 389 | |
| 390 | StringMap<uint32_t> Strings; |
| 391 | uint32_t StringOffset = 0; |
| 392 | |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 393 | uint32_t ContributionOffsets[8] = {}; |
| 394 | |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 395 | for (const auto &Input : Inputs) { |
| 396 | auto ErrOrObj = object::ObjectFile::createObjectFile(Input); |
| 397 | if (!ErrOrObj) |
Kevin Enderby | 3fcdf6a | 2016-04-06 22:14:09 +0000 | [diff] [blame] | 398 | return errorToErrorCode(ErrOrObj.takeError()); |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 399 | |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 400 | UnitIndexEntry CurEntry = {}; |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 401 | |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 402 | StringRef CurStrSection; |
| 403 | StringRef CurStrOffsetSection; |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 404 | std::vector<StringRef> CurTypesSection; |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 405 | StringRef InfoSection; |
| 406 | StringRef AbbrevSection; |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 407 | StringRef CurCUIndexSection; |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 408 | StringRef CurTUIndexSection; |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 409 | |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 410 | SmallVector<SmallString<32>, 4> UncompressedSections; |
| 411 | |
David Blaikie | ddb2736 | 2016-03-01 21:24:04 +0000 | [diff] [blame] | 412 | for (const auto &Section : ErrOrObj->getBinary()->sections()) { |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 413 | if (Section.isBSS()) |
| 414 | continue; |
| 415 | if (Section.isVirtual()) |
| 416 | continue; |
| 417 | |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 418 | StringRef Name; |
| 419 | if (std::error_code Err = Section.getName(Name)) |
| 420 | return Err; |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 421 | |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 422 | Name = Name.substr(Name.find_first_not_of("._")); |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 423 | |
| 424 | StringRef Contents; |
| 425 | if (auto Err = Section.getContents(Contents)) |
| 426 | return Err; |
| 427 | |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 428 | if (Name.startswith("zdebug_")) { |
| 429 | uint64_t OriginalSize; |
| 430 | if (!zlib::isAvailable() || |
| 431 | !consumeCompressedDebugSectionHeader(Contents, OriginalSize)) |
David Blaikie | 9a5d364 | 2016-02-19 02:03:45 +0000 | [diff] [blame] | 432 | return make_error_code(std::errc::invalid_argument); |
David Blaikie | 74f5b28 | 2016-02-19 01:51:44 +0000 | [diff] [blame] | 433 | UncompressedSections.resize(UncompressedSections.size() + 1); |
| 434 | if (zlib::uncompress(Contents, UncompressedSections.back(), OriginalSize) != |
| 435 | zlib::StatusOK) { |
| 436 | UncompressedSections.pop_back(); |
| 437 | continue; |
| 438 | } |
| 439 | Name = Name.substr(1); |
| 440 | Contents = UncompressedSections.back(); |
| 441 | } |
| 442 | |
| 443 | auto SectionPair = KnownSections.find(Name); |
| 444 | if (SectionPair == KnownSections.end()) |
| 445 | continue; |
| 446 | |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 447 | if (DWARFSectionKind Kind = SectionPair->second.second) { |
| 448 | auto Index = Kind - DW_SECT_INFO; |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 449 | if (Kind != DW_SECT_TYPES) { |
| 450 | CurEntry.Contributions[Index].Offset = ContributionOffsets[Index]; |
| 451 | ContributionOffsets[Index] += |
| 452 | (CurEntry.Contributions[Index].Length = Contents.size()); |
| 453 | } |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 454 | |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 455 | switch (Kind) { |
| 456 | case DW_SECT_INFO: |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 457 | InfoSection = Contents; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 458 | break; |
| 459 | case DW_SECT_ABBREV: |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 460 | AbbrevSection = Contents; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 461 | break; |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 462 | default: |
| 463 | break; |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 464 | } |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 465 | } |
| 466 | |
| 467 | MCSection *OutSection = SectionPair->second.first; |
| 468 | if (OutSection == StrOffsetSection) |
| 469 | CurStrOffsetSection = Contents; |
| 470 | else if (OutSection == StrSection) |
| 471 | CurStrSection = Contents; |
David Blaikie | c3826da | 2015-12-09 21:02:33 +0000 | [diff] [blame] | 472 | else if (OutSection == TypesSection) |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 473 | CurTypesSection.push_back(Contents); |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 474 | else if (OutSection == CUIndexSection) |
| 475 | CurCUIndexSection = Contents; |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 476 | else if (OutSection == TUIndexSection) |
| 477 | CurTUIndexSection = Contents; |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 478 | else { |
| 479 | Out.SwitchSection(OutSection); |
| 480 | Out.EmitBytes(Contents); |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 481 | } |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 482 | } |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 483 | |
David Blaikie | 5d6d4dc | 2016-02-26 07:04:58 +0000 | [diff] [blame] | 484 | if (InfoSection.empty()) |
| 485 | continue; |
| 486 | |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 487 | if (!CurCUIndexSection.empty()) { |
| 488 | DWARFUnitIndex CUIndex(DW_SECT_INFO); |
David Blaikie | ddb2736 | 2016-03-01 21:24:04 +0000 | [diff] [blame] | 489 | DataExtractor CUIndexData(CurCUIndexSection, |
| 490 | ErrOrObj->getBinary()->isLittleEndian(), 0); |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 491 | if (!CUIndex.parse(CUIndexData)) |
| 492 | return make_error_code(std::errc::invalid_argument); |
| 493 | |
| 494 | for (const DWARFUnitIndex::Entry &E : CUIndex.getRows()) { |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 495 | auto *I = E.getOffsets(); |
| 496 | if (!I) |
| 497 | continue; |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 498 | auto P = |
| 499 | IndexEntries.insert(std::make_pair(E.getSignature(), CurEntry)); |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 500 | CompileUnitIdentifiers ID = getCUIdentifiers( |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 501 | getSubsection(AbbrevSection, E, DW_SECT_ABBREV), |
| 502 | getSubsection(InfoSection, E, DW_SECT_INFO), |
| 503 | getSubsection(CurStrOffsetSection, E, DW_SECT_STR_OFFSETS), |
| 504 | CurStrSection); |
| 505 | if (!P.second) { |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 506 | printDuplicateError(*P.first, ID, Input); |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 507 | return make_error_code(std::errc::invalid_argument); |
| 508 | } |
David Blaikie | 852c02b | 2016-02-19 21:09:26 +0000 | [diff] [blame] | 509 | auto &NewEntry = P.first->second; |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 510 | NewEntry.Name = ID.Name; |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 511 | NewEntry.DWOName = ID.DWOName; |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 512 | NewEntry.DWPName = Input; |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 513 | for (auto Kind : CUIndex.getColumnKinds()) { |
| 514 | auto &C = NewEntry.Contributions[Kind - DW_SECT_INFO]; |
| 515 | C.Offset += I->Offset; |
| 516 | C.Length = I->Length; |
| 517 | ++I; |
| 518 | } |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 519 | } |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 520 | |
| 521 | if (!CurTypesSection.empty()) { |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 522 | assert(CurTypesSection.size() == 1); |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 523 | if (CurTUIndexSection.empty()) |
| 524 | return make_error_code(std::errc::invalid_argument); |
| 525 | DWARFUnitIndex TUIndex(DW_SECT_TYPES); |
David Blaikie | ddb2736 | 2016-03-01 21:24:04 +0000 | [diff] [blame] | 526 | DataExtractor TUIndexData(CurTUIndexSection, |
| 527 | ErrOrObj->getBinary()->isLittleEndian(), 0); |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 528 | if (!TUIndex.parse(TUIndexData)) |
| 529 | return make_error_code(std::errc::invalid_argument); |
| 530 | addAllTypesFromDWP(Out, TypeIndexEntries, TUIndex, TypesSection, |
David Blaikie | 62be5ae | 2016-04-05 20:26:50 +0000 | [diff] [blame] | 531 | CurTypesSection.front(), CurEntry, |
David Blaikie | 8bce5a0 | 2016-02-17 07:00:24 +0000 | [diff] [blame] | 532 | ContributionOffsets[DW_SECT_TYPES - DW_SECT_INFO]); |
| 533 | } |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 534 | } else { |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 535 | CompileUnitIdentifiers ID = getCUIdentifiers( |
| 536 | AbbrevSection, InfoSection, CurStrOffsetSection, CurStrSection); |
| 537 | auto P = IndexEntries.insert(std::make_pair(ID.Signature, CurEntry)); |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 538 | if (!P.second) { |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 539 | printDuplicateError(*P.first, ID, ""); |
David Blaikie | f1958da | 2016-02-26 07:30:15 +0000 | [diff] [blame] | 540 | return make_error_code(std::errc::invalid_argument); |
| 541 | } |
David Blaikie | ce7c6cf | 2016-03-24 22:17:08 +0000 | [diff] [blame] | 542 | P.first->second.Name = ID.Name; |
David Blaikie | 4dd03f0 | 2016-03-26 20:32:14 +0000 | [diff] [blame] | 543 | P.first->second.DWOName = ID.DWOName; |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 544 | addAllTypes(Out, TypeIndexEntries, TypesSection, CurTypesSection, |
| 545 | CurEntry, ContributionOffsets[DW_SECT_TYPES - DW_SECT_INFO]); |
David Blaikie | 2391937 | 2016-02-06 01:15:26 +0000 | [diff] [blame] | 546 | } |
David Blaikie | ad07b5d | 2015-12-04 17:20:04 +0000 | [diff] [blame] | 547 | |
David Blaikie | bb94e44 | 2015-12-01 19:17:58 +0000 | [diff] [blame] | 548 | if (auto Err = writeStringsAndOffsets(Out, Strings, StringOffset, |
| 549 | StrSection, StrOffsetSection, |
| 550 | CurStrSection, CurStrOffsetSection)) |
David Blaikie | 98ad82a | 2015-12-01 18:07:07 +0000 | [diff] [blame] | 551 | return Err; |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 552 | } |
David Blaikie | b073cb9 | 2015-12-02 06:21:34 +0000 | [diff] [blame] | 553 | |
David Blaikie | 5d6d4dc | 2016-02-26 07:04:58 +0000 | [diff] [blame] | 554 | // Lie about there being no info contributions so the TU index only includes |
| 555 | // the type unit contribution |
| 556 | ContributionOffsets[0] = 0; |
| 557 | writeIndex(Out, MCOFI.getDwarfTUIndexSection(), ContributionOffsets, |
| 558 | TypeIndexEntries); |
David Blaikie | b3757c0 | 2015-12-02 22:01:56 +0000 | [diff] [blame] | 559 | |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 560 | // Lie about the type contribution |
| 561 | ContributionOffsets[DW_SECT_TYPES - DW_SECT_INFO] = 0; |
| 562 | // Unlie about the info contribution |
| 563 | ContributionOffsets[0] = 1; |
David Blaikie | 7c4ffe0 | 2015-12-04 21:30:23 +0000 | [diff] [blame] | 564 | |
David Blaikie | 24c8ac9 | 2015-12-05 03:05:45 +0000 | [diff] [blame] | 565 | writeIndex(Out, MCOFI.getDwarfCUIndexSection(), ContributionOffsets, |
| 566 | IndexEntries); |
David Blaikie | ddb2736 | 2016-03-01 21:24:04 +0000 | [diff] [blame] | 567 | |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 568 | return std::error_code(); |
| 569 | } |
| 570 | |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 571 | int main(int argc, char **argv) { |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 572 | |
| 573 | ParseCommandLineOptions(argc, argv, "merge split dwarf (.dwo) files"); |
| 574 | |
| 575 | llvm::InitializeAllTargetInfos(); |
| 576 | llvm::InitializeAllTargetMCs(); |
| 577 | llvm::InitializeAllTargets(); |
| 578 | llvm::InitializeAllAsmPrinters(); |
| 579 | |
| 580 | std::string ErrorStr; |
| 581 | StringRef Context = "dwarf streamer init"; |
| 582 | |
| 583 | Triple TheTriple("x86_64-linux-gnu"); |
| 584 | |
| 585 | // Get the target. |
| 586 | const Target *TheTarget = |
| 587 | TargetRegistry::lookupTarget("", TheTriple, ErrorStr); |
| 588 | if (!TheTarget) |
| 589 | return error(ErrorStr, Context); |
| 590 | std::string TripleName = TheTriple.getTriple(); |
| 591 | |
| 592 | // Create all the MC Objects. |
| 593 | std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TripleName)); |
| 594 | if (!MRI) |
| 595 | return error(Twine("no register info for target ") + TripleName, Context); |
| 596 | |
| 597 | std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, TripleName)); |
| 598 | if (!MAI) |
| 599 | return error("no asm info for target " + TripleName, Context); |
| 600 | |
| 601 | MCObjectFileInfo MOFI; |
| 602 | MCContext MC(MAI.get(), MRI.get(), &MOFI); |
David Blaikie | 2ed678c | 2015-12-05 03:06:30 +0000 | [diff] [blame] | 603 | MOFI.InitMCObjectFileInfo(TheTriple, Reloc::Default, CodeModel::Default, MC); |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 604 | |
| 605 | auto MAB = TheTarget->createMCAsmBackend(*MRI, TripleName, ""); |
| 606 | if (!MAB) |
| 607 | return error("no asm backend for target " + TripleName, Context); |
| 608 | |
| 609 | std::unique_ptr<MCInstrInfo> MII(TheTarget->createMCInstrInfo()); |
| 610 | if (!MII) |
| 611 | return error("no instr info info for target " + TripleName, Context); |
| 612 | |
| 613 | std::unique_ptr<MCSubtargetInfo> MSTI( |
| 614 | TheTarget->createMCSubtargetInfo(TripleName, "", "")); |
| 615 | if (!MSTI) |
| 616 | return error("no subtarget info for target " + TripleName, Context); |
| 617 | |
| 618 | MCCodeEmitter *MCE = TheTarget->createMCCodeEmitter(*MII, *MRI, MC); |
| 619 | if (!MCE) |
| 620 | return error("no code emitter for target " + TripleName, Context); |
| 621 | |
| 622 | // Create the output file. |
| 623 | std::error_code EC; |
| 624 | raw_fd_ostream OutFile(OutputFilename, EC, sys::fs::F_None); |
| 625 | if (EC) |
| 626 | return error(Twine(OutputFilename) + ": " + EC.message(), Context); |
| 627 | |
David Majnemer | 03e2cc3 | 2015-12-21 22:09:27 +0000 | [diff] [blame] | 628 | MCTargetOptions MCOptions = InitMCTargetOptionsFromFlags(); |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 629 | std::unique_ptr<MCStreamer> MS(TheTarget->createMCObjectStreamer( |
David Majnemer | 03e2cc3 | 2015-12-21 22:09:27 +0000 | [diff] [blame] | 630 | TheTriple, MC, *MAB, OutFile, MCE, *MSTI, MCOptions.MCRelaxAll, |
| 631 | MCOptions.MCIncrementalLinkerCompatible, |
David Blaikie | 242b948 | 2015-12-01 00:48:39 +0000 | [diff] [blame] | 632 | /*DWARFMustBeAtTheEnd*/ false)); |
| 633 | if (!MS) |
| 634 | return error("no object streamer for target " + TripleName, Context); |
| 635 | |
| 636 | if (auto Err = write(*MS, InputFiles)) |
| 637 | return error(Err.message(), "Writing DWP file"); |
David Blaikie | ddb2736 | 2016-03-01 21:24:04 +0000 | [diff] [blame] | 638 | |
| 639 | MS->Finish(); |
David Blaikie | df05525 | 2015-12-01 00:48:34 +0000 | [diff] [blame] | 640 | } |