diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/VMCore/Module.cpp | 5 | ||||
| -rw-r--r-- | lib/VMCore/ValueSymbolTable.cpp | 8 |
2 files changed, 13 insertions, 0 deletions
diff --git a/lib/VMCore/Module.cpp b/lib/VMCore/Module.cpp index 429cf1a4c6..893a5fdcb0 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 fb7c0d8509..f527863e1b 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) { |