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
|
#include <ccan/ptr_valid/ptr_valid.h>
/* Include the C files directly. */
#include <ccan/ptr_valid/ptr_valid.c>
#include <ccan/tap/tap.h>
#include <sys/mman.h>
int main(void)
{
char *page;
struct ptr_valid_batch *batch = malloc(sizeof *batch);
/* This is how many tests you plan to run */
plan_tests(14);
page = mmap(NULL, getpagesize(), PROT_READ|PROT_WRITE,
MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
strcpy(page, "hello");
ok1(ptr_valid_read(page));
ok1(ptr_valid_write(page));
ok1(ptr_valid_string(page));
ok1(ptr_valid_batch_start(batch));
ok1(ptr_valid_batch_string(batch, page));
ptr_valid_batch_end(batch);
/* Check invalid case. */
munmap(page, getpagesize());
ok1(!ptr_valid_string(page));
ok1(ptr_valid_batch_start(batch));
ok1(!ptr_valid_batch_string(batch, page));
ptr_valid_batch_end(batch);
/* Check for overrun. */
page = mmap(NULL, getpagesize()*2, PROT_READ|PROT_WRITE,
MAP_ANONYMOUS|MAP_PRIVATE, -1, 0);
munmap(page + getpagesize(), getpagesize());
memset(page, 'a', getpagesize());
ok1(!ptr_valid_string(page));
ok1(ptr_valid_batch_start(batch));
ok1(!ptr_valid_batch_string(batch, page));
ptr_valid_batch_end(batch);
page[getpagesize()-1] = '\0';
ok1(ptr_valid_string(page));
ok1(ptr_valid_batch_start(batch));
ok1(ptr_valid_batch_string(batch, page));
ptr_valid_batch_end(batch);
munmap(page, getpagesize());
free(batch);
/* This exits depending on whether all tests passed */
return exit_status();
}
|