peerinfo_api_notify.c

Go to the documentation of this file.

/*
     This file is part of GNUnet.
     Copyright (C) 2001, 2002, 2004, 2005, 2007, 2009, 2010 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"
#include "gnunet_peerinfo_service.h"
#include "gnunet_protocols.h"
#include "peerinfo.h"

#define LOG(kind, ...) GNUNET_log_from(kind, "peerinfo-api", __VA_ARGS__)

struct GNUNET_PEERINFO_NotifyContext {
  struct GNUNET_MQ_Handle *mq;

  GNUNET_PEERINFO_Processor callback;

  void *callback_cls;

  const struct GNUNET_CONFIGURATION_Handle *cfg;

  struct GNUNET_SCHEDULER_Task *task;

  int include_friend_only;
};


static void
reconnect(void *cls);


static void
do_reconnect(struct GNUNET_PEERINFO_NotifyContext *nc)
{
  GNUNET_MQ_destroy(nc->mq);
  nc->mq = NULL;
  nc->task = GNUNET_SCHEDULER_add_now(&reconnect,
                                      nc);
}


static void
mq_error_handler(void *cls,
                 enum GNUNET_MQ_Error error)
{
  struct GNUNET_PEERINFO_NotifyContext *nc = cls;

  do_reconnect(nc);
}


static int
check_notification(void *cls,
                   const struct InfoMessage *im)
{
  uint16_t ms = ntohs(im->header.size) - sizeof(*im);

  if (ms >= sizeof(struct GNUNET_MessageHeader))
    {
      const struct GNUNET_HELLO_Message *hello;

      hello = (const struct GNUNET_HELLO_Message *)&im[1];
      if (ms != GNUNET_HELLO_size(hello))
        {
          GNUNET_break(0);
          return GNUNET_SYSERR;
        }
      return GNUNET_OK;
    }
  if (0 != ms)
    {
      GNUNET_break(0);
      return GNUNET_SYSERR;
    }
  return GNUNET_OK;  /* odd... */
}


static void
handle_notification(void *cls,
                    const struct InfoMessage *im)
{
  struct GNUNET_PEERINFO_NotifyContext *nc = cls;
  const struct GNUNET_HELLO_Message *hello;
  uint16_t ms = ntohs(im->header.size) - sizeof(struct InfoMessage);

  if (0 == ms)
    return;
  hello = (const struct GNUNET_HELLO_Message *)&im[1];
  LOG(GNUNET_ERROR_TYPE_DEBUG,
      "Received information about peer `%s' from peerinfo database\n",
      GNUNET_i2s(&im->peer));
  nc->callback(nc->callback_cls,
               &im->peer,
               hello,
               NULL);
}


static void
handle_end_iteration(void *cls,
                     const struct GNUNET_MessageHeader *msg)
{
  /* these are ignored by the notify API */
}


static void
reconnect(void *cls)
{
  struct GNUNET_PEERINFO_NotifyContext *nc = cls;
  struct GNUNET_MQ_MessageHandler handlers[] = {
    GNUNET_MQ_hd_var_size(notification,
                          GNUNET_MESSAGE_TYPE_PEERINFO_INFO,
                          struct InfoMessage,
                          nc),
    GNUNET_MQ_hd_fixed_size(end_iteration,
                            GNUNET_MESSAGE_TYPE_PEERINFO_INFO_END,
                            struct GNUNET_MessageHeader,
                            nc),
    GNUNET_MQ_handler_end()
  };
  struct GNUNET_MQ_Envelope *env;
  struct NotifyMessage *nm;

  nc->task = NULL;
  nc->mq = GNUNET_CLIENT_connect(nc->cfg,
                                 "peerinfo",
                                 handlers,
                                 &mq_error_handler,
                                 nc);
  if (NULL == nc->mq)
    return;
  env = GNUNET_MQ_msg(nm,
                      GNUNET_MESSAGE_TYPE_PEERINFO_NOTIFY);
  nm->include_friend_only = htonl(nc->include_friend_only);
  GNUNET_MQ_send(nc->mq,
                 env);
}


struct GNUNET_PEERINFO_NotifyContext *
GNUNET_PEERINFO_notify(const struct GNUNET_CONFIGURATION_Handle *cfg,
                       int include_friend_only,
                       GNUNET_PEERINFO_Processor callback,
                       void *callback_cls)
{
  struct GNUNET_PEERINFO_NotifyContext *nc;

  nc = GNUNET_new(struct GNUNET_PEERINFO_NotifyContext);
  nc->cfg = cfg;
  nc->callback = callback;
  nc->callback_cls = callback_cls;
  nc->include_friend_only = include_friend_only;
  reconnect(nc);
  if (NULL == nc->mq)
    {
      LOG(GNUNET_ERROR_TYPE_WARNING,
          "Could not connect to PEERINFO service.\n");
      GNUNET_free(nc);
      return NULL;
    }
  return nc;
}


void
GNUNET_PEERINFO_notify_cancel(struct GNUNET_PEERINFO_NotifyContext *nc)
{
  if (NULL != nc->mq)
    {
      GNUNET_MQ_destroy(nc->mq);
      nc->mq = NULL;
    }
  if (NULL != nc->task)
    {
      GNUNET_SCHEDULER_cancel(nc->task);
      nc->task = NULL;
    }
  GNUNET_free(nc);
}

/* end of peerinfo_api_notify.c */