ioctl(SIOCGIFINDEX) does not require clearing of entire ifr
[oweals/busybox.git] / networking / arp.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * arp.c - Manipulate the system ARP cache
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version
8  * 2 of the License, or (at your option) any later version.
9  *
10  * Author: Fred N. van Kempen, <waltje at uwalt.nl.mugnet.org>
11  * Busybox port: Paul van Gool <pvangool at mimotech.com>
12  *
13  * modified for getopt32 by Arne Bernin <arne [at] alamut.de>
14  */
15 //config:config ARP
16 //config:       bool "arp (11 kb)"
17 //config:       default y
18 //config:       select PLATFORM_LINUX
19 //config:       help
20 //config:       Manipulate the system ARP cache.
21
22 //applet:IF_ARP(APPLET(arp, BB_DIR_SBIN, BB_SUID_DROP))
23
24 //kbuild:lib-$(CONFIG_ARP) += arp.o interface.o
25
26 //usage:#define arp_trivial_usage
27 //usage:     "\n[-vn]   [-H HWTYPE] [-i IF] -a [HOSTNAME]"
28 //usage:     "\n[-v]                [-i IF] -d HOSTNAME [pub]"
29 //usage:     "\n[-v]    [-H HWTYPE] [-i IF] -s HOSTNAME HWADDR [temp]"
30 //usage:     "\n[-v]    [-H HWTYPE] [-i IF] -s HOSTNAME HWADDR [netmask MASK] pub"
31 //usage:     "\n[-v]    [-H HWTYPE] [-i IF] -Ds HOSTNAME IFACE [netmask MASK] pub"
32 //usage:#define arp_full_usage "\n\n"
33 //usage:       "Manipulate ARP cache\n"
34 //usage:       "\n      -a              Display (all) hosts"
35 //usage:       "\n      -d              Delete ARP entry"
36 //usage:       "\n      -s              Set new entry"
37 //usage:       "\n      -v              Verbose"
38 //usage:       "\n      -n              Don't resolve names"
39 //usage:       "\n      -i IF           Network interface"
40 //usage:       "\n      -D              Read HWADDR from IFACE"
41 //usage:       "\n      -A,-p AF        Protocol family"
42 //usage:       "\n      -H HWTYPE       Hardware address type"
43
44 #include "libbb.h"
45 #include "common_bufsiz.h"
46 #include "inet_common.h"
47
48 #include <arpa/inet.h>
49 #include <net/if.h>
50 #include <net/if_arp.h>
51 #include <netinet/ether.h>
52 #include <netpacket/packet.h>
53
54 #define DEBUG 0
55
56 #define DFLT_AF "inet"
57 #define DFLT_HW "ether"
58
59 enum {
60         ARP_OPT_A = (1 << 0),
61         ARP_OPT_p = (1 << 1),
62         ARP_OPT_H = (1 << 2),
63         ARP_OPT_t = (1 << 3),
64         ARP_OPT_i = (1 << 4),
65         ARP_OPT_a = (1 << 5),
66         ARP_OPT_d = (1 << 6),
67         ARP_OPT_n = (1 << 7), /* do not resolve addresses */
68         ARP_OPT_D = (1 << 8), /* HW-address is devicename */
69         ARP_OPT_s = (1 << 9),
70         ARP_OPT_v = (1 << 10) * DEBUG, /* debugging output flag */
71 };
72
73 enum {
74         sockfd = 3, /* active socket descriptor */
75 };
76
77 struct globals {
78         const struct aftype *ap; /* current address family */
79         const struct hwtype *hw; /* current hardware type */
80         const char *device;      /* current device */
81         smallint hw_set;         /* flag if hw-type was set (-H) */
82 } FIX_ALIASING;
83 #define G (*(struct globals*)bb_common_bufsiz1)
84 #define ap         (G.ap        )
85 #define hw         (G.hw        )
86 #define device     (G.device    )
87 #define hw_set     (G.hw_set    )
88 #define INIT_G() do { \
89         setup_common_bufsiz(); \
90         device = ""; \
91 } while (0)
92
93
94 static const char options[] ALIGN1 =
95         "pub\0"
96         "priv\0"
97         "temp\0"
98         "trail\0"
99         "dontpub\0"
100         "auto\0"
101         "dev\0"
102         "netmask\0";
103
104 /* Delete an entry from the ARP cache. */
105 /* Called only from main, once */
106 static int arp_del(char **args)
107 {
108         char *host;
109         struct arpreq req;
110         struct sockaddr sa;
111         int flags = 0;
112         int err;
113
114         memset(&req, 0, sizeof(req));
115
116         /* Resolve the host name. */
117         host = *args;
118         if (ap->input(host, &sa) < 0) {
119                 bb_herror_msg_and_die("%s", host);
120         }
121
122         /* If a host has more than one address, use the correct one! */
123         memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr));
124
125         if (hw_set)
126                 req.arp_ha.sa_family = hw->type;
127
128         req.arp_flags = ATF_PERM;
129         args++;
130         while (*args != NULL) {
131                 switch (index_in_strings(options, *args)) {
132                 case 0: /* "pub" */
133                         flags |= 1;
134                         args++;
135                         break;
136                 case 1: /* "priv" */
137                         flags |= 2;
138                         args++;
139                         break;
140                 case 2: /* "temp" */
141                         req.arp_flags &= ~ATF_PERM;
142                         args++;
143                         break;
144                 case 3: /* "trail" */
145                         req.arp_flags |= ATF_USETRAILERS;
146                         args++;
147                         break;
148                 case 4: /* "dontpub" */
149 #ifdef HAVE_ATF_DONTPUB
150                         req.arp_flags |= ATF_DONTPUB;
151 #else
152                         bb_error_msg("feature ATF_DONTPUB is not supported");
153 #endif
154                         args++;
155                         break;
156                 case 5: /* "auto" */
157 #ifdef HAVE_ATF_MAGIC
158                         req.arp_flags |= ATF_MAGIC;
159 #else
160                         bb_error_msg("feature ATF_MAGIC is not supported");
161 #endif
162                         args++;
163                         break;
164                 case 6: /* "dev" */
165                         if (*++args == NULL)
166                                 bb_show_usage();
167                         device = *args;
168                         args++;
169                         break;
170                 case 7: /* "netmask" */
171                         if (*++args == NULL)
172                                 bb_show_usage();
173                         if (strcmp(*args, "255.255.255.255") != 0) {
174                                 host = *args;
175                                 if (ap->input(host, &sa) < 0) {
176                                         bb_herror_msg_and_die("%s", host);
177                                 }
178                                 memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr));
179                                 req.arp_flags |= ATF_NETMASK;
180                         }
181                         args++;
182                         break;
183                 default:
184                         bb_show_usage();
185                         break;
186                 }
187         }
188         if (flags == 0)
189                 flags = 3;
190
191         strncpy_IFNAMSIZ(req.arp_dev, device);
192
193         err = -1;
194
195         /* Call the kernel. */
196         if (flags & 2) {
197                 if (option_mask32 & ARP_OPT_v)
198                         bb_error_msg("SIOCDARP(nopub)");
199                 err = ioctl(sockfd, SIOCDARP, &req);
200                 if (err < 0) {
201                         if (errno == ENXIO) {
202                                 if (flags & 1)
203                                         goto nopub;
204                                 printf("No ARP entry for %s\n", host);
205                                 return -1;
206                         }
207                         bb_perror_msg_and_die("SIOCDARP(priv)");
208                 }
209         }
210         if ((flags & 1) && err) {
211  nopub:
212                 req.arp_flags |= ATF_PUBL;
213                 if (option_mask32 & ARP_OPT_v)
214                         bb_error_msg("SIOCDARP(pub)");
215                 if (ioctl(sockfd, SIOCDARP, &req) < 0) {
216                         if (errno == ENXIO) {
217                                 printf("No ARP entry for %s\n", host);
218                                 return -1;
219                         }
220                         bb_perror_msg_and_die("SIOCDARP(pub)");
221                 }
222         }
223         return 0;
224 }
225
226 /* Get the hardware address to a specified interface name */
227 static void arp_getdevhw(char *ifname, struct sockaddr *sa)
228 {
229         struct ifreq ifr;
230         const struct hwtype *xhw;
231
232         strncpy_IFNAMSIZ(ifr.ifr_name, ifname);
233         ioctl_or_perror_and_die(sockfd, SIOCGIFHWADDR, &ifr,
234                                         "can't get HW-Address for '%s'", ifname);
235         if (hw_set && (ifr.ifr_hwaddr.sa_family != hw->type)) {
236                 bb_error_msg_and_die("protocol type mismatch");
237         }
238         memcpy(sa, &(ifr.ifr_hwaddr), sizeof(struct sockaddr));
239
240         if (option_mask32 & ARP_OPT_v) {
241                 xhw = get_hwntype(ifr.ifr_hwaddr.sa_family);
242                 if (!xhw || !xhw->print) {
243                         xhw = get_hwntype(-1);
244                 }
245                 bb_error_msg("device '%s' has HW address %s '%s'",
246                                 ifname, xhw->name,
247                                 xhw->print((unsigned char *) &ifr.ifr_hwaddr.sa_data));
248         }
249 }
250
251 /* Set an entry in the ARP cache. */
252 /* Called only from main, once */
253 static int arp_set(char **args)
254 {
255         char *host;
256         struct arpreq req;
257         struct sockaddr sa;
258         int flags;
259
260         memset(&req, 0, sizeof(req));
261
262         host = *args++;
263         if (ap->input(host, &sa) < 0) {
264                 bb_herror_msg_and_die("%s", host);
265         }
266         /* If a host has more than one address, use the correct one! */
267         memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr));
268
269         /* Fetch the hardware address. */
270         if (*args == NULL) {
271                 bb_error_msg_and_die("need hardware address");
272         }
273         if (option_mask32 & ARP_OPT_D) {
274                 arp_getdevhw(*args++, &req.arp_ha);
275         } else {
276                 if (hw->input(*args++, &req.arp_ha) < 0) {
277                         bb_error_msg_and_die("invalid hardware address");
278                 }
279         }
280
281         /* Check out any modifiers. */
282         flags = ATF_PERM | ATF_COM;
283         while (*args != NULL) {
284                 switch (index_in_strings(options, *args)) {
285                 case 0: /* "pub" */
286                         flags |= ATF_PUBL;
287                         args++;
288                         break;
289                 case 1: /* "priv" */
290                         flags &= ~ATF_PUBL;
291                         args++;
292                         break;
293                 case 2: /* "temp" */
294                         flags &= ~ATF_PERM;
295                         args++;
296                         break;
297                 case 3: /* "trail" */
298                         flags |= ATF_USETRAILERS;
299                         args++;
300                         break;
301                 case 4: /* "dontpub" */
302 #ifdef HAVE_ATF_DONTPUB
303                         flags |= ATF_DONTPUB;
304 #else
305                         bb_error_msg("feature ATF_DONTPUB is not supported");
306 #endif
307                         args++;
308                         break;
309                 case 5: /* "auto" */
310 #ifdef HAVE_ATF_MAGIC
311                         flags |= ATF_MAGIC;
312 #else
313                         bb_error_msg("feature ATF_MAGIC is not supported");
314 #endif
315                         args++;
316                         break;
317                 case 6: /* "dev" */
318                         if (*++args == NULL)
319                                 bb_show_usage();
320                         device = *args;
321                         args++;
322                         break;
323                 case 7: /* "netmask" */
324                         if (*++args == NULL)
325                                 bb_show_usage();
326                         if (strcmp(*args, "255.255.255.255") != 0) {
327                                 host = *args;
328                                 if (ap->input(host, &sa) < 0) {
329                                         bb_herror_msg_and_die("%s", host);
330                                 }
331                                 memcpy(&req.arp_netmask, &sa, sizeof(struct sockaddr));
332                                 flags |= ATF_NETMASK;
333                         }
334                         args++;
335                         break;
336                 default:
337                         bb_show_usage();
338                         break;
339                 }
340         }
341
342         /* Fill in the remainder of the request. */
343         req.arp_flags = flags;
344
345         strncpy_IFNAMSIZ(req.arp_dev, device);
346
347         /* Call the kernel. */
348         if (option_mask32 & ARP_OPT_v)
349                 bb_error_msg("SIOCSARP()");
350         xioctl(sockfd, SIOCSARP, &req);
351         return 0;
352 }
353
354
355 /* Print the contents of an ARP request block. */
356 static void
357 arp_disp(const char *name, char *ip, int type, int arp_flags,
358                 char *hwa, char *mask, char *dev)
359 {
360         static const int arp_masks[] = {
361                 ATF_PERM, ATF_PUBL,
362 #ifdef HAVE_ATF_MAGIC
363                 ATF_MAGIC,
364 #endif
365 #ifdef HAVE_ATF_DONTPUB
366                 ATF_DONTPUB,
367 #endif
368                 ATF_USETRAILERS,
369         };
370         static const char arp_labels[] ALIGN1 = "PERM\0""PUP\0"
371 #ifdef HAVE_ATF_MAGIC
372                 "AUTO\0"
373 #endif
374 #ifdef HAVE_ATF_DONTPUB
375                 "DONTPUB\0"
376 #endif
377                 "TRAIL\0"
378         ;
379
380         const struct hwtype *xhw;
381
382         xhw = get_hwntype(type);
383         if (xhw == NULL)
384                 xhw = get_hwtype(DFLT_HW);
385
386         printf("%s (%s) at ", name, ip);
387
388         if (!(arp_flags & ATF_COM)) {
389                 if (arp_flags & ATF_PUBL)
390                         printf("* ");
391                 else
392                         printf("<incomplete> ");
393         } else {
394                 printf("%s [%s] ", hwa, xhw->name);
395         }
396
397         if (arp_flags & ATF_NETMASK)
398                 printf("netmask %s ", mask);
399
400         print_flags_separated(arp_masks, arp_labels, arp_flags, " ");
401         printf(" on %s\n", dev);
402 }
403
404 /* Display the contents of the ARP cache in the kernel. */
405 /* Called only from main, once */
406 static int arp_show(char *name)
407 {
408         const char *host;
409         const char *hostname;
410         FILE *fp;
411         struct sockaddr sa;
412         int type, flags;
413         int num;
414         unsigned entries = 0, shown = 0;
415         char ip[128];
416         char hwa[128];
417         char mask[128];
418         char line[128];
419         char dev[128];
420
421         host = NULL;
422         if (name != NULL) {
423                 /* Resolve the host name. */
424                 if (ap->input(name, &sa) < 0) {
425                         bb_herror_msg_and_die("%s", name);
426                 }
427                 host = xstrdup(ap->sprint(&sa, 1));
428         }
429         fp = xfopen_for_read("/proc/net/arp");
430         /* Bypass header -- read one line */
431         fgets(line, sizeof(line), fp);
432
433         /* Read the ARP cache entries. */
434         while (fgets(line, sizeof(line), fp)) {
435
436                 mask[0] = '-'; mask[1] = '\0';
437                 dev[0] = '-'; dev[1] = '\0';
438                 /* All these strings can't overflow
439                  * because fgets above reads limited amount of data */
440                 num = sscanf(line, "%s 0x%x 0x%x %s %s %s\n",
441                                         ip, &type, &flags, hwa, mask, dev);
442                 if (num < 4)
443                         break;
444
445                 entries++;
446                 /* if the user specified hw-type differs, skip it */
447                 if (hw_set && (type != hw->type))
448                         continue;
449
450                 /* if the user specified address differs, skip it */
451                 if (host && strcmp(ip, host) != 0)
452                         continue;
453
454                 /* if the user specified device differs, skip it */
455                 if (device[0] && strcmp(dev, device) != 0)
456                         continue;
457
458                 shown++;
459                 /* This IS ugly but it works -be */
460                 hostname = "?";
461                 if (!(option_mask32 & ARP_OPT_n)) {
462                         if (ap->input(ip, &sa) < 0)
463                                 hostname = ip;
464                         else
465                                 hostname = ap->sprint(&sa, (option_mask32 & ARP_OPT_n) | 0x8000);
466                         if (strcmp(hostname, ip) == 0)
467                                 hostname = "?";
468                 }
469
470                 arp_disp(hostname, ip, type, flags, hwa, mask, dev);
471         }
472         if (option_mask32 & ARP_OPT_v)
473                 printf("Entries: %u\tSkipped: %u\tFound: %u\n",
474                                 entries, entries - shown, shown);
475
476         if (!shown) {
477                 if (hw_set || host || device[0])
478                         printf("No match found in %u entries\n", entries);
479         }
480         if (ENABLE_FEATURE_CLEAN_UP) {
481                 free((char*)host);
482                 fclose(fp);
483         }
484         return 0;
485 }
486
487 int arp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
488 int arp_main(int argc UNUSED_PARAM, char **argv)
489 {
490         const char *hw_type;
491         const char *protocol;
492         unsigned opts;
493
494         INIT_G();
495
496         xmove_fd(xsocket(AF_INET, SOCK_DGRAM, 0), sockfd);
497
498         ap = get_aftype(DFLT_AF);
499         /* Defaults are always supported */
500         //if (!ap)
501         //      bb_error_msg_and_die("%s: %s not supported", DFLT_AF, "address family");
502         hw = get_hwtype(DFLT_HW);
503         //if (!hw)
504         //      bb_error_msg_and_die("%s: %s not supported", DFLT_HW, "hardware type");
505
506         opts = getopt32(argv, "A:p:H:t:i:adnDsv", &protocol, &protocol,
507                                  &hw_type, &hw_type, &device);
508         argv += optind;
509         if (opts & (ARP_OPT_A | ARP_OPT_p)) {
510                 ap = get_aftype(protocol);
511                 if (!ap)
512                         bb_error_msg_and_die("%s: unknown %s", protocol, "address family");
513         }
514         if (opts & (ARP_OPT_H | ARP_OPT_t)) {
515                 hw = get_hwtype(hw_type);
516                 if (!hw)
517                         bb_error_msg_and_die("%s: unknown %s", hw_type, "hardware type");
518                 hw_set = 1;
519         }
520         //if (opts & ARP_OPT_i)... -i
521
522         if (ap->af != AF_INET) {
523                 bb_error_msg_and_die("%s: kernel only supports 'inet'", ap->name);
524         }
525         if (hw->alen <= 0) {
526                 bb_error_msg_and_die("%s: %s without ARP support",
527                                 hw->name, "hardware type");
528         }
529
530         /* Now see what we have to do here... */
531         if (opts & (ARP_OPT_d | ARP_OPT_s)) {
532                 if (argv[0] == NULL)
533                         bb_error_msg_and_die("need host name");
534                 if (opts & ARP_OPT_s)
535                         return arp_set(argv);
536                 return arp_del(argv);
537         }
538
539         //if (opts & ARP_OPT_a) - default
540         return arp_show(argv[0]);
541 }