sys/kern/sysv_msg.c

   1 /* $FreeBSD: src/sys/kern/sysv_msg.c,v 1.23.2.5 2002/12/31 08:54:53 maxim Exp $ */
   2 /* $DragonFly: src/sys/kern/sysv_msg.c,v 1.4 2003/06/25 03:55:57 dillon Exp $ */
   3
   4 /*
   5  * Implementation of SVID messages
   6  *
   7  * Author:  Daniel Boulet
   8  *
   9  * Copyright 1993 Daniel Boulet and RTMX Inc.
  10  *
  11  * This system call was implemented by Daniel Boulet under contract from RTMX.
  12  *
  13  * Redistribution and use in source forms, with and without modification,
  14  * are permitted provided that this entire comment appears intact.
  15  *
  16  * Redistribution in binary form may occur without any restrictions.
  17  * Obviously, it would be nice if you gave credit where credit is due
  18  * but requiring it would be too onerous.
  19  *
  20  * This software is provided ``AS IS'' without any warranties of any kind.
  21  */
  22
  23 #include "opt_sysvipc.h"
  24
  25 #include <sys/param.h>
  26 #include <sys/systm.h>
  27 #include <sys/sysproto.h>
  28 #include <sys/kernel.h>
  29 #include <sys/proc.h>
  30 #include <sys/msg.h>
  31 #include <sys/sysent.h>
  32 #include <sys/sysctl.h>
  33 #include <sys/malloc.h>
  34 #include <sys/jail.h>
  35
  36 static MALLOC_DEFINE(M_MSG, "msg", "SVID compatible message queues");
  37
  38 static void msginit __P((void *));
  39
  40 #define MSG_DEBUG
  41 #undef MSG_DEBUG_OK
  42
  43 static void msg_freehdr __P((struct msg *msghdr));
  44
  45 /* XXX casting to (sy_call_t *) is bogus, as usual. */
  46 static sy_call_t *msgcalls[] = {
  47         (sy_call_t *)msgctl, (sy_call_t *)msgget,
  48         (sy_call_t *)msgsnd, (sy_call_t *)msgrcv
  49 };
  50
  51 struct msg {
  52         struct  msg *msg_next;  /* next msg in the chain */
  53         long    msg_type;       /* type of this message */
  54                                 /* >0 -> type of this message */
  55                                 /* 0 -> free header */
  56         u_short msg_ts;         /* size of this message */
  57         short   msg_spot;       /* location of start of msg in buffer */
  58 };
  59
  60
  61 #ifndef MSGSSZ
  62 #define MSGSSZ  8               /* Each segment must be 2^N long */
  63 #endif
  64 #ifndef MSGSEG
  65 #define MSGSEG  2048            /* must be less than 32767 */
  66 #endif
  67 #define MSGMAX  (MSGSSZ*MSGSEG)
  68 #ifndef MSGMNB
  69 #define MSGMNB  2048            /* max # of bytes in a queue */
  70 #endif
  71 #ifndef MSGMNI
  72 #define MSGMNI  40
  73 #endif
  74 #ifndef MSGTQL
  75 #define MSGTQL  40
  76 #endif
  77
  78 /*
  79  * Based on the configuration parameters described in an SVR2 (yes, two)
  80  * config(1m) man page.
  81  *
  82  * Each message is broken up and stored in segments that are msgssz bytes
  83  * long.  For efficiency reasons, this should be a power of two.  Also,
  84  * it doesn't make sense if it is less than 8 or greater than about 256.
  85  * Consequently, msginit in kern/sysv_msg.c checks that msgssz is a power of
  86  * two between 8 and 1024 inclusive (and panic's if it isn't).
  87  */
  88 struct msginfo msginfo = {
  89                 MSGMAX,         /* max chars in a message */
  90                 MSGMNI,         /* # of message queue identifiers */
  91                 MSGMNB,         /* max chars in a queue */
  92                 MSGTQL,         /* max messages in system */
  93                 MSGSSZ,         /* size of a message segment */
  94                                 /* (must be small power of 2 greater than 4) */
  95                 MSGSEG          /* number of message segments */
  96 };
  97
  98 /*
  99  * macros to convert between msqid_ds's and msqid's.
 100  * (specific to this implementation)
 101  */
 102 #define MSQID(ix,ds)    ((ix) & 0xffff | (((ds).msg_perm.seq << 16) & 0xffff0000))
 103 #define MSQID_IX(id)    ((id) & 0xffff)
 104 #define MSQID_SEQ(id)   (((id) >> 16) & 0xffff)
 105
 106 /*
 107  * The rest of this file is specific to this particular implementation.
 108  */
 109
 110 struct msgmap {
 111         short   next;           /* next segment in buffer */
 112                                 /* -1 -> available */
 113                                 /* 0..(MSGSEG-1) -> index of next segment */
 114 };
 115
 116 #define MSG_LOCKED      01000   /* Is this msqid_ds locked? */
 117
 118 static int nfree_msgmaps;       /* # of free map entries */
 119 static short free_msgmaps;      /* head of linked list of free map entries */
 120 static struct msg *free_msghdrs;/* list of free msg headers */
 121 static char *msgpool;           /* MSGMAX byte long msg buffer pool */
 122 static struct msgmap *msgmaps;  /* MSGSEG msgmap structures */
 123 static struct msg *msghdrs;     /* MSGTQL msg headers */
 124 static struct msqid_ds *msqids; /* MSGMNI msqid_ds struct's */
 125
 126 static void
 127 msginit(dummy)
 128         void *dummy;
 129 {
 130         register int i;
 131
 132         msginfo.msgmax = msginfo.msgseg * msginfo.msgssz;
 133         msgpool = malloc(msginfo.msgmax, M_MSG, M_WAITOK);
 134         if (msgpool == NULL)
 135                 panic("msgpool is NULL");
 136         msgmaps = malloc(sizeof(struct msgmap) * msginfo.msgseg, M_MSG, M_WAITOK);
 137         if (msgmaps == NULL)
 138                 panic("msgmaps is NULL");
 139         msghdrs = malloc(sizeof(struct msg) * msginfo.msgtql, M_MSG, M_WAITOK);
 140         if (msghdrs == NULL)
 141                 panic("msghdrs is NULL");
 142         msqids = malloc(sizeof(struct msqid_ds) * msginfo.msgmni, M_MSG, M_WAITOK);
 143         if (msqids == NULL)
 144                 panic("msqids is NULL");
 145
 146         /*
 147          * msginfo.msgssz should be a power of two for efficiency reasons.
 148          * It is also pretty silly if msginfo.msgssz is less than 8
 149          * or greater than about 256 so ...
 150          */
 151
 152         i = 8;
 153         while (i < 1024 && i != msginfo.msgssz)
 154                 i <<= 1;
 155         if (i != msginfo.msgssz) {
 156                 printf("msginfo.msgssz=%d (0x%x)\n", msginfo.msgssz,
 157                     msginfo.msgssz);
 158                 panic("msginfo.msgssz not a small power of 2");
 159         }
 160
 161         if (msginfo.msgseg > 32767) {
 162                 printf("msginfo.msgseg=%d\n", msginfo.msgseg);
 163                 panic("msginfo.msgseg > 32767");
 164         }
 165
 166         if (msgmaps == NULL)
 167                 panic("msgmaps is NULL");
 168
 169         for (i = 0; i < msginfo.msgseg; i++) {
 170                 if (i > 0)
 171                         msgmaps[i-1].next = i;
 172                 msgmaps[i].next = -1;   /* implies entry is available */
 173         }
 174         free_msgmaps = 0;
 175         nfree_msgmaps = msginfo.msgseg;
 176
 177         if (msghdrs == NULL)
 178                 panic("msghdrs is NULL");
 179
 180         for (i = 0; i < msginfo.msgtql; i++) {
 181                 msghdrs[i].msg_type = 0;
 182                 if (i > 0)
 183                         msghdrs[i-1].msg_next = &msghdrs[i];
 184                 msghdrs[i].msg_next = NULL;
 185         }
 186         free_msghdrs = &msghdrs[0];
 187
 188         if (msqids == NULL)
 189                 panic("msqids is NULL");
 190
 191         for (i = 0; i < msginfo.msgmni; i++) {
 192                 msqids[i].msg_qbytes = 0;       /* implies entry is available */
 193                 msqids[i].msg_perm.seq = 0;     /* reset to a known value */
 194                 msqids[i].msg_perm.mode = 0;
 195         }
 196 }
 197 SYSINIT(sysv_msg, SI_SUB_SYSV_MSG, SI_ORDER_FIRST, msginit, NULL)
 198
 199 /*
 200  * Entry point for all MSG calls
 201  *
 202  * msgsys_args(u_int which, int a2, ...) (VARARGS)
 203  */
 204 int
 205 msgsys(struct msgsys_args *uap)
 206 {
 207         struct proc *p = curproc;
 208
 209         if (!jail_sysvipc_allowed && p->p_ucred->cr_prison != NULL)
 210                 return (ENOSYS);
 211
 212         if (uap->which >= sizeof(msgcalls)/sizeof(msgcalls[0]))
 213                 return (EINVAL);
 214         return ((*msgcalls[uap->which])(&uap->a2));
 215 }
 216
 217 static void
 218 msg_freehdr(struct msg *msghdr)
 219 {
 220         while (msghdr->msg_ts > 0) {
 221                 short next;
 222                 if (msghdr->msg_spot < 0 || msghdr->msg_spot >= msginfo.msgseg)
 223                         panic("msghdr->msg_spot out of range");
 224                 next = msgmaps[msghdr->msg_spot].next;
 225                 msgmaps[msghdr->msg_spot].next = free_msgmaps;
 226                 free_msgmaps = msghdr->msg_spot;
 227                 nfree_msgmaps++;
 228                 msghdr->msg_spot = next;
 229                 if (msghdr->msg_ts >= msginfo.msgssz)
 230                         msghdr->msg_ts -= msginfo.msgssz;
 231                 else
 232                         msghdr->msg_ts = 0;
 233         }
 234         if (msghdr->msg_spot != -1)
 235                 panic("msghdr->msg_spot != -1");
 236         msghdr->msg_next = free_msghdrs;
 237         free_msghdrs = msghdr;
 238 }
 239
 240 #ifndef _SYS_SYSPROTO_H_
 241 struct msgctl_args {
 242         int     msqid;
 243         int     cmd;
 244         struct  msqid_ds *buf;
 245 };
 246 #endif
 247
 248 int
 249 msgctl(struct msgctl_args *uap)
 250 {
 251         struct thread *td = curthread;
 252         struct proc *p = td->td_proc;
 253         int msqid = uap->msqid;
 254         int cmd = uap->cmd;
 255         struct msqid_ds *user_msqptr = uap->buf;
 256         int rval, eval;
 257         struct msqid_ds msqbuf;
 258         register struct msqid_ds *msqptr;
 259
 260 #ifdef MSG_DEBUG_OK
 261         printf("call to msgctl(%d, %d, 0x%x)\n", msqid, cmd, user_msqptr);
 262 #endif
 263
 264         if (!jail_sysvipc_allowed && p->p_ucred->cr_prison != NULL)
 265                 return (ENOSYS);
 266
 267         msqid = IPCID_TO_IX(msqid);
 268
 269         if (msqid < 0 || msqid >= msginfo.msgmni) {
 270 #ifdef MSG_DEBUG_OK
 271                 printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
 272                     msginfo.msgmni);
 273 #endif
 274                 return(EINVAL);
 275         }
 276
 277         msqptr = &msqids[msqid];
 278
 279         if (msqptr->msg_qbytes == 0) {
 280 #ifdef MSG_DEBUG_OK
 281                 printf("no such msqid\n");
 282 #endif
 283                 return(EINVAL);
 284         }
 285         if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
 286 #ifdef MSG_DEBUG_OK
 287                 printf("wrong sequence number\n");
 288 #endif
 289                 return(EINVAL);
 290         }
 291
 292         eval = 0;
 293         rval = 0;
 294
 295         switch (cmd) {
 296
 297         case IPC_RMID:
 298         {
 299                 struct msg *msghdr;
 300                 if ((eval = ipcperm(p, &msqptr->msg_perm, IPC_M)))
 301                         return(eval);
 302                 /* Free the message headers */
 303                 msghdr = msqptr->msg_first;
 304                 while (msghdr != NULL) {
 305                         struct msg *msghdr_tmp;
 306
 307                         /* Free the segments of each message */
 308                         msqptr->msg_cbytes -= msghdr->msg_ts;
 309                         msqptr->msg_qnum--;
 310                         msghdr_tmp = msghdr;
 311                         msghdr = msghdr->msg_next;
 312                         msg_freehdr(msghdr_tmp);
 313                 }
 314
 315                 if (msqptr->msg_cbytes != 0)
 316                         panic("msg_cbytes is screwed up");
 317                 if (msqptr->msg_qnum != 0)
 318                         panic("msg_qnum is screwed up");
 319
 320                 msqptr->msg_qbytes = 0; /* Mark it as free */
 321
 322                 wakeup((caddr_t)msqptr);
 323         }
 324
 325                 break;
 326
 327         case IPC_SET:
 328                 if ((eval = ipcperm(p, &msqptr->msg_perm, IPC_M)))
 329                         return(eval);
 330                 if ((eval = copyin(user_msqptr, &msqbuf, sizeof(msqbuf))) != 0)
 331                         return(eval);
 332                 if (msqbuf.msg_qbytes > msqptr->msg_qbytes) {
 333                         eval = suser(td);
 334                         if (eval)
 335                                 return(eval);
 336                 }
 337                 if (msqbuf.msg_qbytes > msginfo.msgmnb) {
 338 #ifdef MSG_DEBUG_OK
 339                         printf("can't increase msg_qbytes beyond %d (truncating)\n",
 340                             msginfo.msgmnb);
 341 #endif
 342                         msqbuf.msg_qbytes = msginfo.msgmnb;     /* silently restrict qbytes to system limit */
 343                 }
 344                 if (msqbuf.msg_qbytes == 0) {
 345 #ifdef MSG_DEBUG_OK
 346                         printf("can't reduce msg_qbytes to 0\n");
 347 #endif
 348                         return(EINVAL);         /* non-standard errno! */
 349                 }
 350                 msqptr->msg_perm.uid = msqbuf.msg_perm.uid;     /* change the owner */
 351                 msqptr->msg_perm.gid = msqbuf.msg_perm.gid;     /* change the owner */
 352                 msqptr->msg_perm.mode = (msqptr->msg_perm.mode & ~0777) |
 353                     (msqbuf.msg_perm.mode & 0777);
 354                 msqptr->msg_qbytes = msqbuf.msg_qbytes;
 355                 msqptr->msg_ctime = time_second;
 356                 break;
 357
 358         case IPC_STAT:
 359                 if ((eval = ipcperm(p, &msqptr->msg_perm, IPC_R))) {
 360 #ifdef MSG_DEBUG_OK
 361                         printf("requester doesn't have read access\n");
 362 #endif
 363                         return(eval);
 364                 }
 365                 eval = copyout((caddr_t)msqptr, user_msqptr,
 366                     sizeof(struct msqid_ds));
 367                 break;
 368
 369         default:
 370 #ifdef MSG_DEBUG_OK
 371                 printf("invalid command %d\n", cmd);
 372 #endif
 373                 return(EINVAL);
 374         }
 375
 376         if (eval == 0)
 377                 p->p_retval[0] = rval;
 378         return(eval);
 379 }
 380
 381 #ifndef _SYS_SYSPROTO_H_
 382 struct msgget_args {
 383         key_t   key;
 384         int     msgflg;
 385 };
 386 #endif
 387
 388 int
 389 msgget(struct msgget_args *uap)
 390 {
 391         struct proc *p = curproc;
 392         int msqid, eval;
 393         int key = uap->key;
 394         int msgflg = uap->msgflg;
 395         struct ucred *cred = p->p_ucred;
 396         register struct msqid_ds *msqptr = NULL;
 397
 398 #ifdef MSG_DEBUG_OK
 399         printf("msgget(0x%x, 0%o)\n", key, msgflg);
 400 #endif
 401
 402         if (!jail_sysvipc_allowed && p->p_ucred->cr_prison != NULL)
 403                 return (ENOSYS);
 404
 405         if (key != IPC_PRIVATE) {
 406                 for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
 407                         msqptr = &msqids[msqid];
 408                         if (msqptr->msg_qbytes != 0 &&
 409                             msqptr->msg_perm.key == key)
 410                                 break;
 411                 }
 412                 if (msqid < msginfo.msgmni) {
 413 #ifdef MSG_DEBUG_OK
 414                         printf("found public key\n");
 415 #endif
 416                         if ((msgflg & IPC_CREAT) && (msgflg & IPC_EXCL)) {
 417 #ifdef MSG_DEBUG_OK
 418                                 printf("not exclusive\n");
 419 #endif
 420                                 return(EEXIST);
 421                         }
 422                         if ((eval = ipcperm(p, &msqptr->msg_perm, msgflg & 0700 ))) {
 423 #ifdef MSG_DEBUG_OK
 424                                 printf("requester doesn't have 0%o access\n",
 425                                     msgflg & 0700);
 426 #endif
 427                                 return(eval);
 428                         }
 429                         goto found;
 430                 }
 431         }
 432
 433 #ifdef MSG_DEBUG_OK
 434         printf("need to allocate the msqid_ds\n");
 435 #endif
 436         if (key == IPC_PRIVATE || (msgflg & IPC_CREAT)) {
 437                 for (msqid = 0; msqid < msginfo.msgmni; msqid++) {
 438                         /*
 439                          * Look for an unallocated and unlocked msqid_ds.
 440                          * msqid_ds's can be locked by msgsnd or msgrcv while
 441                          * they are copying the message in/out.  We can't
 442                          * re-use the entry until they release it.
 443                          */
 444                         msqptr = &msqids[msqid];
 445                         if (msqptr->msg_qbytes == 0 &&
 446                             (msqptr->msg_perm.mode & MSG_LOCKED) == 0)
 447                                 break;
 448                 }
 449                 if (msqid == msginfo.msgmni) {
 450 #ifdef MSG_DEBUG_OK
 451                         printf("no more msqid_ds's available\n");
 452 #endif
 453                         return(ENOSPC);
 454                 }
 455 #ifdef MSG_DEBUG_OK
 456                 printf("msqid %d is available\n", msqid);
 457 #endif
 458                 msqptr->msg_perm.key = key;
 459                 msqptr->msg_perm.cuid = cred->cr_uid;
 460                 msqptr->msg_perm.uid = cred->cr_uid;
 461                 msqptr->msg_perm.cgid = cred->cr_gid;
 462                 msqptr->msg_perm.gid = cred->cr_gid;
 463                 msqptr->msg_perm.mode = (msgflg & 0777);
 464                 /* Make sure that the returned msqid is unique */
 465                 msqptr->msg_perm.seq = (msqptr->msg_perm.seq + 1) & 0x7fff;
 466                 msqptr->msg_first = NULL;
 467                 msqptr->msg_last = NULL;
 468                 msqptr->msg_cbytes = 0;
 469                 msqptr->msg_qnum = 0;
 470                 msqptr->msg_qbytes = msginfo.msgmnb;
 471                 msqptr->msg_lspid = 0;
 472                 msqptr->msg_lrpid = 0;
 473                 msqptr->msg_stime = 0;
 474                 msqptr->msg_rtime = 0;
 475                 msqptr->msg_ctime = time_second;
 476         } else {
 477 #ifdef MSG_DEBUG_OK
 478                 printf("didn't find it and wasn't asked to create it\n");
 479 #endif
 480                 return(ENOENT);
 481         }
 482
 483 found:
 484         /* Construct the unique msqid */
 485         p->p_retval[0] = IXSEQ_TO_IPCID(msqid, msqptr->msg_perm);
 486         return(0);
 487 }
 488
 489 #ifndef _SYS_SYSPROTO_H_
 490 struct msgsnd_args {
 491         int     msqid;
 492         void    *msgp;
 493         size_t  msgsz;
 494         int     msgflg;
 495 };
 496 #endif
 497
 498 int
 499 msgsnd(struct msgsnd_args *uap)
 500 {
 501         struct proc *p = curproc;
 502         int msqid = uap->msqid;
 503         void *user_msgp = uap->msgp;
 504         size_t msgsz = uap->msgsz;
 505         int msgflg = uap->msgflg;
 506         int segs_needed, eval;
 507         register struct msqid_ds *msqptr;
 508         register struct msg *msghdr;
 509         short next;
 510
 511 #ifdef MSG_DEBUG_OK
 512         printf("call to msgsnd(%d, 0x%x, %d, %d)\n", msqid, user_msgp, msgsz,
 513             msgflg);
 514 #endif
 515
 516         if (!jail_sysvipc_allowed && p->p_ucred->cr_prison != NULL)
 517                 return (ENOSYS);
 518
 519         msqid = IPCID_TO_IX(msqid);
 520
 521         if (msqid < 0 || msqid >= msginfo.msgmni) {
 522 #ifdef MSG_DEBUG_OK
 523                 printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
 524                     msginfo.msgmni);
 525 #endif
 526                 return(EINVAL);
 527         }
 528
 529         msqptr = &msqids[msqid];
 530         if (msqptr->msg_qbytes == 0) {
 531 #ifdef MSG_DEBUG_OK
 532                 printf("no such message queue id\n");
 533 #endif
 534                 return(EINVAL);
 535         }
 536         if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
 537 #ifdef MSG_DEBUG_OK
 538                 printf("wrong sequence number\n");
 539 #endif
 540                 return(EINVAL);
 541         }
 542
 543         if ((eval = ipcperm(p, &msqptr->msg_perm, IPC_W))) {
 544 #ifdef MSG_DEBUG_OK
 545                 printf("requester doesn't have write access\n");
 546 #endif
 547                 return(eval);
 548         }
 549
 550         segs_needed = (msgsz + msginfo.msgssz - 1) / msginfo.msgssz;
 551 #ifdef MSG_DEBUG_OK
 552         printf("msgsz=%d, msgssz=%d, segs_needed=%d\n", msgsz, msginfo.msgssz,
 553             segs_needed);
 554 #endif
 555         for (;;) {
 556                 int need_more_resources = 0;
 557
 558                 /*
 559                  * check msgsz
 560                  * (inside this loop in case msg_qbytes changes while we sleep)
 561                  */
 562
 563                 if (msgsz > msqptr->msg_qbytes) {
 564 #ifdef MSG_DEBUG_OK
 565                         printf("msgsz > msqptr->msg_qbytes\n");
 566 #endif
 567                         return(EINVAL);
 568                 }
 569
 570                 if (msqptr->msg_perm.mode & MSG_LOCKED) {
 571 #ifdef MSG_DEBUG_OK
 572                         printf("msqid is locked\n");
 573 #endif
 574                         need_more_resources = 1;
 575                 }
 576                 if (msgsz + msqptr->msg_cbytes > msqptr->msg_qbytes) {
 577 #ifdef MSG_DEBUG_OK
 578                         printf("msgsz + msg_cbytes > msg_qbytes\n");
 579 #endif
 580                         need_more_resources = 1;
 581                 }
 582                 if (segs_needed > nfree_msgmaps) {
 583 #ifdef MSG_DEBUG_OK
 584                         printf("segs_needed > nfree_msgmaps\n");
 585 #endif
 586                         need_more_resources = 1;
 587                 }
 588                 if (free_msghdrs == NULL) {
 589 #ifdef MSG_DEBUG_OK
 590                         printf("no more msghdrs\n");
 591 #endif
 592                         need_more_resources = 1;
 593                 }
 594
 595                 if (need_more_resources) {
 596                         int we_own_it;
 597
 598                         if ((msgflg & IPC_NOWAIT) != 0) {
 599 #ifdef MSG_DEBUG_OK
 600                                 printf("need more resources but caller doesn't want to wait\n");
 601 #endif
 602                                 return(EAGAIN);
 603                         }
 604
 605                         if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0) {
 606 #ifdef MSG_DEBUG_OK
 607                                 printf("we don't own the msqid_ds\n");
 608 #endif
 609                                 we_own_it = 0;
 610                         } else {
 611                                 /* Force later arrivals to wait for our
 612                                    request */
 613 #ifdef MSG_DEBUG_OK
 614                                 printf("we own the msqid_ds\n");
 615 #endif
 616                                 msqptr->msg_perm.mode |= MSG_LOCKED;
 617                                 we_own_it = 1;
 618                         }
 619 #ifdef MSG_DEBUG_OK
 620                         printf("goodnight\n");
 621 #endif
 622                         eval = tsleep((caddr_t)msqptr, (PZERO - 4) | PCATCH,
 623                             "msgwait", 0);
 624 #ifdef MSG_DEBUG_OK
 625                         printf("good morning, eval=%d\n", eval);
 626 #endif
 627                         if (we_own_it)
 628                                 msqptr->msg_perm.mode &= ~MSG_LOCKED;
 629                         if (eval != 0) {
 630 #ifdef MSG_DEBUG_OK
 631                                 printf("msgsnd:  interrupted system call\n");
 632 #endif
 633                                 return(EINTR);
 634                         }
 635
 636                         /*
 637                          * Make sure that the msq queue still exists
 638                          */
 639
 640                         if (msqptr->msg_qbytes == 0) {
 641 #ifdef MSG_DEBUG_OK
 642                                 printf("msqid deleted\n");
 643 #endif
 644                                 return(EIDRM);
 645                         }
 646
 647                 } else {
 648 #ifdef MSG_DEBUG_OK
 649                         printf("got all the resources that we need\n");
 650 #endif
 651                         break;
 652                 }
 653         }
 654
 655         /*
 656          * We have the resources that we need.
 657          * Make sure!
 658          */
 659
 660         if (msqptr->msg_perm.mode & MSG_LOCKED)
 661                 panic("msg_perm.mode & MSG_LOCKED");
 662         if (segs_needed > nfree_msgmaps)
 663                 panic("segs_needed > nfree_msgmaps");
 664         if (msgsz + msqptr->msg_cbytes > msqptr->msg_qbytes)
 665                 panic("msgsz + msg_cbytes > msg_qbytes");
 666         if (free_msghdrs == NULL)
 667                 panic("no more msghdrs");
 668
 669         /*
 670          * Re-lock the msqid_ds in case we page-fault when copying in the
 671          * message
 672          */
 673
 674         if ((msqptr->msg_perm.mode & MSG_LOCKED) != 0)
 675                 panic("msqid_ds is already locked");
 676         msqptr->msg_perm.mode |= MSG_LOCKED;
 677
 678         /*
 679          * Allocate a message header
 680          */
 681
 682         msghdr = free_msghdrs;
 683         free_msghdrs = msghdr->msg_next;
 684         msghdr->msg_spot = -1;
 685         msghdr->msg_ts = msgsz;
 686
 687         /*
 688          * Allocate space for the message
 689          */
 690
 691         while (segs_needed > 0) {
 692                 if (nfree_msgmaps <= 0)
 693                         panic("not enough msgmaps");
 694                 if (free_msgmaps == -1)
 695                         panic("nil free_msgmaps");
 696                 next = free_msgmaps;
 697                 if (next <= -1)
 698                         panic("next too low #1");
 699                 if (next >= msginfo.msgseg)
 700                         panic("next out of range #1");
 701 #ifdef MSG_DEBUG_OK
 702                 printf("allocating segment %d to message\n", next);
 703 #endif
 704                 free_msgmaps = msgmaps[next].next;
 705                 nfree_msgmaps--;
 706                 msgmaps[next].next = msghdr->msg_spot;
 707                 msghdr->msg_spot = next;
 708                 segs_needed--;
 709         }
 710
 711         /*
 712          * Copy in the message type
 713          */
 714
 715         if ((eval = copyin(user_msgp, &msghdr->msg_type,
 716             sizeof(msghdr->msg_type))) != 0) {
 717 #ifdef MSG_DEBUG_OK
 718                 printf("error %d copying the message type\n", eval);
 719 #endif
 720                 msg_freehdr(msghdr);
 721                 msqptr->msg_perm.mode &= ~MSG_LOCKED;
 722                 wakeup((caddr_t)msqptr);
 723                 return(eval);
 724         }
 725         user_msgp = (char *)user_msgp + sizeof(msghdr->msg_type);
 726
 727         /*
 728          * Validate the message type
 729          */
 730
 731         if (msghdr->msg_type < 1) {
 732                 msg_freehdr(msghdr);
 733                 msqptr->msg_perm.mode &= ~MSG_LOCKED;
 734                 wakeup((caddr_t)msqptr);
 735 #ifdef MSG_DEBUG_OK
 736                 printf("mtype (%d) < 1\n", msghdr->msg_type);
 737 #endif
 738                 return(EINVAL);
 739         }
 740
 741         /*
 742          * Copy in the message body
 743          */
 744
 745         next = msghdr->msg_spot;
 746         while (msgsz > 0) {
 747                 size_t tlen;
 748                 if (msgsz > msginfo.msgssz)
 749                         tlen = msginfo.msgssz;
 750                 else
 751                         tlen = msgsz;
 752                 if (next <= -1)
 753                         panic("next too low #2");
 754                 if (next >= msginfo.msgseg)
 755                         panic("next out of range #2");
 756                 if ((eval = copyin(user_msgp, &msgpool[next * msginfo.msgssz],
 757                     tlen)) != 0) {
 758 #ifdef MSG_DEBUG_OK
 759                         printf("error %d copying in message segment\n", eval);
 760 #endif
 761                         msg_freehdr(msghdr);
 762                         msqptr->msg_perm.mode &= ~MSG_LOCKED;
 763                         wakeup((caddr_t)msqptr);
 764                         return(eval);
 765                 }
 766                 msgsz -= tlen;
 767                 user_msgp = (char *)user_msgp + tlen;
 768                 next = msgmaps[next].next;
 769         }
 770         if (next != -1)
 771                 panic("didn't use all the msg segments");
 772
 773         /*
 774          * We've got the message.  Unlock the msqid_ds.
 775          */
 776
 777         msqptr->msg_perm.mode &= ~MSG_LOCKED;
 778
 779         /*
 780          * Make sure that the msqid_ds is still allocated.
 781          */
 782
 783         if (msqptr->msg_qbytes == 0) {
 784                 msg_freehdr(msghdr);
 785                 wakeup((caddr_t)msqptr);
 786                 return(EIDRM);
 787         }
 788
 789         /*
 790          * Put the message into the queue
 791          */
 792
 793         if (msqptr->msg_first == NULL) {
 794                 msqptr->msg_first = msghdr;
 795                 msqptr->msg_last = msghdr;
 796         } else {
 797                 msqptr->msg_last->msg_next = msghdr;
 798                 msqptr->msg_last = msghdr;
 799         }
 800         msqptr->msg_last->msg_next = NULL;
 801
 802         msqptr->msg_cbytes += msghdr->msg_ts;
 803         msqptr->msg_qnum++;
 804         msqptr->msg_lspid = p->p_pid;
 805         msqptr->msg_stime = time_second;
 806
 807         wakeup((caddr_t)msqptr);
 808         p->p_retval[0] = 0;
 809         return(0);
 810 }
 811
 812 #ifndef _SYS_SYSPROTO_H_
 813 struct msgrcv_args {
 814         int     msqid;
 815         void    *msgp;
 816         size_t  msgsz;
 817         long    msgtyp;
 818         int     msgflg;
 819 };
 820 #endif
 821
 822 int
 823 msgrcv(struct msgrcv_args *uap)
 824 {
 825         struct proc *p = curproc;
 826         int msqid = uap->msqid;
 827         void *user_msgp = uap->msgp;
 828         size_t msgsz = uap->msgsz;
 829         long msgtyp = uap->msgtyp;
 830         int msgflg = uap->msgflg;
 831         size_t len;
 832         register struct msqid_ds *msqptr;
 833         register struct msg *msghdr;
 834         int eval;
 835         short next;
 836
 837 #ifdef MSG_DEBUG_OK
 838         printf("call to msgrcv(%d, 0x%x, %d, %ld, %d)\n", msqid, user_msgp,
 839             msgsz, msgtyp, msgflg);
 840 #endif
 841
 842         if (!jail_sysvipc_allowed && p->p_ucred->cr_prison != NULL)
 843                 return (ENOSYS);
 844
 845         msqid = IPCID_TO_IX(msqid);
 846
 847         if (msqid < 0 || msqid >= msginfo.msgmni) {
 848 #ifdef MSG_DEBUG_OK
 849                 printf("msqid (%d) out of range (0<=msqid<%d)\n", msqid,
 850                     msginfo.msgmni);
 851 #endif
 852                 return(EINVAL);
 853         }
 854
 855         msqptr = &msqids[msqid];
 856         if (msqptr->msg_qbytes == 0) {
 857 #ifdef MSG_DEBUG_OK
 858                 printf("no such message queue id\n");
 859 #endif
 860                 return(EINVAL);
 861         }
 862         if (msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
 863 #ifdef MSG_DEBUG_OK
 864                 printf("wrong sequence number\n");
 865 #endif
 866                 return(EINVAL);
 867         }
 868
 869         if ((eval = ipcperm(p, &msqptr->msg_perm, IPC_R))) {
 870 #ifdef MSG_DEBUG_OK
 871                 printf("requester doesn't have read access\n");
 872 #endif
 873                 return(eval);
 874         }
 875
 876         msghdr = NULL;
 877         while (msghdr == NULL) {
 878                 if (msgtyp == 0) {
 879                         msghdr = msqptr->msg_first;
 880                         if (msghdr != NULL) {
 881                                 if (msgsz < msghdr->msg_ts &&
 882                                     (msgflg & MSG_NOERROR) == 0) {
 883 #ifdef MSG_DEBUG_OK
 884                                         printf("first message on the queue is too big (want %d, got %d)\n",
 885                                             msgsz, msghdr->msg_ts);
 886 #endif
 887                                         return(E2BIG);
 888                                 }
 889                                 if (msqptr->msg_first == msqptr->msg_last) {
 890                                         msqptr->msg_first = NULL;
 891                                         msqptr->msg_last = NULL;
 892                                 } else {
 893                                         msqptr->msg_first = msghdr->msg_next;
 894                                         if (msqptr->msg_first == NULL)
 895                                                 panic("msg_first/last screwed up #1");
 896                                 }
 897                         }
 898                 } else {
 899                         struct msg *previous;
 900                         struct msg **prev;
 901
 902                         previous = NULL;
 903                         prev = &(msqptr->msg_first);
 904                         while ((msghdr = *prev) != NULL) {
 905                                 /*
 906                                  * Is this message's type an exact match or is
 907                                  * this message's type less than or equal to
 908                                  * the absolute value of a negative msgtyp?
 909                                  * Note that the second half of this test can
 910                                  * NEVER be true if msgtyp is positive since
 911                                  * msg_type is always positive!
 912                                  */
 913
 914                                 if (msgtyp == msghdr->msg_type ||
 915                                     msghdr->msg_type <= -msgtyp) {
 916 #ifdef MSG_DEBUG_OK
 917                                         printf("found message type %d, requested %d\n",
 918                                             msghdr->msg_type, msgtyp);
 919 #endif
 920                                         if (msgsz < msghdr->msg_ts &&
 921                                             (msgflg & MSG_NOERROR) == 0) {
 922 #ifdef MSG_DEBUG_OK
 923                                                 printf("requested message on the queue is too big (want %d, got %d)\n",
 924                                                     msgsz, msghdr->msg_ts);
 925 #endif
 926                                                 return(E2BIG);
 927                                         }
 928                                         *prev = msghdr->msg_next;
 929                                         if (msghdr == msqptr->msg_last) {
 930                                                 if (previous == NULL) {
 931                                                         if (prev !=
 932                                                             &msqptr->msg_first)
 933                                                                 panic("msg_first/last screwed up #2");
 934                                                         msqptr->msg_first =
 935                                                             NULL;
 936                                                         msqptr->msg_last =
 937                                                             NULL;
 938                                                 } else {
 939                                                         if (prev ==
 940                                                             &msqptr->msg_first)
 941                                                                 panic("msg_first/last screwed up #3");
 942                                                         msqptr->msg_last =
 943                                                             previous;
 944                                                 }
 945                                         }
 946                                         break;
 947                                 }
 948                                 previous = msghdr;
 949                                 prev = &(msghdr->msg_next);
 950                         }
 951                 }
 952
 953                 /*
 954                  * We've either extracted the msghdr for the appropriate
 955                  * message or there isn't one.
 956                  * If there is one then bail out of this loop.
 957                  */
 958
 959                 if (msghdr != NULL)
 960                         break;
 961
 962                 /*
 963                  * Hmph!  No message found.  Does the user want to wait?
 964                  */
 965
 966                 if ((msgflg & IPC_NOWAIT) != 0) {
 967 #ifdef MSG_DEBUG_OK
 968                         printf("no appropriate message found (msgtyp=%d)\n",
 969                             msgtyp);
 970 #endif
 971                         /* The SVID says to return ENOMSG. */
 972 #ifdef ENOMSG
 973                         return(ENOMSG);
 974 #else
 975                         /* Unfortunately, BSD doesn't define that code yet! */
 976                         return(EAGAIN);
 977 #endif
 978                 }
 979
 980                 /*
 981                  * Wait for something to happen
 982                  */
 983
 984 #ifdef MSG_DEBUG_OK
 985                 printf("msgrcv:  goodnight\n");
 986 #endif
 987                 eval = tsleep((caddr_t)msqptr, (PZERO - 4) | PCATCH, "msgwait",
 988                     0);
 989 #ifdef MSG_DEBUG_OK
 990                 printf("msgrcv:  good morning (eval=%d)\n", eval);
 991 #endif
 992
 993                 if (eval != 0) {
 994 #ifdef MSG_DEBUG_OK
 995                         printf("msgsnd:  interrupted system call\n");
 996 #endif
 997                         return(EINTR);
 998                 }
 999
1000                 /*
1001                  * Make sure that the msq queue still exists
1002                  */
1003
1004                 if (msqptr->msg_qbytes == 0 ||
1005                     msqptr->msg_perm.seq != IPCID_TO_SEQ(uap->msqid)) {
1006 #ifdef MSG_DEBUG_OK
1007                         printf("msqid deleted\n");
1008 #endif
1009                         return(EIDRM);
1010                 }
1011         }
1012
1013         /*
1014          * Return the message to the user.
1015          *
1016          * First, do the bookkeeping (before we risk being interrupted).
1017          */
1018
1019         msqptr->msg_cbytes -= msghdr->msg_ts;
1020         msqptr->msg_qnum--;
1021         msqptr->msg_lrpid = p->p_pid;
1022         msqptr->msg_rtime = time_second;
1023
1024         /*
1025          * Make msgsz the actual amount that we'll be returning.
1026          * Note that this effectively truncates the message if it is too long
1027          * (since msgsz is never increased).
1028          */
1029
1030 #ifdef MSG_DEBUG_OK
1031         printf("found a message, msgsz=%d, msg_ts=%d\n", msgsz,
1032             msghdr->msg_ts);
1033 #endif
1034         if (msgsz > msghdr->msg_ts)
1035                 msgsz = msghdr->msg_ts;
1036
1037         /*
1038          * Return the type to the user.
1039          */
1040
1041         eval = copyout((caddr_t)&(msghdr->msg_type), user_msgp,
1042             sizeof(msghdr->msg_type));
1043         if (eval != 0) {
1044 #ifdef MSG_DEBUG_OK
1045                 printf("error (%d) copying out message type\n", eval);
1046 #endif
1047                 msg_freehdr(msghdr);
1048                 wakeup((caddr_t)msqptr);
1049                 return(eval);
1050         }
1051         user_msgp = (char *)user_msgp + sizeof(msghdr->msg_type);
1052
1053         /*
1054          * Return the segments to the user
1055          */
1056
1057         next = msghdr->msg_spot;
1058         for (len = 0; len < msgsz; len += msginfo.msgssz) {
1059                 size_t tlen;
1060
1061                 if (msgsz - len > msginfo.msgssz)
1062                         tlen = msginfo.msgssz;
1063                 else
1064                         tlen = msgsz - len;
1065                 if (next <= -1)
1066                         panic("next too low #3");
1067                 if (next >= msginfo.msgseg)
1068                         panic("next out of range #3");
1069                 eval = copyout((caddr_t)&msgpool[next * msginfo.msgssz],
1070                     user_msgp, tlen);
1071                 if (eval != 0) {
1072 #ifdef MSG_DEBUG_OK
1073                         printf("error (%d) copying out message segment\n",
1074                             eval);
1075 #endif
1076                         msg_freehdr(msghdr);
1077                         wakeup((caddr_t)msqptr);
1078                         return(eval);
1079                 }
1080                 user_msgp = (char *)user_msgp + tlen;
1081                 next = msgmaps[next].next;
1082         }
1083
1084         /*
1085          * Done, return the actual number of bytes copied out.
1086          */
1087
1088         msg_freehdr(msghdr);
1089         wakeup((caddr_t)msqptr);
1090         p->p_retval[0] = msgsz;
1091         return(0);
1092 }
1093
1094 static int
1095 sysctl_msqids(SYSCTL_HANDLER_ARGS)
1096 {
1097
1098         return (SYSCTL_OUT(req, msqids,
1099             sizeof(struct msqid_ds) * msginfo.msgmni));
1100 }
1101
1102 TUNABLE_INT("kern.ipc.msgseg", &msginfo.msgseg);
1103 TUNABLE_INT("kern.ipc.msgssz", &msginfo.msgssz);
1104 TUNABLE_INT("kern.ipc.msgmni", &msginfo.msgmni);
1105
1106 SYSCTL_DECL(_kern_ipc);
1107 SYSCTL_INT(_kern_ipc, OID_AUTO, msgmax, CTLFLAG_RD, &msginfo.msgmax, 0, "");
1108 SYSCTL_INT(_kern_ipc, OID_AUTO, msgmni, CTLFLAG_RD, &msginfo.msgmni, 0, "");
1109 SYSCTL_INT(_kern_ipc, OID_AUTO, msgmnb, CTLFLAG_RD, &msginfo.msgmnb, 0, "");
1110 SYSCTL_INT(_kern_ipc, OID_AUTO, msgtql, CTLFLAG_RD, &msginfo.msgtql, 0, "");
1111 SYSCTL_INT(_kern_ipc, OID_AUTO, msgssz, CTLFLAG_RD, &msginfo.msgssz, 0, "");
1112 SYSCTL_INT(_kern_ipc, OID_AUTO, msgseg, CTLFLAG_RD, &msginfo.msgseg, 0, "");
1113 SYSCTL_PROC(_kern_ipc, OID_AUTO, msqids, CTLFLAG_RD,
1114     NULL, 0, sysctl_msqids, "", "Message queue IDs");