4479c0b0835808f5dd1af14a873f0dbc1434f1b5
[oweals/gnunet.git] / src / dv / gnunet-service-dv.c
1 /*
2      This file is part of GNUnet.
3      (C) 2013 Christian Grothoff (and other contributing authors)
4
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.
9
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.
14
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.
19 */
20
21 /**
22  * @file dv/gnunet-service-dv.c
23  * @brief the distance vector service, primarily handles gossip of nearby
24  * peers and sending/receiving DV messages from core and decapsulating
25  * them
26  *
27  * @author Christian Grothoff
28  * @author Nathan Evans
29  */
30 #include "platform.h"
31 #include "gnunet_util_lib.h"
32 #include "gnunet_protocols.h"
33 #include "gnunet_core_service.h"
34 #include "gnunet_hello_lib.h"
35 #include "gnunet_peerinfo_service.h"
36 #include "gnunet_statistics_service.h"
37 #include "gnunet_consensus_service.h"
38 #include "gnunet_ats_service.h"
39 #include "dv.h"
40 #include <gcrypt.h>
41
42
43 /**
44  * How often do we establish the consensu?
45  */
46 #define GNUNET_DV_CONSENSUS_FREQUENCY GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MINUTES, 5)
47
48 /**
49  * Maximum number of messages we queue per peer.
50  */
51 #define MAX_QUEUE_SIZE 16
52
53 /**
54  * Maximum number of messages we queue towards the clients/plugin.
55  */
56 #define MAX_QUEUE_SIZE_PLUGIN 1024
57
58 /**
59  * The default fisheye depth, from how many hops away will
60  * we keep peers?
61  */
62 #define DEFAULT_FISHEYE_DEPTH 3
63
64 /**
65  * How many hops is a direct neighbor away?
66  */
67 #define DIRECT_NEIGHBOR_COST 1
68
69
70 GNUNET_NETWORK_STRUCT_BEGIN
71
72 /**
73  * Information about a peer DV can route to.  These entries are what
74  * we use as the binary format to establish consensus to create our
75  * routing table and as the address format in the HELLOs.
76  */
77 struct Target
78 {
79
80   /**
81    * Identity of the peer we can reach.
82    */
83   struct GNUNET_PeerIdentity peer;
84
85   /**
86    * How many hops (1-3) is this peer away? in network byte order
87    */
88   uint32_t distance GNUNET_PACKED;
89
90 };
91
92
93 /**
94  * Message exchanged between DV services (via core), requesting a
95  * message to be routed.  
96  */
97 struct RouteMessage
98 {
99   /**
100    * Type: GNUNET_MESSAGE_TYPE_DV_ROUTE
101    */
102   struct GNUNET_MessageHeader header;
103
104   /**
105    * Expected (remaining) distance.  Must be always smaller than
106    * DEFAULT_FISHEYE_DEPTH, should be zero at the target.  Must
107    * be decremented by one at each hop.  Peers must not forward
108    * these messages further once the counter has reached zero.
109    */
110   uint32_t distance GNUNET_PACKED;
111
112   /**
113    * The (actual) target of the message (this peer, if distance is zero).
114    */
115   struct GNUNET_PeerIdentity target;
116
117   /**
118    * The (actual) sender of the message.
119    */
120   struct GNUNET_PeerIdentity sender;
121
122 };
123
124 GNUNET_NETWORK_STRUCT_END
125
126
127 /**
128  * Linked list of messages to send to clients.
129  */
130 struct PendingMessage
131 {
132   /**
133    * Pointer to next item in the list
134    */
135   struct PendingMessage *next;
136
137   /**
138    * Pointer to previous item in the list
139    */
140   struct PendingMessage *prev;
141
142   /**
143    * Actual message to be sent, allocated after this struct.
144    */
145   const struct GNUNET_MessageHeader *msg;
146
147   /**
148    * Ultimate target for the message.
149    */
150   struct GNUNET_PeerIdentity ultimate_target;
151
152   /**
153    * Unique ID of the message.
154    */
155   uint32_t uid;
156
157 };
158
159
160 /**
161  * Information about a direct neighbor (core-level, excluding
162  * DV-links, only DV-enabled peers).
163  */
164 struct DirectNeighbor
165 {
166
167   /**
168    * Identity of the peer.
169    */
170   struct GNUNET_PeerIdentity peer;
171   
172   /**
173    * Head of linked list of messages to send to this peer.
174    */
175   struct PendingMessage *pm_head;
176
177   /**
178    * Tail of linked list of messages to send to this peer.
179    */
180   struct PendingMessage *pm_tail;
181
182   /**
183    * Transmit handle to core service.
184    */
185   struct GNUNET_CORE_TransmitHandle *cth;
186
187   /**
188    * Routing table of the neighbor, NULL if not yet established.
189    * Keys are peer identities, values are 'struct Target' entries.
190    * Note that the distances in the targets are from the point-of-view
191    * of the peer, not from us!
192    */ 
193   struct GNUNET_CONTAINER_MultiHashMap *neighbor_table;
194
195   /**
196    * Updated routing table of the neighbor, under construction,
197    * NULL if we are not currently building it.
198    * Keys are peer identities, values are 'struct Target' entries.
199    * Note that the distances in the targets are from the point-of-view
200    * of the peer, not from us!
201    */ 
202   struct GNUNET_CONTAINER_MultiHashMap *neighbor_table_consensus;
203
204   /**
205    * Active consensus, if we are currently synchronizing the
206    * routing tables.
207    */
208   struct GNUNET_CONSENSUS_Handle *consensus;
209
210   /**
211    * ID of the task we use to (periodically) update our consensus
212    * with this peer.
213    */
214   GNUNET_SCHEDULER_TaskIdentifier consensus_task;
215
216   /**
217    * At what offset are we, with respect to inserting our own routes
218    * into the consensus?
219    */
220   unsigned int consensus_insertion_offset;
221
222   /**
223    * At what distance are we, with respect to inserting our own routes
224    * into the consensus?
225    */
226   unsigned int consensus_insertion_distance;
227
228   /**
229    * Number of messages currently in the 'pm_XXXX'-DLL.
230    */
231   unsigned int pm_queue_size;
232
233   /**
234    * Flag set within 'check_target_removed' to trigger full global route refresh.
235    */
236   int target_removed;
237
238   /**
239    * Our distance to this peer, 0 for unknown.
240    */
241   uint32_t distance;
242
243   /**
244    * Is this neighbor connected at the core level?
245    */
246   int connected;
247
248 };
249
250
251 /**
252  * A route includes information about the next hop,
253  * the target, and the ultimate distance to the
254  * target.
255  */
256 struct Route
257 {
258
259   /**
260    * Which peer do we need to forward the message to?
261    */
262   struct DirectNeighbor *next_hop;
263
264   /**
265    * What would be the target, and how far is it away?
266    */
267   struct Target target;
268
269   /**
270    * Offset of this target in the respective consensus set.
271    */
272   unsigned int set_offset;
273
274 };
275
276
277 /**
278  * Set of targets we bring to a consensus; all targets in a set have a
279  * distance equal to the sets distance (which is implied by the array
280  * index of the set).
281  */
282 struct ConsensusSet
283 {
284
285   /**
286    * Array of targets in the set, may include NULL entries if a
287    * neighbor has disconnected; the targets are allocated with the
288    * respective container (all_routes), not here.
289    */
290   struct Route **targets;
291
292   /**
293    * Size of the 'targets' array.
294    */
295   unsigned int array_length;
296
297 };
298
299
300 /**
301  * Hashmap of all of our neighbors; processing these usually requires
302  * first checking to see if the peer is core-connected and if the 
303  * distance is 1, in which case they are direct neighbors.
304  */
305 static struct GNUNET_CONTAINER_MultiHashMap *direct_neighbors;
306
307 /**
308  * Hashmap with all routes that we currently support; contains 
309  * routing information for all peers from distance 2
310  * up to distance DEFAULT_FISHEYE_DEPTH.
311  */
312 static struct GNUNET_CONTAINER_MultiHashMap *all_routes;
313
314 /**
315  * Array of consensus sets we expose to the outside world.  Sets
316  * are structured by the distance to the target.
317  */
318 static struct ConsensusSet consensi[DEFAULT_FISHEYE_DEPTH - 1];
319
320 /**
321  * Handle to the core service api.
322  */
323 static struct GNUNET_CORE_Handle *core_api;
324
325 /**
326  * The identity of our peer.
327  */
328 static struct GNUNET_PeerIdentity my_identity;
329
330 /**
331  * The configuration for this service.
332  */
333 static const struct GNUNET_CONFIGURATION_Handle *cfg;
334
335 /**
336  * The client, the DV plugin connected to us (or an event monitor).
337  * Hopefully this client will never change, although if the plugin
338  * dies and returns for some reason it may happen.
339  */
340 static struct GNUNET_SERVER_NotificationContext *nc;
341
342 /**
343  * Handle for the statistics service.
344  */
345 static struct GNUNET_STATISTICS_Handle *stats;
346
347 /**
348  * Handle to ATS service.
349  */
350 static struct GNUNET_ATS_PerformanceHandle *ats;
351  
352
353 /**
354  * Start creating a new consensus from scratch.
355  *
356  * @param cls the 'struct DirectNeighbor' of the peer we're building
357  *        a routing consensus with
358  * @param tc scheduler context
359  */    
360 static void
361 start_consensus (void *cls,
362                  const struct GNUNET_SCHEDULER_TaskContext *tc);
363
364
365 /**
366  * Forward a message from another peer to the plugin.
367  *
368  * @param message the message to send to the plugin
369  * @param origin the original sender of the message
370  * @param distance distance to the original sender of the message
371  */
372 static void
373 send_data_to_plugin (const struct GNUNET_MessageHeader *message, 
374                      const struct GNUNET_PeerIdentity *origin,
375                      uint32_t distance)
376 {
377   struct GNUNET_DV_ReceivedMessage *received_msg;
378   size_t size;
379
380   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
381               "Delivering message from peer `%s'\n",
382               GNUNET_i2s (origin));
383   size = sizeof (struct GNUNET_DV_ReceivedMessage) + 
384     ntohs (message->size);
385   if (size >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
386   {    
387     GNUNET_break (0); /* too big */
388     return;
389   }
390   received_msg = GNUNET_malloc (size);
391   received_msg->header.size = htons (size);
392   received_msg->header.type = htons (GNUNET_MESSAGE_TYPE_DV_RECV);
393   received_msg->distance = htonl (distance);
394   received_msg->sender = *origin;
395   memcpy (&received_msg[1], message, ntohs (message->size));
396   GNUNET_SERVER_notification_context_broadcast (nc, 
397                                                 &received_msg->header,
398                                                 GNUNET_YES);
399   GNUNET_free (received_msg);
400 }
401
402
403 /**
404  * Forward a control message to the plugin.
405  *
406  * @param message the message to send to the plugin
407  */
408 static void
409 send_control_to_plugin (const struct GNUNET_MessageHeader *message)
410 {
411   GNUNET_SERVER_notification_context_broadcast (nc, 
412                                                 message,
413                                                 GNUNET_NO);
414 }
415
416
417 /**
418  * Give an (N)ACK message to the plugin, we transmitted a message for it.
419  *
420  * @param target peer that received the message
421  * @param uid plugin-chosen UID for the message
422  * @param nack GNUNET_NO to send ACK, GNUNET_YES to send NACK
423  */
424 static void
425 send_ack_to_plugin (const struct GNUNET_PeerIdentity *target, 
426                     uint32_t uid,
427                     int nack)
428 {
429   struct GNUNET_DV_AckMessage ack_msg;
430
431   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
432               "Delivering ACK for message to peer `%s'\n",
433               GNUNET_i2s (target));
434   ack_msg.header.size = htons (sizeof (ack_msg));
435   ack_msg.header.type = htons ((GNUNET_YES == nack) 
436                                ? GNUNET_MESSAGE_TYPE_DV_SEND_NACK
437                                : GNUNET_MESSAGE_TYPE_DV_SEND_ACK);
438   ack_msg.uid = htonl (uid);
439   ack_msg.target = *target;
440   send_control_to_plugin (&ack_msg.header);
441 }
442
443
444 /**
445  * Send a DISTANCE_CHANGED message to the plugin.
446  *
447  * @param peer peer with a changed distance
448  * @param distance new distance to the peer
449  */
450 static void
451 send_distance_change_to_plugin (const struct GNUNET_PeerIdentity *peer, 
452                                 uint32_t distance)
453 {
454   struct GNUNET_DV_DistanceUpdateMessage du_msg;
455
456   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
457               "Delivering DISTANCE_CHANGED for message about peer `%s'\n",
458               GNUNET_i2s (peer));
459   du_msg.header.size = htons (sizeof (du_msg));
460   du_msg.header.type = htons (GNUNET_MESSAGE_TYPE_DV_DISTANCE_CHANGED);
461   du_msg.distance = htonl (distance);
462   du_msg.peer = *peer;
463   send_control_to_plugin (&du_msg.header);
464 }
465
466
467 /**
468  * Give a CONNECT message to the plugin.
469  *
470  * @param target peer that connected
471  * @param distance distance to the target
472  */
473 static void
474 send_connect_to_plugin (const struct GNUNET_PeerIdentity *target, 
475                         uint32_t distance)
476 {
477   struct GNUNET_DV_ConnectMessage cm;
478
479   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
480               "Delivering CONNECT about peer `%s'\n",
481               GNUNET_i2s (target));
482   cm.header.size = htons (sizeof (cm));
483   cm.header.type = htons (GNUNET_MESSAGE_TYPE_DV_CONNECT);
484   cm.distance = htonl (distance);
485   cm.peer = *target;
486   send_control_to_plugin (&cm.header);
487 }
488
489
490 /**
491  * Give a DISCONNECT message to the plugin.
492  *
493  * @param target peer that disconnected
494  */
495 static void
496 send_disconnect_to_plugin (const struct GNUNET_PeerIdentity *target)
497 {
498   struct GNUNET_DV_DisconnectMessage dm;
499
500   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
501               "Delivering DISCONNECT about peer `%s'\n",
502               GNUNET_i2s (target));
503   dm.header.size = htons (sizeof (dm));
504   dm.header.type = htons (GNUNET_MESSAGE_TYPE_DV_DISCONNECT);
505   dm.reserved = htonl (0);
506   dm.peer = *target;
507   send_control_to_plugin (&dm.header);
508 }
509
510
511 /**
512  * Function called to transfer a message to another peer
513  * via core.
514  *
515  * @param cls closure with the direct neighbor
516  * @param size number of bytes available in buf
517  * @param buf where the callee should write the message
518  * @return number of bytes written to buf
519  */
520 static size_t
521 core_transmit_notify (void *cls, size_t size, void *buf)
522 {
523   struct DirectNeighbor *dn = cls;
524   char *cbuf = buf;
525   struct PendingMessage *pending;
526   size_t off;
527   size_t msize;
528
529   dn->cth = NULL;
530   if (NULL == buf)
531   {
532     /* peer disconnected */
533     return 0;
534   }
535   off = 0;
536   pending = dn->pm_head;
537   off = 0;
538   while ( (NULL != (pending = dn->pm_head)) &&
539           (size >= off + (msize = ntohs (pending->msg->size))))
540   {
541     dn->pm_queue_size--;
542     GNUNET_CONTAINER_DLL_remove (dn->pm_head,
543                                  dn->pm_tail,
544                                  pending);
545     memcpy (&cbuf[off], pending->msg, msize);
546     if (0 != pending->uid) 
547       send_ack_to_plugin (&pending->ultimate_target,
548                           pending->uid,
549                           GNUNET_NO);
550     GNUNET_free (pending);
551     off += msize;
552   }
553   if (NULL != dn->pm_head)
554     dn->cth =
555       GNUNET_CORE_notify_transmit_ready (core_api,
556                                          GNUNET_YES /* cork */,
557                                          0 /* priority */,
558                                          GNUNET_TIME_UNIT_FOREVER_REL,
559                                          &dn->peer,
560                                          msize,                                  
561                                          &core_transmit_notify, dn);
562   return off;
563 }
564
565
566 /**
567  * Forward the given payload to the given target.
568  *
569  * @param target where to send the message
570  * @param uid unique ID for the message
571  * @param ultimate_target ultimate recipient for the message
572  * @param distance expected (remaining) distance to the target
573  * @param sender original sender of the message
574  * @param payload payload of the message
575  */
576 static void
577 forward_payload (struct DirectNeighbor *target,
578                  uint32_t distance,
579                  uint32_t uid,
580                  const struct GNUNET_PeerIdentity *sender,
581                  const struct GNUNET_PeerIdentity *ultimate_target,
582                  const struct GNUNET_MessageHeader *payload)
583 {
584   struct PendingMessage *pm;
585   struct RouteMessage *rm;
586   size_t msize;
587
588   if ( (target->pm_queue_size >= MAX_QUEUE_SIZE) &&
589        (0 != memcmp (sender,
590                      &my_identity,
591                      sizeof (struct GNUNET_PeerIdentity))) )
592   {
593     GNUNET_break (0 == uid);
594     return;
595   }
596   msize = sizeof (struct RouteMessage) + ntohs (payload->size);
597   if (msize >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
598   {
599     GNUNET_break (0);
600     return;
601   }
602   pm = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
603   pm->ultimate_target = *ultimate_target;
604   pm->uid = uid;
605   pm->msg = (const struct GNUNET_MessageHeader *) &pm[1];
606   rm = (struct RouteMessage *) &pm[1];
607   rm->header.size = htons ((uint16_t) msize);
608   rm->header.type = htons (GNUNET_MESSAGE_TYPE_DV_ROUTE);
609   rm->distance = htonl (distance);
610   rm->target = target->peer;
611   rm->sender = *sender;
612   memcpy (&rm[1], payload, ntohs (payload->size));
613   GNUNET_CONTAINER_DLL_insert_tail (target->pm_head,
614                                     target->pm_tail,
615                                     pm);
616   target->pm_queue_size++;
617   if (NULL == target->cth)
618     target->cth = GNUNET_CORE_notify_transmit_ready (core_api,
619                                                      GNUNET_YES /* cork */,
620                                                      0 /* priority */,
621                                                      GNUNET_TIME_UNIT_FOREVER_REL,
622                                                      &target->peer,
623                                                      msize,                                      
624                                                      &core_transmit_notify, target);
625 }
626
627
628 /**
629  * Find a free slot for storing a 'route' in the 'consensi'
630  * set at the given distance.
631  *
632  * @param distance distance to use for the set slot
633  */
634 static unsigned int
635 get_consensus_slot (uint32_t distance)
636 {
637   struct ConsensusSet *cs;
638   unsigned int i;
639
640   cs = &consensi[distance];
641   i = 0;
642   while ( (i < cs->array_length) &&
643           (NULL != cs->targets[i]) ) i++;
644   if (i == cs->array_length)
645     GNUNET_array_grow (cs->targets,
646                        cs->array_length,
647                        cs->array_length * 2 + 2);
648   return i;
649 }
650
651
652 /**
653  * Allocate a slot in the consensus set for a route.
654  *
655  * @param route route to initialize
656  * @param distance which consensus set to use
657  */
658 static void
659 allocate_route (struct Route *route,
660                 uint32_t distance)
661 {
662   unsigned int i;
663
664   i = get_consensus_slot (distance);
665   route->set_offset = i;
666   consensi[distance].targets[i] = route;
667   route->target.distance = htonl (distance);
668 }
669
670
671 /**
672  * Release a slot in the consensus set for a route.
673  *
674  * @param route route to release the slot from
675  */
676 static void
677 release_route (struct Route *route)
678 {
679   consensi[ntohl (route->target.distance)].targets[route->set_offset] = NULL;
680   route->set_offset = UINT_MAX; /* indicate invalid slot */
681 }
682
683
684 /**
685  * Move a route from one consensus set to another.
686  *
687  * @param route route to move
688  * @param new_distance new distance for the route (destination set)
689  */
690 static void
691 move_route (struct Route *route,
692             uint32_t new_distance)
693 {
694   unsigned int i;
695
696   release_route (route);
697   i = get_consensus_slot (new_distance);
698   route->set_offset = i;
699   consensi[new_distance].targets[i] = route;     
700   route->target.distance = htonl (new_distance);
701 }
702
703
704 /**
705  * A peer is now connected to us at distance 1.  Initiate DV exchange.
706  *
707  * @param neighbor entry for the neighbor at distance 1
708  */
709 static void
710 handle_direct_connect (struct DirectNeighbor *neighbor)
711 {
712   struct Route *route;
713
714   GNUNET_STATISTICS_update (stats,
715                             "# peers connected (1-hop)",
716                             1, GNUNET_NO);
717   route = GNUNET_CONTAINER_multihashmap_get (all_routes, 
718                                              &neighbor->peer.hashPubKey);
719   if (NULL != route)  
720   {
721     send_disconnect_to_plugin (&neighbor->peer);
722     release_route (route);
723     GNUNET_free (route);
724   }
725   neighbor->consensus_task = GNUNET_SCHEDULER_add_now (&start_consensus,
726                                                        neighbor);
727 }
728
729
730 /**
731  * Method called whenever a peer connects.
732  *
733  * @param cls closure
734  * @param peer peer identity this notification is about
735  */
736 static void
737 handle_core_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
738 {
739   struct DirectNeighbor *neighbor;
740  
741   /* Check for connect to self message */
742   if (0 == memcmp (&my_identity, peer, sizeof (struct GNUNET_PeerIdentity)))
743     return;
744   /* check if entry exists */
745   neighbor = GNUNET_CONTAINER_multihashmap_get (direct_neighbors, 
746                                                 &peer->hashPubKey);
747   if (NULL != neighbor)
748   {
749     GNUNET_break (GNUNET_YES != neighbor->connected);
750     neighbor->connected = GNUNET_YES;
751     if (DIRECT_NEIGHBOR_COST != neighbor->distance)
752       return;
753     handle_direct_connect (neighbor);
754     return;
755   }
756   neighbor = GNUNET_malloc (sizeof (struct DirectNeighbor));
757   neighbor->peer = *peer;
758   GNUNET_assert (GNUNET_YES ==
759                  GNUNET_CONTAINER_multihashmap_put (direct_neighbors,
760                                                     &peer->hashPubKey,
761                                                     neighbor,
762                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
763   neighbor->connected = GNUNET_YES;
764   neighbor->distance = 0; /* unknown */
765 }
766
767
768 /**
769  * Called for each 'target' in a neighbor table to free the associated memory.
770  *
771  * @param cls NULL
772  * @param key key of the value
773  * @param value value to free
774  * @return GNUNET_OK to continue to iterate
775  */
776 static int
777 free_targets (void *cls,
778               const struct GNUNET_HashCode *key,
779               void *value)
780 {
781   GNUNET_free (value);
782   return GNUNET_OK;
783 }
784
785
786 /**
787  * Multihashmap iterator for checking if a given route is
788  * (now) useful to this peer.
789  *
790  * @param cls the direct neighbor for the given route
791  * @param key key value stored under
792  * @param value a 'struct Target' that may or may not be useful; not that
793  *        the distance in 'target' does not include the first hop yet
794  * @return GNUNET_YES to continue iteration, GNUNET_NO to stop
795  */
796 static int
797 check_possible_route (void *cls, const struct GNUNET_HashCode * key, void *value)
798 {
799   struct DirectNeighbor *neighbor = cls;
800   struct Target *target = value;
801   struct Route *route;
802   
803   route = GNUNET_CONTAINER_multihashmap_get (all_routes,
804                                            key);
805   if (NULL != route)
806   {
807     if (ntohl (route->target.distance) > ntohl (target->distance) + 1)
808     {
809       /* this 'target' is cheaper than the existing route; switch to alternative route! */
810       move_route (route, ntohl (target->distance) + 1);
811       route->next_hop = neighbor;
812       send_distance_change_to_plugin (&target->peer, ntohl (target->distance) + 1);
813     }
814     return GNUNET_YES; /* got a route to this target already */
815   }
816   route = GNUNET_malloc (sizeof (struct Route));
817   route->next_hop = neighbor;
818   route->target.distance = htonl (ntohl (target->distance) + 1);
819   route->target.peer = target->peer;
820   allocate_route (route, ntohl (route->target.distance));
821   GNUNET_assert (GNUNET_YES ==
822                  GNUNET_CONTAINER_multihashmap_put (all_routes,
823                                                     &route->target.peer.hashPubKey,
824                                                     route,
825                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
826   send_connect_to_plugin (&route->target.peer, ntohl (target->distance));
827   return GNUNET_YES;
828 }
829
830
831 /**
832  * Multihashmap iterator for finding routes that were previously
833  * "hidden" due to a better route (called after a disconnect event).
834  *
835  * @param cls NULL
836  * @param key peer identity of the given direct neighbor
837  * @param value a 'struct DirectNeighbor' to check for additional routes
838  * @return GNUNET_YES to continue iteration
839  */
840 static int
841 refresh_routes (void *cls, const struct GNUNET_HashCode * key, void *value)
842 {
843   struct DirectNeighbor *neighbor = value;
844
845   if ( (GNUNET_YES != neighbor->connected) ||
846        (DIRECT_NEIGHBOR_COST != neighbor->distance) )
847     return GNUNET_YES;    
848   if (NULL != neighbor->neighbor_table)
849     GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table,
850                                            &check_possible_route,
851                                            neighbor);
852   return GNUNET_YES;
853 }
854
855
856 /**
857  * Get distance information from 'atsi'.
858  *
859  * @param atsi performance data
860  * @param atsi_count number of entries in atsi
861  * @return connected transport distance
862  */
863 static uint32_t
864 get_atsi_distance (const struct GNUNET_ATS_Information *atsi,
865                    uint32_t atsi_count)
866 {
867   uint32_t i;
868
869   for (i = 0; i < atsi_count; i++)
870     if (ntohl (atsi[i].type) == GNUNET_ATS_QUALITY_NET_DISTANCE)
871       return ntohl (atsi->value);
872   /* If we do not have explicit distance data, assume direct neighbor. */
873   return DIRECT_NEIGHBOR_COST;
874 }
875
876
877 /**
878  * Multihashmap iterator for freeing routes that go via a particular
879  * neighbor that disconnected and is thus no longer available.
880  *
881  * @param cls the direct neighbor that is now unavailable
882  * @param key key value stored under
883  * @param value a 'struct Route' that may or may not go via neighbor
884  *
885  * @return GNUNET_YES to continue iteration, GNUNET_NO to stop
886  */
887 static int
888 cull_routes (void *cls, const struct GNUNET_HashCode * key, void *value)
889 {
890   struct DirectNeighbor *neighbor = cls;
891   struct Route *route = value;
892
893   if (route->next_hop != neighbor)
894     return GNUNET_YES; /* not affected */
895   GNUNET_assert (GNUNET_YES ==
896                  GNUNET_CONTAINER_multihashmap_remove (all_routes, key, value));
897   release_route (route);
898   send_disconnect_to_plugin (&route->target.peer);
899   GNUNET_free (route);
900   return GNUNET_YES;
901 }
902
903
904 /**
905  * Handle the case that a direct connection to a peer is
906  * disrupted.  Remove all routes via that peer and
907  * stop the consensus with it.
908  *
909  * @param neighbor peer that was disconnected (or at least is no 
910  *    longer at distance 1)
911  */
912 static void
913 handle_direct_disconnect (struct DirectNeighbor *neighbor)
914 {
915   GNUNET_CONTAINER_multihashmap_iterate (all_routes,
916                                          &cull_routes,
917                                          neighbor);
918   if (NULL != neighbor->cth)
919   {
920     GNUNET_CORE_notify_transmit_ready_cancel (neighbor->cth);
921     neighbor->cth = NULL;
922   }
923   if (NULL != neighbor->neighbor_table_consensus)
924   {
925     GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table_consensus,
926                                            &free_targets,
927                                            NULL);
928     GNUNET_CONTAINER_multihashmap_destroy (neighbor->neighbor_table_consensus);
929     neighbor->neighbor_table_consensus = NULL;
930   }
931   if (NULL != neighbor->neighbor_table)
932   {
933     GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table,
934                                            &free_targets,
935                                            NULL);
936     GNUNET_CONTAINER_multihashmap_destroy (neighbor->neighbor_table);
937     neighbor->neighbor_table = NULL;
938   }
939   if (GNUNET_SCHEDULER_NO_TASK != neighbor->consensus_task)
940   {
941     GNUNET_SCHEDULER_cancel (neighbor->consensus_task);
942     neighbor->consensus_task = GNUNET_SCHEDULER_NO_TASK;
943   }
944   if (NULL != neighbor->consensus)
945   {
946     GNUNET_CONSENSUS_destroy (neighbor->consensus);
947     neighbor->consensus = NULL;
948   }
949 }
950
951
952 /**
953  * Function that is called with QoS information about an address; used
954  * to update our current distance to another peer.
955  *
956  * @param cls closure
957  * @param address the address
958  * @param bandwidth_out assigned outbound bandwidth for the connection
959  * @param bandwidth_in assigned inbound bandwidth for the connection
960  * @param ats performance data for the address (as far as known)
961  * @param ats_count number of performance records in 'ats'
962  */
963 static void
964 handle_ats_update (void *cls,
965                    const struct GNUNET_PeerIdentity *peer,
966                    struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
967                    struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
968                    const struct GNUNET_ATS_Information *ats, 
969                    uint32_t ats_count)
970 {
971   struct DirectNeighbor *neighbor;
972   uint32_t distance;
973
974   /* FIXME: ignore CB if this address is not the one that is in use! */
975   distance = get_atsi_distance (ats, ats_count); 
976   /* check if entry exists */
977   neighbor = GNUNET_CONTAINER_multihashmap_get (direct_neighbors, 
978                                                         &peer->hashPubKey);
979   if (NULL != neighbor)
980   {    
981     if ( (DIRECT_NEIGHBOR_COST == neighbor->distance) &&
982          (DIRECT_NEIGHBOR_COST == distance) )
983       return; /* no change */
984     if (DIRECT_NEIGHBOR_COST == neighbor->distance) 
985     {
986       neighbor->distance = distance;
987       GNUNET_STATISTICS_update (stats,
988                                 "# peers connected (1-hop)",
989                                 -1, GNUNET_NO);  
990       handle_direct_disconnect (neighbor);
991       GNUNET_CONTAINER_multihashmap_iterate (direct_neighbors,
992                                              &refresh_routes,
993                                              NULL);
994       return;
995     }
996     neighbor->distance = distance;
997     if (DIRECT_NEIGHBOR_COST != neighbor->distance)
998       return;    
999     if (GNUNET_YES != neighbor->connected)
1000       return;
1001     handle_direct_connect (neighbor);
1002     return;
1003   }
1004   neighbor = GNUNET_malloc (sizeof (struct DirectNeighbor));
1005   neighbor->peer = (*peer);
1006   GNUNET_assert (GNUNET_YES ==
1007                  GNUNET_CONTAINER_multihashmap_put (direct_neighbors,
1008                                                     &peer->hashPubKey,
1009                                                     neighbor,
1010                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
1011   neighbor->connected = GNUNET_NO; /* not yet */
1012   neighbor->distance = distance; 
1013 }
1014
1015
1016 /**
1017  * Check if a target was removed from the set of the other peer; if so,
1018  * if we also used it for our route, we need to remove it from our
1019  * 'all_routes' set (and later check if an alternative path now exists).
1020  *
1021  * @param cls the 'struct DirectNeighbor'
1022  * @param key peer identity for the target
1023  * @param value a 'struct Target' previously reachable via the given neighbor
1024  */
1025 static int
1026 check_target_removed (void *cls,
1027                       const struct GNUNET_HashCode *key,
1028                       void *value)
1029 {
1030   struct DirectNeighbor *neighbor = cls;
1031   struct Target *new_target;
1032   struct Route *current_route;
1033
1034   new_target = GNUNET_CONTAINER_multihashmap_get (neighbor->neighbor_table_consensus,
1035                                                   key);
1036   if (NULL == new_target)
1037   {
1038     /* target was revoked, check if it was used */
1039     current_route = GNUNET_CONTAINER_multihashmap_get (all_routes,
1040                                                        key);
1041     if ( (NULL == current_route) ||
1042          (current_route->next_hop != neighbor) )
1043     {
1044       /* didn't matter, wasn't used */
1045       return GNUNET_OK;
1046     }
1047     /* remove existing route */
1048     GNUNET_assert (GNUNET_YES ==
1049                    GNUNET_CONTAINER_multihashmap_remove (all_routes, key, current_route));
1050     send_disconnect_to_plugin (&current_route->target.peer);
1051     GNUNET_free (current_route);
1052     neighbor->target_removed = GNUNET_YES;
1053     return GNUNET_OK;
1054   }
1055   return GNUNET_OK;
1056 }
1057
1058
1059 /**
1060  * Check if a target was added to the set of the other peer; if it
1061  * was added or impoves the existing route, do the needed updates.
1062  *
1063  * @param cls the 'struct DirectNeighbor'
1064  * @param key peer identity for the target
1065  * @param value a 'struct Target' now reachable via the given neighbor
1066  */
1067 static int
1068 check_target_added (void *cls,
1069                       const struct GNUNET_HashCode *key,
1070                       void *value)
1071 {
1072   struct DirectNeighbor *neighbor = cls;
1073   struct Target *target = value;
1074   struct Route *current_route;
1075
1076   /* target was revoked, check if it was used */
1077   current_route = GNUNET_CONTAINER_multihashmap_get (all_routes,
1078                                                      key);
1079   if (NULL != current_route)
1080   {
1081     /* route exists */
1082     if (current_route->next_hop == neighbor)
1083     {
1084       /* we had the same route before, no change */
1085       if (ntohl (target->distance) + 1 != ntohl (current_route->target.distance))
1086       {
1087         current_route->target.distance = htonl (ntohl (target->distance) + 1);
1088         send_distance_change_to_plugin (&target->peer, ntohl (target->distance) + 1);
1089       }
1090       return GNUNET_OK;
1091     }
1092     if (ntohl (current_route->target.distance) >= ntohl (target->distance) + 1)
1093     {
1094       /* alternative, shorter route exists, ignore */
1095       return GNUNET_OK;
1096     }
1097     /* new route is better than the existing one, take over! */
1098     /* NOTE: minor security issue: malicious peers may advertise
1099        very short routes to take over longer paths; as we don't
1100        check that the shorter routes actually work, a malicious
1101        direct neighbor can use this to DoS our long routes */
1102     current_route->next_hop = neighbor;
1103     current_route->target.distance = htonl (ntohl (target->distance) + 1);
1104     send_distance_change_to_plugin (&target->peer, ntohl (target->distance) + 1);
1105     return GNUNET_OK;
1106   }
1107   /* new route */
1108   current_route = GNUNET_malloc (sizeof (struct Route));
1109   current_route->next_hop = neighbor;
1110   current_route->target.peer = target->peer;
1111   current_route->target.distance = htonl (ntohl (target->distance) + 1);
1112   GNUNET_assert (GNUNET_YES ==
1113                  GNUNET_CONTAINER_multihashmap_put (all_routes,
1114                                                     &current_route->target.peer.hashPubKey,
1115                                                     current_route,
1116                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
1117   send_connect_to_plugin (&current_route->target.peer,
1118                           ntohl (current_route->target.distance));
1119   return GNUNET_OK;
1120 }
1121
1122
1123
1124 /**
1125  * The consensus has concluded, clean up and schedule the next one.
1126  *
1127  * @param cls the 'struct GNUNET_DirectNeighbor' with which we created the consensus
1128  */
1129 static void
1130 consensus_done_cb (void *cls)
1131 {
1132   struct DirectNeighbor *neighbor = cls;
1133
1134   GNUNET_CONSENSUS_destroy (neighbor->consensus);
1135   neighbor->consensus = NULL;
1136   /* remove targets that disappeared */
1137   neighbor->target_removed = GNUNET_NO;
1138   GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table,
1139                                          &check_target_removed,
1140                                          neighbor);
1141   if (GNUNET_YES == neighbor->target_removed)
1142   {
1143     /* check if we got an alternative for the removed routes */
1144     GNUNET_CONTAINER_multihashmap_iterate (direct_neighbors,
1145                                            &refresh_routes,
1146                                            NULL);    
1147   }
1148   /* add targets that appeared (and check for improved routes) */
1149   GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table_consensus,
1150                                          &check_target_added,
1151                                          neighbor);
1152   if (NULL != neighbor->neighbor_table)
1153   {
1154     GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table,
1155                                            &free_targets,
1156                                            NULL);
1157     GNUNET_CONTAINER_multihashmap_destroy (neighbor->neighbor_table);
1158     neighbor->neighbor_table = NULL;
1159   }
1160   neighbor->neighbor_table = neighbor->neighbor_table_consensus;
1161   neighbor->neighbor_table_consensus = NULL;
1162   neighbor->consensus_task = GNUNET_SCHEDULER_add_delayed (GNUNET_DV_CONSENSUS_FREQUENCY,
1163                                                            &start_consensus,
1164                                                            neighbor);
1165 }
1166
1167
1168 /**
1169  * We inserted the last element into the consensus, get ready to
1170  * insert the next element into the consensus or conclude if
1171  * we're done.
1172  *
1173  * @param cls the 'struct DirectNeighbor' of the peer we're building
1174  *        a routing consensus with
1175  * @param success GNUNET_OK if the last element was added successfully,
1176  *                GNUNET_SYSERR if we failed
1177  */
1178 static void
1179 insert_next_element (void *cls,
1180                      int success)
1181 {
1182   struct DirectNeighbor *neighbor = cls;
1183   struct GNUNET_CONSENSUS_Element element;
1184
1185   while ( (DEFAULT_FISHEYE_DEPTH - 1 > neighbor->consensus_insertion_distance) &&
1186           (consensi[neighbor->consensus_insertion_distance].array_length == neighbor->consensus_insertion_offset) )
1187   {
1188     neighbor->consensus_insertion_offset = 0;
1189     neighbor->consensus_insertion_distance++;
1190     /* skip over NULL entries */
1191     while ( (DEFAULT_FISHEYE_DEPTH - 1 > neighbor->consensus_insertion_distance) &&
1192             (consensi[neighbor->consensus_insertion_distance].array_length < neighbor->consensus_insertion_offset) &&
1193             (NULL == consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset]) )
1194       neighbor->consensus_insertion_offset++;
1195   }
1196   if (DEFAULT_FISHEYE_DEPTH - 1 == neighbor->consensus_insertion_distance)
1197   {
1198     /* we're done, conclude! */
1199     GNUNET_CONSENSUS_conclude (neighbor->consensus,
1200                                GNUNET_DV_CONSENSUS_FREQUENCY,
1201                                &consensus_done_cb,
1202                                neighbor);
1203     return;
1204   }
1205   element.size = sizeof (struct Target);
1206   element.data = &consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset++]->target;
1207
1208   /* skip over NULL entries */
1209   while ( (DEFAULT_FISHEYE_DEPTH - 1 > neighbor->consensus_insertion_distance) &&
1210           (consensi[neighbor->consensus_insertion_distance].array_length < neighbor->consensus_insertion_offset) &&
1211           (NULL == consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset]) )
1212     neighbor->consensus_insertion_offset++;  
1213   GNUNET_CONSENSUS_insert (neighbor->consensus,
1214                            &element,
1215                            &insert_next_element,
1216                            neighbor);
1217 }
1218
1219
1220 /**
1221  * We have learned a new route from the other peer.  Add it to the
1222  * route set we're building.
1223  *
1224  * @param cls the 'struct DirectNeighbor' we're building the consensus with
1225  * @param element the new element we have learned
1226  * @return GNUNET_OK if the valid is well-formed and should be added to the consensus,
1227  *         GNUNET_SYSERR if the element should be ignored and not be propagated
1228  */
1229 static int
1230 learn_route_cb (void *cls,
1231                 const struct GNUNET_CONSENSUS_Element *element)
1232 {
1233   struct DirectNeighbor *neighbor = cls;
1234   struct Target *target;
1235
1236   if (NULL == element)
1237   {
1238     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1239                 "Failed to establish DV consensus, will try again later\n");
1240     GNUNET_CONSENSUS_destroy (neighbor->consensus);
1241     if (NULL != neighbor->neighbor_table_consensus)
1242     {
1243       GNUNET_CONTAINER_multihashmap_iterate (neighbor->neighbor_table_consensus,
1244                                              &free_targets,
1245                                              NULL);
1246       GNUNET_CONTAINER_multihashmap_destroy (neighbor->neighbor_table_consensus);
1247       neighbor->neighbor_table_consensus = NULL;
1248     }
1249     neighbor->consensus = NULL;
1250     neighbor->consensus_task = GNUNET_SCHEDULER_add_delayed (GNUNET_DV_CONSENSUS_FREQUENCY,
1251                                                              &start_consensus,
1252                                                              neighbor);
1253     return GNUNET_SYSERR;
1254   }
1255   if (sizeof (struct Target) != element->size)
1256   {
1257     GNUNET_break_op (0);
1258     return GNUNET_SYSERR;
1259   }
1260   target = GNUNET_malloc (sizeof (struct Target));
1261   memcpy (target, element->data, sizeof (struct Target));
1262   if (GNUNET_YES !=
1263       GNUNET_CONTAINER_multihashmap_put (neighbor->neighbor_table_consensus,
1264                                          &target->peer.hashPubKey,
1265                                          target,
1266                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
1267   {
1268     GNUNET_break_op (0);
1269     GNUNET_free (target);
1270     return GNUNET_SYSERR;
1271   }
1272   return GNUNET_OK;
1273 }
1274
1275
1276 /**
1277  * Start creating a new consensus from scratch.
1278  *
1279  * @param cls the 'struct DirectNeighbor' of the peer we're building
1280  *        a routing consensus with
1281  * @param tc scheduler context
1282  */    
1283 static void
1284 start_consensus (void *cls,
1285                  const struct GNUNET_SCHEDULER_TaskContext *tc)
1286 {
1287   struct DirectNeighbor *neighbor = cls;
1288   struct GNUNET_HashCode session_id;
1289   struct GNUNET_HashCode real_session_id;
1290
1291   neighbor->consensus_task = GNUNET_SCHEDULER_NO_TASK;
1292   neighbor->consensus_insertion_offset = 0;
1293   neighbor->consensus_insertion_distance = 0;
1294   GNUNET_assert (NULL == neighbor->neighbor_table_consensus);
1295   GNUNET_assert (NULL == neighbor->consensus);
1296   neighbor->neighbor_table_consensus = GNUNET_CONTAINER_multihashmap_create (1024, GNUNET_YES);
1297   /* construct session ID seed as XOR of both peer's identities */
1298   GNUNET_CRYPTO_hash_xor (&my_identity.hashPubKey, 
1299                           &neighbor->peer.hashPubKey, 
1300                           &session_id);
1301   /* make sure session ID is unique across applications by salting it with 'DV' */
1302   GNUNET_CRYPTO_hkdf (&real_session_id, sizeof (real_session_id),
1303                       GCRY_MD_SHA512, GCRY_MD_SHA256,
1304                       "DV-SALT", 2,
1305                       &session_id, sizeof (session_id),
1306                       NULL, 0);
1307   neighbor->consensus = GNUNET_CONSENSUS_create (cfg,
1308                                                  1,
1309                                                  &neighbor->peer,
1310                                                  &real_session_id,
1311                                                  &learn_route_cb,
1312                                                  neighbor);
1313   if (NULL == neighbor->consensus)
1314   {
1315     neighbor->consensus_task = GNUNET_SCHEDULER_add_delayed (GNUNET_DV_CONSENSUS_FREQUENCY,
1316                                                              &start_consensus,
1317                                                              neighbor);
1318     return;
1319   }
1320   insert_next_element (neighbor, GNUNET_OK);
1321 }
1322
1323
1324 /**
1325  * Core handler for DV data messages.  Whatever this message
1326  * contains all we really have to do is rip it out of its
1327  * DV layering and give it to our pal the DV plugin to report
1328  * in with.
1329  *
1330  * @param cls closure
1331  * @param peer peer which sent the message (immediate sender)
1332  * @param message the message
1333  * @return GNUNET_OK on success, GNUNET_SYSERR if the other peer violated the protocol
1334  */
1335 static int
1336 handle_dv_route_message (void *cls, const struct GNUNET_PeerIdentity *peer,
1337                          const struct GNUNET_MessageHeader *message)
1338 {
1339   const struct RouteMessage *rm;
1340   const struct GNUNET_MessageHeader *payload;
1341   struct Route *route;
1342
1343   if (ntohs (message->size) < sizeof (struct RouteMessage) + sizeof (struct GNUNET_MessageHeader))
1344   {
1345     GNUNET_break_op (0);
1346     return GNUNET_SYSERR;
1347   }
1348   rm = (const struct RouteMessage *) message;
1349   payload = (const struct GNUNET_MessageHeader *) &rm[1];
1350   if (ntohs (message->size) != sizeof (struct RouteMessage) + ntohs (payload->size))
1351   {
1352     GNUNET_break_op (0);
1353     return GNUNET_SYSERR;
1354   }
1355   if (0 == memcmp (&rm->target,
1356                    &my_identity,
1357                    sizeof (struct GNUNET_PeerIdentity)))
1358   {
1359     /* message is for me, check reverse route! */
1360     route = GNUNET_CONTAINER_multihashmap_get (all_routes,
1361                                                &rm->sender.hashPubKey);
1362     if (NULL == route)
1363     {
1364       /* don't have reverse route, drop */
1365       GNUNET_STATISTICS_update (stats,
1366                                 "# message discarded (no reverse route)",
1367                                 1, GNUNET_NO);
1368       return GNUNET_OK;
1369     }
1370     send_data_to_plugin (payload,
1371                          &rm->sender,
1372                          ntohl (route->target.distance));
1373     return GNUNET_OK;
1374   }
1375   route = GNUNET_CONTAINER_multihashmap_get (all_routes,
1376                                              &rm->target.hashPubKey);
1377   if (NULL == route)
1378   {
1379     GNUNET_STATISTICS_update (stats,
1380                               "# messages discarded (no route)",
1381                               1, GNUNET_NO);
1382     return GNUNET_OK;
1383   }
1384   if (ntohl (route->target.distance) > ntohl (rm->distance) + 1)
1385   {
1386     GNUNET_STATISTICS_update (stats,
1387                               "# messages discarded (target too far)",
1388                               1, GNUNET_NO);
1389     return GNUNET_OK;
1390   }
1391   forward_payload (route->next_hop,
1392                    ntohl (route->target.distance),
1393                    0,
1394                    &rm->target,
1395                    &rm->sender,
1396                    payload);
1397   return GNUNET_OK;  
1398 }
1399
1400
1401 /**
1402  * Service server's handler for message send requests (which come
1403  * bubbling up to us through the DV plugin).
1404  *
1405  * @param cls closure
1406  * @param client identification of the client
1407  * @param message the actual message
1408  */
1409 static void
1410 handle_dv_send_message (void *cls, struct GNUNET_SERVER_Client *client,
1411                         const struct GNUNET_MessageHeader *message)
1412 {
1413   struct Route *route;
1414   const struct GNUNET_DV_SendMessage *msg;
1415   const struct GNUNET_MessageHeader *payload;
1416
1417   if (ntohs (message->size) < sizeof (struct GNUNET_DV_SendMessage) + sizeof (struct GNUNET_MessageHeader))
1418   {
1419     GNUNET_break (0);
1420     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1421     return;
1422   }
1423   msg = (const struct GNUNET_DV_SendMessage *) message;
1424   GNUNET_break (0 != ntohl (msg->uid));
1425   payload = (const struct GNUNET_MessageHeader *) &msg[1];
1426   if (ntohs (message->size) != sizeof (struct GNUNET_DV_SendMessage) + ntohs (payload->size))
1427   {
1428     GNUNET_break (0);
1429     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1430     return;
1431   }
1432   route = GNUNET_CONTAINER_multihashmap_get (all_routes,
1433                                              &msg->target.hashPubKey);
1434   if (NULL == route)
1435   {
1436     /* got disconnected */
1437     GNUNET_STATISTICS_update (stats,
1438                               "# local messages discarded (no route)",
1439                               1, GNUNET_NO);
1440     send_ack_to_plugin (&msg->target, ntohl (msg->uid), GNUNET_YES);
1441     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1442     return;
1443   }
1444   forward_payload (route->next_hop,
1445                    ntohl (route->target.distance),
1446                    htonl (msg->uid),
1447                    &msg->target,
1448                    &my_identity,
1449                    payload);
1450   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1451 }
1452
1453
1454 /**
1455  * Cleanup all of the data structures associated with a given neighbor.
1456  *
1457  * @param neighbor neighbor to clean up
1458  */
1459 static void
1460 cleanup_neighbor (struct DirectNeighbor *neighbor)
1461 {
1462   struct PendingMessage *pending;
1463
1464   while (NULL != (pending = neighbor->pm_head))
1465   {
1466     neighbor->pm_queue_size--;
1467     GNUNET_CONTAINER_DLL_remove (neighbor->pm_head,
1468                                  neighbor->pm_tail,
1469                                  pending);    
1470     GNUNET_free (pending);
1471   }
1472   handle_direct_disconnect (neighbor);
1473   GNUNET_assert (GNUNET_YES ==
1474                  GNUNET_CONTAINER_multihashmap_remove (direct_neighbors, 
1475                                                        &neighbor->peer.hashPubKey,
1476                                                        neighbor));
1477   GNUNET_free (neighbor);
1478 }
1479
1480
1481 /**
1482  * Method called whenever a given peer disconnects.
1483  *
1484  * @param cls closure
1485  * @param peer peer identity this notification is about
1486  */
1487 static void
1488 handle_core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
1489 {
1490   struct DirectNeighbor *neighbor;
1491
1492   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1493               "Received core peer disconnect message for peer `%s'!\n",
1494               GNUNET_i2s (peer));
1495   /* Check for disconnect from self message */
1496   if (0 == memcmp (&my_identity, peer, sizeof (struct GNUNET_PeerIdentity)))
1497     return;
1498   neighbor =
1499       GNUNET_CONTAINER_multihashmap_get (direct_neighbors, &peer->hashPubKey);
1500   if (NULL == neighbor)
1501   {
1502     GNUNET_break (0);
1503     return;
1504   }
1505   GNUNET_break (GNUNET_YES == neighbor->connected);
1506   neighbor->connected = GNUNET_NO;
1507   if (DIRECT_NEIGHBOR_COST == neighbor->distance)
1508   {
1509     GNUNET_STATISTICS_update (stats,
1510                               "# peers connected (1-hop)",
1511                               -1, GNUNET_NO);  
1512   }
1513   cleanup_neighbor (neighbor);
1514   GNUNET_CONTAINER_multihashmap_iterate (direct_neighbors,
1515                                          &refresh_routes,
1516                                          NULL);
1517 }
1518
1519
1520 /**
1521  * Multihashmap iterator for freeing routes.  Should never be called.
1522  *
1523  * @param cls NULL
1524  * @param key key value stored under
1525  * @param value the route to be freed
1526  *
1527  * @return GNUNET_YES to continue iteration, GNUNET_NO to stop
1528  */
1529 static int
1530 free_route (void *cls, const struct GNUNET_HashCode * key, void *value)
1531 {
1532   struct Route *route = value;
1533
1534   GNUNET_break (0);
1535   GNUNET_assert (GNUNET_YES ==
1536                  GNUNET_CONTAINER_multihashmap_remove (all_routes, key, value));
1537   release_route (route);
1538   send_disconnect_to_plugin (&route->target.peer);
1539   GNUNET_free (route);
1540   return GNUNET_YES;
1541 }
1542
1543
1544 /**
1545  * Multihashmap iterator for freeing direct neighbors. Should never be called.
1546  *
1547  * @param cls NULL
1548  * @param key key value stored under
1549  * @param value the direct neighbor to be freed
1550  *
1551  * @return GNUNET_YES to continue iteration, GNUNET_NO to stop
1552  */
1553 static int
1554 free_direct_neighbors (void *cls, const struct GNUNET_HashCode * key, void *value)
1555 {
1556   struct DirectNeighbor *neighbor = value;
1557
1558   GNUNET_break (0);
1559   cleanup_neighbor (neighbor);
1560   return GNUNET_YES;
1561 }
1562
1563
1564 /**
1565  * Task run during shutdown.
1566  *
1567  * @param cls unused
1568  * @param tc unused
1569  */
1570 static void
1571 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1572 {
1573   unsigned int i;
1574
1575   GNUNET_CORE_disconnect (core_api);
1576   core_api = NULL;
1577   GNUNET_ATS_performance_done (ats);
1578   ats = NULL;
1579   GNUNET_CONTAINER_multihashmap_iterate (direct_neighbors,
1580                                          &free_direct_neighbors, NULL);
1581   GNUNET_CONTAINER_multihashmap_iterate (all_routes,
1582                                          &free_route, NULL);
1583   GNUNET_CONTAINER_multihashmap_destroy (direct_neighbors);
1584   GNUNET_CONTAINER_multihashmap_destroy (all_routes);
1585   GNUNET_STATISTICS_destroy (stats, GNUNET_NO);
1586   stats = NULL;
1587   GNUNET_SERVER_notification_context_destroy (nc);
1588   nc = NULL;
1589   for (i=0;i<DEFAULT_FISHEYE_DEPTH - 1;i++)
1590     GNUNET_array_grow (consensi[i].targets,
1591                        consensi[i].array_length,
1592                        0);
1593 }
1594
1595
1596 /**
1597  * Notify newly connected client about an existing route.
1598  *
1599  * @param cls the 'struct GNUNET_SERVER_Client'
1600  * @param key peer identity
1601  * @param value the XXX.
1602  * @return GNUNET_OK (continue to iterate)
1603  */
1604 static int
1605 add_route (void *cls,
1606            const struct GNUNET_HashCode *key,
1607            void *value)
1608 {
1609   struct GNUNET_SERVER_Client *client = cls;
1610   struct Route *route = value;
1611   struct GNUNET_DV_ConnectMessage cm;
1612   
1613   cm.header.size = htons (sizeof (cm));
1614   cm.header.type = htons (GNUNET_MESSAGE_TYPE_DV_CONNECT);
1615   cm.distance = htonl (route->target.distance);
1616   cm.peer = route->target.peer;
1617
1618   GNUNET_SERVER_notification_context_unicast (nc, 
1619                                               client,
1620                                               &cm.header,
1621                                               GNUNET_NO);
1622   return GNUNET_OK;
1623 }
1624
1625
1626 /**
1627  * Handle START-message.  This is the first message sent to us
1628  * by the client (can only be one!).
1629  *
1630  * @param cls closure (always NULL)
1631  * @param client identification of the client
1632  * @param message the actual message
1633  */
1634 static void
1635 handle_start (void *cls, struct GNUNET_SERVER_Client *client,
1636               const struct GNUNET_MessageHeader *message)
1637 {
1638   GNUNET_SERVER_notification_context_add (nc, client);  
1639   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1640   GNUNET_CONTAINER_multihashmap_iterate (all_routes,
1641                                          &add_route,
1642                                          client);
1643 }
1644
1645
1646 /**
1647  * Called on core init.
1648  *
1649  * @param cls unused
1650  * @param server legacy
1651  * @param identity this peer's identity
1652  */
1653 static void
1654 core_init (void *cls, struct GNUNET_CORE_Handle *server,
1655            const struct GNUNET_PeerIdentity *identity)
1656 {
1657   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1658               "I am peer: %s\n",
1659               GNUNET_i2s (identity));
1660   my_identity = *identity;
1661 }
1662
1663
1664 /**
1665  * Process dv requests.
1666  *
1667  * @param cls closure
1668  * @param server the initialized server
1669  * @param c configuration to use
1670  */
1671 static void
1672 run (void *cls, struct GNUNET_SERVER_Handle *server,
1673      const struct GNUNET_CONFIGURATION_Handle *c)
1674 {
1675   static struct GNUNET_CORE_MessageHandler core_handlers[] = {
1676     {&handle_dv_route_message, GNUNET_MESSAGE_TYPE_DV_ROUTE, 0},
1677     {NULL, 0, 0}
1678   };
1679   static struct GNUNET_SERVER_MessageHandler plugin_handlers[] = {
1680     {&handle_start, NULL, 
1681      GNUNET_MESSAGE_TYPE_DV_START, 
1682      sizeof (struct GNUNET_MessageHeader) },
1683     { &handle_dv_send_message, NULL, 
1684       GNUNET_MESSAGE_TYPE_DV_SEND, 
1685       0},
1686     {NULL, NULL, 0, 0}
1687   };
1688
1689   cfg = c;
1690   direct_neighbors = GNUNET_CONTAINER_multihashmap_create (128, GNUNET_NO);
1691   all_routes = GNUNET_CONTAINER_multihashmap_create (65536, GNUNET_NO);
1692   core_api = GNUNET_CORE_connect (cfg, NULL,
1693                                   &core_init, 
1694                                   &handle_core_connect,
1695                                   &handle_core_disconnect,
1696                                   NULL, GNUNET_NO, 
1697                                   NULL, GNUNET_NO, 
1698                                   core_handlers);
1699
1700   if (NULL == core_api)
1701     return;
1702   ats = GNUNET_ATS_performance_init (cfg, &handle_ats_update, NULL, NULL, NULL);
1703   if (NULL == ats)
1704   {
1705     GNUNET_CORE_disconnect (core_api);
1706     return;
1707   }
1708   nc = GNUNET_SERVER_notification_context_create (server,
1709                                                   MAX_QUEUE_SIZE_PLUGIN);
1710   stats = GNUNET_STATISTICS_create ("dv", cfg);
1711   GNUNET_SERVER_add_handlers (server, plugin_handlers);
1712   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1713                                 &shutdown_task, NULL);
1714 }
1715
1716
1717 /**
1718  * The main function for the dv service.
1719  *
1720  * @param argc number of arguments from the command line
1721  * @param argv command line arguments
1722  * @return 0 ok, 1 on error
1723  */
1724 int
1725 main (int argc, char *const *argv)
1726 {
1727   return (GNUNET_OK ==
1728           GNUNET_SERVICE_run (argc, argv, "dv", GNUNET_SERVICE_OPTION_NONE,
1729                               &run, NULL)) ? 0 : 1;
1730 }
1731
1732 /* end of gnunet-service-dv.c */