2 This file is part of GNUnet.
3 (C) 2011 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.
22 * @file src/transport/gnunet-transport.c
23 * @brief Tool to help configure, measure and control the transport subsystem.
24 * @author Christian Grothoff
25 * @author Nathan Evans
27 * This utility can be used to test if a transport mechanism for
28 * GNUnet is properly configured.
32 #include "gnunet_util_lib.h"
33 #include "gnunet_resolver_service.h"
34 #include "gnunet_protocols.h"
35 #include "gnunet_transport_service.h"
36 #include "gnunet_nat_lib.h"
39 * How long do we wait for the NAT test to report success?
40 * Should match NAT_SERVER_TIMEOUT in 'nat_test.c'.
42 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 20)
43 #define RESOLUTION_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
44 #define OP_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
47 * Benchmarking block size in KB
52 * Which peer should we connect to?
57 * Handle to transport service.
59 static struct GNUNET_TRANSPORT_Handle *handle;
62 * Configuration handle
64 static struct GNUNET_CONFIGURATION_Handle *cfg;
69 struct GNUNET_TRANSPORT_TryConnectHandle * tc_handle;
74 static int benchmark_send;
79 static int benchmark_receive;
84 static int benchmark_receive;
89 static int iterate_connections;
94 static int iterate_validation;
99 static int iterate_all;
104 static int test_configuration;
109 static int monitor_connects;
114 static int monitor_connections;
119 static int monitor_validation;
124 static int try_connect;
129 static int try_disconnect;
137 * Global return value (0 success).
142 * Current number of connections in monitor mode
144 static int monitor_connect_counter;
147 * Number of bytes of traffic we received so far.
149 static unsigned long long traffic_received;
152 * Number of bytes of traffic we sent so far.
154 static unsigned long long traffic_sent;
157 * Starting time of transmitting/receiving data.
159 static struct GNUNET_TIME_Absolute start_time;
162 * Handle for current transmission request.
164 static struct GNUNET_TRANSPORT_TransmitHandle *th;
167 * Map storing information about monitored peers
169 static struct GNUNET_CONTAINER_MultiPeerMap *monitored_peers;
174 static struct GNUNET_TRANSPORT_PeerMonitoringContext *pic;
176 static struct GNUNET_TRANSPORT_ValidationMonitoringContext *vic;
179 * Identity of the peer we transmit to / connect to.
180 * (equivalent to 'cpid' string).
182 static struct GNUNET_PeerIdentity pid;
185 * Task scheduled for cleanup / termination of the process.
187 static GNUNET_SCHEDULER_TaskIdentifier end;
190 * Task for operation timeout
192 static GNUNET_SCHEDULER_TaskIdentifier op_timeout;
195 * Selected level of verbosity.
197 static int verbosity;
200 * Resolver process handle.
202 struct GNUNET_OS_Process *resolver;
205 * Number of address resolutions pending
207 static unsigned int address_resolutions;
210 * Address resolutions pending in progress
212 static unsigned int address_resolution_in_progress;
215 * DLL for NAT Test Contexts: head
217 struct TestContext *head;
220 * DLL for NAT Test Contexts: tail
222 struct TestContext *tail;
225 * Context for a plugin test.
232 struct TestContext *prev;
237 struct TestContext *next;
240 * Handle to the active NAT test.
242 struct GNUNET_NAT_Test *tst;
245 * Task identifier for the timeout.
247 GNUNET_SCHEDULER_TaskIdentifier tsk;
250 * Name of plugin under test.
257 unsigned long long bnd_port;
262 unsigned long long adv_port;
268 /* NAT returned success */
269 NAT_TEST_SUCCESS = GNUNET_OK,
271 /* NAT returned failure */
272 NAT_TEST_FAIL = GNUNET_NO,
274 /* NAT returned failure while running test */
275 NAT_TEST_INTERNAL_FAIL = GNUNET_SYSERR,
277 /* We could not start the test */
278 NAT_TEST_FAILED_TO_START = 2,
280 /* We had a timeout while running the test */
281 NAT_TEST_TIMEOUT = 3,
284 static struct ValidationResolutionContext *vc_head;
285 static struct ValidationResolutionContext *vc_tail;
287 struct ValidationResolutionContext
289 struct ValidationResolutionContext *next;
290 struct ValidationResolutionContext *prev;
292 struct GNUNET_PeerIdentity id;
293 struct GNUNET_HELLO_Address *addrcp;
294 struct GNUNET_TIME_Absolute last_validation;
295 struct GNUNET_TIME_Absolute valid_until;
296 struct GNUNET_TIME_Absolute next_validation;
297 enum GNUNET_TRANSPORT_ValidationState state;
299 struct GNUNET_TRANSPORT_AddressToStringContext *asc;
307 enum GNUNET_TRANSPORT_PeerState state;
308 struct GNUNET_TIME_Absolute state_timeout;
309 struct GNUNET_HELLO_Address *address;
314 destroy_it (void *cls,
315 const struct GNUNET_PeerIdentity *key,
318 struct MonitoredPeer *m = value;
320 GNUNET_assert (GNUNET_OK == GNUNET_CONTAINER_multipeermap_remove (monitored_peers,
322 GNUNET_free_non_null (m->address);
329 * Task run in monitor mode when the user presses CTRL-C to abort.
330 * Stops monitoring activity.
332 * @param cls the 'struct GNUNET_TRANSPORT_PeerIterateContext *'
333 * @param tc scheduler context
336 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
338 struct GNUNET_TIME_Relative duration;
339 struct ValidationResolutionContext *cur;
340 struct ValidationResolutionContext *next;
341 end = GNUNET_SCHEDULER_NO_TASK;
342 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
344 GNUNET_SCHEDULER_cancel (op_timeout);
345 op_timeout = GNUNET_SCHEDULER_NO_TASK;
347 if (NULL != tc_handle)
349 GNUNET_TRANSPORT_try_connect_cancel (tc_handle);
354 GNUNET_TRANSPORT_monitor_peers_cancel (pic);
359 GNUNET_TRANSPORT_monitor_validation_entries_cancel (vic);
364 for (cur = next; NULL != cur; cur = next)
368 GNUNET_TRANSPORT_address_to_string_cancel (cur->asc);
369 GNUNET_CONTAINER_DLL_remove (vc_head, vc_tail, cur);
370 GNUNET_free (cur->transport);
371 GNUNET_HELLO_address_free (cur->addrcp);
377 GNUNET_TRANSPORT_notify_transmit_ready_cancel (th);
382 GNUNET_TRANSPORT_disconnect (handle);
387 duration = GNUNET_TIME_absolute_get_duration (start_time);
388 FPRINTF (stdout, _("Transmitted %llu bytes/s (%llu bytes in %s)\n"),
389 1000LL * 1000LL * traffic_sent / (1 + duration.rel_value_us),
391 GNUNET_STRINGS_relative_time_to_string (duration, GNUNET_YES));
393 if (benchmark_receive)
395 duration = GNUNET_TIME_absolute_get_duration (start_time);
396 FPRINTF (stdout, _("Received %llu bytes/s (%llu bytes in %s)\n"),
397 1000LL * 1000LL * traffic_received / (1 + duration.rel_value_us),
399 GNUNET_STRINGS_relative_time_to_string (duration, GNUNET_YES));
402 if (NULL != monitored_peers)
404 GNUNET_CONTAINER_multipeermap_iterate (monitored_peers, &destroy_it, NULL);
405 GNUNET_CONTAINER_multipeermap_destroy (monitored_peers);
406 monitored_peers = NULL;
410 static struct PeerResolutionContext *rc_head;
411 static struct PeerResolutionContext *rc_tail;
413 struct PeerResolutionContext
415 struct PeerResolutionContext *next;
416 struct PeerResolutionContext *prev;
417 struct GNUNET_PeerIdentity id;
418 struct GNUNET_HELLO_Address *addrcp;
419 struct GNUNET_TRANSPORT_AddressToStringContext *asc;
420 enum GNUNET_TRANSPORT_PeerState state;
421 struct GNUNET_TIME_Absolute state_timeout;
429 operation_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
431 struct PeerResolutionContext *cur;
432 struct PeerResolutionContext *next;
433 op_timeout = GNUNET_SCHEDULER_NO_TASK;
434 if ((try_connect) || (benchmark_send) || (benchmark_receive))
436 FPRINTF (stdout, _("Failed to connect to `%s'\n"), GNUNET_i2s_full (&pid));
437 if (GNUNET_SCHEDULER_NO_TASK != end)
438 GNUNET_SCHEDULER_cancel (end);
439 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
443 if (iterate_connections)
446 while (NULL != (cur = next))
449 FPRINTF (stdout, _("Failed to resolve address for peer `%s'\n"),
450 GNUNET_i2s (&cur->addrcp->peer));
452 GNUNET_CONTAINER_DLL_remove(rc_head, rc_tail, cur);
453 GNUNET_TRANSPORT_address_to_string_cancel (cur->asc);
454 GNUNET_free(cur->transport);
455 GNUNET_free(cur->addrcp);
459 FPRINTF (stdout, "%s", _("Failed to list connections, timeout occured\n") );
460 if (GNUNET_SCHEDULER_NO_TASK != end)
461 GNUNET_SCHEDULER_cancel (end);
462 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
473 * Display the result of the test.
475 * @param tc test context
476 * @param result #GNUNET_YES on success
479 display_test_result (struct TestContext *tc, enum TestResult result)
483 FPRINTF (stderr, _("Configuration for plugin `%s' did not work!\n"),
486 case NAT_TEST_SUCCESS:
487 FPRINTF (stderr, _("Configuration for plugin `%s' did work!\n"),
490 case NAT_TEST_INTERNAL_FAIL:
491 FPRINTF (stderr, _("Internal NAT error while running test for plugin `%s'\n"),
494 case NAT_TEST_FAILED_TO_START:
495 FPRINTF (stderr, _("Failed to start NAT test for plugin `%s'\n"),
498 case NAT_TEST_TIMEOUT:
499 FPRINTF (stderr, _("Timeout while waiting for result of NAT test for plugin `%s'\n"),
506 if (GNUNET_YES != result)
508 FPRINTF (stderr, "Configuration for plugin `%s' did not work!\n", tc->name);
512 FPRINTF (stderr, "Configuration for plugin `%s' is working!\n", tc->name);
514 if (GNUNET_SCHEDULER_NO_TASK != tc->tsk)
516 GNUNET_SCHEDULER_cancel (tc->tsk);
517 tc->tsk = GNUNET_SCHEDULER_NO_TASK;
521 GNUNET_NAT_test_stop (tc->tst);
525 GNUNET_CONTAINER_DLL_remove (head, tail, tc);
526 GNUNET_free (tc->name);
529 if ((NULL == head) && (NULL != resolver))
531 GNUNET_break(0 == GNUNET_OS_process_kill (resolver, GNUNET_TERM_SIG));
532 GNUNET_OS_process_destroy (resolver);
540 * Function called by NAT on success.
541 * Clean up and update GUI (with success).
543 * @param cls test context
544 * @param success currently always #GNUNET_OK
545 * @param emsg error message, NULL on success
548 result_callback (void *cls, int success, const char *emsg)
550 struct TestContext *tc = cls;
551 display_test_result (tc, success);
555 * Function called if NAT failed to confirm success.
556 * Clean up and update GUI (with failure).
558 * @param cls test context
559 * @param tc scheduler callback
562 fail_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
564 struct TestContext *tstc = cls;
566 tstc->tsk = GNUNET_SCHEDULER_NO_TASK;
567 display_test_result (tstc, NAT_TEST_TIMEOUT);
572 resolve_validation_address (const struct GNUNET_PeerIdentity *id,
573 const struct GNUNET_HELLO_Address *address,
575 struct GNUNET_TIME_Absolute last_validation,
576 struct GNUNET_TIME_Absolute valid_until,
577 struct GNUNET_TIME_Absolute next_validation,
578 enum GNUNET_TRANSPORT_ValidationState state);
582 process_validation_string (void *cls, const char *address)
584 struct ValidationResolutionContext *vc = cls;
589 if (address != NULL )
591 if (GNUNET_TIME_UNIT_ZERO_ABS.abs_value_us == vc->valid_until.abs_value_us)
592 s_valid = GNUNET_strdup("never");
594 s_valid = GNUNET_strdup(GNUNET_STRINGS_absolute_time_to_string (vc->valid_until));
596 if (GNUNET_TIME_UNIT_ZERO_ABS.abs_value_us == vc->last_validation.abs_value_us)
597 s_last = GNUNET_strdup("never");
599 s_last = GNUNET_strdup(GNUNET_STRINGS_absolute_time_to_string (vc->last_validation));
601 if (GNUNET_TIME_UNIT_ZERO_ABS.abs_value_us == vc->next_validation.abs_value_us)
602 s_next = GNUNET_strdup("never");
604 s_next = GNUNET_strdup(GNUNET_STRINGS_absolute_time_to_string (vc->next_validation));
607 _("Peer `%s' %s %s\n\t%s%s\n\t%s%s\n\t%s%s\n"),
608 GNUNET_i2s (&vc->id), address,
609 (monitor_validation) ? GNUNET_TRANSPORT_vs2s (vc->state) : "",
610 "Valid until : ", s_valid,
611 "Last validation: ",s_last,
612 "Next validation: ", s_next);
613 GNUNET_free (s_valid);
614 GNUNET_free (s_last);
615 GNUNET_free (s_next);
616 vc->printed = GNUNET_YES;
621 GNUNET_assert(address_resolutions > 0);
622 address_resolutions--;
623 if (GNUNET_NO == vc->printed)
625 if (numeric == GNUNET_NO)
627 /* Failed to resolve address, try numeric lookup */
628 resolve_validation_address (&vc->id, vc->addrcp, GNUNET_NO,
629 vc->last_validation, vc->valid_until, vc->next_validation,
634 FPRINTF (stdout, _("Peer `%s' %s `%s' \n"),
635 GNUNET_i2s (&vc->id), "<unable to resolve address>",
636 GNUNET_TRANSPORT_vs2s (vc->state));
639 GNUNET_free (vc->transport);
640 GNUNET_free (vc->addrcp);
641 GNUNET_CONTAINER_DLL_remove(vc_head, vc_tail, vc);
643 if ((0 == address_resolutions) && (iterate_validation))
645 if (GNUNET_SCHEDULER_NO_TASK != end)
647 GNUNET_SCHEDULER_cancel (end);
648 end = GNUNET_SCHEDULER_NO_TASK;
650 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
652 GNUNET_SCHEDULER_cancel (op_timeout);
653 op_timeout = GNUNET_SCHEDULER_NO_TASK;
656 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
664 resolve_validation_address (const struct GNUNET_PeerIdentity *id,
665 const struct GNUNET_HELLO_Address *address, int numeric,
666 struct GNUNET_TIME_Absolute last_validation,
667 struct GNUNET_TIME_Absolute valid_until,
668 struct GNUNET_TIME_Absolute next_validation,
669 enum GNUNET_TRANSPORT_ValidationState state)
671 struct ValidationResolutionContext *vc;
673 vc = GNUNET_new (struct ValidationResolutionContext);
674 GNUNET_assert(NULL != vc);
675 GNUNET_CONTAINER_DLL_insert(vc_head, vc_tail, vc);
676 address_resolutions++;
679 vc->transport = GNUNET_strdup(address->transport_name);
680 vc->addrcp = GNUNET_HELLO_address_copy (address);
681 vc->printed = GNUNET_NO;
683 vc->last_validation = last_validation;
684 vc->valid_until = valid_until;
685 vc->next_validation = next_validation;
687 /* Resolve address to string */
688 vc->asc = GNUNET_TRANSPORT_address_to_string (cfg, address, numeric,
689 RESOLUTION_TIMEOUT, &process_validation_string, vc);
694 process_validation_cb (void *cls,
695 const struct GNUNET_PeerIdentity *peer,
696 const struct GNUNET_HELLO_Address *address,
697 struct GNUNET_TIME_Absolute last_validation,
698 struct GNUNET_TIME_Absolute valid_until,
699 struct GNUNET_TIME_Absolute next_validation,
700 enum GNUNET_TRANSPORT_ValidationState state)
702 if ((NULL == peer) && (NULL == address))
706 if (GNUNET_SCHEDULER_NO_TASK != end)
707 GNUNET_SCHEDULER_cancel (end);
708 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
711 if ((NULL == peer) || (NULL == address))
713 /* invalid response */
715 if (GNUNET_SCHEDULER_NO_TASK != end)
716 GNUNET_SCHEDULER_cancel (end);
717 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
720 resolve_validation_address (peer, address,
721 numeric, last_validation,
722 valid_until, next_validation, state);
729 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
730 "Running test for plugin `%s' using bind port %u and advertised port %u \n",
731 head->name, (uint16_t) head->bnd_port, (uint16_t) head->adv_port);
733 head->tst = GNUNET_NAT_test_start (cfg,
734 (0 == strcasecmp (head->name, "udp")) ? GNUNET_NO : GNUNET_YES,
735 (uint16_t) head->bnd_port,
736 (uint16_t) head->adv_port,
737 &result_callback, head);
738 if (NULL == head->tst)
740 display_test_result (head, NAT_TEST_FAILED_TO_START);
743 head->tsk = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &fail_timeout, head);
748 * Test our plugin's configuration (NAT traversal, etc.).
750 * @param cfg configuration to test
753 do_test_configuration (const struct GNUNET_CONFIGURATION_Handle *cfg)
757 unsigned long long bnd_port;
758 unsigned long long adv_port;
759 struct TestContext *tc;
763 != GNUNET_CONFIGURATION_get_value_string (cfg, "transport", "plugins",
766 FPRINTF (stderr, "%s", _
767 ("No transport plugins configured, peer will never communicate\n") );
772 for (tok = strtok (plugins, " "); tok != NULL ; tok = strtok (NULL, " "))
774 char section[12 + strlen (tok)];
775 GNUNET_snprintf (section, sizeof(section), "transport-%s", tok);
777 != GNUNET_CONFIGURATION_get_value_number (cfg, section, "PORT",
781 _("No port configured for plugin `%s', cannot test it\n"), tok);
784 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (cfg, section,
785 "ADVERTISED_PORT", &adv_port))
788 tc = GNUNET_new (struct TestContext);
789 tc->name = GNUNET_strdup (tok);
790 tc->adv_port = adv_port;
791 tc->bnd_port = bnd_port;
792 GNUNET_CONTAINER_DLL_insert_tail (head, tail, tc);
794 GNUNET_free(plugins);
796 if ((NULL != head) && (NULL == resolver))
798 binary = GNUNET_OS_get_libexec_binary_path ("gnunet-service-resolver");
799 resolver = GNUNET_OS_start_process (GNUNET_YES,
800 GNUNET_OS_INHERIT_STD_OUT_AND_ERR,
803 "gnunet-service-resolver", NULL );
804 if (NULL == resolver)
806 FPRINTF (stderr, _("Failed to start resolver!\n"));
811 GNUNET_RESOLVER_connect (cfg);
817 * Function called to notify a client about the socket
818 * begin ready to queue more data. @a buf will be
819 * NULL and @a size zero if the socket was closed for
820 * writing in the meantime.
823 * @param size number of bytes available in @a buf
824 * @param buf where the callee should write the message
825 * @return number of bytes written to @a buf
828 transmit_data (void *cls, size_t size, void *buf)
830 struct GNUNET_MessageHeader *m = buf;
832 if ((NULL == buf) || (0 == size))
838 GNUNET_assert(size >= sizeof(struct GNUNET_MessageHeader));
839 GNUNET_assert(size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
840 m->size = ntohs (size);
841 m->type = ntohs (GNUNET_MESSAGE_TYPE_DUMMY);
842 memset (&m[1], 52, size - sizeof(struct GNUNET_MessageHeader));
843 traffic_sent += size;
844 th = GNUNET_TRANSPORT_notify_transmit_ready (handle, &pid,
846 GNUNET_TIME_UNIT_FOREVER_REL,
847 &transmit_data, NULL );
849 FPRINTF (stdout, _("Transmitting %u bytes to %s\n"), (unsigned int) size,
855 * Function called to notify transport users that another
856 * peer connected to us.
859 * @param peer the peer that connected
862 notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
864 if (0 != memcmp (&pid, peer, sizeof(struct GNUNET_PeerIdentity)))
869 /* all done, terminate instantly */
870 FPRINTF (stdout, _("Successfully connected to `%s'\n"),
871 GNUNET_i2s_full (peer));
874 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
876 GNUNET_SCHEDULER_cancel (op_timeout);
877 op_timeout = GNUNET_SCHEDULER_NO_TASK;
880 if (GNUNET_SCHEDULER_NO_TASK != end)
881 GNUNET_SCHEDULER_cancel (end);
882 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
887 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
889 GNUNET_SCHEDULER_cancel (op_timeout);
890 op_timeout = GNUNET_SCHEDULER_NO_TASK;
894 _("Successfully connected to `%s', starting to send benchmark data in %u Kb blocks\n"),
895 GNUNET_i2s (&pid), BLOCKSIZE);
896 start_time = GNUNET_TIME_absolute_get ();
898 th = GNUNET_TRANSPORT_notify_transmit_ready (handle, peer,
900 GNUNET_TIME_UNIT_FOREVER_REL,
910 * Function called to notify transport users that another
911 * peer disconnected from us.
914 * @param peer the peer that disconnected
917 notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
919 if (0 != memcmp (&pid, peer, sizeof(struct GNUNET_PeerIdentity)))
924 /* all done, terminate instantly */
925 FPRINTF (stdout, _("Successfully disconnected from `%s'\n"),
926 GNUNET_i2s_full (peer));
929 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
931 GNUNET_SCHEDULER_cancel (op_timeout);
932 op_timeout = GNUNET_SCHEDULER_NO_TASK;
935 if (GNUNET_SCHEDULER_NO_TASK != end)
936 GNUNET_SCHEDULER_cancel (end);
937 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
943 GNUNET_TRANSPORT_notify_transmit_ready_cancel (th);
948 FPRINTF (stdout, _("Disconnected from peer `%s' while benchmarking\n"),
950 if (GNUNET_SCHEDULER_NO_TASK != end)
951 GNUNET_SCHEDULER_cancel (end);
957 * Function called to notify transport users that another
958 * peer connected to us.
961 * @param peer the peer that connected
964 monitor_notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
966 monitor_connect_counter++;
967 struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
968 const char *now_str = GNUNET_STRINGS_absolute_time_to_string (now);
970 FPRINTF (stdout, _("%24s: %-17s %4s (%u connections in total)\n"), now_str,
971 _("Connected to"), GNUNET_i2s (peer), monitor_connect_counter);
975 * Function called to notify transport users that another
976 * peer disconnected from us.
979 * @param peer the peer that disconnected
982 monitor_notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
984 struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
985 const char *now_str = GNUNET_STRINGS_absolute_time_to_string (now);
987 GNUNET_assert(monitor_connect_counter > 0);
988 monitor_connect_counter--;
990 FPRINTF (stdout, _("%24s: %-17s %4s (%u connections in total)\n"), now_str,
991 _("Disconnected from"), GNUNET_i2s (peer), monitor_connect_counter);
995 * Function called by the transport for each received message.
998 * @param peer (claimed) identity of the other peer
999 * @param message the message
1002 notify_receive (void *cls, const struct GNUNET_PeerIdentity *peer,
1003 const struct GNUNET_MessageHeader *message)
1005 if (benchmark_receive)
1007 if (GNUNET_MESSAGE_TYPE_DUMMY != ntohs (message->type))
1010 FPRINTF (stdout, _("Received %u bytes from %s\n"),
1011 (unsigned int) ntohs (message->size), GNUNET_i2s (peer));
1013 if (traffic_received == 0)
1014 start_time = GNUNET_TIME_absolute_get ();
1015 traffic_received += ntohs (message->size);
1022 resolve_peer_address (const struct GNUNET_PeerIdentity *id,
1023 const struct GNUNET_HELLO_Address *address,
1025 enum GNUNET_TRANSPORT_PeerState state,
1026 struct GNUNET_TIME_Absolute state_timeout);
1030 print_info (const struct GNUNET_PeerIdentity *id,
1031 const char *transport,
1033 enum GNUNET_TRANSPORT_PeerState state,
1034 struct GNUNET_TIME_Absolute state_timeout)
1037 if ( ((GNUNET_YES == iterate_connections) && (GNUNET_YES == iterate_all)) ||
1038 (GNUNET_YES == monitor_connections) )
1040 FPRINTF (stdout, _("Peer `%s': %s %s in state `%s' until %s\n"),
1042 (NULL == transport) ? "<none>" : transport,
1043 (NULL == transport) ? "<none>" : addr,
1044 GNUNET_TRANSPORT_ps2s (state),
1045 GNUNET_STRINGS_absolute_time_to_string (state_timeout));
1047 else if ( (GNUNET_YES == iterate_connections) &&
1048 (GNUNET_TRANSPORT_is_connected(state)) )
1050 /* Only connected peers, skip state */
1051 FPRINTF (stdout, _("Peer `%s': %s %s\n"), GNUNET_i2s (id), transport, addr);
1057 process_peer_string (void *cls, const char *address)
1059 struct PeerResolutionContext *rc = cls;
1061 if (address != NULL )
1063 print_info (&rc->id, rc->transport, address, rc->state, rc->state_timeout);
1064 rc->printed = GNUNET_YES;
1069 GNUNET_assert(address_resolutions > 0);
1070 address_resolutions--;
1071 if (GNUNET_NO == rc->printed)
1073 if (numeric == GNUNET_NO)
1075 /* Failed to resolve address, try numeric lookup */
1076 resolve_peer_address (&rc->id, rc->addrcp, GNUNET_YES,
1077 rc->state, rc->state_timeout);
1081 print_info (&rc->id, rc->transport, "<unable to resolve address>",
1082 rc->state, rc->state_timeout);
1085 GNUNET_free (rc->transport);
1086 GNUNET_free (rc->addrcp);
1087 GNUNET_CONTAINER_DLL_remove(rc_head, rc_tail, rc);
1089 if ((0 == address_resolutions) && (iterate_connections))
1091 if (GNUNET_SCHEDULER_NO_TASK != end)
1093 GNUNET_SCHEDULER_cancel (end);
1094 end = GNUNET_SCHEDULER_NO_TASK;
1096 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1098 GNUNET_SCHEDULER_cancel (op_timeout);
1099 op_timeout = GNUNET_SCHEDULER_NO_TASK;
1102 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1109 resolve_peer_address (const struct GNUNET_PeerIdentity *id,
1110 const struct GNUNET_HELLO_Address *address,
1112 enum GNUNET_TRANSPORT_PeerState state,
1113 struct GNUNET_TIME_Absolute state_timeout)
1115 struct PeerResolutionContext *rc;
1117 rc = GNUNET_new (struct PeerResolutionContext);
1118 GNUNET_assert(NULL != rc);
1119 GNUNET_CONTAINER_DLL_insert(rc_head, rc_tail, rc);
1120 address_resolutions++;
1123 rc->transport = GNUNET_strdup(address->transport_name);
1124 rc->addrcp = GNUNET_HELLO_address_copy (address);
1125 rc->printed = GNUNET_NO;
1127 rc->state_timeout = state_timeout;
1128 /* Resolve address to string */
1129 rc->asc = GNUNET_TRANSPORT_address_to_string (cfg, address, numeric,
1130 RESOLUTION_TIMEOUT, &process_peer_string, rc);
1135 * Function called with information about a peers during a one shot iteration
1137 * @param cls closure
1138 * @param peer identity of the peer, NULL for final callback when operation done
1139 * @param address binary address used to communicate with this peer,
1140 * NULL on disconnect or when done
1141 * @param state current state this peer is in
1142 * @param state_timeout time out for the current state
1146 process_peer_iteration_cb (void *cls,
1147 const struct GNUNET_PeerIdentity *peer,
1148 const struct GNUNET_HELLO_Address *address,
1149 enum GNUNET_TRANSPORT_PeerState state,
1150 struct GNUNET_TIME_Absolute state_timeout)
1154 if (monitor_connections)
1157 _("Monitor disconnected from transport service. Reconnecting.\n"));
1161 address_resolution_in_progress = GNUNET_NO;
1163 if (GNUNET_SCHEDULER_NO_TASK != end)
1164 GNUNET_SCHEDULER_cancel (end);
1165 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1169 if ((GNUNET_NO == iterate_all) && (GNUNET_NO == GNUNET_TRANSPORT_is_connected(state)) )
1170 return; /* Display only connected peers */
1172 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1173 GNUNET_SCHEDULER_cancel (op_timeout);
1174 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1177 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1178 "Received address for peer `%s': %s\n",
1179 GNUNET_i2s (peer), address->transport_name);
1181 if (NULL != address)
1182 resolve_peer_address (peer, address, numeric, state, state_timeout);
1184 print_info (peer, NULL, NULL, state, state_timeout);
1189 * Function called with information about a peers
1191 * @param cls closure
1192 * @param peer identity of the peer, NULL for final callback when operation done
1193 * @param address binary address used to communicate with this peer,
1194 * NULL on disconnect or when done
1195 * @param state current state this peer is in
1196 * @param state_timeout time out for the current state
1200 process_peer_monitoring_cb (void *cls, const struct GNUNET_PeerIdentity *peer,
1201 const struct GNUNET_HELLO_Address *address,
1202 enum GNUNET_TRANSPORT_PeerState state,
1203 struct GNUNET_TIME_Absolute state_timeout)
1205 struct MonitoredPeer *m;
1210 address_resolution_in_progress = GNUNET_NO;
1212 if (GNUNET_SCHEDULER_NO_TASK != end)
1213 GNUNET_SCHEDULER_cancel (end);
1214 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1218 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1219 GNUNET_SCHEDULER_cancel (op_timeout);
1220 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1223 if (NULL == (m = GNUNET_CONTAINER_multipeermap_get (monitored_peers, peer)))
1225 m = GNUNET_new (struct MonitoredPeer);
1226 GNUNET_CONTAINER_multipeermap_put (monitored_peers, peer,
1227 m, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1231 if ( (m->state == state) &&
1232 (m->state_timeout.abs_value_us == state_timeout.abs_value_us) &&
1233 ((NULL == address) && (NULL == m->address)) )
1235 return; /* No real change */
1237 if ( (m->state == state) && ((NULL != address) && (NULL != m->address)) &&
1238 (0 == GNUNET_HELLO_address_cmp(m->address, address)) )
1239 return; /* No real change */
1242 if (NULL != m->address)
1244 GNUNET_free (m->address);
1247 if (NULL != address)
1248 m->address = GNUNET_HELLO_address_copy (address);
1250 m->state_timeout = state_timeout;
1252 if (NULL != address)
1253 resolve_peer_address (peer, m->address, numeric, m->state, m->state_timeout);
1255 print_info (peer, NULL, NULL, m->state, m->state_timeout);
1259 try_connect_cb (void *cls, const int result)
1261 static int retries = 0;
1262 if (GNUNET_OK == result)
1269 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1273 FPRINTF (stderr, "%s",
1274 _("Failed to send connect request to transport service\n") );
1275 if (GNUNET_SCHEDULER_NO_TASK != end)
1276 GNUNET_SCHEDULER_cancel (end);
1278 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1284 try_disconnect_cb (void *cls, const int result)
1286 static int retries = 0;
1287 if (GNUNET_OK == result)
1294 tc_handle = GNUNET_TRANSPORT_try_disconnect (handle, &pid, try_disconnect_cb,
1298 FPRINTF (stderr, "%s",
1299 _("Failed to send connect request to transport service\n") );
1300 if (GNUNET_SCHEDULER_NO_TASK != end)
1301 GNUNET_SCHEDULER_cancel (end);
1303 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1309 * Function called with the result of the check if the 'transport'
1310 * service is running.
1312 * @param cls closure with our configuration
1313 * @param result #GNUNET_YES if transport is running
1316 testservice_task (void *cls, int result)
1321 if (GNUNET_YES != result)
1323 FPRINTF (stderr, _("Service `%s' is not running\n"), "transport");
1329 != GNUNET_CRYPTO_eddsa_public_key_from_string (cpid, strlen (cpid),
1332 FPRINTF (stderr, _("Failed to parse peer identity `%s'\n"), cpid);
1336 counter = benchmark_send + benchmark_receive + iterate_connections
1337 + monitor_connections + monitor_connects + try_connect + try_disconnect +
1338 + iterate_validation + monitor_validation;
1343 _("Multiple operations given. Please choose only one operation: %s, %s, %s, %s, %s, %s\n"),
1344 "connect", "benchmark send", "benchmark receive", "information",
1345 "monitor", "events");
1351 _("No operation given. Please choose one operation: %s, %s, %s, %s, %s, %s\n"),
1352 "connect", "benchmark send", "benchmark receive", "information",
1353 "monitor", "events");
1357 if (try_connect) /* -C: Connect to peer */
1361 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1366 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1367 ¬ify_connect, ¬ify_disconnect);
1370 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1374 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1376 if (NULL == tc_handle)
1378 FPRINTF (stderr, "%s",
1379 _("Failed to send request to transport service\n") );
1383 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1387 else if (try_disconnect) /* -D: Disconnect from peer */
1391 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1396 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1397 ¬ify_connect, ¬ify_disconnect);
1400 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1404 tc_handle = GNUNET_TRANSPORT_try_disconnect (handle, &pid, try_disconnect_cb,
1406 if (NULL == tc_handle)
1408 FPRINTF (stderr, "%s",
1409 _("Failed to send request to transport service\n") );
1413 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1417 else if (benchmark_send) /* -s: Benchmark sending */
1421 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1426 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1427 ¬ify_connect, ¬ify_disconnect);
1430 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1434 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1436 if (NULL == tc_handle)
1438 FPRINTF (stderr, "%s",
1439 _("Failed to send request to transport service\n") );
1443 start_time = GNUNET_TIME_absolute_get ();
1444 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1447 else if (benchmark_receive) /* -b: Benchmark receiving */
1449 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive, NULL,
1453 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1458 FPRINTF (stdout, "%s", _("Starting to receive benchmark data\n") );
1459 start_time = GNUNET_TIME_absolute_get ();
1462 else if (iterate_connections) /* -i: List information about peers once */
1464 address_resolution_in_progress = GNUNET_YES;
1465 pic = GNUNET_TRANSPORT_monitor_peers (cfg, (NULL == cpid) ? NULL : &pid,
1466 GNUNET_YES, TIMEOUT, &process_peer_iteration_cb, (void *) cfg);
1467 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1470 else if (monitor_connections) /* -m: List information about peers continuously */
1472 monitored_peers = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1473 address_resolution_in_progress = GNUNET_YES;
1474 pic = GNUNET_TRANSPORT_monitor_peers (cfg, (NULL == cpid) ? NULL : &pid,
1475 GNUNET_NO, TIMEOUT, &process_peer_monitoring_cb, (void *) cfg);
1477 else if (iterate_validation) /* -d: Print information about validations */
1479 vic = GNUNET_TRANSPORT_monitor_validation_entries (cfg,
1480 (NULL == cpid) ? NULL : &pid,
1481 GNUNET_YES, TIMEOUT, &process_validation_cb, (void *) cfg);
1483 else if (monitor_validation) /* -f: Print information about validations continuously */
1485 vic = GNUNET_TRANSPORT_monitor_validation_entries (cfg,
1486 (NULL == cpid) ? NULL : &pid,
1487 GNUNET_NO, TIMEOUT, &process_validation_cb, (void *) cfg);
1489 else if (monitor_connects) /* -e : Monitor (dis)connect events continuously */
1491 monitor_connect_counter = 0;
1492 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, NULL,
1493 &monitor_notify_connect, &monitor_notify_disconnect);
1496 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1508 end = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1509 &shutdown_task, NULL );
1514 * Main function that will be run by the scheduler.
1516 * @param cls closure
1517 * @param args remaining command-line arguments
1518 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1519 * @param mycfg configuration
1522 run (void *cls, char * const *args, const char *cfgfile,
1523 const struct GNUNET_CONFIGURATION_Handle *mycfg)
1525 cfg = (struct GNUNET_CONFIGURATION_Handle *) mycfg;
1526 if (test_configuration)
1528 do_test_configuration (cfg);
1531 GNUNET_CLIENT_service_test ("transport", cfg, GNUNET_TIME_UNIT_SECONDS,
1532 &testservice_task, (void *) cfg);
1536 main (int argc, char * const *argv)
1539 static const struct GNUNET_GETOPT_CommandLineOption options[] =
1542 gettext_noop ("print information for all peers (instead of only connected peers )"),
1543 0, &GNUNET_GETOPT_set_one, &iterate_all },
1544 { 'b', "benchmark", NULL,
1545 gettext_noop ("measure how fast we are receiving data from all peers (until CTRL-C)"),
1546 0, &GNUNET_GETOPT_set_one, &benchmark_receive },
1548 NULL, gettext_noop ("connect to a peer"), 0,
1549 &GNUNET_GETOPT_set_one, &try_connect },
1550 { 'D', "disconnect",
1551 NULL, gettext_noop ("disconnect to a peer"), 0,
1552 &GNUNET_GETOPT_set_one, &try_disconnect },
1553 { 'd', "validation", NULL,
1554 gettext_noop ("print information for all pending validations "),
1555 0, &GNUNET_GETOPT_set_one, &iterate_validation },
1556 { 'f', "monitorvalidation", NULL,
1557 gettext_noop ("print information for all pending validations continously"),
1558 0, &GNUNET_GETOPT_set_one, &monitor_validation },
1559 { 'i', "information", NULL,
1560 gettext_noop ("provide information about all current connections (once)"),
1561 0, &GNUNET_GETOPT_set_one, &iterate_connections },
1562 { 'm', "monitor", NULL,
1563 gettext_noop ("provide information about all current connections (continuously)"),
1564 0, &GNUNET_GETOPT_set_one, &monitor_connections },
1565 { 'e', "events", NULL,
1566 gettext_noop ("provide information about all connects and disconnect events (continuously)"),
1567 0, &GNUNET_GETOPT_set_one, &monitor_connects }, { 'n', "numeric",
1568 NULL, gettext_noop ("do not resolve hostnames"), 0,
1569 &GNUNET_GETOPT_set_one, &numeric }, { 'p', "peer", "PEER",
1570 gettext_noop ("peer identity"), 1, &GNUNET_GETOPT_set_string,
1571 &cpid }, { 's', "send", NULL, gettext_noop
1572 ("send data for benchmarking to the other peer (until CTRL-C)"), 0,
1573 &GNUNET_GETOPT_set_one, &benchmark_send },
1574 { 't', "test", NULL,
1575 gettext_noop ("test transport configuration (involves external server)"),
1576 0, &GNUNET_GETOPT_set_one, &test_configuration },
1577 GNUNET_GETOPT_OPTION_VERBOSE (&verbosity),
1578 GNUNET_GETOPT_OPTION_END };
1580 if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
1583 res = GNUNET_PROGRAM_run (argc, argv, "gnunet-transport", gettext_noop
1584 ("Direct access to transport service."), options, &run, NULL );
1585 GNUNET_free((void * ) argv);
1586 if (GNUNET_OK == res)
1591 /* end of gnunet-transport.c */