- send connection handshake before data
[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: %u\n",
1594                 GNUNET_h2s (&c->id), c->state);
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     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
1889       break;
1890
1891     default:
1892       GNUNET_break (0);
1893   }
1894
1895   queue_add (data,
1896              type,
1897              size,
1898              c,
1899              ch,
1900              fwd);
1901 }
1902
1903
1904 /**
1905  * Sends an already built message on a tunnel, choosing the best connection.
1906  *
1907  * @param message Message to send. Function modifies it.
1908  * @param t Tunnel on which this message is transmitted.
1909  * @param ch Channel on which this message is transmitted.
1910  * @param fwd Is this a fwd message?
1911  */
1912 static void
1913 send_prebuilt_message_tunnel (struct GNUNET_MESH_Encrypted *msg,
1914                               struct MeshTunnel2 *t,
1915                               struct MeshChannel *ch,
1916                               int fwd)
1917 {
1918   struct MeshConnection *c;
1919   uint16_t type;
1920
1921   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send on Tunnel %s\n",
1922               peer2s (t->peer));
1923   c = tunnel_get_connection (t, fwd);
1924   if (NULL == c)
1925   {
1926     GNUNET_break (GNUNET_YES == t->destroy);
1927     return;
1928   }
1929   type = ntohs (msg->header.type);
1930   switch (type)
1931   {
1932     case GNUNET_MESSAGE_TYPE_MESH_FWD:
1933     case GNUNET_MESSAGE_TYPE_MESH_BCK:
1934     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
1935     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
1936       msg->cid = c->id;
1937       msg->ttl = htonl (default_ttl);
1938       break;
1939     default:
1940       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "unkown type %s\n",
1941                   GNUNET_MESH_DEBUG_M2S (type));
1942       GNUNET_break (0);
1943   }
1944   msg->reserved = 0;
1945
1946   send_prebuilt_message_connection (&msg->header, c, ch, fwd);
1947 }
1948
1949
1950 /**
1951  * Sends an already built message on a channel, properly registering
1952  * all used resources and encrypting the message with the tunnel's key.
1953  *
1954  * @param message Message to send. Function makes a copy of it.
1955  * @param ch Channel on which this message is transmitted.
1956  * @param fwd Is this a fwd message?
1957  */
1958 static void
1959 send_prebuilt_message_channel (const struct GNUNET_MessageHeader *message,
1960                                struct MeshChannel *ch,
1961                                int fwd)
1962 {
1963   struct GNUNET_MESH_Encrypted *msg;
1964   size_t size = ntohs (message->size);
1965   char *cbuf[sizeof (struct GNUNET_MESH_Encrypted) + size];
1966   uint16_t type;
1967   uint64_t iv;
1968
1969   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send on Channel %s:%X %s\n",
1970               peer2s (ch->t->peer), ch->gid, fwd ? "FWD" : "BCK");
1971   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  %s\n",
1972               GNUNET_MESH_DEBUG_M2S (ntohs (message->type)));
1973
1974   if (channel_is_terminal (ch, fwd))
1975   {
1976     handle_decrypted (ch->t, message, fwd);
1977     return;
1978   }
1979   
1980   type = fwd ? GNUNET_MESSAGE_TYPE_MESH_FWD : GNUNET_MESSAGE_TYPE_MESH_BCK;
1981   iv = GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_NONCE, UINT64_MAX);
1982
1983   msg = (struct GNUNET_MESH_Encrypted *) cbuf;
1984   msg->header.type = htons (type);
1985   msg->header.size = htons (sizeof (struct GNUNET_MESH_Encrypted) + size);
1986   msg->iv = GNUNET_htonll (iv);
1987   tunnel_encrypt (ch->t, &msg[1], message, size, iv, fwd);
1988   send_prebuilt_message_tunnel (msg, ch->t, ch, fwd);
1989 }
1990
1991
1992 /**
1993  * Sends a CREATE CONNECTION message for a path to a peer.
1994  * Changes the connection and tunnel states if necessary.
1995  *
1996  * @param connection Connection to create.
1997  */
1998 static void
1999 send_connection_create (struct MeshConnection *connection)
2000 {
2001   struct MeshTunnel2 *t;
2002
2003   t = connection->t;
2004   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send connection create\n");
2005   queue_add (NULL,
2006              GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
2007              sizeof (struct GNUNET_MESH_ConnectionCreate) +
2008                 (connection->path->length *
2009                  sizeof (struct GNUNET_PeerIdentity)),
2010              connection,
2011              NULL,
2012              GNUNET_YES);
2013   if (NULL != t &&
2014       (MESH_TUNNEL_SEARCHING == t->state || MESH_TUNNEL_NEW == t->state))
2015     tunnel_change_state (t, MESH_TUNNEL_WAITING);
2016   if (MESH_CONNECTION_NEW == connection->state)
2017     connection_change_state (connection, MESH_CONNECTION_SENT);
2018 }
2019
2020
2021 /**
2022  * Sends a CONNECTION ACK message in reponse to a received CONNECTION_CREATE
2023  * directed to us.
2024  *
2025  * @param connection Connection to confirm.
2026  * @param fwd Is this a fwd ACK? (First is bck (SYNACK), second is fwd (ACK))
2027  */
2028 static void
2029 send_connection_ack (struct MeshConnection *connection, int fwd) 
2030 {
2031   struct MeshTunnel2 *t;
2032
2033   t = connection->t;
2034   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Send connection ack\n");
2035   queue_add (NULL,
2036              GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
2037              sizeof (struct GNUNET_MESH_ConnectionACK),
2038              connection,
2039              NULL,
2040              fwd);
2041   if (MESH_TUNNEL_NEW == t->state)
2042     tunnel_change_state (t, MESH_TUNNEL_WAITING);
2043   if (MESH_CONNECTION_READY != connection->state)
2044     connection_change_state (connection, MESH_CONNECTION_SENT);
2045 }
2046
2047
2048 /**
2049   * Core callback to write a pre-constructed data packet to core buffer
2050   *
2051   * @param cls Closure (MeshTransmissionDescriptor with data in "data" member).
2052   * @param size Number of bytes available in buf.
2053   * @param buf Where the to write the message.
2054   *
2055   * @return number of bytes written to buf
2056   */
2057 static size_t
2058 send_core_data_raw (void *cls, size_t size, void *buf)
2059 {
2060   struct GNUNET_MessageHeader *msg = cls;
2061   size_t total_size;
2062
2063   GNUNET_assert (NULL != msg);
2064   total_size = ntohs (msg->size);
2065
2066   if (total_size > size)
2067   {
2068     GNUNET_break (0);
2069     return 0;
2070   }
2071   memcpy (buf, msg, total_size);
2072   GNUNET_free (cls);
2073   return total_size;
2074 }
2075
2076
2077 /**
2078  * Function to send a create connection message to a peer.
2079  *
2080  * @param c Connection to create.
2081  * @param size number of bytes available in buf
2082  * @param buf where the callee should write the message
2083  * @return number of bytes written to buf
2084  */
2085 static size_t
2086 send_core_connection_create (struct MeshConnection *c, size_t size, void *buf)
2087 {
2088   struct GNUNET_MESH_ConnectionCreate *msg;
2089   struct GNUNET_PeerIdentity *peer_ptr;
2090   struct MeshPeerPath *p = c->path;
2091   size_t size_needed;
2092   int i;
2093
2094   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION CREATE...\n");
2095   size_needed =
2096       sizeof (struct GNUNET_MESH_ConnectionCreate) +
2097       p->length * sizeof (struct GNUNET_PeerIdentity);
2098
2099   if (size < size_needed || NULL == buf)
2100   {
2101     GNUNET_break (0);
2102     return 0;
2103   }
2104   msg = (struct GNUNET_MESH_ConnectionCreate *) buf;
2105   msg->header.size = htons (size_needed);
2106   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE);
2107   msg->cid = c->id;
2108
2109   peer_ptr = (struct GNUNET_PeerIdentity *) &msg[1];
2110   for (i = 0; i < p->length; i++)
2111   {
2112     GNUNET_PEER_resolve (p->peers[i], peer_ptr++);
2113   }
2114
2115   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2116               "CONNECTION CREATE (%u bytes long) sent!\n", size_needed);
2117   return size_needed;
2118 }
2119
2120
2121 /**
2122  * Creates a path ack message in buf and frees all unused resources.
2123  *
2124  * @param c Connection to send an ACK on.
2125  * @param size number of bytes available in buf
2126  * @param buf where the callee should write the message
2127  *
2128  * @return number of bytes written to buf
2129  */
2130 static size_t
2131 send_core_connection_ack (struct MeshConnection *c, size_t size, void *buf)
2132 {
2133   struct GNUNET_MESH_ConnectionACK *msg = buf;
2134   struct MeshTunnel2 *t = c->t;
2135
2136   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending CONNECTION ACK...\n");
2137   GNUNET_assert (NULL != t);
2138   if (sizeof (struct GNUNET_MESH_ConnectionACK) > size)
2139   {
2140     GNUNET_break (0);
2141     return 0;
2142   }
2143   msg->header.size = htons (sizeof (struct GNUNET_MESH_ConnectionACK));
2144   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK);
2145   msg->cid = c->id;
2146   msg->reserved = 0;
2147
2148   /* TODO add signature */
2149
2150   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CONNECTION ACK sent!\n");
2151   return sizeof (struct GNUNET_MESH_ConnectionACK);
2152 }
2153
2154
2155 /**
2156  * Destroy the peer_info and free any allocated resources linked to it
2157  *
2158  * @param peer The peer_info to destroy.
2159  *
2160  * @return GNUNET_OK on success
2161  */
2162 static int
2163 peer_destroy (struct MeshPeer *peer)
2164 {
2165   struct GNUNET_PeerIdentity id;
2166   struct MeshPeerPath *p;
2167   struct MeshPeerPath *nextp;
2168
2169   GNUNET_PEER_resolve (peer->id, &id);
2170   GNUNET_PEER_change_rc (peer->id, -1);
2171
2172   if (GNUNET_YES !=
2173       GNUNET_CONTAINER_multihashmap_remove (peers, &id.hashPubKey, peer))
2174   {
2175     GNUNET_break (0);
2176     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
2177                 "removing peer %s, not in hashmap\n", GNUNET_i2s (&id));
2178   }
2179   if (NULL != peer->dhtget)
2180   {
2181     GNUNET_DHT_get_stop (peer->dhtget);
2182   }
2183   p = peer->path_head;
2184   while (NULL != p)
2185   {
2186     nextp = p->next;
2187     GNUNET_CONTAINER_DLL_remove (peer->path_head, peer->path_tail, p);
2188     path_destroy (p);
2189     p = nextp;
2190   }
2191   tunnel_destroy_empty (peer->tunnel);
2192   GNUNET_free (peer);
2193   return GNUNET_OK;
2194 }
2195
2196
2197 /**
2198  * Returns if peer is used (has a tunnel, is neighbor).
2199  *
2200  * @peer Peer to check.
2201  *
2202  * @return GNUNET_YES if peer is in use.
2203  */
2204 static int
2205 peer_is_used (struct MeshPeer *peer)
2206 {
2207   struct MeshPeerPath *p;
2208
2209   if (NULL != peer->tunnel)
2210     return GNUNET_YES;
2211
2212   for (p = peer->path_head; NULL != p; p = p->next)
2213   {
2214     if (p->length < 3)
2215       return GNUNET_YES;
2216   }
2217   return GNUNET_NO;
2218 }
2219
2220 /**
2221  * Iterator over all the peers to get the oldest timestamp.
2222  *
2223  * @param cls Closure (unsued).
2224  * @param key ID of the peer.
2225  * @param value Peer_Info of the peer.
2226  */
2227 static int
2228 peer_get_oldest (void *cls,
2229                  const struct GNUNET_HashCode *key,
2230                  void *value)
2231 {
2232   struct MeshPeer *p = value;
2233   struct GNUNET_TIME_Absolute *abs = cls;
2234
2235   /* Don't count active peers */
2236   if (GNUNET_YES == peer_is_used (p))
2237     return GNUNET_YES;
2238
2239   if (abs->abs_value_us < p->last_contact.abs_value_us)
2240     abs->abs_value_us = p->last_contact.abs_value_us;
2241
2242   return GNUNET_YES;
2243 }
2244
2245
2246 /**
2247  * Iterator over all the peers to remove the oldest entry.
2248  *
2249  * @param cls Closure (unsued).
2250  * @param key ID of the peer.
2251  * @param value Peer_Info of the peer.
2252  */
2253 static int
2254 peer_timeout (void *cls,
2255               const struct GNUNET_HashCode *key,
2256               void *value)
2257 {
2258   struct MeshPeer *p = value;
2259   struct GNUNET_TIME_Absolute *abs = cls;
2260
2261   if (p->last_contact.abs_value_us == abs->abs_value_us &&
2262       GNUNET_NO == peer_is_used (p))
2263   {
2264     peer_destroy (p);
2265     return GNUNET_NO;
2266   }
2267   return GNUNET_YES;
2268 }
2269
2270
2271 /**
2272  * Delete oldest unused peer.
2273  */
2274 static void
2275 peer_delete_oldest (void)
2276 {
2277   struct GNUNET_TIME_Absolute abs;
2278
2279   abs = GNUNET_TIME_UNIT_FOREVER_ABS;
2280
2281   GNUNET_CONTAINER_multihashmap_iterate (peers,
2282                                          &peer_get_oldest,
2283                                          &abs);
2284   GNUNET_CONTAINER_multihashmap_iterate (peers,
2285                                          &peer_timeout,
2286                                          &abs);
2287 }
2288
2289
2290 /**
2291  * Retrieve the MeshPeer stucture associated with the peer, create one
2292  * and insert it in the appropriate structures if the peer is not known yet.
2293  *
2294  * @param peer Full identity of the peer.
2295  *
2296  * @return Existing or newly created peer info.
2297  */
2298 static struct MeshPeer *
2299 peer_get (const struct GNUNET_PeerIdentity *peer_id)
2300 {
2301   struct MeshPeer *peer;
2302
2303   peer = GNUNET_CONTAINER_multihashmap_get (peers, &peer_id->hashPubKey);
2304   if (NULL == peer)
2305   {
2306     peer = GNUNET_new (struct MeshPeer);
2307     if (GNUNET_CONTAINER_multihashmap_size (peers) > max_peers)
2308     {
2309       peer_delete_oldest ();
2310     }
2311     GNUNET_CONTAINER_multihashmap_put (peers, &peer_id->hashPubKey, peer,
2312                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
2313     peer->id = GNUNET_PEER_intern (peer_id);
2314   }
2315   peer->last_contact = GNUNET_TIME_absolute_get();
2316
2317   return peer;
2318 }
2319
2320
2321 /**
2322  * Retrieve the MeshPeer stucture associated with the peer, create one
2323  * and insert it in the appropriate structures if the peer is not known yet.
2324  *
2325  * @param peer Short identity of the peer.
2326  *
2327  * @return Existing or newly created peer info.
2328  */
2329 static struct MeshPeer *
2330 peer_get_short (const GNUNET_PEER_Id peer)
2331 {
2332   return peer_get (GNUNET_PEER_resolve2 (peer));
2333 }
2334
2335
2336 /**
2337  * Get a cost of a path for a peer considering existing tunnel connections.
2338  *
2339  * @param peer Peer towards which the path is considered.
2340  * @param path Candidate path.
2341  *
2342  * @return Cost of the path (path length + number of overlapping nodes)
2343  */
2344 static unsigned int
2345 peer_get_path_cost (const struct MeshPeer *peer,
2346                     const struct MeshPeerPath *path)
2347 {
2348   struct MeshConnection *c;
2349   unsigned int overlap;
2350   unsigned int i;
2351   unsigned int j;
2352
2353   if (NULL == path)
2354     return 0;
2355
2356   overlap = 0;
2357   GNUNET_assert (NULL != peer->tunnel);
2358
2359   for (i = 0; i < path->length; i++)
2360   {
2361     for (c = peer->tunnel->connection_head; NULL != c; c = c->next)
2362     {
2363       for (j = 0; j < c->path->length; j++)
2364       {
2365         if (path->peers[i] == c->path->peers[j])
2366         {
2367           overlap++;
2368           break;
2369         }
2370       }
2371     }
2372   }
2373   return (path->length + overlap) * (path->score * -1);
2374 }
2375
2376
2377 /**
2378  * Choose the best path towards a peer considering the tunnel properties.
2379  *
2380  * @param peer The destination peer.
2381  *
2382  * @return Best current known path towards the peer, if any.
2383  */
2384 static struct MeshPeerPath *
2385 peer_get_best_path (const struct MeshPeer *peer)
2386 {
2387   struct MeshPeerPath *best_p;
2388   struct MeshPeerPath *p;
2389   struct MeshConnection *c;
2390   unsigned int best_cost;
2391   unsigned int cost;
2392
2393   best_cost = UINT_MAX;
2394   best_p = NULL;
2395   for (p = peer->path_head; NULL != p; p = p->next)
2396   {
2397     for (c = peer->tunnel->connection_head; NULL != c; c = c->next)
2398       if (c->path == p)
2399         break;
2400     if (NULL != c)
2401       continue; /* If path is in use in a connection, skip it. */
2402
2403     if ((cost = peer_get_path_cost (peer, p)) < best_cost)
2404     {
2405       best_cost = cost;
2406       best_p = p;
2407     }
2408   }
2409   return best_p;
2410 }
2411
2412 static int
2413 queue_is_sendable (struct MeshPeerQueue *q)
2414 {
2415   struct MeshFlowControl *fc;
2416
2417   /* Is PID-independent? */
2418   switch (q->type)
2419   {
2420     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2421     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2422       return GNUNET_YES;
2423   }
2424
2425   /* Is PID allowed? */
2426   fc = q->fwd ? &q->c->fwd_fc : &q->c->bck_fc;
2427   if (GMC_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2428     return GNUNET_YES;
2429
2430   return GNUNET_NO;
2431 }
2432
2433
2434 /**
2435  * Get first sendable message.
2436  *
2437  * @param peer The destination peer.
2438  *
2439  * @return Best current known path towards the peer, if any.
2440  */
2441 static struct MeshPeerQueue *
2442 peer_get_first_message (const struct MeshPeer *peer)
2443 {
2444   struct MeshPeerQueue *q;
2445
2446   for (q = peer->queue_head; NULL != q; q = q->next)
2447   {
2448     if (queue_is_sendable (q))
2449       return q;
2450   }
2451
2452   return NULL;
2453 }
2454
2455
2456 /**
2457  * Try to establish a new connection to this peer in the given tunnel.
2458  * If the peer doesn't have any path to it yet, try to get one.
2459  * If the peer already has some path, send a CREATE CONNECTION towards it.
2460  *
2461  * @param peer PeerInfo of the peer.
2462  */
2463 static void
2464 peer_connect (struct MeshPeer *peer)
2465 {
2466   struct MeshTunnel2 *t;
2467   struct MeshPeerPath *p;
2468   struct MeshConnection *c;
2469   int rerun_dhtget;
2470
2471   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2472               "peer_connect towards %s\n",
2473               peer2s (peer));
2474   t = peer->tunnel;
2475   c = NULL;
2476   rerun_dhtget = GNUNET_NO;
2477
2478   if (NULL != peer->path_head)
2479   {
2480     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "path exists\n");
2481     p = peer_get_best_path (peer);
2482     if (NULL != p)
2483     {
2484       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  %u hops\n", p->length);
2485       c = tunnel_use_path (t, p);
2486       if (NULL == c)
2487       {
2488         /* This case can happen when the path includes a first hop that is
2489          * not yet known to be connected.
2490          * 
2491          * This happens quite often during testing when running mesh
2492          * under valgrind: core connect notifications come very late and the
2493          * DHT result has already come and created a valid path.
2494          * In this case, the peer->connections hashmap will be NULL and
2495          * tunnel_use_path will not be able to create a connection from that
2496          * path.
2497          *
2498          * Re-running the DHT GET should give core time to callback.
2499          */
2500         GNUNET_break(0);
2501         rerun_dhtget = GNUNET_YES;
2502       }
2503       else
2504       {
2505         send_connection_create (c);
2506         return;
2507       }
2508     }
2509   }
2510
2511   if (NULL != peer->dhtget && GNUNET_YES == rerun_dhtget)
2512   {
2513     GNUNET_DHT_get_stop (peer->dhtget);
2514     peer->dhtget = NULL;
2515     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2516                 "  Stopping DHT GET for peer %s\n", peer2s (peer));
2517   }
2518
2519   if (NULL == peer->dhtget)
2520   {
2521     const struct GNUNET_PeerIdentity *id;
2522
2523     id = GNUNET_PEER_resolve2 (peer->id);
2524     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2525                 "  Starting DHT GET for peer %s\n", peer2s (peer));
2526     peer->dhtget = GNUNET_DHT_get_start (dht_handle,    /* handle */
2527                                          GNUNET_BLOCK_TYPE_MESH_PEER, /* type */
2528                                          &id->hashPubKey,     /* key to search */
2529                                          dht_replication_level, /* replication level */
2530                                          GNUNET_DHT_RO_RECORD_ROUTE |
2531                                          GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
2532                                          NULL,       /* xquery */
2533                                          0,     /* xquery bits */
2534                                          &dht_get_id_handler, peer);
2535     if (MESH_TUNNEL_NEW == t->state)
2536       tunnel_change_state (t, MESH_TUNNEL_SEARCHING);
2537   }
2538 }
2539
2540
2541 /**
2542  * Get the first transmittable message for a connection.
2543  *
2544  * @param c Connection.
2545  * @param fwd Is this FWD?
2546  *
2547  * @return First transmittable message.
2548  */
2549 static struct MeshPeerQueue *
2550 connection_get_first_message (struct MeshConnection *c, int fwd)
2551 {
2552   struct MeshPeerQueue *q;
2553   struct MeshPeer *p;
2554
2555   p = connection_get_hop (c, fwd);
2556
2557   for (q = p->queue_head; NULL != q; q = q->next)
2558   {
2559     if (q->c != c)
2560       continue;
2561     if (queue_is_sendable (q))
2562       return q;
2563   }
2564
2565   return NULL;
2566 }
2567
2568
2569 /**
2570  * @brief Re-initiate traffic on this connection if necessary.
2571  *
2572  * Check if there is traffic queued towards this peer
2573  * and the core transmit handle is NULL (traffic was stalled).
2574  * If so, call core tmt rdy.
2575  *
2576  * @param c Connection on which initiate traffic.
2577  * @param fwd Is this about fwd traffic?
2578  */
2579 static void
2580 connection_unlock_queue (struct MeshConnection *c, int fwd)
2581 {
2582   struct MeshPeer *peer;
2583   struct MeshPeerQueue *q;
2584   size_t size;
2585
2586   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2587               "connection_unlock_queue %s on %s\n",
2588               fwd ? "FWD" : "BCK", GNUNET_h2s (&c->id));
2589
2590   if (connection_is_origin (c, fwd))
2591   {
2592     struct MeshTunnel2 *t = c->t;
2593     struct MeshChannel *ch;
2594     struct MeshChannelReliability *rel;
2595
2596     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " is origin!\n");
2597     /* FIXME randomize channel selection, not always first channel */
2598     for (ch = t->channel_head; NULL != ch; ch = ch->next)
2599     {
2600       rel = fwd ? ch->root_rel : ch->dest_rel;
2601
2602       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  channel %X - %s\n",
2603                   ch->gid, rel->client_ready ? "ready " : "not ready");
2604       if (GNUNET_NO == rel->client_ready)
2605       {
2606         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    sending local ack!\n");
2607         send_local_ack (ch, fwd);
2608         return; /* FIXME authorize all channels? */
2609       }
2610     }
2611     return;
2612   }
2613
2614   peer = connection_get_hop (c, fwd);
2615
2616   if (NULL != peer->core_transmit)
2617   {
2618     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  already unlocked!\n");
2619     return; /* Already unlocked */
2620   }
2621
2622   q = connection_get_first_message (c, fwd);
2623   if (NULL == q)
2624   {
2625     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  queue empty!\n");
2626     return; /* Nothing to transmit */
2627   }
2628
2629   size = q->size;
2630   peer->core_transmit =
2631       GNUNET_CORE_notify_transmit_ready (core_handle,
2632                                          GNUNET_NO,
2633                                          0,
2634                                          GNUNET_TIME_UNIT_FOREVER_REL,
2635                                          GNUNET_PEER_resolve2 (peer->id),
2636                                          size,
2637                                          &queue_send,
2638                                          peer);
2639 }
2640
2641
2642 /**
2643  * Cancel all transmissions that belong to a certain connection.
2644  *
2645  * @param c Connection which to cancel.
2646  * @param fwd Cancel fwd traffic?
2647  */
2648 static void
2649 connection_cancel_queues (struct MeshConnection *c, int fwd)
2650 {
2651   struct MeshPeerQueue *q;
2652   struct MeshPeerQueue *next;
2653   struct MeshFlowControl *fc;
2654   struct MeshPeer *peer;
2655
2656   if (NULL == c)
2657   {
2658     GNUNET_break (0);
2659     return;
2660   }
2661   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2662   peer = connection_get_hop (c, fwd);
2663
2664   for (q = peer->queue_head; NULL != q; q = next)
2665   {
2666     next = q->next;
2667     if (q->c == c)
2668     {
2669       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2670                   "connection_cancel_queue %s\n",
2671                   GNUNET_MESH_DEBUG_M2S (q->type));
2672       queue_destroy (q, GNUNET_YES);
2673     }
2674   }
2675   if (NULL == peer->queue_head)
2676   {
2677     if (NULL != peer->core_transmit)
2678     {
2679       GNUNET_CORE_notify_transmit_ready_cancel (peer->core_transmit);
2680       peer->core_transmit = NULL;
2681     }
2682     if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
2683     {
2684       GNUNET_SCHEDULER_cancel (fc->poll_task);
2685       fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2686     }
2687   }
2688 }
2689
2690
2691 /**
2692  * Add the path to the peer and update the path used to reach it in case this
2693  * is the shortest.
2694  *
2695  * @param peer_info Destination peer to add the path to.
2696  * @param path New path to add. Last peer must be the peer in arg 1.
2697  *             Path will be either used of freed if already known.
2698  * @param trusted Do we trust that this path is real?
2699  */
2700 void
2701 peer_add_path (struct MeshPeer *peer_info, struct MeshPeerPath *path,
2702                     int trusted)
2703 {
2704   struct MeshPeerPath *aux;
2705   unsigned int l;
2706   unsigned int l2;
2707
2708   if ((NULL == peer_info) || (NULL == path))
2709   {
2710     GNUNET_break (0);
2711     path_destroy (path);
2712     return;
2713   }
2714   if (path->peers[path->length - 1] != peer_info->id)
2715   {
2716     GNUNET_break (0);
2717     path_destroy (path);
2718     return;
2719   }
2720   if (2 >= path->length && GNUNET_NO == trusted)
2721   {
2722     /* Only allow CORE to tell us about direct paths */
2723     path_destroy (path);
2724     return;
2725   }
2726   for (l = 1; l < path->length; l++)
2727   {
2728     if (path->peers[l] == myid)
2729     {
2730       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shortening path by %u\n", l);
2731       for (l2 = 0; l2 < path->length - l; l2++)
2732       {
2733         path->peers[l2] = path->peers[l + l2];
2734       }
2735       path->length -= l;
2736       l = 1;
2737       path->peers =
2738           GNUNET_realloc (path->peers, path->length * sizeof (GNUNET_PEER_Id));
2739     }
2740   }
2741
2742   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "adding path [%u] to peer %s\n",
2743               path->length, peer2s (peer_info));
2744
2745   l = path_get_length (path);
2746   if (0 == l)
2747   {
2748     path_destroy (path);
2749     return;
2750   }
2751
2752   GNUNET_assert (peer_info->id == path->peers[path->length - 1]);
2753   for (aux = peer_info->path_head; aux != NULL; aux = aux->next)
2754   {
2755     l2 = path_get_length (aux);
2756     if (l2 > l)
2757     {
2758       GNUNET_CONTAINER_DLL_insert_before (peer_info->path_head,
2759                                           peer_info->path_tail, aux, path);
2760       return;
2761     }
2762     else
2763     {
2764       if (l2 == l && memcmp (path->peers, aux->peers, l) == 0)
2765       {
2766         path_destroy (path);
2767         return;
2768       }
2769     }
2770   }
2771   GNUNET_CONTAINER_DLL_insert_tail (peer_info->path_head, peer_info->path_tail,
2772                                     path);
2773   return;
2774 }
2775
2776
2777 /**
2778  * Add the path to the origin peer and update the path used to reach it in case
2779  * this is the shortest.
2780  * The path is given in peer_info -> destination, therefore we turn the path
2781  * upside down first.
2782  *
2783  * @param peer_info Peer to add the path to, being the origin of the path.
2784  * @param path New path to add after being inversed.
2785  *             Path will be either used or freed.
2786  * @param trusted Do we trust that this path is real?
2787  */
2788 static void
2789 peer_add_path_to_origin (struct MeshPeer *peer_info,
2790                          struct MeshPeerPath *path, int trusted)
2791 {
2792   if (NULL == path)
2793     return;
2794   path_invert (path);
2795   peer_add_path (peer_info, path, trusted);
2796 }
2797
2798
2799
2800 /**
2801  * Function called if a connection has been stalled for a while,
2802  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
2803  *
2804  * @param cls Closure (poll ctx).
2805  * @param tc TaskContext.
2806  */
2807 static void
2808 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2809 {
2810   struct MeshFlowControl *fc = cls;
2811   struct GNUNET_MESH_Poll msg;
2812   struct MeshConnection *c;
2813
2814   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2815   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2816   {
2817     return;
2818   }
2819
2820   c = fc->c;
2821   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** Polling!\n");
2822   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** connection %s[%X]\n", 
2823               peer2s (c->t->peer), c->id);
2824   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ***   %s\n", 
2825               fc == &c->fwd_fc ? "FWD" : "BCK");
2826
2827   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
2828   msg.header.size = htons (sizeof (msg));
2829   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " *** pid (%u)!\n", fc->last_pid_sent);
2830   send_prebuilt_message_connection (&msg.header, c, NULL, fc == &c->fwd_fc);
2831   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
2832   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
2833                                                 &connection_poll, fc);
2834 }
2835
2836
2837 /**
2838  * Build a PeerPath from the paths returned from the DHT, reversing the paths
2839  * to obtain a local peer -> destination path and interning the peer ids.
2840  *
2841  * @return Newly allocated and created path
2842  */
2843 static struct MeshPeerPath *
2844 path_build_from_dht (const struct GNUNET_PeerIdentity *get_path,
2845                      unsigned int get_path_length,
2846                      const struct GNUNET_PeerIdentity *put_path,
2847                      unsigned int put_path_length)
2848 {
2849   struct MeshPeerPath *p;
2850   GNUNET_PEER_Id id;
2851   int i;
2852
2853   p = path_new (1);
2854   p->peers[0] = myid;
2855   GNUNET_PEER_change_rc (myid, 1);
2856   i = get_path_length;
2857   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   GET has %d hops.\n", i);
2858   for (i--; i >= 0; i--)
2859   {
2860     id = GNUNET_PEER_intern (&get_path[i]);
2861     if (p->length > 0 && id == p->peers[p->length - 1])
2862     {
2863       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
2864       GNUNET_PEER_change_rc (id, -1);
2865     }
2866     else
2867     {
2868       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from GET: %s.\n",
2869                   GNUNET_i2s (&get_path[i]));
2870       p->length++;
2871       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
2872       p->peers[p->length - 1] = id;
2873     }
2874   }
2875   i = put_path_length;
2876   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   PUT has %d hops.\n", i);
2877   for (i--; i >= 0; i--)
2878   {
2879     id = GNUNET_PEER_intern (&put_path[i]);
2880     if (id == myid)
2881     {
2882       /* PUT path went through us, so discard the path up until now and start
2883        * from here to get a much shorter (and loop-free) path.
2884        */
2885       path_destroy (p);
2886       p = path_new (0);
2887     }
2888     if (p->length > 0 && id == p->peers[p->length - 1])
2889     {
2890       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Optimizing 1 hop out.\n");
2891       GNUNET_PEER_change_rc (id, -1);
2892     }
2893     else
2894     {
2895       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   Adding from PUT: %s.\n",
2896                   GNUNET_i2s (&put_path[i]));
2897       p->length++;
2898       p->peers = GNUNET_realloc (p->peers, sizeof (GNUNET_PEER_Id) * p->length);
2899       p->peers[p->length - 1] = id;
2900     }
2901   }
2902 #if MESH_DEBUG
2903   if (get_path_length > 0)
2904     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of GET: %s)\n",
2905                 GNUNET_i2s (&get_path[0]));
2906   if (put_path_length > 0)
2907     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (first of PUT: %s)\n",
2908                 GNUNET_i2s (&put_path[0]));
2909   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   In total: %d hops\n",
2910               p->length);
2911   for (i = 0; i < p->length; i++)
2912   {
2913     struct GNUNET_PeerIdentity peer_id;
2914
2915     GNUNET_PEER_resolve (p->peers[i], &peer_id);
2916     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "       %u: %s\n", p->peers[i],
2917                 GNUNET_i2s (&peer_id));
2918   }
2919 #endif
2920   return p;
2921 }
2922
2923
2924 /**
2925  * Adds a path to the peer_infos of all the peers in the path
2926  *
2927  * @param p Path to process.
2928  * @param confirmed Whether we know if the path works or not.
2929  */
2930 static void
2931 path_add_to_peers (struct MeshPeerPath *p, int confirmed)
2932 {
2933   unsigned int i;
2934
2935   /* TODO: invert and add */
2936   for (i = 0; i < p->length && p->peers[i] != myid; i++) /* skip'em */ ;
2937   for (i++; i < p->length; i++)
2938   {
2939     struct MeshPeer *aux;
2940     struct MeshPeerPath *copy;
2941
2942     aux = peer_get_short (p->peers[i]);
2943     copy = path_duplicate (p);
2944     copy->length = i + 1;
2945     peer_add_path (aux, copy, p->length < 3 ? GNUNET_NO : confirmed);
2946   }
2947 }
2948
2949
2950 /**
2951  * Search for a channel among the channels for a client
2952  *
2953  * @param c the client whose channels to search in
2954  * @param chid the local id of the channel
2955  *
2956  * @return channel handler, NULL if doesn't exist
2957  */
2958 static struct MeshChannel *
2959 channel_get_by_local_id (struct MeshClient *c, MESH_ChannelNumber chid)
2960 {
2961   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   -- get CHID %X\n", chid);
2962   if (0 == (chid & GNUNET_MESH_LOCAL_CHANNEL_ID_CLI))
2963   {
2964     GNUNET_break_op (0);
2965     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CHID %X not a local chid\n", chid);
2966     return NULL;
2967   }
2968   if (chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV)
2969     return GNUNET_CONTAINER_multihashmap32_get (c->incoming_channels, chid);
2970   return GNUNET_CONTAINER_multihashmap32_get (c->own_channels, chid);
2971 }
2972
2973 #if 0
2974
2975 static void
2976 channel_debug (struct MeshChannel *ch)
2977 {
2978   if (NULL == ch)
2979   {
2980     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CHANNEL ***\n");
2981     return;
2982   }
2983   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Channel %s:%X\n",
2984               peer2s (ch->t->peer), ch->gid);
2985   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  root %p/%p\n",
2986               ch->root, ch->root_rel);
2987   if (NULL != ch->root)
2988   {
2989     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  cli %u\n", ch->root->id);
2990     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ready %s\n",
2991                 ch->root_rel->client_ready ? "YES" : "NO");
2992     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  id %X\n", ch->lid_root);
2993   }
2994   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  root %p/%p\n",
2995               ch->dest, ch->dest_rel);
2996   if (NULL != ch->dest)
2997   {
2998     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  cli %u\n", ch->dest->id);
2999     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ready %s\n",
3000                 ch->dest_rel->client_ready ? "YES" : "NO");
3001     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  id %X\n", ch->lid_dest);
3002   }
3003 }
3004
3005 static void
3006 fc_debug (struct MeshFlowControl *fc)
3007 {
3008   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    IN: %u/%u\n",
3009               fc->last_pid_recv, fc->last_ack_sent);
3010   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    OUT: %u/%u\n",
3011               fc->last_pid_sent, fc->last_ack_recv);
3012   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    QUEUE: %u/%u\n",
3013               fc->queue_n, fc->queue_max);
3014 }
3015
3016 static void
3017 connection_debug (struct MeshConnection *c)
3018 {
3019   if (NULL == c)
3020   {
3021     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CONNECTION ***\n");
3022     return;
3023   }
3024   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connection %s:%X\n",
3025               peer2s (c->t->peer), GNUNET_h2s (&c->id));
3026   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  state: %u, pending msgs: %u\n", 
3027               c->state, c->pending_messages);
3028   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
3029   fc_debug (&c->fwd_fc);
3030   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
3031   fc_debug (&c->bck_fc);
3032 }
3033
3034 #endif
3035
3036 static struct MeshChannel *
3037 channel_get (struct MeshTunnel2 *t, MESH_ChannelNumber chid)
3038 {
3039   struct MeshChannel *ch;
3040
3041   if (NULL == t)
3042     return NULL;
3043
3044   for (ch = t->channel_head; NULL != ch; ch = ch->next)
3045   {
3046     if (ch->gid == chid)
3047       break;
3048   }
3049
3050   return ch;
3051 }
3052
3053
3054 /**
3055  * Change the tunnel state.
3056  *
3057  * @param t Tunnel whose state to change.
3058  * @param state New state.
3059  */
3060 static void
3061 tunnel_change_state (struct MeshTunnel2* t, enum MeshTunnelState state)
3062 {
3063   if (NULL == t)
3064     return;
3065   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3066               "Tunnel %s state was %s\n",
3067               peer2s (t->peer),
3068               GNUNET_MESH_DEBUG_TS2S (t->state));
3069   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3070               "Tunnel %s state is now %s\n",
3071               peer2s (t->peer),
3072               GNUNET_MESH_DEBUG_TS2S (state));
3073   t->state = state;
3074 }
3075
3076
3077 /**
3078  * Cache a message to be sent once tunnel is online.
3079  *
3080  * @param t Tunnel to hold the message.
3081  * @param ch Channel the message is about.
3082  * @param msg Message itself (copy will be made).
3083  * @param fwd Is this fwd?
3084  */
3085 static void
3086 tunnel_queue_data (struct MeshTunnel2 *t,
3087                    struct MeshChannel *ch,
3088                    struct GNUNET_MessageHeader *msg,
3089                    int fwd)
3090 {
3091   struct MeshTunnelQueue *tq;
3092   uint16_t size = ntohs (msg->size);
3093
3094   tq = GNUNET_malloc (sizeof (struct MeshTunnelQueue) + size);
3095
3096   tq->ch = ch;
3097   memcpy (&tq[1], msg, size);
3098   GNUNET_CONTAINER_DLL_insert_tail (t->tq_head, t->tq_tail, tq);
3099 }
3100
3101
3102 /**
3103  * Send all cached messages that we can, tunnel is online.
3104  *
3105  * @param t Tunnel that holds the messages.
3106  * @param fwd Is this fwd?
3107  */
3108 static void
3109 tunnel_send_queued_data (struct MeshTunnel2 *t, int fwd)
3110 {
3111   struct MeshTunnelQueue *tq;
3112   struct MeshTunnelQueue *next;
3113   unsigned int room;
3114
3115   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3116               "tunnel_send_queued_data on tunnel %s\n",
3117               peer2s (t->peer));
3118   room = tunnel_get_buffer (t, fwd);
3119   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  buffer space: %u\n", room);
3120   for (tq = t->tq_head; NULL != tq && room > 0; tq = next)
3121   {
3122     next = tq->next;
3123     room--;
3124     GNUNET_CONTAINER_DLL_remove (t->tq_head, t->tq_tail, tq);
3125     send_prebuilt_message_channel ((struct GNUNET_MessageHeader *) &tq[1],
3126                                    tq->ch, fwd);
3127
3128     GNUNET_free (tq);
3129   }
3130 }
3131
3132
3133 static void
3134 connection_change_state (struct MeshConnection* c,
3135                          enum MeshConnectionState state)
3136 {
3137   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3138               "Connection %s state was %s\n",
3139               GNUNET_h2s (&c->id), GNUNET_MESH_DEBUG_CS2S (c->state));
3140   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3141               "Connection %s state is now %s\n",
3142               GNUNET_h2s (&c->id), GNUNET_MESH_DEBUG_CS2S (state));
3143   c->state = state;
3144 }
3145
3146
3147 /**
3148  * Add a client to a channel, initializing all needed data structures.
3149  * 
3150  * @param ch Channel to which add the client.
3151  * @param c Client which to add to the channel.
3152  */
3153 static void
3154 channel_add_client (struct MeshChannel *ch, struct MeshClient *c)
3155 {
3156   struct MeshTunnel2 *t = ch->t;
3157
3158   if (NULL != ch->dest)
3159   {
3160     GNUNET_break (0);
3161     return;
3162   }
3163
3164   /* Assign local id as destination */
3165   while (NULL != channel_get_by_local_id (c, t->next_local_chid))
3166     t->next_local_chid = (t->next_local_chid + 1) | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
3167   ch->lid_dest = t->next_local_chid++;
3168   t->next_local_chid = t->next_local_chid | GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
3169
3170   /* Store in client's hashmap */
3171   if (GNUNET_OK !=
3172       GNUNET_CONTAINER_multihashmap32_put (c->incoming_channels,
3173                                            ch->lid_dest, ch,
3174                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
3175   {
3176     GNUNET_break (0);
3177     return;
3178   }
3179
3180   GNUNET_break (NULL == ch->dest_rel);
3181   ch->dest_rel = GNUNET_new (struct MeshChannelReliability);
3182   ch->dest_rel->ch = ch;
3183   ch->dest_rel->expected_delay = MESH_RETRANSMIT_TIME;
3184
3185   ch->dest = c;
3186 }
3187
3188
3189 static struct MeshConnection *
3190 tunnel_use_path (struct MeshTunnel2 *t, struct MeshPeerPath *p)
3191 {
3192   struct MeshConnection *c;
3193   struct GNUNET_HashCode cid;
3194   struct MeshPeer *peer;
3195   unsigned int own_pos;
3196
3197   if (NULL == t || NULL == p)
3198   {
3199     GNUNET_break (0);
3200     return NULL;
3201   }
3202
3203   GNUNET_CRYPTO_hash_create_random (GNUNET_CRYPTO_QUALITY_NONCE, &cid);
3204
3205   c = connection_new (&cid);
3206   c->t = t;
3207   GNUNET_CONTAINER_DLL_insert (t->connection_head, t->connection_tail, c);
3208   for (own_pos = 0; own_pos < p->length; own_pos++)
3209   {
3210     if (p->peers[own_pos] == myid)
3211       break;
3212   }
3213   if (own_pos > p->length - 1)
3214   {
3215     GNUNET_break (0);
3216     connection_destroy (c);
3217     return NULL;
3218   }
3219   c->own_pos = own_pos;
3220   c->path = p;
3221
3222   if (0 == own_pos)
3223   {
3224     c->fwd_maintenance_task =
3225         GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
3226                                       &connection_fwd_keepalive, c);
3227   }
3228
3229   peer = connection_get_next_hop (c);
3230   if (NULL == peer->connections)
3231   {
3232     connection_destroy (c);
3233     return NULL;
3234   }
3235   GNUNET_CONTAINER_multihashmap_put (peer->connections, &c->id, c,
3236                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
3237   peer = connection_get_prev_hop (c);
3238   if (NULL == peer->connections)
3239   {
3240     connection_destroy (c);
3241     return NULL;
3242   }
3243   GNUNET_CONTAINER_multihashmap_put (peer->connections, &c->id, c,
3244                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
3245   return c;
3246 }
3247
3248
3249 /**
3250  * Notifies a tunnel that a connection has broken that affects at least
3251  * some of its peers. Sends a notification towards the root of the tree.
3252  * In case the peer is the owner of the tree, notifies the client that owns
3253  * the tunnel and tries to reconnect.
3254  * 
3255  * FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME 
3256  *
3257  * @param t Tunnel affected.
3258  * @param p1 Peer that got disconnected from p2.
3259  * @param p2 Peer that got disconnected from p1.
3260  *
3261  * @return Short ID of the peer disconnected (either p1 or p2).
3262  *         0 if the tunnel remained unaffected.
3263  */
3264 static GNUNET_PEER_Id
3265 tunnel_notify_connection_broken (struct MeshTunnel2* t,
3266                                  GNUNET_PEER_Id p1, GNUNET_PEER_Id p2)
3267 {
3268 //   if (myid != p1 && myid != p2) FIXME
3269 //   {
3270 //     return;
3271 //   }
3272 // 
3273 //   if (tree_get_predecessor (t->tree) != 0)
3274 //   {
3275 //     /* We are the peer still connected, notify owner of the disconnection. */
3276 //     struct GNUNET_MESH_PathBroken msg;
3277 //     struct GNUNET_PeerIdentity neighbor;
3278 // 
3279 //     msg.header.size = htons (sizeof (msg));
3280 //     msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_PATH_BROKEN);
3281 //     GNUNET_PEER_resolve (t->id.oid, &msg.oid);
3282 //     msg.tid = htonl (t->id.tid);
3283 //     msg.peer1 = my_full_id;
3284 //     GNUNET_PEER_resolve (pid, &msg.peer2);
3285 //     GNUNET_PEER_resolve (tree_get_predecessor (t->tree), &neighbor);
3286 //     send_prebuilt_message (&msg.header, &neighbor, t);
3287 //   }
3288   return 0;
3289 }
3290
3291
3292 /**
3293  * Send an end-to-end ACK message for the most recent in-sequence payload.
3294  *
3295  * If channel is not reliable, do nothing.
3296  *
3297  * @param ch Channel this is about.
3298  * @param fwd Is for FWD traffic? (ACK dest->owner)
3299  */
3300 static void
3301 channel_send_data_ack (struct MeshChannel *ch, int fwd)
3302 {
3303   struct GNUNET_MESH_DataACK msg;
3304   struct MeshChannelReliability *rel;
3305   struct MeshReliableMessage *copy;
3306   unsigned int delta;
3307   uint64_t mask;
3308   uint16_t type;
3309
3310   if (GNUNET_NO == ch->reliable)
3311   {
3312     return;
3313   }
3314   rel = fwd ? ch->dest_rel : ch->root_rel;
3315   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3316               "send_data_ack for %u\n",
3317               rel->mid_recv - 1);
3318
3319   type = GNUNET_MESSAGE_TYPE_MESH_DATA_ACK;
3320   msg.header.type = htons (type);
3321   msg.header.size = htons (sizeof (msg));
3322   msg.chid = htonl (ch->gid);
3323   msg.mid = htonl (rel->mid_recv - 1);
3324   msg.futures = 0;
3325   for (copy = rel->head_recv; NULL != copy; copy = copy->next)
3326   {
3327     if (copy->type != type)
3328       continue;
3329     delta = copy->mid - rel->mid_recv;
3330     if (63 < delta)
3331       break;
3332     mask = 0x1LL << delta;
3333     msg.futures |= mask;
3334     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3335                 " setting bit for %u (delta %u) (%llX) -> %llX\n",
3336                 copy->mid, delta, mask, msg.futures);
3337   }
3338   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " final futures %llX\n", msg.futures);
3339
3340   send_prebuilt_message_channel (&msg.header, ch, fwd);
3341   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_data_ack END\n");
3342 }
3343
3344
3345 /**
3346  * Send an ACK informing the predecessor about the available buffer space.
3347  *
3348  * Note that for fwd ack, the FWD mean forward *traffic* (root->dest),
3349  * the ACK itself goes "back" (dest->root).
3350  *
3351  * @param c Connection on which to send the ACK.
3352  * @param buffer How much space free to advertise?
3353  * @param fwd Is this FWD ACK? (Going dest->owner)
3354  */
3355 static void
3356 connection_send_ack (struct MeshConnection *c, unsigned int buffer, int fwd)
3357 {
3358   struct MeshFlowControl *next_fc;
3359   struct MeshFlowControl *prev_fc;
3360   struct GNUNET_MESH_ACK msg;
3361   uint32_t ack;
3362   int delta;
3363
3364   next_fc = fwd ? &c->fwd_fc : &c->bck_fc;
3365   prev_fc = fwd ? &c->bck_fc : &c->fwd_fc;
3366
3367   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3368               "connection send %s ack on %s\n",
3369               fwd ? "FWD" : "BCK", GNUNET_h2s (&c->id));
3370
3371   /* Check if we need to transmit the ACK */
3372   if (prev_fc->last_ack_sent - prev_fc->last_pid_recv > 3)
3373   {
3374     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, buffer > 3\n");
3375     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3376                 "  last pid recv: %u, last ack sent: %u\n",
3377                 prev_fc->last_pid_recv, prev_fc->last_ack_sent);
3378     return;
3379   }
3380
3381   /* Ok, ACK might be necessary, what PID to ACK? */
3382   delta = next_fc->queue_max - next_fc->queue_n;
3383   ack = prev_fc->last_pid_recv + delta;
3384   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ACK %u\n", ack);
3385   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3386               " last pid %u, last ack %u, qmax %u, q %u\n",
3387               prev_fc->last_pid_recv, prev_fc->last_ack_sent,
3388               next_fc->queue_max, next_fc->queue_n);
3389   if (ack == prev_fc->last_ack_sent)
3390   {
3391     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Not sending FWD ACK, not needed\n");
3392     return;
3393   }
3394
3395   prev_fc->last_ack_sent = ack;
3396
3397   /* Build ACK message and send on connection */
3398   msg.header.size = htons (sizeof (msg));
3399   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_ACK);
3400   msg.ack = htonl (ack);
3401   msg.cid = c->id;
3402
3403   send_prebuilt_message_connection (&msg.header, c, NULL, !fwd);
3404 }
3405
3406
3407 /**
3408  * Modify the mesh message TID from global to local and send to client.
3409  * 
3410  * @param ch Channel on which to send the message.
3411  * @param msg Message to modify and send.
3412  * @param c Client to send to.
3413  * @param tid Tunnel ID to use (c can be both owner and client).
3414  */
3415 static void
3416 channel_send_client_to_tid (struct MeshChannel *ch,
3417                              const struct GNUNET_MESH_Data *msg,
3418                              struct MeshClient *c, MESH_ChannelNumber id)
3419 {
3420   struct GNUNET_MESH_LocalData *copy;
3421   uint16_t size = ntohs (msg->header.size) - sizeof (struct GNUNET_MESH_Data);
3422   char cbuf[size + sizeof (struct GNUNET_MESH_LocalData)];
3423
3424   if (size < sizeof (struct GNUNET_MessageHeader))
3425   {
3426     GNUNET_break_op (0);
3427     return;
3428   }
3429   if (NULL == c)
3430   {
3431     GNUNET_break (0);
3432     return;
3433   }
3434   copy = (struct GNUNET_MESH_LocalData *) cbuf;
3435   memcpy (&copy[1], &msg[1], size);
3436   copy->header.size = htons (sizeof (struct GNUNET_MESH_LocalData) + size);
3437   copy->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA);
3438   copy->id = htonl (id);
3439   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
3440                                               &copy->header, GNUNET_NO);
3441 }
3442
3443
3444 /**
3445  * We have received a message out of order, or the client is not ready.
3446  * Buffer it until we receive an ACK from the client or the missing
3447  * message from the channel.
3448  *
3449  * @param msg Message to buffer (MUST be of type MESH_DATA).
3450  * @param rel Reliability data to the corresponding direction.
3451  */
3452 static void
3453 channel_rel_add_buffered_data (const struct GNUNET_MESH_Data *msg,
3454                                struct MeshChannelReliability *rel)
3455 {
3456   struct MeshReliableMessage *copy;
3457   struct MeshReliableMessage *prev;
3458   uint32_t mid;
3459   uint16_t size;
3460   
3461   size = ntohs (msg->header.size);
3462   mid = ntohl (msg->mid);
3463   
3464   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add_buffered_data %u\n", mid);
3465   
3466   copy = GNUNET_malloc (sizeof (*copy) + size);
3467   copy->mid = mid;
3468   copy->rel = rel;
3469   memcpy (&copy[1], msg, size);
3470   
3471   rel->n_recv++;
3472   
3473   // FIXME do something better than O(n), although n < 64...
3474   // FIXME start from the end (most messages are the latest ones)
3475   for (prev = rel->head_recv; NULL != prev; prev = prev->next)
3476   {
3477     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " prev %u\n", prev->mid);
3478     if (GMC_is_pid_bigger (prev->mid, mid))
3479     {
3480       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " bingo!\n");
3481       GNUNET_CONTAINER_DLL_insert_before (rel->head_recv, rel->tail_recv,
3482                                           prev, copy);
3483       return;
3484     }
3485   }
3486     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " insert at tail!\n");
3487     GNUNET_CONTAINER_DLL_insert_tail (rel->head_recv, rel->tail_recv, copy);
3488     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add_buffered_data END\n");
3489 }
3490
3491
3492 static void
3493 channel_send_client_data (struct MeshChannel *ch,
3494                           const struct GNUNET_MESH_Data *msg,
3495                           int fwd)
3496 {
3497   if (fwd)
3498   {
3499     if (ch->dest_rel->client_ready)
3500       channel_send_client_to_tid (ch, msg, ch->dest, ch->lid_dest);
3501     else
3502       channel_rel_add_buffered_data (msg, ch->dest_rel);
3503   }
3504   else
3505   {
3506     if (ch->root_rel->client_ready)
3507       channel_send_client_to_tid (ch, msg, ch->root, ch->lid_root);
3508     else
3509       channel_rel_add_buffered_data (msg, ch->root_rel);
3510   }
3511 }
3512
3513
3514 /**
3515  * Send a buffered message to the client, for in order delivery or
3516  * as result of client ACK.
3517  *
3518  * @param ch Channel on which to empty the message buffer.
3519  * @param c Client to send to.
3520  * @param rel Reliability structure to corresponding peer.
3521  *            If rel == bck_rel, this is FWD data.
3522  */
3523 static void
3524 channel_send_client_buffered_data (struct MeshChannel *ch,
3525                                    struct MeshClient *c,
3526                                    int fwd)
3527 {
3528   struct MeshReliableMessage *copy;
3529   struct MeshChannelReliability *rel;
3530
3531   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_buffered_data\n");
3532   rel = fwd ? ch->dest_rel : ch->root_rel;
3533   if (GNUNET_NO == rel->client_ready)
3534   {
3535     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client not ready\n");
3536     return;
3537   }
3538
3539   copy = rel->head_recv;
3540   /* We never buffer channel management messages */
3541   if (NULL != copy)
3542   {
3543     if (copy->mid == rel->mid_recv || GNUNET_NO == ch->reliable)
3544     {
3545       struct GNUNET_MESH_Data *msg = (struct GNUNET_MESH_Data *) &copy[1];
3546
3547       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3548                   " have %u! now expecting %u\n",
3549                   copy->mid, rel->mid_recv + 1);
3550       channel_send_client_data (ch, msg, fwd);
3551       rel->n_recv--;
3552       rel->mid_recv++;
3553       GNUNET_CONTAINER_DLL_remove (rel->head_recv, rel->tail_recv, copy);
3554       GNUNET_free (copy);
3555     }
3556     else
3557     {
3558       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3559                   " reliable && don't have %u, next is %u\n",
3560                   rel->mid_recv,
3561                   copy->mid);
3562       return;
3563     }
3564   }
3565   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "send_buffered_data END\n");
3566 }
3567
3568
3569 /**
3570  * Destroy a reliable message after it has been acknowledged, either by
3571  * direct mid ACK or bitfield. Updates the appropriate data structures and
3572  * timers and frees all memory.
3573  * 
3574  * @param copy Message that is no longer needed: remote peer got it.
3575  */
3576 static void
3577 rel_message_free (struct MeshReliableMessage *copy)
3578 {
3579   struct MeshChannelReliability *rel;
3580   struct GNUNET_TIME_Relative time;
3581
3582   rel = copy->rel;
3583   time = GNUNET_TIME_absolute_get_duration (copy->timestamp);
3584   rel->expected_delay.rel_value_us *= 7;
3585   rel->expected_delay.rel_value_us += time.rel_value_us;
3586   rel->expected_delay.rel_value_us /= 8;
3587   rel->n_sent--;
3588   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! Freeing %u\n", copy->mid);
3589   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    n_sent %u\n", rel->n_sent);
3590   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  took %s\n",
3591               GNUNET_STRINGS_relative_time_to_string (time, GNUNET_NO));
3592   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  new expected delay %s\n",
3593               GNUNET_STRINGS_relative_time_to_string (rel->expected_delay,
3594                                                       GNUNET_NO));
3595   rel->retry_timer = rel->expected_delay;
3596   GNUNET_CONTAINER_DLL_remove (rel->head_sent, rel->tail_sent, copy);
3597   GNUNET_free (copy);
3598 }
3599
3600
3601 /**
3602  * Destroy all reliable messages queued for a channel,
3603  * during a channel destruction.
3604  * Frees the reliability structure itself.
3605  *
3606  * @param rel Reliability data for a channel.
3607  */
3608 static void
3609 channel_rel_free_all (struct MeshChannelReliability *rel)
3610 {
3611   struct MeshReliableMessage *copy;
3612   struct MeshReliableMessage *next;
3613
3614   if (NULL == rel)
3615     return;
3616
3617   for (copy = rel->head_recv; NULL != copy; copy = next)
3618   {
3619     next = copy->next;
3620     GNUNET_CONTAINER_DLL_remove (rel->head_recv, rel->tail_recv, copy);
3621     GNUNET_free (copy);
3622   }
3623   for (copy = rel->head_sent; NULL != copy; copy = next)
3624   {
3625     next = copy->next;
3626     GNUNET_CONTAINER_DLL_remove (rel->head_sent, rel->tail_sent, copy);
3627     GNUNET_free (copy);
3628   }
3629   if (GNUNET_SCHEDULER_NO_TASK != rel->retry_task)
3630     GNUNET_SCHEDULER_cancel (rel->retry_task);
3631   GNUNET_free (rel);
3632 }
3633
3634
3635 /**
3636  * Mark future messages as ACK'd.
3637  *
3638  * @param rel Reliability data.
3639  * @param msg DataACK message with a bitfield of future ACK'd messages.
3640  */
3641 static void
3642 channel_rel_free_sent (struct MeshChannelReliability *rel,
3643                        const struct GNUNET_MESH_DataACK *msg)
3644 {
3645   struct MeshReliableMessage *copy;
3646   struct MeshReliableMessage *next;
3647   uint64_t bitfield;
3648   uint64_t mask;
3649   uint32_t mid;
3650   uint32_t target;
3651   unsigned int i;
3652
3653   bitfield = msg->futures;
3654   mid = ntohl (msg->mid);
3655   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3656               "free_sent_reliable %u %llX\n",
3657               mid, bitfield);
3658   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3659               " rel %p, head %p\n",
3660               rel, rel->head_sent);
3661   for (i = 0, copy = rel->head_sent;
3662        i < 64 && NULL != copy && 0 != bitfield;
3663        i++)
3664   {
3665     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3666                 " trying bit %u (mid %u)\n",
3667                 i, mid + i + 1);
3668     mask = 0x1LL << i;
3669     if (0 == (bitfield & mask))
3670      continue;
3671
3672     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " set!\n");
3673     /* Bit was set, clear the bit from the bitfield */
3674     bitfield &= ~mask;
3675
3676     /* The i-th bit was set. Do we have that copy? */
3677     /* Skip copies with mid < target */
3678     target = mid + i + 1;
3679     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " target %u\n", target);
3680     while (NULL != copy && GMC_is_pid_bigger (target, copy->mid))
3681      copy = copy->next;
3682
3683     /* Did we run out of copies? (previously freed, it's ok) */
3684     if (NULL == copy)
3685     {
3686      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "run out of copies...\n");
3687      return;
3688     }
3689
3690     /* Did we overshoot the target? (previously freed, it's ok) */
3691     if (GMC_is_pid_bigger (copy->mid, target))
3692     {
3693      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " next copy %u\n", copy->mid);
3694      continue;
3695     }
3696
3697     /* Now copy->mid == target, free it */
3698     next = copy->next;
3699     rel_message_free (copy);
3700     copy = next;
3701   }
3702   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "free_sent_reliable END\n");
3703 }
3704
3705
3706 /**
3707  * We haven't received an ACK after a certain time: restransmit the message.
3708  *
3709  * @param cls Closure (MeshReliableMessage with the message to restransmit)
3710  * @param tc TaskContext.
3711  */
3712 static void
3713 channel_retransmit_message (void *cls,
3714                             const struct GNUNET_SCHEDULER_TaskContext *tc)
3715 {
3716   struct MeshChannelReliability *rel = cls;
3717   struct MeshReliableMessage *copy;
3718   struct MeshPeerQueue *q;
3719   struct MeshChannel *ch;
3720   struct MeshConnection *c;
3721   struct GNUNET_MESH_Data *payload;
3722   struct MeshPeer *hop;
3723   int fwd;
3724
3725   rel->retry_task = GNUNET_SCHEDULER_NO_TASK;
3726   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
3727     return;
3728
3729   ch = rel->ch;
3730   copy = rel->head_sent;
3731   if (NULL == copy)
3732   {
3733     GNUNET_break (0);
3734     return;
3735   }
3736
3737   /* Search the message to be retransmitted in the outgoing queue.
3738    * Check only the queue for the connection that is going to be used,
3739    * if the message is stuck in some other connection's queue we shouldn't
3740    * act upon it:
3741    * - cancelling it and sending the new one doesn't guarantee it's delivery,
3742    *   the old connection could be temporary stalled or the queue happened to
3743    *   be long at time of insertion.
3744    * - not sending the new one could cause terrible delays the old connection
3745    *   is stalled.
3746    */
3747   payload = (struct GNUNET_MESH_Data *) &copy[1];
3748   fwd = (rel == ch->root_rel);
3749   c = tunnel_get_connection (ch->t, fwd);
3750   hop = connection_get_hop (c, fwd);
3751   for (q = hop->queue_head; NULL != q; q = q->next)
3752   {
3753     if (ntohs (payload->header.type) == q->type && ch == q->ch)
3754     {
3755       struct GNUNET_MESH_Data *queued_data = q->cls;
3756
3757       if (queued_data->mid == payload->mid)
3758         break;
3759     }
3760   }
3761
3762   /* Message not found in the queue that we are going to use. */
3763   if (NULL == q)
3764   {
3765     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! RETRANSMIT %u\n", copy->mid);
3766
3767     send_prebuilt_message_channel (&payload->header, ch, fwd);
3768     GNUNET_STATISTICS_update (stats, "# data retransmitted", 1, GNUNET_NO);
3769   }
3770   else
3771   {
3772     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! ALREADY IN QUEUE %u\n", copy->mid);
3773   }
3774
3775   rel->retry_timer = GNUNET_TIME_STD_BACKOFF (rel->retry_timer);
3776   rel->retry_task = GNUNET_SCHEDULER_add_delayed (rel->retry_timer,
3777                                                   &channel_retransmit_message,
3778                                                   cls);
3779 }
3780
3781
3782 /**
3783  * Send ACK on one or more connections due to buffer space to the client.
3784  *
3785  * Iterates all connections of the tunnel and sends ACKs appropriately.
3786  *
3787  * @param ch Channel which has some free buffer space.
3788  * @param fwd Is this in for FWD traffic? (ACK goes dest->root)
3789  */
3790 static void
3791 channel_send_connections_ack (struct MeshChannel *ch,
3792                               unsigned int buffer,
3793                               int fwd)
3794 {
3795   struct MeshTunnel2 *t = ch->t;
3796   struct MeshConnection *c;
3797   struct MeshFlowControl *fc;
3798   uint32_t allowed;
3799   uint32_t to_allow;
3800   uint32_t allow_per_connection;
3801   unsigned int cs;
3802
3803   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3804               "Channel send connection %s ack on %s:%X\n",
3805               fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3806
3807   /* Count connections, how many messages are already allowed */
3808   for (cs = 0, allowed = 0, c = t->connection_head; NULL != c; c = c->next)
3809   {
3810     fc = fwd ? &c->fwd_fc : &c->bck_fc;
3811     if (GMC_is_pid_bigger(fc->last_pid_recv, fc->last_ack_sent))
3812     {
3813       GNUNET_break (0);
3814       continue;
3815     }
3816     allowed += fc->last_ack_sent - fc->last_pid_recv;
3817     cs++;
3818   }
3819
3820   /* Make sure there is no overflow */
3821   if (allowed > buffer)
3822   {
3823     GNUNET_break (0);
3824     return;
3825   }
3826
3827   /* Authorize connections to send more data */
3828   to_allow = buffer - allowed;
3829
3830   for (c = t->connection_head; NULL != c && to_allow > 0; c = c->next)
3831   {
3832     allow_per_connection = to_allow/cs;
3833     to_allow -= allow_per_connection;
3834     cs--;
3835     fc = fwd ? &c->fwd_fc : &c->bck_fc;
3836     if (fc->last_ack_sent - fc->last_pid_recv > 64 / 3)
3837     {
3838       continue;
3839     }
3840     connection_send_ack (c, allow_per_connection, fwd);
3841   }
3842
3843   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3844                 "Channel send connection %s ack on %s:%X\n",
3845                 fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3846   GNUNET_break (to_allow == 0);
3847 }
3848
3849
3850 /**
3851  * Send an ACK on the appropriate connection/channel, depending on
3852  * the direction and the position of the peer.
3853  *
3854  * @param c Which connection to send the hop-by-hop ACK.
3855  * @param ch Channel, if any.
3856  * @param fwd Is this a fwd ACK? (will go dest->root)
3857  */
3858 static void
3859 send_ack (struct MeshConnection *c, struct MeshChannel *ch, int fwd)
3860 {
3861   unsigned int buffer;
3862
3863   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3864               "send ack %s on %p %p\n",
3865               fwd ? "FWD" : "BCK", c, ch);
3866   if (NULL == c || connection_is_terminal (c, fwd))
3867   {
3868     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  getting from Channel\n");
3869     buffer = tunnel_get_buffer (NULL == c ? ch->t : c->t, fwd);
3870   }
3871   else
3872   {
3873     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  getting from Connection\n");
3874     GNUNET_assert (NULL != c);
3875     buffer = connection_get_buffer (c, fwd);
3876   }
3877   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  buffer available: %u\n", buffer);
3878
3879   if (NULL == c)
3880   {
3881     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on all connections\n");
3882     GNUNET_assert (NULL != ch);
3883     channel_send_connections_ack (ch, buffer, fwd);
3884   }
3885   else if (connection_is_origin (c, fwd))
3886   {
3887     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on channel...\n");
3888     if (0 < buffer)
3889     {
3890       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  really sending!\n");
3891       GNUNET_assert (NULL != ch);
3892       send_local_ack (ch, fwd);
3893     }
3894   }
3895   else
3896   {
3897     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on connection\n");
3898     connection_send_ack (c, buffer, fwd);
3899   }
3900 }
3901
3902
3903 /**
3904  * Channel was ACK'd by remote peer, mark as ready and cancel retransmission.
3905  *
3906  * @param ch Channel to mark as ready.
3907  * @param fwd Was the CREATE message sent fwd?
3908  */
3909 static void
3910 channel_confirm (struct MeshChannel *ch, int fwd)
3911 {
3912   struct MeshChannelReliability *rel;
3913   struct MeshReliableMessage *copy;
3914   struct MeshReliableMessage *next;
3915
3916   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3917               "  channel confirm %s %s:%X\n",
3918               fwd ? "FWD" : "BCK", peer2s (ch->t->peer), ch->gid);
3919   ch->state = MESH_CHANNEL_READY;
3920
3921   rel = fwd ? ch->root_rel : ch->dest_rel;
3922   for (copy = rel->head_sent; NULL != copy; copy = next)
3923   {
3924     struct GNUNET_MessageHeader *msg;
3925
3926     next = copy->next;
3927     msg = (struct GNUNET_MessageHeader *) &copy[1];
3928     if (ntohs (msg->type) == GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE)
3929     {
3930       rel_message_free (copy);
3931       /* TODO return? */
3932     }
3933   }
3934   if (GNUNET_NO == rel->client_ready)
3935     send_local_ack (ch, fwd);
3936 }
3937
3938
3939 /**
3940  * Save a copy to retransmit in case it gets lost.
3941  *
3942  * Initializes all needed callbacks and timers.
3943  *
3944  * @param ch Channel this message goes on.
3945  * @param msg Message to copy.
3946  * @param fwd Is this fwd traffic?
3947  */
3948 static void
3949 channel_save_copy (struct MeshChannel *ch,
3950                    const struct GNUNET_MessageHeader *msg,
3951                    int fwd)
3952 {
3953   struct MeshChannelReliability *rel;
3954   struct MeshReliableMessage *copy;
3955   uint32_t mid;
3956   uint16_t type;
3957   uint16_t size;
3958
3959   rel = fwd ? ch->root_rel : ch->dest_rel;
3960   mid = rel->mid_send;
3961   type = ntohs (msg->type);
3962   size = ntohs (msg->size);
3963
3964   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! SAVE %u\n", mid);
3965   copy = GNUNET_malloc (sizeof (struct MeshReliableMessage) + size);
3966   copy->mid = mid;
3967   copy->timestamp = GNUNET_TIME_absolute_get ();
3968   copy->rel = rel;
3969   copy->type = type;
3970   memcpy (&copy[1], msg, size);
3971   rel->n_sent++;
3972   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " n_sent %u\n", rel->n_sent);
3973   GNUNET_CONTAINER_DLL_insert_tail (rel->head_sent, rel->tail_sent, copy);
3974   if (GNUNET_SCHEDULER_NO_TASK == rel->retry_task)
3975   {
3976     rel->retry_timer =
3977         GNUNET_TIME_relative_multiply (rel->expected_delay,
3978                                         MESH_RETRANSMIT_MARGIN);
3979     rel->retry_task =
3980         GNUNET_SCHEDULER_add_delayed (rel->retry_timer,
3981                                       &channel_retransmit_message,
3982                                       rel);
3983   }
3984 }
3985
3986
3987 /**
3988  * Send keepalive packets for a connection.
3989  *
3990  * @param c Connection to keep alive..
3991  * @param fwd Is this a FWD keepalive? (owner -> dest).
3992  */
3993 static void
3994 connection_keepalive (struct MeshConnection *c, int fwd)
3995 {
3996   struct GNUNET_MESH_ConnectionKeepAlive *msg;
3997   size_t size = sizeof (struct GNUNET_MESH_ConnectionKeepAlive);
3998   char cbuf[size];
3999   uint16_t type;
4000
4001   type = fwd ? GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE :
4002                GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE;
4003
4004   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4005               "sending %s keepalive for connection %s[%d]\n",
4006               fwd ? "FWD" : "BCK",
4007               peer2s (c->t->peer),
4008               c->id);
4009
4010   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) cbuf;
4011   msg->header.size = htons (size);
4012   msg->header.type = htons (type);
4013   msg->cid = c->id;
4014
4015   send_prebuilt_message_connection (&msg->header, c, NULL, fwd);
4016 }
4017
4018
4019 /**
4020  * Send CONNECTION_{CREATE/ACK} packets for a connection.
4021  *
4022  * @param c Connection for which to send the message.
4023  * @param fwd If GNUNET_YES, send CREATE, otherwise send ACK.
4024  */
4025 static void
4026 connection_recreate (struct MeshConnection *c, int fwd)
4027 {
4028   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "sending connection recreate\n");
4029   if (fwd)
4030     send_connection_create (c);
4031   else
4032     send_connection_ack (c, GNUNET_NO);
4033 }
4034
4035
4036 /**
4037  * Generic connection timer management.
4038  * Depending on the role of the peer in the connection will send the
4039  * appropriate message (build or keepalive)
4040  *
4041  * @param c Conncetion to maintain.
4042  * @param fwd Is FWD?
4043  */
4044 static void
4045 connection_maintain (struct MeshConnection *c, int fwd)
4046 {
4047   if (MESH_TUNNEL_SEARCHING == c->t->state)
4048   {
4049     /* TODO DHT GET with RO_BART */
4050     return;
4051   }
4052   switch (c->state)
4053   {
4054     case MESH_CONNECTION_NEW:
4055       GNUNET_break (0);
4056     case MESH_CONNECTION_SENT:
4057       connection_recreate (c, fwd);
4058       break;
4059     case MESH_CONNECTION_READY:
4060       connection_keepalive (c, fwd);
4061       break;
4062     default:
4063       break;
4064   }
4065 }
4066
4067
4068 static void
4069 connection_fwd_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4070 {
4071   struct MeshConnection *c = cls;
4072
4073   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4074   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4075     return;
4076
4077   connection_maintain (c, GNUNET_YES);
4078   c->fwd_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
4079                                                           &connection_fwd_keepalive,
4080                                                           c);
4081 }
4082
4083
4084 static void
4085 connection_bck_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
4086 {
4087   struct MeshConnection *c = cls;
4088
4089   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4090   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4091     return;
4092
4093   connection_maintain (c, GNUNET_NO);
4094   c->bck_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
4095                                                           &connection_bck_keepalive,
4096                                                           c);
4097 }
4098
4099
4100 /**
4101  * Send a message to all peers in this connection that the connection
4102  * is no longer valid.
4103  *
4104  * If some peer should not receive the message, it should be zero'ed out
4105  * before calling this function.
4106  *
4107  * @param c The connection whose peers to notify.
4108  */
4109 static void
4110 connection_send_destroy (struct MeshConnection *c)
4111 {
4112   struct GNUNET_MESH_ConnectionDestroy msg;
4113
4114   msg.header.size = htons (sizeof (msg));
4115   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY);;
4116   msg.cid = c->id;
4117   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4118               "  sending connection destroy for connection %s[%X]\n",
4119               peer2s (c->t->peer),
4120               c->id);
4121
4122   if (GNUNET_NO == connection_is_terminal (c, GNUNET_YES))
4123     send_prebuilt_message_connection (&msg.header, c, NULL, GNUNET_YES);
4124   if (GNUNET_NO == connection_is_terminal (c, GNUNET_NO))
4125     send_prebuilt_message_connection (&msg.header, c, NULL, GNUNET_NO);
4126   c->destroy = GNUNET_YES;
4127 }
4128
4129
4130 /**
4131  * Confirm we got a channel create.
4132  *
4133  * @param ch The channel to confirm.
4134  * @param fwd Should we send the ACK fwd?
4135  */
4136 static void
4137 channel_send_ack (struct MeshChannel *ch, int fwd)
4138 {
4139   struct GNUNET_MESH_ChannelManage msg;
4140
4141   msg.header.size = htons (sizeof (msg));
4142   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_ACK);
4143   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4144               "  sending channel %s ack for channel %s:%X\n",
4145               fwd ? "FWD" : "BCK", peer2s (ch->t->peer),
4146               ch->gid);
4147
4148   msg.chid = htonl (ch->gid);
4149   send_prebuilt_message_channel (&msg.header, ch, !fwd);
4150 }
4151
4152
4153 /**
4154  * Send a message to all clients (local and remote) of this channel
4155  * notifying that the channel is no longer valid.
4156  *
4157  * If some peer or client should not receive the message,
4158  * should be zero'ed out before calling this function.
4159  *
4160  * @param ch The channel whose clients to notify.
4161  */
4162 static void
4163 channel_send_destroy (struct MeshChannel *ch)
4164 {
4165   struct GNUNET_MESH_ChannelManage msg;
4166
4167   msg.header.size = htons (sizeof (msg));
4168   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY);
4169   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4170               "  sending channel destroy for channel %s:%X\n",
4171               peer2s (ch->t->peer),
4172               ch->gid);
4173
4174   if (channel_is_terminal (ch, GNUNET_NO))
4175   {
4176     if (NULL != ch->root && GNUNET_NO == ch->root->shutting_down)
4177     {
4178       msg.chid = htonl (ch->lid_root);
4179       send_local_channel_destroy (ch, GNUNET_NO);
4180     }
4181   }
4182   else
4183   {
4184     msg.chid = htonl (ch->gid);
4185     send_prebuilt_message_channel (&msg.header, ch, GNUNET_NO);
4186   }
4187
4188   if (channel_is_terminal (ch, GNUNET_YES))
4189   {
4190     if (NULL != ch->dest && GNUNET_NO == ch->dest->shutting_down)
4191     {
4192       msg.chid = htonl (ch->lid_dest);
4193       send_local_channel_destroy (ch, GNUNET_YES);
4194     }
4195   }
4196   else
4197   {
4198     msg.chid = htonl (ch->gid);
4199     send_prebuilt_message_channel (&msg.header, ch, GNUNET_YES);
4200   }
4201 }
4202
4203
4204 /**
4205  * Create a tunnel.
4206  */
4207 static struct MeshTunnel2 *
4208 tunnel_new (void)
4209 {
4210   struct MeshTunnel2 *t;
4211
4212   t = GNUNET_new (struct MeshTunnel2);
4213   t->next_chid = 0;
4214   t->next_local_chid = GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
4215 //   if (GNUNET_OK !=
4216 //       GNUNET_CONTAINER_multihashmap_put (tunnels, tid, t,
4217 //                                          GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
4218 //   {
4219 //     GNUNET_break (0);
4220 //     tunnel_destroy (t);
4221 //     return NULL;
4222 //   }
4223
4224 //   char salt[] = "salt";
4225 //   GNUNET_CRYPTO_kdf (&t->e_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4226 //                      salt, sizeof (salt),
4227 //                      &t->e_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4228 //                      &my_full_id, sizeof (struct GNUNET_PeerIdentity),
4229 //                      GNUNET_PEER_resolve2 (t->peer->id), sizeof (struct GNUNET_PeerIdentity),
4230 //                      NULL);
4231 //   GNUNET_CRYPTO_kdf (&t->d_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4232 //                      salt, sizeof (salt),
4233 //                      &t->d_key, sizeof (struct GNUNET_CRYPTO_AesSessionKey),
4234 //                      GNUNET_PEER_resolve2 (t->peer->id), sizeof (struct GNUNET_PeerIdentity),
4235 //                      &my_full_id, sizeof (struct GNUNET_PeerIdentity),
4236 //                      NULL);
4237
4238   return t;
4239 }
4240
4241
4242 /**
4243  * Add a connection to a tunnel.
4244  *
4245  * @param t Tunnel.
4246  * @param c Connection.
4247  */
4248 static void
4249 tunnel_add_connection (struct MeshTunnel2 *t, struct MeshConnection *c)
4250 {
4251   struct MeshConnection *aux;
4252   c->t = t;
4253   for (aux = t->connection_head; aux != NULL; aux = aux->next)
4254     if (aux == c)
4255       return;
4256   GNUNET_CONTAINER_DLL_insert_tail (t->connection_head, t->connection_tail, c);
4257 }
4258
4259
4260 /**
4261  * Initialize a Flow Control structure to the initial state.
4262  * 
4263  * @param fc Flow Control structure to initialize.
4264  */
4265 static void
4266 fc_init (struct MeshFlowControl *fc)
4267 {
4268   fc->next_pid = 0;
4269   fc->last_pid_sent = (uint32_t) -1; /* Next (expected) = 0 */
4270   fc->last_pid_recv = (uint32_t) -1;
4271   fc->last_ack_sent = (uint32_t) 0;
4272   fc->last_ack_recv = (uint32_t) 0;
4273   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
4274   fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
4275   fc->queue_n = 0;
4276   fc->queue_max = (max_msgs_queue / max_connections) + 1;
4277 }
4278
4279
4280 static struct MeshConnection *
4281 connection_new (const struct GNUNET_HashCode *cid)
4282 {
4283   struct MeshConnection *c;
4284
4285   c = GNUNET_new (struct MeshConnection);
4286   c->id = *cid;
4287   GNUNET_CONTAINER_multihashmap_put (connections, &c->id, c,
4288                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
4289   fc_init (&c->fwd_fc);
4290   fc_init (&c->bck_fc);
4291   c->fwd_fc.c = c;
4292   c->bck_fc.c = c;
4293
4294   return c;
4295 }
4296
4297
4298 /**
4299  * Find a connection.
4300  *
4301  * @param cid Connection ID.
4302  */
4303 static struct MeshConnection *
4304 connection_get (const struct GNUNET_HashCode *cid)
4305 {
4306   return GNUNET_CONTAINER_multihashmap_get (connections, cid);
4307 }
4308
4309
4310 static void
4311 connection_destroy (struct MeshConnection *c)
4312 {
4313   struct MeshPeer *peer;
4314
4315   if (NULL == c)
4316     return;
4317
4318   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "destroying connection %s[%X]\n",
4319               peer2s (c->t->peer),
4320               c->id);
4321
4322   /* Cancel all traffic */
4323   connection_cancel_queues (c, GNUNET_YES);
4324   connection_cancel_queues (c, GNUNET_NO);
4325
4326   /* Cancel maintainance task (keepalive/timeout) */
4327   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_maintenance_task)
4328     GNUNET_SCHEDULER_cancel (c->fwd_maintenance_task);
4329   if (GNUNET_SCHEDULER_NO_TASK != c->bck_maintenance_task)
4330     GNUNET_SCHEDULER_cancel (c->bck_maintenance_task);
4331
4332   /* Deregister from neighbors */
4333   peer = connection_get_next_hop (c);
4334   if (NULL != peer && NULL != peer->connections)
4335     GNUNET_CONTAINER_multihashmap_remove (peer->connections, &c->id, c);
4336   peer = connection_get_prev_hop (c);
4337   if (NULL != peer && NULL != peer->connections)
4338     GNUNET_CONTAINER_multihashmap_remove (peer->connections, &c->id, c);
4339
4340   /* Delete */
4341   GNUNET_STATISTICS_update (stats, "# connections", -1, GNUNET_NO);
4342   GNUNET_CONTAINER_DLL_remove (c->t->connection_head, c->t->connection_tail, c);
4343   GNUNET_free (c);
4344 }
4345
4346
4347 static void
4348 tunnel_destroy (struct MeshTunnel2 *t)
4349 {
4350   struct MeshConnection *c;
4351   struct MeshConnection *next;
4352
4353   if (NULL == t)
4354     return;
4355
4356   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "destroying tunnel %s\n",
4357               peer2s (t->peer));
4358
4359 //   if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_remove (tunnels, &t->id, t))
4360 //     GNUNET_break (0);
4361
4362   for (c = t->connection_head; NULL != c; c = next)
4363   {
4364     next = c->next;
4365     connection_destroy (c);
4366   }
4367
4368   GNUNET_STATISTICS_update (stats, "# tunnels", -1, GNUNET_NO);
4369
4370   GNUNET_free (t);
4371 }
4372
4373
4374 /**
4375  * Tunnel is empty: destroy it.
4376  *
4377  * Notifies all connections about the destruction.
4378  *
4379  * @param t Tunnel to destroy. 
4380  */
4381 static void
4382 tunnel_destroy_empty (struct MeshTunnel2 *t)
4383 {
4384   struct MeshConnection *c;
4385
4386   for (c = t->connection_head; NULL != c; c = c->next)
4387   {
4388     if (GNUNET_NO == c->destroy)
4389       connection_send_destroy (c);
4390   }
4391
4392   if (0 == t->pending_messages)
4393     tunnel_destroy (t);
4394   else
4395     t->destroy = GNUNET_YES;
4396 }
4397
4398
4399 /**
4400  * Destroy tunnel if empty (no more channels).
4401  *
4402  * @param t Tunnel to destroy if empty.
4403  */
4404 static void
4405 tunnel_destroy_if_empty (struct MeshTunnel2 *t)
4406 {
4407   if (NULL != t->channel_head)
4408     return;
4409
4410   tunnel_destroy_empty (t);
4411 }
4412
4413
4414 /**
4415  * Destroy a channel and free all resources.
4416  * 
4417  * @param ch Channel to destroy.
4418  */
4419 static void
4420 channel_destroy (struct MeshChannel *ch)
4421 {
4422   struct MeshClient *c;
4423
4424   if (NULL == ch)
4425     return;
4426
4427   c = ch->root;
4428   if (NULL != c)
4429   {
4430     if (GNUNET_YES != GNUNET_CONTAINER_multihashmap32_remove (c->own_channels,
4431                                                               ch->lid_root, ch))
4432     {
4433       GNUNET_break (0);
4434     }
4435   }
4436
4437   c = ch->dest;
4438   if (NULL != c)
4439   {
4440     if (GNUNET_YES !=
4441         GNUNET_CONTAINER_multihashmap32_remove (c->incoming_channels,
4442                                                 ch->lid_dest, ch))
4443     {
4444       GNUNET_break (0);
4445     }
4446   }
4447
4448   channel_rel_free_all (ch->root_rel);
4449   channel_rel_free_all (ch->dest_rel);
4450
4451   GNUNET_CONTAINER_DLL_remove (ch->t->channel_head, ch->t->channel_tail, ch);
4452   GNUNET_STATISTICS_update (stats, "# channels", -1, GNUNET_NO);
4453
4454   GNUNET_free (ch);
4455 }
4456
4457 /**
4458  * Create a new channel.
4459  *
4460  * @param t Tunnel this channel is in.
4461  * @param owner Client that owns the channel, NULL for foreign channels.
4462  * @param lid_root Local ID for root client.
4463  *
4464  * @return A new initialized channel. NULL on error.
4465  */
4466 static struct MeshChannel *
4467 channel_new (struct MeshTunnel2 *t,
4468              struct MeshClient *owner, MESH_ChannelNumber lid_root)
4469 {
4470   struct MeshChannel *ch;
4471
4472   ch = GNUNET_new (struct MeshChannel);
4473   ch->root = owner;
4474   ch->lid_root = lid_root;
4475   ch->t = t;
4476
4477   GNUNET_CONTAINER_DLL_insert (t->channel_head, t->channel_tail, ch);
4478
4479   GNUNET_STATISTICS_update (stats, "# channels", 1, GNUNET_NO);
4480
4481   if (NULL != owner)
4482   {
4483     while (NULL != channel_get (t, t->next_chid))
4484       t->next_chid = (t->next_chid + 1) & ~GNUNET_MESH_LOCAL_CHANNEL_ID_CLI;
4485     ch->gid = t->next_chid;
4486     t->next_chid = (t->next_chid + 1) & ~GNUNET_MESH_LOCAL_CHANNEL_ID_CLI;
4487
4488     if(GNUNET_OK !=
4489        GNUNET_CONTAINER_multihashmap32_put (owner->own_channels, lid_root, ch,
4490                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY))
4491     {
4492       GNUNET_break (0);
4493       channel_destroy (ch);
4494       GNUNET_SERVER_receive_done (owner->handle, GNUNET_SYSERR);
4495       return NULL;
4496     }
4497   }
4498
4499   return ch;
4500 }
4501
4502
4503 /**
4504  * Set options in a channel, extracted from a bit flag field
4505  * 
4506  * @param ch Channel to set options to.
4507  * @param options Bit array in host byte order.
4508  */
4509 static void
4510 channel_set_options (struct MeshChannel *ch, uint32_t options)
4511 {
4512   ch->nobuffer = (options & GNUNET_MESH_OPTION_NOBUFFER) != 0 ?
4513                  GNUNET_YES : GNUNET_NO;
4514   ch->reliable = (options & GNUNET_MESH_OPTION_RELIABLE) != 0 ?
4515                  GNUNET_YES : GNUNET_NO;
4516 }
4517
4518
4519 /**
4520  * Iterator for deleting each channel whose client endpoint disconnected.
4521  *
4522  * @param cls Closure (client that has disconnected).
4523  * @param key The local channel id (used to access the hashmap).
4524  * @param value The value stored at the key (channel to destroy).
4525  *
4526  * @return GNUNET_OK, keep iterating.
4527  */
4528 static int
4529 channel_destroy_iterator (void *cls,
4530                           uint32_t key,
4531                           void *value)
4532 {
4533   struct MeshChannel *ch = value;
4534   struct MeshClient *c = cls;
4535   struct MeshTunnel2 *t;
4536
4537   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4538               " Channel %X (%X / %X) destroy, due to client %u shutdown.\n",
4539               ch->gid, ch->lid_root, ch->lid_dest, c->id);
4540
4541   if (c == ch->dest)
4542   {
4543     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is destination.\n", c->id);
4544   }
4545   if (c == ch->root)
4546   {
4547     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " Client %u is owner.\n", c->id);
4548   }
4549
4550   t = ch->t;
4551   channel_send_destroy (ch);
4552   channel_destroy (ch);
4553   tunnel_destroy_if_empty (t);
4554
4555   return GNUNET_OK;
4556 }
4557
4558
4559 /**
4560  * Remove client's ports from the global hashmap on disconnect.
4561  *
4562  * @param cls Closure (unused).
4563  * @param key Port.
4564  * @param value Client structure.
4565  *
4566  * @return GNUNET_OK, keep iterating.
4567  */
4568 static int
4569 client_release_ports (void *cls,
4570                       uint32_t key,
4571                       void *value)
4572 {
4573   int res;
4574
4575   res = GNUNET_CONTAINER_multihashmap32_remove (ports, key, value);
4576   if (GNUNET_YES != res)
4577   {
4578     GNUNET_break (0);
4579     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4580                 "Port %u by client %p was not registered.\n",
4581                 key, value);
4582   }
4583   return GNUNET_OK;
4584 }
4585
4586
4587 /**
4588  * Timeout function due to lack of keepalive/traffic from the owner.
4589  * Destroys connection if called.
4590  *
4591  * @param cls Closure (connection to destroy).
4592  * @param tc TaskContext.
4593  */
4594 static void
4595 connection_fwd_timeout (void *cls,
4596                         const struct GNUNET_SCHEDULER_TaskContext *tc)
4597 {
4598   struct MeshConnection *c = cls;
4599
4600   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4601   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4602     return;
4603   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4604               "Connection %s[%X] FWD timed out. Destroying.\n",
4605               peer2s (c->t->peer),
4606               c->id);
4607
4608   if (connection_is_origin (c, GNUNET_YES)) /* If local, leave. */
4609     return;
4610
4611   connection_destroy (c);
4612 }
4613
4614
4615 /**
4616  * Timeout function due to lack of keepalive/traffic from the destination.
4617  * Destroys connection if called.
4618  *
4619  * @param cls Closure (connection to destroy).
4620  * @param tc TaskContext
4621  */
4622 static void
4623 connection_bck_timeout (void *cls,
4624                         const struct GNUNET_SCHEDULER_TaskContext *tc)
4625 {
4626   struct MeshConnection *c = cls;
4627
4628   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
4629   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
4630     return;
4631
4632   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4633               "Connection %s[%X] FWD timed out. Destroying.\n",
4634               peer2s (c->t->peer),
4635               c->id);
4636
4637   if (connection_is_origin (c, GNUNET_NO)) /* If local, leave. */
4638     return;
4639
4640   connection_destroy (c);
4641 }
4642
4643
4644 /**
4645  * Resets the connection timeout task, some other message has done the
4646  * task's job.
4647  * - For the first peer on the direction this means to send
4648  *   a keepalive or a path confirmation message (either create or ACK).
4649  * - For all other peers, this means to destroy the connection,
4650  *   due to lack of activity.
4651  * Starts the tiemout if no timeout was running (connection just created).
4652  *
4653  * @param c Connection whose timeout to reset.
4654  * @param fwd Is this forward?
4655  *
4656  * TODO use heap to improve efficiency of scheduler.
4657  */
4658 static void
4659 connection_reset_timeout (struct MeshConnection *c, int fwd)
4660 {
4661   GNUNET_SCHEDULER_TaskIdentifier *ti;
4662   GNUNET_SCHEDULER_Task f;
4663
4664   ti = fwd ? &c->fwd_maintenance_task : &c->bck_maintenance_task;
4665
4666   if (GNUNET_SCHEDULER_NO_TASK != *ti)
4667     GNUNET_SCHEDULER_cancel (*ti);
4668
4669   if (connection_is_origin (c, fwd)) /* Endpoint */
4670   {
4671     f  = fwd ? &connection_fwd_keepalive : &connection_bck_keepalive;
4672     *ti = GNUNET_SCHEDULER_add_delayed (refresh_connection_time, f, c);
4673   }
4674   else /* Relay */
4675   {
4676     struct GNUNET_TIME_Relative delay;
4677
4678     delay = GNUNET_TIME_relative_multiply (refresh_connection_time, 4);
4679     f  = fwd ? &connection_fwd_timeout : &connection_bck_timeout;
4680     *ti = GNUNET_SCHEDULER_add_delayed (delay, f, c);
4681   }
4682 }
4683
4684
4685 /**
4686  * Iterator to notify all connections of a broken link. Mark connections
4687  * to destroy after all traffic has been sent.
4688  *
4689  * @param cls Closure (peer disconnected).
4690  * @param key Current key code (tid).
4691  * @param value Value in the hash map (connection).
4692  *
4693  * @return GNUNET_YES if we should continue to iterate,
4694  *         GNUNET_NO if not.
4695  */
4696 static int
4697 connection_broken (void *cls,
4698                    const struct GNUNET_HashCode *key,
4699                    void *value)
4700 {
4701   struct MeshPeer *peer = cls;
4702   struct MeshConnection *c = value;
4703   struct GNUNET_MESH_ConnectionBroken msg;
4704   int fwd;
4705
4706   fwd = peer == connection_get_prev_hop (c);
4707
4708   connection_cancel_queues (c, !fwd);
4709   if (connection_is_terminal (c, fwd))
4710   {
4711     /* Local shutdown, no one to notify about this. */
4712     connection_destroy (c);
4713     return GNUNET_YES;
4714   }
4715
4716   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectionBroken));
4717   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN);
4718   msg.cid = c->id;
4719   msg.peer1 = my_full_id;
4720   msg.peer2 = *GNUNET_PEER_resolve2 (peer->id);
4721   send_prebuilt_message_connection (&msg.header, c, NULL, fwd);
4722   c->destroy = GNUNET_YES;
4723
4724   return GNUNET_YES;
4725 }
4726
4727 /******************************************************************************/
4728 /****************      MESH NETWORK HANDLER HELPERS     ***********************/
4729 /******************************************************************************/
4730
4731 /**
4732  * Free a transmission that was already queued with all resources
4733  * associated to the request.
4734  *
4735  * @param queue Queue handler to cancel.
4736  * @param clear_cls Is it necessary to free associated cls?
4737  */
4738 static void
4739 queue_destroy (struct MeshPeerQueue *queue, int clear_cls)
4740 {
4741   struct MeshPeer *peer;
4742   struct MeshFlowControl *fc;
4743   int fwd;
4744
4745   fwd = queue->fwd;
4746   peer = queue->peer;
4747   GNUNET_assert (NULL != queue->c);
4748   fc = fwd ? &queue->c->fwd_fc : &queue->c->bck_fc;
4749
4750   if (GNUNET_YES == clear_cls)
4751   {
4752     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   queue destroy type %s\n",
4753                 GNUNET_MESH_DEBUG_M2S (queue->type));
4754     switch (queue->type)
4755     {
4756       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
4757       case GNUNET_MESSAGE_TYPE_MESH_TUNNEL_DESTROY:
4758         GNUNET_log (GNUNET_ERROR_TYPE_INFO, "destroying a DESTROY message\n");
4759         GNUNET_break (GNUNET_YES == queue->c->destroy);
4760         /* fall through */
4761       case GNUNET_MESSAGE_TYPE_MESH_FWD:
4762       case GNUNET_MESSAGE_TYPE_MESH_BCK:
4763       case GNUNET_MESSAGE_TYPE_MESH_ACK:
4764       case GNUNET_MESSAGE_TYPE_MESH_POLL:
4765       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
4766       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
4767       case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
4768         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   prebuilt message\n");;
4769         GNUNET_free_non_null (queue->cls);
4770         break;
4771
4772       default:
4773         GNUNET_break (0);
4774         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "   type %s unknown!\n",
4775                     GNUNET_MESH_DEBUG_M2S (queue->type));
4776     }
4777
4778   }
4779   GNUNET_CONTAINER_DLL_remove (peer->queue_head, peer->queue_tail, queue);
4780
4781   if (queue->type != GNUNET_MESSAGE_TYPE_MESH_ACK &&
4782       queue->type != GNUNET_MESSAGE_TYPE_MESH_POLL)
4783   {
4784     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Q_N- %p %u\n", fc, fc->queue_n);
4785     fc->queue_n--;
4786     peer->queue_n--;
4787   }
4788   if (NULL != queue->c)
4789   {
4790     queue->c->pending_messages--;
4791     if (NULL != queue->c->t)
4792     {
4793       queue->c->t->pending_messages--;
4794     }
4795   }
4796
4797   GNUNET_free (queue);
4798 }
4799
4800
4801 static size_t
4802 queue_send (void *cls, size_t size, void *buf)
4803 {
4804   struct MeshPeer *peer = cls;
4805   struct MeshFlowControl *fc;
4806   struct MeshConnection *c;
4807   struct GNUNET_MessageHeader *msg;
4808   struct MeshPeerQueue *queue;
4809   struct MeshTunnel2 *t;
4810   struct MeshChannel *ch;
4811   const struct GNUNET_PeerIdentity *dst_id;
4812   size_t data_size;
4813   uint32_t pid;
4814   uint16_t type;
4815   int fwd;
4816
4817   peer->core_transmit = NULL;
4818   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Queue send (max %u)\n", size);
4819
4820   if (NULL == buf || 0 == size)
4821   {
4822     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "* Buffer size 0.\n");
4823     return 0;
4824   }
4825
4826   /* Initialize */
4827   queue = peer_get_first_message (peer);
4828   if (NULL == queue)
4829   {
4830     GNUNET_break (0); /* Core tmt_rdy should've been canceled */
4831     return 0;
4832   }
4833   c = queue->c;
4834   fwd = queue->fwd;
4835   fc = fwd ? &c->fwd_fc : &c->bck_fc;
4836
4837
4838   dst_id = GNUNET_PEER_resolve2 (peer->id);
4839   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   towards %s\n", GNUNET_i2s (dst_id));
4840   /* Check if buffer size is enough for the message */
4841   if (queue->size > size)
4842   {
4843       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   not enough room, reissue\n");
4844       peer->core_transmit =
4845           GNUNET_CORE_notify_transmit_ready (core_handle,
4846                                              GNUNET_NO,
4847                                              0,
4848                                              GNUNET_TIME_UNIT_FOREVER_REL,
4849                                              dst_id,
4850                                              queue->size,
4851                                              &queue_send,
4852                                              peer);
4853       return 0;
4854   }
4855   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   size %u ok\n", queue->size);
4856
4857   t = (NULL != c) ? c->t : NULL;
4858   type = 0;
4859
4860   /* Fill buf */
4861   switch (queue->type)
4862   {
4863     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
4864     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
4865     case GNUNET_MESSAGE_TYPE_MESH_FWD:
4866     case GNUNET_MESSAGE_TYPE_MESH_BCK:
4867     case GNUNET_MESSAGE_TYPE_MESH_ACK:
4868     case GNUNET_MESSAGE_TYPE_MESH_POLL:
4869       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4870                   "*   raw: %s\n",
4871                   GNUNET_MESH_DEBUG_M2S (queue->type));
4872       data_size = send_core_data_raw (queue->cls, size, buf);
4873       msg = (struct GNUNET_MessageHeader *) buf;
4874       type = ntohs (msg->type);
4875       break;
4876     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
4877       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path create\n");
4878       if (connection_is_origin (c, GNUNET_YES))
4879         data_size = send_core_connection_create (queue->c, size, buf);
4880       else
4881         data_size = send_core_data_raw (queue->cls, size, buf);
4882       break;
4883     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
4884       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   path ack\n");
4885       if (connection_is_origin (c, GNUNET_NO) ||
4886           connection_is_origin (c, GNUNET_YES))
4887         data_size = send_core_connection_ack (queue->c, size, buf);
4888       else
4889         data_size = send_core_data_raw (queue->cls, size, buf);
4890       break;
4891     case GNUNET_MESSAGE_TYPE_MESH_DATA:
4892     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
4893     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
4894       /* This should be encapsulted */
4895       GNUNET_break (0);
4896       data_size = 0;
4897       break;
4898     default:
4899       GNUNET_break (0);
4900       GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "*   type unknown: %u\n",
4901                   queue->type);
4902       data_size = 0;
4903   }
4904
4905   if (0 < drop_percent &&
4906       GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_WEAK, 101) < drop_percent)
4907   {
4908     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
4909                 "Dropping message of type %s\n",
4910                 GNUNET_MESH_DEBUG_M2S (queue->type));
4911     data_size = 0;
4912   }
4913
4914   /* Free queue, but cls was freed by send_core_* */
4915   ch = queue->ch;
4916   queue_destroy (queue, GNUNET_NO);
4917
4918   /* Send ACK if needed, after accounting for sent ID in fc->queue_n */
4919   switch (type)
4920   {
4921     case GNUNET_MESSAGE_TYPE_MESH_FWD:
4922     case GNUNET_MESSAGE_TYPE_MESH_BCK:
4923       pid = ntohl ( ((struct GNUNET_MESH_Encrypted *) buf)->pid );
4924       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   accounting pid %u\n", pid);
4925       fc->last_pid_sent = pid;
4926       send_ack (c, ch, fwd);
4927       break;
4928     default:
4929       break;
4930   }
4931
4932   /* If more data in queue, send next */
4933   queue = peer_get_first_message (peer);
4934   if (NULL != queue)
4935   {
4936     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   more data!\n");
4937     if (NULL == peer->core_transmit) {
4938       peer->core_transmit =
4939           GNUNET_CORE_notify_transmit_ready(core_handle,
4940                                             0,
4941                                             0,
4942                                             GNUNET_TIME_UNIT_FOREVER_REL,
4943                                             dst_id,
4944                                             queue->size,
4945                                             &queue_send,
4946                                             peer);
4947     }
4948     else
4949     {
4950       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
4951                   "*   tmt rdy called somewhere else\n");
4952     }
4953     if (GNUNET_SCHEDULER_NO_TASK == fc->poll_task)
4954     {
4955       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*   starting poll timeout\n");
4956       fc->poll_task =
4957           GNUNET_SCHEDULER_add_delayed (fc->poll_time, &connection_poll, fc);
4958     }
4959   }
4960   else
4961   {
4962     if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
4963     {
4964       GNUNET_SCHEDULER_cancel (fc->poll_task);
4965       fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
4966     }
4967   }
4968   if (NULL != c)
4969   {
4970     c->pending_messages--;
4971     if (GNUNET_YES == c->destroy && 0 == c->pending_messages)
4972     {
4973       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  destroying connection!\n");
4974       connection_destroy (c);
4975     }
4976   }
4977
4978   if (NULL != t)
4979   {
4980     t->pending_messages--;
4981     if (GNUNET_YES == t->destroy && 0 == t->pending_messages)
4982     {
4983 //       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  destroying tunnel!\n");
4984       tunnel_destroy (t);
4985     }
4986   }
4987   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "*  Return %d\n", data_size);
4988   return data_size;
4989 }
4990
4991
4992 static void
4993 queue_add (void *cls, uint16_t type, size_t size,
4994            struct MeshConnection *c,
4995            struct MeshChannel *ch,
4996            int fwd)
4997 {
4998   struct MeshPeerQueue *queue;
4999   struct MeshFlowControl *fc;
5000   struct MeshPeer *peer;
5001   int priority;
5002   int call_core;
5003
5004   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5005               "queue add %s %s (%u) on c %p, ch %p\n",
5006               fwd ? "FWD" : "BCK",  GNUNET_MESH_DEBUG_M2S (type), size, c, ch);
5007   GNUNET_assert (NULL != c);
5008
5009   fc   = fwd ? &c->fwd_fc : &c->bck_fc;
5010   peer = fwd ? connection_get_next_hop (c) : connection_get_prev_hop (c);
5011
5012   if (NULL == fc)
5013   {
5014     GNUNET_break (0);
5015     return;
5016   }
5017
5018   priority = 0;
5019
5020   if (GNUNET_MESSAGE_TYPE_MESH_POLL == type ||
5021       GNUNET_MESSAGE_TYPE_MESH_ACK == type)
5022   {
5023     priority = 100;
5024   }
5025
5026   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "priority %d\n", priority);
5027   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "fc %p\n", fc);
5028   if (fc->queue_n >= fc->queue_max && 0 == priority)
5029   {
5030     GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)",
5031                               1, GNUNET_NO);
5032     GNUNET_break (0);
5033     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5034                 "queue full: %u/%u\n",
5035                 fc->queue_n, fc->queue_max);
5036     return; /* Drop this message */
5037   }
5038
5039   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "last pid %u\n", fc->last_pid_sent);
5040   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "     ack %u\n", fc->last_ack_recv);
5041   if (GMC_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv))
5042   {
5043     call_core = GNUNET_NO;
5044     if (GNUNET_SCHEDULER_NO_TASK == fc->poll_task &&
5045         GNUNET_MESSAGE_TYPE_MESH_POLL != type)
5046     {
5047       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5048                   "no buffer space (%u > %u): starting poll\n",
5049                   fc->last_pid_sent + 1, fc->last_ack_recv);
5050       fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
5051                                                     &connection_poll,
5052                                                     fc);
5053     }
5054   }
5055   else
5056     call_core = GNUNET_YES;
5057   queue = GNUNET_malloc (sizeof (struct MeshPeerQueue));
5058   queue->cls = cls;
5059   queue->type = type;
5060   queue->size = size;
5061   queue->peer = peer;
5062   queue->c = c;
5063   queue->ch = ch;
5064   queue->fwd = fwd;
5065   if (100 <= priority)
5066   {
5067     GNUNET_CONTAINER_DLL_insert (peer->queue_head, peer->queue_tail, queue);
5068   }
5069   else
5070   {
5071     GNUNET_CONTAINER_DLL_insert_tail (peer->queue_head, peer->queue_tail, queue);
5072     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Q_N+ %p %u\n", fc, fc->queue_n);
5073     fc->queue_n++;
5074     peer->queue_n++;
5075   }
5076
5077   if (NULL == peer->core_transmit && GNUNET_YES == call_core)
5078   {
5079     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5080                 "calling core tmt rdy towards %s for %u bytes\n",
5081                 peer2s (peer), size);
5082     peer->core_transmit =
5083         GNUNET_CORE_notify_transmit_ready (core_handle,
5084                                            0,
5085                                            0,
5086                                            GNUNET_TIME_UNIT_FOREVER_REL,
5087                                            GNUNET_PEER_resolve2 (peer->id),
5088                                            size,
5089                                            &queue_send,
5090                                            peer);
5091   }
5092   else
5093   {
5094     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5095                 "core tmt rdy towards %s already called\n",
5096                 peer2s (peer));
5097
5098   }
5099   c->pending_messages++;
5100   if (NULL != c->t)
5101     c->t->pending_messages++;
5102 }
5103
5104
5105 /******************************************************************************/
5106 /********************      MESH NETWORK HANDLERS     **************************/
5107 /******************************************************************************/
5108
5109
5110 /**
5111  * Generic handler for mesh network payload traffic.
5112  *
5113  * @param t Tunnel on which we got this message.
5114  * @param message Unencryted data message.
5115  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5116  */
5117 static void
5118 handle_data (struct MeshTunnel2 *t, const struct GNUNET_MESH_Data *msg, int fwd)
5119 {
5120   struct MeshChannelReliability *rel;
5121   struct MeshChannel *ch;
5122   struct MeshClient *c;
5123   uint32_t mid;
5124   uint16_t type;
5125   size_t size;
5126
5127   /* Check size */
5128   size = ntohs (msg->header.size);
5129   if (size <
5130       sizeof (struct GNUNET_MESH_Data) +
5131       sizeof (struct GNUNET_MessageHeader))
5132   {
5133     GNUNET_break (0);
5134     return;
5135   }
5136   type = ntohs (msg->header.type);
5137   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a %s message\n",
5138               GNUNET_MESH_DEBUG_M2S (type));
5139   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " payload of type %s\n",
5140               GNUNET_MESH_DEBUG_M2S (ntohs (msg[1].header.type)));
5141
5142   /* Check channel */
5143   ch = channel_get (t, ntohl (msg->chid));
5144   if (NULL == ch)
5145   {
5146     GNUNET_STATISTICS_update (stats, "# data on unknown channel", 1, GNUNET_NO);
5147     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "WARNING channel unknown\n");
5148     return;
5149   }
5150
5151   /*  Initialize FWD/BCK data */
5152   c        = fwd ? ch->dest     : ch->root;
5153   rel      = fwd ? ch->dest_rel : ch->root_rel;
5154
5155   if (NULL == c)
5156   {
5157     GNUNET_break (0);
5158     return;
5159   }
5160
5161   tunnel_change_state (t, MESH_TUNNEL_READY);
5162
5163   GNUNET_STATISTICS_update (stats, "# data received", 1, GNUNET_NO);
5164
5165   mid = ntohl (msg->mid);
5166   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " mid %u\n", mid);
5167
5168   if (GNUNET_NO == ch->reliable ||
5169       ( !GMC_is_pid_bigger (rel->mid_recv, mid) &&
5170         GMC_is_pid_bigger (rel->mid_recv + 64, mid) ) )
5171   {
5172     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! RECV %u\n", mid);
5173     if (GNUNET_YES == ch->reliable)
5174     {
5175       /* Is this the exact next expected messasge? */
5176       if (mid == rel->mid_recv)
5177       {
5178         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "as expected\n");
5179         rel->mid_recv++;
5180         channel_send_client_data (ch, msg, fwd);
5181       }
5182       else
5183       {
5184         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "save for later\n");
5185         channel_rel_add_buffered_data (msg, rel);
5186       }
5187     }
5188     else
5189     {
5190       /* Tunnel is unreliable: send to clients directly */
5191       /* FIXME: accept Out Of Order traffic */
5192       rel->mid_recv = mid + 1;
5193       channel_send_client_data (ch, msg, fwd);
5194     }
5195   }
5196   else
5197   {
5198     GNUNET_break_op (0);
5199     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5200                 " MID %u not expected (%u - %u), dropping!\n",
5201                 mid, rel->mid_recv, rel->mid_recv + 64);
5202   }
5203
5204   channel_send_data_ack (ch, fwd);
5205 }
5206
5207 /**
5208  * Handler for mesh network traffic end-to-end ACKs.
5209  *
5210  * @param t Tunnel on which we got this message.
5211  * @param message Data message.
5212  * @param fwd Is this a fwd ACK? (dest->orig)
5213  */
5214 static void
5215 handle_data_ack (struct MeshTunnel2 *t,
5216                  const struct GNUNET_MESH_DataACK *msg, int fwd)
5217 {
5218   struct MeshChannelReliability *rel;
5219   struct MeshReliableMessage *copy;
5220   struct MeshReliableMessage *next;
5221   struct MeshChannel *ch;
5222   uint32_t ack;
5223   uint16_t type;
5224   int work;
5225
5226   type = ntohs (msg->header.type);
5227   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a %s message!\n",
5228               GNUNET_MESH_DEBUG_M2S (type));
5229   ch = channel_get (t, ntohl (msg->chid));
5230   if (NULL == ch)
5231   {
5232     GNUNET_STATISTICS_update (stats, "# ack on unknown channel", 1, GNUNET_NO);
5233     return;
5234   }
5235   ack = ntohl (msg->mid);
5236   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! %s ACK %u\n",
5237               (GNUNET_YES == fwd) ? "FWD" : "BCK", ack);
5238
5239   if (GNUNET_YES == fwd)
5240   {
5241     rel = ch->root_rel;
5242   }
5243   else
5244   {
5245     rel = ch->dest_rel;
5246   }
5247   if (NULL == rel)
5248   {
5249     GNUNET_break (0);
5250     return;
5251   }
5252
5253   for (work = GNUNET_NO, copy = rel->head_sent; copy != NULL; copy = next)
5254   {
5255     if (GMC_is_pid_bigger (copy->mid, ack))
5256     {
5257       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  head %u, out!\n", copy->mid);
5258       channel_rel_free_sent (rel, msg);
5259       break;
5260     }
5261     work = GNUNET_YES;
5262     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!!  id %u\n", copy->mid);
5263     next = copy->next;
5264     rel_message_free (copy);
5265   }
5266   /* ACK client if needed */
5267 //   channel_send_ack (t, type, GNUNET_MESSAGE_TYPE_MESH_UNICAST_ACK == type);
5268
5269   /* If some message was free'd, update the retransmission delay*/
5270   if (GNUNET_YES == work)
5271   {
5272     if (GNUNET_SCHEDULER_NO_TASK != rel->retry_task)
5273     {
5274       GNUNET_SCHEDULER_cancel (rel->retry_task);
5275       if (NULL == rel->head_sent)
5276       {
5277         rel->retry_task = GNUNET_SCHEDULER_NO_TASK;
5278       }
5279       else
5280       {
5281         struct GNUNET_TIME_Absolute new_target;
5282         struct GNUNET_TIME_Relative delay;
5283
5284         delay = GNUNET_TIME_relative_multiply (rel->retry_timer,
5285                                                MESH_RETRANSMIT_MARGIN);
5286         new_target = GNUNET_TIME_absolute_add (rel->head_sent->timestamp,
5287                                                delay);
5288         delay = GNUNET_TIME_absolute_get_remaining (new_target);
5289         rel->retry_task =
5290             GNUNET_SCHEDULER_add_delayed (delay,
5291                                           &channel_retransmit_message,
5292                                           rel);
5293       }
5294     }
5295     else
5296       GNUNET_break (0);
5297   }
5298 }
5299
5300
5301 /**
5302  * Core handler for connection creation.
5303  *
5304  * @param cls Closure (unused).
5305  * @param peer Sender (neighbor).
5306  * @param message Message.
5307  *
5308  * @return GNUNET_OK to keep the connection open,
5309  *         GNUNET_SYSERR to close it (signal serious error)
5310  */
5311 static int
5312 handle_mesh_connection_create (void *cls,
5313                                const struct GNUNET_PeerIdentity *peer,
5314                                const struct GNUNET_MessageHeader *message)
5315 {
5316   struct GNUNET_MESH_ConnectionCreate *msg;
5317   struct GNUNET_PeerIdentity *id;
5318   struct GNUNET_HashCode *cid;
5319   struct MeshPeerPath *path;
5320   struct MeshPeer *dest_peer;
5321   struct MeshPeer *orig_peer;
5322   struct MeshConnection *c;
5323   unsigned int own_pos;
5324   uint16_t size;
5325   uint16_t i;
5326
5327   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5328   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received a connection create msg\n");
5329
5330   /* Check size */
5331   size = ntohs (message->size);
5332   if (size < sizeof (struct GNUNET_MESH_ConnectionCreate))
5333   {
5334     GNUNET_break_op (0);
5335     return GNUNET_OK;
5336   }
5337
5338   /* Calculate hops */
5339   size -= sizeof (struct GNUNET_MESH_ConnectionCreate);
5340   if (size % sizeof (struct GNUNET_PeerIdentity))
5341   {
5342     GNUNET_break_op (0);
5343     return GNUNET_OK;
5344   }
5345   size /= sizeof (struct GNUNET_PeerIdentity);
5346   if (1 > size)
5347   {
5348     GNUNET_break_op (0);
5349     return GNUNET_OK;
5350   }
5351   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
5352
5353   /* Get parameters */
5354   msg = (struct GNUNET_MESH_ConnectionCreate *) message;
5355   cid = &msg->cid;
5356   id = (struct GNUNET_PeerIdentity *) &msg[1];
5357   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5358               "    connection %s (%s).\n",
5359               GNUNET_h2s (cid), GNUNET_i2s (id));
5360
5361   /* Create connection */
5362   c = connection_get (cid);
5363   if (NULL == c)
5364   {
5365     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating connection\n");
5366     c = connection_new (cid);
5367     if (NULL == c)
5368       return GNUNET_OK;  connection_reset_timeout (c, GNUNET_YES);
5369     tunnel_change_state (c->t,  MESH_TUNNEL_WAITING);
5370
5371     /* Create path */
5372     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
5373     path = path_new (size);
5374     own_pos = 0;
5375     for (i = 0; i < size; i++)
5376     {
5377       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ... adding %s\n",
5378                   GNUNET_i2s (&id[i]));
5379       path->peers[i] = GNUNET_PEER_intern (&id[i]);
5380       if (path->peers[i] == myid)
5381         own_pos = i;
5382     }
5383     if (own_pos == 0 && path->peers[own_pos] != myid)
5384     {
5385       /* create path: self not found in path through self */
5386       GNUNET_break_op (0);
5387       path_destroy (path);
5388       connection_destroy (c);
5389       return GNUNET_OK;
5390     }
5391     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
5392     path_add_to_peers (path, GNUNET_NO);
5393     c->path = path_duplicate (path);
5394     c->own_pos = own_pos;
5395   }
5396   else
5397   {
5398     path = NULL;
5399   }
5400
5401   /* Remember peers */
5402   dest_peer = peer_get (&id[size - 1]);
5403   orig_peer = peer_get (&id[0]);
5404
5405   /* Is it a connection to us? */
5406   if (c->own_pos == size - 1)
5407   {
5408     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
5409     peer_add_path_to_origin (orig_peer, path, GNUNET_YES);
5410
5411     if (NULL == orig_peer->tunnel)
5412     {
5413       orig_peer->tunnel = tunnel_new ();
5414       orig_peer->tunnel->peer = orig_peer;
5415     }
5416     tunnel_add_connection (orig_peer->tunnel, c);
5417
5418     send_connection_ack (c, GNUNET_NO);
5419
5420     /* Keep tunnel alive in direction dest->owner*/
5421     connection_reset_timeout (c, GNUNET_NO); 
5422   }
5423   else
5424   {
5425     /* It's for somebody else! Retransmit. */
5426     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
5427     peer_add_path (dest_peer, path_duplicate (path), GNUNET_NO);
5428     peer_add_path_to_origin (orig_peer, path, GNUNET_NO);
5429     send_prebuilt_message_connection (message, c, NULL, GNUNET_YES);
5430   }
5431   return GNUNET_OK;
5432 }
5433
5434
5435 /**
5436  * Core handler for path ACKs
5437  *
5438  * @param cls closure
5439  * @param message message
5440  * @param peer peer identity this notification is about
5441  *
5442  * @return GNUNET_OK to keep the connection open,
5443  *         GNUNET_SYSERR to close it (signal serious error)
5444  */
5445 static int
5446 handle_mesh_connection_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
5447                             const struct GNUNET_MessageHeader *message)
5448 {
5449   struct GNUNET_MESH_ConnectionACK *msg;
5450   struct MeshPeerPath *p;
5451   struct MeshConnection *c;
5452
5453   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5454   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received a connection ACK msg\n");
5455   msg = (struct GNUNET_MESH_ConnectionACK *) message;
5456   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n",
5457               GNUNET_h2s (&msg->cid));
5458   c = connection_get (&msg->cid);
5459   if (NULL == c)
5460   {
5461     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
5462                               1, GNUNET_NO);
5463     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  don't know the connection!\n");
5464     return GNUNET_OK;
5465   }
5466
5467   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n",
5468               GNUNET_i2s (peer));
5469
5470   /* Add path to peers? */
5471   p = c->path;
5472   if (NULL != p)
5473   {
5474     path_add_to_peers (p, GNUNET_YES);
5475   }
5476   else
5477   {
5478     GNUNET_break (0);
5479   }
5480   connection_change_state (c, MESH_CONNECTION_READY);
5481   connection_reset_timeout (c, GNUNET_NO);
5482
5483   /* Message for us as creator? */
5484   if (connection_is_origin (c, GNUNET_YES))
5485   {
5486     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Connection (SYN)ACK for us!\n");
5487     if (MESH_TUNNEL_READY != c->t->state)
5488       tunnel_change_state (c->t, MESH_TUNNEL_READY);
5489     send_connection_ack (c, GNUNET_NO);
5490     tunnel_send_queued_data (c->t, GNUNET_YES);
5491     if (3 <= tunnel_count_connections (c->t) && NULL != c->t->peer->dhtget)
5492     {
5493       GNUNET_DHT_get_stop (c->t->peer->dhtget);
5494       c->t->peer->dhtget = NULL;
5495     }
5496     connection_change_state (c, MESH_CONNECTION_READY);
5497     return GNUNET_OK;
5498   }
5499
5500   /* Message for us as destination? */
5501   if (connection_is_terminal (c, GNUNET_YES))
5502   {
5503     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Connection ACK for us!\n");
5504     if (MESH_TUNNEL_READY != c->t->state)
5505       tunnel_change_state (c->t, MESH_TUNNEL_READY);
5506     connection_change_state (c, MESH_CONNECTION_READY);
5507     tunnel_send_queued_data (c->t, GNUNET_NO);
5508     return GNUNET_OK;
5509   }
5510
5511   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
5512   send_prebuilt_message_connection (message, c, NULL, GNUNET_NO);
5513   return GNUNET_OK;
5514 }
5515
5516
5517 /**
5518  * Core handler for notifications of broken paths
5519  *
5520  * @param cls Closure (unused).
5521  * @param peer Peer identity of sending neighbor.
5522  * @param message Message.
5523  *
5524  * @return GNUNET_OK to keep the connection open,
5525  *         GNUNET_SYSERR to close it (signal serious error)
5526  */
5527 static int
5528 handle_mesh_connection_broken (void *cls, const struct GNUNET_PeerIdentity *peer,
5529                                const struct GNUNET_MessageHeader *message)
5530 {
5531   struct GNUNET_MESH_ConnectionBroken *msg;
5532   struct MeshConnection *c;
5533
5534   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5535               "Received a CONNECTION BROKEN msg from %s\n", GNUNET_i2s (peer));
5536   msg = (struct GNUNET_MESH_ConnectionBroken *) message;
5537   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
5538               GNUNET_i2s (&msg->peer1));
5539   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
5540               GNUNET_i2s (&msg->peer2));
5541   c = connection_get (&msg->cid);
5542   if (NULL == c)
5543   {
5544     GNUNET_break_op (0);
5545     return GNUNET_OK;
5546   }
5547   tunnel_notify_connection_broken (c->t, GNUNET_PEER_search (&msg->peer1),
5548                                    GNUNET_PEER_search (&msg->peer2));
5549   return GNUNET_OK;
5550
5551 }
5552
5553
5554 /**
5555  * Core handler for tunnel destruction
5556  *
5557  * @param cls Closure (unused).
5558  * @param peer Peer identity of sending neighbor.
5559  * @param message Message.
5560  *
5561  * @return GNUNET_OK to keep the connection open,
5562  *         GNUNET_SYSERR to close it (signal serious error)
5563  */
5564 static int
5565 handle_mesh_connection_destroy (void *cls,
5566                                 const struct GNUNET_PeerIdentity *peer,
5567                                 const struct GNUNET_MessageHeader *message)
5568 {
5569   struct GNUNET_MESH_ConnectionDestroy *msg;
5570   struct MeshConnection *c;
5571   GNUNET_PEER_Id id;
5572   int fwd;
5573
5574   msg = (struct GNUNET_MESH_ConnectionDestroy *) message;
5575   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5576               "Got a CONNECTION DESTROY message from %s\n",
5577               GNUNET_i2s (peer));
5578   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5579               "  for connection %s\n",
5580               GNUNET_h2s (&msg->cid));
5581   c = connection_get (&msg->cid);
5582   if (NULL == c)
5583   {
5584     /* Probably already got the message from another path,
5585      * destroyed the tunnel and retransmitted to children.
5586      * Safe to ignore.
5587      */
5588     GNUNET_STATISTICS_update (stats, "# control on unknown tunnel",
5589                               1, GNUNET_NO);
5590     return GNUNET_OK;
5591   }
5592   id = GNUNET_PEER_search (peer);
5593   if (id == connection_get_prev_hop (c)->id)
5594     fwd = GNUNET_YES;
5595   else if (id == connection_get_next_hop (c)->id)
5596     fwd = GNUNET_NO;
5597   else
5598   {
5599     GNUNET_break_op (0);
5600     return GNUNET_OK;
5601   }
5602   send_prebuilt_message_connection (message, c, NULL, fwd);
5603   c->destroy = GNUNET_YES;
5604
5605   return GNUNET_OK;
5606 }
5607
5608
5609 /**
5610  * Handler for channel create messages.
5611  *
5612  * @param t Tunnel this channel is to be created in.
5613  * @param msg Message.
5614  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5615  */
5616 static void
5617 handle_channel_create (struct MeshTunnel2 *t,
5618                        struct GNUNET_MESH_ChannelCreate *msg,
5619                        int fwd)
5620 {
5621   MESH_ChannelNumber chid;
5622   struct MeshChannel *ch;
5623   struct MeshClient *c;
5624   uint32_t port;
5625
5626   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received Channel Create\n");
5627   /* Check message size */
5628   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelCreate))
5629   {
5630     GNUNET_break_op (0);
5631     return;
5632   }
5633
5634   /* Check if channel exists */
5635   chid = ntohl (msg->chid);
5636   ch = channel_get (t, chid);
5637   if (NULL != ch)
5638   {
5639     /* Probably a retransmission, safe to ignore */
5640     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   already exists...\n");
5641     if (NULL != ch->dest)
5642     {
5643       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   duplicate CC!!\n");
5644       channel_send_ack (ch, !fwd);
5645       return;
5646     }
5647   }
5648   else
5649   {
5650     /* Create channel */
5651     ch = channel_new (t, NULL, 0);
5652     channel_set_options (ch, ntohl (msg->opt));
5653   }
5654
5655   /* Find a destination client */
5656   port = ntohl (msg->port);
5657   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   port %u\n", port);
5658   c = GNUNET_CONTAINER_multihashmap32_get (ports, port);
5659   if (NULL == c)
5660   {
5661     /* TODO send reject */
5662     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  no client has port registered\n");
5663     /* TODO free ch */
5664     return;
5665   }
5666
5667   channel_add_client (ch, c);
5668   if (GNUNET_YES == ch->reliable)
5669     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "!!! Reliable\n");
5670
5671   send_local_channel_create (ch);
5672   channel_send_ack (ch, fwd);
5673   send_local_ack (ch, !fwd);
5674 }
5675
5676
5677 /**
5678  * Handler for channel ack messages.
5679  *
5680  * @param t Tunnel this channel is to be created in.
5681  * @param msg Message.
5682  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5683  */
5684 static void
5685 handle_channel_ack (struct MeshTunnel2 *t,
5686                     struct GNUNET_MESH_ChannelManage *msg,
5687                     int fwd)
5688 {
5689   MESH_ChannelNumber chid;
5690   struct MeshChannel *ch;
5691
5692   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received Channel ACK\n");
5693   /* Check message size */
5694   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelManage))
5695   {
5696     GNUNET_break_op (0);
5697     return;
5698   }
5699
5700   /* Check if channel exists */
5701   chid = ntohl (msg->chid);
5702   ch = channel_get (t, chid);
5703   if (NULL == ch)
5704   {
5705     GNUNET_break_op (0);
5706     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   channel %u unknown!!\n", chid);
5707     return;
5708   }
5709
5710   channel_confirm (ch, !fwd);
5711 }
5712
5713
5714 /**
5715  * Handler for channel destroy messages.
5716  *
5717  * @param t Tunnel this channel is to be destroyed of.
5718  * @param msg Message.
5719  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5720  */
5721 static void
5722 handle_channel_destroy (struct MeshTunnel2 *t,
5723                         struct GNUNET_MESH_ChannelManage *msg,
5724                         int fwd)
5725 {
5726   MESH_ChannelNumber chid;
5727   struct MeshChannel *ch;
5728
5729   /* Check message size */
5730   if (ntohs (msg->header.size) != sizeof (struct GNUNET_MESH_ChannelManage))
5731   {
5732     GNUNET_break_op (0);
5733     return;
5734   }
5735
5736   /* Check if channel exists */
5737   chid = ntohl (msg->chid);
5738   ch = channel_get (t, chid);
5739   if (NULL == ch)
5740   {
5741     /* Probably a retransmission, safe to ignore */
5742     return;
5743   }
5744
5745   send_local_channel_destroy (ch, fwd);
5746   channel_destroy (ch);
5747 }
5748
5749
5750 static void
5751 handle_decrypted (struct MeshTunnel2 *t,
5752                   const struct GNUNET_MessageHeader *msgh,
5753                   int fwd)
5754 {
5755   switch (ntohs (msgh->type))
5756   {
5757     case GNUNET_MESSAGE_TYPE_MESH_DATA:
5758       /* Don't send hop ACK, wait for client to ACK */
5759       handle_data (t, (struct GNUNET_MESH_Data *) msgh, fwd);
5760       break;
5761
5762     case GNUNET_MESSAGE_TYPE_MESH_DATA_ACK:
5763       handle_data_ack (t, (struct GNUNET_MESH_DataACK *) msgh, fwd);
5764       break;
5765
5766     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE:
5767       handle_channel_create (t,
5768                              (struct GNUNET_MESH_ChannelCreate *) msgh,
5769                              fwd);
5770       break;
5771
5772     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_ACK:
5773       handle_channel_ack (t,
5774                           (struct GNUNET_MESH_ChannelManage *) msgh,
5775                           fwd);
5776       break;
5777
5778     case GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY:
5779       handle_channel_destroy (t,
5780                               (struct GNUNET_MESH_ChannelManage *) msgh,
5781                               fwd);
5782       break;
5783
5784     default:
5785       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5786                   "end-to-end message not known (%u)\n",
5787                   ntohs (msgh->type));
5788   }
5789 }
5790
5791
5792 /**
5793  * Generic handler for mesh network encrypted traffic.
5794  *
5795  * @param peer Peer identity this notification is about.
5796  * @param message Encrypted message.
5797  * @param fwd Is this FWD traffic? GNUNET_YES : GNUNET_NO;
5798  *
5799  * @return GNUNET_OK to keep the connection open,
5800  *         GNUNET_SYSERR to close it (signal serious error)
5801  */
5802 static int
5803 handle_mesh_encrypted (const struct GNUNET_PeerIdentity *peer,
5804                        const struct GNUNET_MESH_Encrypted *msg,
5805                        int fwd)
5806 {
5807   struct MeshConnection *c;
5808   struct MeshTunnel2 *t;
5809   struct MeshPeer *neighbor;
5810   struct MeshFlowControl *fc;
5811   uint32_t pid;
5812   uint32_t ttl;
5813   uint16_t type;
5814   size_t size;
5815
5816   /* Check size */
5817   size = ntohs (msg->header.size);
5818   if (size <
5819       sizeof (struct GNUNET_MESH_Encrypted) +
5820       sizeof (struct GNUNET_MessageHeader))
5821   {
5822     GNUNET_break_op (0);
5823     return GNUNET_OK;
5824   }
5825   type = ntohs (msg->header.type);
5826   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5827   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
5828               GNUNET_MESH_DEBUG_M2S (type), GNUNET_i2s (peer));
5829
5830   /* Check connection */
5831   c = connection_get (&msg->cid);
5832   if (NULL == c)
5833   {
5834     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
5835     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "WARNING connection unknown\n");
5836     return GNUNET_OK;
5837   }
5838   t = c->t;
5839   fc = fwd ? &c->bck_fc : &c->fwd_fc;
5840
5841   /* Check if origin is as expected */
5842   neighbor = connection_get_hop (c, !fwd);
5843   if (peer_get (peer)->id != neighbor->id)
5844   {
5845     GNUNET_break_op (0);
5846     return GNUNET_OK;
5847   }
5848
5849   /* Check PID */
5850   pid = ntohl (msg->pid);
5851   if (GMC_is_pid_bigger (pid, fc->last_ack_sent))
5852   {
5853     GNUNET_STATISTICS_update (stats, "# unsolicited message", 1, GNUNET_NO);
5854     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5855                 "WARNING Received PID %u, (prev %u), ACK %u\n",
5856                 pid, fc->last_pid_recv, fc->last_ack_sent);
5857     return GNUNET_OK;
5858   }
5859   if (GNUNET_NO == GMC_is_pid_bigger (pid, fc->last_pid_recv))
5860   {
5861     GNUNET_STATISTICS_update (stats, "# duplicate PID", 1, GNUNET_NO);
5862     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
5863                 " Pid %u not expected (%u+), dropping!\n",
5864                 pid, fc->last_pid_recv + 1);
5865     return GNUNET_OK;
5866   }
5867   if (MESH_CONNECTION_SENT == c->state)
5868     connection_change_state (c, MESH_CONNECTION_READY);
5869   connection_reset_timeout (c, fwd);
5870   fc->last_pid_recv = pid;
5871
5872   /* Is this message for us? */
5873   if (connection_is_terminal (c, fwd))
5874   {
5875     size_t dsize = size - sizeof (struct GNUNET_MESH_Encrypted);
5876     char cbuf[dsize];
5877     struct GNUNET_MessageHeader *msgh;
5878     unsigned int off;
5879
5880     /* TODO signature verification */
5881     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
5882     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
5883
5884     fc->last_pid_recv = pid;
5885     tunnel_decrypt (t, cbuf, &msg[1], dsize, msg->iv, fwd);
5886     off = 0;
5887     while (off < dsize)
5888     {
5889       msgh = (struct GNUNET_MessageHeader *) &cbuf[off];
5890       handle_decrypted (t, msgh, fwd);
5891       off += ntohs (msgh->size);
5892     }
5893     send_ack (c, NULL, fwd);
5894     return GNUNET_OK;
5895   }
5896
5897   /* Message not for us: forward to next hop */
5898   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
5899   ttl = ntohl (msg->ttl);
5900   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
5901   if (ttl == 0)
5902   {
5903     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
5904     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
5905     send_ack (c, NULL, fwd);
5906     return GNUNET_OK;
5907   }
5908   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
5909
5910   send_prebuilt_message_connection (&msg->header, c, NULL, fwd);
5911
5912   return GNUNET_OK;
5913 }
5914
5915
5916 /**
5917  * Core handler for mesh network traffic going orig->dest.
5918  *
5919  * @param cls Closure (unused).
5920  * @param message Message received.
5921  * @param peer Peer who sent the message.
5922  *
5923  * @return GNUNET_OK to keep the connection open,
5924  *         GNUNET_SYSERR to close it (signal serious error)
5925  */
5926 static int
5927 handle_mesh_fwd (void *cls, const struct GNUNET_PeerIdentity *peer,
5928                      const struct GNUNET_MessageHeader *message)
5929 {
5930   return handle_mesh_encrypted (peer,
5931                                 (struct GNUNET_MESH_Encrypted *)message,
5932                                 GNUNET_YES);
5933 }
5934
5935 /**
5936  * Core handler for mesh network traffic going dest->orig.
5937  *
5938  * @param cls Closure (unused).
5939  * @param message Message received.
5940  * @param peer Peer who sent the message.
5941  *
5942  * @return GNUNET_OK to keep the connection open,
5943  *         GNUNET_SYSERR to close it (signal serious error)
5944  */
5945 static int
5946 handle_mesh_bck (void *cls, const struct GNUNET_PeerIdentity *peer,
5947                      const struct GNUNET_MessageHeader *message)
5948 {
5949   return handle_mesh_encrypted (peer,
5950                                 (struct GNUNET_MESH_Encrypted *)message,
5951                                 GNUNET_NO);
5952 }
5953
5954
5955 /**
5956  * Core handler for mesh network traffic point-to-point acks.
5957  *
5958  * @param cls closure
5959  * @param message message
5960  * @param peer peer identity this notification is about
5961  *
5962  * @return GNUNET_OK to keep the connection open,
5963  *         GNUNET_SYSERR to close it (signal serious error)
5964  */
5965 static int
5966 handle_mesh_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
5967                  const struct GNUNET_MessageHeader *message)
5968 {
5969   struct GNUNET_MESH_ACK *msg;
5970   struct MeshConnection *c;
5971   struct MeshFlowControl *fc;
5972   GNUNET_PEER_Id id;
5973   uint32_t ack;
5974
5975   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
5976   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
5977               GNUNET_i2s (peer));
5978   msg = (struct GNUNET_MESH_ACK *) message;
5979
5980   c = connection_get (&msg->cid);
5981
5982   if (NULL == c)
5983   {
5984     GNUNET_STATISTICS_update (stats, "# ack on unknown connection", 1,
5985                               GNUNET_NO);
5986     return GNUNET_OK;
5987   }
5988
5989   /* Is this a forward or backward ACK? */
5990   id = GNUNET_PEER_search (peer);
5991   if (connection_get_next_hop (c)->id == id)
5992   {
5993     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
5994     fc = &c->fwd_fc;
5995   }
5996   else if (connection_get_prev_hop (c)->id == id)
5997   {
5998     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
5999     fc = &c->bck_fc;
6000   }
6001   else
6002   {
6003     GNUNET_break_op (0);
6004     return GNUNET_OK;
6005   }
6006
6007   ack = ntohl (msg->ack);
6008   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u\n", ack);
6009
6010   /* Cancel polling if the ACK is bigger than before. */
6011   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task &&
6012       GMC_is_pid_bigger (ack, fc->last_ack_recv))
6013   {
6014     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  Cancel poll\n");
6015     GNUNET_SCHEDULER_cancel (fc->poll_task);
6016     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
6017     fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
6018   }
6019
6020   fc->last_ack_recv = ack;
6021   connection_unlock_queue (c, fc == &c->fwd_fc);
6022
6023   return GNUNET_OK;
6024 }
6025
6026
6027 /**
6028  * Core handler for mesh network traffic point-to-point ack polls.
6029  *
6030  * @param cls closure
6031  * @param message message
6032  * @param peer peer identity this notification is about
6033  *
6034  * @return GNUNET_OK to keep the connection open,
6035  *         GNUNET_SYSERR to close it (signal serious error)
6036  */
6037 static int
6038 handle_mesh_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
6039                   const struct GNUNET_MessageHeader *message)
6040 {
6041   struct GNUNET_MESH_Poll *msg;
6042   struct MeshConnection *c;
6043   struct MeshFlowControl *fc;
6044   GNUNET_PEER_Id id;
6045   uint32_t pid;
6046   int fwd;
6047
6048   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
6049   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a POLL packet from %s!\n",
6050               GNUNET_i2s (peer));
6051
6052   msg = (struct GNUNET_MESH_Poll *) message;
6053
6054   c = connection_get (&msg->cid);
6055
6056   if (NULL == c)
6057   {
6058     GNUNET_STATISTICS_update (stats, "# poll on unknown connection", 1,
6059                               GNUNET_NO);
6060     GNUNET_break_op (0);
6061     return GNUNET_OK;
6062   }
6063
6064   /* Is this a forward or backward ACK?
6065    * Note: a poll should never be needed in a loopback case,
6066    * since there is no possiblility of packet loss there, so
6067    * this way of discerining FWD/BCK should not be a problem.
6068    */
6069   id = GNUNET_PEER_search (peer);
6070   if (connection_get_next_hop (c)->id == id)
6071   {
6072     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
6073     fc = &c->fwd_fc;
6074   }
6075   else if (connection_get_prev_hop (c)->id == id)
6076   {
6077     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
6078     fc = &c->bck_fc;
6079   }
6080   else
6081   {
6082     GNUNET_break_op (0);
6083     return GNUNET_OK;
6084   }
6085
6086   pid = ntohl (msg->pid);
6087   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  PID %u, OLD %u\n",
6088               pid, fc->last_pid_recv);
6089   fc->last_pid_recv = pid;
6090   fwd = fc == &c->fwd_fc;
6091   send_ack (c, NULL, fwd);
6092
6093   return GNUNET_OK;
6094 }
6095
6096
6097 /**
6098  * Core handler for mesh keepalives.
6099  *
6100  * @param cls closure
6101  * @param message message
6102  * @param peer peer identity this notification is about
6103  * @return GNUNET_OK to keep the connection open,
6104  *         GNUNET_SYSERR to close it (signal serious error)
6105  *
6106  * TODO: Check who we got this from, to validate route.
6107  */
6108 static int
6109 handle_mesh_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
6110                        const struct GNUNET_MessageHeader *message)
6111 {
6112   struct GNUNET_MESH_ConnectionKeepAlive *msg;
6113   struct MeshConnection *c;
6114   struct MeshPeer *neighbor;
6115   int fwd;
6116
6117   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) message;
6118   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
6119               GNUNET_i2s (peer));
6120
6121   c = connection_get (&msg->cid);
6122   if (NULL == c)
6123   {
6124     GNUNET_STATISTICS_update (stats, "# keepalive on unknown connection", 1,
6125                               GNUNET_NO);
6126     return GNUNET_OK;
6127   }
6128
6129   fwd = GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE == ntohs (message->type) ? 
6130         GNUNET_YES : GNUNET_NO;
6131
6132   /* Check if origin is as expected */
6133   neighbor = connection_get_hop (c, fwd);
6134   if (peer_get (peer)->id != neighbor->id)
6135   {
6136     GNUNET_break_op (0);
6137     return GNUNET_OK;
6138   }
6139
6140   connection_change_state (c, MESH_CONNECTION_READY);
6141   connection_reset_timeout (c, fwd);
6142
6143   if (connection_is_terminal (c, fwd))
6144     return GNUNET_OK;
6145
6146   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
6147   send_prebuilt_message_connection (message, c, NULL, fwd);
6148
6149   return GNUNET_OK;
6150 }
6151
6152
6153
6154 /**
6155  * Functions to handle messages from core
6156  */
6157 static struct GNUNET_CORE_MessageHandler core_handlers[] = {
6158   {&handle_mesh_connection_create, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
6159     0},
6160   {&handle_mesh_connection_ack, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
6161     sizeof (struct GNUNET_MESH_ConnectionACK)},
6162   {&handle_mesh_connection_broken, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN,
6163     sizeof (struct GNUNET_MESH_ConnectionBroken)},
6164   {&handle_mesh_connection_destroy, GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY,
6165     sizeof (struct GNUNET_MESH_ConnectionDestroy)},
6166   {&handle_mesh_keepalive, GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE,
6167     sizeof (struct GNUNET_MESH_ConnectionKeepAlive)},
6168   {&handle_mesh_keepalive, GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE,
6169     sizeof (struct GNUNET_MESH_ConnectionKeepAlive)},
6170   {&handle_mesh_ack, GNUNET_MESSAGE_TYPE_MESH_ACK,
6171     sizeof (struct GNUNET_MESH_ACK)},
6172   {&handle_mesh_poll, GNUNET_MESSAGE_TYPE_MESH_POLL,
6173     sizeof (struct GNUNET_MESH_Poll)},
6174   {&handle_mesh_fwd, GNUNET_MESSAGE_TYPE_MESH_FWD, 0},
6175   {&handle_mesh_bck, GNUNET_MESSAGE_TYPE_MESH_BCK, 0},
6176   {NULL, 0, 0}
6177 };
6178
6179
6180 /**
6181  * Function to process paths received for a new peer addition. The recorded
6182  * paths form the initial tunnel, which can be optimized later.
6183  * Called on each result obtained for the DHT search.
6184  *
6185  * @param cls closure
6186  * @param exp when will this value expire
6187  * @param key key of the result
6188  * @param get_path path of the get request
6189  * @param get_path_length lenght of get_path
6190  * @param put_path path of the put request
6191  * @param put_path_length length of the put_path
6192  * @param type type of the result
6193  * @param size number of bytes in data
6194  * @param data pointer to the result data
6195  */
6196 static void
6197 dht_get_id_handler (void *cls, struct GNUNET_TIME_Absolute exp,
6198                     const struct GNUNET_HashCode * key,
6199                     const struct GNUNET_PeerIdentity *get_path,
6200                     unsigned int get_path_length,
6201                     const struct GNUNET_PeerIdentity *put_path,
6202                     unsigned int put_path_length, enum GNUNET_BLOCK_Type type,
6203                     size_t size, const void *data)
6204 {
6205   struct MeshPeer *peer = cls;
6206   struct MeshPeerPath *p;
6207   struct MeshConnection *c;
6208   struct GNUNET_PeerIdentity pi;
6209   int i;
6210
6211   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got results from DHT!\n");
6212   GNUNET_PEER_resolve (peer->id, &pi);
6213   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  for %s\n", GNUNET_i2s (&pi));
6214
6215   p = path_build_from_dht (get_path, get_path_length,
6216                            put_path, put_path_length);
6217   path_add_to_peers (p, GNUNET_NO);
6218   path_destroy (p);
6219
6220   /* Count connections */
6221   for (c = peer->tunnel->connection_head, i = 0; NULL != c; c = c->next, i++);
6222
6223   /* If we already have 3 (or more (?!)) connections, it's enough */
6224   if (3 <= i)
6225     return;
6226
6227   if (peer->tunnel->state == MESH_TUNNEL_SEARCHING)
6228   {
6229     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, " ... connect!\n");
6230     peer_connect (peer);
6231   }
6232   return;
6233 }
6234
6235
6236 /******************************************************************************/
6237 /*********************       MESH LOCAL HANDLES      **************************/
6238 /******************************************************************************/
6239
6240
6241 /**
6242  * Handler for client connection.
6243  *
6244  * @param cls Closure (unused).
6245  * @param client Client handler.
6246  */
6247 static void
6248 handle_local_client_connect (void *cls, struct GNUNET_SERVER_Client *client)
6249 {
6250   struct MeshClient *c;
6251
6252   if (NULL == client)
6253     return;
6254   c = GNUNET_malloc (sizeof (struct MeshClient));
6255   c->handle = client;
6256   c->id = next_client_id++; /* overflow not important: just for debug */
6257   GNUNET_SERVER_client_keep (client);
6258   GNUNET_SERVER_client_set_user_context (client, c);
6259   GNUNET_CONTAINER_DLL_insert (clients_head, clients_tail, c);
6260 }
6261
6262
6263 /**
6264  * Handler for client disconnection
6265  *
6266  * @param cls closure
6267  * @param client identification of the client; NULL
6268  *        for the last call when the server is destroyed
6269  */
6270 static void
6271 handle_local_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
6272 {
6273   struct MeshClient *c;
6274
6275   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disconnected: %p\n", client);
6276   if (client == NULL)
6277   {
6278     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   (SERVER DOWN)\n");
6279     return;
6280   }
6281
6282   c = client_get (client);
6283   if (NULL != c)
6284   {
6285     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "matching client found (%u, %p)\n",
6286                 c->id, c);
6287     GNUNET_SERVER_client_drop (c->handle);
6288     c->shutting_down = GNUNET_YES;
6289     if (NULL != c->own_channels)
6290     {
6291       GNUNET_CONTAINER_multihashmap32_iterate (c->own_channels,
6292                                                &channel_destroy_iterator, c);
6293       GNUNET_CONTAINER_multihashmap32_destroy (c->own_channels);
6294     }
6295
6296     if (NULL != c->incoming_channels)
6297     {
6298       GNUNET_CONTAINER_multihashmap32_iterate (c->incoming_channels,
6299                                                &channel_destroy_iterator, c);
6300       GNUNET_CONTAINER_multihashmap32_destroy (c->incoming_channels);
6301     }
6302
6303     if (NULL != c->ports)
6304     {
6305       GNUNET_CONTAINER_multihashmap32_iterate (c->ports,
6306                                                &client_release_ports, c);
6307       GNUNET_CONTAINER_multihashmap32_destroy (c->ports);
6308     }
6309     GNUNET_CONTAINER_DLL_remove (clients_head, clients_tail, c);
6310     GNUNET_STATISTICS_update (stats, "# clients", -1, GNUNET_NO);
6311     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client free (%p)\n", c);
6312     GNUNET_free (c);
6313   }
6314   else
6315   {
6316     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, " context NULL!\n");
6317   }
6318   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "done!\n");
6319   return;
6320 }
6321
6322
6323 /**
6324  * Handler for new clients
6325  *
6326  * @param cls closure
6327  * @param client identification of the client
6328  * @param message the actual message, which includes messages the client wants
6329  */
6330 static void
6331 handle_local_new_client (void *cls, struct GNUNET_SERVER_Client *client,
6332                          const struct GNUNET_MessageHeader *message)
6333 {
6334   struct GNUNET_MESH_ClientConnect *cc_msg;
6335   struct MeshClient *c;
6336   unsigned int size;
6337   uint32_t *p;
6338   unsigned int i;
6339
6340   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client connected %p\n", client);
6341
6342   /* Check data sanity */
6343   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_ClientConnect);
6344   cc_msg = (struct GNUNET_MESH_ClientConnect *) message;
6345   if (0 != (size % sizeof (uint32_t)))
6346   {
6347     GNUNET_break (0);
6348     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6349     return;
6350   }
6351   size /= sizeof (uint32_t);
6352
6353   /* Initialize new client structure */
6354   c = GNUNET_SERVER_client_get_user_context (client, struct MeshClient);
6355   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client id %u\n", c->id);
6356   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  client has %u ports\n", size);
6357   if (size > 0)
6358   {
6359     uint32_t u32;
6360
6361     p = (uint32_t *) &cc_msg[1];
6362     c->ports = GNUNET_CONTAINER_multihashmap32_create (size);
6363     for (i = 0; i < size; i++)
6364     {
6365       u32 = ntohl (p[i]);
6366       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "    port: %u\n", u32);
6367
6368       /* store in client's hashmap */
6369       GNUNET_CONTAINER_multihashmap32_put (c->ports, u32, c,
6370                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
6371       /* store in global hashmap */
6372       /* FIXME only allow one client to have the port open,
6373        *       have a backup hashmap with waiting clients */
6374       GNUNET_CONTAINER_multihashmap32_put (ports, u32, c,
6375                                            GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
6376     }
6377   }
6378
6379   c->own_channels = GNUNET_CONTAINER_multihashmap32_create (32);
6380   c->incoming_channels = GNUNET_CONTAINER_multihashmap32_create (32);
6381   GNUNET_SERVER_notification_context_add (nc, client);
6382   GNUNET_STATISTICS_update (stats, "# clients", 1, GNUNET_NO);
6383
6384   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6385   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new client processed\n");
6386 }
6387
6388
6389 /**
6390  * Handler for requests of new tunnels
6391  *
6392  * @param cls Closure.
6393  * @param client Identification of the client.
6394  * @param message The actual message.
6395  */
6396 static void
6397 handle_local_channel_create (void *cls, struct GNUNET_SERVER_Client *client,
6398                             const struct GNUNET_MessageHeader *message)
6399 {
6400   struct GNUNET_MESH_ChannelMessage *msg;
6401   struct MeshPeer *peer;
6402   struct MeshTunnel2 *t;
6403   struct MeshChannel *ch;
6404   struct MeshClient *c;
6405   MESH_ChannelNumber chid;
6406
6407   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "new channel requested\n");
6408
6409   /* Sanity check for client registration */
6410   if (NULL == (c = client_get (client)))
6411   {
6412     GNUNET_break (0);
6413     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6414     return;
6415   }
6416   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6417
6418   /* Message size sanity check */
6419   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
6420   {
6421     GNUNET_break (0);
6422     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6423     return;
6424   }
6425
6426   msg = (struct GNUNET_MESH_ChannelMessage *) message;
6427   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  towards %s:%u\n",
6428               GNUNET_i2s (&msg->peer), ntohl (msg->port));
6429   chid = ntohl (msg->channel_id);
6430
6431   /* Sanity check for duplicate channel IDs */
6432   if (NULL != channel_get_by_local_id (c, chid))
6433   {
6434     GNUNET_break (0);
6435     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6436     return;
6437   }
6438
6439   peer = peer_get (&msg->peer);
6440   if (NULL == peer->tunnel)
6441   {
6442     peer->tunnel = tunnel_new ();
6443     peer->tunnel->peer = peer;
6444   }
6445   t = peer->tunnel;
6446
6447   /* Create channel */
6448   ch = channel_new (t, c, chid);
6449   if (NULL == ch)
6450   {
6451     GNUNET_break (0);
6452     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6453     return;
6454   }
6455   ch->port = ntohl (msg->port);
6456   channel_set_options (ch, ntohl (msg->opt));
6457
6458   /* In unreliable channels, we'll use the DLL to buffer data for the root */
6459   ch->root_rel = GNUNET_new (struct MeshChannelReliability);
6460   ch->root_rel->ch = ch;
6461   ch->root_rel->expected_delay = MESH_RETRANSMIT_TIME;
6462
6463   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "CREATED CHANNEL %s[%x]:%u (%x)\n",
6464               peer2s (t->peer), ch->gid, ch->port, ch->lid_root);
6465   peer_connect (peer);
6466
6467   /* Send create channel */
6468   {
6469     struct GNUNET_MESH_ChannelCreate msgcc;
6470
6471     msgcc.header.size = htons (sizeof (msgcc));
6472     msgcc.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE);
6473     msgcc.chid = htonl (ch->gid);
6474     msgcc.port = msg->port;
6475     msgcc.opt = msg->opt;
6476
6477     tunnel_queue_data (t, ch, &msgcc.header, GNUNET_YES);
6478   }
6479
6480   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6481   return;
6482 }
6483
6484
6485 /**
6486  * Handler for requests of deleting tunnels
6487  *
6488  * @param cls closure
6489  * @param client identification of the client
6490  * @param message the actual message
6491  */
6492 static void
6493 handle_local_channel_destroy (void *cls, struct GNUNET_SERVER_Client *client,
6494                              const struct GNUNET_MessageHeader *message)
6495 {
6496   struct GNUNET_MESH_ChannelMessage *msg;
6497   struct MeshClient *c;
6498   struct MeshChannel *ch;
6499   struct MeshTunnel2 *t;
6500   MESH_ChannelNumber chid;
6501
6502   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
6503               "Got a DESTROY CHANNEL from client!\n");
6504
6505   /* Sanity check for client registration */
6506   if (NULL == (c = client_get (client)))
6507   {
6508     GNUNET_break (0);
6509     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6510     return;
6511   }
6512   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6513
6514   /* Message sanity check */
6515   if (sizeof (struct GNUNET_MESH_ChannelMessage) != ntohs (message->size))
6516   {
6517     GNUNET_break (0);
6518     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6519     return;
6520   }
6521
6522   msg = (struct GNUNET_MESH_ChannelMessage *) message;
6523
6524   /* Retrieve tunnel */
6525   chid = ntohl (msg->channel_id);
6526   ch = channel_get_by_local_id (c, chid);
6527   if (NULL == ch)
6528   {
6529     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "  channel %X not found\n", chid);
6530     GNUNET_break (0);
6531     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6532     return;
6533   }
6534
6535   /* Cleanup after the tunnel */
6536   client_delete_channel (c, ch);
6537   if (c == ch->dest && GNUNET_MESH_LOCAL_CHANNEL_ID_SERV <= chid)
6538   {
6539     ch->dest = NULL;
6540   }
6541   else if (c == ch->root && GNUNET_MESH_LOCAL_CHANNEL_ID_SERV > chid)
6542   {
6543     ch->root = NULL;
6544   }
6545   else 
6546   {
6547     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6548                 "  channel %X client %p (%p, %p)\n",
6549                 chid, c, ch->root, ch->dest);
6550     GNUNET_break (0);
6551   }
6552
6553   t = ch->t;
6554   channel_destroy (ch);
6555   tunnel_destroy_if_empty (t);
6556
6557   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6558   return;
6559 }
6560
6561
6562 /**
6563  * Handler for client traffic
6564  *
6565  * @param cls closure
6566  * @param client identification of the client
6567  * @param message the actual message
6568  */
6569 static void
6570 handle_local_data (void *cls, struct GNUNET_SERVER_Client *client,
6571                    const struct GNUNET_MessageHeader *message)
6572 {
6573   struct GNUNET_MESH_LocalData *msg;
6574   struct MeshClient *c;
6575   struct MeshChannel *ch;
6576   struct MeshChannelReliability *rel;
6577   MESH_ChannelNumber chid;
6578   size_t size;
6579   int fwd;
6580
6581   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
6582               "Got data from a client!\n");
6583
6584   /* Sanity check for client registration */
6585   if (NULL == (c = client_get (client)))
6586   {
6587     GNUNET_break (0);
6588     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6589     return;
6590   }
6591   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6592
6593   msg = (struct GNUNET_MESH_LocalData *) message;
6594
6595   /* Sanity check for message size */
6596   size = ntohs (message->size) - sizeof (struct GNUNET_MESH_LocalData);
6597   if (size < sizeof (struct GNUNET_MessageHeader))
6598   {
6599     GNUNET_break (0);
6600     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6601     return;
6602   }
6603
6604   /* Channel exists? */
6605   chid = ntohl (msg->id);
6606   fwd = chid < GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
6607   ch = channel_get_by_local_id (c, chid);
6608   if (NULL == ch)
6609   {
6610     GNUNET_break (0);
6611     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6612     return;
6613   }
6614
6615   /* Is the client in the channel? */
6616   if ( !( (fwd &&
6617            ch->root &&
6618            ch->root->handle == client)
6619          ||
6620           (!fwd &&
6621            ch->dest && 
6622            ch->dest->handle == client) ) )
6623   {
6624     GNUNET_break (0);
6625     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6626     return;
6627   }
6628
6629   rel = fwd ? ch->root_rel : ch->dest_rel;
6630   rel->client_ready = GNUNET_NO;
6631
6632   /* Ok, everything is correct, send the message. */
6633   {
6634     struct GNUNET_MESH_Data *payload;
6635     uint16_t p2p_size = sizeof(struct GNUNET_MESH_Data) + size;
6636     unsigned char cbuf[p2p_size];
6637
6638     payload = (struct GNUNET_MESH_Data *) cbuf;
6639     payload->mid = htonl (rel->mid_send);
6640     rel->mid_send++;
6641     memcpy (&payload[1], &msg[1], size);
6642     payload->header.size = htons (p2p_size);
6643     payload->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_DATA);
6644     payload->chid = htonl (ch->gid);
6645     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  sending on channel...\n");
6646     send_prebuilt_message_channel (&payload->header, ch, fwd);
6647
6648     if (GNUNET_YES == ch->reliable)
6649       channel_save_copy (ch, &payload->header, fwd);
6650   }
6651   if (tunnel_get_buffer (ch->t, fwd) > 0)
6652     send_local_ack (ch, fwd);
6653   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "receive done OK\n");
6654   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6655
6656   return;
6657 }
6658
6659
6660 /**
6661  * Handler for client's ACKs for payload traffic.
6662  *
6663  * @param cls Closure (unused).
6664  * @param client Identification of the client.
6665  * @param message The actual message.
6666  */
6667 static void
6668 handle_local_ack (void *cls, struct GNUNET_SERVER_Client *client,
6669                   const struct GNUNET_MessageHeader *message)
6670 {
6671   struct GNUNET_MESH_LocalAck *msg;
6672   struct MeshChannelReliability *rel;
6673   struct MeshChannel *ch;
6674   struct MeshClient *c;
6675   MESH_ChannelNumber chid;
6676   int fwd;
6677
6678   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got a local ACK\n");
6679
6680   /* Sanity check for client registration */
6681   if (NULL == (c = client_get (client)))
6682   {
6683     GNUNET_break (0);
6684     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6685     return;
6686   }
6687   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  by client %u\n", c->id);
6688
6689   msg = (struct GNUNET_MESH_LocalAck *) message;
6690
6691   /* Channel exists? */
6692   chid = ntohl (msg->channel_id);
6693   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  on channel %X\n", chid);
6694   ch = channel_get_by_local_id (c, chid);
6695   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "   -- ch %p\n", ch);
6696   if (NULL == ch)
6697   {
6698     GNUNET_break (0);
6699     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Channel %X unknown.\n", chid);
6700     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "  for client %u.\n", c->id);
6701     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6702     return;
6703   }
6704
6705   /* If client is root, the ACK is going FWD, therefore this is "BCK". */
6706   /* If client is dest, the ACK is going BCK, therefore this is "FWD" */
6707   fwd = chid >= GNUNET_MESH_LOCAL_CHANNEL_ID_SERV;
6708   rel = fwd ? ch->dest_rel : ch->root_rel;
6709
6710   rel->client_ready = GNUNET_YES;
6711   channel_send_client_buffered_data (ch, c, fwd);
6712   send_ack (NULL, ch, fwd);
6713
6714   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6715
6716   return;
6717 }
6718
6719
6720 /**
6721  * Iterator over all tunnels to send a monitoring client info about each tunnel.
6722  *
6723  * @param cls Closure (client handle).
6724  * @param key Key (hashed tunnel ID, unused).
6725  * @param value Tunnel info.
6726  *
6727  * @return GNUNET_YES, to keep iterating.
6728  */
6729 // static int
6730 // monitor_all_tunnels_iterator (void *cls,
6731 //                               const struct GNUNET_HashCode * key,
6732 //                               void *value)
6733 // {
6734 //   struct GNUNET_SERVER_Client *client = cls;
6735 //   struct MeshChannel *ch = value;
6736 //   struct GNUNET_MESH_LocalMonitor *msg;
6737 // 
6738 //   msg = GNUNET_malloc (sizeof(struct GNUNET_MESH_LocalMonitor));
6739 //   msg->channel_id = htonl (ch->gid);
6740 //   msg->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
6741 //   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS);
6742 // 
6743 //   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6744 //               "*  sending info about tunnel %s\n",
6745 //               GNUNET_i2s (&msg->owner));
6746 // 
6747 //   GNUNET_SERVER_notification_context_unicast (nc, client,
6748 //                                               &msg->header, GNUNET_NO);
6749 //   return GNUNET_YES;
6750 // }
6751
6752
6753 /**
6754  * Handler for client's MONITOR request.
6755  *
6756  * @param cls Closure (unused).
6757  * @param client Identification of the client.
6758  * @param message The actual message.
6759  */
6760 static void
6761 handle_local_get_tunnels (void *cls, struct GNUNET_SERVER_Client *client,
6762                           const struct GNUNET_MessageHeader *message)
6763 {
6764   struct MeshClient *c;
6765
6766   /* Sanity check for client registration */
6767   if (NULL == (c = client_get (client)))
6768   {
6769     GNUNET_break (0);
6770     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6771     return;
6772   }
6773
6774   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6775               "Received get tunnels request from client %u\n",
6776               c->id);
6777 //   GNUNET_CONTAINER_multihashmap_iterate (tunnels,
6778 //                                          monitor_all_tunnels_iterator,
6779 //                                          client);
6780   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6781               "Get tunnels request from client %u completed\n",
6782               c->id);
6783   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6784 }
6785
6786
6787 /**
6788  * Handler for client's MONITOR_TUNNEL request.
6789  *
6790  * @param cls Closure (unused).
6791  * @param client Identification of the client.
6792  * @param message The actual message.
6793  */
6794 static void
6795 handle_local_show_tunnel (void *cls, struct GNUNET_SERVER_Client *client,
6796                           const struct GNUNET_MessageHeader *message)
6797 {
6798   const struct GNUNET_MESH_LocalMonitor *msg;
6799   struct GNUNET_MESH_LocalMonitor *resp;
6800   struct MeshClient *c;
6801   struct MeshChannel *ch;
6802
6803   /* Sanity check for client registration */
6804   if (NULL == (c = client_get (client)))
6805   {
6806     GNUNET_break (0);
6807     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
6808     return;
6809   }
6810
6811   msg = (struct GNUNET_MESH_LocalMonitor *) message;
6812   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6813               "Received tunnel info request from client %u for tunnel %s[%X]\n",
6814               c->id,
6815               &msg->owner,
6816               ntohl (msg->channel_id));
6817 //   ch = channel_get (&msg->owner, ntohl (msg->channel_id));
6818   ch = NULL; // FIXME
6819   if (NULL == ch)
6820   {
6821     /* We don't know the tunnel */
6822     struct GNUNET_MESH_LocalMonitor warn;
6823
6824     warn = *msg;
6825     GNUNET_SERVER_notification_context_unicast (nc, client,
6826                                                 &warn.header,
6827                                                 GNUNET_NO);
6828     GNUNET_SERVER_receive_done (client, GNUNET_OK);
6829     return;
6830   }
6831
6832   /* Initialize context */
6833   resp = GNUNET_malloc (sizeof (struct GNUNET_MESH_LocalMonitor));
6834   *resp = *msg;
6835   resp->header.size = htons (sizeof (struct GNUNET_MESH_LocalMonitor));
6836   GNUNET_SERVER_notification_context_unicast (nc, c->handle,
6837                                               &resp->header, GNUNET_NO);
6838   GNUNET_free (resp);
6839
6840   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
6841               "Monitor tunnel request from client %u completed\n",
6842               c->id);
6843   GNUNET_SERVER_receive_done (client, GNUNET_OK);
6844 }
6845
6846
6847 /**
6848  * Functions to handle messages from clients
6849  */
6850 static struct GNUNET_SERVER_MessageHandler client_handlers[] = {
6851   {&handle_local_new_client, NULL,
6852    GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT, 0},
6853   {&handle_local_channel_create, NULL,
6854    GNUNET_MESSAGE_TYPE_MESH_CHANNEL_CREATE,
6855    sizeof (struct GNUNET_MESH_ChannelMessage)},
6856   {&handle_local_channel_destroy, NULL,
6857    GNUNET_MESSAGE_TYPE_MESH_CHANNEL_DESTROY,
6858    sizeof (struct GNUNET_MESH_ChannelMessage)},
6859   {&handle_local_data, NULL,
6860    GNUNET_MESSAGE_TYPE_MESH_LOCAL_DATA, 0},
6861   {&handle_local_ack, NULL,
6862    GNUNET_MESSAGE_TYPE_MESH_LOCAL_ACK,
6863    sizeof (struct GNUNET_MESH_LocalAck)},
6864   {&handle_local_get_tunnels, NULL,
6865    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNELS,
6866    sizeof (struct GNUNET_MessageHeader)},
6867   {&handle_local_show_tunnel, NULL,
6868    GNUNET_MESSAGE_TYPE_MESH_LOCAL_INFO_TUNNEL,
6869      sizeof (struct GNUNET_MESH_LocalMonitor)},
6870   {NULL, NULL, 0, 0}
6871 };
6872
6873
6874 /**
6875  * Method called whenever a given peer connects.
6876  *
6877  * @param cls closure
6878  * @param peer peer identity this notification is about
6879  */
6880 static void
6881 core_connect (void *cls, const struct GNUNET_PeerIdentity *peer)
6882 {
6883   struct MeshPeer *pi;
6884   struct MeshPeerPath *path;
6885
6886   DEBUG_CONN ("Peer connected\n");
6887   DEBUG_CONN ("     %s\n", GNUNET_i2s (&my_full_id));
6888   pi = peer_get (peer);
6889   if (myid == pi->id)
6890   {
6891     DEBUG_CONN ("     (self)\n");
6892     path = path_new (1);
6893   }
6894   else
6895   {
6896     DEBUG_CONN ("     %s\n", GNUNET_i2s (peer));
6897     path = path_new (2);
6898     path->peers[1] = pi->id;
6899     GNUNET_PEER_change_rc (pi->id, 1);
6900     GNUNET_STATISTICS_update (stats, "# peers", 1, GNUNET_NO);
6901   }
6902   path->peers[0] = myid;
6903   GNUNET_PEER_change_rc (myid, 1);
6904   peer_add_path (pi, path, GNUNET_YES);
6905
6906   pi->connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES);
6907   return;
6908 }
6909
6910
6911 /**
6912  * Method called whenever a peer disconnects.
6913  *
6914  * @param cls closure
6915  * @param peer peer identity this notification is about
6916  */
6917 static void
6918 core_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
6919 {
6920   struct MeshPeer *pi;
6921
6922   DEBUG_CONN ("Peer disconnected\n");
6923   pi = GNUNET_CONTAINER_multihashmap_get (peers, &peer->hashPubKey);
6924   if (NULL == pi)
6925   {
6926     GNUNET_break (0);
6927     return;
6928   }
6929
6930   GNUNET_CONTAINER_multihashmap_iterate (pi->connections,
6931                                          connection_broken,
6932                                          pi);
6933   GNUNET_CONTAINER_multihashmap_destroy (pi->connections);
6934   pi->connections = NULL;
6935   if (myid == pi->id)
6936   {
6937     DEBUG_CONN ("     (self)\n");
6938   }
6939   GNUNET_STATISTICS_update (stats, "# peers", -1, GNUNET_NO);
6940
6941   return;
6942 }
6943
6944
6945 /**
6946  * Install server (service) handlers and start listening to clients.
6947  */
6948 static void
6949 server_init (void)
6950 {
6951   GNUNET_SERVER_add_handlers (server_handle, client_handlers);
6952   GNUNET_SERVER_connect_notify (server_handle,
6953                                 &handle_local_client_connect, NULL);
6954   GNUNET_SERVER_disconnect_notify (server_handle,
6955                                    &handle_local_client_disconnect, NULL);
6956   nc = GNUNET_SERVER_notification_context_create (server_handle, 1);
6957
6958   clients_head = NULL;
6959   clients_tail = NULL;
6960   next_client_id = 0;
6961   GNUNET_SERVER_resume (server_handle);
6962 }
6963
6964
6965 /**
6966  * To be called on core init/fail.
6967  *
6968  * @param cls Closure (config)
6969  * @param identity the public identity of this peer
6970  */
6971 static void
6972 core_init (void *cls, 
6973            const struct GNUNET_PeerIdentity *identity)
6974 {
6975   const struct GNUNET_CONFIGURATION_Handle *c = cls;
6976   static int i = 0;
6977
6978   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Core init\n");
6979   if (0 != memcmp (identity, &my_full_id, sizeof (my_full_id)))
6980   {
6981     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Wrong CORE service\n"));
6982     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6983                 " core id %s\n",
6984                 GNUNET_i2s (identity));
6985     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
6986                 " my id %s\n",
6987                 GNUNET_i2s (&my_full_id));
6988     GNUNET_CORE_disconnect (core_handle);
6989     core_handle = GNUNET_CORE_connect (c, /* Main configuration */
6990                                        NULL,      /* Closure passed to MESH functions */
6991                                        &core_init,        /* Call core_init once connected */
6992                                        &core_connect,     /* Handle connects */
6993                                        &core_disconnect,  /* remove peers on disconnects */
6994                                        NULL,      /* Don't notify about all incoming messages */
6995                                        GNUNET_NO, /* For header only in notification */
6996                                        NULL,      /* Don't notify about all outbound messages */
6997                                        GNUNET_NO, /* For header-only out notification */
6998                                        core_handlers);    /* Register these handlers */
6999     if (10 < i++)
7000       GNUNET_abort();
7001   }
7002   server_init ();
7003   return;
7004 }
7005
7006
7007 /******************************************************************************/
7008 /************************      MAIN FUNCTIONS      ****************************/
7009 /******************************************************************************/
7010
7011 /**
7012  * Iterator over tunnel hash map entries to destroy the tunnel during shutdown.
7013  *
7014  * @param cls closure
7015  * @param key current key code
7016  * @param value value in the hash map
7017  * @return GNUNET_YES if we should continue to iterate,
7018  *         GNUNET_NO if not.
7019  */
7020 static int
7021 shutdown_tunnel (void *cls, const struct GNUNET_HashCode * key, void *value)
7022 {
7023   struct MeshPeer *p = value;
7024   struct MeshTunnel2 *t = p->tunnel;
7025
7026   if (NULL != t)
7027     tunnel_destroy (t);
7028   return GNUNET_YES;
7029 }
7030
7031
7032 /**
7033  * Task run during shutdown.
7034  *
7035  * @param cls unused
7036  * @param tc unused
7037  */
7038 static void
7039 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
7040 {
7041   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shutting down\n");
7042
7043   if (core_handle != NULL)
7044   {
7045     GNUNET_CORE_disconnect (core_handle);
7046     core_handle = NULL;
7047   }
7048   GNUNET_CONTAINER_multihashmap_iterate (peers, &shutdown_tunnel, NULL);
7049   if (dht_handle != NULL)
7050   {
7051     GNUNET_DHT_disconnect (dht_handle);
7052     dht_handle = NULL;
7053   }
7054   if (nc != NULL)
7055   {
7056     GNUNET_SERVER_notification_context_destroy (nc);
7057     nc = NULL;
7058   }
7059   if (GNUNET_SCHEDULER_NO_TASK != announce_id_task)
7060   {
7061     GNUNET_SCHEDULER_cancel (announce_id_task);
7062     announce_id_task = GNUNET_SCHEDULER_NO_TASK;
7063   }
7064   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "shut down\n");
7065 }
7066
7067
7068 /**
7069  * Process mesh requests.
7070  *
7071  * @param cls closure
7072  * @param server the initialized server
7073  * @param c configuration to use
7074  */
7075 static void
7076 run (void *cls, struct GNUNET_SERVER_Handle *server,
7077      const struct GNUNET_CONFIGURATION_Handle *c)
7078 {
7079   char *keyfile;
7080   struct GNUNET_CRYPTO_EccPrivateKey *pk;
7081
7082   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "starting to run\n");
7083   server_handle = server;
7084   GNUNET_SERVER_suspend (server_handle);
7085
7086   if (GNUNET_OK !=
7087       GNUNET_CONFIGURATION_get_value_filename (c, "PEER", "PRIVATE_KEY",
7088                                                &keyfile))
7089   {
7090     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7091                 _
7092                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7093                 "mesh", "peer/privatekey");
7094     GNUNET_SCHEDULER_shutdown ();
7095     return;
7096   }
7097
7098   if (GNUNET_OK !=
7099       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_CONNECTION_TIME",
7100                                            &refresh_connection_time))
7101   {
7102     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7103                 _
7104                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7105                 "mesh", "refresh path time");
7106     GNUNET_SCHEDULER_shutdown ();
7107     return;
7108   }
7109
7110   if (GNUNET_OK !=
7111       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "ID_ANNOUNCE_TIME",
7112                                            &id_announce_time))
7113   {
7114     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7115                 _
7116                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7117                 "mesh", "id announce time");
7118     GNUNET_SCHEDULER_shutdown ();
7119     return;
7120   }
7121
7122   if (GNUNET_OK !=
7123       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "CONNECT_TIMEOUT",
7124                                            &connect_timeout))
7125   {
7126     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7127                 _
7128                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7129                 "mesh", "connect timeout");
7130     GNUNET_SCHEDULER_shutdown ();
7131     return;
7132   }
7133
7134   if (GNUNET_OK !=
7135       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
7136                                              &max_msgs_queue))
7137   {
7138     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7139                 _
7140                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7141                 "mesh", "max msgs queue");
7142     GNUNET_SCHEDULER_shutdown ();
7143     return;
7144   }
7145
7146   if (GNUNET_OK !=
7147       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_CONNECTIONS",
7148                                              &max_connections))
7149   {
7150     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
7151                 _
7152                 ("%s service is lacking key configuration settings (%s).  Exiting.\n"),
7153                 "mesh", "max tunnels");
7154     GNUNET_SCHEDULER_shutdown ();
7155     return;
7156   }
7157
7158   if (GNUNET_OK !=
7159       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DEFAULT_TTL",
7160                                              &default_ttl))
7161   {
7162     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7163                 _
7164                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7165                 "mesh", "default ttl", 64);
7166     default_ttl = 64;
7167   }
7168
7169   if (GNUNET_OK !=
7170       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_PEERS",
7171                                              &max_peers))
7172   {
7173     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7174                 _("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7175                 "mesh", "max peers", 1000);
7176     max_peers = 1000;
7177   }
7178
7179   if (GNUNET_OK !=
7180       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DROP_PERCENT",
7181                                              &drop_percent))
7182   {
7183     drop_percent = 0;
7184   }
7185   else
7186   {
7187     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7188                 "Mesh is running with drop mode enabled. "
7189                 "This is NOT a good idea! "
7190                 "Remove the DROP_PERCENT option from your configuration.\n");
7191   }
7192
7193   if (GNUNET_OK !=
7194       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "DHT_REPLICATION_LEVEL",
7195                                              &dht_replication_level))
7196   {
7197     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
7198                 _
7199                 ("%s service is lacking key configuration settings (%s). Using default (%u).\n"),
7200                 "mesh", "dht replication level", 3);
7201     dht_replication_level = 3;
7202   }
7203
7204   connections = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_YES);
7205   peers = GNUNET_CONTAINER_multihashmap_create (32, GNUNET_NO);
7206   ports = GNUNET_CONTAINER_multihashmap32_create (32);
7207
7208   dht_handle = GNUNET_DHT_connect (c, 64);
7209   if (NULL == dht_handle)
7210   {
7211     GNUNET_break (0);
7212   }
7213   stats = GNUNET_STATISTICS_create ("mesh", c);
7214
7215   /* Scheduled the task to clean up when shutdown is called */
7216   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
7217                                 NULL);
7218   pk = GNUNET_CRYPTO_ecc_key_create_from_file (keyfile);
7219   GNUNET_free (keyfile);
7220   GNUNET_assert (NULL != pk);
7221   my_private_key = pk;
7222   GNUNET_CRYPTO_ecc_key_get_public (my_private_key, &my_public_key);
7223   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
7224                       &my_full_id.hashPubKey);
7225   myid = GNUNET_PEER_intern (&my_full_id);
7226   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
7227               "Mesh for peer [%s] starting\n",
7228               GNUNET_i2s(&my_full_id));
7229
7230   core_handle = GNUNET_CORE_connect (c, /* Main configuration */
7231                                      NULL,      /* Closure passed to MESH functions */
7232                                      &core_init,        /* Call core_init once connected */
7233                                      &core_connect,     /* Handle connects */
7234                                      &core_disconnect,  /* remove peers on disconnects */
7235                                      NULL,      /* Don't notify about all incoming messages */
7236                                      GNUNET_NO, /* For header only in notification */
7237                                      NULL,      /* Don't notify about all outbound messages */
7238                                      GNUNET_NO, /* For header-only out notification */
7239                                      core_handlers);    /* Register these handlers */
7240   if (NULL == core_handle)
7241   {
7242     GNUNET_break (0);
7243     GNUNET_SCHEDULER_shutdown ();
7244     return;
7245   }
7246   announce_id_task = GNUNET_SCHEDULER_add_now (&announce_id, cls);
7247   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Mesh service running\n");
7248 }
7249
7250
7251 /**
7252  * The main function for the mesh service.
7253  *
7254  * @param argc number of arguments from the command line
7255  * @param argv command line arguments
7256  * @return 0 ok, 1 on error
7257  */
7258 int
7259 main (int argc, char *const *argv)
7260 {
7261   int ret;
7262   int r;
7263
7264   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main()\n");
7265   r = GNUNET_SERVICE_run (argc, argv, "mesh", GNUNET_SERVICE_OPTION_NONE, &run,
7266                           NULL);
7267   ret = (GNUNET_OK == r) ? 0 : 1;
7268   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "main() END\n");
7269
7270   INTERVAL_SHOW;
7271
7272   return ret;
7273 }