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