-check NULL, check RV
[oweals/gnunet.git] / src / namestore / test_namestore_api_zone_to_name.c
1 /*
2      This file is part of GNUnet.
3      (C) 2009 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_namestore_api_zone_to_name.c
22  * @brief testcase for zone to name translation
23  */
24 #include "platform.h"
25 #include "gnunet_namestore_service.h"
26 #include "gnunet_testing_lib.h"
27 #include "namestore.h"
28
29 #define RECORDS 5
30
31 #define TEST_RECORD_TYPE 1234
32
33 #define TEST_RECORD_DATALEN 123
34
35 #define TEST_RECORD_DATA 'a'
36
37 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 100)
38
39
40 static struct GNUNET_NAMESTORE_Handle * nsh;
41
42 static GNUNET_SCHEDULER_TaskIdentifier endbadly_task;
43
44 static struct GNUNET_CRYPTO_EcdsaPrivateKey * privkey;
45
46 static struct GNUNET_CRYPTO_EcdsaPublicKey pubkey;
47
48 static struct GNUNET_TIME_Absolute expire;
49
50 static struct GNUNET_CRYPTO_ShortHashCode s_zone;
51
52 static struct GNUNET_CRYPTO_ShortHashCode s_zone_value;
53
54 static char * s_name;
55
56 static struct GNUNET_CRYPTO_EcdsaSignature *s_signature;
57
58 static int res;
59
60 static char *directory;
61
62 /**
63  * Re-establish the connection to the service.
64  *
65  * @param cls handle to use to re-connect.
66  * @param tc scheduler context
67  */
68 static void
69 endbadly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
70 {
71   if (nsh != NULL)
72     GNUNET_NAMESTORE_disconnect (nsh);
73   nsh = NULL;
74   if (privkey != NULL)
75     GNUNET_free (privkey);
76   privkey = NULL;
77   res = 1;
78 }
79
80
81 static void
82 end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
83 {
84   if (endbadly_task != GNUNET_SCHEDULER_NO_TASK)
85   {
86     GNUNET_SCHEDULER_cancel (endbadly_task);
87     endbadly_task = GNUNET_SCHEDULER_NO_TASK;
88   }
89   if (privkey != NULL)
90     GNUNET_free (privkey);
91   privkey = NULL;
92   if (nsh != NULL)
93     GNUNET_NAMESTORE_disconnect (nsh);
94   nsh = NULL;
95 }
96
97
98 static void
99 zone_to_name_proc (void *cls,
100                    const struct GNUNET_CRYPTO_EcdsaPublicKey *zone_key,
101                    struct GNUNET_TIME_Absolute expire,
102                    const char *n,
103                    unsigned int rd_count,
104                    const struct GNUNET_GNSRECORD_Data *rd,
105                    const struct GNUNET_CRYPTO_EcdsaSignature *signature)
106 {
107   int fail = GNUNET_NO;
108
109   if ((zone_key == NULL) && (n == NULL) && (rd_count == 0) && (rd == NULL) && (signature == NULL))
110   {
111     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "No result found\n");
112     res = 1;
113   }
114   else
115   {
116     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Result found: `%s'\n", n);
117     if ((n == NULL) || (0 != strcmp(n, s_name)))
118     {
119       fail = GNUNET_YES;
120       GNUNET_break (0);
121     }
122     if (rd_count != 1)
123     {
124       fail = GNUNET_YES;
125       GNUNET_break (0);
126     }
127     if ((zone_key == NULL) || (0 != memcmp (zone_key, &pubkey, sizeof (struct GNUNET_CRYPTO_EcdsaPublicKey))))
128     {
129       fail = GNUNET_YES;
130       GNUNET_break (0);
131     }
132     if (fail == GNUNET_NO)
133       res = 0;
134     else
135       res = 1;
136   }
137   GNUNET_SCHEDULER_add_now(&end, NULL);
138 }
139
140
141 static void
142 put_cont (void *cls, int32_t success, const char *emsg)
143 {
144   char *name = cls;
145   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Name store added record for `%s': %s\n", name, (success == GNUNET_OK) ? "SUCCESS" : "FAIL");
146   if (success == GNUNET_OK)
147   {
148     res = 0;
149
150     /* create initial record */
151     GNUNET_NAMESTORE_zone_to_name (nsh, &s_zone, &s_zone_value, zone_to_name_proc, NULL);
152
153   }
154   else
155   {
156     res = 1;
157     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
158                 "Failed to put records for name `%s'\n", name);
159     GNUNET_SCHEDULER_add_now(&end, NULL);
160   }
161 }
162
163
164 static void
165 run (void *cls,
166      const struct GNUNET_CONFIGURATION_Handle *cfg,
167      struct GNUNET_TESTING_Peer *peer)
168 {
169   struct GNUNET_TIME_Absolute et;
170
171   directory = NULL;
172   GNUNET_CONFIGURATION_get_value_string(cfg, "PATHS", "GNUNET_TEST_HOME", &directory);
173   GNUNET_DISK_directory_remove (directory);
174
175   endbadly_task = GNUNET_SCHEDULER_add_delayed(TIMEOUT,endbadly, NULL);
176   GNUNET_asprintf (&s_name, "dummy");
177   /* load privat key */
178   char *hostkey_file;
179   GNUNET_asprintf(&hostkey_file,"zonefiles%s%s",DIR_SEPARATOR_STR,
180       "N0UJMP015AFUNR2BTNM3FKPBLG38913BL8IDMCO2H0A1LIB81960.zkey");
181   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
182               "Using zonekey file `%s'\n",
183               hostkey_file);
184   privkey = GNUNET_CRYPTO_ecdsa_key_create_from_file(hostkey_file);
185   GNUNET_free (hostkey_file);
186   GNUNET_assert (privkey != NULL);
187   /* get public key */
188   GNUNET_CRYPTO_ecdsa_key_get_public(privkey, &pubkey);
189
190   /* zone hash */
191   GNUNET_CRYPTO_short_hash (&pubkey, sizeof (struct GNUNET_CRYPTO_EcdsaPublicKey), &s_zone);
192   GNUNET_CRYPTO_short_hash (s_name, strlen (s_name) + 1, &s_zone_value);
193   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
194               "Using PKEY `%s' \n",
195               GNUNET_NAMESTORE_short_h2s (&s_zone_value));
196
197   struct GNUNET_GNSRECORD_Data rd;
198   rd.expiration_time = GNUNET_TIME_absolute_get().abs_value_us;
199   rd.record_type = GNUNET_GNSRECORD_TYPE_PKEY;
200   rd.data_size = sizeof (struct GNUNET_CRYPTO_ShortHashCode);
201   rd.data = GNUNET_malloc(sizeof (struct GNUNET_CRYPTO_ShortHashCode));
202   rd.flags = 0;
203   memcpy ((char *) rd.data, &s_zone_value, sizeof (struct GNUNET_CRYPTO_ShortHashCode));
204   nsh = GNUNET_NAMESTORE_connect (cfg);
205   GNUNET_break (NULL != nsh);
206
207   expire = GNUNET_TIME_absolute_get ();
208   et.abs_value_us = rd.expiration_time;
209   s_signature = GNUNET_NAMESTORE_create_signature(privkey, et, s_name, &rd, 1);
210   GNUNET_NAMESTORE_record_put(nsh, &pubkey, s_name, expire, 1, &rd, s_signature, put_cont, NULL);
211
212   GNUNET_free ((void *) rd.data);
213 }
214
215
216
217 int
218 main (int argc, char *argv[])
219 {
220   res = 1;
221   if (0 !=
222       GNUNET_TESTING_peer_run ("test-namestore-api-zone-to-name",
223                                "test_namestore_api.conf",
224                                &run,
225                                NULL))
226   {
227     res = 1;
228   }
229   if (NULL != directory)
230   {
231       GNUNET_DISK_directory_remove (directory);
232       GNUNET_free (directory);
233   }
234   return res;
235 }
236
237 /* end of test_namestore_api_zone_to_name.c */