Merge remote-tracking branch 'origin/master' into credentials
[oweals/gnunet.git] / src / namestore / test_namestore_api_zone_to_name.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 2009 GNUnet e.V.
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., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, 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 struct GNUNET_SCHEDULER_Task * 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  */
67 static void
68 endbadly (void *cls)
69 {
70   if (nsh != NULL)
71     GNUNET_NAMESTORE_disconnect (nsh);
72   nsh = NULL;
73   if (privkey != NULL)
74     GNUNET_free (privkey);
75   privkey = NULL;
76   res = 1;
77 }
78
79
80 static void
81 end (void *cls)
82 {
83   if (endbadly_task != NULL)
84   {
85     GNUNET_SCHEDULER_cancel (endbadly_task);
86     endbadly_task = NULL;
87   }
88   if (privkey != NULL)
89     GNUNET_free (privkey);
90   privkey = NULL;
91   if (nsh != NULL)
92     GNUNET_NAMESTORE_disconnect (nsh);
93   nsh = NULL;
94 }
95
96
97 static void
98 zone_to_name_proc (void *cls,
99                    const struct GNUNET_CRYPTO_EcdsaPublicKey *zone_key,
100                    struct GNUNET_TIME_Absolute expire,
101                    const char *n,
102                    unsigned int rd_count,
103                    const struct GNUNET_GNSRECORD_Data *rd,
104                    const struct GNUNET_CRYPTO_EcdsaSignature *signature)
105 {
106   int fail = GNUNET_NO;
107
108   if ((zone_key == NULL) && (n == NULL) && (rd_count == 0) && (rd == NULL) && (signature == NULL))
109   {
110     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "No result found\n");
111     res = 1;
112   }
113   else
114   {
115     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Result found: `%s'\n", n);
116     if ((n == NULL) || (0 != strcmp(n, s_name)))
117     {
118       fail = GNUNET_YES;
119       GNUNET_break (0);
120     }
121     if (rd_count != 1)
122     {
123       fail = GNUNET_YES;
124       GNUNET_break (0);
125     }
126     if ((zone_key == NULL) || (0 != memcmp (zone_key, &pubkey, sizeof (struct GNUNET_CRYPTO_EcdsaPublicKey))))
127     {
128       fail = GNUNET_YES;
129       GNUNET_break (0);
130     }
131     if (fail == GNUNET_NO)
132       res = 0;
133     else
134       res = 1;
135   }
136   GNUNET_SCHEDULER_add_now(&end, NULL);
137 }
138
139
140 static void
141 put_cont (void *cls, int32_t success, const char *emsg)
142 {
143   char *name = cls;
144   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Name store added record for `%s': %s\n", name, (success == GNUNET_OK) ? "SUCCESS" : "FAIL");
145   if (success == GNUNET_OK)
146   {
147     res = 0;
148
149     /* create initial record */
150     GNUNET_NAMESTORE_zone_to_name (nsh, &s_zone, &s_zone_value, zone_to_name_proc, NULL);
151
152   }
153   else
154   {
155     res = 1;
156     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
157                 "Failed to put records for name `%s'\n", name);
158     GNUNET_SCHEDULER_add_now(&end, NULL);
159   }
160 }
161
162
163 static void
164 run (void *cls,
165      const struct GNUNET_CONFIGURATION_Handle *cfg,
166      struct GNUNET_TESTING_Peer *peer)
167 {
168   struct GNUNET_TIME_Absolute et;
169
170   directory = NULL;
171   GNUNET_CONFIGURATION_get_value_string(cfg, "PATHS", "GNUNET_TEST_HOME", &directory);
172   GNUNET_DISK_directory_remove (directory);
173
174   endbadly_task = GNUNET_SCHEDULER_add_delayed(TIMEOUT,endbadly, NULL);
175   GNUNET_asprintf (&s_name, "dummy");
176   /* load privat key */
177   char *hostkey_file;
178   GNUNET_asprintf(&hostkey_file,"zonefiles%s%s",DIR_SEPARATOR_STR,
179       "N0UJMP015AFUNR2BTNM3FKPBLG38913BL8IDMCO2H0A1LIB81960.zkey");
180   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
181               "Using zonekey file `%s'\n",
182               hostkey_file);
183   privkey = GNUNET_CRYPTO_ecdsa_key_create_from_file(hostkey_file);
184   GNUNET_free (hostkey_file);
185   GNUNET_assert (privkey != NULL);
186   /* get public key */
187   GNUNET_CRYPTO_ecdsa_key_get_public(privkey, &pubkey);
188
189   /* zone hash */
190   GNUNET_CRYPTO_short_hash (&pubkey, sizeof (struct GNUNET_CRYPTO_EcdsaPublicKey), &s_zone);
191   GNUNET_CRYPTO_short_hash (s_name, strlen (s_name) + 1, &s_zone_value);
192   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
193               "Using PKEY `%s' \n",
194               GNUNET_NAMESTORE_short_h2s (&s_zone_value));
195
196   struct GNUNET_GNSRECORD_Data rd;
197   rd.expiration_time = GNUNET_TIME_absolute_get().abs_value_us;
198   rd.record_type = GNUNET_GNSRECORD_TYPE_PKEY;
199   rd.data_size = sizeof (struct GNUNET_CRYPTO_ShortHashCode);
200   rd.data = GNUNET_malloc(sizeof (struct GNUNET_CRYPTO_ShortHashCode));
201   rd.flags = 0;
202   GNUNET_memcpy ((char *) rd.data, &s_zone_value, sizeof (struct GNUNET_CRYPTO_ShortHashCode));
203   nsh = GNUNET_NAMESTORE_connect (cfg);
204   GNUNET_break (NULL != nsh);
205
206   expire = GNUNET_TIME_absolute_get ();
207   et.abs_value_us = rd.expiration_time;
208   s_signature = GNUNET_NAMESTORE_create_signature(privkey, et, s_name, &rd, 1);
209   GNUNET_NAMESTORE_record_put(nsh, &pubkey, s_name, expire, 1, &rd, s_signature, put_cont, NULL);
210
211   GNUNET_free ((void *) rd.data);
212 }
213
214
215
216 int
217 main (int argc, char *argv[])
218 {
219   res = 1;
220   if (0 !=
221       GNUNET_TESTING_peer_run ("test-namestore-api-zone-to-name",
222                                "test_namestore_api.conf",
223                                &run,
224                                NULL))
225   {
226     res = 1;
227   }
228   if (NULL != directory)
229   {
230       GNUNET_DISK_directory_remove (directory);
231       GNUNET_free (directory);
232   }
233   return res;
234 }
235
236 /* end of test_namestore_api_zone_to_name.c */