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
для определения того момента, когда сокет станет готов для чтения.
