UNIX: разработка сетевых приложений
Часть 116 из 399 Информация о книге
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 else27 sctpstr_cli_echoall(stdin, sock_fd, (SA*)&servaddr,28 sizeof(servaddr));29 Close(sock_fd);30 return(0);31 }Проверка аргументов и создание сокетаКлиент проверяет переданные ему при запуске аргументы командной строки. Сначала проверяется, указан ли в строке IP-адрес узла, на который нужно отправлять сообщения. Затем проверяется, указан ли параметр отправки эхо-сообщений всем (мы воспользуемся им в разделе 10.5). Наконец, клиент создает сокет SCTP типа «один-ко-многим».9-15Подготовка адреса сервераКлиент преобразует IP-адрес сервера, переданный ему в командной строке, с помощью функции16-20. К адресу он добавляет заранее известный номер порта сервера. Полученная структура используется для всех обращений к данному серверу.inet_ptonПодписка на уведомленияКлиент явно указывает, какие именно уведомления он хочет получать от созданного сокета SCTP. События21-23ему не нужны, поэтому он отключает их, оставляя лишь структуруMSG_NOTIFICATION.sctp_sndrcvinfoВызов функции обработки сообщенийЕсли флаг24-28не установлен, клиент вызывает функциюecho_to_all, которая будет обсуждаться в разделе 10.4. В противном случае вызываетсяsctpstr_cli(раздел 10.5, где рассматривается применение потоков SCTP).sctpstr_cli_echoallЗавершение работыЗакончив работу с сообщениями, клиент закрывает сокет SCTP, что приводит к закрытию всех ассоциаций, использующих этот сокет. Затем функция29-31завершается и возвращает код 0 — никаких ошибок не произошло.main10.4. Потоковый эхо-клиент SCTP: функция str_cli
В листинге 10.3 приведена основная функция эхо-клиента SCTP.
Листинг 10.3. Функция sctp_strcli
//sctp/sctp_strcli.c1 #include "unp.h"2 void3 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, поэтому функция считывает и обрабатывает пользовательский ввод до тех пор, пока пользователь не введет завершающий EOF (Ctrl+D). При этом функция завершается и управление передается вызвавшей функции.fpПроверка вводаКлиент проверяет введенный пользователем текст на соответствие шаблону13-16. Если формат строки нарушен, клиент выводит сообщение об ошибке и снова вызывает[#]текст.fgetsПреобразование номера потокаКлиент записывает запрошенный пользователем номер потока из текстовой строки в поле17структурыsinfo_stream.sriОтправка сообщенияПосле инициализации длины структуры адреса и размера пользовательских данных клиент отсылает сообщение серверу при помощи функции18-20.sctp_sendmsgБлокирование в ожидании ответаКлиент блокируется и ожидает получения эхо-ответа сервера.21-23Отображение полученного эхо-ответаКлиент выводит на экран полученное от сервера сообщение, вместе с номером потока и последовательным номером сообщения в этом потоке. После этого клиент возвращается на начало цикла, ожидая, что пользователь введет следующую строку.24-26Запуск программы
Мы запустили эхо-сервер SCTP без аргументов командной строки на компьютере, работающем под управлением FreeBSD. Клиенту при запуске необходимо указать IP-адрес сервера.