Уильям Стивенс - UNIX - разработка сетевых приложений

20 for (;;) {

21 if (gotalarm)

22 return(-3); /* истек таймер */

23 len = pr->salen;

24 n = recvfrom(recvfd, recvbuf, sizeof(recvbuf), 0, pr->sarecv, &len);

25 if (n

26 if (errno == EINTR)

27 continue;

28 else

29 err_sys("recvfrom error");

30 }

31 ip = (struct ip*)recvbuf; /* начало IP-заголовка */

32 hlenl = ip->ip_hl icmp_type == ICMP_TIMXCEED &&

37 icmp->icmp_code == ICMP_TIMXCEED_INTRANS) {

38 if (icmplen

39 continue; /* недостаточно данных для проверки внутреннего IP */

40 hip = (struct ip*)(recvbuf + hlen1 + 8);

41 hlen2 = hip->ip_hl ip_p == IPPROTO_UDP &&

46 udp->uh_sport == htons(sport) &&

47 udp->uh_dport == htons(dport + seq)) {

48 ret = -2; /* ответил промежуточный маршрутизатор */

49 break;

50 }

51 } else if (icmp->icmp_type == ICMP_UNREACH) {

52 if (icmplen

53 continue; /* недостаточно данных для проверки внутреннего IP */

54 hip = (struct ip*)(recvbuf + hlen1 + 8);

55 hlen2 = hip->ip_hl ip_p == IPPROTO_UDP &&

60 udp->uh_sport == htons(sport) &&

61 udp->uh_dport == htons(dport + seq)) {

62 if (icmp->icmp_code == ICMP_UNREACH_PORT)

63 ret = -1; /* цель достигнута */

64 else

65 ret = icmp->icmp_code; /* 0, 1, 2, ... */

66 break;

67 }

68 }

69 if (verbose) {

70 printf(" (from %s: type = %d, code - %d)\n",

71 Sock_ntop_host(pr->sarecv, pr->salen),

72 icmp->icmp_type, icmp->icmp_code);

73 }

74 /* другая ICMP-ошибка, нужно снова вызвать recvfrom() */

75 }

76 alarm(0); /* отключаем таймер */

77 Gettimeofday(tv, NULL); /* время получения пакета */

78 return(ret);

79 }

17-27 Таймер устанавливается на 3 с, и функция входит в цикл, вызывающий recvfrom, считывая каждое ICMPv4-сообщение, возвращаемое на символьный сокет.

Эта функция не создает ситуации гонок, описанной в разделе 20.5, благодаря использованию глобального флага.

31-35 Указатель iруказывает на начало IPv4-заголовка (напомним, что операция чтения на символьном сокете всегда возвращает IP-заголовок), а указатель icmpуказывает на начало ICMP-заголовка. На рис. 28.5 показаны различные заголовки, указатели и длины, используемые в данном коде.

Рис. 28.6. Заголовки, указатели и длины, используемые при обработке ошибки ICMPv6

Мы определяем две функции icmpcode_v4и icmpcode_v6, которые можно вызывать в конце функции traceloopдля вывода строки описания, соответствующей ICMP-ошибке недоступности получателя. В листинге 28.19 приведена IPv6-функция. IPv4-функция аналогична, хотя и длиннее, поскольку существует большее количество ICMPv4-кодов недоступности получателя (см. табл. А.5).

Последней функцией в нашей программе tracerouteявляется обработчик сигнала SIGALRM— функция sig_alrm, приведенная в листинге 28.17. Эта функция лишь возвращает ошибку EINTRиз функции recvfrom, как в случае функции recv_v4, так и в случае recv_v6.

Листинг 28.17. Функция sig_alrm

//traceroutе/sig_alrm.c

1 #include "trace.h"

2 int gotalarm;

3 void

4 sig_alrm(int signo)

5 {

6 gotalarm = 1; /* установка флага, оповещающего о сигнале */

7 return; /* прерывается работа функции recvfrom() */

8 }

Листинг 28.18. Функция recv_v6: чтение и обработка сообщений ICMPv6

//traceroute/recv_v6

1 #include "trace.h"

2 extern int gotalarm;

3 /*

4 * Возвращает; -3 при тайм-ауте

5 * -2 для сообщения ICMP time exceeded in transit (продолжаем поиск

маршрута)

6 * -1 для сообщения ICMP port unreachable (цель достигнута)

7 * неотрицательные значения соответствуют всем прочим ICMP-сообщениям

8 */

9 int

10 recv_v6(int seq, struct timeval *tv)

11 {

12 #ifdef IPV6

13 int hlen2, icmp6len, ret;

14 ssize_t n;

15 socklen_t len;

16 struct ip6_hdr *hip6;

17 struct icmp6_hdr *icmp6;

18 struct udphdr *udp;

19 gotalarm = 0;

20 alarm(3);

21 for (;;) {

22 if (gotalarm)

23 return(-3); /* истек таймер */

24 len = pr->salen;

25 n = recvfrom(recvfd, recvbuf, sizeof(recvbuf), 0, pr->sarecv, &len);

26 if (n

27 if (errno == EINTR)

28 continue;

29 else

30 err_sys("recvfrom error");

31 }

32 icmp6 = (struct icmp6_hdr*)recvbuf; /* ICMP-заголовок */

33 if ((icmp6len = n)

34 continue; /* недостаточно для проверки ICMP-заголовка */

35 if (icmp6->icmp6_type == ICMP6_TIME_EXCEEDED &&