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 }