diff options
Diffstat (limited to 'src/multicast/test_multicast_2peers.c')
-rw-r--r-- | src/multicast/test_multicast_2peers.c | 511 |
1 files changed, 511 insertions, 0 deletions
diff --git a/src/multicast/test_multicast_2peers.c b/src/multicast/test_multicast_2peers.c new file mode 100644 index 000000000..058533e50 --- /dev/null +++ b/src/multicast/test_multicast_2peers.c | |||
@@ -0,0 +1,511 @@ | |||
1 | /* | ||
2 | * This file is part of GNUnet | ||
3 | * Copyright (C) 2013 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 | /** | ||
22 | * @file multicast/test_multicast_2peers.c | ||
23 | * @brief Tests for the Multicast API with two peers doing the ping | ||
24 | * pong test. | ||
25 | * @author xrs | ||
26 | */ | ||
27 | |||
28 | #include <inttypes.h> | ||
29 | |||
30 | #include "platform.h" | ||
31 | #include "gnunet_crypto_lib.h" | ||
32 | #include "gnunet_common.h" | ||
33 | #include "gnunet_util_lib.h" | ||
34 | #include "gnunet_testbed_service.h" | ||
35 | #include "gnunet_multicast_service.h" | ||
36 | |||
37 | #define NUM_PEERS 2 | ||
38 | |||
39 | static struct GNUNET_TESTBED_Operation *op0; | ||
40 | static struct GNUNET_TESTBED_Operation *op1; | ||
41 | static struct GNUNET_TESTBED_Operation *pi_op0; | ||
42 | static struct GNUNET_TESTBED_Operation *pi_op1; | ||
43 | |||
44 | static struct GNUNET_TESTBED_Peer **peers; | ||
45 | const struct GNUNET_PeerIdentity *peer_id[2]; | ||
46 | |||
47 | static struct GNUNET_SCHEDULER_Task *timeout_tid; | ||
48 | |||
49 | static struct GNUNET_MULTICAST_Origin *origin; | ||
50 | static struct GNUNET_MULTICAST_Member *member; | ||
51 | |||
52 | struct GNUNET_CRYPTO_EddsaPrivateKey *group_key; | ||
53 | struct GNUNET_CRYPTO_EddsaPublicKey group_pub_key; | ||
54 | |||
55 | struct GNUNET_CRYPTO_EcdsaPrivateKey *member_key; | ||
56 | struct GNUNET_CRYPTO_EcdsaPublicKey member_pub_key; | ||
57 | |||
58 | /** | ||
59 | * Global result for testcase. | ||
60 | */ | ||
61 | static int result; | ||
62 | |||
63 | |||
64 | /** | ||
65 | * Function run on CTRL-C or shutdown (i.e. success/timeout/etc.). | ||
66 | * Cleans up. | ||
67 | */ | ||
68 | static void | ||
69 | shutdown_task (void *cls) | ||
70 | { | ||
71 | if (NULL != op0) | ||
72 | { | ||
73 | GNUNET_TESTBED_operation_done (op0); | ||
74 | op0 = NULL; | ||
75 | } | ||
76 | if (NULL != op1) | ||
77 | { | ||
78 | GNUNET_TESTBED_operation_done (op1); | ||
79 | op1 = NULL; | ||
80 | } | ||
81 | if (NULL != pi_op0) | ||
82 | { | ||
83 | GNUNET_TESTBED_operation_done (pi_op0); | ||
84 | pi_op0 = NULL; | ||
85 | } | ||
86 | if (NULL != pi_op1) | ||
87 | { | ||
88 | GNUNET_TESTBED_operation_done (pi_op1); | ||
89 | pi_op1 = NULL; | ||
90 | } | ||
91 | if (NULL != timeout_tid) | ||
92 | { | ||
93 | GNUNET_SCHEDULER_cancel (timeout_tid); | ||
94 | timeout_tid = NULL; | ||
95 | } | ||
96 | } | ||
97 | |||
98 | |||
99 | static void | ||
100 | timeout_task (void *cls) | ||
101 | { | ||
102 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, | ||
103 | "Timeout!\n"); | ||
104 | result = GNUNET_SYSERR; | ||
105 | GNUNET_SCHEDULER_shutdown (); | ||
106 | } | ||
107 | |||
108 | |||
109 | static void | ||
110 | member_join_request (void *cls, | ||
111 | const struct GNUNET_CRYPTO_EcdsaPublicKey *member_pub_key, | ||
112 | const struct GNUNET_MessageHeader *join_msg, | ||
113 | struct GNUNET_MULTICAST_JoinHandle *jh) | ||
114 | { | ||
115 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
116 | "Member sent a join request.\n"); | ||
117 | |||
118 | } | ||
119 | |||
120 | |||
121 | static int | ||
122 | notify (void *cls, | ||
123 | size_t *data_size, | ||
124 | void *data) | ||
125 | { | ||
126 | |||
127 | char text[] = "ping"; | ||
128 | *data_size = strlen(text)+1; | ||
129 | GNUNET_memcpy(data, text, *data_size); | ||
130 | |||
131 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
132 | "Member sents message to origin: %s\n", text); | ||
133 | |||
134 | return GNUNET_YES; | ||
135 | } | ||
136 | |||
137 | |||
138 | static void | ||
139 | member_join_decision (void *cls, | ||
140 | int is_admitted, | ||
141 | const struct GNUNET_PeerIdentity *peer, | ||
142 | uint16_t relay_count, | ||
143 | const struct GNUNET_PeerIdentity *relays, | ||
144 | const struct GNUNET_MessageHeader *join_msg) | ||
145 | { | ||
146 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
147 | "Member received a decision from origin: %s\n", | ||
148 | (GNUNET_YES == is_admitted) | ||
149 | ? "accepted" | ||
150 | : "rejected"); | ||
151 | |||
152 | if (GNUNET_YES == is_admitted) | ||
153 | { | ||
154 | struct GNUNET_MULTICAST_MemberTransmitHandle *req; | ||
155 | |||
156 | // FIXME: move to MQ-style API! | ||
157 | req = GNUNET_MULTICAST_member_to_origin (member, | ||
158 | 0, | ||
159 | ¬ify, | ||
160 | NULL); | ||
161 | } | ||
162 | } | ||
163 | |||
164 | |||
165 | static void | ||
166 | member_message (void *cls, | ||
167 | const struct GNUNET_MULTICAST_MessageHeader *msg) | ||
168 | { | ||
169 | if (0 != strncmp ("pong", (char *)&msg[1], 4)) | ||
170 | { | ||
171 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "member did not receive pong\n"); | ||
172 | result = GNUNET_SYSERR; | ||
173 | GNUNET_SCHEDULER_shutdown (); | ||
174 | } | ||
175 | |||
176 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
177 | "member receives: %s\n", (char *)&msg[1]); | ||
178 | |||
179 | // Testcase ends here. | ||
180 | result = GNUNET_YES; | ||
181 | GNUNET_SCHEDULER_shutdown (); | ||
182 | } | ||
183 | |||
184 | static void | ||
185 | origin_join_request (void *cls, | ||
186 | const struct GNUNET_CRYPTO_EcdsaPublicKey *member_pub_key, | ||
187 | const struct GNUNET_MessageHeader *join_msg, | ||
188 | struct GNUNET_MULTICAST_JoinHandle *jh) | ||
189 | { | ||
190 | struct GNUNET_MessageHeader *join_resp; | ||
191 | |||
192 | uint8_t data_size = ntohs (join_msg->size); | ||
193 | |||
194 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
195 | "origin got a join request...\n"); | ||
196 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
197 | "origin receives: '%s'\n", (char *)&join_msg[1]); | ||
198 | |||
199 | const char data[] = "Come in!"; | ||
200 | data_size = strlen (data) + 1; | ||
201 | join_resp = GNUNET_malloc (sizeof (join_resp) + data_size); | ||
202 | join_resp->size = htons (sizeof (join_resp) + data_size); | ||
203 | join_resp->type = htons (123); | ||
204 | GNUNET_memcpy (&join_resp[1], data, data_size); | ||
205 | |||
206 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
207 | "origin sends: '%s'\n", data); | ||
208 | |||
209 | GNUNET_MULTICAST_join_decision (jh, | ||
210 | GNUNET_YES, | ||
211 | 0, | ||
212 | NULL, | ||
213 | join_resp); | ||
214 | GNUNET_free (join_resp); | ||
215 | result = GNUNET_OK; | ||
216 | } | ||
217 | |||
218 | int | ||
219 | origin_notify (void *cls, | ||
220 | size_t *data_size, | ||
221 | void *data) | ||
222 | { | ||
223 | char text[] = "pong"; | ||
224 | *data_size = strlen(text)+1; | ||
225 | memcpy(data, text, *data_size); | ||
226 | |||
227 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "origin sends (to all): %s\n", text); | ||
228 | |||
229 | return GNUNET_YES; | ||
230 | } | ||
231 | |||
232 | |||
233 | static void | ||
234 | origin_request (void *cls, | ||
235 | const struct GNUNET_MULTICAST_RequestHeader *req) | ||
236 | { | ||
237 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "origin receives: %s\n", (char *)&req[1]); | ||
238 | |||
239 | if (0 != strncmp ("ping", (char *)&req[1], 4)) | ||
240 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "origin didn't reveice a correct request"); | ||
241 | |||
242 | GNUNET_MULTICAST_origin_to_all (origin, | ||
243 | 0, | ||
244 | 0, | ||
245 | origin_notify, | ||
246 | NULL); | ||
247 | } | ||
248 | |||
249 | static void | ||
250 | origin_message (void *cls, | ||
251 | const struct GNUNET_MULTICAST_MessageHeader *msg) | ||
252 | { | ||
253 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "origin message msg\n"); | ||
254 | } | ||
255 | |||
256 | |||
257 | static void | ||
258 | service_connect1 (void *cls, | ||
259 | struct GNUNET_TESTBED_Operation *op, | ||
260 | void *ca_result, | ||
261 | const char *emsg) | ||
262 | { | ||
263 | member = ca_result; | ||
264 | |||
265 | if (NULL != member) | ||
266 | { | ||
267 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Connected to multicast service of member\n"); | ||
268 | } | ||
269 | else | ||
270 | { | ||
271 | result = GNUNET_SYSERR; | ||
272 | GNUNET_SCHEDULER_shutdown (); | ||
273 | } | ||
274 | } | ||
275 | |||
276 | static void | ||
277 | multicast_da1 (void *cls, | ||
278 | void * op_result) | ||
279 | { | ||
280 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
281 | "Member parting from multicast group\n"); | ||
282 | |||
283 | GNUNET_MULTICAST_member_part (member, NULL, NULL); | ||
284 | } | ||
285 | |||
286 | |||
287 | static void * | ||
288 | multicast_ca1 (void *cls, | ||
289 | const struct GNUNET_CONFIGURATION_Handle *cfg) | ||
290 | { | ||
291 | struct GNUNET_MessageHeader *join_msg; | ||
292 | void *ret; | ||
293 | |||
294 | // Get members keys | ||
295 | member_key = GNUNET_CRYPTO_ecdsa_key_create (); | ||
296 | GNUNET_CRYPTO_ecdsa_key_get_public (member_key, &member_pub_key); | ||
297 | |||
298 | char data[] = "Hi, can I enter?"; | ||
299 | uint8_t data_size = strlen (data) + 1; | ||
300 | join_msg = GNUNET_malloc (sizeof (join_msg) + data_size); | ||
301 | join_msg->size = htons (sizeof (join_msg) + data_size); | ||
302 | join_msg->type = htons (123); | ||
303 | GNUNET_memcpy (&join_msg[1], data, data_size); | ||
304 | |||
305 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
306 | "Members tries to join multicast group\n"); | ||
307 | |||
308 | ret = GNUNET_MULTICAST_member_join (cfg, | ||
309 | &group_pub_key, | ||
310 | member_key, | ||
311 | peer_id[0], | ||
312 | 0, | ||
313 | NULL, | ||
314 | join_msg, /* join message */ | ||
315 | member_join_request, | ||
316 | member_join_decision, | ||
317 | NULL, /* no test for member_replay_frag */ | ||
318 | NULL, /* no test for member_replay_msg */ | ||
319 | member_message, | ||
320 | NULL); | ||
321 | GNUNET_free (join_msg); | ||
322 | return ret; | ||
323 | } | ||
324 | |||
325 | |||
326 | static void | ||
327 | peer_information_cb (void *cls, | ||
328 | struct GNUNET_TESTBED_Operation *op, | ||
329 | const struct GNUNET_TESTBED_PeerInformation *pinfo, | ||
330 | const char *emsg) | ||
331 | { | ||
332 | int i = (int) (long) cls; | ||
333 | |||
334 | if (NULL == pinfo) | ||
335 | { | ||
336 | result = GNUNET_SYSERR; | ||
337 | GNUNET_SCHEDULER_shutdown (); | ||
338 | } | ||
339 | |||
340 | peer_id[i] = pinfo->result.id; | ||
341 | |||
342 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
343 | "Got peer information of %s (%s)\n", (0==i)?"origin":"member" ,GNUNET_i2s(pinfo->result.id)); | ||
344 | |||
345 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
346 | "Create member peer\n"); | ||
347 | |||
348 | if (0 == i) | ||
349 | { | ||
350 | /* connect to multicast service of member */ | ||
351 | op1 = GNUNET_TESTBED_service_connect (NULL, /* Closure for operation */ | ||
352 | peers[1], /* The peer whose service to connect to */ | ||
353 | "multicast", /* The name of the service */ | ||
354 | service_connect1, /* callback to call after a handle to service | ||
355 | is opened */ | ||
356 | NULL, /* closure for the above callback */ | ||
357 | multicast_ca1, /* callback to call with peer's configuration; | ||
358 | this should open the needed service connection */ | ||
359 | multicast_da1, /* callback to be called when closing the | ||
360 | opened service connection */ | ||
361 | NULL); /* closure for the above two callbacks */ | ||
362 | } | ||
363 | } | ||
364 | |||
365 | /** | ||
366 | * Test logic of peer "0" being origin starts here. | ||
367 | * | ||
368 | * @param cls closure, for the example: NULL | ||
369 | * @param op should be equal to "dht_op" | ||
370 | * @param ca_result result of the connect operation, the | ||
371 | * connection to the DHT service | ||
372 | * @param emsg error message, if testbed somehow failed to | ||
373 | * connect to the DHT. | ||
374 | */ | ||
375 | static void | ||
376 | service_connect0 (void *cls, | ||
377 | struct GNUNET_TESTBED_Operation *op, | ||
378 | void *ca_result, | ||
379 | const char *emsg) | ||
380 | { | ||
381 | origin = ca_result; | ||
382 | |||
383 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
384 | "Connected to multicast service of origin\n"); | ||
385 | |||
386 | // Get GNUnet identity of origin | ||
387 | pi_op0 = GNUNET_TESTBED_peer_get_information (peers[0], | ||
388 | GNUNET_TESTBED_PIT_IDENTITY, | ||
389 | peer_information_cb, | ||
390 | (void *) 0); | ||
391 | // Get GNUnet identity of member | ||
392 | pi_op1 = GNUNET_TESTBED_peer_get_information (peers[1], | ||
393 | GNUNET_TESTBED_PIT_IDENTITY, | ||
394 | peer_information_cb, | ||
395 | (void *) 1); | ||
396 | |||
397 | /* Connection to service successful. Here we'd usually do something with | ||
398 | * the service. */ | ||
399 | result = GNUNET_OK; | ||
400 | //GNUNET_SCHEDULER_shutdown (); /* Also kills the testbed */ | ||
401 | } | ||
402 | |||
403 | |||
404 | |||
405 | /** | ||
406 | * Function run when service multicast has started and is providing us | ||
407 | * with a configuration file. | ||
408 | */ | ||
409 | static void * | ||
410 | multicast_ca0 (void *cls, | ||
411 | const struct GNUNET_CONFIGURATION_Handle *cfg) | ||
412 | { | ||
413 | group_key = GNUNET_CRYPTO_eddsa_key_create (); | ||
414 | GNUNET_CRYPTO_eddsa_key_get_public (group_key, &group_pub_key); | ||
415 | |||
416 | return GNUNET_MULTICAST_origin_start (cfg, | ||
417 | group_key, | ||
418 | 0, | ||
419 | origin_join_request, | ||
420 | NULL, /* no test for origin_replay_frag */ | ||
421 | NULL, /* no test for origin_replay_msg */ | ||
422 | origin_request, | ||
423 | origin_message, | ||
424 | NULL); | ||
425 | } | ||
426 | |||
427 | static void | ||
428 | multicast_da0 (void *cls, | ||
429 | void *op_result) | ||
430 | { | ||
431 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
432 | "Origin closes multicast group\n"); | ||
433 | |||
434 | GNUNET_MULTICAST_origin_stop (origin, NULL, NULL); | ||
435 | } | ||
436 | |||
437 | |||
438 | /** | ||
439 | * Main function inovked from TESTBED once all of the | ||
440 | * peers are up and running. This one then connects | ||
441 | * just to the multicast service of peer 0 and 1. | ||
442 | * Peer 0 is going to be origin. | ||
443 | * Peer 1 is going to be one member. | ||
444 | * Origin will start a multicast group and the member will try to join it. | ||
445 | * After that we execute some multicast test. | ||
446 | * | ||
447 | * @param cls closure | ||
448 | * @param h the run handle | ||
449 | * @param peers started peers for the test | ||
450 | * @param num_peers size of the 'peers' array | ||
451 | * @param links_succeeded number of links between peers that were created | ||
452 | * @param links_failed number of links testbed was unable to establish | ||
453 | */ | ||
454 | static void | ||
455 | testbed_master (void *cls, | ||
456 | struct GNUNET_TESTBED_RunHandle *h, | ||
457 | unsigned int num_peers, | ||
458 | struct GNUNET_TESTBED_Peer **p, | ||
459 | unsigned int links_succeeded, | ||
460 | unsigned int links_failed) | ||
461 | { | ||
462 | /* Testbed is ready with peers running and connected in a pre-defined overlay | ||
463 | topology (FIXME) */ | ||
464 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
465 | "Connected to testbed_master()\n"); | ||
466 | |||
467 | peers = p; | ||
468 | |||
469 | GNUNET_log (GNUNET_ERROR_TYPE_INFO, | ||
470 | "Create origin peer\n"); | ||
471 | op0 = GNUNET_TESTBED_service_connect (NULL, /* Closure for operation */ | ||
472 | peers[0], /* The peer whose service to connect to */ | ||
473 | "multicast", /* The name of the service */ | ||
474 | service_connect0, /* callback to call after a handle to service | ||
475 | is opened */ | ||
476 | NULL, /* closure for the above callback */ | ||
477 | multicast_ca0, /* callback to call with peer's configuration; | ||
478 | this should open the needed service connection */ | ||
479 | multicast_da0, /* callback to be called when closing the | ||
480 | opened service connection */ | ||
481 | NULL); /* closure for the above two callbacks */ | ||
482 | |||
483 | GNUNET_SCHEDULER_add_shutdown (&shutdown_task, NULL); /* Schedule a new task on shutdown */ | ||
484 | |||
485 | /* Schedule the shutdown task with a delay of a few Seconds */ | ||
486 | timeout_tid = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 50), | ||
487 | &timeout_task, NULL); | ||
488 | } | ||
489 | |||
490 | |||
491 | int | ||
492 | main (int argc, char *argv[]) | ||
493 | { | ||
494 | int ret; | ||
495 | |||
496 | result = GNUNET_SYSERR; | ||
497 | ret = GNUNET_TESTBED_test_run | ||
498 | ("test-multicast-multipeer", /* test case name */ | ||
499 | "test_multicast.conf", /* template configuration */ | ||
500 | NUM_PEERS, /* number of peers to start */ | ||
501 | 0LL, /* Event mask - set to 0 for no event notifications */ | ||
502 | NULL, /* Controller event callback */ | ||
503 | NULL, /* Closure for controller event callback */ | ||
504 | testbed_master, /* continuation callback to be called when testbed setup is complete */ | ||
505 | NULL); /* Closure for the test_master callback */ | ||
506 | if ( (GNUNET_OK != ret) || (GNUNET_OK != result) ) | ||
507 | return 1; | ||
508 | return 0; | ||
509 | } | ||
510 | |||
511 | /* end of test_multicast_multipeer.c */ | ||