dd65251a98c33c6e1083e846541a0ff875bd2b4b
[oweals/gnunet.git] / src / mesh / gnunet-service-mesh.c
1 /*
2      This file is part of GNUnet.
3      (C) 2001-2013 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file mesh/gnunet-service-mesh.c
23  * @brief GNUnet MESH service
24  * @author Bartlomiej Polot
25  *
26  *  FIXME in progress:
27  * - keep queues until receiving ACK
28  *
29  * TODO:
30  * - relay corking down to core
31  * - set ttl relative to path length
32  * - add signatures
33  * - add encryption
34  * TODO END
35  */
36
37 #include "platform.h"
38 #include "mesh.h"
39 #include "mesh_protocol.h"
40 #include "mesh_path.h"
41 #include "block_mesh.h"
42 #include "gnunet_dht_service.h"
43 #include "gnunet_statistics_service.h"
44
45 #define MESH_BLOOM_SIZE         128
46
47 #define MESH_DEBUG_DHT          GNUNET_NO
48 #define MESH_DEBUG_CONNECTION   GNUNET_NO
49 #define MESH_DEBUG_TIMING       __LINUX__ && GNUNET_NO
50
51 #define MESH_MAX_POLL_TIME      GNUNET_TIME_relative_multiply (\
52                                   GNUNET_TIME_UNIT_MINUTES,\
53                                   10)
54
55 #if MESH_DEBUG_CONNECTION
56 #define DEBUG_CONN(...) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
57 #else
58 #define DEBUG_CONN(...)
59 #endif
60
61 #if MESH_DEBUG_DHT
62 #define DEBUG_DHT(...) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
63 #else
64 #define DEBUG_DHT(...)
65 #endif
66
67 #if MESH_DEBUG_TIMING
68 #include <time.h>
69 double __sum;
70 uint64_t __count;
71 struct timespec __mesh_start;
72 struct timespec __mesh_end;
73 #define INTERVAL_START clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &(__mesh_start))
74 #define INTERVAL_END \
75 do {\
76   clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &(__mesh_end));\
77   double __diff = __mesh_end.tv_nsec - __mesh_start.tv_nsec;\
78   if (__diff < 0) __diff += 1000000000;\
79   __sum += __diff;\
80   __count++;\
81 } while (0)
82 #define INTERVAL_SHOW \
83 if (0 < __count)\
84   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "AVG process time: %f ns\n", __sum/__count)
85 #else
86 #define INTERVAL_START
87 #define INTERVAL_END
88 #define INTERVAL_SHOW
89 #endif
90
91 /******************************************************************************/
92 /************************      DATA STRUCTURES     ****************************/
93 /******************************************************************************/
94
95 /** FWD declaration */
96 struct MeshPeerInfo;
97 struct MeshClient;
98
99
100 /**
101  * Struct containing info about a queued transmission to this peer
102  */
103 struct MeshPeerQueue
104 {
105     /**
106       * DLL next
107       */
108   struct MeshPeerQueue *next;
109
110     /**
111       * DLL previous
112       */
113   struct MeshPeerQueue *prev;
114
115     /**
116      * Peer this transmission is directed to.
117      */
118   struct MeshPeerInfo *peer;
119
120     /**
121      * Tunnel this message belongs to.
122      */
123   struct MeshTunnel *tunnel;
124
125     /**
126      * Pointer to info stucture used as cls.
127      */
128   void *cls;
129
130     /**
131      * Type of message
132      */
133   uint16_t type;
134
135     /**
136      * Size of the message
137      */
138   size_t size;
139 };
140
141
142 /**
143  * Struct containing all information regarding a given peer
144  */
145 struct MeshPeerInfo
146 {
147     /**
148      * ID of the peer
149      */
150   GNUNET_PEER_Id id;
151
152     /**
153      * Last time we heard from this peer
154      */
155   struct GNUNET_TIME_Absolute last_contact;
156
157     /**
158      * Number of attempts to reconnect so far
159      */
160   int n_reconnect_attempts;
161
162     /**
163      * Paths to reach the peer, ordered by ascending hop count
164      */
165   struct MeshPeerPath *path_head;
166
167     /**
168      * Paths to reach the peer, ordered by ascending hop count
169      */
170   struct MeshPeerPath *path_tail;
171
172     /**
173      * Handle to stop the DHT search for a path to this peer
174      */
175   struct GNUNET_DHT_GetHandle *dhtget;
176
177     /**
178      * Array of tunnels this peer is the target of.
179      * Most probably a small amount, therefore not a hashmap.
180      * When the path to the peer changes, notify these tunnels to let them
181      * re-adjust their path trees.
182      */
183   struct MeshTunnel **tunnels;
184
185     /**
186      * Number of tunnels this peers participates in
187      */
188   unsigned int ntunnels;
189
190    /**
191     * Transmission queue to core DLL head
192     */
193   struct MeshPeerQueue *queue_head;
194
195    /**
196     * Transmission queue to core DLL tail
197     */
198    struct MeshPeerQueue *queue_tail;
199
200    /**
201     * How many messages are in the queue to this peer.
202     */
203    unsigned int queue_n;
204
205    /**
206     * Handle for queued transmissions
207     */
208   struct GNUNET_CORE_TransmitHandle *core_transmit;
209 };
210
211
212 /**
213  * Struct to encapsulate all the Flow Control information to a peer in the
214  * context of a tunnel: Same peer in different tunnels will have independent
215  * flow control structures, allowing to choke/free tunnels according to its
216  * own criteria.
217  */
218 struct MeshFlowControl
219 {
220   /**
221    * ID of the last packet sent towards the peer.
222    */
223   uint32_t last_pid_sent;
224
225   /**
226    * ID of the last packet received from the peer.
227    */
228   uint32_t last_pid_recv;
229
230   /**
231    * Last ACK sent to the peer (peer can't send more than this PID).
232    */
233   uint32_t last_ack_sent;
234
235   /**
236    * Last ACK sent towards the origin (for traffic towards leaf node).
237    */
238   uint32_t last_ack_recv;
239
240   /**
241    * How many payload messages are in the queue towards this peer.
242    */
243   uint32_t queue_n;
244
245   /**
246    * Task to poll the peer in case of a lost ACK causes stall.
247    */
248   GNUNET_SCHEDULER_TaskIdentifier poll_task;
249
250   /**
251    * How frequently to poll for ACKs.
252    */
253   struct GNUNET_TIME_Relative poll_time;
254
255   /**
256    * On which tunnel to poll.
257    * Using an explicit poll_ctx would not help memory wise,
258    * since the allocated context would have to be stored in the
259    * fc struct in order to free it upon cancelling poll_task.
260    */
261   struct MeshTunnel *t;
262 };
263
264
265 /**
266  * Globally unique tunnel identification (owner + number)
267  * DO NOT USE OVER THE NETWORK
268  */
269 struct MESH_TunnelID
270 {
271     /**
272      * Node that owns the tunnel
273      */
274   GNUNET_PEER_Id oid;
275
276     /**
277      * Tunnel number to differentiate all the tunnels owned by the node oid
278      * ( tid < GNUNET_MESH_LOCAL_TUNNEL_ID_CLI )
279      */
280   MESH_TunnelNumber tid;
281 };
282
283
284 /**
285  * Info needed to retry a message in case it gets lost.
286  */
287 struct MeshSentMessage {
288
289   /**
290    * Tunnel this message is in.
291    */
292   struct MeshTunnel                 *t;
293
294   /**
295    * ID of the message (ACK needed to free)
296    */
297   uint32_t                          id;
298
299   /**
300    * Task to resend/poll in case no ACK is received.
301    */
302   GNUNET_SCHEDULER_TaskIdentifier   retry_task; // FIXME move to per tunnel timer?
303
304   /**
305    * Counter for exponential backoff.
306    */
307   struct GNUNET_TIME_Relative       retry_timer;
308
309   /**
310    * Is this a forward or backward going message?
311    */
312   int                               is_forward;
313
314   /* struct GNUNET_MESH_Data with payload */
315 };
316
317 /**
318  * Struct containing all information regarding a tunnel
319  * For an intermediate node the improtant info used will be:
320  * - id        Tunnel unique identification
321  * - paths[0]  To know where to send it next
322  * - metainfo: ready, speeds, accounting
323  */
324 struct MeshTunnel
325 {
326     /**
327      * Tunnel ID
328      */
329   struct MESH_TunnelID id;
330
331     /**
332      * Port of the tunnel.
333      */
334   uint32_t port;
335
336     /**
337      * State of the tunnel.
338      */
339   enum MeshTunnelState state;
340
341     /**
342      * Local tunnel number ( >= GNUNET_MESH_LOCAL_TUNNEL_ID_CLI or 0 )
343      */
344   MESH_TunnelNumber local_tid;
345
346     /**
347      * Local tunnel number for local destination clients (incoming number)
348      * ( >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV or 0). All clients share the same
349      * number.
350      */
351   MESH_TunnelNumber local_tid_dest;
352
353     /**
354      * Is the tunnel bufferless (minimum latency)?
355      */
356   int nobuffer;
357
358     /**
359      * Is the tunnel reliable?
360      */
361   int reliable;
362
363     /**
364      * Force sending ACK? Flag to allow duplicate ACK on POLL.
365      */
366   int force_ack;
367
368     /**
369      * How many messages do we accept in the forward queue.
370      */
371   unsigned int queue_max;
372
373     /**
374      * Last time the tunnel was used
375      */
376   struct GNUNET_TIME_Absolute timestamp;
377
378     /**
379      * Destination of the tunnel.
380      */
381   GNUNET_PEER_Id dest;
382
383     /**
384      * Next hop in the tunnel. If 0, @c client must be set.
385      */
386   GNUNET_PEER_Id next_hop;
387
388     /**
389      * Previous hop in the tunnel. If 0, @c owner must be set.
390      */
391   GNUNET_PEER_Id prev_hop;
392
393     /**
394      * Flow control information about @c next_hop or @c client.
395      */
396   struct MeshFlowControl next_fc;
397
398   /**
399    * Flow control information about @c prev_hop or @c owner.
400    */
401   struct MeshFlowControl prev_fc;
402
403     /**
404      * Client owner of the tunnel, if any
405      */
406   struct MeshClient *owner;
407
408     /**
409      * Client destination of the tunnel, if any.
410      */
411   struct MeshClient *client;
412
413     /**
414      * Task to keep the used paths alive at the owner,
415      * time tunnel out on all the other peers.
416      */
417   GNUNET_SCHEDULER_TaskIdentifier maintenance_task;
418
419     /**
420      * Path being used for the tunnel.
421      */
422   struct MeshPeerPath *path;
423
424     /**
425      * Flag to signal the destruction of the tunnel.
426      * If this is set GNUNET_YES the tunnel will be destroyed
427      * when the queue is empty.
428      */
429   int destroy;
430
431     /**
432      * Total messages pending for this tunnels, payload or not.
433      */
434   unsigned int pending_messages;
435
436   /**
437    * Messages sent and not yet ACK'd.
438    * Only present (non-NULL) at the owner of a tunnel.
439    */
440   struct GNUNET_CONTAINER_MultiHashMap32 *sent_messages_fwd;
441
442   /**
443    * Messages sent and not yet ACK'd.
444    * Only present (non-NULL) at the destination of a tunnel.
445    */
446   struct GNUNET_CONTAINER_MultiHashMap32 *sent_messages_bck;
447 };
448
449
450 /**
451  * Struct containing information about a client of the service
452  * 
453  * TODO: add a list of 'waiting' ports
454  */
455 struct MeshClient
456 {
457     /**
458      * Linked list next
459      */
460   struct MeshClient *next;
461
462     /**
463      * Linked list prev
464      */
465   struct MeshClient *prev;
466
467     /**
468      * Tunnels that belong to this client, indexed by local id
469      */
470   struct GNUNET_CONTAINER_MultiHashMap *own_tunnels;
471
472    /**
473      * Tunnels this client has accepted, indexed by incoming local id
474      */
475   struct GNUNET_CONTAINER_MultiHashMap *incoming_tunnels;
476
477     /**
478      * Handle to communicate with the client
479      */
480   struct GNUNET_SERVER_Client *handle;
481
482     /**
483      * Ports that this client has declared interest in.
484      * Indexed by a GMC_hash32 (type), contains *Client.
485      */
486   struct GNUNET_CONTAINER_MultiHashMap *ports;
487
488     /**
489      * Whether the client is active or shutting down (don't send confirmations
490      * to a client that is shutting down.
491      */
492   int shutting_down;
493
494     /**
495      * ID of the client, mainly for debug messages
496      */
497   unsigned int id;
498
499 };
500
501
502 /******************************************************************************/
503 /************************      DEBUG FUNCTIONS     ****************************/
504 /******************************************************************************/
505
506 #if MESH_DEBUG
507 /**
508  * GNUNET_SCHEDULER_Task for printing a message after some operation is done
509  * @param cls string to print
510  * @param success  GNUNET_OK if the PUT was transmitted,
511  *                GNUNET_NO on timeout,
512  *                GNUNET_SYSERR on disconnect from service
513  *                after the PUT message was transmitted
514  *                (so we don't know if it was received or not)
515  */
516
517 #if 0
518 static void
519 mesh_debug (void *cls, int success)
520 {
521   char *s = cls;
522
523   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "%s (%d)\n", s, success);
524 }
525 #endif
526
527 /* FIXME */
528 unsigned int debug_fwd_ack;
529 unsigned int debug_bck_ack;
530
531 #endif
532
533 /******************************************************************************/
534 /***********************      GLOBAL VARIABLES     ****************************/
535 /******************************************************************************/
536
537 /************************** Configuration parameters **************************/
538
539 /**
540  * How often to send tunnel keepalives. Tunnels timeout after 4 missed.
541  */
542 static struct GNUNET_TIME_Relative refresh_path_time;
543
544 /**
545  * How often to PUT own ID in the DHT.
546  */
547 static struct GNUNET_TIME_Relative id_announce_time;
548
549 /**
550  * Maximum time allowed to connect to a peer found by string.
551  */
552 static struct GNUNET_TIME_Relative connect_timeout;
553
554 /**
555  * Default TTL for payload packets.
556  */
557 static unsigned long long default_ttl;
558
559 /**
560  * DHT replication level, see DHT API: GNUNET_DHT_get_start, GNUNET_DHT_put.
561  */
562 static unsigned long long dht_replication_level;
563
564 /**
565  * How many tunnels are we willing to maintain.
566  * Local tunnels are always allowed, even if there are more tunnels than max.
567  */
568 static unsigned long long max_tunnels;
569
570 /**
571  * How many messages *in total* are we willing to queue, divided by number of 
572  * tunnels to get tunnel queue size.
573  */
574 static unsigned long long max_msgs_queue;
575
576 /**
577  * How many peers do we want to remember?
578  */
579 static unsigned long long max_peers;
580
581 /**
582  * Percentage of messages that will be dropped (for test purposes only).
583  */
584 static unsigned long long drop_percent;
585
586 /*************************** Static global variables **************************/
587
588 /**
589  * Hostkey generation context
590  */
591 static struct GNUNET_CRYPTO_EccKeyGenerationContext *keygen;
592
593 /**
594  * DLL with all the clients, head.
595  */
596 static struct MeshClient *clients_head;
597
598 /**
599  * DLL with all the clients, tail.
600  */
601 static struct MeshClient *clients_tail;
602
603 /**
604  * Tunnels known, indexed by MESH_TunnelID (MeshTunnel).
605  */
606 static struct GNUNET_CONTAINER_MultiHashMap *tunnels;
607
608 /**
609  * Number of tunnels known.
610  */
611 static unsigned long long n_tunnels;
612
613 /**
614  * Tunnels incoming, indexed by MESH_TunnelNumber
615  * (which is greater than GNUNET_MESH_LOCAL_TUNNEL_ID_SERV).
616  */
617 static struct GNUNET_CONTAINER_MultiHashMap *incoming_tunnels;
618
619 /**
620  * Peers known, indexed by PeerIdentity (MeshPeerInfo).
621  */
622 static struct GNUNET_CONTAINER_MultiHashMap *peers;
623
624 /*
625  * Handle to communicate with transport
626  */
627 // static struct GNUNET_TRANSPORT_Handle *transport_handle;
628
629 /**
630  * Handle to communicate with core.
631  */
632 static struct GNUNET_CORE_Handle *core_handle;
633
634 /**
635  * Handle to use DHT.
636  */
637 static struct GNUNET_DHT_Handle *dht_handle;
638
639 /**
640  * Handle to server.
641  */
642 static struct GNUNET_SERVER_Handle *server_handle;
643
644 /**
645  * Handle to the statistics service.
646  */
647 static struct GNUNET_STATISTICS_Handle *stats;
648
649 /**
650  * Notification context, to send messages to local clients.
651  */
652 static struct GNUNET_SERVER_NotificationContext *nc;
653
654 /**
655  * Local peer own ID (memory efficient handle).
656  */
657 static GNUNET_PEER_Id myid;
658
659 /**
660  * Local peer own ID (full value).
661  */
662 static struct GNUNET_PeerIdentity my_full_id;
663
664 /**
665  * Own private key.
666  */
667 static struct GNUNET_CRYPTO_EccPrivateKey *my_private_key;
668
669 /**
670  * Own public key.
671  */
672 static struct GNUNET_CRYPTO_EccPublicKeyBinaryEncoded my_public_key;
673
674 /**
675  * Tunnel ID for the next created tunnel (global tunnel number).
676  */
677 static MESH_TunnelNumber next_tid;
678
679 /**
680  * Tunnel ID for the next incoming tunnel (local tunnel number).
681  */
682 static MESH_TunnelNumber next_local_tid;
683
684 /**
685  * All ports clients of this peer have opened.
686  */
687 static struct GNUNET_CONTAINER_MultiHashMap *ports;
688
689 /**
690  * Task to periodically announce itself in the network.
691  */
692 GNUNET_SCHEDULER_TaskIdentifier announce_id_task;
693
694 /**
695  * Next ID to assign to a client.
696  */
697 unsigned int next_client_id;
698
699
700 /******************************************************************************/
701 /***********************         DECLARATIONS        **************************/
702 /******************************************************************************/
703
704 /**
705  * Function to process paths received for a new peer addition. The recorded
706  * paths form the initial tunnel, which can be optimized later.
707  * Called on each result obtained for the DHT search.
708  *
709  * @param cls closure
710  * @param exp when will this value expire
711  * @param key key of the result
712  * @param type type of the result
713  * @param size number of bytes in data
714  * @param data pointer to the result data
715  */
716 static void
717 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
718                     const struct GNUNET_HashCode * key,
719                     const struct GNUNET_PeerIdentity *get_path,
720                     unsigned int get_path_length,
721                     const struct GNUNET_PeerIdentity *put_path,
722                     unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
723                     size_t size, const void *data);
724
725
726 /**
727  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
728  * and insert it in the appropriate structures if the peer is not known yet.
729  *
730  * @param peer Full identity of the peer.
731  *
732  * @return Existing or newly created peer info.
733  */
734 static struct MeshPeerInfo *
735 peer_get (const struct GNUNET_PeerIdentity *peer);
736
737
738 /**
739  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
740  * and insert it in the appropriate structures if the peer is not known yet.
741  *
742  * @param peer Short identity of the peer.
743  *
744  * @return Existing or newly created peer info.
745  */
746 static struct MeshPeerInfo *
747 peer_get_short (const GNUNET_PEER_Id peer);
748
749
750 /**
751  * Build a PeerPath from the paths returned from the DHT, reversing the paths
752  * to obtain a local peer -> destination path and interning the peer ids.
753  *
754  * @return Newly allocated and created path
755  */
756 static struct MeshPeerPath *
757 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
758                      unsigned int get_path_length,
759                      const struct GNUNET_PeerIdentity *put_path,
760                      unsigned int put_path_length);
761
762
763 /**
764  * Adds a path to the peer_infos of all the peers in the path
765  *
766  * @param p Path to process.
767  * @param confirmed Whether we know if the path works or not.
768  */
769 static void
770 path_add_to_peers (struct MeshPeerPath *p, int confirmed);
771
772
773
774 /**
775  * Search for a tunnel by global ID using full PeerIdentities.
776  *
777  * @param oid owner of the tunnel.
778  * @param tid global tunnel number.
779  *
780  * @return tunnel handler, NULL if doesn't exist.
781  */
782 static struct MeshTunnel *
783 tunnel_get (const struct GNUNET_PeerIdentity *oid, MESH_TunnelNumber tid);
784
785
786 /**
787  * Notify a tunnel that a connection has broken that affects at least
788  * some of its peers.
789  *
790  * @param t Tunnel affected.
791  * @param p1 Peer that got disconnected from p2.
792  * @param p2 Peer that got disconnected from p1.
793  *
794  * @return Short ID of the peer disconnected (either p1 or p2).
795  *         0 if the tunnel remained unaffected.
796  */
797 static GNUNET_PEER_Id
798 tunnel_notify_connection_broken (struct MeshTunnel *t, GNUNET_PEER_Id p1,
799                                  GNUNET_PEER_Id p2);
800
801
802 /**
803  * @brief Use the given path for the tunnel.
804  * Update the next and prev hops (and RCs).
805  * (Re)start the path refresh in case the tunnel is locally owned.
806  * 
807  * @param t Tunnel to update.
808  * @param p Path to use.
809  */
810 static void
811 tunnel_use_path (struct MeshTunnel *t, struct MeshPeerPath *p);
812
813 /**
814  * Tunnel is empty: destroy it.
815  * 
816  * Notifies all participants (peers, cleints) about the destruction.
817  * 
818  * @param t Tunnel to destroy. 
819  */
820 static void
821 tunnel_destroy_empty (struct MeshTunnel *t);
822
823 /**
824  * @brief Queue and pass message to core when possible.
825  * 
826  * If type is payload (UNICAST, TO_ORIGIN, MULTICAST) checks for queue status
827  * and accounts for it. In case the queue is full, the message is dropped and
828  * a break issued.
829  * 
830  * Otherwise, message is treated as internal and allowed to go regardless of 
831  * queue status.
832  *
833  * @param cls Closure (@c type dependant). It will be used by queue_send to
834  *            build the message to be sent if not already prebuilt.
835  * @param type Type of the message, 0 for a raw message.
836  * @param size Size of the message.
837  * @param dst Neighbor to send message to.
838  * @param t Tunnel this message belongs to.
839  */
840 static void
841 queue_add (void *cls, uint16_t type, size_t size,
842            struct MeshPeerInfo *dst, struct MeshTunnel *t);
843
844
845 /**
846  * Free a transmission that was already queued with all resources
847  * associated to the request.
848  *
849  * @param queue Queue handler to cancel.
850  * @param clear_cls Is it necessary to free associated cls?
851  */
852 static void
853 queue_destroy (struct MeshPeerQueue *queue, int clear_cls);
854
855
856 /**
857  * @brief Get the next transmittable message from the queue.
858  *
859  * This will be the head, except in the case of being a data packet
860  * not allowed by the destination peer.
861  *
862  * @param peer Destination peer.
863  *
864  * @return The next viable MeshPeerQueue element to send to that peer.
865  *         NULL when there are no transmittable messages.
866  */
867 struct MeshPeerQueue *
868 queue_get_next (const struct MeshPeerInfo *peer);
869
870
871 /**
872  * Core callback to write a queued packet to core buffer
873  *
874  * @param cls Closure (peer info).
875  * @param size Number of bytes available in buf.
876  * @param buf Where the to write the message.
877  *
878  * @return number of bytes written to buf
879  */
880 static size_t
881 queue_send (void *cls, size_t size, void *buf);
882
883
884 /******************************************************************************/
885 /************************    PERIODIC FUNCTIONS    ****************************/
886 /******************************************************************************/
887
888 /**
889  * Periodically announce self id in the DHT
890  *
891  * @param cls closure
892  * @param tc task context
893  */
894 static void
895 announce_id (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
896 {
897   struct PBlock block;
898
899   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
900   {
901     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
902     return;
903   }
904   /* TODO
905    * - Set data expiration in function of X
906    * - Adapt X to churn
907    */
908   DEBUG_DHT ("DHT_put for ID %s started.\n", GNUNET_i2s (&my_full_id));
909
910   block.id = my_full_id;
911   GNUNET_DHT_put (dht_handle,   /* DHT handle */
912                   &my_full_id.hashPubKey,       /* Key to use */
913                   dht_replication_level,     /* Replication level */
914                   GNUNET_DHT_RO_RECORD_ROUTE | GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,    /* DHT options */
915                   GNUNET_BLOCK_TYPE_MESH_PEER,       /* Block type */
916                   sizeof (block),  /* Size of the data */
917                   (const char *) &block, /* Data itself */
918                   GNUNET_TIME_UNIT_FOREVER_ABS,  /* Data expiration */
919                   GNUNET_TIME_UNIT_FOREVER_REL, /* Retry time */
920                   NULL,         /* Continuation */
921                   NULL);        /* Continuation closure */
922   announce_id_task =
923       GNUNET_SCHEDULER_add_delayed (id_announce_time, &announce_id, cls);
924 }
925
926
927 /******************************************************************************/
928 /******************      GENERAL HELPER FUNCTIONS      ************************/
929 /******************************************************************************/
930
931
932 /**
933  * Check if client has registered with the service and has not disconnected
934  *
935  * @param client the client to check
936  *
937  * @return non-NULL if client exists in the global DLL
938  */
939 static struct MeshClient *
940 client_get (struct GNUNET_SERVER_Client *client)
941 {
942   struct MeshClient *c;
943
944   c = clients_head;
945   while (NULL != c)
946   {
947     if (c->handle == client)
948       return c;
949     c = c->next;
950   }
951   return NULL;
952 }
953
954
955 /**
956  * Deletes a tunnel from a client (either owner or destination). To be used on
957  * tunnel destroy.
958  *
959  * @param c Client whose tunnel to delete.
960  * @param t Tunnel which should be deleted.
961  */
962 static void
963 client_delete_tunnel (struct MeshClient *c, struct MeshTunnel *t)
964 {
965   struct GNUNET_HashCode hash;
966
967   if (c == t->owner)
968   {
969     GMC_hash32 (t->local_tid, &hash);
970     GNUNET_assert (GNUNET_YES ==
971                    GNUNET_CONTAINER_multihashmap_remove (c->own_tunnels,
972                                                          &hash,
973                                                          t));
974   }
975   else if (c == t->client)
976   {
977     GMC_hash32 (t->local_tid_dest, &hash);
978     GNUNET_assert (GNUNET_YES ==
979                    GNUNET_CONTAINER_multihashmap_remove (c->incoming_tunnels,
980                                                          &hash,
981                                                          t));
982   }
983   else
984   {
985     GNUNET_break (0);
986   }
987 }
988
989 /**
990  * Notify the appropriate client that a new incoming tunnel was created.
991  *
992  * @param t Tunnel that was created.
993  */
994 static void
995 send_client_tunnel_create (struct MeshTunnel *t)
996 {
997   struct GNUNET_MESH_TunnelMessage msg;
998
999   if (NULL == t->client)
1000     return;
1001   msg.header.size = htons (sizeof (msg));
1002   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1003   msg.tunnel_id = htonl (t->local_tid_dest);
1004   msg.port = htonl (t->port);
1005   GNUNET_PEER_resolve (t->id.oid, &msg.peer);
1006   GNUNET_SERVER_notification_context_unicast (nc, t->client->handle,
1007                                               &msg.header, GNUNET_NO);
1008 }
1009
1010
1011 /**
1012  * Notify dest client that the incoming tunnel is no longer valid.
1013  *
1014  * @param c Client to notify..
1015  * @param t Tunnel that is destroyed.
1016  */
1017 static void
1018 send_client_tunnel_destroy (struct MeshClient *c, struct MeshTunnel *t)
1019 {
1020   struct GNUNET_MESH_TunnelMessage msg;
1021
1022   if (NULL == c)
1023   {
1024     GNUNET_break (0);
1025     return;
1026   }
1027   if (c != t->client && c != t->owner)
1028   {
1029     GNUNET_break (0);
1030     return;
1031   }
1032   msg.header.size = htons (sizeof (msg));
1033   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1034   msg.tunnel_id = htonl (t->local_tid_dest);
1035   msg.port = htonl (0);
1036   memset(&msg.peer, 0, sizeof (msg.peer));
1037   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1038                                               &msg.header, GNUNET_NO);
1039 }
1040
1041
1042 /**
1043  * Iterator over all the peers to remove the oldest not-used entry.
1044  *
1045  * @param cls Closure (unsued).
1046  * @param key ID of the peer.
1047  * @param value Peer_Info of the peer.
1048  *
1049  * FIXME implement
1050  */
1051 static int
1052 peer_info_timeout (void *cls,
1053                    const struct GNUNET_HashCode *key,
1054                    void *value)
1055 {
1056   return GNUNET_YES;
1057 }
1058
1059 /**
1060  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
1061  * and insert it in the appropriate structures if the peer is not known yet.
1062  *
1063  * @param peer Full identity of the peer.
1064  *
1065  * @return Existing or newly created peer info.
1066  */
1067 static struct MeshPeerInfo *
1068 peer_get (const struct GNUNET_PeerIdentity *peer)
1069 {
1070   struct MeshPeerInfo *peer_info;
1071
1072   peer_info = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
1073   if (NULL == peer_info)
1074   {
1075     peer_info =
1076         (struct MeshPeerInfo *) GNUNET_malloc (sizeof (struct MeshPeerInfo));
1077     if (GNUNET_CONTAINER_multihashmap_size (peers) > max_peers)
1078     {
1079       GNUNET_CONTAINER_multihashmap_iterate (peers,
1080                                              &peer_info_timeout,
1081                                              NULL);
1082     }
1083     GNUNET_CONTAINER_multihashmap_put (peers, &peer->hashPubKey, peer_info,
1084                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1085     peer_info->id = GNUNET_PEER_intern (peer);
1086   }
1087   peer_info->last_contact = GNUNET_TIME_absolute_get();
1088
1089   return peer_info;
1090 }
1091
1092
1093 /**
1094  * Retrieve the MeshPeerInfo stucture associated with the peer, create one
1095  * and insert it in the appropriate structures if the peer is not known yet.
1096  *
1097  * @param peer Short identity of the peer.
1098  *
1099  * @return Existing or newly created peer info.
1100  */
1101 static struct MeshPeerInfo *
1102 peer_get_short (const GNUNET_PEER_Id peer)
1103 {
1104   struct GNUNET_PeerIdentity id;
1105
1106   GNUNET_PEER_resolve (peer, &id);
1107   return peer_get (&id);
1108 }
1109
1110
1111 /**
1112  * Choose the best path towards a peer considering the tunnel properties.
1113  * 
1114  * @param peer The destination peer.
1115  * @param t The tunnel the path is for.
1116  *
1117  * @return Best current known path towards the peer, if any.
1118  */
1119 static struct MeshPeerPath *
1120 peer_get_best_path (const struct MeshPeerInfo *peer, const struct MeshTunnel *t)
1121 {
1122   struct MeshPeerPath *best_p;
1123   struct MeshPeerPath *p;
1124   unsigned int best_cost;
1125   unsigned int cost;
1126
1127   best_p = p = peer->path_head;
1128   best_cost = cost = p->length;
1129   while (NULL != p)
1130   {
1131     if ((cost = p->length) < best_cost)
1132     {
1133       best_cost = cost;
1134       best_p = p;
1135     }
1136         p = p->next;
1137   }
1138   return best_p;
1139 }
1140
1141 /**
1142   * Core callback to write a pre-constructed data packet to core buffer
1143   *
1144   * @param cls Closure (MeshTransmissionDescriptor with data in "data" member).
1145   * @param size Number of bytes available in buf.
1146   * @param buf Where the to write the message.
1147   *
1148   * @return number of bytes written to buf
1149   */
1150 static size_t
1151 send_core_data_raw (void *cls, size_t size, void *buf)
1152 {
1153   struct GNUNET_MessageHeader *msg = cls;
1154   size_t total_size;
1155
1156   GNUNET_assert (NULL != msg);
1157   total_size = ntohs (msg->size);
1158
1159   if (total_size > size)
1160   {
1161     GNUNET_break (0);
1162     return 0;
1163   }
1164   memcpy (buf, msg, total_size);
1165   GNUNET_free (cls);
1166   return total_size;
1167 }
1168
1169
1170 /**
1171  * Sends an already built message to a peer, properly registrating
1172  * all used resources.
1173  *
1174  * @param message Message to send. Function makes a copy of it.
1175  * @param peer Short ID of the neighbor whom to send the message.
1176  * @param t Tunnel on which this message is transmitted.
1177  */
1178 static void
1179 send_prebuilt_message (const struct GNUNET_MessageHeader *message,
1180                        GNUNET_PEER_Id peer,
1181                        struct MeshTunnel *t)
1182 {
1183   struct GNUNET_PeerIdentity id;
1184   struct MeshPeerInfo *neighbor;
1185   struct MeshPeerPath *p;
1186   void *data;
1187   size_t size;
1188   uint16_t type;
1189
1190 //   GNUNET_TRANSPORT_try_connect(); FIXME use?
1191
1192   if (0 == peer)
1193     return;
1194
1195   size = ntohs (message->size);
1196   data = GNUNET_malloc (size);
1197   memcpy (data, message, size);
1198   type = ntohs(message->type);
1199   if (GNUNET_MESSAGE_TYPE_MESH_UNICAST == type ||
1200       GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN == type)
1201   {
1202     struct GNUNET_MESH_Data *u;
1203
1204     u = (struct GNUNET_MESH_Data *) data;
1205     u->ttl = htonl (ntohl (u->ttl) - 1);
1206   }
1207   GNUNET_PEER_resolve (peer, &id);
1208   neighbor = peer_get (&id);
1209   for (p = neighbor->path_head; NULL != p; p = p->next)
1210   {
1211     if (2 >= p->length)
1212     {
1213       break;
1214     }
1215   }
1216   if (NULL == p)
1217   {
1218 #if MESH_DEBUG
1219     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1220                 "  %s IS NOT DIRECTLY CONNECTED\n",
1221                 GNUNET_i2s(&id));
1222     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1223                 "  PATHS TO %s:\n",
1224                 GNUNET_i2s(&id));
1225     for (p = neighbor->path_head; NULL != p; p = p->next)
1226     {
1227       struct GNUNET_PeerIdentity debug_id;
1228       unsigned int i;
1229
1230       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1231                   "    path with %u hops through:\n",
1232                   p->length);
1233       for (i = 0; i < p->length; i++)
1234       {
1235         GNUNET_PEER_resolve(p->peers[i], &debug_id);
1236         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1237                     "      hop %u: %s\n",
1238                     i, GNUNET_i2s(&debug_id));
1239       }
1240     }
1241 #endif
1242     GNUNET_break (0); // FIXME sometimes fails (testing disconnect?)
1243     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1244                     " no direct connection to %s\n",
1245                     GNUNET_i2s (&id));
1246     GNUNET_free (data);
1247     return;
1248   }
1249   if (GNUNET_MESSAGE_TYPE_MESH_PATH_ACK == type)
1250     type = 0;
1251   queue_add (data,
1252              type,
1253              size,
1254              neighbor,
1255              t);
1256 }
1257
1258
1259 /**
1260  * Sends a CREATE PATH message for a path to a peer, properly registrating
1261  * all used resources.
1262  *
1263  * @param t Tunnel for which the path is created.
1264  */
1265 static void
1266 send_create_path (struct MeshTunnel *t)
1267 {
1268   struct MeshPeerInfo *neighbor;
1269
1270   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send create path\n");
1271   neighbor = peer_get_short (t->next_hop);
1272   queue_add (t,
1273              GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE,
1274              sizeof (struct GNUNET_MESH_CreateTunnel) +
1275                 (t->path->length * sizeof (struct GNUNET_PeerIdentity)),
1276              neighbor,
1277              t);
1278   t->state = MESH_TUNNEL_WAITING;
1279 }
1280
1281
1282 /**
1283  * Sends a PATH ACK message in reponse to a received PATH_CREATE directed to us.
1284  *
1285  * @param t Tunnel which to confirm.
1286  */
1287 static void
1288 send_path_ack (struct MeshTunnel *t) 
1289 {
1290   struct MeshPeerInfo *peer;
1291
1292   peer = peer_get_short (t->prev_hop);
1293
1294   queue_add (t,
1295              GNUNET_MESSAGE_TYPE_MESH_PATH_ACK,
1296              sizeof (struct GNUNET_MESH_PathACK),
1297              peer,
1298              t);
1299 }
1300
1301
1302 /**
1303  * Try to establish a new connection to this peer in the given tunnel.
1304  * If the peer doesn't have any path to it yet, try to get one.
1305  * If the peer already has some path, send a CREATE PATH towards it.
1306  *
1307  * @param peer PeerInfo of the peer.
1308  * @param t Tunnel for which to create the path, if possible.
1309  */
1310 static void
1311 peer_connect (struct MeshPeerInfo *peer, struct MeshTunnel *t)
1312 {
1313   struct MeshPeerPath *p;
1314
1315   if (NULL != peer->path_head)
1316   {
1317     p = peer_get_best_path (peer, t);
1318     tunnel_use_path (t, p);
1319     send_create_path (t);
1320   }
1321   else if (NULL == peer->dhtget)
1322   {
1323     struct GNUNET_PeerIdentity id;
1324
1325     GNUNET_PEER_resolve (peer->id, &id);
1326     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1327                 "  Starting DHT GET for peer %s\n", GNUNET_i2s (&id));
1328     peer->dhtget = GNUNET_DHT_get_start (dht_handle,    /* handle */
1329                                          GNUNET_BLOCK_TYPE_MESH_PEER, /* type */
1330                                          &id.hashPubKey,     /* key to search */
1331                                          dht_replication_level, /* replication level */
1332                                          GNUNET_DHT_RO_RECORD_ROUTE |
1333                                          GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
1334                                          NULL,       /* xquery */
1335                                          0,     /* xquery bits */
1336                                          &dht_get_id_handler, peer);
1337     t->state = MESH_TUNNEL_SEARCHING;
1338   }
1339   else
1340   {
1341     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1342                 "There is no path but the DHT GET is already started.\n");
1343   }
1344 }
1345
1346
1347 /**
1348  * Destroy the peer_info and free any allocated resources linked to it
1349  *
1350  * @param pi The peer_info to destroy.
1351  *
1352  * @return GNUNET_OK on success
1353  */
1354 static int
1355 peer_info_destroy (struct MeshPeerInfo *pi)
1356 {
1357   struct GNUNET_PeerIdentity id;
1358   struct MeshPeerPath *p;
1359   struct MeshPeerPath *nextp;
1360   unsigned int i;
1361
1362   GNUNET_PEER_resolve (pi->id, &id);
1363   GNUNET_PEER_change_rc (pi->id, -1);
1364
1365   if (GNUNET_YES !=
1366       GNUNET_CONTAINER_multihashmap_remove (peers, &id.hashPubKey, pi))
1367   {
1368     GNUNET_break (0);
1369     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1370                 "removing peer %s, not in hashmap\n", GNUNET_i2s (&id));
1371   }
1372   if (NULL != pi->dhtget)
1373   {
1374     GNUNET_DHT_get_stop (pi->dhtget);
1375   }
1376   p = pi->path_head;
1377   while (NULL != p)
1378   {
1379     nextp = p->next;
1380     GNUNET_CONTAINER_DLL_remove (pi->path_head, pi->path_tail, p);
1381     path_destroy (p);
1382     p = nextp;
1383   }
1384   for (i = 0; i < pi->ntunnels; i++)
1385     tunnel_destroy_empty (pi->tunnels[i]);
1386   GNUNET_array_grow (pi->tunnels, pi->ntunnels, 0);
1387   GNUNET_free (pi);
1388   return GNUNET_OK;
1389 }
1390
1391
1392 /**
1393  * Remove all paths that rely on a direct connection between p1 and p2
1394  * from the peer itself and notify all tunnels about it.
1395  *
1396  * @param peer PeerInfo of affected peer.
1397  * @param p1 GNUNET_PEER_Id of one peer.
1398  * @param p2 GNUNET_PEER_Id of another peer that was connected to the first and
1399  *           no longer is.
1400  *
1401  * TODO: optimize (see below)
1402  */
1403 static void
1404 peer_remove_path (struct MeshPeerInfo *peer, GNUNET_PEER_Id p1,
1405                        GNUNET_PEER_Id p2)
1406 {
1407   struct MeshPeerPath *p;
1408   struct MeshPeerPath *next;
1409   struct MeshPeerInfo *peer_d;
1410   GNUNET_PEER_Id d;
1411   unsigned int destroyed;
1412   unsigned int i;
1413
1414   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer_info_remove_path\n");
1415   destroyed = 0;
1416   for (p = peer->path_head; NULL != p; p = next)
1417   {
1418     next = p->next;
1419     for (i = 0; i < (p->length - 1); i++)
1420     {
1421       if ((p->peers[i] == p1 && p->peers[i + 1] == p2) ||
1422           (p->peers[i] == p2 && p->peers[i + 1] == p1))
1423       {
1424         GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
1425         path_destroy (p);
1426         destroyed++;
1427         break;
1428       }
1429     }
1430   }
1431   if (0 == destroyed)
1432     return;
1433
1434   for (i = 0; i < peer->ntunnels; i++)
1435   {
1436     d = tunnel_notify_connection_broken (peer->tunnels[i], p1, p2);
1437     if (0 == d)
1438       continue;
1439
1440     peer_d = peer_get_short (d);
1441     next = peer_get_best_path (peer_d, peer->tunnels[i]);
1442     tunnel_use_path (peer->tunnels[i], next);
1443     peer_connect (peer_d, peer->tunnels[i]);
1444   }
1445   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer_info_remove_path END\n");
1446 }
1447
1448
1449 /**
1450  * Add the path to the peer and update the path used to reach it in case this
1451  * is the shortest.
1452  *
1453  * @param peer_info Destination peer to add the path to.
1454  * @param path New path to add. Last peer must be the peer in arg 1.
1455  *             Path will be either used of freed if already known.
1456  * @param trusted Do we trust that this path is real?
1457  */
1458 void
1459 peer_info_add_path (struct MeshPeerInfo *peer_info, struct MeshPeerPath *path,
1460                     int trusted)
1461 {
1462   struct MeshPeerPath *aux;
1463   unsigned int l;
1464   unsigned int l2;
1465
1466   if ((NULL == peer_info) || (NULL == path))
1467   {
1468     GNUNET_break (0);
1469     path_destroy (path);
1470     return;
1471   }
1472   if (path->peers[path->length - 1] != peer_info->id)
1473   {
1474     GNUNET_break (0);
1475     path_destroy (path);
1476     return;
1477   }
1478   if (2 >= path->length && GNUNET_NO == trusted)
1479   {
1480     /* Only allow CORE to tell us about direct paths */
1481     path_destroy (path);
1482     return;
1483   }
1484   for (l = 1; l < path->length; l++)
1485   {
1486     if (path->peers[l] == myid)
1487     {
1488       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shortening path by %u\n", l);
1489       for (l2 = 0; l2 < path->length - l; l2++)
1490       {
1491         path->peers[l2] = path->peers[l + l2];
1492       }
1493       path->length -= l;
1494       l = 1;
1495       path->peers =
1496           GNUNET_realloc (path->peers, path->length * sizeof (GNUNET_PEER_Id));
1497     }
1498   }
1499 #if MESH_DEBUG
1500   {
1501     struct GNUNET_PeerIdentity id;
1502
1503     GNUNET_PEER_resolve (peer_info->id, &id);
1504     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n",
1505                 path->length, GNUNET_i2s (&id));
1506   }
1507 #endif
1508   l = path_get_length (path);
1509   if (0 == l)
1510   {
1511     path_destroy (path);
1512     return;
1513   }
1514
1515   GNUNET_assert (peer_info->id == path->peers[path->length - 1]);
1516   for (aux = peer_info->path_head; aux != NULL; aux = aux->next)
1517   {
1518     l2 = path_get_length (aux);
1519     if (l2 > l)
1520     {
1521       GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head,
1522                                           peer_info->path_tail, aux, path);
1523       return;
1524     }
1525     else
1526     {
1527       if (l2 == l && memcmp (path->peers, aux->peers, l) == 0)
1528       {
1529         path_destroy (path);
1530         return;
1531       }
1532     }
1533   }
1534   GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head, peer_info->path_tail,
1535                                     path);
1536   return;
1537 }
1538
1539
1540 /**
1541  * Add the path to the origin peer and update the path used to reach it in case
1542  * this is the shortest.
1543  * The path is given in peer_info -> destination, therefore we turn the path
1544  * upside down first.
1545  *
1546  * @param peer_info Peer to add the path to, being the origin of the path.
1547  * @param path New path to add after being inversed.
1548  *             Path will be either used or freed.
1549  * @param trusted Do we trust that this path is real?
1550  */
1551 static void
1552 peer_info_add_path_to_origin (struct MeshPeerInfo *peer_info,
1553                               struct MeshPeerPath *path, int trusted)
1554 {
1555   path_invert (path);
1556   peer_info_add_path (peer_info, path, trusted);
1557 }
1558
1559
1560 /**
1561  * Add a tunnel to the list of tunnels a peer participates in.
1562  * Update the tunnel's destination.
1563  * 
1564  * @param p Peer to add to.
1565  * @param t Tunnel to add.
1566  */
1567 static void
1568 peer_info_add_tunnel (struct MeshPeerInfo *p, struct MeshTunnel *t)
1569 {
1570   if (0 != t->dest)
1571   {
1572     GNUNET_break (t->dest == p->id);
1573     return;
1574   }
1575   t->dest = p->id;
1576   GNUNET_PEER_change_rc (t->dest, 1);
1577   GNUNET_array_append (p->tunnels, p->ntunnels, t);
1578 }
1579
1580
1581 /**
1582  * Remove a tunnel from the list of tunnels a peer participates in.
1583  * Free the tunnel's destination.
1584  * 
1585  * @param p Peer to clean.
1586  * @param t Tunnel to remove.
1587  */
1588 static void
1589 peer_info_remove_tunnel (struct MeshPeerInfo *p, struct MeshTunnel *t)
1590 {
1591   unsigned int i;
1592
1593   if (t->dest == p->id)
1594   {
1595       GNUNET_PEER_change_rc (t->dest, -1);
1596       t->dest = 0;
1597   }
1598   for (i = 0; i < p->ntunnels; i++)
1599   {
1600     if (p->tunnels[i] == t)
1601     {
1602       p->tunnels[i] = p->tunnels[p->ntunnels - 1];
1603       GNUNET_array_grow (p->tunnels, p->ntunnels, p->ntunnels - 1);
1604       return;
1605     }
1606   }
1607 }
1608
1609
1610 /**
1611  * Function called if the connection to the peer has been stalled for a while,
1612  * possibly due to a missed ACK. Poll the peer about its ACK status.
1613  *
1614  * @param cls Closure (poll ctx).
1615  * @param tc TaskContext.
1616  */
1617 static void
1618 tunnel_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1619 {
1620   struct MeshFlowControl *fc = cls;
1621   struct GNUNET_MESH_Poll msg;
1622   struct MeshTunnel *t = fc->t;
1623   GNUNET_PEER_Id peer;
1624
1625   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1626   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1627   {
1628     return;
1629   }
1630
1631   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
1632   msg.header.size = htons (sizeof (msg));
1633   msg.tid = htonl (t->id.tid);
1634   GNUNET_PEER_resolve (t->id.oid, &msg.oid);
1635
1636   if (fc == &t->prev_fc)
1637   {
1638     peer = t->prev_hop;
1639   }
1640   else if (fc == &t->next_fc)
1641   {
1642     peer = t->next_hop;
1643   }
1644   else
1645   {
1646     GNUNET_break (0);
1647     return;
1648   }
1649   send_prebuilt_message (&msg.header, peer, t);
1650   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
1651   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
1652                                                 &tunnel_poll, fc);
1653 }
1654
1655
1656 /**
1657  * Build a PeerPath from the paths returned from the DHT, reversing the paths
1658  * to obtain a local peer -> destination path and interning the peer ids.
1659  *
1660  * @return Newly allocated and created path
1661  */
1662 static struct MeshPeerPath *
1663 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
1664                      unsigned int get_path_length,
1665                      const struct GNUNET_PeerIdentity *put_path,
1666                      unsigned int put_path_length)
1667 {
1668   struct MeshPeerPath *p;
1669   GNUNET_PEER_Id id;
1670   int i;
1671
1672   p = path_new (1);
1673   p->peers[0] = myid;
1674   GNUNET_PEER_change_rc (myid, 1);
1675   i = get_path_length;
1676   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   GET has %d hops.\n", i);
1677   for (i--; i >= 0; i--)
1678   {
1679     id = GNUNET_PEER_intern (&get_path[i]);
1680     if (p->length > 0 && id == p->peers[p->length - 1])
1681     {
1682       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
1683       GNUNET_PEER_change_rc (id, -1);
1684     }
1685     else
1686     {
1687       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from GET: %s.\n",
1688                   GNUNET_i2s (&get_path[i]));
1689       p->length++;
1690       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
1691       p->peers[p->length - 1] = id;
1692     }
1693   }
1694   i = put_path_length;
1695   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   PUT has %d hops.\n", i);
1696   for (i--; i >= 0; i--)
1697   {
1698     id = GNUNET_PEER_intern (&put_path[i]);
1699     if (id == myid)
1700     {
1701       /* PUT path went through us, so discard the path up until now and start
1702        * from here to get a much shorter (and loop-free) path.
1703        */
1704       path_destroy (p);
1705       p = path_new (0);
1706     }
1707     if (p->length > 0 && id == p->peers[p->length - 1])
1708     {
1709       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
1710       GNUNET_PEER_change_rc (id, -1);
1711     }
1712     else
1713     {
1714       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from PUT: %s.\n",
1715                   GNUNET_i2s (&put_path[i]));
1716       p->length++;
1717       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
1718       p->peers[p->length - 1] = id;
1719     }
1720   }
1721 #if MESH_DEBUG
1722   if (get_path_length > 0)
1723     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of GET: %s)\n",
1724                 GNUNET_i2s (&get_path[0]));
1725   if (put_path_length > 0)
1726     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of PUT: %s)\n",
1727                 GNUNET_i2s (&put_path[0]));
1728   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   In total: %d hops\n",
1729               p->length);
1730   for (i = 0; i < p->length; i++)
1731   {
1732     struct GNUNET_PeerIdentity peer_id;
1733
1734     GNUNET_PEER_resolve (p->peers[i], &peer_id);
1735     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "       %u: %s\n", p->peers[i],
1736                 GNUNET_i2s (&peer_id));
1737   }
1738 #endif
1739   return p;
1740 }
1741
1742
1743 /**
1744  * Adds a path to the peer_infos of all the peers in the path
1745  *
1746  * @param p Path to process.
1747  * @param confirmed Whether we know if the path works or not.
1748  */
1749 static void
1750 path_add_to_peers (struct MeshPeerPath *p, int confirmed)
1751 {
1752   unsigned int i;
1753
1754   /* TODO: invert and add */
1755   for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ;
1756   for (i++; i < p->length; i++)
1757   {
1758     struct MeshPeerInfo *aux;
1759     struct MeshPeerPath *copy;
1760
1761     aux = peer_get_short (p->peers[i]);
1762     copy = path_duplicate (p);
1763     copy->length = i + 1;
1764     peer_info_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed);
1765   }
1766 }
1767
1768
1769 /**
1770  * Send keepalive packets for a peer
1771  *
1772  * @param cls Closure (tunnel for which to send the keepalive).
1773  * @param tc Notification context.
1774  */
1775 static void
1776 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1777
1778
1779 /**
1780  * Search for a tunnel among the incoming tunnels
1781  *
1782  * @param tid the local id of the tunnel
1783  *
1784  * @return tunnel handler, NULL if doesn't exist
1785  */
1786 static struct MeshTunnel *
1787 tunnel_get_incoming (MESH_TunnelNumber tid)
1788 {
1789   struct GNUNET_HashCode hash;
1790
1791   GNUNET_assert (tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV);
1792   GMC_hash32 (tid, &hash);
1793   return GNUNET_CONTAINER_multihashmap_get (incoming_tunnels, &hash);
1794 }
1795
1796
1797 /**
1798  * Search for a tunnel among the tunnels for a client
1799  *
1800  * @param c the client whose tunnels to search in
1801  * @param tid the local id of the tunnel
1802  *
1803  * @return tunnel handler, NULL if doesn't exist
1804  */
1805 static struct MeshTunnel *
1806 tunnel_get_by_local_id (struct MeshClient *c, MESH_TunnelNumber tid)
1807 {
1808   if (tid >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
1809   {
1810     return tunnel_get_incoming (tid);
1811   }
1812   else
1813   {
1814     struct GNUNET_HashCode hash;
1815
1816     GMC_hash32 (tid, &hash);
1817     return GNUNET_CONTAINER_multihashmap_get (c->own_tunnels, &hash);
1818   }
1819 }
1820
1821
1822 /**
1823  * Search for a tunnel by global ID using PEER_ID
1824  *
1825  * @param pi owner of the tunnel
1826  * @param tid global tunnel number
1827  *
1828  * @return tunnel handler, NULL if doesn't exist
1829  */
1830 static struct MeshTunnel *
1831 tunnel_get_by_pi (GNUNET_PEER_Id pi, MESH_TunnelNumber tid)
1832 {
1833   struct MESH_TunnelID id;
1834   struct GNUNET_HashCode hash;
1835
1836   id.oid = pi;
1837   id.tid = tid;
1838
1839   GNUNET_CRYPTO_hash (&id, sizeof (struct MESH_TunnelID), &hash);
1840   return GNUNET_CONTAINER_multihashmap_get (tunnels, &hash);
1841 }
1842
1843
1844 /**
1845  * Search for a tunnel by global ID using full PeerIdentities
1846  *
1847  * @param oid owner of the tunnel
1848  * @param tid global tunnel number
1849  *
1850  * @return tunnel handler, NULL if doesn't exist
1851  */
1852 static struct MeshTunnel *
1853 tunnel_get (const struct GNUNET_PeerIdentity *oid, MESH_TunnelNumber tid)
1854 {
1855   return tunnel_get_by_pi (GNUNET_PEER_search (oid), tid);
1856 }
1857
1858
1859 /**
1860  * Add a client to a tunnel, initializing all needed data structures.
1861  * 
1862  * @param t Tunnel to which add the client.
1863  * @param c Client which to add to the tunnel.
1864  */
1865 static void
1866 tunnel_add_client (struct MeshTunnel *t, struct MeshClient *c)
1867 {
1868   struct GNUNET_HashCode hash;
1869
1870   if (NULL != t->client)
1871   {
1872     GNUNET_break(0);
1873     return;
1874   }
1875   GMC_hash32 (t->local_tid_dest, &hash);
1876   if (GNUNET_OK !=
1877       GNUNET_CONTAINER_multihashmap_put (c->incoming_tunnels, &hash, t,
1878                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
1879   {
1880     GNUNET_break (0);
1881     return;
1882   }
1883   if (GNUNET_OK !=
1884       GNUNET_CONTAINER_multihashmap_put (incoming_tunnels, &hash, t,
1885                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
1886   {
1887     GNUNET_break (0);
1888     return;
1889   }
1890   t->client = c;
1891 }
1892
1893
1894 static void
1895 tunnel_use_path (struct MeshTunnel *t, struct MeshPeerPath *p)
1896 {
1897   unsigned int own_pos;
1898
1899   for (own_pos = 0; own_pos < p->length; own_pos++)
1900   {
1901     if (p->peers[own_pos] == myid)
1902       break;
1903   }
1904   if (own_pos > p->length - 1)
1905   {
1906     GNUNET_break (0);
1907     return;
1908   }
1909
1910   if (own_pos < p->length - 1)
1911     t->next_hop = p->peers[own_pos + 1];
1912   else
1913     t->next_hop = p->peers[own_pos];
1914   GNUNET_PEER_change_rc (t->next_hop, 1);
1915   if (0 < own_pos)
1916     t->prev_hop = p->peers[own_pos - 1];
1917   else
1918     t->prev_hop = p->peers[0];
1919   GNUNET_PEER_change_rc (t->prev_hop, 1);
1920
1921   if (NULL != t->path)
1922     path_destroy (t->path);
1923   t->path = path_duplicate (p);
1924   if (0 == own_pos)
1925   {
1926     if (GNUNET_SCHEDULER_NO_TASK != t->maintenance_task)
1927       GNUNET_SCHEDULER_cancel (t->maintenance_task);
1928     t->maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_path_time,
1929                                                         &path_refresh, t);
1930   }
1931 }
1932
1933
1934 /**
1935  * Notifies a tunnel that a connection has broken that affects at least
1936  * some of its peers. Sends a notification towards the root of the tree.
1937  * In case the peer is the owner of the tree, notifies the client that owns
1938  * the tunnel and tries to reconnect.
1939  *
1940  * @param t Tunnel affected.
1941  * @param p1 Peer that got disconnected from p2.
1942  * @param p2 Peer that got disconnected from p1.
1943  *
1944  * @return Short ID of the peer disconnected (either p1 or p2).
1945  *         0 if the tunnel remained unaffected.
1946  */
1947 static GNUNET_PEER_Id
1948 tunnel_notify_connection_broken (struct MeshTunnel *t, GNUNET_PEER_Id p1,
1949                                  GNUNET_PEER_Id p2)
1950 {
1951 //   if (myid != p1 && myid != p2) FIXME
1952 //   {
1953 //     return;
1954 //   }
1955 // 
1956 //   if (tree_get_predecessor (t->tree) != 0)
1957 //   {
1958 //     /* We are the peer still connected, notify owner of the disconnection. */
1959 //     struct GNUNET_MESH_PathBroken msg;
1960 //     struct GNUNET_PeerIdentity neighbor;
1961 // 
1962 //     msg.header.size = htons (sizeof (msg));
1963 //     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN);
1964 //     GNUNET_PEER_resolve (t->id.oid, &msg.oid);
1965 //     msg.tid = htonl (t->id.tid);
1966 //     msg.peer1 = my_full_id;
1967 //     GNUNET_PEER_resolve (pid, &msg.peer2);
1968 //     GNUNET_PEER_resolve (tree_get_predecessor (t->tree), &neighbor);
1969 //     send_prebuilt_message (&msg.header, &neighbor, t);
1970 //   }
1971   return 0;
1972 }
1973
1974
1975 /**
1976  * Build a local ACK message and send it to a local client.
1977  * 
1978  * @param t Tunnel on which to send the ACK.
1979  * @param c Client to whom send the ACK.
1980  * @param ack Value of the ACK.
1981  */
1982 static void
1983 send_local_ack (struct MeshTunnel *t, struct MeshClient *c, uint32_t ack)
1984 {
1985   struct GNUNET_MESH_LocalAck msg;
1986
1987   msg.header.size = htons (sizeof (msg));
1988   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
1989   msg.tunnel_id = htonl (t->owner == c ? t->local_tid : t->local_tid_dest);
1990   msg.ack = htonl (ack); 
1991   GNUNET_SERVER_notification_context_unicast(nc,
1992                                               c->handle,
1993                                               &msg.header,
1994                                               GNUNET_NO);
1995 }
1996
1997 /**
1998  * Build an ACK message and queue it to send to the given peer.
1999  * 
2000  * @param t Tunnel on which to send the ACK.
2001  * @param peer Peer to whom send the ACK.
2002  * @param ack Value of the ACK.
2003  */
2004 static void
2005 send_ack (struct MeshTunnel *t, GNUNET_PEER_Id peer,  uint32_t ack)
2006 {
2007   struct GNUNET_MESH_ACK msg;
2008
2009   GNUNET_PEER_resolve (t->id.oid, &msg.oid);
2010   msg.header.size = htons (sizeof (msg));
2011   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_ACK);
2012   msg.pid = htonl (ack);
2013   msg.tid = htonl (t->id.tid);
2014
2015   send_prebuilt_message (&msg.header, peer, t);
2016 }
2017
2018
2019 /**
2020  * Send an ACK informing the predecessor about the available buffer space.
2021  * In case there is no predecessor, inform the owning client.
2022  * If buffering is off, send only on behalf of children or self if endpoint.
2023  * If buffering is on, send when sent to children and buffer space is free.
2024  * Note that although the name is fwd_ack, the FWD mean forward *traffic*,
2025  * the ACK itself goes "back" (towards root).
2026  * 
2027  * @param t Tunnel on which to send the ACK.
2028  * @param type Type of message that triggered the ACK transmission.
2029  */
2030 static void
2031 tunnel_send_fwd_ack (struct MeshTunnel *t, uint16_t type)
2032 {
2033   uint32_t ack;
2034
2035   /* Is it after unicast retransmission? */
2036   switch (type)
2037   {
2038     case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
2039       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2040                   "ACK due to FWD DATA retransmission\n");
2041       if (GNUNET_YES == t->nobuffer)
2042       {
2043         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, nobuffer\n");
2044         return;
2045       }
2046       break;
2047     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2048       if (NULL != t->owner && GNUNET_YES == t->reliable)
2049         return;
2050     case GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK:
2051       break;
2052     case GNUNET_MESSAGE_TYPE_MESH_PATH_ACK:
2053     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2054     case GNUNET_MESSAGE_TYPE_MESH_DATA_ACK:
2055       t->force_ack = GNUNET_YES;
2056       break;
2057     default:
2058       GNUNET_break (0);
2059   }
2060
2061   /* Check if we need to transmit the ACK */
2062   if (t->queue_max > t->next_fc.queue_n * 4 &&
2063       GMC_is_pid_bigger(t->prev_fc.last_ack_sent, t->prev_fc.last_pid_recv) &&
2064       GNUNET_NO == t->force_ack)
2065   {
2066     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, buffer free\n");
2067     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2068                 "  t->qmax: %u, t->qn: %u\n",
2069                 t->queue_max, t->next_fc.queue_n);
2070     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2071                 "  t->pid: %u, t->ack: %u\n",
2072                 t->prev_fc.last_pid_recv, t->prev_fc.last_ack_sent);
2073     return;
2074   }
2075
2076   /* Ok, ACK might be necessary, what PID to ACK? */
2077   ack = t->prev_fc.last_pid_recv + t->queue_max - t->next_fc.queue_n;
2078   if (ack == t->prev_fc.last_ack_sent && GNUNET_NO == t->force_ack)
2079   {
2080     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending FWD ACK, not needed\n");
2081     return;
2082   }
2083
2084   t->prev_fc.last_ack_sent = ack;
2085   if (NULL != t->owner)
2086     send_local_ack (t, t->owner, ack);
2087   else if (0 != t->prev_hop)
2088     send_ack (t, t->prev_hop, ack);
2089   else
2090     GNUNET_break (0);
2091   debug_fwd_ack++;
2092   t->force_ack = GNUNET_NO;
2093 }
2094
2095
2096 /**
2097  * Send an ACK informing the children node/client about the available
2098  * buffer space.
2099  * If buffering is off, send only on behalf of root (can be self).
2100  * If buffering is on, send when sent to predecessor and buffer space is free.
2101  * Note that although the name is bck_ack, the BCK mean backwards *traffic*,
2102  * the ACK itself goes "forward" (towards children/clients).
2103  * 
2104  * @param t Tunnel on which to send the ACK.
2105  * @param type Type of message that triggered the ACK transmission.
2106  */
2107 static void
2108 tunnel_send_bck_ack (struct MeshTunnel *t, uint16_t type)
2109 {
2110   uint32_t ack;
2111   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2112               "Sending BCK ACK on tunnel %u [%u] due to %s\n",
2113               t->id.oid, t->id.tid, GNUNET_MESH_DEBUG_M2S(type));
2114   /* Is it after data to_origin retransmission? */
2115   switch (type)
2116   {
2117     case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
2118       if (GNUNET_YES == t->nobuffer)
2119       {
2120         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2121                     "    Not sending ACK, nobuffer + traffic\n");
2122         return;
2123       }
2124       break;
2125     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2126       if (NULL != t->client && GNUNET_YES == t->reliable)
2127         return;
2128     case GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK:
2129       break;
2130     case GNUNET_MESSAGE_TYPE_MESH_PATH_ACK:
2131     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2132     case GNUNET_MESSAGE_TYPE_MESH_DATA_ACK:
2133       t->force_ack = GNUNET_YES;
2134       break;
2135     default:
2136       GNUNET_break (0);
2137   }
2138
2139   /* TODO: Check if we need to transmit the ACK (as in fwd) */
2140
2141   ack = t->next_fc.last_pid_recv + t->queue_max - t->prev_fc.queue_n;
2142
2143   if (t->next_fc.last_ack_sent == ack && GNUNET_NO == t->force_ack)
2144   {
2145     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2146                 "    Not sending ACK, not needed, last ack sent was %u\n",
2147                 t->next_fc.last_ack_sent);
2148     return;
2149   }
2150   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2151               "    Sending BCK ACK %u (last sent: %u)\n",
2152               ack, t->next_fc.last_ack_sent);
2153   t->next_fc.last_ack_sent = ack;
2154
2155   if (NULL != t->client)
2156     send_local_ack (t, t->client, ack);
2157   else if (0 != t->next_hop)
2158     send_ack (t, t->next_hop, ack);
2159   else
2160     GNUNET_break (0);
2161   t->force_ack = GNUNET_NO;
2162 }
2163
2164
2165 /**
2166  * Modify the unicast message TID from global to local and send to client.
2167  * 
2168  * @param t Tunnel on which to send the message.
2169  * @param msg Message to modify and send.
2170  */
2171 static void
2172 tunnel_send_client_ucast (struct MeshTunnel *t,
2173                           const struct GNUNET_MESH_Data *msg)
2174 {
2175   struct GNUNET_MESH_Data *copy;
2176   uint16_t size = ntohs (msg->header.size);
2177   char cbuf[size];
2178
2179   if (size < sizeof (struct GNUNET_MESH_Data) +
2180              sizeof (struct GNUNET_MessageHeader))
2181   {
2182     GNUNET_break_op (0);
2183     return;
2184   }
2185   if (NULL == t->client)
2186   {
2187     GNUNET_break (0);
2188     return;
2189   }
2190   copy = (struct GNUNET_MESH_Data *) cbuf;
2191   memcpy (copy, msg, size);
2192   copy->tid = htonl (t->local_tid_dest);
2193   GNUNET_SERVER_notification_context_unicast (nc, t->client->handle,
2194                                               &copy->header, GNUNET_NO);
2195 }
2196
2197
2198 /**
2199  * Modify the to_origin  message TID from global to local and send to client.
2200  * 
2201  * @param t Tunnel on which to send the message.
2202  * @param msg Message to modify and send.
2203  */
2204 static void
2205 tunnel_send_client_to_orig (struct MeshTunnel *t,
2206                             const struct GNUNET_MESH_Data *msg)
2207 {
2208   struct GNUNET_MESH_Data *copy;
2209   uint16_t size = ntohs (msg->header.size);
2210   char cbuf[size];
2211
2212   if (size < sizeof (struct GNUNET_MESH_Data) +
2213              sizeof (struct GNUNET_MessageHeader))
2214   {
2215     GNUNET_break_op (0);
2216     return;
2217   }
2218   if (NULL == t->owner)
2219   {
2220     GNUNET_break (0);
2221     return;
2222   }
2223   copy = (struct GNUNET_MESH_Data *) cbuf;
2224   memcpy (cbuf, msg, size);
2225   copy->tid = htonl (t->local_tid);
2226   GNUNET_SERVER_notification_context_unicast (nc, t->owner->handle,
2227                                               &copy->header, GNUNET_NO);
2228 }
2229
2230
2231 /**
2232  * We haven't received an ACK after a certain time: restransmit the message.
2233  *
2234  * @param cls Closure (MeshSentMessage with the message to restransmit)
2235  * @param tc TaskContext.
2236  */
2237 static void
2238 tunnel_retransmit_message (void *cls,
2239                            const struct GNUNET_SCHEDULER_TaskContext *tc)
2240 {
2241   struct MeshSentMessage *copy = cls;
2242   struct GNUNET_MESH_Data *payload;
2243   GNUNET_PEER_Id hop;
2244
2245   copy->retry_task = GNUNET_SCHEDULER_NO_TASK;
2246   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2247     return;
2248
2249   payload = (struct GNUNET_MESH_Data *) &copy[1];
2250   hop = copy->is_forward ? copy->t->next_hop : copy->t->prev_hop;
2251   send_prebuilt_message (&payload->header, hop, copy->t);
2252   GNUNET_STATISTICS_update (stats, "# unicast retransmitted", 1, GNUNET_NO);
2253   copy->retry_timer = GNUNET_TIME_STD_BACKOFF (copy->retry_timer);
2254   copy->retry_task = GNUNET_SCHEDULER_add_delayed (copy->retry_timer,
2255                                                    &tunnel_retransmit_message,
2256                                                    cls);
2257 }
2258
2259
2260 /**
2261  * @brief Re-initiate traffic to this peer if necessary.
2262  *
2263  * Check if there is traffic queued towards this peer
2264  * and the core transmit handle is NULL (traffic was stalled).
2265  * If so, call core tmt rdy.
2266  *
2267  * @param peer_id Short ID of peer to which initiate traffic.
2268  */
2269 static void
2270 peer_unlock_queue(GNUNET_PEER_Id peer_id)
2271 {
2272   struct MeshPeerInfo *peer;
2273   struct GNUNET_PeerIdentity id;
2274   struct MeshPeerQueue *q;
2275   size_t size;
2276
2277   peer = peer_get_short (peer_id);
2278   if (NULL != peer->core_transmit)
2279     return;
2280
2281   q = queue_get_next (peer);
2282   if (NULL == q)
2283   {
2284     /* Might br multicast traffic already sent to this particular peer but
2285      * not to other children in this tunnel.
2286      * This way t->queue_n would be > 0 but the queue of this particular peer
2287      * would be empty.
2288      */
2289     return;
2290   }
2291   size = q->size;
2292   GNUNET_PEER_resolve (peer->id, &id);
2293   peer->core_transmit =
2294         GNUNET_CORE_notify_transmit_ready(core_handle,
2295                                           0,
2296                                           0,
2297                                           GNUNET_TIME_UNIT_FOREVER_REL,
2298                                           &id,
2299                                           size,
2300                                           &queue_send,
2301                                           peer);
2302         return;
2303 }
2304
2305
2306 /**
2307  * Send a message to all peers and clients in this tunnel that the tunnel
2308  * is no longer valid. If some peer or client should not receive the message,
2309  * should be zero'ed out before calling this function.
2310  *
2311  * @param t The tunnel whose peers and clients to notify.
2312  */
2313 static void
2314 tunnel_send_destroy (struct MeshTunnel *t)
2315 {
2316   struct GNUNET_MESH_TunnelDestroy msg;
2317   struct GNUNET_PeerIdentity id;
2318
2319   msg.header.size = htons (sizeof (msg));
2320   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY);
2321   GNUNET_PEER_resolve (t->id.oid, &msg.oid);
2322   msg.tid = htonl (t->id.tid);
2323   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2324               "  sending tunnel destroy for tunnel: %s [%X]\n",
2325               GNUNET_i2s (&msg.oid), t->id.tid);
2326
2327   if (NULL == t->client && 0 != t->next_hop)
2328   {
2329     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  child: %u\n", t->next_hop);
2330     GNUNET_PEER_resolve (t->next_hop, &id);
2331     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2332                 "  sending forward to %s\n",
2333                 GNUNET_i2s (&id));
2334     send_prebuilt_message (&msg.header, t->next_hop, t);
2335   }
2336   if (NULL == t->owner && 0 != t->prev_hop)
2337   {
2338     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  parent: %u\n", t->prev_hop);
2339     GNUNET_PEER_resolve (t->prev_hop, &id);
2340     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2341                 "  sending back to %s\n",
2342                 GNUNET_i2s (&id));
2343     send_prebuilt_message (&msg.header, t->prev_hop, t);
2344   }
2345   if (NULL != t->owner)
2346   {
2347     send_client_tunnel_destroy (t->owner, t);
2348   }
2349   if (NULL != t->client)
2350   {
2351     send_client_tunnel_destroy (t->client, t);
2352   }
2353 }
2354
2355
2356 /**
2357  * Cancel all transmissions towards a neighbor that belongs to a certain tunnel.
2358  *
2359  * @param t Tunnel which to cancel.
2360  * @param neighbor Short ID of the neighbor to whom cancel the transmissions.
2361  */
2362 static void
2363 peer_cancel_queues (GNUNET_PEER_Id neighbor, struct MeshTunnel *t)
2364 {
2365   struct MeshPeerInfo *peer_info;
2366   struct MeshPeerQueue *pq;
2367   struct MeshPeerQueue *next;
2368
2369   if (0 == neighbor)
2370     return; /* Was local peer, 0'ed in tunnel_destroy_iterator */
2371   peer_info = peer_get_short (neighbor);
2372   for (pq = peer_info->queue_head; NULL != pq; pq = next)
2373   {
2374     next = pq->next;
2375     if (pq->tunnel == t)
2376     {
2377       if (GNUNET_MESSAGE_TYPE_MESH_UNICAST == pq->type ||
2378           GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN == pq->type)
2379       {
2380         /* Should have been removed on destroy children */
2381         GNUNET_break (0);
2382       }
2383       queue_destroy (pq, GNUNET_YES);
2384     }
2385   }
2386   if (NULL == peer_info->queue_head && NULL != peer_info->core_transmit)
2387   {
2388     GNUNET_CORE_notify_transmit_ready_cancel(peer_info->core_transmit);
2389     peer_info->core_transmit = NULL;
2390   }
2391 }
2392
2393
2394 /**
2395  * Destroy the tunnel.
2396  * 
2397  * This function does not generate any warning traffic to clients or peers.
2398  * 
2399  * Tasks:
2400  * Remove the tunnel from peer_info's and clients' hashmaps.
2401  * Cancel messages belonging to this tunnel queued to neighbors.
2402  * Free any allocated resources linked to the tunnel.
2403  *
2404  * @param t the tunnel to destroy
2405  *
2406  * @return GNUNET_OK on success
2407  */
2408 static int
2409 tunnel_destroy (struct MeshTunnel *t)
2410 {
2411   struct MeshClient *c;
2412   struct GNUNET_HashCode hash;
2413   int r;
2414
2415   if (NULL == t)
2416     return GNUNET_OK;
2417
2418   r = GNUNET_OK;
2419   c = t->owner;
2420 #if MESH_DEBUG
2421   {
2422     struct GNUNET_PeerIdentity id;
2423
2424     GNUNET_PEER_resolve (t->id.oid, &id);
2425     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "destroying tunnel %s [%x]\n",
2426                 GNUNET_i2s (&id), t->id.tid);
2427     if (NULL != c)
2428       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
2429   }
2430 #endif
2431
2432   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2433   if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (tunnels, &hash, t))
2434   {
2435     GNUNET_break (0);
2436     r = GNUNET_SYSERR;
2437   }
2438
2439   if (NULL != c)
2440   {
2441     GMC_hash32 (t->local_tid, &hash);
2442     if (GNUNET_YES !=
2443         GNUNET_CONTAINER_multihashmap_remove (c->own_tunnels, &hash, t))
2444     {
2445       GNUNET_break (0);
2446       r = GNUNET_SYSERR;
2447     }
2448   }
2449
2450   if (NULL != t->client)
2451   {
2452     c = t->client;
2453     GMC_hash32 (t->local_tid_dest, &hash);
2454     if (GNUNET_YES !=
2455           GNUNET_CONTAINER_multihashmap_remove (c->incoming_tunnels, &hash, t))
2456     {
2457       GNUNET_break (0);
2458       r = GNUNET_SYSERR;
2459     }
2460     if (GNUNET_YES != 
2461       GNUNET_CONTAINER_multihashmap_remove (incoming_tunnels, &hash, t))
2462     {
2463       GNUNET_break (0);
2464       r = GNUNET_SYSERR;
2465     }
2466   }
2467
2468   if (0 != t->prev_hop)
2469   {
2470     peer_cancel_queues (t->prev_hop, t);
2471     GNUNET_PEER_change_rc (t->prev_hop, -1);
2472   }
2473   if (0 != t->next_hop)
2474   {
2475     peer_cancel_queues (t->next_hop, t);
2476     GNUNET_PEER_change_rc (t->next_hop, -1);
2477   }
2478   if (0 != t->dest) {
2479       peer_info_remove_tunnel (peer_get_short (t->dest), t);
2480   }
2481
2482   if (GNUNET_SCHEDULER_NO_TASK != t->maintenance_task)
2483     GNUNET_SCHEDULER_cancel (t->maintenance_task);
2484
2485   n_tunnels--;
2486   GNUNET_STATISTICS_update (stats, "# tunnels", -1, GNUNET_NO);
2487   path_destroy (t->path);
2488   GNUNET_free (t);
2489   return r;
2490 }
2491
2492 /**
2493  * Tunnel is empty: destroy it.
2494  * 
2495  * Notifies all participants (peers, cleints) about the destruction.
2496  * 
2497  * @param t Tunnel to destroy. 
2498  */
2499 static void
2500 tunnel_destroy_empty (struct MeshTunnel *t)
2501 {
2502   #if MESH_DEBUG
2503   {
2504     struct GNUNET_PeerIdentity id;
2505
2506     GNUNET_PEER_resolve (t->id.oid, &id);
2507     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2508                 "executing destruction of empty tunnel %s [%X]\n",
2509                 GNUNET_i2s (&id), t->id.tid);
2510   }
2511   #endif
2512
2513   if (GNUNET_NO == t->destroy)
2514     tunnel_send_destroy (t);
2515   if (0 == t->pending_messages)
2516     tunnel_destroy (t);
2517   else
2518     t->destroy = GNUNET_YES;
2519 }
2520
2521 /**
2522  * Initialize a Flow Control structure to the initial state.
2523  * 
2524  * @param fc Flow Control structure to initialize.
2525  */
2526 static void
2527 fc_init (struct MeshFlowControl *fc)
2528 {
2529   fc->last_pid_sent = (uint32_t) -1; /* Next (expected) = 0 */
2530   fc->last_pid_recv = (uint32_t) -1;
2531   fc->last_ack_sent = (uint32_t) -1; /* No traffic allowed yet */
2532   fc->last_ack_recv = (uint32_t) -1;
2533   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2534   fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
2535   fc->queue_n = 0;
2536 }
2537
2538 /**
2539  * Create a new tunnel
2540  * 
2541  * @param owner Who is the owner of the tunnel (short ID).
2542  * @param tid Tunnel Number of the tunnel.
2543  * @param client Clients that owns the tunnel, NULL for foreign tunnels.
2544  * @param local Tunnel Number for the tunnel, for the client point of view.
2545  * 
2546  * @return A new initialized tunnel. NULL on error.
2547  */
2548 static struct MeshTunnel *
2549 tunnel_new (GNUNET_PEER_Id owner,
2550             MESH_TunnelNumber tid,
2551             struct MeshClient *client,
2552             MESH_TunnelNumber local)
2553 {
2554   struct MeshTunnel *t;
2555   struct GNUNET_HashCode hash;
2556
2557   if (n_tunnels >= max_tunnels && NULL == client)
2558     return NULL;
2559
2560   t = GNUNET_malloc (sizeof (struct MeshTunnel));
2561   t->id.oid = owner;
2562   t->id.tid = tid;
2563   t->queue_max = (max_msgs_queue / max_tunnels) + 1;
2564   t->owner = client;
2565   fc_init (&t->next_fc);
2566   fc_init (&t->prev_fc);
2567   t->local_tid = local;
2568   n_tunnels++;
2569   GNUNET_STATISTICS_update (stats, "# tunnels", 1, GNUNET_NO);
2570
2571   GNUNET_CRYPTO_hash (&t->id, sizeof (struct MESH_TunnelID), &hash);
2572   if (GNUNET_OK !=
2573       GNUNET_CONTAINER_multihashmap_put (tunnels, &hash, t,
2574                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
2575   {
2576     GNUNET_break (0);
2577     tunnel_destroy (t);
2578     if (NULL != client)
2579     {
2580       GNUNET_break (0);
2581       GNUNET_SERVER_receive_done (client->handle, GNUNET_SYSERR);
2582     }
2583     return NULL;
2584   }
2585
2586   if (NULL != client)
2587   {
2588     GMC_hash32 (t->local_tid, &hash);
2589     if (GNUNET_OK !=
2590         GNUNET_CONTAINER_multihashmap_put (client->own_tunnels, &hash, t,
2591                                           GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
2592     {
2593       tunnel_destroy (t);
2594       GNUNET_break (0);
2595       GNUNET_SERVER_receive_done (client->handle, GNUNET_SYSERR);
2596       return NULL;
2597     }
2598   }
2599
2600   return t;
2601 }
2602
2603
2604 /**
2605  * Set options in a tunnel, extracted from a bit flag field
2606  * 
2607  * @param t Tunnel to set options to.
2608  * @param options Bit array in host byte order.
2609  */
2610 static void
2611 tunnel_set_options (struct MeshTunnel *t, uint32_t options)
2612 {
2613   t->nobuffer = options & GNUNET_MESH_OPTION_NOBUFFER;
2614   t->reliable = options & GNUNET_MESH_OPTION_RELIABLE;
2615 }
2616
2617
2618 /**
2619  * Iterator for deleting each tunnel whose client endpoint disconnected.
2620  *
2621  * @param cls Closure (client that has disconnected).
2622  * @param key The hash of the local tunnel id (used to access the hashmap).
2623  * @param value The value stored at the key (tunnel to destroy).
2624  *
2625  * @return GNUNET_OK, keep iterating.
2626  */
2627 static int
2628 tunnel_destroy_iterator (void *cls,
2629                          const struct GNUNET_HashCode * key,
2630                          void *value)
2631 {
2632   struct MeshTunnel *t = value;
2633   struct MeshClient *c = cls;
2634
2635   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2636               " Tunnel %X / %X destroy, due to client %u shutdown.\n",
2637               t->local_tid, t->local_tid_dest, c->id);
2638   client_delete_tunnel (c, t);
2639   if (c == t->client)
2640   {
2641     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is destination.\n", c->id);
2642     t->client = NULL;
2643     if (0 != t->next_hop) { /* destroy could come before a path is used */
2644         GNUNET_PEER_change_rc (t->next_hop, -1);
2645         t->next_hop = 0;
2646     }
2647   }
2648   else if (c == t->owner)
2649   {
2650     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is owner.\n", c->id);
2651     t->owner = NULL;
2652     if (0 != t->prev_hop) { /* destroy could come before a path is used */
2653         GNUNET_PEER_change_rc (t->prev_hop, -1);
2654         t->prev_hop = 0;
2655     }
2656   }
2657   else
2658   {
2659     GNUNET_break (0);
2660   }
2661   tunnel_destroy_empty (t);
2662
2663   return GNUNET_OK;
2664 }
2665
2666
2667 /**
2668  * Timeout function, destroys tunnel if called
2669  *
2670  * @param cls Closure (tunnel to destroy).
2671  * @param tc TaskContext
2672  */
2673 static void
2674 tunnel_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2675 {
2676   struct MeshTunnel *t = cls;
2677   struct GNUNET_PeerIdentity id;
2678
2679   t->maintenance_task = GNUNET_SCHEDULER_NO_TASK;
2680   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2681     return;
2682   GNUNET_PEER_resolve(t->id.oid, &id);
2683   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2684               "Tunnel %s [%X] timed out. Destroying.\n",
2685               GNUNET_i2s(&id), t->id.tid);
2686   if (NULL != t->client)
2687     send_client_tunnel_destroy (t->client, t);
2688   tunnel_destroy (t); /* Do not notify other */
2689 }
2690
2691
2692 /**
2693  * Resets the tunnel timeout. Starts it if no timeout was running.
2694  *
2695  * @param t Tunnel whose timeout to reset.
2696  *
2697  * TODO use heap to improve efficiency of scheduler.
2698  */
2699 static void
2700 tunnel_reset_timeout (struct MeshTunnel *t)
2701 {
2702   if (NULL != t->owner || 0 != t->local_tid || 0 == t->prev_hop)
2703     return;
2704   if (GNUNET_SCHEDULER_NO_TASK != t->maintenance_task)
2705     GNUNET_SCHEDULER_cancel (t->maintenance_task);
2706   t->maintenance_task =
2707       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
2708                                     (refresh_path_time, 4), &tunnel_timeout, t);
2709 }
2710
2711
2712 /******************************************************************************/
2713 /****************      MESH NETWORK HANDLER HELPERS     ***********************/
2714 /******************************************************************************/
2715
2716 /**
2717  * Function to send a create path packet to a peer.
2718  *
2719  * @param cls closure
2720  * @param size number of bytes available in buf
2721  * @param buf where the callee should write the message
2722  * @return number of bytes written to buf
2723  */
2724 static size_t
2725 send_core_path_create (void *cls, size_t size, void *buf)
2726 {
2727   struct MeshTunnel *t = cls;
2728   struct GNUNET_MESH_CreateTunnel *msg;
2729   struct GNUNET_PeerIdentity *peer_ptr;
2730   struct MeshPeerPath *p = t->path;
2731   size_t size_needed;
2732   uint32_t opt;
2733   int i;
2734
2735   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CREATE PATH sending...\n");
2736   size_needed =
2737       sizeof (struct GNUNET_MESH_CreateTunnel) +
2738       p->length * sizeof (struct GNUNET_PeerIdentity);
2739
2740   if (size < size_needed || NULL == buf)
2741   {
2742     GNUNET_break (0);
2743     return 0;
2744   }
2745   msg = (struct GNUNET_MESH_CreateTunnel *) buf;
2746   msg->header.size = htons (size_needed);
2747   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE);
2748   msg->tid = ntohl (t->id.tid);
2749
2750   opt = 0;
2751   if (GNUNET_YES == t->nobuffer)
2752     opt |= GNUNET_MESH_OPTION_NOBUFFER;
2753   if (GNUNET_YES == t->reliable)
2754     opt |= GNUNET_MESH_OPTION_RELIABLE;
2755   msg->opt = htonl (opt);
2756   msg->port = htonl (t->port);
2757
2758   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
2759   for (i = 0; i < p->length; i++)
2760   {
2761     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
2762   }
2763
2764   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2765               "CREATE PATH (%u bytes long) sent!\n", size_needed);
2766   return size_needed;
2767 }
2768
2769
2770 /**
2771  * Creates a path ack message in buf and frees all unused resources.
2772  *
2773  * @param cls closure (MeshTransmissionDescriptor)
2774  * @param size number of bytes available in buf
2775  * @param buf where the callee should write the message
2776  * @return number of bytes written to buf
2777  */
2778 static size_t
2779 send_core_path_ack (void *cls, size_t size, void *buf)
2780 {
2781   struct MeshTunnel *t = cls;
2782   struct GNUNET_MESH_PathACK *msg = buf;
2783
2784   GNUNET_assert (NULL != t);
2785   if (sizeof (struct GNUNET_MESH_PathACK) > size)
2786   {
2787     GNUNET_break (0);
2788     return 0;
2789   }
2790   t->prev_fc.last_ack_sent = t->nobuffer ? 0 : t->queue_max - 1;
2791   msg->header.size = htons (sizeof (struct GNUNET_MESH_PathACK));
2792   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_ACK);
2793   GNUNET_PEER_resolve (t->id.oid, &msg->oid);
2794   msg->tid = htonl (t->id.tid);
2795   msg->peer_id = my_full_id;
2796   msg->ack = htonl (t->prev_fc.last_ack_sent);
2797
2798   /* TODO add signature */
2799
2800   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "PATH ACK sent!\n");
2801   return sizeof (struct GNUNET_MESH_PathACK);
2802 }
2803
2804
2805 /**
2806  * Free a transmission that was already queued with all resources
2807  * associated to the request.
2808  *
2809  * @param queue Queue handler to cancel.
2810  * @param clear_cls Is it necessary to free associated cls?
2811  */
2812 static void
2813 queue_destroy (struct MeshPeerQueue *queue, int clear_cls)
2814 {
2815   struct MeshFlowControl *fc;
2816
2817   if (GNUNET_YES == clear_cls)
2818   {
2819     switch (queue->type)
2820     {
2821       case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY:
2822         GNUNET_log (GNUNET_ERROR_TYPE_INFO, "   cancelling TUNNEL_DESTROY\n");
2823         GNUNET_break (GNUNET_YES == queue->tunnel->destroy);
2824         /* fall through */
2825       case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
2826       case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
2827       case GNUNET_MESSAGE_TYPE_MESH_ACK:
2828       case GNUNET_MESSAGE_TYPE_MESH_POLL:
2829       case GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE:
2830         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2831                     "   prebuilt message\n");
2832         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2833                     "   type %s\n",
2834                     GNUNET_MESH_DEBUG_M2S (queue->type));
2835         break;
2836       case GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE:
2837         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   type create path\n");
2838         break;
2839       default:
2840         GNUNET_break (0);
2841         GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2842                     "   type %s unknown!\n",
2843                     GNUNET_MESH_DEBUG_M2S (queue->type));
2844     }
2845     GNUNET_free_non_null (queue->cls);
2846   }
2847   GNUNET_CONTAINER_DLL_remove (queue->peer->queue_head,
2848                                queue->peer->queue_tail,
2849                                queue);
2850
2851   /* Delete from appropriate fc in the tunnel */
2852   if (GNUNET_MESSAGE_TYPE_MESH_UNICAST == queue->type ||
2853       GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN == queue->type )
2854   {
2855     if (queue->peer->id == queue->tunnel->prev_hop)
2856       fc = &queue->tunnel->prev_fc;
2857     else if (queue->peer->id == queue->tunnel->next_hop)
2858       fc = &queue->tunnel->next_fc;
2859     else
2860     {
2861       GNUNET_break (0);
2862       return;
2863     }
2864     fc->queue_n--;
2865   }
2866   GNUNET_free (queue);
2867 }
2868
2869
2870 /**
2871  * @brief Get the next transmittable message from the queue.
2872  *
2873  * This will be the head, except in the case of being a data packet
2874  * not allowed by the destination peer.
2875  *
2876  * @param peer Destination peer.
2877  *
2878  * @return The next viable MeshPeerQueue element to send to that peer.
2879  *         NULL when there are no transmittable messages.
2880  */
2881 struct MeshPeerQueue *
2882 queue_get_next (const struct MeshPeerInfo *peer)
2883 {
2884   struct MeshPeerQueue *q;
2885
2886   struct GNUNET_MESH_Data *dmsg;
2887   struct MeshTunnel* t;
2888   uint32_t pid;
2889   uint32_t ack;
2890
2891   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   selecting message\n");
2892   for (q = peer->queue_head; NULL != q; q = q->next)
2893   {
2894     t = q->tunnel;
2895     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2896                 "*     %s\n",
2897                 GNUNET_MESH_DEBUG_M2S (q->type));
2898     dmsg = (struct GNUNET_MESH_Data *) q->cls;
2899     pid = ntohl (dmsg->pid);
2900     switch (q->type)
2901     {
2902       case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
2903         ack = t->next_fc.last_ack_recv;
2904         break;
2905       case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
2906         ack = t->prev_fc.last_ack_recv;
2907         break;
2908       default:
2909         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2910                     "*   OK!\n");
2911         return q;
2912     }
2913     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2914                 "*     ACK: %u, PID: %u\n",
2915                 ack, pid);
2916     if (GNUNET_NO == GMC_is_pid_bigger (pid, ack))
2917     {
2918       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2919                   "*   OK!\n");
2920       return q;
2921     }
2922     else
2923     {
2924       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2925                   "*     NEXT!\n");
2926     }
2927   }
2928   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2929                 "*   nothing found\n");
2930   return NULL;
2931 }
2932
2933
2934 static size_t
2935 queue_send (void *cls, size_t size, void *buf)
2936 {
2937   struct MeshPeerInfo *peer = cls;
2938   struct GNUNET_MessageHeader *msg;
2939   struct MeshPeerQueue *queue;
2940   struct MeshTunnel *t;
2941   struct GNUNET_PeerIdentity dst_id;
2942   struct MeshFlowControl *fc;
2943   size_t data_size;
2944   uint32_t pid;
2945   uint16_t type;
2946
2947   peer->core_transmit = NULL;
2948
2949   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Queue send\n");
2950   queue = queue_get_next (peer);
2951
2952   /* Queue has no internal mesh traffic nor sendable payload */
2953   if (NULL == queue)
2954   {
2955     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   not ready, return\n");
2956     if (NULL == peer->queue_head)
2957       GNUNET_break (0); /* Core tmt_rdy should've been canceled */
2958     return 0;
2959   }
2960   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   not empty\n");
2961
2962   GNUNET_PEER_resolve (peer->id, &dst_id);
2963   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2964               "*   towards %s\n",
2965               GNUNET_i2s (&dst_id));
2966   /* Check if buffer size is enough for the message */
2967   if (queue->size > size)
2968   {
2969       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2970                   "*   not enough room, reissue\n");
2971       peer->core_transmit =
2972           GNUNET_CORE_notify_transmit_ready (core_handle,
2973                                              GNUNET_NO,
2974                                              0,
2975                                              GNUNET_TIME_UNIT_FOREVER_REL,
2976                                              &dst_id,
2977                                              queue->size,
2978                                              &queue_send,
2979                                              peer);
2980       return 0;
2981   }
2982   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   size ok\n");
2983
2984   t = queue->tunnel;
2985   GNUNET_assert (0 < t->pending_messages);
2986   t->pending_messages--;
2987   type = 0;
2988
2989   /* Fill buf */
2990   switch (queue->type)
2991   {
2992     case 0:
2993     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2994     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2995     case GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN:
2996     case GNUNET_MESSAGE_TYPE_MESH_PATH_DESTROY:
2997     case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY:
2998     case GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE:
2999       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3000                   "*   raw: %s\n",
3001                   GNUNET_MESH_DEBUG_M2S (queue->type));
3002       /* Fall through */
3003     case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
3004     case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
3005       data_size = send_core_data_raw (queue->cls, size, buf);
3006       msg = (struct GNUNET_MessageHeader *) buf;
3007       type = ntohs (msg->type);
3008       break;
3009     case GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE:
3010       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path create\n");
3011       data_size = send_core_path_create (queue->cls, size, buf);
3012       break;
3013     case GNUNET_MESSAGE_TYPE_MESH_PATH_ACK:
3014       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path ack\n");
3015       data_size = send_core_path_ack (queue->cls, size, buf);
3016       break;
3017     default:
3018       GNUNET_break (0);
3019       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3020                   "*   type unknown: %u\n",
3021                   queue->type);
3022       data_size = 0;
3023   }
3024
3025   if (0 < drop_percent &&
3026       GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent)
3027   {
3028     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3029                 "Dropping message of type %s\n",
3030                 GNUNET_MESH_DEBUG_M2S(queue->type));
3031     data_size = 0;
3032   }
3033   /* Free queue, but cls was freed by send_core_* */
3034   queue_destroy (queue, GNUNET_NO);
3035
3036   /* Send ACK if needed, after accounting for sent ID in fc->queue_n */
3037   pid = ((struct GNUNET_MESH_Data *) buf)->pid;
3038   switch (type)
3039   {
3040     case GNUNET_MESSAGE_TYPE_MESH_UNICAST:
3041       t->next_fc.last_pid_sent = pid;
3042       tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_UNICAST);
3043       break;
3044     case GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN:
3045       t->prev_fc.last_pid_sent = pid;
3046       tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN);
3047       break;
3048     default:
3049       break;
3050   }
3051
3052   if (GNUNET_YES == t->destroy && 0 == t->pending_messages)
3053   {
3054     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  destroying tunnel!\n");
3055     tunnel_destroy (t);
3056   }
3057
3058   /* If more data in queue, send next */
3059   queue = queue_get_next (peer);
3060   if (NULL != queue)
3061   {
3062       struct GNUNET_PeerIdentity id;
3063
3064       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   more data!\n");
3065       GNUNET_PEER_resolve (peer->id, &id);
3066       peer->core_transmit =
3067           GNUNET_CORE_notify_transmit_ready(core_handle,
3068                                             0,
3069                                             0,
3070                                             GNUNET_TIME_UNIT_FOREVER_REL,
3071                                             &id,
3072                                             queue->size,
3073                                             &queue_send,
3074                                             peer);
3075   }
3076   else if (NULL != peer->queue_head)
3077   {
3078     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3079                 "*   %s stalled\n",
3080                 GNUNET_i2s (&my_full_id));
3081     if (peer->id == t->next_hop)
3082       fc = &t->next_fc;
3083     else if (peer->id == t->prev_hop)
3084       fc = &t->prev_fc;
3085     else
3086     {
3087       GNUNET_break (0);
3088       fc = NULL;
3089     }
3090     if (NULL != fc && GNUNET_SCHEDULER_NO_TASK == fc->poll_task)
3091     {
3092       fc->t = t;
3093       fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
3094                                                     &tunnel_poll, fc);
3095     }
3096   }
3097   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  Return %d\n", data_size);
3098   return data_size;
3099 }
3100
3101
3102 /**
3103  * @brief Queue and pass message to core when possible.
3104  * 
3105  * If type is payload (UNICAST, TO_ORIGIN) checks for queue status and
3106  * accounts for it. In case the queue is full, the message is dropped and
3107  * a break issued.
3108  * 
3109  * Otherwise, message is treated as internal and allowed to go regardless of 
3110  * queue status.
3111  *
3112  * @param cls Closure (@c type dependant). It will be used by queue_send to
3113  *            build the message to be sent if not already prebuilt.
3114  * @param type Type of the message, 0 for a raw message.
3115  * @param size Size of the message.
3116  * @param dst Neighbor to send message to.
3117  * @param t Tunnel this message belongs to.
3118  */
3119 static void
3120 queue_add (void *cls, uint16_t type, size_t size,
3121            struct MeshPeerInfo *dst, struct MeshTunnel *t)
3122 {
3123   struct MeshPeerQueue *queue;
3124   struct GNUNET_PeerIdentity id;
3125   unsigned int *n;
3126
3127   n = NULL;
3128   if (GNUNET_MESSAGE_TYPE_MESH_UNICAST == type)
3129   {
3130     n = &t->next_fc.queue_n;
3131   }
3132   else if (GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN == type)
3133   {
3134     n = &t->prev_fc.queue_n;
3135   }
3136   if (NULL != n)
3137   {
3138     if (*n >= t->queue_max)
3139     {
3140       GNUNET_break(0);
3141       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3142                   "queue full: %u/%u\n",
3143                   *n, t->queue_max);
3144       GNUNET_STATISTICS_update(stats,
3145                                "# messages dropped (buffer full)",
3146                                1, GNUNET_NO);
3147       return; /* Drop message */
3148     }
3149     (*n)++;
3150   }
3151   queue = GNUNET_malloc (sizeof (struct MeshPeerQueue));
3152   queue->cls = cls;
3153   queue->type = type;
3154   queue->size = size;
3155   queue->peer = dst;
3156   queue->tunnel = t;
3157   GNUNET_CONTAINER_DLL_insert_tail (dst->queue_head, dst->queue_tail, queue);
3158   if (NULL == dst->core_transmit)
3159   {
3160     GNUNET_PEER_resolve (dst->id, &id);
3161     dst->core_transmit =
3162         GNUNET_CORE_notify_transmit_ready (core_handle,
3163                                            0,
3164                                            0,
3165                                            GNUNET_TIME_UNIT_FOREVER_REL,
3166                                            &id,
3167                                            size,
3168                                            &queue_send,
3169                                            dst);
3170   }
3171   t->pending_messages++;
3172 }
3173
3174
3175 /******************************************************************************/
3176 /********************      MESH NETWORK HANDLERS     **************************/
3177 /******************************************************************************/
3178
3179
3180 /**
3181  * Core handler for path creation
3182  *
3183  * @param cls closure
3184  * @param message message
3185  * @param peer peer identity this notification is about
3186  *
3187  * @return GNUNET_OK to keep the connection open,
3188  *         GNUNET_SYSERR to close it (signal serious error)
3189  */
3190 static int
3191 handle_mesh_path_create (void *cls, const struct GNUNET_PeerIdentity *peer,
3192                          const struct GNUNET_MessageHeader *message)
3193 {
3194   unsigned int own_pos;
3195   uint16_t size;
3196   uint16_t i;
3197   MESH_TunnelNumber tid;
3198   struct GNUNET_MESH_CreateTunnel *msg;
3199   struct GNUNET_PeerIdentity *pi;
3200   struct MeshPeerPath *path;
3201   struct MeshPeerInfo *dest_peer_info;
3202   struct MeshPeerInfo *orig_peer_info;
3203   struct MeshTunnel *t;
3204
3205   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3206               "Received a path create msg [%s]\n",
3207               GNUNET_i2s (&my_full_id));
3208   size = ntohs (message->size);
3209   if (size < sizeof (struct GNUNET_MESH_CreateTunnel))
3210   {
3211     GNUNET_break_op (0);
3212     return GNUNET_OK;
3213   }
3214
3215   size -= sizeof (struct GNUNET_MESH_CreateTunnel);
3216   if (size % sizeof (struct GNUNET_PeerIdentity))
3217   {
3218     GNUNET_break_op (0);
3219     return GNUNET_OK;
3220   }
3221   size /= sizeof (struct GNUNET_PeerIdentity);
3222   if (size < 1)
3223   {
3224     GNUNET_break_op (0);
3225     return GNUNET_OK;
3226   }
3227   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
3228   msg = (struct GNUNET_MESH_CreateTunnel *) message;
3229
3230   tid = ntohl (msg->tid);
3231   pi = (struct GNUNET_PeerIdentity *) &msg[1];
3232   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3233               "    path is for tunnel %s[%X].\n", GNUNET_i2s (pi), tid);
3234   t = tunnel_get (pi, tid);
3235   if (NULL == t) /* might be a local tunnel */
3236   {
3237     uint32_t opt;
3238
3239     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating tunnel\n");
3240     t = tunnel_new (GNUNET_PEER_intern (pi), tid, NULL, 0);
3241     if (NULL == t)
3242     {
3243       GNUNET_break (0);
3244       return GNUNET_OK;
3245     }
3246     t->port = ntohl (msg->port);
3247     opt = ntohl (msg->opt);
3248     if (0 != (opt & GNUNET_MESH_OPTION_NOBUFFER))
3249     {
3250       t->nobuffer = GNUNET_YES;
3251       t->queue_max = 1;
3252     }
3253     if (0 != (opt & GNUNET_MESH_OPTION_RELIABLE))
3254     {
3255       t->reliable = GNUNET_YES;
3256     }
3257     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  nobuffer:%d\n", t->nobuffer);
3258
3259     tunnel_reset_timeout (t);
3260   }
3261   t->state = MESH_TUNNEL_WAITING;
3262   dest_peer_info =
3263       GNUNET_CONTAINER_multihashmap_get (peers, &pi[size - 1].hashPubKey);
3264   if (NULL == dest_peer_info)
3265   {
3266     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3267                 "  Creating PeerInfo for destination.\n");
3268     dest_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
3269     dest_peer_info->id = GNUNET_PEER_intern (&pi[size - 1]);
3270     GNUNET_CONTAINER_multihashmap_put (peers, &pi[size - 1].hashPubKey,
3271                                        dest_peer_info,
3272                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
3273   }
3274   orig_peer_info = GNUNET_CONTAINER_multihashmap_get (peers, &pi->hashPubKey);
3275   if (NULL == orig_peer_info)
3276   {
3277     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3278                 "  Creating PeerInfo for origin.\n");
3279     orig_peer_info = GNUNET_malloc (sizeof (struct MeshPeerInfo));
3280     orig_peer_info->id = GNUNET_PEER_intern (pi);
3281     GNUNET_CONTAINER_multihashmap_put (peers, &pi->hashPubKey, orig_peer_info,
3282                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
3283   }
3284   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
3285   path = path_new (size);
3286   own_pos = 0;
3287   for (i = 0; i < size; i++)
3288   {
3289     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ... adding %s\n",
3290                 GNUNET_i2s (&pi[i]));
3291     path->peers[i] = GNUNET_PEER_intern (&pi[i]);
3292     if (path->peers[i] == myid)
3293       own_pos = i;
3294   }
3295   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
3296   if (own_pos == 0 && path->peers[own_pos] != myid)
3297   {
3298     /* create path: self not found in path through self */
3299     GNUNET_break_op (0);
3300     path_destroy (path);
3301     tunnel_destroy (t);
3302     return GNUNET_OK;
3303   }
3304   path_add_to_peers (path, GNUNET_NO);
3305   tunnel_use_path (t, path);
3306
3307   peer_info_add_tunnel (dest_peer_info, t);
3308
3309   if (own_pos == size - 1)
3310   {
3311     struct MeshClient *c;
3312     struct GNUNET_HashCode hc;
3313
3314     /* Find target client */
3315     GMC_hash32 (t->port, &hc);
3316     c = GNUNET_CONTAINER_multihashmap_get (ports, &hc);
3317     if (NULL == c)
3318     {
3319       /* TODO send reject */
3320       return GNUNET_OK;
3321     }
3322
3323     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
3324     peer_info_add_path_to_origin (orig_peer_info, path, GNUNET_YES);
3325
3326     /* Assign local tid */
3327     while (NULL != tunnel_get_incoming (next_local_tid))
3328       next_local_tid = (next_local_tid + 1) | GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
3329     t->local_tid_dest = next_local_tid++;
3330     next_local_tid = next_local_tid | GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
3331
3332     if (GNUNET_YES == t->reliable)
3333       t->sent_messages_bck =
3334            GNUNET_CONTAINER_multihashmap32_create (t->queue_max);
3335
3336     tunnel_add_client (t, c);
3337     send_client_tunnel_create (t);
3338     send_path_ack (t);
3339   }
3340   else
3341   {
3342     struct MeshPeerPath *path2;
3343
3344     t->next_hop = path->peers[own_pos + 1];
3345     GNUNET_PEER_change_rc(t->next_hop, 1);
3346
3347     /* It's for somebody else! Retransmit. */
3348     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
3349     path2 = path_duplicate (path);
3350     peer_info_add_path (dest_peer_info, path2, GNUNET_NO);
3351     peer_info_add_path_to_origin (orig_peer_info, path, GNUNET_NO);
3352     send_create_path (t);
3353   }
3354   return GNUNET_OK;
3355 }
3356
3357
3358
3359 /**
3360  * Core handler for path ACKs
3361  *
3362  * @param cls closure
3363  * @param message message
3364  * @param peer peer identity this notification is about
3365  *
3366  * @return GNUNET_OK to keep the connection open,
3367  *         GNUNET_SYSERR to close it (signal serious error)
3368  */
3369 static int
3370 handle_mesh_path_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
3371                       const struct GNUNET_MessageHeader *message)
3372 {
3373   struct GNUNET_MESH_PathACK *msg;
3374   struct MeshPeerInfo *peer_info;
3375   struct MeshPeerPath *p;
3376   struct MeshTunnel *t;
3377
3378   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received a path ACK msg [%s]\n",
3379               GNUNET_i2s (&my_full_id));
3380   msg = (struct GNUNET_MESH_PathACK *) message;
3381   t = tunnel_get (&msg->oid, ntohl(msg->tid));
3382   if (NULL == t)
3383   {
3384     /* TODO notify that we don't know the tunnel */
3385     GNUNET_STATISTICS_update (stats, "# control on unknown tunnel", 1, GNUNET_NO);
3386     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  don't know the tunnel %s [%X]!\n",
3387                 GNUNET_i2s (&msg->oid), ntohl(msg->tid));
3388     return GNUNET_OK;
3389   }
3390   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on tunnel %s [%X]\n",
3391               GNUNET_i2s (&msg->oid), ntohl(msg->tid));
3392
3393   peer_info = peer_get (&msg->peer_id);
3394   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by peer %s\n",
3395               GNUNET_i2s (&msg->peer_id));
3396   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n",
3397               GNUNET_i2s (peer));
3398
3399   /* Add path to peers? */
3400   p = t->path;
3401   if (NULL != p)
3402   {
3403     path_add_to_peers (p, GNUNET_YES);
3404   }
3405   else
3406   {
3407     GNUNET_break (0);
3408   }
3409   t->state = MESH_TUNNEL_READY;
3410   t->next_fc.last_ack_recv = (NULL == t->client) ? ntohl (msg->ack) : 0;
3411   t->prev_fc.last_ack_sent = ntohl (msg->ack);
3412
3413   /* Message for us? */
3414   if (0 == memcmp (&msg->oid, &my_full_id, sizeof (struct GNUNET_PeerIdentity)))
3415   {
3416     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
3417     if (NULL == t->owner)
3418     {
3419       GNUNET_break_op (0);
3420       return GNUNET_OK;
3421     }
3422     if (NULL != peer_info->dhtget)
3423     {
3424       GNUNET_DHT_get_stop (peer_info->dhtget);
3425       peer_info->dhtget = NULL;
3426     }
3427     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_PATH_ACK);
3428     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_PATH_ACK);
3429     return GNUNET_OK;
3430   }
3431
3432   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3433               "  not for us, retransmitting...\n");
3434   peer_info = peer_get (&msg->oid);
3435   send_prebuilt_message (message, t->prev_hop, t);
3436   return GNUNET_OK;
3437 }
3438
3439
3440 /**
3441  * Core handler for notifications of broken paths
3442  *
3443  * @param cls closure
3444  * @param message message
3445  * @param peer peer identity this notification is about
3446  *
3447  * @return GNUNET_OK to keep the connection open,
3448  *         GNUNET_SYSERR to close it (signal serious error)
3449  */
3450 static int
3451 handle_mesh_path_broken (void *cls, const struct GNUNET_PeerIdentity *peer,
3452                          const struct GNUNET_MessageHeader *message)
3453 {
3454   struct GNUNET_MESH_PathBroken *msg;
3455   struct MeshTunnel *t;
3456
3457   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3458               "Received a PATH BROKEN msg from %s\n", GNUNET_i2s (peer));
3459   msg = (struct GNUNET_MESH_PathBroken *) message;
3460   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
3461               GNUNET_i2s (&msg->peer1));
3462   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
3463               GNUNET_i2s (&msg->peer2));
3464   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3465   if (NULL == t)
3466   {
3467     GNUNET_break_op (0);
3468     return GNUNET_OK;
3469   }
3470   tunnel_notify_connection_broken (t, GNUNET_PEER_search (&msg->peer1),
3471                                    GNUNET_PEER_search (&msg->peer2));
3472   return GNUNET_OK;
3473
3474 }
3475
3476
3477 /**
3478  * Core handler for tunnel destruction
3479  *
3480  * @param cls closure
3481  * @param message message
3482  * @param peer peer identity this notification is about
3483  *
3484  * @return GNUNET_OK to keep the connection open,
3485  *         GNUNET_SYSERR to close it (signal serious error)
3486  */
3487 static int
3488 handle_mesh_tunnel_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
3489                             const struct GNUNET_MessageHeader *message)
3490 {
3491   struct GNUNET_MESH_TunnelDestroy *msg;
3492   struct MeshTunnel *t;
3493
3494   msg = (struct GNUNET_MESH_TunnelDestroy *) message;
3495   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3496               "Got a TUNNEL DESTROY packet from %s\n",
3497               GNUNET_i2s (peer));
3498   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3499               "  for tunnel %s [%u]\n",
3500               GNUNET_i2s (&msg->oid), ntohl (msg->tid));
3501   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3502   if (NULL == t)
3503   {
3504     /* Probably already got the message from another path,
3505      * destroyed the tunnel and retransmitted to children.
3506      * Safe to ignore.
3507      */
3508     GNUNET_STATISTICS_update (stats, "# control on unknown tunnel",
3509                               1, GNUNET_NO);
3510     return GNUNET_OK;
3511   }
3512   if (t->local_tid_dest >= GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
3513   {
3514     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "INCOMING TUNNEL %X %X\n",
3515                 t->local_tid, t->local_tid_dest);
3516   }
3517   if (GNUNET_PEER_search (peer) == t->prev_hop)
3518   {
3519     // TODO check owner's signature
3520     // TODO add owner's signatue to tunnel for retransmission
3521     peer_cancel_queues (t->prev_hop, t);
3522     GNUNET_PEER_change_rc (t->prev_hop, -1);
3523     t->prev_hop = 0;
3524   }
3525   else if (GNUNET_PEER_search (peer) == t->next_hop)
3526   {
3527     // TODO check dest's signature
3528     // TODO add dest's signatue to tunnel for retransmission
3529     peer_cancel_queues (t->next_hop, t);
3530     GNUNET_PEER_change_rc (t->next_hop, -1);
3531     t->next_hop = 0;
3532   }
3533   else
3534   {
3535     GNUNET_break_op (0);
3536     // TODO check both owner AND destination's signature to see which matches
3537     // TODO restransmit in appropriate direction
3538     return GNUNET_OK;
3539   }
3540   tunnel_destroy_empty (t);
3541
3542   // TODO: add timeout to destroy the tunnel anyway
3543   return GNUNET_OK;
3544 }
3545
3546
3547 /**
3548  * Core handler for mesh network traffic going from the origin to a peer
3549  *
3550  * @param cls closure
3551  * @param peer peer identity this notification is about
3552  * @param message message
3553  * @return GNUNET_OK to keep the connection open,
3554  *         GNUNET_SYSERR to close it (signal serious error)
3555  */
3556 static int
3557 handle_mesh_unicast (void *cls, const struct GNUNET_PeerIdentity *peer,
3558                           const struct GNUNET_MessageHeader *message)
3559 {
3560   struct GNUNET_MESH_Data *msg;
3561   struct MeshTunnel *t;
3562   uint32_t pid;
3563   uint32_t ttl;
3564   size_t size;
3565
3566   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a unicast packet from %s\n",
3567               GNUNET_i2s (peer));
3568   /* Check size */
3569   size = ntohs (message->size);
3570   if (size <
3571       sizeof (struct GNUNET_MESH_Data) +
3572       sizeof (struct GNUNET_MessageHeader))
3573   {
3574     GNUNET_break (0);
3575     return GNUNET_OK;
3576   }
3577   msg = (struct GNUNET_MESH_Data *) message;
3578   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " of type %s\n",
3579               GNUNET_MESH_DEBUG_M2S (ntohs (msg[1].header.type)));
3580   /* Check tunnel */
3581   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3582   if (NULL == t)
3583   {
3584     /* TODO notify back: we don't know this tunnel */
3585     GNUNET_STATISTICS_update (stats, "# data on unknown tunnel", 1, GNUNET_NO);
3586     GNUNET_break_op (0);
3587     return GNUNET_OK;
3588   }
3589   pid = ntohl (msg->pid);
3590   if (t->prev_fc.last_pid_recv == pid)
3591   {
3592     GNUNET_STATISTICS_update (stats, "# duplicate PID drops", 1, GNUNET_NO);
3593     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3594                 " Already seen pid %u, DROPPING!\n", pid);
3595     return GNUNET_OK;
3596   }
3597   else
3598   {
3599     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3600                 " pid %u not seen yet, forwarding\n", pid);
3601   }
3602
3603   if (GMC_is_pid_bigger (pid, t->prev_fc.last_ack_sent))
3604   {
3605     GNUNET_STATISTICS_update (stats, "# unsolicited unicast", 1, GNUNET_NO);
3606     GNUNET_break_op (0);
3607     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3608                 "Received PID %u, ACK %u\n",
3609                 pid, t->prev_fc.last_ack_sent);
3610     tunnel_send_fwd_ack(t, GNUNET_MESSAGE_TYPE_MESH_POLL);
3611     return GNUNET_OK;
3612   }
3613   t->prev_fc.last_pid_recv = pid;
3614
3615   tunnel_reset_timeout (t);
3616   if (t->dest == myid)
3617   {
3618     /* TODO signature verification */
3619     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3620                 "  it's for us! sending to clients...\n");
3621     GNUNET_STATISTICS_update (stats, "# unicast received", 1, GNUNET_NO);
3622     tunnel_send_client_ucast (t, msg);
3623     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_UNICAST);
3624     return GNUNET_OK;
3625   }
3626   if (0 == t->next_hop)
3627   {
3628     GNUNET_break (0);
3629     return GNUNET_OK;
3630   }
3631   ttl = ntohl (msg->ttl);
3632   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
3633   if (ttl == 0)
3634   {
3635     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
3636     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
3637     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_ACK);
3638     return GNUNET_OK;
3639   }
3640   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3641               "  not for us, retransmitting...\n");
3642
3643   send_prebuilt_message (message, t->next_hop, t);
3644   GNUNET_STATISTICS_update (stats, "# unicast forwarded", 1, GNUNET_NO);
3645   return GNUNET_OK;
3646 }
3647
3648
3649 /**
3650  * Core handler for mesh network traffic toward the owner of a tunnel
3651  *
3652  * @param cls closure
3653  * @param message message
3654  * @param peer peer identity this notification is about
3655  *
3656  * @return GNUNET_OK to keep the connection open,
3657  *         GNUNET_SYSERR to close it (signal serious error)
3658  */
3659 static int
3660 handle_mesh_to_orig (void *cls, const struct GNUNET_PeerIdentity *peer,
3661                           const struct GNUNET_MessageHeader *message)
3662 {
3663   struct GNUNET_MESH_Data *msg;
3664   struct MeshTunnel *t;
3665   size_t size;
3666   uint32_t pid;
3667   uint32_t ttl;
3668
3669   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a ToOrigin packet from %s\n",
3670               GNUNET_i2s (peer));
3671   size = ntohs (message->size);
3672   if (size < sizeof (struct GNUNET_MESH_Data) +     /* Payload must be */
3673       sizeof (struct GNUNET_MessageHeader))     /* at least a header */
3674   {
3675     GNUNET_break_op (0);
3676     return GNUNET_OK;
3677   }
3678   msg = (struct GNUNET_MESH_Data *) message;
3679   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " of type %s\n",
3680               GNUNET_MESH_DEBUG_M2S (ntohs (msg[1].header.type)));
3681   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3682   pid = ntohl (msg->pid);
3683
3684   if (NULL == t)
3685   {
3686     /* TODO notify that we dont know this tunnel (whom)? */
3687     GNUNET_STATISTICS_update (stats, "# data on unknown tunnel", 1, GNUNET_NO);
3688     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3689                 "Received to_origin with PID %u on unknown tunnel %s [%u]\n",
3690                 pid, GNUNET_i2s (&msg->oid), ntohl (msg->tid));
3691     return GNUNET_OK;
3692   }
3693
3694   if (t->next_fc.last_pid_recv == pid)
3695   {
3696     /* already seen this packet, drop */
3697     GNUNET_STATISTICS_update (stats, "# duplicate PID drops BCK", 1, GNUNET_NO);
3698     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3699                 " Already seen pid %u, DROPPING!\n", pid);
3700     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_ACK);
3701     return GNUNET_OK;
3702   }
3703
3704   if (GMC_is_pid_bigger (pid, t->next_fc.last_ack_sent))
3705   {
3706     GNUNET_STATISTICS_update (stats, "# unsolicited to_orig", 1, GNUNET_NO);
3707     GNUNET_break_op (0);
3708     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3709                 "Received PID %u, ACK %u\n",
3710                 pid, t->next_fc.last_ack_sent);
3711     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_POLL);
3712     return GNUNET_OK;
3713   }
3714   t->next_fc.last_pid_recv = pid;
3715   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3716               " pid %u not seen yet, forwarding\n", pid);
3717
3718   if (myid == t->id.oid)
3719   {
3720     /* TODO signature verification */
3721     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3722                 "  it's for us! sending to clients...\n");
3723     GNUNET_STATISTICS_update (stats, "# to origin received", 1, GNUNET_NO);
3724     tunnel_send_client_to_orig (t, msg);
3725     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN);
3726     return GNUNET_OK;
3727   }
3728   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3729               "  not for us, retransmitting...\n");
3730
3731   if (0 == t->prev_hop) /* No owner AND no prev hop */
3732   {
3733     if (GNUNET_YES == t->destroy)
3734     {
3735       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3736                   "to orig received on a dying tunnel %s [%X]\n",
3737                   GNUNET_i2s (&msg->oid), ntohl(msg->tid));
3738       return GNUNET_OK;
3739     }
3740     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
3741                 "unknown to origin at %s\n",
3742                 GNUNET_i2s (&my_full_id));
3743     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
3744                 "from peer %s\n",
3745                 GNUNET_i2s (peer));
3746     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
3747                 "on tunnel %s [%X]\n",
3748                 GNUNET_i2s (&msg->oid), ntohl(msg->tid));
3749     return GNUNET_OK;
3750   }
3751   ttl = ntohl (msg->ttl);
3752   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
3753   if (ttl == 0)
3754   {
3755     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
3756     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
3757     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_ACK);
3758     return GNUNET_OK;
3759   }
3760   send_prebuilt_message (message, t->prev_hop, t);
3761   GNUNET_STATISTICS_update (stats, "# to origin forwarded", 1, GNUNET_NO);
3762
3763   return GNUNET_OK;
3764 }
3765
3766
3767 /**
3768  * Core handler for mesh network traffic end-to-end ACKs.
3769  *
3770  * @param cls Closure.
3771  * @param message Message.
3772  * @param peer Peer identity this notification is about.
3773  *
3774  * @return GNUNET_OK to keep the connection open,
3775  *         GNUNET_SYSERR to close it (signal serious error)
3776  */
3777 static int
3778 handle_mesh_data_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
3779                       const struct GNUNET_MessageHeader *message)
3780 {
3781   struct GNUNET_MESH_DataACK *msg;
3782   struct GNUNET_CONTAINER_MultiHashMap32 *hm;
3783   struct MeshSentMessage *copy;
3784   struct MeshTunnel *t;
3785   GNUNET_PEER_Id id;
3786   uint32_t ack;
3787
3788   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a DATA ACK message from %s!\n",
3789               GNUNET_i2s (peer));
3790   msg = (struct GNUNET_MESH_DataACK *) message;
3791
3792   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3793   if (NULL == t)
3794   {
3795     /* TODO notify that we dont know this tunnel (whom)? */
3796     GNUNET_STATISTICS_update (stats, "# ack on unknown tunnel", 1, GNUNET_NO);
3797     return GNUNET_OK;
3798   }
3799   ack = ntohl (msg->pid);
3800   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u\n", ack);
3801
3802   /* Is this a forward or backward ACK? */
3803   id = GNUNET_PEER_search (peer);
3804   if (t->next_hop == id)
3805   {
3806     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
3807     if (NULL == t->owner)
3808     {
3809       send_prebuilt_message (message, t->prev_hop, t);
3810       return GNUNET_OK;
3811     }
3812     hm = t->sent_messages_fwd;
3813     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_DATA_ACK);
3814   }
3815   else if (t->prev_hop == id)
3816   {
3817     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
3818     if (NULL == t->client)
3819     {
3820       send_prebuilt_message (message, t->next_hop, t);
3821       return GNUNET_OK;
3822     }
3823     hm = t->sent_messages_bck;
3824     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_DATA_ACK);
3825   }
3826   else
3827     GNUNET_break_op (0);
3828
3829   copy = GNUNET_CONTAINER_multihashmap32_get (hm, ack);
3830   if (NULL == copy)
3831   {
3832     GNUNET_break (0); // FIXME needed?
3833     return GNUNET_OK;
3834   }
3835   GNUNET_break (GNUNET_YES ==
3836                 GNUNET_CONTAINER_multihashmap32_remove (hm, ack, copy));
3837   if (GNUNET_SCHEDULER_NO_TASK != copy->retry_task)
3838   {
3839     GNUNET_SCHEDULER_cancel (copy->retry_task);
3840   }
3841   else
3842     GNUNET_break (0);
3843   GNUNET_free (copy);
3844   return GNUNET_OK;
3845 }
3846
3847 /**
3848  * Core handler for mesh network traffic point-to-point acks.
3849  *
3850  * @param cls closure
3851  * @param message message
3852  * @param peer peer identity this notification is about
3853  *
3854  * @return GNUNET_OK to keep the connection open,
3855  *         GNUNET_SYSERR to close it (signal serious error)
3856  */
3857 static int
3858 handle_mesh_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
3859                  const struct GNUNET_MessageHeader *message)
3860 {
3861   struct GNUNET_MESH_ACK *msg;
3862   struct MeshTunnel *t;
3863   GNUNET_PEER_Id id;
3864   uint32_t ack;
3865
3866   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
3867               GNUNET_i2s (peer));
3868   msg = (struct GNUNET_MESH_ACK *) message;
3869
3870   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3871
3872   if (NULL == t)
3873   {
3874     /* TODO notify that we dont know this tunnel (whom)? */
3875     GNUNET_STATISTICS_update (stats, "# ack on unknown tunnel", 1, GNUNET_NO);
3876     return GNUNET_OK;
3877   }
3878   ack = ntohl (msg->pid);
3879   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u\n", ack);
3880
3881   /* Is this a forward or backward ACK? */
3882   id = GNUNET_PEER_search (peer);
3883   if (t->next_hop == id)
3884   {
3885     debug_fwd_ack++;
3886     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
3887     if (GNUNET_SCHEDULER_NO_TASK != t->next_fc.poll_task &&
3888         GMC_is_pid_bigger (ack, t->next_fc.last_ack_recv))
3889     {
3890       GNUNET_SCHEDULER_cancel (t->next_fc.poll_task);
3891       t->next_fc.poll_task = GNUNET_SCHEDULER_NO_TASK;
3892       t->next_fc.poll_time = GNUNET_TIME_UNIT_SECONDS;
3893     }
3894     t->next_fc.last_ack_recv = ack;
3895     peer_unlock_queue (t->next_hop);
3896     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_ACK);
3897   }
3898   else if (t->prev_hop == id)
3899   {
3900     debug_bck_ack++;
3901     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
3902     if (GNUNET_SCHEDULER_NO_TASK != t->prev_fc.poll_task &&
3903         GMC_is_pid_bigger (ack, t->prev_fc.last_ack_recv))
3904     {
3905       GNUNET_SCHEDULER_cancel (t->prev_fc.poll_task);
3906       t->prev_fc.poll_task = GNUNET_SCHEDULER_NO_TASK;
3907       t->prev_fc.poll_time = GNUNET_TIME_UNIT_SECONDS;
3908     }
3909     t->prev_fc.last_ack_recv = ack;
3910     peer_unlock_queue (t->prev_hop);
3911     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_ACK);
3912   }
3913   else
3914     GNUNET_break_op (0);
3915   return GNUNET_OK;
3916 }
3917
3918
3919 /**
3920  * Core handler for mesh network traffic point-to-point ack polls.
3921  *
3922  * @param cls closure
3923  * @param message message
3924  * @param peer peer identity this notification is about
3925  *
3926  * @return GNUNET_OK to keep the connection open,
3927  *         GNUNET_SYSERR to close it (signal serious error)
3928  */
3929 static int
3930 handle_mesh_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
3931                   const struct GNUNET_MessageHeader *message)
3932 {
3933   struct GNUNET_MESH_Poll *msg;
3934   struct MeshTunnel *t;
3935   GNUNET_PEER_Id id;
3936
3937   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got an POLL packet from %s!\n",
3938               GNUNET_i2s (peer));
3939
3940   msg = (struct GNUNET_MESH_Poll *) message;
3941
3942   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3943
3944   if (NULL == t)
3945   {
3946     /* TODO notify that we dont know this tunnel (whom)? */
3947     GNUNET_STATISTICS_update (stats, "# poll on unknown tunnel", 1, GNUNET_NO);
3948     GNUNET_break_op (0);
3949     return GNUNET_OK;
3950   }
3951
3952   /* Is this a forward or backward ACK? */
3953   id = GNUNET_PEER_search(peer);
3954   if (t->next_hop == id)
3955   {
3956     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  from FWD\n");
3957     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_POLL);
3958   }
3959   else if (t->prev_hop == id)
3960   {
3961     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  from BCK\n");
3962     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_POLL);
3963   }
3964   else
3965     GNUNET_break (0);
3966
3967   return GNUNET_OK;
3968 }
3969
3970
3971 /**
3972  * Core handler for mesh keepalives.
3973  *
3974  * @param cls closure
3975  * @param message message
3976  * @param peer peer identity this notification is about
3977  * @return GNUNET_OK to keep the connection open,
3978  *         GNUNET_SYSERR to close it (signal serious error)
3979  *
3980  * TODO: Check who we got this from, to validate route.
3981  */
3982 static int
3983 handle_mesh_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
3984                        const struct GNUNET_MessageHeader *message)
3985 {
3986   struct GNUNET_MESH_TunnelKeepAlive *msg;
3987   struct MeshTunnel *t;
3988
3989   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
3990               GNUNET_i2s (peer));
3991
3992   msg = (struct GNUNET_MESH_TunnelKeepAlive *) message;
3993   t = tunnel_get (&msg->oid, ntohl (msg->tid));
3994
3995   if (NULL == t)
3996   {
3997     /* TODO notify that we dont know that tunnel */
3998     GNUNET_STATISTICS_update (stats, "# keepalive on unknown tunnel", 1,
3999                               GNUNET_NO);
4000     return GNUNET_OK;
4001   }
4002
4003   tunnel_reset_timeout (t);
4004
4005   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
4006   send_prebuilt_message (message, t->next_hop, t);
4007   return GNUNET_OK;
4008   }
4009
4010
4011
4012 /**
4013  * Functions to handle messages from core
4014  */
4015 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
4016   {&handle_mesh_path_create, GNUNET_MESSAGE_TYPE_MESH_PATH_CREATE, 0},
4017   {&handle_mesh_path_broken, GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN,
4018    sizeof (struct GNUNET_MESH_PathBroken)},
4019   {&handle_mesh_tunnel_destroy, GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY,
4020    sizeof (struct GNUNET_MESH_TunnelDestroy)},
4021   {&handle_mesh_unicast, GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
4022   {&handle_mesh_to_orig, GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
4023   {&handle_mesh_data_ack, GNUNET_MESSAGE_TYPE_MESH_DATA_ACK,
4024     sizeof (struct GNUNET_MESH_DataACK)},
4025   {&handle_mesh_keepalive, GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE,
4026     sizeof (struct GNUNET_MESH_TunnelKeepAlive)},
4027   {&handle_mesh_ack, GNUNET_MESSAGE_TYPE_MESH_ACK,
4028     sizeof (struct GNUNET_MESH_ACK)},
4029   {&handle_mesh_poll, GNUNET_MESSAGE_TYPE_MESH_POLL,
4030     sizeof (struct GNUNET_MESH_Poll)},
4031   {&handle_mesh_path_ack, GNUNET_MESSAGE_TYPE_MESH_PATH_ACK,
4032    sizeof (struct GNUNET_MESH_PathACK)},
4033   {NULL, 0, 0}
4034 };
4035
4036
4037
4038 /******************************************************************************/
4039 /****************       MESH LOCAL HANDLER HELPERS      ***********************/
4040 /******************************************************************************/
4041
4042
4043 #if LATER
4044 /**
4045  * notify_client_connection_failure: notify a client that the connection to the
4046  * requested remote peer is not possible (for instance, no route found)
4047  * Function called when the socket is ready to queue more data. "buf" will be
4048  * NULL and "size" zero if the socket was closed for writing in the meantime.
4049  *
4050  * @param cls closure
4051  * @param size number of bytes available in buf
4052  * @param buf where the callee should write the message
4053  * @return number of bytes written to buf
4054  */
4055 static size_t
4056 notify_client_connection_failure (void *cls, size_t size, void *buf)
4057 {
4058   int size_needed;
4059   struct MeshPeerInfo *peer_info;
4060   struct GNUNET_MESH_PeerControl *msg;
4061   struct GNUNET_PeerIdentity id;
4062
4063   if (0 == size && NULL == buf)
4064   {
4065     // TODO retry? cancel?
4066     return 0;
4067   }
4068
4069   size_needed = sizeof (struct GNUNET_MESH_PeerControl);
4070   peer_info = (struct MeshPeerInfo *) cls;
4071   msg = (struct GNUNET_MESH_PeerControl *) buf;
4072   msg->header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
4073   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_PEER_DISCONNECTED);
4074 //     msg->tunnel_id = htonl(peer_info->t->tid);
4075   GNUNET_PEER_resolve (peer_info->id, &id);
4076   memcpy (&msg->peer, &id, sizeof (struct GNUNET_PeerIdentity));
4077
4078   return size_needed;
4079 }
4080 #endif
4081
4082
4083 /**
4084  * Send keepalive packets for a tunnel.
4085  *
4086  * @param cls Closure (tunnel for which to send the keepalive).
4087  * @param tc Notification context.
4088  * 
4089  * FIXME: add a refresh reset in case of normal unicast traffic is doing the job
4090  */
4091 static void
4092 path_refresh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4093 {
4094   struct MeshTunnel *t = cls;
4095   struct GNUNET_MESH_TunnelKeepAlive *msg;
4096   size_t size = sizeof (struct GNUNET_MESH_TunnelKeepAlive);
4097   char cbuf[size];
4098
4099   t->maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4100   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) ||
4101       NULL == t->owner || 0 == t->local_tid)
4102   {
4103     return;
4104   }
4105
4106   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4107               "sending keepalive for tunnel %d\n", t->id.tid);
4108
4109   msg = (struct GNUNET_MESH_TunnelKeepAlive *) cbuf;
4110   msg->header.size = htons (size);
4111   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_KEEPALIVE);
4112   msg->oid = my_full_id;
4113   msg->tid = htonl (t->id.tid);
4114   send_prebuilt_message (&msg->header, t->next_hop, t);
4115
4116   t->maintenance_task =
4117       GNUNET_SCHEDULER_add_delayed (refresh_path_time, &path_refresh, t);
4118 }
4119
4120
4121 /**
4122  * Function to process paths received for a new peer addition. The recorded
4123  * paths form the initial tunnel, which can be optimized later.
4124  * Called on each result obtained for the DHT search.
4125  *
4126  * @param cls closure
4127  * @param exp when will this value expire
4128  * @param key key of the result
4129  * @param get_path path of the get request
4130  * @param get_path_length lenght of get_path
4131  * @param put_path path of the put request
4132  * @param put_path_length length of the put_path
4133  * @param type type of the result
4134  * @param size number of bytes in data
4135  * @param data pointer to the result data
4136  */
4137 static void
4138 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
4139                     const struct GNUNET_HashCode * key,
4140                     const struct GNUNET_PeerIdentity *get_path,
4141                     unsigned int get_path_length,
4142                     const struct GNUNET_PeerIdentity *put_path,
4143                     unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
4144                     size_t size, const void *data)
4145 {
4146   struct MeshPeerInfo *peer = cls;
4147   struct MeshPeerPath *p;
4148   struct GNUNET_PeerIdentity pi;
4149   int i;
4150
4151   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got results from DHT!\n");
4152   GNUNET_PEER_resolve (peer->id, &pi);
4153   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  for %s\n", GNUNET_i2s (&pi));
4154
4155   p = path_build_from_dht (get_path, get_path_length,
4156                            put_path, put_path_length);
4157   path_add_to_peers (p, GNUNET_NO);
4158   path_destroy (p);
4159   for (i = 0; i < peer->ntunnels; i++)
4160   {
4161     struct GNUNET_PeerIdentity id;
4162
4163     GNUNET_PEER_resolve (peer->tunnels[i]->id.oid, &id);
4164     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ... tunnel %s:%X (%X / %X)\n",
4165                 GNUNET_i2s (&id), peer->tunnels[i]->id.tid,
4166                 peer->tunnels[i]->local_tid, 
4167                 peer->tunnels[i]->local_tid_dest);
4168     if (peer->tunnels[i]->state == MESH_TUNNEL_SEARCHING)
4169     {
4170       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n");
4171       peer_connect (peer, peer->tunnels[i]);
4172     }
4173   }
4174
4175   return;
4176 }
4177
4178
4179 /******************************************************************************/
4180 /*********************       MESH LOCAL HANDLES      **************************/
4181 /******************************************************************************/
4182
4183
4184 /**
4185  * Handler for client disconnection
4186  *
4187  * @param cls closure
4188  * @param client identification of the client; NULL
4189  *        for the last call when the server is destroyed
4190  */
4191 static void
4192 handle_local_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
4193 {
4194   struct MeshClient *c;
4195   struct MeshClient *next;
4196
4197   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disconnected: %p\n", client);
4198   if (client == NULL)
4199   {
4200     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (SERVER DOWN)\n");
4201     return;
4202   }
4203
4204   c = clients_head;
4205   while (NULL != c)
4206   {
4207     if (c->handle != client)
4208     {
4209       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4210                   "   ... searching %p (%u)\n",
4211                   c->handle, c->id);
4212       c = c->next;
4213       continue;
4214     }
4215     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "matching client found (%u)\n",
4216                 c->id);
4217     GNUNET_SERVER_client_drop (c->handle);
4218     c->shutting_down = GNUNET_YES;
4219     GNUNET_CONTAINER_multihashmap_iterate (c->own_tunnels,
4220                                            &tunnel_destroy_iterator, c);
4221     GNUNET_CONTAINER_multihashmap_iterate (c->incoming_tunnels,
4222                                            &tunnel_destroy_iterator, c);
4223     GNUNET_CONTAINER_multihashmap_destroy (c->own_tunnels);
4224     GNUNET_CONTAINER_multihashmap_destroy (c->incoming_tunnels);
4225
4226     if (NULL != c->ports)
4227       GNUNET_CONTAINER_multihashmap_destroy (c->ports);
4228     next = c->next;
4229     GNUNET_CONTAINER_DLL_remove (clients_head, clients_tail, c);
4230     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  CLIENT FREE at %p\n", c);
4231     GNUNET_free (c);
4232     GNUNET_STATISTICS_update (stats, "# clients", -1, GNUNET_NO);
4233     c = next;
4234   }
4235   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "done!\n");
4236   return;
4237 }
4238
4239
4240 /**
4241  * Handler for new clients
4242  *
4243  * @param cls closure
4244  * @param client identification of the client
4245  * @param message the actual message, which includes messages the client wants
4246  */
4247 static void
4248 handle_local_new_client (void *cls, struct GNUNET_SERVER_Client *client,
4249                          const struct GNUNET_MessageHeader *message)
4250 {
4251   struct GNUNET_MESH_ClientConnect *cc_msg;
4252   struct MeshClient *c;
4253   unsigned int size;
4254   uint32_t *p;
4255   unsigned int i;
4256
4257   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client connected %p\n", client);
4258
4259   /* Check data sanity */
4260   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
4261   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
4262   if (0 != (size % sizeof (uint32_t)))
4263   {
4264     GNUNET_break (0);
4265     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4266     return;
4267   }
4268   size /= sizeof (uint32_t);
4269
4270   /* Create new client structure */
4271   c = GNUNET_malloc (sizeof (struct MeshClient));
4272   c->id = next_client_id++; /* overflow not important: just for debug */
4273   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client id %u\n", c->id);
4274   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client has %u ports\n", size);
4275   c->handle = client;
4276   GNUNET_SERVER_client_keep (client);
4277   if (size > 0)
4278   {
4279     uint32_t u32;
4280     struct GNUNET_HashCode hc;
4281
4282     p = (uint32_t *) &cc_msg[1];
4283     c->ports = GNUNET_CONTAINER_multihashmap_create (size, GNUNET_NO);
4284     for (i = 0; i < size; i++)
4285     {
4286       u32 = ntohl (p[i]);
4287       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    port: %u\n", u32);
4288       GMC_hash32 (u32, &hc);
4289
4290       /* store in client's hashmap */
4291       GNUNET_CONTAINER_multihashmap_put (c->ports, &hc, c,
4292                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
4293       /* store in global hashmap */
4294       /* FIXME only allow one client to have the port open,
4295        *       have a backup hashmap with waiting clients */
4296       GNUNET_CONTAINER_multihashmap_put (ports, &hc, c,
4297                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
4298     }
4299   }
4300
4301   GNUNET_CONTAINER_DLL_insert (clients_head, clients_tail, c);
4302   c->own_tunnels = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
4303   c->incoming_tunnels = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
4304   GNUNET_SERVER_notification_context_add (nc, client);
4305   GNUNET_STATISTICS_update (stats, "# clients", 1, GNUNET_NO);
4306
4307   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4308   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client processed\n");
4309 }
4310
4311
4312 /**
4313  * Handler for requests of new tunnels
4314  *
4315  * @param cls Closure.
4316  * @param client Identification of the client.
4317  * @param message The actual message.
4318  */
4319 static void
4320 handle_local_tunnel_create (void *cls, struct GNUNET_SERVER_Client *client,
4321                             const struct GNUNET_MessageHeader *message)
4322 {
4323   struct GNUNET_MESH_TunnelMessage *t_msg;
4324   struct MeshPeerInfo *peer_info;
4325   struct MeshTunnel *t;
4326   struct MeshClient *c;
4327   MESH_TunnelNumber tid;
4328
4329   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new tunnel requested\n");
4330
4331   /* Sanity check for client registration */
4332   if (NULL == (c = client_get (client)))
4333   {
4334     GNUNET_break (0);
4335     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4336     return;
4337   }
4338   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
4339
4340   /* Message size sanity check */
4341   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
4342   {
4343     GNUNET_break (0);
4344     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4345     return;
4346   }
4347
4348   t_msg = (struct GNUNET_MESH_TunnelMessage *) message;
4349   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  towards %s\n",
4350               GNUNET_i2s (&t_msg->peer));
4351   /* Sanity check for tunnel numbering */
4352   tid = ntohl (t_msg->tunnel_id);
4353   if (0 == (tid & GNUNET_MESH_LOCAL_TUNNEL_ID_CLI))
4354   {
4355     GNUNET_break (0);
4356     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4357     return;
4358   }
4359   /* Sanity check for duplicate tunnel IDs */
4360   if (NULL != tunnel_get_by_local_id (c, tid))
4361   {
4362     GNUNET_break (0);
4363     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4364     return;
4365   }
4366
4367   /* Create tunnel */
4368   while (NULL != tunnel_get_by_pi (myid, next_tid))
4369     next_tid = (next_tid + 1) & ~GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
4370   t = tunnel_new (myid, next_tid, c, tid);
4371   next_tid = (next_tid + 1) & ~GNUNET_MESH_LOCAL_TUNNEL_ID_CLI;
4372   if (NULL == t)
4373   {
4374     GNUNET_break (0);
4375     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4376     return;
4377   }
4378   t->port = ntohl (t_msg->port);
4379   tunnel_set_options (t, ntohl (t_msg->options));
4380   if (GNUNET_YES == t->reliable)
4381     t->sent_messages_fwd =
4382      GNUNET_CONTAINER_multihashmap32_create (t->queue_max);
4383   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CREATED TUNNEL %s[%x]:%u (%x)\n",
4384               GNUNET_i2s (&my_full_id), t->id.tid, t->port, t->local_tid);
4385
4386   peer_info = peer_get (&t_msg->peer);
4387   peer_info_add_tunnel (peer_info, t);
4388   peer_connect (peer_info, t);
4389   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4390   return;
4391 }
4392
4393
4394 /**
4395  * Handler for requests of deleting tunnels
4396  *
4397  * @param cls closure
4398  * @param client identification of the client
4399  * @param message the actual message
4400  */
4401 static void
4402 handle_local_tunnel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
4403                              const struct GNUNET_MessageHeader *message)
4404 {
4405   struct GNUNET_MESH_TunnelMessage *tunnel_msg;
4406   struct MeshClient *c;
4407   struct MeshTunnel *t;
4408   MESH_TunnelNumber tid;
4409
4410   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4411               "Got a DESTROY TUNNEL from client!\n");
4412
4413   /* Sanity check for client registration */
4414   if (NULL == (c = client_get (client)))
4415   {
4416     GNUNET_break (0);
4417     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4418     return;
4419   }
4420   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
4421
4422   /* Message sanity check */
4423   if (sizeof (struct GNUNET_MESH_TunnelMessage) != ntohs (message->size))
4424   {
4425     GNUNET_break (0);
4426     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4427     return;
4428   }
4429
4430   tunnel_msg = (struct GNUNET_MESH_TunnelMessage *) message;
4431
4432   /* Retrieve tunnel */
4433   tid = ntohl (tunnel_msg->tunnel_id);
4434   t = tunnel_get_by_local_id(c, tid);
4435   if (NULL == t)
4436   {
4437     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "  tunnel %X not found\n", tid);
4438     GNUNET_break (0);
4439     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4440     return;
4441   }
4442
4443   /* Cleanup after the tunnel */
4444   client_delete_tunnel (c, t);
4445   if (c == t->client)
4446   {
4447     t->client = NULL;
4448   }
4449   else if (c == t->owner)
4450   {
4451     peer_info_remove_tunnel (peer_get_short (t->dest), t);
4452     t->owner = NULL;
4453   }
4454
4455   /* The tunnel will be destroyed when the last message is transmitted. */
4456   tunnel_destroy_empty (t);
4457
4458   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4459   return;
4460 }
4461
4462
4463 /**
4464  * Handler for client traffic directed to one peer
4465  *
4466  * @param cls closure
4467  * @param client identification of the client
4468  * @param message the actual message
4469  */
4470 static void
4471 handle_local_unicast (void *cls, struct GNUNET_SERVER_Client *client,
4472                       const struct GNUNET_MessageHeader *message)
4473 {
4474   struct MeshClient *c;
4475   struct MeshTunnel *t;
4476   struct GNUNET_MESH_Data *data_msg;
4477   MESH_TunnelNumber tid;
4478   size_t size;
4479
4480   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4481               "Got a unicast request from a client!\n");
4482
4483   /* Sanity check for client registration */
4484   if (NULL == (c = client_get (client)))
4485   {
4486     GNUNET_break (0);
4487     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4488     return;
4489   }
4490   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
4491
4492   data_msg = (struct GNUNET_MESH_Data *) message;
4493
4494   /* Sanity check for message size */
4495   size = ntohs (message->size);
4496   if (sizeof (struct GNUNET_MESH_Data) +
4497       sizeof (struct GNUNET_MessageHeader) > size)
4498   {
4499     GNUNET_break (0);
4500     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4501     return;
4502   }
4503
4504   /* Tunnel exists? */
4505   tid = ntohl (data_msg->tid);
4506   t = tunnel_get_by_local_id (c, tid);
4507   if (NULL == t)
4508   {
4509     GNUNET_break (0);
4510     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4511     return;
4512   }
4513
4514   /*  Is it a local tunnel? Then, does client own the tunnel? */
4515   if (t->owner->handle != client)
4516   {
4517     GNUNET_break (0);
4518     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4519     return;
4520   }
4521
4522   /* PID should be as expected: client<->service communication */
4523   if (ntohl (data_msg->pid) != t->prev_fc.last_pid_recv + 1)
4524   {
4525     GNUNET_break (0);
4526     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4527               "Unicast PID, expected %u, got %u\n",
4528               t->prev_fc.last_pid_recv + 1, ntohl (data_msg->pid));
4529     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4530     return;
4531   }
4532
4533   /* Ok, everything is correct, send the message
4534    * (pretend we got it from a mesh peer)
4535    */
4536   {
4537     struct MeshSentMessage *copy;
4538     struct GNUNET_MESH_Data *payload;
4539
4540     copy = GNUNET_malloc (sizeof (struct MeshSentMessage) + size);
4541     copy->id = ntohl (data_msg->pid);
4542     copy->is_forward = GNUNET_YES;
4543     copy->retry_timer = GNUNET_TIME_UNIT_MINUTES;
4544     copy->retry_task = GNUNET_SCHEDULER_add_delayed (copy->retry_timer,
4545                                                      &tunnel_retransmit_message,
4546                                                      copy);
4547     if (GNUNET_YES == t->reliable &&
4548         GNUNET_OK !=
4549         GNUNET_CONTAINER_multihashmap32_put (t->sent_messages_fwd,
4550                                              copy->id,
4551                                              copy,
4552                                              GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
4553     {
4554       GNUNET_break (0);
4555       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4556       return;
4557     }
4558     payload = (struct GNUNET_MESH_Data *) &copy[1];
4559     memcpy (payload, data_msg, size);
4560     payload->oid = my_full_id;
4561     payload->tid = htonl (t->id.tid);
4562     payload->ttl = htonl (default_ttl);
4563     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4564                 "  calling generic handler...\n");
4565     handle_mesh_unicast (NULL, &my_full_id, &payload->header);
4566   }
4567   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "receive done OK\n");
4568   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4569
4570   return;
4571 }
4572
4573
4574 /**
4575  * Handler for client traffic directed to the origin
4576  *
4577  * @param cls closure
4578  * @param client identification of the client
4579  * @param message the actual message
4580  */
4581 static void
4582 handle_local_to_origin (void *cls, struct GNUNET_SERVER_Client *client,
4583                         const struct GNUNET_MessageHeader *message)
4584 {
4585   struct GNUNET_MESH_Data *data_msg;
4586   struct MeshFlowControl *fc;
4587   struct MeshClient *c;
4588   struct MeshTunnel *t;
4589   MESH_TunnelNumber tid;
4590   size_t size;
4591
4592   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4593               "Got a ToOrigin request from a client!\n");
4594   /* Sanity check for client registration */
4595   if (NULL == (c = client_get (client)))
4596   {
4597     GNUNET_break (0);
4598     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4599     return;
4600   }
4601   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
4602
4603   data_msg = (struct GNUNET_MESH_Data *) message;
4604
4605   /* Sanity check for message size */
4606   size = ntohs (message->size);
4607   if (sizeof (struct GNUNET_MESH_Data) +
4608       sizeof (struct GNUNET_MessageHeader) > size)
4609   {
4610     GNUNET_break (0);
4611     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4612     return;
4613   }
4614
4615   /* Tunnel exists? */
4616   tid = ntohl (data_msg->tid);
4617   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on tunnel %X\n", tid);
4618   if (tid < GNUNET_MESH_LOCAL_TUNNEL_ID_SERV)
4619   {
4620     GNUNET_break (0);
4621     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4622     return;
4623   }
4624   t = tunnel_get_by_local_id (c, tid);
4625   if (NULL == t)
4626   {
4627     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Tunnel %X unknown.\n", tid);
4628     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "  for client %u.\n", c->id);
4629     GNUNET_break (0);
4630     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4631     return;
4632   }
4633
4634   /*  It should be sent by someone who has this as incoming tunnel. */
4635   if (t->client != c)
4636   {
4637     GNUNET_break (0);
4638     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4639     return;
4640   }
4641
4642   /* PID should be as expected */
4643   fc = &t->next_fc;
4644   if (ntohl (data_msg->pid) != fc->last_pid_recv + 1)
4645   {
4646     GNUNET_break (0);
4647     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4648                 "To Origin PID, expected %u, got %u\n",
4649                 fc->last_pid_recv + 1,
4650                 ntohl (data_msg->pid));
4651     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4652     return;
4653   }
4654
4655   /* Ok, everything is correct, send the message
4656    * (pretend we got it from a mesh peer)
4657    */
4658   {
4659     struct MeshSentMessage *copy;
4660     struct GNUNET_MESH_Data *payload;
4661
4662     copy = GNUNET_malloc (sizeof (struct MeshSentMessage) + size);
4663     copy->id = ntohl (data_msg->pid);
4664     copy->is_forward = GNUNET_NO;
4665     copy->retry_timer = GNUNET_TIME_UNIT_MINUTES;
4666     copy->retry_task = GNUNET_SCHEDULER_add_delayed (copy->retry_timer,
4667                                                      &tunnel_retransmit_message,
4668                                                      copy);
4669     if (GNUNET_OK !=
4670         GNUNET_CONTAINER_multihashmap32_put (t->sent_messages_bck,
4671                                              copy->id,
4672                                              copy,
4673                                              GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
4674     {
4675       GNUNET_break (0);
4676       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4677       return;
4678     }
4679     payload = (struct GNUNET_MESH_Data *) &copy[1];
4680     memcpy (payload, data_msg, size);
4681     GNUNET_PEER_resolve (t->id.oid, &payload->oid);
4682     payload->tid = htonl (t->id.tid);
4683     payload->ttl = htonl (default_ttl);
4684
4685     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4686                 "  calling generic handler...\n");
4687     handle_mesh_to_orig (NULL, &my_full_id, &payload->header);
4688   }
4689   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4690
4691   return;
4692 }
4693
4694
4695 /**
4696  * Handler for client's ACKs for payload traffic.
4697  *
4698  * @param cls Closure (unused).
4699  * @param client Identification of the client.
4700  * @param message The actual message.
4701  */
4702 static void
4703 handle_local_ack (void *cls, struct GNUNET_SERVER_Client *client,
4704                   const struct GNUNET_MessageHeader *message)
4705 {
4706   struct GNUNET_MESH_LocalAck *msg;
4707   struct MeshTunnel *t;
4708   struct MeshClient *c;
4709   MESH_TunnelNumber tid;
4710   uint32_t ack;
4711
4712   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a local ACK\n");
4713   /* Sanity check for client registration */
4714   if (NULL == (c = client_get (client)))
4715   {
4716     GNUNET_break (0);
4717     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4718     return;
4719   }
4720   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
4721
4722   msg = (struct GNUNET_MESH_LocalAck *) message;
4723
4724   /* Tunnel exists? */
4725   tid = ntohl (msg->tunnel_id);
4726   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on tunnel %X\n", tid);
4727   t = tunnel_get_by_local_id (c, tid);
4728   if (NULL == t)
4729   {
4730     GNUNET_break (0);
4731     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Tunnel %X unknown.\n", tid);
4732     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "  for client %u.\n", c->id);
4733     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4734     return;
4735   }
4736
4737   ack = ntohl (msg->ack);
4738   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ack %u\n", ack);
4739
4740   /* Does client own tunnel? I.E: Is this an ACK for BCK traffic? */
4741   if (t->owner == c)
4742   {
4743     /* The client owns the tunnel, ACK is for data to_origin, send BCK ACK. */
4744     t->prev_fc.last_ack_recv = ack;
4745     tunnel_send_bck_ack (t, GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
4746   }
4747   else
4748   {
4749     /* The client doesn't own the tunnel, this ACK is for FWD traffic. */
4750     t->next_fc.last_ack_recv = ack;
4751     tunnel_send_fwd_ack (t, GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
4752   }
4753
4754   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4755
4756   return;
4757 }
4758
4759
4760
4761 /**
4762  * Iterator over all tunnels to send a monitoring client info about each tunnel.
4763  *
4764  * @param cls Closure (client handle).
4765  * @param key Key (hashed tunnel ID, unused).
4766  * @param value Tunnel info.
4767  *
4768  * @return GNUNET_YES, to keep iterating.
4769  */
4770 static int
4771 monitor_all_tunnels_iterator (void *cls,
4772                               const struct GNUNET_HashCode * key,
4773                               void *value)
4774 {
4775   struct GNUNET_SERVER_Client *client = cls;
4776   struct MeshTunnel *t = value;
4777   struct GNUNET_MESH_LocalMonitor *msg;
4778
4779   msg = GNUNET_malloc (sizeof(struct GNUNET_MESH_LocalMonitor));
4780   GNUNET_PEER_resolve(t->id.oid, &msg->owner);
4781   msg->tunnel_id = htonl (t->id.tid);
4782   msg->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
4783   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS);
4784   GNUNET_PEER_resolve (t->dest, &msg->destination);
4785
4786   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
4787               "*  sending info about tunnel %s [%u]\n",
4788               GNUNET_i2s (&msg->owner), t->id.tid);
4789
4790   GNUNET_SERVER_notification_context_unicast (nc, client,
4791                                               &msg->header, GNUNET_NO);
4792   return GNUNET_YES;
4793 }
4794
4795
4796 /**
4797  * Handler for client's MONITOR request.
4798  *
4799  * @param cls Closure (unused).
4800  * @param client Identification of the client.
4801  * @param message The actual message.
4802  */
4803 static void
4804 handle_local_get_tunnels (void *cls, struct GNUNET_SERVER_Client *client,
4805                           const struct GNUNET_MessageHeader *message)
4806 {
4807   struct MeshClient *c;
4808
4809   /* Sanity check for client registration */
4810   if (NULL == (c = client_get (client)))
4811   {
4812     GNUNET_break (0);
4813     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4814     return;
4815   }
4816
4817   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
4818               "Received get tunnels request from client %u\n",
4819               c->id);
4820   GNUNET_CONTAINER_multihashmap_iterate (tunnels,
4821                                          monitor_all_tunnels_iterator,
4822                                          client);
4823   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
4824               "Get tunnels request from client %u completed\n",
4825               c->id);
4826   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4827 }
4828
4829
4830 /**
4831  * Handler for client's MONITOR_TUNNEL request.
4832  *
4833  * @param cls Closure (unused).
4834  * @param client Identification of the client.
4835  * @param message The actual message.
4836  */
4837 static void
4838 handle_local_show_tunnel (void *cls, struct GNUNET_SERVER_Client *client,
4839                           const struct GNUNET_MessageHeader *message)
4840 {
4841   const struct GNUNET_MESH_LocalMonitor *msg;
4842   struct GNUNET_MESH_LocalMonitor *resp;
4843   struct MeshClient *c;
4844   struct MeshTunnel *t;
4845
4846   /* Sanity check for client registration */
4847   if (NULL == (c = client_get (client)))
4848   {
4849     GNUNET_break (0);
4850     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
4851     return;
4852   }
4853
4854   msg = (struct GNUNET_MESH_LocalMonitor *) message;
4855   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
4856               "Received tunnel info request from client %u for tunnel %s[%X]\n",
4857               c->id,
4858               &msg->owner,
4859               ntohl (msg->tunnel_id));
4860   t = tunnel_get (&msg->owner, ntohl (msg->tunnel_id));
4861   if (NULL == t)
4862   {
4863     /* We don't know the tunnel FIXME */
4864     struct GNUNET_MESH_LocalMonitor warn;
4865
4866     warn = *msg;
4867     GNUNET_SERVER_notification_context_unicast (nc, client,
4868                                                 &warn.header,
4869                                                 GNUNET_NO);
4870     GNUNET_SERVER_receive_done (client, GNUNET_OK);
4871     return;
4872   }
4873
4874   /* Initialize context */
4875   resp = GNUNET_malloc (sizeof (struct GNUNET_MESH_LocalMonitor));
4876   *resp = *msg;
4877   GNUNET_PEER_resolve (t->dest, &resp->destination);
4878   resp->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
4879   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
4880                                               &resp->header, GNUNET_NO);
4881   GNUNET_free (resp);
4882
4883   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
4884               "Monitor tunnel request from client %u completed\n",
4885               c->id);
4886   GNUNET_SERVER_receive_done (client, GNUNET_OK);
4887 }
4888
4889
4890 /**
4891  * Functions to handle messages from clients
4892  */
4893 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
4894   {&handle_local_new_client, NULL,
4895    GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
4896   {&handle_local_tunnel_create, NULL,
4897    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE,
4898    sizeof (struct GNUNET_MESH_TunnelMessage)},
4899   {&handle_local_tunnel_destroy, NULL,
4900    GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY,
4901    sizeof (struct GNUNET_MESH_TunnelMessage)},
4902   {&handle_local_unicast, NULL,
4903    GNUNET_MESSAGE_TYPE_MESH_UNICAST, 0},
4904   {&handle_local_to_origin, NULL,
4905    GNUNET_MESSAGE_TYPE_MESH_TO_ORIGIN, 0},
4906   {&handle_local_ack, NULL,
4907    GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK,
4908    sizeof (struct GNUNET_MESH_LocalAck)},
4909   {&handle_local_get_tunnels, NULL,
4910    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS,
4911    sizeof (struct GNUNET_MessageHeader)},
4912   {&handle_local_show_tunnel, NULL,
4913    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL,
4914      sizeof (struct GNUNET_MESH_LocalMonitor)},
4915   {NULL, NULL, 0, 0}
4916 };
4917
4918
4919 /**
4920  * Method called whenever a given peer connects.
4921  *
4922  * @param cls closure
4923  * @param peer peer identity this notification is about
4924  */
4925 static void
4926 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
4927 {
4928   struct MeshPeerInfo *peer_info;
4929   struct MeshPeerPath *path;
4930
4931   DEBUG_CONN ("Peer connected\n");
4932   DEBUG_CONN ("     %s\n", GNUNET_i2s (&my_full_id));
4933   peer_info = peer_get (peer);
4934   if (myid == peer_info->id)
4935   {
4936     DEBUG_CONN ("     (self)\n");
4937     path = path_new (1);
4938   }
4939   else
4940   {
4941     DEBUG_CONN ("     %s\n", GNUNET_i2s (peer));
4942     path = path_new (2);
4943     path->peers[1] = peer_info->id;
4944     GNUNET_PEER_change_rc (peer_info->id, 1);
4945     GNUNET_STATISTICS_update (stats, "# peers", 1, GNUNET_NO);
4946   }
4947   path->peers[0] = myid;
4948   GNUNET_PEER_change_rc (myid, 1);
4949   peer_info_add_path (peer_info, path, GNUNET_YES);
4950   return;
4951 }
4952
4953
4954 /**
4955  * Method called whenever a peer disconnects.
4956  *
4957  * @param cls closure
4958  * @param peer peer identity this notification is about
4959  */
4960 static void
4961 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
4962 {
4963   struct MeshPeerInfo *pi;
4964   struct MeshPeerQueue *q;
4965   struct MeshPeerQueue *n;
4966
4967   DEBUG_CONN ("Peer disconnected\n");
4968   pi = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
4969   if (NULL == pi)
4970   {
4971     GNUNET_break (0);
4972     return;
4973   }
4974   q = pi->queue_head;
4975   while (NULL != q)
4976   {
4977       n = q->next;
4978       /* TODO try to reroute this traffic instead */
4979       queue_destroy(q, GNUNET_YES);
4980       q = n;
4981   }
4982   if (NULL != pi->core_transmit)
4983   {
4984     GNUNET_CORE_notify_transmit_ready_cancel(pi->core_transmit);
4985     pi->core_transmit = NULL;
4986   }
4987     peer_remove_path (pi, pi->id, myid);
4988   if (myid == pi->id)
4989   {
4990     DEBUG_CONN ("     (self)\n");
4991   }
4992   GNUNET_STATISTICS_update (stats, "# peers", -1, GNUNET_NO);
4993   return;
4994 }
4995
4996
4997 /**
4998  * Install server (service) handlers and start listening to clients.
4999  */
5000 static void
5001 server_init (void)
5002 {
5003   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
5004   GNUNET_SERVER_disconnect_notify (server_handle,
5005                                    &handle_local_client_disconnect, NULL);
5006   nc = GNUNET_SERVER_notification_context_create (server_handle, 1);
5007
5008   clients_head = NULL;
5009   clients_tail = NULL;
5010   next_client_id = 0;
5011   GNUNET_SERVER_resume (server_handle);
5012 }
5013
5014
5015 /**
5016  * To be called on core init/fail.
5017  *
5018  * @param cls Closure (config)
5019  * @param server handle to the server for this service
5020  * @param identity the public identity of this peer
5021  */
5022 static void
5023 core_init (void *cls, struct GNUNET_CORE_Handle *server,
5024            const struct GNUNET_PeerIdentity *identity)
5025 {
5026   const struct GNUNET_CONFIGURATION_Handle *c = cls;
5027   static int i = 0;
5028
5029   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Core init\n");
5030   GNUNET_break (core_handle == server);
5031   if (0 != memcmp (identity, &my_full_id, sizeof (my_full_id)) ||
5032     NULL == server)
5033   {
5034     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Wrong CORE service\n"));
5035     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5036                 " core id %s\n",
5037                 GNUNET_i2s (identity));
5038     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5039                 " my id %s\n",
5040                 GNUNET_i2s (&my_full_id));
5041     GNUNET_CORE_disconnect (core_handle);
5042     core_handle = GNUNET_CORE_connect (c, /* Main configuration */
5043                                        NULL,      /* Closure passed to MESH functions */
5044                                        &core_init,        /* Call core_init once connected */
5045                                        &core_connect,     /* Handle connects */
5046                                        &core_disconnect,  /* remove peers on disconnects */
5047                                        NULL,      /* Don't notify about all incoming messages */
5048                                        GNUNET_NO, /* For header only in notification */
5049                                        NULL,      /* Don't notify about all outbound messages */
5050                                        GNUNET_NO, /* For header-only out notification */
5051                                        core_handlers);    /* Register these handlers */
5052     if (10 < i++)
5053       GNUNET_abort();
5054   }
5055   server_init ();
5056   return;
5057 }
5058
5059
5060 /******************************************************************************/
5061 /************************      MAIN FUNCTIONS      ****************************/
5062 /******************************************************************************/
5063
5064 /**
5065  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
5066  *
5067  * @param cls closure
5068  * @param key current key code
5069  * @param value value in the hash map
5070  * @return GNUNET_YES if we should continue to iterate,
5071  *         GNUNET_NO if not.
5072  */
5073 static int
5074 shutdown_tunnel (void *cls, const struct GNUNET_HashCode * key, void *value)
5075 {
5076   struct MeshTunnel *t = value;
5077
5078   tunnel_destroy (t);
5079   return GNUNET_YES;
5080 }
5081
5082 /**
5083  * Iterator over peer hash map entries to destroy the tunnel during shutdown.
5084  *
5085  * @param cls closure
5086  * @param key current key code
5087  * @param value value in the hash map
5088  * @return GNUNET_YES if we should continue to iterate,
5089  *         GNUNET_NO if not.
5090  */
5091 static int
5092 shutdown_peer (void *cls, const struct GNUNET_HashCode * key, void *value)
5093 {
5094   struct MeshPeerInfo *p = value;
5095   struct MeshPeerQueue *q;
5096   struct MeshPeerQueue *n;
5097
5098   q = p->queue_head;
5099   while (NULL != q)
5100   {
5101       n = q->next;
5102       if (q->peer == p)
5103       {
5104         queue_destroy(q, GNUNET_YES);
5105       }
5106       q = n;
5107   }
5108   peer_info_destroy (p);
5109   return GNUNET_YES;
5110 }
5111
5112
5113 /**
5114  * Task run during shutdown.
5115  *
5116  * @param cls unused
5117  * @param tc unused
5118  */
5119 static void
5120 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
5121 {
5122   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shutting down\n");
5123
5124   if (core_handle != NULL)
5125   {
5126     GNUNET_CORE_disconnect (core_handle);
5127     core_handle = NULL;
5128   }
5129   if (NULL != keygen)
5130   {
5131     GNUNET_CRYPTO_ecc_key_create_stop (keygen);
5132     keygen = NULL;
5133   }
5134   GNUNET_CONTAINER_multihashmap_iterate (tunnels, &shutdown_tunnel, NULL);
5135   GNUNET_CONTAINER_multihashmap_iterate (peers, &shutdown_peer, NULL);
5136   if (dht_handle != NULL)
5137   {
5138     GNUNET_DHT_disconnect (dht_handle);
5139     dht_handle = NULL;
5140   }
5141   if (nc != NULL)
5142   {
5143     GNUNET_SERVER_notification_context_destroy (nc);
5144     nc = NULL;
5145   }
5146   if (GNUNET_SCHEDULER_NO_TASK != announce_id_task)
5147   {
5148     GNUNET_SCHEDULER_cancel (announce_id_task);
5149     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
5150   }
5151   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shut down\n");
5152 }
5153
5154
5155 /**
5156  * Callback for hostkey read/generation.
5157  *
5158  * @param cls Closure (Configuration handle).
5159  * @param pk The ECC private key.
5160  * @param emsg Error message, if any.
5161  */
5162 static void
5163 key_generation_cb (void *cls,
5164                    struct GNUNET_CRYPTO_EccPrivateKey *pk,
5165                    const char *emsg)
5166 {
5167   const struct GNUNET_CONFIGURATION_Handle *c = cls;
5168
5169   keygen = NULL;
5170   if (NULL == pk)
5171   {
5172     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
5173                 _("Could not access hostkey: %s. Exiting.\n"),
5174                 emsg);
5175     GNUNET_SCHEDULER_shutdown ();
5176     return;
5177   }
5178   my_private_key = pk;
5179   GNUNET_CRYPTO_ecc_key_get_public (my_private_key, &my_public_key);
5180   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
5181                       &my_full_id.hashPubKey);
5182   myid = GNUNET_PEER_intern (&my_full_id);
5183   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
5184               "Mesh for peer [%s] starting\n",
5185               GNUNET_i2s(&my_full_id));
5186
5187   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
5188                                      NULL,      /* Closure passed to MESH functions */
5189                                      &core_init,        /* Call core_init once connected */
5190                                      &core_connect,     /* Handle connects */
5191                                      &core_disconnect,  /* remove peers on disconnects */
5192                                      NULL,      /* Don't notify about all incoming messages */
5193                                      GNUNET_NO, /* For header only in notification */
5194                                      NULL,      /* Don't notify about all outbound messages */
5195                                      GNUNET_NO, /* For header-only out notification */
5196                                      core_handlers);    /* Register these handlers */
5197   if (core_handle == NULL)
5198   {
5199     GNUNET_break (0);
5200     GNUNET_SCHEDULER_shutdown ();
5201     return;
5202   }
5203
5204   next_tid = 0;
5205   next_local_tid = GNUNET_MESH_LOCAL_TUNNEL_ID_SERV;
5206
5207   announce_id_task = GNUNET_SCHEDULER_add_now (&announce_id, cls);
5208
5209   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Mesh service running\n");
5210 }
5211
5212
5213 /**
5214  * Process mesh requests.
5215  *
5216  * @param cls closure
5217  * @param server the initialized server
5218  * @param c configuration to use
5219  */
5220 static void
5221 run (void *cls, struct GNUNET_SERVER_Handle *server,
5222      const struct GNUNET_CONFIGURATION_Handle *c)
5223 {
5224   char *keyfile;
5225
5226   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "starting to run\n");
5227   server_handle = server;
5228   GNUNET_SERVER_suspend (server_handle);
5229
5230   if (GNUNET_OK !=
5231       GNUNET_CONFIGURATION_get_value_filename (c, "PEER", "PRIVATE_KEY",
5232                                                &keyfile))
5233   {
5234     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5235                 _
5236                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5237                 "mesh", "peer/privatekey");
5238     GNUNET_SCHEDULER_shutdown ();
5239     return;
5240   }
5241
5242   if (GNUNET_OK !=
5243       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_PATH_TIME",
5244                                            &refresh_path_time))
5245   {
5246     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5247                 _
5248                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5249                 "mesh", "refresh path time");
5250     GNUNET_SCHEDULER_shutdown ();
5251     return;
5252   }
5253
5254   if (GNUNET_OK !=
5255       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "ID_ANNOUNCE_TIME",
5256                                            &id_announce_time))
5257   {
5258     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5259                 _
5260                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5261                 "mesh", "id announce time");
5262     GNUNET_SCHEDULER_shutdown ();
5263     return;
5264   }
5265
5266   if (GNUNET_OK !=
5267       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "CONNECT_TIMEOUT",
5268                                            &connect_timeout))
5269   {
5270     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5271                 _
5272                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5273                 "mesh", "connect timeout");
5274     GNUNET_SCHEDULER_shutdown ();
5275     return;
5276   }
5277
5278   if (GNUNET_OK !=
5279       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
5280                                              &max_msgs_queue))
5281   {
5282     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5283                 _
5284                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5285                 "mesh", "max msgs queue");
5286     GNUNET_SCHEDULER_shutdown ();
5287     return;
5288   }
5289
5290   if (GNUNET_OK !=
5291       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_TUNNELS",
5292                                              &max_tunnels))
5293   {
5294     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
5295                 _
5296                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
5297                 "mesh", "max tunnels");
5298     GNUNET_SCHEDULER_shutdown ();
5299     return;
5300   }
5301
5302   if (GNUNET_OK !=
5303       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DEFAULT_TTL",
5304                                              &default_ttl))
5305   {
5306     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
5307                 _
5308                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
5309                 "mesh", "default ttl", 64);
5310     default_ttl = 64;
5311   }
5312
5313   if (GNUNET_OK !=
5314       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS",
5315                                              &max_peers))
5316   {
5317     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
5318                 _("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
5319                 "mesh", "max peers", 1000);
5320     max_peers = 1000;
5321   }
5322
5323   if (GNUNET_OK !=
5324     GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DROP_PERCENT",
5325                                            &drop_percent))
5326   {
5327     drop_percent = 0;
5328   }
5329   else
5330   {
5331     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
5332                 "Mesh is running with drop mode enabled. "
5333                 "This is NOT a good idea! "
5334                 "Remove the DROP_PERCENT option from your configuration.\n");
5335   }
5336
5337   if (GNUNET_OK !=
5338       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DHT_REPLICATION_LEVEL",
5339                                              &dht_replication_level))
5340   {
5341     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
5342                 _
5343                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
5344                 "mesh", "dht replication level", 3);
5345     dht_replication_level = 3;
5346   }
5347
5348   tunnels = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
5349   incoming_tunnels = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
5350   peers = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
5351   ports = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
5352
5353   dht_handle = GNUNET_DHT_connect (c, 64);
5354   if (NULL == dht_handle)
5355   {
5356     GNUNET_break (0);
5357   }
5358   stats = GNUNET_STATISTICS_create ("mesh", c);
5359
5360   /* Scheduled the task to clean up when shutdown is called */
5361   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
5362                                 NULL);
5363   keygen = GNUNET_CRYPTO_ecc_key_create_start (keyfile,
5364                                                &key_generation_cb,
5365                                                (void *) c);
5366   GNUNET_free (keyfile);
5367 }
5368
5369
5370 /**
5371  * The main function for the mesh service.
5372  *
5373  * @param argc number of arguments from the command line
5374  * @param argv command line arguments
5375  * @return 0 ok, 1 on error
5376  */
5377 int
5378 main (int argc, char *const *argv)
5379 {
5380   int ret;
5381   int r;
5382
5383   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main()\n");
5384   r = GNUNET_SERVICE_run (argc, argv, "mesh", GNUNET_SERVICE_OPTION_NONE, &run,
5385                           NULL);
5386   ret = (GNUNET_OK == r) ? 0 : 1;
5387   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main() END\n");
5388
5389   INTERVAL_SHOW;
5390
5391   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
5392               "Mesh for peer [%s] FWD ACKs %u, BCK ACKs %u\n",
5393               GNUNET_i2s(&my_full_id), debug_fwd_ack, debug_bck_ack);
5394
5395   return ret;
5396 }