diff options
Diffstat (limited to 'mach/msg.c')
| -rw-r--r-- | mach/msg.c | 85 |
1 files changed, 74 insertions, 11 deletions
diff --git a/mach/msg.c b/mach/msg.c index ece110868d..bccad7fd26 100644 --- a/mach/msg.c +++ b/mach/msg.c @@ -1,31 +1,94 @@ -/* +/* * Mach Operating System * Copyright (c) 1991,1990,1989, 1995 Carnegie Mellon University * All Rights Reserved. - * + * * Permission to use, copy, modify and distribute this software and its * documentation is hereby granted, provided that both the copyright * notice and this permission notice appear in all copies of the * software, derivative works or modified versions, and any portions * thereof, and that both notices appear in supporting documentation. - * + * * CARNEGIE MELLON ALLOWS FREE USE OF THIS SOFTWARE IN ITS "AS IS" * CONDITION. CARNEGIE MELLON DISCLAIMS ANY LIABILITY OF ANY KIND FOR * ANY DAMAGES WHATSOEVER RESULTING FROM THE USE OF THIS SOFTWARE. - * + * * Carnegie Mellon requests users of this software to return to - * + * * Software Distribution Coordinator or Software.Distribution@CS.CMU.EDU * School of Computer Science * Carnegie Mellon University * Pittsburgh PA 15213-3890 - * + * * any improvements or extensions that they make and grant Carnegie Mellon * the rights to redistribute these changes. */ #include <mach/port.h> #include <mach/message.h> +#ifdef MACH_MSG_OVERWRITE +/* In variants with this feature, the actual system call is + __mach_msg_overwrite_trap. */ +mach_msg_return_t +__mach_msg_trap (mach_msg_header_t *msg, + mach_msg_option_t option, + mach_msg_size_t send_size, + mach_msg_size_t rcv_size, + mach_port_t rcv_name, + mach_msg_timeout_t timeout, + mach_port_t notify) +{ + return __mach_msg_overwrite_trap (msg, option, send_size, + rcv_size, rcv_name, timeout, notify, + MACH_MSG_NULL, 0); +} +weak_alias (__mach_msg_trap, mach_msg_trap) + +/* See comments below in __mach_msg. */ +mach_msg_return_t +__mach_msg_overwrite (mach_msg_header_t *msg, + mach_msg_option_t option, + mach_msg_size_t send_size, + mach_msg_size_t rcv_size, + mach_port_t rcv_name, + mach_msg_timeout_t timeout, + mach_port_t notify, + mach_msg_header_t *rcv_msg, + mach_msg_size_t rcv_msg_size) + +{ + mach_msg_return_t ret; + + /* Consider the following cases: + 1. Errors in pseudo-receive (eg, MACH_SEND_INTERRUPTED + plus special bits). + 2. Use of MACH_SEND_INTERRUPT/MACH_RCV_INTERRUPT options. + 3. RPC calls with interruptions in one/both halves. + */ + + ret = __mach_msg_overwrite_trap (msg, option, send_size, + rcv_size, rcv_name, timeout, notify, + rcv_msg, rcv_msg_size); + if (ret == MACH_MSG_SUCCESS) + return MACH_MSG_SUCCESS; + + if (!(option & MACH_SEND_INTERRUPT)) + while (ret == MACH_SEND_INTERRUPTED) + ret = __mach_msg_overwrite_trap (msg, option, send_size, + rcv_size, rcv_name, timeout, notify, + rcv_msg, rcv_msg_size); + + if (!(option & MACH_RCV_INTERRUPT)) + while (ret == MACH_RCV_INTERRUPTED) + ret = __mach_msg_overwrite_trap (msg, option & ~MACH_SEND_MSG, + 0, rcv_size, rcv_name, timeout, notify, + rcv_msg, rcv_msg_size); + + return ret; +} +weak_alias (__mach_msg_overwrite, mach_msg_overwrite) +#endif + mach_msg_return_t __mach_msg (mach_msg_header_t *msg, mach_msg_option_t option, @@ -36,29 +99,29 @@ __mach_msg (mach_msg_header_t *msg, mach_port_t notify) { mach_msg_return_t ret; - + /* Consider the following cases: 1. Errors in pseudo-receive (eg, MACH_SEND_INTERRUPTED plus special bits). 2. Use of MACH_SEND_INTERRUPT/MACH_RCV_INTERRUPT options. 3. RPC calls with interruptions in one/both halves. */ - + ret = __mach_msg_trap (msg, option, send_size, rcv_size, rcv_name, timeout, notify); if (ret == MACH_MSG_SUCCESS) return MACH_MSG_SUCCESS; - + if (!(option & MACH_SEND_INTERRUPT)) while (ret == MACH_SEND_INTERRUPTED) ret = __mach_msg_trap (msg, option, send_size, rcv_size, rcv_name, timeout, notify); - + if (!(option & MACH_RCV_INTERRUPT)) while (ret == MACH_RCV_INTERRUPTED) ret = __mach_msg_trap (msg, option & ~MACH_SEND_MSG, 0, rcv_size, rcv_name, timeout, notify); - + return ret; } weak_alias (__mach_msg, mach_msg) |