- don't try to rekey loopback tunnels (#3254)
[oweals/gnunet.git] / src / mesh / gnunet-service-mesh_connection.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_connection.c
23  * @brief GNUnet MESH service connection handling
24  * @author Bartlomiej Polot
25  */
26
27 #include "platform.h"
28 #include "gnunet_util_lib.h"
29
30 #include "gnunet_statistics_service.h"
31
32 #include "mesh_path.h"
33 #include "mesh_protocol.h"
34 #include "mesh.h"
35 #include "gnunet-service-mesh_connection.h"
36 #include "gnunet-service-mesh_peer.h"
37 #include "gnunet-service-mesh_tunnel.h"
38
39
40 #define LOG(level, ...) GNUNET_log_from (level,"mesh-con",__VA_ARGS__)
41
42 #define MESH_MAX_POLL_TIME      GNUNET_TIME_relative_multiply (\
43                                   GNUNET_TIME_UNIT_MINUTES,\
44                                   10)
45 #define AVG_MSGS                32
46
47
48 /******************************************************************************/
49 /********************************   STRUCTS  **********************************/
50 /******************************************************************************/
51
52 /**
53  * Struct to encapsulate all the Flow Control information to a peer to which
54  * we are directly connected (on a core level).
55  */
56 struct MeshFlowControl
57 {
58   /**
59    * Connection this controls.
60    */
61   struct MeshConnection *c;
62
63   /**
64    * How many messages are in the queue on this connection.
65    */
66   unsigned int queue_n;
67
68   /**
69    * How many messages do we accept in the queue.
70    */
71   unsigned int queue_max;
72
73   /**
74    * Next ID to use.
75    */
76   uint32_t next_pid;
77
78   /**
79    * ID of the last packet sent towards the peer.
80    */
81   uint32_t last_pid_sent;
82
83   /**
84    * ID of the last packet received from the peer.
85    */
86   uint32_t last_pid_recv;
87
88   /**
89    * Last ACK sent to the peer (peer can't send more than this PID).
90    */
91   uint32_t last_ack_sent;
92
93   /**
94    * Last ACK sent towards the origin (for traffic towards leaf node).
95    */
96   uint32_t last_ack_recv;
97
98   /**
99    * Task to poll the peer in case of a lost ACK causes stall.
100    */
101   GNUNET_SCHEDULER_TaskIdentifier poll_task;
102
103   /**
104    * How frequently to poll for ACKs.
105    */
106   struct GNUNET_TIME_Relative poll_time;
107
108   /**
109    * Queued poll message, to cancel if not necessary anymore (got ACK).
110    */
111   struct MeshConnectionQueue *poll_msg;
112
113   /**
114    * Queued poll message, to cancel if not necessary anymore (got ACK).
115    */
116   struct MeshConnectionQueue *ack_msg;
117 };
118
119 /**
120  * Keep a record of the last messages sent on this connection.
121  */
122 struct MeshConnectionPerformance
123 {
124   /**
125    * Circular buffer for storing measurements.
126    */
127   double usecsperbyte[AVG_MSGS];
128
129   /**
130    * Running average of @c usecsperbyte.
131    */
132   double avg;
133
134   /**
135    * How many values of @c usecsperbyte are valid.
136    */
137   uint16_t size;
138
139   /**
140    * Index of the next "free" position in @c usecsperbyte.
141    */
142   uint16_t idx;
143 };
144
145
146 /**
147  * Struct containing all information regarding a connection to a peer.
148  */
149 struct MeshConnection
150 {
151   /**
152    * Tunnel this connection is part of.
153    */
154   struct MeshTunnel3 *t;
155
156   /**
157    * Flow control information for traffic fwd.
158    */
159   struct MeshFlowControl fwd_fc;
160
161   /**
162    * Flow control information for traffic bck.
163    */
164   struct MeshFlowControl bck_fc;
165
166   /**
167    * Measure connection performance on the endpoint.
168    */
169   struct MeshConnectionPerformance *perf;
170
171   /**
172    * ID of the connection.
173    */
174   struct GNUNET_HashCode id;
175
176   /**
177    * State of the connection.
178    */
179   enum MeshConnectionState state;
180
181   /**
182    * Path being used for the tunnel. At the origin of the connection
183    * it's a pointer to the destination's path pool, otherwise just a copy.
184    */
185   struct MeshPeerPath *path;
186
187   /**
188    * Position of the local peer in the path.
189    */
190   unsigned int own_pos;
191
192   /**
193    * Task to keep the used paths alive at the owner,
194    * time tunnel out on all the other peers.
195    */
196   GNUNET_SCHEDULER_TaskIdentifier fwd_maintenance_task;
197
198   /**
199    * Task to keep the used paths alive at the destination,
200    * time tunnel out on all the other peers.
201    */
202   GNUNET_SCHEDULER_TaskIdentifier bck_maintenance_task;
203
204   /**
205    * Queue handle for maintainance traffic. One handle for FWD and BCK since
206    * one peer never needs to maintain both directions (no loopback connections).
207    */
208   struct MeshPeerQueue *maintenance_q;
209
210   /**
211    * Counter to do exponential backoff when creating a connection (max 64).
212    */
213   unsigned short create_retry;
214
215   /**
216    * Pending message count.
217    */
218   int pending_messages;
219
220   /**
221    * Destroy flag: if true, destroy on last message.
222    */
223   int destroy;
224 };
225
226 /**
227  * Handle for messages queued but not yet sent.
228  */
229 struct MeshConnectionQueue
230 {
231   /**
232    * Peer queue handle, to cancel if necessary.
233    */
234   struct MeshPeerQueue *q;
235
236   /**
237    * Was this a forced message? (Do not account for it)
238    */
239   int forced;
240
241   /**
242    * Continuation to call once sent.
243    */
244   GMC_sent cont;
245
246   /**
247    * Closure for @c cont.
248    */
249   void *cont_cls;
250 };
251
252 /******************************************************************************/
253 /*******************************   GLOBALS  ***********************************/
254 /******************************************************************************/
255
256 /**
257  * Global handle to the statistics service.
258  */
259 extern struct GNUNET_STATISTICS_Handle *stats;
260
261 /**
262  * Local peer own ID (memory efficient handle).
263  */
264 extern GNUNET_PEER_Id myid;
265
266 /**
267  * Local peer own ID (full value).
268  */
269 extern struct GNUNET_PeerIdentity my_full_id;
270
271 /**
272  * Connections known, indexed by cid (MeshConnection).
273  */
274 static struct GNUNET_CONTAINER_MultiHashMap *connections;
275
276 /**
277  * How many connections are we willing to maintain.
278  * Local connections are always allowed, even if there are more connections than max.
279  */
280 static unsigned long long max_connections;
281
282 /**
283  * How many messages *in total* are we willing to queue, divide by number of
284  * connections to get connection queue size.
285  */
286 static unsigned long long max_msgs_queue;
287
288 /**
289  * How often to send path keepalives. Paths timeout after 4 missed.
290  */
291 static struct GNUNET_TIME_Relative refresh_connection_time;
292
293 /**
294  * How often to send path create / ACKs.
295  */
296 static struct GNUNET_TIME_Relative create_connection_time;
297
298
299 /******************************************************************************/
300 /********************************   STATIC  ***********************************/
301 /******************************************************************************/
302
303 #if 0 // avoid compiler warning for unused static function
304 static void
305 fc_debug (struct MeshFlowControl *fc)
306 {
307   LOG (GNUNET_ERROR_TYPE_DEBUG, "    IN: %u/%u\n",
308               fc->last_pid_recv, fc->last_ack_sent);
309   LOG (GNUNET_ERROR_TYPE_DEBUG, "    OUT: %u/%u\n",
310               fc->last_pid_sent, fc->last_ack_recv);
311   LOG (GNUNET_ERROR_TYPE_DEBUG, "    QUEUE: %u/%u\n",
312               fc->queue_n, fc->queue_max);
313 }
314
315 static void
316 connection_debug (struct MeshConnection *c)
317 {
318   if (NULL == c)
319   {
320     LOG (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CONNECTION ***\n");
321     return;
322   }
323   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s:%X\n",
324               peer2s (c->t->peer), GMC_2s (c));
325   LOG (GNUNET_ERROR_TYPE_DEBUG, "  state: %u, pending msgs: %u\n",
326               c->state, c->pending_messages);
327   LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
328   fc_debug (&c->fwd_fc);
329   LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
330   fc_debug (&c->bck_fc);
331 }
332 #endif
333
334
335 /**
336  * Schedule next keepalive task, taking in consideration
337  * the connection state and number of retries.
338  *
339  * @param c Connection for which to schedule the next keepalive.
340  * @param fwd Direction for the next keepalive.
341  */
342 static void
343 schedule_next_keepalive (struct MeshConnection *c, int fwd);
344
345
346 /**
347  * Get string description for tunnel state.
348  *
349  * @param s Tunnel state.
350  *
351  * @return String representation.
352  */
353 static const char *
354 GMC_state2s (enum MeshConnectionState s)
355 {
356   switch (s)
357   {
358     case MESH_CONNECTION_NEW:
359       return "MESH_CONNECTION_NEW";
360     case MESH_CONNECTION_SENT:
361       return "MESH_CONNECTION_SENT";
362     case MESH_CONNECTION_ACK:
363       return "MESH_CONNECTION_ACK";
364     case MESH_CONNECTION_READY:
365       return "MESH_CONNECTION_READY";
366     case MESH_CONNECTION_DESTROYED:
367       return "MESH_CONNECTION_DESTROYED";
368     default:
369       return "MESH_CONNECTION_STATE_ERROR";
370   }
371 }
372
373
374 /**
375  * Initialize a Flow Control structure to the initial state.
376  *
377  * @param fc Flow Control structure to initialize.
378  */
379 static void
380 fc_init (struct MeshFlowControl *fc)
381 {
382   fc->next_pid = 0;
383   fc->last_pid_sent = (uint32_t) -1; /* Next (expected) = 0 */
384   fc->last_pid_recv = (uint32_t) -1;
385   fc->last_ack_sent = (uint32_t) 0;
386   fc->last_ack_recv = (uint32_t) 0;
387   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
388   fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
389   fc->queue_n = 0;
390   fc->queue_max = (max_msgs_queue / max_connections) + 1;
391 }
392
393
394 /**
395  * Find a connection.
396  *
397  * @param cid Connection ID.
398  */
399 static struct MeshConnection *
400 connection_get (const struct GNUNET_HashCode *cid)
401 {
402   return GNUNET_CONTAINER_multihashmap_get (connections, cid);
403 }
404
405
406 static void
407 connection_change_state (struct MeshConnection* c,
408                          enum MeshConnectionState state)
409 {
410   LOG (GNUNET_ERROR_TYPE_DEBUG,
411               "Connection %s state was %s\n",
412               GMC_2s (c), GMC_state2s (c->state));
413   if (MESH_CONNECTION_DESTROYED == c->state)
414   {
415     LOG (GNUNET_ERROR_TYPE_DEBUG, "state not changing anymore\n");
416     return;
417   }
418   LOG (GNUNET_ERROR_TYPE_DEBUG,
419               "Connection %s state is now %s\n",
420               GMC_2s (c), GMC_state2s (state));
421   c->state = state;
422   if (MESH_CONNECTION_READY == state)
423     c->create_retry = 1;
424 }
425
426
427 /**
428  * Callback called when a queued ACK message is sent.
429  *
430  * @param cls Closure (FC).
431  * @param c Connection this message was on.
432  * @param q Queue handler this call invalidates.
433  * @param type Type of message sent.
434  * @param fwd Was this a FWD going message?
435  * @param size Size of the message.
436  */
437 static void
438 ack_sent (void *cls,
439           struct MeshConnection *c,
440           struct MeshConnectionQueue *q,
441           uint16_t type, int fwd, size_t size)
442 {
443   struct MeshFlowControl *fc = cls;
444
445   fc->ack_msg = NULL;
446 }
447
448
449 /**
450  * Send an ACK on the connection, informing the predecessor about
451  * the available buffer space. Should not be called in case the peer
452  * is origin (no predecessor) in the @c fwd direction.
453  *
454  * Note that for fwd ack, the FWD mean forward *traffic* (root->dest),
455  * the ACK itself goes "back" (dest->root).
456  *
457  * @param c Connection on which to send the ACK.
458  * @param buffer How much space free to advertise?
459  * @param fwd Is this FWD ACK? (Going dest -> root)
460  * @param force Don't optimize out.
461  */
462 static void
463 send_ack (struct MeshConnection *c, unsigned int buffer, int fwd, int force)
464 {
465   struct MeshFlowControl *next_fc;
466   struct MeshFlowControl *prev_fc;
467   struct GNUNET_MESH_ACK msg;
468   uint32_t ack;
469   int delta;
470
471   /* If origin, there is no connection to send ACKs. Wrong function! */
472   if (GMC_is_origin (c, fwd))
473   {
474     LOG (GNUNET_ERROR_TYPE_DEBUG, "connection %s is origin in %s\n",
475          GMC_2s (c), GM_f2s (fwd));
476     GNUNET_break (0);
477     return;
478   }
479
480   next_fc = fwd ? &c->fwd_fc : &c->bck_fc;
481   prev_fc = fwd ? &c->bck_fc : &c->fwd_fc;
482
483   LOG (GNUNET_ERROR_TYPE_DEBUG, "connection send %s ack on %s\n",
484        GM_f2s (fwd), GMC_2s (c));
485
486   /* Check if we need to transmit the ACK. */
487   delta = prev_fc->last_ack_sent - prev_fc->last_pid_recv;
488   if (3 < delta && buffer < delta && GNUNET_NO == force)
489   {
490     LOG (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, buffer > 3\n");
491     LOG (GNUNET_ERROR_TYPE_DEBUG,
492          "  last pid recv: %u, last ack sent: %u\n",
493          prev_fc->last_pid_recv, prev_fc->last_ack_sent);
494     return;
495   }
496
497   /* Ok, ACK might be necessary, what PID to ACK? */
498   ack = prev_fc->last_pid_recv + buffer;
499   LOG (GNUNET_ERROR_TYPE_DEBUG, " ACK %u\n", ack);
500   LOG (GNUNET_ERROR_TYPE_DEBUG,
501        " last pid %u, last ack %u, qmax %u, q %u\n",
502        prev_fc->last_pid_recv, prev_fc->last_ack_sent,
503        next_fc->queue_max, next_fc->queue_n);
504   if (ack == prev_fc->last_ack_sent && GNUNET_NO == force)
505   {
506     LOG (GNUNET_ERROR_TYPE_DEBUG, "Not sending FWD ACK, not needed\n");
507     return;
508   }
509
510   /* Check if message is already in queue */
511   if (NULL != prev_fc->ack_msg)
512   {
513     if (GM_is_pid_bigger (ack, prev_fc->last_ack_sent))
514     {
515       LOG (GNUNET_ERROR_TYPE_DEBUG, " canceling old ACK\n");
516       GMC_cancel (prev_fc->ack_msg);
517       /* GMC_cancel triggers ack_sent(), which clears fc->ack_msg */
518     }
519     else
520     {
521       LOG (GNUNET_ERROR_TYPE_DEBUG, " same ACK already in queue\n");
522       return;
523     }
524   }
525
526   prev_fc->last_ack_sent = ack;
527
528   /* Build ACK message and send on connection */
529   msg.header.size = htons (sizeof (msg));
530   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_ACK);
531   msg.ack = htonl (ack);
532   msg.cid = c->id;
533
534   prev_fc->ack_msg = GMC_send_prebuilt_message (&msg.header, c,
535                                                 !fwd, GNUNET_YES,
536                                                 &ack_sent, prev_fc);
537 }
538
539
540 /**
541  * Callback called when a queued message is sent.
542  *
543  * Calculates the average time and connection packet tracking.
544  *
545  * @param cls Closure (ConnectionQueue Handle).
546  * @param c Connection this message was on.
547  * @param type Type of message sent.
548  * @param fwd Was this a FWD going message?
549  * @param size Size of the message.
550  * @param wait Time spent waiting for core (only the time for THIS message)
551  */
552 static void
553 message_sent (void *cls,
554               struct MeshConnection *c, uint16_t type,
555               int fwd, size_t size,
556               struct GNUNET_TIME_Relative wait)
557 {
558   struct MeshConnectionPerformance *p;
559   struct MeshFlowControl *fc;
560   struct MeshConnectionQueue *q = cls;
561   double usecsperbyte;
562   int forced;
563
564   fc = fwd ? &c->fwd_fc : &c->bck_fc;
565   LOG (GNUNET_ERROR_TYPE_DEBUG,
566        "!  sent %s %s\n",
567        GM_f2s (fwd),
568        GM_m2s (type));
569   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  C_P- %p %u\n", c, c->pending_messages);
570   if (NULL != q)
571   {
572     forced = q->forced;
573     if (NULL != q->cont)
574     {
575       LOG (GNUNET_ERROR_TYPE_DEBUG, "!  calling cont\n");
576       q->cont (q->cont_cls, c, q, type, fwd, size);
577     }
578     GNUNET_free (q);
579   }
580   else if (type == GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED)
581   {
582     /* If NULL == q and ENCRYPTED == type, message must have been ch_mngmnt */
583     forced = GNUNET_YES;
584   }
585   else
586   {
587     forced = GNUNET_NO;
588   }
589   c->pending_messages--;
590   if (GNUNET_YES == c->destroy && 0 == c->pending_messages)
591   {
592     LOG (GNUNET_ERROR_TYPE_DEBUG, "!  destroying connection!\n");
593     GMC_destroy (c);
594     return;
595   }
596   /* Send ACK if needed, after accounting for sent ID in fc->queue_n */
597   switch (type)
598   {
599     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
600     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
601       c->maintenance_q = NULL;
602       /* Don't trigger a keepalive for sent ACKs, only SYN and SYNACKs */
603       if (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE == type || !fwd)
604         schedule_next_keepalive (c, fwd);
605       break;
606
607     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
608       fc->last_pid_sent++;
609       LOG (GNUNET_ERROR_TYPE_DEBUG, "!  Q_N- %p %u\n", fc, fc->queue_n);
610       if (GNUNET_NO == forced)
611       {
612         fc->queue_n--;
613         LOG (GNUNET_ERROR_TYPE_DEBUG,
614             "!   accounting pid %u\n",
615             fc->last_pid_sent);
616       }
617       else
618       {
619         LOG (GNUNET_ERROR_TYPE_DEBUG,
620              "!   forced, Q_N not accounting pid %u\n",
621              fc->last_pid_sent);
622       }
623       GMC_send_ack (c, fwd, GNUNET_NO);
624       break;
625
626     case GNUNET_MESSAGE_TYPE_MESH_POLL:
627       fc->poll_msg = NULL;
628       break;
629
630     case GNUNET_MESSAGE_TYPE_MESH_ACK:
631       fc->ack_msg = NULL;
632       break;
633
634     default:
635       break;
636   }
637   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  message sent!\n");
638
639   if (NULL == c->perf)
640     return; /* Only endpoints are interested in timing. */
641
642   p = c->perf;
643   usecsperbyte = ((double) wait.rel_value_us) / size;
644   if (p->size == AVG_MSGS)
645   {
646     /* Array is full. Substract oldest value, add new one and store. */
647     p->avg -= (p->usecsperbyte[p->idx] / AVG_MSGS);
648     p->usecsperbyte[p->idx] = usecsperbyte;
649     p->avg += (p->usecsperbyte[p->idx] / AVG_MSGS);
650   }
651   else
652   {
653     /* Array not yet full. Add current value to avg and store. */
654     p->usecsperbyte[p->idx] = usecsperbyte;
655     p->avg *= p->size;
656     p->avg += p->usecsperbyte[p->idx];
657     p->size++;
658     p->avg /= p->size;
659   }
660   p->idx = (p->idx + 1) % AVG_MSGS;
661 }
662
663
664 /**
665  * Get the previous hop in a connection
666  *
667  * @param c Connection.
668  *
669  * @return Previous peer in the connection.
670  */
671 static struct MeshPeer *
672 get_prev_hop (const struct MeshConnection *c)
673 {
674   GNUNET_PEER_Id id;
675
676   LOG (GNUNET_ERROR_TYPE_DEBUG, "  get prev hop %s [%u/%u]\n",
677        GMC_2s (c), c->own_pos, c->path->length);
678   if (0 == c->own_pos || c->path->length < 2)
679   {
680     LOG (GNUNET_ERROR_TYPE_DEBUG, "  own pos is zero\n");
681     id = c->path->peers[0];
682   }
683   else
684   {
685     LOG (GNUNET_ERROR_TYPE_DEBUG, "  own pos is NOT zero\n");
686     id = c->path->peers[c->own_pos - 1];
687   }
688
689   LOG (GNUNET_ERROR_TYPE_DEBUG, "  id: %u\n", id);
690   LOG (GNUNET_ERROR_TYPE_DEBUG, "  ID: %s\n", GNUNET_i2s (GNUNET_PEER_resolve2 (id)));
691   return GMP_get_short (id);
692 }
693
694
695 /**
696  * Get the next hop in a connection
697  *
698  * @param c Connection.
699  *
700  * @return Next peer in the connection.
701  */
702 static struct MeshPeer *
703 get_next_hop (const struct MeshConnection *c)
704 {
705   GNUNET_PEER_Id id;
706
707   LOG (GNUNET_ERROR_TYPE_DEBUG, "  get next hop %s [%u/%u]\n",
708        GMC_2s (c), c->own_pos, c->path->length);
709   if ((c->path->length - 1) == c->own_pos || c->path->length < 2)
710   {
711     LOG (GNUNET_ERROR_TYPE_DEBUG, "  own pos is end\n");
712     id = c->path->peers[c->path->length - 1];
713   }
714   else
715   {
716     LOG (GNUNET_ERROR_TYPE_DEBUG, "  own pos is NOT end\n");
717     id = c->path->peers[c->own_pos + 1];
718   }
719
720   LOG (GNUNET_ERROR_TYPE_DEBUG, "  id: %u\n", id);
721   LOG (GNUNET_ERROR_TYPE_DEBUG, "  ID: %s\n", GNUNET_i2s (GNUNET_PEER_resolve2 (id)));
722
723   return GMP_get_short (id);
724 }
725
726
727 /**
728  * Get the hop in a connection.
729  *
730  * @param c Connection.
731  * @param fwd Next hop?
732  *
733  * @return Next peer in the connection.
734  */
735 static struct MeshPeer *
736 get_hop (struct MeshConnection *c, int fwd)
737 {
738   if (fwd)
739     return get_next_hop (c);
740   return get_prev_hop (c);
741 }
742
743
744 /**
745  * Is traffic coming from this sender 'FWD' traffic?
746  *
747  * @param c Connection to check.
748  * @param sender Peer identity of neighbor.
749  *
750  * @return #GNUNET_YES in case the sender is the 'prev' hop and therefore
751  *         the traffic is 'FWD'.
752  *         #GNUNET_NO for BCK.
753  *         #GNUNET_SYSERR for errors.
754  */
755 static int
756 is_fwd (const struct MeshConnection *c,
757         const struct GNUNET_PeerIdentity *sender)
758 {
759   GNUNET_PEER_Id id;
760
761   id = GNUNET_PEER_search (sender);
762   if (GMP_get_short_id (get_prev_hop (c)) == id)
763     return GNUNET_YES;
764
765   if (GMP_get_short_id (get_next_hop (c)) == id)
766     return GNUNET_NO;
767
768   GNUNET_break (0);
769   return GNUNET_SYSERR;
770 }
771
772
773 /**
774  * Sends a CONNECTION ACK message in reponse to a received CONNECTION_CREATE
775  * or a first CONNECTION_ACK directed to us.
776  *
777  * @param connection Connection to confirm.
778  * @param fwd Should we send it FWD? (root->dest)
779  *            (First (~SYNACK) goes BCK, second (~ACK) goes FWD)
780  */
781 static void
782 send_connection_ack (struct MeshConnection *connection, int fwd)
783 {
784   struct MeshTunnel3 *t;
785
786   t = connection->t;
787   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send connection %s ACK\n",
788        GM_f2s (!fwd));
789   GMP_queue_add (get_hop (connection, fwd), NULL,
790                  GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
791                  sizeof (struct GNUNET_MESH_ConnectionACK),
792                  connection, fwd, &message_sent, NULL);
793   connection->pending_messages++;
794   if (MESH_TUNNEL3_NEW == GMT_get_cstate (t))
795     GMT_change_cstate (t, MESH_TUNNEL3_WAITING);
796   if (MESH_CONNECTION_READY != connection->state)
797     connection_change_state (connection, MESH_CONNECTION_SENT);
798 }
799
800
801 /**
802  * Send a notification that a connection is broken.
803  *
804  * @param c Connection that is broken.
805  * @param id1 Peer that has disconnected.
806  * @param id2 Peer that has disconnected.
807  * @param fwd Direction towards which to send it.
808  */
809 static void
810 send_broken (struct MeshConnection *c,
811              const struct GNUNET_PeerIdentity *id1,
812              const struct GNUNET_PeerIdentity *id2,
813              int fwd)
814 {
815   struct GNUNET_MESH_ConnectionBroken msg;
816
817   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectionBroken));
818   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN);
819   msg.cid = c->id;
820   msg.peer1 = *id1;
821   msg.peer2 = *id2;
822   GMC_send_prebuilt_message (&msg.header, c, fwd, GNUNET_YES, NULL, NULL);
823 }
824
825
826 /**
827  * Send a notification that a connection is broken, when a connection
828  * isn't even created.
829  *
830  * @param connection_id Connection ID.
831  * @param id1 Peer that has disconnected.
832  * @param id2 Peer that has disconnected.
833  * @param peer Peer to notify (neighbor who sent the connection).
834  */
835 static void
836 send_broken2 (struct GNUNET_HashCode *connection_id,
837              const struct GNUNET_PeerIdentity *id1,
838              const struct GNUNET_PeerIdentity *id2,
839              GNUNET_PEER_Id peer_id)
840 {
841   struct GNUNET_MESH_ConnectionBroken *msg;
842   struct MeshPeer *neighbor;
843
844   msg = GNUNET_new (struct GNUNET_MESH_ConnectionBroken);
845   msg->header.size = htons (sizeof (struct GNUNET_MESH_ConnectionBroken));
846   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN);
847   msg->cid = *connection_id;
848   msg->peer1 = *id1;
849   msg->peer2 = *id2;
850   neighbor = GMP_get_short (peer_id);
851   GMP_queue_add (neighbor, msg,
852                  GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED,
853                  sizeof (struct GNUNET_MESH_ConnectionBroken),
854                  NULL, GNUNET_SYSERR, /* connection, fwd */
855                  NULL, NULL); /* continuation */
856 }
857
858
859 /**
860  * Send keepalive packets for a connection.
861  *
862  * @param c Connection to keep alive..
863  * @param fwd Is this a FWD keepalive? (owner -> dest).
864  */
865 static void
866 connection_keepalive (struct MeshConnection *c, int fwd)
867 {
868   struct GNUNET_MESH_ConnectionKeepAlive *msg;
869   size_t size = sizeof (struct GNUNET_MESH_ConnectionKeepAlive);
870   char cbuf[size];
871
872   LOG (GNUNET_ERROR_TYPE_DEBUG,
873        "sending %s keepalive for connection %s]\n",
874        GM_f2s (fwd), GMC_2s (c));
875
876   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) cbuf;
877   msg->header.size = htons (size);
878   msg->header.type = htons (GNUNET_MESSAGE_TYPE_MESH_KEEPALIVE);
879   msg->cid = c->id;
880   msg->reserved = htonl (0);
881
882   GMC_send_prebuilt_message (&msg->header, c, fwd, GNUNET_YES, NULL, NULL);
883 }
884
885
886 /**
887  * Send CONNECTION_{CREATE/ACK} packets for a connection.
888  *
889  * @param c Connection for which to send the message.
890  * @param fwd If #GNUNET_YES, send CREATE, otherwise send ACK.
891  */
892 static void
893 connection_recreate (struct MeshConnection *c, int fwd)
894 {
895   LOG (GNUNET_ERROR_TYPE_DEBUG, "sending connection recreate\n");
896   if (fwd)
897     GMC_send_create (c);
898   else
899     send_connection_ack (c, GNUNET_NO);
900 }
901
902
903 /**
904  * Generic connection timer management.
905  * Depending on the role of the peer in the connection will send the
906  * appropriate message (build or keepalive)
907  *
908  * @param c Conncetion to maintain.
909  * @param fwd Is FWD?
910  */
911 static void
912 connection_maintain (struct MeshConnection *c, int fwd)
913 {
914   if (GNUNET_NO != c->destroy)
915     return;
916
917   if (MESH_TUNNEL3_SEARCHING == GMT_get_cstate (c->t))
918   {
919     /* TODO DHT GET with RO_BART */
920     return;
921   }
922   switch (c->state)
923   {
924     case MESH_CONNECTION_NEW:
925       GNUNET_break (0);
926       /* fall-through */
927     case MESH_CONNECTION_SENT:
928       connection_recreate (c, fwd);
929       break;
930     case MESH_CONNECTION_READY:
931       connection_keepalive (c, fwd);
932       break;
933     default:
934       break;
935   }
936 }
937
938
939 /**
940  * Keep the connection alive in the FWD direction.
941  *
942  * @param cls Closure (connection to keepalive).
943  * @param tc TaskContext.
944  */
945 static void
946 connection_fwd_keepalive (void *cls,
947                           const struct GNUNET_SCHEDULER_TaskContext *tc)
948 {
949   struct MeshConnection *c = cls;
950
951   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
952   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
953     return;
954
955   connection_maintain (c, GNUNET_YES);
956
957   /* Next execution will be scheduled by message_sent */
958 }
959
960
961 /**
962  * Keep the connection alive in the BCK direction.
963  *
964  * TODO refactor and merge with connection_fwd_keepalive.
965  *
966  * @param cls Closure (connection to keepalive).
967  * @param tc TaskContext.
968  */
969 static void
970 connection_bck_keepalive (void *cls,
971                           const struct GNUNET_SCHEDULER_TaskContext *tc)
972 {
973   struct MeshConnection *c = cls;
974
975   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
976   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
977     return;
978
979   connection_maintain (c, GNUNET_NO);
980
981   /* Next execution will be scheduled by message_sent */
982 }
983
984
985 /**
986  * Schedule next keepalive task, taking in consideration
987  * the connection state and number of retries.
988  *
989  * If the peer is not the origin, do nothing.
990  *
991  * @param c Connection for which to schedule the next keepalive.
992  * @param fwd Direction for the next keepalive.
993  */
994 static void
995 schedule_next_keepalive (struct MeshConnection *c, int fwd)
996 {
997   struct GNUNET_TIME_Relative delay;
998   GNUNET_SCHEDULER_TaskIdentifier *task_id;
999   GNUNET_SCHEDULER_Task keepalive_task;
1000
1001   if (GNUNET_NO == GMC_is_origin (c, fwd))
1002     return;
1003
1004   /* Calculate delay to use, depending on the state of the connection */
1005   if (MESH_CONNECTION_READY == c->state)
1006   {
1007     delay = refresh_connection_time;
1008   }
1009   else
1010   {
1011     if (1 > c->create_retry)
1012       c->create_retry = 1;
1013     delay = GNUNET_TIME_relative_multiply (create_connection_time,
1014                                            c->create_retry);
1015     if (c->create_retry < 64)
1016       c->create_retry *= 2;
1017   }
1018
1019   /* Select direction-dependent parameters */
1020   if (GNUNET_YES == fwd)
1021   {
1022     task_id = &c->fwd_maintenance_task;
1023     keepalive_task = &connection_fwd_keepalive;
1024   }
1025   else
1026   {
1027     task_id = &c->bck_maintenance_task;
1028     keepalive_task = &connection_bck_keepalive;
1029   }
1030
1031   /* Check that no one scheduled it before us (and alert in that case) */
1032   if (GNUNET_SCHEDULER_NO_TASK != *task_id)
1033   {
1034     GNUNET_break (0);
1035     GNUNET_SCHEDULER_cancel (*task_id);
1036   }
1037
1038   /* Schedule the task */
1039   *task_id = GNUNET_SCHEDULER_add_delayed (delay, keepalive_task, c);
1040 }
1041
1042
1043 /**
1044  * @brief Re-initiate traffic on this connection if necessary.
1045  *
1046  * Check if there is traffic queued towards this peer
1047  * and the core transmit handle is NULL (traffic was stalled).
1048  * If so, call core tmt rdy.
1049  *
1050  * @param c Connection on which initiate traffic.
1051  * @param fwd Is this about fwd traffic?
1052  */
1053 static void
1054 connection_unlock_queue (struct MeshConnection *c, int fwd)
1055 {
1056   struct MeshPeer *peer;
1057
1058   LOG (GNUNET_ERROR_TYPE_DEBUG,
1059               "connection_unlock_queue %s on %s\n",
1060               GM_f2s (fwd), GMC_2s (c));
1061
1062   if (GMC_is_terminal (c, fwd))
1063   {
1064     LOG (GNUNET_ERROR_TYPE_DEBUG, " is terminal!\n");
1065     return;
1066   }
1067
1068   peer = get_hop (c, fwd);
1069   GMP_queue_unlock (peer, c);
1070 }
1071
1072
1073 /**
1074  * Cancel all transmissions that belong to a certain connection.
1075  *
1076  * If the connection is scheduled for destruction and no more messages are left,
1077  * the connection will be destroyed by the continuation call.
1078  *
1079  * @param c Connection which to cancel. Might be destroyed during this call.
1080  * @param fwd Cancel fwd traffic?
1081  */
1082 static void
1083 connection_cancel_queues (struct MeshConnection *c, int fwd)
1084 {
1085   struct MeshFlowControl *fc;
1086   struct MeshPeer *peer;
1087
1088   LOG (GNUNET_ERROR_TYPE_DEBUG,
1089        " *** Cancel %s queues for connection %s\n",
1090        GM_f2s (fwd), GMC_2s (c));
1091   if (NULL == c)
1092   {
1093     GNUNET_break (0);
1094     return;
1095   }
1096
1097   fc = fwd ? &c->fwd_fc : &c->bck_fc;
1098   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
1099   {
1100     GNUNET_SCHEDULER_cancel (fc->poll_task);
1101     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1102     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** Cancel POLL in ccq for fc %p\n", fc);
1103   }
1104   peer = get_hop (c, fwd);
1105   GMP_queue_cancel (peer, c);
1106 }
1107
1108
1109 /**
1110  * Function called if a connection has been stalled for a while,
1111  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
1112  *
1113  * @param cls Closure (poll ctx).
1114  * @param tc TaskContext.
1115  */
1116 static void
1117 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1118
1119
1120 /**
1121  * Callback called when a queued POLL message is sent.
1122  *
1123  * @param cls Closure (FC).
1124  * @param c Connection this message was on.
1125  * @param q Queue handler this call invalidates.
1126  * @param type Type of message sent.
1127  * @param fwd Was this a FWD going message?
1128  * @param size Size of the message.
1129  */
1130 static void
1131 poll_sent (void *cls,
1132            struct MeshConnection *c,
1133            struct MeshConnectionQueue *q,
1134            uint16_t type, int fwd, size_t size)
1135 {
1136   struct MeshFlowControl *fc = cls;
1137
1138   if (2 == c->destroy)
1139   {
1140     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL canceled on shutdown\n");
1141     return;
1142   }
1143   LOG (GNUNET_ERROR_TYPE_DEBUG,
1144        " *** POLL sent for , scheduling new one!\n");
1145   fc->poll_msg = NULL;
1146   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
1147   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
1148                                                 &connection_poll, fc);
1149   LOG (GNUNET_ERROR_TYPE_DEBUG, " task %u\n", fc->poll_task);
1150
1151 }
1152
1153 /**
1154  * Function called if a connection has been stalled for a while,
1155  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
1156  *
1157  * @param cls Closure (poll ctx).
1158  * @param tc TaskContext.
1159  */
1160 static void
1161 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1162 {
1163   struct MeshFlowControl *fc = cls;
1164   struct GNUNET_MESH_Poll msg;
1165   struct MeshConnection *c;
1166
1167   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1168   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1169   {
1170     return;
1171   }
1172
1173   c = fc->c;
1174   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** Polling!\n");
1175   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** connection [%s]\n", GMC_2s (c));
1176   LOG (GNUNET_ERROR_TYPE_DEBUG, " ***   %s\n",
1177        fc == &c->fwd_fc ? "FWD" : "BCK");
1178
1179   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
1180   msg.header.size = htons (sizeof (msg));
1181   msg.pid = htonl (fc->last_pid_sent);
1182   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** last pid sent: %u!\n", fc->last_pid_sent);
1183   fc->poll_msg = GMC_send_prebuilt_message (&msg.header, c,
1184                                             fc == &c->fwd_fc, GNUNET_YES,
1185                                             &poll_sent, fc);
1186 }
1187
1188
1189 /**
1190  * Timeout function due to lack of keepalive/traffic from the owner.
1191  * Destroys connection if called.
1192  *
1193  * @param cls Closure (connection to destroy).
1194  * @param tc TaskContext.
1195  */
1196 static void
1197 connection_fwd_timeout (void *cls,
1198                         const struct GNUNET_SCHEDULER_TaskContext *tc)
1199 {
1200   struct MeshConnection *c = cls;
1201
1202   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
1203   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1204     return;
1205
1206   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s FWD timed out. Destroying.\n",
1207        GMC_2s (c));
1208   if (GMC_is_origin (c, GNUNET_YES)) /* If local, leave. */
1209   {
1210     GNUNET_break (0);
1211     return;
1212   }
1213
1214   GMC_destroy (c);
1215 }
1216
1217
1218 /**
1219  * Timeout function due to lack of keepalive/traffic from the destination.
1220  * Destroys connection if called.
1221  *
1222  * @param cls Closure (connection to destroy).
1223  * @param tc TaskContext
1224  */
1225 static void
1226 connection_bck_timeout (void *cls,
1227                         const struct GNUNET_SCHEDULER_TaskContext *tc)
1228 {
1229   struct MeshConnection *c = cls;
1230
1231   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
1232   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1233     return;
1234
1235   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s BCK timed out. Destroying.\n",
1236        GMC_2s (c));
1237
1238   if (GMC_is_origin (c, GNUNET_NO)) /* If local, leave. */
1239   {
1240     GNUNET_break (0);
1241     return;
1242   }
1243
1244   GMC_destroy (c);
1245 }
1246
1247
1248 /**
1249  * Resets the connection timeout task, some other message has done the
1250  * task's job.
1251  * - For the first peer on the direction this means to send
1252  *   a keepalive or a path confirmation message (either create or ACK).
1253  * - For all other peers, this means to destroy the connection,
1254  *   due to lack of activity.
1255  * Starts the timeout if no timeout was running (connection just created).
1256  *
1257  * @param c Connection whose timeout to reset.
1258  * @param fwd Is this forward?
1259  *
1260  * TODO use heap to improve efficiency of scheduler.
1261  */
1262 static void
1263 connection_reset_timeout (struct MeshConnection *c, int fwd)
1264 {
1265   GNUNET_SCHEDULER_TaskIdentifier *ti;
1266   GNUNET_SCHEDULER_Task f;
1267
1268   ti = fwd ? &c->fwd_maintenance_task : &c->bck_maintenance_task;
1269
1270   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s reset timeout\n", GM_f2s (fwd));
1271
1272   if (GNUNET_SCHEDULER_NO_TASK != *ti)
1273     GNUNET_SCHEDULER_cancel (*ti);
1274
1275   if (GMC_is_origin (c, fwd)) /* Startpoint */
1276   {
1277     f = fwd ? &connection_fwd_keepalive : &connection_bck_keepalive;
1278     *ti = GNUNET_SCHEDULER_add_delayed (refresh_connection_time, f, c);
1279   }
1280   else /* Relay, endpoint. */
1281   {
1282     struct GNUNET_TIME_Relative delay;
1283
1284     delay = GNUNET_TIME_relative_multiply (refresh_connection_time, 4);
1285     f = fwd ? &connection_fwd_timeout : &connection_bck_timeout;
1286     *ti = GNUNET_SCHEDULER_add_delayed (delay, f, c);
1287   }
1288 }
1289
1290
1291 /**
1292  * Add the connection to the list of both neighbors.
1293  *
1294  * @param c Connection.
1295  *
1296  * @return #GNUNET_OK if everything went fine
1297  *         #GNUNET_SYSERR if the was an error and @c c is malformed.
1298  */
1299 static int
1300 register_neighbors (struct MeshConnection *c)
1301 {
1302   struct MeshPeer *next_peer;
1303   struct MeshPeer *prev_peer;
1304
1305   next_peer = get_next_hop (c);
1306   prev_peer = get_prev_hop (c);
1307
1308   LOG (GNUNET_ERROR_TYPE_DEBUG, "register neighbors for connection %s\n",
1309        GMC_2s (c));
1310   path_debug (c->path);
1311   LOG (GNUNET_ERROR_TYPE_DEBUG, "own pos %u\n", c->own_pos);
1312   LOG (GNUNET_ERROR_TYPE_DEBUG, "putting connection %s to next peer %p\n",
1313        GMC_2s (c), next_peer);
1314   LOG (GNUNET_ERROR_TYPE_DEBUG, "next peer %p %s\n", next_peer, GMP_2s (next_peer));
1315   LOG (GNUNET_ERROR_TYPE_DEBUG, "putting connection %s to prev peer %p\n",
1316        GMC_2s (c), prev_peer);
1317   LOG (GNUNET_ERROR_TYPE_DEBUG, "prev peer %p %s\n", prev_peer, GMP_2s (prev_peer));
1318
1319   if (GNUNET_NO == GMP_is_neighbor (next_peer)
1320       || GNUNET_NO == GMP_is_neighbor (prev_peer))
1321   {
1322     if (GMC_is_origin (c, GNUNET_YES))
1323       GNUNET_STATISTICS_update (stats, "# local bad paths", 1, GNUNET_NO);
1324     GNUNET_STATISTICS_update (stats, "# bad paths", 1, GNUNET_NO);
1325
1326     LOG (GNUNET_ERROR_TYPE_DEBUG, "  register neighbors failed\n");
1327     LOG (GNUNET_ERROR_TYPE_DEBUG, "  prev: %s, neighbor?: %d\n",
1328          GMP_2s (prev_peer), GMP_is_neighbor (prev_peer));
1329     LOG (GNUNET_ERROR_TYPE_DEBUG, "  next: %s, neighbor?: %d\n",
1330          GMP_2s (next_peer), GMP_is_neighbor (next_peer));
1331     return GNUNET_SYSERR;
1332   }
1333
1334   GMP_add_connection (next_peer, c);
1335   GMP_add_connection (prev_peer, c);
1336
1337   return GNUNET_OK;
1338 }
1339
1340
1341 /**
1342  * Remove the connection from the list of both neighbors.
1343  *
1344  * @param c Connection.
1345  */
1346 static void
1347 unregister_neighbors (struct MeshConnection *c)
1348 {
1349   struct MeshPeer *peer;
1350
1351   peer = get_next_hop (c);
1352   if (GNUNET_OK != GMP_remove_connection (peer, c))
1353   {
1354     GNUNET_assert (MESH_CONNECTION_NEW == c->state
1355                   || MESH_CONNECTION_DESTROYED == c->state);
1356     LOG (GNUNET_ERROR_TYPE_DEBUG, "  cstate: %u\n", c->state);
1357     if (NULL != c->t) GMT_debug (c->t);
1358   }
1359
1360   peer = get_prev_hop (c);
1361   if (GNUNET_OK != GMP_remove_connection (peer, c))
1362   {
1363     GNUNET_assert (MESH_CONNECTION_NEW == c->state
1364                   || MESH_CONNECTION_DESTROYED == c->state);
1365     LOG (GNUNET_ERROR_TYPE_DEBUG, "  cstate: %u\n", c->state);
1366     if (NULL != c->t) GMT_debug (c->t);
1367   }
1368 }
1369
1370
1371 /**
1372  * Bind the connection to the peer and the tunnel to that peer.
1373  *
1374  * If the peer has no tunnel, create one. Update tunnel and connection
1375  * data structres to reflect new status.
1376  *
1377  * @param c Connection.
1378  * @param peer Peer.
1379  */
1380 static void
1381 add_to_peer (struct MeshConnection *c, struct MeshPeer *peer)
1382 {
1383   GMP_add_tunnel (peer);
1384   c->t = GMP_get_tunnel (peer);
1385   GMT_add_connection (c->t, c);
1386 }
1387
1388
1389 /**
1390  * Builds a path from a PeerIdentity array.
1391  *
1392  * @param peers PeerIdentity array.
1393  * @param size Size of the @c peers array.
1394  * @param own_pos Output parameter: own position in the path.
1395  *
1396  * @return Fixed and shortened path.
1397  */
1398 static struct MeshPeerPath *
1399 build_path_from_peer_ids (struct GNUNET_PeerIdentity *peers,
1400                           unsigned int size,
1401                           unsigned int *own_pos)
1402 {
1403   struct MeshPeerPath *path;
1404   GNUNET_PEER_Id shortid;
1405   unsigned int i;
1406   unsigned int j;
1407   unsigned int offset;
1408
1409   /* Create path */
1410   LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
1411   path = path_new (size);
1412   *own_pos = 0;
1413   offset = 0;
1414   for (i = 0; i < size; i++)
1415   {
1416     LOG (GNUNET_ERROR_TYPE_DEBUG, "  - %u: taking %s\n",
1417          i, GNUNET_i2s (&peers[i]));
1418     shortid = GNUNET_PEER_intern (&peers[i]);
1419
1420     /* Check for loops / duplicates */
1421     for (j = 0; j < i - offset; j++)
1422     {
1423       if (path->peers[j] == shortid)
1424       {
1425         LOG (GNUNET_ERROR_TYPE_DEBUG, "    already exists at pos %u\n", j);
1426         offset += i - j;
1427         LOG (GNUNET_ERROR_TYPE_DEBUG, "    offset now\n", offset);
1428         GNUNET_PEER_change_rc (shortid, -1);
1429       }
1430     }
1431     LOG (GNUNET_ERROR_TYPE_DEBUG, "    storing at %u\n", i - offset);
1432     path->peers[i - offset] = shortid;
1433     if (path->peers[i] == myid)
1434       *own_pos = i;
1435   }
1436   path->length -= offset;
1437
1438   if (path->peers[*own_pos] != myid)
1439   {
1440     /* create path: self not found in path through self */
1441     GNUNET_break_op (0);
1442     path_destroy (path);
1443     return NULL;
1444   }
1445
1446   return path;
1447 }
1448
1449 /******************************************************************************/
1450 /********************************    API    ***********************************/
1451 /******************************************************************************/
1452
1453 /**
1454  * Core handler for connection creation.
1455  *
1456  * @param cls Closure (unused).
1457  * @param peer Sender (neighbor).
1458  * @param message Message.
1459  *
1460  * @return GNUNET_OK to keep the connection open,
1461  *         GNUNET_SYSERR to close it (signal serious error)
1462  */
1463 int
1464 GMC_handle_create (void *cls, const struct GNUNET_PeerIdentity *peer,
1465                    const struct GNUNET_MessageHeader *message)
1466 {
1467   struct GNUNET_MESH_ConnectionCreate *msg;
1468   struct GNUNET_PeerIdentity *id;
1469   struct GNUNET_HashCode *cid;
1470   struct MeshPeerPath *path;
1471   struct MeshPeer *dest_peer;
1472   struct MeshPeer *orig_peer;
1473   struct MeshConnection *c;
1474   unsigned int own_pos;
1475   uint16_t size;
1476
1477   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1478   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection create msg\n");
1479
1480   /* Check size */
1481   size = ntohs (message->size);
1482   if (size < sizeof (struct GNUNET_MESH_ConnectionCreate))
1483   {
1484     GNUNET_break_op (0);
1485     return GNUNET_OK;
1486   }
1487
1488   /* Calculate hops */
1489   size -= sizeof (struct GNUNET_MESH_ConnectionCreate);
1490   if (size % sizeof (struct GNUNET_PeerIdentity))
1491   {
1492     GNUNET_break_op (0);
1493     return GNUNET_OK;
1494   }
1495   size /= sizeof (struct GNUNET_PeerIdentity);
1496   if (1 > size)
1497   {
1498     GNUNET_break_op (0);
1499     return GNUNET_OK;
1500   }
1501   LOG (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
1502
1503   /* Get parameters */
1504   msg = (struct GNUNET_MESH_ConnectionCreate *) message;
1505   cid = &msg->cid;
1506   id = (struct GNUNET_PeerIdentity *) &msg[1];
1507   LOG (GNUNET_ERROR_TYPE_DEBUG, "    connection %s (%s->).\n",
1508        GNUNET_h2s (cid), GNUNET_i2s (id));
1509
1510   /* Create connection */
1511   c = connection_get (cid);
1512   if (NULL == c)
1513   {
1514     path = build_path_from_peer_ids ((struct GNUNET_PeerIdentity *) &msg[1],
1515                                      size, &own_pos);
1516     if (NULL == path)
1517       return GNUNET_OK;
1518     if (0 == own_pos)
1519     {
1520       GNUNET_break_op (0);
1521       path_destroy (path);
1522       return GNUNET_OK;
1523     }
1524     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
1525     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating connection\n");
1526     c = GMC_new (cid, NULL, path_duplicate (path), own_pos);
1527     if (NULL == c)
1528     {
1529       if (path->length - 1 == own_pos)
1530       {
1531         /* If we are destination, why did the creation fail? */
1532         GNUNET_break (0);
1533         return GNUNET_OK;
1534       }
1535       send_broken2 (cid, &my_full_id,
1536                     GNUNET_PEER_resolve2 (path->peers[own_pos + 1]),
1537                     path->peers[own_pos - 1]);
1538       path_destroy (path);
1539       return GNUNET_OK;
1540     }
1541     GMP_add_path_to_all (path, GNUNET_NO);
1542     connection_reset_timeout (c, GNUNET_YES);
1543   }
1544   else
1545   {
1546     path = path_duplicate (c->path);
1547   }
1548   if (MESH_CONNECTION_NEW == c->state)
1549     connection_change_state (c, MESH_CONNECTION_SENT);
1550
1551   /* Remember peers */
1552   dest_peer = GMP_get (&id[size - 1]);
1553   orig_peer = GMP_get (&id[0]);
1554
1555   /* Is it a connection to us? */
1556   if (c->own_pos == size - 1)
1557   {
1558     LOG (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
1559     GMP_add_path_to_origin (orig_peer, path_duplicate (path), GNUNET_YES);
1560
1561     add_to_peer (c, orig_peer);
1562     if (MESH_TUNNEL3_NEW == GMT_get_cstate (c->t))
1563       GMT_change_cstate (c->t,  MESH_TUNNEL3_WAITING);
1564
1565     send_connection_ack (c, GNUNET_NO);
1566     if (MESH_CONNECTION_SENT == c->state)
1567       connection_change_state (c, MESH_CONNECTION_ACK);
1568   }
1569   else
1570   {
1571     /* It's for somebody else! Retransmit. */
1572     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
1573     GMP_add_path (dest_peer, path_duplicate (path), GNUNET_NO);
1574     GMP_add_path_to_origin (orig_peer, path_duplicate (path), GNUNET_NO);
1575     GMC_send_prebuilt_message (message, c, GNUNET_YES, GNUNET_YES,
1576                                NULL, NULL);
1577   }
1578   path_destroy (path);
1579   return GNUNET_OK;
1580 }
1581
1582
1583 /**
1584  * Core handler for path confirmations.
1585  *
1586  * @param cls closure
1587  * @param message message
1588  * @param peer peer identity this notification is about
1589  *
1590  * @return GNUNET_OK to keep the connection open,
1591  *         GNUNET_SYSERR to close it (signal serious error)
1592  */
1593 int
1594 GMC_handle_confirm (void *cls, const struct GNUNET_PeerIdentity *peer,
1595                     const struct GNUNET_MessageHeader *message)
1596 {
1597   struct GNUNET_MESH_ConnectionACK *msg;
1598   struct MeshConnection *c;
1599   struct MeshPeerPath *p;
1600   struct MeshPeer *pi;
1601   enum MeshConnectionState oldstate;
1602   int fwd;
1603
1604   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1605   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection ACK msg\n");
1606   msg = (struct GNUNET_MESH_ConnectionACK *) message;
1607   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n",
1608               GNUNET_h2s (&msg->cid));
1609   c = connection_get (&msg->cid);
1610   if (NULL == c)
1611   {
1612     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
1613                               1, GNUNET_NO);
1614     LOG (GNUNET_ERROR_TYPE_DEBUG, "  don't know the connection!\n");
1615     return GNUNET_OK;
1616   }
1617
1618   if (GNUNET_NO != c->destroy)
1619   {
1620     LOG (GNUNET_ERROR_TYPE_DEBUG, "  connection being destroyed\n");
1621     return GNUNET_OK;
1622   }
1623
1624   oldstate = c->state;
1625   LOG (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n", GNUNET_i2s (peer));
1626   pi = GMP_get (peer);
1627   if (get_next_hop (c) == pi)
1628   {
1629     LOG (GNUNET_ERROR_TYPE_DEBUG, "  SYNACK\n");
1630     fwd = GNUNET_NO;
1631     if (MESH_CONNECTION_SENT == oldstate)
1632       connection_change_state (c, MESH_CONNECTION_ACK);
1633   }
1634   else if (get_prev_hop (c) == pi)
1635   {
1636     LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK\n");
1637     fwd = GNUNET_YES;
1638     connection_change_state (c, MESH_CONNECTION_READY);
1639   }
1640   else
1641   {
1642     GNUNET_break_op (0);
1643     return GNUNET_OK;
1644   }
1645
1646   connection_reset_timeout (c, fwd);
1647
1648   /* Add path to peers? */
1649   p = c->path;
1650   if (NULL != p)
1651   {
1652     GMP_add_path_to_all (p, GNUNET_YES);
1653   }
1654   else
1655   {
1656     GNUNET_break (0);
1657   }
1658
1659   /* Message for us as creator? */
1660   if (GMC_is_origin (c, GNUNET_YES))
1661   {
1662     if (GNUNET_NO != fwd)
1663     {
1664       GNUNET_break_op (0);
1665       return GNUNET_OK;
1666     }
1667     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection (SYN)ACK for us!\n");
1668
1669     /* If just created, cancel the short timeout and start a long one */
1670     if (MESH_CONNECTION_SENT == oldstate)
1671       connection_reset_timeout (c, GNUNET_YES);
1672
1673     /* Change connection state */
1674     connection_change_state (c, MESH_CONNECTION_READY);
1675     send_connection_ack (c, GNUNET_YES);
1676
1677     /* Change tunnel state, trigger KX */
1678     if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
1679       GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
1680
1681     return GNUNET_OK;
1682   }
1683
1684   /* Message for us as destination? */
1685   if (GMC_is_terminal (c, GNUNET_YES))
1686   {
1687     if (GNUNET_YES != fwd)
1688     {
1689       GNUNET_break_op (0);
1690       return GNUNET_OK;
1691     }
1692     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection ACK for us!\n");
1693
1694     /* If just created, cancel the short timeout and start a long one */
1695     if (MESH_CONNECTION_ACK == oldstate)
1696       connection_reset_timeout (c, GNUNET_NO);
1697
1698     /* Change tunnel state */
1699     if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
1700       GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
1701
1702     return GNUNET_OK;
1703   }
1704
1705   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1706   GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1707   return GNUNET_OK;
1708 }
1709
1710
1711 /**
1712  * Core handler for notifications of broken paths
1713  *
1714  * @param cls Closure (unused).
1715  * @param id Peer identity of sending neighbor.
1716  * @param message Message.
1717  *
1718  * @return GNUNET_OK to keep the connection open,
1719  *         GNUNET_SYSERR to close it (signal serious error)
1720  */
1721 int
1722 GMC_handle_broken (void* cls,
1723                    const struct GNUNET_PeerIdentity* id,
1724                    const struct GNUNET_MessageHeader* message)
1725 {
1726   struct GNUNET_MESH_ConnectionBroken *msg;
1727   struct MeshConnection *c;
1728   int fwd;
1729
1730   LOG (GNUNET_ERROR_TYPE_DEBUG,
1731               "Received a CONNECTION BROKEN msg from %s\n", GNUNET_i2s (id));
1732   msg = (struct GNUNET_MESH_ConnectionBroken *) message;
1733   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1734               GNUNET_i2s (&msg->peer1));
1735   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1736               GNUNET_i2s (&msg->peer2));
1737   c = connection_get (&msg->cid);
1738   if (NULL == c)
1739   {
1740     GNUNET_break_op (0);
1741     return GNUNET_OK;
1742   }
1743
1744   fwd = is_fwd (c, id);
1745   if (GMC_is_terminal (c, fwd))
1746   {
1747     path_invalidate (c->path);
1748     if (0 < c->pending_messages)
1749       c->destroy = GNUNET_YES;
1750     else
1751       GMC_destroy (c);
1752   }
1753   else
1754   {
1755     GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1756     c->destroy = GNUNET_YES;
1757     connection_cancel_queues (c, !fwd);
1758   }
1759
1760   return GNUNET_OK;
1761
1762 }
1763
1764
1765 /**
1766  * Core handler for tunnel destruction
1767  *
1768  * @param cls Closure (unused).
1769  * @param peer Peer identity of sending neighbor.
1770  * @param message Message.
1771  *
1772  * @return GNUNET_OK to keep the connection open,
1773  *         GNUNET_SYSERR to close it (signal serious error)
1774  */
1775 int
1776 GMC_handle_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
1777                     const struct GNUNET_MessageHeader *message)
1778 {
1779   struct GNUNET_MESH_ConnectionDestroy *msg;
1780   struct MeshConnection *c;
1781   int fwd;
1782
1783   msg = (struct GNUNET_MESH_ConnectionDestroy *) message;
1784   LOG (GNUNET_ERROR_TYPE_DEBUG,
1785               "Got a CONNECTION DESTROY message from %s\n",
1786               GNUNET_i2s (peer));
1787   LOG (GNUNET_ERROR_TYPE_DEBUG,
1788               "  for connection %s\n",
1789               GNUNET_h2s (&msg->cid));
1790   c = connection_get (&msg->cid);
1791   if (NULL == c)
1792   {
1793     /* Probably already got the message from another path,
1794      * destroyed the tunnel and retransmitted to children.
1795      * Safe to ignore.
1796      */
1797     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
1798                               1, GNUNET_NO);
1799     LOG (GNUNET_ERROR_TYPE_DEBUG, "  connection unknown: already destroyed?\n");
1800     return GNUNET_OK;
1801   }
1802   fwd = is_fwd (c, peer);
1803   if (GNUNET_SYSERR == fwd)
1804   {
1805     GNUNET_break_op (0); /* FIXME */
1806     return GNUNET_OK;
1807   }
1808   if (GNUNET_NO == GMC_is_terminal (c, fwd))
1809     GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1810   else if (0 == c->pending_messages)
1811   {
1812     LOG (GNUNET_ERROR_TYPE_DEBUG, "!  directly destroying connection!\n");
1813     GMC_destroy (c);
1814     return GNUNET_OK;
1815   }
1816   c->destroy = GNUNET_YES;
1817   c->state = MESH_CONNECTION_DESTROYED;
1818   if (NULL != c->t)
1819   {
1820     GMT_remove_connection (c->t, c);
1821     c->t = NULL;
1822   }
1823
1824   return GNUNET_OK;
1825 }
1826
1827 /**
1828  * Generic handler for mesh network encrypted traffic.
1829  *
1830  * @param peer Peer identity this notification is about.
1831  * @param msg Encrypted message.
1832  *
1833  * @return GNUNET_OK to keep the connection open,
1834  *         GNUNET_SYSERR to close it (signal serious error)
1835  */
1836 static int
1837 handle_mesh_encrypted (const struct GNUNET_PeerIdentity *peer,
1838                        const struct GNUNET_MESH_Encrypted *msg)
1839 {
1840   struct MeshConnection *c;
1841   struct MeshPeer *neighbor;
1842   struct MeshFlowControl *fc;
1843   GNUNET_PEER_Id peer_id;
1844   uint32_t pid;
1845   uint32_t ttl;
1846   uint16_t type;
1847   size_t size;
1848   int fwd;
1849
1850   /* Check size */
1851   size = ntohs (msg->header.size);
1852   if (size <
1853       sizeof (struct GNUNET_MESH_Encrypted) +
1854       sizeof (struct GNUNET_MessageHeader))
1855   {
1856     GNUNET_break_op (0);
1857     return GNUNET_OK;
1858   }
1859   type = ntohs (msg->header.type);
1860   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1861   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message (#%u) from %s\n",
1862        GM_m2s (type), ntohl (msg->pid), GNUNET_i2s (peer));
1863
1864   /* Check connection */
1865   c = connection_get (&msg->cid);
1866   if (NULL == c)
1867   {
1868     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1869     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING enc on unknown connection %s\n",
1870          GNUNET_h2s (&msg->cid));
1871     return GNUNET_OK;
1872   }
1873
1874   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n", GMC_2s (c));
1875
1876   /* Check if origin is as expected */
1877   neighbor = get_prev_hop (c);
1878   peer_id = GNUNET_PEER_search (peer);
1879   if (peer_id == GMP_get_short_id (neighbor))
1880   {
1881     fwd = GNUNET_YES;
1882   }
1883   else
1884   {
1885     neighbor = get_next_hop (c);
1886     if (peer_id == GMP_get_short_id (neighbor))
1887     {
1888       fwd = GNUNET_NO;
1889     }
1890     else
1891     {
1892       /* Unexpected peer sending traffic on a connection. */
1893       GNUNET_break_op (0);
1894       return GNUNET_OK;
1895     }
1896   }
1897
1898   /* Check PID */
1899   fc = fwd ? &c->bck_fc : &c->fwd_fc;
1900   pid = ntohl (msg->pid);
1901   if (GM_is_pid_bigger (pid, fc->last_ack_sent))
1902   {
1903     GNUNET_STATISTICS_update (stats, "# unsolicited message", 1, GNUNET_NO);
1904     LOG (GNUNET_ERROR_TYPE_DEBUG,
1905                 "WARNING Received PID %u, (prev %u), ACK %u\n",
1906                 pid, fc->last_pid_recv, fc->last_ack_sent);
1907     return GNUNET_OK;
1908   }
1909   if (GNUNET_NO == GM_is_pid_bigger (pid, fc->last_pid_recv))
1910   {
1911     GNUNET_STATISTICS_update (stats, "# duplicate PID", 1, GNUNET_NO);
1912     LOG (GNUNET_ERROR_TYPE_DEBUG,
1913                 " Pid %u not expected (%u+), dropping!\n",
1914                 pid, fc->last_pid_recv + 1);
1915     return GNUNET_OK;
1916   }
1917   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
1918     connection_change_state (c, MESH_CONNECTION_READY);
1919   connection_reset_timeout (c, fwd);
1920   fc->last_pid_recv = pid;
1921
1922   /* Is this message for us? */
1923   if (GMC_is_terminal (c, fwd))
1924   {
1925     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
1926     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
1927
1928     if (NULL == c->t)
1929     {
1930       GNUNET_break (GNUNET_NO != c->destroy);
1931       return GNUNET_OK;
1932     }
1933     fc->last_pid_recv = pid;
1934     GMT_handle_encrypted (c->t, msg);
1935     GMC_send_ack (c, fwd, GNUNET_NO);
1936     return GNUNET_OK;
1937   }
1938
1939   /* Message not for us: forward to next hop */
1940   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1941   ttl = ntohl (msg->ttl);
1942   LOG (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
1943   if (ttl == 0)
1944   {
1945     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
1946     LOG (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
1947     GMC_send_ack (c, fwd, GNUNET_NO);
1948     return GNUNET_OK;
1949   }
1950
1951   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
1952   GMC_send_prebuilt_message (&msg->header, c, fwd, GNUNET_NO, NULL, NULL);
1953
1954   return GNUNET_OK;
1955 }
1956
1957 /**
1958  * Generic handler for mesh network encrypted traffic.
1959  *
1960  * @param peer Peer identity this notification is about.
1961  * @param msg Encrypted message.
1962  *
1963  * @return GNUNET_OK to keep the connection open,
1964  *         GNUNET_SYSERR to close it (signal serious error)
1965  */
1966 static int
1967 handle_mesh_kx (const struct GNUNET_PeerIdentity *peer,
1968                 const struct GNUNET_MESH_KX *msg)
1969 {
1970   struct MeshConnection *c;
1971   struct MeshPeer *neighbor;
1972   GNUNET_PEER_Id peer_id;
1973   size_t size;
1974   uint16_t type;
1975   int fwd;
1976
1977   /* Check size */
1978   size = ntohs (msg->header.size);
1979   if (size <
1980       sizeof (struct GNUNET_MESH_Encrypted) +
1981       sizeof (struct GNUNET_MessageHeader))
1982   {
1983     GNUNET_break_op (0);
1984     return GNUNET_OK;
1985   }
1986   type = ntohs (msg->header.type);
1987   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1988   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
1989        GM_m2s (type), GNUNET_i2s (peer));
1990
1991   /* Check connection */
1992   c = connection_get (&msg->cid);
1993   if (NULL == c)
1994   {
1995     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1996     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING kx on unknown connection %s\n",
1997          GNUNET_h2s (&msg->cid));
1998     return GNUNET_OK;
1999   }
2000   LOG (GNUNET_ERROR_TYPE_DEBUG, " on connection %s\n", GMC_2s (c));
2001
2002   /* Check if origin is as expected */
2003   neighbor = get_prev_hop (c);
2004   peer_id = GNUNET_PEER_search (peer);
2005   if (peer_id == GMP_get_short_id (neighbor))
2006   {
2007     fwd = GNUNET_YES;
2008   }
2009   else
2010   {
2011     neighbor = get_next_hop (c);
2012     if (peer_id == GMP_get_short_id (neighbor))
2013     {
2014       fwd = GNUNET_NO;
2015     }
2016     else
2017     {
2018       /* Unexpected peer sending traffic on a connection. */
2019       GNUNET_break_op (0);
2020       return GNUNET_OK;
2021     }
2022   }
2023
2024   /* Count as connection confirmation. */
2025   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
2026   {
2027     connection_change_state (c, MESH_CONNECTION_READY);
2028     if (NULL != c->t)
2029     {
2030       if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
2031         GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
2032     }
2033   }
2034   connection_reset_timeout (c, fwd);
2035
2036   /* Is this message for us? */
2037   if (GMC_is_terminal (c, fwd))
2038   {
2039     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
2040     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
2041     if (NULL == c->t)
2042     {
2043       GNUNET_break (0);
2044       return GNUNET_OK;
2045     }
2046     GMT_handle_kx (c->t, &msg[1].header);
2047     return GNUNET_OK;
2048   }
2049
2050   /* Message not for us: forward to next hop */
2051   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
2052   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
2053   GMC_send_prebuilt_message (&msg->header, c, fwd, GNUNET_NO, NULL, NULL);
2054
2055   return GNUNET_OK;
2056 }
2057
2058
2059 /**
2060  * Core handler for encrypted mesh network traffic (channel mgmt, data).
2061  *
2062  * @param cls Closure (unused).
2063  * @param message Message received.
2064  * @param peer Peer who sent the message.
2065  *
2066  * @return GNUNET_OK to keep the connection open,
2067  *         GNUNET_SYSERR to close it (signal serious error)
2068  */
2069 int
2070 GMC_handle_encrypted (void *cls, const struct GNUNET_PeerIdentity *peer,
2071                       const struct GNUNET_MessageHeader *message)
2072 {
2073   return handle_mesh_encrypted (peer,
2074                                 (struct GNUNET_MESH_Encrypted *)message);
2075 }
2076
2077
2078 /**
2079  * Core handler for key exchange traffic (ephemeral key, ping, pong).
2080  *
2081  * @param cls Closure (unused).
2082  * @param message Message received.
2083  * @param peer Peer who sent the message.
2084  *
2085  * @return GNUNET_OK to keep the connection open,
2086  *         GNUNET_SYSERR to close it (signal serious error)
2087  */
2088 int
2089 GMC_handle_kx (void *cls, const struct GNUNET_PeerIdentity *peer,
2090                const struct GNUNET_MessageHeader *message)
2091 {
2092   return handle_mesh_kx (peer,
2093                          (struct GNUNET_MESH_KX *) message);
2094 }
2095
2096
2097 /**
2098  * Core handler for mesh network traffic point-to-point acks.
2099  *
2100  * @param cls closure
2101  * @param message message
2102  * @param peer peer identity this notification is about
2103  *
2104  * @return GNUNET_OK to keep the connection open,
2105  *         GNUNET_SYSERR to close it (signal serious error)
2106  */
2107 int
2108 GMC_handle_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
2109                 const struct GNUNET_MessageHeader *message)
2110 {
2111   struct GNUNET_MESH_ACK *msg;
2112   struct MeshConnection *c;
2113   struct MeshFlowControl *fc;
2114   GNUNET_PEER_Id id;
2115   uint32_t ack;
2116   int fwd;
2117
2118   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
2119   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
2120               GNUNET_i2s (peer));
2121   msg = (struct GNUNET_MESH_ACK *) message;
2122
2123   c = connection_get (&msg->cid);
2124
2125   if (NULL == c)
2126   {
2127     GNUNET_STATISTICS_update (stats, "# ack on unknown connection", 1,
2128                               GNUNET_NO);
2129     return GNUNET_OK;
2130   }
2131
2132   /* Is this a forward or backward ACK? */
2133   id = GNUNET_PEER_search (peer);
2134   if (GMP_get_short_id (get_next_hop (c)) == id)
2135   {
2136     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
2137     fc = &c->fwd_fc;
2138     fwd = GNUNET_YES;
2139   }
2140   else if (GMP_get_short_id (get_prev_hop (c)) == id)
2141   {
2142     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
2143     fc = &c->bck_fc;
2144     fwd = GNUNET_NO;
2145   }
2146   else
2147   {
2148     GNUNET_break_op (0);
2149     return GNUNET_OK;
2150   }
2151
2152   ack = ntohl (msg->ack);
2153   LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u (was %u)\n",
2154               ack, fc->last_ack_recv);
2155   if (GM_is_pid_bigger (ack, fc->last_ack_recv))
2156     fc->last_ack_recv = ack;
2157
2158   /* Cancel polling if the ACK is big enough. */
2159   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task &&
2160       GM_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2161   {
2162     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Cancel poll\n");
2163     GNUNET_SCHEDULER_cancel (fc->poll_task);
2164     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2165     fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
2166   }
2167
2168   connection_unlock_queue (c, fwd);
2169
2170   return GNUNET_OK;
2171 }
2172
2173
2174 /**
2175  * Core handler for mesh network traffic point-to-point ack polls.
2176  *
2177  * @param cls closure
2178  * @param message message
2179  * @param peer peer identity this notification is about
2180  *
2181  * @return GNUNET_OK to keep the connection open,
2182  *         GNUNET_SYSERR to close it (signal serious error)
2183  */
2184 int
2185 GMC_handle_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
2186                  const struct GNUNET_MessageHeader *message)
2187 {
2188   struct GNUNET_MESH_Poll *msg;
2189   struct MeshConnection *c;
2190   struct MeshFlowControl *fc;
2191   GNUNET_PEER_Id id;
2192   uint32_t pid;
2193   int fwd;
2194
2195   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
2196   LOG (GNUNET_ERROR_TYPE_DEBUG,
2197        "Got a POLL message from %s!\n",
2198        GNUNET_i2s (peer));
2199
2200   msg = (struct GNUNET_MESH_Poll *) message;
2201
2202   c = connection_get (&msg->cid);
2203
2204   if (NULL == c)
2205   {
2206     GNUNET_STATISTICS_update (stats, "# poll on unknown connection", 1,
2207                               GNUNET_NO);
2208     LOG (GNUNET_ERROR_TYPE_DEBUG,
2209          "WARNING POLL message on unknown connection %s!\n",
2210          GNUNET_h2s (&msg->cid));
2211     return GNUNET_OK;
2212   }
2213
2214   /* Is this a forward or backward ACK?
2215    * Note: a poll should never be needed in a loopback case,
2216    * since there is no possiblility of packet loss there, so
2217    * this way of discerining FWD/BCK should not be a problem.
2218    */
2219   id = GNUNET_PEER_search (peer);
2220   if (GMP_get_short_id (get_next_hop (c)) == id)
2221   {
2222     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
2223     fc = &c->fwd_fc;
2224   }
2225   else if (GMP_get_short_id (get_prev_hop (c)) == id)
2226   {
2227     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
2228     fc = &c->bck_fc;
2229   }
2230   else
2231   {
2232     GNUNET_break_op (0);
2233     return GNUNET_OK;
2234   }
2235
2236   pid = ntohl (msg->pid);
2237   LOG (GNUNET_ERROR_TYPE_DEBUG, "  PID %u, OLD %u\n", pid, fc->last_pid_recv);
2238   fc->last_pid_recv = pid;
2239   fwd = fc == &c->bck_fc;
2240   GMC_send_ack (c, fwd, GNUNET_YES);
2241
2242   return GNUNET_OK;
2243 }
2244
2245
2246 /**
2247  * Core handler for mesh keepalives.
2248  *
2249  * @param cls closure
2250  * @param message message
2251  * @param peer peer identity this notification is about
2252  * @return GNUNET_OK to keep the connection open,
2253  *         GNUNET_SYSERR to close it (signal serious error)
2254  *
2255  * TODO: Check who we got this from, to validate route.
2256  */
2257 int
2258 GMC_handle_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
2259                       const struct GNUNET_MessageHeader *message)
2260 {
2261   struct GNUNET_MESH_ConnectionKeepAlive *msg;
2262   struct MeshConnection *c;
2263   struct MeshPeer *neighbor;
2264   GNUNET_PEER_Id peer_id;
2265   int fwd;
2266
2267   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) message;
2268   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
2269               GNUNET_i2s (peer));
2270
2271   c = connection_get (&msg->cid);
2272   if (NULL == c)
2273   {
2274     GNUNET_STATISTICS_update (stats, "# keepalive on unknown connection", 1,
2275                               GNUNET_NO);
2276     return GNUNET_OK;
2277   }
2278
2279   /* Check if origin is as expected TODO refactor and reuse */
2280   peer_id = GNUNET_PEER_search (peer);
2281   neighbor = get_prev_hop (c);
2282   if (peer_id == GMP_get_short_id (neighbor))
2283   {
2284     fwd = GNUNET_YES;
2285   }
2286   else
2287   {
2288     neighbor = get_next_hop (c);
2289     if (peer_id == GMP_get_short_id (neighbor))
2290     {
2291       fwd = GNUNET_NO;
2292     }
2293     else
2294     {
2295       GNUNET_break_op (0);
2296       return GNUNET_OK;
2297     }
2298   }
2299
2300   connection_change_state (c, MESH_CONNECTION_READY);
2301   connection_reset_timeout (c, fwd);
2302
2303   if (GMC_is_terminal (c, fwd))
2304     return GNUNET_OK;
2305
2306   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
2307   GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
2308
2309   return GNUNET_OK;
2310 }
2311
2312
2313 /**
2314  * Send an ACK on the appropriate connection/channel, depending on
2315  * the direction and the position of the peer.
2316  *
2317  * @param c Which connection to send the hop-by-hop ACK.
2318  * @param fwd Is this a fwd ACK? (will go dest->root).
2319  * @param force Send the ACK even if suboptimal (e.g. requested by POLL).
2320  */
2321 void
2322 GMC_send_ack (struct MeshConnection *c, int fwd, int force)
2323 {
2324   unsigned int buffer;
2325
2326   LOG (GNUNET_ERROR_TYPE_DEBUG,
2327        "GMC send %s ACK on %s\n",
2328        GM_f2s (fwd), GMC_2s (c));
2329
2330   if (NULL == c)
2331   {
2332     GNUNET_break (0);
2333     return;
2334   }
2335
2336   if (GNUNET_NO != c->destroy)
2337   {
2338     LOG (GNUNET_ERROR_TYPE_DEBUG, "  being destroyed, why bother...\n");
2339     return;
2340   }
2341
2342   /* Get available buffer space */
2343   if (GMC_is_terminal (c, fwd))
2344   {
2345     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from all channels\n");
2346     buffer = GMT_get_channels_buffer (c->t);
2347   }
2348   else
2349   {
2350     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from one connection\n");
2351     buffer = GMC_get_buffer (c, fwd);
2352   }
2353   LOG (GNUNET_ERROR_TYPE_DEBUG, "  buffer available: %u\n", buffer);
2354   if (0 == buffer && GNUNET_NO == force)
2355     return;
2356
2357   /* Send available buffer space */
2358   if (GMC_is_origin (c, fwd))
2359   {
2360     GNUNET_assert (NULL != c->t);
2361     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on channels...\n");
2362     GMT_unchoke_channels (c->t);
2363   }
2364   else
2365   {
2366     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on connection\n");
2367     send_ack (c, buffer, fwd, force);
2368   }
2369 }
2370
2371
2372 /**
2373  * Initialize the connections subsystem
2374  *
2375  * @param c Configuration handle.
2376  */
2377 void
2378 GMC_init (const struct GNUNET_CONFIGURATION_Handle *c)
2379 {
2380   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
2381   if (GNUNET_OK !=
2382       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
2383                                              &max_msgs_queue))
2384   {
2385     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2386                                "MESH", "MAX_MSGS_QUEUE", "MISSING");
2387     GNUNET_SCHEDULER_shutdown ();
2388     return;
2389   }
2390
2391   if (GNUNET_OK !=
2392       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_CONNECTIONS",
2393                                              &max_connections))
2394   {
2395     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2396                                "MESH", "MAX_CONNECTIONS", "MISSING");
2397     GNUNET_SCHEDULER_shutdown ();
2398     return;
2399   }
2400
2401   if (GNUNET_OK !=
2402       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_CONNECTION_TIME",
2403                                            &refresh_connection_time))
2404   {
2405     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2406                                "MESH", "REFRESH_CONNECTION_TIME", "MISSING");
2407     GNUNET_SCHEDULER_shutdown ();
2408     return;
2409   }
2410   create_connection_time = GNUNET_TIME_UNIT_SECONDS;
2411   connections = GNUNET_CONTAINER_multihashmap_create (1024, GNUNET_YES);
2412 }
2413
2414
2415 /**
2416  * Destroy each connection on shutdown.
2417  *
2418  * @param cls Closure (unused).
2419  * @param key Current key code (CID, unused).
2420  * @param value Value in the hash map (connection)
2421  *
2422  * @return #GNUNET_YES, because we should continue to iterate,
2423  */
2424 static int
2425 shutdown_iterator (void *cls,
2426                    const struct GNUNET_HashCode *key,
2427                    void *value)
2428 {
2429   struct MeshConnection *c = value;
2430
2431   GMC_destroy (c);
2432   return GNUNET_YES;
2433 }
2434
2435
2436 /**
2437  * Shut down the connections subsystem.
2438  */
2439 void
2440 GMC_shutdown (void)
2441 {
2442   GNUNET_CONTAINER_multihashmap_iterate (connections, &shutdown_iterator, NULL);
2443   GNUNET_CONTAINER_multihashmap_destroy (connections);
2444   connections = NULL;
2445 }
2446
2447
2448 struct MeshConnection *
2449 GMC_new (const struct GNUNET_HashCode *cid,
2450          struct MeshTunnel3 *t,
2451          struct MeshPeerPath *p,
2452          unsigned int own_pos)
2453 {
2454   struct MeshConnection *c;
2455
2456   c = GNUNET_new (struct MeshConnection);
2457   c->id = *cid;
2458   GNUNET_assert (GNUNET_OK ==
2459                  GNUNET_CONTAINER_multihashmap_put (connections,
2460                                                     &c->id, c,
2461                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
2462   fc_init (&c->fwd_fc);
2463   fc_init (&c->bck_fc);
2464   c->fwd_fc.c = c;
2465   c->bck_fc.c = c;
2466
2467   c->t = t;
2468   GNUNET_assert (own_pos <= p->length - 1);
2469   c->own_pos = own_pos;
2470   c->path = p;
2471
2472   if (GNUNET_OK != register_neighbors (c))
2473   {
2474     if (0 == own_pos)
2475     {
2476       path_invalidate (c->path);
2477       c->t = NULL;
2478       c->path = NULL;
2479     }
2480     GMC_destroy (c);
2481     return NULL;
2482   }
2483
2484   return c;
2485 }
2486
2487
2488 void
2489 GMC_destroy (struct MeshConnection *c)
2490 {
2491   if (NULL == c)
2492   {
2493     GNUNET_break (0);
2494     return;
2495   }
2496
2497   if (2 == c->destroy) /* cancel queues -> GMP_queue_cancel -> q_destroy -> */
2498     return;            /* -> message_sent -> GMC_destroy. Don't loop. */
2499   c->destroy = 2;
2500
2501   LOG (GNUNET_ERROR_TYPE_DEBUG, "destroying connection %s\n", GMC_2s (c));
2502   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc's f: %p, b: %p\n",
2503        &c->fwd_fc, &c->bck_fc);
2504   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc tasks f: %u, b: %u\n",
2505        c->fwd_fc.poll_task, c->bck_fc.poll_task);
2506
2507   /* Cancel all traffic */
2508   if (NULL != c->path)
2509   {
2510     connection_cancel_queues (c, GNUNET_YES);
2511     connection_cancel_queues (c, GNUNET_NO);
2512     unregister_neighbors (c);
2513   }
2514
2515   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc tasks f: %u, b: %u\n",
2516        c->fwd_fc.poll_task, c->bck_fc.poll_task);
2517
2518   /* Cancel maintainance task (keepalive/timeout) */
2519   if (NULL != c->fwd_fc.poll_msg)
2520   {
2521     GMC_cancel (c->fwd_fc.poll_msg);
2522     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL msg FWD canceled\n");
2523   }
2524   if (NULL != c->bck_fc.poll_msg)
2525   {
2526     GMC_cancel (c->bck_fc.poll_msg);
2527     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL msg BCK canceled\n");
2528   }
2529
2530   /* Delete from tunnel */
2531   if (NULL != c->t)
2532     GMT_remove_connection (c->t, c);
2533
2534   if (GNUNET_NO == GMC_is_origin (c, GNUNET_YES) && NULL != c->path)
2535     path_destroy (c->path);
2536   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_maintenance_task)
2537     GNUNET_SCHEDULER_cancel (c->fwd_maintenance_task);
2538   if (GNUNET_SCHEDULER_NO_TASK != c->bck_maintenance_task)
2539     GNUNET_SCHEDULER_cancel (c->bck_maintenance_task);
2540   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_fc.poll_task)
2541   {
2542     GNUNET_SCHEDULER_cancel (c->fwd_fc.poll_task);
2543     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL FWD canceled\n");
2544   }
2545   if (GNUNET_SCHEDULER_NO_TASK != c->bck_fc.poll_task)
2546   {
2547     GNUNET_SCHEDULER_cancel (c->bck_fc.poll_task);
2548     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL BCK canceled\n");
2549   }
2550
2551   GNUNET_break (GNUNET_YES ==
2552                 GNUNET_CONTAINER_multihashmap_remove (connections, &c->id, c));
2553
2554   GNUNET_STATISTICS_update (stats, "# connections", -1, GNUNET_NO);
2555   GNUNET_free (c);
2556 }
2557
2558 /**
2559  * Get the connection ID.
2560  *
2561  * @param c Connection to get the ID from.
2562  *
2563  * @return ID of the connection.
2564  */
2565 const struct GNUNET_HashCode *
2566 GMC_get_id (const struct MeshConnection *c)
2567 {
2568   return &c->id;
2569 }
2570
2571
2572 /**
2573  * Get the connection path.
2574  *
2575  * @param c Connection to get the path from.
2576  *
2577  * @return path used by the connection.
2578  */
2579 const struct MeshPeerPath *
2580 GMC_get_path (const struct MeshConnection *c)
2581 {
2582   if (GNUNET_NO == c->destroy)
2583     return c->path;
2584   return NULL;
2585 }
2586
2587
2588 /**
2589  * Get the connection state.
2590  *
2591  * @param c Connection to get the state from.
2592  *
2593  * @return state of the connection.
2594  */
2595 enum MeshConnectionState
2596 GMC_get_state (const struct MeshConnection *c)
2597 {
2598   return c->state;
2599 }
2600
2601 /**
2602  * Get the connection tunnel.
2603  *
2604  * @param c Connection to get the tunnel from.
2605  *
2606  * @return tunnel of the connection.
2607  */
2608 struct MeshTunnel3 *
2609 GMC_get_tunnel (const struct MeshConnection *c)
2610 {
2611   return c->t;
2612 }
2613
2614
2615 /**
2616  * Get free buffer space in a connection.
2617  *
2618  * @param c Connection.
2619  * @param fwd Is query about FWD traffic?
2620  *
2621  * @return Free buffer space [0 - max_msgs_queue/max_connections]
2622  */
2623 unsigned int
2624 GMC_get_buffer (struct MeshConnection *c, int fwd)
2625 {
2626   struct MeshFlowControl *fc;
2627
2628   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2629
2630   return (fc->queue_max - fc->queue_n);
2631 }
2632
2633 /**
2634  * Get how many messages have we allowed to send to us from a direction.
2635  *
2636  * @param c Connection.
2637  * @param fwd Are we asking about traffic from FWD (BCK messages)?
2638  *
2639  * @return last_ack_sent - last_pid_recv
2640  */
2641 unsigned int
2642 GMC_get_allowed (struct MeshConnection *c, int fwd)
2643 {
2644   struct MeshFlowControl *fc;
2645
2646   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2647   if (GM_is_pid_bigger(fc->last_pid_recv, fc->last_ack_sent))
2648   {
2649     return 0;
2650   }
2651   return (fc->last_ack_sent - fc->last_pid_recv);
2652 }
2653
2654 /**
2655  * Get messages queued in a connection.
2656  *
2657  * @param c Connection.
2658  * @param fwd Is query about FWD traffic?
2659  *
2660  * @return Number of messages queued.
2661  */
2662 unsigned int
2663 GMC_get_qn (struct MeshConnection *c, int fwd)
2664 {
2665   struct MeshFlowControl *fc;
2666
2667   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2668
2669   return fc->queue_n;
2670 }
2671
2672
2673 /**
2674  * Allow the connection to advertise a buffer of the given size.
2675  *
2676  * The connection will send an @c fwd ACK message (so: in direction !fwd)
2677  * allowing up to last_pid_recv + buffer.
2678  *
2679  * @param c Connection.
2680  * @param buffer How many more messages the connection can accept.
2681  * @param fwd Is this about FWD traffic? (The ack will go dest->root).
2682  */
2683 void
2684 GMC_allow (struct MeshConnection *c, unsigned int buffer, int fwd)
2685 {
2686   LOG (GNUNET_ERROR_TYPE_DEBUG, "  allowing %s %u messages %s\n",
2687        GMC_2s (c), buffer, GM_f2s (fwd));
2688   send_ack (c, buffer, fwd, GNUNET_NO);
2689 }
2690
2691
2692 /**
2693  * Notify other peers on a connection of a broken link. Mark connections
2694  * to destroy after all traffic has been sent.
2695  *
2696  * @param c Connection on which there has been a disconnection.
2697  * @param peer Peer that disconnected.
2698  */
2699 void
2700 GMC_notify_broken (struct MeshConnection *c,
2701                    struct MeshPeer *peer)
2702 {
2703   int fwd;
2704
2705   LOG (GNUNET_ERROR_TYPE_DEBUG,
2706        " notify broken on %s due to %s disconnect\n",
2707        GMC_2s (c), GMP_2s (peer));
2708
2709   fwd = peer == get_prev_hop (c);
2710
2711   if (GNUNET_YES == GMC_is_terminal (c, fwd))
2712   {
2713     /* Local shutdown, no one to notify about this. */
2714     GMC_destroy (c);
2715     return;
2716   }
2717   if (GNUNET_NO == c->destroy)
2718     send_broken (c, &my_full_id, GMP_get_id (peer), fwd);
2719
2720   /* Connection will have at least one pending message
2721    * (the one we just scheduled), so no point in checking whether to
2722    * destroy immediately. */
2723   c->destroy = GNUNET_YES;
2724   c->state = MESH_CONNECTION_DESTROYED;
2725
2726   /**
2727    * Cancel all queues, if no message is left, connection will be destroyed.
2728    */
2729   connection_cancel_queues (c, !fwd);
2730
2731   return;
2732 }
2733
2734
2735 /**
2736  * Is this peer the first one on the connection?
2737  *
2738  * @param c Connection.
2739  * @param fwd Is this about fwd traffic?
2740  *
2741  * @return #GNUNET_YES if origin, #GNUNET_NO if relay/terminal.
2742  */
2743 int
2744 GMC_is_origin (struct MeshConnection *c, int fwd)
2745 {
2746   if (!fwd && c->path->length - 1 == c->own_pos )
2747     return GNUNET_YES;
2748   if (fwd && 0 == c->own_pos)
2749     return GNUNET_YES;
2750   return GNUNET_NO;
2751 }
2752
2753
2754 /**
2755  * Is this peer the last one on the connection?
2756  *
2757  * @param c Connection.
2758  * @param fwd Is this about fwd traffic?
2759  *            Note that the ROOT is the terminal for BCK traffic!
2760  *
2761  * @return #GNUNET_YES if terminal, #GNUNET_NO if relay/origin.
2762  */
2763 int
2764 GMC_is_terminal (struct MeshConnection *c, int fwd)
2765 {
2766   return GMC_is_origin (c, !fwd);
2767 }
2768
2769
2770 /**
2771  * See if we are allowed to send by the next hop in the given direction.
2772  *
2773  * @param c Connection.
2774  * @param fwd Is this about fwd traffic?
2775  *
2776  * @return #GNUNET_YES in case it's OK to send.
2777  */
2778 int
2779 GMC_is_sendable (struct MeshConnection *c, int fwd)
2780 {
2781   struct MeshFlowControl *fc;
2782
2783   if (NULL == c)
2784   {
2785     GNUNET_break (0);
2786     return GNUNET_YES;
2787   }
2788   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2789   if (GM_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2790     return GNUNET_YES;
2791   return GNUNET_NO;
2792 }
2793
2794 /**
2795  * Sends an already built message on a connection, properly registering
2796  * all used resources.
2797  *
2798  * @param message Message to send. Function makes a copy of it.
2799  *                If message is not hop-by-hop, decrements TTL of copy.
2800  * @param c Connection on which this message is transmitted.
2801  * @param fwd Is this a fwd message?
2802  * @param force Force the connection to accept the message (buffer overfill).
2803  * @param cont Continuation called once message is sent. Can be NULL.
2804  * @param cont_cls Closure for @c cont.
2805  *
2806  * @return Handle to cancel the message before it's sent.
2807  *         NULL on error or if @c cont is NULL.
2808  *         Invalid on @c cont call.
2809  */
2810 struct MeshConnectionQueue *
2811 GMC_send_prebuilt_message (const struct GNUNET_MessageHeader *message,
2812                            struct MeshConnection *c, int fwd, int force,
2813                            GMC_sent cont, void *cont_cls)
2814 {
2815   struct MeshFlowControl *fc;
2816   struct MeshConnectionQueue *q;
2817   void *data;
2818   size_t size;
2819   uint16_t type;
2820   int droppable;
2821
2822   size = ntohs (message->size);
2823   data = GNUNET_malloc (size);
2824   memcpy (data, message, size);
2825   type = ntohs (message->type);
2826   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send %s (%u bytes) on connection %s\n",
2827        GM_m2s (type), size, GMC_2s (c));
2828
2829   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2830   droppable = GNUNET_NO == force;
2831   switch (type)
2832   {
2833     struct GNUNET_MESH_Encrypted *emsg;
2834     struct GNUNET_MESH_KX        *kmsg;
2835     struct GNUNET_MESH_ACK       *amsg;
2836     struct GNUNET_MESH_Poll      *pmsg;
2837     struct GNUNET_MESH_ConnectionDestroy *dmsg;
2838     struct GNUNET_MESH_ConnectionBroken  *bmsg;
2839     uint32_t ttl;
2840
2841     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
2842       emsg = (struct GNUNET_MESH_Encrypted *) data;
2843       ttl = ntohl (emsg->ttl);
2844       if (0 == ttl)
2845       {
2846         GNUNET_break_op (0);
2847         GNUNET_free (data);
2848         return NULL;
2849       }
2850       emsg->cid = c->id;
2851       emsg->ttl = htonl (ttl - 1);
2852       emsg->pid = htonl (fc->next_pid++);
2853       LOG (GNUNET_ERROR_TYPE_DEBUG, "  Q_N+ %p %u\n", fc, fc->queue_n);
2854       if (GNUNET_YES == droppable)
2855       {
2856         fc->queue_n++;
2857         LOG (GNUNET_ERROR_TYPE_DEBUG, "pid %u\n", ntohl (emsg->pid));
2858         LOG (GNUNET_ERROR_TYPE_DEBUG, "last pid sent %u\n", fc->last_pid_sent);
2859         LOG (GNUNET_ERROR_TYPE_DEBUG, "     ack recv %u\n", fc->last_ack_recv);
2860       }
2861       else
2862       {
2863         LOG (GNUNET_ERROR_TYPE_DEBUG, "  not droppable, Q_N stays the same\n");
2864       }
2865       if (GM_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv))
2866       {
2867         GMC_start_poll (c, fwd);
2868       }
2869       break;
2870
2871     case GNUNET_MESSAGE_TYPE_MESH_KX:
2872       kmsg = (struct GNUNET_MESH_KX *) data;
2873       kmsg->cid = c->id;
2874       break;
2875
2876     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2877       amsg = (struct GNUNET_MESH_ACK *) data;
2878       amsg->cid = c->id;
2879       LOG (GNUNET_ERROR_TYPE_DEBUG, " ack %u\n", ntohl (amsg->ack));
2880       droppable = GNUNET_NO;
2881       break;
2882
2883     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2884       pmsg = (struct GNUNET_MESH_Poll *) data;
2885       pmsg->cid = c->id;
2886       LOG (GNUNET_ERROR_TYPE_DEBUG, " poll %u\n", ntohl (pmsg->pid));
2887       droppable = GNUNET_NO;
2888       break;
2889
2890     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
2891       dmsg = (struct GNUNET_MESH_ConnectionDestroy *) data;
2892       dmsg->cid = c->id;
2893       dmsg->reserved = 0;
2894       break;
2895
2896     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
2897       bmsg = (struct GNUNET_MESH_ConnectionBroken *) data;
2898       bmsg->cid = c->id;
2899       bmsg->reserved = 0;
2900       break;
2901
2902     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
2903     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
2904     case GNUNET_MESSAGE_TYPE_MESH_KEEPALIVE:
2905       break;
2906
2907     default:
2908       GNUNET_break (0);
2909       GNUNET_free (data);
2910       return NULL;
2911   }
2912
2913   if (fc->queue_n > fc->queue_max && droppable)
2914   {
2915     GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)",
2916                               1, GNUNET_NO);
2917     GNUNET_break (0);
2918     LOG (GNUNET_ERROR_TYPE_DEBUG,
2919                 "queue full: %u/%u\n",
2920                 fc->queue_n, fc->queue_max);
2921     if (GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED == type)
2922     {
2923       fc->queue_n--;
2924       fc->next_pid--;
2925     }
2926     GNUNET_free (data);
2927     return NULL; /* Drop this message */
2928   }
2929
2930   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u\n", c, c->pending_messages);
2931   c->pending_messages++;
2932
2933   q = GNUNET_new (struct MeshConnectionQueue);
2934   q->forced = !droppable;
2935   q->q = GMP_queue_add (get_hop (c, fwd), data, type, size, c, fwd,
2936                         &message_sent, q);
2937   if (NULL == q->q)
2938   {
2939     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING dropping msg on %s\n", GMC_2s (c));
2940     GNUNET_free (data);
2941     GNUNET_free (q);
2942     return NULL;
2943   }
2944   q->cont = cont;
2945   q->cont_cls = cont_cls;
2946   return q;
2947 }
2948
2949
2950 /**
2951  * Cancel a previously sent message while it's in the queue.
2952  *
2953  * ONLY can be called before the continuation given to the send function
2954  * is called. Once the continuation is called, the message is no longer in the
2955  * queue.
2956  *
2957  * @param q Handle to the queue.
2958  */
2959 void
2960 GMC_cancel (struct MeshConnectionQueue *q)
2961 {
2962   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  GMC cancel message\n");
2963
2964   /* queue destroy calls message_sent, which calls q->cont and frees q */
2965   GMP_queue_destroy (q->q, GNUNET_YES);
2966 }
2967
2968
2969 /**
2970  * Sends a CREATE CONNECTION message for a path to a peer.
2971  * Changes the connection and tunnel states if necessary.
2972  *
2973  * @param connection Connection to create.
2974  */
2975 void
2976 GMC_send_create (struct MeshConnection *connection)
2977 {
2978   enum MeshTunnel3CState state;
2979   size_t size;
2980
2981   size = sizeof (struct GNUNET_MESH_ConnectionCreate);
2982   size += connection->path->length * sizeof (struct GNUNET_PeerIdentity);
2983
2984   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send connection create\n");
2985   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u (create)\n",
2986        connection, connection->pending_messages);
2987   connection->pending_messages++;
2988
2989   connection->maintenance_q =
2990     GMP_queue_add (get_next_hop (connection), NULL,
2991                    GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
2992                    size, connection, GNUNET_YES, &message_sent, NULL);
2993
2994   state = GMT_get_cstate (connection->t);
2995   if (MESH_TUNNEL3_SEARCHING == state || MESH_TUNNEL3_NEW == state)
2996     GMT_change_cstate (connection->t, MESH_TUNNEL3_WAITING);
2997   if (MESH_CONNECTION_NEW == connection->state)
2998     connection_change_state (connection, MESH_CONNECTION_SENT);
2999 }
3000
3001
3002 /**
3003  * Send a message to all peers in this connection that the connection
3004  * is no longer valid.
3005  *
3006  * If some peer should not receive the message, it should be zero'ed out
3007  * before calling this function.
3008  *
3009  * @param c The connection whose peers to notify.
3010  */
3011 void
3012 GMC_send_destroy (struct MeshConnection *c)
3013 {
3014   struct GNUNET_MESH_ConnectionDestroy msg;
3015
3016   if (GNUNET_YES == c->destroy)
3017     return;
3018
3019   msg.header.size = htons (sizeof (msg));
3020   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY);;
3021   msg.cid = c->id;
3022   LOG (GNUNET_ERROR_TYPE_DEBUG,
3023               "  sending connection destroy for connection %s\n",
3024               GMC_2s (c));
3025
3026   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_YES))
3027     GMC_send_prebuilt_message (&msg.header, c,
3028                                GNUNET_YES, GNUNET_YES, NULL, NULL);
3029   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_NO))
3030     GMC_send_prebuilt_message (&msg.header, c,
3031                                GNUNET_NO, GNUNET_YES, NULL, NULL);
3032   c->destroy = GNUNET_YES;
3033   c->state = MESH_CONNECTION_DESTROYED;
3034 }
3035
3036
3037 /**
3038  * @brief Start a polling timer for the connection.
3039  *
3040  * When a neighbor does not accept more traffic on the connection it could be
3041  * caused by a simple congestion or by a lost ACK. Polling enables to check
3042  * for the lastest ACK status for a connection.
3043  *
3044  * @param c Connection.
3045  * @param fwd Should we poll in the FWD direction?
3046  */
3047 void
3048 GMC_start_poll (struct MeshConnection *c, int fwd)
3049 {
3050   struct MeshFlowControl *fc;
3051
3052   fc = fwd ? &c->fwd_fc : &c->bck_fc;
3053   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL %s requested\n",
3054        GM_f2s (fwd));
3055   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task || NULL != fc->poll_msg)
3056   {
3057     LOG (GNUNET_ERROR_TYPE_DEBUG, " ***   not needed (%u, %p)\n",
3058          fc->poll_task, fc->poll_msg);
3059     return;
3060   }
3061   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL started on request\n");
3062   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
3063                                                 &connection_poll,
3064                                                 fc);
3065 }
3066
3067
3068 /**
3069  * @brief Stop polling a connection for ACKs.
3070  *
3071  * Once we have enough ACKs for future traffic, polls are no longer necessary.
3072  *
3073  * @param c Connection.
3074  * @param fwd Should we stop the poll in the FWD direction?
3075  */
3076 void
3077 GMC_stop_poll (struct MeshConnection *c, int fwd)
3078 {
3079   struct MeshFlowControl *fc;
3080
3081   fc = fwd ? &c->fwd_fc : &c->bck_fc;
3082   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
3083   {
3084     GNUNET_SCHEDULER_cancel (fc->poll_task);
3085     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
3086   }
3087 }
3088
3089 /**
3090  * Get a (static) string for a connection.
3091  *
3092  * @param c Connection.
3093  */
3094 const char *
3095 GMC_2s (const struct MeshConnection *c)
3096 {
3097   if (NULL == c)
3098     return "NULL";
3099
3100   if (NULL != c->t)
3101   {
3102     static char buf[128];
3103
3104     sprintf (buf, "%s (->%s)", GNUNET_h2s (&c->id), GMT_2s (c->t));
3105     return buf;
3106   }
3107   return GNUNET_h2s (&c->id);
3108 }