From 2bcfcbe7956d60bb4ce609b80ee302ca8202f10a Mon Sep 17 00:00:00 2001
From: Reid Spencer <rspencer@reidspencer.com>
Date: Sat, 6 Nov 2004 08:56:40 +0000
Subject: [PATCH] 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
---
 lib/Bytecode/Reader/ReaderWrappers.cpp | 39 ++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/lib/Bytecode/Reader/ReaderWrappers.cpp b/lib/Bytecode/Reader/ReaderWrappers.cpp
index 09bc55b3361..b70c3ef4a33 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