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

Эхо-клиент UDP, использующий демон icmpd

Эхо-клиент UDP, использующий демон icmpd

Теперь модифицируем наш эхо-клиент UDP (функцию dg_cli) для использования нашего демона icmpd. В листинге 28.21 приведена первая половина функции.

Листинг 28.21. Первая часть приложения dg_cli

//icmpd/dgcli01.c
 1 #include "unpicmpd.h"
 2 void
 3 dg_cli(FILE *fp, int sockfd, const SA *pservadd, socklen_t servlen)
 4 {
 5  int icmpfd, maxfdp1;
 6  char sendline[MAXLINE], recvline[MAXLINE + 1];
 7  fd_set rset;
 8  ssize_t n;
 9  struct timeval tv;
10  struct icmpd_err icmpd_err;
11  struct sockaddr_un sun;
12  Sock_bind_wild(sockfd, pservaddr->sa_family);
13  icmpfd = Socket(AF_LOCAL, SOCK_STREAM, 0);
14  sun.sun_family = AF_LOCAL;
15  strcpy(sun.sun_path, ICMPD_PATH);
16  Connect(icmpfd, (SA*)&sun, sizeof(sun));
17  Write_fd(icmpfd, "1", 1, sockfd);
18  n = Read(icmpfd, recvline, 1);
19  if (n != 1 || recvline[0] != '1')
20   err_quit("error creating icmp socket, n = %d, char = %c",
21   n, recvline[0]);
22  FD_ZERO(&rset);
23  maxfdp1 = max(sockfd, icmpfd) + 1;
2-3 Аргументы функции те же, что и во всех ее предыдущих версиях.

12 Вызываем функцию sock_bind_wild для связывания при помощи функции bind универсального IP-адреса и динамически назначаемого порта с UDP-сокетом. Таким образом копия сокета, который пересылается демону, оказывается связана с портом, поскольку демону необходимо знать этот порт.

13-16 Мы создаем сокет семейства AF_INET и подключаемся к известному имени сервера при помощи вызова connect.

17-21 Вызываем функцию write_fd, приведенную в листинге 15.11 для отправки копии UDP-сокета демону. Мы также посылаем одиночный байт данных — символ "1", поскольку некоторые реализации не передают дескриптор без данных. Демон посылает обратно одиночный байт данных, состоящий из символа "1", для обозначения успешного выполнения. Любой другой ответ означает ошибку.

22-23 Инициализируем набор дескрипторов и вычисляем первый аргумент для функции select (максимальный из двух дескрипторов, увеличенный на единицу).

Вторая половина нашего клиента приведена в листинге 28.22. Это цикл, который считывает данные из стандартного ввода, посылает строку серверу, считывает ответ сервера и записывает ответ в стандартный вывод.

Листинг 28.22. Вторая часть приложения dg_cli

//icmpd/dgcli01.c
24  while (Fgets(sendline, MAXLINE, fp) != NULL) {
25   Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
26   tv.tv_sec = 5;
27   tv.tv_usec = 0;
28   FD_SET(sockfd, &rset);
29   FD_SET(icmpfd, &rset);
30   if ((n = Select(maxfdp1, &rset, NULL, NULL, &tv)) == 0) {
31    fprintf(stderr, "socket timeoutn");
32    continue;
33   }
34   if (FD_ISSET(sockfd, &rset)) {
35    n = Recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
36    recvline[n] = 0; /* завершающий нуль */
37    Fputs(recvline, stdout);
38   }
39   if (FD_ISSET(icmpfd, &rset)) {
40    if ((n = Read(icmpfd, &icmpd_err, sizeof(icmpd_err))) == 0)
41     err_quit("ICMP daemon terminated");
42    else if (n != sizeof(icmpd_err))
43     err_quit("n = %d, expected %d", n, sizeof(icmpd_err)),
44    printf("ICMP error: dest = %s, %s, type = %d, code = %dn",
45     Sock_ntop(&icmpd_err.icmpd_dest, icmpd_err.icmpd_len);
46    strerror(icmpd_err.icmpd_errno),
47     icmpd_err.icmpd_type, icmpd_err.icmpd_code);
48   }
49  }
50 }

26-33 Поскольку мы вызываем функцию select, мы можем легко установить время ожидания ответа от эхо-сервера. Задаем его равным 5 с, открываем оба дескриптора для чтения и вызываем функцию select. Если происходит превышение времени, выводится соответствующее сообщение и осуществляется переход в начало цикла.

34-38 Если дейтаграмма возвращается сервером, она выводится в стандартный поток вывода.

39-48 Если наше доменное соединение Unix с демоном icmpd готово для чтения, мы пытаемся прочитать структуру icmpd_err. Если это удается, выводится соответствующая информация, возвращаемая демоном.

Оглавление книги