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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/fastos/fastos.h>
#include <stddef.h>
#include <vespa/log/log.h>
LOG_SETUP("alloc_test");
#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/vespalib/util/alloc.h>
#include <vespa/vespalib/util/exceptions.h>
using namespace vespalib;
using namespace vespalib::alloc;
template <typename T>
void
testSwap(T & a, T & b)
{
void * tmpA(a.get());
void * tmpB(b.get());
EXPECT_EQUAL(4096u, a.size());
EXPECT_EQUAL(8192, b.size());
std::swap(a, b);
EXPECT_EQUAL(4096u, b.size());
EXPECT_EQUAL(8192, a.size());
EXPECT_EQUAL(tmpA, b.get());
EXPECT_EQUAL(tmpB, a.get());
}
TEST("test basics") {
{
Alloc h = Alloc::allocHeap(100);
EXPECT_EQUAL(100u, h.size());
EXPECT_TRUE(h.get() != nullptr);
}
{
EXPECT_EXCEPTION(Alloc::allocAlignedHeap(100, 7), IllegalArgumentException, "Alloc::allocAlignedHeap(100, 7) does not support 7 alignment");
Alloc h = Alloc::allocAlignedHeap(100, 1024);
EXPECT_EQUAL(100u, h.size());
EXPECT_TRUE(h.get() != nullptr);
}
{
Alloc h = Alloc::allocMMap(100);
EXPECT_EQUAL(4096u, h.size());
EXPECT_TRUE(h.get() != nullptr);
}
{
Alloc a = Alloc::allocHeap(4096), b = Alloc::allocHeap(8192);
testSwap(a, b);
}
{
Alloc a = Alloc::allocMMap(4096), b = Alloc::allocMMap(8192);
testSwap(a, b);
}
{
Alloc a = Alloc::allocAlignedHeap(4096, 1024), b = Alloc::allocAlignedHeap(8192, 1024);
testSwap(a, b);
}
{
Alloc a = Alloc::allocHeap(4096);
Alloc b = Alloc::allocMMap(8192);
testSwap(a, b);
}
{
Alloc a = Alloc::allocHeap(100);
Alloc b = Alloc::allocHeap(100);
a = std::move(b);
EXPECT_TRUE(b.get() == nullptr);
}
}
TEST("test correct alignment") {
{
Alloc buf = Alloc::alloc(10, MemoryAllocator::HUGEPAGE_SIZE, 1024);
EXPECT_TRUE(reinterpret_cast<ptrdiff_t>(buf.get()) % 1024 == 0);
}
{
// Mmapped pointers are page-aligned, but sanity test anyway.
Alloc buf = Alloc::alloc(3000000, MemoryAllocator::HUGEPAGE_SIZE, 512);
EXPECT_TRUE(reinterpret_cast<ptrdiff_t>(buf.get()) % 512 == 0);
}
}
TEST("no rounding of small heap buffer") {
Alloc buf = Alloc::alloc(3, MemoryAllocator::HUGEPAGE_SIZE);
EXPECT_EQUAL(3ul, buf.size());
}
TEST("no rounding of large heap buffer") {
Alloc buf = Alloc::alloc(MemoryAllocator::HUGEPAGE_SIZE*11+3, MemoryAllocator::HUGEPAGE_SIZE*16);
EXPECT_EQUAL(MemoryAllocator::HUGEPAGE_SIZE*11+3, buf.size());
}
TEST("rounding of small mmaped buffer") {
Alloc buf = Alloc::alloc(MemoryAllocator::HUGEPAGE_SIZE);
EXPECT_EQUAL(MemoryAllocator::HUGEPAGE_SIZE, buf.size());
buf = Alloc::alloc(MemoryAllocator::HUGEPAGE_SIZE+1);
EXPECT_EQUAL(MemoryAllocator::HUGEPAGE_SIZE*2, buf.size());
}
TEST("rounding of large mmaped buffer") {
Alloc buf = Alloc::alloc(MemoryAllocator::HUGEPAGE_SIZE*11+3);
EXPECT_EQUAL(MemoryAllocator::HUGEPAGE_SIZE*12, buf.size());
}
TEST("heap alloc can not be extended") {
Alloc buf = Alloc::allocHeap(100);
void * oldPtr = buf.get();
EXPECT_EQUAL(100, buf.size());
EXPECT_FALSE(buf.resize_inplace(101));
EXPECT_EQUAL(oldPtr, buf.get());
EXPECT_EQUAL(100, buf.size());
}
TEST("mmap alloc can be extended") {
Alloc buf = Alloc::allocMMap(100);
void * oldPtr = buf.get();
EXPECT_EQUAL(4096, buf.size());
EXPECT_TRUE(buf.resize_inplace(4097));
EXPECT_EQUAL(oldPtr, buf.get());
EXPECT_EQUAL(8192, buf.size());
}
TEST_MAIN() { TEST_RUN_ALL(); }
|