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