/* $NetBSD: t_bitstring.c,v 1.4 2012/03/25 06:54:04 joerg Exp $ */

/*-
 * Copyright (c) 1993, 2008, 2010 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <assert.h>
#include <bitstring.h>
#include <stdio.h>
#include <stdlib.h>

#include <atf-c.h>

static void
clearbits(bitstr_t *b, int n)
{
	int i = bitstr_size(n);

	while(i--)
		*(b + i) = 0;
}

static void
printbits(FILE *file, bitstr_t *b, int n)
{
	int i;
	int jc, js;

	bit_ffc(b, n, &jc);
	bit_ffs(b, n, &js);

	(void) fprintf(file, "%3d %3d ", jc, js);

	for (i=0; i < n; i++) {
		(void) fprintf(file, "%c", (bit_test(b, i) ? '1' : '0'));
	}

	(void) fprintf(file, "%c", '\n');
}

static void
calculate_data(FILE *file, const int test_length)
{
	int i;
	bitstr_t *bs;

	assert(test_length >= 4);

	(void) fprintf(file, "Testing with TEST_LENGTH = %d\n\n", test_length);

	(void) fprintf(file, "test _bit_byte, _bit_mask, and bitstr_size\n");
	(void) fprintf(file, "  i   _bit_byte(i)   _bit_mask(i) bitstr_size(i)\n");

	for (i=0; i < test_length; i++) {
		(void) fprintf(file, "%3d%15u%15u%15zu\n",
			i, _bit_byte(i), _bit_mask(i), bitstr_size(i));
	}

	bs = bit_alloc(test_length);
	clearbits(bs, test_length);
	(void) fprintf(file, "\ntest bit_alloc, clearbits, bit_ffc, bit_ffs\n");
	(void) fprintf(file, "be:   0  -1 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", '0');
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);

	(void) fprintf(file, "\ntest bit_set\n");
	for (i=0; i < test_length; i+=3)
		bit_set(bs, i);
	(void) fprintf(file, "be:   1   0 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", "100"[i % 3]);
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);

	(void) fprintf(file, "\ntest bit_clear\n");
	for (i=0; i < test_length; i+=6)
		bit_clear(bs, i);
	(void) fprintf(file, "be:   0   3 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", "000100"[i % 6]);
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);

	(void) fprintf(file, "\ntest bit_test using previous bitstring\n");
	(void) fprintf(file, "  i    bit_test(i)\n");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%3d%15d\n", i, bit_test(bs, i));

	clearbits(bs, test_length);
	(void) fprintf(file, "\ntest clearbits\n");
	(void) fprintf(file, "be:   0  -1 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", '0');
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);

	(void) fprintf(file, "\ntest bit_nset and bit_nclear\n");
	bit_nset(bs, 1, test_length - 2);
	(void) fprintf(file, "be:   0   1 0");
	for (i=0; i < test_length - 2; i++)
		(void) fprintf(file, "%c", '1');
	(void) fprintf(file, "0\nis: ");
	printbits(file, bs, test_length);

	bit_nclear(bs, 2, test_length - 3);
	(void) fprintf(file, "be:   0   1 01");
	for (i=0; i < test_length - 4; i++)
		(void) fprintf(file, "%c", '0');
	(void) fprintf(file, "10\nis: ");
	printbits(file, bs, test_length);

	bit_nclear(bs, 0, test_length - 1);
	(void) fprintf(file, "be:   0  -1 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", '0');
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);
	bit_nset(bs, 0, test_length - 2);
	(void) fprintf(file, "be: %3d   0 ",test_length - 1);
	for (i=0; i < test_length - 1; i++)
		(void) fprintf(file, "%c", '1');
	fprintf(file, "%c", '0');
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);
	bit_nclear(bs, 0, test_length - 1);
	(void) fprintf(file, "be:   0  -1 ");
	for (i=0; i < test_length; i++)
		(void) fprintf(file, "%c", '0');
	(void) fprintf(file, "\nis: ");
	printbits(file, bs, test_length);

	(void) fprintf(file, "\n");
	(void) fprintf(file, "first 1 bit should move right 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nclear(bs, 0, test_length - 1);
		bit_nset(bs, i, test_length - 1);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) fprintf(file, "\n");
	(void) fprintf(file, "first 0 bit should move right 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nset(bs, 0, test_length - 1);
		bit_nclear(bs, i, test_length - 1);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) fprintf(file, "\n");
	(void) fprintf(file, "first 0 bit should move left 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nclear(bs, 0, test_length - 1);
		bit_nset(bs, 0, test_length - 1 - i);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) fprintf(file, "\n");
	(void) fprintf(file, "first 1 bit should move left 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nset(bs, 0, test_length - 1);
		bit_nclear(bs, 0, test_length - 1 - i);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) fprintf(file, "\n");
	(void) fprintf(file, "0 bit should move right 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nset(bs, 0, test_length - 1);
		bit_nclear(bs, i, i);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) fprintf(file, "\n");
	(void) fprintf(file, "1 bit should move right 1 position each line\n");
	for (i=0; i < test_length; i++) {
		bit_nclear(bs, 0, test_length - 1);
		bit_nset(bs, i, i);
		(void) fprintf(file, "%3d ", i); printbits(file, bs, test_length);
	}

	(void) free(bs);
}

static void
one_check(const atf_tc_t *tc, const int test_length)
{
	FILE *out;
	char command[1024];

	ATF_REQUIRE((out = fopen("out", "w")) != NULL);
	calculate_data(out, test_length);
	fclose(out);

	/* XXX The following is a huge hack that was added to simplify the
	 * conversion of these tests from src/regress/ to src/tests/.  The
	 * tests in this file should be checking their own results, without
	 * having to resort to external data files. */
	snprintf(command, sizeof(command), "diff -u %s/d_bitstring_%d.out out",
	    atf_tc_get_config_var(tc, "srcdir"), test_length);
	if (system(command) != EXIT_SUCCESS)
		atf_tc_fail("Test failed; see output for details");
}

ATF_TC(bits_8);
ATF_TC_HEAD(bits_8, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 8-bit long bitstrings");
}
ATF_TC_BODY(bits_8, tc)
{
	one_check(tc, 8);
}

ATF_TC(bits_27);
ATF_TC_HEAD(bits_27, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 27-bit long bitstrings");
}
ATF_TC_BODY(bits_27, tc)
{
	one_check(tc, 27);
}

ATF_TC(bits_32);
ATF_TC_HEAD(bits_32, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 32-bit long bitstrings");
}
ATF_TC_BODY(bits_32, tc)
{
	one_check(tc, 32);
}

ATF_TC(bits_49);
ATF_TC_HEAD(bits_49, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 49-bit long bitstrings");
}
ATF_TC_BODY(bits_49, tc)
{
	one_check(tc, 49);
}

ATF_TC(bits_64);
ATF_TC_HEAD(bits_64, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 64-bit long bitstrings");
}
ATF_TC_BODY(bits_64, tc)
{
	one_check(tc, 64);
}

ATF_TC(bits_67);
ATF_TC_HEAD(bits_67, tc)
{
	atf_tc_set_md_var(tc, "descr", "Checks 67-bit long bitstrings");
}
ATF_TC_BODY(bits_67, tc)
{
	one_check(tc, 67);
}

ATF_TP_ADD_TCS(tp)
{

	ATF_TP_ADD_TC(tp, bits_8);
	ATF_TP_ADD_TC(tp, bits_27);
	ATF_TP_ADD_TC(tp, bits_32);
	ATF_TP_ADD_TC(tp, bits_49);
	ATF_TP_ADD_TC(tp, bits_64);
	ATF_TP_ADD_TC(tp, bits_67);

	return atf_no_error();
}