nc.c

Go to the documentation of this file.

/*
     This file is part of GNUnet.
     Copyright (C) 2010, 2016 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_util_lib.h"
 
#define LOG(kind, ...) GNUNET_log_from (kind, "util-nc", __VA_ARGS__)
 
 
struct SubscriberList
{
  struct SubscriberList *next;
 
  struct SubscriberList *prev;
 
  struct GNUNET_NotificationContext *nc;
 
  struct GNUNET_MQ_DestroyNotificationHandle *mq_nh;
 
  struct GNUNET_MQ_Handle *mq;
};
 
 
struct GNUNET_NotificationContext
{
  struct SubscriberList *subscribers_head;
 
  struct SubscriberList *subscribers_tail;
 
  unsigned int queue_length;
};
 
 
static void
handle_mq_destroy (void *cls)
{
  struct SubscriberList *pos = cls;
  struct GNUNET_NotificationContext *nc = pos->nc;
 
  GNUNET_CONTAINER_DLL_remove (nc->subscribers_head,
                               nc->subscribers_tail,
                               pos);
  GNUNET_free (pos);
}
 
 
struct GNUNET_NotificationContext *
GNUNET_notification_context_create (unsigned int queue_length)
{
  struct GNUNET_NotificationContext *nc;
 
  nc = GNUNET_new (struct GNUNET_NotificationContext);
  nc->queue_length = queue_length;
  return nc;
}
 
 
void
GNUNET_notification_context_destroy (struct GNUNET_NotificationContext *nc)
{
  struct SubscriberList *pos;
 
  while (NULL != (pos = nc->subscribers_head))
  {
    GNUNET_CONTAINER_DLL_remove (nc->subscribers_head,
                                 nc->subscribers_tail,
                                 pos);
    GNUNET_MQ_destroy_notify_cancel (pos->mq_nh);
    GNUNET_free (pos);
  }
  GNUNET_free (nc);
}
 
 
void
GNUNET_notification_context_add (struct GNUNET_NotificationContext *nc,
                                 struct GNUNET_MQ_Handle *mq)
{
  struct SubscriberList *cl;
 
  for (cl = nc->subscribers_head; NULL != cl; cl = cl->next)
    if (cl->mq == mq)
      return;
  /* already present */
  cl = GNUNET_new (struct SubscriberList);
  GNUNET_CONTAINER_DLL_insert (nc->subscribers_head,
                               nc->subscribers_tail,
                               cl);
  cl->nc = nc;
  cl->mq = mq;
  cl->mq_nh = GNUNET_MQ_destroy_notify (cl->mq,
                                        &handle_mq_destroy,
                                        cl);
}
 
 
void
GNUNET_notification_context_broadcast (struct GNUNET_NotificationContext *nc,
                                       const struct GNUNET_MessageHeader *msg,
                                       int can_drop)
{
  struct SubscriberList *pos;
  struct GNUNET_MQ_Envelope *env;
 
  for (pos = nc->subscribers_head; NULL != pos; pos = pos->next)
  {
    if ((GNUNET_YES == can_drop) &&
        (GNUNET_MQ_get_length (pos->mq) > nc->queue_length))
      continue;
    env = GNUNET_MQ_msg_copy (msg);
    GNUNET_MQ_send (pos->mq,
                    env);
  }
}
 
 
unsigned int
GNUNET_notification_context_get_size (struct GNUNET_NotificationContext *nc)
{
  unsigned int num;
  struct SubscriberList *pos;
 
  num = 0;
  for (pos = nc->subscribers_head; NULL != pos; pos = pos->next)
    num++;
  return num;
}
 
 
/* end of nc.c */