forked from OSchip/llvm-project
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
//===- llvm/unittest/Bitcode/BitReaderTest.cpp - Tests for BitReader ------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/Bitcode/BitstreamWriter.h"
|
|
#include "llvm/Bitcode/ReaderWriter.h"
|
|
#include "llvm/IR/Constants.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/Verifier.h"
|
|
#include "llvm/PassManager.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
namespace llvm {
|
|
namespace {
|
|
|
|
static Module *makeLLVMModule() {
|
|
Module* Mod = new Module("test-mem", getGlobalContext());
|
|
|
|
FunctionType* FuncTy =
|
|
FunctionType::get(Type::getVoidTy(Mod->getContext()), false);
|
|
Function* Func = Function::Create(FuncTy,GlobalValue::ExternalLinkage,
|
|
"func", Mod);
|
|
|
|
BasicBlock* Entry = BasicBlock::Create(Mod->getContext(), "entry", Func);
|
|
new UnreachableInst(Mod->getContext(), Entry);
|
|
|
|
BasicBlock* BB = BasicBlock::Create(Mod->getContext(), "bb", Func);
|
|
new UnreachableInst(Mod->getContext(), BB);
|
|
|
|
PointerType* Int8Ptr = Type::getInt8PtrTy(Mod->getContext());
|
|
new GlobalVariable(*Mod, Int8Ptr, /*isConstant=*/true,
|
|
GlobalValue::ExternalLinkage,
|
|
BlockAddress::get(BB), "table");
|
|
|
|
return Mod;
|
|
}
|
|
|
|
static void writeModuleToBuffer(SmallVectorImpl<char> &Buffer) {
|
|
OwningPtr<Module> Mod(makeLLVMModule());
|
|
raw_svector_ostream OS(Buffer);
|
|
WriteBitcodeToFile(Mod.get(), OS);
|
|
}
|
|
|
|
TEST(BitReaderTest, MaterializeFunctionsForBlockAddr) { // PR11677
|
|
SmallString<1024> Mem;
|
|
writeModuleToBuffer(Mem);
|
|
MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "test", false);
|
|
ErrorOr<Module *> ModuleOrErr =
|
|
getLazyBitcodeModule(Buffer, getGlobalContext());
|
|
OwningPtr<Module> m(ModuleOrErr.get());
|
|
PassManager passes;
|
|
passes.add(createVerifierPass());
|
|
passes.run(*m);
|
|
}
|
|
|
|
}
|
|
}
|