]> git.saurik.com Git - apple/xnu.git/blob - osfmk/ipc/mach_msg.c
xnu-6153.41.3.tar.gz
[apple/xnu.git] / osfmk / ipc / mach_msg.c
1 /*
2 * Copyright (c) 2000-2007 Apple Inc. All rights reserved.
3 *
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
14 *
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
25 *
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27 */
28 /*
29 * @OSF_COPYRIGHT@
30 */
31 /*
32 * Mach Operating System
33 * Copyright (c) 1991,1990,1989 Carnegie Mellon University
34 * All Rights Reserved.
35 *
36 * Permission to use, copy, modify and distribute this software and its
37 * documentation is hereby granted, provided that both the copyright
38 * notice and this permission notice appear in all copies of the
39 * software, derivative works or modified versions, and any portions
40 * thereof, and that both notices appear in supporting documentation.
41 *
42 * CARNEGIE MELLON ALLOWS FREE USE OF THIS SOFTWARE IN ITS "AS IS"
43 * CONDITION. CARNEGIE MELLON DISCLAIMS ANY LIABILITY OF ANY KIND FOR
44 * ANY DAMAGES WHATSOEVER RESULTING FROM THE USE OF THIS SOFTWARE.
45 *
46 * Carnegie Mellon requests users of this software to return to
47 *
48 * Software Distribution Coordinator or Software.Distribution@CS.CMU.EDU
49 * School of Computer Science
50 * Carnegie Mellon University
51 * Pittsburgh PA 15213-3890
52 *
53 * any improvements or extensions that they make and grant Carnegie Mellon
54 * the rights to redistribute these changes.
55 */
56 /*
57 * NOTICE: This file was modified by McAfee Research in 2004 to introduce
58 * support for mandatory and extensible security protections. This notice
59 * is included in support of clause 2.2 (b) of the Apple Public License,
60 * Version 2.0.
61 * Copyright (c) 2005 SPARTA, Inc.
62 */
63 /*
64 */
65 /*
66 * File: ipc/mach_msg.c
67 * Author: Rich Draves
68 * Date: 1989
69 *
70 * Exported message traps. See mach/message.h.
71 */
72
73 #include <mach/mach_types.h>
74 #include <mach/kern_return.h>
75 #include <mach/port.h>
76 #include <mach/message.h>
77 #include <mach/mig_errors.h>
78 #include <mach/mach_traps.h>
79
80 #include <kern/kern_types.h>
81 #include <kern/assert.h>
82 #include <kern/counters.h>
83 #include <kern/cpu_number.h>
84 #include <kern/ipc_kobject.h>
85 #include <kern/ipc_mig.h>
86 #include <kern/task.h>
87 #include <kern/thread.h>
88 #include <kern/sched_prim.h>
89 #include <kern/exception.h>
90 #include <kern/misc_protos.h>
91 #include <kern/kalloc.h>
92 #include <kern/processor.h>
93 #include <kern/syscall_subr.h>
94 #include <kern/policy_internal.h>
95
96 #include <vm/vm_map.h>
97
98 #include <ipc/port.h>
99 #include <ipc/ipc_types.h>
100 #include <ipc/ipc_kmsg.h>
101 #include <ipc/ipc_mqueue.h>
102 #include <ipc/ipc_object.h>
103 #include <ipc/ipc_notify.h>
104 #include <ipc/ipc_port.h>
105 #include <ipc/ipc_pset.h>
106 #include <ipc/ipc_space.h>
107 #include <ipc/ipc_entry.h>
108 #include <ipc/ipc_importance.h>
109 #include <ipc/ipc_voucher.h>
110
111 #include <machine/machine_routines.h>
112 #include <security/mac_mach_internal.h>
113
114 #include <sys/kdebug.h>
115
116 #ifndef offsetof
117 #define offsetof(type, member) ((size_t)(&((type *)0)->member))
118 #endif /* offsetof */
119
120 /*
121 * Forward declarations - kernel internal routines
122 */
123
124 mach_msg_return_t mach_msg_send(
125 mach_msg_header_t *msg,
126 mach_msg_option_t option,
127 mach_msg_size_t send_size,
128 mach_msg_timeout_t send_timeout,
129 mach_port_name_t notify);
130
131 mach_msg_return_t mach_msg_receive(
132 mach_msg_header_t *msg,
133 mach_msg_option_t option,
134 mach_msg_size_t rcv_size,
135 mach_port_name_t rcv_name,
136 mach_msg_timeout_t rcv_timeout,
137 void (*continuation)(mach_msg_return_t),
138 mach_msg_size_t slist_size);
139
140
141 mach_msg_return_t msg_receive_error(
142 ipc_kmsg_t kmsg,
143 mach_msg_option_t option,
144 mach_vm_address_t rcv_addr,
145 mach_msg_size_t rcv_size,
146 mach_port_seqno_t seqno,
147 ipc_space_t space,
148 mach_msg_size_t *out_size);
149
150 static mach_msg_return_t
151 mach_msg_rcv_link_special_reply_port(
152 ipc_port_t special_reply_port,
153 mach_port_name_t dest_name_port);
154
155 void
156 mach_msg_receive_results_complete(ipc_object_t object);
157
158 const security_token_t KERNEL_SECURITY_TOKEN = KERNEL_SECURITY_TOKEN_VALUE;
159 const audit_token_t KERNEL_AUDIT_TOKEN = KERNEL_AUDIT_TOKEN_VALUE;
160
161 mach_msg_format_0_trailer_t trailer_template = {
162 /* mach_msg_trailer_type_t */ MACH_MSG_TRAILER_FORMAT_0,
163 /* mach_msg_trailer_size_t */ MACH_MSG_TRAILER_MINIMUM_SIZE,
164 /* mach_port_seqno_t */ 0,
165 /* security_token_t */ KERNEL_SECURITY_TOKEN_VALUE
166 };
167
168 /*
169 * Routine: mach_msg_send [Kernel Internal]
170 * Purpose:
171 * Routine for kernel-task threads to send a message.
172 *
173 * Unlike mach_msg_send_from_kernel(), this routine
174 * looks port names up in the kernel's port namespace
175 * and copies in the kernel virtual memory (instead
176 * of taking a vm_map_copy_t pointer for OOL descriptors).
177 * Conditions:
178 * Nothing locked.
179 * Returns:
180 * MACH_MSG_SUCCESS Sent the message.
181 * MACH_SEND_MSG_TOO_SMALL Message smaller than a header.
182 * MACH_SEND_NO_BUFFER Couldn't allocate buffer.
183 * MACH_SEND_INVALID_DATA Couldn't copy message data.
184 * MACH_SEND_INVALID_HEADER
185 * Illegal value in the message header bits.
186 * MACH_SEND_INVALID_DEST The space is dead.
187 * MACH_SEND_INVALID_NOTIFY Bad notify port.
188 * MACH_SEND_INVALID_DEST Can't copyin destination port.
189 * MACH_SEND_INVALID_REPLY Can't copyin reply port.
190 * MACH_SEND_TIMED_OUT Timeout expired without delivery.
191 * MACH_SEND_INTERRUPTED Delivery interrupted.
192 */
193
194 mach_msg_return_t
195 mach_msg_send(
196 mach_msg_header_t *msg,
197 mach_msg_option_t option,
198 mach_msg_size_t send_size,
199 mach_msg_timeout_t send_timeout,
200 mach_msg_priority_t override)
201 {
202 ipc_space_t space = current_space();
203 vm_map_t map = current_map();
204 ipc_kmsg_t kmsg;
205 mach_msg_return_t mr;
206 mach_msg_size_t msg_and_trailer_size;
207 mach_msg_max_trailer_t *trailer;
208
209 option |= MACH_SEND_KERNEL;
210
211 if ((send_size & 3) ||
212 send_size < sizeof(mach_msg_header_t) ||
213 (send_size < sizeof(mach_msg_base_t) && (msg->msgh_bits & MACH_MSGH_BITS_COMPLEX))) {
214 return MACH_SEND_MSG_TOO_SMALL;
215 }
216
217 if (send_size > MACH_MSG_SIZE_MAX - MAX_TRAILER_SIZE) {
218 return MACH_SEND_TOO_LARGE;
219 }
220
221 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_START);
222
223 msg_and_trailer_size = send_size + MAX_TRAILER_SIZE;
224
225 kmsg = ipc_kmsg_alloc(msg_and_trailer_size);
226
227 if (kmsg == IKM_NULL) {
228 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, MACH_SEND_NO_BUFFER);
229 return MACH_SEND_NO_BUFFER;
230 }
231
232 KERNEL_DEBUG_CONSTANT(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_LINK) | DBG_FUNC_NONE,
233 (uintptr_t)0, /* this should only be called from the kernel! */
234 VM_KERNEL_ADDRPERM((uintptr_t)kmsg),
235 0, 0,
236 0);
237 (void) memcpy((void *) kmsg->ikm_header, (const void *) msg, send_size);
238
239 kmsg->ikm_header->msgh_size = send_size;
240
241 /*
242 * reserve for the trailer the largest space (MAX_TRAILER_SIZE)
243 * However, the internal size field of the trailer (msgh_trailer_size)
244 * is initialized to the minimum (sizeof(mach_msg_trailer_t)), to optimize
245 * the cases where no implicit data is requested.
246 */
247 trailer = (mach_msg_max_trailer_t *) ((vm_offset_t)kmsg->ikm_header + send_size);
248 trailer->msgh_sender = current_thread()->task->sec_token;
249 trailer->msgh_audit = current_thread()->task->audit_token;
250 trailer->msgh_trailer_type = MACH_MSG_TRAILER_FORMAT_0;
251 trailer->msgh_trailer_size = MACH_MSG_TRAILER_MINIMUM_SIZE;
252
253 mr = ipc_kmsg_copyin(kmsg, space, map, override, &option);
254
255 if (mr != MACH_MSG_SUCCESS) {
256 ipc_kmsg_free(kmsg);
257 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, mr);
258 return mr;
259 }
260
261 mr = ipc_kmsg_send(kmsg, option, send_timeout);
262
263 if (mr != MACH_MSG_SUCCESS) {
264 mr |= ipc_kmsg_copyout_pseudo(kmsg, space, map, MACH_MSG_BODY_NULL);
265 (void) memcpy((void *) msg, (const void *) kmsg->ikm_header,
266 kmsg->ikm_header->msgh_size);
267 ipc_kmsg_free(kmsg);
268 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, mr);
269 }
270
271 return mr;
272 }
273
274 /*
275 * message header as seen at user-space
276 * (for MACH_RCV_LARGE/IDENTITY updating)
277 */
278 typedef struct{
279 mach_msg_bits_t msgh_bits;
280 mach_msg_size_t msgh_size;
281 mach_port_name_t msgh_remote_port;
282 mach_port_name_t msgh_local_port;
283 mach_msg_size_t msgh_reserved;
284 mach_msg_id_t msgh_id;
285 } mach_msg_user_header_t;
286
287 /*
288 * Routine: mach_msg_receive_results
289 * Purpose:
290 * Receive a message.
291 * Conditions:
292 * Nothing locked.
293 * Returns:
294 * MACH_MSG_SUCCESS Received a message.
295 * MACH_RCV_INVALID_NAME The name doesn't denote a right,
296 * or the denoted right is not receive or port set.
297 * MACH_RCV_IN_SET Receive right is a member of a set.
298 * MACH_RCV_TOO_LARGE Message wouldn't fit into buffer.
299 * MACH_RCV_TIMED_OUT Timeout expired without a message.
300 * MACH_RCV_INTERRUPTED Reception interrupted.
301 * MACH_RCV_PORT_DIED Port/set died while receiving.
302 * MACH_RCV_PORT_CHANGED Port moved into set while receiving.
303 * MACH_RCV_INVALID_DATA Couldn't copy to user buffer.
304 * MACH_RCV_INVALID_NOTIFY Bad notify port.
305 * MACH_RCV_HEADER_ERROR
306 */
307
308 mach_msg_return_t
309 mach_msg_receive_results(
310 mach_msg_size_t *sizep)
311 {
312 thread_t self = current_thread();
313 ipc_space_t space = current_space();
314 vm_map_t map = current_map();
315
316 ipc_object_t object = self->ith_object;
317 mach_msg_return_t mr = self->ith_state;
318 mach_vm_address_t rcv_addr = self->ith_msg_addr;
319 mach_msg_size_t rcv_size = self->ith_rsize;
320 mach_msg_option_t option = self->ith_option;
321 ipc_kmsg_t kmsg = self->ith_kmsg;
322 mach_port_seqno_t seqno = self->ith_seqno;
323
324 mach_msg_trailer_size_t trailer_size;
325 mach_msg_size_t size = 0;
326
327 /*
328 * unlink the special_reply_port before releasing reference to object.
329 * get the thread's turnstile, if the thread donated it's turnstile to the port
330 */
331 mach_msg_receive_results_complete(object);
332 io_release(object);
333
334 if (mr != MACH_MSG_SUCCESS) {
335 if (mr == MACH_RCV_TOO_LARGE) {
336 /*
337 * If the receive operation occurs with MACH_RCV_LARGE set
338 * then no message was extracted from the queue, and the size
339 * and (optionally) receiver names were the only thing captured.
340 * Just copyout the size (and optional port name) in a fake
341 * header.
342 */
343 if (option & MACH_RCV_LARGE) {
344 if ((option & MACH_RCV_STACK) == 0 &&
345 rcv_size >= offsetof(mach_msg_user_header_t, msgh_reserved)) {
346 /*
347 * We need to inform the user-level code that it needs more
348 * space. The value for how much space was returned in the
349 * msize save area instead of the message (which was left on
350 * the queue).
351 */
352 if (option & MACH_RCV_LARGE_IDENTITY) {
353 if (copyout((char *) &self->ith_receiver_name,
354 rcv_addr + offsetof(mach_msg_user_header_t, msgh_local_port),
355 sizeof(mach_port_name_t))) {
356 mr = MACH_RCV_INVALID_DATA;
357 }
358 }
359 if (copyout((char *) &self->ith_msize,
360 rcv_addr + offsetof(mach_msg_user_header_t, msgh_size),
361 sizeof(mach_msg_size_t))) {
362 mr = MACH_RCV_INVALID_DATA;
363 }
364 }
365 } else {
366 /* discard importance in message */
367 ipc_importance_clean(kmsg);
368
369 if (msg_receive_error(kmsg, option, rcv_addr, rcv_size, seqno, space, &size)
370 == MACH_RCV_INVALID_DATA) {
371 mr = MACH_RCV_INVALID_DATA;
372 }
373 }
374 }
375
376 if (sizep) {
377 *sizep = size;
378 }
379 return mr;
380 }
381
382 /* MACH_MSG_SUCCESS */
383
384 #if IMPORTANCE_INHERITANCE
385
386 /* adopt/transform any importance attributes carried in the message */
387 ipc_importance_receive(kmsg, option);
388
389 #endif /* IMPORTANCE_INHERITANCE */
390
391 /* auto redeem the voucher in the message */
392 ipc_voucher_receive_postprocessing(kmsg, option);
393
394 trailer_size = ipc_kmsg_add_trailer(kmsg, space, option, self, seqno, FALSE,
395 kmsg->ikm_header->msgh_remote_port->ip_context);
396
397 mr = ipc_kmsg_copyout(kmsg, space, map, MACH_MSG_BODY_NULL, option);
398
399 if (mr != MACH_MSG_SUCCESS) {
400 /* already received importance, so have to undo that here */
401 ipc_importance_unreceive(kmsg, option);
402
403 /* if we had a body error copyout what we have, otherwise a simple header/trailer */
404 if ((mr & ~MACH_MSG_MASK) == MACH_RCV_BODY_ERROR) {
405 if (ipc_kmsg_put(kmsg, option, rcv_addr, rcv_size, trailer_size, &size) == MACH_RCV_INVALID_DATA) {
406 mr = MACH_RCV_INVALID_DATA;
407 }
408 } else {
409 if (msg_receive_error(kmsg, option, rcv_addr, rcv_size, seqno, space, &size)
410 == MACH_RCV_INVALID_DATA) {
411 mr = MACH_RCV_INVALID_DATA;
412 }
413 }
414 } else {
415 /* capture ksmg QoS values to the thread continuation state */
416 self->ith_qos = kmsg->ikm_qos;
417 self->ith_qos_override = kmsg->ikm_qos_override;
418 mr = ipc_kmsg_put(kmsg, option, rcv_addr, rcv_size, trailer_size, &size);
419 }
420
421 if (sizep) {
422 *sizep = size;
423 }
424 return mr;
425 }
426
427 /*
428 * Routine: mach_msg_receive [Kernel Internal]
429 * Purpose:
430 * Routine for kernel-task threads to actively receive a message.
431 *
432 * Unlike being dispatched to by ipc_kobject_server() or the
433 * reply part of mach_msg_rpc_from_kernel(), this routine
434 * looks up the receive port name in the kernel's port
435 * namespace and copies out received port rights to that namespace
436 * as well. Out-of-line memory is copied out the kernel's
437 * address space (rather than just providing the vm_map_copy_t).
438 * Conditions:
439 * Nothing locked.
440 * Returns:
441 * MACH_MSG_SUCCESS Received a message.
442 * See <mach/message.h> for list of MACH_RCV_XXX errors.
443 */
444 mach_msg_return_t
445 mach_msg_receive(
446 mach_msg_header_t *msg,
447 mach_msg_option_t option,
448 mach_msg_size_t rcv_size,
449 mach_port_name_t rcv_name,
450 mach_msg_timeout_t rcv_timeout,
451 void (*continuation)(mach_msg_return_t),
452 __unused mach_msg_size_t slist_size)
453 {
454 thread_t self = current_thread();
455 ipc_space_t space = current_space();
456 ipc_object_t object;
457 ipc_mqueue_t mqueue;
458 mach_msg_return_t mr;
459
460 mr = ipc_mqueue_copyin(space, rcv_name, &mqueue, &object);
461 if (mr != MACH_MSG_SUCCESS) {
462 return mr;
463 }
464 /* hold ref for object */
465
466 self->ith_msg_addr = CAST_DOWN(mach_vm_address_t, msg);
467 self->ith_object = object;
468 self->ith_rsize = rcv_size;
469 self->ith_msize = 0;
470 self->ith_option = option;
471 self->ith_continuation = continuation;
472 self->ith_knote = ITH_KNOTE_NULL;
473
474 ipc_mqueue_receive(mqueue, option, rcv_size, rcv_timeout, THREAD_ABORTSAFE);
475 if ((option & MACH_RCV_TIMEOUT) && rcv_timeout == 0) {
476 thread_poll_yield(self);
477 }
478 return mach_msg_receive_results(NULL);
479 }
480
481 void
482 mach_msg_receive_continue(void)
483 {
484 mach_msg_return_t mr;
485 thread_t self = current_thread();
486
487 if (self->ith_state == MACH_PEEK_READY) {
488 mr = MACH_PEEK_READY;
489 } else {
490 mr = mach_msg_receive_results(NULL);
491 }
492 (*self->ith_continuation)(mr);
493 }
494
495
496 /*
497 * Routine: mach_msg_overwrite_trap [mach trap]
498 * Purpose:
499 * Possibly send a message; possibly receive a message.
500 * Conditions:
501 * Nothing locked.
502 * Returns:
503 * All of mach_msg_send and mach_msg_receive error codes.
504 */
505
506 mach_msg_return_t
507 mach_msg_overwrite_trap(
508 struct mach_msg_overwrite_trap_args *args)
509 {
510 mach_vm_address_t msg_addr = args->msg;
511 mach_msg_option_t option = args->option;
512 mach_msg_size_t send_size = args->send_size;
513 mach_msg_size_t rcv_size = args->rcv_size;
514 mach_port_name_t rcv_name = args->rcv_name;
515 mach_msg_timeout_t msg_timeout = args->timeout;
516 mach_msg_priority_t override = args->override;
517 mach_vm_address_t rcv_msg_addr = args->rcv_msg;
518 __unused mach_port_seqno_t temp_seqno = 0;
519
520 mach_msg_return_t mr = MACH_MSG_SUCCESS;
521 vm_map_t map = current_map();
522
523 /* Only accept options allowed by the user */
524 option &= MACH_MSG_OPTION_USER;
525
526 if (option & MACH_SEND_MSG) {
527 ipc_space_t space = current_space();
528 ipc_kmsg_t kmsg;
529
530 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_START);
531
532 mr = ipc_kmsg_get(msg_addr, send_size, &kmsg);
533
534 if (mr != MACH_MSG_SUCCESS) {
535 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, mr);
536 return mr;
537 }
538
539 KERNEL_DEBUG_CONSTANT(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_LINK) | DBG_FUNC_NONE,
540 (uintptr_t)msg_addr,
541 VM_KERNEL_ADDRPERM((uintptr_t)kmsg),
542 0, 0,
543 0);
544
545 mr = ipc_kmsg_copyin(kmsg, space, map, override, &option);
546
547 if (mr != MACH_MSG_SUCCESS) {
548 ipc_kmsg_free(kmsg);
549 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, mr);
550 return mr;
551 }
552
553 mr = ipc_kmsg_send(kmsg, option, msg_timeout);
554
555 if (mr != MACH_MSG_SUCCESS) {
556 mr |= ipc_kmsg_copyout_pseudo(kmsg, space, map, MACH_MSG_BODY_NULL);
557 (void) ipc_kmsg_put(kmsg, option, msg_addr, send_size, 0, NULL);
558 KDBG(MACHDBG_CODE(DBG_MACH_IPC, MACH_IPC_KMSG_INFO) | DBG_FUNC_END, mr);
559 return mr;
560 }
561 }
562
563 if (option & MACH_RCV_MSG) {
564 thread_t self = current_thread();
565 ipc_space_t space = current_space();
566 ipc_object_t object;
567 ipc_mqueue_t mqueue;
568
569 mr = ipc_mqueue_copyin(space, rcv_name, &mqueue, &object);
570 if (mr != MACH_MSG_SUCCESS) {
571 return mr;
572 }
573 /* hold ref for object */
574
575 if ((option & MACH_RCV_SYNC_WAIT) && !(option & MACH_SEND_SYNC_OVERRIDE)) {
576 ipc_port_t special_reply_port;
577 special_reply_port = ip_object_to_port(object);
578 /* link the special reply port to the destination */
579 mr = mach_msg_rcv_link_special_reply_port(special_reply_port,
580 (mach_port_name_t)override);
581 if (mr != MACH_MSG_SUCCESS) {
582 io_release(object);
583 return mr;
584 }
585 }
586
587 if (rcv_msg_addr != (mach_vm_address_t)0) {
588 self->ith_msg_addr = rcv_msg_addr;
589 } else {
590 self->ith_msg_addr = msg_addr;
591 }
592 self->ith_object = object;
593 self->ith_rsize = rcv_size;
594 self->ith_msize = 0;
595 self->ith_option = option;
596 self->ith_receiver_name = MACH_PORT_NULL;
597 self->ith_continuation = thread_syscall_return;
598 self->ith_knote = ITH_KNOTE_NULL;
599
600 ipc_mqueue_receive(mqueue, option, rcv_size, msg_timeout, THREAD_ABORTSAFE);
601 if ((option & MACH_RCV_TIMEOUT) && msg_timeout == 0) {
602 thread_poll_yield(self);
603 }
604 return mach_msg_receive_results(NULL);
605 }
606
607 return MACH_MSG_SUCCESS;
608 }
609
610 /*
611 * Routine: mach_msg_rcv_link_special_reply_port
612 * Purpose:
613 * Link the special reply port(rcv right) to the
614 * other end of the sync ipc channel.
615 * Conditions:
616 * Nothing locked.
617 * Returns:
618 * None.
619 */
620 static mach_msg_return_t
621 mach_msg_rcv_link_special_reply_port(
622 ipc_port_t special_reply_port,
623 mach_port_name_t dest_name_port)
624 {
625 ipc_port_t dest_port = IP_NULL;
626 kern_return_t kr;
627
628 if (current_thread()->ith_special_reply_port != special_reply_port) {
629 return MACH_RCV_INVALID_NOTIFY;
630 }
631
632 /* Copyin the destination port */
633 if (!MACH_PORT_VALID(dest_name_port)) {
634 return MACH_RCV_INVALID_NOTIFY;
635 }
636
637 kr = ipc_port_translate_send(current_space(), dest_name_port, &dest_port);
638 if (kr == KERN_SUCCESS) {
639 ip_reference(dest_port);
640 ip_unlock(dest_port);
641
642 /*
643 * The receive right of dest port might have gone away,
644 * do not fail the receive in that case.
645 */
646 ipc_port_link_special_reply_port(special_reply_port,
647 dest_port, FALSE);
648
649 ip_release(dest_port);
650 }
651 return MACH_MSG_SUCCESS;
652 }
653
654 /*
655 * Routine: mach_msg_receive_results_complete
656 * Purpose:
657 * Get thread's turnstile back from the object and
658 * if object is a special reply port then reset its
659 * linkage.
660 * Condition:
661 * Nothing locked.
662 * Returns:
663 * None.
664 */
665 void
666 mach_msg_receive_results_complete(ipc_object_t object)
667 {
668 thread_t self = current_thread();
669 ipc_port_t port = IPC_PORT_NULL;
670 boolean_t get_turnstile = (self->turnstile == TURNSTILE_NULL);
671
672 if (io_otype(object) == IOT_PORT) {
673 port = ip_object_to_port(object);
674 } else {
675 assert(self->turnstile != TURNSTILE_NULL);
676 return;
677 }
678
679 uint8_t flags = IPC_PORT_ADJUST_SR_ALLOW_SYNC_LINKAGE;
680
681 /*
682 * Don't clear the ip_srp_msg_sent bit if...
683 */
684 if (!((self->ith_state == MACH_RCV_TOO_LARGE && self->ith_option & MACH_RCV_LARGE) || //msg was too large and the next receive will get it
685 self->ith_state == MACH_RCV_INTERRUPTED ||
686 self->ith_state == MACH_RCV_TIMED_OUT ||
687 self->ith_state == MACH_RCV_PORT_CHANGED ||
688 self->ith_state == MACH_PEEK_READY)) {
689 flags |= IPC_PORT_ADJUST_SR_RECEIVED_MSG;
690 }
691
692 if (port->ip_specialreply || get_turnstile) {
693 ip_lock(port);
694 ipc_port_adjust_special_reply_port_locked(port, NULL,
695 flags, get_turnstile);
696 }
697 assert(self->turnstile != TURNSTILE_NULL);
698 /* thread now has a turnstile */
699 }
700
701 /*
702 * Routine: mach_msg_trap [mach trap]
703 * Purpose:
704 * Possibly send a message; possibly receive a message.
705 * Conditions:
706 * Nothing locked.
707 * Returns:
708 * All of mach_msg_send and mach_msg_receive error codes.
709 */
710
711 mach_msg_return_t
712 mach_msg_trap(
713 struct mach_msg_overwrite_trap_args *args)
714 {
715 kern_return_t kr;
716 args->rcv_msg = (mach_vm_address_t)0;
717
718 kr = mach_msg_overwrite_trap(args);
719 return kr;
720 }
721
722
723 /*
724 * Routine: msg_receive_error [internal]
725 * Purpose:
726 * Builds a minimal header/trailer and copies it to
727 * the user message buffer. Invoked when in the case of a
728 * MACH_RCV_TOO_LARGE or MACH_RCV_BODY_ERROR error.
729 * Conditions:
730 * Nothing locked.
731 * size - maximum buffer size on input,
732 * actual copied-out size on output
733 * Returns:
734 * MACH_MSG_SUCCESS minimal header/trailer copied
735 * MACH_RCV_INVALID_DATA copyout to user buffer failed
736 */
737
738 mach_msg_return_t
739 msg_receive_error(
740 ipc_kmsg_t kmsg,
741 mach_msg_option_t option,
742 mach_vm_address_t rcv_addr,
743 mach_msg_size_t rcv_size,
744 mach_port_seqno_t seqno,
745 ipc_space_t space,
746 mach_msg_size_t *sizep)
747 {
748 mach_vm_address_t context;
749 mach_msg_trailer_size_t trailer_size;
750 mach_msg_max_trailer_t *trailer;
751
752 context = kmsg->ikm_header->msgh_remote_port->ip_context;
753
754 /*
755 * Copy out the destination port in the message.
756 * Destroy all other rights and memory in the message.
757 */
758 ipc_kmsg_copyout_dest(kmsg, space);
759
760 /*
761 * Build a minimal message with the requested trailer.
762 */
763 trailer = (mach_msg_max_trailer_t *)
764 ((vm_offset_t)kmsg->ikm_header +
765 round_msg(sizeof(mach_msg_header_t)));
766 kmsg->ikm_header->msgh_size = sizeof(mach_msg_header_t);
767 bcopy((char *)&trailer_template,
768 (char *)trailer,
769 sizeof(trailer_template));
770
771 trailer_size = ipc_kmsg_add_trailer(kmsg, space,
772 option, current_thread(), seqno,
773 TRUE, context);
774
775 /*
776 * Copy the message to user space and return the size
777 * (note that ipc_kmsg_put may also adjust the actual
778 * size copied out to user-space).
779 */
780 if (ipc_kmsg_put(kmsg, option, rcv_addr, rcv_size, trailer_size, sizep) == MACH_RCV_INVALID_DATA) {
781 return MACH_RCV_INVALID_DATA;
782 } else {
783 return MACH_MSG_SUCCESS;
784 }
785 }