krb5/os/hostaddr.c

	hostaddr.c revision 159d09a20817016f09b3ea28d1bdada4a336bb91
/*
 * 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 "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)
    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:
    if (addrs) {
    for (i = 0; addrs[i]; i++) {
        free (addrs[i]->contents);
        free (addrs[i]);
    }
    krb5_free_addresses(context, addrs);
    }
    if (ai)
    freeaddrinfo(ai);
    return retval;

}