#include <check.h>
#include <stdint.h>
#include <bitfield/bitfield.h>

START_TEST (test_large_bitmask)
{
    uint64_t result = bitmask(32);
    fail_if(result != 0xffffffff);
}
END_TEST

START_TEST (test_one_bit_not_swapped)
{
    uint64_t data = 0x80;
    uint64_t result = eightbyte_get_bitfield(data, 0, 1, false);
    fail_if(result == 1);
}
END_TEST

START_TEST (test_one_bit)
{
    uint64_t data = 0x8000000000000000;
    uint64_t result = eightbyte_get_bitfield(data, 0, 1, false);
    fail_unless(result == 0x1,
            "First bit in 0x%llx was 0x%llx instead of 0x1", data, result);
}
END_TEST

START_TEST (test_32_bit_parse)
{
    uint64_t data = 0x0402574d555a0401;
    uint64_t result = eightbyte_get_bitfield(data, 16, 32, false);
    uint64_t expectedValue = 0x574d555a;
    fail_unless(result == expectedValue,
            "Field retrieved in 0x%llx was 0x%llx instead of 0x%llx", data,
            result, expectedValue);
}
END_TEST

START_TEST (test_16_bit_parse)
{
    uint64_t data = 0xF34DFCFF00000000;
    uint64_t result = eightbyte_get_bitfield(data, 16, 16, false);
    uint64_t expectedValue = 0xFCFF;
    fail_unless(result == expectedValue,
            "Field retrieved in 0x%llx was 0x%llx instead of 0x%llx", data,
            result, expectedValue);
}
END_TEST

START_TEST (test_one_byte)
{
    uint64_t data = 0xFA00000000000000;
    uint64_t result = eightbyte_get_bitfield(data, 0, 4, false);
    fail_unless(result == 0xF,
            "First nibble in 0x%llx was 0x%llx instead of 0xF", data, result);
    result = eightbyte_get_bitfield(data, 4, 4, false);
    fail_unless(result == 0xA,
            "Second nibble in 0x%llx was 0x%llx instead of 0xA", data, result);
    result = eightbyte_get_bitfield(data, 0, 8, false);
    fail_unless(result == 0xFA,
            "All bits in 0x%llx were 0x%llx instead of 0x%llx", data, result, data);
}
END_TEST

START_TEST (test_multi_byte)
{
    uint64_t data = 0x12FA000000000000;
    uint64_t result = eightbyte_get_bitfield(data, 0, 4, false);
    fail_unless(result == 0x1,
            "First 4 bits in 0x%llx was 0x%llx instead of 0xF", (data >> 60) & 0xF,
            result);
    result = eightbyte_get_bitfield(data, 4, 4, false);
    fail_unless(result == 0x2,
            "Second 4 bits in 0x%llx was 0x%llx instead of 0xA", (data >> 56) & 0xF,
            result);
    result = eightbyte_get_bitfield(data, 8, 4, false);
    fail_unless(result == 0xF,
            "First 4 bits in 0x%llx was 0x%llx instead of 0x1", (data >> 52) & 0xF,
            result);
    result = eightbyte_get_bitfield(data, 12, 4, false);
    fail_unless(result == 0xA,
            "Second 4 bits in 0x%llx was 0x%llx instead of 0x2", (data >> 48) % 0xF,
            result);
}
END_TEST

START_TEST (test_get_multi_byte)
{
    uint64_t data = 0x12FA000000000000;
    uint64_t result = eightbyte_get_bitfield(data, 0, 9, false);
    ck_assert_int_eq(result, 0x25);
}
END_TEST

START_TEST (test_get_off_byte_boundary)
{
    uint64_t data = 0x000012FA00000000;
    uint64_t result = eightbyte_get_bitfield(data, 12, 8, false);
    ck_assert_int_eq(result, 0x01);
} END_TEST

START_TEST (test_set_wont_fit)
{
    uint64_t data = 0;
    fail_if(eightbyte_set_bitfield(&data, 100, 0, 1));
}
END_TEST

START_TEST (test_set_field)
{
    uint64_t data = 0;
    fail_unless(eightbyte_set_bitfield(&data, 1, 0, 1));
    uint64_t result = eightbyte_get_bitfield(data, 0, 1, false);
    ck_assert_int_eq(result, 0x1);
    data = 0;
    fail_unless(eightbyte_set_bitfield(&data, 1, 1, 1));
    result = eightbyte_get_bitfield(data, 1, 1, false);
    ck_assert_int_eq(result, 0x1);

    data = 0;
    fail_unless(eightbyte_set_bitfield(&data, 0xf, 3, 4));
    result = eightbyte_get_bitfield(data, 3, 4, false);
    ck_assert_int_eq(result, 0xf);
}
END_TEST

START_TEST (test_set_doesnt_clobber_existing_data)
{
    uint64_t data = 0xFFFC4DF300000000;
    fail_unless(eightbyte_set_bitfield(&data, 0x4fc8, 16, 16));
    uint64_t result = eightbyte_get_bitfield(data, 16, 16, false);
    fail_unless(result == 0x4fc8,
            "Field retrieved in 0x%llx was 0x%llx instead of 0x%x", data, result,
            0xc84f);

    data = 0x8000000000000000;
    fail_unless(eightbyte_set_bitfield(&data, 1, 21, 1));
    fail_unless(data == 0x8000040000000000LLU,
            "Expected combined value 0x8000040000000000 but got 0x%llx%llx",
            data >> 32, data);
}
END_TEST

START_TEST (test_set_off_byte_boundary)
{
    uint64_t data = 0xFFFC4DF300000000;
    fail_unless(eightbyte_set_bitfield(&data, 0x12, 12, 8));
    uint64_t result = eightbyte_get_bitfield(data, 12, 12, false);
    ck_assert_int_eq(result,0x12d);
}
END_TEST

START_TEST (test_set_odd_number_of_bits)
{
    uint64_t data = 0xFFFC4DF300000000LLU;
    fail_unless(eightbyte_set_bitfield(&data, 0x12, 11, 5));
    uint64_t result = eightbyte_get_bitfield(data, 11, 5, false);
    fail_unless(result == 0x12,
            "Field set in 0x%llx%llx%llx%llx was 0x%llx instead of 0x%llx", data, result,
            0x12);

    data = 0xFFFC4DF300000000LLU;
    fail_unless(eightbyte_set_bitfield(&data, 0x2, 11, 5));
    result = eightbyte_get_bitfield(data, 11, 5, false);
    fail_unless(result == 0x2,
            "Field set in 0x%llx%llx%llx%llx was 0x%llx instead of 0x%llx", data, result,
            0x2);
}
END_TEST

START_TEST(test_eightbyte_get_byte)
{
    uint64_t data = 0x00000000F34DFCFF;
    uint8_t result = eightbyte_get_byte(data, 0, false);
    uint8_t expected = 0x0;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_byte(data, 4, false);
    expected = 0xF3;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_byte(data, 5, false);
    expected = 0x4D;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_byte(data, 6, false);
    expected = 0xFC;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_byte(data, 7, false);
    expected = 0xFF;
    ck_assert_int_eq(result, expected);
}
END_TEST

START_TEST(test_eightbyte_get_nibble)
{
    uint64_t data = 0x00000000F34DFCFF;
    uint8_t result = eightbyte_get_nibble(data, 0, false);
    uint8_t expected = 0x0;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_nibble(data, 2, false);
    expected = 0x0;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_nibble(data, 8, false);
    expected = 0xF;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_nibble(data, 9, false);
    expected = 0x3;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_nibble(data, 10, false);
    expected = 0x4;
    ck_assert_int_eq(result, expected);

    result = eightbyte_get_nibble(data, 13, false);
    expected = 0xC;
    ck_assert_int_eq(result, expected);
}
END_TEST

Suite* bitfieldSuite(void) {
    Suite* s = suite_create("bitfield");
    TCase *tc_core = tcase_create("core");
    tcase_add_test(tc_core, test_large_bitmask);
    tcase_add_test(tc_core, test_one_bit);
    tcase_add_test(tc_core, test_one_bit_not_swapped);
    tcase_add_test(tc_core, test_one_byte);
    tcase_add_test(tc_core, test_16_bit_parse);
    tcase_add_test(tc_core, test_32_bit_parse);
    tcase_add_test(tc_core, test_multi_byte);
    tcase_add_test(tc_core, test_get_multi_byte);
    tcase_add_test(tc_core, test_get_off_byte_boundary);
    tcase_add_test(tc_core, test_set_wont_fit);
    tcase_add_test(tc_core, test_set_field);
    tcase_add_test(tc_core, test_set_doesnt_clobber_existing_data);
    tcase_add_test(tc_core, test_set_off_byte_boundary);
    tcase_add_test(tc_core, test_set_odd_number_of_bits);
    tcase_add_test(tc_core, test_eightbyte_get_nibble);
    tcase_add_test(tc_core, test_eightbyte_get_byte);
    suite_add_tcase(s, tc_core);

    return s;
}

int main(void) {
    int numberFailed;
    Suite* s = bitfieldSuite();
    SRunner *sr = srunner_create(s);
    // Don't fork so we can actually use gdb
    srunner_set_fork_status(sr, CK_NOFORK);
    srunner_run_all(sr, CK_NORMAL);
    numberFailed = srunner_ntests_failed(sr);
    srunner_free(sr);
    return (numberFailed == 0) ? 0 : 1;
}