isc/tests/sockaddr_test.c

	sockaddr_test.c revision 0c27b3fe77ac1d5094ba3521e8142d9e7973133f
/*
 * Copyright (C) 2012, 2015, 2016  Internet Systems Consortium, Inc. ("ISC")
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/* $Id$ */

/*! \file */

#include <config.h>

#include <atf-c.h>

#include <unistd.h>

#include <isc/sockaddr.h>
#include <isc/print.h>

#include "isctest.h"

/*
 * Individual unit tests
 */

ATF_TC(sockaddr_hash);
ATF_TC_HEAD(sockaddr_hash, tc) {
    atf_tc_set_md_var(tc, "descr", "sockaddr hash");
}
ATF_TC_BODY(sockaddr_hash, tc) {
    isc_result_t result;
    isc_sockaddr_t addr;
    struct in_addr in;
    struct in6_addr in6;
    unsigned int h1, h2, h3, h4;
    int ret;

    UNUSED(tc);

    result = isc_test_begin(NULL, ISC_TRUE);
    ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);

    in.s_addr = inet_addr("127.0.0.1");
    isc_sockaddr_fromin(&addr, &in, 1);
    h1 = isc_sockaddr_hash(&addr, ISC_TRUE);
    h2 = isc_sockaddr_hash(&addr, ISC_FALSE);
    ATF_CHECK(h1 != h2);

    ret = inet_pton(AF_INET6, "::ffff:127.0.0.1", &in6);
    ATF_CHECK(ret == 1);
    isc_sockaddr_fromin6(&addr, &in6, 1);
    h3 = isc_sockaddr_hash(&addr, ISC_TRUE);
    h4 = isc_sockaddr_hash(&addr, ISC_FALSE);
    ATF_CHECK(h1 == h3);
    ATF_CHECK(h2 == h4);

    isc_test_end();
}

ATF_TC(sockaddr_isnetzero);
ATF_TC_HEAD(sockaddr_isnetzero, tc) {
    atf_tc_set_md_var(tc, "descr", "sockaddr is net zero");
}
ATF_TC_BODY(sockaddr_isnetzero, tc) {
    isc_sockaddr_t addr;
    struct in_addr in;
    struct in6_addr in6;
    isc_boolean_t r;
    int ret;
    size_t i;
    struct {
        const char *string;
        isc_boolean_t expect;
    } data4[] = {
        { "0.0.0.0", ISC_TRUE },
        { "0.0.0.1", ISC_TRUE },
        { "0.0.1.0", ISC_TRUE },
        { "0.1.0.0", ISC_TRUE },
        { "1.0.0.0", ISC_FALSE },
        { "0.0.0.127", ISC_TRUE },
        { "0.0.0.255", ISC_TRUE },
        { "127.0.0.1", ISC_FALSE },
        { "255.255.255.255", ISC_FALSE },
    };
    /*
     * Mapped addresses are currently not netzero.
     */
    struct {
        const char *string;
        isc_boolean_t expect;
    } data6[] = {
        { "::ffff:0.0.0.0", ISC_FALSE },
        { "::ffff:0.0.0.1", ISC_FALSE },
        { "::ffff:0.0.0.127", ISC_FALSE },
        { "::ffff:0.0.0.255", ISC_FALSE },
        { "::ffff:127.0.0.1", ISC_FALSE },
        { "::ffff:255.255.255.255", ISC_FALSE },
    };

    UNUSED(tc);

    for (i = 0; i < sizeof(data4)/sizeof(data4[0]); i++) {
        in.s_addr = inet_addr(data4[0].string);
        isc_sockaddr_fromin(&addr, &in, 1);
        r = isc_sockaddr_isnetzero(&addr);
        ATF_CHECK_EQ(r, data4[0].expect);
    }

    for (i = 0; i < sizeof(data6)/sizeof(data6[0]); i++) {
        ret = inet_pton(AF_INET6, "::ffff:127.0.0.1", &in6);
        ATF_CHECK_EQ(ret, 1);
        isc_sockaddr_fromin6(&addr, &in6, 1);
        r = isc_sockaddr_isnetzero(&addr);
        ATF_CHECK_EQ(r, data6[0].expect);
    }
}

/*
 * Main
 */
ATF_TP_ADD_TCS(tp) {
    ATF_TP_ADD_TC(tp, sockaddr_hash);
    ATF_TP_ADD_TC(tp, sockaddr_isnetzero);

    return (atf_no_error());
}