23 Setsockopt(sock_fd, IPPROTO_SCTP, SCTP_EVENTS, &evnts, sizeof(evnts));
24 if (echo_to_all == 0)
25 sctpstr_cli(stdin, sock_fd, (SA*)&servaddr, sizeof(servaddr));
26 else
27 sctpstr_cli_echoall(stdin, sock_fd, (SA*)&servaddr,
28 sizeof(servaddr));
29 Close(sock_fd);
30 return(0);
31 }
Проверка аргументов и создание сокета
9-15
Клиент проверяет переданные ему при запуске аргументы командной строки. Сначала проверяется, указан ли в строке IP-адрес узла, на который нужно отправлять сообщения. Затем проверяется, указан ли параметр отправки эхо-сообщений всем (мы воспользуемся им в разделе 10.5). Наконец, клиент создает сокет SCTP типа «один-ко-многим».
Подготовка адреса сервера
16-20
Клиент преобразует IP-адрес сервера, переданный ему в командной строке, с помощью функции
inet_pton
. К адресу он добавляет заранее известный номер порта сервера. Полученная структура используется для всех обращений к данному серверу.
Подписка на уведомления
21-23
Клиент явно указывает, какие именно уведомления он хочет получать от созданного сокета SCTP. События
MSG_NOTIFICATION
ему не нужны, поэтому он отключает их, оставляя лишь структуру
sctp_sndrcvinfo
.
Вызов функции обработки сообщений
24-28
Если флаг
echo_to_all
не установлен, клиент вызывает функцию
sctpstr_cli
, которая будет обсуждаться в разделе 10.4. В противном случае вызывается
sctpstr_cli_echoall
(раздел 10.5, где рассматривается применение потоков SCTP).
Завершение работы
29-31
Закончив работу с сообщениями, клиент закрывает сокет SCTP, что приводит к закрытию всех ассоциаций, использующих этот сокет. Затем функция
main
завершается и возвращает код 0 — никаких ошибок не произошло.
10.4. Потоковый эхо-клиент SCTP: функция str_cli
В листинге 10.3 приведена основная функция эхо-клиента SCTP.
Листинг 10.3. Функция sctp_strcli
//sctp/sctp_strcli.c
1 #include "unp.h"
2 void
3 sctpstr_cli(FILE *fp, int sock_fd, struct sockaddr *to, socklen_t tolen)
4 {
5 struct sockaddr_in peeraddr;
6 struct sctp_sndrcvinfo sri;
7 char sendline[MAXLINE], recvline[MAXLINE];
8 socklen_t len;
9 int out_sz, rd_sz;
10 int msg_flags;
11 bzero(&sri, sizeof(sri));
12 while (fgets(sendline, MAXLINE, fp) != NULL) {
13 if (sendline[0] != '[') {
14 printf("Error, line must be of the form '[streamnum]text'\n");
15 continue;
16 }
17 sri.sinfo_stream = strtol(&sendline[1], NULL, 0);
18 out_sz = strlen(sendline);
19 Sctp_sendmsg(sock_fd, sendline, out_sz,
20 to, tolen, 0, 0, sri.sinfo_stream, 0, 0);
21 len = sizeof(peeraddr);
22 rd_sz = Sctp_recvmsg(sock_fd, recvline, sizeof(recvline),
23 (SA*)&peeraddr, &len, &sri, &msg_flags);
24 printf("From str:%d seq:%d (assoc:0x%x):",
25 sri.sinfo_stream.sri.sinfo_ssn, (u_int)sri.sinfo_assoc_id);
26 printf("%*s", rd_sz.recvline);
27 }
28 }
Инициализация структуры sri и вход в цикл
11-12
Основная функция клиента начинает работу с очистки структуры
sctp_sndrcvinfo
(переменная
sri
). Затем функция входит в цикл, считывающий из дескриптора
fp
, переданного вызывающей функцией, при помощи блокирующего вызова
fgets
. Главная программа (
main
) передает этой функции
stdin
в качестве аргумента
fp
, поэтому функция считывает и обрабатывает пользовательский ввод до тех пор, пока пользователь не введет завершающий EOF (Ctrl+D). При этом функция завершается и управление передается вызвавшей функции.
Проверка ввода
13-16
Клиент проверяет введенный пользователем текст на соответствие шаблону
[#]текст
. Если формат строки нарушен, клиент выводит сообщение об ошибке и снова вызывает
fgets
.
Преобразование номера потока
17
Клиент записывает запрошенный пользователем номер потока из текстовой строки в поле
sinfo_stream
структуры
sri
.
Отправка сообщения
18-20
После инициализации длины структуры адреса и размера пользовательских данных клиент отсылает сообщение серверу при помощи функции
sctp_sendmsg
.
Блокирование в ожидании ответа
21-23
Клиент блокируется и ожидает получения эхо-ответа сервера.
Отображение полученного эхо-ответа
24-26
Клиент выводит на экран полученное от сервера сообщение, вместе с номером потока и последовательным номером сообщения в этом потоке. После этого клиент возвращается на начало цикла, ожидая, что пользователь введет следующую строку.
Запуск программы
Мы запустили эхо-сервер SCTP без аргументов командной строки на компьютере, работающем под управлением FreeBSD. Клиенту при запуске необходимо указать IP-адрес сервера.