dev.c

   1 #include <string.h>
   2 #include <errno.h>
   3 #include <sys/ioctl.h>
   4 #include <sys/types.h>
   5 #include <sys/socket.h>
   6 #include <linux/if_arp.h>
   7 #include <ifaddrs.h>
   8
   9 #include "dev.h"
  10 #include "str.h"
  11 #include "sock.h"
  12 #include "die.h"
  13 #include "link.h"
  14 #include "built_in.h"
  15
  16 int device_ifindex_get(const char *ifname)
  17 {
  18         int ret, sock, index;
  19         struct ifreq ifr;
  20
  21         if (!strncmp("any", ifname, strlen("any")))
  22                 return 0;
  23
  24         sock = af_socket(AF_INET);
  25
  26         memset(&ifr, 0, sizeof(ifr));
  27         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
  28
  29         ret = ioctl(sock, SIOCGIFINDEX, &ifr);
  30         if (ret)
  31                 index = -1;
  32         else
  33                 index = ifr.ifr_ifindex;
  34
  35         close(sock);
  36
  37         return index;
  38 }
  39
  40 int device_ifindex(const char *ifname)
  41 {
  42         int index = device_ifindex_get(ifname);
  43
  44         if (unlikely(index < 0))
  45                 panic("Cannot get ifindex from device!\n");
  46
  47         return index;
  48 }
  49
  50 int device_type(const char *ifname)
  51 {
  52         int ret, sock, type;
  53         struct ifreq ifr;
  54
  55         if (!strncmp("any", ifname, strlen("any")))
  56                 return ARPHRD_ETHER;
  57
  58         sock = af_socket(AF_INET);
  59
  60         memset(&ifr, 0, sizeof(ifr));
  61         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
  62
  63         ret = ioctl(sock, SIOCGIFHWADDR, &ifr);
  64         if (unlikely(ret))
  65                 panic("Cannot get iftype from device!\n");
  66
  67         /* dev->type */
  68         type = ifr.ifr_hwaddr.sa_family;
  69         close(sock);
  70
  71         return type;
  72 }
  73
  74 static int __device_address6(const char *ifname, struct sockaddr_storage *ss)
  75 {
  76         int ret, family, found = -EINVAL;
  77         struct ifaddrs *ifaddr, *ifa;
  78
  79         ret = getifaddrs(&ifaddr);
  80         if (unlikely(ret < 0))
  81                 panic("Cannot get device addresses for IPv6!\n");
  82
  83         for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
  84                 family = ifa->ifa_addr->sa_family;
  85                 if (family != AF_INET6)
  86                         continue;
  87                 if (strcmp(ifa->ifa_name, ifname))
  88                         continue;
  89
  90                 memcpy(ss, ifa->ifa_addr, sizeof(*ss));
  91                 found = 0;
  92                 break;
  93         }
  94
  95         freeifaddrs(ifaddr);
  96         return found;
  97 }
  98
  99 int device_address(const char *ifname, int af, struct sockaddr_storage *ss)
 100 {
 101         int ret, sock;
 102         struct ifreq ifr;
 103
 104         if (unlikely(!ss))
 105                 return -EINVAL;
 106         if (!strncmp("any", ifname, strlen("any")))
 107                 return -EINVAL;
 108         if (af == AF_INET6)
 109                 return __device_address6(ifname, ss);
 110
 111         sock = af_socket(af);
 112
 113         memset(&ifr, 0, sizeof(ifr));
 114         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 115
 116         ifr.ifr_addr.sa_family = af;
 117
 118         ret = ioctl(sock, SIOCGIFADDR, &ifr);
 119         if (likely(!ret))
 120                 memcpy(ss, &ifr.ifr_addr, sizeof(ifr.ifr_addr));
 121
 122         close(sock);
 123         return ret;
 124 }
 125
 126 size_t device_mtu(const char *ifname)
 127 {
 128         size_t mtu = 0;
 129         int ret, sock;
 130         struct ifreq ifr;
 131
 132         sock = af_socket(AF_INET);
 133
 134         memset(&ifr, 0, sizeof(ifr));
 135         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 136
 137         ret = ioctl(sock, SIOCGIFMTU, &ifr);
 138         if (likely(!ret))
 139                 mtu = ifr.ifr_mtu;
 140
 141         close(sock);
 142         return mtu;
 143 }
 144
 145 short device_get_flags(const char *ifname)
 146 {
 147         short flags = 0;
 148         int ret, sock;
 149         struct ifreq ifr;
 150
 151         sock = af_socket(AF_INET);
 152
 153         memset(&ifr, 0, sizeof(ifr));
 154         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 155
 156         ret = ioctl(sock, SIOCGIFFLAGS, &ifr);
 157         if (likely(!ret))
 158                 flags = ifr.ifr_flags;
 159
 160         close(sock);
 161         return flags;
 162 }
 163
 164 void device_set_flags(const char *ifname, const short flags)
 165 {
 166         int ret, sock;
 167         struct ifreq ifr;
 168
 169         sock = af_socket(AF_INET);
 170
 171         memset(&ifr, 0, sizeof(ifr));
 172         strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 173         ifr.ifr_flags = flags;
 174
 175         ret = ioctl(sock, SIOCSIFFLAGS, &ifr);
 176         if (unlikely(ret < 0))
 177                 panic("Cannot set NIC flags!\n");
 178
 179         close(sock);
 180 }
 181
 182 int device_up_and_running(const char *ifname)
 183 {
 184         if (!ifname)
 185                 return -EINVAL;
 186         if (!strncmp("any", ifname, strlen("any")))
 187                 return 1;
 188
 189         return (device_get_flags(ifname) &
 190                 (IFF_UP | IFF_RUNNING)) ==
 191                 (IFF_UP | IFF_RUNNING);
 192 }
 193
 194 u32 device_bitrate(const char *ifname)
 195 {
 196         u32 scopper, swireless;
 197
 198         scopper   = ethtool_bitrate(ifname);
 199         swireless = wireless_bitrate(ifname);
 200
 201         return scopper ? : swireless;
 202 }
 203
 204 short device_enter_promiscuous_mode(const char *ifname)
 205 {
 206         short ifflags;
 207
 208         if (!strncmp("any", ifname, strlen("any")))
 209                 return 0;
 210
 211         ifflags = device_get_flags(ifname);
 212         device_set_flags(ifname, ifflags | IFF_PROMISC);
 213
 214         return ifflags;
 215 }
 216
 217 void device_leave_promiscuous_mode(const char *ifname, short oldflags)
 218 {
 219         if (!strncmp("any", ifname, strlen("any")))
 220                 return;
 221
 222         device_set_flags(ifname, oldflags);
 223 }