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
|
#pragma once
#include <initializer_list>
#include <stdio.h>
#include <sys/mman.h>
#include <unistd.h>
void *malloc_hugetlbfs(size_t size) {
void *r;
char tmpName[] = "/mnt/hugetlbfs/mmallocXXXXXX";
int fd = mkstemp(tmpName);
if (fd < 0) goto fail1;
if (ftruncate(fd, size) != 0) goto fail2;
r = mmap(NULL,
size,
PROT_READ | PROT_WRITE,
MAP_SHARED | MAP_POPULATE,
fd, 0);
if (r == MAP_FAILED) goto fail2;
return r;
fail2:
close(fd);
fail1:
perror(NULL);
return MAP_FAILED;
}
void *malloc_mmap(size_t size) {
void *r;
for (auto flag : {MAP_POPULATE | MAP_HUGETLB, MAP_POPULATE, 0}) {
fprintf(stderr, "try %i", flag);
r = mmap(NULL, size,
PROT_READ | PROT_WRITE,
flag | MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (r != MAP_FAILED) break;
perror("mmap");
}
return r;
}
template<typename T>
T* mmalloc() {
void *r;
if ((r = malloc_hugetlbfs(sizeof(T))) == MAP_FAILED)
if ((r = malloc_mmap(sizeof(T))) == MAP_FAILED)
exit(EXIT_FAILURE);
return (T*) r;
}
|
