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