isc/win32/net.c

	net.c revision a687db7ce86c97d884b7ee1a68f59fcddf4d1c2a
/*
 * Copyright (C) 2004, 2005, 2007, 2008  Internet Systems Consortium, Inc. ("ISC")
 * Copyright (C) 1999-2003  Internet Software Consortium.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

/* $Id: net.c,v 1.16 2008/04/02 23:46:57 tbox Exp $ */

#include <config.h>

#include <errno.h>
#include <unistd.h>

#include <isc/log.h>
#include <isc/msgs.h>
#include <isc/net.h>
#include <isc/once.h>
#include <isc/strerror.h>
#include <isc/string.h>
#include <isc/util.h>

#if defined(ISC_PLATFORM_HAVEIPV6) && defined(ISC_PLATFORM_NEEDIN6ADDRANY)
const struct in6_addr isc_net_in6addrany = IN6ADDR_ANY_INIT;
#endif

static isc_once_t   once = ISC_ONCE_INIT;
static isc_once_t   once_ipv6only = ISC_ONCE_INIT;
static isc_once_t   once_ipv6pktinfo = ISC_ONCE_INIT;
static isc_result_t ipv4_result = ISC_R_NOTFOUND;
static isc_result_t ipv6_result = ISC_R_NOTFOUND;
static isc_result_t ipv6only_result = ISC_R_NOTFOUND;
static isc_result_t ipv6pktinfo_result = ISC_R_NOTFOUND;

void InitSockets(void);

static isc_result_t
try_proto(int domain) {
    SOCKET s;
    isc_result_t result = ISC_R_SUCCESS;
    char strbuf[ISC_STRERRORSIZE];
    int errval;

    s = socket(domain, SOCK_STREAM, IPPROTO_TCP);
    if (s == INVALID_SOCKET) {
        errval = WSAGetLastError();
        switch (errval) {
        case WSAEAFNOSUPPORT:
        case WSAEPROTONOSUPPORT:
        case WSAEINVAL:
            return (ISC_R_NOTFOUND);
        default:
            isc__strerror(errval, strbuf, sizeof(strbuf));
            UNEXPECTED_ERROR(__FILE__, __LINE__,
                     "socket() %s: %s",
                     isc_msgcat_get(isc_msgcat,
                            ISC_MSGSET_GENERAL,
                            ISC_MSG_FAILED,
                            "failed"),
                     strbuf);
            return (ISC_R_UNEXPECTED);
        }
    }

    closesocket(s);

    return (ISC_R_SUCCESS);
}

static void
initialize_action(void) {
    InitSockets();
    ipv4_result = try_proto(PF_INET);
#ifdef ISC_PLATFORM_HAVEIPV6
#ifdef WANT_IPV6
#ifdef ISC_PLATFORM_HAVEIN6PKTINFO
    ipv6_result = try_proto(PF_INET6);
#endif
#endif
#endif
}

static void
initialize(void) {
    RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
}

isc_result_t
isc_net_probeipv4(void) {
    initialize();
    return (ipv4_result);
}

isc_result_t
isc_net_probeipv6(void) {
    initialize();
    return (ipv6_result);
}

isc_result_t
isc_net_probeunix(void) {
    return (ISC_R_NOTFOUND);
}

#ifdef ISC_PLATFORM_HAVEIPV6
#ifdef WANT_IPV6
static void
try_ipv6only(void) {
#ifdef IPV6_V6ONLY
    SOCKET s;
    int on;
    char strbuf[ISC_STRERRORSIZE];
#endif
    isc_result_t result;

    result = isc_net_probeipv6();
    if (result != ISC_R_SUCCESS) {
        ipv6only_result = result;
        return;
    }

#ifndef IPV6_V6ONLY
    ipv6only_result = ISC_R_NOTFOUND;
    return;
#else
    /* check for TCP sockets */
    s = socket(PF_INET6, SOCK_STREAM, 0);
    if (s == INVALID_SOCKET) {
        isc__strerror(errno, strbuf, sizeof(strbuf));
        UNEXPECTED_ERROR(__FILE__, __LINE__,
                 "socket() %s: %s",
                 isc_msgcat_get(isc_msgcat,
                        ISC_MSGSET_GENERAL,
                        ISC_MSG_FAILED,
                        "failed"),
                 strbuf);
        ipv6only_result = ISC_R_UNEXPECTED;
        return;
    }

    on = 1;
    if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) < 0) {
        ipv6only_result = ISC_R_NOTFOUND;
        goto close;
    }

    closesocket(s);

    /* check for UDP sockets */
    s = socket(PF_INET6, SOCK_DGRAM, 0);
    if (s == INVALID_SOCKET) {
        isc__strerror(errno, strbuf, sizeof(strbuf));
        UNEXPECTED_ERROR(__FILE__, __LINE__,
                 "socket() %s: %s",
                 isc_msgcat_get(isc_msgcat,
                        ISC_MSGSET_GENERAL,
                        ISC_MSG_FAILED,
                        "failed"),
                 strbuf);
        ipv6only_result = ISC_R_UNEXPECTED;
        return;
    }

    on = 1;
    if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) < 0) {
        ipv6only_result = ISC_R_NOTFOUND;
        goto close;
    }

    ipv6only_result = ISC_R_SUCCESS;

close:
    closeocket(s);
    return;
#endif /* IPV6_V6ONLY */
}

static void
initialize_ipv6only(void) {
    RUNTIME_CHECK(isc_once_do(&once_ipv6only,
                  try_ipv6only) == ISC_R_SUCCESS);
}

static void
try_ipv6pktinfo(void) {
    int s, on;
    char strbuf[ISC_STRERRORSIZE];
    isc_result_t result;
    int optname;

    result = isc_net_probeipv6();
    if (result != ISC_R_SUCCESS) {
        ipv6pktinfo_result = result;
        return;
    }

    /* we only use this for UDP sockets */
    s = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP);
    if (s == INVALID_SOCKET) {
        isc__strerror(errno, strbuf, sizeof(strbuf));
        UNEXPECTED_ERROR(__FILE__, __LINE__,
                 "socket() %s: %s",
                 isc_msgcat_get(isc_msgcat,
                        ISC_MSGSET_GENERAL,
                        ISC_MSG_FAILED,
                        "failed"),
                 strbuf);
        ipv6pktinfo_result = ISC_R_UNEXPECTED;
        return;
    }

#ifdef IPV6_RECVPKTINFO
    optname = IPV6_RECVPKTINFO;
#else
    optname = IPV6_PKTINFO;
#endif
    on = 1;
    if (setsockopt(s, IPPROTO_IPV6, optname, &on, sizeof(on)) < 0) {
        ipv6pktinfo_result = ISC_R_NOTFOUND;
        goto close;
    }

    ipv6pktinfo_result = ISC_R_SUCCESS;

close:
    closesocket(s);
    return;
}

static void
initialize_ipv6pktinfo(void) {
    RUNTIME_CHECK(isc_once_do(&once_ipv6pktinfo,
                  try_ipv6pktinfo) == ISC_R_SUCCESS);
}
#endif /* WANT_IPV6 */
#endif /* ISC_PLATFORM_HAVEIPV6 */

isc_result_t
isc_net_probe_ipv6only(void) {
#ifdef ISC_PLATFORM_HAVEIPV6
#ifdef WANT_IPV6
    initialize_ipv6only();
#else
    ipv6only_result = ISC_R_NOTFOUND;
#endif
#endif
    return (ipv6only_result);
}

isc_result_t
isc_net_probe_ipv6pktinfo(void) {
#ifdef ISC_PLATFORM_HAVEIPV6
#ifdef WANT_IPV6
    initialize_ipv6pktinfo();
#else
    ipv6pktinfo_result = ISC_R_NOTFOUND;
#endif
#endif
    return (ipv6pktinfo_result);
}

void
isc_net_disableipv4(void) {
    initialize();
    if (ipv4_result == ISC_R_SUCCESS)
        ipv4_result = ISC_R_DISABLED;
}

void
isc_net_disableipv6(void) {
    initialize();
    if (ipv6_result == ISC_R_SUCCESS)
        ipv6_result = ISC_R_DISABLED;
}

void
isc_net_enableipv4(void) {
    initialize();
    if (ipv4_result == ISC_R_DISABLED)
        ipv4_result = ISC_R_SUCCESS;
}

void
isc_net_enableipv6(void) {
    initialize();
    if (ipv6_result == ISC_R_DISABLED)
        ipv6_result = ISC_R_SUCCESS;
}