xref: /freebsd/sys/kern/sysv_msg.c (revision 2b14f991e64ebe31ca31a1a41237061c22a753d0)
12b14f991SJulian Elischer /*	$Id: sysv_msg.c,v 1.5 1995/07/29 11:40:13 bde Exp $ */
23d903220SDoug Rabson 
33d903220SDoug Rabson /*
43d903220SDoug Rabson  * Implementation of SVID messages
53d903220SDoug Rabson  *
63d903220SDoug Rabson  * Author:  Daniel Boulet
73d903220SDoug Rabson  *
83d903220SDoug Rabson  * Copyright 1993 Daniel Boulet and RTMX Inc.
93d903220SDoug Rabson  *
103d903220SDoug Rabson  * This system call was implemented by Daniel Boulet under contract from RTMX.
113d903220SDoug Rabson  *
123d903220SDoug Rabson  * Redistribution and use in source forms, with and without modification,
133d903220SDoug Rabson  * are permitted provided that this entire comment appears intact.
143d903220SDoug Rabson  *
153d903220SDoug Rabson  * Redistribution in binary form may occur without any restrictions.
163d903220SDoug Rabson  * Obviously, it would be nice if you gave credit where credit is due
173d903220SDoug Rabson  * but requiring it would be too onerous.
183d903220SDoug Rabson  *
193d903220SDoug Rabson  * This software is provided ``AS IS'' without any warranties of any kind.
203d903220SDoug Rabson  */
213d903220SDoug Rabson 
223d903220SDoug Rabson #include <sys/param.h>
233d903220SDoug Rabson #include <sys/systm.h>
243d903220SDoug Rabson #include <sys/kernel.h>
253d903220SDoug Rabson #include <sys/proc.h>
263d903220SDoug Rabson #include <sys/msg.h>
273d903220SDoug Rabson #include <sys/malloc.h>
283d903220SDoug Rabson 
292b14f991SJulian Elischer /*
302b14f991SJulian Elischer  * System initialization
312b14f991SJulian Elischer  */
322b14f991SJulian Elischer 
332b14f991SJulian Elischer extern void msginit();				/* should be static*/
342b14f991SJulian Elischer SYSINIT(sysv_msg, SI_SUB_SYSV_MSG, SI_ORDER_FIRST, msginit, NULL)
352b14f991SJulian Elischer 
363d903220SDoug Rabson #define MSG_DEBUG
373d903220SDoug Rabson #undef MSG_DEBUG_OK
383d903220SDoug Rabson 
393d903220SDoug Rabson static int	msgctl(), msgget(), msgsnd(), msgrcv();
403d903220SDoug Rabson 
413d903220SDoug Rabson int	(*msgcalls[])() = { msgctl, msgget, msgsnd, msgrcv };
423d903220SDoug Rabson 
433d903220SDoug Rabson int nfree_msgmaps;		/* # of free map entries */
443d903220SDoug Rabson short free_msgmaps;		/* head of linked list of free map entries */
453d903220SDoug Rabson struct msg *free_msghdrs;	/* list of free msg headers */
4628f8db14SBruce Evans char *msgpool;			/* MSGMAX byte long msg buffer pool */
4728f8db14SBruce Evans struct msgmap *msgmaps;		/* MSGSEG msgmap structures */
4828f8db14SBruce Evans struct msg *msghdrs;		/* MSGTQL msg headers */
4928f8db14SBruce Evans struct msqid_ds *msqids;	/* MSGMNI msqid_ds struct's */
503d903220SDoug Rabson 
51789668e2SDavid Greenman void
523d903220SDoug Rabson msginit()
533d903220SDoug Rabson {
543d903220SDoug Rabson 	register int i;
553d903220SDoug Rabson 
563d903220SDoug Rabson 	/*
573d903220SDoug Rabson 	 * msginfo.msgssz should be a power of two for efficiency reasons.
583d903220SDoug Rabson 	 * It is also pretty silly if msginfo.msgssz is less than 8
593d903220SDoug Rabson 	 * or greater than about 256 so ...
603d903220SDoug Rabson 	 */
613d903220SDoug Rabson 
623d903220SDoug Rabson 	i = 8;
633d903220SDoug Rabson 	while (i < 1024 && i != msginfo.msgssz)
643d903220SDoug Rabson 		i <<= 1;
653d903220SDoug Rabson     	if (i != msginfo.msgssz) {
663d903220SDoug Rabson 		printf("msginfo.msgssz=%d (0x%x)\n", msginfo.msgssz,
673d903220SDoug Rabson 		    msginfo.msgssz);
683d903220SDoug Rabson 		panic("msginfo.msgssz not a small power of 2");
693d903220SDoug Rabson 	}
703d903220SDoug Rabson 
713d903220SDoug Rabson 	if (msginfo.msgseg > 32767) {
723d903220SDoug Rabson 		printf("msginfo.msgseg=%d\n", msginfo.msgseg);
733d903220SDoug Rabson 		panic("msginfo.msgseg > 32767");
743d903220SDoug Rabson 	}
753d903220SDoug Rabson 
763d903220SDoug Rabson 	if (msgmaps == NULL)
773d903220SDoug Rabson 		panic("msgmaps is NULL");
783d903220SDoug Rabson 
793d903220SDoug Rabson 	for (i = 0; i < msginfo.msgseg; i++) {
803d903220SDoug Rabson 		if (i > 0)
813d903220SDoug Rabson 			msgmaps[i-1].next = i;
823d903220SDoug Rabson 		msgmaps[i].next = -1;	/* implies entry is available */
833d903220SDoug Rabson 	}
843d903220SDoug Rabson 	free_msgmaps = 0;
853d903220SDoug Rabson 	nfree_msgmaps = msginfo.msgseg;
863d903220SDoug Rabson 
873d903220SDoug Rabson 	if (msghdrs == NULL)
883d903220SDoug Rabson 		panic("msghdrs is NULL");
893d903220SDoug Rabson 
903d903220SDoug Rabson 	for (i = 0; i < msginfo.msgtql; i++) {
913d903220SDoug Rabson 		msghdrs[i].msg_type = 0;
923d903220SDoug Rabson 		if (i > 0)
933d903220SDoug Rabson 			msghdrs[i-1].msg_next = &msghdrs[i];
943d903220SDoug Rabson 		msghdrs[i].msg_next = NULL;
953d903220SDoug Rabson     	}
963d903220SDoug Rabson 	free_msghdrs = &msghdrs[0];
973d903220SDoug Rabson 
983d903220SDoug Rabson 	if (msqids == NULL)
993d903220SDoug Rabson 		panic("msqids is NULL");
1003d903220SDoug Rabson 
1013d903220SDoug Rabson 	for (i = 0; i < msginfo.msgmni; i++) {
1023d903220SDoug Rabson 		msqids[i].msg_qbytes = 0;	/* implies entry is available */
1033d903220SDoug Rabson 		msqids[i].msg_perm.seq = 0;	/* reset to a known value */
1043d903220SDoug Rabson 	}
1053d903220SDoug Rabson }
1063d903220SDoug Rabson 
1073d903220SDoug Rabson /*
1083d903220SDoug Rabson  * Entry point for all MSG calls
1093d903220SDoug Rabson  */
1103d903220SDoug Rabson 
1113d903220SDoug Rabson struct msgsys_args {
1123d903220SDoug Rabson 	u_int	which;
1133d903220SDoug Rabson };
1143d903220SDoug Rabson 
1153d903220SDoug Rabson int
1163d903220SDoug Rabson msgsys(p, uap, retval)
1173d903220SDoug Rabson 	struct caller *p;
1183d903220SDoug Rabson 	struct msgsys_args *uap;
1193d903220SDoug Rabson 	int *retval;
1203d903220SDoug Rabson {
1213d903220SDoug Rabson 
1223d903220SDoug Rabson 	if (uap->which >= sizeof(msgcalls)/sizeof(msgcalls[0]))
1233d903220SDoug Rabson 		return (EINVAL);
1243d903220SDoug Rabson 	return ((*msgcalls[uap->which])(p, &uap[1], retval));
1253d903220SDoug Rabson }
1263d903220SDoug Rabson 
1273d903220SDoug Rabson static void
1283d903220SDoug Rabson msg_freehdr(msghdr)
1293d903220SDoug Rabson 	struct msg *msghdr;
1303d903220SDoug Rabson {
1313d903220SDoug Rabson 	while (msghdr->msg_ts > 0) {
1323d903220SDoug Rabson 		short next;
1333d903220SDoug Rabson 		if (msghdr->msg_spot < 0 || msghdr->msg_spot >= msginfo.msgseg)
1343d903220SDoug Rabson 			panic("msghdr->msg_spot out of range");
1353d903220SDoug Rabson 		next = msgmaps[msghdr->msg_spot].next;
1363d903220SDoug Rabson 		msgmaps[msghdr->msg_spot].next = free_msgmaps;
1373d903220SDoug Rabson 		free_msgmaps = msghdr->msg_spot;
1383d903220SDoug Rabson 		nfree_msgmaps++;
1393d903220SDoug Rabson 		msghdr->msg_spot = next;
1403d903220SDoug Rabson 		if (msghdr->msg_ts >= msginfo.msgssz)
1413d903220SDoug Rabson 			msghdr->msg_ts -= msginfo.msgssz;
1423d903220SDoug Rabson 		else
1433d903220SDoug Rabson 			msghdr->msg_ts = 0;
1443d903220SDoug Rabson 	}
1453d903220SDoug Rabson 	if (msghdr->msg_spot != -1)
1463d903220SDoug Rabson 		panic("msghdr->msg_spot != -1");
1473d903220SDoug Rabson 	msghdr->msg_next = free_msghdrs;
1483d903220SDoug Rabson 	free_msghdrs = msghdr;
1493d903220SDoug Rabson }
1503d903220SDoug Rabson 
1513d903220SDoug Rabson struct msgctl_args {
1523d903220SDoug Rabson 	int	msqid;
1533d903220SDoug Rabson 	int	cmd;
1543d903220SDoug Rabson 	struct	msqid_ds *user_msqptr;
1553d903220SDoug Rabson };
1563d903220SDoug Rabson 
1573d903220SDoug Rabson int
1583d903220SDoug Rabson msgctl(p, uap, retval)
1593d903220SDoug Rabson 	struct proc *p;
1603d903220SDoug Rabson 	register struct msgctl_args *uap;
1613d903220SDoug Rabson 	int *retval;
1623d903220SDoug Rabson {
1633d903220SDoug Rabson 	int msqid = uap->msqid;
1643d903220SDoug Rabson 	int cmd = uap->cmd;
1653d903220SDoug Rabson 	struct msqid_ds *user_msqptr = uap->user_msqptr;
1663d903220SDoug Rabson 	struct ucred *cred = p->p_ucred;
167797f2d22SPoul-Henning Kamp 	int rval, eval;
1683d903220SDoug Rabson 	struct msqid_ds msqbuf;
1693d903220SDoug Rabson 	register struct msqid_ds *msqptr;
1703d903220SDoug Rabson 
1713d903220SDoug Rabson #ifdef MSG_DEBUG_OK
1723d903220SDoug Rabson 	printf("call to msgctl(%d, %d, 0x%x)\n", msqid, cmd, user_msqptr);
1733d903220SDoug Rabson #endif
1743d903220SDoug Rabson 
1753d903220SDoug Rabson 	msqid = IPCID_TO_IX(msqid);
1763d903220SDoug Rabson 
1773d903220SDoug Rabson 	if (msqid < 0 || msqid >= msginfo.msgmni) {
1783d903220SDoug Rabson #ifdef MSG_DEBUG_OK
1793d903220SDoug Rabson 		printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
1803d903220SDoug Rabson 		    msginfo.msgmni);
1813d903220SDoug Rabson #endif
1823d903220SDoug Rabson 		return(EINVAL);
1833d903220SDoug Rabson 	}
1843d903220SDoug Rabson 
1853d903220SDoug Rabson 	msqptr = &msqids[msqid];
1863d903220SDoug Rabson 
1873d903220SDoug Rabson 	if (msqptr->msg_qbytes == 0) {
1883d903220SDoug Rabson #ifdef MSG_DEBUG_OK
1893d903220SDoug Rabson 		printf("no such msqid\n");
1903d903220SDoug Rabson #endif
1913d903220SDoug Rabson 		return(EINVAL);
1923d903220SDoug Rabson 	}
1933d903220SDoug Rabson 	if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
1943d903220SDoug Rabson #ifdef MSG_DEBUG_OK
1953d903220SDoug Rabson 		printf("wrong sequence number\n");
1963d903220SDoug Rabson #endif
1973d903220SDoug Rabson 		return(EINVAL);
1983d903220SDoug Rabson 	}
1993d903220SDoug Rabson 
2003d903220SDoug Rabson 	eval = 0;
2013d903220SDoug Rabson 	rval = 0;
2023d903220SDoug Rabson 
2033d903220SDoug Rabson 	switch (cmd) {
2043d903220SDoug Rabson 
2053d903220SDoug Rabson 	case IPC_RMID:
2063d903220SDoug Rabson 	{
2073d903220SDoug Rabson 		struct msg *msghdr;
2083d903220SDoug Rabson 		if ((eval = ipcperm(cred, &msqptr->msg_perm, IPC_M)))
2093d903220SDoug Rabson 			return(eval);
2103d903220SDoug Rabson 		/* Free the message headers */
2113d903220SDoug Rabson 		msghdr = msqptr->msg_first;
2123d903220SDoug Rabson 		while (msghdr != NULL) {
2133d903220SDoug Rabson 			struct msg *msghdr_tmp;
2143d903220SDoug Rabson 
2153d903220SDoug Rabson 			/* Free the segments of each message */
2163d903220SDoug Rabson 			msqptr->msg_cbytes -= msghdr->msg_ts;
2173d903220SDoug Rabson 			msqptr->msg_qnum--;
2183d903220SDoug Rabson 			msghdr_tmp = msghdr;
2193d903220SDoug Rabson 			msghdr = msghdr->msg_next;
2203d903220SDoug Rabson 			msg_freehdr(msghdr_tmp);
2213d903220SDoug Rabson 		}
2223d903220SDoug Rabson 
2233d903220SDoug Rabson 		if (msqptr->msg_cbytes != 0)
2243d903220SDoug Rabson 			panic("msg_cbytes is screwed up");
2253d903220SDoug Rabson 		if (msqptr->msg_qnum != 0)
2263d903220SDoug Rabson 			panic("msg_qnum is screwed up");
2273d903220SDoug Rabson 
2283d903220SDoug Rabson 		msqptr->msg_qbytes = 0;	/* Mark it as free */
2293d903220SDoug Rabson 
2303d903220SDoug Rabson 		wakeup((caddr_t)msqptr);
2313d903220SDoug Rabson 	}
2323d903220SDoug Rabson 
2333d903220SDoug Rabson 		break;
2343d903220SDoug Rabson 
2353d903220SDoug Rabson 	case IPC_SET:
2363d903220SDoug Rabson 		if ((eval = ipcperm(cred, &msqptr->msg_perm, IPC_M)))
2373d903220SDoug Rabson 			return(eval);
2383d903220SDoug Rabson 		if ((eval = copyin(user_msqptr, &msqbuf, sizeof(msqbuf))) != 0)
2393d903220SDoug Rabson 			return(eval);
2403d903220SDoug Rabson 		if (msqbuf.msg_qbytes > msqptr->msg_qbytes && cred->cr_uid != 0)
2413d903220SDoug Rabson 			return(EPERM);
2423d903220SDoug Rabson 		if (msqbuf.msg_qbytes > msginfo.msgmnb) {
2433d903220SDoug Rabson #ifdef MSG_DEBUG_OK
2443d903220SDoug Rabson 			printf("can't increase msg_qbytes beyond %d (truncating)\n",
2453d903220SDoug Rabson 			    msginfo.msgmnb);
2463d903220SDoug Rabson #endif
2473d903220SDoug Rabson 			msqbuf.msg_qbytes = msginfo.msgmnb;	/* silently restrict qbytes to system limit */
2483d903220SDoug Rabson 		}
2493d903220SDoug Rabson 		if (msqbuf.msg_qbytes == 0) {
2503d903220SDoug Rabson #ifdef MSG_DEBUG_OK
2513d903220SDoug Rabson 			printf("can't reduce msg_qbytes to 0\n");
2523d903220SDoug Rabson #endif
2533d903220SDoug Rabson 			return(EINVAL);		/* non-standard errno! */
2543d903220SDoug Rabson 		}
2553d903220SDoug Rabson 		msqptr->msg_perm.uid = msqbuf.msg_perm.uid;	/* change the owner */
2563d903220SDoug Rabson 		msqptr->msg_perm.gid = msqbuf.msg_perm.gid;	/* change the owner */
2573d903220SDoug Rabson 		msqptr->msg_perm.mode = (msqptr->msg_perm.mode & ~0777) |
2583d903220SDoug Rabson 		    (msqbuf.msg_perm.mode & 0777);
2593d903220SDoug Rabson 		msqptr->msg_qbytes = msqbuf.msg_qbytes;
2603d903220SDoug Rabson 		msqptr->msg_ctime = time.tv_sec;
2613d903220SDoug Rabson 		break;
2623d903220SDoug Rabson 
2633d903220SDoug Rabson 	case IPC_STAT:
2643d903220SDoug Rabson 		if ((eval = ipcperm(cred, &msqptr->msg_perm, IPC_R))) {
2653d903220SDoug Rabson #ifdef MSG_DEBUG_OK
2663d903220SDoug Rabson 			printf("requester doesn't have read access\n");
2673d903220SDoug Rabson #endif
2683d903220SDoug Rabson 			return(eval);
2693d903220SDoug Rabson 		}
2703d903220SDoug Rabson 		eval = copyout((caddr_t)msqptr, user_msqptr,
2713d903220SDoug Rabson 		    sizeof(struct msqid_ds));
2723d903220SDoug Rabson 		break;
2733d903220SDoug Rabson 
2743d903220SDoug Rabson 	default:
2753d903220SDoug Rabson #ifdef MSG_DEBUG_OK
2763d903220SDoug Rabson 		printf("invalid command %d\n", cmd);
2773d903220SDoug Rabson #endif
2783d903220SDoug Rabson 		return(EINVAL);
2793d903220SDoug Rabson 	}
2803d903220SDoug Rabson 
2813d903220SDoug Rabson 	if (eval == 0)
2823d903220SDoug Rabson 		*retval = rval;
2833d903220SDoug Rabson 	return(eval);
2843d903220SDoug Rabson }
2853d903220SDoug Rabson 
2863d903220SDoug Rabson struct msgget_args {
2873d903220SDoug Rabson 	key_t	key;
2883d903220SDoug Rabson 	int	msgflg;
2893d903220SDoug Rabson };
2903d903220SDoug Rabson 
2913d903220SDoug Rabson int
2923d903220SDoug Rabson msgget(p, uap, retval)
2933d903220SDoug Rabson 	struct proc *p;
2943d903220SDoug Rabson 	register struct msgget_args *uap;
2953d903220SDoug Rabson 	int *retval;
2963d903220SDoug Rabson {
2973d903220SDoug Rabson 	int msqid, eval;
2983d903220SDoug Rabson 	int key = uap->key;
2993d903220SDoug Rabson 	int msgflg = uap->msgflg;
3003d903220SDoug Rabson 	struct ucred *cred = p->p_ucred;
301789668e2SDavid Greenman 	register struct msqid_ds *msqptr = NULL;
3023d903220SDoug Rabson 
3033d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3043d903220SDoug Rabson 	printf("msgget(0x%x, 0%o)\n", key, msgflg);
3053d903220SDoug Rabson #endif
3063d903220SDoug Rabson 
3073d903220SDoug Rabson 	if (key != IPC_PRIVATE) {
3083d903220SDoug Rabson 		for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
3093d903220SDoug Rabson 			msqptr = &msqids[msqid];
3103d903220SDoug Rabson 			if (msqptr->msg_qbytes != 0 &&
3113d903220SDoug Rabson 			    msqptr->msg_perm.key == key)
3123d903220SDoug Rabson 				break;
3133d903220SDoug Rabson 		}
3143d903220SDoug Rabson 		if (msqid < msginfo.msgmni) {
3153d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3163d903220SDoug Rabson 			printf("found public key\n");
3173d903220SDoug Rabson #endif
3183d903220SDoug Rabson 			if ((msgflg & IPC_CREAT) && (msgflg & IPC_EXCL)) {
3193d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3203d903220SDoug Rabson 				printf("not exclusive\n");
3213d903220SDoug Rabson #endif
3223d903220SDoug Rabson 				return(EEXIST);
3233d903220SDoug Rabson 			}
3243d903220SDoug Rabson 			if ((eval = ipcperm(cred, &msqptr->msg_perm, msgflg & 0700 ))) {
3253d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3263d903220SDoug Rabson 				printf("requester doesn't have 0%o access\n",
3273d903220SDoug Rabson 				    msgflg & 0700);
3283d903220SDoug Rabson #endif
3293d903220SDoug Rabson 				return(eval);
3303d903220SDoug Rabson 			}
3313d903220SDoug Rabson 			goto found;
3323d903220SDoug Rabson 		}
3333d903220SDoug Rabson 	}
3343d903220SDoug Rabson 
3353d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3363d903220SDoug Rabson 	printf("need to allocate the msqid_ds\n");
3373d903220SDoug Rabson #endif
3383d903220SDoug Rabson 	if (key == IPC_PRIVATE || (msgflg & IPC_CREAT)) {
3393d903220SDoug Rabson 		for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
3403d903220SDoug Rabson 			/*
3413d903220SDoug Rabson 			 * Look for an unallocated and unlocked msqid_ds.
3423d903220SDoug Rabson 			 * msqid_ds's can be locked by msgsnd or msgrcv while
3433d903220SDoug Rabson 			 * they are copying the message in/out.  We can't
3443d903220SDoug Rabson 			 * re-use the entry until they release it.
3453d903220SDoug Rabson 			 */
3463d903220SDoug Rabson 			msqptr = &msqids[msqid];
3473d903220SDoug Rabson 			if (msqptr->msg_qbytes == 0 &&
3483d903220SDoug Rabson 			    (msqptr->msg_perm.mode & MSG_LOCKED) == 0)
3493d903220SDoug Rabson 				break;
3503d903220SDoug Rabson 		}
3513d903220SDoug Rabson 		if (msqid == msginfo.msgmni) {
3523d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3533d903220SDoug Rabson 			printf("no more msqid_ds's available\n");
3543d903220SDoug Rabson #endif
3553d903220SDoug Rabson 			return(ENOSPC);
3563d903220SDoug Rabson 		}
3573d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3583d903220SDoug Rabson 		printf("msqid %d is available\n", msqid);
3593d903220SDoug Rabson #endif
3603d903220SDoug Rabson 		msqptr->msg_perm.key = key;
3613d903220SDoug Rabson 		msqptr->msg_perm.cuid = cred->cr_uid;
3623d903220SDoug Rabson 		msqptr->msg_perm.uid = cred->cr_uid;
3633d903220SDoug Rabson 		msqptr->msg_perm.cgid = cred->cr_gid;
3643d903220SDoug Rabson 		msqptr->msg_perm.gid = cred->cr_gid;
3653d903220SDoug Rabson 		msqptr->msg_perm.mode = (msgflg & 0777);
3663d903220SDoug Rabson 		/* Make sure that the returned msqid is unique */
3673d903220SDoug Rabson 		msqptr->msg_perm.seq++;
3683d903220SDoug Rabson 		msqptr->msg_first = NULL;
3693d903220SDoug Rabson 		msqptr->msg_last = NULL;
3703d903220SDoug Rabson 		msqptr->msg_cbytes = 0;
3713d903220SDoug Rabson 		msqptr->msg_qnum = 0;
3723d903220SDoug Rabson 		msqptr->msg_qbytes = msginfo.msgmnb;
3733d903220SDoug Rabson 		msqptr->msg_lspid = 0;
3743d903220SDoug Rabson 		msqptr->msg_lrpid = 0;
3753d903220SDoug Rabson 		msqptr->msg_stime = 0;
3763d903220SDoug Rabson 		msqptr->msg_rtime = 0;
3773d903220SDoug Rabson 		msqptr->msg_ctime = time.tv_sec;
3783d903220SDoug Rabson 	} else {
3793d903220SDoug Rabson #ifdef MSG_DEBUG_OK
3803d903220SDoug Rabson 		printf("didn't find it and wasn't asked to create it\n");
3813d903220SDoug Rabson #endif
3823d903220SDoug Rabson 		return(ENOENT);
3833d903220SDoug Rabson 	}
3843d903220SDoug Rabson 
3853d903220SDoug Rabson found:
3863d903220SDoug Rabson 	/* Construct the unique msqid */
3873d903220SDoug Rabson 	*retval = IXSEQ_TO_IPCID(msqid, msqptr->msg_perm);
3883d903220SDoug Rabson 	return(0);
3893d903220SDoug Rabson }
3903d903220SDoug Rabson 
3913d903220SDoug Rabson struct msgsnd_args {
3923d903220SDoug Rabson 	int	msqid;
3933d903220SDoug Rabson 	void	*user_msgp;
3943d903220SDoug Rabson 	size_t	msgsz;
3953d903220SDoug Rabson 	int	msgflg;
3963d903220SDoug Rabson };
3973d903220SDoug Rabson 
3983d903220SDoug Rabson int
3993d903220SDoug Rabson msgsnd(p, uap, retval)
4003d903220SDoug Rabson 	struct proc *p;
4013d903220SDoug Rabson 	register struct msgsnd_args *uap;
4023d903220SDoug Rabson 	int *retval;
4033d903220SDoug Rabson {
4043d903220SDoug Rabson 	int msqid = uap->msqid;
4053d903220SDoug Rabson 	void *user_msgp = uap->user_msgp;
4063d903220SDoug Rabson 	size_t msgsz = uap->msgsz;
4073d903220SDoug Rabson 	int msgflg = uap->msgflg;
4083d903220SDoug Rabson 	int segs_needed, eval;
4093d903220SDoug Rabson 	struct ucred *cred = p->p_ucred;
4103d903220SDoug Rabson 	register struct msqid_ds *msqptr;
4113d903220SDoug Rabson 	register struct msg *msghdr;
4123d903220SDoug Rabson 	short next;
4133d903220SDoug Rabson 
4143d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4153d903220SDoug Rabson 	printf("call to msgsnd(%d, 0x%x, %d, %d)\n", msqid, user_msgp, msgsz,
4163d903220SDoug Rabson 	    msgflg);
4173d903220SDoug Rabson #endif
4183d903220SDoug Rabson 
4193d903220SDoug Rabson 	msqid = IPCID_TO_IX(msqid);
4203d903220SDoug Rabson 
4213d903220SDoug Rabson 	if (msqid < 0 || msqid >= msginfo.msgmni) {
4223d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4233d903220SDoug Rabson 		printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
4243d903220SDoug Rabson 		    msginfo.msgmni);
4253d903220SDoug Rabson #endif
4263d903220SDoug Rabson 		return(EINVAL);
4273d903220SDoug Rabson 	}
4283d903220SDoug Rabson 
4293d903220SDoug Rabson 	msqptr = &msqids[msqid];
4303d903220SDoug Rabson 	if (msqptr->msg_qbytes == 0) {
4313d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4323d903220SDoug Rabson 		printf("no such message queue id\n");
4333d903220SDoug Rabson #endif
4343d903220SDoug Rabson 		return(EINVAL);
4353d903220SDoug Rabson 	}
4363d903220SDoug Rabson 	if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
4373d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4383d903220SDoug Rabson 		printf("wrong sequence number\n");
4393d903220SDoug Rabson #endif
4403d903220SDoug Rabson 		return(EINVAL);
4413d903220SDoug Rabson 	}
4423d903220SDoug Rabson 
4433d903220SDoug Rabson 	if ((eval = ipcperm(cred, &msqptr->msg_perm, IPC_W))) {
4443d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4453d903220SDoug Rabson 		printf("requester doesn't have write access\n");
4463d903220SDoug Rabson #endif
4473d903220SDoug Rabson 		return(eval);
4483d903220SDoug Rabson 	}
4493d903220SDoug Rabson 
4503d903220SDoug Rabson 	segs_needed = (msgsz + msginfo.msgssz - 1) / msginfo.msgssz;
4513d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4523d903220SDoug Rabson 	printf("msgsz=%d, msgssz=%d, segs_needed=%d\n", msgsz, msginfo.msgssz,
4533d903220SDoug Rabson 	    segs_needed);
4543d903220SDoug Rabson #endif
4553d903220SDoug Rabson 	for (;;) {
4563d903220SDoug Rabson 		int need_more_resources = 0;
4573d903220SDoug Rabson 
4583d903220SDoug Rabson 		/*
4593d903220SDoug Rabson 		 * check msgsz
4603d903220SDoug Rabson 		 * (inside this loop in case msg_qbytes changes while we sleep)
4613d903220SDoug Rabson 		 */
4623d903220SDoug Rabson 
463789668e2SDavid Greenman 		if (msgsz > msqptr->msg_qbytes) {
4643d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4653d903220SDoug Rabson 			printf("msgsz > msqptr->msg_qbytes\n");
4663d903220SDoug Rabson #endif
4673d903220SDoug Rabson 			return(EINVAL);
4683d903220SDoug Rabson 		}
4693d903220SDoug Rabson 
4703d903220SDoug Rabson 		if (msqptr->msg_perm.mode & MSG_LOCKED) {
4713d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4723d903220SDoug Rabson 			printf("msqid is locked\n");
4733d903220SDoug Rabson #endif
4743d903220SDoug Rabson 			need_more_resources = 1;
4753d903220SDoug Rabson 		}
4763d903220SDoug Rabson 		if (msgsz + msqptr->msg_cbytes > msqptr->msg_qbytes) {
4773d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4783d903220SDoug Rabson 			printf("msgsz + msg_cbytes > msg_qbytes\n");
4793d903220SDoug Rabson #endif
4803d903220SDoug Rabson 			need_more_resources = 1;
4813d903220SDoug Rabson 		}
4823d903220SDoug Rabson 		if (segs_needed > nfree_msgmaps) {
4833d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4843d903220SDoug Rabson 			printf("segs_needed > nfree_msgmaps\n");
4853d903220SDoug Rabson #endif
4863d903220SDoug Rabson 			need_more_resources = 1;
4873d903220SDoug Rabson 		}
4883d903220SDoug Rabson 		if (free_msghdrs == NULL) {
4893d903220SDoug Rabson #ifdef MSG_DEBUG_OK
4903d903220SDoug Rabson 			printf("no more msghdrs\n");
4913d903220SDoug Rabson #endif
4923d903220SDoug Rabson 			need_more_resources = 1;
4933d903220SDoug Rabson 		}
4943d903220SDoug Rabson 
4953d903220SDoug Rabson 		if (need_more_resources) {
4963d903220SDoug Rabson 			int we_own_it;
4973d903220SDoug Rabson 
4983d903220SDoug Rabson 			if ((msgflg & IPC_NOWAIT) != 0) {
4993d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5003d903220SDoug Rabson 				printf("need more resources but caller doesn't want to wait\n");
5013d903220SDoug Rabson #endif
5023d903220SDoug Rabson 				return(EAGAIN);
5033d903220SDoug Rabson 			}
5043d903220SDoug Rabson 
5053d903220SDoug Rabson 			if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0) {
5063d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5073d903220SDoug Rabson 				printf("we don't own the msqid_ds\n");
5083d903220SDoug Rabson #endif
5093d903220SDoug Rabson 				we_own_it = 0;
5103d903220SDoug Rabson 			} else {
5113d903220SDoug Rabson 				/* Force later arrivals to wait for our
5123d903220SDoug Rabson 				   request */
5133d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5143d903220SDoug Rabson 				printf("we own the msqid_ds\n");
5153d903220SDoug Rabson #endif
5163d903220SDoug Rabson 				msqptr->msg_perm.mode |= MSG_LOCKED;
5173d903220SDoug Rabson 				we_own_it = 1;
5183d903220SDoug Rabson 			}
5193d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5203d903220SDoug Rabson 			printf("goodnight\n");
5213d903220SDoug Rabson #endif
5223d903220SDoug Rabson 			eval = tsleep((caddr_t)msqptr, (PZERO - 4) | PCATCH,
5233d903220SDoug Rabson 			    "msgwait", 0);
5243d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5253d903220SDoug Rabson 			printf("good morning, eval=%d\n", eval);
5263d903220SDoug Rabson #endif
5273d903220SDoug Rabson 			if (we_own_it)
5283d903220SDoug Rabson 				msqptr->msg_perm.mode &= ~MSG_LOCKED;
5293d903220SDoug Rabson 			if (eval != 0) {
5303d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5313d903220SDoug Rabson 				printf("msgsnd:  interrupted system call\n");
5323d903220SDoug Rabson #endif
5333d903220SDoug Rabson 				return(EINTR);
5343d903220SDoug Rabson 			}
5353d903220SDoug Rabson 
5363d903220SDoug Rabson 			/*
5373d903220SDoug Rabson 			 * Make sure that the msq queue still exists
5383d903220SDoug Rabson 			 */
5393d903220SDoug Rabson 
5403d903220SDoug Rabson 			if (msqptr->msg_qbytes == 0) {
5413d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5423d903220SDoug Rabson 				printf("msqid deleted\n");
5433d903220SDoug Rabson #endif
5443d903220SDoug Rabson 				/* The SVID says to return EIDRM. */
5453d903220SDoug Rabson #ifdef EIDRM
5463d903220SDoug Rabson 				return(EIDRM);
5473d903220SDoug Rabson #else
5483d903220SDoug Rabson 				/* Unfortunately, BSD doesn't define that code
5493d903220SDoug Rabson 				   yet! */
5503d903220SDoug Rabson 				return(EINVAL);
5513d903220SDoug Rabson #endif
5523d903220SDoug Rabson 			}
5533d903220SDoug Rabson 
5543d903220SDoug Rabson 		} else {
5553d903220SDoug Rabson #ifdef MSG_DEBUG_OK
5563d903220SDoug Rabson 			printf("got all the resources that we need\n");
5573d903220SDoug Rabson #endif
5583d903220SDoug Rabson 			break;
5593d903220SDoug Rabson 		}
5603d903220SDoug Rabson 	}
5613d903220SDoug Rabson 
5623d903220SDoug Rabson 	/*
5633d903220SDoug Rabson 	 * We have the resources that we need.
5643d903220SDoug Rabson 	 * Make sure!
5653d903220SDoug Rabson 	 */
5663d903220SDoug Rabson 
5673d903220SDoug Rabson 	if (msqptr->msg_perm.mode & MSG_LOCKED)
5683d903220SDoug Rabson 		panic("msg_perm.mode & MSG_LOCKED");
5693d903220SDoug Rabson 	if (segs_needed > nfree_msgmaps)
5703d903220SDoug Rabson 		panic("segs_needed > nfree_msgmaps");
5713d903220SDoug Rabson 	if (msgsz + msqptr->msg_cbytes > msqptr->msg_qbytes)
5723d903220SDoug Rabson 		panic("msgsz + msg_cbytes > msg_qbytes");
5733d903220SDoug Rabson 	if (free_msghdrs == NULL)
5743d903220SDoug Rabson 		panic("no more msghdrs");
5753d903220SDoug Rabson 
5763d903220SDoug Rabson 	/*
5773d903220SDoug Rabson 	 * Re-lock the msqid_ds in case we page-fault when copying in the
5783d903220SDoug Rabson 	 * message
5793d903220SDoug Rabson 	 */
5803d903220SDoug Rabson 
5813d903220SDoug Rabson 	if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0)
5823d903220SDoug Rabson 		panic("msqid_ds is already locked");
5833d903220SDoug Rabson 	msqptr->msg_perm.mode |= MSG_LOCKED;
5843d903220SDoug Rabson 
5853d903220SDoug Rabson 	/*
5863d903220SDoug Rabson 	 * Allocate a message header
5873d903220SDoug Rabson 	 */
5883d903220SDoug Rabson 
5893d903220SDoug Rabson 	msghdr = free_msghdrs;
5903d903220SDoug Rabson 	free_msghdrs = msghdr->msg_next;
5913d903220SDoug Rabson 	msghdr->msg_spot = -1;
5923d903220SDoug Rabson 	msghdr->msg_ts = msgsz;
5933d903220SDoug Rabson 
5943d903220SDoug Rabson 	/*
5953d903220SDoug Rabson 	 * Allocate space for the message
5963d903220SDoug Rabson 	 */
5973d903220SDoug Rabson 
5983d903220SDoug Rabson 	while (segs_needed > 0) {
5993d903220SDoug Rabson 		if (nfree_msgmaps <= 0)
6003d903220SDoug Rabson 			panic("not enough msgmaps");
6013d903220SDoug Rabson 		if (free_msgmaps == -1)
6023d903220SDoug Rabson 			panic("nil free_msgmaps");
6033d903220SDoug Rabson 		next = free_msgmaps;
6043d903220SDoug Rabson 		if (next <= -1)
6053d903220SDoug Rabson 			panic("next too low #1");
6063d903220SDoug Rabson 		if (next >= msginfo.msgseg)
6073d903220SDoug Rabson 			panic("next out of range #1");
6083d903220SDoug Rabson #ifdef MSG_DEBUG_OK
6093d903220SDoug Rabson 		printf("allocating segment %d to message\n", next);
6103d903220SDoug Rabson #endif
6113d903220SDoug Rabson 		free_msgmaps = msgmaps[next].next;
6123d903220SDoug Rabson 		nfree_msgmaps--;
6133d903220SDoug Rabson 		msgmaps[next].next = msghdr->msg_spot;
6143d903220SDoug Rabson 		msghdr->msg_spot = next;
6153d903220SDoug Rabson 		segs_needed--;
6163d903220SDoug Rabson 	}
6173d903220SDoug Rabson 
6183d903220SDoug Rabson 	/*
6193d903220SDoug Rabson 	 * Copy in the message type
6203d903220SDoug Rabson 	 */
6213d903220SDoug Rabson 
6223d903220SDoug Rabson 	if ((eval = copyin(user_msgp, &msghdr->msg_type,
6233d903220SDoug Rabson 	    sizeof(msghdr->msg_type))) != 0) {
6243d903220SDoug Rabson #ifdef MSG_DEBUG_OK
6253d903220SDoug Rabson 		printf("error %d copying the message type\n", eval);
6263d903220SDoug Rabson #endif
6273d903220SDoug Rabson 		msg_freehdr(msghdr);
6283d903220SDoug Rabson 		msqptr->msg_perm.mode &= ~MSG_LOCKED;
6293d903220SDoug Rabson 		wakeup((caddr_t)msqptr);
6303d903220SDoug Rabson 		return(eval);
6313d903220SDoug Rabson 	}
6323d903220SDoug Rabson 	user_msgp += sizeof(msghdr->msg_type);
6333d903220SDoug Rabson 
6343d903220SDoug Rabson 	/*
6353d903220SDoug Rabson 	 * Validate the message type
6363d903220SDoug Rabson 	 */
6373d903220SDoug Rabson 
6383d903220SDoug Rabson 	if (msghdr->msg_type < 1) {
6393d903220SDoug Rabson 		msg_freehdr(msghdr);
6403d903220SDoug Rabson 		msqptr->msg_perm.mode &= ~MSG_LOCKED;
6413d903220SDoug Rabson 		wakeup((caddr_t)msqptr);
6423d903220SDoug Rabson #ifdef MSG_DEBUG_OK
6433d903220SDoug Rabson 		printf("mtype (%d) < 1\n", msghdr->msg_type);
6443d903220SDoug Rabson #endif
6453d903220SDoug Rabson 		return(EINVAL);
6463d903220SDoug Rabson 	}
6473d903220SDoug Rabson 
6483d903220SDoug Rabson 	/*
6493d903220SDoug Rabson 	 * Copy in the message body
6503d903220SDoug Rabson 	 */
6513d903220SDoug Rabson 
6523d903220SDoug Rabson 	next = msghdr->msg_spot;
6533d903220SDoug Rabson 	while (msgsz > 0) {
6543d903220SDoug Rabson 		size_t tlen;
6553d903220SDoug Rabson 		if (msgsz > msginfo.msgssz)
6563d903220SDoug Rabson 			tlen = msginfo.msgssz;
6573d903220SDoug Rabson 		else
6583d903220SDoug Rabson 			tlen = msgsz;
6593d903220SDoug Rabson 		if (next <= -1)
6603d903220SDoug Rabson 			panic("next too low #2");
6613d903220SDoug Rabson 		if (next >= msginfo.msgseg)
6623d903220SDoug Rabson 			panic("next out of range #2");
6633d903220SDoug Rabson 		if ((eval = copyin(user_msgp, &msgpool[next * msginfo.msgssz],
6643d903220SDoug Rabson 		    tlen)) != 0) {
6653d903220SDoug Rabson #ifdef MSG_DEBUG_OK
6663d903220SDoug Rabson 			printf("error %d copying in message segment\n", eval);
6673d903220SDoug Rabson #endif
6683d903220SDoug Rabson 			msg_freehdr(msghdr);
6693d903220SDoug Rabson 			msqptr->msg_perm.mode &= ~MSG_LOCKED;
6703d903220SDoug Rabson 			wakeup((caddr_t)msqptr);
6713d903220SDoug Rabson 			return(eval);
6723d903220SDoug Rabson 		}
6733d903220SDoug Rabson 		msgsz -= tlen;
6743d903220SDoug Rabson 		user_msgp += tlen;
6753d903220SDoug Rabson 		next = msgmaps[next].next;
6763d903220SDoug Rabson 	}
6773d903220SDoug Rabson 	if (next != -1)
6783d903220SDoug Rabson 		panic("didn't use all the msg segments");
6793d903220SDoug Rabson 
6803d903220SDoug Rabson 	/*
6813d903220SDoug Rabson 	 * We've got the message.  Unlock the msqid_ds.
6823d903220SDoug Rabson 	 */
6833d903220SDoug Rabson 
6843d903220SDoug Rabson 	msqptr->msg_perm.mode &= ~MSG_LOCKED;
6853d903220SDoug Rabson 
6863d903220SDoug Rabson 	/*
6873d903220SDoug Rabson 	 * Make sure that the msqid_ds is still allocated.
6883d903220SDoug Rabson 	 */
6893d903220SDoug Rabson 
6903d903220SDoug Rabson 	if (msqptr->msg_qbytes == 0) {
6913d903220SDoug Rabson 		msg_freehdr(msghdr);
6923d903220SDoug Rabson 		wakeup((caddr_t)msqptr);
6933d903220SDoug Rabson 		/* The SVID says to return EIDRM. */
6943d903220SDoug Rabson #ifdef EIDRM
6953d903220SDoug Rabson 		return(EIDRM);
6963d903220SDoug Rabson #else
6973d903220SDoug Rabson 		/* Unfortunately, BSD doesn't define that code yet! */
6983d903220SDoug Rabson 		return(EINVAL);
6993d903220SDoug Rabson #endif
7003d903220SDoug Rabson 	}
7013d903220SDoug Rabson 
7023d903220SDoug Rabson 	/*
7033d903220SDoug Rabson 	 * Put the message into the queue
7043d903220SDoug Rabson 	 */
7053d903220SDoug Rabson 
7063d903220SDoug Rabson 	if (msqptr->msg_first == NULL) {
7073d903220SDoug Rabson 		msqptr->msg_first = msghdr;
7083d903220SDoug Rabson 		msqptr->msg_last = msghdr;
7093d903220SDoug Rabson 	} else {
7103d903220SDoug Rabson 		msqptr->msg_last->msg_next = msghdr;
7113d903220SDoug Rabson 		msqptr->msg_last = msghdr;
7123d903220SDoug Rabson 	}
7133d903220SDoug Rabson 	msqptr->msg_last->msg_next = NULL;
7143d903220SDoug Rabson 
7153d903220SDoug Rabson 	msqptr->msg_cbytes += msghdr->msg_ts;
7163d903220SDoug Rabson 	msqptr->msg_qnum++;
7173d903220SDoug Rabson 	msqptr->msg_lspid = p->p_pid;
7183d903220SDoug Rabson 	msqptr->msg_stime = time.tv_sec;
7193d903220SDoug Rabson 
7203d903220SDoug Rabson 	wakeup((caddr_t)msqptr);
7213d903220SDoug Rabson 	*retval = 0;
7223d903220SDoug Rabson 	return(0);
7233d903220SDoug Rabson }
7243d903220SDoug Rabson 
7253d903220SDoug Rabson struct msgrcv_args {
7263d903220SDoug Rabson 	int	msqid;
7273d903220SDoug Rabson 	void	*msgp;
7283d903220SDoug Rabson 	size_t	msgsz;
7293d903220SDoug Rabson 	long	msgtyp;
7303d903220SDoug Rabson 	int	msgflg;
7313d903220SDoug Rabson };
7323d903220SDoug Rabson 
7333d903220SDoug Rabson int
7343d903220SDoug Rabson msgrcv(p, uap, retval)
7353d903220SDoug Rabson 	struct proc *p;
7363d903220SDoug Rabson 	register struct msgrcv_args *uap;
7373d903220SDoug Rabson 	int *retval;
7383d903220SDoug Rabson {
7393d903220SDoug Rabson 	int msqid = uap->msqid;
7403d903220SDoug Rabson 	void *user_msgp = uap->msgp;
7413d903220SDoug Rabson 	size_t msgsz = uap->msgsz;
7423d903220SDoug Rabson 	long msgtyp = uap->msgtyp;
7433d903220SDoug Rabson 	int msgflg = uap->msgflg;
7443d903220SDoug Rabson 	size_t len;
7453d903220SDoug Rabson 	struct ucred *cred = p->p_ucred;
7463d903220SDoug Rabson 	register struct msqid_ds *msqptr;
7473d903220SDoug Rabson 	register struct msg *msghdr;
7483d903220SDoug Rabson 	int eval;
7493d903220SDoug Rabson 	short next;
7503d903220SDoug Rabson 
7513d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7523d903220SDoug Rabson 	printf("call to msgrcv(%d, 0x%x, %d, %ld, %d)\n", msqid, user_msgp,
7533d903220SDoug Rabson 	    msgsz, msgtyp, msgflg);
7543d903220SDoug Rabson #endif
7553d903220SDoug Rabson 
7563d903220SDoug Rabson 	msqid = IPCID_TO_IX(msqid);
7573d903220SDoug Rabson 
7583d903220SDoug Rabson 	if (msqid < 0 || msqid >= msginfo.msgmni) {
7593d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7603d903220SDoug Rabson 		printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
7613d903220SDoug Rabson 		    msginfo.msgmni);
7623d903220SDoug Rabson #endif
7633d903220SDoug Rabson 		return(EINVAL);
7643d903220SDoug Rabson 	}
7653d903220SDoug Rabson 
7663d903220SDoug Rabson 	msqptr = &msqids[msqid];
7673d903220SDoug Rabson 	if (msqptr->msg_qbytes == 0) {
7683d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7693d903220SDoug Rabson 		printf("no such message queue id\n");
7703d903220SDoug Rabson #endif
7713d903220SDoug Rabson 		return(EINVAL);
7723d903220SDoug Rabson 	}
7733d903220SDoug Rabson 	if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
7743d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7753d903220SDoug Rabson 		printf("wrong sequence number\n");
7763d903220SDoug Rabson #endif
7773d903220SDoug Rabson 		return(EINVAL);
7783d903220SDoug Rabson 	}
7793d903220SDoug Rabson 
7803d903220SDoug Rabson 	if ((eval = ipcperm(cred, &msqptr->msg_perm, IPC_R))) {
7813d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7823d903220SDoug Rabson 		printf("requester doesn't have read access\n");
7833d903220SDoug Rabson #endif
7843d903220SDoug Rabson 		return(eval);
7853d903220SDoug Rabson 	}
7863d903220SDoug Rabson 
7873d903220SDoug Rabson 	msghdr = NULL;
7883d903220SDoug Rabson 	while (msghdr == NULL) {
7893d903220SDoug Rabson 		if (msgtyp == 0) {
7903d903220SDoug Rabson 			msghdr = msqptr->msg_first;
7913d903220SDoug Rabson 			if (msghdr != NULL) {
7923d903220SDoug Rabson 				if (msgsz < msghdr->msg_ts &&
7933d903220SDoug Rabson 				    (msgflg & MSG_NOERROR) == 0) {
7943d903220SDoug Rabson #ifdef MSG_DEBUG_OK
7953d903220SDoug Rabson 					printf("first message on the queue is too big (want %d, got %d)\n",
7963d903220SDoug Rabson 					    msgsz, msghdr->msg_ts);
7973d903220SDoug Rabson #endif
7983d903220SDoug Rabson 					return(E2BIG);
7993d903220SDoug Rabson 				}
8003d903220SDoug Rabson 				if (msqptr->msg_first == msqptr->msg_last) {
8013d903220SDoug Rabson 					msqptr->msg_first = NULL;
8023d903220SDoug Rabson 					msqptr->msg_last = NULL;
8033d903220SDoug Rabson 				} else {
8043d903220SDoug Rabson 					msqptr->msg_first = msghdr->msg_next;
8053d903220SDoug Rabson 					if (msqptr->msg_first == NULL)
8063d903220SDoug Rabson 						panic("msg_first/last screwed up #1");
8073d903220SDoug Rabson 				}
8083d903220SDoug Rabson 			}
8093d903220SDoug Rabson 		} else {
8103d903220SDoug Rabson 			struct msg *previous;
8113d903220SDoug Rabson 			struct msg **prev;
8123d903220SDoug Rabson 
8133d903220SDoug Rabson 			previous = NULL;
8143d903220SDoug Rabson 			prev = &(msqptr->msg_first);
8153d903220SDoug Rabson 			while ((msghdr = *prev) != NULL) {
8163d903220SDoug Rabson 				/*
8173d903220SDoug Rabson 				 * Is this message's type an exact match or is
8183d903220SDoug Rabson 				 * this message's type less than or equal to
8193d903220SDoug Rabson 				 * the absolute value of a negative msgtyp?
8203d903220SDoug Rabson 				 * Note that the second half of this test can
8213d903220SDoug Rabson 				 * NEVER be true if msgtyp is positive since
8223d903220SDoug Rabson 				 * msg_type is always positive!
8233d903220SDoug Rabson 				 */
8243d903220SDoug Rabson 
8253d903220SDoug Rabson 				if (msgtyp == msghdr->msg_type ||
8263d903220SDoug Rabson 				    msghdr->msg_type <= -msgtyp) {
8273d903220SDoug Rabson #ifdef MSG_DEBUG_OK
8283d903220SDoug Rabson 					printf("found message type %d, requested %d\n",
8293d903220SDoug Rabson 					    msghdr->msg_type, msgtyp);
8303d903220SDoug Rabson #endif
8313d903220SDoug Rabson 					if (msgsz < msghdr->msg_ts &&
8323d903220SDoug Rabson 					    (msgflg & MSG_NOERROR) == 0) {
8333d903220SDoug Rabson #ifdef MSG_DEBUG_OK
8343d903220SDoug Rabson 						printf("requested message on the queue is too big (want %d, got %d)\n",
8353d903220SDoug Rabson 						    msgsz, msghdr->msg_ts);
8363d903220SDoug Rabson #endif
8373d903220SDoug Rabson 						return(E2BIG);
8383d903220SDoug Rabson 					}
8393d903220SDoug Rabson 					*prev = msghdr->msg_next;
8403d903220SDoug Rabson 					if (msghdr == msqptr->msg_last) {
8413d903220SDoug Rabson 						if (previous == NULL) {
8423d903220SDoug Rabson 							if (prev !=
8433d903220SDoug Rabson 							    &msqptr->msg_first)
8443d903220SDoug Rabson 								panic("msg_first/last screwed up #2");
8453d903220SDoug Rabson 							msqptr->msg_first =
8463d903220SDoug Rabson 							    NULL;
8473d903220SDoug Rabson 							msqptr->msg_last =
8483d903220SDoug Rabson 							    NULL;
8493d903220SDoug Rabson 						} else {
8503d903220SDoug Rabson 							if (prev ==
8513d903220SDoug Rabson 							    &msqptr->msg_first)
8523d903220SDoug Rabson 								panic("msg_first/last screwed up #3");
8533d903220SDoug Rabson 							msqptr->msg_last =
8543d903220SDoug Rabson 							    previous;
8553d903220SDoug Rabson 						}
8563d903220SDoug Rabson 					}
8573d903220SDoug Rabson 					break;
8583d903220SDoug Rabson 				}
8593d903220SDoug Rabson 				previous = msghdr;
8603d903220SDoug Rabson 				prev = &(msghdr->msg_next);
8613d903220SDoug Rabson 			}
8623d903220SDoug Rabson 		}
8633d903220SDoug Rabson 
8643d903220SDoug Rabson 		/*
8653d903220SDoug Rabson 		 * We've either extracted the msghdr for the appropriate
8663d903220SDoug Rabson 		 * message or there isn't one.
8673d903220SDoug Rabson 		 * If there is one then bail out of this loop.
8683d903220SDoug Rabson 		 */
8693d903220SDoug Rabson 
8703d903220SDoug Rabson 		if (msghdr != NULL)
8713d903220SDoug Rabson 			break;
8723d903220SDoug Rabson 
8733d903220SDoug Rabson 		/*
8743d903220SDoug Rabson 		 * Hmph!  No message found.  Does the user want to wait?
8753d903220SDoug Rabson 		 */
8763d903220SDoug Rabson 
8773d903220SDoug Rabson 		if ((msgflg & IPC_NOWAIT) != 0) {
8783d903220SDoug Rabson #ifdef MSG_DEBUG_OK
8793d903220SDoug Rabson 			printf("no appropriate message found (msgtyp=%d)\n",
8803d903220SDoug Rabson 			    msgtyp);
8813d903220SDoug Rabson #endif
8823d903220SDoug Rabson 			/* The SVID says to return ENOMSG. */
8833d903220SDoug Rabson #ifdef ENOMSG
8843d903220SDoug Rabson 			return(ENOMSG);
8853d903220SDoug Rabson #else
8863d903220SDoug Rabson 			/* Unfortunately, BSD doesn't define that code yet! */
8873d903220SDoug Rabson 			return(EAGAIN);
8883d903220SDoug Rabson #endif
8893d903220SDoug Rabson 		}
8903d903220SDoug Rabson 
8913d903220SDoug Rabson 		/*
8923d903220SDoug Rabson 		 * Wait for something to happen
8933d903220SDoug Rabson 		 */
8943d903220SDoug Rabson 
8953d903220SDoug Rabson #ifdef MSG_DEBUG_OK
8963d903220SDoug Rabson 		printf("msgrcv:  goodnight\n");
8973d903220SDoug Rabson #endif
8983d903220SDoug Rabson 		eval = tsleep((caddr_t)msqptr, (PZERO - 4) | PCATCH, "msgwait",
8993d903220SDoug Rabson 		    0);
9003d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9013d903220SDoug Rabson 		printf("msgrcv:  good morning (eval=%d)\n", eval);
9023d903220SDoug Rabson #endif
9033d903220SDoug Rabson 
9043d903220SDoug Rabson 		if (eval != 0) {
9053d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9063d903220SDoug Rabson 			printf("msgsnd:  interrupted system call\n");
9073d903220SDoug Rabson #endif
9083d903220SDoug Rabson 			return(EINTR);
9093d903220SDoug Rabson 		}
9103d903220SDoug Rabson 
9113d903220SDoug Rabson 		/*
9123d903220SDoug Rabson 		 * Make sure that the msq queue still exists
9133d903220SDoug Rabson 		 */
9143d903220SDoug Rabson 
9153d903220SDoug Rabson 		if (msqptr->msg_qbytes == 0 ||
9163d903220SDoug Rabson 		    msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
9173d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9183d903220SDoug Rabson 			printf("msqid deleted\n");
9193d903220SDoug Rabson #endif
9203d903220SDoug Rabson 			/* The SVID says to return EIDRM. */
9213d903220SDoug Rabson #ifdef EIDRM
9223d903220SDoug Rabson 			return(EIDRM);
9233d903220SDoug Rabson #else
9243d903220SDoug Rabson 			/* Unfortunately, BSD doesn't define that code yet! */
9253d903220SDoug Rabson 			return(EINVAL);
9263d903220SDoug Rabson #endif
9273d903220SDoug Rabson 		}
9283d903220SDoug Rabson 	}
9293d903220SDoug Rabson 
9303d903220SDoug Rabson 	/*
9313d903220SDoug Rabson 	 * Return the message to the user.
9323d903220SDoug Rabson 	 *
9333d903220SDoug Rabson 	 * First, do the bookkeeping (before we risk being interrupted).
9343d903220SDoug Rabson 	 */
9353d903220SDoug Rabson 
9363d903220SDoug Rabson 	msqptr->msg_cbytes -= msghdr->msg_ts;
9373d903220SDoug Rabson 	msqptr->msg_qnum--;
9383d903220SDoug Rabson 	msqptr->msg_lrpid = p->p_pid;
9393d903220SDoug Rabson 	msqptr->msg_rtime = time.tv_sec;
9403d903220SDoug Rabson 
9413d903220SDoug Rabson 	/*
9423d903220SDoug Rabson 	 * Make msgsz the actual amount that we'll be returning.
9433d903220SDoug Rabson 	 * Note that this effectively truncates the message if it is too long
9443d903220SDoug Rabson 	 * (since msgsz is never increased).
9453d903220SDoug Rabson 	 */
9463d903220SDoug Rabson 
9473d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9483d903220SDoug Rabson 	printf("found a message, msgsz=%d, msg_ts=%d\n", msgsz,
9493d903220SDoug Rabson 	    msghdr->msg_ts);
9503d903220SDoug Rabson #endif
9513d903220SDoug Rabson 	if (msgsz > msghdr->msg_ts)
9523d903220SDoug Rabson 		msgsz = msghdr->msg_ts;
9533d903220SDoug Rabson 
9543d903220SDoug Rabson 	/*
9553d903220SDoug Rabson 	 * Return the type to the user.
9563d903220SDoug Rabson 	 */
9573d903220SDoug Rabson 
9583d903220SDoug Rabson 	eval = copyout((caddr_t)&(msghdr->msg_type), user_msgp,
9593d903220SDoug Rabson 	    sizeof(msghdr->msg_type));
9603d903220SDoug Rabson 	if (eval != 0) {
9613d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9623d903220SDoug Rabson 		printf("error (%d) copying out message type\n", eval);
9633d903220SDoug Rabson #endif
9643d903220SDoug Rabson 		msg_freehdr(msghdr);
9653d903220SDoug Rabson 		wakeup((caddr_t)msqptr);
9663d903220SDoug Rabson 		return(eval);
9673d903220SDoug Rabson 	}
9683d903220SDoug Rabson 	user_msgp += sizeof(msghdr->msg_type);
9693d903220SDoug Rabson 
9703d903220SDoug Rabson 	/*
9713d903220SDoug Rabson 	 * Return the segments to the user
9723d903220SDoug Rabson 	 */
9733d903220SDoug Rabson 
9743d903220SDoug Rabson 	next = msghdr->msg_spot;
9753d903220SDoug Rabson 	for (len = 0; len < msgsz; len += msginfo.msgssz) {
9763d903220SDoug Rabson 		size_t tlen;
9773d903220SDoug Rabson 
9783d903220SDoug Rabson 		if (msgsz > msginfo.msgssz)
9793d903220SDoug Rabson 			tlen = msginfo.msgssz;
9803d903220SDoug Rabson 		else
9813d903220SDoug Rabson 			tlen = msgsz;
9823d903220SDoug Rabson 		if (next <= -1)
9833d903220SDoug Rabson 			panic("next too low #3");
9843d903220SDoug Rabson 		if (next >= msginfo.msgseg)
9853d903220SDoug Rabson 			panic("next out of range #3");
9863d903220SDoug Rabson 		eval = copyout((caddr_t)&msgpool[next * msginfo.msgssz],
9873d903220SDoug Rabson 		    user_msgp, tlen);
9883d903220SDoug Rabson 		if (eval != 0) {
9893d903220SDoug Rabson #ifdef MSG_DEBUG_OK
9903d903220SDoug Rabson 			printf("error (%d) copying out message segment\n",
9913d903220SDoug Rabson 			    eval);
9923d903220SDoug Rabson #endif
9933d903220SDoug Rabson 			msg_freehdr(msghdr);
9943d903220SDoug Rabson 			wakeup((caddr_t)msqptr);
9953d903220SDoug Rabson 			return(eval);
9963d903220SDoug Rabson 		}
9973d903220SDoug Rabson 		user_msgp += tlen;
9983d903220SDoug Rabson 		next = msgmaps[next].next;
9993d903220SDoug Rabson 	}
10003d903220SDoug Rabson 
10013d903220SDoug Rabson 	/*
10023d903220SDoug Rabson 	 * Done, return the actual number of bytes copied out.
10033d903220SDoug Rabson 	 */
10043d903220SDoug Rabson 
10053d903220SDoug Rabson 	msg_freehdr(msghdr);
10063d903220SDoug Rabson 	wakeup((caddr_t)msqptr);
10073d903220SDoug Rabson 	*retval = msgsz;
10083d903220SDoug Rabson 	return(0);
10093d903220SDoug Rabson }
1010