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))
704 if (monitor_validation)
708 _("Monitor disconnected from transport service. Reconnecting.\n"));
714 if (GNUNET_SCHEDULER_NO_TASK != end)
715 GNUNET_SCHEDULER_cancel (end);
716 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
719 if ((NULL == peer) || (NULL == address))
721 /* invalid response */
723 if (GNUNET_SCHEDULER_NO_TASK != end)
724 GNUNET_SCHEDULER_cancel (end);
725 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
728 resolve_validation_address (peer, address,
729 numeric, last_validation,
730 valid_until, next_validation, state);
737 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
738 "Running test for plugin `%s' using bind port %u and advertised port %u \n",
739 head->name, (uint16_t) head->bnd_port, (uint16_t) head->adv_port);
741 head->tst = GNUNET_NAT_test_start (cfg,
742 (0 == strcasecmp (head->name, "udp")) ? GNUNET_NO : GNUNET_YES,
743 (uint16_t) head->bnd_port,
744 (uint16_t) head->adv_port,
745 &result_callback, head);
746 if (NULL == head->tst)
748 display_test_result (head, NAT_TEST_FAILED_TO_START);
751 head->tsk = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &fail_timeout, head);
756 * Test our plugin's configuration (NAT traversal, etc.).
758 * @param cfg configuration to test
761 do_test_configuration (const struct GNUNET_CONFIGURATION_Handle *cfg)
765 unsigned long long bnd_port;
766 unsigned long long adv_port;
767 struct TestContext *tc;
771 != GNUNET_CONFIGURATION_get_value_string (cfg, "transport", "plugins",
774 FPRINTF (stderr, "%s", _
775 ("No transport plugins configured, peer will never communicate\n") );
780 for (tok = strtok (plugins, " "); tok != NULL ; tok = strtok (NULL, " "))
782 char section[12 + strlen (tok)];
783 GNUNET_snprintf (section, sizeof(section), "transport-%s", tok);
785 != GNUNET_CONFIGURATION_get_value_number (cfg, section, "PORT",
789 _("No port configured for plugin `%s', cannot test it\n"), tok);
792 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (cfg, section,
793 "ADVERTISED_PORT", &adv_port))
796 tc = GNUNET_new (struct TestContext);
797 tc->name = GNUNET_strdup (tok);
798 tc->adv_port = adv_port;
799 tc->bnd_port = bnd_port;
800 GNUNET_CONTAINER_DLL_insert_tail (head, tail, tc);
802 GNUNET_free(plugins);
804 if ((NULL != head) && (NULL == resolver))
806 binary = GNUNET_OS_get_libexec_binary_path ("gnunet-service-resolver");
807 resolver = GNUNET_OS_start_process (GNUNET_YES,
808 GNUNET_OS_INHERIT_STD_OUT_AND_ERR,
811 "gnunet-service-resolver", NULL );
812 if (NULL == resolver)
814 FPRINTF (stderr, _("Failed to start resolver!\n"));
819 GNUNET_RESOLVER_connect (cfg);
825 * Function called to notify a client about the socket
826 * begin ready to queue more data. @a buf will be
827 * NULL and @a size zero if the socket was closed for
828 * writing in the meantime.
831 * @param size number of bytes available in @a buf
832 * @param buf where the callee should write the message
833 * @return number of bytes written to @a buf
836 transmit_data (void *cls, size_t size, void *buf)
838 struct GNUNET_MessageHeader *m = buf;
840 if ((NULL == buf) || (0 == size))
846 GNUNET_assert(size >= sizeof(struct GNUNET_MessageHeader));
847 GNUNET_assert(size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
848 m->size = ntohs (size);
849 m->type = ntohs (GNUNET_MESSAGE_TYPE_DUMMY);
850 memset (&m[1], 52, size - sizeof(struct GNUNET_MessageHeader));
851 traffic_sent += size;
852 th = GNUNET_TRANSPORT_notify_transmit_ready (handle, &pid,
854 GNUNET_TIME_UNIT_FOREVER_REL,
855 &transmit_data, NULL );
857 FPRINTF (stdout, _("Transmitting %u bytes to %s\n"), (unsigned int) size,
864 * Function called to notify transport users that another
865 * peer connected to us.
868 * @param peer the peer that connected
871 notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
873 if (0 != memcmp (&pid, peer, sizeof(struct GNUNET_PeerIdentity)))
878 /* all done, terminate instantly */
879 FPRINTF (stdout, _("Successfully connected to `%s'\n"),
880 GNUNET_i2s_full (peer));
883 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
885 GNUNET_SCHEDULER_cancel (op_timeout);
886 op_timeout = GNUNET_SCHEDULER_NO_TASK;
889 if (GNUNET_SCHEDULER_NO_TASK != end)
890 GNUNET_SCHEDULER_cancel (end);
891 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
896 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
898 GNUNET_SCHEDULER_cancel (op_timeout);
899 op_timeout = GNUNET_SCHEDULER_NO_TASK;
903 _("Successfully connected to `%s', starting to send benchmark data in %u Kb blocks\n"),
904 GNUNET_i2s (&pid), BLOCKSIZE);
905 start_time = GNUNET_TIME_absolute_get ();
907 th = GNUNET_TRANSPORT_notify_transmit_ready (handle, peer,
909 GNUNET_TIME_UNIT_FOREVER_REL,
919 * Function called to notify transport users that another
920 * peer disconnected from us.
923 * @param peer the peer that disconnected
926 notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
928 if (0 != memcmp (&pid, peer, sizeof(struct GNUNET_PeerIdentity)))
933 /* all done, terminate instantly */
934 FPRINTF (stdout, _("Successfully disconnected from `%s'\n"),
935 GNUNET_i2s_full (peer));
938 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
940 GNUNET_SCHEDULER_cancel (op_timeout);
941 op_timeout = GNUNET_SCHEDULER_NO_TASK;
944 if (GNUNET_SCHEDULER_NO_TASK != end)
945 GNUNET_SCHEDULER_cancel (end);
946 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
952 GNUNET_TRANSPORT_notify_transmit_ready_cancel (th);
957 FPRINTF (stdout, _("Disconnected from peer `%s' while benchmarking\n"),
959 if (GNUNET_SCHEDULER_NO_TASK != end)
960 GNUNET_SCHEDULER_cancel (end);
966 * Function called to notify transport users that another
967 * peer connected to us.
970 * @param peer the peer that connected
973 monitor_notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
975 monitor_connect_counter++;
976 struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
977 const char *now_str = GNUNET_STRINGS_absolute_time_to_string (now);
979 FPRINTF (stdout, _("%24s: %-17s %4s (%u connections in total)\n"), now_str,
980 _("Connected to"), GNUNET_i2s (peer), monitor_connect_counter);
985 * Function called to notify transport users that another
986 * peer disconnected from us.
989 * @param peer the peer that disconnected
992 monitor_notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
994 struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
995 const char *now_str = GNUNET_STRINGS_absolute_time_to_string (now);
997 GNUNET_assert(monitor_connect_counter > 0);
998 monitor_connect_counter--;
1000 FPRINTF (stdout, _("%24s: %-17s %4s (%u connections in total)\n"), now_str,
1001 _("Disconnected from"), GNUNET_i2s (peer), monitor_connect_counter);
1006 * Function called by the transport for each received message.
1008 * @param cls closure
1009 * @param peer (claimed) identity of the other peer
1010 * @param message the message
1013 notify_receive (void *cls,
1014 const struct GNUNET_PeerIdentity *peer,
1015 const struct GNUNET_MessageHeader *message)
1017 if (benchmark_receive)
1019 if (GNUNET_MESSAGE_TYPE_DUMMY != ntohs (message->type))
1022 FPRINTF (stdout, _("Received %u bytes from %s\n"),
1023 (unsigned int) ntohs (message->size), GNUNET_i2s (peer));
1025 if (traffic_received == 0)
1026 start_time = GNUNET_TIME_absolute_get ();
1027 traffic_received += ntohs (message->size);
1034 resolve_peer_address (const struct GNUNET_PeerIdentity *id,
1035 const struct GNUNET_HELLO_Address *address,
1037 enum GNUNET_TRANSPORT_PeerState state,
1038 struct GNUNET_TIME_Absolute state_timeout);
1042 print_info (const struct GNUNET_PeerIdentity *id,
1043 const char *transport,
1045 enum GNUNET_TRANSPORT_PeerState state,
1046 struct GNUNET_TIME_Absolute state_timeout)
1049 if ( ((GNUNET_YES == iterate_connections) && (GNUNET_YES == iterate_all)) ||
1050 (GNUNET_YES == monitor_connections) )
1052 FPRINTF (stdout, _("Peer `%s': %s %s in state `%s' until %s\n"),
1054 (NULL == transport) ? "<none>" : transport,
1055 (NULL == transport) ? "<none>" : addr,
1056 GNUNET_TRANSPORT_ps2s (state),
1057 GNUNET_STRINGS_absolute_time_to_string (state_timeout));
1059 else if ( (GNUNET_YES == iterate_connections) &&
1060 (GNUNET_TRANSPORT_is_connected(state)) )
1062 /* Only connected peers, skip state */
1063 FPRINTF (stdout, _("Peer `%s': %s %s\n"), GNUNET_i2s (id), transport, addr);
1069 process_peer_string (void *cls, const char *address)
1071 struct PeerResolutionContext *rc = cls;
1073 if (address != NULL )
1075 print_info (&rc->id, rc->transport, address, rc->state, rc->state_timeout);
1076 rc->printed = GNUNET_YES;
1081 GNUNET_assert(address_resolutions > 0);
1082 address_resolutions--;
1083 if (GNUNET_NO == rc->printed)
1085 if (numeric == GNUNET_NO)
1087 /* Failed to resolve address, try numeric lookup */
1088 resolve_peer_address (&rc->id, rc->addrcp, GNUNET_YES,
1089 rc->state, rc->state_timeout);
1093 print_info (&rc->id, rc->transport, "<unable to resolve address>",
1094 rc->state, rc->state_timeout);
1097 GNUNET_free (rc->transport);
1098 GNUNET_free (rc->addrcp);
1099 GNUNET_CONTAINER_DLL_remove(rc_head, rc_tail, rc);
1101 if ((0 == address_resolutions) && (iterate_connections))
1103 if (GNUNET_SCHEDULER_NO_TASK != end)
1105 GNUNET_SCHEDULER_cancel (end);
1106 end = GNUNET_SCHEDULER_NO_TASK;
1108 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1110 GNUNET_SCHEDULER_cancel (op_timeout);
1111 op_timeout = GNUNET_SCHEDULER_NO_TASK;
1114 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1121 resolve_peer_address (const struct GNUNET_PeerIdentity *id,
1122 const struct GNUNET_HELLO_Address *address,
1124 enum GNUNET_TRANSPORT_PeerState state,
1125 struct GNUNET_TIME_Absolute state_timeout)
1127 struct PeerResolutionContext *rc;
1129 rc = GNUNET_new (struct PeerResolutionContext);
1130 GNUNET_assert(NULL != rc);
1131 GNUNET_CONTAINER_DLL_insert(rc_head, rc_tail, rc);
1132 address_resolutions++;
1135 rc->transport = GNUNET_strdup(address->transport_name);
1136 rc->addrcp = GNUNET_HELLO_address_copy (address);
1137 rc->printed = GNUNET_NO;
1139 rc->state_timeout = state_timeout;
1140 /* Resolve address to string */
1141 rc->asc = GNUNET_TRANSPORT_address_to_string (cfg, address, numeric,
1142 RESOLUTION_TIMEOUT, &process_peer_string, rc);
1147 * Function called with information about a peers during a one shot iteration
1149 * @param cls closure
1150 * @param peer identity of the peer, NULL for final callback when operation done
1151 * @param address binary address used to communicate with this peer,
1152 * NULL on disconnect or when done
1153 * @param state current state this peer is in
1154 * @param state_timeout time out for the current state
1157 process_peer_iteration_cb (void *cls,
1158 const struct GNUNET_PeerIdentity *peer,
1159 const struct GNUNET_HELLO_Address *address,
1160 enum GNUNET_TRANSPORT_PeerState state,
1161 struct GNUNET_TIME_Absolute state_timeout)
1166 address_resolution_in_progress = GNUNET_NO;
1168 if (GNUNET_SCHEDULER_NO_TASK != end)
1169 GNUNET_SCHEDULER_cancel (end);
1170 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1174 if ( (GNUNET_NO == iterate_all) &&
1175 (GNUNET_NO == GNUNET_TRANSPORT_is_connected(state)) )
1176 return; /* Display only connected peers */
1178 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1179 GNUNET_SCHEDULER_cancel (op_timeout);
1180 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1183 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1184 "Received address for peer `%s': %s\n",
1185 GNUNET_i2s (peer), address->transport_name);
1187 if (NULL != address)
1188 resolve_peer_address (peer, address, numeric, state, state_timeout);
1190 print_info (peer, NULL, NULL, state, state_timeout);
1195 * Function called with information about a peers
1197 * @param cls closure
1198 * @param peer identity of the peer, NULL for final callback when operation done
1199 * @param address binary address used to communicate with this peer,
1200 * NULL on disconnect or when done
1201 * @param state current state this peer is in
1202 * @param state_timeout time out for the current state
1206 process_peer_monitoring_cb (void *cls,
1207 const struct GNUNET_PeerIdentity *peer,
1208 const struct GNUNET_HELLO_Address *address,
1209 enum GNUNET_TRANSPORT_PeerState state,
1210 struct GNUNET_TIME_Absolute state_timeout)
1212 struct MonitoredPeer *m;
1218 _("Monitor disconnected from transport service. Reconnecting.\n"));
1222 if (GNUNET_SCHEDULER_NO_TASK != op_timeout)
1223 GNUNET_SCHEDULER_cancel (op_timeout);
1224 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1227 if (NULL == (m = GNUNET_CONTAINER_multipeermap_get (monitored_peers, peer)))
1229 m = GNUNET_new (struct MonitoredPeer);
1230 GNUNET_CONTAINER_multipeermap_put (monitored_peers, peer,
1231 m, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1235 if ( (m->state == state) &&
1236 (m->state_timeout.abs_value_us == state_timeout.abs_value_us) &&
1237 ((NULL == address) && (NULL == m->address)) )
1239 return; /* No real change */
1241 if ( (m->state == state) && ((NULL != address) && (NULL != m->address)) &&
1242 (0 == GNUNET_HELLO_address_cmp(m->address, address)) )
1243 return; /* No real change */
1246 if (NULL != m->address)
1248 GNUNET_free (m->address);
1251 if (NULL != address)
1252 m->address = GNUNET_HELLO_address_copy (address);
1254 m->state_timeout = state_timeout;
1256 if (NULL != address)
1257 resolve_peer_address (peer, m->address, numeric, m->state, m->state_timeout);
1259 print_info (peer, NULL, NULL, m->state, m->state_timeout);
1263 try_connect_cb (void *cls, const int result)
1265 static int retries = 0;
1266 if (GNUNET_OK == result)
1273 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1277 FPRINTF (stderr, "%s",
1278 _("Failed to send connect request to transport service\n") );
1279 if (GNUNET_SCHEDULER_NO_TASK != end)
1280 GNUNET_SCHEDULER_cancel (end);
1282 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1288 try_disconnect_cb (void *cls, const int result)
1290 static int retries = 0;
1291 if (GNUNET_OK == result)
1298 tc_handle = GNUNET_TRANSPORT_try_disconnect (handle, &pid, try_disconnect_cb,
1302 FPRINTF (stderr, "%s",
1303 _("Failed to send connect request to transport service\n") );
1304 if (GNUNET_SCHEDULER_NO_TASK != end)
1305 GNUNET_SCHEDULER_cancel (end);
1307 end = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL );
1313 * Function called with the result of the check if the 'transport'
1314 * service is running.
1316 * @param cls closure with our configuration
1317 * @param result #GNUNET_YES if transport is running
1320 testservice_task (void *cls, int result)
1325 if (GNUNET_YES != result)
1327 FPRINTF (stderr, _("Service `%s' is not running\n"), "transport");
1333 != GNUNET_CRYPTO_eddsa_public_key_from_string (cpid, strlen (cpid),
1336 FPRINTF (stderr, _("Failed to parse peer identity `%s'\n"), cpid);
1340 counter = benchmark_send + benchmark_receive + iterate_connections
1341 + monitor_connections + monitor_connects + try_connect + try_disconnect +
1342 + iterate_validation + monitor_validation;
1347 _("Multiple operations given. Please choose only one operation: %s, %s, %s, %s, %s, %s\n"),
1348 "connect", "benchmark send", "benchmark receive", "information",
1349 "monitor", "events");
1355 _("No operation given. Please choose one operation: %s, %s, %s, %s, %s, %s\n"),
1356 "connect", "benchmark send", "benchmark receive", "information",
1357 "monitor", "events");
1361 if (try_connect) /* -C: Connect to peer */
1365 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1370 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1371 ¬ify_connect, ¬ify_disconnect);
1374 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1378 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1380 if (NULL == tc_handle)
1382 FPRINTF (stderr, "%s",
1383 _("Failed to send request to transport service\n") );
1387 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1391 else if (try_disconnect) /* -D: Disconnect from peer */
1395 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1400 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1401 ¬ify_connect, ¬ify_disconnect);
1404 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1408 tc_handle = GNUNET_TRANSPORT_try_disconnect (handle, &pid, try_disconnect_cb,
1410 if (NULL == tc_handle)
1412 FPRINTF (stderr, "%s",
1413 _("Failed to send request to transport service\n") );
1417 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1421 else if (benchmark_send) /* -s: Benchmark sending */
1425 FPRINTF (stderr, _("Option `%s' makes no sense without option `%s'.\n"),
1430 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive,
1431 ¬ify_connect, ¬ify_disconnect);
1434 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1438 tc_handle = GNUNET_TRANSPORT_try_connect (handle, &pid, try_connect_cb,
1440 if (NULL == tc_handle)
1442 FPRINTF (stderr, "%s",
1443 _("Failed to send request to transport service\n") );
1447 start_time = GNUNET_TIME_absolute_get ();
1448 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1451 else if (benchmark_receive) /* -b: Benchmark receiving */
1453 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, ¬ify_receive, NULL,
1457 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1462 FPRINTF (stdout, "%s", _("Starting to receive benchmark data\n") );
1463 start_time = GNUNET_TIME_absolute_get ();
1466 else if (iterate_connections) /* -i: List information about peers once */
1468 address_resolution_in_progress = GNUNET_YES;
1469 pic = GNUNET_TRANSPORT_monitor_peers (cfg, (NULL == cpid) ? NULL : &pid,
1470 GNUNET_YES, TIMEOUT, &process_peer_iteration_cb, (void *) cfg);
1471 op_timeout = GNUNET_SCHEDULER_add_delayed (OP_TIMEOUT, &operation_timeout,
1474 else if (monitor_connections) /* -m: List information about peers continuously */
1476 monitored_peers = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1477 address_resolution_in_progress = GNUNET_YES;
1478 pic = GNUNET_TRANSPORT_monitor_peers (cfg, (NULL == cpid) ? NULL : &pid,
1479 GNUNET_NO, TIMEOUT, &process_peer_monitoring_cb, (void *) cfg);
1481 else if (iterate_validation) /* -d: Print information about validations */
1483 vic = GNUNET_TRANSPORT_monitor_validation_entries (cfg,
1484 (NULL == cpid) ? NULL : &pid,
1485 GNUNET_YES, TIMEOUT, &process_validation_cb, (void *) cfg);
1487 else if (monitor_validation) /* -f: Print information about validations continuously */
1489 vic = GNUNET_TRANSPORT_monitor_validation_entries (cfg,
1490 (NULL == cpid) ? NULL : &pid,
1491 GNUNET_NO, TIMEOUT, &process_validation_cb, (void *) cfg);
1493 else if (monitor_connects) /* -e : Monitor (dis)connect events continuously */
1495 monitor_connect_counter = 0;
1496 handle = GNUNET_TRANSPORT_connect (cfg, NULL, NULL, NULL,
1497 &monitor_notify_connect,
1498 &monitor_notify_disconnect);
1501 FPRINTF (stderr, "%s", _("Failed to connect to transport service\n") );
1513 end = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1514 &shutdown_task, NULL );
1519 * Main function that will be run by the scheduler.
1521 * @param cls closure
1522 * @param args remaining command-line arguments
1523 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1524 * @param mycfg configuration
1527 run (void *cls, char * const *args, const char *cfgfile,
1528 const struct GNUNET_CONFIGURATION_Handle *mycfg)
1530 cfg = (struct GNUNET_CONFIGURATION_Handle *) mycfg;
1531 if (test_configuration)
1533 do_test_configuration (cfg);
1536 GNUNET_CLIENT_service_test ("transport", cfg, GNUNET_TIME_UNIT_SECONDS,
1537 &testservice_task, (void *) cfg);
1541 main (int argc, char * const *argv)
1544 static const struct GNUNET_GETOPT_CommandLineOption options[] =
1547 gettext_noop ("print information for all peers (instead of only connected peers )"),
1548 0, &GNUNET_GETOPT_set_one, &iterate_all },
1549 { 'b', "benchmark", NULL,
1550 gettext_noop ("measure how fast we are receiving data from all peers (until CTRL-C)"),
1551 0, &GNUNET_GETOPT_set_one, &benchmark_receive },
1553 NULL, gettext_noop ("connect to a peer"), 0,
1554 &GNUNET_GETOPT_set_one, &try_connect },
1555 { 'D', "disconnect",
1556 NULL, gettext_noop ("disconnect to a peer"), 0,
1557 &GNUNET_GETOPT_set_one, &try_disconnect },
1558 { 'd', "validation", NULL,
1559 gettext_noop ("print information for all pending validations "),
1560 0, &GNUNET_GETOPT_set_one, &iterate_validation },
1561 { 'f', "monitorvalidation", NULL,
1562 gettext_noop ("print information for all pending validations continously"),
1563 0, &GNUNET_GETOPT_set_one, &monitor_validation },
1564 { 'i', "information", NULL,
1565 gettext_noop ("provide information about all current connections (once)"),
1566 0, &GNUNET_GETOPT_set_one, &iterate_connections },
1567 { 'm', "monitor", NULL,
1568 gettext_noop ("provide information about all current connections (continuously)"),
1569 0, &GNUNET_GETOPT_set_one, &monitor_connections },
1570 { 'e', "events", NULL,
1571 gettext_noop ("provide information about all connects and disconnect events (continuously)"),
1572 0, &GNUNET_GETOPT_set_one, &monitor_connects }, { 'n', "numeric",
1573 NULL, gettext_noop ("do not resolve hostnames"), 0,
1574 &GNUNET_GETOPT_set_one, &numeric }, { 'p', "peer", "PEER",
1575 gettext_noop ("peer identity"), 1, &GNUNET_GETOPT_set_string,
1576 &cpid }, { 's', "send", NULL, gettext_noop
1577 ("send data for benchmarking to the other peer (until CTRL-C)"), 0,
1578 &GNUNET_GETOPT_set_one, &benchmark_send },
1579 { 't', "test", NULL,
1580 gettext_noop ("test transport configuration (involves external server)"),
1581 0, &GNUNET_GETOPT_set_one, &test_configuration },
1582 GNUNET_GETOPT_OPTION_VERBOSE (&verbosity),
1583 GNUNET_GETOPT_OPTION_END };
1585 if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
1588 res = GNUNET_PROGRAM_run (argc, argv, "gnunet-transport", gettext_noop
1589 ("Direct access to transport service."), options, &run, NULL );
1590 GNUNET_free((void * ) argv);
1591 if (GNUNET_OK == res)
1596 /* end of gnunet-transport.c */