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