/**
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0.
 */

#include <aws/common/math.h>

#include <aws/testing/aws_test_harness.h>

#include <float.h>
#include <stdio.h>

#define CHECK_SAT(fn, a, b, result)                                                                                    \
    do {                                                                                                               \
        ASSERT_UINT_EQUALS(                                                                                            \
            (result),                                                                                                  \
            fn((a), (b)),                                                                                              \
            "%s(0x%016llx, 0x%016llx) = 0x%016llx",                                                                    \
            #fn,                                                                                                       \
            (unsigned long long)(a),                                                                                   \
            (unsigned long long)(b),                                                                                   \
            (unsigned long long)(result));                                                                             \
        ASSERT_UINT_EQUALS(                                                                                            \
            (result),                                                                                                  \
            fn((b), (a)),                                                                                              \
            "%s(0x%016llx, 0x%016llx) = 0x%016llx",                                                                    \
            #fn,                                                                                                       \
            (unsigned long long)(b),                                                                                   \
            (unsigned long long)(a),                                                                                   \
            (unsigned long long)(result));                                                                             \
    } while (0)

AWS_TEST_CASE(test_is_power_of_two, s_test_is_power_of_two_fn)
static int s_test_is_power_of_two_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    ASSERT_FALSE(aws_is_power_of_two(0));
    for (size_t i = 0; i < SIZE_BITS; ++i) {
        const size_t ith_power = (size_t)1 << i;
        ASSERT_TRUE(aws_is_power_of_two(ith_power));
        ASSERT_FALSE(aws_is_power_of_two(ith_power + 9));
        ASSERT_FALSE(aws_is_power_of_two(ith_power - 9));
        ASSERT_FALSE(aws_is_power_of_two(ith_power + 100));
        ASSERT_FALSE(aws_is_power_of_two(ith_power - 100));
    }
    return 0;
}

#define CHECK_ROUND_OVERFLOWS(a)                                                                                       \
    do {                                                                                                               \
        size_t result_val;                                                                                             \
        ASSERT_TRUE(aws_round_up_to_power_of_two((a), &result_val));                                                   \
    } while (0)

#define CHECK_ROUND_SUCCEEDS(a, r)                                                                                     \
    do {                                                                                                               \
        size_t result_val;                                                                                             \
        ASSERT_FALSE(aws_round_up_to_power_of_two((a), &result_val));                                                  \
        ASSERT_TRUE(aws_is_power_of_two(result_val));                                                                  \
        ASSERT_INT_EQUALS(                                                                                             \
            (uint64_t)result_val,                                                                                      \
            (uint64_t)(r),                                                                                             \
            "Expected %s(%016llx) = %016llx; got %016llx",                                                             \
            "aws_round_up_to_power_of_two",                                                                            \
            (unsigned long long)(a),                                                                                   \
            (unsigned long long)(r),                                                                                   \
            (unsigned long long)(result_val));                                                                         \
    } while (0)

AWS_TEST_CASE(test_round_up_to_power_of_two, s_test_round_up_to_power_of_two_fn)
static int s_test_round_up_to_power_of_two_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    /*special case 0, 1 and 2, where subtracting from the number doesn't cause it to round back up */
    CHECK_ROUND_SUCCEEDS(0, 1);
    CHECK_ROUND_SUCCEEDS(1, 1);
    CHECK_ROUND_SUCCEEDS(2, 2);

    for (size_t i = 2; i < SIZE_BITS - 1; ++i) {
        const size_t ith_power = (size_t)1 << i;
        CHECK_ROUND_SUCCEEDS(ith_power, ith_power);
        CHECK_ROUND_SUCCEEDS(ith_power - 1, ith_power);
        CHECK_ROUND_SUCCEEDS(ith_power + 1, ith_power << 1);
    }

    /* Special case for the largest representable power of two, where addition causes overflow */
    CHECK_ROUND_SUCCEEDS(SIZE_MAX_POWER_OF_TWO, SIZE_MAX_POWER_OF_TWO);
    CHECK_ROUND_SUCCEEDS(SIZE_MAX_POWER_OF_TWO - 1, SIZE_MAX_POWER_OF_TWO);
    CHECK_ROUND_OVERFLOWS(SIZE_MAX_POWER_OF_TWO + 1);
    return 0;
}

AWS_TEST_CASE(test_mul_u64_saturating, s_test_mul_u64_saturating_fn)
static int s_test_mul_u64_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    CHECK_SAT(aws_mul_u64_saturating, 0, 0, 0);
    CHECK_SAT(aws_mul_u64_saturating, 0, 1, 0);
    CHECK_SAT(aws_mul_u64_saturating, 0, ~0LLU, 0);
    CHECK_SAT(aws_mul_u64_saturating, 4, 5, 20);
    CHECK_SAT(aws_mul_u64_saturating, 1234, 4321, 5332114);

    CHECK_SAT(aws_mul_u64_saturating, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_SAT(aws_mul_u64_saturating, 0xFFFFFFFF, 0xFFFFFFFF, 0xfffffffe00000001LLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x100000000, 0xFFFFFFFF, 0xFFFFFFFF00000000LLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x100000001, 0xFFFFFFFF, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x100000001, 0xFFFFFFFE, 0xFFFFFFFEFFFFFFFELLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x100000002, 0xFFFFFFFE, 0xFFFFFFFFFFFFFFFCLLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x100000003, 0xFFFFFFFE, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_u64_saturating, 0xFFFFFFFE, 0xFFFFFFFE, 0xFFFFFFFC00000004LLU);
    CHECK_SAT(aws_mul_u64_saturating, 0x1FFFFFFFF, 0x1FFFFFFFF, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_u64_saturating, ~0LLU, ~0LLU, ~0LLU);

    return 0;
}

AWS_TEST_CASE(test_mul_u32_saturating, s_test_mul_u32_saturating_fn)
static int s_test_mul_u32_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    CHECK_SAT(aws_mul_u32_saturating, 0, 0, 0);
    CHECK_SAT(aws_mul_u32_saturating, 0, 1, 0);
    CHECK_SAT(aws_mul_u32_saturating, 0, ~0U, 0);
    CHECK_SAT(aws_mul_u32_saturating, 4, 5, 20);
    CHECK_SAT(aws_mul_u32_saturating, 1234, 4321, 5332114);

    CHECK_SAT(aws_mul_u32_saturating, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_SAT(aws_mul_u32_saturating, 0xFFFF, 1, 0xFFFF);
    CHECK_SAT(aws_mul_u32_saturating, 0xFFFF, 0xFFFF, 0xfffe0001);
    CHECK_SAT(aws_mul_u32_saturating, 0x10000, 0xFFFF, 0xFFFF0000U);
    CHECK_SAT(aws_mul_u32_saturating, 0x10001, 0xFFFF, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_u32_saturating, 0x10001, 0xFFFE, 0xFFFEFFFEU);
    CHECK_SAT(aws_mul_u32_saturating, 0x10002, 0xFFFE, 0xFFFFFFFCU);
    CHECK_SAT(aws_mul_u32_saturating, 0x10003, 0xFFFE, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_u32_saturating, 0xFFFE, 0xFFFE, 0xFFFC0004U);
    CHECK_SAT(aws_mul_u32_saturating, 0x1FFFF, 0x1FFFF, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_u32_saturating, ~0U, ~0U, ~0U);

    return 0;
}

AWS_TEST_CASE(test_mul_size_saturating, s_test_mul_size_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_mul_size_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

#if SIZE_BITS == 32
    CHECK_SAT(aws_mul_size_saturating, 0, 0, 0);
    CHECK_SAT(aws_mul_size_saturating, 0, 1, 0);
    CHECK_SAT(aws_mul_size_saturating, 0, ~0U, 0);
    CHECK_SAT(aws_mul_size_saturating, 4, 5, 20);
    CHECK_SAT(aws_mul_size_saturating, 1234, 4321, 5332114);

    CHECK_SAT(aws_mul_size_saturating, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_SAT(aws_mul_size_saturating, 0xFFFF, 1, 0xFFFF);
    CHECK_SAT(aws_mul_size_saturating, 0xFFFF, 0xFFFF, 0xfffe0001);
    CHECK_SAT(aws_mul_size_saturating, 0x10000, 0xFFFF, 0xFFFF0000U);
    CHECK_SAT(aws_mul_size_saturating, 0x10001, 0xFFFF, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_size_saturating, 0x10001, 0xFFFE, 0xFFFEFFFEU);
    CHECK_SAT(aws_mul_size_saturating, 0x10002, 0xFFFE, 0xFFFFFFFCU);
    CHECK_SAT(aws_mul_size_saturating, 0x10003, 0xFFFE, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_size_saturating, 0xFFFE, 0xFFFE, 0xFFFC0004U);
    CHECK_SAT(aws_mul_size_saturating, 0x1FFFF, 0x1FFFF, 0xFFFFFFFFU);
    CHECK_SAT(aws_mul_size_saturating, ~0U, ~0U, ~0U);
#elif SIZE_BITS == 64
    CHECK_SAT(aws_mul_size_saturating, 0, 0, 0);
    CHECK_SAT(aws_mul_size_saturating, 0, 1, 0);
    CHECK_SAT(aws_mul_size_saturating, 0, ~0LLU, 0);
    CHECK_SAT(aws_mul_size_saturating, 4, 5, 20);
    CHECK_SAT(aws_mul_size_saturating, 1234, 4321, 5332114);

    CHECK_SAT(aws_mul_size_saturating, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_SAT(aws_mul_size_saturating, 0xFFFFFFFF, 0xFFFFFFFF, 0xfffffffe00000001LLU);
    CHECK_SAT(aws_mul_size_saturating, 0x100000000, 0xFFFFFFFF, 0xFFFFFFFF00000000LLU);
    CHECK_SAT(aws_mul_size_saturating, 0x100000001, 0xFFFFFFFF, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_size_saturating, 0x100000001, 0xFFFFFFFE, 0xFFFFFFFEFFFFFFFELLU);
    CHECK_SAT(aws_mul_size_saturating, 0x100000002, 0xFFFFFFFE, 0xFFFFFFFFFFFFFFFCLLU);
    CHECK_SAT(aws_mul_size_saturating, 0x100000003, 0xFFFFFFFE, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_size_saturating, 0xFFFFFFFE, 0xFFFFFFFE, 0xFFFFFFFC00000004LLU);
    CHECK_SAT(aws_mul_size_saturating, 0x1FFFFFFFF, 0x1FFFFFFFF, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_SAT(aws_mul_size_saturating, ~0LLU, ~0LLU, ~0LLU);
#else
    FAIL("Unexpected size for size_t: %zu", sizeof(size_t));
#endif

    return 0;
}

#define CHECK_OVF_0(fn, type, a, b)                                                                                    \
    do {                                                                                                               \
        type result_val = 0;                                                                                           \
        ASSERT_TRUE(fn((a), (b), &result_val));                                                                        \
    } while (0)

#define CHECK_OVF(fn, type, a, b)                                                                                      \
    do {                                                                                                               \
        CHECK_OVF_0(fn, type, a, b);                                                                                   \
        CHECK_OVF_0(fn, type, b, a);                                                                                   \
    } while (0)

#define CHECK_NO_OVF_0(fn, type, a, b, r)                                                                              \
    do {                                                                                                               \
        type result_val = 0;                                                                                           \
        ASSERT_FALSE(fn((a), (b), &result_val));                                                                       \
        ASSERT_INT_EQUALS(                                                                                             \
            (uint64_t)result_val,                                                                                      \
            (uint64_t)(r),                                                                                             \
            "Expected %s(%016llx, %016llx) = %016llx; got %016llx",                                                    \
            #fn,                                                                                                       \
            (unsigned long long)(a),                                                                                   \
            (unsigned long long)(b),                                                                                   \
            (unsigned long long)(r),                                                                                   \
            (unsigned long long)(result_val));                                                                         \
    } while (0)

#define CHECK_NO_OVF(fn, type, a, b, r)                                                                                \
    do {                                                                                                               \
        CHECK_NO_OVF_0(fn, type, a, b, r);                                                                             \
        CHECK_NO_OVF_0(fn, type, b, a, r);                                                                             \
    } while (0)

AWS_TEST_CASE(test_mul_u64_checked, s_test_mul_u64_checked_fn)
static int s_test_mul_u64_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0, 0, 0);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0, 1, 0);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0, ~0LLU, 0);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 4, 5, 20);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 1234, 4321, 5332114);

    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0xFFFFFFFFLLU, 1LLU, 0xFFFFFFFFLLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0xFFFFFFFFLLU, 0xFFFFFFFFLLU, 0xfffffffe00000001LLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0x100000000LLU, 0xFFFFFFFFLLU, 0xFFFFFFFF00000000LLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0x100000001LLU, 0xFFFFFFFFLLU, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0x100000001LLU, 0xFFFFFFFELLU, 0xFFFFFFFEFFFFFFFELLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0x100000002LLU, 0xFFFFFFFELLU, 0xFFFFFFFFFFFFFFFCLLU);
    CHECK_OVF(aws_mul_u64_checked, uint64_t, 0x100000003LLU, 0xFFFFFFFELLU);
    CHECK_NO_OVF(aws_mul_u64_checked, uint64_t, 0xFFFFFFFELLU, 0xFFFFFFFELLU, 0xFFFFFFFC00000004LLU);
    CHECK_OVF(aws_mul_u64_checked, uint64_t, 0x1FFFFFFFFLLU, 0x1FFFFFFFFLLU);
    CHECK_OVF(aws_mul_u64_checked, uint64_t, ~0LLU, ~0LLU);

    return 0;
}

AWS_TEST_CASE(test_mul_u32_checked, s_test_mul_u32_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_mul_u32_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0, 0, 0);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0, 1, 0);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0, ~0U, 0);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 4, 5, 20);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 1234, 4321, 5332114);

    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0xFFFF, 1, 0xFFFF);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0xFFFF, 0xFFFF, 0xfffe0001U);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0x10000, 0xFFFF, 0xFFFF0000U);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0x10001, 0xFFFF, 0xFFFFFFFFu);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0x10001, 0xFFFE, 0xFFFEFFFEu);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0x10002, 0xFFFE, 0xFFFFFFFCu);
    CHECK_OVF(aws_mul_u32_checked, uint32_t, 0x10003, 0xFFFE);
    CHECK_NO_OVF(aws_mul_u32_checked, uint32_t, 0xFFFE, 0xFFFE, 0xFFFC0004u);
    CHECK_OVF(aws_mul_u32_checked, uint32_t, 0x1FFFF, 0x1FFFF);
    CHECK_OVF(aws_mul_u32_checked, uint32_t, ~0U, ~0U);

    return 0;
}

AWS_TEST_CASE(test_mul_size_checked, s_test_mul_size_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_mul_size_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

#if SIZE_BITS == 32
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, 0, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, 1, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, ~0U, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 4, 5, 20);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 1234, 4321, 5332114);

    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFFFFFF, 1, 0xFFFFFFFF);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFF, 1, 0xFFFF);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFF, 0xFFFF, 0xfffe0001U);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x10000, 0xFFFF, 0xFFFF0000U);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x10001, 0xFFFF, 0xFFFFFFFFu);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x10001, 0xFFFE, 0xFFFEFFFEu);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x10002, 0xFFFE, 0xFFFFFFFCu);
    CHECK_OVF(aws_mul_size_checked, size_t, 0x10003, 0xFFFE);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFE, 0xFFFE, 0xFFFC0004u);
    CHECK_OVF(aws_mul_size_checked, size_t, 0x1FFFF, 0x1FFFF);
    CHECK_OVF(aws_mul_size_checked, size_t, ~0U, ~0U);
#elif SIZE_BITS == 64
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, 0, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, 1, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0, ~0LLU, 0);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 4, 5, 20);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 1234, 4321, 5332114);

    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFFFFFFLLU, 1LLU, 0xFFFFFFFFLLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFFFFFFLLU, 0xFFFFFFFFLLU, 0xfffffffe00000001LLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x100000000LLU, 0xFFFFFFFFLLU, 0xFFFFFFFF00000000LLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x100000001LLU, 0xFFFFFFFFLLU, 0xFFFFFFFFFFFFFFFFLLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x100000001LLU, 0xFFFFFFFELLU, 0xFFFFFFFEFFFFFFFELLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0x100000002LLU, 0xFFFFFFFELLU, 0xFFFFFFFFFFFFFFFCLLU);
    CHECK_OVF(aws_mul_size_checked, size_t, 0x100000003LLU, 0xFFFFFFFELLU);
    CHECK_NO_OVF(aws_mul_size_checked, size_t, 0xFFFFFFFELLU, 0xFFFFFFFELLU, 0xFFFFFFFC00000004LLU);
    CHECK_OVF(aws_mul_size_checked, size_t, 0x1FFFFFFFFLLU, 0x1FFFFFFFFLLU);
    CHECK_OVF(aws_mul_size_checked, size_t, ~0LLU, ~0LLU);
#else
    FAIL("Unexpected size for size_t: %zu", sizeof(size_t));
#endif
    return 0;
}

AWS_TEST_CASE(test_add_size_checked, s_test_add_size_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_size_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

#if SIZE_BITS == 32
    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;
#elif SIZE_BITS == 64
    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;
#else
    FAIL("Unexpected size for size_t: %zu", sizeof(size_t));
#endif

    CHECK_NO_OVF(aws_add_size_checked, size_t, 0, 0, 0);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 0, 1, 1);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 4, 5, 9);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 1234, 4321, 5555);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_size_checked, size_t, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_NO_OVF(aws_add_size_checked, size_t, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_NO_OVF(aws_add_size_checked, size_t, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    CHECK_OVF(aws_add_size_checked, size_t, 1, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, 100, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, HALF_MAX + 1, HALF_MAX + 1);
    CHECK_OVF(aws_add_size_checked, size_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_size_checked, size_t, 100, ACTUAL_MAX - 99);
    CHECK_OVF(aws_add_size_checked, size_t, 100, ACTUAL_MAX - 1);
    return 0;
}

AWS_TEST_CASE(test_sub_size_checked, s_test_sub_size_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_size_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const size_t HALF_MAX = SIZE_MAX / 2;
    const size_t ACTUAL_MAX = SIZE_MAX;

    /* No overflow expected */
    CHECK_NO_OVF(aws_sub_size_checked, size_t, 0, 0, 0);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, 1, 0, 1);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, 9, 4, 5);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, 5555, 1234, 4321);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_NO_OVF(aws_sub_size_checked, size_t, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    /* Overflow expected */
    CHECK_OVF(aws_sub_size_checked, size_t, 0, 1);
    CHECK_OVF(aws_sub_size_checked, size_t, 0, 100);
    CHECK_OVF(aws_sub_size_checked, size_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_size_checked, size_t, 0, ACTUAL_MAX);
    CHECK_OVF(aws_sub_size_checked, size_t, HALF_MAX, HALF_MAX + 1);
    CHECK_OVF(aws_sub_size_checked, size_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_size_checked, size_t, 99, 100);
    CHECK_OVF(aws_sub_size_checked, size_t, 1, 100);
    return 0;
}

#define CHECK_OVF_VARARGS(fn, type, num, ...)                                                                          \
    do {                                                                                                               \
        type result_val;                                                                                               \
        ASSERT_TRUE(fn(num, &result_val, __VA_ARGS__));                                                                \
    } while (0)

#define CHECK_NO_OVF_VARARGS(fn, type, num, r, ...)                                                                    \
    do {                                                                                                               \
        type result_val;                                                                                               \
        ASSERT_FALSE(fn(num, &result_val, __VA_ARGS__));                                                               \
        ASSERT_INT_EQUALS(                                                                                             \
            (uint64_t)result_val,                                                                                      \
            (uint64_t)(r),                                                                                             \
            "From fn %s num %016llx: Expected %016llx; got %016llx",                                                   \
            #fn,                                                                                                       \
            (unsigned long long)(num),                                                                                 \
            (unsigned long long)(r),                                                                                   \
            (unsigned long long)(result_val));                                                                         \
    } while (0)

void print_array(size_t *a, size_t len) {
    for (size_t i = 0; i < len; ++i) {
        fprintf(AWS_TESTING_REPORT_FD, "a: %zu\t%zu\n", i, a[i]);
    }
}

static int check_add_varargs_no_overflow(size_t a, size_t b, size_t r) {
    // Check for 2 inputs
    for (size_t i = 0; i < 2; ++i) {
        for (size_t j = 0; i < 2; ++i) {
            size_t array[2] = {0};
            if (i != j) {
                // fprintf(AWS_TESTING_REPORT_FD,"i: %zu j: %zu\n", i, j);
                array[i] = a;
                array[j] = b;
                // print_array(array,2);

                CHECK_NO_OVF_VARARGS(aws_add_size_checked_varargs, size_t, 2, r, array[0], array[1]);
            }
        }
    }

    // Check for 3 inputs
    for (size_t i = 0; i < 3; ++i) {
        for (size_t j = 0; i < 3; ++i) {
            size_t array[3] = {0};
            if (i != j) {
                array[i] = a;
                array[j] = b;
                CHECK_NO_OVF_VARARGS(aws_add_size_checked_varargs, size_t, 3, r, array[0], array[1], array[2]);
            }
        }
    }

    // Check for 5 inputs
    for (size_t i = 0; i < 5; ++i) {
        for (size_t j = 0; i < 5; ++i) {
            size_t array[5] = {0};
            if (i != j) {
                array[i] = a;
                array[j] = b;
                CHECK_NO_OVF_VARARGS(
                    aws_add_size_checked_varargs, size_t, 5, r, array[0], array[1], array[2], array[3], array[4]);
            }
        }
    }

    return 0;
}

static int check_add_varargs_overflow(size_t a, size_t b) {
    // Check for 2 inputs
    for (size_t i = 0; i < 2; ++i) {
        for (size_t j = 0; i < 2; ++i) {
            size_t array[2] = {0};
            if (i != j) {
                // fprintf(AWS_TESTING_REPORT_FD,"i: %zu j: %zu\n", i, j);
                array[i] = a;
                array[j] = b;
                // print_array(array,2);

                CHECK_OVF_VARARGS(aws_add_size_checked_varargs, size_t, 2, array[0], array[1]);
            }
        }
    }

    // Check for 3 inputs
    for (size_t i = 0; i < 3; ++i) {
        for (size_t j = 0; i < 3; ++i) {
            size_t array[3] = {0};
            if (i != j) {
                array[i] = a;
                array[j] = b;
                CHECK_OVF_VARARGS(aws_add_size_checked_varargs, size_t, 3, array[0], array[1], array[2]);
            }
        }
    }

    // Check for 5 inputs
    for (size_t i = 0; i < 5; ++i) {
        for (size_t j = 0; i < 5; ++i) {
            size_t array[5] = {0};
            if (i != j) {
                array[i] = a;
                array[j] = b;
                CHECK_OVF_VARARGS(
                    aws_add_size_checked_varargs, size_t, 5, array[0], array[1], array[2], array[3], array[4]);
            }
        }
    }

    return 0;
}

AWS_TEST_CASE(test_aws_add_size_checked_varargs, s_test_add_size_checked_varargs_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_size_checked_varargs_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

#if SIZE_BITS == 32
    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;
#elif SIZE_BITS == 64
    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;
#else
    FAIL("Unexpected size for size_t: %zu", sizeof(size_t));
#endif

    ASSERT_SUCCESS(check_add_varargs_no_overflow(0, 0, 0));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(0, 1, 1));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(4, 5, 9));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(1234, 4321, 5555));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(0, ACTUAL_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(HALF_MAX, HALF_MAX, ACTUAL_MAX - 1));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(HALF_MAX + 1, HALF_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(100, ACTUAL_MAX - 102, ACTUAL_MAX - 2));
    ASSERT_SUCCESS(check_add_varargs_no_overflow(100, ACTUAL_MAX - 100, ACTUAL_MAX));

    ASSERT_SUCCESS(check_add_varargs_overflow(1, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(100, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(HALF_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(ACTUAL_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(HALF_MAX + 1, HALF_MAX + 1));
    ASSERT_SUCCESS(check_add_varargs_overflow(HALF_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(HALF_MAX, ACTUAL_MAX));
    ASSERT_SUCCESS(check_add_varargs_overflow(100, ACTUAL_MAX - 99));
    ASSERT_SUCCESS(check_add_varargs_overflow(100, ACTUAL_MAX - 1));

    return 0;
}

AWS_TEST_CASE(test_add_size_saturating, s_test_add_size_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_size_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

#if SIZE_BITS == 32
    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;
#elif SIZE_BITS == 64
    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;
#else
    FAIL("Unexpected size for size_t: %zu", sizeof(size_t));
#endif
    (void)HALF_MAX;
    (void)ACTUAL_MAX;
    /* No overflow expected */
    CHECK_SAT(aws_add_size_saturating, 0, 0, 0);
    CHECK_SAT(aws_add_size_saturating, 0, 1, 1);
    CHECK_SAT(aws_add_size_saturating, 4, 5, 9);
    CHECK_SAT(aws_add_size_saturating, 1234, 4321, 5555);
    CHECK_SAT(aws_add_size_saturating, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_SAT(aws_add_size_saturating, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    /* Overflow expected */
    CHECK_SAT(aws_add_size_saturating, 1, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, 100, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, ACTUAL_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX + 1, HALF_MAX + 1, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, 100, ACTUAL_MAX - 99, ACTUAL_MAX);
    CHECK_SAT(aws_add_size_saturating, 100, ACTUAL_MAX - 1, ACTUAL_MAX);
    return 0;
}

AWS_TEST_CASE(test_sub_size_saturating, s_test_sub_size_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_size_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const size_t HALF_MAX = SIZE_MAX / 2;
    const size_t ACTUAL_MAX = SIZE_MAX;

    /* No overflow expected */
    CHECK_SAT(aws_sub_size_saturating, 0, 0, 0);
    CHECK_SAT(aws_sub_size_saturating, 1, 0, 1);
    CHECK_SAT(aws_sub_size_saturating, 9, 4, 5);
    CHECK_SAT(aws_sub_size_saturating, 5555, 1234, 4321);
    CHECK_SAT(aws_sub_size_saturating, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_SAT(aws_sub_size_saturating, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_SAT(aws_sub_size_saturating, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_SAT(aws_sub_size_saturating, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_SAT(aws_sub_size_saturating, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    /* Overflow expected */
    CHECK_SAT(aws_sub_size_saturating, 0, 1, 0);
    CHECK_SAT(aws_sub_size_saturating, 0, 100, 0);
    CHECK_SAT(aws_sub_size_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_size_saturating, 0, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_size_saturating, HALF_MAX, HALF_MAX + 1, 0);
    CHECK_SAT(aws_sub_size_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_size_saturating, 99, 100, 0);
    CHECK_SAT(aws_sub_size_saturating, 1, 100, 0);
    return 0;
}

AWS_TEST_CASE(test_add_u32_checked, s_test_add_u32_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_u32_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 0, 0, 0);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 0, 1, 1);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 4, 5, 9);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 1234, 4321, 5555);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_NO_OVF(aws_add_u32_checked, uint32_t, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    CHECK_OVF(aws_add_u32_checked, uint32_t, 1, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, 100, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, HALF_MAX + 1, HALF_MAX + 1);
    CHECK_OVF(aws_add_u32_checked, uint32_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u32_checked, uint32_t, 100, ACTUAL_MAX - 99);
    CHECK_OVF(aws_add_u32_checked, uint32_t, 100, ACTUAL_MAX - 1);
    return 0;
}

AWS_TEST_CASE(test_add_u32_saturating, s_test_add_u32_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_u32_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;

    /* No overflow expected */
    CHECK_SAT(aws_add_u32_saturating, 0, 0, 0);
    CHECK_SAT(aws_add_u32_saturating, 0, 1, 1);
    CHECK_SAT(aws_add_u32_saturating, 4, 5, 9);
    CHECK_SAT(aws_add_u32_saturating, 1234, 4321, 5555);
    CHECK_SAT(aws_add_u32_saturating, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_SAT(aws_add_u32_saturating, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    /* Overflow expected */
    CHECK_SAT(aws_add_u32_saturating, 1, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, 100, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, ACTUAL_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX + 1, HALF_MAX + 1, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, 100, ACTUAL_MAX - 99, ACTUAL_MAX);
    CHECK_SAT(aws_add_u32_saturating, 100, ACTUAL_MAX - 1, ACTUAL_MAX);
    return 0;
}

AWS_TEST_CASE(test_sub_u32_checked, s_test_sub_u32_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_u32_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;

    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, 0, 0, 0);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, 1, 0, 1);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, 9, 4, 5);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, 5555, 1234, 4321);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_NO_OVF(aws_sub_u32_checked, uint32_t, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    CHECK_OVF(aws_sub_u32_checked, uint32_t, 0, 1);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, 0, 100);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, 0, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, HALF_MAX, HALF_MAX + 1);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, 99, 100);
    CHECK_OVF(aws_sub_u32_checked, uint32_t, 1, 100);

    return 0;
}

AWS_TEST_CASE(test_sub_u32_saturating, s_test_sub_u32_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_u32_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint32_t HALF_MAX = UINT32_MAX / 2;
    const uint32_t ACTUAL_MAX = UINT32_MAX;

    /* No overflow expected */
    CHECK_SAT(aws_sub_u32_saturating, 0, 0, 0);
    CHECK_SAT(aws_sub_u32_saturating, 1, 0, 1);
    CHECK_SAT(aws_sub_u32_saturating, 9, 4, 5);
    CHECK_SAT(aws_sub_u32_saturating, 5555, 1234, 4321);
    CHECK_SAT(aws_sub_u32_saturating, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_SAT(aws_sub_u32_saturating, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_SAT(aws_sub_u32_saturating, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_SAT(aws_sub_u32_saturating, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_SAT(aws_sub_u32_saturating, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    /* Overflow expected */
    CHECK_SAT(aws_sub_u32_saturating, 0, 1, 0);
    CHECK_SAT(aws_sub_u32_saturating, 0, 100, 0);
    CHECK_SAT(aws_sub_u32_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u32_saturating, 0, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u32_saturating, HALF_MAX, HALF_MAX + 1, 0);
    CHECK_SAT(aws_sub_u32_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u32_saturating, 99, 100, 0);
    CHECK_SAT(aws_sub_u32_saturating, 1, 100, 0);

    return 0;
}

AWS_TEST_CASE(test_add_u64_checked, s_test_add_u64_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_u64_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 0, 0, 0);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 0, 1, 1);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 4, 5, 9);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 1234, 4321, 5555);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_NO_OVF(aws_add_u64_checked, uint64_t, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    CHECK_OVF(aws_add_u64_checked, uint64_t, 1, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, 100, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, HALF_MAX + 1, HALF_MAX + 1);
    CHECK_OVF(aws_add_u64_checked, uint64_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_add_u64_checked, uint64_t, 100, ACTUAL_MAX - 99);
    CHECK_OVF(aws_add_u64_checked, uint64_t, 100, ACTUAL_MAX - 1);
    return 0;
}

AWS_TEST_CASE(test_add_u64_saturating, s_test_add_u64_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_add_u64_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;

    /* No overflow expected */
    CHECK_SAT(aws_add_u64_saturating, 0, 0, 0);
    CHECK_SAT(aws_add_u64_saturating, 0, 1, 1);
    CHECK_SAT(aws_add_u64_saturating, 4, 5, 9);
    CHECK_SAT(aws_add_u64_saturating, 1234, 4321, 5555);
    CHECK_SAT(aws_add_u64_saturating, 0, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX, HALF_MAX, ACTUAL_MAX - 1);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX + 1, HALF_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, 100, ACTUAL_MAX - 102, ACTUAL_MAX - 2);
    CHECK_SAT(aws_add_u64_saturating, 100, ACTUAL_MAX - 100, ACTUAL_MAX);

    /* Overflow expected */
    CHECK_SAT(aws_add_u64_saturating, 1, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, 100, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, ACTUAL_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX + 1, HALF_MAX + 1, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, HALF_MAX, ACTUAL_MAX, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, 100, ACTUAL_MAX - 99, ACTUAL_MAX);
    CHECK_SAT(aws_add_u64_saturating, 100, ACTUAL_MAX - 1, ACTUAL_MAX);
    return 0;
}

AWS_TEST_CASE(test_sub_u64_checked, s_test_sub_u64_checked_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_u64_checked_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;

    /* No overflow expected */
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, 0, 0, 0);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, 1, 0, 1);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, 9, 4, 5);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, 5555, 1234, 4321);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_NO_OVF(aws_sub_u64_checked, uint64_t, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    /* Overflow expected */
    CHECK_OVF(aws_sub_u64_checked, uint64_t, 0, 1);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, 0, 100);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, 0, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, HALF_MAX, HALF_MAX + 1);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, HALF_MAX, ACTUAL_MAX);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, 99, 100);
    CHECK_OVF(aws_sub_u64_checked, uint64_t, 1, 100);

    return 0;
}

AWS_TEST_CASE(test_sub_u64_saturating, s_test_sub_u64_saturating_fn)
/* NOLINTNEXTLINE(readability-function-size) */
static int s_test_sub_u64_saturating_fn(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    const uint64_t HALF_MAX = UINT64_MAX / 2;
    const uint64_t ACTUAL_MAX = UINT64_MAX;

    /* No overflow expected */
    CHECK_SAT(aws_sub_u64_saturating, 0, 0, 0);
    CHECK_SAT(aws_sub_u64_saturating, 1, 0, 1);
    CHECK_SAT(aws_sub_u64_saturating, 9, 4, 5);
    CHECK_SAT(aws_sub_u64_saturating, 5555, 1234, 4321);
    CHECK_SAT(aws_sub_u64_saturating, ACTUAL_MAX, 0, ACTUAL_MAX);
    CHECK_SAT(aws_sub_u64_saturating, ACTUAL_MAX - 1, HALF_MAX, HALF_MAX);
    CHECK_SAT(aws_sub_u64_saturating, ACTUAL_MAX, HALF_MAX + 1, HALF_MAX);
    CHECK_SAT(aws_sub_u64_saturating, ACTUAL_MAX - 2, 100, ACTUAL_MAX - 102);
    CHECK_SAT(aws_sub_u64_saturating, ACTUAL_MAX, 100, ACTUAL_MAX - 100);

    /* Overflow expected */
    CHECK_SAT(aws_sub_u64_saturating, 0, 1, 0);
    CHECK_SAT(aws_sub_u64_saturating, 0, 100, 0);
    CHECK_SAT(aws_sub_u64_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u64_saturating, 0, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u64_saturating, HALF_MAX, HALF_MAX + 1, 0);
    CHECK_SAT(aws_sub_u64_saturating, HALF_MAX, ACTUAL_MAX, 0);
    CHECK_SAT(aws_sub_u64_saturating, 99, 100, 0);
    CHECK_SAT(aws_sub_u64_saturating, 1, 100, 0);
    return 0;
}

AWS_TEST_CASE(test_min_max, s_test_min_max)
static int s_test_min_max(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    ASSERT_UINT_EQUALS(0, aws_min_u8(0, UINT8_MAX));
    ASSERT_UINT_EQUALS(0, aws_min_u8(UINT8_MAX, 0));
    ASSERT_UINT_EQUALS(UINT8_MAX, aws_max_u8(0, UINT8_MAX));
    ASSERT_UINT_EQUALS(UINT8_MAX, aws_max_u8(UINT8_MAX, 0));
    ASSERT_INT_EQUALS(INT8_MIN, aws_min_i8(INT8_MIN, INT8_MAX));
    ASSERT_INT_EQUALS(INT8_MIN, aws_min_i8(INT8_MAX, INT8_MIN));
    ASSERT_INT_EQUALS(INT8_MAX, aws_max_i8(INT8_MIN, INT8_MAX));
    ASSERT_INT_EQUALS(INT8_MAX, aws_max_i8(INT8_MAX, INT8_MIN));

    ASSERT_UINT_EQUALS(0, aws_min_u16(0, UINT16_MAX));
    ASSERT_UINT_EQUALS(0, aws_min_u16(UINT16_MAX, 0));
    ASSERT_UINT_EQUALS(UINT16_MAX, aws_max_u16(0, UINT16_MAX));
    ASSERT_UINT_EQUALS(UINT16_MAX, aws_max_u16(UINT16_MAX, 0));
    ASSERT_INT_EQUALS(INT16_MIN, aws_min_i16(INT16_MIN, INT16_MAX));
    ASSERT_INT_EQUALS(INT16_MIN, aws_min_i16(INT16_MAX, INT16_MIN));
    ASSERT_INT_EQUALS(INT16_MAX, aws_max_i16(INT16_MIN, INT16_MAX));
    ASSERT_INT_EQUALS(INT16_MAX, aws_max_i16(INT16_MAX, INT16_MIN));

    ASSERT_UINT_EQUALS(0, aws_min_u32(0, UINT32_MAX));
    ASSERT_UINT_EQUALS(0, aws_min_u32(UINT32_MAX, 0));
    ASSERT_UINT_EQUALS(UINT32_MAX, aws_max_u32(0, UINT32_MAX));
    ASSERT_UINT_EQUALS(UINT32_MAX, aws_max_u32(UINT32_MAX, 0));
    ASSERT_INT_EQUALS(INT32_MIN, aws_min_i32(INT32_MIN, INT32_MAX));
    ASSERT_INT_EQUALS(INT32_MIN, aws_min_i32(INT32_MAX, INT32_MIN));
    ASSERT_INT_EQUALS(INT32_MAX, aws_max_i32(INT32_MIN, INT32_MAX));
    ASSERT_INT_EQUALS(INT32_MAX, aws_max_i32(INT32_MAX, INT32_MIN));

    ASSERT_UINT_EQUALS(0, aws_min_u64(0, UINT64_MAX));
    ASSERT_UINT_EQUALS(0, aws_min_u64(UINT64_MAX, 0));
    ASSERT_UINT_EQUALS(UINT64_MAX, aws_max_u64(0, UINT64_MAX));
    ASSERT_UINT_EQUALS(UINT64_MAX, aws_max_u64(UINT64_MAX, 0));
    ASSERT_INT_EQUALS(INT64_MIN, aws_min_i64(INT64_MIN, INT64_MAX));
    ASSERT_INT_EQUALS(INT64_MIN, aws_min_i64(INT64_MAX, INT64_MIN));
    ASSERT_INT_EQUALS(INT64_MAX, aws_max_i64(INT64_MIN, INT64_MAX));
    ASSERT_INT_EQUALS(INT64_MAX, aws_max_i64(INT64_MAX, INT64_MIN));

    ASSERT_UINT_EQUALS(0, aws_min_size(0, SIZE_MAX));
    ASSERT_UINT_EQUALS(0, aws_min_size(SIZE_MAX, 0));
    ASSERT_UINT_EQUALS(SIZE_MAX, aws_max_size(0, SIZE_MAX));
    ASSERT_UINT_EQUALS(SIZE_MAX, aws_max_size(SIZE_MAX, 0));

    ASSERT_INT_EQUALS(INT_MIN, aws_min_int(INT_MIN, INT_MAX));
    ASSERT_INT_EQUALS(INT_MIN, aws_min_int(INT_MAX, INT_MIN));
    ASSERT_INT_EQUALS(INT_MAX, aws_max_int(INT_MIN, INT_MAX));
    ASSERT_INT_EQUALS(INT_MAX, aws_max_int(INT_MAX, INT_MIN));

    ASSERT_TRUE(FLT_MIN == aws_min_float(FLT_MIN, FLT_MAX));
    ASSERT_TRUE(FLT_MIN == aws_min_float(FLT_MAX, FLT_MIN));
    ASSERT_TRUE(FLT_MAX == aws_max_float(FLT_MIN, FLT_MAX));
    ASSERT_TRUE(FLT_MAX == aws_max_float(FLT_MAX, FLT_MIN));

    ASSERT_TRUE(DBL_MIN == aws_min_double(DBL_MIN, DBL_MAX));
    ASSERT_TRUE(DBL_MIN == aws_min_double(DBL_MAX, DBL_MIN));
    ASSERT_TRUE(DBL_MAX == aws_max_double(DBL_MIN, DBL_MAX));
    ASSERT_TRUE(DBL_MAX == aws_max_double(DBL_MAX, DBL_MIN));

    return 0;
}

AWS_TEST_CASE(test_clz, s_test_clz)
static int s_test_clz(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    ASSERT_UINT_EQUALS(0, aws_clz_u32(UINT32_MAX));
    ASSERT_UINT_EQUALS(1, aws_clz_u32(INT32_MAX));
    ASSERT_UINT_EQUALS(32, aws_clz_u32(0));

    ASSERT_UINT_EQUALS(0, aws_clz_i32(-1));
    ASSERT_UINT_EQUALS(1, aws_clz_i32(INT32_MAX));
    ASSERT_UINT_EQUALS(32, aws_clz_i32(0));

    ASSERT_UINT_EQUALS(0, aws_clz_u64(UINT64_MAX));
    ASSERT_UINT_EQUALS(1, aws_clz_u64(INT64_MAX));
    ASSERT_UINT_EQUALS(64, aws_clz_u64(0));

    ASSERT_UINT_EQUALS(0, aws_clz_i64(-1));
    ASSERT_UINT_EQUALS(1, aws_clz_i64(INT64_MAX));
    ASSERT_UINT_EQUALS(64, aws_clz_i64(0));

    return 0;
}

AWS_TEST_CASE(test_ctz, s_test_ctz)
static int s_test_ctz(struct aws_allocator *allocator, void *ctx) {
    (void)allocator;
    (void)ctx;

    ASSERT_UINT_EQUALS(0, aws_ctz_u32(1));
    ASSERT_UINT_EQUALS(1, aws_ctz_u32(2));
    ASSERT_UINT_EQUALS(32, aws_ctz_u32(0));

    ASSERT_UINT_EQUALS(0, aws_ctz_i32(1));
    ASSERT_UINT_EQUALS(1, aws_ctz_i32(2));
    ASSERT_UINT_EQUALS(32, aws_ctz_i32(0));

    ASSERT_UINT_EQUALS(0, aws_ctz_u64(1));
    ASSERT_UINT_EQUALS(1, aws_ctz_u64(2));
    ASSERT_UINT_EQUALS(64, aws_ctz_u64(0));

    ASSERT_UINT_EQUALS(0, aws_ctz_i64(1));
    ASSERT_UINT_EQUALS(1, aws_ctz_i64(2));
    ASSERT_UINT_EQUALS(64, aws_ctz_i64(0));

    return 0;
}
