diff options
Diffstat (limited to 'src/service/namestore/test_namestore_api_lookup_nick.c')
-rw-r--r-- | src/service/namestore/test_namestore_api_lookup_nick.c | 346 |
1 files changed, 346 insertions, 0 deletions
diff --git a/src/service/namestore/test_namestore_api_lookup_nick.c b/src/service/namestore/test_namestore_api_lookup_nick.c new file mode 100644 index 000000000..8b4799363 --- /dev/null +++ b/src/service/namestore/test_namestore_api_lookup_nick.c | |||
@@ -0,0 +1,346 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet. | ||
3 | Copyright (C) 2012 GNUnet e.V. | ||
4 | |||
5 | GNUnet is free software: you can redistribute it and/or modify it | ||
6 | under the terms of the GNU Affero General Public License as published | ||
7 | by the Free Software Foundation, either version 3 of the License, | ||
8 | or (at your 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 | Affero General Public License for more details. | ||
14 | |||
15 | You should have received a copy of the GNU Affero General Public License | ||
16 | along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
17 | |||
18 | SPDX-License-Identifier: AGPL3.0-or-later | ||
19 | */ | ||
20 | /** | ||
21 | * @file namestore/test_namestore_api_lookup_nick.c | ||
22 | * @brief testcase for namestore_api.c: NICK records | ||
23 | */ | ||
24 | #include "platform.h" | ||
25 | #include "gnunet_namestore_service.h" | ||
26 | #include "gnunet_testing_lib.h" | ||
27 | |||
28 | #define TEST_RECORD_TYPE GNUNET_DNSPARSER_TYPE_TXT | ||
29 | |||
30 | #define TEST_RECORD_DATALEN 123 | ||
31 | |||
32 | #define TEST_NICK "gnunettestnick" | ||
33 | |||
34 | #define TEST_RECORD_DATA 'a' | ||
35 | |||
36 | #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10) | ||
37 | |||
38 | static struct GNUNET_NAMESTORE_Handle *nsh; | ||
39 | |||
40 | static struct GNUNET_SCHEDULER_Task *endbadly_task; | ||
41 | |||
42 | static struct GNUNET_CRYPTO_PrivateKey privkey; | ||
43 | |||
44 | static struct GNUNET_CRYPTO_PublicKey pubkey; | ||
45 | |||
46 | static int res; | ||
47 | |||
48 | static struct GNUNET_GNSRECORD_Data rd_orig; | ||
49 | |||
50 | static struct GNUNET_NAMESTORE_QueueEntry *nsqe; | ||
51 | |||
52 | // static const char * name = "dummy.dummy.gnunet"; | ||
53 | static const char *name = "d"; | ||
54 | |||
55 | static char *record_data; | ||
56 | |||
57 | static void | ||
58 | cleanup () | ||
59 | { | ||
60 | GNUNET_free (record_data); | ||
61 | if (NULL != nsh) | ||
62 | { | ||
63 | GNUNET_NAMESTORE_disconnect (nsh); | ||
64 | nsh = NULL; | ||
65 | } | ||
66 | GNUNET_SCHEDULER_shutdown (); | ||
67 | } | ||
68 | |||
69 | |||
70 | /** | ||
71 | * Re-establish the connection to the service. | ||
72 | * | ||
73 | * @param cls handle to use to re-connect. | ||
74 | * @param tc scheduler context | ||
75 | */ | ||
76 | static void | ||
77 | endbadly (void *cls) | ||
78 | { | ||
79 | if (NULL != nsqe) | ||
80 | { | ||
81 | GNUNET_NAMESTORE_cancel (nsqe); | ||
82 | nsqe = NULL; | ||
83 | } | ||
84 | cleanup (); | ||
85 | res = 1; | ||
86 | } | ||
87 | |||
88 | |||
89 | static void | ||
90 | end (void *cls) | ||
91 | { | ||
92 | cleanup (); | ||
93 | res = 0; | ||
94 | } | ||
95 | |||
96 | |||
97 | static void | ||
98 | lookup_it (void *cls, | ||
99 | const struct GNUNET_CRYPTO_PrivateKey *zone, | ||
100 | const char *label, | ||
101 | unsigned int rd_count, | ||
102 | const struct GNUNET_GNSRECORD_Data *rd) | ||
103 | { | ||
104 | nsqe = NULL; | ||
105 | int c; | ||
106 | int found_record = GNUNET_NO; | ||
107 | int found_nick = GNUNET_NO; | ||
108 | |||
109 | if (0 != GNUNET_memcmp (&privkey, zone)) | ||
110 | { | ||
111 | GNUNET_break (0); | ||
112 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
113 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
114 | return; | ||
115 | } | ||
116 | |||
117 | if (NULL == label) | ||
118 | { | ||
119 | GNUNET_break (0); | ||
120 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
121 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
122 | return; | ||
123 | } | ||
124 | |||
125 | if (0 != strcmp (label, name)) | ||
126 | { | ||
127 | GNUNET_break (0); | ||
128 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
129 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
130 | return; | ||
131 | } | ||
132 | |||
133 | if (2 != rd_count) | ||
134 | { | ||
135 | GNUNET_break (0); | ||
136 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
137 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
138 | return; | ||
139 | } | ||
140 | |||
141 | for (c = 0; c < rd_count; c++) | ||
142 | { | ||
143 | if (GNUNET_GNSRECORD_TYPE_NICK == rd[c].record_type) | ||
144 | { | ||
145 | if (rd[c].data_size != strlen (TEST_NICK) + 1) | ||
146 | { | ||
147 | GNUNET_break (0); | ||
148 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
149 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
150 | return; | ||
151 | } | ||
152 | if (0 != (rd[c].flags & GNUNET_GNSRECORD_RF_PRIVATE)) | ||
153 | { | ||
154 | GNUNET_break (0); | ||
155 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
156 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
157 | return; | ||
158 | } | ||
159 | if (0 != strcmp (rd[c].data, TEST_NICK)) | ||
160 | { | ||
161 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
162 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
163 | return; | ||
164 | } | ||
165 | found_nick = GNUNET_YES; | ||
166 | } | ||
167 | else | ||
168 | { | ||
169 | if (rd[c].record_type != TEST_RECORD_TYPE) | ||
170 | { | ||
171 | GNUNET_break (0); | ||
172 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
173 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
174 | return; | ||
175 | } | ||
176 | if (rd[c].data_size != TEST_RECORD_DATALEN) | ||
177 | { | ||
178 | GNUNET_break (0); | ||
179 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
180 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
181 | return; | ||
182 | } | ||
183 | if (0 != memcmp (rd[c].data, rd_orig.data, TEST_RECORD_DATALEN)) | ||
184 | { | ||
185 | GNUNET_break (0); | ||
186 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
187 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
188 | return; | ||
189 | } | ||
190 | if (rd[c].flags != rd->flags) | ||
191 | { | ||
192 | GNUNET_break (0); | ||
193 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
194 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
195 | return; | ||
196 | } | ||
197 | found_record = GNUNET_YES; | ||
198 | } | ||
199 | } | ||
200 | |||
201 | /* Done */ | ||
202 | if ((GNUNET_YES == found_nick) && (GNUNET_YES == found_record)) | ||
203 | { | ||
204 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
205 | endbadly_task = NULL; | ||
206 | GNUNET_SCHEDULER_add_now (&end, NULL); | ||
207 | } | ||
208 | else | ||
209 | { | ||
210 | GNUNET_break (0); | ||
211 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
212 | endbadly_task = NULL; | ||
213 | GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
214 | } | ||
215 | } | ||
216 | |||
217 | |||
218 | static void | ||
219 | fail_cb (void *cls) | ||
220 | { | ||
221 | GNUNET_assert (0); | ||
222 | } | ||
223 | |||
224 | |||
225 | static void | ||
226 | put_cont (void *cls, enum GNUNET_ErrorCode ec) | ||
227 | { | ||
228 | const char *name = cls; | ||
229 | |||
230 | nsqe = NULL; | ||
231 | GNUNET_assert (NULL != cls); | ||
232 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, | ||
233 | "Name store added record for `%s': %s\n", | ||
234 | name, | ||
235 | (ec == GNUNET_EC_NONE) ? "SUCCESS" : "FAIL"); | ||
236 | |||
237 | if (GNUNET_EC_NONE != ec) | ||
238 | { | ||
239 | GNUNET_SCHEDULER_cancel (endbadly_task); | ||
240 | endbadly_task = GNUNET_SCHEDULER_add_now (&endbadly, NULL); | ||
241 | return; | ||
242 | } | ||
243 | /* Lookup */ | ||
244 | nsqe = GNUNET_NAMESTORE_records_lookup (nsh, | ||
245 | &privkey, | ||
246 | name, | ||
247 | &fail_cb, | ||
248 | NULL, | ||
249 | &lookup_it, | ||
250 | NULL); | ||
251 | } | ||
252 | |||
253 | |||
254 | static void | ||
255 | nick_cont (void *cls, enum GNUNET_ErrorCode ec) | ||
256 | { | ||
257 | const char *name = cls; | ||
258 | |||
259 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, | ||
260 | "Nick added : %s\n", | ||
261 | (ec == GNUNET_EC_NONE) ? "SUCCESS" : "FAIL"); | ||
262 | |||
263 | rd_orig.expiration_time = GNUNET_TIME_UNIT_HOURS.rel_value_us; | ||
264 | rd_orig.record_type = TEST_RECORD_TYPE; | ||
265 | rd_orig.data_size = TEST_RECORD_DATALEN; | ||
266 | record_data = GNUNET_malloc (TEST_RECORD_DATALEN); | ||
267 | rd_orig.data = record_data; | ||
268 | rd_orig.flags = GNUNET_GNSRECORD_RF_RELATIVE_EXPIRATION; | ||
269 | memset ((char *) rd_orig.data, 'a', TEST_RECORD_DATALEN); | ||
270 | |||
271 | nsqe = GNUNET_NAMESTORE_record_set_store (nsh, &privkey, | ||
272 | name, | ||
273 | 1, | ||
274 | &rd_orig, | ||
275 | &put_cont, (void *) name); | ||
276 | } | ||
277 | |||
278 | |||
279 | static void | ||
280 | run (void *cls, | ||
281 | const struct GNUNET_CONFIGURATION_Handle *cfg, | ||
282 | struct GNUNET_TESTING_Peer *peer) | ||
283 | { | ||
284 | struct GNUNET_GNSRECORD_Data rd; | ||
285 | |||
286 | endbadly_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, | ||
287 | &endbadly, | ||
288 | NULL); | ||
289 | privkey.type = htonl (GNUNET_GNSRECORD_TYPE_PKEY); | ||
290 | GNUNET_CRYPTO_ecdsa_key_create (&privkey.ecdsa_key); | ||
291 | GNUNET_CRYPTO_key_get_public (&privkey, | ||
292 | &pubkey); | ||
293 | |||
294 | nsh = GNUNET_NAMESTORE_connect (cfg); | ||
295 | GNUNET_break (NULL != nsh); | ||
296 | |||
297 | memset (&rd, 0, sizeof(rd)); | ||
298 | rd.data = TEST_NICK; | ||
299 | rd.data_size = strlen (TEST_NICK) + 1; | ||
300 | rd.record_type = GNUNET_GNSRECORD_TYPE_NICK; | ||
301 | rd.expiration_time = GNUNET_TIME_UNIT_FOREVER_ABS.abs_value_us; | ||
302 | rd.flags |= GNUNET_GNSRECORD_RF_PRIVATE; | ||
303 | nsqe = GNUNET_NAMESTORE_record_set_store (nsh, | ||
304 | &privkey, | ||
305 | GNUNET_GNS_EMPTY_LABEL_AT, | ||
306 | 1, | ||
307 | &rd, | ||
308 | &nick_cont, | ||
309 | (void *) name); | ||
310 | |||
311 | if (NULL == nsqe) | ||
312 | { | ||
313 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, | ||
314 | _ ("Namestore cannot store no block\n")); | ||
315 | } | ||
316 | } | ||
317 | |||
318 | |||
319 | #include "test_common.c" | ||
320 | |||
321 | |||
322 | int | ||
323 | main (int argc, char *argv[]) | ||
324 | { | ||
325 | char *plugin_name; | ||
326 | char *cfg_name; | ||
327 | |||
328 | SETUP_CFG (plugin_name, cfg_name); | ||
329 | res = 1; | ||
330 | if (0 != | ||
331 | GNUNET_TESTING_peer_run ("test-namestore-api-lookup-nick", | ||
332 | cfg_name, | ||
333 | &run, | ||
334 | NULL)) | ||
335 | { | ||
336 | res = 1; | ||
337 | } | ||
338 | GNUNET_DISK_purge_cfg_dir (cfg_name, | ||
339 | "GNUNET_TEST_HOME"); | ||
340 | GNUNET_free (plugin_name); | ||
341 | GNUNET_free (cfg_name); | ||
342 | return res; | ||
343 | } | ||
344 | |||
345 | |||
346 | /* end of test_namestore_api_store.c */ | ||