networking/udhcp/arpping.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * arpping.c
   4  *
   5  * Mostly stolen from: dhcpcd - DHCP client daemon
   6  * by Yoichi Hariguchi <yoichi@fore.com>
   7  *
   8  * Licensed under GPLv2, see file LICENSE in this tarball for details.
   9  */
  10
  11 #include <netinet/if_ether.h>
  12 #include <net/if_arp.h>
  13
  14 #include "common.h"
  15 #include "dhcpd.h"
  16
  17
  18 struct arpMsg {
  19         /* Ethernet header */
  20         uint8_t  h_dest[6];     /* 00 destination ether addr */
  21         uint8_t  h_source[6];   /* 06 source ether addr */
  22         uint16_t h_proto;       /* 0c packet type ID field */
  23
  24         /* ARP packet */
  25         uint16_t htype;         /* 0e hardware type (must be ARPHRD_ETHER) */
  26         uint16_t ptype;         /* 10 protocol type (must be ETH_P_IP) */
  27         uint8_t  hlen;          /* 12 hardware address length (must be 6) */
  28         uint8_t  plen;          /* 13 protocol address length (must be 4) */
  29         uint16_t operation;     /* 14 ARP opcode */
  30         uint8_t  sHaddr[6];     /* 16 sender's hardware address */
  31         uint8_t  sInaddr[4];    /* 1c sender's IP address */
  32         uint8_t  tHaddr[6];     /* 20 target's hardware address */
  33         uint8_t  tInaddr[4];    /* 26 target's IP address */
  34         uint8_t  pad[18];       /* 2a pad for min. ethernet payload (60 bytes) */
  35 } PACKED;
  36
  37 enum {
  38         ARP_MSG_SIZE = 0x2a
  39 };
  40
  41
  42 /* Returns 1 if no reply received */
  43
  44 int FAST_FUNC arpping(uint32_t test_ip, uint32_t from_ip, uint8_t *from_mac, const char *interface)
  45 {
  46         int timeout_ms;
  47         struct pollfd pfd[1];
  48 #define s (pfd[0].fd)           /* socket */
  49         int rv = 1;             /* "no reply received" yet */
  50         struct sockaddr addr;   /* for interface name */
  51         struct arpMsg arp;
  52
  53         s = socket(PF_PACKET, SOCK_PACKET, htons(ETH_P_ARP));
  54         if (s == -1) {
  55                 bb_perror_msg(bb_msg_can_not_create_raw_socket);
  56                 return -1;
  57         }
  58
  59         if (setsockopt_broadcast(s) == -1) {
  60                 bb_perror_msg("cannot enable bcast on raw socket");
  61                 goto ret;
  62         }
  63
  64         /* send arp request */
  65         memset(&arp, 0, sizeof(arp));
  66         memset(arp.h_dest, 0xff, 6);                    /* MAC DA */
  67         memcpy(arp.h_source, from_mac, 6);              /* MAC SA */
  68         arp.h_proto = htons(ETH_P_ARP);                 /* protocol type (Ethernet) */
  69         arp.htype = htons(ARPHRD_ETHER);                /* hardware type */
  70         arp.ptype = htons(ETH_P_IP);                    /* protocol type (ARP message) */
  71         arp.hlen = 6;                                   /* hardware address length */
  72         arp.plen = 4;                                   /* protocol address length */
  73         arp.operation = htons(ARPOP_REQUEST);           /* ARP op code */
  74         memcpy(arp.sHaddr, from_mac, 6);                /* source hardware address */
  75         memcpy(arp.sInaddr, &from_ip, sizeof(from_ip)); /* source IP address */
  76         /* tHaddr is zero-fiiled */                     /* target hardware address */
  77         memcpy(arp.tInaddr, &test_ip, sizeof(test_ip)); /* target IP address */
  78
  79         memset(&addr, 0, sizeof(addr));
  80         safe_strncpy(addr.sa_data, interface, sizeof(addr.sa_data));
  81         if (sendto(s, &arp, sizeof(arp), 0, &addr, sizeof(addr)) < 0) {
  82                 // TODO: error message? caller didn't expect us to fail,
  83                 // just returning 1 "no reply received" misleads it.
  84                 goto ret;
  85         }
  86
  87         /* wait for arp reply, and check it */
  88         timeout_ms = 2000;
  89         do {
  90                 int r;
  91                 unsigned prevTime = monotonic_us();
  92
  93                 pfd[0].events = POLLIN;
  94                 r = safe_poll(pfd, 1, timeout_ms);
  95                 if (r < 0)
  96                         break;
  97                 if (r) {
  98                         r = read(s, &arp, sizeof(arp));
  99                         if (r < 0)
 100                                 break;
 101                         if (r >= ARP_MSG_SIZE
 102                          && arp.operation == htons(ARPOP_REPLY)
 103                          /* don't check it: Linux doesn't return proper tHaddr (fixed in 2.6.24?) */
 104                          /* && memcmp(arp.tHaddr, from_mac, 6) == 0 */
 105                          && *((uint32_t *) arp.sInaddr) == test_ip
 106                         ) {
 107                                 rv = 0;
 108                                 break;
 109                         }
 110                 }
 111                 timeout_ms -= ((unsigned)monotonic_us() - prevTime) / 1000;
 112         } while (timeout_ms > 0);
 113
 114  ret:
 115         close(s);
 116         DEBUG("%srp reply received for this address", rv ? "No a" : "A");
 117         return rv;
 118 }