src/mesh/gnunet-service-mesh.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2001 - 2011 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file mesh/gnunet-service-mesh.c
  23  * @brief GNUnet MESH service
  24  * @author Bartlomiej Polot
  25  *
  26  * STRUCTURE:
  27  * - DATA STRUCTURES
  28  * - GLOBAL VARIABLES
  29  * - GENERAL HELPERS
  30  * - PERIODIC FUNCTIONS
  31  * - MESH NETWORK HANDLER HELPERS
  32  * - MESH NETWORK HANDLES
  33  * - MESH LOCAL HANDLER HELPERS
  34  * - MESH LOCAL HANDLES
  35  * - MAIN FUNCTIONS (main & run)
  36  *
  37  * TODO:
  38  * - error reporting (CREATE/CHANGE/ADD/DEL?) -- new message!
  39  * - partial disconnect reporting -- same as error reporting?
  40  * - add vs create? change vs. keep-alive? same msg or different ones? -- thinking...
  41  * - speed requirement specification (change?) in mesh API -- API call
  42  * - add ping message
  43  * - relay corking down to core
  44  */
  45
  46 #include "platform.h"
  47 #include "mesh.h"
  48 #include "mesh_protocol.h"
  49 #include "gnunet_dht_service.h"
  50 #include "mesh_tunnel_tree.h"
  51
  52 /* TODO: move into configuration file */
  53 #define REFRESH_PATH_TIME       GNUNET_TIME_relative_multiply(\
  54                                     GNUNET_TIME_UNIT_SECONDS,\
  55                                     300)
  56 #define APP_ANNOUNCE_TIME       GNUNET_TIME_relative_multiply(\
  57                                     GNUNET_TIME_UNIT_SECONDS,\
  58                                     5)
  59
  60 #define ID_ANNOUNCE_TIME        GNUNET_TIME_relative_multiply(\
  61                                     GNUNET_TIME_UNIT_SECONDS,\
  62                                     10)
  63
  64 #define GET_RESTART_TIME        GNUNET_TIME_relative_multiply(\
  65                                     GNUNET_TIME_UNIT_SECONDS,\
  66                                     5)
  67
  68 #define UNACKNOWLEDGED_WAIT     GNUNET_TIME_relative_multiply(\
  69                                     GNUNET_TIME_UNIT_SECONDS,\
  70                                     2)
  71
  72 /******************************************************************************/
  73 /************************      DATA STRUCTURES     ****************************/
  74 /******************************************************************************/
  75
  76 /** FWD declaration */
  77 struct MeshPeerInfo;
  78
  79 /**
  80  * Struct containing all info possibly needed to build a package when called
  81  * back by core.
  82  */
  83 struct MeshDataDescriptor
  84 {
  85     /** ID of the tunnel this packet travels in */
  86   struct MESH_TunnelID *origin;
  87
  88     /** Data itself */
  89   void *data;
  90
  91     /** Client that asked for the transmission, if any */
  92   struct GNUNET_SERVER_Client *client;
  93
  94     /** Who was this message being sent to */
  95   struct MeshPeerInfo *peer;
  96
  97     /** Ultimate destination of the packet */
  98   GNUNET_PEER_Id destination;
  99
 100     /** Number of identical messages sent to different hops (multicast) */
 101   unsigned int *copies;
 102
 103     /** Which handler was used to request the transmission */
 104   unsigned int handler_n;
 105
 106     /** Size of the data */
 107   size_t size;
 108
 109     /** Used to allow a client send more traffic to the service after a
 110      * previous packet was tried to be sent to a neighbor and couldn't */
 111   GNUNET_SCHEDULER_TaskIdentifier *timeout_task;
 112 };
 113
 114
 115 /**
 116  * Struct containing all information regarding a given peer
 117  */
 118 struct MeshPeerInfo
 119 {
 120     /**
 121      * ID of the peer
 122      */
 123   GNUNET_PEER_Id id;
 124
 125     /**
 126      * Last time we heard from this peer
 127      */
 128   struct GNUNET_TIME_Absolute last_contact;
 129
 130     /**
 131      * Number of attempts to reconnect so far
 132      */
 133   int n_reconnect_attempts;
 134
 135     /**
 136      * Paths to reach the peer, ordered by ascending hop count
 137      */
 138   struct MeshPeerPath *path_head;
 139
 140     /**
 141      * Paths to reach the peer, ordered by ascending hop count
 142      */
 143   struct MeshPeerPath *path_tail;
 144
 145     /**
 146      * Handle to stop the DHT search for a path to this peer
 147      */
 148   struct GNUNET_DHT_GetHandle *dhtget;
 149
 150     /**
 151      * Closure given to the DHT GET
 152      */
 153   struct MeshPathInfo *dhtgetcls;
 154
 155     /**
 156      * Handles to stop queued transmissions for this peer
 157      */
 158   struct GNUNET_CORE_TransmitHandle *core_transmit[CORE_QUEUE_SIZE];
 159
 160     /**
 161      * Pointer to info stuctures used as cls for queued transmissions
 162      */
 163   void *infos[CORE_QUEUE_SIZE];
 164
 165     /**
 166      * Type of message being in each transmission
 167      */
 168   uint16_t types[CORE_QUEUE_SIZE];
 169
 170     /**
 171      * Array of tunnels this peer participates in
 172      * (most probably a small amount, therefore not a hashmap)
 173      * When the path to the peer changes, notify these tunnels to let them
 174      * re-adjust their path trees.
 175      */
 176   struct MeshTunnel **tunnels;
 177
 178     /**
 179      * Number of tunnels this peers participates in
 180      */
 181   unsigned int ntunnels;
 182 };
 183
 184
 185 /**
 186  * Data scheduled to transmit (to local client or remote peer)
 187  */
 188 struct MeshQueue
 189 {
 190     /**
 191      * Double linked list
 192      */
 193   struct MeshQueue *next;
 194   struct MeshQueue *prev;
 195
 196     /**
 197      * Target of the data (NULL if target is client)
 198      */
 199   struct MeshPeerInfo *peer;
 200
 201     /**
 202      * Client to send the data to (NULL if target is peer)
 203      */
 204   struct MeshClient *client;
 205
 206     /**
 207      * Size of the message to transmit
 208      */
 209   unsigned int size;
 210
 211     /**
 212      * How old is the data?
 213      */
 214   struct GNUNET_TIME_Absolute timestamp;
 215
 216     /**
 217      * Data itself
 218      */
 219   struct GNUNET_MessageHeader *data;
 220 };
 221
 222 /**
 223  * Globally unique tunnel identification (owner + number)
 224  * DO NOT USE OVER THE NETWORK
 225  */
 226 struct MESH_TunnelID
 227 {
 228     /**
 229      * Node that owns the tunnel
 230      */
 231   GNUNET_PEER_Id oid;
 232
 233     /**
 234      * Tunnel number to differentiate all the tunnels owned by the node oid
 235      * ( tid < GNUNET_MESH_LOCAL_TUNNEL_ID_CLI )
 236      */
 237   MESH_TunnelNumber tid;
 238 };
 239
 240
 241 struct MeshClient;              /* FWD declaration */
 242
 243 /**
 244  * Struct containing all information regarding a tunnel
 245  * For an intermediate node the improtant info used will be:
 246  * - id        Tunnel unique identification
 247  * - paths[0]  To know where to send it next
 248  * - metainfo: ready, speeds, accounting
 249  */
 250 struct MeshTunnel
 251 {
 252     /**
 253      * Tunnel ID
 254      */
 255   struct MESH_TunnelID id;
 256
 257     /**
 258      * Local tunnel number ( >= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI or 0 )
 259      */
 260   MESH_TunnelNumber local_tid;
 261
 262     /**
 263      * Last time the tunnel was used
 264      */
 265   struct GNUNET_TIME_Absolute timestamp;
 266
 267     /**
 268      * Peers in the tunnel, indexed by PeerIdentity -> (MeshPeerInfo)
 269      * containing peers added by id or by type, not intermediate peers.
 270      */
 271   struct GNUNET_CONTAINER_MultiHashMap *peers;
 272
 273     /**
 274      * Number of peers that are connected and potentially ready to receive data
 275      */
 276   unsigned int peers_ready;
 277
 278     /**
 279      * Number of peers that have been added to the tunnel
 280      */
 281   unsigned int peers_total;
 282
 283     /**
 284      * Client owner of the tunnel, if any
 285      */
 286   struct MeshClient *client;
 287
 288     /**
 289      * Messages ready to transmit
 290      */
 291   struct MeshQueue *queue_head;
 292   struct MeshQueue *queue_tail;
 293
 294   /**
 295    * Tunnel paths
 296    */
 297   struct MeshTunnelTree *tree;
 298
 299   /**
 300    * Application type we are looking for in this tunnel
 301    */
 302   GNUNET_MESH_ApplicationType type;
 303
 304     /**
 305      * Used to search peers offering a service
 306      */
 307   struct GNUNET_DHT_GetHandle *dht_get_type;
 308
 309   /**
 310    * Task to keep the used paths alive
 311    */
 312   GNUNET_SCHEDULER_TaskIdentifier path_refresh_task;
 313
 314   /**
 315    * Task to destroy the tunnel after timeout
 316    *
 317    * FIXME: merge the two? a tunnel will have either
 318    * a path refresh OR a timeout, never both!
 319    */
 320   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
 321 };
 322
 323
 324 /**
 325  * Info needed to work with tunnel paths and peers
 326  */
 327 struct MeshPathInfo
 328 {
 329   /**
 330    * Tunnel
 331    */
 332   struct MeshTunnel *t;
 333
 334   /**
 335    * Neighbouring peer to whom we send the packet to
 336    */
 337   struct MeshPeerInfo *peer;
 338
 339   /**
 340    * Path itself
 341    */
 342   struct MeshPeerPath *path;
 343
 344   /**
 345    * Position in peer's transmit queue
 346    */
 347   unsigned int pos;
 348 };
 349
 350
 351 /**
 352  * Struct containing information about a client of the service
 353  */
 354 struct MeshClient
 355 {
 356     /**
 357      * Linked list
 358      */
 359   struct MeshClient *next;
 360   struct MeshClient *prev;
 361
 362     /**
 363      * Tunnels that belong to this client, indexed by local id
 364      */
 365   struct GNUNET_CONTAINER_MultiHashMap *tunnels;
 366
 367     /**
 368      * Handle to communicate with the client
 369      */
 370   struct GNUNET_SERVER_Client *handle;
 371
 372     /**
 373      * Applications that this client has claimed to provide
 374      */
 375   struct GNUNET_CONTAINER_MultiHashMap *apps;
 376
 377     /**
 378      * Messages that this client has declared interest in
 379      */
 380   struct GNUNET_CONTAINER_MultiHashMap *types;
 381
 382 #if MESH_DEBUG
 383     /**
 384      * ID of the client, for debug messages
 385      */
 386   unsigned int id;
 387 #endif
 388
 389 };
 390
 391
 392
 393 /******************************************************************************/
 394 /************************      DEBUG FUNCTIONS     ****************************/
 395 /******************************************************************************/
 396
 397
 398 /**
 399  * GNUNET_SCHEDULER_Task for printing a message after some operation is done
 400  * @param cls string to print
 401  * @param tc task context
 402  */
 403 static void
 404 mesh_debug (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 405 {
 406   char *s = cls;
 407
 408   if (GNUNET_SCHEDULER_REASON_SHUTDOWN == tc->reason)
 409   {
 410     return;
 411   }
 412   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: %s\n", s);
 413 }
 414
 415
 416 /******************************************************************************/
 417 /***********************      GLOBAL VARIABLES     ****************************/
 418 /******************************************************************************/
 419
 420 /**
 421  * All the clients
 422  */
 423 static struct MeshClient *clients;
 424 static struct MeshClient *clients_tail;
 425
 426 /**
 427  * Tunnels known, indexed by MESH_TunnelID (MeshTunnel)
 428  */
 429 static struct GNUNET_CONTAINER_MultiHashMap *tunnels;
 430
 431 /**
 432  * Tunnels incoming, indexed by MESH_TunnelNumber
 433  * (which is greater than GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
 434  */
 435 static struct GNUNET_CONTAINER_MultiHashMap *incoming_tunnels;
 436
 437 /**
 438  * Peers known, indexed by PeerIdentity (MeshPeerInfo)
 439  */
 440 static struct GNUNET_CONTAINER_MultiHashMap *peers;
 441
 442 /**
 443  * Handle to communicate with core
 444  */
 445 static struct GNUNET_CORE_Handle *core_handle;
 446
 447 /**
 448  * Handle to use DHT
 449  */
 450 static struct GNUNET_DHT_Handle *dht_handle;
 451
 452 /**
 453  * Handle to server
 454  */
 455 static struct GNUNET_SERVER_Handle *server_handle;
 456
 457 /**
 458  * Notification context, to send messages to local clients
 459  */
 460 static struct GNUNET_SERVER_NotificationContext *nc;
 461
 462 /**
 463  * Local peer own ID (memory efficient handle)
 464  */
 465 static GNUNET_PEER_Id myid;
 466
 467 /**
 468  * Local peer own ID (full value)
 469  */
 470 static struct GNUNET_PeerIdentity my_full_id;
 471
 472 /**
 473  * Own private key
 474  */
 475 static struct GNUNET_CRYPTO_RsaPrivateKey* my_private_key;
 476
 477 /**
 478  * Own public key.
 479  */
 480 static struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded my_public_key;
 481
 482 /**
 483  * Tunnel ID for the next created tunnel (global tunnel number)
 484  */
 485 static MESH_TunnelNumber next_tid;
 486
 487 /**
 488  * Tunnel ID for the next incoming tunnel (local tunnel number)
 489  */
 490 static MESH_TunnelNumber next_local_tid;
 491
 492 /**
 493  * All application types provided by this peer
 494  */
 495 static struct GNUNET_CONTAINER_MultiHashMap *applications;
 496
 497 /**
 498  * All message types clients of this peer are interested in
 499  */
 500 static struct GNUNET_CONTAINER_MultiHashMap *types;
 501
 502 /**
 503  * Task to periodically announce provided applications
 504  */
 505 GNUNET_SCHEDULER_TaskIdentifier announce_applications_task;
 506
 507 /**
 508  * Task to periodically announce itself in the network
 509  */
 510 GNUNET_SCHEDULER_TaskIdentifier announce_id_task;
 511
 512 #if MESH_DEBUG
 513 unsigned int next_client_id;
 514 #endif
 515
 516
 517 /******************************************************************************/
 518 /************************         ITERATORS        ****************************/
 519 /******************************************************************************/
 520
 521 /* FIXME move iterators here */
 522
 523
 524 /******************************************************************************/
 525 /************************    PERIODIC FUNCTIONS    ****************************/
 526 /******************************************************************************/
 527
 528 /**
 529  * Announce iterator over for each application provided by the peer
 530  *
 531  * @param cls closure
 532  * @param key current key code
 533  * @param value value in the hash map
 534  * @return GNUNET_YES if we should continue to
 535  *         iterate,
 536  *         GNUNET_NO if not.
 537  */
 538 static int
 539 announce_application (void *cls, const GNUNET_HashCode * key, void *value)
 540 {
 541   /* FIXME are hashes in multihash map equal on all aquitectures? */
 542   GNUNET_DHT_put (dht_handle,
 543                   key,
 544                   10U,
 545                   GNUNET_DHT_RO_RECORD_ROUTE |
 546                     GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
 547                   GNUNET_BLOCK_TYPE_TEST,
 548                   sizeof (struct GNUNET_PeerIdentity),
 549                   (const char *) &my_full_id,
 550 #if MESH_DEBUG
 551                   GNUNET_TIME_UNIT_FOREVER_ABS, GNUNET_TIME_UNIT_FOREVER_REL,
 552                   &mesh_debug, "DHT_put for app completed");
 553 #else
 554                   GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get (),
 555                                             APP_ANNOUNCE_TIME),
 556                   APP_ANNOUNCE_TIME, NULL, NULL);
 557 #endif
 558   return GNUNET_OK;
 559 }
 560
 561
 562 /**
 563  * Periodically announce what applications are provided by local clients
 564  *
 565  * @param cls closure
 566  * @param tc task context
 567  */
 568 static void
 569 announce_applications (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 570 {
 571   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
 572   {
 573     announce_applications_task = GNUNET_SCHEDULER_NO_TASK;
 574     return;
 575   }
 576   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Starting PUT for apps\n");
 577   GNUNET_CONTAINER_multihashmap_iterate (applications, &announce_application,
 578                                          NULL);
 579   announce_applications_task =
 580       GNUNET_SCHEDULER_add_delayed (APP_ANNOUNCE_TIME, &announce_applications,
 581                                     cls);
 582   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Finished PUT for apps\n");
 583   return;
 584 }
 585
 586
 587 /**
 588  * Periodically announce self id in the DHT
 589  *
 590  * @param cls closure
 591  * @param tc task context
 592  */
 593 static void
 594 announce_id (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 595 {
 596   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
 597   {
 598     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
 599     return;
 600   }
 601   /* TODO
 602    * - Set data expiration in function of X
 603    * - Adapt X to churn
 604    */
 605   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: DHT_put for ID %s started.\n",
 606               GNUNET_i2s (&my_full_id));
 607   GNUNET_DHT_put (dht_handle,   /* DHT handle */
 608                   &my_full_id.hashPubKey,       /* Key to use */
 609                   10U,          /* Replication level */
 610                   GNUNET_DHT_RO_RECORD_ROUTE |
 611                     GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,   /* DHT options */
 612                   GNUNET_BLOCK_TYPE_TEST,       /* Block type */
 613                   sizeof(my_full_id),   /* Size of the data */
 614                   (char *)&my_full_id,  /* Data itself */
 615                   GNUNET_TIME_absolute_get_forever (),  /* Data expiration */
 616                   GNUNET_TIME_UNIT_FOREVER_REL, /* Retry time */
 617 #if MESH_DEBUG
 618                   &mesh_debug, "DHT_put for id completed");
 619 #else
 620                   NULL,         /* Continuation */
 621                   NULL);        /* Continuation closure */
 622 #endif
 623   announce_id_task =
 624       GNUNET_SCHEDULER_add_delayed (ID_ANNOUNCE_TIME, &announce_id, cls);
 625 }
 626
 627
 628 /**
 629  * Function to process paths received for a new peer addition. The recorded
 630  * paths form the initial tunnel, which can be optimized later.
 631  * Called on each result obtained for the DHT search.
 632  *
 633  * @param cls closure
 634  * @param exp when will this value expire
 635  * @param key key of the result
 636  * @param type type of the result
 637  * @param size number of bytes in data
 638  * @param data pointer to the result data
 639  */
 640 static void
 641 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
 642                     const GNUNET_HashCode * key,
 643                     const struct GNUNET_PeerIdentity *get_path,
 644                     unsigned int get_path_length,
 645                     const struct GNUNET_PeerIdentity *put_path,
 646                     unsigned int put_path_length,
 647                     enum GNUNET_BLOCK_Type type, size_t size, const void *data);
 648
 649
 650 /******************************************************************************/
 651 /******************      GENERAL HELPER FUNCTIONS      ************************/
 652 /******************************************************************************/
 653
 654 /**
 655  * Check if client has registered with the service and has not disconnected
 656  *
 657  * @param client the client to check
 658  *
 659  * @return non-NULL if client exists in the global DLL
 660  */
 661 static struct MeshClient *
 662 client_get (struct GNUNET_SERVER_Client *client)
 663 {
 664   struct MeshClient *c;
 665
 666   c = clients;
 667   while (NULL != c)
 668   {
 669     if (c->handle == client)
 670       return c;
 671     c = c->next;
 672   }
 673   return NULL;
 674 }
 675
 676
 677 /**
 678  * Checks if a given client has subscribed to certain message type
 679  *
 680  * @param message_type Type of message to check
 681  * @param c Client to check
 682  *
 683  * @return GNUNET_YES or GNUNET_NO, depending on subscription status
 684  *
 685  * TODO inline?
 686  */
 687 static int
 688 client_is_subscribed (uint16_t message_type, struct MeshClient *c)
 689 {
 690   GNUNET_HashCode hc;
 691
 692   GNUNET_CRYPTO_hash (&message_type, sizeof (uint16_t), &hc);
 693   return GNUNET_CONTAINER_multihashmap_contains (c->types, &hc);
 694 }
 695
 696
 697 /**
 698  * Allow a client to send more data after transmitting a multicast message
 699  * which some neighbor has not yet accepted altough a reasonable time has
 700  * passed.
 701  *
 702  * @param cls Closure (DataDescriptor containing the task identifier)
 703  * @param tc Task Context
 704  */
 705 static void
 706 client_allow_send(void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 707 {
 708   struct MeshDataDescriptor *info = cls;
 709
 710   if (GNUNET_SCHEDULER_REASON_SHUTDOWN == tc->reason)
 711     return;
 712 #if MESH_DEBUG
 713   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 714               "MESH: CLIENT ALLOW SEND DESPITE %u COPIES PENDING\n",
 715               *(info->copies));
 716 #endif
 717   *(info->timeout_task) = GNUNET_SCHEDULER_NO_TASK;
 718   GNUNET_SERVER_receive_done(info->client, GNUNET_OK);
 719 }
 720
 721
 722 /**
 723  * Search for a tunnel by global ID using full PeerIdentities
 724  *
 725  * @param oid owner of the tunnel
 726  * @param tid global tunnel number
 727  *
 728  * @return tunnel handler, NULL if doesn't exist
 729  */
 730 static struct MeshTunnel *
 731 tunnel_get (struct GNUNET_PeerIdentity *oid, MESH_TunnelNumber tid);
 732
 733
 734 /**
 735  * Notify a tunnel that a connection has broken that affects at least
 736  * some of its peers.
 737  *
 738  * @param t Tunnel affected.
 739  * @param p1 Peer that got disconnected from p2.
 740  * @param p2 Peer that got disconnected from p1.
 741  *
 742  * @return Short ID of the peer disconnected (either p1 or p2).
 743  *         0 if the tunnel remained unaffected.
 744  */
 745 static GNUNET_PEER_Id
 746 tunnel_notify_connection_broken (struct MeshTunnel *t,
 747                                  GNUNET_PEER_Id p1,
 748                                  GNUNET_PEER_Id p2);
 749
 750
 751 /**
 752  * Send the message to all clients that have subscribed to its type
 753  *
 754  * @param msg Pointer to the message itself
 755  * @return number of clients this message was sent to
 756  */
 757 static unsigned int
 758 send_subscribed_clients (const struct GNUNET_MessageHeader *msg,
 759                          const struct GNUNET_MessageHeader *payload)
 760 {
 761   struct GNUNET_PeerIdentity *oid;
 762   struct MeshClient *c;
 763   struct MeshTunnel *t;
 764   MESH_TunnelNumber *tid;
 765   unsigned int count;
 766   uint16_t type;
 767   char cbuf[htons(msg->size)];
 768
 769   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Sending to clients...\n");
 770   type = ntohs (payload->type);
 771   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: message of type %u\n", type);
 772
 773   memcpy (cbuf, msg, sizeof(cbuf));
 774   switch (htons(msg->type))
 775   {
 776     struct GNUNET_MESH_Unicast *uc;
 777     struct GNUNET_MESH_Multicast *mc;
 778     struct GNUNET_MESH_ToOrigin *to;
 779
 780     case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 781       uc = (struct GNUNET_MESH_Unicast *) cbuf;
 782       tid = &uc->tid;
 783       oid = &uc->oid;
 784       break;
 785     case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 786       mc = (struct GNUNET_MESH_Multicast *) cbuf;
 787       tid = &mc->tid;
 788       oid = &mc->oid;
 789       break;
 790     case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 791       to = (struct GNUNET_MESH_ToOrigin *) cbuf;
 792       tid = &to->tid;
 793       oid = &to->oid;
 794       break;
 795     default:
 796       GNUNET_break (0);
 797       return 0;
 798   }
 799   t = tunnel_get (oid, ntohl(*tid));
 800   if (NULL == t)
 801   {
 802     GNUNET_break (0);
 803     return 0;
 804   }
 805   *tid = htonl (t->local_tid);
 806   for (count = 0, c = clients; c != NULL; c = c->next)
 807   {
 808     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    client %u\n", c->id);
 809     if (client_is_subscribed (type, c))
 810     {
 811       count++;
 812       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      sending\n");
 813       GNUNET_SERVER_notification_context_unicast (
 814           nc,
 815           c->handle,
 816           (struct GNUNET_MessageHeader *) cbuf,
 817           GNUNET_YES);
 818     }
 819   }
 820   return count;
 821 }
 822
 823
 824 /**
 825  * Notify the client that owns the tunnel that a peer has connected to it
 826  * (the requested path to it has been confirmed).
 827  *
 828  * @param t Tunnel whose owner to notify
 829  * @param id Short id of the peer that has connected
 830  */
 831 static void
 832 send_client_peer_connected (const struct MeshTunnel *t, const GNUNET_PEER_Id id)
 833 {
 834   struct GNUNET_MESH_PeerControl pc;
 835
 836   pc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD);
 837   pc.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
 838   pc.tunnel_id = htonl (t->local_tid);
 839   GNUNET_PEER_resolve (id, &pc.peer);
 840   GNUNET_SERVER_notification_context_unicast (nc, t->client->handle,
 841                                               &pc.header, GNUNET_NO);
 842 }
 843
 844
 845 /**
 846  * Notify all clients (not depending on registration status) that the incoming
 847  * tunnel is no longer valid.
 848  *
 849  * @param t Tunnel that was destroyed.
 850  */
 851 static void
 852 send_clients_tunnel_destroy (struct MeshTunnel *t)
 853 {
 854     struct GNUNET_MESH_TunnelMessage msg;
 855
 856     msg.header.size = htons (sizeof (msg));
 857     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
 858     msg.tunnel_id = htonl (t->local_tid);
 859     GNUNET_SERVER_notification_context_broadcast(nc, &msg.header, GNUNET_NO);
 860 }
 861
 862
 863 /**
 864  * Function called to notify a client about the socket
 865  * being ready to queue more data.  "buf" will be
 866  * NULL and "size" zero if the socket was closed for
 867  * writing in the meantime.
 868  *
 869  * @param cls closure
 870  * @param size number of bytes available in buf
 871  * @param buf where the callee should write the message
 872  * @return number of bytes written to buf
 873  */
 874 static size_t
 875 send_core_create_path (void *cls, size_t size, void *buf);
 876
 877
 878 /**
 879  * Function called to notify a client about the socket
 880  * being ready to queue more data.  "buf" will be
 881  * NULL and "size" zero if the socket was closed for
 882  * writing in the meantime.
 883  *
 884  * @param cls closure (data itself)
 885  * @param size number of bytes available in buf
 886  * @param buf where the callee should write the message
 887  *
 888  * @return number of bytes written to buf
 889  */
 890 static size_t
 891 send_core_data_multicast (void *cls, size_t size, void *buf);
 892
 893
 894 /**
 895  * Decrements the reference counter and frees all resources if needed
 896  *
 897  * @param dd Data Descriptor used in a multicast message
 898  */
 899 static void
 900 data_descriptor_decrement_multicast (struct MeshDataDescriptor *dd)
 901 {
 902   if (0 == --(*(dd->copies)))
 903   {
 904     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Last copy!\n");
 905     if (NULL != dd->client)
 906     {
 907       if (GNUNET_SCHEDULER_NO_TASK != *(dd->timeout_task))
 908       {
 909         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 910                     "MESH:  cancelling client timeout (%u)...\n",
 911                     *(dd->timeout_task));
 912         GNUNET_SCHEDULER_cancel(*(dd->timeout_task));
 913         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  notifying client...\n");
 914         GNUNET_SERVER_receive_done (dd->client, GNUNET_OK);
 915       }
 916       GNUNET_free (dd->timeout_task);
 917     }
 918     GNUNET_free (dd->copies);
 919     GNUNET_free (dd->data);
 920   }
 921 }
 922
 923
 924 /**
 925  * Cancel a core transmission that was already requested and free all resources
 926  * associated to the request.
 927  *
 928  * @param peer PeeInfo of the peer whose transmission is cancelled.
 929  * @param i Position of the transmission to be cancelled.
 930  */
 931 static void
 932 peer_info_cancel_transmission(struct MeshPeerInfo *peer, unsigned int i)
 933 {
 934   if (NULL != peer->core_transmit[i])
 935   {
 936     struct MeshDataDescriptor *dd;
 937     struct MeshPathInfo *path_info;
 938
 939 #if MESH_DEBUG
 940     {
 941       struct GNUNET_PeerIdentity id;
 942
 943       GNUNET_PEER_resolve (peer->id, &id);
 944       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 945                   "MESH:   Cancelling data transmission at %s [%u]\n",
 946                   GNUNET_i2s (&id),
 947                   i);
 948       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 949                   "MESH:    message type %u\n",
 950                   peer->types[i]);
 951     }
 952 #endif
 953     /* TODO: notify that tranmission has failed */
 954     switch (peer->types[i])
 955     {
 956       case GNUNET_MESSAGE_TYPE_MESH_MULTICAST:
 957       case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
 958       case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
 959         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    type payload\n");
 960         dd = peer->infos[i];
 961         data_descriptor_decrement_multicast (dd);
 962         break;
 963       case GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE:
 964         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    type create path\n");
 965         path_info = peer->infos[i];
 966         path_destroy(path_info->path);
 967         break;
 968       default:
 969         GNUNET_break (0);
 970         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    type unknown!\n");
 971     }
 972     GNUNET_CORE_notify_transmit_ready_cancel (peer->core_transmit[i]);
 973     peer->core_transmit[i] = NULL;
 974     GNUNET_free (peer->infos[i]);
 975   }
 976 }
 977
 978
 979 /**
 980  * Get a unused CORE slot to transmit a message to a peer. If all the slots
 981  * are used, cancel one and return it's position.
 982  *
 983  * @param peer PeerInfo of the neighbor we want to transmit to.
 984  *
 985  * @return The index of an available slot to transmit to the neighbor.
 986  */
 987 static unsigned int
 988 peer_info_transmit_slot (struct MeshPeerInfo *peer)
 989 {
 990   unsigned int i;
 991
 992   for (i = 0; peer->core_transmit[i]; i++)
 993   {
 994     if (i == (CORE_QUEUE_SIZE - 1))
 995     {
 996       /* All positions are taken! Overwriting! */
 997       GNUNET_break (0);
 998       peer_info_cancel_transmission(peer, 0);
 999       return 0;
1000     }
1001   }
1002   return i;
1003 }
1004
1005
1006 /**
1007  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
1008  * and insert it in the appropiate structures if the peer is not known yet.
1009  *
1010  * @param peer Full identity of the peer.
1011  *
1012  * @return Existing or newly created peer info.
1013  */
1014 static struct MeshPeerInfo *
1015 peer_info_get (const struct GNUNET_PeerIdentity *peer)
1016 {
1017   struct MeshPeerInfo *peer_info;
1018
1019   peer_info = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
1020   if (NULL == peer_info)
1021   {
1022     peer_info =
1023         (struct MeshPeerInfo *) GNUNET_malloc (sizeof (struct MeshPeerInfo));
1024     GNUNET_CONTAINER_multihashmap_put (peers, &peer->hashPubKey, peer_info,
1025                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1026     peer_info->id = GNUNET_PEER_intern (peer);
1027   }
1028
1029   return peer_info;
1030 }
1031
1032
1033 /**
1034  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
1035  * and insert it in the appropiate structures if the peer is not known yet.
1036  *
1037  * @param peer Short identity of the peer.
1038  *
1039  * @return Existing or newly created peer info.
1040  */
1041 static struct MeshPeerInfo *
1042 peer_info_get_short (const GNUNET_PEER_Id peer)
1043 {
1044   struct GNUNET_PeerIdentity id;
1045
1046   GNUNET_PEER_resolve(peer, &id);
1047   return peer_info_get(&id);
1048 }
1049
1050
1051 /**
1052  * Iterator to remove the tunnel from the list of tunnels a peer participates
1053  * in.
1054  *
1055  * @param cls Closure (tunnel info)
1056  * @param key GNUNET_PeerIdentity of the peer (unused)
1057  * @param value PeerInfo of the peer
1058  *
1059  * @return always GNUNET_YES, to keep iterating
1060  */
1061 static int
1062 peer_info_delete_tunnel (void* cls, const GNUNET_HashCode* key, void* value)
1063 {
1064   struct MeshTunnel *t = cls;
1065   struct MeshPeerInfo *peer = value;
1066   unsigned int i;
1067
1068   for (i = 0; i < peer->ntunnels; i++)
1069   {
1070     if (0 == memcmp(&peer->tunnels[i]->id,
1071                     &t->id,
1072                     sizeof(struct MESH_TunnelID)))
1073     {
1074       peer->ntunnels--;
1075       peer->tunnels[i] = peer->tunnels[peer->ntunnels];
1076       peer->tunnels = GNUNET_realloc (peer->tunnels, peer->ntunnels);
1077       return GNUNET_YES;
1078     }
1079   }
1080   return GNUNET_YES;
1081 }
1082
1083
1084 /**
1085   * Core callback to write a
1086   *
1087   * @param cls Closure (MeshDataDescriptor with data in "data" member).
1088   * @param size Number of bytes available in buf.
1089   * @param buf Where the to write the message.
1090   *
1091   * @return number of bytes written to buf
1092   */
1093 static size_t
1094 send_core_data_raw (void *cls, size_t size, void *buf)
1095 {
1096   struct MeshDataDescriptor *info = cls;
1097   struct GNUNET_MessageHeader *msg;
1098   size_t total_size;
1099
1100   GNUNET_assert (NULL != info);
1101   GNUNET_assert (NULL != info->data);
1102   msg = (struct GNUNET_MessageHeader *) info->data;
1103   total_size = ntohs (msg->size);
1104
1105   if (total_size > size)
1106   {
1107     struct GNUNET_PeerIdentity id;
1108
1109     GNUNET_PEER_resolve (info->peer->id, &id);
1110     info->peer->core_transmit[info->handler_n] =
1111       GNUNET_CORE_notify_transmit_ready(core_handle,
1112                                         0,
1113                                         100,
1114                                         GNUNET_TIME_UNIT_FOREVER_REL,
1115                                         &id,
1116                                         size,
1117                                         &send_core_data_raw,
1118                                         info);
1119       return 0;
1120   }
1121   info->peer->core_transmit[info->handler_n] = NULL;
1122   memcpy (buf, msg, total_size);
1123   GNUNET_free (info->data);
1124   GNUNET_free (info);
1125   return total_size;
1126 }
1127
1128
1129 /**
1130  * Sends an already built message to a peer, properly registrating
1131  * all used resources.
1132  *
1133  * @param message Message to send. Fucntion makes a copy of it.
1134  * @param peer Short ID of the neighbor whom to send the message.
1135  */
1136 static void
1137 send_message (const struct GNUNET_MessageHeader *message,
1138               const struct GNUNET_PeerIdentity *peer)
1139 {
1140   struct MeshDataDescriptor *info;
1141   struct MeshPeerInfo *neighbor;
1142   unsigned int i;
1143   size_t size;
1144
1145   size = ntohs (message->size);
1146   info = GNUNET_malloc (sizeof (struct MeshDataDescriptor));
1147   info->data = GNUNET_malloc (size);
1148   memcpy (info->data, message, size);
1149   neighbor = peer_info_get (peer);
1150   i = peer_info_transmit_slot (neighbor);
1151   info->handler_n = i;
1152   info->peer = neighbor;
1153   neighbor->types[i] = GNUNET_MESSAGE_TYPE_MESH_UNICAST;
1154   neighbor->infos[i] = info;
1155   neighbor->core_transmit[i] =
1156       GNUNET_CORE_notify_transmit_ready(core_handle,
1157                                         0,
1158                                         100,
1159                                         GNUNET_TIME_UNIT_FOREVER_REL,
1160                                         peer,
1161                                         size,
1162                                         &send_core_data_raw,
1163                                         info);
1164
1165 }
1166
1167
1168 /**
1169  * Sends a CREATE PATH message for a path to a peer, properly registrating
1170  * all used resources.
1171  *
1172  * @param peer PeerInfo of the final peer for whom this path is being created.
1173  * @param p Path itself.
1174  * @param t Tunnel for which the path is created.
1175  */
1176 static void
1177 send_create_path (struct MeshPeerInfo *peer,
1178                   struct MeshPeerPath *p,
1179                   struct MeshTunnel *t)
1180 {
1181   struct GNUNET_PeerIdentity id;
1182   struct MeshPathInfo *path_info;
1183   struct MeshPeerInfo *neighbor;
1184   unsigned int i;
1185
1186   if (NULL == p)
1187   {
1188     p = tree_get_path_to_peer(t->tree, peer->id);
1189     if (NULL == p)
1190     {
1191       GNUNET_break (0);
1192       return;
1193     }
1194   }
1195   for (i = 0; i < p->length; i++)
1196   {
1197     if (p->peers[i] == myid)
1198       break;
1199   }
1200   if (i >= p->length - 1)
1201   {
1202     path_destroy(p);
1203     GNUNET_break (0);
1204     return;
1205   }
1206   GNUNET_PEER_resolve(p->peers[i + 1], &id);
1207
1208   path_info = GNUNET_malloc (sizeof (struct MeshPathInfo));
1209   path_info->path = p;
1210   path_info->t = t;
1211   neighbor = peer_info_get(&id);
1212   path_info->peer = neighbor;
1213   path_info->pos = peer_info_transmit_slot(neighbor);
1214   neighbor->types[path_info->pos] = GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE;
1215   neighbor->infos[path_info->pos] = path_info;
1216   neighbor->core_transmit[path_info->pos] =
1217       GNUNET_CORE_notify_transmit_ready (
1218           core_handle, /* handle */
1219           0, /* cork */
1220           0, /* priority */
1221           GNUNET_TIME_UNIT_FOREVER_REL, /* timeout */
1222           &id, /* target */
1223           sizeof (struct GNUNET_MESH_ManipulatePath)
1224           + (p->length * sizeof (struct GNUNET_PeerIdentity)), /*size */
1225           &send_core_create_path, /* callback */
1226           path_info);        /* cls */
1227 }
1228
1229
1230 /**
1231  * Sends a DESTROY PATH message to free resources for a path in a tunnel
1232  *
1233  * @param t Tunnel whose path to destroy.
1234  * @param destination Short ID of the peer to whom the path to destroy.
1235  */
1236 static void
1237 send_destroy_path (struct MeshTunnel *t, GNUNET_PEER_Id destination)
1238 {
1239   struct MeshPeerPath *p;
1240   size_t size;
1241
1242   p = tree_get_path_to_peer(t->tree, destination);
1243   if (NULL == p)
1244   {
1245     GNUNET_break (0);
1246     return;
1247   }
1248   size = sizeof (struct GNUNET_MESH_ManipulatePath);
1249   size += p->length * sizeof (struct GNUNET_PeerIdentity);
1250   {
1251     struct GNUNET_MESH_ManipulatePath *msg;
1252     struct GNUNET_PeerIdentity *pi;
1253     char cbuf[size];
1254     unsigned int i;
1255
1256     msg = (struct GNUNET_MESH_ManipulatePath *) cbuf;
1257     msg->header.size = htons (size);
1258     msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_DESTROY);
1259     msg->tid = htonl (t->id.tid);
1260     pi = (struct GNUNET_PeerIdentity *) &msg[1];
1261     for (i = 0; i < p->length; i++)
1262     {
1263       GNUNET_PEER_resolve(p->peers[i], &pi[i]);
1264     }
1265     send_message (&msg->header, path_get_first_hop(t->tree, destination));
1266   }
1267   path_destroy (p);
1268 }
1269
1270
1271 /**
1272  * Try to establish a new connection to this peer.
1273  * Use the best path for the given tunnel.
1274  * If the peer doesn't have any path to it yet, try to get one.
1275  * If the peer already has some path, send a CREATE PATH towards it.
1276  *
1277  * @param peer PeerInfo of the peer.
1278  * @param t Tunnel for which to create the path, if possible.
1279  */
1280 static void
1281 peer_info_connect (struct MeshPeerInfo *peer, struct MeshTunnel *t)
1282 {
1283   struct MeshPeerPath *p;
1284   struct MeshPathInfo *path_info;
1285
1286   if (NULL != peer->path_head)
1287   {
1288     p = tree_get_path_to_peer(t->tree, peer->id);
1289     if (p->length > 1)
1290     {
1291       send_create_path(peer, p, t);
1292     }
1293     else
1294     {
1295       path_destroy(p);
1296       send_client_peer_connected(t, myid);
1297     }
1298   }
1299   else if (NULL == peer->dhtget)
1300   {
1301     struct GNUNET_PeerIdentity id;
1302
1303     GNUNET_PEER_resolve(peer->id, &id);
1304     path_info = GNUNET_malloc(sizeof(struct MeshPathInfo));
1305     path_info->peer = peer;
1306     path_info->t = t;
1307     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1308                 "MESH:   Starting DHT GET for peer %s\n",
1309                 GNUNET_i2s (&id));
1310     peer->dhtgetcls = path_info;
1311     peer->dhtget =
1312         GNUNET_DHT_get_start(dht_handle,       /* handle */
1313                              GNUNET_TIME_UNIT_FOREVER_REL,     /* timeout */
1314                              GNUNET_BLOCK_TYPE_TEST,   /* type */
1315                              &id.hashPubKey,       /* key to search */
1316                              4,         /* replication level */
1317                              GNUNET_DHT_RO_RECORD_ROUTE |
1318                                GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
1319                              NULL,     /* xquery */
1320                              0,        /* xquery bits */
1321                              &dht_get_id_handler,
1322                              path_info);
1323   }
1324   /* Otherwise, there is no path but the DHT get is already started. */
1325 }
1326
1327
1328 /**
1329  * Destroy the peer_info and free any allocated resources linked to it
1330  *
1331  * @param pi The peer_info to destroy.
1332  *
1333  * @return GNUNET_OK on success
1334  */
1335 static int
1336 peer_info_destroy (struct MeshPeerInfo *pi)
1337 {
1338   struct GNUNET_PeerIdentity id;
1339   struct MeshPeerPath *p;
1340   struct MeshPeerPath *nextp;
1341   unsigned int i;
1342
1343   GNUNET_PEER_resolve (pi->id, &id);
1344   GNUNET_PEER_change_rc (pi->id, -1);
1345
1346   if (GNUNET_YES !=
1347       GNUNET_CONTAINER_multihashmap_remove (peers,&id.hashPubKey, pi))
1348   {
1349     GNUNET_break (0);
1350     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1351                 "MESH: removing peer %s, not in hashmap\n",
1352                 GNUNET_i2s (&id));
1353   }
1354   if (NULL != pi->dhtget)
1355   {
1356     GNUNET_DHT_get_stop(pi->dhtget);
1357     GNUNET_free (pi->dhtgetcls);
1358   }
1359   for (i = 0; i < CORE_QUEUE_SIZE; i++)
1360   {
1361     peer_info_cancel_transmission(pi, i);
1362   }
1363   p = pi->path_head;
1364   while (NULL != p)
1365   {
1366     nextp = p->next;
1367     GNUNET_CONTAINER_DLL_remove (pi->path_head, pi->path_tail, p);
1368     path_destroy (p);
1369     p = nextp;
1370   }
1371   GNUNET_free (pi);
1372   return GNUNET_OK;
1373 }
1374
1375
1376 /**
1377  * Remove all paths that rely on a direct connection between p1 and p2
1378  * from the peer itself and notify all tunnels about it.
1379  *
1380  * @param peer PeerInfo of affected peer.
1381  * @param p1 GNUNET_PEER_Id of one peer.
1382  * @param p2 GNUNET_PEER_Id of another peer that was connected to the first and
1383  *           no longer is.
1384  *
1385  * TODO: optimize (see below)
1386  */
1387 static void
1388 path_remove_from_peer (struct MeshPeerInfo *peer,
1389                        GNUNET_PEER_Id p1,
1390                        GNUNET_PEER_Id p2)
1391 {
1392   struct MeshPeerPath *p;
1393   struct MeshPeerPath *aux;
1394   struct MeshPeerInfo *peer_d;
1395   GNUNET_PEER_Id d;
1396   unsigned int destroyed;
1397   unsigned int best;
1398   unsigned int cost;
1399   unsigned int i;
1400
1401   destroyed = 0;
1402   p = peer->path_head;
1403   while (NULL != p)
1404   {
1405     aux = p->next;
1406     for (i = 0; i < (p->length - 1); i++)
1407     {
1408       if ((p->peers[i] == p1 && p->peers[i + 1] == p2) ||
1409           (p->peers[i] == p2 && p->peers[i + 1] == p1))
1410       {
1411         GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
1412         path_destroy (p);
1413         destroyed++;
1414         break;
1415       }
1416     }
1417     p = aux;
1418   }
1419   if (0 == destroyed)
1420     return;
1421
1422   for (i = 0; i < peer->ntunnels; i++)
1423   {
1424     d = tunnel_notify_connection_broken (peer->tunnels[i], p1, p2);
1425     /* TODO
1426      * Problem: one or more peers have been deleted from the tunnel tree.
1427      * We don't know who they are to try to add them again.
1428      * We need to try to find a new path for each of the disconnected peers.
1429      * Some of them might already have a path to reach them that does not
1430      * involve p1 and p2. Adding all anew might render in a better tree than
1431      * the trivial immediate fix.
1432      *
1433      * Trivial immiediate fix: try to reconnect to the disconnected node. All
1434      * its children will be reachable trough him.
1435      */
1436     peer_d = peer_info_get_short(d);
1437     best = UINT_MAX;
1438     aux = NULL;
1439     for (p = peer_d->path_head; NULL != p; p = p->next)
1440     {
1441       if ((cost = path_get_cost(peer->tunnels[i]->tree, p)) < best)
1442       {
1443         best = cost;
1444         aux = p;
1445       }
1446     }
1447     if (NULL != aux)
1448     {
1449       /* No callback, as peer will be already disconnected */
1450       tree_add_path(peer->tunnels[i]->tree, aux, NULL);
1451     }
1452     else
1453     {
1454       peer_info_connect (peer_d, peer->tunnels[i]);
1455     }
1456   }
1457 }
1458
1459
1460 /**
1461  * Add the path to the peer and update the path used to reach it in case this
1462  * is the shortest.
1463  *
1464  * @param peer_info Destination peer to add the path to.
1465  * @param path New path to add. Last peer must be the peer in arg 1.
1466  *             Path will be either used of freed if already known.
1467  *
1468  * TODO: trim the part from origin to us? Add it as path to origin?
1469  */
1470 void
1471 path_add_to_peer (struct MeshPeerInfo *peer_info, struct MeshPeerPath *path)
1472 {
1473   struct MeshPeerPath *aux;
1474   unsigned int l;
1475   unsigned int l2;
1476
1477 #if MESH_DEBUG
1478   struct GNUNET_PeerIdentity id;
1479
1480   GNUNET_PEER_resolve (peer_info->id, &id);
1481   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1482               "MESH: adding path [%u] to peer %s\n",
1483               path->length,
1484               GNUNET_i2s (&id));
1485 #endif
1486   if (NULL == peer_info || NULL == path)
1487   {
1488     GNUNET_break (0);
1489     path_destroy (path);
1490     return;
1491   }
1492
1493   l = path_get_length (path);
1494   if (0 == l)
1495   {
1496     GNUNET_free (path);
1497     return;
1498   }
1499
1500   for (aux = peer_info->path_head; aux != NULL; aux = aux->next)
1501   {
1502     l2 = path_get_length (aux);
1503     if (l2 > l)
1504     {
1505       GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head,
1506                                           peer_info->path_tail,
1507                                           aux,
1508                                           path);
1509       return;
1510     }
1511     else
1512     {
1513       if (l2 == l && memcmp(path->peers, aux->peers, l) == 0)
1514       {
1515         path_destroy(path);
1516         return;
1517       }
1518     }
1519   }
1520   GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head,
1521                                     peer_info->path_tail,
1522                                     path);
1523   return;
1524 }
1525
1526
1527 /**
1528  * Add the path to the origin peer and update the path used to reach it in case
1529  * this is the shortest.
1530  * The path is given in peer_info -> destination, therefore we turn the path
1531  * upside down first.
1532  *
1533  * @param peer_info Peer to add the path to, being the origin of the path.
1534  * @param path New path to add after being inversed.
1535  */
1536 static void
1537 path_add_to_origin (struct MeshPeerInfo *peer_info, struct MeshPeerPath *path)
1538 {
1539   path_invert(path);
1540   path_add_to_peer (peer_info, path);
1541 }
1542
1543
1544 /**
1545  * Build a PeerPath from the paths returned from the DHT, reversing the paths
1546  * to obtain a local peer -> destination path and interning the peer ids.
1547  *
1548  * @return Newly allocated and created path
1549  */
1550 static struct MeshPeerPath *
1551 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
1552                      unsigned int get_path_length,
1553                      const struct GNUNET_PeerIdentity *put_path,
1554                      unsigned int put_path_length)
1555 {
1556   struct MeshPeerPath *p;
1557   GNUNET_PEER_Id id;
1558   int i;
1559
1560   p = path_new (1);
1561   p->peers[0] = myid;
1562   GNUNET_PEER_change_rc(myid, 1);
1563   i = get_path_length;
1564   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    GET has %d hops.\n", i);
1565   for (i--; i >= 0; i--)
1566   {
1567     id = GNUNET_PEER_intern (&get_path[i]);
1568     if (p->length > 0 && id == p->peers[p->length - 1])
1569     {
1570       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    Optimizing 1 hop out.\n");
1571       GNUNET_PEER_change_rc(id, -1);
1572     }
1573     else
1574     {
1575       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1576                  "MESH:    Adding from GET: %s.\n",
1577                  GNUNET_i2s(&get_path[i]));
1578       p->length++;
1579       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
1580       p->peers[p->length - 1] = id;
1581     }
1582   }
1583   i = put_path_length;
1584   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    PUT has %d hops.\n", i);
1585   for (i--; i >= 0; i--)
1586   {
1587     id = GNUNET_PEER_intern (&put_path[i]);
1588     if (id == myid)
1589     {
1590       /* PUT path went through us, so discard the path up until now and start
1591        * from here to get a much shorter (and loop-free) path.
1592        */
1593       path_destroy (p);
1594       p = path_new (0);
1595     }
1596     if (p->length > 0 && id == p->peers[p->length - 1])
1597     {
1598       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    Optimizing 1 hop out.\n");
1599       GNUNET_PEER_change_rc(id, -1);
1600     }
1601     else
1602     {
1603       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1604             "MESH:    Adding from PUT: %s.\n",
1605             GNUNET_i2s(&put_path[i]));
1606       p->length++;
1607       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
1608       p->peers[p->length - 1] = id;
1609     }
1610   }
1611 #if MESH_DEBUG
1612   if (get_path_length > 0)
1613     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1614                 "MESH:    (first of GET: %s)\n",
1615                 GNUNET_i2s(&get_path[0]));
1616   if (put_path_length > 0)
1617     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1618                 "MESH:    (first of PUT: %s)\n",
1619                 GNUNET_i2s(&put_path[0]));
1620   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1621               "MESH:    In total: %d hops\n",
1622               p->length);
1623   for (i = 0; i < p->length; i++)
1624   {
1625     struct GNUNET_PeerIdentity peer_id;
1626
1627     GNUNET_PEER_resolve(p->peers[i], &peer_id);
1628     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1629               "MESH:        %u: %s\n",
1630               p->peers[i],
1631               GNUNET_i2s(&peer_id));
1632   }
1633 #endif
1634   return p;
1635 }
1636
1637
1638 /**
1639  * Send keepalive packets for a peer
1640  *
1641  * @param cls Closure (tunnel for which to send the keepalive).
1642  * @param tc Notification context.
1643  *
1644  * TODO: implement explicit multicast keepalive?
1645  */
1646 static void
1647 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1648
1649
1650 /**
1651  * Search for a tunnel among the tunnels for a client
1652  *
1653  * @param c the client whose tunnels to search in
1654  * @param tid the local id of the tunnel
1655  *
1656  * @return tunnel handler, NULL if doesn't exist
1657  */
1658 static struct MeshTunnel *
1659 tunnel_get_by_local_id (struct MeshClient *c, MESH_TunnelNumber tid)
1660 {
1661   GNUNET_HashCode hash;
1662
1663   GNUNET_CRYPTO_hash (&tid, sizeof (MESH_TunnelNumber), &hash);
1664   if (tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1665   {
1666     return GNUNET_CONTAINER_multihashmap_get (incoming_tunnels, &hash);
1667   }
1668   return GNUNET_CONTAINER_multihashmap_get (c->tunnels, &hash);
1669 }
1670
1671
1672 /**
1673  * Search for a tunnel by global ID using PEER_ID
1674  *
1675  * @param pi owner of the tunnel
1676  * @param tid global tunnel number
1677  *
1678  * @return tunnel handler, NULL if doesn't exist
1679  */
1680 static struct MeshTunnel *
1681 tunnel_get_by_pi (GNUNET_PEER_Id pi, MESH_TunnelNumber tid)
1682 {
1683   struct MESH_TunnelID id;
1684   GNUNET_HashCode hash;
1685
1686   id.oid = pi;
1687   id.tid = tid;
1688
1689   GNUNET_CRYPTO_hash (&id, sizeof (struct MESH_TunnelID), &hash);
1690   return GNUNET_CONTAINER_multihashmap_get (tunnels, &hash);
1691 }
1692
1693
1694 /**
1695  * Search for a tunnel by global ID using full PeerIdentities
1696  *
1697  * @param oid owner of the tunnel
1698  * @param tid global tunnel number
1699  *
1700  * @return tunnel handler, NULL if doesn't exist
1701  */
1702 static struct MeshTunnel *
1703 tunnel_get (struct GNUNET_PeerIdentity *oid, MESH_TunnelNumber tid)
1704 {
1705   return tunnel_get_by_pi (GNUNET_PEER_search (oid), tid);
1706 }
1707
1708
1709 /**
1710  * Callback used to notify a client owner of a tunnel that a peer has
1711  * disconnected, most likely because of a path change.
1712  *
1713  * @param n Node in the tree representing the disconnected peer
1714  *
1715  * FIXME: pass tunnel via cls, make param just a peer identity
1716  */
1717 void
1718 notify_peer_disconnected (const struct MeshTunnelTreeNode *n)
1719 {
1720   struct MeshPeerInfo *peer;
1721
1722   if (NULL != n->t->client && NULL != nc)
1723   {
1724     struct GNUNET_MESH_PeerControl msg;
1725     msg.header.size = htons (sizeof (msg));
1726     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1727     msg.tunnel_id = htonl (n->t->local_tid);
1728     GNUNET_PEER_resolve (n->peer, &msg.peer);
1729     GNUNET_SERVER_notification_context_unicast (nc, n->t->client->handle,
1730                                                 &msg.header, GNUNET_NO);
1731   }
1732   peer = peer_info_get_short(n->peer);
1733   peer_info_connect(peer, n->t);
1734 }
1735
1736
1737 /**
1738  * Add a peer to a tunnel, accomodating paths accordingly and initializing all
1739  * needed rescources.
1740  * If peer already exists, reevaluate shortest path and change if different.
1741  *
1742  * @param t Tunnel we want to add a new peer to
1743  * @param peer PeerInfo of the peer being added
1744  *
1745  */
1746 static void
1747 tunnel_add_peer (struct MeshTunnel *t, struct MeshPeerInfo *peer)
1748 {
1749   struct GNUNET_PeerIdentity id;
1750   struct MeshPeerPath *best_p;
1751   struct MeshPeerPath *p;
1752   unsigned int best_cost;
1753   unsigned int cost;
1754
1755   GNUNET_PEER_resolve(peer->id, &id);
1756   if (GNUNET_NO ==
1757       GNUNET_CONTAINER_multihashmap_contains(t->peers, &id.hashPubKey))
1758   {
1759     t->peers_total++;
1760     GNUNET_array_append (peer->tunnels, peer->ntunnels, t);
1761     GNUNET_CONTAINER_multihashmap_put(
1762       t->peers,
1763       &id.hashPubKey,
1764       peer,
1765       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1766   }
1767
1768   if (NULL != (p = peer->path_head))
1769   {
1770     best_p = p;
1771     best_cost = path_get_cost(t->tree, p);
1772     while (NULL != p)
1773     {
1774       if ((cost = path_get_cost (t->tree, p)) < best_cost)
1775       {
1776         best_cost = cost;
1777         best_p = p;
1778       }
1779       p = p->next;
1780     }
1781     tree_add_path (t->tree, best_p, &notify_peer_disconnected);
1782     if (GNUNET_SCHEDULER_NO_TASK == t->path_refresh_task)
1783       t->path_refresh_task =
1784           GNUNET_SCHEDULER_add_delayed (t->tree->refresh, &path_refresh, t);
1785   }
1786   else
1787   {
1788     /* Start a DHT get if necessary */
1789     peer_info_connect(peer, t);
1790   }
1791 }
1792
1793 /**
1794  * Add a path to a tunnel which we don't own, just to remember the next hop.
1795  * If destination node was already in the tunnel, the first hop information
1796  * will be replaced with the new path.
1797  *
1798  * @param t Tunnel we want to add a new peer to
1799  * @param p Path to add
1800  * @param own_pos Position of local node in path.
1801  *
1802  */
1803 static void
1804 tunnel_add_path (struct MeshTunnel *t,
1805                  struct MeshPeerPath *p,
1806                  unsigned int own_pos)
1807 {
1808   struct GNUNET_PeerIdentity id;
1809
1810   GNUNET_assert (0 != own_pos);
1811   tree_add_path(t->tree, p, NULL);
1812   if (NULL == t->tree->me)
1813     t->tree->me = tree_find_peer(t->tree->root, p->peers[own_pos]);
1814   if (own_pos < p->length - 1)
1815   {
1816     GNUNET_PEER_resolve (p->peers[own_pos + 1], &id);
1817     tree_update_first_hops(t->tree, t->tree->me, &id);
1818   }
1819 }
1820
1821
1822 /**
1823  * Notifies a tunnel that a connection has broken that affects at least
1824  * some of its peers. Sends a notification towards the root of the tree.
1825  * In case the peer is the owner of the tree, notifies the client that owns
1826  * the tunnel and tries to reconnect.
1827  *
1828  * @param t Tunnel affected.
1829  * @param p1 Peer that got disconnected from p2.
1830  * @param p2 Peer that got disconnected from p1.
1831  *
1832  * @return Short ID of the peer disconnected (either p1 or p2).
1833  *         0 if the tunnel remained unaffected.
1834  */
1835 static GNUNET_PEER_Id
1836 tunnel_notify_connection_broken (struct MeshTunnel *t,
1837                                  GNUNET_PEER_Id p1,
1838                                  GNUNET_PEER_Id p2)
1839 {
1840   GNUNET_PEER_Id pid;
1841
1842   pid = tree_notify_connection_broken (t->tree,
1843                                        p1,
1844                                        p2,
1845                                        &notify_peer_disconnected);
1846   if (myid != p1 && myid != p2)
1847   {
1848     return pid;
1849   }
1850   if (pid != myid)
1851   {
1852     if (NULL != t->tree->me->parent)
1853     {
1854       /* We are the peer still connected, notify owner of the disconnection. */
1855       struct GNUNET_MESH_PathBroken msg;
1856       struct GNUNET_PeerIdentity neighbor;
1857
1858       msg.header.size = htons (sizeof (msg));
1859       msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN);
1860       GNUNET_PEER_resolve (t->id.oid, &msg.oid);
1861       msg.tid = htonl (t->id.tid);
1862       msg.peer1 = my_full_id;
1863       GNUNET_PEER_resolve (pid, &msg.peer2);
1864       GNUNET_PEER_resolve (t->tree->me->parent->peer, &neighbor);
1865       send_message (&msg.header, &neighbor);
1866     }
1867   }
1868   return pid;
1869 }
1870
1871
1872 /**
1873  * Send a message in a tunnel in multicast, sending a copy to each child node
1874  * down the local one in the tunnel tree.
1875  *
1876  * @param t Tunnel in which to send the data.
1877  * @param msg Message to be sent
1878  *
1879  * @return Number of copies sent.
1880  *
1881  * TODO: unifiy shared resources and reference counter management
1882  */
1883 static int
1884 tunnel_send_multicast (struct MeshTunnel *t,
1885                        const struct GNUNET_MessageHeader *msg)
1886 {
1887   struct GNUNET_PeerIdentity neighbor;
1888   struct MeshDataDescriptor *info;
1889   struct MeshTunnelTreeNode *n;
1890   struct MeshTunnelTreeNode *counter;
1891   GNUNET_SCHEDULER_TaskIdentifier *task;
1892   unsigned int *copies;
1893   unsigned int i;
1894   size_t size;
1895   void *data;
1896
1897   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1898               "MESH:  sending a multicast packet...\n");
1899   size = ntohs (msg->size);
1900   GNUNET_assert (NULL != t->tree->me);
1901   n = t->tree->me->children_head;
1902   if (NULL == n)
1903   {
1904     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1905               "MESH:  no children in the tree, no one to send.\n");
1906     return 0;
1907   }
1908   copies = GNUNET_malloc (sizeof (unsigned int));
1909   for (counter = n; NULL != counter; counter = counter->next)
1910     (*copies)++;
1911   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  (%u copies)\n", *copies);
1912   data = GNUNET_malloc (size);
1913   memcpy (data, msg, size);
1914   if (NULL != t->client)
1915   {
1916     task = GNUNET_malloc (sizeof (GNUNET_SCHEDULER_TaskIdentifier));
1917     *task = GNUNET_SCHEDULER_add_delayed (UNACKNOWLEDGED_WAIT,
1918                                          &client_allow_send,
1919                                          t->client->handle);
1920   }
1921   else
1922     task = NULL; // So GCC shuts up about task being potentially uninitialized
1923   while (NULL != n)
1924   {
1925     info = GNUNET_malloc (sizeof (struct MeshDataDescriptor));
1926     info->data = data;
1927     info->size = size;
1928     info->copies = copies;
1929     if (NULL != t->client)
1930     {
1931       info->client = t->client->handle;
1932       info->timeout_task = task;
1933     }
1934     info->destination = n->peer;
1935     GNUNET_PEER_resolve (n->peer, &neighbor);
1936     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1937               "MESH:  sending to %s...\n",
1938               GNUNET_i2s (&neighbor));
1939     info->peer = peer_info_get(&neighbor);
1940     GNUNET_assert (NULL != info->peer);
1941     i = peer_info_transmit_slot(info->peer);
1942     info->handler_n = i;
1943     info->peer->infos[i] = info;
1944     info->peer->types[i] = GNUNET_MESSAGE_TYPE_MESH_MULTICAST;
1945     info->peer->core_transmit[i] =
1946         GNUNET_CORE_notify_transmit_ready (core_handle,
1947                                            0,
1948                                            0,
1949                                            GNUNET_TIME_UNIT_FOREVER_REL,
1950                                            &neighbor,
1951                                            size,
1952                                            &send_core_data_multicast, info);
1953     n = n->next;
1954   }
1955   return *copies;
1956 }
1957
1958
1959 /**
1960  * Send a message to all peers in this tunnel that the tunnel is no longer
1961  * valid.
1962  *
1963  * @param t The tunnel whose peers to notify.
1964  */
1965 static void
1966 tunnel_send_destroy (struct MeshTunnel *t)
1967 {
1968   struct GNUNET_MESH_TunnelDestroy msg;
1969
1970   msg.header.size = htons (sizeof (msg));
1971   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY);
1972   GNUNET_PEER_resolve (t->id.oid, &msg.oid);
1973   msg.tid = htonl (t->id.tid);
1974   tunnel_send_multicast (t, &msg.header);
1975 }
1976
1977
1978
1979 /**
1980  * Destroy the tunnel and free any allocated resources linked to it
1981  *
1982  * @param t the tunnel to destroy
1983  *
1984  * @return GNUNET_OK on success
1985  */
1986 static int
1987 tunnel_destroy (struct MeshTunnel *t)
1988 {
1989   struct MeshClient *c;
1990   struct MeshQueue *q;
1991   struct MeshQueue *qn;
1992   GNUNET_HashCode hash;
1993   int r;
1994
1995   if (NULL == t)
1996     return GNUNET_OK;
1997
1998   c = t->client;
1999 #if MESH_DEBUG
2000   {
2001     struct GNUNET_PeerIdentity id;
2002
2003     GNUNET_PEER_resolve(t->id.oid, &id);
2004     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2005                 "MESH: destroying tunnel %s [%x]\n",
2006                 GNUNET_i2s (&id),
2007                 t->id.tid);
2008     if (NULL != c)
2009       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
2010   }
2011 #endif
2012
2013   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2014   if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (tunnels, &hash, t))
2015   {
2016     r = GNUNET_SYSERR;
2017   }
2018
2019   GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2020   if (NULL != c &&
2021       GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (c->tunnels, &hash, t))
2022   {
2023     r = GNUNET_SYSERR;
2024   }
2025   if (t->local_tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
2026   {
2027     GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2028     GNUNET_break (GNUNET_YES ==
2029       GNUNET_CONTAINER_multihashmap_remove (incoming_tunnels, &hash, t));
2030   }
2031
2032   if (NULL != t->peers)
2033   {
2034     GNUNET_CONTAINER_multihashmap_iterate(t->peers,
2035                                           &peer_info_delete_tunnel, t);
2036     GNUNET_CONTAINER_multihashmap_destroy (t->peers);
2037   }
2038   q = t->queue_head;
2039   while (NULL != q)
2040   {
2041     if (NULL != q->data)
2042       GNUNET_free (q->data);
2043     qn = q->next;
2044     GNUNET_free (q);
2045     q = qn;
2046     /* TODO cancel core transmit ready in case it was active */
2047   }
2048   tree_destroy(t->tree);
2049   if (NULL != t->dht_get_type)
2050     GNUNET_DHT_get_stop(t->dht_get_type);
2051   if (GNUNET_SCHEDULER_NO_TASK != t->timeout_task)
2052     GNUNET_SCHEDULER_cancel(t->timeout_task);
2053   if (GNUNET_SCHEDULER_NO_TASK != t->path_refresh_task)
2054     GNUNET_SCHEDULER_cancel(t->path_refresh_task);
2055   GNUNET_free (t);
2056   return r;
2057 }
2058
2059
2060 /**
2061  * Removes an explicit path from a tunnel, freeing all intermediate nodes
2062  * that are no longer needed, as well as nodes of no longer reachable peers.
2063  * The tunnel itself is also destoyed if results in a remote empty tunnel.
2064  *
2065  * @param t Tunnel from which to remove the path.
2066  * @param p Peer which should be removed.
2067  */
2068 static void
2069 tunnel_delete_peer (struct MeshTunnel *t,
2070                     GNUNET_PEER_Id peer)
2071 {
2072   GNUNET_break (GNUNET_OK == tree_del_peer (t->tree, peer, NULL));
2073   if (NULL == t->tree->root)
2074     tunnel_destroy (t);
2075 }
2076
2077
2078 /**
2079  * tunnel_destroy_iterator: iterator for deleting each tunnel that belongs to a
2080  * client when the client disconnects.
2081  *
2082  * @param cls closure (client that is disconnecting)
2083  * @param key the hash of the local tunnel id (used to access the hashmap)
2084  * @param value the value stored at the key (tunnel to destroy)
2085  *
2086  * @return GNUNET_OK on success
2087  */
2088 static int
2089 tunnel_destroy_iterator (void *cls, const GNUNET_HashCode * key, void *value)
2090 {
2091   struct MeshTunnel *t = value;
2092   int r;
2093
2094   if (NULL != t->dht_get_type)
2095   {
2096     GNUNET_DHT_get_stop (t->dht_get_type);
2097   }
2098   r = tunnel_destroy (t);
2099   return r;
2100 }
2101
2102
2103 /**
2104  * Timeout function, destroys tunnel if called
2105  *
2106  * @param cls Closure (tunnel to destroy).
2107  * @param tc TaskContext
2108  */
2109 static void
2110 tunnel_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2111 {
2112   struct MeshTunnel *t = cls;
2113
2114   if (GNUNET_SCHEDULER_REASON_SHUTDOWN == tc->reason)
2115     return;
2116   t->timeout_task = GNUNET_SCHEDULER_NO_TASK;
2117   tunnel_destroy (t);
2118 }
2119
2120 /**
2121  * Resets the tunnel timeout. Starts it if no timeout was running.
2122  *
2123  * @param t Tunnel whose timeout to reset.
2124  */
2125 static void
2126 tunnel_reset_timeout (struct MeshTunnel *t)
2127 {
2128   if (GNUNET_SCHEDULER_NO_TASK != t->timeout_task)
2129     GNUNET_SCHEDULER_cancel (t->timeout_task);
2130   t->timeout_task = GNUNET_SCHEDULER_add_delayed (
2131       GNUNET_TIME_relative_multiply(REFRESH_PATH_TIME, 4),
2132       &tunnel_timeout,
2133       t);
2134 }
2135
2136
2137 /******************************************************************************/
2138 /****************      MESH NETWORK HANDLER HELPERS     ***********************/
2139 /******************************************************************************/
2140
2141 /**
2142  * Function called to notify a client about the socket
2143  * being ready to queue more data.  "buf" will be
2144  * NULL and "size" zero if the socket was closed for
2145  * writing in the meantime.
2146  *
2147  * @param cls closure
2148  * @param size number of bytes available in buf
2149  * @param buf where the callee should write the message
2150  * @return number of bytes written to buf
2151  */
2152 static size_t
2153 send_core_create_path (void *cls, size_t size, void *buf)
2154 {
2155   struct MeshPathInfo *info = cls;
2156   struct GNUNET_MESH_ManipulatePath *msg;
2157   struct GNUNET_PeerIdentity *peer_ptr;
2158   struct MeshPeerInfo *peer = info->peer;
2159   struct MeshTunnel *t = info->t;
2160   struct MeshPeerPath *p = info->path;
2161   size_t size_needed;
2162   int i;
2163
2164   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2165               "MESH: CREATE PATH sending...\n");
2166   size_needed =
2167       sizeof (struct GNUNET_MESH_ManipulatePath) +
2168       p->length * sizeof (struct GNUNET_PeerIdentity);
2169
2170   if (size < size_needed || NULL == buf)
2171   {
2172     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: create path retransmit!\n");
2173     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   buf:  %p\n", buf);
2174     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   size: (%u/%u)\n",
2175                 size, size_needed);
2176     info->peer->core_transmit[info->pos] =
2177       GNUNET_CORE_notify_transmit_ready (core_handle, 0, 0,
2178                                         GNUNET_TIME_UNIT_FOREVER_REL,
2179                                         path_get_first_hop (t->tree, peer->id),
2180                                         size_needed,
2181                                         &send_core_create_path,
2182                                         info);
2183     return 0;
2184   }
2185   info->peer->core_transmit[info->pos] = NULL;
2186 #if MESH_DEBUG
2187   {
2188     struct GNUNET_PeerIdentity id;
2189
2190     GNUNET_PEER_resolve (peer->id, &id);
2191     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2192                 "MESH:   setting core_transmit %s [%u] to NULL\n",
2193                 GNUNET_i2s (&id),
2194                 info->pos);
2195   }
2196 #endif
2197   msg = (struct GNUNET_MESH_ManipulatePath *) buf;
2198   msg->header.size = htons (size_needed);
2199   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE);
2200   msg->tid = ntohl (t->id.tid);
2201
2202   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
2203   for (i = 0; i < p->length; i++)
2204   {
2205     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
2206   }
2207
2208   path_destroy (p);
2209   GNUNET_free (info);
2210
2211   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2212               "MESH: CREATE PATH (%u bytes long) sent!\n",
2213               size_needed);
2214   return size_needed;
2215 }
2216
2217
2218 /**
2219  * Function called to notify a client about the socket
2220  * being ready to queue more data.  "buf" will be
2221  * NULL and "size" zero if the socket was closed for
2222  * writing in the meantime.
2223  *
2224  * @param cls closure (data itself)
2225  * @param size number of bytes available in buf
2226  * @param buf where the callee should write the message
2227  *
2228  * @return number of bytes written to buf
2229  */
2230 static size_t
2231 send_core_data_multicast (void *cls, size_t size, void *buf)
2232 {
2233   struct MeshDataDescriptor *info = cls;
2234   size_t total_size;
2235
2236   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Multicast callback.\n");
2237   GNUNET_assert (NULL != info);
2238   GNUNET_assert (NULL != info->peer);
2239   total_size = info->size;
2240   GNUNET_assert (total_size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
2241
2242   if (total_size > size)
2243   {
2244     /* Retry */
2245     struct GNUNET_PeerIdentity id;
2246
2247     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2248                 "MESH: Multicast: retransmitting... (%u/%u)\n",
2249                 size, total_size);
2250     GNUNET_PEER_resolve(info->peer->id, &id);
2251     info->peer->core_transmit[info->handler_n] =
2252       GNUNET_CORE_notify_transmit_ready (core_handle,
2253                                          0,
2254                                          0,
2255                                          GNUNET_TIME_UNIT_FOREVER_REL,
2256                                          &id,
2257                                          total_size,
2258                                          &send_core_data_multicast,
2259                                          info);
2260     return 0;
2261   }
2262   info->peer->core_transmit[info->handler_n] = NULL;
2263   info->peer->infos[info->handler_n] = NULL;
2264   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  copying data...\n");
2265   memcpy (buf, info->data, total_size);
2266 #if MESH_DEBUG
2267   {
2268     struct GNUNET_MESH_Multicast *mc;
2269     struct GNUNET_MessageHeader *mh;
2270
2271     mh = buf;
2272     if (ntohs (mh->type) == GNUNET_MESSAGE_TYPE_MESH_MULTICAST)
2273     {
2274       mc = (struct GNUNET_MESH_Multicast *) mh;
2275       mh = (struct GNUNET_MessageHeader *) &mc[1];
2276       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2277                   "MESH:  multicast, payload type %u\n",
2278                   ntohs (mh->type));
2279       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2280                   "MESH:  multicast, payload size %u\n",
2281                   ntohs (mh->size));
2282     }
2283     else
2284     {
2285       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2286                   "MESH:  type %u\n",
2287                   ntohs (mh->type));
2288     }
2289   }
2290 #endif
2291   data_descriptor_decrement_multicast (info);
2292   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: freeing info...\n");
2293   GNUNET_free (info);
2294   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: return %u\n", total_size);
2295   return total_size;
2296 }
2297
2298
2299 /**
2300  * Function called to notify a client about the socket
2301  * being ready to queue more data.  "buf" will be
2302  * NULL and "size" zero if the socket was closed for
2303  * writing in the meantime.
2304  *
2305  * @param cls closure (MeshDataDescriptor)
2306  * @param size number of bytes available in buf
2307  * @param buf where the callee should write the message
2308  * @return number of bytes written to buf
2309  */
2310 static size_t
2311 send_core_path_ack (void *cls, size_t size, void *buf)
2312 {
2313   struct MeshDataDescriptor *info = cls;
2314   struct GNUNET_MESH_PathACK *msg = buf;
2315
2316   GNUNET_assert (NULL != info);
2317   if (info->peer)
2318   {
2319     info->peer->core_transmit[info->handler_n] = NULL;
2320   }
2321   if (sizeof (struct GNUNET_MESH_PathACK) > size)
2322   {
2323     GNUNET_break (0);
2324     return 0;
2325   }
2326   msg->header.size = htons (sizeof (struct GNUNET_MESH_PathACK));
2327   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_ACK);
2328   GNUNET_PEER_resolve (info->origin->oid, &msg->oid);
2329   msg->tid = htonl (info->origin->tid);
2330   msg->peer_id = my_full_id;
2331   GNUNET_free (info);
2332   /* TODO add signature */
2333
2334   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: PATH ACK sent!\n");
2335   return sizeof (struct GNUNET_MESH_PathACK);
2336 }
2337
2338
2339 /******************************************************************************/
2340 /********************      MESH NETWORK HANDLERS     **************************/
2341 /******************************************************************************/
2342
2343
2344 /**
2345  * Core handler for path creation
2346  *
2347  * @param cls closure
2348  * @param message message
2349  * @param peer peer identity this notification is about
2350  * @param atsi performance data
2351  * @param atsi_count number of records in 'atsi'
2352  *
2353  * @return GNUNET_OK to keep the connection open,
2354  *         GNUNET_SYSERR to close it (signal serious error)
2355  */
2356 static int
2357 handle_mesh_path_create (void *cls, const struct GNUNET_PeerIdentity *peer,
2358                          const struct GNUNET_MessageHeader *message,
2359                          const struct GNUNET_ATS_Information *atsi,
2360                          unsigned int atsi_count)
2361 {
2362   unsigned int own_pos;
2363   uint16_t size;
2364   uint16_t i;
2365   MESH_TunnelNumber tid;
2366   struct GNUNET_MESH_ManipulatePath *msg;
2367   struct GNUNET_PeerIdentity *pi;
2368   GNUNET_HashCode hash;
2369   struct MeshPeerPath *path;
2370   struct MeshPeerInfo *dest_peer_info;
2371   struct MeshPeerInfo *orig_peer_info;
2372   struct MeshTunnel *t;
2373
2374   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2375               "MESH: Received a path create msg [%s]\n",
2376               GNUNET_i2s(&my_full_id));
2377   size = ntohs (message->size);
2378   if (size < sizeof (struct GNUNET_MESH_ManipulatePath))
2379   {
2380     GNUNET_break_op (0);
2381     return GNUNET_OK;
2382   }
2383
2384   size -= sizeof (struct GNUNET_MESH_ManipulatePath);
2385   if (size % sizeof (struct GNUNET_PeerIdentity))
2386   {
2387     GNUNET_break_op (0);
2388     return GNUNET_OK;
2389   }
2390   size /= sizeof (struct GNUNET_PeerIdentity);
2391   if (size < 2)
2392   {
2393     GNUNET_break_op (0);
2394     return GNUNET_OK;
2395   }
2396   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2397               "MESH:     path has %u hops.\n",
2398               size);
2399   msg = (struct GNUNET_MESH_ManipulatePath *) message;
2400
2401   tid = ntohl (msg->tid);
2402   pi = (struct GNUNET_PeerIdentity *) &msg[1];
2403   t = tunnel_get (pi, tid);
2404   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2405               "MESH:     path is for tunnel %s [%X].\n",
2406               GNUNET_i2s(pi),
2407               tid);
2408   if (NULL == t)
2409   {
2410     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating tunnel\n");
2411     t = GNUNET_malloc (sizeof (struct MeshTunnel));
2412     t->id.oid = GNUNET_PEER_intern (pi);
2413     t->id.tid = tid;
2414     t->local_tid = next_local_tid++;
2415     /* FIXME test if taken */
2416     next_local_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
2417     t->tree = tree_new(t, t->id.oid);
2418
2419     GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2420     if (GNUNET_OK !=
2421         GNUNET_CONTAINER_multihashmap_put (
2422             tunnels,
2423             &hash,
2424             t,
2425             GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
2426     {
2427       tunnel_destroy (t);
2428       GNUNET_break (0);
2429       return GNUNET_OK;
2430     }
2431     tunnel_reset_timeout (t);
2432     GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2433     if (GNUNET_OK !=
2434         GNUNET_CONTAINER_multihashmap_put (
2435             incoming_tunnels,
2436             &hash,
2437             t,
2438             GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
2439     {
2440       tunnel_destroy (t);
2441       GNUNET_break (0);
2442       return GNUNET_OK;
2443     }
2444   }
2445   dest_peer_info =
2446       GNUNET_CONTAINER_multihashmap_get (peers, &pi[size - 1].hashPubKey);
2447   if (NULL == dest_peer_info)
2448   {
2449     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2450                 "MESH:   Creating PeerInfo for destination.\n");
2451     dest_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
2452     dest_peer_info->id = GNUNET_PEER_intern (&pi[size - 1]);
2453     GNUNET_CONTAINER_multihashmap_put (peers, &pi[size - 1].hashPubKey,
2454                                        dest_peer_info,
2455                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
2456   }
2457   orig_peer_info = GNUNET_CONTAINER_multihashmap_get (peers, &pi->hashPubKey);
2458   if (NULL == orig_peer_info)
2459   {
2460     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2461                 "MESH:   Creating PeerInfo for origin.\n");
2462     orig_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
2463     orig_peer_info->id = GNUNET_PEER_intern (pi);
2464     GNUNET_CONTAINER_multihashmap_put (peers, &pi->hashPubKey, orig_peer_info,
2465                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
2466   }
2467   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating path...\n");
2468   path = path_new (size);
2469   own_pos = 0;
2470   for (i = 0; i < size; i++)
2471   {
2472     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2473                 "MESH:   ... adding %s\n",
2474                 GNUNET_i2s(&pi[i]));
2475     path->peers[i] = GNUNET_PEER_intern (&pi[i]);
2476     if (path->peers[i] == myid)
2477       own_pos = i;
2478   }
2479   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2480                 "MESH:   Own position: %u\n", own_pos);
2481   if (own_pos == 0)
2482   {
2483     /* cannot be self, must be 'not found' */
2484     /* create path: self not found in path through self */
2485     GNUNET_break_op (0);
2486     path_destroy (path);
2487     /* FIXME error. destroy tunnel? leave for timeout? */
2488     return 0;
2489   }
2490   tunnel_add_path (t, path, own_pos);
2491   if (own_pos == size - 1)
2492   {
2493     /* It is for us! Send ack. */
2494     struct GNUNET_MESH_TunnelNotification cmsg;
2495     struct MeshDataDescriptor *info;
2496     unsigned int j;
2497
2498     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2499                 "MESH:   It's for us!\n");
2500     path_add_to_origin (orig_peer_info, path);
2501     t->peers = GNUNET_CONTAINER_multihashmap_create(4);
2502     GNUNET_break (GNUNET_OK == GNUNET_CONTAINER_multihashmap_put (
2503         t->peers,
2504         &my_full_id.hashPubKey,
2505         peer_info_get(&my_full_id),
2506         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST));
2507     info = GNUNET_malloc (sizeof (struct MeshDataDescriptor));
2508     info->origin = &t->id;
2509     info->peer = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
2510     GNUNET_assert (NULL != info->peer);
2511     j = peer_info_transmit_slot(info->peer);
2512     info->handler_n = j;
2513     info->peer->types[j] = GNUNET_MESSAGE_TYPE_MESH_PATH_ACK;
2514     info->peer->infos[j] = info;
2515     info->peer->core_transmit[j] =
2516         GNUNET_CORE_notify_transmit_ready (core_handle, 0, 100,
2517                                            GNUNET_TIME_UNIT_FOREVER_REL, peer,
2518                                            sizeof (struct GNUNET_MESH_PathACK),
2519                                            &send_core_path_ack, info);
2520     cmsg.header.size = htons(sizeof(cmsg));
2521     cmsg.header.type = htons(GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
2522     GNUNET_PEER_resolve(t->id.oid, &cmsg.peer);
2523     cmsg.tunnel_id = htonl(t->local_tid);
2524     GNUNET_SERVER_notification_context_broadcast(nc, &cmsg.header, GNUNET_NO);
2525   }
2526   else
2527   {
2528     struct MeshPeerPath *path2;
2529
2530     /* It's for somebody else! Retransmit. */
2531     path2 = path_duplicate(path);
2532     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2533                 "MESH:   Retransmitting.\n");
2534     path_add_to_peer(dest_peer_info, path);
2535     path = path_duplicate(path2);
2536     path_add_to_origin(orig_peer_info, path2);
2537     send_create_path(dest_peer_info, path, t);
2538   }
2539   return GNUNET_OK;
2540 }
2541
2542
2543 /**
2544  * Core handler for path destruction
2545  *
2546  * @param cls closure
2547  * @param message message
2548  * @param peer peer identity this notification is about
2549  * @param atsi performance data
2550  * @param atsi_count number of records in 'atsi'
2551  *
2552  * @return GNUNET_OK to keep the connection open,
2553  *         GNUNET_SYSERR to close it (signal serious error)
2554  */
2555 static int
2556 handle_mesh_path_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
2557                           const struct GNUNET_MessageHeader *message,
2558                           const struct GNUNET_ATS_Information *atsi,
2559                           unsigned int atsi_count)
2560 {
2561   struct GNUNET_MESH_ManipulatePath *msg;
2562   struct GNUNET_PeerIdentity *pi;
2563   struct MeshPeerPath *path;
2564   struct MeshTunnel *t;
2565   unsigned int own_pos;
2566   unsigned int i;
2567   size_t size;
2568
2569   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2570               "MESH: Received a PATH DESTROY msg from %s\n",
2571               GNUNET_i2s(peer));
2572   size = ntohs (message->size);
2573   if (size < sizeof (struct GNUNET_MESH_ManipulatePath))
2574   {
2575     GNUNET_break_op (0);
2576     return GNUNET_OK;
2577   }
2578
2579   size -= sizeof (struct GNUNET_MESH_ManipulatePath);
2580   if (size % sizeof (struct GNUNET_PeerIdentity))
2581   {
2582     GNUNET_break_op (0);
2583     return GNUNET_OK;
2584   }
2585   size /= sizeof (struct GNUNET_PeerIdentity);
2586   if (size < 2)
2587   {
2588     GNUNET_break_op (0);
2589     return GNUNET_OK;
2590   }
2591   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2592               "MESH:     path has %u hops.\n",
2593               size);
2594
2595   msg = (struct GNUNET_MESH_ManipulatePath *) message;
2596   pi = (struct GNUNET_PeerIdentity *) &msg[1];
2597   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2598               "MESH:     path is for tunnel %s [%X].\n",
2599               GNUNET_i2s(pi),
2600               msg->tid);
2601   t = tunnel_get (pi, ntohl (msg->tid));
2602   if (NULL == t)
2603   {
2604     /* TODO notify back: we don't know this tunnel */
2605     GNUNET_break_op (0);
2606     return GNUNET_OK;
2607   }
2608   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating path...\n");
2609   path = path_new (size);
2610   own_pos = 0;
2611   for (i = 0; i < size; i++)
2612   {
2613     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2614                 "MESH:   ... adding %s\n",
2615                 GNUNET_i2s(&pi[i]));
2616     path->peers[i] = GNUNET_PEER_intern (&pi[i]);
2617     if (path->peers[i] == myid)
2618       own_pos = i;
2619   }
2620   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2621                 "MESH:   Own position: %u\n", own_pos);
2622   if (own_pos < path->length - 1)
2623     send_message (message, &pi[own_pos + 1]);
2624   tunnel_delete_peer (t, path->peers[path->length - 1]);
2625   path_destroy (path);
2626   return GNUNET_OK;
2627 }
2628
2629
2630 /**
2631  * Core handler for notifications of broken paths
2632  *
2633  * @param cls closure
2634  * @param message message
2635  * @param peer peer identity this notification is about
2636  * @param atsi performance data
2637  * @param atsi_count number of records in 'atsi'
2638  *
2639  * @return GNUNET_OK to keep the connection open,
2640  *         GNUNET_SYSERR to close it (signal serious error)
2641  */
2642 static int
2643 handle_mesh_path_broken (void *cls, const struct GNUNET_PeerIdentity *peer,
2644                           const struct GNUNET_MessageHeader *message,
2645                           const struct GNUNET_ATS_Information *atsi,
2646                           unsigned int atsi_count)
2647 {
2648   struct GNUNET_MESH_PathBroken *msg;
2649   struct MeshTunnel *t;
2650
2651   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2652               "MESH: Received a PATH BROKEN msg from %s\n",
2653               GNUNET_i2s(peer));
2654   msg = (struct GNUNET_MESH_PathBroken *) message;
2655   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2656               "MESH:   regarding %s\n",
2657               GNUNET_i2s(&msg->peer1));
2658   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2659               "MESH:   regarding %s\n",
2660               GNUNET_i2s(&msg->peer2));
2661   t = tunnel_get(&msg->oid, ntohl (msg->tid));
2662   if (NULL == t)
2663   {
2664     GNUNET_break_op (0);
2665     return GNUNET_OK;
2666   }
2667   tunnel_notify_connection_broken(t,
2668                                   GNUNET_PEER_search(&msg->peer1),
2669                                   GNUNET_PEER_search(&msg->peer2));
2670   return GNUNET_OK;
2671
2672 }
2673
2674
2675 /**
2676  * Core handler for tunnel destruction
2677  *
2678  * @param cls closure
2679  * @param message message
2680  * @param peer peer identity this notification is about
2681  * @param atsi performance data
2682  * @param atsi_count number of records in 'atsi'
2683  *
2684  * @return GNUNET_OK to keep the connection open,
2685  *         GNUNET_SYSERR to close it (signal serious error)
2686  */
2687 static int
2688 handle_mesh_tunnel_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
2689                             const struct GNUNET_MessageHeader *message,
2690                             const struct GNUNET_ATS_Information *atsi,
2691                             unsigned int atsi_count)
2692 {
2693   struct GNUNET_MESH_TunnelDestroy *msg;
2694   struct MeshTunnel *t;
2695
2696   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2697               "MESH: Got a TUNNEL DESTROY packet from %s\n",
2698               GNUNET_i2s (peer));
2699   msg = (struct GNUNET_MESH_TunnelDestroy *) message;
2700   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2701               "MESH:   for tunnel %s [%u]\n",
2702               GNUNET_i2s (&msg->oid),
2703               ntohl (msg->tid));
2704   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2705   if (NULL == t)
2706   {
2707     /* Probably already got the message from another path,
2708      * destroyed the tunnel and retransmitted to children.
2709      * Safe to ignore.
2710      */
2711     return GNUNET_OK;
2712   }
2713   if (t->id.oid == myid)
2714   {
2715     GNUNET_break_op (0);
2716     return GNUNET_OK;
2717   }
2718   if (t->local_tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
2719   {
2720     /* Tunnel was incoming, notify clients */
2721     send_clients_tunnel_destroy (t);
2722   }
2723   tunnel_send_destroy (t);
2724   tunnel_destroy (t);
2725   return GNUNET_OK;
2726 }
2727
2728
2729 /**
2730  * Core handler for mesh network traffic going from the origin to a peer
2731  *
2732  * @param cls closure
2733  * @param peer peer identity this notification is about
2734  * @param message message
2735  * @param atsi performance data
2736  * @param atsi_count number of records in 'atsi'
2737  * @return GNUNET_OK to keep the connection open,
2738  *         GNUNET_SYSERR to close it (signal serious error)
2739  */
2740 static int
2741 handle_mesh_data_unicast (void *cls, const struct GNUNET_PeerIdentity *peer,
2742                           const struct GNUNET_MessageHeader *message,
2743                           const struct GNUNET_ATS_Information *atsi,
2744                           unsigned int atsi_count)
2745 {
2746   struct GNUNET_MESH_Unicast *msg;
2747   struct MeshTunnel *t;
2748   GNUNET_PEER_Id pid;
2749   size_t size;
2750
2751   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2752               "MESH: got a unicast packet from %s\n",
2753               GNUNET_i2s (peer));
2754   size = ntohs (message->size);
2755   if (size <
2756       sizeof (struct GNUNET_MESH_Unicast) +
2757       sizeof (struct GNUNET_MessageHeader))
2758   {
2759     GNUNET_break (0);
2760     return GNUNET_OK;
2761   }
2762   msg = (struct GNUNET_MESH_Unicast *) message;
2763   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2764               "MESH:  of type %u\n",
2765               ntohs (msg[1].header.type));
2766   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2767   if (NULL == t)
2768   {
2769     /* TODO notify back: we don't know this tunnel */
2770     GNUNET_break_op (0);
2771     return GNUNET_OK;
2772   }
2773   tunnel_reset_timeout (t);
2774   pid = GNUNET_PEER_search(&msg->destination);
2775   if (pid == myid)
2776   {
2777     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2778                 "MESH:   it's for us! sending to clients...\n");
2779     send_subscribed_clients (message, (struct GNUNET_MessageHeader *) &msg[1]);
2780     return GNUNET_OK;
2781   }
2782   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2783               "MESH:   not for us, retransmitting...\n");
2784   send_message (message, path_get_first_hop(t->tree, pid));
2785   return GNUNET_OK;
2786 }
2787
2788
2789 /**
2790  * Core handler for mesh network traffic going from the origin to all peers
2791  *
2792  * @param cls closure
2793  * @param message message
2794  * @param peer peer identity this notification is about
2795  * @param atsi performance data
2796  * @param atsi_count number of records in 'atsi'
2797  * @return GNUNET_OK to keep the connection open,
2798  *         GNUNET_SYSERR to close it (signal serious error)
2799  *
2800  * TODO: Check who we got this from, to validate route.
2801  */
2802 static int
2803 handle_mesh_data_multicast (void *cls, const struct GNUNET_PeerIdentity *peer,
2804                             const struct GNUNET_MessageHeader *message,
2805                             const struct GNUNET_ATS_Information *atsi,
2806                             unsigned int atsi_count)
2807 {
2808   struct GNUNET_MESH_Multicast *msg;
2809   struct MeshTunnel *t;
2810   size_t size;
2811
2812   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2813               "MESH: got a multicast packet from %s\n",
2814               GNUNET_i2s (peer));
2815   size = ntohs (message->size);
2816   if (sizeof (struct GNUNET_MESH_Multicast) +
2817       sizeof (struct GNUNET_MessageHeader) > size)
2818   {
2819     GNUNET_break_op (0);
2820     return GNUNET_OK;
2821   }
2822   msg = (struct GNUNET_MESH_Multicast *) message;
2823   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2824
2825   if (NULL == t)
2826   {
2827     /* TODO notify that we dont know that tunnel */
2828     GNUNET_break_op (0);
2829     return GNUNET_OK;
2830   }
2831   tunnel_reset_timeout (t);
2832
2833   /* Transmit to locally interested clients */
2834   if (NULL != t->peers &&
2835       GNUNET_CONTAINER_multihashmap_contains (t->peers, &my_full_id.hashPubKey))
2836   {
2837     send_subscribed_clients (message, &msg[1].header);
2838   }
2839   tunnel_send_multicast(t, message);
2840   return GNUNET_OK;
2841 }
2842
2843
2844 /**
2845  * Core handler for mesh network traffic toward the owner of a tunnel
2846  *
2847  * @param cls closure
2848  * @param message message
2849  * @param peer peer identity this notification is about
2850  * @param atsi performance data
2851  * @param atsi_count number of records in 'atsi'
2852  *
2853  * @return GNUNET_OK to keep the connection open,
2854  *         GNUNET_SYSERR to close it (signal serious error)
2855  */
2856 static int
2857 handle_mesh_data_to_orig (void *cls, const struct GNUNET_PeerIdentity *peer,
2858                           const struct GNUNET_MessageHeader *message,
2859                           const struct GNUNET_ATS_Information *atsi,
2860                           unsigned int atsi_count)
2861 {
2862   struct GNUNET_MESH_ToOrigin *msg;
2863   struct GNUNET_PeerIdentity id;
2864   struct MeshPeerInfo *peer_info;
2865   struct MeshTunnel *t;
2866   size_t size;
2867
2868   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2869               "MESH: got a ToOrigin packet from %s\n",
2870               GNUNET_i2s (peer));
2871   size = ntohs (message->size);
2872   if (size < sizeof (struct GNUNET_MESH_ToOrigin) +     /* Payload must be */
2873       sizeof (struct GNUNET_MessageHeader))     /* at least a header */
2874   {
2875     GNUNET_break_op (0);
2876     return GNUNET_OK;
2877   }
2878   msg = (struct GNUNET_MESH_ToOrigin *) message;
2879   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2880               "MESH:  of type %u\n",
2881               ntohs (msg[1].header.type));
2882   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2883
2884   if (NULL == t)
2885   {
2886     /* TODO notify that we dont know this tunnel (whom)? */
2887     GNUNET_break_op (0);
2888     return GNUNET_OK;
2889   }
2890
2891   if (t->id.oid == myid)
2892   {
2893     char cbuf[size];
2894     struct GNUNET_MESH_ToOrigin *copy;
2895
2896     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2897                 "MESH:   it's for us! sending to clients...\n");
2898     if (NULL == t->client)
2899     {
2900       /* got data packet for ownerless tunnel */
2901       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2902                 "MESH:   no clients!\n");
2903       GNUNET_break_op (0);
2904       return GNUNET_OK;
2905     }
2906     /* TODO signature verification */
2907     memcpy (cbuf, message, size);
2908     copy = (struct GNUNET_MESH_ToOrigin *) cbuf;
2909     copy->tid = htonl (t->local_tid);
2910     GNUNET_SERVER_notification_context_unicast (nc,
2911                                                 t->client->handle,
2912                                                 &copy->header,
2913                                                 GNUNET_YES);
2914     return GNUNET_OK;
2915   }
2916   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2917               "MESH:   not for us, retransmitting...\n");
2918
2919   peer_info = peer_info_get (&msg->oid);
2920   if (NULL == peer_info)
2921   {
2922     /* unknown origin of tunnel */
2923     GNUNET_break (0);
2924     return GNUNET_OK;
2925   }
2926   GNUNET_PEER_resolve (t->tree->me->parent->peer, &id);
2927   send_message (message, &id);
2928
2929   return GNUNET_OK;
2930 }
2931
2932
2933 /**
2934  * Core handler for path ACKs
2935  *
2936  * @param cls closure
2937  * @param message message
2938  * @param peer peer identity this notification is about
2939  * @param atsi performance data
2940  * @param atsi_count number of records in 'atsi'
2941  *
2942  * @return GNUNET_OK to keep the connection open,
2943  *         GNUNET_SYSERR to close it (signal serious error)
2944  */
2945 static int
2946 handle_mesh_path_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
2947                       const struct GNUNET_MessageHeader *message,
2948                       const struct GNUNET_ATS_Information *atsi,
2949                       unsigned int atsi_count)
2950 {
2951   struct GNUNET_MESH_PathACK *msg;
2952   struct GNUNET_PeerIdentity id;
2953   struct MeshTunnelTreeNode *n;
2954   struct MeshPeerInfo *peer_info;
2955   struct MeshTunnel *t;
2956
2957   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2958               "MESH: Received a path ACK msg [%s]\n",
2959               GNUNET_i2s(&my_full_id));
2960   msg = (struct GNUNET_MESH_PathACK *) message;
2961   t = tunnel_get (&msg->oid, msg->tid);
2962   if (NULL == t)
2963   {
2964     /* TODO notify that we don't know the tunnel */
2965     return GNUNET_OK;
2966   }
2967
2968   /* Message for us? */
2969   if (0 == memcmp (&msg->oid, &my_full_id, sizeof (struct GNUNET_PeerIdentity)))
2970   {
2971     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   It's for us!\n");
2972     if (NULL == t->client)
2973     {
2974       GNUNET_break_op (0);
2975       return GNUNET_OK;
2976     }
2977     if (NULL != t->dht_get_type)
2978     {
2979       GNUNET_DHT_get_stop (t->dht_get_type);
2980       t->dht_get_type = NULL;
2981     }
2982     peer_info = peer_info_get (&msg->peer_id);
2983     n = tree_find_peer(t->tree->root, peer_info->id);
2984     if (NULL == n)
2985     {
2986       GNUNET_break_op (0);
2987       return GNUNET_OK;
2988     }
2989     n->status = MESH_PEER_READY;
2990     send_client_peer_connected(t, peer_info->id);
2991     return GNUNET_OK;
2992   }
2993
2994   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2995               "MESH:   not for us, retransmitting...\n");
2996   GNUNET_PEER_resolve(t->tree->me->parent->peer, &id);
2997   peer_info = peer_info_get (&msg->oid);
2998   if (NULL == peer_info)
2999   {
3000     /* If we know the tunnel, we should DEFINITELY know the peer */
3001     GNUNET_break (0);
3002     return GNUNET_OK;
3003   }
3004   send_message (message, &id);
3005   return GNUNET_OK;
3006 }
3007
3008
3009 /**
3010  * Functions to handle messages from core
3011  */
3012 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
3013   {&handle_mesh_path_create, GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE, 0},
3014   {&handle_mesh_path_destroy, GNUNET_MESSAGE_TYPE_MESH_PATH_DESTROY, 0},
3015   {&handle_mesh_path_broken, GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN,
3016     sizeof (struct GNUNET_MESH_PathBroken)},
3017   {&handle_mesh_tunnel_destroy, GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY, 0},
3018   {&handle_mesh_data_unicast, GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
3019   {&handle_mesh_data_multicast, GNUNET_MESSAGE_TYPE_MESH_MULTICAST, 0},
3020   {&handle_mesh_data_to_orig, GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
3021   {&handle_mesh_path_ack, GNUNET_MESSAGE_TYPE_MESH_PATH_ACK,
3022     sizeof (struct GNUNET_MESH_PathACK)},
3023   {NULL, 0, 0}
3024 };
3025
3026
3027
3028 /******************************************************************************/
3029 /****************       MESH LOCAL HANDLER HELPERS      ***********************/
3030 /******************************************************************************/
3031
3032 /**
3033  * deregister_app: iterator for removing each application registered by a client
3034  *
3035  * @param cls closure
3036  * @param key the hash of the application id (used to access the hashmap)
3037  * @param value the value stored at the key (client)
3038  *
3039  * @return GNUNET_OK on success
3040  */
3041 static int
3042 deregister_app (void *cls, const GNUNET_HashCode * key, void *value)
3043 {
3044   GNUNET_break (GNUNET_YES ==
3045       GNUNET_CONTAINER_multihashmap_remove (applications, key, value));
3046   return GNUNET_OK;
3047 }
3048
3049 #if LATER
3050 /**
3051  * notify_client_connection_failure: notify a client that the connection to the
3052  * requested remote peer is not possible (for instance, no route found)
3053  * Function called when the socket is ready to queue more data. "buf" will be
3054  * NULL and "size" zero if the socket was closed for writing in the meantime.
3055  *
3056  * @param cls closure
3057  * @param size number of bytes available in buf
3058  * @param buf where the callee should write the message
3059  * @return number of bytes written to buf
3060  */
3061 static size_t
3062 notify_client_connection_failure (void *cls, size_t size, void *buf)
3063 {
3064   int size_needed;
3065   struct MeshPeerInfo *peer_info;
3066   struct GNUNET_MESH_PeerControl *msg;
3067   struct GNUNET_PeerIdentity id;
3068
3069   if (0 == size && NULL == buf)
3070   {
3071     // TODO retry? cancel?
3072     return 0;
3073   }
3074
3075   size_needed = sizeof (struct GNUNET_MESH_PeerControl);
3076   peer_info = (struct MeshPeerInfo *) cls;
3077   msg = (struct GNUNET_MESH_PeerControl *) buf;
3078   msg->header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
3079   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DISCONNECTED);
3080 //     msg->tunnel_id = htonl(peer_info->t->tid);
3081   GNUNET_PEER_resolve (peer_info->id, &id);
3082   memcpy (&msg->peer, &id, sizeof (struct GNUNET_PeerIdentity));
3083
3084   return size_needed;
3085 }
3086 #endif
3087
3088
3089 /**
3090  * Send keepalive packets for a peer
3091  *
3092  * @param cls Closure (tunnel for which to send the keepalive).
3093  * @param tc Notification context.
3094  *
3095  * TODO: implement explicit multicast keepalive?
3096  */
3097 static void
3098 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
3099 {
3100   struct MeshTunnel *t = cls;
3101   struct GNUNET_MessageHeader *payload;
3102   struct GNUNET_MESH_Multicast *msg;
3103   size_t size = sizeof(struct GNUNET_MESH_Multicast) +
3104                 sizeof(struct GNUNET_MessageHeader);
3105   char cbuf[size];
3106
3107   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
3108   {
3109     return;
3110   }
3111   t->path_refresh_task = GNUNET_SCHEDULER_NO_TASK;
3112
3113   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3114               "MESH: sending keepalive for tunnel %d\n",
3115               t->id.tid);
3116
3117   msg = (struct GNUNET_MESH_Multicast *) cbuf;
3118   msg->header.size = htons (size);
3119   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
3120   msg->oid = my_full_id;
3121   msg->tid = htonl(t->id.tid);
3122   payload = (struct GNUNET_MessageHeader *) &msg[1];
3123   payload->size = htons (sizeof(struct GNUNET_MessageHeader));
3124   payload->type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE);
3125   tunnel_send_multicast (t, &msg->header);
3126
3127   t->path_refresh_task =
3128       GNUNET_SCHEDULER_add_delayed (t->tree->refresh, &path_refresh, t);
3129   return;
3130 }
3131
3132
3133 /**
3134  * Function to process paths received for a new peer addition. The recorded
3135  * paths form the initial tunnel, which can be optimized later.
3136  * Called on each result obtained for the DHT search.
3137  *
3138  * @param cls closure
3139  * @param exp when will this value expire
3140  * @param key key of the result
3141  * @param type type of the result
3142  * @param size number of bytes in data
3143  * @param data pointer to the result data
3144  *
3145  * TODO: re-issue the request after certain time? cancel after X results?
3146  */
3147 static void
3148 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
3149                     const GNUNET_HashCode * key,
3150                     const struct GNUNET_PeerIdentity *get_path,
3151                     unsigned int get_path_length,
3152                     const struct GNUNET_PeerIdentity *put_path,
3153                     unsigned int put_path_length,
3154                     enum GNUNET_BLOCK_Type type, size_t size, const void *data)
3155 {
3156   struct MeshPathInfo *path_info = cls;
3157   struct MeshPeerPath *p;
3158   struct GNUNET_PeerIdentity pi;
3159   int i;
3160
3161   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
3162              "MESH: Got results from DHT!\n");
3163   GNUNET_PEER_resolve (path_info->peer->id, &pi);
3164   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
3165              "MESH:   for %s\n",
3166              GNUNET_i2s(&pi));
3167 //   GNUNET_DHT_get_stop(path_info->peer->dhtget);
3168 //   path_info->peer->dhtget = NULL;
3169
3170   p = path_build_from_dht (get_path, get_path_length,
3171                            put_path, put_path_length);
3172   path_add_to_peer (path_info->peer, p);
3173   for (i = 0; i < path_info->peer->ntunnels; i++)
3174   {
3175     tunnel_add_peer (path_info->peer->tunnels[i], path_info->peer);
3176     peer_info_connect(path_info->peer, path_info->t);
3177   }
3178 //   GNUNET_free (path_info);
3179
3180   return;
3181 }
3182
3183
3184 /**
3185  * Function to process paths received for a new peer addition. The recorded
3186  * paths form the initial tunnel, which can be optimized later.
3187  * Called on each result obtained for the DHT search.
3188  *
3189  * @param cls closure
3190  * @param exp when will this value expire
3191  * @param key key of the result
3192  * @param type type of the result
3193  * @param size number of bytes in data
3194  * @param data pointer to the result data
3195  */
3196 static void
3197 dht_get_type_handler (void *cls, struct GNUNET_TIME_Absolute exp,
3198                       const GNUNET_HashCode * key,
3199                       const struct GNUNET_PeerIdentity *get_path,
3200                       unsigned int get_path_length,
3201                       const struct GNUNET_PeerIdentity *put_path,
3202                       unsigned int put_path_length,
3203                       enum GNUNET_BLOCK_Type type, size_t size,
3204                       const void *data)
3205 {
3206   const struct GNUNET_PeerIdentity *pi = data;
3207   struct MeshTunnel *t = cls;
3208   struct MeshPeerInfo *peer_info;
3209   struct MeshPeerPath *p;
3210
3211   if (size != sizeof (struct GNUNET_PeerIdentity))
3212   {
3213     GNUNET_break_op (0);
3214     return;
3215   }
3216   GNUNET_assert (NULL != t->client);
3217   peer_info = peer_info_get (pi);
3218   (void) GNUNET_CONTAINER_multihashmap_put (
3219       t->peers,
3220       &pi->hashPubKey,
3221       peer_info,
3222       GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE);
3223
3224   p = path_build_from_dht (get_path, get_path_length,
3225                            put_path, put_path_length);
3226   path_add_to_peer (peer_info, p);
3227   tunnel_add_peer(t, peer_info);
3228   peer_info_connect (peer_info, t);
3229 }
3230
3231
3232 /******************************************************************************/
3233 /*********************       MESH LOCAL HANDLES      **************************/
3234 /******************************************************************************/
3235
3236
3237 /**
3238  * Handler for client disconnection
3239  *
3240  * @param cls closure
3241  * @param client identification of the client; NULL
3242  *        for the last call when the server is destroyed
3243  */
3244 static void
3245 handle_local_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
3246 {
3247   struct MeshClient *c;
3248   struct MeshClient *next;
3249
3250   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: client disconnected\n");
3251   if (client == NULL)
3252   {
3253     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    (SERVER DOWN)\n");
3254     return;
3255   }
3256   c = clients;
3257   while (NULL != c)
3258   {
3259     if (c->handle != client && NULL != client)
3260     {
3261       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    ... searching\n");
3262       c = c->next;
3263       continue;
3264     }
3265     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: matching client found\n");
3266     GNUNET_SERVER_client_drop (c->handle);
3267     if (NULL != c->tunnels)
3268     {
3269       GNUNET_CONTAINER_multihashmap_iterate (c->tunnels,
3270                                              &tunnel_destroy_iterator,
3271                                              c);
3272       GNUNET_CONTAINER_multihashmap_destroy (c->tunnels);
3273     }
3274
3275     /* deregister clients applications */
3276     if (NULL != c->apps)
3277     {
3278       GNUNET_CONTAINER_multihashmap_iterate (c->apps, &deregister_app, NULL);
3279       GNUNET_CONTAINER_multihashmap_destroy (c->apps);
3280     }
3281     if (0 == GNUNET_CONTAINER_multihashmap_size (applications) &&
3282         GNUNET_SCHEDULER_NO_TASK != announce_applications_task)
3283     {
3284       GNUNET_SCHEDULER_cancel (announce_applications_task);
3285       announce_applications_task = GNUNET_SCHEDULER_NO_TASK;
3286     }
3287     if (NULL != c->types)
3288       GNUNET_CONTAINER_multihashmap_destroy (c->types);
3289     next = c->next;
3290     GNUNET_CONTAINER_DLL_remove (clients, clients_tail, c);
3291     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   CLIENT FREE at %p\n", c);
3292     GNUNET_free (c);
3293     c = next;
3294   }
3295
3296   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    done!\n");
3297   return;
3298 }
3299
3300
3301 /**
3302  * Handler for new clients
3303  *
3304  * @param cls closure
3305  * @param client identification of the client
3306  * @param message the actual message, which includes messages the client wants
3307  */
3308 static void
3309 handle_local_new_client (void *cls, struct GNUNET_SERVER_Client *client,
3310                          const struct GNUNET_MessageHeader *message)
3311 {
3312   struct GNUNET_MESH_ClientConnect *cc_msg;
3313   struct MeshClient *c;
3314   GNUNET_MESH_ApplicationType *a;
3315   unsigned int size;
3316   uint16_t ntypes;
3317   uint16_t *t;
3318   uint16_t napps;
3319   uint16_t i;
3320
3321   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new client connected\n");
3322   /* Check data sanity */
3323   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
3324   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
3325   ntypes = ntohs (cc_msg->types);
3326   napps = ntohs (cc_msg->applications);
3327   if (size !=
3328       ntypes * sizeof (uint16_t) + napps * sizeof (GNUNET_MESH_ApplicationType))
3329   {
3330     GNUNET_break (0);
3331     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3332     return;
3333   }
3334
3335   /* Create new client structure */
3336   c = GNUNET_malloc (sizeof (struct MeshClient));
3337 #if MESH_DEBUG
3338   c->id = next_client_id++;
3339 #endif
3340   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   CLIENT NEW %u at %p\n", c->id, c);
3341   c->handle = client;
3342   GNUNET_SERVER_client_keep (client);
3343   a = (GNUNET_MESH_ApplicationType *) &cc_msg[1];
3344   if (napps > 0)
3345   {
3346     GNUNET_MESH_ApplicationType at;
3347     GNUNET_HashCode hc;
3348
3349     c->apps = GNUNET_CONTAINER_multihashmap_create (napps);
3350     for (i = 0; i < napps; i++)
3351     {
3352       at = ntohl (a[i]);
3353       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   app type: %u\n", at);
3354       GNUNET_CRYPTO_hash (&at, sizeof (at), &hc);
3355       /* store in clients hashmap */
3356       GNUNET_CONTAINER_multihashmap_put (c->apps, &hc, c,
3357                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3358       /* store in global hashmap, for announcements */
3359       GNUNET_CONTAINER_multihashmap_put (applications, &hc, c,
3360                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3361     }
3362     if (GNUNET_SCHEDULER_NO_TASK == announce_applications_task)
3363       announce_applications_task =
3364           GNUNET_SCHEDULER_add_now (&announce_applications, NULL);
3365
3366   }
3367   if (ntypes > 0)
3368   {
3369     uint16_t u16;
3370     GNUNET_HashCode hc;
3371
3372     t = (uint16_t *) & a[napps];
3373     c->types = GNUNET_CONTAINER_multihashmap_create (ntypes);
3374     for (i = 0; i < ntypes; i++)
3375     {
3376       u16 = ntohs (t[i]);
3377       GNUNET_CRYPTO_hash (&u16, sizeof (u16), &hc);
3378
3379       /* store in clients hashmap */
3380       GNUNET_CONTAINER_multihashmap_put (c->types, &hc, c,
3381                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3382       /* store in global hashmap */
3383       GNUNET_CONTAINER_multihashmap_put (types, &hc, c,
3384                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3385     }
3386   }
3387   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3388               "MESH:  client has %u+%u subscriptions\n", napps, ntypes);
3389
3390   GNUNET_CONTAINER_DLL_insert (clients, clients_tail, c);
3391   c->tunnels = GNUNET_CONTAINER_multihashmap_create (32);
3392   GNUNET_SERVER_notification_context_add (nc, client);
3393
3394   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3395 #if MESH_DEBUG
3396   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new client processed\n");
3397 #endif
3398 }
3399
3400
3401 /**
3402  * Handler for requests of new tunnels
3403  *
3404  * @param cls closure
3405  * @param client identification of the client
3406  * @param message the actual message
3407  */
3408 static void
3409 handle_local_tunnel_create (void *cls, struct GNUNET_SERVER_Client *client,
3410                             const struct GNUNET_MessageHeader *message)
3411 {
3412   struct GNUNET_MESH_TunnelMessage *t_msg;
3413   struct MeshTunnel *t;
3414   struct MeshClient *c;
3415   GNUNET_HashCode hash;
3416
3417   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new tunnel requested\n");
3418
3419   /* Sanity check for client registration */
3420   if (NULL == (c = client_get (client)))
3421   {
3422     GNUNET_break (0);
3423     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3424     return;
3425   }
3426 #if MESH_DEBUG
3427   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
3428 #endif
3429
3430   /* Message sanity check */
3431   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
3432   {
3433     GNUNET_break (0);
3434     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3435     return;
3436   }
3437
3438   t_msg = (struct GNUNET_MESH_TunnelMessage *) message;
3439   /* Sanity check for tunnel numbering */
3440   if (0 == (ntohl (t_msg->tunnel_id) & GNUNET_MESH_LOCAL_TUNNEL_ID_CLI))
3441   {
3442     GNUNET_break (0);
3443     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3444     return;
3445   }
3446   /* Sanity check for duplicate tunnel IDs */
3447   if (NULL != tunnel_get_by_local_id (c, ntohl (t_msg->tunnel_id)))
3448   {
3449     GNUNET_break (0);
3450     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3451     return;
3452   }
3453
3454   t = GNUNET_malloc (sizeof (struct MeshTunnel));
3455   while (NULL != tunnel_get_by_pi (myid, next_tid))
3456     next_tid = (next_tid + 1) & ~GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
3457   t->id.tid = next_tid++;
3458   t->id.oid = myid;
3459   t->local_tid = ntohl (t_msg->tunnel_id);
3460 #if MESH_DEBUG
3461   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3462               "MESH: CREATED TUNNEL %s [%x] (%x)\n",
3463               GNUNET_i2s (&my_full_id),
3464               t->id.tid,
3465               t->local_tid);
3466 #endif
3467   t->client = c;
3468   t->peers = GNUNET_CONTAINER_multihashmap_create (32);
3469
3470   GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
3471   if (GNUNET_OK !=
3472       GNUNET_CONTAINER_multihashmap_put (
3473         c->tunnels, &hash, t,
3474         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
3475   {
3476     GNUNET_break (0);
3477     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3478     return;
3479   }
3480
3481   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
3482   if (GNUNET_OK !=
3483       GNUNET_CONTAINER_multihashmap_put (
3484         tunnels, &hash, t,
3485         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
3486   {
3487     GNUNET_break (0);
3488     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3489     return;
3490   }
3491   t->tree = tree_new (t, myid);
3492   t->tree->refresh = REFRESH_PATH_TIME;
3493   t->tree->root->status = MESH_PEER_READY;
3494   t->tree->me = t->tree->root;
3495
3496   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3497   return;
3498 }
3499
3500
3501 /**
3502  * Handler for requests of deleting tunnels
3503  *
3504  * @param cls closure
3505  * @param client identification of the client
3506  * @param message the actual message
3507  */
3508 static void
3509 handle_local_tunnel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
3510                              const struct GNUNET_MessageHeader *message)
3511 {
3512   struct GNUNET_MESH_TunnelMessage *tunnel_msg;
3513   struct MeshClient *c;
3514   struct MeshTunnel *t;
3515   MESH_TunnelNumber tid;
3516   GNUNET_HashCode hash;
3517
3518   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3519               "MESH: Got a DESTROY TUNNEL from client!\n");
3520
3521   /* Sanity check for client registration */
3522   if (NULL == (c = client_get (client)))
3523   {
3524     GNUNET_break (0);
3525     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3526     return;
3527   }
3528   /* Message sanity check */
3529   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
3530   {
3531     GNUNET_break (0);
3532     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3533     return;
3534   }
3535 #if MESH_DEBUG
3536   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
3537 #endif
3538   tunnel_msg = (struct GNUNET_MESH_TunnelMessage *) message;
3539
3540   /* Retrieve tunnel */
3541   tid = ntohl (tunnel_msg->tunnel_id);
3542
3543   /* Remove from local id hashmap */
3544   GNUNET_CRYPTO_hash (&tid, sizeof (MESH_TunnelNumber), &hash);
3545   t = GNUNET_CONTAINER_multihashmap_get (c->tunnels, &hash);
3546   GNUNET_CONTAINER_multihashmap_remove (c->tunnels, &hash, t);
3547
3548   t->client = NULL;
3549   tunnel_send_destroy (t);
3550   tunnel_destroy(t);
3551   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3552   return;
3553 }
3554
3555
3556 /**
3557  * Handler for connection requests to new peers
3558  *
3559  * @param cls closure
3560  * @param client identification of the client
3561  * @param message the actual message (PeerControl)
3562  */
3563 static void
3564 handle_local_connect_add (void *cls, struct GNUNET_SERVER_Client *client,
3565                           const struct GNUNET_MessageHeader *message)
3566 {
3567   struct GNUNET_MESH_PeerControl *peer_msg;
3568   struct MeshPeerInfo *peer_info;
3569   struct MeshClient *c;
3570   struct MeshTunnel *t;
3571   MESH_TunnelNumber tid;
3572
3573   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH: Got connection request\n");
3574   /* Sanity check for client registration */
3575   if (NULL == (c = client_get (client)))
3576   {
3577     GNUNET_break (0);
3578     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3579     return;
3580   }
3581
3582   peer_msg = (struct GNUNET_MESH_PeerControl *) message;
3583   /* Sanity check for message size */
3584   if (sizeof (struct GNUNET_MESH_PeerControl) != ntohs (peer_msg->header.size))
3585   {
3586     GNUNET_break (0);
3587     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3588     return;
3589   }
3590
3591   /* Tunnel exists? */
3592   tid = ntohl (peer_msg->tunnel_id);
3593   t = tunnel_get_by_local_id (c, tid);
3594   if (NULL == t)
3595   {
3596     GNUNET_break (0);
3597     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3598     return;
3599   }
3600
3601   /* Does client own tunnel? */
3602   if (t->client->handle != client)
3603   {
3604     GNUNET_break (0);
3605     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3606     return;
3607   }
3608   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:      for %s\n",
3609              GNUNET_i2s(&peer_msg->peer));
3610   peer_info = peer_info_get (&peer_msg->peer);
3611
3612   tunnel_add_peer(t, peer_info);
3613   peer_info_connect(peer_info, t);
3614
3615   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3616   return;
3617 }
3618
3619
3620 /**
3621  * Handler for disconnection requests of peers in a tunnel
3622  *
3623  * @param cls closure
3624  * @param client identification of the client
3625  * @param message the actual message (PeerControl)
3626  */
3627 static void
3628 handle_local_connect_del (void *cls, struct GNUNET_SERVER_Client *client,
3629                           const struct GNUNET_MessageHeader *message)
3630 {
3631   struct GNUNET_MESH_PeerControl *peer_msg;
3632   struct MeshPeerInfo *peer_info;
3633   struct MeshClient *c;
3634   struct MeshTunnel *t;
3635   MESH_TunnelNumber tid;
3636
3637   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Got a PEER DEL request\n");
3638   /* Sanity check for client registration */
3639   if (NULL == (c = client_get (client)))
3640   {
3641     GNUNET_break (0);
3642     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3643     return;
3644   }
3645   peer_msg = (struct GNUNET_MESH_PeerControl *) message;
3646   /* Sanity check for message size */
3647   if (sizeof (struct GNUNET_MESH_PeerControl) != ntohs (peer_msg->header.size))
3648   {
3649     GNUNET_break (0);
3650     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3651     return;
3652   }
3653
3654   /* Tunnel exists? */
3655   tid = ntohl (peer_msg->tunnel_id);
3656   t = tunnel_get_by_local_id (c, tid);
3657   if (NULL == t)
3658   {
3659     GNUNET_break (0);
3660     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3661     return;
3662   }
3663   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   on tunnel %X\n", t->id.tid);
3664
3665   /* Does client own tunnel? */
3666   if (t->client->handle != client)
3667   {
3668     GNUNET_break (0);
3669     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3670     return;
3671   }
3672
3673   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3674               "MESH:   for peer %s\n",
3675               GNUNET_i2s(&peer_msg->peer));
3676   /* Is the peer in the tunnel? */
3677   peer_info = GNUNET_CONTAINER_multihashmap_get(t->peers,
3678                                                 &peer_msg->peer.hashPubKey);
3679   if (NULL == peer_info)
3680   {
3681     GNUNET_break (0);
3682     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3683     return;
3684   }
3685
3686   /* Ok, delete peer from tunnel */
3687   GNUNET_CONTAINER_multihashmap_remove_all (t->peers,
3688                                             &peer_msg->peer.hashPubKey);
3689
3690   send_destroy_path (t, peer_info->id);
3691   tunnel_delete_peer(t, peer_info->id);
3692   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3693   return;
3694 }
3695
3696
3697 /**
3698  * Handler for connection requests to new peers by type
3699  *
3700  * @param cls closure
3701  * @param client identification of the client
3702  * @param message the actual message (ConnectPeerByType)
3703  */
3704 static void
3705 handle_local_connect_by_type (void *cls, struct GNUNET_SERVER_Client *client,
3706                               const struct GNUNET_MessageHeader *message)
3707 {
3708   struct GNUNET_MESH_ConnectPeerByType *connect_msg;
3709   struct MeshClient *c;
3710   struct MeshTunnel *t;
3711   GNUNET_HashCode hash;
3712   MESH_TunnelNumber tid;
3713
3714   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: got connect by type request\n");
3715   /* Sanity check for client registration */
3716   if (NULL == (c = client_get (client)))
3717   {
3718     GNUNET_break (0);
3719     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3720     return;
3721   }
3722
3723   connect_msg = (struct GNUNET_MESH_ConnectPeerByType *) message;
3724   /* Sanity check for message size */
3725   if (sizeof (struct GNUNET_MESH_ConnectPeerByType) !=
3726       ntohs (connect_msg->header.size))
3727   {
3728     GNUNET_break (0);
3729     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3730     return;
3731   }
3732
3733   /* Tunnel exists? */
3734   tid = ntohl (connect_msg->tunnel_id);
3735   t = tunnel_get_by_local_id (c, tid);
3736   if (NULL == t)
3737   {
3738     GNUNET_break (0);
3739     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3740     return;
3741   }
3742
3743   /* Does client own tunnel? */
3744   if (t->client->handle != client)
3745   {
3746     GNUNET_break (0);
3747     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3748     return;
3749   }
3750
3751   /* Do WE have the service? */
3752   t->type = ntohl (connect_msg->type);
3753   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  type requested: %u\n", t->type);
3754   GNUNET_CRYPTO_hash (&t->type, sizeof (GNUNET_MESH_ApplicationType), &hash);
3755   if (GNUNET_CONTAINER_multihashmap_contains (applications, &hash) ==
3756       GNUNET_YES)
3757   {
3758     /* Yes! Fast forward, add ourselves to the tunnel and send the
3759      * good news to the client
3760      */
3761     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  available locally\n");
3762     GNUNET_CONTAINER_multihashmap_put (t->peers, &my_full_id.hashPubKey,
3763                                        peer_info_get (&my_full_id),
3764                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
3765
3766     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  notifying client\n");
3767     send_client_peer_connected(t, myid);
3768     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  Done\n");
3769     GNUNET_SERVER_receive_done (client, GNUNET_OK);
3770     return;
3771   }
3772   /* Ok, lets find a peer offering the service */
3773   if (NULL != t->dht_get_type)
3774   {
3775     GNUNET_DHT_get_stop (t->dht_get_type);
3776   }
3777   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  looking in DHT for %s\n",
3778               GNUNET_h2s (&hash));
3779   t->dht_get_type =
3780       GNUNET_DHT_get_start (dht_handle,
3781                             GNUNET_TIME_UNIT_FOREVER_REL,
3782                             GNUNET_BLOCK_TYPE_TEST,
3783                             &hash,
3784                             10U,
3785                             GNUNET_DHT_RO_RECORD_ROUTE |
3786                               GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
3787                             NULL, 0,
3788                             &dht_get_type_handler, t);
3789
3790   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3791   return;
3792 }
3793
3794
3795 /**
3796  * Handler for client traffic directed to one peer
3797  *
3798  * @param cls closure
3799  * @param client identification of the client
3800  * @param message the actual message
3801  */
3802 static void
3803 handle_local_unicast (void *cls, struct GNUNET_SERVER_Client *client,
3804                       const struct GNUNET_MessageHeader *message)
3805 {
3806   struct MeshClient *c;
3807   struct MeshTunnel *t;
3808   struct MeshPeerInfo *pi;
3809   struct GNUNET_MESH_Unicast *data_msg;
3810   MESH_TunnelNumber tid;
3811   size_t size;
3812
3813   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3814               "MESH: Got a unicast request from a client!\n");
3815
3816   /* Sanity check for client registration */
3817   if (NULL == (c = client_get (client)))
3818   {
3819     GNUNET_break (0);
3820     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3821     return;
3822   }
3823   data_msg = (struct GNUNET_MESH_Unicast *) message;
3824   /* Sanity check for message size */
3825   size = ntohs (message->size);
3826   if (sizeof (struct GNUNET_MESH_Unicast) +
3827       sizeof (struct GNUNET_MessageHeader) > size)
3828   {
3829     GNUNET_break (0);
3830     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3831     return;
3832   }
3833
3834   /* Tunnel exists? */
3835   tid = ntohl (data_msg->tid);
3836   t = tunnel_get_by_local_id (c, tid);
3837   if (NULL == t)
3838   {
3839     GNUNET_break (0);
3840     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3841     return;
3842   }
3843
3844   /*  Is it a local tunnel? Then, does client own the tunnel? */
3845   if (NULL != t->client &&
3846       NULL != t->client->handle &&
3847       t->client->handle != client)
3848   {
3849     GNUNET_break (0);
3850     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3851     return;
3852   }
3853
3854   pi = GNUNET_CONTAINER_multihashmap_get (t->peers,
3855                                           &data_msg->destination.hashPubKey);
3856   /* Is the selected peer in the tunnel? */
3857   if (NULL == pi)
3858   {
3859     GNUNET_break (0);
3860     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3861     return;
3862   }
3863
3864   /* Ok, everything is correct, send the message
3865    * (pretend we got it from a mesh peer)
3866    */
3867   {
3868     char buf[ntohs (message->size)];
3869     struct GNUNET_MESH_Unicast *copy;
3870
3871     /* Work around const limitation */
3872     copy = (struct GNUNET_MESH_Unicast *) buf;
3873     memcpy (buf, data_msg, size);
3874     copy->oid = my_full_id;
3875     copy->tid = htonl (t->id.tid);
3876     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3877                 "MESH:   calling generic handler...\n");
3878     handle_mesh_data_unicast (NULL, &my_full_id, &copy->header, NULL, 0);
3879   }
3880   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3881   return;
3882 }
3883
3884
3885 /**
3886  * Handler for client traffic directed to the origin
3887  *
3888  * @param cls closure
3889  * @param client identification of the client
3890  * @param message the actual message
3891  */
3892 static void
3893 handle_local_to_origin (void *cls, struct GNUNET_SERVER_Client *client,
3894                         const struct GNUNET_MessageHeader *message)
3895 {
3896   struct GNUNET_MESH_ToOrigin *data_msg;
3897   struct GNUNET_PeerIdentity id;
3898   struct MeshClient *c;
3899   struct MeshTunnel *t;
3900   MESH_TunnelNumber tid;
3901   size_t size;
3902
3903   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3904               "MESH: Got a ToOrigin request from a client!\n");
3905
3906   /* Sanity check for client registration */
3907   if (NULL == (c = client_get (client)))
3908   {
3909     GNUNET_break (0);
3910     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3911     return;
3912   }
3913   data_msg = (struct GNUNET_MESH_ToOrigin *) message;
3914   /* Sanity check for message size */
3915   size = ntohs (message->size);
3916   if (sizeof (struct GNUNET_MESH_ToOrigin) +
3917       sizeof (struct GNUNET_MessageHeader) > size)
3918   {
3919     GNUNET_break (0);
3920     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3921     return;
3922   }
3923
3924   /* Tunnel exists? */
3925   tid = ntohl (data_msg->tid);
3926   if (tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
3927   {
3928     GNUNET_break (0);
3929     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3930     return;
3931   }
3932   t = tunnel_get_by_local_id (c, tid);
3933   if (NULL == t)
3934   {
3935     GNUNET_break (0);
3936     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3937     return;
3938   }
3939
3940   /*  It shouldn't be a local tunnel.  */
3941   if (NULL != t->client)
3942   {
3943     GNUNET_break (0);
3944     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3945     return;
3946   }
3947   GNUNET_PEER_resolve(t->id.oid, &id);
3948
3949   /* Ok, everything is correct, send the message
3950    * (pretend we got it from a mesh peer)
3951    */
3952   {
3953     char buf[ntohs (message->size)];
3954     struct GNUNET_MESH_ToOrigin *copy;
3955
3956     /* Work around const limitation */
3957     copy = (struct GNUNET_MESH_ToOrigin *) buf;
3958     memcpy (buf, data_msg, size);
3959     copy->oid = id;
3960     copy->tid = htonl (t->id.tid);
3961     copy->sender = my_full_id;
3962     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3963                 "MESH:   calling generic handler...\n");
3964     handle_mesh_data_to_orig (NULL, &my_full_id, &copy->header, NULL, 0);
3965   }
3966   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3967   return;
3968 }
3969
3970
3971 /**
3972  * Handler for client traffic directed to all peers in a tunnel
3973  *
3974  * @param cls closure
3975  * @param client identification of the client
3976  * @param message the actual message
3977  */
3978 static void
3979 handle_local_multicast (void *cls, struct GNUNET_SERVER_Client *client,
3980                         const struct GNUNET_MessageHeader *message)
3981 {
3982   struct MeshClient *c;
3983   struct MeshTunnel *t;
3984   struct GNUNET_MESH_Multicast *data_msg;
3985   MESH_TunnelNumber tid;
3986
3987   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3988               "MESH: Got a multicast request from a client!\n");
3989
3990   /* Sanity check for client registration */
3991   if (NULL == (c = client_get (client)))
3992   {
3993     GNUNET_break (0);
3994     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3995     return;
3996   }
3997   data_msg = (struct GNUNET_MESH_Multicast *) message;
3998   /* Sanity check for message size */
3999   if (sizeof (struct GNUNET_MESH_Multicast) +
4000       sizeof (struct GNUNET_MessageHeader) > ntohs (data_msg->header.size))
4001   {
4002     GNUNET_break (0);
4003     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4004     return;
4005   }
4006
4007   /* Tunnel exists? */
4008   tid = ntohl (data_msg->tid);
4009   t = tunnel_get_by_local_id (c, tid);
4010   if (NULL == t)
4011   {
4012     GNUNET_break (0);
4013     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4014     return;
4015   }
4016
4017   /* Does client own tunnel? */
4018   if (t->client->handle != client)
4019   {
4020     GNUNET_break (0);
4021     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4022     return;
4023   }
4024
4025   {
4026     char buf[ntohs(message->size)];
4027     struct GNUNET_MESH_Multicast *copy;
4028
4029     copy = (struct GNUNET_MESH_Multicast *)buf;
4030     memcpy(buf, message, ntohs(message->size));
4031     copy->oid = my_full_id;
4032     copy->tid = htonl(t->id.tid);
4033     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4034                 "MESH:   calling generic handler...\n");
4035     handle_mesh_data_multicast(client, &my_full_id, &copy->header, NULL, 0);
4036   }
4037
4038   /* receive done gets called when last copy is sent to a neighbor */
4039   return;
4040 }
4041
4042 /**
4043  * Functions to handle messages from clients
4044  */
4045 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
4046   {&handle_local_new_client, NULL,
4047    GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
4048   {&handle_local_tunnel_create, NULL,
4049    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE,
4050    sizeof (struct GNUNET_MESH_TunnelMessage)},
4051   {&handle_local_tunnel_destroy, NULL,
4052    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY,
4053    sizeof (struct GNUNET_MESH_TunnelMessage)},
4054   {&handle_local_connect_add, NULL,
4055    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD,
4056    sizeof (struct GNUNET_MESH_PeerControl)},
4057   {&handle_local_connect_del, NULL,
4058    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL,
4059    sizeof (struct GNUNET_MESH_PeerControl)},
4060   {&handle_local_connect_by_type, NULL,
4061    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE,
4062    sizeof (struct GNUNET_MESH_ConnectPeerByType)},
4063   {&handle_local_unicast, NULL,
4064    GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
4065   {&handle_local_to_origin, NULL,
4066    GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
4067   {&handle_local_multicast, NULL,
4068    GNUNET_MESSAGE_TYPE_MESH_MULTICAST, 0},
4069   {NULL, NULL, 0, 0}
4070 };
4071
4072
4073 /**
4074  * To be called on core init/fail.
4075  *
4076  * @param cls service closure
4077  * @param server handle to the server for this service
4078  * @param identity the public identity of this peer
4079  */
4080 static void
4081 core_init (void *cls, struct GNUNET_CORE_Handle *server,
4082            const struct GNUNET_PeerIdentity *identity)
4083 {
4084   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Core init\n");
4085   core_handle = server;
4086   if (0 != memcmp(identity, &my_full_id, sizeof(my_full_id)) || NULL == server)
4087   {
4088     GNUNET_log(GNUNET_ERROR_TYPE_ERROR, _("MESH: Wrong CORE service\n"));
4089     GNUNET_SCHEDULER_shutdown();
4090   }
4091   return;
4092 }
4093
4094 /**
4095  * Method called whenever a given peer connects.
4096  *
4097  * @param cls closure
4098  * @param peer peer identity this notification is about
4099  * @param atsi performance data for the connection
4100  * @param atsi_count number of records in 'atsi'
4101  */
4102 static void
4103 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer,
4104               const struct GNUNET_ATS_Information *atsi,
4105               unsigned int atsi_count)
4106 {
4107   struct MeshPeerInfo *peer_info;
4108   struct MeshPeerPath *path;
4109
4110 #if MESH_DEBUG_CONNECTION
4111   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Peer connected\n");
4112   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      %s\n",
4113               GNUNET_i2s(&my_full_id));
4114 #endif
4115   peer_info = peer_info_get (peer);
4116   if (myid == peer_info->id)
4117   {
4118 #if MESH_DEBUG_CONNECTION
4119     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      (self)\n");
4120 #endif
4121     return;
4122   }
4123 #if MESH_DEBUG_CONNECTION
4124   else
4125   {
4126     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      %s\n",
4127                 GNUNET_i2s(peer));
4128   }
4129 #endif
4130   path = path_new (2);
4131   path->peers[0] = myid;
4132   path->peers[1] = peer_info->id;
4133   GNUNET_PEER_change_rc(myid, 1);
4134   GNUNET_PEER_change_rc(peer_info->id, 1);
4135   path_add_to_peer (peer_info, path);
4136   return;
4137 }
4138
4139 /**
4140  * Method called whenever a peer disconnects.
4141  *
4142  * @param cls closure
4143  * @param peer peer identity this notification is about
4144  */
4145 static void
4146 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
4147 {
4148   struct MeshPeerInfo *pi;
4149   unsigned int i;
4150
4151 #if MESH_DEBUG_CONNECTION
4152   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Peer disconnected\n");
4153 #endif
4154   pi = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
4155   if (NULL == pi)
4156   {
4157     GNUNET_break (0);
4158     return;
4159   }
4160   for (i = 0; i < CORE_QUEUE_SIZE; i++)
4161   {
4162     /* TODO: notify that the transmission failed */
4163     peer_info_cancel_transmission(pi, i);
4164   }
4165   path_remove_from_peer (pi, pi->id, myid);
4166 #if MESH_DEBUG_CONNECTION
4167   if (myid == pi->id)
4168   {
4169     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      (self)\n");
4170   }
4171 #endif
4172   return;
4173 }
4174
4175
4176 /******************************************************************************/
4177 /************************      MAIN FUNCTIONS      ****************************/
4178 /******************************************************************************/
4179
4180 /**
4181  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
4182  *
4183  * @param cls closure
4184  * @param key current key code
4185  * @param value value in the hash map
4186  * @return GNUNET_YES if we should continue to iterate,
4187  *         GNUNET_NO if not.
4188  */
4189 static int
4190 shutdown_tunnel (void *cls, const GNUNET_HashCode * key, void *value)
4191 {
4192   struct MeshTunnel *t = value;
4193   tunnel_destroy(t);
4194   return GNUNET_YES;
4195 }
4196
4197 /**
4198  * Iterator over peer hash map entries to destroy the tunnel during shutdown.
4199  *
4200  * @param cls closure
4201  * @param key current key code
4202  * @param value value in the hash map
4203  * @return GNUNET_YES if we should continue to iterate,
4204  *         GNUNET_NO if not.
4205  */
4206 static int
4207 shutdown_peer (void *cls, const GNUNET_HashCode * key, void *value)
4208 {
4209   struct MeshPeerInfo *p = value;
4210   peer_info_destroy (p);
4211   return GNUNET_YES;
4212 }
4213
4214 /**
4215  * Task run during shutdown.
4216  *
4217  * @param cls unused
4218  * @param tc unused
4219  */
4220 static void
4221 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4222 {
4223   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: shutting down\n");
4224   /* TODO: destroy tunnels? */
4225   if (core_handle != NULL)
4226   {
4227     GNUNET_CORE_disconnect (core_handle);
4228     core_handle = NULL;
4229   }
4230   GNUNET_CONTAINER_multihashmap_iterate(tunnels, &shutdown_tunnel, NULL);
4231   GNUNET_CONTAINER_multihashmap_iterate(peers, &shutdown_peer, NULL);
4232   if (dht_handle != NULL)
4233   {
4234     GNUNET_DHT_disconnect (dht_handle);
4235     dht_handle = NULL;
4236   }
4237   if (nc != NULL)
4238   {
4239     GNUNET_SERVER_notification_context_destroy (nc);
4240     nc = NULL;
4241   }
4242   if (GNUNET_SCHEDULER_NO_TASK != announce_id_task)
4243   {
4244     GNUNET_SCHEDULER_cancel (announce_id_task);
4245     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
4246   }
4247   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: shut down\n");
4248 }
4249
4250 /**
4251  * Process mesh requests.
4252  *
4253  * @param cls closure
4254  * @param server the initialized server
4255  * @param c configuration to use
4256  */
4257 static void
4258 run (void *cls, struct GNUNET_SERVER_Handle *server,
4259      const struct GNUNET_CONFIGURATION_Handle *c)
4260 {
4261   struct MeshPeerInfo *peer;
4262   struct MeshPeerPath *p;
4263   char *keyfile;
4264
4265   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: starting to run\n");
4266   server_handle = server;
4267   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
4268                                      CORE_QUEUE_SIZE,   /* queue size */
4269                                      NULL,      /* Closure passed to MESH functions */
4270                                      &core_init,        /* Call core_init once connected */
4271                                      &core_connect,     /* Handle connects */
4272                                      &core_disconnect,  /* remove peers on disconnects */
4273                                      NULL,      /* Don't notify about all incoming messages */
4274                                      GNUNET_NO, /* For header only in notification */
4275                                      NULL,      /* Don't notify about all outbound messages */
4276                                      GNUNET_NO, /* For header-only out notification */
4277                                      core_handlers);    /* Register these handlers */
4278   if (core_handle == NULL)
4279   {
4280     GNUNET_break (0);
4281     GNUNET_SCHEDULER_shutdown ();
4282     return;
4283   }
4284
4285   if (GNUNET_OK !=
4286        GNUNET_CONFIGURATION_get_value_filename (c, "GNUNETD", "HOSTKEY",
4287                                                 &keyfile))
4288   {
4289     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
4290                 _
4291                 ("Mesh service is lacking key configuration settings.  Exiting.\n"));
4292     GNUNET_SCHEDULER_shutdown ();
4293     return;
4294   }
4295   my_private_key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
4296   GNUNET_free (keyfile);
4297   if (my_private_key == NULL)
4298   {
4299     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
4300                 _("Mesh service could not access hostkey.  Exiting.\n"));
4301     GNUNET_SCHEDULER_shutdown ();
4302     return;
4303   }
4304   GNUNET_CRYPTO_rsa_key_get_public (my_private_key, &my_public_key);
4305   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
4306                       &my_full_id.hashPubKey);
4307   myid = GNUNET_PEER_intern (&my_full_id);
4308
4309   dht_handle = GNUNET_DHT_connect (c, 64);
4310   if (dht_handle == NULL)
4311   {
4312     GNUNET_break (0);
4313   }
4314
4315   next_tid = 0;
4316   next_local_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
4317
4318   tunnels = GNUNET_CONTAINER_multihashmap_create (32);
4319   incoming_tunnels = GNUNET_CONTAINER_multihashmap_create (32);
4320   peers = GNUNET_CONTAINER_multihashmap_create (32);
4321   applications = GNUNET_CONTAINER_multihashmap_create (32);
4322   types = GNUNET_CONTAINER_multihashmap_create (32);
4323
4324   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
4325   nc = GNUNET_SERVER_notification_context_create (server_handle,
4326                                                   LOCAL_QUEUE_SIZE);
4327   GNUNET_SERVER_disconnect_notify (server_handle,
4328                                    &handle_local_client_disconnect,
4329                                    NULL);
4330
4331
4332   clients = NULL;
4333   clients_tail = NULL;
4334 #if MESH_DEBUG
4335   next_client_id = 0;
4336 #endif
4337
4338   announce_applications_task = GNUNET_SCHEDULER_NO_TASK;
4339   announce_id_task = GNUNET_SCHEDULER_add_now (&announce_id, cls);
4340
4341   /* Create a peer_info for the local peer */
4342   peer = peer_info_get(&my_full_id);
4343   p = path_new (1);
4344   p->peers[0] = myid;
4345   path_add_to_peer(peer, p);
4346
4347   /* Scheduled the task to clean up when shutdown is called */
4348   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
4349                                 NULL);
4350
4351   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: end of run()\n");
4352 }
4353
4354 /**
4355  * The main function for the mesh service.
4356  *
4357  * @param argc number of arguments from the command line
4358  * @param argv command line arguments
4359  * @return 0 ok, 1 on error
4360  */
4361 int
4362 main (int argc, char *const *argv)
4363 {
4364   int ret;
4365
4366   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: main()\n");
4367   ret =
4368       (GNUNET_OK ==
4369        GNUNET_SERVICE_run (argc, argv, "mesh", GNUNET_SERVICE_OPTION_NONE, &run,
4370                            NULL)) ? 0 : 1;
4371   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: main() END\n");
4372
4373   return ret;
4374 }