/*
 * Copyright (c) 1999, 2010, Oracle and/or its affiliates. All rights reserved.
 */
/*
 * lib/krb5/os/hostaddr.c
 *
 * Copyright 1990,1991 by the Massachusetts Institute of Technology.
 * All Rights Reserved.
 *
 * Export of this software from the United States of America may
 *   require a specific license from the United States Government.
 *   It is the responsibility of any person or organization contemplating
 *   export to obtain such a license before exporting.
 *
 * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
 * distribute this software and its documentation for any purpose and
 * without fee is hereby granted, provided that the above copyright
 * notice appear in all copies and that both that copyright notice and
 * this permission notice appear in supporting documentation, and that
 * the name of M.I.T. not be used in advertising or publicity pertaining
 * to distribution of the software without specific, written prior
 * permission.  Furthermore if you modify this software you must label
 * your software as modified software and not distribute it in such a
 * fashion that it might be confused with the original M.I.T. software.
 * M.I.T. makes no representations about the suitability of
 * this software for any purpose.  It is provided "as is" without express
 * or implied warranty.
 *
 * This routine returns a list of krb5 addresses given a hostname.
 *
 */

#include "k5-int.h"
#include <locale.h>
#include "fake-addrinfo.h"

krb5_error_code
krb5_os_hostaddr(krb5_context context, const char *name, krb5_address ***ret_addrs)
{
    krb5_error_code 	retval;
    krb5_address 	**addrs;
    int			i, j, r;
    struct addrinfo hints, *ai, *aip;

    if (!name) {
	return KRB5_ERR_BAD_HOSTNAME;
    }

    memset (&hints, 0, sizeof (hints));
    hints.ai_flags = AI_NUMERICHOST;
    /* We don't care what kind at this point, really, but without
       this, we can get back multiple sockaddrs per address, for
       SOCK_DGRAM, SOCK_STREAM, and SOCK_RAW.  I haven't checked if
       that's what the spec indicates.  */
    hints.ai_socktype = SOCK_DGRAM;

    r = getaddrinfo (name, 0, &hints, &ai);
    if (r && AI_NUMERICHOST != 0) {
	hints.ai_flags &= ~AI_NUMERICHOST;
	r = getaddrinfo (name, 0, &hints, &ai);
    }
    if (r) {
        krb5_set_error_message(context, KRB5_ERR_BAD_HOSTNAME,
			    dgettext(TEXT_DOMAIN,
				    "Hostname cannot be canonicalized for '%s': %s"),
			    name, strerror(r));
	return KRB5_ERR_BAD_HOSTNAME;
    }

    for (i = 0, aip = ai; aip; aip = aip->ai_next) {
	switch (aip->ai_addr->sa_family) {
	case AF_INET:
#ifdef KRB5_USE_INET6
	case AF_INET6:
#endif
	    i++;
	default:
	    /* Ignore addresses of unknown families.  */
	    ;
	}
    }

    addrs = malloc ((i+1) * sizeof(*addrs));
    if (!addrs)
	return errno;

    for (j = 0; j < i + 1; j++)
	addrs[j] = 0;

    for (i = 0, aip = ai; aip; aip = aip->ai_next) {
	void *ptr;
	size_t addrlen;
	int atype;

	switch (aip->ai_addr->sa_family) {
	case AF_INET:
	    addrlen = sizeof (struct in_addr);
	    /*LINTED*/
	    ptr = &sa2sin(aip->ai_addr)->sin_addr;
	    atype = ADDRTYPE_INET;
	    break;
#ifdef KRB5_USE_INET6
	case AF_INET6:
	    addrlen = sizeof (struct in6_addr);
	    /*LINTED*/
	    ptr = &sa2sin6(aip->ai_addr)->sin6_addr;
	    atype = ADDRTYPE_INET6;
	    break;
#endif
	default:
	    continue;
	}
	addrs[i] = (krb5_address *) malloc(sizeof(krb5_address));
	if (!addrs[i]) {
	    retval = ENOMEM;
	    goto errout;
	}
	addrs[i]->magic = KV5M_ADDRESS;
	addrs[i]->addrtype = atype;
	addrs[i]->length = addrlen;
	addrs[i]->contents = malloc(addrs[i]->length);
	if (!addrs[i]->contents) {
	    retval = ENOMEM;
	    goto errout;
	}
	memcpy (addrs[i]->contents, ptr, addrs[i]->length);
	i++;
    }

    *ret_addrs = addrs;
    if (ai)
	freeaddrinfo(ai);
    return 0;

errout:
    /* Solaris Kerberos */
    if (addrs)
	krb5_free_addresses(context, addrs);
    if (ai)
	freeaddrinfo(ai);
    return retval;

}