2 This file is part of GNUnet.
3 (C) 2010-2013 Christian Grothoff (and other contributing authors)
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.
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.
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., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
21 * @file ats-tests/ats-testing.c
22 * @brief ats testing library: setup topology
24 * @author Christian Grothoff
25 * @author Matthias Wachs
27 #include "ats-testing.h"
31 * Connect peers with testbed
33 struct TestbedConnectOperation
36 * The benchmarking master initiating this connection
38 struct BenchmarkPeer *master;
41 * The benchmarking slave to connect to
43 struct BenchmarkPeer *slave;
46 * Testbed operation to connect peers
48 struct GNUNET_TESTBED_Operation *connect_op;
51 struct GNUNET_CONFIGURATION_Handle *cfg;
53 struct GNUNET_ATS_TEST_Topology *top;
59 * @param tc the task context
62 do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
67 struct BenchmarkPeer *p;
69 top->shutdown_task = GNUNET_SCHEDULER_NO_TASK;
70 top->state.benchmarking = GNUNET_NO;
72 GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Benchmarking done\n"));
74 GNUNET_ATS_TEST_generate_traffic_stop_all ();
76 for (c_m = 0; c_m < top->num_masters; c_m++)
79 if (NULL != top->mps[c_m].peer_id_op)
81 GNUNET_TESTBED_operation_done (p->peer_id_op);
85 if (GNUNET_SCHEDULER_NO_TASK != p->ats_task)
86 GNUNET_SCHEDULER_cancel (p->ats_task);
87 p->ats_task = GNUNET_SCHEDULER_NO_TASK;
89 for (c_op = 0; c_op < p->num_partners; c_op++)
91 if (NULL != p->partners[c_op].cth)
93 GNUNET_CORE_notify_transmit_ready_cancel (p->partners[c_op].cth);
94 p->partners[c_op].cth = NULL;
96 if (NULL != p->partners[c_op].tth)
98 GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->partners[c_op].tth);
99 p->partners[c_op].tth = NULL;
101 if ( (NULL != p->core_connect_ops) &&
102 (NULL != p->core_connect_ops[c_op].connect_op) )
104 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
105 _("Failed to connect peer 0 and %u\n"), c_op);
106 GNUNET_TESTBED_operation_done (
107 p->core_connect_ops[c_op].connect_op);
108 p->core_connect_ops[c_op].connect_op = NULL;
112 if (NULL != p->ats_perf_op)
114 GNUNET_TESTBED_operation_done (p->ats_perf_op);
115 p->ats_perf_op = NULL;
118 if (NULL != p->comm_op)
120 GNUNET_TESTBED_operation_done (p->comm_op);
123 GNUNET_free_non_null (p->core_connect_ops);
124 GNUNET_free(p->partners);
128 for (c_s = 0; c_s < top->num_slaves; c_s++)
131 if (NULL != p->peer_id_op)
133 GNUNET_TESTBED_operation_done (p->peer_id_op);
134 p->peer_id_op = NULL;
137 for (c_op = 0; c_op < p->num_partners; c_op++)
139 if (NULL != p->partners[c_op].cth)
141 GNUNET_CORE_notify_transmit_ready_cancel (p->partners[c_op].cth);
142 p->partners[c_op].cth = NULL;
144 if (NULL != p->partners[c_op].tth)
146 GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->partners[c_op].tth);
147 p->partners[c_op].tth = NULL;
150 if (NULL != p->ats_perf_op)
152 GNUNET_TESTBED_operation_done (p->ats_perf_op);
153 p->ats_perf_op = NULL;
155 if (NULL != p->comm_op)
157 GNUNET_TESTBED_operation_done (p->comm_op);
160 GNUNET_free(p->partners);
163 GNUNET_SCHEDULER_shutdown ();
168 static struct BenchmarkPartner *
169 find_partner (struct BenchmarkPeer *me, const struct GNUNET_PeerIdentity * peer)
172 GNUNET_assert (NULL != me);
173 GNUNET_assert (NULL != peer);
175 for (c_m = 0; c_m < me->num_partners; c_m++)
177 /* Find a partner with other as destination */
178 if (0 == memcmp (peer, &me->partners[c_m].dest->id,
179 sizeof(struct GNUNET_PeerIdentity)))
181 return &me->partners[c_m];
189 static struct BenchmarkPeer *
190 find_peer (const struct GNUNET_PeerIdentity * peer)
194 for (c_p = 0; c_p < top->num_masters; c_p++)
196 if (0 == memcmp (&top->mps[c_p].id, peer, sizeof(struct GNUNET_PeerIdentity)))
197 return &top->mps[c_p];
200 for (c_p = 0; c_p < top->num_slaves; c_p++)
202 if (0 == memcmp (&top->sps[c_p].id, peer, sizeof(struct GNUNET_PeerIdentity)))
203 return &top->sps[c_p];
210 * Method called whenever a given peer connects.
213 * @param peer peer identity this notification is about
216 comm_connect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
218 struct BenchmarkPeer *me = cls;
219 struct BenchmarkPeer *remote;
224 remote = find_peer (peer);
227 GNUNET_log(GNUNET_ERROR_TYPE_WARNING, "Unknown peer connected: `%s'\n", GNUNET_i2s (peer));
232 id = GNUNET_strdup (GNUNET_i2s (&me->id));
233 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s [%u] `%s' connected to %s [%u] %s\n",
234 (me->master == GNUNET_YES) ? "Master": "Slave", me->no, id,
235 (remote->master == GNUNET_YES) ? "Master": "Slave", remote->no,
238 me->core_connections++;
239 if ((GNUNET_YES == me->master) && (GNUNET_NO == remote->master)
240 && (GNUNET_NO == top->state.connected_CORE))
242 me->core_slave_connections++;
244 if (me->core_slave_connections == top->num_slaves)
246 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Master [%u] connected all slaves\n",
249 completed = GNUNET_YES;
250 for (c = 0; c < top->num_masters; c++)
252 if (top->mps[c].core_slave_connections != top->num_slaves)
253 completed = GNUNET_NO;
255 if (GNUNET_YES == completed)
257 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
258 "All master peers connected all slave peers\n", id,
260 top->state.connected_CORE = GNUNET_YES;
261 /* Notify about setup done */
262 if (NULL != top->done_cb)
263 top->done_cb (top->done_cb_cls, top->mps, top->sps);
270 comm_disconnect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
272 struct BenchmarkPeer *me = cls;
273 struct BenchmarkPartner *p;
276 if (NULL == (p = find_partner (me, peer)))
279 id = GNUNET_strdup (GNUNET_i2s (&me->id));
280 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s disconnected from %s \n", id,
282 GNUNET_assert(me->core_connections > 0);
283 me->core_connections--;
285 if ((GNUNET_YES == top->state.benchmarking)
286 && ((GNUNET_YES == me->master) || (GNUNET_YES == p->dest->master)))
288 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
289 "%s disconnected from %s while benchmarking \n", id, GNUNET_i2s (peer));
292 GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->tth);
297 GNUNET_CORE_notify_transmit_ready_cancel (p->cth);
306 core_connect_adapter (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
308 struct BenchmarkPeer *me = cls;
310 me->ch = GNUNET_CORE_connect (cfg, me, NULL, comm_connect_cb,
311 comm_disconnect_cb, NULL, GNUNET_NO, NULL, GNUNET_NO, top->handlers);
313 GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Failed to create core connection \n");
318 core_disconnect_adapter (void *cls, void *op_result)
320 struct BenchmarkPeer *me = cls;
322 GNUNET_CORE_disconnect (me->ch);
330 comm_handle_pong (void *cls, const struct GNUNET_PeerIdentity *other,
331 const struct GNUNET_MessageHeader *message)
333 struct BenchmarkPeer *me = cls;
334 struct BenchmarkPartner *p = NULL;
336 if (NULL == (p = find_partner (me, other)))
339 return GNUNET_SYSERR;
342 GNUNET_ATS_TEST_traffic_handle_pong (p);
348 comm_handle_ping (void *cls, const struct GNUNET_PeerIdentity *other,
349 const struct GNUNET_MessageHeader *message)
351 struct BenchmarkPeer *me = cls;
352 struct BenchmarkPartner *p = NULL;
354 if (NULL == (p = find_partner(me, other)))
357 return GNUNET_SYSERR;
359 GNUNET_ATS_TEST_traffic_handle_ping (p);
364 test_recv_cb (void *cls,
365 const struct GNUNET_PeerIdentity * peer,
366 const struct GNUNET_MessageHeader * message)
368 if (TEST_MESSAGE_SIZE != ntohs (message->size) ||
369 (TEST_MESSAGE_TYPE_PING != ntohs (message->type) &&
370 TEST_MESSAGE_TYPE_PONG != ntohs (message->type)))
374 if (TEST_MESSAGE_TYPE_PING == ntohs (message->type))
375 comm_handle_ping (cls, peer, message);
376 if (TEST_MESSAGE_TYPE_PONG == ntohs (message->type))
377 comm_handle_pong (cls, peer, message);
382 transport_connect_adapter (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
384 struct BenchmarkPeer *me = cls;
386 me->th = GNUNET_TRANSPORT_connect (cfg, &me->id, me, &test_recv_cb,
387 &comm_connect_cb, &comm_disconnect_cb);
389 GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Failed to create transport connection \n");
394 transport_disconnect_adapter (void *cls, void *op_result)
396 struct BenchmarkPeer *me = cls;
398 GNUNET_TRANSPORT_disconnect (me->th);
404 connect_completion_callback (void *cls, struct GNUNET_TESTBED_Operation *op,
407 struct TestbedConnectOperation *cop = cls;
412 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
413 _("Connected master [%u] with slave [%u]\n"), cop->master->no,
418 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
419 _("Failed to connect master peer [%u] with slave [%u]\n"),
420 cop->master->no, cop->slave->no);
422 if (GNUNET_SCHEDULER_NO_TASK != top->shutdown_task)
423 GNUNET_SCHEDULER_cancel (top->shutdown_task);
424 top->shutdown_task = GNUNET_SCHEDULER_add_now (do_shutdown, NULL );
426 GNUNET_TESTBED_operation_done (op);
428 for (c = 0; c < top->num_slaves; c++)
430 if (cop == &cop->master->core_connect_ops[c])
431 cop->master->core_connect_ops[c].connect_op = NULL;
433 if (ops == top->num_masters * top->num_slaves)
435 top->state.connected_PEERS = GNUNET_YES;
440 do_connect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
444 struct BenchmarkPeer *p;
446 if ((top->state.connected_ATS_service == GNUNET_NO) ||
447 (top->state.connected_COMM_service == GNUNET_NO))
450 GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Connecting peers on CORE level\n"));
452 for (c_m = 0; c_m < top->num_masters; c_m++)
455 p->core_connect_ops = GNUNET_malloc (top->num_slaves *
456 sizeof (struct TestbedConnectOperation));
458 for (c_s = 0; c_s < top->num_slaves; c_s++)
460 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
461 _("Connecting master [%u] with slave [%u]\n"), p->no, top->sps[c_s].no);
462 p->core_connect_ops[c_s].master = p;
463 p->core_connect_ops[c_s].slave = &top->sps[c_s];
464 p->core_connect_ops[c_s].connect_op = GNUNET_TESTBED_overlay_connect (
465 NULL, &connect_completion_callback, &p->core_connect_ops[c_s],
466 top->sps[c_s].peer, p->peer);
467 if (NULL == p->core_connect_ops[c_s].connect_op)
469 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
470 _("Could not connect master [%u] and slave [%u]\n"), p->no,
473 if (GNUNET_SCHEDULER_NO_TASK != top->shutdown_task)
474 GNUNET_SCHEDULER_cancel (top->shutdown_task);
475 top->shutdown_task = GNUNET_SCHEDULER_add_now (do_shutdown, NULL );
484 comm_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
485 void *ca_result, const char *emsg)
487 static int comm_done = 0;
488 if ((NULL != emsg) || (NULL == ca_result))
490 GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Initialization failed, shutdown\n"));
492 if (GNUNET_SCHEDULER_NO_TASK != top->shutdown_task)
493 GNUNET_SCHEDULER_cancel (top->shutdown_task);
494 top->shutdown_task = GNUNET_SCHEDULER_add_now (do_shutdown, NULL );
499 if (comm_done == top->num_slaves + top->num_masters)
501 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connected to all %s services\n",
502 (GNUNET_YES == top->test_core) ? "CORE" : "TRANSPORT");
503 top->state.connected_COMM_service = GNUNET_YES;
504 GNUNET_SCHEDULER_add_now (&do_connect_peers, NULL );
509 do_comm_connect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
513 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connecting to all %s services\n",
514 (GNUNET_YES == top->test_core) ? "CORE" : "TRANSPORT");
515 for (c_m = 0; c_m < top->num_masters; c_m++)
517 if (GNUNET_YES == top->test_core)
518 top->mps[c_m].comm_op = GNUNET_TESTBED_service_connect (NULL, top->mps[c_m].peer,
519 "core", &comm_connect_completion_cb, NULL, &core_connect_adapter,
520 &core_disconnect_adapter, &top->mps[c_m]);
523 top->mps[c_m].comm_op = GNUNET_TESTBED_service_connect (NULL, top->mps[c_m].peer,
524 "transport", &comm_connect_completion_cb, NULL, &transport_connect_adapter,
525 &transport_disconnect_adapter, &top->mps[c_m]);
529 for (c_s = 0; c_s < top->num_slaves; c_s++)
531 if (GNUNET_YES == top->test_core)
532 top->sps[c_s].comm_op = GNUNET_TESTBED_service_connect (NULL, top->sps[c_s].peer,
533 "core", &comm_connect_completion_cb, NULL, &core_connect_adapter,
534 &core_disconnect_adapter, &top->sps[c_s]);
537 top->sps[c_s].comm_op = GNUNET_TESTBED_service_connect (NULL, top->sps[c_s].peer,
538 "transport", &comm_connect_completion_cb, NULL, &transport_connect_adapter,
539 &transport_disconnect_adapter, &top->sps[c_s]);
546 ats_performance_info_cb (void *cls,
547 const struct GNUNET_HELLO_Address *address,
549 struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
550 struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
551 const struct GNUNET_ATS_Information *ats,
554 struct BenchmarkPeer *me = cls;
555 struct BenchmarkPartner *p;
560 p = find_partner (me, &address->peer);
563 /* This is not one of my partners
564 * Will happen since the peers will connect to each other due to gossiping
568 peer_id = GNUNET_strdup (GNUNET_i2s (&me->id));
571 if ((p->bandwidth_in != ntohl (bandwidth_in.value__)) ||
572 (p->bandwidth_out != ntohl (bandwidth_out.value__)))
574 p->bandwidth_in = ntohl (bandwidth_in.value__);
575 p->bandwidth_out = ntohl (bandwidth_out.value__);
577 for (c_a = 0; c_a < ats_count; c_a++)
579 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "%s [%u] received ATS information: %s %s %u\n",
580 (GNUNET_YES == p->me->master) ? "Master" : "Slave",
582 GNUNET_i2s (&p->dest->id),
583 GNUNET_ATS_print_property_type(ntohl(ats[c_a].type)),
584 ntohl(ats[c_a].value));
585 switch (ntohl (ats[c_a].type ))
587 case GNUNET_ATS_ARRAY_TERMINATOR:
589 case GNUNET_ATS_UTILIZATION_OUT:
590 if (p->ats_utilization_up != ntohl (ats[c_a].value))
592 p->ats_utilization_up = ntohl (ats[c_a].value);
595 case GNUNET_ATS_UTILIZATION_IN:
596 if (p->ats_utilization_down != ntohl (ats[c_a].value))
598 p->ats_utilization_down = ntohl (ats[c_a].value);
600 case GNUNET_ATS_NETWORK_TYPE:
601 if (p->ats_network_type != ntohl (ats[c_a].value))
603 p->ats_network_type = ntohl (ats[c_a].value);
605 case GNUNET_ATS_QUALITY_NET_DELAY:
606 if (p->ats_delay != ntohl (ats[c_a].value))
608 p->ats_delay = ntohl (ats[c_a].value);
610 case GNUNET_ATS_QUALITY_NET_DISTANCE:
611 if (p->ats_distance != ntohl (ats[c_a].value))
613 p->ats_distance = ntohl (ats[c_a].value);
616 case GNUNET_ATS_COST_WAN:
617 if (p->ats_cost_wan != ntohl (ats[c_a].value))
619 p->ats_cost_wan = ntohl (ats[c_a].value);
621 case GNUNET_ATS_COST_LAN:
622 if (p->ats_cost_lan != ntohl (ats[c_a].value))
624 p->ats_cost_lan = ntohl (ats[c_a].value);
626 case GNUNET_ATS_COST_WLAN:
627 if (p->ats_cost_wlan != ntohl (ats[c_a].value))
629 p->ats_cost_wlan = ntohl (ats[c_a].value);
635 if (GNUNET_YES == log)
636 top->ats_perf_cb (cls, address,
641 GNUNET_free(peer_id);
646 ats_perf_connect_adapter (void *cls,
647 const struct GNUNET_CONFIGURATION_Handle *cfg)
649 struct BenchmarkPeer *me = cls;
651 me->ats_perf_handle = GNUNET_ATS_performance_init (cfg,
652 &ats_performance_info_cb,
654 if (NULL == me->ats_perf_handle)
655 GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
656 "Failed to create ATS performance handle \n");
657 return me->ats_perf_handle;
661 ats_perf_disconnect_adapter (void *cls, void *op_result)
663 struct BenchmarkPeer *me = cls;
665 GNUNET_ATS_performance_done (me->ats_perf_handle);
666 me->ats_perf_handle = NULL;
670 ats_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
671 void *ca_result, const char *emsg)
673 static int op_done = 0;
675 if ((NULL != emsg) || (NULL == ca_result))
677 GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Initialization failed, shutdown\n"));
679 if (GNUNET_SCHEDULER_NO_TASK != top->shutdown_task)
680 GNUNET_SCHEDULER_cancel (top->shutdown_task);
681 top->shutdown_task = GNUNET_SCHEDULER_add_now (do_shutdown, NULL );
685 if (op_done == (top->num_masters + top->num_slaves))
687 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connected to all ATS services\n");
688 top->state.connected_ATS_service = GNUNET_YES;
689 GNUNET_SCHEDULER_add_now (&do_comm_connect, NULL );
695 do_connect_ats (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
700 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connecting to all ATS services\n");
701 for (c_m = 0; c_m < top->num_masters; c_m++)
703 top->mps[c_m].ats_perf_op = GNUNET_TESTBED_service_connect (NULL,
705 "ats", ats_connect_completion_cb, NULL,
706 &ats_perf_connect_adapter,
707 &ats_perf_disconnect_adapter, &top->mps[c_m]);
710 for (c_s = 0; c_s < top->num_slaves; c_s++)
712 top->sps[c_s].ats_perf_op = GNUNET_TESTBED_service_connect (NULL, top->sps[c_s].peer,
713 "ats", ats_connect_completion_cb, NULL, &ats_perf_connect_adapter,
714 &ats_perf_disconnect_adapter, &top->sps[c_s]);
721 peerinformation_cb (void *cb_cls, struct GNUNET_TESTBED_Operation *op,
722 const struct GNUNET_TESTBED_PeerInformation*pinfo, const char *emsg)
724 struct BenchmarkPeer *p = cb_cls;
727 GNUNET_assert(pinfo->pit == GNUNET_TESTBED_PIT_IDENTITY);
729 p->id = *pinfo->result.id;
730 GNUNET_log(GNUNET_ERROR_TYPE_INFO, "%s [%u] has peer id `%s'\n",
731 (p->master == GNUNET_YES) ? "Master" : "Slave", p->no,
732 GNUNET_i2s (&p->id));
734 GNUNET_TESTBED_operation_done (op);
735 p->peer_id_op = NULL;
738 if (done == top->num_slaves + top->num_masters)
740 GNUNET_log(GNUNET_ERROR_TYPE_INFO,
741 "Retrieved all peer ID, connect to ATS\n");
742 GNUNET_SCHEDULER_add_now (&do_connect_ats, NULL );
747 * Signature of a main function for a testcase.
750 * @param h testbed handle
751 * @param num_peers number of peers in 'peers'
752 * @param peers_ handle to peers run in the testbed
753 * @param links_succeeded the number of overlay link connection attempts that
755 * @param links_failed the number of overlay link connection attempts that
759 main_run (void *cls, struct GNUNET_TESTBED_RunHandle *h,
760 unsigned int num_peers,
761 struct GNUNET_TESTBED_Peer **peers_,
762 unsigned int links_succeeded,
763 unsigned int links_failed)
767 GNUNET_assert(NULL == cls);
768 GNUNET_assert(top->num_masters + top->num_slaves == num_peers);
769 GNUNET_assert(NULL != peers_);
771 top->shutdown_task = GNUNET_SCHEDULER_add_delayed (
772 GNUNET_TIME_UNIT_FOREVER_REL, &do_shutdown, top);
774 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Setting up %u masters and %u slaves\n",
775 top->num_masters, top->num_slaves);
777 /* Setup master peers */
778 for (c_m = 0; c_m < top->num_masters; c_m++)
780 GNUNET_assert(NULL != peers_[c_m]);
781 top->mps[c_m].peer = peers_[c_m];
782 top->mps[c_m].no = c_m;
783 top->mps[c_m].master = GNUNET_YES;
784 top->mps[c_m].pref_partner = &top->sps[c_m];
785 top->mps[c_m].pref_value = TEST_ATS_PREFERENCE_DEFAULT;
786 top->mps[c_m].partners =
787 GNUNET_malloc (top->num_slaves * sizeof (struct BenchmarkPartner));
788 top->mps[c_m].num_partners = top->num_slaves;
789 /* Initialize partners */
790 for (c_s = 0; c_s < top->num_slaves; c_s++)
792 top->mps[c_m].partners[c_s].me = &top->mps[c_m];
793 top->mps[c_m].partners[c_s].dest = &top->sps[c_s];
795 /* Get configuration */
796 top->mps[c_m].peer_id_op = GNUNET_TESTBED_peer_get_information (top->mps[c_m].peer,
797 GNUNET_TESTBED_PIT_IDENTITY, &peerinformation_cb, &top->mps[c_m]);
800 /* Setup slave peers */
801 for (c_s = 0; c_s < top->num_slaves; c_s++)
803 GNUNET_assert(NULL != peers_[c_s + top->num_masters]);
804 top->sps[c_s].peer = peers_[c_s + top->num_masters];
805 top->sps[c_s].no = c_s + top->num_masters;
806 top->sps[c_s].master = GNUNET_NO;
807 top->sps[c_s].partners =
808 GNUNET_malloc (top->num_masters * sizeof (struct BenchmarkPartner));
809 top->sps[c_s].num_partners = top->num_masters;
810 /* Initialize partners */
811 for (c_m = 0; c_m < top->num_masters; c_m++)
813 top->sps[c_s].partners[c_m].me = &top->sps[c_s];
814 top->sps[c_s].partners[c_m].dest = &top->mps[c_m];
816 /* Get configuration */
817 top->sps[c_s].peer_id_op = GNUNET_TESTBED_peer_get_information (top->sps[c_s].peer,
818 GNUNET_TESTBED_PIT_IDENTITY, &peerinformation_cb, &top->sps[c_s]);
823 * Controller event callback
826 * @param event the controller event
829 controller_event_cb (void *cls,
830 const struct GNUNET_TESTBED_EventInformation *event)
832 struct GNUNET_ATS_TEST_Topology *top = cls;
835 case GNUNET_TESTBED_ET_CONNECT:
837 case GNUNET_TESTBED_ET_OPERATION_FINISHED:
841 GNUNET_SCHEDULER_cancel (top->shutdown_task);
842 top->shutdown_task = GNUNET_SCHEDULER_add_now (&do_shutdown, NULL );
846 struct BenchmarkPeer *
847 GNUNET_ATS_TEST_get_peer (int src)
849 if (src > top->num_masters)
851 return &top->mps[src];
854 struct BenchmarkPartner *
855 GNUNET_ATS_TEST_get_partner (int src, int dest)
857 if (src > top->num_masters)
859 if (dest > top->num_slaves)
861 return &top->mps[src].partners[dest];
866 * Create a topology for ats testing
868 * @param name test name
869 * @param cfg_file configuration file to use for the peers
870 * @param num_slaves number of slaves
871 * @param num_masters number of masters
872 * @param test_core connect to CORE service (GNUNET_YES) or transport (GNUNET_NO)
873 * @param done_cb function to call when topology is setup
874 * @param done_cb_cls cls for callback
875 * @param transport_recv_cb callback to call when data are received
876 * @param log_request_cb callback to call when logging is required
879 GNUNET_ATS_TEST_create_topology (char *name, char *cfg_file,
880 unsigned int num_slaves,
881 unsigned int num_masters,
883 GNUNET_ATS_TEST_TopologySetupDoneCallback done_cb,
885 GNUNET_TRANSPORT_ReceiveCallback transport_recv_cb,
886 GNUNET_ATS_AddressInformationCallback log_request_cb)
888 static struct GNUNET_CORE_MessageHandler handlers[] = {
889 {&comm_handle_ping, TEST_MESSAGE_TYPE_PING, 0 },
890 {&comm_handle_pong, TEST_MESSAGE_TYPE_PONG, 0 },
893 top = GNUNET_new (struct GNUNET_ATS_TEST_Topology);
894 top->num_masters = num_masters;
895 top->num_slaves = num_slaves;
896 top->handlers = handlers;
897 top->done_cb = done_cb;
898 top->done_cb_cls = done_cb_cls;
899 top->test_core = test_core;
900 top->transport_recv_cb = transport_recv_cb;
901 top->ats_perf_cb = log_request_cb;
903 top->mps = GNUNET_malloc (num_masters * sizeof (struct BenchmarkPeer));
904 top->sps = GNUNET_malloc (num_slaves * sizeof (struct BenchmarkPeer));
909 event_mask |= (1LL << GNUNET_TESTBED_ET_CONNECT);
910 event_mask |= (1LL << GNUNET_TESTBED_ET_OPERATION_FINISHED);
911 (void) GNUNET_TESTBED_test_run (name, cfg_file,
912 num_slaves + num_masters,
914 &controller_event_cb, NULL,
923 GNUNET_ATS_TEST_shutdown_topology (void)
927 GNUNET_SCHEDULER_shutdown();
934 /* end of file ats-testing.c */