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
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <malloc.h>
#include <sys/mman.h>
extern void timestamp(const char *msg, int val);
extern void testfunc_malloc(void);
extern void testfunc_deepstack(void);
void prefault_stack(void)
{
#define STACK_SIZE (7680 * 1024) /* 7.5 MiB */
char buf[STACK_SIZE];
long pagesize;
int i;
pagesize = sysconf(_SC_PAGESIZE);
for (i = 0; i < STACK_SIZE; i += pagesize)
buf[i] = 0;
}
void prefault_heap(void)
{
#define HEAP_SIZE (20 * 1024 * 1024) /* 20 MiB */
long pagesize;
char *buf;
int i;
pagesize = sysconf(_SC_PAGESIZE);
buf = malloc(HEAP_SIZE);
if (!buf) {
fprintf(stderr, "MALLOC FAILED: %s:%d\n", __FILE__, __LINE__);
return;
}
for (i = 0; i < HEAP_SIZE; i += pagesize)
buf[i] = 0;
free(buf);
}
void setup_rt(unsigned int opts)
{
if (opts & 0x1) {
if (mallopt(M_TRIM_THRESHOLD, -1) == 0) {
fprintf(stderr, "MALLOPT FAILED: %s:%d\n",
__FILE__, __LINE__);
}
if (mallopt(M_MMAP_MAX, 0) == 0) {
fprintf(stderr, "MALLOPT FAILED: %s:%d\n",
__FILE__, __LINE__);
}
}
if (opts & 0x2) {
if (mlockall(MCL_CURRENT | MCL_FUTURE) == -1) {
fprintf(stderr, "MLOCKALL FAILED: %s:%d\n",
__FILE__, __LINE__);
}
}
if (opts & 0x4)
prefault_stack();
if (opts & 0x8)
prefault_heap();
}
|
