tools/regression/mqueue/mqtest2/mqtest2.c

   1 /* $FreeBSD$ */
   2
   3 #include <sys/types.h>
   4 #include <sys/wait.h>
   5 #include <err.h>
   6 #include <fcntl.h>
   7 #include <mqueue.h>
   8 #include <signal.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <unistd.h>
  12
  13 #define MQNAME  "/mytstqueue2"
  14 #define LOOPS   1000
  15 #define PRIO    10
  16
  17 void alarmhandler(int sig)
  18 {
  19         write(1, "timeout\n", 8);
  20         _exit(1);
  21 }
  22
  23 int main()
  24 {
  25         struct mq_attr attr;
  26         mqd_t mq;
  27         int status, pid;
  28
  29         mq_unlink(MQNAME);
  30
  31         attr.mq_maxmsg  = 5;
  32         attr.mq_msgsize = 128;
  33         mq = mq_open(MQNAME, O_CREAT | O_RDWR | O_EXCL, 0666, &attr);
  34         if (mq == (mqd_t)-1)
  35                 err(1, "mq_open");
  36         status = mq_getattr(mq, &attr);
  37         if (status)
  38                 err(1, "mq_getattr");
  39         pid = fork();
  40         if (pid == 0) { /* child */
  41                 int prio, j, i;
  42                 char *buf;
  43
  44                 mq_close(mq);
  45
  46                 signal(SIGALRM, alarmhandler);
  47
  48                 mq = mq_open(MQNAME, O_RDWR);
  49                 if (mq == (mqd_t)-1)
  50                         err(1, "child: mq_open");
  51                 buf = malloc(attr.mq_msgsize);
  52                 for (j = 0; j < LOOPS; ++j) {
  53                         alarm(3);
  54                         status = mq_receive(mq, buf, attr.mq_msgsize, &prio);
  55                         if (status == -1)
  56                                 err(2, "child: mq_receive");
  57                         for (i = 0; i < attr.mq_msgsize; ++i)
  58                                 if (buf[i] != i)
  59                                         err(3, "child: message data corrupted");
  60                         if (prio != PRIO)
  61                                 err(4, "child: priority is incorrect: %d",
  62                                         prio);
  63                 }
  64                 alarm(0);
  65                 free(buf);
  66                 mq_close(mq);
  67                 return (0);
  68         } else if (pid == -1) {
  69                 err(1, "fork()");
  70         } else {
  71                 char *buf;
  72                 int i, j, prio;
  73
  74                 signal(SIGALRM, alarmhandler);
  75                 buf = malloc(attr.mq_msgsize);
  76                 for (j = 0; j < LOOPS; ++j) {
  77                         for (i = 0; i < attr.mq_msgsize; ++i)
  78                                 buf[i] = i;
  79                         alarm(3);
  80                         status = mq_send(mq, buf, attr.mq_msgsize, PRIO);
  81                         if (status)
  82                                 err(1, "mq_send");
  83                 }
  84                 alarm(3);
  85                 wait(&status);
  86                 alarm(0);
  87         }
  88         status = mq_close(mq);
  89         if (status)
  90                 err(1, "mq_close");
  91         mq_unlink(MQNAME);
  92         return (0);
  93 }