blob: b8e39c415270776079089b5e7a4153efbdb5ee06 [file] [log] [blame]
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001//===- MicrosoftDemangle.cpp ----------------------------------------------===//
2//
Chandler Carruth57b08b02019-01-19 10:56:40 +00003// 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
Zachary Turnerf435a7e2018-07-20 17:27:48 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This file defines a demangler for MSVC-style mangled symbols.
10//
11// This file has no dependencies on the rest of LLVM so that it can be
12// easily reused in other programs such as libcxxabi.
13//
14//===----------------------------------------------------------------------===//
15
Zachary Turner7ba90562018-11-01 15:07:32 +000016#include "llvm/Demangle/MicrosoftDemangle.h"
Zachary Turnerf435a7e2018-07-20 17:27:48 +000017#include "llvm/Demangle/Demangle.h"
Zachary Turner7ba90562018-11-01 15:07:32 +000018#include "llvm/Demangle/MicrosoftDemangleNodes.h"
Zachary Turnerf435a7e2018-07-20 17:27:48 +000019
Erik Pilkington5094e5e2019-01-17 20:37:51 +000020#include "llvm/Demangle/DemangleConfig.h"
Richard Smith8a57f2e2018-08-20 19:44:01 +000021#include "llvm/Demangle/StringView.h"
22#include "llvm/Demangle/Utility.h"
Zachary Turnerf435a7e2018-07-20 17:27:48 +000023
Zachary Turner66555a72018-08-20 19:15:35 +000024#include <array>
Zachary Turnerf435a7e2018-07-20 17:27:48 +000025#include <cctype>
Zachary Turner5ae08b82018-08-01 18:44:12 +000026#include <cstdio>
Zachary Turnerd742d642018-07-26 19:56:09 +000027#include <tuple>
Zachary Turnerf435a7e2018-07-20 17:27:48 +000028
Zachary Turner03312862018-08-27 03:48:03 +000029using namespace llvm;
30using namespace ms_demangle;
Zachary Turnerf435a7e2018-07-20 17:27:48 +000031
32static bool startsWithDigit(StringView S) {
33 return !S.empty() && std::isdigit(S.front());
34}
35
Zachary Turnerd742d642018-07-26 19:56:09 +000036
Zachary Turner03312862018-08-27 03:48:03 +000037struct NodeList {
38 Node *N = nullptr;
39 NodeList *Next = nullptr;
Zachary Turnerf435a7e2018-07-20 17:27:48 +000040};
41
Zachary Turner8fb9a712018-12-14 18:10:13 +000042static bool isMemberPointer(StringView MangledName, bool &Error) {
43 Error = false;
Zachary Turnerd742d642018-07-26 19:56:09 +000044 switch (MangledName.popFront()) {
Zachary Turner931e8792018-07-30 23:02:10 +000045 case '$':
46 // This is probably an rvalue reference (e.g. $$Q), and you cannot have an
47 // rvalue reference to a member.
48 return false;
Zachary Turnerd742d642018-07-26 19:56:09 +000049 case 'A':
50 // 'A' indicates a reference, and you cannot have a reference to a member
Zachary Turner931e8792018-07-30 23:02:10 +000051 // function or member.
Zachary Turnerd742d642018-07-26 19:56:09 +000052 return false;
53 case 'P':
54 case 'Q':
55 case 'R':
56 case 'S':
57 // These 4 values indicate some kind of pointer, but we still don't know
58 // what.
59 break;
60 default:
Zachary Turner8fb9a712018-12-14 18:10:13 +000061 Error = true;
62 return false;
Zachary Turnerd742d642018-07-26 19:56:09 +000063 }
64
65 // If it starts with a number, then 6 indicates a non-member function
66 // pointer, and 8 indicates a member function pointer.
67 if (startsWithDigit(MangledName)) {
68 assert(MangledName[0] == '6' || MangledName[0] == '8');
69 return (MangledName[0] == '8');
70 }
71
72 // Remove ext qualifiers since those can appear on either type and are
73 // therefore not indicative.
74 MangledName.consumeFront('E'); // 64-bit
75 MangledName.consumeFront('I'); // restrict
76 MangledName.consumeFront('F'); // unaligned
77
78 assert(!MangledName.empty());
79
80 // The next value should be either ABCD (non-member) or QRST (member).
81 switch (MangledName.front()) {
82 case 'A':
83 case 'B':
84 case 'C':
85 case 'D':
86 return false;
87 case 'Q':
88 case 'R':
89 case 'S':
90 case 'T':
91 return true;
92 default:
Zachary Turner8fb9a712018-12-14 18:10:13 +000093 Error = true;
94 return false;
Zachary Turnerd742d642018-07-26 19:56:09 +000095 }
Zachary Turnerd742d642018-07-26 19:56:09 +000096}
97
Zachary Turner03312862018-08-27 03:48:03 +000098static SpecialIntrinsicKind
99consumeSpecialIntrinsicKind(StringView &MangledName) {
100 if (MangledName.consumeFront("?_7"))
101 return SpecialIntrinsicKind::Vftable;
102 if (MangledName.consumeFront("?_8"))
103 return SpecialIntrinsicKind::Vbtable;
104 if (MangledName.consumeFront("?_9"))
105 return SpecialIntrinsicKind::VcallThunk;
106 if (MangledName.consumeFront("?_A"))
107 return SpecialIntrinsicKind::Typeof;
108 if (MangledName.consumeFront("?_B"))
109 return SpecialIntrinsicKind::LocalStaticGuard;
110 if (MangledName.consumeFront("?_C"))
111 return SpecialIntrinsicKind::StringLiteralSymbol;
112 if (MangledName.consumeFront("?_P"))
113 return SpecialIntrinsicKind::UdtReturning;
114 if (MangledName.consumeFront("?_R0"))
115 return SpecialIntrinsicKind::RttiTypeDescriptor;
116 if (MangledName.consumeFront("?_R1"))
117 return SpecialIntrinsicKind::RttiBaseClassDescriptor;
118 if (MangledName.consumeFront("?_R2"))
119 return SpecialIntrinsicKind::RttiBaseClassArray;
120 if (MangledName.consumeFront("?_R3"))
121 return SpecialIntrinsicKind::RttiClassHierarchyDescriptor;
122 if (MangledName.consumeFront("?_R4"))
123 return SpecialIntrinsicKind::RttiCompleteObjLocator;
124 if (MangledName.consumeFront("?_S"))
125 return SpecialIntrinsicKind::LocalVftable;
126 if (MangledName.consumeFront("?__E"))
127 return SpecialIntrinsicKind::DynamicInitializer;
128 if (MangledName.consumeFront("?__F"))
129 return SpecialIntrinsicKind::DynamicAtexitDestructor;
130 if (MangledName.consumeFront("?__J"))
131 return SpecialIntrinsicKind::LocalStaticThreadGuard;
132 return SpecialIntrinsicKind::None;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000133}
134
Zachary Turner71c91f92018-07-30 03:12:34 +0000135static bool startsWithLocalScopePattern(StringView S) {
136 if (!S.consumeFront('?'))
137 return false;
138 if (S.size() < 2)
139 return false;
140
141 size_t End = S.find('?');
142 if (End == StringView::npos)
143 return false;
144 StringView Candidate = S.substr(0, End);
145 if (Candidate.empty())
146 return false;
147
148 // \?[0-9]\?
149 // ?@? is the discriminator 0.
150 if (Candidate.size() == 1)
151 return Candidate[0] == '@' || (Candidate[0] >= '0' && Candidate[0] <= '9');
152
153 // If it's not 0-9, then it's an encoded number terminated with an @
154 if (Candidate.back() != '@')
155 return false;
156 Candidate = Candidate.dropBack();
157
158 // An encoded number starts with B-P and all subsequent digits are in A-P.
159 // Note that the reason the first digit cannot be A is two fold. First, it
160 // would create an ambiguity with ?A which delimits the beginning of an
161 // anonymous namespace. Second, A represents 0, and you don't start a multi
162 // digit number with a leading 0. Presumably the anonymous namespace
163 // ambiguity is also why single digit encoded numbers use 0-9 rather than A-J.
164 if (Candidate[0] < 'B' || Candidate[0] > 'P')
165 return false;
166 Candidate = Candidate.dropFront();
167 while (!Candidate.empty()) {
168 if (Candidate[0] < 'A' || Candidate[0] > 'P')
169 return false;
170 Candidate = Candidate.dropFront();
171 }
172
173 return true;
174}
175
Zachary Turner03312862018-08-27 03:48:03 +0000176static bool isTagType(StringView S) {
177 switch (S.front()) {
178 case 'T': // union
179 case 'U': // struct
180 case 'V': // class
181 case 'W': // enum
182 return true;
Zachary Turner38b78a72018-07-26 20:20:10 +0000183 }
Zachary Turner03312862018-08-27 03:48:03 +0000184 return false;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000185}
186
Zachary Turnerb2fef1a2018-08-29 04:12:44 +0000187static bool isCustomType(StringView S) { return S[0] == '?'; }
188
Zachary Turner03312862018-08-27 03:48:03 +0000189static bool isPointerType(StringView S) {
190 if (S.startsWith("$$Q")) // foo &&
191 return true;
192
193 switch (S.front()) {
194 case 'A': // foo &
195 case 'P': // foo *
196 case 'Q': // foo *const
197 case 'R': // foo *volatile
198 case 'S': // foo *const volatile
199 return true;
200 }
201 return false;
202}
203
204static bool isArrayType(StringView S) { return S[0] == 'Y'; }
205
206static bool isFunctionType(StringView S) {
207 return S.startsWith("$$A8@@") || S.startsWith("$$A6");
208}
209
210static FunctionRefQualifier
211demangleFunctionRefQualifier(StringView &MangledName) {
212 if (MangledName.consumeFront('G'))
213 return FunctionRefQualifier::Reference;
214 else if (MangledName.consumeFront('H'))
215 return FunctionRefQualifier::RValueReference;
216 return FunctionRefQualifier::None;
217}
218
219static std::pair<Qualifiers, PointerAffinity>
220demanglePointerCVQualifiers(StringView &MangledName) {
221 if (MangledName.consumeFront("$$Q"))
222 return std::make_pair(Q_None, PointerAffinity::RValueReference);
223
224 switch (MangledName.popFront()) {
225 case 'A':
226 return std::make_pair(Q_None, PointerAffinity::Reference);
227 case 'P':
228 return std::make_pair(Q_None, PointerAffinity::Pointer);
229 case 'Q':
230 return std::make_pair(Q_Const, PointerAffinity::Pointer);
231 case 'R':
232 return std::make_pair(Q_Volatile, PointerAffinity::Pointer);
233 case 'S':
234 return std::make_pair(Qualifiers(Q_Const | Q_Volatile),
235 PointerAffinity::Pointer);
Zachary Turner970fdc32018-08-16 16:17:36 +0000236 default:
Zachary Turner03312862018-08-27 03:48:03 +0000237 assert(false && "Ty is not a pointer type!");
Zachary Turner970fdc32018-08-16 16:17:36 +0000238 }
Zachary Turner03312862018-08-27 03:48:03 +0000239 return std::make_pair(Q_None, PointerAffinity::Pointer);
Zachary Turner970fdc32018-08-16 16:17:36 +0000240}
241
Zachary Turner71c91f92018-07-30 03:12:34 +0000242StringView Demangler::copyString(StringView Borrowed) {
243 char *Stable = Arena.allocUnalignedBuffer(Borrowed.size() + 1);
244 std::strcpy(Stable, Borrowed.begin());
245
246 return {Stable, Borrowed.size()};
247}
248
Zachary Turner03312862018-08-27 03:48:03 +0000249SpecialTableSymbolNode *
250Demangler::demangleSpecialTableSymbolNode(StringView &MangledName,
251 SpecialIntrinsicKind K) {
252 NamedIdentifierNode *NI = Arena.alloc<NamedIdentifierNode>();
253 switch (K) {
254 case SpecialIntrinsicKind::Vftable:
255 NI->Name = "`vftable'";
256 break;
257 case SpecialIntrinsicKind::Vbtable:
258 NI->Name = "`vbtable'";
259 break;
260 case SpecialIntrinsicKind::LocalVftable:
261 NI->Name = "`local vftable'";
262 break;
263 case SpecialIntrinsicKind::RttiCompleteObjLocator:
264 NI->Name = "`RTTI Complete Object Locator'";
265 break;
266 default:
Erik Pilkington5094e5e2019-01-17 20:37:51 +0000267 DEMANGLE_UNREACHABLE;
Zachary Turner03312862018-08-27 03:48:03 +0000268 }
269 QualifiedNameNode *QN = demangleNameScopeChain(MangledName, NI);
270 SpecialTableSymbolNode *STSN = Arena.alloc<SpecialTableSymbolNode>();
271 STSN->Name = QN;
Zachary Turner469f0762018-08-17 21:18:05 +0000272 bool IsMember = false;
Zachary Turner03312862018-08-27 03:48:03 +0000273 char Front = MangledName.popFront();
274 if (Front != '6' && Front != '7') {
275 Error = true;
276 return nullptr;
277 }
278
279 std::tie(STSN->Quals, IsMember) = demangleQualifiers(MangledName);
280 if (!MangledName.consumeFront('@'))
281 STSN->TargetName = demangleFullyQualifiedTypeName(MangledName);
282 return STSN;
283}
284
285LocalStaticGuardVariableNode *
286Demangler::demangleLocalStaticGuard(StringView &MangledName) {
287 LocalStaticGuardIdentifierNode *LSGI =
288 Arena.alloc<LocalStaticGuardIdentifierNode>();
289 QualifiedNameNode *QN = demangleNameScopeChain(MangledName, LSGI);
290 LocalStaticGuardVariableNode *LSGVN =
291 Arena.alloc<LocalStaticGuardVariableNode>();
292 LSGVN->Name = QN;
293
294 if (MangledName.consumeFront("4IA"))
295 LSGVN->IsVisible = false;
296 else if (MangledName.consumeFront("5"))
297 LSGVN->IsVisible = true;
298 else {
299 Error = true;
300 return nullptr;
301 }
302
303 if (!MangledName.empty())
304 LSGI->ScopeIndex = demangleUnsigned(MangledName);
305 return LSGVN;
306}
307
308static NamedIdentifierNode *synthesizeNamedIdentifier(ArenaAllocator &Arena,
309 StringView Name) {
310 NamedIdentifierNode *Id = Arena.alloc<NamedIdentifierNode>();
311 Id->Name = Name;
312 return Id;
313}
314
315static QualifiedNameNode *synthesizeQualifiedName(ArenaAllocator &Arena,
316 IdentifierNode *Identifier) {
317 QualifiedNameNode *QN = Arena.alloc<QualifiedNameNode>();
318 QN->Components = Arena.alloc<NodeArrayNode>();
319 QN->Components->Count = 1;
320 QN->Components->Nodes = Arena.allocArray<Node *>(1);
321 QN->Components->Nodes[0] = Identifier;
322 return QN;
323}
324
325static QualifiedNameNode *synthesizeQualifiedName(ArenaAllocator &Arena,
326 StringView Name) {
327 NamedIdentifierNode *Id = synthesizeNamedIdentifier(Arena, Name);
328 return synthesizeQualifiedName(Arena, Id);
329}
330
331static VariableSymbolNode *synthesizeVariable(ArenaAllocator &Arena,
332 TypeNode *Type,
333 StringView VariableName) {
334 VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
335 VSN->Type = Type;
336 VSN->Name = synthesizeQualifiedName(Arena, VariableName);
337 return VSN;
338}
339
340VariableSymbolNode *Demangler::demangleUntypedVariable(
341 ArenaAllocator &Arena, StringView &MangledName, StringView VariableName) {
342 NamedIdentifierNode *NI = synthesizeNamedIdentifier(Arena, VariableName);
343 QualifiedNameNode *QN = demangleNameScopeChain(MangledName, NI);
344 VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
345 VSN->Name = QN;
346 if (MangledName.consumeFront("8"))
347 return VSN;
348
349 Error = true;
350 return nullptr;
351}
352
353VariableSymbolNode *
354Demangler::demangleRttiBaseClassDescriptorNode(ArenaAllocator &Arena,
355 StringView &MangledName) {
356 RttiBaseClassDescriptorNode *RBCDN =
357 Arena.alloc<RttiBaseClassDescriptorNode>();
358 RBCDN->NVOffset = demangleUnsigned(MangledName);
359 RBCDN->VBPtrOffset = demangleSigned(MangledName);
360 RBCDN->VBTableOffset = demangleUnsigned(MangledName);
361 RBCDN->Flags = demangleUnsigned(MangledName);
362 if (Error)
363 return nullptr;
364
365 VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
366 VSN->Name = demangleNameScopeChain(MangledName, RBCDN);
367 MangledName.consumeFront('8');
368 return VSN;
369}
370
Zachary Turner32a8a202018-08-29 23:56:09 +0000371FunctionSymbolNode *Demangler::demangleInitFiniStub(StringView &MangledName,
372 bool IsDestructor) {
Zachary Turner03312862018-08-27 03:48:03 +0000373 DynamicStructorIdentifierNode *DSIN =
374 Arena.alloc<DynamicStructorIdentifierNode>();
375 DSIN->IsDestructor = IsDestructor;
Zachary Turner32a8a202018-08-29 23:56:09 +0000376
Zachary Turner78ab3cb2018-08-30 20:53:29 +0000377 bool IsKnownStaticDataMember = false;
378 if (MangledName.consumeFront('?'))
379 IsKnownStaticDataMember = true;
380
Zachary Turner32a8a202018-08-29 23:56:09 +0000381 QualifiedNameNode *QN = demangleFullyQualifiedSymbolName(MangledName);
382
383 SymbolNode *Symbol = demangleEncodedSymbol(MangledName, QN);
384 FunctionSymbolNode *FSN = nullptr;
385 Symbol->Name = QN;
386
387 if (Symbol->kind() == NodeKind::VariableSymbol) {
388 DSIN->Variable = static_cast<VariableSymbolNode *>(Symbol);
Zachary Turner78ab3cb2018-08-30 20:53:29 +0000389
390 // Older versions of clang mangled this type of symbol incorrectly. They
391 // would omit the leading ? and they would only emit a single @ at the end.
392 // The correct mangling is a leading ? and 2 trailing @ signs. Handle
393 // both cases.
394 int AtCount = IsKnownStaticDataMember ? 2 : 1;
395 for (int I = 0; I < AtCount; ++I) {
396 if (MangledName.consumeFront('@'))
397 continue;
Zachary Turner32a8a202018-08-29 23:56:09 +0000398 Error = true;
399 return nullptr;
400 }
401
402 FSN = demangleFunctionEncoding(MangledName);
403 FSN->Name = synthesizeQualifiedName(Arena, DSIN);
404 } else {
Zachary Turner78ab3cb2018-08-30 20:53:29 +0000405 if (IsKnownStaticDataMember) {
406 // This was supposed to be a static data member, but we got a function.
407 Error = true;
408 return nullptr;
409 }
410
Zachary Turner32a8a202018-08-29 23:56:09 +0000411 FSN = static_cast<FunctionSymbolNode *>(Symbol);
412 DSIN->Name = Symbol->Name;
413 FSN->Name = synthesizeQualifiedName(Arena, DSIN);
414 }
415
Zachary Turner03312862018-08-27 03:48:03 +0000416 return FSN;
417}
418
419SymbolNode *Demangler::demangleSpecialIntrinsic(StringView &MangledName) {
420 SpecialIntrinsicKind SIK = consumeSpecialIntrinsicKind(MangledName);
421 if (SIK == SpecialIntrinsicKind::None)
422 return nullptr;
423
424 switch (SIK) {
425 case SpecialIntrinsicKind::StringLiteralSymbol:
426 return demangleStringLiteral(MangledName);
427 case SpecialIntrinsicKind::Vftable:
428 case SpecialIntrinsicKind::Vbtable:
429 case SpecialIntrinsicKind::LocalVftable:
430 case SpecialIntrinsicKind::RttiCompleteObjLocator:
431 return demangleSpecialTableSymbolNode(MangledName, SIK);
432 case SpecialIntrinsicKind::VcallThunk:
433 return demangleVcallThunkNode(MangledName);
434 case SpecialIntrinsicKind::LocalStaticGuard:
435 return demangleLocalStaticGuard(MangledName);
436 case SpecialIntrinsicKind::RttiTypeDescriptor: {
437 TypeNode *T = demangleType(MangledName, QualifierMangleMode::Result);
Zachary Turner469f0762018-08-17 21:18:05 +0000438 if (Error)
439 break;
440 if (!MangledName.consumeFront("@8"))
Zachary Turner03312862018-08-27 03:48:03 +0000441 break;
Zachary Turner469f0762018-08-17 21:18:05 +0000442 if (!MangledName.empty())
Zachary Turner03312862018-08-27 03:48:03 +0000443 break;
444 return synthesizeVariable(Arena, T, "`RTTI Type Descriptor'");
445 }
446 case SpecialIntrinsicKind::RttiBaseClassArray:
447 return demangleUntypedVariable(Arena, MangledName,
448 "`RTTI Base Class Array'");
449 case SpecialIntrinsicKind::RttiClassHierarchyDescriptor:
450 return demangleUntypedVariable(Arena, MangledName,
451 "`RTTI Class Hierarchy Descriptor'");
452 case SpecialIntrinsicKind::RttiBaseClassDescriptor:
453 return demangleRttiBaseClassDescriptorNode(Arena, MangledName);
454 case SpecialIntrinsicKind::DynamicInitializer:
Zachary Turner32a8a202018-08-29 23:56:09 +0000455 return demangleInitFiniStub(MangledName, false);
Zachary Turner03312862018-08-27 03:48:03 +0000456 case SpecialIntrinsicKind::DynamicAtexitDestructor:
Zachary Turner32a8a202018-08-29 23:56:09 +0000457 return demangleInitFiniStub(MangledName, true);
Zachary Turner469f0762018-08-17 21:18:05 +0000458 default:
Zachary Turner469f0762018-08-17 21:18:05 +0000459 break;
460 }
Zachary Turner03312862018-08-27 03:48:03 +0000461 Error = true;
462 return nullptr;
Zachary Turner469f0762018-08-17 21:18:05 +0000463}
464
Zachary Turner03312862018-08-27 03:48:03 +0000465IdentifierNode *
466Demangler::demangleFunctionIdentifierCode(StringView &MangledName) {
467 assert(MangledName.startsWith('?'));
468 MangledName = MangledName.dropFront();
469
470 if (MangledName.consumeFront("__"))
471 return demangleFunctionIdentifierCode(
472 MangledName, FunctionIdentifierCodeGroup::DoubleUnder);
473 else if (MangledName.consumeFront("_"))
474 return demangleFunctionIdentifierCode(MangledName,
475 FunctionIdentifierCodeGroup::Under);
476 return demangleFunctionIdentifierCode(MangledName,
477 FunctionIdentifierCodeGroup::Basic);
478}
479
480StructorIdentifierNode *
481Demangler::demangleStructorIdentifier(StringView &MangledName,
482 bool IsDestructor) {
483 StructorIdentifierNode *N = Arena.alloc<StructorIdentifierNode>();
484 N->IsDestructor = IsDestructor;
485 return N;
486}
487
488ConversionOperatorIdentifierNode *
489Demangler::demangleConversionOperatorIdentifier(StringView &MangledName) {
490 ConversionOperatorIdentifierNode *N =
491 Arena.alloc<ConversionOperatorIdentifierNode>();
492 return N;
493}
494
495LiteralOperatorIdentifierNode *
496Demangler::demangleLiteralOperatorIdentifier(StringView &MangledName) {
497 LiteralOperatorIdentifierNode *N =
498 Arena.alloc<LiteralOperatorIdentifierNode>();
499 N->Name = demangleSimpleString(MangledName, false);
500 return N;
501}
502
Benjamin Kramerc55e9972018-10-13 22:18:22 +0000503static IntrinsicFunctionKind
Zachary Turner03312862018-08-27 03:48:03 +0000504translateIntrinsicFunctionCode(char CH, FunctionIdentifierCodeGroup Group) {
505 // Not all ? identifiers are intrinsics *functions*. This function only maps
506 // operator codes for the special functions, all others are handled elsewhere,
507 // hence the IFK::None entries in the table.
508 using IFK = IntrinsicFunctionKind;
509 static IFK Basic[36] = {
510 IFK::None, // ?0 # Foo::Foo()
511 IFK::None, // ?1 # Foo::~Foo()
512 IFK::New, // ?2 # operator new
513 IFK::Delete, // ?3 # operator delete
514 IFK::Assign, // ?4 # operator=
515 IFK::RightShift, // ?5 # operator>>
516 IFK::LeftShift, // ?6 # operator<<
517 IFK::LogicalNot, // ?7 # operator!
518 IFK::Equals, // ?8 # operator==
519 IFK::NotEquals, // ?9 # operator!=
520 IFK::ArraySubscript, // ?A # operator[]
521 IFK::None, // ?B # Foo::operator <type>()
522 IFK::Pointer, // ?C # operator->
523 IFK::Dereference, // ?D # operator*
524 IFK::Increment, // ?E # operator++
525 IFK::Decrement, // ?F # operator--
526 IFK::Minus, // ?G # operator-
527 IFK::Plus, // ?H # operator+
528 IFK::BitwiseAnd, // ?I # operator&
529 IFK::MemberPointer, // ?J # operator->*
530 IFK::Divide, // ?K # operator/
531 IFK::Modulus, // ?L # operator%
532 IFK::LessThan, // ?M operator<
533 IFK::LessThanEqual, // ?N operator<=
534 IFK::GreaterThan, // ?O operator>
535 IFK::GreaterThanEqual, // ?P operator>=
536 IFK::Comma, // ?Q operator,
537 IFK::Parens, // ?R operator()
538 IFK::BitwiseNot, // ?S operator~
539 IFK::BitwiseXor, // ?T operator^
540 IFK::BitwiseOr, // ?U operator|
541 IFK::LogicalAnd, // ?V operator&&
542 IFK::LogicalOr, // ?W operator||
543 IFK::TimesEqual, // ?X operator*=
544 IFK::PlusEqual, // ?Y operator+=
545 IFK::MinusEqual, // ?Z operator-=
546 };
547 static IFK Under[36] = {
548 IFK::DivEqual, // ?_0 operator/=
549 IFK::ModEqual, // ?_1 operator%=
550 IFK::RshEqual, // ?_2 operator>>=
551 IFK::LshEqual, // ?_3 operator<<=
552 IFK::BitwiseAndEqual, // ?_4 operator&=
553 IFK::BitwiseOrEqual, // ?_5 operator|=
554 IFK::BitwiseXorEqual, // ?_6 operator^=
555 IFK::None, // ?_7 # vftable
556 IFK::None, // ?_8 # vbtable
557 IFK::None, // ?_9 # vcall
558 IFK::None, // ?_A # typeof
559 IFK::None, // ?_B # local static guard
560 IFK::None, // ?_C # string literal
561 IFK::VbaseDtor, // ?_D # vbase destructor
562 IFK::VecDelDtor, // ?_E # vector deleting destructor
563 IFK::DefaultCtorClosure, // ?_F # default constructor closure
564 IFK::ScalarDelDtor, // ?_G # scalar deleting destructor
565 IFK::VecCtorIter, // ?_H # vector constructor iterator
566 IFK::VecDtorIter, // ?_I # vector destructor iterator
567 IFK::VecVbaseCtorIter, // ?_J # vector vbase constructor iterator
568 IFK::VdispMap, // ?_K # virtual displacement map
569 IFK::EHVecCtorIter, // ?_L # eh vector constructor iterator
570 IFK::EHVecDtorIter, // ?_M # eh vector destructor iterator
571 IFK::EHVecVbaseCtorIter, // ?_N # eh vector vbase constructor iterator
572 IFK::CopyCtorClosure, // ?_O # copy constructor closure
573 IFK::None, // ?_P<name> # udt returning <name>
574 IFK::None, // ?_Q # <unknown>
575 IFK::None, // ?_R0 - ?_R4 # RTTI Codes
576 IFK::None, // ?_S # local vftable
577 IFK::LocalVftableCtorClosure, // ?_T # local vftable constructor closure
578 IFK::ArrayNew, // ?_U operator new[]
579 IFK::ArrayDelete, // ?_V operator delete[]
580 IFK::None, // ?_W <unused>
581 IFK::None, // ?_X <unused>
582 IFK::None, // ?_Y <unused>
583 IFK::None, // ?_Z <unused>
584 };
585 static IFK DoubleUnder[36] = {
586 IFK::None, // ?__0 <unused>
587 IFK::None, // ?__1 <unused>
588 IFK::None, // ?__2 <unused>
589 IFK::None, // ?__3 <unused>
590 IFK::None, // ?__4 <unused>
591 IFK::None, // ?__5 <unused>
592 IFK::None, // ?__6 <unused>
593 IFK::None, // ?__7 <unused>
594 IFK::None, // ?__8 <unused>
595 IFK::None, // ?__9 <unused>
596 IFK::ManVectorCtorIter, // ?__A managed vector ctor iterator
597 IFK::ManVectorDtorIter, // ?__B managed vector dtor iterator
598 IFK::EHVectorCopyCtorIter, // ?__C EH vector copy ctor iterator
599 IFK::EHVectorVbaseCopyCtorIter, // ?__D EH vector vbase copy ctor iter
600 IFK::None, // ?__E dynamic initializer for `T'
601 IFK::None, // ?__F dynamic atexit destructor for `T'
602 IFK::VectorCopyCtorIter, // ?__G vector copy constructor iter
603 IFK::VectorVbaseCopyCtorIter, // ?__H vector vbase copy ctor iter
604 IFK::ManVectorVbaseCopyCtorIter, // ?__I managed vector vbase copy ctor
605 // iter
606 IFK::None, // ?__J local static thread guard
607 IFK::None, // ?__K operator ""_name
608 IFK::CoAwait, // ?__L co_await
609 IFK::None, // ?__M <unused>
610 IFK::None, // ?__N <unused>
611 IFK::None, // ?__O <unused>
612 IFK::None, // ?__P <unused>
613 IFK::None, // ?__Q <unused>
614 IFK::None, // ?__R <unused>
615 IFK::None, // ?__S <unused>
616 IFK::None, // ?__T <unused>
617 IFK::None, // ?__U <unused>
618 IFK::None, // ?__V <unused>
619 IFK::None, // ?__W <unused>
620 IFK::None, // ?__X <unused>
621 IFK::None, // ?__Y <unused>
622 IFK::None, // ?__Z <unused>
623 };
624
625 int Index = (CH >= '0' && CH <= '9') ? (CH - '0') : (CH - 'A' + 10);
626 switch (Group) {
627 case FunctionIdentifierCodeGroup::Basic:
628 return Basic[Index];
629 case FunctionIdentifierCodeGroup::Under:
630 return Under[Index];
631 case FunctionIdentifierCodeGroup::DoubleUnder:
632 return DoubleUnder[Index];
633 }
Erik Pilkington5094e5e2019-01-17 20:37:51 +0000634 DEMANGLE_UNREACHABLE;
Zachary Turner03312862018-08-27 03:48:03 +0000635}
636
637IdentifierNode *
638Demangler::demangleFunctionIdentifierCode(StringView &MangledName,
639 FunctionIdentifierCodeGroup Group) {
640 switch (Group) {
641 case FunctionIdentifierCodeGroup::Basic:
642 switch (char CH = MangledName.popFront()) {
643 case '0':
644 case '1':
645 return demangleStructorIdentifier(MangledName, CH == '1');
646 case 'B':
647 return demangleConversionOperatorIdentifier(MangledName);
648 default:
649 return Arena.alloc<IntrinsicFunctionIdentifierNode>(
650 translateIntrinsicFunctionCode(CH, Group));
651 }
652 break;
653 case FunctionIdentifierCodeGroup::Under:
654 return Arena.alloc<IntrinsicFunctionIdentifierNode>(
655 translateIntrinsicFunctionCode(MangledName.popFront(), Group));
656 case FunctionIdentifierCodeGroup::DoubleUnder:
657 switch (char CH = MangledName.popFront()) {
658 case 'K':
659 return demangleLiteralOperatorIdentifier(MangledName);
660 default:
661 return Arena.alloc<IntrinsicFunctionIdentifierNode>(
662 translateIntrinsicFunctionCode(CH, Group));
663 }
664 }
665 // No Mangling Yet: Spaceship, // operator<=>
666
667 return nullptr;
668}
669
670SymbolNode *Demangler::demangleEncodedSymbol(StringView &MangledName,
671 QualifiedNameNode *Name) {
Zachary Turner469f0762018-08-17 21:18:05 +0000672 // Read a variable.
673 switch (MangledName.front()) {
674 case '0':
675 case '1':
676 case '2':
677 case '3':
Zachary Turner03312862018-08-27 03:48:03 +0000678 case '4': {
679 StorageClass SC = demangleVariableStorageClass(MangledName);
680 return demangleVariableEncoding(MangledName, SC);
Zachary Turner469f0762018-08-17 21:18:05 +0000681 }
Zachary Turner03312862018-08-27 03:48:03 +0000682 case '8':
683 return nullptr;
684 }
685 FunctionSymbolNode *FSN = demangleFunctionEncoding(MangledName);
686
687 IdentifierNode *UQN = Name->getUnqualifiedIdentifier();
688 if (UQN->kind() == NodeKind::ConversionOperatorIdentifier) {
689 ConversionOperatorIdentifierNode *COIN =
690 static_cast<ConversionOperatorIdentifierNode *>(UQN);
691 COIN->TargetType = FSN->Signature->ReturnType;
692 }
693 return FSN;
Zachary Turner469f0762018-08-17 21:18:05 +0000694}
695
696// Parser entry point.
Zachary Turner03312862018-08-27 03:48:03 +0000697SymbolNode *Demangler::parse(StringView &MangledName) {
Zachary Turner83313f82018-08-16 16:17:17 +0000698 // We can't demangle MD5 names, just output them as-is.
Zachary Turner469f0762018-08-17 21:18:05 +0000699 // Also, MSVC-style mangled symbols must start with '?'.
Zachary Turner03312862018-08-27 03:48:03 +0000700 if (MangledName.startsWith("??@")) {
701 // This is an MD5 mangled name. We can't demangle it, just return the
702 // mangled name.
703 SymbolNode *S = Arena.alloc<SymbolNode>(NodeKind::Md5Symbol);
704 S->Name = synthesizeQualifiedName(Arena, MangledName);
Zachary Turner83313f82018-08-16 16:17:17 +0000705 return S;
706 }
707
Zachary Turner03312862018-08-27 03:48:03 +0000708 if (!MangledName.startsWith('?')) {
709 Error = true;
710 return nullptr;
711 }
712
Zachary Turner469f0762018-08-17 21:18:05 +0000713 MangledName.consumeFront('?');
Zachary Turner970fdc32018-08-16 16:17:36 +0000714
Zachary Turner469f0762018-08-17 21:18:05 +0000715 // ?$ is a template instantiation, but all other names that start with ? are
716 // operators / special names.
Zachary Turner03312862018-08-27 03:48:03 +0000717 if (SymbolNode *SI = demangleSpecialIntrinsic(MangledName))
718 return SI;
Zachary Turner469f0762018-08-17 21:18:05 +0000719
Zachary Turner469f0762018-08-17 21:18:05 +0000720 // What follows is a main symbol name. This may include namespaces or class
721 // back references.
Zachary Turner03312862018-08-27 03:48:03 +0000722 QualifiedNameNode *QN = demangleFullyQualifiedSymbolName(MangledName);
Zachary Turner32a8a202018-08-29 23:56:09 +0000723 if (Error)
724 return nullptr;
Zachary Turner3461bfa2018-08-17 16:14:05 +0000725
Zachary Turner03312862018-08-27 03:48:03 +0000726 SymbolNode *Symbol = demangleEncodedSymbol(MangledName, QN);
727 if (Symbol) {
728 Symbol->Name = QN;
729 }
Zachary Turner44ebbc22018-08-01 18:32:47 +0000730
Zachary Turner54d4ffe2018-08-01 18:32:28 +0000731 if (Error)
732 return nullptr;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000733
Zachary Turner03312862018-08-27 03:48:03 +0000734 return Symbol;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000735}
736
Zachary Turner7ba90562018-11-01 15:07:32 +0000737TagTypeNode *Demangler::parseTagUniqueName(StringView &MangledName) {
738 if (!MangledName.consumeFront(".?A"))
739 return nullptr;
740 MangledName.consumeFront(".?A");
741 if (MangledName.empty())
742 return nullptr;
743
744 return demangleClassType(MangledName);
745}
746
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000747// <type-encoding> ::= <storage-class> <variable-type>
748// <storage-class> ::= 0 # private static member
749// ::= 1 # protected static member
750// ::= 2 # public static member
751// ::= 3 # global
752// ::= 4 # static local
753
Zachary Turner03312862018-08-27 03:48:03 +0000754VariableSymbolNode *Demangler::demangleVariableEncoding(StringView &MangledName,
755 StorageClass SC) {
756 VariableSymbolNode *VSN = Arena.alloc<VariableSymbolNode>();
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000757
Zachary Turner03312862018-08-27 03:48:03 +0000758 VSN->Type = demangleType(MangledName, QualifierMangleMode::Drop);
759 VSN->SC = SC;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000760
761 // <variable-type> ::= <type> <cvr-qualifiers>
762 // ::= <type> <pointee-cvr-qualifiers> # pointers, references
Zachary Turner03312862018-08-27 03:48:03 +0000763 switch (VSN->Type->kind()) {
764 case NodeKind::PointerType: {
765 PointerTypeNode *PTN = static_cast<PointerTypeNode *>(VSN->Type);
766
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000767 Qualifiers ExtraChildQuals = Q_None;
Zachary Turner03312862018-08-27 03:48:03 +0000768 PTN->Quals = Qualifiers(VSN->Type->Quals |
769 demanglePointerExtQualifiers(MangledName));
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000770
Zachary Turnerd742d642018-07-26 19:56:09 +0000771 bool IsMember = false;
Zachary Turner316109b2018-07-29 16:38:02 +0000772 std::tie(ExtraChildQuals, IsMember) = demangleQualifiers(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000773
Zachary Turner03312862018-08-27 03:48:03 +0000774 if (PTN->ClassParent) {
775 QualifiedNameNode *BackRefName =
776 demangleFullyQualifiedTypeName(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000777 (void)BackRefName;
778 }
Zachary Turner03312862018-08-27 03:48:03 +0000779 PTN->Pointee->Quals = Qualifiers(PTN->Pointee->Quals | ExtraChildQuals);
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000780
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000781 break;
782 }
783 default:
Zachary Turner03312862018-08-27 03:48:03 +0000784 VSN->Type->Quals = demangleQualifiers(MangledName).first;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000785 break;
786 }
787
Zachary Turner03312862018-08-27 03:48:03 +0000788 return VSN;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000789}
790
791// Sometimes numbers are encoded in mangled symbols. For example,
792// "int (*x)[20]" is a valid C type (x is a pointer to an array of
793// length 20), so we need some way to embed numbers as part of symbols.
794// This function parses it.
795//
796// <number> ::= [?] <non-negative integer>
797//
798// <non-negative integer> ::= <decimal digit> # when 1 <= Number <= 10
799// ::= <hex digit>+ @ # when Numbrer == 0 or >= 10
800//
801// <hex-digit> ::= [A-P] # A = 0, B = 1, ...
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000802std::pair<uint64_t, bool> Demangler::demangleNumber(StringView &MangledName) {
803 bool IsNegative = MangledName.consumeFront('?');
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000804
805 if (startsWithDigit(MangledName)) {
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000806 uint64_t Ret = MangledName[0] - '0' + 1;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000807 MangledName = MangledName.dropFront(1);
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000808 return {Ret, IsNegative};
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000809 }
810
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000811 uint64_t Ret = 0;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000812 for (size_t i = 0; i < MangledName.size(); ++i) {
813 char C = MangledName[i];
814 if (C == '@') {
815 MangledName = MangledName.dropFront(i + 1);
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000816 return {Ret, IsNegative};
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000817 }
818 if ('A' <= C && C <= 'P') {
819 Ret = (Ret << 4) + (C - 'A');
820 continue;
821 }
822 break;
823 }
824
825 Error = true;
Zachary Turnerdbefc6c2018-08-10 14:31:04 +0000826 return {0ULL, false};
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000827}
828
Zachary Turner469f0762018-08-17 21:18:05 +0000829uint64_t Demangler::demangleUnsigned(StringView &MangledName) {
830 bool IsNegative = false;
831 uint64_t Number = 0;
832 std::tie(Number, IsNegative) = demangleNumber(MangledName);
833 if (IsNegative)
834 Error = true;
835 return Number;
836}
837
838int64_t Demangler::demangleSigned(StringView &MangledName) {
839 bool IsNegative = false;
840 uint64_t Number = 0;
841 std::tie(Number, IsNegative) = demangleNumber(MangledName);
842 if (Number > INT64_MAX)
843 Error = true;
844 int64_t I = static_cast<int64_t>(Number);
845 return IsNegative ? -I : I;
846}
847
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000848// First 10 strings can be referenced by special BackReferences ?0, ?1, ..., ?9.
849// Memorize it.
850void Demangler::memorizeString(StringView S) {
Zachary Turnerd346cba2018-08-08 17:17:04 +0000851 if (Backrefs.NamesCount >= BackrefContext::Max)
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000852 return;
Zachary Turnerd346cba2018-08-08 17:17:04 +0000853 for (size_t i = 0; i < Backrefs.NamesCount; ++i)
Zachary Turner03312862018-08-27 03:48:03 +0000854 if (S == Backrefs.Names[i]->Name)
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000855 return;
Zachary Turner03312862018-08-27 03:48:03 +0000856 NamedIdentifierNode *N = Arena.alloc<NamedIdentifierNode>();
857 N->Name = S;
858 Backrefs.Names[Backrefs.NamesCount++] = N;
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000859}
860
Zachary Turner03312862018-08-27 03:48:03 +0000861NamedIdentifierNode *Demangler::demangleBackRefName(StringView &MangledName) {
Zachary Turnera7dffb12018-07-28 22:10:42 +0000862 assert(startsWithDigit(MangledName));
Zachary Turnerd9e925f2018-08-18 18:49:48 +0000863
864 size_t I = MangledName[0] - '0';
865 if (I >= Backrefs.NamesCount) {
866 Error = true;
867 return nullptr;
868 }
869
Zachary Turner172aea12018-08-02 17:08:03 +0000870 MangledName = MangledName.dropFront();
Zachary Turner03312862018-08-27 03:48:03 +0000871 return Backrefs.Names[I];
Zachary Turnerf435a7e2018-07-20 17:27:48 +0000872}
873
Zachary Turnerb2fef1a2018-08-29 04:12:44 +0000874void Demangler::memorizeIdentifier(IdentifierNode *Identifier) {
875 // Render this class template name into a string buffer so that we can
876 // memorize it for the purpose of back-referencing.
Nico Weber1359d652018-09-15 18:24:20 +0000877 OutputStream OS;
Nico Weber6808bc02018-11-11 10:04:00 +0000878 if (!initializeOutputStream(nullptr, nullptr, OS, 1024))
Nico Weber1359d652018-09-15 18:24:20 +0000879 // FIXME: Propagate out-of-memory as an error?
880 std::terminate();
Zachary Turnerb2fef1a2018-08-29 04:12:44 +0000881 Identifier->output(OS, OF_Default);
882 OS << '\0';
883 char *Name = OS.getBuffer();
884
885 StringView Owned = copyString(Name);
886 memorizeString(Owned);
887 std::free(Name);
888}
889
Zachary Turner03312862018-08-27 03:48:03 +0000890IdentifierNode *
891Demangler::demangleTemplateInstantiationName(StringView &MangledName,
892 NameBackrefBehavior NBB) {
Zachary Turnera7dffb12018-07-28 22:10:42 +0000893 assert(MangledName.startsWith("?$"));
894 MangledName.consumeFront("?$");
895
Zachary Turnerd346cba2018-08-08 17:17:04 +0000896 BackrefContext OuterContext;
897 std::swap(OuterContext, Backrefs);
Zachary Turner54d4ffe2018-08-01 18:32:28 +0000898
Zachary Turner03312862018-08-27 03:48:03 +0000899 IdentifierNode *Identifier =
900 demangleUnqualifiedSymbolName(MangledName, NBB_Simple);
Zachary Turnerd346cba2018-08-08 17:17:04 +0000901 if (!Error)
Zachary Turner03312862018-08-27 03:48:03 +0000902 Identifier->TemplateParams = demangleTemplateParameterList(MangledName);
Zachary Turnerd346cba2018-08-08 17:17:04 +0000903
904 std::swap(OuterContext, Backrefs);
Zachary Turner54d4ffe2018-08-01 18:32:28 +0000905 if (Error)
906 return nullptr;
Zachary Turner71c91f92018-07-30 03:12:34 +0000907
Zachary Turnerb2fef1a2018-08-29 04:12:44 +0000908 if (NBB & NBB_Template)
909 memorizeIdentifier(Identifier);
Zachary Turner71c91f92018-07-30 03:12:34 +0000910
Zachary Turner03312862018-08-27 03:48:03 +0000911 return Identifier;
Zachary Turnera7dffb12018-07-28 22:10:42 +0000912}
913
Zachary Turner03312862018-08-27 03:48:03 +0000914NamedIdentifierNode *Demangler::demangleSimpleName(StringView &MangledName,
915 bool Memorize) {
Zachary Turner931e8792018-07-30 23:02:10 +0000916 StringView S = demangleSimpleString(MangledName, Memorize);
917 if (Error)
918 return nullptr;
919
Zachary Turner03312862018-08-27 03:48:03 +0000920 NamedIdentifierNode *Name = Arena.alloc<NamedIdentifierNode>();
921 Name->Name = S;
922 return Name;
Zachary Turner931e8792018-07-30 23:02:10 +0000923}
924
Zachary Turner970fdc32018-08-16 16:17:36 +0000925static bool isRebasedHexDigit(char C) { return (C >= 'A' && C <= 'P'); }
926
927static uint8_t rebasedHexDigitToNumber(char C) {
928 assert(isRebasedHexDigit(C));
929 return (C <= 'J') ? (C - 'A') : (10 + C - 'K');
930}
931
932uint8_t Demangler::demangleCharLiteral(StringView &MangledName) {
933 if (!MangledName.startsWith('?'))
934 return MangledName.popFront();
935
936 MangledName = MangledName.dropFront();
937 if (MangledName.empty())
938 goto CharLiteralError;
939
940 if (MangledName.consumeFront('$')) {
941 // Two hex digits
942 if (MangledName.size() < 2)
943 goto CharLiteralError;
944 StringView Nibbles = MangledName.substr(0, 2);
945 if (!isRebasedHexDigit(Nibbles[0]) || !isRebasedHexDigit(Nibbles[1]))
946 goto CharLiteralError;
947 // Don't append the null terminator.
948 uint8_t C1 = rebasedHexDigitToNumber(Nibbles[0]);
949 uint8_t C2 = rebasedHexDigitToNumber(Nibbles[1]);
950 MangledName = MangledName.dropFront(2);
951 return (C1 << 4) | C2;
952 }
953
954 if (startsWithDigit(MangledName)) {
955 const char *Lookup = ",/\\:. \n\t'-";
956 char C = Lookup[MangledName[0] - '0'];
957 MangledName = MangledName.dropFront();
958 return C;
959 }
960
961 if (MangledName[0] >= 'a' && MangledName[0] <= 'z') {
962 char Lookup[26] = {'\xE1', '\xE2', '\xE3', '\xE4', '\xE5', '\xE6', '\xE7',
963 '\xE8', '\xE9', '\xEA', '\xEB', '\xEC', '\xED', '\xEE',
964 '\xEF', '\xF0', '\xF1', '\xF2', '\xF3', '\xF4', '\xF5',
965 '\xF6', '\xF7', '\xF8', '\xF9', '\xFA'};
966 char C = Lookup[MangledName[0] - 'a'];
967 MangledName = MangledName.dropFront();
968 return C;
969 }
970
971 if (MangledName[0] >= 'A' && MangledName[0] <= 'Z') {
972 char Lookup[26] = {'\xC1', '\xC2', '\xC3', '\xC4', '\xC5', '\xC6', '\xC7',
973 '\xC8', '\xC9', '\xCA', '\xCB', '\xCC', '\xCD', '\xCE',
974 '\xCF', '\xD0', '\xD1', '\xD2', '\xD3', '\xD4', '\xD5',
975 '\xD6', '\xD7', '\xD8', '\xD9', '\xDA'};
976 char C = Lookup[MangledName[0] - 'A'];
977 MangledName = MangledName.dropFront();
978 return C;
979 }
980
981CharLiteralError:
982 Error = true;
983 return '\0';
984}
985
986wchar_t Demangler::demangleWcharLiteral(StringView &MangledName) {
Zachary Turnerd78fe2f2018-08-16 16:30:27 +0000987 uint8_t C1, C2;
988
989 C1 = demangleCharLiteral(MangledName);
Zachary Turner970fdc32018-08-16 16:17:36 +0000990 if (Error)
991 goto WCharLiteralError;
Zachary Turnerd78fe2f2018-08-16 16:30:27 +0000992 C2 = demangleCharLiteral(MangledName);
Zachary Turner970fdc32018-08-16 16:17:36 +0000993 if (Error)
994 goto WCharLiteralError;
995
996 return ((wchar_t)C1 << 8) | (wchar_t)C2;
997
998WCharLiteralError:
999 Error = true;
1000 return L'\0';
1001}
1002
1003static void writeHexDigit(char *Buffer, uint8_t Digit) {
1004 assert(Digit <= 15);
1005 *Buffer = (Digit < 10) ? ('0' + Digit) : ('A' + Digit - 10);
1006}
1007
1008static void outputHex(OutputStream &OS, unsigned C) {
1009 if (C == 0) {
1010 OS << "\\x00";
1011 return;
1012 }
1013 // It's easier to do the math if we can work from right to left, but we need
1014 // to print the numbers from left to right. So render this into a temporary
1015 // buffer first, then output the temporary buffer. Each byte is of the form
1016 // \xAB, which means that each byte needs 4 characters. Since there are at
1017 // most 4 bytes, we need a 4*4+1 = 17 character temporary buffer.
1018 char TempBuffer[17];
1019
1020 ::memset(TempBuffer, 0, sizeof(TempBuffer));
1021 constexpr int MaxPos = 15;
1022
1023 int Pos = MaxPos - 1;
1024 while (C != 0) {
1025 for (int I = 0; I < 2; ++I) {
1026 writeHexDigit(&TempBuffer[Pos--], C % 16);
1027 C /= 16;
1028 }
1029 TempBuffer[Pos--] = 'x';
1030 TempBuffer[Pos--] = '\\';
1031 assert(Pos >= 0);
1032 }
1033 OS << StringView(&TempBuffer[Pos + 1]);
1034}
1035
1036static void outputEscapedChar(OutputStream &OS, unsigned C) {
1037 switch (C) {
1038 case '\'': // single quote
1039 OS << "\\\'";
1040 return;
1041 case '\"': // double quote
1042 OS << "\\\"";
1043 return;
1044 case '\\': // backslash
1045 OS << "\\\\";
1046 return;
1047 case '\a': // bell
1048 OS << "\\a";
1049 return;
1050 case '\b': // backspace
1051 OS << "\\b";
1052 return;
1053 case '\f': // form feed
1054 OS << "\\f";
1055 return;
1056 case '\n': // new line
1057 OS << "\\n";
1058 return;
1059 case '\r': // carriage return
1060 OS << "\\r";
1061 return;
1062 case '\t': // tab
1063 OS << "\\t";
1064 return;
1065 case '\v': // vertical tab
1066 OS << "\\v";
1067 return;
1068 default:
1069 break;
1070 }
1071
1072 if (C > 0x1F && C < 0x7F) {
1073 // Standard ascii char.
1074 OS << (char)C;
1075 return;
1076 }
1077
1078 outputHex(OS, C);
1079}
1080
Benjamin Kramerc55e9972018-10-13 22:18:22 +00001081static unsigned countTrailingNullBytes(const uint8_t *StringBytes, int Length) {
Zachary Turner970fdc32018-08-16 16:17:36 +00001082 const uint8_t *End = StringBytes + Length - 1;
Zachary Turner3461bfa2018-08-17 16:14:05 +00001083 unsigned Count = 0;
Zachary Turner970fdc32018-08-16 16:17:36 +00001084 while (Length > 0 && *End == 0) {
1085 --Length;
1086 --End;
Zachary Turner3461bfa2018-08-17 16:14:05 +00001087 ++Count;
Zachary Turner970fdc32018-08-16 16:17:36 +00001088 }
Zachary Turner3461bfa2018-08-17 16:14:05 +00001089 return Count;
Zachary Turner970fdc32018-08-16 16:17:36 +00001090}
1091
Benjamin Kramerc55e9972018-10-13 22:18:22 +00001092static unsigned countEmbeddedNulls(const uint8_t *StringBytes,
1093 unsigned Length) {
Zachary Turner970fdc32018-08-16 16:17:36 +00001094 unsigned Result = 0;
1095 for (unsigned I = 0; I < Length; ++I) {
1096 if (*StringBytes++ == 0)
1097 ++Result;
1098 }
1099 return Result;
1100}
1101
Benjamin Kramerc55e9972018-10-13 22:18:22 +00001102static unsigned guessCharByteSize(const uint8_t *StringBytes, unsigned NumChars,
1103 unsigned NumBytes) {
Zachary Turner970fdc32018-08-16 16:17:36 +00001104 assert(NumBytes > 0);
1105
1106 // If the number of bytes is odd, this is guaranteed to be a char string.
1107 if (NumBytes % 2 == 1)
1108 return 1;
1109
1110 // All strings can encode at most 32 bytes of data. If it's less than that,
1111 // then we encoded the entire string. In this case we check for a 1-byte,
1112 // 2-byte, or 4-byte null terminator.
1113 if (NumBytes < 32) {
1114 unsigned TrailingNulls = countTrailingNullBytes(StringBytes, NumChars);
1115 if (TrailingNulls >= 4)
1116 return 4;
1117 if (TrailingNulls >= 2)
1118 return 2;
1119 return 1;
1120 }
1121
1122 // The whole string was not able to be encoded. Try to look at embedded null
1123 // terminators to guess. The heuristic is that we count all embedded null
1124 // terminators. If more than 2/3 are null, it's a char32. If more than 1/3
1125 // are null, it's a char16. Otherwise it's a char8. This obviously isn't
1126 // perfect and is biased towards languages that have ascii alphabets, but this
1127 // was always going to be best effort since the encoding is lossy.
1128 unsigned Nulls = countEmbeddedNulls(StringBytes, NumChars);
1129 if (Nulls >= 2 * NumChars / 3)
1130 return 4;
1131 if (Nulls >= NumChars / 3)
1132 return 2;
1133 return 1;
1134}
1135
1136static unsigned decodeMultiByteChar(const uint8_t *StringBytes,
1137 unsigned CharIndex, unsigned CharBytes) {
1138 assert(CharBytes == 1 || CharBytes == 2 || CharBytes == 4);
1139 unsigned Offset = CharIndex * CharBytes;
1140 unsigned Result = 0;
1141 StringBytes = StringBytes + Offset;
1142 for (unsigned I = 0; I < CharBytes; ++I) {
1143 unsigned C = static_cast<unsigned>(StringBytes[I]);
1144 Result |= C << (8 * I);
1145 }
1146 return Result;
1147}
1148
Zachary Turner03312862018-08-27 03:48:03 +00001149FunctionSymbolNode *Demangler::demangleVcallThunkNode(StringView &MangledName) {
1150 FunctionSymbolNode *FSN = Arena.alloc<FunctionSymbolNode>();
1151 VcallThunkIdentifierNode *VTIN = Arena.alloc<VcallThunkIdentifierNode>();
1152 FSN->Signature = Arena.alloc<ThunkSignatureNode>();
1153 FSN->Signature->FunctionClass = FC_NoParameterList;
1154
1155 FSN->Name = demangleNameScopeChain(MangledName, VTIN);
1156 if (!Error)
1157 Error = !MangledName.consumeFront("$B");
1158 if (!Error)
1159 VTIN->OffsetInVTable = demangleUnsigned(MangledName);
1160 if (!Error)
1161 Error = !MangledName.consumeFront('A');
1162 if (!Error)
1163 FSN->Signature->CallConvention = demangleCallingConvention(MangledName);
1164 return (Error) ? nullptr : FSN;
1165}
1166
1167EncodedStringLiteralNode *
1168Demangler::demangleStringLiteral(StringView &MangledName) {
Zachary Turnerd78fe2f2018-08-16 16:30:27 +00001169 // This function uses goto, so declare all variables up front.
Zachary Turner970fdc32018-08-16 16:17:36 +00001170 OutputStream OS;
1171 StringView CRC;
Zachary Turnerd78fe2f2018-08-16 16:30:27 +00001172 uint64_t StringByteSize;
1173 bool IsWcharT = false;
1174 bool IsNegative = false;
1175 size_t CrcEndPos = 0;
1176 char *ResultBuffer = nullptr;
1177
Zachary Turner03312862018-08-27 03:48:03 +00001178 EncodedStringLiteralNode *Result = Arena.alloc<EncodedStringLiteralNode>();
Zachary Turner970fdc32018-08-16 16:17:36 +00001179
1180 // Prefix indicating the beginning of a string literal
Zachary Turner3461bfa2018-08-17 16:14:05 +00001181 if (!MangledName.consumeFront("@_"))
1182 goto StringLiteralError;
Zachary Turner970fdc32018-08-16 16:17:36 +00001183 if (MangledName.empty())
1184 goto StringLiteralError;
1185
1186 // Char Type (regular or wchar_t)
Zachary Turner970fdc32018-08-16 16:17:36 +00001187 switch (MangledName.popFront()) {
1188 case '1':
1189 IsWcharT = true;
Erik Pilkington5094e5e2019-01-17 20:37:51 +00001190 DEMANGLE_FALLTHROUGH;
Zachary Turner970fdc32018-08-16 16:17:36 +00001191 case '0':
1192 break;
1193 default:
1194 goto StringLiteralError;
1195 }
1196
1197 // Encoded Length
Zachary Turner970fdc32018-08-16 16:17:36 +00001198 std::tie(StringByteSize, IsNegative) = demangleNumber(MangledName);
1199 if (Error || IsNegative)
1200 goto StringLiteralError;
1201
1202 // CRC 32 (always 8 characters plus a terminator)
Zachary Turnerd78fe2f2018-08-16 16:30:27 +00001203 CrcEndPos = MangledName.find('@');
Zachary Turner970fdc32018-08-16 16:17:36 +00001204 if (CrcEndPos == StringView::npos)
1205 goto StringLiteralError;
1206 CRC = MangledName.substr(0, CrcEndPos);
1207 MangledName = MangledName.dropFront(CrcEndPos + 1);
1208 if (MangledName.empty())
1209 goto StringLiteralError;
1210
Nico Weber6808bc02018-11-11 10:04:00 +00001211 if (!initializeOutputStream(nullptr, nullptr, OS, 1024))
Nico Weber1359d652018-09-15 18:24:20 +00001212 // FIXME: Propagate out-of-memory as an error?
1213 std::terminate();
Zachary Turner970fdc32018-08-16 16:17:36 +00001214 if (IsWcharT) {
Zachary Turner03312862018-08-27 03:48:03 +00001215 Result->Char = CharKind::Wchar;
Zachary Turner970fdc32018-08-16 16:17:36 +00001216 if (StringByteSize > 64)
Zachary Turner3461bfa2018-08-17 16:14:05 +00001217 Result->IsTruncated = true;
Zachary Turner970fdc32018-08-16 16:17:36 +00001218
1219 while (!MangledName.consumeFront('@')) {
1220 assert(StringByteSize >= 2);
1221 wchar_t W = demangleWcharLiteral(MangledName);
Zachary Turner3461bfa2018-08-17 16:14:05 +00001222 if (StringByteSize != 2 || Result->IsTruncated)
Zachary Turner970fdc32018-08-16 16:17:36 +00001223 outputEscapedChar(OS, W);
1224 StringByteSize -= 2;
1225 if (Error)
1226 goto StringLiteralError;
1227 }
1228 } else {
Zachary Turner32a8a202018-08-29 23:56:09 +00001229 // The max byte length is actually 32, but some compilers mangled strings
1230 // incorrectly, so we have to assume it can go higher.
1231 constexpr unsigned MaxStringByteLength = 32 * 4;
Zachary Turner970fdc32018-08-16 16:17:36 +00001232 uint8_t StringBytes[MaxStringByteLength];
1233
1234 unsigned BytesDecoded = 0;
1235 while (!MangledName.consumeFront('@')) {
1236 assert(StringByteSize >= 1);
1237 StringBytes[BytesDecoded++] = demangleCharLiteral(MangledName);
1238 }
1239
Zachary Turner32a8a202018-08-29 23:56:09 +00001240 if (StringByteSize > BytesDecoded)
1241 Result->IsTruncated = true;
1242
Zachary Turner970fdc32018-08-16 16:17:36 +00001243 unsigned CharBytes =
1244 guessCharByteSize(StringBytes, BytesDecoded, StringByteSize);
1245 assert(StringByteSize % CharBytes == 0);
1246 switch (CharBytes) {
1247 case 1:
Zachary Turner03312862018-08-27 03:48:03 +00001248 Result->Char = CharKind::Char;
Zachary Turner970fdc32018-08-16 16:17:36 +00001249 break;
1250 case 2:
Zachary Turner03312862018-08-27 03:48:03 +00001251 Result->Char = CharKind::Char16;
Zachary Turner970fdc32018-08-16 16:17:36 +00001252 break;
1253 case 4:
Zachary Turner03312862018-08-27 03:48:03 +00001254 Result->Char = CharKind::Char32;
Zachary Turner970fdc32018-08-16 16:17:36 +00001255 break;
1256 default:
Erik Pilkington5094e5e2019-01-17 20:37:51 +00001257 DEMANGLE_UNREACHABLE;
Zachary Turner970fdc32018-08-16 16:17:36 +00001258 }
1259 const unsigned NumChars = BytesDecoded / CharBytes;
1260 for (unsigned CharIndex = 0; CharIndex < NumChars; ++CharIndex) {
1261 unsigned NextChar =
1262 decodeMultiByteChar(StringBytes, CharIndex, CharBytes);
Zachary Turner3461bfa2018-08-17 16:14:05 +00001263 if (CharIndex + 1 < NumChars || Result->IsTruncated)
Zachary Turner970fdc32018-08-16 16:17:36 +00001264 outputEscapedChar(OS, NextChar);
1265 }
1266 }
1267
1268 OS << '\0';
Zachary Turnerd78fe2f2018-08-16 16:30:27 +00001269 ResultBuffer = OS.getBuffer();
Zachary Turner03312862018-08-27 03:48:03 +00001270 Result->DecodedString = copyString(ResultBuffer);
Zachary Turneraf738f72018-08-16 17:48:32 +00001271 std::free(ResultBuffer);
Zachary Turner970fdc32018-08-16 16:17:36 +00001272 return Result;
1273
1274StringLiteralError:
1275 Error = true;
1276 return nullptr;
1277}
1278
Nico Weberc7444dd2019-04-03 23:15:56 +00001279// Returns MangledName's prefix before the first '@', or an error if
1280// MangledName contains no '@' or the prefix has length 0.
Zachary Turner931e8792018-07-30 23:02:10 +00001281StringView Demangler::demangleSimpleString(StringView &MangledName,
1282 bool Memorize) {
1283 StringView S;
Zachary Turnera7dffb12018-07-28 22:10:42 +00001284 for (size_t i = 0; i < MangledName.size(); ++i) {
1285 if (MangledName[i] != '@')
1286 continue;
Nico Weberc7444dd2019-04-03 23:15:56 +00001287 if (i == 0)
1288 break;
Zachary Turner931e8792018-07-30 23:02:10 +00001289 S = MangledName.substr(0, i);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001290 MangledName = MangledName.dropFront(i + 1);
1291
1292 if (Memorize)
Zachary Turner931e8792018-07-30 23:02:10 +00001293 memorizeString(S);
1294 return S;
Zachary Turnera7dffb12018-07-28 22:10:42 +00001295 }
1296
1297 Error = true;
Zachary Turner931e8792018-07-30 23:02:10 +00001298 return {};
Zachary Turnera7dffb12018-07-28 22:10:42 +00001299}
1300
Zachary Turner03312862018-08-27 03:48:03 +00001301NamedIdentifierNode *
1302Demangler::demangleAnonymousNamespaceName(StringView &MangledName) {
Zachary Turnera7dffb12018-07-28 22:10:42 +00001303 assert(MangledName.startsWith("?A"));
1304 MangledName.consumeFront("?A");
1305
Zachary Turner03312862018-08-27 03:48:03 +00001306 NamedIdentifierNode *Node = Arena.alloc<NamedIdentifierNode>();
1307 Node->Name = "`anonymous namespace'";
Zachary Turner91c98a82018-08-20 23:58:35 +00001308 size_t EndPos = MangledName.find('@');
1309 if (EndPos == StringView::npos) {
1310 Error = true;
1311 return nullptr;
1312 }
Zachary Turner0002dd42018-08-20 23:58:58 +00001313 StringView NamespaceKey = MangledName.substr(0, EndPos);
1314 memorizeString(NamespaceKey);
Zachary Turner91c98a82018-08-20 23:58:35 +00001315 MangledName = MangledName.substr(EndPos + 1);
1316 return Node;
Zachary Turnera7dffb12018-07-28 22:10:42 +00001317}
1318
Zachary Turner03312862018-08-27 03:48:03 +00001319NamedIdentifierNode *
1320Demangler::demangleLocallyScopedNamePiece(StringView &MangledName) {
Zachary Turner71c91f92018-07-30 03:12:34 +00001321 assert(startsWithLocalScopePattern(MangledName));
1322
Zachary Turner03312862018-08-27 03:48:03 +00001323 NamedIdentifierNode *Identifier = Arena.alloc<NamedIdentifierNode>();
Zachary Turner71c91f92018-07-30 03:12:34 +00001324 MangledName.consumeFront('?');
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001325 auto Number = demangleNumber(MangledName);
1326 assert(!Number.second);
Zachary Turner71c91f92018-07-30 03:12:34 +00001327
1328 // One ? to terminate the number
1329 MangledName.consumeFront('?');
1330
1331 assert(!Error);
Zachary Turner03312862018-08-27 03:48:03 +00001332 Node *Scope = parse(MangledName);
Zachary Turner71c91f92018-07-30 03:12:34 +00001333 if (Error)
1334 return nullptr;
1335
1336 // Render the parent symbol's name into a buffer.
Nico Weber1359d652018-09-15 18:24:20 +00001337 OutputStream OS;
Nico Weber6808bc02018-11-11 10:04:00 +00001338 if (!initializeOutputStream(nullptr, nullptr, OS, 1024))
Nico Weber1359d652018-09-15 18:24:20 +00001339 // FIXME: Propagate out-of-memory as an error?
1340 std::terminate();
Zachary Turner71c91f92018-07-30 03:12:34 +00001341 OS << '`';
Zachary Turner38d2edd2018-08-29 03:59:17 +00001342 Scope->output(OS, OF_Default);
Zachary Turner71c91f92018-07-30 03:12:34 +00001343 OS << '\'';
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001344 OS << "::`" << Number.first << "'";
Zachary Turner71c91f92018-07-30 03:12:34 +00001345 OS << '\0';
1346 char *Result = OS.getBuffer();
Zachary Turner03312862018-08-27 03:48:03 +00001347 Identifier->Name = copyString(Result);
Zachary Turner71c91f92018-07-30 03:12:34 +00001348 std::free(Result);
Zachary Turner03312862018-08-27 03:48:03 +00001349 return Identifier;
Zachary Turner71c91f92018-07-30 03:12:34 +00001350}
1351
Zachary Turnera7dffb12018-07-28 22:10:42 +00001352// Parses a type name in the form of A@B@C@@ which represents C::B::A.
Zachary Turner03312862018-08-27 03:48:03 +00001353QualifiedNameNode *
1354Demangler::demangleFullyQualifiedTypeName(StringView &MangledName) {
1355 IdentifierNode *Identifier = demangleUnqualifiedTypeName(MangledName, true);
Zachary Turner54d4ffe2018-08-01 18:32:28 +00001356 if (Error)
1357 return nullptr;
Zachary Turner03312862018-08-27 03:48:03 +00001358 assert(Identifier);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001359
Zachary Turner03312862018-08-27 03:48:03 +00001360 QualifiedNameNode *QN = demangleNameScopeChain(MangledName, Identifier);
Zachary Turner54d4ffe2018-08-01 18:32:28 +00001361 if (Error)
1362 return nullptr;
Zachary Turner03312862018-08-27 03:48:03 +00001363 assert(QN);
1364 return QN;
Zachary Turnera7dffb12018-07-28 22:10:42 +00001365}
1366
1367// Parses a symbol name in the form of A@B@C@@ which represents C::B::A.
1368// Symbol names have slightly different rules regarding what can appear
1369// so we separate out the implementations for flexibility.
Zachary Turner03312862018-08-27 03:48:03 +00001370QualifiedNameNode *
1371Demangler::demangleFullyQualifiedSymbolName(StringView &MangledName) {
Zachary Turner58d29cf2018-08-08 00:43:31 +00001372 // This is the final component of a symbol name (i.e. the leftmost component
1373 // of a mangled name. Since the only possible template instantiation that
1374 // can appear in this context is a function template, and since those are
1375 // not saved for the purposes of name backreferences, only backref simple
1376 // names.
Zachary Turner03312862018-08-27 03:48:03 +00001377 IdentifierNode *Identifier =
1378 demangleUnqualifiedSymbolName(MangledName, NBB_Simple);
Zachary Turner54d4ffe2018-08-01 18:32:28 +00001379 if (Error)
1380 return nullptr;
Zachary Turner3461bfa2018-08-17 16:14:05 +00001381
Zachary Turner03312862018-08-27 03:48:03 +00001382 QualifiedNameNode *QN = demangleNameScopeChain(MangledName, Identifier);
Zachary Turner54d4ffe2018-08-01 18:32:28 +00001383 if (Error)
1384 return nullptr;
Zachary Turner03312862018-08-27 03:48:03 +00001385
1386 if (Identifier->kind() == NodeKind::StructorIdentifier) {
1387 StructorIdentifierNode *SIN =
1388 static_cast<StructorIdentifierNode *>(Identifier);
1389 assert(QN->Components->Count >= 2);
1390 Node *ClassNode = QN->Components->Nodes[QN->Components->Count - 2];
1391 SIN->Class = static_cast<IdentifierNode *>(ClassNode);
1392 }
1393 assert(QN);
1394 return QN;
Zachary Turnera7dffb12018-07-28 22:10:42 +00001395}
1396
Zachary Turner03312862018-08-27 03:48:03 +00001397IdentifierNode *Demangler::demangleUnqualifiedTypeName(StringView &MangledName,
1398 bool Memorize) {
Zachary Turnera7dffb12018-07-28 22:10:42 +00001399 // An inner-most name can be a back-reference, because a fully-qualified name
1400 // (e.g. Scope + Inner) can contain other fully qualified names inside of
1401 // them (for example template parameters), and these nested parameters can
1402 // refer to previously mangled types.
1403 if (startsWithDigit(MangledName))
Zachary Turner316109b2018-07-29 16:38:02 +00001404 return demangleBackRefName(MangledName);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001405
1406 if (MangledName.startsWith("?$"))
Zachary Turner58d29cf2018-08-08 00:43:31 +00001407 return demangleTemplateInstantiationName(MangledName, NBB_Template);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001408
Zachary Turner44ebbc22018-08-01 18:32:47 +00001409 return demangleSimpleName(MangledName, Memorize);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001410}
1411
Zachary Turner03312862018-08-27 03:48:03 +00001412IdentifierNode *
1413Demangler::demangleUnqualifiedSymbolName(StringView &MangledName,
1414 NameBackrefBehavior NBB) {
Zachary Turner71c91f92018-07-30 03:12:34 +00001415 if (startsWithDigit(MangledName))
1416 return demangleBackRefName(MangledName);
1417 if (MangledName.startsWith("?$"))
Zachary Turner58d29cf2018-08-08 00:43:31 +00001418 return demangleTemplateInstantiationName(MangledName, NBB);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001419 if (MangledName.startsWith('?'))
Zachary Turner03312862018-08-27 03:48:03 +00001420 return demangleFunctionIdentifierCode(MangledName);
Zachary Turner58d29cf2018-08-08 00:43:31 +00001421 return demangleSimpleName(MangledName, (NBB & NBB_Simple) != 0);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001422}
1423
Zachary Turner03312862018-08-27 03:48:03 +00001424IdentifierNode *Demangler::demangleNameScopePiece(StringView &MangledName) {
Zachary Turnera7dffb12018-07-28 22:10:42 +00001425 if (startsWithDigit(MangledName))
Zachary Turner316109b2018-07-29 16:38:02 +00001426 return demangleBackRefName(MangledName);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001427
1428 if (MangledName.startsWith("?$"))
Zachary Turner58d29cf2018-08-08 00:43:31 +00001429 return demangleTemplateInstantiationName(MangledName, NBB_Template);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001430
1431 if (MangledName.startsWith("?A"))
Zachary Turner316109b2018-07-29 16:38:02 +00001432 return demangleAnonymousNamespaceName(MangledName);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001433
Zachary Turner71c91f92018-07-30 03:12:34 +00001434 if (startsWithLocalScopePattern(MangledName))
1435 return demangleLocallyScopedNamePiece(MangledName);
1436
Zachary Turner316109b2018-07-29 16:38:02 +00001437 return demangleSimpleName(MangledName, true);
Zachary Turnera7dffb12018-07-28 22:10:42 +00001438}
1439
Zachary Turner03312862018-08-27 03:48:03 +00001440static NodeArrayNode *nodeListToNodeArray(ArenaAllocator &Arena, NodeList *Head,
1441 size_t Count) {
1442 NodeArrayNode *N = Arena.alloc<NodeArrayNode>();
1443 N->Count = Count;
1444 N->Nodes = Arena.allocArray<Node *>(Count);
1445 for (size_t I = 0; I < Count; ++I) {
1446 N->Nodes[I] = Head->N;
1447 Head = Head->Next;
1448 }
1449 return N;
1450}
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001451
Zachary Turner03312862018-08-27 03:48:03 +00001452QualifiedNameNode *
1453Demangler::demangleNameScopeChain(StringView &MangledName,
1454 IdentifierNode *UnqualifiedName) {
1455 NodeList *Head = Arena.alloc<NodeList>();
1456
1457 Head->N = UnqualifiedName;
1458
1459 size_t Count = 1;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001460 while (!MangledName.consumeFront("@")) {
Zachary Turner03312862018-08-27 03:48:03 +00001461 ++Count;
1462 NodeList *NewHead = Arena.alloc<NodeList>();
1463 NewHead->Next = Head;
1464 Head = NewHead;
1465
Zachary Turnera7dffb12018-07-28 22:10:42 +00001466 if (MangledName.empty()) {
1467 Error = true;
1468 return nullptr;
1469 }
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001470
1471 assert(!Error);
Zachary Turner03312862018-08-27 03:48:03 +00001472 IdentifierNode *Elem = demangleNameScopePiece(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001473 if (Error)
1474 return nullptr;
1475
Zachary Turner03312862018-08-27 03:48:03 +00001476 Head->N = Elem;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001477 }
Zachary Turner03312862018-08-27 03:48:03 +00001478
1479 QualifiedNameNode *QN = Arena.alloc<QualifiedNameNode>();
1480 QN->Components = nodeListToNodeArray(Arena, Head, Count);
1481 return QN;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001482}
1483
Zachary Turner316109b2018-07-29 16:38:02 +00001484FuncClass Demangler::demangleFunctionClass(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001485 switch (MangledName.popFront()) {
Zachary Turner29ec67b2018-08-10 21:09:05 +00001486 case '9':
Zachary Turner03312862018-08-27 03:48:03 +00001487 return FuncClass(FC_ExternC | FC_NoParameterList);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001488 case 'A':
Zachary Turner03312862018-08-27 03:48:03 +00001489 return FC_Private;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001490 case 'B':
Zachary Turner03312862018-08-27 03:48:03 +00001491 return FuncClass(FC_Private | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001492 case 'C':
Zachary Turner03312862018-08-27 03:48:03 +00001493 return FuncClass(FC_Private | FC_Static);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001494 case 'D':
Zachary Turner03312862018-08-27 03:48:03 +00001495 return FuncClass(FC_Private | FC_Static);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001496 case 'E':
Zachary Turner03312862018-08-27 03:48:03 +00001497 return FuncClass(FC_Private | FC_Virtual);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001498 case 'F':
Zachary Turner03312862018-08-27 03:48:03 +00001499 return FuncClass(FC_Private | FC_Virtual);
Zachary Turner32a8a202018-08-29 23:56:09 +00001500 case 'G':
1501 return FuncClass(FC_Private | FC_StaticThisAdjust);
1502 case 'H':
1503 return FuncClass(FC_Private | FC_StaticThisAdjust | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001504 case 'I':
Zachary Turner03312862018-08-27 03:48:03 +00001505 return FuncClass(FC_Protected);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001506 case 'J':
Zachary Turner03312862018-08-27 03:48:03 +00001507 return FuncClass(FC_Protected | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001508 case 'K':
Zachary Turner03312862018-08-27 03:48:03 +00001509 return FuncClass(FC_Protected | FC_Static);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001510 case 'L':
Zachary Turner03312862018-08-27 03:48:03 +00001511 return FuncClass(FC_Protected | FC_Static | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001512 case 'M':
Zachary Turner03312862018-08-27 03:48:03 +00001513 return FuncClass(FC_Protected | FC_Virtual);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001514 case 'N':
Zachary Turner03312862018-08-27 03:48:03 +00001515 return FuncClass(FC_Protected | FC_Virtual | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001516 case 'O':
Zachary Turner03312862018-08-27 03:48:03 +00001517 return FuncClass(FC_Protected | FC_Virtual | FC_StaticThisAdjust);
Zachary Turner469f0762018-08-17 21:18:05 +00001518 case 'P':
Zachary Turner03312862018-08-27 03:48:03 +00001519 return FuncClass(FC_Protected | FC_Virtual | FC_StaticThisAdjust | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001520 case 'Q':
Zachary Turner03312862018-08-27 03:48:03 +00001521 return FuncClass(FC_Public);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001522 case 'R':
Zachary Turner03312862018-08-27 03:48:03 +00001523 return FuncClass(FC_Public | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001524 case 'S':
Zachary Turner03312862018-08-27 03:48:03 +00001525 return FuncClass(FC_Public | FC_Static);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001526 case 'T':
Zachary Turner03312862018-08-27 03:48:03 +00001527 return FuncClass(FC_Public | FC_Static | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001528 case 'U':
Zachary Turner03312862018-08-27 03:48:03 +00001529 return FuncClass(FC_Public | FC_Virtual);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001530 case 'V':
Zachary Turner03312862018-08-27 03:48:03 +00001531 return FuncClass(FC_Public | FC_Virtual | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001532 case 'W':
Zachary Turner03312862018-08-27 03:48:03 +00001533 return FuncClass(FC_Public | FC_Virtual | FC_StaticThisAdjust);
Zachary Turner469f0762018-08-17 21:18:05 +00001534 case 'X':
Zachary Turner03312862018-08-27 03:48:03 +00001535 return FuncClass(FC_Public | FC_Virtual | FC_StaticThisAdjust | FC_Far);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001536 case 'Y':
Zachary Turner03312862018-08-27 03:48:03 +00001537 return FuncClass(FC_Global);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001538 case 'Z':
Zachary Turner03312862018-08-27 03:48:03 +00001539 return FuncClass(FC_Global | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001540 case '$': {
Zachary Turner03312862018-08-27 03:48:03 +00001541 FuncClass VFlag = FC_VirtualThisAdjust;
Zachary Turner469f0762018-08-17 21:18:05 +00001542 if (MangledName.consumeFront('R'))
Zachary Turner03312862018-08-27 03:48:03 +00001543 VFlag = FuncClass(VFlag | FC_VirtualThisAdjustEx);
Zachary Turner469f0762018-08-17 21:18:05 +00001544
1545 switch (MangledName.popFront()) {
1546 case '0':
Zachary Turner03312862018-08-27 03:48:03 +00001547 return FuncClass(FC_Private | FC_Virtual | VFlag);
Zachary Turner469f0762018-08-17 21:18:05 +00001548 case '1':
Zachary Turner03312862018-08-27 03:48:03 +00001549 return FuncClass(FC_Private | FC_Virtual | VFlag | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001550 case '2':
Zachary Turner03312862018-08-27 03:48:03 +00001551 return FuncClass(FC_Protected | FC_Virtual | VFlag);
Zachary Turner469f0762018-08-17 21:18:05 +00001552 case '3':
Zachary Turner03312862018-08-27 03:48:03 +00001553 return FuncClass(FC_Protected | FC_Virtual | VFlag | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001554 case '4':
Zachary Turner03312862018-08-27 03:48:03 +00001555 return FuncClass(FC_Public | FC_Virtual | VFlag);
Zachary Turner469f0762018-08-17 21:18:05 +00001556 case '5':
Zachary Turner03312862018-08-27 03:48:03 +00001557 return FuncClass(FC_Public | FC_Virtual | VFlag | FC_Far);
Zachary Turner469f0762018-08-17 21:18:05 +00001558 }
1559 }
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001560 }
1561
1562 Error = true;
Zachary Turner03312862018-08-27 03:48:03 +00001563 return FC_Public;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001564}
1565
Zachary Turner316109b2018-07-29 16:38:02 +00001566CallingConv Demangler::demangleCallingConvention(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001567 switch (MangledName.popFront()) {
1568 case 'A':
1569 case 'B':
1570 return CallingConv::Cdecl;
1571 case 'C':
1572 case 'D':
1573 return CallingConv::Pascal;
1574 case 'E':
1575 case 'F':
1576 return CallingConv::Thiscall;
1577 case 'G':
1578 case 'H':
1579 return CallingConv::Stdcall;
1580 case 'I':
1581 case 'J':
1582 return CallingConv::Fastcall;
1583 case 'M':
1584 case 'N':
1585 return CallingConv::Clrcall;
1586 case 'O':
1587 case 'P':
1588 return CallingConv::Eabi;
1589 case 'Q':
1590 return CallingConv::Vectorcall;
1591 }
1592
1593 return CallingConv::None;
Martin Storsjo0f2abd82018-07-20 18:43:42 +00001594}
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001595
Zachary Turner316109b2018-07-29 16:38:02 +00001596StorageClass Demangler::demangleVariableStorageClass(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001597 assert(std::isdigit(MangledName.front()));
1598
1599 switch (MangledName.popFront()) {
1600 case '0':
1601 return StorageClass::PrivateStatic;
1602 case '1':
1603 return StorageClass::ProtectedStatic;
1604 case '2':
1605 return StorageClass::PublicStatic;
1606 case '3':
1607 return StorageClass::Global;
1608 case '4':
1609 return StorageClass::FunctionLocalStatic;
1610 }
1611 Error = true;
1612 return StorageClass::None;
1613}
1614
Zachary Turner316109b2018-07-29 16:38:02 +00001615std::pair<Qualifiers, bool>
1616Demangler::demangleQualifiers(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001617
1618 switch (MangledName.popFront()) {
Zachary Turnerd742d642018-07-26 19:56:09 +00001619 // Member qualifiers
1620 case 'Q':
1621 return std::make_pair(Q_None, true);
1622 case 'R':
1623 return std::make_pair(Q_Const, true);
1624 case 'S':
1625 return std::make_pair(Q_Volatile, true);
1626 case 'T':
1627 return std::make_pair(Qualifiers(Q_Const | Q_Volatile), true);
1628 // Non-Member qualifiers
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001629 case 'A':
Zachary Turnerd742d642018-07-26 19:56:09 +00001630 return std::make_pair(Q_None, false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001631 case 'B':
Zachary Turnerd742d642018-07-26 19:56:09 +00001632 return std::make_pair(Q_Const, false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001633 case 'C':
Zachary Turnerd742d642018-07-26 19:56:09 +00001634 return std::make_pair(Q_Volatile, false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001635 case 'D':
Zachary Turnerd742d642018-07-26 19:56:09 +00001636 return std::make_pair(Qualifiers(Q_Const | Q_Volatile), false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001637 }
1638 Error = true;
Zachary Turnerd742d642018-07-26 19:56:09 +00001639 return std::make_pair(Q_None, false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001640}
1641
1642// <variable-type> ::= <type> <cvr-qualifiers>
1643// ::= <type> <pointee-cvr-qualifiers> # pointers, references
Zachary Turner03312862018-08-27 03:48:03 +00001644TypeNode *Demangler::demangleType(StringView &MangledName,
1645 QualifierMangleMode QMM) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001646 Qualifiers Quals = Q_None;
Zachary Turnerd742d642018-07-26 19:56:09 +00001647 bool IsMember = false;
Zachary Turnerd742d642018-07-26 19:56:09 +00001648 if (QMM == QualifierMangleMode::Mangle) {
Zachary Turner316109b2018-07-29 16:38:02 +00001649 std::tie(Quals, IsMember) = demangleQualifiers(MangledName);
Zachary Turnerd742d642018-07-26 19:56:09 +00001650 } else if (QMM == QualifierMangleMode::Result) {
Zachary Turnerdf4cd7c2018-08-21 21:23:49 +00001651 if (MangledName.consumeFront('?'))
Zachary Turner316109b2018-07-29 16:38:02 +00001652 std::tie(Quals, IsMember) = demangleQualifiers(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001653 }
1654
Zachary Turner03312862018-08-27 03:48:03 +00001655 TypeNode *Ty = nullptr;
Zachary Turner931e8792018-07-30 23:02:10 +00001656 if (isTagType(MangledName))
Zachary Turner316109b2018-07-29 16:38:02 +00001657 Ty = demangleClassType(MangledName);
Zachary Turner931e8792018-07-30 23:02:10 +00001658 else if (isPointerType(MangledName)) {
Zachary Turner8fb9a712018-12-14 18:10:13 +00001659 if (isMemberPointer(MangledName, Error))
Zachary Turner316109b2018-07-29 16:38:02 +00001660 Ty = demangleMemberPointerType(MangledName);
Zachary Turner8fb9a712018-12-14 18:10:13 +00001661 else if (!Error)
Zachary Turner316109b2018-07-29 16:38:02 +00001662 Ty = demanglePointerType(MangledName);
Zachary Turner8fb9a712018-12-14 18:10:13 +00001663 else
1664 return nullptr;
Zachary Turner931e8792018-07-30 23:02:10 +00001665 } else if (isArrayType(MangledName))
Zachary Turner316109b2018-07-29 16:38:02 +00001666 Ty = demangleArrayType(MangledName);
Zachary Turner931e8792018-07-30 23:02:10 +00001667 else if (isFunctionType(MangledName)) {
1668 if (MangledName.consumeFront("$$A8@@"))
Zachary Turner03312862018-08-27 03:48:03 +00001669 Ty = demangleFunctionType(MangledName, true);
Zachary Turner931e8792018-07-30 23:02:10 +00001670 else {
1671 assert(MangledName.startsWith("$$A6"));
1672 MangledName.consumeFront("$$A6");
Zachary Turner03312862018-08-27 03:48:03 +00001673 Ty = demangleFunctionType(MangledName, false);
Zachary Turner931e8792018-07-30 23:02:10 +00001674 }
Zachary Turnerb2fef1a2018-08-29 04:12:44 +00001675 } else if (isCustomType(MangledName)) {
1676 Ty = demangleCustomType(MangledName);
Zachary Turner931e8792018-07-30 23:02:10 +00001677 } else {
Zachary Turner03312862018-08-27 03:48:03 +00001678 Ty = demanglePrimitiveType(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001679 }
Zachary Turner931e8792018-07-30 23:02:10 +00001680
Zachary Turner2cd32862018-12-14 17:43:56 +00001681 if (!Ty || Error)
1682 return Ty;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001683 Ty->Quals = Qualifiers(Ty->Quals | Quals);
1684 return Ty;
1685}
1686
Zachary Turner2fe49002019-01-08 21:05:51 +00001687bool Demangler::demangleThrowSpecification(StringView &MangledName) {
1688 if (MangledName.consumeFront("_E"))
1689 return true;
Zachary Turner38b78a72018-07-26 20:20:10 +00001690 if (MangledName.consumeFront('Z'))
Zachary Turner2fe49002019-01-08 21:05:51 +00001691 return false;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001692
Zachary Turner38b78a72018-07-26 20:20:10 +00001693 Error = true;
Zachary Turner2fe49002019-01-08 21:05:51 +00001694 return false;
Zachary Turner38b78a72018-07-26 20:20:10 +00001695}
1696
Zachary Turner03312862018-08-27 03:48:03 +00001697FunctionSignatureNode *Demangler::demangleFunctionType(StringView &MangledName,
1698 bool HasThisQuals) {
1699 FunctionSignatureNode *FTy = Arena.alloc<FunctionSignatureNode>();
Zachary Turner38b78a72018-07-26 20:20:10 +00001700
1701 if (HasThisQuals) {
Zachary Turner316109b2018-07-29 16:38:02 +00001702 FTy->Quals = demanglePointerExtQualifiers(MangledName);
Zachary Turner03312862018-08-27 03:48:03 +00001703 FTy->RefQualifier = demangleFunctionRefQualifier(MangledName);
Zachary Turner316109b2018-07-29 16:38:02 +00001704 FTy->Quals = Qualifiers(FTy->Quals | demangleQualifiers(MangledName).first);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001705 }
1706
1707 // Fields that appear on both member and non-member functions.
Zachary Turner316109b2018-07-29 16:38:02 +00001708 FTy->CallConvention = demangleCallingConvention(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001709
1710 // <return-type> ::= <type>
1711 // ::= @ # structors (they have no declared return type)
1712 bool IsStructor = MangledName.consumeFront('@');
1713 if (!IsStructor)
Zachary Turner316109b2018-07-29 16:38:02 +00001714 FTy->ReturnType = demangleType(MangledName, QualifierMangleMode::Result);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001715
Zachary Turner316109b2018-07-29 16:38:02 +00001716 FTy->Params = demangleFunctionParameterList(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001717
Zachary Turner2fe49002019-01-08 21:05:51 +00001718 FTy->IsNoexcept = demangleThrowSpecification(MangledName);
Zachary Turner38b78a72018-07-26 20:20:10 +00001719
1720 return FTy;
1721}
1722
Zachary Turner03312862018-08-27 03:48:03 +00001723FunctionSymbolNode *
1724Demangler::demangleFunctionEncoding(StringView &MangledName) {
1725 FuncClass ExtraFlags = FC_None;
Zachary Turner469f0762018-08-17 21:18:05 +00001726 if (MangledName.consumeFront("$$J0"))
Zachary Turner03312862018-08-27 03:48:03 +00001727 ExtraFlags = FC_ExternC;
Zachary Turner469f0762018-08-17 21:18:05 +00001728
Zachary Turner316109b2018-07-29 16:38:02 +00001729 FuncClass FC = demangleFunctionClass(MangledName);
Zachary Turner469f0762018-08-17 21:18:05 +00001730 FC = FuncClass(ExtraFlags | FC);
1731
Zachary Turner03312862018-08-27 03:48:03 +00001732 FunctionSignatureNode *FSN = nullptr;
1733 ThunkSignatureNode *TTN = nullptr;
1734 if (FC & FC_StaticThisAdjust) {
1735 TTN = Arena.alloc<ThunkSignatureNode>();
1736 TTN->ThisAdjust.StaticOffset = demangleSigned(MangledName);
1737 } else if (FC & FC_VirtualThisAdjust) {
1738 TTN = Arena.alloc<ThunkSignatureNode>();
1739 if (FC & FC_VirtualThisAdjustEx) {
1740 TTN->ThisAdjust.VBPtrOffset = demangleSigned(MangledName);
1741 TTN->ThisAdjust.VBOffsetOffset = demangleSigned(MangledName);
Zachary Turner469f0762018-08-17 21:18:05 +00001742 }
Zachary Turner03312862018-08-27 03:48:03 +00001743 TTN->ThisAdjust.VtordispOffset = demangleSigned(MangledName);
1744 TTN->ThisAdjust.StaticOffset = demangleSigned(MangledName);
Zachary Turner469f0762018-08-17 21:18:05 +00001745 }
1746
Zachary Turner03312862018-08-27 03:48:03 +00001747 if (FC & FC_NoParameterList) {
Zachary Turner29ec67b2018-08-10 21:09:05 +00001748 // This is an extern "C" function whose full signature hasn't been mangled.
1749 // This happens when we need to mangle a local symbol inside of an extern
1750 // "C" function.
Zachary Turner03312862018-08-27 03:48:03 +00001751 FSN = Arena.alloc<FunctionSignatureNode>();
Zachary Turner29ec67b2018-08-10 21:09:05 +00001752 } else {
Zachary Turner03312862018-08-27 03:48:03 +00001753 bool HasThisQuals = !(FC & (FC_Global | FC_Static));
1754 FSN = demangleFunctionType(MangledName, HasThisQuals);
Zachary Turner29ec67b2018-08-10 21:09:05 +00001755 }
Zachary Turner03312862018-08-27 03:48:03 +00001756 if (TTN) {
1757 *static_cast<FunctionSignatureNode *>(TTN) = *FSN;
1758 FSN = TTN;
1759 }
1760 FSN->FunctionClass = FC;
Zachary Turner38b78a72018-07-26 20:20:10 +00001761
Zachary Turner03312862018-08-27 03:48:03 +00001762 FunctionSymbolNode *Symbol = Arena.alloc<FunctionSymbolNode>();
1763 Symbol->Signature = FSN;
1764 return Symbol;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001765}
1766
Zachary Turnerb2fef1a2018-08-29 04:12:44 +00001767CustomTypeNode *Demangler::demangleCustomType(StringView &MangledName) {
1768 assert(MangledName.startsWith('?'));
1769 MangledName.popFront();
1770
1771 CustomTypeNode *CTN = Arena.alloc<CustomTypeNode>();
1772 CTN->Identifier = demangleUnqualifiedTypeName(MangledName, true);
1773 if (!MangledName.consumeFront('@'))
1774 Error = true;
1775 if (Error)
1776 return nullptr;
1777 return CTN;
1778}
1779
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001780// Reads a primitive type.
Zachary Turner03312862018-08-27 03:48:03 +00001781PrimitiveTypeNode *Demangler::demanglePrimitiveType(StringView &MangledName) {
1782 if (MangledName.consumeFront("$$T"))
1783 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Nullptr);
Zachary Turner931e8792018-07-30 23:02:10 +00001784
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001785 switch (MangledName.popFront()) {
1786 case 'X':
Zachary Turner03312862018-08-27 03:48:03 +00001787 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Void);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001788 case 'D':
Zachary Turner03312862018-08-27 03:48:03 +00001789 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001790 case 'C':
Zachary Turner03312862018-08-27 03:48:03 +00001791 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Schar);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001792 case 'E':
Zachary Turner03312862018-08-27 03:48:03 +00001793 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uchar);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001794 case 'F':
Zachary Turner03312862018-08-27 03:48:03 +00001795 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Short);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001796 case 'G':
Zachary Turner03312862018-08-27 03:48:03 +00001797 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ushort);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001798 case 'H':
Zachary Turner03312862018-08-27 03:48:03 +00001799 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Int);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001800 case 'I':
Zachary Turner03312862018-08-27 03:48:03 +00001801 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uint);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001802 case 'J':
Zachary Turner03312862018-08-27 03:48:03 +00001803 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Long);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001804 case 'K':
Zachary Turner03312862018-08-27 03:48:03 +00001805 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ulong);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001806 case 'M':
Zachary Turner03312862018-08-27 03:48:03 +00001807 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Float);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001808 case 'N':
Zachary Turner03312862018-08-27 03:48:03 +00001809 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Double);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001810 case 'O':
Zachary Turner03312862018-08-27 03:48:03 +00001811 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Ldouble);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001812 case '_': {
Zachary Turner91ecedd2018-07-20 18:07:33 +00001813 if (MangledName.empty()) {
1814 Error = true;
1815 return nullptr;
1816 }
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001817 switch (MangledName.popFront()) {
1818 case 'N':
Zachary Turner03312862018-08-27 03:48:03 +00001819 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Bool);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001820 case 'J':
Zachary Turner03312862018-08-27 03:48:03 +00001821 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Int64);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001822 case 'K':
Zachary Turner03312862018-08-27 03:48:03 +00001823 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Uint64);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001824 case 'W':
Zachary Turner03312862018-08-27 03:48:03 +00001825 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Wchar);
Zachary Turner931e8792018-07-30 23:02:10 +00001826 case 'S':
Zachary Turner03312862018-08-27 03:48:03 +00001827 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char16);
Zachary Turner931e8792018-07-30 23:02:10 +00001828 case 'U':
Zachary Turner03312862018-08-27 03:48:03 +00001829 return Arena.alloc<PrimitiveTypeNode>(PrimitiveKind::Char32);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001830 }
1831 break;
1832 }
1833 }
Zachary Turner03312862018-08-27 03:48:03 +00001834 Error = true;
1835 return nullptr;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001836}
1837
Zachary Turner03312862018-08-27 03:48:03 +00001838TagTypeNode *Demangler::demangleClassType(StringView &MangledName) {
1839 TagTypeNode *TT = nullptr;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001840
1841 switch (MangledName.popFront()) {
1842 case 'T':
Zachary Turner03312862018-08-27 03:48:03 +00001843 TT = Arena.alloc<TagTypeNode>(TagKind::Union);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001844 break;
1845 case 'U':
Zachary Turner03312862018-08-27 03:48:03 +00001846 TT = Arena.alloc<TagTypeNode>(TagKind::Struct);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001847 break;
1848 case 'V':
Zachary Turner03312862018-08-27 03:48:03 +00001849 TT = Arena.alloc<TagTypeNode>(TagKind::Class);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001850 break;
1851 case 'W':
1852 if (MangledName.popFront() != '4') {
1853 Error = true;
1854 return nullptr;
1855 }
Zachary Turner03312862018-08-27 03:48:03 +00001856 TT = Arena.alloc<TagTypeNode>(TagKind::Enum);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001857 break;
1858 default:
1859 assert(false);
1860 }
1861
Zachary Turner03312862018-08-27 03:48:03 +00001862 TT->QualifiedName = demangleFullyQualifiedTypeName(MangledName);
1863 return TT;
Zachary Turnerd742d642018-07-26 19:56:09 +00001864}
1865
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001866// <pointer-type> ::= E? <pointer-cvr-qualifiers> <ext-qualifiers> <type>
1867// # the E is required for 64-bit non-static pointers
Zachary Turner03312862018-08-27 03:48:03 +00001868PointerTypeNode *Demangler::demanglePointerType(StringView &MangledName) {
1869 PointerTypeNode *Pointer = Arena.alloc<PointerTypeNode>();
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001870
Zachary Turner931e8792018-07-30 23:02:10 +00001871 std::tie(Pointer->Quals, Pointer->Affinity) =
1872 demanglePointerCVQualifiers(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001873
1874 if (MangledName.consumeFront("6")) {
Zachary Turner03312862018-08-27 03:48:03 +00001875 Pointer->Pointee = demangleFunctionType(MangledName, false);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001876 return Pointer;
1877 }
1878
Zachary Turner316109b2018-07-29 16:38:02 +00001879 Qualifiers ExtQuals = demanglePointerExtQualifiers(MangledName);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001880 Pointer->Quals = Qualifiers(Pointer->Quals | ExtQuals);
1881
Zachary Turner316109b2018-07-29 16:38:02 +00001882 Pointer->Pointee = demangleType(MangledName, QualifierMangleMode::Mangle);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001883 return Pointer;
1884}
1885
Zachary Turner03312862018-08-27 03:48:03 +00001886PointerTypeNode *Demangler::demangleMemberPointerType(StringView &MangledName) {
1887 PointerTypeNode *Pointer = Arena.alloc<PointerTypeNode>();
Zachary Turnerd742d642018-07-26 19:56:09 +00001888
Zachary Turner03312862018-08-27 03:48:03 +00001889 std::tie(Pointer->Quals, Pointer->Affinity) =
1890 demanglePointerCVQualifiers(MangledName);
1891 assert(Pointer->Affinity == PointerAffinity::Pointer);
Zachary Turnerd742d642018-07-26 19:56:09 +00001892
Zachary Turner316109b2018-07-29 16:38:02 +00001893 Qualifiers ExtQuals = demanglePointerExtQualifiers(MangledName);
Zachary Turnerd742d642018-07-26 19:56:09 +00001894 Pointer->Quals = Qualifiers(Pointer->Quals | ExtQuals);
1895
Zachary Turner38b78a72018-07-26 20:20:10 +00001896 if (MangledName.consumeFront("8")) {
Zachary Turner32a8a202018-08-29 23:56:09 +00001897 Pointer->ClassParent = demangleFullyQualifiedTypeName(MangledName);
Zachary Turner03312862018-08-27 03:48:03 +00001898 Pointer->Pointee = demangleFunctionType(MangledName, true);
Zachary Turner38b78a72018-07-26 20:20:10 +00001899 } else {
1900 Qualifiers PointeeQuals = Q_None;
1901 bool IsMember = false;
Zachary Turner316109b2018-07-29 16:38:02 +00001902 std::tie(PointeeQuals, IsMember) = demangleQualifiers(MangledName);
Zachary Turner38b78a72018-07-26 20:20:10 +00001903 assert(IsMember);
Zachary Turner32a8a202018-08-29 23:56:09 +00001904 Pointer->ClassParent = demangleFullyQualifiedTypeName(MangledName);
Zachary Turnerd742d642018-07-26 19:56:09 +00001905
Zachary Turner316109b2018-07-29 16:38:02 +00001906 Pointer->Pointee = demangleType(MangledName, QualifierMangleMode::Drop);
Zachary Turner38b78a72018-07-26 20:20:10 +00001907 Pointer->Pointee->Quals = PointeeQuals;
1908 }
1909
Zachary Turnerd742d642018-07-26 19:56:09 +00001910 return Pointer;
1911}
1912
Zachary Turner316109b2018-07-29 16:38:02 +00001913Qualifiers Demangler::demanglePointerExtQualifiers(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001914 Qualifiers Quals = Q_None;
1915 if (MangledName.consumeFront('E'))
1916 Quals = Qualifiers(Quals | Q_Pointer64);
1917 if (MangledName.consumeFront('I'))
1918 Quals = Qualifiers(Quals | Q_Restrict);
1919 if (MangledName.consumeFront('F'))
1920 Quals = Qualifiers(Quals | Q_Unaligned);
1921
1922 return Quals;
1923}
1924
Zachary Turner03312862018-08-27 03:48:03 +00001925ArrayTypeNode *Demangler::demangleArrayType(StringView &MangledName) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001926 assert(MangledName.front() == 'Y');
1927 MangledName.popFront();
1928
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001929 uint64_t Rank = 0;
1930 bool IsNegative = false;
1931 std::tie(Rank, IsNegative) = demangleNumber(MangledName);
1932 if (IsNegative || Rank == 0) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001933 Error = true;
1934 return nullptr;
1935 }
1936
Zachary Turner03312862018-08-27 03:48:03 +00001937 ArrayTypeNode *ATy = Arena.alloc<ArrayTypeNode>();
1938 NodeList *Head = Arena.alloc<NodeList>();
1939 NodeList *Tail = Head;
1940
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001941 for (uint64_t I = 0; I < Rank; ++I) {
Zachary Turner03312862018-08-27 03:48:03 +00001942 uint64_t D = 0;
1943 std::tie(D, IsNegative) = demangleNumber(MangledName);
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001944 if (IsNegative) {
1945 Error = true;
1946 return nullptr;
1947 }
Zachary Turner03312862018-08-27 03:48:03 +00001948 Tail->N = Arena.alloc<IntegerLiteralNode>(D, IsNegative);
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001949 if (I + 1 < Rank) {
Zachary Turner03312862018-08-27 03:48:03 +00001950 Tail->Next = Arena.alloc<NodeList>();
1951 Tail = Tail->Next;
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00001952 }
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001953 }
Zachary Turner03312862018-08-27 03:48:03 +00001954 ATy->Dimensions = nodeListToNodeArray(Arena, Head, Rank);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001955
1956 if (MangledName.consumeFront("$$C")) {
Zachary Turner2bbb23b2018-08-14 18:54:28 +00001957 bool IsMember = false;
1958 std::tie(ATy->Quals, IsMember) = demangleQualifiers(MangledName);
1959 if (IsMember) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001960 Error = true;
Zachary Turner2bbb23b2018-08-14 18:54:28 +00001961 return nullptr;
1962 }
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001963 }
1964
Zachary Turner316109b2018-07-29 16:38:02 +00001965 ATy->ElementType = demangleType(MangledName, QualifierMangleMode::Drop);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001966 return ATy;
1967}
1968
1969// Reads a function or a template parameters.
Zachary Turner03312862018-08-27 03:48:03 +00001970NodeArrayNode *
Zachary Turnerd30700f2018-07-31 17:16:44 +00001971Demangler::demangleFunctionParameterList(StringView &MangledName) {
Zachary Turner38b78a72018-07-26 20:20:10 +00001972 // Empty parameter list.
Zachary Turner38b78a72018-07-26 20:20:10 +00001973 if (MangledName.consumeFront('X'))
1974 return {};
1975
Zachary Turner03312862018-08-27 03:48:03 +00001976 NodeList *Head = Arena.alloc<NodeList>();
1977 NodeList **Current = &Head;
1978 size_t Count = 0;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001979 while (!Error && !MangledName.startsWith('@') &&
1980 !MangledName.startsWith('Z')) {
Zachary Turner03312862018-08-27 03:48:03 +00001981 ++Count;
Zachary Turner23df1312018-07-26 22:13:39 +00001982
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001983 if (startsWithDigit(MangledName)) {
Zachary Turner30375de2018-07-26 22:24:01 +00001984 size_t N = MangledName[0] - '0';
Zachary Turnerd346cba2018-08-08 17:17:04 +00001985 if (N >= Backrefs.FunctionParamCount) {
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001986 Error = true;
1987 return {};
1988 }
1989 MangledName = MangledName.dropFront();
1990
Zachary Turner03312862018-08-27 03:48:03 +00001991 *Current = Arena.alloc<NodeList>();
1992 (*Current)->N = Backrefs.FunctionParams[N];
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001993 Current = &(*Current)->Next;
1994 continue;
1995 }
1996
Zachary Turner23df1312018-07-26 22:13:39 +00001997 size_t OldSize = MangledName.size();
Zachary Turnerf435a7e2018-07-20 17:27:48 +00001998
Zachary Turner03312862018-08-27 03:48:03 +00001999 *Current = Arena.alloc<NodeList>();
2000 TypeNode *TN = demangleType(MangledName, QualifierMangleMode::Drop);
Zachary Turner8fb9a712018-12-14 18:10:13 +00002001 if (!TN || Error)
2002 return nullptr;
Zachary Turner03312862018-08-27 03:48:03 +00002003
2004 (*Current)->N = TN;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002005
Zachary Turner23df1312018-07-26 22:13:39 +00002006 size_t CharsConsumed = OldSize - MangledName.size();
2007 assert(CharsConsumed != 0);
2008
2009 // Single-letter types are ignored for backreferences because memorizing
2010 // them doesn't save anything.
Zachary Turnerd346cba2018-08-08 17:17:04 +00002011 if (Backrefs.FunctionParamCount <= 9 && CharsConsumed > 1)
Zachary Turner03312862018-08-27 03:48:03 +00002012 Backrefs.FunctionParams[Backrefs.FunctionParamCount++] = TN;
Zachary Turner23df1312018-07-26 22:13:39 +00002013
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002014 Current = &(*Current)->Next;
2015 }
2016
Zachary Turner38b78a72018-07-26 20:20:10 +00002017 if (Error)
2018 return {};
2019
Zachary Turner03312862018-08-27 03:48:03 +00002020 NodeArrayNode *NA = nodeListToNodeArray(Arena, Head, Count);
Zachary Turner38b78a72018-07-26 20:20:10 +00002021 // A non-empty parameter list is terminated by either 'Z' (variadic) parameter
2022 // list or '@' (non variadic). Careful not to consume "@Z", as in that case
2023 // the following Z could be a throw specifier.
2024 if (MangledName.consumeFront('@'))
Zachary Turner03312862018-08-27 03:48:03 +00002025 return NA;
Zachary Turner38b78a72018-07-26 20:20:10 +00002026
2027 if (MangledName.consumeFront('Z')) {
Zachary Turner03312862018-08-27 03:48:03 +00002028 // This is a variadic parameter list. We probably need a variadic node to
2029 // append to the end.
2030 return NA;
Zachary Turner38b78a72018-07-26 20:20:10 +00002031 }
2032
2033 Error = true;
2034 return {};
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002035}
2036
Zachary Turner03312862018-08-27 03:48:03 +00002037NodeArrayNode *
Zachary Turnerd30700f2018-07-31 17:16:44 +00002038Demangler::demangleTemplateParameterList(StringView &MangledName) {
Zachary Turner03312862018-08-27 03:48:03 +00002039 NodeList *Head;
2040 NodeList **Current = &Head;
2041 size_t Count = 0;
2042
Zachary Turner23df1312018-07-26 22:13:39 +00002043 while (!Error && !MangledName.startsWith('@')) {
Zachary Turner32a8a202018-08-29 23:56:09 +00002044 if (MangledName.consumeFront("$S") || MangledName.consumeFront("$$V") ||
Zachary Turner78ab3cb2018-08-30 20:53:29 +00002045 MangledName.consumeFront("$$$V") || MangledName.consumeFront("$$Z")) {
2046 // parameter pack separator
Zachary Turner32a8a202018-08-29 23:56:09 +00002047 continue;
2048 }
2049
Zachary Turner03312862018-08-27 03:48:03 +00002050 ++Count;
Zachary Turner32a8a202018-08-29 23:56:09 +00002051
Zachary Turner23df1312018-07-26 22:13:39 +00002052 // Template parameter lists don't participate in back-referencing.
Zachary Turner03312862018-08-27 03:48:03 +00002053 *Current = Arena.alloc<NodeList>();
Zachary Turner931e8792018-07-30 23:02:10 +00002054
Zachary Turner03312862018-08-27 03:48:03 +00002055 NodeList &TP = **Current;
Zachary Turner66555a72018-08-20 19:15:35 +00002056
Zachary Turner03312862018-08-27 03:48:03 +00002057 TemplateParameterReferenceNode *TPRN = nullptr;
Zachary Turner32a8a202018-08-29 23:56:09 +00002058 if (MangledName.consumeFront("$$Y")) {
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00002059 // Template alias
Zachary Turner03312862018-08-27 03:48:03 +00002060 TP.N = demangleFullyQualifiedTypeName(MangledName);
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00002061 } else if (MangledName.consumeFront("$$B")) {
2062 // Array
Zachary Turner03312862018-08-27 03:48:03 +00002063 TP.N = demangleType(MangledName, QualifierMangleMode::Drop);
Zachary Turner073620b2018-08-10 19:57:36 +00002064 } else if (MangledName.consumeFront("$$C")) {
2065 // Type has qualifiers.
Zachary Turner03312862018-08-27 03:48:03 +00002066 TP.N = demangleType(MangledName, QualifierMangleMode::Mangle);
Zachary Turner66555a72018-08-20 19:15:35 +00002067 } else if (MangledName.startsWith("$1") || MangledName.startsWith("$H") ||
2068 MangledName.startsWith("$I") || MangledName.startsWith("$J")) {
Zachary Turner03312862018-08-27 03:48:03 +00002069 // Pointer to member
2070 TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2071 TPRN->IsMemberPointer = true;
2072
Zachary Turner66555a72018-08-20 19:15:35 +00002073 MangledName = MangledName.dropFront();
2074 // 1 - single inheritance <name>
2075 // H - multiple inheritance <name> <number>
2076 // I - virtual inheritance <name> <number> <number> <number>
2077 // J - unspecified inheritance <name> <number> <number> <number>
2078 char InheritanceSpecifier = MangledName.popFront();
Zachary Turnerb2fef1a2018-08-29 04:12:44 +00002079 SymbolNode *S = nullptr;
2080 if (MangledName.startsWith('?')) {
2081 S = parse(MangledName);
2082 memorizeIdentifier(S->Name->getUnqualifiedIdentifier());
2083 }
2084
Zachary Turner66555a72018-08-20 19:15:35 +00002085 switch (InheritanceSpecifier) {
2086 case 'J':
Zachary Turner03312862018-08-27 03:48:03 +00002087 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2088 demangleSigned(MangledName);
Erik Pilkington5094e5e2019-01-17 20:37:51 +00002089 DEMANGLE_FALLTHROUGH;
Zachary Turner66555a72018-08-20 19:15:35 +00002090 case 'I':
Zachary Turner03312862018-08-27 03:48:03 +00002091 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2092 demangleSigned(MangledName);
Erik Pilkington5094e5e2019-01-17 20:37:51 +00002093 DEMANGLE_FALLTHROUGH;
Zachary Turner66555a72018-08-20 19:15:35 +00002094 case 'H':
Zachary Turner03312862018-08-27 03:48:03 +00002095 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2096 demangleSigned(MangledName);
Erik Pilkington5094e5e2019-01-17 20:37:51 +00002097 DEMANGLE_FALLTHROUGH;
Zachary Turner66555a72018-08-20 19:15:35 +00002098 case '1':
2099 break;
2100 default:
2101 Error = true;
2102 break;
2103 }
Zachary Turner03312862018-08-27 03:48:03 +00002104 TPRN->Affinity = PointerAffinity::Pointer;
2105 TPRN->Symbol = S;
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00002106 } else if (MangledName.startsWith("$E?")) {
2107 MangledName.consumeFront("$E");
2108 // Reference to symbol
Zachary Turner03312862018-08-27 03:48:03 +00002109 TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2110 TPRN->Symbol = parse(MangledName);
2111 TPRN->Affinity = PointerAffinity::Reference;
Zachary Turner66555a72018-08-20 19:15:35 +00002112 } else if (MangledName.startsWith("$F") || MangledName.startsWith("$G")) {
Zachary Turner03312862018-08-27 03:48:03 +00002113 TP.N = TPRN = Arena.alloc<TemplateParameterReferenceNode>();
2114
Zachary Turner66555a72018-08-20 19:15:35 +00002115 // Data member pointer.
2116 MangledName = MangledName.dropFront();
2117 char InheritanceSpecifier = MangledName.popFront();
2118
2119 switch (InheritanceSpecifier) {
2120 case 'G':
Zachary Turner03312862018-08-27 03:48:03 +00002121 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2122 demangleSigned(MangledName);
Erik Pilkington5094e5e2019-01-17 20:37:51 +00002123 DEMANGLE_FALLTHROUGH;
Zachary Turner66555a72018-08-20 19:15:35 +00002124 case 'F':
Zachary Turner03312862018-08-27 03:48:03 +00002125 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2126 demangleSigned(MangledName);
2127 TPRN->ThunkOffsets[TPRN->ThunkOffsetCount++] =
2128 demangleSigned(MangledName);
Erik Pilkington5094e5e2019-01-17 20:37:51 +00002129 DEMANGLE_FALLTHROUGH;
Zachary Turner66555a72018-08-20 19:15:35 +00002130 case '0':
2131 break;
2132 default:
2133 Error = true;
2134 break;
2135 }
Zachary Turner03312862018-08-27 03:48:03 +00002136 TPRN->IsMemberPointer = true;
Zachary Turner66555a72018-08-20 19:15:35 +00002137
Zachary Turnerdbefc6c2018-08-10 14:31:04 +00002138 } else if (MangledName.consumeFront("$0")) {
2139 // Integral non-type template parameter
2140 bool IsNegative = false;
2141 uint64_t Value = 0;
2142 std::tie(Value, IsNegative) = demangleNumber(MangledName);
2143
Zachary Turner03312862018-08-27 03:48:03 +00002144 TP.N = Arena.alloc<IntegerLiteralNode>(Value, IsNegative);
Zachary Turnerd30700f2018-07-31 17:16:44 +00002145 } else {
Zachary Turner03312862018-08-27 03:48:03 +00002146 TP.N = demangleType(MangledName, QualifierMangleMode::Drop);
Zachary Turnerd30700f2018-07-31 17:16:44 +00002147 }
Zachary Turner54d4ffe2018-08-01 18:32:28 +00002148 if (Error)
2149 return nullptr;
Zachary Turner23df1312018-07-26 22:13:39 +00002150
Zachary Turner66555a72018-08-20 19:15:35 +00002151 Current = &TP.Next;
Zachary Turner23df1312018-07-26 22:13:39 +00002152 }
2153
2154 if (Error)
Zachary Turner54d4ffe2018-08-01 18:32:28 +00002155 return nullptr;
Zachary Turner23df1312018-07-26 22:13:39 +00002156
2157 // Template parameter lists cannot be variadic, so it can only be terminated
2158 // by @.
2159 if (MangledName.consumeFront('@'))
Zachary Turner03312862018-08-27 03:48:03 +00002160 return nodeListToNodeArray(Arena, Head, Count);
Zachary Turner23df1312018-07-26 22:13:39 +00002161 Error = true;
Zachary Turner54d4ffe2018-08-01 18:32:28 +00002162 return nullptr;
Zachary Turner23df1312018-07-26 22:13:39 +00002163}
2164
Zachary Turner3a758e22018-08-01 18:33:04 +00002165void Demangler::dumpBackReferences() {
Zachary Turner5ae08b82018-08-01 18:44:12 +00002166 std::printf("%d function parameter backreferences\n",
Zachary Turnerd346cba2018-08-08 17:17:04 +00002167 (int)Backrefs.FunctionParamCount);
Zachary Turner3a758e22018-08-01 18:33:04 +00002168
2169 // Create an output stream so we can render each type.
Nico Weber1359d652018-09-15 18:24:20 +00002170 OutputStream OS;
Nico Weber6808bc02018-11-11 10:04:00 +00002171 if (!initializeOutputStream(nullptr, nullptr, OS, 1024))
Nico Weber1359d652018-09-15 18:24:20 +00002172 std::terminate();
Zachary Turnerd346cba2018-08-08 17:17:04 +00002173 for (size_t I = 0; I < Backrefs.FunctionParamCount; ++I) {
Zachary Turner3a758e22018-08-01 18:33:04 +00002174 OS.setCurrentPosition(0);
2175
Zachary Turner03312862018-08-27 03:48:03 +00002176 TypeNode *T = Backrefs.FunctionParams[I];
Zachary Turner38d2edd2018-08-29 03:59:17 +00002177 T->output(OS, OF_Default);
Zachary Turner3a758e22018-08-01 18:33:04 +00002178
Zachary Turner7563ebe2018-08-02 17:08:24 +00002179 std::printf(" [%d] - %.*s\n", (int)I, (int)OS.getCurrentPosition(),
Zachary Turner5ae08b82018-08-01 18:44:12 +00002180 OS.getBuffer());
Zachary Turner3a758e22018-08-01 18:33:04 +00002181 }
2182 std::free(OS.getBuffer());
2183
Zachary Turnerd346cba2018-08-08 17:17:04 +00002184 if (Backrefs.FunctionParamCount > 0)
Zachary Turner5ae08b82018-08-01 18:44:12 +00002185 std::printf("\n");
Zachary Turnerd346cba2018-08-08 17:17:04 +00002186 std::printf("%d name backreferences\n", (int)Backrefs.NamesCount);
2187 for (size_t I = 0; I < Backrefs.NamesCount; ++I) {
Zachary Turner03312862018-08-27 03:48:03 +00002188 std::printf(" [%d] - %.*s\n", (int)I, (int)Backrefs.Names[I]->Name.size(),
2189 Backrefs.Names[I]->Name.begin());
Zachary Turner3a758e22018-08-01 18:33:04 +00002190 }
Zachary Turnerd346cba2018-08-08 17:17:04 +00002191 if (Backrefs.NamesCount > 0)
Zachary Turner5ae08b82018-08-01 18:44:12 +00002192 std::printf("\n");
Zachary Turner3a758e22018-08-01 18:33:04 +00002193}
2194
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002195char *llvm::microsoftDemangle(const char *MangledName, char *Buf, size_t *N,
Zachary Turner3a758e22018-08-01 18:33:04 +00002196 int *Status, MSDemangleFlags Flags) {
Nico Weber1359d652018-09-15 18:24:20 +00002197 int InternalStatus = demangle_success;
Zachary Turner316109b2018-07-29 16:38:02 +00002198 Demangler D;
Nico Weber1359d652018-09-15 18:24:20 +00002199 OutputStream S;
2200
Zachary Turner316109b2018-07-29 16:38:02 +00002201 StringView Name{MangledName};
Nico Weber1359d652018-09-15 18:24:20 +00002202 SymbolNode *AST = D.parse(Name);
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002203
Zachary Turner3a758e22018-08-01 18:33:04 +00002204 if (Flags & MSDF_DumpBackrefs)
2205 D.dumpBackReferences();
Nico Weber1359d652018-09-15 18:24:20 +00002206
2207 if (D.Error)
2208 InternalStatus = demangle_invalid_mangled_name;
Nico Weber6808bc02018-11-11 10:04:00 +00002209 else if (!initializeOutputStream(Buf, N, S, 1024))
Nico Weber1359d652018-09-15 18:24:20 +00002210 InternalStatus = demangle_memory_alloc_failure;
2211 else {
2212 AST->output(S, OF_Default);
2213 S += '\0';
2214 if (N != nullptr)
2215 *N = S.getCurrentPosition();
2216 Buf = S.getBuffer();
Zachary Turner54d4ffe2018-08-01 18:32:28 +00002217 }
2218
Nico Weber1359d652018-09-15 18:24:20 +00002219 if (Status)
2220 *Status = InternalStatus;
2221 return InternalStatus == demangle_success ? Buf : nullptr;
Zachary Turnerf435a7e2018-07-20 17:27:48 +00002222}