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
126
127
128
129
|
// SPDX-License-Identifier: BSD-3-Clause
/* Copyright 2018-2020, Intel Corporation */
#include <libpmemobj++/make_persistent_array_atomic.hpp>
#include <libpmemobj++/persistent_ptr.hpp>
#include <libpmemobj++/pool.hpp>
#include "unittest.hpp"
struct root {
};
using Object = int[10240];
void
run_ctl_pool_prefault(pmem::obj::pool<struct root> &pop)
try {
pop.ctl_set<int>("prefault.at_open", 1);
auto prefault_at_open = pop.ctl_get<int>("prefault.at_open");
UT_ASSERTeq(prefault_at_open, 1);
pop.ctl_set<int>("prefault.at_open", 0);
prefault_at_open = pop.ctl_get<int>("prefault.at_open");
UT_ASSERTeq(prefault_at_open, 0);
} catch (...) {
UT_ASSERT(0);
}
void
run_ctl_pool_extend(pmem::obj::pool<struct root> &pop)
try {
/* disable auto-extend */
pop.ctl_set<uint64_t>("heap.size.granularity", 0);
pmem::obj::persistent_ptr<Object> ptr;
try {
/* allocate until OOM */
while (true) {
pmem::obj::make_persistent_atomic<Object>(pop, ptr);
}
} catch (...) {
}
pop.ctl_exec<uint64_t>("heap.size.extend", (1 << 20) * 10);
/* next allocation should succeed */
try {
pmem::obj::make_persistent_atomic<Object>(pop, ptr);
} catch (...) {
UT_ASSERT(0);
}
} catch (...) {
UT_ASSERT(0);
}
void
run_ctl_global()
try {
pmem::obj::ctl_set<int>("prefault.at_create", 1);
auto prefault_at_create = pmem::obj::ctl_get<int>("prefault.at_create");
UT_ASSERTeq(prefault_at_create, 1);
pmem::obj::ctl_set<int>("prefault.at_create", 0);
prefault_at_create = pmem::obj::ctl_get<int>("prefault.at_create");
UT_ASSERTeq(prefault_at_create, 0);
} catch (...) {
UT_ASSERT(0);
}
void
run_ctl_exception()
{
try {
/* run query with non-existing entry point */
pmem::obj::ctl_set<int>("prefault.non_existing_entry_point", 1);
UT_ASSERT(0);
} catch (pmem::ctl_error &e) {
} catch (...) {
UT_ASSERT(0);
}
try {
/* run query with non-existing entry point */
pmem::obj::ctl_get<int>("prefault.non_existing_entry_point");
UT_ASSERT(0);
} catch (pmem::ctl_error &e) {
} catch (...) {
UT_ASSERT(0);
}
try {
/* run query with non-existing entry point */
pmem::obj::ctl_exec<int>("prefault.non_existing_entry_point",
1);
UT_ASSERT(0);
} catch (pmem::ctl_error &e) {
} catch (...) {
UT_ASSERT(0);
}
}
static void
test(int argc, char *argv[])
{
if (argc < 2) {
UT_FATAL("usage: %s file-name", argv[0]);
}
auto path = argv[1];
auto pop = pmem::obj::pool<root>::create(path, "ctl_test", 0,
S_IWUSR | S_IRUSR);
run_ctl_pool_prefault(pop);
run_ctl_pool_extend(pop);
run_ctl_global();
run_ctl_exception();
pop.close();
}
int
main(int argc, char *argv[])
{
return run_test([&] { test(argc, argv); });
}
|
