Bitcode: Don't expose WriteBitcodeToStream to clients.

llvm-svn: 151747
This commit is contained in:
Daniel Dunbar
2012-02-29 20:30:56 +00:00
parent ff3f99bf0a
commit 5fa5ecf852
3 changed files with 8 additions and 13 deletions

View File

@@ -7,6 +7,7 @@
//
//===----------------------------------------------------------------------===//
#include "llvm/ADT/SmallString.h"
#include "llvm/Analysis/Verifier.h"
#include "llvm/Bitcode/BitstreamWriter.h"
#include "llvm/Bitcode/ReaderWriter.h"
@@ -43,17 +44,16 @@ static Module *makeLLVMModule() {
return Mod;
}
static void writeModuleToBuffer(std::vector<unsigned char> &Buffer) {
static void writeModuleToBuffer(SmallVectorImpl<char> &Buffer) {
Module *Mod = makeLLVMModule();
BitstreamWriter Stream(Buffer);
WriteBitcodeToStream(Mod, Stream);
raw_svector_ostream OS(Buffer);
WriteBitcodeToFile(Mod, OS);
}
TEST(BitReaderTest, MaterializeFunctionsForBlockAddr) { // PR11677
std::vector<unsigned char> Mem;
SmallString<1024> Mem;
writeModuleToBuffer(Mem);
StringRef Data((const char*)&Mem[0], Mem.size());
MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Data, "test", false);
MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "test", false);
std::string errMsg;
Module *m = getLazyBitcodeModule(Buffer, getGlobalContext(), &errMsg);
PassManager passes;