shinys000114/libssh
1
0
Fork 0
mirror of https://git.libssh.org/projects/libssh.git synced 2026-02-11 10:40:27 +09:00
Code Issues Packages Projects Releases Wiki Activity

include: Add macros for bytearray handling

Browse Source 
Signed-off-by: Andreas Schneider <asn@cryptomilk.org>
Reviewed-by: Jakub Jelen <jjelen@redhat.com>
This commit is contained in:
Andreas Schneider
2018-09-29 14:06:00 +02:00
parent e210b61148
commit 85a274ff3c
3 changed files with 473 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
include/libssh/bytearray.h Normal file
View File

@@ -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 */

1
tests/unittests/CMakeLists.txt
View File

@@ -4,6 +4,7 @@ include_directories(${OPENSSL_INCLUDE_DIR})
set(LIBSSH_UNIT_TESTS set(LIBSSH_UNIT_TESTS
torture_buffer torture_buffer
torture_bytearray
torture_callbacks torture_callbacks
torture_crypto torture_crypto
torture_init torture_init

382
tests/unittests/torture_bytearray.c Normal file
View 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;
}