src/datacache/test_datacache.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2006, 2009, 2010 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20 /*
  21  * @file datacache/test_datacache.c
  22  * @brief Test for the datacache implementations.
  23  * @author Nils Durner
  24  */
  25 #include "platform.h"
  26 #include "gnunet_util_lib.h"
  27 #include "gnunet_datacache_lib.h"
  28
  29 #define VERBOSE GNUNET_EXTRA_LOGGING
  30
  31 #define ASSERT(x) do { if (! (x)) { printf("Error at %s:%d\n", __FILE__, __LINE__); goto FAILURE;} } while (0)
  32
  33 static int ok;
  34
  35 /**
  36  * Name of plugin under test.
  37  */
  38 static const char *plugin_name;
  39
  40
  41 static int
  42 checkIt (void *cls, struct GNUNET_TIME_Absolute exp,
  43          const GNUNET_HashCode * key, size_t size, const char *data,
  44          enum GNUNET_BLOCK_Type type)
  45 {
  46   if (size != sizeof (GNUNET_HashCode))
  47   {
  48     printf ("ERROR: Invalid size\n");
  49     ok = 2;
  50   }
  51   if (0 != memcmp (data, cls, size))
  52   {
  53     printf ("ERROR: Invalid data\n");
  54     ok = 3;
  55   }
  56   return GNUNET_OK;
  57 }
  58
  59
  60 static void
  61 run (void *cls, char *const *args, const char *cfgfile,
  62      const struct GNUNET_CONFIGURATION_Handle *cfg)
  63 {
  64   struct GNUNET_DATACACHE_Handle *h;
  65   GNUNET_HashCode k;
  66   GNUNET_HashCode n;
  67   struct GNUNET_TIME_Absolute exp;
  68   unsigned int i;
  69
  70   ok = 0;
  71   h = GNUNET_DATACACHE_create (cfg, "testcache");
  72   if (h == NULL)
  73   {
  74     fprintf (stderr,
  75              "Failed to initialize datacache.  Database likely not setup, skipping test.\n");
  76     return;
  77   }
  78   exp = GNUNET_TIME_absolute_get ();
  79   exp.abs_value += 5 * 60 * 1000;
  80   memset (&k, 0, sizeof (GNUNET_HashCode));
  81   for (i = 0; i < 100; i++)
  82   {
  83     GNUNET_CRYPTO_hash (&k, sizeof (GNUNET_HashCode), &n);
  84     ASSERT (GNUNET_OK ==
  85             GNUNET_DATACACHE_put (h, &k, sizeof (GNUNET_HashCode),
  86                                   (const char *) &n, 1 + i % 16, exp));
  87     k = n;
  88   }
  89   memset (&k, 0, sizeof (GNUNET_HashCode));
  90   for (i = 0; i < 100; i++)
  91   {
  92     GNUNET_CRYPTO_hash (&k, sizeof (GNUNET_HashCode), &n);
  93     ASSERT (1 == GNUNET_DATACACHE_get (h, &k, 1 + i % 16, &checkIt, &n));
  94     k = n;
  95   }
  96
  97   memset (&k, 42, sizeof (GNUNET_HashCode));
  98   GNUNET_CRYPTO_hash (&k, sizeof (GNUNET_HashCode), &n);
  99   ASSERT (GNUNET_OK ==
 100           GNUNET_DATACACHE_put (h, &k, sizeof (GNUNET_HashCode),
 101                                 (const char *) &n, 792,
 102                                 GNUNET_TIME_UNIT_FOREVER_ABS));
 103   ASSERT (0 != GNUNET_DATACACHE_get (h, &k, 792, &checkIt, &n));
 104
 105   GNUNET_DATACACHE_destroy (h);
 106   ASSERT (ok == 0);
 107   return;
 108 FAILURE:
 109   if (h != NULL)
 110     GNUNET_DATACACHE_destroy (h);
 111   ok = GNUNET_SYSERR;
 112 }
 113
 114
 115 int
 116 main (int argc, char *argv[])
 117 {
 118   char *pos;
 119   char cfg_name[128];
 120
 121   char *const xargv[] = {
 122     "test-datacache",
 123     "-c",
 124     cfg_name,
 125 #if VERBOSE
 126     "-L", "DEBUG",
 127 #endif
 128     NULL
 129   };
 130   struct GNUNET_GETOPT_CommandLineOption options[] = {
 131     GNUNET_GETOPT_OPTION_END
 132   };
 133
 134   GNUNET_log_setup ("test-datacache",
 135 #if VERBOSE
 136                     "DEBUG",
 137 #else
 138                     "WARNING",
 139 #endif
 140                     NULL);
 141   /* determine name of plugin to use */
 142   plugin_name = argv[0];
 143   while (NULL != (pos = strstr (plugin_name, "_")))
 144     plugin_name = pos + 1;
 145   if (NULL != (pos = strstr (plugin_name, ".")))
 146     pos[0] = 0;
 147   else
 148     pos = (char *) plugin_name;
 149
 150   GNUNET_snprintf (cfg_name, sizeof (cfg_name), "test_datacache_data_%s.conf",
 151                    plugin_name);
 152   if (pos != plugin_name)
 153     pos[0] = '.';
 154   GNUNET_PROGRAM_run ((sizeof (xargv) / sizeof (char *)) - 1, xargv,
 155                       "test-datacache", "nohelp", options, &run, NULL);
 156   if (ok != 0)
 157     fprintf (stderr, "Missed some testcases: %d\n", ok);
 158   return ok;
 159 }
 160
 161 /* end of test_datacache.c */