transport_api_blacklist.c

Go to the documentation of this file.

/*
     This file is part of GNUnet.
     Copyright (C) 2010-2014, 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"
#include "gnunet_arm_service.h"
#include "gnunet_hello_lib.h"
#include "gnunet_protocols.h"
#include "gnunet_transport_service.h"
#include "transport.h"

struct GNUNET_TRANSPORT_Blacklist {
  struct GNUNET_MQ_Handle *mq;

  const struct GNUNET_CONFIGURATION_Handle *cfg;

  GNUNET_TRANSPORT_BlacklistCallback cb;

  void *cb_cls;
};


static void
reconnect(struct GNUNET_TRANSPORT_Blacklist *br);


static void
handle_query(void *cls,
             const struct BlacklistMessage *bm)
{
  struct GNUNET_TRANSPORT_Blacklist *br = cls;
  struct GNUNET_MQ_Envelope *env;
  struct BlacklistMessage *res;

  GNUNET_break(0 == ntohl(bm->is_allowed));
  env = GNUNET_MQ_msg(res,
                      GNUNET_MESSAGE_TYPE_TRANSPORT_BLACKLIST_REPLY);
  res->is_allowed = htonl(br->cb(br->cb_cls,
                                 &bm->peer));
  res->peer = bm->peer;
  GNUNET_MQ_send(br->mq,
                 env);
}

static void
mq_error_handler(void *cls,
                 enum GNUNET_MQ_Error error)
{
  struct GNUNET_TRANSPORT_Blacklist *br = cls;

  reconnect(br);
}


static void
reconnect(struct GNUNET_TRANSPORT_Blacklist *br)
{
  struct GNUNET_MQ_MessageHandler handlers[] = {
    GNUNET_MQ_hd_fixed_size(query,
                            GNUNET_MESSAGE_TYPE_TRANSPORT_BLACKLIST_QUERY,
                            struct BlacklistMessage,
                            br),
    GNUNET_MQ_handler_end()
  };
  struct GNUNET_MQ_Envelope *env;
  struct GNUNET_MessageHeader *req;

  if (NULL != br->mq)
    GNUNET_MQ_destroy(br->mq);
  br->mq = GNUNET_CLIENT_connect(br->cfg,
                                 "transport",
                                 handlers,
                                 &mq_error_handler,
                                 br);
  if (NULL == br->mq)
    return;
  env = GNUNET_MQ_msg(req,
                      GNUNET_MESSAGE_TYPE_TRANSPORT_BLACKLIST_INIT);
  GNUNET_MQ_send(br->mq,
                 env);
}


struct GNUNET_TRANSPORT_Blacklist *
GNUNET_TRANSPORT_blacklist(const struct GNUNET_CONFIGURATION_Handle *cfg,
                           GNUNET_TRANSPORT_BlacklistCallback cb,
                           void *cb_cls)
{
  struct GNUNET_TRANSPORT_Blacklist *br;

  br = GNUNET_new(struct GNUNET_TRANSPORT_Blacklist);
  br->cfg = cfg;
  br->cb = cb;
  br->cb_cls = cb_cls;
  reconnect(br);
  if (NULL == br->mq)
    {
      GNUNET_free(br);
      return NULL;
    }
  return br;
}


void
GNUNET_TRANSPORT_blacklist_cancel(struct GNUNET_TRANSPORT_Blacklist *br)
{
  GNUNET_MQ_destroy(br->mq);
  GNUNET_free(br);
}


/* end of transport_api_blacklist.c */