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

Листинг 26.6. Вторая часть функции readline, безопасной в многопоточной среде

 //threads/readline.c
 19 static ssize_t
 20 my_read(Rline *tsd, int fd, char *ptr)
 21 {
 22 if (tsd->rl_cnt <= 0) {
 23 again:
 24 if ((tsd->rl_cnt = read(fd, tsd->rl_buf, MAXLINE)) < 0) {
 25 if (errno == EINTR)
 26 goto again;
 27 return (-1);
 28 } else if (tsd->rl_cnt == 0)
 29 return (0);
 30 tsd->rl_bufptr = tsd->rl_buf;
 31 }
 32 tsd->rl_cnt--;
 33 *ptr = *tsd->rl_bufptr++;
 34 return (1);
 35 }

 36 ssize_t
 37 readline(int fd, void *vptr, size_t maxlen)
 38 {
 39 int n, rc;
 40 char c, *ptr;
 41 Rline *tsd;

 42 Pthread_once(&rl_once, readline_once);
 43 if ((tsd = pthread_getspecific(rl_key)) == NULL) {
44 tsd = Calloc(1, sizeof(Rline)); /* инициализируется нулем */
 45 Pthread_setspecifiс(rl_key, tsd);
 46 }

 47 ptr = vptr;
 48 for (n = 1; n < maxlen; n++) {
 49 if ((rc = my_read(tsd, fd, &c)) == 1) {
 50 *ptr++ = c;
 51 if (c == '\n')
 52 break;
 53 } else if (rc == 0) {
 54 *ptr = 0;
 55 return (n-1); /* EOF, данные не были считаны */
 56 } else
 57 return (-1); /* ошибка, errno устанавливается функцией read */
 58 }

 59 *ptr = 0;
 60 return (n);
 61 }

Функция my_read

19-35 Первым аргументом функции теперь является указатель на структуру Rline, которая была размещена в памяти для данного потока (и содержит собственные данные этого потока).

Размещение собственных данных потока в памяти

42 Сначала мы вызываем функцию pthread_once, так чтобы первый поток, вызывающий функцию readline в этом процессе, вызвал бы функцию readline_once для создания ключа собственных данных потока.

Получение указателя на собственные данные потока

43-46 Функция pthread_getspecific возвращает указатель на структуру Rline для данного потока. Но если это первый вызов функции readline данным потоком, то возвращаемым значением будет пустой указатель. В таком случае мы выделяем в памяти место для структуры Rline, а элемент rl_cnt этой структуры инициализируется нулем с помощью функции calloc. Затем мы записываем этот указатель для данного потока, вызывая функцию pthread_setspecific. Когда этот поток вызовет функцию readline в следующий раз, функция pthread_getspecific возвратит этот указатель, который был только что записан.

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)