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

  
26.6. Веб-клиент и одновременное соединение (продолжение)
 
Вернемся к нашему примеру с веб-клиентом из раздела 16.5 и перепишем его с использованием потоков вместо неблокируемой функции 
connect
. Мы можем оставить сокеты в их заданном по умолчанию виде — блокируемыми, и создать один поток на каждое соединение. Каждый поток может блокироваться в вызове функции 
connect
, так как ядро будет просто выполнять какой-либо другой поток, готовый к работе.
В листинге 26.7 показана первая часть нашей программы, глобальные переменные и начало функции 
main
.
Листинг 26.7. Глобальные переменные и начало функции main
//threads/web01.c
 1 #include "unpthread.h"
 2 #include <thread.h> /* потоки Solaris */
 3 #define MAXFILES 20
 4 #define SERV    "80" /* номер порта или имя службы */
 5 struct file {
 6  char      *f_name; /* имя файла */
 7  char      *f_host; /* имя узла или IP-адрес */
 8  int       f_fd;    /* дескриптор */
 9  int       f_flags; /* F_xxx ниже */
10  pthread_t f_tid;   /* идентификатор потока */
11 } file[MAXFILES];
12 #define F_CONNECTING 1 /* функция connect () в процессе
                             выполнения */
13 #define F_READING 2    /* функция connect() завершена;
                             выполняется считывание */
14 #define F_DONE 4       /* все сделано */
15 #define GET_CMD "GET %s HTTP/1.0\r\n\r\n"
16 int nconn, nfiles, nlefttoconn, nlefttoread;
17 void *do_get_read(void*);
18 void home_page(const char*, const char*);
19 void write_get_cmd(struct file*);
20 int
21 main(int argc, char **argv)
22 {
23  int i, n, maxnconn;
24  pthread_t tid;
25  struct file *fptr;
26  if (argc < 5)
27   err_quit("usage: web <#conns> <IPaddr> <homepage> file1 ...");
28  maxnconn = atoi(argv[1]);
29  nfiles = min(argc - 4, MAXFILES);
30  for (i = 0; i < nfiles; i++) {
31   file[i].f_name = argv[i + 4];
32   file[i].f_host = argv[2];
33   file[i].f_flags = 0;
  
34  }
35  printf("nfiles = %d\n", nfiles);
36  home_page(argv[2], argv[3]);
37  nlefttoread = nlefttoconn = nfiles;
38  nconn = 0;
Глобальные переменные
1-16
 Мы подключаем заголовочный файл 
<thread.h>
 вдобавок к обычному 
<pthread.h>
, так как нам требуется использовать потоки Solaris в дополнение к потокам Pthreads, как мы вскоре покажем.
10
 Мы добавили к структуре 
file
 один элемент — идентификатор потока f
_tid
. Остальная часть этого кода аналогична коду в листинге 16.9. В этой версии нам не нужно использовать функцию 
select
, а следовательно, не нужны наборы дескрипторов и переменная 
maxfd
.
36
 Функция 
home_page
 не изменилась относительно листинга 16.10. В листинге 26.8 показан основной рабочий цикл потока main.
Листинг 26.8. Основной рабочий цикл потока main
//threads/web01.c
39  while (nlefttoread > 0) {
40   while (nconn < maxnconn && nlefttoconn > 0) {
41    /* находим файл для считывания */
42    for (i = 0; i < nfiles; i++)
43     if (file[i].f_flags == 0)
44      break;
45    if (i == nfiles)
46     err_quit("nlefttoconn = %d but nothing found", nlefttoconn);
47    file[i].f_flags = F_CONNECTING;
48    Pthread_create(&tid, NULL, &do_get_read, &file[i]);
49    file[i].f_tid = tid;
50    nconn++;
51    nlefttoconn--;
52   }
53   if ((n = thr_join(0, &tid, (void**)&fptr)) != 0)
54    errno = n, err_sys("thr_join error");
55   nconn--;
56   nlefttoread--;
57   printf("thread id %d for %s done\n", tid, fptr->f_name);
58  }
59  exit(0);
60 }
По возможности создаем другой поток
40-52
 Если имеется возможность создать другой поток (
nconn
 меньше, чем 
maxconn
), мы так и делаем. Функция, которую выполняет каждый новый поток, — это 
do_get_read
, а ее аргументом является указатель на структуру 
file
.