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

В листинге 16.12 показана функция 
write_get_cmd
, посылающая серверу команду HTTP GET.
Листинг 16.12. Отправка команды HTTP GET серверу
//nonblock/write_get_cmd.c
 1 #include "web.h"
 2 void
 3 write_get_cmd(struct file *fptr)
 4 {
 5  int n;
 6  char line[MAXLINE];
 7  n = snprintf(line, sizeof(line), GET_CMD, fptr->f_name);
 8  Writen(fptr->f_fd, line, n);
 9  printf("wrote %d bytes for %s\n", n, fptr->f_name);
10  fptr->f_flags = F_READING; /* сброс F_CONNECTING */
11  FD_SET(fptr->f_fd, &rset); /* прочитаем ответ сервера */
12  if (fptr->f_fd > maxfd)
13   maxfd = fptr->f_fd;
14 }
Создание команды и ее отправка
7-9
 Команда создается и пишется в сокет.
Установка флагов
10-13
 Устанавливается флаг 
F_READING
, при этом также сбрасывается флаг 
F_CONNECTING
 (если он установлен). Это указывает основному циклу, что данный дескриптор готов для ввода. Также включается дескриптор в наборе чтения, и при необходимости обновляется значение 
maxfd
.
Теперь мы возвращаемся в функцию 
main
, показанную в листинге 16.13, начиная с того места, где закончили в листинге 16.9. Это основной цикл программы: пока имеется ненулевое количество файлов для обработки (значение 
nlefttoread
 больше нуля), устанавливается, если это возможно, другое соединение и затем вызывается функция 
select
 для всех активных дескрипторов, обрабатывающая как завершение неблокируемых соединений, так и прием данных.
Можем ли мы инициировать другое соединение?
24-35
 Если мы не дошли до заданного предела одновременных соединений и есть дополнительные соединения, которые нужно установить, мы ищем еще не обработанный файл (на него указывает нулевое значение 
f_flags
) и вызываем функцию 
start_connect
 для инициирования соединения. Число активных соединений увеличивается на единицу (
nconn
), а число соединений, которые нужно установить, на единицу уменьшается (
nlefttoconn
).
Функция select: ожидание событий
36-37
 Функция 
select
 ожидает готовности сокета либо для чтения, либо для записи. Дескрипторы, для которых в настоящий момент происходит установление соединения (неблокируемая функция 
connect
 находится в процессе выполнения), будут включены в обоих наборах, в то время как дескрипторы с завершенным соединением, ожидающие данных от сервера, будут включены только в наборе чтения.
Листинг 16.13. Основной цикл функции main
  
//nonblock/web.c
24  while (nlefttoread > 0) {
25   while (nconn < maxnconn && nlefttoconn > 0) {
26    /* find a file to read */
27    for (i =0; i < nfiles; i++)
28     if (file[i].f_flags == 0)
29      break;
30    if (i == nfiles)
31     err_quit("nlefttoconn = %d but nothing found", nlefttoconn);
32    start_connect(&file[i]);
33    nconn++;
34    nlefttoconn--;
35   }
36   rs = rset:
37   ws = wset;
38   n = Select(maxfd + 1, &rs, &ws, NULL, NULL);
39   for (i = 0; i < nfiles; i++) {
40    flags = file[i].f_flags;
41    if (flags == 0 || flags & F_DONE)
42     continue;
43    fd = file[i].f_fd;
44    if (flags & F_CONNECTING &&
45     (FD_ISSET(fd, &rs) || FD_ISSET(fd, &ws))) {
46     n = sizeof(error);
47     if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &error, &n) < 0 ||
48      error != 0) {
49      err_ret("nonblocking connect failed
50       for %s", file[i].f_name);
51     }
52     /* соединение установлено */
53     printf("connection established for %s\n", file[i].f_name);
54     FD_CLR(fd, &wset); /* отключаем запись в этот сокет */
55     write_get_cmd(&file[i]); /* передаем команду GET */
56    } else if (flags & F_READING && FD_ISSET(fd, &rs)) {
57     if ((n = Read(fd, buf, sizeof(buf))) == 0) {
58      printf("end-of-file on %s\n", file[i].f_name);
59      Close(fd);
60      file[i].f_flags = F_DONE; /* сбрасывает флаг F_READING */
61      FD_CLR(fd, &rset);
62      nconn--;
63      nlefttoread--;
64     } else {
65      printf("read %d bytes from %s\n", n, file[i].f_name);
66     }