]> git.meshlink.io Git - catta/blobdiff - src/iface-pfroute.c
Merge branch 'release/0.0.1'
[catta] / src / iface-pfroute.c
diff --git a/src/iface-pfroute.c b/src/iface-pfroute.c
new file mode 100644 (file)
index 0000000..3db53a3
--- /dev/null
@@ -0,0 +1,543 @@
+/***
+  This file is part of catta.
+
+  catta is free software; you can redistribute it and/or modify it
+  under the terms of the GNU Lesser General Public License as
+  published by the Free Software Foundation; either version 2.1 of the
+  License, or (at your option) any later version.
+
+  catta is distributed in the hope that it will be useful, but WITHOUT
+  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+  or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
+  Public License for more details.
+
+  You should have received a copy of the GNU Lesser General Public
+  License along with catta; if not, write to the Free Software
+  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+  USA.
+***/
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <catta/malloc.h>
+
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/param.h>
+#ifdef HAVE_SYS_SYSCTL_H
+#include <sys/sysctl.h>
+#else
+#include <sys/sockio.h>
+#endif
+
+#include <net/route.h>
+#include <net/if.h>
+#include <net/if_dl.h>
+#include <netinet/in.h>
+
+#include <catta/log.h>
+#include "iface.h"
+#include "iface-pfroute.h"
+#include "util.h"
+
+static int bitcount (unsigned int n)
+{
+  int count=0 ;
+  while (n)
+    {
+      count++ ;
+      n &= (n - 1) ;
+    }
+  return count ;
+}
+
+static void rtm_info(struct rt_msghdr *rtm, CattaInterfaceMonitor *m)
+{
+  CattaHwInterface *hw;
+  struct if_msghdr *ifm = (struct if_msghdr *)rtm;
+  struct sockaddr_dl *sdl = (struct sockaddr_dl *)(ifm + 1);
+
+  if (sdl->sdl_family != AF_LINK)
+    return;
+
+  if (ifm->ifm_addrs == 0 && ifm->ifm_index > 0) {
+    if (!(hw = catta_interface_monitor_get_hw_interface(m, (CattaIfIndex) ifm->ifm_index)))
+      return;
+    catta_hw_interface_free(hw, 0);
+    return;
+  }
+
+  if (!(hw = catta_interface_monitor_get_hw_interface(m, ifm->ifm_index)))
+    if (!(hw = catta_hw_interface_new(m, (CattaIfIndex) ifm->ifm_index)))
+      return; /* OOM */
+
+  hw->flags_ok =
+    (ifm->ifm_flags & IFF_UP) &&
+    (!m->server->config.use_iff_running || (ifm->ifm_flags & IFF_RUNNING)) &&
+    !(ifm->ifm_flags & IFF_LOOPBACK) &&
+    (ifm->ifm_flags & IFF_MULTICAST) &&
+    (m->server->config.allow_point_to_point || !(ifm->ifm_flags & IFF_POINTOPOINT));
+
+  catta_free(hw->name);
+  hw->name = catta_strndup(sdl->sdl_data, sdl->sdl_nlen);
+
+  hw->mtu = ifm->ifm_data.ifi_mtu;
+
+  hw->mac_address_size = sdl->sdl_alen;
+  if (hw->mac_address_size > CATTA_MAC_ADDRESS_MAX)
+    hw->mac_address_size = CATTA_MAC_ADDRESS_MAX;
+
+  memcpy(hw->mac_address, sdl->sdl_data + sdl->sdl_nlen, hw->mac_address_size);
+
+/*   { */
+/*     char mac[256]; */
+/*     catta_log_debug("======\n name: %s\n index:%d\n mtu:%d\n mac:%s\n flags_ok:%d\n======",  */
+/*                 hw->name, hw->index,  */
+/*                 hw->mtu,  */
+/*                 catta_format_mac_address(mac, sizeof(mac), hw->mac_address, hw->mac_address_size), */
+/*                 hw->flags_ok); */
+/*   } */
+
+  catta_hw_interface_check_relevant(hw);
+  catta_hw_interface_update_rrs(hw, 0);
+}
+
+#define ROUNDUP(a) \
+     ((a) > 0 ? (1 + (((a) - 1) | (sizeof(long) - 1))) : sizeof(long))
+#ifdef HAVE_SYS_SYSCTL_H
+#define ADVANCE(x, n) (x += ROUNDUP((n)->sa_len))
+#else
+#define ADVANCE(x, n) (x += ROUNDUP(sizeof(struct sockaddr)))
+#endif
+
+static void rtm_addr(struct rt_msghdr *rtm, CattaInterfaceMonitor *m)
+{
+  CattaInterface *iface;
+  CattaAddress raddr;
+  int raddr_valid = 0;
+  struct ifa_msghdr *ifam = (struct ifa_msghdr *) rtm;
+  char *cp = (char *)(ifam + 1);
+  char *cp0;
+  int i;
+  int prefixlen = 0;
+  struct sockaddr *sa  =NULL;
+
+#if defined(__NetBSD__) || defined(__OpenBSD__)
+  if(((struct sockaddr *)cp)->sa_family == AF_UNSPEC)
+    ((struct sockaddr *)cp)->sa_family = AF_INET;
+#endif
+
+  for (cp0 = cp, i = 0; i < RTAX_MAX; i++) {
+    if (!(ifam->ifam_addrs & (1<<i)))
+      continue;
+    sa = (struct sockaddr *)cp;
+    if (i == RTAX_IFA)
+      break;
+#ifdef SA_SIZE
+    cp += SA_SIZE(sa);
+#else
+    ADVANCE(cp, sa);
+#endif
+  }
+
+  if(sa->sa_family != AF_INET && sa->sa_family != AF_INET6)
+    return;
+
+  if (!(iface = catta_interface_monitor_get_interface(m, (CattaIfIndex) ifam->ifam_index, catta_af_to_proto(sa->sa_family))))
+    return;
+
+  raddr.proto = catta_af_to_proto(sa->sa_family);
+
+  for(cp = cp0, i = 0; i < RTAX_MAX; i++)
+    {
+      if (!(ifam->ifam_addrs & (1<<i)))
+       continue;
+      sa = (struct sockaddr *)cp;
+#ifdef HAVE_SYS_SYSCTL_H
+      if (sa->sa_len == 0)
+       continue;
+#endif
+      switch(sa->sa_family) {
+      case AF_INET:
+       switch (1<<i) {
+       case RTA_NETMASK:
+         prefixlen = bitcount((unsigned int)((struct sockaddr_in *)sa)->sin_addr.s_addr);
+         break;
+       case RTA_IFA:
+         memcpy(raddr.data.data, &((struct sockaddr_in *)sa)->sin_addr,  sizeof(struct in_addr));
+         raddr_valid = 1;
+       default:
+         break;
+       }
+       break;
+      case AF_INET6:
+       switch (1<<i) {
+       case RTA_NETMASK:
+         prefixlen = bitcount((unsigned int)((struct sockaddr_in6 *)sa)->sin6_addr.s6_addr);
+         break;
+       case RTA_IFA:
+         memcpy(raddr.data.data, &((struct sockaddr_in6 *)sa)->sin6_addr,  sizeof(struct in6_addr));
+#ifdef __KAME__
+         if (IN6_IS_ADDR_LINKLOCAL((struct in6_addr *)raddr.data.data))
+           {
+             ((struct in6_addr *)raddr.data.data)->s6_addr[2] = 0;
+             ((struct in6_addr *)raddr.data.data)->s6_addr[3] = 0;
+           }
+#endif
+         raddr_valid = 1;
+       default:
+         break;
+       }
+       break;
+      default:
+       break;
+      }
+#ifdef SA_SIZE
+      cp += SA_SIZE(sa);
+#else
+      ADVANCE(cp, sa);
+#endif
+    }
+
+  if (!raddr_valid)
+    return;
+
+  if(rtm->rtm_type == RTM_NEWADDR)
+    {
+      CattaInterfaceAddress *addriface;
+      if (!(addriface = catta_interface_monitor_get_address(m, iface, &raddr)))
+       if (!(addriface = catta_interface_address_new(m, iface, &raddr, prefixlen)))
+         return; /* OOM */
+      if (raddr.proto == CATTA_PROTO_INET6)
+        {
+         addriface->global_scope = !(IN6_IS_ADDR_LINKLOCAL((struct in6_addr *)raddr.data.data) || IN6_IS_ADDR_MULTICAST((struct in6_addr *)raddr.data.data));
+       }
+      else
+        addriface->global_scope = 1;
+    }
+  else
+    {
+      CattaInterfaceAddress *addriface;
+      assert(rtm->rtm_type == RTM_DELADDR);
+      if (!(addriface = catta_interface_monitor_get_address(m, iface, &raddr)))
+       return;
+      catta_interface_address_free(addriface);
+    }
+
+  catta_interface_check_relevant(iface);
+  catta_interface_update_rrs(iface, 0);
+}
+
+static void parse_rtmsg(struct rt_msghdr *rtm, CattaInterfaceMonitor *m)
+{
+  assert(m);
+  assert(rtm);
+
+  if (rtm->rtm_version != RTM_VERSION) {
+    catta_log_warn("routing message version %d not understood",
+                  rtm->rtm_version);
+    return;
+  }
+
+  switch (rtm->rtm_type) {
+  case RTM_IFINFO:
+    rtm_info(rtm,m);
+    break;
+  case RTM_NEWADDR:
+  case RTM_DELADDR:
+    rtm_addr(rtm,m);
+    break;
+  default:
+    break;
+  }
+}
+
+static void socket_event(CattaWatch *w, int fd, CATTA_GCC_UNUSED CattaWatchEvent event,void *userdata) {
+  CattaInterfaceMonitor *m = (CattaInterfaceMonitor *)userdata;
+  CattaPfRoute *nl = m->osdep.pfroute;
+    ssize_t bytes;
+    char msg[2048];
+
+    assert(m);
+    assert(w);
+    assert(nl);
+    assert(fd == nl->fd);
+
+    do {
+      if((bytes = recv(nl->fd, msg, 2048, MSG_DONTWAIT)) < 0) {
+       if (errno == EAGAIN || errno == EINTR)
+         return;
+       catta_log_error(__FILE__": recv() failed: %s", strerror(errno));
+       return;
+      }
+      parse_rtmsg((struct rt_msghdr *)msg, m);
+    }
+    while (bytes > 0);
+}
+
+int catta_interface_monitor_init_osdep(CattaInterfaceMonitor *m) {
+    int fd = -1;
+
+    assert(m);
+
+    m->osdep.pfroute = NULL;
+
+    if ((fd = socket(PF_ROUTE, SOCK_RAW, AF_UNSPEC)) < 0) {
+        catta_log_error(__FILE__": socket(PF_ROUTE): %s", strerror(errno));
+        goto fail;
+    }
+
+    if (!(m->osdep.pfroute = catta_new(CattaPfRoute , 1))) {
+        catta_log_error(__FILE__": catta_new() failed.");
+        goto fail;
+    }
+    m->osdep.pfroute->fd = fd;
+
+    if (!(m->osdep.pfroute->watch = m->server->poll_api->watch_new(m->server->poll_api,
+                                                                  m->osdep.pfroute->fd,
+                                                                  CATTA_WATCH_IN,
+                                                                  socket_event,
+                                                                  m))) {
+      catta_log_error(__FILE__": Failed to create watch.");
+      goto fail;
+    }
+
+    return 0;
+
+fail:
+
+    if (m->osdep.pfroute) {
+      if (m->osdep.pfroute->watch)
+        m->server->poll_api->watch_free(m->osdep.pfroute->watch);
+
+      if (fd >= 0)
+        close(fd);
+
+      m->osdep.pfroute = NULL;
+    }
+
+    return -1;
+}
+
+void catta_interface_monitor_free_osdep(CattaInterfaceMonitor *m) {
+    assert(m);
+
+    if (m->osdep.pfroute) {
+      if (m->osdep.pfroute->watch)
+        m->server->poll_api->watch_free(m->osdep.pfroute->watch);
+
+      if (m->osdep.pfroute->fd >= 0)
+        close(m->osdep.pfroute->fd);
+
+      catta_free(m->osdep.pfroute);
+      m->osdep.pfroute = NULL;
+    }
+}
+
+#if defined (SIOCGLIFNUM) && defined(HAVE_STRUCT_LIFCONF) /* Solaris 8 and later; Sol 7? */
+/*
+ * I got this function from GNU zsbra
+ */
+static int ip6_masklen (struct in6_addr netmask) {
+    int len = 0;
+    unsigned char val;
+    unsigned char *pnt;
+
+    pnt = (unsigned char *) & netmask;
+
+    while ((*pnt == 0xff) && len < 128) {
+        len += 8;
+        pnt++;
+    }
+
+    if (len < 128) {
+        val = *pnt;
+        while (val) {
+            len++;
+            val <<= 1;
+        }
+    }
+    return len;
+}
+
+static void if_add_interface(struct lifreq *lifreq, CattaInterfaceMonitor *m, int fd, int count)
+{
+    CattaHwInterface *hw;
+    CattaAddress addr;
+    struct lifreq lifrcopy;
+    unsigned int index;
+    int flags;
+    int mtu;
+    int prefixlen;
+    CattaInterfaceAddress *addriface;
+    CattaInterface *iface;
+    struct sockaddr_in mask;
+    struct sockaddr_in6 mask6;
+    char caddr[CATTA_ADDRESS_STR_MAX];
+
+    lifrcopy = *lifreq;
+
+    if (ioctl(fd, SIOCGLIFFLAGS, &lifrcopy) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFFLAGS) %s", strerror(errno));
+        return;
+    }
+    flags = lifrcopy.lifr_flags;
+
+    if (ioctl(fd, SIOCGLIFMTU, &lifrcopy) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFMTU) %s", strerror(errno));
+        return;
+    }
+    mtu = lifrcopy.lifr_metric;
+
+    if (ioctl(fd, SIOCGLIFADDR, &lifrcopy) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFADDR) %s", strerror(errno));
+        return;
+    }
+    addr.proto = catta_af_to_proto(lifreq->lifr_addr.ss_family);
+    if (ioctl(fd, SIOCGLIFNETMASK, &lifrcopy) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFNETMASK) %s", strerror(errno));
+        return;
+    }
+    switch (lifreq->lifr_addr.ss_family) {
+        case AF_INET:
+           memcpy(addr.data.data, &((struct sockaddr_in *)&lifreq->lifr_addr)->sin_addr,  sizeof(struct in_addr));
+           memcpy(&mask, &((struct sockaddr_in *)&lifrcopy.lifr_addr)->sin_addr,  sizeof(struct in_addr));
+            prefixlen = bitcount((unsigned int) mask.sin_addr.s_addr);
+            break;
+        case AF_INET6:
+           memcpy(addr.data.data, &((struct sockaddr_in6 *)&lifreq->lifr_addr)->sin6_addr,  sizeof(struct in6_addr));
+           memcpy(&mask6, &((struct sockaddr_in6 *)&lifrcopy.lifr_addr)->sin6_addr,  sizeof(struct in6_addr));
+            prefixlen = lifrcopy.lifr_addrlen;
+            break;
+        default:
+            break;
+    }
+    index = if_nametoindex(lifreq->lifr_name);
+
+    if (!(hw = catta_interface_monitor_get_hw_interface(m, (CattaIfIndex) index))) {
+        if (!(hw = catta_hw_interface_new(m, (CattaIfIndex) index)))
+            return; /* OOM */
+
+        hw->flags_ok =
+            (flags & IFF_UP) &&
+            (!m->server->config.use_iff_running || (flags & IFF_RUNNING)) &&
+            !(flags & IFF_LOOPBACK) &&
+            (flags & IFF_MULTICAST) &&
+            (m->server->config.allow_point_to_point || !(flags & IFF_POINTOPOINT));
+        hw->name = catta_strdup(lifreq->lifr_name);
+        hw->mtu = mtu;
+        /* TODO get mac address */
+    }
+
+    if (!(iface = catta_interface_monitor_get_interface(m, (CattaIfIndex)index, addr.proto)))
+        return;
+
+    if (!(addriface = catta_interface_monitor_get_address(m, iface, &addr)))
+        if (!(addriface = catta_interface_address_new(m, iface, &addr, prefixlen)))
+            return; /* OOM */
+
+    addriface->global_scope = 1;
+
+    catta_hw_interface_check_relevant(hw);
+    catta_hw_interface_update_rrs(hw, 0);
+}
+#endif
+
+void catta_interface_monitor_sync(CattaInterfaceMonitor *m) {
+#ifndef HAVE_STRUCT_LIFCONF
+  size_t needed;
+  int mib[6];
+  char *buf, *lim, *next, count = 0;
+  struct rt_msghdr *rtm;
+
+  assert(m);
+
+ retry2:
+  mib[0] = CTL_NET;
+  mib[1] = PF_ROUTE;
+  mib[2] = 0;             /* protocol */
+  mib[3] = 0;             /* wildcard address family */
+  mib[4] = NET_RT_IFLIST;
+  mib[5] = 0;             /* no flags */
+  if (sysctl(mib, 6, NULL, &needed, NULL, 0) < 0)
+    {
+      catta_log_error("sysctl failed: %s", strerror(errno));
+      catta_log_error("route-sysctl-estimate");
+      return;
+    }
+  if ((buf = catta_malloc(needed)) == NULL)
+    {
+      catta_log_error("malloc failed in catta_interface_monitor_sync");
+      return;
+    }
+  if (sysctl(mib, 6, buf, &needed, NULL, 0) < 0) {
+    catta_log_warn("sysctl failed: %s", strerror(errno));
+    if (errno == ENOMEM && count++ < 10) {
+      catta_log_warn("Routing table grew, retrying");
+      sleep(1);
+      catta_free(buf);
+      goto retry2;
+    }
+  }
+  lim = buf + needed;
+  for (next = buf; next < lim; next += rtm->rtm_msglen) {
+    rtm = (struct rt_msghdr *)next;
+    parse_rtmsg(rtm, m);
+  }
+
+  m->list_complete = 1;
+  catta_interface_monitor_check_relevant(m);
+  catta_interface_monitor_update_rrs(m, 0);
+  catta_log_info("Network interface enumeration completed.");
+#elif defined (SIOCGLIFNUM) && defined(HAVE_STRUCT_LIFCONF) /* Solaris 8 and later; Sol 7? */
+    int sockfd;
+    int ret;
+    int n;
+    struct lifnum lifn;
+    struct lifconf lifc;
+    struct lifreq *lifreq;
+
+    if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
+        catta_log_error(__FILE__": socket(PFROUTE): %s", strerror(errno));
+        return;
+    }
+    lifc.lifc_buf = NULL;
+    lifn.lifn_family = AF_UNSPEC;
+    lifn.lifn_flags = 0;
+    if (ioctl(sockfd, SIOCGLIFNUM, &lifn) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFNUM): %s", strerror(errno));
+        goto end;
+    }
+    lifc.lifc_len = lifn.lifn_count * sizeof (struct lifreq);
+    if ((lifc.lifc_buf = catta_malloc(lifc.lifc_len)) == NULL) {
+            catta_log_error("malloc failed in catta_interface_monitor_sync");
+            goto end;
+    }
+    lifc.lifc_family = NULL;
+    lifc.lifc_flags = 0;
+    if(ioctl(sockfd, SIOCGLIFCONF, &lifc) < 0) {
+        catta_log_error(__FILE__": ioctl(SIOCGLIFCONF): %s", strerror(errno));
+        goto end;
+    }
+    lifreq = lifc.lifc_req;
+
+    for (n = 0; n < lifc.lifc_len; n += sizeof(struct lifreq)) {
+        if_add_interface(lifreq, m, sockfd, lifn.lifn_count);
+        lifreq++;
+    }
+    m->list_complete = 1;
+    catta_interface_monitor_check_relevant(m);
+    catta_interface_monitor_update_rrs(m, 0);
+end:
+    close(sockfd);
+    catta_free(lifc.lifc_buf);
+
+    catta_log_info("Network interface enumeration completed.");
+#endif
+}