testing_api_cmd_barrier_reached.c

Go to the documentation of this file.

/*
      This file is part of GNUnet
      Copyright (C) 2022 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 "gnunet_testing_lib.h"
#include "testing_cmds.h"
#include "gnunet_testing_plugin.h"
#include "gnunet_testing_barrier.h"
#include "gnunet_testing_netjail_lib.h"
#include "testing.h"
 
#define LOG(kind, ...) GNUNET_log (kind, __VA_ARGS__)
 
struct BarrierReachedState
{
  GNUNET_TESTING_cmd_helper_write_cb write_message;
 
  struct GNUNET_TESTING_AsyncContext ac;
 
  const char *label;
 
  const char *barrier_name;
 
  /*
   * The global numer of the node the cmd runs on.
   */
  unsigned int node_number;
 
  unsigned int asynchronous_finish;
 
  unsigned int running_on_master;
};
 
 
static void
barrier_reached_run (void *cls,
                     struct GNUNET_TESTING_Interpreter *is)
{
  struct BarrierReachedState *brs = cls;
  struct GNUNET_TESTING_Barrier *barrier;
  struct GNUNET_TESTING_Command *cmd =
    GNUNET_TESTING_interpreter_get_current_command (is);
  struct CommandListEntry *cle;
  size_t msg_length;
  struct GNUNET_TESTING_CommandBarrierReached *msg;
  size_t name_len;
 
  barrier = TST_interpreter_get_barrier (is, brs->barrier_name);
  if (NULL == barrier)
  {
    barrier = GNUNET_new (struct GNUNET_TESTING_Barrier);
    barrier->name = brs->barrier_name;
    TST_interpreter_add_barrier (is, barrier);
    LOG (GNUNET_ERROR_TYPE_DEBUG,
         "barrier %s added locally\n",
         brs->barrier_name);
  }
  barrier->reached++;
  if (GNUNET_TESTING_barrier_crossable (barrier))
  {
    GNUNET_assert (NULL != cmd);
    cmd->asynchronous_finish = GNUNET_YES;
    TST_interpreter_finish_attached_cmds (is, barrier->name);
  }
  else if (GNUNET_NO == brs->asynchronous_finish)
  {
    cle = GNUNET_new (struct CommandListEntry);
    cle->command = cmd;
    GNUNET_CONTAINER_DLL_insert (barrier->cmds_head,
                                 barrier->cmds_tail,
                                 cle);
    LOG (GNUNET_ERROR_TYPE_DEBUG,
         "added cle for %p %s\n",
         barrier,
         barrier->name);
  }
 
  if (GNUNET_NO == brs->running_on_master)
  {
    char *terminator = "\0";
 
    name_len = strlen (barrier->name);
    msg_length = sizeof(struct GNUNET_TESTING_CommandBarrierReached)
                 + name_len + 1;
    msg = GNUNET_malloc (msg_length);
    msg->header.size = htons ((uint16_t) msg_length);
    msg->header.type = htons (GNUNET_MESSAGE_TYPE_CMDS_HELPER_BARRIER_REACHED);
    msg->node_number = brs->node_number;
    memcpy (&msg[1], barrier->name, name_len + 1);
    memcpy (&msg[name_len + 1],terminator,1);
    brs->write_message ((struct GNUNET_MessageHeader *) msg, msg_length);
  }
}
 
 
static void
barrier_reached_cleanup (void *cls)
{
  struct BarrierReachedState *brs = cls;
 
  GNUNET_free (brs);
}
 
 
static enum GNUNET_GenericReturnValue
barrier_reached_traits (void *cls,
                        const void **ret,
                        const char *trait,
                        unsigned int index)
{
  struct BarrierReachedState *brs = cls;
  struct GNUNET_TESTING_AsyncContext *ac = &brs->ac;
 
  struct GNUNET_TESTING_Trait traits[] = {
    GNUNET_TESTING_make_trait_async_context ((const void *) ac),
    GNUNET_TESTING_trait_end ()
  };
 
  return GNUNET_TESTING_get_trait (traits,
                                   ret,
                                   trait,
                                   index);
}
 
 
struct GNUNET_TESTING_Command
GNUNET_TESTING_cmd_barrier_reached (
  const char *label,
  const char *barrier_label,
  unsigned int asynchronous_finish,
  unsigned int node_number,
  unsigned int running_on_master,
  GNUNET_TESTING_cmd_helper_write_cb write_message)
{
  struct BarrierReachedState *brs;
 
  brs = GNUNET_new (struct BarrierReachedState);
  brs->label = label;
  brs->barrier_name = barrier_label;
  brs->asynchronous_finish = asynchronous_finish;
  brs->node_number = node_number;
  brs->running_on_master = running_on_master;
  brs->write_message = write_message;
  return GNUNET_TESTING_command_new (brs, label,
                                     &barrier_reached_run,
                                     &barrier_reached_cleanup,
                                     &barrier_reached_traits,
                                     &brs->ac);
}