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
20#include <netinet/in.h>
21#include <arpa/inet.h>
22#include <linux/errqueue.h>
23
24#ifdef ANDROID
25#include <linux/icmp.h>
26#endif
27
28#include "SNAPSHOT.h"
29
30#define	DEFDATALEN	(64 - 8)	/* default data length */
31
32#define	MAXWAIT		10		/* max seconds to wait for response */
33#define MININTERVAL	10		/* Minimal interpacket gap */
34#define MINUSERINTERVAL	200		/* Minimal allowed interval for non-root */
35
36#define SCHINT(a)	(((a) <= MININTERVAL) ? MININTERVAL : (a))
37
38#define	A(bit)		rcvd_tbl[(bit)>>3]	/* identify byte in array */
39#define	B(bit)		(1 << ((bit) & 0x07))	/* identify bit in byte */
40#define	SET(bit)	(A(bit) |= B(bit))
41#define	CLR(bit)	(A(bit) &= (~B(bit)))
42#define	TST(bit)	(A(bit) & B(bit))
43
44/* various options */
45extern int options;
46#define	F_FLOOD		0x001
47#define	F_INTERVAL	0x002
48#define	F_NUMERIC	0x004
49#define	F_PINGFILLED	0x008
50#define	F_QUIET		0x010
51#define	F_RROUTE	0x020
52#define	F_SO_DEBUG	0x040
53#define	F_SO_DONTROUTE	0x080
54#define	F_VERBOSE	0x100
55#define	F_TIMESTAMP	0x200
56#define	F_FLOWINFO	0x200
57#define	F_SOURCEROUTE	0x400
58#define	F_TCLASS	0x400
59#define	F_FLOOD_POLL	0x800
60#define	F_LATENCY	0x1000
61#define	F_AUDIBLE	0x2000
62#define	F_ADAPTIVE	0x4000
63#define	F_STRICTSOURCE	0x8000
64#define F_NOLOOP	0x10000
65#define F_TTL		0x20000
66
67/*
68 * MAX_DUP_CHK is the number of bits in received table, i.e. the maximum
69 * number of received sequence numbers we can keep track of.
70 */
71#define	MAX_DUP_CHK	0x10000
72extern int mx_dup_ck;
73extern char rcvd_tbl[MAX_DUP_CHK / 8];
74
75
76extern u_char outpack[];
77extern int maxpacket;
78
79extern int datalen;
80extern char *hostname;
81extern int uid;
82extern int ident;			/* process id to identify our packets */
83
84extern int sndbuf;
85extern int ttl;
86
87extern long npackets;			/* max packets to transmit */
88extern long nreceived;			/* # of packets we got back */
89extern long nrepeats;			/* number of duplicates */
90extern long ntransmitted;		/* sequence # for outbound packets = #sent */
91extern long nchecksum;			/* replies with bad checksum */
92extern long nerrors;			/* icmp errors */
93extern int interval;			/* interval between packets (msec) */
94extern int preload;
95extern int deadline;			/* time to die */
96extern int lingertime;
97extern struct timeval start_time, cur_time;
98extern volatile int exiting;
99extern volatile int status_snapshot;
100extern int confirm;
101extern int confirm_flag;
102extern int working_recverr;
103
104#ifndef MSG_CONFIRM
105#define MSG_CONFIRM 0
106#endif
107
108
109/* timing */
110extern int timing;			/* flag to do timing */
111extern long tmin;			/* minimum round trip time */
112extern long tmax;			/* maximum round trip time */
113extern long long tsum;			/* sum of all times, for doing average */
114extern long long tsum2;
115extern int rtt;
116extern __u16 acked;
117extern int pipesize;
118
119#define COMMON_OPTIONS \
120case 'a': case 'U': case 'c': case 'd': \
121case 'f': case 'i': case 'w': case 'l': \
122case 'S': case 'n': case 'p': case 'q': \
123case 'r': case 's': case 'v': case 'L': \
124case 't': case 'A': case 'W': case 'B':
125
126#define COMMON_OPTSTR "h?VQ:I:M:aUc:dfi:w:l:S:np:qrs:vLt:AW:B"
127
128
129/*
130 * tvsub --
131 *	Subtract 2 timeval structs:  out = out - in.  Out is assumed to
132 * be >= in.
133 */
134static inline void tvsub(struct timeval *out, struct timeval *in)
135{
136	if ((out->tv_usec -= in->tv_usec) < 0) {
137		--out->tv_sec;
138		out->tv_usec += 1000000;
139	}
140	out->tv_sec -= in->tv_sec;
141}
142
143static inline void set_signal(int signo, void (*handler)(int))
144{
145	struct sigaction sa;
146
147	memset(&sa, 0, sizeof(sa));
148
149	sa.sa_handler = (void (*)(int))handler;
150#ifdef SA_INTERRUPT
151	sa.sa_flags = SA_INTERRUPT;
152#endif
153	sigaction(signo, &sa, NULL);
154}
155
156extern int __schedule_exit(int next);
157
158static inline int schedule_exit(int next)
159{
160	if (npackets && ntransmitted >= npackets && !deadline)
161		next = __schedule_exit(next);
162	return next;
163}
164
165static inline int in_flight(void)
166{
167	__u16 diff = (__u16)ntransmitted - acked;
168	return (diff<=0x7FFF) ? diff : ntransmitted-nreceived-nerrors;
169}
170
171static inline void acknowledge(__u16 seq)
172{
173	__u16 diff = (__u16)ntransmitted - seq;
174	if (diff <= 0x7FFF) {
175		if ((int)diff+1 > pipesize)
176			pipesize = (int)diff+1;
177		if ((__s16)(seq - acked) > 0 ||
178		    (__u16)ntransmitted - acked > 0x7FFF)
179			acked = seq;
180	}
181}
182
183static inline void advance_ntransmitted(void)
184{
185	ntransmitted++;
186	/* Invalidate acked, if 16 bit seq overflows. */
187	if ((__u16)ntransmitted - acked > 0x7FFF)
188		acked = (__u16)ntransmitted + 1;
189}
190
191
192extern int send_probe(void);
193extern int receive_error_msg(void);
194extern int parse_reply(struct msghdr *msg, int len, void *addr, struct timeval *);
195extern void install_filter(void);
196
197extern int pinger(void);
198extern void sock_setbufs(int icmp_sock, int alloc);
199extern void setup(int icmp_sock);
200extern void main_loop(int icmp_sock, __u8 *buf, int buflen) __attribute__((noreturn));
201extern void finish(void) __attribute__((noreturn));
202extern void status(void);
203extern void common_options(int ch);
204extern int gather_statistics(__u8 *ptr, int cc, __u16 seq, int hops,
205			     int csfailed, struct timeval *tv, char *from);
206