[RISCV] Use RISCVSubtarget::is64Bit() instead of hasFeature(RISCV::Feature64Bit)...
[llvm-project.git] / llvm / lib / Bitcode / Reader / MetadataLoader.cpp
blob1caa8480efc5182d478847b897252354560614b5
1 //===- MetadataLoader.cpp - Internal BitcodeReader implementation ---------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
9 #include "MetadataLoader.h"
10 #include "ValueList.h"
12 #include "llvm/ADT/APInt.h"
13 #include "llvm/ADT/ArrayRef.h"
14 #include "llvm/ADT/BitmaskEnum.h"
15 #include "llvm/ADT/DenseMap.h"
16 #include "llvm/ADT/DenseSet.h"
17 #include "llvm/ADT/STLFunctionalExtras.h"
18 #include "llvm/ADT/SetVector.h"
19 #include "llvm/ADT/SmallString.h"
20 #include "llvm/ADT/SmallVector.h"
21 #include "llvm/ADT/Statistic.h"
22 #include "llvm/ADT/StringRef.h"
23 #include "llvm/ADT/Twine.h"
24 #include "llvm/ADT/ilist_iterator.h"
25 #include "llvm/BinaryFormat/Dwarf.h"
26 #include "llvm/Bitcode/BitcodeReader.h"
27 #include "llvm/Bitcode/LLVMBitCodes.h"
28 #include "llvm/Bitstream/BitstreamReader.h"
29 #include "llvm/IR/AutoUpgrade.h"
30 #include "llvm/IR/BasicBlock.h"
31 #include "llvm/IR/Constants.h"
32 #include "llvm/IR/DebugInfoMetadata.h"
33 #include "llvm/IR/Function.h"
34 #include "llvm/IR/GlobalObject.h"
35 #include "llvm/IR/GlobalVariable.h"
36 #include "llvm/IR/Instruction.h"
37 #include "llvm/IR/IntrinsicInst.h"
38 #include "llvm/IR/LLVMContext.h"
39 #include "llvm/IR/Metadata.h"
40 #include "llvm/IR/Module.h"
41 #include "llvm/IR/TrackingMDRef.h"
42 #include "llvm/IR/Type.h"
43 #include "llvm/Support/Casting.h"
44 #include "llvm/Support/CommandLine.h"
45 #include "llvm/Support/Compiler.h"
46 #include "llvm/Support/ErrorHandling.h"
47 #include "llvm/Support/type_traits.h"
49 #include <algorithm>
50 #include <cassert>
51 #include <cstddef>
52 #include <cstdint>
53 #include <deque>
54 #include <iterator>
55 #include <limits>
56 #include <map>
57 #include <optional>
58 #include <string>
59 #include <tuple>
60 #include <type_traits>
61 #include <utility>
62 #include <vector>
63 namespace llvm {
64 class Argument;
67 using namespace llvm;
69 #define DEBUG_TYPE "bitcode-reader"
71 STATISTIC(NumMDStringLoaded, "Number of MDStrings loaded");
72 STATISTIC(NumMDNodeTemporary, "Number of MDNode::Temporary created");
73 STATISTIC(NumMDRecordLoaded, "Number of Metadata records loaded");
75 /// Flag whether we need to import full type definitions for ThinLTO.
76 /// Currently needed for Darwin and LLDB.
77 static cl::opt<bool> ImportFullTypeDefinitions(
78 "import-full-type-definitions", cl::init(false), cl::Hidden,
79 cl::desc("Import full type definitions for ThinLTO."));
81 static cl::opt<bool> DisableLazyLoading(
82 "disable-ondemand-mds-loading", cl::init(false), cl::Hidden,
83 cl::desc("Force disable the lazy-loading on-demand of metadata when "
84 "loading bitcode for importing."));
86 namespace {
88 static int64_t unrotateSign(uint64_t U) { return (U & 1) ? ~(U >> 1) : U >> 1; }
90 class BitcodeReaderMetadataList {
91 /// Array of metadata references.
92 ///
93 /// Don't use std::vector here. Some versions of libc++ copy (instead of
94 /// move) on resize, and TrackingMDRef is very expensive to copy.
95 SmallVector<TrackingMDRef, 1> MetadataPtrs;
97 /// The set of indices in MetadataPtrs above of forward references that were
98 /// generated.
99 SmallDenseSet<unsigned, 1> ForwardReference;
101 /// The set of indices in MetadataPtrs above of Metadata that need to be
102 /// resolved.
103 SmallDenseSet<unsigned, 1> UnresolvedNodes;
105 /// Structures for resolving old type refs.
106 struct {
107 SmallDenseMap<MDString *, TempMDTuple, 1> Unknown;
108 SmallDenseMap<MDString *, DICompositeType *, 1> Final;
109 SmallDenseMap<MDString *, DICompositeType *, 1> FwdDecls;
110 SmallVector<std::pair<TrackingMDRef, TempMDTuple>, 1> Arrays;
111 } OldTypeRefs;
113 LLVMContext &Context;
115 /// Maximum number of valid references. Forward references exceeding the
116 /// maximum must be invalid.
117 unsigned RefsUpperBound;
119 public:
120 BitcodeReaderMetadataList(LLVMContext &C, size_t RefsUpperBound)
121 : Context(C),
122 RefsUpperBound(std::min((size_t)std::numeric_limits<unsigned>::max(),
123 RefsUpperBound)) {}
125 // vector compatibility methods
126 unsigned size() const { return MetadataPtrs.size(); }
127 void resize(unsigned N) { MetadataPtrs.resize(N); }
128 void push_back(Metadata *MD) { MetadataPtrs.emplace_back(MD); }
129 void clear() { MetadataPtrs.clear(); }
130 Metadata *back() const { return MetadataPtrs.back(); }
131 void pop_back() { MetadataPtrs.pop_back(); }
132 bool empty() const { return MetadataPtrs.empty(); }
134 Metadata *operator[](unsigned i) const {
135 assert(i < MetadataPtrs.size());
136 return MetadataPtrs[i];
139 Metadata *lookup(unsigned I) const {
140 if (I < MetadataPtrs.size())
141 return MetadataPtrs[I];
142 return nullptr;
145 void shrinkTo(unsigned N) {
146 assert(N <= size() && "Invalid shrinkTo request!");
147 assert(ForwardReference.empty() && "Unexpected forward refs");
148 assert(UnresolvedNodes.empty() && "Unexpected unresolved node");
149 MetadataPtrs.resize(N);
152 /// Return the given metadata, creating a replaceable forward reference if
153 /// necessary.
154 Metadata *getMetadataFwdRef(unsigned Idx);
156 /// Return the given metadata only if it is fully resolved.
158 /// Gives the same result as \a lookup(), unless \a MDNode::isResolved()
159 /// would give \c false.
160 Metadata *getMetadataIfResolved(unsigned Idx);
162 MDNode *getMDNodeFwdRefOrNull(unsigned Idx);
163 void assignValue(Metadata *MD, unsigned Idx);
164 void tryToResolveCycles();
165 bool hasFwdRefs() const { return !ForwardReference.empty(); }
166 int getNextFwdRef() {
167 assert(hasFwdRefs());
168 return *ForwardReference.begin();
171 /// Upgrade a type that had an MDString reference.
172 void addTypeRef(MDString &UUID, DICompositeType &CT);
174 /// Upgrade a type that had an MDString reference.
175 Metadata *upgradeTypeRef(Metadata *MaybeUUID);
177 /// Upgrade a type ref array that may have MDString references.
178 Metadata *upgradeTypeRefArray(Metadata *MaybeTuple);
180 private:
181 Metadata *resolveTypeRefArray(Metadata *MaybeTuple);
184 void BitcodeReaderMetadataList::assignValue(Metadata *MD, unsigned Idx) {
185 if (auto *MDN = dyn_cast<MDNode>(MD))
186 if (!MDN->isResolved())
187 UnresolvedNodes.insert(Idx);
189 if (Idx == size()) {
190 push_back(MD);
191 return;
194 if (Idx >= size())
195 resize(Idx + 1);
197 TrackingMDRef &OldMD = MetadataPtrs[Idx];
198 if (!OldMD) {
199 OldMD.reset(MD);
200 return;
203 // If there was a forward reference to this value, replace it.
204 TempMDTuple PrevMD(cast<MDTuple>(OldMD.get()));
205 PrevMD->replaceAllUsesWith(MD);
206 ForwardReference.erase(Idx);
209 Metadata *BitcodeReaderMetadataList::getMetadataFwdRef(unsigned Idx) {
210 // Bail out for a clearly invalid value.
211 if (Idx >= RefsUpperBound)
212 return nullptr;
214 if (Idx >= size())
215 resize(Idx + 1);
217 if (Metadata *MD = MetadataPtrs[Idx])
218 return MD;
220 // Track forward refs to be resolved later.
221 ForwardReference.insert(Idx);
223 // Create and return a placeholder, which will later be RAUW'd.
224 ++NumMDNodeTemporary;
225 Metadata *MD = MDNode::getTemporary(Context, {}).release();
226 MetadataPtrs[Idx].reset(MD);
227 return MD;
230 Metadata *BitcodeReaderMetadataList::getMetadataIfResolved(unsigned Idx) {
231 Metadata *MD = lookup(Idx);
232 if (auto *N = dyn_cast_or_null<MDNode>(MD))
233 if (!N->isResolved())
234 return nullptr;
235 return MD;
238 MDNode *BitcodeReaderMetadataList::getMDNodeFwdRefOrNull(unsigned Idx) {
239 return dyn_cast_or_null<MDNode>(getMetadataFwdRef(Idx));
242 void BitcodeReaderMetadataList::tryToResolveCycles() {
243 if (!ForwardReference.empty())
244 // Still forward references... can't resolve cycles.
245 return;
247 // Give up on finding a full definition for any forward decls that remain.
248 for (const auto &Ref : OldTypeRefs.FwdDecls)
249 OldTypeRefs.Final.insert(Ref);
250 OldTypeRefs.FwdDecls.clear();
252 // Upgrade from old type ref arrays. In strange cases, this could add to
253 // OldTypeRefs.Unknown.
254 for (const auto &Array : OldTypeRefs.Arrays)
255 Array.second->replaceAllUsesWith(resolveTypeRefArray(Array.first.get()));
256 OldTypeRefs.Arrays.clear();
258 // Replace old string-based type refs with the resolved node, if possible.
259 // If we haven't seen the node, leave it to the verifier to complain about
260 // the invalid string reference.
261 for (const auto &Ref : OldTypeRefs.Unknown) {
262 if (DICompositeType *CT = OldTypeRefs.Final.lookup(Ref.first))
263 Ref.second->replaceAllUsesWith(CT);
264 else
265 Ref.second->replaceAllUsesWith(Ref.first);
267 OldTypeRefs.Unknown.clear();
269 if (UnresolvedNodes.empty())
270 // Nothing to do.
271 return;
273 // Resolve any cycles.
274 for (unsigned I : UnresolvedNodes) {
275 auto &MD = MetadataPtrs[I];
276 auto *N = dyn_cast_or_null<MDNode>(MD);
277 if (!N)
278 continue;
280 assert(!N->isTemporary() && "Unexpected forward reference");
281 N->resolveCycles();
284 // Make sure we return early again until there's another unresolved ref.
285 UnresolvedNodes.clear();
288 void BitcodeReaderMetadataList::addTypeRef(MDString &UUID,
289 DICompositeType &CT) {
290 assert(CT.getRawIdentifier() == &UUID && "Mismatched UUID");
291 if (CT.isForwardDecl())
292 OldTypeRefs.FwdDecls.insert(std::make_pair(&UUID, &CT));
293 else
294 OldTypeRefs.Final.insert(std::make_pair(&UUID, &CT));
297 Metadata *BitcodeReaderMetadataList::upgradeTypeRef(Metadata *MaybeUUID) {
298 auto *UUID = dyn_cast_or_null<MDString>(MaybeUUID);
299 if (LLVM_LIKELY(!UUID))
300 return MaybeUUID;
302 if (auto *CT = OldTypeRefs.Final.lookup(UUID))
303 return CT;
305 auto &Ref = OldTypeRefs.Unknown[UUID];
306 if (!Ref)
307 Ref = MDNode::getTemporary(Context, {});
308 return Ref.get();
311 Metadata *BitcodeReaderMetadataList::upgradeTypeRefArray(Metadata *MaybeTuple) {
312 auto *Tuple = dyn_cast_or_null<MDTuple>(MaybeTuple);
313 if (!Tuple || Tuple->isDistinct())
314 return MaybeTuple;
316 // Look through the array immediately if possible.
317 if (!Tuple->isTemporary())
318 return resolveTypeRefArray(Tuple);
320 // Create and return a placeholder to use for now. Eventually
321 // resolveTypeRefArrays() will be resolve this forward reference.
322 OldTypeRefs.Arrays.emplace_back(
323 std::piecewise_construct, std::forward_as_tuple(Tuple),
324 std::forward_as_tuple(MDTuple::getTemporary(Context, {})));
325 return OldTypeRefs.Arrays.back().second.get();
328 Metadata *BitcodeReaderMetadataList::resolveTypeRefArray(Metadata *MaybeTuple) {
329 auto *Tuple = dyn_cast_or_null<MDTuple>(MaybeTuple);
330 if (!Tuple || Tuple->isDistinct())
331 return MaybeTuple;
333 // Look through the DITypeRefArray, upgrading each DIType *.
334 SmallVector<Metadata *, 32> Ops;
335 Ops.reserve(Tuple->getNumOperands());
336 for (Metadata *MD : Tuple->operands())
337 Ops.push_back(upgradeTypeRef(MD));
339 return MDTuple::get(Context, Ops);
342 namespace {
344 class PlaceholderQueue {
345 // Placeholders would thrash around when moved, so store in a std::deque
346 // instead of some sort of vector.
347 std::deque<DistinctMDOperandPlaceholder> PHs;
349 public:
350 ~PlaceholderQueue() {
351 assert(empty() &&
352 "PlaceholderQueue hasn't been flushed before being destroyed");
354 bool empty() const { return PHs.empty(); }
355 DistinctMDOperandPlaceholder &getPlaceholderOp(unsigned ID);
356 void flush(BitcodeReaderMetadataList &MetadataList);
358 /// Return the list of temporaries nodes in the queue, these need to be
359 /// loaded before we can flush the queue.
360 void getTemporaries(BitcodeReaderMetadataList &MetadataList,
361 DenseSet<unsigned> &Temporaries) {
362 for (auto &PH : PHs) {
363 auto ID = PH.getID();
364 auto *MD = MetadataList.lookup(ID);
365 if (!MD) {
366 Temporaries.insert(ID);
367 continue;
369 auto *N = dyn_cast_or_null<MDNode>(MD);
370 if (N && N->isTemporary())
371 Temporaries.insert(ID);
376 } // end anonymous namespace
378 DistinctMDOperandPlaceholder &PlaceholderQueue::getPlaceholderOp(unsigned ID) {
379 PHs.emplace_back(ID);
380 return PHs.back();
383 void PlaceholderQueue::flush(BitcodeReaderMetadataList &MetadataList) {
384 while (!PHs.empty()) {
385 auto *MD = MetadataList.lookup(PHs.front().getID());
386 assert(MD && "Flushing placeholder on unassigned MD");
387 #ifndef NDEBUG
388 if (auto *MDN = dyn_cast<MDNode>(MD))
389 assert(MDN->isResolved() &&
390 "Flushing Placeholder while cycles aren't resolved");
391 #endif
392 PHs.front().replaceUseWith(MD);
393 PHs.pop_front();
397 } // anonymous namespace
399 static Error error(const Twine &Message) {
400 return make_error<StringError>(
401 Message, make_error_code(BitcodeError::CorruptedBitcode));
404 class MetadataLoader::MetadataLoaderImpl {
405 BitcodeReaderMetadataList MetadataList;
406 BitcodeReaderValueList &ValueList;
407 BitstreamCursor &Stream;
408 LLVMContext &Context;
409 Module &TheModule;
410 MetadataLoaderCallbacks Callbacks;
412 /// Cursor associated with the lazy-loading of Metadata. This is the easy way
413 /// to keep around the right "context" (Abbrev list) to be able to jump in
414 /// the middle of the metadata block and load any record.
415 BitstreamCursor IndexCursor;
417 /// Index that keeps track of MDString values.
418 std::vector<StringRef> MDStringRef;
420 /// On-demand loading of a single MDString. Requires the index above to be
421 /// populated.
422 MDString *lazyLoadOneMDString(unsigned Idx);
424 /// Index that keeps track of where to find a metadata record in the stream.
425 std::vector<uint64_t> GlobalMetadataBitPosIndex;
427 /// Cursor position of the start of the global decl attachments, to enable
428 /// loading using the index built for lazy loading, instead of forward
429 /// references.
430 uint64_t GlobalDeclAttachmentPos = 0;
432 #ifndef NDEBUG
433 /// Baisic correctness check that we end up parsing all of the global decl
434 /// attachments.
435 unsigned NumGlobalDeclAttachSkipped = 0;
436 unsigned NumGlobalDeclAttachParsed = 0;
437 #endif
439 /// Load the global decl attachments, using the index built for lazy loading.
440 Expected<bool> loadGlobalDeclAttachments();
442 /// Populate the index above to enable lazily loading of metadata, and load
443 /// the named metadata as well as the transitively referenced global
444 /// Metadata.
445 Expected<bool> lazyLoadModuleMetadataBlock();
447 /// On-demand loading of a single metadata. Requires the index above to be
448 /// populated.
449 void lazyLoadOneMetadata(unsigned Idx, PlaceholderQueue &Placeholders);
451 // Keep mapping of seens pair of old-style CU <-> SP, and update pointers to
452 // point from SP to CU after a block is completly parsed.
453 std::vector<std::pair<DICompileUnit *, Metadata *>> CUSubprograms;
455 /// Functions that need to be matched with subprograms when upgrading old
456 /// metadata.
457 SmallDenseMap<Function *, DISubprogram *, 16> FunctionsWithSPs;
459 // Map the bitcode's custom MDKind ID to the Module's MDKind ID.
460 DenseMap<unsigned, unsigned> MDKindMap;
462 bool StripTBAA = false;
463 bool HasSeenOldLoopTags = false;
464 bool NeedUpgradeToDIGlobalVariableExpression = false;
465 bool NeedDeclareExpressionUpgrade = false;
467 /// Map DILocalScope to the enclosing DISubprogram, if any.
468 DenseMap<DILocalScope *, DISubprogram *> ParentSubprogram;
470 /// True if metadata is being parsed for a module being ThinLTO imported.
471 bool IsImporting = false;
473 Error parseOneMetadata(SmallVectorImpl<uint64_t> &Record, unsigned Code,
474 PlaceholderQueue &Placeholders, StringRef Blob,
475 unsigned &NextMetadataNo);
476 Error parseMetadataStrings(ArrayRef<uint64_t> Record, StringRef Blob,
477 function_ref<void(StringRef)> CallBack);
478 Error parseGlobalObjectAttachment(GlobalObject &GO,
479 ArrayRef<uint64_t> Record);
480 Error parseMetadataKindRecord(SmallVectorImpl<uint64_t> &Record);
482 void resolveForwardRefsAndPlaceholders(PlaceholderQueue &Placeholders);
484 /// Upgrade old-style CU <-> SP pointers to point from SP to CU.
485 void upgradeCUSubprograms() {
486 for (auto CU_SP : CUSubprograms)
487 if (auto *SPs = dyn_cast_or_null<MDTuple>(CU_SP.second))
488 for (auto &Op : SPs->operands())
489 if (auto *SP = dyn_cast_or_null<DISubprogram>(Op))
490 SP->replaceUnit(CU_SP.first);
491 CUSubprograms.clear();
494 /// Upgrade old-style bare DIGlobalVariables to DIGlobalVariableExpressions.
495 void upgradeCUVariables() {
496 if (!NeedUpgradeToDIGlobalVariableExpression)
497 return;
499 // Upgrade list of variables attached to the CUs.
500 if (NamedMDNode *CUNodes = TheModule.getNamedMetadata("llvm.dbg.cu"))
501 for (unsigned I = 0, E = CUNodes->getNumOperands(); I != E; ++I) {
502 auto *CU = cast<DICompileUnit>(CUNodes->getOperand(I));
503 if (auto *GVs = dyn_cast_or_null<MDTuple>(CU->getRawGlobalVariables()))
504 for (unsigned I = 0; I < GVs->getNumOperands(); I++)
505 if (auto *GV =
506 dyn_cast_or_null<DIGlobalVariable>(GVs->getOperand(I))) {
507 auto *DGVE = DIGlobalVariableExpression::getDistinct(
508 Context, GV, DIExpression::get(Context, {}));
509 GVs->replaceOperandWith(I, DGVE);
513 // Upgrade variables attached to globals.
514 for (auto &GV : TheModule.globals()) {
515 SmallVector<MDNode *, 1> MDs;
516 GV.getMetadata(LLVMContext::MD_dbg, MDs);
517 GV.eraseMetadata(LLVMContext::MD_dbg);
518 for (auto *MD : MDs)
519 if (auto *DGV = dyn_cast<DIGlobalVariable>(MD)) {
520 auto *DGVE = DIGlobalVariableExpression::getDistinct(
521 Context, DGV, DIExpression::get(Context, {}));
522 GV.addMetadata(LLVMContext::MD_dbg, *DGVE);
523 } else
524 GV.addMetadata(LLVMContext::MD_dbg, *MD);
528 DISubprogram *findEnclosingSubprogram(DILocalScope *S) {
529 if (!S)
530 return nullptr;
531 if (auto *SP = ParentSubprogram[S]) {
532 return SP;
535 DILocalScope *InitialScope = S;
536 DenseSet<DILocalScope *> Visited;
537 while (S && !isa<DISubprogram>(S)) {
538 S = dyn_cast_or_null<DILocalScope>(S->getScope());
539 if (!Visited.insert(S).second)
540 break;
542 ParentSubprogram[InitialScope] = llvm::dyn_cast_or_null<DISubprogram>(S);
544 return ParentSubprogram[InitialScope];
547 /// Move local imports from DICompileUnit's 'imports' field to
548 /// DISubprogram's retainedNodes.
549 void upgradeCULocals() {
550 if (NamedMDNode *CUNodes = TheModule.getNamedMetadata("llvm.dbg.cu")) {
551 for (MDNode *N : CUNodes->operands()) {
552 auto *CU = dyn_cast<DICompileUnit>(N);
553 if (!CU)
554 continue;
556 if (CU->getRawImportedEntities()) {
557 // Collect a set of imported entities to be moved.
558 SetVector<Metadata *> EntitiesToRemove;
559 for (Metadata *Op : CU->getImportedEntities()->operands()) {
560 auto *IE = cast<DIImportedEntity>(Op);
561 if (dyn_cast_or_null<DILocalScope>(IE->getScope())) {
562 EntitiesToRemove.insert(IE);
566 if (!EntitiesToRemove.empty()) {
567 // Make a new list of CU's 'imports'.
568 SmallVector<Metadata *> NewImports;
569 for (Metadata *Op : CU->getImportedEntities()->operands()) {
570 if (!EntitiesToRemove.contains(cast<DIImportedEntity>(Op))) {
571 NewImports.push_back(Op);
575 // Find DISubprogram corresponding to each entity.
576 std::map<DISubprogram *, SmallVector<Metadata *>> SPToEntities;
577 for (auto *I : EntitiesToRemove) {
578 auto *Entity = cast<DIImportedEntity>(I);
579 if (auto *SP = findEnclosingSubprogram(
580 cast<DILocalScope>(Entity->getScope()))) {
581 SPToEntities[SP].push_back(Entity);
585 // Update DISubprograms' retainedNodes.
586 for (auto I = SPToEntities.begin(); I != SPToEntities.end(); ++I) {
587 auto *SP = I->first;
588 auto RetainedNodes = SP->getRetainedNodes();
589 SmallVector<Metadata *> MDs(RetainedNodes.begin(),
590 RetainedNodes.end());
591 MDs.append(I->second);
592 SP->replaceRetainedNodes(MDNode::get(Context, MDs));
595 // Remove entities with local scope from CU.
596 CU->replaceImportedEntities(MDTuple::get(Context, NewImports));
602 ParentSubprogram.clear();
605 /// Remove a leading DW_OP_deref from DIExpressions in a dbg.declare that
606 /// describes a function argument.
607 void upgradeDeclareExpressions(Function &F) {
608 if (!NeedDeclareExpressionUpgrade)
609 return;
611 auto UpdateDeclareIfNeeded = [&](auto *Declare) {
612 auto *DIExpr = Declare->getExpression();
613 if (!DIExpr || !DIExpr->startsWithDeref() ||
614 !isa_and_nonnull<Argument>(Declare->getAddress()))
615 return;
616 SmallVector<uint64_t, 8> Ops;
617 Ops.append(std::next(DIExpr->elements_begin()), DIExpr->elements_end());
618 Declare->setExpression(DIExpression::get(Context, Ops));
621 for (auto &BB : F)
622 for (auto &I : BB) {
623 for (DbgVariableRecord &DVR : filterDbgVars(I.getDbgRecordRange())) {
624 if (DVR.isDbgDeclare())
625 UpdateDeclareIfNeeded(&DVR);
627 if (auto *DDI = dyn_cast<DbgDeclareInst>(&I))
628 UpdateDeclareIfNeeded(DDI);
632 /// Upgrade the expression from previous versions.
633 Error upgradeDIExpression(uint64_t FromVersion,
634 MutableArrayRef<uint64_t> &Expr,
635 SmallVectorImpl<uint64_t> &Buffer) {
636 auto N = Expr.size();
637 switch (FromVersion) {
638 default:
639 return error("Invalid record");
640 case 0:
641 if (N >= 3 && Expr[N - 3] == dwarf::DW_OP_bit_piece)
642 Expr[N - 3] = dwarf::DW_OP_LLVM_fragment;
643 [[fallthrough]];
644 case 1:
645 // Move DW_OP_deref to the end.
646 if (N && Expr[0] == dwarf::DW_OP_deref) {
647 auto End = Expr.end();
648 if (Expr.size() >= 3 &&
649 *std::prev(End, 3) == dwarf::DW_OP_LLVM_fragment)
650 End = std::prev(End, 3);
651 std::move(std::next(Expr.begin()), End, Expr.begin());
652 *std::prev(End) = dwarf::DW_OP_deref;
654 NeedDeclareExpressionUpgrade = true;
655 [[fallthrough]];
656 case 2: {
657 // Change DW_OP_plus to DW_OP_plus_uconst.
658 // Change DW_OP_minus to DW_OP_uconst, DW_OP_minus
659 auto SubExpr = ArrayRef<uint64_t>(Expr);
660 while (!SubExpr.empty()) {
661 // Skip past other operators with their operands
662 // for this version of the IR, obtained from
663 // from historic DIExpression::ExprOperand::getSize().
664 size_t HistoricSize;
665 switch (SubExpr.front()) {
666 default:
667 HistoricSize = 1;
668 break;
669 case dwarf::DW_OP_constu:
670 case dwarf::DW_OP_minus:
671 case dwarf::DW_OP_plus:
672 HistoricSize = 2;
673 break;
674 case dwarf::DW_OP_LLVM_fragment:
675 HistoricSize = 3;
676 break;
679 // If the expression is malformed, make sure we don't
680 // copy more elements than we should.
681 HistoricSize = std::min(SubExpr.size(), HistoricSize);
682 ArrayRef<uint64_t> Args = SubExpr.slice(1, HistoricSize - 1);
684 switch (SubExpr.front()) {
685 case dwarf::DW_OP_plus:
686 Buffer.push_back(dwarf::DW_OP_plus_uconst);
687 Buffer.append(Args.begin(), Args.end());
688 break;
689 case dwarf::DW_OP_minus:
690 Buffer.push_back(dwarf::DW_OP_constu);
691 Buffer.append(Args.begin(), Args.end());
692 Buffer.push_back(dwarf::DW_OP_minus);
693 break;
694 default:
695 Buffer.push_back(*SubExpr.begin());
696 Buffer.append(Args.begin(), Args.end());
697 break;
700 // Continue with remaining elements.
701 SubExpr = SubExpr.slice(HistoricSize);
703 Expr = MutableArrayRef<uint64_t>(Buffer);
704 [[fallthrough]];
706 case 3:
707 // Up-to-date!
708 break;
711 return Error::success();
714 void upgradeDebugInfo(bool ModuleLevel) {
715 upgradeCUSubprograms();
716 upgradeCUVariables();
717 if (ModuleLevel)
718 upgradeCULocals();
721 void callMDTypeCallback(Metadata **Val, unsigned TypeID);
723 public:
724 MetadataLoaderImpl(BitstreamCursor &Stream, Module &TheModule,
725 BitcodeReaderValueList &ValueList,
726 MetadataLoaderCallbacks Callbacks, bool IsImporting)
727 : MetadataList(TheModule.getContext(), Stream.SizeInBytes()),
728 ValueList(ValueList), Stream(Stream), Context(TheModule.getContext()),
729 TheModule(TheModule), Callbacks(std::move(Callbacks)),
730 IsImporting(IsImporting) {}
732 Error parseMetadata(bool ModuleLevel);
734 bool hasFwdRefs() const { return MetadataList.hasFwdRefs(); }
736 Metadata *getMetadataFwdRefOrLoad(unsigned ID) {
737 if (ID < MDStringRef.size())
738 return lazyLoadOneMDString(ID);
739 if (auto *MD = MetadataList.lookup(ID))
740 return MD;
741 // If lazy-loading is enabled, we try recursively to load the operand
742 // instead of creating a temporary.
743 if (ID < (MDStringRef.size() + GlobalMetadataBitPosIndex.size())) {
744 PlaceholderQueue Placeholders;
745 lazyLoadOneMetadata(ID, Placeholders);
746 resolveForwardRefsAndPlaceholders(Placeholders);
747 return MetadataList.lookup(ID);
749 return MetadataList.getMetadataFwdRef(ID);
752 DISubprogram *lookupSubprogramForFunction(Function *F) {
753 return FunctionsWithSPs.lookup(F);
756 bool hasSeenOldLoopTags() const { return HasSeenOldLoopTags; }
758 Error parseMetadataAttachment(Function &F,
759 ArrayRef<Instruction *> InstructionList);
761 Error parseMetadataKinds();
763 void setStripTBAA(bool Value) { StripTBAA = Value; }
764 bool isStrippingTBAA() const { return StripTBAA; }
766 unsigned size() const { return MetadataList.size(); }
767 void shrinkTo(unsigned N) { MetadataList.shrinkTo(N); }
768 void upgradeDebugIntrinsics(Function &F) { upgradeDeclareExpressions(F); }
771 Expected<bool>
772 MetadataLoader::MetadataLoaderImpl::lazyLoadModuleMetadataBlock() {
773 IndexCursor = Stream;
774 SmallVector<uint64_t, 64> Record;
775 GlobalDeclAttachmentPos = 0;
776 // Get the abbrevs, and preload record positions to make them lazy-loadable.
777 while (true) {
778 uint64_t SavedPos = IndexCursor.GetCurrentBitNo();
779 BitstreamEntry Entry;
780 if (Error E =
781 IndexCursor
782 .advanceSkippingSubblocks(BitstreamCursor::AF_DontPopBlockAtEnd)
783 .moveInto(Entry))
784 return std::move(E);
786 switch (Entry.Kind) {
787 case BitstreamEntry::SubBlock: // Handled for us already.
788 case BitstreamEntry::Error:
789 return error("Malformed block");
790 case BitstreamEntry::EndBlock: {
791 return true;
793 case BitstreamEntry::Record: {
794 // The interesting case.
795 ++NumMDRecordLoaded;
796 uint64_t CurrentPos = IndexCursor.GetCurrentBitNo();
797 unsigned Code;
798 if (Error E = IndexCursor.skipRecord(Entry.ID).moveInto(Code))
799 return std::move(E);
800 switch (Code) {
801 case bitc::METADATA_STRINGS: {
802 // Rewind and parse the strings.
803 if (Error Err = IndexCursor.JumpToBit(CurrentPos))
804 return std::move(Err);
805 StringRef Blob;
806 Record.clear();
807 if (Expected<unsigned> MaybeRecord =
808 IndexCursor.readRecord(Entry.ID, Record, &Blob))
810 else
811 return MaybeRecord.takeError();
812 unsigned NumStrings = Record[0];
813 MDStringRef.reserve(NumStrings);
814 auto IndexNextMDString = [&](StringRef Str) {
815 MDStringRef.push_back(Str);
817 if (auto Err = parseMetadataStrings(Record, Blob, IndexNextMDString))
818 return std::move(Err);
819 break;
821 case bitc::METADATA_INDEX_OFFSET: {
822 // This is the offset to the index, when we see this we skip all the
823 // records and load only an index to these.
824 if (Error Err = IndexCursor.JumpToBit(CurrentPos))
825 return std::move(Err);
826 Record.clear();
827 if (Expected<unsigned> MaybeRecord =
828 IndexCursor.readRecord(Entry.ID, Record))
830 else
831 return MaybeRecord.takeError();
832 if (Record.size() != 2)
833 return error("Invalid record");
834 auto Offset = Record[0] + (Record[1] << 32);
835 auto BeginPos = IndexCursor.GetCurrentBitNo();
836 if (Error Err = IndexCursor.JumpToBit(BeginPos + Offset))
837 return std::move(Err);
838 Expected<BitstreamEntry> MaybeEntry =
839 IndexCursor.advanceSkippingSubblocks(
840 BitstreamCursor::AF_DontPopBlockAtEnd);
841 if (!MaybeEntry)
842 return MaybeEntry.takeError();
843 Entry = MaybeEntry.get();
844 assert(Entry.Kind == BitstreamEntry::Record &&
845 "Corrupted bitcode: Expected `Record` when trying to find the "
846 "Metadata index");
847 Record.clear();
848 if (Expected<unsigned> MaybeCode =
849 IndexCursor.readRecord(Entry.ID, Record))
850 assert(MaybeCode.get() == bitc::METADATA_INDEX &&
851 "Corrupted bitcode: Expected `METADATA_INDEX` when trying to "
852 "find the Metadata index");
853 else
854 return MaybeCode.takeError();
855 // Delta unpack
856 auto CurrentValue = BeginPos;
857 GlobalMetadataBitPosIndex.reserve(Record.size());
858 for (auto &Elt : Record) {
859 CurrentValue += Elt;
860 GlobalMetadataBitPosIndex.push_back(CurrentValue);
862 break;
864 case bitc::METADATA_INDEX:
865 // We don't expect to get there, the Index is loaded when we encounter
866 // the offset.
867 return error("Corrupted Metadata block");
868 case bitc::METADATA_NAME: {
869 // Named metadata need to be materialized now and aren't deferred.
870 if (Error Err = IndexCursor.JumpToBit(CurrentPos))
871 return std::move(Err);
872 Record.clear();
874 unsigned Code;
875 if (Expected<unsigned> MaybeCode =
876 IndexCursor.readRecord(Entry.ID, Record)) {
877 Code = MaybeCode.get();
878 assert(Code == bitc::METADATA_NAME);
879 } else
880 return MaybeCode.takeError();
882 // Read name of the named metadata.
883 SmallString<8> Name(Record.begin(), Record.end());
884 if (Expected<unsigned> MaybeCode = IndexCursor.ReadCode())
885 Code = MaybeCode.get();
886 else
887 return MaybeCode.takeError();
889 // Named Metadata comes in two parts, we expect the name to be followed
890 // by the node
891 Record.clear();
892 if (Expected<unsigned> MaybeNextBitCode =
893 IndexCursor.readRecord(Code, Record))
894 assert(MaybeNextBitCode.get() == bitc::METADATA_NAMED_NODE);
895 else
896 return MaybeNextBitCode.takeError();
898 // Read named metadata elements.
899 unsigned Size = Record.size();
900 NamedMDNode *NMD = TheModule.getOrInsertNamedMetadata(Name);
901 for (unsigned i = 0; i != Size; ++i) {
902 // FIXME: We could use a placeholder here, however NamedMDNode are
903 // taking MDNode as operand and not using the Metadata infrastructure.
904 // It is acknowledged by 'TODO: Inherit from Metadata' in the
905 // NamedMDNode class definition.
906 MDNode *MD = MetadataList.getMDNodeFwdRefOrNull(Record[i]);
907 assert(MD && "Invalid metadata: expect fwd ref to MDNode");
908 NMD->addOperand(MD);
910 break;
912 case bitc::METADATA_GLOBAL_DECL_ATTACHMENT: {
913 if (!GlobalDeclAttachmentPos)
914 GlobalDeclAttachmentPos = SavedPos;
915 #ifndef NDEBUG
916 NumGlobalDeclAttachSkipped++;
917 #endif
918 break;
920 case bitc::METADATA_KIND:
921 case bitc::METADATA_STRING_OLD:
922 case bitc::METADATA_OLD_FN_NODE:
923 case bitc::METADATA_OLD_NODE:
924 case bitc::METADATA_VALUE:
925 case bitc::METADATA_DISTINCT_NODE:
926 case bitc::METADATA_NODE:
927 case bitc::METADATA_LOCATION:
928 case bitc::METADATA_GENERIC_DEBUG:
929 case bitc::METADATA_SUBRANGE:
930 case bitc::METADATA_ENUMERATOR:
931 case bitc::METADATA_BASIC_TYPE:
932 case bitc::METADATA_STRING_TYPE:
933 case bitc::METADATA_DERIVED_TYPE:
934 case bitc::METADATA_COMPOSITE_TYPE:
935 case bitc::METADATA_SUBROUTINE_TYPE:
936 case bitc::METADATA_MODULE:
937 case bitc::METADATA_FILE:
938 case bitc::METADATA_COMPILE_UNIT:
939 case bitc::METADATA_SUBPROGRAM:
940 case bitc::METADATA_LEXICAL_BLOCK:
941 case bitc::METADATA_LEXICAL_BLOCK_FILE:
942 case bitc::METADATA_NAMESPACE:
943 case bitc::METADATA_COMMON_BLOCK:
944 case bitc::METADATA_MACRO:
945 case bitc::METADATA_MACRO_FILE:
946 case bitc::METADATA_TEMPLATE_TYPE:
947 case bitc::METADATA_TEMPLATE_VALUE:
948 case bitc::METADATA_GLOBAL_VAR:
949 case bitc::METADATA_LOCAL_VAR:
950 case bitc::METADATA_ASSIGN_ID:
951 case bitc::METADATA_LABEL:
952 case bitc::METADATA_EXPRESSION:
953 case bitc::METADATA_OBJC_PROPERTY:
954 case bitc::METADATA_IMPORTED_ENTITY:
955 case bitc::METADATA_GLOBAL_VAR_EXPR:
956 case bitc::METADATA_GENERIC_SUBRANGE:
957 // We don't expect to see any of these, if we see one, give up on
958 // lazy-loading and fallback.
959 MDStringRef.clear();
960 GlobalMetadataBitPosIndex.clear();
961 return false;
963 break;
969 // Load the global decl attachments after building the lazy loading index.
970 // We don't load them "lazily" - all global decl attachments must be
971 // parsed since they aren't materialized on demand. However, by delaying
972 // their parsing until after the index is created, we can use the index
973 // instead of creating temporaries.
974 Expected<bool> MetadataLoader::MetadataLoaderImpl::loadGlobalDeclAttachments() {
975 // Nothing to do if we didn't find any of these metadata records.
976 if (!GlobalDeclAttachmentPos)
977 return true;
978 // Use a temporary cursor so that we don't mess up the main Stream cursor or
979 // the lazy loading IndexCursor (which holds the necessary abbrev ids).
980 BitstreamCursor TempCursor = Stream;
981 SmallVector<uint64_t, 64> Record;
982 // Jump to the position before the first global decl attachment, so we can
983 // scan for the first BitstreamEntry record.
984 if (Error Err = TempCursor.JumpToBit(GlobalDeclAttachmentPos))
985 return std::move(Err);
986 while (true) {
987 BitstreamEntry Entry;
988 if (Error E =
989 TempCursor
990 .advanceSkippingSubblocks(BitstreamCursor::AF_DontPopBlockAtEnd)
991 .moveInto(Entry))
992 return std::move(E);
994 switch (Entry.Kind) {
995 case BitstreamEntry::SubBlock: // Handled for us already.
996 case BitstreamEntry::Error:
997 return error("Malformed block");
998 case BitstreamEntry::EndBlock:
999 // Check that we parsed them all.
1000 assert(NumGlobalDeclAttachSkipped == NumGlobalDeclAttachParsed);
1001 return true;
1002 case BitstreamEntry::Record:
1003 break;
1005 uint64_t CurrentPos = TempCursor.GetCurrentBitNo();
1006 Expected<unsigned> MaybeCode = TempCursor.skipRecord(Entry.ID);
1007 if (!MaybeCode)
1008 return MaybeCode.takeError();
1009 if (MaybeCode.get() != bitc::METADATA_GLOBAL_DECL_ATTACHMENT) {
1010 // Anything other than a global decl attachment signals the end of
1011 // these records. Check that we parsed them all.
1012 assert(NumGlobalDeclAttachSkipped == NumGlobalDeclAttachParsed);
1013 return true;
1015 #ifndef NDEBUG
1016 NumGlobalDeclAttachParsed++;
1017 #endif
1018 // FIXME: we need to do this early because we don't materialize global
1019 // value explicitly.
1020 if (Error Err = TempCursor.JumpToBit(CurrentPos))
1021 return std::move(Err);
1022 Record.clear();
1023 if (Expected<unsigned> MaybeRecord =
1024 TempCursor.readRecord(Entry.ID, Record))
1026 else
1027 return MaybeRecord.takeError();
1028 if (Record.size() % 2 == 0)
1029 return error("Invalid record");
1030 unsigned ValueID = Record[0];
1031 if (ValueID >= ValueList.size())
1032 return error("Invalid record");
1033 if (auto *GO = dyn_cast<GlobalObject>(ValueList[ValueID])) {
1034 // Need to save and restore the current position since
1035 // parseGlobalObjectAttachment will resolve all forward references which
1036 // would require parsing from locations stored in the index.
1037 CurrentPos = TempCursor.GetCurrentBitNo();
1038 if (Error Err = parseGlobalObjectAttachment(
1039 *GO, ArrayRef<uint64_t>(Record).slice(1)))
1040 return std::move(Err);
1041 if (Error Err = TempCursor.JumpToBit(CurrentPos))
1042 return std::move(Err);
1047 void MetadataLoader::MetadataLoaderImpl::callMDTypeCallback(Metadata **Val,
1048 unsigned TypeID) {
1049 if (Callbacks.MDType) {
1050 (*Callbacks.MDType)(Val, TypeID, Callbacks.GetTypeByID,
1051 Callbacks.GetContainedTypeID);
1055 /// Parse a METADATA_BLOCK. If ModuleLevel is true then we are parsing
1056 /// module level metadata.
1057 Error MetadataLoader::MetadataLoaderImpl::parseMetadata(bool ModuleLevel) {
1058 if (!ModuleLevel && MetadataList.hasFwdRefs())
1059 return error("Invalid metadata: fwd refs into function blocks");
1061 // Record the entry position so that we can jump back here and efficiently
1062 // skip the whole block in case we lazy-load.
1063 auto EntryPos = Stream.GetCurrentBitNo();
1065 if (Error Err = Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
1066 return Err;
1068 SmallVector<uint64_t, 64> Record;
1069 PlaceholderQueue Placeholders;
1071 // We lazy-load module-level metadata: we build an index for each record, and
1072 // then load individual record as needed, starting with the named metadata.
1073 if (ModuleLevel && IsImporting && MetadataList.empty() &&
1074 !DisableLazyLoading) {
1075 auto SuccessOrErr = lazyLoadModuleMetadataBlock();
1076 if (!SuccessOrErr)
1077 return SuccessOrErr.takeError();
1078 if (SuccessOrErr.get()) {
1079 // An index was successfully created and we will be able to load metadata
1080 // on-demand.
1081 MetadataList.resize(MDStringRef.size() +
1082 GlobalMetadataBitPosIndex.size());
1084 // Now that we have built the index, load the global decl attachments
1085 // that were deferred during that process. This avoids creating
1086 // temporaries.
1087 SuccessOrErr = loadGlobalDeclAttachments();
1088 if (!SuccessOrErr)
1089 return SuccessOrErr.takeError();
1090 assert(SuccessOrErr.get());
1092 // Reading the named metadata created forward references and/or
1093 // placeholders, that we flush here.
1094 resolveForwardRefsAndPlaceholders(Placeholders);
1095 upgradeDebugInfo(ModuleLevel);
1096 // Return at the beginning of the block, since it is easy to skip it
1097 // entirely from there.
1098 Stream.ReadBlockEnd(); // Pop the abbrev block context.
1099 if (Error Err = IndexCursor.JumpToBit(EntryPos))
1100 return Err;
1101 if (Error Err = Stream.SkipBlock()) {
1102 // FIXME this drops the error on the floor, which
1103 // ThinLTO/X86/debuginfo-cu-import.ll relies on.
1104 consumeError(std::move(Err));
1105 return Error::success();
1107 return Error::success();
1109 // Couldn't load an index, fallback to loading all the block "old-style".
1112 unsigned NextMetadataNo = MetadataList.size();
1114 // Read all the records.
1115 while (true) {
1116 BitstreamEntry Entry;
1117 if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
1118 return E;
1120 switch (Entry.Kind) {
1121 case BitstreamEntry::SubBlock: // Handled for us already.
1122 case BitstreamEntry::Error:
1123 return error("Malformed block");
1124 case BitstreamEntry::EndBlock:
1125 resolveForwardRefsAndPlaceholders(Placeholders);
1126 upgradeDebugInfo(ModuleLevel);
1127 return Error::success();
1128 case BitstreamEntry::Record:
1129 // The interesting case.
1130 break;
1133 // Read a record.
1134 Record.clear();
1135 StringRef Blob;
1136 ++NumMDRecordLoaded;
1137 if (Expected<unsigned> MaybeCode =
1138 Stream.readRecord(Entry.ID, Record, &Blob)) {
1139 if (Error Err = parseOneMetadata(Record, MaybeCode.get(), Placeholders,
1140 Blob, NextMetadataNo))
1141 return Err;
1142 } else
1143 return MaybeCode.takeError();
1147 MDString *MetadataLoader::MetadataLoaderImpl::lazyLoadOneMDString(unsigned ID) {
1148 ++NumMDStringLoaded;
1149 if (Metadata *MD = MetadataList.lookup(ID))
1150 return cast<MDString>(MD);
1151 auto MDS = MDString::get(Context, MDStringRef[ID]);
1152 MetadataList.assignValue(MDS, ID);
1153 return MDS;
1156 void MetadataLoader::MetadataLoaderImpl::lazyLoadOneMetadata(
1157 unsigned ID, PlaceholderQueue &Placeholders) {
1158 assert(ID < (MDStringRef.size()) + GlobalMetadataBitPosIndex.size());
1159 assert(ID >= MDStringRef.size() && "Unexpected lazy-loading of MDString");
1160 // Lookup first if the metadata hasn't already been loaded.
1161 if (auto *MD = MetadataList.lookup(ID)) {
1162 auto *N = cast<MDNode>(MD);
1163 if (!N->isTemporary())
1164 return;
1166 SmallVector<uint64_t, 64> Record;
1167 StringRef Blob;
1168 if (Error Err = IndexCursor.JumpToBit(
1169 GlobalMetadataBitPosIndex[ID - MDStringRef.size()]))
1170 report_fatal_error("lazyLoadOneMetadata failed jumping: " +
1171 Twine(toString(std::move(Err))));
1172 BitstreamEntry Entry;
1173 if (Error E = IndexCursor.advanceSkippingSubblocks().moveInto(Entry))
1174 // FIXME this drops the error on the floor.
1175 report_fatal_error("lazyLoadOneMetadata failed advanceSkippingSubblocks: " +
1176 Twine(toString(std::move(E))));
1177 ++NumMDRecordLoaded;
1178 if (Expected<unsigned> MaybeCode =
1179 IndexCursor.readRecord(Entry.ID, Record, &Blob)) {
1180 if (Error Err =
1181 parseOneMetadata(Record, MaybeCode.get(), Placeholders, Blob, ID))
1182 report_fatal_error("Can't lazyload MD, parseOneMetadata: " +
1183 Twine(toString(std::move(Err))));
1184 } else
1185 report_fatal_error("Can't lazyload MD: " +
1186 Twine(toString(MaybeCode.takeError())));
1189 /// Ensure that all forward-references and placeholders are resolved.
1190 /// Iteratively lazy-loading metadata on-demand if needed.
1191 void MetadataLoader::MetadataLoaderImpl::resolveForwardRefsAndPlaceholders(
1192 PlaceholderQueue &Placeholders) {
1193 DenseSet<unsigned> Temporaries;
1194 while (true) {
1195 // Populate Temporaries with the placeholders that haven't been loaded yet.
1196 Placeholders.getTemporaries(MetadataList, Temporaries);
1198 // If we don't have any temporary, or FwdReference, we're done!
1199 if (Temporaries.empty() && !MetadataList.hasFwdRefs())
1200 break;
1202 // First, load all the temporaries. This can add new placeholders or
1203 // forward references.
1204 for (auto ID : Temporaries)
1205 lazyLoadOneMetadata(ID, Placeholders);
1206 Temporaries.clear();
1208 // Second, load the forward-references. This can also add new placeholders
1209 // or forward references.
1210 while (MetadataList.hasFwdRefs())
1211 lazyLoadOneMetadata(MetadataList.getNextFwdRef(), Placeholders);
1213 // At this point we don't have any forward reference remaining, or temporary
1214 // that haven't been loaded. We can safely drop RAUW support and mark cycles
1215 // as resolved.
1216 MetadataList.tryToResolveCycles();
1218 // Finally, everything is in place, we can replace the placeholders operands
1219 // with the final node they refer to.
1220 Placeholders.flush(MetadataList);
1223 static Value *getValueFwdRef(BitcodeReaderValueList &ValueList, unsigned Idx,
1224 Type *Ty, unsigned TyID) {
1225 Value *V = ValueList.getValueFwdRef(Idx, Ty, TyID,
1226 /*ConstExprInsertBB*/ nullptr);
1227 if (V)
1228 return V;
1230 // This is a reference to a no longer supported constant expression.
1231 // Pretend that the constant was deleted, which will replace metadata
1232 // references with undef.
1233 // TODO: This is a rather indirect check. It would be more elegant to use
1234 // a separate ErrorInfo for constant materialization failure and thread
1235 // the error reporting through getValueFwdRef().
1236 if (Idx < ValueList.size() && ValueList[Idx] &&
1237 ValueList[Idx]->getType() == Ty)
1238 return UndefValue::get(Ty);
1240 return nullptr;
1243 Error MetadataLoader::MetadataLoaderImpl::parseOneMetadata(
1244 SmallVectorImpl<uint64_t> &Record, unsigned Code,
1245 PlaceholderQueue &Placeholders, StringRef Blob, unsigned &NextMetadataNo) {
1247 bool IsDistinct = false;
1248 auto getMD = [&](unsigned ID) -> Metadata * {
1249 if (ID < MDStringRef.size())
1250 return lazyLoadOneMDString(ID);
1251 if (!IsDistinct) {
1252 if (auto *MD = MetadataList.lookup(ID))
1253 return MD;
1254 // If lazy-loading is enabled, we try recursively to load the operand
1255 // instead of creating a temporary.
1256 if (ID < (MDStringRef.size() + GlobalMetadataBitPosIndex.size())) {
1257 // Create a temporary for the node that is referencing the operand we
1258 // will lazy-load. It is needed before recursing in case there are
1259 // uniquing cycles.
1260 MetadataList.getMetadataFwdRef(NextMetadataNo);
1261 lazyLoadOneMetadata(ID, Placeholders);
1262 return MetadataList.lookup(ID);
1264 // Return a temporary.
1265 return MetadataList.getMetadataFwdRef(ID);
1267 if (auto *MD = MetadataList.getMetadataIfResolved(ID))
1268 return MD;
1269 return &Placeholders.getPlaceholderOp(ID);
1271 auto getMDOrNull = [&](unsigned ID) -> Metadata * {
1272 if (ID)
1273 return getMD(ID - 1);
1274 return nullptr;
1276 auto getMDOrNullWithoutPlaceholders = [&](unsigned ID) -> Metadata * {
1277 if (ID)
1278 return MetadataList.getMetadataFwdRef(ID - 1);
1279 return nullptr;
1281 auto getMDString = [&](unsigned ID) -> MDString * {
1282 // This requires that the ID is not really a forward reference. In
1283 // particular, the MDString must already have been resolved.
1284 auto MDS = getMDOrNull(ID);
1285 return cast_or_null<MDString>(MDS);
1288 // Support for old type refs.
1289 auto getDITypeRefOrNull = [&](unsigned ID) {
1290 return MetadataList.upgradeTypeRef(getMDOrNull(ID));
1293 #define GET_OR_DISTINCT(CLASS, ARGS) \
1294 (IsDistinct ? CLASS::getDistinct ARGS : CLASS::get ARGS)
1296 switch (Code) {
1297 default: // Default behavior: ignore.
1298 break;
1299 case bitc::METADATA_NAME: {
1300 // Read name of the named metadata.
1301 SmallString<8> Name(Record.begin(), Record.end());
1302 Record.clear();
1303 if (Error E = Stream.ReadCode().moveInto(Code))
1304 return E;
1306 ++NumMDRecordLoaded;
1307 if (Expected<unsigned> MaybeNextBitCode = Stream.readRecord(Code, Record)) {
1308 if (MaybeNextBitCode.get() != bitc::METADATA_NAMED_NODE)
1309 return error("METADATA_NAME not followed by METADATA_NAMED_NODE");
1310 } else
1311 return MaybeNextBitCode.takeError();
1313 // Read named metadata elements.
1314 unsigned Size = Record.size();
1315 NamedMDNode *NMD = TheModule.getOrInsertNamedMetadata(Name);
1316 for (unsigned i = 0; i != Size; ++i) {
1317 MDNode *MD = MetadataList.getMDNodeFwdRefOrNull(Record[i]);
1318 if (!MD)
1319 return error("Invalid named metadata: expect fwd ref to MDNode");
1320 NMD->addOperand(MD);
1322 break;
1324 case bitc::METADATA_OLD_FN_NODE: {
1325 // Deprecated, but still needed to read old bitcode files.
1326 // This is a LocalAsMetadata record, the only type of function-local
1327 // metadata.
1328 if (Record.size() % 2 == 1)
1329 return error("Invalid record");
1331 // If this isn't a LocalAsMetadata record, we're dropping it. This used
1332 // to be legal, but there's no upgrade path.
1333 auto dropRecord = [&] {
1334 MetadataList.assignValue(MDNode::get(Context, {}), NextMetadataNo);
1335 NextMetadataNo++;
1337 if (Record.size() != 2) {
1338 dropRecord();
1339 break;
1342 unsigned TyID = Record[0];
1343 Type *Ty = Callbacks.GetTypeByID(TyID);
1344 if (!Ty || Ty->isMetadataTy() || Ty->isVoidTy()) {
1345 dropRecord();
1346 break;
1349 Value *V = ValueList.getValueFwdRef(Record[1], Ty, TyID,
1350 /*ConstExprInsertBB*/ nullptr);
1351 if (!V)
1352 return error("Invalid value reference from old fn metadata");
1354 MetadataList.assignValue(LocalAsMetadata::get(V), NextMetadataNo);
1355 NextMetadataNo++;
1356 break;
1358 case bitc::METADATA_OLD_NODE: {
1359 // Deprecated, but still needed to read old bitcode files.
1360 if (Record.size() % 2 == 1)
1361 return error("Invalid record");
1363 unsigned Size = Record.size();
1364 SmallVector<Metadata *, 8> Elts;
1365 for (unsigned i = 0; i != Size; i += 2) {
1366 unsigned TyID = Record[i];
1367 Type *Ty = Callbacks.GetTypeByID(TyID);
1368 if (!Ty)
1369 return error("Invalid record");
1370 if (Ty->isMetadataTy())
1371 Elts.push_back(getMD(Record[i + 1]));
1372 else if (!Ty->isVoidTy()) {
1373 Value *V = getValueFwdRef(ValueList, Record[i + 1], Ty, TyID);
1374 if (!V)
1375 return error("Invalid value reference from old metadata");
1376 Metadata *MD = ValueAsMetadata::get(V);
1377 assert(isa<ConstantAsMetadata>(MD) &&
1378 "Expected non-function-local metadata");
1379 callMDTypeCallback(&MD, TyID);
1380 Elts.push_back(MD);
1381 } else
1382 Elts.push_back(nullptr);
1384 MetadataList.assignValue(MDNode::get(Context, Elts), NextMetadataNo);
1385 NextMetadataNo++;
1386 break;
1388 case bitc::METADATA_VALUE: {
1389 if (Record.size() != 2)
1390 return error("Invalid record");
1392 unsigned TyID = Record[0];
1393 Type *Ty = Callbacks.GetTypeByID(TyID);
1394 if (!Ty || Ty->isMetadataTy() || Ty->isVoidTy())
1395 return error("Invalid record");
1397 Value *V = getValueFwdRef(ValueList, Record[1], Ty, TyID);
1398 if (!V)
1399 return error("Invalid value reference from metadata");
1401 Metadata *MD = ValueAsMetadata::get(V);
1402 callMDTypeCallback(&MD, TyID);
1403 MetadataList.assignValue(MD, NextMetadataNo);
1404 NextMetadataNo++;
1405 break;
1407 case bitc::METADATA_DISTINCT_NODE:
1408 IsDistinct = true;
1409 [[fallthrough]];
1410 case bitc::METADATA_NODE: {
1411 SmallVector<Metadata *, 8> Elts;
1412 Elts.reserve(Record.size());
1413 for (unsigned ID : Record)
1414 Elts.push_back(getMDOrNull(ID));
1415 MetadataList.assignValue(IsDistinct ? MDNode::getDistinct(Context, Elts)
1416 : MDNode::get(Context, Elts),
1417 NextMetadataNo);
1418 NextMetadataNo++;
1419 break;
1421 case bitc::METADATA_LOCATION: {
1422 if (Record.size() != 5 && Record.size() != 6)
1423 return error("Invalid record");
1425 IsDistinct = Record[0];
1426 unsigned Line = Record[1];
1427 unsigned Column = Record[2];
1428 Metadata *Scope = getMD(Record[3]);
1429 Metadata *InlinedAt = getMDOrNull(Record[4]);
1430 bool ImplicitCode = Record.size() == 6 && Record[5];
1431 MetadataList.assignValue(
1432 GET_OR_DISTINCT(DILocation, (Context, Line, Column, Scope, InlinedAt,
1433 ImplicitCode)),
1434 NextMetadataNo);
1435 NextMetadataNo++;
1436 break;
1438 case bitc::METADATA_GENERIC_DEBUG: {
1439 if (Record.size() < 4)
1440 return error("Invalid record");
1442 IsDistinct = Record[0];
1443 unsigned Tag = Record[1];
1444 unsigned Version = Record[2];
1446 if (Tag >= 1u << 16 || Version != 0)
1447 return error("Invalid record");
1449 auto *Header = getMDString(Record[3]);
1450 SmallVector<Metadata *, 8> DwarfOps;
1451 for (unsigned I = 4, E = Record.size(); I != E; ++I)
1452 DwarfOps.push_back(getMDOrNull(Record[I]));
1453 MetadataList.assignValue(
1454 GET_OR_DISTINCT(GenericDINode, (Context, Tag, Header, DwarfOps)),
1455 NextMetadataNo);
1456 NextMetadataNo++;
1457 break;
1459 case bitc::METADATA_SUBRANGE: {
1460 Metadata *Val = nullptr;
1461 // Operand 'count' is interpreted as:
1462 // - Signed integer (version 0)
1463 // - Metadata node (version 1)
1464 // Operand 'lowerBound' is interpreted as:
1465 // - Signed integer (version 0 and 1)
1466 // - Metadata node (version 2)
1467 // Operands 'upperBound' and 'stride' are interpreted as:
1468 // - Metadata node (version 2)
1469 switch (Record[0] >> 1) {
1470 case 0:
1471 Val = GET_OR_DISTINCT(DISubrange,
1472 (Context, Record[1], unrotateSign(Record[2])));
1473 break;
1474 case 1:
1475 Val = GET_OR_DISTINCT(DISubrange, (Context, getMDOrNull(Record[1]),
1476 unrotateSign(Record[2])));
1477 break;
1478 case 2:
1479 Val = GET_OR_DISTINCT(
1480 DISubrange, (Context, getMDOrNull(Record[1]), getMDOrNull(Record[2]),
1481 getMDOrNull(Record[3]), getMDOrNull(Record[4])));
1482 break;
1483 default:
1484 return error("Invalid record: Unsupported version of DISubrange");
1487 MetadataList.assignValue(Val, NextMetadataNo);
1488 IsDistinct = Record[0] & 1;
1489 NextMetadataNo++;
1490 break;
1492 case bitc::METADATA_GENERIC_SUBRANGE: {
1493 Metadata *Val = nullptr;
1494 Val = GET_OR_DISTINCT(DIGenericSubrange,
1495 (Context, getMDOrNull(Record[1]),
1496 getMDOrNull(Record[2]), getMDOrNull(Record[3]),
1497 getMDOrNull(Record[4])));
1499 MetadataList.assignValue(Val, NextMetadataNo);
1500 IsDistinct = Record[0] & 1;
1501 NextMetadataNo++;
1502 break;
1504 case bitc::METADATA_ENUMERATOR: {
1505 if (Record.size() < 3)
1506 return error("Invalid record");
1508 IsDistinct = Record[0] & 1;
1509 bool IsUnsigned = Record[0] & 2;
1510 bool IsBigInt = Record[0] & 4;
1511 APInt Value;
1513 if (IsBigInt) {
1514 const uint64_t BitWidth = Record[1];
1515 const size_t NumWords = Record.size() - 3;
1516 Value = readWideAPInt(ArrayRef(&Record[3], NumWords), BitWidth);
1517 } else
1518 Value = APInt(64, unrotateSign(Record[1]), !IsUnsigned);
1520 MetadataList.assignValue(
1521 GET_OR_DISTINCT(DIEnumerator,
1522 (Context, Value, IsUnsigned, getMDString(Record[2]))),
1523 NextMetadataNo);
1524 NextMetadataNo++;
1525 break;
1527 case bitc::METADATA_BASIC_TYPE: {
1528 if (Record.size() < 6 || Record.size() > 8)
1529 return error("Invalid record");
1531 IsDistinct = Record[0];
1532 DINode::DIFlags Flags = (Record.size() > 6)
1533 ? static_cast<DINode::DIFlags>(Record[6])
1534 : DINode::FlagZero;
1535 uint32_t NumExtraInhabitants = (Record.size() > 7) ? Record[7] : 0;
1537 MetadataList.assignValue(
1538 GET_OR_DISTINCT(DIBasicType,
1539 (Context, Record[1], getMDString(Record[2]), Record[3],
1540 Record[4], Record[5], NumExtraInhabitants, Flags)),
1541 NextMetadataNo);
1542 NextMetadataNo++;
1543 break;
1545 case bitc::METADATA_STRING_TYPE: {
1546 if (Record.size() > 9 || Record.size() < 8)
1547 return error("Invalid record");
1549 IsDistinct = Record[0];
1550 bool SizeIs8 = Record.size() == 8;
1551 // StringLocationExp (i.e. Record[5]) is added at a later time
1552 // than the other fields. The code here enables backward compatibility.
1553 Metadata *StringLocationExp = SizeIs8 ? nullptr : getMDOrNull(Record[5]);
1554 unsigned Offset = SizeIs8 ? 5 : 6;
1555 MetadataList.assignValue(
1556 GET_OR_DISTINCT(DIStringType,
1557 (Context, Record[1], getMDString(Record[2]),
1558 getMDOrNull(Record[3]), getMDOrNull(Record[4]),
1559 StringLocationExp, Record[Offset], Record[Offset + 1],
1560 Record[Offset + 2])),
1561 NextMetadataNo);
1562 NextMetadataNo++;
1563 break;
1565 case bitc::METADATA_DERIVED_TYPE: {
1566 if (Record.size() < 12 || Record.size() > 15)
1567 return error("Invalid record");
1569 // DWARF address space is encoded as N->getDWARFAddressSpace() + 1. 0 means
1570 // that there is no DWARF address space associated with DIDerivedType.
1571 std::optional<unsigned> DWARFAddressSpace;
1572 if (Record.size() > 12 && Record[12])
1573 DWARFAddressSpace = Record[12] - 1;
1575 Metadata *Annotations = nullptr;
1576 std::optional<DIDerivedType::PtrAuthData> PtrAuthData;
1578 // Only look for annotations/ptrauth if both are allocated.
1579 // If not, we can't tell which was intended to be embedded, as both ptrauth
1580 // and annotations have been expected at Record[13] at various times.
1581 if (Record.size() > 14) {
1582 if (Record[13])
1583 Annotations = getMDOrNull(Record[13]);
1584 if (Record[14])
1585 PtrAuthData.emplace(Record[14]);
1588 IsDistinct = Record[0];
1589 DINode::DIFlags Flags = static_cast<DINode::DIFlags>(Record[10]);
1590 MetadataList.assignValue(
1591 GET_OR_DISTINCT(DIDerivedType,
1592 (Context, Record[1], getMDString(Record[2]),
1593 getMDOrNull(Record[3]), Record[4],
1594 getDITypeRefOrNull(Record[5]),
1595 getDITypeRefOrNull(Record[6]), Record[7], Record[8],
1596 Record[9], DWARFAddressSpace, PtrAuthData, Flags,
1597 getDITypeRefOrNull(Record[11]), Annotations)),
1598 NextMetadataNo);
1599 NextMetadataNo++;
1600 break;
1602 case bitc::METADATA_COMPOSITE_TYPE: {
1603 if (Record.size() < 16 || Record.size() > 24)
1604 return error("Invalid record");
1606 // If we have a UUID and this is not a forward declaration, lookup the
1607 // mapping.
1608 IsDistinct = Record[0] & 0x1;
1609 bool IsNotUsedInTypeRef = Record[0] >= 2;
1610 unsigned Tag = Record[1];
1611 MDString *Name = getMDString(Record[2]);
1612 Metadata *File = getMDOrNull(Record[3]);
1613 unsigned Line = Record[4];
1614 Metadata *Scope = getDITypeRefOrNull(Record[5]);
1615 Metadata *BaseType = nullptr;
1616 uint64_t SizeInBits = Record[7];
1617 if (Record[8] > (uint64_t)std::numeric_limits<uint32_t>::max())
1618 return error("Alignment value is too large");
1619 uint32_t AlignInBits = Record[8];
1620 uint64_t OffsetInBits = 0;
1621 uint32_t NumExtraInhabitants = (Record.size() > 22) ? Record[22] : 0;
1622 DINode::DIFlags Flags = static_cast<DINode::DIFlags>(Record[10]);
1623 Metadata *Elements = nullptr;
1624 unsigned RuntimeLang = Record[12];
1625 Metadata *VTableHolder = nullptr;
1626 Metadata *TemplateParams = nullptr;
1627 Metadata *Discriminator = nullptr;
1628 Metadata *DataLocation = nullptr;
1629 Metadata *Associated = nullptr;
1630 Metadata *Allocated = nullptr;
1631 Metadata *Rank = nullptr;
1632 Metadata *Annotations = nullptr;
1633 Metadata *Specification = nullptr;
1634 auto *Identifier = getMDString(Record[15]);
1635 // If this module is being parsed so that it can be ThinLTO imported
1636 // into another module, composite types only need to be imported as
1637 // type declarations (unless full type definitions are requested).
1638 // Create type declarations up front to save memory. This is only
1639 // done for types which have an Identifier, and are therefore
1640 // subject to the ODR.
1642 // buildODRType handles the case where this is type ODRed with a
1643 // definition needed by the importing module, in which case the
1644 // existing definition is used.
1646 // We always import full definitions for anonymous composite types,
1647 // as without a name, debuggers cannot easily resolve a declaration
1648 // to its definition.
1649 if (IsImporting && !ImportFullTypeDefinitions && Identifier && Name &&
1650 (Tag == dwarf::DW_TAG_enumeration_type ||
1651 Tag == dwarf::DW_TAG_class_type ||
1652 Tag == dwarf::DW_TAG_structure_type ||
1653 Tag == dwarf::DW_TAG_union_type)) {
1654 Flags = Flags | DINode::FlagFwdDecl;
1655 // This is a hack around preserving template parameters for simplified
1656 // template names - it should probably be replaced with a
1657 // DICompositeType flag specifying whether template parameters are
1658 // required on declarations of this type.
1659 StringRef NameStr = Name->getString();
1660 if (!NameStr.contains('<') || NameStr.starts_with("_STN|"))
1661 TemplateParams = getMDOrNull(Record[14]);
1662 } else {
1663 BaseType = getDITypeRefOrNull(Record[6]);
1664 OffsetInBits = Record[9];
1665 Elements = getMDOrNull(Record[11]);
1666 VTableHolder = getDITypeRefOrNull(Record[13]);
1667 TemplateParams = getMDOrNull(Record[14]);
1668 if (Record.size() > 16)
1669 Discriminator = getMDOrNull(Record[16]);
1670 if (Record.size() > 17)
1671 DataLocation = getMDOrNull(Record[17]);
1672 if (Record.size() > 19) {
1673 Associated = getMDOrNull(Record[18]);
1674 Allocated = getMDOrNull(Record[19]);
1676 if (Record.size() > 20) {
1677 Rank = getMDOrNull(Record[20]);
1679 if (Record.size() > 21) {
1680 Annotations = getMDOrNull(Record[21]);
1682 if (Record.size() > 23) {
1683 Specification = getMDOrNull(Record[23]);
1686 DICompositeType *CT = nullptr;
1687 if (Identifier)
1688 CT = DICompositeType::buildODRType(
1689 Context, *Identifier, Tag, Name, File, Line, Scope, BaseType,
1690 SizeInBits, AlignInBits, OffsetInBits, Specification,
1691 NumExtraInhabitants, Flags, Elements, RuntimeLang, VTableHolder,
1692 TemplateParams, Discriminator, DataLocation, Associated, Allocated,
1693 Rank, Annotations);
1695 // Create a node if we didn't get a lazy ODR type.
1696 if (!CT)
1697 CT = GET_OR_DISTINCT(DICompositeType,
1698 (Context, Tag, Name, File, Line, Scope, BaseType,
1699 SizeInBits, AlignInBits, OffsetInBits, Flags,
1700 Elements, RuntimeLang, VTableHolder, TemplateParams,
1701 Identifier, Discriminator, DataLocation, Associated,
1702 Allocated, Rank, Annotations, Specification,
1703 NumExtraInhabitants));
1704 if (!IsNotUsedInTypeRef && Identifier)
1705 MetadataList.addTypeRef(*Identifier, *cast<DICompositeType>(CT));
1707 MetadataList.assignValue(CT, NextMetadataNo);
1708 NextMetadataNo++;
1709 break;
1711 case bitc::METADATA_SUBROUTINE_TYPE: {
1712 if (Record.size() < 3 || Record.size() > 4)
1713 return error("Invalid record");
1714 bool IsOldTypeRefArray = Record[0] < 2;
1715 unsigned CC = (Record.size() > 3) ? Record[3] : 0;
1717 IsDistinct = Record[0] & 0x1;
1718 DINode::DIFlags Flags = static_cast<DINode::DIFlags>(Record[1]);
1719 Metadata *Types = getMDOrNull(Record[2]);
1720 if (LLVM_UNLIKELY(IsOldTypeRefArray))
1721 Types = MetadataList.upgradeTypeRefArray(Types);
1723 MetadataList.assignValue(
1724 GET_OR_DISTINCT(DISubroutineType, (Context, Flags, CC, Types)),
1725 NextMetadataNo);
1726 NextMetadataNo++;
1727 break;
1730 case bitc::METADATA_MODULE: {
1731 if (Record.size() < 5 || Record.size() > 9)
1732 return error("Invalid record");
1734 unsigned Offset = Record.size() >= 8 ? 2 : 1;
1735 IsDistinct = Record[0];
1736 MetadataList.assignValue(
1737 GET_OR_DISTINCT(
1738 DIModule,
1739 (Context, Record.size() >= 8 ? getMDOrNull(Record[1]) : nullptr,
1740 getMDOrNull(Record[0 + Offset]), getMDString(Record[1 + Offset]),
1741 getMDString(Record[2 + Offset]), getMDString(Record[3 + Offset]),
1742 getMDString(Record[4 + Offset]),
1743 Record.size() <= 7 ? 0 : Record[7],
1744 Record.size() <= 8 ? false : Record[8])),
1745 NextMetadataNo);
1746 NextMetadataNo++;
1747 break;
1750 case bitc::METADATA_FILE: {
1751 if (Record.size() != 3 && Record.size() != 5 && Record.size() != 6)
1752 return error("Invalid record");
1754 IsDistinct = Record[0];
1755 std::optional<DIFile::ChecksumInfo<MDString *>> Checksum;
1756 // The BitcodeWriter writes null bytes into Record[3:4] when the Checksum
1757 // is not present. This matches up with the old internal representation,
1758 // and the old encoding for CSK_None in the ChecksumKind. The new
1759 // representation reserves the value 0 in the ChecksumKind to continue to
1760 // encode None in a backwards-compatible way.
1761 if (Record.size() > 4 && Record[3] && Record[4])
1762 Checksum.emplace(static_cast<DIFile::ChecksumKind>(Record[3]),
1763 getMDString(Record[4]));
1764 MetadataList.assignValue(
1765 GET_OR_DISTINCT(DIFile,
1766 (Context, getMDString(Record[1]),
1767 getMDString(Record[2]), Checksum,
1768 Record.size() > 5 ? getMDString(Record[5]) : nullptr)),
1769 NextMetadataNo);
1770 NextMetadataNo++;
1771 break;
1773 case bitc::METADATA_COMPILE_UNIT: {
1774 if (Record.size() < 14 || Record.size() > 22)
1775 return error("Invalid record");
1777 // Ignore Record[0], which indicates whether this compile unit is
1778 // distinct. It's always distinct.
1779 IsDistinct = true;
1780 auto *CU = DICompileUnit::getDistinct(
1781 Context, Record[1], getMDOrNull(Record[2]), getMDString(Record[3]),
1782 Record[4], getMDString(Record[5]), Record[6], getMDString(Record[7]),
1783 Record[8], getMDOrNull(Record[9]), getMDOrNull(Record[10]),
1784 getMDOrNull(Record[12]), getMDOrNull(Record[13]),
1785 Record.size() <= 15 ? nullptr : getMDOrNull(Record[15]),
1786 Record.size() <= 14 ? 0 : Record[14],
1787 Record.size() <= 16 ? true : Record[16],
1788 Record.size() <= 17 ? false : Record[17],
1789 Record.size() <= 18 ? 0 : Record[18],
1790 Record.size() <= 19 ? false : Record[19],
1791 Record.size() <= 20 ? nullptr : getMDString(Record[20]),
1792 Record.size() <= 21 ? nullptr : getMDString(Record[21]));
1794 MetadataList.assignValue(CU, NextMetadataNo);
1795 NextMetadataNo++;
1797 // Move the Upgrade the list of subprograms.
1798 if (Metadata *SPs = getMDOrNullWithoutPlaceholders(Record[11]))
1799 CUSubprograms.push_back({CU, SPs});
1800 break;
1802 case bitc::METADATA_SUBPROGRAM: {
1803 if (Record.size() < 18 || Record.size() > 21)
1804 return error("Invalid record");
1806 bool HasSPFlags = Record[0] & 4;
1808 DINode::DIFlags Flags;
1809 DISubprogram::DISPFlags SPFlags;
1810 if (!HasSPFlags)
1811 Flags = static_cast<DINode::DIFlags>(Record[11 + 2]);
1812 else {
1813 Flags = static_cast<DINode::DIFlags>(Record[11]);
1814 SPFlags = static_cast<DISubprogram::DISPFlags>(Record[9]);
1817 // Support for old metadata when
1818 // subprogram specific flags are placed in DIFlags.
1819 const unsigned DIFlagMainSubprogram = 1 << 21;
1820 bool HasOldMainSubprogramFlag = Flags & DIFlagMainSubprogram;
1821 if (HasOldMainSubprogramFlag)
1822 // Remove old DIFlagMainSubprogram from DIFlags.
1823 // Note: This assumes that any future use of bit 21 defaults to it
1824 // being 0.
1825 Flags &= ~static_cast<DINode::DIFlags>(DIFlagMainSubprogram);
1827 if (HasOldMainSubprogramFlag && HasSPFlags)
1828 SPFlags |= DISubprogram::SPFlagMainSubprogram;
1829 else if (!HasSPFlags)
1830 SPFlags = DISubprogram::toSPFlags(
1831 /*IsLocalToUnit=*/Record[7], /*IsDefinition=*/Record[8],
1832 /*IsOptimized=*/Record[14], /*Virtuality=*/Record[11],
1833 /*IsMainSubprogram=*/HasOldMainSubprogramFlag);
1835 // All definitions should be distinct.
1836 IsDistinct = (Record[0] & 1) || (SPFlags & DISubprogram::SPFlagDefinition);
1837 // Version 1 has a Function as Record[15].
1838 // Version 2 has removed Record[15].
1839 // Version 3 has the Unit as Record[15].
1840 // Version 4 added thisAdjustment.
1841 // Version 5 repacked flags into DISPFlags, changing many element numbers.
1842 bool HasUnit = Record[0] & 2;
1843 if (!HasSPFlags && HasUnit && Record.size() < 19)
1844 return error("Invalid record");
1845 if (HasSPFlags && !HasUnit)
1846 return error("Invalid record");
1847 // Accommodate older formats.
1848 bool HasFn = false;
1849 bool HasThisAdj = true;
1850 bool HasThrownTypes = true;
1851 bool HasAnnotations = false;
1852 bool HasTargetFuncName = false;
1853 unsigned OffsetA = 0;
1854 unsigned OffsetB = 0;
1855 if (!HasSPFlags) {
1856 OffsetA = 2;
1857 OffsetB = 2;
1858 if (Record.size() >= 19) {
1859 HasFn = !HasUnit;
1860 OffsetB++;
1862 HasThisAdj = Record.size() >= 20;
1863 HasThrownTypes = Record.size() >= 21;
1864 } else {
1865 HasAnnotations = Record.size() >= 19;
1866 HasTargetFuncName = Record.size() >= 20;
1868 Metadata *CUorFn = getMDOrNull(Record[12 + OffsetB]);
1869 DISubprogram *SP = GET_OR_DISTINCT(
1870 DISubprogram,
1871 (Context,
1872 getDITypeRefOrNull(Record[1]), // scope
1873 getMDString(Record[2]), // name
1874 getMDString(Record[3]), // linkageName
1875 getMDOrNull(Record[4]), // file
1876 Record[5], // line
1877 getMDOrNull(Record[6]), // type
1878 Record[7 + OffsetA], // scopeLine
1879 getDITypeRefOrNull(Record[8 + OffsetA]), // containingType
1880 Record[10 + OffsetA], // virtualIndex
1881 HasThisAdj ? Record[16 + OffsetB] : 0, // thisAdjustment
1882 Flags, // flags
1883 SPFlags, // SPFlags
1884 HasUnit ? CUorFn : nullptr, // unit
1885 getMDOrNull(Record[13 + OffsetB]), // templateParams
1886 getMDOrNull(Record[14 + OffsetB]), // declaration
1887 getMDOrNull(Record[15 + OffsetB]), // retainedNodes
1888 HasThrownTypes ? getMDOrNull(Record[17 + OffsetB])
1889 : nullptr, // thrownTypes
1890 HasAnnotations ? getMDOrNull(Record[18 + OffsetB])
1891 : nullptr, // annotations
1892 HasTargetFuncName ? getMDString(Record[19 + OffsetB])
1893 : nullptr // targetFuncName
1895 MetadataList.assignValue(SP, NextMetadataNo);
1896 NextMetadataNo++;
1898 // Upgrade sp->function mapping to function->sp mapping.
1899 if (HasFn) {
1900 if (auto *CMD = dyn_cast_or_null<ConstantAsMetadata>(CUorFn))
1901 if (auto *F = dyn_cast<Function>(CMD->getValue())) {
1902 if (F->isMaterializable())
1903 // Defer until materialized; unmaterialized functions may not have
1904 // metadata.
1905 FunctionsWithSPs[F] = SP;
1906 else if (!F->empty())
1907 F->setSubprogram(SP);
1910 break;
1912 case bitc::METADATA_LEXICAL_BLOCK: {
1913 if (Record.size() != 5)
1914 return error("Invalid record");
1916 IsDistinct = Record[0];
1917 MetadataList.assignValue(
1918 GET_OR_DISTINCT(DILexicalBlock,
1919 (Context, getMDOrNull(Record[1]),
1920 getMDOrNull(Record[2]), Record[3], Record[4])),
1921 NextMetadataNo);
1922 NextMetadataNo++;
1923 break;
1925 case bitc::METADATA_LEXICAL_BLOCK_FILE: {
1926 if (Record.size() != 4)
1927 return error("Invalid record");
1929 IsDistinct = Record[0];
1930 MetadataList.assignValue(
1931 GET_OR_DISTINCT(DILexicalBlockFile,
1932 (Context, getMDOrNull(Record[1]),
1933 getMDOrNull(Record[2]), Record[3])),
1934 NextMetadataNo);
1935 NextMetadataNo++;
1936 break;
1938 case bitc::METADATA_COMMON_BLOCK: {
1939 IsDistinct = Record[0] & 1;
1940 MetadataList.assignValue(
1941 GET_OR_DISTINCT(DICommonBlock,
1942 (Context, getMDOrNull(Record[1]),
1943 getMDOrNull(Record[2]), getMDString(Record[3]),
1944 getMDOrNull(Record[4]), Record[5])),
1945 NextMetadataNo);
1946 NextMetadataNo++;
1947 break;
1949 case bitc::METADATA_NAMESPACE: {
1950 // Newer versions of DINamespace dropped file and line.
1951 MDString *Name;
1952 if (Record.size() == 3)
1953 Name = getMDString(Record[2]);
1954 else if (Record.size() == 5)
1955 Name = getMDString(Record[3]);
1956 else
1957 return error("Invalid record");
1959 IsDistinct = Record[0] & 1;
1960 bool ExportSymbols = Record[0] & 2;
1961 MetadataList.assignValue(
1962 GET_OR_DISTINCT(DINamespace,
1963 (Context, getMDOrNull(Record[1]), Name, ExportSymbols)),
1964 NextMetadataNo);
1965 NextMetadataNo++;
1966 break;
1968 case bitc::METADATA_MACRO: {
1969 if (Record.size() != 5)
1970 return error("Invalid record");
1972 IsDistinct = Record[0];
1973 MetadataList.assignValue(
1974 GET_OR_DISTINCT(DIMacro,
1975 (Context, Record[1], Record[2], getMDString(Record[3]),
1976 getMDString(Record[4]))),
1977 NextMetadataNo);
1978 NextMetadataNo++;
1979 break;
1981 case bitc::METADATA_MACRO_FILE: {
1982 if (Record.size() != 5)
1983 return error("Invalid record");
1985 IsDistinct = Record[0];
1986 MetadataList.assignValue(
1987 GET_OR_DISTINCT(DIMacroFile,
1988 (Context, Record[1], Record[2], getMDOrNull(Record[3]),
1989 getMDOrNull(Record[4]))),
1990 NextMetadataNo);
1991 NextMetadataNo++;
1992 break;
1994 case bitc::METADATA_TEMPLATE_TYPE: {
1995 if (Record.size() < 3 || Record.size() > 4)
1996 return error("Invalid record");
1998 IsDistinct = Record[0];
1999 MetadataList.assignValue(
2000 GET_OR_DISTINCT(DITemplateTypeParameter,
2001 (Context, getMDString(Record[1]),
2002 getDITypeRefOrNull(Record[2]),
2003 (Record.size() == 4) ? getMDOrNull(Record[3])
2004 : getMDOrNull(false))),
2005 NextMetadataNo);
2006 NextMetadataNo++;
2007 break;
2009 case bitc::METADATA_TEMPLATE_VALUE: {
2010 if (Record.size() < 5 || Record.size() > 6)
2011 return error("Invalid record");
2013 IsDistinct = Record[0];
2015 MetadataList.assignValue(
2016 GET_OR_DISTINCT(
2017 DITemplateValueParameter,
2018 (Context, Record[1], getMDString(Record[2]),
2019 getDITypeRefOrNull(Record[3]),
2020 (Record.size() == 6) ? getMDOrNull(Record[4]) : getMDOrNull(false),
2021 (Record.size() == 6) ? getMDOrNull(Record[5])
2022 : getMDOrNull(Record[4]))),
2023 NextMetadataNo);
2024 NextMetadataNo++;
2025 break;
2027 case bitc::METADATA_GLOBAL_VAR: {
2028 if (Record.size() < 11 || Record.size() > 13)
2029 return error("Invalid record");
2031 IsDistinct = Record[0] & 1;
2032 unsigned Version = Record[0] >> 1;
2034 if (Version == 2) {
2035 Metadata *Annotations = nullptr;
2036 if (Record.size() > 12)
2037 Annotations = getMDOrNull(Record[12]);
2039 MetadataList.assignValue(
2040 GET_OR_DISTINCT(DIGlobalVariable,
2041 (Context, getMDOrNull(Record[1]),
2042 getMDString(Record[2]), getMDString(Record[3]),
2043 getMDOrNull(Record[4]), Record[5],
2044 getDITypeRefOrNull(Record[6]), Record[7], Record[8],
2045 getMDOrNull(Record[9]), getMDOrNull(Record[10]),
2046 Record[11], Annotations)),
2047 NextMetadataNo);
2049 NextMetadataNo++;
2050 } else if (Version == 1) {
2051 // No upgrade necessary. A null field will be introduced to indicate
2052 // that no parameter information is available.
2053 MetadataList.assignValue(
2054 GET_OR_DISTINCT(
2055 DIGlobalVariable,
2056 (Context, getMDOrNull(Record[1]), getMDString(Record[2]),
2057 getMDString(Record[3]), getMDOrNull(Record[4]), Record[5],
2058 getDITypeRefOrNull(Record[6]), Record[7], Record[8],
2059 getMDOrNull(Record[10]), nullptr, Record[11], nullptr)),
2060 NextMetadataNo);
2062 NextMetadataNo++;
2063 } else if (Version == 0) {
2064 // Upgrade old metadata, which stored a global variable reference or a
2065 // ConstantInt here.
2066 NeedUpgradeToDIGlobalVariableExpression = true;
2067 Metadata *Expr = getMDOrNull(Record[9]);
2068 uint32_t AlignInBits = 0;
2069 if (Record.size() > 11) {
2070 if (Record[11] > (uint64_t)std::numeric_limits<uint32_t>::max())
2071 return error("Alignment value is too large");
2072 AlignInBits = Record[11];
2074 GlobalVariable *Attach = nullptr;
2075 if (auto *CMD = dyn_cast_or_null<ConstantAsMetadata>(Expr)) {
2076 if (auto *GV = dyn_cast<GlobalVariable>(CMD->getValue())) {
2077 Attach = GV;
2078 Expr = nullptr;
2079 } else if (auto *CI = dyn_cast<ConstantInt>(CMD->getValue())) {
2080 Expr = DIExpression::get(Context,
2081 {dwarf::DW_OP_constu, CI->getZExtValue(),
2082 dwarf::DW_OP_stack_value});
2083 } else {
2084 Expr = nullptr;
2087 DIGlobalVariable *DGV = GET_OR_DISTINCT(
2088 DIGlobalVariable,
2089 (Context, getMDOrNull(Record[1]), getMDString(Record[2]),
2090 getMDString(Record[3]), getMDOrNull(Record[4]), Record[5],
2091 getDITypeRefOrNull(Record[6]), Record[7], Record[8],
2092 getMDOrNull(Record[10]), nullptr, AlignInBits, nullptr));
2094 DIGlobalVariableExpression *DGVE = nullptr;
2095 if (Attach || Expr)
2096 DGVE = DIGlobalVariableExpression::getDistinct(
2097 Context, DGV, Expr ? Expr : DIExpression::get(Context, {}));
2098 if (Attach)
2099 Attach->addDebugInfo(DGVE);
2101 auto *MDNode = Expr ? cast<Metadata>(DGVE) : cast<Metadata>(DGV);
2102 MetadataList.assignValue(MDNode, NextMetadataNo);
2103 NextMetadataNo++;
2104 } else
2105 return error("Invalid record");
2107 break;
2109 case bitc::METADATA_ASSIGN_ID: {
2110 if (Record.size() != 1)
2111 return error("Invalid DIAssignID record.");
2113 IsDistinct = Record[0] & 1;
2114 if (!IsDistinct)
2115 return error("Invalid DIAssignID record. Must be distinct");
2117 MetadataList.assignValue(DIAssignID::getDistinct(Context), NextMetadataNo);
2118 NextMetadataNo++;
2119 break;
2121 case bitc::METADATA_LOCAL_VAR: {
2122 // 10th field is for the obseleted 'inlinedAt:' field.
2123 if (Record.size() < 8 || Record.size() > 10)
2124 return error("Invalid record");
2126 IsDistinct = Record[0] & 1;
2127 bool HasAlignment = Record[0] & 2;
2128 // 2nd field used to be an artificial tag, either DW_TAG_auto_variable or
2129 // DW_TAG_arg_variable, if we have alignment flag encoded it means, that
2130 // this is newer version of record which doesn't have artificial tag.
2131 bool HasTag = !HasAlignment && Record.size() > 8;
2132 DINode::DIFlags Flags = static_cast<DINode::DIFlags>(Record[7 + HasTag]);
2133 uint32_t AlignInBits = 0;
2134 Metadata *Annotations = nullptr;
2135 if (HasAlignment) {
2136 if (Record[8] > (uint64_t)std::numeric_limits<uint32_t>::max())
2137 return error("Alignment value is too large");
2138 AlignInBits = Record[8];
2139 if (Record.size() > 9)
2140 Annotations = getMDOrNull(Record[9]);
2143 MetadataList.assignValue(
2144 GET_OR_DISTINCT(DILocalVariable,
2145 (Context, getMDOrNull(Record[1 + HasTag]),
2146 getMDString(Record[2 + HasTag]),
2147 getMDOrNull(Record[3 + HasTag]), Record[4 + HasTag],
2148 getDITypeRefOrNull(Record[5 + HasTag]),
2149 Record[6 + HasTag], Flags, AlignInBits, Annotations)),
2150 NextMetadataNo);
2151 NextMetadataNo++;
2152 break;
2154 case bitc::METADATA_LABEL: {
2155 if (Record.size() != 5)
2156 return error("Invalid record");
2158 IsDistinct = Record[0] & 1;
2159 MetadataList.assignValue(
2160 GET_OR_DISTINCT(DILabel, (Context, getMDOrNull(Record[1]),
2161 getMDString(Record[2]),
2162 getMDOrNull(Record[3]), Record[4])),
2163 NextMetadataNo);
2164 NextMetadataNo++;
2165 break;
2167 case bitc::METADATA_EXPRESSION: {
2168 if (Record.size() < 1)
2169 return error("Invalid record");
2171 IsDistinct = Record[0] & 1;
2172 uint64_t Version = Record[0] >> 1;
2173 auto Elts = MutableArrayRef<uint64_t>(Record).slice(1);
2175 SmallVector<uint64_t, 6> Buffer;
2176 if (Error Err = upgradeDIExpression(Version, Elts, Buffer))
2177 return Err;
2179 MetadataList.assignValue(GET_OR_DISTINCT(DIExpression, (Context, Elts)),
2180 NextMetadataNo);
2181 NextMetadataNo++;
2182 break;
2184 case bitc::METADATA_GLOBAL_VAR_EXPR: {
2185 if (Record.size() != 3)
2186 return error("Invalid record");
2188 IsDistinct = Record[0];
2189 Metadata *Expr = getMDOrNull(Record[2]);
2190 if (!Expr)
2191 Expr = DIExpression::get(Context, {});
2192 MetadataList.assignValue(
2193 GET_OR_DISTINCT(DIGlobalVariableExpression,
2194 (Context, getMDOrNull(Record[1]), Expr)),
2195 NextMetadataNo);
2196 NextMetadataNo++;
2197 break;
2199 case bitc::METADATA_OBJC_PROPERTY: {
2200 if (Record.size() != 8)
2201 return error("Invalid record");
2203 IsDistinct = Record[0];
2204 MetadataList.assignValue(
2205 GET_OR_DISTINCT(DIObjCProperty,
2206 (Context, getMDString(Record[1]),
2207 getMDOrNull(Record[2]), Record[3],
2208 getMDString(Record[4]), getMDString(Record[5]),
2209 Record[6], getDITypeRefOrNull(Record[7]))),
2210 NextMetadataNo);
2211 NextMetadataNo++;
2212 break;
2214 case bitc::METADATA_IMPORTED_ENTITY: {
2215 if (Record.size() < 6 || Record.size() > 8)
2216 return error("Invalid DIImportedEntity record");
2218 IsDistinct = Record[0];
2219 bool HasFile = (Record.size() >= 7);
2220 bool HasElements = (Record.size() >= 8);
2221 MetadataList.assignValue(
2222 GET_OR_DISTINCT(DIImportedEntity,
2223 (Context, Record[1], getMDOrNull(Record[2]),
2224 getDITypeRefOrNull(Record[3]),
2225 HasFile ? getMDOrNull(Record[6]) : nullptr,
2226 HasFile ? Record[4] : 0, getMDString(Record[5]),
2227 HasElements ? getMDOrNull(Record[7]) : nullptr)),
2228 NextMetadataNo);
2229 NextMetadataNo++;
2230 break;
2232 case bitc::METADATA_STRING_OLD: {
2233 std::string String(Record.begin(), Record.end());
2235 // Test for upgrading !llvm.loop.
2236 HasSeenOldLoopTags |= mayBeOldLoopAttachmentTag(String);
2237 ++NumMDStringLoaded;
2238 Metadata *MD = MDString::get(Context, String);
2239 MetadataList.assignValue(MD, NextMetadataNo);
2240 NextMetadataNo++;
2241 break;
2243 case bitc::METADATA_STRINGS: {
2244 auto CreateNextMDString = [&](StringRef Str) {
2245 ++NumMDStringLoaded;
2246 MetadataList.assignValue(MDString::get(Context, Str), NextMetadataNo);
2247 NextMetadataNo++;
2249 if (Error Err = parseMetadataStrings(Record, Blob, CreateNextMDString))
2250 return Err;
2251 break;
2253 case bitc::METADATA_GLOBAL_DECL_ATTACHMENT: {
2254 if (Record.size() % 2 == 0)
2255 return error("Invalid record");
2256 unsigned ValueID = Record[0];
2257 if (ValueID >= ValueList.size())
2258 return error("Invalid record");
2259 if (auto *GO = dyn_cast<GlobalObject>(ValueList[ValueID]))
2260 if (Error Err = parseGlobalObjectAttachment(
2261 *GO, ArrayRef<uint64_t>(Record).slice(1)))
2262 return Err;
2263 break;
2265 case bitc::METADATA_KIND: {
2266 // Support older bitcode files that had METADATA_KIND records in a
2267 // block with METADATA_BLOCK_ID.
2268 if (Error Err = parseMetadataKindRecord(Record))
2269 return Err;
2270 break;
2272 case bitc::METADATA_ARG_LIST: {
2273 SmallVector<ValueAsMetadata *, 4> Elts;
2274 Elts.reserve(Record.size());
2275 for (uint64_t Elt : Record) {
2276 Metadata *MD = getMD(Elt);
2277 if (isa<MDNode>(MD) && cast<MDNode>(MD)->isTemporary())
2278 return error(
2279 "Invalid record: DIArgList should not contain forward refs");
2280 if (!isa<ValueAsMetadata>(MD))
2281 return error("Invalid record");
2282 Elts.push_back(cast<ValueAsMetadata>(MD));
2285 MetadataList.assignValue(DIArgList::get(Context, Elts), NextMetadataNo);
2286 NextMetadataNo++;
2287 break;
2290 return Error::success();
2291 #undef GET_OR_DISTINCT
2294 Error MetadataLoader::MetadataLoaderImpl::parseMetadataStrings(
2295 ArrayRef<uint64_t> Record, StringRef Blob,
2296 function_ref<void(StringRef)> CallBack) {
2297 // All the MDStrings in the block are emitted together in a single
2298 // record. The strings are concatenated and stored in a blob along with
2299 // their sizes.
2300 if (Record.size() != 2)
2301 return error("Invalid record: metadata strings layout");
2303 unsigned NumStrings = Record[0];
2304 unsigned StringsOffset = Record[1];
2305 if (!NumStrings)
2306 return error("Invalid record: metadata strings with no strings");
2307 if (StringsOffset > Blob.size())
2308 return error("Invalid record: metadata strings corrupt offset");
2310 StringRef Lengths = Blob.slice(0, StringsOffset);
2311 SimpleBitstreamCursor R(Lengths);
2313 StringRef Strings = Blob.drop_front(StringsOffset);
2314 do {
2315 if (R.AtEndOfStream())
2316 return error("Invalid record: metadata strings bad length");
2318 uint32_t Size;
2319 if (Error E = R.ReadVBR(6).moveInto(Size))
2320 return E;
2321 if (Strings.size() < Size)
2322 return error("Invalid record: metadata strings truncated chars");
2324 CallBack(Strings.slice(0, Size));
2325 Strings = Strings.drop_front(Size);
2326 } while (--NumStrings);
2328 return Error::success();
2331 Error MetadataLoader::MetadataLoaderImpl::parseGlobalObjectAttachment(
2332 GlobalObject &GO, ArrayRef<uint64_t> Record) {
2333 assert(Record.size() % 2 == 0);
2334 for (unsigned I = 0, E = Record.size(); I != E; I += 2) {
2335 auto K = MDKindMap.find(Record[I]);
2336 if (K == MDKindMap.end())
2337 return error("Invalid ID");
2338 MDNode *MD =
2339 dyn_cast_or_null<MDNode>(getMetadataFwdRefOrLoad(Record[I + 1]));
2340 if (!MD)
2341 return error("Invalid metadata attachment: expect fwd ref to MDNode");
2342 GO.addMetadata(K->second, *MD);
2344 return Error::success();
2347 /// Parse metadata attachments.
2348 Error MetadataLoader::MetadataLoaderImpl::parseMetadataAttachment(
2349 Function &F, ArrayRef<Instruction *> InstructionList) {
2350 if (Error Err = Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
2351 return Err;
2353 SmallVector<uint64_t, 64> Record;
2354 PlaceholderQueue Placeholders;
2356 while (true) {
2357 BitstreamEntry Entry;
2358 if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
2359 return E;
2361 switch (Entry.Kind) {
2362 case BitstreamEntry::SubBlock: // Handled for us already.
2363 case BitstreamEntry::Error:
2364 return error("Malformed block");
2365 case BitstreamEntry::EndBlock:
2366 resolveForwardRefsAndPlaceholders(Placeholders);
2367 return Error::success();
2368 case BitstreamEntry::Record:
2369 // The interesting case.
2370 break;
2373 // Read a metadata attachment record.
2374 Record.clear();
2375 ++NumMDRecordLoaded;
2376 Expected<unsigned> MaybeRecord = Stream.readRecord(Entry.ID, Record);
2377 if (!MaybeRecord)
2378 return MaybeRecord.takeError();
2379 switch (MaybeRecord.get()) {
2380 default: // Default behavior: ignore.
2381 break;
2382 case bitc::METADATA_ATTACHMENT: {
2383 unsigned RecordLength = Record.size();
2384 if (Record.empty())
2385 return error("Invalid record");
2386 if (RecordLength % 2 == 0) {
2387 // A function attachment.
2388 if (Error Err = parseGlobalObjectAttachment(F, Record))
2389 return Err;
2390 continue;
2393 // An instruction attachment.
2394 Instruction *Inst = InstructionList[Record[0]];
2395 for (unsigned i = 1; i != RecordLength; i = i + 2) {
2396 unsigned Kind = Record[i];
2397 DenseMap<unsigned, unsigned>::iterator I = MDKindMap.find(Kind);
2398 if (I == MDKindMap.end())
2399 return error("Invalid ID");
2400 if (I->second == LLVMContext::MD_tbaa && StripTBAA)
2401 continue;
2403 auto Idx = Record[i + 1];
2404 if (Idx < (MDStringRef.size() + GlobalMetadataBitPosIndex.size()) &&
2405 !MetadataList.lookup(Idx)) {
2406 // Load the attachment if it is in the lazy-loadable range and hasn't
2407 // been loaded yet.
2408 lazyLoadOneMetadata(Idx, Placeholders);
2409 resolveForwardRefsAndPlaceholders(Placeholders);
2412 Metadata *Node = MetadataList.getMetadataFwdRef(Idx);
2413 if (isa<LocalAsMetadata>(Node))
2414 // Drop the attachment. This used to be legal, but there's no
2415 // upgrade path.
2416 break;
2417 MDNode *MD = dyn_cast_or_null<MDNode>(Node);
2418 if (!MD)
2419 return error("Invalid metadata attachment");
2421 if (HasSeenOldLoopTags && I->second == LLVMContext::MD_loop)
2422 MD = upgradeInstructionLoopAttachment(*MD);
2424 if (I->second == LLVMContext::MD_tbaa) {
2425 assert(!MD->isTemporary() && "should load MDs before attachments");
2426 MD = UpgradeTBAANode(*MD);
2428 Inst->setMetadata(I->second, MD);
2430 break;
2436 /// Parse a single METADATA_KIND record, inserting result in MDKindMap.
2437 Error MetadataLoader::MetadataLoaderImpl::parseMetadataKindRecord(
2438 SmallVectorImpl<uint64_t> &Record) {
2439 if (Record.size() < 2)
2440 return error("Invalid record");
2442 unsigned Kind = Record[0];
2443 SmallString<8> Name(Record.begin() + 1, Record.end());
2445 unsigned NewKind = TheModule.getMDKindID(Name.str());
2446 if (!MDKindMap.insert(std::make_pair(Kind, NewKind)).second)
2447 return error("Conflicting METADATA_KIND records");
2448 return Error::success();
2451 /// Parse the metadata kinds out of the METADATA_KIND_BLOCK.
2452 Error MetadataLoader::MetadataLoaderImpl::parseMetadataKinds() {
2453 if (Error Err = Stream.EnterSubBlock(bitc::METADATA_KIND_BLOCK_ID))
2454 return Err;
2456 SmallVector<uint64_t, 64> Record;
2458 // Read all the records.
2459 while (true) {
2460 BitstreamEntry Entry;
2461 if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
2462 return E;
2464 switch (Entry.Kind) {
2465 case BitstreamEntry::SubBlock: // Handled for us already.
2466 case BitstreamEntry::Error:
2467 return error("Malformed block");
2468 case BitstreamEntry::EndBlock:
2469 return Error::success();
2470 case BitstreamEntry::Record:
2471 // The interesting case.
2472 break;
2475 // Read a record.
2476 Record.clear();
2477 ++NumMDRecordLoaded;
2478 Expected<unsigned> MaybeCode = Stream.readRecord(Entry.ID, Record);
2479 if (!MaybeCode)
2480 return MaybeCode.takeError();
2481 switch (MaybeCode.get()) {
2482 default: // Default behavior: ignore.
2483 break;
2484 case bitc::METADATA_KIND: {
2485 if (Error Err = parseMetadataKindRecord(Record))
2486 return Err;
2487 break;
2493 MetadataLoader &MetadataLoader::operator=(MetadataLoader &&RHS) {
2494 Pimpl = std::move(RHS.Pimpl);
2495 return *this;
2497 MetadataLoader::MetadataLoader(MetadataLoader &&RHS)
2498 : Pimpl(std::move(RHS.Pimpl)) {}
2500 MetadataLoader::~MetadataLoader() = default;
2501 MetadataLoader::MetadataLoader(BitstreamCursor &Stream, Module &TheModule,
2502 BitcodeReaderValueList &ValueList,
2503 bool IsImporting,
2504 MetadataLoaderCallbacks Callbacks)
2505 : Pimpl(std::make_unique<MetadataLoaderImpl>(
2506 Stream, TheModule, ValueList, std::move(Callbacks), IsImporting)) {}
2508 Error MetadataLoader::parseMetadata(bool ModuleLevel) {
2509 return Pimpl->parseMetadata(ModuleLevel);
2512 bool MetadataLoader::hasFwdRefs() const { return Pimpl->hasFwdRefs(); }
2514 /// Return the given metadata, creating a replaceable forward reference if
2515 /// necessary.
2516 Metadata *MetadataLoader::getMetadataFwdRefOrLoad(unsigned Idx) {
2517 return Pimpl->getMetadataFwdRefOrLoad(Idx);
2520 DISubprogram *MetadataLoader::lookupSubprogramForFunction(Function *F) {
2521 return Pimpl->lookupSubprogramForFunction(F);
2524 Error MetadataLoader::parseMetadataAttachment(
2525 Function &F, ArrayRef<Instruction *> InstructionList) {
2526 return Pimpl->parseMetadataAttachment(F, InstructionList);
2529 Error MetadataLoader::parseMetadataKinds() {
2530 return Pimpl->parseMetadataKinds();
2533 void MetadataLoader::setStripTBAA(bool StripTBAA) {
2534 return Pimpl->setStripTBAA(StripTBAA);
2537 bool MetadataLoader::isStrippingTBAA() { return Pimpl->isStrippingTBAA(); }
2539 unsigned MetadataLoader::size() const { return Pimpl->size(); }
2540 void MetadataLoader::shrinkTo(unsigned N) { return Pimpl->shrinkTo(N); }
2542 void MetadataLoader::upgradeDebugIntrinsics(Function &F) {
2543 return Pimpl->upgradeDebugIntrinsics(F);