blob: 98489e4dc802fc03903273612b9747d730e729a4 [file] [log] [blame]
Richard Smith4344cbd2018-08-20 20:14:49 +00001//===------------------------- ItaniumDemangle.h ----------------*- C++ -*-===//
2//
Chandler Carruth87bc9b42019-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
Richard Smith4344cbd2018-08-20 20:14:49 +00006//
7//===----------------------------------------------------------------------===//
8//
Erik Pilkington870950f2019-01-17 20:37:51 +00009// Generic itanium demangler library. This file has two byte-per-byte identical
10// copies in the source tree, one in libcxxabi, and the other in llvm.
Richard Smith4344cbd2018-08-20 20:14:49 +000011//
12//===----------------------------------------------------------------------===//
13
Erik Pilkington870950f2019-01-17 20:37:51 +000014#ifndef DEMANGLE_ITANIUMDEMANGLE_H
15#define DEMANGLE_ITANIUMDEMANGLE_H
Richard Smith4344cbd2018-08-20 20:14:49 +000016
17// FIXME: (possibly) incomplete list of features that clang mangles that this
18// file does not yet support:
19// - C++ modules TS
20
Erik Pilkington870950f2019-01-17 20:37:51 +000021#include "DemangleConfig.h"
Richard Smith4344cbd2018-08-20 20:14:49 +000022#include "StringView.h"
23#include "Utility.h"
Nathan Sidwell96c73312022-01-24 04:11:59 -080024#include <algorithm>
Richard Smith4344cbd2018-08-20 20:14:49 +000025#include <cassert>
26#include <cctype>
27#include <cstdio>
28#include <cstdlib>
29#include <cstring>
Nathan Sidwell96c73312022-01-24 04:11:59 -080030#include <limits>
Richard Smith4344cbd2018-08-20 20:14:49 +000031#include <utility>
32
33#define FOR_EACH_NODE_KIND(X) \
34 X(NodeArrayNode) \
35 X(DotSuffix) \
36 X(VendorExtQualType) \
37 X(QualType) \
38 X(ConversionOperatorType) \
39 X(PostfixQualifiedType) \
40 X(ElaboratedTypeSpefType) \
41 X(NameType) \
42 X(AbiTagAttr) \
43 X(EnableIfAttr) \
44 X(ObjCProtoName) \
45 X(PointerType) \
46 X(ReferenceType) \
47 X(PointerToMemberType) \
48 X(ArrayType) \
49 X(FunctionType) \
50 X(NoexceptSpec) \
51 X(DynamicExceptionSpec) \
52 X(FunctionEncoding) \
53 X(LiteralOperator) \
54 X(SpecialName) \
55 X(CtorVtableSpecialName) \
56 X(QualifiedName) \
57 X(NestedName) \
58 X(LocalName) \
59 X(VectorType) \
60 X(PixelVectorType) \
Pengfei Wang036d3152021-09-23 11:02:25 +080061 X(BinaryFPType) \
Richard Smithcbffe542019-09-06 23:53:21 +000062 X(SyntheticTemplateParamName) \
63 X(TypeTemplateParamDecl) \
64 X(NonTypeTemplateParamDecl) \
65 X(TemplateTemplateParamDecl) \
66 X(TemplateParamPackDecl) \
Richard Smith4344cbd2018-08-20 20:14:49 +000067 X(ParameterPack) \
68 X(TemplateArgumentPack) \
69 X(ParameterPackExpansion) \
70 X(TemplateArgs) \
71 X(ForwardTemplateReference) \
72 X(NameWithTemplateArgs) \
73 X(GlobalQualifiedName) \
74 X(StdQualifiedName) \
75 X(ExpandedSpecialSubstitution) \
76 X(SpecialSubstitution) \
77 X(CtorDtorName) \
78 X(DtorName) \
79 X(UnnamedTypeName) \
80 X(ClosureTypeName) \
81 X(StructuredBindingName) \
82 X(BinaryExpr) \
83 X(ArraySubscriptExpr) \
84 X(PostfixExpr) \
85 X(ConditionalExpr) \
86 X(MemberExpr) \
Richard Smith2b38a692020-10-22 19:29:36 -070087 X(SubobjectExpr) \
Richard Smith4344cbd2018-08-20 20:14:49 +000088 X(EnclosingExpr) \
89 X(CastExpr) \
90 X(SizeofParamPackExpr) \
91 X(CallExpr) \
92 X(NewExpr) \
93 X(DeleteExpr) \
94 X(PrefixExpr) \
95 X(FunctionParam) \
96 X(ConversionExpr) \
Richard Smith2b38a692020-10-22 19:29:36 -070097 X(PointerToMemberConversionExpr) \
Richard Smith4344cbd2018-08-20 20:14:49 +000098 X(InitListExpr) \
99 X(FoldExpr) \
100 X(ThrowExpr) \
101 X(BoolExpr) \
Richard Smithcbffe542019-09-06 23:53:21 +0000102 X(StringLiteral) \
103 X(LambdaExpr) \
Erik Pilkington27011a82020-05-13 14:13:37 -0400104 X(EnumLiteral) \
Richard Smith4344cbd2018-08-20 20:14:49 +0000105 X(IntegerLiteral) \
106 X(FloatLiteral) \
107 X(DoubleLiteral) \
108 X(LongDoubleLiteral) \
109 X(BracedExpr) \
110 X(BracedRangeExpr)
111
Erik Pilkington870950f2019-01-17 20:37:51 +0000112DEMANGLE_NAMESPACE_BEGIN
113
Mikhail Borisov5f4d2ce2021-08-17 18:06:53 -0400114template <class T, size_t N> class PODSmallVector {
115 static_assert(std::is_pod<T>::value,
116 "T is required to be a plain old data type");
117
118 T *First = nullptr;
119 T *Last = nullptr;
120 T *Cap = nullptr;
121 T Inline[N] = {0};
122
123 bool isInline() const { return First == Inline; }
124
125 void clearInline() {
126 First = Inline;
127 Last = Inline;
128 Cap = Inline + N;
129 }
130
131 void reserve(size_t NewCap) {
132 size_t S = size();
133 if (isInline()) {
134 auto *Tmp = static_cast<T *>(std::malloc(NewCap * sizeof(T)));
135 if (Tmp == nullptr)
136 std::terminate();
137 std::copy(First, Last, Tmp);
138 First = Tmp;
139 } else {
140 First = static_cast<T *>(std::realloc(First, NewCap * sizeof(T)));
141 if (First == nullptr)
142 std::terminate();
143 }
144 Last = First + S;
145 Cap = First + NewCap;
146 }
147
148public:
149 PODSmallVector() : First(Inline), Last(First), Cap(Inline + N) {}
150
151 PODSmallVector(const PODSmallVector &) = delete;
152 PODSmallVector &operator=(const PODSmallVector &) = delete;
153
154 PODSmallVector(PODSmallVector &&Other) : PODSmallVector() {
155 if (Other.isInline()) {
156 std::copy(Other.begin(), Other.end(), First);
157 Last = First + Other.size();
158 Other.clear();
159 return;
160 }
161
162 First = Other.First;
163 Last = Other.Last;
164 Cap = Other.Cap;
165 Other.clearInline();
166 }
167
168 PODSmallVector &operator=(PODSmallVector &&Other) {
169 if (Other.isInline()) {
170 if (!isInline()) {
171 std::free(First);
172 clearInline();
173 }
174 std::copy(Other.begin(), Other.end(), First);
175 Last = First + Other.size();
176 Other.clear();
177 return *this;
178 }
179
180 if (isInline()) {
181 First = Other.First;
182 Last = Other.Last;
183 Cap = Other.Cap;
184 Other.clearInline();
185 return *this;
186 }
187
188 std::swap(First, Other.First);
189 std::swap(Last, Other.Last);
190 std::swap(Cap, Other.Cap);
191 Other.clear();
192 return *this;
193 }
194
195 // NOLINTNEXTLINE(readability-identifier-naming)
196 void push_back(const T &Elem) {
197 if (Last == Cap)
198 reserve(size() * 2);
199 *Last++ = Elem;
200 }
201
202 // NOLINTNEXTLINE(readability-identifier-naming)
203 void pop_back() {
204 assert(Last != First && "Popping empty vector!");
205 --Last;
206 }
207
208 void dropBack(size_t Index) {
209 assert(Index <= size() && "dropBack() can't expand!");
210 Last = First + Index;
211 }
212
213 T *begin() { return First; }
214 T *end() { return Last; }
215
216 bool empty() const { return First == Last; }
217 size_t size() const { return static_cast<size_t>(Last - First); }
218 T &back() {
219 assert(Last != First && "Calling back() on empty vector!");
220 return *(Last - 1);
221 }
222 T &operator[](size_t Index) {
223 assert(Index < size() && "Invalid access!");
224 return *(begin() + Index);
225 }
226 void clear() { Last = First; }
227
228 ~PODSmallVector() {
229 if (!isInline())
230 std::free(First);
231 }
232};
233
Richard Smith4344cbd2018-08-20 20:14:49 +0000234// Base class of all AST nodes. The AST is built by the parser, then is
235// traversed by the printLeft/Right functions to produce a demangled string.
236class Node {
237public:
238 enum Kind : unsigned char {
239#define ENUMERATOR(NodeKind) K ## NodeKind,
240 FOR_EACH_NODE_KIND(ENUMERATOR)
241#undef ENUMERATOR
242 };
243
244 /// Three-way bool to track a cached value. Unknown is possible if this node
245 /// has an unexpanded parameter pack below it that may affect this cache.
246 enum class Cache : unsigned char { Yes, No, Unknown, };
247
248private:
249 Kind K;
250
251 // FIXME: Make these protected.
252public:
253 /// Tracks if this node has a component on its right side, in which case we
254 /// need to call printRight.
255 Cache RHSComponentCache;
256
257 /// Track if this node is a (possibly qualified) array type. This can affect
258 /// how we format the output string.
259 Cache ArrayCache;
260
261 /// Track if this node is a (possibly qualified) function type. This can
262 /// affect how we format the output string.
263 Cache FunctionCache;
264
265public:
266 Node(Kind K_, Cache RHSComponentCache_ = Cache::No,
267 Cache ArrayCache_ = Cache::No, Cache FunctionCache_ = Cache::No)
268 : K(K_), RHSComponentCache(RHSComponentCache_), ArrayCache(ArrayCache_),
269 FunctionCache(FunctionCache_) {}
270
271 /// Visit the most-derived object corresponding to this object.
272 template<typename Fn> void visit(Fn F) const;
273
274 // The following function is provided by all derived classes:
275 //
276 // Call F with arguments that, when passed to the constructor of this node,
277 // would construct an equivalent node.
278 //template<typename Fn> void match(Fn F) const;
279
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700280 bool hasRHSComponent(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000281 if (RHSComponentCache != Cache::Unknown)
282 return RHSComponentCache == Cache::Yes;
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700283 return hasRHSComponentSlow(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000284 }
285
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700286 bool hasArray(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000287 if (ArrayCache != Cache::Unknown)
288 return ArrayCache == Cache::Yes;
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700289 return hasArraySlow(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000290 }
291
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700292 bool hasFunction(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000293 if (FunctionCache != Cache::Unknown)
294 return FunctionCache == Cache::Yes;
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700295 return hasFunctionSlow(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000296 }
297
298 Kind getKind() const { return K; }
299
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700300 virtual bool hasRHSComponentSlow(OutputBuffer &) const { return false; }
301 virtual bool hasArraySlow(OutputBuffer &) const { return false; }
302 virtual bool hasFunctionSlow(OutputBuffer &) const { return false; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000303
304 // Dig through "glue" nodes like ParameterPack and ForwardTemplateReference to
305 // get at a node that actually represents some concrete syntax.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700306 virtual const Node *getSyntaxNode(OutputBuffer &) const { return this; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000307
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700308 void print(OutputBuffer &OB) const {
309 printLeft(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000310 if (RHSComponentCache != Cache::No)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700311 printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000312 }
313
Nathan Sidwell64e94ec2022-01-20 07:40:12 -0800314 // Print the "left" side of this Node into OutputBuffer.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700315 virtual void printLeft(OutputBuffer &) const = 0;
Richard Smith4344cbd2018-08-20 20:14:49 +0000316
317 // Print the "right". This distinction is necessary to represent C++ types
318 // that appear on the RHS of their subtype, such as arrays or functions.
319 // Since most types don't have such a component, provide a default
320 // implementation.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700321 virtual void printRight(OutputBuffer &) const {}
Richard Smith4344cbd2018-08-20 20:14:49 +0000322
323 virtual StringView getBaseName() const { return StringView(); }
324
325 // Silence compiler warnings, this dtor will never be called.
326 virtual ~Node() = default;
327
328#ifndef NDEBUG
Erik Pilkington870950f2019-01-17 20:37:51 +0000329 DEMANGLE_DUMP_METHOD void dump() const;
Richard Smith4344cbd2018-08-20 20:14:49 +0000330#endif
331};
332
333class NodeArray {
334 Node **Elements;
335 size_t NumElements;
336
337public:
338 NodeArray() : Elements(nullptr), NumElements(0) {}
339 NodeArray(Node **Elements_, size_t NumElements_)
340 : Elements(Elements_), NumElements(NumElements_) {}
341
342 bool empty() const { return NumElements == 0; }
343 size_t size() const { return NumElements; }
344
345 Node **begin() const { return Elements; }
346 Node **end() const { return Elements + NumElements; }
347
348 Node *operator[](size_t Idx) const { return Elements[Idx]; }
349
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700350 void printWithComma(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000351 bool FirstElement = true;
352 for (size_t Idx = 0; Idx != NumElements; ++Idx) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700353 size_t BeforeComma = OB.getCurrentPosition();
Richard Smith4344cbd2018-08-20 20:14:49 +0000354 if (!FirstElement)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700355 OB += ", ";
356 size_t AfterComma = OB.getCurrentPosition();
357 Elements[Idx]->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000358
359 // Elements[Idx] is an empty parameter pack expansion, we should erase the
360 // comma we just printed.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700361 if (AfterComma == OB.getCurrentPosition()) {
362 OB.setCurrentPosition(BeforeComma);
Richard Smith4344cbd2018-08-20 20:14:49 +0000363 continue;
364 }
365
366 FirstElement = false;
367 }
368 }
369};
370
371struct NodeArrayNode : Node {
372 NodeArray Array;
373 NodeArrayNode(NodeArray Array_) : Node(KNodeArrayNode), Array(Array_) {}
374
375 template<typename Fn> void match(Fn F) const { F(Array); }
376
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700377 void printLeft(OutputBuffer &OB) const override { Array.printWithComma(OB); }
Richard Smith4344cbd2018-08-20 20:14:49 +0000378};
379
380class DotSuffix final : public Node {
381 const Node *Prefix;
382 const StringView Suffix;
383
384public:
385 DotSuffix(const Node *Prefix_, StringView Suffix_)
386 : Node(KDotSuffix), Prefix(Prefix_), Suffix(Suffix_) {}
387
388 template<typename Fn> void match(Fn F) const { F(Prefix, Suffix); }
389
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700390 void printLeft(OutputBuffer &OB) const override {
391 Prefix->print(OB);
392 OB += " (";
393 OB += Suffix;
394 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +0000395 }
396};
397
398class VendorExtQualType final : public Node {
399 const Node *Ty;
400 StringView Ext;
Alex Orlov1540db32021-03-24 10:21:32 +0400401 const Node *TA;
Richard Smith4344cbd2018-08-20 20:14:49 +0000402
403public:
Alex Orlov1540db32021-03-24 10:21:32 +0400404 VendorExtQualType(const Node *Ty_, StringView Ext_, const Node *TA_)
405 : Node(KVendorExtQualType), Ty(Ty_), Ext(Ext_), TA(TA_) {}
Richard Smith4344cbd2018-08-20 20:14:49 +0000406
Alex Orlov1540db32021-03-24 10:21:32 +0400407 template <typename Fn> void match(Fn F) const { F(Ty, Ext, TA); }
Richard Smith4344cbd2018-08-20 20:14:49 +0000408
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700409 void printLeft(OutputBuffer &OB) const override {
410 Ty->print(OB);
411 OB += " ";
412 OB += Ext;
Alex Orlov1540db32021-03-24 10:21:32 +0400413 if (TA != nullptr)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700414 TA->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000415 }
416};
417
418enum FunctionRefQual : unsigned char {
419 FrefQualNone,
420 FrefQualLValue,
421 FrefQualRValue,
422};
423
424enum Qualifiers {
425 QualNone = 0,
426 QualConst = 0x1,
427 QualVolatile = 0x2,
428 QualRestrict = 0x4,
429};
430
431inline Qualifiers operator|=(Qualifiers &Q1, Qualifiers Q2) {
432 return Q1 = static_cast<Qualifiers>(Q1 | Q2);
433}
434
Richard Smithcbffe542019-09-06 23:53:21 +0000435class QualType final : public Node {
Richard Smith4344cbd2018-08-20 20:14:49 +0000436protected:
437 const Qualifiers Quals;
438 const Node *Child;
439
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700440 void printQuals(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000441 if (Quals & QualConst)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700442 OB += " const";
Richard Smith4344cbd2018-08-20 20:14:49 +0000443 if (Quals & QualVolatile)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700444 OB += " volatile";
Richard Smith4344cbd2018-08-20 20:14:49 +0000445 if (Quals & QualRestrict)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700446 OB += " restrict";
Richard Smith4344cbd2018-08-20 20:14:49 +0000447 }
448
449public:
450 QualType(const Node *Child_, Qualifiers Quals_)
451 : Node(KQualType, Child_->RHSComponentCache,
452 Child_->ArrayCache, Child_->FunctionCache),
453 Quals(Quals_), Child(Child_) {}
454
455 template<typename Fn> void match(Fn F) const { F(Child, Quals); }
456
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700457 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
458 return Child->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000459 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700460 bool hasArraySlow(OutputBuffer &OB) const override {
461 return Child->hasArray(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000462 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700463 bool hasFunctionSlow(OutputBuffer &OB) const override {
464 return Child->hasFunction(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000465 }
466
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700467 void printLeft(OutputBuffer &OB) const override {
468 Child->printLeft(OB);
469 printQuals(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000470 }
471
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700472 void printRight(OutputBuffer &OB) const override { Child->printRight(OB); }
Richard Smith4344cbd2018-08-20 20:14:49 +0000473};
474
475class ConversionOperatorType final : public Node {
476 const Node *Ty;
477
478public:
479 ConversionOperatorType(const Node *Ty_)
480 : Node(KConversionOperatorType), Ty(Ty_) {}
481
482 template<typename Fn> void match(Fn F) const { F(Ty); }
483
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700484 void printLeft(OutputBuffer &OB) const override {
485 OB += "operator ";
486 Ty->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000487 }
488};
489
490class PostfixQualifiedType final : public Node {
491 const Node *Ty;
492 const StringView Postfix;
493
494public:
495 PostfixQualifiedType(Node *Ty_, StringView Postfix_)
496 : Node(KPostfixQualifiedType), Ty(Ty_), Postfix(Postfix_) {}
497
498 template<typename Fn> void match(Fn F) const { F(Ty, Postfix); }
499
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700500 void printLeft(OutputBuffer &OB) const override {
501 Ty->printLeft(OB);
502 OB += Postfix;
Richard Smith4344cbd2018-08-20 20:14:49 +0000503 }
504};
505
506class NameType final : public Node {
507 const StringView Name;
508
509public:
510 NameType(StringView Name_) : Node(KNameType), Name(Name_) {}
511
512 template<typename Fn> void match(Fn F) const { F(Name); }
513
514 StringView getName() const { return Name; }
515 StringView getBaseName() const override { return Name; }
516
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700517 void printLeft(OutputBuffer &OB) const override { OB += Name; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000518};
519
520class ElaboratedTypeSpefType : public Node {
521 StringView Kind;
522 Node *Child;
523public:
524 ElaboratedTypeSpefType(StringView Kind_, Node *Child_)
525 : Node(KElaboratedTypeSpefType), Kind(Kind_), Child(Child_) {}
526
527 template<typename Fn> void match(Fn F) const { F(Kind, Child); }
528
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700529 void printLeft(OutputBuffer &OB) const override {
530 OB += Kind;
531 OB += ' ';
532 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000533 }
534};
535
536struct AbiTagAttr : Node {
537 Node *Base;
538 StringView Tag;
539
540 AbiTagAttr(Node* Base_, StringView Tag_)
541 : Node(KAbiTagAttr, Base_->RHSComponentCache,
542 Base_->ArrayCache, Base_->FunctionCache),
543 Base(Base_), Tag(Tag_) {}
544
545 template<typename Fn> void match(Fn F) const { F(Base, Tag); }
546
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700547 void printLeft(OutputBuffer &OB) const override {
548 Base->printLeft(OB);
549 OB += "[abi:";
550 OB += Tag;
551 OB += "]";
Richard Smith4344cbd2018-08-20 20:14:49 +0000552 }
553};
554
555class EnableIfAttr : public Node {
556 NodeArray Conditions;
557public:
558 EnableIfAttr(NodeArray Conditions_)
559 : Node(KEnableIfAttr), Conditions(Conditions_) {}
560
561 template<typename Fn> void match(Fn F) const { F(Conditions); }
562
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700563 void printLeft(OutputBuffer &OB) const override {
564 OB += " [enable_if:";
565 Conditions.printWithComma(OB);
566 OB += ']';
Richard Smith4344cbd2018-08-20 20:14:49 +0000567 }
568};
569
570class ObjCProtoName : public Node {
571 const Node *Ty;
572 StringView Protocol;
573
574 friend class PointerType;
575
576public:
577 ObjCProtoName(const Node *Ty_, StringView Protocol_)
578 : Node(KObjCProtoName), Ty(Ty_), Protocol(Protocol_) {}
579
580 template<typename Fn> void match(Fn F) const { F(Ty, Protocol); }
581
582 bool isObjCObject() const {
583 return Ty->getKind() == KNameType &&
584 static_cast<const NameType *>(Ty)->getName() == "objc_object";
585 }
586
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700587 void printLeft(OutputBuffer &OB) const override {
588 Ty->print(OB);
589 OB += "<";
590 OB += Protocol;
591 OB += ">";
Richard Smith4344cbd2018-08-20 20:14:49 +0000592 }
593};
594
595class PointerType final : public Node {
596 const Node *Pointee;
597
598public:
599 PointerType(const Node *Pointee_)
600 : Node(KPointerType, Pointee_->RHSComponentCache),
601 Pointee(Pointee_) {}
602
603 template<typename Fn> void match(Fn F) const { F(Pointee); }
604
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700605 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
606 return Pointee->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000607 }
608
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700609 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +0000610 // We rewrite objc_object<SomeProtocol>* into id<SomeProtocol>.
611 if (Pointee->getKind() != KObjCProtoName ||
612 !static_cast<const ObjCProtoName *>(Pointee)->isObjCObject()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700613 Pointee->printLeft(OB);
614 if (Pointee->hasArray(OB))
615 OB += " ";
616 if (Pointee->hasArray(OB) || Pointee->hasFunction(OB))
617 OB += "(";
618 OB += "*";
Richard Smith4344cbd2018-08-20 20:14:49 +0000619 } else {
620 const auto *objcProto = static_cast<const ObjCProtoName *>(Pointee);
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700621 OB += "id<";
622 OB += objcProto->Protocol;
623 OB += ">";
Richard Smith4344cbd2018-08-20 20:14:49 +0000624 }
625 }
626
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700627 void printRight(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +0000628 if (Pointee->getKind() != KObjCProtoName ||
629 !static_cast<const ObjCProtoName *>(Pointee)->isObjCObject()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700630 if (Pointee->hasArray(OB) || Pointee->hasFunction(OB))
631 OB += ")";
632 Pointee->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000633 }
634 }
635};
636
637enum class ReferenceKind {
638 LValue,
639 RValue,
640};
641
642// Represents either a LValue or an RValue reference type.
643class ReferenceType : public Node {
644 const Node *Pointee;
645 ReferenceKind RK;
646
647 mutable bool Printing = false;
648
649 // Dig through any refs to refs, collapsing the ReferenceTypes as we go. The
650 // rule here is rvalue ref to rvalue ref collapses to a rvalue ref, and any
651 // other combination collapses to a lvalue ref.
Mikhail Borisovcf1381e2021-08-17 18:10:57 -0400652 //
653 // A combination of a TemplateForwardReference and a back-ref Substitution
654 // from an ill-formed string may have created a cycle; use cycle detection to
655 // avoid looping forever.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700656 std::pair<ReferenceKind, const Node *> collapse(OutputBuffer &OB) const {
Richard Smith4344cbd2018-08-20 20:14:49 +0000657 auto SoFar = std::make_pair(RK, Pointee);
Mikhail Borisovcf1381e2021-08-17 18:10:57 -0400658 // Track the chain of nodes for the Floyd's 'tortoise and hare'
659 // cycle-detection algorithm, since getSyntaxNode(S) is impure
660 PODSmallVector<const Node *, 8> Prev;
Richard Smith4344cbd2018-08-20 20:14:49 +0000661 for (;;) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700662 const Node *SN = SoFar.second->getSyntaxNode(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000663 if (SN->getKind() != KReferenceType)
664 break;
665 auto *RT = static_cast<const ReferenceType *>(SN);
666 SoFar.second = RT->Pointee;
667 SoFar.first = std::min(SoFar.first, RT->RK);
Mikhail Borisovcf1381e2021-08-17 18:10:57 -0400668
669 // The middle of Prev is the 'slow' pointer moving at half speed
670 Prev.push_back(SoFar.second);
671 if (Prev.size() > 1 && SoFar.second == Prev[(Prev.size() - 1) / 2]) {
672 // Cycle detected
673 SoFar.second = nullptr;
674 break;
675 }
Richard Smith4344cbd2018-08-20 20:14:49 +0000676 }
677 return SoFar;
678 }
679
680public:
681 ReferenceType(const Node *Pointee_, ReferenceKind RK_)
682 : Node(KReferenceType, Pointee_->RHSComponentCache),
683 Pointee(Pointee_), RK(RK_) {}
684
685 template<typename Fn> void match(Fn F) const { F(Pointee, RK); }
686
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700687 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
688 return Pointee->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000689 }
690
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700691 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +0000692 if (Printing)
693 return;
694 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700695 std::pair<ReferenceKind, const Node *> Collapsed = collapse(OB);
Mikhail Borisovcf1381e2021-08-17 18:10:57 -0400696 if (!Collapsed.second)
697 return;
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700698 Collapsed.second->printLeft(OB);
699 if (Collapsed.second->hasArray(OB))
700 OB += " ";
701 if (Collapsed.second->hasArray(OB) || Collapsed.second->hasFunction(OB))
702 OB += "(";
Richard Smith4344cbd2018-08-20 20:14:49 +0000703
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700704 OB += (Collapsed.first == ReferenceKind::LValue ? "&" : "&&");
Richard Smith4344cbd2018-08-20 20:14:49 +0000705 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700706 void printRight(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +0000707 if (Printing)
708 return;
709 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700710 std::pair<ReferenceKind, const Node *> Collapsed = collapse(OB);
Mikhail Borisovcf1381e2021-08-17 18:10:57 -0400711 if (!Collapsed.second)
712 return;
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700713 if (Collapsed.second->hasArray(OB) || Collapsed.second->hasFunction(OB))
714 OB += ")";
715 Collapsed.second->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000716 }
717};
718
719class PointerToMemberType final : public Node {
720 const Node *ClassType;
721 const Node *MemberType;
722
723public:
724 PointerToMemberType(const Node *ClassType_, const Node *MemberType_)
725 : Node(KPointerToMemberType, MemberType_->RHSComponentCache),
726 ClassType(ClassType_), MemberType(MemberType_) {}
727
728 template<typename Fn> void match(Fn F) const { F(ClassType, MemberType); }
729
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700730 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
731 return MemberType->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000732 }
733
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700734 void printLeft(OutputBuffer &OB) const override {
735 MemberType->printLeft(OB);
736 if (MemberType->hasArray(OB) || MemberType->hasFunction(OB))
737 OB += "(";
Richard Smith4344cbd2018-08-20 20:14:49 +0000738 else
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700739 OB += " ";
740 ClassType->print(OB);
741 OB += "::*";
Richard Smith4344cbd2018-08-20 20:14:49 +0000742 }
743
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700744 void printRight(OutputBuffer &OB) const override {
745 if (MemberType->hasArray(OB) || MemberType->hasFunction(OB))
746 OB += ")";
747 MemberType->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000748 }
749};
750
Richard Smith4344cbd2018-08-20 20:14:49 +0000751class ArrayType final : public Node {
752 const Node *Base;
Erik Pilkington87529ae2019-11-04 10:47:44 -0800753 Node *Dimension;
Richard Smith4344cbd2018-08-20 20:14:49 +0000754
755public:
Erik Pilkington87529ae2019-11-04 10:47:44 -0800756 ArrayType(const Node *Base_, Node *Dimension_)
Richard Smith4344cbd2018-08-20 20:14:49 +0000757 : Node(KArrayType,
758 /*RHSComponentCache=*/Cache::Yes,
759 /*ArrayCache=*/Cache::Yes),
760 Base(Base_), Dimension(Dimension_) {}
761
762 template<typename Fn> void match(Fn F) const { F(Base, Dimension); }
763
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700764 bool hasRHSComponentSlow(OutputBuffer &) const override { return true; }
765 bool hasArraySlow(OutputBuffer &) const override { return true; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000766
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700767 void printLeft(OutputBuffer &OB) const override { Base->printLeft(OB); }
Richard Smith4344cbd2018-08-20 20:14:49 +0000768
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700769 void printRight(OutputBuffer &OB) const override {
770 if (OB.back() != ']')
771 OB += " ";
772 OB += "[";
Erik Pilkington87529ae2019-11-04 10:47:44 -0800773 if (Dimension)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700774 Dimension->print(OB);
775 OB += "]";
776 Base->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000777 }
778};
779
780class FunctionType final : public Node {
781 const Node *Ret;
782 NodeArray Params;
783 Qualifiers CVQuals;
784 FunctionRefQual RefQual;
785 const Node *ExceptionSpec;
786
787public:
788 FunctionType(const Node *Ret_, NodeArray Params_, Qualifiers CVQuals_,
789 FunctionRefQual RefQual_, const Node *ExceptionSpec_)
790 : Node(KFunctionType,
791 /*RHSComponentCache=*/Cache::Yes, /*ArrayCache=*/Cache::No,
792 /*FunctionCache=*/Cache::Yes),
793 Ret(Ret_), Params(Params_), CVQuals(CVQuals_), RefQual(RefQual_),
794 ExceptionSpec(ExceptionSpec_) {}
795
796 template<typename Fn> void match(Fn F) const {
797 F(Ret, Params, CVQuals, RefQual, ExceptionSpec);
798 }
799
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700800 bool hasRHSComponentSlow(OutputBuffer &) const override { return true; }
801 bool hasFunctionSlow(OutputBuffer &) const override { return true; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000802
803 // Handle C++'s ... quirky decl grammar by using the left & right
804 // distinction. Consider:
805 // int (*f(float))(char) {}
806 // f is a function that takes a float and returns a pointer to a function
807 // that takes a char and returns an int. If we're trying to print f, start
808 // by printing out the return types's left, then print our parameters, then
809 // finally print right of the return type.
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700810 void printLeft(OutputBuffer &OB) const override {
811 Ret->printLeft(OB);
812 OB += " ";
Richard Smith4344cbd2018-08-20 20:14:49 +0000813 }
814
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700815 void printRight(OutputBuffer &OB) const override {
816 OB += "(";
817 Params.printWithComma(OB);
818 OB += ")";
819 Ret->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000820
821 if (CVQuals & QualConst)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700822 OB += " const";
Richard Smith4344cbd2018-08-20 20:14:49 +0000823 if (CVQuals & QualVolatile)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700824 OB += " volatile";
Richard Smith4344cbd2018-08-20 20:14:49 +0000825 if (CVQuals & QualRestrict)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700826 OB += " restrict";
Richard Smith4344cbd2018-08-20 20:14:49 +0000827
828 if (RefQual == FrefQualLValue)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700829 OB += " &";
Richard Smith4344cbd2018-08-20 20:14:49 +0000830 else if (RefQual == FrefQualRValue)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700831 OB += " &&";
Richard Smith4344cbd2018-08-20 20:14:49 +0000832
833 if (ExceptionSpec != nullptr) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700834 OB += ' ';
835 ExceptionSpec->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000836 }
837 }
838};
839
840class NoexceptSpec : public Node {
841 const Node *E;
842public:
843 NoexceptSpec(const Node *E_) : Node(KNoexceptSpec), E(E_) {}
844
845 template<typename Fn> void match(Fn F) const { F(E); }
846
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700847 void printLeft(OutputBuffer &OB) const override {
848 OB += "noexcept(";
849 E->print(OB);
850 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +0000851 }
852};
853
854class DynamicExceptionSpec : public Node {
855 NodeArray Types;
856public:
857 DynamicExceptionSpec(NodeArray Types_)
858 : Node(KDynamicExceptionSpec), Types(Types_) {}
859
860 template<typename Fn> void match(Fn F) const { F(Types); }
861
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700862 void printLeft(OutputBuffer &OB) const override {
863 OB += "throw(";
864 Types.printWithComma(OB);
865 OB += ')';
Richard Smith4344cbd2018-08-20 20:14:49 +0000866 }
867};
868
869class FunctionEncoding final : public Node {
870 const Node *Ret;
871 const Node *Name;
872 NodeArray Params;
873 const Node *Attrs;
874 Qualifiers CVQuals;
875 FunctionRefQual RefQual;
876
877public:
878 FunctionEncoding(const Node *Ret_, const Node *Name_, NodeArray Params_,
879 const Node *Attrs_, Qualifiers CVQuals_,
880 FunctionRefQual RefQual_)
881 : Node(KFunctionEncoding,
882 /*RHSComponentCache=*/Cache::Yes, /*ArrayCache=*/Cache::No,
883 /*FunctionCache=*/Cache::Yes),
884 Ret(Ret_), Name(Name_), Params(Params_), Attrs(Attrs_),
885 CVQuals(CVQuals_), RefQual(RefQual_) {}
886
887 template<typename Fn> void match(Fn F) const {
888 F(Ret, Name, Params, Attrs, CVQuals, RefQual);
889 }
890
891 Qualifiers getCVQuals() const { return CVQuals; }
892 FunctionRefQual getRefQual() const { return RefQual; }
893 NodeArray getParams() const { return Params; }
894 const Node *getReturnType() const { return Ret; }
895
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700896 bool hasRHSComponentSlow(OutputBuffer &) const override { return true; }
897 bool hasFunctionSlow(OutputBuffer &) const override { return true; }
Richard Smith4344cbd2018-08-20 20:14:49 +0000898
899 const Node *getName() const { return Name; }
900
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700901 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +0000902 if (Ret) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700903 Ret->printLeft(OB);
904 if (!Ret->hasRHSComponent(OB))
905 OB += " ";
Richard Smith4344cbd2018-08-20 20:14:49 +0000906 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700907 Name->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000908 }
909
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700910 void printRight(OutputBuffer &OB) const override {
911 OB += "(";
912 Params.printWithComma(OB);
913 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +0000914 if (Ret)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700915 Ret->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000916
917 if (CVQuals & QualConst)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700918 OB += " const";
Richard Smith4344cbd2018-08-20 20:14:49 +0000919 if (CVQuals & QualVolatile)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700920 OB += " volatile";
Richard Smith4344cbd2018-08-20 20:14:49 +0000921 if (CVQuals & QualRestrict)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700922 OB += " restrict";
Richard Smith4344cbd2018-08-20 20:14:49 +0000923
924 if (RefQual == FrefQualLValue)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700925 OB += " &";
Richard Smith4344cbd2018-08-20 20:14:49 +0000926 else if (RefQual == FrefQualRValue)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700927 OB += " &&";
Richard Smith4344cbd2018-08-20 20:14:49 +0000928
929 if (Attrs != nullptr)
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700930 Attrs->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000931 }
932};
933
934class LiteralOperator : public Node {
935 const Node *OpName;
936
937public:
938 LiteralOperator(const Node *OpName_)
939 : Node(KLiteralOperator), OpName(OpName_) {}
940
941 template<typename Fn> void match(Fn F) const { F(OpName); }
942
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700943 void printLeft(OutputBuffer &OB) const override {
944 OB += "operator\"\" ";
945 OpName->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000946 }
947};
948
949class SpecialName final : public Node {
950 const StringView Special;
951 const Node *Child;
952
953public:
954 SpecialName(StringView Special_, const Node *Child_)
955 : Node(KSpecialName), Special(Special_), Child(Child_) {}
956
957 template<typename Fn> void match(Fn F) const { F(Special, Child); }
958
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700959 void printLeft(OutputBuffer &OB) const override {
960 OB += Special;
961 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000962 }
963};
964
965class CtorVtableSpecialName final : public Node {
966 const Node *FirstType;
967 const Node *SecondType;
968
969public:
970 CtorVtableSpecialName(const Node *FirstType_, const Node *SecondType_)
971 : Node(KCtorVtableSpecialName),
972 FirstType(FirstType_), SecondType(SecondType_) {}
973
974 template<typename Fn> void match(Fn F) const { F(FirstType, SecondType); }
975
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700976 void printLeft(OutputBuffer &OB) const override {
977 OB += "construction vtable for ";
978 FirstType->print(OB);
979 OB += "-in-";
980 SecondType->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000981 }
982};
983
984struct NestedName : Node {
985 Node *Qual;
986 Node *Name;
987
988 NestedName(Node *Qual_, Node *Name_)
989 : Node(KNestedName), Qual(Qual_), Name(Name_) {}
990
991 template<typename Fn> void match(Fn F) const { F(Qual, Name); }
992
993 StringView getBaseName() const override { return Name->getBaseName(); }
994
Luís Ferreira4fb856a2021-10-21 17:31:53 -0700995 void printLeft(OutputBuffer &OB) const override {
996 Qual->print(OB);
997 OB += "::";
998 Name->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +0000999 }
1000};
1001
1002struct LocalName : Node {
1003 Node *Encoding;
1004 Node *Entity;
1005
1006 LocalName(Node *Encoding_, Node *Entity_)
1007 : Node(KLocalName), Encoding(Encoding_), Entity(Entity_) {}
1008
1009 template<typename Fn> void match(Fn F) const { F(Encoding, Entity); }
1010
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001011 void printLeft(OutputBuffer &OB) const override {
1012 Encoding->print(OB);
1013 OB += "::";
1014 Entity->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001015 }
1016};
1017
1018class QualifiedName final : public Node {
1019 // qualifier::name
1020 const Node *Qualifier;
1021 const Node *Name;
1022
1023public:
1024 QualifiedName(const Node *Qualifier_, const Node *Name_)
1025 : Node(KQualifiedName), Qualifier(Qualifier_), Name(Name_) {}
1026
1027 template<typename Fn> void match(Fn F) const { F(Qualifier, Name); }
1028
1029 StringView getBaseName() const override { return Name->getBaseName(); }
1030
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001031 void printLeft(OutputBuffer &OB) const override {
1032 Qualifier->print(OB);
1033 OB += "::";
1034 Name->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001035 }
1036};
1037
1038class VectorType final : public Node {
1039 const Node *BaseType;
Erik Pilkington87529ae2019-11-04 10:47:44 -08001040 const Node *Dimension;
Richard Smith4344cbd2018-08-20 20:14:49 +00001041
1042public:
Erik Pilkington87529ae2019-11-04 10:47:44 -08001043 VectorType(const Node *BaseType_, Node *Dimension_)
Richard Smith4344cbd2018-08-20 20:14:49 +00001044 : Node(KVectorType), BaseType(BaseType_),
1045 Dimension(Dimension_) {}
1046
1047 template<typename Fn> void match(Fn F) const { F(BaseType, Dimension); }
1048
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001049 void printLeft(OutputBuffer &OB) const override {
1050 BaseType->print(OB);
1051 OB += " vector[";
Erik Pilkington87529ae2019-11-04 10:47:44 -08001052 if (Dimension)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001053 Dimension->print(OB);
1054 OB += "]";
Richard Smith4344cbd2018-08-20 20:14:49 +00001055 }
1056};
1057
1058class PixelVectorType final : public Node {
Erik Pilkington87529ae2019-11-04 10:47:44 -08001059 const Node *Dimension;
Richard Smith4344cbd2018-08-20 20:14:49 +00001060
1061public:
Erik Pilkington87529ae2019-11-04 10:47:44 -08001062 PixelVectorType(const Node *Dimension_)
Richard Smith4344cbd2018-08-20 20:14:49 +00001063 : Node(KPixelVectorType), Dimension(Dimension_) {}
1064
1065 template<typename Fn> void match(Fn F) const { F(Dimension); }
1066
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001067 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001068 // FIXME: This should demangle as "vector pixel".
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001069 OB += "pixel vector[";
1070 Dimension->print(OB);
1071 OB += "]";
Richard Smith4344cbd2018-08-20 20:14:49 +00001072 }
1073};
1074
Pengfei Wang036d3152021-09-23 11:02:25 +08001075class BinaryFPType final : public Node {
1076 const Node *Dimension;
1077
1078public:
1079 BinaryFPType(const Node *Dimension_)
1080 : Node(KBinaryFPType), Dimension(Dimension_) {}
1081
1082 template<typename Fn> void match(Fn F) const { F(Dimension); }
1083
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001084 void printLeft(OutputBuffer &OB) const override {
1085 OB += "_Float";
1086 Dimension->print(OB);
Pengfei Wang036d3152021-09-23 11:02:25 +08001087 }
1088};
1089
Richard Smithcbffe542019-09-06 23:53:21 +00001090enum class TemplateParamKind { Type, NonType, Template };
1091
1092/// An invented name for a template parameter for which we don't have a
1093/// corresponding template argument.
1094///
1095/// This node is created when parsing the <lambda-sig> for a lambda with
1096/// explicit template arguments, which might be referenced in the parameter
1097/// types appearing later in the <lambda-sig>.
1098class SyntheticTemplateParamName final : public Node {
1099 TemplateParamKind Kind;
1100 unsigned Index;
1101
1102public:
1103 SyntheticTemplateParamName(TemplateParamKind Kind_, unsigned Index_)
1104 : Node(KSyntheticTemplateParamName), Kind(Kind_), Index(Index_) {}
1105
1106 template<typename Fn> void match(Fn F) const { F(Kind, Index); }
1107
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001108 void printLeft(OutputBuffer &OB) const override {
Richard Smithcbffe542019-09-06 23:53:21 +00001109 switch (Kind) {
1110 case TemplateParamKind::Type:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001111 OB += "$T";
Richard Smithcbffe542019-09-06 23:53:21 +00001112 break;
1113 case TemplateParamKind::NonType:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001114 OB += "$N";
Richard Smithcbffe542019-09-06 23:53:21 +00001115 break;
1116 case TemplateParamKind::Template:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001117 OB += "$TT";
Richard Smithcbffe542019-09-06 23:53:21 +00001118 break;
1119 }
1120 if (Index > 0)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001121 OB << Index - 1;
Richard Smithcbffe542019-09-06 23:53:21 +00001122 }
1123};
1124
1125/// A template type parameter declaration, 'typename T'.
1126class TypeTemplateParamDecl final : public Node {
1127 Node *Name;
1128
1129public:
1130 TypeTemplateParamDecl(Node *Name_)
1131 : Node(KTypeTemplateParamDecl, Cache::Yes), Name(Name_) {}
1132
1133 template<typename Fn> void match(Fn F) const { F(Name); }
1134
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001135 void printLeft(OutputBuffer &OB) const override { OB += "typename "; }
Richard Smithcbffe542019-09-06 23:53:21 +00001136
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001137 void printRight(OutputBuffer &OB) const override { Name->print(OB); }
Richard Smithcbffe542019-09-06 23:53:21 +00001138};
1139
1140/// A non-type template parameter declaration, 'int N'.
1141class NonTypeTemplateParamDecl final : public Node {
1142 Node *Name;
1143 Node *Type;
1144
1145public:
1146 NonTypeTemplateParamDecl(Node *Name_, Node *Type_)
1147 : Node(KNonTypeTemplateParamDecl, Cache::Yes), Name(Name_), Type(Type_) {}
1148
1149 template<typename Fn> void match(Fn F) const { F(Name, Type); }
1150
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001151 void printLeft(OutputBuffer &OB) const override {
1152 Type->printLeft(OB);
1153 if (!Type->hasRHSComponent(OB))
1154 OB += " ";
Richard Smithcbffe542019-09-06 23:53:21 +00001155 }
1156
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001157 void printRight(OutputBuffer &OB) const override {
1158 Name->print(OB);
1159 Type->printRight(OB);
Richard Smithcbffe542019-09-06 23:53:21 +00001160 }
1161};
1162
1163/// A template template parameter declaration,
1164/// 'template<typename T> typename N'.
1165class TemplateTemplateParamDecl final : public Node {
1166 Node *Name;
1167 NodeArray Params;
1168
1169public:
1170 TemplateTemplateParamDecl(Node *Name_, NodeArray Params_)
1171 : Node(KTemplateTemplateParamDecl, Cache::Yes), Name(Name_),
1172 Params(Params_) {}
1173
1174 template<typename Fn> void match(Fn F) const { F(Name, Params); }
1175
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001176 void printLeft(OutputBuffer &OB) const override {
1177 OB += "template<";
1178 Params.printWithComma(OB);
1179 OB += "> typename ";
Richard Smithcbffe542019-09-06 23:53:21 +00001180 }
1181
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001182 void printRight(OutputBuffer &OB) const override { Name->print(OB); }
Richard Smithcbffe542019-09-06 23:53:21 +00001183};
1184
1185/// A template parameter pack declaration, 'typename ...T'.
1186class TemplateParamPackDecl final : public Node {
1187 Node *Param;
1188
1189public:
1190 TemplateParamPackDecl(Node *Param_)
1191 : Node(KTemplateParamPackDecl, Cache::Yes), Param(Param_) {}
1192
1193 template<typename Fn> void match(Fn F) const { F(Param); }
1194
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001195 void printLeft(OutputBuffer &OB) const override {
1196 Param->printLeft(OB);
1197 OB += "...";
Richard Smithcbffe542019-09-06 23:53:21 +00001198 }
1199
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001200 void printRight(OutputBuffer &OB) const override { Param->printRight(OB); }
Richard Smithcbffe542019-09-06 23:53:21 +00001201};
1202
Richard Smith4344cbd2018-08-20 20:14:49 +00001203/// An unexpanded parameter pack (either in the expression or type context). If
1204/// this AST is correct, this node will have a ParameterPackExpansion node above
1205/// it.
1206///
1207/// This node is created when some <template-args> are found that apply to an
1208/// <encoding>, and is stored in the TemplateParams table. In order for this to
1209/// appear in the final AST, it has to referenced via a <template-param> (ie,
1210/// T_).
1211class ParameterPack final : public Node {
1212 NodeArray Data;
1213
Nathan Sidwell64e94ec2022-01-20 07:40:12 -08001214 // Setup OutputBuffer for a pack expansion, unless we're already expanding
1215 // one.
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001216 void initializePackExpansion(OutputBuffer &OB) const {
1217 if (OB.CurrentPackMax == std::numeric_limits<unsigned>::max()) {
1218 OB.CurrentPackMax = static_cast<unsigned>(Data.size());
1219 OB.CurrentPackIndex = 0;
Richard Smith4344cbd2018-08-20 20:14:49 +00001220 }
1221 }
1222
1223public:
1224 ParameterPack(NodeArray Data_) : Node(KParameterPack), Data(Data_) {
1225 ArrayCache = FunctionCache = RHSComponentCache = Cache::Unknown;
1226 if (std::all_of(Data.begin(), Data.end(), [](Node* P) {
1227 return P->ArrayCache == Cache::No;
1228 }))
1229 ArrayCache = Cache::No;
1230 if (std::all_of(Data.begin(), Data.end(), [](Node* P) {
1231 return P->FunctionCache == Cache::No;
1232 }))
1233 FunctionCache = Cache::No;
1234 if (std::all_of(Data.begin(), Data.end(), [](Node* P) {
1235 return P->RHSComponentCache == Cache::No;
1236 }))
1237 RHSComponentCache = Cache::No;
1238 }
1239
1240 template<typename Fn> void match(Fn F) const { F(Data); }
1241
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001242 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
1243 initializePackExpansion(OB);
1244 size_t Idx = OB.CurrentPackIndex;
1245 return Idx < Data.size() && Data[Idx]->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001246 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001247 bool hasArraySlow(OutputBuffer &OB) const override {
1248 initializePackExpansion(OB);
1249 size_t Idx = OB.CurrentPackIndex;
1250 return Idx < Data.size() && Data[Idx]->hasArray(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001251 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001252 bool hasFunctionSlow(OutputBuffer &OB) const override {
1253 initializePackExpansion(OB);
1254 size_t Idx = OB.CurrentPackIndex;
1255 return Idx < Data.size() && Data[Idx]->hasFunction(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001256 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001257 const Node *getSyntaxNode(OutputBuffer &OB) const override {
1258 initializePackExpansion(OB);
1259 size_t Idx = OB.CurrentPackIndex;
1260 return Idx < Data.size() ? Data[Idx]->getSyntaxNode(OB) : this;
Richard Smith4344cbd2018-08-20 20:14:49 +00001261 }
1262
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001263 void printLeft(OutputBuffer &OB) const override {
1264 initializePackExpansion(OB);
1265 size_t Idx = OB.CurrentPackIndex;
Richard Smith4344cbd2018-08-20 20:14:49 +00001266 if (Idx < Data.size())
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001267 Data[Idx]->printLeft(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001268 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001269 void printRight(OutputBuffer &OB) const override {
1270 initializePackExpansion(OB);
1271 size_t Idx = OB.CurrentPackIndex;
Richard Smith4344cbd2018-08-20 20:14:49 +00001272 if (Idx < Data.size())
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001273 Data[Idx]->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001274 }
1275};
1276
1277/// A variadic template argument. This node represents an occurrence of
1278/// J<something>E in some <template-args>. It isn't itself unexpanded, unless
1279/// one of it's Elements is. The parser inserts a ParameterPack into the
1280/// TemplateParams table if the <template-args> this pack belongs to apply to an
1281/// <encoding>.
1282class TemplateArgumentPack final : public Node {
1283 NodeArray Elements;
1284public:
1285 TemplateArgumentPack(NodeArray Elements_)
1286 : Node(KTemplateArgumentPack), Elements(Elements_) {}
1287
1288 template<typename Fn> void match(Fn F) const { F(Elements); }
1289
1290 NodeArray getElements() const { return Elements; }
1291
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001292 void printLeft(OutputBuffer &OB) const override {
1293 Elements.printWithComma(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001294 }
1295};
1296
1297/// A pack expansion. Below this node, there are some unexpanded ParameterPacks
1298/// which each have Child->ParameterPackSize elements.
1299class ParameterPackExpansion final : public Node {
1300 const Node *Child;
1301
1302public:
1303 ParameterPackExpansion(const Node *Child_)
1304 : Node(KParameterPackExpansion), Child(Child_) {}
1305
1306 template<typename Fn> void match(Fn F) const { F(Child); }
1307
1308 const Node *getChild() const { return Child; }
1309
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001310 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001311 constexpr unsigned Max = std::numeric_limits<unsigned>::max();
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001312 SwapAndRestore<unsigned> SavePackIdx(OB.CurrentPackIndex, Max);
1313 SwapAndRestore<unsigned> SavePackMax(OB.CurrentPackMax, Max);
1314 size_t StreamPos = OB.getCurrentPosition();
Richard Smith4344cbd2018-08-20 20:14:49 +00001315
1316 // Print the first element in the pack. If Child contains a ParameterPack,
1317 // it will set up S.CurrentPackMax and print the first element.
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001318 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001319
1320 // No ParameterPack was found in Child. This can occur if we've found a pack
1321 // expansion on a <function-param>.
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001322 if (OB.CurrentPackMax == Max) {
1323 OB += "...";
Richard Smith4344cbd2018-08-20 20:14:49 +00001324 return;
1325 }
1326
1327 // We found a ParameterPack, but it has no elements. Erase whatever we may
1328 // of printed.
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001329 if (OB.CurrentPackMax == 0) {
1330 OB.setCurrentPosition(StreamPos);
Richard Smith4344cbd2018-08-20 20:14:49 +00001331 return;
1332 }
1333
1334 // Else, iterate through the rest of the elements in the pack.
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001335 for (unsigned I = 1, E = OB.CurrentPackMax; I < E; ++I) {
1336 OB += ", ";
1337 OB.CurrentPackIndex = I;
1338 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001339 }
1340 }
1341};
1342
1343class TemplateArgs final : public Node {
1344 NodeArray Params;
1345
1346public:
1347 TemplateArgs(NodeArray Params_) : Node(KTemplateArgs), Params(Params_) {}
1348
1349 template<typename Fn> void match(Fn F) const { F(Params); }
1350
1351 NodeArray getParams() { return Params; }
1352
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001353 void printLeft(OutputBuffer &OB) const override {
1354 OB += "<";
1355 Params.printWithComma(OB);
1356 if (OB.back() == '>')
1357 OB += " ";
1358 OB += ">";
Richard Smith4344cbd2018-08-20 20:14:49 +00001359 }
1360};
1361
Richard Smith7918dea2018-08-24 23:30:26 +00001362/// A forward-reference to a template argument that was not known at the point
1363/// where the template parameter name was parsed in a mangling.
1364///
1365/// This is created when demangling the name of a specialization of a
1366/// conversion function template:
1367///
1368/// \code
1369/// struct A {
1370/// template<typename T> operator T*();
1371/// };
1372/// \endcode
1373///
1374/// When demangling a specialization of the conversion function template, we
1375/// encounter the name of the template (including the \c T) before we reach
1376/// the template argument list, so we cannot substitute the parameter name
1377/// for the corresponding argument while parsing. Instead, we create a
1378/// \c ForwardTemplateReference node that is resolved after we parse the
1379/// template arguments.
Richard Smith4344cbd2018-08-20 20:14:49 +00001380struct ForwardTemplateReference : Node {
1381 size_t Index;
1382 Node *Ref = nullptr;
1383
1384 // If we're currently printing this node. It is possible (though invalid) for
1385 // a forward template reference to refer to itself via a substitution. This
1386 // creates a cyclic AST, which will stack overflow printing. To fix this, bail
1387 // out if more than one print* function is active.
1388 mutable bool Printing = false;
1389
1390 ForwardTemplateReference(size_t Index_)
1391 : Node(KForwardTemplateReference, Cache::Unknown, Cache::Unknown,
1392 Cache::Unknown),
1393 Index(Index_) {}
1394
1395 // We don't provide a matcher for these, because the value of the node is
1396 // not determined by its construction parameters, and it generally needs
1397 // special handling.
1398 template<typename Fn> void match(Fn F) const = delete;
1399
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001400 bool hasRHSComponentSlow(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001401 if (Printing)
1402 return false;
1403 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001404 return Ref->hasRHSComponent(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001405 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001406 bool hasArraySlow(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001407 if (Printing)
1408 return false;
1409 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001410 return Ref->hasArray(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001411 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001412 bool hasFunctionSlow(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001413 if (Printing)
1414 return false;
1415 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001416 return Ref->hasFunction(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001417 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001418 const Node *getSyntaxNode(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001419 if (Printing)
1420 return this;
1421 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001422 return Ref->getSyntaxNode(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001423 }
1424
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001425 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001426 if (Printing)
1427 return;
1428 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001429 Ref->printLeft(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001430 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001431 void printRight(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001432 if (Printing)
1433 return;
1434 SwapAndRestore<bool> SavePrinting(Printing, true);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001435 Ref->printRight(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001436 }
1437};
1438
1439struct NameWithTemplateArgs : Node {
1440 // name<template_args>
1441 Node *Name;
1442 Node *TemplateArgs;
1443
1444 NameWithTemplateArgs(Node *Name_, Node *TemplateArgs_)
1445 : Node(KNameWithTemplateArgs), Name(Name_), TemplateArgs(TemplateArgs_) {}
1446
1447 template<typename Fn> void match(Fn F) const { F(Name, TemplateArgs); }
1448
1449 StringView getBaseName() const override { return Name->getBaseName(); }
1450
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001451 void printLeft(OutputBuffer &OB) const override {
1452 Name->print(OB);
1453 TemplateArgs->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001454 }
1455};
1456
1457class GlobalQualifiedName final : public Node {
1458 Node *Child;
1459
1460public:
1461 GlobalQualifiedName(Node* Child_)
1462 : Node(KGlobalQualifiedName), Child(Child_) {}
1463
1464 template<typename Fn> void match(Fn F) const { F(Child); }
1465
1466 StringView getBaseName() const override { return Child->getBaseName(); }
1467
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001468 void printLeft(OutputBuffer &OB) const override {
1469 OB += "::";
1470 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001471 }
1472};
1473
1474struct StdQualifiedName : Node {
1475 Node *Child;
1476
1477 StdQualifiedName(Node *Child_) : Node(KStdQualifiedName), Child(Child_) {}
1478
1479 template<typename Fn> void match(Fn F) const { F(Child); }
1480
1481 StringView getBaseName() const override { return Child->getBaseName(); }
1482
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001483 void printLeft(OutputBuffer &OB) const override {
1484 OB += "std::";
1485 Child->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001486 }
1487};
1488
1489enum class SpecialSubKind {
1490 allocator,
1491 basic_string,
1492 string,
1493 istream,
1494 ostream,
1495 iostream,
1496};
1497
1498class ExpandedSpecialSubstitution final : public Node {
1499 SpecialSubKind SSK;
1500
1501public:
1502 ExpandedSpecialSubstitution(SpecialSubKind SSK_)
1503 : Node(KExpandedSpecialSubstitution), SSK(SSK_) {}
1504
1505 template<typename Fn> void match(Fn F) const { F(SSK); }
1506
1507 StringView getBaseName() const override {
1508 switch (SSK) {
1509 case SpecialSubKind::allocator:
1510 return StringView("allocator");
1511 case SpecialSubKind::basic_string:
1512 return StringView("basic_string");
1513 case SpecialSubKind::string:
1514 return StringView("basic_string");
1515 case SpecialSubKind::istream:
1516 return StringView("basic_istream");
1517 case SpecialSubKind::ostream:
1518 return StringView("basic_ostream");
1519 case SpecialSubKind::iostream:
1520 return StringView("basic_iostream");
1521 }
Erik Pilkington870950f2019-01-17 20:37:51 +00001522 DEMANGLE_UNREACHABLE;
Richard Smith4344cbd2018-08-20 20:14:49 +00001523 }
1524
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001525 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001526 switch (SSK) {
1527 case SpecialSubKind::allocator:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001528 OB += "std::allocator";
Richard Smith4344cbd2018-08-20 20:14:49 +00001529 break;
1530 case SpecialSubKind::basic_string:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001531 OB += "std::basic_string";
Richard Smith7918dea2018-08-24 23:30:26 +00001532 break;
Richard Smith4344cbd2018-08-20 20:14:49 +00001533 case SpecialSubKind::string:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001534 OB += "std::basic_string<char, std::char_traits<char>, "
1535 "std::allocator<char> >";
Richard Smith4344cbd2018-08-20 20:14:49 +00001536 break;
1537 case SpecialSubKind::istream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001538 OB += "std::basic_istream<char, std::char_traits<char> >";
Richard Smith4344cbd2018-08-20 20:14:49 +00001539 break;
1540 case SpecialSubKind::ostream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001541 OB += "std::basic_ostream<char, std::char_traits<char> >";
Richard Smith4344cbd2018-08-20 20:14:49 +00001542 break;
1543 case SpecialSubKind::iostream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001544 OB += "std::basic_iostream<char, std::char_traits<char> >";
Richard Smith4344cbd2018-08-20 20:14:49 +00001545 break;
1546 }
1547 }
1548};
1549
1550class SpecialSubstitution final : public Node {
1551public:
1552 SpecialSubKind SSK;
1553
1554 SpecialSubstitution(SpecialSubKind SSK_)
1555 : Node(KSpecialSubstitution), SSK(SSK_) {}
1556
1557 template<typename Fn> void match(Fn F) const { F(SSK); }
1558
1559 StringView getBaseName() const override {
1560 switch (SSK) {
1561 case SpecialSubKind::allocator:
1562 return StringView("allocator");
1563 case SpecialSubKind::basic_string:
1564 return StringView("basic_string");
1565 case SpecialSubKind::string:
1566 return StringView("string");
1567 case SpecialSubKind::istream:
1568 return StringView("istream");
1569 case SpecialSubKind::ostream:
1570 return StringView("ostream");
1571 case SpecialSubKind::iostream:
1572 return StringView("iostream");
1573 }
Erik Pilkington870950f2019-01-17 20:37:51 +00001574 DEMANGLE_UNREACHABLE;
Richard Smith4344cbd2018-08-20 20:14:49 +00001575 }
1576
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001577 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001578 switch (SSK) {
1579 case SpecialSubKind::allocator:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001580 OB += "std::allocator";
Richard Smith4344cbd2018-08-20 20:14:49 +00001581 break;
1582 case SpecialSubKind::basic_string:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001583 OB += "std::basic_string";
Richard Smith4344cbd2018-08-20 20:14:49 +00001584 break;
1585 case SpecialSubKind::string:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001586 OB += "std::string";
Richard Smith4344cbd2018-08-20 20:14:49 +00001587 break;
1588 case SpecialSubKind::istream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001589 OB += "std::istream";
Richard Smith4344cbd2018-08-20 20:14:49 +00001590 break;
1591 case SpecialSubKind::ostream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001592 OB += "std::ostream";
Richard Smith4344cbd2018-08-20 20:14:49 +00001593 break;
1594 case SpecialSubKind::iostream:
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001595 OB += "std::iostream";
Richard Smith4344cbd2018-08-20 20:14:49 +00001596 break;
1597 }
1598 }
1599};
1600
1601class CtorDtorName final : public Node {
1602 const Node *Basename;
1603 const bool IsDtor;
Pavel Labath36050092018-10-10 08:39:16 +00001604 const int Variant;
Richard Smith4344cbd2018-08-20 20:14:49 +00001605
1606public:
Pavel Labath36050092018-10-10 08:39:16 +00001607 CtorDtorName(const Node *Basename_, bool IsDtor_, int Variant_)
1608 : Node(KCtorDtorName), Basename(Basename_), IsDtor(IsDtor_),
1609 Variant(Variant_) {}
Richard Smith4344cbd2018-08-20 20:14:49 +00001610
Pavel Labath36050092018-10-10 08:39:16 +00001611 template<typename Fn> void match(Fn F) const { F(Basename, IsDtor, Variant); }
Richard Smith4344cbd2018-08-20 20:14:49 +00001612
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001613 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001614 if (IsDtor)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001615 OB += "~";
1616 OB += Basename->getBaseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00001617 }
1618};
1619
1620class DtorName : public Node {
1621 const Node *Base;
1622
1623public:
1624 DtorName(const Node *Base_) : Node(KDtorName), Base(Base_) {}
1625
1626 template<typename Fn> void match(Fn F) const { F(Base); }
1627
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001628 void printLeft(OutputBuffer &OB) const override {
1629 OB += "~";
1630 Base->printLeft(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001631 }
1632};
1633
1634class UnnamedTypeName : public Node {
1635 const StringView Count;
1636
1637public:
1638 UnnamedTypeName(StringView Count_) : Node(KUnnamedTypeName), Count(Count_) {}
1639
1640 template<typename Fn> void match(Fn F) const { F(Count); }
1641
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001642 void printLeft(OutputBuffer &OB) const override {
1643 OB += "'unnamed";
1644 OB += Count;
1645 OB += "\'";
Richard Smith4344cbd2018-08-20 20:14:49 +00001646 }
1647};
1648
1649class ClosureTypeName : public Node {
Richard Smithcbffe542019-09-06 23:53:21 +00001650 NodeArray TemplateParams;
Richard Smith4344cbd2018-08-20 20:14:49 +00001651 NodeArray Params;
1652 StringView Count;
1653
1654public:
Richard Smithcbffe542019-09-06 23:53:21 +00001655 ClosureTypeName(NodeArray TemplateParams_, NodeArray Params_,
1656 StringView Count_)
1657 : Node(KClosureTypeName), TemplateParams(TemplateParams_),
1658 Params(Params_), Count(Count_) {}
Richard Smith4344cbd2018-08-20 20:14:49 +00001659
Richard Smithcbffe542019-09-06 23:53:21 +00001660 template<typename Fn> void match(Fn F) const {
1661 F(TemplateParams, Params, Count);
1662 }
1663
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001664 void printDeclarator(OutputBuffer &OB) const {
Richard Smithcbffe542019-09-06 23:53:21 +00001665 if (!TemplateParams.empty()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001666 OB += "<";
1667 TemplateParams.printWithComma(OB);
1668 OB += ">";
Richard Smithcbffe542019-09-06 23:53:21 +00001669 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001670 OB += "(";
1671 Params.printWithComma(OB);
1672 OB += ")";
Richard Smithcbffe542019-09-06 23:53:21 +00001673 }
Richard Smith4344cbd2018-08-20 20:14:49 +00001674
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001675 void printLeft(OutputBuffer &OB) const override {
1676 OB += "\'lambda";
1677 OB += Count;
1678 OB += "\'";
1679 printDeclarator(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001680 }
1681};
1682
1683class StructuredBindingName : public Node {
1684 NodeArray Bindings;
1685public:
1686 StructuredBindingName(NodeArray Bindings_)
1687 : Node(KStructuredBindingName), Bindings(Bindings_) {}
1688
1689 template<typename Fn> void match(Fn F) const { F(Bindings); }
1690
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001691 void printLeft(OutputBuffer &OB) const override {
1692 OB += '[';
1693 Bindings.printWithComma(OB);
1694 OB += ']';
Richard Smith4344cbd2018-08-20 20:14:49 +00001695 }
1696};
1697
1698// -- Expression Nodes --
1699
1700class BinaryExpr : public Node {
1701 const Node *LHS;
1702 const StringView InfixOperator;
1703 const Node *RHS;
1704
1705public:
1706 BinaryExpr(const Node *LHS_, StringView InfixOperator_, const Node *RHS_)
1707 : Node(KBinaryExpr), LHS(LHS_), InfixOperator(InfixOperator_), RHS(RHS_) {
1708 }
1709
1710 template<typename Fn> void match(Fn F) const { F(LHS, InfixOperator, RHS); }
1711
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001712 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001713 // might be a template argument expression, then we need to disambiguate
1714 // with parens.
1715 if (InfixOperator == ">")
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001716 OB += "(";
Richard Smith4344cbd2018-08-20 20:14:49 +00001717
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001718 OB += "(";
1719 LHS->print(OB);
1720 OB += ") ";
1721 OB += InfixOperator;
1722 OB += " (";
1723 RHS->print(OB);
1724 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001725
1726 if (InfixOperator == ">")
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001727 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001728 }
1729};
1730
1731class ArraySubscriptExpr : public Node {
1732 const Node *Op1;
1733 const Node *Op2;
1734
1735public:
1736 ArraySubscriptExpr(const Node *Op1_, const Node *Op2_)
1737 : Node(KArraySubscriptExpr), Op1(Op1_), Op2(Op2_) {}
1738
1739 template<typename Fn> void match(Fn F) const { F(Op1, Op2); }
1740
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001741 void printLeft(OutputBuffer &OB) const override {
1742 OB += "(";
1743 Op1->print(OB);
1744 OB += ")[";
1745 Op2->print(OB);
1746 OB += "]";
Richard Smith4344cbd2018-08-20 20:14:49 +00001747 }
1748};
1749
1750class PostfixExpr : public Node {
1751 const Node *Child;
1752 const StringView Operator;
1753
1754public:
1755 PostfixExpr(const Node *Child_, StringView Operator_)
1756 : Node(KPostfixExpr), Child(Child_), Operator(Operator_) {}
1757
1758 template<typename Fn> void match(Fn F) const { F(Child, Operator); }
1759
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001760 void printLeft(OutputBuffer &OB) const override {
1761 OB += "(";
1762 Child->print(OB);
1763 OB += ")";
1764 OB += Operator;
Richard Smith4344cbd2018-08-20 20:14:49 +00001765 }
1766};
1767
1768class ConditionalExpr : public Node {
1769 const Node *Cond;
1770 const Node *Then;
1771 const Node *Else;
1772
1773public:
1774 ConditionalExpr(const Node *Cond_, const Node *Then_, const Node *Else_)
1775 : Node(KConditionalExpr), Cond(Cond_), Then(Then_), Else(Else_) {}
1776
1777 template<typename Fn> void match(Fn F) const { F(Cond, Then, Else); }
1778
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001779 void printLeft(OutputBuffer &OB) const override {
1780 OB += "(";
1781 Cond->print(OB);
1782 OB += ") ? (";
1783 Then->print(OB);
1784 OB += ") : (";
1785 Else->print(OB);
1786 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001787 }
1788};
1789
1790class MemberExpr : public Node {
1791 const Node *LHS;
1792 const StringView Kind;
1793 const Node *RHS;
1794
1795public:
1796 MemberExpr(const Node *LHS_, StringView Kind_, const Node *RHS_)
1797 : Node(KMemberExpr), LHS(LHS_), Kind(Kind_), RHS(RHS_) {}
1798
1799 template<typename Fn> void match(Fn F) const { F(LHS, Kind, RHS); }
1800
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001801 void printLeft(OutputBuffer &OB) const override {
1802 LHS->print(OB);
1803 OB += Kind;
1804 RHS->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001805 }
1806};
1807
Richard Smith2b38a692020-10-22 19:29:36 -07001808class SubobjectExpr : public Node {
1809 const Node *Type;
1810 const Node *SubExpr;
1811 StringView Offset;
1812 NodeArray UnionSelectors;
1813 bool OnePastTheEnd;
1814
1815public:
1816 SubobjectExpr(const Node *Type_, const Node *SubExpr_, StringView Offset_,
1817 NodeArray UnionSelectors_, bool OnePastTheEnd_)
1818 : Node(KSubobjectExpr), Type(Type_), SubExpr(SubExpr_), Offset(Offset_),
1819 UnionSelectors(UnionSelectors_), OnePastTheEnd(OnePastTheEnd_) {}
1820
1821 template<typename Fn> void match(Fn F) const {
1822 F(Type, SubExpr, Offset, UnionSelectors, OnePastTheEnd);
1823 }
1824
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001825 void printLeft(OutputBuffer &OB) const override {
1826 SubExpr->print(OB);
1827 OB += ".<";
1828 Type->print(OB);
1829 OB += " at offset ";
Richard Smith2b38a692020-10-22 19:29:36 -07001830 if (Offset.empty()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001831 OB += "0";
Richard Smith2b38a692020-10-22 19:29:36 -07001832 } else if (Offset[0] == 'n') {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001833 OB += "-";
1834 OB += Offset.dropFront();
Richard Smith2b38a692020-10-22 19:29:36 -07001835 } else {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001836 OB += Offset;
Richard Smith2b38a692020-10-22 19:29:36 -07001837 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001838 OB += ">";
Richard Smith2b38a692020-10-22 19:29:36 -07001839 }
1840};
1841
Richard Smith4344cbd2018-08-20 20:14:49 +00001842class EnclosingExpr : public Node {
1843 const StringView Prefix;
1844 const Node *Infix;
1845 const StringView Postfix;
1846
1847public:
1848 EnclosingExpr(StringView Prefix_, Node *Infix_, StringView Postfix_)
1849 : Node(KEnclosingExpr), Prefix(Prefix_), Infix(Infix_),
1850 Postfix(Postfix_) {}
1851
1852 template<typename Fn> void match(Fn F) const { F(Prefix, Infix, Postfix); }
1853
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001854 void printLeft(OutputBuffer &OB) const override {
1855 OB += Prefix;
1856 Infix->print(OB);
1857 OB += Postfix;
Richard Smith4344cbd2018-08-20 20:14:49 +00001858 }
1859};
1860
1861class CastExpr : public Node {
1862 // cast_kind<to>(from)
1863 const StringView CastKind;
1864 const Node *To;
1865 const Node *From;
1866
1867public:
1868 CastExpr(StringView CastKind_, const Node *To_, const Node *From_)
1869 : Node(KCastExpr), CastKind(CastKind_), To(To_), From(From_) {}
1870
1871 template<typename Fn> void match(Fn F) const { F(CastKind, To, From); }
1872
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001873 void printLeft(OutputBuffer &OB) const override {
1874 OB += CastKind;
1875 OB += "<";
1876 To->printLeft(OB);
1877 OB += ">(";
1878 From->printLeft(OB);
1879 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001880 }
1881};
1882
1883class SizeofParamPackExpr : public Node {
1884 const Node *Pack;
1885
1886public:
1887 SizeofParamPackExpr(const Node *Pack_)
1888 : Node(KSizeofParamPackExpr), Pack(Pack_) {}
1889
1890 template<typename Fn> void match(Fn F) const { F(Pack); }
1891
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001892 void printLeft(OutputBuffer &OB) const override {
1893 OB += "sizeof...(";
Richard Smith4344cbd2018-08-20 20:14:49 +00001894 ParameterPackExpansion PPE(Pack);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001895 PPE.printLeft(OB);
1896 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001897 }
1898};
1899
1900class CallExpr : public Node {
1901 const Node *Callee;
1902 NodeArray Args;
1903
1904public:
1905 CallExpr(const Node *Callee_, NodeArray Args_)
1906 : Node(KCallExpr), Callee(Callee_), Args(Args_) {}
1907
1908 template<typename Fn> void match(Fn F) const { F(Callee, Args); }
1909
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001910 void printLeft(OutputBuffer &OB) const override {
1911 Callee->print(OB);
1912 OB += "(";
1913 Args.printWithComma(OB);
1914 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001915 }
1916};
1917
1918class NewExpr : public Node {
1919 // new (expr_list) type(init_list)
1920 NodeArray ExprList;
1921 Node *Type;
1922 NodeArray InitList;
1923 bool IsGlobal; // ::operator new ?
1924 bool IsArray; // new[] ?
1925public:
1926 NewExpr(NodeArray ExprList_, Node *Type_, NodeArray InitList_, bool IsGlobal_,
1927 bool IsArray_)
1928 : Node(KNewExpr), ExprList(ExprList_), Type(Type_), InitList(InitList_),
1929 IsGlobal(IsGlobal_), IsArray(IsArray_) {}
1930
1931 template<typename Fn> void match(Fn F) const {
1932 F(ExprList, Type, InitList, IsGlobal, IsArray);
1933 }
1934
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001935 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001936 if (IsGlobal)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001937 OB += "::operator ";
1938 OB += "new";
Richard Smith4344cbd2018-08-20 20:14:49 +00001939 if (IsArray)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001940 OB += "[]";
1941 OB += ' ';
Richard Smith4344cbd2018-08-20 20:14:49 +00001942 if (!ExprList.empty()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001943 OB += "(";
1944 ExprList.printWithComma(OB);
1945 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001946 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001947 Type->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001948 if (!InitList.empty()) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001949 OB += "(";
1950 InitList.printWithComma(OB);
1951 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001952 }
Richard Smith4344cbd2018-08-20 20:14:49 +00001953 }
1954};
1955
1956class DeleteExpr : public Node {
1957 Node *Op;
1958 bool IsGlobal;
1959 bool IsArray;
1960
1961public:
1962 DeleteExpr(Node *Op_, bool IsGlobal_, bool IsArray_)
1963 : Node(KDeleteExpr), Op(Op_), IsGlobal(IsGlobal_), IsArray(IsArray_) {}
1964
1965 template<typename Fn> void match(Fn F) const { F(Op, IsGlobal, IsArray); }
1966
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001967 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00001968 if (IsGlobal)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001969 OB += "::";
1970 OB += "delete";
Richard Smith4344cbd2018-08-20 20:14:49 +00001971 if (IsArray)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001972 OB += "[] ";
1973 Op->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00001974 }
1975};
1976
1977class PrefixExpr : public Node {
1978 StringView Prefix;
1979 Node *Child;
1980
1981public:
1982 PrefixExpr(StringView Prefix_, Node *Child_)
1983 : Node(KPrefixExpr), Prefix(Prefix_), Child(Child_) {}
1984
1985 template<typename Fn> void match(Fn F) const { F(Prefix, Child); }
1986
Luís Ferreira4fb856a2021-10-21 17:31:53 -07001987 void printLeft(OutputBuffer &OB) const override {
1988 OB += Prefix;
1989 OB += "(";
1990 Child->print(OB);
1991 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00001992 }
1993};
1994
1995class FunctionParam : public Node {
1996 StringView Number;
1997
1998public:
1999 FunctionParam(StringView Number_) : Node(KFunctionParam), Number(Number_) {}
2000
2001 template<typename Fn> void match(Fn F) const { F(Number); }
2002
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002003 void printLeft(OutputBuffer &OB) const override {
2004 OB += "fp";
2005 OB += Number;
Richard Smith4344cbd2018-08-20 20:14:49 +00002006 }
2007};
2008
2009class ConversionExpr : public Node {
2010 const Node *Type;
2011 NodeArray Expressions;
2012
2013public:
2014 ConversionExpr(const Node *Type_, NodeArray Expressions_)
2015 : Node(KConversionExpr), Type(Type_), Expressions(Expressions_) {}
2016
2017 template<typename Fn> void match(Fn F) const { F(Type, Expressions); }
2018
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002019 void printLeft(OutputBuffer &OB) const override {
2020 OB += "(";
2021 Type->print(OB);
2022 OB += ")(";
2023 Expressions.printWithComma(OB);
2024 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00002025 }
2026};
2027
Richard Smith2b38a692020-10-22 19:29:36 -07002028class PointerToMemberConversionExpr : public Node {
2029 const Node *Type;
2030 const Node *SubExpr;
2031 StringView Offset;
2032
2033public:
2034 PointerToMemberConversionExpr(const Node *Type_, const Node *SubExpr_,
2035 StringView Offset_)
2036 : Node(KPointerToMemberConversionExpr), Type(Type_), SubExpr(SubExpr_),
2037 Offset(Offset_) {}
2038
2039 template<typename Fn> void match(Fn F) const { F(Type, SubExpr, Offset); }
2040
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002041 void printLeft(OutputBuffer &OB) const override {
2042 OB += "(";
2043 Type->print(OB);
2044 OB += ")(";
2045 SubExpr->print(OB);
2046 OB += ")";
Richard Smith2b38a692020-10-22 19:29:36 -07002047 }
2048};
2049
Richard Smith4344cbd2018-08-20 20:14:49 +00002050class InitListExpr : public Node {
2051 const Node *Ty;
2052 NodeArray Inits;
2053public:
2054 InitListExpr(const Node *Ty_, NodeArray Inits_)
2055 : Node(KInitListExpr), Ty(Ty_), Inits(Inits_) {}
2056
2057 template<typename Fn> void match(Fn F) const { F(Ty, Inits); }
2058
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002059 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00002060 if (Ty)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002061 Ty->print(OB);
2062 OB += '{';
2063 Inits.printWithComma(OB);
2064 OB += '}';
Richard Smith4344cbd2018-08-20 20:14:49 +00002065 }
2066};
2067
2068class BracedExpr : public Node {
2069 const Node *Elem;
2070 const Node *Init;
2071 bool IsArray;
2072public:
2073 BracedExpr(const Node *Elem_, const Node *Init_, bool IsArray_)
2074 : Node(KBracedExpr), Elem(Elem_), Init(Init_), IsArray(IsArray_) {}
2075
2076 template<typename Fn> void match(Fn F) const { F(Elem, Init, IsArray); }
2077
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002078 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00002079 if (IsArray) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002080 OB += '[';
2081 Elem->print(OB);
2082 OB += ']';
Richard Smith4344cbd2018-08-20 20:14:49 +00002083 } else {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002084 OB += '.';
2085 Elem->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00002086 }
2087 if (Init->getKind() != KBracedExpr && Init->getKind() != KBracedRangeExpr)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002088 OB += " = ";
2089 Init->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00002090 }
2091};
2092
2093class BracedRangeExpr : public Node {
2094 const Node *First;
2095 const Node *Last;
2096 const Node *Init;
2097public:
2098 BracedRangeExpr(const Node *First_, const Node *Last_, const Node *Init_)
2099 : Node(KBracedRangeExpr), First(First_), Last(Last_), Init(Init_) {}
2100
2101 template<typename Fn> void match(Fn F) const { F(First, Last, Init); }
2102
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002103 void printLeft(OutputBuffer &OB) const override {
2104 OB += '[';
2105 First->print(OB);
2106 OB += " ... ";
2107 Last->print(OB);
2108 OB += ']';
Richard Smith4344cbd2018-08-20 20:14:49 +00002109 if (Init->getKind() != KBracedExpr && Init->getKind() != KBracedRangeExpr)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002110 OB += " = ";
2111 Init->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00002112 }
2113};
2114
2115class FoldExpr : public Node {
2116 const Node *Pack, *Init;
2117 StringView OperatorName;
2118 bool IsLeftFold;
2119
2120public:
2121 FoldExpr(bool IsLeftFold_, StringView OperatorName_, const Node *Pack_,
2122 const Node *Init_)
2123 : Node(KFoldExpr), Pack(Pack_), Init(Init_), OperatorName(OperatorName_),
2124 IsLeftFold(IsLeftFold_) {}
2125
2126 template<typename Fn> void match(Fn F) const {
2127 F(IsLeftFold, OperatorName, Pack, Init);
2128 }
2129
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002130 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00002131 auto PrintPack = [&] {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002132 OB += '(';
2133 ParameterPackExpansion(Pack).print(OB);
2134 OB += ')';
Richard Smith4344cbd2018-08-20 20:14:49 +00002135 };
2136
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002137 OB += '(';
Richard Smith4344cbd2018-08-20 20:14:49 +00002138
2139 if (IsLeftFold) {
2140 // init op ... op pack
2141 if (Init != nullptr) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002142 Init->print(OB);
2143 OB += ' ';
2144 OB += OperatorName;
2145 OB += ' ';
Richard Smith4344cbd2018-08-20 20:14:49 +00002146 }
2147 // ... op pack
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002148 OB += "... ";
2149 OB += OperatorName;
2150 OB += ' ';
Richard Smith4344cbd2018-08-20 20:14:49 +00002151 PrintPack();
2152 } else { // !IsLeftFold
2153 // pack op ...
2154 PrintPack();
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002155 OB += ' ';
2156 OB += OperatorName;
2157 OB += " ...";
Richard Smith4344cbd2018-08-20 20:14:49 +00002158 // pack op ... op init
2159 if (Init != nullptr) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002160 OB += ' ';
2161 OB += OperatorName;
2162 OB += ' ';
2163 Init->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00002164 }
2165 }
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002166 OB += ')';
Richard Smith4344cbd2018-08-20 20:14:49 +00002167 }
2168};
2169
2170class ThrowExpr : public Node {
2171 const Node *Op;
2172
2173public:
2174 ThrowExpr(const Node *Op_) : Node(KThrowExpr), Op(Op_) {}
2175
2176 template<typename Fn> void match(Fn F) const { F(Op); }
2177
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002178 void printLeft(OutputBuffer &OB) const override {
2179 OB += "throw ";
2180 Op->print(OB);
Richard Smith4344cbd2018-08-20 20:14:49 +00002181 }
2182};
2183
2184class BoolExpr : public Node {
2185 bool Value;
2186
2187public:
2188 BoolExpr(bool Value_) : Node(KBoolExpr), Value(Value_) {}
2189
2190 template<typename Fn> void match(Fn F) const { F(Value); }
2191
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002192 void printLeft(OutputBuffer &OB) const override {
2193 OB += Value ? StringView("true") : StringView("false");
Richard Smith4344cbd2018-08-20 20:14:49 +00002194 }
2195};
2196
Richard Smithcbffe542019-09-06 23:53:21 +00002197class StringLiteral : public Node {
2198 const Node *Type;
2199
2200public:
2201 StringLiteral(const Node *Type_) : Node(KStringLiteral), Type(Type_) {}
2202
2203 template<typename Fn> void match(Fn F) const { F(Type); }
2204
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002205 void printLeft(OutputBuffer &OB) const override {
2206 OB += "\"<";
2207 Type->print(OB);
2208 OB += ">\"";
Richard Smithcbffe542019-09-06 23:53:21 +00002209 }
2210};
2211
2212class LambdaExpr : public Node {
2213 const Node *Type;
2214
Richard Smithcbffe542019-09-06 23:53:21 +00002215public:
2216 LambdaExpr(const Node *Type_) : Node(KLambdaExpr), Type(Type_) {}
2217
2218 template<typename Fn> void match(Fn F) const { F(Type); }
2219
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002220 void printLeft(OutputBuffer &OB) const override {
2221 OB += "[]";
Richard Smith069f19d2019-09-09 22:26:04 +00002222 if (Type->getKind() == KClosureTypeName)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002223 static_cast<const ClosureTypeName *>(Type)->printDeclarator(OB);
2224 OB += "{...}";
Richard Smithcbffe542019-09-06 23:53:21 +00002225 }
2226};
2227
Erik Pilkington27011a82020-05-13 14:13:37 -04002228class EnumLiteral : public Node {
Richard Smith4344cbd2018-08-20 20:14:49 +00002229 // ty(integer)
2230 const Node *Ty;
2231 StringView Integer;
2232
2233public:
Erik Pilkington27011a82020-05-13 14:13:37 -04002234 EnumLiteral(const Node *Ty_, StringView Integer_)
2235 : Node(KEnumLiteral), Ty(Ty_), Integer(Integer_) {}
Richard Smith4344cbd2018-08-20 20:14:49 +00002236
2237 template<typename Fn> void match(Fn F) const { F(Ty, Integer); }
2238
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002239 void printLeft(OutputBuffer &OB) const override {
2240 OB << "(";
2241 Ty->print(OB);
2242 OB << ")";
Erik Pilkington27011a82020-05-13 14:13:37 -04002243
2244 if (Integer[0] == 'n')
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002245 OB << "-" << Integer.dropFront(1);
Erik Pilkington27011a82020-05-13 14:13:37 -04002246 else
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002247 OB << Integer;
Richard Smith4344cbd2018-08-20 20:14:49 +00002248 }
2249};
2250
2251class IntegerLiteral : public Node {
2252 StringView Type;
2253 StringView Value;
2254
2255public:
2256 IntegerLiteral(StringView Type_, StringView Value_)
2257 : Node(KIntegerLiteral), Type(Type_), Value(Value_) {}
2258
2259 template<typename Fn> void match(Fn F) const { F(Type, Value); }
2260
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002261 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00002262 if (Type.size() > 3) {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002263 OB += "(";
2264 OB += Type;
2265 OB += ")";
Richard Smith4344cbd2018-08-20 20:14:49 +00002266 }
2267
2268 if (Value[0] == 'n') {
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002269 OB += "-";
2270 OB += Value.dropFront(1);
Richard Smith4344cbd2018-08-20 20:14:49 +00002271 } else
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002272 OB += Value;
Richard Smith4344cbd2018-08-20 20:14:49 +00002273
2274 if (Type.size() <= 3)
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002275 OB += Type;
Richard Smith4344cbd2018-08-20 20:14:49 +00002276 }
2277};
2278
2279template <class Float> struct FloatData;
2280
2281namespace float_literal_impl {
2282constexpr Node::Kind getFloatLiteralKind(float *) {
2283 return Node::KFloatLiteral;
2284}
2285constexpr Node::Kind getFloatLiteralKind(double *) {
2286 return Node::KDoubleLiteral;
2287}
2288constexpr Node::Kind getFloatLiteralKind(long double *) {
2289 return Node::KLongDoubleLiteral;
2290}
2291}
2292
2293template <class Float> class FloatLiteralImpl : public Node {
2294 const StringView Contents;
2295
2296 static constexpr Kind KindForClass =
2297 float_literal_impl::getFloatLiteralKind((Float *)nullptr);
2298
2299public:
2300 FloatLiteralImpl(StringView Contents_)
2301 : Node(KindForClass), Contents(Contents_) {}
2302
2303 template<typename Fn> void match(Fn F) const { F(Contents); }
2304
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002305 void printLeft(OutputBuffer &OB) const override {
Richard Smith4344cbd2018-08-20 20:14:49 +00002306 const char *first = Contents.begin();
2307 const char *last = Contents.end() + 1;
2308
2309 const size_t N = FloatData<Float>::mangled_size;
2310 if (static_cast<std::size_t>(last - first) > N) {
2311 last = first + N;
2312 union {
2313 Float value;
2314 char buf[sizeof(Float)];
2315 };
2316 const char *t = first;
2317 char *e = buf;
2318 for (; t != last; ++t, ++e) {
2319 unsigned d1 = isdigit(*t) ? static_cast<unsigned>(*t - '0')
2320 : static_cast<unsigned>(*t - 'a' + 10);
2321 ++t;
2322 unsigned d0 = isdigit(*t) ? static_cast<unsigned>(*t - '0')
2323 : static_cast<unsigned>(*t - 'a' + 10);
2324 *e = static_cast<char>((d1 << 4) + d0);
2325 }
2326#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
2327 std::reverse(buf, e);
2328#endif
2329 char num[FloatData<Float>::max_demangled_size] = {0};
2330 int n = snprintf(num, sizeof(num), FloatData<Float>::spec, value);
Luís Ferreira4fb856a2021-10-21 17:31:53 -07002331 OB += StringView(num, num + n);
Richard Smith4344cbd2018-08-20 20:14:49 +00002332 }
2333 }
2334};
2335
2336using FloatLiteral = FloatLiteralImpl<float>;
2337using DoubleLiteral = FloatLiteralImpl<double>;
2338using LongDoubleLiteral = FloatLiteralImpl<long double>;
2339
2340/// Visit the node. Calls \c F(P), where \c P is the node cast to the
2341/// appropriate derived class.
2342template<typename Fn>
2343void Node::visit(Fn F) const {
2344 switch (K) {
2345#define CASE(X) case K ## X: return F(static_cast<const X*>(this));
2346 FOR_EACH_NODE_KIND(CASE)
2347#undef CASE
2348 }
2349 assert(0 && "unknown mangling node kind");
2350}
2351
2352/// Determine the kind of a node from its type.
2353template<typename NodeT> struct NodeKind;
2354#define SPECIALIZATION(X) \
2355 template<> struct NodeKind<X> { \
2356 static constexpr Node::Kind Kind = Node::K##X; \
2357 static constexpr const char *name() { return #X; } \
2358 };
2359FOR_EACH_NODE_KIND(SPECIALIZATION)
2360#undef SPECIALIZATION
2361
2362#undef FOR_EACH_NODE_KIND
2363
Pavel Labathb26cabc2018-10-16 14:29:14 +00002364template <typename Derived, typename Alloc> struct AbstractManglingParser {
Richard Smith4344cbd2018-08-20 20:14:49 +00002365 const char *First;
2366 const char *Last;
2367
2368 // Name stack, this is used by the parser to hold temporary names that were
2369 // parsed. The parser collapses multiple names into new nodes to construct
2370 // the AST. Once the parser is finished, names.size() == 1.
2371 PODSmallVector<Node *, 32> Names;
2372
2373 // Substitution table. Itanium supports name substitutions as a means of
2374 // compression. The string "S42_" refers to the 44nd entry (base-36) in this
2375 // table.
2376 PODSmallVector<Node *, 32> Subs;
2377
Richard Smithcbffe542019-09-06 23:53:21 +00002378 using TemplateParamList = PODSmallVector<Node *, 8>;
2379
2380 class ScopedTemplateParamList {
2381 AbstractManglingParser *Parser;
2382 size_t OldNumTemplateParamLists;
2383 TemplateParamList Params;
2384
2385 public:
Louis Dionne83a75212020-10-30 17:33:02 -04002386 ScopedTemplateParamList(AbstractManglingParser *TheParser)
2387 : Parser(TheParser),
2388 OldNumTemplateParamLists(TheParser->TemplateParams.size()) {
Richard Smithcbffe542019-09-06 23:53:21 +00002389 Parser->TemplateParams.push_back(&Params);
2390 }
2391 ~ScopedTemplateParamList() {
2392 assert(Parser->TemplateParams.size() >= OldNumTemplateParamLists);
2393 Parser->TemplateParams.dropBack(OldNumTemplateParamLists);
2394 }
Richard Smithcbffe542019-09-06 23:53:21 +00002395 };
2396
Richard Smith4344cbd2018-08-20 20:14:49 +00002397 // Template parameter table. Like the above, but referenced like "T42_".
2398 // This has a smaller size compared to Subs and Names because it can be
2399 // stored on the stack.
Richard Smithcbffe542019-09-06 23:53:21 +00002400 TemplateParamList OuterTemplateParams;
2401
2402 // Lists of template parameters indexed by template parameter depth,
2403 // referenced like "TL2_4_". If nonempty, element 0 is always
2404 // OuterTemplateParams; inner elements are always template parameter lists of
2405 // lambda expressions. For a generic lambda with no explicit template
2406 // parameter list, the corresponding parameter list pointer will be null.
2407 PODSmallVector<TemplateParamList *, 4> TemplateParams;
Richard Smith4344cbd2018-08-20 20:14:49 +00002408
2409 // Set of unresolved forward <template-param> references. These can occur in a
2410 // conversion operator's type, and are resolved in the enclosing <encoding>.
2411 PODSmallVector<ForwardTemplateReference *, 4> ForwardTemplateRefs;
2412
Richard Smith4344cbd2018-08-20 20:14:49 +00002413 bool TryToParseTemplateArgs = true;
2414 bool PermitForwardTemplateReferences = false;
Richard Smithcbffe542019-09-06 23:53:21 +00002415 size_t ParsingLambdaParamsAtLevel = (size_t)-1;
2416
2417 unsigned NumSyntheticTemplateParameters[3] = {};
Richard Smith4344cbd2018-08-20 20:14:49 +00002418
2419 Alloc ASTAllocator;
2420
Pavel Labathb26cabc2018-10-16 14:29:14 +00002421 AbstractManglingParser(const char *First_, const char *Last_)
2422 : First(First_), Last(Last_) {}
2423
2424 Derived &getDerived() { return static_cast<Derived &>(*this); }
Richard Smith4344cbd2018-08-20 20:14:49 +00002425
2426 void reset(const char *First_, const char *Last_) {
2427 First = First_;
2428 Last = Last_;
2429 Names.clear();
2430 Subs.clear();
2431 TemplateParams.clear();
Richard Smithcbffe542019-09-06 23:53:21 +00002432 ParsingLambdaParamsAtLevel = (size_t)-1;
Richard Smith4344cbd2018-08-20 20:14:49 +00002433 TryToParseTemplateArgs = true;
2434 PermitForwardTemplateReferences = false;
Richard Smithb38f7002019-09-07 00:11:53 +00002435 for (int I = 0; I != 3; ++I)
2436 NumSyntheticTemplateParameters[I] = 0;
Richard Smith4344cbd2018-08-20 20:14:49 +00002437 ASTAllocator.reset();
2438 }
2439
Richard Smith7918dea2018-08-24 23:30:26 +00002440 template <class T, class... Args> Node *make(Args &&... args) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002441 return ASTAllocator.template makeNode<T>(std::forward<Args>(args)...);
2442 }
2443
2444 template <class It> NodeArray makeNodeArray(It begin, It end) {
2445 size_t sz = static_cast<size_t>(end - begin);
2446 void *mem = ASTAllocator.allocateNodeArray(sz);
2447 Node **data = new (mem) Node *[sz];
2448 std::copy(begin, end, data);
2449 return NodeArray(data, sz);
2450 }
2451
2452 NodeArray popTrailingNodeArray(size_t FromPosition) {
2453 assert(FromPosition <= Names.size());
2454 NodeArray res =
2455 makeNodeArray(Names.begin() + (long)FromPosition, Names.end());
2456 Names.dropBack(FromPosition);
2457 return res;
2458 }
2459
2460 bool consumeIf(StringView S) {
2461 if (StringView(First, Last).startsWith(S)) {
2462 First += S.size();
2463 return true;
2464 }
2465 return false;
2466 }
2467
2468 bool consumeIf(char C) {
2469 if (First != Last && *First == C) {
2470 ++First;
2471 return true;
2472 }
2473 return false;
2474 }
2475
2476 char consume() { return First != Last ? *First++ : '\0'; }
2477
Nathan Sidwell64e94ec2022-01-20 07:40:12 -08002478 char look(unsigned Lookahead = 0) const {
Richard Smith4344cbd2018-08-20 20:14:49 +00002479 if (static_cast<size_t>(Last - First) <= Lookahead)
2480 return '\0';
2481 return First[Lookahead];
2482 }
2483
2484 size_t numLeft() const { return static_cast<size_t>(Last - First); }
2485
2486 StringView parseNumber(bool AllowNegative = false);
2487 Qualifiers parseCVQualifiers();
2488 bool parsePositiveInteger(size_t *Out);
2489 StringView parseBareSourceName();
2490
2491 bool parseSeqId(size_t *Out);
2492 Node *parseSubstitution();
2493 Node *parseTemplateParam();
Richard Smithcbffe542019-09-06 23:53:21 +00002494 Node *parseTemplateParamDecl();
Richard Smith4344cbd2018-08-20 20:14:49 +00002495 Node *parseTemplateArgs(bool TagTemplates = false);
2496 Node *parseTemplateArg();
2497
2498 /// Parse the <expr> production.
2499 Node *parseExpr();
2500 Node *parsePrefixExpr(StringView Kind);
2501 Node *parseBinaryExpr(StringView Kind);
2502 Node *parseIntegerLiteral(StringView Lit);
2503 Node *parseExprPrimary();
2504 template <class Float> Node *parseFloatingLiteral();
2505 Node *parseFunctionParam();
2506 Node *parseNewExpr();
2507 Node *parseConversionExpr();
2508 Node *parseBracedExpr();
2509 Node *parseFoldExpr();
Richard Smith2b38a692020-10-22 19:29:36 -07002510 Node *parsePointerToMemberConversionExpr();
2511 Node *parseSubobjectExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00002512
2513 /// Parse the <type> production.
2514 Node *parseType();
2515 Node *parseFunctionType();
2516 Node *parseVectorType();
2517 Node *parseDecltype();
2518 Node *parseArrayType();
2519 Node *parsePointerToMemberType();
2520 Node *parseClassEnumType();
2521 Node *parseQualifiedType();
2522
2523 Node *parseEncoding();
2524 bool parseCallOffset();
2525 Node *parseSpecialName();
2526
2527 /// Holds some extra information about a <name> that is being parsed. This
2528 /// information is only pertinent if the <name> refers to an <encoding>.
2529 struct NameState {
2530 bool CtorDtorConversion = false;
2531 bool EndsWithTemplateArgs = false;
2532 Qualifiers CVQualifiers = QualNone;
2533 FunctionRefQual ReferenceQualifier = FrefQualNone;
2534 size_t ForwardTemplateRefsBegin;
2535
Pavel Labathb26cabc2018-10-16 14:29:14 +00002536 NameState(AbstractManglingParser *Enclosing)
Richard Smith4344cbd2018-08-20 20:14:49 +00002537 : ForwardTemplateRefsBegin(Enclosing->ForwardTemplateRefs.size()) {}
2538 };
2539
2540 bool resolveForwardTemplateRefs(NameState &State) {
2541 size_t I = State.ForwardTemplateRefsBegin;
2542 size_t E = ForwardTemplateRefs.size();
2543 for (; I < E; ++I) {
2544 size_t Idx = ForwardTemplateRefs[I]->Index;
Richard Smithcbffe542019-09-06 23:53:21 +00002545 if (TemplateParams.empty() || !TemplateParams[0] ||
2546 Idx >= TemplateParams[0]->size())
Richard Smith4344cbd2018-08-20 20:14:49 +00002547 return true;
Richard Smithcbffe542019-09-06 23:53:21 +00002548 ForwardTemplateRefs[I]->Ref = (*TemplateParams[0])[Idx];
Richard Smith4344cbd2018-08-20 20:14:49 +00002549 }
2550 ForwardTemplateRefs.dropBack(State.ForwardTemplateRefsBegin);
2551 return false;
2552 }
2553
2554 /// Parse the <name> production>
2555 Node *parseName(NameState *State = nullptr);
2556 Node *parseLocalName(NameState *State);
2557 Node *parseOperatorName(NameState *State);
2558 Node *parseUnqualifiedName(NameState *State);
2559 Node *parseUnnamedTypeName(NameState *State);
2560 Node *parseSourceName(NameState *State);
2561 Node *parseUnscopedName(NameState *State);
2562 Node *parseNestedName(NameState *State);
2563 Node *parseCtorDtorName(Node *&SoFar, NameState *State);
2564
2565 Node *parseAbiTags(Node *N);
2566
2567 /// Parse the <unresolved-name> production.
2568 Node *parseUnresolvedName();
2569 Node *parseSimpleId();
2570 Node *parseBaseUnresolvedName();
2571 Node *parseUnresolvedType();
2572 Node *parseDestructorName();
2573
2574 /// Top-level entry point into the parser.
2575 Node *parse();
2576};
2577
2578const char* parse_discriminator(const char* first, const char* last);
2579
2580// <name> ::= <nested-name> // N
2581// ::= <local-name> # See Scope Encoding below // Z
2582// ::= <unscoped-template-name> <template-args>
2583// ::= <unscoped-name>
2584//
2585// <unscoped-template-name> ::= <unscoped-name>
2586// ::= <substitution>
Pavel Labathb26cabc2018-10-16 14:29:14 +00002587template <typename Derived, typename Alloc>
2588Node *AbstractManglingParser<Derived, Alloc>::parseName(NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002589 consumeIf('L'); // extension
2590
2591 if (look() == 'N')
Pavel Labathb26cabc2018-10-16 14:29:14 +00002592 return getDerived().parseNestedName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002593 if (look() == 'Z')
Pavel Labathb26cabc2018-10-16 14:29:14 +00002594 return getDerived().parseLocalName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002595
Nathan Sidwellbbc24432022-01-24 04:28:09 -08002596 Node *Result = nullptr;
2597 bool IsSubst = look() == 'S' && look(1) != 't';
2598 if (IsSubst) {
2599 // A substitution must lead to:
2600 // ::= <unscoped-template-name> <template-args>
2601 Result = getDerived().parseSubstitution();
2602 } else {
2603 // An unscoped name can be one of:
2604 // ::= <unscoped-name>
2605 // ::= <unscoped-template-name> <template-args>
2606 Result = getDerived().parseUnscopedName(State);
2607 }
2608 if (Result == nullptr)
2609 return nullptr;
2610
2611 if (look() == 'I') {
2612 // ::= <unscoped-template-name> <template-args>
2613 if (!IsSubst)
2614 // An unscoped-template-name is substitutable.
2615 Subs.push_back(Result);
Pavel Labathb26cabc2018-10-16 14:29:14 +00002616 Node *TA = getDerived().parseTemplateArgs(State != nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00002617 if (TA == nullptr)
2618 return nullptr;
Nathan Sidwellbbc24432022-01-24 04:28:09 -08002619 if (State)
2620 State->EndsWithTemplateArgs = true;
2621 Result = make<NameWithTemplateArgs>(Result, TA);
2622 } else if (IsSubst) {
2623 // The substitution case must be followed by <template-args>.
2624 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00002625 }
2626
Nathan Sidwellbbc24432022-01-24 04:28:09 -08002627 return Result;
Richard Smith4344cbd2018-08-20 20:14:49 +00002628}
2629
2630// <local-name> := Z <function encoding> E <entity name> [<discriminator>]
2631// := Z <function encoding> E s [<discriminator>]
2632// := Z <function encoding> Ed [ <parameter number> ] _ <entity name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00002633template <typename Derived, typename Alloc>
2634Node *AbstractManglingParser<Derived, Alloc>::parseLocalName(NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002635 if (!consumeIf('Z'))
2636 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00002637 Node *Encoding = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00002638 if (Encoding == nullptr || !consumeIf('E'))
2639 return nullptr;
2640
2641 if (consumeIf('s')) {
2642 First = parse_discriminator(First, Last);
Richard Smith7918dea2018-08-24 23:30:26 +00002643 auto *StringLitName = make<NameType>("string literal");
2644 if (!StringLitName)
2645 return nullptr;
2646 return make<LocalName>(Encoding, StringLitName);
Richard Smith4344cbd2018-08-20 20:14:49 +00002647 }
2648
2649 if (consumeIf('d')) {
2650 parseNumber(true);
2651 if (!consumeIf('_'))
2652 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00002653 Node *N = getDerived().parseName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002654 if (N == nullptr)
2655 return nullptr;
2656 return make<LocalName>(Encoding, N);
2657 }
2658
Pavel Labathb26cabc2018-10-16 14:29:14 +00002659 Node *Entity = getDerived().parseName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002660 if (Entity == nullptr)
2661 return nullptr;
2662 First = parse_discriminator(First, Last);
2663 return make<LocalName>(Encoding, Entity);
2664}
2665
2666// <unscoped-name> ::= <unqualified-name>
2667// ::= St <unqualified-name> # ::std::
2668// extension ::= StL<unqualified-name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00002669template <typename Derived, typename Alloc>
2670Node *
2671AbstractManglingParser<Derived, Alloc>::parseUnscopedName(NameState *State) {
Nathan Sidwellbbc24432022-01-24 04:28:09 -08002672 bool IsStd = consumeIf("St");
2673 if (IsStd)
2674 consumeIf('L');
2675
2676 Node *Result = getDerived().parseUnqualifiedName(State);
2677 if (Result == nullptr)
2678 return nullptr;
2679 if (IsStd)
2680 Result = make<StdQualifiedName>(Result);
2681
2682 return Result;
Richard Smith4344cbd2018-08-20 20:14:49 +00002683}
2684
2685// <unqualified-name> ::= <operator-name> [abi-tags]
2686// ::= <ctor-dtor-name>
2687// ::= <source-name>
2688// ::= <unnamed-type-name>
2689// ::= DC <source-name>+ E # structured binding declaration
Pavel Labathb26cabc2018-10-16 14:29:14 +00002690template <typename Derived, typename Alloc>
2691Node *
2692AbstractManglingParser<Derived, Alloc>::parseUnqualifiedName(NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002693 // <ctor-dtor-name>s are special-cased in parseNestedName().
2694 Node *Result;
2695 if (look() == 'U')
Pavel Labathb26cabc2018-10-16 14:29:14 +00002696 Result = getDerived().parseUnnamedTypeName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002697 else if (look() >= '1' && look() <= '9')
Pavel Labathb26cabc2018-10-16 14:29:14 +00002698 Result = getDerived().parseSourceName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002699 else if (consumeIf("DC")) {
2700 size_t BindingsBegin = Names.size();
2701 do {
Pavel Labathb26cabc2018-10-16 14:29:14 +00002702 Node *Binding = getDerived().parseSourceName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002703 if (Binding == nullptr)
2704 return nullptr;
2705 Names.push_back(Binding);
2706 } while (!consumeIf('E'));
2707 Result = make<StructuredBindingName>(popTrailingNodeArray(BindingsBegin));
2708 } else
Pavel Labathb26cabc2018-10-16 14:29:14 +00002709 Result = getDerived().parseOperatorName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002710 if (Result != nullptr)
Pavel Labathb26cabc2018-10-16 14:29:14 +00002711 Result = getDerived().parseAbiTags(Result);
Richard Smith4344cbd2018-08-20 20:14:49 +00002712 return Result;
2713}
2714
2715// <unnamed-type-name> ::= Ut [<nonnegative number>] _
2716// ::= <closure-type-name>
2717//
2718// <closure-type-name> ::= Ul <lambda-sig> E [ <nonnegative number> ] _
2719//
2720// <lambda-sig> ::= <parameter type>+ # Parameter types or "v" if the lambda has no parameters
Pavel Labathb26cabc2018-10-16 14:29:14 +00002721template <typename Derived, typename Alloc>
2722Node *
Richard Smithcbffe542019-09-06 23:53:21 +00002723AbstractManglingParser<Derived, Alloc>::parseUnnamedTypeName(NameState *State) {
2724 // <template-params> refer to the innermost <template-args>. Clear out any
2725 // outer args that we may have inserted into TemplateParams.
2726 if (State != nullptr)
2727 TemplateParams.clear();
2728
Richard Smith4344cbd2018-08-20 20:14:49 +00002729 if (consumeIf("Ut")) {
2730 StringView Count = parseNumber();
2731 if (!consumeIf('_'))
2732 return nullptr;
2733 return make<UnnamedTypeName>(Count);
2734 }
2735 if (consumeIf("Ul")) {
Richard Smithcbffe542019-09-06 23:53:21 +00002736 SwapAndRestore<size_t> SwapParams(ParsingLambdaParamsAtLevel,
2737 TemplateParams.size());
2738 ScopedTemplateParamList LambdaTemplateParams(this);
2739
2740 size_t ParamsBegin = Names.size();
2741 while (look() == 'T' &&
2742 StringView("yptn").find(look(1)) != StringView::npos) {
2743 Node *T = parseTemplateParamDecl();
2744 if (!T)
2745 return nullptr;
Richard Smithcbffe542019-09-06 23:53:21 +00002746 Names.push_back(T);
2747 }
2748 NodeArray TempParams = popTrailingNodeArray(ParamsBegin);
2749
2750 // FIXME: If TempParams is empty and none of the function parameters
2751 // includes 'auto', we should remove LambdaTemplateParams from the
2752 // TemplateParams list. Unfortunately, we don't find out whether there are
2753 // any 'auto' parameters until too late in an example such as:
2754 //
2755 // template<typename T> void f(
2756 // decltype([](decltype([]<typename T>(T v) {}),
2757 // auto) {})) {}
2758 // template<typename T> void f(
2759 // decltype([](decltype([]<typename T>(T w) {}),
2760 // int) {})) {}
2761 //
2762 // Here, the type of v is at level 2 but the type of w is at level 1. We
2763 // don't find this out until we encounter the type of the next parameter.
2764 //
2765 // However, compilers can't actually cope with the former example in
2766 // practice, and it's likely to be made ill-formed in future, so we don't
2767 // need to support it here.
2768 //
2769 // If we encounter an 'auto' in the function parameter types, we will
2770 // recreate a template parameter scope for it, but any intervening lambdas
2771 // will be parsed in the 'wrong' template parameter depth.
2772 if (TempParams.empty())
2773 TemplateParams.pop_back();
2774
Richard Smith4344cbd2018-08-20 20:14:49 +00002775 if (!consumeIf("vE")) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002776 do {
Pavel Labathb26cabc2018-10-16 14:29:14 +00002777 Node *P = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00002778 if (P == nullptr)
2779 return nullptr;
2780 Names.push_back(P);
2781 } while (!consumeIf('E'));
Richard Smith4344cbd2018-08-20 20:14:49 +00002782 }
Richard Smithcbffe542019-09-06 23:53:21 +00002783 NodeArray Params = popTrailingNodeArray(ParamsBegin);
2784
Richard Smith4344cbd2018-08-20 20:14:49 +00002785 StringView Count = parseNumber();
2786 if (!consumeIf('_'))
2787 return nullptr;
Richard Smithcbffe542019-09-06 23:53:21 +00002788 return make<ClosureTypeName>(TempParams, Params, Count);
Richard Smith4344cbd2018-08-20 20:14:49 +00002789 }
Erik Pilkingtonf0b74392019-01-17 21:37:51 +00002790 if (consumeIf("Ub")) {
2791 (void)parseNumber();
2792 if (!consumeIf('_'))
2793 return nullptr;
2794 return make<NameType>("'block-literal'");
2795 }
Richard Smith4344cbd2018-08-20 20:14:49 +00002796 return nullptr;
2797}
2798
2799// <source-name> ::= <positive length number> <identifier>
Pavel Labathb26cabc2018-10-16 14:29:14 +00002800template <typename Derived, typename Alloc>
2801Node *AbstractManglingParser<Derived, Alloc>::parseSourceName(NameState *) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002802 size_t Length = 0;
2803 if (parsePositiveInteger(&Length))
2804 return nullptr;
2805 if (numLeft() < Length || Length == 0)
2806 return nullptr;
2807 StringView Name(First, First + Length);
2808 First += Length;
2809 if (Name.startsWith("_GLOBAL__N"))
2810 return make<NameType>("(anonymous namespace)");
2811 return make<NameType>(Name);
2812}
2813
2814// <operator-name> ::= aa # &&
2815// ::= ad # & (unary)
2816// ::= an # &
2817// ::= aN # &=
2818// ::= aS # =
2819// ::= cl # ()
2820// ::= cm # ,
2821// ::= co # ~
2822// ::= cv <type> # (cast)
2823// ::= da # delete[]
2824// ::= de # * (unary)
2825// ::= dl # delete
2826// ::= dv # /
2827// ::= dV # /=
2828// ::= eo # ^
2829// ::= eO # ^=
2830// ::= eq # ==
2831// ::= ge # >=
2832// ::= gt # >
2833// ::= ix # []
2834// ::= le # <=
2835// ::= li <source-name> # operator ""
2836// ::= ls # <<
2837// ::= lS # <<=
2838// ::= lt # <
2839// ::= mi # -
2840// ::= mI # -=
2841// ::= ml # *
2842// ::= mL # *=
2843// ::= mm # -- (postfix in <expression> context)
2844// ::= na # new[]
2845// ::= ne # !=
2846// ::= ng # - (unary)
2847// ::= nt # !
2848// ::= nw # new
2849// ::= oo # ||
2850// ::= or # |
2851// ::= oR # |=
2852// ::= pm # ->*
2853// ::= pl # +
2854// ::= pL # +=
2855// ::= pp # ++ (postfix in <expression> context)
2856// ::= ps # + (unary)
2857// ::= pt # ->
2858// ::= qu # ?
2859// ::= rm # %
2860// ::= rM # %=
2861// ::= rs # >>
2862// ::= rS # >>=
2863// ::= ss # <=> C++2a
2864// ::= v <digit> <source-name> # vendor extended operator
Pavel Labathb26cabc2018-10-16 14:29:14 +00002865template <typename Derived, typename Alloc>
2866Node *
2867AbstractManglingParser<Derived, Alloc>::parseOperatorName(NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00002868 switch (look()) {
2869 case 'a':
2870 switch (look(1)) {
2871 case 'a':
2872 First += 2;
2873 return make<NameType>("operator&&");
2874 case 'd':
2875 case 'n':
2876 First += 2;
2877 return make<NameType>("operator&");
2878 case 'N':
2879 First += 2;
2880 return make<NameType>("operator&=");
2881 case 'S':
2882 First += 2;
2883 return make<NameType>("operator=");
2884 }
2885 return nullptr;
2886 case 'c':
2887 switch (look(1)) {
2888 case 'l':
2889 First += 2;
2890 return make<NameType>("operator()");
2891 case 'm':
2892 First += 2;
2893 return make<NameType>("operator,");
2894 case 'o':
2895 First += 2;
2896 return make<NameType>("operator~");
2897 // ::= cv <type> # (cast)
2898 case 'v': {
2899 First += 2;
2900 SwapAndRestore<bool> SaveTemplate(TryToParseTemplateArgs, false);
2901 // If we're parsing an encoding, State != nullptr and the conversion
2902 // operators' <type> could have a <template-param> that refers to some
2903 // <template-arg>s further ahead in the mangled name.
2904 SwapAndRestore<bool> SavePermit(PermitForwardTemplateReferences,
2905 PermitForwardTemplateReferences ||
2906 State != nullptr);
Pavel Labathb26cabc2018-10-16 14:29:14 +00002907 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00002908 if (Ty == nullptr)
2909 return nullptr;
2910 if (State) State->CtorDtorConversion = true;
2911 return make<ConversionOperatorType>(Ty);
2912 }
2913 }
2914 return nullptr;
2915 case 'd':
2916 switch (look(1)) {
2917 case 'a':
2918 First += 2;
2919 return make<NameType>("operator delete[]");
2920 case 'e':
2921 First += 2;
2922 return make<NameType>("operator*");
2923 case 'l':
2924 First += 2;
2925 return make<NameType>("operator delete");
2926 case 'v':
2927 First += 2;
2928 return make<NameType>("operator/");
2929 case 'V':
2930 First += 2;
2931 return make<NameType>("operator/=");
2932 }
2933 return nullptr;
2934 case 'e':
2935 switch (look(1)) {
2936 case 'o':
2937 First += 2;
2938 return make<NameType>("operator^");
2939 case 'O':
2940 First += 2;
2941 return make<NameType>("operator^=");
2942 case 'q':
2943 First += 2;
2944 return make<NameType>("operator==");
2945 }
2946 return nullptr;
2947 case 'g':
2948 switch (look(1)) {
2949 case 'e':
2950 First += 2;
2951 return make<NameType>("operator>=");
2952 case 't':
2953 First += 2;
2954 return make<NameType>("operator>");
2955 }
2956 return nullptr;
2957 case 'i':
2958 if (look(1) == 'x') {
2959 First += 2;
2960 return make<NameType>("operator[]");
2961 }
2962 return nullptr;
2963 case 'l':
2964 switch (look(1)) {
2965 case 'e':
2966 First += 2;
2967 return make<NameType>("operator<=");
2968 // ::= li <source-name> # operator ""
2969 case 'i': {
2970 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00002971 Node *SN = getDerived().parseSourceName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00002972 if (SN == nullptr)
2973 return nullptr;
2974 return make<LiteralOperator>(SN);
2975 }
2976 case 's':
2977 First += 2;
2978 return make<NameType>("operator<<");
2979 case 'S':
2980 First += 2;
2981 return make<NameType>("operator<<=");
2982 case 't':
2983 First += 2;
2984 return make<NameType>("operator<");
2985 }
2986 return nullptr;
2987 case 'm':
2988 switch (look(1)) {
2989 case 'i':
2990 First += 2;
2991 return make<NameType>("operator-");
2992 case 'I':
2993 First += 2;
2994 return make<NameType>("operator-=");
2995 case 'l':
2996 First += 2;
2997 return make<NameType>("operator*");
2998 case 'L':
2999 First += 2;
3000 return make<NameType>("operator*=");
3001 case 'm':
3002 First += 2;
3003 return make<NameType>("operator--");
3004 }
3005 return nullptr;
3006 case 'n':
3007 switch (look(1)) {
3008 case 'a':
3009 First += 2;
3010 return make<NameType>("operator new[]");
3011 case 'e':
3012 First += 2;
3013 return make<NameType>("operator!=");
3014 case 'g':
3015 First += 2;
3016 return make<NameType>("operator-");
3017 case 't':
3018 First += 2;
3019 return make<NameType>("operator!");
3020 case 'w':
3021 First += 2;
3022 return make<NameType>("operator new");
3023 }
3024 return nullptr;
3025 case 'o':
3026 switch (look(1)) {
3027 case 'o':
3028 First += 2;
3029 return make<NameType>("operator||");
3030 case 'r':
3031 First += 2;
3032 return make<NameType>("operator|");
3033 case 'R':
3034 First += 2;
3035 return make<NameType>("operator|=");
3036 }
3037 return nullptr;
3038 case 'p':
3039 switch (look(1)) {
3040 case 'm':
3041 First += 2;
3042 return make<NameType>("operator->*");
3043 case 'l':
3044 First += 2;
3045 return make<NameType>("operator+");
3046 case 'L':
3047 First += 2;
3048 return make<NameType>("operator+=");
3049 case 'p':
3050 First += 2;
3051 return make<NameType>("operator++");
3052 case 's':
3053 First += 2;
3054 return make<NameType>("operator+");
3055 case 't':
3056 First += 2;
3057 return make<NameType>("operator->");
3058 }
3059 return nullptr;
3060 case 'q':
3061 if (look(1) == 'u') {
3062 First += 2;
3063 return make<NameType>("operator?");
3064 }
3065 return nullptr;
3066 case 'r':
3067 switch (look(1)) {
3068 case 'm':
3069 First += 2;
3070 return make<NameType>("operator%");
3071 case 'M':
3072 First += 2;
3073 return make<NameType>("operator%=");
3074 case 's':
3075 First += 2;
3076 return make<NameType>("operator>>");
3077 case 'S':
3078 First += 2;
3079 return make<NameType>("operator>>=");
3080 }
3081 return nullptr;
3082 case 's':
3083 if (look(1) == 's') {
3084 First += 2;
3085 return make<NameType>("operator<=>");
3086 }
3087 return nullptr;
3088 // ::= v <digit> <source-name> # vendor extended operator
3089 case 'v':
3090 if (std::isdigit(look(1))) {
3091 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003092 Node *SN = getDerived().parseSourceName(State);
Richard Smith4344cbd2018-08-20 20:14:49 +00003093 if (SN == nullptr)
3094 return nullptr;
3095 return make<ConversionOperatorType>(SN);
3096 }
3097 return nullptr;
3098 }
3099 return nullptr;
3100}
3101
3102// <ctor-dtor-name> ::= C1 # complete object constructor
3103// ::= C2 # base object constructor
3104// ::= C3 # complete object allocating constructor
Nico Weberba88b372019-04-03 23:14:33 +00003105// extension ::= C4 # gcc old-style "[unified]" constructor
3106// extension ::= C5 # the COMDAT used for ctors
Richard Smith4344cbd2018-08-20 20:14:49 +00003107// ::= D0 # deleting destructor
3108// ::= D1 # complete object destructor
3109// ::= D2 # base object destructor
Nico Weberba88b372019-04-03 23:14:33 +00003110// extension ::= D4 # gcc old-style "[unified]" destructor
3111// extension ::= D5 # the COMDAT used for dtors
Pavel Labathb26cabc2018-10-16 14:29:14 +00003112template <typename Derived, typename Alloc>
3113Node *
3114AbstractManglingParser<Derived, Alloc>::parseCtorDtorName(Node *&SoFar,
3115 NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003116 if (SoFar->getKind() == Node::KSpecialSubstitution) {
3117 auto SSK = static_cast<SpecialSubstitution *>(SoFar)->SSK;
3118 switch (SSK) {
3119 case SpecialSubKind::string:
3120 case SpecialSubKind::istream:
3121 case SpecialSubKind::ostream:
3122 case SpecialSubKind::iostream:
3123 SoFar = make<ExpandedSpecialSubstitution>(SSK);
Richard Smith7918dea2018-08-24 23:30:26 +00003124 if (!SoFar)
3125 return nullptr;
Reid Kleckner9ddef322018-11-01 18:24:03 +00003126 break;
Richard Smith4344cbd2018-08-20 20:14:49 +00003127 default:
3128 break;
3129 }
3130 }
3131
3132 if (consumeIf('C')) {
3133 bool IsInherited = consumeIf('I');
Nico Weberba88b372019-04-03 23:14:33 +00003134 if (look() != '1' && look() != '2' && look() != '3' && look() != '4' &&
3135 look() != '5')
Richard Smith4344cbd2018-08-20 20:14:49 +00003136 return nullptr;
Pavel Labath36050092018-10-10 08:39:16 +00003137 int Variant = look() - '0';
Richard Smith4344cbd2018-08-20 20:14:49 +00003138 ++First;
3139 if (State) State->CtorDtorConversion = true;
3140 if (IsInherited) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003141 if (getDerived().parseName(State) == nullptr)
Richard Smith4344cbd2018-08-20 20:14:49 +00003142 return nullptr;
3143 }
Nico Weberba88b372019-04-03 23:14:33 +00003144 return make<CtorDtorName>(SoFar, /*IsDtor=*/false, Variant);
Richard Smith4344cbd2018-08-20 20:14:49 +00003145 }
3146
Nico Weberba88b372019-04-03 23:14:33 +00003147 if (look() == 'D' && (look(1) == '0' || look(1) == '1' || look(1) == '2' ||
3148 look(1) == '4' || look(1) == '5')) {
Pavel Labath36050092018-10-10 08:39:16 +00003149 int Variant = look(1) - '0';
Richard Smith4344cbd2018-08-20 20:14:49 +00003150 First += 2;
3151 if (State) State->CtorDtorConversion = true;
Nico Weberba88b372019-04-03 23:14:33 +00003152 return make<CtorDtorName>(SoFar, /*IsDtor=*/true, Variant);
Richard Smith4344cbd2018-08-20 20:14:49 +00003153 }
3154
3155 return nullptr;
3156}
3157
3158// <nested-name> ::= N [<CV-Qualifiers>] [<ref-qualifier>] <prefix> <unqualified-name> E
3159// ::= N [<CV-Qualifiers>] [<ref-qualifier>] <template-prefix> <template-args> E
3160//
3161// <prefix> ::= <prefix> <unqualified-name>
3162// ::= <template-prefix> <template-args>
3163// ::= <template-param>
3164// ::= <decltype>
3165// ::= # empty
3166// ::= <substitution>
3167// ::= <prefix> <data-member-prefix>
3168// extension ::= L
3169//
3170// <data-member-prefix> := <member source-name> [<template-args>] M
3171//
3172// <template-prefix> ::= <prefix> <template unqualified-name>
3173// ::= <template-param>
3174// ::= <substitution>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003175template <typename Derived, typename Alloc>
3176Node *
3177AbstractManglingParser<Derived, Alloc>::parseNestedName(NameState *State) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003178 if (!consumeIf('N'))
3179 return nullptr;
3180
3181 Qualifiers CVTmp = parseCVQualifiers();
3182 if (State) State->CVQualifiers = CVTmp;
3183
3184 if (consumeIf('O')) {
3185 if (State) State->ReferenceQualifier = FrefQualRValue;
3186 } else if (consumeIf('R')) {
3187 if (State) State->ReferenceQualifier = FrefQualLValue;
3188 } else
3189 if (State) State->ReferenceQualifier = FrefQualNone;
3190
3191 Node *SoFar = nullptr;
3192 auto PushComponent = [&](Node *Comp) {
Richard Smith7918dea2018-08-24 23:30:26 +00003193 if (!Comp) return false;
Richard Smith4344cbd2018-08-20 20:14:49 +00003194 if (SoFar) SoFar = make<NestedName>(SoFar, Comp);
3195 else SoFar = Comp;
3196 if (State) State->EndsWithTemplateArgs = false;
Richard Smith7918dea2018-08-24 23:30:26 +00003197 return SoFar != nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003198 };
3199
Richard Smith7918dea2018-08-24 23:30:26 +00003200 if (consumeIf("St")) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003201 SoFar = make<NameType>("std");
Richard Smith7918dea2018-08-24 23:30:26 +00003202 if (!SoFar)
3203 return nullptr;
3204 }
Richard Smith4344cbd2018-08-20 20:14:49 +00003205
3206 while (!consumeIf('E')) {
3207 consumeIf('L'); // extension
3208
3209 // <data-member-prefix> := <member source-name> [<template-args>] M
3210 if (consumeIf('M')) {
3211 if (SoFar == nullptr)
3212 return nullptr;
3213 continue;
3214 }
3215
3216 // ::= <template-param>
3217 if (look() == 'T') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003218 if (!PushComponent(getDerived().parseTemplateParam()))
Richard Smith4344cbd2018-08-20 20:14:49 +00003219 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003220 Subs.push_back(SoFar);
3221 continue;
3222 }
3223
3224 // ::= <template-prefix> <template-args>
3225 if (look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003226 Node *TA = getDerived().parseTemplateArgs(State != nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00003227 if (TA == nullptr || SoFar == nullptr)
3228 return nullptr;
3229 SoFar = make<NameWithTemplateArgs>(SoFar, TA);
Richard Smith7918dea2018-08-24 23:30:26 +00003230 if (!SoFar)
3231 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003232 if (State) State->EndsWithTemplateArgs = true;
3233 Subs.push_back(SoFar);
3234 continue;
3235 }
3236
3237 // ::= <decltype>
3238 if (look() == 'D' && (look(1) == 't' || look(1) == 'T')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003239 if (!PushComponent(getDerived().parseDecltype()))
Richard Smith4344cbd2018-08-20 20:14:49 +00003240 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003241 Subs.push_back(SoFar);
3242 continue;
3243 }
3244
3245 // ::= <substitution>
3246 if (look() == 'S' && look(1) != 't') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003247 Node *S = getDerived().parseSubstitution();
Richard Smith7918dea2018-08-24 23:30:26 +00003248 if (!PushComponent(S))
Richard Smith4344cbd2018-08-20 20:14:49 +00003249 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003250 if (SoFar != S)
3251 Subs.push_back(S);
3252 continue;
3253 }
3254
3255 // Parse an <unqualified-name> thats actually a <ctor-dtor-name>.
3256 if (look() == 'C' || (look() == 'D' && look(1) != 'C')) {
3257 if (SoFar == nullptr)
3258 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003259 if (!PushComponent(getDerived().parseCtorDtorName(SoFar, State)))
Richard Smith4344cbd2018-08-20 20:14:49 +00003260 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003261 SoFar = getDerived().parseAbiTags(SoFar);
Richard Smith4344cbd2018-08-20 20:14:49 +00003262 if (SoFar == nullptr)
3263 return nullptr;
3264 Subs.push_back(SoFar);
3265 continue;
3266 }
3267
3268 // ::= <prefix> <unqualified-name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003269 if (!PushComponent(getDerived().parseUnqualifiedName(State)))
Richard Smith4344cbd2018-08-20 20:14:49 +00003270 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003271 Subs.push_back(SoFar);
3272 }
3273
3274 if (SoFar == nullptr || Subs.empty())
3275 return nullptr;
3276
3277 Subs.pop_back();
3278 return SoFar;
3279}
3280
3281// <simple-id> ::= <source-name> [ <template-args> ]
Pavel Labathb26cabc2018-10-16 14:29:14 +00003282template <typename Derived, typename Alloc>
3283Node *AbstractManglingParser<Derived, Alloc>::parseSimpleId() {
3284 Node *SN = getDerived().parseSourceName(/*NameState=*/nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00003285 if (SN == nullptr)
3286 return nullptr;
3287 if (look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003288 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00003289 if (TA == nullptr)
3290 return nullptr;
3291 return make<NameWithTemplateArgs>(SN, TA);
3292 }
3293 return SN;
3294}
3295
3296// <destructor-name> ::= <unresolved-type> # e.g., ~T or ~decltype(f())
3297// ::= <simple-id> # e.g., ~A<2*N>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003298template <typename Derived, typename Alloc>
3299Node *AbstractManglingParser<Derived, Alloc>::parseDestructorName() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003300 Node *Result;
3301 if (std::isdigit(look()))
Pavel Labathb26cabc2018-10-16 14:29:14 +00003302 Result = getDerived().parseSimpleId();
Richard Smith4344cbd2018-08-20 20:14:49 +00003303 else
Pavel Labathb26cabc2018-10-16 14:29:14 +00003304 Result = getDerived().parseUnresolvedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003305 if (Result == nullptr)
3306 return nullptr;
3307 return make<DtorName>(Result);
3308}
3309
3310// <unresolved-type> ::= <template-param>
3311// ::= <decltype>
3312// ::= <substitution>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003313template <typename Derived, typename Alloc>
3314Node *AbstractManglingParser<Derived, Alloc>::parseUnresolvedType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003315 if (look() == 'T') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003316 Node *TP = getDerived().parseTemplateParam();
Richard Smith4344cbd2018-08-20 20:14:49 +00003317 if (TP == nullptr)
3318 return nullptr;
3319 Subs.push_back(TP);
3320 return TP;
3321 }
3322 if (look() == 'D') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003323 Node *DT = getDerived().parseDecltype();
Richard Smith4344cbd2018-08-20 20:14:49 +00003324 if (DT == nullptr)
3325 return nullptr;
3326 Subs.push_back(DT);
3327 return DT;
3328 }
Pavel Labathb26cabc2018-10-16 14:29:14 +00003329 return getDerived().parseSubstitution();
Richard Smith4344cbd2018-08-20 20:14:49 +00003330}
3331
3332// <base-unresolved-name> ::= <simple-id> # unresolved name
3333// extension ::= <operator-name> # unresolved operator-function-id
3334// extension ::= <operator-name> <template-args> # unresolved operator template-id
3335// ::= on <operator-name> # unresolved operator-function-id
3336// ::= on <operator-name> <template-args> # unresolved operator template-id
3337// ::= dn <destructor-name> # destructor or pseudo-destructor;
3338// # e.g. ~X or ~X<N-1>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003339template <typename Derived, typename Alloc>
3340Node *AbstractManglingParser<Derived, Alloc>::parseBaseUnresolvedName() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003341 if (std::isdigit(look()))
Pavel Labathb26cabc2018-10-16 14:29:14 +00003342 return getDerived().parseSimpleId();
Richard Smith4344cbd2018-08-20 20:14:49 +00003343
3344 if (consumeIf("dn"))
Pavel Labathb26cabc2018-10-16 14:29:14 +00003345 return getDerived().parseDestructorName();
Richard Smith4344cbd2018-08-20 20:14:49 +00003346
3347 consumeIf("on");
3348
Pavel Labathb26cabc2018-10-16 14:29:14 +00003349 Node *Oper = getDerived().parseOperatorName(/*NameState=*/nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00003350 if (Oper == nullptr)
3351 return nullptr;
3352 if (look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003353 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00003354 if (TA == nullptr)
3355 return nullptr;
3356 return make<NameWithTemplateArgs>(Oper, TA);
3357 }
3358 return Oper;
3359}
3360
3361// <unresolved-name>
3362// extension ::= srN <unresolved-type> [<template-args>] <unresolved-qualifier-level>* E <base-unresolved-name>
3363// ::= [gs] <base-unresolved-name> # x or (with "gs") ::x
3364// ::= [gs] sr <unresolved-qualifier-level>+ E <base-unresolved-name>
3365// # A::x, N::y, A<T>::z; "gs" means leading "::"
3366// ::= sr <unresolved-type> <base-unresolved-name> # T::x / decltype(p)::x
3367// extension ::= sr <unresolved-type> <template-args> <base-unresolved-name>
3368// # T::N::x /decltype(p)::N::x
3369// (ignored) ::= srN <unresolved-type> <unresolved-qualifier-level>+ E <base-unresolved-name>
3370//
3371// <unresolved-qualifier-level> ::= <simple-id>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003372template <typename Derived, typename Alloc>
3373Node *AbstractManglingParser<Derived, Alloc>::parseUnresolvedName() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003374 Node *SoFar = nullptr;
3375
3376 // srN <unresolved-type> [<template-args>] <unresolved-qualifier-level>* E <base-unresolved-name>
3377 // srN <unresolved-type> <unresolved-qualifier-level>+ E <base-unresolved-name>
3378 if (consumeIf("srN")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003379 SoFar = getDerived().parseUnresolvedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003380 if (SoFar == nullptr)
3381 return nullptr;
3382
3383 if (look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003384 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00003385 if (TA == nullptr)
3386 return nullptr;
3387 SoFar = make<NameWithTemplateArgs>(SoFar, TA);
Richard Smith7918dea2018-08-24 23:30:26 +00003388 if (!SoFar)
3389 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003390 }
3391
3392 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003393 Node *Qual = getDerived().parseSimpleId();
Richard Smith4344cbd2018-08-20 20:14:49 +00003394 if (Qual == nullptr)
3395 return nullptr;
3396 SoFar = make<QualifiedName>(SoFar, Qual);
Richard Smith7918dea2018-08-24 23:30:26 +00003397 if (!SoFar)
3398 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003399 }
3400
Pavel Labathb26cabc2018-10-16 14:29:14 +00003401 Node *Base = getDerived().parseBaseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00003402 if (Base == nullptr)
3403 return nullptr;
3404 return make<QualifiedName>(SoFar, Base);
3405 }
3406
3407 bool Global = consumeIf("gs");
3408
3409 // [gs] <base-unresolved-name> # x or (with "gs") ::x
3410 if (!consumeIf("sr")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003411 SoFar = getDerived().parseBaseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00003412 if (SoFar == nullptr)
3413 return nullptr;
3414 if (Global)
3415 SoFar = make<GlobalQualifiedName>(SoFar);
3416 return SoFar;
3417 }
3418
3419 // [gs] sr <unresolved-qualifier-level>+ E <base-unresolved-name>
3420 if (std::isdigit(look())) {
3421 do {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003422 Node *Qual = getDerived().parseSimpleId();
Richard Smith4344cbd2018-08-20 20:14:49 +00003423 if (Qual == nullptr)
3424 return nullptr;
3425 if (SoFar)
3426 SoFar = make<QualifiedName>(SoFar, Qual);
3427 else if (Global)
3428 SoFar = make<GlobalQualifiedName>(Qual);
3429 else
3430 SoFar = Qual;
Richard Smith7918dea2018-08-24 23:30:26 +00003431 if (!SoFar)
3432 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003433 } while (!consumeIf('E'));
3434 }
3435 // sr <unresolved-type> <base-unresolved-name>
3436 // sr <unresolved-type> <template-args> <base-unresolved-name>
3437 else {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003438 SoFar = getDerived().parseUnresolvedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003439 if (SoFar == nullptr)
3440 return nullptr;
3441
3442 if (look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003443 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00003444 if (TA == nullptr)
3445 return nullptr;
3446 SoFar = make<NameWithTemplateArgs>(SoFar, TA);
Richard Smith7918dea2018-08-24 23:30:26 +00003447 if (!SoFar)
3448 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003449 }
3450 }
3451
3452 assert(SoFar != nullptr);
3453
Pavel Labathb26cabc2018-10-16 14:29:14 +00003454 Node *Base = getDerived().parseBaseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00003455 if (Base == nullptr)
3456 return nullptr;
3457 return make<QualifiedName>(SoFar, Base);
3458}
3459
3460// <abi-tags> ::= <abi-tag> [<abi-tags>]
3461// <abi-tag> ::= B <source-name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003462template <typename Derived, typename Alloc>
3463Node *AbstractManglingParser<Derived, Alloc>::parseAbiTags(Node *N) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003464 while (consumeIf('B')) {
3465 StringView SN = parseBareSourceName();
3466 if (SN.empty())
3467 return nullptr;
3468 N = make<AbiTagAttr>(N, SN);
Richard Smith7918dea2018-08-24 23:30:26 +00003469 if (!N)
3470 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003471 }
3472 return N;
3473}
3474
3475// <number> ::= [n] <non-negative decimal integer>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003476template <typename Alloc, typename Derived>
3477StringView
3478AbstractManglingParser<Alloc, Derived>::parseNumber(bool AllowNegative) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003479 const char *Tmp = First;
3480 if (AllowNegative)
3481 consumeIf('n');
3482 if (numLeft() == 0 || !std::isdigit(*First))
3483 return StringView();
3484 while (numLeft() != 0 && std::isdigit(*First))
3485 ++First;
3486 return StringView(Tmp, First);
3487}
3488
3489// <positive length number> ::= [0-9]*
Pavel Labathb26cabc2018-10-16 14:29:14 +00003490template <typename Alloc, typename Derived>
3491bool AbstractManglingParser<Alloc, Derived>::parsePositiveInteger(size_t *Out) {
Richard Smith4344cbd2018-08-20 20:14:49 +00003492 *Out = 0;
3493 if (look() < '0' || look() > '9')
3494 return true;
3495 while (look() >= '0' && look() <= '9') {
3496 *Out *= 10;
3497 *Out += static_cast<size_t>(consume() - '0');
3498 }
3499 return false;
3500}
3501
Pavel Labathb26cabc2018-10-16 14:29:14 +00003502template <typename Alloc, typename Derived>
3503StringView AbstractManglingParser<Alloc, Derived>::parseBareSourceName() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003504 size_t Int = 0;
3505 if (parsePositiveInteger(&Int) || numLeft() < Int)
3506 return StringView();
3507 StringView R(First, First + Int);
3508 First += Int;
3509 return R;
3510}
3511
3512// <function-type> ::= [<CV-qualifiers>] [<exception-spec>] [Dx] F [Y] <bare-function-type> [<ref-qualifier>] E
3513//
3514// <exception-spec> ::= Do # non-throwing exception-specification (e.g., noexcept, throw())
3515// ::= DO <expression> E # computed (instantiation-dependent) noexcept
3516// ::= Dw <type>+ E # dynamic exception specification with instantiation-dependent types
3517//
3518// <ref-qualifier> ::= R # & ref-qualifier
3519// <ref-qualifier> ::= O # && ref-qualifier
Pavel Labathb26cabc2018-10-16 14:29:14 +00003520template <typename Derived, typename Alloc>
3521Node *AbstractManglingParser<Derived, Alloc>::parseFunctionType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003522 Qualifiers CVQuals = parseCVQualifiers();
3523
3524 Node *ExceptionSpec = nullptr;
3525 if (consumeIf("Do")) {
3526 ExceptionSpec = make<NameType>("noexcept");
Richard Smith7918dea2018-08-24 23:30:26 +00003527 if (!ExceptionSpec)
3528 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003529 } else if (consumeIf("DO")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003530 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00003531 if (E == nullptr || !consumeIf('E'))
3532 return nullptr;
3533 ExceptionSpec = make<NoexceptSpec>(E);
Richard Smith7918dea2018-08-24 23:30:26 +00003534 if (!ExceptionSpec)
3535 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003536 } else if (consumeIf("Dw")) {
3537 size_t SpecsBegin = Names.size();
3538 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003539 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003540 if (T == nullptr)
3541 return nullptr;
3542 Names.push_back(T);
3543 }
3544 ExceptionSpec =
3545 make<DynamicExceptionSpec>(popTrailingNodeArray(SpecsBegin));
Richard Smith7918dea2018-08-24 23:30:26 +00003546 if (!ExceptionSpec)
3547 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003548 }
3549
3550 consumeIf("Dx"); // transaction safe
3551
3552 if (!consumeIf('F'))
3553 return nullptr;
3554 consumeIf('Y'); // extern "C"
Pavel Labathb26cabc2018-10-16 14:29:14 +00003555 Node *ReturnType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003556 if (ReturnType == nullptr)
3557 return nullptr;
3558
3559 FunctionRefQual ReferenceQualifier = FrefQualNone;
3560 size_t ParamsBegin = Names.size();
3561 while (true) {
3562 if (consumeIf('E'))
3563 break;
3564 if (consumeIf('v'))
3565 continue;
3566 if (consumeIf("RE")) {
3567 ReferenceQualifier = FrefQualLValue;
3568 break;
3569 }
3570 if (consumeIf("OE")) {
3571 ReferenceQualifier = FrefQualRValue;
3572 break;
3573 }
Pavel Labathb26cabc2018-10-16 14:29:14 +00003574 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003575 if (T == nullptr)
3576 return nullptr;
3577 Names.push_back(T);
3578 }
3579
3580 NodeArray Params = popTrailingNodeArray(ParamsBegin);
3581 return make<FunctionType>(ReturnType, Params, CVQuals,
3582 ReferenceQualifier, ExceptionSpec);
3583}
3584
3585// extension:
3586// <vector-type> ::= Dv <positive dimension number> _ <extended element type>
3587// ::= Dv [<dimension expression>] _ <element type>
3588// <extended element type> ::= <element type>
3589// ::= p # AltiVec vector pixel
Pavel Labathb26cabc2018-10-16 14:29:14 +00003590template <typename Derived, typename Alloc>
3591Node *AbstractManglingParser<Derived, Alloc>::parseVectorType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003592 if (!consumeIf("Dv"))
3593 return nullptr;
3594 if (look() >= '1' && look() <= '9') {
Erik Pilkington87529ae2019-11-04 10:47:44 -08003595 Node *DimensionNumber = make<NameType>(parseNumber());
3596 if (!DimensionNumber)
3597 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003598 if (!consumeIf('_'))
3599 return nullptr;
3600 if (consumeIf('p'))
3601 return make<PixelVectorType>(DimensionNumber);
Pavel Labathb26cabc2018-10-16 14:29:14 +00003602 Node *ElemType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003603 if (ElemType == nullptr)
3604 return nullptr;
3605 return make<VectorType>(ElemType, DimensionNumber);
3606 }
3607
3608 if (!consumeIf('_')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003609 Node *DimExpr = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00003610 if (!DimExpr)
3611 return nullptr;
3612 if (!consumeIf('_'))
3613 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003614 Node *ElemType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003615 if (!ElemType)
3616 return nullptr;
3617 return make<VectorType>(ElemType, DimExpr);
3618 }
Pavel Labathb26cabc2018-10-16 14:29:14 +00003619 Node *ElemType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003620 if (!ElemType)
3621 return nullptr;
Erik Pilkington87529ae2019-11-04 10:47:44 -08003622 return make<VectorType>(ElemType, /*Dimension=*/nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00003623}
3624
3625// <decltype> ::= Dt <expression> E # decltype of an id-expression or class member access (C++0x)
3626// ::= DT <expression> E # decltype of an expression (C++0x)
Pavel Labathb26cabc2018-10-16 14:29:14 +00003627template <typename Derived, typename Alloc>
3628Node *AbstractManglingParser<Derived, Alloc>::parseDecltype() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003629 if (!consumeIf('D'))
3630 return nullptr;
3631 if (!consumeIf('t') && !consumeIf('T'))
3632 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003633 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00003634 if (E == nullptr)
3635 return nullptr;
3636 if (!consumeIf('E'))
3637 return nullptr;
3638 return make<EnclosingExpr>("decltype(", E, ")");
3639}
3640
3641// <array-type> ::= A <positive dimension number> _ <element type>
3642// ::= A [<dimension expression>] _ <element type>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003643template <typename Derived, typename Alloc>
3644Node *AbstractManglingParser<Derived, Alloc>::parseArrayType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003645 if (!consumeIf('A'))
3646 return nullptr;
3647
Erik Pilkington87529ae2019-11-04 10:47:44 -08003648 Node *Dimension = nullptr;
Pavel Labath36050092018-10-10 08:39:16 +00003649
Richard Smith4344cbd2018-08-20 20:14:49 +00003650 if (std::isdigit(look())) {
Erik Pilkington87529ae2019-11-04 10:47:44 -08003651 Dimension = make<NameType>(parseNumber());
3652 if (!Dimension)
3653 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003654 if (!consumeIf('_'))
3655 return nullptr;
Pavel Labath36050092018-10-10 08:39:16 +00003656 } else if (!consumeIf('_')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003657 Node *DimExpr = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00003658 if (DimExpr == nullptr)
3659 return nullptr;
3660 if (!consumeIf('_'))
3661 return nullptr;
Pavel Labath36050092018-10-10 08:39:16 +00003662 Dimension = DimExpr;
Richard Smith4344cbd2018-08-20 20:14:49 +00003663 }
3664
Pavel Labathb26cabc2018-10-16 14:29:14 +00003665 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003666 if (Ty == nullptr)
3667 return nullptr;
Pavel Labath36050092018-10-10 08:39:16 +00003668 return make<ArrayType>(Ty, Dimension);
Richard Smith4344cbd2018-08-20 20:14:49 +00003669}
3670
3671// <pointer-to-member-type> ::= M <class type> <member type>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003672template <typename Derived, typename Alloc>
3673Node *AbstractManglingParser<Derived, Alloc>::parsePointerToMemberType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003674 if (!consumeIf('M'))
3675 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003676 Node *ClassType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003677 if (ClassType == nullptr)
3678 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003679 Node *MemberType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003680 if (MemberType == nullptr)
3681 return nullptr;
3682 return make<PointerToMemberType>(ClassType, MemberType);
3683}
3684
3685// <class-enum-type> ::= <name> # non-dependent type name, dependent type name, or dependent typename-specifier
3686// ::= Ts <name> # dependent elaborated type specifier using 'struct' or 'class'
3687// ::= Tu <name> # dependent elaborated type specifier using 'union'
3688// ::= Te <name> # dependent elaborated type specifier using 'enum'
Pavel Labathb26cabc2018-10-16 14:29:14 +00003689template <typename Derived, typename Alloc>
3690Node *AbstractManglingParser<Derived, Alloc>::parseClassEnumType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003691 StringView ElabSpef;
3692 if (consumeIf("Ts"))
3693 ElabSpef = "struct";
3694 else if (consumeIf("Tu"))
3695 ElabSpef = "union";
3696 else if (consumeIf("Te"))
3697 ElabSpef = "enum";
3698
Pavel Labathb26cabc2018-10-16 14:29:14 +00003699 Node *Name = getDerived().parseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00003700 if (Name == nullptr)
3701 return nullptr;
3702
3703 if (!ElabSpef.empty())
3704 return make<ElaboratedTypeSpefType>(ElabSpef, Name);
3705
3706 return Name;
3707}
3708
3709// <qualified-type> ::= <qualifiers> <type>
3710// <qualifiers> ::= <extended-qualifier>* <CV-qualifiers>
3711// <extended-qualifier> ::= U <source-name> [<template-args>] # vendor extended type qualifier
Pavel Labathb26cabc2018-10-16 14:29:14 +00003712template <typename Derived, typename Alloc>
3713Node *AbstractManglingParser<Derived, Alloc>::parseQualifiedType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003714 if (consumeIf('U')) {
3715 StringView Qual = parseBareSourceName();
3716 if (Qual.empty())
3717 return nullptr;
3718
Richard Smith4344cbd2018-08-20 20:14:49 +00003719 // extension ::= U <objc-name> <objc-type> # objc-type<identifier>
3720 if (Qual.startsWith("objcproto")) {
3721 StringView ProtoSourceName = Qual.dropFront(std::strlen("objcproto"));
3722 StringView Proto;
3723 {
3724 SwapAndRestore<const char *> SaveFirst(First, ProtoSourceName.begin()),
3725 SaveLast(Last, ProtoSourceName.end());
3726 Proto = parseBareSourceName();
3727 }
3728 if (Proto.empty())
3729 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003730 Node *Child = getDerived().parseQualifiedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003731 if (Child == nullptr)
3732 return nullptr;
3733 return make<ObjCProtoName>(Child, Proto);
3734 }
3735
Alex Orlov1540db32021-03-24 10:21:32 +04003736 Node *TA = nullptr;
3737 if (look() == 'I') {
3738 TA = getDerived().parseTemplateArgs();
3739 if (TA == nullptr)
3740 return nullptr;
3741 }
3742
Pavel Labathb26cabc2018-10-16 14:29:14 +00003743 Node *Child = getDerived().parseQualifiedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003744 if (Child == nullptr)
3745 return nullptr;
Alex Orlov1540db32021-03-24 10:21:32 +04003746 return make<VendorExtQualType>(Child, Qual, TA);
Richard Smith4344cbd2018-08-20 20:14:49 +00003747 }
3748
3749 Qualifiers Quals = parseCVQualifiers();
Pavel Labathb26cabc2018-10-16 14:29:14 +00003750 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003751 if (Ty == nullptr)
3752 return nullptr;
3753 if (Quals != QualNone)
3754 Ty = make<QualType>(Ty, Quals);
3755 return Ty;
3756}
3757
3758// <type> ::= <builtin-type>
3759// ::= <qualified-type>
3760// ::= <function-type>
3761// ::= <class-enum-type>
3762// ::= <array-type>
3763// ::= <pointer-to-member-type>
3764// ::= <template-param>
3765// ::= <template-template-param> <template-args>
3766// ::= <decltype>
3767// ::= P <type> # pointer
3768// ::= R <type> # l-value reference
3769// ::= O <type> # r-value reference (C++11)
3770// ::= C <type> # complex pair (C99)
3771// ::= G <type> # imaginary (C99)
3772// ::= <substitution> # See Compression below
3773// extension ::= U <objc-name> <objc-type> # objc-type<identifier>
3774// extension ::= <vector-type> # <vector-type> starts with Dv
3775//
3776// <objc-name> ::= <k0 number> objcproto <k1 number> <identifier> # k0 = 9 + <number of digits in k1> + k1
3777// <objc-type> ::= <source-name> # PU<11+>objcproto 11objc_object<source-name> 11objc_object -> id<source-name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00003778template <typename Derived, typename Alloc>
3779Node *AbstractManglingParser<Derived, Alloc>::parseType() {
Richard Smith4344cbd2018-08-20 20:14:49 +00003780 Node *Result = nullptr;
3781
Richard Smith4344cbd2018-08-20 20:14:49 +00003782 switch (look()) {
3783 // ::= <qualified-type>
3784 case 'r':
3785 case 'V':
3786 case 'K': {
3787 unsigned AfterQuals = 0;
3788 if (look(AfterQuals) == 'r') ++AfterQuals;
3789 if (look(AfterQuals) == 'V') ++AfterQuals;
3790 if (look(AfterQuals) == 'K') ++AfterQuals;
3791
3792 if (look(AfterQuals) == 'F' ||
3793 (look(AfterQuals) == 'D' &&
3794 (look(AfterQuals + 1) == 'o' || look(AfterQuals + 1) == 'O' ||
3795 look(AfterQuals + 1) == 'w' || look(AfterQuals + 1) == 'x'))) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003796 Result = getDerived().parseFunctionType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003797 break;
3798 }
Erik Pilkington870950f2019-01-17 20:37:51 +00003799 DEMANGLE_FALLTHROUGH;
Richard Smith4344cbd2018-08-20 20:14:49 +00003800 }
3801 case 'U': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003802 Result = getDerived().parseQualifiedType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003803 break;
3804 }
3805 // <builtin-type> ::= v # void
3806 case 'v':
3807 ++First;
3808 return make<NameType>("void");
3809 // ::= w # wchar_t
3810 case 'w':
3811 ++First;
3812 return make<NameType>("wchar_t");
3813 // ::= b # bool
3814 case 'b':
3815 ++First;
3816 return make<NameType>("bool");
3817 // ::= c # char
3818 case 'c':
3819 ++First;
3820 return make<NameType>("char");
3821 // ::= a # signed char
3822 case 'a':
3823 ++First;
3824 return make<NameType>("signed char");
3825 // ::= h # unsigned char
3826 case 'h':
3827 ++First;
3828 return make<NameType>("unsigned char");
3829 // ::= s # short
3830 case 's':
3831 ++First;
3832 return make<NameType>("short");
3833 // ::= t # unsigned short
3834 case 't':
3835 ++First;
3836 return make<NameType>("unsigned short");
3837 // ::= i # int
3838 case 'i':
3839 ++First;
3840 return make<NameType>("int");
3841 // ::= j # unsigned int
3842 case 'j':
3843 ++First;
3844 return make<NameType>("unsigned int");
3845 // ::= l # long
3846 case 'l':
3847 ++First;
3848 return make<NameType>("long");
3849 // ::= m # unsigned long
3850 case 'm':
3851 ++First;
3852 return make<NameType>("unsigned long");
3853 // ::= x # long long, __int64
3854 case 'x':
3855 ++First;
3856 return make<NameType>("long long");
3857 // ::= y # unsigned long long, __int64
3858 case 'y':
3859 ++First;
3860 return make<NameType>("unsigned long long");
3861 // ::= n # __int128
3862 case 'n':
3863 ++First;
3864 return make<NameType>("__int128");
3865 // ::= o # unsigned __int128
3866 case 'o':
3867 ++First;
3868 return make<NameType>("unsigned __int128");
3869 // ::= f # float
3870 case 'f':
3871 ++First;
3872 return make<NameType>("float");
3873 // ::= d # double
3874 case 'd':
3875 ++First;
3876 return make<NameType>("double");
3877 // ::= e # long double, __float80
3878 case 'e':
3879 ++First;
3880 return make<NameType>("long double");
3881 // ::= g # __float128
3882 case 'g':
3883 ++First;
3884 return make<NameType>("__float128");
3885 // ::= z # ellipsis
3886 case 'z':
3887 ++First;
3888 return make<NameType>("...");
3889
3890 // <builtin-type> ::= u <source-name> # vendor extended type
3891 case 'u': {
3892 ++First;
3893 StringView Res = parseBareSourceName();
3894 if (Res.empty())
3895 return nullptr;
Erik Pilkingtona11a6222019-06-10 21:02:39 +00003896 // Typically, <builtin-type>s are not considered substitution candidates,
3897 // but the exception to that exception is vendor extended types (Itanium C++
3898 // ABI 5.9.1).
3899 Result = make<NameType>(Res);
3900 break;
Richard Smith4344cbd2018-08-20 20:14:49 +00003901 }
3902 case 'D':
3903 switch (look(1)) {
3904 // ::= Dd # IEEE 754r decimal floating point (64 bits)
3905 case 'd':
3906 First += 2;
3907 return make<NameType>("decimal64");
3908 // ::= De # IEEE 754r decimal floating point (128 bits)
3909 case 'e':
3910 First += 2;
3911 return make<NameType>("decimal128");
3912 // ::= Df # IEEE 754r decimal floating point (32 bits)
3913 case 'f':
3914 First += 2;
3915 return make<NameType>("decimal32");
3916 // ::= Dh # IEEE 754r half-precision floating point (16 bits)
3917 case 'h':
3918 First += 2;
Stuart Brady19025d12021-06-07 16:30:22 +01003919 return make<NameType>("half");
Pengfei Wang036d3152021-09-23 11:02:25 +08003920 // ::= DF <number> _ # ISO/IEC TS 18661 binary floating point (N bits)
3921 case 'F': {
3922 First += 2;
3923 Node *DimensionNumber = make<NameType>(parseNumber());
3924 if (!DimensionNumber)
3925 return nullptr;
3926 if (!consumeIf('_'))
3927 return nullptr;
3928 return make<BinaryFPType>(DimensionNumber);
3929 }
Richard Smith4344cbd2018-08-20 20:14:49 +00003930 // ::= Di # char32_t
3931 case 'i':
3932 First += 2;
3933 return make<NameType>("char32_t");
3934 // ::= Ds # char16_t
3935 case 's':
3936 First += 2;
3937 return make<NameType>("char16_t");
Erik Pilkingtoncabc7f02019-06-28 19:54:19 +00003938 // ::= Du # char8_t (C++2a, not yet in the Itanium spec)
3939 case 'u':
3940 First += 2;
3941 return make<NameType>("char8_t");
Richard Smith4344cbd2018-08-20 20:14:49 +00003942 // ::= Da # auto (in dependent new-expressions)
3943 case 'a':
3944 First += 2;
3945 return make<NameType>("auto");
3946 // ::= Dc # decltype(auto)
3947 case 'c':
3948 First += 2;
3949 return make<NameType>("decltype(auto)");
3950 // ::= Dn # std::nullptr_t (i.e., decltype(nullptr))
3951 case 'n':
3952 First += 2;
3953 return make<NameType>("std::nullptr_t");
3954
3955 // ::= <decltype>
3956 case 't':
3957 case 'T': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003958 Result = getDerived().parseDecltype();
Richard Smith4344cbd2018-08-20 20:14:49 +00003959 break;
3960 }
3961 // extension ::= <vector-type> # <vector-type> starts with Dv
3962 case 'v': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003963 Result = getDerived().parseVectorType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003964 break;
3965 }
3966 // ::= Dp <type> # pack expansion (C++0x)
3967 case 'p': {
3968 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00003969 Node *Child = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003970 if (!Child)
3971 return nullptr;
3972 Result = make<ParameterPackExpansion>(Child);
3973 break;
3974 }
3975 // Exception specifier on a function type.
3976 case 'o':
3977 case 'O':
3978 case 'w':
3979 // Transaction safe function type.
3980 case 'x':
Pavel Labathb26cabc2018-10-16 14:29:14 +00003981 Result = getDerived().parseFunctionType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003982 break;
3983 }
3984 break;
3985 // ::= <function-type>
3986 case 'F': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003987 Result = getDerived().parseFunctionType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003988 break;
3989 }
3990 // ::= <array-type>
3991 case 'A': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003992 Result = getDerived().parseArrayType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003993 break;
3994 }
3995 // ::= <pointer-to-member-type>
3996 case 'M': {
Pavel Labathb26cabc2018-10-16 14:29:14 +00003997 Result = getDerived().parsePointerToMemberType();
Richard Smith4344cbd2018-08-20 20:14:49 +00003998 break;
3999 }
4000 // ::= <template-param>
4001 case 'T': {
4002 // This could be an elaborate type specifier on a <class-enum-type>.
4003 if (look(1) == 's' || look(1) == 'u' || look(1) == 'e') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004004 Result = getDerived().parseClassEnumType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004005 break;
4006 }
4007
Pavel Labathb26cabc2018-10-16 14:29:14 +00004008 Result = getDerived().parseTemplateParam();
Richard Smith4344cbd2018-08-20 20:14:49 +00004009 if (Result == nullptr)
4010 return nullptr;
4011
4012 // Result could be either of:
4013 // <type> ::= <template-param>
4014 // <type> ::= <template-template-param> <template-args>
4015 //
4016 // <template-template-param> ::= <template-param>
4017 // ::= <substitution>
4018 //
4019 // If this is followed by some <template-args>, and we're permitted to
4020 // parse them, take the second production.
4021
4022 if (TryToParseTemplateArgs && look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004023 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00004024 if (TA == nullptr)
4025 return nullptr;
4026 Result = make<NameWithTemplateArgs>(Result, TA);
4027 }
4028 break;
4029 }
4030 // ::= P <type> # pointer
4031 case 'P': {
4032 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004033 Node *Ptr = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004034 if (Ptr == nullptr)
4035 return nullptr;
4036 Result = make<PointerType>(Ptr);
4037 break;
4038 }
4039 // ::= R <type> # l-value reference
4040 case 'R': {
4041 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004042 Node *Ref = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004043 if (Ref == nullptr)
4044 return nullptr;
4045 Result = make<ReferenceType>(Ref, ReferenceKind::LValue);
4046 break;
4047 }
4048 // ::= O <type> # r-value reference (C++11)
4049 case 'O': {
4050 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004051 Node *Ref = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004052 if (Ref == nullptr)
4053 return nullptr;
4054 Result = make<ReferenceType>(Ref, ReferenceKind::RValue);
4055 break;
4056 }
4057 // ::= C <type> # complex pair (C99)
4058 case 'C': {
4059 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004060 Node *P = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004061 if (P == nullptr)
4062 return nullptr;
4063 Result = make<PostfixQualifiedType>(P, " complex");
4064 break;
4065 }
4066 // ::= G <type> # imaginary (C99)
4067 case 'G': {
4068 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004069 Node *P = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004070 if (P == nullptr)
4071 return P;
4072 Result = make<PostfixQualifiedType>(P, " imaginary");
4073 break;
4074 }
4075 // ::= <substitution> # See Compression below
4076 case 'S': {
Nathan Sidwellbbc24432022-01-24 04:28:09 -08004077 if (look(1) != 't') {
4078 Result = getDerived().parseSubstitution();
4079 if (Result == nullptr)
Richard Smith4344cbd2018-08-20 20:14:49 +00004080 return nullptr;
4081
4082 // Sub could be either of:
4083 // <type> ::= <substitution>
4084 // <type> ::= <template-template-param> <template-args>
4085 //
4086 // <template-template-param> ::= <template-param>
4087 // ::= <substitution>
4088 //
4089 // If this is followed by some <template-args>, and we're permitted to
4090 // parse them, take the second production.
4091
4092 if (TryToParseTemplateArgs && look() == 'I') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004093 Node *TA = getDerived().parseTemplateArgs();
Richard Smith4344cbd2018-08-20 20:14:49 +00004094 if (TA == nullptr)
4095 return nullptr;
Nathan Sidwellbbc24432022-01-24 04:28:09 -08004096 Result = make<NameWithTemplateArgs>(Result, TA);
4097 } else {
4098 // If all we parsed was a substitution, don't re-insert into the
4099 // substitution table.
4100 return Result;
Richard Smith4344cbd2018-08-20 20:14:49 +00004101 }
Nathan Sidwellbbc24432022-01-24 04:28:09 -08004102 break;
Richard Smith4344cbd2018-08-20 20:14:49 +00004103 }
Erik Pilkington870950f2019-01-17 20:37:51 +00004104 DEMANGLE_FALLTHROUGH;
Richard Smith4344cbd2018-08-20 20:14:49 +00004105 }
4106 // ::= <class-enum-type>
4107 default: {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004108 Result = getDerived().parseClassEnumType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004109 break;
4110 }
4111 }
4112
4113 // If we parsed a type, insert it into the substitution table. Note that all
4114 // <builtin-type>s and <substitution>s have already bailed out, because they
4115 // don't get substitutions.
4116 if (Result != nullptr)
4117 Subs.push_back(Result);
4118 return Result;
4119}
4120
Pavel Labathb26cabc2018-10-16 14:29:14 +00004121template <typename Derived, typename Alloc>
4122Node *AbstractManglingParser<Derived, Alloc>::parsePrefixExpr(StringView Kind) {
4123 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004124 if (E == nullptr)
4125 return nullptr;
4126 return make<PrefixExpr>(Kind, E);
4127}
4128
Pavel Labathb26cabc2018-10-16 14:29:14 +00004129template <typename Derived, typename Alloc>
4130Node *AbstractManglingParser<Derived, Alloc>::parseBinaryExpr(StringView Kind) {
4131 Node *LHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004132 if (LHS == nullptr)
4133 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004134 Node *RHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004135 if (RHS == nullptr)
4136 return nullptr;
4137 return make<BinaryExpr>(LHS, Kind, RHS);
4138}
4139
Pavel Labathb26cabc2018-10-16 14:29:14 +00004140template <typename Derived, typename Alloc>
4141Node *
4142AbstractManglingParser<Derived, Alloc>::parseIntegerLiteral(StringView Lit) {
Richard Smith4344cbd2018-08-20 20:14:49 +00004143 StringView Tmp = parseNumber(true);
4144 if (!Tmp.empty() && consumeIf('E'))
4145 return make<IntegerLiteral>(Lit, Tmp);
4146 return nullptr;
4147}
4148
4149// <CV-Qualifiers> ::= [r] [V] [K]
Pavel Labathb26cabc2018-10-16 14:29:14 +00004150template <typename Alloc, typename Derived>
4151Qualifiers AbstractManglingParser<Alloc, Derived>::parseCVQualifiers() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004152 Qualifiers CVR = QualNone;
4153 if (consumeIf('r'))
4154 CVR |= QualRestrict;
4155 if (consumeIf('V'))
4156 CVR |= QualVolatile;
4157 if (consumeIf('K'))
4158 CVR |= QualConst;
4159 return CVR;
4160}
4161
4162// <function-param> ::= fp <top-level CV-Qualifiers> _ # L == 0, first parameter
4163// ::= fp <top-level CV-Qualifiers> <parameter-2 non-negative number> _ # L == 0, second and later parameters
4164// ::= fL <L-1 non-negative number> p <top-level CV-Qualifiers> _ # L > 0, first parameter
4165// ::= fL <L-1 non-negative number> p <top-level CV-Qualifiers> <parameter-2 non-negative number> _ # L > 0, second and later parameters
Erik Pilkingtondc9d5ff2020-05-13 22:19:45 -04004166// ::= fpT # 'this' expression (not part of standard?)
Pavel Labathb26cabc2018-10-16 14:29:14 +00004167template <typename Derived, typename Alloc>
4168Node *AbstractManglingParser<Derived, Alloc>::parseFunctionParam() {
Erik Pilkingtondc9d5ff2020-05-13 22:19:45 -04004169 if (consumeIf("fpT"))
4170 return make<NameType>("this");
Richard Smith4344cbd2018-08-20 20:14:49 +00004171 if (consumeIf("fp")) {
4172 parseCVQualifiers();
4173 StringView Num = parseNumber();
4174 if (!consumeIf('_'))
4175 return nullptr;
4176 return make<FunctionParam>(Num);
4177 }
4178 if (consumeIf("fL")) {
4179 if (parseNumber().empty())
4180 return nullptr;
4181 if (!consumeIf('p'))
4182 return nullptr;
4183 parseCVQualifiers();
4184 StringView Num = parseNumber();
4185 if (!consumeIf('_'))
4186 return nullptr;
4187 return make<FunctionParam>(Num);
4188 }
4189 return nullptr;
4190}
4191
4192// [gs] nw <expression>* _ <type> E # new (expr-list) type
4193// [gs] nw <expression>* _ <type> <initializer> # new (expr-list) type (init)
4194// [gs] na <expression>* _ <type> E # new[] (expr-list) type
4195// [gs] na <expression>* _ <type> <initializer> # new[] (expr-list) type (init)
4196// <initializer> ::= pi <expression>* E # parenthesized initialization
Pavel Labathb26cabc2018-10-16 14:29:14 +00004197template <typename Derived, typename Alloc>
4198Node *AbstractManglingParser<Derived, Alloc>::parseNewExpr() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004199 bool Global = consumeIf("gs");
4200 bool IsArray = look(1) == 'a';
4201 if (!consumeIf("nw") && !consumeIf("na"))
4202 return nullptr;
4203 size_t Exprs = Names.size();
4204 while (!consumeIf('_')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004205 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004206 if (Ex == nullptr)
4207 return nullptr;
4208 Names.push_back(Ex);
4209 }
4210 NodeArray ExprList = popTrailingNodeArray(Exprs);
Pavel Labathb26cabc2018-10-16 14:29:14 +00004211 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004212 if (Ty == nullptr)
4213 return Ty;
4214 if (consumeIf("pi")) {
4215 size_t InitsBegin = Names.size();
4216 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004217 Node *Init = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004218 if (Init == nullptr)
4219 return Init;
4220 Names.push_back(Init);
4221 }
4222 NodeArray Inits = popTrailingNodeArray(InitsBegin);
4223 return make<NewExpr>(ExprList, Ty, Inits, Global, IsArray);
4224 } else if (!consumeIf('E'))
4225 return nullptr;
4226 return make<NewExpr>(ExprList, Ty, NodeArray(), Global, IsArray);
4227}
4228
4229// cv <type> <expression> # conversion with one argument
4230// cv <type> _ <expression>* E # conversion with a different number of arguments
Pavel Labathb26cabc2018-10-16 14:29:14 +00004231template <typename Derived, typename Alloc>
4232Node *AbstractManglingParser<Derived, Alloc>::parseConversionExpr() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004233 if (!consumeIf("cv"))
4234 return nullptr;
4235 Node *Ty;
4236 {
4237 SwapAndRestore<bool> SaveTemp(TryToParseTemplateArgs, false);
Pavel Labathb26cabc2018-10-16 14:29:14 +00004238 Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004239 }
4240
4241 if (Ty == nullptr)
4242 return nullptr;
4243
4244 if (consumeIf('_')) {
4245 size_t ExprsBegin = Names.size();
4246 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004247 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004248 if (E == nullptr)
4249 return E;
4250 Names.push_back(E);
4251 }
4252 NodeArray Exprs = popTrailingNodeArray(ExprsBegin);
4253 return make<ConversionExpr>(Ty, Exprs);
4254 }
4255
Pavel Labathb26cabc2018-10-16 14:29:14 +00004256 Node *E[1] = {getDerived().parseExpr()};
Richard Smith4344cbd2018-08-20 20:14:49 +00004257 if (E[0] == nullptr)
4258 return nullptr;
4259 return make<ConversionExpr>(Ty, makeNodeArray(E, E + 1));
4260}
4261
4262// <expr-primary> ::= L <type> <value number> E # integer literal
4263// ::= L <type> <value float> E # floating literal
4264// ::= L <string type> E # string literal
4265// ::= L <nullptr type> E # nullptr literal (i.e., "LDnE")
Richard Smithcbffe542019-09-06 23:53:21 +00004266// ::= L <lambda type> E # lambda expression
Richard Smith4344cbd2018-08-20 20:14:49 +00004267// FIXME: ::= L <type> <real-part float> _ <imag-part float> E # complex floating point literal (C 2000)
4268// ::= L <mangled-name> E # external name
Pavel Labathb26cabc2018-10-16 14:29:14 +00004269template <typename Derived, typename Alloc>
4270Node *AbstractManglingParser<Derived, Alloc>::parseExprPrimary() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004271 if (!consumeIf('L'))
4272 return nullptr;
4273 switch (look()) {
4274 case 'w':
4275 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004276 return getDerived().parseIntegerLiteral("wchar_t");
Richard Smith4344cbd2018-08-20 20:14:49 +00004277 case 'b':
4278 if (consumeIf("b0E"))
4279 return make<BoolExpr>(0);
4280 if (consumeIf("b1E"))
4281 return make<BoolExpr>(1);
4282 return nullptr;
4283 case 'c':
4284 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004285 return getDerived().parseIntegerLiteral("char");
Richard Smith4344cbd2018-08-20 20:14:49 +00004286 case 'a':
4287 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004288 return getDerived().parseIntegerLiteral("signed char");
Richard Smith4344cbd2018-08-20 20:14:49 +00004289 case 'h':
4290 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004291 return getDerived().parseIntegerLiteral("unsigned char");
Richard Smith4344cbd2018-08-20 20:14:49 +00004292 case 's':
4293 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004294 return getDerived().parseIntegerLiteral("short");
Richard Smith4344cbd2018-08-20 20:14:49 +00004295 case 't':
4296 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004297 return getDerived().parseIntegerLiteral("unsigned short");
Richard Smith4344cbd2018-08-20 20:14:49 +00004298 case 'i':
4299 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004300 return getDerived().parseIntegerLiteral("");
Richard Smith4344cbd2018-08-20 20:14:49 +00004301 case 'j':
4302 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004303 return getDerived().parseIntegerLiteral("u");
Richard Smith4344cbd2018-08-20 20:14:49 +00004304 case 'l':
4305 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004306 return getDerived().parseIntegerLiteral("l");
Richard Smith4344cbd2018-08-20 20:14:49 +00004307 case 'm':
4308 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004309 return getDerived().parseIntegerLiteral("ul");
Richard Smith4344cbd2018-08-20 20:14:49 +00004310 case 'x':
4311 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004312 return getDerived().parseIntegerLiteral("ll");
Richard Smith4344cbd2018-08-20 20:14:49 +00004313 case 'y':
4314 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004315 return getDerived().parseIntegerLiteral("ull");
Richard Smith4344cbd2018-08-20 20:14:49 +00004316 case 'n':
4317 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004318 return getDerived().parseIntegerLiteral("__int128");
Richard Smith4344cbd2018-08-20 20:14:49 +00004319 case 'o':
4320 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004321 return getDerived().parseIntegerLiteral("unsigned __int128");
Richard Smith4344cbd2018-08-20 20:14:49 +00004322 case 'f':
4323 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004324 return getDerived().template parseFloatingLiteral<float>();
Richard Smith4344cbd2018-08-20 20:14:49 +00004325 case 'd':
4326 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004327 return getDerived().template parseFloatingLiteral<double>();
Richard Smith4344cbd2018-08-20 20:14:49 +00004328 case 'e':
4329 ++First;
Xing Xuebe1ae5a2020-04-15 09:59:06 -04004330#if defined(__powerpc__) || defined(__s390__)
4331 // Handle cases where long doubles encoded with e have the same size
4332 // and representation as doubles.
4333 return getDerived().template parseFloatingLiteral<double>();
4334#else
Pavel Labathb26cabc2018-10-16 14:29:14 +00004335 return getDerived().template parseFloatingLiteral<long double>();
Xing Xuebe1ae5a2020-04-15 09:59:06 -04004336#endif
Richard Smith4344cbd2018-08-20 20:14:49 +00004337 case '_':
4338 if (consumeIf("_Z")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004339 Node *R = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00004340 if (R != nullptr && consumeIf('E'))
4341 return R;
4342 }
4343 return nullptr;
Richard Smithcbffe542019-09-06 23:53:21 +00004344 case 'A': {
4345 Node *T = getDerived().parseType();
4346 if (T == nullptr)
4347 return nullptr;
4348 // FIXME: We need to include the string contents in the mangling.
4349 if (consumeIf('E'))
4350 return make<StringLiteral>(T);
4351 return nullptr;
4352 }
4353 case 'D':
4354 if (consumeIf("DnE"))
4355 return make<NameType>("nullptr");
4356 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00004357 case 'T':
4358 // Invalid mangled name per
4359 // http://sourcerytools.com/pipermail/cxx-abi-dev/2011-August/002422.html
4360 return nullptr;
Richard Smith069f19d2019-09-09 22:26:04 +00004361 case 'U': {
4362 // FIXME: Should we support LUb... for block literals?
4363 if (look(1) != 'l')
4364 return nullptr;
4365 Node *T = parseUnnamedTypeName(nullptr);
4366 if (!T || !consumeIf('E'))
4367 return nullptr;
4368 return make<LambdaExpr>(T);
4369 }
Richard Smith4344cbd2018-08-20 20:14:49 +00004370 default: {
4371 // might be named type
Pavel Labathb26cabc2018-10-16 14:29:14 +00004372 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004373 if (T == nullptr)
4374 return nullptr;
Erik Pilkington27011a82020-05-13 14:13:37 -04004375 StringView N = parseNumber(/*AllowNegative=*/true);
Richard Smith069f19d2019-09-09 22:26:04 +00004376 if (N.empty())
4377 return nullptr;
4378 if (!consumeIf('E'))
4379 return nullptr;
Erik Pilkington27011a82020-05-13 14:13:37 -04004380 return make<EnumLiteral>(T, N);
Richard Smith4344cbd2018-08-20 20:14:49 +00004381 }
4382 }
4383}
4384
4385// <braced-expression> ::= <expression>
4386// ::= di <field source-name> <braced-expression> # .name = expr
4387// ::= dx <index expression> <braced-expression> # [expr] = expr
4388// ::= dX <range begin expression> <range end expression> <braced-expression>
Pavel Labathb26cabc2018-10-16 14:29:14 +00004389template <typename Derived, typename Alloc>
4390Node *AbstractManglingParser<Derived, Alloc>::parseBracedExpr() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004391 if (look() == 'd') {
4392 switch (look(1)) {
4393 case 'i': {
4394 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004395 Node *Field = getDerived().parseSourceName(/*NameState=*/nullptr);
Richard Smith4344cbd2018-08-20 20:14:49 +00004396 if (Field == nullptr)
4397 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004398 Node *Init = getDerived().parseBracedExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004399 if (Init == nullptr)
4400 return nullptr;
4401 return make<BracedExpr>(Field, Init, /*isArray=*/false);
4402 }
4403 case 'x': {
4404 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004405 Node *Index = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004406 if (Index == nullptr)
4407 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004408 Node *Init = getDerived().parseBracedExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004409 if (Init == nullptr)
4410 return nullptr;
4411 return make<BracedExpr>(Index, Init, /*isArray=*/true);
4412 }
4413 case 'X': {
4414 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004415 Node *RangeBegin = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004416 if (RangeBegin == nullptr)
4417 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004418 Node *RangeEnd = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004419 if (RangeEnd == nullptr)
4420 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004421 Node *Init = getDerived().parseBracedExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004422 if (Init == nullptr)
4423 return nullptr;
4424 return make<BracedRangeExpr>(RangeBegin, RangeEnd, Init);
4425 }
4426 }
4427 }
Pavel Labathb26cabc2018-10-16 14:29:14 +00004428 return getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004429}
4430
4431// (not yet in the spec)
4432// <fold-expr> ::= fL <binary-operator-name> <expression> <expression>
4433// ::= fR <binary-operator-name> <expression> <expression>
4434// ::= fl <binary-operator-name> <expression>
4435// ::= fr <binary-operator-name> <expression>
Pavel Labathb26cabc2018-10-16 14:29:14 +00004436template <typename Derived, typename Alloc>
4437Node *AbstractManglingParser<Derived, Alloc>::parseFoldExpr() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004438 if (!consumeIf('f'))
4439 return nullptr;
4440
4441 char FoldKind = look();
4442 bool IsLeftFold, HasInitializer;
4443 HasInitializer = FoldKind == 'L' || FoldKind == 'R';
4444 if (FoldKind == 'l' || FoldKind == 'L')
4445 IsLeftFold = true;
4446 else if (FoldKind == 'r' || FoldKind == 'R')
4447 IsLeftFold = false;
4448 else
4449 return nullptr;
4450 ++First;
4451
4452 // FIXME: This map is duplicated in parseOperatorName and parseExpr.
4453 StringView OperatorName;
4454 if (consumeIf("aa")) OperatorName = "&&";
4455 else if (consumeIf("an")) OperatorName = "&";
4456 else if (consumeIf("aN")) OperatorName = "&=";
4457 else if (consumeIf("aS")) OperatorName = "=";
4458 else if (consumeIf("cm")) OperatorName = ",";
4459 else if (consumeIf("ds")) OperatorName = ".*";
4460 else if (consumeIf("dv")) OperatorName = "/";
4461 else if (consumeIf("dV")) OperatorName = "/=";
4462 else if (consumeIf("eo")) OperatorName = "^";
4463 else if (consumeIf("eO")) OperatorName = "^=";
4464 else if (consumeIf("eq")) OperatorName = "==";
4465 else if (consumeIf("ge")) OperatorName = ">=";
4466 else if (consumeIf("gt")) OperatorName = ">";
4467 else if (consumeIf("le")) OperatorName = "<=";
4468 else if (consumeIf("ls")) OperatorName = "<<";
4469 else if (consumeIf("lS")) OperatorName = "<<=";
4470 else if (consumeIf("lt")) OperatorName = "<";
4471 else if (consumeIf("mi")) OperatorName = "-";
4472 else if (consumeIf("mI")) OperatorName = "-=";
4473 else if (consumeIf("ml")) OperatorName = "*";
4474 else if (consumeIf("mL")) OperatorName = "*=";
4475 else if (consumeIf("ne")) OperatorName = "!=";
4476 else if (consumeIf("oo")) OperatorName = "||";
4477 else if (consumeIf("or")) OperatorName = "|";
4478 else if (consumeIf("oR")) OperatorName = "|=";
4479 else if (consumeIf("pl")) OperatorName = "+";
4480 else if (consumeIf("pL")) OperatorName = "+=";
4481 else if (consumeIf("rm")) OperatorName = "%";
4482 else if (consumeIf("rM")) OperatorName = "%=";
4483 else if (consumeIf("rs")) OperatorName = ">>";
4484 else if (consumeIf("rS")) OperatorName = ">>=";
4485 else return nullptr;
4486
Pavel Labathb26cabc2018-10-16 14:29:14 +00004487 Node *Pack = getDerived().parseExpr(), *Init = nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00004488 if (Pack == nullptr)
4489 return nullptr;
4490 if (HasInitializer) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004491 Init = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004492 if (Init == nullptr)
4493 return nullptr;
4494 }
4495
4496 if (IsLeftFold && Init)
4497 std::swap(Pack, Init);
4498
4499 return make<FoldExpr>(IsLeftFold, OperatorName, Pack, Init);
4500}
4501
Richard Smith2b38a692020-10-22 19:29:36 -07004502// <expression> ::= mc <parameter type> <expr> [<offset number>] E
4503//
4504// Not yet in the spec: https://github.com/itanium-cxx-abi/cxx-abi/issues/47
4505template <typename Derived, typename Alloc>
4506Node *AbstractManglingParser<Derived, Alloc>::parsePointerToMemberConversionExpr() {
4507 Node *Ty = getDerived().parseType();
4508 if (!Ty)
4509 return nullptr;
4510 Node *Expr = getDerived().parseExpr();
4511 if (!Expr)
4512 return nullptr;
4513 StringView Offset = getDerived().parseNumber(true);
4514 if (!consumeIf('E'))
4515 return nullptr;
4516 return make<PointerToMemberConversionExpr>(Ty, Expr, Offset);
4517}
4518
4519// <expression> ::= so <referent type> <expr> [<offset number>] <union-selector>* [p] E
4520// <union-selector> ::= _ [<number>]
4521//
4522// Not yet in the spec: https://github.com/itanium-cxx-abi/cxx-abi/issues/47
4523template <typename Derived, typename Alloc>
4524Node *AbstractManglingParser<Derived, Alloc>::parseSubobjectExpr() {
4525 Node *Ty = getDerived().parseType();
4526 if (!Ty)
4527 return nullptr;
4528 Node *Expr = getDerived().parseExpr();
4529 if (!Expr)
4530 return nullptr;
4531 StringView Offset = getDerived().parseNumber(true);
4532 size_t SelectorsBegin = Names.size();
4533 while (consumeIf('_')) {
4534 Node *Selector = make<NameType>(parseNumber());
4535 if (!Selector)
4536 return nullptr;
4537 Names.push_back(Selector);
4538 }
4539 bool OnePastTheEnd = consumeIf('p');
4540 if (!consumeIf('E'))
4541 return nullptr;
4542 return make<SubobjectExpr>(
4543 Ty, Expr, Offset, popTrailingNodeArray(SelectorsBegin), OnePastTheEnd);
4544}
4545
Richard Smith4344cbd2018-08-20 20:14:49 +00004546// <expression> ::= <unary operator-name> <expression>
4547// ::= <binary operator-name> <expression> <expression>
4548// ::= <ternary operator-name> <expression> <expression> <expression>
4549// ::= cl <expression>+ E # call
4550// ::= cv <type> <expression> # conversion with one argument
4551// ::= cv <type> _ <expression>* E # conversion with a different number of arguments
4552// ::= [gs] nw <expression>* _ <type> E # new (expr-list) type
4553// ::= [gs] nw <expression>* _ <type> <initializer> # new (expr-list) type (init)
4554// ::= [gs] na <expression>* _ <type> E # new[] (expr-list) type
4555// ::= [gs] na <expression>* _ <type> <initializer> # new[] (expr-list) type (init)
4556// ::= [gs] dl <expression> # delete expression
4557// ::= [gs] da <expression> # delete[] expression
4558// ::= pp_ <expression> # prefix ++
4559// ::= mm_ <expression> # prefix --
4560// ::= ti <type> # typeid (type)
4561// ::= te <expression> # typeid (expression)
4562// ::= dc <type> <expression> # dynamic_cast<type> (expression)
4563// ::= sc <type> <expression> # static_cast<type> (expression)
4564// ::= cc <type> <expression> # const_cast<type> (expression)
4565// ::= rc <type> <expression> # reinterpret_cast<type> (expression)
4566// ::= st <type> # sizeof (a type)
4567// ::= sz <expression> # sizeof (an expression)
4568// ::= at <type> # alignof (a type)
4569// ::= az <expression> # alignof (an expression)
4570// ::= nx <expression> # noexcept (expression)
4571// ::= <template-param>
4572// ::= <function-param>
4573// ::= dt <expression> <unresolved-name> # expr.name
4574// ::= pt <expression> <unresolved-name> # expr->name
4575// ::= ds <expression> <expression> # expr.*expr
4576// ::= sZ <template-param> # size of a parameter pack
4577// ::= sZ <function-param> # size of a function parameter pack
4578// ::= sP <template-arg>* E # sizeof...(T), size of a captured template parameter pack from an alias template
4579// ::= sp <expression> # pack expansion
4580// ::= tw <expression> # throw expression
4581// ::= tr # throw with no operand (rethrow)
4582// ::= <unresolved-name> # f(p), N::f(p), ::f(p),
4583// # freestanding dependent name (e.g., T::x),
4584// # objectless nonstatic member reference
4585// ::= fL <binary-operator-name> <expression> <expression>
4586// ::= fR <binary-operator-name> <expression> <expression>
4587// ::= fl <binary-operator-name> <expression>
4588// ::= fr <binary-operator-name> <expression>
4589// ::= <expr-primary>
Pavel Labathb26cabc2018-10-16 14:29:14 +00004590template <typename Derived, typename Alloc>
4591Node *AbstractManglingParser<Derived, Alloc>::parseExpr() {
Richard Smith4344cbd2018-08-20 20:14:49 +00004592 bool Global = consumeIf("gs");
4593 if (numLeft() < 2)
4594 return nullptr;
4595
4596 switch (*First) {
4597 case 'L':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004598 return getDerived().parseExprPrimary();
Richard Smith4344cbd2018-08-20 20:14:49 +00004599 case 'T':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004600 return getDerived().parseTemplateParam();
Richard Smith4344cbd2018-08-20 20:14:49 +00004601 case 'f': {
4602 // Disambiguate a fold expression from a <function-param>.
4603 if (look(1) == 'p' || (look(1) == 'L' && std::isdigit(look(2))))
Pavel Labathb26cabc2018-10-16 14:29:14 +00004604 return getDerived().parseFunctionParam();
4605 return getDerived().parseFoldExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004606 }
4607 case 'a':
4608 switch (First[1]) {
4609 case 'a':
4610 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004611 return getDerived().parseBinaryExpr("&&");
Richard Smith4344cbd2018-08-20 20:14:49 +00004612 case 'd':
4613 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004614 return getDerived().parsePrefixExpr("&");
Richard Smith4344cbd2018-08-20 20:14:49 +00004615 case 'n':
4616 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004617 return getDerived().parseBinaryExpr("&");
Richard Smith4344cbd2018-08-20 20:14:49 +00004618 case 'N':
4619 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004620 return getDerived().parseBinaryExpr("&=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004621 case 'S':
4622 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004623 return getDerived().parseBinaryExpr("=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004624 case 't': {
4625 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004626 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004627 if (Ty == nullptr)
4628 return nullptr;
4629 return make<EnclosingExpr>("alignof (", Ty, ")");
4630 }
4631 case 'z': {
4632 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004633 Node *Ty = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004634 if (Ty == nullptr)
4635 return nullptr;
4636 return make<EnclosingExpr>("alignof (", Ty, ")");
4637 }
4638 }
4639 return nullptr;
4640 case 'c':
4641 switch (First[1]) {
4642 // cc <type> <expression> # const_cast<type>(expression)
4643 case 'c': {
4644 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004645 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004646 if (Ty == nullptr)
4647 return Ty;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004648 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004649 if (Ex == nullptr)
4650 return Ex;
4651 return make<CastExpr>("const_cast", Ty, Ex);
4652 }
4653 // cl <expression>+ E # call
4654 case 'l': {
4655 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004656 Node *Callee = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004657 if (Callee == nullptr)
4658 return Callee;
4659 size_t ExprsBegin = Names.size();
4660 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004661 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004662 if (E == nullptr)
4663 return E;
4664 Names.push_back(E);
4665 }
4666 return make<CallExpr>(Callee, popTrailingNodeArray(ExprsBegin));
4667 }
4668 case 'm':
4669 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004670 return getDerived().parseBinaryExpr(",");
Richard Smith4344cbd2018-08-20 20:14:49 +00004671 case 'o':
4672 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004673 return getDerived().parsePrefixExpr("~");
Richard Smith4344cbd2018-08-20 20:14:49 +00004674 case 'v':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004675 return getDerived().parseConversionExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004676 }
4677 return nullptr;
4678 case 'd':
4679 switch (First[1]) {
4680 case 'a': {
4681 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004682 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004683 if (Ex == nullptr)
4684 return Ex;
4685 return make<DeleteExpr>(Ex, Global, /*is_array=*/true);
4686 }
4687 case 'c': {
4688 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004689 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004690 if (T == nullptr)
4691 return T;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004692 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004693 if (Ex == nullptr)
4694 return Ex;
4695 return make<CastExpr>("dynamic_cast", T, Ex);
4696 }
4697 case 'e':
4698 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004699 return getDerived().parsePrefixExpr("*");
Richard Smith4344cbd2018-08-20 20:14:49 +00004700 case 'l': {
4701 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004702 Node *E = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004703 if (E == nullptr)
4704 return E;
4705 return make<DeleteExpr>(E, Global, /*is_array=*/false);
4706 }
4707 case 'n':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004708 return getDerived().parseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00004709 case 's': {
4710 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004711 Node *LHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004712 if (LHS == nullptr)
4713 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004714 Node *RHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004715 if (RHS == nullptr)
4716 return nullptr;
4717 return make<MemberExpr>(LHS, ".*", RHS);
4718 }
4719 case 't': {
4720 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004721 Node *LHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004722 if (LHS == nullptr)
4723 return LHS;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004724 Node *RHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004725 if (RHS == nullptr)
4726 return nullptr;
4727 return make<MemberExpr>(LHS, ".", RHS);
4728 }
4729 case 'v':
4730 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004731 return getDerived().parseBinaryExpr("/");
Richard Smith4344cbd2018-08-20 20:14:49 +00004732 case 'V':
4733 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004734 return getDerived().parseBinaryExpr("/=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004735 }
4736 return nullptr;
4737 case 'e':
4738 switch (First[1]) {
4739 case 'o':
4740 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004741 return getDerived().parseBinaryExpr("^");
Richard Smith4344cbd2018-08-20 20:14:49 +00004742 case 'O':
4743 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004744 return getDerived().parseBinaryExpr("^=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004745 case 'q':
4746 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004747 return getDerived().parseBinaryExpr("==");
Richard Smith4344cbd2018-08-20 20:14:49 +00004748 }
4749 return nullptr;
4750 case 'g':
4751 switch (First[1]) {
4752 case 'e':
4753 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004754 return getDerived().parseBinaryExpr(">=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004755 case 't':
4756 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004757 return getDerived().parseBinaryExpr(">");
Richard Smith4344cbd2018-08-20 20:14:49 +00004758 }
4759 return nullptr;
4760 case 'i':
4761 switch (First[1]) {
4762 case 'x': {
4763 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004764 Node *Base = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004765 if (Base == nullptr)
4766 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004767 Node *Index = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004768 if (Index == nullptr)
4769 return Index;
4770 return make<ArraySubscriptExpr>(Base, Index);
4771 }
4772 case 'l': {
4773 First += 2;
4774 size_t InitsBegin = Names.size();
4775 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004776 Node *E = getDerived().parseBracedExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004777 if (E == nullptr)
4778 return nullptr;
4779 Names.push_back(E);
4780 }
4781 return make<InitListExpr>(nullptr, popTrailingNodeArray(InitsBegin));
4782 }
4783 }
4784 return nullptr;
4785 case 'l':
4786 switch (First[1]) {
4787 case 'e':
4788 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004789 return getDerived().parseBinaryExpr("<=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004790 case 's':
4791 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004792 return getDerived().parseBinaryExpr("<<");
Richard Smith4344cbd2018-08-20 20:14:49 +00004793 case 'S':
4794 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004795 return getDerived().parseBinaryExpr("<<=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004796 case 't':
4797 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004798 return getDerived().parseBinaryExpr("<");
Richard Smith4344cbd2018-08-20 20:14:49 +00004799 }
4800 return nullptr;
4801 case 'm':
4802 switch (First[1]) {
Richard Smith2b38a692020-10-22 19:29:36 -07004803 case 'c':
4804 First += 2;
4805 return parsePointerToMemberConversionExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004806 case 'i':
4807 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004808 return getDerived().parseBinaryExpr("-");
Richard Smith4344cbd2018-08-20 20:14:49 +00004809 case 'I':
4810 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004811 return getDerived().parseBinaryExpr("-=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004812 case 'l':
4813 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004814 return getDerived().parseBinaryExpr("*");
Richard Smith4344cbd2018-08-20 20:14:49 +00004815 case 'L':
4816 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004817 return getDerived().parseBinaryExpr("*=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004818 case 'm':
4819 First += 2;
4820 if (consumeIf('_'))
Pavel Labathb26cabc2018-10-16 14:29:14 +00004821 return getDerived().parsePrefixExpr("--");
4822 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004823 if (Ex == nullptr)
4824 return nullptr;
4825 return make<PostfixExpr>(Ex, "--");
4826 }
4827 return nullptr;
4828 case 'n':
4829 switch (First[1]) {
4830 case 'a':
4831 case 'w':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004832 return getDerived().parseNewExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004833 case 'e':
4834 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004835 return getDerived().parseBinaryExpr("!=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004836 case 'g':
4837 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004838 return getDerived().parsePrefixExpr("-");
Richard Smith4344cbd2018-08-20 20:14:49 +00004839 case 't':
4840 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004841 return getDerived().parsePrefixExpr("!");
Richard Smith4344cbd2018-08-20 20:14:49 +00004842 case 'x':
4843 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004844 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004845 if (Ex == nullptr)
4846 return Ex;
4847 return make<EnclosingExpr>("noexcept (", Ex, ")");
4848 }
4849 return nullptr;
4850 case 'o':
4851 switch (First[1]) {
4852 case 'n':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004853 return getDerived().parseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00004854 case 'o':
4855 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004856 return getDerived().parseBinaryExpr("||");
Richard Smith4344cbd2018-08-20 20:14:49 +00004857 case 'r':
4858 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004859 return getDerived().parseBinaryExpr("|");
Richard Smith4344cbd2018-08-20 20:14:49 +00004860 case 'R':
4861 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004862 return getDerived().parseBinaryExpr("|=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004863 }
4864 return nullptr;
4865 case 'p':
4866 switch (First[1]) {
4867 case 'm':
4868 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004869 return getDerived().parseBinaryExpr("->*");
Richard Smith4344cbd2018-08-20 20:14:49 +00004870 case 'l':
4871 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004872 return getDerived().parseBinaryExpr("+");
Richard Smith4344cbd2018-08-20 20:14:49 +00004873 case 'L':
4874 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004875 return getDerived().parseBinaryExpr("+=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004876 case 'p': {
4877 First += 2;
4878 if (consumeIf('_'))
Pavel Labathb26cabc2018-10-16 14:29:14 +00004879 return getDerived().parsePrefixExpr("++");
4880 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004881 if (Ex == nullptr)
4882 return Ex;
4883 return make<PostfixExpr>(Ex, "++");
4884 }
4885 case 's':
4886 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004887 return getDerived().parsePrefixExpr("+");
Richard Smith4344cbd2018-08-20 20:14:49 +00004888 case 't': {
4889 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004890 Node *L = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004891 if (L == nullptr)
4892 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004893 Node *R = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004894 if (R == nullptr)
4895 return nullptr;
4896 return make<MemberExpr>(L, "->", R);
4897 }
4898 }
4899 return nullptr;
4900 case 'q':
4901 if (First[1] == 'u') {
4902 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004903 Node *Cond = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004904 if (Cond == nullptr)
4905 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004906 Node *LHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004907 if (LHS == nullptr)
4908 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004909 Node *RHS = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004910 if (RHS == nullptr)
4911 return nullptr;
4912 return make<ConditionalExpr>(Cond, LHS, RHS);
4913 }
4914 return nullptr;
4915 case 'r':
4916 switch (First[1]) {
4917 case 'c': {
4918 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004919 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004920 if (T == nullptr)
4921 return T;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004922 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004923 if (Ex == nullptr)
4924 return Ex;
4925 return make<CastExpr>("reinterpret_cast", T, Ex);
4926 }
4927 case 'm':
4928 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004929 return getDerived().parseBinaryExpr("%");
Richard Smith4344cbd2018-08-20 20:14:49 +00004930 case 'M':
4931 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004932 return getDerived().parseBinaryExpr("%=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004933 case 's':
4934 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004935 return getDerived().parseBinaryExpr(">>");
Richard Smith4344cbd2018-08-20 20:14:49 +00004936 case 'S':
4937 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004938 return getDerived().parseBinaryExpr(">>=");
Richard Smith4344cbd2018-08-20 20:14:49 +00004939 }
4940 return nullptr;
4941 case 's':
4942 switch (First[1]) {
4943 case 'c': {
4944 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004945 Node *T = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004946 if (T == nullptr)
4947 return T;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004948 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004949 if (Ex == nullptr)
4950 return Ex;
4951 return make<CastExpr>("static_cast", T, Ex);
4952 }
Richard Smith2b38a692020-10-22 19:29:36 -07004953 case 'o':
4954 First += 2;
4955 return parseSubobjectExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004956 case 'p': {
4957 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004958 Node *Child = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004959 if (Child == nullptr)
4960 return nullptr;
4961 return make<ParameterPackExpansion>(Child);
4962 }
4963 case 'r':
Pavel Labathb26cabc2018-10-16 14:29:14 +00004964 return getDerived().parseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00004965 case 't': {
4966 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004967 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00004968 if (Ty == nullptr)
4969 return Ty;
4970 return make<EnclosingExpr>("sizeof (", Ty, ")");
4971 }
4972 case 'z': {
4973 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00004974 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00004975 if (Ex == nullptr)
4976 return Ex;
4977 return make<EnclosingExpr>("sizeof (", Ex, ")");
4978 }
4979 case 'Z':
4980 First += 2;
4981 if (look() == 'T') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004982 Node *R = getDerived().parseTemplateParam();
Richard Smith4344cbd2018-08-20 20:14:49 +00004983 if (R == nullptr)
4984 return nullptr;
4985 return make<SizeofParamPackExpr>(R);
4986 } else if (look() == 'f') {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004987 Node *FP = getDerived().parseFunctionParam();
Richard Smith4344cbd2018-08-20 20:14:49 +00004988 if (FP == nullptr)
4989 return nullptr;
4990 return make<EnclosingExpr>("sizeof... (", FP, ")");
4991 }
4992 return nullptr;
4993 case 'P': {
4994 First += 2;
4995 size_t ArgsBegin = Names.size();
4996 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00004997 Node *Arg = getDerived().parseTemplateArg();
Richard Smith4344cbd2018-08-20 20:14:49 +00004998 if (Arg == nullptr)
4999 return nullptr;
5000 Names.push_back(Arg);
5001 }
Richard Smith7918dea2018-08-24 23:30:26 +00005002 auto *Pack = make<NodeArrayNode>(popTrailingNodeArray(ArgsBegin));
5003 if (!Pack)
5004 return nullptr;
5005 return make<EnclosingExpr>("sizeof... (", Pack, ")");
Richard Smith4344cbd2018-08-20 20:14:49 +00005006 }
5007 }
5008 return nullptr;
5009 case 't':
5010 switch (First[1]) {
5011 case 'e': {
5012 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005013 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00005014 if (Ex == nullptr)
5015 return Ex;
5016 return make<EnclosingExpr>("typeid (", Ex, ")");
5017 }
5018 case 'i': {
5019 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005020 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005021 if (Ty == nullptr)
5022 return Ty;
5023 return make<EnclosingExpr>("typeid (", Ty, ")");
5024 }
5025 case 'l': {
5026 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005027 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005028 if (Ty == nullptr)
5029 return nullptr;
5030 size_t InitsBegin = Names.size();
5031 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005032 Node *E = getDerived().parseBracedExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00005033 if (E == nullptr)
5034 return nullptr;
5035 Names.push_back(E);
5036 }
5037 return make<InitListExpr>(Ty, popTrailingNodeArray(InitsBegin));
5038 }
5039 case 'r':
5040 First += 2;
5041 return make<NameType>("throw");
5042 case 'w': {
5043 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005044 Node *Ex = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00005045 if (Ex == nullptr)
5046 return nullptr;
5047 return make<ThrowExpr>(Ex);
5048 }
5049 }
5050 return nullptr;
James Y Knight415432d2020-12-07 10:26:49 -05005051 case 'u': {
5052 ++First;
5053 Node *Name = getDerived().parseSourceName(/*NameState=*/nullptr);
5054 if (!Name)
5055 return nullptr;
5056 // Special case legacy __uuidof mangling. The 't' and 'z' appear where the
5057 // standard encoding expects a <template-arg>, and would be otherwise be
5058 // interpreted as <type> node 'short' or 'ellipsis'. However, neither
5059 // __uuidof(short) nor __uuidof(...) can actually appear, so there is no
5060 // actual conflict here.
5061 if (Name->getBaseName() == "__uuidof") {
5062 if (numLeft() < 2)
5063 return nullptr;
5064 if (*First == 't') {
5065 ++First;
5066 Node *Ty = getDerived().parseType();
5067 if (!Ty)
5068 return nullptr;
5069 return make<CallExpr>(Name, makeNodeArray(&Ty, &Ty + 1));
5070 }
5071 if (*First == 'z') {
5072 ++First;
5073 Node *Ex = getDerived().parseExpr();
5074 if (!Ex)
5075 return nullptr;
5076 return make<CallExpr>(Name, makeNodeArray(&Ex, &Ex + 1));
5077 }
5078 }
5079 size_t ExprsBegin = Names.size();
5080 while (!consumeIf('E')) {
5081 Node *E = getDerived().parseTemplateArg();
5082 if (E == nullptr)
5083 return E;
5084 Names.push_back(E);
5085 }
5086 return make<CallExpr>(Name, popTrailingNodeArray(ExprsBegin));
5087 }
Richard Smith4344cbd2018-08-20 20:14:49 +00005088 case '1':
5089 case '2':
5090 case '3':
5091 case '4':
5092 case '5':
5093 case '6':
5094 case '7':
5095 case '8':
5096 case '9':
Pavel Labathb26cabc2018-10-16 14:29:14 +00005097 return getDerived().parseUnresolvedName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005098 }
5099 return nullptr;
5100}
5101
5102// <call-offset> ::= h <nv-offset> _
5103// ::= v <v-offset> _
5104//
5105// <nv-offset> ::= <offset number>
5106// # non-virtual base override
5107//
5108// <v-offset> ::= <offset number> _ <virtual offset number>
5109// # virtual base override, with vcall offset
Pavel Labathb26cabc2018-10-16 14:29:14 +00005110template <typename Alloc, typename Derived>
5111bool AbstractManglingParser<Alloc, Derived>::parseCallOffset() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005112 // Just scan through the call offset, we never add this information into the
5113 // output.
5114 if (consumeIf('h'))
5115 return parseNumber(true).empty() || !consumeIf('_');
5116 if (consumeIf('v'))
5117 return parseNumber(true).empty() || !consumeIf('_') ||
5118 parseNumber(true).empty() || !consumeIf('_');
5119 return true;
5120}
5121
5122// <special-name> ::= TV <type> # virtual table
5123// ::= TT <type> # VTT structure (construction vtable index)
5124// ::= TI <type> # typeinfo structure
5125// ::= TS <type> # typeinfo name (null-terminated byte string)
5126// ::= Tc <call-offset> <call-offset> <base encoding>
5127// # base is the nominal target function of thunk
5128// # first call-offset is 'this' adjustment
5129// # second call-offset is result adjustment
5130// ::= T <call-offset> <base encoding>
5131// # base is the nominal target function of thunk
5132// ::= GV <object name> # Guard variable for one-time initialization
5133// # No <type>
5134// ::= TW <object name> # Thread-local wrapper
5135// ::= TH <object name> # Thread-local initialization
5136// ::= GR <object name> _ # First temporary
5137// ::= GR <object name> <seq-id> _ # Subsequent temporaries
5138// extension ::= TC <first type> <number> _ <second type> # construction vtable for second-in-first
5139// extension ::= GR <object name> # reference temporary for object
Pavel Labathb26cabc2018-10-16 14:29:14 +00005140template <typename Derived, typename Alloc>
5141Node *AbstractManglingParser<Derived, Alloc>::parseSpecialName() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005142 switch (look()) {
5143 case 'T':
5144 switch (look(1)) {
Richard Smith2b38a692020-10-22 19:29:36 -07005145 // TA <template-arg> # template parameter object
5146 //
5147 // Not yet in the spec: https://github.com/itanium-cxx-abi/cxx-abi/issues/63
5148 case 'A': {
5149 First += 2;
5150 Node *Arg = getDerived().parseTemplateArg();
5151 if (Arg == nullptr)
5152 return nullptr;
5153 return make<SpecialName>("template parameter object for ", Arg);
5154 }
Richard Smith4344cbd2018-08-20 20:14:49 +00005155 // TV <type> # virtual table
5156 case 'V': {
5157 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005158 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005159 if (Ty == nullptr)
5160 return nullptr;
5161 return make<SpecialName>("vtable for ", Ty);
5162 }
5163 // TT <type> # VTT structure (construction vtable index)
5164 case 'T': {
5165 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005166 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005167 if (Ty == nullptr)
5168 return nullptr;
5169 return make<SpecialName>("VTT for ", Ty);
5170 }
5171 // TI <type> # typeinfo structure
5172 case 'I': {
5173 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005174 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005175 if (Ty == nullptr)
5176 return nullptr;
5177 return make<SpecialName>("typeinfo for ", Ty);
5178 }
5179 // TS <type> # typeinfo name (null-terminated byte string)
5180 case 'S': {
5181 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005182 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005183 if (Ty == nullptr)
5184 return nullptr;
5185 return make<SpecialName>("typeinfo name for ", Ty);
5186 }
5187 // Tc <call-offset> <call-offset> <base encoding>
5188 case 'c': {
5189 First += 2;
5190 if (parseCallOffset() || parseCallOffset())
5191 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005192 Node *Encoding = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00005193 if (Encoding == nullptr)
5194 return nullptr;
5195 return make<SpecialName>("covariant return thunk to ", Encoding);
5196 }
5197 // extension ::= TC <first type> <number> _ <second type>
5198 // # construction vtable for second-in-first
5199 case 'C': {
5200 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005201 Node *FirstType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005202 if (FirstType == nullptr)
5203 return nullptr;
5204 if (parseNumber(true).empty() || !consumeIf('_'))
5205 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005206 Node *SecondType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005207 if (SecondType == nullptr)
5208 return nullptr;
5209 return make<CtorVtableSpecialName>(SecondType, FirstType);
5210 }
5211 // TW <object name> # Thread-local wrapper
5212 case 'W': {
5213 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005214 Node *Name = getDerived().parseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005215 if (Name == nullptr)
5216 return nullptr;
5217 return make<SpecialName>("thread-local wrapper routine for ", Name);
5218 }
5219 // TH <object name> # Thread-local initialization
5220 case 'H': {
5221 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005222 Node *Name = getDerived().parseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005223 if (Name == nullptr)
5224 return nullptr;
5225 return make<SpecialName>("thread-local initialization routine for ", Name);
5226 }
5227 // T <call-offset> <base encoding>
5228 default: {
5229 ++First;
5230 bool IsVirt = look() == 'v';
5231 if (parseCallOffset())
5232 return nullptr;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005233 Node *BaseEncoding = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00005234 if (BaseEncoding == nullptr)
5235 return nullptr;
5236 if (IsVirt)
5237 return make<SpecialName>("virtual thunk to ", BaseEncoding);
5238 else
5239 return make<SpecialName>("non-virtual thunk to ", BaseEncoding);
5240 }
5241 }
5242 case 'G':
5243 switch (look(1)) {
5244 // GV <object name> # Guard variable for one-time initialization
5245 case 'V': {
5246 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005247 Node *Name = getDerived().parseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005248 if (Name == nullptr)
5249 return nullptr;
5250 return make<SpecialName>("guard variable for ", Name);
5251 }
5252 // GR <object name> # reference temporary for object
5253 // GR <object name> _ # First temporary
5254 // GR <object name> <seq-id> _ # Subsequent temporaries
5255 case 'R': {
5256 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005257 Node *Name = getDerived().parseName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005258 if (Name == nullptr)
5259 return nullptr;
5260 size_t Count;
5261 bool ParsedSeqId = !parseSeqId(&Count);
5262 if (!consumeIf('_') && ParsedSeqId)
5263 return nullptr;
5264 return make<SpecialName>("reference temporary for ", Name);
5265 }
5266 }
5267 }
5268 return nullptr;
5269}
5270
5271// <encoding> ::= <function name> <bare-function-type>
5272// ::= <data name>
5273// ::= <special-name>
Pavel Labathb26cabc2018-10-16 14:29:14 +00005274template <typename Derived, typename Alloc>
5275Node *AbstractManglingParser<Derived, Alloc>::parseEncoding() {
Richard Smith252bc9b2020-07-09 21:08:39 -07005276 // The template parameters of an encoding are unrelated to those of the
5277 // enclosing context.
5278 class SaveTemplateParams {
5279 AbstractManglingParser *Parser;
5280 decltype(TemplateParams) OldParams;
Justin Lebara3b738a2021-06-09 16:57:22 -07005281 decltype(OuterTemplateParams) OldOuterParams;
Richard Smith252bc9b2020-07-09 21:08:39 -07005282
5283 public:
Louis Dionne83a75212020-10-30 17:33:02 -04005284 SaveTemplateParams(AbstractManglingParser *TheParser) : Parser(TheParser) {
Richard Smith252bc9b2020-07-09 21:08:39 -07005285 OldParams = std::move(Parser->TemplateParams);
Justin Lebara3b738a2021-06-09 16:57:22 -07005286 OldOuterParams = std::move(Parser->OuterTemplateParams);
Richard Smith252bc9b2020-07-09 21:08:39 -07005287 Parser->TemplateParams.clear();
Justin Lebara3b738a2021-06-09 16:57:22 -07005288 Parser->OuterTemplateParams.clear();
Richard Smith252bc9b2020-07-09 21:08:39 -07005289 }
5290 ~SaveTemplateParams() {
5291 Parser->TemplateParams = std::move(OldParams);
Justin Lebara3b738a2021-06-09 16:57:22 -07005292 Parser->OuterTemplateParams = std::move(OldOuterParams);
Richard Smith252bc9b2020-07-09 21:08:39 -07005293 }
5294 } SaveTemplateParams(this);
Richard Smith95ed96b2020-07-09 20:36:04 -07005295
Richard Smith4344cbd2018-08-20 20:14:49 +00005296 if (look() == 'G' || look() == 'T')
Pavel Labathb26cabc2018-10-16 14:29:14 +00005297 return getDerived().parseSpecialName();
Richard Smith4344cbd2018-08-20 20:14:49 +00005298
5299 auto IsEndOfEncoding = [&] {
5300 // The set of chars that can potentially follow an <encoding> (none of which
5301 // can start a <type>). Enumerating these allows us to avoid speculative
5302 // parsing.
5303 return numLeft() == 0 || look() == 'E' || look() == '.' || look() == '_';
5304 };
5305
5306 NameState NameInfo(this);
Pavel Labathb26cabc2018-10-16 14:29:14 +00005307 Node *Name = getDerived().parseName(&NameInfo);
Richard Smith4344cbd2018-08-20 20:14:49 +00005308 if (Name == nullptr)
5309 return nullptr;
5310
5311 if (resolveForwardTemplateRefs(NameInfo))
5312 return nullptr;
5313
5314 if (IsEndOfEncoding())
5315 return Name;
5316
5317 Node *Attrs = nullptr;
5318 if (consumeIf("Ua9enable_ifI")) {
5319 size_t BeforeArgs = Names.size();
5320 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005321 Node *Arg = getDerived().parseTemplateArg();
Richard Smith4344cbd2018-08-20 20:14:49 +00005322 if (Arg == nullptr)
5323 return nullptr;
5324 Names.push_back(Arg);
5325 }
5326 Attrs = make<EnableIfAttr>(popTrailingNodeArray(BeforeArgs));
Richard Smith7918dea2018-08-24 23:30:26 +00005327 if (!Attrs)
5328 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00005329 }
5330
5331 Node *ReturnType = nullptr;
5332 if (!NameInfo.CtorDtorConversion && NameInfo.EndsWithTemplateArgs) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005333 ReturnType = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005334 if (ReturnType == nullptr)
5335 return nullptr;
5336 }
5337
5338 if (consumeIf('v'))
5339 return make<FunctionEncoding>(ReturnType, Name, NodeArray(),
5340 Attrs, NameInfo.CVQualifiers,
5341 NameInfo.ReferenceQualifier);
5342
5343 size_t ParamsBegin = Names.size();
5344 do {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005345 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005346 if (Ty == nullptr)
5347 return nullptr;
5348 Names.push_back(Ty);
5349 } while (!IsEndOfEncoding());
5350
5351 return make<FunctionEncoding>(ReturnType, Name,
5352 popTrailingNodeArray(ParamsBegin),
5353 Attrs, NameInfo.CVQualifiers,
5354 NameInfo.ReferenceQualifier);
5355}
5356
5357template <class Float>
5358struct FloatData;
5359
5360template <>
5361struct FloatData<float>
5362{
5363 static const size_t mangled_size = 8;
5364 static const size_t max_demangled_size = 24;
5365 static constexpr const char* spec = "%af";
5366};
5367
5368template <>
5369struct FloatData<double>
5370{
5371 static const size_t mangled_size = 16;
5372 static const size_t max_demangled_size = 32;
5373 static constexpr const char* spec = "%a";
5374};
5375
5376template <>
5377struct FloatData<long double>
5378{
5379#if defined(__mips__) && defined(__mips_n64) || defined(__aarch64__) || \
5380 defined(__wasm__)
5381 static const size_t mangled_size = 32;
5382#elif defined(__arm__) || defined(__mips__) || defined(__hexagon__)
5383 static const size_t mangled_size = 16;
5384#else
5385 static const size_t mangled_size = 20; // May need to be adjusted to 16 or 24 on other platforms
5386#endif
Elliott Hughesd77a6342020-04-10 17:42:00 -07005387 // `-0x1.ffffffffffffffffffffffffffffp+16383` + 'L' + '\0' == 42 bytes.
5388 // 28 'f's * 4 bits == 112 bits, which is the number of mantissa bits.
5389 // Negatives are one character longer than positives.
5390 // `0x1.` and `p` are constant, and exponents `+16383` and `-16382` are the
5391 // same length. 1 sign bit, 112 mantissa bits, and 15 exponent bits == 128.
5392 static const size_t max_demangled_size = 42;
Richard Smith4344cbd2018-08-20 20:14:49 +00005393 static constexpr const char *spec = "%LaL";
5394};
5395
Pavel Labathb26cabc2018-10-16 14:29:14 +00005396template <typename Alloc, typename Derived>
5397template <class Float>
5398Node *AbstractManglingParser<Alloc, Derived>::parseFloatingLiteral() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005399 const size_t N = FloatData<Float>::mangled_size;
5400 if (numLeft() <= N)
5401 return nullptr;
5402 StringView Data(First, First + N);
5403 for (char C : Data)
5404 if (!std::isxdigit(C))
5405 return nullptr;
5406 First += N;
5407 if (!consumeIf('E'))
5408 return nullptr;
5409 return make<FloatLiteralImpl<Float>>(Data);
5410}
5411
5412// <seq-id> ::= <0-9A-Z>+
Pavel Labathb26cabc2018-10-16 14:29:14 +00005413template <typename Alloc, typename Derived>
5414bool AbstractManglingParser<Alloc, Derived>::parseSeqId(size_t *Out) {
Richard Smith4344cbd2018-08-20 20:14:49 +00005415 if (!(look() >= '0' && look() <= '9') &&
5416 !(look() >= 'A' && look() <= 'Z'))
5417 return true;
5418
5419 size_t Id = 0;
5420 while (true) {
5421 if (look() >= '0' && look() <= '9') {
5422 Id *= 36;
5423 Id += static_cast<size_t>(look() - '0');
5424 } else if (look() >= 'A' && look() <= 'Z') {
5425 Id *= 36;
5426 Id += static_cast<size_t>(look() - 'A') + 10;
5427 } else {
5428 *Out = Id;
5429 return false;
5430 }
5431 ++First;
5432 }
5433}
5434
5435// <substitution> ::= S <seq-id> _
5436// ::= S_
5437// <substitution> ::= Sa # ::std::allocator
5438// <substitution> ::= Sb # ::std::basic_string
5439// <substitution> ::= Ss # ::std::basic_string < char,
5440// ::std::char_traits<char>,
5441// ::std::allocator<char> >
5442// <substitution> ::= Si # ::std::basic_istream<char, std::char_traits<char> >
5443// <substitution> ::= So # ::std::basic_ostream<char, std::char_traits<char> >
5444// <substitution> ::= Sd # ::std::basic_iostream<char, std::char_traits<char> >
Pavel Labathb26cabc2018-10-16 14:29:14 +00005445template <typename Derived, typename Alloc>
5446Node *AbstractManglingParser<Derived, Alloc>::parseSubstitution() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005447 if (!consumeIf('S'))
5448 return nullptr;
5449
Nathan Sidwell64e94ec2022-01-20 07:40:12 -08005450 if (look() >= 'a' && look() <= 'z') {
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005451 SpecialSubKind Kind;
Richard Smith4344cbd2018-08-20 20:14:49 +00005452 switch (look()) {
5453 case 'a':
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005454 Kind = SpecialSubKind::allocator;
Richard Smith4344cbd2018-08-20 20:14:49 +00005455 break;
5456 case 'b':
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005457 Kind = SpecialSubKind::basic_string;
Richard Smith4344cbd2018-08-20 20:14:49 +00005458 break;
5459 case 'd':
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005460 Kind = SpecialSubKind::iostream;
5461 break;
5462 case 'i':
5463 Kind = SpecialSubKind::istream;
5464 break;
5465 case 'o':
5466 Kind = SpecialSubKind::ostream;
5467 break;
5468 case 's':
5469 Kind = SpecialSubKind::string;
Richard Smith4344cbd2018-08-20 20:14:49 +00005470 break;
5471 default:
5472 return nullptr;
5473 }
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005474 ++First;
5475 auto *SpecialSub = make<SpecialSubstitution>(Kind);
Richard Smith7918dea2018-08-24 23:30:26 +00005476 if (!SpecialSub)
5477 return nullptr;
Nathan Sidwell02ad8742022-01-24 07:59:57 -08005478
Richard Smith4344cbd2018-08-20 20:14:49 +00005479 // Itanium C++ ABI 5.1.2: If a name that would use a built-in <substitution>
5480 // has ABI tags, the tags are appended to the substitution; the result is a
5481 // substitutable component.
Pavel Labathb26cabc2018-10-16 14:29:14 +00005482 Node *WithTags = getDerived().parseAbiTags(SpecialSub);
Richard Smith4344cbd2018-08-20 20:14:49 +00005483 if (WithTags != SpecialSub) {
5484 Subs.push_back(WithTags);
5485 SpecialSub = WithTags;
5486 }
5487 return SpecialSub;
5488 }
5489
5490 // ::= S_
5491 if (consumeIf('_')) {
5492 if (Subs.empty())
5493 return nullptr;
5494 return Subs[0];
5495 }
5496
5497 // ::= S <seq-id> _
5498 size_t Index = 0;
5499 if (parseSeqId(&Index))
5500 return nullptr;
5501 ++Index;
5502 if (!consumeIf('_') || Index >= Subs.size())
5503 return nullptr;
5504 return Subs[Index];
5505}
5506
5507// <template-param> ::= T_ # first template parameter
5508// ::= T <parameter-2 non-negative number> _
Richard Smithcbffe542019-09-06 23:53:21 +00005509// ::= TL <level-1> __
5510// ::= TL <level-1> _ <parameter-2 non-negative number> _
Pavel Labathb26cabc2018-10-16 14:29:14 +00005511template <typename Derived, typename Alloc>
5512Node *AbstractManglingParser<Derived, Alloc>::parseTemplateParam() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005513 if (!consumeIf('T'))
5514 return nullptr;
5515
Richard Smithcbffe542019-09-06 23:53:21 +00005516 size_t Level = 0;
5517 if (consumeIf('L')) {
5518 if (parsePositiveInteger(&Level))
5519 return nullptr;
5520 ++Level;
5521 if (!consumeIf('_'))
5522 return nullptr;
5523 }
5524
Richard Smith4344cbd2018-08-20 20:14:49 +00005525 size_t Index = 0;
5526 if (!consumeIf('_')) {
5527 if (parsePositiveInteger(&Index))
5528 return nullptr;
5529 ++Index;
5530 if (!consumeIf('_'))
5531 return nullptr;
5532 }
5533
Richard Smith4344cbd2018-08-20 20:14:49 +00005534 // If we're in a context where this <template-param> refers to a
5535 // <template-arg> further ahead in the mangled name (currently just conversion
5536 // operator types), then we should only look it up in the right context.
Richard Smithcbffe542019-09-06 23:53:21 +00005537 // This can only happen at the outermost level.
5538 if (PermitForwardTemplateReferences && Level == 0) {
Richard Smith7918dea2018-08-24 23:30:26 +00005539 Node *ForwardRef = make<ForwardTemplateReference>(Index);
5540 if (!ForwardRef)
5541 return nullptr;
5542 assert(ForwardRef->getKind() == Node::KForwardTemplateReference);
5543 ForwardTemplateRefs.push_back(
5544 static_cast<ForwardTemplateReference *>(ForwardRef));
5545 return ForwardRef;
Richard Smith4344cbd2018-08-20 20:14:49 +00005546 }
5547
Richard Smithcbffe542019-09-06 23:53:21 +00005548 if (Level >= TemplateParams.size() || !TemplateParams[Level] ||
5549 Index >= TemplateParams[Level]->size()) {
5550 // Itanium ABI 5.1.8: In a generic lambda, uses of auto in the parameter
5551 // list are mangled as the corresponding artificial template type parameter.
5552 if (ParsingLambdaParamsAtLevel == Level && Level <= TemplateParams.size()) {
5553 // This will be popped by the ScopedTemplateParamList in
5554 // parseUnnamedTypeName.
5555 if (Level == TemplateParams.size())
5556 TemplateParams.push_back(nullptr);
5557 return make<NameType>("auto");
5558 }
5559
Richard Smith4344cbd2018-08-20 20:14:49 +00005560 return nullptr;
Richard Smithcbffe542019-09-06 23:53:21 +00005561 }
5562
5563 return (*TemplateParams[Level])[Index];
5564}
5565
5566// <template-param-decl> ::= Ty # type parameter
5567// ::= Tn <type> # non-type parameter
5568// ::= Tt <template-param-decl>* E # template parameter
5569// ::= Tp <template-param-decl> # parameter pack
5570template <typename Derived, typename Alloc>
5571Node *AbstractManglingParser<Derived, Alloc>::parseTemplateParamDecl() {
5572 auto InventTemplateParamName = [&](TemplateParamKind Kind) {
5573 unsigned Index = NumSyntheticTemplateParameters[(int)Kind]++;
5574 Node *N = make<SyntheticTemplateParamName>(Kind, Index);
5575 if (N) TemplateParams.back()->push_back(N);
5576 return N;
5577 };
5578
5579 if (consumeIf("Ty")) {
5580 Node *Name = InventTemplateParamName(TemplateParamKind::Type);
5581 if (!Name)
5582 return nullptr;
5583 return make<TypeTemplateParamDecl>(Name);
5584 }
5585
5586 if (consumeIf("Tn")) {
5587 Node *Name = InventTemplateParamName(TemplateParamKind::NonType);
5588 if (!Name)
5589 return nullptr;
5590 Node *Type = parseType();
5591 if (!Type)
5592 return nullptr;
5593 return make<NonTypeTemplateParamDecl>(Name, Type);
5594 }
5595
5596 if (consumeIf("Tt")) {
5597 Node *Name = InventTemplateParamName(TemplateParamKind::Template);
5598 if (!Name)
5599 return nullptr;
5600 size_t ParamsBegin = Names.size();
5601 ScopedTemplateParamList TemplateTemplateParamParams(this);
5602 while (!consumeIf("E")) {
5603 Node *P = parseTemplateParamDecl();
5604 if (!P)
5605 return nullptr;
5606 Names.push_back(P);
5607 }
5608 NodeArray Params = popTrailingNodeArray(ParamsBegin);
5609 return make<TemplateTemplateParamDecl>(Name, Params);
5610 }
5611
5612 if (consumeIf("Tp")) {
5613 Node *P = parseTemplateParamDecl();
5614 if (!P)
5615 return nullptr;
5616 return make<TemplateParamPackDecl>(P);
5617 }
5618
5619 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00005620}
5621
5622// <template-arg> ::= <type> # type or template
5623// ::= X <expression> E # expression
5624// ::= <expr-primary> # simple expressions
5625// ::= J <template-arg>* E # argument pack
5626// ::= LZ <encoding> E # extension
Pavel Labathb26cabc2018-10-16 14:29:14 +00005627template <typename Derived, typename Alloc>
5628Node *AbstractManglingParser<Derived, Alloc>::parseTemplateArg() {
Richard Smith4344cbd2018-08-20 20:14:49 +00005629 switch (look()) {
5630 case 'X': {
5631 ++First;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005632 Node *Arg = getDerived().parseExpr();
Richard Smith4344cbd2018-08-20 20:14:49 +00005633 if (Arg == nullptr || !consumeIf('E'))
5634 return nullptr;
5635 return Arg;
5636 }
5637 case 'J': {
5638 ++First;
5639 size_t ArgsBegin = Names.size();
5640 while (!consumeIf('E')) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005641 Node *Arg = getDerived().parseTemplateArg();
Richard Smith4344cbd2018-08-20 20:14:49 +00005642 if (Arg == nullptr)
5643 return nullptr;
5644 Names.push_back(Arg);
5645 }
5646 NodeArray Args = popTrailingNodeArray(ArgsBegin);
5647 return make<TemplateArgumentPack>(Args);
5648 }
5649 case 'L': {
5650 // ::= LZ <encoding> E # extension
5651 if (look(1) == 'Z') {
5652 First += 2;
Pavel Labathb26cabc2018-10-16 14:29:14 +00005653 Node *Arg = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00005654 if (Arg == nullptr || !consumeIf('E'))
5655 return nullptr;
5656 return Arg;
5657 }
5658 // ::= <expr-primary> # simple expressions
Pavel Labathb26cabc2018-10-16 14:29:14 +00005659 return getDerived().parseExprPrimary();
Richard Smith4344cbd2018-08-20 20:14:49 +00005660 }
5661 default:
Pavel Labathb26cabc2018-10-16 14:29:14 +00005662 return getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005663 }
5664}
5665
5666// <template-args> ::= I <template-arg>* E
5667// extension, the abi says <template-arg>+
Pavel Labathb26cabc2018-10-16 14:29:14 +00005668template <typename Derived, typename Alloc>
5669Node *
5670AbstractManglingParser<Derived, Alloc>::parseTemplateArgs(bool TagTemplates) {
Richard Smith4344cbd2018-08-20 20:14:49 +00005671 if (!consumeIf('I'))
5672 return nullptr;
5673
5674 // <template-params> refer to the innermost <template-args>. Clear out any
5675 // outer args that we may have inserted into TemplateParams.
Richard Smithcbffe542019-09-06 23:53:21 +00005676 if (TagTemplates) {
Richard Smith4344cbd2018-08-20 20:14:49 +00005677 TemplateParams.clear();
Richard Smithcbffe542019-09-06 23:53:21 +00005678 TemplateParams.push_back(&OuterTemplateParams);
5679 OuterTemplateParams.clear();
5680 }
Richard Smith4344cbd2018-08-20 20:14:49 +00005681
5682 size_t ArgsBegin = Names.size();
5683 while (!consumeIf('E')) {
5684 if (TagTemplates) {
5685 auto OldParams = std::move(TemplateParams);
Pavel Labathb26cabc2018-10-16 14:29:14 +00005686 Node *Arg = getDerived().parseTemplateArg();
Richard Smith4344cbd2018-08-20 20:14:49 +00005687 TemplateParams = std::move(OldParams);
5688 if (Arg == nullptr)
5689 return nullptr;
5690 Names.push_back(Arg);
5691 Node *TableEntry = Arg;
5692 if (Arg->getKind() == Node::KTemplateArgumentPack) {
5693 TableEntry = make<ParameterPack>(
5694 static_cast<TemplateArgumentPack*>(TableEntry)->getElements());
Richard Smith7918dea2018-08-24 23:30:26 +00005695 if (!TableEntry)
5696 return nullptr;
Richard Smith4344cbd2018-08-20 20:14:49 +00005697 }
Richard Smithcbffe542019-09-06 23:53:21 +00005698 TemplateParams.back()->push_back(TableEntry);
Richard Smith4344cbd2018-08-20 20:14:49 +00005699 } else {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005700 Node *Arg = getDerived().parseTemplateArg();
Richard Smith4344cbd2018-08-20 20:14:49 +00005701 if (Arg == nullptr)
5702 return nullptr;
5703 Names.push_back(Arg);
5704 }
5705 }
5706 return make<TemplateArgs>(popTrailingNodeArray(ArgsBegin));
5707}
5708
5709// <mangled-name> ::= _Z <encoding>
5710// ::= <type>
5711// extension ::= ___Z <encoding> _block_invoke
5712// extension ::= ___Z <encoding> _block_invoke<decimal-digit>+
5713// extension ::= ___Z <encoding> _block_invoke_<decimal-digit>+
Pavel Labathb26cabc2018-10-16 14:29:14 +00005714template <typename Derived, typename Alloc>
5715Node *AbstractManglingParser<Derived, Alloc>::parse() {
Erik Pilkington18cf1982019-01-17 21:37:36 +00005716 if (consumeIf("_Z") || consumeIf("__Z")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005717 Node *Encoding = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00005718 if (Encoding == nullptr)
5719 return nullptr;
5720 if (look() == '.') {
5721 Encoding = make<DotSuffix>(Encoding, StringView(First, Last));
5722 First = Last;
5723 }
5724 if (numLeft() != 0)
5725 return nullptr;
5726 return Encoding;
5727 }
5728
Erik Pilkington18cf1982019-01-17 21:37:36 +00005729 if (consumeIf("___Z") || consumeIf("____Z")) {
Pavel Labathb26cabc2018-10-16 14:29:14 +00005730 Node *Encoding = getDerived().parseEncoding();
Richard Smith4344cbd2018-08-20 20:14:49 +00005731 if (Encoding == nullptr || !consumeIf("_block_invoke"))
5732 return nullptr;
5733 bool RequireNumber = consumeIf('_');
5734 if (parseNumber().empty() && RequireNumber)
5735 return nullptr;
5736 if (look() == '.')
5737 First = Last;
5738 if (numLeft() != 0)
5739 return nullptr;
5740 return make<SpecialName>("invocation function for block in ", Encoding);
5741 }
5742
Pavel Labathb26cabc2018-10-16 14:29:14 +00005743 Node *Ty = getDerived().parseType();
Richard Smith4344cbd2018-08-20 20:14:49 +00005744 if (numLeft() != 0)
5745 return nullptr;
5746 return Ty;
5747}
5748
Pavel Labathb26cabc2018-10-16 14:29:14 +00005749template <typename Alloc>
5750struct ManglingParser : AbstractManglingParser<ManglingParser<Alloc>, Alloc> {
5751 using AbstractManglingParser<ManglingParser<Alloc>,
5752 Alloc>::AbstractManglingParser;
5753};
5754
Erik Pilkington870950f2019-01-17 20:37:51 +00005755DEMANGLE_NAMESPACE_END
Richard Smith4344cbd2018-08-20 20:14:49 +00005756
Erik Pilkington870950f2019-01-17 20:37:51 +00005757#endif // DEMANGLE_ITANIUMDEMANGLE_H