xref: /xnu-10063.121.3/tests/restrict_jit.c (revision 2c2f96dc2b9a4408a43d3150ae9c105355ca3daa)
1*2c2f96dcSApple OSS Distributions #include <stdlib.h>
2*2c2f96dcSApple OSS Distributions #include <unistd.h>
3*2c2f96dcSApple OSS Distributions #include <sys/sysctl.h>
4*2c2f96dcSApple OSS Distributions #include <sys/mman.h>
5*2c2f96dcSApple OSS Distributions 
6*2c2f96dcSApple OSS Distributions #include <darwintest.h>
7*2c2f96dcSApple OSS Distributions 
8*2c2f96dcSApple OSS Distributions 
9*2c2f96dcSApple OSS Distributions /*
10*2c2f96dcSApple OSS Distributions  * macOS only test. Try to map 2 different MAP_JIT regions. 2nd should fail.
11*2c2f96dcSApple OSS Distributions  */
12*2c2f96dcSApple OSS Distributions T_DECL(restrict_jit, "macOS restricted JIT entitlement test")
13*2c2f96dcSApple OSS Distributions {
14*2c2f96dcSApple OSS Distributions #if TARGET_OS_OSX
15*2c2f96dcSApple OSS Distributions 	void *addr1;
16*2c2f96dcSApple OSS Distributions 	void *addr2;
17*2c2f96dcSApple OSS Distributions 	size_t size = 64 * 1024;
18*2c2f96dcSApple OSS Distributions 
19*2c2f96dcSApple OSS Distributions 
20*2c2f96dcSApple OSS Distributions 	addr1 = mmap(NULL, size, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_ANON | MAP_PRIVATE | MAP_JIT, -1, 0);
21*2c2f96dcSApple OSS Distributions 	T_ASSERT_NE_PTR(addr1, MAP_FAILED, "First map MAP_JIT");
22*2c2f96dcSApple OSS Distributions 
23*2c2f96dcSApple OSS Distributions 	addr2 = mmap(NULL, size, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_ANON | MAP_PRIVATE | MAP_JIT, -1, 0);
24*2c2f96dcSApple OSS Distributions 	if (addr2 == MAP_FAILED) {
25*2c2f96dcSApple OSS Distributions 		T_PASS("Only one MAP_JIT was allowed");
26*2c2f96dcSApple OSS Distributions 	} else {
27*2c2f96dcSApple OSS Distributions 		T_FAIL("Second MAP_JIT was allowed");
28*2c2f96dcSApple OSS Distributions 	}
29*2c2f96dcSApple OSS Distributions 
30*2c2f96dcSApple OSS Distributions #else
31*2c2f96dcSApple OSS Distributions 	T_SKIP("Not macOS");
32*2c2f96dcSApple OSS Distributions #endif
33*2c2f96dcSApple OSS Distributions }
34