1/* arping - send ARP REQUEST to a neighbour host.
2 *
3 * Copyright 2013 Sandeep Sharma <sandeep.jack2756@gmail.com>
4 * Copyright 2013 Kyungwan Han <asura321@gmail.com>
5 *
6 * No Standard.
7
8USE_ARPING(NEWTOY(arping, "<1>1s:I:w#<0c#<0AUDbqf[+AU][+Df]", TOYFLAG_USR|TOYFLAG_SBIN))
9
10config ARPING
11  bool "arping"
12  default n
13  help
14    usage: arping [-fqbDUA] [-c CNT] [-w TIMEOUT] [-I IFACE] [-s SRC_IP] DST_IP
15
16    Send ARP requests/replies
17
18    -f         Quit on first ARP reply
19    -q         Quiet
20    -b         Keep broadcasting, don't go unicast
21    -D         Duplicated address detection mode
22    -U         Unsolicited ARP mode, update your neighbors
23    -A         ARP answer mode, update your neighbors
24    -c N       Stop after sending N ARP requests
25    -w TIMEOUT Time to wait for ARP reply, seconds
26    -I IFACE   Interface to use (default eth0)
27    -s SRC_IP  Sender IP address
28    DST_IP     Target IP address
29*/
30#define FOR_arping
31#include "toys.h"
32#include <netinet/ether.h>
33#include <netpacket/packet.h>
34
35GLOBALS(
36    long count;
37    unsigned long time_out;
38    char *iface;
39    char *src_ip;
40
41    int sockfd;
42    unsigned long start, end;
43    unsigned sent_at, sent_nr, rcvd_nr, brd_sent, rcvd_req, brd_rcv,
44             unicast_flag;
45)
46
47struct sockaddr_ll src_pk, dst_pk;
48struct in_addr src_addr, dest_addr;
49extern void *mempcpy(void *dest, const void *src, size_t n);
50
51// Gets information of INTERFACE and updates IFINDEX, MAC and IP.
52static void get_interface(char *interface, int *ifindex, uint32_t *oip,
53    uint8_t *mac)
54{
55  struct ifreq req;
56  struct sockaddr_in *ip;
57  int fd = xsocket(AF_INET, SOCK_RAW, IPPROTO_RAW);
58
59  req.ifr_addr.sa_family = AF_INET;
60  xstrncpy(req.ifr_name, interface, IFNAMSIZ);
61  req.ifr_name[IFNAMSIZ-1] = '\0';
62
63  xioctl(fd, SIOCGIFFLAGS, &req);
64  if (!(req.ifr_flags & IFF_UP)) return;
65
66  if (oip) {
67    xioctl(fd, SIOCGIFADDR, &req);
68    ip = (struct sockaddr_in*) &req.ifr_addr;
69    *oip = ntohl(ip->sin_addr.s_addr);
70  }
71  if (ifindex) {
72    xioctl(fd, SIOCGIFINDEX, &req);
73    *ifindex = req.ifr_ifindex;
74  }
75  if (mac) {
76    xioctl(fd, SIOCGIFHWADDR, &req);
77    memcpy(mac, req.ifr_hwaddr.sa_data, 6);
78  }
79  xclose(fd);
80}
81
82// SIGINT handler, Print Number of Packets send or receive details.
83static void done(int sig)
84{
85  if (!(toys.optflags & FLAG_q)) {
86    xprintf("Sent %u probe(s) (%u broadcast(s))\n", TT.sent_nr, TT.brd_sent);
87    xprintf("Received %u repl%s (%u request(s), %u broadcast(s))\n",
88        TT.rcvd_nr, TT.rcvd_nr == 1 ? "y":"ies", TT.rcvd_req, TT.brd_rcv);
89  }
90  if (toys.optflags & FLAG_D) exit(!!TT.rcvd_nr);
91  //In -U mode, No reply is expected.
92  if (toys.optflags & FLAG_U) exit(EXIT_SUCCESS);
93  exit(!TT.rcvd_nr);
94}
95
96// Create and Send Packet
97static void send_packet()
98{
99  int ret;
100  unsigned char sbuf[256] = {0,};
101  struct arphdr *arp_h = (struct arphdr *) sbuf;
102  unsigned char *ptr = (unsigned char *)(arp_h + 1);
103
104  arp_h->ar_hrd = htons(ARPHRD_ETHER);
105  arp_h->ar_pro = htons(ETH_P_IP);
106  arp_h->ar_hln = src_pk.sll_halen;
107  arp_h->ar_pln = 4;
108  arp_h->ar_op = (toys.optflags & FLAG_A) ? htons(ARPOP_REPLY)
109    : htons(ARPOP_REQUEST);
110
111  ptr = mempcpy(ptr, &src_pk.sll_addr, src_pk.sll_halen);
112  ptr = mempcpy(ptr, &src_addr, 4);
113  ptr = mempcpy(ptr,
114                (toys.optflags & FLAG_A) ? &src_pk.sll_addr : &dst_pk.sll_addr,
115                src_pk.sll_halen);
116  ptr = mempcpy(ptr, &dest_addr, 4);
117
118  ret = sendto(TT.sockfd, sbuf, ptr - sbuf, 0,
119      (struct sockaddr *)&dst_pk, sizeof(dst_pk));
120  if (ret == ptr - sbuf) {
121    struct timeval tval;
122
123    gettimeofday(&tval, NULL);
124    TT.sent_at = tval.tv_sec * 1000000ULL + tval.tv_usec;
125    TT.sent_nr++;
126    if (!TT.unicast_flag) TT.brd_sent++;
127  }
128}
129
130// Receive Packet and filter with valid checks.
131static void recv_from(struct sockaddr_ll *from, int *recv_len)
132{
133  struct in_addr s_ip, d_ip;
134  struct arphdr *arp_hdr = (struct arphdr *)toybuf;
135  unsigned char *p = (unsigned char *)(arp_hdr + 1);
136
137  if (arp_hdr->ar_op != htons(ARPOP_REQUEST) &&
138      arp_hdr->ar_op != htons(ARPOP_REPLY)) return;
139
140  if (from->sll_pkttype != PACKET_HOST && from->sll_pkttype != PACKET_BROADCAST
141      && from->sll_pkttype != PACKET_MULTICAST) return;
142
143  if (arp_hdr->ar_pro != htons(ETH_P_IP) || (arp_hdr->ar_pln != 4)
144      || (arp_hdr->ar_hln != src_pk.sll_halen)
145      || (*recv_len < (int)(sizeof(*arp_hdr) + 2 * (4 + arp_hdr->ar_hln))))
146    return;
147
148  memcpy(&s_ip.s_addr, p + arp_hdr->ar_hln, 4);
149  memcpy(&d_ip.s_addr, p + arp_hdr->ar_hln + 4 + arp_hdr->ar_hln, 4);
150
151  if (dest_addr.s_addr != s_ip.s_addr) return;
152  if (toys.optflags & FLAG_D) {
153    if (src_addr.s_addr && src_addr.s_addr != d_ip.s_addr) return;
154    if (!memcmp(p, &src_pk.sll_addr, src_pk.sll_halen)) return;
155  } else if (src_addr.s_addr != d_ip.s_addr ) return;
156
157  if (!(toys.optflags & FLAG_q)) {
158    printf("%scast re%s from %s [%s]",
159        from->sll_pkttype == PACKET_HOST ? "Uni" : "Broad",
160        arp_hdr->ar_op == htons(ARPOP_REPLY) ? "ply" : "quest",
161        inet_ntoa(s_ip), ether_ntoa((struct ether_addr *) p));
162    if (TT.sent_at) {
163      unsigned delta;
164      struct timeval tval;
165
166      gettimeofday(&tval, NULL);
167      delta = (tval.tv_sec * 1000000ULL + (tval.tv_usec)) - TT.sent_at;
168      xprintf(" %u.%03ums\n", delta / 1000, delta % 1000);
169      xflush();
170    }
171  }
172  TT.rcvd_nr++;
173  if (from->sll_pkttype != PACKET_HOST) TT.brd_rcv++;
174  if (arp_hdr->ar_op == htons(ARPOP_REQUEST)) TT.rcvd_req++;
175  if (toys.optflags & FLAG_f) done(0);
176  if (!(toys.optflags & FLAG_b)) {
177    memcpy(dst_pk.sll_addr, p, src_pk.sll_halen);
178    TT.unicast_flag = 1;
179  }
180}
181
182// Alarm signal Handle, send packets in one second interval.
183static void send_signal(int sig)
184{
185  struct timeval start;
186
187  gettimeofday(&start, NULL);
188  if (!TT.start)
189    TT.end = TT.start = start.tv_sec * 1000 + start.tv_usec / 1000;
190  else TT.end = start.tv_sec*1000 + start.tv_usec / 1000;
191  if (toys.optflags & FLAG_c) {
192    if (!TT.count) done(0);
193    TT.count--;
194  }
195  if ((toys.optflags & FLAG_w) && ((TT.end - TT.start) >
196        ((TT.time_out)*1000))) done(0);
197  send_packet();
198  alarm(1);
199}
200
201void arping_main(void)
202{
203  struct ifreq ifr;
204  struct sockaddr_ll from;
205  socklen_t len;
206  int if_index, recv_len;
207
208  if (!(toys.optflags & FLAG_I)) TT.iface = "eth0";
209  TT.sockfd = xsocket(AF_PACKET, SOCK_DGRAM, 0);
210
211  memset(&ifr, 0, sizeof(ifr));
212  xstrncpy(ifr.ifr_name, TT.iface, IFNAMSIZ);
213  get_interface(TT.iface, &if_index, NULL, NULL);
214  src_pk.sll_ifindex = if_index;
215
216  xioctl(TT.sockfd, SIOCGIFFLAGS, (char*)&ifr);
217  if (!(ifr.ifr_flags & IFF_UP) && !(toys.optflags & FLAG_q))
218    error_exit("Interface \"%s\" is down", TT.iface);
219  if ((ifr.ifr_flags & (IFF_NOARP | IFF_LOOPBACK))
220      && !(toys.optflags & FLAG_q)) {
221    xprintf("Interface \"%s\" is not ARPable\n", TT.iface);
222    toys.exitval = (toys.optflags & FLAG_D) ? 0 : 2;
223    return;
224  }
225  if (!inet_aton(*toys.optargs, &dest_addr)) {
226    struct hostent *hp = gethostbyname2(*toys.optargs, AF_INET);
227
228    if (!hp) perror_exit("bad address '%s'", *toys.optargs);
229    memcpy(&dest_addr, hp->h_addr, 4);
230  }
231  if ((toys.optflags & FLAG_s) && !(inet_aton(TT.src_ip, &src_addr)))
232    perror_exit("invalid source address '%s'",TT.src_ip);
233  if (!(toys.optflags & FLAG_D) && (toys.optflags & FLAG_U)
234      && !src_addr.s_addr) src_addr = dest_addr;
235  if (!(toys.optflags & FLAG_D) || src_addr.s_addr) {
236    struct sockaddr_in saddr;
237    int p_fd = xsocket(AF_INET, SOCK_DGRAM, 0);
238
239    if (setsockopt(p_fd, SOL_SOCKET, SO_BINDTODEVICE, TT.iface,
240          strlen(TT.iface))) perror_exit("setsockopt");
241
242    memset(&saddr, 0, sizeof(saddr));
243    saddr.sin_family = AF_INET;
244    if (src_addr.s_addr) {
245      saddr.sin_addr = src_addr;
246      if (bind(p_fd, (struct sockaddr*)&saddr, sizeof(saddr)))
247        perror_exit("bind");
248    } else {
249      uint32_t oip;
250
251      saddr.sin_port = htons(1025);
252      saddr.sin_addr = dest_addr;
253      if (connect(p_fd, (struct sockaddr *) &saddr, sizeof(saddr)))
254        perror_exit("cannot connect to remote host");
255      get_interface(TT.iface, NULL, &oip, NULL);
256      src_addr.s_addr = htonl(oip);
257    }
258    xclose(p_fd);
259  }
260
261  src_pk.sll_family = AF_PACKET;
262  src_pk.sll_protocol = htons(ETH_P_ARP);
263  if (bind(TT.sockfd, (struct sockaddr *)&src_pk, sizeof(src_pk)))
264    perror_exit("bind");
265
266  socklen_t alen = sizeof(src_pk);
267  getsockname(TT.sockfd, (struct sockaddr *)&src_pk, &alen);
268  if (!src_pk.sll_halen) {
269    perror_msg("src is not arpable");
270    toys.exitval = (toys.optflags & FLAG_D) ? 0 : 2;
271    return;
272  }
273  if (!(toys.optflags & FLAG_q)) {
274    xprintf("ARPING to %s", inet_ntoa(dest_addr));
275    xprintf(" from %s via %s\n", inet_ntoa(src_addr), TT.iface);
276  }
277
278  dst_pk = src_pk;
279  //First packet always broadcasts.
280  memset(dst_pk.sll_addr, -1, dst_pk.sll_halen);
281  signal(SIGINT, done);
282  signal(SIGALRM, send_signal);
283
284  send_signal(0); // Send first Broadcast message.
285  while (1) {
286    len = sizeof(from);
287    recv_len = recvfrom(TT.sockfd, toybuf, 4096, 0,
288        (struct sockaddr *)&from, &len);
289    if (recv_len < 0) continue;
290    recv_from(&from, &recv_len);
291  }
292}
293