Implement an isBytecodeArchive method to determine if an archive contains
bytecode file members or not.
Patch Contributed By Adam Treat
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@20338 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/include/llvm/Bytecode/Archive.h b/include/llvm/Bytecode/Archive.h
index 102bb5c..3805d0b 100644
--- a/include/llvm/Bytecode/Archive.h
+++ b/include/llvm/Bytecode/Archive.h
@@ -414,6 +414,15 @@
std::set<std::string>& symbols, ///< Symbols to be sought
std::set<ModuleProvider*>& modules ///< The modules matching \p symbols
);
+
+ /// This method determines whether the archive is a properly formed llvm
+ /// bytecode archive. It first makes sure the symbol table has been loaded
+ /// and has a non-zero size. If it does, then it is an archive. If not,
+ /// then it tries to load all the bytecode modules of the archive. Finally,
+ /// it returns whether it was successfull.
+ /// @returns true if the archive is a proper llvm bytecode archive
+ /// @brief Determine whether the archive is a proper llvm bytecode archive.
+ bool isBytecodeArchive();
/// @}
/// @name Mutators
diff --git a/lib/Archive/ArchiveReader.cpp b/lib/Archive/ArchiveReader.cpp
index 56b42ff..7eab19d 100644
--- a/lib/Archive/ArchiveReader.cpp
+++ b/lib/Archive/ArchiveReader.cpp
@@ -503,3 +503,32 @@
}
}
}
+
+bool
+Archive::isBytecodeArchive()
+{
+ //Make sure the symTab has been loaded...
+ //in most cases this should have been done
+ //when the archive was constructed, but still,
+ //this is just in case.
+ if ( !symTab.size() )
+ loadSymbolTable();
+
+ //Now that we know it's been loaded, return true
+ //if it has a size
+ if ( symTab.size() ) return true;
+
+ //We still can't be sure it isn't a bytecode archive
+ loadArchive();
+
+ std::vector<Module *> Modules;
+ std::string ErrorMessage;
+
+ //If getAllModules gives an error then this isn't a proper
+ //bytecode archive
+ if ( getAllModules( Modules, &ErrorMessage ) ) return false;
+
+ //Finally, if we find any bytecode modules then this is a proper
+ //bytecode archive
+ return Modules.size();
+}
diff --git a/lib/Bytecode/Archive/ArchiveReader.cpp b/lib/Bytecode/Archive/ArchiveReader.cpp
index 56b42ff..7eab19d 100644
--- a/lib/Bytecode/Archive/ArchiveReader.cpp
+++ b/lib/Bytecode/Archive/ArchiveReader.cpp
@@ -503,3 +503,32 @@
}
}
}
+
+bool
+Archive::isBytecodeArchive()
+{
+ //Make sure the symTab has been loaded...
+ //in most cases this should have been done
+ //when the archive was constructed, but still,
+ //this is just in case.
+ if ( !symTab.size() )
+ loadSymbolTable();
+
+ //Now that we know it's been loaded, return true
+ //if it has a size
+ if ( symTab.size() ) return true;
+
+ //We still can't be sure it isn't a bytecode archive
+ loadArchive();
+
+ std::vector<Module *> Modules;
+ std::string ErrorMessage;
+
+ //If getAllModules gives an error then this isn't a proper
+ //bytecode archive
+ if ( getAllModules( Modules, &ErrorMessage ) ) return false;
+
+ //Finally, if we find any bytecode modules then this is a proper
+ //bytecode archive
+ return Modules.size();
+}