2 This file is part of GNUnet.
3 (C) 2010,2011 Christian Grothoff (and other contributing authors)
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
22 * @file transport/gnunet-service-transport_clients.c
23 * @brief plugin management API
24 * @author Christian Grothoff
27 #include "gnunet-service-transport_clients.h"
28 #include "gnunet-service-transport_hello.h"
29 #include "gnunet-service-transport_neighbours.h"
30 #include "gnunet-service-transport_plugins.h"
31 #include "gnunet-service-transport_validation.h"
32 #include "gnunet-service-transport.h"
33 #include "transport.h"
36 * How many messages can we have pending for a given client process
37 * before we start to drop incoming messages? We typically should
38 * have only one client and so this would be the primary buffer for
39 * messages, so the number should be chosen rather generously.
41 * The expectation here is that most of the time the queue is large
42 * enough so that a drop is virtually never required. Note that
43 * this value must be about as large as 'TOTAL_MSGS' in the
44 * 'test_transport_api_reliability.c', otherwise that testcase may
47 #define MAX_PENDING (128 * 1024)
51 * Linked list of messages to be transmitted to the client. Each
52 * entry is followed by the actual message.
54 struct ClientMessageQueueEntry
57 * This is a doubly-linked list.
59 struct ClientMessageQueueEntry *next;
62 * This is a doubly-linked list.
64 struct ClientMessageQueueEntry *prev;
69 * Client connected to the transport service.
71 struct TransportClient
75 * This is a doubly-linked list.
77 struct TransportClient *next;
80 * This is a doubly-linked list.
82 struct TransportClient *prev;
85 * Handle to the client.
87 struct GNUNET_SERVER_Client *client;
90 * Linked list of messages yet to be transmitted to
93 struct ClientMessageQueueEntry *message_queue_head;
96 * Tail of linked list of messages yet to be transmitted to the
99 struct ClientMessageQueueEntry *message_queue_tail;
102 * Current transmit request handle.
104 struct GNUNET_CONNECTION_TransmitHandle *th;
107 * Length of the list of messages pending for this client.
109 unsigned int message_count;
115 * Head of linked list of all clients to this service.
117 static struct TransportClient *clients_head;
120 * Tail of linked list of all clients to this service.
122 static struct TransportClient *clients_tail;
126 * Find the internal handle associated with the given client handle
128 * @param client server's client handle to look up
129 * @return internal client handle
131 static struct TransportClient *
132 lookup_client (struct GNUNET_SERVER_Client *client)
134 struct TransportClient *tc;
139 if (tc->client == client)
148 * Create the internal handle for the given server client handle
150 * @param client server's client handle to create our internal handle for
151 * @return fresh internal client handle
153 static struct TransportClient *
154 setup_client (struct GNUNET_SERVER_Client *client)
156 struct TransportClient *tc;
158 tc = GNUNET_malloc (sizeof (struct TransportClient));
160 GNUNET_CONTAINER_DLL_insert (clients_head,
168 * Function called to notify a client about the socket being ready to
169 * queue more data. "buf" will be NULL and "size" zero if the socket
170 * was closed for writing in the meantime.
173 * @param size number of bytes available in buf
174 * @param buf where the callee should write the message
175 * @return number of bytes written to buf
178 transmit_to_client_callback (void *cls,
182 struct TransportClient *tc = cls;
183 struct ClientMessageQueueEntry *q;
184 const struct GNUNET_MessageHeader *msg;
193 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
194 "Transmission to client failed, closing connection.\n");
200 while (NULL != (q = tc->message_queue_head))
202 msg = (const struct GNUNET_MessageHeader *) &q[1];
203 msize = ntohs (msg->size);
204 if (msize + tsize > size)
207 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
208 "Transmitting message of type %u to client.\n",
211 GNUNET_CONTAINER_DLL_remove (tc->message_queue_head,
212 tc->message_queue_tail,
215 memcpy (&cbuf[tsize],
223 GNUNET_assert (msize >= sizeof (struct GNUNET_MessageHeader));
224 tc->th = GNUNET_SERVER_notify_transmit_ready (tc->client,
226 GNUNET_TIME_UNIT_FOREVER_REL,
227 &transmit_to_client_callback,
229 GNUNET_assert (tc->th != NULL);
236 * Queue the given message for transmission to the given client
238 * @param client target of the message
239 * @param msg message to transmit
240 * @param may_drop GNUNET_YES if the message can be dropped
243 unicast (struct TransportClient *tc,
244 const struct GNUNET_MessageHeader *msg,
247 struct ClientMessageQueueEntry *q;
250 if ( (tc->message_count >= MAX_PENDING) &&
251 (GNUNET_YES == may_drop) )
253 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
254 _("Dropping message of type %u and size %u, have %u/%u messages pending\n"),
259 GNUNET_STATISTICS_update (GST_stats,
260 gettext_noop ("# messages dropped due to slow client"),
265 msize = ntohs (msg->size);
266 GNUNET_assert (msize >= sizeof (struct GNUNET_MessageHeader));
267 q = GNUNET_malloc (sizeof (struct ClientMessageQueueEntry) + msize);
268 memcpy (&q[1], msg, msize);
269 GNUNET_CONTAINER_DLL_insert_tail (tc->message_queue_head,
270 tc->message_queue_tail,
275 tc->th = GNUNET_SERVER_notify_transmit_ready (tc->client,
277 GNUNET_TIME_UNIT_FOREVER_REL,
278 &transmit_to_client_callback,
280 GNUNET_assert (tc->th != NULL);
285 * Called whenever a client is disconnected. Frees our
286 * resources associated with that client.
289 * @param client identification of the client
292 client_disconnect_notification (void *cls,
293 struct GNUNET_SERVER_Client *client)
295 struct TransportClient *tc;
296 struct ClientMessageQueueEntry *mqe;
300 tc = lookup_client (client);
304 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
305 "Client disconnected, cleaning up.\n");
307 while (NULL != (mqe = tc->message_queue_head))
309 GNUNET_CONTAINER_DLL_remove (tc->message_queue_head,
310 tc->message_queue_tail,
315 GNUNET_CONTAINER_DLL_remove (clients_head,
320 GNUNET_CONNECTION_notify_transmit_ready_cancel (tc->th);
323 GNUNET_break (0 == tc->message_count);
329 * Function called for each of our connected neighbours. Notify the
330 * client about the existing neighbour.
332 * @param cls the 'struct TransportClient' to notify
333 * @param peer identity of the neighbour
334 * @param ats performance data
335 * @param ats_count number of entries in ats (excluding 0-termination)
338 notify_client_about_neighbour (void *cls,
339 const struct GNUNET_PeerIdentity *peer,
340 const struct GNUNET_TRANSPORT_ATS_Information *ats,
343 struct TransportClient *tc = cls;
344 struct ConnectInfoMessage *cim;
347 size = sizeof (struct ConnectInfoMessage) + ats_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information);
348 GNUNET_assert (size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
349 cim = GNUNET_malloc (size);
350 cim->header.size = htons (size);
351 cim->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_CONNECT);
352 cim->ats_count = htonl(ats_count);
356 ats_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information));
357 unicast (tc, &cim->header, GNUNET_NO);
363 * Initialize a normal client. We got a start message from this
364 * client, add him to the list of clients for broadcasting of inbound
368 * @param client the client
369 * @param message the start message that was sent
372 clients_handle_start (void *cls,
373 struct GNUNET_SERVER_Client *client,
374 const struct GNUNET_MessageHeader *message)
376 const struct StartMessage *start;
377 struct TransportClient *tc;
379 tc = lookup_client (client);
382 /* got 'start' twice from the same client, not allowed */
384 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
387 start = (const struct StartMessage*) message;
388 if ( (GNUNET_NO != ntohl (start->do_check)) &&
389 (0 != memcmp (&start->self,
391 sizeof (struct GNUNET_PeerIdentity))) )
393 /* client thinks this is a different peer, reject */
394 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
395 _("Rejecting control connection from peer `%s', which is not me!\n"),
396 GNUNET_i2s (&start->self));
397 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
400 tc = setup_client (client);
401 unicast (tc, GST_hello_get(), GNUNET_NO);
402 GST_neighbours_iterate (¬ify_client_about_neighbour, tc);
403 GNUNET_SERVER_receive_done (client, GNUNET_OK);
408 * Client sent us a HELLO. Process the request.
411 * @param client the client
412 * @param message the HELLO message
415 clients_handle_hello (void *cls,
416 struct GNUNET_SERVER_Client *client,
417 const struct GNUNET_MessageHeader *message)
419 GST_validation_handle_hello (message);
420 GNUNET_SERVER_receive_done (client, GNUNET_OK);
425 * Closure for 'handle_send_transmit_continuation'
427 struct SendTransmitContinuationContext
430 * Client that made the request.
432 struct GNUNET_SERVER_Client *client;
435 * Peer that was the target.
437 struct GNUNET_PeerIdentity target;
442 * Function called after the transmission is done. Notify the client that it is
443 * OK to send the next message.
446 * @param success GNUNET_OK on success, GNUNET_NO on failure, GNUNET_SYSERR if we're not connected
449 handle_send_transmit_continuation (void *cls,
452 struct SendTransmitContinuationContext *stcc = cls;
453 struct SendOkMessage send_ok_msg;
455 send_ok_msg.header.size = htons (sizeof (send_ok_msg));
456 send_ok_msg.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_SEND_OK);
457 send_ok_msg.success = htonl (success);
458 send_ok_msg.latency = GNUNET_TIME_relative_hton (GNUNET_TIME_UNIT_FOREVER_REL);
459 send_ok_msg.peer = stcc->target;
460 GST_clients_unicast (stcc->client, &send_ok_msg.header, GNUNET_NO);
461 GNUNET_SERVER_client_drop (stcc->client);
467 * Client asked for transmission to a peer. Process the request.
470 * @param client the client
471 * @param message the send message that was sent
474 clients_handle_send (void *cls,
475 struct GNUNET_SERVER_Client *client,
476 const struct GNUNET_MessageHeader *message)
478 const struct OutboundMessage *obm;
479 const struct GNUNET_MessageHeader *obmm;
480 struct SendTransmitContinuationContext *stcc;
484 size = ntohs (message->size);
485 if (size < sizeof (struct OutboundMessage) + sizeof (struct GNUNET_MessageHeader))
488 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
491 obm = (const struct OutboundMessage *) message;
492 obmm = (const struct GNUNET_MessageHeader *) &obm[1];
493 msize = size - sizeof (struct OutboundMessage);
494 if (msize < sizeof (struct GNUNET_MessageHeader))
497 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
500 GNUNET_STATISTICS_update (GST_stats,
501 gettext_noop ("# bytes payload received for other peers"),
505 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
506 "Received `%s' request from client with target `%4s' and first message of type %u and total size %u\n",
508 GNUNET_i2s (&obm->peer),
513 GST_neighbours_test_connected (&obm->peer))
515 /* not connected, not allowed to send; can happen due to asynchronous operations */
516 GNUNET_STATISTICS_update (GST_stats,
517 gettext_noop ("# bytes payload dropped (other peer was not connected)"),
520 GNUNET_SERVER_receive_done (client, GNUNET_OK);
523 GNUNET_SERVER_receive_done (client, GNUNET_OK);
524 stcc = GNUNET_malloc (sizeof (struct SendTransmitContinuationContext));
525 stcc->target = obm->peer;
526 stcc->client = client;
527 GNUNET_SERVER_client_keep (client);
528 GST_neighbours_send (&obm->peer,
530 GNUNET_TIME_relative_ntoh (obm->timeout),
531 &handle_send_transmit_continuation,
537 * Client asked for a quota change for a particular peer. Process the request.
540 * @param client the client
541 * @param message the quota changing message
544 clients_handle_set_quota (void *cls,
545 struct GNUNET_SERVER_Client *client,
546 const struct GNUNET_MessageHeader *message)
548 const struct QuotaSetMessage *qsm;
550 qsm = (const struct QuotaSetMessage *) message;
551 GNUNET_STATISTICS_update (GST_stats,
552 gettext_noop ("# SET QUOTA messages received"),
556 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
557 "Received `%s' request (new quota %u) from client for peer `%4s'\n",
559 (unsigned int) ntohl (qsm->quota.value__),
560 GNUNET_i2s (&qsm->peer));
562 GST_neighbours_set_incoming_quota (&qsm->peer,
564 GNUNET_SERVER_receive_done (client, GNUNET_OK);
569 * Take the given address and append it to the set of results sent back to
572 * @param cls the transmission context used ('struct GNUNET_SERVER_TransmitContext*')
573 * @param address the resolved name, NULL to indicate the last response
576 transmit_address_to_client (void *cls,
579 struct GNUNET_SERVER_TransmitContext *tc = cls;
583 GNUNET_SERVER_transmit_context_append_data (tc, NULL, 0,
584 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_REPLY);
585 GNUNET_SERVER_transmit_context_run (tc,
586 GNUNET_TIME_UNIT_FOREVER_REL);
589 GNUNET_SERVER_transmit_context_append_data (tc,
590 address, strlen (address) + 1,
591 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_REPLY);
596 * Client asked to resolve an address. Process the request.
599 * @param client the client
600 * @param message the resolution request
603 clients_handle_address_lookup (void *cls,
604 struct GNUNET_SERVER_Client *client,
605 const struct GNUNET_MessageHeader *message)
607 const struct AddressLookupMessage *alum;
608 struct GNUNET_TRANSPORT_PluginFunctions *papi;
609 const char *plugin_name;
611 uint32_t address_len;
613 struct GNUNET_SERVER_TransmitContext *tc;
614 struct GNUNET_TIME_Relative rtimeout;
617 size = ntohs (message->size);
618 if (size < sizeof (struct AddressLookupMessage))
621 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
624 alum = (const struct AddressLookupMessage *) message;
625 address_len = ntohl (alum->addrlen);
626 if (size <= sizeof (struct AddressLookupMessage) + address_len)
629 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
632 address = (const char *) &alum[1];
633 plugin_name = (const char *) &address[address_len];
635 [size - sizeof (struct AddressLookupMessage) - address_len - 1] != '\0')
638 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
641 rtimeout = GNUNET_TIME_relative_ntoh (alum->timeout);
642 numeric = ntohl (alum->numeric_only);
643 tc = GNUNET_SERVER_transmit_context_create (client);
644 papi = GST_plugins_find (plugin_name);
647 GNUNET_SERVER_transmit_context_append_data (tc, NULL, 0,
648 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_REPLY);
649 GNUNET_SERVER_transmit_context_run (tc, rtimeout);
652 GNUNET_SERVER_disable_receive_done_warning (client);
653 papi->address_pretty_printer (papi->cls,
655 address, address_len,
658 &transmit_address_to_client, tc);
663 * Send an address to the client.
665 * @param cls our 'struct GNUNET_SERVER_TransmitContext' (for sending)
666 * @param public_key public key for the peer, never NULL
667 * @param target peer this change is about, never NULL
668 * @param valid_until until what time do we consider the address valid?
669 * @param validation_block is FOREVER if the address is for an unsupported plugin (from PEERINFO)
670 * is ZERO if the address is considered valid (no validation needed)
671 * is a time in the future if we're currently denying re-validation
672 * @param plugin_name name of the plugin
673 * @param plugin_address binary address
674 * @param plugin_address_len length of address
677 send_address_to_client (void *cls,
678 const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *public_key,
679 const struct GNUNET_PeerIdentity *target,
680 struct GNUNET_TIME_Absolute valid_until,
681 struct GNUNET_TIME_Absolute validation_block,
682 const char *plugin_name,
683 const void *plugin_address,
684 size_t plugin_address_len)
686 struct GNUNET_SERVER_TransmitContext *tc = cls;
689 /* FIXME: move to a binary format!!! */
690 GNUNET_asprintf (&addr_buf, "%s --- %s, %s",
691 GST_plugins_a2s (plugin_name,
694 (GNUNET_YES == GST_neighbours_test_connected (target))
697 (GNUNET_TIME_absolute_get_remaining (valid_until).rel_value > 0)
700 transmit_address_to_client (tc, addr_buf);
701 GNUNET_free (addr_buf);
706 * Client asked to obtain information about a peer's addresses.
707 * Process the request.
708 * FIXME: use better name!
711 * @param client the client
712 * @param message the peer address information request
715 clients_handle_peer_address_lookup (void *cls,
716 struct GNUNET_SERVER_Client *client,
717 const struct GNUNET_MessageHeader *message)
719 const struct PeerAddressLookupMessage *peer_address_lookup;
720 struct GNUNET_SERVER_TransmitContext *tc;
722 peer_address_lookup = (const struct PeerAddressLookupMessage *) message;
723 GNUNET_break (ntohl (peer_address_lookup->reserved) == 0);
724 tc = GNUNET_SERVER_transmit_context_create (client);
725 GST_validation_get_addresses (&peer_address_lookup->peer,
726 &send_address_to_client,
728 GNUNET_SERVER_transmit_context_append_data (tc,
730 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_REPLY);
731 GNUNET_SERVER_transmit_context_run (tc,
732 GNUNET_TIME_UNIT_FOREVER_REL);
737 * Output the active address of connected neighbours to the given client.
739 * @param cls the 'struct GNUNET_SERVER_TransmitContext' for transmission to the client
740 * @param neighbour identity of the neighbour
741 * @param ats performance data
742 * @param ats_count number of entries in ats (excluding 0-termination)
745 output_addresses (void *cls,
746 const struct GNUNET_PeerIdentity *neighbour,
747 const struct GNUNET_TRANSPORT_ATS_Information *ats,
750 struct GNUNET_SERVER_TransmitContext *tc = cls;
753 /* FIXME: move to a binary format!!! */
754 GNUNET_asprintf (&addr_buf,
756 GNUNET_i2s(neighbour),
757 GST_plugins_a2s ("FIXME", NULL, 0));
758 transmit_address_to_client (tc, addr_buf);
759 GNUNET_free (addr_buf);
764 * Client asked to obtain information about all actively used addresses.
765 * Process the request. FIXME: use better name!
768 * @param client the client
769 * @param message the peer address information request
772 clients_handle_address_iterate (void *cls,
773 struct GNUNET_SERVER_Client *client,
774 const struct GNUNET_MessageHeader *message)
776 struct GNUNET_SERVER_TransmitContext *tc;
778 GNUNET_SERVER_disable_receive_done_warning (client);
779 tc = GNUNET_SERVER_transmit_context_create (client);
780 GST_neighbours_iterate (&output_addresses,
782 GNUNET_SERVER_transmit_context_append_data (tc, NULL, 0,
783 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_REPLY);
784 GNUNET_SERVER_transmit_context_run (tc, GNUNET_TIME_UNIT_FOREVER_REL);
789 * Start handling requests from clients.
791 * @param server server used to accept clients from.
794 GST_clients_start (struct GNUNET_SERVER_Handle *server)
796 static const struct GNUNET_SERVER_MessageHandler handlers[] = {
797 { &clients_handle_start, NULL, sizeof (struct StartMessage)},
798 { &clients_handle_hello, NULL, 0},
799 { &clients_handle_send, NULL, 0},
800 { &clients_handle_set_quota, NULL, sizeof (struct QuotaSetMessage)},
801 { &clients_handle_address_lookup, NULL, 0},
802 { &clients_handle_peer_address_lookup, NULL, sizeof (struct PeerAddressLookupMessage)},
803 { &clients_handle_address_iterate, NULL, sizeof (struct GNUNET_MessageHeader)},
806 GNUNET_SERVER_add_handlers (server, handlers);
807 GNUNET_SERVER_disconnect_notify (server,
808 &client_disconnect_notification, NULL);
813 * Stop processing clients.
823 * Broadcast the given message to all of our clients.
825 * @param msg message to broadcast
826 * @param may_drop GNUNET_YES if the message can be dropped
829 GST_clients_broadcast (const struct GNUNET_MessageHeader *msg,
832 struct TransportClient *tc;
834 for (tc = clients_head; tc != NULL; tc = tc->next)
835 unicast (tc, msg, may_drop);
840 * Send the given message to a particular client
842 * @param client target of the message
843 * @param msg message to transmit
844 * @param may_drop GNUNET_YES if the message can be dropped
847 GST_clients_unicast (struct GNUNET_SERVER_Client *client,
848 const struct GNUNET_MessageHeader *msg,
851 struct TransportClient *tc;
853 tc = lookup_client (client);
855 tc = setup_client (client);
856 unicast (tc, msg, may_drop);
860 /* end of file gnunet-service-transport_clients.c */