aboutsummaryrefslogtreecommitdiff
path: root/src/transport/transport-testing.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/transport/transport-testing.c')
-rw-r--r--src/transport/transport-testing.c931
1 files changed, 0 insertions, 931 deletions
diff --git a/src/transport/transport-testing.c b/src/transport/transport-testing.c
deleted file mode 100644
index 00c4a08dd..000000000
--- a/src/transport/transport-testing.c
+++ /dev/null
@@ -1,931 +0,0 @@
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-testing.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->oh)
228 {
229 GNUNET_TRANSPORT_offer_hello_cancel (cc->oh);
230 cc->oh = NULL;
231 }
232 if (NULL != cc->ats_sh)
233 {
234 GNUNET_ATS_connectivity_suggest_cancel (cc->ats_sh);
235 cc->ats_sh = NULL;
236 }
237 }
238 }
239 /* then notify application */
240 for (cc = tth->cc_head; NULL != cc; cc = ccn)
241 {
242 ccn = cc->next;
243 if ((GNUNET_YES == cc->connected) &&
244 (NULL != cc->cb))
245 {
246 cc->cb (cc->cb_cls);
247 cc->cb = NULL; /* only notify once! */
248 }
249 }
250 return ret;
251}
252
253
254/**
255 * Offer the current HELLO of P2 to P1.
256 *
257 * @param cls our `struct GNUNET_TRANSPORT_TESTING_ConnectRequest`
258 */
259static void
260offer_hello (void *cls);
261
262
263static void
264notify_disconnect (void *cls,
265 const struct GNUNET_PeerIdentity *peer,
266 void *handler_cls)
267{
268 struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cls;
269 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p->tth;
270 char *p2_s;
271 /* Find PeerContext */
272 int no = 0;
273 struct GNUNET_TRANSPORT_TESTING_PeerContext *p2 = NULL;
274 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
275
276 p2 = find_peer_context (p->tth,
277 peer);
278 no = p->no;
279 if (NULL != p2)
280 GNUNET_asprintf (&p2_s,
281 "%u (`%s')",
282 p2->no,
283 GNUNET_i2s (&p2->id));
284 else
285 GNUNET_asprintf (&p2_s,
286 "`%s'",
287 GNUNET_i2s (peer));
288 LOG (GNUNET_ERROR_TYPE_DEBUG,
289 "Peers %s disconnected from peer %u (`%s')\n",
290 p2_s,
291 no,
292 GNUNET_i2s (&p->id));
293 GNUNET_free (p2_s);
294 /* notify about disconnect */
295 if (NULL != p->nd)
296 p->nd (p->cb_cls,
297 peer,
298 handler_cls);
299 if (NULL == p2)
300 return;
301 /* clear MQ, it is now invalid */
302 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
303 p2,
304 &set_mq,
305 NULL);
306 /* update set connected flags for all requests */
307 GNUNET_TRANSPORT_TESTING_find_connecting_context (p,
308 p2,
309 &clear_p1c,
310 NULL);
311 GNUNET_TRANSPORT_TESTING_find_connecting_context (p2,
312 p,
313 &clear_p2c,
314 NULL);
315 /* resume connectivity requests as necessary */
316 for (cc = tth->cc_head; NULL != cc; cc = cc->next)
317 {
318 if (GNUNET_NO == cc->connected)
319 continue;
320 if ((GNUNET_YES != cc->p1_c) ||
321 (GNUNET_YES != cc->p2_c))
322 {
323 cc->connected = GNUNET_NO;
324 /* start trying to connect */
325 if ((NULL == cc->tct) &&
326 (NULL == cc->oh))
327 cc->tct = GNUNET_SCHEDULER_add_now (&offer_hello,
328 cc);
329 if (NULL == cc->ats_sh)
330 cc->ats_sh = GNUNET_ATS_connectivity_suggest (cc->p1->ats,
331 &p2->id,
332 1);
333 }
334 }
335}
336
337
338static void
339get_hello (void *cb_cls,
340 const struct GNUNET_MessageHeader *message)
341{
342 struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cb_cls;
343 struct GNUNET_PeerIdentity hello_id;
344
345 GNUNET_assert (GNUNET_OK ==
346 GNUNET_HELLO_get_id ((const struct
347 GNUNET_HELLO_Message *) message,
348 &hello_id));
349 GNUNET_assert (0 == memcmp (&hello_id,
350 &p->id,
351 sizeof(hello_id)));
352 GNUNET_free (p->hello);
353 p->hello = (struct GNUNET_HELLO_Message *) GNUNET_copy_message (message);
354
355 if (NULL != p->start_cb)
356 {
357 LOG (GNUNET_ERROR_TYPE_DEBUG,
358 "Peer %u (`%s') successfully started\n",
359 p->no,
360 GNUNET_i2s (&p->id));
361 p->start_cb (p->start_cb_cls);
362 p->start_cb = NULL;
363 }
364}
365
366
367/**
368 * Start a peer with the given configuration
369 * @param tth the testing handle
370 * @param cfgname configuration file
371 * @param peer_id a unique number to identify the peer
372 * @param handlers functions for receiving messages
373 * @param nc connect callback
374 * @param nd disconnect callback
375 * @param cb_cls closure for callback
376 * @param start_cb start callback
377 * @param start_cb_cls closure for callback
378 * @return the peer context
379 */
380struct GNUNET_TRANSPORT_TESTING_PeerContext *
381GNUNET_TRANSPORT_TESTING_start_peer (struct
382 GNUNET_TRANSPORT_TESTING_Handle *tth,
383 const char *cfgname,
384 int peer_id,
385 const struct
386 GNUNET_MQ_MessageHandler *handlers,
387 GNUNET_TRANSPORT_NotifyConnect nc,
388 GNUNET_TRANSPORT_NotifyDisconnect nd,
389 void *cb_cls,
390 GNUNET_SCHEDULER_TaskCallback start_cb,
391 void *start_cb_cls)
392{
393 char *emsg = NULL;
394 struct GNUNET_TRANSPORT_TESTING_PeerContext *p;
395 struct GNUNET_PeerIdentity dummy;
396 unsigned int i;
397
398 if (GNUNET_NO == GNUNET_DISK_file_test (cfgname))
399 {
400 LOG (GNUNET_ERROR_TYPE_ERROR,
401 "File not found: `%s'\n",
402 cfgname);
403 return NULL;
404 }
405
406 p = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_PeerContext);
407 p->tth = tth;
408 p->nc = nc;
409 p->nd = nd;
410 if (NULL != handlers)
411 {
412 for (i = 0; NULL != handlers[i].cb; i++)
413 ;
414 p->handlers = GNUNET_new_array (i + 1,
415 struct GNUNET_MQ_MessageHandler);
416 GNUNET_memcpy (p->handlers,
417 handlers,
418 i * sizeof(struct GNUNET_MQ_MessageHandler));
419 }
420 if (NULL != cb_cls)
421 p->cb_cls = cb_cls;
422 else
423 p->cb_cls = p;
424 p->start_cb = start_cb;
425 if (NULL != start_cb_cls)
426 p->start_cb_cls = start_cb_cls;
427 else
428 p->start_cb_cls = p;
429 GNUNET_CONTAINER_DLL_insert (tth->p_head,
430 tth->p_tail,
431 p);
432
433 /* Create configuration and call testing lib to modify it */
434 p->cfg = GNUNET_CONFIGURATION_create ();
435 GNUNET_assert (GNUNET_OK ==
436 GNUNET_CONFIGURATION_load (p->cfg, cfgname));
437 if (GNUNET_SYSERR ==
438 GNUNET_TESTING_configuration_create (tth->tl_system,
439 p->cfg))
440 {
441 LOG (GNUNET_ERROR_TYPE_ERROR,
442 "Testing library failed to create unique configuration based on `%s'\n",
443 cfgname);
444 GNUNET_CONFIGURATION_destroy (p->cfg);
445 GNUNET_free (p);
446 return NULL;
447 }
448
449 p->no = peer_id;
450 /* Configure peer with configuration */
451 p->peer = GNUNET_TESTING_peer_configure (tth->tl_system,
452 p->cfg,
453 p->no,
454 NULL,
455 &emsg);
456 if (NULL == p->peer)
457 {
458 LOG (GNUNET_ERROR_TYPE_ERROR,
459 "Testing library failed to create unique configuration based on `%s': `%s'\n",
460 cfgname,
461 emsg);
462 GNUNET_TRANSPORT_TESTING_stop_peer (p);
463 GNUNET_free (emsg);
464 return NULL;
465 }
466
467 if (GNUNET_OK != GNUNET_TESTING_peer_start (p->peer))
468 {
469 LOG (GNUNET_ERROR_TYPE_ERROR,
470 "Testing library failed to create unique configuration based on `%s'\n",
471 cfgname);
472 GNUNET_TRANSPORT_TESTING_stop_peer (p);
473 return NULL;
474 }
475
476 memset (&dummy,
477 '\0',
478 sizeof(dummy));
479 GNUNET_TESTING_peer_get_identity (p->peer,
480 &p->id);
481 if (0 == memcmp (&dummy,
482 &p->id,
483 sizeof(struct GNUNET_PeerIdentity)))
484 {
485 LOG (GNUNET_ERROR_TYPE_ERROR,
486 "Testing library failed to obtain peer identity for peer %u\n",
487 p->no);
488 GNUNET_TRANSPORT_TESTING_stop_peer (p);
489 return NULL;
490 }
491 LOG (GNUNET_ERROR_TYPE_DEBUG,
492 "Peer %u configured with identity `%s'\n",
493 p->no,
494 GNUNET_i2s_full (&p->id));
495 p->tmh = GNUNET_TRANSPORT_manipulation_connect (p->cfg);
496 p->th = GNUNET_TRANSPORT_core_connect (p->cfg,
497 NULL,
498 handlers,
499 p,
500 &notify_connect,
501 &notify_disconnect,
502 NULL);
503 if ((NULL == p->th) ||
504 (NULL == p->tmh))
505 {
506 LOG (GNUNET_ERROR_TYPE_ERROR,
507 "Failed to connect to transport service for peer `%s': `%s'\n",
508 cfgname,
509 emsg);
510 GNUNET_TRANSPORT_TESTING_stop_peer (p);
511 GNUNET_free (emsg);
512 return NULL;
513 }
514 p->ats = GNUNET_ATS_connectivity_init (p->cfg);
515 if (NULL == p->ats)
516 {
517 LOG (GNUNET_ERROR_TYPE_ERROR,
518 "Failed to connect to ATS service for peer `%s': `%s'\n",
519 cfgname,
520 emsg);
521 GNUNET_TRANSPORT_TESTING_stop_peer (p);
522 GNUNET_free (emsg);
523 return NULL;
524 }
525 p->ghh = GNUNET_TRANSPORT_hello_get (p->cfg,
526 GNUNET_TRANSPORT_AC_ANY,
527 &get_hello,
528 p);
529 GNUNET_assert (NULL != p->ghh);
530 return p;
531}
532
533
534/**
535 * Stops and restarts the given peer, sleeping (!) for 5s in between.
536 *
537 * @param p the peer
538 * @param restart_cb callback to call when restarted
539 * @param restart_cb_cls callback closure
540 * @return #GNUNET_OK in success otherwise #GNUNET_SYSERR
541 */
542int
543GNUNET_TRANSPORT_TESTING_restart_peer (struct
544 GNUNET_TRANSPORT_TESTING_PeerContext *p,
545 GNUNET_SCHEDULER_TaskCallback restart_cb,
546 void *restart_cb_cls)
547{
548 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
549 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
550
551 /* shutdown */
552 LOG (GNUNET_ERROR_TYPE_DEBUG,
553 "Stopping peer %u (`%s')\n",
554 p->no,
555 GNUNET_i2s (&p->id));
556 if (NULL != p->ghh)
557 {
558 GNUNET_TRANSPORT_hello_get_cancel (p->ghh);
559 p->ghh = NULL;
560 }
561 if (NULL != p->th)
562 {
563 GNUNET_TRANSPORT_core_disconnect (p->th);
564 p->th = NULL;
565 }
566 if (NULL != p->tmh)
567 {
568 GNUNET_TRANSPORT_manipulation_disconnect (p->tmh);
569 p->tmh = NULL;
570 }
571 for (cc = p->tth->cc_head; NULL != cc; cc = ccn)
572 {
573 ccn = cc->next;
574 if ((cc->p1 == p) ||
575 (cc->p2 == p))
576 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
577 }
578 if (NULL != p->ats)
579 {
580 GNUNET_ATS_connectivity_done (p->ats);
581 p->ats = NULL;
582 }
583 if (GNUNET_SYSERR ==
584 GNUNET_TESTING_peer_stop (p->peer))
585 {
586 LOG (GNUNET_ERROR_TYPE_ERROR,
587 "Failed to stop peer %u (`%s')\n",
588 p->no,
589 GNUNET_i2s (&p->id));
590 return GNUNET_SYSERR;
591 }
592
593 sleep (5); // YUCK!
594
595 LOG (GNUNET_ERROR_TYPE_DEBUG,
596 "Restarting peer %u (`%s')\n",
597 p->no,
598 GNUNET_i2s (&p->id));
599 /* restart */
600 if (GNUNET_SYSERR == GNUNET_TESTING_peer_start (p->peer))
601 {
602 LOG (GNUNET_ERROR_TYPE_ERROR,
603 "Failed to restart peer %u (`%s')\n",
604 p->no,
605 GNUNET_i2s (&p->id));
606 return GNUNET_SYSERR;
607 }
608
609 GNUNET_assert (NULL == p->start_cb);
610 p->start_cb = restart_cb;
611 p->start_cb_cls = restart_cb_cls;
612
613 p->th = GNUNET_TRANSPORT_core_connect (p->cfg,
614 NULL,
615 p->handlers,
616 p,
617 &notify_connect,
618 &notify_disconnect,
619 NULL);
620 GNUNET_assert (NULL != p->th);
621 p->ats = GNUNET_ATS_connectivity_init (p->cfg);
622 p->ghh = GNUNET_TRANSPORT_hello_get (p->cfg,
623 GNUNET_TRANSPORT_AC_ANY,
624 &get_hello,
625 p);
626 GNUNET_assert (NULL != p->ghh);
627 return GNUNET_OK;
628}
629
630
631/**
632 * Shutdown the given peer
633 *
634 * @param p the peer
635 */
636void
637GNUNET_TRANSPORT_TESTING_stop_peer (struct
638 GNUNET_TRANSPORT_TESTING_PeerContext *p)
639{
640 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p->tth;
641 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
642 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
643
644 for (cc = tth->cc_head; NULL != cc; cc = ccn)
645 {
646 ccn = cc->next;
647 if ((cc->p1 == p) ||
648 (cc->p2 == p))
649 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
650 }
651 if (NULL != p->ghh)
652 {
653 GNUNET_TRANSPORT_hello_get_cancel (p->ghh);
654 p->ghh = NULL;
655 }
656 if (NULL != p->tmh)
657 {
658 GNUNET_TRANSPORT_manipulation_disconnect (p->tmh);
659 p->tmh = NULL;
660 }
661 if (NULL != p->th)
662 {
663 GNUNET_TRANSPORT_core_disconnect (p->th);
664 p->th = NULL;
665 }
666 if (NULL != p->peer)
667 {
668 if (GNUNET_OK !=
669 GNUNET_TESTING_peer_stop (p->peer))
670 {
671 LOG (GNUNET_ERROR_TYPE_DEBUG,
672 "Testing lib failed to stop peer %u (`%s')\n",
673 p->no,
674 GNUNET_i2s (&p->id));
675 }
676 GNUNET_TESTING_peer_destroy (p->peer);
677 p->peer = NULL;
678 }
679 if (NULL != p->ats)
680 {
681 GNUNET_ATS_connectivity_done (p->ats);
682 p->ats = NULL;
683 }
684 if (NULL != p->hello)
685 {
686 GNUNET_free (p->hello);
687 p->hello = NULL;
688 }
689 if (NULL != p->cfg)
690 {
691 GNUNET_CONFIGURATION_destroy (p->cfg);
692 p->cfg = NULL;
693 }
694 if (NULL != p->handlers)
695 {
696 GNUNET_free (p->handlers);
697 p->handlers = NULL;
698 }
699 GNUNET_CONTAINER_DLL_remove (tth->p_head,
700 tth->p_tail,
701 p);
702 LOG (GNUNET_ERROR_TYPE_DEBUG,
703 "Peer %u (`%s') stopped\n",
704 p->no,
705 GNUNET_i2s (&p->id));
706 GNUNET_free (p);
707}
708
709
710/**
711 * Function called after the HELLO was passed to the
712 * transport service.
713 */
714static void
715hello_offered (void *cls)
716{
717 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc = cls;
718
719 cc->oh = NULL;
720 cc->tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
721 &offer_hello,
722 cc);
723}
724
725
726/**
727 * Offer the current HELLO of P2 to P1.
728 *
729 * @param cls our `struct GNUNET_TRANSPORT_TESTING_ConnectRequest`
730 */
731static void
732offer_hello (void *cls)
733{
734 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc = cls;
735 struct GNUNET_TRANSPORT_TESTING_PeerContext *p1 = cc->p1;
736 struct GNUNET_TRANSPORT_TESTING_PeerContext *p2 = cc->p2;
737
738 cc->tct = NULL;
739 {
740 char *p2_s = GNUNET_strdup (GNUNET_i2s (&p2->id));
741
742 LOG (GNUNET_ERROR_TYPE_DEBUG,
743 "Asking peer %u (`%s') to connect peer %u (`%s'), providing HELLO with %u bytes\n",
744 p1->no,
745 GNUNET_i2s (&p1->id),
746 p2->no,
747 p2_s,
748 GNUNET_HELLO_size (cc->p2->hello));
749 GNUNET_free (p2_s);
750 }
751
752 if (NULL != cc->oh)
753 GNUNET_TRANSPORT_offer_hello_cancel (cc->oh);
754 cc->oh =
755 GNUNET_TRANSPORT_offer_hello (cc->p1->cfg,
756 (const struct
757 GNUNET_MessageHeader *) cc->p2->hello,
758 &hello_offered,
759 cc);
760}
761
762
763/**
764 * Initiate a connection from p1 to p2 by offering p1 p2's HELLO message
765 *
766 * Remarks: start_peer's notify_connect callback can be called before.
767 *
768 * @param tth transport testing handle
769 * @param p1 peer 1
770 * @param p2 peer 2
771 * @param cb the callback to call when both peers notified that they are connected
772 * @param cls callback cls
773 * @return a connect request handle
774 */
775struct GNUNET_TRANSPORT_TESTING_ConnectRequest *
776GNUNET_TRANSPORT_TESTING_connect_peers (struct
777 GNUNET_TRANSPORT_TESTING_PeerContext *p1,
778 struct
779 GNUNET_TRANSPORT_TESTING_PeerContext *p2,
780 GNUNET_SCHEDULER_TaskCallback cb,
781 void *cls)
782{
783 struct GNUNET_TRANSPORT_TESTING_Handle *tth = p1->tth;
784 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
785 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ccn;
786
787 ccn = NULL;
788 for (cc = tth->cc_head; NULL != cc; cc = cc->next)
789 {
790 if ((cc->p1 == p1) &&
791 (cc->p2 == p2))
792 {
793 ccn = cc;
794 break;
795 }
796 }
797
798 cc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_ConnectRequest);
799 cc->p1 = p1;
800 cc->p2 = p2;
801 cc->cb = cb;
802 if (NULL != cls)
803 cc->cb_cls = cls;
804 else
805 cc->cb_cls = cc;
806 if (NULL != ccn)
807 {
808 cc->p1_c = ccn->p1_c;
809 cc->p2_c = ccn->p2_c;
810 cc->connected = ccn->connected;
811 }
812 GNUNET_CONTAINER_DLL_insert (tth->cc_head,
813 tth->cc_tail,
814 cc);
815 cc->tct = GNUNET_SCHEDULER_add_now (&offer_hello,
816 cc);
817 cc->ats_sh = GNUNET_ATS_connectivity_suggest (cc->p1->ats,
818 &p2->id,
819 1);
820 LOG (GNUNET_ERROR_TYPE_DEBUG,
821 "New connect request %p\n",
822 cc);
823 return cc;
824}
825
826
827/**
828 * Cancel the request to connect two peers
829 * Tou MUST cancel the request if you stop the peers before the peers connected successfully
830 *
831 * @param tth transport testing handle
832 * @param cc a connect request handle
833 */
834void
835GNUNET_TRANSPORT_TESTING_connect_peers_cancel (struct
836 GNUNET_TRANSPORT_TESTING_ConnectRequest
837 *cc)
838{
839 struct GNUNET_TRANSPORT_TESTING_Handle *tth = cc->p1->tth;
840
841 LOG (GNUNET_ERROR_TYPE_DEBUG,
842 "Canceling connect request!\n");
843 if (NULL != cc->tct)
844 {
845 GNUNET_SCHEDULER_cancel (cc->tct);
846 cc->tct = NULL;
847 }
848 if (NULL != cc->oh)
849 {
850 GNUNET_TRANSPORT_offer_hello_cancel (cc->oh);
851 cc->oh = NULL;
852 }
853 if (NULL != cc->ats_sh)
854 {
855 GNUNET_ATS_connectivity_suggest_cancel (cc->ats_sh);
856 cc->ats_sh = NULL;
857 }
858 GNUNET_CONTAINER_DLL_remove (tth->cc_head,
859 tth->cc_tail,
860 cc);
861 GNUNET_free (cc);
862}
863
864
865/**
866 * Clean up the transport testing
867 *
868 * @param tth transport testing handle
869 */
870void
871GNUNET_TRANSPORT_TESTING_done (struct GNUNET_TRANSPORT_TESTING_Handle *tth)
872{
873 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *cc;
874 struct GNUNET_TRANSPORT_TESTING_ConnectRequest *ct;
875 struct GNUNET_TRANSPORT_TESTING_PeerContext *p;
876 struct GNUNET_TRANSPORT_TESTING_PeerContext *t;
877
878 if (NULL == tth)
879 return;
880 cc = tth->cc_head;
881 while (NULL != cc)
882 {
883 ct = cc->next;
884 LOG (GNUNET_ERROR_TYPE_ERROR,
885 "Developer forgot to cancel connect request!\n");
886 GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc);
887 cc = ct;
888 }
889 p = tth->p_head;
890 while (NULL != p)
891 {
892 t = p->next;
893 LOG (GNUNET_ERROR_TYPE_ERROR,
894 "Developer forgot to stop peer!\n");
895 GNUNET_TRANSPORT_TESTING_stop_peer (p);
896 p = t;
897 }
898 GNUNET_TESTING_system_destroy (tth->tl_system,
899 GNUNET_YES);
900
901 GNUNET_free (tth);
902}
903
904
905/**
906 * Initialize the transport testing
907 *
908 * @return transport testing handle
909 */
910struct GNUNET_TRANSPORT_TESTING_Handle *
911GNUNET_TRANSPORT_TESTING_init ()
912{
913 struct GNUNET_TRANSPORT_TESTING_Handle *tth;
914
915 tth = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_Handle);
916 tth->tl_system = GNUNET_TESTING_system_create ("transport-testing",
917 NULL,
918 NULL,
919 NULL);
920 if (NULL == tth->tl_system)
921 {
922 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
923 "Failed to initialize testing library!\n");
924 GNUNET_free (tth);
925 return NULL;
926 }
927 return tth;
928}
929
930
931/* end of transport-testing.c */
generated by cgit v1.2.3 (git 2.39.1) at 2024-04-20 02:18:37 +0000