/*
    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_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;

static int ret;

static struct GNUNET_TIME_Absolute start_short;

static struct GNUNET_TIME_Absolute start_long;

static struct GNUNET_TRANSPORT_TESTING_TransportCommunicatorQueue *my_tc;

#define SHORT_MESSAGE_SIZE 128

#define LONG_MESSAGE_SIZE 32000

#define SHORT_BURST_SECONDS 2

#define LONG_BURST_SECONDS 2

#define SHORT_BURST_WINDOW \
  GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,SHORT_BURST_SECONDS)

#define LONG_BURST_WINDOW \
  GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,SHORT_BURST_SECONDS)

#define BURST_SHORT 0

#define BURST_LONG 1

#define SIZE_CHECK 2

#define MAX_BUF_LEN 10

static int buf_len = 0;

static char short_payload[SHORT_MESSAGE_SIZE];

static char long_payload[LONG_MESSAGE_SIZE];

static uint32_t ack = 0;

static int phase;

static size_t long_received = 0;

static size_t short_received = 0;

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[NUM_PEERS - 1]))
    GNUNET_TRANSPORT_TESTING_transport_communicator_open_queue (tc_hs[0],
                                                                &peer_id[
                                                                  NUM_PEERS
                                                                  - 1],
                                                                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 void
size_test (void *cls)
{
  char payload[ack];
  phase = SIZE_CHECK;

  memset (payload, 0, ack);
  if (ack < 64000) //Leave some room for our protocol.
  {
    GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
                                                          &payload,
                                                          sizeof(payload));
    return;
  }
  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
              "Finished\n");
  GNUNET_SCHEDULER_shutdown ();
  // Finished!
}


static void
long_test (void *cls)
{
  struct GNUNET_TIME_Relative duration = GNUNET_TIME_absolute_get_duration (
    start_long);
  if (LONG_BURST_WINDOW.rel_value_us > duration.rel_value_us)
  {
    if (buf_len < MAX_BUF_LEN)
    {
      GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
                                                            &long_payload,
                                                            sizeof(long_payload));
      buf_len++;
    }
    GNUNET_SCHEDULER_add_now (&long_test, NULL);
    return;
  }
  GNUNET_log (GNUNET_ERROR_TYPE_MESSAGE,
              "LONG Goodput (bytes/s): %lu - received packets: %lu\n",
              (LONG_MESSAGE_SIZE * long_received) / LONG_BURST_SECONDS,
              long_received);
  ack = 5;
  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, &size_test, NULL);
}


static void
short_test (void *cls)
{
  struct GNUNET_TIME_Relative duration = GNUNET_TIME_absolute_get_duration (
    start_short);
  if (SHORT_BURST_WINDOW.rel_value_us > duration.rel_value_us)
  {
    if (buf_len < MAX_BUF_LEN)
    {
      GNUNET_TRANSPORT_TESTING_transport_communicator_send (my_tc,
                                                            &short_payload,
                                                            sizeof(short_payload));
      buf_len++;
    }
    GNUNET_SCHEDULER_add_now (&short_test, NULL);
    return;
  }
  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
              "Short test done!\n");
  GNUNET_log (GNUNET_ERROR_TYPE_MESSAGE,
              "SHORT Goodput (bytes/s): %lu - received packets: %lu\n",
              (SHORT_MESSAGE_SIZE * short_received) / SHORT_BURST_SECONDS,
              short_received);
  start_long = GNUNET_TIME_absolute_get ();
  phase = BURST_LONG;
  buf_len = 0;
  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, &long_test, NULL);
}


/**
 * @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)
{
  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_queue;
  buf_len = 0;
  phase = BURST_SHORT;
  GNUNET_SCHEDULER_add_now (&short_test, tc_queue);
}


/**
 * @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
 */
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]))
    return; // TODO?
  if (phase == BURST_SHORT)
  {
    GNUNET_assert (SHORT_MESSAGE_SIZE == payload_len);
    short_received++;
    buf_len--;
  }
  else if (phase == BURST_LONG)
  {
    if (LONG_MESSAGE_SIZE != payload_len)
      return; // Ignore
    long_received++;
    buf_len--;
  }
  else         // if (phase == SIZE_CHECK) {
  {
    // GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
    //          "Receiving payload with size %lu...\n", payload_len);
    if (ack != payload_len)
    {
      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
                  "Error receiving message, corrupted.\n");
      ret = 1;
      GNUNET_SCHEDULER_shutdown ();
      return;
    }
    ack += 5; // Next expected message size
    GNUNET_SCHEDULER_add_now (&size_test, NULL);
  }
}


/**
 * @brief Main function called by the scheduler
 *
 * @param cls Closure - Handle to configuration
 */
static void
run (void *cls)
{
  ret = 0;
  memset (long_payload, 0, LONG_MESSAGE_SIZE);
  memset (short_payload, 0, SHORT_MESSAGE_SIZE);
  for (int i = 0; i < NUM_PEERS; i++)
  {
    tc_hs[i] = GNUNET_TRANSPORT_TESTING_transport_communicator_service_start (
      "transport",
      communicator_binary,
      cfg_peers_name[i],
      &communicator_available_cb,
      &add_address_cb,
      &queue_create_reply_cb,
      &add_queue_cb,
      &incoming_message_cb,
      cfg_peers_name[i]); /* cls */
  }
}


int
main (int argc,
      char *const *argv)
{
  struct GNUNET_CRYPTO_EddsaPrivateKey *private_key;
  char *communicator_name;
  char *cfg_peer;
  ret = 1;

  communicator_name = GNUNET_TESTING_get_testname_from_underscore (argv[0]);
  GNUNET_asprintf (&communicator_binary, "gnunet-communicator-%s",
                   communicator_name);
  cfg_peers_name = GNUNET_malloc (sizeof(char*) * NUM_PEERS);
  if (GNUNET_OK != GNUNET_log_setup ("test_communicator_basic",
                                     "DEBUG",
                                     "test_communicator_basic.log"))
  {
    fprintf (stderr, "Unable to setup log\n");
    GNUNET_break (0);
    return 2;
  }
  for (int i = 0; i < NUM_PEERS; i++)
  {
    GNUNET_asprintf ((&cfg_peer),
                     "test_communicator_%s_peer%u.conf",
                     communicator_name, 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);
    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
                "Identity of peer %u is %s\n",
                i, GNUNET_i2s_full (&peer_id[i]));
  }
  fprintf (stderr, "Starting test...\n");
  GNUNET_SCHEDULER_run (&run,
                        NULL);
  return ret;
}