include: Add macros for bytearray handling
Signed-off-by: Andreas Schneider <asn@cryptomilk.org> Reviewed-by: Jakub Jelen <jjelen@redhat.com>
Этот коммит содержится в:
Andreas Schneider
родитель
e210b61148
Коммит
85a274ff3c
90
include/libssh/bytearray.h
Обычный файл
90
include/libssh/bytearray.h
Обычный файл
@ -0,0 +1,90 @@
|
||||
/*
|
||||
* This file is part of the SSH Library
|
||||
*
|
||||
* Copyright (c) 2018 Andreas Schneider <asn@cryptomilk.org>
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Lesser General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2.1 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public
|
||||
* License along with this library; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
*/
|
||||
#ifndef _BYTEARRAY_H
|
||||
#define _BYTEARRAY_H
|
||||
|
||||
#define _DATA_BYTE_CONST(data, pos) \
|
||||
((uint8_t)(((const uint8_t *)(data))[(pos)]))
|
||||
|
||||
#define _DATA_BYTE(data, pos) \
|
||||
(((uint8_t *)(data))[(pos)])
|
||||
|
||||
/*
|
||||
* These macros pull or push integer values from byte arrays stored in
|
||||
* little-endian byte order.
|
||||
*/
|
||||
#define PULL_LE_U8(data, pos) \
|
||||
(_DATA_BYTE_CONST(data, pos))
|
||||
|
||||
#define PULL_LE_U16(data, pos) \
|
||||
((uint16_t)PULL_LE_U8(data, pos) | ((uint16_t)(PULL_LE_U8(data, (pos) + 1))) << 8)
|
||||
|
||||
#define PULL_LE_U32(data, pos) \
|
||||
((uint32_t)(PULL_LE_U16(data, pos) | ((uint32_t)PULL_LE_U16(data, (pos) + 2)) << 16))
|
||||
|
||||
#define PULL_LE_U64(data, pos) \
|
||||
((uint64_t)(PULL_LE_U32(data, pos) | ((uint64_t)PULL_LE_U32(data, (pos) + 4)) << 32))
|
||||
|
||||
|
||||
#define PUSH_LE_U8(data, pos, val) \
|
||||
(_DATA_BYTE(data, pos) = ((uint8_t)(val)))
|
||||
|
||||
#define PUSH_LE_U16(data, pos, val) \
|
||||
(PUSH_LE_U8((data), (pos), (uint8_t)((uint16_t)(val) & 0xff)), PUSH_LE_U8((data), (pos) + 1, (uint8_t)((uint16_t)(val) >> 8)))
|
||||
|
||||
#define PUSH_LE_U32(data, pos, val) \
|
||||
(PUSH_LE_U16((data), (pos), (uint16_t)((uint32_t)(val) & 0xffff)), PUSH_LE_U16((data), (pos) + 2, (uint16_t)((uint32_t)(val) >> 16)))
|
||||
|
||||
#define PUSH_LE_U64(data, pos, val) \
|
||||
(PUSH_LE_U32((data), (pos), (uint32_t)((uint64_t)(val) & 0xffffffff)), PUSH_LE_U32((data), (pos) + 4, (uint32_t)((uint64_t)(val) >> 32)))
|
||||
|
||||
|
||||
|
||||
/*
|
||||
* These macros pull or push integer values from byte arrays stored in
|
||||
* big-endian byte order (network byte order).
|
||||
*/
|
||||
#define PULL_BE_U8(data, pos) \
|
||||
(_DATA_BYTE_CONST(data, pos))
|
||||
|
||||
#define PULL_BE_U16(data, pos) \
|
||||
((((uint16_t)(PULL_BE_U8(data, pos))) << 8) | (uint16_t)PULL_BE_U8(data, (pos) + 1))
|
||||
|
||||
#define PULL_BE_U32(data, pos) \
|
||||
((((uint32_t)PULL_BE_U16(data, pos)) << 16) | (uint32_t)(PULL_BE_U16(data, (pos) + 2)))
|
||||
|
||||
#define PULL_BE_U64(data, pos) \
|
||||
((((uint64_t)PULL_BE_U32(data, pos)) << 32) | (uint64_t)(PULL_BE_U32(data, (pos) + 4)))
|
||||
|
||||
|
||||
|
||||
#define PUSH_BE_U8(data, pos, val) \
|
||||
(_DATA_BYTE(data, pos) = ((uint8_t)(val)))
|
||||
|
||||
#define PUSH_BE_U16(data, pos, val) \
|
||||
(PUSH_BE_U8((data), (pos), (uint8_t)(((uint16_t)(val)) >> 8)), PUSH_BE_U8((data), (pos) + 1, (uint8_t)((val) & 0xff)))
|
||||
|
||||
#define PUSH_BE_U32(data, pos, val) \
|
||||
(PUSH_BE_U16((data), (pos), (uint16_t)(((uint32_t)(val)) >> 16)), PUSH_BE_U16((data), (pos) + 2, (uint16_t)((val) & 0xffff)))
|
||||
|
||||
#define PUSH_BE_U64(data, pos, val) \
|
||||
(PUSH_BE_U32((data), (pos), (uint32_t)(((uint64_t)(val)) >> 32)), PUSH_BE_U32((data), (pos) + 4, (uint32_t)((val) & 0xffffffff)))
|
||||
|
||||
#endif /* _BYTEARRAY_H */
|
||||
@ -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
Обычный файл
382
tests/unittests/torture_bytearray.c
Обычный файл
@ -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;
|
||||
}
|
||||
Загрузка…
x
Ссылка в новой задаче
Block a user