2 This file is part of GNUnet.
3 (C) 2009, 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 core/gnunet-service-core_neighbours.c
23 * @brief code for managing low-level 'plaintext' connections with transport (key exchange may or may not be done yet)
24 * @author Christian Grothoff
27 #include "gnunet_util_lib.h"
28 #include "gnunet_statistics_service.h"
29 #include "gnunet_transport_service.h"
30 #include "gnunet-service-core.h"
31 #include "gnunet-service-core_neighbours.h"
32 #include "gnunet-service-core_kx.h"
33 #include "gnunet-service-core_sessions.h"
34 #include "gnunet_constants.h"
38 * Message ready for transmission via transport service. This struct
39 * is followed by the actual content of the message.
41 struct NeighbourMessageEntry
45 * We keep messages in a doubly linked list.
47 struct NeighbourMessageEntry *next;
50 * We keep messages in a doubly linked list.
52 struct NeighbourMessageEntry *prev;
55 * By when are we supposed to transmit this message?
57 struct GNUNET_TIME_Absolute deadline;
60 * How long is the message? (number of bytes following the "struct
61 * MessageEntry", but not including the size of "struct
62 * MessageEntry" itself!)
70 * Data kept per transport-connected peer.
76 * Head of the batched message queue (already ordered, transmit
77 * starting with the head).
79 struct NeighbourMessageEntry *message_head;
82 * Tail of the batched message queue (already ordered, append new
85 struct NeighbourMessageEntry *message_tail;
88 * Handle for pending requests for transmission to this peer
89 * with the transport service. NULL if no request is pending.
91 struct GNUNET_TRANSPORT_TransmitHandle *th;
94 * Information about the key exchange with the other peer.
96 struct GSC_KeyExchangeInfo *kxinfo;
99 * Identity of the other peer.
101 struct GNUNET_PeerIdentity peer;
104 * ID of task used for re-trying plaintext scheduling.
106 GNUNET_SCHEDULER_TaskIdentifier retry_plaintext_task;
112 * Map of peer identities to 'struct Neighbour'.
114 static struct GNUNET_CONTAINER_MultiHashMap *neighbours;
119 static struct GNUNET_TRANSPORT_Handle *transport;
123 * Find the entry for the given neighbour.
125 * @param peer identity of the neighbour
126 * @return NULL if we are not connected, otherwise the
129 static struct Neighbour *
130 find_neighbour (const struct GNUNET_PeerIdentity *peer)
132 if (NULL == neighbours)
134 return GNUNET_CONTAINER_multihashmap_get (neighbours, &peer->hashPubKey);
139 * Free the given entry for the neighbour.
141 * @param n neighbour to free
144 free_neighbour (struct Neighbour *n)
146 struct NeighbourMessageEntry *m;
149 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
150 "Destroying neighbour entry for peer `%4s'\n",
151 GNUNET_i2s (&n->peer));
153 while (NULL != (m = n->message_head))
155 GNUNET_CONTAINER_DLL_remove (n->message_head, n->message_tail, m);
160 GNUNET_TRANSPORT_notify_transmit_ready_cancel (n->th);
163 GNUNET_STATISTICS_update (GSC_stats,
165 ("# sessions terminated by transport disconnect"),
167 GSC_SESSIONS_end (&n->peer);
168 if (NULL != n->kxinfo)
170 GSC_KX_stop (n->kxinfo);
173 if (n->retry_plaintext_task != GNUNET_SCHEDULER_NO_TASK)
175 GNUNET_SCHEDULER_cancel (n->retry_plaintext_task);
176 n->retry_plaintext_task = GNUNET_SCHEDULER_NO_TASK;
178 GNUNET_assert (GNUNET_OK ==
179 GNUNET_CONTAINER_multihashmap_remove (neighbours,
180 &n->peer.hashPubKey, n));
181 GNUNET_STATISTICS_set (GSC_stats,
182 gettext_noop ("# neighbour entries allocated"),
183 GNUNET_CONTAINER_multihashmap_size (neighbours),
190 * Check if we have encrypted messages for the specified neighbour
191 * pending, and if so, check with the transport about sending them
194 * @param n neighbour to check.
197 process_queue (struct Neighbour *n);
201 * Function called when the transport service is ready to receive a
202 * message for the respective peer
204 * @param cls neighbour to use message from
205 * @param size number of bytes we can transmit
206 * @param buf where to copy the message
207 * @return number of bytes transmitted
210 transmit_ready (void *cls, size_t size, void *buf)
212 struct Neighbour *n = cls;
213 struct NeighbourMessageEntry *m;
224 GNUNET_CONTAINER_DLL_remove (n->message_head, n->message_tail, m);
228 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
229 "Transmission of message of type %u and size %u failed\n",
231 ntohs (((struct GNUNET_MessageHeader *) &m[1])->type),
232 (unsigned int) m->size);
239 GNUNET_assert (size >= m->size);
240 memcpy (cbuf, &m[1], m->size);
243 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
244 "Copied message of type %u and size %u into transport buffer for `%4s'\n",
246 ntohs (((struct GNUNET_MessageHeader *) &m[1])->type),
247 (unsigned int) ret, GNUNET_i2s (&n->peer));
251 GNUNET_STATISTICS_update (GSC_stats,
253 ("# encrypted bytes given to transport"), ret,
260 * Check if we have messages for the specified neighbour pending, and
261 * if so, check with the transport about sending them out.
263 * @param n neighbour to check.
266 process_queue (struct Neighbour *n)
268 struct NeighbourMessageEntry *m;
271 return; /* request already pending */
275 /* notify sessions that the queue is empty and more messages
276 * could thus be queued now */
277 GSC_SESSIONS_solicit (&n->peer);
281 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
282 "Asking transport for transmission of %u bytes to `%4s' in next %llu ms\n",
283 (unsigned int) m->size, GNUNET_i2s (&n->peer),
285 GNUNET_TIME_absolute_get_remaining (m->deadline).rel_value);
288 GNUNET_TRANSPORT_notify_transmit_ready (transport, &n->peer, m->size, 0,
289 GNUNET_TIME_absolute_get_remaining
290 (m->deadline), &transmit_ready,
294 /* message request too large or duplicate request */
296 /* discard encrypted message */
297 GNUNET_CONTAINER_DLL_remove (n->message_head, n->message_tail, m);
305 * Function called by transport to notify us that
306 * a peer connected to us (on the network level).
309 * @param peer the peer that connected
310 * @param atsi performance data
311 * @param atsi_count number of entries in ats (excluding 0-termination)
314 handle_transport_notify_connect (void *cls,
315 const struct GNUNET_PeerIdentity *peer,
316 const struct GNUNET_ATS_Information *atsi,
321 if (0 == memcmp (peer, &GSC_my_identity, sizeof (struct GNUNET_PeerIdentity)))
326 n = find_neighbour (peer);
329 /* duplicate connect notification!? */
334 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received connection from `%4s'.\n",
337 n = GNUNET_malloc (sizeof (struct Neighbour));
339 GNUNET_assert (GNUNET_OK ==
340 GNUNET_CONTAINER_multihashmap_put (neighbours,
341 &n->peer.hashPubKey, n,
342 GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
343 GNUNET_STATISTICS_set (GSC_stats,
344 gettext_noop ("# neighbour entries allocated"),
345 GNUNET_CONTAINER_multihashmap_size (neighbours),
347 n->kxinfo = GSC_KX_start (peer);
352 * Function called by transport telling us that a peer
356 * @param peer the peer that disconnected
359 handle_transport_notify_disconnect (void *cls,
360 const struct GNUNET_PeerIdentity *peer)
365 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
366 "Peer `%4s' disconnected from us; received notification from transport.\n",
369 n = find_neighbour (peer);
380 * Function called by the transport for each received message.
383 * @param peer (claimed) identity of the other peer
384 * @param message the message
385 * @param atsi performance data
386 * @param atsi_count number of entries in ats (excluding 0-termination)
389 handle_transport_receive (void *cls, const struct GNUNET_PeerIdentity *peer,
390 const struct GNUNET_MessageHeader *message,
391 const struct GNUNET_ATS_Information *atsi,
398 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
399 "Received message of type %u from `%4s', demultiplexing.\n",
400 (unsigned int) ntohs (message->type), GNUNET_i2s (peer));
402 if (0 == memcmp (peer, &GSC_my_identity, sizeof (struct GNUNET_PeerIdentity)))
407 n = find_neighbour (peer);
410 /* received message from peer that is not connected!? */
414 type = ntohs (message->type);
417 case GNUNET_MESSAGE_TYPE_CORE_SET_KEY:
418 GSC_KX_handle_set_key (n->kxinfo, message);
420 case GNUNET_MESSAGE_TYPE_CORE_PING:
421 GSC_KX_handle_ping (n->kxinfo, message);
423 case GNUNET_MESSAGE_TYPE_CORE_PONG:
424 GSC_KX_handle_pong (n->kxinfo, message);
426 case GNUNET_MESSAGE_TYPE_CORE_ENCRYPTED_MESSAGE:
427 GSC_KX_handle_encrypted_message (n->kxinfo, message, atsi, atsi_count);
430 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
432 ("Unsupported message of type %u (%u bytes) received from peer `%s'\n"),
433 (unsigned int) type, (unsigned int) ntohs (message->size),
441 * Transmit the given message to the given target.
443 * @param target peer that should receive the message (must be connected)
444 * @param msg message to transmit
445 * @param timeout by when should the transmission be done?
448 GSC_NEIGHBOURS_transmit (const struct GNUNET_PeerIdentity *target,
449 const struct GNUNET_MessageHeader *msg,
450 struct GNUNET_TIME_Relative timeout)
452 struct NeighbourMessageEntry *me;
456 n = find_neighbour (target);
462 msize = ntohs (msg->size);
463 me = GNUNET_malloc (sizeof (struct NeighbourMessageEntry) + msize);
464 me->deadline = GNUNET_TIME_relative_to_absolute (timeout);
466 memcpy (&me[1], msg, msize);
467 GNUNET_CONTAINER_DLL_insert_tail (n->message_head, n->message_tail, me);
473 * Initialize neighbours subsystem.
476 GSC_NEIGHBOURS_init ()
478 neighbours = GNUNET_CONTAINER_multihashmap_create (128);
480 GNUNET_TRANSPORT_connect (GSC_cfg, &GSC_my_identity, NULL,
481 &handle_transport_receive,
482 &handle_transport_notify_connect,
483 &handle_transport_notify_disconnect);
484 if (NULL == transport)
486 GNUNET_CONTAINER_multihashmap_destroy (neighbours);
488 return GNUNET_SYSERR;
495 * Wrapper around 'free_neighbour'.
498 * @param key peer identity
499 * @param value the 'struct Neighbour' to free
500 * @return GNUNET_OK (continue to iterate)
503 free_neighbour_helper (void *cls, const GNUNET_HashCode * key, void *value)
505 struct Neighbour *n = value;
507 /* transport should have 'disconnected' all neighbours... */
515 * Shutdown neighbours subsystem.
518 GSC_NEIGHBOURS_done ()
520 if (NULL == transport)
522 GNUNET_TRANSPORT_disconnect (transport);
524 GNUNET_CONTAINER_multihashmap_iterate (neighbours, &free_neighbour_helper,
526 GNUNET_CONTAINER_multihashmap_destroy (neighbours);
530 /* end of gnunet-service-core_neighbours.c */