#include "krb5_locl.h"

RCSID("$Id$");

#if defined(HAVE_SYS_IOCTL_H) && SunOS != 4
#include <sys/ioctl.h>
#endif
#ifdef HAVE_NET_IF_H
#include <net/if.h>
#endif

#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];
     struct hostent *hostent;

     if (gethostname (hostname, sizeof(hostname)))
	  return errno;
     hostent = gethostbyname (hostname);
     if (hostent == NULL)
	  return errno;
     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 somewhat kludgy, but it seems to work. */

	  size_t sz = sizeof(*ifr);
#ifdef SOCKADDR_HAS_SA_LEN
	  if(ifr->ifr_addr.sa_len)
	      sz = sizeof(ifr->ifr_name) + ifr->ifr_addr.sa_len;
#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: {
			unsigned char addr[4];
			struct sockaddr_in *sin;
			res->val[j].addr_type = AF_INET;
			/* This is somewhat XXX */
			sin = (struct sockaddr_in*)&ifr->ifr_addr;
			memcpy(addr, 
			       &sin->sin_addr, 
			       4);
			err = krb5_data_copy(&res->val[j].address,
					     addr, 4);
			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.
 */

krb5_error_code
krb5_get_all_server_addrs ()
{
    return 0;
}