* more portability work

 * support for PF_ROUTE


git-svn-id: file:///home/lennart/svn/public/avahi/trunk@805 941a03a8-eaeb-0310-b9a0-b1bbd8fe43fe

1 files changed, 300 insertions, 0 deletions

diff --git a/avahi-core/iface-pfroute.c b/avahi-core/iface-pfroute.c
new file mode 100644
index 0000000..e677fca
--- /dev/null
+++ b/avahi-core/iface-pfroute.c
@@ -0,0 +1,300 @@
+/* $Id$ */
+
+/***
+  This file is part of avahi.
+ 
+  avahi 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.
+ 
+  avahi 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 avahi; 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 <avahi-common/malloc.h>
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <err.h>
+#include <sysexits.h>
+
+#include <sys/types.h>
+#include <sys/socket.h>
+
+#include <sys/sysctl.h>
+#include <net/route.h>
+#include <net/if.h>
+#include <netinet/in.h>
+#include <net/if_dl.h>
+
+#include <arpa/inet.h>
+
+
+#include "log.h"
+#include "iface.h"
+#include "iface-pfroute.h"
+#include "util.h"
+
+static void rtm_info(struct rt_msghdr *rtm, AvahiInterfaceMonitor *m)
+{
+  AvahiHwInterface *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 = avahi_interface_monitor_get_hw_interface(m, (AvahiIfIndex) ifm->ifm_index)))
+      return;
+    avahi_hw_interface_free(hw, 0);
+    return;
+  }
+  
+  if (!(hw = avahi_interface_monitor_get_hw_interface(m, ifm->ifm_index)))
+    if (!(hw = avahi_hw_interface_new(m, (AvahiIfIndex) 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) &&
+    !(ifm->ifm_flags & IFF_POINTOPOINT);
+  
+  avahi_free(hw->name);
+  hw->name = avahi_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 > AVAHI_MAX_MAC_ADDRESS)
+    hw->mac_address_size = AVAHI_MAX_MAC_ADDRESS;
+  
+  memcpy(hw->mac_address, sdl->sdl_data + sdl->sdl_nlen, hw->mac_address_size);
+  
+  printf("======\n name: %s\n index:%d\n mtu:%d\n mac:%s\n flags_ok:%d\n======", 
+	 hw->name, hw->index, 
+	 hw->mtu, 
+	 avahi_format_mac_address(hw->mac_address, hw->mac_address_size),
+	 hw->flags_ok);
+  
+  avahi_hw_interface_check_relevant(hw);
+  avahi_hw_interface_update_rrs(hw, 0);
+}
+
+#define ROUNDUP(a) \
+     ((a) > 0 ? (1 + (((a) - 1) | (sizeof(long) - 1))) : sizeof(long))
+#define ADVANCE(x, n) (x += ROUNDUP((n)->sa_len))
+
+static void rtm_addr(struct rt_msghdr *rtm, AvahiInterfaceMonitor *m)
+{
+  AvahiInterface *iface;
+  AvahiAddress raddr;
+  int raddr_valid = 0;
+  struct ifa_msghdr *ifam = (struct ifa_msghdr *) rtm;
+  char *cp = (char *)(ifam + 1);
+  int addrs = ifam->ifam_addrs;
+  int i;
+
+  if(addrs == 0)
+    return;
+
+  for(i = 1; i; i <<=1)
+    {
+      struct sockaddr *sa;
+      sa = (struct sockaddr *)cp;
+
+      /* interface addr sockaddr present */
+      if (i & addrs && i & RTA_BRD)
+	{
+	  if(sa->sa_family == AF_INET)
+	    {
+	      struct in_addr in = ((struct sockaddr_in *)sa)->sin_addr;
+	      printf("\n%s\n",
+		     inet_ntoa(in)); 
+	    }
+	  else
+	    if(sa->sa_family == AF_INET6)
+	      {
+		/* 	    struct sockaddr_in6 sin6; */
+		
+	      }
+#ifdef SA_SIZE
+	  cp += SA_SIZE(sa);
+#else
+	  ADVANCE(cp, sa);
+#endif
+
+	}
+    }
+}
+
+static void parse_rtmsg(struct rt_msghdr *rtm, int msglen, AvahiInterfaceMonitor *m)
+{
+  assert(m);
+  assert(rtm);
+  
+  if (rtm->rtm_version != RTM_VERSION) {
+    avahi_log_warn("routing message version %d not understood\n",
+		      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 int wild_dump_interfaces(AvahiInterfaceMonitor *m)
+{
+  size_t needed;
+  int mib[6];
+  char *buf, *lim, *next, count = 0;
+  struct rt_msghdr *rtm;
+
+ 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)
+    err(EX_OSERR, "route-sysctl-estimate");
+  if ((buf = malloc(needed)) == NULL)
+    errx(EX_OSERR, "malloc failed");
+  if (sysctl(mib, 6, buf, &needed, NULL, 0) < 0) {
+    if (errno == ENOMEM && count++ < 10) {
+      warnx("Routing table grew, retrying");
+      sleep(1);
+      free(buf);
+      goto retry2;
+    }
+    err(EX_OSERR, "actual retrieval of interface table");
+  }
+  lim = buf + needed;
+  for (next = buf; next < lim; next += rtm->rtm_msglen) {
+    rtm = (struct rt_msghdr *)next;
+    parse_rtmsg(rtm, rtm->rtm_msglen, m);
+  }
+  return 0;
+}
+
+static void socket_event(AvahiWatch *w, int fd, AvahiWatchEvent event,void *userdata) {
+  AvahiInterfaceMonitor *m = (AvahiInterfaceMonitor *)userdata;
+  AvahiPfRoute *nl = m->osdep.pfroute;
+    ssize_t bytes;
+    char msg[2048];
+
+    assert(m);
+    assert(w);
+    assert(nl);
+    assert(fd == nl->fd);
+
+    do {
+      time_t now = time(NULL);
+      if((bytes = recv(nl->fd, msg, 2048, MSG_DONTWAIT)) < 0) {
+	if (errno == EAGAIN || errno == EINTR)
+	  return;
+	avahi_log_error(__FILE__": recv() failed: %s", strerror(errno));
+	return;
+      }
+
+      (void) printf("\ngot message of size %d on %s", bytes, ctime(&now));
+      printf("parse_rtmsg\n");
+      parse_rtmsg((struct rt_msghdr *)msg, bytes ,m);
+    }
+    while (bytes > 0);
+}
+
+int avahi_interface_monitor_init_osdep(AvahiInterfaceMonitor *m) {
+    int fd = -1;
+    m->osdep.pfroute = NULL;
+
+    assert(m);
+
+    if ((fd = socket(PF_ROUTE, SOCK_RAW, AF_UNSPEC)) < 0) {
+        avahi_log_error(__FILE__": socket(PF_ROUTE): %s", strerror(errno));
+        goto fail;
+    }
+
+    if (!(m->osdep.pfroute = avahi_new(AvahiPfRoute , 1))) {
+        avahi_log_error(__FILE__": avahi_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, 
+								   AVAHI_WATCH_IN, 
+								   socket_event, 
+								   m))) {
+      avahi_log_error(__FILE__": Failed to create watch.");
+      goto fail;
+    }
+    
+    printf("avahi_interface_monitor_init_osdep\n");
+    return 0;
+
+fail:
+
+    if (m->osdep.pfroute) {
+      if (m->osdep.pfroute->watch)
+        m->osdep.pfroute->poll_api->watch_free(m->osdep.pfroute->watch);
+      
+      if (fd >= 0)
+        close(fd);
+      
+      m->osdep.pfroute = NULL;
+    }
+
+    return -1;
+}
+
+void avahi_interface_monitor_free_osdep(AvahiInterfaceMonitor *m) {
+    assert(m);
+
+    if (m->osdep.pfroute) {
+      if (m->osdep.pfroute->watch)
+        m->osdep.pfroute->poll_api->watch_free(m->osdep.pfroute->watch);
+      
+      if (m->osdep.pfroute->fd >= 0)
+        close(m->osdep.pfroute->fd);
+      
+      m->osdep.pfroute = NULL;
+    }
+}
+
+void avahi_interface_monitor_sync(AvahiInterfaceMonitor *m) {
+    assert(m);
+
+    wild_dump_interfaces(m);
+
+}
+