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