aboutsummaryrefslogtreecommitdiff
path: root/src/transport/transport-testing2.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/transport/transport-testing2.c')
-rw-r--r--src/transport/transport-testing2.c923
1 files changed, 923 insertions, 0 deletions
diff --git a/src/transport/transport-testing2.c b/src/transport/transport-testing2.c
new file mode 100644
index 000000000..ca338c52f
--- /dev/null
+++ b/src/transport/transport-testing2.c
@@ -0,0 +1,923 @@
1/*
2 This file is part of GNUnet.
3 Copyright (C) 2006, 2009, 2015, 2016 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 transport-testing.c
22 * @brief testing lib for transport service
23 * @author Matthias Wachs
24 * @author Christian Grothoff
25 */
26#include "transport-testing2.h"
27
28
29#define LOG(kind, ...) GNUNET_log_from (kind, "transport-testing", __VA_ARGS__)
30
31
32static struct GNUNET_TRANSPORT_TESTING_PeerContext *
33find_peer_context (struct GNUNET_TRANSPORT_TESTING_Handle *tth,
34 const struct GNUNET_PeerIdentity *peer)
35{
36 struct GNUNET_TRANSPORT_TESTING_PeerContext *t;
37
38 for (t = tth->p_head; NULL != t; t = t->next)
39 if (0 == memcmp (&t->id,
40 peer,
41 sizeof(struct GNUNET_PeerIdentity)))
42 return t;
43 return NULL;
44}
45
46
47/**
48 * Find any connecting context matching the given pair of peers.
49 *
50 * @param p1 first peer
51 * @param p2 second peer
52 * @param cb function to call
53 * @param cb_cls closure for @a cb
54 */
55void
56GNUNET_TRANSPORT_TESTING_find_connecting_context (struct
57 GNUNET_TRANSPORT_TESTING_PeerContext
58 *p1,
59 struct
60 GNUNET_TRANSPORT_TESTING_PeerContext
61 *p2,
62 GNUNET_TRANSPORT_TESTING_ConnectContextCallback
63 cb,
64 void *cb_cls)
65{
66 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p1->tth;
67 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
68 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
69
70 for (cc = tth->cc_head; NULL != cc; cc = ccn)
71 {
72 ccn = cc->next;
73 if ((cc->p1 == p1) &&
74 (cc->p2 == p2))
75 cb (cb_cls,
76 cc);
77 }
78}
79
80
81static void
82set_p1c (void *cls,
83 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cx)
84{
85 int *found = cls;
86
87 if (NULL != found)
88 *found = GNUNET_YES;
89 cx->p1_c = GNUNET_YES;
90}
91
92
93static void
94set_mq (void *cls,
95 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cx)
96{
97 struct GNUNET_MQ_Handle *mq = cls;
98
99 cx->mq = mq;
100}
101
102
103static void
104set_p2c (void *cls,
105 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cx)
106{
107 int *found = cls;
108
109 if (NULL != found)
110 *found = GNUNET_YES;
111 cx->p2_c = GNUNET_YES;
112}
113
114
115static void
116clear_p1c (void *cls,
117 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cx)
118{
119 int *found = cls;
120
121 if (NULL != found)
122 *found = GNUNET_YES;
123 cx->p1_c = GNUNET_NO;
124}
125
126
127static void
128clear_p2c (void *cls,
129 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cx)
130{
131 int *found = cls;
132
133 if (NULL != found)
134 *found = GNUNET_YES;
135 cx->p2_c = GNUNET_NO;
136}
137
138
139static void *
140notify_connect (void *cls,
141 const struct GNUNET_PeerIdentity *peer,
142 struct GNUNET_MQ_Handle *mq)
143{
144 struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cls;
145 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p->tth;
146 char *p2_s;
147 struct GNUNET_TRANSPORT_TESTING_PeerContext *p2;
148 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
149 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
150 int found;
151 void *ret;
152
153 p2 = find_peer_context (p->tth,
154 peer);
155 if (NULL != p->nc)
156 ret = p->nc (p->cb_cls,
157 peer,
158 mq);
159 else
160 ret = NULL;
161
162 if (NULL != p2)
163 GNUNET_asprintf (&p2_s,
164 "%u (`%s')",
165 p2->no,
166 GNUNET_i2s (&p2->id));
167 else
168 GNUNET_asprintf (&p2_s,
169 "`%s'",
170 GNUNET_i2s (peer));
171 LOG (GNUNET_ERROR_TYPE_DEBUG,
172 "Peers %s connected to peer %u (`%s')\n",
173 p2_s,
174 p->no,
175 GNUNET_i2s (&p->id));
176 GNUNET_free (p2_s);
177 /* update flags in connecting contexts */
178 found = GNUNET_NO;
179 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
180 p2,
181 &set_p1c,
182 &found);
183 if (GNUNET_NO == found)
184 {
185 cc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_ConnectRequest);
186 cc->p1 = p;
187 cc->p2 = p2;
188 cc->p1_c = GNUNET_YES;
189 GNUNET_CONTAINER_DLL_insert (tth->cc_head,
190 tth->cc_tail,
191 cc);
192 }
193 found = GNUNET_NO;
194 GNUNET_TRANSPORT_TESTING_find_connecting_context (p2,
195 p,
196 &set_p2c,
197 &found);
198 if (GNUNET_NO == found)
199 {
200 cc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_ConnectRequest);
201 cc->p1 = p2;
202 cc->p2 = p;
203 cc->p1_c = GNUNET_YES;
204 GNUNET_CONTAINER_DLL_insert (tth->cc_head,
205 tth->cc_tail,
206 cc);
207 }
208 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
209 p2,
210 &set_mq,
211 mq);
212 /* update set connected flag for all requests */
213 for (cc = tth->cc_head; NULL != cc; cc = cc->next)
214 {
215 if (GNUNET_YES == cc->connected)
216 continue;
217 if ((GNUNET_YES == cc->p1_c) &&
218 (GNUNET_YES == cc->p2_c))
219 {
220 cc->connected = GNUNET_YES;
221 /* stop trying to connect */
222 if (NULL != cc->tct)
223 {
224 GNUNET_SCHEDULER_cancel (cc->tct);
225 cc->tct = NULL;
226 }
227 if (NULL != cc->ats_sh)
228 {
229 GNUNET_ATS_connectivity_suggest_cancel (cc->ats_sh);
230 cc->ats_sh = NULL;
231 }
232 }
233 }
234 /* then notify application */
235 for (cc = tth->cc_head; NULL != cc; cc = ccn)
236 {
237 ccn = cc->next;
238 if ((GNUNET_YES == cc->connected) &&
239 (NULL != cc->cb))
240 {
241 cc->cb (cc->cb_cls);
242 cc->cb = NULL; /* only notify once! */
243 }
244 }
245 return ret;
246}
247
248
249/**
250 * Offer the current HELLO of P2 to P1.
251 *
252 * @param cls our `struct GNUNET_TRANSPORT_TESTING_ConnectRequest`
253 */
254static void
255offer_hello (void *cls);
256
257
258static void
259notify_disconnect (void *cls,
260 const struct GNUNET_PeerIdentity *peer,
261 void *handler_cls)
262{
263 struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cls;
264 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p->tth;
265 char *p2_s;
266 /* Find PeerContext */
267 int no = 0;
268 struct GNUNET_TRANSPORT_TESTING_PeerContext *p2 = NULL;
269 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
270
271 p2 = find_peer_context (p->tth,
272 peer);
273 no = p->no;
274 if (NULL != p2)
275 GNUNET_asprintf (&p2_s,
276 "%u (`%s')",
277 p2->no,
278 GNUNET_i2s (&p2->id));
279 else
280 GNUNET_asprintf (&p2_s,
281 "`%s'",
282 GNUNET_i2s (peer));
283 LOG (GNUNET_ERROR_TYPE_DEBUG,
284 "Peers %s disconnected from peer %u (`%s')\n",
285 p2_s,
286 no,
287 GNUNET_i2s (&p->id));
288 GNUNET_free (p2_s);
289 /* notify about disconnect */
290 if (NULL != p->nd)
291 p->nd (p->cb_cls,
292 peer,
293 handler_cls);
294 if (NULL == p2)
295 return;
296 /* clear MQ, it is now invalid */
297 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
298 p2,
299 &set_mq,
300 NULL);
301 /* update set connected flags for all requests */
302 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
303 p2,
304 &clear_p1c,
305 NULL);
306 GNUNET_TRANSPORT_TESTING_find_connecting_context (p2,
307 p,
308 &clear_p2c,
309 NULL);
310 /* resume connectivity requests as necessary */
311 for (cc = tth->cc_head; NULL != cc; cc = cc->next)
312 {
313 if (GNUNET_NO == cc->connected)
314 continue;
315 if ((GNUNET_YES != cc->p1_c) ||
316 (GNUNET_YES != cc->p2_c))
317 {
318 cc->connected = GNUNET_NO;
319 /* start trying to connect */
320 if (NULL == cc->tct)
321 cc->tct = GNUNET_SCHEDULER_add_now (&offer_hello,
322 cc);
323 if (NULL == cc->ats_sh)
324 cc->ats_sh = GNUNET_ATS_connectivity_suggest (cc->p1->ats,
325 &p2->id,
326 1);
327 }
328 }
329}
330
331
332static void
333hello_iter_cb (void *cb_cls,
334 const struct GNUNET_PEERSTORE_Record *record,
335 const char *emsg)
336{
337 struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cb_cls;
338 //Check record type et al?
339 p->hello = GNUNET_malloc (record->value_size);
340 memcpy (p->hello, record->value, record->value_size);
341
342 GNUNET_PEERSTORE_iterate_cancel (p->pic);
343 if (NULL != p->start_cb)
344 {
345 LOG (GNUNET_ERROR_TYPE_DEBUG,
346 "Peer %u (`%s') successfully started\n",
347 p->no,
348 GNUNET_i2s (&p->id));
349 p->start_cb (p->start_cb_cls);
350 p->start_cb = NULL;
351 }
352}
353
354
355/**
356 * Start a peer with the given configuration
357 * @param tth the testing handle
358 * @param cfgname configuration file
359 * @param peer_id a unique number to identify the peer
360 * @param handlers functions for receiving messages
361 * @param nc connect callback
362 * @param nd disconnect callback
363 * @param cb_cls closure for callback
364 * @param start_cb start callback
365 * @param start_cb_cls closure for callback
366 * @return the peer context
367 */
368struct GNUNET_TRANSPORT_TESTING_PeerContext *
369GNUNET_TRANSPORT_TESTING_start_peer (struct
370 GNUNET_TRANSPORT_TESTING_Handle *tth,
371 const char *cfgname,
372 int peer_id,
373 const struct
374 GNUNET_MQ_MessageHandler *handlers,
375 GNUNET_TRANSPORT_NotifyConnect nc,
376 GNUNET_TRANSPORT_NotifyDisconnect nd,
377 void *cb_cls,
378 GNUNET_SCHEDULER_TaskCallback start_cb,
379 void *start_cb_cls)
380{
381 char *emsg = NULL;
382 struct GNUNET_TRANSPORT_TESTING_PeerContext *p;
383 struct GNUNET_PeerIdentity dummy;
384 unsigned int i;
385
386 if (GNUNET_NO == GNUNET_DISK_file_test (cfgname))
387 {
388 LOG (GNUNET_ERROR_TYPE_ERROR,
389 "File not found: `%s'\n",
390 cfgname);
391 return NULL;
392 }
393
394 p = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_PeerContext);
395 p->tth = tth;
396 p->nc = nc;
397 p->nd = nd;
398 if (NULL != handlers)
399 {
400 for (i = 0; NULL != handlers[i].cb; i++)
401 ;
402 p->handlers = GNUNET_new_array (i + 1,
403 struct GNUNET_MQ_MessageHandler);
404 GNUNET_memcpy (p->handlers,
405 handlers,
406 i * sizeof(struct GNUNET_MQ_MessageHandler));
407 }
408 if (NULL != cb_cls)
409 p->cb_cls = cb_cls;
410 else
411 p->cb_cls = p;
412 p->start_cb = start_cb;
413 if (NULL != start_cb_cls)
414 p->start_cb_cls = start_cb_cls;
415 else
416 p->start_cb_cls = p;
417 GNUNET_CONTAINER_DLL_insert (tth->p_head,
418 tth->p_tail,
419 p);
420
421 /* Create configuration and call testing lib to modify it */
422 p->cfg = GNUNET_CONFIGURATION_create ();
423 GNUNET_assert (GNUNET_OK ==
424 GNUNET_CONFIGURATION_load (p->cfg, cfgname));
425 if (GNUNET_SYSERR ==
426 GNUNET_TESTING_configuration_create (tth->tl_system,
427 p->cfg))
428 {
429 LOG (GNUNET_ERROR_TYPE_ERROR,
430 "Testing library failed to create unique configuration based on `%s'\n",
431 cfgname);
432 GNUNET_CONFIGURATION_destroy (p->cfg);
433 GNUNET_free (p);
434 return NULL;
435 }
436
437 p->no = peer_id;
438 /* Configure peer with configuration */
439 p->peer = GNUNET_TESTING_peer_configure (tth->tl_system,
440 p->cfg,
441 p->no,
442 NULL,
443 &emsg);
444 if (NULL == p->peer)
445 {
446 LOG (GNUNET_ERROR_TYPE_ERROR,
447 "Testing library failed to create unique configuration based on `%s': `%s'\n",
448 cfgname,
449 emsg);
450 GNUNET_TRANSPORT_TESTING_stop_peer (p);
451 GNUNET_free (emsg);
452 return NULL;
453 }
454 GNUNET_free (emsg);
455 if (GNUNET_OK != GNUNET_TESTING_peer_start (p->peer))
456 {
457 LOG (GNUNET_ERROR_TYPE_ERROR,
458 "Testing library failed to create unique configuration based on `%s'\n",
459 cfgname);
460 GNUNET_TRANSPORT_TESTING_stop_peer (p);
461 return NULL;
462 }
463
464 memset (&dummy,
465 '\0',
466 sizeof(dummy));
467 GNUNET_TESTING_peer_get_identity (p->peer,
468 &p->id);
469 if (0 == memcmp (&dummy,
470 &p->id,
471 sizeof(struct GNUNET_PeerIdentity)))
472 {
473 LOG (GNUNET_ERROR_TYPE_ERROR,
474 "Testing library failed to obtain peer identity for peer %u\n",
475 p->no);
476 GNUNET_TRANSPORT_TESTING_stop_peer (p);
477 return NULL;
478 }
479 LOG (GNUNET_ERROR_TYPE_DEBUG,
480 "Peer %u configured with identity `%s'\n",
481 p->no,
482 GNUNET_i2s_full (&p->id));
483 p->tmh = GNUNET_TRANSPORT_manipulation_connect (p->cfg);
484 p->th = GNUNET_TRANSPORT_core_connect (p->cfg,
485 NULL,
486 handlers,
487 p,
488 &notify_connect,
489 &notify_disconnect);
490 if ((NULL == p->th) ||
491 (NULL == p->tmh))
492 {
493 LOG (GNUNET_ERROR_TYPE_ERROR,
494 "Failed to connect to transport service for peer `%s': `%s'\n",
495 cfgname,
496 emsg);
497 GNUNET_TRANSPORT_TESTING_stop_peer (p);
498 return NULL;
499 }
500 p->ats = GNUNET_ATS_connectivity_init (p->cfg);
501 if (NULL == p->ats)
502 {
503 LOG (GNUNET_ERROR_TYPE_ERROR,
504 "Failed to connect to ATS service for peer `%s': `%s'\n",
505 cfgname,
506 emsg);
507 GNUNET_TRANSPORT_TESTING_stop_peer (p);
508 return NULL;
509 }
510 p->pic = GNUNET_PEERSTORE_iterate (p->ph,
511 "transport",
512 &p->id,
513 GNUNET_PEERSTORE_TRANSPORT_HELLO_KEY,
514 hello_iter_cb,
515 p);
516 GNUNET_assert (NULL != p->pic);
517
518 return p;
519}
520
521
522/**
523 * Stops and restarts the given peer, sleeping (!) for 5s in between.
524 *
525 * @param p the peer
526 * @param restart_cb callback to call when restarted
527 * @param restart_cb_cls callback closure
528 * @return #GNUNET_OK in success otherwise #GNUNET_SYSERR
529 */
530int
531GNUNET_TRANSPORT_TESTING_restart_peer (struct
532 GNUNET_TRANSPORT_TESTING_PeerContext *p,
533 GNUNET_SCHEDULER_TaskCallback restart_cb,
534 void *restart_cb_cls)
535{
536 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
537 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
538
539 /* shutdown */
540 LOG (GNUNET_ERROR_TYPE_DEBUG,
541 "Stopping peer %u (`%s')\n",
542 p->no,
543 GNUNET_i2s (&p->id));
544 if (NULL != p->pic)
545 {
546 GNUNET_PEERSTORE_iterate_cancel (p->pic);
547 p->pic = NULL;
548 }
549 if (NULL != p->th)
550 {
551 GNUNET_TRANSPORT_core_disconnect (p->th);
552 p->th = NULL;
553 }
554 if (NULL != p->tmh)
555 {
556 GNUNET_TRANSPORT_manipulation_disconnect (p->tmh);
557 p->tmh = NULL;
558 }
559 for (cc = p->tth->cc_head; NULL != cc; cc = ccn)
560 {
561 ccn = cc->next;
562 if ((cc->p1 == p) ||
563 (cc->p2 == p))
564 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
565 }
566 if (NULL != p->ats)
567 {
568 GNUNET_ATS_connectivity_done (p->ats);
569 p->ats = NULL;
570 }
571 if (GNUNET_SYSERR ==
572 GNUNET_TESTING_peer_stop (p->peer))
573 {
574 LOG (GNUNET_ERROR_TYPE_ERROR,
575 "Failed to stop peer %u (`%s')\n",
576 p->no,
577 GNUNET_i2s (&p->id));
578 return GNUNET_SYSERR;
579 }
580
581 sleep (5); // YUCK!
582
583 LOG (GNUNET_ERROR_TYPE_DEBUG,
584 "Restarting peer %u (`%s')\n",
585 p->no,
586 GNUNET_i2s (&p->id));
587 /* restart */
588 if (GNUNET_SYSERR == GNUNET_TESTING_peer_start (p->peer))
589 {
590 LOG (GNUNET_ERROR_TYPE_ERROR,
591 "Failed to restart peer %u (`%s')\n",
592 p->no,
593 GNUNET_i2s (&p->id));
594 return GNUNET_SYSERR;
595 }
596
597 GNUNET_assert (NULL == p->start_cb);
598 p->start_cb = restart_cb;
599 p->start_cb_cls = restart_cb_cls;
600
601 p->th = GNUNET_TRANSPORT_core_connect (p->cfg,
602 NULL,
603 p->handlers,
604 p,
605 &notify_connect,
606 &notify_disconnect);
607 GNUNET_assert (NULL != p->th);
608 p->ats = GNUNET_ATS_connectivity_init (p->cfg);
609 p->pic = GNUNET_PEERSTORE_iterate (p->ph,
610 "transport",
611 &p->id,
612 GNUNET_PEERSTORE_TRANSPORT_HELLO_KEY,
613 hello_iter_cb,
614 p);
615 GNUNET_assert (NULL != p->pic);
616 return GNUNET_OK;
617}
618
619
620/**
621 * Shutdown the given peer
622 *
623 * @param p the peer
624 */
625void
626GNUNET_TRANSPORT_TESTING_stop_peer (struct
627 GNUNET_TRANSPORT_TESTING_PeerContext *p)
628{
629 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p->tth;
630 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
631 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
632
633 for (cc = tth->cc_head; NULL != cc; cc = ccn)
634 {
635 ccn = cc->next;
636 if ((cc->p1 == p) ||
637 (cc->p2 == p))
638 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
639 }
640 if (NULL != p->pic)
641 {
642 GNUNET_PEERSTORE_iterate_cancel (p->pic);
643 p->pic = NULL;
644 }
645 if (NULL != p->tmh)
646 {
647 GNUNET_TRANSPORT_manipulation_disconnect (p->tmh);
648 p->tmh = NULL;
649 }
650 if (NULL != p->th)
651 {
652 GNUNET_TRANSPORT_core_disconnect (p->th);
653 p->th = NULL;
654 }
655 if (NULL != p->peer)
656 {
657 if (GNUNET_OK !=
658 GNUNET_TESTING_peer_stop (p->peer))
659 {
660 LOG (GNUNET_ERROR_TYPE_DEBUG,
661 "Testing lib failed to stop peer %u (`%s')\n",
662 p->no,
663 GNUNET_i2s (&p->id));
664 }
665 GNUNET_TESTING_peer_destroy (p->peer);
666 p->peer = NULL;
667 }
668 if (NULL != p->ats)
669 {
670 GNUNET_ATS_connectivity_done (p->ats);
671 p->ats = NULL;
672 }
673 if (NULL != p->hello)
674 {
675 GNUNET_free (p->hello);
676 p->hello = NULL;
677 }
678 if (NULL != p->cfg)
679 {
680 GNUNET_CONFIGURATION_destroy (p->cfg);
681 p->cfg = NULL;
682 }
683 if (NULL != p->handlers)
684 {
685 GNUNET_free (p->handlers);
686 p->handlers = NULL;
687 }
688 GNUNET_CONTAINER_DLL_remove (tth->p_head,
689 tth->p_tail,
690 p);
691 LOG (GNUNET_ERROR_TYPE_DEBUG,
692 "Peer %u (`%s') stopped\n",
693 p->no,
694 GNUNET_i2s (&p->id));
695 GNUNET_free (p);
696}
697
698
699/**
700 * Function called after the HELLO was passed to the
701 * transport service.
702 * FIXME maybe schedule the application_validate somehow
703 */
704/*
705static void
706hello_offered (void *cls)
707{
708 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc = cls;
709
710 cc->oh = NULL;
711 cc->tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
712 &offer_hello,
713 cc);
714}*/
715
716
717/**
718 * Offer the current HELLO of P2 to P1.
719 *
720 * @param cls our `struct GNUNET_TRANSPORT_TESTING_ConnectRequest`
721 */
722static void
723offer_hello (void *cls)
724{
725 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc = cls;
726 struct GNUNET_TRANSPORT_TESTING_PeerContext *p1 = cc->p1;
727 struct GNUNET_TRANSPORT_TESTING_PeerContext *p2 = cc->p2;
728 struct GNUNET_TIME_Absolute t;
729 enum GNUNET_NetworkType nt;
730 char *addr;
731
732 cc->tct = NULL;
733 {
734 char *p2_s = GNUNET_strdup (GNUNET_i2s (&p2->id));
735
736 LOG (GNUNET_ERROR_TYPE_DEBUG,
737 "Asking peer %u (`%s') to connect peer %u (`%s'), providing HELLO with %s\n",
738 p1->no,
739 GNUNET_i2s (&p1->id),
740 p2->no,
741 p2_s,
742 cc->p2->hello);
743 GNUNET_free (p2_s);
744 }
745
746 addr = GNUNET_HELLO_extract_address (cc->p2->hello,
747 cc->p2->hello_size,
748 &cc->p2->id,
749 &nt,
750 &t);
751 GNUNET_assert (NULL == addr);
752 GNUNET_TRANSPORT_application_validate (cc->p1->ah,
753 &cc->p2->id,
754 nt,
755 addr);
756 GNUNET_free (addr);
757}
758
759
760/**
761 * Initiate a connection from p1 to p2 by offering p1 p2's HELLO message
762 *
763 * Remarks: start_peer's notify_connect callback can be called before.
764 *
765 * @param tth transport testing handle
766 * @param p1 peer 1
767 * @param p2 peer 2
768 * @param cb the callback to call when both peers notified that they are connected
769 * @param cls callback cls
770 * @return a connect request handle
771 */
772struct GNUNET_TRANSPORT_TESTING_ConnectRequest *
773GNUNET_TRANSPORT_TESTING_connect_peers (struct
774 GNUNET_TRANSPORT_TESTING_PeerContext *p1,
775 struct
776 GNUNET_TRANSPORT_TESTING_PeerContext *p2,
777 GNUNET_SCHEDULER_TaskCallback cb,
778 void *cls)
779{
780 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p1->tth;
781 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
782 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
783
784 ccn = NULL;
785 for (cc = tth->cc_head; NULL != cc; cc = cc->next)
786 {
787 if ((cc->p1 == p1) &&
788 (cc->p2 == p2))
789 {
790 ccn = cc;
791 break;
792 }
793 }
794
795 cc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_ConnectRequest);
796 cc->p1 = p1;
797 cc->p2 = p2;
798 cc->cb = cb;
799 if (NULL != cls)
800 cc->cb_cls = cls;
801 else
802 cc->cb_cls = cc;
803 if (NULL != ccn)
804 {
805 cc->p1_c = ccn->p1_c;
806 cc->p2_c = ccn->p2_c;
807 cc->connected = ccn->connected;
808 }
809 GNUNET_CONTAINER_DLL_insert (tth->cc_head,
810 tth->cc_tail,
811 cc);
812 cc->tct = GNUNET_SCHEDULER_add_now (&offer_hello,
813 cc);
814 cc->ats_sh = GNUNET_ATS_connectivity_suggest (cc->p1->ats,
815 &p2->id,
816 1);
817 LOG (GNUNET_ERROR_TYPE_DEBUG,
818 "New connect request %p\n",
819 cc);
820 return cc;
821}
822
823
824/**
825 * Cancel the request to connect two peers
826 * Tou MUST cancel the request if you stop the peers before the peers connected succesfully
827 *
828 * @param tth transport testing handle
829 * @param cc a connect request handle
830 */
831void
832GNUNET_TRANSPORT_TESTING_connect_peers_cancel (struct
833 GNUNET_TRANSPORT_TESTING_ConnectRequest
834 *cc)
835{
836 struct GNUNET_TRANSPORT_TESTING_Handle *tth = cc->p1->tth;
837
838 LOG (GNUNET_ERROR_TYPE_DEBUG,
839 "Canceling connect request!\n");
840 if (NULL != cc->tct)
841 {
842 GNUNET_SCHEDULER_cancel (cc->tct);
843 cc->tct = NULL;
844 }
845 if (NULL != cc->ats_sh)
846 {
847 GNUNET_ATS_connectivity_suggest_cancel (cc->ats_sh);
848 cc->ats_sh = NULL;
849 }
850 GNUNET_CONTAINER_DLL_remove (tth->cc_head,
851 tth->cc_tail,
852 cc);
853 GNUNET_free (cc);
854}
855
856
857/**
858 * Clean up the transport testing
859 *
860 * @param tth transport testing handle
861 */
862void
863GNUNET_TRANSPORT_TESTING_done (struct GNUNET_TRANSPORT_TESTING_Handle *tth)
864{
865 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
866 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ct;
867 struct GNUNET_TRANSPORT_TESTING_PeerContext *p;
868 struct GNUNET_TRANSPORT_TESTING_PeerContext *t;
869
870 if (NULL == tth)
871 return;
872 cc = tth->cc_head;
873 while (NULL != cc)
874 {
875 ct = cc->next;
876 LOG (GNUNET_ERROR_TYPE_ERROR,
877 "Developer forgot to cancel connect request!\n");
878 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
879 cc = ct;
880 }
881 p = tth->p_head;
882 while (NULL != p)
883 {
884 t = p->next;
885 LOG (GNUNET_ERROR_TYPE_ERROR,
886 "Developer forgot to stop peer!\n");
887 GNUNET_TRANSPORT_TESTING_stop_peer (p);
888 p = t;
889 }
890 GNUNET_TESTING_system_destroy (tth->tl_system,
891 GNUNET_YES);
892
893 GNUNET_free (tth);
894}
895
896
897/**
898 * Initialize the transport testing
899 *
900 * @return transport testing handle
901 */
902struct GNUNET_TRANSPORT_TESTING_Handle *
903GNUNET_TRANSPORT_TESTING_init ()
904{
905 struct GNUNET_TRANSPORT_TESTING_Handle *tth;
906
907 tth = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_Handle);
908 tth->tl_system = GNUNET_TESTING_system_create ("transport-testing",
909 NULL,
910 NULL,
911 NULL);
912 if (NULL == tth->tl_system)
913 {
914 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
915 "Failed to initialize testing library!\n");
916 GNUNET_free (tth);
917 return NULL;
918 }
919 return tth;
920}
921
922
923/* end of transport-testing.c */
generated by cgit v1.2.3 (git 2.39.1) at 2024-04-23 18:57:26 +0000