[llvm-commits] CVS: llvm/lib/Bytecode/Reader/ReaderWrappers.cpp

[Reid Spencer]

Changes in directory llvm/lib/Bytecode/Reader:

ReaderWrappers.cpp updated: 1.32 -> 1.33
---
Log message:

Implement the GetBytecodeSymbols interface function to extract just the
externally visible defined symbols from a bytecode file.


---
Diffs of the changes:  (+39 -0)

Index: llvm/lib/Bytecode/Reader/ReaderWrappers.cpp
diff -u llvm/lib/Bytecode/Reader/ReaderWrappers.cpp:1.32 llvm/lib/Bytecode/Reader/ReaderWrappers.cpp:1.33
--- llvm/lib/Bytecode/Reader/ReaderWrappers.cpp:1.32	Sun Sep 12 15:56:38 2004
+++ llvm/lib/Bytecode/Reader/ReaderWrappers.cpp	Sat Nov  6 02:56:40 2004
@@ -336,6 +336,7 @@
   try {
     std::auto_ptr<ModuleProvider> AMP( getBytecodeModuleProvider(fname));
     Module* M = AMP->releaseModule();
+
     deplibs = M->getLibraries();
     delete M;
     return true;
@@ -345,4 +346,42 @@
   }
 }
 
+// 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