Home | History | Annotate | Line # | Download | only in kern
sysv_msg.c revision 1.44.4.3
      1  1.44.4.3        ad /*	$NetBSD: sysv_msg.c,v 1.44.4.3 2007/01/12 01:04:07 ad Exp $	*/
      2      1.26   thorpej 
      3      1.26   thorpej /*-
      4  1.44.4.2        ad  * Copyright (c) 1999, 2006 The NetBSD Foundation, Inc.
      5      1.26   thorpej  * All rights reserved.
      6      1.26   thorpej  *
      7      1.26   thorpej  * This code is derived from software contributed to The NetBSD Foundation
      8      1.26   thorpej  * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
      9  1.44.4.2        ad  * NASA Ames Research Center, and by Andrew Doran.
     10      1.26   thorpej  *
     11      1.26   thorpej  * Redistribution and use in source and binary forms, with or without
     12      1.26   thorpej  * modification, are permitted provided that the following conditions
     13      1.26   thorpej  * are met:
     14      1.26   thorpej  * 1. Redistributions of source code must retain the above copyright
     15      1.26   thorpej  *    notice, this list of conditions and the following disclaimer.
     16      1.26   thorpej  * 2. Redistributions in binary form must reproduce the above copyright
     17      1.26   thorpej  *    notice, this list of conditions and the following disclaimer in the
     18      1.26   thorpej  *    documentation and/or other materials provided with the distribution.
     19      1.26   thorpej  * 3. All advertising materials mentioning features or use of this software
     20      1.26   thorpej  *    must display the following acknowledgement:
     21      1.26   thorpej  *	This product includes software developed by the NetBSD
     22      1.26   thorpej  *	Foundation, Inc. and its contributors.
     23      1.26   thorpej  * 4. Neither the name of The NetBSD Foundation nor the names of its
     24      1.26   thorpej  *    contributors may be used to endorse or promote products derived
     25      1.26   thorpej  *    from this software without specific prior written permission.
     26      1.26   thorpej  *
     27      1.26   thorpej  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     28      1.26   thorpej  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     29      1.26   thorpej  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     30      1.26   thorpej  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     31      1.26   thorpej  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     32      1.26   thorpej  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     33      1.26   thorpej  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     34      1.26   thorpej  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     35      1.26   thorpej  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     36      1.26   thorpej  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     37      1.26   thorpej  * POSSIBILITY OF SUCH DAMAGE.
     38      1.26   thorpej  */
     39       1.9       cgd 
     40       1.1       cgd /*
     41       1.1       cgd  * Implementation of SVID messages
     42       1.1       cgd  *
     43      1.26   thorpej  * Author: Daniel Boulet
     44       1.1       cgd  *
     45       1.1       cgd  * Copyright 1993 Daniel Boulet and RTMX Inc.
     46       1.1       cgd  *
     47       1.1       cgd  * This system call was implemented by Daniel Boulet under contract from RTMX.
     48       1.1       cgd  *
     49       1.1       cgd  * Redistribution and use in source forms, with and without modification,
     50       1.1       cgd  * are permitted provided that this entire comment appears intact.
     51       1.1       cgd  *
     52       1.1       cgd  * Redistribution in binary form may occur without any restrictions.
     53       1.1       cgd  * Obviously, it would be nice if you gave credit where credit is due
     54       1.1       cgd  * but requiring it would be too onerous.
     55       1.1       cgd  *
     56       1.1       cgd  * This software is provided ``AS IS'' without any warranties of any kind.
     57       1.1       cgd  */
     58      1.33     lukem 
     59      1.33     lukem #include <sys/cdefs.h>
     60  1.44.4.3        ad __KERNEL_RCSID(0, "$NetBSD: sysv_msg.c,v 1.44.4.3 2007/01/12 01:04:07 ad Exp $");
     61      1.23      tron 
     62      1.24      tron #define SYSVMSG
     63       1.1       cgd 
     64       1.2   mycroft #include <sys/param.h>
     65       1.2   mycroft #include <sys/kernel.h>
     66       1.2   mycroft #include <sys/msg.h>
     67      1.29    simonb #include <sys/sysctl.h>
     68      1.29    simonb #include <sys/mount.h>		/* XXX for <sys/syscallargs.h> */
     69      1.35   thorpej #include <sys/sa.h>
     70      1.10       cgd #include <sys/syscallargs.h>
     71      1.42      elad #include <sys/kauth.h>
     72  1.44.4.1        ad #include <sys/mutex.h>
     73      1.18  christos 
     74       1.1       cgd #define MSG_DEBUG
     75       1.1       cgd #undef MSG_DEBUG_OK
     76       1.1       cgd 
     77      1.20  christos #ifdef MSG_DEBUG_OK
     78      1.21  christos #define MSG_PRINTF(a)	printf a
     79      1.20  christos #else
     80      1.20  christos #define MSG_PRINTF(a)
     81      1.20  christos #endif
     82      1.20  christos 
     83      1.36  jdolecek static int	nfree_msgmaps;		/* # of free map entries */
     84      1.36  jdolecek static short	free_msgmaps;	/* head of linked list of free map entries */
     85      1.36  jdolecek static struct	__msg *free_msghdrs;	/* list of free msg headers */
     86      1.36  jdolecek static char	*msgpool;		/* MSGMAX byte long msg buffer pool */
     87      1.36  jdolecek static struct	msgmap *msgmaps;	/* MSGSEG msgmap structures */
     88      1.36  jdolecek static struct __msg *msghdrs;		/* MSGTQL msg headers */
     89      1.36  jdolecek struct	msqid_ds *msqids;		/* MSGMNI msqid_ds struct's */
     90  1.44.4.1        ad static kmutex_t msgmutex;
     91       1.1       cgd 
     92      1.37  junyoung static void msg_freehdr(struct __msg *);
     93      1.18  christos 
     94      1.18  christos void
     95      1.40   thorpej msginit(void)
     96       1.1       cgd {
     97      1.36  jdolecek 	int i, sz;
     98      1.36  jdolecek 	vaddr_t v;
     99       1.1       cgd 
    100       1.3   mycroft 	/*
    101       1.3   mycroft 	 * msginfo.msgssz should be a power of two for efficiency reasons.
    102       1.3   mycroft 	 * It is also pretty silly if msginfo.msgssz is less than 8
    103       1.3   mycroft 	 * or greater than about 256 so ...
    104       1.3   mycroft 	 */
    105       1.1       cgd 
    106       1.3   mycroft 	i = 8;
    107       1.3   mycroft 	while (i < 1024 && i != msginfo.msgssz)
    108       1.3   mycroft 		i <<= 1;
    109       1.3   mycroft     	if (i != msginfo.msgssz) {
    110      1.20  christos 		MSG_PRINTF(("msginfo.msgssz=%d (0x%x)\n", msginfo.msgssz,
    111      1.20  christos 		    msginfo.msgssz));
    112       1.3   mycroft 		panic("msginfo.msgssz not a small power of 2");
    113       1.3   mycroft 	}
    114       1.3   mycroft 
    115       1.3   mycroft 	if (msginfo.msgseg > 32767) {
    116      1.20  christos 		MSG_PRINTF(("msginfo.msgseg=%d\n", msginfo.msgseg));
    117       1.3   mycroft 		panic("msginfo.msgseg > 32767");
    118       1.3   mycroft 	}
    119       1.3   mycroft 
    120      1.36  jdolecek 	/* Allocate pageable memory for our structures */
    121      1.36  jdolecek 	sz = msginfo.msgmax
    122      1.36  jdolecek 		+ msginfo.msgseg * sizeof(struct msgmap)
    123      1.36  jdolecek 		+ msginfo.msgtql * sizeof(struct __msg)
    124      1.36  jdolecek 		+ msginfo.msgmni * sizeof(struct msqid_ds);
    125      1.39      yamt 	v = uvm_km_alloc(kernel_map, round_page(sz), 0,
    126      1.39      yamt 	    UVM_KMF_WIRED|UVM_KMF_ZERO);
    127      1.39      yamt 	if (v == 0)
    128      1.36  jdolecek 		panic("sysv_msg: cannot allocate memory");
    129      1.36  jdolecek 	msgpool = (void *)v;
    130      1.36  jdolecek 	msgmaps = (void *) (msgpool + msginfo.msgmax);
    131      1.36  jdolecek 	msghdrs = (void *) (msgmaps + msginfo.msgseg);
    132      1.36  jdolecek 	msqids = (void *) (msghdrs + msginfo.msgtql);
    133       1.3   mycroft 
    134       1.3   mycroft 	for (i = 0; i < msginfo.msgseg; i++) {
    135       1.3   mycroft 		if (i > 0)
    136       1.3   mycroft 			msgmaps[i-1].next = i;
    137       1.3   mycroft 		msgmaps[i].next = -1;	/* implies entry is available */
    138       1.3   mycroft 	}
    139       1.3   mycroft 	free_msgmaps = 0;
    140       1.3   mycroft 	nfree_msgmaps = msginfo.msgseg;
    141       1.3   mycroft 
    142       1.3   mycroft 	if (msghdrs == NULL)
    143       1.3   mycroft 		panic("msghdrs is NULL");
    144       1.3   mycroft 
    145       1.3   mycroft 	for (i = 0; i < msginfo.msgtql; i++) {
    146       1.3   mycroft 		msghdrs[i].msg_type = 0;
    147       1.3   mycroft 		if (i > 0)
    148       1.3   mycroft 			msghdrs[i-1].msg_next = &msghdrs[i];
    149       1.3   mycroft 		msghdrs[i].msg_next = NULL;
    150       1.3   mycroft     	}
    151       1.3   mycroft 	free_msghdrs = &msghdrs[0];
    152       1.3   mycroft 
    153       1.3   mycroft 	if (msqids == NULL)
    154       1.3   mycroft 		panic("msqids is NULL");
    155       1.3   mycroft 
    156       1.4   mycroft 	for (i = 0; i < msginfo.msgmni; i++) {
    157       1.3   mycroft 		msqids[i].msg_qbytes = 0;	/* implies entry is available */
    158      1.26   thorpej 		msqids[i].msg_perm._seq = 0;	/* reset to a known value */
    159       1.3   mycroft 	}
    160  1.44.4.1        ad 
    161  1.44.4.1        ad 	mutex_init(&msgmutex, MUTEX_DEFAULT, IPL_NONE);
    162       1.1       cgd }
    163       1.1       cgd 
    164       1.3   mycroft static void
    165      1.40   thorpej msg_freehdr(struct __msg *msghdr)
    166       1.1       cgd {
    167  1.44.4.1        ad 
    168  1.44.4.1        ad 	KASSERT(mutex_owned(&msgmutex));
    169  1.44.4.1        ad 
    170       1.3   mycroft 	while (msghdr->msg_ts > 0) {
    171       1.3   mycroft 		short next;
    172       1.3   mycroft 		if (msghdr->msg_spot < 0 || msghdr->msg_spot >= msginfo.msgseg)
    173       1.3   mycroft 			panic("msghdr->msg_spot out of range");
    174       1.3   mycroft 		next = msgmaps[msghdr->msg_spot].next;
    175       1.3   mycroft 		msgmaps[msghdr->msg_spot].next = free_msgmaps;
    176       1.3   mycroft 		free_msgmaps = msghdr->msg_spot;
    177       1.5   mycroft 		nfree_msgmaps++;
    178       1.3   mycroft 		msghdr->msg_spot = next;
    179       1.3   mycroft 		if (msghdr->msg_ts >= msginfo.msgssz)
    180       1.3   mycroft 			msghdr->msg_ts -= msginfo.msgssz;
    181       1.3   mycroft 		else
    182       1.3   mycroft 			msghdr->msg_ts = 0;
    183       1.3   mycroft 	}
    184       1.3   mycroft 	if (msghdr->msg_spot != -1)
    185       1.3   mycroft 		panic("msghdr->msg_spot != -1");
    186       1.3   mycroft 	msghdr->msg_next = free_msghdrs;
    187       1.3   mycroft 	free_msghdrs = msghdr;
    188       1.1       cgd }
    189       1.1       cgd 
    190       1.1       cgd int
    191      1.40   thorpej sys___msgctl13(struct lwp *l, void *v, register_t *retval)
    192      1.16   thorpej {
    193      1.26   thorpej 	struct sys___msgctl13_args /* {
    194      1.10       cgd 		syscallarg(int) msqid;
    195      1.10       cgd 		syscallarg(int) cmd;
    196      1.10       cgd 		syscallarg(struct msqid_ds *) buf;
    197      1.16   thorpej 	} */ *uap = v;
    198      1.26   thorpej 	struct msqid_ds msqbuf;
    199      1.26   thorpej 	int cmd, error;
    200      1.26   thorpej 
    201      1.26   thorpej 	cmd = SCARG(uap, cmd);
    202      1.26   thorpej 
    203      1.26   thorpej 	if (cmd == IPC_SET) {
    204      1.26   thorpej 		error = copyin(SCARG(uap, buf), &msqbuf, sizeof(msqbuf));
    205      1.26   thorpej 		if (error)
    206      1.26   thorpej 			return (error);
    207      1.26   thorpej 	}
    208      1.26   thorpej 
    209      1.44        ad 	error = msgctl1(l, SCARG(uap, msqid), cmd,
    210      1.26   thorpej 	    (cmd == IPC_SET || cmd == IPC_STAT) ? &msqbuf : NULL);
    211      1.26   thorpej 
    212      1.26   thorpej 	if (error == 0 && cmd == IPC_STAT)
    213      1.26   thorpej 		error = copyout(&msqbuf, SCARG(uap, buf), sizeof(msqbuf));
    214      1.26   thorpej 
    215      1.26   thorpej 	return (error);
    216      1.26   thorpej }
    217      1.26   thorpej 
    218      1.26   thorpej int
    219      1.44        ad msgctl1(struct lwp *l, int msqid, int cmd, struct msqid_ds *msqbuf)
    220      1.26   thorpej {
    221      1.44        ad 	kauth_cred_t cred = l->l_cred;
    222      1.26   thorpej 	struct msqid_ds *msqptr;
    223      1.26   thorpej 	int error = 0, ix;
    224       1.1       cgd 
    225      1.26   thorpej 	MSG_PRINTF(("call to msgctl1(%d, %d)\n", msqid, cmd));
    226       1.1       cgd 
    227      1.26   thorpej 	ix = IPCID_TO_IX(msqid);
    228       1.1       cgd 
    229  1.44.4.1        ad 	mutex_enter(&msgmutex);
    230  1.44.4.1        ad 
    231      1.26   thorpej 	if (ix < 0 || ix >= msginfo.msgmni) {
    232      1.26   thorpej 		MSG_PRINTF(("msqid (%d) out of range (0<=msqid<%d)\n", ix,
    233      1.20  christos 		    msginfo.msgmni));
    234  1.44.4.1        ad 		error = EINVAL;
    235  1.44.4.1        ad 		goto unlock;
    236       1.3   mycroft 	}
    237       1.1       cgd 
    238      1.26   thorpej 	msqptr = &msqids[ix];
    239       1.1       cgd 
    240       1.3   mycroft 	if (msqptr->msg_qbytes == 0) {
    241      1.20  christos 		MSG_PRINTF(("no such msqid\n"));
    242  1.44.4.1        ad 		error = EINVAL;
    243  1.44.4.1        ad 		goto unlock;
    244       1.3   mycroft 	}
    245      1.26   thorpej 	if (msqptr->msg_perm._seq != IPCID_TO_SEQ(msqid)) {
    246      1.20  christos 		MSG_PRINTF(("wrong sequence number\n"));
    247  1.44.4.1        ad 		error = EINVAL;
    248  1.44.4.1        ad 		goto unlock;
    249       1.3   mycroft 	}
    250       1.1       cgd 
    251       1.3   mycroft 	switch (cmd) {
    252       1.3   mycroft 	case IPC_RMID:
    253       1.1       cgd 	{
    254      1.26   thorpej 		struct __msg *msghdr;
    255      1.26   thorpej 		if ((error = ipcperm(cred, &msqptr->msg_perm, IPC_M)) != 0)
    256  1.44.4.1        ad 			break;
    257       1.3   mycroft 		/* Free the message headers */
    258      1.26   thorpej 		msghdr = msqptr->_msg_first;
    259       1.3   mycroft 		while (msghdr != NULL) {
    260      1.26   thorpej 			struct __msg *msghdr_tmp;
    261       1.3   mycroft 
    262       1.3   mycroft 			/* Free the segments of each message */
    263      1.26   thorpej 			msqptr->_msg_cbytes -= msghdr->msg_ts;
    264       1.5   mycroft 			msqptr->msg_qnum--;
    265       1.3   mycroft 			msghdr_tmp = msghdr;
    266       1.3   mycroft 			msghdr = msghdr->msg_next;
    267       1.3   mycroft 			msg_freehdr(msghdr_tmp);
    268       1.3   mycroft 		}
    269       1.1       cgd 
    270      1.26   thorpej 		if (msqptr->_msg_cbytes != 0)
    271       1.3   mycroft 			panic("msg_cbytes is screwed up");
    272       1.3   mycroft 		if (msqptr->msg_qnum != 0)
    273       1.3   mycroft 			panic("msg_qnum is screwed up");
    274       1.1       cgd 
    275       1.3   mycroft 		msqptr->msg_qbytes = 0;	/* Mark it as free */
    276       1.1       cgd 
    277      1.26   thorpej 		wakeup(msqptr);
    278       1.1       cgd 	}
    279       1.3   mycroft 		break;
    280       1.1       cgd 
    281       1.3   mycroft 	case IPC_SET:
    282      1.26   thorpej 		if ((error = ipcperm(cred, &msqptr->msg_perm, IPC_M)))
    283  1.44.4.2        ad 			break;
    284      1.44        ad 		if (msqbuf->msg_qbytes > msqptr->msg_qbytes &&
    285  1.44.4.3        ad 		    kauth_authorize_generic(cred, KAUTH_GENERIC_ISSUSER,
    286  1.44.4.3        ad 		    NULL) != 0) {
    287  1.44.4.2        ad 			error = EPERM;
    288  1.44.4.2        ad 			break;
    289  1.44.4.2        ad 		}
    290      1.26   thorpej 		if (msqbuf->msg_qbytes > msginfo.msgmnb) {
    291      1.26   thorpej 			MSG_PRINTF(("can't increase msg_qbytes beyond %d "
    292      1.26   thorpej 			    "(truncating)\n", msginfo.msgmnb));
    293      1.26   thorpej 			/* silently restrict qbytes to system limit */
    294      1.26   thorpej 			msqbuf->msg_qbytes = msginfo.msgmnb;
    295       1.3   mycroft 		}
    296      1.26   thorpej 		if (msqbuf->msg_qbytes == 0) {
    297      1.20  christos 			MSG_PRINTF(("can't reduce msg_qbytes to 0\n"));
    298  1.44.4.1        ad 			error = EINVAL;		/* XXX non-standard errno! */
    299  1.44.4.1        ad 			break;
    300       1.3   mycroft 		}
    301      1.26   thorpej 		msqptr->msg_perm.uid = msqbuf->msg_perm.uid;
    302      1.26   thorpej 		msqptr->msg_perm.gid = msqbuf->msg_perm.gid;
    303       1.3   mycroft 		msqptr->msg_perm.mode = (msqptr->msg_perm.mode & ~0777) |
    304      1.26   thorpej 		    (msqbuf->msg_perm.mode & 0777);
    305      1.26   thorpej 		msqptr->msg_qbytes = msqbuf->msg_qbytes;
    306      1.43    kardel 		msqptr->msg_ctime = time_second;
    307       1.3   mycroft 		break;
    308       1.1       cgd 
    309       1.3   mycroft 	case IPC_STAT:
    310      1.26   thorpej 		if ((error = ipcperm(cred, &msqptr->msg_perm, IPC_R))) {
    311      1.20  christos 			MSG_PRINTF(("requester doesn't have read access\n"));
    312      1.26   thorpej 			return (error);
    313       1.3   mycroft 		}
    314      1.26   thorpej 		memcpy(msqbuf, msqptr, sizeof(struct msqid_ds));
    315       1.3   mycroft 		break;
    316       1.1       cgd 
    317       1.3   mycroft 	default:
    318      1.20  christos 		MSG_PRINTF(("invalid command %d\n", cmd));
    319  1.44.4.1        ad 		error = EINVAL;
    320  1.44.4.1        ad 		break;
    321       1.3   mycroft 	}
    322       1.3   mycroft 
    323  1.44.4.1        ad  unlock:
    324  1.44.4.1        ad 	mutex_exit(&msgmutex);
    325      1.26   thorpej 	return (error);
    326       1.1       cgd }
    327       1.1       cgd 
    328       1.1       cgd int
    329      1.40   thorpej sys_msgget(struct lwp *l, void *v, register_t *retval)
    330      1.16   thorpej {
    331      1.26   thorpej 	struct sys_msgget_args /* {
    332      1.10       cgd 		syscallarg(key_t) key;
    333      1.10       cgd 		syscallarg(int) msgflg;
    334      1.16   thorpej 	} */ *uap = v;
    335  1.44.4.1        ad 	int msqid, error = 0;
    336      1.10       cgd 	int key = SCARG(uap, key);
    337      1.10       cgd 	int msgflg = SCARG(uap, msgflg);
    338      1.44        ad 	kauth_cred_t cred = l->l_cred;
    339      1.26   thorpej 	struct msqid_ds *msqptr = NULL;
    340       1.1       cgd 
    341  1.44.4.1        ad 	mutex_enter(&msgmutex);
    342  1.44.4.1        ad 
    343      1.20  christos 	MSG_PRINTF(("msgget(0x%x, 0%o)\n", key, msgflg));
    344       1.1       cgd 
    345       1.5   mycroft 	if (key != IPC_PRIVATE) {
    346       1.5   mycroft 		for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
    347       1.3   mycroft 			msqptr = &msqids[msqid];
    348       1.3   mycroft 			if (msqptr->msg_qbytes != 0 &&
    349      1.26   thorpej 			    msqptr->msg_perm._key == key)
    350       1.3   mycroft 				break;
    351       1.3   mycroft 		}
    352       1.3   mycroft 		if (msqid < msginfo.msgmni) {
    353      1.20  christos 			MSG_PRINTF(("found public key\n"));
    354       1.3   mycroft 			if ((msgflg & IPC_CREAT) && (msgflg & IPC_EXCL)) {
    355      1.20  christos 				MSG_PRINTF(("not exclusive\n"));
    356  1.44.4.1        ad 				error = EEXIST;
    357  1.44.4.1        ad 				goto unlock;
    358       1.3   mycroft 			}
    359      1.26   thorpej 			if ((error = ipcperm(cred, &msqptr->msg_perm,
    360      1.26   thorpej 			    msgflg & 0700 ))) {
    361      1.20  christos 				MSG_PRINTF(("requester doesn't have 0%o access\n",
    362      1.20  christos 				    msgflg & 0700));
    363  1.44.4.1        ad 				goto unlock;
    364       1.3   mycroft 			}
    365       1.5   mycroft 			goto found;
    366       1.3   mycroft 		}
    367       1.1       cgd 	}
    368       1.1       cgd 
    369      1.20  christos 	MSG_PRINTF(("need to allocate the msqid_ds\n"));
    370       1.5   mycroft 	if (key == IPC_PRIVATE || (msgflg & IPC_CREAT)) {
    371       1.5   mycroft 		for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
    372       1.5   mycroft 			/*
    373       1.5   mycroft 			 * Look for an unallocated and unlocked msqid_ds.
    374       1.5   mycroft 			 * msqid_ds's can be locked by msgsnd or msgrcv while
    375       1.5   mycroft 			 * they are copying the message in/out.  We can't
    376       1.5   mycroft 			 * re-use the entry until they release it.
    377       1.5   mycroft 			 */
    378       1.5   mycroft 			msqptr = &msqids[msqid];
    379       1.5   mycroft 			if (msqptr->msg_qbytes == 0 &&
    380       1.5   mycroft 			    (msqptr->msg_perm.mode & MSG_LOCKED) == 0)
    381       1.5   mycroft 				break;
    382       1.5   mycroft 		}
    383       1.5   mycroft 		if (msqid == msginfo.msgmni) {
    384      1.20  christos 			MSG_PRINTF(("no more msqid_ds's available\n"));
    385  1.44.4.1        ad 			error = ENOSPC;
    386  1.44.4.1        ad 			goto unlock;
    387       1.5   mycroft 		}
    388      1.20  christos 		MSG_PRINTF(("msqid %d is available\n", msqid));
    389      1.26   thorpej 		msqptr->msg_perm._key = key;
    390      1.42      elad 		msqptr->msg_perm.cuid = kauth_cred_geteuid(cred);
    391      1.42      elad 		msqptr->msg_perm.uid = kauth_cred_geteuid(cred);
    392      1.42      elad 		msqptr->msg_perm.cgid = kauth_cred_getegid(cred);
    393      1.42      elad 		msqptr->msg_perm.gid = kauth_cred_getegid(cred);
    394       1.5   mycroft 		msqptr->msg_perm.mode = (msgflg & 0777);
    395       1.5   mycroft 		/* Make sure that the returned msqid is unique */
    396      1.26   thorpej 		msqptr->msg_perm._seq++;
    397      1.26   thorpej 		msqptr->_msg_first = NULL;
    398      1.26   thorpej 		msqptr->_msg_last = NULL;
    399      1.26   thorpej 		msqptr->_msg_cbytes = 0;
    400       1.5   mycroft 		msqptr->msg_qnum = 0;
    401       1.5   mycroft 		msqptr->msg_qbytes = msginfo.msgmnb;
    402       1.5   mycroft 		msqptr->msg_lspid = 0;
    403       1.5   mycroft 		msqptr->msg_lrpid = 0;
    404       1.5   mycroft 		msqptr->msg_stime = 0;
    405       1.5   mycroft 		msqptr->msg_rtime = 0;
    406      1.43    kardel 		msqptr->msg_ctime = time_second;
    407       1.5   mycroft 	} else {
    408      1.20  christos 		MSG_PRINTF(("didn't find it and wasn't asked to create it\n"));
    409  1.44.4.1        ad 		error = ENOENT;
    410  1.44.4.1        ad 		goto unlock;
    411       1.1       cgd 	}
    412       1.1       cgd 
    413      1.26   thorpej  found:
    414       1.3   mycroft 	/* Construct the unique msqid */
    415       1.3   mycroft 	*retval = IXSEQ_TO_IPCID(msqid, msqptr->msg_perm);
    416  1.44.4.1        ad 
    417  1.44.4.1        ad  unlock:
    418  1.44.4.1        ad   	mutex_exit(&msgmutex);
    419  1.44.4.1        ad 	return (error);
    420       1.1       cgd }
    421       1.1       cgd 
    422       1.1       cgd int
    423      1.40   thorpej sys_msgsnd(struct lwp *l, void *v, register_t *retval)
    424      1.16   thorpej {
    425      1.26   thorpej 	struct sys_msgsnd_args /* {
    426      1.10       cgd 		syscallarg(int) msqid;
    427      1.22    kleink 		syscallarg(const void *) msgp;
    428      1.10       cgd 		syscallarg(size_t) msgsz;
    429      1.10       cgd 		syscallarg(int) msgflg;
    430      1.16   thorpej 	} */ *uap = v;
    431      1.41      cube 
    432      1.44        ad 	return msgsnd1(l, SCARG(uap, msqid), SCARG(uap, msgp),
    433      1.41      cube 	    SCARG(uap, msgsz), SCARG(uap, msgflg), sizeof(long), copyin);
    434      1.41      cube }
    435      1.41      cube 
    436      1.41      cube int
    437      1.44        ad msgsnd1(struct lwp *l, int msqidr, const char *user_msgp, size_t msgsz,
    438      1.41      cube     int msgflg, size_t typesz, copyin_t fetch_type)
    439      1.41      cube {
    440  1.44.4.1        ad 	int segs_needed, error = 0, msqid;
    441      1.44        ad 	kauth_cred_t cred = l->l_cred;
    442      1.26   thorpej 	struct msqid_ds *msqptr;
    443      1.26   thorpej 	struct __msg *msghdr;
    444       1.3   mycroft 	short next;
    445       1.1       cgd 
    446      1.34   nathanw 	MSG_PRINTF(("call to msgsnd(%d, %p, %lld, %d)\n", msqid, user_msgp,
    447      1.34   nathanw 	    (long long)msgsz, msgflg));
    448       1.1       cgd 
    449      1.41      cube 	msqid = IPCID_TO_IX(msqidr);
    450       1.1       cgd 
    451  1.44.4.1        ad 	mutex_enter(&msgmutex);
    452  1.44.4.1        ad 
    453       1.3   mycroft 	if (msqid < 0 || msqid >= msginfo.msgmni) {
    454      1.20  christos 		MSG_PRINTF(("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
    455      1.20  christos 		    msginfo.msgmni));
    456  1.44.4.1        ad 		error = EINVAL;
    457  1.44.4.1        ad 		goto unlock;
    458       1.3   mycroft 	}
    459       1.1       cgd 
    460       1.3   mycroft 	msqptr = &msqids[msqid];
    461       1.3   mycroft 	if (msqptr->msg_qbytes == 0) {
    462      1.20  christos 		MSG_PRINTF(("no such message queue id\n"));
    463  1.44.4.1        ad 		error = EINVAL;
    464  1.44.4.1        ad 		goto unlock;
    465       1.3   mycroft 	}
    466      1.41      cube 	if (msqptr->msg_perm._seq != IPCID_TO_SEQ(msqidr)) {
    467      1.20  christos 		MSG_PRINTF(("wrong sequence number\n"));
    468  1.44.4.1        ad 		error = EINVAL;
    469  1.44.4.1        ad 		goto unlock;
    470       1.3   mycroft 	}
    471       1.1       cgd 
    472      1.26   thorpej 	if ((error = ipcperm(cred, &msqptr->msg_perm, IPC_W))) {
    473      1.20  christos 		MSG_PRINTF(("requester doesn't have write access\n"));
    474  1.44.4.1        ad 		goto unlock;
    475       1.3   mycroft 	}
    476       1.1       cgd 
    477       1.3   mycroft 	segs_needed = (msgsz + msginfo.msgssz - 1) / msginfo.msgssz;
    478      1.34   nathanw 	MSG_PRINTF(("msgsz=%lld, msgssz=%d, segs_needed=%d\n",
    479      1.34   nathanw 	    (long long)msgsz, msginfo.msgssz, segs_needed));
    480       1.3   mycroft 	for (;;) {
    481       1.3   mycroft 		int need_more_resources = 0;
    482       1.1       cgd 
    483       1.3   mycroft 		/*
    484      1.18  christos 		 * check msgsz [cannot be negative since it is unsigned]
    485       1.3   mycroft 		 * (inside this loop in case msg_qbytes changes while we sleep)
    486       1.3   mycroft 		 */
    487       1.1       cgd 
    488      1.18  christos 		if (msgsz > msqptr->msg_qbytes) {
    489      1.20  christos 			MSG_PRINTF(("msgsz > msqptr->msg_qbytes\n"));
    490  1.44.4.1        ad 			error = EINVAL;
    491  1.44.4.1        ad 			goto unlock;
    492       1.3   mycroft 		}
    493       1.1       cgd 
    494       1.3   mycroft 		if (msqptr->msg_perm.mode & MSG_LOCKED) {
    495      1.20  christos 			MSG_PRINTF(("msqid is locked\n"));
    496       1.3   mycroft 			need_more_resources = 1;
    497       1.3   mycroft 		}
    498      1.26   thorpej 		if (msgsz + msqptr->_msg_cbytes > msqptr->msg_qbytes) {
    499      1.20  christos 			MSG_PRINTF(("msgsz + msg_cbytes > msg_qbytes\n"));
    500       1.3   mycroft 			need_more_resources = 1;
    501       1.3   mycroft 		}
    502       1.3   mycroft 		if (segs_needed > nfree_msgmaps) {
    503      1.20  christos 			MSG_PRINTF(("segs_needed > nfree_msgmaps\n"));
    504       1.3   mycroft 			need_more_resources = 1;
    505       1.3   mycroft 		}
    506       1.3   mycroft 		if (free_msghdrs == NULL) {
    507      1.20  christos 			MSG_PRINTF(("no more msghdrs\n"));
    508       1.3   mycroft 			need_more_resources = 1;
    509       1.3   mycroft 		}
    510       1.1       cgd 
    511       1.3   mycroft 		if (need_more_resources) {
    512       1.3   mycroft 			int we_own_it;
    513       1.1       cgd 
    514       1.3   mycroft 			if ((msgflg & IPC_NOWAIT) != 0) {
    515      1.26   thorpej 				MSG_PRINTF(("need more resources but caller "
    516      1.26   thorpej 				    "doesn't want to wait\n"));
    517  1.44.4.1        ad 				error = EAGAIN;
    518  1.44.4.1        ad 				goto unlock;
    519       1.3   mycroft 			}
    520       1.1       cgd 
    521       1.3   mycroft 			if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0) {
    522      1.20  christos 				MSG_PRINTF(("we don't own the msqid_ds\n"));
    523       1.3   mycroft 				we_own_it = 0;
    524       1.3   mycroft 			} else {
    525       1.3   mycroft 				/* Force later arrivals to wait for our
    526       1.3   mycroft 				   request */
    527      1.20  christos 				MSG_PRINTF(("we own the msqid_ds\n"));
    528       1.3   mycroft 				msqptr->msg_perm.mode |= MSG_LOCKED;
    529       1.3   mycroft 				we_own_it = 1;
    530       1.3   mycroft 			}
    531      1.20  christos 			MSG_PRINTF(("goodnight\n"));
    532  1.44.4.1        ad 			error = mtsleep(msqptr, (PZERO - 4) | PCATCH,
    533  1.44.4.1        ad 			    "msgwait", 0, &msgmutex);
    534      1.26   thorpej 			MSG_PRINTF(("good morning, error=%d\n", error));
    535       1.3   mycroft 			if (we_own_it)
    536       1.3   mycroft 				msqptr->msg_perm.mode &= ~MSG_LOCKED;
    537      1.26   thorpej 			if (error != 0) {
    538      1.26   thorpej 				MSG_PRINTF(("msgsnd: interrupted system "
    539      1.26   thorpej 				    "call\n"));
    540  1.44.4.1        ad 				error = EINTR;
    541  1.44.4.1        ad 				goto unlock;
    542       1.3   mycroft 			}
    543       1.1       cgd 
    544       1.3   mycroft 			/*
    545       1.3   mycroft 			 * Make sure that the msq queue still exists
    546       1.3   mycroft 			 */
    547       1.1       cgd 
    548       1.3   mycroft 			if (msqptr->msg_qbytes == 0) {
    549      1.20  christos 				MSG_PRINTF(("msqid deleted\n"));
    550  1.44.4.1        ad 				error = EIDRM;
    551  1.44.4.1        ad 				goto unlock;
    552       1.3   mycroft 			}
    553       1.3   mycroft 		} else {
    554      1.20  christos 			MSG_PRINTF(("got all the resources that we need\n"));
    555       1.3   mycroft 			break;
    556       1.3   mycroft 		}
    557       1.1       cgd 	}
    558       1.1       cgd 
    559       1.3   mycroft 	/*
    560       1.3   mycroft 	 * We have the resources that we need.
    561       1.3   mycroft 	 * Make sure!
    562       1.3   mycroft 	 */
    563       1.1       cgd 
    564       1.3   mycroft 	if (msqptr->msg_perm.mode & MSG_LOCKED)
    565       1.3   mycroft 		panic("msg_perm.mode & MSG_LOCKED");
    566       1.3   mycroft 	if (segs_needed > nfree_msgmaps)
    567       1.3   mycroft 		panic("segs_needed > nfree_msgmaps");
    568      1.26   thorpej 	if (msgsz + msqptr->_msg_cbytes > msqptr->msg_qbytes)
    569       1.3   mycroft 		panic("msgsz + msg_cbytes > msg_qbytes");
    570       1.3   mycroft 	if (free_msghdrs == NULL)
    571       1.3   mycroft 		panic("no more msghdrs");
    572       1.1       cgd 
    573       1.3   mycroft 	/*
    574       1.3   mycroft 	 * Re-lock the msqid_ds in case we page-fault when copying in the
    575       1.3   mycroft 	 * message
    576       1.3   mycroft 	 */
    577       1.1       cgd 
    578       1.3   mycroft 	if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0)
    579       1.3   mycroft 		panic("msqid_ds is already locked");
    580       1.3   mycroft 	msqptr->msg_perm.mode |= MSG_LOCKED;
    581       1.1       cgd 
    582       1.3   mycroft 	/*
    583       1.3   mycroft 	 * Allocate a message header
    584       1.3   mycroft 	 */
    585       1.1       cgd 
    586       1.3   mycroft 	msghdr = free_msghdrs;
    587       1.3   mycroft 	free_msghdrs = msghdr->msg_next;
    588       1.3   mycroft 	msghdr->msg_spot = -1;
    589       1.3   mycroft 	msghdr->msg_ts = msgsz;
    590       1.1       cgd 
    591       1.3   mycroft 	/*
    592       1.3   mycroft 	 * Allocate space for the message
    593       1.3   mycroft 	 */
    594       1.1       cgd 
    595       1.3   mycroft 	while (segs_needed > 0) {
    596       1.3   mycroft 		if (nfree_msgmaps <= 0)
    597       1.3   mycroft 			panic("not enough msgmaps");
    598       1.3   mycroft 		if (free_msgmaps == -1)
    599       1.3   mycroft 			panic("nil free_msgmaps");
    600       1.3   mycroft 		next = free_msgmaps;
    601       1.3   mycroft 		if (next <= -1)
    602       1.3   mycroft 			panic("next too low #1");
    603       1.3   mycroft 		if (next >= msginfo.msgseg)
    604       1.3   mycroft 			panic("next out of range #1");
    605      1.20  christos 		MSG_PRINTF(("allocating segment %d to message\n", next));
    606       1.3   mycroft 		free_msgmaps = msgmaps[next].next;
    607       1.5   mycroft 		nfree_msgmaps--;
    608       1.3   mycroft 		msgmaps[next].next = msghdr->msg_spot;
    609       1.3   mycroft 		msghdr->msg_spot = next;
    610       1.5   mycroft 		segs_needed--;
    611       1.1       cgd 	}
    612       1.1       cgd 
    613       1.3   mycroft 	/*
    614       1.3   mycroft 	 * Copy in the message type
    615       1.3   mycroft 	 */
    616  1.44.4.1        ad 	mutex_exit(&msgmutex);
    617  1.44.4.1        ad 	error = (*fetch_type)(user_msgp, &msghdr->msg_type, typesz);
    618  1.44.4.1        ad 	mutex_enter(&msgmutex);
    619  1.44.4.1        ad 	if (error != 0) {
    620      1.26   thorpej 		MSG_PRINTF(("error %d copying the message type\n", error));
    621       1.3   mycroft 		msg_freehdr(msghdr);
    622       1.3   mycroft 		msqptr->msg_perm.mode &= ~MSG_LOCKED;
    623      1.26   thorpej 		wakeup(msqptr);
    624  1.44.4.1        ad 		goto unlock;
    625       1.3   mycroft 	}
    626      1.41      cube 	user_msgp += typesz;
    627       1.1       cgd 
    628       1.3   mycroft 	/*
    629       1.3   mycroft 	 * Validate the message type
    630       1.3   mycroft 	 */
    631       1.1       cgd 
    632       1.3   mycroft 	if (msghdr->msg_type < 1) {
    633       1.3   mycroft 		msg_freehdr(msghdr);
    634       1.3   mycroft 		msqptr->msg_perm.mode &= ~MSG_LOCKED;
    635      1.26   thorpej 		wakeup(msqptr);
    636      1.34   nathanw 		MSG_PRINTF(("mtype (%ld) < 1\n", msghdr->msg_type));
    637  1.44.4.1        ad 		goto unlock;
    638       1.3   mycroft 	}
    639       1.1       cgd 
    640       1.3   mycroft 	/*
    641       1.3   mycroft 	 * Copy in the message body
    642       1.3   mycroft 	 */
    643       1.3   mycroft 
    644       1.3   mycroft 	next = msghdr->msg_spot;
    645       1.3   mycroft 	while (msgsz > 0) {
    646       1.3   mycroft 		size_t tlen;
    647       1.3   mycroft 		if (msgsz > msginfo.msgssz)
    648       1.3   mycroft 			tlen = msginfo.msgssz;
    649       1.3   mycroft 		else
    650       1.3   mycroft 			tlen = msgsz;
    651       1.3   mycroft 		if (next <= -1)
    652       1.3   mycroft 			panic("next too low #2");
    653       1.3   mycroft 		if (next >= msginfo.msgseg)
    654       1.3   mycroft 			panic("next out of range #2");
    655  1.44.4.1        ad 		mutex_exit(&msgmutex);
    656  1.44.4.1        ad 		error = copyin(user_msgp, &msgpool[next * msginfo.msgssz], tlen);
    657  1.44.4.1        ad 		mutex_enter(&msgmutex);
    658  1.44.4.1        ad 		if (error != 0) {
    659      1.26   thorpej 			MSG_PRINTF(("error %d copying in message segment\n",
    660      1.26   thorpej 			    error));
    661       1.3   mycroft 			msg_freehdr(msghdr);
    662       1.3   mycroft 			msqptr->msg_perm.mode &= ~MSG_LOCKED;
    663      1.26   thorpej 			wakeup(msqptr);
    664  1.44.4.1        ad 			goto unlock;
    665       1.3   mycroft 		}
    666       1.3   mycroft 		msgsz -= tlen;
    667       1.3   mycroft 		user_msgp += tlen;
    668       1.3   mycroft 		next = msgmaps[next].next;
    669       1.1       cgd 	}
    670       1.3   mycroft 	if (next != -1)
    671       1.3   mycroft 		panic("didn't use all the msg segments");
    672       1.1       cgd 
    673       1.3   mycroft 	/*
    674       1.3   mycroft 	 * We've got the message.  Unlock the msqid_ds.
    675       1.3   mycroft 	 */
    676       1.1       cgd 
    677       1.3   mycroft 	msqptr->msg_perm.mode &= ~MSG_LOCKED;
    678       1.1       cgd 
    679       1.3   mycroft 	/*
    680       1.3   mycroft 	 * Make sure that the msqid_ds is still allocated.
    681       1.3   mycroft 	 */
    682       1.1       cgd 
    683       1.3   mycroft 	if (msqptr->msg_qbytes == 0) {
    684       1.3   mycroft 		msg_freehdr(msghdr);
    685      1.26   thorpej 		wakeup(msqptr);
    686  1.44.4.1        ad 		error = EIDRM;
    687  1.44.4.1        ad 		goto unlock;
    688       1.3   mycroft 	}
    689       1.3   mycroft 
    690       1.3   mycroft 	/*
    691       1.3   mycroft 	 * Put the message into the queue
    692       1.3   mycroft 	 */
    693       1.1       cgd 
    694      1.26   thorpej 	if (msqptr->_msg_first == NULL) {
    695      1.26   thorpej 		msqptr->_msg_first = msghdr;
    696      1.26   thorpej 		msqptr->_msg_last = msghdr;
    697       1.3   mycroft 	} else {
    698      1.26   thorpej 		msqptr->_msg_last->msg_next = msghdr;
    699      1.26   thorpej 		msqptr->_msg_last = msghdr;
    700       1.3   mycroft 	}
    701      1.26   thorpej 	msqptr->_msg_last->msg_next = NULL;
    702       1.3   mycroft 
    703      1.26   thorpej 	msqptr->_msg_cbytes += msghdr->msg_ts;
    704       1.5   mycroft 	msqptr->msg_qnum++;
    705      1.44        ad 	msqptr->msg_lspid = l->l_proc->p_pid;
    706      1.43    kardel 	msqptr->msg_stime = time_second;
    707       1.3   mycroft 
    708      1.26   thorpej 	wakeup(msqptr);
    709  1.44.4.1        ad 
    710  1.44.4.1        ad  unlock:
    711  1.44.4.1        ad  	mutex_exit(&msgmutex);
    712  1.44.4.1        ad 	return error;
    713       1.1       cgd }
    714       1.1       cgd 
    715       1.1       cgd int
    716      1.40   thorpej sys_msgrcv(struct lwp *l, void *v, register_t *retval)
    717      1.16   thorpej {
    718      1.27  augustss 	struct sys_msgrcv_args /* {
    719      1.10       cgd 		syscallarg(int) msqid;
    720      1.10       cgd 		syscallarg(void *) msgp;
    721      1.10       cgd 		syscallarg(size_t) msgsz;
    722      1.10       cgd 		syscallarg(long) msgtyp;
    723      1.10       cgd 		syscallarg(int) msgflg;
    724      1.16   thorpej 	} */ *uap = v;
    725      1.41      cube 
    726      1.44        ad 	return msgrcv1(l, SCARG(uap, msqid), SCARG(uap, msgp),
    727      1.41      cube 	    SCARG(uap, msgsz), SCARG(uap, msgtyp), SCARG(uap, msgflg),
    728      1.41      cube 	    sizeof(long), copyout, retval);
    729      1.41      cube }
    730      1.41      cube 
    731      1.41      cube int
    732      1.44        ad msgrcv1(struct lwp *l, int msqidr, char *user_msgp, size_t msgsz, long msgtyp,
    733      1.41      cube     int msgflg, size_t typesz, copyout_t put_type, register_t *retval)
    734      1.41      cube {
    735       1.3   mycroft 	size_t len;
    736      1.44        ad 	kauth_cred_t cred = l->l_cred;
    737      1.27  augustss 	struct msqid_ds *msqptr;
    738      1.27  augustss 	struct __msg *msghdr;
    739  1.44.4.1        ad 	int error = 0, msqid;
    740       1.3   mycroft 	short next;
    741       1.1       cgd 
    742      1.34   nathanw 	MSG_PRINTF(("call to msgrcv(%d, %p, %lld, %ld, %d)\n", msqid,
    743      1.34   nathanw 	    user_msgp, (long long)msgsz, msgtyp, msgflg));
    744       1.1       cgd 
    745      1.41      cube 	msqid = IPCID_TO_IX(msqidr);
    746       1.1       cgd 
    747  1.44.4.1        ad 	mutex_enter(&msgmutex);
    748  1.44.4.1        ad 
    749       1.3   mycroft 	if (msqid < 0 || msqid >= msginfo.msgmni) {
    750      1.20  christos 		MSG_PRINTF(("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
    751      1.20  christos 		    msginfo.msgmni));
    752  1.44.4.1        ad 		error = EINVAL;
    753  1.44.4.1        ad 		goto unlock;
    754       1.3   mycroft 	}
    755       1.1       cgd 
    756       1.3   mycroft 	msqptr = &msqids[msqid];
    757       1.3   mycroft 	if (msqptr->msg_qbytes == 0) {
    758      1.20  christos 		MSG_PRINTF(("no such message queue id\n"));
    759  1.44.4.1        ad 		error = EINVAL;
    760  1.44.4.1        ad 		goto unlock;
    761       1.3   mycroft 	}
    762      1.41      cube 	if (msqptr->msg_perm._seq != IPCID_TO_SEQ(msqidr)) {
    763      1.20  christos 		MSG_PRINTF(("wrong sequence number\n"));
    764  1.44.4.1        ad 		error = EINVAL;
    765  1.44.4.1        ad 		goto unlock;
    766       1.3   mycroft 	}
    767       1.1       cgd 
    768      1.26   thorpej 	if ((error = ipcperm(cred, &msqptr->msg_perm, IPC_R))) {
    769      1.20  christos 		MSG_PRINTF(("requester doesn't have read access\n"));
    770  1.44.4.1        ad 		goto unlock;
    771       1.3   mycroft 	}
    772       1.1       cgd 
    773      1.18  christos #if 0
    774      1.18  christos 	/* cannot happen, msgsz is unsigned */
    775       1.3   mycroft 	if (msgsz < 0) {
    776      1.20  christos 		MSG_PRINTF(("msgsz < 0\n"));
    777  1.44.4.1        ad 		error = EINVAL;
    778  1.44.4.1        ad 		goto unlock;
    779       1.3   mycroft 	}
    780      1.18  christos #endif
    781       1.1       cgd 
    782       1.3   mycroft 	msghdr = NULL;
    783       1.3   mycroft 	while (msghdr == NULL) {
    784       1.3   mycroft 		if (msgtyp == 0) {
    785      1.26   thorpej 			msghdr = msqptr->_msg_first;
    786       1.3   mycroft 			if (msghdr != NULL) {
    787       1.3   mycroft 				if (msgsz < msghdr->msg_ts &&
    788       1.3   mycroft 				    (msgflg & MSG_NOERROR) == 0) {
    789      1.26   thorpej 					MSG_PRINTF(("first message on the "
    790      1.26   thorpej 					    "queue is too big "
    791      1.34   nathanw 					    "(want %lld, got %d)\n",
    792      1.34   nathanw 					    (long long)msgsz, msghdr->msg_ts));
    793  1.44.4.1        ad 					error = E2BIG;
    794  1.44.4.1        ad 					goto unlock;
    795       1.3   mycroft 				}
    796      1.26   thorpej 				if (msqptr->_msg_first == msqptr->_msg_last) {
    797      1.26   thorpej 					msqptr->_msg_first = NULL;
    798      1.26   thorpej 					msqptr->_msg_last = NULL;
    799       1.3   mycroft 				} else {
    800      1.26   thorpej 					msqptr->_msg_first = msghdr->msg_next;
    801      1.26   thorpej 					if (msqptr->_msg_first == NULL)
    802      1.26   thorpej 						panic("msg_first/last screwed "
    803      1.26   thorpej 						    "up #1");
    804       1.3   mycroft 				}
    805       1.3   mycroft 			}
    806       1.3   mycroft 		} else {
    807      1.26   thorpej 			struct __msg *previous;
    808      1.26   thorpej 			struct __msg **prev;
    809       1.1       cgd 
    810      1.26   thorpej 			for (previous = NULL, prev = &msqptr->_msg_first;
    811      1.12   mycroft 			     (msghdr = *prev) != NULL;
    812      1.12   mycroft 			     previous = msghdr, prev = &msghdr->msg_next) {
    813       1.3   mycroft 				/*
    814       1.3   mycroft 				 * Is this message's type an exact match or is
    815       1.3   mycroft 				 * this message's type less than or equal to
    816       1.3   mycroft 				 * the absolute value of a negative msgtyp?
    817       1.3   mycroft 				 * Note that the second half of this test can
    818       1.3   mycroft 				 * NEVER be true if msgtyp is positive since
    819       1.3   mycroft 				 * msg_type is always positive!
    820       1.3   mycroft 				 */
    821       1.3   mycroft 
    822       1.3   mycroft 				if (msgtyp == msghdr->msg_type ||
    823       1.3   mycroft 				    msghdr->msg_type <= -msgtyp) {
    824      1.34   nathanw 					MSG_PRINTF(("found message type %ld, "
    825      1.34   nathanw 					    "requested %ld\n",
    826      1.20  christos 					    msghdr->msg_type, msgtyp));
    827       1.3   mycroft 					if (msgsz < msghdr->msg_ts &&
    828       1.3   mycroft 					    (msgflg & MSG_NOERROR) == 0) {
    829      1.26   thorpej 						MSG_PRINTF(("requested message "
    830      1.26   thorpej 						    "on the queue is too big "
    831      1.34   nathanw 						    "(want %lld, got %d)\n",
    832      1.34   nathanw 						    (long long)msgsz,
    833      1.34   nathanw 						    msghdr->msg_ts));
    834  1.44.4.1        ad 						error = E2BIG;
    835  1.44.4.1        ad 						goto unlock;
    836       1.3   mycroft 					}
    837       1.3   mycroft 					*prev = msghdr->msg_next;
    838      1.26   thorpej 					if (msghdr == msqptr->_msg_last) {
    839       1.3   mycroft 						if (previous == NULL) {
    840       1.3   mycroft 							if (prev !=
    841      1.26   thorpej 							    &msqptr->_msg_first)
    842       1.3   mycroft 								panic("msg_first/last screwed up #2");
    843      1.26   thorpej 							msqptr->_msg_first =
    844       1.3   mycroft 							    NULL;
    845      1.26   thorpej 							msqptr->_msg_last =
    846       1.3   mycroft 							    NULL;
    847       1.3   mycroft 						} else {
    848       1.3   mycroft 							if (prev ==
    849      1.26   thorpej 							    &msqptr->_msg_first)
    850       1.3   mycroft 								panic("msg_first/last screwed up #3");
    851      1.26   thorpej 							msqptr->_msg_last =
    852       1.3   mycroft 							    previous;
    853       1.3   mycroft 						}
    854       1.3   mycroft 					}
    855       1.3   mycroft 					break;
    856       1.3   mycroft 				}
    857       1.3   mycroft 			}
    858       1.1       cgd 		}
    859       1.1       cgd 
    860       1.3   mycroft 		/*
    861       1.3   mycroft 		 * We've either extracted the msghdr for the appropriate
    862       1.3   mycroft 		 * message or there isn't one.
    863       1.3   mycroft 		 * If there is one then bail out of this loop.
    864       1.3   mycroft 		 */
    865       1.1       cgd 
    866       1.3   mycroft 		if (msghdr != NULL)
    867       1.3   mycroft 			break;
    868       1.1       cgd 
    869       1.1       cgd 		/*
    870       1.3   mycroft 		 * Hmph!  No message found.  Does the user want to wait?
    871       1.1       cgd 		 */
    872       1.1       cgd 
    873       1.3   mycroft 		if ((msgflg & IPC_NOWAIT) != 0) {
    874      1.34   nathanw 			MSG_PRINTF(("no appropriate message found (msgtyp=%ld)\n",
    875      1.20  christos 			    msgtyp));
    876       1.3   mycroft 			/* The SVID says to return ENOMSG. */
    877       1.1       cgd #ifdef ENOMSG
    878  1.44.4.1        ad 			error = ENOMSG;
    879       1.1       cgd #else
    880       1.3   mycroft 			/* Unfortunately, BSD doesn't define that code yet! */
    881  1.44.4.1        ad 			error = EAGAIN;
    882       1.1       cgd #endif
    883  1.44.4.1        ad 			goto unlock;
    884       1.3   mycroft 		}
    885       1.1       cgd 
    886       1.3   mycroft 		/*
    887       1.3   mycroft 		 * Wait for something to happen
    888       1.3   mycroft 		 */
    889       1.1       cgd 
    890      1.20  christos 		MSG_PRINTF(("msgrcv:  goodnight\n"));
    891  1.44.4.1        ad 		error = mtsleep(msqptr, (PZERO - 4) | PCATCH, "msgwait",
    892  1.44.4.1        ad 		    0, &msgmutex);
    893      1.26   thorpej 		MSG_PRINTF(("msgrcv: good morning (error=%d)\n", error));
    894       1.1       cgd 
    895      1.26   thorpej 		if (error != 0) {
    896      1.26   thorpej 			MSG_PRINTF(("msgsnd: interrupted system call\n"));
    897  1.44.4.1        ad 			error = EINTR;
    898  1.44.4.1        ad 			goto unlock;
    899       1.3   mycroft 		}
    900       1.1       cgd 
    901       1.3   mycroft 		/*
    902       1.3   mycroft 		 * Make sure that the msq queue still exists
    903       1.3   mycroft 		 */
    904       1.1       cgd 
    905       1.3   mycroft 		if (msqptr->msg_qbytes == 0 ||
    906      1.41      cube 		    msqptr->msg_perm._seq != IPCID_TO_SEQ(msqidr)) {
    907      1.20  christos 			MSG_PRINTF(("msqid deleted\n"));
    908  1.44.4.1        ad 			error = EIDRM;
    909  1.44.4.1        ad 			goto unlock;
    910       1.3   mycroft 		}
    911       1.1       cgd 	}
    912       1.1       cgd 
    913       1.3   mycroft 	/*
    914       1.3   mycroft 	 * Return the message to the user.
    915       1.3   mycroft 	 *
    916       1.3   mycroft 	 * First, do the bookkeeping (before we risk being interrupted).
    917       1.3   mycroft 	 */
    918       1.1       cgd 
    919      1.26   thorpej 	msqptr->_msg_cbytes -= msghdr->msg_ts;
    920       1.5   mycroft 	msqptr->msg_qnum--;
    921      1.44        ad 	msqptr->msg_lrpid = l->l_proc->p_pid;
    922      1.43    kardel 	msqptr->msg_rtime = time_second;
    923       1.1       cgd 
    924       1.3   mycroft 	/*
    925       1.3   mycroft 	 * Make msgsz the actual amount that we'll be returning.
    926       1.3   mycroft 	 * Note that this effectively truncates the message if it is too long
    927       1.3   mycroft 	 * (since msgsz is never increased).
    928       1.3   mycroft 	 */
    929       1.1       cgd 
    930      1.34   nathanw 	MSG_PRINTF(("found a message, msgsz=%lld, msg_ts=%d\n",
    931      1.34   nathanw 	    (long long)msgsz, msghdr->msg_ts));
    932       1.3   mycroft 	if (msgsz > msghdr->msg_ts)
    933       1.3   mycroft 		msgsz = msghdr->msg_ts;
    934       1.1       cgd 
    935       1.3   mycroft 	/*
    936       1.3   mycroft 	 * Return the type to the user.
    937       1.3   mycroft 	 */
    938  1.44.4.1        ad 	mutex_exit(&msgmutex);
    939      1.41      cube 	error = (*put_type)(&msghdr->msg_type, user_msgp, typesz);
    940  1.44.4.1        ad 	mutex_enter(&msgmutex);
    941      1.26   thorpej 	if (error != 0) {
    942      1.26   thorpej 		MSG_PRINTF(("error (%d) copying out message type\n", error));
    943       1.3   mycroft 		msg_freehdr(msghdr);
    944      1.26   thorpej 		wakeup(msqptr);
    945  1.44.4.1        ad 		goto unlock;
    946       1.3   mycroft 	}
    947      1.41      cube 	user_msgp += typesz;
    948       1.3   mycroft 
    949       1.3   mycroft 	/*
    950       1.3   mycroft 	 * Return the segments to the user
    951       1.3   mycroft 	 */
    952       1.1       cgd 
    953       1.3   mycroft 	next = msghdr->msg_spot;
    954       1.3   mycroft 	for (len = 0; len < msgsz; len += msginfo.msgssz) {
    955       1.3   mycroft 		size_t tlen;
    956       1.3   mycroft 
    957      1.25       mrg 		if (msgsz - len > msginfo.msgssz)
    958       1.3   mycroft 			tlen = msginfo.msgssz;
    959       1.3   mycroft 		else
    960      1.25       mrg 			tlen = msgsz - len;
    961       1.3   mycroft 		if (next <= -1)
    962       1.3   mycroft 			panic("next too low #3");
    963       1.3   mycroft 		if (next >= msginfo.msgseg)
    964       1.3   mycroft 			panic("next out of range #3");
    965  1.44.4.1        ad 		mutex_exit(&msgmutex);
    966      1.26   thorpej 		error = copyout(&msgpool[next * msginfo.msgssz],
    967       1.3   mycroft 		    user_msgp, tlen);
    968  1.44.4.1        ad 		mutex_enter(&msgmutex);
    969      1.26   thorpej 		if (error != 0) {
    970      1.20  christos 			MSG_PRINTF(("error (%d) copying out message segment\n",
    971      1.26   thorpej 			    error));
    972       1.3   mycroft 			msg_freehdr(msghdr);
    973      1.26   thorpej 			wakeup(msqptr);
    974  1.44.4.1        ad 			goto unlock;
    975       1.3   mycroft 		}
    976       1.3   mycroft 		user_msgp += tlen;
    977       1.3   mycroft 		next = msgmaps[next].next;
    978       1.1       cgd 	}
    979       1.1       cgd 
    980       1.3   mycroft 	/*
    981       1.3   mycroft 	 * Done, return the actual number of bytes copied out.
    982       1.3   mycroft 	 */
    983       1.1       cgd 
    984       1.3   mycroft 	msg_freehdr(msghdr);
    985      1.26   thorpej 	wakeup(msqptr);
    986       1.3   mycroft 	*retval = msgsz;
    987  1.44.4.1        ad 
    988  1.44.4.1        ad  unlock:
    989  1.44.4.1        ad  	mutex_exit(&msgmutex);
    990  1.44.4.1        ad 	return error;
    991       1.1       cgd }
    992