Current Path : /usr/src/tools/regression/mqueue/mqtest3/ |
FreeBSD hs32.drive.ne.jp 9.1-RELEASE FreeBSD 9.1-RELEASE #1: Wed Jan 14 12:18:08 JST 2015 root@hs32.drive.ne.jp:/sys/amd64/compile/hs32 amd64 |
Current File : //usr/src/tools/regression/mqueue/mqtest3/mqtest3.c |
/* $FreeBSD: release/9.1.0/tools/regression/mqueue/mqtest3/mqtest3.c 205225 2010-03-16 20:42:54Z kib $ */ #include <sys/types.h> #include <sys/select.h> #include <sys/wait.h> #include <err.h> #include <fcntl.h> #include <mqueue.h> #include <signal.h> #include <stdio.h> #include <stdlib.h> #include <unistd.h> #define MQNAME "/mytstqueue3" #define LOOPS 1000 #define PRIO 10 void sighandler(int sig) { write(1, "timeout\n", 8); _exit(1); } int main() { mqd_t mq; int status; struct mq_attr attr; int pid; fd_set set; mq_unlink(MQNAME); attr.mq_maxmsg = 5; attr.mq_msgsize = 128; mq = mq_open(MQNAME, O_CREAT | O_RDWR | O_EXCL, 0666, &attr); if (mq == (mqd_t)-1) err(1, "mq_open()"); status = mq_getattr(mq, &attr); if (status) err(1, "mq_getattr()"); pid = fork(); if (pid == 0) { /* child */ int prio, j, i; char *buf; mq_close(mq); signal(SIGALRM, sighandler); mq = mq_open(MQNAME, O_RDWR); if (mq == (mqd_t)-1) err(1, "child process: mq_open"); buf = malloc(attr.mq_msgsize); for (j = 0; j < LOOPS; ++j) { FD_ZERO(&set); FD_SET(__mq_oshandle(mq), &set); alarm(3); status = select(__mq_oshandle(mq)+1, &set, NULL, NULL, NULL); if (status != 1) err(1, "child process: select()"); status = mq_receive(mq, buf, attr.mq_msgsize, &prio); if (status == -1) err(2, "child process: mq_receive"); for (i = 0; i < attr.mq_msgsize; ++i) if (buf[i] != i) err(3, "message data corrupted"); if (prio != PRIO) err(4, "priority is incorrect: %d", prio); } alarm(0); free(buf); mq_close(mq); return (0); } else if (pid == -1) { err(1, "fork()"); } else { char *buf; int i, j, prio; signal(SIGALRM, sighandler); buf = malloc(attr.mq_msgsize); for (j = 0; j < LOOPS; ++j) { for (i = 0; i < attr.mq_msgsize; ++i) { buf[i] = i; } alarm(3); FD_ZERO(&set); FD_SET(__mq_oshandle(mq), &set); status = select(__mq_oshandle(mq)+1, NULL, &set, NULL, NULL); if (status != 1) err(1, "select()"); status = mq_send(mq, buf, attr.mq_msgsize, PRIO); if (status) { kill(pid, SIGKILL); err(2, "mq_send()"); } } alarm(3); wait(&status); alarm(0); } status = mq_close(mq); if (status) err(1, "mq_close"); mq_unlink(MQNAME); return (0); }