tools/regression/mqueue/mqtest5/mqtest5.c

   1 /* $FreeBSD$ */
   2 #include <stdio.h>
   3 #include <mqueue.h>
   4 #include <fcntl.h>
   5 #include <signal.h>
   6 #include <unistd.h>
   7 #include <sys/select.h>
   8 #include <sys/event.h>
   9 #include <signal.h>
  10
  11 #define MQNAME  "/mytstqueue5"
  12 #define LOOPS   1000
  13 #define PRIO    10
  14
  15 void sighandler(int sig)
  16 {
  17         write(1, "timeout\n", 8);
  18         _exit(1);
  19 }
  20
  21 int main()
  22 {
  23         mqd_t mq;
  24         int status;
  25         struct mq_attr attr;
  26         int pid;
  27         sigset_t set;
  28         struct sigaction sa;
  29         siginfo_t info;
  30
  31         mq_unlink(MQNAME);
  32
  33         sigemptyset(&set);
  34         sigaddset(&set, SIGRTMIN);
  35         sigprocmask(SIG_BLOCK, &set, NULL);
  36         sigemptyset(&sa.sa_mask);
  37         sa.sa_flags = SA_SIGINFO;
  38         sa.sa_sigaction = (void *) SIG_DFL;
  39         sigaction(SIGRTMIN, &sa, NULL);
  40
  41         attr.mq_maxmsg  = 5;
  42         attr.mq_msgsize = 128;
  43         mq = mq_open(MQNAME, O_CREAT | O_RDWR | O_EXCL, 0666, &attr);
  44         if (mq == (mqd_t)-1)
  45                 err(1, "mq_open()");
  46         status = mq_getattr(mq, &attr);
  47         if (status)
  48                 err(1, "mq_getattr()");
  49         pid = fork();
  50         if (pid == 0) { /* child */
  51                 int prio, j, i;
  52                 char *buf;
  53                 struct sigevent sigev;
  54
  55                 signal(SIGALRM, sighandler);
  56
  57                 sigev.sigev_notify = SIGEV_SIGNAL;
  58                 sigev.sigev_signo = SIGRTMIN;
  59                 sigev.sigev_value.sival_int = 2;
  60
  61                 mq_close(mq);
  62                 mq = mq_open(MQNAME, O_RDWR | O_NONBLOCK);
  63                 if (mq == (mqd_t)-1)
  64                         err(1, "child: mq_open");
  65                 buf = malloc(attr.mq_msgsize);
  66                 for (j = 0; j < LOOPS; ++j) {
  67                         alarm(3);
  68                         status = mq_notify(mq, &sigev);
  69                         if (status)
  70                                 err(1, "child: mq_notify");
  71                         status = sigwaitinfo(&set, &info);
  72                         if (status == -1)
  73                                 err(1, "child: sigwaitinfo");
  74                         if (info.si_value.sival_int != 2)
  75                                 err(1, "child: sival_int");
  76                         status = mq_receive(mq, buf, attr.mq_msgsize, &prio);
  77                         if (status == -1)
  78                                 err(2, "child: mq_receive");
  79                         for (i = 0; i < attr.mq_msgsize; ++i)
  80                                 if (buf[i] != i)
  81                                         err(3, "child: message data corrupted");
  82                         if (prio != PRIO)
  83                                 err(4, "child: priority is incorrect: %d",
  84                                          prio);
  85                 }
  86                 alarm(0);
  87                 free(buf);
  88                 mq_close(mq);
  89                 return (0);
  90         } else if (pid == -1) {
  91                 err(1, "fork()");
  92         } else {
  93                 char *buf;
  94                 int i, j, prio;
  95
  96                 signal(SIGALRM, sighandler);
  97                 buf = malloc(attr.mq_msgsize);
  98                 for (j = 0; j < LOOPS; ++j) {
  99                         for (i = 0; i < attr.mq_msgsize; ++i) {
 100                                 buf[i] = i;
 101                         }
 102                         alarm(3);
 103                         status = mq_send(mq, buf, attr.mq_msgsize, PRIO);
 104                         if (status) {
 105                                 kill(pid, SIGKILL);
 106                                 err(2, "mq_send()");
 107                         }
 108                 }
 109                 alarm(3);
 110                 wait(&status);
 111                 alarm(0);
 112         }
 113         status = mq_close(mq);
 114         if (status)
 115                 err(1, "mq_close");
 116         mq_unlink(MQNAME);
 117         return (0);
 118 }