xref: /xnu-8792.81.2/tests/vm_memory_tests_src/vm_tests.c (revision 19c3b8c28c31cb8130e034cfb5df6bf9ba342d90)
1*19c3b8c2SApple OSS Distributions //
2*19c3b8c2SApple OSS Distributions //  vmremaptest.c
3*19c3b8c2SApple OSS Distributions //
4*19c3b8c2SApple OSS Distributions //  Created by Lionel Desai on 9/16/19.
5*19c3b8c2SApple OSS Distributions //  Copyright © 2019 Apple. All rights reserved.
6*19c3b8c2SApple OSS Distributions //
7*19c3b8c2SApple OSS Distributions 
8*19c3b8c2SApple OSS Distributions #include "mach_vm_tests.h"
9*19c3b8c2SApple OSS Distributions #include <sys/sysctl.h>
10*19c3b8c2SApple OSS Distributions 
11*19c3b8c2SApple OSS Distributions 
12*19c3b8c2SApple OSS Distributions #define TESTSZ (140 * 1024 * 1024ULL)
13*19c3b8c2SApple OSS Distributions 
14*19c3b8c2SApple OSS Distributions void
mach_vm_client(mach_port_t port)15*19c3b8c2SApple OSS Distributions mach_vm_client(mach_port_t port)
16*19c3b8c2SApple OSS Distributions {
17*19c3b8c2SApple OSS Distributions 	mach_port_t memport = MACH_PORT_NULL;
18*19c3b8c2SApple OSS Distributions 	mach_vm_address_t       src = 0, dest = 0, tmp = 0;
19*19c3b8c2SApple OSS Distributions 	mach_vm_size_t          size = 0;
20*19c3b8c2SApple OSS Distributions 	vm_prot_t cur_prot, max_prot;
21*19c3b8c2SApple OSS Distributions 	mach_port_name_t        lport = 0;
22*19c3b8c2SApple OSS Distributions 	kern_return_t           ret = 0;
23*19c3b8c2SApple OSS Distributions 	boolean_t                       copy = FALSE;
24*19c3b8c2SApple OSS Distributions 	mach_vm_offset_t misoffset = 0;
25*19c3b8c2SApple OSS Distributions 
26*19c3b8c2SApple OSS Distributions 	mach_msg_type_number_t countp;
27*19c3b8c2SApple OSS Distributions 	mach_msg_size_t messageSize = 0;
28*19c3b8c2SApple OSS Distributions 	ipc_message_t *message = NULL;
29*19c3b8c2SApple OSS Distributions 
30*19c3b8c2SApple OSS Distributions 	char buffer[PATH_MAX];
31*19c3b8c2SApple OSS Distributions 	ret = proc_pidpath(getpid(), buffer, sizeof(buffer));
32*19c3b8c2SApple OSS Distributions 	assert(ret != -1);
33*19c3b8c2SApple OSS Distributions 
34*19c3b8c2SApple OSS Distributions 	messageSize = sizeof(ipc_message_t) + sizeof(mach_msg_trailer_t) + 64;
35*19c3b8c2SApple OSS Distributions 	message = (ipc_message_t *)calloc(1, messageSize);
36*19c3b8c2SApple OSS Distributions 
37*19c3b8c2SApple OSS Distributions 	message->header.msgh_bits = MACH_MSGH_BITS_ZERO;
38*19c3b8c2SApple OSS Distributions 	message->header.msgh_size = messageSize;
39*19c3b8c2SApple OSS Distributions 	message->header.msgh_remote_port = MACH_PORT_NULL;
40*19c3b8c2SApple OSS Distributions 	message->header.msgh_local_port = port;
41*19c3b8c2SApple OSS Distributions 
42*19c3b8c2SApple OSS Distributions 	while (1) {
43*19c3b8c2SApple OSS Distributions 		/* Awaiting the pid/src. addr/size from the server so we know what to remap from where */
44*19c3b8c2SApple OSS Distributions 		ret = mach_msg(&message->header, MACH_RCV_MSG | MACH_RCV_LARGE, 0, messageSize, port, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
45*19c3b8c2SApple OSS Distributions 		if (ret == KERN_SUCCESS) {
46*19c3b8c2SApple OSS Distributions 			if (debug) {
47*19c3b8c2SApple OSS Distributions 				T_LOG("CLIENT: received info from server... 0x%llx, %lld, 0x%llx, %d - %d\n", message->address, message->size, message->misoffset, message->vm_op, message->copy);
48*19c3b8c2SApple OSS Distributions 			}
49*19c3b8c2SApple OSS Distributions 
50*19c3b8c2SApple OSS Distributions 			switch (message->vm_op) {
51*19c3b8c2SApple OSS Distributions 			case VM_OP_REMAP:
52*19c3b8c2SApple OSS Distributions 				ret = task_for_pid(mach_task_self(), (pid_t) message->pid, &lport);
53*19c3b8c2SApple OSS Distributions 				T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "task_for_pid");
54*19c3b8c2SApple OSS Distributions 
55*19c3b8c2SApple OSS Distributions 				copy = message->copy;
56*19c3b8c2SApple OSS Distributions 				size = message->size;
57*19c3b8c2SApple OSS Distributions 				src = message->address;
58*19c3b8c2SApple OSS Distributions 				misoffset = 0;
59*19c3b8c2SApple OSS Distributions 
60*19c3b8c2SApple OSS Distributions 				ret = mach_vm_allocate(mach_task_self(), &tmp, size + 16384, VM_FLAGS_ANYWHERE);
61*19c3b8c2SApple OSS Distributions 				T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "mach_vm_allocate");
62*19c3b8c2SApple OSS Distributions 				mach_vm_deallocate(mach_task_self(), tmp, size + 16384);
63*19c3b8c2SApple OSS Distributions 
64*19c3b8c2SApple OSS Distributions 				dest = tmp + 4096;
65*19c3b8c2SApple OSS Distributions 
66*19c3b8c2SApple OSS Distributions 				ret = mach_vm_remap(mach_task_self(), &dest, size, 0, VM_FLAGS_ANYWHERE | VM_FLAGS_RETURN_DATA_ADDR,
67*19c3b8c2SApple OSS Distributions 				    lport, src, copy,
68*19c3b8c2SApple OSS Distributions 				    &cur_prot,
69*19c3b8c2SApple OSS Distributions 				    &max_prot,
70*19c3b8c2SApple OSS Distributions 				    VM_INHERIT_NONE);
71*19c3b8c2SApple OSS Distributions 
72*19c3b8c2SApple OSS Distributions 				if (ret) {
73*19c3b8c2SApple OSS Distributions 					char dstval[64];
74*19c3b8c2SApple OSS Distributions 					memcpy(dstval, (void*) dest, 64);
75*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_remap FAILED: %s -- src 0x%llx, dest 0x%llx (%s)\n", mach_error_string(ret), src, dest, dstval);
76*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_remap FAILED");
77*19c3b8c2SApple OSS Distributions 				}
78*19c3b8c2SApple OSS Distributions 
79*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*)dest, 64);
80*19c3b8c2SApple OSS Distributions 				break;
81*19c3b8c2SApple OSS Distributions 
82*19c3b8c2SApple OSS Distributions 			case VM_OP_READ_OVERWRITE:
83*19c3b8c2SApple OSS Distributions 				ret = task_for_pid(mach_task_self(), (pid_t) message->pid, &lport);
84*19c3b8c2SApple OSS Distributions 				T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "task_for_pid");
85*19c3b8c2SApple OSS Distributions 
86*19c3b8c2SApple OSS Distributions 				size = message->size;
87*19c3b8c2SApple OSS Distributions 				src = message->address;
88*19c3b8c2SApple OSS Distributions 				misoffset = 0;
89*19c3b8c2SApple OSS Distributions 
90*19c3b8c2SApple OSS Distributions 				mach_vm_size_t  dest_size = 0;
91*19c3b8c2SApple OSS Distributions 				ret = mach_vm_allocate(mach_task_self(), &tmp, size + 16384, VM_FLAGS_ANYWHERE);
92*19c3b8c2SApple OSS Distributions 				assert(KERN_SUCCESS == ret);
93*19c3b8c2SApple OSS Distributions 
94*19c3b8c2SApple OSS Distributions 				dest = tmp + 4096;
95*19c3b8c2SApple OSS Distributions 
96*19c3b8c2SApple OSS Distributions 				ret = mach_vm_read_overwrite(lport, src, size, dest, &dest_size);
97*19c3b8c2SApple OSS Distributions 
98*19c3b8c2SApple OSS Distributions 				if (ret) {
99*19c3b8c2SApple OSS Distributions 					char dstval[64];
100*19c3b8c2SApple OSS Distributions 					memcpy(dstval, (void*) dest, 64);
101*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_read_overwrite FAILED: %s -- src 0x%llx, dest 0x%llx (%s)\n", mach_error_string(ret), src, dest, dstval);
102*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_read_overwrite FAILED");
103*19c3b8c2SApple OSS Distributions 				}
104*19c3b8c2SApple OSS Distributions 
105*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*)dest, 64);
106*19c3b8c2SApple OSS Distributions 				break;
107*19c3b8c2SApple OSS Distributions 
108*19c3b8c2SApple OSS Distributions 			case VM_OP_READ:
109*19c3b8c2SApple OSS Distributions 				ret = task_for_pid(mach_task_self(), (pid_t) message->pid, &lport);
110*19c3b8c2SApple OSS Distributions 				T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "task_for_pid");
111*19c3b8c2SApple OSS Distributions 
112*19c3b8c2SApple OSS Distributions 				size = message->size;
113*19c3b8c2SApple OSS Distributions 				src = message->address;
114*19c3b8c2SApple OSS Distributions 				misoffset = 0;
115*19c3b8c2SApple OSS Distributions 
116*19c3b8c2SApple OSS Distributions 				ret = mach_vm_read(lport, src, size, (vm_offset_t*)&dest, &countp);
117*19c3b8c2SApple OSS Distributions 				if (ret) {
118*19c3b8c2SApple OSS Distributions 					char dstval[64];
119*19c3b8c2SApple OSS Distributions 					memcpy(dstval, (void*) dest, 64);
120*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_read FAILED: %s -- src 0x%llx, dest 0x%llx (%s)\n", mach_error_string(ret), src, dest, dstval);
121*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_read FAILED");
122*19c3b8c2SApple OSS Distributions 					exit(1);
123*19c3b8c2SApple OSS Distributions 				}
124*19c3b8c2SApple OSS Distributions 
125*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*)dest, 64);
126*19c3b8c2SApple OSS Distributions 				break;
127*19c3b8c2SApple OSS Distributions 
128*19c3b8c2SApple OSS Distributions #if 0
129*19c3b8c2SApple OSS Distributions 			case VM_OP_WRITE:
130*19c3b8c2SApple OSS Distributions 				ret = task_for_pid(mach_task_self(), (pid_t) message->pid, &lport);
131*19c3b8c2SApple OSS Distributions 				T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "task_for_pid");
132*19c3b8c2SApple OSS Distributions 
133*19c3b8c2SApple OSS Distributions 				size = message->size;
134*19c3b8c2SApple OSS Distributions 				src = message->address;
135*19c3b8c2SApple OSS Distributions 				misoffset = 0;
136*19c3b8c2SApple OSS Distributions 
137*19c3b8c2SApple OSS Distributions 				ret = mach_vm_write(lport, src, dest, countp);
138*19c3b8c2SApple OSS Distributions 				if (ret) {
139*19c3b8c2SApple OSS Distributions 					char dstval[64];
140*19c3b8c2SApple OSS Distributions 					memcpy(dstval, (void*) dest, 64);
141*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_write FAILED: %s -- src 0x%llx, dest 0x%llx (%s)\n", mach_error_string(ret), src, dest, dstval);
142*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_write FAILED");
143*19c3b8c2SApple OSS Distributions 				}
144*19c3b8c2SApple OSS Distributions 
145*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*)dest, 64);
146*19c3b8c2SApple OSS Distributions 				break;
147*19c3b8c2SApple OSS Distributions #endif
148*19c3b8c2SApple OSS Distributions 			case VM_OP_MEMENTRY:
149*19c3b8c2SApple OSS Distributions 				assert(message->body.msgh_descriptor_count == 1);
150*19c3b8c2SApple OSS Distributions 				dest = 0;
151*19c3b8c2SApple OSS Distributions 				size = message->size;
152*19c3b8c2SApple OSS Distributions 				memport = message->port_descriptor.name;
153*19c3b8c2SApple OSS Distributions 				copy = message->copy;
154*19c3b8c2SApple OSS Distributions 				if (copy) {
155*19c3b8c2SApple OSS Distributions 					misoffset = 0;
156*19c3b8c2SApple OSS Distributions 				} else {
157*19c3b8c2SApple OSS Distributions 					misoffset = message->misoffset;
158*19c3b8c2SApple OSS Distributions 				}
159*19c3b8c2SApple OSS Distributions 
160*19c3b8c2SApple OSS Distributions 				cur_prot = max_prot = VM_PROT_READ;
161*19c3b8c2SApple OSS Distributions #if 0
162*19c3b8c2SApple OSS Distributions 				/* This + VM_FLAGS_FIXED in mach_vm_map() will return KERN_INVALID_ARG expectedly */
163*19c3b8c2SApple OSS Distributions 				ret = mach_vm_allocate(mach_task_self(), &tmp, size + 16384, VM_FLAGS_ANYWHERE);
164*19c3b8c2SApple OSS Distributions 				dest = tmp + 4095;
165*19c3b8c2SApple OSS Distributions 				mach_vm_deallocate(mach_task_self(), tmp, size + 16384);
166*19c3b8c2SApple OSS Distributions #endif
167*19c3b8c2SApple OSS Distributions 				ret = mach_vm_map(mach_task_self(), &dest, size, 0 /*mask*/,
168*19c3b8c2SApple OSS Distributions 				    VM_FLAGS_ANYWHERE | VM_FLAGS_RETURN_DATA_ADDR,
169*19c3b8c2SApple OSS Distributions 				    memport, 0 /*offset*/, copy, cur_prot, max_prot, VM_INHERIT_NONE);
170*19c3b8c2SApple OSS Distributions 
171*19c3b8c2SApple OSS Distributions 				if (ret) {
172*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_map FAILED: %s -- port 0x%x\n", mach_error_string(ret), memport);
173*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_map FAILED");
174*19c3b8c2SApple OSS Distributions 				}
175*19c3b8c2SApple OSS Distributions 
176*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*)(dest + misoffset), 64);
177*19c3b8c2SApple OSS Distributions 				break;
178*19c3b8c2SApple OSS Distributions 
179*19c3b8c2SApple OSS Distributions 			case VM_OP_UNMAP:
180*19c3b8c2SApple OSS Distributions 				assert(dest);
181*19c3b8c2SApple OSS Distributions 				ret = mach_vm_deallocate(mach_task_self(), dest, size);
182*19c3b8c2SApple OSS Distributions 				if (ret) {
183*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT: mach_vm_deallocate FAILED: %s -- dest 0x%llx, size 0x%llx\n", mach_error_string(ret), dest, size);
184*19c3b8c2SApple OSS Distributions 					T_FAIL("CLIENT: mach_vm_deallocate FAILED");
185*19c3b8c2SApple OSS Distributions 				}
186*19c3b8c2SApple OSS Distributions 				/* No message to send here */
187*19c3b8c2SApple OSS Distributions 				continue;
188*19c3b8c2SApple OSS Distributions 
189*19c3b8c2SApple OSS Distributions 			case VM_OP_NONE:
190*19c3b8c2SApple OSS Distributions 				memcpy(message->value, (void*) (dest + misoffset), 64);
191*19c3b8c2SApple OSS Distributions 				break;
192*19c3b8c2SApple OSS Distributions 
193*19c3b8c2SApple OSS Distributions 			case VM_OP_EXIT:
194*19c3b8c2SApple OSS Distributions 				if (debug) {
195*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT EXITING ****** \n");
196*19c3b8c2SApple OSS Distributions 				}
197*19c3b8c2SApple OSS Distributions 				return;
198*19c3b8c2SApple OSS Distributions 
199*19c3b8c2SApple OSS Distributions 			case VM_OP_EXIT_ERROR:
200*19c3b8c2SApple OSS Distributions 				if (debug) {
201*19c3b8c2SApple OSS Distributions 					T_LOG("CLIENT EXITING WITH ERROR****** \n");
202*19c3b8c2SApple OSS Distributions 					T_FAIL("Revieved VM_OP_EXIT_ERROR");
203*19c3b8c2SApple OSS Distributions 				}
204*19c3b8c2SApple OSS Distributions 				return;
205*19c3b8c2SApple OSS Distributions 			default:
206*19c3b8c2SApple OSS Distributions 				break;
207*19c3b8c2SApple OSS Distributions 			}
208*19c3b8c2SApple OSS Distributions 
209*19c3b8c2SApple OSS Distributions 			char dstval[64];
210*19c3b8c2SApple OSS Distributions 			memcpy(dstval, (void*) message->value, 64);
211*19c3b8c2SApple OSS Distributions 			dstval[63] = '\0';
212*19c3b8c2SApple OSS Distributions 
213*19c3b8c2SApple OSS Distributions 			if (debug) {
214*19c3b8c2SApple OSS Distributions 				T_LOG("CLIENT: dest 0x%llx -> 0x%llx (0x%llx), *dest %s\n", dest, dest + misoffset, misoffset, dstval);
215*19c3b8c2SApple OSS Distributions 				/*memcpy(dstval, (void*) (dest + misoffset), 64);
216*19c3b8c2SApple OSS Distributions 				 *  dstval[63]='\0';
217*19c3b8c2SApple OSS Distributions 				 *  T_LOG("*dest %s\n", dstval);*/
218*19c3b8c2SApple OSS Distributions 			}
219*19c3b8c2SApple OSS Distributions 
220*19c3b8c2SApple OSS Distributions 			message->header.msgh_local_port = MACH_PORT_NULL;
221*19c3b8c2SApple OSS Distributions 
222*19c3b8c2SApple OSS Distributions 			ret = mach_msg(&message->header, MACH_SEND_MSG, message->header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
223*19c3b8c2SApple OSS Distributions 			T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "CLIENT: mach_msg_send FAILED");
224*19c3b8c2SApple OSS Distributions 		} else {
225*19c3b8c2SApple OSS Distributions 			T_QUIET; T_ASSERT_MACH_SUCCESS(ret, "CLIENT: mach_msg_rcv FAILED");
226*19c3b8c2SApple OSS Distributions 		}
227*19c3b8c2SApple OSS Distributions 	}
228*19c3b8c2SApple OSS Distributions }
229*19c3b8c2SApple OSS Distributions 
230*19c3b8c2SApple OSS Distributions void
mach_server_make_memory_entry(mach_port_t replyPort)231*19c3b8c2SApple OSS Distributions mach_server_make_memory_entry(mach_port_t replyPort)
232*19c3b8c2SApple OSS Distributions {
233*19c3b8c2SApple OSS Distributions 	mach_vm_address_t       src = 0, lsrc = 0;
234*19c3b8c2SApple OSS Distributions 	mach_vm_size_t          size = TESTSZ;
235*19c3b8c2SApple OSS Distributions 	memory_object_size_t memsz = 0;
236*19c3b8c2SApple OSS Distributions 	kern_return_t           kr;
237*19c3b8c2SApple OSS Distributions 	boolean_t                       modified_in_server = FALSE, perm_changed = FALSE;
238*19c3b8c2SApple OSS Distributions 	ipc_message_t               message;
239*19c3b8c2SApple OSS Distributions 	ipc_message_t               *reply = NULL;
240*19c3b8c2SApple OSS Distributions 	char                            src_val[64], dst_val[64];
241*19c3b8c2SApple OSS Distributions 	mach_msg_size_t             replySize = 0;
242*19c3b8c2SApple OSS Distributions 	void                            *buffer = NULL;
243*19c3b8c2SApple OSS Distributions 	int                                     flags = 0;
244*19c3b8c2SApple OSS Distributions 	mach_port_t                     memport = 0;
245*19c3b8c2SApple OSS Distributions 	int                                     mementry_pass_idx = 0;
246*19c3b8c2SApple OSS Distributions 	mach_vm_offset_t        misoffset = 0;
247*19c3b8c2SApple OSS Distributions 
248*19c3b8c2SApple OSS Distributions 	if (debug) {
249*19c3b8c2SApple OSS Distributions 		T_LOG("\n*************** make_memory_entry_test START ***************\n");
250*19c3b8c2SApple OSS Distributions 	}
251*19c3b8c2SApple OSS Distributions 
252*19c3b8c2SApple OSS Distributions 	if (mach_server_data_setup(&buffer) != 0) {
253*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
254*19c3b8c2SApple OSS Distributions 	}
255*19c3b8c2SApple OSS Distributions 
256*19c3b8c2SApple OSS Distributions 	if (buffer == NULL) {
257*19c3b8c2SApple OSS Distributions 		mach_server_data_cleanup(NULL, 0, 0);
258*19c3b8c2SApple OSS Distributions 		exit(0);
259*19c3b8c2SApple OSS Distributions 	}
260*19c3b8c2SApple OSS Distributions 
261*19c3b8c2SApple OSS Distributions 	replySize = sizeof(ipc_message_t) + sizeof(mach_msg_trailer_t) + 64;
262*19c3b8c2SApple OSS Distributions 	reply = calloc(1, replySize);
263*19c3b8c2SApple OSS Distributions 
264*19c3b8c2SApple OSS Distributions test_different_mementry_mode:
265*19c3b8c2SApple OSS Distributions 	/* create message to send over rights/address/pid/size */
266*19c3b8c2SApple OSS Distributions 	mach_server_construct_header(&message, replyPort);
267*19c3b8c2SApple OSS Distributions 
268*19c3b8c2SApple OSS Distributions 	/* allocation that we plan to remap in the client */
269*19c3b8c2SApple OSS Distributions 	mach_server_create_allocation(&src, size, buffer);
270*19c3b8c2SApple OSS Distributions 
271*19c3b8c2SApple OSS Distributions 	memsz = 8191;
272*19c3b8c2SApple OSS Distributions 	lsrc = src + 94095;
273*19c3b8c2SApple OSS Distributions 	int pgmask = (getpagesize() - 1);
274*19c3b8c2SApple OSS Distributions 	misoffset = 94095 - (94095 & ~pgmask);
275*19c3b8c2SApple OSS Distributions 
276*19c3b8c2SApple OSS Distributions 	if (mementry_pass_idx < 2) {
277*19c3b8c2SApple OSS Distributions 		if (mementry_pass_idx == 0) {
278*19c3b8c2SApple OSS Distributions 			flags = VM_PROT_DEFAULT | MAP_MEM_VM_COPY | MAP_MEM_USE_DATA_ADDR;
279*19c3b8c2SApple OSS Distributions 			T_LOG("mach_make_memory_entry VM_COPY | USE_DATA_ADDR test...");
280*19c3b8c2SApple OSS Distributions 		} else {
281*19c3b8c2SApple OSS Distributions 			flags = VM_PROT_READ | MAP_MEM_VM_SHARE;
282*19c3b8c2SApple OSS Distributions 			T_LOG("mach_make_memory_entry VM_SHARE test...");
283*19c3b8c2SApple OSS Distributions 		}
284*19c3b8c2SApple OSS Distributions 		kr = mach_vm_protect(mach_task_self(), (mach_vm_address_t) lsrc, (mach_vm_size_t)getpagesize(), FALSE, VM_PROT_READ);
285*19c3b8c2SApple OSS Distributions 		assert(kr == KERN_SUCCESS);
286*19c3b8c2SApple OSS Distributions 		perm_changed = TRUE;
287*19c3b8c2SApple OSS Distributions 	} else {
288*19c3b8c2SApple OSS Distributions 		flags = VM_PROT_DEFAULT;
289*19c3b8c2SApple OSS Distributions 		perm_changed = FALSE;
290*19c3b8c2SApple OSS Distributions 		T_LOG("mach_make_memory_entry DEFAULT test...");
291*19c3b8c2SApple OSS Distributions 	}
292*19c3b8c2SApple OSS Distributions 
293*19c3b8c2SApple OSS Distributions 	kr = mach_make_memory_entry_64(mach_task_self(), &memsz, lsrc, flags, &memport, MACH_PORT_NULL);
294*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
295*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: mach_make_memory_entry_64 try (%d) failed in Client: (%d) %s\n",
296*19c3b8c2SApple OSS Distributions 		    mementry_pass_idx + 1, kr, mach_error_string(kr));
297*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
298*19c3b8c2SApple OSS Distributions 	}
299*19c3b8c2SApple OSS Distributions 
300*19c3b8c2SApple OSS Distributions 	mach_server_contruct_payload(&message, lsrc, memport, memsz, misoffset, ((flags & MAP_MEM_VM_COPY) == MAP_MEM_VM_COPY) /*copy*/, VM_OP_MEMENTRY);
301*19c3b8c2SApple OSS Distributions 
302*19c3b8c2SApple OSS Distributions 	memcpy(src_val, (void*) lsrc, 64);
303*19c3b8c2SApple OSS Distributions 	src_val[63] = '\0';
304*19c3b8c2SApple OSS Distributions 
305*19c3b8c2SApple OSS Distributions check_again:
306*19c3b8c2SApple OSS Distributions 	/* Sending over pid/src address/size */
307*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
308*19c3b8c2SApple OSS Distributions 
309*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
310*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
311*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
312*19c3b8c2SApple OSS Distributions 	}
313*19c3b8c2SApple OSS Distributions 
314*19c3b8c2SApple OSS Distributions 	/* Ack from client that it worked */
315*19c3b8c2SApple OSS Distributions 
316*19c3b8c2SApple OSS Distributions 	bzero(reply, replySize);
317*19c3b8c2SApple OSS Distributions 
318*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&reply->header, MACH_RCV_MSG, 0, replySize, replyPort, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
319*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
320*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to get reply from client: (%d) %s\n", kr, mach_error_string(kr));
321*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
322*19c3b8c2SApple OSS Distributions 	}
323*19c3b8c2SApple OSS Distributions 
324*19c3b8c2SApple OSS Distributions 	memcpy(dst_val, &reply->value, 64);
325*19c3b8c2SApple OSS Distributions 	dst_val[63] = '\0';
326*19c3b8c2SApple OSS Distributions 
327*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
328*19c3b8c2SApple OSS Distributions 		if (strncmp(src_val, dst_val, 64)) {
329*19c3b8c2SApple OSS Distributions 			T_LOG("FAILED\n");
330*19c3b8c2SApple OSS Distributions 			T_LOG("(%d) Pre modification mach_make_memory_entry() FAILED: copy(%d) src_val: %s  dest_val: %s\n", mementry_pass_idx + 1, message.copy, src_val, dst_val);
331*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
332*19c3b8c2SApple OSS Distributions 		}
333*19c3b8c2SApple OSS Distributions 	} else {
334*19c3b8c2SApple OSS Distributions 		if (message.copy == TRUE) {
335*19c3b8c2SApple OSS Distributions 			if (strncmp(src_val, dst_val, 64) == 0) {
336*19c3b8c2SApple OSS Distributions 				T_LOG("FAILED\n");
337*19c3b8c2SApple OSS Distributions 				T_LOG("(%d) Data mismatch with Copy: %d src_val: %s  dest_val: %s\n",
338*19c3b8c2SApple OSS Distributions 				    mementry_pass_idx + 1, message.copy, src_val, dst_val);
339*19c3b8c2SApple OSS Distributions 				server_error_out(replyPort);
340*19c3b8c2SApple OSS Distributions 			}
341*19c3b8c2SApple OSS Distributions 		} else {
342*19c3b8c2SApple OSS Distributions 			if (strncmp(src_val, dst_val, 64)) {
343*19c3b8c2SApple OSS Distributions 				T_LOG("FAILED\n");
344*19c3b8c2SApple OSS Distributions 				T_LOG("(%d) Data mismatch with Copy: %d src_val: %s  dest_val: %s\n",
345*19c3b8c2SApple OSS Distributions 				    mementry_pass_idx + 1, message.copy, src_val, dst_val);
346*19c3b8c2SApple OSS Distributions 				server_error_out(replyPort);
347*19c3b8c2SApple OSS Distributions 			}
348*19c3b8c2SApple OSS Distributions 		}
349*19c3b8c2SApple OSS Distributions 	}
350*19c3b8c2SApple OSS Distributions 
351*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
352*19c3b8c2SApple OSS Distributions 		/* Now we change our data that has been mapped elsewhere */
353*19c3b8c2SApple OSS Distributions 		if (perm_changed) {
354*19c3b8c2SApple OSS Distributions 			kr = mach_vm_protect(mach_task_self(), (mach_vm_address_t) lsrc, (mach_vm_size_t)getpagesize(), FALSE, VM_PROT_READ | VM_PROT_WRITE);
355*19c3b8c2SApple OSS Distributions 			assert(kr == KERN_SUCCESS);
356*19c3b8c2SApple OSS Distributions 		}
357*19c3b8c2SApple OSS Distributions 
358*19c3b8c2SApple OSS Distributions 		memcpy((void*) lsrc, "THIS IS DIFFERENT -- BUT WE DON'T know if that's expecTED", 64);
359*19c3b8c2SApple OSS Distributions 
360*19c3b8c2SApple OSS Distributions 		if (perm_changed) {
361*19c3b8c2SApple OSS Distributions 			kr = mach_vm_protect(mach_task_self(), (mach_vm_address_t) lsrc, (mach_vm_size_t)getpagesize(), FALSE, VM_PROT_READ);
362*19c3b8c2SApple OSS Distributions 			assert(kr == KERN_SUCCESS);
363*19c3b8c2SApple OSS Distributions 		}
364*19c3b8c2SApple OSS Distributions 
365*19c3b8c2SApple OSS Distributions 		memcpy(src_val, (void*) lsrc, 64);
366*19c3b8c2SApple OSS Distributions 		src_val[63] = '\0';
367*19c3b8c2SApple OSS Distributions 		modified_in_server = TRUE;
368*19c3b8c2SApple OSS Distributions 		message.vm_op = VM_OP_NONE;
369*19c3b8c2SApple OSS Distributions 
370*19c3b8c2SApple OSS Distributions 		/* Check to see if the data in the other process is as expected */
371*19c3b8c2SApple OSS Distributions 		goto check_again;
372*19c3b8c2SApple OSS Distributions 	}
373*19c3b8c2SApple OSS Distributions 
374*19c3b8c2SApple OSS Distributions 	if (mementry_pass_idx < 2) {
375*19c3b8c2SApple OSS Distributions 		/* Next remap mode...so ask the other process to unmap the older mapping. */
376*19c3b8c2SApple OSS Distributions 		message.vm_op = VM_OP_UNMAP;
377*19c3b8c2SApple OSS Distributions 		kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
378*19c3b8c2SApple OSS Distributions 		if (kr != KERN_SUCCESS) {
379*19c3b8c2SApple OSS Distributions 			T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
380*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
381*19c3b8c2SApple OSS Distributions 		}
382*19c3b8c2SApple OSS Distributions 
383*19c3b8c2SApple OSS Distributions 		mach_port_deallocate(mach_task_self(), memport);
384*19c3b8c2SApple OSS Distributions 		memport = MACH_PORT_NULL;
385*19c3b8c2SApple OSS Distributions 		mach_vm_deallocate(mach_task_self(), src, size);
386*19c3b8c2SApple OSS Distributions 
387*19c3b8c2SApple OSS Distributions 		T_LOG("PASSED\n");
388*19c3b8c2SApple OSS Distributions 
389*19c3b8c2SApple OSS Distributions 		mementry_pass_idx++;
390*19c3b8c2SApple OSS Distributions 		modified_in_server = FALSE;
391*19c3b8c2SApple OSS Distributions 
392*19c3b8c2SApple OSS Distributions 		goto test_different_mementry_mode;
393*19c3b8c2SApple OSS Distributions 	}
394*19c3b8c2SApple OSS Distributions 
395*19c3b8c2SApple OSS Distributions 	T_LOG("PASSED\n");
396*19c3b8c2SApple OSS Distributions 
397*19c3b8c2SApple OSS Distributions 	/* Unmap old mapping in the other process. */
398*19c3b8c2SApple OSS Distributions 	message.vm_op = VM_OP_UNMAP;
399*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
400*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
401*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
402*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
403*19c3b8c2SApple OSS Distributions 	}
404*19c3b8c2SApple OSS Distributions 
405*19c3b8c2SApple OSS Distributions 	free(reply);
406*19c3b8c2SApple OSS Distributions 	reply = NULL;
407*19c3b8c2SApple OSS Distributions 
408*19c3b8c2SApple OSS Distributions 	mach_port_deallocate(mach_task_self(), memport);
409*19c3b8c2SApple OSS Distributions 	memport = MACH_PORT_NULL;
410*19c3b8c2SApple OSS Distributions 
411*19c3b8c2SApple OSS Distributions 	mach_server_data_cleanup(buffer, src, size);
412*19c3b8c2SApple OSS Distributions 	buffer = NULL;
413*19c3b8c2SApple OSS Distributions 
414*19c3b8c2SApple OSS Distributions 	if (debug) {
415*19c3b8c2SApple OSS Distributions 		T_LOG("*************** mach_make_memory_entry_test END ***************\n");
416*19c3b8c2SApple OSS Distributions 	}
417*19c3b8c2SApple OSS Distributions }
418*19c3b8c2SApple OSS Distributions 
419*19c3b8c2SApple OSS Distributions void
mach_server_read(mach_port_t replyPort,int op)420*19c3b8c2SApple OSS Distributions mach_server_read(mach_port_t replyPort, int op)
421*19c3b8c2SApple OSS Distributions {
422*19c3b8c2SApple OSS Distributions 	mach_vm_address_t       src;
423*19c3b8c2SApple OSS Distributions 	mach_vm_size_t          size = TESTSZ;
424*19c3b8c2SApple OSS Distributions 	kern_return_t           kr;
425*19c3b8c2SApple OSS Distributions 	boolean_t                       modified_in_server = FALSE;
426*19c3b8c2SApple OSS Distributions 	ipc_message_t               message;
427*19c3b8c2SApple OSS Distributions 	char                            src_val[64], dst_val[64];
428*19c3b8c2SApple OSS Distributions 	mach_msg_size_t             replySize = 0;
429*19c3b8c2SApple OSS Distributions 	ipc_message_t               *reply = NULL;
430*19c3b8c2SApple OSS Distributions 	void                            *buffer = NULL;
431*19c3b8c2SApple OSS Distributions 
432*19c3b8c2SApple OSS Distributions 	if (debug) {
433*19c3b8c2SApple OSS Distributions 		T_LOG("\n*************** vm_read / write / overwrite_test START ***************\n");
434*19c3b8c2SApple OSS Distributions 	}
435*19c3b8c2SApple OSS Distributions 
436*19c3b8c2SApple OSS Distributions 	{
437*19c3b8c2SApple OSS Distributions 		char opname[16];
438*19c3b8c2SApple OSS Distributions 		if (op == VM_OP_READ) {
439*19c3b8c2SApple OSS Distributions 			strlcpy(opname, "read", 5);
440*19c3b8c2SApple OSS Distributions 		}
441*19c3b8c2SApple OSS Distributions 		if (op == VM_OP_WRITE) {
442*19c3b8c2SApple OSS Distributions 			strlcpy(opname, "write", 6);
443*19c3b8c2SApple OSS Distributions 		}
444*19c3b8c2SApple OSS Distributions 		if (op == VM_OP_READ_OVERWRITE) {
445*19c3b8c2SApple OSS Distributions 			strlcpy(opname, "read_overwrite", 15);
446*19c3b8c2SApple OSS Distributions 		}
447*19c3b8c2SApple OSS Distributions 
448*19c3b8c2SApple OSS Distributions 		T_LOG("vm_%s test...", opname);
449*19c3b8c2SApple OSS Distributions 	}
450*19c3b8c2SApple OSS Distributions 
451*19c3b8c2SApple OSS Distributions 	if (mach_server_data_setup(&buffer) != 0) {
452*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
453*19c3b8c2SApple OSS Distributions 	}
454*19c3b8c2SApple OSS Distributions 
455*19c3b8c2SApple OSS Distributions 	if (buffer == NULL) {
456*19c3b8c2SApple OSS Distributions 		mach_server_data_cleanup(NULL, 0, 0);
457*19c3b8c2SApple OSS Distributions 		exit(0);
458*19c3b8c2SApple OSS Distributions 	}
459*19c3b8c2SApple OSS Distributions 
460*19c3b8c2SApple OSS Distributions 	replySize = sizeof(ipc_message_t) + sizeof(mach_msg_trailer_t) + 64;
461*19c3b8c2SApple OSS Distributions 	reply = calloc(1, replySize);
462*19c3b8c2SApple OSS Distributions 
463*19c3b8c2SApple OSS Distributions 	/* create message to send over rights/address/pid/size */
464*19c3b8c2SApple OSS Distributions 	mach_server_construct_header(&message, replyPort);
465*19c3b8c2SApple OSS Distributions 
466*19c3b8c2SApple OSS Distributions 	/* allocation that we plan to remap in the client */
467*19c3b8c2SApple OSS Distributions 	mach_server_create_allocation(&src, size, buffer);
468*19c3b8c2SApple OSS Distributions 
469*19c3b8c2SApple OSS Distributions 	mach_server_contruct_payload(&message, src, MACH_PORT_NULL /* port */, size, 0, TRUE /*copy*/, op);
470*19c3b8c2SApple OSS Distributions 	if (debug) {
471*19c3b8c2SApple OSS Distributions 		T_LOG("server COPY: Sending 0x%llx, %d, 0x%llx\n", message.address, getpid(), message.size);
472*19c3b8c2SApple OSS Distributions 	}
473*19c3b8c2SApple OSS Distributions 	memcpy(src_val, (void*)message.address, 64);
474*19c3b8c2SApple OSS Distributions 
475*19c3b8c2SApple OSS Distributions check_again:
476*19c3b8c2SApple OSS Distributions 	/* Sending over pid/src address/size */
477*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
478*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
479*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
480*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
481*19c3b8c2SApple OSS Distributions 	}
482*19c3b8c2SApple OSS Distributions 
483*19c3b8c2SApple OSS Distributions 	/* Ack from client that it worked */
484*19c3b8c2SApple OSS Distributions 
485*19c3b8c2SApple OSS Distributions 	bzero(reply, replySize);
486*19c3b8c2SApple OSS Distributions 
487*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&reply->header, MACH_RCV_MSG, 0, replySize, replyPort, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
488*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
489*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to get reply from client: (%d) %s\n", kr, mach_error_string(kr));
490*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
491*19c3b8c2SApple OSS Distributions 	}
492*19c3b8c2SApple OSS Distributions 
493*19c3b8c2SApple OSS Distributions 	memcpy(dst_val, &reply->value, 64);
494*19c3b8c2SApple OSS Distributions 
495*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
496*19c3b8c2SApple OSS Distributions 		if (strncmp(src_val, dst_val, 64)) {
497*19c3b8c2SApple OSS Distributions 			T_LOG("Pre modification (op: %d) FAILED: src_val: %s  dest_val: %s\n", op, src_val, dst_val);
498*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
499*19c3b8c2SApple OSS Distributions 		}
500*19c3b8c2SApple OSS Distributions 	} else {
501*19c3b8c2SApple OSS Distributions 		if (strncmp(src_val, dst_val, 64) == 0) {
502*19c3b8c2SApple OSS Distributions 			T_LOG("Data mismatch (op:%d) with Copy: %d src_val: %s  dest_val: %s\n", op, message.copy, src_val, dst_val);
503*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
504*19c3b8c2SApple OSS Distributions 		}
505*19c3b8c2SApple OSS Distributions 	}
506*19c3b8c2SApple OSS Distributions 
507*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
508*19c3b8c2SApple OSS Distributions 		/* Now we change our data that has been mapped elsewhere */
509*19c3b8c2SApple OSS Distributions 		memcpy((void*)message.address, "THIS IS DIFFERENT -- BUT WE DON'T know if that's expecTED", 64);
510*19c3b8c2SApple OSS Distributions 		memcpy(src_val, (void*)message.address, 64);
511*19c3b8c2SApple OSS Distributions 		modified_in_server = TRUE;
512*19c3b8c2SApple OSS Distributions 		message.vm_op = VM_OP_NONE;
513*19c3b8c2SApple OSS Distributions 
514*19c3b8c2SApple OSS Distributions 		/* Check to see if the data in the other process is as expected */
515*19c3b8c2SApple OSS Distributions 		goto check_again;
516*19c3b8c2SApple OSS Distributions 	}
517*19c3b8c2SApple OSS Distributions 
518*19c3b8c2SApple OSS Distributions 	/* Unmap old mapping in the other process. */
519*19c3b8c2SApple OSS Distributions 	message.vm_op = VM_OP_UNMAP;
520*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
521*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
522*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
523*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
524*19c3b8c2SApple OSS Distributions 	}
525*19c3b8c2SApple OSS Distributions 
526*19c3b8c2SApple OSS Distributions 	free(reply);
527*19c3b8c2SApple OSS Distributions 	reply = NULL;
528*19c3b8c2SApple OSS Distributions 
529*19c3b8c2SApple OSS Distributions 	mach_server_data_cleanup(buffer, src, size);
530*19c3b8c2SApple OSS Distributions 	buffer = NULL;
531*19c3b8c2SApple OSS Distributions 
532*19c3b8c2SApple OSS Distributions 	if (debug) {
533*19c3b8c2SApple OSS Distributions 		T_LOG("*************** vm_read_test END ***************\n");
534*19c3b8c2SApple OSS Distributions 	}
535*19c3b8c2SApple OSS Distributions 
536*19c3b8c2SApple OSS Distributions 	T_LOG("PASSED\n");
537*19c3b8c2SApple OSS Distributions }
538*19c3b8c2SApple OSS Distributions 
539*19c3b8c2SApple OSS Distributions void
mach_server_remap(mach_port_t replyPort)540*19c3b8c2SApple OSS Distributions mach_server_remap(mach_port_t replyPort)
541*19c3b8c2SApple OSS Distributions {
542*19c3b8c2SApple OSS Distributions 	mach_vm_address_t       src = 0, lsrc = 0;
543*19c3b8c2SApple OSS Distributions 	mach_vm_size_t          size = TESTSZ;
544*19c3b8c2SApple OSS Distributions 	kern_return_t           kr;
545*19c3b8c2SApple OSS Distributions 	int                                     remap_copy_pass_idx = 0;
546*19c3b8c2SApple OSS Distributions 	boolean_t                       modified_in_server = FALSE;
547*19c3b8c2SApple OSS Distributions 	void                            *buffer;
548*19c3b8c2SApple OSS Distributions 	ipc_message_t               message;
549*19c3b8c2SApple OSS Distributions 	char                            src_val[64], dst_val[64];
550*19c3b8c2SApple OSS Distributions 	mach_msg_size_t             replySize = 0;
551*19c3b8c2SApple OSS Distributions 	ipc_message_t               *reply = NULL;
552*19c3b8c2SApple OSS Distributions 
553*19c3b8c2SApple OSS Distributions 	if (debug) {
554*19c3b8c2SApple OSS Distributions 		T_LOG("\n*************** vm_remap_test START ***************\n");
555*19c3b8c2SApple OSS Distributions 	}
556*19c3b8c2SApple OSS Distributions 
557*19c3b8c2SApple OSS Distributions 	if (mach_server_data_setup(&buffer) != 0) {
558*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
559*19c3b8c2SApple OSS Distributions 	}
560*19c3b8c2SApple OSS Distributions 
561*19c3b8c2SApple OSS Distributions 	if (buffer == NULL) {
562*19c3b8c2SApple OSS Distributions 		mach_server_data_cleanup(NULL, 0, 0);
563*19c3b8c2SApple OSS Distributions 		exit(0);
564*19c3b8c2SApple OSS Distributions 	}
565*19c3b8c2SApple OSS Distributions 
566*19c3b8c2SApple OSS Distributions 	replySize = sizeof(ipc_message_t) + sizeof(mach_msg_trailer_t) + 64;
567*19c3b8c2SApple OSS Distributions 	reply = calloc(1, replySize);
568*19c3b8c2SApple OSS Distributions 
569*19c3b8c2SApple OSS Distributions remap_again:
570*19c3b8c2SApple OSS Distributions 
571*19c3b8c2SApple OSS Distributions 	T_LOG("vm_remap (copy = %s) test...", ((remap_copy_pass_idx == 0) ? "FALSE" : "TRUE"));
572*19c3b8c2SApple OSS Distributions 
573*19c3b8c2SApple OSS Distributions 	/* create message to send over rights/address/pid/size */
574*19c3b8c2SApple OSS Distributions 	mach_server_construct_header(&message, replyPort);
575*19c3b8c2SApple OSS Distributions 
576*19c3b8c2SApple OSS Distributions 	/* server allocation that we plan to remap in the client */
577*19c3b8c2SApple OSS Distributions 	mach_server_create_allocation(&src, size, buffer);
578*19c3b8c2SApple OSS Distributions 
579*19c3b8c2SApple OSS Distributions 	lsrc = src + 8193;
580*19c3b8c2SApple OSS Distributions 
581*19c3b8c2SApple OSS Distributions 	mach_server_contruct_payload(&message, lsrc, MACH_PORT_NULL /* port */, size - 9000, 0, remap_copy_pass_idx /*copy*/, VM_OP_REMAP);
582*19c3b8c2SApple OSS Distributions 	if (debug) {
583*19c3b8c2SApple OSS Distributions 		T_LOG("server COPY: Sending 0x%llx, %d, 0x%llx\n", message.address, getpid(), message.size);
584*19c3b8c2SApple OSS Distributions 	}
585*19c3b8c2SApple OSS Distributions 
586*19c3b8c2SApple OSS Distributions 	memcpy(src_val, (void*)lsrc, 64);
587*19c3b8c2SApple OSS Distributions 	src_val[63] = '\0';
588*19c3b8c2SApple OSS Distributions 
589*19c3b8c2SApple OSS Distributions check_again:
590*19c3b8c2SApple OSS Distributions 	/* Sending over pid/src address/size */
591*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
592*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
593*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
594*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
595*19c3b8c2SApple OSS Distributions 	}
596*19c3b8c2SApple OSS Distributions 
597*19c3b8c2SApple OSS Distributions 	/* Ack from client that it worked */
598*19c3b8c2SApple OSS Distributions 
599*19c3b8c2SApple OSS Distributions 	bzero(reply, replySize);
600*19c3b8c2SApple OSS Distributions 
601*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&reply->header, MACH_RCV_MSG, 0, replySize, replyPort, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
602*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
603*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to get reply from client: (%d) %s\n", kr, mach_error_string(kr));
604*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
605*19c3b8c2SApple OSS Distributions 	}
606*19c3b8c2SApple OSS Distributions 
607*19c3b8c2SApple OSS Distributions 	memcpy(dst_val, &reply->value, 64);
608*19c3b8c2SApple OSS Distributions 	dst_val[63] = '\0';
609*19c3b8c2SApple OSS Distributions 
610*19c3b8c2SApple OSS Distributions 
611*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
612*19c3b8c2SApple OSS Distributions 		if (strncmp(src_val, dst_val, 64)) {
613*19c3b8c2SApple OSS Distributions 			T_LOG("Pre modification remap() FAILED: copy(%d) src_val: %s  dest_val: %s\n",
614*19c3b8c2SApple OSS Distributions 			    message.copy, src_val, dst_val);
615*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
616*19c3b8c2SApple OSS Distributions 		}
617*19c3b8c2SApple OSS Distributions 	} else {
618*19c3b8c2SApple OSS Distributions 		if (message.copy == TRUE) {
619*19c3b8c2SApple OSS Distributions 			if (strcmp(src_val, dst_val) == 0) {
620*19c3b8c2SApple OSS Distributions 				T_LOG("Data mismatch with Copy: %d src_val: %s  dest_val: %s\n",
621*19c3b8c2SApple OSS Distributions 				    message.copy, src_val, dst_val);
622*19c3b8c2SApple OSS Distributions 				server_error_out(replyPort);
623*19c3b8c2SApple OSS Distributions 			}
624*19c3b8c2SApple OSS Distributions 		} else {
625*19c3b8c2SApple OSS Distributions 			if (strcmp(src_val, dst_val)) {
626*19c3b8c2SApple OSS Distributions 				T_LOG("Data mismatch with Copy: %d src_val: %s  dest_val: %s\n",
627*19c3b8c2SApple OSS Distributions 				    message.copy, src_val, dst_val);
628*19c3b8c2SApple OSS Distributions 				server_error_out(replyPort);
629*19c3b8c2SApple OSS Distributions 			}
630*19c3b8c2SApple OSS Distributions 		}
631*19c3b8c2SApple OSS Distributions 	}
632*19c3b8c2SApple OSS Distributions 
633*19c3b8c2SApple OSS Distributions 	if (modified_in_server == FALSE) {
634*19c3b8c2SApple OSS Distributions 		/* Now we change our data that has been mapped elsewhere */
635*19c3b8c2SApple OSS Distributions 		memcpy((void*)message.address, "THIS IS DIFFERENT -- BUT WE DON'T know if that's expecTED", 64);
636*19c3b8c2SApple OSS Distributions 		memcpy(src_val, (void*)message.address, 64);
637*19c3b8c2SApple OSS Distributions 		src_val[63] = '\0';
638*19c3b8c2SApple OSS Distributions 
639*19c3b8c2SApple OSS Distributions 		modified_in_server = TRUE;
640*19c3b8c2SApple OSS Distributions 		message.vm_op = VM_OP_NONE;
641*19c3b8c2SApple OSS Distributions 
642*19c3b8c2SApple OSS Distributions 		/* Check to see if the data in the other process is as expected */
643*19c3b8c2SApple OSS Distributions 		goto check_again;
644*19c3b8c2SApple OSS Distributions 	}
645*19c3b8c2SApple OSS Distributions 
646*19c3b8c2SApple OSS Distributions 	if (remap_copy_pass_idx == 0) {
647*19c3b8c2SApple OSS Distributions 		/* Next remap mode...so ask the other process to unmap the older mapping. */
648*19c3b8c2SApple OSS Distributions 		message.vm_op = VM_OP_UNMAP;
649*19c3b8c2SApple OSS Distributions 		kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
650*19c3b8c2SApple OSS Distributions 		if (kr != KERN_SUCCESS) {
651*19c3b8c2SApple OSS Distributions 			T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
652*19c3b8c2SApple OSS Distributions 			server_error_out(replyPort);
653*19c3b8c2SApple OSS Distributions 		}
654*19c3b8c2SApple OSS Distributions 
655*19c3b8c2SApple OSS Distributions 		mach_vm_deallocate(mach_task_self(), src, size);
656*19c3b8c2SApple OSS Distributions 
657*19c3b8c2SApple OSS Distributions 		T_LOG("PASSED\n");
658*19c3b8c2SApple OSS Distributions 
659*19c3b8c2SApple OSS Distributions 		remap_copy_pass_idx++;
660*19c3b8c2SApple OSS Distributions 		modified_in_server = FALSE;
661*19c3b8c2SApple OSS Distributions 
662*19c3b8c2SApple OSS Distributions 		/* Next remap pass to test (copy == TRUE). Send data out again to the other process to remap. */
663*19c3b8c2SApple OSS Distributions 		goto remap_again;
664*19c3b8c2SApple OSS Distributions 	}
665*19c3b8c2SApple OSS Distributions 
666*19c3b8c2SApple OSS Distributions 	T_LOG("PASSED\n");
667*19c3b8c2SApple OSS Distributions 
668*19c3b8c2SApple OSS Distributions 	/* Unmap old mapping in the other process. */
669*19c3b8c2SApple OSS Distributions 	message.vm_op = VM_OP_UNMAP;
670*19c3b8c2SApple OSS Distributions 	kr = mach_msg(&message.header, MACH_SEND_MSG, message.header.msgh_size, 0, MACH_PORT_NULL, MACH_MSG_TIMEOUT_NONE, MACH_PORT_NULL);
671*19c3b8c2SApple OSS Distributions 	if (kr != KERN_SUCCESS) {
672*19c3b8c2SApple OSS Distributions 		T_LOG("ERROR: Failed to send message to client: (%d) %s\n", kr, mach_error_string(kr));
673*19c3b8c2SApple OSS Distributions 		server_error_out(replyPort);
674*19c3b8c2SApple OSS Distributions 	}
675*19c3b8c2SApple OSS Distributions 
676*19c3b8c2SApple OSS Distributions 	free(reply);
677*19c3b8c2SApple OSS Distributions 	reply = NULL;
678*19c3b8c2SApple OSS Distributions 
679*19c3b8c2SApple OSS Distributions 	mach_server_data_cleanup(buffer, src, size);
680*19c3b8c2SApple OSS Distributions 	buffer = NULL;
681*19c3b8c2SApple OSS Distributions 
682*19c3b8c2SApple OSS Distributions 	if (debug) {
683*19c3b8c2SApple OSS Distributions 		T_LOG("*************** vm_remap_test END ***************\n");
684*19c3b8c2SApple OSS Distributions 	}
685*19c3b8c2SApple OSS Distributions }
686