diff options
| -rw-r--r-- | include/llvm/Module.h | 1 | ||||
| -rw-r--r-- | include/llvm/ValueSymbolTable.h | 1 | ||||
| -rw-r--r-- | lib/VMCore/Module.cpp | 5 | ||||
| -rw-r--r-- | lib/VMCore/ValueSymbolTable.cpp | 8 |
4 files changed, 15 insertions, 0 deletions
diff --git a/include/llvm/Module.h b/include/llvm/Module.h index bce671d..56207a0 100644 --- a/include/llvm/Module.h +++ b/include/llvm/Module.h @@ -209,6 +209,7 @@ public: /// getFunction - Look up the specified function in the module symbol table. /// If it does not exist, return null. Function *getFunction(const std::string &Name) const; + Function *getFunction(const char *Name) const; /// @} /// @name Global Variable Accessors diff --git a/include/llvm/ValueSymbolTable.h b/include/llvm/ValueSymbolTable.h index 54522df..6f79f6f 100644 --- a/include/llvm/ValueSymbolTable.h +++ b/include/llvm/ValueSymbolTable.h @@ -67,6 +67,7 @@ public: /// @returns the value associated with the \p name /// @brief Lookup a named Value. Value *lookup(const std::string &name) const; + Value *lookup(const char *NameBegin, const char *NameEnd) const; /// @returns true iff the symbol table is empty /// @brief Determine if the symbol table is empty diff --git a/lib/VMCore/Module.cpp b/lib/VMCore/Module.cpp index 429cf1a..893a5fd 100644 --- a/lib/VMCore/Module.cpp +++ b/lib/VMCore/Module.cpp @@ -201,6 +201,11 @@ Function *Module::getFunction(const std::string &Name) const { return dyn_cast_or_null<Function>(SymTab.lookup(Name)); } +Function *Module::getFunction(const char *Name) const { + const ValueSymbolTable &SymTab = getValueSymbolTable(); + return dyn_cast_or_null<Function>(SymTab.lookup(Name, Name+strlen(Name))); +} + //===----------------------------------------------------------------------===// // Methods for easy access to the global variables in the module. // diff --git a/lib/VMCore/ValueSymbolTable.cpp b/lib/VMCore/ValueSymbolTable.cpp index fb7c0d8..f527863 100644 --- a/lib/VMCore/ValueSymbolTable.cpp +++ b/lib/VMCore/ValueSymbolTable.cpp @@ -54,6 +54,14 @@ Value *ValueSymbolTable::lookup(const std::string &Name) const { return 0; } +Value *ValueSymbolTable::lookup(const char *NameBegin, + const char *NameEnd) const { + const_iterator VI = vmap.find(NameBegin, NameEnd); + if (VI != vmap.end()) // We found the symbol + return VI->getValue(); + return 0; +} + // Insert a value into the symbol table with the specified name... // void ValueSymbolTable::reinsertValue(Value* V) { |