/*
This file is part of GNUnet.
- (C) 2011-2015 Christian Grothoff (and other contributing authors)
+ Copyright (C) 2011-2015 Christian Grothoff (and other contributing authors)
GNUnet is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published
* @brief ats service, interaction with 'connecivity' API
* @author Matthias Wachs
* @author Christian Grothoff
- *
- * FIXME:
- * - we should track requests by client, and if a client
- * disconnects cancel all associated requests; right
- * now, they will persist forever unless the client
- * explicitly sends us a cancel before disconnecting!
*/
#include "platform.h"
#include "gnunet-service-ats.h"
#include "gnunet-service-ats_addresses.h"
#include "gnunet-service-ats_connectivity.h"
+#include "gnunet-service-ats_plugins.h"
#include "ats.h"
+
+/**
+ * Active connection requests.
+ */
+struct ConnectionRequest
+{
+ /**
+ * Client that made the request.
+ */
+ struct GNUNET_SERVER_Client *client;
+
+ /* TODO: allow client to express a 'strength' for this request */
+};
+
+
+/**
+ * Address suggestion requests by peer.
+ */
+static struct GNUNET_CONTAINER_MultiPeerMap *connection_requests;
+
+
+/**
+ * Is the given peer in the list of peers for which we
+ * have an address request?
+ *
+ * @param cls unused, NULL
+ * @param peer peer to query for
+ * @return #GNUNET_YES if so, #GNUNET_NO if not
+ */
+unsigned int
+GAS_connectivity_has_peer (void *cls,
+ const struct GNUNET_PeerIdentity *peer)
+{
+ if (NULL == connection_requests)
+ return 0;
+ /* TODO: return sum of 'strength's of connectivity requests */
+ return GNUNET_CONTAINER_multipeermap_contains (connection_requests,
+ peer);
+}
+
+
/**
* Handle 'request address' messages from clients.
*
{
const struct RequestAddressMessage *msg =
(const struct RequestAddressMessage *) message;
+ struct ConnectionRequest *cr;
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
"Received `%s' message\n",
"REQUEST_ADDRESS");
GNUNET_break (0 == ntohl (msg->reserved));
- GAS_addresses_request_address (&msg->peer);
+ cr = GNUNET_new (struct ConnectionRequest);
+ cr->client = client;
+ (void) GNUNET_CONTAINER_multipeermap_put (connection_requests,
+ &msg->peer,
+ cr,
+ GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
+ GAS_plugin_request_connect_start (&msg->peer);
GNUNET_SERVER_receive_done (client, GNUNET_OK);
}
/**
- * Handle 'request address' messages from clients.
+ * Free the connection request from the map if the
+ * closure matches the client.
+ *
+ * @param cls the client to match
+ * @param pid peer for which the request was made
+ * @param value the `struct ConnectionRequest`
+ * @return #GNUNET_OK (continue to iterate)
+ */
+static int
+free_matching_requests (void *cls,
+ const struct GNUNET_PeerIdentity *pid,
+ void *value)
+{
+ struct GNUNET_SERVER_Client *client = cls;
+ struct ConnectionRequest *cr = value;
+
+ if (cr->client == client)
+ {
+ GAS_plugin_request_connect_stop (pid);
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Removed request pending for peer `%s\n",
+ GNUNET_i2s (pid));
+ GNUNET_assert (GNUNET_YES ==
+ GNUNET_CONTAINER_multipeermap_remove (connection_requests,
+ pid,
+ cr));
+ GNUNET_free (cr);
+ }
+ return GNUNET_OK;
+}
+
+
+/**
+ * Handle 'request address cancel' messages from clients.
*
* @param cls unused, NULL
* @param client client that sent the request
(const struct RequestAddressMessage *) message;
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Received `%s' message\n",
- "REQUEST_ADDRESS_CANCEL");
+ "Received REQUEST_ADDRESS_CANCEL message for peer %s\n",
+ GNUNET_i2s (&msg->peer));
GNUNET_break (0 == ntohl (msg->reserved));
- GAS_addresses_request_address_cancel (&msg->peer);
+ GNUNET_CONTAINER_multipeermap_get_multiple (connection_requests,
+ &msg->peer,
+ &free_matching_requests,
+ client);
GNUNET_SERVER_receive_done (client, GNUNET_OK);
}
+
+/**
+ * Unregister a client (which may have been a connectivity client,
+ * but this is not assured).
+ *
+ * @param client handle of the (now dead) client
+ */
+void
+GAS_connectivity_remove_client (struct GNUNET_SERVER_Client *client)
+{
+ GNUNET_CONTAINER_multipeermap_iterate (connection_requests,
+ &free_matching_requests,
+ client);
+}
+
+
+/**
+ * Shutdown connectivity subsystem.
+ */
+void
+GAS_connectivity_init ()
+{
+ connection_requests = GNUNET_CONTAINER_multipeermap_create (32, GNUNET_NO);
+}
+
+
+/**
+ * Free the connection request from the map.
+ *
+ * @param cls NULL
+ * @param pid peer for which the request was made
+ * @param value the `struct ConnectionRequest`
+ * @return #GNUNET_OK (continue to iterate)
+ */
+static int
+free_request (void *cls,
+ const struct GNUNET_PeerIdentity *pid,
+ void *value)
+{
+ struct ConnectionRequest *cr = value;
+
+ free_matching_requests (cr->client,
+ pid,
+ cr);
+ return GNUNET_OK;
+}
+
+
+/**
+ * Shutdown connectivity subsystem.
+ */
+void
+GAS_connectivity_done ()
+{
+ GNUNET_CONTAINER_multipeermap_iterate (connection_requests,
+ &free_request,
+ NULL);
+ GNUNET_CONTAINER_multipeermap_destroy (connection_requests);
+ connection_requests = NULL;
+}
+
+
/* end of gnunet-service-ats_connectivity.c */
projects / oweals / gnunet.git / blobdiff