LCOV - code coverage report
Current view: top level - egg - test-secmem.c (source / functions) Coverage Total Hit
Test: coverage.info Lines: 97.2 % 143 139
Test Date: 2024-05-07 18:02:03 Functions: 100.0 % 16 16

            Line data    Source code
       1              : /* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
       2              : /* unit-test-secmem.c: Test low level secure memory allocation functionality
       3              : 
       4              :    Copyright (C) 2007 Stefan Walter
       5              : 
       6              :    The Gnome Keyring Library is free software; you can redistribute it and/or
       7              :    modify it under the terms of the GNU Library General Public License as
       8              :    published by the Free Software Foundation; either version 2 of the
       9              :    License, or (at your option) any later version.
      10              : 
      11              :    The Gnome Keyring Library is distributed in the hope that it will be useful,
      12              :    but WITHOUT ANY WARRANTY; without even the implied warranty of
      13              :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      14              :    Library General Public License for more details.
      15              : 
      16              :    You should have received a copy of the GNU Library General Public
      17              :    License along with the Gnome Library; see the file COPYING.LIB.  If not,
      18              :    <http://www.gnu.org/licenses/>.
      19              : 
      20              :    Author: Stef Walter <stef@memberwebs.com>
      21              : */
      22              : 
      23              : #include "config.h"
      24              : 
      25              : #include "egg/egg-secure-memory.h"
      26              : 
      27              : #include <glib.h>
      28              : 
      29              : #include <stdlib.h>
      30              : #include <stdio.h>
      31              : #include <string.h>
      32              : 
      33              : 
      34       433936 : EGG_SECURE_DEFINE_GLIB_GLOBALS ();
      35              : 
      36              : /* Declared in egg-secure-memory.c */
      37              : extern int egg_secure_warnings;
      38              : 
      39        66729 : EGG_SECURE_DECLARE (tests);
      40              : 
      41              : /*
      42              :  * Each test looks like (on one line):
      43              :  *     void unit_test_xxxxx (CuTest* cu)
      44              :  *
      45              :  * Each setup looks like (on one line):
      46              :  *     void unit_setup_xxxxx (void);
      47              :  *
      48              :  * Each teardown looks like (on one line):
      49              :  *     void unit_teardown_xxxxx (void);
      50              :  *
      51              :  * Tests be run in the order specified here.
      52              :  */
      53              : 
      54              : static gsize
      55            7 : find_non_zero (gpointer mem, gsize len)
      56              : {
      57              :         guchar *b, *e;
      58            7 :         gsize sz = 0;
      59        34523 :         for (b = (guchar*)mem, e = ((guchar*)mem) + len; b != e; ++b, ++sz) {
      60        34516 :                 if (*b != 0x00)
      61            0 :                         return sz;
      62              :         }
      63              : 
      64            7 :         return G_MAXSIZE;
      65              : }
      66              : 
      67              : static void
      68            1 : test_alloc_free (void)
      69              : {
      70              :         gpointer p;
      71              :         gboolean ret;
      72              : 
      73            1 :         p = egg_secure_alloc_full ("tests", 512, 0);
      74            1 :         g_assert (p != NULL);
      75            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p, 512));
      76              : 
      77            1 :         memset (p, 0x67, 512);
      78              : 
      79            1 :         ret = egg_secure_check (p);
      80            1 :         g_assert (ret == TRUE);
      81              : 
      82            1 :         egg_secure_free_full (p, 0);
      83            1 : }
      84              : 
      85              : static void
      86            1 : test_realloc_across (void)
      87              : {
      88              :         gpointer p, p2;
      89              : 
      90              :         /* Tiny allocation */
      91            1 :         p = egg_secure_realloc_full ("tests", NULL, 1088, 0);
      92            1 :         g_assert (p != NULL);
      93            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p, 1088));
      94              : 
      95              :         /* Reallocate to a large one, will have to have changed blocks */
      96            1 :         p2 = egg_secure_realloc_full ("tests", p, 16200, 0);
      97            1 :         g_assert (p2 != NULL);
      98            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p2, 16200));
      99              : 
     100            1 :         egg_secure_free (p2);
     101            1 : }
     102              : 
     103              : static void
     104            1 : test_alloc_two (void)
     105              : {
     106              :         gpointer p, p2;
     107              :         gboolean ret;
     108              : 
     109            1 :         p2 = egg_secure_alloc_full ("tests", 4, 0);
     110            1 :         g_assert (p2 != NULL);
     111            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p2, 4));
     112              : 
     113            1 :         memset (p2, 0x67, 4);
     114              : 
     115            1 :         p = egg_secure_alloc_full ("tests", 16200, 0);
     116            1 :         g_assert (p != NULL);
     117            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p, 16200));
     118              : 
     119            1 :         memset (p, 0x67, 16200);
     120              : 
     121            1 :         ret = egg_secure_check (p);
     122            1 :         g_assert (ret == TRUE);
     123              : 
     124            1 :         egg_secure_free_full (p2, 0);
     125            1 :         egg_secure_free_full (p, 0);
     126            1 : }
     127              : 
     128              : static void
     129            1 : test_realloc (void)
     130              : {
     131            1 :         gchar *str = "a test string to see if realloc works properly";
     132              :         gpointer p, p2;
     133              :         gsize len;
     134              : 
     135            1 :         len = strlen (str) + 1;
     136              : 
     137            1 :         p = egg_secure_realloc_full ("tests", NULL, len, 0);
     138            1 :         g_assert (p != NULL);
     139            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (p, len));
     140              : 
     141            1 :         strcpy ((gchar*)p, str);
     142              : 
     143            1 :         p2 = egg_secure_realloc_full ("tests", p, 512, 0);
     144            1 :         g_assert (p2 != NULL);
     145            1 :         g_assert_cmpint (G_MAXSIZE, ==, find_non_zero (((gchar*)p2) + len, 512 - len));
     146              : 
     147            1 :         g_assert (strcmp (p2, str) == 0);
     148              : 
     149            1 :         p = egg_secure_realloc_full ("tests", p2, 0, 0);
     150            1 :         g_assert (p == NULL);
     151            1 : }
     152              : 
     153              : static void
     154            1 : test_multialloc (void)
     155              : {
     156              :         GPtrArray *memory;
     157              :         gpointer data;
     158              :         gsize size;
     159              :         int i, action, index;
     160              : 
     161              :         /* A predetermined seed to get a predetermined pattern */
     162            1 :         g_random_set_seed (15);
     163            1 :         memory = g_ptr_array_new ();
     164              : 
     165              :         /* Don't print "can't allocate" warnings */
     166            1 :         egg_secure_warnings = 0;
     167              : 
     168       100174 :         for (i = 0; TRUE; ++i) {
     169              : 
     170              :                 /* Determine what we want to do */
     171       100174 :                 if (memory->len > 0) {
     172       100018 :                         if (i > 100000) /* Once we've done 100000 alocations start freeing */
     173          173 :                                 action = 2;
     174              :                         else
     175        99845 :                                 action = g_random_int_range (0, 3);
     176              :                 } else {
     177          156 :                         action = 0; /* No allocations, so allocate */
     178              :                 }
     179              : 
     180       100174 :                 switch (action) {
     181        33449 :                 case 0: /* Allocate some memory */
     182        33449 :                         size = g_random_int_range (1, 16384);
     183        33449 :                         data = egg_secure_alloc (size);
     184        33449 :                         g_assert (data != NULL);
     185        33449 :                         memset (data, 0xCAFEBABE, size);
     186        33449 :                         g_ptr_array_add (memory, data);
     187        33449 :                         break;
     188        33276 :                 case 1: /* Reallocate some memory */
     189        33276 :                         index = g_random_int_range (0, memory->len);
     190        33276 :                         data = g_ptr_array_index (memory, index);
     191        33276 :                         g_assert (data != NULL);
     192        33276 :                         size = g_random_int_range (1, 16384);
     193        33276 :                         data = egg_secure_realloc (data, size);
     194        33276 :                         g_assert (data != NULL);
     195        33276 :                         memset (data, 0xCAFEBABE, size);
     196        33276 :                         g_ptr_array_index (memory, index) = data;
     197        33276 :                         break;
     198        33449 :                 case 2: /* Free some memory */
     199        33449 :                         index = g_random_int_range (0, memory->len);
     200        33449 :                         data = g_ptr_array_remove_index_fast (memory, index);
     201        33449 :                         g_assert (data != NULL);
     202        33449 :                         egg_secure_free (data);
     203        33449 :                         break;
     204            0 :                 default:
     205            0 :                         g_assert_not_reached ();
     206              :                 }
     207              : 
     208       100174 :                 egg_secure_validate ();
     209              : 
     210       100174 :                 if (i > 100000 && !memory->len)
     211            1 :                         break;
     212              :         }
     213              : 
     214            1 :         g_assert (memory->len == 0);
     215            1 :         for (i = 0; i < memory->len; i++)
     216            0 :                 egg_secure_free (memory->pdata[i]);
     217            1 :         g_ptr_array_free (memory, TRUE);
     218              : 
     219            1 :         egg_secure_warnings = 1;
     220            1 : }
     221              : 
     222              : static void
     223            1 : test_clear (void)
     224              : {
     225              :         gpointer p;
     226              : 
     227            1 :         p = egg_secure_alloc_full ("tests", 188, 0);
     228            1 :         g_assert (p != NULL);
     229            1 :         memset (p, 0x89, 188);
     230            1 :         g_assert (memchr (p, 0x89, 188) == p);
     231              : 
     232            1 :         egg_secure_clear (p, 188);
     233            1 :         g_assert (memchr (p, 0x89, 188) == NULL);
     234              : 
     235            1 :         egg_secure_free_full (p, 0);
     236            1 : }
     237              : 
     238              : static void
     239            1 : test_strclear (void)
     240              : {
     241              :         gchar *str;
     242              : 
     243            1 :         str = egg_secure_strdup ("secret");
     244            1 :         g_assert (str != NULL);
     245            1 :         g_assert_cmpuint (strlen (str), ==, 6);
     246            1 :         g_assert (strchr (str, 't') == str + 5);
     247              : 
     248            1 :         egg_secure_strclear (str);
     249            1 :         g_assert_cmpuint (strlen (str), ==, 6);
     250            1 :         g_assert (strchr (str, 't') == NULL);
     251              : 
     252            1 :         egg_secure_free_full (str, 0);
     253            1 : }
     254              : 
     255              : static void
     256            1 : test_strndup (void)
     257              : {
     258              :         gchar *str;
     259              : 
     260            1 :         str = egg_secure_strndup ("secret", 6);
     261            1 :         g_assert (str != NULL);
     262            1 :         g_assert_cmpuint (strlen (str), ==, 6);
     263              : 
     264            1 :         egg_secure_free_full (str, 0);
     265              : 
     266            1 :         str = egg_secure_strndup ("secret", 10);
     267            1 :         g_assert (str != NULL);
     268            1 :         g_assert_cmpuint (strlen (str), ==, 6);
     269              : 
     270            1 :         egg_secure_free_full (str, 0);
     271              : 
     272            1 :         str = egg_secure_strndup ("long secret", 6);
     273            1 :         g_assert (str != NULL);
     274            1 :         g_assert_cmpuint (strlen (str), ==, 6);
     275              : 
     276            1 :         egg_secure_free_full (str, 0);
     277            1 : }
     278              : 
     279              : int
     280            1 : main (int argc, char **argv)
     281              : {
     282            1 :         g_test_init (&argc, &argv, NULL);
     283              : 
     284            1 :         g_test_add_func ("/secmem/alloc_free", test_alloc_free);
     285            1 :         g_test_add_func ("/secmem/realloc_across", test_realloc_across);
     286            1 :         g_test_add_func ("/secmem/alloc_two", test_alloc_two);
     287            1 :         g_test_add_func ("/secmem/realloc", test_realloc);
     288            1 :         g_test_add_func ("/secmem/multialloc", test_multialloc);
     289            1 :         g_test_add_func ("/secmem/clear", test_clear);
     290            1 :         g_test_add_func ("/secmem/strclear", test_strclear);
     291            1 :         g_test_add_func ("/secmem/strndup", test_strndup);
     292              : 
     293            1 :         return g_test_run ();
     294              : }
        

Generated by: LCOV version 2.0-1