Fixed a bug when adding a path to a peer who is already a relay in the tunnel
[oweals/gnunet.git] / 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 (NULL == p)
1290     {
1291       GNUNET_break (0);
1292       return;
1293     }
1294
1295     if (p->length > 1)
1296     {
1297       send_create_path(peer, p, t);
1298     }
1299     else
1300     {
1301       path_destroy(p);
1302       send_client_peer_connected(t, myid);
1303     }
1304   }
1305   else if (NULL == peer->dhtget)
1306   {
1307     struct GNUNET_PeerIdentity id;
1308
1309     GNUNET_PEER_resolve(peer->id, &id);
1310     path_info = GNUNET_malloc(sizeof(struct MeshPathInfo));
1311     path_info->peer = peer;
1312     path_info->t = t;
1313     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1314                 "MESH:   Starting DHT GET for peer %s\n",
1315                 GNUNET_i2s (&id));
1316     peer->dhtgetcls = path_info;
1317     peer->dhtget =
1318         GNUNET_DHT_get_start(dht_handle,       /* handle */
1319                              GNUNET_TIME_UNIT_FOREVER_REL,     /* timeout */
1320                              GNUNET_BLOCK_TYPE_TEST,   /* type */
1321                              &id.hashPubKey,       /* key to search */
1322                              4,         /* replication level */
1323                              GNUNET_DHT_RO_RECORD_ROUTE |
1324                                GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
1325                              NULL,     /* xquery */
1326                              0,        /* xquery bits */
1327                              &dht_get_id_handler,
1328                              path_info);
1329   }
1330   /* Otherwise, there is no path but the DHT get is already started. */
1331 }
1332
1333
1334 /**
1335  * Task to delay the connection of a peer
1336  *
1337  * @param cls Closure (path info with tunnel and peer to connect).
1338  *            Will be free'd on exection.
1339  * @param tc TaskContext
1340  */
1341 static void
1342 peer_info_connect_task (void *cls,
1343                         const struct GNUNET_SCHEDULER_TaskContext *tc)
1344 {
1345   struct MeshPathInfo *path_info = cls;
1346
1347   if (GNUNET_SCHEDULER_REASON_SHUTDOWN == tc->reason)
1348   {
1349     GNUNET_free (cls);
1350   }
1351   peer_info_connect (path_info->peer, path_info->t);
1352   GNUNET_free (cls);
1353 }
1354
1355
1356 /**
1357  * Destroy the peer_info and free any allocated resources linked to it
1358  * 
1359  * @param pi The peer_info to destroy.
1360  * 
1361  * @return GNUNET_OK on success
1362  */
1363 static int
1364 peer_info_destroy (struct MeshPeerInfo *pi)
1365 {
1366   struct GNUNET_PeerIdentity id;
1367   struct MeshPeerPath *p;
1368   struct MeshPeerPath *nextp;
1369   unsigned int i;
1370
1371   GNUNET_PEER_resolve (pi->id, &id);
1372   GNUNET_PEER_change_rc (pi->id, -1);
1373
1374   if (GNUNET_YES !=
1375       GNUNET_CONTAINER_multihashmap_remove (peers,&id.hashPubKey, pi))
1376   {
1377     GNUNET_break (0);
1378     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1379                 "MESH: removing peer %s, not in hashmap\n",
1380                 GNUNET_i2s (&id));
1381   }
1382   if (NULL != pi->dhtget)
1383   {
1384     GNUNET_DHT_get_stop(pi->dhtget);
1385     GNUNET_free (pi->dhtgetcls);
1386   }
1387   for (i = 0; i < CORE_QUEUE_SIZE; i++)
1388   {
1389     peer_info_cancel_transmission(pi, i);
1390   }
1391   p = pi->path_head;
1392   while (NULL != p)
1393   {
1394     nextp = p->next;
1395     GNUNET_CONTAINER_DLL_remove (pi->path_head, pi->path_tail, p);
1396     path_destroy (p);
1397     p = nextp;
1398   }
1399   GNUNET_free (pi);
1400   return GNUNET_OK;
1401 }
1402
1403
1404 /**
1405  * Remove all paths that rely on a direct connection between p1 and p2
1406  * from the peer itself and notify all tunnels about it.
1407  *
1408  * @param peer PeerInfo of affected peer.
1409  * @param p1 GNUNET_PEER_Id of one peer.
1410  * @param p2 GNUNET_PEER_Id of another peer that was connected to the first and
1411  *           no longer is.
1412  *
1413  * TODO: optimize (see below)
1414  */
1415 static void
1416 path_remove_from_peer (struct MeshPeerInfo *peer,
1417                        GNUNET_PEER_Id p1,
1418                        GNUNET_PEER_Id p2)
1419 {
1420   struct MeshPeerPath *p;
1421   struct MeshPeerPath *aux;
1422   struct MeshPeerInfo *peer_d;
1423   GNUNET_PEER_Id d;
1424   unsigned int destroyed;
1425   unsigned int best;
1426   unsigned int cost;
1427   unsigned int i;
1428
1429   destroyed = 0;
1430   p = peer->path_head;
1431   while (NULL != p)
1432   {
1433     aux = p->next;
1434     for (i = 0; i < (p->length - 1); i++)
1435     {
1436       if ((p->peers[i] == p1 && p->peers[i + 1] == p2) ||
1437           (p->peers[i] == p2 && p->peers[i + 1] == p1))
1438       {
1439         GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
1440         path_destroy (p);
1441         destroyed++;
1442         break;
1443       }
1444     }
1445     p = aux;
1446   }
1447   if (0 == destroyed)
1448     return;
1449
1450   for (i = 0; i < peer->ntunnels; i++)
1451   {
1452     d = tunnel_notify_connection_broken (peer->tunnels[i], p1, p2);
1453     /* TODO
1454      * Problem: one or more peers have been deleted from the tunnel tree.
1455      * We don't know who they are to try to add them again.
1456      * We need to try to find a new path for each of the disconnected peers.
1457      * Some of them might already have a path to reach them that does not
1458      * involve p1 and p2. Adding all anew might render in a better tree than
1459      * the trivial immediate fix.
1460      * 
1461      * Trivial immiediate fix: try to reconnect to the disconnected node. All
1462      * its children will be reachable trough him.
1463      */
1464     peer_d = peer_info_get_short(d);
1465     best = UINT_MAX;
1466     aux = NULL;
1467     for (p = peer_d->path_head; NULL != p; p = p->next)
1468     {
1469       if ((cost = path_get_cost(peer->tunnels[i]->tree, p)) < best)
1470       {
1471         best = cost;
1472         aux = p;
1473       }
1474     }
1475     if (NULL != aux)
1476     {
1477       /* No callback, as peer will be already disconnected */
1478       tree_add_path(peer->tunnels[i]->tree, aux, NULL);
1479     }
1480     else
1481     {
1482       peer_info_connect (peer_d, peer->tunnels[i]);
1483     }
1484   }
1485 }
1486
1487
1488 /**
1489  * Add the path to the peer and update the path used to reach it in case this
1490  * is the shortest.
1491  *
1492  * @param peer_info Destination peer to add the path to.
1493  * @param path New path to add. Last peer must be the peer in arg 1.
1494  *             Path will be either used of freed if already known.
1495  *
1496  * TODO: trim the part from origin to us? Add it as path to origin?
1497  */
1498 void
1499 path_add_to_peer (struct MeshPeerInfo *peer_info, struct MeshPeerPath *path)
1500 {
1501   struct MeshPeerPath *aux;
1502   unsigned int l;
1503   unsigned int l2;
1504
1505 #if MESH_DEBUG
1506   struct GNUNET_PeerIdentity id;
1507
1508   GNUNET_PEER_resolve (peer_info->id, &id);
1509   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1510               "MESH: adding path [%u] to peer %s\n",
1511               path->length,
1512               GNUNET_i2s (&id));
1513 #endif
1514   if (NULL == peer_info || NULL == path)
1515   {
1516     GNUNET_break (0);
1517     path_destroy (path);
1518     return;
1519   }
1520
1521   l = path_get_length (path);
1522   if (0 == l)
1523   {
1524     GNUNET_free (path);
1525     return;
1526   }
1527
1528   for (aux = peer_info->path_head; aux != NULL; aux = aux->next)
1529   {
1530     l2 = path_get_length (aux);
1531     if (l2 > l)
1532     {
1533       GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head,
1534                                           peer_info->path_tail,
1535                                           aux,
1536                                           path);
1537       return;
1538     }
1539     else
1540     {
1541       if (l2 == l && memcmp(path->peers, aux->peers, l) == 0)
1542       {
1543         path_destroy(path);
1544         return;
1545       }
1546     }
1547   }
1548   GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head,
1549                                     peer_info->path_tail,
1550                                     path);
1551   return;
1552 }
1553
1554
1555 /**
1556  * Add the path to the origin peer and update the path used to reach it in case
1557  * this is the shortest.
1558  * The path is given in peer_info -> destination, therefore we turn the path
1559  * upside down first.
1560  *
1561  * @param peer_info Peer to add the path to, being the origin of the path.
1562  * @param path New path to add after being inversed.
1563  */
1564 static void
1565 path_add_to_origin (struct MeshPeerInfo *peer_info, struct MeshPeerPath *path)
1566 {
1567   path_invert(path);
1568   path_add_to_peer (peer_info, path);
1569 }
1570
1571
1572 /**
1573  * Build a PeerPath from the paths returned from the DHT, reversing the paths
1574  * to obtain a local peer -> destination path and interning the peer ids.
1575  *
1576  * @return Newly allocated and created path
1577  */
1578 static struct MeshPeerPath *
1579 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
1580                      unsigned int get_path_length,
1581                      const struct GNUNET_PeerIdentity *put_path,
1582                      unsigned int put_path_length)
1583 {
1584   struct MeshPeerPath *p;
1585   GNUNET_PEER_Id id;
1586   int i;
1587
1588   p = path_new (1);
1589   p->peers[0] = myid;
1590   GNUNET_PEER_change_rc(myid, 1);
1591   i = get_path_length;
1592   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    GET has %d hops.\n", i);
1593   for (i--; i >= 0; i--)
1594   {
1595     id = GNUNET_PEER_intern (&get_path[i]);
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 GET: %s.\n",
1605                  GNUNET_i2s(&get_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   i = put_path_length;
1612   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    PUT has %d hops.\n", i);
1613   for (i--; i >= 0; i--)
1614   {
1615     id = GNUNET_PEER_intern (&put_path[i]);
1616     if (id == myid)
1617     {
1618       /* PUT path went through us, so discard the path up until now and start
1619        * from here to get a much shorter (and loop-free) path.
1620        */
1621       path_destroy (p);
1622       p = path_new (0);
1623     }
1624     if (p->length > 0 && id == p->peers[p->length - 1])
1625     {
1626       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:    Optimizing 1 hop out.\n");
1627       GNUNET_PEER_change_rc(id, -1);
1628     }
1629     else
1630     {
1631       GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
1632             "MESH:    Adding from PUT: %s.\n",
1633             GNUNET_i2s(&put_path[i]));
1634       p->length++;
1635       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
1636       p->peers[p->length - 1] = id;
1637     }
1638   }
1639 #if MESH_DEBUG
1640   if (get_path_length > 0)
1641     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1642                 "MESH:    (first of GET: %s)\n",
1643                 GNUNET_i2s(&get_path[0]));
1644   if (put_path_length > 0)
1645     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1646                 "MESH:    (first of PUT: %s)\n",
1647                 GNUNET_i2s(&put_path[0]));
1648   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1649               "MESH:    In total: %d hops\n",
1650               p->length);
1651   for (i = 0; i < p->length; i++)
1652   {
1653     struct GNUNET_PeerIdentity peer_id;
1654
1655     GNUNET_PEER_resolve(p->peers[i], &peer_id);
1656     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1657               "MESH:        %u: %s\n",
1658               p->peers[i],
1659               GNUNET_i2s(&peer_id));
1660   }
1661 #endif
1662   return p;
1663 }
1664
1665
1666 /**
1667  * Send keepalive packets for a peer
1668  *
1669  * @param cls Closure (tunnel for which to send the keepalive).
1670  * @param tc Notification context.
1671  *
1672  * TODO: implement explicit multicast keepalive?
1673  */
1674 static void
1675 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1676
1677
1678 /**
1679  * Search for a tunnel among the tunnels for a client
1680  *
1681  * @param c the client whose tunnels to search in
1682  * @param tid the local id of the tunnel
1683  *
1684  * @return tunnel handler, NULL if doesn't exist
1685  */
1686 static struct MeshTunnel *
1687 tunnel_get_by_local_id (struct MeshClient *c, MESH_TunnelNumber tid)
1688 {
1689   GNUNET_HashCode hash;
1690
1691   GNUNET_CRYPTO_hash (&tid, sizeof (MESH_TunnelNumber), &hash);
1692   if (tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1693   {
1694     return GNUNET_CONTAINER_multihashmap_get (incoming_tunnels, &hash);
1695   }
1696   return GNUNET_CONTAINER_multihashmap_get (c->tunnels, &hash);
1697 }
1698
1699
1700 /**
1701  * Search for a tunnel by global ID using PEER_ID
1702  *
1703  * @param pi owner of the tunnel
1704  * @param tid global tunnel number
1705  *
1706  * @return tunnel handler, NULL if doesn't exist
1707  */
1708 static struct MeshTunnel *
1709 tunnel_get_by_pi (GNUNET_PEER_Id pi, MESH_TunnelNumber tid)
1710 {
1711   struct MESH_TunnelID id;
1712   GNUNET_HashCode hash;
1713
1714   id.oid = pi;
1715   id.tid = tid;
1716
1717   GNUNET_CRYPTO_hash (&id, sizeof (struct MESH_TunnelID), &hash);
1718   return GNUNET_CONTAINER_multihashmap_get (tunnels, &hash);
1719 }
1720
1721
1722 /**
1723  * Search for a tunnel by global ID using full PeerIdentities
1724  *
1725  * @param oid owner of the tunnel
1726  * @param tid global tunnel number
1727  *
1728  * @return tunnel handler, NULL if doesn't exist
1729  */
1730 static struct MeshTunnel *
1731 tunnel_get (struct GNUNET_PeerIdentity *oid, MESH_TunnelNumber tid)
1732 {
1733   return tunnel_get_by_pi (GNUNET_PEER_search (oid), tid);
1734 }
1735
1736
1737 /**
1738  * Callback used to notify a client owner of a tunnel that a peer has
1739  * disconnected, most likely because of a path change.
1740  *
1741  * @param n Node in the tree representing the disconnected peer
1742  * 
1743  * FIXME: pass tunnel via cls, make param just a peer identity
1744  */
1745 void
1746 notify_peer_disconnected (const struct MeshTunnelTreeNode *n)
1747 {
1748   struct MeshPeerInfo *peer;
1749   struct MeshPathInfo *path_info;
1750
1751   if (NULL != n->t->client && NULL != nc)
1752   {
1753     struct GNUNET_MESH_PeerControl msg;
1754     msg.header.size = htons (sizeof (msg));
1755     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL);
1756     msg.tunnel_id = htonl (n->t->local_tid);
1757     GNUNET_PEER_resolve (n->peer, &msg.peer);
1758     GNUNET_SERVER_notification_context_unicast (nc, n->t->client->handle,
1759                                                 &msg.header, GNUNET_NO);
1760   }
1761   peer = peer_info_get_short(n->peer);
1762   path_info = GNUNET_malloc (sizeof (struct MeshPathInfo));
1763   path_info->peer = peer;
1764   path_info->t = n->t;
1765   GNUNET_SCHEDULER_add_now(&peer_info_connect_task, path_info);
1766 }
1767
1768
1769 /**
1770  * Add a peer to a tunnel, accomodating paths accordingly and initializing all
1771  * needed rescources.
1772  * If peer already exists, reevaluate shortest path and change if different.
1773  *
1774  * @param t Tunnel we want to add a new peer to
1775  * @param peer PeerInfo of the peer being added
1776  *
1777  */
1778 static void
1779 tunnel_add_peer (struct MeshTunnel *t, struct MeshPeerInfo *peer)
1780 {
1781   struct GNUNET_PeerIdentity id;
1782   struct MeshPeerPath *best_p;
1783   struct MeshPeerPath *p;
1784   unsigned int best_cost;
1785   unsigned int cost;
1786
1787   GNUNET_PEER_resolve(peer->id, &id);
1788   if (GNUNET_NO ==
1789       GNUNET_CONTAINER_multihashmap_contains(t->peers, &id.hashPubKey))
1790   {
1791     t->peers_total++;
1792     GNUNET_array_append (peer->tunnels, peer->ntunnels, t);
1793     GNUNET_CONTAINER_multihashmap_put(
1794       t->peers,
1795       &id.hashPubKey,
1796       peer,
1797       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1798   }
1799
1800   if (NULL != (p = peer->path_head))
1801   {
1802     best_p = p;
1803     best_cost = path_get_cost(t->tree, p);
1804     while (NULL != p)
1805     {
1806       if ((cost = path_get_cost (t->tree, p)) < best_cost)
1807       {
1808         best_cost = cost;
1809         best_p = p;
1810       }
1811       p = p->next;
1812     }
1813     tree_add_path (t->tree, best_p, &notify_peer_disconnected);
1814     if (GNUNET_SCHEDULER_NO_TASK == t->path_refresh_task)
1815       t->path_refresh_task =
1816           GNUNET_SCHEDULER_add_delayed (t->tree->refresh, &path_refresh, t);
1817   }
1818   else
1819   {
1820     /* Start a DHT get if necessary */
1821     peer_info_connect(peer, t);
1822   }
1823 }
1824
1825 /**
1826  * Add a path to a tunnel which we don't own, just to remember the next hop.
1827  * If destination node was already in the tunnel, the first hop information
1828  * will be replaced with the new path.
1829  *
1830  * @param t Tunnel we want to add a new peer to
1831  * @param p Path to add
1832  * @param own_pos Position of local node in path.
1833  *
1834  */
1835 static void
1836 tunnel_add_path (struct MeshTunnel *t,
1837                  struct MeshPeerPath *p,
1838                  unsigned int own_pos)
1839 {
1840   struct GNUNET_PeerIdentity id;
1841
1842   GNUNET_assert (0 != own_pos);
1843   tree_add_path(t->tree, p, NULL);
1844   if (NULL == t->tree->me)
1845     t->tree->me = tree_find_peer(t->tree->root, p->peers[own_pos]);
1846   if (own_pos < p->length - 1)
1847   {
1848     GNUNET_PEER_resolve (p->peers[own_pos + 1], &id);
1849     tree_update_first_hops(t->tree, t->tree->me, &id);
1850   }
1851 }
1852
1853
1854 /**
1855  * Notifies a tunnel that a connection has broken that affects at least
1856  * some of its peers. Sends a notification towards the root of the tree.
1857  * In case the peer is the owner of the tree, notifies the client that owns
1858  * the tunnel and tries to reconnect.
1859  *
1860  * @param t Tunnel affected.
1861  * @param p1 Peer that got disconnected from p2.
1862  * @param p2 Peer that got disconnected from p1.
1863  *
1864  * @return Short ID of the peer disconnected (either p1 or p2).
1865  *         0 if the tunnel remained unaffected.
1866  */
1867 static GNUNET_PEER_Id
1868 tunnel_notify_connection_broken (struct MeshTunnel *t,
1869                                  GNUNET_PEER_Id p1,
1870                                  GNUNET_PEER_Id p2)
1871 {
1872   GNUNET_PEER_Id pid;
1873
1874   pid = tree_notify_connection_broken (t->tree,
1875                                        p1,
1876                                        p2,
1877                                        &notify_peer_disconnected);
1878   if (myid != p1 && myid != p2)
1879   {
1880     return pid;
1881   }
1882   if (pid != myid)
1883   {
1884     if (NULL != t->tree->me->parent)
1885     {
1886       /* We are the peer still connected, notify owner of the disconnection. */
1887       struct GNUNET_MESH_PathBroken msg;
1888       struct GNUNET_PeerIdentity neighbor;
1889
1890       msg.header.size = htons (sizeof (msg));
1891       msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN);
1892       GNUNET_PEER_resolve (t->id.oid, &msg.oid);
1893       msg.tid = htonl (t->id.tid);
1894       msg.peer1 = my_full_id;
1895       GNUNET_PEER_resolve (pid, &msg.peer2);
1896       GNUNET_PEER_resolve (t->tree->me->parent->peer, &neighbor);
1897       send_message (&msg.header, &neighbor);
1898     }
1899   }
1900   return pid;
1901 }
1902
1903
1904 /**
1905  * Send a message in a tunnel in multicast, sending a copy to each child node
1906  * down the local one in the tunnel tree.
1907  *
1908  * @param t Tunnel in which to send the data.
1909  * @param msg Message to be sent
1910  *
1911  * @return Number of copies sent.
1912  * 
1913  * TODO: unifiy shared resources and reference counter management
1914  */
1915 static int
1916 tunnel_send_multicast (struct MeshTunnel *t,
1917                        const struct GNUNET_MessageHeader *msg)
1918 {
1919   struct GNUNET_PeerIdentity neighbor;
1920   struct MeshDataDescriptor *info;
1921   struct MeshTunnelTreeNode *n;
1922   struct MeshTunnelTreeNode *counter;
1923   GNUNET_SCHEDULER_TaskIdentifier *task;
1924   unsigned int *copies;
1925   unsigned int i;
1926   size_t size;
1927   void *data;
1928
1929   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1930               "MESH:  sending a multicast packet...\n");
1931   size = ntohs (msg->size);
1932   GNUNET_assert (NULL != t->tree->me);
1933   n = t->tree->me->children_head;
1934   if (NULL == n)
1935   {
1936     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1937               "MESH:  no children in the tree, no one to send.\n");
1938     return 0;
1939   }
1940   copies = GNUNET_malloc (sizeof (unsigned int));
1941   for (counter = n; NULL != counter; counter = counter->next)
1942     (*copies)++;
1943   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  (%u copies)\n", *copies);
1944   data = GNUNET_malloc (size);
1945   memcpy (data, msg, size);
1946   if (NULL != t->client)
1947   {
1948     task = GNUNET_malloc (sizeof (GNUNET_SCHEDULER_TaskIdentifier));
1949     *task = GNUNET_SCHEDULER_add_delayed (UNACKNOWLEDGED_WAIT,
1950                                          &client_allow_send,
1951                                          t->client->handle);
1952   }
1953   else
1954     task = NULL; // So GCC shuts up about task being potentially uninitialized
1955   while (NULL != n)
1956   {
1957     info = GNUNET_malloc (sizeof (struct MeshDataDescriptor));
1958     info->data = data;
1959     info->size = size;
1960     info->copies = copies;
1961     if (NULL != t->client)
1962     {
1963       info->client = t->client->handle;
1964       info->timeout_task = task;
1965     }
1966     info->destination = n->peer;
1967     GNUNET_PEER_resolve (n->peer, &neighbor);
1968     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1969               "MESH:  sending to %s...\n",
1970               GNUNET_i2s (&neighbor));
1971     info->peer = peer_info_get(&neighbor);
1972     GNUNET_assert (NULL != info->peer);
1973     i = peer_info_transmit_slot(info->peer);
1974     info->handler_n = i;
1975     info->peer->infos[i] = info;
1976     info->peer->types[i] = GNUNET_MESSAGE_TYPE_MESH_MULTICAST;
1977     info->peer->core_transmit[i] =
1978         GNUNET_CORE_notify_transmit_ready (core_handle,
1979                                            0,
1980                                            0,
1981                                            GNUNET_TIME_UNIT_FOREVER_REL,
1982                                            &neighbor,
1983                                            size,
1984                                            &send_core_data_multicast, info);
1985     n = n->next;
1986   }
1987   return *copies;
1988 }
1989
1990
1991 /**
1992  * Send a message to all peers in this tunnel that the tunnel is no longer
1993  * valid.
1994  *
1995  * @param t The tunnel whose peers to notify.
1996  */
1997 static void
1998 tunnel_send_destroy (struct MeshTunnel *t)
1999 {
2000   struct GNUNET_MESH_TunnelDestroy msg;
2001
2002   msg.header.size = htons (sizeof (msg));
2003   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY);
2004   GNUNET_PEER_resolve (t->id.oid, &msg.oid);
2005   msg.tid = htonl (t->id.tid);
2006   tunnel_send_multicast (t, &msg.header);
2007 }
2008
2009
2010
2011 /**
2012  * Destroy the tunnel and free any allocated resources linked to it
2013  *
2014  * @param t the tunnel to destroy
2015  *
2016  * @return GNUNET_OK on success
2017  */
2018 static int
2019 tunnel_destroy (struct MeshTunnel *t)
2020 {
2021   struct MeshClient *c;
2022   struct MeshQueue *q;
2023   struct MeshQueue *qn;
2024   GNUNET_HashCode hash;
2025   int r;
2026
2027   if (NULL == t)
2028     return GNUNET_OK;
2029
2030   c = t->client;
2031 #if MESH_DEBUG
2032   {
2033     struct GNUNET_PeerIdentity id;
2034
2035     GNUNET_PEER_resolve(t->id.oid, &id);
2036     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2037                 "MESH: destroying tunnel %s [%x]\n",
2038                 GNUNET_i2s (&id),
2039                 t->id.tid);
2040     if (NULL != c)
2041       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
2042   }
2043 #endif
2044
2045   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2046   if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (tunnels, &hash, t))
2047   {
2048     r = GNUNET_SYSERR;
2049   }
2050
2051   GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2052   if (NULL != c &&
2053       GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (c->tunnels, &hash, t))
2054   {
2055     r = GNUNET_SYSERR;
2056   }
2057   if (t->local_tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
2058   {
2059     GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2060     GNUNET_break (GNUNET_YES ==
2061       GNUNET_CONTAINER_multihashmap_remove (incoming_tunnels, &hash, t));
2062   }
2063
2064   if (NULL != t->peers)
2065   {
2066     GNUNET_CONTAINER_multihashmap_iterate(t->peers,
2067                                           &peer_info_delete_tunnel, t);
2068     GNUNET_CONTAINER_multihashmap_destroy (t->peers);
2069   }
2070   q = t->queue_head;
2071   while (NULL != q)
2072   {
2073     if (NULL != q->data)
2074       GNUNET_free (q->data);
2075     qn = q->next;
2076     GNUNET_free (q);
2077     q = qn;
2078     /* TODO cancel core transmit ready in case it was active */
2079   }
2080   tree_destroy(t->tree);
2081   if (NULL != t->dht_get_type)
2082     GNUNET_DHT_get_stop(t->dht_get_type);
2083   if (GNUNET_SCHEDULER_NO_TASK != t->timeout_task)
2084     GNUNET_SCHEDULER_cancel(t->timeout_task);
2085   if (GNUNET_SCHEDULER_NO_TASK != t->path_refresh_task)
2086     GNUNET_SCHEDULER_cancel(t->path_refresh_task);
2087   GNUNET_free (t);
2088   return r;
2089 }
2090
2091
2092 /**
2093  * Removes an explicit path from a tunnel, freeing all intermediate nodes
2094  * that are no longer needed, as well as nodes of no longer reachable peers.
2095  * The tunnel itself is also destoyed if results in a remote empty tunnel.
2096  *
2097  * @param t Tunnel from which to remove the path.
2098  * @param p Peer which should be removed.
2099  */
2100 static void
2101 tunnel_delete_peer (struct MeshTunnel *t,
2102                     GNUNET_PEER_Id peer)
2103 {
2104   GNUNET_break (GNUNET_OK == tree_del_peer (t->tree, peer, NULL));
2105   if (NULL == t->tree->root)
2106     tunnel_destroy (t);
2107 }
2108
2109
2110 /**
2111  * tunnel_destroy_iterator: iterator for deleting each tunnel that belongs to a
2112  * client when the client disconnects.
2113  * 
2114  * @param cls closure (client that is disconnecting)
2115  * @param key the hash of the local tunnel id (used to access the hashmap)
2116  * @param value the value stored at the key (tunnel to destroy)
2117  * 
2118  * @return GNUNET_OK on success
2119  */
2120 static int
2121 tunnel_destroy_iterator (void *cls, const GNUNET_HashCode * key, void *value)
2122 {
2123   struct MeshTunnel *t = value;
2124   int r;
2125
2126   if (NULL != t->dht_get_type)
2127   {
2128     GNUNET_DHT_get_stop (t->dht_get_type);
2129   }
2130   r = tunnel_destroy (t);
2131   return r;
2132 }
2133
2134
2135 /**
2136  * Timeout function, destroys tunnel if called
2137  *
2138  * @param cls Closure (tunnel to destroy).
2139  * @param tc TaskContext
2140  */
2141 static void
2142 tunnel_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2143 {
2144   struct MeshTunnel *t = cls;
2145
2146   if (GNUNET_SCHEDULER_REASON_SHUTDOWN == tc->reason)
2147     return;
2148   t->timeout_task = GNUNET_SCHEDULER_NO_TASK;
2149   tunnel_destroy (t);
2150 }
2151
2152 /**
2153  * Resets the tunnel timeout. Starts it if no timeout was running.
2154  *
2155  * @param t Tunnel whose timeout to reset.
2156  */
2157 static void
2158 tunnel_reset_timeout (struct MeshTunnel *t)
2159 {
2160   if (GNUNET_SCHEDULER_NO_TASK != t->timeout_task)
2161     GNUNET_SCHEDULER_cancel (t->timeout_task);
2162   t->timeout_task = GNUNET_SCHEDULER_add_delayed (
2163       GNUNET_TIME_relative_multiply(REFRESH_PATH_TIME, 4),
2164       &tunnel_timeout,
2165       t);
2166 }
2167
2168
2169 /******************************************************************************/
2170 /****************      MESH NETWORK HANDLER HELPERS     ***********************/
2171 /******************************************************************************/
2172
2173 /**
2174  * Function called to notify a client about the socket
2175  * being ready to queue more data.  "buf" will be
2176  * NULL and "size" zero if the socket was closed for
2177  * writing in the meantime.
2178  *
2179  * @param cls closure
2180  * @param size number of bytes available in buf
2181  * @param buf where the callee should write the message
2182  * @return number of bytes written to buf
2183  */
2184 static size_t
2185 send_core_create_path (void *cls, size_t size, void *buf)
2186 {
2187   struct MeshPathInfo *info = cls;
2188   struct GNUNET_MESH_ManipulatePath *msg;
2189   struct GNUNET_PeerIdentity *peer_ptr;
2190   struct MeshPeerInfo *peer = info->peer;
2191   struct MeshTunnel *t = info->t;
2192   struct MeshPeerPath *p = info->path;
2193   size_t size_needed;
2194   int i;
2195
2196   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2197               "MESH: CREATE PATH sending...\n");
2198   size_needed =
2199       sizeof (struct GNUNET_MESH_ManipulatePath) +
2200       p->length * sizeof (struct GNUNET_PeerIdentity);
2201
2202   if (size < size_needed || NULL == buf)
2203   {
2204     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: create path retransmit!\n");
2205     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   buf:  %p\n", buf);
2206     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   size: (%u/%u)\n",
2207                 size, size_needed);
2208     info->peer->core_transmit[info->pos] = 
2209       GNUNET_CORE_notify_transmit_ready (core_handle, 0, 0,
2210                                         GNUNET_TIME_UNIT_FOREVER_REL,
2211                                         path_get_first_hop (t->tree, peer->id),
2212                                         size_needed,
2213                                         &send_core_create_path,
2214                                         info);
2215     return 0;
2216   }
2217   info->peer->core_transmit[info->pos] = NULL;
2218 #if MESH_DEBUG
2219   {
2220     struct GNUNET_PeerIdentity id;
2221
2222     GNUNET_PEER_resolve (peer->id, &id);
2223     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2224                 "MESH:   setting core_transmit %s [%u] to NULL\n",
2225                 GNUNET_i2s (&id),
2226                 info->pos);
2227   }
2228 #endif
2229   msg = (struct GNUNET_MESH_ManipulatePath *) buf;
2230   msg->header.size = htons (size_needed);
2231   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE);
2232   msg->tid = ntohl (t->id.tid);
2233
2234   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
2235   for (i = 0; i < p->length; i++)
2236   {
2237     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
2238   }
2239
2240   path_destroy (p);
2241   GNUNET_free (info);
2242
2243   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2244               "MESH: CREATE PATH (%u bytes long) sent!\n",
2245               size_needed);
2246   return size_needed;
2247 }
2248
2249
2250 /**
2251  * Function called to notify a client about the socket
2252  * being ready to queue more data.  "buf" will be
2253  * NULL and "size" zero if the socket was closed for
2254  * writing in the meantime.
2255  *
2256  * @param cls closure (data itself)
2257  * @param size number of bytes available in buf
2258  * @param buf where the callee should write the message
2259  * 
2260  * @return number of bytes written to buf
2261  */
2262 static size_t
2263 send_core_data_multicast (void *cls, size_t size, void *buf)
2264 {
2265   struct MeshDataDescriptor *info = cls;
2266   size_t total_size;
2267
2268   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Multicast callback.\n");
2269   GNUNET_assert (NULL != info);
2270   GNUNET_assert (NULL != info->peer);
2271   total_size = info->size;
2272   GNUNET_assert (total_size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
2273
2274   if (total_size > size)
2275   {
2276     /* Retry */
2277     struct GNUNET_PeerIdentity id;
2278
2279     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2280                 "MESH: Multicast: retransmitting... (%u/%u)\n",
2281                 size, total_size);
2282     GNUNET_PEER_resolve(info->peer->id, &id);
2283     info->peer->core_transmit[info->handler_n] =
2284       GNUNET_CORE_notify_transmit_ready (core_handle,
2285                                          0,
2286                                          0,
2287                                          GNUNET_TIME_UNIT_FOREVER_REL,
2288                                          &id,
2289                                          total_size,
2290                                          &send_core_data_multicast,
2291                                          info);
2292     return 0;
2293   }
2294   info->peer->core_transmit[info->handler_n] = NULL;
2295   info->peer->infos[info->handler_n] = NULL;
2296   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  copying data...\n");
2297   memcpy (buf, info->data, total_size);
2298 #if MESH_DEBUG
2299   {
2300     struct GNUNET_MESH_Multicast *mc;
2301     struct GNUNET_MessageHeader *mh;
2302
2303     mh = buf;
2304     if (ntohs (mh->type) == GNUNET_MESSAGE_TYPE_MESH_MULTICAST)
2305     {
2306       mc = (struct GNUNET_MESH_Multicast *) mh;
2307       mh = (struct GNUNET_MessageHeader *) &mc[1];
2308       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2309                   "MESH:  multicast, payload type %u\n",
2310                   ntohs (mh->type));
2311       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2312                   "MESH:  multicast, payload size %u\n",
2313                   ntohs (mh->size));
2314     }
2315     else
2316     {
2317       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2318                   "MESH:  type %u\n",
2319                   ntohs (mh->type));
2320     }
2321   }
2322 #endif
2323   data_descriptor_decrement_multicast (info);
2324   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: freeing info...\n");
2325   GNUNET_free (info);
2326   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: return %u\n", total_size);
2327   return total_size;
2328 }
2329
2330
2331 /**
2332  * Function called to notify a client about the socket
2333  * being ready to queue more data.  "buf" will be
2334  * NULL and "size" zero if the socket was closed for
2335  * writing in the meantime.
2336  *
2337  * @param cls closure (MeshDataDescriptor)
2338  * @param size number of bytes available in buf
2339  * @param buf where the callee should write the message
2340  * @return number of bytes written to buf
2341  */
2342 static size_t
2343 send_core_path_ack (void *cls, size_t size, void *buf)
2344 {
2345   struct MeshDataDescriptor *info = cls;
2346   struct GNUNET_MESH_PathACK *msg = buf;
2347
2348   GNUNET_assert (NULL != info);
2349   if (info->peer)
2350   {
2351     info->peer->core_transmit[info->handler_n] = NULL;
2352   }
2353   if (sizeof (struct GNUNET_MESH_PathACK) > size)
2354   {
2355     GNUNET_break (0);
2356     return 0;
2357   }
2358   msg->header.size = htons (sizeof (struct GNUNET_MESH_PathACK));
2359   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_ACK);
2360   GNUNET_PEER_resolve (info->origin->oid, &msg->oid);
2361   msg->tid = htonl (info->origin->tid);
2362   msg->peer_id = my_full_id;
2363   GNUNET_free (info);
2364   /* TODO add signature */
2365
2366   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: PATH ACK sent!\n");
2367   return sizeof (struct GNUNET_MESH_PathACK);
2368 }
2369
2370
2371 /******************************************************************************/
2372 /********************      MESH NETWORK HANDLERS     **************************/
2373 /******************************************************************************/
2374
2375
2376 /**
2377  * Core handler for path creation
2378  *
2379  * @param cls closure
2380  * @param message message
2381  * @param peer peer identity this notification is about
2382  * @param atsi performance data
2383  * @param atsi_count number of records in 'atsi'
2384  * 
2385  * @return GNUNET_OK to keep the connection open,
2386  *         GNUNET_SYSERR to close it (signal serious error)
2387  */
2388 static int
2389 handle_mesh_path_create (void *cls, const struct GNUNET_PeerIdentity *peer,
2390                          const struct GNUNET_MessageHeader *message,
2391                          const struct GNUNET_ATS_Information *atsi,
2392                          unsigned int atsi_count)
2393 {
2394   unsigned int own_pos;
2395   uint16_t size;
2396   uint16_t i;
2397   MESH_TunnelNumber tid;
2398   struct GNUNET_MESH_ManipulatePath *msg;
2399   struct GNUNET_PeerIdentity *pi;
2400   GNUNET_HashCode hash;
2401   struct MeshPeerPath *path;
2402   struct MeshPeerInfo *dest_peer_info;
2403   struct MeshPeerInfo *orig_peer_info;
2404   struct MeshTunnel *t;
2405
2406   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2407               "MESH: Received a path create msg [%s]\n",
2408               GNUNET_i2s(&my_full_id));
2409   size = ntohs (message->size);
2410   if (size < sizeof (struct GNUNET_MESH_ManipulatePath))
2411   {
2412     GNUNET_break_op (0);
2413     return GNUNET_OK;
2414   }
2415
2416   size -= sizeof (struct GNUNET_MESH_ManipulatePath);
2417   if (size % sizeof (struct GNUNET_PeerIdentity))
2418   {
2419     GNUNET_break_op (0);
2420     return GNUNET_OK;
2421   }
2422   size /= sizeof (struct GNUNET_PeerIdentity);
2423   if (size < 2)
2424   {
2425     GNUNET_break_op (0);
2426     return GNUNET_OK;
2427   }
2428   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2429               "MESH:     path has %u hops.\n",
2430               size);
2431   msg = (struct GNUNET_MESH_ManipulatePath *) message;
2432
2433   tid = ntohl (msg->tid);
2434   pi = (struct GNUNET_PeerIdentity *) &msg[1];
2435   t = tunnel_get (pi, tid);
2436   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2437               "MESH:     path is for tunnel %s [%X].\n",
2438               GNUNET_i2s(pi),
2439               tid);
2440   if (NULL == t)
2441   {
2442     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating tunnel\n");
2443     t = GNUNET_malloc (sizeof (struct MeshTunnel));
2444     t->id.oid = GNUNET_PEER_intern (pi);
2445     t->id.tid = tid;
2446     t->local_tid = next_local_tid++;
2447     /* FIXME test if taken */
2448     next_local_tid |= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
2449     t->tree = tree_new(t, t->id.oid);
2450
2451     GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2452     if (GNUNET_OK !=
2453         GNUNET_CONTAINER_multihashmap_put (
2454             tunnels,
2455             &hash,
2456             t,
2457             GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
2458     {
2459       tunnel_destroy (t);
2460       GNUNET_break (0);
2461       return GNUNET_OK;
2462     }
2463     tunnel_reset_timeout (t);
2464     GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
2465     if (GNUNET_OK !=
2466         GNUNET_CONTAINER_multihashmap_put (
2467             incoming_tunnels,
2468             &hash,
2469             t,
2470             GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
2471     {
2472       tunnel_destroy (t);
2473       GNUNET_break (0);
2474       return GNUNET_OK;
2475     }
2476   }
2477   dest_peer_info =
2478       GNUNET_CONTAINER_multihashmap_get (peers, &pi[size - 1].hashPubKey);
2479   if (NULL == dest_peer_info)
2480   {
2481     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2482                 "MESH:   Creating PeerInfo for destination.\n");
2483     dest_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
2484     dest_peer_info->id = GNUNET_PEER_intern (&pi[size - 1]);
2485     GNUNET_CONTAINER_multihashmap_put (peers, &pi[size - 1].hashPubKey,
2486                                        dest_peer_info,
2487                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
2488   }
2489   orig_peer_info = GNUNET_CONTAINER_multihashmap_get (peers, &pi->hashPubKey);
2490   if (NULL == orig_peer_info)
2491   {
2492     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2493                 "MESH:   Creating PeerInfo for origin.\n");
2494     orig_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
2495     orig_peer_info->id = GNUNET_PEER_intern (pi);
2496     GNUNET_CONTAINER_multihashmap_put (peers, &pi->hashPubKey, orig_peer_info,
2497                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
2498   }
2499   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating path...\n");
2500   path = path_new (size);
2501   own_pos = 0;
2502   for (i = 0; i < size; i++)
2503   {
2504     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2505                 "MESH:   ... adding %s\n",
2506                 GNUNET_i2s(&pi[i]));
2507     path->peers[i] = GNUNET_PEER_intern (&pi[i]);
2508     if (path->peers[i] == myid)
2509       own_pos = i;
2510   }
2511   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2512                 "MESH:   Own position: %u\n", own_pos);
2513   if (own_pos == 0)
2514   {
2515     /* cannot be self, must be 'not found' */
2516     /* create path: self not found in path through self */
2517     GNUNET_break_op (0);
2518     path_destroy (path);
2519     /* FIXME error. destroy tunnel? leave for timeout? */
2520     return 0;
2521   }
2522   tunnel_add_path (t, path, own_pos);
2523   if (own_pos == size - 1)
2524   {
2525     /* It is for us! Send ack. */
2526     struct GNUNET_MESH_TunnelNotification cmsg;
2527     struct MeshDataDescriptor *info;
2528     unsigned int j;
2529
2530     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2531                 "MESH:   It's for us!\n");
2532     path_add_to_origin (orig_peer_info, path);
2533     t->peers = GNUNET_CONTAINER_multihashmap_create(4);
2534     GNUNET_break (GNUNET_OK == GNUNET_CONTAINER_multihashmap_put (
2535         t->peers,
2536         &my_full_id.hashPubKey,
2537         peer_info_get(&my_full_id),
2538         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST));
2539     info = GNUNET_malloc (sizeof (struct MeshDataDescriptor));
2540     info->origin = &t->id;
2541     info->peer = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
2542     GNUNET_assert (NULL != info->peer);
2543     j = peer_info_transmit_slot(info->peer);
2544     info->handler_n = j;
2545     info->peer->types[j] = GNUNET_MESSAGE_TYPE_MESH_PATH_ACK;
2546     info->peer->infos[j] = info;
2547     info->peer->core_transmit[j] =
2548         GNUNET_CORE_notify_transmit_ready (core_handle, 0, 100,
2549                                            GNUNET_TIME_UNIT_FOREVER_REL, peer,
2550                                            sizeof (struct GNUNET_MESH_PathACK),
2551                                            &send_core_path_ack, info);
2552     cmsg.header.size = htons(sizeof(cmsg));
2553     cmsg.header.type = htons(GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
2554     GNUNET_PEER_resolve(t->id.oid, &cmsg.peer);
2555     cmsg.tunnel_id = htonl(t->local_tid);
2556     GNUNET_SERVER_notification_context_broadcast(nc, &cmsg.header, GNUNET_NO);
2557   }
2558   else
2559   {
2560     struct MeshPeerPath *path2;
2561
2562     /* It's for somebody else! Retransmit. */
2563     path2 = path_duplicate(path);
2564     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2565                 "MESH:   Retransmitting.\n");
2566     path_add_to_peer(dest_peer_info, path);
2567     path = path_duplicate(path2);
2568     path_add_to_origin(orig_peer_info, path2);
2569     send_create_path(dest_peer_info, path, t);
2570   }
2571   return GNUNET_OK;
2572 }
2573
2574
2575 /**
2576  * Core handler for path destruction
2577  *
2578  * @param cls closure
2579  * @param message message
2580  * @param peer peer identity this notification is about
2581  * @param atsi performance data
2582  * @param atsi_count number of records in 'atsi'
2583  *
2584  * @return GNUNET_OK to keep the connection open,
2585  *         GNUNET_SYSERR to close it (signal serious error)
2586  */
2587 static int
2588 handle_mesh_path_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
2589                           const struct GNUNET_MessageHeader *message,
2590                           const struct GNUNET_ATS_Information *atsi,
2591                           unsigned int atsi_count)
2592 {
2593   struct GNUNET_MESH_ManipulatePath *msg;
2594   struct GNUNET_PeerIdentity *pi;
2595   struct MeshPeerPath *path;
2596   struct MeshTunnel *t;
2597   unsigned int own_pos;
2598   unsigned int i;
2599   size_t size;
2600
2601   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2602               "MESH: Received a PATH DESTROY msg from %s\n",
2603               GNUNET_i2s(peer));
2604   size = ntohs (message->size);
2605   if (size < sizeof (struct GNUNET_MESH_ManipulatePath))
2606   {
2607     GNUNET_break_op (0);
2608     return GNUNET_OK;
2609   }
2610
2611   size -= sizeof (struct GNUNET_MESH_ManipulatePath);
2612   if (size % sizeof (struct GNUNET_PeerIdentity))
2613   {
2614     GNUNET_break_op (0);
2615     return GNUNET_OK;
2616   }
2617   size /= sizeof (struct GNUNET_PeerIdentity);
2618   if (size < 2)
2619   {
2620     GNUNET_break_op (0);
2621     return GNUNET_OK;
2622   }
2623   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2624               "MESH:     path has %u hops.\n",
2625               size);
2626
2627   msg = (struct GNUNET_MESH_ManipulatePath *) message;
2628   pi = (struct GNUNET_PeerIdentity *) &msg[1];
2629   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2630               "MESH:     path is for tunnel %s [%X].\n",
2631               GNUNET_i2s(pi),
2632               msg->tid);
2633   t = tunnel_get (pi, ntohl (msg->tid));
2634   if (NULL == t)
2635   {
2636     /* TODO notify back: we don't know this tunnel */
2637     GNUNET_break_op (0);
2638     return GNUNET_OK;
2639   }
2640   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   Creating path...\n");
2641   path = path_new (size);
2642   own_pos = 0;
2643   for (i = 0; i < size; i++)
2644   {
2645     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2646                 "MESH:   ... adding %s\n",
2647                 GNUNET_i2s(&pi[i]));
2648     path->peers[i] = GNUNET_PEER_intern (&pi[i]);
2649     if (path->peers[i] == myid)
2650       own_pos = i;
2651   }
2652   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2653                 "MESH:   Own position: %u\n", own_pos);
2654   if (own_pos < path->length - 1)
2655     send_message (message, &pi[own_pos + 1]);
2656   tunnel_delete_peer (t, path->peers[path->length - 1]);
2657   path_destroy (path);
2658   return GNUNET_OK;
2659 }
2660
2661
2662 /**
2663  * Core handler for notifications of broken paths
2664  *
2665  * @param cls closure
2666  * @param message message
2667  * @param peer peer identity this notification is about
2668  * @param atsi performance data
2669  * @param atsi_count number of records in 'atsi'
2670  *
2671  * @return GNUNET_OK to keep the connection open,
2672  *         GNUNET_SYSERR to close it (signal serious error)
2673  */
2674 static int
2675 handle_mesh_path_broken (void *cls, const struct GNUNET_PeerIdentity *peer,
2676                           const struct GNUNET_MessageHeader *message,
2677                           const struct GNUNET_ATS_Information *atsi,
2678                           unsigned int atsi_count)
2679 {
2680   struct GNUNET_MESH_PathBroken *msg;
2681   struct MeshTunnel *t;
2682
2683   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2684               "MESH: Received a PATH BROKEN msg from %s\n",
2685               GNUNET_i2s(peer));
2686   msg = (struct GNUNET_MESH_PathBroken *) message;
2687   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2688               "MESH:   regarding %s\n",
2689               GNUNET_i2s(&msg->peer1));
2690   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2691               "MESH:   regarding %s\n",
2692               GNUNET_i2s(&msg->peer2));
2693   t = tunnel_get(&msg->oid, ntohl (msg->tid));
2694   if (NULL == t)
2695   {
2696     GNUNET_break_op (0);
2697     return GNUNET_OK;
2698   }
2699   tunnel_notify_connection_broken(t,
2700                                   GNUNET_PEER_search(&msg->peer1),
2701                                   GNUNET_PEER_search(&msg->peer2));
2702   return GNUNET_OK;
2703
2704 }
2705
2706
2707 /**
2708  * Core handler for tunnel destruction
2709  *
2710  * @param cls closure
2711  * @param message message
2712  * @param peer peer identity this notification is about
2713  * @param atsi performance data
2714  * @param atsi_count number of records in 'atsi'
2715  *
2716  * @return GNUNET_OK to keep the connection open,
2717  *         GNUNET_SYSERR to close it (signal serious error)
2718  */
2719 static int
2720 handle_mesh_tunnel_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
2721                             const struct GNUNET_MessageHeader *message,
2722                             const struct GNUNET_ATS_Information *atsi,
2723                             unsigned int atsi_count)
2724 {
2725   struct GNUNET_MESH_TunnelDestroy *msg;
2726   struct MeshTunnel *t;
2727
2728   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2729               "MESH: Got a TUNNEL DESTROY packet from %s\n",
2730               GNUNET_i2s (peer));
2731   msg = (struct GNUNET_MESH_TunnelDestroy *) message;
2732   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2733               "MESH:   for tunnel %s [%u]\n",
2734               GNUNET_i2s (&msg->oid),
2735               ntohl (msg->tid));
2736   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2737   if (NULL == t)
2738   {
2739     /* Probably already got the message from another path,
2740      * destroyed the tunnel and retransmitted to children.
2741      * Safe to ignore.
2742      */
2743     return GNUNET_OK;
2744   }
2745   if (t->id.oid == myid)
2746   {
2747     GNUNET_break_op (0);
2748     return GNUNET_OK;
2749   }
2750   if (t->local_tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
2751   {
2752     /* Tunnel was incoming, notify clients */
2753     send_clients_tunnel_destroy (t);
2754   }
2755   tunnel_send_destroy (t);
2756   tunnel_destroy (t);
2757   return GNUNET_OK;
2758 }
2759
2760
2761 /**
2762  * Core handler for mesh network traffic going from the origin to a peer
2763  *
2764  * @param cls closure
2765  * @param peer peer identity this notification is about
2766  * @param message message
2767  * @param atsi performance data
2768  * @param atsi_count number of records in 'atsi'
2769  * @return GNUNET_OK to keep the connection open,
2770  *         GNUNET_SYSERR to close it (signal serious error)
2771  */
2772 static int
2773 handle_mesh_data_unicast (void *cls, const struct GNUNET_PeerIdentity *peer,
2774                           const struct GNUNET_MessageHeader *message,
2775                           const struct GNUNET_ATS_Information *atsi,
2776                           unsigned int atsi_count)
2777 {
2778   struct GNUNET_MESH_Unicast *msg;
2779   struct MeshTunnel *t;
2780   GNUNET_PEER_Id pid;
2781   size_t size;
2782
2783   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2784               "MESH: got a unicast packet from %s\n",
2785               GNUNET_i2s (peer));
2786   size = ntohs (message->size);
2787   if (size <
2788       sizeof (struct GNUNET_MESH_Unicast) +
2789       sizeof (struct GNUNET_MessageHeader))
2790   {
2791     GNUNET_break (0);
2792     return GNUNET_OK;
2793   }
2794   msg = (struct GNUNET_MESH_Unicast *) message;
2795   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2796               "MESH:  of type %u\n",
2797               ntohs (msg[1].header.type));
2798   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2799   if (NULL == t)
2800   {
2801     /* TODO notify back: we don't know this tunnel */
2802     GNUNET_break_op (0);
2803     return GNUNET_OK;
2804   }
2805   tunnel_reset_timeout (t);
2806   pid = GNUNET_PEER_search(&msg->destination);
2807   if (pid == myid)
2808   {
2809     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2810                 "MESH:   it's for us! sending to clients...\n");
2811     send_subscribed_clients (message, (struct GNUNET_MessageHeader *) &msg[1]);
2812     return GNUNET_OK;
2813   }
2814   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2815               "MESH:   not for us, retransmitting...\n");
2816   send_message (message, path_get_first_hop(t->tree, pid));
2817   return GNUNET_OK;
2818 }
2819
2820
2821 /**
2822  * Core handler for mesh network traffic going from the origin to all peers
2823  *
2824  * @param cls closure
2825  * @param message message
2826  * @param peer peer identity this notification is about
2827  * @param atsi performance data
2828  * @param atsi_count number of records in 'atsi'
2829  * @return GNUNET_OK to keep the connection open,
2830  *         GNUNET_SYSERR to close it (signal serious error)
2831  *
2832  * TODO: Check who we got this from, to validate route.
2833  */
2834 static int
2835 handle_mesh_data_multicast (void *cls, const struct GNUNET_PeerIdentity *peer,
2836                             const struct GNUNET_MessageHeader *message,
2837                             const struct GNUNET_ATS_Information *atsi,
2838                             unsigned int atsi_count)
2839 {
2840   struct GNUNET_MESH_Multicast *msg;
2841   struct MeshTunnel *t;
2842   size_t size;
2843
2844   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2845               "MESH: got a multicast packet from %s\n",
2846               GNUNET_i2s (peer));
2847   size = ntohs (message->size);
2848   if (sizeof (struct GNUNET_MESH_Multicast) +
2849       sizeof (struct GNUNET_MessageHeader) > size)
2850   {
2851     GNUNET_break_op (0);
2852     return GNUNET_OK;
2853   }
2854   msg = (struct GNUNET_MESH_Multicast *) message;
2855   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2856
2857   if (NULL == t)
2858   {
2859     /* TODO notify that we dont know that tunnel */
2860     GNUNET_break_op (0);
2861     return GNUNET_OK;
2862   }
2863   tunnel_reset_timeout (t);
2864
2865   /* Transmit to locally interested clients */
2866   if (NULL != t->peers &&
2867       GNUNET_CONTAINER_multihashmap_contains (t->peers, &my_full_id.hashPubKey))
2868   {
2869     send_subscribed_clients (message, &msg[1].header);
2870   }
2871   tunnel_send_multicast(t, message);
2872   return GNUNET_OK;
2873 }
2874
2875
2876 /**
2877  * Core handler for mesh network traffic toward the owner of a tunnel
2878  *
2879  * @param cls closure
2880  * @param message message
2881  * @param peer peer identity this notification is about
2882  * @param atsi performance data
2883  * @param atsi_count number of records in 'atsi'
2884  *
2885  * @return GNUNET_OK to keep the connection open,
2886  *         GNUNET_SYSERR to close it (signal serious error)
2887  */
2888 static int
2889 handle_mesh_data_to_orig (void *cls, const struct GNUNET_PeerIdentity *peer,
2890                           const struct GNUNET_MessageHeader *message,
2891                           const struct GNUNET_ATS_Information *atsi,
2892                           unsigned int atsi_count)
2893 {
2894   struct GNUNET_MESH_ToOrigin *msg;
2895   struct GNUNET_PeerIdentity id;
2896   struct MeshPeerInfo *peer_info;
2897   struct MeshTunnel *t;
2898   size_t size;
2899
2900   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2901               "MESH: got a ToOrigin packet from %s\n",
2902               GNUNET_i2s (peer));
2903   size = ntohs (message->size);
2904   if (size < sizeof (struct GNUNET_MESH_ToOrigin) +     /* Payload must be */
2905       sizeof (struct GNUNET_MessageHeader))     /* at least a header */
2906   {
2907     GNUNET_break_op (0);
2908     return GNUNET_OK;
2909   }
2910   msg = (struct GNUNET_MESH_ToOrigin *) message;
2911   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2912               "MESH:  of type %u\n",
2913               ntohs (msg[1].header.type));
2914   t = tunnel_get (&msg->oid, ntohl (msg->tid));
2915
2916   if (NULL == t)
2917   {
2918     /* TODO notify that we dont know this tunnel (whom)? */
2919     GNUNET_break_op (0);
2920     return GNUNET_OK;
2921   }
2922
2923   if (t->id.oid == myid)
2924   {
2925     char cbuf[size];
2926     struct GNUNET_MESH_ToOrigin *copy;
2927
2928     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2929                 "MESH:   it's for us! sending to clients...\n");
2930     if (NULL == t->client)
2931     {
2932       /* got data packet for ownerless tunnel */
2933       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2934                 "MESH:   no clients!\n");
2935       GNUNET_break_op (0);
2936       return GNUNET_OK;
2937     }
2938     /* TODO signature verification */
2939     memcpy (cbuf, message, size);
2940     copy = (struct GNUNET_MESH_ToOrigin *) cbuf;
2941     copy->tid = htonl (t->local_tid);
2942     GNUNET_SERVER_notification_context_unicast (nc,
2943                                                 t->client->handle,
2944                                                 &copy->header,
2945                                                 GNUNET_YES);
2946     return GNUNET_OK;
2947   }
2948   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2949               "MESH:   not for us, retransmitting...\n");
2950
2951   peer_info = peer_info_get (&msg->oid);
2952   if (NULL == peer_info)
2953   {
2954     /* unknown origin of tunnel */
2955     GNUNET_break (0);
2956     return GNUNET_OK;
2957   }
2958   GNUNET_PEER_resolve (t->tree->me->parent->peer, &id);
2959   send_message (message, &id);
2960
2961   return GNUNET_OK;
2962 }
2963
2964
2965 /**
2966  * Core handler for path ACKs
2967  *
2968  * @param cls closure
2969  * @param message message
2970  * @param peer peer identity this notification is about
2971  * @param atsi performance data
2972  * @param atsi_count number of records in 'atsi'
2973  *
2974  * @return GNUNET_OK to keep the connection open,
2975  *         GNUNET_SYSERR to close it (signal serious error)
2976  */
2977 static int
2978 handle_mesh_path_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
2979                       const struct GNUNET_MessageHeader *message,
2980                       const struct GNUNET_ATS_Information *atsi,
2981                       unsigned int atsi_count)
2982 {
2983   struct GNUNET_MESH_PathACK *msg;
2984   struct GNUNET_PeerIdentity id;
2985   struct MeshTunnelTreeNode *n;
2986   struct MeshPeerInfo *peer_info;
2987   struct MeshTunnel *t;
2988
2989   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2990               "MESH: Received a path ACK msg [%s]\n",
2991               GNUNET_i2s(&my_full_id));
2992   msg = (struct GNUNET_MESH_PathACK *) message;
2993   t = tunnel_get (&msg->oid, msg->tid);
2994   if (NULL == t)
2995   {
2996     /* TODO notify that we don't know the tunnel */
2997     return GNUNET_OK;
2998   }
2999
3000   /* Message for us? */
3001   if (0 == memcmp (&msg->oid, &my_full_id, sizeof (struct GNUNET_PeerIdentity)))
3002   {
3003     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   It's for us!\n");
3004     if (NULL == t->client)
3005     {
3006       GNUNET_break_op (0);
3007       return GNUNET_OK;
3008     }
3009     if (NULL != t->dht_get_type)
3010     {
3011       GNUNET_DHT_get_stop (t->dht_get_type);
3012       t->dht_get_type = NULL;
3013     }
3014     peer_info = peer_info_get (&msg->peer_id);
3015     n = tree_find_peer(t->tree->root, peer_info->id);
3016     if (NULL == n)
3017     {
3018       GNUNET_break_op (0);
3019       return GNUNET_OK;
3020     }
3021     n->status = MESH_PEER_READY;
3022     send_client_peer_connected(t, peer_info->id);
3023     return GNUNET_OK;
3024   }
3025   
3026   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3027               "MESH:   not for us, retransmitting...\n");
3028   GNUNET_PEER_resolve(t->tree->me->parent->peer, &id);
3029   peer_info = peer_info_get (&msg->oid);
3030   if (NULL == peer_info)
3031   {
3032     /* If we know the tunnel, we should DEFINITELY know the peer */
3033     GNUNET_break (0);
3034     return GNUNET_OK;
3035   }
3036   send_message (message, &id);
3037   return GNUNET_OK;
3038 }
3039
3040
3041 /**
3042  * Functions to handle messages from core
3043  */
3044 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
3045   {&handle_mesh_path_create, GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE, 0},
3046   {&handle_mesh_path_destroy, GNUNET_MESSAGE_TYPE_MESH_PATH_DESTROY, 0},
3047   {&handle_mesh_path_broken, GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN,
3048     sizeof (struct GNUNET_MESH_PathBroken)},
3049   {&handle_mesh_tunnel_destroy, GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY, 0},
3050   {&handle_mesh_data_unicast, GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
3051   {&handle_mesh_data_multicast, GNUNET_MESSAGE_TYPE_MESH_MULTICAST, 0},
3052   {&handle_mesh_data_to_orig, GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
3053   {&handle_mesh_path_ack, GNUNET_MESSAGE_TYPE_MESH_PATH_ACK,
3054     sizeof (struct GNUNET_MESH_PathACK)},
3055   {NULL, 0, 0}
3056 };
3057
3058
3059
3060 /******************************************************************************/
3061 /****************       MESH LOCAL HANDLER HELPERS      ***********************/
3062 /******************************************************************************/
3063
3064 /**
3065  * deregister_app: iterator for removing each application registered by a client
3066  * 
3067  * @param cls closure
3068  * @param key the hash of the application id (used to access the hashmap)
3069  * @param value the value stored at the key (client)
3070  * 
3071  * @return GNUNET_OK on success
3072  */
3073 static int
3074 deregister_app (void *cls, const GNUNET_HashCode * key, void *value)
3075 {
3076   GNUNET_break (GNUNET_YES == 
3077       GNUNET_CONTAINER_multihashmap_remove (applications, key, value));
3078   return GNUNET_OK;
3079 }
3080
3081 #if LATER
3082 /**
3083  * notify_client_connection_failure: notify a client that the connection to the
3084  * requested remote peer is not possible (for instance, no route found)
3085  * Function called when the socket is ready to queue more data. "buf" will be
3086  * NULL and "size" zero if the socket was closed for writing in the meantime.
3087  *
3088  * @param cls closure
3089  * @param size number of bytes available in buf
3090  * @param buf where the callee should write the message
3091  * @return number of bytes written to buf
3092  */
3093 static size_t
3094 notify_client_connection_failure (void *cls, size_t size, void *buf)
3095 {
3096   int size_needed;
3097   struct MeshPeerInfo *peer_info;
3098   struct GNUNET_MESH_PeerControl *msg;
3099   struct GNUNET_PeerIdentity id;
3100
3101   if (0 == size && NULL == buf)
3102   {
3103     // TODO retry? cancel?
3104     return 0;
3105   }
3106
3107   size_needed = sizeof (struct GNUNET_MESH_PeerControl);
3108   peer_info = (struct MeshPeerInfo *) cls;
3109   msg = (struct GNUNET_MESH_PeerControl *) buf;
3110   msg->header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
3111   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DISCONNECTED);
3112 //     msg->tunnel_id = htonl(peer_info->t->tid);
3113   GNUNET_PEER_resolve (peer_info->id, &id);
3114   memcpy (&msg->peer, &id, sizeof (struct GNUNET_PeerIdentity));
3115
3116   return size_needed;
3117 }
3118 #endif
3119
3120
3121 /**
3122  * Send keepalive packets for a peer
3123  *
3124  * @param cls Closure (tunnel for which to send the keepalive).
3125  * @param tc Notification context.
3126  *
3127  * TODO: implement explicit multicast keepalive?
3128  */
3129 static void
3130 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
3131 {
3132   struct MeshTunnel *t = cls;
3133   struct GNUNET_MessageHeader *payload;
3134   struct GNUNET_MESH_Multicast *msg;
3135   size_t size = sizeof(struct GNUNET_MESH_Multicast) +
3136                 sizeof(struct GNUNET_MessageHeader);
3137   char cbuf[size];
3138
3139   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
3140   {
3141     return;
3142   }
3143   t->path_refresh_task = GNUNET_SCHEDULER_NO_TASK;
3144
3145   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3146               "MESH: sending keepalive for tunnel %d\n",
3147               t->id.tid);
3148
3149   msg = (struct GNUNET_MESH_Multicast *) cbuf;
3150   msg->header.size = htons (size);
3151   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_MULTICAST);
3152   msg->oid = my_full_id;
3153   msg->tid = htonl(t->id.tid);
3154   payload = (struct GNUNET_MessageHeader *) &msg[1];
3155   payload->size = htons (sizeof(struct GNUNET_MessageHeader));
3156   payload->type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE);
3157   tunnel_send_multicast (t, &msg->header);
3158
3159   t->path_refresh_task =
3160       GNUNET_SCHEDULER_add_delayed (t->tree->refresh, &path_refresh, t);
3161   return;
3162 }
3163
3164
3165 /**
3166  * Function to process paths received for a new peer addition. The recorded
3167  * paths form the initial tunnel, which can be optimized later.
3168  * Called on each result obtained for the DHT search.
3169  *
3170  * @param cls closure
3171  * @param exp when will this value expire
3172  * @param key key of the result
3173  * @param type type of the result
3174  * @param size number of bytes in data
3175  * @param data pointer to the result data
3176  *
3177  * TODO: re-issue the request after certain time? cancel after X results?
3178  */
3179 static void
3180 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
3181                     const GNUNET_HashCode * key,
3182                     const struct GNUNET_PeerIdentity *get_path,
3183                     unsigned int get_path_length,
3184                     const struct GNUNET_PeerIdentity *put_path,
3185                     unsigned int put_path_length,
3186                     enum GNUNET_BLOCK_Type type, size_t size, const void *data)
3187 {
3188   struct MeshPathInfo *path_info = cls;
3189   struct MeshPeerPath *p;
3190   struct GNUNET_PeerIdentity pi;
3191   int i;
3192
3193   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
3194              "MESH: Got results from DHT!\n");
3195   GNUNET_PEER_resolve (path_info->peer->id, &pi);
3196   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
3197              "MESH:   for %s\n",
3198              GNUNET_i2s(&pi));
3199 //   GNUNET_DHT_get_stop(path_info->peer->dhtget);
3200 //   path_info->peer->dhtget = NULL;
3201
3202   p = path_build_from_dht (get_path, get_path_length,
3203                            put_path, put_path_length);
3204   path_add_to_peer (path_info->peer, p);
3205   for (i = 0; i < path_info->peer->ntunnels; i++)
3206   {
3207     tunnel_add_peer (path_info->peer->tunnels[i], path_info->peer);
3208     peer_info_connect(path_info->peer, path_info->t);
3209   }
3210 //   GNUNET_free (path_info);
3211
3212   return;
3213 }
3214
3215
3216 /**
3217  * Function to process paths received for a new peer addition. The recorded
3218  * paths form the initial tunnel, which can be optimized later.
3219  * Called on each result obtained for the DHT search.
3220  *
3221  * @param cls closure
3222  * @param exp when will this value expire
3223  * @param key key of the result
3224  * @param type type of the result
3225  * @param size number of bytes in data
3226  * @param data pointer to the result data
3227  */
3228 static void
3229 dht_get_type_handler (void *cls, struct GNUNET_TIME_Absolute exp,
3230                       const GNUNET_HashCode * key,
3231                       const struct GNUNET_PeerIdentity *get_path,
3232                       unsigned int get_path_length,
3233                       const struct GNUNET_PeerIdentity *put_path,
3234                       unsigned int put_path_length,
3235                       enum GNUNET_BLOCK_Type type, size_t size,
3236                       const void *data)
3237 {
3238   const struct GNUNET_PeerIdentity *pi = data;
3239   struct MeshTunnel *t = cls;
3240   struct MeshPeerInfo *peer_info;
3241   struct MeshPeerPath *p;
3242
3243   if (size != sizeof (struct GNUNET_PeerIdentity))
3244   {
3245     GNUNET_break_op (0);
3246     return;
3247   }
3248   GNUNET_assert (NULL != t->client);
3249   peer_info = peer_info_get (pi);
3250   (void) GNUNET_CONTAINER_multihashmap_put (
3251       t->peers,
3252       &pi->hashPubKey,
3253       peer_info,
3254       GNUNET_CONTAINER_MULTIHASHMAPOPTION_REPLACE);
3255
3256   p = path_build_from_dht (get_path, get_path_length,
3257                            put_path, put_path_length);
3258   path_add_to_peer (peer_info, p);
3259   tunnel_add_peer(t, peer_info);
3260   peer_info_connect (peer_info, t);
3261 }
3262
3263
3264 /******************************************************************************/
3265 /*********************       MESH LOCAL HANDLES      **************************/
3266 /******************************************************************************/
3267
3268
3269 /**
3270  * Handler for client disconnection
3271  *
3272  * @param cls closure
3273  * @param client identification of the client; NULL
3274  *        for the last call when the server is destroyed
3275  */
3276 static void
3277 handle_local_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
3278 {
3279   struct MeshClient *c;
3280   struct MeshClient *next;
3281
3282   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: client disconnected\n");
3283   if (client == NULL)
3284   {
3285     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    (SERVER DOWN)\n");
3286     return;
3287   }
3288   c = clients;
3289   while (NULL != c)
3290   {
3291     if (c->handle != client && NULL != client)
3292     {
3293       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    ... searching\n");
3294       c = c->next;
3295       continue;
3296     }
3297     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: matching client found\n");
3298     GNUNET_SERVER_client_drop (c->handle);
3299     if (NULL != c->tunnels)
3300     {
3301       GNUNET_CONTAINER_multihashmap_iterate (c->tunnels,
3302                                              &tunnel_destroy_iterator,
3303                                              c);
3304       GNUNET_CONTAINER_multihashmap_destroy (c->tunnels);
3305     }
3306
3307     /* deregister clients applications */
3308     if (NULL != c->apps)
3309     {
3310       GNUNET_CONTAINER_multihashmap_iterate (c->apps, &deregister_app, NULL);
3311       GNUNET_CONTAINER_multihashmap_destroy (c->apps);
3312     }
3313     if (0 == GNUNET_CONTAINER_multihashmap_size (applications) &&
3314         GNUNET_SCHEDULER_NO_TASK != announce_applications_task)
3315     {
3316       GNUNET_SCHEDULER_cancel (announce_applications_task);
3317       announce_applications_task = GNUNET_SCHEDULER_NO_TASK;
3318     }
3319     if (NULL != c->types)
3320       GNUNET_CONTAINER_multihashmap_destroy (c->types);
3321     next = c->next;
3322     GNUNET_CONTAINER_DLL_remove (clients, clients_tail, c);
3323     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   CLIENT FREE at %p\n", c);
3324     GNUNET_free (c);
3325     c = next;
3326   }
3327
3328   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:    done!\n");
3329   return;
3330 }
3331
3332
3333 /**
3334  * Handler for new clients
3335  *
3336  * @param cls closure
3337  * @param client identification of the client
3338  * @param message the actual message, which includes messages the client wants
3339  */
3340 static void
3341 handle_local_new_client (void *cls, struct GNUNET_SERVER_Client *client,
3342                          const struct GNUNET_MessageHeader *message)
3343 {
3344   struct GNUNET_MESH_ClientConnect *cc_msg;
3345   struct MeshClient *c;
3346   GNUNET_MESH_ApplicationType *a;
3347   unsigned int size;
3348   uint16_t ntypes;
3349   uint16_t *t;
3350   uint16_t napps;
3351   uint16_t i;
3352
3353   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new client connected\n");
3354   /* Check data sanity */
3355   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
3356   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
3357   ntypes = ntohs (cc_msg->types);
3358   napps = ntohs (cc_msg->applications);
3359   if (size !=
3360       ntypes * sizeof (uint16_t) + napps * sizeof (GNUNET_MESH_ApplicationType))
3361   {
3362     GNUNET_break (0);
3363     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3364     return;
3365   }
3366
3367   /* Create new client structure */
3368   c = GNUNET_malloc (sizeof (struct MeshClient));
3369 #if MESH_DEBUG
3370   c->id = next_client_id++;
3371 #endif
3372   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   CLIENT NEW %u at %p\n", c->id, c);
3373   c->handle = client;
3374   GNUNET_SERVER_client_keep (client);
3375   a = (GNUNET_MESH_ApplicationType *) &cc_msg[1];
3376   if (napps > 0)
3377   {
3378     GNUNET_MESH_ApplicationType at;
3379     GNUNET_HashCode hc;
3380
3381     c->apps = GNUNET_CONTAINER_multihashmap_create (napps);
3382     for (i = 0; i < napps; i++)
3383     {
3384       at = ntohl (a[i]);
3385       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   app type: %u\n", at);
3386       GNUNET_CRYPTO_hash (&at, sizeof (at), &hc);
3387       /* store in clients hashmap */
3388       GNUNET_CONTAINER_multihashmap_put (c->apps, &hc, c,
3389                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3390       /* store in global hashmap, for announcements */
3391       GNUNET_CONTAINER_multihashmap_put (applications, &hc, c,
3392                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3393     }
3394     if (GNUNET_SCHEDULER_NO_TASK == announce_applications_task)
3395       announce_applications_task =
3396           GNUNET_SCHEDULER_add_now (&announce_applications, NULL);
3397
3398   }
3399   if (ntypes > 0)
3400   {
3401     uint16_t u16;
3402     GNUNET_HashCode hc;
3403
3404     t = (uint16_t *) & a[napps];
3405     c->types = GNUNET_CONTAINER_multihashmap_create (ntypes);
3406     for (i = 0; i < ntypes; i++)
3407     {
3408       u16 = ntohs (t[i]);
3409       GNUNET_CRYPTO_hash (&u16, sizeof (u16), &hc);
3410
3411       /* store in clients hashmap */
3412       GNUNET_CONTAINER_multihashmap_put (c->types, &hc, c,
3413                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3414       /* store in global hashmap */
3415       GNUNET_CONTAINER_multihashmap_put (types, &hc, c,
3416                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
3417     }
3418   }
3419   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3420               "MESH:  client has %u+%u subscriptions\n", napps, ntypes);
3421
3422   GNUNET_CONTAINER_DLL_insert (clients, clients_tail, c);
3423   c->tunnels = GNUNET_CONTAINER_multihashmap_create (32);
3424   GNUNET_SERVER_notification_context_add (nc, client);
3425
3426   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3427 #if MESH_DEBUG
3428   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new client processed\n");
3429 #endif
3430 }
3431
3432
3433 /**
3434  * Handler for requests of new tunnels
3435  *
3436  * @param cls closure
3437  * @param client identification of the client
3438  * @param message the actual message
3439  */
3440 static void
3441 handle_local_tunnel_create (void *cls, struct GNUNET_SERVER_Client *client,
3442                             const struct GNUNET_MessageHeader *message)
3443 {
3444   struct GNUNET_MESH_TunnelMessage *t_msg;
3445   struct MeshTunnel *t;
3446   struct MeshClient *c;
3447   GNUNET_HashCode hash;
3448
3449   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: new tunnel requested\n");
3450
3451   /* Sanity check for client registration */
3452   if (NULL == (c = client_get (client)))
3453   {
3454     GNUNET_break (0);
3455     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3456     return;
3457   }
3458 #if MESH_DEBUG
3459   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
3460 #endif
3461
3462   /* Message sanity check */
3463   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
3464   {
3465     GNUNET_break (0);
3466     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3467     return;
3468   }
3469
3470   t_msg = (struct GNUNET_MESH_TunnelMessage *) message;
3471   /* Sanity check for tunnel numbering */
3472   if (0 == (ntohl (t_msg->tunnel_id) & GNUNET_MESH_LOCAL_TUNNEL_ID_CLI))
3473   {
3474     GNUNET_break (0);
3475     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3476     return;
3477   }
3478   /* Sanity check for duplicate tunnel IDs */
3479   if (NULL != tunnel_get_by_local_id (c, ntohl (t_msg->tunnel_id)))
3480   {
3481     GNUNET_break (0);
3482     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3483     return;
3484   }
3485
3486   t = GNUNET_malloc (sizeof (struct MeshTunnel));
3487   while (NULL != tunnel_get_by_pi (myid, next_tid))
3488     next_tid = (next_tid + 1) & ~GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
3489   t->id.tid = next_tid++;
3490   t->id.oid = myid;
3491   t->local_tid = ntohl (t_msg->tunnel_id);
3492 #if MESH_DEBUG
3493   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3494               "MESH: CREATED TUNNEL %s [%x] (%x)\n",
3495               GNUNET_i2s (&my_full_id),
3496               t->id.tid,
3497               t->local_tid);
3498 #endif
3499   t->client = c;
3500   t->peers = GNUNET_CONTAINER_multihashmap_create (32);
3501
3502   GNUNET_CRYPTO_hash (&t->local_tid, sizeof (MESH_TunnelNumber), &hash);
3503   if (GNUNET_OK !=
3504       GNUNET_CONTAINER_multihashmap_put (
3505         c->tunnels, &hash, t,
3506         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
3507   {
3508     GNUNET_break (0);
3509     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3510     return;
3511   }
3512
3513   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
3514   if (GNUNET_OK !=
3515       GNUNET_CONTAINER_multihashmap_put (
3516         tunnels, &hash, t,
3517         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
3518   {
3519     GNUNET_break (0);
3520     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3521     return;
3522   }
3523   t->tree = tree_new (t, myid);
3524   t->tree->refresh = REFRESH_PATH_TIME;
3525   t->tree->root->status = MESH_PEER_READY;
3526   t->tree->me = t->tree->root;
3527
3528   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3529   return;
3530 }
3531
3532
3533 /**
3534  * Handler for requests of deleting tunnels
3535  *
3536  * @param cls closure
3537  * @param client identification of the client
3538  * @param message the actual message
3539  */
3540 static void
3541 handle_local_tunnel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
3542                              const struct GNUNET_MessageHeader *message)
3543 {
3544   struct GNUNET_MESH_TunnelMessage *tunnel_msg;
3545   struct MeshClient *c;
3546   struct MeshTunnel *t;
3547   MESH_TunnelNumber tid;
3548   GNUNET_HashCode hash;
3549
3550   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3551               "MESH: Got a DESTROY TUNNEL from client!\n");
3552
3553   /* Sanity check for client registration */
3554   if (NULL == (c = client_get (client)))
3555   {
3556     GNUNET_break (0);
3557     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3558     return;
3559   }
3560   /* Message sanity check */
3561   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
3562   {
3563     GNUNET_break (0);
3564     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3565     return;
3566   }
3567 #if MESH_DEBUG
3568   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   by client %u\n", c->id);
3569 #endif
3570   tunnel_msg = (struct GNUNET_MESH_TunnelMessage *) message;
3571
3572   /* Retrieve tunnel */
3573   tid = ntohl (tunnel_msg->tunnel_id);
3574
3575   /* Remove from local id hashmap */
3576   GNUNET_CRYPTO_hash (&tid, sizeof (MESH_TunnelNumber), &hash);
3577   t = GNUNET_CONTAINER_multihashmap_get (c->tunnels, &hash);
3578   GNUNET_CONTAINER_multihashmap_remove (c->tunnels, &hash, t);
3579
3580   t->client = NULL;
3581   tunnel_send_destroy (t);
3582   tunnel_destroy(t);
3583   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3584   return;
3585 }
3586
3587
3588 /**
3589  * Handler for connection requests to new peers
3590  *
3591  * @param cls closure
3592  * @param client identification of the client
3593  * @param message the actual message (PeerControl)
3594  */
3595 static void
3596 handle_local_connect_add (void *cls, struct GNUNET_SERVER_Client *client,
3597                           const struct GNUNET_MessageHeader *message)
3598 {
3599   struct GNUNET_MESH_PeerControl *peer_msg;
3600   struct MeshPeerInfo *peer_info;
3601   struct MeshClient *c;
3602   struct MeshTunnel *t;
3603   MESH_TunnelNumber tid;
3604
3605   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH: Got connection request\n");
3606   /* Sanity check for client registration */
3607   if (NULL == (c = client_get (client)))
3608   {
3609     GNUNET_break (0);
3610     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3611     return;
3612   }
3613
3614   peer_msg = (struct GNUNET_MESH_PeerControl *) message;
3615   /* Sanity check for message size */
3616   if (sizeof (struct GNUNET_MESH_PeerControl) != ntohs (peer_msg->header.size))
3617   {
3618     GNUNET_break (0);
3619     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3620     return;
3621   }
3622
3623   /* Tunnel exists? */
3624   tid = ntohl (peer_msg->tunnel_id);
3625   t = tunnel_get_by_local_id (c, tid);
3626   if (NULL == t)
3627   {
3628     GNUNET_break (0);
3629     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3630     return;
3631   }
3632
3633   /* Does client own tunnel? */
3634   if (t->client->handle != client)
3635   {
3636     GNUNET_break (0);
3637     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3638     return;
3639   }
3640   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "MESH:      for %s\n",
3641              GNUNET_i2s(&peer_msg->peer));
3642   peer_info = peer_info_get (&peer_msg->peer);
3643
3644   tunnel_add_peer(t, peer_info);
3645   peer_info_connect(peer_info, t);
3646
3647   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3648   return;
3649 }
3650
3651
3652 /**
3653  * Handler for disconnection requests of peers in a tunnel
3654  *
3655  * @param cls closure
3656  * @param client identification of the client
3657  * @param message the actual message (PeerControl)
3658  */
3659 static void
3660 handle_local_connect_del (void *cls, struct GNUNET_SERVER_Client *client,
3661                           const struct GNUNET_MessageHeader *message)
3662 {
3663   struct GNUNET_MESH_PeerControl *peer_msg;
3664   struct MeshPeerInfo *peer_info;
3665   struct MeshClient *c;
3666   struct MeshTunnel *t;
3667   MESH_TunnelNumber tid;
3668
3669   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Got a PEER DEL request\n");
3670   /* Sanity check for client registration */
3671   if (NULL == (c = client_get (client)))
3672   {
3673     GNUNET_break (0);
3674     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3675     return;
3676   }
3677   peer_msg = (struct GNUNET_MESH_PeerControl *) message;
3678   /* Sanity check for message size */
3679   if (sizeof (struct GNUNET_MESH_PeerControl) != ntohs (peer_msg->header.size))
3680   {
3681     GNUNET_break (0);
3682     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3683     return;
3684   }
3685
3686   /* Tunnel exists? */
3687   tid = ntohl (peer_msg->tunnel_id);
3688   t = tunnel_get_by_local_id (c, tid);
3689   if (NULL == t)
3690   {
3691     GNUNET_break (0);
3692     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3693     return;
3694   }
3695   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:   on tunnel %X\n", t->id.tid);
3696
3697   /* Does client own tunnel? */
3698   if (t->client->handle != client)
3699   {
3700     GNUNET_break (0);
3701     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3702     return;
3703   }
3704
3705   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3706               "MESH:   for peer %s\n",
3707               GNUNET_i2s(&peer_msg->peer));
3708   /* Is the peer in the tunnel? */
3709   peer_info = GNUNET_CONTAINER_multihashmap_get(t->peers,
3710                                                 &peer_msg->peer.hashPubKey);
3711   if (NULL == peer_info)
3712   {
3713     GNUNET_break (0);
3714     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3715     return;
3716   }
3717
3718   /* Ok, delete peer from tunnel */
3719   GNUNET_CONTAINER_multihashmap_remove_all (t->peers,
3720                                             &peer_msg->peer.hashPubKey);
3721
3722   send_destroy_path (t, peer_info->id);
3723   tunnel_delete_peer(t, peer_info->id);
3724   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3725   return;
3726 }
3727
3728
3729 /**
3730  * Handler for connection requests to new peers by type
3731  *
3732  * @param cls closure
3733  * @param client identification of the client
3734  * @param message the actual message (ConnectPeerByType)
3735  */
3736 static void
3737 handle_local_connect_by_type (void *cls, struct GNUNET_SERVER_Client *client,
3738                               const struct GNUNET_MessageHeader *message)
3739 {
3740   struct GNUNET_MESH_ConnectPeerByType *connect_msg;
3741   struct MeshClient *c;
3742   struct MeshTunnel *t;
3743   GNUNET_HashCode hash;
3744   MESH_TunnelNumber tid;
3745
3746   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: got connect by type request\n");
3747   /* Sanity check for client registration */
3748   if (NULL == (c = client_get (client)))
3749   {
3750     GNUNET_break (0);
3751     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3752     return;
3753   }
3754
3755   connect_msg = (struct GNUNET_MESH_ConnectPeerByType *) message;
3756   /* Sanity check for message size */
3757   if (sizeof (struct GNUNET_MESH_ConnectPeerByType) !=
3758       ntohs (connect_msg->header.size))
3759   {
3760     GNUNET_break (0);
3761     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3762     return;
3763   }
3764
3765   /* Tunnel exists? */
3766   tid = ntohl (connect_msg->tunnel_id);
3767   t = tunnel_get_by_local_id (c, tid);
3768   if (NULL == t)
3769   {
3770     GNUNET_break (0);
3771     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3772     return;
3773   }
3774
3775   /* Does client own tunnel? */
3776   if (t->client->handle != client)
3777   {
3778     GNUNET_break (0);
3779     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3780     return;
3781   }
3782
3783   /* Do WE have the service? */
3784   t->type = ntohl (connect_msg->type);
3785   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  type requested: %u\n", t->type);
3786   GNUNET_CRYPTO_hash (&t->type, sizeof (GNUNET_MESH_ApplicationType), &hash);
3787   if (GNUNET_CONTAINER_multihashmap_contains (applications, &hash) ==
3788       GNUNET_YES)
3789   {
3790     /* Yes! Fast forward, add ourselves to the tunnel and send the
3791      * good news to the client
3792      */
3793     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  available locally\n");
3794     GNUNET_CONTAINER_multihashmap_put (t->peers, &my_full_id.hashPubKey,
3795                                        peer_info_get (&my_full_id),
3796                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
3797
3798     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  notifying client\n");
3799     send_client_peer_connected(t, myid);
3800     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  Done\n");
3801     GNUNET_SERVER_receive_done (client, GNUNET_OK);
3802     return;
3803   }
3804   /* Ok, lets find a peer offering the service */
3805   if (NULL != t->dht_get_type)
3806   {
3807     GNUNET_DHT_get_stop (t->dht_get_type);
3808   }
3809   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:  looking in DHT for %s\n",
3810               GNUNET_h2s (&hash));
3811   t->dht_get_type =
3812       GNUNET_DHT_get_start (dht_handle,
3813                             GNUNET_TIME_UNIT_FOREVER_REL,
3814                             GNUNET_BLOCK_TYPE_TEST,
3815                             &hash,
3816                             10U,
3817                             GNUNET_DHT_RO_RECORD_ROUTE |
3818                               GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
3819                             NULL, 0,
3820                             &dht_get_type_handler, t);
3821
3822   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3823   return;
3824 }
3825
3826
3827 /**
3828  * Handler for client traffic directed to one peer
3829  *
3830  * @param cls closure
3831  * @param client identification of the client
3832  * @param message the actual message
3833  */
3834 static void
3835 handle_local_unicast (void *cls, struct GNUNET_SERVER_Client *client,
3836                       const struct GNUNET_MessageHeader *message)
3837 {
3838   struct MeshClient *c;
3839   struct MeshTunnel *t;
3840   struct MeshPeerInfo *pi;
3841   struct GNUNET_MESH_Unicast *data_msg;
3842   MESH_TunnelNumber tid;
3843   size_t size;
3844
3845   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3846               "MESH: Got a unicast request from a client!\n");
3847
3848   /* Sanity check for client registration */
3849   if (NULL == (c = client_get (client)))
3850   {
3851     GNUNET_break (0);
3852     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3853     return;
3854   }
3855   data_msg = (struct GNUNET_MESH_Unicast *) message;
3856   /* Sanity check for message size */
3857   size = ntohs (message->size);
3858   if (sizeof (struct GNUNET_MESH_Unicast) +
3859       sizeof (struct GNUNET_MessageHeader) > size)
3860   {
3861     GNUNET_break (0);
3862     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3863     return;
3864   }
3865
3866   /* Tunnel exists? */
3867   tid = ntohl (data_msg->tid);
3868   t = tunnel_get_by_local_id (c, tid);
3869   if (NULL == t)
3870   {
3871     GNUNET_break (0);
3872     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3873     return;
3874   }
3875
3876   /*  Is it a local tunnel? Then, does client own the tunnel? */
3877   if (NULL != t->client &&
3878       NULL != t->client->handle &&
3879       t->client->handle != client)
3880   {
3881     GNUNET_break (0);
3882     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3883     return;
3884   }
3885
3886   pi = GNUNET_CONTAINER_multihashmap_get (t->peers,
3887                                           &data_msg->destination.hashPubKey);
3888   /* Is the selected peer in the tunnel? */
3889   if (NULL == pi)
3890   {
3891     GNUNET_break (0);
3892     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3893     return;
3894   }
3895
3896   /* Ok, everything is correct, send the message
3897    * (pretend we got it from a mesh peer)
3898    */
3899   {
3900     char buf[ntohs (message->size)];
3901     struct GNUNET_MESH_Unicast *copy;
3902
3903     /* Work around const limitation */
3904     copy = (struct GNUNET_MESH_Unicast *) buf;
3905     memcpy (buf, data_msg, size);
3906     copy->oid = my_full_id;
3907     copy->tid = htonl (t->id.tid);
3908     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3909                 "MESH:   calling generic handler...\n");
3910     handle_mesh_data_unicast (NULL, &my_full_id, &copy->header, NULL, 0);
3911   }
3912   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3913   return;
3914 }
3915
3916
3917 /**
3918  * Handler for client traffic directed to the origin
3919  *
3920  * @param cls closure
3921  * @param client identification of the client
3922  * @param message the actual message
3923  */
3924 static void
3925 handle_local_to_origin (void *cls, struct GNUNET_SERVER_Client *client,
3926                         const struct GNUNET_MessageHeader *message)
3927 {
3928   struct GNUNET_MESH_ToOrigin *data_msg;
3929   struct GNUNET_PeerIdentity id;
3930   struct MeshClient *c;
3931   struct MeshTunnel *t;
3932   MESH_TunnelNumber tid;
3933   size_t size;
3934
3935   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3936               "MESH: Got a ToOrigin request from a client!\n");
3937
3938   /* Sanity check for client registration */
3939   if (NULL == (c = client_get (client)))
3940   {
3941     GNUNET_break (0);
3942     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3943     return;
3944   }
3945   data_msg = (struct GNUNET_MESH_ToOrigin *) message;
3946   /* Sanity check for message size */
3947   size = ntohs (message->size);
3948   if (sizeof (struct GNUNET_MESH_ToOrigin) +
3949       sizeof (struct GNUNET_MessageHeader) > size)
3950   {
3951     GNUNET_break (0);
3952     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3953     return;
3954   }
3955
3956   /* Tunnel exists? */
3957   tid = ntohl (data_msg->tid);
3958   if (tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
3959   {
3960     GNUNET_break (0);
3961     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3962     return;
3963   }
3964   t = tunnel_get_by_local_id (c, tid);
3965   if (NULL == t)
3966   {
3967     GNUNET_break (0);
3968     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3969     return;
3970   }
3971
3972   /*  It shouldn't be a local tunnel.  */
3973   if (NULL != t->client)
3974   {
3975     GNUNET_break (0);
3976     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
3977     return;
3978   }
3979   GNUNET_PEER_resolve(t->id.oid, &id);
3980
3981   /* Ok, everything is correct, send the message
3982    * (pretend we got it from a mesh peer)
3983    */
3984   {
3985     char buf[ntohs (message->size)];
3986     struct GNUNET_MESH_ToOrigin *copy;
3987
3988     /* Work around const limitation */
3989     copy = (struct GNUNET_MESH_ToOrigin *) buf;
3990     memcpy (buf, data_msg, size);
3991     copy->oid = id;
3992     copy->tid = htonl (t->id.tid);
3993     copy->sender = my_full_id;
3994     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3995                 "MESH:   calling generic handler...\n");
3996     handle_mesh_data_to_orig (NULL, &my_full_id, &copy->header, NULL, 0);
3997   }
3998   GNUNET_SERVER_receive_done (client, GNUNET_OK);
3999   return;
4000 }
4001
4002
4003 /**
4004  * Handler for client traffic directed to all peers in a tunnel
4005  *
4006  * @param cls closure
4007  * @param client identification of the client
4008  * @param message the actual message
4009  */
4010 static void
4011 handle_local_multicast (void *cls, struct GNUNET_SERVER_Client *client,
4012                         const struct GNUNET_MessageHeader *message)
4013 {
4014   struct MeshClient *c;
4015   struct MeshTunnel *t;
4016   struct GNUNET_MESH_Multicast *data_msg;
4017   MESH_TunnelNumber tid;
4018
4019   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4020               "MESH: Got a multicast request from a client!\n");
4021
4022   /* Sanity check for client registration */
4023   if (NULL == (c = client_get (client)))
4024   {
4025     GNUNET_break (0);
4026     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4027     return;
4028   }
4029   data_msg = (struct GNUNET_MESH_Multicast *) message;
4030   /* Sanity check for message size */
4031   if (sizeof (struct GNUNET_MESH_Multicast) +
4032       sizeof (struct GNUNET_MessageHeader) > ntohs (data_msg->header.size))
4033   {
4034     GNUNET_break (0);
4035     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4036     return;
4037   }
4038
4039   /* Tunnel exists? */
4040   tid = ntohl (data_msg->tid);
4041   t = tunnel_get_by_local_id (c, tid);
4042   if (NULL == t)
4043   {
4044     GNUNET_break (0);
4045     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4046     return;
4047   }
4048
4049   /* Does client own tunnel? */
4050   if (t->client->handle != client)
4051   {
4052     GNUNET_break (0);
4053     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4054     return;
4055   }
4056
4057   {
4058     char buf[ntohs(message->size)];
4059     struct GNUNET_MESH_Multicast *copy;
4060
4061     copy = (struct GNUNET_MESH_Multicast *)buf;
4062     memcpy(buf, message, ntohs(message->size));
4063     copy->oid = my_full_id;
4064     copy->tid = htonl(t->id.tid);
4065     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4066                 "MESH:   calling generic handler...\n");
4067     handle_mesh_data_multicast(client, &my_full_id, &copy->header, NULL, 0);
4068   }
4069
4070   /* receive done gets called when last copy is sent to a neighbor */
4071   return;
4072 }
4073
4074 /**
4075  * Functions to handle messages from clients
4076  */
4077 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
4078   {&handle_local_new_client, NULL,
4079    GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
4080   {&handle_local_tunnel_create, NULL,
4081    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE,
4082    sizeof (struct GNUNET_MESH_TunnelMessage)},
4083   {&handle_local_tunnel_destroy, NULL,
4084    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY,
4085    sizeof (struct GNUNET_MESH_TunnelMessage)},
4086   {&handle_local_connect_add, NULL,
4087    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD,
4088    sizeof (struct GNUNET_MESH_PeerControl)},
4089   {&handle_local_connect_del, NULL,
4090    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DEL,
4091    sizeof (struct GNUNET_MESH_PeerControl)},
4092   {&handle_local_connect_by_type, NULL,
4093    GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_ADD_BY_TYPE,
4094    sizeof (struct GNUNET_MESH_ConnectPeerByType)},
4095   {&handle_local_unicast, NULL,
4096    GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
4097   {&handle_local_to_origin, NULL,
4098    GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
4099   {&handle_local_multicast, NULL,
4100    GNUNET_MESSAGE_TYPE_MESH_MULTICAST, 0},
4101   {NULL, NULL, 0, 0}
4102 };
4103
4104
4105 /**
4106  * To be called on core init/fail.
4107  *
4108  * @param cls service closure
4109  * @param server handle to the server for this service
4110  * @param identity the public identity of this peer
4111  */
4112 static void
4113 core_init (void *cls, struct GNUNET_CORE_Handle *server,
4114            const struct GNUNET_PeerIdentity *identity)
4115 {
4116   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Core init\n");
4117   core_handle = server;
4118   if (0 != memcmp(identity, &my_full_id, sizeof(my_full_id)) || NULL == server)
4119   {
4120     GNUNET_log(GNUNET_ERROR_TYPE_ERROR, _("MESH: Wrong CORE service\n"));
4121     GNUNET_SCHEDULER_shutdown();   
4122   }
4123   return;
4124 }
4125
4126 /**
4127  * Method called whenever a given peer connects.
4128  *
4129  * @param cls closure
4130  * @param peer peer identity this notification is about
4131  * @param atsi performance data for the connection
4132  * @param atsi_count number of records in 'atsi'
4133  */
4134 static void
4135 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer,
4136               const struct GNUNET_ATS_Information *atsi,
4137               unsigned int atsi_count)
4138 {
4139   struct MeshPeerInfo *peer_info;
4140   struct MeshPeerPath *path;
4141
4142 #if MESH_DEBUG_CONNECTION
4143   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Peer connected\n");
4144   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      %s\n",
4145               GNUNET_i2s(&my_full_id));
4146 #endif
4147   peer_info = peer_info_get (peer);
4148   if (myid == peer_info->id)
4149   {
4150 #if MESH_DEBUG_CONNECTION
4151     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      (self)\n");
4152 #endif
4153     return;
4154   }
4155 #if MESH_DEBUG_CONNECTION
4156   else
4157   {
4158     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      %s\n",
4159                 GNUNET_i2s(peer));
4160   }
4161 #endif
4162   path = path_new (2);
4163   path->peers[0] = myid;
4164   path->peers[1] = peer_info->id;
4165   GNUNET_PEER_change_rc(myid, 1);
4166   GNUNET_PEER_change_rc(peer_info->id, 1);
4167   path_add_to_peer (peer_info, path);
4168   return;
4169 }
4170
4171 /**
4172  * Method called whenever a peer disconnects.
4173  *
4174  * @param cls closure
4175  * @param peer peer identity this notification is about
4176  */
4177 static void
4178 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
4179 {
4180   struct MeshPeerInfo *pi;
4181   unsigned int i;
4182
4183 #if MESH_DEBUG_CONNECTION
4184   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: Peer disconnected\n");
4185 #endif
4186   pi = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
4187   if (NULL == pi)
4188   {
4189     GNUNET_break (0);
4190     return;
4191   }
4192   for (i = 0; i < CORE_QUEUE_SIZE; i++)
4193   {
4194     /* TODO: notify that the transmission failed */
4195     peer_info_cancel_transmission(pi, i);
4196   }
4197   path_remove_from_peer (pi, pi->id, myid);
4198 #if MESH_DEBUG_CONNECTION
4199   if (myid == pi->id)
4200   {
4201     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH:      (self)\n");
4202   }
4203 #endif
4204   return;
4205 }
4206
4207
4208 /******************************************************************************/
4209 /************************      MAIN FUNCTIONS      ****************************/
4210 /******************************************************************************/
4211
4212 /**
4213  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
4214  *
4215  * @param cls closure
4216  * @param key current key code
4217  * @param value value in the hash map
4218  * @return GNUNET_YES if we should continue to iterate,
4219  *         GNUNET_NO if not.
4220  */
4221 static int
4222 shutdown_tunnel (void *cls, const GNUNET_HashCode * key, void *value)
4223 {
4224   struct MeshTunnel *t = value;
4225   tunnel_destroy(t);
4226   return GNUNET_YES;
4227 }
4228
4229 /**
4230  * Iterator over peer hash map entries to destroy the tunnel during shutdown.
4231  *
4232  * @param cls closure
4233  * @param key current key code
4234  * @param value value in the hash map
4235  * @return GNUNET_YES if we should continue to iterate,
4236  *         GNUNET_NO if not.
4237  */
4238 static int
4239 shutdown_peer (void *cls, const GNUNET_HashCode * key, void *value)
4240 {
4241   struct MeshPeerInfo *p = value;
4242   peer_info_destroy (p);
4243   return GNUNET_YES;
4244 }
4245
4246 /**
4247  * Task run during shutdown.
4248  *
4249  * @param cls unused
4250  * @param tc unused
4251  */
4252 static void
4253 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4254 {
4255   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: shutting down\n");
4256   /* TODO: destroy tunnels? */
4257   if (core_handle != NULL)
4258   {
4259     GNUNET_CORE_disconnect (core_handle);
4260     core_handle = NULL;
4261   }
4262   GNUNET_CONTAINER_multihashmap_iterate(tunnels, &shutdown_tunnel, NULL);
4263   GNUNET_CONTAINER_multihashmap_iterate(peers, &shutdown_peer, NULL);
4264   if (dht_handle != NULL)
4265   {
4266     GNUNET_DHT_disconnect (dht_handle);
4267     dht_handle = NULL;
4268   }
4269   if (nc != NULL)
4270   {
4271     GNUNET_SERVER_notification_context_destroy (nc);
4272     nc = NULL;
4273   }
4274   if (GNUNET_SCHEDULER_NO_TASK != announce_id_task)
4275   {
4276     GNUNET_SCHEDULER_cancel (announce_id_task);
4277     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
4278   }
4279   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: shut down\n");
4280 }
4281
4282 /**
4283  * Process mesh requests.
4284  *
4285  * @param cls closure
4286  * @param server the initialized server
4287  * @param c configuration to use
4288  */
4289 static void
4290 run (void *cls, struct GNUNET_SERVER_Handle *server,
4291      const struct GNUNET_CONFIGURATION_Handle *c)
4292 {
4293   struct MeshPeerInfo *peer;
4294   struct MeshPeerPath *p;
4295   char *keyfile;
4296
4297   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: starting to run\n");
4298   server_handle = server;
4299   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
4300                                      CORE_QUEUE_SIZE,   /* queue size */
4301                                      NULL,      /* Closure passed to MESH functions */
4302                                      &core_init,        /* Call core_init once connected */
4303                                      &core_connect,     /* Handle connects */
4304                                      &core_disconnect,  /* remove peers on disconnects */
4305                                      NULL,      /* Don't notify about all incoming messages */
4306                                      GNUNET_NO, /* For header only in notification */
4307                                      NULL,      /* Don't notify about all outbound messages */
4308                                      GNUNET_NO, /* For header-only out notification */
4309                                      core_handlers);    /* Register these handlers */
4310   if (core_handle == NULL)
4311   {
4312     GNUNET_break (0);
4313     GNUNET_SCHEDULER_shutdown ();
4314     return;
4315   }
4316
4317   if (GNUNET_OK !=
4318        GNUNET_CONFIGURATION_get_value_filename (c, "GNUNETD", "HOSTKEY",
4319                                                 &keyfile))
4320   {
4321     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
4322                 _
4323                 ("Mesh service is lacking key configuration settings.  Exiting.\n"));
4324     GNUNET_SCHEDULER_shutdown ();
4325     return;
4326   }
4327   my_private_key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
4328   GNUNET_free (keyfile);
4329   if (my_private_key == NULL)
4330   {
4331     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
4332                 _("Mesh service could not access hostkey.  Exiting.\n"));
4333     GNUNET_SCHEDULER_shutdown ();
4334     return;
4335   }
4336   GNUNET_CRYPTO_rsa_key_get_public (my_private_key, &my_public_key);
4337   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
4338                       &my_full_id.hashPubKey);
4339   myid = GNUNET_PEER_intern (&my_full_id);
4340
4341   dht_handle = GNUNET_DHT_connect (c, 64);
4342   if (dht_handle == NULL)
4343   {
4344     GNUNET_break (0);
4345   }
4346
4347   next_tid = 0;
4348   next_local_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
4349
4350   tunnels = GNUNET_CONTAINER_multihashmap_create (32);
4351   incoming_tunnels = GNUNET_CONTAINER_multihashmap_create (32);
4352   peers = GNUNET_CONTAINER_multihashmap_create (32);
4353   applications = GNUNET_CONTAINER_multihashmap_create (32);
4354   types = GNUNET_CONTAINER_multihashmap_create (32);
4355
4356   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
4357   nc = GNUNET_SERVER_notification_context_create (server_handle,
4358                                                   LOCAL_QUEUE_SIZE);
4359   GNUNET_SERVER_disconnect_notify (server_handle,
4360                                    &handle_local_client_disconnect,
4361                                    NULL);
4362
4363
4364   clients = NULL;
4365   clients_tail = NULL;
4366 #if MESH_DEBUG
4367   next_client_id = 0;
4368 #endif
4369
4370   announce_applications_task = GNUNET_SCHEDULER_NO_TASK;
4371   announce_id_task = GNUNET_SCHEDULER_add_now (&announce_id, cls);
4372
4373   /* Create a peer_info for the local peer */
4374   peer = peer_info_get(&my_full_id);
4375   p = path_new (1);
4376   p->peers[0] = myid;
4377   path_add_to_peer(peer, p);
4378
4379   /* Scheduled the task to clean up when shutdown is called */
4380   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
4381                                 NULL);
4382
4383   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: end of run()\n");
4384 }
4385
4386 /**
4387  * The main function for the mesh service.
4388  *
4389  * @param argc number of arguments from the command line
4390  * @param argv command line arguments
4391  * @return 0 ok, 1 on error
4392  */
4393 int
4394 main (int argc, char *const *argv)
4395 {
4396   int ret;
4397
4398   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: main()\n");
4399   ret =
4400       (GNUNET_OK ==
4401        GNUNET_SERVICE_run (argc, argv, "mesh", GNUNET_SERVICE_OPTION_NONE, &run,
4402                            NULL)) ? 0 : 1;
4403   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MESH: main() END\n");
4404
4405   return ret;
4406 }