UNIX: разработка сетевых приложений
Шрифт:
13 echo_to_all = 1;
14 }
15 sock_fd = Socket(AF_INET, SOCK_SEQPACKET, IPPROTO_SCTP);
16 bzero(&servaddr, sizeof(servaddr));
17 servaddr.sin_family = AF_INET;
18 servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
19 servaddr.sin_port = htons(SERV_PORT);
20 Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
21 bzero(&evnts, sizeof(evnts));
22 evnts.sctp_data_io_event = 1;
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
. Отправка сообщения
Поделиться с друзьями: