xref: /freebsd/contrib/llvm-project/llvm/include/llvm/IR/Module.h (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1 //===- llvm/Module.h - C++ class to represent a VM module -------*- 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 /// @file
10 /// Module.h This file contains the declarations for the Module class.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_IR_MODULE_H
15 #define LLVM_IR_MODULE_H
16 
17 #include "llvm-c/Types.h"
18 #include "llvm/ADT/STLExtras.h"
19 #include "llvm/ADT/StringMap.h"
20 #include "llvm/ADT/StringRef.h"
21 #include "llvm/ADT/iterator_range.h"
22 #include "llvm/IR/Attributes.h"
23 #include "llvm/IR/Comdat.h"
24 #include "llvm/IR/DataLayout.h"
25 #include "llvm/IR/Function.h"
26 #include "llvm/IR/GlobalAlias.h"
27 #include "llvm/IR/GlobalIFunc.h"
28 #include "llvm/IR/GlobalVariable.h"
29 #include "llvm/IR/Metadata.h"
30 #include "llvm/IR/ProfileSummary.h"
31 #include "llvm/IR/SymbolTableListTraits.h"
32 #include "llvm/Support/CBindingWrapping.h"
33 #include "llvm/Support/CodeGen.h"
34 #include <cstddef>
35 #include <cstdint>
36 #include <iterator>
37 #include <memory>
38 #include <optional>
39 #include <string>
40 #include <vector>
41 
42 namespace llvm {
43 
44 class Error;
45 class FunctionType;
46 class GVMaterializer;
47 class LLVMContext;
48 class MemoryBuffer;
49 class ModuleSummaryIndex;
50 class RandomNumberGenerator;
51 class StructType;
52 class VersionTuple;
53 
54 /// A Module instance is used to store all the information related to an
55 /// LLVM module. Modules are the top level container of all other LLVM
56 /// Intermediate Representation (IR) objects. Each module directly contains a
57 /// list of globals variables, a list of functions, a list of libraries (or
58 /// other modules) this module depends on, a symbol table, and various data
59 /// about the target's characteristics.
60 ///
61 /// A module maintains a GlobalList object that is used to hold all
62 /// constant references to global variables in the module.  When a global
63 /// variable is destroyed, it should have no entries in the GlobalList.
64 /// The main container class for the LLVM Intermediate Representation.
65 class LLVM_EXTERNAL_VISIBILITY Module {
66   /// @name Types And Enumerations
67   /// @{
68 public:
69   /// The type for the list of global variables.
70   using GlobalListType = SymbolTableList<GlobalVariable>;
71   /// The type for the list of functions.
72   using FunctionListType = SymbolTableList<Function>;
73   /// The type for the list of aliases.
74   using AliasListType = SymbolTableList<GlobalAlias>;
75   /// The type for the list of ifuncs.
76   using IFuncListType = SymbolTableList<GlobalIFunc>;
77   /// The type for the list of named metadata.
78   using NamedMDListType = ilist<NamedMDNode>;
79   /// The type of the comdat "symbol" table.
80   using ComdatSymTabType = StringMap<Comdat>;
81   /// The type for mapping names to named metadata.
82   using NamedMDSymTabType = StringMap<NamedMDNode *>;
83 
84   /// The Global Variable iterator.
85   using global_iterator = GlobalListType::iterator;
86   /// The Global Variable constant iterator.
87   using const_global_iterator = GlobalListType::const_iterator;
88 
89   /// The Function iterators.
90   using iterator = FunctionListType::iterator;
91   /// The Function constant iterator
92   using const_iterator = FunctionListType::const_iterator;
93 
94   /// The Function reverse iterator.
95   using reverse_iterator = FunctionListType::reverse_iterator;
96   /// The Function constant reverse iterator.
97   using const_reverse_iterator = FunctionListType::const_reverse_iterator;
98 
99   /// The Global Alias iterators.
100   using alias_iterator = AliasListType::iterator;
101   /// The Global Alias constant iterator
102   using const_alias_iterator = AliasListType::const_iterator;
103 
104   /// The Global IFunc iterators.
105   using ifunc_iterator = IFuncListType::iterator;
106   /// The Global IFunc constant iterator
107   using const_ifunc_iterator = IFuncListType::const_iterator;
108 
109   /// The named metadata iterators.
110   using named_metadata_iterator = NamedMDListType::iterator;
111   /// The named metadata constant iterators.
112   using const_named_metadata_iterator = NamedMDListType::const_iterator;
113 
114   /// This enumeration defines the supported behaviors of module flags.
115   enum ModFlagBehavior {
116     /// Emits an error if two values disagree, otherwise the resulting value is
117     /// that of the operands.
118     Error = 1,
119 
120     /// Emits a warning if two values disagree. The result value will be the
121     /// operand for the flag from the first module being linked.
122     Warning = 2,
123 
124     /// Adds a requirement that another module flag be present and have a
125     /// specified value after linking is performed. The value must be a metadata
126     /// pair, where the first element of the pair is the ID of the module flag
127     /// to be restricted, and the second element of the pair is the value the
128     /// module flag should be restricted to. This behavior can be used to
129     /// restrict the allowable results (via triggering of an error) of linking
130     /// IDs with the **Override** behavior.
131     Require = 3,
132 
133     /// Uses the specified value, regardless of the behavior or value of the
134     /// other module. If both modules specify **Override**, but the values
135     /// differ, an error will be emitted.
136     Override = 4,
137 
138     /// Appends the two values, which are required to be metadata nodes.
139     Append = 5,
140 
141     /// Appends the two values, which are required to be metadata
142     /// nodes. However, duplicate entries in the second list are dropped
143     /// during the append operation.
144     AppendUnique = 6,
145 
146     /// Takes the max of the two values, which are required to be integers.
147     Max = 7,
148 
149     /// Takes the min of the two values, which are required to be integers.
150     Min = 8,
151 
152     // Markers:
153     ModFlagBehaviorFirstVal = Error,
154     ModFlagBehaviorLastVal = Min
155   };
156 
157   /// Checks if Metadata represents a valid ModFlagBehavior, and stores the
158   /// converted result in MFB.
159   static bool isValidModFlagBehavior(Metadata *MD, ModFlagBehavior &MFB);
160 
161   /// Check if the given module flag metadata represents a valid module flag,
162   /// and store the flag behavior, the key string and the value metadata.
163   static bool isValidModuleFlag(const MDNode &ModFlag, ModFlagBehavior &MFB,
164                                 MDString *&Key, Metadata *&Val);
165 
166   struct ModuleFlagEntry {
167     ModFlagBehavior Behavior;
168     MDString *Key;
169     Metadata *Val;
170 
ModuleFlagEntryModuleFlagEntry171     ModuleFlagEntry(ModFlagBehavior B, MDString *K, Metadata *V)
172         : Behavior(B), Key(K), Val(V) {}
173   };
174 
175 /// @}
176 /// @name Member Variables
177 /// @{
178 private:
179   LLVMContext &Context;           ///< The LLVMContext from which types and
180                                   ///< constants are allocated.
181   GlobalListType GlobalList;      ///< The Global Variables in the module
182   FunctionListType FunctionList;  ///< The Functions in the module
183   AliasListType AliasList;        ///< The Aliases in the module
184   IFuncListType IFuncList;        ///< The IFuncs in the module
185   NamedMDListType NamedMDList;    ///< The named metadata in the module
186   std::string GlobalScopeAsm;     ///< Inline Asm at global scope.
187   std::unique_ptr<ValueSymbolTable> ValSymTab; ///< Symbol table for values
188   ComdatSymTabType ComdatSymTab;  ///< Symbol table for COMDATs
189   std::unique_ptr<MemoryBuffer>
190   OwnedMemoryBuffer;              ///< Memory buffer directly owned by this
191                                   ///< module, for legacy clients only.
192   std::unique_ptr<GVMaterializer>
193   Materializer;                   ///< Used to materialize GlobalValues
194   std::string ModuleID;           ///< Human readable identifier for the module
195   std::string SourceFileName;     ///< Original source file name for module,
196                                   ///< recorded in bitcode.
197   std::string TargetTriple;       ///< Platform target triple Module compiled on
198                                   ///< Format: (arch)(sub)-(vendor)-(sys0-(abi)
199   NamedMDSymTabType NamedMDSymTab;  ///< NamedMDNode names.
200   DataLayout DL;                  ///< DataLayout associated with the module
201   StringMap<unsigned>
202       CurrentIntrinsicIds; ///< Keep track of the current unique id count for
203                            ///< the specified intrinsic basename.
204   DenseMap<std::pair<Intrinsic::ID, const FunctionType *>, unsigned>
205       UniquedIntrinsicNames; ///< Keep track of uniqued names of intrinsics
206                              ///< based on unnamed types. The combination of
207                              ///< ID and FunctionType maps to the extension that
208                              ///< is used to make the intrinsic name unique.
209 
210   friend class Constant;
211 
212 /// @}
213 /// @name Constructors
214 /// @{
215 public:
216   /// Is this Module using intrinsics to record the position of debugging
217   /// information, or non-intrinsic records? See IsNewDbgInfoFormat in
218   /// \ref BasicBlock.
219   bool IsNewDbgInfoFormat;
220 
221   /// Used when printing this module in the new debug info format; removes all
222   /// declarations of debug intrinsics that are replaced by non-intrinsic
223   /// records in the new format.
224   void removeDebugIntrinsicDeclarations();
225 
226   /// \see BasicBlock::convertToNewDbgValues.
convertToNewDbgValues()227   void convertToNewDbgValues() {
228     for (auto &F : *this) {
229       F.convertToNewDbgValues();
230     }
231     IsNewDbgInfoFormat = true;
232   }
233 
234   /// \see BasicBlock::convertFromNewDbgValues.
convertFromNewDbgValues()235   void convertFromNewDbgValues() {
236     for (auto &F : *this) {
237       F.convertFromNewDbgValues();
238     }
239     IsNewDbgInfoFormat = false;
240   }
241 
setIsNewDbgInfoFormat(bool UseNewFormat)242   void setIsNewDbgInfoFormat(bool UseNewFormat) {
243     if (UseNewFormat && !IsNewDbgInfoFormat)
244       convertToNewDbgValues();
245     else if (!UseNewFormat && IsNewDbgInfoFormat)
246       convertFromNewDbgValues();
247   }
setNewDbgInfoFormatFlag(bool NewFlag)248   void setNewDbgInfoFormatFlag(bool NewFlag) {
249     for (auto &F : *this) {
250       F.setNewDbgInfoFormatFlag(NewFlag);
251     }
252     IsNewDbgInfoFormat = NewFlag;
253   }
254 
255   /// The Module constructor. Note that there is no default constructor. You
256   /// must provide a name for the module upon construction.
257   explicit Module(StringRef ModuleID, LLVMContext& C);
258   /// The module destructor. This will dropAllReferences.
259   ~Module();
260 
261 /// @}
262 /// @name Module Level Accessors
263 /// @{
264 
265   /// Get the module identifier which is, essentially, the name of the module.
266   /// @returns the module identifier as a string
getModuleIdentifier()267   const std::string &getModuleIdentifier() const { return ModuleID; }
268 
269   /// Returns the number of non-debug IR instructions in the module.
270   /// This is equivalent to the sum of the IR instruction counts of each
271   /// function contained in the module.
272   unsigned getInstructionCount() const;
273 
274   /// Get the module's original source file name. When compiling from
275   /// bitcode, this is taken from a bitcode record where it was recorded.
276   /// For other compiles it is the same as the ModuleID, which would
277   /// contain the source file name.
getSourceFileName()278   const std::string &getSourceFileName() const { return SourceFileName; }
279 
280   /// Get a short "name" for the module.
281   ///
282   /// This is useful for debugging or logging. It is essentially a convenience
283   /// wrapper around getModuleIdentifier().
getName()284   StringRef getName() const { return ModuleID; }
285 
286   /// Get the data layout string for the module's target platform. This is
287   /// equivalent to getDataLayout()->getStringRepresentation().
getDataLayoutStr()288   const std::string &getDataLayoutStr() const {
289     return DL.getStringRepresentation();
290   }
291 
292   /// Get the data layout for the module's target platform.
getDataLayout()293   const DataLayout &getDataLayout() const { return DL; }
294 
295   /// Get the target triple which is a string describing the target host.
296   /// @returns a string containing the target triple.
getTargetTriple()297   const std::string &getTargetTriple() const { return TargetTriple; }
298 
299   /// Get the global data context.
300   /// @returns LLVMContext - a container for LLVM's global information
getContext()301   LLVMContext &getContext() const { return Context; }
302 
303   /// Get any module-scope inline assembly blocks.
304   /// @returns a string containing the module-scope inline assembly blocks.
getModuleInlineAsm()305   const std::string &getModuleInlineAsm() const { return GlobalScopeAsm; }
306 
307   /// Get a RandomNumberGenerator salted for use with this module. The
308   /// RNG can be seeded via -rng-seed=<uint64> and is salted with the
309   /// ModuleID and the provided pass salt. The returned RNG should not
310   /// be shared across threads or passes.
311   ///
312   /// A unique RNG per pass ensures a reproducible random stream even
313   /// when other randomness consuming passes are added or removed. In
314   /// addition, the random stream will be reproducible across LLVM
315   /// versions when the pass does not change.
316   std::unique_ptr<RandomNumberGenerator> createRNG(const StringRef Name) const;
317 
318   /// Return true if size-info optimization remark is enabled, false
319   /// otherwise.
shouldEmitInstrCountChangedRemark()320   bool shouldEmitInstrCountChangedRemark() {
321     return getContext().getDiagHandlerPtr()->isAnalysisRemarkEnabled(
322         "size-info");
323   }
324 
325   /// @}
326   /// @name Module Level Mutators
327   /// @{
328 
329   /// Set the module identifier.
setModuleIdentifier(StringRef ID)330   void setModuleIdentifier(StringRef ID) { ModuleID = std::string(ID); }
331 
332   /// Set the module's original source file name.
setSourceFileName(StringRef Name)333   void setSourceFileName(StringRef Name) { SourceFileName = std::string(Name); }
334 
335   /// Set the data layout
336   void setDataLayout(StringRef Desc);
337   void setDataLayout(const DataLayout &Other);
338 
339   /// Set the target triple.
setTargetTriple(StringRef T)340   void setTargetTriple(StringRef T) { TargetTriple = std::string(T); }
341 
342   /// Set the module-scope inline assembly blocks.
343   /// A trailing newline is added if the input doesn't have one.
setModuleInlineAsm(StringRef Asm)344   void setModuleInlineAsm(StringRef Asm) {
345     GlobalScopeAsm = std::string(Asm);
346     if (!GlobalScopeAsm.empty() && GlobalScopeAsm.back() != '\n')
347       GlobalScopeAsm += '\n';
348   }
349 
350   /// Append to the module-scope inline assembly blocks.
351   /// A trailing newline is added if the input doesn't have one.
appendModuleInlineAsm(StringRef Asm)352   void appendModuleInlineAsm(StringRef Asm) {
353     GlobalScopeAsm += Asm;
354     if (!GlobalScopeAsm.empty() && GlobalScopeAsm.back() != '\n')
355       GlobalScopeAsm += '\n';
356   }
357 
358 /// @}
359 /// @name Generic Value Accessors
360 /// @{
361 
362   /// Return the global value in the module with the specified name, of
363   /// arbitrary type. This method returns null if a global with the specified
364   /// name is not found.
365   GlobalValue *getNamedValue(StringRef Name) const;
366 
367   /// Return the number of global values in the module.
368   unsigned getNumNamedValues() const;
369 
370   /// Return a unique non-zero ID for the specified metadata kind. This ID is
371   /// uniqued across modules in the current LLVMContext.
372   unsigned getMDKindID(StringRef Name) const;
373 
374   /// Populate client supplied SmallVector with the name for custom metadata IDs
375   /// registered in this LLVMContext.
376   void getMDKindNames(SmallVectorImpl<StringRef> &Result) const;
377 
378   /// Populate client supplied SmallVector with the bundle tags registered in
379   /// this LLVMContext.  The bundle tags are ordered by increasing bundle IDs.
380   /// \see LLVMContext::getOperandBundleTagID
381   void getOperandBundleTags(SmallVectorImpl<StringRef> &Result) const;
382 
383   std::vector<StructType *> getIdentifiedStructTypes() const;
384 
385   /// Return a unique name for an intrinsic whose mangling is based on an
386   /// unnamed type. The Proto represents the function prototype.
387   std::string getUniqueIntrinsicName(StringRef BaseName, Intrinsic::ID Id,
388                                      const FunctionType *Proto);
389 
390 /// @}
391 /// @name Function Accessors
392 /// @{
393 
394   /// Look up the specified function in the module symbol table. If it does not
395   /// exist, add a prototype for the function and return it. Otherwise, return
396   /// the existing function.
397   ///
398   /// In all cases, the returned value is a FunctionCallee wrapper around the
399   /// 'FunctionType *T' passed in, as well as the 'Value*' of the Function. The
400   /// function type of the function may differ from the function type stored in
401   /// FunctionCallee if it was previously created with a different type.
402   ///
403   /// Note: For library calls getOrInsertLibFunc() should be used instead.
404   FunctionCallee getOrInsertFunction(StringRef Name, FunctionType *T,
405                                      AttributeList AttributeList);
406 
407   FunctionCallee getOrInsertFunction(StringRef Name, FunctionType *T);
408 
409   /// Same as above, but takes a list of function arguments, which makes it
410   /// easier for clients to use.
411   template <typename... ArgsTy>
getOrInsertFunction(StringRef Name,AttributeList AttributeList,Type * RetTy,ArgsTy...Args)412   FunctionCallee getOrInsertFunction(StringRef Name,
413                                      AttributeList AttributeList, Type *RetTy,
414                                      ArgsTy... Args) {
415     SmallVector<Type*, sizeof...(ArgsTy)> ArgTys{Args...};
416     return getOrInsertFunction(Name,
417                                FunctionType::get(RetTy, ArgTys, false),
418                                AttributeList);
419   }
420 
421   /// Same as above, but without the attributes.
422   template <typename... ArgsTy>
getOrInsertFunction(StringRef Name,Type * RetTy,ArgsTy...Args)423   FunctionCallee getOrInsertFunction(StringRef Name, Type *RetTy,
424                                      ArgsTy... Args) {
425     return getOrInsertFunction(Name, AttributeList{}, RetTy, Args...);
426   }
427 
428   // Avoid an incorrect ordering that'd otherwise compile incorrectly.
429   template <typename... ArgsTy>
430   FunctionCallee
431   getOrInsertFunction(StringRef Name, AttributeList AttributeList,
432                       FunctionType *Invalid, ArgsTy... Args) = delete;
433 
434   /// Look up the specified function in the module symbol table. If it does not
435   /// exist, return null.
436   Function *getFunction(StringRef Name) const;
437 
438 /// @}
439 /// @name Global Variable Accessors
440 /// @{
441 
442   /// Look up the specified global variable in the module symbol table. If it
443   /// does not exist, return null. If AllowInternal is set to true, this
444   /// function will return types that have InternalLinkage. By default, these
445   /// types are not returned.
getGlobalVariable(StringRef Name)446   GlobalVariable *getGlobalVariable(StringRef Name) const {
447     return getGlobalVariable(Name, false);
448   }
449 
450   GlobalVariable *getGlobalVariable(StringRef Name, bool AllowInternal) const;
451 
452   GlobalVariable *getGlobalVariable(StringRef Name,
453                                     bool AllowInternal = false) {
454     return static_cast<const Module *>(this)->getGlobalVariable(Name,
455                                                                 AllowInternal);
456   }
457 
458   /// Return the global variable in the module with the specified name, of
459   /// arbitrary type. This method returns null if a global with the specified
460   /// name is not found.
getNamedGlobal(StringRef Name)461   const GlobalVariable *getNamedGlobal(StringRef Name) const {
462     return getGlobalVariable(Name, true);
463   }
getNamedGlobal(StringRef Name)464   GlobalVariable *getNamedGlobal(StringRef Name) {
465     return const_cast<GlobalVariable *>(
466                        static_cast<const Module *>(this)->getNamedGlobal(Name));
467   }
468 
469   /// Look up the specified global in the module symbol table.
470   /// If it does not exist, invoke a callback to create a declaration of the
471   /// global and return it. The global is constantexpr casted to the expected
472   /// type if necessary.
473   Constant *
474   getOrInsertGlobal(StringRef Name, Type *Ty,
475                     function_ref<GlobalVariable *()> CreateGlobalCallback);
476 
477   /// Look up the specified global in the module symbol table. If required, this
478   /// overload constructs the global variable using its constructor's defaults.
479   Constant *getOrInsertGlobal(StringRef Name, Type *Ty);
480 
481 /// @}
482 /// @name Global Alias Accessors
483 /// @{
484 
485   /// Return the global alias in the module with the specified name, of
486   /// arbitrary type. This method returns null if a global with the specified
487   /// name is not found.
488   GlobalAlias *getNamedAlias(StringRef Name) const;
489 
490 /// @}
491 /// @name Global IFunc Accessors
492 /// @{
493 
494   /// Return the global ifunc in the module with the specified name, of
495   /// arbitrary type. This method returns null if a global with the specified
496   /// name is not found.
497   GlobalIFunc *getNamedIFunc(StringRef Name) const;
498 
499 /// @}
500 /// @name Named Metadata Accessors
501 /// @{
502 
503   /// Return the first NamedMDNode in the module with the specified name. This
504   /// method returns null if a NamedMDNode with the specified name is not found.
505   NamedMDNode *getNamedMetadata(const Twine &Name) const;
506 
507   /// Return the named MDNode in the module with the specified name. This method
508   /// returns a new NamedMDNode if a NamedMDNode with the specified name is not
509   /// found.
510   NamedMDNode *getOrInsertNamedMetadata(StringRef Name);
511 
512   /// Remove the given NamedMDNode from this module and delete it.
513   void eraseNamedMetadata(NamedMDNode *NMD);
514 
515 /// @}
516 /// @name Comdat Accessors
517 /// @{
518 
519   /// Return the Comdat in the module with the specified name. It is created
520   /// if it didn't already exist.
521   Comdat *getOrInsertComdat(StringRef Name);
522 
523 /// @}
524 /// @name Module Flags Accessors
525 /// @{
526 
527   /// Returns the module flags in the provided vector.
528   void getModuleFlagsMetadata(SmallVectorImpl<ModuleFlagEntry> &Flags) const;
529 
530   /// Return the corresponding value if Key appears in module flags, otherwise
531   /// return null.
532   Metadata *getModuleFlag(StringRef Key) const;
533 
534   /// Returns the NamedMDNode in the module that represents module-level flags.
535   /// This method returns null if there are no module-level flags.
536   NamedMDNode *getModuleFlagsMetadata() const;
537 
538   /// Returns the NamedMDNode in the module that represents module-level flags.
539   /// If module-level flags aren't found, it creates the named metadata that
540   /// contains them.
541   NamedMDNode *getOrInsertModuleFlagsMetadata();
542 
543   /// Add a module-level flag to the module-level flags metadata. It will create
544   /// the module-level flags named metadata if it doesn't already exist.
545   void addModuleFlag(ModFlagBehavior Behavior, StringRef Key, Metadata *Val);
546   void addModuleFlag(ModFlagBehavior Behavior, StringRef Key, Constant *Val);
547   void addModuleFlag(ModFlagBehavior Behavior, StringRef Key, uint32_t Val);
548   void addModuleFlag(MDNode *Node);
549   /// Like addModuleFlag but replaces the old module flag if it already exists.
550   void setModuleFlag(ModFlagBehavior Behavior, StringRef Key, Metadata *Val);
551   void setModuleFlag(ModFlagBehavior Behavior, StringRef Key, Constant *Val);
552   void setModuleFlag(ModFlagBehavior Behavior, StringRef Key, uint32_t Val);
553 
554   /// @}
555   /// @name Materialization
556   /// @{
557 
558   /// Sets the GVMaterializer to GVM. This module must not yet have a
559   /// Materializer. To reset the materializer for a module that already has one,
560   /// call materializeAll first. Destroying this module will destroy
561   /// its materializer without materializing any more GlobalValues. Without
562   /// destroying the Module, there is no way to detach or destroy a materializer
563   /// without materializing all the GVs it controls, to avoid leaving orphan
564   /// unmaterialized GVs.
565   void setMaterializer(GVMaterializer *GVM);
566   /// Retrieves the GVMaterializer, if any, for this Module.
getMaterializer()567   GVMaterializer *getMaterializer() const { return Materializer.get(); }
isMaterialized()568   bool isMaterialized() const { return !getMaterializer(); }
569 
570   /// Make sure the GlobalValue is fully read.
571   llvm::Error materialize(GlobalValue *GV);
572 
573   /// Make sure all GlobalValues in this Module are fully read and clear the
574   /// Materializer.
575   llvm::Error materializeAll();
576 
577   llvm::Error materializeMetadata();
578 
579   /// Detach global variable \p GV from the list but don't delete it.
removeGlobalVariable(GlobalVariable * GV)580   void removeGlobalVariable(GlobalVariable *GV) { GlobalList.remove(GV); }
581   /// Remove global variable \p GV from the list and delete it.
eraseGlobalVariable(GlobalVariable * GV)582   void eraseGlobalVariable(GlobalVariable *GV) { GlobalList.erase(GV); }
583   /// Insert global variable \p GV at the end of the global variable list and
584   /// take ownership.
insertGlobalVariable(GlobalVariable * GV)585   void insertGlobalVariable(GlobalVariable *GV) {
586     insertGlobalVariable(GlobalList.end(), GV);
587   }
588   /// Insert global variable \p GV into the global variable list before \p
589   /// Where and take ownership.
insertGlobalVariable(GlobalListType::iterator Where,GlobalVariable * GV)590   void insertGlobalVariable(GlobalListType::iterator Where, GlobalVariable *GV) {
591     GlobalList.insert(Where, GV);
592   }
593   // Use global_size() to get the total number of global variables.
594   // Use globals() to get the range of all global variables.
595 
596 private:
597 /// @}
598 /// @name Direct access to the globals list, functions list, and symbol table
599 /// @{
600 
601   /// Get the Module's list of global variables (constant).
getGlobalList()602   const GlobalListType   &getGlobalList() const       { return GlobalList; }
603   /// Get the Module's list of global variables.
getGlobalList()604   GlobalListType         &getGlobalList()             { return GlobalList; }
605 
getSublistAccess(GlobalVariable *)606   static GlobalListType Module::*getSublistAccess(GlobalVariable*) {
607     return &Module::GlobalList;
608   }
609   friend class llvm::SymbolTableListTraits<llvm::GlobalVariable>;
610 
611 public:
612   /// Get the Module's list of functions (constant).
getFunctionList()613   const FunctionListType &getFunctionList() const     { return FunctionList; }
614   /// Get the Module's list of functions.
getFunctionList()615   FunctionListType       &getFunctionList()           { return FunctionList; }
getSublistAccess(Function *)616   static FunctionListType Module::*getSublistAccess(Function*) {
617     return &Module::FunctionList;
618   }
619 
620   /// Detach \p Alias from the list but don't delete it.
removeAlias(GlobalAlias * Alias)621   void removeAlias(GlobalAlias *Alias) { AliasList.remove(Alias); }
622   /// Remove \p Alias from the list and delete it.
eraseAlias(GlobalAlias * Alias)623   void eraseAlias(GlobalAlias *Alias) { AliasList.erase(Alias); }
624   /// Insert \p Alias at the end of the alias list and take ownership.
insertAlias(GlobalAlias * Alias)625   void insertAlias(GlobalAlias *Alias) { AliasList.insert(AliasList.end(), Alias); }
626   // Use alias_size() to get the size of AliasList.
627   // Use aliases() to get a range of all Alias objects in AliasList.
628 
629   /// Detach \p IFunc from the list but don't delete it.
removeIFunc(GlobalIFunc * IFunc)630   void removeIFunc(GlobalIFunc *IFunc) { IFuncList.remove(IFunc); }
631   /// Remove \p IFunc from the list and delete it.
eraseIFunc(GlobalIFunc * IFunc)632   void eraseIFunc(GlobalIFunc *IFunc) { IFuncList.erase(IFunc); }
633   /// Insert \p IFunc at the end of the alias list and take ownership.
insertIFunc(GlobalIFunc * IFunc)634   void insertIFunc(GlobalIFunc *IFunc) { IFuncList.push_back(IFunc); }
635   // Use ifunc_size() to get the number of functions in IFuncList.
636   // Use ifuncs() to get the range of all IFuncs.
637 
638   /// Detach \p MDNode from the list but don't delete it.
removeNamedMDNode(NamedMDNode * MDNode)639   void removeNamedMDNode(NamedMDNode *MDNode) { NamedMDList.remove(MDNode); }
640   /// Remove \p MDNode from the list and delete it.
eraseNamedMDNode(NamedMDNode * MDNode)641   void eraseNamedMDNode(NamedMDNode *MDNode) { NamedMDList.erase(MDNode); }
642   /// Insert \p MDNode at the end of the alias list and take ownership.
insertNamedMDNode(NamedMDNode * MDNode)643   void insertNamedMDNode(NamedMDNode *MDNode) {
644     NamedMDList.push_back(MDNode);
645   }
646   // Use named_metadata_size() to get the size of the named meatadata list.
647   // Use named_metadata() to get the range of all named metadata.
648 
649 private: // Please use functions like insertAlias(), removeAlias() etc.
650   /// Get the Module's list of aliases (constant).
getAliasList()651   const AliasListType    &getAliasList() const        { return AliasList; }
652   /// Get the Module's list of aliases.
getAliasList()653   AliasListType          &getAliasList()              { return AliasList; }
654 
getSublistAccess(GlobalAlias *)655   static AliasListType Module::*getSublistAccess(GlobalAlias*) {
656     return &Module::AliasList;
657   }
658   friend class llvm::SymbolTableListTraits<llvm::GlobalAlias>;
659 
660   /// Get the Module's list of ifuncs (constant).
getIFuncList()661   const IFuncListType    &getIFuncList() const        { return IFuncList; }
662   /// Get the Module's list of ifuncs.
getIFuncList()663   IFuncListType          &getIFuncList()              { return IFuncList; }
664 
getSublistAccess(GlobalIFunc *)665   static IFuncListType Module::*getSublistAccess(GlobalIFunc*) {
666     return &Module::IFuncList;
667   }
668   friend class llvm::SymbolTableListTraits<llvm::GlobalIFunc>;
669 
670   /// Get the Module's list of named metadata (constant).
getNamedMDList()671   const NamedMDListType  &getNamedMDList() const      { return NamedMDList; }
672   /// Get the Module's list of named metadata.
getNamedMDList()673   NamedMDListType        &getNamedMDList()            { return NamedMDList; }
674 
getSublistAccess(NamedMDNode *)675   static NamedMDListType Module::*getSublistAccess(NamedMDNode*) {
676     return &Module::NamedMDList;
677   }
678 
679 public:
680   /// Get the symbol table of global variable and function identifiers
getValueSymbolTable()681   const ValueSymbolTable &getValueSymbolTable() const { return *ValSymTab; }
682   /// Get the Module's symbol table of global variable and function identifiers.
getValueSymbolTable()683   ValueSymbolTable       &getValueSymbolTable()       { return *ValSymTab; }
684 
685   /// Get the Module's symbol table for COMDATs (constant).
getComdatSymbolTable()686   const ComdatSymTabType &getComdatSymbolTable() const { return ComdatSymTab; }
687   /// Get the Module's symbol table for COMDATs.
getComdatSymbolTable()688   ComdatSymTabType &getComdatSymbolTable() { return ComdatSymTab; }
689 
690 /// @}
691 /// @name Global Variable Iteration
692 /// @{
693 
global_begin()694   global_iterator       global_begin()       { return GlobalList.begin(); }
global_begin()695   const_global_iterator global_begin() const { return GlobalList.begin(); }
global_end()696   global_iterator       global_end  ()       { return GlobalList.end(); }
global_end()697   const_global_iterator global_end  () const { return GlobalList.end(); }
global_size()698   size_t                global_size () const { return GlobalList.size(); }
global_empty()699   bool                  global_empty() const { return GlobalList.empty(); }
700 
globals()701   iterator_range<global_iterator> globals() {
702     return make_range(global_begin(), global_end());
703   }
globals()704   iterator_range<const_global_iterator> globals() const {
705     return make_range(global_begin(), global_end());
706   }
707 
708 /// @}
709 /// @name Function Iteration
710 /// @{
711 
begin()712   iterator                begin()       { return FunctionList.begin(); }
begin()713   const_iterator          begin() const { return FunctionList.begin(); }
end()714   iterator                end  ()       { return FunctionList.end();   }
end()715   const_iterator          end  () const { return FunctionList.end();   }
rbegin()716   reverse_iterator        rbegin()      { return FunctionList.rbegin(); }
rbegin()717   const_reverse_iterator  rbegin() const{ return FunctionList.rbegin(); }
rend()718   reverse_iterator        rend()        { return FunctionList.rend(); }
rend()719   const_reverse_iterator  rend() const  { return FunctionList.rend(); }
size()720   size_t                  size() const  { return FunctionList.size(); }
empty()721   bool                    empty() const { return FunctionList.empty(); }
722 
functions()723   iterator_range<iterator> functions() {
724     return make_range(begin(), end());
725   }
functions()726   iterator_range<const_iterator> functions() const {
727     return make_range(begin(), end());
728   }
729 
730 /// @}
731 /// @name Alias Iteration
732 /// @{
733 
alias_begin()734   alias_iterator       alias_begin()            { return AliasList.begin(); }
alias_begin()735   const_alias_iterator alias_begin() const      { return AliasList.begin(); }
alias_end()736   alias_iterator       alias_end  ()            { return AliasList.end();   }
alias_end()737   const_alias_iterator alias_end  () const      { return AliasList.end();   }
alias_size()738   size_t               alias_size () const      { return AliasList.size();  }
alias_empty()739   bool                 alias_empty() const      { return AliasList.empty(); }
740 
aliases()741   iterator_range<alias_iterator> aliases() {
742     return make_range(alias_begin(), alias_end());
743   }
aliases()744   iterator_range<const_alias_iterator> aliases() const {
745     return make_range(alias_begin(), alias_end());
746   }
747 
748 /// @}
749 /// @name IFunc Iteration
750 /// @{
751 
ifunc_begin()752   ifunc_iterator       ifunc_begin()            { return IFuncList.begin(); }
ifunc_begin()753   const_ifunc_iterator ifunc_begin() const      { return IFuncList.begin(); }
ifunc_end()754   ifunc_iterator       ifunc_end  ()            { return IFuncList.end();   }
ifunc_end()755   const_ifunc_iterator ifunc_end  () const      { return IFuncList.end();   }
ifunc_size()756   size_t               ifunc_size () const      { return IFuncList.size();  }
ifunc_empty()757   bool                 ifunc_empty() const      { return IFuncList.empty(); }
758 
ifuncs()759   iterator_range<ifunc_iterator> ifuncs() {
760     return make_range(ifunc_begin(), ifunc_end());
761   }
ifuncs()762   iterator_range<const_ifunc_iterator> ifuncs() const {
763     return make_range(ifunc_begin(), ifunc_end());
764   }
765 
766   /// @}
767   /// @name Convenience iterators
768   /// @{
769 
770   using global_object_iterator =
771       concat_iterator<GlobalObject, iterator, global_iterator>;
772   using const_global_object_iterator =
773       concat_iterator<const GlobalObject, const_iterator,
774                       const_global_iterator>;
775 
776   iterator_range<global_object_iterator> global_objects();
777   iterator_range<const_global_object_iterator> global_objects() const;
778 
779   using global_value_iterator =
780       concat_iterator<GlobalValue, iterator, global_iterator, alias_iterator,
781                       ifunc_iterator>;
782   using const_global_value_iterator =
783       concat_iterator<const GlobalValue, const_iterator, const_global_iterator,
784                       const_alias_iterator, const_ifunc_iterator>;
785 
786   iterator_range<global_value_iterator> global_values();
787   iterator_range<const_global_value_iterator> global_values() const;
788 
789   /// @}
790   /// @name Named Metadata Iteration
791   /// @{
792 
named_metadata_begin()793   named_metadata_iterator named_metadata_begin() { return NamedMDList.begin(); }
named_metadata_begin()794   const_named_metadata_iterator named_metadata_begin() const {
795     return NamedMDList.begin();
796   }
797 
named_metadata_end()798   named_metadata_iterator named_metadata_end() { return NamedMDList.end(); }
named_metadata_end()799   const_named_metadata_iterator named_metadata_end() const {
800     return NamedMDList.end();
801   }
802 
named_metadata_size()803   size_t named_metadata_size() const { return NamedMDList.size();  }
named_metadata_empty()804   bool named_metadata_empty() const { return NamedMDList.empty(); }
805 
named_metadata()806   iterator_range<named_metadata_iterator> named_metadata() {
807     return make_range(named_metadata_begin(), named_metadata_end());
808   }
named_metadata()809   iterator_range<const_named_metadata_iterator> named_metadata() const {
810     return make_range(named_metadata_begin(), named_metadata_end());
811   }
812 
813   /// An iterator for DICompileUnits that skips those marked NoDebug.
814   class debug_compile_units_iterator {
815     NamedMDNode *CUs;
816     unsigned Idx;
817 
818     void SkipNoDebugCUs();
819 
820   public:
821     using iterator_category = std::input_iterator_tag;
822     using value_type = DICompileUnit *;
823     using difference_type = std::ptrdiff_t;
824     using pointer = value_type *;
825     using reference = value_type &;
826 
debug_compile_units_iterator(NamedMDNode * CUs,unsigned Idx)827     explicit debug_compile_units_iterator(NamedMDNode *CUs, unsigned Idx)
828         : CUs(CUs), Idx(Idx) {
829       SkipNoDebugCUs();
830     }
831 
832     debug_compile_units_iterator &operator++() {
833       ++Idx;
834       SkipNoDebugCUs();
835       return *this;
836     }
837 
838     debug_compile_units_iterator operator++(int) {
839       debug_compile_units_iterator T(*this);
840       ++Idx;
841       return T;
842     }
843 
844     bool operator==(const debug_compile_units_iterator &I) const {
845       return Idx == I.Idx;
846     }
847 
848     bool operator!=(const debug_compile_units_iterator &I) const {
849       return Idx != I.Idx;
850     }
851 
852     DICompileUnit *operator*() const;
853     DICompileUnit *operator->() const;
854   };
855 
debug_compile_units_begin()856   debug_compile_units_iterator debug_compile_units_begin() const {
857     auto *CUs = getNamedMetadata("llvm.dbg.cu");
858     return debug_compile_units_iterator(CUs, 0);
859   }
860 
debug_compile_units_end()861   debug_compile_units_iterator debug_compile_units_end() const {
862     auto *CUs = getNamedMetadata("llvm.dbg.cu");
863     return debug_compile_units_iterator(CUs, CUs ? CUs->getNumOperands() : 0);
864   }
865 
866   /// Return an iterator for all DICompileUnits listed in this Module's
867   /// llvm.dbg.cu named metadata node and aren't explicitly marked as
868   /// NoDebug.
debug_compile_units()869   iterator_range<debug_compile_units_iterator> debug_compile_units() const {
870     auto *CUs = getNamedMetadata("llvm.dbg.cu");
871     return make_range(
872         debug_compile_units_iterator(CUs, 0),
873         debug_compile_units_iterator(CUs, CUs ? CUs->getNumOperands() : 0));
874   }
875 /// @}
876 
877   /// Destroy ConstantArrays in LLVMContext if they are not used.
878   /// ConstantArrays constructed during linking can cause quadratic memory
879   /// explosion. Releasing all unused constants can cause a 20% LTO compile-time
880   /// slowdown for a large application.
881   ///
882   /// NOTE: Constants are currently owned by LLVMContext. This can then only
883   /// be called where all uses of the LLVMContext are understood.
884   void dropTriviallyDeadConstantArrays();
885 
886 /// @name Utility functions for printing and dumping Module objects
887 /// @{
888 
889   /// Print the module to an output stream with an optional
890   /// AssemblyAnnotationWriter.  If \c ShouldPreserveUseListOrder, then include
891   /// uselistorder directives so that use-lists can be recreated when reading
892   /// the assembly.
893   void print(raw_ostream &OS, AssemblyAnnotationWriter *AAW,
894              bool ShouldPreserveUseListOrder = false,
895              bool IsForDebug = false) const;
896 
897   /// Dump the module to stderr (for debugging).
898   void dump() const;
899 
900   /// This function causes all the subinstructions to "let go" of all references
901   /// that they are maintaining.  This allows one to 'delete' a whole class at
902   /// a time, even though there may be circular references... first all
903   /// references are dropped, and all use counts go to zero.  Then everything
904   /// is delete'd for real.  Note that no operations are valid on an object
905   /// that has "dropped all references", except operator delete.
906   void dropAllReferences();
907 
908 /// @}
909 /// @name Utility functions for querying Debug information.
910 /// @{
911 
912   /// Returns the Number of Register ParametersDwarf Version by checking
913   /// module flags.
914   unsigned getNumberRegisterParameters() const;
915 
916   /// Returns the Dwarf Version by checking module flags.
917   unsigned getDwarfVersion() const;
918 
919   /// Returns the DWARF format by checking module flags.
920   bool isDwarf64() const;
921 
922   /// Returns the CodeView Version by checking module flags.
923   /// Returns zero if not present in module.
924   unsigned getCodeViewFlag() const;
925 
926 /// @}
927 /// @name Utility functions for querying and setting PIC level
928 /// @{
929 
930   /// Returns the PIC level (small or large model)
931   PICLevel::Level getPICLevel() const;
932 
933   /// Set the PIC level (small or large model)
934   void setPICLevel(PICLevel::Level PL);
935 /// @}
936 
937 /// @}
938 /// @name Utility functions for querying and setting PIE level
939 /// @{
940 
941   /// Returns the PIE level (small or large model)
942   PIELevel::Level getPIELevel() const;
943 
944   /// Set the PIE level (small or large model)
945   void setPIELevel(PIELevel::Level PL);
946 /// @}
947 
948   /// @}
949   /// @name Utility function for querying and setting code model
950   /// @{
951 
952   /// Returns the code model (tiny, small, kernel, medium or large model)
953   std::optional<CodeModel::Model> getCodeModel() const;
954 
955   /// Set the code model (tiny, small, kernel, medium or large)
956   void setCodeModel(CodeModel::Model CL);
957   /// @}
958 
959   /// @}
960   /// @name Utility function for querying and setting the large data threshold
961   /// @{
962 
963   /// Returns the code model (tiny, small, kernel, medium or large model)
964   std::optional<uint64_t> getLargeDataThreshold() const;
965 
966   /// Set the code model (tiny, small, kernel, medium or large)
967   void setLargeDataThreshold(uint64_t Threshold);
968   /// @}
969 
970   /// @name Utility functions for querying and setting PGO summary
971   /// @{
972 
973   /// Attach profile summary metadata to this module.
974   void setProfileSummary(Metadata *M, ProfileSummary::Kind Kind);
975 
976   /// Returns profile summary metadata. When IsCS is true, use the context
977   /// sensitive profile summary.
978   Metadata *getProfileSummary(bool IsCS) const;
979   /// @}
980 
981   /// Returns whether semantic interposition is to be respected.
982   bool getSemanticInterposition() const;
983 
984   /// Set whether semantic interposition is to be respected.
985   void setSemanticInterposition(bool);
986 
987   /// Returns true if PLT should be avoided for RTLib calls.
988   bool getRtLibUseGOT() const;
989 
990   /// Set that PLT should be avoid for RTLib calls.
991   void setRtLibUseGOT();
992 
993   /// Get/set whether referencing global variables can use direct access
994   /// relocations on ELF targets.
995   bool getDirectAccessExternalData() const;
996   void setDirectAccessExternalData(bool Value);
997 
998   /// Get/set whether synthesized functions should get the uwtable attribute.
999   UWTableKind getUwtable() const;
1000   void setUwtable(UWTableKind Kind);
1001 
1002   /// Get/set whether synthesized functions should get the "frame-pointer"
1003   /// attribute.
1004   FramePointerKind getFramePointer() const;
1005   void setFramePointer(FramePointerKind Kind);
1006 
1007   /// Get/set what kind of stack protector guard to use.
1008   StringRef getStackProtectorGuard() const;
1009   void setStackProtectorGuard(StringRef Kind);
1010 
1011   /// Get/set which register to use as the stack protector guard register. The
1012   /// empty string is equivalent to "global". Other values may be "tls" or
1013   /// "sysreg".
1014   StringRef getStackProtectorGuardReg() const;
1015   void setStackProtectorGuardReg(StringRef Reg);
1016 
1017   /// Get/set a symbol to use as the stack protector guard.
1018   StringRef getStackProtectorGuardSymbol() const;
1019   void setStackProtectorGuardSymbol(StringRef Symbol);
1020 
1021   /// Get/set what offset from the stack protector to use.
1022   int getStackProtectorGuardOffset() const;
1023   void setStackProtectorGuardOffset(int Offset);
1024 
1025   /// Get/set the stack alignment overridden from the default.
1026   unsigned getOverrideStackAlignment() const;
1027   void setOverrideStackAlignment(unsigned Align);
1028 
1029   unsigned getMaxTLSAlignment() const;
1030 
1031   /// @name Utility functions for querying and setting the build SDK version
1032   /// @{
1033 
1034   /// Attach a build SDK version metadata to this module.
1035   void setSDKVersion(const VersionTuple &V);
1036 
1037   /// Get the build SDK version metadata.
1038   ///
1039   /// An empty version is returned if no such metadata is attached.
1040   VersionTuple getSDKVersion() const;
1041   /// @}
1042 
1043   /// Take ownership of the given memory buffer.
1044   void setOwnedMemoryBuffer(std::unique_ptr<MemoryBuffer> MB);
1045 
1046   /// Set the partial sample profile ratio in the profile summary module flag,
1047   /// if applicable.
1048   void setPartialSampleProfileRatio(const ModuleSummaryIndex &Index);
1049 
1050   /// Get the target variant triple which is a string describing a variant of
1051   /// the target host platform. For example, Mac Catalyst can be a variant
1052   /// target triple for a macOS target.
1053   /// @returns a string containing the target variant triple.
1054   StringRef getDarwinTargetVariantTriple() const;
1055 
1056   /// Set the target variant triple which is a string describing a variant of
1057   /// the target host platform.
1058   void setDarwinTargetVariantTriple(StringRef T);
1059 
1060   /// Get the target variant version build SDK version metadata.
1061   ///
1062   /// An empty version is returned if no such metadata is attached.
1063   VersionTuple getDarwinTargetVariantSDKVersion() const;
1064 
1065   /// Set the target variant version build SDK version metadata.
1066   void setDarwinTargetVariantSDKVersion(VersionTuple Version);
1067 };
1068 
1069 /// Given "llvm.used" or "llvm.compiler.used" as a global name, collect the
1070 /// initializer elements of that global in a SmallVector and return the global
1071 /// itself.
1072 GlobalVariable *collectUsedGlobalVariables(const Module &M,
1073                                            SmallVectorImpl<GlobalValue *> &Vec,
1074                                            bool CompilerUsed);
1075 
1076 /// An raw_ostream inserter for modules.
1077 inline raw_ostream &operator<<(raw_ostream &O, const Module &M) {
1078   M.print(O, nullptr);
1079   return O;
1080 }
1081 
1082 // Create wrappers for C Binding types (see CBindingWrapping.h).
DEFINE_SIMPLE_CONVERSION_FUNCTIONS(Module,LLVMModuleRef)1083 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(Module, LLVMModuleRef)
1084 
1085 /* LLVMModuleProviderRef exists for historical reasons, but now just holds a
1086  * Module.
1087  */
1088 inline Module *unwrap(LLVMModuleProviderRef MP) {
1089   return reinterpret_cast<Module*>(MP);
1090 }
1091 
1092 } // end namespace llvm
1093 
1094 #endif // LLVM_IR_MODULE_H
1095