Шрифт:
Листинг 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" /* номер порта или имя службы */