1 | //===--- Mangle.cpp - Mangle C++ Names --------------------------*- C++ -*-===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // Implements generic name mangling support for blocks and Objective-C. |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | #include "clang/AST/Attr.h" |
13 | #include "clang/AST/ASTContext.h" |
14 | #include "clang/AST/Decl.h" |
15 | #include "clang/AST/DeclCXX.h" |
16 | #include "clang/AST/DeclObjC.h" |
17 | #include "clang/AST/DeclTemplate.h" |
18 | #include "clang/AST/ExprCXX.h" |
19 | #include "clang/AST/Mangle.h" |
20 | #include "clang/AST/VTableBuilder.h" |
21 | #include "clang/Basic/ABI.h" |
22 | #include "clang/Basic/SourceManager.h" |
23 | #include "clang/Basic/TargetInfo.h" |
24 | #include "llvm/ADT/StringExtras.h" |
25 | #include "llvm/IR/DataLayout.h" |
26 | #include "llvm/IR/Mangler.h" |
27 | #include "llvm/Support/ErrorHandling.h" |
28 | #include "llvm/Support/Format.h" |
29 | #include "llvm/Support/raw_ostream.h" |
30 | |
31 | using namespace clang; |
32 | |
33 | // FIXME: For blocks we currently mimic GCC's mangling scheme, which leaves |
34 | // much to be desired. Come up with a better mangling scheme. |
35 | |
36 | static void mangleFunctionBlock(MangleContext &Context, |
37 | StringRef Outer, |
38 | const BlockDecl *BD, |
39 | raw_ostream &Out) { |
40 | unsigned discriminator = Context.getBlockId(BD, Local: true); |
41 | if (discriminator == 0) |
42 | Out << "__" << Outer << "_block_invoke" ; |
43 | else |
44 | Out << "__" << Outer << "_block_invoke_" << discriminator+1; |
45 | } |
46 | |
47 | void MangleContext::anchor() { } |
48 | |
49 | enum CCMangling { |
50 | CCM_Other, |
51 | CCM_Fast, |
52 | CCM_RegCall, |
53 | CCM_Vector, |
54 | CCM_Std, |
55 | CCM_WasmMainArgcArgv |
56 | }; |
57 | |
58 | static bool isExternC(const NamedDecl *ND) { |
59 | if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Val: ND)) |
60 | return FD->isExternC(); |
61 | if (const VarDecl *VD = dyn_cast<VarDecl>(Val: ND)) |
62 | return VD->isExternC(); |
63 | return false; |
64 | } |
65 | |
66 | static CCMangling getCallingConvMangling(const ASTContext &Context, |
67 | const NamedDecl *ND) { |
68 | const TargetInfo &TI = Context.getTargetInfo(); |
69 | const llvm::Triple &Triple = TI.getTriple(); |
70 | |
71 | // On wasm, the argc/argv form of "main" is renamed so that the startup code |
72 | // can call it with the correct function signature. |
73 | if (Triple.isWasm()) |
74 | if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(Val: ND)) |
75 | if (FD->isMain() && FD->getNumParams() == 2) |
76 | return CCM_WasmMainArgcArgv; |
77 | |
78 | if (!Triple.isOSWindows() || !Triple.isX86()) |
79 | return CCM_Other; |
80 | |
81 | if (Context.getLangOpts().CPlusPlus && !isExternC(ND) && |
82 | TI.getCXXABI() == TargetCXXABI::Microsoft) |
83 | return CCM_Other; |
84 | |
85 | const FunctionDecl *FD = dyn_cast<FunctionDecl>(Val: ND); |
86 | if (!FD) |
87 | return CCM_Other; |
88 | QualType T = FD->getType(); |
89 | |
90 | const FunctionType *FT = T->castAs<FunctionType>(); |
91 | |
92 | CallingConv CC = FT->getCallConv(); |
93 | switch (CC) { |
94 | default: |
95 | return CCM_Other; |
96 | case CC_X86FastCall: |
97 | return CCM_Fast; |
98 | case CC_X86StdCall: |
99 | return CCM_Std; |
100 | case CC_X86VectorCall: |
101 | return CCM_Vector; |
102 | } |
103 | } |
104 | |
105 | bool MangleContext::shouldMangleDeclName(const NamedDecl *D) { |
106 | const ASTContext &ASTContext = getASTContext(); |
107 | |
108 | CCMangling CC = getCallingConvMangling(Context: ASTContext, ND: D); |
109 | if (CC != CCM_Other) |
110 | return true; |
111 | |
112 | // If the declaration has an owning module for linkage purposes that needs to |
113 | // be mangled, we must mangle its name. |
114 | if (!D->hasExternalFormalLinkage() && D->getOwningModuleForLinkage()) |
115 | return true; |
116 | |
117 | // C functions with internal linkage have to be mangled with option |
118 | // -funique-internal-linkage-names. |
119 | if (!getASTContext().getLangOpts().CPlusPlus && |
120 | isUniqueInternalLinkageDecl(ND: D)) |
121 | return true; |
122 | |
123 | // In C, functions with no attributes never need to be mangled. Fastpath them. |
124 | if (!getASTContext().getLangOpts().CPlusPlus && !D->hasAttrs()) |
125 | return false; |
126 | |
127 | // Any decl can be declared with __asm("foo") on it, and this takes precedence |
128 | // over all other naming in the .o file. |
129 | if (D->hasAttr<AsmLabelAttr>()) |
130 | return true; |
131 | |
132 | // Declarations that don't have identifier names always need to be mangled. |
133 | if (isa<MSGuidDecl>(Val: D)) |
134 | return true; |
135 | |
136 | return shouldMangleCXXName(D); |
137 | } |
138 | |
139 | void MangleContext::mangleName(GlobalDecl GD, raw_ostream &Out) { |
140 | const ASTContext &ASTContext = getASTContext(); |
141 | const NamedDecl *D = cast<NamedDecl>(Val: GD.getDecl()); |
142 | |
143 | // Any decl can be declared with __asm("foo") on it, and this takes precedence |
144 | // over all other naming in the .o file. |
145 | if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) { |
146 | // If we have an asm name, then we use it as the mangling. |
147 | |
148 | // If the label isn't literal, or if this is an alias for an LLVM intrinsic, |
149 | // do not add a "\01" prefix. |
150 | if (!ALA->getIsLiteralLabel() || ALA->getLabel().starts_with(Prefix: "llvm." )) { |
151 | Out << ALA->getLabel(); |
152 | return; |
153 | } |
154 | |
155 | // Adding the prefix can cause problems when one file has a "foo" and |
156 | // another has a "\01foo". That is known to happen on ELF with the |
157 | // tricks normally used for producing aliases (PR9177). Fortunately the |
158 | // llvm mangler on ELF is a nop, so we can just avoid adding the \01 |
159 | // marker. |
160 | StringRef UserLabelPrefix = |
161 | getASTContext().getTargetInfo().getUserLabelPrefix(); |
162 | #ifndef NDEBUG |
163 | char GlobalPrefix = |
164 | llvm::DataLayout(getASTContext().getTargetInfo().getDataLayoutString()) |
165 | .getGlobalPrefix(); |
166 | assert((UserLabelPrefix.empty() && !GlobalPrefix) || |
167 | (UserLabelPrefix.size() == 1 && UserLabelPrefix[0] == GlobalPrefix)); |
168 | #endif |
169 | if (!UserLabelPrefix.empty()) |
170 | Out << '\01'; // LLVM IR Marker for __asm("foo") |
171 | |
172 | Out << ALA->getLabel(); |
173 | return; |
174 | } |
175 | |
176 | if (auto *GD = dyn_cast<MSGuidDecl>(Val: D)) |
177 | return mangleMSGuidDecl(GD, Out); |
178 | |
179 | CCMangling CC = getCallingConvMangling(Context: ASTContext, ND: D); |
180 | |
181 | if (CC == CCM_WasmMainArgcArgv) { |
182 | Out << "__main_argc_argv" ; |
183 | return; |
184 | } |
185 | |
186 | bool MCXX = shouldMangleCXXName(D); |
187 | const TargetInfo &TI = Context.getTargetInfo(); |
188 | if (CC == CCM_Other || (MCXX && TI.getCXXABI() == TargetCXXABI::Microsoft)) { |
189 | if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(Val: D)) |
190 | mangleObjCMethodNameAsSourceName(MD: OMD, Out); |
191 | else |
192 | mangleCXXName(GD, Out); |
193 | return; |
194 | } |
195 | |
196 | Out << '\01'; |
197 | if (CC == CCM_Std) |
198 | Out << '_'; |
199 | else if (CC == CCM_Fast) |
200 | Out << '@'; |
201 | else if (CC == CCM_RegCall) { |
202 | if (getASTContext().getLangOpts().RegCall4) |
203 | Out << "__regcall4__" ; |
204 | else |
205 | Out << "__regcall3__" ; |
206 | } |
207 | |
208 | if (!MCXX) |
209 | Out << D->getIdentifier()->getName(); |
210 | else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(Val: D)) |
211 | mangleObjCMethodNameAsSourceName(MD: OMD, Out); |
212 | else |
213 | mangleCXXName(GD, Out); |
214 | |
215 | const FunctionDecl *FD = cast<FunctionDecl>(Val: D); |
216 | const FunctionType *FT = FD->getType()->castAs<FunctionType>(); |
217 | const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(Val: FT); |
218 | if (CC == CCM_Vector) |
219 | Out << '@'; |
220 | Out << '@'; |
221 | if (!Proto) { |
222 | Out << '0'; |
223 | return; |
224 | } |
225 | assert(!Proto->isVariadic()); |
226 | unsigned ArgWords = 0; |
227 | if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(Val: FD)) |
228 | if (MD->isImplicitObjectMemberFunction()) |
229 | ++ArgWords; |
230 | uint64_t DefaultPtrWidth = TI.getPointerWidth(AddrSpace: LangAS::Default); |
231 | for (const auto &AT : Proto->param_types()) { |
232 | // If an argument type is incomplete there is no way to get its size to |
233 | // correctly encode into the mangling scheme. |
234 | // Follow GCCs behaviour by simply breaking out of the loop. |
235 | if (AT->isIncompleteType()) |
236 | break; |
237 | // Size should be aligned to pointer size. |
238 | ArgWords += llvm::alignTo(Value: ASTContext.getTypeSize(T: AT), Align: DefaultPtrWidth) / |
239 | DefaultPtrWidth; |
240 | } |
241 | Out << ((DefaultPtrWidth / 8) * ArgWords); |
242 | } |
243 | |
244 | void MangleContext::mangleMSGuidDecl(const MSGuidDecl *GD, raw_ostream &Out) { |
245 | // For now, follow the MSVC naming convention for GUID objects on all |
246 | // targets. |
247 | MSGuidDecl::Parts P = GD->getParts(); |
248 | Out << llvm::format(Fmt: "_GUID_%08" PRIx32 "_%04" PRIx32 "_%04" PRIx32 "_" , |
249 | Vals: P.Part1, Vals: P.Part2, Vals: P.Part3); |
250 | unsigned I = 0; |
251 | for (uint8_t C : P.Part4And5) { |
252 | Out << llvm::format(Fmt: "%02" PRIx8, Vals: C); |
253 | if (++I == 2) |
254 | Out << "_" ; |
255 | } |
256 | } |
257 | |
258 | void MangleContext::mangleGlobalBlock(const BlockDecl *BD, |
259 | const NamedDecl *ID, |
260 | raw_ostream &Out) { |
261 | unsigned discriminator = getBlockId(BD, Local: false); |
262 | if (ID) { |
263 | if (shouldMangleDeclName(D: ID)) |
264 | mangleName(GD: ID, Out); |
265 | else { |
266 | Out << ID->getIdentifier()->getName(); |
267 | } |
268 | } |
269 | if (discriminator == 0) |
270 | Out << "_block_invoke" ; |
271 | else |
272 | Out << "_block_invoke_" << discriminator+1; |
273 | } |
274 | |
275 | void MangleContext::mangleCtorBlock(const CXXConstructorDecl *CD, |
276 | CXXCtorType CT, const BlockDecl *BD, |
277 | raw_ostream &ResStream) { |
278 | SmallString<64> Buffer; |
279 | llvm::raw_svector_ostream Out(Buffer); |
280 | mangleName(GD: GlobalDecl(CD, CT), Out); |
281 | mangleFunctionBlock(Context&: *this, Outer: Buffer, BD, Out&: ResStream); |
282 | } |
283 | |
284 | void MangleContext::mangleDtorBlock(const CXXDestructorDecl *DD, |
285 | CXXDtorType DT, const BlockDecl *BD, |
286 | raw_ostream &ResStream) { |
287 | SmallString<64> Buffer; |
288 | llvm::raw_svector_ostream Out(Buffer); |
289 | mangleName(GD: GlobalDecl(DD, DT), Out); |
290 | mangleFunctionBlock(Context&: *this, Outer: Buffer, BD, Out&: ResStream); |
291 | } |
292 | |
293 | void MangleContext::mangleBlock(const DeclContext *DC, const BlockDecl *BD, |
294 | raw_ostream &Out) { |
295 | assert(!isa<CXXConstructorDecl>(DC) && !isa<CXXDestructorDecl>(DC)); |
296 | |
297 | SmallString<64> Buffer; |
298 | llvm::raw_svector_ostream Stream(Buffer); |
299 | if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(Val: DC)) { |
300 | mangleObjCMethodNameAsSourceName(MD: Method, Stream); |
301 | } else { |
302 | assert((isa<NamedDecl>(DC) || isa<BlockDecl>(DC)) && |
303 | "expected a NamedDecl or BlockDecl" ); |
304 | for (; isa_and_nonnull<BlockDecl>(Val: DC); DC = DC->getParent()) |
305 | (void)getBlockId(BD: cast<BlockDecl>(Val: DC), Local: true); |
306 | assert((isa<TranslationUnitDecl>(DC) || isa<NamedDecl>(DC)) && |
307 | "expected a TranslationUnitDecl or a NamedDecl" ); |
308 | if (const auto *CD = dyn_cast<CXXConstructorDecl>(Val: DC)) |
309 | mangleCtorBlock(CD, /*CT*/ Ctor_Complete, BD, ResStream&: Out); |
310 | else if (const auto *DD = dyn_cast<CXXDestructorDecl>(Val: DC)) |
311 | mangleDtorBlock(DD, /*DT*/ Dtor_Complete, BD, ResStream&: Out); |
312 | else if (auto ND = dyn_cast<NamedDecl>(Val: DC)) { |
313 | if (!shouldMangleDeclName(D: ND) && ND->getIdentifier()) |
314 | Stream << ND->getIdentifier()->getName(); |
315 | else { |
316 | // FIXME: We were doing a mangleUnqualifiedName() before, but that's |
317 | // a private member of a class that will soon itself be private to the |
318 | // Itanium C++ ABI object. What should we do now? Right now, I'm just |
319 | // calling the mangleName() method on the MangleContext; is there a |
320 | // better way? |
321 | mangleName(GD: ND, Out&: Stream); |
322 | } |
323 | } |
324 | } |
325 | mangleFunctionBlock(Context&: *this, Outer: Buffer, BD, Out); |
326 | } |
327 | |
328 | void MangleContext::mangleObjCMethodName(const ObjCMethodDecl *MD, |
329 | raw_ostream &OS, |
330 | bool includePrefixByte, |
331 | bool includeCategoryNamespace) { |
332 | if (getASTContext().getLangOpts().ObjCRuntime.isGNUFamily()) { |
333 | // This is the mangling we've always used on the GNU runtimes, but it |
334 | // has obvious collisions in the face of underscores within class |
335 | // names, category names, and selectors; maybe we should improve it. |
336 | |
337 | OS << (MD->isClassMethod() ? "_c_" : "_i_" ) |
338 | << MD->getClassInterface()->getName() << '_'; |
339 | |
340 | if (includeCategoryNamespace) { |
341 | if (auto category = MD->getCategory()) |
342 | OS << category->getName(); |
343 | } |
344 | OS << '_'; |
345 | |
346 | auto selector = MD->getSelector(); |
347 | for (unsigned slotIndex = 0, |
348 | numArgs = selector.getNumArgs(), |
349 | slotEnd = std::max(a: numArgs, b: 1U); |
350 | slotIndex != slotEnd; ++slotIndex) { |
351 | if (auto name = selector.getIdentifierInfoForSlot(argIndex: slotIndex)) |
352 | OS << name->getName(); |
353 | |
354 | // Replace all the positions that would've been ':' with '_'. |
355 | // That's after each slot except that a unary selector doesn't |
356 | // end in ':'. |
357 | if (numArgs) |
358 | OS << '_'; |
359 | } |
360 | |
361 | return; |
362 | } |
363 | |
364 | // \01+[ContainerName(CategoryName) SelectorName] |
365 | if (includePrefixByte) { |
366 | OS << '\01'; |
367 | } |
368 | OS << (MD->isInstanceMethod() ? '-' : '+') << '['; |
369 | if (const auto *CID = MD->getCategory()) { |
370 | OS << CID->getClassInterface()->getName(); |
371 | if (includeCategoryNamespace) { |
372 | OS << '(' << *CID << ')'; |
373 | } |
374 | } else if (const auto *CD = |
375 | dyn_cast<ObjCContainerDecl>(Val: MD->getDeclContext())) { |
376 | OS << CD->getName(); |
377 | } else { |
378 | llvm_unreachable("Unexpected ObjC method decl context" ); |
379 | } |
380 | OS << ' '; |
381 | MD->getSelector().print(OS); |
382 | OS << ']'; |
383 | } |
384 | |
385 | void MangleContext::mangleObjCMethodNameAsSourceName(const ObjCMethodDecl *MD, |
386 | raw_ostream &Out) { |
387 | SmallString<64> Name; |
388 | llvm::raw_svector_ostream OS(Name); |
389 | |
390 | mangleObjCMethodName(MD, OS, /*includePrefixByte=*/false, |
391 | /*includeCategoryNamespace=*/true); |
392 | Out << OS.str().size() << OS.str(); |
393 | } |
394 | |
395 | class ASTNameGenerator::Implementation { |
396 | std::unique_ptr<MangleContext> MC; |
397 | llvm::DataLayout DL; |
398 | |
399 | public: |
400 | explicit Implementation(ASTContext &Ctx) |
401 | : MC(Ctx.createMangleContext()), |
402 | DL(Ctx.getTargetInfo().getDataLayoutString()) {} |
403 | |
404 | bool writeName(const Decl *D, raw_ostream &OS) { |
405 | // First apply frontend mangling. |
406 | SmallString<128> FrontendBuf; |
407 | llvm::raw_svector_ostream FrontendBufOS(FrontendBuf); |
408 | if (auto *FD = dyn_cast<FunctionDecl>(Val: D)) { |
409 | if (FD->isDependentContext()) |
410 | return true; |
411 | if (writeFuncOrVarName(D: FD, OS&: FrontendBufOS)) |
412 | return true; |
413 | } else if (auto *VD = dyn_cast<VarDecl>(Val: D)) { |
414 | if (writeFuncOrVarName(D: VD, OS&: FrontendBufOS)) |
415 | return true; |
416 | } else if (auto *MD = dyn_cast<ObjCMethodDecl>(Val: D)) { |
417 | MC->mangleObjCMethodName(MD, OS, /*includePrefixByte=*/false, |
418 | /*includeCategoryNamespace=*/true); |
419 | return false; |
420 | } else if (auto *ID = dyn_cast<ObjCInterfaceDecl>(Val: D)) { |
421 | writeObjCClassName(D: ID, OS&: FrontendBufOS); |
422 | } else { |
423 | return true; |
424 | } |
425 | |
426 | // Now apply backend mangling. |
427 | llvm::Mangler::getNameWithPrefix(OS, GVName: FrontendBufOS.str(), DL); |
428 | return false; |
429 | } |
430 | |
431 | std::string getName(const Decl *D) { |
432 | std::string Name; |
433 | { |
434 | llvm::raw_string_ostream OS(Name); |
435 | writeName(D, OS); |
436 | } |
437 | return Name; |
438 | } |
439 | |
440 | enum ObjCKind { |
441 | ObjCClass, |
442 | ObjCMetaclass, |
443 | }; |
444 | |
445 | static StringRef getClassSymbolPrefix(ObjCKind Kind, |
446 | const ASTContext &Context) { |
447 | if (Context.getLangOpts().ObjCRuntime.isGNUFamily()) |
448 | return Kind == ObjCMetaclass ? "_OBJC_METACLASS_" : "_OBJC_CLASS_" ; |
449 | return Kind == ObjCMetaclass ? "OBJC_METACLASS_$_" : "OBJC_CLASS_$_" ; |
450 | } |
451 | |
452 | std::vector<std::string> getAllManglings(const ObjCContainerDecl *OCD) { |
453 | StringRef ClassName; |
454 | if (const auto *OID = dyn_cast<ObjCInterfaceDecl>(Val: OCD)) |
455 | ClassName = OID->getObjCRuntimeNameAsString(); |
456 | else if (const auto *OID = dyn_cast<ObjCImplementationDecl>(Val: OCD)) |
457 | ClassName = OID->getObjCRuntimeNameAsString(); |
458 | |
459 | if (ClassName.empty()) |
460 | return {}; |
461 | |
462 | auto Mangle = [&](ObjCKind Kind, StringRef ClassName) -> std::string { |
463 | SmallString<40> Mangled; |
464 | auto Prefix = getClassSymbolPrefix(Kind, Context: OCD->getASTContext()); |
465 | llvm::Mangler::getNameWithPrefix(OutName&: Mangled, GVName: Prefix + ClassName, DL); |
466 | return std::string(Mangled); |
467 | }; |
468 | |
469 | return { |
470 | Mangle(ObjCClass, ClassName), |
471 | Mangle(ObjCMetaclass, ClassName), |
472 | }; |
473 | } |
474 | |
475 | std::vector<std::string> getAllManglings(const Decl *D) { |
476 | if (const auto *OCD = dyn_cast<ObjCContainerDecl>(Val: D)) |
477 | return getAllManglings(OCD); |
478 | |
479 | if (!(isa<CXXRecordDecl>(Val: D) || isa<CXXMethodDecl>(Val: D))) |
480 | return {}; |
481 | |
482 | const NamedDecl *ND = cast<NamedDecl>(Val: D); |
483 | |
484 | ASTContext &Ctx = ND->getASTContext(); |
485 | std::unique_ptr<MangleContext> M(Ctx.createMangleContext()); |
486 | |
487 | std::vector<std::string> Manglings; |
488 | |
489 | auto hasDefaultCXXMethodCC = [](ASTContext &C, const CXXMethodDecl *MD) { |
490 | auto DefaultCC = C.getDefaultCallingConvention(/*IsVariadic=*/false, |
491 | /*IsCXXMethod=*/true); |
492 | auto CC = MD->getType()->castAs<FunctionProtoType>()->getCallConv(); |
493 | return CC == DefaultCC; |
494 | }; |
495 | |
496 | if (const auto *CD = dyn_cast_or_null<CXXConstructorDecl>(Val: ND)) { |
497 | Manglings.emplace_back(args: getMangledStructor(ND: CD, StructorType: Ctor_Base)); |
498 | |
499 | if (Ctx.getTargetInfo().getCXXABI().isItaniumFamily()) |
500 | if (!CD->getParent()->isAbstract()) |
501 | Manglings.emplace_back(args: getMangledStructor(ND: CD, StructorType: Ctor_Complete)); |
502 | |
503 | if (Ctx.getTargetInfo().getCXXABI().isMicrosoft()) |
504 | if (CD->hasAttr<DLLExportAttr>() && CD->isDefaultConstructor()) |
505 | if (!(hasDefaultCXXMethodCC(Ctx, CD) && CD->getNumParams() == 0)) |
506 | Manglings.emplace_back(args: getMangledStructor(ND: CD, StructorType: Ctor_DefaultClosure)); |
507 | } else if (const auto *DD = dyn_cast_or_null<CXXDestructorDecl>(Val: ND)) { |
508 | Manglings.emplace_back(args: getMangledStructor(ND: DD, StructorType: Dtor_Base)); |
509 | if (Ctx.getTargetInfo().getCXXABI().isItaniumFamily()) { |
510 | Manglings.emplace_back(args: getMangledStructor(ND: DD, StructorType: Dtor_Complete)); |
511 | if (DD->isVirtual()) |
512 | Manglings.emplace_back(args: getMangledStructor(ND: DD, StructorType: Dtor_Deleting)); |
513 | } |
514 | } else if (const auto *MD = dyn_cast_or_null<CXXMethodDecl>(Val: ND)) { |
515 | Manglings.emplace_back(args: getName(D: ND)); |
516 | if (MD->isVirtual()) { |
517 | if (const auto *TIV = Ctx.getVTableContext()->getThunkInfo(GD: MD)) { |
518 | for (const auto &T : *TIV) { |
519 | std::string ThunkName; |
520 | std::string ContextualizedName = |
521 | getMangledThunk(MD, T, /* ElideOverrideInfo */ false); |
522 | if (Ctx.useAbbreviatedThunkName(VirtualMethodDecl: MD, MangledName: ContextualizedName)) |
523 | ThunkName = getMangledThunk(MD, T, /* ElideOverrideInfo */ true); |
524 | else |
525 | ThunkName = ContextualizedName; |
526 | Manglings.emplace_back(args&: ThunkName); |
527 | } |
528 | } |
529 | } |
530 | } |
531 | |
532 | return Manglings; |
533 | } |
534 | |
535 | private: |
536 | bool writeFuncOrVarName(const NamedDecl *D, raw_ostream &OS) { |
537 | if (MC->shouldMangleDeclName(D)) { |
538 | GlobalDecl GD; |
539 | if (const auto *CtorD = dyn_cast<CXXConstructorDecl>(Val: D)) |
540 | GD = GlobalDecl(CtorD, Ctor_Complete); |
541 | else if (const auto *DtorD = dyn_cast<CXXDestructorDecl>(Val: D)) |
542 | GD = GlobalDecl(DtorD, Dtor_Complete); |
543 | else if (D->hasAttr<CUDAGlobalAttr>()) |
544 | GD = GlobalDecl(cast<FunctionDecl>(Val: D)); |
545 | else |
546 | GD = GlobalDecl(D); |
547 | MC->mangleName(GD, Out&: OS); |
548 | return false; |
549 | } else { |
550 | IdentifierInfo *II = D->getIdentifier(); |
551 | if (!II) |
552 | return true; |
553 | OS << II->getName(); |
554 | return false; |
555 | } |
556 | } |
557 | |
558 | void writeObjCClassName(const ObjCInterfaceDecl *D, raw_ostream &OS) { |
559 | OS << getClassSymbolPrefix(Kind: ObjCClass, Context: D->getASTContext()); |
560 | OS << D->getObjCRuntimeNameAsString(); |
561 | } |
562 | |
563 | std::string getMangledStructor(const NamedDecl *ND, unsigned StructorType) { |
564 | std::string FrontendBuf; |
565 | llvm::raw_string_ostream FOS(FrontendBuf); |
566 | |
567 | GlobalDecl GD; |
568 | if (const auto *CD = dyn_cast_or_null<CXXConstructorDecl>(Val: ND)) |
569 | GD = GlobalDecl(CD, static_cast<CXXCtorType>(StructorType)); |
570 | else if (const auto *DD = dyn_cast_or_null<CXXDestructorDecl>(Val: ND)) |
571 | GD = GlobalDecl(DD, static_cast<CXXDtorType>(StructorType)); |
572 | MC->mangleName(GD, Out&: FOS); |
573 | |
574 | std::string BackendBuf; |
575 | llvm::raw_string_ostream BOS(BackendBuf); |
576 | |
577 | llvm::Mangler::getNameWithPrefix(OS&: BOS, GVName: FOS.str(), DL); |
578 | |
579 | return BOS.str(); |
580 | } |
581 | |
582 | std::string getMangledThunk(const CXXMethodDecl *MD, const ThunkInfo &T, |
583 | bool ElideOverrideInfo) { |
584 | std::string FrontendBuf; |
585 | llvm::raw_string_ostream FOS(FrontendBuf); |
586 | |
587 | MC->mangleThunk(MD, Thunk: T, ElideOverrideInfo, FOS); |
588 | |
589 | std::string BackendBuf; |
590 | llvm::raw_string_ostream BOS(BackendBuf); |
591 | |
592 | llvm::Mangler::getNameWithPrefix(OS&: BOS, GVName: FOS.str(), DL); |
593 | |
594 | return BOS.str(); |
595 | } |
596 | }; |
597 | |
598 | ASTNameGenerator::ASTNameGenerator(ASTContext &Ctx) |
599 | : Impl(std::make_unique<Implementation>(args&: Ctx)) {} |
600 | |
601 | ASTNameGenerator::~ASTNameGenerator() {} |
602 | |
603 | bool ASTNameGenerator::writeName(const Decl *D, raw_ostream &OS) { |
604 | return Impl->writeName(D, OS); |
605 | } |
606 | |
607 | std::string ASTNameGenerator::getName(const Decl *D) { |
608 | return Impl->getName(D); |
609 | } |
610 | |
611 | std::vector<std::string> ASTNameGenerator::getAllManglings(const Decl *D) { |
612 | return Impl->getAllManglings(D); |
613 | } |
614 | |