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