1 //===- MicrosoftDemangle.cpp ----------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines a demangler for MSVC-style mangled symbols.
11 //
12 // This file has no dependencies on the rest of LLVM so that it can be
13 // easily reused in other programs such as libcxxabi.
14 //
15 //===----------------------------------------------------------------------===//
16 
17 #include "MicrosoftDemangleNodes.h"
18 #include "llvm/Demangle/Demangle.h"
19 
20 #include "llvm/Demangle/Compiler.h"
21 #include "llvm/Demangle/StringView.h"
22 #include "llvm/Demangle/Utility.h"
23 
24 #include <array>
25 #include <cctype>
26 #include <cstdio>
27 #include <tuple>
28 
29 using namespace llvm;
30 using namespace ms_demangle;
31 
32 static bool startsWithDigit(StringView S) {
33   return !S.empty() && std::isdigit(S.front());
34 }
35 
36 enum class QualifierMangleMode { Drop, Mangle, Result };
37 
38 struct NodeList {
39   Node *N = nullptr;
40   NodeList *Next = nullptr;
41 };
42 
43 enum class FunctionIdentifierCodeGroup { Basic, Under, DoubleUnder };
44 
45 enum NameBackrefBehavior : uint8_t {
46   NBB_None = 0,          // don't save any names as backrefs.
47   NBB_Template = 1 << 0, // save template instanations.
48   NBB_Simple = 1 << 1,   // save simple names.
49 };
50 
51 static bool isMemberPointer(StringView MangledName) {
52   switch (MangledName.popFront()) {
53   case '$':
54     // This is probably an rvalue reference (e.g. $$Q), and you cannot have an
55     // rvalue reference to a member.
56     return false;
57   case 'A':
58     // 'A' indicates a reference, and you cannot have a reference to a member
59     // function or member.
60     return false;
61   case 'P':
62   case 'Q':
63   case 'R':
64   case 'S':
65     // These 4 values indicate some kind of pointer, but we still don't know
66     // what.
67     break;
68   default:
69     assert(false && "Ty is not a pointer type!");
70   }
71 
72   // If it starts with a number, then 6 indicates a non-member function
73   // pointer, and 8 indicates a member function pointer.
74   if (startsWithDigit(MangledName)) {
75     assert(MangledName[0] == '6' || MangledName[0] == '8');
76     return (MangledName[0] == '8');
77   }
78 
79   // Remove ext qualifiers since those can appear on either type and are
80   // therefore not indicative.
81   MangledName.consumeFront('E'); // 64-bit
82   MangledName.consumeFront('I'); // restrict
83   MangledName.consumeFront('F'); // unaligned
84 
85   assert(!MangledName.empty());
86 
87   // The next value should be either ABCD (non-member) or QRST (member).
88   switch (MangledName.front()) {
89   case 'A':
90   case 'B':
91   case 'C':
92   case 'D':
93     return false;
94   case 'Q':
95   case 'R':
96   case 'S':
97   case 'T':
98     return true;
99   default:
100     assert(false);
101   }
102   return false;
103 }
104 
105 static SpecialIntrinsicKind
106 consumeSpecialIntrinsicKind(StringView &MangledName) {
107   if (MangledName.consumeFront("?_7"))
108     return SpecialIntrinsicKind::Vftable;
109   if (MangledName.consumeFront("?_8"))
110     return SpecialIntrinsicKind::Vbtable;
111   if (MangledName.consumeFront("?_9"))
112     return SpecialIntrinsicKind::VcallThunk;
113   if (MangledName.consumeFront("?_A"))
114     return SpecialIntrinsicKind::Typeof;
115   if (MangledName.consumeFront("?_B"))
116     return SpecialIntrinsicKind::LocalStaticGuard;
117   if (MangledName.consumeFront("?_C"))
118     return SpecialIntrinsicKind::StringLiteralSymbol;
119   if (MangledName.consumeFront("?_P"))
120     return SpecialIntrinsicKind::UdtReturning;
121   if (MangledName.consumeFront("?_R0"))
122     return SpecialIntrinsicKind::RttiTypeDescriptor;
123   if (MangledName.consumeFront("?_R1"))
124     return SpecialIntrinsicKind::RttiBaseClassDescriptor;
125   if (MangledName.consumeFront("?_R2"))
126     return SpecialIntrinsicKind::RttiBaseClassArray;
127   if (MangledName.consumeFront("?_R3"))
128     return SpecialIntrinsicKind::RttiClassHierarchyDescriptor;
129   if (MangledName.consumeFront("?_R4"))
130     return SpecialIntrinsicKind::RttiCompleteObjLocator;
131   if (MangledName.consumeFront("?_S"))
132     return SpecialIntrinsicKind::LocalVftable;
133   if (MangledName.consumeFront("?__E"))
134     return SpecialIntrinsicKind::DynamicInitializer;
135   if (MangledName.consumeFront("?__F"))
136     return SpecialIntrinsicKind::DynamicAtexitDestructor;
137   if (MangledName.consumeFront("?__J"))
138     return SpecialIntrinsicKind::LocalStaticThreadGuard;
139   return SpecialIntrinsicKind::None;
140 }
141 
142 static bool startsWithLocalScopePattern(StringView S) {
143   if (!S.consumeFront('?'))
144     return false;
145   if (S.size() < 2)
146     return false;
147 
148   size_t End = S.find('?');
149   if (End == StringView::npos)
150     return false;
151   StringView Candidate = S.substr(0, End);
152   if (Candidate.empty())
153     return false;
154 
155   // \?[0-9]\?
156   // ?@? is the discriminator 0.
157   if (Candidate.size() == 1)
158     return Candidate[0] == '@' || (Candidate[0] >= '0' && Candidate[0] <= '9');
159 
160   // If it's not 0-9, then it's an encoded number terminated with an @
161   if (Candidate.back() != '@')
162     return false;
163   Candidate = Candidate.dropBack();
164 
165   // An encoded number starts with B-P and all subsequent digits are in A-P.
166   // Note that the reason the first digit cannot be A is two fold.  First, it
167   // would create an ambiguity with ?A which delimits the beginning of an
168   // anonymous namespace.  Second, A represents 0, and you don't start a multi
169   // digit number with a leading 0.  Presumably the anonymous namespace
170   // ambiguity is also why single digit encoded numbers use 0-9 rather than A-J.
171   if (Candidate[0] < 'B' || Candidate[0] > 'P')
172     return false;
173   Candidate = Candidate.dropFront();
174   while (!Candidate.empty()) {
175     if (Candidate[0] < 'A' || Candidate[0] > 'P')
176       return false;
177     Candidate = Candidate.dropFront();
178   }
179 
180   return true;
181 }
182 
183 static bool isTagType(StringView S) {
184   switch (S.front()) {
185   case 'T': // union
186   case 'U': // struct
187   case 'V': // class
188   case 'W': // enum
189     return true;
190   }
191   return false;
192 }
193 
194 static bool isCustomType(StringView S) { return S[0] == '?'; }
195 
196 static bool isPointerType(StringView S) {
197   if (S.startsWith("$$Q")) // foo &&
198     return true;
199 
200   switch (S.front()) {
201   case 'A': // foo &
202   case 'P': // foo *
203   case 'Q': // foo *const
204   case 'R': // foo *volatile
205   case 'S': // foo *const volatile
206     return true;
207   }
208   return false;
209 }
210 
211 static bool isArrayType(StringView S) { return S[0] == 'Y'; }
212 
213 static bool isFunctionType(StringView S) {
214   return S.startsWith("$$A8@@") || S.startsWith("$$A6");
215 }
216 
217 static FunctionRefQualifier
218 demangleFunctionRefQualifier(StringView &MangledName) {
219   if (MangledName.consumeFront('G'))
220     return FunctionRefQualifier::Reference;
221   else if (MangledName.consumeFront('H'))
222     return FunctionRefQualifier::RValueReference;
223   return FunctionRefQualifier::None;
224 }
225 
226 static std::pair<Qualifiers, PointerAffinity>
227 demanglePointerCVQualifiers(StringView &MangledName) {
228   if (MangledName.consumeFront("$$Q"))
229     return std::make_pair(Q_None, PointerAffinity::RValueReference);
230 
231   switch (MangledName.popFront()) {
232   case 'A':
233     return std::make_pair(Q_None, PointerAffinity::Reference);
234   case 'P':
235     return std::make_pair(Q_None, PointerAffinity::Pointer);
236   case 'Q':
237     return std::make_pair(Q_Const, PointerAffinity::Pointer);
238   case 'R':
239     return std::make_pair(Q_Volatile, PointerAffinity::Pointer);
240   case 'S':
241     return std::make_pair(Qualifiers(Q_Const | Q_Volatile),
242                           PointerAffinity::Pointer);
243   default:
244     assert(false && "Ty is not a pointer type!");
245   }
246   return std::make_pair(Q_None, PointerAffinity::Pointer);
247 }
248 
249 namespace {
250 
251 struct BackrefContext {
252   static constexpr size_t Max = 10;
253 
254   TypeNode *FunctionParams[Max];
255   size_t FunctionParamCount = 0;
256 
257   // The first 10 BackReferences in a mangled name can be back-referenced by
258   // special name @[0-9]. This is a storage for the first 10 BackReferences.
259   NamedIdentifierNode *Names[Max];
260   size_t NamesCount = 0;
261 };
262 
263 // Demangler class takes the main role in demangling symbols.
264 // It has a set of functions to parse mangled symbols into Type instances.
265 // It also has a set of functions to cnovert Type instances to strings.
266 class Demangler {
267 public:
268   Demangler() = default;
269   virtual ~Demangler() = default;
270 
271   // You are supposed to call parse() first and then check if error is true.  If
272   // it is false, call output() to write the formatted name to the given stream.
273   SymbolNode *parse(StringView &MangledName);
274 
275   // True if an error occurred.
276   bool Error = false;
277 
278   void dumpBackReferences();
279 
280 private:
281   SymbolNode *demangleEncodedSymbol(StringView &MangledName,
282                                     QualifiedNameNode *QN);
283 
284   VariableSymbolNode *demangleVariableEncoding(StringView &MangledName,
285                                                StorageClass SC);
286   FunctionSymbolNode *demangleFunctionEncoding(StringView &MangledName);
287 
288   Qualifiers demanglePointerExtQualifiers(StringView &MangledName);
289 
290   // Parser functions. This is a recursive-descent parser.
291   TypeNode *demangleType(StringView &MangledName, QualifierMangleMode QMM);
292   PrimitiveTypeNode *demanglePrimitiveType(StringView &MangledName);
293   CustomTypeNode *demangleCustomType(StringView &MangledName);
294   TagTypeNode *demangleClassType(StringView &MangledName);
295   PointerTypeNode *demanglePointerType(StringView &MangledName);
296   PointerTypeNode *demangleMemberPointerType(StringView &MangledName);
297   FunctionSignatureNode *demangleFunctionType(StringView &MangledName,
298                                               bool HasThisQuals);
299 
300   ArrayTypeNode *demangleArrayType(StringView &MangledName);
301 
302   NodeArrayNode *demangleTemplateParameterList(StringView &MangledName);
303   NodeArrayNode *demangleFunctionParameterList(StringView &MangledName);
304 
305   std::pair<uint64_t, bool> demangleNumber(StringView &MangledName);
306   uint64_t demangleUnsigned(StringView &MangledName);
307   int64_t demangleSigned(StringView &MangledName);
308 
309   void memorizeString(StringView s);
310   void memorizeIdentifier(IdentifierNode *Identifier);
311 
312   /// Allocate a copy of \p Borrowed into memory that we own.
313   StringView copyString(StringView Borrowed);
314 
315   QualifiedNameNode *demangleFullyQualifiedTypeName(StringView &MangledName);
316   QualifiedNameNode *demangleFullyQualifiedSymbolName(StringView &MangledName);
317 
318   IdentifierNode *demangleUnqualifiedTypeName(StringView &MangledName,
319                                               bool Memorize);
320   IdentifierNode *demangleUnqualifiedSymbolName(StringView &MangledName,
321                                                 NameBackrefBehavior NBB);
322 
323   QualifiedNameNode *demangleNameScopeChain(StringView &MangledName,
324                                             IdentifierNode *UnqualifiedName);
325   IdentifierNode *demangleNameScopePiece(StringView &MangledName);
326 
327   NamedIdentifierNode *demangleBackRefName(StringView &MangledName);
328   IdentifierNode *demangleTemplateInstantiationName(StringView &MangledName,
329                                                     NameBackrefBehavior NBB);
330   IdentifierNode *demangleFunctionIdentifierCode(StringView &MangledName);
331   IdentifierNode *
332   demangleFunctionIdentifierCode(StringView &MangledName,
333                                  FunctionIdentifierCodeGroup Group);
334   StructorIdentifierNode *demangleStructorIdentifier(StringView &MangledName,
335                                                      bool IsDestructor);
336   ConversionOperatorIdentifierNode *
337   demangleConversionOperatorIdentifier(StringView &MangledName);
338   LiteralOperatorIdentifierNode *
339   demangleLiteralOperatorIdentifier(StringView &MangledName);
340 
341   SymbolNode *demangleSpecialIntrinsic(StringView &MangledName);
342   SpecialTableSymbolNode *
343   demangleSpecialTableSymbolNode(StringView &MangledName,
344                                  SpecialIntrinsicKind SIK);
345   LocalStaticGuardVariableNode *
346   demangleLocalStaticGuard(StringView &MangledName);
347   VariableSymbolNode *demangleUntypedVariable(ArenaAllocator &Arena,
348                                               StringView &MangledName,
349                                               StringView VariableName);
350   VariableSymbolNode *
351   demangleRttiBaseClassDescriptorNode(ArenaAllocator &Arena,
352                                       StringView &MangledName);
353   FunctionSymbolNode *demangleInitFiniStub(StringView &MangledName,
354                                            bool IsDestructor);
355 
356   NamedIdentifierNode *demangleSimpleName(StringView &MangledName,
357                                           bool Memorize);
358   NamedIdentifierNode *demangleAnonymousNamespaceName(StringView &MangledName);
359   NamedIdentifierNode *demangleLocallyScopedNamePiece(StringView &MangledName);
360   EncodedStringLiteralNode *demangleStringLiteral(StringView &MangledName);
361   FunctionSymbolNode *demangleVcallThunkNode(StringView &MangledName);
362 
363   StringView demangleSimpleString(StringView &MangledName, bool Memorize);
364 
365   FuncClass demangleFunctionClass(StringView &MangledName);
366   CallingConv demangleCallingConvention(StringView &MangledName);
367   StorageClass demangleVariableStorageClass(StringView &MangledName);
368   void demangleThrowSpecification(StringView &MangledName);
369   wchar_t demangleWcharLiteral(StringView &MangledName);
370   uint8_t demangleCharLiteral(StringView &MangledName);
371 
372   std::pair<Qualifiers, bool> demangleQualifiers(StringView &MangledName);
373 
374   // Memory allocator.
375   ArenaAllocator Arena;
376 
377   // A single type uses one global back-ref table for all function params.
378   // This means back-refs can even go "into" other types.  Examples:
379   //
380   //  // Second int* is a back-ref to first.
381   //  void foo(int *, int*);
382   //
383   //  // Second int* is not a back-ref to first (first is not a function param).
384   //  int* foo(int*);
385   //
386   //  // Second int* is a back-ref to first (ALL function types share the same
387   //  // back-ref map.
388   //  using F = void(*)(int*);
389   //  F G(int *);
390   BackrefContext Backrefs;
391 };
392 } // namespace
393 
394 StringView Demangler::copyString(StringView Borrowed) {
395   char *Stable = Arena.allocUnalignedBuffer(Borrowed.size() + 1);
396   std::strcpy(Stable, Borrowed.begin());
397 
398   return {Stable, Borrowed.size()};
399 }
400 
401 SpecialTableSymbolNode *
402 Demangler::demangleSpecialTableSymbolNode(StringView &MangledName,
403                                           SpecialIntrinsicKind K) {
404   NamedIdentifierNode *NI = Arena.alloc<NamedIdentifierNode>();
405   switch (K) {
406   case SpecialIntrinsicKind::Vftable:
407     NI->Name = "`vftable'";
408     break;
409   case SpecialIntrinsicKind::Vbtable:
410     NI->Name = "`vbtable'";
411     break;
412   case SpecialIntrinsicKind::LocalVftable:
413     NI->Name = "`local vftable'";
414     break;
415   case SpecialIntrinsicKind::RttiCompleteObjLocator:
416     NI->Name = "`RTTI Complete Object Locator'";
417     break;
418   default:
419     LLVM_BUILTIN_UNREACHABLE;
420   }
421   QualifiedNameNode *QN = demangleNameScopeChain(MangledName, NI);
422   SpecialTableSymbolNode *STSN = Arena.alloc<SpecialTableSymbolNode>();
423   STSN->Name = QN;
424   bool IsMember = false;
425   char Front = MangledName.popFront();
426   if (Front != '6' && Front != '7') {
427     Error = true;
428     return nullptr;
429   }
430 
431   std::tie(STSN->Quals, IsMember) = demangleQualifiers(MangledName);
432   if (!MangledName.consumeFront('@'))
433     STSN->TargetName = demangleFullyQualifiedTypeName(MangledName);
434   return STSN;
435 }
436 
437 LocalStaticGuardVariableNode *
438 Demangler::demangleLocalStaticGuard(StringView &MangledName) {
439   LocalStaticGuardIdentifierNode *LSGI =
440       Arena.alloc<LocalStaticGuardIdentifierNode>();
441   QualifiedNameNode *QN = demangleNameScopeChain(MangledName, LSGI);
442   LocalStaticGuardVariableNode *LSGVN =
443       Arena.alloc<LocalStaticGuardVariableNode>();
444   LSGVN->Name = QN;
445 
446   if (MangledName.consumeFront("4IA"))
447     LSGVN->IsVisible = false;
448   else if (MangledName.consumeFront("5"))
449     LSGVN->IsVisible = true;
450   else {
451     Error = true;
452     return nullptr;
453   }
454 
455   if (!MangledName.empty())
456     LSGI->ScopeIndex = demangleUnsigned(MangledName);
457   return LSGVN;
458 }
459 
460 static NamedIdentifierNode *synthesizeNamedIdentifier(ArenaAllocator &Arena,
461                                                       StringView Name) {
462   NamedIdentifierNode *Id = Arena.alloc<NamedIdentifierNode>();
463   Id->Name = Name;
464   return Id;
465 }
466 
467 static QualifiedNameNode *synthesizeQualifiedName(ArenaAllocator &Arena,
468                                                   IdentifierNode *Identifier) {
469   QualifiedNameNode *QN = Arena.alloc<QualifiedNameNode>();
470   QN->Components = Arena.alloc<NodeArrayNode>();
471   QN->Components->Count = 1;
472   QN->Components->Nodes = Arena.allocArray<Node *>(1);
473   QN->Components->Nodes[0] = Identifier;
474   return QN;
475 }
476 
477 static QualifiedNameNode *synthesizeQualifiedName(ArenaAllocator &Arena,
478                                                   StringView Name) {
479   NamedIdentifierNode *Id = synthesizeNamedIdentifier(Arena, Name);
480   return synthesizeQualifiedName(Arena, Id);
481 }
482 
483 static VariableSymbolNode *synthesizeVariable(ArenaAllocator &Arena,
484                                               TypeNode *Type,
485                                               StringView VariableName) {
486   VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
487   VSN->Type = Type;
488   VSN->Name = synthesizeQualifiedName(Arena, VariableName);
489   return VSN;
490 }
491 
492 VariableSymbolNode *Demangler::demangleUntypedVariable(
493     ArenaAllocator &Arena, StringView &MangledName, StringView VariableName) {
494   NamedIdentifierNode *NI = synthesizeNamedIdentifier(Arena, VariableName);
495   QualifiedNameNode *QN = demangleNameScopeChain(MangledName, NI);
496   VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
497   VSN->Name = QN;
498   if (MangledName.consumeFront("8"))
499     return VSN;
500 
501   Error = true;
502   return nullptr;
503 }
504 
505 VariableSymbolNode *
506 Demangler::demangleRttiBaseClassDescriptorNode(ArenaAllocator &Arena,
507                                                StringView &MangledName) {
508   RttiBaseClassDescriptorNode *RBCDN =
509       Arena.alloc<RttiBaseClassDescriptorNode>();
510   RBCDN->NVOffset = demangleUnsigned(MangledName);
511   RBCDN->VBPtrOffset = demangleSigned(MangledName);
512   RBCDN->VBTableOffset = demangleUnsigned(MangledName);
513   RBCDN->Flags = demangleUnsigned(MangledName);
514   if (Error)
515     return nullptr;
516 
517   VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
518   VSN->Name = demangleNameScopeChain(MangledName, RBCDN);
519   MangledName.consumeFront('8');
520   return VSN;
521 }
522 
523 FunctionSymbolNode *Demangler::demangleInitFiniStub(StringView &MangledName,
524                                                     bool IsDestructor) {
525   DynamicStructorIdentifierNode *DSIN =
526       Arena.alloc<DynamicStructorIdentifierNode>();
527   DSIN->IsDestructor = IsDestructor;
528 
529   bool IsKnownStaticDataMember = false;
530   if (MangledName.consumeFront('?'))
531     IsKnownStaticDataMember = true;
532 
533   QualifiedNameNode *QN = demangleFullyQualifiedSymbolName(MangledName);
534 
535   SymbolNode *Symbol = demangleEncodedSymbol(MangledName, QN);
536   FunctionSymbolNode *FSN = nullptr;
537   Symbol->Name = QN;
538 
539   if (Symbol->kind() == NodeKind::VariableSymbol) {
540     DSIN->Variable = static_cast<VariableSymbolNode *>(Symbol);
541 
542     // Older versions of clang mangled this type of symbol incorrectly.  They
543     // would omit the leading ? and they would only emit a single @ at the end.
544     // The correct mangling is a leading ? and 2 trailing @ signs.  Handle
545     // both cases.
546     int AtCount = IsKnownStaticDataMember ? 2 : 1;
547     for (int I = 0; I < AtCount; ++I) {
548       if (MangledName.consumeFront('@'))
549         continue;
550       Error = true;
551       return nullptr;
552     }
553 
554     FSN = demangleFunctionEncoding(MangledName);
555     FSN->Name = synthesizeQualifiedName(Arena, DSIN);
556   } else {
557     if (IsKnownStaticDataMember) {
558       // This was supposed to be a static data member, but we got a function.
559       Error = true;
560       return nullptr;
561     }
562 
563     FSN = static_cast<FunctionSymbolNode *>(Symbol);
564     DSIN->Name = Symbol->Name;
565     FSN->Name = synthesizeQualifiedName(Arena, DSIN);
566   }
567 
568   return FSN;
569 }
570 
571 SymbolNode *Demangler::demangleSpecialIntrinsic(StringView &MangledName) {
572   SpecialIntrinsicKind SIK = consumeSpecialIntrinsicKind(MangledName);
573   if (SIK == SpecialIntrinsicKind::None)
574     return nullptr;
575 
576   switch (SIK) {
577   case SpecialIntrinsicKind::StringLiteralSymbol:
578     return demangleStringLiteral(MangledName);
579   case SpecialIntrinsicKind::Vftable:
580   case SpecialIntrinsicKind::Vbtable:
581   case SpecialIntrinsicKind::LocalVftable:
582   case SpecialIntrinsicKind::RttiCompleteObjLocator:
583     return demangleSpecialTableSymbolNode(MangledName, SIK);
584   case SpecialIntrinsicKind::VcallThunk:
585     return demangleVcallThunkNode(MangledName);
586   case SpecialIntrinsicKind::LocalStaticGuard:
587     return demangleLocalStaticGuard(MangledName);
588   case SpecialIntrinsicKind::RttiTypeDescriptor: {
589     TypeNode *T = demangleType(MangledName, QualifierMangleMode::Result);
590     if (Error)
591       break;
592     if (!MangledName.consumeFront("@8"))
593       break;
594     if (!MangledName.empty())
595       break;
596     return synthesizeVariable(Arena, T, "`RTTI Type Descriptor'");
597   }
598   case SpecialIntrinsicKind::RttiBaseClassArray:
599     return demangleUntypedVariable(Arena, MangledName,
600                                    "`RTTI Base Class Array'");
601   case SpecialIntrinsicKind::RttiClassHierarchyDescriptor:
602     return demangleUntypedVariable(Arena, MangledName,
603                                    "`RTTI Class Hierarchy Descriptor'");
604   case SpecialIntrinsicKind::RttiBaseClassDescriptor:
605     return demangleRttiBaseClassDescriptorNode(Arena, MangledName);
606   case SpecialIntrinsicKind::DynamicInitializer:
607     return demangleInitFiniStub(MangledName, false);
608   case SpecialIntrinsicKind::DynamicAtexitDestructor:
609     return demangleInitFiniStub(MangledName, true);
610   default:
611     break;
612   }
613   Error = true;
614   return nullptr;
615 }
616 
617 IdentifierNode *
618 Demangler::demangleFunctionIdentifierCode(StringView &MangledName) {
619   assert(MangledName.startsWith('?'));
620   MangledName = MangledName.dropFront();
621 
622   if (MangledName.consumeFront("__"))
623     return demangleFunctionIdentifierCode(
624         MangledName, FunctionIdentifierCodeGroup::DoubleUnder);
625   else if (MangledName.consumeFront("_"))
626     return demangleFunctionIdentifierCode(MangledName,
627                                           FunctionIdentifierCodeGroup::Under);
628   return demangleFunctionIdentifierCode(MangledName,
629                                         FunctionIdentifierCodeGroup::Basic);
630 }
631 
632 StructorIdentifierNode *
633 Demangler::demangleStructorIdentifier(StringView &MangledName,
634                                       bool IsDestructor) {
635   StructorIdentifierNode *N = Arena.alloc<StructorIdentifierNode>();
636   N->IsDestructor = IsDestructor;
637   return N;
638 }
639 
640 ConversionOperatorIdentifierNode *
641 Demangler::demangleConversionOperatorIdentifier(StringView &MangledName) {
642   ConversionOperatorIdentifierNode *N =
643       Arena.alloc<ConversionOperatorIdentifierNode>();
644   return N;
645 }
646 
647 LiteralOperatorIdentifierNode *
648 Demangler::demangleLiteralOperatorIdentifier(StringView &MangledName) {
649   LiteralOperatorIdentifierNode *N =
650       Arena.alloc<LiteralOperatorIdentifierNode>();
651   N->Name = demangleSimpleString(MangledName, false);
652   return N;
653 }
654 
655 IntrinsicFunctionKind
656 translateIntrinsicFunctionCode(char CH, FunctionIdentifierCodeGroup Group) {
657   // Not all ? identifiers are intrinsics *functions*.  This function only maps
658   // operator codes for the special functions, all others are handled elsewhere,
659   // hence the IFK::None entries in the table.
660   using IFK = IntrinsicFunctionKind;
661   static IFK Basic[36] = {
662       IFK::None,             // ?0 # Foo::Foo()
663       IFK::None,             // ?1 # Foo::~Foo()
664       IFK::New,              // ?2 # operator new
665       IFK::Delete,           // ?3 # operator delete
666       IFK::Assign,           // ?4 # operator=
667       IFK::RightShift,       // ?5 # operator>>
668       IFK::LeftShift,        // ?6 # operator<<
669       IFK::LogicalNot,       // ?7 # operator!
670       IFK::Equals,           // ?8 # operator==
671       IFK::NotEquals,        // ?9 # operator!=
672       IFK::ArraySubscript,   // ?A # operator[]
673       IFK::None,             // ?B # Foo::operator <type>()
674       IFK::Pointer,          // ?C # operator->
675       IFK::Dereference,      // ?D # operator*
676       IFK::Increment,        // ?E # operator++
677       IFK::Decrement,        // ?F # operator--
678       IFK::Minus,            // ?G # operator-
679       IFK::Plus,             // ?H # operator+
680       IFK::BitwiseAnd,       // ?I # operator&
681       IFK::MemberPointer,    // ?J # operator->*
682       IFK::Divide,           // ?K # operator/
683       IFK::Modulus,          // ?L # operator%
684       IFK::LessThan,         // ?M operator<
685       IFK::LessThanEqual,    // ?N operator<=
686       IFK::GreaterThan,      // ?O operator>
687       IFK::GreaterThanEqual, // ?P operator>=
688       IFK::Comma,            // ?Q operator,
689       IFK::Parens,           // ?R operator()
690       IFK::BitwiseNot,       // ?S operator~
691       IFK::BitwiseXor,       // ?T operator^
692       IFK::BitwiseOr,        // ?U operator|
693       IFK::LogicalAnd,       // ?V operator&&
694       IFK::LogicalOr,        // ?W operator||
695       IFK::TimesEqual,       // ?X operator*=
696       IFK::PlusEqual,        // ?Y operator+=
697       IFK::MinusEqual,       // ?Z operator-=
698   };
699   static IFK Under[36] = {
700       IFK::DivEqual,           // ?_0 operator/=
701       IFK::ModEqual,           // ?_1 operator%=
702       IFK::RshEqual,           // ?_2 operator>>=
703       IFK::LshEqual,           // ?_3 operator<<=
704       IFK::BitwiseAndEqual,    // ?_4 operator&=
705       IFK::BitwiseOrEqual,     // ?_5 operator|=
706       IFK::BitwiseXorEqual,    // ?_6 operator^=
707       IFK::None,               // ?_7 # vftable
708       IFK::None,               // ?_8 # vbtable
709       IFK::None,               // ?_9 # vcall
710       IFK::None,               // ?_A # typeof
711       IFK::None,               // ?_B # local static guard
712       IFK::None,               // ?_C # string literal
713       IFK::VbaseDtor,          // ?_D # vbase destructor
714       IFK::VecDelDtor,         // ?_E # vector deleting destructor
715       IFK::DefaultCtorClosure, // ?_F # default constructor closure
716       IFK::ScalarDelDtor,      // ?_G # scalar deleting destructor
717       IFK::VecCtorIter,        // ?_H # vector constructor iterator
718       IFK::VecDtorIter,        // ?_I # vector destructor iterator
719       IFK::VecVbaseCtorIter,   // ?_J # vector vbase constructor iterator
720       IFK::VdispMap,           // ?_K # virtual displacement map
721       IFK::EHVecCtorIter,      // ?_L # eh vector constructor iterator
722       IFK::EHVecDtorIter,      // ?_M # eh vector destructor iterator
723       IFK::EHVecVbaseCtorIter, // ?_N # eh vector vbase constructor iterator
724       IFK::CopyCtorClosure,    // ?_O # copy constructor closure
725       IFK::None,               // ?_P<name> # udt returning <name>
726       IFK::None,               // ?_Q # <unknown>
727       IFK::None,               // ?_R0 - ?_R4 # RTTI Codes
728       IFK::None,               // ?_S # local vftable
729       IFK::LocalVftableCtorClosure, // ?_T # local vftable constructor closure
730       IFK::ArrayNew,                // ?_U operator new[]
731       IFK::ArrayDelete,             // ?_V operator delete[]
732       IFK::None,                    // ?_W <unused>
733       IFK::None,                    // ?_X <unused>
734       IFK::None,                    // ?_Y <unused>
735       IFK::None,                    // ?_Z <unused>
736   };
737   static IFK DoubleUnder[36] = {
738       IFK::None,                       // ?__0 <unused>
739       IFK::None,                       // ?__1 <unused>
740       IFK::None,                       // ?__2 <unused>
741       IFK::None,                       // ?__3 <unused>
742       IFK::None,                       // ?__4 <unused>
743       IFK::None,                       // ?__5 <unused>
744       IFK::None,                       // ?__6 <unused>
745       IFK::None,                       // ?__7 <unused>
746       IFK::None,                       // ?__8 <unused>
747       IFK::None,                       // ?__9 <unused>
748       IFK::ManVectorCtorIter,          // ?__A managed vector ctor iterator
749       IFK::ManVectorDtorIter,          // ?__B managed vector dtor iterator
750       IFK::EHVectorCopyCtorIter,       // ?__C EH vector copy ctor iterator
751       IFK::EHVectorVbaseCopyCtorIter,  // ?__D EH vector vbase copy ctor iter
752       IFK::None,                       // ?__E dynamic initializer for `T'
753       IFK::None,                       // ?__F dynamic atexit destructor for `T'
754       IFK::VectorCopyCtorIter,         // ?__G vector copy constructor iter
755       IFK::VectorVbaseCopyCtorIter,    // ?__H vector vbase copy ctor iter
756       IFK::ManVectorVbaseCopyCtorIter, // ?__I managed vector vbase copy ctor
757                                        // iter
758       IFK::None,                       // ?__J local static thread guard
759       IFK::None,                       // ?__K operator ""_name
760       IFK::CoAwait,                    // ?__L co_await
761       IFK::None,                       // ?__M <unused>
762       IFK::None,                       // ?__N <unused>
763       IFK::None,                       // ?__O <unused>
764       IFK::None,                       // ?__P <unused>
765       IFK::None,                       // ?__Q <unused>
766       IFK::None,                       // ?__R <unused>
767       IFK::None,                       // ?__S <unused>
768       IFK::None,                       // ?__T <unused>
769       IFK::None,                       // ?__U <unused>
770       IFK::None,                       // ?__V <unused>
771       IFK::None,                       // ?__W <unused>
772       IFK::None,                       // ?__X <unused>
773       IFK::None,                       // ?__Y <unused>
774       IFK::None,                       // ?__Z <unused>
775   };
776 
777   int Index = (CH >= '0' && CH <= '9') ? (CH - '0') : (CH - 'A' + 10);
778   switch (Group) {
779   case FunctionIdentifierCodeGroup::Basic:
780     return Basic[Index];
781   case FunctionIdentifierCodeGroup::Under:
782     return Under[Index];
783   case FunctionIdentifierCodeGroup::DoubleUnder:
784     return DoubleUnder[Index];
785   }
786   LLVM_BUILTIN_UNREACHABLE;
787 }
788 
789 IdentifierNode *
790 Demangler::demangleFunctionIdentifierCode(StringView &MangledName,
791                                           FunctionIdentifierCodeGroup Group) {
792   switch (Group) {
793   case FunctionIdentifierCodeGroup::Basic:
794     switch (char CH = MangledName.popFront()) {
795     case '0':
796     case '1':
797       return demangleStructorIdentifier(MangledName, CH == '1');
798     case 'B':
799       return demangleConversionOperatorIdentifier(MangledName);
800     default:
801       return Arena.alloc<IntrinsicFunctionIdentifierNode>(
802           translateIntrinsicFunctionCode(CH, Group));
803     }
804     break;
805   case FunctionIdentifierCodeGroup::Under:
806     return Arena.alloc<IntrinsicFunctionIdentifierNode>(
807         translateIntrinsicFunctionCode(MangledName.popFront(), Group));
808   case FunctionIdentifierCodeGroup::DoubleUnder:
809     switch (char CH = MangledName.popFront()) {
810     case 'K':
811       return demangleLiteralOperatorIdentifier(MangledName);
812     default:
813       return Arena.alloc<IntrinsicFunctionIdentifierNode>(
814           translateIntrinsicFunctionCode(CH, Group));
815     }
816   }
817   // No Mangling Yet:      Spaceship,                    // operator<=>
818 
819   return nullptr;
820 }
821 
822 SymbolNode *Demangler::demangleEncodedSymbol(StringView &MangledName,
823                                              QualifiedNameNode *Name) {
824   // Read a variable.
825   switch (MangledName.front()) {
826   case '0':
827   case '1':
828   case '2':
829   case '3':
830   case '4': {
831     StorageClass SC = demangleVariableStorageClass(MangledName);
832     return demangleVariableEncoding(MangledName, SC);
833   }
834   case '8':
835     return nullptr;
836   }
837   FunctionSymbolNode *FSN = demangleFunctionEncoding(MangledName);
838 
839   IdentifierNode *UQN = Name->getUnqualifiedIdentifier();
840   if (UQN->kind() == NodeKind::ConversionOperatorIdentifier) {
841     ConversionOperatorIdentifierNode *COIN =
842         static_cast<ConversionOperatorIdentifierNode *>(UQN);
843     COIN->TargetType = FSN->Signature->ReturnType;
844   }
845   return FSN;
846 }
847 
848 // Parser entry point.
849 SymbolNode *Demangler::parse(StringView &MangledName) {
850   // We can't demangle MD5 names, just output them as-is.
851   // Also, MSVC-style mangled symbols must start with '?'.
852   if (MangledName.startsWith("??@")) {
853     // This is an MD5 mangled name.  We can't demangle it, just return the
854     // mangled name.
855     SymbolNode *S = Arena.alloc<SymbolNode>(NodeKind::Md5Symbol);
856     S->Name = synthesizeQualifiedName(Arena, MangledName);
857     return S;
858   }
859 
860   if (!MangledName.startsWith('?')) {
861     Error = true;
862     return nullptr;
863   }
864 
865   MangledName.consumeFront('?');
866 
867   // ?$ is a template instantiation, but all other names that start with ? are
868   // operators / special names.
869   if (SymbolNode *SI = demangleSpecialIntrinsic(MangledName))
870     return SI;
871 
872   // What follows is a main symbol name. This may include namespaces or class
873   // back references.
874   QualifiedNameNode *QN = demangleFullyQualifiedSymbolName(MangledName);
875   if (Error)
876     return nullptr;
877 
878   SymbolNode *Symbol = demangleEncodedSymbol(MangledName, QN);
879   if (Symbol) {
880     Symbol->Name = QN;
881   }
882 
883   if (Error)
884     return nullptr;
885 
886   return Symbol;
887 }
888 
889 // <type-encoding> ::= <storage-class> <variable-type>
890 // <storage-class> ::= 0  # private static member
891 //                 ::= 1  # protected static member
892 //                 ::= 2  # public static member
893 //                 ::= 3  # global
894 //                 ::= 4  # static local
895 
896 VariableSymbolNode *Demangler::demangleVariableEncoding(StringView &MangledName,
897                                                         StorageClass SC) {
898   VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
899 
900   VSN->Type = demangleType(MangledName, QualifierMangleMode::Drop);
901   VSN->SC = SC;
902 
903   // <variable-type> ::= <type> <cvr-qualifiers>
904   //                 ::= <type> <pointee-cvr-qualifiers> # pointers, references
905   switch (VSN->Type->kind()) {
906   case NodeKind::PointerType: {
907     PointerTypeNode *PTN = static_cast<PointerTypeNode *>(VSN->Type);
908 
909     Qualifiers ExtraChildQuals = Q_None;
910     PTN->Quals = Qualifiers(VSN->Type->Quals |
911                             demanglePointerExtQualifiers(MangledName));
912 
913     bool IsMember = false;
914     std::tie(ExtraChildQuals, IsMember) = demangleQualifiers(MangledName);
915 
916     if (PTN->ClassParent) {
917       QualifiedNameNode *BackRefName =
918           demangleFullyQualifiedTypeName(MangledName);
919       (void)BackRefName;
920     }
921     PTN->Pointee->Quals = Qualifiers(PTN->Pointee->Quals | ExtraChildQuals);
922 
923     break;
924   }
925   default:
926     VSN->Type->Quals = demangleQualifiers(MangledName).first;
927     break;
928   }
929 
930   return VSN;
931 }
932 
933 // Sometimes numbers are encoded in mangled symbols. For example,
934 // "int (*x)[20]" is a valid C type (x is a pointer to an array of
935 // length 20), so we need some way to embed numbers as part of symbols.
936 // This function parses it.
937 //
938 // <number>               ::= [?] <non-negative integer>
939 //
940 // <non-negative integer> ::= <decimal digit> # when 1 <= Number <= 10
941 //                        ::= <hex digit>+ @  # when Numbrer == 0 or >= 10
942 //
943 // <hex-digit>            ::= [A-P]           # A = 0, B = 1, ...
944 std::pair<uint64_t, bool> Demangler::demangleNumber(StringView &MangledName) {
945   bool IsNegative = MangledName.consumeFront('?');
946 
947   if (startsWithDigit(MangledName)) {
948     uint64_t Ret = MangledName[0] - '0' + 1;
949     MangledName = MangledName.dropFront(1);
950     return {Ret, IsNegative};
951   }
952 
953   uint64_t Ret = 0;
954   for (size_t i = 0; i < MangledName.size(); ++i) {
955     char C = MangledName[i];
956     if (C == '@') {
957       MangledName = MangledName.dropFront(i + 1);
958       return {Ret, IsNegative};
959     }
960     if ('A' <= C && C <= 'P') {
961       Ret = (Ret << 4) + (C - 'A');
962       continue;
963     }
964     break;
965   }
966 
967   Error = true;
968   return {0ULL, false};
969 }
970 
971 uint64_t Demangler::demangleUnsigned(StringView &MangledName) {
972   bool IsNegative = false;
973   uint64_t Number = 0;
974   std::tie(Number, IsNegative) = demangleNumber(MangledName);
975   if (IsNegative)
976     Error = true;
977   return Number;
978 }
979 
980 int64_t Demangler::demangleSigned(StringView &MangledName) {
981   bool IsNegative = false;
982   uint64_t Number = 0;
983   std::tie(Number, IsNegative) = demangleNumber(MangledName);
984   if (Number > INT64_MAX)
985     Error = true;
986   int64_t I = static_cast<int64_t>(Number);
987   return IsNegative ? -I : I;
988 }
989 
990 // First 10 strings can be referenced by special BackReferences ?0, ?1, ..., ?9.
991 // Memorize it.
992 void Demangler::memorizeString(StringView S) {
993   if (Backrefs.NamesCount >= BackrefContext::Max)
994     return;
995   for (size_t i = 0; i < Backrefs.NamesCount; ++i)
996     if (S == Backrefs.Names[i]->Name)
997       return;
998   NamedIdentifierNode *N = Arena.alloc<NamedIdentifierNode>();
999   N->Name = S;
1000   Backrefs.Names[Backrefs.NamesCount++] = N;
1001 }
1002 
1003 NamedIdentifierNode *Demangler::demangleBackRefName(StringView &MangledName) {
1004   assert(startsWithDigit(MangledName));
1005 
1006   size_t I = MangledName[0] - '0';
1007   if (I >= Backrefs.NamesCount) {
1008     Error = true;
1009     return nullptr;
1010   }
1011 
1012   MangledName = MangledName.dropFront();
1013   return Backrefs.Names[I];
1014 }
1015 
1016 void Demangler::memorizeIdentifier(IdentifierNode *Identifier) {
1017   // Render this class template name into a string buffer so that we can
1018   // memorize it for the purpose of back-referencing.
1019   OutputStream OS = OutputStream::create(nullptr, nullptr, 1024);
1020   Identifier->output(OS, OF_Default);
1021   OS << '\0';
1022   char *Name = OS.getBuffer();
1023 
1024   StringView Owned = copyString(Name);
1025   memorizeString(Owned);
1026   std::free(Name);
1027 }
1028 
1029 IdentifierNode *
1030 Demangler::demangleTemplateInstantiationName(StringView &MangledName,
1031                                              NameBackrefBehavior NBB) {
1032   assert(MangledName.startsWith("?$"));
1033   MangledName.consumeFront("?$");
1034 
1035   BackrefContext OuterContext;
1036   std::swap(OuterContext, Backrefs);
1037 
1038   IdentifierNode *Identifier =
1039       demangleUnqualifiedSymbolName(MangledName, NBB_Simple);
1040   if (!Error)
1041     Identifier->TemplateParams = demangleTemplateParameterList(MangledName);
1042 
1043   std::swap(OuterContext, Backrefs);
1044   if (Error)
1045     return nullptr;
1046 
1047   if (NBB & NBB_Template)
1048     memorizeIdentifier(Identifier);
1049 
1050   return Identifier;
1051 }
1052 
1053 NamedIdentifierNode *Demangler::demangleSimpleName(StringView &MangledName,
1054                                                    bool Memorize) {
1055   StringView S = demangleSimpleString(MangledName, Memorize);
1056   if (Error)
1057     return nullptr;
1058 
1059   NamedIdentifierNode *Name = Arena.alloc<NamedIdentifierNode>();
1060   Name->Name = S;
1061   return Name;
1062 }
1063 
1064 static bool isRebasedHexDigit(char C) { return (C >= 'A' && C <= 'P'); }
1065 
1066 static uint8_t rebasedHexDigitToNumber(char C) {
1067   assert(isRebasedHexDigit(C));
1068   return (C <= 'J') ? (C - 'A') : (10 + C - 'K');
1069 }
1070 
1071 uint8_t Demangler::demangleCharLiteral(StringView &MangledName) {
1072   if (!MangledName.startsWith('?'))
1073     return MangledName.popFront();
1074 
1075   MangledName = MangledName.dropFront();
1076   if (MangledName.empty())
1077     goto CharLiteralError;
1078 
1079   if (MangledName.consumeFront('$')) {
1080     // Two hex digits
1081     if (MangledName.size() < 2)
1082       goto CharLiteralError;
1083     StringView Nibbles = MangledName.substr(0, 2);
1084     if (!isRebasedHexDigit(Nibbles[0]) || !isRebasedHexDigit(Nibbles[1]))
1085       goto CharLiteralError;
1086     // Don't append the null terminator.
1087     uint8_t C1 = rebasedHexDigitToNumber(Nibbles[0]);
1088     uint8_t C2 = rebasedHexDigitToNumber(Nibbles[1]);
1089     MangledName = MangledName.dropFront(2);
1090     return (C1 << 4) | C2;
1091   }
1092 
1093   if (startsWithDigit(MangledName)) {
1094     const char *Lookup = ",/\\:. \n\t'-";
1095     char C = Lookup[MangledName[0] - '0'];
1096     MangledName = MangledName.dropFront();
1097     return C;
1098   }
1099 
1100   if (MangledName[0] >= 'a' && MangledName[0] <= 'z') {
1101     char Lookup[26] = {'\xE1', '\xE2', '\xE3', '\xE4', '\xE5', '\xE6', '\xE7',
1102                        '\xE8', '\xE9', '\xEA', '\xEB', '\xEC', '\xED', '\xEE',
1103                        '\xEF', '\xF0', '\xF1', '\xF2', '\xF3', '\xF4', '\xF5',
1104                        '\xF6', '\xF7', '\xF8', '\xF9', '\xFA'};
1105     char C = Lookup[MangledName[0] - 'a'];
1106     MangledName = MangledName.dropFront();
1107     return C;
1108   }
1109 
1110   if (MangledName[0] >= 'A' && MangledName[0] <= 'Z') {
1111     char Lookup[26] = {'\xC1', '\xC2', '\xC3', '\xC4', '\xC5', '\xC6', '\xC7',
1112                        '\xC8', '\xC9', '\xCA', '\xCB', '\xCC', '\xCD', '\xCE',
1113                        '\xCF', '\xD0', '\xD1', '\xD2', '\xD3', '\xD4', '\xD5',
1114                        '\xD6', '\xD7', '\xD8', '\xD9', '\xDA'};
1115     char C = Lookup[MangledName[0] - 'A'];
1116     MangledName = MangledName.dropFront();
1117     return C;
1118   }
1119 
1120 CharLiteralError:
1121   Error = true;
1122   return '\0';
1123 }
1124 
1125 wchar_t Demangler::demangleWcharLiteral(StringView &MangledName) {
1126   uint8_t C1, C2;
1127 
1128   C1 = demangleCharLiteral(MangledName);
1129   if (Error)
1130     goto WCharLiteralError;
1131   C2 = demangleCharLiteral(MangledName);
1132   if (Error)
1133     goto WCharLiteralError;
1134 
1135   return ((wchar_t)C1 << 8) | (wchar_t)C2;
1136 
1137 WCharLiteralError:
1138   Error = true;
1139   return L'\0';
1140 }
1141 
1142 static void writeHexDigit(char *Buffer, uint8_t Digit) {
1143   assert(Digit <= 15);
1144   *Buffer = (Digit < 10) ? ('0' + Digit) : ('A' + Digit - 10);
1145 }
1146 
1147 static void outputHex(OutputStream &OS, unsigned C) {
1148   if (C == 0) {
1149     OS << "\\x00";
1150     return;
1151   }
1152   // It's easier to do the math if we can work from right to left, but we need
1153   // to print the numbers from left to right.  So render this into a temporary
1154   // buffer first, then output the temporary buffer.  Each byte is of the form
1155   // \xAB, which means that each byte needs 4 characters.  Since there are at
1156   // most 4 bytes, we need a 4*4+1 = 17 character temporary buffer.
1157   char TempBuffer[17];
1158 
1159   ::memset(TempBuffer, 0, sizeof(TempBuffer));
1160   constexpr int MaxPos = 15;
1161 
1162   int Pos = MaxPos - 1;
1163   while (C != 0) {
1164     for (int I = 0; I < 2; ++I) {
1165       writeHexDigit(&TempBuffer[Pos--], C % 16);
1166       C /= 16;
1167     }
1168     TempBuffer[Pos--] = 'x';
1169     TempBuffer[Pos--] = '\\';
1170     assert(Pos >= 0);
1171   }
1172   OS << StringView(&TempBuffer[Pos + 1]);
1173 }
1174 
1175 static void outputEscapedChar(OutputStream &OS, unsigned C) {
1176   switch (C) {
1177   case '\'': // single quote
1178     OS << "\\\'";
1179     return;
1180   case '\"': // double quote
1181     OS << "\\\"";
1182     return;
1183   case '\\': // backslash
1184     OS << "\\\\";
1185     return;
1186   case '\a': // bell
1187     OS << "\\a";
1188     return;
1189   case '\b': // backspace
1190     OS << "\\b";
1191     return;
1192   case '\f': // form feed
1193     OS << "\\f";
1194     return;
1195   case '\n': // new line
1196     OS << "\\n";
1197     return;
1198   case '\r': // carriage return
1199     OS << "\\r";
1200     return;
1201   case '\t': // tab
1202     OS << "\\t";
1203     return;
1204   case '\v': // vertical tab
1205     OS << "\\v";
1206     return;
1207   default:
1208     break;
1209   }
1210 
1211   if (C > 0x1F && C < 0x7F) {
1212     // Standard ascii char.
1213     OS << (char)C;
1214     return;
1215   }
1216 
1217   outputHex(OS, C);
1218 }
1219 
1220 unsigned countTrailingNullBytes(const uint8_t *StringBytes, int Length) {
1221   const uint8_t *End = StringBytes + Length - 1;
1222   unsigned Count = 0;
1223   while (Length > 0 && *End == 0) {
1224     --Length;
1225     --End;
1226     ++Count;
1227   }
1228   return Count;
1229 }
1230 
1231 unsigned countEmbeddedNulls(const uint8_t *StringBytes, unsigned Length) {
1232   unsigned Result = 0;
1233   for (unsigned I = 0; I < Length; ++I) {
1234     if (*StringBytes++ == 0)
1235       ++Result;
1236   }
1237   return Result;
1238 }
1239 
1240 unsigned guessCharByteSize(const uint8_t *StringBytes, unsigned NumChars,
1241                            unsigned NumBytes) {
1242   assert(NumBytes > 0);
1243 
1244   // If the number of bytes is odd, this is guaranteed to be a char string.
1245   if (NumBytes % 2 == 1)
1246     return 1;
1247 
1248   // All strings can encode at most 32 bytes of data.  If it's less than that,
1249   // then we encoded the entire string.  In this case we check for a 1-byte,
1250   // 2-byte, or 4-byte null terminator.
1251   if (NumBytes < 32) {
1252     unsigned TrailingNulls = countTrailingNullBytes(StringBytes, NumChars);
1253     if (TrailingNulls >= 4)
1254       return 4;
1255     if (TrailingNulls >= 2)
1256       return 2;
1257     return 1;
1258   }
1259 
1260   // The whole string was not able to be encoded.  Try to look at embedded null
1261   // terminators to guess.  The heuristic is that we count all embedded null
1262   // terminators.  If more than 2/3 are null, it's a char32.  If more than 1/3
1263   // are null, it's a char16.  Otherwise it's a char8.  This obviously isn't
1264   // perfect and is biased towards languages that have ascii alphabets, but this
1265   // was always going to be best effort since the encoding is lossy.
1266   unsigned Nulls = countEmbeddedNulls(StringBytes, NumChars);
1267   if (Nulls >= 2 * NumChars / 3)
1268     return 4;
1269   if (Nulls >= NumChars / 3)
1270     return 2;
1271   return 1;
1272 }
1273 
1274 static unsigned decodeMultiByteChar(const uint8_t *StringBytes,
1275                                     unsigned CharIndex, unsigned CharBytes) {
1276   assert(CharBytes == 1 || CharBytes == 2 || CharBytes == 4);
1277   unsigned Offset = CharIndex * CharBytes;
1278   unsigned Result = 0;
1279   StringBytes = StringBytes + Offset;
1280   for (unsigned I = 0; I < CharBytes; ++I) {
1281     unsigned C = static_cast<unsigned>(StringBytes[I]);
1282     Result |= C << (8 * I);
1283   }
1284   return Result;
1285 }
1286 
1287 FunctionSymbolNode *Demangler::demangleVcallThunkNode(StringView &MangledName) {
1288   FunctionSymbolNode *FSN = Arena.alloc<FunctionSymbolNode>();
1289   VcallThunkIdentifierNode *VTIN = Arena.alloc<VcallThunkIdentifierNode>();
1290   FSN->Signature = Arena.alloc<ThunkSignatureNode>();
1291   FSN->Signature->FunctionClass = FC_NoParameterList;
1292 
1293   FSN->Name = demangleNameScopeChain(MangledName, VTIN);
1294   if (!Error)
1295     Error = !MangledName.consumeFront("$B");
1296   if (!Error)
1297     VTIN->OffsetInVTable = demangleUnsigned(MangledName);
1298   if (!Error)
1299     Error = !MangledName.consumeFront('A');
1300   if (!Error)
1301     FSN->Signature->CallConvention = demangleCallingConvention(MangledName);
1302   return (Error) ? nullptr : FSN;
1303 }
1304 
1305 EncodedStringLiteralNode *
1306 Demangler::demangleStringLiteral(StringView &MangledName) {
1307   // This function uses goto, so declare all variables up front.
1308   OutputStream OS;
1309   StringView CRC;
1310   uint64_t StringByteSize;
1311   bool IsWcharT = false;
1312   bool IsNegative = false;
1313   size_t CrcEndPos = 0;
1314   char *ResultBuffer = nullptr;
1315 
1316   EncodedStringLiteralNode *Result = Arena.alloc<EncodedStringLiteralNode>();
1317 
1318   // Prefix indicating the beginning of a string literal
1319   if (!MangledName.consumeFront("@_"))
1320     goto StringLiteralError;
1321   if (MangledName.empty())
1322     goto StringLiteralError;
1323 
1324   // Char Type (regular or wchar_t)
1325   switch (MangledName.popFront()) {
1326   case '1':
1327     IsWcharT = true;
1328     LLVM_FALLTHROUGH;
1329   case '0':
1330     break;
1331   default:
1332     goto StringLiteralError;
1333   }
1334 
1335   // Encoded Length
1336   std::tie(StringByteSize, IsNegative) = demangleNumber(MangledName);
1337   if (Error || IsNegative)
1338     goto StringLiteralError;
1339 
1340   // CRC 32 (always 8 characters plus a terminator)
1341   CrcEndPos = MangledName.find('@');
1342   if (CrcEndPos == StringView::npos)
1343     goto StringLiteralError;
1344   CRC = MangledName.substr(0, CrcEndPos);
1345   MangledName = MangledName.dropFront(CrcEndPos + 1);
1346   if (MangledName.empty())
1347     goto StringLiteralError;
1348 
1349   OS = OutputStream::create(nullptr, nullptr, 1024);
1350   if (IsWcharT) {
1351     Result->Char = CharKind::Wchar;
1352     if (StringByteSize > 64)
1353       Result->IsTruncated = true;
1354 
1355     while (!MangledName.consumeFront('@')) {
1356       assert(StringByteSize >= 2);
1357       wchar_t W = demangleWcharLiteral(MangledName);
1358       if (StringByteSize != 2 || Result->IsTruncated)
1359         outputEscapedChar(OS, W);
1360       StringByteSize -= 2;
1361       if (Error)
1362         goto StringLiteralError;
1363     }
1364   } else {
1365     // The max byte length is actually 32, but some compilers mangled strings
1366     // incorrectly, so we have to assume it can go higher.
1367     constexpr unsigned MaxStringByteLength = 32 * 4;
1368     uint8_t StringBytes[MaxStringByteLength];
1369 
1370     unsigned BytesDecoded = 0;
1371     while (!MangledName.consumeFront('@')) {
1372       assert(StringByteSize >= 1);
1373       StringBytes[BytesDecoded++] = demangleCharLiteral(MangledName);
1374     }
1375 
1376     if (StringByteSize > BytesDecoded)
1377       Result->IsTruncated = true;
1378 
1379     unsigned CharBytes =
1380         guessCharByteSize(StringBytes, BytesDecoded, StringByteSize);
1381     assert(StringByteSize % CharBytes == 0);
1382     switch (CharBytes) {
1383     case 1:
1384       Result->Char = CharKind::Char;
1385       break;
1386     case 2:
1387       Result->Char = CharKind::Char16;
1388       break;
1389     case 4:
1390       Result->Char = CharKind::Char32;
1391       break;
1392     default:
1393       LLVM_BUILTIN_UNREACHABLE;
1394     }
1395     const unsigned NumChars = BytesDecoded / CharBytes;
1396     for (unsigned CharIndex = 0; CharIndex < NumChars; ++CharIndex) {
1397       unsigned NextChar =
1398           decodeMultiByteChar(StringBytes, CharIndex, CharBytes);
1399       if (CharIndex + 1 < NumChars || Result->IsTruncated)
1400         outputEscapedChar(OS, NextChar);
1401     }
1402   }
1403 
1404   OS << '\0';
1405   ResultBuffer = OS.getBuffer();
1406   Result->DecodedString = copyString(ResultBuffer);
1407   std::free(ResultBuffer);
1408   return Result;
1409 
1410 StringLiteralError:
1411   Error = true;
1412   return nullptr;
1413 }
1414 
1415 StringView Demangler::demangleSimpleString(StringView &MangledName,
1416                                            bool Memorize) {
1417   StringView S;
1418   for (size_t i = 0; i < MangledName.size(); ++i) {
1419     if (MangledName[i] != '@')
1420       continue;
1421     S = MangledName.substr(0, i);
1422     MangledName = MangledName.dropFront(i + 1);
1423 
1424     if (Memorize)
1425       memorizeString(S);
1426     return S;
1427   }
1428 
1429   Error = true;
1430   return {};
1431 }
1432 
1433 NamedIdentifierNode *
1434 Demangler::demangleAnonymousNamespaceName(StringView &MangledName) {
1435   assert(MangledName.startsWith("?A"));
1436   MangledName.consumeFront("?A");
1437 
1438   NamedIdentifierNode *Node = Arena.alloc<NamedIdentifierNode>();
1439   Node->Name = "`anonymous namespace'";
1440   size_t EndPos = MangledName.find('@');
1441   if (EndPos == StringView::npos) {
1442     Error = true;
1443     return nullptr;
1444   }
1445   StringView NamespaceKey = MangledName.substr(0, EndPos);
1446   memorizeString(NamespaceKey);
1447   MangledName = MangledName.substr(EndPos + 1);
1448   return Node;
1449 }
1450 
1451 NamedIdentifierNode *
1452 Demangler::demangleLocallyScopedNamePiece(StringView &MangledName) {
1453   assert(startsWithLocalScopePattern(MangledName));
1454 
1455   NamedIdentifierNode *Identifier = Arena.alloc<NamedIdentifierNode>();
1456   MangledName.consumeFront('?');
1457   auto Number = demangleNumber(MangledName);
1458   assert(!Number.second);
1459 
1460   // One ? to terminate the number
1461   MangledName.consumeFront('?');
1462 
1463   assert(!Error);
1464   Node *Scope = parse(MangledName);
1465   if (Error)
1466     return nullptr;
1467 
1468   // Render the parent symbol's name into a buffer.
1469   OutputStream OS = OutputStream::create(nullptr, nullptr, 1024);
1470   OS << '`';
1471   Scope->output(OS, OF_Default);
1472   OS << '\'';
1473   OS << "::`" << Number.first << "'";
1474   OS << '\0';
1475   char *Result = OS.getBuffer();
1476   Identifier->Name = copyString(Result);
1477   std::free(Result);
1478   return Identifier;
1479 }
1480 
1481 // Parses a type name in the form of A@B@C@@ which represents C::B::A.
1482 QualifiedNameNode *
1483 Demangler::demangleFullyQualifiedTypeName(StringView &MangledName) {
1484   IdentifierNode *Identifier = demangleUnqualifiedTypeName(MangledName, true);
1485   if (Error)
1486     return nullptr;
1487   assert(Identifier);
1488 
1489   QualifiedNameNode *QN = demangleNameScopeChain(MangledName, Identifier);
1490   if (Error)
1491     return nullptr;
1492   assert(QN);
1493   return QN;
1494 }
1495 
1496 // Parses a symbol name in the form of A@B@C@@ which represents C::B::A.
1497 // Symbol names have slightly different rules regarding what can appear
1498 // so we separate out the implementations for flexibility.
1499 QualifiedNameNode *
1500 Demangler::demangleFullyQualifiedSymbolName(StringView &MangledName) {
1501   // This is the final component of a symbol name (i.e. the leftmost component
1502   // of a mangled name.  Since the only possible template instantiation that
1503   // can appear in this context is a function template, and since those are
1504   // not saved for the purposes of name backreferences, only backref simple
1505   // names.
1506   IdentifierNode *Identifier =
1507       demangleUnqualifiedSymbolName(MangledName, NBB_Simple);
1508   if (Error)
1509     return nullptr;
1510 
1511   QualifiedNameNode *QN = demangleNameScopeChain(MangledName, Identifier);
1512   if (Error)
1513     return nullptr;
1514 
1515   if (Identifier->kind() == NodeKind::StructorIdentifier) {
1516     StructorIdentifierNode *SIN =
1517         static_cast<StructorIdentifierNode *>(Identifier);
1518     assert(QN->Components->Count >= 2);
1519     Node *ClassNode = QN->Components->Nodes[QN->Components->Count - 2];
1520     SIN->Class = static_cast<IdentifierNode *>(ClassNode);
1521   }
1522   assert(QN);
1523   return QN;
1524 }
1525 
1526 IdentifierNode *Demangler::demangleUnqualifiedTypeName(StringView &MangledName,
1527                                                        bool Memorize) {
1528   // An inner-most name can be a back-reference, because a fully-qualified name
1529   // (e.g. Scope + Inner) can contain other fully qualified names inside of
1530   // them (for example template parameters), and these nested parameters can
1531   // refer to previously mangled types.
1532   if (startsWithDigit(MangledName))
1533     return demangleBackRefName(MangledName);
1534 
1535   if (MangledName.startsWith("?$"))
1536     return demangleTemplateInstantiationName(MangledName, NBB_Template);
1537 
1538   return demangleSimpleName(MangledName, Memorize);
1539 }
1540 
1541 IdentifierNode *
1542 Demangler::demangleUnqualifiedSymbolName(StringView &MangledName,
1543                                          NameBackrefBehavior NBB) {
1544   if (startsWithDigit(MangledName))
1545     return demangleBackRefName(MangledName);
1546   if (MangledName.startsWith("?$"))
1547     return demangleTemplateInstantiationName(MangledName, NBB);
1548   if (MangledName.startsWith('?'))
1549     return demangleFunctionIdentifierCode(MangledName);
1550   return demangleSimpleName(MangledName, (NBB & NBB_Simple) != 0);
1551 }
1552 
1553 IdentifierNode *Demangler::demangleNameScopePiece(StringView &MangledName) {
1554   if (startsWithDigit(MangledName))
1555     return demangleBackRefName(MangledName);
1556 
1557   if (MangledName.startsWith("?$"))
1558     return demangleTemplateInstantiationName(MangledName, NBB_Template);
1559 
1560   if (MangledName.startsWith("?A"))
1561     return demangleAnonymousNamespaceName(MangledName);
1562 
1563   if (startsWithLocalScopePattern(MangledName))
1564     return demangleLocallyScopedNamePiece(MangledName);
1565 
1566   return demangleSimpleName(MangledName, true);
1567 }
1568 
1569 static NodeArrayNode *nodeListToNodeArray(ArenaAllocator &Arena, NodeList *Head,
1570                                           size_t Count) {
1571   NodeArrayNode *N = Arena.alloc<NodeArrayNode>();
1572   N->Count = Count;
1573   N->Nodes = Arena.allocArray<Node *>(Count);
1574   for (size_t I = 0; I < Count; ++I) {
1575     N->Nodes[I] = Head->N;
1576     Head = Head->Next;
1577   }
1578   return N;
1579 }
1580 
1581 QualifiedNameNode *
1582 Demangler::demangleNameScopeChain(StringView &MangledName,
1583                                   IdentifierNode *UnqualifiedName) {
1584   NodeList *Head = Arena.alloc<NodeList>();
1585 
1586   Head->N = UnqualifiedName;
1587 
1588   size_t Count = 1;
1589   while (!MangledName.consumeFront("@")) {
1590     ++Count;
1591     NodeList *NewHead = Arena.alloc<NodeList>();
1592     NewHead->Next = Head;
1593     Head = NewHead;
1594 
1595     if (MangledName.empty()) {
1596       Error = true;
1597       return nullptr;
1598     }
1599 
1600     assert(!Error);
1601     IdentifierNode *Elem = demangleNameScopePiece(MangledName);
1602     if (Error)
1603       return nullptr;
1604 
1605     Head->N = Elem;
1606   }
1607 
1608   QualifiedNameNode *QN = Arena.alloc<QualifiedNameNode>();
1609   QN->Components = nodeListToNodeArray(Arena, Head, Count);
1610   return QN;
1611 }
1612 
1613 FuncClass Demangler::demangleFunctionClass(StringView &MangledName) {
1614   switch (MangledName.popFront()) {
1615   case '9':
1616     return FuncClass(FC_ExternC | FC_NoParameterList);
1617   case 'A':
1618     return FC_Private;
1619   case 'B':
1620     return FuncClass(FC_Private | FC_Far);
1621   case 'C':
1622     return FuncClass(FC_Private | FC_Static);
1623   case 'D':
1624     return FuncClass(FC_Private | FC_Static);
1625   case 'E':
1626     return FuncClass(FC_Private | FC_Virtual);
1627   case 'F':
1628     return FuncClass(FC_Private | FC_Virtual);
1629   case 'G':
1630     return FuncClass(FC_Private | FC_StaticThisAdjust);
1631   case 'H':
1632     return FuncClass(FC_Private | FC_StaticThisAdjust | FC_Far);
1633   case 'I':
1634     return FuncClass(FC_Protected);
1635   case 'J':
1636     return FuncClass(FC_Protected | FC_Far);
1637   case 'K':
1638     return FuncClass(FC_Protected | FC_Static);
1639   case 'L':
1640     return FuncClass(FC_Protected | FC_Static | FC_Far);
1641   case 'M':
1642     return FuncClass(FC_Protected | FC_Virtual);
1643   case 'N':
1644     return FuncClass(FC_Protected | FC_Virtual | FC_Far);
1645   case 'O':
1646     return FuncClass(FC_Protected | FC_Virtual | FC_StaticThisAdjust);
1647   case 'P':
1648     return FuncClass(FC_Protected | FC_Virtual | FC_StaticThisAdjust | FC_Far);
1649   case 'Q':
1650     return FuncClass(FC_Public);
1651   case 'R':
1652     return FuncClass(FC_Public | FC_Far);
1653   case 'S':
1654     return FuncClass(FC_Public | FC_Static);
1655   case 'T':
1656     return FuncClass(FC_Public | FC_Static | FC_Far);
1657   case 'U':
1658     return FuncClass(FC_Public | FC_Virtual);
1659   case 'V':
1660     return FuncClass(FC_Public | FC_Virtual | FC_Far);
1661   case 'W':
1662     return FuncClass(FC_Public | FC_Virtual | FC_StaticThisAdjust);
1663   case 'X':
1664     return FuncClass(FC_Public | FC_Virtual | FC_StaticThisAdjust | FC_Far);
1665   case 'Y':
1666     return FuncClass(FC_Global);
1667   case 'Z':
1668     return FuncClass(FC_Global | FC_Far);
1669   case '$': {
1670     FuncClass VFlag = FC_VirtualThisAdjust;
1671     if (MangledName.consumeFront('R'))
1672       VFlag = FuncClass(VFlag | FC_VirtualThisAdjustEx);
1673 
1674     switch (MangledName.popFront()) {
1675     case '0':
1676       return FuncClass(FC_Private | FC_Virtual | VFlag);
1677     case '1':
1678       return FuncClass(FC_Private | FC_Virtual | VFlag | FC_Far);
1679     case '2':
1680       return FuncClass(FC_Protected | FC_Virtual | VFlag);
1681     case '3':
1682       return FuncClass(FC_Protected | FC_Virtual | VFlag | FC_Far);
1683     case '4':
1684       return FuncClass(FC_Public | FC_Virtual | VFlag);
1685     case '5':
1686       return FuncClass(FC_Public | FC_Virtual | VFlag | FC_Far);
1687     }
1688   }
1689   }
1690 
1691   Error = true;
1692   return FC_Public;
1693 }
1694 
1695 CallingConv Demangler::demangleCallingConvention(StringView &MangledName) {
1696   switch (MangledName.popFront()) {
1697   case 'A':
1698   case 'B':
1699     return CallingConv::Cdecl;
1700   case 'C':
1701   case 'D':
1702     return CallingConv::Pascal;
1703   case 'E':
1704   case 'F':
1705     return CallingConv::Thiscall;
1706   case 'G':
1707   case 'H':
1708     return CallingConv::Stdcall;
1709   case 'I':
1710   case 'J':
1711     return CallingConv::Fastcall;
1712   case 'M':
1713   case 'N':
1714     return CallingConv::Clrcall;
1715   case 'O':
1716   case 'P':
1717     return CallingConv::Eabi;
1718   case 'Q':
1719     return CallingConv::Vectorcall;
1720   }
1721 
1722   return CallingConv::None;
1723 }
1724 
1725 StorageClass Demangler::demangleVariableStorageClass(StringView &MangledName) {
1726   assert(std::isdigit(MangledName.front()));
1727 
1728   switch (MangledName.popFront()) {
1729   case '0':
1730     return StorageClass::PrivateStatic;
1731   case '1':
1732     return StorageClass::ProtectedStatic;
1733   case '2':
1734     return StorageClass::PublicStatic;
1735   case '3':
1736     return StorageClass::Global;
1737   case '4':
1738     return StorageClass::FunctionLocalStatic;
1739   }
1740   Error = true;
1741   return StorageClass::None;
1742 }
1743 
1744 std::pair<Qualifiers, bool>
1745 Demangler::demangleQualifiers(StringView &MangledName) {
1746 
1747   switch (MangledName.popFront()) {
1748   // Member qualifiers
1749   case 'Q':
1750     return std::make_pair(Q_None, true);
1751   case 'R':
1752     return std::make_pair(Q_Const, true);
1753   case 'S':
1754     return std::make_pair(Q_Volatile, true);
1755   case 'T':
1756     return std::make_pair(Qualifiers(Q_Const | Q_Volatile), true);
1757   // Non-Member qualifiers
1758   case 'A':
1759     return std::make_pair(Q_None, false);
1760   case 'B':
1761     return std::make_pair(Q_Const, false);
1762   case 'C':
1763     return std::make_pair(Q_Volatile, false);
1764   case 'D':
1765     return std::make_pair(Qualifiers(Q_Const | Q_Volatile), false);
1766   }
1767   Error = true;
1768   return std::make_pair(Q_None, false);
1769 }
1770 
1771 // <variable-type> ::= <type> <cvr-qualifiers>
1772 //                 ::= <type> <pointee-cvr-qualifiers> # pointers, references
1773 TypeNode *Demangler::demangleType(StringView &MangledName,
1774                                   QualifierMangleMode QMM) {
1775   Qualifiers Quals = Q_None;
1776   bool IsMember = false;
1777   if (QMM == QualifierMangleMode::Mangle) {
1778     std::tie(Quals, IsMember) = demangleQualifiers(MangledName);
1779   } else if (QMM == QualifierMangleMode::Result) {
1780     if (MangledName.consumeFront('?'))
1781       std::tie(Quals, IsMember) = demangleQualifiers(MangledName);
1782   }
1783 
1784   TypeNode *Ty = nullptr;
1785   if (isTagType(MangledName))
1786     Ty = demangleClassType(MangledName);
1787   else if (isPointerType(MangledName)) {
1788     if (isMemberPointer(MangledName))
1789       Ty = demangleMemberPointerType(MangledName);
1790     else
1791       Ty = demanglePointerType(MangledName);
1792   } else if (isArrayType(MangledName))
1793     Ty = demangleArrayType(MangledName);
1794   else if (isFunctionType(MangledName)) {
1795     if (MangledName.consumeFront("$$A8@@"))
1796       Ty = demangleFunctionType(MangledName, true);
1797     else {
1798       assert(MangledName.startsWith("$$A6"));
1799       MangledName.consumeFront("$$A6");
1800       Ty = demangleFunctionType(MangledName, false);
1801     }
1802   } else if (isCustomType(MangledName)) {
1803     Ty = demangleCustomType(MangledName);
1804   } else {
1805     Ty = demanglePrimitiveType(MangledName);
1806     if (!Ty || Error)
1807       return Ty;
1808   }
1809 
1810   Ty->Quals = Qualifiers(Ty->Quals | Quals);
1811   return Ty;
1812 }
1813 
1814 void Demangler::demangleThrowSpecification(StringView &MangledName) {
1815   if (MangledName.consumeFront('Z'))
1816     return;
1817 
1818   Error = true;
1819 }
1820 
1821 FunctionSignatureNode *Demangler::demangleFunctionType(StringView &MangledName,
1822                                                        bool HasThisQuals) {
1823   FunctionSignatureNode *FTy = Arena.alloc<FunctionSignatureNode>();
1824 
1825   if (HasThisQuals) {
1826     FTy->Quals = demanglePointerExtQualifiers(MangledName);
1827     FTy->RefQualifier = demangleFunctionRefQualifier(MangledName);
1828     FTy->Quals = Qualifiers(FTy->Quals | demangleQualifiers(MangledName).first);
1829   }
1830 
1831   // Fields that appear on both member and non-member functions.
1832   FTy->CallConvention = demangleCallingConvention(MangledName);
1833 
1834   // <return-type> ::= <type>
1835   //               ::= @ # structors (they have no declared return type)
1836   bool IsStructor = MangledName.consumeFront('@');
1837   if (!IsStructor)
1838     FTy->ReturnType = demangleType(MangledName, QualifierMangleMode::Result);
1839 
1840   FTy->Params = demangleFunctionParameterList(MangledName);
1841 
1842   demangleThrowSpecification(MangledName);
1843 
1844   return FTy;
1845 }
1846 
1847 FunctionSymbolNode *
1848 Demangler::demangleFunctionEncoding(StringView &MangledName) {
1849   FuncClass ExtraFlags = FC_None;
1850   if (MangledName.consumeFront("$$J0"))
1851     ExtraFlags = FC_ExternC;
1852 
1853   FuncClass FC = demangleFunctionClass(MangledName);
1854   FC = FuncClass(ExtraFlags | FC);
1855 
1856   FunctionSignatureNode *FSN = nullptr;
1857   ThunkSignatureNode *TTN = nullptr;
1858   if (FC & FC_StaticThisAdjust) {
1859     TTN = Arena.alloc<ThunkSignatureNode>();
1860     TTN->ThisAdjust.StaticOffset = demangleSigned(MangledName);
1861   } else if (FC & FC_VirtualThisAdjust) {
1862     TTN = Arena.alloc<ThunkSignatureNode>();
1863     if (FC & FC_VirtualThisAdjustEx) {
1864       TTN->ThisAdjust.VBPtrOffset = demangleSigned(MangledName);
1865       TTN->ThisAdjust.VBOffsetOffset = demangleSigned(MangledName);
1866     }
1867     TTN->ThisAdjust.VtordispOffset = demangleSigned(MangledName);
1868     TTN->ThisAdjust.StaticOffset = demangleSigned(MangledName);
1869   }
1870 
1871   if (FC & FC_NoParameterList) {
1872     // This is an extern "C" function whose full signature hasn't been mangled.
1873     // This happens when we need to mangle a local symbol inside of an extern
1874     // "C" function.
1875     FSN = Arena.alloc<FunctionSignatureNode>();
1876   } else {
1877     bool HasThisQuals = !(FC & (FC_Global | FC_Static));
1878     FSN = demangleFunctionType(MangledName, HasThisQuals);
1879   }
1880   if (TTN) {
1881     *static_cast<FunctionSignatureNode *>(TTN) = *FSN;
1882     FSN = TTN;
1883   }
1884   FSN->FunctionClass = FC;
1885 
1886   FunctionSymbolNode *Symbol = Arena.alloc<FunctionSymbolNode>();
1887   Symbol->Signature = FSN;
1888   return Symbol;
1889 }
1890 
1891 CustomTypeNode *Demangler::demangleCustomType(StringView &MangledName) {
1892   assert(MangledName.startsWith('?'));
1893   MangledName.popFront();
1894 
1895   CustomTypeNode *CTN = Arena.alloc<CustomTypeNode>();
1896   CTN->Identifier = demangleUnqualifiedTypeName(MangledName, true);
1897   if (!MangledName.consumeFront('@'))
1898     Error = true;
1899   if (Error)
1900     return nullptr;
1901   return CTN;
1902 }
1903 
1904 // Reads a primitive type.
1905 PrimitiveTypeNode *Demangler::demanglePrimitiveType(StringView &MangledName) {
1906   if (MangledName.consumeFront("$$T"))
1907     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Nullptr);
1908 
1909   switch (MangledName.popFront()) {
1910   case 'X':
1911     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Void);
1912   case 'D':
1913     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char);
1914   case 'C':
1915     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Schar);
1916   case 'E':
1917     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uchar);
1918   case 'F':
1919     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Short);
1920   case 'G':
1921     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ushort);
1922   case 'H':
1923     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Int);
1924   case 'I':
1925     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uint);
1926   case 'J':
1927     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Long);
1928   case 'K':
1929     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ulong);
1930   case 'M':
1931     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Float);
1932   case 'N':
1933     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Double);
1934   case 'O':
1935     return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ldouble);
1936   case '_': {
1937     if (MangledName.empty()) {
1938       Error = true;
1939       return nullptr;
1940     }
1941     switch (MangledName.popFront()) {
1942     case 'N':
1943       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Bool);
1944     case 'J':
1945       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Int64);
1946     case 'K':
1947       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uint64);
1948     case 'W':
1949       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Wchar);
1950     case 'S':
1951       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char16);
1952     case 'U':
1953       return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char32);
1954     }
1955     break;
1956   }
1957   }
1958   Error = true;
1959   return nullptr;
1960 }
1961 
1962 TagTypeNode *Demangler::demangleClassType(StringView &MangledName) {
1963   TagTypeNode *TT = nullptr;
1964 
1965   switch (MangledName.popFront()) {
1966   case 'T':
1967     TT = Arena.alloc<TagTypeNode>(TagKind::Union);
1968     break;
1969   case 'U':
1970     TT = Arena.alloc<TagTypeNode>(TagKind::Struct);
1971     break;
1972   case 'V':
1973     TT = Arena.alloc<TagTypeNode>(TagKind::Class);
1974     break;
1975   case 'W':
1976     if (MangledName.popFront() != '4') {
1977       Error = true;
1978       return nullptr;
1979     }
1980     TT = Arena.alloc<TagTypeNode>(TagKind::Enum);
1981     break;
1982   default:
1983     assert(false);
1984   }
1985 
1986   TT->QualifiedName = demangleFullyQualifiedTypeName(MangledName);
1987   return TT;
1988 }
1989 
1990 // <pointer-type> ::= E? <pointer-cvr-qualifiers> <ext-qualifiers> <type>
1991 //                       # the E is required for 64-bit non-static pointers
1992 PointerTypeNode *Demangler::demanglePointerType(StringView &MangledName) {
1993   PointerTypeNode *Pointer = Arena.alloc<PointerTypeNode>();
1994 
1995   std::tie(Pointer->Quals, Pointer->Affinity) =
1996       demanglePointerCVQualifiers(MangledName);
1997 
1998   if (MangledName.consumeFront("6")) {
1999     Pointer->Pointee = demangleFunctionType(MangledName, false);
2000     return Pointer;
2001   }
2002 
2003   Qualifiers ExtQuals = demanglePointerExtQualifiers(MangledName);
2004   Pointer->Quals = Qualifiers(Pointer->Quals | ExtQuals);
2005 
2006   Pointer->Pointee = demangleType(MangledName, QualifierMangleMode::Mangle);
2007   return Pointer;
2008 }
2009 
2010 PointerTypeNode *Demangler::demangleMemberPointerType(StringView &MangledName) {
2011   PointerTypeNode *Pointer = Arena.alloc<PointerTypeNode>();
2012 
2013   std::tie(Pointer->Quals, Pointer->Affinity) =
2014       demanglePointerCVQualifiers(MangledName);
2015   assert(Pointer->Affinity == PointerAffinity::Pointer);
2016 
2017   Qualifiers ExtQuals = demanglePointerExtQualifiers(MangledName);
2018   Pointer->Quals = Qualifiers(Pointer->Quals | ExtQuals);
2019 
2020   if (MangledName.consumeFront("8")) {
2021     Pointer->ClassParent = demangleFullyQualifiedTypeName(MangledName);
2022     Pointer->Pointee = demangleFunctionType(MangledName, true);
2023   } else {
2024     Qualifiers PointeeQuals = Q_None;
2025     bool IsMember = false;
2026     std::tie(PointeeQuals, IsMember) = demangleQualifiers(MangledName);
2027     assert(IsMember);
2028     Pointer->ClassParent = demangleFullyQualifiedTypeName(MangledName);
2029 
2030     Pointer->Pointee = demangleType(MangledName, QualifierMangleMode::Drop);
2031     Pointer->Pointee->Quals = PointeeQuals;
2032   }
2033 
2034   return Pointer;
2035 }
2036 
2037 Qualifiers Demangler::demanglePointerExtQualifiers(StringView &MangledName) {
2038   Qualifiers Quals = Q_None;
2039   if (MangledName.consumeFront('E'))
2040     Quals = Qualifiers(Quals | Q_Pointer64);
2041   if (MangledName.consumeFront('I'))
2042     Quals = Qualifiers(Quals | Q_Restrict);
2043   if (MangledName.consumeFront('F'))
2044     Quals = Qualifiers(Quals | Q_Unaligned);
2045 
2046   return Quals;
2047 }
2048 
2049 ArrayTypeNode *Demangler::demangleArrayType(StringView &MangledName) {
2050   assert(MangledName.front() == 'Y');
2051   MangledName.popFront();
2052 
2053   uint64_t Rank = 0;
2054   bool IsNegative = false;
2055   std::tie(Rank, IsNegative) = demangleNumber(MangledName);
2056   if (IsNegative || Rank == 0) {
2057     Error = true;
2058     return nullptr;
2059   }
2060 
2061   ArrayTypeNode *ATy = Arena.alloc<ArrayTypeNode>();
2062   NodeList *Head = Arena.alloc<NodeList>();
2063   NodeList *Tail = Head;
2064 
2065   for (uint64_t I = 0; I < Rank; ++I) {
2066     uint64_t D = 0;
2067     std::tie(D, IsNegative) = demangleNumber(MangledName);
2068     if (IsNegative) {
2069       Error = true;
2070       return nullptr;
2071     }
2072     Tail->N = Arena.alloc<IntegerLiteralNode>(D, IsNegative);
2073     if (I + 1 < Rank) {
2074       Tail->Next = Arena.alloc<NodeList>();
2075       Tail = Tail->Next;
2076     }
2077   }
2078   ATy->Dimensions = nodeListToNodeArray(Arena, Head, Rank);
2079 
2080   if (MangledName.consumeFront("$$C")) {
2081     bool IsMember = false;
2082     std::tie(ATy->Quals, IsMember) = demangleQualifiers(MangledName);
2083     if (IsMember) {
2084       Error = true;
2085       return nullptr;
2086     }
2087   }
2088 
2089   ATy->ElementType = demangleType(MangledName, QualifierMangleMode::Drop);
2090   return ATy;
2091 }
2092 
2093 // Reads a function or a template parameters.
2094 NodeArrayNode *
2095 Demangler::demangleFunctionParameterList(StringView &MangledName) {
2096   // Empty parameter list.
2097   if (MangledName.consumeFront('X'))
2098     return {};
2099 
2100   NodeList *Head = Arena.alloc<NodeList>();
2101   NodeList **Current = &Head;
2102   size_t Count = 0;
2103   while (!Error && !MangledName.startsWith('@') &&
2104          !MangledName.startsWith('Z')) {
2105     ++Count;
2106 
2107     if (startsWithDigit(MangledName)) {
2108       size_t N = MangledName[0] - '0';
2109       if (N >= Backrefs.FunctionParamCount) {
2110         Error = true;
2111         return {};
2112       }
2113       MangledName = MangledName.dropFront();
2114 
2115       *Current = Arena.alloc<NodeList>();
2116       (*Current)->N = Backrefs.FunctionParams[N];
2117       Current = &(*Current)->Next;
2118       continue;
2119     }
2120 
2121     size_t OldSize = MangledName.size();
2122 
2123     *Current = Arena.alloc<NodeList>();
2124     TypeNode *TN = demangleType(MangledName, QualifierMangleMode::Drop);
2125 
2126     (*Current)->N = TN;
2127 
2128     size_t CharsConsumed = OldSize - MangledName.size();
2129     assert(CharsConsumed != 0);
2130 
2131     // Single-letter types are ignored for backreferences because memorizing
2132     // them doesn't save anything.
2133     if (Backrefs.FunctionParamCount <= 9 && CharsConsumed > 1)
2134       Backrefs.FunctionParams[Backrefs.FunctionParamCount++] = TN;
2135 
2136     Current = &(*Current)->Next;
2137   }
2138 
2139   if (Error)
2140     return {};
2141 
2142   NodeArrayNode *NA = nodeListToNodeArray(Arena, Head, Count);
2143   // A non-empty parameter list is terminated by either 'Z' (variadic) parameter
2144   // list or '@' (non variadic).  Careful not to consume "@Z", as in that case
2145   // the following Z could be a throw specifier.
2146   if (MangledName.consumeFront('@'))
2147     return NA;
2148 
2149   if (MangledName.consumeFront('Z')) {
2150     // This is a variadic parameter list.  We probably need a variadic node to
2151     // append to the end.
2152     return NA;
2153   }
2154 
2155   Error = true;
2156   return {};
2157 }
2158 
2159 NodeArrayNode *
2160 Demangler::demangleTemplateParameterList(StringView &MangledName) {
2161   NodeList *Head;
2162   NodeList **Current = &Head;
2163   size_t Count = 0;
2164 
2165   while (!Error && !MangledName.startsWith('@')) {
2166     if (MangledName.consumeFront("$S") || MangledName.consumeFront("$$V") ||
2167         MangledName.consumeFront("$$$V") || MangledName.consumeFront("$$Z")) {
2168       // parameter pack separator
2169       continue;
2170     }
2171 
2172     ++Count;
2173 
2174     // Template parameter lists don't participate in back-referencing.
2175     *Current = Arena.alloc<NodeList>();
2176 
2177     NodeList &TP = **Current;
2178 
2179     TemplateParameterReferenceNode *TPRN = nullptr;
2180     if (MangledName.consumeFront("$$Y")) {
2181       // Template alias
2182       TP.N = demangleFullyQualifiedTypeName(MangledName);
2183     } else if (MangledName.consumeFront("$$B")) {
2184       // Array
2185       TP.N = demangleType(MangledName, QualifierMangleMode::Drop);
2186     } else if (MangledName.consumeFront("$$C")) {
2187       // Type has qualifiers.
2188       TP.N = demangleType(MangledName, QualifierMangleMode::Mangle);
2189     } else if (MangledName.startsWith("$1") || MangledName.startsWith("$H") ||
2190                MangledName.startsWith("$I") || MangledName.startsWith("$J")) {
2191       // Pointer to member
2192       TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2193       TPRN->IsMemberPointer = true;
2194 
2195       MangledName = MangledName.dropFront();
2196       // 1 - single inheritance       <name>
2197       // H - multiple inheritance     <name> <number>
2198       // I - virtual inheritance      <name> <number> <number> <number>
2199       // J - unspecified inheritance  <name> <number> <number> <number>
2200       char InheritanceSpecifier = MangledName.popFront();
2201       SymbolNode *S = nullptr;
2202       if (MangledName.startsWith('?')) {
2203         S = parse(MangledName);
2204         memorizeIdentifier(S->Name->getUnqualifiedIdentifier());
2205       }
2206 
2207       switch (InheritanceSpecifier) {
2208       case 'J':
2209         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2210             demangleSigned(MangledName);
2211         LLVM_FALLTHROUGH;
2212       case 'I':
2213         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2214             demangleSigned(MangledName);
2215         LLVM_FALLTHROUGH;
2216       case 'H':
2217         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2218             demangleSigned(MangledName);
2219         LLVM_FALLTHROUGH;
2220       case '1':
2221         break;
2222       default:
2223         Error = true;
2224         break;
2225       }
2226       TPRN->Affinity = PointerAffinity::Pointer;
2227       TPRN->Symbol = S;
2228     } else if (MangledName.startsWith("$E?")) {
2229       MangledName.consumeFront("$E");
2230       // Reference to symbol
2231       TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2232       TPRN->Symbol = parse(MangledName);
2233       TPRN->Affinity = PointerAffinity::Reference;
2234     } else if (MangledName.startsWith("$F") || MangledName.startsWith("$G")) {
2235       TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2236 
2237       // Data member pointer.
2238       MangledName = MangledName.dropFront();
2239       char InheritanceSpecifier = MangledName.popFront();
2240 
2241       switch (InheritanceSpecifier) {
2242       case 'G':
2243         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2244             demangleSigned(MangledName);
2245         LLVM_FALLTHROUGH;
2246       case 'F':
2247         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2248             demangleSigned(MangledName);
2249         TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2250             demangleSigned(MangledName);
2251         LLVM_FALLTHROUGH;
2252       case '0':
2253         break;
2254       default:
2255         Error = true;
2256         break;
2257       }
2258       TPRN->IsMemberPointer = true;
2259 
2260     } else if (MangledName.consumeFront("$0")) {
2261       // Integral non-type template parameter
2262       bool IsNegative = false;
2263       uint64_t Value = 0;
2264       std::tie(Value, IsNegative) = demangleNumber(MangledName);
2265 
2266       TP.N = Arena.alloc<IntegerLiteralNode>(Value, IsNegative);
2267     } else {
2268       TP.N = demangleType(MangledName, QualifierMangleMode::Drop);
2269     }
2270     if (Error)
2271       return nullptr;
2272 
2273     Current = &TP.Next;
2274   }
2275 
2276   if (Error)
2277     return nullptr;
2278 
2279   // Template parameter lists cannot be variadic, so it can only be terminated
2280   // by @.
2281   if (MangledName.consumeFront('@'))
2282     return nodeListToNodeArray(Arena, Head, Count);
2283   Error = true;
2284   return nullptr;
2285 }
2286 
2287 void Demangler::dumpBackReferences() {
2288   std::printf("%d function parameter backreferences\n",
2289               (int)Backrefs.FunctionParamCount);
2290 
2291   // Create an output stream so we can render each type.
2292   OutputStream OS = OutputStream::create(nullptr, 0, 1024);
2293   for (size_t I = 0; I < Backrefs.FunctionParamCount; ++I) {
2294     OS.setCurrentPosition(0);
2295 
2296     TypeNode *T = Backrefs.FunctionParams[I];
2297     T->output(OS, OF_Default);
2298 
2299     std::printf("  [%d] - %.*s\n", (int)I, (int)OS.getCurrentPosition(),
2300                 OS.getBuffer());
2301   }
2302   std::free(OS.getBuffer());
2303 
2304   if (Backrefs.FunctionParamCount > 0)
2305     std::printf("\n");
2306   std::printf("%d name backreferences\n", (int)Backrefs.NamesCount);
2307   for (size_t I = 0; I < Backrefs.NamesCount; ++I) {
2308     std::printf("  [%d] - %.*s\n", (int)I, (int)Backrefs.Names[I]->Name.size(),
2309                 Backrefs.Names[I]->Name.begin());
2310   }
2311   if (Backrefs.NamesCount > 0)
2312     std::printf("\n");
2313 }
2314 
2315 char *llvm::microsoftDemangle(const char *MangledName, char *Buf, size_t *N,
2316                               int *Status, MSDemangleFlags Flags) {
2317   Demangler D;
2318   StringView Name{MangledName};
2319   SymbolNode *S = D.parse(Name);
2320 
2321   if (Flags & MSDF_DumpBackrefs)
2322     D.dumpBackReferences();
2323   OutputStream OS = OutputStream::create(Buf, N, 1024);
2324   if (D.Error) {
2325     OS << MangledName;
2326     *Status = llvm::demangle_invalid_mangled_name;
2327   } else {
2328     S->output(OS, OF_Default);
2329     *Status = llvm::demangle_success;
2330   }
2331 
2332   OS << '\0';
2333   return OS.getBuffer();
2334 }
2335