From: Andreas Schneider Date: Thu, 16 May 2019 06:31:40 +0000 (+0200) Subject: lib:util: Add a test for byteorder.h X-Git-Tag: samba-4.11.0rc1~885 X-Git-Url: http://git.samba.org/?p=samba.git;a=commitdiff_plain;h=0b8b04dbd3f053f11da15a8f28bbd5cbd9f4d48b lib:util: Add a test for byteorder.h Signed-off-by: Andreas Schneider Reviewed-by: Andrew Bartlett Autobuild-User(master): Andrew Bartlett Autobuild-Date(master): Fri May 17 20:44:36 UTC 2019 on sn-devel-184 --- diff --git a/lib/util/tests/test_byteorder.c b/lib/util/tests/test_byteorder.c new file mode 100644 index 00000000000..9faa0380826 --- /dev/null +++ b/lib/util/tests/test_byteorder.c @@ -0,0 +1,435 @@ +/* + * Unix SMB/CIFS implementation. + * + * Copyright (C) 2018-2019 Andreas Schneider + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * This program 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include +#include +#include +#include +#include + +#include "lib/replace/replace.h" +#include "lib/util/byteorder.h" + +static void torture_pull_le_u8(void **state) +{ + uint8_t data[2] = {0}; + uint8_t result; + + (void)state; + + result = CVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x2a; + result = CVAL(data, 0); + assert_int_equal(result, 42); + + + data[0] = 0xf; + result = CVAL(data, 0); + assert_int_equal(result, 0xf); + + data[0] = 0xff; + result = CVAL(data, 0); + assert_int_equal(result, 0xff); + + data[1] = 0x2a; + result = CVAL(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 = SVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x2a; + data[1] = 0x00; + result = SVAL(data, 0); + assert_int_equal(result, 42); + + data[0] = 0xff; + data[1] = 0x00; + result = SVAL(data, 0); + assert_int_equal(result, 0x00ff); + + data[0] = 0x00; + data[1] = 0xff; + result = SVAL(data, 0); + assert_int_equal(result, 0xff00); + + data[0] = 0xff; + data[1] = 0xff; + result = SVAL(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 = IVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x2a; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0x00; + result = IVAL(data, 0); + assert_int_equal(result, 42); + + data[0] = 0xff; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0x00; + result = IVAL(data, 0); + assert_int_equal(result, 0x00ff); + + data[0] = 0x00; + data[1] = 0xff; + data[2] = 0x00; + data[3] = 0x00; + result = IVAL(data, 0); + assert_int_equal(result, 0xff00); + + data[0] = 0x00; + data[1] = 0x00; + data[2] = 0xff; + data[3] = 0x00; + result = IVAL(data, 0); + assert_int_equal(result, 0xff0000); + + data[0] = 0x00; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0xff; + result = IVAL(data, 0); + assert_int_equal(result, 0xff000000); + + data[0] = 0xff; + data[1] = 0xff; + data[2] = 0xff; + data[3] = 0xff; + result = IVAL(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; + + SCVAL(data, 0, 42); + SCVAL(data, 1, 42); + SCVAL(data, 2, 42); + SCVAL(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; + + SSVALX(data, 0, 32678); + SSVALX(data, 2, 42); + assert_memory_equal(data, data2, sizeof(data)); + + result = SVAL(data, 2); + assert_int_equal(result, 42); + + result = SVAL(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; + + SIVALX(data, 0, 32678); + SIVALX(data, 4, 42); + assert_memory_equal(data, data2, sizeof(data)); + + result = IVAL(data, 4); + assert_int_equal(result, 42); + + result = IVAL(data, 0); + assert_int_equal(result, 32678); + + SIVALX(data, 0, 0xfffefffe); + result = IVAL(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; + + SBVAL(data, 0, 32678); + + result = BVAL(data, 0); + assert_int_equal(result, 32678); + + SBVAL(data, 0, 0xfffefffefffefffeUL); + + result = BVAL(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 = CVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x2a; + result = CVAL(data, 0); + assert_int_equal(result, 42); + + + data[0] = 0xf; + result = CVAL(data, 0); + assert_int_equal(result, 0xf); + + data[0] = 0xff; + result = CVAL(data, 0); + assert_int_equal(result, 0xff); + + data[1] = 0x2a; + result = CVAL(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 = RSVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x00; + data[1] = 0x2a; + result = RSVAL(data, 0); + assert_int_equal(result, 42); + + data[0] = 0x00; + data[1] = 0xff; + result = RSVAL(data, 0); + assert_int_equal(result, 0x00ff); + + data[0] = 0xff; + data[1] = 0x00; + result = RSVAL(data, 0); + assert_int_equal(result, 0xff00); + + data[0] = 0xff; + data[1] = 0xff; + result = RSVAL(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 = RIVAL(data, 0); + assert_int_equal(result, 0); + + data[0] = 0x00; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0x2a; + result = RIVAL(data, 0); + assert_int_equal(result, 42); + + data[0] = 0x00; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0xff; + result = RIVAL(data, 0); + assert_int_equal(result, 0x00ff); + + data[0] = 0x00; + data[1] = 0x00; + data[2] = 0xff; + data[3] = 0x00; + result = RIVAL(data, 0); + assert_int_equal(result, 0xff00); + + data[0] = 0x00; + data[1] = 0xff; + data[2] = 0x00; + data[3] = 0x00; + result = RIVAL(data, 0); + assert_int_equal(result, 0xff0000); + + data[0] = 0xff; + data[1] = 0x00; + data[2] = 0x00; + data[3] = 0x00; + result = RIVAL(data, 0); + assert_int_equal(result, 0xff000000); + + data[0] = 0xff; + data[1] = 0xff; + data[2] = 0xff; + data[3] = 0xff; + result = RIVAL(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; + + SCVAL(data, 0, 42); + SCVAL(data, 1, 42); + SCVAL(data, 2, 42); + SCVAL(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; + + RSSVALS(data, 0, 32678); + RSSVALS(data, 2, 42); + assert_memory_equal(data, data2, sizeof(data)); + + result = RSVAL(data, 2); + assert_int_equal(result, 42); + + result = RSVAL(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; + + RSIVALS(data, 0, 32678); + RSIVALS(data, 4, 42); + assert_memory_equal(data, data2, sizeof(data)); + + result = RIVAL(data, 4); + assert_int_equal(result, 42); + + result = RIVAL(data, 0); + assert_int_equal(result, 32678); + + RSIVALS(data, 0, 0xfffefffe); + result = RIVAL(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; + + RSBVALS(data, 0, 32678); + + result = RBVAL(data, 0); + assert_int_equal(result, 32678); + + SBVAL(data, 8, 0xfffefffe); + + result = BVAL(data, 8); + assert_int_equal(result, 0xfffefffe); +} + +int main(int argc, char *argv[]) +{ + int rc; + const 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), + }; + + if (argc == 2) { + cmocka_set_test_filter(argv[1]); + } + cmocka_set_message_output(CM_OUTPUT_SUBUNIT); + + rc = cmocka_run_group_tests(tests, NULL, NULL); + + return rc; +} diff --git a/lib/util/wscript_build b/lib/util/wscript_build index 9af1bb5d801..ff1c76e3686 100644 --- a/lib/util/wscript_build +++ b/lib/util/wscript_build @@ -275,3 +275,9 @@ else: deps='cmocka replace samba-util', local_include=False, install=False) + + bld.SAMBA_BINARY('test_byteorder', + source='tests/test_byteorder.c', + deps='cmocka replace samba-util', + local_include=False, + install=False) diff --git a/selftest/tests.py b/selftest/tests.py index 133f227ab82..1568f29b212 100644 --- a/selftest/tests.py +++ b/selftest/tests.py @@ -257,6 +257,8 @@ plantestsuite("samba.unittests.kerberos", "none", [os.path.join(bindir(), "test_kerberos")]) plantestsuite("samba.unittests.ms_fnmatch", "none", [os.path.join(bindir(), "default/lib/util/test_ms_fnmatch")]) +plantestsuite("samba.unittests.byteorder", "none", + [os.path.join(bindir(), "default/lib/util/test_byteorder")]) plantestsuite("samba.unittests.ntlm_check", "none", [os.path.join(bindir(), "default/libcli/auth/test_ntlm_check")]) plantestsuite("samba.unittests.test_registry_regfio", "none",