--- a/netselect.c +++ b/netselect.c @@ -39,14 +39,14 @@ * hey, great! Let me know. -- apenwarr */ -#ifdef __EMX__ -# include -# include -# include -# include -# include -#else -# include +#ifdef __linux__ +#include +#include +#elif defined(__OpenBSD__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__APPLE__) +#if defined(__OpenBSD__) +#include +#endif +#include #endif #include @@ -167,6 +167,7 @@ extern int optind; int hostcount, startcount, endcount = 0, sent_one, lag, min_lag = 100; int ch, seq, ttl, max_ttl = 30, num_score = 1; + int on = 1; int use_icmp = 0; unsigned int min_tries = 10; struct timeval now; @@ -234,7 +235,12 @@ if (verbose >= 1) fprintf(stderr, "Running netselect to choose %d out of %d address%s.\n", num_score, numhosts, numhosts==1 ? "" : "es"); - + + if (setsockopt(sndsock, IPPROTO_IP, IP_HDRINCL, &on, sizeof(on)) < 0) { + perror("IP_HDRINCL"); + _exit(EXIT_FAILURE); + } + /* keep going until most of the hosts have been finished */ must_continue = numhosts; while (must_continue && must_continue >= numhosts/2) @@ -712,7 +718,7 @@ ip->ip_off = 0; ip->ip_hl = sizeof(*ip) >> 2; ip->ip_p = IPPROTO_UDP; - ip->ip_len = 0; /* kernel fills this in */ + ip->ip_len = sizeof(OPacket); ip->ip_ttl = ttl; ip->ip_v = IPVERSION; ip->ip_id = htons(ident + seq);