src/dv/gnunet-service-dv.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2013, 2016 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, 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_set_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  * Information about a direct neighbor (core-level, excluding
 129  * DV-links, only DV-enabled peers).
 130  */
 131 struct DirectNeighbor
 132 {
 133
 134   /**
 135    * Identity of the peer.
 136    */
 137   struct GNUNET_PeerIdentity peer;
 138
 139   /**
 140    * Session ID we use whenever we create a set union with
 141    * this neighbor; constructed from the XOR of our peer
 142    * IDs and then salted with "DV-SALT" to avoid conflicts
 143    * with other applications.
 144    */
 145   struct GNUNET_HashCode real_session_id;
 146
 147   /**
 148    * Transmit handle to core service.
 149    */
 150   struct GNUNET_MQ_Handle *mq;
 151
 152   /**
 153    * Routing table of the neighbor, NULL if not yet established.
 154    * Keys are peer identities, values are 'struct Target' entries.
 155    * Note that the distances in the targets are from the point-of-view
 156    * of the peer, not from us!
 157    */
 158   struct GNUNET_CONTAINER_MultiPeerMap *neighbor_table;
 159
 160   /**
 161    * Updated routing table of the neighbor, under construction,
 162    * NULL if we are not currently building it.
 163    * Keys are peer identities, values are 'struct Target' entries.
 164    * Note that the distances in the targets are from the point-of-view
 165    * of the other peer, not from us!
 166    */
 167   struct GNUNET_CONTAINER_MultiPeerMap *neighbor_table_consensus;
 168
 169   /**
 170    * Our current (exposed) routing table as a set.
 171    */
 172   struct GNUNET_SET_Handle *my_set;
 173
 174   /**
 175    * Handle for our current active set union operation.
 176    */
 177   struct GNUNET_SET_OperationHandle *set_op;
 178
 179   /**
 180    * Handle used if we are listening for this peer, waiting for the
 181    * other peer to initiate construction of the set union.  NULL if
 182    * we ar the initiating peer.
 183    */
 184   struct GNUNET_SET_ListenHandle *listen_handle;
 185
 186   /**
 187    * ID of the task we use to (periodically) update our consensus
 188    * with this peer.  Used if we are the initiating peer.
 189    */
 190   struct GNUNET_SCHEDULER_Task *initiate_task;
 191
 192   /**
 193    * At what offset are we, with respect to inserting our own routes
 194    * into the consensus?
 195    */
 196   unsigned int consensus_insertion_offset;
 197
 198   /**
 199    * At what distance are we, with respect to inserting our own routes
 200    * into the consensus?
 201    */
 202   unsigned int consensus_insertion_distance;
 203
 204   /**
 205    * Elements in consensus
 206    */
 207   unsigned int consensus_elements;
 208
 209   /**
 210    * Direct one hop route
 211    */
 212   struct Route *direct_route;
 213
 214   /**
 215    * Flag set within 'check_target_removed' to trigger full global route refresh.
 216    */
 217   int target_removed;
 218
 219   /**
 220    * Our distance to this peer, 0 for unknown.
 221    */
 222   uint32_t distance;
 223
 224   /**
 225    * The network this peer is in
 226    */
 227   enum GNUNET_ATS_Network_Type network;
 228
 229   /**
 230    * Is this neighbor connected at the core level?
 231    */
 232   int connected;
 233
 234 };
 235
 236
 237 /**
 238  * A route includes information about the next hop,
 239  * the target, and the ultimate distance to the
 240  * target.
 241  */
 242 struct Route
 243 {
 244
 245   /**
 246    * Which peer do we need to forward the message to?
 247    */
 248   struct DirectNeighbor *next_hop;
 249
 250   /**
 251    * What would be the target, and how far is it away?
 252    */
 253   struct Target target;
 254
 255   /**
 256    * Offset of this target in the respective consensus set.
 257    */
 258   unsigned int set_offset;
 259
 260 };
 261
 262
 263 /**
 264  * Set of targets we bring to a consensus; all targets in a set have a
 265  * distance equal to the sets distance (which is implied by the array
 266  * index of the set).
 267  */
 268 struct ConsensusSet
 269 {
 270
 271   /**
 272    * Array of targets in the set, may include NULL entries if a
 273    * neighbor has disconnected; the targets are allocated with the
 274    * respective container (all_routes), not here.
 275    */
 276   struct Route **targets;
 277
 278   /**
 279    * Size of the @e targets array.
 280    */
 281   unsigned int array_length;
 282
 283 };
 284
 285
 286 /**
 287  * Peermap of all of our neighbors; processing these usually requires
 288  * first checking to see if the peer is core-connected and if the
 289  * distance is 1, in which case they are direct neighbors.
 290  */
 291 static struct GNUNET_CONTAINER_MultiPeerMap *direct_neighbors;
 292
 293 /**
 294  * Hashmap with all routes that we currently support; contains
 295  * routing information for all peers from distance 2
 296  * up to distance #DEFAULT_FISHEYE_DEPTH.
 297  */
 298 static struct GNUNET_CONTAINER_MultiPeerMap *all_routes;
 299
 300 /**
 301  * Array of consensus sets we expose to the outside world.  Sets
 302  * are structured by the distance to the target.
 303  */
 304 static struct ConsensusSet consensi[DEFAULT_FISHEYE_DEPTH];
 305
 306 /**
 307  * Handle to the core service api.
 308  */
 309 static struct GNUNET_CORE_Handle *core_api;
 310
 311 /**
 312  * The identity of our peer.
 313  */
 314 static struct GNUNET_PeerIdentity my_identity;
 315
 316 /**
 317  * The configuration for this service.
 318  */
 319 static const struct GNUNET_CONFIGURATION_Handle *cfg;
 320
 321 /**
 322  * The client, the DV plugin connected to us (or an event monitor).
 323  * Hopefully this client will never change, although if the plugin
 324  * dies and returns for some reason it may happen.
 325  */
 326 static struct GNUNET_NotificationContext *nc;
 327
 328 /**
 329  * Handle for the statistics service.
 330  */
 331 static struct GNUNET_STATISTICS_Handle *stats;
 332
 333 /**
 334  * Handle to ATS service.
 335  */
 336 static struct GNUNET_ATS_PerformanceHandle *ats;
 337
 338 /**
 339  * Task scheduled to refresh routes based on direct neighbours.
 340  */
 341 static struct GNUNET_SCHEDULER_Task *rr_task;
 342
 343 /**
 344  * #GNUNET_YES if we are shutting down.
 345  */
 346 static int in_shutdown;
 347
 348 /**
 349  * Start creating a new DV set union by initiating the connection.
 350  *
 351  * @param cls the 'struct DirectNeighbor' of the peer we're building
 352  *        a routing consensus with
 353  */
 354 static void
 355 initiate_set_union (void *cls);
 356
 357
 358 /**
 359  * Start creating a new DV set union construction, our neighbour has
 360  * asked for it (callback for listening peer).
 361  *
 362  * @param cls the 'struct DirectNeighbor' of the peer we're building
 363  *        a routing consensus with
 364  * @param other_peer the other peer
 365  * @param context_msg message with application specific information from
 366  *        the other peer
 367  * @param request request from the other peer, use GNUNET_SET_accept
 368  *        to accept it, otherwise the request will be refused
 369  *        Note that we don't use a return value here, as it is also
 370  *        necessary to specify the set we want to do the operation with,
 371  *        whith sometimes can be derived from the context message.
 372  *        Also necessary to specify the timeout.
 373  */
 374 static void
 375 listen_set_union (void *cls,
 376                   const struct GNUNET_PeerIdentity *other_peer,
 377                   const struct GNUNET_MessageHeader *context_msg,
 378                   struct GNUNET_SET_Request *request);
 379
 380
 381 /**
 382  * Forward a message from another peer to the plugin.
 383  *
 384  * @param message the message to send to the plugin
 385  * @param origin the original sender of the message
 386  * @param distance distance to the original sender of the message
 387  */
 388 static void
 389 send_data_to_plugin (const struct GNUNET_MessageHeader *message,
 390                      const struct GNUNET_PeerIdentity *origin,
 391                      uint32_t distance)
 392 {
 393   struct GNUNET_DV_ReceivedMessage *received_msg;
 394   size_t size;
 395
 396   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 397               "Delivering message from peer `%s' at distance %u\n",
 398               GNUNET_i2s (origin),
 399               (unsigned int) distance);
 400   size = sizeof (struct GNUNET_DV_ReceivedMessage) +
 401     ntohs (message->size);
 402   if (size >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
 403   {
 404     GNUNET_break (0); /* too big */
 405     return;
 406   }
 407   received_msg = GNUNET_malloc (size);
 408   received_msg->header.size = htons (size);
 409   received_msg->header.type = htons (GNUNET_MESSAGE_TYPE_DV_RECV);
 410   received_msg->distance = htonl (distance);
 411   received_msg->sender = *origin;
 412   GNUNET_memcpy (&received_msg[1], message, ntohs (message->size));
 413   GNUNET_notification_context_broadcast (nc,
 414                                          &received_msg->header,
 415                                          GNUNET_YES);
 416   GNUNET_free (received_msg);
 417 }
 418
 419
 420 /**
 421  * Forward a control message to the plugin.
 422  *
 423  * @param message the message to send to the plugin
 424  */
 425 static void
 426 send_control_to_plugin (const struct GNUNET_MessageHeader *message)
 427 {
 428   GNUNET_notification_context_broadcast (nc,
 429                                          message,
 430                                          GNUNET_NO);
 431 }
 432
 433
 434 /**
 435  * Send a DISTANCE_CHANGED message to the plugin.
 436  *
 437  * @param peer peer with a changed distance
 438  * @param distance new distance to the peer
 439  * @param network network used by the neighbor
 440  */
 441 static void
 442 send_distance_change_to_plugin (const struct GNUNET_PeerIdentity *peer,
 443                                 uint32_t distance,
 444                                 enum GNUNET_ATS_Network_Type network)
 445 {
 446   struct GNUNET_DV_DistanceUpdateMessage du_msg;
 447
 448   GNUNET_break (GNUNET_ATS_NET_UNSPECIFIED != network);
 449   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 450               "Delivering DISTANCE_CHANGED for message about peer `%s'\n",
 451               GNUNET_i2s (peer));
 452   du_msg.header.size = htons (sizeof (du_msg));
 453   du_msg.header.type = htons (GNUNET_MESSAGE_TYPE_DV_DISTANCE_CHANGED);
 454   du_msg.distance = htonl (distance);
 455   du_msg.peer = *peer;
 456   du_msg.network = htonl ((uint32_t) network);
 457   send_control_to_plugin (&du_msg.header);
 458 }
 459
 460
 461 /**
 462  * Give a CONNECT message to the plugin.
 463  *
 464  * @param target peer that connected
 465  * @param distance distance to the target
 466  * @param network the network the next hop is located in
 467  */
 468 static void
 469 send_connect_to_plugin (const struct GNUNET_PeerIdentity *target,
 470                         uint32_t distance,
 471                         enum GNUNET_ATS_Network_Type network)
 472 {
 473   struct GNUNET_DV_ConnectMessage cm;
 474
 475   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 476               "Delivering CONNECT about peer %s with distance %u\n",
 477               GNUNET_i2s (target), distance);
 478   cm.header.size = htons (sizeof (cm));
 479   cm.header.type = htons (GNUNET_MESSAGE_TYPE_DV_CONNECT);
 480   cm.distance = htonl (distance);
 481   cm.network = htonl ((uint32_t) network);
 482   cm.peer = *target;
 483   send_control_to_plugin (&cm.header);
 484 }
 485
 486
 487 /**
 488  * Give a DISCONNECT message to the plugin.
 489  *
 490  * @param target peer that disconnected
 491  */
 492 static void
 493 send_disconnect_to_plugin (const struct GNUNET_PeerIdentity *target)
 494 {
 495   struct GNUNET_DV_DisconnectMessage dm;
 496
 497   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 498               "Delivering DISCONNECT about peer `%s'\n",
 499               GNUNET_i2s (target));
 500   dm.header.size = htons (sizeof (dm));
 501   dm.header.type = htons (GNUNET_MESSAGE_TYPE_DV_DISCONNECT);
 502   dm.reserved = htonl (0);
 503   dm.peer = *target;
 504   send_control_to_plugin (&dm.header);
 505 }
 506
 507
 508 /**
 509  * Forward the given payload to the given target.
 510  *
 511  * @param target where to send the message
 512  * @param distance distance to the @a sender
 513  * @param sender original sender of the message
 514  * @param actual_target ultimate recipient for the message
 515  * @param payload payload of the message
 516  */
 517 static void
 518 forward_payload (struct DirectNeighbor *target,
 519                  uint32_t distance,
 520                  const struct GNUNET_PeerIdentity *sender,
 521                  const struct GNUNET_PeerIdentity *actual_target,
 522                  const struct GNUNET_MessageHeader *payload)
 523 {
 524   struct GNUNET_MQ_Envelope *env;
 525   struct RouteMessage *rm;
 526
 527   if ( (GNUNET_MQ_get_length (target->mq) >= MAX_QUEUE_SIZE) &&
 528        (0 != memcmp (sender,
 529                      &my_identity,
 530                      sizeof (struct GNUNET_PeerIdentity))) )
 531   {
 532     /* not _our_ client and queue is full, drop */
 533     GNUNET_STATISTICS_update (stats,
 534                               "# messages dropped",
 535                               1,
 536                               GNUNET_NO);
 537     return;
 538   }
 539   if (sizeof (struct RouteMessage) + ntohs (payload->size)
 540       >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
 541   {
 542     GNUNET_break (0);
 543     return;
 544   }
 545   env = GNUNET_MQ_msg_nested_mh (rm,
 546                                  GNUNET_MESSAGE_TYPE_DV_ROUTE,
 547                                  payload);
 548   rm->distance = htonl (distance);
 549   rm->target = *actual_target;
 550   rm->sender = *sender;
 551   GNUNET_MQ_send (target->mq,
 552                   env);
 553 }
 554
 555
 556 /**
 557  * Find a free slot for storing a 'route' in the 'consensi'
 558  * set at the given distance.
 559  *
 560  * @param distance distance to use for the set slot
 561  */
 562 static unsigned int
 563 get_consensus_slot (uint32_t distance)
 564 {
 565   struct ConsensusSet *cs;
 566   unsigned int i;
 567
 568   GNUNET_assert (distance < DEFAULT_FISHEYE_DEPTH);
 569   cs = &consensi[distance];
 570   i = 0;
 571   while ( (i < cs->array_length) &&
 572           (NULL != cs->targets[i]) ) i++;
 573   if (i == cs->array_length)
 574   {
 575     GNUNET_array_grow (cs->targets,
 576                        cs->array_length,
 577                        cs->array_length * 2 + 2);
 578   }
 579   return i;
 580 }
 581
 582
 583 /**
 584  * Allocate a slot in the consensus set for a route.
 585  *
 586  * @param route route to initialize
 587  * @param distance which consensus set to use
 588  */
 589 static void
 590 allocate_route (struct Route *route,
 591                 uint32_t distance)
 592 {
 593   unsigned int i;
 594
 595   if (distance >= DEFAULT_FISHEYE_DEPTH)
 596   {
 597     route->target.distance = htonl (distance);
 598     route->set_offset = UINT_MAX; /* invalid slot */
 599     return;
 600   }
 601   i = get_consensus_slot (distance);
 602   route->set_offset = i;
 603   consensi[distance].targets[i] = route;
 604   route->target.distance = htonl (distance);
 605 }
 606
 607
 608 /**
 609  * Release a slot in the consensus set for a route.
 610  *
 611  * @param route route to release the slot from
 612  */
 613 static void
 614 release_route (struct Route *route)
 615 {
 616   if (UINT_MAX == route->set_offset)
 617     return;
 618   GNUNET_assert (ntohl (route->target.distance) < DEFAULT_FISHEYE_DEPTH);
 619   consensi[ntohl (route->target.distance)].targets[route->set_offset] = NULL;
 620   route->set_offset = UINT_MAX; /* indicate invalid slot */
 621 }
 622
 623
 624 /**
 625  * Move a route from one consensus set to another.
 626  *
 627  * @param route route to move
 628  * @param new_distance new distance for the route (destination set)
 629  */
 630 static void
 631 move_route (struct Route *route,
 632             uint32_t new_distance)
 633 {
 634   release_route (route);
 635   allocate_route (route, new_distance);
 636 }
 637
 638
 639 /**
 640  * Initialize this neighbors 'my_set' and when done give
 641  * it to the pending set operation for execution.
 642  *
 643  * Add a single element to the set per call:
 644  *
 645  * If we reached the last element of a consensus element: increase distance
 646  *
 647  *
 648  * @param cls the neighbor for which we are building the set
 649  */
 650 static void
 651 build_set (void *cls)
 652 {
 653   struct DirectNeighbor *neighbor = cls;
 654   struct GNUNET_SET_Element element;
 655   struct Target *target;
 656   struct Route *route;
 657
 658   target = NULL;
 659   /* skip over NULL entries */
 660   while ( (DEFAULT_FISHEYE_DEPTH > neighbor->consensus_insertion_distance) &&
 661           (consensi[neighbor->consensus_insertion_distance].array_length > neighbor->consensus_insertion_offset) &&
 662           (NULL == consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset]) )
 663     neighbor->consensus_insertion_offset++;
 664   while ( (DEFAULT_FISHEYE_DEPTH > neighbor->consensus_insertion_distance) &&
 665           (consensi[neighbor->consensus_insertion_distance].array_length == neighbor->consensus_insertion_offset) )
 666   {
 667     /* If we reached the last element of a consensus array element: increase distance and start with next array */
 668     neighbor->consensus_insertion_offset = 0;
 669     neighbor->consensus_insertion_distance++;
 670     /* skip over NULL entries */
 671     while ( (DEFAULT_FISHEYE_DEPTH > neighbor->consensus_insertion_distance) &&
 672             (consensi[neighbor->consensus_insertion_distance].array_length  > neighbor->consensus_insertion_offset) &&
 673             (NULL == consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset]) )
 674       neighbor->consensus_insertion_offset++;
 675   }
 676   if (DEFAULT_FISHEYE_DEPTH == neighbor->consensus_insertion_distance)
 677   {
 678     /* we have added all elements to the set, run the operation */
 679     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 680                 "Finished building my SET for peer `%s' with %u elements, committing\n",
 681                 GNUNET_i2s (&neighbor->peer),
 682                 neighbor->consensus_elements);
 683     GNUNET_SET_commit (neighbor->set_op,
 684                        neighbor->my_set);
 685     GNUNET_SET_destroy (neighbor->my_set);
 686     neighbor->my_set = NULL;
 687     return;
 688   }
 689
 690   route = consensi[neighbor->consensus_insertion_distance].targets[neighbor->consensus_insertion_offset];
 691   GNUNET_assert (NULL != route);
 692   target = &route->target;
 693   GNUNET_assert (ntohl (target->distance) < DEFAULT_FISHEYE_DEPTH);
 694   element.size = sizeof (struct Target);
 695   element.element_type = htons (0);
 696   element.data = target;
 697
 698   /* Find next non-NULL entry */
 699   neighbor->consensus_insertion_offset++;
 700   if ( (0 != memcmp (&target->peer,
 701                      &my_identity,
 702                      sizeof (my_identity))) &&
 703        (0 != memcmp (&target->peer,
 704                      &neighbor->peer,
 705                      sizeof (struct GNUNET_PeerIdentity))) )
 706   {
 707     /* Add target if it is not the neighbor or this peer */
 708     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 709                 "Adding peer `%s' with distance %u to SET\n",
 710                 GNUNET_i2s (&target->peer),
 711                 ntohl (target->distance) + 1);
 712     GNUNET_SET_add_element (neighbor->my_set,
 713                             &element,
 714                             &build_set, neighbor);
 715     neighbor->consensus_elements++;
 716   }
 717   else
 718     build_set (neighbor);
 719 }
 720
 721
 722 /**
 723  * A peer is now connected to us at distance 1.  Initiate DV exchange.
 724  *
 725  * @param neighbor entry for the neighbor at distance 1
 726  */
 727 static void
 728 handle_direct_connect (struct DirectNeighbor *neighbor)
 729 {
 730   struct Route *route;
 731   struct GNUNET_HashCode h1;
 732   struct GNUNET_HashCode h2;
 733   struct GNUNET_HashCode session_id;
 734
 735   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 736               "Direct connection to %s established, routing table exchange begins.\n",
 737               GNUNET_i2s (&neighbor->peer));
 738   GNUNET_STATISTICS_update (stats,
 739                             "# peers connected (1-hop)",
 740                             1, GNUNET_NO);
 741   route = GNUNET_CONTAINER_multipeermap_get (all_routes,
 742                                              &neighbor->peer);
 743   if (NULL != route)
 744   {
 745     GNUNET_assert (GNUNET_YES ==
 746                    GNUNET_CONTAINER_multipeermap_remove (all_routes,
 747                                                          &neighbor->peer,
 748                                                          route));
 749     send_disconnect_to_plugin (&neighbor->peer);
 750     release_route (route);
 751     GNUNET_free (route);
 752   }
 753
 754   neighbor->direct_route = GNUNET_new (struct Route);
 755   neighbor->direct_route->next_hop = neighbor;
 756   neighbor->direct_route->target.peer = neighbor->peer;
 757   allocate_route (neighbor->direct_route, DIRECT_NEIGHBOR_COST);
 758
 759   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 760               "Adding direct route to %s\n",
 761               GNUNET_i2s (&neighbor->direct_route->target.peer));
 762
 763
 764   /* construct session ID seed as XOR of both peer's identities */
 765   GNUNET_CRYPTO_hash (&my_identity,
 766                       sizeof (my_identity),
 767                       &h1);
 768   GNUNET_CRYPTO_hash (&neighbor->peer,
 769                       sizeof (struct GNUNET_PeerIdentity),
 770                       &h2);
 771   GNUNET_CRYPTO_hash_xor (&h1,
 772                           &h2,
 773                           &session_id);
 774   /* make sure session ID is unique across applications by salting it with 'DV' */
 775   GNUNET_CRYPTO_hkdf (&neighbor->real_session_id, sizeof (struct GNUNET_HashCode),
 776                       GCRY_MD_SHA512, GCRY_MD_SHA256,
 777                       "DV-SALT", 2,
 778                       &session_id, sizeof (session_id),
 779                       NULL, 0);
 780   if (0 < memcmp (&neighbor->peer,
 781                   &my_identity,
 782                   sizeof (struct GNUNET_PeerIdentity)))
 783   {
 784     if (NULL != neighbor->listen_handle)
 785     {
 786       GNUNET_break (0);
 787     }
 788     else
 789       neighbor->initiate_task = GNUNET_SCHEDULER_add_now (&initiate_set_union,
 790                                                           neighbor);
 791   }
 792   else
 793   {
 794     if (NULL != neighbor->listen_handle)
 795     {
 796       GNUNET_break (0);
 797     }
 798     else
 799     {
 800       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 801                   "Starting SET listen operation with peer `%s'\n",
 802                   GNUNET_i2s (&neighbor->peer));
 803       neighbor->listen_handle = GNUNET_SET_listen (cfg,
 804                                                    GNUNET_SET_OPERATION_UNION,
 805                                                    &neighbor->real_session_id,
 806                                                    &listen_set_union,
 807                                                    neighbor);
 808     }
 809   }
 810 }
 811
 812
 813 /**
 814  * Method called whenever a peer connects.
 815  *
 816  * @param cls closure
 817  * @param peer peer identity this notification is about
 818  * @param mq message queue for sending data to @a peer
 819  * @return our `struct DirectNeighbour` for this peer
 820  */
 821 static void *
 822 handle_core_connect (void *cls,
 823                      const struct GNUNET_PeerIdentity *peer,
 824                      struct GNUNET_MQ_Handle *mq)
 825 {
 826   struct DirectNeighbor *neighbor;
 827
 828   /* Check for connect to self message */
 829   if (0 == memcmp (&my_identity,
 830                    peer,
 831                    sizeof (struct GNUNET_PeerIdentity)))
 832     return NULL;
 833   /* check if entry exists */
 834   neighbor = GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
 835                                                 peer);
 836   if (NULL != neighbor)
 837   {
 838     GNUNET_break (GNUNET_ATS_NET_UNSPECIFIED != neighbor->network);
 839     GNUNET_break (GNUNET_YES != neighbor->connected);
 840     neighbor->connected = GNUNET_YES;
 841     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 842                 "Core connected to %s (distance %u)\n",
 843                 GNUNET_i2s (peer),
 844                 (unsigned int) neighbor->distance);
 845     if (DIRECT_NEIGHBOR_COST != neighbor->distance)
 846       return NULL;
 847     handle_direct_connect (neighbor);
 848     return NULL;
 849   }
 850   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 851               "Core connected to %s (distance unknown)\n",
 852               GNUNET_i2s (peer));
 853   neighbor = GNUNET_new (struct DirectNeighbor);
 854   neighbor->peer = *peer;
 855   GNUNET_assert (GNUNET_YES ==
 856                  GNUNET_CONTAINER_multipeermap_put (direct_neighbors,
 857                                                     &neighbor->peer,
 858                                                     neighbor,
 859                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 860   neighbor->connected = GNUNET_YES;
 861   neighbor->distance = 0; /* unknown */
 862   neighbor->network = GNUNET_ATS_NET_UNSPECIFIED;
 863   return neighbor;
 864 }
 865
 866
 867 /**
 868  * Called for each 'target' in a neighbor table to free the associated memory.
 869  *
 870  * @param cls NULL
 871  * @param key key of the value
 872  * @param value value to free
 873  * @return #GNUNET_OK to continue to iterate
 874  */
 875 static int
 876 free_targets (void *cls,
 877               const struct GNUNET_PeerIdentity *key,
 878               void *value)
 879 {
 880   GNUNET_free (value);
 881   return GNUNET_OK;
 882 }
 883
 884
 885 /**
 886  * Add a new route to the given @a target via the given @a neighbor.
 887  *
 888  * @param target the target of the route
 889  * @param neighbor the next hop for communicating with the @a target
 890  */
 891 static void
 892 add_new_route (struct Target *target,
 893                struct DirectNeighbor *neighbor)
 894 {
 895   struct Route *route;
 896
 897   route = GNUNET_new (struct Route);
 898   route->next_hop = neighbor;
 899   route->target.peer = target->peer;
 900   allocate_route (route, ntohl (target->distance) + 1);
 901   GNUNET_assert (GNUNET_YES ==
 902                  GNUNET_CONTAINER_multipeermap_put (all_routes,
 903                                                     &route->target.peer,
 904                                                     route,
 905                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 906   send_connect_to_plugin (&route->target.peer,
 907                           ntohl (route->target.distance),
 908                           neighbor->network);
 909 }
 910
 911
 912 /**
 913  * Multipeerhmap iterator for checking if a given route is
 914  * (now) useful to this peer.
 915  *
 916  * @param cls the direct neighbor for the given route
 917  * @param key key value stored under
 918  * @param value a 'struct Target' that may or may not be useful; not that
 919  *        the distance in 'target' does not include the first hop yet
 920  * @return #GNUNET_YES to continue iteration, #GNUNET_NO to stop
 921  */
 922 static int
 923 check_possible_route (void *cls,
 924                       const struct GNUNET_PeerIdentity *key,
 925                       void *value)
 926 {
 927   struct DirectNeighbor *neighbor = cls;
 928   struct Target *target = value;
 929   struct Route *route;
 930
 931   if (GNUNET_YES ==
 932       GNUNET_CONTAINER_multipeermap_contains (direct_neighbors,
 933                                               key))
 934     return GNUNET_YES; /* direct route, do not care about alternatives */
 935   route = GNUNET_CONTAINER_multipeermap_get (all_routes,
 936                                              key);
 937   if (NULL != route)
 938   {
 939     /* we have an existing route, check how it compares with going via 'target' */
 940     if (ntohl (route->target.distance) > ntohl (target->distance) + 1)
 941     {
 942       /* via 'target' is cheaper than the existing route; switch to alternative route! */
 943       move_route (route, ntohl (target->distance) + 1);
 944       route->next_hop = neighbor;
 945       send_distance_change_to_plugin (&target->peer,
 946                                       ntohl (target->distance) + 1,
 947                                       neighbor->network);
 948     }
 949     return GNUNET_YES; /* got a route to this target already */
 950   }
 951   if (ntohl (target->distance) >= DEFAULT_FISHEYE_DEPTH)
 952     return GNUNET_YES; /* distance is too large to be interesting */
 953   add_new_route (target, neighbor);
 954   return GNUNET_YES;
 955 }
 956
 957
 958 /**
 959  * Multipeermap iterator for finding routes that were previously
 960  * "hidden" due to a better route (called after a disconnect event).
 961  *
 962  * @param cls NULL
 963  * @param key peer identity of the given direct neighbor
 964  * @param value a `struct DirectNeighbor` to check for additional routes
 965  * @return #GNUNET_YES to continue iteration
 966  */
 967 static int
 968 refresh_routes (void *cls,
 969                 const struct GNUNET_PeerIdentity *key,
 970                 void *value)
 971 {
 972   struct DirectNeighbor *neighbor = value;
 973
 974   if ( (GNUNET_YES != neighbor->connected) ||
 975        (DIRECT_NEIGHBOR_COST != neighbor->distance) )
 976     return GNUNET_YES;
 977   if (NULL != neighbor->neighbor_table)
 978     GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table,
 979                                            &check_possible_route,
 980                                            neighbor);
 981   return GNUNET_YES;
 982 }
 983
 984
 985 /**
 986  * Task to run #refresh_routes() on all direct neighbours.
 987  *
 988  * @param cls NULL
 989  */
 990 static void
 991 refresh_routes_task (void *cls)
 992 {
 993   rr_task = NULL;
 994   GNUNET_CONTAINER_multipeermap_iterate (direct_neighbors,
 995                                          &refresh_routes,
 996                                          NULL);
 997 }
 998
 999
1000 /**
1001  * Asynchronously run #refresh_routes() at the next opportunity
1002  * on all direct neighbours.
1003  */
1004 static void
1005 schedule_refresh_routes ()
1006 {
1007   if (NULL == rr_task)
1008     rr_task = GNUNET_SCHEDULER_add_now (&refresh_routes_task,
1009                                         NULL);
1010 }
1011
1012
1013 /**
1014  * Multipeermap iterator for freeing routes that go via a particular
1015  * neighbor that disconnected and is thus no longer available.
1016  *
1017  * @param cls the direct neighbor that is now unavailable
1018  * @param key key value stored under
1019  * @param value a `struct Route` that may or may not go via neighbor
1020  *
1021  * @return #GNUNET_YES to continue iteration, #GNUNET_NO to stop
1022  */
1023 static int
1024 cull_routes (void *cls,
1025              const struct GNUNET_PeerIdentity *key,
1026              void *value)
1027 {
1028   struct DirectNeighbor *neighbor = cls;
1029   struct Route *route = value;
1030
1031   if (route->next_hop != neighbor)
1032     return GNUNET_YES; /* not affected */
1033   GNUNET_assert (GNUNET_YES ==
1034                  GNUNET_CONTAINER_multipeermap_remove (all_routes, key, value));
1035   release_route (route);
1036   send_disconnect_to_plugin (&route->target.peer);
1037   GNUNET_free (route);
1038   return GNUNET_YES;
1039 }
1040
1041
1042 /**
1043  * Handle the case that a direct connection to a peer is
1044  * disrupted.  Remove all routes via that peer and
1045  * stop the consensus with it.
1046  *
1047  * @param neighbor peer that was disconnected (or at least is no
1048  *    longer at distance 1)
1049  */
1050 static void
1051 handle_direct_disconnect (struct DirectNeighbor *neighbor)
1052 {
1053   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1054               "Culling routes via %s due to direct disconnect\n",
1055               GNUNET_i2s (&neighbor->peer));
1056   GNUNET_CONTAINER_multipeermap_iterate (all_routes,
1057                                          &cull_routes,
1058                                          neighbor);
1059   if (NULL != neighbor->direct_route)
1060   {
1061     release_route (neighbor->direct_route);
1062     GNUNET_free (neighbor->direct_route);
1063     neighbor->direct_route = NULL;
1064   }
1065   if (NULL != neighbor->neighbor_table_consensus)
1066   {
1067     GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table_consensus,
1068                                            &free_targets,
1069                                            NULL);
1070     GNUNET_CONTAINER_multipeermap_destroy (neighbor->neighbor_table_consensus);
1071     neighbor->neighbor_table_consensus = NULL;
1072   }
1073   if (NULL != neighbor->neighbor_table)
1074   {
1075     GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table,
1076                                            &free_targets,
1077                                            NULL);
1078     GNUNET_CONTAINER_multipeermap_destroy (neighbor->neighbor_table);
1079     neighbor->neighbor_table = NULL;
1080   }
1081   if (NULL != neighbor->set_op)
1082   {
1083     GNUNET_SET_operation_cancel (neighbor->set_op);
1084     neighbor->set_op = NULL;
1085   }
1086   if (NULL != neighbor->my_set)
1087   {
1088     GNUNET_SET_destroy (neighbor->my_set);
1089     neighbor->my_set = NULL;
1090   }
1091   if (NULL != neighbor->listen_handle)
1092   {
1093     GNUNET_SET_listen_cancel (neighbor->listen_handle);
1094     neighbor->listen_handle = NULL;
1095   }
1096   if (NULL != neighbor->initiate_task)
1097   {
1098     GNUNET_SCHEDULER_cancel (neighbor->initiate_task);
1099     neighbor->initiate_task = NULL;
1100   }
1101 }
1102
1103
1104 /**
1105  * Function that is called with QoS information about an address; used
1106  * to update our current distance to another peer.
1107  *
1108  * @param cls closure
1109  * @param address the address
1110  * @param active #GNUNET_YES if this address is actively used
1111  *        to maintain a connection to a peer;
1112  *        #GNUNET_NO if the address is not actively used;
1113  *        #GNUNET_SYSERR if this address is no longer available for ATS
1114  * @param bandwidth_out assigned outbound bandwidth for the connection
1115  * @param bandwidth_in assigned inbound bandwidth for the connection
1116  * @param prop performance data for the address (as far as known)
1117  */
1118 static void
1119 handle_ats_update (void *cls,
1120                    const struct GNUNET_HELLO_Address *address,
1121                    int active,
1122                    struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out,
1123                    struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
1124                    const struct GNUNET_ATS_Properties *prop)
1125 {
1126   struct DirectNeighbor *neighbor;
1127   uint32_t distance;
1128   enum GNUNET_ATS_Network_Type network;
1129
1130   if (NULL == address)
1131   {
1132     /* ATS service temporarily disconnected */
1133     return;
1134   }
1135
1136   if (GNUNET_YES != active)
1137   {
1138     // FIXME: handle disconnect/inactive case too!
1139     return;
1140   }
1141   distance = prop->distance;
1142   network = prop->scope;
1143   GNUNET_break (GNUNET_ATS_NET_UNSPECIFIED != network);
1144   /* check if entry exists */
1145   neighbor = GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
1146                                                 &address->peer);
1147   if (NULL != neighbor)
1148   {
1149     neighbor->network = network;
1150     if (neighbor->distance == distance)
1151       return; /* nothing new to see here, move along */
1152     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1153                 "ATS says distance to %s is now %u\n",
1154                 GNUNET_i2s (&address->peer),
1155                 (unsigned int) distance);
1156     if ( (DIRECT_NEIGHBOR_COST == neighbor->distance) &&
1157          (DIRECT_NEIGHBOR_COST == distance) )
1158       return; /* no change */
1159     if (DIRECT_NEIGHBOR_COST == neighbor->distance)
1160     {
1161       neighbor->distance = distance;
1162       GNUNET_STATISTICS_update (stats,
1163                                 "# peers connected (1-hop)",
1164                                 -1, GNUNET_NO);
1165       handle_direct_disconnect (neighbor);
1166       schedule_refresh_routes ();
1167       return;
1168     }
1169     neighbor->distance = distance;
1170     if (DIRECT_NEIGHBOR_COST != neighbor->distance)
1171       return;
1172     if (GNUNET_YES != neighbor->connected)
1173       return;
1174     handle_direct_connect (neighbor);
1175     return;
1176   }
1177   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1178               "ATS says distance to %s is now %u\n",
1179               GNUNET_i2s (&address->peer),
1180               (unsigned int) distance);
1181   neighbor = GNUNET_new (struct DirectNeighbor);
1182   neighbor->peer = address->peer;
1183   GNUNET_assert (GNUNET_YES ==
1184                  GNUNET_CONTAINER_multipeermap_put (direct_neighbors,
1185                                                     &neighbor->peer,
1186                                                     neighbor,
1187                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
1188   neighbor->connected = GNUNET_NO; /* not yet */
1189   neighbor->distance = distance;
1190   neighbor->network = network;
1191 }
1192
1193
1194 /**
1195  * Check if a target was removed from the set of the other peer; if so,
1196  * if we also used it for our route, we need to remove it from our
1197  * 'all_routes' set (and later check if an alternative path now exists).
1198  *
1199  * @param cls the `struct DirectNeighbor`
1200  * @param key peer identity for the target
1201  * @param value a `struct Target` previously reachable via the given neighbor
1202  */
1203 static int
1204 check_target_removed (void *cls,
1205                       const struct GNUNET_PeerIdentity *key,
1206                       void *value)
1207 {
1208   struct DirectNeighbor *neighbor = cls;
1209   struct Target *new_target;
1210   struct Route *current_route;
1211
1212   new_target = GNUNET_CONTAINER_multipeermap_get (neighbor->neighbor_table_consensus,
1213                                                   key);
1214   current_route = GNUNET_CONTAINER_multipeermap_get (all_routes,
1215                                                      key);
1216   if (NULL != new_target)
1217   {
1218     /* target was in old set, is in new set */
1219     if ( (NULL != current_route) &&
1220          (current_route->next_hop == neighbor) &&
1221          (current_route->target.distance != new_target->distance) )
1222     {
1223       /* need to recalculate routes due to distance change */
1224       neighbor->target_removed = GNUNET_YES;
1225     }
1226     return GNUNET_OK;
1227   }
1228   /* target was revoked, check if it was used */
1229   if ( (NULL == current_route) ||
1230        (current_route->next_hop != neighbor) )
1231   {
1232     /* didn't matter, wasn't used */
1233     return GNUNET_OK;
1234   }
1235   /* remove existing route */
1236   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1237               "Lost route to %s\n",
1238               GNUNET_i2s (&current_route->target.peer));
1239   GNUNET_assert (GNUNET_YES ==
1240                  GNUNET_CONTAINER_multipeermap_remove (all_routes, key, current_route));
1241   send_disconnect_to_plugin (&current_route->target.peer);
1242   release_route (current_route);
1243   GNUNET_free (current_route);
1244   neighbor->target_removed = GNUNET_YES;
1245   return GNUNET_OK;
1246 }
1247
1248
1249 /**
1250  * Check if a target was added to the set of the other peer; if it
1251  * was added or impoves the existing route, do the needed updates.
1252  *
1253  * @param cls the `struct DirectNeighbor`
1254  * @param key peer identity for the target
1255  * @param value a `struct Target` now reachable via the given neighbor
1256  */
1257 static int
1258 check_target_added (void *cls,
1259                     const struct GNUNET_PeerIdentity *key,
1260                     void *value)
1261 {
1262   struct DirectNeighbor *neighbor = cls;
1263   struct Target *target = value;
1264   struct Route *current_route;
1265
1266   /* target was revoked, check if it was used */
1267   current_route = GNUNET_CONTAINER_multipeermap_get (all_routes,
1268                                                      key);
1269   if (NULL != current_route)
1270   {
1271     /* route exists */
1272     if (current_route->next_hop == neighbor)
1273     {
1274       /* we had the same route before, no change in target */
1275       if (ntohl (target->distance) + 1 != ntohl (current_route->target.distance))
1276       {
1277         /* but distance changed! */
1278         if (ntohl (target->distance) + 1 > DEFAULT_FISHEYE_DEPTH)
1279         {
1280           /* distance increased beyond what is allowed, kill route */
1281           GNUNET_assert (GNUNET_YES ==
1282                          GNUNET_CONTAINER_multipeermap_remove (all_routes,
1283                                                                key,
1284                                                                current_route));
1285           send_disconnect_to_plugin (key);
1286           release_route (current_route);
1287           GNUNET_free (current_route);
1288         }
1289         else
1290         {
1291           /* distance decreased, update route */
1292           move_route (current_route,
1293                       ntohl (target->distance) + 1);
1294           send_distance_change_to_plugin (&target->peer,
1295                                           ntohl (target->distance) + 1,
1296                                           neighbor->network);
1297         }
1298       }
1299       return GNUNET_OK;
1300     }
1301     if (ntohl (current_route->target.distance) <= ntohl (target->distance) + 1)
1302     {
1303       /* alternative, shorter route exists, ignore */
1304       return GNUNET_OK;
1305     }
1306     /* new route is better than the existing one, take over! */
1307     /* NOTE: minor security issue: malicious peers may advertise
1308        very short routes to take over longer paths; as we don't
1309        check that the shorter routes actually work, a malicious
1310        direct neighbor can use this to DoS our long routes */
1311
1312     move_route (current_route, ntohl (target->distance) + 1);
1313     current_route->next_hop = neighbor;
1314     send_distance_change_to_plugin (&target->peer,
1315                                     ntohl (target->distance) + 1,
1316                                     neighbor->network);
1317     return GNUNET_OK;
1318   }
1319   /* new route */
1320   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1321               "Discovered new route to %s using %u hops\n",
1322               GNUNET_i2s (&target->peer),
1323               (unsigned int) (ntohl (target->distance) + 1));
1324   current_route = GNUNET_new (struct Route);
1325   current_route->next_hop = neighbor;
1326   current_route->target.peer = target->peer;
1327   allocate_route (current_route, ntohl (target->distance) + 1);
1328   GNUNET_assert (GNUNET_YES ==
1329                  GNUNET_CONTAINER_multipeermap_put (all_routes,
1330                                                     &current_route->target.peer,
1331                                                     current_route,
1332                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
1333
1334   send_connect_to_plugin (&current_route->target.peer,
1335                           ntohl (current_route->target.distance),
1336                           neighbor->network);
1337   return GNUNET_OK;
1338 }
1339
1340
1341 /**
1342  * Callback for set operation results. Called for each element
1343  * in the result set.
1344  * We have learned a new route from the other peer.  Add it to the
1345  * route set we're building.
1346  *
1347  * @param cls the `struct DirectNeighbor` we're building the consensus with
1348  * @param element a result element, only valid if status is #GNUNET_SET_STATUS_OK
1349  * @param status see `enum GNUNET_SET_Status`
1350  */
1351 static void
1352 handle_set_union_result (void *cls,
1353                          const struct GNUNET_SET_Element *element,
1354                          enum GNUNET_SET_Status status)
1355 {
1356   struct DirectNeighbor *neighbor = cls;
1357   struct DirectNeighbor *dn;
1358   struct Target *target;
1359   const struct Target *ctarget;
1360   char *status_str;
1361
1362   switch (status)
1363   {
1364   case GNUNET_SET_STATUS_OK:
1365     status_str = "GNUNET_SET_STATUS_OK";
1366     break;
1367   case GNUNET_SET_STATUS_FAILURE:
1368     status_str = "GNUNET_SET_STATUS_FAILURE";
1369     break;
1370   case GNUNET_SET_STATUS_HALF_DONE:
1371     status_str = "GNUNET_SET_STATUS_HALF_DONE";
1372     break;
1373   case GNUNET_SET_STATUS_DONE:
1374     status_str = "GNUNET_SET_STATUS_DONE";
1375     break;
1376   default:
1377     status_str = "UNDEFINED";
1378     break;
1379   }
1380
1381   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1382               "Got SET union result: %s\n",
1383               status_str);
1384   switch (status)
1385   {
1386   case GNUNET_SET_STATUS_OK:
1387     if (sizeof (struct Target) != element->size)
1388     {
1389       GNUNET_break_op (0);
1390       return;
1391     }
1392     ctarget = element->data;
1393     if ( (NULL !=
1394           (dn = GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
1395                                                    &ctarget->peer))) &&
1396          (DIRECT_NEIGHBOR_COST == dn->distance) )
1397     {
1398       /* this is a direct neighbor of ours, we do not care about routes
1399          to this peer */
1400       return;
1401     }
1402     target = GNUNET_new (struct Target);
1403     GNUNET_memcpy (target, element->data, sizeof (struct Target));
1404     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1405                 "Received information about peer `%s' with distance %u from SET\n",
1406                 GNUNET_i2s (&target->peer),
1407                 ntohl (target->distance) + 1);
1408
1409     if (NULL == neighbor->neighbor_table_consensus)
1410       neighbor->neighbor_table_consensus
1411         = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1412     if (GNUNET_YES !=
1413         GNUNET_CONTAINER_multipeermap_put (neighbor->neighbor_table_consensus,
1414                                            &target->peer,
1415                                            target,
1416                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
1417     {
1418       GNUNET_break_op (0);
1419       GNUNET_free (target);
1420     }
1421     break;
1422   case GNUNET_SET_STATUS_FAILURE:
1423     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1424                 "Failed to establish DV union, will try again later\n");
1425     neighbor->set_op = NULL;
1426     if (NULL != neighbor->neighbor_table_consensus)
1427     {
1428       GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table_consensus,
1429                                              &free_targets,
1430                                              NULL);
1431       GNUNET_CONTAINER_multipeermap_destroy (neighbor->neighbor_table_consensus);
1432       neighbor->neighbor_table_consensus = NULL;
1433     }
1434     if (0 < memcmp (&neighbor->peer,
1435                     &my_identity,
1436                     sizeof (struct GNUNET_PeerIdentity)))
1437       neighbor->initiate_task = GNUNET_SCHEDULER_add_delayed (GNUNET_DV_CONSENSUS_FREQUENCY,
1438                                                               &initiate_set_union,
1439                                                               neighbor);
1440     break;
1441   case GNUNET_SET_STATUS_HALF_DONE:
1442     break;
1443   case GNUNET_SET_STATUS_DONE:
1444     /* we got all of our updates; integrate routing table! */
1445     neighbor->target_removed = GNUNET_NO;
1446     if (NULL == neighbor->neighbor_table_consensus)
1447       neighbor->neighbor_table_consensus = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1448     if (NULL != neighbor->neighbor_table)
1449       GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table,
1450                                              &check_target_removed,
1451                                              neighbor);
1452     if (GNUNET_YES == neighbor->target_removed)
1453     {
1454       /* check if we got an alternative for the removed routes */
1455       schedule_refresh_routes ();
1456     }
1457     /* add targets that appeared (and check for improved routes) */
1458     GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table_consensus,
1459                                            &check_target_added,
1460                                            neighbor);
1461     if (NULL != neighbor->neighbor_table)
1462     {
1463       GNUNET_CONTAINER_multipeermap_iterate (neighbor->neighbor_table,
1464                                              &free_targets,
1465                                              NULL);
1466       GNUNET_CONTAINER_multipeermap_destroy (neighbor->neighbor_table);
1467       neighbor->neighbor_table = NULL;
1468     }
1469     neighbor->neighbor_table = neighbor->neighbor_table_consensus;
1470     neighbor->neighbor_table_consensus = NULL;
1471
1472     /* operation done, schedule next run! */
1473     neighbor->set_op = NULL;
1474     if (0 < memcmp (&neighbor->peer,
1475                     &my_identity,
1476                     sizeof (struct GNUNET_PeerIdentity)))
1477       neighbor->initiate_task = GNUNET_SCHEDULER_add_delayed (GNUNET_DV_CONSENSUS_FREQUENCY,
1478                                                               &initiate_set_union,
1479                                                               neighbor);
1480     break;
1481   default:
1482     GNUNET_break (0);
1483     return;
1484   }
1485 }
1486
1487
1488 /**
1489  * Start creating a new DV set union construction, our neighbour has
1490  * asked for it (callback for listening peer).
1491  *
1492  * @param cls the 'struct DirectNeighbor' of the peer we're building
1493  *        a routing consensus with
1494  * @param other_peer the other peer
1495  * @param context_msg message with application specific information from
1496  *        the other peer
1497  * @param request request from the other peer, use GNUNET_SET_accept
1498  *        to accept it, otherwise the request will be refused
1499  *        Note that we don't use a return value here, as it is also
1500  *        necessary to specify the set we want to do the operation with,
1501  *        whith sometimes can be derived from the context message.
1502  *        Also necessary to specify the timeout.
1503  */
1504 static void
1505 listen_set_union (void *cls,
1506                   const struct GNUNET_PeerIdentity *other_peer,
1507                   const struct GNUNET_MessageHeader *context_msg,
1508                   struct GNUNET_SET_Request *request)
1509 {
1510   struct DirectNeighbor *neighbor = cls;
1511
1512   if (NULL == request)
1513     return; /* why??? */
1514   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1515               "Starting to create consensus with %s\n",
1516               GNUNET_i2s (&neighbor->peer));
1517   if (NULL != neighbor->set_op)
1518   {
1519     GNUNET_SET_operation_cancel (neighbor->set_op);
1520     neighbor->set_op = NULL;
1521   }
1522   if (NULL != neighbor->my_set)
1523   {
1524     GNUNET_SET_destroy (neighbor->my_set);
1525     neighbor->my_set = NULL;
1526   }
1527   neighbor->my_set = GNUNET_SET_create (cfg,
1528                                         GNUNET_SET_OPERATION_UNION);
1529   neighbor->set_op = GNUNET_SET_accept (request,
1530                                         GNUNET_SET_RESULT_ADDED,
1531                                         &handle_set_union_result,
1532                                         neighbor);
1533   neighbor->consensus_insertion_offset = 0;
1534   neighbor->consensus_insertion_distance = 0;
1535   neighbor->consensus_elements = 0;
1536   build_set (neighbor);
1537 }
1538
1539
1540 /**
1541  * Start creating a new DV set union by initiating the connection.
1542  *
1543  * @param cls the `struct DirectNeighbor *` of the peer we're building
1544  *        a routing consensus with
1545  */
1546 static void
1547 initiate_set_union (void *cls)
1548 {
1549   struct DirectNeighbor *neighbor = cls;
1550
1551   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1552               "Initiating SET union with peer `%s'\n",
1553               GNUNET_i2s (&neighbor->peer));
1554   neighbor->initiate_task = NULL;
1555   neighbor->my_set = GNUNET_SET_create (cfg,
1556                                         GNUNET_SET_OPERATION_UNION);
1557   neighbor->set_op = GNUNET_SET_prepare (&neighbor->peer,
1558                                          &neighbor->real_session_id,
1559                                          NULL,
1560                                          GNUNET_SET_RESULT_ADDED,
1561                                          &handle_set_union_result,
1562                                          neighbor);
1563   neighbor->consensus_insertion_offset = 0;
1564   neighbor->consensus_insertion_distance = 0;
1565   neighbor->consensus_elements = 0;
1566   build_set (neighbor);
1567 }
1568
1569
1570 /**
1571  * Check that @a rm is well-formed.
1572  *
1573  * @param cls closure
1574  * @param rm the message
1575  * @return #GNUNET_OK if @a rm is well-formed.
1576  */
1577 static int
1578 check_dv_route_message (void *cls,
1579                         const struct RouteMessage *rm)
1580 {
1581   const struct GNUNET_MessageHeader *payload;
1582
1583   if (ntohs (rm->header.size) < sizeof (struct RouteMessage) + sizeof (struct GNUNET_MessageHeader))
1584   {
1585     GNUNET_break_op (0);
1586     return GNUNET_SYSERR;
1587   }
1588   payload = (const struct GNUNET_MessageHeader *) &rm[1];
1589   if (ntohs (rm->header.size) != sizeof (struct RouteMessage) + ntohs (payload->size))
1590   {
1591     GNUNET_break_op (0);
1592     return GNUNET_SYSERR;
1593   }
1594   return GNUNET_OK;
1595 }
1596
1597
1598 /**
1599  * Core handler for DV data messages.  Whatever this message
1600  * contains all we really have to do is rip it out of its
1601  * DV layering and give it to our pal the DV plugin to report
1602  * in with.
1603  *
1604  * @param cls closure
1605  * @param rm the message
1606  */
1607 static void
1608 handle_dv_route_message (void *cls,
1609                          const struct RouteMessage *rm)
1610 {
1611   struct DirectNeighbor *neighbor = cls;
1612   const struct GNUNET_MessageHeader *payload;
1613   struct Route *route;
1614   struct DirectNeighbor *nneighbor;
1615   struct DirectNeighbor *dn;
1616   struct Target *target;
1617   uint32_t distance;
1618   char me[5];
1619   char src[5];
1620   char prev[5];
1621   char dst[5];
1622
1623   distance = ntohl (rm->distance);
1624   payload = (const struct GNUNET_MessageHeader *) &rm[1];
1625   strncpy (prev, GNUNET_i2s (&neighbor->peer), 4);
1626   strncpy (me, GNUNET_i2s (&my_identity), 4);
1627   strncpy (src, GNUNET_i2s (&rm->sender), 4);
1628   strncpy (dst, GNUNET_i2s (&rm->target), 4);
1629   prev[4] = me[4] = src[4] = dst[4] = '\0';
1630   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1631               "Handling DV message with %u bytes payload of type %u from %s to %s routed by %s to me (%s @ hop %u)\n",
1632               (unsigned int) (ntohs (rm->header.size) - sizeof (struct RouteMessage)),
1633               ntohs (payload->type),
1634               src,
1635               dst,
1636               prev,
1637               me,
1638               (unsigned int) distance + 1);
1639
1640   if (0 == memcmp (&rm->target,
1641                    &my_identity,
1642                    sizeof (struct GNUNET_PeerIdentity)))
1643   {
1644     if ((NULL !=
1645          (dn = GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
1646                                                   &rm->sender))) &&
1647         (DIRECT_NEIGHBOR_COST == dn->distance))
1648     {
1649       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1650                   "Discarding DV message, as %s is a direct neighbor\n",
1651                   GNUNET_i2s (&rm->sender));
1652       GNUNET_STATISTICS_update (stats,
1653                                 "# messages discarded (direct neighbor)",
1654                                 1, GNUNET_NO);
1655       return;
1656     }
1657     /* message is for me, check reverse route! */
1658     route = GNUNET_CONTAINER_multipeermap_get (all_routes,
1659                                                &rm->sender);
1660     if ( (NULL == route) &&
1661          (distance < DEFAULT_FISHEYE_DEPTH) )
1662     {
1663       /* don't have reverse route yet, learn it! */
1664       target = GNUNET_new (struct Target);
1665       target->peer = rm->sender;
1666       target->distance = htonl (distance);
1667       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1668                   "Learning sender %s at distance %u from delivery!\n",
1669                   GNUNET_i2s (&rm->sender),
1670                   (unsigned int) distance + 1);
1671       if (NULL == neighbor->neighbor_table)
1672         neighbor->neighbor_table = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1673       if (GNUNET_YES !=
1674           GNUNET_CONTAINER_multipeermap_put (neighbor->neighbor_table,
1675                                              &target->peer,
1676                                              target,
1677                                              GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
1678       {
1679         GNUNET_break_op (0);
1680         GNUNET_free (target);
1681         return;
1682       }
1683       add_new_route (target, neighbor);
1684     }
1685     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1686                 "Delivering %u bytes from %s to myself!\n",
1687                 ntohs (payload->size),
1688                 GNUNET_i2s (&rm->sender));
1689     send_data_to_plugin (payload,
1690                          &rm->sender,
1691                          1 + distance);
1692     return;
1693   }
1694   if ( (NULL == GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
1695                                                    &rm->sender)) &&
1696        (NULL == GNUNET_CONTAINER_multipeermap_get (all_routes,
1697                                                    &rm->sender)) )
1698   {
1699     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1700                 "Learning sender %s at distance %u from forwarding!\n",
1701                 GNUNET_i2s (&rm->sender),
1702                 1 + distance);
1703     target = GNUNET_new (struct Target);
1704     target->peer = rm->sender;
1705     target->distance = htonl (distance);
1706     if (NULL == neighbor->neighbor_table)
1707       neighbor->neighbor_table = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
1708     if (GNUNET_YES !=
1709         GNUNET_CONTAINER_multipeermap_put (neighbor->neighbor_table,
1710                                            &target->peer,
1711                                            target,
1712                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
1713     {
1714       GNUNET_break_op (0);
1715       GNUNET_free (target);
1716       return;
1717     }
1718     add_new_route (target, neighbor);
1719   }
1720
1721   route = GNUNET_CONTAINER_multipeermap_get (all_routes,
1722                                              &rm->target);
1723   if (NULL == route)
1724   {
1725     nneighbor = GNUNET_CONTAINER_multipeermap_get (direct_neighbors,
1726                                                   &rm->target);
1727     if (NULL == nneighbor)
1728     {
1729       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1730                   "No route to %s, not routing %u bytes!\n",
1731                   GNUNET_i2s (&rm->target),
1732                   ntohs (payload->size));
1733       GNUNET_STATISTICS_update (stats,
1734                                 "# messages discarded (no route)",
1735                                 1, GNUNET_NO);
1736       return;
1737     }
1738   }
1739   else
1740   {
1741     nneighbor = route->next_hop;
1742   }
1743   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1744               "Forwarding message to %s\n",
1745               GNUNET_i2s (&nneighbor->peer));
1746   forward_payload (nneighbor,
1747                    distance + 1,
1748                    &rm->sender,
1749                    &rm->target,
1750                    payload);
1751 }
1752
1753
1754 /**
1755  * Check that @a msg is well-formed
1756  *
1757  * @param cls identification of the client
1758  * @param message the actual message
1759  * @return #GNUNET_OK if @a msg is well-formed
1760  */
1761 static int
1762 check_dv_send_message (void *cls,
1763                        const struct GNUNET_DV_SendMessage *msg)
1764 {
1765   const struct GNUNET_MessageHeader *payload;
1766
1767   if (ntohs (msg->header.size) < sizeof (struct GNUNET_DV_SendMessage) +
1768       sizeof (struct GNUNET_MessageHeader))
1769   {
1770     GNUNET_break (0);
1771     return GNUNET_SYSERR;
1772   }
1773   payload = (const struct GNUNET_MessageHeader *) &msg[1];
1774   if (ntohs (msg->header.size) != sizeof (struct GNUNET_DV_SendMessage) + ntohs (payload->size))
1775   {
1776     GNUNET_break (0);
1777     return GNUNET_SYSERR;
1778   }
1779   return GNUNET_OK;
1780 }
1781
1782
1783 /**
1784  * Service server's handler for message send requests (which come
1785  * bubbling up to us through the DV plugin).
1786  *
1787  * @param cls identification of the client
1788  * @param message the actual message
1789  */
1790 static void
1791 handle_dv_send_message (void *cls,
1792                         const struct GNUNET_DV_SendMessage *msg)
1793 {
1794   struct GNUNET_SERVICE_Client *client = cls;
1795   struct Route *route;
1796   const struct GNUNET_MessageHeader *payload;
1797
1798   payload = (const struct GNUNET_MessageHeader *) &msg[1];
1799   route = GNUNET_CONTAINER_multipeermap_get (all_routes,
1800                                              &msg->target);
1801   if (NULL == route)
1802   {
1803     /* got disconnected */
1804     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1805                 "No route to %s, dropping local message of type %u\n",
1806                 GNUNET_i2s (&msg->target),
1807                 ntohs (payload->type));
1808     GNUNET_STATISTICS_update (stats,
1809                               "# local messages discarded (no route)",
1810                               1, GNUNET_NO);
1811     GNUNET_SERVICE_client_continue (client);
1812     return;
1813   }
1814   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1815               "Forwarding %u bytes of type %u to %s\n",
1816               ntohs (payload->size),
1817               ntohs (payload->type),
1818               GNUNET_i2s (&msg->target));
1819
1820   forward_payload (route->next_hop,
1821                    0 /* first hop, distance is zero */,
1822                    &my_identity,
1823                    &msg->target,
1824                    payload);
1825   GNUNET_SERVICE_client_continue (client);
1826 }
1827
1828
1829 /**
1830  * Cleanup all of the data structures associated with a given neighbor.
1831  *
1832  * @param neighbor neighbor to clean up
1833  */
1834 static void
1835 cleanup_neighbor (struct DirectNeighbor *neighbor)
1836 {
1837   handle_direct_disconnect (neighbor);
1838   GNUNET_assert (GNUNET_YES ==
1839                  GNUNET_CONTAINER_multipeermap_remove (direct_neighbors,
1840                                                        &neighbor->peer,
1841                                                        neighbor));
1842   GNUNET_free (neighbor);
1843 }
1844
1845
1846 /**
1847  * Method called whenever a given peer disconnects.
1848  *
1849  * @param cls closure
1850  * @param peer peer identity this notification is about
1851  * @param internal_cls the corresponding `struct DirectNeighbor`
1852  */
1853 static void
1854 handle_core_disconnect (void *cls,
1855                         const struct GNUNET_PeerIdentity *peer,
1856                         void *internal_cls)
1857 {
1858   struct DirectNeighbor *neighbor = internal_cls;
1859
1860   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1861               "Received core peer disconnect message for peer `%s'!\n",
1862               GNUNET_i2s (peer));
1863   /* Check for disconnect from self message */
1864   if (NULL == neighbor)
1865     return;
1866   GNUNET_break (GNUNET_YES == neighbor->connected);
1867   neighbor->connected = GNUNET_NO;
1868   if (DIRECT_NEIGHBOR_COST == neighbor->distance)
1869   {
1870     GNUNET_STATISTICS_update (stats,
1871                               "# peers connected (1-hop)",
1872                               -1,
1873                               GNUNET_NO);
1874   }
1875   cleanup_neighbor (neighbor);
1876   if (GNUNET_YES == in_shutdown)
1877     return;
1878   schedule_refresh_routes ();
1879 }
1880
1881
1882 /**
1883  * Multipeermap iterator for freeing routes.  Should never be called.
1884  *
1885  * @param cls NULL
1886  * @param key key value stored under
1887  * @param value the route to be freed
1888  * @return #GNUNET_YES to continue iteration, #GNUNET_NO to stop
1889  */
1890 static int
1891 free_route (void *cls,
1892             const struct GNUNET_PeerIdentity *key,
1893             void *value)
1894 {
1895   struct Route *route = value;
1896
1897   GNUNET_break (0);
1898   GNUNET_assert (GNUNET_YES ==
1899                  GNUNET_CONTAINER_multipeermap_remove (all_routes, key, value));
1900   release_route (route);
1901   send_disconnect_to_plugin (&route->target.peer);
1902   GNUNET_free (route);
1903   return GNUNET_YES;
1904 }
1905
1906
1907 /**
1908  * Multipeermap iterator for freeing direct neighbors. Should never be called.
1909  *
1910  * @param cls NULL
1911  * @param key key value stored under
1912  * @param value the direct neighbor to be freed
1913  * @return #GNUNET_YES to continue iteration, #GNUNET_NO to stop
1914  */
1915 static int
1916 free_direct_neighbors (void *cls,
1917                        const struct GNUNET_PeerIdentity *key,
1918                        void *value)
1919 {
1920   struct DirectNeighbor *neighbor = value;
1921
1922   cleanup_neighbor (neighbor);
1923   return GNUNET_YES;
1924 }
1925
1926
1927 /**
1928  * Task run during shutdown.
1929  *
1930  * @param cls unused
1931  */
1932 static void
1933 shutdown_task (void *cls)
1934 {
1935   unsigned int i;
1936
1937   in_shutdown = GNUNET_YES;
1938   GNUNET_assert (NULL != core_api);
1939   GNUNET_CORE_disconnect (core_api);
1940   core_api = NULL;
1941   GNUNET_ATS_performance_done (ats);
1942   ats = NULL;
1943   GNUNET_CONTAINER_multipeermap_iterate (direct_neighbors,
1944                                          &free_direct_neighbors,
1945                                          NULL);
1946   GNUNET_CONTAINER_multipeermap_iterate (all_routes,
1947                                          &free_route,
1948                                          NULL);
1949   GNUNET_CONTAINER_multipeermap_destroy (direct_neighbors);
1950   GNUNET_CONTAINER_multipeermap_destroy (all_routes);
1951   GNUNET_STATISTICS_destroy (stats, GNUNET_NO);
1952   stats = NULL;
1953   GNUNET_notification_context_destroy (nc);
1954   nc = NULL;
1955   for (i=0;i<DEFAULT_FISHEYE_DEPTH;i++)
1956   {
1957     GNUNET_array_grow (consensi[i].targets,
1958                        consensi[i].array_length,
1959                        0);
1960   }
1961   if (NULL != rr_task)
1962   {
1963     GNUNET_SCHEDULER_cancel (rr_task);
1964     rr_task = NULL;
1965   }
1966 }
1967
1968
1969 /**
1970  * Notify newly connected client about an existing route.
1971  *
1972  * @param cls the `struct GNUNET_SERVICE_Client *`
1973  * @param key peer identity
1974  * @param value the `struct Route *`
1975  * @return #GNUNET_OK (continue to iterate)
1976  */
1977 static int
1978 notify_client_about_route (void *cls,
1979                            const struct GNUNET_PeerIdentity *key,
1980                            void *value)
1981 {
1982   struct GNUNET_SERVICE_Client *client = cls;
1983   struct Route *route = value;
1984   struct GNUNET_MQ_Envelope *env;
1985   struct GNUNET_DV_ConnectMessage *cm;
1986
1987   env = GNUNET_MQ_msg (cm,
1988                        GNUNET_MESSAGE_TYPE_DV_CONNECT);
1989   cm->distance = htonl (route->target.distance);
1990   cm->peer = route->target.peer;
1991   GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
1992                   env);
1993   return GNUNET_OK;
1994 }
1995
1996
1997 /**
1998  * Handle START-message.  This is the first message sent to us
1999  * by the client (can only be one!).
2000  *
2001  * @param cls closure (always NULL)
2002  * @param client identification of the client
2003  * @param message the actual message
2004  */
2005 static void
2006 handle_start (void *cls,
2007               const struct GNUNET_MessageHeader *message)
2008 {
2009   struct GNUNET_SERVICE_Client *client = cls;
2010
2011   GNUNET_notification_context_add (nc,
2012                                    GNUNET_SERVICE_client_get_mq (client));
2013   GNUNET_SERVICE_client_continue (client);
2014   GNUNET_CONTAINER_multipeermap_iterate (all_routes,
2015                                          &notify_client_about_route,
2016                                          client);
2017 }
2018
2019
2020 /**
2021  * Called on core init.
2022  *
2023  * @param cls unused
2024  * @param identity this peer's identity
2025  */
2026 static void
2027 core_init (void *cls,
2028            const struct GNUNET_PeerIdentity *identity)
2029 {
2030   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2031               "I am peer: %s\n",
2032               GNUNET_i2s (identity));
2033   my_identity = *identity;
2034 }
2035
2036
2037 /**
2038  * Process dv requests.
2039  *
2040  * @param cls closure
2041  * @param c configuration to use
2042  * @param service the initialized service
2043  */
2044 static void
2045 run (void *cls,
2046      const struct GNUNET_CONFIGURATION_Handle *c,
2047      struct GNUNET_SERVICE_Handle *service)
2048 {
2049   struct GNUNET_MQ_MessageHandler core_handlers[] = {
2050     GNUNET_MQ_hd_var_size (dv_route_message,
2051                            GNUNET_MESSAGE_TYPE_DV_ROUTE,
2052                            struct RouteMessage,
2053                            NULL),
2054     GNUNET_MQ_handler_end ()
2055   };
2056   in_shutdown = GNUNET_NO;
2057   cfg = c;
2058   direct_neighbors = GNUNET_CONTAINER_multipeermap_create (128,
2059                                                            GNUNET_NO);
2060   all_routes = GNUNET_CONTAINER_multipeermap_create (65536,
2061                                                      GNUNET_NO);
2062   core_api = GNUNET_CORE_connect (cfg,
2063                                   NULL,
2064                                   &core_init,
2065                                   &handle_core_connect,
2066                                   &handle_core_disconnect,
2067                                   core_handlers);
2068
2069   if (NULL == core_api)
2070     return;
2071   ats = GNUNET_ATS_performance_init (cfg,
2072                                      &handle_ats_update,
2073                                      NULL);
2074   if (NULL == ats)
2075   {
2076     GNUNET_CORE_disconnect (core_api);
2077     core_api = NULL;
2078     return;
2079   }
2080   nc = GNUNET_notification_context_create (MAX_QUEUE_SIZE_PLUGIN);
2081   stats = GNUNET_STATISTICS_create ("dv",
2082                                     cfg);
2083   GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
2084                                  NULL);
2085 }
2086
2087
2088 /**
2089  * Callback called when a client connects to the service.
2090  *
2091  * @param cls closure for the service
2092  * @param c the new client that connected to the service
2093  * @param mq the message queue used to send messages to the client
2094  * @return @a c
2095  */
2096 static void *
2097 client_connect_cb (void *cls,
2098                    struct GNUNET_SERVICE_Client *c,
2099                    struct GNUNET_MQ_Handle *mq)
2100 {
2101   return c;
2102 }
2103
2104
2105 /**
2106  * Callback called when a client disconnected from the service
2107  *
2108  * @param cls closure for the service
2109  * @param c the client that disconnected
2110  * @param internal_cls should be equal to @a c
2111  */
2112 static void
2113 client_disconnect_cb (void *cls,
2114                       struct GNUNET_SERVICE_Client *c,
2115                       void *internal_cls)
2116 {
2117   GNUNET_assert (c == internal_cls);
2118 }
2119
2120
2121 /**
2122  * Define "main" method using service macro.
2123  */
2124 GNUNET_SERVICE_MAIN
2125 ("dv",
2126  GNUNET_SERVICE_OPTION_NONE,
2127  &run,
2128  &client_connect_cb,
2129  &client_disconnect_cb,
2130  NULL,
2131  GNUNET_MQ_hd_fixed_size (start,
2132                           GNUNET_MESSAGE_TYPE_DV_START,
2133                           struct GNUNET_MessageHeader,
2134                           NULL),
2135  GNUNET_MQ_hd_var_size (dv_send_message,
2136                         GNUNET_MESSAGE_TYPE_DV_SEND,
2137                         struct GNUNET_DV_SendMessage,
2138                         NULL),
2139  GNUNET_MQ_handler_end ());
2140
2141
2142 /* end of gnunet-service-dv.c */