figure out all local addresses, possibly even IPv6!

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@2048 ec53bebd-3082-4978-b11e-865c3cabbd6b

lib/krb5/get_addrs.c

@@ -2,13 +2,19 @@
 
 RCSID("$Id$");
 
-/*
- * Try to get all addresses, but return the one corresponding to
- * `hostname' if we fail.
- */
+#if defined(HAVE_SYS_IOCTL_H) && SunOS != 4
+#include <sys/ioctl.h>
+#endif
+#ifdef HAVE_NET_IF_H
+#include <net/if.h>
+#endif
 
-krb5_error_code
-krb5_get_all_client_addrs (krb5_addresses *res)
+#ifdef HAVE_SYS_SOCKIO_H
+#include <sys/sockio.h>
+#endif /* HAVE_SYS_SOCKIO_H */
+
+static krb5_error_code
+gethostname_fallback (krb5_addresses *res)
 {
      krb5_error_code err;
      char hostname[MAXHOSTNAMELEN];
@@ -19,20 +25,153 @@ krb5_get_all_client_addrs (krb5_addresses *res)
      hostent = gethostbyname (hostname);
      if (hostent == NULL)
 	  return errno;
-     res->number = 1;
-     res->addrs = malloc (sizeof(*res->addrs));
-     res->addrs[0].addr_type = hostent->h_addrtype;
-     res->addrs[0].address.data = NULL;
-     res->addrs[0].address.length = 0;
-     err = krb5_data_alloc (&res->addrs[0].address, hostent->h_length);
-     if (err)
-	  return err;
-     memcpy (res->addrs[0].address.data,
+     res->len = 1;
+     res->val = malloc (sizeof(*res->val));
+     if (res->val == NULL)
+	 return ENOMEM;
+     res->val[0].addr_type = hostent->h_addrtype;
+     res->val[0].address.data = NULL;
+     res->val[0].address.length = 0;
+     err = krb5_data_copy (&res->val[0].address,
 			   hostent->h_addr,
 			   hostent->h_length);
+     if (err) {
+	 free (res->val);
+	 return err;
+     }
      return 0;
 }
 
+#if defined(SIOCGIFCONF) && defined(SIOCGIFFLAGS) && defined(SIOCGIFADDR)
+static krb5_error_code
+find_all_addresses (krb5_addresses *res)
+{
+     krb5_error_code err;
+     int fd;
+     char buf[BUFSIZ];
+     struct ifreq ifreq;
+     struct ifconf ifconf;
+     int num, j;
+     char *p;
+
+     fd = socket(AF_INET, SOCK_DGRAM, 0);
+     if (fd < 0)
+	  return -1;
+
+     ifconf.ifc_len = sizeof(buf);
+     ifconf.ifc_buf = buf;
+     if(ioctl(fd, SIOCGIFCONF, &ifconf) < 0)
+	  return -1;
+     num = ifconf.ifc_len / sizeof(struct ifreq);
+     res->len = num;
+     res->val = calloc(num, sizeof(*res->val));
+     if (res->val == NULL) {
+	 close (fd);
+	 return ENOMEM;
+     }
+
+     j = 0;
+     ifreq.ifr_name[0] = '\0';
+     for (p = ifconf.ifc_buf; p < ifconf.ifc_buf + ifconf.ifc_len;) {
+          struct ifreq *ifr = (struct ifreq *)p;
+
+	  /*
+	   * This is not the real test and fails on Cray, but I've not
+	   * found any more reliable test.
+	   */
+
+#ifdef SOCKADDR_HAS_SA_LEN
+	  size_t sz = sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len;
+#else
+	  size_t sz = sizeof(*ifr);
+#endif
+	  if(strncmp(ifreq.ifr_name, ifr->ifr_name, sizeof(ifr->ifr_name))) {
+	       if(ioctl(fd, SIOCGIFFLAGS, ifr) < 0) {
+		    close (fd);
+		    free (res->val);
+		    return errno;
+	       }
+	       if (ifr->ifr_flags & IFF_UP) {
+		    if(ioctl(fd, SIOCGIFADDR, ifr) < 0) {
+			 close (fd);
+			 free (res->val);
+			 return errno;
+		    }
+		    switch (ifr->ifr_addr.sa_family) {
+#ifdef AF_INET
+		    case AF_INET: {
+			res->val[j].addr_type = AF_INET;
+			err = krb5_data_copy(&res->val[j].address,
+					     &ifr->ifr_addr,
+					     sizeof(struct sockaddr_in));
+			if (err) {
+			    close (fd);
+			    free (res->val);
+			    return ENOMEM;
+			}
+			++j;
+			break;
+		    }
+#endif /* AF_INET */
+
+/*
+ * This is not an correct nor ideal test.
+ */
+
+#if defined(AF_INET6) && defined(HAVE_NETINET_IN6_H)
+		    case AF_INET6: {
+			res->val[j].addr_type = AF_INET6;
+			err = krb5_data_copy(&res->val[j].address,
+					     &ifr->ifr_addr,
+					     sizeof(struct sockaddr_in6));
+			if (err) {
+			    close (fd);
+			    free (res->val);
+			    return ENOMEM;
+			}
+			++j;
+			break;
+		    }
+#endif /* AF_INET6 */
+		    default:
+			break;
+		    }
+	       }
+	       ifreq = *ifr;
+	  }
+	  p = p + sz;
+     }
+     close (fd);
+     if (j != num) {
+	 void *tmp;
+
+	 res->len = j;
+	 tmp = realloc (res->val, j * sizeof(*res->val));
+	 if (tmp == NULL) {
+	     free (res->val);
+	     return ENOMEM;
+	 }
+	 res->val = tmp;
+     }
+     return 0;
+}
+#endif /* SIOCGIFCONF */
+
+/*
+ * Try to get all addresses, but return the one corresponding to
+ * `hostname' if we fail.
+ */
+
+krb5_error_code
+krb5_get_all_client_addrs (krb5_addresses *res)
+{
+#if !defined(SIOCGIFCONF) || !defined(SIOCGIFFLAGS) || !defined(SIOCGIFADDR)
+    return gethostname_fallback (res);
+#else
+    return find_all_addresses (res);
+#endif /* SIOCGIFCONF */
+}
+
 /*
  * Same as above, but with the fall-back to INADDR_ANY.
  */