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
130
131
132
|
// 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>(ut_toUTF8(L"prefault.at_open"), 1);
auto prefault_at_open =
pop.ctl_get<int>(ut_toUTF8(L"prefault.at_open"));
UT_ASSERTeq(prefault_at_open, 1);
pop.ctl_set<int>(ut_toUTF8(L"prefault.at_open"), 0);
prefault_at_open = pop.ctl_get<int>(ut_toUTF8(L"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>(ut_toUTF8(L"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>(ut_toUTF8(L"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>(ut_toUTF8(L"prefault.at_create"), 1);
auto prefault_at_create =
pmem::obj::ctl_get<int>(ut_toUTF8(L"prefault.at_create"));
UT_ASSERTeq(prefault_at_create, 1);
pmem::obj::ctl_set<int>(ut_toUTF8(L"prefault.at_create"), 0);
prefault_at_create =
pmem::obj::ctl_get<int>(ut_toUTF8(L"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>(
ut_toUTF8(L"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>(
ut_toUTF8(L"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>(
ut_toUTF8(L"prefault.non_existing_entry_point"), 1);
UT_ASSERT(0);
} catch (pmem::ctl_error &e) {
} catch (...) {
UT_ASSERT(0);
}
}
static void
test(int argc, wchar_t *argv[])
{
if (argc < 2) {
UT_FATAL("usage: %s file-name", ut_toUTF8(argv[0]));
}
auto path = argv[1];
auto pop = pmem::obj::pool<root>::create(path, L"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
wmain(int argc, wchar_t *argv[])
{
return run_test([&] { test(argc, argv); });
}
|
