dht_test_lib.c

Go to the documentation of this file.

/*
     This file is part of GNUnet.
     Copyright (C) 2012 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
 */
#include "platform.h"
#include "dht_test_lib.h"
 
struct GNUNET_DHT_TEST_Context
{
  struct GNUNET_TESTBED_Peer **peers;
 
  struct GNUNET_DHT_Handle **dhts;
 
  struct GNUNET_TESTBED_Operation **ops;
 
  GNUNET_DHT_TEST_AppMain app_main;
 
  void *app_main_cls;
 
  unsigned int num_peers;
};
 
 
static void *
dht_connect_adapter (void *cls,
                     const struct GNUNET_CONFIGURATION_Handle *cfg)
{
  return GNUNET_DHT_connect (cfg, 16);
}
 
 
static void
dht_disconnect_adapter (void *cls,
                        void *op_result)
{
  struct GNUNET_DHT_Handle *dht = op_result;
 
  GNUNET_DHT_disconnect (dht);
}
 
 
static void
dht_connect_cb (void *cls,
                struct GNUNET_TESTBED_Operation *op,
                void *ca_result,
                const char *emsg)
{
  struct GNUNET_DHT_TEST_Context *ctx = cls;
 
  if (NULL != emsg)
  {
    fprintf (stderr,
             "Failed to connect to DHT service: %s\n",
             emsg);
    GNUNET_SCHEDULER_shutdown ();
    return;
  }
  for (unsigned int i = 0; i < ctx->num_peers; i++)
    if (op == ctx->ops[i])
      ctx->dhts[i] = ca_result;
  for (unsigned int i = 0; i < ctx->num_peers; i++)
    if (NULL == ctx->dhts[i])
      return;
  /* still some DHT connections missing */
  /* all DHT connections ready! */
  ctx->app_main (ctx->app_main_cls,
                 ctx,
                 ctx->num_peers,
                 ctx->peers,
                 ctx->dhts);
}
 
 
void
GNUNET_DHT_TEST_cleanup (struct GNUNET_DHT_TEST_Context *ctx)
{
  for (unsigned int i = 0; i < ctx->num_peers; i++)
    GNUNET_TESTBED_operation_done (ctx->ops[i]);
  GNUNET_free (ctx->ops);
  GNUNET_free (ctx->dhts);
  GNUNET_free (ctx);
  GNUNET_SCHEDULER_shutdown ();
}
 
 
static void
dht_test_run (void *cls,
              struct GNUNET_TESTBED_RunHandle *h,
              unsigned int num_peers,
              struct GNUNET_TESTBED_Peer **peers,
              unsigned int links_succeeded,
              unsigned int links_failed)
{
  struct GNUNET_DHT_TEST_Context *ctx = cls;
 
  GNUNET_assert (num_peers == ctx->num_peers);
  ctx->peers = peers;
  for (unsigned int i = 0; i < num_peers; i++)
    ctx->ops[i] = GNUNET_TESTBED_service_connect (ctx,
                                                  peers[i],
                                                  "dht",
                                                  &dht_connect_cb,
                                                  ctx,
                                                  &dht_connect_adapter,
                                                  &dht_disconnect_adapter,
                                                  ctx);
}
 
 
void
GNUNET_DHT_TEST_run (const char *testname,
                     const char *cfgname,
                     unsigned int num_peers,
                     GNUNET_DHT_TEST_AppMain tmain,
                     void *tmain_cls)
{
  struct GNUNET_DHT_TEST_Context *ctx;
 
  ctx = GNUNET_new (struct GNUNET_DHT_TEST_Context);
  ctx->num_peers = num_peers;
  ctx->ops = GNUNET_new_array (num_peers,
                               struct GNUNET_TESTBED_Operation *);
  ctx->dhts = GNUNET_new_array (num_peers,
                                struct GNUNET_DHT_Handle *);
  ctx->app_main = tmain;
  ctx->app_main_cls = tmain_cls;
  (void) GNUNET_TESTBED_test_run (testname,
                                  cfgname,
                                  num_peers,
                                  0LL, NULL, NULL,
                                  &dht_test_run, ctx);
}
 
 
/* end of dht_test_lib.c */