123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651 |
- /*
- This file is part of GNUnet.
- Copyright (C) 2019 GNUnet e.V.
- GNUnet is free software: you can redistribute it and/or modify it
- under the terms of the GNU Affero General Public License as published
- by the Free Software Foundation, either version 3 of the License,
- or (at your option) any later version.
- GNUnet is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Affero General Public License for more details.
- You should have received a copy of the GNU Affero General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>.
- SPDX-License-Identifier: AGPL3.0-or-later
- */
- /**
- * @file transport/test_communicator_basic.c
- * @brief test the communicators
- * @author Julius Bünger
- * @author Martin Schanzenbach
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "transport-testing2.h"
- #include "gnunet_ats_transport_service.h"
- #include "gnunet_signatures.h"
- #include "gnunet_testing_lib.h"
- #include "transport.h"
- #include <inttypes.h>
- #define LOG(kind, ...) GNUNET_log_from (kind, \
- "test_transport_communicator", \
- __VA_ARGS__)
- #define NUM_PEERS 2
- static struct GNUNET_SCHEDULER_Task *to_task;
- static int queue_est = GNUNET_NO;
- static struct GNUNET_PeerIdentity peer_id[NUM_PEERS];
- static char *communicator_binary;
- static struct
- GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *tc_hs[NUM_PEERS];
- static struct GNUNET_CONFIGURATION_Handle *cfg_peers[NUM_PEERS];
- static char *cfg_peers_name[NUM_PEERS];
- static int ret;
- static size_t long_message_size;
- static struct GNUNET_TIME_Absolute start_short;
- static struct GNUNET_TIME_Absolute start_long;
- static struct GNUNET_TIME_Absolute timeout;
- static struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *my_tc;
- #define SHORT_MESSAGE_SIZE 128
- #define LONG_MESSAGE_SIZE 32000 /* FIXME */
- #define BURST_PACKETS 5000
- #define TOTAL_ITERATIONS 1
- #define PEER_A 0
- #define PEER_B 1
- static unsigned int iterations_left = TOTAL_ITERATIONS;
- #define SHORT_BURST_WINDOW \
- GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,2)
- #define LONG_BURST_WINDOW \
- GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,2)
- enum TestPhase
- {
- TP_INIT,
- TP_BURST_SHORT,
- TP_BURST_LONG,
- TP_SIZE_CHECK
- };
- static size_t num_sent = 0;
- static uint32_t ack = 0;
- static enum TestPhase phase;
- static size_t num_received = 0;
- static uint64_t avg_latency = 0;
- static struct GNUNET_TIME_Relative duration;
- static void
- communicator_available_cb (void *cls,
- struct
- GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle
- *tc_h,
- enum GNUNET_TRANSPORT_CommunicatorCharacteristics cc,
- char *address_prefix)
- {
- LOG (GNUNET_ERROR_TYPE_INFO,
- "Communicator available. (cc: %u, prefix: %s)\n",
- cc,
- address_prefix);
- }
- static void
- add_address_cb (void *cls,
- struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *
- tc_h,
- const char *address,
- struct GNUNET_TIME_Relative expiration,
- uint32_t aid,
- enum GNUNET_NetworkType nt)
- {
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "New address. (addr: %s, expir: %" PRIu32 ", ID: %" PRIu32 ", nt: %u\n",
- address,
- expiration.rel_value_us,
- aid,
- nt);
- // addresses[1] = GNUNET_strdup (address);
- if ((0 == strcmp ((char*) cls, cfg_peers_name[PEER_B])) &&
- (GNUNET_NO == queue_est))
- {
- queue_est = GNUNET_YES;
- GNUNET_TRANSPORT_TESTING_transport_communicator_open_queue (tc_hs[PEER_A],
- &peer_id[PEER_B],
- address);
- }
- }
- /**
- * @brief Callback that informs whether the requested queue will be
- * established
- *
- * Implements #GNUNET_TRANSPORT_TESTING_QueueCreateReplyCallback.
- *
- * @param cls Closure - unused
- * @param tc_h Communicator handle - unused
- * @param will_try #GNUNET_YES if queue will be established
- * #GNUNET_NO if queue will not be established (bogous address)
- */
- static void
- queue_create_reply_cb (void *cls,
- struct
- GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *
- tc_h,
- int will_try)
- {
- if (GNUNET_YES == will_try)
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Queue will be established!\n");
- else
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "Queue won't be established (bougus address?)!\n");
- }
- static struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *
- handle_backchannel_cb (void *cls,
- struct GNUNET_MessageHeader *msg,
- struct GNUNET_PeerIdentity *pid)
- {
- struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *tc_h = cls;
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Handling BC message...\n");
- if (0 == memcmp (&peer_id[PEER_A], pid, sizeof (*pid)))
- return tc_hs[PEER_A];
- else
- return tc_hs[PEER_B];
- }
- static char*
- make_payload (size_t payload_size)
- {
- struct GNUNET_TIME_Absolute ts;
- struct GNUNET_TIME_AbsoluteNBO ts_n;
- char *payload = GNUNET_malloc (payload_size);
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Making payload of size %lu\n", payload_size);
- GNUNET_assert (payload_size >= 8); // So that out timestamp fits
- ts = GNUNET_TIME_absolute_get ();
- ts_n = GNUNET_TIME_absolute_hton (ts);
- memset (payload, 0, payload_size);
- memcpy (payload, &ts_n, sizeof (struct GNUNET_TIME_AbsoluteNBO));
- return payload;
- }
- static void
- latency_timeout (void *cls)
- {
- to_task = NULL;
- if (GNUNET_TIME_absolute_get_remaining (timeout).rel_value_us > 0)
- {
- to_task = GNUNET_SCHEDULER_add_at (timeout,
- &latency_timeout,
- NULL);
- return;
- }
- LOG (GNUNET_ERROR_TYPE_ERROR,
- "Latency too high. Test failed. (Phase: %d. Sent: %lu, Received: %lu)\n",
- phase, num_sent, num_received);
- ret = 2;
- GNUNET_SCHEDULER_shutdown ();
- }
- static void
- size_test (void *cls)
- {
- char *payload;
- size_t max_size = 64000;
- GNUNET_assert (TP_SIZE_CHECK == phase);
- if (LONG_MESSAGE_SIZE != long_message_size)
- max_size = long_message_size;
- if (ack + 10 > max_size)
- return; /* Leave some room for our protocol, so not 2^16 exactly */
- ack += 10;
- payload = make_payload (ack);
- num_sent++;
- GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
- (ack < max_size)
- ? &size_test
- : NULL,
- NULL,
- payload,
- ack);
- GNUNET_free (payload);
- timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_SECONDS);
- }
- static void
- long_test (void *cls)
- {
- char *payload;
- payload = make_payload (long_message_size);
- num_sent++;
- GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
- (BURST_PACKETS ==
- num_sent)
- ? NULL
- : &long_test,
- NULL,
- payload,
- long_message_size);
- GNUNET_free (payload);
- timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_SECONDS);
- }
- static void
- short_test (void *cls)
- {
- char *payload;
- payload = make_payload (SHORT_MESSAGE_SIZE);
- num_sent++;
- GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
- (BURST_PACKETS ==
- num_sent)
- ? NULL
- : &short_test,
- NULL,
- payload,
- SHORT_MESSAGE_SIZE);
- GNUNET_free (payload);
- timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_SECONDS);
- }
- static int test_prepared = GNUNET_NO;
- /**
- * This helps establishing the backchannel
- */
- static void
- prepare_test (void *cls)
- {
- char *payload;
- if (GNUNET_YES == test_prepared)
- {
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
- &short_test,
- NULL);
- return;
- }
- test_prepared = GNUNET_YES;
- payload = make_payload (SHORT_MESSAGE_SIZE);
- GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
- &prepare_test,
- NULL,
- payload,
- SHORT_MESSAGE_SIZE);
- GNUNET_free (payload);
- }
- /**
- * @brief Handle opening of queue
- *
- * Issues sending of test data
- *
- * Implements #GNUNET_TRANSPORT_TESTING_AddQueueCallback
- *
- * @param cls Closure
- * @param tc_h Communicator handle
- * @param tc_queue Handle to newly opened queue
- */
- static void
- add_queue_cb (void *cls,
- struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle *tc_h,
- struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorQueue *
- tc_queue,
- size_t mtu)
- {
- if (TP_INIT != phase)
- return;
- if (0 != strcmp ((char*) cls, cfg_peers_name[0]))
- return; // TODO?
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Queue established, starting test...\n");
- start_short = GNUNET_TIME_absolute_get ();
- my_tc = tc_h;
- if (0 != mtu)
- long_message_size = mtu - 4; /* Dummy message header overhead */
- else
- long_message_size = LONG_MESSAGE_SIZE;
- phase = TP_BURST_SHORT;
- timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_MINUTES);
- GNUNET_assert (NULL == to_task);
- to_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MINUTES,
- &latency_timeout,
- NULL);
- prepare_test (NULL);
- }
- static void
- update_avg_latency (const char*payload)
- {
- struct GNUNET_TIME_AbsoluteNBO *ts_n;
- struct GNUNET_TIME_Absolute ts;
- struct GNUNET_TIME_Relative latency;
- ts_n = (struct GNUNET_TIME_AbsoluteNBO *) payload;
- ts = GNUNET_TIME_absolute_ntoh (*ts_n);
- latency = GNUNET_TIME_absolute_get_duration (ts);
- if (1 >= num_received)
- avg_latency = latency.rel_value_us;
- else
- avg_latency = ((avg_latency * (num_received - 1)) + latency.rel_value_us)
- / num_received;
- }
- /**
- * @brief Handle an incoming message
- *
- * Implements #GNUNET_TRANSPORT_TESTING_IncomingMessageCallback
- * @param cls Closure
- * @param tc_h Handle to the receiving communicator
- * @param msg Received message
- */
- static void
- incoming_message_cb (void *cls,
- struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorHandle
- *tc_h,
- const char*payload,
- size_t payload_len)
- {
- if (0 != strcmp ((char*) cls, cfg_peers_name[NUM_PEERS - 1]))
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "unexpected receiver...\n");
- return;
- }
- /* Reset timeout */
- timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_SECONDS);
- switch (phase)
- {
- case TP_INIT:
- GNUNET_break (0);
- break;
- case TP_BURST_SHORT:
- {
- GNUNET_assert (SHORT_MESSAGE_SIZE == payload_len);
- num_received++;
- duration = GNUNET_TIME_absolute_get_duration (start_short);
- update_avg_latency (payload);
- if (num_received == BURST_PACKETS)
- {
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "Short size packet test done.\n");
- char *goodput = GNUNET_STRINGS_byte_size_fancy ((SHORT_MESSAGE_SIZE
- * num_received * 1000
- * 1000)
- / duration.rel_value_us);
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "%lu/%lu packets in %llu us (%s/s) -- avg latency: %llu us\n",
- (unsigned long) num_received,
- (unsigned long) num_sent,
- (unsigned long long) duration.rel_value_us,
- goodput,
- (unsigned long long) avg_latency);
- GNUNET_free (goodput);
- start_long = GNUNET_TIME_absolute_get ();
- phase = TP_BURST_LONG;
- num_sent = 0;
- avg_latency = 0;
- num_received = 0;
- long_test (NULL);
- }
- break;
- }
- case TP_BURST_LONG:
- {
- if (long_message_size != payload_len)
- {
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "Ignoring packet with wrong length\n");
- return; // Ignore
- }
- num_received++;
- duration = GNUNET_TIME_absolute_get_duration (start_long);
- update_avg_latency (payload);
- if (num_received == BURST_PACKETS)
- {
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "Long size packet test done.\n");
- char *goodput = GNUNET_STRINGS_byte_size_fancy ((long_message_size
- * num_received * 1000
- * 1000)
- / duration.rel_value_us);
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "%lu/%lu packets in %llu us (%s/s) -- avg latency: %llu us\n",
- (unsigned long) num_received,
- (unsigned long) num_sent,
- (unsigned long long) duration.rel_value_us,
- goodput,
- (unsigned long long) avg_latency);
- GNUNET_free (goodput);
- ack = 0;
- phase = TP_SIZE_CHECK;
- num_received = 0;
- num_sent = 0;
- avg_latency = 0;
- size_test (NULL);
- }
- break;
- }
- case TP_SIZE_CHECK:
- {
- size_t max_size = 64000;
- GNUNET_assert (TP_SIZE_CHECK == phase);
- if (LONG_MESSAGE_SIZE != long_message_size)
- max_size = long_message_size;
- num_received++;
- update_avg_latency (payload);
- if (num_received >= (max_size) / 10)
- {
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "Size packet test done.\n");
- LOG (GNUNET_ERROR_TYPE_MESSAGE,
- "%lu/%lu packets -- avg latency: %llu us\n",
- (unsigned long) num_received,
- (unsigned long) num_sent,
- (unsigned long long) avg_latency);
- num_received = 0;
- num_sent = 0;
- avg_latency = 0;
- iterations_left--;
- if (0 != iterations_left)
- {
- start_short = GNUNET_TIME_absolute_get ();
- phase = TP_BURST_SHORT;
- short_test (NULL);
- break;
- }
- LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Finished\n");
- GNUNET_SCHEDULER_shutdown ();
- }
- break;
- }
- }
- }
- static void
- do_shutdown (void *cls)
- {
- if (NULL != to_task)
- {
- GNUNET_SCHEDULER_cancel (to_task);
- to_task = NULL;
- }
- for (unsigned int i = 0; i < NUM_PEERS; i++)
- {
- GNUNET_TRANSPORT_TESTING_transport_communicator_service_stop (tc_hs[i]);
- }
- }
- /**
- * @brief Main function called by the scheduler
- *
- * @param cls Closure - Handle to configuration
- */
- static void
- run (void *cls)
- {
- ret = 0;
- num_received = 0;
- num_sent = 0;
- for (unsigned int i = 0; i < NUM_PEERS; i++)
- {
- tc_hs[i] = GNUNET_TRANSPORT_TESTING_transport_communicator_service_start (
- "transport",
- communicator_binary,
- cfg_peers_name[i],
- &peer_id[i],
- &communicator_available_cb,
- &add_address_cb,
- &queue_create_reply_cb,
- &add_queue_cb,
- &incoming_message_cb,
- &handle_backchannel_cb,
- cfg_peers_name[i]); /* cls */
- }
- GNUNET_SCHEDULER_add_shutdown (&do_shutdown,
- NULL);
- }
- int
- main (int argc,
- char *const *argv)
- {
- struct GNUNET_CRYPTO_EddsaPrivateKey *private_key;
- char *communicator_name;
- char *test_mode;
- char *test_name;
- char *cfg_peer;
- phase = TP_INIT;
- ret = 1;
- test_name = GNUNET_TESTING_get_testname_from_underscore (argv[0]);
- communicator_name = strchr (test_name, '-');
- communicator_name[0] = '\0';
- communicator_name++;
- test_mode = test_name;
- GNUNET_asprintf (&communicator_binary,
- "gnunet-communicator-%s",
- communicator_name);
- if (GNUNET_OK !=
- GNUNET_log_setup ("test_communicator_basic",
- "DEBUG",
- NULL))
- {
- fprintf (stderr, "Unable to setup log\n");
- GNUNET_break (0);
- return 2;
- }
- for (unsigned int i = 0; i < NUM_PEERS; i++)
- {
- GNUNET_asprintf ((&cfg_peer),
- "test_communicator_%s_%s_peer%u.conf",
- communicator_name, test_mode, i + 1);
- cfg_peers_name[i] = cfg_peer;
- cfg_peers[i] = GNUNET_CONFIGURATION_create ();
- if (GNUNET_YES ==
- GNUNET_DISK_file_test (cfg_peers_name[i]))
- {
- if (GNUNET_SYSERR ==
- GNUNET_CONFIGURATION_load (cfg_peers[i],
- cfg_peers_name[i]))
- {
- fprintf (stderr,
- "Malformed configuration file `%s', exiting ...\n",
- cfg_peers_name[i]);
- return 1;
- }
- }
- else
- {
- if (GNUNET_SYSERR ==
- GNUNET_CONFIGURATION_load (cfg_peers[i],
- NULL))
- {
- fprintf (stderr,
- "Configuration file %s does not exist, exiting ...\n",
- cfg_peers_name[i]);
- return 1;
- }
- }
- private_key =
- GNUNET_CRYPTO_eddsa_key_create_from_configuration (cfg_peers[i]);
- if (NULL == private_key)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR,
- "Unable to get peer ID\n");
- return 1;
- }
- GNUNET_CRYPTO_eddsa_key_get_public (private_key,
- &peer_id[i].public_key);
- GNUNET_free (private_key);
- LOG (GNUNET_ERROR_TYPE_INFO,
- "Identity of peer %u is %s\n",
- i,
- GNUNET_i2s_full (&peer_id[i]));
- }
- LOG (GNUNET_ERROR_TYPE_MESSAGE, "Starting test...\n");
- GNUNET_SCHEDULER_run (&run,
- NULL);
- return ret;
- }
|