Шрифт:
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 < 0) {
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) < 8)
34 continue; /* недостаточно для проверки ICMP-заголовка */
35 if (icmp6->icmp6_type == ICMP6_TIME_EXCEEDED &&
36 icmp6->icmp6_code == ICMP6_TIME_EXCEED_TRANSIT) {
37 if (icmp6len < 8 + sizeof(struct ip6_hdr) + 4)
38 continue; /* недостаточно для проверки внутреннего заголовка */
39 hip6 = (struct ip6_hdr*)(recvbuf + 8);
40 hlen2 = sizeof(struct ip6_hdr);
41 udp = (struct udphdr*)(recvbuf + 8 + hlen2);
42 if (hip6->ip6_nxt == IPPROTO_UDP &&
43 udp->uh_sport == htons(sport) &&
44 udp->uh_dport == htons(dport + seq))
45 ret = -2; /* ответил промежуточный маршрутизатор */
46 break;
47 } else if (icmp6->icmp6_type == ICMP6_DST_UNREACH) {
48 if (icmp6len < 8 + sizeof(struct ip6_hdr) + 4)
49 continue; /* недостаточно для проверки внутреннего заголовка */
50 hip6 = (struct ip6_hdr*)(recvbuf + 8);
51 hlen2 = sizeof(struct ip6_hdr);
52 udp = (struct udphdr*)(recvbuf + 8 + hlen2);
53 if (hip6->ip6_nxt == IPPROTO_UDP &&