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