UNIX: взаимодействие процессов
вернуться

Стивенс Уильям Ричард
Шрифт:
Проверка заполненности очереди

39-48 Если очередь переполнена и установлен флаг O_NONBLOCK, мы возвращаем ошибку с кодом EAGAIN. В противном случае мы ожидаем сигнала по условной переменной mqh_wait, который, как мы увидим, отправляется функцией mq_receive при считывании сообщения из переполненной очереди.

ПРИМЕЧАНИЕ

Наша реализация упрощает ситуацию с возвращением ошибки EINTR при прерывании вызова mq_send сигналом, перехватываемым вызвавшим процессом. Проблема в том, что функция pthread_cond_wait не возвращает ошибки при возврате из обработчика сигнала: она может вернуть либо 0 (что рассматривается как ложное пробуждение), либо вообще не завершить работу. Все эти проблемы можно обойти, но это непросто. 

В листинге 5.26 приведена вторая половина функции mq_send. К моменту ее выполнения мы уже знаем о наличии в очереди свободного места для нашего сообщения.

Листинг 5.25. Функция mq_send: первая половина

//my_pxmsg_mmap/mq_send.с

1 #include "unpipc.h"

2 #include "mqueue.h"

3 int

4 mymq_send(mymqd_t mqd, const char *ptr, size_t len, unsigned int prio)

5 {

6 int n;

7 long index, freeindex;

8 int8_t *mptr;

9 struct sigevent *sigev;

10 struct mymq_hdr *mqhdr;

11 struct mymq_attr *attr;

12 struct mymsg_hdr *msghdr, *nmsghdr, *pmsghdr;

13 struct mymq_info *mqinfo;

14 mqinfo = mqd;

15 if (mqinfo->mqi_magic != MQI_MAGIC) {

16 errno = EBADF;

17 return(-1);

18 }

19 mqhdr = mqinfo->mqi_hdr; /* указатель типа struct */

20 mptr = (int8_t *) mqhdr; /* указатель на байт */

21 attr = &mqhdr->mqh_attr;

22 if ((n = pthread_mutex_lock(&mqhdr->mqh_lock)) != 0) {

23 errno = n;

24 return(-1);

25 }

26 if (len > attr->mq_msgsize) {

27 errno = EMSGSIZE;

28 goto err;

29 }

30 if (attr->mq_curmsgs == 0) {

31 if (mqhdr->mqh_pid != 0 && mqhdr->mqh_nwait == 0) {

32 sigev = &mqhdr->mqh_event;

33 if (sigev->sigev_notify == SIGEV_SIGNAL) {

34 sigqueue(mqhdr->mqh_pid, sigev->sigev_signo,

35 sigev->sigev_value);

36 }

37 mqhdr->mqh_pid = 0; /* снятие с регистрации */

38 }

39 } else if (attr->mq_curmsgs >= attr->mq_maxmsg) {

40 /* 4queue is full */

41 if (mqinfo->mqi_flags & O_NONBLOCK) {

32 errno = EAGAIN;

43 goto err;

44 }

45 /* ожидание освобождения места в очереди */

46 while (attr->mq_curmsgs >= attr->mq_maxmsg)

47 pthread_cond_wait(&mqhdr->mqh_wait, &mqhdr->mqh_lock);

48 }

Листинг 5.25. Функция mq_send: вторая половина

//my_pxmsg_mmap/mq_send.с

49 /* nmsghdr будет указывать на новое сообщение*/

50 if ((freeindex = mqhdr->mqh_free) == 0)

51 err_dump("mymq_send: curmsgs = %ld; free = 0", attr->mq_curmsgs);

52 nmsghdr = (struct mymsg_hdr *) &mptr[freeindex];

53 nmsghdr->msg_prio = prio;

54 nmsghdr->msg_len = len;

55 memcpy(nmsghdr + 1, ptr, len); /* копирование сообщения в очередь */

56 mqhdr->mqh_free = nmsghdr->msg_next; /* новое начало списка пустых сообщений */

57 /* поиск места в списке для нового сообщения */

58 index = mqhdr->mqh_head;