mirror of
https://github.com/shadow-maint/shadow.git
synced 2026-01-26 22:12:26 +00:00
0d9193dcf6
The 'T' in the name notes that this API is a type-safe variant of the API it wraps. This makes the names more explicative. Signed-off-by: Alejandro Colomar <alx@kernel.org>
97 lines
1.5 KiB
C
97 lines
1.5 KiB
C
// SPDX-FileCopyrightText: 2023-2025, Alejandro Colomar <alx@kernel.org>
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
|
|
#include "alloc/malloc.h"
|
|
|
|
#include <setjmp.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <stdarg.h> // Required by <cmocka.h>
|
|
#include <stddef.h> // Required by <cmocka.h>
|
|
#include <setjmp.h> // Required by <cmocka.h>
|
|
#include <stdint.h> // Required by <cmocka.h>
|
|
#include <cmocka.h>
|
|
|
|
#include "sizeof.h"
|
|
|
|
|
|
#define assert_unreachable() assert_true(0)
|
|
|
|
#define EXIT_CALLED (42)
|
|
|
|
|
|
static jmp_buf jmpb;
|
|
|
|
|
|
void __wrap_exit(int);
|
|
|
|
static void test_exit_if_null_exit(void **);
|
|
static void test_exit_if_null_ok(void **);
|
|
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test(test_exit_if_null_exit),
|
|
cmocka_unit_test(test_exit_if_null_ok),
|
|
};
|
|
|
|
return cmocka_run_group_tests(tests, NULL, NULL);
|
|
}
|
|
|
|
|
|
void
|
|
__wrap_exit(int)
|
|
{
|
|
longjmp(jmpb, EXIT_CALLED);
|
|
}
|
|
|
|
|
|
static void
|
|
test_exit_if_null_exit(void **)
|
|
{
|
|
char *volatile p;
|
|
|
|
p = "before";
|
|
|
|
switch (setjmp(jmpb)) {
|
|
case 0:
|
|
p = "called";
|
|
p = xmalloc_T(SIZE_MAX, char);
|
|
assert_unreachable();
|
|
break;
|
|
case EXIT_CALLED:
|
|
assert_string_equal(p, "called");
|
|
p = "test_ok";
|
|
break;
|
|
default:
|
|
assert_unreachable();
|
|
break;
|
|
}
|
|
|
|
assert_string_equal(p, "test_ok");
|
|
}
|
|
|
|
|
|
static void
|
|
test_exit_if_null_ok(void **)
|
|
{
|
|
char *p;
|
|
|
|
static const char foo[] = "foo1bar";
|
|
|
|
p = xmalloc_T(countof(foo), char);
|
|
assert_true(p != NULL);
|
|
strcpy(p, foo);
|
|
assert_string_equal(p, "foo1bar");
|
|
free(p);
|
|
|
|
p = xmalloc_T(0, char);
|
|
assert_true(p != NULL);
|
|
free(p);
|
|
}
|