1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/*
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
*
* SPDX-License-Identifier: MPL-2.0
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
*
* See the COPYRIGHT file distributed with this work for additional
* information regarding copyright ownership.
*/
/* ! \file */
#if HAVE_CMOCKA
#include <sched.h> /* IWYU pragma: keep */
#include <setjmp.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#define UNIT_TESTING
#include <cmocka.h>
#include <isc/safe.h>
#include <isc/util.h>
/* test isc_safe_memequal() */
static void
isc_safe_memequal_test(void **state) {
UNUSED(state);
assert_true(isc_safe_memequal("test", "test", 4));
assert_true(!isc_safe_memequal("test", "tesc", 4));
assert_true(
isc_safe_memequal("\x00\x00\x00\x00", "\x00\x00\x00\x00", 4));
assert_true(
!isc_safe_memequal("\x00\x00\x00\x00", "\x00\x00\x00\x01", 4));
assert_true(
!isc_safe_memequal("\x00\x00\x00\x02", "\x00\x00\x00\x00", 4));
}
/* test isc_safe_memwipe() */
static void
isc_safe_memwipe_test(void **state) {
UNUSED(state);
/* These should pass. */
isc_safe_memwipe(NULL, 0);
isc_safe_memwipe((void *)-1, 0);
/*
* isc_safe_memwipe(ptr, size) should function same as
* memset(ptr, 0, size);
*/
{
char buf1[4] = { 1, 2, 3, 4 };
char buf2[4] = { 1, 2, 3, 4 };
isc_safe_memwipe(buf1, sizeof(buf1));
memset(buf2, 0, sizeof(buf2));
assert_int_equal(memcmp(buf1, buf2, sizeof(buf1)), 0);
}
/*
* Boundary test.
*/
{
char buf1[4] = { 1, 2, 3, 4 };
char buf2[4] = { 1, 2, 3, 4 };
/*
* We wipe 3 elements on purpose, keeping the 4th in
* place.
*/
isc_safe_memwipe(buf1, 3);
memset(buf2, 0, 3);
assert_int_equal(memcmp(buf1, buf2, sizeof(buf1)), 0);
}
}
int
main(void) {
const struct CMUnitTest tests[] = {
cmocka_unit_test(isc_safe_memequal_test),
cmocka_unit_test(isc_safe_memwipe_test),
};
return (cmocka_run_group_tests(tests, NULL, NULL));
}
#else /* HAVE_CMOCKA */
#include <stdio.h>
int
main(void) {
printf("1..0 # Skipped: cmocka not available\n");
return (SKIPPED_TEST_EXIT_CODE);
}
#endif /* if HAVE_CMOCKA */
|
