1*0f4c859eSApple OSS Distributions #include "test_fault_helper.h"
2*0f4c859eSApple OSS Distributions #include "fail.h"
3*0f4c859eSApple OSS Distributions #include <sys/mman.h>
4*0f4c859eSApple OSS Distributions #include <stdlib.h>
5*0f4c859eSApple OSS Distributions #include <unistd.h>
6*0f4c859eSApple OSS Distributions #include <assert.h>
7*0f4c859eSApple OSS Distributions #include <TargetConditionals.h>
8*0f4c859eSApple OSS Distributions
9*0f4c859eSApple OSS Distributions #if (TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
10*0f4c859eSApple OSS Distributions #define MEMSIZE (1L<<28)
11*0f4c859eSApple OSS Distributions #else
12*0f4c859eSApple OSS Distributions #define MEMSIZE (1L<<30)
13*0f4c859eSApple OSS Distributions #endif
14*0f4c859eSApple OSS Distributions
15*0f4c859eSApple OSS Distributions static char* memblock;
16*0f4c859eSApple OSS Distributions
17*0f4c859eSApple OSS Distributions int
test_fault_setup()18*0f4c859eSApple OSS Distributions test_fault_setup()
19*0f4c859eSApple OSS Distributions {
20*0f4c859eSApple OSS Distributions char *ptr;
21*0f4c859eSApple OSS Distributions int pgsz = getpagesize();
22*0f4c859eSApple OSS Distributions int retval;
23*0f4c859eSApple OSS Distributions
24*0f4c859eSApple OSS Distributions memblock = (char *)mmap(NULL, MEMSIZE, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE, -1, 0);
25*0f4c859eSApple OSS Distributions VERIFY(memblock != MAP_FAILED, "mmap failed");
26*0f4c859eSApple OSS Distributions
27*0f4c859eSApple OSS Distributions /* make sure memory is paged */
28*0f4c859eSApple OSS Distributions for (ptr = memblock; ptr < memblock + MEMSIZE; ptr += pgsz) {
29*0f4c859eSApple OSS Distributions *ptr = 1;
30*0f4c859eSApple OSS Distributions }
31*0f4c859eSApple OSS Distributions
32*0f4c859eSApple OSS Distributions /* set to read only, then back to read write so it faults on first write */
33*0f4c859eSApple OSS Distributions retval = mprotect(memblock, MEMSIZE, PROT_READ);
34*0f4c859eSApple OSS Distributions VERIFY(retval == 0, "mprotect failed");
35*0f4c859eSApple OSS Distributions
36*0f4c859eSApple OSS Distributions retval = mprotect(memblock, MEMSIZE, PROT_READ | PROT_WRITE);
37*0f4c859eSApple OSS Distributions VERIFY(retval == 0, "mprotect failed");
38*0f4c859eSApple OSS Distributions
39*0f4c859eSApple OSS Distributions return PERFINDEX_SUCCESS;
40*0f4c859eSApple OSS Distributions }
41*0f4c859eSApple OSS Distributions
42*0f4c859eSApple OSS Distributions int
test_fault_helper(int thread_id,int num_threads,long long length,testtype_t testtype)43*0f4c859eSApple OSS Distributions test_fault_helper(int thread_id, int num_threads, long long length, testtype_t testtype)
44*0f4c859eSApple OSS Distributions {
45*0f4c859eSApple OSS Distributions char *ptr;
46*0f4c859eSApple OSS Distributions int pgsz = getpagesize();
47*0f4c859eSApple OSS Distributions int retval;
48*0f4c859eSApple OSS Distributions
49*0f4c859eSApple OSS Distributions long long num_pages = MEMSIZE / pgsz;
50*0f4c859eSApple OSS Distributions long long region_len = num_pages / num_threads;
51*0f4c859eSApple OSS Distributions long long region_start = region_len * thread_id;
52*0f4c859eSApple OSS Distributions long long region_end;
53*0f4c859eSApple OSS Distributions
54*0f4c859eSApple OSS Distributions if (thread_id < num_pages % num_threads) {
55*0f4c859eSApple OSS Distributions region_start += thread_id;
56*0f4c859eSApple OSS Distributions region_len++;
57*0f4c859eSApple OSS Distributions } else {
58*0f4c859eSApple OSS Distributions region_start += num_pages % num_threads;
59*0f4c859eSApple OSS Distributions }
60*0f4c859eSApple OSS Distributions
61*0f4c859eSApple OSS Distributions region_start *= pgsz;
62*0f4c859eSApple OSS Distributions region_len *= pgsz;
63*0f4c859eSApple OSS Distributions region_end = region_start + region_len;
64*0f4c859eSApple OSS Distributions
65*0f4c859eSApple OSS Distributions long long left = length;
66*0f4c859eSApple OSS Distributions
67*0f4c859eSApple OSS Distributions while (1) {
68*0f4c859eSApple OSS Distributions for (ptr = memblock + region_start; ptr < memblock + region_end; ptr += pgsz) {
69*0f4c859eSApple OSS Distributions *ptr = 1;
70*0f4c859eSApple OSS Distributions left--;
71*0f4c859eSApple OSS Distributions if (left == 0) {
72*0f4c859eSApple OSS Distributions break;
73*0f4c859eSApple OSS Distributions }
74*0f4c859eSApple OSS Distributions }
75*0f4c859eSApple OSS Distributions
76*0f4c859eSApple OSS Distributions if (left == 0) {
77*0f4c859eSApple OSS Distributions break;
78*0f4c859eSApple OSS Distributions }
79*0f4c859eSApple OSS Distributions
80*0f4c859eSApple OSS Distributions if (testtype == TESTFAULT) {
81*0f4c859eSApple OSS Distributions retval = mprotect(memblock + region_start, region_len, PROT_READ) == 0;
82*0f4c859eSApple OSS Distributions VERIFY(retval == 0, "mprotect failed");
83*0f4c859eSApple OSS Distributions retval = mprotect(memblock + region_start, region_len, PROT_READ | PROT_WRITE) == 0;
84*0f4c859eSApple OSS Distributions VERIFY(retval == 0, "mprotect failed");
85*0f4c859eSApple OSS Distributions } else if (testtype == TESTZFOD) {
86*0f4c859eSApple OSS Distributions retval = munmap(memblock + region_start, region_len) == 0;
87*0f4c859eSApple OSS Distributions VERIFY(retval == 0, "munmap failed");
88*0f4c859eSApple OSS Distributions ptr = mmap(memblock + region_start, region_len, PROT_READ | PROT_WRITE, MAP_ANON | MAP_PRIVATE | MAP_FIXED, -1, 0);
89*0f4c859eSApple OSS Distributions VERIFY(ptr != 0, "mmap failed");
90*0f4c859eSApple OSS Distributions }
91*0f4c859eSApple OSS Distributions }
92*0f4c859eSApple OSS Distributions return PERFINDEX_SUCCESS;
93*0f4c859eSApple OSS Distributions }
94