#include "config.h"

#include <stdint.h>

#include "torture.h"
#include "libssh/bytearray.h"

static void torture_pull_le_u8(void **state)
{
    uint8_t data[2] = {0};
    uint8_t result;

    (void)state;

    result = PULL_LE_U8(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x2a;
    result = PULL_LE_U8(data, 0);
    assert_int_equal(result, 42);


    data[0] = 0xf;
    result = PULL_LE_U8(data, 0);
    assert_int_equal(result, 0xf);

    data[0] = 0xff;
    result = PULL_LE_U8(data, 0);
    assert_int_equal(result, 0xff);

    data[1] = 0x2a;
    result = PULL_LE_U8(data, 1);
    assert_int_equal(result, 42);
}

static void torture_pull_le_u16(void **state)
{
    uint8_t data[2] = {0, 0};
    uint16_t result;

    (void)state;

    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x2a;
    data[1] = 0x00;
    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 42);

    data[0] = 0xff;
    data[1] = 0x00;
    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 0x00ff);

    data[0] = 0x00;
    data[1] = 0xff;
    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 0xff00);

    data[0] = 0xff;
    data[1] = 0xff;
    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 0xffff);
}

static void torture_pull_le_u32(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint32_t result;

    (void)state;

    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x2a;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0x00;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 42);

    data[0] = 0xff;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0x00;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0x00ff);

    data[0] = 0x00;
    data[1] = 0xff;
    data[2] = 0x00;
    data[3] = 0x00;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0xff00);

    data[0] = 0x00;
    data[1] = 0x00;
    data[2] = 0xff;
    data[3] = 0x00;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0xff0000);

    data[0] = 0x00;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0xff;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0xff000000);

    data[0] = 0xff;
    data[1] = 0xff;
    data[2] = 0xff;
    data[3] = 0xff;
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0xffffffff);
}

static void torture_push_le_u8(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint8_t data2[4] = {42, 42, 42, 42};

    (void)state;

    PUSH_LE_U8(data, 0, 42);
    PUSH_LE_U8(data, 1, 42);
    PUSH_LE_U8(data, 2, 42);
    PUSH_LE_U8(data, 3, 42);
    assert_memory_equal(data, data2, sizeof(data));
}

static void torture_push_le_u16(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint8_t data2[4] = {0xa6, 0x7f, 0x2a, 0x00};
    uint16_t result;

    (void)state;

    PUSH_LE_U16(data, 0, 32678);
    PUSH_LE_U16(data, 2, 42);
    assert_memory_equal(data, data2, sizeof(data));

    result = PULL_LE_U16(data, 2);
    assert_int_equal(result, 42);

    result = PULL_LE_U16(data, 0);
    assert_int_equal(result, 32678);
}

static void torture_push_le_u32(void **state)
{
    uint8_t data[8] = {0};
    uint8_t data2[8] = {0xa6, 0x7f, 0x00, 0x00, 0x2a, 0x00, 0x00, 0x00};
    uint32_t result;

    (void)state;

    PUSH_LE_U32(data, 0, 32678);
    PUSH_LE_U32(data, 4, 42);
    assert_memory_equal(data, data2, sizeof(data));

    result = PULL_LE_U32(data, 4);
    assert_int_equal(result, 42);

    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 32678);

    PUSH_LE_U32(data, 0, 0xfffefffe);
    result = PULL_LE_U32(data, 0);
    assert_int_equal(result, 0xfffefffe);
}

static void torture_push_le_u64(void **state)
{
    uint8_t data[16] = {0};
    uint64_t result;

    (void)state;

    PUSH_LE_U64(data, 0, 32678);

    result = PULL_LE_U64(data, 0);
    assert_int_equal(result, 32678);

    PUSH_LE_U64(data, 0, 0xfffefffefffefffeUL);

    result = PULL_LE_U64(data, 0);
    assert_int_equal(result, 0xfffefffefffefffeUL);
}

/****************** BIG ENDIAN ********************/

static void torture_pull_be_u8(void **state)
{
    uint8_t data[2] = {0};
    uint8_t result;

    (void)state;

    result = PULL_BE_U8(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x2a;
    result = PULL_BE_U8(data, 0);
    assert_int_equal(result, 42);


    data[0] = 0xf;
    result = PULL_BE_U8(data, 0);
    assert_int_equal(result, 0xf);

    data[0] = 0xff;
    result = PULL_BE_U8(data, 0);
    assert_int_equal(result, 0xff);

    data[1] = 0x2a;
    result = PULL_BE_U8(data, 1);
    assert_int_equal(result, 42);
}

static void torture_pull_be_u16(void **state)
{
    uint8_t data[2] = {0, 0};
    uint16_t result;

    (void)state;

    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x00;
    data[1] = 0x2a;
    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 42);

    data[0] = 0x00;
    data[1] = 0xff;
    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 0x00ff);

    data[0] = 0xff;
    data[1] = 0x00;
    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 0xff00);

    data[0] = 0xff;
    data[1] = 0xff;
    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 0xffff);
}

static void torture_pull_be_u32(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint32_t result;

    (void)state;

    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0);

    data[0] = 0x00;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0x2a;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 42);

    data[0] = 0x00;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0xff;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0x00ff);

    data[0] = 0x00;
    data[1] = 0x00;
    data[2] = 0xff;
    data[3] = 0x00;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0xff00);

    data[0] = 0x00;
    data[1] = 0xff;
    data[2] = 0x00;
    data[3] = 0x00;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0xff0000);

    data[0] = 0xff;
    data[1] = 0x00;
    data[2] = 0x00;
    data[3] = 0x00;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0xff000000);

    data[0] = 0xff;
    data[1] = 0xff;
    data[2] = 0xff;
    data[3] = 0xff;
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0xffffffff);
}

static void torture_push_be_u8(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint8_t data2[4] = {42, 42, 42, 42};

    (void)state;

    PUSH_BE_U8(data, 0, 42);
    PUSH_BE_U8(data, 1, 42);
    PUSH_BE_U8(data, 2, 42);
    PUSH_BE_U8(data, 3, 42);
    assert_memory_equal(data, data2, sizeof(data));
}

static void torture_push_be_u16(void **state)
{
    uint8_t data[4] = {0, 0, 0, 0};
    uint8_t data2[4] = {0x7f, 0xa6, 0x00, 0x2a};
    uint16_t result;

    (void)state;

    PUSH_BE_U16(data, 0, 32678);
    PUSH_BE_U16(data, 2, 42);
    assert_memory_equal(data, data2, sizeof(data));

    result = PULL_BE_U16(data, 2);
    assert_int_equal(result, 42);

    result = PULL_BE_U16(data, 0);
    assert_int_equal(result, 32678);
}

static void torture_push_be_u32(void **state)
{
    uint8_t data[8] = {0};
    uint8_t data2[8] = {0x00, 0x00, 0x7f, 0xa6, 0x00, 0x00, 0x00, 0x2a};
    uint32_t result;

    (void)state;

    PUSH_BE_U32(data, 0, 32678);
    PUSH_BE_U32(data, 4, 42);
    assert_memory_equal(data, data2, sizeof(data));

    result = PULL_BE_U32(data, 4);
    assert_int_equal(result, 42);

    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 32678);

    PUSH_BE_U32(data, 0, 0xfffefffe);
    result = PULL_BE_U32(data, 0);
    assert_int_equal(result, 0xfffefffe);
}

static void torture_push_be_u64(void **state)
{
    uint8_t data[16] = {0};
    uint64_t result;

    (void)state;

    PUSH_BE_U64(data, 0, 32678);

    result = PULL_BE_U64(data, 0);
    assert_int_equal(result, 32678);

    PUSH_LE_U64(data, 8, 0xfffefffe);

    result = PULL_LE_U64(data, 8);
    assert_int_equal(result, 0xfffefffe);
}

int torture_run_tests(void) {
    int rc;
    struct CMUnitTest tests[] = {
        cmocka_unit_test(torture_pull_le_u8),
        cmocka_unit_test(torture_pull_le_u16),
        cmocka_unit_test(torture_pull_le_u32),

        cmocka_unit_test(torture_push_le_u8),
        cmocka_unit_test(torture_push_le_u16),
        cmocka_unit_test(torture_push_le_u32),
        cmocka_unit_test(torture_push_le_u64),

        /* BIG ENDIAN */
        cmocka_unit_test(torture_pull_be_u8),
        cmocka_unit_test(torture_pull_be_u16),
        cmocka_unit_test(torture_pull_be_u32),

        cmocka_unit_test(torture_push_be_u8),
        cmocka_unit_test(torture_push_be_u16),
        cmocka_unit_test(torture_push_be_u32),
        cmocka_unit_test(torture_push_be_u64),
    };

    torture_filter_tests(tests);

    rc = cmocka_run_group_tests(tests, NULL, NULL);

    return rc;
}