1 //===-- llvm/IR/Mangler.h - Self-contained name mangler ---------*- 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 // Unified name mangler for various backends.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #ifndef LLVM_IR_MANGLER_H
14 #define LLVM_IR_MANGLER_H
15
16 #include "llvm/ADT/DenseMap.h"
17 #include "llvm/ADT/StringRef.h"
18
19 namespace llvm {
20
21 class DataLayout;
22 class GlobalValue;
23 template <typename T> class SmallVectorImpl;
24 class Triple;
25 class Twine;
26 class raw_ostream;
27
28 class Mangler {
29 /// We need to give global values the same name every time they are mangled.
30 /// This keeps track of the number we give to anonymous ones.
31 mutable DenseMap<const GlobalValue*, unsigned> AnonGlobalIDs;
32
33 public:
34 /// Print the appropriate prefix and the specified global variable's name.
35 /// If the global variable doesn't have a name, this fills in a unique name
36 /// for the global.
37 void getNameWithPrefix(raw_ostream &OS, const GlobalValue *GV,
38 bool CannotUsePrivateLabel) const;
39 void getNameWithPrefix(SmallVectorImpl<char> &OutName, const GlobalValue *GV,
40 bool CannotUsePrivateLabel) const;
41
42 /// Print the appropriate prefix and the specified name as the global variable
43 /// name. GVName must not be empty.
44 static void getNameWithPrefix(raw_ostream &OS, const Twine &GVName,
45 const DataLayout &DL);
46 static void getNameWithPrefix(SmallVectorImpl<char> &OutName,
47 const Twine &GVName, const DataLayout &DL);
48 };
49
50 void emitLinkerFlagsForGlobalCOFF(raw_ostream &OS, const GlobalValue *GV,
51 const Triple &TT, Mangler &Mangler);
52
53 void emitLinkerFlagsForUsedCOFF(raw_ostream &OS, const GlobalValue *GV,
54 const Triple &T, Mangler &M);
55
56 std::optional<std::string> getArm64ECMangledFunctionName(StringRef Name);
57 std::optional<std::string> getArm64ECDemangledFunctionName(StringRef Name);
58
59 /// Check if an ARM64EC function name is mangled.
isArm64ECMangledFunctionName(StringRef Name)60 bool inline isArm64ECMangledFunctionName(StringRef Name) {
61 return Name[0] == '#' ||
62 (Name[0] == '?' && Name.find("@$$h") != StringRef::npos);
63 }
64
65 } // End llvm namespace
66
67 #endif
68