UNIX: взаимодействие процессов, Стивенс Уильям Ричард

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

на обложку

Стивенс Уильям Ричард

Шрифт:

//my_pxmsg_mniap/mq_setattr.с

1 #include "unpipc.h"

2 #include "mqueue.h"

3 int

4 mymq_setattr(mymqd_t mqd. const struct mymq_attr *mqstat,

5 struct mymq attr *omqstat)

6 {

7 int n;

8 struct mymq_hdr *mqhdr;

9 struct mymq_attr *attr;

10 struct mymq_info *mqinfo;

11 mqinfo = mqd;

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

13 errno = EBADF;

14 return(-1);

15 }

16 mqhdr = mqinfo->mqi_hdr;

17 attr = &mqhdr->mqh_attr;

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

19 errno = n;

20 return(-1);

21 }

22 if (omqstat != NULL) {

23 omqstat->mq_flags = mqinfo->mqi_flags; /* исходные атрибуты */

24 omqstat->mq_maxmsg = attr->mq_maxmsg;

25 omqstat->mq_msgsize = attr->mq_msgsize;

26 omqstat->mq_curmsgs = attr->mq_curmsgs; /* текущий статус */

27 }

28 if (mqstat->mq_flags & O_NONBLOCK)

29 mqinfo->mqi flags |= O_NONBLOCK;

30 else

31 mqinfo->ntqi_flags &= ~O_NONBLOCK;

32 pthread_mutex_unlock(&mqhdr->mqh_lock);

33 return(0);

34 }

Функция mq_notify

Функция mq_notify, текст которой приведен в листинге 5.24, позволяет регистрировать процесс на уведомление для текущей очереди и снимать его с регистрации. Информация о зарегистрированных процессах (их идентификаторы) хранится в поле mqh_pid структуры mq_hdr. Только один процесс может быть зарегистрирован на уведомление в любой момент времени. При регистрации процесса мы сохраняем его структуру sigevent в структуре mqh_event.

Листинг 5.24. Функция mq_notify

//my_pxmsg_mmap/mq_notify.с

1 #include "unpipc.h"

2 #include "mqueue.h"

3 int

4 mymq_notify(mymqd_t mqd, const struct sigevent *notification)

5 {

6 int n;

7 pid_t pid;

8 struct mymq_hdr *mqhdr;

9 struct mymq_info *mqinfo;

10 mqinfo = mqd;

11 if (mqinfo->mqi magic != MQI_MAGIC) {

12 errno = EBADF;

13 return(-1);

14 }

15 mqhdr = mqinfo->mqi_hdr;

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

17 errno = n;

18 return(-1);

19 }

20 pid = getpid;

21 if (notification == NULL) {

22 if (mqhdr->mqh_pid == pid) {

23 mqhdr->mqh_pid = 0; /* снятие вызвавшего процесса с регистрации */

24 } /* если вызвавший процесс не зарегистрирован – 61К */

25 } else {

26 if (mqhdr->mqh_pid != 0) {

27 if (kill(mqhdr->mqh_pid, 0) != –1 || errno != ESRCH) {

28 errno = EBUSY;

29 goto err;

30 }

31 }

32 mqhdr->mqh_pid = pid;

33 mqhdr->mqh_event = *notification;

34 }

35 pthread_mutex_unlock(&mqhdr->mqh_lock);

36 return(0);

37 err:

38 pthread_mutex_unlock(&mqhdr->mqh_lock);

39 return(-1);

40 }

Снятие процесса с регистрации

20-24 Если второй аргумент представляет собой нулевой указатель, вызвавший процесс снимается с регистрации. Если он не зарегистрирован, никакой ошибки не возвращается.

Регистрация вызвавшего процесса

25-34 Если какой-либо процесс уже зарегистрирован, мы проверяем, существует ли он, отправкой ему сигнала с кодом 0 (называемого нулевым сигналом — null signal). Это обычная проверка на возможность ошибки, на самом деле при этом никакого сигнала процессу не отправляется, но при его отсутствии возвращается ошибка с кодом ESRCH. Если какой-либо процесс уже зарегистрирован на уведомление, функция возвращает ошибку EBUSY. В противном случае сохраняется идентификатор процесса вместе с его структурой sigevent.

ПРИМЕЧАНИЕ

Наш метод проверки существования вызвавшего процесса не идеален. Процесс мог завершить работу, а его идентификатор мог быть использован другим процессом.

Функция mq_send

В листинге 5.25 приведен текст первой половины нашей функции mqsend.

Инициализация

14-29 Мы получаем указатели на используемые структуры и блокируем взаимное исключение для данной очереди. Проверяем, не превышает ли размер сообщения максимально допустимый для данной очереди.