UNIX: разработка сетевых приложений

  
Тайм-аут для функции recvfrom (сигнал SIGALRM)
 
В листинге 14.2 показана новая версия функции 
dg_cli
, приведенной в листинге 8.4, в которую добавлен вызов функции 
alarm
 для прерывания функции recvfrom при отсутствии ответа в течение 5 с.
Листинг 14.2. Функция dg_cli, в которой при установке тайм-аута для функции recvfrom используется функция alarm
//advio/dgclitimeo3.c
 1 #include "unp.h"
 2 static void signalrm(int);
 3 void
 4 dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
 5 {
 6  int n;
 7  char sendline[MAXLINE], recvline[MAXLINE + 1];
 8  Signal(SIGALRM, signalrm);
 9  while (Fgets(sendline, MAXLINE, fp) != NULL) {
10   Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
11   alarm(5);
12   if ((n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL)) < 0) {
13    if (errno == EINTR)
14     fprintf(stderr, "socket timeout\n");
15    else
16     err_sys("recvfrom error");
17   } else {
18    alarm(0);
19    recvline[n] = 0; /* завершающий нуль */
20    Fputs(recvline, stdout);
21   }
22  }
23 }
24 static void
25 sig_alrm(int signo)
26 {
27  return; /* просто прерываем recvfrom() */
28 }
Обработка тайм-аута из функции recvfrom
8-22
 Мы устанавливаем обработчик для сигнала 
SIGALRM
 и затем вызываем функцию 
alarm
 для 5-секундного тайм-аута при каждом вызове функции 
recvfrom
. Если функция 
recvfrom
 прерывается нашим обработчиком сигнала, мы выводим сообщение об ошибке и продолжаем работу. Если получена строка от сервера, мы отключаем функцию alarm и выводим ответ.
Обработчик сигнала SIGALRM
24-28
 Наш обработчик сигналов возвращает управление, прерывая блокированную функцию 
recvfrom
.
Этот пример работает корректно, потому что каждый раз, когда мы устанавливаем функцию 
alarm
, мы читаем только один ответ. В разделе 20.4 мы попытаемся использовать ту же технологию, но поскольку мы будем считывать множество ответов для данной функции 
alarm
, возникнет ситуация гонок, которую нам придется разрешить.
    
Тайм-аут для функции recvfrom (функция select)
 
Мы демонстрируем вторую технологию для установки тайм-аута (использование функции 
select
) в листинге 14.3. Здесь показана наша функция 
readable_timeo
, которая ждет, когда дескриптор станет готов для чтения, но не более заданного числа секунд.
Листинг 14.3. Функция readable_timeo: ожидание, когда дескриптор станет готов для чтения
//lib/readable_timео.c
 1 #include "unp.h"
 2 int
 3 readable_timeo(int fd, int sec)
 4 {
 5  fd_set rset;
 6  struct timeval tv;
 7  FD_ZERO(&rset);
 8  FD_SET(fd, &rset);
 9  tv.tv_sec = sec;
10  tv.tv_usec = 0;
11  return (select(fd + 1, &rset, NULL, NULL, &tv));
12  /* > если дескриптор готов для чтения */
13 }
Подготовка аргументов для функции select
7-10
 В наборе дескрипторов для чтения включается бит, соответствующий данному дескриптору. В структуре 
timeval
 устанавливается время (число секунд), в течение которого вызывающий процесс готов ждать.
Блокирование в функции select
11-12
 Функция 
select
 ждет, когда дескриптор станет готов для чтения или истечет заданное время ожидания. Возвращаемое значение этой функции — это возвращаемое значение функции 
select
: -1 в случае ошибки, 0, если истекло время ожидания, и положительное значение, задающее число готовых дескрипторов, если таковые появились.
Эта функция не выполняет операции чтения — она просто ждет, когда дескриптор будет готов к чтению. Следовательно, эту функцию можно использовать с любым типом сокета — TCP или UDP.
Создание аналогичной функции, называемой 
writable_timeo
, тривиально. Эта функция ждет, когда дескриптор будет готов для записи.
Мы используем эту функцию в листинге 14.4, где показана еще одна версия нашей функции 
dg_cli
, приведенной в листинге 8.4. Эта новая версия вызывает функцию 
recvfrom
, только когда наша функция 
readable_timeo
 возвращает положительное значение.
Мы не вызываем функцию 
recvfrom
, пока функция 
readable_timeo
 не сообщит нам, что дескриптор готов для чтения. Тем самым мы гарантируем, что функция 
recvfrom
 не заблокируется.
Листинг 14.4. Функция dg_cli, вызывающая функцию readable_timeo для установки тайм-аута
//advio/dgclitimeo1.c