path: root/src/mesh/test_mesh_regex.c

/*
     This file is part of GNUnet.
     (C) 2012 Christian Grothoff (and other contributing authors)

     GNUnet is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published
     by the Free Software Foundation; either version 3, 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
     General Public License for more details.

     You should have received a copy of the GNU General Public License
     along with GNUnet; see the file COPYING.  If not, write to the
     Free Software Foundation, Inc., 59 Temple Place - Suite 330,
     Boston, MA 02111-1307, USA.
*/
/**
 * @file mesh/test_mesh_regex.c
 *
 * @brief Test for regex announce / by_string connect.
 * based on the 2dtorus testcase
 */
#include "platform.h"
#include "mesh_test_lib.h"
#include "gnunet_mesh_service.h"

#define REMOVE_DIR GNUNET_YES
#define MESH_REGEX_PEERS 4

/**
 * How long until we give up on connecting the peers?
 */
#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 300)

/**
 * Time to wait for stuff that should be rather fast
 */
#define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60)

/**
 * Which strings have been found & connected.
 */
static int ok[MESH_REGEX_PEERS];

/**
 * How many connects have happened.
 */
static int regex_peers;

/**
 * Total number of currently running peers.
 */
static unsigned long long peers_running;

/**
 * Task called to disconnect peers
 */
static GNUNET_SCHEDULER_TaskIdentifier disconnect_task;

/**
 * Task called to shutdown test.
 */
static GNUNET_SCHEDULER_TaskIdentifier shutdown_handle;

/**
 * Mesh handle for connecting peer.
 */
static struct GNUNET_MESH_Handle *h1;

/**
 * Mesh handle for announcing peers.
 */
static struct GNUNET_MESH_Handle *h2[MESH_REGEX_PEERS];

/**
 * Tunnel handles for announcing peer.
 */
static struct GNUNET_MESH_Tunnel *t[MESH_REGEX_PEERS];

/**
 * Incoming tunnels for announcing peers.
 */
static struct GNUNET_MESH_Tunnel *incoming_t[MESH_REGEX_PEERS];

/**
 * Test context (to shut down).
 */
struct GNUNET_MESH_TEST_Context *test_ctx;

/**
 * Regular expressions for the announces.
 */
static char *regexes[MESH_REGEX_PEERS] = {"(0|1)"
                                          "(0|1)"
                                          "23456789ABC",

                                          "0123456789A*BC",

                                          "1234567890123456789012340*123456789ABC*",

                                          "GNUNETVPN0001000IPEX401110011101100100000111(0|1)*"};


/**
 * Service strings to look for.
 */
static char *strings[MESH_REGEX_PEERS] = {"1123456789ABC",

                                          "0123456789AABC",

                                          "12345678901234567890123400123456789ABCCCC",

                                          "GNUNETVPN0001000IPEX401110011101100100000111"};



/**
 * Task to run for shutdown: stops peers, ends test.
 *
 * @param cls Closure (not used).
 * @param tc TaskContext.
 *
 */
static void
shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Ending test.\n");
  shutdown_handle = GNUNET_SCHEDULER_NO_TASK;
}


/**
 * Ends test: Disconnects peers and calls shutdown.
 * @param cls Closure (not used).
 * @param tc TaskContext.
 */
static void
disconnect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
  unsigned int i;

  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: disconnecting peers\n");

  for (i = 0; i < MESH_REGEX_PEERS; i++)
  {
    GNUNET_MESH_tunnel_destroy (t[i]);
  }
  if (GNUNET_SCHEDULER_NO_TASK != shutdown_handle)
  {
    GNUNET_SCHEDULER_cancel (shutdown_handle);
  }
  GNUNET_MESH_TEST_cleanup (test_ctx);
  shutdown_handle = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
}

/**
 * Function called whenever an inbound tunnel is destroyed.  Should clean up
 * any associated state.
 *
 * @param cls closure (set from GNUNET_MESH_connect)
 * @param tunnel connection to the other end (henceforth invalid)
 * @param tunnel_ctx place where local state associated
 *                   with the tunnel is stored
 */
static void
tunnel_cleaner (void *cls, const struct GNUNET_MESH_Tunnel *tunnel,
                void *tunnel_ctx)
{
  long i = (long) cls;

  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
              "Incoming tunnel disconnected at peer %d\n",
              i);
  return;
}


/**
 * Method called whenever a tunnel falls apart.
 *
 * @param cls closure
 * @param peer peer identity the tunnel stopped working with
 */
static void
dh (void *cls, const struct GNUNET_PeerIdentity *peer)
{
  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
              "peer %s disconnected\n",
              GNUNET_i2s (peer));
  return;
}

/**
 * Function called to notify a client about the connection
 * begin ready to queue more data.  "buf" will be
 * NULL and "size" zero if the connection was closed for
 * writing in the meantime.
 *
 * @param cls closure
 * @param size number of bytes available in buf
 * @param buf where the callee should write the message
 * @return number of bytes written to buf
 */
static size_t
data_ready (void *cls, size_t size, void *buf)
{
  struct GNUNET_MessageHeader *m = buf;

  if (NULL == buf || size < sizeof(struct GNUNET_MessageHeader))
    return 0;
  m->type = htons (1);
  m->size = htons (sizeof(struct GNUNET_MessageHeader));
  return sizeof(struct GNUNET_MessageHeader);
}

/**
 * Method called whenever a peer connects to a tunnel.
 *
 * @param cls closure
 * @param peer peer identity the tunnel was created to, NULL on timeout
 * @param atsi performance data for the connection
 */
static void
ch (void *cls, const struct GNUNET_PeerIdentity *peer,
    const struct GNUNET_ATS_Information *atsi)
{
  long i = (long) cls;

  GNUNET_log (GNUNET_ERROR_TYPE_INFO,
              "Peer connected: %s\n",
              GNUNET_i2s (peer));
  regex_peers++;

  GNUNET_MESH_notify_transmit_ready(t[i], GNUNET_NO,
                                    GNUNET_TIME_UNIT_FOREVER_REL,
                                    peer,
                                    sizeof(struct GNUNET_MessageHeader),
                                    &data_ready, NULL);
}

/**
 * Method called whenever another peer has added us to a tunnel
 * the other peer initiated.
 *
 * @param cls closure
 * @param tunnel new handle to the tunnel
 * @param initiator peer that started the tunnel
 * @param atsi performance information for the tunnel
 * @return initial tunnel context for the tunnel
 *         (can be NULL -- that's not an error)
 */
static void *
incoming_tunnel (void *cls, struct GNUNET_MESH_Tunnel *tunnel,
                 const struct GNUNET_PeerIdentity *initiator,
                 const struct GNUNET_ATS_Information *atsi)
{
  long i = (long) cls;

  GNUNET_log (GNUNET_ERROR_TYPE_INFO,
              "Incoming tunnel from %s to peer %d\n",
              GNUNET_i2s (initiator), i);
  if ( (i >= 10L) && (i < 10L + MESH_REGEX_PEERS))
  {
    incoming_t[i - 10L] = tunnel;
    ok[i - 10L] = GNUNET_OK;
  }
  else
  {
    GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
                "Incoming tunnel for unknown client %lu\n", (long) cls);
  }
  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
  {
    GNUNET_SCHEDULER_cancel (disconnect_task);
    disconnect_task =
        GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_peers, NULL);
  }
  return NULL;
}

/**
 * Function is called whenever a message is received.
 *
 * @param cls closure (set from GNUNET_MESH_connect)
 * @param tunnel connection to the other end
 * @param tunnel_ctx place to store local state associated with the tunnel
 * @param sender who sent the message
 * @param message the actual message
 * @param atsi performance data for the connection
 * @return GNUNET_OK to keep the connection open,
 *         GNUNET_SYSERR to close it (signal serious error)
 */
int
data_callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
               const struct GNUNET_PeerIdentity *sender,
               const struct GNUNET_MessageHeader *message,
               const struct GNUNET_ATS_Information *atsi)
{
  unsigned int i;
  long peer_number = (long) cls;

  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "got data on peer %ld!\n", peer_number);
  for (i = 0; i < MESH_REGEX_PEERS; i++)
  {
    if (GNUNET_OK != ok[i]) {
      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
              "data from peer %u still missing!\n", i + 10);
      return GNUNET_OK;
    }
  }
  GNUNET_log (GNUNET_ERROR_TYPE_INFO,
              "test: EVERYONE GOT DATA, FINISHING!\n");
  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
  {
    GNUNET_SCHEDULER_cancel (disconnect_task);
  }
  disconnect_task =
      GNUNET_SCHEDULER_add_now (&disconnect_peers, NULL);
  return GNUNET_OK;
}

/**
 * Handlers, for diverse services
 */
static struct GNUNET_MESH_MessageHandler handlers[] = {
  {&data_callback, 1, sizeof (struct GNUNET_MessageHeader)},
  {NULL, 0, 0}
};


/**
 * test main: start test when all peers are connected
 *
 * @param cls Closure.
 * @param ctx Argument to give to GNUNET_MESH_TEST_cleanup on test end.
 * @param num_peers Number of peers that are running.
 * @param peers Array of peers.
 * @param meshes Handle to each of the MESHs of the peers.
 */
static void
tmain (void *cls,
       struct GNUNET_MESH_TEST_Context *ctx,
       unsigned int num_peers,
       struct GNUNET_TESTBED_Peer **peers,
       struct GNUNET_MESH_Handle **meshes)
{
  unsigned int i;

  shutdown_handle = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
                                                  &shutdown_task, NULL);
  test_ctx = ctx;
  if (16 != num_peers)
  {
    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
                "running peers mismatch, aborting test!\n");
    GNUNET_MESH_TEST_cleanup (ctx);
    return;
  }
  peers_running = num_peers;
  disconnect_task =
    GNUNET_SCHEDULER_add_delayed (TIMEOUT, &disconnect_peers, NULL);

//   h1 = GNUNET_MESH_connect (d->cfg, (void *) 1L,
//                             NULL,
//                             NULL,
//                             handlers,
//                             &app);
  h1 = meshes[0];
  regex_peers = 0;
  for (i = 0; i < MESH_REGEX_PEERS; i++)
  {
    ok[i] = GNUNET_NO;
//     h2[i] = GNUNET_MESH_connect (d->cfg, (void *) (long) (i + 2),
//                                  &incoming_tunnel,
//                                  &tunnel_cleaner,
//                                  handlers,
//                                  &app);
    h2[i] = meshes[10 + i];
    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
                "Announce REGEX %u: %s\n", i, regexes[i]);
    GNUNET_MESH_announce_regex (h2[i], regexes[i], 1);
  }

  for (i = 0; i < MESH_REGEX_PEERS; i++)
  {
    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
                "Create tunnel\n");
    t[i] = GNUNET_MESH_tunnel_create (h1, NULL, &ch, &dh, (void *) (long) i);
    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
                "Connect by string %s\n", strings[i]);
    GNUNET_MESH_peer_request_connect_by_string (t[i], strings[i]);
  }
  /* connect handler = success, timeout = error */

}


/**
 * Main: start test
 */
int
main (int argc, char *argv[])
{
  int result;
  unsigned int i;

  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Start\n");

  GNUNET_MESH_TEST_run ("test_mesh_regex",
                        "test_mesh_2dtorus.conf",
                        16,
                        &tmain,
                        NULL,
                        &incoming_tunnel,
                        &tunnel_cleaner,
                        handlers,
                        NULL);

  result = GNUNET_OK;
  for (i = 0; i < MESH_REGEX_PEERS; i++)
  {
    if (GNUNET_OK != ok[i])
    {
      result = GNUNET_SYSERR;
      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
                  "COULD NOT CONNECT TO %u: %s!\n",
                  i, strings[i]);
    }
  }
  if (GNUNET_OK != result || regex_peers != MESH_REGEX_PEERS)
  {
    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
                "FAILED! %u connected peers\n",
                regex_peers);
    return 1;
  }
  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "success :)\n");
  return 0;
}

/* end of test_mesh_regex.c */