- Changes for long churn (test with 10 peers)
[oweals/gnunet.git] / src / namestore / test_plugin_namestore.c
1 /*
2      This file is part of GNUnet.
3      (C) 2012 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 namestore/test_plugin_namestore.c
22  * @brief Test for the namestore plugins
23  * @author Christian Grothoff
24  */
25 #include "platform.h"
26 #include "gnunet_util_lib.h"
27 #include "gnunet_namestore_plugin.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 /**
42  * Function called when the service shuts down.  Unloads our namestore
43  * plugin.
44  *
45  * @param api api to unload
46  */
47 static void
48 unload_plugin (struct GNUNET_NAMESTORE_PluginFunctions *api)
49 {
50   char *libname;
51
52   GNUNET_asprintf (&libname, "libgnunet_plugin_namestore_%s", plugin_name);
53   GNUNET_break (NULL == GNUNET_PLUGIN_unload (libname, api));
54   GNUNET_free (libname);
55 }
56
57
58 /**
59  * Load the namestore plugin.
60  *
61  * @param cfg configuration to pass
62  * @return NULL on error
63  */
64 static struct GNUNET_NAMESTORE_PluginFunctions *
65 load_plugin (const struct GNUNET_CONFIGURATION_Handle *cfg)
66 {
67   struct GNUNET_NAMESTORE_PluginFunctions *ret;
68   char *libname;
69
70   GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Loading `%s' namestore plugin\n"),
71               plugin_name);
72   GNUNET_asprintf (&libname, "libgnunet_plugin_namestore_%s", plugin_name);
73   if (NULL == (ret = GNUNET_PLUGIN_load (libname, (void*) cfg)))
74   {
75     FPRINTF (stderr, "Failed to load plugin `%s'!\n", plugin_name);
76     return NULL;
77   }
78   GNUNET_free (libname);
79   return ret;
80 }
81
82
83 /**
84  * Function called by for each matching record.
85  *
86  * @param cls closure
87  * @param zone_key public key of the zone
88  * @param expire when does the corresponding block in the DHT expire (until
89  *               when should we never do a DHT lookup for the same name again)?
90  * @param name name that is being mapped (at most 255 characters long)
91  * @param rd_count number of entries in 'rd' array
92  * @param rd array of records with data to store
93  * @param signature signature of the record block, NULL if signature is unavailable (i.e. 
94  *        because the user queried for a particular record type only)
95  */
96 static void 
97 test_record (void *cls,
98              const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key,
99              struct GNUNET_TIME_Absolute expire,
100              const char *name,
101              unsigned int rd_count,
102              const struct GNUNET_NAMESTORE_RecordData *rd,
103              const struct GNUNET_CRYPTO_RsaSignature *signature)
104 {
105 }
106
107
108 static void
109 get_record (struct GNUNET_NAMESTORE_PluginFunctions *nsp, int id)
110 {
111   GNUNET_assert (1 == nsp->iterate_records (nsp->cls,
112                                             NULL, NULL, 0,
113                                             &test_record, &id));
114 }
115
116
117 static void
118 put_record (struct GNUNET_NAMESTORE_PluginFunctions *nsp, int id)
119 {
120 }
121
122
123 static void
124 run (void *cls, char *const *args, const char *cfgfile,
125      const struct GNUNET_CONFIGURATION_Handle *cfg)
126 {
127   struct GNUNET_NAMESTORE_PluginFunctions *nsp;  
128   GNUNET_HashCode zone;
129   
130   ok = 0;
131   nsp = load_plugin (cfg);
132   if (NULL == nsp)
133   {
134     FPRINTF (stderr,
135              "%s", 
136              "Failed to initialize namestore.  Database likely not setup, skipping test.\n");
137     return;
138   }
139   put_record (nsp, 1);
140   get_record (nsp, 1);
141
142   memset (&zone, 42, sizeof (zone));  
143   nsp->delete_zone (nsp->cls, &zone);
144   unload_plugin (nsp);
145 }
146
147
148 int
149 main (int argc, char *argv[])
150 {
151   char *pos;
152   char cfg_name[128];
153
154   char *const xargv[] = {
155     "test-plugin-namestore",
156     "-c",
157     cfg_name,
158 #if VERBOSE
159     "-L", "DEBUG",
160 #endif
161     NULL
162   };
163   struct GNUNET_GETOPT_CommandLineOption options[] = {
164     GNUNET_GETOPT_OPTION_END
165   };
166
167   GNUNET_DISK_directory_remove ("/tmp/gnunet-test-plugin-namestore-sqlite");
168   GNUNET_log_setup ("test-plugin-namestore",
169 #if VERBOSE
170                     "DEBUG",
171 #else
172                     "WARNING",
173 #endif
174                     NULL);
175   /* determine name of plugin to use */
176   plugin_name = argv[0];
177   while (NULL != (pos = strstr (plugin_name, "_")))
178     plugin_name = pos + 1;
179   if (NULL != (pos = strstr (plugin_name, ".")))
180     pos[0] = 0;
181   else
182     pos = (char *) plugin_name;
183
184   GNUNET_snprintf (cfg_name, sizeof (cfg_name), "test_plugin_namestore_%s.conf",
185                    plugin_name);
186   if (pos != plugin_name)
187     pos[0] = '.';
188   GNUNET_PROGRAM_run ((sizeof (xargv) / sizeof (char *)) - 1, xargv,
189                       "test-plugin-namestore", "nohelp", options, &run, NULL);
190   if (ok != 0)
191     FPRINTF (stderr, "Missed some testcases: %d\n", ok);
192   GNUNET_DISK_directory_remove ("/tmp/gnunet-test-plugin-namestore-sqlite");
193   return ok;
194 }
195
196 /* end of test_plugin_namestore.c */