ping_common.h

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