xref: /llvm-project/llvm/lib/Bitcode/Reader/MetadataLoader.cpp (revision 67fb2686fba9abd6e607ff9a09b7018b2b8ae31b)
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 //===----------------------------------------------------------------------===//
8 
9 #include "MetadataLoader.h"
10 #include "ValueList.h"
11 
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"
48 
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;
65 }
66 
67 using namespace llvm;
68 
69 #define DEBUG_TYPE "bitcode-reader"
70 
71 STATISTIC(NumMDStringLoaded, "Number of MDStrings loaded");
72 STATISTIC(NumMDNodeTemporary, "Number of MDNode::Temporary created");
73 STATISTIC(NumMDRecordLoaded, "Number of Metadata records loaded");
74 
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."));
80 
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."));
85 
86 namespace {
87 
88 static int64_t unrotateSign(uint64_t U) { return (U & 1) ? ~(U >> 1) : U >> 1; }
89 
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;
96 
97   /// The set of indices in MetadataPtrs above of forward references that were
98   /// generated.
99   SmallDenseSet<unsigned, 1> ForwardReference;
100 
101   /// The set of indices in MetadataPtrs above of Metadata that need to be
102   /// resolved.
103   SmallDenseSet<unsigned, 1> UnresolvedNodes;
104 
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;
112 
113   LLVMContext &Context;
114 
115   /// Maximum number of valid references. Forward references exceeding the
116   /// maximum must be invalid.
117   unsigned RefsUpperBound;
118 
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)) {}
124 
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(); }
133 
134   Metadata *operator[](unsigned i) const {
135     assert(i < MetadataPtrs.size());
136     return MetadataPtrs[i];
137   }
138 
139   Metadata *lookup(unsigned I) const {
140     if (I < MetadataPtrs.size())
141       return MetadataPtrs[I];
142     return nullptr;
143   }
144 
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);
150   }
151 
152   /// Return the given metadata, creating a replaceable forward reference if
153   /// necessary.
154   Metadata *getMetadataFwdRef(unsigned Idx);
155 
156   /// Return the given metadata only if it is fully resolved.
157   ///
158   /// Gives the same result as \a lookup(), unless \a MDNode::isResolved()
159   /// would give \c false.
160   Metadata *getMetadataIfResolved(unsigned Idx);
161 
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();
169   }
170 
171   /// Upgrade a type that had an MDString reference.
172   void addTypeRef(MDString &UUID, DICompositeType &CT);
173 
174   /// Upgrade a type that had an MDString reference.
175   Metadata *upgradeTypeRef(Metadata *MaybeUUID);
176 
177   /// Upgrade a type ref array that may have MDString references.
178   Metadata *upgradeTypeRefArray(Metadata *MaybeTuple);
179 
180 private:
181   Metadata *resolveTypeRefArray(Metadata *MaybeTuple);
182 };
183 
184 void BitcodeReaderMetadataList::assignValue(Metadata *MD, unsigned Idx) {
185   if (auto *MDN = dyn_cast<MDNode>(MD))
186     if (!MDN->isResolved())
187       UnresolvedNodes.insert(Idx);
188 
189   if (Idx == size()) {
190     push_back(MD);
191     return;
192   }
193 
194   if (Idx >= size())
195     resize(Idx + 1);
196 
197   TrackingMDRef &OldMD = MetadataPtrs[Idx];
198   if (!OldMD) {
199     OldMD.reset(MD);
200     return;
201   }
202 
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);
207 }
208 
209 Metadata *BitcodeReaderMetadataList::getMetadataFwdRef(unsigned Idx) {
210   // Bail out for a clearly invalid value.
211   if (Idx >= RefsUpperBound)
212     return nullptr;
213 
214   if (Idx >= size())
215     resize(Idx + 1);
216 
217   if (Metadata *MD = MetadataPtrs[Idx])
218     return MD;
219 
220   // Track forward refs to be resolved later.
221   ForwardReference.insert(Idx);
222 
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;
228 }
229 
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;
236 }
237 
238 MDNode *BitcodeReaderMetadataList::getMDNodeFwdRefOrNull(unsigned Idx) {
239   return dyn_cast_or_null<MDNode>(getMetadataFwdRef(Idx));
240 }
241 
242 void BitcodeReaderMetadataList::tryToResolveCycles() {
243   if (!ForwardReference.empty())
244     // Still forward references... can't resolve cycles.
245     return;
246 
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();
251 
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();
257 
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);
266   }
267   OldTypeRefs.Unknown.clear();
268 
269   if (UnresolvedNodes.empty())
270     // Nothing to do.
271     return;
272 
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;
279 
280     assert(!N->isTemporary() && "Unexpected forward reference");
281     N->resolveCycles();
282   }
283 
284   // Make sure we return early again until there's another unresolved ref.
285   UnresolvedNodes.clear();
286 }
287 
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));
295 }
296 
297 Metadata *BitcodeReaderMetadataList::upgradeTypeRef(Metadata *MaybeUUID) {
298   auto *UUID = dyn_cast_or_null<MDString>(MaybeUUID);
299   if (LLVM_LIKELY(!UUID))
300     return MaybeUUID;
301 
302   if (auto *CT = OldTypeRefs.Final.lookup(UUID))
303     return CT;
304 
305   auto &Ref = OldTypeRefs.Unknown[UUID];
306   if (!Ref)
307     Ref = MDNode::getTemporary(Context, {});
308   return Ref.get();
309 }
310 
311 Metadata *BitcodeReaderMetadataList::upgradeTypeRefArray(Metadata *MaybeTuple) {
312   auto *Tuple = dyn_cast_or_null<MDTuple>(MaybeTuple);
313   if (!Tuple || Tuple->isDistinct())
314     return MaybeTuple;
315 
316   // Look through the array immediately if possible.
317   if (!Tuple->isTemporary())
318     return resolveTypeRefArray(Tuple);
319 
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();
326 }
327 
328 Metadata *BitcodeReaderMetadataList::resolveTypeRefArray(Metadata *MaybeTuple) {
329   auto *Tuple = dyn_cast_or_null<MDTuple>(MaybeTuple);
330   if (!Tuple || Tuple->isDistinct())
331     return MaybeTuple;
332 
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));
338 
339   return MDTuple::get(Context, Ops);
340 }
341 
342 namespace {
343 
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;
348 
349 public:
350   ~PlaceholderQueue() {
351     assert(empty() &&
352            "PlaceholderQueue hasn't been flushed before being destroyed");
353   }
354   bool empty() const { return PHs.empty(); }
355   DistinctMDOperandPlaceholder &getPlaceholderOp(unsigned ID);
356   void flush(BitcodeReaderMetadataList &MetadataList);
357 
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;
368       }
369       auto *N = dyn_cast_or_null<MDNode>(MD);
370       if (N && N->isTemporary())
371         Temporaries.insert(ID);
372     }
373   }
374 };
375 
376 } // end anonymous namespace
377 
378 DistinctMDOperandPlaceholder &PlaceholderQueue::getPlaceholderOp(unsigned ID) {
379   PHs.emplace_back(ID);
380   return PHs.back();
381 }
382 
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();
394   }
395 }
396 
397 } // anonymous namespace
398 
399 static Error error(const Twine &Message) {
400   return make_error<StringError>(
401       Message, make_error_code(BitcodeError::CorruptedBitcode));
402 }
403 
404 class MetadataLoader::MetadataLoaderImpl {
405   BitcodeReaderMetadataList MetadataList;
406   BitcodeReaderValueList &ValueList;
407   BitstreamCursor &Stream;
408   LLVMContext &Context;
409   Module &TheModule;
410   MetadataLoaderCallbacks Callbacks;
411 
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;
416 
417   /// Index that keeps track of MDString values.
418   std::vector<StringRef> MDStringRef;
419 
420   /// On-demand loading of a single MDString. Requires the index above to be
421   /// populated.
422   MDString *lazyLoadOneMDString(unsigned Idx);
423 
424   /// Index that keeps track of where to find a metadata record in the stream.
425   std::vector<uint64_t> GlobalMetadataBitPosIndex;
426 
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;
431 
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
438 
439   /// Load the global decl attachments, using the index built for lazy loading.
440   Expected<bool> loadGlobalDeclAttachments();
441 
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();
446 
447   /// On-demand loading of a single metadata. Requires the index above to be
448   /// populated.
449   void lazyLoadOneMetadata(unsigned Idx, PlaceholderQueue &Placeholders);
450 
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;
454 
455   /// Functions that need to be matched with subprograms when upgrading old
456   /// metadata.
457   SmallDenseMap<Function *, DISubprogram *, 16> FunctionsWithSPs;
458 
459   // Map the bitcode's custom MDKind ID to the Module's MDKind ID.
460   DenseMap<unsigned, unsigned> MDKindMap;
461 
462   bool StripTBAA = false;
463   bool HasSeenOldLoopTags = false;
464   bool NeedUpgradeToDIGlobalVariableExpression = false;
465   bool NeedDeclareExpressionUpgrade = false;
466 
467   /// Map DILocalScope to the enclosing DISubprogram, if any.
468   DenseMap<DILocalScope *, DISubprogram *> ParentSubprogram;
469 
470   /// True if metadata is being parsed for a module being ThinLTO imported.
471   bool IsImporting = false;
472 
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);
481 
482   void resolveForwardRefsAndPlaceholders(PlaceholderQueue &Placeholders);
483 
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();
492   }
493 
494   /// Upgrade old-style bare DIGlobalVariables to DIGlobalVariableExpressions.
495   void upgradeCUVariables() {
496     if (!NeedUpgradeToDIGlobalVariableExpression)
497       return;
498 
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);
510             }
511       }
512 
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);
525     }
526   }
527 
528   DISubprogram *findEnclosingSubprogram(DILocalScope *S) {
529     if (!S)
530       return nullptr;
531     if (auto *SP = ParentSubprogram[S]) {
532       return SP;
533     }
534 
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;
541     }
542     ParentSubprogram[InitialScope] = llvm::dyn_cast_or_null<DISubprogram>(S);
543 
544     return ParentSubprogram[InitialScope];
545   }
546 
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;
555 
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);
563             }
564           }
565 
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);
572               }
573             }
574 
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);
582               }
583             }
584 
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));
593             }
594 
595             // Remove entities with local scope from CU.
596             CU->replaceImportedEntities(MDTuple::get(Context, NewImports));
597           }
598         }
599       }
600     }
601 
602     ParentSubprogram.clear();
603   }
604 
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;
610 
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));
619     };
620 
621     for (auto &BB : F)
622       for (auto &I : BB) {
623         for (DbgVariableRecord &DVR : filterDbgVars(I.getDbgRecordRange())) {
624           if (DVR.isDbgDeclare())
625             UpdateDeclareIfNeeded(&DVR);
626         }
627         if (auto *DDI = dyn_cast<DbgDeclareInst>(&I))
628           UpdateDeclareIfNeeded(DDI);
629       }
630   }
631 
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;
653       }
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;
677         }
678 
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);
683 
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;
698         }
699 
700         // Continue with remaining elements.
701         SubExpr = SubExpr.slice(HistoricSize);
702       }
703       Expr = MutableArrayRef<uint64_t>(Buffer);
704       [[fallthrough]];
705     }
706     case 3:
707       // Up-to-date!
708       break;
709     }
710 
711     return Error::success();
712   }
713 
714   void upgradeDebugInfo(bool ModuleLevel) {
715     upgradeCUSubprograms();
716     upgradeCUVariables();
717     if (ModuleLevel)
718       upgradeCULocals();
719   }
720 
721   void callMDTypeCallback(Metadata **Val, unsigned TypeID);
722 
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) {}
731 
732   Error parseMetadata(bool ModuleLevel);
733 
734   bool hasFwdRefs() const { return MetadataList.hasFwdRefs(); }
735 
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);
748     }
749     return MetadataList.getMetadataFwdRef(ID);
750   }
751 
752   DISubprogram *lookupSubprogramForFunction(Function *F) {
753     return FunctionsWithSPs.lookup(F);
754   }
755 
756   bool hasSeenOldLoopTags() const { return HasSeenOldLoopTags; }
757 
758   Error parseMetadataAttachment(Function &F,
759                                 ArrayRef<Instruction *> InstructionList);
760 
761   Error parseMetadataKinds();
762 
763   void setStripTBAA(bool Value) { StripTBAA = Value; }
764   bool isStrippingTBAA() const { return StripTBAA; }
765 
766   unsigned size() const { return MetadataList.size(); }
767   void shrinkTo(unsigned N) { MetadataList.shrinkTo(N); }
768   void upgradeDebugIntrinsics(Function &F) { upgradeDeclareExpressions(F); }
769 };
770 
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);
785 
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;
792     }
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))
809           ;
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);
816         };
817         if (auto Err = parseMetadataStrings(Record, Blob, IndexNextMDString))
818           return std::move(Err);
819         break;
820       }
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))
829           ;
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);
861         }
862         break;
863       }
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();
873 
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();
881 
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();
888 
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();
897 
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);
909         }
910         break;
911       }
912       case bitc::METADATA_GLOBAL_DECL_ATTACHMENT: {
913         if (!GlobalDeclAttachmentPos)
914           GlobalDeclAttachmentPos = SavedPos;
915 #ifndef NDEBUG
916         NumGlobalDeclAttachSkipped++;
917 #endif
918         break;
919       }
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;
962       }
963       break;
964     }
965     }
966   }
967 }
968 
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);
993 
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;
1004     }
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;
1014     }
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))
1025       ;
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);
1043     }
1044   }
1045 }
1046 
1047 void MetadataLoader::MetadataLoaderImpl::callMDTypeCallback(Metadata **Val,
1048                                                             unsigned TypeID) {
1049   if (Callbacks.MDType) {
1050     (*Callbacks.MDType)(Val, TypeID, Callbacks.GetTypeByID,
1051                         Callbacks.GetContainedTypeID);
1052   }
1053 }
1054 
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");
1060 
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();
1064 
1065   if (Error Err = Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
1066     return Err;
1067 
1068   SmallVector<uint64_t, 64> Record;
1069   PlaceholderQueue Placeholders;
1070 
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());
1083 
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());
1091 
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();
1106       }
1107       return Error::success();
1108     }
1109     // Couldn't load an index, fallback to loading all the block "old-style".
1110   }
1111 
1112   unsigned NextMetadataNo = MetadataList.size();
1113 
1114   // Read all the records.
1115   while (true) {
1116     BitstreamEntry Entry;
1117     if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
1118       return E;
1119 
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;
1131     }
1132 
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();
1144   }
1145 }
1146 
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;
1154 }
1155 
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;
1165   }
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())));
1187 }
1188 
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);
1197 
1198     // If we don't have any temporary, or FwdReference, we're done!
1199     if (Temporaries.empty() && !MetadataList.hasFwdRefs())
1200       break;
1201 
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();
1207 
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);
1212   }
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();
1217 
1218   // Finally, everything is in place, we can replace the placeholders operands
1219   // with the final node they refer to.
1220   Placeholders.flush(MetadataList);
1221 }
1222 
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;
1229 
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);
1239 
1240   return nullptr;
1241 }
1242 
1243 Error MetadataLoader::MetadataLoaderImpl::parseOneMetadata(
1244     SmallVectorImpl<uint64_t> &Record, unsigned Code,
1245     PlaceholderQueue &Placeholders, StringRef Blob, unsigned &NextMetadataNo) {
1246 
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);
1263       }
1264       // Return a temporary.
1265       return MetadataList.getMetadataFwdRef(ID);
1266     }
1267     if (auto *MD = MetadataList.getMetadataIfResolved(ID))
1268       return MD;
1269     return &Placeholders.getPlaceholderOp(ID);
1270   };
1271   auto getMDOrNull = [&](unsigned ID) -> Metadata * {
1272     if (ID)
1273       return getMD(ID - 1);
1274     return nullptr;
1275   };
1276   auto getMDOrNullWithoutPlaceholders = [&](unsigned ID) -> Metadata * {
1277     if (ID)
1278       return MetadataList.getMetadataFwdRef(ID - 1);
1279     return nullptr;
1280   };
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);
1286   };
1287 
1288   // Support for old type refs.
1289   auto getDITypeRefOrNull = [&](unsigned ID) {
1290     return MetadataList.upgradeTypeRef(getMDOrNull(ID));
1291   };
1292 
1293 #define GET_OR_DISTINCT(CLASS, ARGS)                                           \
1294   (IsDistinct ? CLASS::getDistinct ARGS : CLASS::get ARGS)
1295 
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;
1305 
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();
1312 
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);
1321     }
1322     break;
1323   }
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");
1330 
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++;
1336     };
1337     if (Record.size() != 2) {
1338       dropRecord();
1339       break;
1340     }
1341 
1342     unsigned TyID = Record[0];
1343     Type *Ty = Callbacks.GetTypeByID(TyID);
1344     if (!Ty || Ty->isMetadataTy() || Ty->isVoidTy()) {
1345       dropRecord();
1346       break;
1347     }
1348 
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");
1353 
1354     MetadataList.assignValue(LocalAsMetadata::get(V), NextMetadataNo);
1355     NextMetadataNo++;
1356     break;
1357   }
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");
1362 
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);
1383     }
1384     MetadataList.assignValue(MDNode::get(Context, Elts), NextMetadataNo);
1385     NextMetadataNo++;
1386     break;
1387   }
1388   case bitc::METADATA_VALUE: {
1389     if (Record.size() != 2)
1390       return error("Invalid record");
1391 
1392     unsigned TyID = Record[0];
1393     Type *Ty = Callbacks.GetTypeByID(TyID);
1394     if (!Ty || Ty->isMetadataTy() || Ty->isVoidTy())
1395       return error("Invalid record");
1396 
1397     Value *V = getValueFwdRef(ValueList, Record[1], Ty, TyID);
1398     if (!V)
1399       return error("Invalid value reference from metadata");
1400 
1401     Metadata *MD = ValueAsMetadata::get(V);
1402     callMDTypeCallback(&MD, TyID);
1403     MetadataList.assignValue(MD, NextMetadataNo);
1404     NextMetadataNo++;
1405     break;
1406   }
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;
1420   }
1421   case bitc::METADATA_LOCATION: {
1422     if (Record.size() != 5 && Record.size() != 6)
1423       return error("Invalid record");
1424 
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;
1437   }
1438   case bitc::METADATA_GENERIC_DEBUG: {
1439     if (Record.size() < 4)
1440       return error("Invalid record");
1441 
1442     IsDistinct = Record[0];
1443     unsigned Tag = Record[1];
1444     unsigned Version = Record[2];
1445 
1446     if (Tag >= 1u << 16 || Version != 0)
1447       return error("Invalid record");
1448 
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;
1458   }
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");
1485     }
1486 
1487     MetadataList.assignValue(Val, NextMetadataNo);
1488     IsDistinct = Record[0] & 1;
1489     NextMetadataNo++;
1490     break;
1491   }
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])));
1498 
1499     MetadataList.assignValue(Val, NextMetadataNo);
1500     IsDistinct = Record[0] & 1;
1501     NextMetadataNo++;
1502     break;
1503   }
1504   case bitc::METADATA_ENUMERATOR: {
1505     if (Record.size() < 3)
1506       return error("Invalid record");
1507 
1508     IsDistinct = Record[0] & 1;
1509     bool IsUnsigned = Record[0] & 2;
1510     bool IsBigInt = Record[0] & 4;
1511     APInt Value;
1512 
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);
1519 
1520     MetadataList.assignValue(
1521         GET_OR_DISTINCT(DIEnumerator,
1522                         (Context, Value, IsUnsigned, getMDString(Record[2]))),
1523         NextMetadataNo);
1524     NextMetadataNo++;
1525     break;
1526   }
1527   case bitc::METADATA_BASIC_TYPE: {
1528     if (Record.size() < 6 || Record.size() > 8)
1529       return error("Invalid record");
1530 
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;
1536 
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;
1544   }
1545   case bitc::METADATA_STRING_TYPE: {
1546     if (Record.size() > 9 || Record.size() < 8)
1547       return error("Invalid record");
1548 
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;
1564   }
1565   case bitc::METADATA_DERIVED_TYPE: {
1566     if (Record.size() < 12 || Record.size() > 15)
1567       return error("Invalid record");
1568 
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;
1574 
1575     Metadata *Annotations = nullptr;
1576     std::optional<DIDerivedType::PtrAuthData> PtrAuthData;
1577 
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]);
1586     }
1587 
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;
1601   }
1602   case bitc::METADATA_COMPOSITE_TYPE: {
1603     if (Record.size() < 16 || Record.size() > 24)
1604       return error("Invalid record");
1605 
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.
1641     //
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.
1645     //
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]);
1675       }
1676       if (Record.size() > 20) {
1677         Rank = getMDOrNull(Record[20]);
1678       }
1679       if (Record.size() > 21) {
1680         Annotations = getMDOrNull(Record[21]);
1681       }
1682       if (Record.size() > 23) {
1683         Specification = getMDOrNull(Record[23]);
1684       }
1685     }
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);
1694 
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));
1706 
1707     MetadataList.assignValue(CT, NextMetadataNo);
1708     NextMetadataNo++;
1709     break;
1710   }
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;
1716 
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);
1722 
1723     MetadataList.assignValue(
1724         GET_OR_DISTINCT(DISubroutineType, (Context, Flags, CC, Types)),
1725         NextMetadataNo);
1726     NextMetadataNo++;
1727     break;
1728   }
1729 
1730   case bitc::METADATA_MODULE: {
1731     if (Record.size() < 5 || Record.size() > 9)
1732       return error("Invalid record");
1733 
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;
1748   }
1749 
1750   case bitc::METADATA_FILE: {
1751     if (Record.size() != 3 && Record.size() != 5 && Record.size() != 6)
1752       return error("Invalid record");
1753 
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;
1772   }
1773   case bitc::METADATA_COMPILE_UNIT: {
1774     if (Record.size() < 14 || Record.size() > 22)
1775       return error("Invalid record");
1776 
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]));
1793 
1794     MetadataList.assignValue(CU, NextMetadataNo);
1795     NextMetadataNo++;
1796 
1797     // Move the Upgrade the list of subprograms.
1798     if (Metadata *SPs = getMDOrNullWithoutPlaceholders(Record[11]))
1799       CUSubprograms.push_back({CU, SPs});
1800     break;
1801   }
1802   case bitc::METADATA_SUBPROGRAM: {
1803     if (Record.size() < 18 || Record.size() > 21)
1804       return error("Invalid record");
1805 
1806     bool HasSPFlags = Record[0] & 4;
1807 
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]);
1815     }
1816 
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);
1826 
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);
1834 
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++;
1861       }
1862       HasThisAdj = Record.size() >= 20;
1863       HasThrownTypes = Record.size() >= 21;
1864     } else {
1865       HasAnnotations = Record.size() >= 19;
1866       HasTargetFuncName = Record.size() >= 20;
1867     }
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
1894          ));
1895     MetadataList.assignValue(SP, NextMetadataNo);
1896     NextMetadataNo++;
1897 
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);
1908         }
1909     }
1910     break;
1911   }
1912   case bitc::METADATA_LEXICAL_BLOCK: {
1913     if (Record.size() != 5)
1914       return error("Invalid record");
1915 
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;
1924   }
1925   case bitc::METADATA_LEXICAL_BLOCK_FILE: {
1926     if (Record.size() != 4)
1927       return error("Invalid record");
1928 
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;
1937   }
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;
1948   }
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");
1958 
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;
1967   }
1968   case bitc::METADATA_MACRO: {
1969     if (Record.size() != 5)
1970       return error("Invalid record");
1971 
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;
1980   }
1981   case bitc::METADATA_MACRO_FILE: {
1982     if (Record.size() != 5)
1983       return error("Invalid record");
1984 
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;
1993   }
1994   case bitc::METADATA_TEMPLATE_TYPE: {
1995     if (Record.size() < 3 || Record.size() > 4)
1996       return error("Invalid record");
1997 
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;
2008   }
2009   case bitc::METADATA_TEMPLATE_VALUE: {
2010     if (Record.size() < 5 || Record.size() > 6)
2011       return error("Invalid record");
2012 
2013     IsDistinct = Record[0];
2014 
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;
2026   }
2027   case bitc::METADATA_GLOBAL_VAR: {
2028     if (Record.size() < 11 || Record.size() > 13)
2029       return error("Invalid record");
2030 
2031     IsDistinct = Record[0] & 1;
2032     unsigned Version = Record[0] >> 1;
2033 
2034     if (Version == 2) {
2035       Metadata *Annotations = nullptr;
2036       if (Record.size() > 12)
2037         Annotations = getMDOrNull(Record[12]);
2038 
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);
2048 
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);
2061 
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];
2073       }
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;
2085         }
2086       }
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));
2093 
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);
2100 
2101       auto *MDNode = Expr ? cast<Metadata>(DGVE) : cast<Metadata>(DGV);
2102       MetadataList.assignValue(MDNode, NextMetadataNo);
2103       NextMetadataNo++;
2104     } else
2105       return error("Invalid record");
2106 
2107     break;
2108   }
2109   case bitc::METADATA_ASSIGN_ID: {
2110     if (Record.size() != 1)
2111       return error("Invalid DIAssignID record.");
2112 
2113     IsDistinct = Record[0] & 1;
2114     if (!IsDistinct)
2115       return error("Invalid DIAssignID record. Must be distinct");
2116 
2117     MetadataList.assignValue(DIAssignID::getDistinct(Context), NextMetadataNo);
2118     NextMetadataNo++;
2119     break;
2120   }
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");
2125 
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]);
2141     }
2142 
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;
2153   }
2154   case bitc::METADATA_LABEL: {
2155     if (Record.size() != 5)
2156       return error("Invalid record");
2157 
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;
2166   }
2167   case bitc::METADATA_EXPRESSION: {
2168     if (Record.size() < 1)
2169       return error("Invalid record");
2170 
2171     IsDistinct = Record[0] & 1;
2172     uint64_t Version = Record[0] >> 1;
2173     auto Elts = MutableArrayRef<uint64_t>(Record).slice(1);
2174 
2175     SmallVector<uint64_t, 6> Buffer;
2176     if (Error Err = upgradeDIExpression(Version, Elts, Buffer))
2177       return Err;
2178 
2179     MetadataList.assignValue(GET_OR_DISTINCT(DIExpression, (Context, Elts)),
2180                              NextMetadataNo);
2181     NextMetadataNo++;
2182     break;
2183   }
2184   case bitc::METADATA_GLOBAL_VAR_EXPR: {
2185     if (Record.size() != 3)
2186       return error("Invalid record");
2187 
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;
2198   }
2199   case bitc::METADATA_OBJC_PROPERTY: {
2200     if (Record.size() != 8)
2201       return error("Invalid record");
2202 
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;
2213   }
2214   case bitc::METADATA_IMPORTED_ENTITY: {
2215     if (Record.size() < 6 || Record.size() > 8)
2216       return error("Invalid DIImportedEntity record");
2217 
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;
2231   }
2232   case bitc::METADATA_STRING_OLD: {
2233     std::string String(Record.begin(), Record.end());
2234 
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;
2242   }
2243   case bitc::METADATA_STRINGS: {
2244     auto CreateNextMDString = [&](StringRef Str) {
2245       ++NumMDStringLoaded;
2246       MetadataList.assignValue(MDString::get(Context, Str), NextMetadataNo);
2247       NextMetadataNo++;
2248     };
2249     if (Error Err = parseMetadataStrings(Record, Blob, CreateNextMDString))
2250       return Err;
2251     break;
2252   }
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;
2264   }
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;
2271   }
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));
2283     }
2284 
2285     MetadataList.assignValue(DIArgList::get(Context, Elts), NextMetadataNo);
2286     NextMetadataNo++;
2287     break;
2288   }
2289   }
2290   return Error::success();
2291 #undef GET_OR_DISTINCT
2292 }
2293 
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");
2302 
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");
2309 
2310   StringRef Lengths = Blob.slice(0, StringsOffset);
2311   SimpleBitstreamCursor R(Lengths);
2312 
2313   StringRef Strings = Blob.drop_front(StringsOffset);
2314   do {
2315     if (R.AtEndOfStream())
2316       return error("Invalid record: metadata strings bad length");
2317 
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");
2323 
2324     CallBack(Strings.slice(0, Size));
2325     Strings = Strings.drop_front(Size);
2326   } while (--NumStrings);
2327 
2328   return Error::success();
2329 }
2330 
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);
2343   }
2344   return Error::success();
2345 }
2346 
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;
2352 
2353   SmallVector<uint64_t, 64> Record;
2354   PlaceholderQueue Placeholders;
2355 
2356   while (true) {
2357     BitstreamEntry Entry;
2358     if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
2359       return E;
2360 
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;
2371     }
2372 
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;
2391       }
2392 
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;
2402 
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);
2410         }
2411 
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");
2420 
2421         if (HasSeenOldLoopTags && I->second == LLVMContext::MD_loop)
2422           MD = upgradeInstructionLoopAttachment(*MD);
2423 
2424         if (I->second == LLVMContext::MD_tbaa) {
2425           assert(!MD->isTemporary() && "should load MDs before attachments");
2426           MD = UpgradeTBAANode(*MD);
2427         }
2428         Inst->setMetadata(I->second, MD);
2429       }
2430       break;
2431     }
2432     }
2433   }
2434 }
2435 
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");
2441 
2442   unsigned Kind = Record[0];
2443   SmallString<8> Name(Record.begin() + 1, Record.end());
2444 
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();
2449 }
2450 
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;
2455 
2456   SmallVector<uint64_t, 64> Record;
2457 
2458   // Read all the records.
2459   while (true) {
2460     BitstreamEntry Entry;
2461     if (Error E = Stream.advanceSkippingSubblocks().moveInto(Entry))
2462       return E;
2463 
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;
2473     }
2474 
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;
2488     }
2489     }
2490   }
2491 }
2492 
2493 MetadataLoader &MetadataLoader::operator=(MetadataLoader &&RHS) {
2494   Pimpl = std::move(RHS.Pimpl);
2495   return *this;
2496 }
2497 MetadataLoader::MetadataLoader(MetadataLoader &&RHS)
2498     : Pimpl(std::move(RHS.Pimpl)) {}
2499 
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)) {}
2507 
2508 Error MetadataLoader::parseMetadata(bool ModuleLevel) {
2509   return Pimpl->parseMetadata(ModuleLevel);
2510 }
2511 
2512 bool MetadataLoader::hasFwdRefs() const { return Pimpl->hasFwdRefs(); }
2513 
2514 /// Return the given metadata, creating a replaceable forward reference if
2515 /// necessary.
2516 Metadata *MetadataLoader::getMetadataFwdRefOrLoad(unsigned Idx) {
2517   return Pimpl->getMetadataFwdRefOrLoad(Idx);
2518 }
2519 
2520 DISubprogram *MetadataLoader::lookupSubprogramForFunction(Function *F) {
2521   return Pimpl->lookupSubprogramForFunction(F);
2522 }
2523 
2524 Error MetadataLoader::parseMetadataAttachment(
2525     Function &F, ArrayRef<Instruction *> InstructionList) {
2526   return Pimpl->parseMetadataAttachment(F, InstructionList);
2527 }
2528 
2529 Error MetadataLoader::parseMetadataKinds() {
2530   return Pimpl->parseMetadataKinds();
2531 }
2532 
2533 void MetadataLoader::setStripTBAA(bool StripTBAA) {
2534   return Pimpl->setStripTBAA(StripTBAA);
2535 }
2536 
2537 bool MetadataLoader::isStrippingTBAA() { return Pimpl->isStrippingTBAA(); }
2538 
2539 unsigned MetadataLoader::size() const { return Pimpl->size(); }
2540 void MetadataLoader::shrinkTo(unsigned N) { return Pimpl->shrinkTo(N); }
2541 
2542 void MetadataLoader::upgradeDebugIntrinsics(Function &F) {
2543   return Pimpl->upgradeDebugIntrinsics(F);
2544 }
2545