| OLD | NEW |
| 1 //===- llvm/unittest/Bitcode/NaClBitReaderTest.cpp - Tests for BitReader --===// | 1 //===- llvm/unittest/Bitcode/NaClBitReaderTest.cpp - Tests for BitReader --===// |
| 2 // | 2 // |
| 3 // The LLVM Compiler Infrastructure | 3 // The LLVM Compiler Infrastructure |
| 4 // | 4 // |
| 5 // This file is distributed under the University of Illinois Open Source | 5 // This file is distributed under the University of Illinois Open Source |
| 6 // License. See LICENSE.TXT for details. | 6 // License. See LICENSE.TXT for details. |
| 7 // | 7 // |
| 8 //===----------------------------------------------------------------------===// | 8 //===----------------------------------------------------------------------===// |
| 9 | 9 |
| 10 #include "llvm/ADT/SmallString.h" | 10 #include "llvm/ADT/SmallString.h" |
| 11 #include "llvm/Bitcode/BitstreamWriter.h" | 11 #include "llvm/Bitcode/BitstreamWriter.h" |
| 12 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" | 12 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" |
| 13 #include "llvm/IR/Constants.h" | 13 #include "llvm/IR/Constants.h" |
| 14 #include "llvm/IR/Instructions.h" | 14 #include "llvm/IR/Instructions.h" |
| 15 #include "llvm/IR/LLVMContext.h" | 15 #include "llvm/IR/LLVMContext.h" |
| 16 #include "llvm/IR/Module.h" | 16 #include "llvm/IR/Module.h" |
| 17 #include "llvm/IR/Verifier.h" | 17 #include "llvm/IR/Verifier.h" |
| 18 #include "llvm/PassManager.h" | 18 #include "llvm/PassManager.h" |
| 19 #include "llvm/Support/Debug.h" |
| 19 #include "llvm/Support/MemoryBuffer.h" | 20 #include "llvm/Support/MemoryBuffer.h" |
| 20 #include "gtest/gtest.h" | 21 #include "gtest/gtest.h" |
| 21 | 22 |
| 22 namespace llvm { | 23 namespace llvm { |
| 23 namespace { | 24 namespace { |
| 24 | 25 |
| 25 static Module *makeLLVMModule() { | 26 static std::unique_ptr<Module> makeLLVMModule() { |
| 26 Module* Mod = new Module("test-mem", getGlobalContext()); | 27 std::unique_ptr<Module> Mod(new Module("test-mem", getGlobalContext())); |
| 27 | 28 |
| 28 FunctionType* FuncTy = | 29 FunctionType* FuncTy = |
| 29 FunctionType::get(Type::getVoidTy(Mod->getContext()), false); | 30 FunctionType::get(Type::getVoidTy(Mod->getContext()), false); |
| 30 Function* Func = Function::Create(FuncTy,GlobalValue::ExternalLinkage, | 31 Function* Func = Function::Create(FuncTy,GlobalValue::ExternalLinkage, |
| 31 "func", Mod); | 32 "func", Mod.get()); |
| 32 | 33 |
| 33 BasicBlock* Entry = BasicBlock::Create(Mod->getContext(), "entry", Func); | 34 BasicBlock* Entry = BasicBlock::Create(Mod->getContext(), "entry", Func); |
| 34 new UnreachableInst(Mod->getContext(), Entry); | 35 new UnreachableInst(Mod->getContext(), Entry); |
| 35 | 36 |
| 36 BasicBlock* BB = BasicBlock::Create(Mod->getContext(), "bb", Func); | 37 BasicBlock* BB = BasicBlock::Create(Mod->getContext(), "bb", Func); |
| 37 new UnreachableInst(Mod->getContext(), BB); | 38 new UnreachableInst(Mod->getContext(), BB); |
| 38 | 39 |
| 39 return Mod; | 40 return Mod; |
| 40 } | 41 } |
| 41 | 42 |
| 42 static void writeModuleToBuffer(SmallVectorImpl<char> &Buffer) { | 43 static void writeModuleToBuffer(SmallVectorImpl<char> &Buffer) { |
| 43 std::unique_ptr<Module> Mod(makeLLVMModule()); | 44 std::unique_ptr<Module> Mod = makeLLVMModule(); |
| 44 raw_svector_ostream OS(Buffer); | 45 raw_svector_ostream OS(Buffer); |
| 45 NaClWriteBitcodeToFile(Mod.get(), OS); | 46 NaClWriteBitcodeToFile(Mod.get(), OS); |
| 46 } | 47 } |
| 47 | 48 |
| 48 // Check that we can parse a good bitcode file. | 49 // Check that we can parse a good bitcode file. |
| 49 TEST(NaClBitReaderTest, MaterializeSimpleModule) { | 50 TEST(NaClBitReaderTest, MaterializeSimpleModule) { |
| 50 SmallString<1024> Mem; | 51 SmallString<1024> Mem; |
| 51 writeModuleToBuffer(Mem); | 52 writeModuleToBuffer(Mem); |
| 52 MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "test", false); | 53 std::unique_ptr<MemoryBuffer> Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "
test", false); |
| 53 ErrorOr<Module *> ModuleOrErr = | 54 ErrorOr<Module *> ModuleOrErr = |
| 54 getNaClLazyBitcodeModule(Buffer, getGlobalContext()); | 55 getNaClLazyBitcodeModule(std::move(Buffer), getGlobalContext()); |
| 55 EXPECT_EQ(true, bool(ModuleOrErr)); | 56 EXPECT_EQ(true, bool(ModuleOrErr)); |
| 56 // Do something with the module just to make sure it was built. | 57 // Do something with the module just to make sure it was built. |
| 57 std::unique_ptr<Module> m(ModuleOrErr.get()); | 58 std::unique_ptr<Module> M(ModuleOrErr.get()); |
| 58 EXPECT_NE((Module *)nullptr, m.get()); | 59 EXPECT_NE((Module *)nullptr, M.get()); |
| 59 PassManager passes; | 60 M->getFunction("func")->materialize(); |
| 60 passes.add(createVerifierPass()); | 61 EXPECT_FALSE(verifyModule(*M, &dbgs())); |
| 61 passes.run(*m); | |
| 62 } | 62 } |
| 63 | 63 |
| 64 // Test that we catch bad stuff at the end of a bitcode file. | 64 // Test that we catch bad stuff at the end of a bitcode file. |
| 65 TEST(NaClBitReaderTest, BadDataAfterModule) { | 65 TEST(NaClBitReaderTest, BadDataAfterModule) { |
| 66 SmallString<1024> Mem; | 66 SmallString<1024> Mem; |
| 67 writeModuleToBuffer(Mem); | 67 writeModuleToBuffer(Mem); |
| 68 Mem.append("more"); // Length must be divisible by 4! | 68 Mem.append("more"); // Length must be divisible by 4! |
| 69 MemoryBuffer *Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "test", false); | 69 std::unique_ptr<MemoryBuffer> Buffer = MemoryBuffer::getMemBuffer(Mem.str(), "
test", false); |
| 70 ErrorOr<Module *> ModuleOrErr = | 70 ErrorOr<Module *> ModuleOrErr = |
| 71 getNaClLazyBitcodeModule(Buffer, getGlobalContext()); | 71 getNaClLazyBitcodeModule(std::move(Buffer), getGlobalContext()); |
| 72 EXPECT_EQ(false, bool(ModuleOrErr)); | 72 EXPECT_EQ(false, bool(ModuleOrErr)); |
| 73 std::string BadMessage("Invalid data after module"); | 73 std::string BadMessage("Invalid data after module"); |
| 74 EXPECT_EQ(BadMessage, ModuleOrErr.getError().message()); | 74 EXPECT_EQ(BadMessage, ModuleOrErr.getError().message()); |
| 75 } | 75 } |
| 76 | 76 |
| 77 } | 77 } |
| 78 } | 78 } |
| OLD | NEW |