1*d4514f0bSApple OSS Distributions #include <pthread.h>
2*d4514f0bSApple OSS Distributions #include <stdlib.h>
3*d4514f0bSApple OSS Distributions #include <stdio.h>
4*d4514f0bSApple OSS Distributions #include <unistd.h>
5*d4514f0bSApple OSS Distributions
6*d4514f0bSApple OSS Distributions #include <mach/mach.h>
7*d4514f0bSApple OSS Distributions #include <mach/mach_vm.h>
8*d4514f0bSApple OSS Distributions
9*d4514f0bSApple OSS Distributions static void*
loop(void * arg)10*d4514f0bSApple OSS Distributions loop(__attribute__ ((unused)) void *arg)
11*d4514f0bSApple OSS Distributions {
12*d4514f0bSApple OSS Distributions while (1) {
13*d4514f0bSApple OSS Distributions }
14*d4514f0bSApple OSS Distributions }
15*d4514f0bSApple OSS Distributions
16*d4514f0bSApple OSS Distributions
17*d4514f0bSApple OSS Distributions static int
run_additional_threads(int nthreads)18*d4514f0bSApple OSS Distributions run_additional_threads(int nthreads)
19*d4514f0bSApple OSS Distributions {
20*d4514f0bSApple OSS Distributions for (int i = 0; i < nthreads; i++) {
21*d4514f0bSApple OSS Distributions pthread_t pthread;
22*d4514f0bSApple OSS Distributions int err;
23*d4514f0bSApple OSS Distributions
24*d4514f0bSApple OSS Distributions err = pthread_create(&pthread, NULL, loop, NULL);
25*d4514f0bSApple OSS Distributions if (err) {
26*d4514f0bSApple OSS Distributions return err;
27*d4514f0bSApple OSS Distributions }
28*d4514f0bSApple OSS Distributions }
29*d4514f0bSApple OSS Distributions
30*d4514f0bSApple OSS Distributions return 0;
31*d4514f0bSApple OSS Distributions }
32*d4514f0bSApple OSS Distributions
33*d4514f0bSApple OSS Distributions static int
allocate_and_wire_memory(mach_vm_size_t size)34*d4514f0bSApple OSS Distributions allocate_and_wire_memory(mach_vm_size_t size)
35*d4514f0bSApple OSS Distributions {
36*d4514f0bSApple OSS Distributions int err;
37*d4514f0bSApple OSS Distributions task_t task = mach_task_self();
38*d4514f0bSApple OSS Distributions mach_vm_address_t addr;
39*d4514f0bSApple OSS Distributions
40*d4514f0bSApple OSS Distributions if (size <= 0) {
41*d4514f0bSApple OSS Distributions return 0;
42*d4514f0bSApple OSS Distributions }
43*d4514f0bSApple OSS Distributions
44*d4514f0bSApple OSS Distributions err = mach_vm_allocate(task, &addr, size, VM_FLAGS_ANYWHERE);
45*d4514f0bSApple OSS Distributions if (err != KERN_SUCCESS) {
46*d4514f0bSApple OSS Distributions printf("mach_vm_allocate returned non-zero: %s\n", mach_error_string(err));
47*d4514f0bSApple OSS Distributions return err;
48*d4514f0bSApple OSS Distributions }
49*d4514f0bSApple OSS Distributions err = mach_vm_protect(task, addr, size, 0, VM_PROT_READ | VM_PROT_WRITE);
50*d4514f0bSApple OSS Distributions if (err != KERN_SUCCESS) {
51*d4514f0bSApple OSS Distributions printf("mach_vm_protect returned non-zero: %s\n", mach_error_string(err));
52*d4514f0bSApple OSS Distributions return err;
53*d4514f0bSApple OSS Distributions }
54*d4514f0bSApple OSS Distributions host_t host_priv_port;
55*d4514f0bSApple OSS Distributions err = host_get_host_priv_port(mach_host_self(), &host_priv_port);
56*d4514f0bSApple OSS Distributions if (err != KERN_SUCCESS) {
57*d4514f0bSApple OSS Distributions printf("host_get_host_priv_port returned non-zero: %s\n", mach_error_string(err));
58*d4514f0bSApple OSS Distributions return err;
59*d4514f0bSApple OSS Distributions }
60*d4514f0bSApple OSS Distributions err = mach_vm_wire(host_priv_port, task, addr, size, VM_PROT_READ | VM_PROT_WRITE);
61*d4514f0bSApple OSS Distributions if (err != KERN_SUCCESS) {
62*d4514f0bSApple OSS Distributions printf("mach_vm_wire returned non-zero: %s\n", mach_error_string(err));
63*d4514f0bSApple OSS Distributions return err;
64*d4514f0bSApple OSS Distributions }
65*d4514f0bSApple OSS Distributions
66*d4514f0bSApple OSS Distributions return 0;
67*d4514f0bSApple OSS Distributions }
68*d4514f0bSApple OSS Distributions
69*d4514f0bSApple OSS Distributions int
main(int argc,char * argv[])70*d4514f0bSApple OSS Distributions main(int argc, char *argv[])
71*d4514f0bSApple OSS Distributions {
72*d4514f0bSApple OSS Distributions int nthreads = 0;
73*d4514f0bSApple OSS Distributions int err;
74*d4514f0bSApple OSS Distributions mach_vm_size_t wired_mem = 0;
75*d4514f0bSApple OSS Distributions
76*d4514f0bSApple OSS Distributions if (argc > 1) {
77*d4514f0bSApple OSS Distributions nthreads = (int)strtoul(argv[1], NULL, 10);
78*d4514f0bSApple OSS Distributions }
79*d4514f0bSApple OSS Distributions if (argc > 2) {
80*d4514f0bSApple OSS Distributions wired_mem = (mach_vm_size_t)strtoul(argv[2], NULL, 10);
81*d4514f0bSApple OSS Distributions }
82*d4514f0bSApple OSS Distributions
83*d4514f0bSApple OSS Distributions err = allocate_and_wire_memory(wired_mem);
84*d4514f0bSApple OSS Distributions if (err) {
85*d4514f0bSApple OSS Distributions return err;
86*d4514f0bSApple OSS Distributions }
87*d4514f0bSApple OSS Distributions
88*d4514f0bSApple OSS Distributions err = run_additional_threads(nthreads);
89*d4514f0bSApple OSS Distributions if (err) {
90*d4514f0bSApple OSS Distributions return err;
91*d4514f0bSApple OSS Distributions }
92*d4514f0bSApple OSS Distributions
93*d4514f0bSApple OSS Distributions return 0;
94*d4514f0bSApple OSS Distributions }
95