182 lines
5.8 KiB
C++
182 lines
5.8 KiB
C++
/*
|
|
* Copyright (c) 2019 Facebook
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#include <ztest.h>
|
|
#include <sys/math_extras.h>
|
|
#include <inttypes.h>
|
|
|
|
static void VNAME(u32_add)(void)
|
|
{
|
|
uint32_t result = 42;
|
|
|
|
zassert_false(u32_add_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 5, NULL);
|
|
|
|
zassert_false(u32_add_overflow(2, 0, &result), NULL);
|
|
zassert_equal(result, 2, NULL);
|
|
|
|
zassert_false(u32_add_overflow(0, 3, &result), NULL);
|
|
zassert_equal(result, 3, NULL);
|
|
|
|
zassert_false(u32_add_overflow(0, UINT32_MAX, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX, NULL);
|
|
zassert_true(u32_add_overflow(1, UINT32_MAX, &result), NULL);
|
|
zassert_equal(result, 0, NULL);
|
|
|
|
zassert_false(u32_add_overflow(UINT32_MAX, 0, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX, NULL);
|
|
zassert_true(u32_add_overflow(UINT32_MAX, 2, &result), NULL);
|
|
zassert_equal(result, 1, NULL);
|
|
}
|
|
|
|
static void VNAME(u32_mul)(void)
|
|
{
|
|
uint32_t result = 42;
|
|
|
|
zassert_false(u32_mul_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 6, NULL);
|
|
|
|
zassert_false(u32_mul_overflow(UINT32_MAX, 1, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX, NULL);
|
|
zassert_true(u32_mul_overflow(UINT32_MAX, 2, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX * 2, NULL);
|
|
|
|
zassert_false(u32_mul_overflow(1, UINT32_MAX, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX, NULL);
|
|
zassert_true(u32_mul_overflow(2, UINT32_MAX, &result), NULL);
|
|
zassert_equal(result, UINT32_MAX * 2, NULL);
|
|
}
|
|
|
|
static void VNAME(u64_add)(void)
|
|
{
|
|
uint64_t result = 42;
|
|
|
|
zassert_false(u64_add_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 5, NULL);
|
|
|
|
zassert_false(u64_add_overflow(2, 0, &result), NULL);
|
|
zassert_equal(result, 2, NULL);
|
|
|
|
zassert_false(u64_add_overflow(0, 3, &result), NULL);
|
|
zassert_equal(result, 3, NULL);
|
|
|
|
zassert_false(u64_add_overflow(0, UINT64_MAX, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX, NULL);
|
|
zassert_true(u64_add_overflow(1, UINT64_MAX, &result), NULL);
|
|
zassert_equal(result, 0, NULL);
|
|
|
|
zassert_false(u64_add_overflow(UINT64_MAX, 0, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX, NULL);
|
|
zassert_true(u64_add_overflow(UINT64_MAX, 2, &result), NULL);
|
|
zassert_equal(result, 1, NULL);
|
|
}
|
|
|
|
static void VNAME(u64_mul)(void)
|
|
{
|
|
uint64_t result = 42;
|
|
|
|
zassert_false(u64_mul_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 6, NULL);
|
|
|
|
zassert_false(u64_mul_overflow(UINT64_MAX, 1, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX, NULL);
|
|
zassert_true(u64_mul_overflow(UINT64_MAX, 2, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX * 2, NULL);
|
|
|
|
zassert_false(u64_mul_overflow(1, UINT64_MAX, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX, NULL);
|
|
zassert_true(u64_mul_overflow(2, UINT64_MAX, &result), NULL);
|
|
zassert_equal(result, UINT64_MAX * 2, NULL);
|
|
}
|
|
|
|
static void VNAME(size_add)(void)
|
|
{
|
|
size_t result = 42;
|
|
|
|
zassert_false(size_add_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 5, NULL);
|
|
|
|
zassert_false(size_add_overflow(2, 0, &result), NULL);
|
|
zassert_equal(result, 2, NULL);
|
|
|
|
zassert_false(size_add_overflow(0, 3, &result), NULL);
|
|
zassert_equal(result, 3, NULL);
|
|
|
|
zassert_false(size_add_overflow(0, SIZE_MAX, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX, NULL);
|
|
zassert_true(size_add_overflow(1, SIZE_MAX, &result), NULL);
|
|
zassert_equal(result, 0, NULL);
|
|
|
|
zassert_false(size_add_overflow(SIZE_MAX, 0, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX, NULL);
|
|
zassert_true(size_add_overflow(SIZE_MAX, 2, &result), NULL);
|
|
zassert_equal(result, 1, NULL);
|
|
}
|
|
|
|
static void VNAME(size_mul)(void)
|
|
{
|
|
size_t result = 42;
|
|
|
|
zassert_false(size_mul_overflow(2, 3, &result), NULL);
|
|
zassert_equal(result, 6, NULL);
|
|
|
|
zassert_false(size_mul_overflow(SIZE_MAX, 1, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX, NULL);
|
|
zassert_true(size_mul_overflow(SIZE_MAX, 2, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX * 2, NULL);
|
|
|
|
zassert_false(size_mul_overflow(1, SIZE_MAX, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX, NULL);
|
|
zassert_true(size_mul_overflow(2, SIZE_MAX, &result), NULL);
|
|
zassert_equal(result, SIZE_MAX * 2, NULL);
|
|
}
|
|
|
|
static void VNAME(u32_clz)(void)
|
|
{
|
|
zassert_equal(u32_count_leading_zeros(0), 32, NULL);
|
|
zassert_equal(u32_count_leading_zeros(1), 31, NULL);
|
|
zassert_equal(u32_count_leading_zeros(0xf00f), 16, NULL);
|
|
zassert_equal(u32_count_leading_zeros(0xf00ff00f), 0, NULL);
|
|
zassert_equal(u32_count_leading_zeros(0xffffffff), 0, NULL);
|
|
}
|
|
|
|
static void VNAME(u64_clz)(void)
|
|
{
|
|
zassert_equal(u64_count_leading_zeros(0), 64, NULL);
|
|
zassert_equal(u64_count_leading_zeros(1), 63, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xf00f), 48, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xf00ff00f), 32, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xffffffff), 32, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xf00f00000000ull), 16, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xf00ff00f00000000ull), 0, NULL);
|
|
zassert_equal(u64_count_leading_zeros(0xffffffff00000000ull), 0, NULL);
|
|
}
|
|
|
|
static void VNAME(u32_ctz)(void)
|
|
{
|
|
zassert_equal(u32_count_trailing_zeros(0), 32, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(1), 0, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(6), 1, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(0x00f00f00), 8, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(0xf00ffc00), 10, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(0xffffffff), 0, NULL);
|
|
zassert_equal(u32_count_trailing_zeros(0x80000000), 31, NULL);
|
|
}
|
|
|
|
static void VNAME(u64_ctz)(void)
|
|
{
|
|
zassert_equal(u64_count_trailing_zeros(0), 64, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(1), 0, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(6), 1, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(0x00f00f00), 8, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(0xf00ffc00), 10, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(0xffffffffffffffffull), 0, NULL);
|
|
zassert_equal(u64_count_trailing_zeros(0x8000000080000000ull), 31,
|
|
NULL);
|
|
zassert_equal(u64_count_trailing_zeros(0xc000000000000000ull), 62,
|
|
NULL);
|
|
}
|