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