diff options
| -rw-r--r-- | toys/pending/arp.c | 306 |
1 files changed, 306 insertions, 0 deletions
diff --git a/toys/pending/arp.c b/toys/pending/arp.c new file mode 100644 index 00000000..831facb9 --- /dev/null +++ b/toys/pending/arp.c @@ -0,0 +1,306 @@ +/* arp.c - manipulate the system ARP cache + * + * Copyright 2014 Sandeep Sharma <sandeep.jack2756@gmail.com> + * Copyright 2014 Kyungwan Han <asura321@gamil.com> + * No Standard + +USE_ARP(NEWTOY(arp, "vi:nDsdap:A:H:[+Ap][!sd]", TOYFLAG_USR|TOYFLAG_BIN)) + +config ARP + bool "arp" + default n + help + Usage: arp + [-vn] [-H HWTYPE] [-i IF] -a [HOSTNAME] + [-v] [-i IF] -d HOSTNAME [pub] + [-v] [-H HWTYPE] [-i IF] -s HOSTNAME HWADDR [temp] + [-v] [-H HWTYPE] [-i IF] -s HOSTNAME HWADDR [netmask MASK] pub + [-v] [-H HWTYPE] [-i IF] -Ds HOSTNAME IFACE [netmask MASK] pub + + Manipulate ARP cache + + -a Display (all) hosts + -s Set new ARP entry + -d Delete a specified entry + -v Verbose + -n Don't resolve names + -i IF Network interface + -D Read <hwaddr> from given device + -A,-p AF Protocol family + -H HWTYPE Hardware address type + +*/ + +#define FOR_arp +#include "toys.h" +#include <net/if_arp.h> + +GLOBALS( + char *hw_type; + char *af_type_A; + char *af_type_p; + char *interface; + + int sockfd; + char *device; +) + +struct arpreq req; //Global request structure + +struct type { + char *name; + int val; +}; + +struct type hwtype[] = { + {"ether", ARPHRD_ETHER }, + {"loop" ,ARPHRD_LOOPBACK}, + {"ppp" ,ARPHRD_PPP}, + {"infiniband" ,ARPHRD_INFINIBAND}, + {NULL, -1}, +}; + +struct type aftype[] = { + {"inet", AF_INET }, + {"inet6" ,AF_INET6}, + {"unspec" ,AF_UNSPEC}, + {NULL, -1}, +}; + +struct type flag_type[] = { + {"PERM", ATF_PERM }, + {"PUB" ,ATF_PUBL}, + {"DONTPUB" ,ATF_DONTPUB}, + {"TRAIL" ,ATF_USETRAILERS}, + {NULL, -1}, +}; + +static int get_index(struct type arr[], char *name) +{ + int i; + + for (i = 0; arr[i].name; i++) + if (!strcmp(arr[i].name, name)) break; + return arr[i].val; +} + + +void get_hw_add(char *hw_addr, char *ptr) +{ + char *p = ptr, *hw = hw_addr; + + while (*hw_addr && (p-ptr) < 6) { + int val, len = 0; + + if (*hw_addr == ':') hw_addr++; + sscanf(hw_addr, "%2x%n", &val, &len); + if (!len || len > 2) break; + hw_addr += len; + *p++ = val; + } + + if ((p-ptr) != 6 || *hw_addr) + error_exit("bad hw addr '%s'", hw); +} + +static void resolve_host(char *host, struct sockaddr *sa) +{ + struct addrinfo hints, *res = NULL; + int ret; + + memset(&hints, 0, sizeof hints); + hints.ai_family = AF_INET; + hints.ai_socktype = SOCK_STREAM; + if ((ret = getaddrinfo(host, NULL, &hints, &res))) + perror_exit("%s", gai_strerror(ret)); + + memcpy(sa, res->ai_addr, res->ai_addrlen); + freeaddrinfo(res); +} + +static void check_flags(int *i, char** argv) +{ + struct sockaddr sa; + int flag = *i, j; + struct flags { + char *name; + int or, flag; + } f[] = { + {"pub", 1 ,ATF_PUBL}, + {"priv", 0 ,~ATF_PUBL}, + {"trail", 1, ATF_USETRAILERS}, + {"temp", 0, ~ATF_PERM}, + {"dontpub",1, ATF_DONTPUB}, + }; + + for (;*argv; argv++) { + for (j = 0; j < ARRAY_LEN(f); j++) { + if (!strcmp(*argv, f[j].name)) { + (f[j].or) ?(flag |= f[j].flag):(flag &= f[j].flag); + break; + } + } + if (j > 4 && !strcmp(*argv, "netmask")) { + if (!*++argv) error_exit("NULL netmask"); + if (strcmp(*argv, "255.255.255.255")) { + resolve_host(toys.optargs[0], &sa); + memcpy(&req.arp_netmask, &sa, sizeof(sa)); + flag |= ATF_NETMASK; + } else argv++; + } else if (j > 4 && !strcmp(*argv, "dev")) { + if (!*++argv) error_exit("NULL dev"); + TT.device = *argv; + } else if (j > 4) error_exit("invalid arg"); + } + *i = flag; +} + +static int set_entry(void) +{ + int flags = 0; + + if (!toys.optargs[1]) error_exit("bad syntax"); + + if (!(toys.optflags & FLAG_D)) get_hw_add(toys.optargs[1], (char*)&req.arp_ha.sa_data); + else { + struct ifreq ifre; + + xstrncpy(ifre.ifr_name, toys.optargs[1], IFNAMSIZ); + xioctl(TT.sockfd, SIOCGIFHWADDR, &ifre); + if ((toys.optflags & FLAG_H) && (ifre.ifr_hwaddr.sa_family != ARPHRD_ETHER)) + error_exit("protocol type mismatch"); + memcpy(&req.arp_ha, &(ifre.ifr_hwaddr), sizeof(req.arp_ha)); + } + + flags = ATF_PERM | ATF_COM; + if (toys.optargs[2]) check_flags(&flags, (toys.optargs+2)); + req.arp_flags = flags; + strncpy(req.arp_dev, TT.device, sizeof(TT.device)); + xioctl(TT.sockfd, SIOCSARP, &req); + + if (toys.optflags & FLAG_v) xprintf("Entry set for %s\n", toys.optargs[0]); + return 0; +} + +static int ip_to_host(struct sockaddr *sa, int flag) +{ + int status = 0; + char hbuf[NI_MAXHOST] = {0,}, sbuf[NI_MAXSERV] = {0,}; + socklen_t len = sizeof(struct sockaddr_in6); + + *toybuf = 0; + if (!(status = getnameinfo(sa, len, hbuf, sizeof(hbuf), sbuf, + sizeof(sbuf), flag))) { + strcpy(toybuf, hbuf); + return 0; + } + return 1; +} + +static int delete_entry(void) +{ + int flags; + + flags = ATF_PERM; + if (toys.optargs[1]) check_flags(&flags, (toys.optargs+1)); + req.arp_flags = flags; + strncpy(req.arp_dev, TT.device, sizeof(TT.device)); + xioctl(TT.sockfd, SIOCDARP, &req); + + if (toys.optflags & FLAG_v) xprintf("Delete entry for %s\n", toys.optargs[0]); + return 0; +} + +void arp_main(void) +{ + struct sockaddr sa; + char ip[128], hw_addr[128], mask[12], dev[128], *host_ip = NULL, *buf; + int h_type, type, flag, i, fd, entries = 0, disp = 0; + + TT.device = ""; + memset(&sa, 0, sizeof(sa)); + memset(&req, 0, sizeof(req)); + TT.sockfd = xsocket(AF_INET, SOCK_STREAM, 0); + + if ((toys.optflags & FLAG_A) || (toys.optflags & FLAG_p)) { + if ((type = get_index(aftype, + (TT.af_type_A)?TT.af_type_A:TT.af_type_p)) != AF_INET) + error_exit((type != -1)?"only inet supported by kernel":"unknown family"); + } + + req.arp_ha.sa_family = ARPHRD_ETHER; + if (toys.optflags & FLAG_H) { + if ((type = get_index(hwtype, TT.hw_type)) != ARPHRD_ETHER) + error_exit((type != -1)?"h/w type not supported":"unknown h/w type"); + req.arp_ha.sa_family = type; + } + + if (((toys.optflags & FLAG_s) || toys.optflags & FLAG_d)) { + if (!toys.optargs[0]) error_exit("host name req"); + resolve_host(toys.optargs[0], &sa); + memcpy(&req.arp_pa, &sa, sizeof(struct sockaddr)); + } + + if ((toys.optflags & FLAG_s) && !set_entry()) return; + if ((toys.optflags & FLAG_d) && !delete_entry()) return; + + //show arp chache + fd = xopen("/proc/net/arp", O_RDONLY); + buf = get_line(fd); + free(buf); //skip first line + + if (toys.optargs[0]) { + resolve_host(toys.optargs[0], &sa); + ip_to_host(&sa, NI_NUMERICHOST); + host_ip = xstrdup(toybuf); + } + + while ((buf = get_line(fd))) { + char *host_name = "?"; + + if ((sscanf(buf, "%s 0x%x 0x%x %s %s %s\n", ip, + &h_type, &flag, hw_addr, mask, dev )) != 6) break; + entries++; + if (((toys.optflags & FLAG_H) && (get_index(hwtype, TT.hw_type) != h_type)) + || ((toys.optflags & FLAG_i) && strcmp(TT.interface, dev)) + || (toys.optargs[0] && strcmp(host_ip, ip))) { + free(buf); + continue; + } + + resolve_host(buf, &sa); + if (!(toys.optflags & FLAG_n)) { + if (!ip_to_host(&sa, NI_NAMEREQD)) host_name = toybuf; + } else ip_to_host(&sa, NI_NUMERICHOST); + + disp++; + printf("%s (%s) at" , host_name, ip); + + for (i = 0; hwtype[i].name; i++) + if (hwtype[i].val & h_type) break; + if (!hwtype[i].name) error_exit("unknown h/w type"); + + if (!(flag & ATF_COM)) { + if ((flag & ATF_PUBL)) printf(" *"); + else printf(" <incomplete>"); + } else printf(" %s [%s]", hw_addr, hwtype[i].name); + + if (flag & ATF_NETMASK) printf("netmask %s ", mask); + + for (i = 0; flag_type[i].name; i++) + if (flag_type[i].val & flag) printf(" %s", flag_type[i].name); + + printf(" on %s\n", dev); + free(buf); + } + + if (toys.optflags & FLAG_v) + xprintf("Entries: %d\tSkipped: %d\tFound: %d\n", + entries, entries - disp, disp); + if (!disp) xprintf("No Match found in %d entries\n", entries); + + if (CFG_TOYBOX_FREE) { + free(host_ip); + xclose(fd); + } +} |