- check hashmap for initialization
[oweals/gnunet.git] / src / mesh / gnunet-service-mesh-enc.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-enc.c
23  * @brief GNUnet MESH service with encryption
24  * @author Bartlomiej Polot
25  *
26  *  FIXME in progress:
27  * - when sending in-order buffered data, wait for client ACKs
28  * - add signatures
29  * - add encryption
30  * - set connection IDs independently from tunnel, tunnel has no ID
31  *
32  * TODO:
33  * - relay corking down to core
34  * - set ttl relative to path length
35  * TODO END
36  * 
37  * Dictionary:
38  * - peer: other mesh instance. If there is direct connection it's a neighbor.
39  * - tunnel: encrypted connection to a peer, neighbor or not.
40  * - channel: connection between two clients, on the same or different peers.
41  *            have properties like reliability.
42  * - path: series of directly connected peer from one peer to another.
43  * - connection: path which is being used in a tunnel.
44  */
45
46 #include "platform.h"
47 #include "gnunet_crypto_lib.h"
48 #include "mesh_enc.h"
49 #include "mesh_protocol_enc.h"
50 #include "mesh_path.h"
51 #include "block_mesh.h"
52 #include "gnunet_dht_service.h"
53 #include "gnunet_statistics_service.h"
54
55 #define MESH_BLOOM_SIZE         128
56
57 #define MESH_DEBUG_DHT          GNUNET_NO
58 #define MESH_DEBUG_CONNECTION   GNUNET_YES
59 #define MESH_DEBUG_TIMING       __LINUX__ && GNUNET_NO
60
61 #define MESH_MAX_POLL_TIME      GNUNET_TIME_relative_multiply (\
62                                   GNUNET_TIME_UNIT_MINUTES,\
63                                   10)
64 #define MESH_RETRANSMIT_TIME    GNUNET_TIME_UNIT_SECONDS
65 #define MESH_RETRANSMIT_MARGIN  4
66
67 #if MESH_DEBUG_CONNECTION
68 #define DEBUG_CONN(...) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
69 #else
70 #define DEBUG_CONN(...)
71 #endif
72
73 #if MESH_DEBUG_DHT
74 #define DEBUG_DHT(...) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
75 #else
76 #define DEBUG_DHT(...)
77 #endif
78
79 #if MESH_DEBUG_TIMING
80 #include <time.h>
81 double __sum;
82 uint64_t __count;
83 struct timespec __mesh_start;
84 struct timespec __mesh_end;
85 #define INTERVAL_START clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &(__mesh_start))
86 #define INTERVAL_END \
87 do {\
88   clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &(__mesh_end));\
89   double __diff = __mesh_end.tv_nsec - __mesh_start.tv_nsec;\
90   if (__diff < 0) __diff += 1000000000;\
91   __sum += __diff;\
92   __count++;\
93 } while (0)
94 #define INTERVAL_SHOW \
95 if (0 < __count)\
96   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "AVG process time: %f ns\n", __sum/__count)
97 #else
98 #define INTERVAL_START
99 #define INTERVAL_END
100 #define INTERVAL_SHOW
101 #endif
102
103 /**
104  * All the states a tunnel can be in.
105  */
106 enum MeshTunnelState
107 {
108     /**
109      * Uninitialized status, should never appear in operation.
110      */
111   MESH_TUNNEL_NEW,
112
113     /**
114      * Path to the peer not known yet
115      */
116   MESH_TUNNEL_SEARCHING,
117
118     /**
119      * Request sent, not yet answered.
120      */
121   MESH_TUNNEL_WAITING,
122
123     /**
124      * Peer connected and ready to accept data
125      */
126   MESH_TUNNEL_READY,
127
128     /**
129      * Peer connected previosly but not responding
130      */
131   MESH_TUNNEL_RECONNECTING
132 };
133
134
135 /**
136  * All the states a connection can be in.
137  */
138 enum MeshConnectionState
139 {
140   /**
141    * Uninitialized status, should never appear in operation.
142    */
143   MESH_CONNECTION_NEW,
144
145   /**
146    * Connection create message sent, waiting for ACK.
147    */
148   MESH_CONNECTION_SENT,
149
150   /**
151    * Connection ACK sent, waiting for ACK.
152    */
153   MESH_CONNECTION_ACK,
154
155   /**
156    * Connection confirmed, ready to carry traffic.
157    */
158   MESH_CONNECTION_READY,
159 };
160
161
162 /**
163  * All the states a connection can be in.
164  */
165 enum MeshChannelState
166 {
167   /**
168    * Uninitialized status, should never appear in operation.
169    */
170   MESH_CHANNEL_NEW,
171
172   /**
173    * Connection create message sent, waiting for ACK.
174    */
175   MESH_CHANNEL_SENT,
176
177   /**
178    * Connection confirmed, ready to carry traffic..
179    */
180   MESH_CHANNEL_READY,
181 };
182
183
184 /******************************************************************************/
185 /************************      DATA STRUCTURES     ****************************/
186 /******************************************************************************/
187
188 /** FWD declaration */
189 struct MeshClient;
190 struct MeshPeer;
191 struct MeshTunnel2;
192 struct MeshConnection;
193 struct MeshChannel;
194 struct MeshChannelReliability;
195
196
197 /**
198  * Struct containing info about a queued transmission to this peer
199  */
200 struct MeshPeerQueue
201 {
202     /**
203       * DLL next
204       */
205   struct MeshPeerQueue *next;
206
207     /**
208       * DLL previous
209       */
210   struct MeshPeerQueue *prev;
211
212     /**
213      * Peer this transmission is directed to.
214      */
215   struct MeshPeer *peer;
216
217     /**
218      * Connection this message belongs to.
219      */
220   struct MeshConnection *c;
221
222     /**
223      * Is FWD in c?
224      */
225   int fwd;
226
227     /**
228      * Channel this message belongs to, if known.
229      */
230   struct MeshChannel *ch;
231
232     /**
233      * Pointer to info stucture used as cls.
234      */
235   void *cls;
236
237     /**
238      * Type of message
239      */
240   uint16_t type;
241
242     /**
243      * Size of the message
244      */
245   size_t size;
246 };
247
248
249 /**
250  * Struct to encapsulate all the Flow Control information to a peer to which
251  * we are directly connected (on a core level).
252  */
253 struct MeshFlowControl
254 {
255   /**
256    * Connection this controls.
257    */
258   struct MeshConnection *c;
259
260   /**
261    * How many messages are in the queue on this connection.
262    */
263   unsigned int queue_n;
264
265   /**
266    * How many messages do we accept in the queue.
267    */
268   unsigned int queue_max;
269
270   /**
271    * Next ID to use.
272    */
273   uint32_t next_pid;
274
275   /**
276    * ID of the last packet sent towards the peer.
277    */
278   uint32_t last_pid_sent;
279
280   /**
281    * ID of the last packet received from the peer.
282    */
283   uint32_t last_pid_recv;
284
285   /**
286    * Last ACK sent to the peer (peer can't send more than this PID).
287    */
288   uint32_t last_ack_sent;
289
290   /**
291    * Last ACK sent towards the origin (for traffic towards leaf node).
292    */
293   uint32_t last_ack_recv;
294
295   /**
296    * Task to poll the peer in case of a lost ACK causes stall.
297    */
298   GNUNET_SCHEDULER_TaskIdentifier poll_task;
299
300   /**
301    * How frequently to poll for ACKs.
302    */
303   struct GNUNET_TIME_Relative poll_time;
304 };
305
306
307 /**
308  * Struct containing all information regarding a given peer
309  */
310 struct MeshPeer
311 {
312     /**
313      * ID of the peer
314      */
315   GNUNET_PEER_Id id;
316
317     /**
318      * Last time we heard from this peer
319      */
320   struct GNUNET_TIME_Absolute last_contact;
321
322     /**
323      * Paths to reach the peer, ordered by ascending hop count
324      */
325   struct MeshPeerPath *path_head;
326
327     /**
328      * Paths to reach the peer, ordered by ascending hop count
329      */
330   struct MeshPeerPath *path_tail;
331
332     /**
333      * Handle to stop the DHT search for paths to this peer
334      */
335   struct GNUNET_DHT_GetHandle *dhtget;
336
337     /**
338      * Tunnel to this peer, if any.
339      */
340   struct MeshTunnel2 *tunnel;
341
342     /**
343      * Connections that go through this peer, indexed by tid;
344      */
345   struct GNUNET_CONTAINER_MultiHashMap *connections;
346
347     /**
348      * Handle for queued transmissions
349      */
350   struct GNUNET_CORE_TransmitHandle *core_transmit;
351
352   /**
353    * Transmission queue to core DLL head
354    */
355   struct MeshPeerQueue *queue_head;
356   
357   /**
358    * Transmission queue to core DLL tail
359    */
360   struct MeshPeerQueue *queue_tail;
361
362   /**
363    * How many messages are in the queue to this peer.
364    */
365   unsigned int queue_n;
366 };
367
368
369 /**
370  * Info needed to retry a message in case it gets lost.
371  */
372 struct MeshReliableMessage
373 {
374     /**
375      * Double linked list, FIFO style
376      */
377   struct MeshReliableMessage    *next;
378   struct MeshReliableMessage    *prev;
379
380     /**
381      * Type of message (payload, channel management).
382      */
383   int16_t type;
384
385     /**
386      * Tunnel Reliability queue this message is in.
387      */
388   struct MeshChannelReliability  *rel;
389
390     /**
391      * ID of the message (ACK needed to free)
392      */
393   uint32_t                      mid;
394
395     /**
396      * When was this message issued (to calculate ACK delay)
397      */
398   struct GNUNET_TIME_Absolute   timestamp;
399
400   /* struct GNUNET_MESH_Data with payload */
401 };
402
403
404 /**
405  * Info about the traffic state for a client in a channel.
406  */
407 struct MeshChannelReliability
408 {
409     /**
410      * Channel this is about.
411      */
412   struct MeshChannel *ch;
413
414     /**
415      * DLL of messages sent and not yet ACK'd.
416      */
417   struct MeshReliableMessage        *head_sent;
418   struct MeshReliableMessage        *tail_sent;
419
420     /**
421      * Messages pending to send.
422      */
423   unsigned int                      n_sent;
424
425     /**
426      * DLL of messages received out of order.
427      */
428   struct MeshReliableMessage        *head_recv;
429   struct MeshReliableMessage        *tail_recv;
430
431     /**
432      * Messages received.
433      */
434   unsigned int                      n_recv;
435
436     /**
437      * Next MID to use for outgoing traffic.
438      */
439   uint32_t                          mid_send;
440
441     /**
442      * Next MID expected for incoming traffic.
443      */
444   uint32_t                          mid_recv;
445
446     /**
447      * Can we send data to the client?
448      */
449   int                               client_ready;
450
451     /**
452      * Task to resend/poll in case no ACK is received.
453      */
454   GNUNET_SCHEDULER_TaskIdentifier   retry_task;
455
456     /**
457      * Counter for exponential backoff.
458      */
459   struct GNUNET_TIME_Relative       retry_timer;
460
461     /**
462      * How long does it usually take to get an ACK.
463      */
464   struct GNUNET_TIME_Relative       expected_delay;
465 };
466
467
468 /**
469  * Struct containing all information regarding a channel to a remote client.
470  */
471 struct MeshChannel
472 {
473     /**
474      * Tunnel this channel is in.
475      */
476   struct MeshTunnel2 *t;
477
478     /**
479      * Double linked list.
480      */
481   struct MeshChannel    *next;
482   struct MeshChannel    *prev;
483
484     /**
485      * Destination port of the channel.
486      */
487   uint32_t port;
488
489     /**
490      * Global channel number ( < GNUNET_MESH_LOCAL_CHANNEL_ID_CLI)
491      */
492   MESH_ChannelNumber gid;
493
494     /**
495      * Local tunnel number for root (owner) client.
496      * ( >= GNUNET_MESH_LOCAL_CHANNEL_ID_CLI or 0 )
497      */
498   MESH_ChannelNumber lid_root;
499
500     /**
501      * Local tunnel number for local destination clients (incoming number)
502      * ( >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV or 0).
503      */
504   MESH_ChannelNumber lid_dest;
505
506     /**
507      * Channel state.
508      */
509   enum MeshChannelState state;
510
511     /**
512      * Is the tunnel bufferless (minimum latency)?
513      */
514   int nobuffer;
515
516     /**
517      * Is the tunnel reliable?
518      */
519   int reliable;
520
521     /**
522      * Last time the channel was used
523      */
524   struct GNUNET_TIME_Absolute timestamp;
525
526     /**
527      * Client owner of the tunnel, if any
528      */
529   struct MeshClient *root;
530
531     /**
532      * Client destination of the tunnel, if any.
533      */
534   struct MeshClient *dest;
535
536     /**
537      * Flag to signal the destruction of the channel.
538      * If this is set GNUNET_YES the channel will be destroyed
539      * when the queue is empty.
540      */
541   int destroy;
542
543     /**
544      * Total messages pending for this channel, payload or not.
545      */
546   unsigned int pending_messages;
547
548     /**
549      * Reliability data.
550      * Only present (non-NULL) at the owner of a tunnel.
551      */
552   struct MeshChannelReliability *root_rel;
553
554     /**
555      * Reliability data.
556      * Only present (non-NULL) at the destination of a tunnel.
557      */
558   struct MeshChannelReliability *dest_rel;
559
560 };
561
562
563 /**
564  * Struct containing all information regarding a connection to a peer.
565  */
566 struct MeshConnection
567 {
568   /**
569    * DLL
570    */
571   struct MeshConnection *next;
572   struct MeshConnection *prev;
573
574   /**
575    * Tunnel this connection is part of.
576    */
577   struct MeshTunnel2 *t;
578
579   /**
580    * Flow control information for traffic fwd.
581    */
582   struct MeshFlowControl fwd_fc;
583
584   /**
585    * Flow control information for traffic bck.
586    */
587   struct MeshFlowControl bck_fc;
588
589   /**
590    * ID of the connection.
591    */
592   struct GNUNET_HashCode id;
593
594   /**
595    * State of the connection.
596    */
597   enum MeshConnectionState state;
598
599   /**
600    * Path being used for the tunnel.
601    */
602   struct MeshPeerPath *path;
603
604   /**
605    * Position of the local peer in the path.
606    */
607   unsigned int own_pos;
608
609   /**
610    * Task to keep the used paths alive at the owner,
611    * time tunnel out on all the other peers.
612    */
613   GNUNET_SCHEDULER_TaskIdentifier fwd_maintenance_task;
614
615   /**
616    * Task to keep the used paths alive at the destination,
617    * time tunnel out on all the other peers.
618    */
619   GNUNET_SCHEDULER_TaskIdentifier bck_maintenance_task;
620
621   /**
622    * Pending message count.
623    */
624   int pending_messages;
625
626   /**
627    * Destroy flag: if true, destroy on last message.
628    */
629   int destroy;
630 };
631
632
633 /**
634  * Struct used to queue messages in a tunnel.
635  */
636 struct MeshTunnelQueue
637 {
638   /**
639    * DLL
640    */
641   struct MeshTunnelQueue *next;
642   struct MeshTunnelQueue *prev;
643
644   /**
645    * Channel.
646    */
647   struct MeshChannel *ch;
648
649   /**
650    * Message to send.
651    */
652   /* struct GNUNET_MessageHeader *msg; */
653 };
654
655
656 /**
657  * Struct containing all information regarding a tunnel to a peer.
658  */
659 struct MeshTunnel2
660 {
661     /**
662      * Endpoint of the tunnel.
663      */
664   struct MeshPeer *peer;
665
666     /**
667      * State of the tunnel.
668      */
669   enum MeshTunnelState state;
670
671   /**
672    * Local peer ephemeral private key
673    */
674   struct GNUNET_CRYPTO_EccPrivateKey *my_eph_key;
675
676   /**
677    * Local peer ephemeral public key
678    */
679   struct GNUNET_CRYPTO_EccPublicKey *my_eph;
680
681   /**
682    * Remote peer's public key.
683    */
684   struct GNUNET_CRYPTO_EccPublicKey *peers_eph;
685
686   /**
687    * Encryption ("our") key.
688    */
689   struct GNUNET_CRYPTO_AesSessionKey e_key;
690
691   /**
692    * Decryption ("their") key.
693    */
694   struct GNUNET_CRYPTO_AesSessionKey d_key;
695
696   /**
697    * Paths that are actively used to reach the destination peer.
698    */
699   struct MeshConnection *connection_head;
700   struct MeshConnection *connection_tail;
701
702   /**
703    * Next connection number.
704    */
705   uint32_t next_cid;
706
707   /**
708    * Channels inside this tunnel.
709    */
710   struct MeshChannel *channel_head;
711   struct MeshChannel *channel_tail;
712
713   /**
714    * Channel ID for the next created channel.
715    */
716   MESH_ChannelNumber next_chid;
717
718   /**
719    * Channel ID for the next incoming channel.
720    */
721   MESH_ChannelNumber next_local_chid;
722
723   /**
724    * Pending message count.
725    */
726   int pending_messages;
727
728   /**
729    * Destroy flag: if true, destroy on last message.
730    */
731   int destroy;
732
733   /**
734    * Queued messages, to transmit once tunnel gets connected.
735    */
736   struct MeshTunnelQueue *tq_head;
737   struct MeshTunnelQueue *tq_tail;
738 };
739
740
741
742 /**
743  * Struct containing information about a client of the service
744  * 
745  * TODO: add a list of 'waiting' ports
746  */
747 struct MeshClient
748 {
749     /**
750      * Linked list next
751      */
752   struct MeshClient *next;
753
754     /**
755      * Linked list prev
756      */
757   struct MeshClient *prev;
758
759     /**
760      * Tunnels that belong to this client, indexed by local id
761      */
762   struct GNUNET_CONTAINER_MultiHashMap32 *own_channels;
763
764    /**
765      * Tunnels this client has accepted, indexed by incoming local id
766      */
767   struct GNUNET_CONTAINER_MultiHashMap32 *incoming_channels;
768
769     /**
770      * Handle to communicate with the client
771      */
772   struct GNUNET_SERVER_Client *handle;
773
774     /**
775      * Ports that this client has declared interest in.
776      * Indexed by port, contains *Client.
777      */
778   struct GNUNET_CONTAINER_MultiHashMap32 *ports;
779
780     /**
781      * Whether the client is active or shutting down (don't send confirmations
782      * to a client that is shutting down.
783      */
784   int shutting_down;
785
786     /**
787      * ID of the client, mainly for debug messages
788      */
789   unsigned int id;
790 };
791
792
793 /******************************************************************************/
794 /************************      DEBUG FUNCTIONS     ****************************/
795 /******************************************************************************/
796
797 #if MESH_DEBUG
798 /**
799  * GNUNET_SCHEDULER_Task for printing a message after some operation is done
800  * @param cls string to print
801  * @param success  GNUNET_OK if the PUT was transmitted,
802  *                GNUNET_NO on timeout,
803  *                GNUNET_SYSERR on disconnect from service
804  *                after the PUT message was transmitted
805  *                (so we don't know if it was received or not)
806  */
807
808 #if 0
809 static void
810 mesh_debug (void *cls, int success)
811 {
812   char *s = cls;
813
814   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "%s (%d)\n", s, success);
815 }
816 #endif
817
818 #endif
819
820 /******************************************************************************/
821 /***********************      GLOBAL VARIABLES     ****************************/
822 /******************************************************************************/
823
824 /************************** Configuration parameters **************************/
825
826 /**
827  * How often to send path keepalives. Paths timeout after 4 missed.
828  */
829 static struct GNUNET_TIME_Relative refresh_connection_time;
830
831 /**
832  * How often to PUT own ID in the DHT.
833  */
834 static struct GNUNET_TIME_Relative id_announce_time;
835
836 /**
837  * Maximum time allowed to connect to a peer found by string.
838  */
839 static struct GNUNET_TIME_Relative connect_timeout;
840
841 /**
842  * Default TTL for payload packets.
843  */
844 static unsigned long long default_ttl;
845
846 /**
847  * DHT replication level, see DHT API: GNUNET_DHT_get_start, GNUNET_DHT_put.
848  */
849 static unsigned long long dht_replication_level;
850
851 /**
852  * How many connections are we willing to maintain.
853  * Local connections are always allowed, even if there are more connections than max.
854  */
855 static unsigned long long max_connections;
856
857 /**
858  * How many messages *in total* are we willing to queue, divide by number of 
859  * connections to get connection queue size.
860  */
861 static unsigned long long max_msgs_queue;
862
863 /**
864  * How many peers do we want to remember?
865  */
866 static unsigned long long max_peers;
867
868 /**
869  * Percentage of messages that will be dropped (for test purposes only).
870  */
871 static unsigned long long drop_percent;
872
873 /*************************** Static global variables **************************/
874
875 /**
876  * DLL with all the clients, head.
877  */
878 static struct MeshClient *clients_head;
879
880 /**
881  * DLL with all the clients, tail.
882  */
883 static struct MeshClient *clients_tail;
884
885 /**
886  * Connections known, indexed by cid (MeshConnection).
887  */
888 static struct GNUNET_CONTAINER_MultiHashMap *connections;
889
890 /**
891  * Peers known, indexed by PeerIdentity (MeshPeer).
892  */
893 static struct GNUNET_CONTAINER_MultiHashMap *peers;
894
895 /**
896  * Handle to communicate with core.
897  */
898 static struct GNUNET_CORE_Handle *core_handle;
899
900 /**
901  * Handle to use DHT.
902  */
903 static struct GNUNET_DHT_Handle *dht_handle;
904
905 /**
906  * Handle to server lib.
907  */
908 static struct GNUNET_SERVER_Handle *server_handle;
909
910 /**
911  * Handle to the statistics service.
912  */
913 static struct GNUNET_STATISTICS_Handle *stats;
914
915 /**
916  * Notification context, to send messages to local clients.
917  */
918 static struct GNUNET_SERVER_NotificationContext *nc;
919
920 /**
921  * Local peer own ID (memory efficient handle).
922  */
923 static GNUNET_PEER_Id myid;
924
925 /**
926  * Local peer own ID (full value).
927  */
928 static struct GNUNET_PeerIdentity my_full_id;
929
930 /**
931  * Own private key.
932  */
933 static struct GNUNET_CRYPTO_EccPrivateKey *my_private_key;
934
935 /**
936  * Own public key.
937  */
938 static struct GNUNET_CRYPTO_EccPublicKey my_public_key;
939
940 /**
941  * All ports clients of this peer have opened.
942  */
943 static struct GNUNET_CONTAINER_MultiHashMap32 *ports;
944
945 /**
946  * Task to periodically announce itself in the network.
947  */
948 GNUNET_SCHEDULER_TaskIdentifier announce_id_task;
949
950 /**
951  * Next ID to assign to a client.
952  */
953 unsigned int next_client_id;
954
955
956 /******************************************************************************/
957 /***********************         DECLARATIONS        **************************/
958 /******************************************************************************/
959
960 /**
961  * Function to process paths received for a new peer addition. The recorded
962  * paths form the initial tunnel, which can be optimized later.
963  * Called on each result obtained for the DHT search.
964  *
965  * @param cls closure
966  * @param exp when will this value expire
967  * @param key key of the result
968  * @param type type of the result
969  * @param size number of bytes in data
970  * @param data pointer to the result data
971  */
972 static void
973 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
974                     const struct GNUNET_HashCode * key,
975                     const struct GNUNET_PeerIdentity *get_path,
976                     unsigned int get_path_length,
977                     const struct GNUNET_PeerIdentity *put_path,
978                     unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
979                     size_t size, const void *data);
980
981
982 /**
983  * Retrieve the MeshPeer stucture associated with the peer, create one
984  * and insert it in the appropriate structures if the peer is not known yet.
985  *
986  * @param peer Full identity of the peer.
987  *
988  * @return Existing or newly created peer info.
989  */
990 static struct MeshPeer *
991 peer_get (const struct GNUNET_PeerIdentity *peer);
992
993
994 /**
995  * Retrieve the MeshPeer stucture associated with the peer, create one
996  * and insert it in the appropriate structures if the peer is not known yet.
997  *
998  * @param peer Short identity of the peer.
999  *
1000  * @return Existing or newly created peer info.
1001  */
1002 static struct MeshPeer *
1003 peer_get_short (const GNUNET_PEER_Id peer);
1004
1005
1006 /**
1007  * Build a PeerPath from the paths returned from the DHT, reversing the paths
1008  * to obtain a local peer -> destination path and interning the peer ids.
1009  *
1010  * @return Newly allocated and created path
1011  */
1012 static struct MeshPeerPath *
1013 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
1014                      unsigned int get_path_length,
1015                      const struct GNUNET_PeerIdentity *put_path,
1016                      unsigned int put_path_length);
1017
1018
1019 /**
1020  * Adds a path to the data structs of all the peers in the path
1021  *
1022  * @param p Path to process.
1023  * @param confirmed Whether we know if the path works or not.
1024  */
1025 static void
1026 path_add_to_peers (struct MeshPeerPath *p, int confirmed);
1027
1028
1029 /**
1030  * Search for a tunnel by global ID using full PeerIdentities.
1031  *
1032  * @param t Tunnel containing the channel.
1033  * @param chid Public channel number.
1034  *
1035  * @return channel handler, NULL if doesn't exist
1036  */
1037 static struct MeshChannel *
1038 channel_get (struct MeshTunnel2 *t, MESH_ChannelNumber chid);
1039
1040 /**
1041  * Modify the data message ID from global to local and send to client.
1042  * 
1043  * @param ch Channel on which to send the message.
1044  * @param msg Message to modify and send.
1045  * @param fwd Forward?
1046  */
1047 static void
1048 channel_send_client_data (struct MeshChannel *ch,
1049                           const struct GNUNET_MESH_Data *msg,
1050                           int fwd);
1051
1052
1053 /**
1054  * Change the tunnel state.
1055  *
1056  * @param t Tunnel whose state to change.
1057  * @param state New state.
1058  */
1059 static void
1060 tunnel_change_state (struct MeshTunnel2 *t, enum MeshTunnelState state);
1061
1062
1063 /**
1064  * Notify a tunnel that a connection has broken that affects at least
1065  * some of its peers.
1066  *
1067  * @param t Tunnel affected.
1068  * @param p1 Peer that got disconnected from p2.
1069  * @param p2 Peer that got disconnected from p1.
1070  *
1071  * @return Short ID of the peer disconnected (either p1 or p2).
1072  *         0 if the tunnel remained unaffected.
1073  */
1074 static GNUNET_PEER_Id
1075 tunnel_notify_connection_broken (struct MeshTunnel2 *t,
1076                                  GNUNET_PEER_Id p1, GNUNET_PEER_Id p2);
1077
1078 /**
1079  * @brief Use the given path for the tunnel.
1080  * Update the next and prev hops (and RCs).
1081  * (Re)start the path refresh in case the tunnel is locally owned.
1082  * 
1083  * @param t Tunnel to update.
1084  * @param p Path to use.
1085  *
1086  * @return Connection created.
1087  */
1088 static struct MeshConnection *
1089 tunnel_use_path (struct MeshTunnel2 *t, struct MeshPeerPath *p);
1090
1091 /**
1092  * Tunnel is empty: destroy it.
1093  * 
1094  * Notifies all participants (peers, cleints) about the destruction.
1095  * 
1096  * @param t Tunnel to destroy. 
1097  */
1098 static void
1099 tunnel_destroy_empty (struct MeshTunnel2 *t);
1100
1101 /**
1102  * Destroy the tunnel.
1103  *
1104  * This function does not generate any warning traffic to clients or peers.
1105  *
1106  * Tasks:
1107  * Cancel messages belonging to this tunnel queued to neighbors.
1108  * Free any allocated resources linked to the tunnel.
1109  *
1110  * @param t The tunnel to destroy.
1111  */
1112 static void
1113 tunnel_destroy (struct MeshTunnel2 *t);
1114
1115 /**
1116  * Create a connection.
1117  *
1118  * @param cid Connection ID.
1119  */
1120 static struct MeshConnection *
1121 connection_new (const struct GNUNET_HashCode *cid);
1122
1123 /**
1124  * Connection is no longer needed: destroy it and remove from tunnel.
1125  *
1126  * @param c Connection to destroy.
1127  */
1128 static void
1129 connection_destroy (struct MeshConnection *c);
1130
1131 /**
1132  * Send FWD keepalive packets for a connection.
1133  *
1134  * @param cls Closure (connection for which to send the keepalive).
1135  * @param tc Notification context.
1136  */
1137 static void
1138 connection_fwd_keepalive (void *cls,
1139                           const struct GNUNET_SCHEDULER_TaskContext *tc);
1140
1141 /**
1142  * Send BCK keepalive packets for a connection.
1143  *
1144  * @param cls Closure (connection for which to send the keepalive).
1145  * @param tc Notification context.
1146  */
1147 static void
1148 connection_bck_keepalive (void *cls,
1149                           const struct GNUNET_SCHEDULER_TaskContext *tc);
1150
1151
1152 /**
1153  * Change the tunnel state.
1154  *
1155  * @param c Connection whose state to change.
1156  * @param state New state.
1157  */
1158 static void
1159 connection_change_state (struct MeshConnection* c,
1160                          enum MeshConnectionState state);
1161
1162
1163
1164 /**
1165  * @brief Queue and pass message to core when possible.
1166  *
1167  * @param cls Closure (@c type dependant). It will be used by queue_send to
1168  *            build the message to be sent if not already prebuilt.
1169  * @param type Type of the message, 0 for a raw message.
1170  * @param size Size of the message.
1171  * @param c Connection this message belongs to (cannot be NULL).
1172  * @param ch Channel this message belongs to, if applicable (otherwise NULL).
1173  * @param fwd Is this a message going root->dest? (FWD ACK are NOT FWD!)
1174  */
1175 static void
1176 queue_add (void* cls,
1177            uint16_t type,
1178            size_t size,
1179            struct MeshConnection* c,
1180            struct MeshChannel* ch,
1181            int fwd);
1182
1183
1184 /**
1185  * Free a transmission that was already queued with all resources
1186  * associated to the request.
1187  *
1188  * @param queue Queue handler to cancel.
1189  * @param clear_cls Is it necessary to free associated cls?
1190  */
1191 static void
1192 queue_destroy (struct MeshPeerQueue *queue, int clear_cls);
1193
1194
1195 /**
1196  * Core callback to write a queued packet to core buffer
1197  *
1198  * @param cls Closure (peer info).
1199  * @param size Number of bytes available in buf.
1200  * @param buf Where the to write the message.
1201  *
1202  * @return number of bytes written to buf
1203  */
1204 static size_t
1205 queue_send (void *cls, size_t size, void *buf);
1206
1207
1208 /**
1209  * Demultiplex by message type and call appropriate handler for a message
1210  * towards a channel of a local tunnel.
1211  *
1212  * @param t Tunnel this message came on.
1213  * @param msgh Message header.
1214  * @param fwd Is this message fwd?
1215  */
1216 static void
1217 handle_decrypted (struct MeshTunnel2 *t,
1218                   const struct GNUNET_MessageHeader *msgh,
1219                   int fwd);
1220
1221
1222 /**
1223  * Dummy function to separate declarations from definitions in function list.
1224  */
1225 void
1226 __mesh_divider______________________________________________________________();
1227
1228
1229 /**
1230  * Get string description for tunnel state.
1231  *
1232  * @param s Tunnel state.
1233  *
1234  * @return String representation. 
1235  */
1236 static const char *
1237 GNUNET_MESH_DEBUG_TS2S (enum MeshTunnelState s)
1238 {
1239   static char buf[128];
1240
1241   switch (s)
1242   {
1243     case MESH_TUNNEL_NEW:
1244       return "MESH_TUNNEL_NEW";
1245     case MESH_TUNNEL_SEARCHING:
1246       return "MESH_TUNNEL_SEARCHING";
1247     case MESH_TUNNEL_WAITING:
1248       return "MESH_TUNNEL_WAITING";
1249     case MESH_TUNNEL_READY:
1250       return "MESH_TUNNEL_READY";
1251     case MESH_TUNNEL_RECONNECTING:
1252       return "MESH_TUNNEL_RECONNECTING";
1253
1254     default:
1255       sprintf (buf, "%u (UNKNOWN STATE)", s);
1256       return buf;
1257   }
1258 }
1259
1260
1261 /**
1262  * Get string description for tunnel state.
1263  *
1264  * @param s Tunnel state.
1265  *
1266  * @return String representation. 
1267  */
1268 static const char *
1269 GNUNET_MESH_DEBUG_CS2S (enum MeshTunnelState s)
1270 {
1271   switch (s) 
1272   {
1273     case MESH_CONNECTION_NEW:
1274       return "MESH_CONNECTION_NEW";
1275     case MESH_CONNECTION_SENT:
1276       return "MESH_CONNECTION_SENT";
1277     case MESH_CONNECTION_READY:
1278       return "MESH_CONNECTION_READY";
1279     default:
1280       return "MESH_CONNECTION_STATE_ERROR";
1281   }
1282 }
1283
1284
1285
1286 /******************************************************************************/
1287 /************************    PERIODIC FUNCTIONS    ****************************/
1288 /******************************************************************************/
1289
1290 /**
1291  * Periodically announce self id in the DHT
1292  *
1293  * @param cls closure
1294  * @param tc task context
1295  */
1296 static void
1297 announce_id (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1298 {
1299   struct PBlock block;
1300
1301   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1302   {
1303     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
1304     return;
1305   }
1306   /* TODO
1307    * - Set data expiration in function of X
1308    * - Adapt X to churn
1309    */
1310   DEBUG_DHT ("DHT_put for ID %s started.\n", GNUNET_i2s (&my_full_id));
1311
1312   block.id = my_full_id;
1313   GNUNET_DHT_put (dht_handle,   /* DHT handle */
1314                   &my_full_id.hashPubKey,       /* Key to use */
1315                   dht_replication_level,     /* Replication level */
1316                   GNUNET_DHT_RO_RECORD_ROUTE | GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,    /* DHT options */
1317                   GNUNET_BLOCK_TYPE_MESH_PEER,       /* Block type */
1318                   sizeof (block),  /* Size of the data */
1319                   (const char *) &block, /* Data itself */
1320                   GNUNET_TIME_UNIT_FOREVER_ABS,  /* Data expiration */
1321                   GNUNET_TIME_UNIT_FOREVER_REL, /* Retry time */
1322                   NULL,         /* Continuation */
1323                   NULL);        /* Continuation closure */
1324   announce_id_task =
1325       GNUNET_SCHEDULER_add_delayed (id_announce_time, &announce_id, cls);
1326 }
1327
1328
1329 /******************************************************************************/
1330 /******************      GENERAL HELPER FUNCTIONS      ************************/
1331 /******************************************************************************/
1332
1333
1334 /**
1335  * Get the static string for a peer ID.
1336  *
1337  * @param peer Peer.
1338  *
1339  * @return Static string for it's ID.
1340  */
1341 static const char *
1342 peer2s (const struct MeshPeer *peer)
1343 {
1344   if (NULL == peer)
1345     return "(NULL)";
1346   return GNUNET_i2s (GNUNET_PEER_resolve2 (peer->id));
1347 }
1348
1349
1350 /**
1351  * Get the previous hop in a connection
1352  *
1353  * @param c Connection.
1354  *
1355  * @return Previous peer in the connection.
1356  */
1357 static struct MeshPeer *
1358 connection_get_prev_hop (struct MeshConnection *c)
1359 {
1360   GNUNET_PEER_Id id;
1361
1362   if (0 == c->own_pos || c->path->length < 2)
1363     id = c->path->peers[0];
1364   else
1365     id = c->path->peers[c->own_pos - 1];
1366
1367   return peer_get_short (id);
1368 }
1369
1370
1371 /**
1372  * Get the next hop in a connection
1373  *
1374  * @param c Connection.
1375  *
1376  * @return Next peer in the connection. 
1377  */
1378 static struct MeshPeer *
1379 connection_get_next_hop (struct MeshConnection *c)
1380 {
1381   GNUNET_PEER_Id id;
1382
1383   if ((c->path->length - 1) == c->own_pos || c->path->length < 2)
1384     id = c->path->peers[c->path->length - 1];
1385   else
1386     id = c->path->peers[c->own_pos + 1];
1387
1388   return peer_get_short (id);
1389 }
1390
1391
1392 /**
1393  * Get the hop in a connection.
1394  *
1395  * @param c Connection.
1396  * @param fwd Next hop?
1397  *
1398  * @return Next peer in the connection. 
1399  */
1400 static struct MeshPeer *
1401 connection_get_hop (struct MeshConnection *c, int fwd)
1402 {
1403   if (fwd)
1404     return connection_get_next_hop (c);
1405   return connection_get_prev_hop (c);
1406 }
1407
1408 /**
1409  * Check if client has registered with the service and has not disconnected
1410  *
1411  * @param client the client to check
1412  *
1413  * @return non-NULL if client exists in the global DLL
1414  */
1415 static struct MeshClient *
1416 client_get (struct GNUNET_SERVER_Client *client)
1417 {
1418   return GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
1419 }
1420
1421
1422 /**
1423  * Deletes a tunnel from a client (either owner or destination).
1424  *
1425  * @param c Client whose tunnel to delete.
1426  * @param ch Channel which should be deleted.
1427  */
1428 static void
1429 client_delete_channel (struct MeshClient *c, struct MeshChannel *ch)
1430 {
1431   int res;
1432
1433   if (c == ch->root)
1434   {
1435     res = GNUNET_CONTAINER_multihashmap32_remove (c->own_channels,
1436                                                   ch->lid_root, ch);
1437     if (GNUNET_YES != res)
1438       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client_delete_channel owner KO\n");
1439   }
1440   if (c == ch->dest)
1441   {
1442     res = GNUNET_CONTAINER_multihashmap32_remove (c->incoming_channels,
1443                                                   ch->lid_dest, ch);
1444     if (GNUNET_YES != res)
1445       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client_delete_tunnel client KO\n");
1446   }
1447 }
1448
1449
1450 /**
1451  * Notify the appropriate client that a new incoming channel was created.
1452  *
1453  * @param ch Channel that was created.
1454  */
1455 static void
1456 send_local_channel_create (struct MeshChannel *ch)
1457 {
1458   struct GNUNET_MESH_ChannelMessage msg;
1459   struct MeshTunnel2 *t = ch->t;
1460
1461   if (NULL == ch->dest)
1462     return;
1463   msg.header.size = htons (sizeof (msg));
1464   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_CREATE);
1465   msg.channel_id = htonl (ch->lid_dest);
1466   msg.port = htonl (ch->port);
1467   msg.opt = 0;
1468   msg.opt |= GNUNET_YES == ch->reliable ? GNUNET_MESH_OPTION_RELIABLE : 0;
1469   msg.opt |= GNUNET_YES == ch->nobuffer ? GNUNET_MESH_OPTION_NOBUFFER : 0;
1470   msg.opt = htonl (msg.opt);
1471   GNUNET_PEER_resolve (t->peer->id, &msg.peer);
1472   GNUNET_SERVER_notification_context_unicast (nc, ch->dest->handle,
1473                                               &msg.header, GNUNET_NO);
1474 }
1475
1476
1477 /**
1478  * Notify a client that the incoming tunnel is no longer valid.
1479  *
1480  * @param ch Channel that is destroyed.
1481  * @param fwd Forward notification (owner->dest)?
1482  */
1483 static void
1484 send_local_channel_destroy (struct MeshChannel *ch, int fwd)
1485 {
1486   struct GNUNET_MESH_ChannelMessage msg;
1487   struct MeshClient *c;
1488
1489   c = fwd ? ch->dest : ch->root;
1490   if (NULL == c)
1491   {
1492     GNUNET_break (0);
1493     return;
1494   }
1495   msg.header.size = htons (sizeof (msg));
1496   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_TUNNEL_DESTROY);
1497   msg.channel_id = htonl (fwd ? ch->lid_dest : ch->lid_root);
1498   msg.port = htonl (0);
1499   memset (&msg.peer, 0, sizeof (msg.peer));
1500   msg.opt = htonl (0);
1501   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
1502                                               &msg.header, GNUNET_NO);
1503 }
1504
1505
1506 /**
1507  * Build a local ACK message and send it to a local client, if needed.
1508  *
1509  * If the client was already allowed to send data, do nothing.
1510  *
1511  * @param ch Channel on which to send the ACK.
1512  * @param c Client to whom send the ACK.
1513  * @param fwd Set to GNUNET_YES for FWD ACK (dest->root)
1514  */
1515 static void
1516 send_local_ack (struct MeshChannel *ch, int fwd)
1517 {
1518   struct GNUNET_MESH_LocalAck msg;
1519   struct MeshChannelReliability *rel;
1520   struct MeshClient *c;
1521
1522   c   = fwd ? ch->root     : ch->dest;
1523   rel = fwd ? ch->root_rel : ch->dest_rel;
1524
1525   if (GNUNET_YES == rel->client_ready)
1526     return; /* don't send double ACKs to client */
1527
1528   rel->client_ready = GNUNET_YES;
1529
1530   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1531               "send local %s ack on %s:%X towards %p\n",
1532               fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid, c);
1533
1534   if (NULL == c
1535       || ( fwd && (0 == ch->lid_root || c != ch->root))
1536       || (!fwd && (0 == ch->lid_dest || c != ch->dest)) )
1537   {
1538     GNUNET_break (0);
1539     return;
1540   }
1541   msg.header.size = htons (sizeof (msg));
1542   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK);
1543   msg.channel_id = htonl (fwd ? ch->lid_root : ch->lid_dest);
1544   GNUNET_SERVER_notification_context_unicast (nc,
1545                                               c->handle,
1546                                               &msg.header,
1547                                               GNUNET_NO);
1548
1549 }
1550
1551
1552 /**
1553  * Count established (ready) connections of a tunnel.
1554  *
1555  * @param t Tunnel on which to send the message.
1556  *
1557  * @return Number of connections.
1558  */
1559 static unsigned int
1560 tunnel_count_connections (struct MeshTunnel2 *t)
1561 {
1562   struct MeshConnection *c;
1563   unsigned int i;
1564
1565   for (c = t->connection_head, i = 0; NULL != c; c = c->next, i++);
1566
1567   return i;
1568 }
1569
1570
1571 /**
1572  * Pick a connection on which send the next data message.
1573  *
1574  * @param t Tunnel on which to send the message.
1575  * @param fwd Is this a fwd message?
1576  *
1577  * @return The connection on which to send the next message.
1578  */
1579 static struct MeshConnection *
1580 tunnel_get_connection (struct MeshTunnel2 *t, int fwd)
1581 {
1582   struct MeshConnection *c;
1583   struct MeshConnection *best;
1584   struct MeshFlowControl *fc;
1585   unsigned int lowest_q;
1586
1587   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "tunnel_get_connection %s\n",
1588               peer2s (t->peer));
1589   best = NULL;
1590   lowest_q = UINT_MAX;
1591   for (c = t->connection_head; NULL != c; c = c->next)
1592   {
1593     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  connection %s, \n",
1594                 GNUNET_h2s (&c->id));
1595     if (MESH_CONNECTION_READY == c->state)
1596     {
1597       fc = fwd ? &c->fwd_fc : &c->bck_fc;
1598       if (NULL == fc)
1599       {
1600         GNUNET_break (0);
1601         continue;
1602       }
1603       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    q_n %u, \n", fc->queue_n);
1604       if (fc->queue_n < lowest_q)
1605       {
1606         best = c;
1607         lowest_q = fc->queue_n;
1608       }
1609     }
1610   }
1611   return best;
1612 }
1613
1614
1615 /**
1616  * Is this peer the first one on the connection?
1617  *
1618  * @param c Connection.
1619  * @param fwd Is this about fwd traffic?
1620  *
1621  * @return GNUNET_YES if origin, GNUNET_NO if relay/terminal.
1622  */
1623 static int
1624 connection_is_origin (struct MeshConnection *c, int fwd)
1625 {
1626   if (!fwd && c->own_pos == c->path->length - 1)
1627     return GNUNET_YES;
1628   if (fwd && c->own_pos == 0)
1629     return GNUNET_YES;
1630   return GNUNET_NO;
1631 }
1632
1633
1634 /**
1635  * Is this peer the last one on the connection?
1636  *
1637  * @param c Connection.
1638  * @param fwd Is this about fwd traffic?
1639  *            Note that the ROOT is the terminal for BCK traffic!
1640  *
1641  * @return GNUNET_YES if terminal, GNUNET_NO if relay/origin.
1642  */
1643 static int
1644 connection_is_terminal (struct MeshConnection *c, int fwd)
1645 {
1646   if (fwd && c->own_pos == c->path->length - 1)
1647     return GNUNET_YES;
1648   if (!fwd && c->own_pos == 0)
1649     return GNUNET_YES;
1650   return GNUNET_NO;
1651 }
1652
1653
1654 /**
1655  * Is the recipient client for this channel on this peer?
1656  *
1657  * @param ch Channel.
1658  * @param fwd Is this for fwd traffic?
1659  *
1660  * @return GNUNET_YES in case it is.
1661  */
1662 static int
1663 channel_is_terminal (struct MeshChannel *ch, int fwd)
1664 {
1665   if (NULL == ch->t || NULL == ch->t->connection_head)
1666   {
1667     GNUNET_break (0);
1668     return GNUNET_NO;
1669   }
1670   return connection_is_terminal (ch->t->connection_head, fwd);
1671 }
1672
1673
1674 /**
1675  * Get free buffer space towards the client on a specific channel.
1676  *
1677  * @param ch Channel.
1678  * @param fwd Is query about FWD traffic?
1679  *
1680  * @return Free buffer space [0 - 64]
1681  */
1682 static unsigned int
1683 channel_get_buffer (struct MeshChannel *ch, int fwd)
1684 {
1685   struct MeshChannelReliability *rel;
1686   
1687   rel = fwd ? ch->dest_rel : ch->root_rel;
1688
1689   /* If rel is NULL it means that the end is not yet created,
1690    * most probably is a loopback channel at the point of sending
1691    * the ChannelCreate to itself.
1692    */
1693   if (NULL == rel)
1694     return 64;
1695
1696   return (64 - rel->n_recv);
1697 }
1698
1699
1700 /**
1701  * Get free buffer space in a connection.
1702  *
1703  * @param c Connection.
1704  * @param fwd Is query about FWD traffic?
1705  *
1706  * @return Free buffer space [0 - max_msgs_queue/max_connections]
1707  */
1708 static unsigned int
1709 connection_get_buffer (struct MeshConnection *c, int fwd)
1710 {
1711   struct MeshFlowControl *fc;
1712   
1713   fc = fwd ? &c->fwd_fc : &c->bck_fc;
1714   
1715   return (fc->queue_max - fc->queue_n);
1716 }
1717
1718
1719 /**
1720  * Get the total buffer space for a tunnel.
1721  */
1722 static unsigned int
1723 tunnel_get_buffer (struct MeshTunnel2 *t, int fwd)
1724 {
1725   struct MeshConnection *c;
1726   struct MeshFlowControl *fc;
1727   unsigned int buffer;
1728
1729   c = t->connection_head;
1730   buffer = 0;
1731
1732   if (NULL == c)
1733   {
1734     GNUNET_break (0);
1735     return 0;
1736   }
1737
1738   /* If terminal, return biggest channel buffer */
1739   if (connection_is_terminal (c, fwd))
1740   {
1741     struct MeshChannel *ch;
1742     unsigned int ch_buf;
1743
1744     if (NULL == t->channel_head)
1745       return 64;
1746
1747     for (ch = t->channel_head; NULL != ch; ch = ch->next)
1748     {
1749       ch_buf = channel_get_buffer (ch, fwd);
1750       if (ch_buf > buffer)
1751         buffer = ch_buf;
1752     }
1753     return buffer;
1754   }
1755
1756   /* If not terminal, return sum of connection buffers */
1757   while (NULL != c)
1758   {
1759     if (c->state != MESH_CONNECTION_READY)
1760       continue;
1761
1762     fc = fwd ? &c->fwd_fc : &c->bck_fc;
1763     buffer += fc->last_ack_recv - fc->last_pid_sent;
1764     c = c->next;
1765   }
1766
1767   return buffer;
1768 }
1769
1770
1771 /**
1772  * FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME 
1773  * Encrypt data with the tunnel key.
1774  *
1775  * @param t Tunnel whose key to use.
1776  * @param dst Destination for the encrypted data.
1777  * @param src Source of the plaintext.
1778  * @param size Size of the plaintext.
1779  * @param iv Initialization Vector to use.
1780  * @param fwd Is this a fwd message?
1781  */
1782 static void
1783 tunnel_encrypt (struct MeshTunnel2 *t,
1784                 void *dst, const void *src,
1785                 size_t size, uint64_t iv, int fwd)
1786 {
1787   memcpy (dst, src, size);
1788 }
1789
1790
1791 /**
1792  * FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME 
1793  * Decrypt data with the tunnel key.
1794  *
1795  * @param t Tunnel whose key to use.
1796  * @param dst Destination for the plaintext.
1797  * @param src Source of the encrypted data.
1798  * @param size Size of the encrypted data.
1799  * @param iv Initialization Vector to use.
1800  * @param fwd Is this a fwd message?
1801  */
1802 static void
1803 tunnel_decrypt (struct MeshTunnel2 *t,
1804                 void *dst, const void *src,
1805                 size_t size, uint64_t iv, int fwd)
1806 {
1807   memcpy (dst, src, size);
1808 }
1809
1810
1811 /**
1812  * Sends an already built message on a connection, properly registering
1813  * all used resources.
1814  *
1815  * @param message Message to send. Function makes a copy of it.
1816  *                If message is not hop-by-hop, decrements TTL of copy.
1817  * @param c Connection on which this message is transmitted.
1818  * @param ch Channel on which this message is transmitted, or NULL.
1819  * @param fwd Is this a fwd message?
1820  */
1821 static void
1822 send_prebuilt_message_connection (const struct GNUNET_MessageHeader *message,
1823                                   struct MeshConnection *c,
1824                                   struct MeshChannel *ch,
1825                                   int fwd)
1826 {
1827   void *data;
1828   size_t size;
1829   uint16_t type;
1830
1831   size = ntohs (message->size);
1832   data = GNUNET_malloc (size);
1833   memcpy (data, message, size);
1834   type = ntohs (message->type);
1835   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send %s (%u) on connection %s\n",
1836               GNUNET_MESH_DEBUG_M2S (type), size, GNUNET_h2s (&c->id));
1837
1838   switch (type)
1839   {
1840     struct GNUNET_MESH_Encrypted *emsg;
1841     struct GNUNET_MESH_ACK       *amsg;
1842     struct GNUNET_MESH_Poll      *pmsg;
1843     struct GNUNET_MESH_ConnectionDestroy *dmsg;
1844     struct GNUNET_MESH_ConnectionBroken  *bmsg;
1845     uint32_t ttl;
1846
1847     case GNUNET_MESSAGE_TYPE_MESH_FWD:
1848     case GNUNET_MESSAGE_TYPE_MESH_BCK:
1849       emsg = (struct GNUNET_MESH_Encrypted *) data;
1850       ttl = ntohl (emsg->ttl);
1851       if (0 == ttl)
1852       {
1853         GNUNET_break_op (0);
1854         return;
1855       }
1856       emsg->cid = c->id;
1857       emsg->ttl = htonl (ttl - 1);
1858       emsg->pid = htonl (fwd ? c->fwd_fc.next_pid++ : c->bck_fc.next_pid++);
1859       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " pid %u\n", ntohl (emsg->pid));
1860       break;
1861
1862     case GNUNET_MESSAGE_TYPE_MESH_ACK:
1863       amsg = (struct GNUNET_MESH_ACK *) data;
1864       amsg->cid = c->id;
1865       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ack %u\n", ntohl (amsg->ack));
1866       break;
1867
1868     case GNUNET_MESSAGE_TYPE_MESH_POLL:
1869       pmsg = (struct GNUNET_MESH_Poll *) data;
1870       pmsg->cid = c->id;
1871       pmsg->pid = htonl (fwd ? c->fwd_fc.last_pid_sent : c->bck_fc.last_pid_sent);
1872       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " poll %u\n", ntohl (pmsg->pid));
1873       break;
1874
1875     case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY:
1876       dmsg = (struct GNUNET_MESH_ConnectionDestroy *) data;
1877       dmsg->cid = c->id;
1878       dmsg->reserved = 0;
1879       break;
1880
1881     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
1882       bmsg = (struct GNUNET_MESH_ConnectionBroken *) data;
1883       bmsg->cid = c->id;
1884       bmsg->reserved = 0;
1885       break;
1886
1887     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
1888       break;
1889
1890     default:
1891       GNUNET_break (0);
1892   }
1893
1894   queue_add (data,
1895              type,
1896              size,
1897              c,
1898              ch,
1899              fwd);
1900 }
1901
1902
1903 /**
1904  * Sends an already built message on a tunnel, choosing the best connection.
1905  *
1906  * @param message Message to send. Function modifies it.
1907  * @param t Tunnel on which this message is transmitted.
1908  * @param ch Channel on which this message is transmitted.
1909  * @param fwd Is this a fwd message?
1910  */
1911 static void
1912 send_prebuilt_message_tunnel (struct GNUNET_MESH_Encrypted *msg,
1913                               struct MeshTunnel2 *t,
1914                               struct MeshChannel *ch,
1915                               int fwd)
1916 {
1917   struct MeshConnection *c;
1918   uint16_t type;
1919
1920   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send on Tunnel %s\n",
1921               peer2s (t->peer));
1922   c = tunnel_get_connection (t, fwd);
1923   if (NULL == c)
1924   {
1925     GNUNET_break (GNUNET_YES == t->destroy);
1926     return;
1927   }
1928   type = ntohs (msg->header.type);
1929   switch (type)
1930   {
1931     case GNUNET_MESSAGE_TYPE_MESH_FWD:
1932     case GNUNET_MESSAGE_TYPE_MESH_BCK:
1933     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
1934     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
1935       msg->cid = c->id;
1936       msg->ttl = htonl (default_ttl);
1937       break;
1938     default:
1939       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "unkown type %s\n",
1940                   GNUNET_MESH_DEBUG_M2S (type));
1941       GNUNET_break (0);
1942   }
1943   msg->reserved = 0;
1944
1945   send_prebuilt_message_connection (&msg->header, c, ch, fwd);
1946 }
1947
1948
1949 /**
1950  * Sends an already built message on a channel, properly registering
1951  * all used resources and encrypting the message with the tunnel's key.
1952  *
1953  * @param message Message to send. Function makes a copy of it.
1954  * @param ch Channel on which this message is transmitted.
1955  * @param fwd Is this a fwd message?
1956  */
1957 static void
1958 send_prebuilt_message_channel (const struct GNUNET_MessageHeader *message,
1959                                struct MeshChannel *ch,
1960                                int fwd)
1961 {
1962   struct GNUNET_MESH_Encrypted *msg;
1963   size_t size = ntohs (message->size);
1964   char *cbuf[sizeof (struct GNUNET_MESH_Encrypted) + size];
1965   uint16_t type;
1966   uint64_t iv;
1967
1968   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send on Channel %s:%X\n",
1969               peer2s (ch->t->peer), ch->gid);
1970   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  %s\n",
1971               GNUNET_MESH_DEBUG_M2S (ntohs (message->type)));
1972
1973   if (channel_is_terminal (ch, fwd))
1974   {
1975     handle_decrypted (ch->t, message, fwd);
1976     return;
1977   }
1978   
1979   type = fwd ? GNUNET_MESSAGE_TYPE_MESH_FWD : GNUNET_MESSAGE_TYPE_MESH_BCK;
1980   iv = GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_NONCE, UINT64_MAX);
1981
1982   msg = (struct GNUNET_MESH_Encrypted *) cbuf;
1983   msg->header.type = htons (type);
1984   msg->header.size = htons (sizeof (struct GNUNET_MESH_Encrypted) + size);
1985   msg->iv = GNUNET_htonll (iv);
1986   tunnel_encrypt (ch->t, &msg[1], message, size, iv, fwd);
1987   send_prebuilt_message_tunnel (msg, ch->t, ch, fwd);
1988 }
1989
1990
1991 /**
1992  * Sends a CREATE CONNECTION message for a path to a peer.
1993  * Changes the connection and tunnel states if necessary.
1994  *
1995  * @param connection Connection to create.
1996  */
1997 static void
1998 send_connection_create (struct MeshConnection *connection)
1999 {
2000   struct MeshTunnel2 *t;
2001
2002   t = connection->t;
2003   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send connection create\n");
2004   queue_add (NULL,
2005              GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
2006              sizeof (struct GNUNET_MESH_ConnectionCreate) +
2007                 (connection->path->length *
2008                  sizeof (struct GNUNET_PeerIdentity)),
2009              connection,
2010              NULL,
2011              GNUNET_YES);
2012   if (NULL != t &&
2013       (MESH_TUNNEL_SEARCHING == t->state || MESH_TUNNEL_NEW == t->state))
2014     tunnel_change_state (t, MESH_TUNNEL_WAITING);
2015   if (MESH_CONNECTION_NEW == connection->state)
2016     connection_change_state (connection, MESH_CONNECTION_SENT);
2017 }
2018
2019
2020 /**
2021  * Sends a CONNECTION ACK message in reponse to a received CONNECTION_CREATE
2022  * directed to us.
2023  *
2024  * @param connection Connection to confirm.
2025  * @param fwd Is this a fwd ACK? (First is bck (SYNACK), second is fwd (ACK))
2026  */
2027 static void
2028 send_connection_ack (struct MeshConnection *connection, int fwd) 
2029 {
2030   struct MeshTunnel2 *t;
2031
2032   t = connection->t;
2033   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send connection ack\n");
2034   queue_add (NULL,
2035              GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
2036              sizeof (struct GNUNET_MESH_ConnectionACK),
2037              connection,
2038              NULL,
2039              fwd);
2040   if (MESH_TUNNEL_NEW == t->state)
2041     tunnel_change_state (t, MESH_TUNNEL_WAITING);
2042 }
2043
2044
2045 /**
2046   * Core callback to write a pre-constructed data packet to core buffer
2047   *
2048   * @param cls Closure (MeshTransmissionDescriptor with data in "data" member).
2049   * @param size Number of bytes available in buf.
2050   * @param buf Where the to write the message.
2051   *
2052   * @return number of bytes written to buf
2053   */
2054 static size_t
2055 send_core_data_raw (void *cls, size_t size, void *buf)
2056 {
2057   struct GNUNET_MessageHeader *msg = cls;
2058   size_t total_size;
2059
2060   GNUNET_assert (NULL != msg);
2061   total_size = ntohs (msg->size);
2062
2063   if (total_size > size)
2064   {
2065     GNUNET_break (0);
2066     return 0;
2067   }
2068   memcpy (buf, msg, total_size);
2069   GNUNET_free (cls);
2070   return total_size;
2071 }
2072
2073
2074 /**
2075  * Function to send a create connection message to a peer.
2076  *
2077  * @param c Connection to create.
2078  * @param size number of bytes available in buf
2079  * @param buf where the callee should write the message
2080  * @return number of bytes written to buf
2081  */
2082 static size_t
2083 send_core_connection_create (struct MeshConnection *c, size_t size, void *buf)
2084 {
2085   struct GNUNET_MESH_ConnectionCreate *msg;
2086   struct GNUNET_PeerIdentity *peer_ptr;
2087   struct MeshPeerPath *p = c->path;
2088   size_t size_needed;
2089   int i;
2090
2091   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION CREATE...\n");
2092   size_needed =
2093       sizeof (struct GNUNET_MESH_ConnectionCreate) +
2094       p->length * sizeof (struct GNUNET_PeerIdentity);
2095
2096   if (size < size_needed || NULL == buf)
2097   {
2098     GNUNET_break (0);
2099     return 0;
2100   }
2101   msg = (struct GNUNET_MESH_ConnectionCreate *) buf;
2102   msg->header.size = htons (size_needed);
2103   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE);
2104   msg->cid = c->id;
2105
2106   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
2107   for (i = 0; i < p->length; i++)
2108   {
2109     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
2110   }
2111
2112   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2113               "CONNECTION CREATE (%u bytes long) sent!\n", size_needed);
2114   return size_needed;
2115 }
2116
2117
2118 /**
2119  * Creates a path ack message in buf and frees all unused resources.
2120  *
2121  * @param c Connection to send an ACK on.
2122  * @param size number of bytes available in buf
2123  * @param buf where the callee should write the message
2124  *
2125  * @return number of bytes written to buf
2126  */
2127 static size_t
2128 send_core_connection_ack (struct MeshConnection *c, size_t size, void *buf)
2129 {
2130   struct GNUNET_MESH_ConnectionACK *msg = buf;
2131   struct MeshTunnel2 *t = c->t;
2132
2133   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION ACK...\n");
2134   GNUNET_assert (NULL != t);
2135   if (sizeof (struct GNUNET_MESH_ConnectionACK) > size)
2136   {
2137     GNUNET_break (0);
2138     return 0;
2139   }
2140   msg->header.size = htons (sizeof (struct GNUNET_MESH_ConnectionACK));
2141   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK);
2142   msg->cid = c->id;
2143
2144   /* TODO add signature */
2145
2146   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CONNECTION ACK sent!\n");
2147   return sizeof (struct GNUNET_MESH_ConnectionACK);
2148 }
2149
2150
2151 /**
2152  * Destroy the peer_info and free any allocated resources linked to it
2153  *
2154  * @param peer The peer_info to destroy.
2155  *
2156  * @return GNUNET_OK on success
2157  */
2158 static int
2159 peer_destroy (struct MeshPeer *peer)
2160 {
2161   struct GNUNET_PeerIdentity id;
2162   struct MeshPeerPath *p;
2163   struct MeshPeerPath *nextp;
2164
2165   GNUNET_PEER_resolve (peer->id, &id);
2166   GNUNET_PEER_change_rc (peer->id, -1);
2167
2168   if (GNUNET_YES !=
2169       GNUNET_CONTAINER_multihashmap_remove (peers, &id.hashPubKey, peer))
2170   {
2171     GNUNET_break (0);
2172     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
2173                 "removing peer %s, not in hashmap\n", GNUNET_i2s (&id));
2174   }
2175   if (NULL != peer->dhtget)
2176   {
2177     GNUNET_DHT_get_stop (peer->dhtget);
2178   }
2179   p = peer->path_head;
2180   while (NULL != p)
2181   {
2182     nextp = p->next;
2183     GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
2184     path_destroy (p);
2185     p = nextp;
2186   }
2187   tunnel_destroy_empty (peer->tunnel);
2188   GNUNET_free (peer);
2189   return GNUNET_OK;
2190 }
2191
2192
2193 /**
2194  * Returns if peer is used (has a tunnel, is neighbor).
2195  *
2196  * @peer Peer to check.
2197  *
2198  * @return GNUNET_YES if peer is in use.
2199  */
2200 static int
2201 peer_is_used (struct MeshPeer *peer)
2202 {
2203   struct MeshPeerPath *p;
2204
2205   if (NULL != peer->tunnel)
2206     return GNUNET_YES;
2207
2208   for (p = peer->path_head; NULL != p; p = p->next)
2209   {
2210     if (p->length < 3)
2211       return GNUNET_YES;
2212   }
2213   return GNUNET_NO;
2214 }
2215
2216 /**
2217  * Iterator over all the peers to get the oldest timestamp.
2218  *
2219  * @param cls Closure (unsued).
2220  * @param key ID of the peer.
2221  * @param value Peer_Info of the peer.
2222  */
2223 static int
2224 peer_get_oldest (void *cls,
2225                  const struct GNUNET_HashCode *key,
2226                  void *value)
2227 {
2228   struct MeshPeer *p = value;
2229   struct GNUNET_TIME_Absolute *abs = cls;
2230
2231   /* Don't count active peers */
2232   if (GNUNET_YES == peer_is_used (p))
2233     return GNUNET_YES;
2234
2235   if (abs->abs_value_us < p->last_contact.abs_value_us)
2236     abs->abs_value_us = p->last_contact.abs_value_us;
2237
2238   return GNUNET_YES;
2239 }
2240
2241
2242 /**
2243  * Iterator over all the peers to remove the oldest entry.
2244  *
2245  * @param cls Closure (unsued).
2246  * @param key ID of the peer.
2247  * @param value Peer_Info of the peer.
2248  */
2249 static int
2250 peer_timeout (void *cls,
2251               const struct GNUNET_HashCode *key,
2252               void *value)
2253 {
2254   struct MeshPeer *p = value;
2255   struct GNUNET_TIME_Absolute *abs = cls;
2256
2257   if (p->last_contact.abs_value_us == abs->abs_value_us &&
2258       GNUNET_NO == peer_is_used (p))
2259   {
2260     peer_destroy (p);
2261     return GNUNET_NO;
2262   }
2263   return GNUNET_YES;
2264 }
2265
2266
2267 /**
2268  * Delete oldest unused peer.
2269  */
2270 static void
2271 peer_delete_oldest (void)
2272 {
2273   struct GNUNET_TIME_Absolute abs;
2274
2275   abs = GNUNET_TIME_UNIT_FOREVER_ABS;
2276
2277   GNUNET_CONTAINER_multihashmap_iterate (peers,
2278                                          &peer_get_oldest,
2279                                          &abs);
2280   GNUNET_CONTAINER_multihashmap_iterate (peers,
2281                                          &peer_timeout,
2282                                          &abs);
2283 }
2284
2285
2286 /**
2287  * Retrieve the MeshPeer stucture associated with the peer, create one
2288  * and insert it in the appropriate structures if the peer is not known yet.
2289  *
2290  * @param peer Full identity of the peer.
2291  *
2292  * @return Existing or newly created peer info.
2293  */
2294 static struct MeshPeer *
2295 peer_get (const struct GNUNET_PeerIdentity *peer_id)
2296 {
2297   struct MeshPeer *peer;
2298
2299   peer = GNUNET_CONTAINER_multihashmap_get (peers, &peer_id->hashPubKey);
2300   if (NULL == peer)
2301   {
2302     peer = GNUNET_new (struct MeshPeer);
2303     if (GNUNET_CONTAINER_multihashmap_size (peers) > max_peers)
2304     {
2305       peer_delete_oldest ();
2306     }
2307     GNUNET_CONTAINER_multihashmap_put (peers, &peer_id->hashPubKey, peer,
2308                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
2309     peer->id = GNUNET_PEER_intern (peer_id);
2310   }
2311   peer->last_contact = GNUNET_TIME_absolute_get();
2312
2313   return peer;
2314 }
2315
2316
2317 /**
2318  * Retrieve the MeshPeer stucture associated with the peer, create one
2319  * and insert it in the appropriate structures if the peer is not known yet.
2320  *
2321  * @param peer Short identity of the peer.
2322  *
2323  * @return Existing or newly created peer info.
2324  */
2325 static struct MeshPeer *
2326 peer_get_short (const GNUNET_PEER_Id peer)
2327 {
2328   return peer_get (GNUNET_PEER_resolve2 (peer));
2329 }
2330
2331
2332 /**
2333  * Get a cost of a path for a peer considering existing tunnel connections.
2334  *
2335  * @param peer Peer towards which the path is considered.
2336  * @param path Candidate path.
2337  *
2338  * @return Cost of the path (path length + number of overlapping nodes)
2339  */
2340 static unsigned int
2341 peer_get_path_cost (const struct MeshPeer *peer,
2342                     const struct MeshPeerPath *path)
2343 {
2344   struct MeshConnection *c;
2345   unsigned int overlap;
2346   unsigned int i;
2347   unsigned int j;
2348
2349   if (NULL == path)
2350     return 0;
2351
2352   overlap = 0;
2353   GNUNET_assert (NULL != peer->tunnel);
2354
2355   for (i = 0; i < path->length; i++)
2356   {
2357     for (c = peer->tunnel->connection_head; NULL != c; c = c->next)
2358     {
2359       for (j = 0; j < c->path->length; j++)
2360       {
2361         if (path->peers[i] == c->path->peers[j])
2362         {
2363           overlap++;
2364           break;
2365         }
2366       }
2367     }
2368   }
2369   return (path->length + overlap) * (path->score * -1);
2370 }
2371
2372
2373 /**
2374  * Choose the best path towards a peer considering the tunnel properties.
2375  *
2376  * @param peer The destination peer.
2377  *
2378  * @return Best current known path towards the peer, if any.
2379  */
2380 static struct MeshPeerPath *
2381 peer_get_best_path (const struct MeshPeer *peer)
2382 {
2383   struct MeshPeerPath *best_p;
2384   struct MeshPeerPath *p;
2385   struct MeshConnection *c;
2386   unsigned int best_cost;
2387   unsigned int cost;
2388
2389   best_cost = UINT_MAX;
2390   best_p = NULL;
2391   for (p = peer->path_head; NULL != p; p = p->next)
2392   {
2393     for (c = peer->tunnel->connection_head; NULL != c; c = c->next)
2394       if (c->path == p)
2395         break;
2396     if (NULL != c)
2397       continue; /* If path is in use in a connection, skip it. */
2398
2399     if ((cost = peer_get_path_cost (peer, p)) < best_cost)
2400     {
2401       best_cost = cost;
2402       best_p = p;
2403     }
2404   }
2405   return best_p;
2406 }
2407
2408 static int
2409 queue_is_sendable (struct MeshPeerQueue *q)
2410 {
2411   struct MeshFlowControl *fc;
2412
2413   /* Is PID-independent? */
2414   switch (q->type)
2415   {
2416     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2417     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2418       return GNUNET_YES;
2419   }
2420
2421   /* Is PID allowed? */
2422   fc = q->fwd ? &q->c->fwd_fc : &q->c->bck_fc;
2423   if (GMC_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2424     return GNUNET_YES;
2425
2426   return GNUNET_NO;
2427 }
2428
2429
2430 /**
2431  * Get first sendable message.
2432  *
2433  * @param peer The destination peer.
2434  *
2435  * @return Best current known path towards the peer, if any.
2436  */
2437 static struct MeshPeerQueue *
2438 peer_get_first_message (const struct MeshPeer *peer)
2439 {
2440   struct MeshPeerQueue *q;
2441
2442   for (q = peer->queue_head; NULL != q; q = q->next)
2443   {
2444     if (queue_is_sendable (q))
2445       return q;
2446   }
2447
2448   return NULL;
2449 }
2450
2451
2452 /**
2453  * Try to establish a new connection to this peer in the given tunnel.
2454  * If the peer doesn't have any path to it yet, try to get one.
2455  * If the peer already has some path, send a CREATE CONNECTION towards it.
2456  *
2457  * @param peer PeerInfo of the peer.
2458  */
2459 static void
2460 peer_connect (struct MeshPeer *peer)
2461 {
2462   struct MeshTunnel2 *t;
2463   struct MeshPeerPath *p;
2464   struct MeshConnection *c;
2465
2466   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2467               "peer_connect towards %s\n",
2468               peer2s (peer));
2469   t = peer->tunnel;
2470   if (NULL != peer->path_head)
2471   {
2472     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "path exists\n");
2473     p = peer_get_best_path (peer);
2474     if (NULL != p)
2475     {
2476       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  %u hops\n", p->length);
2477       c = tunnel_use_path (t, p);
2478       send_connection_create (c);
2479     }
2480   }
2481   else if (NULL == peer->dhtget)
2482   {
2483     const struct GNUNET_PeerIdentity *id;
2484
2485     id = GNUNET_PEER_resolve2 (peer->id);
2486     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2487                 "  Starting DHT GET for peer %s\n", peer2s (peer));
2488     peer->dhtget = GNUNET_DHT_get_start (dht_handle,    /* handle */
2489                                          GNUNET_BLOCK_TYPE_MESH_PEER, /* type */
2490                                          &id->hashPubKey,     /* key to search */
2491                                          dht_replication_level, /* replication level */
2492                                          GNUNET_DHT_RO_RECORD_ROUTE |
2493                                          GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
2494                                          NULL,       /* xquery */
2495                                          0,     /* xquery bits */
2496                                          &dht_get_id_handler, peer);
2497     if (MESH_TUNNEL_NEW == t->state)
2498       tunnel_change_state (t, MESH_TUNNEL_SEARCHING);
2499   }
2500   else
2501   {
2502     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2503                 "There is no path but the DHT GET is already started.\n");
2504   }
2505 }
2506
2507
2508 /**
2509  * Get the first transmittable message for a connection.
2510  *
2511  * @param c Connection.
2512  * @param fwd Is this FWD?
2513  *
2514  * @return First transmittable message.
2515  */
2516 static struct MeshPeerQueue *
2517 connection_get_first_message (struct MeshConnection *c, int fwd)
2518 {
2519   struct MeshPeerQueue *q;
2520   struct MeshPeer *p;
2521
2522   p = connection_get_hop (c, fwd);
2523
2524   for (q = p->queue_head; NULL != q; q = q->next)
2525   {
2526     if (q->c != c)
2527       continue;
2528     if (queue_is_sendable (q))
2529       return q;
2530   }
2531
2532   return NULL;
2533 }
2534
2535
2536 /**
2537  * @brief Re-initiate traffic on this connection if necessary.
2538  *
2539  * Check if there is traffic queued towards this peer
2540  * and the core transmit handle is NULL (traffic was stalled).
2541  * If so, call core tmt rdy.
2542  *
2543  * @param c Connection on which initiate traffic.
2544  * @param fwd Is this about fwd traffic?
2545  */
2546 static void
2547 connection_unlock_queue (struct MeshConnection *c, int fwd)
2548 {
2549   struct MeshPeer *peer;
2550   struct MeshPeerQueue *q;
2551   size_t size;
2552
2553   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2554               "connection_unlock_queue %s on %s\n",
2555               fwd ? "FWD" : "BCK", GNUNET_h2s (&c->id));
2556
2557   if (connection_is_origin (c, fwd))
2558   {
2559     struct MeshTunnel2 *t = c->t;
2560     struct MeshChannel *ch;
2561     struct MeshChannelReliability *rel;
2562
2563     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " is origin!\n");
2564     /* FIXME randomize channel selection, not always first channel */
2565     for (ch = t->channel_head; NULL != ch; ch = ch->next)
2566     {
2567       rel = fwd ? ch->root_rel : ch->dest_rel;
2568
2569       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  channel %X - %s\n",
2570                   ch->gid, rel->client_ready ? "ready " : "not ready");
2571       if (GNUNET_NO == rel->client_ready)
2572       {
2573         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    sending local ack!\n");
2574         send_local_ack (ch, fwd);
2575         return; /* FIXME authorize all channels? */
2576       }
2577     }
2578     return;
2579   }
2580
2581   peer = connection_get_hop (c, fwd);
2582
2583   if (NULL != peer->core_transmit)
2584   {
2585     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  already unlocked!\n");
2586     return; /* Already unlocked */
2587   }
2588
2589   q = connection_get_first_message (c, fwd);
2590   if (NULL == q)
2591   {
2592     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  queue empty!\n");
2593     return; /* Nothing to transmit */
2594   }
2595
2596   size = q->size;
2597   peer->core_transmit =
2598       GNUNET_CORE_notify_transmit_ready (core_handle,
2599                                          GNUNET_NO,
2600                                          0,
2601                                          GNUNET_TIME_UNIT_FOREVER_REL,
2602                                          GNUNET_PEER_resolve2 (peer->id),
2603                                          size,
2604                                          &queue_send,
2605                                          peer);
2606 }
2607
2608
2609 /**
2610  * Cancel all transmissions that belong to a certain connection.
2611  *
2612  * @param c Connection which to cancel.
2613  * @param fwd Cancel fwd traffic?
2614  */
2615 static void
2616 connection_cancel_queues (struct MeshConnection *c, int fwd)
2617 {
2618   struct MeshPeerQueue *q;
2619   struct MeshPeerQueue *next;
2620   struct MeshFlowControl *fc;
2621   struct MeshPeer *peer;
2622
2623   if (NULL == c)
2624   {
2625     GNUNET_break (0);
2626     return;
2627   }
2628   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2629   peer = connection_get_hop (c, fwd);
2630
2631   for (q = peer->queue_head; NULL != q; q = next)
2632   {
2633     next = q->next;
2634     if (q->c == c)
2635     {
2636       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2637                   "connection_cancel_queue %s\n",
2638                   GNUNET_MESH_DEBUG_M2S (q->type));
2639       queue_destroy (q, GNUNET_YES);
2640     }
2641   }
2642   if (NULL == peer->queue_head)
2643   {
2644     if (NULL != peer->core_transmit)
2645     {
2646       GNUNET_CORE_notify_transmit_ready_cancel (peer->core_transmit);
2647       peer->core_transmit = NULL;
2648     }
2649     if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
2650     {
2651       GNUNET_SCHEDULER_cancel (fc->poll_task);
2652       fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2653     }
2654   }
2655 }
2656
2657
2658 /**
2659  * Remove all paths that rely on a direct connection between p1 and p2
2660  * from the peer itself and notify all tunnels about it.
2661  *
2662  * @param peer PeerInfo of affected peer.
2663  * @param p1 GNUNET_PEER_Id of one peer.
2664  * @param p2 GNUNET_PEER_Id of another peer that was connected to the first and
2665  *           no longer is.
2666  *
2667  * TODO: optimize (see below)
2668  */
2669 static void
2670 peer_remove_path (struct MeshPeer *peer, GNUNET_PEER_Id p1,
2671                   GNUNET_PEER_Id p2)
2672 {
2673   struct MeshPeerPath *p;
2674   struct MeshPeerPath *next;
2675   struct MeshPeer *peer_d;
2676   GNUNET_PEER_Id d;
2677   unsigned int destroyed;
2678   unsigned int i;
2679
2680   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer_info_remove_path\n");
2681   destroyed = 0;
2682   for (p = peer->path_head; NULL != p; p = next)
2683   {
2684     next = p->next;
2685     for (i = 0; i < (p->length - 1); i++)
2686     {
2687       if ((p->peers[i] == p1 && p->peers[i + 1] == p2) ||
2688           (p->peers[i] == p2 && p->peers[i + 1] == p1))
2689       {
2690         GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
2691         path_destroy (p);
2692         destroyed++;
2693         break;
2694       }
2695     }
2696   }
2697   if (0 == destroyed)
2698     return;
2699
2700
2701   d = tunnel_notify_connection_broken (peer->tunnel, p1, p2);
2702
2703   peer_d = peer_get_short (d); // FIXME
2704   next = peer_get_best_path (peer_d);
2705   tunnel_use_path (peer->tunnel, next);
2706   peer_connect (peer_d);
2707
2708   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer_info_remove_path END\n");
2709 }
2710
2711
2712 /**
2713  * Add the path to the peer and update the path used to reach it in case this
2714  * is the shortest.
2715  *
2716  * @param peer_info Destination peer to add the path to.
2717  * @param path New path to add. Last peer must be the peer in arg 1.
2718  *             Path will be either used of freed if already known.
2719  * @param trusted Do we trust that this path is real?
2720  */
2721 void
2722 peer_add_path (struct MeshPeer *peer_info, struct MeshPeerPath *path,
2723                     int trusted)
2724 {
2725   struct MeshPeerPath *aux;
2726   unsigned int l;
2727   unsigned int l2;
2728
2729   if ((NULL == peer_info) || (NULL == path))
2730   {
2731     GNUNET_break (0);
2732     path_destroy (path);
2733     return;
2734   }
2735   if (path->peers[path->length - 1] != peer_info->id)
2736   {
2737     GNUNET_break (0);
2738     path_destroy (path);
2739     return;
2740   }
2741   if (2 >= path->length && GNUNET_NO == trusted)
2742   {
2743     /* Only allow CORE to tell us about direct paths */
2744     path_destroy (path);
2745     return;
2746   }
2747   for (l = 1; l < path->length; l++)
2748   {
2749     if (path->peers[l] == myid)
2750     {
2751       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shortening path by %u\n", l);
2752       for (l2 = 0; l2 < path->length - l; l2++)
2753       {
2754         path->peers[l2] = path->peers[l + l2];
2755       }
2756       path->length -= l;
2757       l = 1;
2758       path->peers =
2759           GNUNET_realloc (path->peers, path->length * sizeof (GNUNET_PEER_Id));
2760     }
2761   }
2762
2763   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n",
2764               path->length, peer2s (peer_info));
2765
2766   l = path_get_length (path);
2767   if (0 == l)
2768   {
2769     path_destroy (path);
2770     return;
2771   }
2772
2773   GNUNET_assert (peer_info->id == path->peers[path->length - 1]);
2774   for (aux = peer_info->path_head; aux != NULL; aux = aux->next)
2775   {
2776     l2 = path_get_length (aux);
2777     if (l2 > l)
2778     {
2779       GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head,
2780                                           peer_info->path_tail, aux, path);
2781       return;
2782     }
2783     else
2784     {
2785       if (l2 == l && memcmp (path->peers, aux->peers, l) == 0)
2786       {
2787         path_destroy (path);
2788         return;
2789       }
2790     }
2791   }
2792   GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head, peer_info->path_tail,
2793                                     path);
2794   return;
2795 }
2796
2797
2798 /**
2799  * Add the path to the origin peer and update the path used to reach it in case
2800  * this is the shortest.
2801  * The path is given in peer_info -> destination, therefore we turn the path
2802  * upside down first.
2803  *
2804  * @param peer_info Peer to add the path to, being the origin of the path.
2805  * @param path New path to add after being inversed.
2806  *             Path will be either used or freed.
2807  * @param trusted Do we trust that this path is real?
2808  */
2809 static void
2810 peer_add_path_to_origin (struct MeshPeer *peer_info,
2811                          struct MeshPeerPath *path, int trusted)
2812 {
2813   if (NULL == path)
2814     return;
2815   path_invert (path);
2816   peer_add_path (peer_info, path, trusted);
2817 }
2818
2819
2820
2821 /**
2822  * Function called if a connection has been stalled for a while,
2823  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
2824  *
2825  * @param cls Closure (poll ctx).
2826  * @param tc TaskContext.
2827  */
2828 static void
2829 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2830 {
2831   struct MeshFlowControl *fc = cls;
2832   struct GNUNET_MESH_Poll msg;
2833   struct MeshConnection *c;
2834
2835   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2836   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2837   {
2838     return;
2839   }
2840
2841   c = fc->c;
2842   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** Polling!\n");
2843   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** connection %s[%X]\n", 
2844               peer2s (c->t->peer), c->id);
2845   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ***   %s\n", 
2846               fc == &c->fwd_fc ? "FWD" : "BCK");
2847
2848   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
2849   msg.header.size = htons (sizeof (msg));
2850   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** pid (%u)!\n", fc->last_pid_sent);
2851   send_prebuilt_message_connection (&msg.header, c, NULL, fc == &c->fwd_fc);
2852   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
2853   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
2854                                                 &connection_poll, fc);
2855 }
2856
2857
2858 /**
2859  * Build a PeerPath from the paths returned from the DHT, reversing the paths
2860  * to obtain a local peer -> destination path and interning the peer ids.
2861  *
2862  * @return Newly allocated and created path
2863  */
2864 static struct MeshPeerPath *
2865 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
2866                      unsigned int get_path_length,
2867                      const struct GNUNET_PeerIdentity *put_path,
2868                      unsigned int put_path_length)
2869 {
2870   struct MeshPeerPath *p;
2871   GNUNET_PEER_Id id;
2872   int i;
2873
2874   p = path_new (1);
2875   p->peers[0] = myid;
2876   GNUNET_PEER_change_rc (myid, 1);
2877   i = get_path_length;
2878   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   GET has %d hops.\n", i);
2879   for (i--; i >= 0; i--)
2880   {
2881     id = GNUNET_PEER_intern (&get_path[i]);
2882     if (p->length > 0 && id == p->peers[p->length - 1])
2883     {
2884       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
2885       GNUNET_PEER_change_rc (id, -1);
2886     }
2887     else
2888     {
2889       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from GET: %s.\n",
2890                   GNUNET_i2s (&get_path[i]));
2891       p->length++;
2892       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
2893       p->peers[p->length - 1] = id;
2894     }
2895   }
2896   i = put_path_length;
2897   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   PUT has %d hops.\n", i);
2898   for (i--; i >= 0; i--)
2899   {
2900     id = GNUNET_PEER_intern (&put_path[i]);
2901     if (id == myid)
2902     {
2903       /* PUT path went through us, so discard the path up until now and start
2904        * from here to get a much shorter (and loop-free) path.
2905        */
2906       path_destroy (p);
2907       p = path_new (0);
2908     }
2909     if (p->length > 0 && id == p->peers[p->length - 1])
2910     {
2911       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
2912       GNUNET_PEER_change_rc (id, -1);
2913     }
2914     else
2915     {
2916       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from PUT: %s.\n",
2917                   GNUNET_i2s (&put_path[i]));
2918       p->length++;
2919       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
2920       p->peers[p->length - 1] = id;
2921     }
2922   }
2923 #if MESH_DEBUG
2924   if (get_path_length > 0)
2925     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of GET: %s)\n",
2926                 GNUNET_i2s (&get_path[0]));
2927   if (put_path_length > 0)
2928     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of PUT: %s)\n",
2929                 GNUNET_i2s (&put_path[0]));
2930   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   In total: %d hops\n",
2931               p->length);
2932   for (i = 0; i < p->length; i++)
2933   {
2934     struct GNUNET_PeerIdentity peer_id;
2935
2936     GNUNET_PEER_resolve (p->peers[i], &peer_id);
2937     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "       %u: %s\n", p->peers[i],
2938                 GNUNET_i2s (&peer_id));
2939   }
2940 #endif
2941   return p;
2942 }
2943
2944
2945 /**
2946  * Adds a path to the peer_infos of all the peers in the path
2947  *
2948  * @param p Path to process.
2949  * @param confirmed Whether we know if the path works or not.
2950  */
2951 static void
2952 path_add_to_peers (struct MeshPeerPath *p, int confirmed)
2953 {
2954   unsigned int i;
2955
2956   /* TODO: invert and add */
2957   for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ;
2958   for (i++; i < p->length; i++)
2959   {
2960     struct MeshPeer *aux;
2961     struct MeshPeerPath *copy;
2962
2963     aux = peer_get_short (p->peers[i]);
2964     copy = path_duplicate (p);
2965     copy->length = i + 1;
2966     peer_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed);
2967   }
2968 }
2969
2970
2971 /**
2972  * Search for a channel among the channels for a client
2973  *
2974  * @param c the client whose channels to search in
2975  * @param chid the local id of the channel
2976  *
2977  * @return channel handler, NULL if doesn't exist
2978  */
2979 static struct MeshChannel *
2980 channel_get_by_local_id (struct MeshClient *c, MESH_ChannelNumber chid)
2981 {
2982   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   -- get CHID %X\n", chid);
2983   if (0 == (chid & GNUNET_MESH_LOCAL_CHANNEL_ID_CLI))
2984   {
2985     GNUNET_break_op (0);
2986     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CHID %X not a local chid\n", chid);
2987     return NULL;
2988   }
2989   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
2990     return GNUNET_CONTAINER_multihashmap32_get (c->incoming_channels, chid);
2991   return GNUNET_CONTAINER_multihashmap32_get (c->own_channels, chid);
2992 }
2993
2994 #if 0
2995
2996 static void
2997 channel_debug (struct MeshChannel *ch)
2998 {
2999   if (NULL == ch)
3000   {
3001     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CHANNEL ***\n");
3002     return;
3003   }
3004   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Channel %s:%X\n",
3005               peer2s (ch->t->peer), ch->gid);
3006   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  root %p/%p\n",
3007               ch->root, ch->root_rel);
3008   if (NULL != ch->root)
3009   {
3010     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  cli %u\n", ch->root->id);
3011     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ready %s\n",
3012                 ch->root_rel->client_ready ? "YES" : "NO");
3013     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  id %X\n", ch->lid_root);
3014   }
3015   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  root %p/%p\n",
3016               ch->dest, ch->dest_rel);
3017   if (NULL != ch->dest)
3018   {
3019     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  cli %u\n", ch->dest->id);
3020     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ready %s\n",
3021                 ch->dest_rel->client_ready ? "YES" : "NO");
3022     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  id %X\n", ch->lid_dest);
3023   }
3024 }
3025
3026 static void
3027 fc_debug (struct MeshFlowControl *fc)
3028 {
3029   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    IN: %u/%u\n",
3030               fc->last_pid_recv, fc->last_ack_sent);
3031   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    OUT: %u/%u\n",
3032               fc->last_pid_sent, fc->last_ack_recv);
3033   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    QUEUE: %u/%u\n",
3034               fc->queue_n, fc->queue_max);
3035 }
3036
3037 static void
3038 connection_debug (struct MeshConnection *c)
3039 {
3040   if (NULL == c)
3041   {
3042     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CONNECTION ***\n");
3043     return;
3044   }
3045   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connection %s:%X\n",
3046               peer2s (c->t->peer), GNUNET_h2s (&c->id));
3047   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  state: %u, pending msgs: %u\n", 
3048               c->state, c->pending_messages);
3049   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
3050   fc_debug (&c->fwd_fc);
3051   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
3052   fc_debug (&c->bck_fc);
3053 }
3054
3055 #endif
3056
3057 static struct MeshChannel *
3058 channel_get (struct MeshTunnel2 *t, MESH_ChannelNumber chid)
3059 {
3060   struct MeshChannel *ch;
3061
3062   if (NULL == t)
3063     return NULL;
3064
3065   for (ch = t->channel_head; NULL != ch; ch = ch->next)
3066   {
3067     if (ch->gid == chid)
3068       break;
3069   }
3070
3071   return ch;
3072 }
3073
3074
3075 /**
3076  * Change the tunnel state.
3077  *
3078  * @param t Tunnel whose state to change.
3079  * @param state New state.
3080  */
3081 static void
3082 tunnel_change_state (struct MeshTunnel2* t, enum MeshTunnelState state)
3083 {
3084   if (NULL == t)
3085     return;
3086   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3087               "Tunnel %s state was %s\n",
3088               peer2s (t->peer),
3089               GNUNET_MESH_DEBUG_TS2S (t->state));
3090   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3091               "Tunnel %s state is now %s\n",
3092               peer2s (t->peer),
3093               GNUNET_MESH_DEBUG_TS2S (state));
3094   t->state = state;
3095 }
3096
3097
3098 /**
3099  * Cache a message to be sent once tunnel is online.
3100  *
3101  * @param t Tunnel to hold the message.
3102  * @param ch Channel the message is about.
3103  * @param msg Message itself (copy will be made).
3104  * @param fwd Is this fwd?
3105  */
3106 static void
3107 tunnel_queue_data (struct MeshTunnel2 *t,
3108                    struct MeshChannel *ch,
3109                    struct GNUNET_MessageHeader *msg,
3110                    int fwd)
3111 {
3112   struct MeshTunnelQueue *tq;
3113   uint16_t size = ntohs (msg->size);
3114
3115   tq = GNUNET_malloc (sizeof (struct MeshTunnelQueue) + size);
3116
3117   tq->ch = ch;
3118   memcpy (&tq[1], msg, size);
3119   GNUNET_CONTAINER_DLL_insert_tail (t->tq_head, t->tq_tail, tq);
3120 }
3121
3122
3123 /**
3124  * Send all cached messages that we can, tunnel is online.
3125  *
3126  * @param t Tunnel that holds the messages.
3127  * @param fwd Is this fwd?
3128  */
3129 static void
3130 tunnel_send_queued_data (struct MeshTunnel2 *t, int fwd)
3131 {
3132   struct MeshTunnelQueue *tq;
3133   struct MeshTunnelQueue *next;
3134   unsigned int room;
3135
3136   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3137               "tunnel_send_queued_data on tunnel %s\n",
3138               peer2s (t->peer));
3139   room = tunnel_get_buffer (t, fwd);
3140   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  buffer space: %u\n", room);
3141   for (tq = t->tq_head; NULL != tq && room > 0; tq = next)
3142   {
3143     next = tq->next;
3144     room--;
3145     GNUNET_CONTAINER_DLL_remove (t->tq_head, t->tq_tail, tq);
3146     send_prebuilt_message_channel ((struct GNUNET_MessageHeader *) &tq[1],
3147                                    tq->ch, fwd);
3148
3149     GNUNET_free (tq);
3150   }
3151 }
3152
3153
3154 static void
3155 connection_change_state (struct MeshConnection* c,
3156                          enum MeshConnectionState state)
3157 {
3158   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3159               "Connection %s state was %s\n",
3160               GNUNET_h2s (&c->id), GNUNET_MESH_DEBUG_CS2S (c->state));
3161   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3162               "Connection %s state is now %s\n",
3163               GNUNET_h2s (&c->id), GNUNET_MESH_DEBUG_CS2S (state));
3164   c->state = state;
3165 }
3166
3167
3168 /**
3169  * Add a client to a channel, initializing all needed data structures.
3170  * 
3171  * @param ch Channel to which add the client.
3172  * @param c Client which to add to the channel.
3173  */
3174 static void
3175 channel_add_client (struct MeshChannel *ch, struct MeshClient *c)
3176 {
3177   struct MeshTunnel2 *t = ch->t;
3178
3179   if (NULL != ch->dest)
3180   {
3181     GNUNET_break (0);
3182     return;
3183   }
3184
3185   /* Assign local id as destination */
3186   while (NULL != channel_get_by_local_id (c, t->next_local_chid))
3187     t->next_local_chid = (t->next_local_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
3188   ch->lid_dest = t->next_local_chid++;
3189   t->next_local_chid = t->next_local_chid | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
3190
3191   /* Store in client's hashmap */
3192   if (GNUNET_OK !=
3193       GNUNET_CONTAINER_multihashmap32_put (c->incoming_channels,
3194                                            ch->lid_dest, ch,
3195                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
3196   {
3197     GNUNET_break (0);
3198     return;
3199   }
3200
3201   GNUNET_break (NULL == ch->dest_rel);
3202   ch->dest_rel = GNUNET_new (struct MeshChannelReliability);
3203   ch->dest_rel->ch = ch;
3204   ch->dest_rel->expected_delay = MESH_RETRANSMIT_TIME;
3205
3206   ch->dest = c;
3207 }
3208
3209
3210 static struct MeshConnection *
3211 tunnel_use_path (struct MeshTunnel2 *t, struct MeshPeerPath *p)
3212 {
3213   struct MeshConnection *c;
3214   struct GNUNET_HashCode cid;
3215   struct MeshPeer *peer;
3216   unsigned int own_pos;
3217
3218   GNUNET_CRYPTO_hash_create_random (GNUNET_CRYPTO_QUALITY_NONCE, &cid);
3219
3220   c = connection_new (&cid);
3221   c->t = t;
3222   for (own_pos = 0; own_pos < p->length; own_pos++)
3223   {
3224     if (p->peers[own_pos] == myid)
3225       break;
3226   }
3227   if (own_pos > p->length - 1)
3228   {
3229     GNUNET_break (0);
3230     connection_destroy (c);
3231     return NULL;
3232   }
3233   c->own_pos = own_pos;
3234   c->path = p;
3235
3236   if (0 == own_pos)
3237   {
3238     c->fwd_maintenance_task =
3239         GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
3240                                       &connection_fwd_keepalive, c);
3241   }
3242
3243   peer = connection_get_next_hop (c);
3244   if (NULL == peer->connections)
3245   {
3246     connection_destroy (c);
3247     return NULL;
3248   }
3249   GNUNET_CONTAINER_multihashmap_put (peer->connections, &c->id, c,
3250                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
3251   peer = connection_get_prev_hop (c);
3252   if (NULL == peer->connections)
3253   {
3254     connection_destroy (c);
3255     return NULL;
3256   }
3257   GNUNET_CONTAINER_multihashmap_put (peer->connections, &c->id, c,
3258                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
3259   return c;
3260 }
3261
3262
3263 /**
3264  * Notifies a tunnel that a connection has broken that affects at least
3265  * some of its peers. Sends a notification towards the root of the tree.
3266  * In case the peer is the owner of the tree, notifies the client that owns
3267  * the tunnel and tries to reconnect.
3268  * 
3269  * FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME 
3270  *
3271  * @param t Tunnel affected.
3272  * @param p1 Peer that got disconnected from p2.
3273  * @param p2 Peer that got disconnected from p1.
3274  *
3275  * @return Short ID of the peer disconnected (either p1 or p2).
3276  *         0 if the tunnel remained unaffected.
3277  */
3278 static GNUNET_PEER_Id
3279 tunnel_notify_connection_broken (struct MeshTunnel2* t,
3280                                  GNUNET_PEER_Id p1, GNUNET_PEER_Id p2)
3281 {
3282 //   if (myid != p1 && myid != p2) FIXME
3283 //   {
3284 //     return;
3285 //   }
3286 // 
3287 //   if (tree_get_predecessor (t->tree) != 0)
3288 //   {
3289 //     /* We are the peer still connected, notify owner of the disconnection. */
3290 //     struct GNUNET_MESH_PathBroken msg;
3291 //     struct GNUNET_PeerIdentity neighbor;
3292 // 
3293 //     msg.header.size = htons (sizeof (msg));
3294 //     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN);
3295 //     GNUNET_PEER_resolve (t->id.oid, &msg.oid);
3296 //     msg.tid = htonl (t->id.tid);
3297 //     msg.peer1 = my_full_id;
3298 //     GNUNET_PEER_resolve (pid, &msg.peer2);
3299 //     GNUNET_PEER_resolve (tree_get_predecessor (t->tree), &neighbor);
3300 //     send_prebuilt_message (&msg.header, &neighbor, t);
3301 //   }
3302   return 0;
3303 }
3304
3305
3306 /**
3307  * Send an end-to-end ACK message for the most recent in-sequence payload.
3308  *
3309  * If channel is not reliable, do nothing.
3310  *
3311  * @param ch Channel this is about.
3312  * @param fwd Is for FWD traffic? (ACK dest->owner)
3313  */
3314 static void
3315 channel_send_data_ack (struct MeshChannel *ch, int fwd)
3316 {
3317   struct GNUNET_MESH_DataACK msg;
3318   struct MeshChannelReliability *rel;
3319   struct MeshReliableMessage *copy;
3320   unsigned int delta;
3321   uint64_t mask;
3322   uint16_t type;
3323
3324   if (GNUNET_NO == ch->reliable)
3325   {
3326     return;
3327   }
3328   rel = fwd ? ch->dest_rel : ch->root_rel;
3329   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3330               "send_data_ack for %u\n",
3331               rel->mid_recv - 1);
3332
3333   type = GNUNET_MESSAGE_TYPE_MESH_DATA_ACK;
3334   msg.header.type = htons (type);
3335   msg.header.size = htons (sizeof (msg));
3336   msg.chid = htonl (ch->gid);
3337   msg.mid = htonl (rel->mid_recv - 1);
3338   msg.futures = 0;
3339   for (copy = rel->head_recv; NULL != copy; copy = copy->next)
3340   {
3341     if (copy->type != type)
3342       continue;
3343     delta = copy->mid - rel->mid_recv;
3344     if (63 < delta)
3345       break;
3346     mask = 0x1LL << delta;
3347     msg.futures |= mask;
3348     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3349                 " setting bit for %u (delta %u) (%llX) -> %llX\n",
3350                 copy->mid, delta, mask, msg.futures);
3351   }
3352   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " final futures %llX\n", msg.futures);
3353
3354   send_prebuilt_message_channel (&msg.header, ch, fwd);
3355   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_data_ack END\n");
3356 }
3357
3358
3359 /**
3360  * Send an ACK informing the predecessor about the available buffer space.
3361  *
3362  * Note that for fwd ack, the FWD mean forward *traffic* (root->dest),
3363  * the ACK itself goes "back" (dest->root).
3364  *
3365  * @param c Connection on which to send the ACK.
3366  * @param buffer How much space free to advertise?
3367  * @param fwd Is this FWD ACK? (Going dest->owner)
3368  */
3369 static void
3370 connection_send_ack (struct MeshConnection *c, unsigned int buffer, int fwd)
3371 {
3372   struct MeshFlowControl *next_fc;
3373   struct MeshFlowControl *prev_fc;
3374   struct GNUNET_MESH_ACK msg;
3375   uint32_t ack;
3376   int delta;
3377
3378   next_fc = fwd ? &c->fwd_fc : &c->bck_fc;
3379   prev_fc = fwd ? &c->bck_fc : &c->fwd_fc;
3380
3381   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3382               "connection send %s ack on %s\n",
3383               fwd ? "FWD" : "BCK", GNUNET_h2s (&c->id));
3384
3385   /* Check if we need to transmit the ACK */
3386   if (prev_fc->last_ack_sent - prev_fc->last_pid_recv > 3)
3387   {
3388     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, buffer > 3\n");
3389     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3390                 "  last pid recv: %u, last ack sent: %u\n",
3391                 prev_fc->last_pid_recv, prev_fc->last_ack_sent);
3392     return;
3393   }
3394
3395   /* Ok, ACK might be necessary, what PID to ACK? */
3396   delta = next_fc->queue_max - next_fc->queue_n;
3397   ack = prev_fc->last_pid_recv + delta;
3398   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ACK %u\n", ack);
3399   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3400               " last pid %u, last ack %u, qmax %u, q %u\n",
3401               prev_fc->last_pid_recv, prev_fc->last_ack_sent,
3402               next_fc->queue_max, next_fc->queue_n);
3403   if (ack == prev_fc->last_ack_sent)
3404   {
3405     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending FWD ACK, not needed\n");
3406     return;
3407   }
3408
3409   prev_fc->last_ack_sent = ack;
3410
3411   /* Build ACK message and send on connection */
3412   msg.header.size = htons (sizeof (msg));
3413   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_ACK);
3414   msg.ack = htonl (ack);
3415   msg.cid = c->id;
3416
3417   send_prebuilt_message_connection (&msg.header, c, NULL, !fwd);
3418 }
3419
3420
3421 /**
3422  * Modify the mesh message TID from global to local and send to client.
3423  * 
3424  * @param ch Channel on which to send the message.
3425  * @param msg Message to modify and send.
3426  * @param c Client to send to.
3427  * @param tid Tunnel ID to use (c can be both owner and client).
3428  */
3429 static void
3430 channel_send_client_to_tid (struct MeshChannel *ch,
3431                              const struct GNUNET_MESH_Data *msg,
3432                              struct MeshClient *c, MESH_ChannelNumber id)
3433 {
3434   struct GNUNET_MESH_LocalData *copy;
3435   uint16_t size = ntohs (msg->header.size) - sizeof (struct GNUNET_MESH_Data);
3436   char cbuf[size + sizeof (struct GNUNET_MESH_LocalData)];
3437
3438   if (size < sizeof (struct GNUNET_MessageHeader))
3439   {
3440     GNUNET_break_op (0);
3441     return;
3442   }
3443   if (NULL == c)
3444   {
3445     GNUNET_break (0);
3446     return;
3447   }
3448   copy = (struct GNUNET_MESH_LocalData *) cbuf;
3449   memcpy (&copy[1], &msg[1], size);
3450   copy->header.size = htons (sizeof (struct GNUNET_MESH_LocalData) + size);
3451   copy->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA);
3452   copy->id = htonl (id);
3453   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
3454                                               &copy->header, GNUNET_NO);
3455 }
3456
3457
3458 /**
3459  * We have received a message out of order, or the client is not ready.
3460  * Buffer it until we receive an ACK from the client or the missing
3461  * message from the channel.
3462  *
3463  * @param msg Message to buffer (MUST be of type MESH_DATA).
3464  * @param rel Reliability data to the corresponding direction.
3465  */
3466 static void
3467 channel_rel_add_buffered_data (const struct GNUNET_MESH_Data *msg,
3468                                struct MeshChannelReliability *rel)
3469 {
3470   struct MeshReliableMessage *copy;
3471   struct MeshReliableMessage *prev;
3472   uint32_t mid;
3473   uint16_t size;
3474   
3475   size = ntohs (msg->header.size);
3476   mid = ntohl (msg->mid);
3477   
3478   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add_buffered_data %u\n", mid);
3479   
3480   copy = GNUNET_malloc (sizeof (*copy) + size);
3481   copy->mid = mid;
3482   copy->rel = rel;
3483   memcpy (&copy[1], msg, size);
3484   
3485   rel->n_recv++;
3486   
3487   // FIXME do something better than O(n), although n < 64...
3488   // FIXME start from the end (most messages are the latest ones)
3489   for (prev = rel->head_recv; NULL != prev; prev = prev->next)
3490   {
3491     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " prev %u\n", prev->mid);
3492     if (GMC_is_pid_bigger (prev->mid, mid))
3493     {
3494       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " bingo!\n");
3495       GNUNET_CONTAINER_DLL_insert_before (rel->head_recv, rel->tail_recv,
3496                                           prev, copy);
3497       return;
3498     }
3499   }
3500     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " insert at tail!\n");
3501     GNUNET_CONTAINER_DLL_insert_tail (rel->head_recv, rel->tail_recv, copy);
3502     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add_buffered_data END\n");
3503 }
3504
3505
3506 static void
3507 channel_send_client_data (struct MeshChannel *ch,
3508                           const struct GNUNET_MESH_Data *msg,
3509                           int fwd)
3510 {
3511   if (fwd)
3512   {
3513     if (ch->dest_rel->client_ready)
3514       channel_send_client_to_tid (ch, msg, ch->dest, ch->lid_dest);
3515     else
3516       channel_rel_add_buffered_data (msg, ch->dest_rel);
3517   }
3518   else
3519   {
3520     if (ch->root_rel->client_ready)
3521       channel_send_client_to_tid (ch, msg, ch->root, ch->lid_root);
3522     else
3523       channel_rel_add_buffered_data (msg, ch->root_rel);
3524   }
3525 }
3526
3527
3528 /**
3529  * Send a buffered message to the client, for in order delivery or
3530  * as result of client ACK.
3531  *
3532  * @param ch Channel on which to empty the message buffer.
3533  * @param c Client to send to.
3534  * @param rel Reliability structure to corresponding peer.
3535  *            If rel == bck_rel, this is FWD data.
3536  */
3537 static void
3538 channel_send_client_buffered_data (struct MeshChannel *ch,
3539                                    struct MeshClient *c,
3540                                    int fwd)
3541 {
3542   struct MeshReliableMessage *copy;
3543   struct MeshChannelReliability *rel;
3544
3545   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_buffered_data\n");
3546   rel = fwd ? ch->dest_rel : ch->root_rel;
3547   if (GNUNET_NO == rel->client_ready)
3548   {
3549     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client not ready\n");
3550     return;
3551   }
3552
3553   copy = rel->head_recv;
3554   /* We never buffer channel management messages */
3555   if (NULL != copy)
3556   {
3557     if (copy->mid == rel->mid_recv || GNUNET_NO == ch->reliable)
3558     {
3559       struct GNUNET_MESH_Data *msg = (struct GNUNET_MESH_Data *) &copy[1];
3560
3561       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3562                   " have %u! now expecting %u\n",
3563                   copy->mid, rel->mid_recv + 1);
3564       channel_send_client_data (ch, msg, fwd);
3565       rel->n_recv--;
3566       rel->mid_recv++;
3567       GNUNET_CONTAINER_DLL_remove (rel->head_recv, rel->tail_recv, copy);
3568       GNUNET_free (copy);
3569     }
3570     else
3571     {
3572       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3573                   " reliable && don't have %u, next is %u\n",
3574                   rel->mid_recv,
3575                   copy->mid);
3576       return;
3577     }
3578   }
3579   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_buffered_data END\n");
3580 }
3581
3582
3583 /**
3584  * Destroy a reliable message after it has been acknowledged, either by
3585  * direct mid ACK or bitfield. Updates the appropriate data structures and
3586  * timers and frees all memory.
3587  * 
3588  * @param copy Message that is no longer needed: remote peer got it.
3589  */
3590 static void
3591 rel_message_free (struct MeshReliableMessage *copy)
3592 {
3593   struct MeshChannelReliability *rel;
3594   struct GNUNET_TIME_Relative time;
3595
3596   rel = copy->rel;
3597   time = GNUNET_TIME_absolute_get_duration (copy->timestamp);
3598   rel->expected_delay.rel_value_us *= 7;
3599   rel->expected_delay.rel_value_us += time.rel_value_us;
3600   rel->expected_delay.rel_value_us /= 8;
3601   rel->n_sent--;
3602   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! Freeing %u\n", copy->mid);
3603   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    n_sent %u\n", rel->n_sent);
3604   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  took %s\n",
3605               GNUNET_STRINGS_relative_time_to_string (time, GNUNET_NO));
3606   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  new expected delay %s\n",
3607               GNUNET_STRINGS_relative_time_to_string (rel->expected_delay,
3608                                                       GNUNET_NO));
3609   rel->retry_timer = rel->expected_delay;
3610   GNUNET_CONTAINER_DLL_remove (rel->head_sent, rel->tail_sent, copy);
3611   GNUNET_free (copy);
3612 }
3613
3614
3615 /**
3616  * Destroy all reliable messages queued for a channel,
3617  * during a channel destruction.
3618  * Frees the reliability structure itself.
3619  *
3620  * @param rel Reliability data for a channel.
3621  */
3622 static void
3623 channel_rel_free_all (struct MeshChannelReliability *rel)
3624 {
3625   struct MeshReliableMessage *copy;
3626   struct MeshReliableMessage *next;
3627
3628   if (NULL == rel)
3629     return;
3630
3631   for (copy = rel->head_recv; NULL != copy; copy = next)
3632   {
3633     next = copy->next;
3634     GNUNET_CONTAINER_DLL_remove (rel->head_recv, rel->tail_recv, copy);
3635     GNUNET_free (copy);
3636   }
3637   for (copy = rel->head_sent; NULL != copy; copy = next)
3638   {
3639     next = copy->next;
3640     GNUNET_CONTAINER_DLL_remove (rel->head_sent, rel->tail_sent, copy);
3641     GNUNET_free (copy);
3642   }
3643   if (GNUNET_SCHEDULER_NO_TASK != rel->retry_task)
3644     GNUNET_SCHEDULER_cancel (rel->retry_task);
3645   GNUNET_free (rel);
3646 }
3647
3648
3649 /**
3650  * Mark future messages as ACK'd.
3651  *
3652  * @param rel Reliability data.
3653  * @param msg DataACK message with a bitfield of future ACK'd messages.
3654  */
3655 static void
3656 channel_rel_free_sent (struct MeshChannelReliability *rel,
3657                        const struct GNUNET_MESH_DataACK *msg)
3658 {
3659   struct MeshReliableMessage *copy;
3660   struct MeshReliableMessage *next;
3661   uint64_t bitfield;
3662   uint64_t mask;
3663   uint32_t mid;
3664   uint32_t target;
3665   unsigned int i;
3666
3667   bitfield = msg->futures;
3668   mid = ntohl (msg->mid);
3669   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3670               "free_sent_reliable %u %llX\n",
3671               mid, bitfield);
3672   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3673               " rel %p, head %p\n",
3674               rel, rel->head_sent);
3675   for (i = 0, copy = rel->head_sent;
3676        i < 64 && NULL != copy && 0 != bitfield;
3677        i++)
3678   {
3679     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3680                 " trying bit %u (mid %u)\n",
3681                 i, mid + i + 1);
3682     mask = 0x1LL << i;
3683     if (0 == (bitfield & mask))
3684      continue;
3685
3686     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " set!\n");
3687     /* Bit was set, clear the bit from the bitfield */
3688     bitfield &= ~mask;
3689
3690     /* The i-th bit was set. Do we have that copy? */
3691     /* Skip copies with mid < target */
3692     target = mid + i + 1;
3693     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " target %u\n", target);
3694     while (NULL != copy && GMC_is_pid_bigger (target, copy->mid))
3695      copy = copy->next;
3696
3697     /* Did we run out of copies? (previously freed, it's ok) */
3698     if (NULL == copy)
3699     {
3700      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "run out of copies...\n");
3701      return;
3702     }
3703
3704     /* Did we overshoot the target? (previously freed, it's ok) */
3705     if (GMC_is_pid_bigger (copy->mid, target))
3706     {
3707      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " next copy %u\n", copy->mid);
3708      continue;
3709     }
3710
3711     /* Now copy->mid == target, free it */
3712     next = copy->next;
3713     rel_message_free (copy);
3714     copy = next;
3715   }
3716   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "free_sent_reliable END\n");
3717 }
3718
3719
3720 /**
3721  * We haven't received an ACK after a certain time: restransmit the message.
3722  *
3723  * @param cls Closure (MeshReliableMessage with the message to restransmit)
3724  * @param tc TaskContext.
3725  */
3726 static void
3727 channel_retransmit_message (void *cls,
3728                             const struct GNUNET_SCHEDULER_TaskContext *tc)
3729 {
3730   struct MeshChannelReliability *rel = cls;
3731   struct MeshReliableMessage *copy;
3732   struct MeshPeerQueue *q;
3733   struct MeshChannel *ch;
3734   struct MeshConnection *c;
3735   struct GNUNET_MESH_Data *payload;
3736   struct MeshPeer *hop;
3737   int fwd;
3738
3739   rel->retry_task = GNUNET_SCHEDULER_NO_TASK;
3740   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
3741     return;
3742
3743   ch = rel->ch;
3744   copy = rel->head_sent;
3745   if (NULL == copy)
3746   {
3747     GNUNET_break (0);
3748     return;
3749   }
3750
3751   /* Search the message to be retransmitted in the outgoing queue.
3752    * Check only the queue for the connection that is going to be used,
3753    * if the message is stuck in some other connection's queue we shouldn't
3754    * act upon it:
3755    * - cancelling it and sending the new one doesn't guarantee it's delivery,
3756    *   the old connection could be temporary stalled or the queue happened to
3757    *   be long at time of insertion.
3758    * - not sending the new one could cause terrible delays the old connection
3759    *   is stalled.
3760    */
3761   payload = (struct GNUNET_MESH_Data *) &copy[1];
3762   fwd = (rel == ch->root_rel);
3763   c = tunnel_get_connection (ch->t, fwd);
3764   hop = connection_get_hop (c, fwd);
3765   for (q = hop->queue_head; NULL != q; q = q->next)
3766   {
3767     if (ntohs (payload->header.type) == q->type && ch == q->ch)
3768     {
3769       struct GNUNET_MESH_Data *queued_data = q->cls;
3770
3771       if (queued_data->mid == payload->mid)
3772         break;
3773     }
3774   }
3775
3776   /* Message not found in the queue that we are going to use. */
3777   if (NULL == q)
3778   {
3779     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! RETRANSMIT %u\n", copy->mid);
3780
3781     send_prebuilt_message_channel (&payload->header, ch, fwd);
3782     GNUNET_STATISTICS_update (stats, "# data retransmitted", 1, GNUNET_NO);
3783   }
3784   else
3785   {
3786     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! ALREADY IN QUEUE %u\n", copy->mid);
3787   }
3788
3789   rel->retry_timer = GNUNET_TIME_STD_BACKOFF (rel->retry_timer);
3790   rel->retry_task = GNUNET_SCHEDULER_add_delayed (rel->retry_timer,
3791                                                   &channel_retransmit_message,
3792                                                   cls);
3793 }
3794
3795
3796 /**
3797  * Send ACK on one or more connections due to buffer space to the client.
3798  *
3799  * Iterates all connections of the tunnel and sends ACKs appropriately.
3800  *
3801  * @param ch Channel which has some free buffer space.
3802  * @param fwd Is this in for FWD traffic? (ACK goes dest->root)
3803  */
3804 static void
3805 channel_send_connections_ack (struct MeshChannel *ch,
3806                               unsigned int buffer,
3807                               int fwd)
3808 {
3809   struct MeshTunnel2 *t = ch->t;
3810   struct MeshConnection *c;
3811   struct MeshFlowControl *fc;
3812   uint32_t allowed;
3813   uint32_t to_allow;
3814   uint32_t allow_per_connection;
3815   unsigned int cs;
3816
3817   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3818               "Channel send connection %s ack on %s:%X\n",
3819               fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3820
3821   /* Count connections, how many messages are already allowed */
3822   for (cs = 0, allowed = 0, c = t->connection_head; NULL != c; c = c->next)
3823   {
3824     fc = fwd ? &c->fwd_fc : &c->bck_fc;
3825     if (GMC_is_pid_bigger(fc->last_pid_recv, fc->last_ack_sent))
3826     {
3827       GNUNET_break (0);
3828       continue;
3829     }
3830     allowed += fc->last_ack_sent - fc->last_pid_recv;
3831     cs++;
3832   }
3833
3834   /* Make sure there is no overflow */
3835   if (allowed > buffer)
3836   {
3837     GNUNET_break (0);
3838     return;
3839   }
3840
3841   /* Authorize connections to send more data */
3842   to_allow = buffer - allowed;
3843
3844   for (c = t->connection_head; NULL != c && to_allow > 0; c = c->next)
3845   {
3846     allow_per_connection = to_allow/cs;
3847     to_allow -= allow_per_connection;
3848     cs--;
3849     fc = fwd ? &c->fwd_fc : &c->bck_fc;
3850     if (fc->last_ack_sent - fc->last_pid_recv > 64 / 3)
3851     {
3852       continue;
3853     }
3854     connection_send_ack (c, allow_per_connection, fwd);
3855   }
3856
3857   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3858                 "Channel send connection %s ack on %s:%X\n",
3859                 fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3860   GNUNET_break (to_allow == 0);
3861 }
3862
3863
3864 /**
3865  * Send an ACK on the appropriate connection/channel, depending on
3866  * the direction and the position of the peer.
3867  *
3868  * @param c Which connection to send the hop-by-hop ACK.
3869  * @param ch Channel, if any.
3870  * @param fwd Is this a fwd ACK? (will go dest->root)
3871  */
3872 static void
3873 send_ack (struct MeshConnection *c, struct MeshChannel *ch, int fwd)
3874 {
3875   unsigned int buffer;
3876
3877   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3878               "send ack %s on %p %p\n",
3879               fwd ? "FWD" : "BCK", c, ch);
3880   if (NULL == c || connection_is_terminal (c, fwd))
3881   {
3882     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  getting from Channel\n");
3883     buffer = tunnel_get_buffer (NULL == c ? ch->t : c->t, fwd);
3884   }
3885   else
3886   {
3887     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  getting from Connection\n");
3888     GNUNET_assert (NULL != c);
3889     buffer = connection_get_buffer (c, fwd);
3890   }
3891   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  buffer available: %u\n", buffer);
3892
3893   if (NULL == c)
3894   {
3895     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on all connections\n");
3896     GNUNET_assert (NULL != ch);
3897     channel_send_connections_ack (ch, buffer, fwd);
3898   }
3899   else if (connection_is_origin (c, fwd))
3900   {
3901     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on channel...\n");
3902     if (0 < buffer)
3903     {
3904       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  really sending!\n");
3905       GNUNET_assert (NULL != ch);
3906       send_local_ack (ch, fwd);
3907     }
3908   }
3909   else
3910   {
3911     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on connection\n");
3912     connection_send_ack (c, buffer, fwd);
3913   }
3914 }
3915
3916
3917 /**
3918  * Channel was ACK'd by remote peer, mark as ready and cancel retransmission.
3919  *
3920  * @param ch Channel to mark as ready.
3921  * @param fwd Was the CREATE message sent fwd?
3922  */
3923 static void
3924 channel_confirm (struct MeshChannel *ch, int fwd)
3925 {
3926   struct MeshChannelReliability *rel;
3927   struct MeshReliableMessage *copy;
3928   struct MeshReliableMessage *next;
3929
3930   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3931               "  channel confirm %s %s:%X\n",
3932               fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3933   ch->state = MESH_CHANNEL_READY;
3934
3935   rel = fwd ? ch->root_rel : ch->dest_rel;
3936   for (copy = rel->head_sent; NULL != copy; copy = next)
3937   {
3938     struct GNUNET_MessageHeader *msg;
3939
3940     next = copy->next;
3941     msg = (struct GNUNET_MessageHeader *) &copy[1];
3942     if (ntohs (msg->type) == GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE)
3943     {
3944       rel_message_free (copy);
3945       /* TODO return? */
3946     }
3947   }
3948   if (GNUNET_NO == rel->client_ready)
3949     send_local_ack (ch, fwd);
3950 }
3951
3952
3953 /**
3954  * Save a copy to retransmit in case it gets lost.
3955  *
3956  * Initializes all needed callbacks and timers.
3957  *
3958  * @param ch Channel this message goes on.
3959  * @param msg Message to copy.
3960  * @param fwd Is this fwd traffic?
3961  */
3962 static void
3963 channel_save_copy (struct MeshChannel *ch,
3964                    const struct GNUNET_MessageHeader *msg,
3965                    int fwd)
3966 {
3967   struct MeshChannelReliability *rel;
3968   struct MeshReliableMessage *copy;
3969   uint32_t mid;
3970   uint16_t type;
3971   uint16_t size;
3972
3973   rel = fwd ? ch->root_rel : ch->dest_rel;
3974   mid = rel->mid_send;
3975   type = ntohs (msg->type);
3976   size = ntohs (msg->size);
3977
3978   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! SAVE %u\n", mid);
3979   copy = GNUNET_malloc (sizeof (struct MeshReliableMessage) + size);
3980   copy->mid = mid;
3981   copy->timestamp = GNUNET_TIME_absolute_get ();
3982   copy->rel = rel;
3983   copy->type = type;
3984   memcpy (&copy[1], msg, size);
3985   rel->n_sent++;
3986   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " n_sent %u\n", rel->n_sent);
3987   GNUNET_CONTAINER_DLL_insert_tail (rel->head_sent, rel->tail_sent, copy);
3988   if (GNUNET_SCHEDULER_NO_TASK == rel->retry_task)
3989   {
3990     rel->retry_timer =
3991         GNUNET_TIME_relative_multiply (rel->expected_delay,
3992                                         MESH_RETRANSMIT_MARGIN);
3993     rel->retry_task =
3994         GNUNET_SCHEDULER_add_delayed (rel->retry_timer,
3995                                       &channel_retransmit_message,
3996                                       rel);
3997   }
3998 }
3999
4000
4001 /**
4002  * Send keepalive packets for a connection.
4003  *
4004  * @param c Connection to keep alive..
4005  * @param fwd Is this a FWD keepalive? (owner -> dest).
4006  */
4007 static void
4008 connection_keepalive (struct MeshConnection *c, int fwd)
4009 {
4010   struct GNUNET_MESH_ConnectionKeepAlive *msg;
4011   size_t size = sizeof (struct GNUNET_MESH_ConnectionKeepAlive);
4012   char cbuf[size];
4013   uint16_t type;
4014
4015   type = fwd ? GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE :
4016                GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE;
4017
4018   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4019               "sending %s keepalive for connection %s[%d]\n",
4020               fwd ? "FWD" : "BCK",
4021               peer2s (c->t->peer),
4022               c->id);
4023
4024   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) cbuf;
4025   msg->header.size = htons (size);
4026   msg->header.type = htons (type);
4027   msg->cid = c->id;
4028
4029   send_prebuilt_message_connection (&msg->header, c, NULL, fwd);
4030 }
4031
4032
4033 /**
4034  * Send CONNECTION_{CREATE/ACK} packets for a connection.
4035  *
4036  * @param c Connection for which to send the message.
4037  * @param fwd If GNUNET_YES, send CREATE, otherwise send ACK.
4038  */
4039 static void
4040 connection_recreate (struct MeshConnection *c, int fwd)
4041 {
4042   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "sending connection recreate\n");
4043   if (fwd)
4044     send_connection_create (c);
4045   else
4046     send_connection_ack (c, GNUNET_NO);
4047 }
4048
4049
4050 /**
4051  * Generic connection timer management.
4052  * Depending on the role of the peer in the connection will send the
4053  * appropriate message (build or keepalive)
4054  *
4055  * @param c Conncetion to maintain.
4056  * @param fwd Is FWD?
4057  */
4058 static void
4059 connection_maintain (struct MeshConnection *c, int fwd)
4060 {
4061   if (MESH_TUNNEL_SEARCHING == c->t->state)
4062   {
4063     /* TODO DHT GET with RO_BART */
4064     return;
4065   }
4066   switch (c->state)
4067   {
4068     case MESH_CONNECTION_NEW:
4069       GNUNET_break (0);
4070     case MESH_CONNECTION_SENT:
4071       connection_recreate (c, fwd);
4072       break;
4073     case MESH_CONNECTION_READY:
4074       connection_keepalive (c, fwd);
4075       break;
4076     default:
4077       break;
4078   }
4079 }
4080
4081
4082 static void
4083 connection_fwd_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4084 {
4085   struct MeshConnection *c = cls;
4086
4087   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4088   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4089     return;
4090
4091   connection_maintain (c, GNUNET_YES);
4092   c->fwd_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
4093                                                           &connection_fwd_keepalive,
4094                                                           c);
4095 }
4096
4097
4098 static void
4099 connection_bck_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4100 {
4101   struct MeshConnection *c = cls;
4102
4103   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4104   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4105     return;
4106
4107   connection_maintain (c, GNUNET_NO);
4108   c->bck_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
4109                                                           &connection_bck_keepalive,
4110                                                           c);
4111 }
4112
4113
4114 /**
4115  * Send a message to all peers in this connection that the connection
4116  * is no longer valid.
4117  *
4118  * If some peer should not receive the message, it should be zero'ed out
4119  * before calling this function.
4120  *
4121  * @param c The connection whose peers to notify.
4122  */
4123 static void
4124 connection_send_destroy (struct MeshConnection *c)
4125 {
4126   struct GNUNET_MESH_ConnectionDestroy msg;
4127
4128   msg.header.size = htons (sizeof (msg));
4129   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY);;
4130   msg.cid = c->id;
4131   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4132               "  sending connection destroy for connection %s[%X]\n",
4133               peer2s (c->t->peer),
4134               c->id);
4135
4136   if (GNUNET_NO == connection_is_terminal (c, GNUNET_YES))
4137     send_prebuilt_message_connection (&msg.header, c, NULL, GNUNET_YES);
4138   if (GNUNET_NO == connection_is_terminal (c, GNUNET_NO))
4139     send_prebuilt_message_connection (&msg.header, c, NULL, GNUNET_NO);
4140   c->destroy = GNUNET_YES;
4141 }
4142
4143
4144 /**
4145  * Confirm we got a channel create.
4146  *
4147  * @param ch The channel to confirm.
4148  * @param fwd Should we send the ACK fwd?
4149  */
4150 static void
4151 channel_send_ack (struct MeshChannel *ch, int fwd)
4152 {
4153   struct GNUNET_MESH_ChannelManage msg;
4154
4155   msg.header.size = htons (sizeof (msg));
4156   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_ACK);
4157   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4158               "  sending channel ack for channel %s:%X\n",
4159               peer2s (ch->t->peer),
4160               ch->gid);
4161
4162   msg.chid = htonl (ch->gid);
4163   send_prebuilt_message_channel (&msg.header, ch, fwd);
4164 }
4165
4166
4167 /**
4168  * Send a message to all clients (local and remote) of this channel
4169  * notifying that the channel is no longer valid.
4170  *
4171  * If some peer or client should not receive the message,
4172  * should be zero'ed out before calling this function.
4173  *
4174  * @param ch The channel whose clients to notify.
4175  */
4176 static void
4177 channel_send_destroy (struct MeshChannel *ch)
4178 {
4179   struct GNUNET_MESH_ChannelManage msg;
4180
4181   msg.header.size = htons (sizeof (msg));
4182   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY);
4183   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4184               "  sending channel destroy for channel %s:%X\n",
4185               peer2s (ch->t->peer),
4186               ch->gid);
4187
4188   if (channel_is_terminal (ch, GNUNET_NO))
4189   {
4190     if (NULL != ch->root && GNUNET_NO == ch->root->shutting_down)
4191     {
4192       msg.chid = htonl (ch->lid_root);
4193       send_local_channel_destroy (ch, GNUNET_NO);
4194     }
4195   }
4196   else
4197   {
4198     msg.chid = htonl (ch->gid);
4199     send_prebuilt_message_channel (&msg.header, ch, GNUNET_NO);
4200   }
4201
4202   if (channel_is_terminal (ch, GNUNET_YES))
4203   {
4204     if (NULL != ch->dest && GNUNET_NO == ch->dest->shutting_down)
4205     {
4206       msg.chid = htonl (ch->lid_dest);
4207       send_local_channel_destroy (ch, GNUNET_YES);
4208     }
4209   }
4210   else
4211   {
4212     msg.chid = htonl (ch->gid);
4213     send_prebuilt_message_channel (&msg.header, ch, GNUNET_YES);
4214   }
4215 }
4216
4217
4218 /**
4219  * Create a tunnel.
4220  */
4221 static struct MeshTunnel2 *
4222 tunnel_new (void)
4223 {
4224   struct MeshTunnel2 *t;
4225
4226   t = GNUNET_new (struct MeshTunnel2);
4227   t->next_chid = 0;
4228   t->next_local_chid = GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
4229 //   if (GNUNET_OK !=
4230 //       GNUNET_CONTAINER_multihashmap_put (tunnels, tid, t,
4231 //                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
4232 //   {
4233 //     GNUNET_break (0);
4234 //     tunnel_destroy (t);
4235 //     return NULL;
4236 //   }
4237
4238 //   char salt[] = "salt";
4239 //   GNUNET_CRYPTO_kdf (&t->e_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4240 //                      salt, sizeof (salt),
4241 //                      &t->e_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4242 //                      &my_full_id, sizeof (struct GNUNET_PeerIdentity),
4243 //                      GNUNET_PEER_resolve2 (t->peer->id), sizeof (struct GNUNET_PeerIdentity),
4244 //                      NULL);
4245 //   GNUNET_CRYPTO_kdf (&t->d_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4246 //                      salt, sizeof (salt),
4247 //                      &t->d_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4248 //                      GNUNET_PEER_resolve2 (t->peer->id), sizeof (struct GNUNET_PeerIdentity),
4249 //                      &my_full_id, sizeof (struct GNUNET_PeerIdentity),
4250 //                      NULL);
4251
4252   return t;
4253 }
4254
4255
4256 /**
4257  * Add a connection to a tunnel.
4258  *
4259  * @param t Tunnel.
4260  * @param c Connection.
4261  */
4262 static void
4263 tunnel_add_connection (struct MeshTunnel2 *t, struct MeshConnection *c)
4264 {
4265   struct MeshConnection *aux;
4266   c->t = t;
4267   for (aux = t->connection_head; aux != NULL; aux = aux->next)
4268     if (aux == c)
4269       return;
4270   GNUNET_CONTAINER_DLL_insert_tail (t->connection_head, t->connection_tail, c);
4271 }
4272
4273
4274 /**
4275  * Initialize a Flow Control structure to the initial state.
4276  * 
4277  * @param fc Flow Control structure to initialize.
4278  */
4279 static void
4280 fc_init (struct MeshFlowControl *fc)
4281 {
4282   fc->next_pid = 0;
4283   fc->last_pid_sent = (uint32_t) -1; /* Next (expected) = 0 */
4284   fc->last_pid_recv = (uint32_t) -1;
4285   fc->last_ack_sent = (uint32_t) 0;
4286   fc->last_ack_recv = (uint32_t) 0;
4287   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
4288   fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
4289   fc->queue_n = 0;
4290   fc->queue_max = (max_msgs_queue / max_connections) + 1;
4291 }
4292
4293
4294 static struct MeshConnection *
4295 connection_new (const struct GNUNET_HashCode *cid)
4296 {
4297   struct MeshConnection *c;
4298
4299   c = GNUNET_new (struct MeshConnection);
4300   c->id = *cid;
4301   GNUNET_CONTAINER_multihashmap_put (connections, &c->id, c,
4302                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
4303   fc_init (&c->fwd_fc);
4304   fc_init (&c->bck_fc);
4305   c->fwd_fc.c = c;
4306   c->bck_fc.c = c;
4307
4308   return c;
4309 }
4310
4311
4312 /**
4313  * Find a connection.
4314  *
4315  * @param cid Connection ID.
4316  */
4317 static struct MeshConnection *
4318 connection_get (const struct GNUNET_HashCode *cid)
4319 {
4320   return GNUNET_CONTAINER_multihashmap_get (connections, cid);
4321 }
4322
4323
4324 static void
4325 connection_destroy (struct MeshConnection *c)
4326 {
4327   struct MeshPeer *peer;
4328
4329   if (NULL == c)
4330     return;
4331
4332   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "destroying connection %s[%X]\n",
4333               peer2s (c->t->peer),
4334               c->id);
4335
4336   /* Cancel all traffic */
4337   connection_cancel_queues (c, GNUNET_YES);
4338   connection_cancel_queues (c, GNUNET_NO);
4339
4340   /* Cancel maintainance task (keepalive/timeout) */
4341   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_maintenance_task)
4342     GNUNET_SCHEDULER_cancel (c->fwd_maintenance_task);
4343   if (GNUNET_SCHEDULER_NO_TASK != c->bck_maintenance_task)
4344     GNUNET_SCHEDULER_cancel (c->bck_maintenance_task);
4345
4346   /* Deregister from neighbors */
4347   peer = connection_get_next_hop (c);
4348   if (NULL != peer)
4349     GNUNET_CONTAINER_multihashmap_remove (peer->connections, &c->id, c);
4350   peer = connection_get_prev_hop (c);
4351   if (NULL != peer)
4352     GNUNET_CONTAINER_multihashmap_remove (peer->connections, &c->id, c);
4353
4354   /* Delete */
4355   GNUNET_STATISTICS_update (stats, "# connections", -1, GNUNET_NO);
4356   GNUNET_CONTAINER_DLL_remove (c->t->connection_head, c->t->connection_tail, c);
4357   GNUNET_free (c);
4358 }
4359
4360
4361 static void
4362 tunnel_destroy (struct MeshTunnel2 *t)
4363 {
4364   struct MeshConnection *c;
4365   struct MeshConnection *next;
4366
4367   if (NULL == t)
4368     return;
4369
4370   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "destroying tunnel %s\n",
4371               peer2s (t->peer));
4372
4373 //   if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (tunnels, &t->id, t))
4374 //     GNUNET_break (0);
4375
4376   for (c = t->connection_head; NULL != c; c = next)
4377   {
4378     next = c->next;
4379     connection_destroy (c);
4380   }
4381
4382   GNUNET_STATISTICS_update (stats, "# tunnels", -1, GNUNET_NO);
4383
4384   GNUNET_free (t);
4385 }
4386
4387
4388 /**
4389  * Tunnel is empty: destroy it.
4390  *
4391  * Notifies all connections about the destruction.
4392  *
4393  * @param t Tunnel to destroy. 
4394  */
4395 static void
4396 tunnel_destroy_empty (struct MeshTunnel2 *t)
4397 {
4398   struct MeshConnection *c;
4399
4400   for (c = t->connection_head; NULL != c; c = c->next)
4401   {
4402     if (GNUNET_NO == c->destroy)
4403       connection_send_destroy (c);
4404   }
4405
4406   if (0 == t->pending_messages)
4407     tunnel_destroy (t);
4408   else
4409     t->destroy = GNUNET_YES;
4410 }
4411
4412
4413 /**
4414  * Destroy tunnel if empty (no more channels).
4415  *
4416  * @param t Tunnel to destroy if empty.
4417  */
4418 static void
4419 tunnel_destroy_if_empty (struct MeshTunnel2 *t)
4420 {
4421   if (NULL != t->channel_head)
4422     return;
4423
4424   tunnel_destroy_empty (t);
4425 }
4426
4427
4428 /**
4429  * Destroy a channel and free all resources.
4430  * 
4431  * @param ch Channel to destroy.
4432  */
4433 static void
4434 channel_destroy (struct MeshChannel *ch)
4435 {
4436   struct MeshClient *c;
4437
4438   if (NULL == ch)
4439     return;
4440
4441   c = ch->root;
4442   if (NULL != c)
4443   {
4444     if (GNUNET_YES != GNUNET_CONTAINER_multihashmap32_remove (c->own_channels,
4445                                                               ch->lid_root, ch))
4446     {
4447       GNUNET_break (0);
4448     }
4449   }
4450
4451   c = ch->dest;
4452   if (NULL != c)
4453   {
4454     if (GNUNET_YES !=
4455         GNUNET_CONTAINER_multihashmap32_remove (c->incoming_channels,
4456                                                 ch->lid_dest, ch))
4457     {
4458       GNUNET_break (0);
4459     }
4460   }
4461
4462   channel_rel_free_all (ch->root_rel);
4463   channel_rel_free_all (ch->dest_rel);
4464
4465   GNUNET_CONTAINER_DLL_remove (ch->t->channel_head, ch->t->channel_tail, ch);
4466   GNUNET_STATISTICS_update (stats, "# channels", -1, GNUNET_NO);
4467
4468   GNUNET_free (ch);
4469 }
4470
4471 /**
4472  * Create a new channel.
4473  *
4474  * @param t Tunnel this channel is in.
4475  * @param owner Client that owns the channel, NULL for foreign channels.
4476  * @param lid_root Local ID for root client.
4477  *
4478  * @return A new initialized channel. NULL on error.
4479  */
4480 static struct MeshChannel *
4481 channel_new (struct MeshTunnel2 *t,
4482              struct MeshClient *owner, MESH_ChannelNumber lid_root)
4483 {
4484   struct MeshChannel *ch;
4485
4486   ch = GNUNET_new (struct MeshChannel);
4487   ch->root = owner;
4488   ch->lid_root = lid_root;
4489   ch->t = t;
4490
4491   GNUNET_CONTAINER_DLL_insert (t->channel_head, t->channel_tail, ch);
4492
4493   GNUNET_STATISTICS_update (stats, "# channels", 1, GNUNET_NO);
4494
4495   if (NULL != owner)
4496   {
4497     while (NULL != channel_get (t, t->next_chid))
4498       t->next_chid = (t->next_chid + 1) & ~GNUNET_MESH_LOCAL_CHANNEL_ID_CLI;
4499     ch->gid = t->next_chid;
4500     t->next_chid = (t->next_chid + 1) & ~GNUNET_MESH_LOCAL_CHANNEL_ID_CLI;
4501
4502     if(GNUNET_OK !=
4503        GNUNET_CONTAINER_multihashmap32_put (owner->own_channels, lid_root, ch,
4504                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
4505     {
4506       GNUNET_break (0);
4507       channel_destroy (ch);
4508       GNUNET_SERVER_receive_done (owner->handle, GNUNET_SYSERR);
4509       return NULL;
4510     }
4511   }
4512
4513   return ch;
4514 }
4515
4516
4517 /**
4518  * Set options in a channel, extracted from a bit flag field
4519  * 
4520  * @param ch Channel to set options to.
4521  * @param options Bit array in host byte order.
4522  */
4523 static void
4524 channel_set_options (struct MeshChannel *ch, uint32_t options)
4525 {
4526   ch->nobuffer = (options & GNUNET_MESH_OPTION_NOBUFFER) != 0 ?
4527                  GNUNET_YES : GNUNET_NO;
4528   ch->reliable = (options & GNUNET_MESH_OPTION_RELIABLE) != 0 ?
4529                  GNUNET_YES : GNUNET_NO;
4530 }
4531
4532
4533 /**
4534  * Iterator for deleting each channel whose client endpoint disconnected.
4535  *
4536  * @param cls Closure (client that has disconnected).
4537  * @param key The local channel id (used to access the hashmap).
4538  * @param value The value stored at the key (channel to destroy).
4539  *
4540  * @return GNUNET_OK, keep iterating.
4541  */
4542 static int
4543 channel_destroy_iterator (void *cls,
4544                           uint32_t key,
4545                           void *value)
4546 {
4547   struct MeshChannel *ch = value;
4548   struct MeshClient *c = cls;
4549   struct MeshTunnel2 *t;
4550
4551   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4552               " Channel %X (%X / %X) destroy, due to client %u shutdown.\n",
4553               ch->gid, ch->lid_root, ch->lid_dest, c->id);
4554
4555   if (c == ch->dest)
4556   {
4557     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is destination.\n", c->id);
4558   }
4559   if (c == ch->root)
4560   {
4561     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is owner.\n", c->id);
4562   }
4563
4564   t = ch->t;
4565   channel_send_destroy (ch);
4566   channel_destroy (ch);
4567   tunnel_destroy_if_empty (t);
4568
4569   return GNUNET_OK;
4570 }
4571
4572
4573 /**
4574  * Remove client's ports from the global hashmap on disconnect.
4575  *
4576  * @param cls Closure (unused).
4577  * @param key Port.
4578  * @param value Client structure.
4579  *
4580  * @return GNUNET_OK, keep iterating.
4581  */
4582 static int
4583 client_release_ports (void *cls,
4584                       uint32_t key,
4585                       void *value)
4586 {
4587   int res;
4588
4589   res = GNUNET_CONTAINER_multihashmap32_remove (ports, key, value);
4590   if (GNUNET_YES != res)
4591   {
4592     GNUNET_break (0);
4593     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4594                 "Port %u by client %p was not registered.\n",
4595                 key, value);
4596   }
4597   return GNUNET_OK;
4598 }
4599
4600
4601 /**
4602  * Timeout function due to lack of keepalive/traffic from the owner.
4603  * Destroys connection if called.
4604  *
4605  * @param cls Closure (connection to destroy).
4606  * @param tc TaskContext.
4607  */
4608 static void
4609 connection_fwd_timeout (void *cls,
4610                         const struct GNUNET_SCHEDULER_TaskContext *tc)
4611 {
4612   struct MeshConnection *c = cls;
4613
4614   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4615   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4616     return;
4617   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4618               "Connection %s[%X] FWD timed out. Destroying.\n",
4619               peer2s (c->t->peer),
4620               c->id);
4621
4622   if (connection_is_origin (c, GNUNET_YES)) /* If local, leave. */
4623     return;
4624
4625   connection_destroy (c);
4626 }
4627
4628
4629 /**
4630  * Timeout function due to lack of keepalive/traffic from the destination.
4631  * Destroys connection if called.
4632  *
4633  * @param cls Closure (connection to destroy).
4634  * @param tc TaskContext
4635  */
4636 static void
4637 connection_bck_timeout (void *cls,
4638                         const struct GNUNET_SCHEDULER_TaskContext *tc)
4639 {
4640   struct MeshConnection *c = cls;
4641
4642   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4643   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4644     return;
4645
4646   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4647               "Connection %s[%X] FWD timed out. Destroying.\n",
4648               peer2s (c->t->peer),
4649               c->id);
4650
4651   if (connection_is_origin (c, GNUNET_NO)) /* If local, leave. */
4652     return;
4653
4654   connection_destroy (c);
4655 }
4656
4657
4658 /**
4659  * Resets the connection timeout task, some other message has done the
4660  * task's job.
4661  * - For the first peer on the direction this means to send
4662  *   a keepalive or a path confirmation message (either create or ACK).
4663  * - For all other peers, this means to destroy the connection,
4664  *   due to lack of activity.
4665  * Starts the tiemout if no timeout was running (connection just created).
4666  *
4667  * @param c Connection whose timeout to reset.
4668  * @param fwd Is this forward?
4669  *
4670  * TODO use heap to improve efficiency of scheduler.
4671  */
4672 static void
4673 connection_reset_timeout (struct MeshConnection *c, int fwd)
4674 {
4675   GNUNET_SCHEDULER_TaskIdentifier *ti;
4676   GNUNET_SCHEDULER_Task f;
4677
4678   ti = fwd ? &c->fwd_maintenance_task : &c->bck_maintenance_task;
4679
4680   if (GNUNET_SCHEDULER_NO_TASK != *ti)
4681     GNUNET_SCHEDULER_cancel (*ti);
4682
4683   if (connection_is_origin (c, fwd)) /* Endpoint */
4684   {
4685     f  = fwd ? &connection_fwd_keepalive : &connection_bck_keepalive;
4686     *ti = GNUNET_SCHEDULER_add_delayed (refresh_connection_time, f, c);
4687   }
4688   else /* Relay */
4689   {
4690     struct GNUNET_TIME_Relative delay;
4691
4692     delay = GNUNET_TIME_relative_multiply (refresh_connection_time, 4);
4693     f  = fwd ? &connection_fwd_timeout : &connection_bck_timeout;
4694     *ti = GNUNET_SCHEDULER_add_delayed (delay, f, c);
4695   }
4696 }
4697
4698
4699 /**
4700  * Iterator to notify all connections of a broken link. Mark connections
4701  * to destroy after all traffic has been sent.
4702  *
4703  * @param cls Closure (peer disconnected).
4704  * @param key Current key code (tid).
4705  * @param value Value in the hash map (connection).
4706  *
4707  * @return GNUNET_YES if we should continue to iterate,
4708  *         GNUNET_NO if not.
4709  */
4710 static int
4711 connection_broken (void *cls,
4712                    const struct GNUNET_HashCode *key,
4713                    void *value)
4714 {
4715   struct MeshPeer *peer = cls;
4716   struct MeshConnection *c = value;
4717   struct GNUNET_MESH_ConnectionBroken msg;
4718   int fwd;
4719
4720   fwd = peer == connection_get_prev_hop (c);
4721
4722   if (connection_is_terminal (c, fwd))
4723   {
4724     /* Local shutdown: no point in iterating anymore */
4725     connection_destroy (c);
4726     return GNUNET_NO;
4727   }
4728   connection_cancel_queues (c, !fwd);
4729
4730   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectionBroken));
4731   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN);
4732   msg.cid = c->id;
4733   msg.peer1 = my_full_id;
4734   msg.peer2 = *GNUNET_PEER_resolve2 (peer->id);
4735   send_prebuilt_message_connection (&msg.header, c, NULL, fwd);
4736   c->destroy = GNUNET_YES;
4737
4738   return GNUNET_YES;
4739 }
4740
4741 /******************************************************************************/
4742 /****************      MESH NETWORK HANDLER HELPERS     ***********************/
4743 /******************************************************************************/
4744
4745 /**
4746  * Free a transmission that was already queued with all resources
4747  * associated to the request.
4748  *
4749  * @param queue Queue handler to cancel.
4750  * @param clear_cls Is it necessary to free associated cls?
4751  */
4752 static void
4753 queue_destroy (struct MeshPeerQueue *queue, int clear_cls)
4754 {
4755   struct MeshPeer *peer;
4756   struct MeshFlowControl *fc;
4757   int fwd;
4758
4759   fwd = queue->fwd;
4760   peer = queue->peer;
4761   GNUNET_assert (NULL != queue->c);
4762   fc = fwd ? &queue->c->fwd_fc : &queue->c->bck_fc;
4763
4764   if (GNUNET_YES == clear_cls)
4765   {
4766     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   queue destroy type %s\n",
4767                 GNUNET_MESH_DEBUG_M2S (queue->type));
4768     switch (queue->type)
4769     {
4770       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
4771       case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY:
4772         GNUNET_log (GNUNET_ERROR_TYPE_INFO, "destroying a DESTROY message\n");
4773         GNUNET_break (GNUNET_YES == queue->c->destroy);
4774         /* fall through */
4775       case GNUNET_MESSAGE_TYPE_MESH_FWD:
4776       case GNUNET_MESSAGE_TYPE_MESH_BCK:
4777       case GNUNET_MESSAGE_TYPE_MESH_ACK:
4778       case GNUNET_MESSAGE_TYPE_MESH_POLL:
4779       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
4780       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
4781       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
4782         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   prebuilt message\n");;
4783         GNUNET_free_non_null (queue->cls);
4784         break;
4785
4786       default:
4787         GNUNET_break (0);
4788         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "   type %s unknown!\n",
4789                     GNUNET_MESH_DEBUG_M2S (queue->type));
4790     }
4791
4792   }
4793   GNUNET_CONTAINER_DLL_remove (peer->queue_head, peer->queue_tail, queue);
4794
4795   if (queue->type != GNUNET_MESSAGE_TYPE_MESH_ACK &&
4796       queue->type != GNUNET_MESSAGE_TYPE_MESH_POLL)
4797   {
4798     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Q_N- %p %u, \n", fc, fc->queue_n);
4799     fc->queue_n--;
4800     peer->queue_n--;
4801   }
4802   if (NULL != queue->c)
4803   {
4804     queue->c->pending_messages--;
4805     if (NULL != queue->c->t)
4806     {
4807       queue->c->t->pending_messages--;
4808     }
4809   }
4810
4811   GNUNET_free (queue);
4812 }
4813
4814
4815 static size_t
4816 queue_send (void *cls, size_t size, void *buf)
4817 {
4818   struct MeshPeer *peer = cls;
4819   struct MeshFlowControl *fc;
4820   struct MeshConnection *c;
4821   struct GNUNET_MessageHeader *msg;
4822   struct MeshPeerQueue *queue;
4823   struct MeshTunnel2 *t;
4824   struct MeshChannel *ch;
4825   const struct GNUNET_PeerIdentity *dst_id;
4826   size_t data_size;
4827   uint32_t pid;
4828   uint16_t type;
4829   int fwd;
4830
4831   peer->core_transmit = NULL;
4832   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Queue send (max %u)\n", size);
4833
4834   if (NULL == buf || 0 == size)
4835   {
4836     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Buffer size 0.\n");
4837     return 0;
4838   }
4839
4840   /* Initialize */
4841   queue = peer_get_first_message (peer);
4842   if (NULL == queue)
4843   {
4844     GNUNET_break (0); /* Core tmt_rdy should've been canceled */
4845     return 0;
4846   }
4847   c = queue->c;
4848   fwd = queue->fwd;
4849   fc = fwd ? &c->fwd_fc : &c->bck_fc;
4850
4851
4852   dst_id = GNUNET_PEER_resolve2 (peer->id);
4853   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   towards %s\n", GNUNET_i2s (dst_id));
4854   /* Check if buffer size is enough for the message */
4855   if (queue->size > size)
4856   {
4857       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   not enough room, reissue\n");
4858       peer->core_transmit =
4859           GNUNET_CORE_notify_transmit_ready (core_handle,
4860                                              GNUNET_NO,
4861                                              0,
4862                                              GNUNET_TIME_UNIT_FOREVER_REL,
4863                                              dst_id,
4864                                              queue->size,
4865                                              &queue_send,
4866                                              peer);
4867       return 0;
4868   }
4869   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   size %u ok\n", queue->size);
4870
4871   t = (NULL != c) ? c->t : NULL;
4872   type = 0;
4873
4874   /* Fill buf */
4875   switch (queue->type)
4876   {
4877     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
4878     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
4879     case GNUNET_MESSAGE_TYPE_MESH_FWD:
4880     case GNUNET_MESSAGE_TYPE_MESH_BCK:
4881     case GNUNET_MESSAGE_TYPE_MESH_ACK:
4882     case GNUNET_MESSAGE_TYPE_MESH_POLL:
4883       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4884                   "*   raw: %s\n",
4885                   GNUNET_MESH_DEBUG_M2S (queue->type));
4886       data_size = send_core_data_raw (queue->cls, size, buf);
4887       msg = (struct GNUNET_MessageHeader *) buf;
4888       type = ntohs (msg->type);
4889       break;
4890     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
4891       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path create\n");
4892       if (connection_is_origin (c, GNUNET_YES))
4893         data_size = send_core_connection_create (queue->c, size, buf);
4894       else
4895         data_size = send_core_data_raw (queue->cls, size, buf);
4896       break;
4897     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
4898       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path ack\n");
4899       if (connection_is_origin (c, GNUNET_NO))
4900         data_size = send_core_connection_ack (queue->c, size, buf);
4901       else
4902         data_size = send_core_data_raw (queue->cls, size, buf);
4903       break;
4904     case GNUNET_MESSAGE_TYPE_MESH_DATA:
4905     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
4906     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
4907       /* This should be encapsulted */
4908       GNUNET_break (0);
4909       break;
4910     default:
4911       GNUNET_break (0);
4912       GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "*   type unknown: %u\n",
4913                   queue->type);
4914       data_size = 0;
4915   }
4916
4917   if (0 < drop_percent &&
4918       GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent)
4919   {
4920     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4921                 "Dropping message of type %s\n",
4922                 GNUNET_MESH_DEBUG_M2S (queue->type));
4923     data_size = 0;
4924   }
4925
4926   /* Free queue, but cls was freed by send_core_* */
4927   ch = queue->ch;
4928   queue_destroy (queue, GNUNET_NO);
4929
4930   /* Send ACK if needed, after accounting for sent ID in fc->queue_n */
4931   switch (type)
4932   {
4933     case GNUNET_MESSAGE_TYPE_MESH_FWD:
4934     case GNUNET_MESSAGE_TYPE_MESH_BCK:
4935       pid = ntohl ( ((struct GNUNET_MESH_Encrypted *) buf)->pid );
4936       fc->last_pid_sent = pid;
4937       send_ack (c, ch, fwd);
4938       break;
4939     default:
4940       break;
4941   }
4942
4943   /* If more data in queue, send next */
4944   queue = peer_get_first_message (peer);
4945   if (NULL != queue)
4946   {
4947     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   more data!\n");
4948     if (NULL == peer->core_transmit) {
4949       peer->core_transmit =
4950           GNUNET_CORE_notify_transmit_ready(core_handle,
4951                                             0,
4952                                             0,
4953                                             GNUNET_TIME_UNIT_FOREVER_REL,
4954                                             dst_id,
4955                                             queue->size,
4956                                             &queue_send,
4957                                             peer);
4958     }
4959     else
4960     {
4961       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4962                   "*   tmt rdy called somewhere else\n");
4963     }
4964     if (GNUNET_SCHEDULER_NO_TASK == fc->poll_task)
4965     {
4966       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   starting poll timeout\n");
4967       fc->poll_task =
4968           GNUNET_SCHEDULER_add_delayed (fc->poll_time, &connection_poll, fc);
4969     }
4970   }
4971   else
4972   {
4973     if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
4974     {
4975       GNUNET_SCHEDULER_cancel (fc->poll_task);
4976       fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
4977     }
4978   }
4979   if (NULL != c)
4980   {
4981     c->pending_messages--;
4982     if (GNUNET_YES == c->destroy && 0 == c->pending_messages)
4983     {
4984       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  destroying connection!\n");
4985       connection_destroy (c);
4986     }
4987   }
4988
4989   if (NULL != t)
4990   {
4991     t->pending_messages--;
4992     if (GNUNET_YES == t->destroy && 0 == t->pending_messages)
4993     {
4994 //       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  destroying tunnel!\n");
4995       tunnel_destroy (t);
4996     }
4997   }
4998   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  Return %d\n", data_size);
4999   return data_size;
5000 }
5001
5002
5003 static void
5004 queue_add (void *cls, uint16_t type, size_t size,
5005            struct MeshConnection *c,
5006            struct MeshChannel *ch,
5007            int fwd)
5008 {
5009   struct MeshPeerQueue *queue;
5010   struct MeshFlowControl *fc;
5011   struct MeshPeer *peer;
5012   int priority;
5013
5014   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5015               "queue add %s %s (%u) on c %p, ch %p\n",
5016               fwd ? "FWD" : "BCK",  GNUNET_MESH_DEBUG_M2S (type), size, c, ch);
5017   GNUNET_assert (NULL != c);
5018
5019   fc   = fwd ? &c->fwd_fc : &c->bck_fc;
5020   peer = fwd ? connection_get_next_hop (c) : connection_get_prev_hop (c);
5021
5022   if (NULL == fc)
5023   {
5024     GNUNET_break (0);
5025     return;
5026   }
5027
5028   priority = 0;
5029
5030   if (GNUNET_MESSAGE_TYPE_MESH_POLL == type ||
5031       GNUNET_MESSAGE_TYPE_MESH_ACK == type)
5032   {
5033     priority = 100;
5034   }
5035
5036   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "priority %d\n", priority);
5037   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "fc %p\n", fc);
5038   if (fc->queue_n >= fc->queue_max && 0 == priority)
5039   {
5040     GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)",
5041                               1, GNUNET_NO);
5042     GNUNET_break (0);
5043     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5044                 "queue full: %u/%u\n",
5045                 fc->queue_n, fc->queue_max);
5046     return; /* Drop this message */
5047   }
5048
5049   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "last pid %u\n", fc->last_pid_sent);
5050   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "     ack %u\n", fc->last_ack_recv);
5051   if (GMC_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv) &&
5052       GNUNET_SCHEDULER_NO_TASK == fc->poll_task &&
5053       GNUNET_MESSAGE_TYPE_MESH_POLL != type)
5054   {
5055     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5056                 "no buffer space (%u > %u): starting poll\n",
5057                 fc->last_pid_sent + 1, fc->last_ack_recv);
5058     fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
5059                                                   &connection_poll,
5060                                                   fc);
5061   }
5062   queue = GNUNET_malloc (sizeof (struct MeshPeerQueue));
5063   queue->cls = cls;
5064   queue->type = type;
5065   queue->size = size;
5066   queue->peer = peer;
5067   queue->c = c;
5068   queue->ch = ch;
5069   queue->fwd = fwd;
5070   if (100 <= priority)
5071   {
5072     GNUNET_CONTAINER_DLL_insert (peer->queue_head, peer->queue_tail, queue);
5073   }
5074   else
5075   {
5076     GNUNET_CONTAINER_DLL_insert_tail (peer->queue_head, peer->queue_tail, queue);
5077     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Q_N+ %p %u, \n", fc, fc->queue_n);
5078     fc->queue_n++;
5079     peer->queue_n++;
5080   }
5081
5082   if (NULL == peer->core_transmit)
5083   {
5084     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5085                 "calling core tmt rdy towards %s for %u bytes\n",
5086                 peer2s (peer), size);
5087     peer->core_transmit =
5088         GNUNET_CORE_notify_transmit_ready (core_handle,
5089                                            0,
5090                                            0,
5091                                            GNUNET_TIME_UNIT_FOREVER_REL,
5092                                            GNUNET_PEER_resolve2 (peer->id),
5093                                            size,
5094                                            &queue_send,
5095                                            peer);
5096   }
5097   else
5098   {
5099     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5100                 "core tmt rdy towards %s already called\n",
5101                 peer2s (peer));
5102
5103   }
5104   c->pending_messages++;
5105   if (NULL != c->t)
5106     c->t->pending_messages++;
5107 }
5108
5109
5110 /******************************************************************************/
5111 /********************      MESH NETWORK HANDLERS     **************************/
5112 /******************************************************************************/
5113
5114
5115 /**
5116  * Generic handler for mesh network payload traffic.
5117  *
5118  * @param t Tunnel on which we got this message.
5119  * @param message Unencryted data message.
5120  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5121  */
5122 static void
5123 handle_data (struct MeshTunnel2 *t, const struct GNUNET_MESH_Data *msg, int fwd)
5124 {
5125   struct MeshChannelReliability *rel;
5126   struct MeshChannel *ch;
5127   struct MeshClient *c;
5128   uint32_t mid;
5129   uint16_t type;
5130   size_t size;
5131
5132   /* Check size */
5133   size = ntohs (msg->header.size);
5134   if (size <
5135       sizeof (struct GNUNET_MESH_Data) +
5136       sizeof (struct GNUNET_MessageHeader))
5137   {
5138     GNUNET_break (0);
5139     return;
5140   }
5141   type = ntohs (msg->header.type);
5142   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a %s message\n",
5143               GNUNET_MESH_DEBUG_M2S (type));
5144   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " payload of type %s\n",
5145               GNUNET_MESH_DEBUG_M2S (ntohs (msg[1].header.type)));
5146
5147   /* Check channel */
5148   ch = channel_get (t, ntohl (msg->chid));
5149   if (NULL == ch)
5150   {
5151     GNUNET_STATISTICS_update (stats, "# data on unknown channel", 1, GNUNET_NO);
5152     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "WARNING channel unknown\n");
5153     return;
5154   }
5155
5156   /*  Initialize FWD/BCK data */
5157   c        = fwd ? ch->dest     : ch->root;
5158   rel      = fwd ? ch->dest_rel : ch->root_rel;
5159
5160   if (NULL == c)
5161   {
5162     GNUNET_break (0);
5163     return;
5164   }
5165
5166   tunnel_change_state (t, MESH_TUNNEL_READY);
5167
5168   GNUNET_STATISTICS_update (stats, "# data received", 1, GNUNET_NO);
5169
5170   mid = ntohl (msg->mid);
5171   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " mid %u\n", mid);
5172
5173   if (GNUNET_NO == ch->reliable ||
5174       ( !GMC_is_pid_bigger (rel->mid_recv, mid) &&
5175         GMC_is_pid_bigger (rel->mid_recv + 64, mid) ) )
5176   {
5177     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! RECV %u\n", mid);
5178     if (GNUNET_YES == ch->reliable)
5179     {
5180       /* Is this the exact next expected messasge? */
5181       if (mid == rel->mid_recv)
5182       {
5183         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "as expected\n");
5184         rel->mid_recv++;
5185         channel_send_client_data (ch, msg, fwd);
5186       }
5187       else
5188       {
5189         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "save for later\n");
5190         channel_rel_add_buffered_data (msg, rel);
5191       }
5192     }
5193     else
5194     {
5195       /* Tunnel is unreliable: send to clients directly */
5196       /* FIXME: accept Out Of Order traffic */
5197       rel->mid_recv = mid + 1;
5198       channel_send_client_data (ch, msg, fwd);
5199     }
5200   }
5201   else
5202   {
5203     GNUNET_break_op (0);
5204     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5205                 " MID %u not expected (%u - %u), dropping!\n",
5206                 mid, rel->mid_recv, rel->mid_recv + 64);
5207   }
5208
5209   channel_send_data_ack (ch, fwd);
5210 }
5211
5212 /**
5213  * Handler for mesh network traffic end-to-end ACKs.
5214  *
5215  * @param t Tunnel on which we got this message.
5216  * @param message Data message.
5217  * @param fwd Is this a fwd ACK? (dest->orig)
5218  */
5219 static void
5220 handle_data_ack (struct MeshTunnel2 *t,
5221                  const struct GNUNET_MESH_DataACK *msg, int fwd)
5222 {
5223   struct MeshChannelReliability *rel;
5224   struct MeshReliableMessage *copy;
5225   struct MeshReliableMessage *next;
5226   struct MeshChannel *ch;
5227   uint32_t ack;
5228   uint16_t type;
5229   int work;
5230
5231   type = ntohs (msg->header.type);
5232   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a %s message!\n",
5233               GNUNET_MESH_DEBUG_M2S (type));
5234   ch = channel_get (t, ntohl (msg->chid));
5235   if (NULL == ch)
5236   {
5237     GNUNET_STATISTICS_update (stats, "# ack on unknown channel", 1, GNUNET_NO);
5238     return;
5239   }
5240   ack = ntohl (msg->mid);
5241   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! %s ACK %u\n",
5242               (GNUNET_YES == fwd) ? "FWD" : "BCK", ack);
5243
5244   if (GNUNET_YES == fwd)
5245   {
5246     rel = ch->root_rel;
5247   }
5248   else
5249   {
5250     rel = ch->dest_rel;
5251   }
5252   if (NULL == rel)
5253   {
5254     GNUNET_break (0);
5255     return;
5256   }
5257
5258   for (work = GNUNET_NO, copy = rel->head_sent; copy != NULL; copy = next)
5259   {
5260     if (GMC_is_pid_bigger (copy->mid, ack))
5261     {
5262       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  head %u, out!\n", copy->mid);
5263       channel_rel_free_sent (rel, msg);
5264       break;
5265     }
5266     work = GNUNET_YES;
5267     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  id %u\n", copy->mid);
5268     next = copy->next;
5269     rel_message_free (copy);
5270   }
5271   /* ACK client if needed */
5272 //   channel_send_ack (t, type, GNUNET_MESSAGE_TYPE_MESH_UNICAST_ACK == type);
5273
5274   /* If some message was free'd, update the retransmission delay*/
5275   if (GNUNET_YES == work)
5276   {
5277     if (GNUNET_SCHEDULER_NO_TASK != rel->retry_task)
5278     {
5279       GNUNET_SCHEDULER_cancel (rel->retry_task);
5280       if (NULL == rel->head_sent)
5281       {
5282         rel->retry_task = GNUNET_SCHEDULER_NO_TASK;
5283       }
5284       else
5285       {
5286         struct GNUNET_TIME_Absolute new_target;
5287         struct GNUNET_TIME_Relative delay;
5288
5289         delay = GNUNET_TIME_relative_multiply (rel->retry_timer,
5290                                                MESH_RETRANSMIT_MARGIN);
5291         new_target = GNUNET_TIME_absolute_add (rel->head_sent->timestamp,
5292                                                delay);
5293         delay = GNUNET_TIME_absolute_get_remaining (new_target);
5294         rel->retry_task =
5295             GNUNET_SCHEDULER_add_delayed (delay,
5296                                           &channel_retransmit_message,
5297                                           rel);
5298       }
5299     }
5300     else
5301       GNUNET_break (0);
5302   }
5303 }
5304
5305
5306 /**
5307  * Core handler for connection creation.
5308  *
5309  * @param cls Closure (unused).
5310  * @param peer Sender (neighbor).
5311  * @param message Message.
5312  *
5313  * @return GNUNET_OK to keep the connection open,
5314  *         GNUNET_SYSERR to close it (signal serious error)
5315  */
5316 static int
5317 handle_mesh_connection_create (void *cls,
5318                                const struct GNUNET_PeerIdentity *peer,
5319                                const struct GNUNET_MessageHeader *message)
5320 {
5321   struct GNUNET_MESH_ConnectionCreate *msg;
5322   struct GNUNET_PeerIdentity *id;
5323   struct GNUNET_HashCode *cid;
5324   struct MeshPeerPath *path;
5325   struct MeshPeer *dest_peer;
5326   struct MeshPeer *orig_peer;
5327   struct MeshConnection *c;
5328   unsigned int own_pos;
5329   uint16_t size;
5330   uint16_t i;
5331
5332   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5333   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received a connection create msg\n");
5334
5335   /* Check size */
5336   size = ntohs (message->size);
5337   if (size < sizeof (struct GNUNET_MESH_ConnectionCreate))
5338   {
5339     GNUNET_break_op (0);
5340     return GNUNET_OK;
5341   }
5342
5343   /* Calculate hops */
5344   size -= sizeof (struct GNUNET_MESH_ConnectionCreate);
5345   if (size % sizeof (struct GNUNET_PeerIdentity))
5346   {
5347     GNUNET_break_op (0);
5348     return GNUNET_OK;
5349   }
5350   size /= sizeof (struct GNUNET_PeerIdentity);
5351   if (1 > size)
5352   {
5353     GNUNET_break_op (0);
5354     return GNUNET_OK;
5355   }
5356   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
5357
5358   /* Get parameters */
5359   msg = (struct GNUNET_MESH_ConnectionCreate *) message;
5360   cid = &msg->cid;
5361   id = (struct GNUNET_PeerIdentity *) &msg[1];
5362   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5363               "    connection %s (%s).\n",
5364               GNUNET_h2s (cid), GNUNET_i2s (id));
5365
5366   /* Create connection */
5367   c = connection_get (cid);
5368   if (NULL == c)
5369   {
5370     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating connection\n");
5371     c = connection_new (cid);
5372     if (NULL == c)
5373       return GNUNET_OK;  connection_reset_timeout (c, GNUNET_YES);
5374     tunnel_change_state (c->t,  MESH_TUNNEL_WAITING);
5375
5376     /* Create path */
5377     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
5378     path = path_new (size);
5379     own_pos = 0;
5380     for (i = 0; i < size; i++)
5381     {
5382       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ... adding %s\n",
5383                   GNUNET_i2s (&id[i]));
5384       path->peers[i] = GNUNET_PEER_intern (&id[i]);
5385       if (path->peers[i] == myid)
5386         own_pos = i;
5387     }
5388     if (own_pos == 0 && path->peers[own_pos] != myid)
5389     {
5390       /* create path: self not found in path through self */
5391       GNUNET_break_op (0);
5392       path_destroy (path);
5393       connection_destroy (c);
5394       return GNUNET_OK;
5395     }
5396     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
5397     path_add_to_peers (path, GNUNET_NO);
5398     c->path = path_duplicate (path);
5399     c->own_pos = own_pos;
5400   }
5401   else
5402   {
5403     path = NULL;
5404   }
5405
5406   /* Remember peers */
5407   dest_peer = peer_get (&id[size - 1]);
5408   orig_peer = peer_get (&id[0]);
5409
5410   /* Is it a connection to us? */
5411   if (c->own_pos == size - 1)
5412   {
5413     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
5414     peer_add_path_to_origin (orig_peer, path, GNUNET_YES);
5415
5416     if (NULL == orig_peer->tunnel)
5417       orig_peer->tunnel = tunnel_new ();
5418     tunnel_add_connection (orig_peer->tunnel, c);
5419
5420     send_connection_ack (c, GNUNET_NO);
5421
5422     /* Keep tunnel alive in direction dest->owner*/
5423     connection_reset_timeout (c, GNUNET_NO); 
5424   }
5425   else
5426   {
5427     /* It's for somebody else! Retransmit. */
5428     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
5429     peer_add_path (dest_peer, path_duplicate (path), GNUNET_NO);
5430     peer_add_path_to_origin (orig_peer, path, GNUNET_NO);
5431     send_prebuilt_message_connection (message, c, NULL, GNUNET_YES);
5432   }
5433   return GNUNET_OK;
5434 }
5435
5436
5437 /**
5438  * Core handler for path ACKs
5439  *
5440  * @param cls closure
5441  * @param message message
5442  * @param peer peer identity this notification is about
5443  *
5444  * @return GNUNET_OK to keep the connection open,
5445  *         GNUNET_SYSERR to close it (signal serious error)
5446  */
5447 static int
5448 handle_mesh_connection_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
5449                             const struct GNUNET_MessageHeader *message)
5450 {
5451   struct GNUNET_MESH_ConnectionACK *msg;
5452   struct MeshPeerPath *p;
5453   struct MeshConnection *c;
5454
5455   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5456   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received a connection ACK msg\n");
5457   msg = (struct GNUNET_MESH_ConnectionACK *) message;
5458   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n",
5459               GNUNET_h2s (&msg->cid));
5460   c = connection_get (&msg->cid);
5461   if (NULL == c)
5462   {
5463     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
5464                               1, GNUNET_NO);
5465     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  don't know the connection!\n");
5466     return GNUNET_OK;
5467   }
5468
5469   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n",
5470               GNUNET_i2s (peer));
5471
5472   /* Add path to peers? */
5473   p = c->path;
5474   if (NULL != p)
5475   {
5476     path_add_to_peers (p, GNUNET_YES);
5477   }
5478   else
5479   {
5480     GNUNET_break (0);
5481   }
5482   connection_change_state (c, MESH_CONNECTION_READY);
5483   connection_reset_timeout (c, GNUNET_NO);
5484   if (MESH_TUNNEL_READY != c->t->state)
5485     tunnel_change_state (c->t, MESH_TUNNEL_READY);
5486   tunnel_send_queued_data (c->t, GNUNET_YES);
5487
5488   /* Message for us? */
5489   if (connection_is_terminal (c, GNUNET_NO))
5490   {
5491     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Connection ACK for us!\n");
5492     if (3 <= tunnel_count_connections (c->t) && NULL != c->t->peer->dhtget)
5493     {
5494       GNUNET_DHT_get_stop (c->t->peer->dhtget);
5495       c->t->peer->dhtget = NULL;
5496     }
5497     //connection_send_ack (c, GNUNET_NO); /* FIXME */
5498     return GNUNET_OK;
5499   }
5500
5501   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
5502   send_prebuilt_message_connection (message, c, NULL, GNUNET_NO);
5503   return GNUNET_OK;
5504 }
5505
5506
5507 /**
5508  * Core handler for notifications of broken paths
5509  *
5510  * @param cls Closure (unused).
5511  * @param peer Peer identity of sending neighbor.
5512  * @param message Message.
5513  *
5514  * @return GNUNET_OK to keep the connection open,
5515  *         GNUNET_SYSERR to close it (signal serious error)
5516  */
5517 static int
5518 handle_mesh_connection_broken (void *cls, const struct GNUNET_PeerIdentity *peer,
5519                                const struct GNUNET_MessageHeader *message)
5520 {
5521   struct GNUNET_MESH_ConnectionBroken *msg;
5522   struct MeshConnection *c;
5523
5524   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5525               "Received a CONNECTION BROKEN msg from %s\n", GNUNET_i2s (peer));
5526   msg = (struct GNUNET_MESH_ConnectionBroken *) message;
5527   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
5528               GNUNET_i2s (&msg->peer1));
5529   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
5530               GNUNET_i2s (&msg->peer2));
5531   c = connection_get (&msg->cid);
5532   if (NULL == c)
5533   {
5534     GNUNET_break_op (0);
5535     return GNUNET_OK;
5536   }
5537   tunnel_notify_connection_broken (c->t, GNUNET_PEER_search (&msg->peer1),
5538                                    GNUNET_PEER_search (&msg->peer2));
5539   return GNUNET_OK;
5540
5541 }
5542
5543
5544 /**
5545  * Core handler for tunnel destruction
5546  *
5547  * @param cls Closure (unused).
5548  * @param peer Peer identity of sending neighbor.
5549  * @param message Message.
5550  *
5551  * @return GNUNET_OK to keep the connection open,
5552  *         GNUNET_SYSERR to close it (signal serious error)
5553  */
5554 static int
5555 handle_mesh_connection_destroy (void *cls,
5556                                 const struct GNUNET_PeerIdentity *peer,
5557                                 const struct GNUNET_MessageHeader *message)
5558 {
5559   struct GNUNET_MESH_ConnectionDestroy *msg;
5560   struct MeshConnection *c;
5561   GNUNET_PEER_Id id;
5562   int fwd;
5563
5564   msg = (struct GNUNET_MESH_ConnectionDestroy *) message;
5565   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5566               "Got a CONNECTION DESTROY message from %s\n",
5567               GNUNET_i2s (peer));
5568   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5569               "  for connection %s\n",
5570               GNUNET_h2s (&msg->cid));
5571   c = connection_get (&msg->cid);
5572   if (NULL == c)
5573   {
5574     /* Probably already got the message from another path,
5575      * destroyed the tunnel and retransmitted to children.
5576      * Safe to ignore.
5577      */
5578     GNUNET_STATISTICS_update (stats, "# control on unknown tunnel",
5579                               1, GNUNET_NO);
5580     return GNUNET_OK;
5581   }
5582   id = GNUNET_PEER_search (peer);
5583   if (id == connection_get_prev_hop (c)->id)
5584     fwd = GNUNET_YES;
5585   else if (id == connection_get_next_hop (c)->id)
5586     fwd = GNUNET_NO;
5587   else
5588   {
5589     GNUNET_break_op (0);
5590     return GNUNET_OK;
5591   }
5592   send_prebuilt_message_connection (message, c, NULL, fwd);
5593   c->destroy = GNUNET_YES;
5594
5595   return GNUNET_OK;
5596 }
5597
5598
5599 /**
5600  * Handler for channel create messages.
5601  *
5602  * @param t Tunnel this channel is to be created in.
5603  * @param msg Message.
5604  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5605  */
5606 static void
5607 handle_channel_create (struct MeshTunnel2 *t,
5608                        struct GNUNET_MESH_ChannelCreate *msg,
5609                        int fwd)
5610 {
5611   MESH_ChannelNumber chid;
5612   struct MeshChannel *ch;
5613   struct MeshClient *c;
5614   uint32_t port;
5615
5616   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received Channel Create\n");
5617   /* Check message size */
5618   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelCreate))
5619   {
5620     GNUNET_break_op (0);
5621     return;
5622   }
5623
5624   /* Check if channel exists */
5625   chid = ntohl (msg->chid);
5626   ch = channel_get (t, chid);
5627   if (NULL != ch)
5628   {
5629     /* Probably a retransmission, safe to ignore */
5630     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   already exists...\n");
5631     if (NULL != ch->dest)
5632     {
5633       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   duplicate CC!!\n");
5634       channel_send_ack (ch, !fwd);
5635       return;
5636     }
5637   }
5638   else
5639   {
5640     /* Create channel */
5641     ch = channel_new (t, NULL, 0);
5642     channel_set_options (ch, ntohl (msg->opt));
5643   }
5644
5645   /* Find a destination client */
5646   port = ntohl (msg->port);
5647   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   port %u\n", port);
5648   c = GNUNET_CONTAINER_multihashmap32_get (ports, port);
5649   if (NULL == c)
5650   {
5651     /* TODO send reject */
5652     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  no client has port registered\n");
5653     /* TODO free ch */
5654     return;
5655   }
5656
5657   channel_add_client (ch, c);
5658   if (GNUNET_YES == ch->reliable)
5659     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! Reliable\n");
5660
5661   send_local_channel_create (ch);
5662   channel_send_ack (ch, !fwd);
5663   send_local_ack (ch, !fwd);
5664 }
5665
5666
5667 /**
5668  * Handler for channel ack messages.
5669  *
5670  * @param t Tunnel this channel is to be created in.
5671  * @param msg Message.
5672  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5673  */
5674 static void
5675 handle_channel_ack (struct MeshTunnel2 *t,
5676                     struct GNUNET_MESH_ChannelManage *msg,
5677                     int fwd)
5678 {
5679   MESH_ChannelNumber chid;
5680   struct MeshChannel *ch;
5681
5682   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received Channel ACK\n");
5683   /* Check message size */
5684   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelManage))
5685   {
5686     GNUNET_break_op (0);
5687     return;
5688   }
5689
5690   /* Check if channel exists */
5691   chid = ntohl (msg->chid);
5692   ch = channel_get (t, chid);
5693   if (NULL == ch)
5694   {
5695     GNUNET_break_op (0);
5696     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   channel %u unknown!!\n", chid);
5697     return;
5698   }
5699
5700   channel_confirm (ch, !fwd);
5701 }
5702
5703
5704 /**
5705  * Handler for channel destroy messages.
5706  *
5707  * @param t Tunnel this channel is to be destroyed of.
5708  * @param msg Message.
5709  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5710  */
5711 static void
5712 handle_channel_destroy (struct MeshTunnel2 *t,
5713                         struct GNUNET_MESH_ChannelManage *msg,
5714                         int fwd)
5715 {
5716   MESH_ChannelNumber chid;
5717   struct MeshChannel *ch;
5718
5719   /* Check message size */
5720   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelManage))
5721   {
5722     GNUNET_break_op (0);
5723     return;
5724   }
5725
5726   /* Check if channel exists */
5727   chid = ntohl (msg->chid);
5728   ch = channel_get (t, chid);
5729   if (NULL == ch)
5730   {
5731     /* Probably a retransmission, safe to ignore */
5732     return;
5733   }
5734
5735   send_local_channel_destroy (ch, fwd);
5736   channel_destroy (ch);
5737 }
5738
5739
5740 static void
5741 handle_decrypted (struct MeshTunnel2 *t,
5742                   const struct GNUNET_MessageHeader *msgh,
5743                   int fwd)
5744 {
5745   switch (ntohs (msgh->type))
5746   {
5747     case GNUNET_MESSAGE_TYPE_MESH_DATA:
5748       /* Don't send hop ACK, wait for client to ACK */
5749       handle_data (t, (struct GNUNET_MESH_Data *) msgh, fwd);
5750       break;
5751
5752     case GNUNET_MESSAGE_TYPE_MESH_DATA_ACK:
5753       handle_data_ack (t, (struct GNUNET_MESH_DataACK *) msgh, fwd);
5754       break;
5755
5756     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
5757       handle_channel_create (t,
5758                              (struct GNUNET_MESH_ChannelCreate *) msgh,
5759                              fwd);
5760       break;
5761
5762     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_ACK:
5763       handle_channel_ack (t,
5764                           (struct GNUNET_MESH_ChannelManage *) msgh,
5765                           fwd);
5766       break;
5767
5768     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
5769       handle_channel_destroy (t,
5770                               (struct GNUNET_MESH_ChannelManage *) msgh,
5771                               fwd);
5772       break;
5773
5774     default:
5775       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5776                   "end-to-end message not known (%u)\n",
5777                   ntohs (msgh->type));
5778   }
5779 }
5780
5781
5782 /**
5783  * Generic handler for mesh network encrypted traffic.
5784  *
5785  * @param peer Peer identity this notification is about.
5786  * @param message Encrypted message.
5787  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5788  *
5789  * @return GNUNET_OK to keep the connection open,
5790  *         GNUNET_SYSERR to close it (signal serious error)
5791  */
5792 static int
5793 handle_mesh_encrypted (const struct GNUNET_PeerIdentity *peer,
5794                        const struct GNUNET_MESH_Encrypted *msg,
5795                        int fwd)
5796 {
5797   struct MeshConnection *c;
5798   struct MeshTunnel2 *t;
5799   struct MeshPeer *neighbor;
5800   struct MeshFlowControl *fc;
5801   uint32_t pid;
5802   uint32_t ttl;
5803   uint16_t type;
5804   size_t size;
5805
5806   /* Check size */
5807   size = ntohs (msg->header.size);
5808   if (size <
5809       sizeof (struct GNUNET_MESH_Encrypted) +
5810       sizeof (struct GNUNET_MessageHeader))
5811   {
5812     GNUNET_break_op (0);
5813     return GNUNET_OK;
5814   }
5815   type = ntohs (msg->header.type);
5816   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5817   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
5818               GNUNET_MESH_DEBUG_M2S (type), GNUNET_i2s (peer));
5819
5820   /* Check connection */
5821   c = connection_get (&msg->cid);
5822   if (NULL == c)
5823   {
5824     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
5825     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "WARNING connection unknown\n");
5826     return GNUNET_OK;
5827   }
5828   t = c->t;
5829   fc = fwd ? &c->fwd_fc : &c->bck_fc;
5830
5831   /* Check if origin is as expected */
5832   neighbor = connection_get_hop (c, fwd);
5833   if (peer_get (peer)->id != neighbor->id)
5834   {
5835     GNUNET_break_op (0);
5836     return GNUNET_OK;
5837   }
5838
5839   /* Check PID */
5840   pid = ntohl (msg->pid);
5841   if (GMC_is_pid_bigger (pid, fc->last_ack_sent))
5842   {
5843     GNUNET_STATISTICS_update (stats, "# unsolicited message", 1, GNUNET_NO);
5844     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5845                 "WARNING Received PID %u, (prev %u), ACK %u\n",
5846                 pid, fc->last_pid_recv, fc->last_ack_sent);
5847     return GNUNET_OK;
5848   }
5849   if (GNUNET_NO == GMC_is_pid_bigger (pid, fc->last_pid_recv))
5850   {
5851     GNUNET_STATISTICS_update (stats, "# duplicate PID", 1, GNUNET_NO);
5852     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5853                 " Pid %u not expected (%u+), dropping!\n",
5854                 pid, fc->last_pid_recv + 1);
5855     return GNUNET_OK;
5856   }
5857   if (MESH_CONNECTION_SENT == c->state)
5858     connection_change_state (c, MESH_CONNECTION_READY);
5859   connection_reset_timeout (c, fwd);
5860   fc->last_pid_recv = pid;
5861
5862   /* Is this message for us? */
5863   if (connection_is_terminal (c, fwd))
5864   {
5865     size_t dsize = size - sizeof (struct GNUNET_MESH_Encrypted);
5866     char cbuf[dsize];
5867     struct GNUNET_MessageHeader *msgh;
5868     unsigned int off;
5869
5870     /* TODO signature verification */
5871     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
5872     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
5873
5874     fc->last_pid_recv = pid;
5875     tunnel_decrypt (t, cbuf, &msg[1], dsize, msg->iv, fwd);
5876     off = 0;
5877     while (off < dsize)
5878     {
5879       msgh = (struct GNUNET_MessageHeader *) &cbuf[off];
5880       handle_decrypted (t, msgh, fwd);
5881       off += ntohs (msgh->size);
5882     }
5883     send_ack (c, NULL, fwd);
5884     return GNUNET_OK;
5885   }
5886
5887   /* Message not for us: forward to next hop */
5888   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
5889   ttl = ntohl (msg->ttl);
5890   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
5891   if (ttl == 0)
5892   {
5893     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
5894     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
5895     send_ack (c, NULL, fwd);
5896     return GNUNET_OK;
5897   }
5898   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
5899
5900   send_prebuilt_message_connection (&msg->header, c, NULL, fwd);
5901
5902   return GNUNET_OK;
5903 }
5904
5905
5906 /**
5907  * Core handler for mesh network traffic going orig->dest.
5908  *
5909  * @param cls Closure (unused).
5910  * @param message Message received.
5911  * @param peer Peer who sent the message.
5912  *
5913  * @return GNUNET_OK to keep the connection open,
5914  *         GNUNET_SYSERR to close it (signal serious error)
5915  */
5916 static int
5917 handle_mesh_fwd (void *cls, const struct GNUNET_PeerIdentity *peer,
5918                      const struct GNUNET_MessageHeader *message)
5919 {
5920   return handle_mesh_encrypted (peer,
5921                                 (struct GNUNET_MESH_Encrypted *)message,
5922                                 GNUNET_YES);
5923 }
5924
5925 /**
5926  * Core handler for mesh network traffic going dest->orig.
5927  *
5928  * @param cls Closure (unused).
5929  * @param message Message received.
5930  * @param peer Peer who sent the message.
5931  *
5932  * @return GNUNET_OK to keep the connection open,
5933  *         GNUNET_SYSERR to close it (signal serious error)
5934  */
5935 static int
5936 handle_mesh_bck (void *cls, const struct GNUNET_PeerIdentity *peer,
5937                      const struct GNUNET_MessageHeader *message)
5938 {
5939   return handle_mesh_encrypted (peer,
5940                                 (struct GNUNET_MESH_Encrypted *)message,
5941                                 GNUNET_NO);
5942 }
5943
5944
5945 /**
5946  * Core handler for mesh network traffic point-to-point acks.
5947  *
5948  * @param cls closure
5949  * @param message message
5950  * @param peer peer identity this notification is about
5951  *
5952  * @return GNUNET_OK to keep the connection open,
5953  *         GNUNET_SYSERR to close it (signal serious error)
5954  */
5955 static int
5956 handle_mesh_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
5957                  const struct GNUNET_MessageHeader *message)
5958 {
5959   struct GNUNET_MESH_ACK *msg;
5960   struct MeshConnection *c;
5961   struct MeshFlowControl *fc;
5962   GNUNET_PEER_Id id;
5963   uint32_t ack;
5964
5965   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5966   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
5967               GNUNET_i2s (peer));
5968   msg = (struct GNUNET_MESH_ACK *) message;
5969
5970   c = connection_get (&msg->cid);
5971
5972   if (NULL == c)
5973   {
5974     GNUNET_STATISTICS_update (stats, "# ack on unknown connection", 1,
5975                               GNUNET_NO);
5976     return GNUNET_OK;
5977   }
5978
5979   /* Is this a forward or backward ACK? */
5980   id = GNUNET_PEER_search (peer);
5981   if (connection_get_next_hop (c)->id == id)
5982   {
5983     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
5984     fc = &c->fwd_fc;
5985   }
5986   else if (connection_get_prev_hop (c)->id == id)
5987   {
5988     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
5989     fc = &c->bck_fc;
5990   }
5991   else
5992   {
5993     GNUNET_break_op (0);
5994     return GNUNET_OK;
5995   }
5996
5997   ack = ntohl (msg->ack);
5998   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u\n", ack);
5999
6000   /* Cancel polling if the ACK is bigger than before. */
6001   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task &&
6002       GMC_is_pid_bigger (ack, fc->last_ack_recv))
6003   {
6004     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Cancel poll\n");
6005     GNUNET_SCHEDULER_cancel (fc->poll_task);
6006     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
6007     fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
6008   }
6009
6010   fc->last_ack_recv = ack;
6011   connection_unlock_queue (c, fc == &c->fwd_fc);
6012
6013   return GNUNET_OK;
6014 }
6015
6016
6017 /**
6018  * Core handler for mesh network traffic point-to-point ack polls.
6019  *
6020  * @param cls closure
6021  * @param message message
6022  * @param peer peer identity this notification is about
6023  *
6024  * @return GNUNET_OK to keep the connection open,
6025  *         GNUNET_SYSERR to close it (signal serious error)
6026  */
6027 static int
6028 handle_mesh_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
6029                   const struct GNUNET_MessageHeader *message)
6030 {
6031   struct GNUNET_MESH_Poll *msg;
6032   struct MeshConnection *c;
6033   struct MeshFlowControl *fc;
6034   GNUNET_PEER_Id id;
6035   uint32_t pid;
6036   int fwd;
6037
6038   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
6039   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a POLL packet from %s!\n",
6040               GNUNET_i2s (peer));
6041
6042   msg = (struct GNUNET_MESH_Poll *) message;
6043
6044   c = connection_get (&msg->cid);
6045
6046   if (NULL == c)
6047   {
6048     GNUNET_STATISTICS_update (stats, "# poll on unknown connection", 1,
6049                               GNUNET_NO);
6050     GNUNET_break_op (0);
6051     return GNUNET_OK;
6052   }
6053
6054   /* Is this a forward or backward ACK?
6055    * Note: a poll should never be needed in a loopback case,
6056    * since there is no possiblility of packet loss there, so
6057    * this way of discerining FWD/BCK should not be a problem.
6058    */
6059   id = GNUNET_PEER_search (peer);
6060   if (connection_get_next_hop (c)->id == id)
6061   {
6062     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
6063     fc = &c->fwd_fc;
6064   }
6065   else if (connection_get_prev_hop (c)->id == id)
6066   {
6067     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
6068     fc = &c->bck_fc;
6069   }
6070   else
6071   {
6072     GNUNET_break_op (0);
6073     return GNUNET_OK;
6074   }
6075
6076   pid = ntohl (msg->pid);
6077   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  PID %u, OLD %u\n",
6078               pid, fc->last_pid_recv);
6079   fc->last_pid_recv = pid;
6080   fwd = fc == &c->fwd_fc;
6081   send_ack (c, NULL, fwd);
6082
6083   return GNUNET_OK;
6084 }
6085
6086
6087 /**
6088  * Core handler for mesh keepalives.
6089  *
6090  * @param cls closure
6091  * @param message message
6092  * @param peer peer identity this notification is about
6093  * @return GNUNET_OK to keep the connection open,
6094  *         GNUNET_SYSERR to close it (signal serious error)
6095  *
6096  * TODO: Check who we got this from, to validate route.
6097  */
6098 static int
6099 handle_mesh_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
6100                        const struct GNUNET_MessageHeader *message)
6101 {
6102   struct GNUNET_MESH_ConnectionKeepAlive *msg;
6103   struct MeshConnection *c;
6104   struct MeshPeer *neighbor;
6105   int fwd;
6106
6107   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) message;
6108   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
6109               GNUNET_i2s (peer));
6110
6111   c = connection_get (&msg->cid);
6112   if (NULL == c)
6113   {
6114     GNUNET_STATISTICS_update (stats, "# keepalive on unknown connection", 1,
6115                               GNUNET_NO);
6116     return GNUNET_OK;
6117   }
6118
6119   fwd = GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE == ntohs (message->type) ? 
6120         GNUNET_YES : GNUNET_NO;
6121
6122   /* Check if origin is as expected */
6123   neighbor = connection_get_hop (c, fwd);
6124   if (peer_get (peer)->id != neighbor->id)
6125   {
6126     GNUNET_break_op (0);
6127     return GNUNET_OK;
6128   }
6129
6130   connection_change_state (c, MESH_CONNECTION_READY);
6131   connection_reset_timeout (c, fwd);
6132
6133   if (connection_is_terminal (c, fwd))
6134     return GNUNET_OK;
6135
6136   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
6137   send_prebuilt_message_connection (message, c, NULL, fwd);
6138
6139   return GNUNET_OK;
6140 }
6141
6142
6143
6144 /**
6145  * Functions to handle messages from core
6146  */
6147 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
6148   {&handle_mesh_connection_create, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
6149     0},
6150   {&handle_mesh_connection_ack, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
6151     sizeof (struct GNUNET_MESH_ConnectionACK)},
6152   {&handle_mesh_connection_broken, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN,
6153     sizeof (struct GNUNET_MESH_ConnectionBroken)},
6154   {&handle_mesh_connection_destroy, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY,
6155     sizeof (struct GNUNET_MESH_ConnectionDestroy)},
6156   {&handle_mesh_keepalive, GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE,
6157     sizeof (struct GNUNET_MESH_ConnectionKeepAlive)},
6158   {&handle_mesh_keepalive, GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE,
6159     sizeof (struct GNUNET_MESH_ConnectionKeepAlive)},
6160   {&handle_mesh_ack, GNUNET_MESSAGE_TYPE_MESH_ACK,
6161     sizeof (struct GNUNET_MESH_ACK)},
6162   {&handle_mesh_poll, GNUNET_MESSAGE_TYPE_MESH_POLL,
6163     sizeof (struct GNUNET_MESH_Poll)},
6164   {&handle_mesh_fwd, GNUNET_MESSAGE_TYPE_MESH_FWD, 0},
6165   {&handle_mesh_bck, GNUNET_MESSAGE_TYPE_MESH_BCK, 0},
6166   {NULL, 0, 0}
6167 };
6168
6169
6170 /**
6171  * Function to process paths received for a new peer addition. The recorded
6172  * paths form the initial tunnel, which can be optimized later.
6173  * Called on each result obtained for the DHT search.
6174  *
6175  * @param cls closure
6176  * @param exp when will this value expire
6177  * @param key key of the result
6178  * @param get_path path of the get request
6179  * @param get_path_length lenght of get_path
6180  * @param put_path path of the put request
6181  * @param put_path_length length of the put_path
6182  * @param type type of the result
6183  * @param size number of bytes in data
6184  * @param data pointer to the result data
6185  */
6186 static void
6187 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
6188                     const struct GNUNET_HashCode * key,
6189                     const struct GNUNET_PeerIdentity *get_path,
6190                     unsigned int get_path_length,
6191                     const struct GNUNET_PeerIdentity *put_path,
6192                     unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
6193                     size_t size, const void *data)
6194 {
6195   struct MeshPeer *peer = cls;
6196   struct MeshPeerPath *p;
6197   struct MeshConnection *c;
6198   struct GNUNET_PeerIdentity pi;
6199   int i;
6200
6201   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got results from DHT!\n");
6202   GNUNET_PEER_resolve (peer->id, &pi);
6203   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  for %s\n", GNUNET_i2s (&pi));
6204
6205   p = path_build_from_dht (get_path, get_path_length,
6206                            put_path, put_path_length);
6207   path_add_to_peers (p, GNUNET_NO);
6208   path_destroy (p);
6209
6210   /* Count connections */
6211   for (c = peer->tunnel->connection_head, i = 0; NULL != c; c = c->next, i++);
6212
6213   /* If we already have 3 (or more (?!)) connections, it's enough */
6214   if (3 <= i)
6215     return;
6216
6217   if (peer->tunnel->state == MESH_TUNNEL_SEARCHING)
6218   {
6219     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n");
6220     peer_connect (peer);
6221   }
6222   return;
6223 }
6224
6225
6226 /******************************************************************************/
6227 /*********************       MESH LOCAL HANDLES      **************************/
6228 /******************************************************************************/
6229
6230
6231 /**
6232  * Handler for client connection.
6233  *
6234  * @param cls Closure (unused).
6235  * @param client Client handler.
6236  */
6237 static void
6238 handle_local_client_connect (void *cls, struct GNUNET_SERVER_Client *client)
6239 {
6240   struct MeshClient *c;
6241
6242   if (NULL == client)
6243     return;
6244   c = GNUNET_malloc (sizeof (struct MeshClient));
6245   c->handle = client;
6246   c->id = next_client_id++; /* overflow not important: just for debug */
6247   GNUNET_SERVER_client_keep (client);
6248   GNUNET_SERVER_client_set_user_context (client, c);
6249   GNUNET_CONTAINER_DLL_insert (clients_head, clients_tail, c);
6250 }
6251
6252
6253 /**
6254  * Handler for client disconnection
6255  *
6256  * @param cls closure
6257  * @param client identification of the client; NULL
6258  *        for the last call when the server is destroyed
6259  */
6260 static void
6261 handle_local_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
6262 {
6263   struct MeshClient *c;
6264
6265   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disconnected: %p\n", client);
6266   if (client == NULL)
6267   {
6268     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (SERVER DOWN)\n");
6269     return;
6270   }
6271
6272   c = client_get (client);
6273   if (NULL != c)
6274   {
6275     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "matching client found (%u, %p)\n",
6276                 c->id, c);
6277     GNUNET_SERVER_client_drop (c->handle);
6278     c->shutting_down = GNUNET_YES;
6279     if (NULL != c->own_channels)
6280     {
6281       GNUNET_CONTAINER_multihashmap32_iterate (c->own_channels,
6282                                                &channel_destroy_iterator, c);
6283       GNUNET_CONTAINER_multihashmap32_destroy (c->own_channels);
6284     }
6285
6286     if (NULL != c->incoming_channels)
6287     {
6288       GNUNET_CONTAINER_multihashmap32_iterate (c->incoming_channels,
6289                                                &channel_destroy_iterator, c);
6290       GNUNET_CONTAINER_multihashmap32_destroy (c->incoming_channels);
6291     }
6292
6293     if (NULL != c->ports)
6294     {
6295       GNUNET_CONTAINER_multihashmap32_iterate (c->ports,
6296                                                &client_release_ports, c);
6297       GNUNET_CONTAINER_multihashmap32_destroy (c->ports);
6298     }
6299     GNUNET_CONTAINER_DLL_remove (clients_head, clients_tail, c);
6300     GNUNET_STATISTICS_update (stats, "# clients", -1, GNUNET_NO);
6301     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client free (%p)\n", c);
6302     GNUNET_free (c);
6303   }
6304   else
6305   {
6306     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " context NULL!\n");
6307   }
6308   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "done!\n");
6309   return;
6310 }
6311
6312
6313 /**
6314  * Handler for new clients
6315  *
6316  * @param cls closure
6317  * @param client identification of the client
6318  * @param message the actual message, which includes messages the client wants
6319  */
6320 static void
6321 handle_local_new_client (void *cls, struct GNUNET_SERVER_Client *client,
6322                          const struct GNUNET_MessageHeader *message)
6323 {
6324   struct GNUNET_MESH_ClientConnect *cc_msg;
6325   struct MeshClient *c;
6326   unsigned int size;
6327   uint32_t *p;
6328   unsigned int i;
6329
6330   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client connected %p\n", client);
6331
6332   /* Check data sanity */
6333   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
6334   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
6335   if (0 != (size % sizeof (uint32_t)))
6336   {
6337     GNUNET_break (0);
6338     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6339     return;
6340   }
6341   size /= sizeof (uint32_t);
6342
6343   /* Initialize new client structure */
6344   c = GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
6345   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client id %u\n", c->id);
6346   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client has %u ports\n", size);
6347   if (size > 0)
6348   {
6349     uint32_t u32;
6350
6351     p = (uint32_t *) &cc_msg[1];
6352     c->ports = GNUNET_CONTAINER_multihashmap32_create (size);
6353     for (i = 0; i < size; i++)
6354     {
6355       u32 = ntohl (p[i]);
6356       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    port: %u\n", u32);
6357
6358       /* store in client's hashmap */
6359       GNUNET_CONTAINER_multihashmap32_put (c->ports, u32, c,
6360                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
6361       /* store in global hashmap */
6362       /* FIXME only allow one client to have the port open,
6363        *       have a backup hashmap with waiting clients */
6364       GNUNET_CONTAINER_multihashmap32_put (ports, u32, c,
6365                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
6366     }
6367   }
6368
6369   c->own_channels = GNUNET_CONTAINER_multihashmap32_create (32);
6370   c->incoming_channels = GNUNET_CONTAINER_multihashmap32_create (32);
6371   GNUNET_SERVER_notification_context_add (nc, client);
6372   GNUNET_STATISTICS_update (stats, "# clients", 1, GNUNET_NO);
6373
6374   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6375   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client processed\n");
6376 }
6377
6378
6379 /**
6380  * Handler for requests of new tunnels
6381  *
6382  * @param cls Closure.
6383  * @param client Identification of the client.
6384  * @param message The actual message.
6385  */
6386 static void
6387 handle_local_channel_create (void *cls, struct GNUNET_SERVER_Client *client,
6388                             const struct GNUNET_MessageHeader *message)
6389 {
6390   struct GNUNET_MESH_ChannelMessage *msg;
6391   struct MeshPeer *peer;
6392   struct MeshTunnel2 *t;
6393   struct MeshChannel *ch;
6394   struct MeshClient *c;
6395   MESH_ChannelNumber chid;
6396
6397   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new channel requested\n");
6398
6399   /* Sanity check for client registration */
6400   if (NULL == (c = client_get (client)))
6401   {
6402     GNUNET_break (0);
6403     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6404     return;
6405   }
6406   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6407
6408   /* Message size sanity check */
6409   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
6410   {
6411     GNUNET_break (0);
6412     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6413     return;
6414   }
6415
6416   msg = (struct GNUNET_MESH_ChannelMessage *) message;
6417   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  towards %s:%u\n",
6418               GNUNET_i2s (&msg->peer), ntohl (msg->port));
6419   chid = ntohl (msg->channel_id);
6420
6421   /* Sanity check for duplicate channel IDs */
6422   if (NULL != channel_get_by_local_id (c, chid))
6423   {
6424     GNUNET_break (0);
6425     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6426     return;
6427   }
6428
6429   peer = peer_get (&msg->peer);
6430   if (NULL == peer->tunnel)
6431   {
6432     peer->tunnel = tunnel_new ();
6433     peer->tunnel->peer = peer;
6434   }
6435   t = peer->tunnel;
6436
6437   /* Create channel */
6438   ch = channel_new (t, c, chid);
6439   if (NULL == ch)
6440   {
6441     GNUNET_break (0);
6442     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6443     return;
6444   }
6445   ch->port = ntohl (msg->port);
6446   channel_set_options (ch, ntohl (msg->opt));
6447
6448   /* In unreliable channels, we'll use the DLL to buffer data for the root */
6449   ch->root_rel = GNUNET_new (struct MeshChannelReliability);
6450   ch->root_rel->ch = ch;
6451   ch->root_rel->expected_delay = MESH_RETRANSMIT_TIME;
6452
6453   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CREATED CHANNEL %s[%x]:%u (%x)\n",
6454               peer2s (t->peer), ch->gid, ch->port, ch->lid_root);
6455   peer_connect (peer);
6456
6457   /* Send create channel */
6458   {
6459     struct GNUNET_MESH_ChannelCreate msgcc;
6460
6461     msgcc.header.size = htons (sizeof (msgcc));
6462     msgcc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE);
6463     msgcc.chid = htonl (ch->gid);
6464     msgcc.port = msg->port;
6465     msgcc.opt = msg->opt;
6466
6467     tunnel_queue_data (t, ch, &msgcc.header, GNUNET_YES);
6468   }
6469
6470   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6471   return;
6472 }
6473
6474
6475 /**
6476  * Handler for requests of deleting tunnels
6477  *
6478  * @param cls closure
6479  * @param client identification of the client
6480  * @param message the actual message
6481  */
6482 static void
6483 handle_local_channel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
6484                              const struct GNUNET_MessageHeader *message)
6485 {
6486   struct GNUNET_MESH_ChannelMessage *msg;
6487   struct MeshClient *c;
6488   struct MeshChannel *ch;
6489   struct MeshTunnel2 *t;
6490   MESH_ChannelNumber chid;
6491
6492   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
6493               "Got a DESTROY CHANNEL from client!\n");
6494
6495   /* Sanity check for client registration */
6496   if (NULL == (c = client_get (client)))
6497   {
6498     GNUNET_break (0);
6499     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6500     return;
6501   }
6502   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6503
6504   /* Message sanity check */
6505   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
6506   {
6507     GNUNET_break (0);
6508     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6509     return;
6510   }
6511
6512   msg = (struct GNUNET_MESH_ChannelMessage *) message;
6513
6514   /* Retrieve tunnel */
6515   chid = ntohl (msg->channel_id);
6516   ch = channel_get_by_local_id (c, chid);
6517   if (NULL == ch)
6518   {
6519     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "  channel %X not found\n", chid);
6520     GNUNET_break (0);
6521     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6522     return;
6523   }
6524
6525   /* Cleanup after the tunnel */
6526   client_delete_channel (c, ch);
6527   if (c == ch->dest && GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= chid)
6528   {
6529     ch->dest = NULL;
6530   }
6531   else if (c == ch->root && GNUNET_MESH_LOCAL_CHANNEL_ID_SERV > chid)
6532   {
6533     ch->root = NULL;
6534   }
6535   else 
6536   {
6537     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6538                 "  channel %X client %p (%p, %p)\n",
6539                 chid, c, ch->root, ch->dest);
6540     GNUNET_break (0);
6541   }
6542
6543   t = ch->t;
6544   channel_destroy (ch);
6545   tunnel_destroy_if_empty (t);
6546
6547   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6548   return;
6549 }
6550
6551
6552 /**
6553  * Handler for client traffic
6554  *
6555  * @param cls closure
6556  * @param client identification of the client
6557  * @param message the actual message
6558  */
6559 static void
6560 handle_local_data (void *cls, struct GNUNET_SERVER_Client *client,
6561                    const struct GNUNET_MessageHeader *message)
6562 {
6563   struct GNUNET_MESH_LocalData *msg;
6564   struct MeshClient *c;
6565   struct MeshChannel *ch;
6566   struct MeshChannelReliability *rel;
6567   MESH_ChannelNumber chid;
6568   size_t size;
6569   int fwd;
6570
6571   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
6572               "Got data from a client!\n");
6573
6574   /* Sanity check for client registration */
6575   if (NULL == (c = client_get (client)))
6576   {
6577     GNUNET_break (0);
6578     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6579     return;
6580   }
6581   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6582
6583   msg = (struct GNUNET_MESH_LocalData *) message;
6584
6585   /* Sanity check for message size */
6586   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_LocalData);
6587   if (size < sizeof (struct GNUNET_MessageHeader))
6588   {
6589     GNUNET_break (0);
6590     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6591     return;
6592   }
6593
6594   /* Channel exists? */
6595   chid = ntohl (msg->id);
6596   fwd = chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
6597   ch = channel_get_by_local_id (c, chid);
6598   if (NULL == ch)
6599   {
6600     GNUNET_break (0);
6601     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6602     return;
6603   }
6604
6605   /* Is the client in the channel? */
6606   if ( !( (fwd &&
6607            ch->root &&
6608            ch->root->handle == client)
6609          ||
6610           (!fwd &&
6611            ch->dest && 
6612            ch->dest->handle == client) ) )
6613   {
6614     GNUNET_break (0);
6615     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6616     return;
6617   }
6618
6619   rel = fwd ? ch->root_rel : ch->dest_rel;
6620   rel->client_ready = GNUNET_NO;
6621
6622   /* Ok, everything is correct, send the message. */
6623   {
6624     struct GNUNET_MESH_Data *payload;
6625     uint16_t p2p_size = sizeof(struct GNUNET_MESH_Data) + size;
6626     unsigned char cbuf[p2p_size];
6627
6628     payload = (struct GNUNET_MESH_Data *) cbuf;
6629     payload->mid = htonl (rel->mid_send);
6630     rel->mid_send++;
6631     memcpy (&payload[1], &msg[1], size);
6632     payload->header.size = htons (p2p_size);
6633     payload->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_DATA);
6634     payload->chid = htonl (ch->gid);
6635     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on channel...\n");
6636     send_prebuilt_message_channel (&payload->header, ch, fwd);
6637
6638     if (GNUNET_YES == ch->reliable)
6639       channel_save_copy (ch, &payload->header, fwd);
6640   }
6641   if (tunnel_get_buffer (ch->t, fwd) > 0)
6642     send_local_ack (ch, fwd);
6643   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "receive done OK\n");
6644   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6645
6646   return;
6647 }
6648
6649
6650 /**
6651  * Handler for client's ACKs for payload traffic.
6652  *
6653  * @param cls Closure (unused).
6654  * @param client Identification of the client.
6655  * @param message The actual message.
6656  */
6657 static void
6658 handle_local_ack (void *cls, struct GNUNET_SERVER_Client *client,
6659                   const struct GNUNET_MessageHeader *message)
6660 {
6661   struct GNUNET_MESH_LocalAck *msg;
6662   struct MeshChannelReliability *rel;
6663   struct MeshChannel *ch;
6664   struct MeshClient *c;
6665   MESH_ChannelNumber chid;
6666   int fwd;
6667
6668   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a local ACK\n");
6669
6670   /* Sanity check for client registration */
6671   if (NULL == (c = client_get (client)))
6672   {
6673     GNUNET_break (0);
6674     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6675     return;
6676   }
6677   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6678
6679   msg = (struct GNUNET_MESH_LocalAck *) message;
6680
6681   /* Channel exists? */
6682   chid = ntohl (msg->channel_id);
6683   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on channel %X\n", chid);
6684   ch = channel_get_by_local_id (c, chid);
6685   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   -- ch %p\n", ch);
6686   if (NULL == ch)
6687   {
6688     GNUNET_break (0);
6689     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Channel %X unknown.\n", chid);
6690     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "  for client %u.\n", c->id);
6691     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6692     return;
6693   }
6694
6695   /* If client is root, the ACK is going FWD, therefore this is "BCK". */
6696   /* If client is dest, the ACK is going BCK, therefore this is "FWD" */
6697   fwd = chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
6698   rel = fwd ? ch->dest_rel : ch->root_rel;
6699
6700   rel->client_ready = GNUNET_YES;
6701   channel_send_client_buffered_data (ch, c, fwd);
6702   send_ack (NULL, ch, fwd);
6703
6704   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6705
6706   return;
6707 }
6708
6709
6710 /**
6711  * Iterator over all tunnels to send a monitoring client info about each tunnel.
6712  *
6713  * @param cls Closure (client handle).
6714  * @param key Key (hashed tunnel ID, unused).
6715  * @param value Tunnel info.
6716  *
6717  * @return GNUNET_YES, to keep iterating.
6718  */
6719 // static int
6720 // monitor_all_tunnels_iterator (void *cls,
6721 //                               const struct GNUNET_HashCode * key,
6722 //                               void *value)
6723 // {
6724 //   struct GNUNET_SERVER_Client *client = cls;
6725 //   struct MeshChannel *ch = value;
6726 //   struct GNUNET_MESH_LocalMonitor *msg;
6727 // 
6728 //   msg = GNUNET_malloc (sizeof(struct GNUNET_MESH_LocalMonitor));
6729 //   msg->channel_id = htonl (ch->gid);
6730 //   msg->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
6731 //   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS);
6732 // 
6733 //   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6734 //               "*  sending info about tunnel %s\n",
6735 //               GNUNET_i2s (&msg->owner));
6736 // 
6737 //   GNUNET_SERVER_notification_context_unicast (nc, client,
6738 //                                               &msg->header, GNUNET_NO);
6739 //   return GNUNET_YES;
6740 // }
6741
6742
6743 /**
6744  * Handler for client's MONITOR request.
6745  *
6746  * @param cls Closure (unused).
6747  * @param client Identification of the client.
6748  * @param message The actual message.
6749  */
6750 static void
6751 handle_local_get_tunnels (void *cls, struct GNUNET_SERVER_Client *client,
6752                           const struct GNUNET_MessageHeader *message)
6753 {
6754   struct MeshClient *c;
6755
6756   /* Sanity check for client registration */
6757   if (NULL == (c = client_get (client)))
6758   {
6759     GNUNET_break (0);
6760     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6761     return;
6762   }
6763
6764   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6765               "Received get tunnels request from client %u\n",
6766               c->id);
6767 //   GNUNET_CONTAINER_multihashmap_iterate (tunnels,
6768 //                                          monitor_all_tunnels_iterator,
6769 //                                          client);
6770   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6771               "Get tunnels request from client %u completed\n",
6772               c->id);
6773   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6774 }
6775
6776
6777 /**
6778  * Handler for client's MONITOR_TUNNEL request.
6779  *
6780  * @param cls Closure (unused).
6781  * @param client Identification of the client.
6782  * @param message The actual message.
6783  */
6784 static void
6785 handle_local_show_tunnel (void *cls, struct GNUNET_SERVER_Client *client,
6786                           const struct GNUNET_MessageHeader *message)
6787 {
6788   const struct GNUNET_MESH_LocalMonitor *msg;
6789   struct GNUNET_MESH_LocalMonitor *resp;
6790   struct MeshClient *c;
6791   struct MeshChannel *ch;
6792
6793   /* Sanity check for client registration */
6794   if (NULL == (c = client_get (client)))
6795   {
6796     GNUNET_break (0);
6797     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6798     return;
6799   }
6800
6801   msg = (struct GNUNET_MESH_LocalMonitor *) message;
6802   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6803               "Received tunnel info request from client %u for tunnel %s[%X]\n",
6804               c->id,
6805               &msg->owner,
6806               ntohl (msg->channel_id));
6807 //   ch = channel_get (&msg->owner, ntohl (msg->channel_id));
6808   ch = NULL; // FIXME
6809   if (NULL == ch)
6810   {
6811     /* We don't know the tunnel */
6812     struct GNUNET_MESH_LocalMonitor warn;
6813
6814     warn = *msg;
6815     GNUNET_SERVER_notification_context_unicast (nc, client,
6816                                                 &warn.header,
6817                                                 GNUNET_NO);
6818     GNUNET_SERVER_receive_done (client, GNUNET_OK);
6819     return;
6820   }
6821
6822   /* Initialize context */
6823   resp = GNUNET_malloc (sizeof (struct GNUNET_MESH_LocalMonitor));
6824   *resp = *msg;
6825   resp->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
6826   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
6827                                               &resp->header, GNUNET_NO);
6828   GNUNET_free (resp);
6829
6830   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6831               "Monitor tunnel request from client %u completed\n",
6832               c->id);
6833   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6834 }
6835
6836
6837 /**
6838  * Functions to handle messages from clients
6839  */
6840 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
6841   {&handle_local_new_client, NULL,
6842    GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
6843   {&handle_local_channel_create, NULL,
6844    GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE,
6845    sizeof (struct GNUNET_MESH_ChannelMessage)},
6846   {&handle_local_channel_destroy, NULL,
6847    GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY,
6848    sizeof (struct GNUNET_MESH_ChannelMessage)},
6849   {&handle_local_data, NULL,
6850    GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA, 0},
6851   {&handle_local_ack, NULL,
6852    GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK,
6853    sizeof (struct GNUNET_MESH_LocalAck)},
6854   {&handle_local_get_tunnels, NULL,
6855    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS,
6856    sizeof (struct GNUNET_MessageHeader)},
6857   {&handle_local_show_tunnel, NULL,
6858    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL,
6859      sizeof (struct GNUNET_MESH_LocalMonitor)},
6860   {NULL, NULL, 0, 0}
6861 };
6862
6863
6864 /**
6865  * Method called whenever a given peer connects.
6866  *
6867  * @param cls closure
6868  * @param peer peer identity this notification is about
6869  */
6870 static void
6871 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
6872 {
6873   struct MeshPeer *pi;
6874   struct MeshPeerPath *path;
6875
6876   DEBUG_CONN ("Peer connected\n");
6877   DEBUG_CONN ("     %s\n", GNUNET_i2s (&my_full_id));
6878   pi = peer_get (peer);
6879   if (myid == pi->id)
6880   {
6881     DEBUG_CONN ("     (self)\n");
6882     path = path_new (1);
6883   }
6884   else
6885   {
6886     DEBUG_CONN ("     %s\n", GNUNET_i2s (peer));
6887     path = path_new (2);
6888     path->peers[1] = pi->id;
6889     GNUNET_PEER_change_rc (pi->id, 1);
6890     GNUNET_STATISTICS_update (stats, "# peers", 1, GNUNET_NO);
6891   }
6892   path->peers[0] = myid;
6893   GNUNET_PEER_change_rc (myid, 1);
6894   peer_add_path (pi, path, GNUNET_YES);
6895
6896   pi->connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES);
6897   return;
6898 }
6899
6900
6901 /**
6902  * Method called whenever a peer disconnects.
6903  *
6904  * @param cls closure
6905  * @param peer peer identity this notification is about
6906  */
6907 static void
6908 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
6909 {
6910   struct MeshPeer *pi;
6911
6912   DEBUG_CONN ("Peer disconnected\n");
6913   pi = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
6914   if (NULL == pi)
6915   {
6916     GNUNET_break (0);
6917     return;
6918   }
6919
6920   peer_remove_path (pi, myid, pi->id);
6921
6922   GNUNET_CONTAINER_multihashmap_iterate (pi->connections,
6923                                          connection_broken,
6924                                          pi);
6925   GNUNET_CONTAINER_multihashmap_destroy (pi->connections);
6926   pi->connections = NULL;
6927   if (myid == pi->id)
6928   {
6929     DEBUG_CONN ("     (self)\n");
6930   }
6931   GNUNET_STATISTICS_update (stats, "# peers", -1, GNUNET_NO);
6932
6933   return;
6934 }
6935
6936
6937 /**
6938  * Install server (service) handlers and start listening to clients.
6939  */
6940 static void
6941 server_init (void)
6942 {
6943   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
6944   GNUNET_SERVER_connect_notify (server_handle,
6945                                 &handle_local_client_connect, NULL);
6946   GNUNET_SERVER_disconnect_notify (server_handle,
6947                                    &handle_local_client_disconnect, NULL);
6948   nc = GNUNET_SERVER_notification_context_create (server_handle, 1);
6949
6950   clients_head = NULL;
6951   clients_tail = NULL;
6952   next_client_id = 0;
6953   GNUNET_SERVER_resume (server_handle);
6954 }
6955
6956
6957 /**
6958  * To be called on core init/fail.
6959  *
6960  * @param cls Closure (config)
6961  * @param identity the public identity of this peer
6962  */
6963 static void
6964 core_init (void *cls, 
6965            const struct GNUNET_PeerIdentity *identity)
6966 {
6967   const struct GNUNET_CONFIGURATION_Handle *c = cls;
6968   static int i = 0;
6969
6970   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Core init\n");
6971   if (0 != memcmp (identity, &my_full_id, sizeof (my_full_id)))
6972   {
6973     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Wrong CORE service\n"));
6974     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6975                 " core id %s\n",
6976                 GNUNET_i2s (identity));
6977     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6978                 " my id %s\n",
6979                 GNUNET_i2s (&my_full_id));
6980     GNUNET_CORE_disconnect (core_handle);
6981     core_handle = GNUNET_CORE_connect (c, /* Main configuration */
6982                                        NULL,      /* Closure passed to MESH functions */
6983                                        &core_init,        /* Call core_init once connected */
6984                                        &core_connect,     /* Handle connects */
6985                                        &core_disconnect,  /* remove peers on disconnects */
6986                                        NULL,      /* Don't notify about all incoming messages */
6987                                        GNUNET_NO, /* For header only in notification */
6988                                        NULL,      /* Don't notify about all outbound messages */
6989                                        GNUNET_NO, /* For header-only out notification */
6990                                        core_handlers);    /* Register these handlers */
6991     if (10 < i++)
6992       GNUNET_abort();
6993   }
6994   server_init ();
6995   return;
6996 }
6997
6998
6999 /******************************************************************************/
7000 /************************      MAIN FUNCTIONS      ****************************/
7001 /******************************************************************************/
7002
7003 /**
7004  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
7005  *
7006  * @param cls closure
7007  * @param key current key code
7008  * @param value value in the hash map
7009  * @return GNUNET_YES if we should continue to iterate,
7010  *         GNUNET_NO if not.
7011  */
7012 static int
7013 shutdown_tunnel (void *cls, const struct GNUNET_HashCode * key, void *value)
7014 {
7015   struct MeshPeer *p = value;
7016   struct MeshTunnel2 *t = p->tunnel;
7017
7018   if (NULL != t)
7019     tunnel_destroy (t);
7020   return GNUNET_YES;
7021 }
7022
7023
7024 /**
7025  * Task run during shutdown.
7026  *
7027  * @param cls unused
7028  * @param tc unused
7029  */
7030 static void
7031 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
7032 {
7033   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shutting down\n");
7034
7035   if (core_handle != NULL)
7036   {
7037     GNUNET_CORE_disconnect (core_handle);
7038     core_handle = NULL;
7039   }
7040   GNUNET_CONTAINER_multihashmap_iterate (peers, &shutdown_tunnel, NULL);
7041   if (dht_handle != NULL)
7042   {
7043     GNUNET_DHT_disconnect (dht_handle);
7044     dht_handle = NULL;
7045   }
7046   if (nc != NULL)
7047   {
7048     GNUNET_SERVER_notification_context_destroy (nc);
7049     nc = NULL;
7050   }
7051   if (GNUNET_SCHEDULER_NO_TASK != announce_id_task)
7052   {
7053     GNUNET_SCHEDULER_cancel (announce_id_task);
7054     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
7055   }
7056   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shut down\n");
7057 }
7058
7059
7060 /**
7061  * Process mesh requests.
7062  *
7063  * @param cls closure
7064  * @param server the initialized server
7065  * @param c configuration to use
7066  */
7067 static void
7068 run (void *cls, struct GNUNET_SERVER_Handle *server,
7069      const struct GNUNET_CONFIGURATION_Handle *c)
7070 {
7071   char *keyfile;
7072   struct GNUNET_CRYPTO_EccPrivateKey *pk;
7073
7074   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "starting to run\n");
7075   server_handle = server;
7076   GNUNET_SERVER_suspend (server_handle);
7077
7078   if (GNUNET_OK !=
7079       GNUNET_CONFIGURATION_get_value_filename (c, "PEER", "PRIVATE_KEY",
7080                                                &keyfile))
7081   {
7082     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7083                 _
7084                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7085                 "mesh", "peer/privatekey");
7086     GNUNET_SCHEDULER_shutdown ();
7087     return;
7088   }
7089
7090   if (GNUNET_OK !=
7091       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_CONNECTION_TIME",
7092                                            &refresh_connection_time))
7093   {
7094     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7095                 _
7096                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7097                 "mesh", "refresh path time");
7098     GNUNET_SCHEDULER_shutdown ();
7099     return;
7100   }
7101
7102   if (GNUNET_OK !=
7103       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "ID_ANNOUNCE_TIME",
7104                                            &id_announce_time))
7105   {
7106     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7107                 _
7108                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7109                 "mesh", "id announce time");
7110     GNUNET_SCHEDULER_shutdown ();
7111     return;
7112   }
7113
7114   if (GNUNET_OK !=
7115       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "CONNECT_TIMEOUT",
7116                                            &connect_timeout))
7117   {
7118     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7119                 _
7120                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7121                 "mesh", "connect timeout");
7122     GNUNET_SCHEDULER_shutdown ();
7123     return;
7124   }
7125
7126   if (GNUNET_OK !=
7127       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
7128                                              &max_msgs_queue))
7129   {
7130     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7131                 _
7132                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7133                 "mesh", "max msgs queue");
7134     GNUNET_SCHEDULER_shutdown ();
7135     return;
7136   }
7137
7138   if (GNUNET_OK !=
7139       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_CONNECTIONS",
7140                                              &max_connections))
7141   {
7142     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7143                 _
7144                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7145                 "mesh", "max tunnels");
7146     GNUNET_SCHEDULER_shutdown ();
7147     return;
7148   }
7149
7150   if (GNUNET_OK !=
7151       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DEFAULT_TTL",
7152                                              &default_ttl))
7153   {
7154     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7155                 _
7156                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7157                 "mesh", "default ttl", 64);
7158     default_ttl = 64;
7159   }
7160
7161   if (GNUNET_OK !=
7162       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS",
7163                                              &max_peers))
7164   {
7165     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7166                 _("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7167                 "mesh", "max peers", 1000);
7168     max_peers = 1000;
7169   }
7170
7171   if (GNUNET_OK !=
7172       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DROP_PERCENT",
7173                                              &drop_percent))
7174   {
7175     drop_percent = 0;
7176   }
7177   else
7178   {
7179     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7180                 "Mesh is running with drop mode enabled. "
7181                 "This is NOT a good idea! "
7182                 "Remove the DROP_PERCENT option from your configuration.\n");
7183   }
7184
7185   if (GNUNET_OK !=
7186       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DHT_REPLICATION_LEVEL",
7187                                              &dht_replication_level))
7188   {
7189     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7190                 _
7191                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7192                 "mesh", "dht replication level", 3);
7193     dht_replication_level = 3;
7194   }
7195
7196   connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES);
7197   peers = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
7198   ports = GNUNET_CONTAINER_multihashmap32_create (32);
7199
7200   dht_handle = GNUNET_DHT_connect (c, 64);
7201   if (NULL == dht_handle)
7202   {
7203     GNUNET_break (0);
7204   }
7205   stats = GNUNET_STATISTICS_create ("mesh", c);
7206
7207   /* Scheduled the task to clean up when shutdown is called */
7208   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
7209                                 NULL);
7210   pk = GNUNET_CRYPTO_ecc_key_create_from_file (keyfile);
7211   GNUNET_free (keyfile);
7212   GNUNET_assert (NULL != pk);
7213   my_private_key = pk;
7214   GNUNET_CRYPTO_ecc_key_get_public (my_private_key, &my_public_key);
7215   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
7216                       &my_full_id.hashPubKey);
7217   myid = GNUNET_PEER_intern (&my_full_id);
7218   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
7219               "Mesh for peer [%s] starting\n",
7220               GNUNET_i2s(&my_full_id));
7221
7222   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
7223                                      NULL,      /* Closure passed to MESH functions */
7224                                      &core_init,        /* Call core_init once connected */
7225                                      &core_connect,     /* Handle connects */
7226                                      &core_disconnect,  /* remove peers on disconnects */
7227                                      NULL,      /* Don't notify about all incoming messages */
7228                                      GNUNET_NO, /* For header only in notification */
7229                                      NULL,      /* Don't notify about all outbound messages */
7230                                      GNUNET_NO, /* For header-only out notification */
7231                                      core_handlers);    /* Register these handlers */
7232   if (NULL == core_handle)
7233   {
7234     GNUNET_break (0);
7235     GNUNET_SCHEDULER_shutdown ();
7236     return;
7237   }
7238   announce_id_task = GNUNET_SCHEDULER_add_now (&announce_id, cls);
7239   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Mesh service running\n");
7240 }
7241
7242
7243 /**
7244  * The main function for the mesh service.
7245  *
7246  * @param argc number of arguments from the command line
7247  * @param argv command line arguments
7248  * @return 0 ok, 1 on error
7249  */
7250 int
7251 main (int argc, char *const *argv)
7252 {
7253   int ret;
7254   int r;
7255
7256   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main()\n");
7257   r = GNUNET_SERVICE_run (argc, argv, "mesh", GNUNET_SERVICE_OPTION_NONE, &run,
7258                           NULL);
7259   ret = (GNUNET_OK == r) ? 0 : 1;
7260   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main() END\n");
7261
7262   INTERVAL_SHOW;
7263
7264   return ret;
7265 }