summaryrefslogtreecommitdiff
path: root/ccan/ptr_valid/test/run-string.c
blob: 59d36a44f6dfe154064f15dfbbdce8bbf655957a (plain)
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();
}