1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
//===- lld/unittest/WinLinkModuleDefTest.cpp ------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include <memory>
#include "gtest/gtest.h"
#include "lld/Driver/WinLinkModuleDef.h"
#include "llvm/Support/MemoryBuffer.h"
using namespace llvm;
using namespace lld;
class ParserTest : public testing::Test {
protected:
llvm::Optional<moduledef::Directive *> parse(const char *contents,
llvm::BumpPtrAllocator &alloc) {
auto membuf =
std::unique_ptr<MemoryBuffer>(MemoryBuffer::getMemBuffer(contents));
moduledef::Lexer lexer(std::move(membuf));
moduledef::Parser parser(lexer, alloc);
return parser.parse();
}
};
TEST_F(ParserTest, Exports) {
llvm::BumpPtrAllocator alloc;
llvm::Optional<moduledef::Directive *> dir = parse("EXPORTS\n"
" sym1\n"
" sym2 @5\n"
" sym3 @8 NONAME\n"
" sym4 DATA\n"
" sym5 @10 NONAME DATA\n",
alloc);
EXPECT_TRUE(dir.hasValue());
EXPECT_EQ(moduledef::Directive::Kind::exports, dir.getValue()->getKind());
auto *exportsDir = dyn_cast<moduledef::Exports>(dir.getValue());
EXPECT_TRUE(exportsDir != nullptr);
const std::vector<PECOFFLinkingContext::ExportDesc> &exports =
exportsDir->getExports();
EXPECT_EQ(5U, exports.size());
EXPECT_EQ(exports[0].name, "sym1");
EXPECT_EQ(exports[0].ordinal, -1);
EXPECT_EQ(exports[0].noname, false);
EXPECT_EQ(exports[0].isData, false);
EXPECT_EQ(exports[1].name, "sym2");
EXPECT_EQ(exports[1].ordinal, 5);
EXPECT_EQ(exports[1].noname, false);
EXPECT_EQ(exports[1].isData, false);
EXPECT_EQ(exports[2].name, "sym3");
EXPECT_EQ(exports[2].ordinal, 8);
EXPECT_EQ(exports[2].noname, true);
EXPECT_EQ(exports[2].isData, false);
EXPECT_EQ(exports[3].name, "sym4");
EXPECT_EQ(exports[3].ordinal, -1);
EXPECT_EQ(exports[3].noname, false);
EXPECT_EQ(exports[3].isData, true);
EXPECT_EQ(exports[4].name, "sym5");
EXPECT_EQ(exports[4].ordinal, 10);
EXPECT_EQ(exports[4].noname, true);
EXPECT_EQ(exports[4].isData, true);
}
TEST_F(ParserTest, Heapsize1) {
llvm::BumpPtrAllocator alloc;
llvm::Optional<moduledef::Directive *> dir = parse("HEAPSIZE 65536", alloc);
EXPECT_TRUE(dir.hasValue());
auto *heapsize = dyn_cast<moduledef::Heapsize>(dir.getValue());
EXPECT_TRUE(heapsize != nullptr);
EXPECT_EQ(65536U, heapsize->getReserve());
EXPECT_EQ(0U, heapsize->getCommit());
}
TEST_F(ParserTest, Heapsize2) {
llvm::BumpPtrAllocator alloc;
llvm::Optional<moduledef::Directive *> dir = parse("HEAPSIZE 65536, 8192", alloc);
EXPECT_TRUE(dir.hasValue());
auto *heapsize = dyn_cast<moduledef::Heapsize>(dir.getValue());
EXPECT_TRUE(heapsize != nullptr);
EXPECT_EQ(65536U, heapsize->getReserve());
EXPECT_EQ(8192U, heapsize->getCommit());
}
|
