Home | History | Annotate | Download | only in iputils
      1 #include <stdio.h>
      2 #include <stdlib.h>
      3 #include <unistd.h>
      4 #include <time.h>
      5 #include <sys/param.h>
      6 #include <sys/socket.h>
      7 #include <linux/sockios.h>
      8 #include <sys/file.h>
      9 #include <sys/time.h>
     10 #include <sys/signal.h>
     11 #include <sys/ioctl.h>
     12 #include <net/if.h>
     13 #include <sys/uio.h>
     14 #include <sys/poll.h>
     15 #include <ctype.h>
     16 #include <errno.h>
     17 #include <string.h>
     18 #include <netdb.h>
     19 #include <setjmp.h>
     20 
     21 #ifdef CAPABILITIES
     22 #include <sys/prctl.h>
     23 #include <sys/capability.h>
     24 #endif
     25 
     26 #ifdef USE_IDN
     27 #include <locale.h>
     28 #include <idna.h>
     29 #endif
     30 
     31 #include <netinet/in.h>
     32 #include <arpa/inet.h>
     33 #include <linux/types.h>
     34 #include <linux/errqueue.h>
     35 
     36 #ifdef ANDROID
     37 #include <linux/icmp.h>
     38 #include <sys/auxv.h>
     39 #endif
     40 
     41 #include "SNAPSHOT.h"
     42 
     43 #define	DEFDATALEN	(64 - 8)	/* default data length */
     44 
     45 #define	MAXWAIT		10		/* max seconds to wait for response */
     46 #define MININTERVAL	10		/* Minimal interpacket gap */
     47 #define MINUSERINTERVAL	200		/* Minimal allowed interval for non-root */
     48 
     49 #define SCHINT(a)	(((a) <= MININTERVAL) ? MININTERVAL : (a))
     50 
     51 /* various options */
     52 extern int options;
     53 #define	F_FLOOD		0x001
     54 #define	F_INTERVAL	0x002
     55 #define	F_NUMERIC	0x004
     56 #define	F_PINGFILLED	0x008
     57 #define	F_QUIET		0x010
     58 #define	F_RROUTE	0x020
     59 #define	F_SO_DEBUG	0x040
     60 #define	F_SO_DONTROUTE	0x080
     61 #define	F_VERBOSE	0x100
     62 #define	F_TIMESTAMP	0x200
     63 #define	F_FLOWINFO	0x200
     64 #define	F_SOURCEROUTE	0x400
     65 #define	F_TCLASS	0x400
     66 #define	F_FLOOD_POLL	0x800
     67 #define	F_LATENCY	0x1000
     68 #define	F_AUDIBLE	0x2000
     69 #define	F_ADAPTIVE	0x4000
     70 #define	F_STRICTSOURCE	0x8000
     71 #define F_NOLOOP	0x10000
     72 #define F_TTL		0x20000
     73 #define F_MARK		0x40000
     74 #define F_PTIMEOFDAY	0x80000
     75 #define F_OUTSTANDING	0x100000
     76 
     77 /*
     78  * MAX_DUP_CHK is the number of bits in received table, i.e. the maximum
     79  * number of received sequence numbers we can keep track of.
     80  */
     81 #define	MAX_DUP_CHK	0x10000
     82 
     83 #if defined(__WORDSIZE) && __WORDSIZE == 64
     84 # define USE_BITMAP64
     85 #endif
     86 
     87 #ifdef USE_BITMAP64
     88 typedef __u64	bitmap_t;
     89 # define BITMAP_SHIFT	6
     90 #else
     91 typedef __u32	bitmap_t;
     92 # define BITMAP_SHIFT	5
     93 #endif
     94 
     95 #if ((MAX_DUP_CHK >> (BITMAP_SHIFT + 3)) << (BITMAP_SHIFT + 3)) != MAX_DUP_CHK
     96 # error Please MAX_DUP_CHK and/or BITMAP_SHIFT
     97 #endif
     98 
     99 struct rcvd_table {
    100 	bitmap_t bitmap[MAX_DUP_CHK / (sizeof(bitmap_t) * 8)];
    101 };
    102 
    103 extern struct rcvd_table rcvd_tbl;
    104 extern int using_ping_socket;
    105 
    106 #define	A(bit)	(rcvd_tbl.bitmap[(bit) >> BITMAP_SHIFT])	/* identify word in array */
    107 #define	B(bit)	(((bitmap_t)1) << ((bit) & ((1 << BITMAP_SHIFT) - 1)))	/* identify bit in word */
    108 
    109 static inline void rcvd_set(__u16 seq)
    110 {
    111 	unsigned bit = seq % MAX_DUP_CHK;
    112 	A(bit) |= B(bit);
    113 }
    114 
    115 static inline void rcvd_clear(__u16 seq)
    116 {
    117 	unsigned bit = seq % MAX_DUP_CHK;
    118 	A(bit) &= ~B(bit);
    119 }
    120 
    121 static inline bitmap_t rcvd_test(__u16 seq)
    122 {
    123 	unsigned bit = seq % MAX_DUP_CHK;
    124 	return A(bit) & B(bit);
    125 }
    126 
    127 extern u_char outpack[];
    128 extern int maxpacket;
    129 
    130 extern int datalen;
    131 extern char *hostname;
    132 extern int uid;
    133 extern int ident;			/* process id to identify our packets */
    134 
    135 extern int sndbuf;
    136 extern int ttl;
    137 
    138 extern long npackets;			/* max packets to transmit */
    139 extern long nreceived;			/* # of packets we got back */
    140 extern long nrepeats;			/* number of duplicates */
    141 extern long ntransmitted;		/* sequence # for outbound packets = #sent */
    142 extern long nchecksum;			/* replies with bad checksum */
    143 extern long nerrors;			/* icmp errors */
    144 extern int interval;			/* interval between packets (msec) */
    145 extern int preload;
    146 extern int deadline;			/* time to die */
    147 extern int lingertime;
    148 extern struct timeval start_time, cur_time;
    149 extern volatile int exiting;
    150 extern volatile int status_snapshot;
    151 extern int confirm;
    152 extern int confirm_flag;
    153 extern int working_recverr;
    154 
    155 extern volatile int in_pr_addr;		/* pr_addr() is executing */
    156 extern jmp_buf pr_addr_jmp;
    157 
    158 #ifndef MSG_CONFIRM
    159 #define MSG_CONFIRM 0
    160 #endif
    161 
    162 
    163 /* timing */
    164 extern int timing;			/* flag to do timing */
    165 extern long tmin;			/* minimum round trip time */
    166 extern long tmax;			/* maximum round trip time */
    167 extern long long tsum;			/* sum of all times, for doing average */
    168 extern long long tsum2;
    169 extern int rtt;
    170 extern __u16 acked;
    171 extern int pipesize;
    172 
    173 #define COMMON_OPTIONS \
    174 case 'a': case 'U': case 'c': case 'd': \
    175 case 'f': case 'i': case 'w': case 'l': \
    176 case 'S': case 'n': case 'p': case 'q': \
    177 case 'r': case 's': case 'v': case 'L': \
    178 case 't': case 'A': case 'W': case 'B': case 'm': \
    179 case 'D': case 'O':
    180 
    181 #define COMMON_OPTSTR "h?VQ:I:M:aUc:dfi:w:l:S:np:qrs:vLt:AW:Bm:DO"
    182 
    183 /*
    184  * Write to stdout
    185  */
    186 static inline void write_stdout(const char *str, size_t len)
    187 {
    188 	size_t o = 0;
    189 	ssize_t cc;
    190 	do {
    191 		cc = write(STDOUT_FILENO, str + o, len - o);
    192 		o += cc;
    193 	} while (len > o || cc < 0);
    194 }
    195 
    196 /*
    197  * tvsub --
    198  *	Subtract 2 timeval structs:  out = out - in.  Out is assumed to
    199  * be >= in.
    200  */
    201 static inline void tvsub(struct timeval *out, struct timeval *in)
    202 {
    203 	if ((out->tv_usec -= in->tv_usec) < 0) {
    204 		--out->tv_sec;
    205 		out->tv_usec += 1000000;
    206 	}
    207 	out->tv_sec -= in->tv_sec;
    208 }
    209 
    210 static inline void set_signal(int signo, void (*handler)(int))
    211 {
    212 	struct sigaction sa;
    213 
    214 	memset(&sa, 0, sizeof(sa));
    215 
    216 	sa.sa_handler = (void (*)(int))handler;
    217 #ifdef SA_INTERRUPT
    218 	sa.sa_flags = SA_INTERRUPT;
    219 #endif
    220 	sigaction(signo, &sa, NULL);
    221 }
    222 
    223 extern int __schedule_exit(int next);
    224 
    225 static inline int schedule_exit(int next)
    226 {
    227 	if (npackets && ntransmitted >= npackets && !deadline)
    228 		next = __schedule_exit(next);
    229 	return next;
    230 }
    231 
    232 static inline int in_flight(void)
    233 {
    234 	__u16 diff = (__u16)ntransmitted - acked;
    235 	return (diff<=0x7FFF) ? diff : ntransmitted-nreceived-nerrors;
    236 }
    237 
    238 static inline void acknowledge(__u16 seq)
    239 {
    240 	__u16 diff = (__u16)ntransmitted - seq;
    241 	if (diff <= 0x7FFF) {
    242 		if ((int)diff+1 > pipesize)
    243 			pipesize = (int)diff+1;
    244 		if ((__s16)(seq - acked) > 0 ||
    245 		    (__u16)ntransmitted - acked > 0x7FFF)
    246 			acked = seq;
    247 	}
    248 }
    249 
    250 static inline void advance_ntransmitted(void)
    251 {
    252 	ntransmitted++;
    253 	/* Invalidate acked, if 16 bit seq overflows. */
    254 	if ((__u16)ntransmitted - acked > 0x7FFF)
    255 		acked = (__u16)ntransmitted + 1;
    256 }
    257 
    258 extern void limit_capabilities(void);
    259 static int enable_capability_raw(void);
    260 static int disable_capability_raw(void);
    261 static int enable_capability_admin(void);
    262 static int disable_capability_admin(void);
    263 #ifdef CAPABILITIES
    264 extern int modify_capability(cap_value_t, cap_flag_value_t);
    265 static inline int enable_capability_raw(void)		{ return modify_capability(CAP_NET_RAW,   CAP_SET);   };
    266 static inline int disable_capability_raw(void)		{ return modify_capability(CAP_NET_RAW,   CAP_CLEAR); };
    267 static inline int enable_capability_admin(void)		{ return modify_capability(CAP_NET_ADMIN, CAP_SET);   };
    268 static inline int disable_capability_admin(void)	{ return modify_capability(CAP_NET_ADMIN, CAP_CLEAR); };
    269 #else
    270 extern int modify_capability(int);
    271 static inline int enable_capability_raw(void)		{ return modify_capability(1); };
    272 static inline int disable_capability_raw(void)		{ return modify_capability(0); };
    273 static inline int enable_capability_admin(void)		{ return modify_capability(1); };
    274 static inline int disable_capability_admin(void)	{ return modify_capability(0); };
    275 #endif
    276 extern void drop_capabilities(void);
    277 extern void android_check_security(void);
    278 
    279 extern int send_probe(void);
    280 extern int receive_error_msg(void);
    281 extern int parse_reply(struct msghdr *msg, int len, void *addr, struct timeval *);
    282 extern void install_filter(void);
    283 extern int is_ours(uint16_t id);
    284 
    285 extern int pinger(void);
    286 extern void sock_setbufs(int icmp_sock, int alloc);
    287 extern void setup(int icmp_sock);
    288 extern void main_loop(int icmp_sock, __u8 *buf, int buflen) __attribute__((noreturn));
    289 extern void finish(void) __attribute__((noreturn));
    290 extern void status(void);
    291 extern void common_options(int ch);
    292 extern int gather_statistics(__u8 *ptr, int icmplen,
    293 			     int cc, __u16 seq, int hops,
    294 			     int csfailed, struct timeval *tv, char *from,
    295 			     void (*pr_reply)(__u8 *ptr, int cc));
    296 extern void print_timestamp(void);
    297