mirror of
https://git.libssh.org/projects/libssh.git
synced 2026-02-06 18:29:50 +09:00
include: Add macros for bytearray handling
Signed-off-by: Andreas Schneider <asn@cryptomilk.org> Reviewed-by: Jakub Jelen <jjelen@redhat.com>
This commit is contained in:
Andreas Schneider
@@ -4,6 +4,7 @@ include_directories(${OPENSSL_INCLUDE_DIR})
|
||||
|
||||
set(LIBSSH_UNIT_TESTS
|
||||
torture_buffer
|
||||
torture_bytearray
|
||||
torture_callbacks
|
||||
torture_crypto
|
||||
torture_init
|
||||
|
||||
382
tests/unittests/torture_bytearray.c
Normal file
382
tests/unittests/torture_bytearray.c
Normal file
@@ -0,0 +1,382 @@
|
||||
#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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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};
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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};
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
|
||||
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;
|
||||
}
|
||||
Reference in New Issue
Block a user