3 * Report MTU on path to specified host
5 * (c) 2008 Straylight/Edgeware
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of Trivial IP Encryption (TrIPE).
12 * TrIPE is free software: you can redistribute it and/or modify it under
13 * the terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 3 of the License, or (at your
15 * option) any later version.
17 * TrIPE is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
22 * You should have received a copy of the GNU General Public License
23 * along with TrIPE. If not, see <https://www.gnu.org/licenses/>.
26 /*----- Header files ------------------------------------------------------*/
38 #include <sys/types.h>
42 #include <sys/socket.h>
43 #include <netinet/in.h>
44 #include <arpa/inet.h>
47 #include <netinet/in_systm.h>
48 #include <netinet/ip.h>
49 #include <netinet/ip_icmp.h>
50 #include <netinet/udp.h>
54 #include <sys/ioctl.h>
56 #include <mLib/alloc.h>
57 #include <mLib/bits.h>
58 #include <mLib/dstr.h>
60 #include <mLib/mdwopt.h>
61 #include <mLib/quis.h>
62 #include <mLib/report.h>
65 /*----- Static variables --------------------------------------------------*/
67 static unsigned char buf
[65536];
71 /*----- Utility functions -------------------------------------------------*/
73 /* Step a value according to a simple LFSR. */
75 do (q) = ((q) & 0x8000) ? ((q) << 1) ^ POLY : ((q) << 1); while (0)
77 /* Fill buffer with a constant but pseudorandom string. Uses a simple
80 static void fillbuffer(unsigned char *p
, size_t sz
)
82 unsigned int y
= 0xbc20;
83 const unsigned char *l
= p
+ sz
;
88 for (i
= 0; i
< 8; i
++) STEP(y
);
92 /* Convert a string to floating point. */
93 static double s2f(const char *s
, const char *what
)
100 if (errno
|| *q
) die(EXIT_FAILURE
, "bad %s", what
);
104 /* Convert a floating-point value into a struct timeval. */
105 static void f2tv(struct timeval
*tv
, double t
)
106 { tv
->tv_sec
= t
; tv
->tv_usec
= (t
- tv
->tv_sec
)*MILLION
; }
110 struct sockaddr_in sin
;
111 struct sockaddr_in6 sin6
;
114 /* Check whether an address family is even slightly supported. */
115 static int addrfamok(int af
)
118 case AF_INET
: case AF_INET6
: return (1);
123 /* Return the size of a socket address. */
124 static size_t addrsz(const union addr
*a
)
126 switch (a
->sa
.sa_family
) {
127 case AF_INET
: return (sizeof(a
->sin
));
128 case AF_INET6
: return (sizeof(a
->sin6
));
133 /* Compare two addresses. Maybe compare the port numbers too. */
135 static int addreq(const union addr
*a
, const union addr
*b
, unsigned f
)
137 switch (a
->sa
.sa_family
) {
139 return (a
->sin
.sin_addr
.s_addr
== b
->sin
.sin_addr
.s_addr
&&
140 (!(f
&AEF_PORT
) || a
->sin
.sin_port
== b
->sin
.sin_port
));
142 return (!memcmp(a
->sin6
.sin6_addr
.s6_addr
,
143 b
->sin6
.sin6_addr
.s6_addr
, 16) &&
144 (!(f
&AEF_PORT
) || a
->sin6
.sin6_port
== b
->sin6
.sin6_port
));
150 /*----- Main algorithm skeleton -------------------------------------------*/
153 unsigned f
; /* Various flags */
154 #define F_VERBOSE 1u /* Give a running commentary */
155 double retx
; /* Initial retransmit interval */
156 double regr
; /* Retransmit growth factor */
157 double timeout
; /* Retransmission timeout */
158 int seqoff
; /* Offset to write sequence number */
159 const struct probe_ops
*pops
; /* Probe algorithm description */
160 union addr a
; /* Destination address */
164 const struct param
*pp
;
170 const struct probe_ops
*next
;
172 int (*setup
)(void *, int, const struct param
*);
173 void (*finish
)(void *);
174 void (*selprep
)(void *, int *, fd_set
*);
175 int (*xmit
)(void *, int);
176 int (*selproc
)(void *, fd_set
*, struct probestate
*);
187 /* or a positive MTU upper-bound */
190 /* Add a file descriptor FD to the set `fd_in', updating `*maxfd'. */
192 do { FD_SET(fd, fd_in); if (*maxfd < fd) *maxfd = fd; } while (0)
194 /* Check whether a buffer contains a packet from our current probe. */
195 static int mypacketp(struct probestate
*ps
,
196 const unsigned char *p
, size_t sz
)
198 const struct param
*pp
= ps
->pp
;
200 return (sz
>= pp
->seqoff
+ 2 && LOAD16(p
+ pp
->seqoff
) == ps
->q
);
203 /* See whether MTU is an acceptable MTU value. Return an appropriate
204 * RC_... code or a new suggested MTU.
206 static int probe(struct probestate
*ps
, void *st
, int mtu
)
208 const struct param
*pp
= ps
->pp
;
210 struct timeval tv
, now
, when
, done
;
211 double timer
= pp
->retx
;
214 /* Set up the first retransmit and give-up timers. */
215 gettimeofday(&now
, 0);
216 f2tv(&tv
, pp
->timeout
); TV_ADD(&done
, &now
, &tv
);
217 f2tv(&tv
, timer
); TV_ADD(&when
, &now
, &tv
);
218 if (TV_CMP(&when
, >, &done
)) when
= done
;
220 /* Send the initial probe. */
221 if (pp
->f
& F_VERBOSE
)
222 moan("sending probe of size %d (seq = %04x)", mtu
, ps
->q
);
224 STORE16(buf
+ pp
->seqoff
, ps
->q
);
225 if ((rc
= pp
->pops
->xmit(st
, mtu
)) != RC_OK
) return (rc
);
229 /* Wait for something interesting to happen. */
230 maxfd
= 0; FD_ZERO(&fd_in
);
231 pp
->pops
->selprep(st
, &maxfd
, &fd_in
);
232 TV_SUB(&tv
, &when
, &now
);
233 if (select(maxfd
+ 1, &fd_in
, 0, 0, &tv
) < 0) return (RC_FAIL
);
234 gettimeofday(&now
, 0);
236 /* See whether the probe method has any answers for us. */
237 if ((rc
= pp
->pops
->selproc(st
, &fd_in
, ps
)) != RC_OK
) return (rc
);
239 /* If we've waited too long, give up. If we should retransmit, do
242 if (TV_CMP(&now
, >, &done
))
244 else if (TV_CMP(&now
, >, &when
)) {
245 if (pp
->f
& F_VERBOSE
) moan("re-sending probe of size %d", mtu
);
246 if ((rc
= pp
->pops
->xmit(st
, mtu
)) != RC_OK
) return (rc
);
248 timer
*= pp
->regr
; f2tv(&tv
, timer
); TV_ADD(&when
, &when
, &tv
);
249 } while (TV_CMP(&when
, <, &now
));
250 if (TV_CMP(&when
, >, &done
)) when
= done
;
255 /* Discover the path MTU to the destination address. */
256 static int pathmtu(const struct param
*pp
)
262 struct probestate ps
;
264 /* Build and connect a UDP socket. We'll need this to know the local port
265 * number to use if nothing else. Set other stuff up.
267 if ((sk
= socket(pp
->a
.sa
.sa_family
, SOCK_DGRAM
, IPPROTO_UDP
)) < 0)
269 if (connect(sk
, &pp
->a
.sa
, addrsz(&pp
->a
))) goto fail_1
;
270 st
= xmalloc(pp
->pops
->statesz
);
271 if ((mtu
= pp
->pops
->setup(st
, sk
, pp
)) < 0) goto fail_2
;
272 ps
.pp
= pp
; ps
.q
= rand() & 0xffff;
273 switch (pp
->a
.sa
.sa_family
) {
274 case AF_INET
: lo
= 576; break;
275 case AF_INET6
: lo
= 1280; break;
279 if (hi
< lo
) { errno
= EMSGSIZE
; return (-1); }
281 /* And now we do a thing which is sort of like a binary search, except that
282 * we also take explicit clues as establishing a new upper bound, and we
283 * try to hug that initially.
286 assert(lo
<= mtu
&& mtu
<= hi
);
287 if (pp
->f
& F_VERBOSE
) moan("probe: %d <= %d <= %d", lo
, mtu
, hi
);
288 rc
= probe(&ps
, st
, mtu
);
292 if (pp
->f
& F_VERBOSE
) moan("probe failed");
296 /* If we've not seen a dropped packet before then we don't know what
297 * this means yet -- in particular, we don't know which bit of the
298 * network is swallowing packets. Send a minimum-size probe. If
299 * that doesn't come back then assume that the remote host is
300 * swallowing our packets. If it does, then we assume that dropped
301 * packets are a result of ICMP fragmentation-needed reports being
302 * lost or suppressed.
304 if (pp
->f
& F_VERBOSE
) moan("gave up: black hole detected");
306 if (pp
->f
& F_VERBOSE
) moan("sending minimum-size probe");
307 switch (probe(&ps
, st
, lo
)) {
311 if (pp
->f
& F_VERBOSE
) {
312 moan("no reply from min-size probe: "
313 "assume black hole at target");
318 if (pp
->f
& F_VERBOSE
) {
319 moan("reply from min-size probe OK: "
320 "assume black hole in network");
325 if (pp
->f
& F_VERBOSE
)
326 moan("unexpected return code from probe");
332 if (droppy
) goto higher
; else goto lower
;
337 if (pp
->f
& F_VERBOSE
)
338 moan("probe returned: remote host is not a black hole");
342 if (pp
->f
& F_VERBOSE
) moan("probe returned: found correct MTU");
347 /* Now we must make a new guess, between lo and hi. We know that lo
348 * is good; but we're not so sure about hi here. We know that hi >
349 * lo, so this will find an approximate midpoint, greater than lo and
352 if (pp
->f
& F_VERBOSE
) moan("probe returned: guessing higher");
353 mtu
+= (hi
- lo
+ 1)/2;
358 /* If this didn't work, and we're already at the bottom of our
359 * possible range, then something has gone horribly wrong.
364 if (pp
->f
& F_VERBOSE
) moan("error returned: found correct MTU");
369 /* We must make a new guess, between lo and hi. We're probably
370 * fairly sure that lo will succeed, since either it's the minimum
371 * MTU or we've tested it already; but we're not quite sure about hi,
372 * so we want to aim high.
374 if (pp
->f
& F_VERBOSE
) moan("error returned: guessing lower");
375 mtu
-= (hi
- lo
+ 1)/2;
379 if (pp
->f
& F_VERBOSE
) moan("error returned with new MTU estimate");
386 /* Clean up and return our result. */
387 pp
->pops
->finish(st
);
393 pp
->pops
->finish(st
);
402 /*----- Doing it the hard way ---------------------------------------------*/
404 #if defined(linux) || defined(__OpenBSD__)
409 # define sane_htons htons
410 # define sane_htonl htonl
416 static int rawicmp
= -1, rawudp
= -1, rawerr
= 0;
418 #define IPCK_INIT 0xffff
420 /* Compute an IP checksum over some data. This is a restartable interface:
421 * initialize A to `IPCK_INIT' for the first call.
423 static unsigned ipcksum(const void *buf
, size_t n
, unsigned a
)
425 unsigned long aa
= a
^ 0xffff;
426 const unsigned char *p
= buf
, *l
= p
+ n
;
428 while (p
< l
- 1) { aa
+= LOAD16_B(p
); p
+= 2; }
429 if (p
< l
) { aa
+= (unsigned)(*p
) << 8; }
430 do aa
= (aa
& 0xffff) + (aa
>> 16); while (aa
>= 0x10000);
431 return (aa
== 0xffff ? aa
: aa
^ 0xffff);
434 /* TCP/UDP pseudoheader structure. */
436 struct in_addr ph_src
, ph_dst
;
443 int sk
, rawicmp
, rawudp
;
444 uint16_t srcport
, dstport
;
448 static int raw_setup(void *stv
, int sk
, const struct param
*pp
)
450 struct raw_state
*st
= stv
;
453 struct ifaddrs
*ifa
, *ifaa
, *ifap
;
456 /* Check that the address is OK, and that we have the necessary raw
459 switch (pp
->a
.sa
.sa_family
) {
461 if (rawerr
) { errno
= rawerr
; goto fail_0
; }
462 st
->rawicmp
= rawicmp
; st
->rawudp
= rawudp
; st
->sk
= sk
;
465 errno
= EPFNOSUPPORT
; goto fail_0
;
468 /* Initialize the sequence number. */
469 st
->q
= rand() & 0xffff;
471 /* Snaffle the local and remote address and port number. */
474 if (getsockname(sk
, &st
->me
.sa
, &sz
))
477 /* An unfortunate bodge which will make sense in the future. */
478 st
->srcport
= st
->me
.sin
.sin_port
; st
->me
.sin
.sin_port
= 0;
479 st
->dstport
= st
->a
.sin
.sin_port
; st
->a
.sin
.sin_port
= 0;
481 /* There isn't a portable way to force the DF flag onto a packet through
482 * UDP, or even through raw IP, unless we write the entire IP header
483 * ourselves. This is somewhat annoying, especially since we have an
484 * uphill struggle keeping track of which systems randomly expect which
485 * header fields to be presented in host byte order. Oh, well.
488 if (setsockopt(rawudp
, IPPROTO_IP
, IP_HDRINCL
, &i
, sizeof(i
))) goto fail_0
;
490 /* Find an upper bound on the MTU. Do two passes over the interface
491 * list. If we can find matches for our local address then use the
492 * highest one of those; otherwise do a second pass and simply take the
493 * highest MTU of any network interface.
495 if (getifaddrs(&ifaa
)) goto fail_0
;
496 for (i
= 0; i
< 2; i
++) {
497 for (ifap
= 0, ifa
= ifaa
; ifa
; ifa
= ifa
->ifa_next
) {
498 if (!(ifa
->ifa_flags
& IFF_UP
) || !ifa
->ifa_addr
||
499 ifa
->ifa_addr
->sa_family
!= st
->me
.sa
.sa_family
||
501 !addreq((union addr
*)ifa
->ifa_addr
, &st
->me
, 0)) ||
502 (i
== 1 && ifap
&& strcmp(ifap
->ifa_name
, ifa
->ifa_name
) == 0) ||
503 strlen(ifa
->ifa_name
) >= sizeof(ifr
.ifr_name
))
506 strcpy(ifr
.ifr_name
, ifa
->ifa_name
);
507 if (ioctl(sk
, SIOCGIFMTU
, &ifr
)) goto fail_1
;
508 if (mtu
< ifr
.ifr_mtu
) mtu
= ifr
.ifr_mtu
;
512 if (mtu
< 0) { errno
= ENOTCONN
; goto fail_1
; }
524 static void raw_finish(void *stv
) { ; }
526 static void raw_selprep(void *stv
, int *maxfd
, fd_set
*fd_in
)
527 { struct raw_state
*st
= stv
; ADDFD(st
->sk
); ADDFD(st
->rawicmp
); }
529 static int raw_xmit(void *stv
, int mtu
)
531 struct raw_state
*st
= stv
;
532 unsigned char b
[65536], *p
;
538 /* Build the IP header. */
541 ip
->ip_hl
= sizeof(*ip
)/4;
542 ip
->ip_tos
= IPTOS_RELIABILITY
;
543 ip
->ip_len
= sane_htons(mtu
);
544 STEP(st
->q
); ip
->ip_id
= htons(st
->q
);
545 ip
->ip_off
= sane_htons(0 | IP_DF
);
547 ip
->ip_p
= IPPROTO_UDP
;
549 ip
->ip_src
= st
->me
.sin
.sin_addr
;
550 ip
->ip_dst
= st
->a
.sin
.sin_addr
;
552 /* Build a UDP packet in the output buffer. */
553 udp
= (struct udphdr
*)(ip
+ 1);
554 udp
->uh_sport
= st
->srcport
;
555 udp
->uh_dport
= st
->dstport
;
556 udp
->uh_ulen
= htons(mtu
- sizeof(*ip
));
559 /* Copy the payload. */
560 p
= (unsigned char *)(udp
+ 1);
561 memcpy(p
, buf
, mtu
- (p
- b
));
563 /* Calculate the UDP checksum. */
564 ph
.ph_src
= ip
->ip_src
;
565 ph
.ph_dst
= ip
->ip_dst
;
567 ph
.ph_p
= IPPROTO_UDP
;
568 ph
.ph_len
= udp
->uh_ulen
;
570 ck
= ipcksum(&ph
, sizeof(ph
), ck
);
571 ck
= ipcksum(udp
, mtu
- sizeof(*ip
), ck
);
572 udp
->uh_sum
= htons(ck
);
574 /* Send the whole thing off. If we're too big for the interface then we
575 * might need to trim immediately.
577 if (sendto(st
->rawudp
, b
, mtu
, 0, &st
->a
.sa
, addrsz(&st
->a
)) < 0) {
578 if (errno
== EMSGSIZE
) return (RC_LOWER
);
589 static int raw_selproc(void *stv
, fd_set
*fd_in
, struct probestate
*ps
)
591 struct raw_state
*st
= stv
;
592 unsigned char b
[65536];
596 const unsigned char *payload
;
599 /* An ICMP packet: see what's inside. */
600 if (FD_ISSET(st
->rawicmp
, fd_in
)) {
601 if ((n
= read(st
->rawicmp
, b
, sizeof(b
))) < 0) goto fail_0
;
604 if (n
< sizeof(*ip
) || n
< sizeof(4*ip
->ip_hl
) ||
605 ip
->ip_v
!= 4 || ip
->ip_p
!= IPPROTO_ICMP
)
607 n
-= sizeof(4*ip
->ip_hl
);
609 icmp
= (struct icmp
*)(b
+ 4*ip
->ip_hl
);
610 if (n
< sizeof(*icmp
) || icmp
->icmp_type
!= ICMP_UNREACH
)
612 n
-= offsetof(struct icmp
, icmp_ip
);
615 if (n
< sizeof(*ip
) ||
616 ip
->ip_p
!= IPPROTO_UDP
|| ip
->ip_hl
!= sizeof(*ip
)/4 ||
617 ip
->ip_id
!= htons(st
->q
) ||
618 ip
->ip_src
.s_addr
!= st
->me
.sin
.sin_addr
.s_addr
||
619 ip
->ip_dst
.s_addr
!= st
->a
.sin
.sin_addr
.s_addr
)
623 udp
= (struct udphdr
*)(ip
+ 1);
624 if (n
< sizeof(*udp
) || udp
->uh_sport
!= st
->srcport
||
625 udp
->uh_dport
!= st
->dstport
)
629 payload
= (const unsigned char *)(udp
+ 1);
630 if (!mypacketp(ps
, payload
, n
)) goto skip_icmp
;
632 if (icmp
->icmp_code
== ICMP_UNREACH_PORT
) return (RC_HIGHER
);
633 else if (icmp
->icmp_code
!= ICMP_UNREACH_NEEDFRAG
) goto skip_icmp
;
634 else if (icmp
->icmp_nextmtu
) return (htons(icmp
->icmp_nextmtu
));
635 else return (RC_LOWER
);
639 /* If we got a reply to the current probe then we're good. If we got an
640 * error, or the packet's sequence number is wrong, then ignore it.
642 if (FD_ISSET(st
->sk
, fd_in
)) {
643 if ((n
= read(st
->sk
, b
, sizeof(b
))) < 0) return (RC_OK
);
644 else if (mypacketp(ps
, b
, n
)) return (RC_HIGHER
);
654 static const struct probe_ops raw_ops
= {
655 "raw", OPS_CHAIN
, sizeof(struct raw_state
),
656 raw_setup
, raw_finish
,
657 raw_selprep
, raw_xmit
, raw_selproc
661 #define OPS_CHAIN &raw_ops
663 /*----- Doing the job on Linux --------------------------------------------*/
668 # define IP_MTU 14 /* Blech! */
672 int sol
, so_mtu_discover
, so_mtu
;
677 static int linux_setup(void *stv
, int sk
, const struct param
*pp
)
679 struct linux_state
*st
= stv
;
683 /* Check that the address is OK. */
684 switch (pp
->a
.sa
.sa_family
) {
686 st
->sol
= IPPROTO_IP
;
687 st
->so_mtu_discover
= IP_MTU_DISCOVER
;
692 st
->sol
= IPPROTO_IPV6
;
693 st
->so_mtu_discover
= IPV6_MTU_DISCOVER
;
694 st
->so_mtu
= IPV6_MTU
;
698 errno
= EPFNOSUPPORT
;
702 /* Snaffle the UDP socket. */
705 /* Turn on kernel path-MTU discovery and force DF on. */
706 i
= IP_PMTUDISC_PROBE
;
707 if (setsockopt(st
->sk
, st
->sol
, st
->so_mtu_discover
, &i
, sizeof(i
)))
710 /* Read the initial MTU guess back and report it. */
712 if (getsockopt(st
->sk
, st
->sol
, st
->so_mtu
, &mtu
, &sz
))
719 static void linux_finish(void *stv
) { ; }
721 static void linux_selprep(void *stv
, int *maxfd
, fd_set
*fd_in
)
722 { struct linux_state
*st
= stv
; ADDFD(st
->sk
); }
724 static int linux_xmit(void *stv
, int mtu
)
726 struct linux_state
*st
= stv
;
728 /* Write the packet. */
729 if (write(st
->sk
, buf
, mtu
- st
->hdrlen
) >= 0) return (RC_OK
);
730 else if (errno
== EMSGSIZE
) return (RC_LOWER
);
731 else return (RC_FAIL
);
734 static int linux_selproc(void *stv
, fd_set
*fd_in
, struct probestate
*ps
)
736 struct linux_state
*st
= stv
;
740 unsigned char b
[65536];
742 /* Read an answer. If it looks like the right kind of error then report a
743 * success. This is potentially wrong, since we can't tell whether an
744 * error was delayed from an earlier probe. However, we never return
745 * RC_LOWER from this method, so the packet sizes ought to be monotonically
746 * decreasing and this won't cause trouble. Otherwise update from the
747 * kernel's idea of the right MTU.
749 if (FD_ISSET(st
->sk
, fd_in
)) {
750 n
= read(st
->sk
, &buf
, sizeof(buf
));
752 mypacketp(ps
, b
, n
) :
753 errno
== ECONNREFUSED
|| errno
== EHOSTUNREACH
)
756 if (getsockopt(st
->sk
, st
->sol
, st
->so_mtu
, &mtu
, &sz
))
763 static const struct probe_ops linux_ops
= {
764 "linux", OPS_CHAIN
, sizeof(struct linux_state
),
765 linux_setup
, linux_finish
,
766 linux_selprep
, linux_xmit
, linux_selproc
770 #define OPS_CHAIN &linux_ops
774 /*----- Help options ------------------------------------------------------*/
776 static const struct probe_ops
*probe_ops
= OPS_CHAIN
;
778 static void version(FILE *fp
)
779 { pquis(fp
, "$, TrIPE version " VERSION
"\n"); }
781 static void usage(FILE *fp
)
783 pquis(fp
, "Usage: $ [-46v] [-H HEADER] [-m METHOD]\n\
784 [-r SECS] [-g FACTOR] [-t SECS] HOST [PORT]\n");
787 static void help(FILE *fp
)
789 const struct probe_ops
*ops
;
798 -h, --help Show this help text.\n\
799 -V, --version Show version number.\n\
800 -u, --usage Show brief usage message.\n\
802 -4, --ipv4 Restrict to IPv4 only.\n\
803 -6, --ipv6 Restrict to IPv6 only.\n\
804 -g, --growth=FACTOR Growth factor for retransmit interval.\n\
805 -m, --method=METHOD Use METHOD to probe for MTU.\n\
806 -r, --retransmit=SECS Retransmit if no reply after SEC.\n\
807 -t, --timeout=SECS Give up expecting a reply after SECS.\n\
808 -v, --verbose Write a running commentary to stderr.\n\
809 -H, --header=HEX Packet header, in hexadecimal.\n\
813 for (ops
= probe_ops
; ops
; ops
= ops
->next
)
814 printf("\t%s\n", ops
->name
);
817 /*----- Main code ---------------------------------------------------------*/
819 int main(int argc
, char *argv
[])
821 struct param pp
= { 0, 0.333, 3.0, 8.0, 0, OPS_CHAIN
};
826 struct addrinfo aihint
= { 0 }, *ailist
, *ai
;
827 const char *host
, *svc
= "7";
832 if ((rawicmp
= socket(PF_INET
, SOCK_RAW
, IPPROTO_ICMP
)) < 0 ||
833 (rawudp
= socket(PF_INET
, SOCK_RAW
, IPPROTO_UDP
)) < 0)
835 if (setuid(getuid()))
839 fillbuffer(buf
, sizeof(buf
));
841 aihint
.ai_family
= AF_UNSPEC
;
842 aihint
.ai_protocol
= IPPROTO_UDP
;
843 aihint
.ai_socktype
= SOCK_DGRAM
;
844 aihint
.ai_flags
= AI_ADDRCONFIG
;
847 static const struct option opts
[] = {
848 { "help", 0, 0, 'h' },
849 { "version", 0, 0, 'V' },
850 { "usage", 0, 0, 'u' },
851 { "ipv4", 0, 0, '4' },
852 { "ipv6", 0, 0, '6' },
853 { "header", OPTF_ARGREQ
, 0, 'H' },
854 { "growth", OPTF_ARGREQ
, 0, 'g' },
855 { "method", OPTF_ARGREQ
, 0, 'm' },
856 { "retransmit", OPTF_ARGREQ
, 0, 'r' },
857 { "timeout", OPTF_ARGREQ
, 0, 't' },
858 { "verbose", 0, 0, 'v' },
862 i
= mdwopt(argc
, argv
, "hVu" "46H:g:m:r:t:v", opts
, 0, 0, 0);
865 case 'h': help(stdout
); exit(0);
866 case 'V': version(stdout
); exit(0);
867 case 'u': usage(stdout
); exit(0);
872 hex_decode(&hc
, optarg
, strlen(optarg
), &d
);
873 hex_decode(&hc
, 0, 0, &d
);
874 sz
= d
.len
< 532 ? d
.len
: 532;
875 memcpy(buf
, d
.buf
, sz
);
879 case '4': aihint
.ai_family
= AF_INET
; break;
880 case '6': aihint
.ai_family
= AF_INET6
; break;
881 case 'g': pp
.regr
= s2f(optarg
, "retransmit growth factor"); break;
882 case 'r': pp
.retx
= s2f(optarg
, "retransmit interval"); break;
883 case 't': pp
.timeout
= s2f(optarg
, "timeout"); break;
886 for (pp
.pops
= OPS_CHAIN
; pp
.pops
; pp
.pops
= pp
.pops
->next
)
887 if (strcmp(pp
.pops
->name
, optarg
) == 0) goto found_alg
;
888 die(EXIT_FAILURE
, "unknown probe algorithm `%s'", optarg
);
892 case 'v': pp
.f
|= F_VERBOSE
; break;
899 argv
+= optind
; argc
-= optind
;
900 if ((f
& f_bogus
) || 1 > argc
|| argc
> 2) {
906 if (argv
[1]) svc
= argv
[1];
907 if ((err
= getaddrinfo(host
, svc
, &aihint
, &ailist
)) != 0) {
908 die(EXIT_FAILURE
, "unknown host `%s' or service `%s': %s",
909 host
, svc
, gai_strerror(err
));
911 for (ai
= ailist
; ai
&& !addrfamok(ai
->ai_family
); ai
= ai
->ai_next
);
912 if (!ai
) die(EXIT_FAILURE
, "no supported address families for `%s'", host
);
913 assert(ai
->ai_addrlen
<= sizeof(pp
.a
));
914 memcpy(&pp
.a
, ai
->ai_addr
, ai
->ai_addrlen
);
918 die(EXIT_FAILURE
, "failed to discover MTU: %s", strerror(errno
));
920 if (ferror(stdout
) || fflush(stdout
) || fclose(stdout
))
921 die(EXIT_FAILURE
, "failed to write result: %s", strerror(errno
));
925 /*----- That's all, folks -------------------------------------------------*/