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_api_loop.h"
#include "testing_cmds.h"
 
#define LOG(kind, ...) GNUNET_log (kind, __VA_ARGS__)
 
struct BarrierReachedState
{
  struct GNUNET_TESTING_AsyncContext ac;
 
  const char *label;
 
  const char *barrier_name;
 
};
 
 
static void
barrier_reached_run (void *cls,
                     struct GNUNET_TESTING_Interpreter *is)
{
  struct BarrierReachedState *brs = cls;
  struct GNUNET_TESTING_Barrier *barrier;
 
  barrier = GNUNET_TESTING_get_barrier_ (is,
                                         brs->barrier_name);
  if (NULL == barrier)
  {
    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
                "No barrier `%s'\n",
                brs->barrier_name);
    GNUNET_TESTING_async_fail (&brs->ac);
    return;
  }
  if (barrier->satisfied)
  {
    GNUNET_TESTING_async_finish (&brs->ac);
    return;
  }
  GNUNET_array_append (barrier->waiting,
                       barrier->cnt_waiting,
                       &brs->ac);
  if (barrier->inherited)
  {
    struct GNUNET_TESTING_CommandBarrierReached cbr = {
      .header.size = htons (sizeof (cbr)),
      .header.type = htons (GNUNET_MESSAGE_TYPE_CMDS_HELPER_BARRIER_REACHED)
    };
 
    GNUNET_TESTING_barrier_name_hash_ (brs->barrier_name,
                                       &cbr.barrier_key);
    GNUNET_TESTING_loop_notify_parent_ (is,
                                        &cbr.header);
    return;
  }
  barrier->reached++;
  if (barrier->reached == barrier->expected_reaches)
  {
    struct GNUNET_TESTING_CommandBarrierSatisfied cbs = {
      .header.size = htons (sizeof (cbs)),
      .header.type = htons (GNUNET_MESSAGE_TYPE_CMDS_HELPER_BARRIER_CROSSABLE)
    };
 
    GNUNET_TESTING_barrier_name_hash_ (brs->barrier_name,
                                       &cbs.barrier_key);
    barrier->satisfied = true;
    GNUNET_TESTING_loop_notify_children_ (is,
                                          &cbs.header);
  }
  if (barrier->satisfied)
  {
    GNUNET_TESTING_async_finish (&brs->ac);
    for (unsigned int i = 0; i<barrier->cnt_waiting; i++)
      GNUNET_TESTING_async_finish (barrier->waiting[i]);
    GNUNET_array_grow (barrier->waiting,
                       barrier->cnt_waiting,
                       0);
    return;
  }
}
 
 
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 GNUNET_TESTING_Trait traits[] = {
    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)
{
  struct BarrierReachedState *brs;
 
  brs = GNUNET_new (struct BarrierReachedState);
  brs->label = label;
  brs->barrier_name = barrier_label;
  return GNUNET_TESTING_command_new_ac (
    brs,
    label,
    &barrier_reached_run,
    &barrier_reached_cleanup,
    &barrier_reached_traits,
    &brs->ac);
}