yapf format.
[oweals/gnunet.git] / src / testbed / test_testbed_api_topology_clique.c
1 /*
2    This file is part of GNUnet
3    Copyright (C) 2008--2013 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 /**
22  * @file src/testbed/test_testbed_api_topology.c
23  * @brief testing cases for testing high level testbed api helper functions
24  * @author Sree Harsha Totakura <sreeharsha@totakura.in>
25  */
26
27 #include "platform.h"
28 #include "gnunet_util_lib.h"
29 #include "gnunet_testbed_service.h"
30
31 /**
32  * Number of peers we want to start
33  */
34 #define NUM_PEERS 10
35
36 /**
37  * Array of peers
38  */
39 static struct GNUNET_TESTBED_Peer **peers;
40
41 /**
42  * Operation handle
43  */
44 static struct GNUNET_TESTBED_Operation *op;
45
46 /**
47  * Shutdown task
48  */
49 static struct GNUNET_SCHEDULER_Task *shutdown_task;
50
51 /**
52  * Testing result
53  */
54 static int result;
55
56 /**
57  * Counter for counting overlay connections
58  */
59 static unsigned int overlay_connects;
60
61
62 /**
63  * Shutdown nicely
64  *
65  * @param cls NULL
66  */
67 static void
68 do_shutdown (void *cls)
69 {
70   shutdown_task = NULL;
71   if (NULL != op)
72   {
73     GNUNET_TESTBED_operation_done (op);
74     op = NULL;
75   }
76   GNUNET_SCHEDULER_shutdown ();
77 }
78
79 /**
80  * Controller event callback
81  *
82  * @param cls NULL
83  * @param event the controller event
84  */
85 static void
86 controller_event_cb (void *cls,
87                      const struct GNUNET_TESTBED_EventInformation *event)
88 {
89   switch (event->type)
90   {
91   case GNUNET_TESTBED_ET_CONNECT:
92     overlay_connects++;
93     if ((NUM_PEERS * (NUM_PEERS - 1)) == overlay_connects)
94     {
95       result = GNUNET_OK;
96       GNUNET_SCHEDULER_cancel (shutdown_task);
97       shutdown_task = GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
98     }
99     break;
100
101   case GNUNET_TESTBED_ET_OPERATION_FINISHED:
102     GNUNET_assert (NULL != event->details.operation_finished.emsg);
103     break;
104
105   default:
106     GNUNET_break (0);
107     result = GNUNET_SYSERR;
108     GNUNET_SCHEDULER_cancel (shutdown_task);
109     shutdown_task = GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
110   }
111 }
112
113
114 /**
115  * Signature of a main function for a testcase.
116  *
117  * @param cls closure
118  * @param h the run handle
119  * @param num_peers number of peers in 'peers'
120  * @param peers_ handle to peers run in the testbed
121  * @param links_succeeded the number of overlay link connection attempts that
122  *          succeeded
123  * @param links_failed the number of overlay link connection attempts that
124  *          failed
125  */
126 static void
127 test_master (void *cls,
128              struct GNUNET_TESTBED_RunHandle *h,
129              unsigned int num_peers,
130              struct GNUNET_TESTBED_Peer **peers_,
131              unsigned int links_succeeded,
132              unsigned int links_failed)
133 {
134   unsigned int peer;
135
136   GNUNET_assert (NULL == cls);
137   if (NULL == peers_)
138   {
139     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failing test due to timeout\n");
140     return;
141   }
142   GNUNET_assert (NUM_PEERS == num_peers);
143   for (peer = 0; peer < num_peers; peer++)
144     GNUNET_assert (NULL != peers_[peer]);
145   peers = peers_;
146   overlay_connects = 0;
147   op = GNUNET_TESTBED_overlay_configure_topology (NULL, NUM_PEERS, peers, NULL,
148                                                   NULL,
149                                                   NULL,
150                                                   GNUNET_TESTBED_TOPOLOGY_CLIQUE,
151                                                   /* GNUNET_TESTBED_TOPOLOGY_ERDOS_RENYI, */
152                                                   /* NUM_PEERS, */
153                                                   GNUNET_TESTBED_TOPOLOGY_OPTION_END);
154   GNUNET_assert (NULL != op);
155   shutdown_task =
156     GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
157                                     (GNUNET_TIME_UNIT_SECONDS, 300),
158                                   do_shutdown, NULL);
159 }
160
161
162 /**
163  * Main function
164  */
165 int
166 main (int argc, char **argv)
167 {
168   uint64_t event_mask;
169
170   result = GNUNET_SYSERR;
171   event_mask = 0;
172   event_mask |= (1LL << GNUNET_TESTBED_ET_CONNECT);
173   event_mask |= (1LL << GNUNET_TESTBED_ET_OPERATION_FINISHED);
174   (void) GNUNET_TESTBED_test_run ("test_testbed_api_test",
175                                   "test_testbed_api.conf", NUM_PEERS,
176                                   event_mask, &controller_event_cb, NULL,
177                                   &test_master, NULL);
178   if (GNUNET_OK != result)
179     return 1;
180   return 0;
181 }
182
183 /* end of test_testbed_api_topology.c */