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
|
#include <sys/prctl.h>
#include <sys/mman.h>
#include "zdtmtst.h"
#include "get_smaps_bits.h"
#ifndef MADV_DONTDUMP
#define MADV_DONTDUMP 16
#endif
const char *test_doc = "Test prctl(THP_DISABLE) behaviour";
const char *test_author = "Mike Rapoport <rppt@linux.ibm.com>";
#define MEM_SIZE (2 << 20)
int main(int argc, char **argv)
{
unsigned long orig_flags = 0, new_flags = 0;
unsigned long orig_madv = 0, new_madv = 0;
void *area;
test_init(argc, argv);
area = mmap(NULL, MEM_SIZE, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (area == MAP_FAILED) {
pr_perror("mmap failed");
return -1;
}
test_msg("Fetch existing flags/adv\n");
if (get_smaps_bits((unsigned long)area, &orig_flags, &orig_madv))
return -1;
if (prctl(PR_SET_THP_DISABLE, 1, 0, 0, 0)) {
pr_perror("Disabling THP failed");
return -1;
}
test_daemon();
test_waitsig();
if (prctl(PR_SET_THP_DISABLE, 0, 0, 0, 0)) {
pr_perror("Enabling THP failed");
return -1;
}
test_msg("Fetch restored flags/adv\n");
if (get_smaps_bits((unsigned long)area, &new_flags, &new_madv))
return -1;
if (orig_flags != new_flags) {
pr_err("Flags are changed %lx -> %lx\n", orig_flags, new_flags);
fail();
return -1;
}
if (orig_madv != new_madv) {
pr_err("Madvs are changed %lx -> %lx\n", orig_madv, new_madv);
fail();
return -1;
}
pass();
return 0;
}
|