isc/tests/print_test.c

	print_test.c revision 46ee7c3260725be660e4db900ad792ae186b33fc
/*
 * Copyright (C) 2014, 2015  Internet Systems Consortium, Inc. ("ISC")
 *
 * 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.
 */

#include <config.h>

#include <atf-c.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

/*
 * Workout if we need to force the inclusion of print.c so we can test
 * it on all platforms even if we don't include it in libisc.
 */
#include <isc/platform.h>
#if !defined(ISC_PLATFORM_NEEDPRINTF) && \
    !defined(ISC_PLATFORM_NEEDFPRINTF) && \
    !defined(ISC_PLATFORM_NEEDSPRINTF) && \
    !defined(ISC_PLATFORM_NEEDVSNPRINTF)
#define ISC__PRINT_SOURCE
#include "../print.c"
#else
#if !defined(ISC_PLATFORM_NEEDPRINTF) || \
    !defined(ISC_PLATFORM_NEEDFPRINTF) || \
    !defined(ISC_PLATFORM_NEEDSPRINTF) || \
    !defined(ISC_PLATFORM_NEEDVSNPRINTF)
#define ISC__PRINT_SOURCE
#endif
#include <isc/print.h>
#include <isc/types.h>
#include <isc/util.h>
#endif

ATF_TC(snprintf);
ATF_TC_HEAD(snprintf, tc) {
    atf_tc_set_md_var(tc, "descr", "snprintf implementation");
}
ATF_TC_BODY(snprintf, tc) {
    char buf[10000];
    isc_uint64_t ll = 8589934592ULL;
    int n;
    size_t size;

    UNUSED(tc);

    /*
     * 4294967296 <= 8589934592 < 1000000000^2 to verify fix for
     * RT#36505.
     */

    memset(buf, 0xff, sizeof(buf));
    n = isc_print_snprintf(buf, sizeof(buf), "%qu", ll);
    ATF_CHECK_EQ(n, 10);
    ATF_CHECK_STREQ(buf, "8589934592");

    memset(buf, 0xff, sizeof(buf));
    n = isc_print_snprintf(buf, sizeof(buf), "%llu", ll);
    ATF_CHECK_EQ(n, 10);
    ATF_CHECK_STREQ(buf, "8589934592");

    size = 1000;
    memset(buf, 0xff, sizeof(buf));
    n = isc_print_snprintf(buf, sizeof(buf), "%zu", size);
    ATF_CHECK_EQ(n, 4);
    ATF_CHECK_STREQ(buf, "1000");

    size = 1000;
    memset(buf, 0xff, sizeof(buf));
    n = isc_print_snprintf(buf, sizeof(buf), "%zx", size);
    ATF_CHECK_EQ(n, 3);
    ATF_CHECK_STREQ(buf, "3e8");

    size = 1000;
    memset(buf, 0xff, sizeof(buf));
    n = isc_print_snprintf(buf, sizeof(buf), "%zo", size);
    ATF_CHECK_EQ(n, 4);
    ATF_CHECK_STREQ(buf, "1750");
}

ATF_TC(fprintf);
ATF_TC_HEAD(fprintf, tc) {
    atf_tc_set_md_var(tc, "descr", "fprintf implementation");
}
ATF_TC_BODY(fprintf, tc) {
    FILE *f;
    int n;
    size_t size;
    char buf[10000];

    UNUSED(tc);

    f = fopen("fprintf.test", "w+");
    ATF_REQUIRE(f != NULL);

    size = 1000;
    n = isc_print_fprintf(f, "%zu", size);
    ATF_CHECK_EQ(n, 4);

    rewind(f);

    memset(buf, 0, sizeof(buf));
    n = fread(buf, 1, sizeof(buf), f);
    ATF_CHECK_EQ(n, 4);

    fclose(f);

    ATF_CHECK_STREQ(buf, "1000");

    if ((n > 0) && (!strcmp(buf, "1000")))
        unlink("fprintf.test");
}

/*
 * Main
 */
ATF_TP_ADD_TCS(tp) {
    ATF_TP_ADD_TC(tp, snprintf);
    ATF_TP_ADD_TC(tp, fprintf);
    return (atf_no_error());
}