2 This file is part of GNUnet.
3 Copyright (C) 2017 GNUnet e.V.
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., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
22 * @file cadet/test_cadet_local.c
23 * @brief test cadet local: test of cadet channels with just one peer
24 * @author Bartlomiej Polot
28 #include "gnunet_util_lib.h"
29 #include "gnunet_dht_service.h"
30 #include "gnunet_testing_lib.h"
31 #include "gnunet_cadet_service.h"
33 #define TEST_MESSAGE_TYPE 1
34 #define TEST_PORT_ID 1
37 * Test message structure.
39 struct GNUNET_CADET_TestMsg
42 * Type: #TEST_MESSAGE_TYPE
44 * Size: sizeof(struct GNUNET_CADET_TestMsg)
46 struct GNUNET_MessageHeader header;
54 struct GNUNET_TESTING_Peer *me;
56 static struct GNUNET_CADET_Handle *cadet_peer_1;
58 static struct GNUNET_CADET_Handle *cadet_peer_2;
60 static struct GNUNET_CADET_Channel *ch;
62 static int result = GNUNET_OK;
64 static int got_data = GNUNET_NO;
66 static struct GNUNET_SCHEDULER_Task *abort_task;
68 static struct GNUNET_SCHEDULER_Task *connect_task;
72 * Connect to other client and send data
74 * @param cls Closue (unused).
77 do_connect (void *cls);
84 do_shutdown (void *cls)
86 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
88 if (NULL != abort_task)
90 GNUNET_SCHEDULER_cancel (abort_task);
95 GNUNET_CADET_channel_destroy (ch);
98 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
99 "Disconnect client 1\n");
100 if (NULL != cadet_peer_1)
102 GNUNET_CADET_disconnect (cadet_peer_1);
105 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
106 "Disconnect client 2\n");
107 if (NULL != cadet_peer_2)
109 GNUNET_CADET_disconnect (cadet_peer_2);
112 if (NULL != connect_task)
114 GNUNET_SCHEDULER_cancel (connect_task);
121 * Something went wrong and timed out. Kill everything and set error flag
126 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "ABORT from line %ld\n", (long) cls);
127 result = GNUNET_SYSERR;
129 GNUNET_SCHEDULER_shutdown ();
133 * Method called whenever a peer connects to a port in MQ-based CADET.
135 * @param cls Closure from #GNUNET_CADET_open_porT.
136 * @param channel New handle to the channel.
137 * @param source Peer that started this channel.
138 * @return Closure for the incoming @a channel. It's given to:
139 * - The #GNUNET_CADET_DisconnectEventHandler (given to
140 * #GNUNET_CADET_open_porT) when the channel dies.
141 * - Each the #GNUNET_MQ_MessageCallback handlers for each message
142 * received on the @a channel.
145 connected (void *cls,
146 struct GNUNET_CADET_Channel *channel,
147 const struct GNUNET_PeerIdentity *source)
149 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
150 "connected %s, cls: %p\n",
151 GNUNET_i2s(source), cls);
156 * Function called whenever an MQ-channel is destroyed, even if the destruction
157 * was requested by #GNUNET_CADET_channel_destroy.
158 * It must NOT call #GNUNET_CADET_channel_destroy on the channel.
160 * It should clean up any associated state, including cancelling any pending
161 * transmission on this channel.
163 * @param cls Channel closure.
164 * @param channel Connection to the other end (henceforth invalid).
167 disconnected (void *cls,
168 const struct GNUNET_CADET_Channel *channel)
170 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
171 "disconnected channel %p, cls: %p\n",
181 * @param h The cadet handle
182 * @param msg A message with the details of the new incoming channel
185 handle_data_received (void *cls,
186 const struct GNUNET_CADET_TestMsg *msg)
190 payload = GNUNET_ntohll (msg->payload);
191 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
192 "Data callback payload %llu with cls: %p! Shutting down.\n",
193 (unsigned long long) payload,
195 GNUNET_assert (42 == payload);
196 got_data = GNUNET_YES;
197 GNUNET_SCHEDULER_shutdown ();
202 * Signature of the main function of a task.
204 * @param cls Closure (unused).
207 message_sent (void *cls)
209 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "message sent\n");
214 * Connect to other client and send data
216 * @param cls Closure (unused).
219 do_connect (void *cls)
221 struct GNUNET_PeerIdentity id;
222 struct GNUNET_MQ_Handle *mq;
223 struct GNUNET_MQ_Envelope *env;
224 struct GNUNET_CADET_TestMsg *msg;
226 struct GNUNET_MQ_MessageHandler handlers[] = {
227 GNUNET_MQ_hd_fixed_size (data_received,
229 struct GNUNET_CADET_TestMsg,
231 GNUNET_MQ_handler_end ()
235 GNUNET_TESTING_peer_get_identity (me, &id);
236 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
237 "creating channel\n");
238 ch = GNUNET_CADET_channel_creatE (cadet_peer_1, /* cadet handle */
239 NULL, /* channel cls */
240 &id, /* destination */
241 GC_u2h (TEST_MESSAGE_TYPE), /* port */
242 GNUNET_CADET_OPTION_DEFAULT, /* opt */
243 NULL, /* window change */
244 &disconnected, /* disconnect handler */
245 handlers /* traffic handlers */
247 env = GNUNET_MQ_msg (msg, TEST_MESSAGE_TYPE);
248 msg->payload = GNUNET_htonll (42);
249 mq = GNUNET_CADET_get_mq (ch);
250 GNUNET_MQ_notify_sent (env, &message_sent, NULL);
251 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
252 "sending message\n");
253 GNUNET_MQ_send (mq, env);
258 * Initialize framework and start test
260 * @param cls Closure (unused).
261 * @param cfg Configuration handle.
262 * @param peer Testing peer handle.
266 const struct GNUNET_CONFIGURATION_Handle *cfg,
267 struct GNUNET_TESTING_Peer *peer)
269 struct GNUNET_MQ_MessageHandler handlers[] = {
270 GNUNET_MQ_hd_fixed_size (data_received,
272 struct GNUNET_CADET_TestMsg,
274 GNUNET_MQ_handler_end ()
276 struct GNUNET_TIME_Relative delay;
279 GNUNET_SCHEDULER_add_shutdown (&do_shutdown,
281 delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 15);
282 abort_task = GNUNET_SCHEDULER_add_delayed (delay,
284 (void *) (long) __LINE__);
285 cadet_peer_1 = GNUNET_CADET_connecT (cfg);
286 cadet_peer_2 = GNUNET_CADET_connecT (cfg);
288 if ( (NULL == cadet_peer_1) ||
289 (NULL == cadet_peer_2) )
291 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
292 "Couldn't connect to cadet\n");
293 result = GNUNET_SYSERR;
294 GNUNET_SCHEDULER_shutdown ();
297 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CADET 1: %p\n", cadet_peer_1);
298 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CADET 2: %p\n", cadet_peer_2);
299 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "handlers 2: %p\n", handlers);
300 GNUNET_CADET_open_porT (cadet_peer_2, /* cadet handle */
301 GC_u2h (TEST_PORT_ID), /* port id */
302 &connected, /* connect handler */
303 (void *) 2L, /* handle for #connected */
304 NULL, /* window size handler */
305 &disconnected, /* disconnect handler */
306 handlers); /* traffic handlers */
307 delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2);
308 if (NULL == connect_task)
309 connect_task = GNUNET_SCHEDULER_add_delayed (delay,
319 main (int argc, char *argv[])
321 if (0 != GNUNET_TESTING_peer_run ("test-cadet-local",
325 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "run failed\n");
328 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Final result: %d\n", result);
329 return (result == GNUNET_OK) ? 0 : 1;
332 /* end of test_cadet_local_1.c */