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   LOG (GNUNET_ERROR_TYPE_DEBUG, "FWD keepalive for %s\n", GMC_2s (c));
 952   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 953   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 954     return;
 955
 956   connection_maintain (c, GNUNET_YES);
 957
 958   /* Next execution will be scheduled by message_sent */
 959 }
 960
 961
 962 /**
 963  * Keep the connection alive in the BCK direction.
 964  *
 965  * TODO refactor and merge with connection_fwd_keepalive.
 966  *
 967  * @param cls Closure (connection to keepalive).
 968  * @param tc TaskContext.
 969  */
 970 static void
 971 connection_bck_keepalive (void *cls,
 972                           const struct GNUNET_SCHEDULER_TaskContext *tc)
 973 {
 974   struct MeshConnection *c = cls;
 975
 976   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 977   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 978     return;
 979
 980   connection_maintain (c, GNUNET_NO);
 981
 982   /* Next execution will be scheduled by message_sent */
 983 }
 984
 985
 986 /**
 987  * Schedule next keepalive task, taking in consideration
 988  * the connection state and number of retries.
 989  *
 990  * If the peer is not the origin, do nothing.
 991  *
 992  * @param c Connection for which to schedule the next keepalive.
 993  * @param fwd Direction for the next keepalive.
 994  */
 995 static void
 996 schedule_next_keepalive (struct MeshConnection *c, int fwd)
 997 {
 998   struct GNUNET_TIME_Relative delay;
 999   GNUNET_SCHEDULER_TaskIdentifier *task_id;
1000   GNUNET_SCHEDULER_Task keepalive_task;
1001
1002   if (GNUNET_NO == GMC_is_origin (c, fwd))
1003     return;
1004
1005   /* Calculate delay to use, depending on the state of the connection */
1006   if (MESH_CONNECTION_READY == c->state)
1007   {
1008     delay = refresh_connection_time;
1009   }
1010   else
1011   {
1012     if (1 > c->create_retry)
1013       c->create_retry = 1;
1014     delay = GNUNET_TIME_relative_multiply (create_connection_time,
1015                                            c->create_retry);
1016     if (c->create_retry < 64)
1017       c->create_retry *= 2;
1018   }
1019
1020   /* Select direction-dependent parameters */
1021   if (GNUNET_YES == fwd)
1022   {
1023     task_id = &c->fwd_maintenance_task;
1024     keepalive_task = &connection_fwd_keepalive;
1025   }
1026   else
1027   {
1028     task_id = &c->bck_maintenance_task;
1029     keepalive_task = &connection_bck_keepalive;
1030   }
1031
1032   /* Check that no one scheduled it before us (and alert in that case) */
1033   if (GNUNET_SCHEDULER_NO_TASK != *task_id)
1034   {
1035     GNUNET_break (0);
1036     GNUNET_SCHEDULER_cancel (*task_id);
1037   }
1038
1039   /* Schedule the task */
1040   *task_id = GNUNET_SCHEDULER_add_delayed (delay, keepalive_task, c);
1041 }
1042
1043
1044 /**
1045  * @brief Re-initiate traffic on this connection if necessary.
1046  *
1047  * Check if there is traffic queued towards this peer
1048  * and the core transmit handle is NULL (traffic was stalled).
1049  * If so, call core tmt rdy.
1050  *
1051  * @param c Connection on which initiate traffic.
1052  * @param fwd Is this about fwd traffic?
1053  */
1054 static void
1055 connection_unlock_queue (struct MeshConnection *c, int fwd)
1056 {
1057   struct MeshPeer *peer;
1058
1059   LOG (GNUNET_ERROR_TYPE_DEBUG,
1060               "connection_unlock_queue %s on %s\n",
1061               GM_f2s (fwd), GMC_2s (c));
1062
1063   if (GMC_is_terminal (c, fwd))
1064   {
1065     LOG (GNUNET_ERROR_TYPE_DEBUG, " is terminal!\n");
1066     return;
1067   }
1068
1069   peer = get_hop (c, fwd);
1070   GMP_queue_unlock (peer, c);
1071 }
1072
1073
1074 /**
1075  * Cancel all transmissions that belong to a certain connection.
1076  *
1077  * If the connection is scheduled for destruction and no more messages are left,
1078  * the connection will be destroyed by the continuation call.
1079  *
1080  * @param c Connection which to cancel. Might be destroyed during this call.
1081  * @param fwd Cancel fwd traffic?
1082  */
1083 static void
1084 connection_cancel_queues (struct MeshConnection *c, int fwd)
1085 {
1086   struct MeshFlowControl *fc;
1087   struct MeshPeer *peer;
1088
1089   LOG (GNUNET_ERROR_TYPE_DEBUG,
1090        " *** Cancel %s queues for connection %s\n",
1091        GM_f2s (fwd), GMC_2s (c));
1092   if (NULL == c)
1093   {
1094     GNUNET_break (0);
1095     return;
1096   }
1097
1098   fc = fwd ? &c->fwd_fc : &c->bck_fc;
1099   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
1100   {
1101     GNUNET_SCHEDULER_cancel (fc->poll_task);
1102     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1103     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** Cancel POLL in ccq for fc %p\n", fc);
1104   }
1105   peer = get_hop (c, fwd);
1106   GMP_queue_cancel (peer, c);
1107 }
1108
1109
1110 /**
1111  * Function called if a connection has been stalled for a while,
1112  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
1113  *
1114  * @param cls Closure (poll ctx).
1115  * @param tc TaskContext.
1116  */
1117 static void
1118 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1119
1120
1121 /**
1122  * Callback called when a queued POLL message is sent.
1123  *
1124  * @param cls Closure (FC).
1125  * @param c Connection this message was on.
1126  * @param q Queue handler this call invalidates.
1127  * @param type Type of message sent.
1128  * @param fwd Was this a FWD going message?
1129  * @param size Size of the message.
1130  */
1131 static void
1132 poll_sent (void *cls,
1133            struct MeshConnection *c,
1134            struct MeshConnectionQueue *q,
1135            uint16_t type, int fwd, size_t size)
1136 {
1137   struct MeshFlowControl *fc = cls;
1138
1139   if (2 == c->destroy)
1140   {
1141     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL canceled on shutdown\n");
1142     return;
1143   }
1144   LOG (GNUNET_ERROR_TYPE_DEBUG,
1145        " *** POLL sent for , scheduling new one!\n");
1146   fc->poll_msg = NULL;
1147   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
1148   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
1149                                                 &connection_poll, fc);
1150   LOG (GNUNET_ERROR_TYPE_DEBUG, " task %u\n", fc->poll_task);
1151
1152 }
1153
1154 /**
1155  * Function called if a connection has been stalled for a while,
1156  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
1157  *
1158  * @param cls Closure (poll ctx).
1159  * @param tc TaskContext.
1160  */
1161 static void
1162 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1163 {
1164   struct MeshFlowControl *fc = cls;
1165   struct GNUNET_MESH_Poll msg;
1166   struct MeshConnection *c;
1167
1168   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1169   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1170   {
1171     return;
1172   }
1173
1174   c = fc->c;
1175   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** Polling!\n");
1176   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** connection [%s]\n", GMC_2s (c));
1177   LOG (GNUNET_ERROR_TYPE_DEBUG, " ***   %s\n",
1178        fc == &c->fwd_fc ? "FWD" : "BCK");
1179
1180   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
1181   msg.header.size = htons (sizeof (msg));
1182   msg.pid = htonl (fc->last_pid_sent);
1183   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** last pid sent: %u!\n", fc->last_pid_sent);
1184   fc->poll_msg = GMC_send_prebuilt_message (&msg.header, c,
1185                                             fc == &c->fwd_fc, GNUNET_YES,
1186                                             &poll_sent, fc);
1187 }
1188
1189
1190 /**
1191  * Timeout function due to lack of keepalive/traffic from the owner.
1192  * Destroys connection if called.
1193  *
1194  * @param cls Closure (connection to destroy).
1195  * @param tc TaskContext.
1196  */
1197 static void
1198 connection_fwd_timeout (void *cls,
1199                         const struct GNUNET_SCHEDULER_TaskContext *tc)
1200 {
1201   struct MeshConnection *c = cls;
1202
1203   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
1204   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1205     return;
1206
1207   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s FWD timed out. Destroying.\n",
1208        GMC_2s (c));
1209   if (GMC_is_origin (c, GNUNET_YES)) /* If local, leave. */
1210   {
1211     GNUNET_break (0);
1212     return;
1213   }
1214
1215   GMC_destroy (c);
1216 }
1217
1218
1219 /**
1220  * Timeout function due to lack of keepalive/traffic from the destination.
1221  * Destroys connection if called.
1222  *
1223  * @param cls Closure (connection to destroy).
1224  * @param tc TaskContext
1225  */
1226 static void
1227 connection_bck_timeout (void *cls,
1228                         const struct GNUNET_SCHEDULER_TaskContext *tc)
1229 {
1230   struct MeshConnection *c = cls;
1231
1232   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
1233   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1234     return;
1235
1236   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s BCK timed out. Destroying.\n",
1237        GMC_2s (c));
1238
1239   if (GMC_is_origin (c, GNUNET_NO)) /* If local, leave. */
1240   {
1241     GNUNET_break (0);
1242     return;
1243   }
1244
1245   GMC_destroy (c);
1246 }
1247
1248
1249 /**
1250  * Resets the connection timeout task, some other message has done the
1251  * task's job.
1252  * - For the first peer on the direction this means to send
1253  *   a keepalive or a path confirmation message (either create or ACK).
1254  * - For all other peers, this means to destroy the connection,
1255  *   due to lack of activity.
1256  * Starts the timeout if no timeout was running (connection just created).
1257  *
1258  * @param c Connection whose timeout to reset.
1259  * @param fwd Is this forward?
1260  *
1261  * TODO use heap to improve efficiency of scheduler.
1262  */
1263 static void
1264 connection_reset_timeout (struct MeshConnection *c, int fwd)
1265 {
1266   GNUNET_SCHEDULER_TaskIdentifier *ti;
1267   GNUNET_SCHEDULER_Task f;
1268
1269   ti = fwd ? &c->fwd_maintenance_task : &c->bck_maintenance_task;
1270
1271   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s reset timeout\n", GM_f2s (fwd));
1272
1273   if (GNUNET_SCHEDULER_NO_TASK != *ti)
1274     GNUNET_SCHEDULER_cancel (*ti);
1275
1276   if (GMC_is_origin (c, fwd)) /* Startpoint */
1277   {
1278     f = fwd ? &connection_fwd_keepalive : &connection_bck_keepalive;
1279     *ti = GNUNET_SCHEDULER_add_delayed (refresh_connection_time, f, c);
1280   }
1281   else /* Relay, endpoint. */
1282   {
1283     struct GNUNET_TIME_Relative delay;
1284
1285     delay = GNUNET_TIME_relative_multiply (refresh_connection_time, 4);
1286     f = fwd ? &connection_fwd_timeout : &connection_bck_timeout;
1287     *ti = GNUNET_SCHEDULER_add_delayed (delay, f, c);
1288   }
1289 }
1290
1291
1292 /**
1293  * Add the connection to the list of both neighbors.
1294  *
1295  * @param c Connection.
1296  *
1297  * @return #GNUNET_OK if everything went fine
1298  *         #GNUNET_SYSERR if the was an error and @c c is malformed.
1299  */
1300 static int
1301 register_neighbors (struct MeshConnection *c)
1302 {
1303   struct MeshPeer *next_peer;
1304   struct MeshPeer *prev_peer;
1305
1306   next_peer = get_next_hop (c);
1307   prev_peer = get_prev_hop (c);
1308
1309   LOG (GNUNET_ERROR_TYPE_DEBUG, "register neighbors for connection %s\n",
1310        GMC_2s (c));
1311   path_debug (c->path);
1312   LOG (GNUNET_ERROR_TYPE_DEBUG, "own pos %u\n", c->own_pos);
1313   LOG (GNUNET_ERROR_TYPE_DEBUG, "putting connection %s to next peer %p\n",
1314        GMC_2s (c), next_peer);
1315   LOG (GNUNET_ERROR_TYPE_DEBUG, "next peer %p %s\n", next_peer, GMP_2s (next_peer));
1316   LOG (GNUNET_ERROR_TYPE_DEBUG, "putting connection %s to prev peer %p\n",
1317        GMC_2s (c), prev_peer);
1318   LOG (GNUNET_ERROR_TYPE_DEBUG, "prev peer %p %s\n", prev_peer, GMP_2s (prev_peer));
1319
1320   if (GNUNET_NO == GMP_is_neighbor (next_peer)
1321       || GNUNET_NO == GMP_is_neighbor (prev_peer))
1322   {
1323     if (GMC_is_origin (c, GNUNET_YES))
1324       GNUNET_STATISTICS_update (stats, "# local bad paths", 1, GNUNET_NO);
1325     GNUNET_STATISTICS_update (stats, "# bad paths", 1, GNUNET_NO);
1326
1327     LOG (GNUNET_ERROR_TYPE_DEBUG, "  register neighbors failed\n");
1328     LOG (GNUNET_ERROR_TYPE_DEBUG, "  prev: %s, neighbor?: %d\n",
1329          GMP_2s (prev_peer), GMP_is_neighbor (prev_peer));
1330     LOG (GNUNET_ERROR_TYPE_DEBUG, "  next: %s, neighbor?: %d\n",
1331          GMP_2s (next_peer), GMP_is_neighbor (next_peer));
1332     return GNUNET_SYSERR;
1333   }
1334
1335   GMP_add_connection (next_peer, c);
1336   GMP_add_connection (prev_peer, c);
1337
1338   return GNUNET_OK;
1339 }
1340
1341
1342 /**
1343  * Remove the connection from the list of both neighbors.
1344  *
1345  * @param c Connection.
1346  */
1347 static void
1348 unregister_neighbors (struct MeshConnection *c)
1349 {
1350   struct MeshPeer *peer;
1351
1352   peer = get_next_hop (c);
1353   if (GNUNET_OK != GMP_remove_connection (peer, c))
1354   {
1355     GNUNET_assert (MESH_CONNECTION_NEW == c->state
1356                   || MESH_CONNECTION_DESTROYED == c->state);
1357     LOG (GNUNET_ERROR_TYPE_DEBUG, "  cstate: %u\n", c->state);
1358     if (NULL != c->t) GMT_debug (c->t);
1359   }
1360
1361   peer = get_prev_hop (c);
1362   if (GNUNET_OK != GMP_remove_connection (peer, c))
1363   {
1364     GNUNET_assert (MESH_CONNECTION_NEW == c->state
1365                   || MESH_CONNECTION_DESTROYED == c->state);
1366     LOG (GNUNET_ERROR_TYPE_DEBUG, "  cstate: %u\n", c->state);
1367     if (NULL != c->t) GMT_debug (c->t);
1368   }
1369 }
1370
1371
1372 /**
1373  * Bind the connection to the peer and the tunnel to that peer.
1374  *
1375  * If the peer has no tunnel, create one. Update tunnel and connection
1376  * data structres to reflect new status.
1377  *
1378  * @param c Connection.
1379  * @param peer Peer.
1380  */
1381 static void
1382 add_to_peer (struct MeshConnection *c, struct MeshPeer *peer)
1383 {
1384   GMP_add_tunnel (peer);
1385   c->t = GMP_get_tunnel (peer);
1386   GMT_add_connection (c->t, c);
1387 }
1388
1389
1390 /**
1391  * Builds a path from a PeerIdentity array.
1392  *
1393  * @param peers PeerIdentity array.
1394  * @param size Size of the @c peers array.
1395  * @param own_pos Output parameter: own position in the path.
1396  *
1397  * @return Fixed and shortened path.
1398  */
1399 static struct MeshPeerPath *
1400 build_path_from_peer_ids (struct GNUNET_PeerIdentity *peers,
1401                           unsigned int size,
1402                           unsigned int *own_pos)
1403 {
1404   struct MeshPeerPath *path;
1405   GNUNET_PEER_Id shortid;
1406   unsigned int i;
1407   unsigned int j;
1408   unsigned int offset;
1409
1410   /* Create path */
1411   LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
1412   path = path_new (size);
1413   *own_pos = 0;
1414   offset = 0;
1415   for (i = 0; i < size; i++)
1416   {
1417     LOG (GNUNET_ERROR_TYPE_DEBUG, "  - %u: taking %s\n",
1418          i, GNUNET_i2s (&peers[i]));
1419     shortid = GNUNET_PEER_intern (&peers[i]);
1420
1421     /* Check for loops / duplicates */
1422     for (j = 0; j < i - offset; j++)
1423     {
1424       if (path->peers[j] == shortid)
1425       {
1426         LOG (GNUNET_ERROR_TYPE_DEBUG, "    already exists at pos %u\n", j);
1427         offset += i - j;
1428         LOG (GNUNET_ERROR_TYPE_DEBUG, "    offset now\n", offset);
1429         GNUNET_PEER_change_rc (shortid, -1);
1430       }
1431     }
1432     LOG (GNUNET_ERROR_TYPE_DEBUG, "    storing at %u\n", i - offset);
1433     path->peers[i - offset] = shortid;
1434     if (path->peers[i] == myid)
1435       *own_pos = i;
1436   }
1437   path->length -= offset;
1438
1439   if (path->peers[*own_pos] != myid)
1440   {
1441     /* create path: self not found in path through self */
1442     GNUNET_break_op (0);
1443     path_destroy (path);
1444     return NULL;
1445   }
1446
1447   return path;
1448 }
1449
1450 /******************************************************************************/
1451 /********************************    API    ***********************************/
1452 /******************************************************************************/
1453
1454 /**
1455  * Core handler for connection creation.
1456  *
1457  * @param cls Closure (unused).
1458  * @param peer Sender (neighbor).
1459  * @param message Message.
1460  *
1461  * @return GNUNET_OK to keep the connection open,
1462  *         GNUNET_SYSERR to close it (signal serious error)
1463  */
1464 int
1465 GMC_handle_create (void *cls, const struct GNUNET_PeerIdentity *peer,
1466                    const struct GNUNET_MessageHeader *message)
1467 {
1468   struct GNUNET_MESH_ConnectionCreate *msg;
1469   struct GNUNET_PeerIdentity *id;
1470   struct GNUNET_HashCode *cid;
1471   struct MeshPeerPath *path;
1472   struct MeshPeer *dest_peer;
1473   struct MeshPeer *orig_peer;
1474   struct MeshConnection *c;
1475   unsigned int own_pos;
1476   uint16_t size;
1477
1478   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1479   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection create msg\n");
1480
1481   /* Check size */
1482   size = ntohs (message->size);
1483   if (size < sizeof (struct GNUNET_MESH_ConnectionCreate))
1484   {
1485     GNUNET_break_op (0);
1486     return GNUNET_OK;
1487   }
1488
1489   /* Calculate hops */
1490   size -= sizeof (struct GNUNET_MESH_ConnectionCreate);
1491   if (size % sizeof (struct GNUNET_PeerIdentity))
1492   {
1493     GNUNET_break_op (0);
1494     return GNUNET_OK;
1495   }
1496   size /= sizeof (struct GNUNET_PeerIdentity);
1497   if (1 > size)
1498   {
1499     GNUNET_break_op (0);
1500     return GNUNET_OK;
1501   }
1502   LOG (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
1503
1504   /* Get parameters */
1505   msg = (struct GNUNET_MESH_ConnectionCreate *) message;
1506   cid = &msg->cid;
1507   id = (struct GNUNET_PeerIdentity *) &msg[1];
1508   LOG (GNUNET_ERROR_TYPE_DEBUG, "    connection %s (%s->).\n",
1509        GNUNET_h2s (cid), GNUNET_i2s (id));
1510
1511   /* Create connection */
1512   c = connection_get (cid);
1513   if (NULL == c)
1514   {
1515     path = build_path_from_peer_ids ((struct GNUNET_PeerIdentity *) &msg[1],
1516                                      size, &own_pos);
1517     if (NULL == path)
1518       return GNUNET_OK;
1519     if (0 == own_pos)
1520     {
1521       GNUNET_break_op (0);
1522       path_destroy (path);
1523       return GNUNET_OK;
1524     }
1525     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
1526     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating connection\n");
1527     c = GMC_new (cid, NULL, path_duplicate (path), own_pos);
1528     if (NULL == c)
1529     {
1530       if (path->length - 1 == own_pos)
1531       {
1532         /* If we are destination, why did the creation fail? */
1533         GNUNET_break (0);
1534         return GNUNET_OK;
1535       }
1536       send_broken2 (cid, &my_full_id,
1537                     GNUNET_PEER_resolve2 (path->peers[own_pos + 1]),
1538                     path->peers[own_pos - 1]);
1539       path_destroy (path);
1540       return GNUNET_OK;
1541     }
1542     GMP_add_path_to_all (path, GNUNET_NO);
1543     connection_reset_timeout (c, GNUNET_YES);
1544   }
1545   else
1546   {
1547     path = path_duplicate (c->path);
1548   }
1549   if (MESH_CONNECTION_NEW == c->state)
1550     connection_change_state (c, MESH_CONNECTION_SENT);
1551
1552   /* Remember peers */
1553   dest_peer = GMP_get (&id[size - 1]);
1554   orig_peer = GMP_get (&id[0]);
1555
1556   /* Is it a connection to us? */
1557   if (c->own_pos == size - 1)
1558   {
1559     LOG (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
1560     GMP_add_path_to_origin (orig_peer, path_duplicate (path), GNUNET_YES);
1561
1562     add_to_peer (c, orig_peer);
1563     if (MESH_TUNNEL3_NEW == GMT_get_cstate (c->t))
1564       GMT_change_cstate (c->t,  MESH_TUNNEL3_WAITING);
1565
1566     send_connection_ack (c, GNUNET_NO);
1567     if (MESH_CONNECTION_SENT == c->state)
1568       connection_change_state (c, MESH_CONNECTION_ACK);
1569   }
1570   else
1571   {
1572     /* It's for somebody else! Retransmit. */
1573     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
1574     GMP_add_path (dest_peer, path_duplicate (path), GNUNET_NO);
1575     GMP_add_path_to_origin (orig_peer, path_duplicate (path), GNUNET_NO);
1576     GMC_send_prebuilt_message (message, c, GNUNET_YES, GNUNET_YES,
1577                                NULL, NULL);
1578   }
1579   path_destroy (path);
1580   return GNUNET_OK;
1581 }
1582
1583
1584 /**
1585  * Core handler for path confirmations.
1586  *
1587  * @param cls closure
1588  * @param message message
1589  * @param peer peer identity this notification is about
1590  *
1591  * @return GNUNET_OK to keep the connection open,
1592  *         GNUNET_SYSERR to close it (signal serious error)
1593  */
1594 int
1595 GMC_handle_confirm (void *cls, const struct GNUNET_PeerIdentity *peer,
1596                     const struct GNUNET_MessageHeader *message)
1597 {
1598   struct GNUNET_MESH_ConnectionACK *msg;
1599   struct MeshConnection *c;
1600   struct MeshPeerPath *p;
1601   struct MeshPeer *pi;
1602   enum MeshConnectionState oldstate;
1603   int fwd;
1604
1605   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1606   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection ACK msg\n");
1607   msg = (struct GNUNET_MESH_ConnectionACK *) message;
1608   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n",
1609               GNUNET_h2s (&msg->cid));
1610   c = connection_get (&msg->cid);
1611   if (NULL == c)
1612   {
1613     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
1614                               1, GNUNET_NO);
1615     LOG (GNUNET_ERROR_TYPE_DEBUG, "  don't know the connection!\n");
1616     return GNUNET_OK;
1617   }
1618
1619   if (GNUNET_NO != c->destroy)
1620   {
1621     LOG (GNUNET_ERROR_TYPE_DEBUG, "  connection being destroyed\n");
1622     return GNUNET_OK;
1623   }
1624
1625   oldstate = c->state;
1626   LOG (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n", GNUNET_i2s (peer));
1627   pi = GMP_get (peer);
1628   if (get_next_hop (c) == pi)
1629   {
1630     LOG (GNUNET_ERROR_TYPE_DEBUG, "  SYNACK\n");
1631     fwd = GNUNET_NO;
1632     if (MESH_CONNECTION_SENT == oldstate)
1633       connection_change_state (c, MESH_CONNECTION_ACK);
1634   }
1635   else if (get_prev_hop (c) == pi)
1636   {
1637     LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK\n");
1638     fwd = GNUNET_YES;
1639     connection_change_state (c, MESH_CONNECTION_READY);
1640   }
1641   else
1642   {
1643     GNUNET_break_op (0);
1644     return GNUNET_OK;
1645   }
1646
1647   connection_reset_timeout (c, fwd);
1648
1649   /* Add path to peers? */
1650   p = c->path;
1651   if (NULL != p)
1652   {
1653     GMP_add_path_to_all (p, GNUNET_YES);
1654   }
1655   else
1656   {
1657     GNUNET_break (0);
1658   }
1659
1660   /* Message for us as creator? */
1661   if (GMC_is_origin (c, GNUNET_YES))
1662   {
1663     if (GNUNET_NO != fwd)
1664     {
1665       GNUNET_break_op (0);
1666       return GNUNET_OK;
1667     }
1668     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection (SYN)ACK for us!\n");
1669
1670     /* If just created, cancel the short timeout and start a long one */
1671     if (MESH_CONNECTION_SENT == oldstate)
1672       connection_reset_timeout (c, GNUNET_YES);
1673
1674     /* Change connection state */
1675     connection_change_state (c, MESH_CONNECTION_READY);
1676     send_connection_ack (c, GNUNET_YES);
1677
1678     /* Change tunnel state, trigger KX */
1679     if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
1680       GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
1681
1682     return GNUNET_OK;
1683   }
1684
1685   /* Message for us as destination? */
1686   if (GMC_is_terminal (c, GNUNET_YES))
1687   {
1688     if (GNUNET_YES != fwd)
1689     {
1690       GNUNET_break_op (0);
1691       return GNUNET_OK;
1692     }
1693     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection ACK for us!\n");
1694
1695     /* If just created, cancel the short timeout and start a long one */
1696     if (MESH_CONNECTION_ACK == oldstate)
1697       connection_reset_timeout (c, GNUNET_NO);
1698
1699     /* Change tunnel state */
1700     if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
1701       GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
1702
1703     return GNUNET_OK;
1704   }
1705
1706   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1707   GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1708   return GNUNET_OK;
1709 }
1710
1711
1712 /**
1713  * Core handler for notifications of broken paths
1714  *
1715  * @param cls Closure (unused).
1716  * @param id Peer identity of sending neighbor.
1717  * @param message Message.
1718  *
1719  * @return GNUNET_OK to keep the connection open,
1720  *         GNUNET_SYSERR to close it (signal serious error)
1721  */
1722 int
1723 GMC_handle_broken (void* cls,
1724                    const struct GNUNET_PeerIdentity* id,
1725                    const struct GNUNET_MessageHeader* message)
1726 {
1727   struct GNUNET_MESH_ConnectionBroken *msg;
1728   struct MeshConnection *c;
1729   int fwd;
1730
1731   LOG (GNUNET_ERROR_TYPE_DEBUG,
1732               "Received a CONNECTION BROKEN msg from %s\n", GNUNET_i2s (id));
1733   msg = (struct GNUNET_MESH_ConnectionBroken *) message;
1734   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1735               GNUNET_i2s (&msg->peer1));
1736   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1737               GNUNET_i2s (&msg->peer2));
1738   c = connection_get (&msg->cid);
1739   if (NULL == c)
1740   {
1741     GNUNET_break_op (0);
1742     return GNUNET_OK;
1743   }
1744
1745   fwd = is_fwd (c, id);
1746   if (GMC_is_terminal (c, fwd))
1747   {
1748     path_invalidate (c->path);
1749     if (0 < c->pending_messages)
1750       c->destroy = GNUNET_YES;
1751     else
1752       GMC_destroy (c);
1753   }
1754   else
1755   {
1756     GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1757     c->destroy = GNUNET_YES;
1758     connection_cancel_queues (c, !fwd);
1759   }
1760
1761   return GNUNET_OK;
1762
1763 }
1764
1765
1766 /**
1767  * Core handler for tunnel destruction
1768  *
1769  * @param cls Closure (unused).
1770  * @param peer Peer identity of sending neighbor.
1771  * @param message Message.
1772  *
1773  * @return GNUNET_OK to keep the connection open,
1774  *         GNUNET_SYSERR to close it (signal serious error)
1775  */
1776 int
1777 GMC_handle_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
1778                     const struct GNUNET_MessageHeader *message)
1779 {
1780   struct GNUNET_MESH_ConnectionDestroy *msg;
1781   struct MeshConnection *c;
1782   int fwd;
1783
1784   msg = (struct GNUNET_MESH_ConnectionDestroy *) message;
1785   LOG (GNUNET_ERROR_TYPE_DEBUG,
1786               "Got a CONNECTION DESTROY message from %s\n",
1787               GNUNET_i2s (peer));
1788   LOG (GNUNET_ERROR_TYPE_DEBUG,
1789               "  for connection %s\n",
1790               GNUNET_h2s (&msg->cid));
1791   c = connection_get (&msg->cid);
1792   if (NULL == c)
1793   {
1794     /* Probably already got the message from another path,
1795      * destroyed the tunnel and retransmitted to children.
1796      * Safe to ignore.
1797      */
1798     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
1799                               1, GNUNET_NO);
1800     LOG (GNUNET_ERROR_TYPE_DEBUG, "  connection unknown: already destroyed?\n");
1801     return GNUNET_OK;
1802   }
1803   fwd = is_fwd (c, peer);
1804   if (GNUNET_SYSERR == fwd)
1805   {
1806     GNUNET_break_op (0); /* FIXME */
1807     return GNUNET_OK;
1808   }
1809   if (GNUNET_NO == GMC_is_terminal (c, fwd))
1810     GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
1811   else if (0 == c->pending_messages)
1812   {
1813     LOG (GNUNET_ERROR_TYPE_DEBUG, "!  directly destroying connection!\n");
1814     GMC_destroy (c);
1815     return GNUNET_OK;
1816   }
1817   c->destroy = GNUNET_YES;
1818   c->state = MESH_CONNECTION_DESTROYED;
1819   if (NULL != c->t)
1820   {
1821     GMT_remove_connection (c->t, c);
1822     c->t = NULL;
1823   }
1824
1825   return GNUNET_OK;
1826 }
1827
1828 /**
1829  * Generic handler for mesh network encrypted traffic.
1830  *
1831  * @param peer Peer identity this notification is about.
1832  * @param msg Encrypted message.
1833  *
1834  * @return GNUNET_OK to keep the connection open,
1835  *         GNUNET_SYSERR to close it (signal serious error)
1836  */
1837 static int
1838 handle_mesh_encrypted (const struct GNUNET_PeerIdentity *peer,
1839                        const struct GNUNET_MESH_Encrypted *msg)
1840 {
1841   struct MeshConnection *c;
1842   struct MeshPeer *neighbor;
1843   struct MeshFlowControl *fc;
1844   GNUNET_PEER_Id peer_id;
1845   uint32_t pid;
1846   uint32_t ttl;
1847   uint16_t type;
1848   size_t size;
1849   int fwd;
1850
1851   /* Check size */
1852   size = ntohs (msg->header.size);
1853   if (size <
1854       sizeof (struct GNUNET_MESH_Encrypted) +
1855       sizeof (struct GNUNET_MessageHeader))
1856   {
1857     GNUNET_break_op (0);
1858     return GNUNET_OK;
1859   }
1860   type = ntohs (msg->header.type);
1861   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1862   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message (#%u) from %s\n",
1863        GM_m2s (type), ntohl (msg->pid), GNUNET_i2s (peer));
1864
1865   /* Check connection */
1866   c = connection_get (&msg->cid);
1867   if (NULL == c)
1868   {
1869     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1870     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING enc on unknown connection %s\n",
1871          GNUNET_h2s (&msg->cid));
1872     return GNUNET_OK;
1873   }
1874
1875   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n", GMC_2s (c));
1876
1877   /* Check if origin is as expected */
1878   neighbor = get_prev_hop (c);
1879   peer_id = GNUNET_PEER_search (peer);
1880   if (peer_id == GMP_get_short_id (neighbor))
1881   {
1882     fwd = GNUNET_YES;
1883   }
1884   else
1885   {
1886     neighbor = get_next_hop (c);
1887     if (peer_id == GMP_get_short_id (neighbor))
1888     {
1889       fwd = GNUNET_NO;
1890     }
1891     else
1892     {
1893       /* Unexpected peer sending traffic on a connection. */
1894       GNUNET_break_op (0);
1895       return GNUNET_OK;
1896     }
1897   }
1898
1899   /* Check PID */
1900   fc = fwd ? &c->bck_fc : &c->fwd_fc;
1901   pid = ntohl (msg->pid);
1902   if (GM_is_pid_bigger (pid, fc->last_ack_sent))
1903   {
1904     GNUNET_STATISTICS_update (stats, "# unsolicited message", 1, GNUNET_NO);
1905     LOG (GNUNET_ERROR_TYPE_DEBUG,
1906                 "WARNING Received PID %u, (prev %u), ACK %u\n",
1907                 pid, fc->last_pid_recv, fc->last_ack_sent);
1908     return GNUNET_OK;
1909   }
1910   if (GNUNET_NO == GM_is_pid_bigger (pid, fc->last_pid_recv))
1911   {
1912     GNUNET_STATISTICS_update (stats, "# duplicate PID", 1, GNUNET_NO);
1913     LOG (GNUNET_ERROR_TYPE_DEBUG,
1914                 " Pid %u not expected (%u+), dropping!\n",
1915                 pid, fc->last_pid_recv + 1);
1916     return GNUNET_OK;
1917   }
1918   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
1919     connection_change_state (c, MESH_CONNECTION_READY);
1920   connection_reset_timeout (c, fwd);
1921   fc->last_pid_recv = pid;
1922
1923   /* Is this message for us? */
1924   if (GMC_is_terminal (c, fwd))
1925   {
1926     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
1927     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
1928
1929     if (NULL == c->t)
1930     {
1931       GNUNET_break (GNUNET_NO != c->destroy);
1932       return GNUNET_OK;
1933     }
1934     fc->last_pid_recv = pid;
1935     GMT_handle_encrypted (c->t, msg);
1936     GMC_send_ack (c, fwd, GNUNET_NO);
1937     return GNUNET_OK;
1938   }
1939
1940   /* Message not for us: forward to next hop */
1941   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1942   ttl = ntohl (msg->ttl);
1943   LOG (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
1944   if (ttl == 0)
1945   {
1946     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
1947     LOG (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
1948     GMC_send_ack (c, fwd, GNUNET_NO);
1949     return GNUNET_OK;
1950   }
1951
1952   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
1953   GMC_send_prebuilt_message (&msg->header, c, fwd, GNUNET_NO, NULL, NULL);
1954
1955   return GNUNET_OK;
1956 }
1957
1958 /**
1959  * Generic handler for mesh network encrypted traffic.
1960  *
1961  * @param peer Peer identity this notification is about.
1962  * @param msg Encrypted message.
1963  *
1964  * @return GNUNET_OK to keep the connection open,
1965  *         GNUNET_SYSERR to close it (signal serious error)
1966  */
1967 static int
1968 handle_mesh_kx (const struct GNUNET_PeerIdentity *peer,
1969                 const struct GNUNET_MESH_KX *msg)
1970 {
1971   struct MeshConnection *c;
1972   struct MeshPeer *neighbor;
1973   GNUNET_PEER_Id peer_id;
1974   size_t size;
1975   uint16_t type;
1976   int fwd;
1977
1978   /* Check size */
1979   size = ntohs (msg->header.size);
1980   if (size <
1981       sizeof (struct GNUNET_MESH_Encrypted) +
1982       sizeof (struct GNUNET_MessageHeader))
1983   {
1984     GNUNET_break_op (0);
1985     return GNUNET_OK;
1986   }
1987   type = ntohs (msg->header.type);
1988   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1989   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
1990        GM_m2s (type), GNUNET_i2s (peer));
1991
1992   /* Check connection */
1993   c = connection_get (&msg->cid);
1994   if (NULL == c)
1995   {
1996     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1997     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING kx on unknown connection %s\n",
1998          GNUNET_h2s (&msg->cid));
1999     return GNUNET_OK;
2000   }
2001   LOG (GNUNET_ERROR_TYPE_DEBUG, " on connection %s\n", GMC_2s (c));
2002
2003   /* Check if origin is as expected */
2004   neighbor = get_prev_hop (c);
2005   peer_id = GNUNET_PEER_search (peer);
2006   if (peer_id == GMP_get_short_id (neighbor))
2007   {
2008     fwd = GNUNET_YES;
2009   }
2010   else
2011   {
2012     neighbor = get_next_hop (c);
2013     if (peer_id == GMP_get_short_id (neighbor))
2014     {
2015       fwd = GNUNET_NO;
2016     }
2017     else
2018     {
2019       /* Unexpected peer sending traffic on a connection. */
2020       GNUNET_break_op (0);
2021       return GNUNET_OK;
2022     }
2023   }
2024
2025   /* Count as connection confirmation. */
2026   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
2027   {
2028     connection_change_state (c, MESH_CONNECTION_READY);
2029     if (NULL != c->t)
2030     {
2031       if (MESH_TUNNEL3_WAITING == GMT_get_cstate (c->t))
2032         GMT_change_cstate (c->t, MESH_TUNNEL3_READY);
2033     }
2034   }
2035   connection_reset_timeout (c, fwd);
2036
2037   /* Is this message for us? */
2038   if (GMC_is_terminal (c, fwd))
2039   {
2040     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
2041     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
2042     if (NULL == c->t)
2043     {
2044       GNUNET_break (0);
2045       return GNUNET_OK;
2046     }
2047     GMT_handle_kx (c->t, &msg[1].header);
2048     return GNUNET_OK;
2049   }
2050
2051   /* Message not for us: forward to next hop */
2052   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
2053   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
2054   GMC_send_prebuilt_message (&msg->header, c, fwd, GNUNET_NO, NULL, NULL);
2055
2056   return GNUNET_OK;
2057 }
2058
2059
2060 /**
2061  * Core handler for encrypted mesh network traffic (channel mgmt, data).
2062  *
2063  * @param cls Closure (unused).
2064  * @param message Message received.
2065  * @param peer Peer who sent the message.
2066  *
2067  * @return GNUNET_OK to keep the connection open,
2068  *         GNUNET_SYSERR to close it (signal serious error)
2069  */
2070 int
2071 GMC_handle_encrypted (void *cls, const struct GNUNET_PeerIdentity *peer,
2072                       const struct GNUNET_MessageHeader *message)
2073 {
2074   return handle_mesh_encrypted (peer,
2075                                 (struct GNUNET_MESH_Encrypted *)message);
2076 }
2077
2078
2079 /**
2080  * Core handler for key exchange traffic (ephemeral key, ping, pong).
2081  *
2082  * @param cls Closure (unused).
2083  * @param message Message received.
2084  * @param peer Peer who sent the message.
2085  *
2086  * @return GNUNET_OK to keep the connection open,
2087  *         GNUNET_SYSERR to close it (signal serious error)
2088  */
2089 int
2090 GMC_handle_kx (void *cls, const struct GNUNET_PeerIdentity *peer,
2091                const struct GNUNET_MessageHeader *message)
2092 {
2093   return handle_mesh_kx (peer,
2094                          (struct GNUNET_MESH_KX *) message);
2095 }
2096
2097
2098 /**
2099  * Core handler for mesh network traffic point-to-point acks.
2100  *
2101  * @param cls closure
2102  * @param message message
2103  * @param peer peer identity this notification is about
2104  *
2105  * @return GNUNET_OK to keep the connection open,
2106  *         GNUNET_SYSERR to close it (signal serious error)
2107  */
2108 int
2109 GMC_handle_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
2110                 const struct GNUNET_MessageHeader *message)
2111 {
2112   struct GNUNET_MESH_ACK *msg;
2113   struct MeshConnection *c;
2114   struct MeshFlowControl *fc;
2115   GNUNET_PEER_Id id;
2116   uint32_t ack;
2117   int fwd;
2118
2119   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
2120   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
2121               GNUNET_i2s (peer));
2122   msg = (struct GNUNET_MESH_ACK *) message;
2123
2124   c = connection_get (&msg->cid);
2125
2126   if (NULL == c)
2127   {
2128     GNUNET_STATISTICS_update (stats, "# ack on unknown connection", 1,
2129                               GNUNET_NO);
2130     return GNUNET_OK;
2131   }
2132
2133   /* Is this a forward or backward ACK? */
2134   id = GNUNET_PEER_search (peer);
2135   if (GMP_get_short_id (get_next_hop (c)) == id)
2136   {
2137     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
2138     fc = &c->fwd_fc;
2139     fwd = GNUNET_YES;
2140   }
2141   else if (GMP_get_short_id (get_prev_hop (c)) == id)
2142   {
2143     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
2144     fc = &c->bck_fc;
2145     fwd = GNUNET_NO;
2146   }
2147   else
2148   {
2149     GNUNET_break_op (0);
2150     return GNUNET_OK;
2151   }
2152
2153   ack = ntohl (msg->ack);
2154   LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u (was %u)\n",
2155               ack, fc->last_ack_recv);
2156   if (GM_is_pid_bigger (ack, fc->last_ack_recv))
2157     fc->last_ack_recv = ack;
2158
2159   /* Cancel polling if the ACK is big enough. */
2160   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task &&
2161       GM_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2162   {
2163     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Cancel poll\n");
2164     GNUNET_SCHEDULER_cancel (fc->poll_task);
2165     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2166     fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
2167   }
2168
2169   connection_unlock_queue (c, fwd);
2170
2171   return GNUNET_OK;
2172 }
2173
2174
2175 /**
2176  * Core handler for mesh network traffic point-to-point ack polls.
2177  *
2178  * @param cls closure
2179  * @param message message
2180  * @param peer peer identity this notification is about
2181  *
2182  * @return GNUNET_OK to keep the connection open,
2183  *         GNUNET_SYSERR to close it (signal serious error)
2184  */
2185 int
2186 GMC_handle_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
2187                  const struct GNUNET_MessageHeader *message)
2188 {
2189   struct GNUNET_MESH_Poll *msg;
2190   struct MeshConnection *c;
2191   struct MeshFlowControl *fc;
2192   GNUNET_PEER_Id id;
2193   uint32_t pid;
2194   int fwd;
2195
2196   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
2197   LOG (GNUNET_ERROR_TYPE_DEBUG,
2198        "Got a POLL message from %s!\n",
2199        GNUNET_i2s (peer));
2200
2201   msg = (struct GNUNET_MESH_Poll *) message;
2202
2203   c = connection_get (&msg->cid);
2204
2205   if (NULL == c)
2206   {
2207     GNUNET_STATISTICS_update (stats, "# poll on unknown connection", 1,
2208                               GNUNET_NO);
2209     LOG (GNUNET_ERROR_TYPE_DEBUG,
2210          "WARNING POLL message on unknown connection %s!\n",
2211          GNUNET_h2s (&msg->cid));
2212     return GNUNET_OK;
2213   }
2214
2215   /* Is this a forward or backward ACK?
2216    * Note: a poll should never be needed in a loopback case,
2217    * since there is no possiblility of packet loss there, so
2218    * this way of discerining FWD/BCK should not be a problem.
2219    */
2220   id = GNUNET_PEER_search (peer);
2221   if (GMP_get_short_id (get_next_hop (c)) == id)
2222   {
2223     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
2224     fc = &c->fwd_fc;
2225   }
2226   else if (GMP_get_short_id (get_prev_hop (c)) == id)
2227   {
2228     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
2229     fc = &c->bck_fc;
2230   }
2231   else
2232   {
2233     GNUNET_break_op (0);
2234     return GNUNET_OK;
2235   }
2236
2237   pid = ntohl (msg->pid);
2238   LOG (GNUNET_ERROR_TYPE_DEBUG, "  PID %u, OLD %u\n", pid, fc->last_pid_recv);
2239   fc->last_pid_recv = pid;
2240   fwd = fc == &c->bck_fc;
2241   GMC_send_ack (c, fwd, GNUNET_YES);
2242
2243   return GNUNET_OK;
2244 }
2245
2246
2247 /**
2248  * Core handler for mesh keepalives.
2249  *
2250  * @param cls closure
2251  * @param message message
2252  * @param peer peer identity this notification is about
2253  * @return GNUNET_OK to keep the connection open,
2254  *         GNUNET_SYSERR to close it (signal serious error)
2255  *
2256  * TODO: Check who we got this from, to validate route.
2257  */
2258 int
2259 GMC_handle_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
2260                       const struct GNUNET_MessageHeader *message)
2261 {
2262   struct GNUNET_MESH_ConnectionKeepAlive *msg;
2263   struct MeshConnection *c;
2264   struct MeshPeer *neighbor;
2265   GNUNET_PEER_Id peer_id;
2266   int fwd;
2267
2268   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) message;
2269   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
2270               GNUNET_i2s (peer));
2271
2272   c = connection_get (&msg->cid);
2273   if (NULL == c)
2274   {
2275     GNUNET_STATISTICS_update (stats, "# keepalive on unknown connection", 1,
2276                               GNUNET_NO);
2277     return GNUNET_OK;
2278   }
2279
2280   /* Check if origin is as expected TODO refactor and reuse */
2281   peer_id = GNUNET_PEER_search (peer);
2282   neighbor = get_prev_hop (c);
2283   if (peer_id == GMP_get_short_id (neighbor))
2284   {
2285     fwd = GNUNET_YES;
2286   }
2287   else
2288   {
2289     neighbor = get_next_hop (c);
2290     if (peer_id == GMP_get_short_id (neighbor))
2291     {
2292       fwd = GNUNET_NO;
2293     }
2294     else
2295     {
2296       GNUNET_break_op (0);
2297       return GNUNET_OK;
2298     }
2299   }
2300
2301   connection_change_state (c, MESH_CONNECTION_READY);
2302   connection_reset_timeout (c, fwd);
2303
2304   if (GMC_is_terminal (c, fwd))
2305     return GNUNET_OK;
2306
2307   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
2308   GMC_send_prebuilt_message (message, c, fwd, GNUNET_YES, NULL, NULL);
2309
2310   return GNUNET_OK;
2311 }
2312
2313
2314 /**
2315  * Send an ACK on the appropriate connection/channel, depending on
2316  * the direction and the position of the peer.
2317  *
2318  * @param c Which connection to send the hop-by-hop ACK.
2319  * @param fwd Is this a fwd ACK? (will go dest->root).
2320  * @param force Send the ACK even if suboptimal (e.g. requested by POLL).
2321  */
2322 void
2323 GMC_send_ack (struct MeshConnection *c, int fwd, int force)
2324 {
2325   unsigned int buffer;
2326
2327   LOG (GNUNET_ERROR_TYPE_DEBUG,
2328        "GMC send %s ACK on %s\n",
2329        GM_f2s (fwd), GMC_2s (c));
2330
2331   if (NULL == c)
2332   {
2333     GNUNET_break (0);
2334     return;
2335   }
2336
2337   if (GNUNET_NO != c->destroy)
2338   {
2339     LOG (GNUNET_ERROR_TYPE_DEBUG, "  being destroyed, why bother...\n");
2340     return;
2341   }
2342
2343   /* Get available buffer space */
2344   if (GMC_is_terminal (c, fwd))
2345   {
2346     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from all channels\n");
2347     buffer = GMT_get_channels_buffer (c->t);
2348   }
2349   else
2350   {
2351     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from one connection\n");
2352     buffer = GMC_get_buffer (c, fwd);
2353   }
2354   LOG (GNUNET_ERROR_TYPE_DEBUG, "  buffer available: %u\n", buffer);
2355   if (0 == buffer && GNUNET_NO == force)
2356     return;
2357
2358   /* Send available buffer space */
2359   if (GMC_is_origin (c, fwd))
2360   {
2361     GNUNET_assert (NULL != c->t);
2362     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on channels...\n");
2363     GMT_unchoke_channels (c->t);
2364   }
2365   else
2366   {
2367     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on connection\n");
2368     send_ack (c, buffer, fwd, force);
2369   }
2370 }
2371
2372
2373 /**
2374  * Initialize the connections subsystem
2375  *
2376  * @param c Configuration handle.
2377  */
2378 void
2379 GMC_init (const struct GNUNET_CONFIGURATION_Handle *c)
2380 {
2381   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
2382   if (GNUNET_OK !=
2383       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
2384                                              &max_msgs_queue))
2385   {
2386     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2387                                "MESH", "MAX_MSGS_QUEUE", "MISSING");
2388     GNUNET_SCHEDULER_shutdown ();
2389     return;
2390   }
2391
2392   if (GNUNET_OK !=
2393       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_CONNECTIONS",
2394                                              &max_connections))
2395   {
2396     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2397                                "MESH", "MAX_CONNECTIONS", "MISSING");
2398     GNUNET_SCHEDULER_shutdown ();
2399     return;
2400   }
2401
2402   if (GNUNET_OK !=
2403       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_CONNECTION_TIME",
2404                                            &refresh_connection_time))
2405   {
2406     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
2407                                "MESH", "REFRESH_CONNECTION_TIME", "MISSING");
2408     GNUNET_SCHEDULER_shutdown ();
2409     return;
2410   }
2411   create_connection_time = GNUNET_TIME_UNIT_SECONDS;
2412   connections = GNUNET_CONTAINER_multihashmap_create (1024, GNUNET_YES);
2413 }
2414
2415
2416 /**
2417  * Destroy each connection on shutdown.
2418  *
2419  * @param cls Closure (unused).
2420  * @param key Current key code (CID, unused).
2421  * @param value Value in the hash map (connection)
2422  *
2423  * @return #GNUNET_YES, because we should continue to iterate,
2424  */
2425 static int
2426 shutdown_iterator (void *cls,
2427                    const struct GNUNET_HashCode *key,
2428                    void *value)
2429 {
2430   struct MeshConnection *c = value;
2431
2432   GMC_destroy (c);
2433   return GNUNET_YES;
2434 }
2435
2436
2437 /**
2438  * Shut down the connections subsystem.
2439  */
2440 void
2441 GMC_shutdown (void)
2442 {
2443   GNUNET_CONTAINER_multihashmap_iterate (connections, &shutdown_iterator, NULL);
2444   GNUNET_CONTAINER_multihashmap_destroy (connections);
2445   connections = NULL;
2446 }
2447
2448
2449 struct MeshConnection *
2450 GMC_new (const struct GNUNET_HashCode *cid,
2451          struct MeshTunnel3 *t,
2452          struct MeshPeerPath *p,
2453          unsigned int own_pos)
2454 {
2455   struct MeshConnection *c;
2456
2457   c = GNUNET_new (struct MeshConnection);
2458   c->id = *cid;
2459   GNUNET_assert (GNUNET_OK ==
2460                  GNUNET_CONTAINER_multihashmap_put (connections,
2461                                                     &c->id, c,
2462                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
2463   fc_init (&c->fwd_fc);
2464   fc_init (&c->bck_fc);
2465   c->fwd_fc.c = c;
2466   c->bck_fc.c = c;
2467
2468   c->t = t;
2469   GNUNET_assert (own_pos <= p->length - 1);
2470   c->own_pos = own_pos;
2471   c->path = p;
2472
2473   if (GNUNET_OK != register_neighbors (c))
2474   {
2475     if (0 == own_pos)
2476     {
2477       path_invalidate (c->path);
2478       c->t = NULL;
2479       c->path = NULL;
2480     }
2481     GMC_destroy (c);
2482     return NULL;
2483   }
2484
2485   return c;
2486 }
2487
2488
2489 void
2490 GMC_destroy (struct MeshConnection *c)
2491 {
2492   if (NULL == c)
2493   {
2494     GNUNET_break (0);
2495     return;
2496   }
2497
2498   if (2 == c->destroy) /* cancel queues -> GMP_queue_cancel -> q_destroy -> */
2499     return;            /* -> message_sent -> GMC_destroy. Don't loop. */
2500   c->destroy = 2;
2501
2502   LOG (GNUNET_ERROR_TYPE_DEBUG, "destroying connection %s\n", GMC_2s (c));
2503   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc's f: %p, b: %p\n",
2504        &c->fwd_fc, &c->bck_fc);
2505   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc tasks f: %u, b: %u\n",
2506        c->fwd_fc.poll_task, c->bck_fc.poll_task);
2507
2508   /* Cancel all traffic */
2509   if (NULL != c->path)
2510   {
2511     connection_cancel_queues (c, GNUNET_YES);
2512     connection_cancel_queues (c, GNUNET_NO);
2513     unregister_neighbors (c);
2514   }
2515
2516   LOG (GNUNET_ERROR_TYPE_DEBUG, " fc tasks f: %u, b: %u\n",
2517        c->fwd_fc.poll_task, c->bck_fc.poll_task);
2518
2519   /* Cancel maintainance task (keepalive/timeout) */
2520   if (NULL != c->fwd_fc.poll_msg)
2521   {
2522     GMC_cancel (c->fwd_fc.poll_msg);
2523     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL msg FWD canceled\n");
2524   }
2525   if (NULL != c->bck_fc.poll_msg)
2526   {
2527     GMC_cancel (c->bck_fc.poll_msg);
2528     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL msg BCK canceled\n");
2529   }
2530
2531   /* Delete from tunnel */
2532   if (NULL != c->t)
2533     GMT_remove_connection (c->t, c);
2534
2535   if (GNUNET_NO == GMC_is_origin (c, GNUNET_YES) && NULL != c->path)
2536     path_destroy (c->path);
2537   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_maintenance_task)
2538     GNUNET_SCHEDULER_cancel (c->fwd_maintenance_task);
2539   if (GNUNET_SCHEDULER_NO_TASK != c->bck_maintenance_task)
2540     GNUNET_SCHEDULER_cancel (c->bck_maintenance_task);
2541   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_fc.poll_task)
2542   {
2543     GNUNET_SCHEDULER_cancel (c->fwd_fc.poll_task);
2544     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL FWD canceled\n");
2545   }
2546   if (GNUNET_SCHEDULER_NO_TASK != c->bck_fc.poll_task)
2547   {
2548     GNUNET_SCHEDULER_cancel (c->bck_fc.poll_task);
2549     LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL BCK canceled\n");
2550   }
2551
2552   GNUNET_break (GNUNET_YES ==
2553                 GNUNET_CONTAINER_multihashmap_remove (connections, &c->id, c));
2554
2555   GNUNET_STATISTICS_update (stats, "# connections", -1, GNUNET_NO);
2556   GNUNET_free (c);
2557 }
2558
2559 /**
2560  * Get the connection ID.
2561  *
2562  * @param c Connection to get the ID from.
2563  *
2564  * @return ID of the connection.
2565  */
2566 const struct GNUNET_HashCode *
2567 GMC_get_id (const struct MeshConnection *c)
2568 {
2569   return &c->id;
2570 }
2571
2572
2573 /**
2574  * Get the connection path.
2575  *
2576  * @param c Connection to get the path from.
2577  *
2578  * @return path used by the connection.
2579  */
2580 const struct MeshPeerPath *
2581 GMC_get_path (const struct MeshConnection *c)
2582 {
2583   if (GNUNET_NO == c->destroy)
2584     return c->path;
2585   return NULL;
2586 }
2587
2588
2589 /**
2590  * Get the connection state.
2591  *
2592  * @param c Connection to get the state from.
2593  *
2594  * @return state of the connection.
2595  */
2596 enum MeshConnectionState
2597 GMC_get_state (const struct MeshConnection *c)
2598 {
2599   return c->state;
2600 }
2601
2602 /**
2603  * Get the connection tunnel.
2604  *
2605  * @param c Connection to get the tunnel from.
2606  *
2607  * @return tunnel of the connection.
2608  */
2609 struct MeshTunnel3 *
2610 GMC_get_tunnel (const struct MeshConnection *c)
2611 {
2612   return c->t;
2613 }
2614
2615
2616 /**
2617  * Get free buffer space in a connection.
2618  *
2619  * @param c Connection.
2620  * @param fwd Is query about FWD traffic?
2621  *
2622  * @return Free buffer space [0 - max_msgs_queue/max_connections]
2623  */
2624 unsigned int
2625 GMC_get_buffer (struct MeshConnection *c, int fwd)
2626 {
2627   struct MeshFlowControl *fc;
2628
2629   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2630
2631   return (fc->queue_max - fc->queue_n);
2632 }
2633
2634 /**
2635  * Get how many messages have we allowed to send to us from a direction.
2636  *
2637  * @param c Connection.
2638  * @param fwd Are we asking about traffic from FWD (BCK messages)?
2639  *
2640  * @return last_ack_sent - last_pid_recv
2641  */
2642 unsigned int
2643 GMC_get_allowed (struct MeshConnection *c, int fwd)
2644 {
2645   struct MeshFlowControl *fc;
2646
2647   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2648   if (GM_is_pid_bigger(fc->last_pid_recv, fc->last_ack_sent))
2649   {
2650     return 0;
2651   }
2652   return (fc->last_ack_sent - fc->last_pid_recv);
2653 }
2654
2655 /**
2656  * Get messages queued in a connection.
2657  *
2658  * @param c Connection.
2659  * @param fwd Is query about FWD traffic?
2660  *
2661  * @return Number of messages queued.
2662  */
2663 unsigned int
2664 GMC_get_qn (struct MeshConnection *c, int fwd)
2665 {
2666   struct MeshFlowControl *fc;
2667
2668   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2669
2670   return fc->queue_n;
2671 }
2672
2673
2674 /**
2675  * Allow the connection to advertise a buffer of the given size.
2676  *
2677  * The connection will send an @c fwd ACK message (so: in direction !fwd)
2678  * allowing up to last_pid_recv + buffer.
2679  *
2680  * @param c Connection.
2681  * @param buffer How many more messages the connection can accept.
2682  * @param fwd Is this about FWD traffic? (The ack will go dest->root).
2683  */
2684 void
2685 GMC_allow (struct MeshConnection *c, unsigned int buffer, int fwd)
2686 {
2687   LOG (GNUNET_ERROR_TYPE_DEBUG, "  allowing %s %u messages %s\n",
2688        GMC_2s (c), buffer, GM_f2s (fwd));
2689   send_ack (c, buffer, fwd, GNUNET_NO);
2690 }
2691
2692
2693 /**
2694  * Notify other peers on a connection of a broken link. Mark connections
2695  * to destroy after all traffic has been sent.
2696  *
2697  * @param c Connection on which there has been a disconnection.
2698  * @param peer Peer that disconnected.
2699  */
2700 void
2701 GMC_notify_broken (struct MeshConnection *c,
2702                    struct MeshPeer *peer)
2703 {
2704   int fwd;
2705
2706   LOG (GNUNET_ERROR_TYPE_DEBUG,
2707        " notify broken on %s due to %s disconnect\n",
2708        GMC_2s (c), GMP_2s (peer));
2709
2710   fwd = peer == get_prev_hop (c);
2711
2712   if (GNUNET_YES == GMC_is_terminal (c, fwd))
2713   {
2714     /* Local shutdown, no one to notify about this. */
2715     GMC_destroy (c);
2716     return;
2717   }
2718   if (GNUNET_NO == c->destroy)
2719     send_broken (c, &my_full_id, GMP_get_id (peer), fwd);
2720
2721   /* Connection will have at least one pending message
2722    * (the one we just scheduled), so no point in checking whether to
2723    * destroy immediately. */
2724   c->destroy = GNUNET_YES;
2725   c->state = MESH_CONNECTION_DESTROYED;
2726
2727   /**
2728    * Cancel all queues, if no message is left, connection will be destroyed.
2729    */
2730   connection_cancel_queues (c, !fwd);
2731
2732   return;
2733 }
2734
2735
2736 /**
2737  * Is this peer the first one on the connection?
2738  *
2739  * @param c Connection.
2740  * @param fwd Is this about fwd traffic?
2741  *
2742  * @return #GNUNET_YES if origin, #GNUNET_NO if relay/terminal.
2743  */
2744 int
2745 GMC_is_origin (struct MeshConnection *c, int fwd)
2746 {
2747   if (!fwd && c->path->length - 1 == c->own_pos )
2748     return GNUNET_YES;
2749   if (fwd && 0 == c->own_pos)
2750     return GNUNET_YES;
2751   return GNUNET_NO;
2752 }
2753
2754
2755 /**
2756  * Is this peer the last one on the connection?
2757  *
2758  * @param c Connection.
2759  * @param fwd Is this about fwd traffic?
2760  *            Note that the ROOT is the terminal for BCK traffic!
2761  *
2762  * @return #GNUNET_YES if terminal, #GNUNET_NO if relay/origin.
2763  */
2764 int
2765 GMC_is_terminal (struct MeshConnection *c, int fwd)
2766 {
2767   return GMC_is_origin (c, !fwd);
2768 }
2769
2770
2771 /**
2772  * See if we are allowed to send by the next hop in the given direction.
2773  *
2774  * @param c Connection.
2775  * @param fwd Is this about fwd traffic?
2776  *
2777  * @return #GNUNET_YES in case it's OK to send.
2778  */
2779 int
2780 GMC_is_sendable (struct MeshConnection *c, int fwd)
2781 {
2782   struct MeshFlowControl *fc;
2783
2784   if (NULL == c)
2785   {
2786     GNUNET_break (0);
2787     return GNUNET_YES;
2788   }
2789   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2790   if (GM_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2791     return GNUNET_YES;
2792   return GNUNET_NO;
2793 }
2794
2795 /**
2796  * Sends an already built message on a connection, properly registering
2797  * all used resources.
2798  *
2799  * @param message Message to send. Function makes a copy of it.
2800  *                If message is not hop-by-hop, decrements TTL of copy.
2801  * @param c Connection on which this message is transmitted.
2802  * @param fwd Is this a fwd message?
2803  * @param force Force the connection to accept the message (buffer overfill).
2804  * @param cont Continuation called once message is sent. Can be NULL.
2805  * @param cont_cls Closure for @c cont.
2806  *
2807  * @return Handle to cancel the message before it's sent.
2808  *         NULL on error or if @c cont is NULL.
2809  *         Invalid on @c cont call.
2810  */
2811 struct MeshConnectionQueue *
2812 GMC_send_prebuilt_message (const struct GNUNET_MessageHeader *message,
2813                            struct MeshConnection *c, int fwd, int force,
2814                            GMC_sent cont, void *cont_cls)
2815 {
2816   struct MeshFlowControl *fc;
2817   struct MeshConnectionQueue *q;
2818   void *data;
2819   size_t size;
2820   uint16_t type;
2821   int droppable;
2822
2823   size = ntohs (message->size);
2824   data = GNUNET_malloc (size);
2825   memcpy (data, message, size);
2826   type = ntohs (message->type);
2827   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send %s (%u bytes) on connection %s\n",
2828        GM_m2s (type), size, GMC_2s (c));
2829
2830   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2831   droppable = GNUNET_NO == force;
2832   switch (type)
2833   {
2834     struct GNUNET_MESH_Encrypted *emsg;
2835     struct GNUNET_MESH_KX        *kmsg;
2836     struct GNUNET_MESH_ACK       *amsg;
2837     struct GNUNET_MESH_Poll      *pmsg;
2838     struct GNUNET_MESH_ConnectionDestroy *dmsg;
2839     struct GNUNET_MESH_ConnectionBroken  *bmsg;
2840     uint32_t ttl;
2841
2842     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
2843       emsg = (struct GNUNET_MESH_Encrypted *) data;
2844       ttl = ntohl (emsg->ttl);
2845       if (0 == ttl)
2846       {
2847         GNUNET_break_op (0);
2848         GNUNET_free (data);
2849         return NULL;
2850       }
2851       emsg->cid = c->id;
2852       emsg->ttl = htonl (ttl - 1);
2853       emsg->pid = htonl (fc->next_pid++);
2854       LOG (GNUNET_ERROR_TYPE_DEBUG, "  Q_N+ %p %u\n", fc, fc->queue_n);
2855       if (GNUNET_YES == droppable)
2856       {
2857         fc->queue_n++;
2858         LOG (GNUNET_ERROR_TYPE_DEBUG, "pid %u\n", ntohl (emsg->pid));
2859         LOG (GNUNET_ERROR_TYPE_DEBUG, "last pid sent %u\n", fc->last_pid_sent);
2860         LOG (GNUNET_ERROR_TYPE_DEBUG, "     ack recv %u\n", fc->last_ack_recv);
2861       }
2862       else
2863       {
2864         LOG (GNUNET_ERROR_TYPE_DEBUG, "  not droppable, Q_N stays the same\n");
2865       }
2866       if (GM_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv))
2867       {
2868         GMC_start_poll (c, fwd);
2869       }
2870       break;
2871
2872     case GNUNET_MESSAGE_TYPE_MESH_KX:
2873       kmsg = (struct GNUNET_MESH_KX *) data;
2874       kmsg->cid = c->id;
2875       break;
2876
2877     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2878       amsg = (struct GNUNET_MESH_ACK *) data;
2879       amsg->cid = c->id;
2880       LOG (GNUNET_ERROR_TYPE_DEBUG, " ack %u\n", ntohl (amsg->ack));
2881       droppable = GNUNET_NO;
2882       break;
2883
2884     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2885       pmsg = (struct GNUNET_MESH_Poll *) data;
2886       pmsg->cid = c->id;
2887       LOG (GNUNET_ERROR_TYPE_DEBUG, " poll %u\n", ntohl (pmsg->pid));
2888       droppable = GNUNET_NO;
2889       break;
2890
2891     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
2892       dmsg = (struct GNUNET_MESH_ConnectionDestroy *) data;
2893       dmsg->cid = c->id;
2894       dmsg->reserved = 0;
2895       break;
2896
2897     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
2898       bmsg = (struct GNUNET_MESH_ConnectionBroken *) data;
2899       bmsg->cid = c->id;
2900       bmsg->reserved = 0;
2901       break;
2902
2903     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
2904     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
2905     case GNUNET_MESSAGE_TYPE_MESH_KEEPALIVE:
2906       break;
2907
2908     default:
2909       GNUNET_break (0);
2910       GNUNET_free (data);
2911       return NULL;
2912   }
2913
2914   if (fc->queue_n > fc->queue_max && droppable)
2915   {
2916     GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)",
2917                               1, GNUNET_NO);
2918     GNUNET_break (0);
2919     LOG (GNUNET_ERROR_TYPE_DEBUG,
2920                 "queue full: %u/%u\n",
2921                 fc->queue_n, fc->queue_max);
2922     if (GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED == type)
2923     {
2924       fc->queue_n--;
2925       fc->next_pid--;
2926     }
2927     GNUNET_free (data);
2928     return NULL; /* Drop this message */
2929   }
2930
2931   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u\n", c, c->pending_messages);
2932   c->pending_messages++;
2933
2934   q = GNUNET_new (struct MeshConnectionQueue);
2935   q->forced = !droppable;
2936   q->q = GMP_queue_add (get_hop (c, fwd), data, type, size, c, fwd,
2937                         &message_sent, q);
2938   if (NULL == q->q)
2939   {
2940     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING dropping msg on %s\n", GMC_2s (c));
2941     GNUNET_free (data);
2942     GNUNET_free (q);
2943     return NULL;
2944   }
2945   q->cont = cont;
2946   q->cont_cls = cont_cls;
2947   return q;
2948 }
2949
2950
2951 /**
2952  * Cancel a previously sent message while it's in the queue.
2953  *
2954  * ONLY can be called before the continuation given to the send function
2955  * is called. Once the continuation is called, the message is no longer in the
2956  * queue.
2957  *
2958  * @param q Handle to the queue.
2959  */
2960 void
2961 GMC_cancel (struct MeshConnectionQueue *q)
2962 {
2963   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  GMC cancel message\n");
2964
2965   /* queue destroy calls message_sent, which calls q->cont and frees q */
2966   GMP_queue_destroy (q->q, GNUNET_YES);
2967 }
2968
2969
2970 /**
2971  * Sends a CREATE CONNECTION message for a path to a peer.
2972  * Changes the connection and tunnel states if necessary.
2973  *
2974  * @param connection Connection to create.
2975  */
2976 void
2977 GMC_send_create (struct MeshConnection *connection)
2978 {
2979   enum MeshTunnel3CState state;
2980   size_t size;
2981
2982   size = sizeof (struct GNUNET_MESH_ConnectionCreate);
2983   size += connection->path->length * sizeof (struct GNUNET_PeerIdentity);
2984
2985   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send connection create\n");
2986   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u (create)\n",
2987        connection, connection->pending_messages);
2988   connection->pending_messages++;
2989
2990   connection->maintenance_q =
2991     GMP_queue_add (get_next_hop (connection), NULL,
2992                    GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
2993                    size, connection, GNUNET_YES, &message_sent, NULL);
2994
2995   state = GMT_get_cstate (connection->t);
2996   if (MESH_TUNNEL3_SEARCHING == state || MESH_TUNNEL3_NEW == state)
2997     GMT_change_cstate (connection->t, MESH_TUNNEL3_WAITING);
2998   if (MESH_CONNECTION_NEW == connection->state)
2999     connection_change_state (connection, MESH_CONNECTION_SENT);
3000 }
3001
3002
3003 /**
3004  * Send a message to all peers in this connection that the connection
3005  * is no longer valid.
3006  *
3007  * If some peer should not receive the message, it should be zero'ed out
3008  * before calling this function.
3009  *
3010  * @param c The connection whose peers to notify.
3011  */
3012 void
3013 GMC_send_destroy (struct MeshConnection *c)
3014 {
3015   struct GNUNET_MESH_ConnectionDestroy msg;
3016
3017   if (GNUNET_YES == c->destroy)
3018     return;
3019
3020   msg.header.size = htons (sizeof (msg));
3021   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY);;
3022   msg.cid = c->id;
3023   LOG (GNUNET_ERROR_TYPE_DEBUG,
3024               "  sending connection destroy for connection %s\n",
3025               GMC_2s (c));
3026
3027   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_YES))
3028     GMC_send_prebuilt_message (&msg.header, c,
3029                                GNUNET_YES, GNUNET_YES, NULL, NULL);
3030   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_NO))
3031     GMC_send_prebuilt_message (&msg.header, c,
3032                                GNUNET_NO, GNUNET_YES, NULL, NULL);
3033   c->destroy = GNUNET_YES;
3034   c->state = MESH_CONNECTION_DESTROYED;
3035 }
3036
3037
3038 /**
3039  * @brief Start a polling timer for the connection.
3040  *
3041  * When a neighbor does not accept more traffic on the connection it could be
3042  * caused by a simple congestion or by a lost ACK. Polling enables to check
3043  * for the lastest ACK status for a connection.
3044  *
3045  * @param c Connection.
3046  * @param fwd Should we poll in the FWD direction?
3047  */
3048 void
3049 GMC_start_poll (struct MeshConnection *c, int fwd)
3050 {
3051   struct MeshFlowControl *fc;
3052
3053   fc = fwd ? &c->fwd_fc : &c->bck_fc;
3054   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL %s requested\n",
3055        GM_f2s (fwd));
3056   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task || NULL != fc->poll_msg)
3057   {
3058     LOG (GNUNET_ERROR_TYPE_DEBUG, " ***   not needed (%u, %p)\n",
3059          fc->poll_task, fc->poll_msg);
3060     return;
3061   }
3062   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** POLL started on request\n");
3063   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
3064                                                 &connection_poll,
3065                                                 fc);
3066 }
3067
3068
3069 /**
3070  * @brief Stop polling a connection for ACKs.
3071  *
3072  * Once we have enough ACKs for future traffic, polls are no longer necessary.
3073  *
3074  * @param c Connection.
3075  * @param fwd Should we stop the poll in the FWD direction?
3076  */
3077 void
3078 GMC_stop_poll (struct MeshConnection *c, int fwd)
3079 {
3080   struct MeshFlowControl *fc;
3081
3082   fc = fwd ? &c->fwd_fc : &c->bck_fc;
3083   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
3084   {
3085     GNUNET_SCHEDULER_cancel (fc->poll_task);
3086     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
3087   }
3088 }
3089
3090 /**
3091  * Get a (static) string for a connection.
3092  *
3093  * @param c Connection.
3094  */
3095 const char *
3096 GMC_2s (const struct MeshConnection *c)
3097 {
3098   if (NULL == c)
3099     return "NULL";
3100
3101   if (NULL != c->t)
3102   {
3103     static char buf[128];
3104
3105     sprintf (buf, "%s (->%s)", GNUNET_h2s (&c->id), GMT_2s (c->t));
3106     return buf;
3107   }
3108   return GNUNET_h2s (&c->id);
3109 }