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

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

Билл Феннер , Уильям Ричард Стивенс , Эндрю М. Рудофф

Итак, по мере того как приходят клиенты, мы записываем дескриптор их присоединенного сокета в первый свободный элемент массива client(то есть в первый элемент со значением -1). Следует также добавить присоединенный сокет в набор дескрипторов для чтения. Переменная maxi— это наибольший используемый в данный момент индекс в массиве client, а переменная maxfd(плюс один) — это текущее значение первого аргумента функции select. Единственным ограничением на количество обслуживаемых сервером клиентов является минимальное из двух значений: FD_SETSIZEи максимального числа дескрипторов, которое допускается для данного процесса ядром (о чем мы говорили в конце раздела 6.3).

В листинге 6.3 показана первая половина этой версии сервера.

Листинг 6.3. Сервер TCP, использующий одиночный процесс и функцию select: инициализация

//tcpcliserv/tcpservselect01.c

1 #include "unp.h"

2 int

3 main(int argc, char **argv)

4 {

5 int i, maxi, maxfd, listenfd, connfd, sockfd;

6 int nready, client[FD_SETSIZE],

7 ssize_t n;

8 fd_set rset, allset;

9 char buf[MAXLINE];

10 socklen_t clilen;

11 struct sockaddr_in cliaddr, servaddr;

12 listenfd = Socket(AF_INET, SOCK_STREAM, 0);

13 bzero(&servaddr, sizeof(servaddr));

14 servaddr.sin_family = AF_INET;

15 servaddr.sin_addr.s_addr = htonl(INADDR_ANY);

16 servaddr.sin_port = htons(SERV_PORT);

17 Bind(listenfd, (SA*)&servaddr, sizeof(servaddr));

18 Listen(listenfd, LISTENQ);

19 maxfd = listenfd; /* инициализация */

20 maxi = -1; /* индекс в массиве client[] */

21 for (i = 0; i < FD_SETSIZE; i++)

22 client[i] = -1; /* -1 означает свободный элемент */

23 FD_ZERO(&allset);

24 FD_SET(listenfd, &allset);

Создание прослушиваемого сокета и инициализация функции select

12-24 Этапы создания прослушиваемого сокета те же, что и раньше: вызов функций socket, bindи listen. Мы инициализируем структуры данных при том условии, что единственный дескриптор, который мы с помощью функции selectвыберем, изначально является прослушиваемым сокетом.

Вторая половина функции mainпоказана в листинге 6.4.

Листинг 6.4. Сервер TCP, использующей одиночный процесс и функцию select: цикл

//tcpcliserv/tcpservselect01.c

25 for (;;) {

26 rset = allset; /* присваивание значения структуре */

27 nready = Select(maxfd + 1, &rset, NULL, NULL, NULL);

28 if (FD_ISSET(listenfd, &rset)) { /* соединение с новым клиентом */

29 clilen = sizeof(cliaddr);

30 connfd = Accept(listenfd, (SA*)&cliaddr, &clilen);

31 for (i = 0; i < FD_SETSIZE; i++)

32 if (client[i] < 0) {

33 client[i] = connfd; /* сохраняем дескриптор */