src/mesh/gnunet-service-mesh_peer.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 #include "platform.h"
  23 #include "gnunet_util_lib.h"
  24
  25 #include "gnunet_transport_service.h"
  26 #include "gnunet_core_service.h"
  27 #include "gnunet_statistics_service.h"
  28
  29 #include "mesh_protocol.h"
  30
  31 #include "gnunet-service-mesh_peer.h"
  32 #include "gnunet-service-mesh_dht.h"
  33 #include "gnunet-service-mesh_connection.h"
  34 #include "gnunet-service-mesh_tunnel.h"
  35 #include "mesh_path.h"
  36
  37 #define LOG(level, ...) GNUNET_log_from (level,"mesh-p2p",__VA_ARGS__)
  38
  39 /******************************************************************************/
  40 /********************************   STRUCTS  **********************************/
  41 /******************************************************************************/
  42
  43 /**
  44  * Struct containing info about a queued transmission to this peer
  45  */
  46 struct MeshPeerQueue
  47 {
  48     /**
  49       * DLL next
  50       */
  51   struct MeshPeerQueue *next;
  52
  53     /**
  54       * DLL previous
  55       */
  56   struct MeshPeerQueue *prev;
  57
  58     /**
  59      * Peer this transmission is directed to.
  60      */
  61   struct MeshPeer *peer;
  62
  63     /**
  64      * Connection this message belongs to.
  65      */
  66   struct MeshConnection *c;
  67
  68     /**
  69      * Is FWD in c?
  70      */
  71   int fwd;
  72
  73     /**
  74      * Pointer to info stucture used as cls.
  75      */
  76   void *cls;
  77
  78     /**
  79      * Type of message
  80      */
  81   uint16_t type;
  82
  83     /**
  84      * Size of the message
  85      */
  86   size_t size;
  87
  88     /**
  89      * Set when this message starts waiting for CORE.
  90      */
  91   struct GNUNET_TIME_Absolute start_waiting;
  92
  93     /**
  94      * Function to call on sending.
  95      */
  96   GMP_sent callback;
  97
  98     /**
  99      * Closure for callback.
 100      */
 101   void *callback_cls;
 102 };
 103
 104 /**
 105  * Struct containing all information regarding a given peer
 106  */
 107 struct MeshPeer
 108 {
 109     /**
 110      * ID of the peer
 111      */
 112   GNUNET_PEER_Id id;
 113
 114     /**
 115      * Last time we heard from this peer
 116      */
 117   struct GNUNET_TIME_Absolute last_contact;
 118
 119     /**
 120      * Paths to reach the peer, ordered by ascending hop count
 121      */
 122   struct MeshPeerPath *path_head;
 123
 124     /**
 125      * Paths to reach the peer, ordered by ascending hop count
 126      */
 127   struct MeshPeerPath *path_tail;
 128
 129     /**
 130      * Handle to stop the DHT search for paths to this peer
 131      */
 132   struct GMD_search_handle *search_h;
 133
 134     /**
 135      * Tunnel to this peer, if any.
 136      */
 137   struct MeshTunnel3 *tunnel;
 138
 139     /**
 140      * Connections that go through this peer, indexed by tid;
 141      */
 142   struct GNUNET_CONTAINER_MultiHashMap *connections;
 143
 144     /**
 145      * Handle for queued transmissions
 146      */
 147   struct GNUNET_CORE_TransmitHandle *core_transmit;
 148
 149   /**
 150    * Transmission queue to core DLL head
 151    */
 152   struct MeshPeerQueue *queue_head;
 153
 154   /**
 155    * Transmission queue to core DLL tail
 156    */
 157   struct MeshPeerQueue *queue_tail;
 158
 159   /**
 160    * How many messages are in the queue to this peer.
 161    */
 162   unsigned int queue_n;
 163
 164   /**
 165    * Hello message.
 166    */
 167   struct GNUNET_HELLO_Message* hello;
 168 };
 169
 170
 171 /******************************************************************************/
 172 /*******************************   GLOBALS  ***********************************/
 173 /******************************************************************************/
 174
 175 /**
 176  * Global handle to the statistics service.
 177  */
 178 extern struct GNUNET_STATISTICS_Handle *stats;
 179
 180 /**
 181  * Local peer own ID (full value).
 182  */
 183 extern struct GNUNET_PeerIdentity my_full_id;
 184
 185 /**
 186  * Local peer own ID (short)
 187  */
 188 extern GNUNET_PEER_Id myid;
 189
 190 /**
 191  * Peers known, indexed by PeerIdentity (MeshPeer).
 192  */
 193 static struct GNUNET_CONTAINER_MultiPeerMap *peers;
 194
 195 /**
 196  * How many peers do we want to remember?
 197  */
 198 static unsigned long long max_peers;
 199
 200 /**
 201  * Percentage of messages that will be dropped (for test purposes only).
 202  */
 203 static unsigned long long drop_percent;
 204
 205 /**
 206  * Handle to communicate with core.
 207  */
 208 static struct GNUNET_CORE_Handle *core_handle;
 209
 210 /**
 211  * Handle to try to start new connections.
 212  */
 213 static struct GNUNET_TRANSPORT_Handle *transport_handle;
 214
 215 /******************************************************************************/
 216 /***************************** CORE CALLBACKS *********************************/
 217 /******************************************************************************/
 218
 219
 220 /**
 221  * Iterator to notify all connections of a broken link. Mark connections
 222  * to destroy after all traffic has been sent.
 223  *
 224  * @param cls Closure (peer disconnected).
 225  * @param key Current key code (peer id).
 226  * @param value Value in the hash map (connection).
 227  *
 228  * @return #GNUNET_YES to continue to iterate.
 229  */
 230 static int
 231 notify_broken (void *cls,
 232                const struct GNUNET_HashCode *key,
 233                void *value)
 234 {
 235   struct MeshPeer *peer = cls;
 236   struct MeshConnection *c = value;
 237
 238   LOG (GNUNET_ERROR_TYPE_DEBUG, "  notifying %s due to %s\n",
 239        GMC_2s (c), GMP_2s (peer));
 240   GMC_notify_broken (c, peer);
 241
 242   return GNUNET_YES;
 243 }
 244
 245
 246 /**
 247  * Remove the direct path to the peer.
 248  *
 249  * @param peer Peer to remove the direct path from.
 250  *
 251  */
 252 static struct MeshPeerPath *
 253 pop_direct_path (struct MeshPeer *peer)
 254 {
 255   struct MeshPeerPath *iter;
 256
 257   for (iter = peer->path_head; NULL != iter; iter = iter->next)
 258   {
 259     if (2 <= iter->length)
 260     {
 261       GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, iter);
 262       return iter;
 263     }
 264   }
 265   return NULL;
 266 }
 267
 268
 269
 270 /**
 271  * Method called whenever a given peer connects.
 272  *
 273  * @param cls closure
 274  * @param peer peer identity this notification is about
 275  */
 276 static void
 277 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
 278 {
 279   struct MeshPeer *mp;
 280   struct MeshPeerPath *path;
 281   char own_id[16];
 282
 283   strncpy (own_id, GNUNET_i2s (&my_full_id), 15);
 284   mp = GMP_get (peer);
 285   if (myid == mp->id)
 286   {
 287     LOG (GNUNET_ERROR_TYPE_INFO, "CONNECTED %s (self)\n", own_id);
 288     path = path_new (1);
 289   }
 290   else
 291   {
 292     LOG (GNUNET_ERROR_TYPE_DEBUG, "CONNECTED %s <= %s\n",
 293          own_id, GNUNET_i2s (peer));
 294     path = path_new (2);
 295     path->peers[1] = mp->id;
 296     GNUNET_PEER_change_rc (mp->id, 1);
 297     GNUNET_STATISTICS_update (stats, "# peers", 1, GNUNET_NO);
 298   }
 299   path->peers[0] = myid;
 300   GNUNET_PEER_change_rc (myid, 1);
 301   GMP_add_path (mp, path, GNUNET_YES);
 302
 303   mp->connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES);
 304   return;
 305 }
 306
 307
 308 /**
 309  * Method called whenever a peer disconnects.
 310  *
 311  * @param cls closure
 312  * @param peer peer identity this notification is about
 313  */
 314 static void
 315 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
 316 {
 317   struct MeshPeer *p;
 318   struct MeshPeerPath *direct_path;
 319   char own_id[16];
 320
 321   strncpy (own_id, GNUNET_i2s (&my_full_id), 15);
 322   p = GNUNET_CONTAINER_multipeermap_get (peers, peer);
 323   if (NULL == p)
 324   {
 325     GNUNET_break (0);
 326     return;
 327   }
 328   if (myid == p->id)
 329     LOG (GNUNET_ERROR_TYPE_INFO, "DISCONNECTED %s (self)\n", own_id);
 330   else
 331     LOG (GNUNET_ERROR_TYPE_DEBUG, "DISCONNECTED %s <= %s\n",
 332          own_id, GNUNET_i2s (peer));
 333   direct_path = pop_direct_path (p);
 334   GNUNET_CONTAINER_multihashmap_iterate (p->connections, &notify_broken, p);
 335   GNUNET_CONTAINER_multihashmap_destroy (p->connections);
 336   p->connections = NULL;
 337   if (NULL != p->core_transmit)
 338     {
 339       GNUNET_CORE_notify_transmit_ready_cancel (p->core_transmit);
 340       p->core_transmit = NULL;
 341     }
 342   GNUNET_STATISTICS_update (stats, "# peers", -1, GNUNET_NO);
 343
 344   path_destroy (direct_path);
 345   return;
 346 }
 347
 348
 349 /**
 350  * Functions to handle messages from core
 351  */
 352 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
 353   {&GMC_handle_create, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE, 0},
 354   {&GMC_handle_confirm, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
 355     sizeof (struct GNUNET_MESH_ConnectionACK)},
 356   {&GMC_handle_broken, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN,
 357     sizeof (struct GNUNET_MESH_ConnectionBroken)},
 358   {&GMC_handle_destroy, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY,
 359     sizeof (struct GNUNET_MESH_ConnectionDestroy)},
 360   {&GMC_handle_ack, GNUNET_MESSAGE_TYPE_MESH_ACK,
 361     sizeof (struct GNUNET_MESH_ACK)},
 362   {&GMC_handle_poll, GNUNET_MESSAGE_TYPE_MESH_POLL,
 363     sizeof (struct GNUNET_MESH_Poll)},
 364   {&GMC_handle_encrypted, GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED, 0},
 365   {&GMC_handle_kx, GNUNET_MESSAGE_TYPE_MESH_KX, 0},
 366   {NULL, 0, 0}
 367 };
 368
 369
 370 /**
 371  * To be called on core init/fail.
 372  *
 373  * @param cls Closure (config)
 374  * @param identity the public identity of this peer
 375  */
 376 static void
 377 core_init (void *cls,
 378            const struct GNUNET_PeerIdentity *identity)
 379 {
 380   const struct GNUNET_CONFIGURATION_Handle *c = cls;
 381   static int i = 0;
 382
 383   LOG (GNUNET_ERROR_TYPE_DEBUG, "Core init\n");
 384   if (0 != memcmp (identity, &my_full_id, sizeof (my_full_id)))
 385   {
 386     LOG (GNUNET_ERROR_TYPE_ERROR, _("Wrong CORE service\n"));
 387     LOG (GNUNET_ERROR_TYPE_ERROR, " core id %s\n", GNUNET_i2s (identity));
 388     LOG (GNUNET_ERROR_TYPE_ERROR, " my id %s\n", GNUNET_i2s (&my_full_id));
 389     GNUNET_CORE_disconnect (core_handle);
 390     core_handle = GNUNET_CORE_connect (c, /* Main configuration */
 391                                        NULL,      /* Closure passed to MESH functions */
 392                                        &core_init,        /* Call core_init once connected */
 393                                        &core_connect,     /* Handle connects */
 394                                        &core_disconnect,  /* remove peers on disconnects */
 395                                        NULL,      /* Don't notify about all incoming messages */
 396                                        GNUNET_NO, /* For header only in notification */
 397                                        NULL,      /* Don't notify about all outbound messages */
 398                                        GNUNET_NO, /* For header-only out notification */
 399                                        core_handlers);    /* Register these handlers */
 400     if (10 < i++)
 401       GNUNET_abort();
 402   }
 403   GML_start ();
 404   return;
 405 }
 406
 407 /**
 408   * Core callback to write a pre-constructed data packet to core buffer
 409   *
 410   * @param cls Closure (MeshTransmissionDescriptor with data in "data" member).
 411   * @param size Number of bytes available in buf.
 412   * @param buf Where the to write the message.
 413   *
 414   * @return number of bytes written to buf
 415   */
 416 static size_t
 417 send_core_data_raw (void *cls, size_t size, void *buf)
 418 {
 419   struct GNUNET_MessageHeader *msg = cls;
 420   size_t total_size;
 421
 422   GNUNET_assert (NULL != msg);
 423   total_size = ntohs (msg->size);
 424
 425   if (total_size > size)
 426   {
 427     GNUNET_break (0);
 428     return 0;
 429   }
 430   memcpy (buf, msg, total_size);
 431   GNUNET_free (cls);
 432   return total_size;
 433 }
 434
 435
 436 /**
 437  * Function to send a create connection message to a peer.
 438  *
 439  * @param c Connection to create.
 440  * @param size number of bytes available in buf
 441  * @param buf where the callee should write the message
 442  * @return number of bytes written to buf
 443  */
 444 static size_t
 445 send_core_connection_create (struct MeshConnection *c, size_t size, void *buf)
 446 {
 447   struct GNUNET_MESH_ConnectionCreate *msg;
 448   struct GNUNET_PeerIdentity *peer_ptr;
 449   const struct MeshPeerPath *p = GMC_get_path (c);
 450   size_t size_needed;
 451   int i;
 452
 453   if (NULL == p)
 454     return 0;
 455
 456   LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION CREATE...\n");
 457   size_needed =
 458       sizeof (struct GNUNET_MESH_ConnectionCreate) +
 459       p->length * sizeof (struct GNUNET_PeerIdentity);
 460
 461   if (size < size_needed || NULL == buf)
 462   {
 463     GNUNET_break (0);
 464     return 0;
 465   }
 466   msg = (struct GNUNET_MESH_ConnectionCreate *) buf;
 467   msg->header.size = htons (size_needed);
 468   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE);
 469   msg->cid = *GMC_get_id (c);
 470
 471   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
 472   for (i = 0; i < p->length; i++)
 473   {
 474     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
 475   }
 476
 477   LOG (GNUNET_ERROR_TYPE_DEBUG,
 478        "CONNECTION CREATE (%u bytes long) sent!\n",
 479        size_needed);
 480   return size_needed;
 481 }
 482
 483
 484 /**
 485  * Creates a path ack message in buf and frees all unused resources.
 486  *
 487  * @param c Connection to send an ACK on.
 488  * @param size number of bytes available in buf
 489  * @param buf where the callee should write the message
 490  *
 491  * @return number of bytes written to buf
 492  */
 493 static size_t
 494 send_core_connection_ack (struct MeshConnection *c, size_t size, void *buf)
 495 {
 496   struct GNUNET_MESH_ConnectionACK *msg = buf;
 497
 498   LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION ACK...\n");
 499   if (sizeof (struct GNUNET_MESH_ConnectionACK) > size)
 500   {
 501     GNUNET_break (0);
 502     return 0;
 503   }
 504   msg->header.size = htons (sizeof (struct GNUNET_MESH_ConnectionACK));
 505   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK);
 506   msg->cid = *GMC_get_id (c);
 507
 508   LOG (GNUNET_ERROR_TYPE_DEBUG, "CONNECTION ACK sent!\n");
 509   return sizeof (struct GNUNET_MESH_ConnectionACK);
 510 }
 511
 512
 513 /******************************************************************************/
 514 /********************************   STATIC  ***********************************/
 515 /******************************************************************************/
 516
 517
 518 /**
 519  * Get priority for a queued message.
 520  *
 521  * @param q Queued message
 522  *
 523  * @return CORE priority to use.
 524  */
 525 static enum GNUNET_CORE_Priority
 526 get_priority (struct MeshPeerQueue *q)
 527 {
 528   enum GNUNET_CORE_Priority low;
 529   enum GNUNET_CORE_Priority high;
 530
 531   if (NULL == q)
 532   {
 533     GNUNET_break (0);
 534     return GNUNET_CORE_PRIO_BACKGROUND;
 535   }
 536
 537   /* Relayed traffic has lower priority, our own traffic has higher */
 538   if (NULL == q->c || GNUNET_NO == GMC_is_origin (q->c, q->fwd))
 539   {
 540     low = GNUNET_CORE_PRIO_BEST_EFFORT;
 541     high = GNUNET_CORE_PRIO_URGENT;
 542   }
 543   else
 544   {
 545     low = GNUNET_CORE_PRIO_URGENT;
 546     high = GNUNET_CORE_PRIO_CRITICAL_CONTROL;
 547   }
 548
 549   /* Bulky payload has lower priority, control traffic has higher. */
 550   if (GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED == q->type)
 551     return low;
 552   else
 553     return high;
 554 }
 555
 556
 557 /**
 558  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
 559  *
 560  * @param cls closure
 561  * @param key current key code
 562  * @param value value in the hash map
 563  * @return #GNUNET_YES if we should continue to iterate,
 564  *         #GNUNET_NO if not.
 565  */
 566 static int
 567 shutdown_tunnel (void *cls,
 568                  const struct GNUNET_PeerIdentity *key,
 569                  void *value)
 570 {
 571   struct MeshPeer *p = value;
 572   struct MeshTunnel3 *t = p->tunnel;
 573
 574   if (NULL != t)
 575     GMT_destroy (t);
 576   return GNUNET_YES;
 577 }
 578
 579
 580 /**
 581  * Destroy the peer_info and free any allocated resources linked to it
 582  *
 583  * @param peer The peer_info to destroy.
 584  *
 585  * @return GNUNET_OK on success
 586  */
 587 static int
 588 peer_destroy (struct MeshPeer *peer)
 589 {
 590   struct GNUNET_PeerIdentity id;
 591   struct MeshPeerPath *p;
 592   struct MeshPeerPath *nextp;
 593
 594   GNUNET_PEER_resolve (peer->id, &id);
 595   GNUNET_PEER_change_rc (peer->id, -1);
 596
 597   LOG (GNUNET_ERROR_TYPE_WARNING, "destroying peer %s\n", GNUNET_i2s (&id));
 598
 599   if (GNUNET_YES !=
 600     GNUNET_CONTAINER_multipeermap_remove (peers, &id, peer))
 601   {
 602     GNUNET_break (0);
 603     LOG (GNUNET_ERROR_TYPE_WARNING, " not in peermap!!\n");
 604   }
 605   if (NULL != peer->search_h)
 606   {
 607     GMD_search_stop (peer->search_h);
 608   }
 609   p = peer->path_head;
 610   while (NULL != p)
 611   {
 612     nextp = p->next;
 613     GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
 614     path_destroy (p);
 615     p = nextp;
 616   }
 617   GMT_destroy_empty (peer->tunnel);
 618   GNUNET_free (peer);
 619   return GNUNET_OK;
 620 }
 621
 622
 623 /**
 624  * Returns if peer is used (has a tunnel or is neighbor).
 625  *
 626  * @param peer Peer to check.
 627  *
 628  * @return #GNUNET_YES if peer is in use.
 629  */
 630 static int
 631 peer_is_used (struct MeshPeer *peer)
 632 {
 633   struct MeshPeerPath *p;
 634
 635   if (NULL != peer->tunnel)
 636     return GNUNET_YES;
 637
 638   for (p = peer->path_head; NULL != p; p = p->next)
 639   {
 640     if (p->length < 3)
 641       return GNUNET_YES;
 642   }
 643     return GNUNET_NO;
 644 }
 645
 646
 647 /**
 648  * Iterator over all the peers to get the oldest timestamp.
 649  *
 650  * @param cls Closure (unsued).
 651  * @param key ID of the peer.
 652  * @param value Peer_Info of the peer.
 653  */
 654 static int
 655 peer_get_oldest (void *cls,
 656                  const struct GNUNET_PeerIdentity *key,
 657                  void *value)
 658 {
 659   struct MeshPeer *p = value;
 660   struct GNUNET_TIME_Absolute *abs = cls;
 661
 662   /* Don't count active peers */
 663   if (GNUNET_YES == peer_is_used (p))
 664     return GNUNET_YES;
 665
 666   if (abs->abs_value_us < p->last_contact.abs_value_us)
 667     abs->abs_value_us = p->last_contact.abs_value_us;
 668
 669   return GNUNET_YES;
 670 }
 671
 672
 673 /**
 674  * Iterator over all the peers to remove the oldest entry.
 675  *
 676  * @param cls Closure (unsued).
 677  * @param key ID of the peer.
 678  * @param value Peer_Info of the peer.
 679  */
 680 static int
 681 peer_timeout (void *cls,
 682               const struct GNUNET_PeerIdentity *key,
 683               void *value)
 684 {
 685   struct MeshPeer *p = value;
 686   struct GNUNET_TIME_Absolute *abs = cls;
 687
 688   LOG (GNUNET_ERROR_TYPE_WARNING,
 689        "peer %s timeout\n", GNUNET_i2s (key));
 690
 691   if (p->last_contact.abs_value_us == abs->abs_value_us &&
 692       GNUNET_NO == peer_is_used (p))
 693   {
 694     peer_destroy (p);
 695     return GNUNET_NO;
 696   }
 697     return GNUNET_YES;
 698 }
 699
 700
 701 /**
 702  * Delete oldest unused peer.
 703  */
 704 static void
 705 peer_delete_oldest (void)
 706 {
 707   struct GNUNET_TIME_Absolute abs;
 708
 709   abs = GNUNET_TIME_UNIT_FOREVER_ABS;
 710
 711   GNUNET_CONTAINER_multipeermap_iterate (peers,
 712                                          &peer_get_oldest,
 713                                          &abs);
 714   GNUNET_CONTAINER_multipeermap_iterate (peers,
 715                                          &peer_timeout,
 716                                          &abs);
 717 }
 718
 719
 720 /**
 721  * Choose the best (yet unused) path towards a peer,
 722  * considering the tunnel properties.
 723  *
 724  * @param peer The destination peer.
 725  *
 726  * @return Best current known path towards the peer, if any.
 727  */
 728 static struct MeshPeerPath *
 729 peer_get_best_path (const struct MeshPeer *peer)
 730 {
 731   struct MeshPeerPath *best_p;
 732   struct MeshPeerPath *p;
 733   unsigned int best_cost;
 734   unsigned int cost;
 735
 736   best_cost = UINT_MAX;
 737   best_p = NULL;
 738   for (p = peer->path_head; NULL != p; p = p->next)
 739   {
 740     if (GNUNET_YES == GMT_is_path_used (peer->tunnel, p))
 741       continue; /* If path is already in use, skip it. */
 742
 743     if (GNUNET_NO == path_is_valid (p))
 744       continue; /* Don't use invalid paths. */
 745
 746     if ((cost = GMT_get_path_cost (peer->tunnel, p)) < best_cost)
 747     {
 748       best_cost = cost;
 749       best_p = p;
 750     }
 751   }
 752   return best_p;
 753 }
 754
 755
 756 static int
 757 queue_is_sendable (struct MeshPeerQueue *q)
 758 {
 759   /* Is PID-independent? */
 760   switch (q->type)
 761   {
 762     case GNUNET_MESSAGE_TYPE_MESH_ACK:
 763     case GNUNET_MESSAGE_TYPE_MESH_POLL:
 764     case GNUNET_MESSAGE_TYPE_MESH_KX:
 765     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
 766     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
 767     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
 768     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
 769     case GNUNET_MESSAGE_TYPE_MESH_KEEPALIVE:
 770       return GNUNET_YES;
 771
 772     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
 773       break;
 774
 775     default:
 776       GNUNET_break (0);
 777   }
 778
 779   if (GMC_is_sendable (q->c, q->fwd))
 780     return GNUNET_YES;
 781
 782   return GNUNET_NO;
 783 }
 784
 785
 786 /**
 787  * Get first sendable message.
 788  *
 789  * @param peer The destination peer.
 790  *
 791  * @return Best current known path towards the peer, if any.
 792  */
 793 static struct MeshPeerQueue *
 794 peer_get_first_message (const struct MeshPeer *peer)
 795 {
 796   struct MeshPeerQueue *q;
 797
 798   for (q = peer->queue_head; NULL != q; q = q->next)
 799   {
 800     if (queue_is_sendable (q))
 801       return q;
 802   }
 803
 804   return NULL;
 805 }
 806
 807
 808 /**
 809  * Function to process paths received for a new peer addition. The recorded
 810  * paths form the initial tunnel, which can be optimized later.
 811  * Called on each result obtained for the DHT search.
 812  *
 813  * @param cls closure
 814  * @param path
 815  */
 816 static void
 817 search_handler (void *cls, const struct MeshPeerPath *path)
 818 {
 819   struct MeshPeer *peer = cls;
 820   unsigned int connection_count;
 821
 822   GMP_add_path_to_all (path, GNUNET_NO);
 823
 824   /* Count connections */
 825   connection_count = GMT_count_connections (peer->tunnel);
 826
 827   /* If we already have 3 (or more (?!)) connections, it's enough */
 828   if (3 <= connection_count)
 829     return;
 830
 831   if (MESH_TUNNEL3_SEARCHING == GMT_get_cstate (peer->tunnel))
 832   {
 833     LOG (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n");
 834     GMP_connect (peer);
 835   }
 836   return;
 837 }
 838
 839
 840
 841 /**
 842  * Core callback to write a queued packet to core buffer
 843  *
 844  * @param cls Closure (peer info).
 845  * @param size Number of bytes available in buf.
 846  * @param buf Where the to write the message.
 847  *
 848  * @return number of bytes written to buf
 849  */
 850 static size_t
 851 queue_send (void *cls, size_t size, void *buf)
 852 {
 853   struct MeshPeer *peer = cls;
 854   struct MeshConnection *c;
 855   struct MeshPeerQueue *queue;
 856   const struct GNUNET_PeerIdentity *dst_id;
 857   size_t data_size;
 858
 859   peer->core_transmit = NULL;
 860   LOG (GNUNET_ERROR_TYPE_DEBUG, "* Queue send towards %s (max %u)\n",
 861        GMP_2s (peer), size);
 862
 863   if (NULL == buf || 0 == size)
 864   {
 865     LOG (GNUNET_ERROR_TYPE_DEBUG, "* Buffer size 0.\n");
 866     return 0;
 867   }
 868
 869   /* Initialize */
 870   queue = peer_get_first_message (peer);
 871   if (NULL == queue)
 872   {
 873     GNUNET_break (0); /* Core tmt_rdy should've been canceled */
 874     return 0;
 875   }
 876   c = queue->c;
 877
 878   dst_id = GNUNET_PEER_resolve2 (peer->id);
 879   LOG (GNUNET_ERROR_TYPE_DEBUG, "*   on connection %s\n", GMC_2s (c));
 880   /* Check if buffer size is enough for the message */
 881   if (queue->size > size)
 882   {
 883     LOG (GNUNET_ERROR_TYPE_WARNING, "not enough room (%u vs %u), reissue\n",
 884          queue->size, size);
 885     peer->core_transmit =
 886       GNUNET_CORE_notify_transmit_ready (core_handle,
 887                                          GNUNET_NO, get_priority (queue),
 888                                          GNUNET_TIME_UNIT_FOREVER_REL,
 889                                          dst_id,
 890                                          queue->size,
 891                                          &queue_send,
 892                                          peer);
 893     return 0;
 894   }
 895   LOG (GNUNET_ERROR_TYPE_DEBUG, "*   size %u ok\n", queue->size);
 896
 897   /* Fill buf */
 898   switch (queue->type)
 899   {
 900     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
 901     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
 902     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
 903     case GNUNET_MESSAGE_TYPE_MESH_KX:
 904     case GNUNET_MESSAGE_TYPE_MESH_ACK:
 905     case GNUNET_MESSAGE_TYPE_MESH_POLL:
 906       LOG (GNUNET_ERROR_TYPE_DEBUG, "*   raw: %s\n", GM_m2s (queue->type));
 907       data_size = send_core_data_raw (queue->cls, size, buf);
 908       break;
 909     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
 910       LOG (GNUNET_ERROR_TYPE_DEBUG, "*   path create\n");
 911       if (GMC_is_origin (c, GNUNET_YES))
 912         data_size = send_core_connection_create (queue->c, size, buf);
 913       else
 914         data_size = send_core_data_raw (queue->cls, size, buf);
 915       break;
 916     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
 917       LOG (GNUNET_ERROR_TYPE_DEBUG, "*   path ack\n");
 918       if (GMC_is_origin (c, GNUNET_NO) ||
 919           GMC_is_origin (c, GNUNET_YES))
 920         data_size = send_core_connection_ack (queue->c, size, buf);
 921       else
 922         data_size = send_core_data_raw (queue->cls, size, buf);
 923       break;
 924     case GNUNET_MESSAGE_TYPE_MESH_DATA:
 925     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
 926     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
 927       /* This should be encapsulted */
 928       GNUNET_break (0);
 929       data_size = 0;
 930       break;
 931     default:
 932       GNUNET_break (0);
 933       LOG (GNUNET_ERROR_TYPE_WARNING, "*   type unknown: %u\n", queue->type);
 934       data_size = 0;
 935   }
 936
 937   if (0 < drop_percent &&
 938       GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent)
 939   {
 940     LOG (GNUNET_ERROR_TYPE_WARNING, "DD %s on connection\n",
 941          GM_m2s (queue->type), GMC_2s (c));
 942     data_size = 0;
 943   }
 944   else
 945   {
 946     LOG (GNUNET_ERROR_TYPE_INFO,
 947         "ss %s on connection %s (%p) %s (size %u)\n",
 948         GM_m2s (queue->type), GMC_2s (c), c, GM_f2s (queue->fwd), data_size);
 949   }
 950
 951   /* Free queue, but cls was freed by send_core_* */
 952   GMP_queue_destroy (queue, GNUNET_NO);
 953
 954   /* If more data in queue, send next */
 955   queue = peer_get_first_message (peer);
 956   if (NULL != queue)
 957   {
 958     LOG (GNUNET_ERROR_TYPE_DEBUG, "*   more data!\n");
 959     if (NULL == peer->core_transmit)
 960     {
 961       peer->core_transmit =
 962           GNUNET_CORE_notify_transmit_ready (core_handle,
 963                                              GNUNET_NO, get_priority (queue),
 964                                              GNUNET_TIME_UNIT_FOREVER_REL,
 965                                              dst_id,
 966                                              queue->size,
 967                                              &queue_send,
 968                                              peer);
 969       queue->start_waiting = GNUNET_TIME_absolute_get ();
 970     }
 971     else
 972     {
 973       LOG (GNUNET_ERROR_TYPE_DEBUG,
 974                   "*   tmt rdy called somewhere else\n");
 975     }
 976 //     GMC_start_poll (); FIXME needed?
 977   }
 978   else
 979   {
 980 //     GMC_stop_poll(); FIXME needed?
 981   }
 982
 983   LOG (GNUNET_ERROR_TYPE_DEBUG, "*  Return %d\n", data_size);
 984   return data_size;
 985 }
 986
 987
 988 /******************************************************************************/
 989 /********************************    API    ***********************************/
 990 /******************************************************************************/
 991
 992
 993 /**
 994  * Free a transmission that was already queued with all resources
 995  * associated to the request.
 996  *
 997  * @param queue Queue handler to cancel.
 998  * @param clear_cls Is it necessary to free associated cls?
 999  */
1000 void
1001 GMP_queue_destroy (struct MeshPeerQueue *queue, int clear_cls)
1002 {
1003   struct MeshPeer *peer;
1004
1005   peer = queue->peer;
1006
1007   if (GNUNET_YES == clear_cls)
1008   {
1009     LOG (GNUNET_ERROR_TYPE_DEBUG, "#   queue destroy type %s\n",
1010                 GM_m2s (queue->type));
1011     switch (queue->type)
1012     {
1013       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
1014         LOG (GNUNET_ERROR_TYPE_INFO, "destroying a DESTROY message\n");
1015         /* fall through */
1016       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
1017       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
1018       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
1019       case GNUNET_MESSAGE_TYPE_MESH_KEEPALIVE:
1020       case GNUNET_MESSAGE_TYPE_MESH_KX:
1021       case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
1022       case GNUNET_MESSAGE_TYPE_MESH_ACK:
1023       case GNUNET_MESSAGE_TYPE_MESH_POLL:
1024         GNUNET_free_non_null (queue->cls);
1025         break;
1026
1027       default:
1028         GNUNET_break (0);
1029         LOG (GNUNET_ERROR_TYPE_ERROR, "#   type %s unknown!\n",
1030                     GM_m2s (queue->type));
1031     }
1032   }
1033   GNUNET_CONTAINER_DLL_remove (peer->queue_head, peer->queue_tail, queue);
1034
1035   if (queue->type != GNUNET_MESSAGE_TYPE_MESH_ACK &&
1036       queue->type != GNUNET_MESSAGE_TYPE_MESH_POLL)
1037   {
1038     peer->queue_n--;
1039   }
1040
1041   if (NULL != queue->callback)
1042   {
1043     LOG (GNUNET_ERROR_TYPE_DEBUG, "#   Calling callback\n");
1044     queue->callback (queue->callback_cls,
1045                      queue->c, queue->type,
1046                      queue->fwd, queue->size,
1047                      GNUNET_TIME_absolute_get_duration (queue->start_waiting));
1048   }
1049
1050   GNUNET_free (queue);
1051 }
1052
1053
1054 /**
1055  * @brief Queue and pass message to core when possible.
1056  *
1057  * @param peer Peer towards which to queue the message.
1058  * @param cls Closure (@c type dependant). It will be used by queue_send to
1059  *            build the message to be sent if not already prebuilt.
1060  * @param type Type of the message, 0 for a raw message.
1061  * @param size Size of the message.
1062  * @param c Connection this message belongs to (can be NULL).
1063  * @param fwd Is this a message going root->dest? (FWD ACK are NOT FWD!)
1064  * @param cont Continuation to be called once CORE has taken the message.
1065  * @param cont_cls Closure for @c cont.
1066  *
1067  * @return Handle to cancel the message before it is sent. Once cont is called
1068  *         message has been sent and therefore the handle is no longer valid.
1069  */
1070 struct MeshPeerQueue *
1071 GMP_queue_add (struct MeshPeer *peer, void *cls, uint16_t type, size_t size,
1072                struct MeshConnection *c, int fwd,
1073                GMP_sent cont, void *cont_cls)
1074 {
1075   struct MeshPeerQueue *queue;
1076   int priority;
1077   int call_core;
1078
1079   LOG (GNUNET_ERROR_TYPE_INFO, "qq %s on connection %s (%p) %s towards %s (size %u)\n",
1080        GM_m2s (type), GMC_2s (c), c, GM_f2s (fwd), GMP_2s(peer), size);
1081
1082   if (NULL == peer->connections)
1083   {
1084     /* We are not connected to this peer, ignore request. */
1085     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING %s not a neighbor\n", GMP_2s (peer));
1086     GNUNET_STATISTICS_update (stats, "# messages dropped due to wrong hop", 1,
1087                               GNUNET_NO);
1088     return NULL;
1089   }
1090
1091   priority = 0;
1092
1093   if (GNUNET_MESSAGE_TYPE_MESH_POLL == type ||
1094       GNUNET_MESSAGE_TYPE_MESH_ACK == type)
1095   {
1096     priority = 100;
1097   }
1098
1099   LOG (GNUNET_ERROR_TYPE_DEBUG, "priority %d\n", priority);
1100
1101   call_core = NULL == c ? GNUNET_YES : GMC_is_sendable (c, fwd);
1102   queue = GNUNET_new (struct MeshPeerQueue);
1103   queue->cls = cls;
1104   queue->type = type;
1105   queue->size = size;
1106   queue->peer = peer;
1107   queue->c = c;
1108   queue->fwd = fwd;
1109   queue->callback = cont;
1110   queue->callback_cls = cont_cls;
1111   if (100 > priority)
1112   {
1113     GNUNET_CONTAINER_DLL_insert_tail (peer->queue_head, peer->queue_tail, queue);
1114     peer->queue_n++;
1115   }
1116   else
1117   {
1118     GNUNET_CONTAINER_DLL_insert (peer->queue_head, peer->queue_tail, queue);
1119     call_core = GNUNET_YES;
1120   }
1121
1122   if (NULL == peer->core_transmit && GNUNET_YES == call_core)
1123   {
1124     LOG (GNUNET_ERROR_TYPE_DEBUG,
1125                 "calling core tmt rdy towards %s for %u bytes\n",
1126                 GMP_2s (peer), size);
1127     peer->core_transmit =
1128         GNUNET_CORE_notify_transmit_ready (core_handle,
1129                                            GNUNET_NO, get_priority (queue),
1130                                            GNUNET_TIME_UNIT_FOREVER_REL,
1131                                            GNUNET_PEER_resolve2 (peer->id),
1132                                            size,
1133                                            &queue_send,
1134                                            peer);
1135     queue->start_waiting = GNUNET_TIME_absolute_get ();
1136   }
1137   else
1138   {
1139     LOG (GNUNET_ERROR_TYPE_DEBUG,
1140                 "core tmt rdy towards %s already called\n",
1141                 GMP_2s (peer));
1142
1143   }
1144   return queue;
1145 }
1146
1147
1148 /**
1149  * Cancel all queued messages to a peer that belong to a certain connection.
1150  *
1151  * @param peer Peer towards whom to cancel.
1152  * @param c Connection whose queued messages to cancel. Might be destroyed by
1153  *          the sent continuation call.
1154  */
1155 void
1156 GMP_queue_cancel (struct MeshPeer *peer, struct MeshConnection *c)
1157 {
1158   struct MeshPeerQueue *q;
1159   struct MeshPeerQueue *next;
1160   struct MeshPeerQueue *prev;
1161
1162   for (q = peer->queue_head; NULL != q; q = next)
1163   {
1164     prev = q->prev;
1165     if (q->c == c)
1166     {
1167       LOG (GNUNET_ERROR_TYPE_DEBUG, "GMP_cancel_queue %s\n", GM_m2s (q->type));
1168       GMP_queue_destroy (q, GNUNET_YES);
1169
1170       /* Get next from prev, q->next might be already freed:
1171        * queue destroy -> callback -> GMC_destroy -> cancel_queues -> here
1172        */
1173       if (NULL == prev)
1174         next = peer->queue_head;
1175       else
1176         next = prev->next;
1177     }
1178     else
1179     {
1180       next = q->next;
1181     }
1182   }
1183   if (NULL == peer->queue_head)
1184   {
1185     if (NULL != peer->core_transmit)
1186     {
1187       GNUNET_CORE_notify_transmit_ready_cancel (peer->core_transmit);
1188       peer->core_transmit = NULL;
1189     }
1190   }
1191 }
1192
1193
1194 /**
1195  * Get the first transmittable message for a connection.
1196  *
1197  * @param peer Neighboring peer.
1198  * @param c Connection.
1199  *
1200  * @return First transmittable message.
1201  */
1202 static struct MeshPeerQueue *
1203 connection_get_first_message (struct MeshPeer *peer, struct MeshConnection *c)
1204 {
1205   struct MeshPeerQueue *q;
1206
1207   for (q = peer->queue_head; NULL != q; q = q->next)
1208   {
1209     if (q->c != c)
1210       continue;
1211     if (queue_is_sendable (q))
1212     {
1213       LOG (GNUNET_ERROR_TYPE_DEBUG, "  sendable!!\n");
1214       return q;
1215     }
1216     LOG (GNUNET_ERROR_TYPE_DEBUG, "  not sendable\n");
1217   }
1218
1219   return NULL;
1220 }
1221
1222
1223 /**
1224  * Get the first message for a connection and unqueue it.
1225  *
1226  * @param peer Neighboring peer.
1227  * @param c Connection.
1228  *
1229  * @return First message for this connection.
1230  */
1231 struct GNUNET_MessageHeader *
1232 GMP_connection_pop (struct MeshPeer *peer, struct MeshConnection *c)
1233 {
1234   struct MeshPeerQueue *q;
1235   struct GNUNET_MessageHeader *msg;
1236
1237   for (q = peer->queue_head; NULL != q; q = q->next)
1238   {
1239     if (q->c != c)
1240       continue;
1241     switch (q->type)
1242     {
1243       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
1244       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
1245       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
1246       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
1247       case GNUNET_MESSAGE_TYPE_MESH_ACK:
1248       case GNUNET_MESSAGE_TYPE_MESH_POLL:
1249         GMP_queue_destroy (q, GNUNET_YES);
1250         continue;
1251
1252       case GNUNET_MESSAGE_TYPE_MESH_KX:
1253       case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
1254         msg = (struct GNUNET_MessageHeader *) q->cls;
1255         GMP_queue_destroy (q, GNUNET_NO);
1256         return msg;
1257
1258       default:
1259         GNUNET_break (0);
1260     }
1261   }
1262
1263   return NULL;
1264 }
1265
1266
1267 void
1268 GMP_queue_unlock (struct MeshPeer *peer, struct MeshConnection *c)
1269 {
1270   struct MeshPeerQueue *q;
1271   size_t size;
1272
1273   if (NULL != peer->core_transmit)
1274   {
1275     LOG (GNUNET_ERROR_TYPE_DEBUG, "  already unlocked!\n");
1276     return; /* Already unlocked */
1277   }
1278
1279   q = connection_get_first_message (peer, c);
1280   if (NULL == q)
1281   {
1282     LOG (GNUNET_ERROR_TYPE_DEBUG, "  queue empty!\n");
1283     return; /* Nothing to transmit */
1284   }
1285
1286   size = q->size;
1287   peer->core_transmit =
1288       GNUNET_CORE_notify_transmit_ready (core_handle,
1289                                          GNUNET_NO, get_priority (q),
1290                                          GNUNET_TIME_UNIT_FOREVER_REL,
1291                                          GNUNET_PEER_resolve2 (peer->id),
1292                                          size,
1293                                          &queue_send,
1294                                          peer);
1295 }
1296
1297
1298 /**
1299  * Initialize the peer subsystem.
1300  *
1301  * @param c Configuration.
1302  */
1303 void
1304 GMP_init (const struct GNUNET_CONFIGURATION_Handle *c)
1305 {
1306   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
1307   peers = GNUNET_CONTAINER_multipeermap_create (128, GNUNET_NO);
1308   if (GNUNET_OK !=
1309       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS",
1310                                              &max_peers))
1311   {
1312     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_WARNING,
1313                                "MESH", "MAX_PEERS", "USING DEFAULT");
1314     max_peers = 1000;
1315   }
1316
1317   if (GNUNET_OK !=
1318       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DROP_PERCENT",
1319                                              &drop_percent))
1320   {
1321     drop_percent = 0;
1322   }
1323   else
1324   {
1325     LOG (GNUNET_ERROR_TYPE_WARNING, "**************************************\n");
1326     LOG (GNUNET_ERROR_TYPE_WARNING, "Mesh is running with DROP enabled.\n");
1327     LOG (GNUNET_ERROR_TYPE_WARNING, "This is NOT a good idea!\n");
1328     LOG (GNUNET_ERROR_TYPE_WARNING, "Remove DROP_PERCENT from config file.\n");
1329     LOG (GNUNET_ERROR_TYPE_WARNING, "**************************************\n");
1330   }
1331
1332   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
1333                                      NULL,      /* Closure passed to MESH functions */
1334                                      &core_init,        /* Call core_init once connected */
1335                                      &core_connect,     /* Handle connects */
1336                                      &core_disconnect,  /* remove peers on disconnects */
1337                                      NULL,      /* Don't notify about all incoming messages */
1338                                      GNUNET_NO, /* For header only in notification */
1339                                      NULL,      /* Don't notify about all outbound messages */
1340                                      GNUNET_NO, /* For header-only out notification */
1341                                      core_handlers);    /* Register these handlers */
1342   if (GNUNET_YES !=
1343     GNUNET_CONFIGURATION_get_value_yesno (c, "MESH", "DISABLE_TRY_CONNECT"))
1344   {
1345     transport_handle = GNUNET_TRANSPORT_connect (c, &my_full_id, NULL, /* cls */
1346                                                  /* Notify callbacks */
1347                                                  NULL, NULL, NULL);
1348
1349   }
1350
1351
1352   if (NULL == core_handle)
1353   {
1354     GNUNET_break (0);
1355     GNUNET_SCHEDULER_shutdown ();
1356     return;
1357   }
1358
1359 }
1360
1361 /**
1362  * Shut down the peer subsystem.
1363  */
1364 void
1365 GMP_shutdown (void)
1366 {
1367   GNUNET_CONTAINER_multipeermap_iterate (peers, &shutdown_tunnel, NULL);
1368
1369   if (core_handle != NULL)
1370   {
1371     GNUNET_CORE_disconnect (core_handle);
1372     core_handle = NULL;
1373   }
1374   if (transport_handle != NULL)
1375   {
1376     GNUNET_TRANSPORT_disconnect (transport_handle);
1377     transport_handle = NULL;
1378   }
1379   GNUNET_PEER_change_rc (myid, -1);
1380 }
1381
1382 /**
1383  * Retrieve the MeshPeer stucture associated with the peer, create one
1384  * and insert it in the appropriate structures if the peer is not known yet.
1385  *
1386  * @param peer_id Full identity of the peer.
1387  *
1388  * @return Existing or newly created peer structure.
1389  */
1390 struct MeshPeer *
1391 GMP_get (const struct GNUNET_PeerIdentity *peer_id)
1392 {
1393   struct MeshPeer *peer;
1394
1395   peer = GNUNET_CONTAINER_multipeermap_get (peers, peer_id);
1396   if (NULL == peer)
1397   {
1398     peer = GNUNET_new (struct MeshPeer);
1399     if (GNUNET_CONTAINER_multipeermap_size (peers) > max_peers)
1400     {
1401       peer_delete_oldest ();
1402     }
1403         GNUNET_CONTAINER_multipeermap_put (peers, peer_id, peer,
1404                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1405         peer->id = GNUNET_PEER_intern (peer_id);
1406   }
1407   peer->last_contact = GNUNET_TIME_absolute_get();
1408
1409   return peer;
1410 }
1411
1412
1413 /**
1414  * Retrieve the MeshPeer stucture associated with the peer, create one
1415  * and insert it in the appropriate structures if the peer is not known yet.
1416  *
1417  * @param peer Short identity of the peer.
1418  *
1419  * @return Existing or newly created peer structure.
1420  */
1421 struct MeshPeer *
1422 GMP_get_short (const GNUNET_PEER_Id peer)
1423 {
1424   return GMP_get (GNUNET_PEER_resolve2 (peer));
1425 }
1426
1427
1428 /**
1429  * Try to connect to a peer on transport level.
1430  *
1431  * @param cls Closure (peer).
1432  * @param tc TaskContext.
1433  */
1434 static void
1435 try_connect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1436 {
1437   struct MeshPeer *peer = cls;
1438
1439   if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
1440     return;
1441
1442   GNUNET_TRANSPORT_try_connect (transport_handle,
1443                                 GNUNET_PEER_resolve2 (peer->id), NULL, NULL);
1444 }
1445
1446
1447 /**
1448  * Try to establish a new connection to this peer (in its tunnel).
1449  * If the peer doesn't have any path to it yet, try to get one.
1450  * If the peer already has some path, send a CREATE CONNECTION towards it.
1451  *
1452  * @param peer Peer to connect to.
1453  */
1454 void
1455 GMP_connect (struct MeshPeer *peer)
1456 {
1457   struct MeshTunnel3 *t;
1458   struct MeshPeerPath *p;
1459   struct MeshConnection *c;
1460   int rerun_search;
1461
1462   LOG (GNUNET_ERROR_TYPE_DEBUG, "peer_connect towards %s\n", GMP_2s (peer));
1463
1464   /* If we have a current hello, try to connect using it. */
1465   GMP_try_connect (peer);
1466
1467   t = peer->tunnel;
1468   c = NULL;
1469   rerun_search = GNUNET_NO;
1470
1471   if (NULL != peer->path_head)
1472   {
1473     LOG (GNUNET_ERROR_TYPE_DEBUG, "  some path exists\n");
1474     p = peer_get_best_path (peer);
1475     if (NULL != p)
1476     {
1477       LOG (GNUNET_ERROR_TYPE_DEBUG, "  %u hops\n", p->length);
1478       c = GMT_use_path (t, p);
1479       if (NULL == c)
1480       {
1481         /* This case can happen when the path includes a first hop that is
1482          * not yet known to be connected.
1483          *
1484          * This happens quite often during testing when running mesh
1485          * under valgrind: core connect notifications come very late and the
1486          * DHT result has already come and created a valid path.
1487          * In this case, the peer->connections hashmap will be NULL and
1488          * tunnel_use_path will not be able to create a connection from that
1489          * path.
1490          *
1491          * Re-running the DHT GET should give core time to callback.
1492          *
1493          * GMT_use_path -> GMC_new -> register_neighbors takes care of
1494          * updating statistics about this issue.
1495          */
1496         rerun_search = GNUNET_YES;
1497       }
1498       else
1499       {
1500         GMC_send_create (c);
1501         return;
1502       }
1503     }
1504     else
1505     {
1506       LOG (GNUNET_ERROR_TYPE_DEBUG, "  but is NULL, all paths are in use\n");
1507     }
1508   }
1509
1510   if (NULL != peer->search_h && GNUNET_YES == rerun_search)
1511   {
1512     GMD_search_stop (peer->search_h);
1513     peer->search_h = NULL;
1514     LOG (GNUNET_ERROR_TYPE_DEBUG,
1515          "  Stopping DHT GET for peer %s\n",
1516          GMP_2s (peer));
1517   }
1518
1519   if (NULL == peer->search_h)
1520   {
1521     const struct GNUNET_PeerIdentity *id;
1522
1523     id = GNUNET_PEER_resolve2 (peer->id);
1524     LOG (GNUNET_ERROR_TYPE_DEBUG,
1525                 "  Starting DHT GET for peer %s\n", GMP_2s (peer));
1526     peer->search_h = GMD_search (id, &search_handler, peer);
1527     if (MESH_TUNNEL3_NEW == GMT_get_cstate (t))
1528       GMT_change_cstate (t, MESH_TUNNEL3_SEARCHING);
1529   }
1530 }
1531
1532
1533 /**
1534  * Chech whether there is a direct (core level)  connection to peer.
1535  *
1536  * @param peer Peer to check.
1537  *
1538  * @return #GNUNET_YES if there is a direct connection.
1539  */
1540 int
1541 GMP_is_neighbor (const struct MeshPeer *peer)
1542 {
1543   struct MeshPeerPath *path;
1544
1545   if (NULL == peer->connections)
1546     return GNUNET_NO;
1547
1548   for (path = peer->path_head; NULL != path; path = path->next)
1549   {
1550     if (3 > path->length)
1551       return GNUNET_YES;
1552   }
1553
1554   /* Is not a neighbor but connections is not NULL, probably disconnecting */
1555   return GNUNET_NO;
1556 }
1557
1558
1559 /**
1560  * Create and initialize a new tunnel towards a peer, in case it has none.
1561  * In case the peer already has a tunnel, nothing is done.
1562  *
1563  * Does not generate any traffic, just creates the local data structures.
1564  *
1565  * @param peer Peer towards which to create the tunnel.
1566  */
1567 void
1568 GMP_add_tunnel (struct MeshPeer *peer)
1569 {
1570   if (NULL != peer->tunnel)
1571     return;
1572   peer->tunnel = GMT_new (peer);
1573 }
1574
1575
1576 /**
1577  * Add a connection to a neighboring peer.
1578  *
1579  * Store that the peer is the first hop of the connection in one
1580  * direction and that on peer disconnect the connection must be
1581  * notified and destroyed, for it will no longer be valid.
1582  *
1583  * @param peer Peer to add connection to.
1584  * @param c Connection to add.
1585  *
1586  * @return GNUNET_OK on success.
1587  */
1588 int
1589 GMP_add_connection (struct MeshPeer *peer,
1590                     struct MeshConnection *c)
1591 {
1592   int result;
1593   LOG (GNUNET_ERROR_TYPE_DEBUG, "adding connection %s\n", GMC_2s (c));
1594   LOG (GNUNET_ERROR_TYPE_DEBUG, "to peer %s\n", GMP_2s (peer));
1595
1596   if (NULL == peer->connections)
1597   {
1598     GNUNET_break (0);
1599     LOG (GNUNET_ERROR_TYPE_DEBUG,
1600          "Peer %s is not a neighbor!\n",
1601          GMP_2s (peer));
1602     return GNUNET_SYSERR;
1603   }
1604   LOG (GNUNET_ERROR_TYPE_DEBUG,
1605        "peer %s ok, has %u connections.\n",
1606        GMP_2s (peer), GNUNET_CONTAINER_multihashmap_size (peer->connections));
1607   result = GNUNET_CONTAINER_multihashmap_put (peer->connections,
1608                                               GMC_get_h (c),
1609                                               c,
1610                                               GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1611   LOG (GNUNET_ERROR_TYPE_DEBUG,
1612        " now has %u connections.\n",
1613        GNUNET_CONTAINER_multihashmap_size (peer->connections));
1614   LOG (GNUNET_ERROR_TYPE_DEBUG, "result %u\n", result);
1615
1616   return result;
1617 }
1618
1619
1620 /**
1621  * Add the path to the peer and update the path used to reach it in case this
1622  * is the shortest.
1623  *
1624  * @param peer Destination peer to add the path to.
1625  * @param path New path to add. Last peer must be the peer in arg 1.
1626  *             Path will be either used of freed if already known.
1627  * @param trusted Do we trust that this path is real?
1628  *
1629  * @return path if path was taken, pointer to existing duplicate if exists
1630  *         NULL on error.
1631  */
1632 struct MeshPeerPath *
1633 GMP_add_path (struct MeshPeer *peer, struct MeshPeerPath *path,
1634               int trusted)
1635 {
1636   struct MeshPeerPath *aux;
1637   unsigned int l;
1638   unsigned int l2;
1639
1640   LOG (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n",
1641        path->length, GMP_2s (peer));
1642
1643   if ((NULL == peer) || (NULL == path))
1644   {
1645     GNUNET_break (0);
1646     path_destroy (path);
1647     return NULL;
1648   }
1649   if (path->peers[path->length - 1] != peer->id)
1650   {
1651     GNUNET_break (0);
1652     path_destroy (path);
1653     return NULL;
1654   }
1655   if (2 >= path->length && GNUNET_NO == trusted)
1656   {
1657     /* Only allow CORE to tell us about direct paths */
1658     path_destroy (path);
1659     return NULL;
1660   }
1661   for (l = 1; l < path->length; l++)
1662   {
1663     if (path->peers[l] == myid)
1664     {
1665       LOG (GNUNET_ERROR_TYPE_DEBUG, " shortening path by %u\n", l);
1666       for (l2 = 0; l2 < path->length - l; l2++)
1667       {
1668         path->peers[l2] = path->peers[l + l2];
1669       }
1670       path->length -= l;
1671       l = 1;
1672       path->peers = GNUNET_realloc (path->peers,
1673                                     path->length * sizeof (GNUNET_PEER_Id));
1674     }
1675   }
1676
1677   LOG (GNUNET_ERROR_TYPE_DEBUG, " final length: %u\n", path->length);
1678
1679   l = path_get_length (path);
1680   if (0 == l)
1681   {
1682     path_destroy (path);
1683     return NULL;
1684   }
1685
1686   GNUNET_assert (peer->id == path->peers[path->length - 1]);
1687   for (aux = peer->path_head; aux != NULL; aux = aux->next)
1688   {
1689     l2 = path_get_length (aux);
1690     if (l2 > l)
1691     {
1692       LOG (GNUNET_ERROR_TYPE_DEBUG, "  added\n");
1693       GNUNET_CONTAINER_DLL_insert_before (peer->path_head,
1694                                           peer->path_tail, aux, path);
1695       if (NULL != peer->tunnel && 3 < GMT_count_connections (peer->tunnel))
1696       {
1697         GMP_connect (peer);
1698       }
1699       return path;
1700     }
1701     else
1702     {
1703       if (l2 == l && memcmp (path->peers, aux->peers, l) == 0)
1704       {
1705         LOG (GNUNET_ERROR_TYPE_DEBUG, "  already known\n");
1706         path_destroy (path);
1707         return aux;
1708       }
1709     }
1710   }
1711   GNUNET_CONTAINER_DLL_insert_tail (peer->path_head, peer->path_tail,
1712                                     path);
1713   LOG (GNUNET_ERROR_TYPE_DEBUG, "  added last\n");
1714   if (NULL != peer->tunnel && 3 < GMT_count_connections (peer->tunnel))
1715   {
1716     GMP_connect (peer);
1717   }
1718   return path;
1719 }
1720
1721
1722 /**
1723  * Add the path to the origin peer and update the path used to reach it in case
1724  * this is the shortest.
1725  * The path is given in peer_info -> destination, therefore we turn the path
1726  * upside down first.
1727  *
1728  * @param peer Peer to add the path to, being the origin of the path.
1729  * @param path New path to add after being inversed.
1730  *             Path will be either used or freed.
1731  * @param trusted Do we trust that this path is real?
1732  *
1733  * @return path if path was taken, pointer to existing duplicate if exists
1734  *         NULL on error.
1735  */
1736 struct MeshPeerPath *
1737 GMP_add_path_to_origin (struct MeshPeer *peer,
1738                         struct MeshPeerPath *path,
1739                         int trusted)
1740 {
1741   if (NULL == path)
1742     return NULL;
1743   path_invert (path);
1744   return GMP_add_path (peer, path, trusted);
1745 }
1746
1747
1748 /**
1749  * Adds a path to the info of all the peers in the path
1750  *
1751  * @param p Path to process.
1752  * @param confirmed Whether we know if the path works or not.
1753  */
1754 void
1755 GMP_add_path_to_all (const struct MeshPeerPath *p, int confirmed)
1756 {
1757   unsigned int i;
1758
1759   /* TODO: invert and add */
1760   for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ;
1761   for (i++; i < p->length; i++)
1762   {
1763     struct MeshPeer *aux;
1764     struct MeshPeerPath *copy;
1765
1766     aux = GMP_get_short (p->peers[i]);
1767     copy = path_duplicate (p);
1768     copy->length = i + 1;
1769     GMP_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed);
1770   }
1771 }
1772
1773
1774 /**
1775  * Remove any path to the peer that has the extact same peers as the one given.
1776  *
1777  * @param peer Peer to remove the path from.
1778  * @param path Path to remove. Is always destroyed .
1779  */
1780 void
1781 GMP_remove_path (struct MeshPeer *peer, struct MeshPeerPath *path)
1782 {
1783   struct MeshPeerPath *iter;
1784   struct MeshPeerPath *next;
1785
1786   GNUNET_assert (myid == path->peers[0]);
1787   GNUNET_assert (peer->id == path->peers[path->length - 1]);
1788
1789   for (iter = peer->path_head; NULL != iter; iter = next)
1790   {
1791     next = iter->next;
1792     if (0 == memcmp (path->peers, iter->peers,
1793                      sizeof (GNUNET_PEER_Id) * path->length))
1794     {
1795       GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, iter);
1796       path_destroy (iter);
1797       if (path == iter)
1798         return;
1799     }
1800   }
1801   path_destroy (path);
1802 }
1803
1804
1805 /**
1806  * Remove a connection from a neighboring peer.
1807  *
1808  * @param peer Peer to remove connection from.
1809  * @param c Connection to remove.
1810  *
1811  * @return GNUNET_OK on success.
1812  */
1813 int
1814 GMP_remove_connection (struct MeshPeer *peer,
1815                        const struct MeshConnection *c)
1816 {
1817   LOG (GNUNET_ERROR_TYPE_DEBUG, "removing connection %s\n", GMC_2s (c));
1818   LOG (GNUNET_ERROR_TYPE_DEBUG, "from peer %s\n", GMP_2s (peer));
1819
1820   if (NULL == peer || NULL == peer->connections)
1821   {
1822     LOG (GNUNET_ERROR_TYPE_DEBUG,
1823          "Peer %s is not a neighbor!\n",
1824          GMP_2s (peer));
1825     return GNUNET_SYSERR;
1826   }
1827   LOG (GNUNET_ERROR_TYPE_DEBUG,
1828        "peer %s ok, has %u connections.\n",
1829        GMP_2s (peer), GNUNET_CONTAINER_multihashmap_size (peer->connections));
1830
1831   return GNUNET_CONTAINER_multihashmap_remove (peer->connections,
1832                                                GMC_get_h (c),
1833                                                c);
1834 }
1835
1836 /**
1837  * Start the DHT search for new paths towards the peer: we don't have
1838  * enough good connections.
1839  *
1840  * @param peer Destination peer.
1841  */
1842 void
1843 GMP_start_search (struct MeshPeer *peer)
1844 {
1845   if (NULL != peer->search_h)
1846   {
1847     GNUNET_break (0);
1848     return;
1849   }
1850
1851   peer->search_h = GMD_search (GMP_get_id (peer), &search_handler, peer);
1852 }
1853
1854
1855 /**
1856  * Stop the DHT search for new paths towards the peer: we already have
1857  * enough good connections.
1858  *
1859  * @param peer Destination peer.
1860  */
1861 void
1862 GMP_stop_search (struct MeshPeer *peer)
1863 {
1864   if (NULL == peer->search_h)
1865   {
1866     GNUNET_break (0);
1867     return;
1868   }
1869
1870   GMD_search_stop (peer->search_h);
1871   peer->search_h = NULL;
1872 }
1873
1874
1875 /**
1876  * Get the Full ID of a peer.
1877  *
1878  * @param peer Peer to get from.
1879  *
1880  * @return Full ID of peer.
1881  */
1882 const struct GNUNET_PeerIdentity *
1883 GMP_get_id (const struct MeshPeer *peer)
1884 {
1885   return GNUNET_PEER_resolve2 (peer->id);
1886 }
1887
1888
1889 /**
1890  * Get the Short ID of a peer.
1891  *
1892  * @param peer Peer to get from.
1893  *
1894  * @return Short ID of peer.
1895  */
1896 GNUNET_PEER_Id
1897 GMP_get_short_id (const struct MeshPeer *peer)
1898 {
1899   return peer->id;
1900 }
1901
1902
1903 /**
1904  * Set tunnel.
1905  *
1906  * @param peer Peer.
1907  * @param t Tunnel.
1908  */
1909 void
1910 GMP_set_tunnel (struct MeshPeer *peer, struct MeshTunnel3 *t)
1911 {
1912   peer->tunnel = t;
1913 }
1914
1915
1916 /**
1917  * Get the tunnel towards a peer.
1918  *
1919  * @param peer Peer to get from.
1920  *
1921  * @return Tunnel towards peer.
1922  */
1923 struct MeshTunnel3 *
1924 GMP_get_tunnel (const struct MeshPeer *peer)
1925 {
1926   return peer->tunnel;
1927 }
1928
1929
1930 /**
1931  * Set the hello message.
1932  *
1933  * @param peer Peer whose message to set.
1934  * @param hello Hello message.
1935  */
1936 void
1937 GMP_set_hello (struct MeshPeer *peer, const struct GNUNET_HELLO_Message *hello)
1938 {
1939   struct GNUNET_HELLO_Message *old;
1940   size_t size;
1941
1942   LOG (GNUNET_ERROR_TYPE_DEBUG, "set hello for %s\n", GMP_2s (peer));
1943   if (NULL == hello)
1944     return;
1945
1946   old = GMP_get_hello (peer);
1947   if (NULL == old)
1948   {
1949     size = GNUNET_HELLO_size (hello);
1950     LOG (GNUNET_ERROR_TYPE_DEBUG, " new (%u bytes)\n", size);
1951     peer->hello = GNUNET_malloc (size);
1952     memcpy (peer->hello, hello, size);
1953   }
1954   else
1955   {
1956     peer->hello = GNUNET_HELLO_merge (old, hello);
1957     LOG (GNUNET_ERROR_TYPE_DEBUG, " merge into %p (%u bytes)\n",
1958          peer->hello, GNUNET_HELLO_size (hello));
1959     GNUNET_free (old);
1960   }
1961 }
1962
1963
1964 /**
1965  * Get the hello message.
1966  *
1967  * @param peer Peer whose message to get.
1968  *
1969  * @return Hello message.
1970  */
1971 struct GNUNET_HELLO_Message *
1972 GMP_get_hello (struct MeshPeer *peer)
1973 {
1974   struct GNUNET_TIME_Absolute expiration;
1975   struct GNUNET_TIME_Relative remaining;
1976
1977   if (NULL == peer->hello)
1978     return NULL;
1979
1980   expiration = GNUNET_HELLO_get_last_expiration (peer->hello);
1981   remaining = GNUNET_TIME_absolute_get_remaining (expiration);
1982   if (0 == remaining.rel_value_us)
1983   {
1984     LOG (GNUNET_ERROR_TYPE_DEBUG, " get - hello expired on %s\n",
1985          GNUNET_STRINGS_absolute_time_to_string (expiration));
1986     GNUNET_free (peer->hello);
1987     peer->hello = NULL;
1988   }
1989   return peer->hello;
1990 }
1991
1992
1993 /**
1994  * Try to connect to a peer on TRANSPORT level.
1995  *
1996  * @param peer Peer to whom to connect.
1997  */
1998 void
1999 GMP_try_connect (struct MeshPeer *peer)
2000 {
2001   struct GNUNET_HELLO_Message *hello;
2002   struct GNUNET_MessageHeader *mh;
2003
2004   if (NULL == transport_handle)
2005     return;
2006
2007   hello = GMP_get_hello (peer);
2008   if (NULL == hello)
2009     return;
2010
2011   mh = GNUNET_HELLO_get_header (hello);
2012   GNUNET_TRANSPORT_offer_hello (transport_handle, mh, try_connect, peer);
2013 }
2014
2015
2016 /**
2017  * Count the number of known paths toward the peer.
2018  *
2019  * @param peer Peer to get path info.
2020  *
2021  * @return Number of known paths.
2022  */
2023 unsigned int
2024 GMP_count_paths (const struct MeshPeer *peer)
2025 {
2026   struct MeshPeerPath *iter;
2027   unsigned int i;
2028
2029   for (iter = peer->path_head, i = 0; NULL != iter; iter = iter->next)
2030     i++;
2031
2032   return i;
2033 }
2034
2035
2036 /**
2037  * Iterate all known peers.
2038  *
2039  * @param iter Iterator.
2040  * @param cls Closure for @c iter.
2041  */
2042 void
2043 GMP_iterate_all (GNUNET_CONTAINER_PeerMapIterator iter, void *cls)
2044 {
2045   GNUNET_CONTAINER_multipeermap_iterate (peers, iter, cls);
2046 }
2047
2048
2049 /**
2050  * Get the static string for a peer ID.
2051  *
2052  * @param peer Peer.
2053  *
2054  * @return Static string for it's ID.
2055  */
2056 const char *
2057 GMP_2s (const struct MeshPeer *peer)
2058 {
2059   if (NULL == peer)
2060     return "(NULL)";
2061   return GNUNET_i2s (GNUNET_PEER_resolve2 (peer->id));
2062 }