Implement the GetBytecodeSymbols interface function to extract just the

externally visible defined symbols from a bytecode file.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@17503 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 39 insertions, 0 deletions

diff --git a/lib/Bytecode/Reader/ReaderWrappers.cpp b/lib/Bytecode/Reader/ReaderWrappers.cpp
index 09bc55b..b70c3ef 100644
--- a/lib/Bytecode/Reader/ReaderWrappers.cpp
+++ b/lib/Bytecode/Reader/ReaderWrappers.cpp
@@ -336,6 +336,7 @@ bool llvm::GetBytecodeDependentLibraries(const std::string &fname,
   try {
     std::auto_ptr<ModuleProvider> AMP( getBytecodeModuleProvider(fname));
     Module* M = AMP->releaseModule();
+
     deplibs = M->getLibraries();
     delete M;
     return true;
@@ -345,4 +346,42 @@ bool llvm::GetBytecodeDependentLibraries(const std::string &fname,
   }
 }
 
+// Get just the externally visible defined symbols from the bytecode
+bool llvm::GetBytecodeSymbols(const sys::Path& fName,
+                              std::vector<std::string>& symbols) {
+  try {
+    std::auto_ptr<ModuleProvider> AMP( getBytecodeModuleProvider(fName.get()));
+
+    // Get the module from the provider
+    Module* M = AMP->releaseModule();
+
+    // Loop over global variables
+    for (Module::giterator GI = M->gbegin(), GE=M->gend(); GI != GE; ++GI) {
+      if (GI->hasInitializer()) {
+        std::string name ( GI->getName() );
+        if (!name.empty()) {
+          symbols.push_back(name);
+        }
+      }
+    }
+
+    //Loop over functions
+    for (Module::iterator FI = M->begin(), FE=M->end(); FI != FE; ++FI) {
+      if (!FI->isExternal()) {
+        std::string name ( FI->getName() );
+        if (!name.empty()) {
+          symbols.push_back(name);
+        }
+      }
+    }
+
+    // Done with the module
+    delete M;
+    return true;
+
+  } catch (...) {
+    return false;
+  }
+}
+
 // vim: sw=2 ai