2012-01-02 17:19:48 +08:00
|
|
|
//===- llvm/unittest/Bitcode/BitReaderTest.cpp - Tests for BitReader ------===//
|
2012-01-02 15:49:53 +08:00
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2015-06-17 09:15:47 +08:00
|
|
|
#include "llvm/ADT/STLExtras.h"
|
2016-11-11 13:34:58 +08:00
|
|
|
#include "llvm/ADT/SmallString.h"
|
2015-01-14 19:23:27 +08:00
|
|
|
#include "llvm/AsmParser/Parser.h"
|
2016-11-11 13:34:58 +08:00
|
|
|
#include "llvm/Bitcode/BitcodeReader.h"
|
|
|
|
#include "llvm/Bitcode/BitcodeWriter.h"
|
2013-01-02 19:36:10 +08:00
|
|
|
#include "llvm/IR/LLVMContext.h"
|
|
|
|
#include "llvm/IR/Module.h"
|
2014-01-13 17:26:24 +08:00
|
|
|
#include "llvm/IR/Verifier.h"
|
2014-08-02 05:01:04 +08:00
|
|
|
#include "llvm/Support/Debug.h"
|
2016-11-10 01:49:19 +08:00
|
|
|
#include "llvm/Support/Error.h"
|
2012-01-02 15:49:53 +08:00
|
|
|
#include "llvm/Support/MemoryBuffer.h"
|
2014-08-02 05:01:04 +08:00
|
|
|
#include "llvm/Support/SourceMgr.h"
|
2012-01-02 15:49:53 +08:00
|
|
|
#include "gtest/gtest.h"
|
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
using namespace llvm;
|
2012-01-02 15:49:53 +08:00
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
namespace {
|
2012-01-02 15:49:53 +08:00
|
|
|
|
2016-04-15 05:59:01 +08:00
|
|
|
std::unique_ptr<Module> parseAssembly(LLVMContext &Context,
|
|
|
|
const char *Assembly) {
|
2014-08-02 05:01:04 +08:00
|
|
|
SMDiagnostic Error;
|
2016-04-15 05:59:01 +08:00
|
|
|
std::unique_ptr<Module> M = parseAssemblyString(Assembly, Error, Context);
|
2012-01-02 15:49:53 +08:00
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
std::string ErrMsg;
|
|
|
|
raw_string_ostream OS(ErrMsg);
|
|
|
|
Error.print("", OS);
|
2012-01-02 15:49:53 +08:00
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
// A failure here means that the test itself is buggy.
|
2014-08-20 00:58:54 +08:00
|
|
|
if (!M)
|
2014-08-02 05:01:04 +08:00
|
|
|
report_fatal_error(OS.str().c_str());
|
2012-01-02 15:49:53 +08:00
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
return M;
|
2012-01-02 15:49:53 +08:00
|
|
|
}
|
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
static void writeModuleToBuffer(std::unique_ptr<Module> Mod,
|
|
|
|
SmallVectorImpl<char> &Buffer) {
|
2012-03-01 04:30:56 +08:00
|
|
|
raw_svector_ostream OS(Buffer);
|
2018-02-15 03:11:32 +08:00
|
|
|
WriteBitcodeToFile(*Mod, OS);
|
2012-01-02 15:49:53 +08:00
|
|
|
}
|
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
static std::unique_ptr<Module> getLazyModuleFromAssembly(LLVMContext &Context,
|
|
|
|
SmallString<1024> &Mem,
|
|
|
|
const char *Assembly) {
|
2016-04-15 05:59:01 +08:00
|
|
|
writeModuleToBuffer(parseAssembly(Context, Assembly), Mem);
|
2016-11-13 15:00:17 +08:00
|
|
|
Expected<std::unique_ptr<Module>> ModuleOrErr =
|
2016-11-08 14:03:43 +08:00
|
|
|
getLazyBitcodeModule(MemoryBufferRef(Mem.str(), "test"), Context);
|
2016-11-13 15:00:17 +08:00
|
|
|
if (!ModuleOrErr)
|
|
|
|
report_fatal_error("Could not parse bitcode module");
|
2015-06-17 06:27:55 +08:00
|
|
|
return std::move(ModuleOrErr.get());
|
2012-01-02 15:49:53 +08:00
|
|
|
}
|
2012-01-02 17:19:48 +08:00
|
|
|
|
2015-05-07 00:52:35 +08:00
|
|
|
// Tests that lazy evaluation can parse functions out of order.
|
|
|
|
TEST(BitReaderTest, MaterializeFunctionsOutOfOrder) {
|
|
|
|
SmallString<1024> Mem;
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = getLazyModuleFromAssembly(
|
|
|
|
Context, Mem, "define void @f() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @g() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @h() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @j() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n");
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
Function *F = M->getFunction("f");
|
|
|
|
Function *G = M->getFunction("g");
|
|
|
|
Function *H = M->getFunction("h");
|
|
|
|
Function *J = M->getFunction("j");
|
|
|
|
|
|
|
|
// Initially all functions are not materialized (no basic blocks).
|
|
|
|
EXPECT_TRUE(F->empty());
|
|
|
|
EXPECT_TRUE(G->empty());
|
|
|
|
EXPECT_TRUE(H->empty());
|
|
|
|
EXPECT_TRUE(J->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize h.
|
2016-11-10 01:49:19 +08:00
|
|
|
ASSERT_FALSE(H->materialize());
|
2015-05-07 00:52:35 +08:00
|
|
|
EXPECT_TRUE(F->empty());
|
|
|
|
EXPECT_TRUE(G->empty());
|
|
|
|
EXPECT_FALSE(H->empty());
|
|
|
|
EXPECT_TRUE(J->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize g.
|
2016-11-10 01:49:19 +08:00
|
|
|
ASSERT_FALSE(G->materialize());
|
2015-05-07 00:52:35 +08:00
|
|
|
EXPECT_TRUE(F->empty());
|
|
|
|
EXPECT_FALSE(G->empty());
|
|
|
|
EXPECT_FALSE(H->empty());
|
|
|
|
EXPECT_TRUE(J->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize j.
|
2016-11-10 01:49:19 +08:00
|
|
|
ASSERT_FALSE(J->materialize());
|
2015-05-07 00:52:35 +08:00
|
|
|
EXPECT_TRUE(F->empty());
|
|
|
|
EXPECT_FALSE(G->empty());
|
|
|
|
EXPECT_FALSE(H->empty());
|
|
|
|
EXPECT_FALSE(J->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize f.
|
2016-11-10 01:49:19 +08:00
|
|
|
ASSERT_FALSE(F->materialize());
|
2015-05-07 00:52:35 +08:00
|
|
|
EXPECT_FALSE(F->empty());
|
|
|
|
EXPECT_FALSE(G->empty());
|
|
|
|
EXPECT_FALSE(H->empty());
|
|
|
|
EXPECT_FALSE(J->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
}
|
|
|
|
|
2014-08-02 05:01:04 +08:00
|
|
|
TEST(BitReaderTest, MaterializeFunctionsForBlockAddr) { // PR11677
|
|
|
|
SmallString<1024> Mem;
|
|
|
|
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = getLazyModuleFromAssembly(
|
|
|
|
Context, Mem, "@table = constant i8* blockaddress(@func, %bb)\n"
|
|
|
|
"define void @func() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"bb:\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n");
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
2016-11-02 08:08:19 +08:00
|
|
|
EXPECT_FALSE(M->getFunction("func")->empty());
|
2014-08-02 05:11:34 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST(BitReaderTest, MaterializeFunctionsForBlockAddrInFunctionBefore) {
|
|
|
|
SmallString<1024> Mem;
|
|
|
|
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = getLazyModuleFromAssembly(
|
|
|
|
Context, Mem, "define i8* @before() {\n"
|
|
|
|
" ret i8* blockaddress(@func, %bb)\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @other() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @func() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"bb:\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n");
|
|
|
|
EXPECT_TRUE(M->getFunction("before")->empty());
|
|
|
|
EXPECT_TRUE(M->getFunction("func")->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize @before, pulling in @func.
|
2014-10-25 06:50:48 +08:00
|
|
|
EXPECT_FALSE(M->getFunction("before")->materialize());
|
2014-08-02 05:11:34 +08:00
|
|
|
EXPECT_FALSE(M->getFunction("func")->empty());
|
|
|
|
EXPECT_TRUE(M->getFunction("other")->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(BitReaderTest, MaterializeFunctionsForBlockAddrInFunctionAfter) {
|
|
|
|
SmallString<1024> Mem;
|
|
|
|
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = getLazyModuleFromAssembly(
|
|
|
|
Context, Mem, "define void @func() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"bb:\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define void @other() {\n"
|
|
|
|
" unreachable\n"
|
|
|
|
"}\n"
|
|
|
|
"define i8* @after() {\n"
|
|
|
|
" ret i8* blockaddress(@func, %bb)\n"
|
|
|
|
"}\n");
|
|
|
|
EXPECT_TRUE(M->getFunction("after")->empty());
|
|
|
|
EXPECT_TRUE(M->getFunction("func")->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
|
|
|
|
|
|
|
// Materialize @after, pulling in @func.
|
2014-10-25 06:50:48 +08:00
|
|
|
EXPECT_FALSE(M->getFunction("after")->materialize());
|
2014-08-02 05:11:34 +08:00
|
|
|
EXPECT_FALSE(M->getFunction("func")->empty());
|
|
|
|
EXPECT_TRUE(M->getFunction("other")->empty());
|
|
|
|
EXPECT_FALSE(verifyModule(*M, &dbgs()));
|
2012-01-02 15:49:53 +08:00
|
|
|
}
|
2014-08-02 05:01:04 +08:00
|
|
|
|
|
|
|
} // end namespace
|