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

19-23
 Инициализируются два набора дескрипторов, по одному для чтения и для записи. 
maxfd
 — это максимальный дескриптор для функции 
select
 (который мы инициализируем значением -1, поскольку дескрипторы неотрицательны), 
nlefttoread
 — число файлов, которые осталось прочитать (когда это значение становится нулевым, чтение заканчивается), 
nlefttoconn
 — это количество файлов, для которых пока еще требуется соединение TCP, a 
nconn
 — это число соединений, открытых в настоящий момент (оно никогда не может превышать первый аргумент командной строки).
В листинге 16.10 показана функция 
home_page
, вызываемая один раз, когда начинается выполнение функции main.
Листинг 16.10. Функция home_page
//nonblock/home_page.c
 1 #include "web.h"
 2 void
 3 home_page(const char *host, const char *fname)
 4 {
 5  int fd, n;
 6  char line[MAXLINE];
 7  fd = Tcp_connect(host, SERV); /* блокируемая функция connect() */
 8  n = snprintf(line, sizeof(line), GET_CMD, fname);
 9  Writen(fd, line, n);
10  for (;;) {
11   if ((n = Read(fd, line, MAXLINE)) == 0)
12    break; /* сервер закрыл соединение */
13   printf("read %d bytes of home page\n", n);
14   /* обрабатываем полученные данные */
15  }
16  printf("end-of-file on home page\n");
17  Close(fd);
18 }
Установление соединения с сервером
7
 Наша функция 
tcp_connect
 устанавливает соединение с сервером.
Отправка команды HTTP серверу, чтение ответа
8-17
 Запускается команда HTTP 
GET
 для домашней страницы (часто обозначается символом 
/
). Читается ответ (с ответом мы в данном случае ничего не делаем), и соединение закрывается.
Следующая функция, 
start_connect
, показанная в листинге 16.11, инициирует вызов неблокируемой функции connect.
Листинг 16.11. Инициирование неблокируемой функции connect
//nonblock/start_connect.c
 1 #include "web.h"
 2 void
 3 start_connect(struct file *fptr)
 4 {
 5  int fd, flags, n;
 6  struct addrinfo *ai;
 7  ai = Host_serv(fptr->f_host, SERV, 0, SOCK_STREAM);
  
 8  fd = Socket(ai->ai_family; ai->ai_socktype, ai->ai_protocol);
 9  fptr->f_fd = fd;
10  printf("start_connect for %s, fd %d\n", fptr->f_name, fd);
11  /* отключаем блокирование сокета */
12  flags = Fcntl(fd, F_GETFL, 0);
13  Fcntl(fd, F_SETFL, flags | O_NONBLOCK);
14  /* инициируем неблокируемое соединение с сервером */
15  if ((n = connected, ai->ai_addr, ai->ai_addrlen)) < 0) {
16   if (errno != EINPROGRESS)
17    err_sys("nonblocking connect error");
18   fptr->f_flags = F_CONNECTING;
19   FD_SET(fd, &rset); /* включаем дескриптор сокета в наборе чтения
                           и записи */
20   FD_SET(fd, &wset);
21   if (fd > maxfd)
22    maxfd = fd;
23  } else if (n >= 0) /* соединение уже установлено */
24   write_get_cmd(fptr); /* отправляем команду GET серверу */
25 }
Создание сокета, отключение блокировки сокета
7-13
 Мы вызываем нашу функцию 
host_serv
 для поиска и преобразования имени узла и имени службы. Она возвращает указатель на массив структур 
addrinfo
. Мы используем только первую структуру. Создается сокет TCP, и он становится неблокируемым.
Вызов неблокируемой функции connect
14-22
 Вызывается неблокируемая функция 
connect
, и флагу файла присваивается значение 
F_CONNECTING
. Включается дескриптор сокета и в наборе чтения, и в наборе записи, поскольку функция 
select
 будет ожидать любого из этих условий как указания на то, что установление соединения завершилось. При необходимости мы также обновляем значение 
maxfd
.
Обработка завершения установления соединения
23-24
 Если функция 
connect
 успешно завершается, значит, соединение уже установлено, и функция 
write_get_cmd
 (она показана в следующем листинге) посылает команду серверу.
Мы делаем сокет неблокируемым для функции 
connect
, но никогда не переустанавливаем его в блокируемый режим, заданный по умолчанию. Это нормально, поскольку мы записываем в сокет только небольшое количество данных (команда GET следующей функции) и считаем, что эти данные занимают значительно меньше места, чем имеется в буфере отправки сокета. Даже если из-за установленного флага отсутствия блокировки при вызове функции 
write
 происходит частичное копирование, наша функция 
writen
 обрабатывает эту ситуацию. Если оставить сокет неблокируемым, это не повлияет на последующее выполнение функций 
read
, потому что мы всегда вызываем функцию 
select
 для определения того момента, когда сокет станет готов для чтения.