Bitcode: Don't expose WriteBitcodeToStream to clients.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@151747 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Daniel Dunbar
2012-02-29 20:30:56 +00:00
parent 9fd58f05d5
commit fdc8f785cd
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;