xref: /xnu-12377.41.6/tests/get_shared_cache_address.c (revision bbb1b6f9e71b8cdde6e5cd6f4841f207dee3d828)
1*bbb1b6f9SApple OSS Distributions #include <sys/resource.h>
2*bbb1b6f9SApple OSS Distributions #include <unistd.h>
3*bbb1b6f9SApple OSS Distributions #include <stdlib.h>
4*bbb1b6f9SApple OSS Distributions #include <stdio.h>
5*bbb1b6f9SApple OSS Distributions #include <string.h>
6*bbb1b6f9SApple OSS Distributions #include <libproc.h>
7*bbb1b6f9SApple OSS Distributions #include <mach-o/dyld.h>
8*bbb1b6f9SApple OSS Distributions #include <mach-o/dyld_priv.h>
9*bbb1b6f9SApple OSS Distributions 
10*bbb1b6f9SApple OSS Distributions /*
11*bbb1b6f9SApple OSS Distributions  * Test helper to retrieve the address of the shared cache. The helper
12*bbb1b6f9SApple OSS Distributions  * also verifies that the process is correctly marked to have the shared
13*bbb1b6f9SApple OSS Distributions  * cache reslid when interrogated through proc_pid_rusage()
14*bbb1b6f9SApple OSS Distributions  */
15*bbb1b6f9SApple OSS Distributions int
main(int argc,char ** argv)16*bbb1b6f9SApple OSS Distributions main(int argc, char **argv)
17*bbb1b6f9SApple OSS Distributions {
18*bbb1b6f9SApple OSS Distributions 	size_t shared_cache_len = 0;
19*bbb1b6f9SApple OSS Distributions 	struct rusage_info_v5 ru = {};
20*bbb1b6f9SApple OSS Distributions 
21*bbb1b6f9SApple OSS Distributions 	if (argc != 2) {
22*bbb1b6f9SApple OSS Distributions 		fprintf(stderr, "Invalid helper invocation");
23*bbb1b6f9SApple OSS Distributions 		exit(1);
24*bbb1b6f9SApple OSS Distributions 	}
25*bbb1b6f9SApple OSS Distributions 
26*bbb1b6f9SApple OSS Distributions 	if (proc_pid_rusage(getpid(), RUSAGE_INFO_V5, (rusage_info_t *)&ru) != 0) {
27*bbb1b6f9SApple OSS Distributions 		perror("proc_pid_rusage() helper");
28*bbb1b6f9SApple OSS Distributions 		exit(1);
29*bbb1b6f9SApple OSS Distributions 	}
30*bbb1b6f9SApple OSS Distributions 
31*bbb1b6f9SApple OSS Distributions 	if (strcmp(argv[1], "check_rusage_flag") == 0) {
32*bbb1b6f9SApple OSS Distributions 		if (!(ru.ri_flags & RU_PROC_RUNS_RESLIDE)) {
33*bbb1b6f9SApple OSS Distributions 			fprintf(stderr, "Helper rusage flag check failed\n");
34*bbb1b6f9SApple OSS Distributions 			exit(1);
35*bbb1b6f9SApple OSS Distributions 		}
36*bbb1b6f9SApple OSS Distributions 	}
37*bbb1b6f9SApple OSS Distributions 
38*bbb1b6f9SApple OSS Distributions 	printf("%p\n", _dyld_get_shared_cache_range(&shared_cache_len));
39*bbb1b6f9SApple OSS Distributions 	exit(0);
40*bbb1b6f9SApple OSS Distributions }
41