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