lib:util: Add test for talloc_keep_secret()
[nivanova/samba-autobuild/.git] / lib / util / tests / test_talloc_keep_secret.c
1 #include <stdarg.h>
2 #include <stddef.h>
3 #include <stdint.h>
4 #include <setjmp.h>
5 #include <cmocka.h>
6
7 #include <string.h>
8 #include <talloc.h>
9 #include "lib/util/talloc_keep_secret.h"
10
11 int rep_memset_s(void *dest, size_t destsz, int ch, size_t count);
12
13 int rep_memset_s(void *dest, size_t destsz, int ch, size_t count)
14 {
15         check_expected_ptr(dest);
16         check_expected(destsz);
17         check_expected(ch);
18         check_expected(count);
19
20         return 0;
21 }
22
23 static void test_talloc_keep_secret(void ** state)
24 {
25         TALLOC_CTX *pool = NULL;
26         char *ptr1 = NULL;
27         char *ptr2 = NULL;
28         const char *ptr1_talloc_name = NULL;
29         size_t ptr1_size;
30         size_t i;
31
32         pool = talloc_pool(NULL, 256);
33         assert_non_null(pool);
34
35         ptr1 = talloc_strdup(pool, "secret");
36         assert_non_null(ptr1);
37         assert_string_equal(ptr1, "secret");
38
39         talloc_keep_secret(ptr1);
40
41         ptr1_talloc_name = talloc_get_name(ptr1);
42         assert_string_equal(ptr1_talloc_name, "ptr1");
43
44         ptr1_size = talloc_get_size(ptr1);
45         assert_int_equal(ptr1_size, strlen(ptr1) + 1);
46
47         expect_string(rep_memset_s, dest, "secret");
48         expect_value(rep_memset_s, destsz, strlen(ptr1) + 1);
49         expect_value(rep_memset_s, ch, (int)'\0');
50         expect_value(rep_memset_s, count, strlen(ptr1) + 1);
51
52         talloc_free(ptr1);
53
54         ptr2 = talloc_size(pool, ptr1_size);
55         assert_ptr_equal(ptr1, ptr2);
56
57         for (i = 1; i < ptr1_size; i++) {
58                 assert_int_not_equal(ptr2[0], ptr2[i]);
59         }
60
61         talloc_free(pool);
62 }
63
64 static void test_talloc_keep_secret_validate_memset(void **state)
65 {
66         TALLOC_CTX *mem_ctx = NULL;
67         char *password = NULL;
68
69         mem_ctx = talloc_new(NULL);
70         assert_non_null(mem_ctx);
71
72         password = talloc_strdup(mem_ctx, "secret");
73         assert_non_null(password);
74         talloc_keep_secret(password);
75
76         expect_string(rep_memset_s, dest, "secret");
77         expect_value(rep_memset_s, destsz, strlen(password) + 1);
78         expect_value(rep_memset_s, ch, (int)'\0');
79         expect_value(rep_memset_s, count, strlen(password) + 1);
80
81         talloc_free(mem_ctx);
82 }
83
84 int main(void)
85 {
86     const struct CMUnitTest tests[] = {
87         cmocka_unit_test(test_talloc_keep_secret),
88         cmocka_unit_test(test_talloc_keep_secret_validate_memset),
89     };
90
91     cmocka_set_message_output(CM_OUTPUT_SUBUNIT);
92
93     return cmocka_run_group_tests(tests, NULL, NULL);
94 }