libcollectdclient: Implement if_nametoindex() for Windows.

[collectd.git] / src / libcollectdclient / network.c

diff --git a/src/libcollectdclient/network.c b/src/libcollectdclient/network.c
index 02a8e96..7105774 100644 (file)
--- a/src/libcollectdclient/network.c
+++ b/src/libcollectdclient/network.c
@@ -1,6 +1,7 @@
 /**
  * collectd - src/libcollectdclient/network.c
- * Copyright (C) 2005-2012  Florian octo Forster
+ * Copyright (C) 2005-2014  Florian Forster
+ * Copyright (C) 2010       Max Henkel
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -21,9 +22,23 @@
  * DEALINGS IN THE SOFTWARE.
  *
  * Authors:
- *   Florian octo Forster <octo at collectd.org>
+ *   Florian Forster <octo at collectd.org>
+ *   Max Henkel <henkel at gmx.at>
  **/
 
+#if WIN32
+
+#include <winsock2.h>
+#include <wspiapi.h>
+#include <iphlpapi.h>
+#include <windows.h>
+#include <io.h>
+#include <assert.h>
+
+#else
+
+#include "collectd.h"
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <unistd.h>
@@ -35,6 +50,16 @@
 #include <sys/socket.h>
 #include <netdb.h>
 
+#if HAVE_NETINET_IN_H
+# include <netinet/in.h>
+#endif
+
+#if HAVE_NET_IF_H
+# include <net/if.h>
+#endif
+
+#endif /* !WIN32 */
+
 #include "collectd/network.h"
 #include "collectd/network_buffer.h"
 
@@ -152,7 +177,7 @@ static int server_open_socket (lcc_server_t *srv) /* {{{ */
         optname = IP_TTL;
 
       setsockopt (srv->fd, IPPROTO_IP, optname,
-          &srv->ttl,
+          (void *) &srv->ttl,
           sizeof (srv->ttl));
     }
     else if (ai_ptr->ai_family == AF_INET6)
@@ -167,7 +192,7 @@ static int server_open_socket (lcc_server_t *srv) /* {{{ */
         optname = IPV6_UNICAST_HOPS;
 
       setsockopt (srv->fd, IPPROTO_IPV6, optname,
-          &srv->ttl,
+          (void *) &srv->ttl,
           sizeof (srv->ttl));
     }
 
@@ -373,6 +398,118 @@ int lcc_server_set_ttl (lcc_server_t *srv, uint8_t ttl) /* {{{ */
   return (0);
 } /* }}} int lcc_server_set_ttl */
 
+#if WIN32
+static unsigned if_nametoindex (LPCSTR pszIfName) /* {{{ */
+{
+  IP_ADAPTER_ADDRESSES *pAddrList;
+  IP_ADAPTER_ADDRESSES *pAddrPtr;
+  BYTE bBuffer[8192];
+  ULONG dwSize;
+  ULONG dwStatus;
+  unsigned dwIndex = 0;
+
+  dwSize = (ULONG) sizeof (bBuffer);
+  pAddrList = (IP_ADAPTER_ADDRESSES *) &bBuffer[0];
+
+  dwStatus = GetAdaptersAddresses (
+      /* Family           = */ AF_UNSPEC,
+      /* Flags            = */ GAA_FLAG_SKIP_ANYCAST,
+      /* Reserved         = */ NULL,
+      /* AdapterAddresses = */ pAddrList,
+      /* SizePointer      = */ &dwSize);
+  if (dwStatus != ERROR_SUCCESS)
+    return (0);
+
+  for (pAddrPtr = pAddrList;
+      pAddrPtr != NULL;
+      pAddrPtr = pAddrPtr->Next)
+  {
+    if (strcmp (pszIfName, pAddrPtr->AdapterName) != 0)
+      continue;
+
+    dwIndex = (unsigned) pAddrPtr->IfIndex;
+    break;
+  }
+
+  return (dwIndex);
+} /* }}} unsigned if_nametoindex */
+#endif /* WIN32 */
+
+int lcc_server_set_interface (lcc_server_t *srv, char const *ifname) /* {{{ */
+{
+  int if_index;
+  int status;
+
+  if ((srv == NULL) || (ifname == NULL))
+    return (EINVAL);
+
+  if_index = if_nametoindex (ifname);
+  if (if_index == 0)
+    return (ENOENT);
+
+  /* IPv4 multicast */
+  if (srv->sa->sa_family == AF_INET)
+  {
+    struct sockaddr_in *addr = (struct sockaddr_in *) srv->sa;
+
+    if (IN_MULTICAST (ntohl (addr->sin_addr.s_addr)))
+    {
+#if HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
+      /* If possible, use the "ip_mreqn" structure which has
+       * an "interface index" member. Using the interface
+       * index is preferred here, because of its similarity
+       * to the way IPv6 handles this. Unfortunately, it
+       * appears not to be portable. */
+      struct ip_mreqn mreq;
+
+      memset (&mreq, 0, sizeof (mreq));
+      mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
+      mreq.imr_address.s_addr = ntohl (INADDR_ANY);
+      mreq.imr_ifindex = if_index;
+#else
+      struct ip_mreq mreq;
+
+      memset (&mreq, 0, sizeof (mreq));
+      mreq.imr_multiaddr.s_addr = addr->sin_addr.s_addr;
+      mreq.imr_interface.s_addr = ntohl (INADDR_ANY);
+#endif
+
+      status = setsockopt (srv->fd, IPPROTO_IP, IP_MULTICAST_IF,
+          (void *) &mreq, sizeof (mreq));
+      if (status != 0)
+        return (status);
+
+      return (0);
+    }
+  }
+
+  /* IPv6 multicast */
+  if (srv->sa->sa_family == AF_INET6)
+  {
+    struct sockaddr_in6 *addr = (struct sockaddr_in6 *) srv->sa;
+
+    if (IN6_IS_ADDR_MULTICAST (&addr->sin6_addr))
+    {
+      status = setsockopt (srv->fd, IPPROTO_IPV6, IPV6_MULTICAST_IF,
+          (void *) &if_index, sizeof (if_index));
+      if (status != 0)
+        return (status);
+
+      return (0);
+    }
+  }
+
+  /* else: Not a multicast interface. */
+#if defined(SO_BINDTODEVICE)
+  status = setsockopt (srv->fd, SOL_SOCKET, SO_BINDTODEVICE,
+      ifname, strlen (ifname) + 1);
+  if (status != 0)
+    return (-1);
+#endif
+
+  return (0);
+} /* }}} int lcc_server_set_interface */
+
 int lcc_server_set_security_level (lcc_server_t *srv, /* {{{ */
     lcc_security_level_t level,
     const char *username, const char *password)