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_enc.h"
  34 #include "mesh_enc.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 /**
 110  * Keep a record of the last messages sent on this connection.
 111  */
 112 struct MeshConnectionPerformance
 113 {
 114   /**
 115    * Circular buffer for storing measurements.
 116    */
 117   double usecsperbyte[AVG_MSGS];
 118
 119   /**
 120    * Running average of @c usecsperbyte.
 121    */
 122   double avg;
 123
 124   /**
 125    * How many values of @c usecsperbyte are valid.
 126    */
 127   uint16_t size;
 128
 129   /**
 130    * Index of the next "free" position in @c usecsperbyte.
 131    */
 132   uint16_t idx;
 133 };
 134
 135
 136 /**
 137  * Struct containing all information regarding a connection to a peer.
 138  */
 139 struct MeshConnection
 140 {
 141   /**
 142    * Tunnel this connection is part of.
 143    */
 144   struct MeshTunnel3 *t;
 145
 146   /**
 147    * Flow control information for traffic fwd.
 148    */
 149   struct MeshFlowControl fwd_fc;
 150
 151   /**
 152    * Flow control information for traffic bck.
 153    */
 154   struct MeshFlowControl bck_fc;
 155
 156   /**
 157    * Measure connection performance on the endpoint.
 158    */
 159   struct MeshConnectionPerformance *perf;
 160
 161   /**
 162    * ID of the connection.
 163    */
 164   struct GNUNET_HashCode id;
 165
 166   /**
 167    * State of the connection.
 168    */
 169   enum MeshConnectionState state;
 170
 171   /**
 172    * Path being used for the tunnel.
 173    */
 174   struct MeshPeerPath *path;
 175
 176   /**
 177    * Position of the local peer in the path.
 178    */
 179   unsigned int own_pos;
 180
 181   /**
 182    * Task to keep the used paths alive at the owner,
 183    * time tunnel out on all the other peers.
 184    */
 185   GNUNET_SCHEDULER_TaskIdentifier fwd_maintenance_task;
 186
 187   /**
 188    * Task to keep the used paths alive at the destination,
 189    * time tunnel out on all the other peers.
 190    */
 191   GNUNET_SCHEDULER_TaskIdentifier bck_maintenance_task;
 192
 193   /**
 194    * Pending message count.
 195    */
 196   int pending_messages;
 197
 198   /**
 199    * Destroy flag: if true, destroy on last message.
 200    */
 201   int destroy;
 202 };
 203
 204 /******************************************************************************/
 205 /*******************************   GLOBALS  ***********************************/
 206 /******************************************************************************/
 207
 208 /**
 209  * Global handle to the statistics service.
 210  */
 211 extern struct GNUNET_STATISTICS_Handle *stats;
 212
 213 /**
 214  * Local peer own ID (memory efficient handle).
 215  */
 216 extern GNUNET_PEER_Id myid;
 217
 218 /**
 219  * Local peer own ID (full value).
 220  */
 221 extern struct GNUNET_PeerIdentity my_full_id;
 222
 223 /**
 224  * Connections known, indexed by cid (MeshConnection).
 225  */
 226 static struct GNUNET_CONTAINER_MultiHashMap *connections;
 227
 228 /**
 229  * How many connections are we willing to maintain.
 230  * Local connections are always allowed, even if there are more connections than max.
 231  */
 232 static unsigned long long max_connections;
 233
 234 /**
 235  * How many messages *in total* are we willing to queue, divide by number of
 236  * connections to get connection queue size.
 237  */
 238 static unsigned long long max_msgs_queue;
 239
 240 /**
 241  * How often to send path keepalives. Paths timeout after 4 missed.
 242  */
 243 static struct GNUNET_TIME_Relative refresh_connection_time;
 244
 245
 246 /******************************************************************************/
 247 /********************************   STATIC  ***********************************/
 248 /******************************************************************************/
 249
 250 #if 0 // avoid compiler warning for unused static function
 251 static void
 252 fc_debug (struct MeshFlowControl *fc)
 253 {
 254   LOG (GNUNET_ERROR_TYPE_DEBUG, "    IN: %u/%u\n",
 255               fc->last_pid_recv, fc->last_ack_sent);
 256   LOG (GNUNET_ERROR_TYPE_DEBUG, "    OUT: %u/%u\n",
 257               fc->last_pid_sent, fc->last_ack_recv);
 258   LOG (GNUNET_ERROR_TYPE_DEBUG, "    QUEUE: %u/%u\n",
 259               fc->queue_n, fc->queue_max);
 260 }
 261
 262 static void
 263 connection_debug (struct MeshConnection *c)
 264 {
 265   if (NULL == c)
 266   {
 267     LOG (GNUNET_ERROR_TYPE_DEBUG, "*** DEBUG NULL CONNECTION ***\n");
 268     return;
 269   }
 270   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connection %s:%X\n",
 271               peer2s (c->t->peer), GMC_2s (c));
 272   LOG (GNUNET_ERROR_TYPE_DEBUG, "  state: %u, pending msgs: %u\n",
 273               c->state, c->pending_messages);
 274   LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD FC\n");
 275   fc_debug (&c->fwd_fc);
 276   LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK FC\n");
 277   fc_debug (&c->bck_fc);
 278 }
 279 #endif
 280
 281 /**
 282  * Get string description for tunnel state.
 283  *
 284  * @param s Tunnel state.
 285  *
 286  * @return String representation.
 287  */
 288 static const char *
 289 GMC_state2s (enum MeshConnectionState s)
 290 {
 291   switch (s)
 292   {
 293     case MESH_CONNECTION_NEW:
 294       return "MESH_CONNECTION_NEW";
 295     case MESH_CONNECTION_SENT:
 296       return "MESH_CONNECTION_SENT";
 297     case MESH_CONNECTION_ACK:
 298       return "MESH_CONNECTION_ACK";
 299     case MESH_CONNECTION_READY:
 300       return "MESH_CONNECTION_READY";
 301     default:
 302       return "MESH_CONNECTION_STATE_ERROR";
 303   }
 304 }
 305
 306
 307 /**
 308  * Initialize a Flow Control structure to the initial state.
 309  *
 310  * @param fc Flow Control structure to initialize.
 311  */
 312 static void
 313 fc_init (struct MeshFlowControl *fc)
 314 {
 315   fc->next_pid = 0;
 316   fc->last_pid_sent = (uint32_t) -1; /* Next (expected) = 0 */
 317   fc->last_pid_recv = (uint32_t) -1;
 318   fc->last_ack_sent = (uint32_t) 0;
 319   fc->last_ack_recv = (uint32_t) 0;
 320   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
 321   fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
 322   fc->queue_n = 0;
 323   fc->queue_max = (max_msgs_queue / max_connections) + 1;
 324 }
 325
 326
 327 /**
 328  * Find a connection.
 329  *
 330  * @param cid Connection ID.
 331  */
 332 static struct MeshConnection *
 333 connection_get (const struct GNUNET_HashCode *cid)
 334 {
 335   return GNUNET_CONTAINER_multihashmap_get (connections, cid);
 336 }
 337
 338
 339 static void
 340 connection_change_state (struct MeshConnection* c,
 341                          enum MeshConnectionState state)
 342 {
 343   LOG (GNUNET_ERROR_TYPE_DEBUG,
 344               "Connection %s state was %s\n",
 345               GMC_2s (c), GMC_state2s (c->state));
 346   LOG (GNUNET_ERROR_TYPE_DEBUG,
 347               "Connection %s state is now %s\n",
 348               GMC_2s (c), GMC_state2s (state));
 349   c->state = state;
 350 }
 351
 352
 353 /**
 354  * Send an ACK on the connection, informing the predecessor about
 355  * the available buffer space. Should not be called in case the peer
 356  * is origin (no predecessor).
 357  *
 358  * Note that for fwd ack, the FWD mean forward *traffic* (root->dest),
 359  * the ACK itself goes "back" (dest->root).
 360  *
 361  * @param c Connection on which to send the ACK.
 362  * @param buffer How much space free to advertise?
 363  * @param fwd Is this FWD ACK? (Going dest->owner)
 364  */
 365 static void
 366 send_ack (struct MeshConnection *c, unsigned int buffer, int fwd)
 367 {
 368   struct MeshFlowControl *next_fc;
 369   struct MeshFlowControl *prev_fc;
 370   struct GNUNET_MESH_ACK msg;
 371   uint32_t ack;
 372   int delta;
 373
 374   /* If origin, there is no connection to send ACKs. Wrong function! */
 375   if (GMC_is_origin (c, fwd))
 376   {
 377     GNUNET_break (0);
 378     return;
 379   }
 380
 381   next_fc = fwd ? &c->fwd_fc : &c->bck_fc;
 382   prev_fc = fwd ? &c->bck_fc : &c->fwd_fc;
 383
 384   LOG (GNUNET_ERROR_TYPE_DEBUG,
 385               "connection send %s ack on %s\n",
 386               fwd ? "FWD" : "BCK", GMC_2s (c));
 387
 388   /* Check if we need to transmit the ACK */
 389   delta = prev_fc->last_ack_sent - prev_fc->last_pid_recv;
 390   if (3 < delta && buffer < delta)
 391   {
 392     LOG (GNUNET_ERROR_TYPE_DEBUG, "Not sending ACK, buffer > 3\n");
 393     LOG (GNUNET_ERROR_TYPE_DEBUG,
 394          "  last pid recv: %u, last ack sent: %u\n",
 395          prev_fc->last_pid_recv, prev_fc->last_ack_sent);
 396     return;
 397   }
 398
 399   /* Ok, ACK might be necessary, what PID to ACK? */
 400   ack = prev_fc->last_pid_recv + buffer;
 401   LOG (GNUNET_ERROR_TYPE_DEBUG, " ACK %u\n", ack);
 402   LOG (GNUNET_ERROR_TYPE_DEBUG,
 403        " last pid %u, last ack %u, qmax %u, q %u\n",
 404        prev_fc->last_pid_recv, prev_fc->last_ack_sent,
 405        next_fc->queue_max, next_fc->queue_n);
 406   if (ack == prev_fc->last_ack_sent)
 407   {
 408     LOG (GNUNET_ERROR_TYPE_DEBUG, "Not sending FWD ACK, not needed\n");
 409     return;
 410   }
 411
 412   prev_fc->last_ack_sent = ack;
 413
 414   /* Build ACK message and send on connection */
 415   msg.header.size = htons (sizeof (msg));
 416   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_ACK);
 417   msg.ack = htonl (ack);
 418   msg.cid = c->id;
 419
 420   GMC_send_prebuilt_message (&msg.header, c, !fwd);
 421 }
 422
 423
 424 /**
 425  * Callback called when a queued message is sent.
 426  *
 427  * Calculates the average time and connection packet tracking.
 428  *
 429  * @param cls Closure.
 430  * @param c Connection this message was on.
 431  * @param type Type of message sent.
 432  * @param fwd Was this a FWD going message?
 433  * @param size Size of the message.
 434  * @param wait Time spent waiting for core (only the time for THIS message)
 435  */
 436 static void
 437 message_sent (void *cls,
 438               struct MeshConnection *c, uint16_t type,
 439               int fwd, size_t size,
 440               struct GNUNET_TIME_Relative wait)
 441 {
 442   struct MeshConnectionPerformance *p;
 443   struct MeshFlowControl *fc;
 444   double usecsperbyte;
 445
 446   fc = fwd ? &c->fwd_fc : &c->bck_fc;
 447   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  sent %s\n", GNUNET_MESH_DEBUG_M2S (type));
 448   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  Q_N- %p %u\n", fc, fc->queue_n);
 449   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  C_P- %p %u\n", c, c->pending_messages);
 450   c->pending_messages--;
 451   if (GNUNET_YES == c->destroy && 0 == c->pending_messages)
 452   {
 453     LOG (GNUNET_ERROR_TYPE_DEBUG, "!  destroying connection!\n");
 454     GMC_destroy (c);
 455     return;
 456   }
 457   /* Send ACK if needed, after accounting for sent ID in fc->queue_n */
 458   switch (type)
 459   {
 460     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
 461       fc->last_pid_sent++;
 462       fc->queue_n--;
 463       LOG (GNUNET_ERROR_TYPE_DEBUG,
 464            "!   accounting pid %u\n",
 465            fc->last_pid_sent);
 466       GMC_send_ack (c, fwd);
 467       break;
 468     default:
 469       break;
 470   }
 471   LOG (GNUNET_ERROR_TYPE_DEBUG, "!  message sent!\n");
 472
 473   if (NULL == c->perf)
 474     return; /* Only endpoints are interested in timing. */
 475
 476   p = c->perf;
 477   usecsperbyte = ((double) wait.rel_value_us) / size;
 478   if (p->size == AVG_MSGS)
 479   {
 480     /* Array is full. Substract oldest value, add new one and store. */
 481     p->avg -= (p->usecsperbyte[p->idx] / AVG_MSGS);
 482     p->usecsperbyte[p->idx] = usecsperbyte;
 483     p->avg += (p->usecsperbyte[p->idx] / AVG_MSGS);
 484   }
 485   else
 486   {
 487     /* Array not yet full. Add current value to avg and store. */
 488     p->usecsperbyte[p->idx] = usecsperbyte;
 489     p->avg *= p->size;
 490     p->avg += p->usecsperbyte[p->idx];
 491     p->size++;
 492     p->avg /= p->size;
 493   }
 494   p->idx = (p->idx + 1) % AVG_MSGS;
 495 }
 496
 497
 498 /**
 499  * Get the previous hop in a connection
 500  *
 501  * @param c Connection.
 502  *
 503  * @return Previous peer in the connection.
 504  */
 505 static struct MeshPeer *
 506 get_prev_hop (const struct MeshConnection *c)
 507 {
 508   GNUNET_PEER_Id id;
 509
 510   LOG (GNUNET_ERROR_TYPE_DEBUG, "Get prev hop, own pos %u\n", c->own_pos);
 511   if (0 == c->own_pos || c->path->length < 2)
 512     id = c->path->peers[0];
 513   else
 514     id = c->path->peers[c->own_pos - 1];
 515
 516   return GMP_get_short (id);
 517 }
 518
 519
 520 /**
 521  * Get the next hop in a connection
 522  *
 523  * @param c Connection.
 524  *
 525  * @return Next peer in the connection.
 526  */
 527 static struct MeshPeer *
 528 get_next_hop (const struct MeshConnection *c)
 529 {
 530   GNUNET_PEER_Id id;
 531
 532   if ((c->path->length - 1) == c->own_pos || c->path->length < 2)
 533     id = c->path->peers[c->path->length - 1];
 534   else
 535     id = c->path->peers[c->own_pos + 1];
 536
 537   return GMP_get_short (id);
 538 }
 539
 540
 541 /**
 542  * Get the hop in a connection.
 543  *
 544  * @param c Connection.
 545  * @param fwd Next hop?
 546  *
 547  * @return Next peer in the connection.
 548  */
 549 static struct MeshPeer *
 550 get_hop (struct MeshConnection *c, int fwd)
 551 {
 552   if (fwd)
 553     return get_next_hop (c);
 554   return get_prev_hop (c);
 555 }
 556
 557
 558 /**
 559  * Is traffic coming from this sender 'FWD' traffic?
 560  *
 561  * @param c Connection to check.
 562  * @param sender Peer identity of neighbor.
 563  *
 564  * @return #GNUNET_YES in case the sender is the 'prev' hop and therefore
 565  *         the traffic is 'FWD'.
 566  *         #GNUNET_NO for BCK.
 567  *         #GNUNET_SYSERR for errors.
 568  */
 569 static int
 570 is_fwd (const struct MeshConnection *c,
 571         const struct GNUNET_PeerIdentity *sender)
 572 {
 573   GNUNET_PEER_Id id;
 574
 575   id = GNUNET_PEER_search (sender);
 576   if (GMP_get_short_id (get_prev_hop (c)) == id)
 577     return GNUNET_YES;
 578
 579   if (GMP_get_short_id (get_next_hop (c)) == id)
 580     return GNUNET_NO;
 581
 582   GNUNET_break (0);
 583   return GNUNET_SYSERR;
 584 }
 585
 586
 587 /**
 588  * Sends a CONNECTION ACK message in reponse to a received CONNECTION_CREATE
 589  * or a first CONNECTION_ACK directed to us.
 590  *
 591  * @param connection Connection to confirm.
 592  * @param fwd Should we send it FWD? (root->dest)
 593  *            (First (~SYNACK) goes BCK, second (~ACK) goes FWD)
 594  */
 595 static void
 596 send_connection_ack (struct MeshConnection *connection, int fwd)
 597 {
 598   struct MeshTunnel3 *t;
 599
 600   t = connection->t;
 601   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send connection %s ACK\n",
 602        !fwd ? "FWD" : "BCK");
 603   GMP_queue_add (get_hop (connection, fwd), NULL,
 604                  GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK,
 605                  sizeof (struct GNUNET_MESH_ConnectionACK),
 606                  connection, fwd, &message_sent, NULL);
 607   if (MESH_TUNNEL3_NEW == GMT_get_state (t))
 608     GMT_change_state (t, MESH_TUNNEL3_WAITING);
 609   if (MESH_CONNECTION_READY != connection->state)
 610     connection_change_state (connection, MESH_CONNECTION_SENT);
 611 }
 612
 613
 614 /**
 615  * Send a notification that a connection is broken.
 616  *
 617  * @param c Connection that is broken.
 618  * @param id1 Peer that has disconnected.
 619  * @param id2 Peer that has disconnected.
 620  * @param fwd Direction towards which to send it.
 621  */
 622 static void
 623 send_broken (struct MeshConnection *c,
 624              const struct GNUNET_PeerIdentity *id1,
 625              const struct GNUNET_PeerIdentity *id2,
 626              int fwd)
 627 {
 628   struct GNUNET_MESH_ConnectionBroken msg;
 629
 630   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectionBroken));
 631   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN);
 632   msg.cid = c->id;
 633   msg.peer1 = *id1;
 634   msg.peer2 = *id2;
 635   GMC_send_prebuilt_message (&msg.header, c, fwd);
 636 }
 637
 638
 639
 640 /**
 641  * Send keepalive packets for a connection.
 642  *
 643  * @param c Connection to keep alive..
 644  * @param fwd Is this a FWD keepalive? (owner -> dest).
 645  */
 646 static void
 647 connection_keepalive (struct MeshConnection *c, int fwd)
 648 {
 649   struct GNUNET_MESH_ConnectionKeepAlive *msg;
 650   size_t size = sizeof (struct GNUNET_MESH_ConnectionKeepAlive);
 651   char cbuf[size];
 652   uint16_t type;
 653
 654   type = fwd ? GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE :
 655                GNUNET_MESSAGE_TYPE_MESH_BCK_KEEPALIVE;
 656
 657   LOG (GNUNET_ERROR_TYPE_DEBUG,
 658        "sending %s keepalive for connection %s]\n",
 659        fwd ? "FWD" : "BCK", GMC_2s (c));
 660
 661   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) cbuf;
 662   msg->header.size = htons (size);
 663   msg->header.type = htons (type);
 664   msg->cid = c->id;
 665
 666   GMC_send_prebuilt_message (&msg->header, c, fwd);
 667 }
 668
 669
 670 /**
 671  * Send CONNECTION_{CREATE/ACK} packets for a connection.
 672  *
 673  * @param c Connection for which to send the message.
 674  * @param fwd If #GNUNET_YES, send CREATE, otherwise send ACK.
 675  */
 676 static void
 677 connection_recreate (struct MeshConnection *c, int fwd)
 678 {
 679   LOG (GNUNET_ERROR_TYPE_DEBUG, "sending connection recreate\n");
 680   if (fwd)
 681     GMC_send_create (c);
 682   else
 683     send_connection_ack (c, GNUNET_NO);
 684 }
 685
 686
 687 /**
 688  * Generic connection timer management.
 689  * Depending on the role of the peer in the connection will send the
 690  * appropriate message (build or keepalive)
 691  *
 692  * @param c Conncetion to maintain.
 693  * @param fwd Is FWD?
 694  */
 695 static void
 696 connection_maintain (struct MeshConnection *c, int fwd)
 697 {
 698   if (MESH_TUNNEL3_SEARCHING == GMT_get_state (c->t))
 699   {
 700     /* TODO DHT GET with RO_BART */
 701     return;
 702   }
 703   switch (c->state)
 704   {
 705     case MESH_CONNECTION_NEW:
 706       GNUNET_break (0);
 707     case MESH_CONNECTION_SENT:
 708       connection_recreate (c, fwd);
 709       break;
 710     case MESH_CONNECTION_READY:
 711       connection_keepalive (c, fwd);
 712       break;
 713     default:
 714       break;
 715   }
 716 }
 717
 718
 719 static void
 720 connection_fwd_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 721 {
 722   struct MeshConnection *c = cls;
 723
 724   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 725   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 726     return;
 727
 728   connection_maintain (c, GNUNET_YES);
 729   c->fwd_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
 730                                                           &connection_fwd_keepalive,
 731                                                           c);
 732 }
 733
 734
 735 static void
 736 connection_bck_keepalive (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 737 {
 738   struct MeshConnection *c = cls;
 739
 740   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 741   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 742     return;
 743
 744   connection_maintain (c, GNUNET_NO);
 745   c->bck_maintenance_task = GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
 746                                                           &connection_bck_keepalive,
 747                                                           c);
 748 }
 749
 750
 751 /**
 752  * @brief Re-initiate traffic on this connection if necessary.
 753  *
 754  * Check if there is traffic queued towards this peer
 755  * and the core transmit handle is NULL (traffic was stalled).
 756  * If so, call core tmt rdy.
 757  *
 758  * @param c Connection on which initiate traffic.
 759  * @param fwd Is this about fwd traffic?
 760  */
 761 static void
 762 connection_unlock_queue (struct MeshConnection *c, int fwd)
 763 {
 764   struct MeshPeer *peer;
 765
 766   LOG (GNUNET_ERROR_TYPE_DEBUG,
 767               "connection_unlock_queue %s on %s\n",
 768               fwd ? "FWD" : "BCK", GMC_2s (c));
 769
 770   if (GMC_is_terminal (c, fwd))
 771   {
 772     LOG (GNUNET_ERROR_TYPE_DEBUG, " is terminal!\n");
 773     return;
 774   }
 775
 776   peer = get_hop (c, fwd);
 777   GMP_queue_unlock (peer, c);
 778 }
 779
 780
 781 /**
 782  * Cancel all transmissions that belong to a certain connection.
 783  *
 784  * If the connection is scheduled for destruction and no more messages are left,
 785  * the connection will be destroyed by the continuation call.
 786  *
 787  * @param c Connection which to cancel. Might be destroyed during this call.
 788  * @param fwd Cancel fwd traffic?
 789  */
 790 static void
 791 connection_cancel_queues (struct MeshConnection *c, int fwd)
 792 {
 793   struct MeshFlowControl *fc;
 794   struct MeshPeer *peer;
 795
 796   if (NULL == c)
 797   {
 798     GNUNET_break (0);
 799     return;
 800   }
 801
 802   fc = fwd ? &c->fwd_fc : &c->bck_fc;
 803   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
 804   {
 805     GNUNET_SCHEDULER_cancel (fc->poll_task);
 806     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
 807   }
 808   peer = get_hop (c, fwd);
 809   GMP_queue_cancel (peer, c);
 810 }
 811
 812
 813 /**
 814  * Function called if a connection has been stalled for a while,
 815  * possibly due to a missed ACK. Poll the neighbor about its ACK status.
 816  *
 817  * @param cls Closure (poll ctx).
 818  * @param tc TaskContext.
 819  */
 820 static void
 821 connection_poll (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 822 {
 823   struct MeshFlowControl *fc = cls;
 824   struct GNUNET_MESH_Poll msg;
 825   struct MeshConnection *c;
 826
 827   fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
 828   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 829   {
 830     return;
 831   }
 832
 833   c = fc->c;
 834   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** Polling!\n");
 835   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** connection [%s]\n", GMC_2s (c));
 836   LOG (GNUNET_ERROR_TYPE_DEBUG, " ***   %s\n",
 837               fc == &c->fwd_fc ? "FWD" : "BCK");
 838
 839   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_POLL);
 840   msg.header.size = htons (sizeof (msg));
 841   LOG (GNUNET_ERROR_TYPE_DEBUG, " *** pid (%u)!\n", fc->last_pid_sent);
 842   GMC_send_prebuilt_message (&msg.header, c, fc == &c->fwd_fc);
 843   fc->poll_time = GNUNET_TIME_STD_BACKOFF (fc->poll_time);
 844   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
 845                                                 &connection_poll, fc);
 846 }
 847
 848
 849 /**
 850  * Timeout function due to lack of keepalive/traffic from the owner.
 851  * Destroys connection if called.
 852  *
 853  * @param cls Closure (connection to destroy).
 854  * @param tc TaskContext.
 855  */
 856 static void
 857 connection_fwd_timeout (void *cls,
 858                         const struct GNUNET_SCHEDULER_TaskContext *tc)
 859 {
 860   struct MeshConnection *c = cls;
 861
 862   c->fwd_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 863   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 864     return;
 865   LOG (GNUNET_ERROR_TYPE_DEBUG,
 866               "Connection %s[%X] FWD timed out. Destroying.\n",
 867               GMT_2s (c->t),
 868               c->id);
 869
 870   if (GMC_is_origin (c, GNUNET_YES)) /* If local, leave. */
 871     return;
 872
 873   GMC_destroy (c);
 874 }
 875
 876
 877 /**
 878  * Timeout function due to lack of keepalive/traffic from the destination.
 879  * Destroys connection if called.
 880  *
 881  * @param cls Closure (connection to destroy).
 882  * @param tc TaskContext
 883  */
 884 static void
 885 connection_bck_timeout (void *cls,
 886                         const struct GNUNET_SCHEDULER_TaskContext *tc)
 887 {
 888   struct MeshConnection *c = cls;
 889
 890   c->bck_maintenance_task = GNUNET_SCHEDULER_NO_TASK;
 891   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 892     return;
 893
 894   LOG (GNUNET_ERROR_TYPE_DEBUG,
 895               "Connection %s[%X] FWD timed out. Destroying.\n",
 896               GMT_2s (c->t), c->id);
 897
 898   if (GMC_is_origin (c, GNUNET_NO)) /* If local, leave. */
 899     return;
 900
 901   GMC_destroy (c);
 902 }
 903
 904
 905 /**
 906  * Resets the connection timeout task, some other message has done the
 907  * task's job.
 908  * - For the first peer on the direction this means to send
 909  *   a keepalive or a path confirmation message (either create or ACK).
 910  * - For all other peers, this means to destroy the connection,
 911  *   due to lack of activity.
 912  * Starts the tiemout if no timeout was running (connection just created).
 913  *
 914  * @param c Connection whose timeout to reset.
 915  * @param fwd Is this forward?
 916  *
 917  * TODO use heap to improve efficiency of scheduler.
 918  */
 919 static void
 920 connection_reset_timeout (struct MeshConnection *c, int fwd)
 921 {
 922   GNUNET_SCHEDULER_TaskIdentifier *ti;
 923   GNUNET_SCHEDULER_Task f;
 924
 925   ti = fwd ? &c->fwd_maintenance_task : &c->bck_maintenance_task;
 926
 927   if (GNUNET_SCHEDULER_NO_TASK != *ti)
 928     GNUNET_SCHEDULER_cancel (*ti);
 929
 930   if (GMC_is_origin (c, fwd)) /* Endpoint */
 931   {
 932     f  = fwd ? &connection_fwd_keepalive : &connection_bck_keepalive;
 933     *ti = GNUNET_SCHEDULER_add_delayed (refresh_connection_time, f, c);
 934   }
 935   else /* Relay */
 936   {
 937     struct GNUNET_TIME_Relative delay;
 938
 939     delay = GNUNET_TIME_relative_multiply (refresh_connection_time, 4);
 940     f  = fwd ? &connection_fwd_timeout : &connection_bck_timeout;
 941     *ti = GNUNET_SCHEDULER_add_delayed (delay, f, c);
 942   }
 943 }
 944
 945
 946 /**
 947  * Add the connection to the list of both neighbors.
 948  *
 949  * @param c Connection.
 950  */
 951 static void
 952 register_neighbors (struct MeshConnection *c)
 953 {
 954   struct MeshPeer *peer;
 955
 956   peer = get_next_hop (c);
 957   if (GNUNET_NO == GMP_is_neighbor (peer))
 958   {
 959     GMC_destroy (c);
 960     return;
 961   }
 962   GMP_add_connection (peer, c);
 963   peer = get_prev_hop (c);
 964   if (GNUNET_NO == GMP_is_neighbor (peer))
 965   {
 966     GMC_destroy (c);
 967     return;
 968   }
 969   GMP_add_connection (peer, c);
 970 }
 971
 972
 973 /**
 974  * Remove the connection from the list of both neighbors.
 975  *
 976  * @param c Connection.
 977  */
 978 static void
 979 unregister_neighbors (struct MeshConnection *c)
 980 {
 981   struct MeshPeer *peer;
 982
 983   peer = get_next_hop (c);
 984   GMP_remove_connection (peer, c);
 985
 986   peer = get_prev_hop (c);
 987   GMP_remove_connection (peer, c);
 988
 989 }
 990
 991
 992 /**
 993  * Bind the connection to the peer and the tunnel to that peer.
 994  *
 995  * If the peer has no tunnel, create one. Update tunnel and connection
 996  * data structres to reflect new status.
 997  *
 998  * @param c Connection.
 999  * @param peer Peer.
1000  */
1001 static void
1002 add_to_peer (struct MeshConnection *c, struct MeshPeer *peer)
1003 {
1004   GMP_add_tunnel (peer);
1005   c->t = GMP_get_tunnel (peer);
1006   GMT_add_connection (c->t, c);
1007 }
1008
1009 /******************************************************************************/
1010 /********************************    API    ***********************************/
1011 /******************************************************************************/
1012
1013 /**
1014  * Core handler for connection creation.
1015  *
1016  * @param cls Closure (unused).
1017  * @param peer Sender (neighbor).
1018  * @param message Message.
1019  *
1020  * @return GNUNET_OK to keep the connection open,
1021  *         GNUNET_SYSERR to close it (signal serious error)
1022  */
1023 int
1024 GMC_handle_create (void *cls, const struct GNUNET_PeerIdentity *peer,
1025                    const struct GNUNET_MessageHeader *message)
1026 {
1027   struct GNUNET_MESH_ConnectionCreate *msg;
1028   struct GNUNET_PeerIdentity *id;
1029   struct GNUNET_HashCode *cid;
1030   struct MeshPeerPath *path;
1031   struct MeshPeer *dest_peer;
1032   struct MeshPeer *orig_peer;
1033   struct MeshConnection *c;
1034   unsigned int own_pos;
1035   uint16_t size;
1036   uint16_t i;
1037
1038   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1039   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection create msg\n");
1040
1041   /* Check size */
1042   size = ntohs (message->size);
1043   if (size < sizeof (struct GNUNET_MESH_ConnectionCreate))
1044   {
1045     GNUNET_break_op (0);
1046     return GNUNET_OK;
1047   }
1048
1049   /* Calculate hops */
1050   size -= sizeof (struct GNUNET_MESH_ConnectionCreate);
1051   if (size % sizeof (struct GNUNET_PeerIdentity))
1052   {
1053     GNUNET_break_op (0);
1054     return GNUNET_OK;
1055   }
1056   size /= sizeof (struct GNUNET_PeerIdentity);
1057   if (1 > size)
1058   {
1059     GNUNET_break_op (0);
1060     return GNUNET_OK;
1061   }
1062   LOG (GNUNET_ERROR_TYPE_DEBUG, "    path has %u hops.\n", size);
1063
1064   /* Get parameters */
1065   msg = (struct GNUNET_MESH_ConnectionCreate *) message;
1066   cid = &msg->cid;
1067   id = (struct GNUNET_PeerIdentity *) &msg[1];
1068   LOG (GNUNET_ERROR_TYPE_DEBUG,
1069               "    connection %s (%s).\n",
1070               GNUNET_h2s (cid), GNUNET_i2s (id));
1071
1072   /* Create connection */
1073   c = connection_get (cid);
1074   if (NULL == c)
1075   {
1076     /* Create path */
1077     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating path...\n");
1078     path = path_new (size);
1079     own_pos = 0;
1080     for (i = 0; i < size; i++)
1081     {
1082       LOG (GNUNET_ERROR_TYPE_DEBUG, "  ... adding %s\n",
1083                   GNUNET_i2s (&id[i]));
1084       path->peers[i] = GNUNET_PEER_intern (&id[i]);
1085       if (path->peers[i] == myid)
1086         own_pos = i;
1087     }
1088     if (own_pos == 0 && path->peers[own_pos] != myid)
1089     {
1090       /* create path: self not found in path through self */
1091       GNUNET_break_op (0);
1092       path_destroy (path);
1093       return GNUNET_OK;
1094     }
1095     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Own position: %u\n", own_pos);
1096     GMP_add_path_to_all (path, GNUNET_NO);
1097         LOG (GNUNET_ERROR_TYPE_DEBUG, "  Creating connection\n");
1098     c = GMC_new (cid, NULL, path_duplicate (path), own_pos);
1099     if (NULL == c)
1100       return GNUNET_OK;
1101     connection_reset_timeout (c, GNUNET_YES);
1102   }
1103   else
1104   {
1105     path = NULL;
1106   }
1107   if (MESH_CONNECTION_NEW == c->state)
1108     connection_change_state (c, MESH_CONNECTION_SENT);
1109
1110   /* Remember peers */
1111   dest_peer = GMP_get (&id[size - 1]);
1112   orig_peer = GMP_get (&id[0]);
1113
1114   /* Is it a connection to us? */
1115   if (c->own_pos == size - 1)
1116   {
1117     LOG (GNUNET_ERROR_TYPE_DEBUG, "  It's for us!\n");
1118     GMP_add_path_to_origin (orig_peer, path, GNUNET_YES);
1119
1120     add_to_peer (c, orig_peer);
1121     if (MESH_TUNNEL3_NEW == GMT_get_state (c->t))
1122       GMT_change_state (c->t,  MESH_TUNNEL3_WAITING);
1123
1124     send_connection_ack (c, GNUNET_NO);
1125     if (MESH_CONNECTION_SENT == c->state)
1126       connection_change_state (c, MESH_CONNECTION_ACK);
1127
1128     /* Keep tunnel alive in direction dest->owner*/
1129     connection_reset_timeout (c, GNUNET_NO);
1130   }
1131   else
1132   {
1133     /* It's for somebody else! Retransmit. */
1134     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Retransmitting.\n");
1135     GMP_add_path (dest_peer, path_duplicate (path), GNUNET_NO);
1136     GMP_add_path_to_origin (orig_peer, path, GNUNET_NO);
1137     GMC_send_prebuilt_message (message, c, GNUNET_YES);
1138   }
1139   return GNUNET_OK;
1140 }
1141
1142
1143 /**
1144  * Core handler for path confirmations.
1145  *
1146  * @param cls closure
1147  * @param message message
1148  * @param peer peer identity this notification is about
1149  *
1150  * @return GNUNET_OK to keep the connection open,
1151  *         GNUNET_SYSERR to close it (signal serious error)
1152  */
1153 int
1154 GMC_handle_confirm (void *cls, const struct GNUNET_PeerIdentity *peer,
1155                     const struct GNUNET_MessageHeader *message)
1156 {
1157   struct GNUNET_MESH_ConnectionACK *msg;
1158   struct MeshConnection *c;
1159   struct MeshPeerPath *p;
1160   struct MeshPeer *pi;
1161   int fwd;
1162
1163   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1164   LOG (GNUNET_ERROR_TYPE_DEBUG, "Received a connection ACK msg\n");
1165   msg = (struct GNUNET_MESH_ConnectionACK *) message;
1166   LOG (GNUNET_ERROR_TYPE_DEBUG, "  on connection %s\n",
1167               GNUNET_h2s (&msg->cid));
1168   c = connection_get (&msg->cid);
1169   if (NULL == c)
1170   {
1171     GNUNET_STATISTICS_update (stats, "# control on unknown connection",
1172                               1, GNUNET_NO);
1173     LOG (GNUNET_ERROR_TYPE_DEBUG, "  don't know the connection!\n");
1174     return GNUNET_OK;
1175   }
1176
1177
1178   LOG (GNUNET_ERROR_TYPE_DEBUG, "  via peer %s\n",
1179               GNUNET_i2s (peer));
1180   pi = GMP_get (peer);
1181   if (get_next_hop (c) == pi)
1182   {
1183     LOG (GNUNET_ERROR_TYPE_DEBUG, "  SYNACK\n");
1184     fwd = GNUNET_NO;
1185     if (MESH_CONNECTION_SENT == c->state)
1186       connection_change_state (c, MESH_CONNECTION_ACK);
1187   }
1188   else if (get_prev_hop (c) == pi)
1189   {
1190     LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK\n");
1191     fwd = GNUNET_YES;
1192     connection_change_state (c, MESH_CONNECTION_READY);
1193   }
1194   else
1195   {
1196     GNUNET_break_op (0);
1197     return GNUNET_OK;
1198   }
1199   connection_reset_timeout (c, fwd);
1200
1201   /* Add path to peers? */
1202   p = c->path;
1203   if (NULL != p)
1204   {
1205     GMP_add_path_to_all (p, GNUNET_YES);
1206   }
1207   else
1208   {
1209     GNUNET_break (0);
1210   }
1211
1212   /* Message for us as creator? */
1213   if (GMC_is_origin (c, GNUNET_YES))
1214   {
1215     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection (SYN)ACK for us!\n");
1216     connection_change_state (c, MESH_CONNECTION_READY);
1217     GMT_change_state (c->t, MESH_TUNNEL3_READY);
1218     send_connection_ack (c, GNUNET_YES);
1219     return GNUNET_OK;
1220   }
1221
1222   /* Message for us as destination? */
1223   if (GMC_is_terminal (c, GNUNET_YES))
1224   {
1225     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Connection ACK for us!\n");
1226     connection_change_state (c, MESH_CONNECTION_READY);
1227     GMT_change_state (c->t, MESH_TUNNEL3_READY);
1228     return GNUNET_OK;
1229   }
1230
1231   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1232   GMC_send_prebuilt_message (message, c, fwd);
1233   return GNUNET_OK;
1234 }
1235
1236
1237 /**
1238  * Core handler for notifications of broken paths
1239  *
1240  * @param cls Closure (unused).
1241  * @param id Peer identity of sending neighbor.
1242  * @param message Message.
1243  *
1244  * @return GNUNET_OK to keep the connection open,
1245  *         GNUNET_SYSERR to close it (signal serious error)
1246  */
1247 int
1248 GMC_handle_broken (void* cls,
1249                    const struct GNUNET_PeerIdentity* id,
1250                    const struct GNUNET_MessageHeader* message)
1251 {
1252   struct GNUNET_MESH_ConnectionBroken *msg;
1253   struct MeshConnection *c;
1254   int fwd;
1255
1256   LOG (GNUNET_ERROR_TYPE_DEBUG,
1257               "Received a CONNECTION BROKEN msg from %s\n", GNUNET_i2s (id));
1258   msg = (struct GNUNET_MESH_ConnectionBroken *) message;
1259   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1260               GNUNET_i2s (&msg->peer1));
1261   LOG (GNUNET_ERROR_TYPE_DEBUG, "  regarding %s\n",
1262               GNUNET_i2s (&msg->peer2));
1263   c = connection_get (&msg->cid);
1264   if (NULL == c)
1265   {
1266     GNUNET_break_op (0);
1267     return GNUNET_OK;
1268   }
1269
1270   fwd = is_fwd (c, id);
1271   connection_cancel_queues (c, !fwd);
1272   if (GMC_is_terminal (c, fwd))
1273   {
1274     if (0 < c->pending_messages)
1275       c->destroy = GNUNET_YES;
1276     else
1277       GMC_destroy (c);
1278   }
1279   else
1280   {
1281     GMC_send_prebuilt_message (message, c, fwd);
1282     c->destroy = GNUNET_YES;
1283   }
1284
1285   return GNUNET_OK;
1286
1287 }
1288
1289
1290 /**
1291  * Core handler for tunnel destruction
1292  *
1293  * @param cls Closure (unused).
1294  * @param peer Peer identity of sending neighbor.
1295  * @param message Message.
1296  *
1297  * @return GNUNET_OK to keep the connection open,
1298  *         GNUNET_SYSERR to close it (signal serious error)
1299  */
1300 int
1301 GMC_handle_destroy (void *cls, const struct GNUNET_PeerIdentity *peer,
1302                     const struct GNUNET_MessageHeader *message)
1303 {
1304   struct GNUNET_MESH_ConnectionDestroy *msg;
1305   struct MeshConnection *c;
1306   int fwd;
1307
1308   msg = (struct GNUNET_MESH_ConnectionDestroy *) message;
1309   LOG (GNUNET_ERROR_TYPE_DEBUG,
1310               "Got a CONNECTION DESTROY message from %s\n",
1311               GNUNET_i2s (peer));
1312   LOG (GNUNET_ERROR_TYPE_DEBUG,
1313               "  for connection %s\n",
1314               GNUNET_h2s (&msg->cid));
1315   c = connection_get (&msg->cid);
1316   if (NULL == c)
1317   {
1318     /* Probably already got the message from another path,
1319      * destroyed the tunnel and retransmitted to children.
1320      * Safe to ignore.
1321      */
1322     GNUNET_STATISTICS_update (stats, "# control on unknown tunnel",
1323                               1, GNUNET_NO);
1324     return GNUNET_OK;
1325   }
1326   fwd = is_fwd (c, peer);
1327   if (GNUNET_SYSERR == fwd)
1328   {
1329     GNUNET_break_op (0);
1330     return GNUNET_OK;
1331   }
1332   GMC_send_prebuilt_message (message, c, fwd);
1333   c->destroy = GNUNET_YES;
1334
1335   return GNUNET_OK;
1336 }
1337
1338 /**
1339  * Generic handler for mesh network encrypted traffic.
1340  *
1341  * @param peer Peer identity this notification is about.
1342  * @param msg Encrypted message.
1343  *
1344  * @return GNUNET_OK to keep the connection open,
1345  *         GNUNET_SYSERR to close it (signal serious error)
1346  */
1347 static int
1348 handle_mesh_encrypted (const struct GNUNET_PeerIdentity *peer,
1349                        const struct GNUNET_MESH_Encrypted *msg)
1350 {
1351   struct MeshConnection *c;
1352   struct MeshPeer *neighbor;
1353   struct MeshFlowControl *fc;
1354   GNUNET_PEER_Id peer_id;
1355   uint32_t pid;
1356   uint32_t ttl;
1357   uint16_t type;
1358   size_t size;
1359   int fwd;
1360
1361   /* Check size */
1362   size = ntohs (msg->header.size);
1363   if (size <
1364       sizeof (struct GNUNET_MESH_Encrypted) +
1365       sizeof (struct GNUNET_MessageHeader))
1366   {
1367     GNUNET_break_op (0);
1368     return GNUNET_OK;
1369   }
1370   type = ntohs (msg->header.type);
1371   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1372   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
1373               GNUNET_MESH_DEBUG_M2S (type), GNUNET_i2s (peer));
1374
1375   /* Check connection */
1376   c = connection_get (&msg->cid);
1377   if (NULL == c)
1378   {
1379     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1380     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING connection unknown\n");
1381     return GNUNET_OK;
1382   }
1383
1384   /* Check if origin is as expected */
1385   neighbor = get_prev_hop (c);
1386   peer_id = GNUNET_PEER_search (peer);
1387   if (peer_id == GMP_get_short_id (neighbor))
1388   {
1389     fwd = GNUNET_YES;
1390   }
1391   else
1392   {
1393     neighbor = get_next_hop (c);
1394     if (peer_id == GMP_get_short_id (neighbor))
1395     {
1396       fwd = GNUNET_NO;
1397     }
1398     else
1399     {
1400       /* Unexpected peer sending traffic on a connection. */
1401       GNUNET_break_op (0);
1402       return GNUNET_OK;
1403     }
1404   }
1405
1406   /* Check PID */
1407   fc = fwd ? &c->bck_fc : &c->fwd_fc;
1408   pid = ntohl (msg->pid);
1409   if (GMC_is_pid_bigger (pid, fc->last_ack_sent))
1410   {
1411     GNUNET_STATISTICS_update (stats, "# unsolicited message", 1, GNUNET_NO);
1412     LOG (GNUNET_ERROR_TYPE_DEBUG,
1413                 "WARNING Received PID %u, (prev %u), ACK %u\n",
1414                 pid, fc->last_pid_recv, fc->last_ack_sent);
1415     return GNUNET_OK;
1416   }
1417   if (GNUNET_NO == GMC_is_pid_bigger (pid, fc->last_pid_recv))
1418   {
1419     GNUNET_STATISTICS_update (stats, "# duplicate PID", 1, GNUNET_NO);
1420     LOG (GNUNET_ERROR_TYPE_DEBUG,
1421                 " Pid %u not expected (%u+), dropping!\n",
1422                 pid, fc->last_pid_recv + 1);
1423     return GNUNET_OK;
1424   }
1425   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
1426     connection_change_state (c, MESH_CONNECTION_READY);
1427   connection_reset_timeout (c, fwd);
1428   fc->last_pid_recv = pid;
1429
1430   /* Is this message for us? */
1431   if (GMC_is_terminal (c, fwd))
1432   {
1433     /* TODO signature verification */
1434     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
1435     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
1436
1437     if (NULL == c->t)
1438     {
1439       GNUNET_break (0);
1440       return GNUNET_OK;
1441     }
1442     fc->last_pid_recv = pid;
1443     GMT_handle_encrypted (c->t, msg);
1444     GMC_send_ack (c, fwd);
1445     return GNUNET_OK;
1446   }
1447
1448   /* Message not for us: forward to next hop */
1449   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1450   ttl = ntohl (msg->ttl);
1451   LOG (GNUNET_ERROR_TYPE_DEBUG, "   ttl: %u\n", ttl);
1452   if (ttl == 0)
1453   {
1454     GNUNET_STATISTICS_update (stats, "# TTL drops", 1, GNUNET_NO);
1455     LOG (GNUNET_ERROR_TYPE_WARNING, " TTL is 0, DROPPING!\n");
1456     GMC_send_ack (c, fwd);
1457     return GNUNET_OK;
1458   }
1459   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
1460
1461   GMC_send_prebuilt_message (&msg->header, c, fwd);
1462
1463   return GNUNET_OK;
1464 }
1465
1466 /**
1467  * Generic handler for mesh network encrypted traffic.
1468  *
1469  * @param peer Peer identity this notification is about.
1470  * @param msg Encrypted message.
1471  *
1472  * @return GNUNET_OK to keep the connection open,
1473  *         GNUNET_SYSERR to close it (signal serious error)
1474  */
1475 static int
1476 handle_mesh_kx (const struct GNUNET_PeerIdentity *peer,
1477                 const struct GNUNET_MESH_KX *msg)
1478 {
1479   struct MeshConnection *c;
1480   struct MeshPeer *neighbor;
1481   GNUNET_PEER_Id peer_id;
1482   size_t size;
1483   uint16_t type;
1484   int fwd;
1485
1486   /* Check size */
1487   size = ntohs (msg->header.size);
1488   if (size <
1489       sizeof (struct GNUNET_MESH_Encrypted) +
1490       sizeof (struct GNUNET_MessageHeader))
1491   {
1492     GNUNET_break_op (0);
1493     return GNUNET_OK;
1494   }
1495   type = ntohs (msg->header.type);
1496   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1497   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a %s message from %s\n",
1498               GNUNET_MESH_DEBUG_M2S (type), GNUNET_i2s (peer));
1499
1500   /* Check connection */
1501   c = connection_get (&msg->cid);
1502   if (NULL == c)
1503   {
1504     GNUNET_STATISTICS_update (stats, "# unknown connection", 1, GNUNET_NO);
1505     LOG (GNUNET_ERROR_TYPE_DEBUG, "WARNING connection unknown\n");
1506     return GNUNET_OK;
1507   }
1508
1509   /* Check if origin is as expected */
1510   neighbor = get_prev_hop (c);
1511   peer_id = GNUNET_PEER_search (peer);
1512   if (peer_id == GMP_get_short_id (neighbor))
1513   {
1514     fwd = GNUNET_YES;
1515   }
1516   else
1517   {
1518     neighbor = get_next_hop (c);
1519     if (peer_id == GMP_get_short_id (neighbor))
1520     {
1521       fwd = GNUNET_NO;
1522     }
1523     else
1524     {
1525       /* Unexpected peer sending traffic on a connection. */
1526       GNUNET_break_op (0);
1527       return GNUNET_OK;
1528     }
1529   }
1530
1531   /* Count as connection confirmation. */
1532   if (MESH_CONNECTION_SENT == c->state || MESH_CONNECTION_ACK == c->state)
1533     connection_change_state (c, MESH_CONNECTION_READY);
1534   connection_reset_timeout (c, fwd);
1535   if (NULL != c->t)
1536   {
1537     if (MESH_TUNNEL3_WAITING == GMT_get_state (c->t))
1538       GMT_change_state (c->t, MESH_TUNNEL3_READY);
1539   }
1540
1541   /* Is this message for us? */
1542   if (GMC_is_terminal (c, fwd))
1543   {
1544     LOG (GNUNET_ERROR_TYPE_DEBUG, "  message for us!\n");
1545     GNUNET_STATISTICS_update (stats, "# messages received", 1, GNUNET_NO);
1546     if (NULL == c->t)
1547     {
1548       GNUNET_break (0);
1549       return GNUNET_OK;
1550     }
1551     GMT_handle_kx (c->t, &msg[1].header);
1552     return GNUNET_OK;
1553   }
1554
1555   /* Message not for us: forward to next hop */
1556   LOG (GNUNET_ERROR_TYPE_DEBUG, "  not for us, retransmitting...\n");
1557   GNUNET_STATISTICS_update (stats, "# messages forwarded", 1, GNUNET_NO);
1558
1559   GMC_send_prebuilt_message (&msg->header, c, fwd);
1560
1561   return GNUNET_OK;
1562 }
1563
1564
1565 /**
1566  * Core handler for encrypted mesh network traffic (channel mgmt, data).
1567  *
1568  * @param cls Closure (unused).
1569  * @param message Message received.
1570  * @param peer Peer who sent the message.
1571  *
1572  * @return GNUNET_OK to keep the connection open,
1573  *         GNUNET_SYSERR to close it (signal serious error)
1574  */
1575 int
1576 GMC_handle_encrypted (void *cls, const struct GNUNET_PeerIdentity *peer,
1577                       const struct GNUNET_MessageHeader *message)
1578 {
1579   return handle_mesh_encrypted (peer,
1580                                 (struct GNUNET_MESH_Encrypted *)message);
1581 }
1582
1583
1584 /**
1585  * Core handler for key exchange traffic (ephemeral key, ping, pong).
1586  *
1587  * @param cls Closure (unused).
1588  * @param message Message received.
1589  * @param peer Peer who sent the message.
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_kx (void *cls, const struct GNUNET_PeerIdentity *peer,
1596                const struct GNUNET_MessageHeader *message)
1597 {
1598   return handle_mesh_kx (peer,
1599                          (struct GNUNET_MESH_KX *) message);
1600 }
1601
1602
1603 /**
1604  * Core handler for mesh network traffic point-to-point acks.
1605  *
1606  * @param cls closure
1607  * @param message message
1608  * @param peer peer identity this notification is about
1609  *
1610  * @return GNUNET_OK to keep the connection open,
1611  *         GNUNET_SYSERR to close it (signal serious error)
1612  */
1613 int
1614 GMC_handle_ack (void *cls, const struct GNUNET_PeerIdentity *peer,
1615                 const struct GNUNET_MessageHeader *message)
1616 {
1617   struct GNUNET_MESH_ACK *msg;
1618   struct MeshConnection *c;
1619   struct MeshFlowControl *fc;
1620   GNUNET_PEER_Id id;
1621   uint32_t ack;
1622   int fwd;
1623
1624   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1625   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got an ACK packet from %s!\n",
1626               GNUNET_i2s (peer));
1627   msg = (struct GNUNET_MESH_ACK *) message;
1628
1629   c = connection_get (&msg->cid);
1630
1631   if (NULL == c)
1632   {
1633     GNUNET_STATISTICS_update (stats, "# ack on unknown connection", 1,
1634                               GNUNET_NO);
1635     return GNUNET_OK;
1636   }
1637
1638   /* Is this a forward or backward ACK? */
1639   id = GNUNET_PEER_search (peer);
1640   if (GMP_get_short_id (get_next_hop (c)) == id)
1641   {
1642     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
1643     fc = &c->fwd_fc;
1644     fwd = GNUNET_YES;
1645   }
1646   else if (GMP_get_short_id (get_prev_hop (c)) == id)
1647   {
1648     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
1649     fc = &c->bck_fc;
1650     fwd = GNUNET_NO;
1651   }
1652   else
1653   {
1654     GNUNET_break_op (0);
1655     return GNUNET_OK;
1656   }
1657
1658   ack = ntohl (msg->ack);
1659   LOG (GNUNET_ERROR_TYPE_DEBUG, "  ACK %u (was %u)\n",
1660               ack, fc->last_ack_recv);
1661   if (GMC_is_pid_bigger (ack, fc->last_ack_recv))
1662     fc->last_ack_recv = ack;
1663
1664   /* Cancel polling if the ACK is big enough. */
1665   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task &&
1666       GMC_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
1667   {
1668     LOG (GNUNET_ERROR_TYPE_DEBUG, "  Cancel poll\n");
1669     GNUNET_SCHEDULER_cancel (fc->poll_task);
1670     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
1671     fc->poll_time = GNUNET_TIME_UNIT_SECONDS;
1672   }
1673
1674   connection_unlock_queue (c, fwd);
1675
1676   return GNUNET_OK;
1677 }
1678
1679
1680 /**
1681  * Core handler for mesh network traffic point-to-point ack polls.
1682  *
1683  * @param cls closure
1684  * @param message message
1685  * @param peer peer identity this notification is about
1686  *
1687  * @return GNUNET_OK to keep the connection open,
1688  *         GNUNET_SYSERR to close it (signal serious error)
1689  */
1690 int
1691 GMC_handle_poll (void *cls, const struct GNUNET_PeerIdentity *peer,
1692                  const struct GNUNET_MessageHeader *message)
1693 {
1694   struct GNUNET_MESH_Poll *msg;
1695   struct MeshConnection *c;
1696   struct MeshFlowControl *fc;
1697   GNUNET_PEER_Id id;
1698   uint32_t pid;
1699   int fwd;
1700
1701   LOG (GNUNET_ERROR_TYPE_DEBUG, "\n\n");
1702   LOG (GNUNET_ERROR_TYPE_DEBUG, "Got a POLL packet from %s!\n",
1703               GNUNET_i2s (peer));
1704
1705   msg = (struct GNUNET_MESH_Poll *) message;
1706
1707   c = connection_get (&msg->cid);
1708
1709   if (NULL == c)
1710   {
1711     GNUNET_STATISTICS_update (stats, "# poll on unknown connection", 1,
1712                               GNUNET_NO);
1713     GNUNET_break_op (0);
1714     return GNUNET_OK;
1715   }
1716
1717   /* Is this a forward or backward ACK?
1718    * Note: a poll should never be needed in a loopback case,
1719    * since there is no possiblility of packet loss there, so
1720    * this way of discerining FWD/BCK should not be a problem.
1721    */
1722   id = GNUNET_PEER_search (peer);
1723   if (GMP_get_short_id (get_next_hop (c)) == id)
1724   {
1725     LOG (GNUNET_ERROR_TYPE_DEBUG, "  FWD ACK\n");
1726     fc = &c->fwd_fc;
1727   }
1728   else if (GMP_get_short_id (get_prev_hop (c)) == id)
1729   {
1730     LOG (GNUNET_ERROR_TYPE_DEBUG, "  BCK ACK\n");
1731     fc = &c->bck_fc;
1732   }
1733   else
1734   {
1735     GNUNET_break_op (0);
1736     return GNUNET_OK;
1737   }
1738
1739   pid = ntohl (msg->pid);
1740   LOG (GNUNET_ERROR_TYPE_DEBUG, "  PID %u, OLD %u\n",
1741               pid, fc->last_pid_recv);
1742   fc->last_pid_recv = pid;
1743   fwd = fc == &c->fwd_fc;
1744   GMC_send_ack (c, fwd);
1745
1746   return GNUNET_OK;
1747 }
1748
1749
1750 /**
1751  * Core handler for mesh keepalives.
1752  *
1753  * @param cls closure
1754  * @param message message
1755  * @param peer peer identity this notification is about
1756  * @return GNUNET_OK to keep the connection open,
1757  *         GNUNET_SYSERR to close it (signal serious error)
1758  *
1759  * TODO: Check who we got this from, to validate route.
1760  */
1761 int
1762 GMC_handle_keepalive (void *cls, const struct GNUNET_PeerIdentity *peer,
1763                       const struct GNUNET_MessageHeader *message)
1764 {
1765   struct GNUNET_MESH_ConnectionKeepAlive *msg;
1766   struct MeshConnection *c;
1767   struct MeshPeer *neighbor;
1768   int fwd;
1769
1770   msg = (struct GNUNET_MESH_ConnectionKeepAlive *) message;
1771   LOG (GNUNET_ERROR_TYPE_DEBUG, "got a keepalive packet from %s\n",
1772               GNUNET_i2s (peer));
1773
1774   c = connection_get (&msg->cid);
1775   if (NULL == c)
1776   {
1777     GNUNET_STATISTICS_update (stats, "# keepalive on unknown connection", 1,
1778                               GNUNET_NO);
1779     return GNUNET_OK;
1780   }
1781
1782   fwd = GNUNET_MESSAGE_TYPE_MESH_FWD_KEEPALIVE == ntohs (message->type) ?
1783         GNUNET_YES : GNUNET_NO;
1784
1785   /* Check if origin is as expected */
1786   neighbor = get_hop (c, fwd);
1787   if (GNUNET_PEER_search (peer) != GMP_get_short_id (neighbor))
1788   {
1789     GNUNET_break_op (0);
1790     return GNUNET_OK;
1791   }
1792
1793   connection_change_state (c, MESH_CONNECTION_READY);
1794   connection_reset_timeout (c, fwd);
1795
1796   if (GMC_is_terminal (c, fwd))
1797     return GNUNET_OK;
1798
1799   GNUNET_STATISTICS_update (stats, "# keepalives forwarded", 1, GNUNET_NO);
1800   GMC_send_prebuilt_message (message, c, fwd);
1801
1802   return GNUNET_OK;
1803 }
1804
1805
1806 /**
1807  * Send an ACK on the appropriate connection/channel, depending on
1808  * the direction and the position of the peer.
1809  *
1810  * @param c Which connection to send the hop-by-hop ACK.
1811  * @param fwd Is this a fwd ACK? (will go dest->root)
1812  */
1813 void
1814 GMC_send_ack (struct MeshConnection *c, int fwd)
1815 {
1816   unsigned int buffer;
1817
1818   LOG (GNUNET_ERROR_TYPE_DEBUG,
1819        "GMC send %s ACK on %s\n",
1820        fwd ? "FWD" : "BCK", GMC_2s (c));
1821
1822   if (NULL == c)
1823   {
1824     GNUNET_break (0);
1825     return;
1826   }
1827
1828   /* Get available buffer space */
1829   if (GMC_is_terminal (c, fwd))
1830   {
1831     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from all channels\n");
1832     buffer = GMT_get_channels_buffer (c->t);
1833   }
1834   else
1835   {
1836     LOG (GNUNET_ERROR_TYPE_DEBUG, "  getting from one connection\n");
1837     buffer = GMC_get_buffer (c, fwd);
1838   }
1839   LOG (GNUNET_ERROR_TYPE_DEBUG, "  buffer available: %u\n", buffer);
1840
1841   /* Send available buffer space */
1842   if (GMC_is_origin (c, fwd))
1843   {
1844     GNUNET_assert (NULL != c->t);
1845     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on channels...\n");
1846     if (0 < buffer)
1847     {
1848       LOG (GNUNET_ERROR_TYPE_DEBUG, "  really sending!\n");
1849       GMT_unchoke_channels (c->t);
1850     }
1851   }
1852   else
1853   {
1854     LOG (GNUNET_ERROR_TYPE_DEBUG, "  sending on connection\n");
1855     send_ack (c, buffer, fwd);
1856   }
1857 }
1858
1859
1860 /**
1861  * Initialize the connections subsystem
1862  *
1863  * @param c Configuration handle.
1864  */
1865 void
1866 GMC_init (const struct GNUNET_CONFIGURATION_Handle *c)
1867 {
1868   LOG (GNUNET_ERROR_TYPE_DEBUG, "init\n");
1869   if (GNUNET_OK !=
1870       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_MSGS_QUEUE",
1871                                              &max_msgs_queue))
1872   {
1873     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
1874                                "MESH", "MAX_MSGS_QUEUE", "MISSING");
1875     GNUNET_SCHEDULER_shutdown ();
1876     return;
1877   }
1878
1879   if (GNUNET_OK !=
1880       GNUNET_CONFIGURATION_get_value_number (c, "MESH", "MAX_CONNECTIONS",
1881                                              &max_connections))
1882   {
1883     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
1884                                "MESH", "MAX_CONNECTIONS", "MISSING");
1885     GNUNET_SCHEDULER_shutdown ();
1886     return;
1887   }
1888
1889   if (GNUNET_OK !=
1890       GNUNET_CONFIGURATION_get_value_time (c, "MESH", "REFRESH_CONNECTION_TIME",
1891                                            &refresh_connection_time))
1892   {
1893     GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR,
1894                                "MESH", "REFRESH_CONNECTION_TIME", "MISSING");
1895     GNUNET_SCHEDULER_shutdown ();
1896     return;
1897   }
1898   connections = GNUNET_CONTAINER_multihashmap_create (1024, GNUNET_YES);
1899 }
1900
1901 /**
1902  * Shut down the connections subsystem.
1903  */
1904 void
1905 GMC_shutdown (void)
1906 {
1907   GNUNET_CONTAINER_multihashmap_destroy (connections);
1908 }
1909
1910
1911 struct MeshConnection *
1912 GMC_new (const struct GNUNET_HashCode *cid,
1913          struct MeshTunnel3 *t,
1914          struct MeshPeerPath *p,
1915          unsigned int own_pos)
1916 {
1917   struct MeshConnection *c;
1918
1919   c = GNUNET_new (struct MeshConnection);
1920   c->id = *cid;
1921   GNUNET_CONTAINER_multihashmap_put (connections, &c->id, c,
1922                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST);
1923   fc_init (&c->fwd_fc);
1924   fc_init (&c->bck_fc);
1925   c->fwd_fc.c = c;
1926   c->bck_fc.c = c;
1927
1928   c->t = t;
1929   if (own_pos > p->length - 1)
1930   {
1931     GNUNET_break (0);
1932     GMC_destroy (c);
1933     return NULL;
1934   }
1935   c->own_pos = own_pos;
1936   c->path = p;
1937
1938   if (0 == own_pos)
1939   {
1940     c->fwd_maintenance_task =
1941             GNUNET_SCHEDULER_add_delayed (refresh_connection_time,
1942                                           &connection_fwd_keepalive, c);
1943   }
1944   register_neighbors (c);
1945   return c;
1946 }
1947
1948
1949 void
1950 GMC_destroy (struct MeshConnection *c)
1951 {
1952   if (NULL == c)
1953     return;
1954
1955   if (2 == c->destroy) /* cancel queues -> GMP_queue_cancel -> q_destroy -> */
1956     return;            /* -> message_sent -> GMC_destroy. Don't loop. */
1957   c->destroy = 2;
1958
1959   LOG (GNUNET_ERROR_TYPE_DEBUG, "destroying connection %s\n", GMC_2s (c));
1960
1961   /* Cancel all traffic */
1962   connection_cancel_queues (c, GNUNET_YES);
1963   connection_cancel_queues (c, GNUNET_NO);
1964
1965   /* Cancel maintainance task (keepalive/timeout) */
1966   if (GNUNET_SCHEDULER_NO_TASK != c->fwd_maintenance_task)
1967     GNUNET_SCHEDULER_cancel (c->fwd_maintenance_task);
1968   if (GNUNET_SCHEDULER_NO_TASK != c->bck_maintenance_task)
1969     GNUNET_SCHEDULER_cancel (c->bck_maintenance_task);
1970
1971   /* Unregister from neighbors */
1972   unregister_neighbors (c);
1973
1974   /* Delete */
1975   GNUNET_STATISTICS_update (stats, "# connections", -1, GNUNET_NO);
1976   if (NULL != c->t)
1977     GMT_remove_connection (c->t, c);
1978
1979   if (GNUNET_NO == GMC_is_origin (c, GNUNET_YES))
1980     path_destroy (c->path);
1981
1982   GNUNET_free (c);
1983 }
1984
1985 /**
1986  * Get the connection ID.
1987  *
1988  * @param c Connection to get the ID from.
1989  *
1990  * @return ID of the connection.
1991  */
1992 const struct GNUNET_HashCode *
1993 GMC_get_id (const struct MeshConnection *c)
1994 {
1995   return &c->id;
1996 }
1997
1998
1999 /**
2000  * Get the connection path.
2001  *
2002  * @param c Connection to get the path from.
2003  *
2004  * @return path used by the connection.
2005  */
2006 const struct MeshPeerPath *
2007 GMC_get_path (const struct MeshConnection *c)
2008 {
2009   return c->path;
2010 }
2011
2012
2013 /**
2014  * Get the connection state.
2015  *
2016  * @param c Connection to get the state from.
2017  *
2018  * @return state of the connection.
2019  */
2020 enum MeshConnectionState
2021 GMC_get_state (const struct MeshConnection *c)
2022 {
2023   return c->state;
2024 }
2025
2026 /**
2027  * Get the connection tunnel.
2028  *
2029  * @param c Connection to get the tunnel from.
2030  *
2031  * @return tunnel of the connection.
2032  */
2033 struct MeshTunnel3 *
2034 GMC_get_tunnel (const struct MeshConnection *c)
2035 {
2036   return c->t;
2037 }
2038
2039
2040 /**
2041  * Get free buffer space in a connection.
2042  *
2043  * @param c Connection.
2044  * @param fwd Is query about FWD traffic?
2045  *
2046  * @return Free buffer space [0 - max_msgs_queue/max_connections]
2047  */
2048 unsigned int
2049 GMC_get_buffer (struct MeshConnection *c, int fwd)
2050 {
2051   struct MeshFlowControl *fc;
2052
2053   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2054
2055   return (fc->queue_max - fc->queue_n);
2056 }
2057
2058 /**
2059  * Get how many messages have we allowed to send to us from a direction.
2060  *
2061  * @param c Connection.
2062  * @param fwd Are we asking about traffic from FWD (BCK messages)?
2063  *
2064  * @return last_ack_sent - last_pid_recv
2065  */
2066 unsigned int
2067 GMC_get_allowed (struct MeshConnection *c, int fwd)
2068 {
2069   struct MeshFlowControl *fc;
2070
2071   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2072   if (GMC_is_pid_bigger(fc->last_pid_recv, fc->last_ack_sent))
2073   {
2074     return 0;
2075   }
2076   return (fc->last_ack_sent - fc->last_pid_recv);
2077 }
2078
2079 /**
2080  * Get messages queued in a connection.
2081  *
2082  * @param c Connection.
2083  * @param fwd Is query about FWD traffic?
2084  *
2085  * @return Number of messages queued.
2086  */
2087 unsigned int
2088 GMC_get_qn (struct MeshConnection *c, int fwd)
2089 {
2090   struct MeshFlowControl *fc;
2091
2092   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2093
2094   return fc->queue_n;
2095 }
2096
2097
2098 /**
2099  * Allow the connection to advertise a buffer of the given size.
2100  *
2101  * The connection will send an @c fwd ACK message (so: in direction !fwd)
2102  * allowing up to last_pid_recv + buffer.
2103  *
2104  * @param c Connection.
2105  * @param buffer How many more messages the connection can accept.
2106  * @param fwd Is this about FWD traffic? (The ack will go dest->root).
2107  */
2108 void
2109 GMC_allow (struct MeshConnection *c, unsigned int buffer, int fwd)
2110 {
2111   send_ack (c, buffer, fwd);
2112 }
2113
2114
2115 /**
2116  * Notify other peers on a connection of a broken link. Mark connections
2117  * to destroy after all traffic has been sent.
2118  *
2119  * @param c Connection on which there has been a disconnection.
2120  * @param peer Peer that disconnected.
2121  */
2122 void
2123 GMC_notify_broken (struct MeshConnection *c,
2124                    struct MeshPeer *peer)
2125 {
2126   int fwd;
2127
2128   fwd = peer == get_prev_hop (c);
2129
2130   if (GNUNET_YES == GMC_is_terminal (c, fwd))
2131   {
2132     /* Local shutdown, no one to notify about this. */
2133     GMC_destroy (c);
2134     return;
2135   }
2136   if (GNUNET_NO == c->destroy)
2137     send_broken (c, &my_full_id, GMP_get_id (peer), fwd);
2138
2139   /* Connection will have at least one pending message
2140    * (the one we just scheduled), so no point in checking whether to
2141    * destroy immediately. */
2142   c->destroy = GNUNET_YES;
2143
2144   /**
2145    * Cancel all queues, if no message is left, connection will be destroyed.
2146    */
2147   connection_cancel_queues (c, !fwd);
2148
2149   return;
2150 }
2151
2152
2153 /**
2154  * Is this peer the first one on the connection?
2155  *
2156  * @param c Connection.
2157  * @param fwd Is this about fwd traffic?
2158  *
2159  * @return #GNUNET_YES if origin, #GNUNET_NO if relay/terminal.
2160  */
2161 int
2162 GMC_is_origin (struct MeshConnection *c, int fwd)
2163 {
2164   if (!fwd && c->path->length - 1 == c->own_pos )
2165     return GNUNET_YES;
2166   if (fwd && 0 == c->own_pos)
2167     return GNUNET_YES;
2168   return GNUNET_NO;
2169 }
2170
2171
2172 /**
2173  * Is this peer the last one on the connection?
2174  *
2175  * @param c Connection.
2176  * @param fwd Is this about fwd traffic?
2177  *            Note that the ROOT is the terminal for BCK traffic!
2178  *
2179  * @return #GNUNET_YES if terminal, #GNUNET_NO if relay/origin.
2180  */
2181 int
2182 GMC_is_terminal (struct MeshConnection *c, int fwd)
2183 {
2184   return GMC_is_origin (c, !fwd);
2185 }
2186
2187
2188 /**
2189  * See if we are allowed to send by the next hop in the given direction.
2190  *
2191  * @param c Connection.
2192  * @param fwd Is this about fwd traffic?
2193  *
2194  * @return #GNUNET_YES in case it's OK to send.
2195  */
2196 int
2197 GMC_is_sendable (struct MeshConnection *c, int fwd)
2198 {
2199   struct MeshFlowControl *fc;
2200
2201   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2202   if (GMC_is_pid_bigger (fc->last_ack_recv, fc->last_pid_sent))
2203     return GNUNET_YES;
2204   return GNUNET_NO;
2205 }
2206
2207 /**
2208  * Sends an already built message on a connection, properly registering
2209  * all used resources.
2210  *
2211  * @param message Message to send. Function makes a copy of it.
2212  *                If message is not hop-by-hop, decrements TTL of copy.
2213  * @param c Connection on which this message is transmitted.
2214  * @param fwd Is this a fwd message?
2215  */
2216 void
2217 GMC_send_prebuilt_message (const struct GNUNET_MessageHeader *message,
2218                            struct MeshConnection *c,
2219                            int fwd)
2220 {
2221   struct MeshFlowControl *fc;
2222   void *data;
2223   size_t size;
2224   uint16_t type;
2225   int droppable;
2226
2227   size = ntohs (message->size);
2228   data = GNUNET_malloc (size);
2229   memcpy (data, message, size);
2230   type = ntohs (message->type);
2231   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send %s (%u) on connection %s\n",
2232               GNUNET_MESH_DEBUG_M2S (type), size, GMC_2s (c));
2233
2234   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2235   droppable = GNUNET_YES;
2236   switch (type)
2237   {
2238     struct GNUNET_MESH_Encrypted *emsg;
2239     struct GNUNET_MESH_KX        *kmsg;
2240     struct GNUNET_MESH_ACK       *amsg;
2241     struct GNUNET_MESH_Poll      *pmsg;
2242     struct GNUNET_MESH_ConnectionDestroy *dmsg;
2243     struct GNUNET_MESH_ConnectionBroken  *bmsg;
2244     uint32_t ttl;
2245
2246     case GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED:
2247       emsg = (struct GNUNET_MESH_Encrypted *) data;
2248       ttl = ntohl (emsg->ttl);
2249       if (0 == ttl)
2250       {
2251         GNUNET_break_op (0);
2252         return;
2253       }
2254       emsg->cid = c->id;
2255       emsg->ttl = htonl (ttl - 1);
2256       emsg->pid = htonl (fwd ? c->fwd_fc.next_pid++ : c->bck_fc.next_pid++);
2257       LOG (GNUNET_ERROR_TYPE_DEBUG, "  Q_N+ %p %u\n", fc, fc->queue_n);
2258       fc->queue_n++;
2259       LOG (GNUNET_ERROR_TYPE_DEBUG, "pid %u\n", ntohl (emsg->pid));
2260       LOG (GNUNET_ERROR_TYPE_DEBUG, "last pid %u\n", fc->last_pid_sent);
2261       LOG (GNUNET_ERROR_TYPE_DEBUG, "     ack %u\n", fc->last_ack_recv);
2262       if (GMC_is_pid_bigger (fc->last_pid_sent + 1, fc->last_ack_recv))
2263       {
2264         GMC_start_poll (c, fwd);
2265       }
2266       break;
2267
2268     case GNUNET_MESSAGE_TYPE_MESH_KX:
2269       kmsg = (struct GNUNET_MESH_KX *) data;
2270       kmsg->cid = c->id;
2271       break;
2272
2273     case GNUNET_MESSAGE_TYPE_MESH_ACK:
2274       amsg = (struct GNUNET_MESH_ACK *) data;
2275       amsg->cid = c->id;
2276       LOG (GNUNET_ERROR_TYPE_DEBUG, " ack %u\n", ntohl (amsg->ack));
2277       droppable = GNUNET_NO;
2278       break;
2279
2280     case GNUNET_MESSAGE_TYPE_MESH_POLL:
2281       pmsg = (struct GNUNET_MESH_Poll *) data;
2282       pmsg->cid = c->id;
2283       pmsg->pid = htonl (fwd ? c->fwd_fc.last_pid_sent : c->bck_fc.last_pid_sent);
2284       LOG (GNUNET_ERROR_TYPE_DEBUG, " poll %u\n", ntohl (pmsg->pid));
2285       droppable = GNUNET_NO;
2286       break;
2287
2288     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY:
2289       dmsg = (struct GNUNET_MESH_ConnectionDestroy *) data;
2290       dmsg->cid = c->id;
2291       dmsg->reserved = 0;
2292       break;
2293
2294     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_BROKEN:
2295       bmsg = (struct GNUNET_MESH_ConnectionBroken *) data;
2296       bmsg->cid = c->id;
2297       bmsg->reserved = 0;
2298       break;
2299
2300     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE:
2301     case GNUNET_MESSAGE_TYPE_MESH_CONNECTION_ACK:
2302       break;
2303
2304     default:
2305       GNUNET_break (0);
2306   }
2307
2308   if (fc->queue_n > fc->queue_max && droppable)
2309   {
2310     GNUNET_STATISTICS_update (stats, "# messages dropped (buffer full)",
2311                               1, GNUNET_NO);
2312     GNUNET_break (0);
2313     LOG (GNUNET_ERROR_TYPE_DEBUG,
2314                 "queue full: %u/%u\n",
2315                 fc->queue_n, fc->queue_max);
2316     if (GNUNET_MESSAGE_TYPE_MESH_ENCRYPTED == type)
2317       fc->queue_n--;
2318     return; /* Drop this message */
2319   }
2320
2321   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u\n", c, c->pending_messages);
2322   c->pending_messages++;
2323
2324   GMP_queue_add (get_hop (c, fwd), data, type, size, c, fwd,
2325                  &message_sent, NULL);
2326 }
2327
2328
2329 /**
2330  * Sends a CREATE CONNECTION message for a path to a peer.
2331  * Changes the connection and tunnel states if necessary.
2332  *
2333  * @param connection Connection to create.
2334  */
2335 void
2336 GMC_send_create (struct MeshConnection *connection)
2337 {
2338   enum MeshTunnel3State state;
2339   size_t size;
2340
2341   size = sizeof (struct GNUNET_MESH_ConnectionCreate);
2342   size += connection->path->length * sizeof (struct GNUNET_PeerIdentity);
2343   LOG (GNUNET_ERROR_TYPE_DEBUG, "Send connection create\n");
2344   GMP_queue_add (get_next_hop (connection), NULL,
2345                  GNUNET_MESSAGE_TYPE_MESH_CONNECTION_CREATE,
2346                  size, connection, GNUNET_YES, &message_sent, NULL);
2347   LOG (GNUNET_ERROR_TYPE_DEBUG, "  C_P+ %p %u (create)\n",
2348        connection, connection->pending_messages);
2349   connection->pending_messages++;
2350   state = GMT_get_state (connection->t);
2351   if (MESH_TUNNEL3_SEARCHING == state || MESH_TUNNEL3_NEW == state)
2352     GMT_change_state (connection->t, MESH_TUNNEL3_WAITING);
2353   if (MESH_CONNECTION_NEW == connection->state)
2354     connection_change_state (connection, MESH_CONNECTION_SENT);
2355 }
2356
2357
2358 /**
2359  * Send a message to all peers in this connection that the connection
2360  * is no longer valid.
2361  *
2362  * If some peer should not receive the message, it should be zero'ed out
2363  * before calling this function.
2364  *
2365  * @param c The connection whose peers to notify.
2366  */
2367 void
2368 GMC_send_destroy (struct MeshConnection *c)
2369 {
2370   struct GNUNET_MESH_ConnectionDestroy msg;
2371
2372   if (GNUNET_YES == c->destroy)
2373     return;
2374
2375   msg.header.size = htons (sizeof (msg));
2376   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_CONNECTION_DESTROY);;
2377   msg.cid = c->id;
2378   LOG (GNUNET_ERROR_TYPE_DEBUG,
2379               "  sending connection destroy for connection %s\n",
2380               GMC_2s (c));
2381
2382   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_YES))
2383     GMC_send_prebuilt_message (&msg.header, c, GNUNET_YES);
2384   if (GNUNET_NO == GMC_is_terminal (c, GNUNET_NO))
2385     GMC_send_prebuilt_message (&msg.header, c, GNUNET_NO);
2386   c->destroy = GNUNET_YES;
2387 }
2388
2389
2390 /**
2391  * @brief Start a polling timer for the connection.
2392  *
2393  * When a neighbor does not accept more traffic on the connection it could be
2394  * caused by a simple congestion or by a lost ACK. Polling enables to check
2395  * for the lastest ACK status for a connection.
2396  *
2397  * @param c Connection.
2398  * @param fwd Should we poll in the FWD direction?
2399  */
2400 void
2401 GMC_start_poll (struct MeshConnection *c, int fwd)
2402 {
2403   struct MeshFlowControl *fc;
2404
2405   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2406   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
2407   {
2408     return;
2409   }
2410   fc->poll_task = GNUNET_SCHEDULER_add_delayed (fc->poll_time,
2411                                                 &connection_poll,
2412                                                 fc);
2413 }
2414
2415
2416 /**
2417  * @brief Stop polling a connection for ACKs.
2418  *
2419  * Once we have enough ACKs for future traffic, polls are no longer necessary.
2420  *
2421  * @param c Connection.
2422  * @param fwd Should we stop the poll in the FWD direction?
2423  */
2424 void
2425 GMC_stop_poll (struct MeshConnection *c, int fwd)
2426 {
2427   struct MeshFlowControl *fc;
2428
2429   fc = fwd ? &c->fwd_fc : &c->bck_fc;
2430   if (GNUNET_SCHEDULER_NO_TASK != fc->poll_task)
2431   {
2432     GNUNET_SCHEDULER_cancel (fc->poll_task);
2433     fc->poll_task = GNUNET_SCHEDULER_NO_TASK;
2434   }
2435 }
2436
2437 /**
2438  * Get a (static) string for a connection.
2439  *
2440  * @param c Connection.
2441  */
2442 const char *
2443 GMC_2s (struct MeshConnection *c)
2444 {
2445   if (NULL != c->t)
2446   {
2447     static char buf[128];
2448
2449     sprintf (buf, "%s (->%s)", GNUNET_h2s (&c->id), GMT_2s (c->t));
2450     return buf;
2451   }
2452   return GNUNET_h2s (&c->id);
2453 }