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