src/transport/transport_api.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009, 2010 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 transport/transport_api.c
  23  * @brief library to access the low-level P2P IO service
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_bandwidth_lib.h"
  28 #include "gnunet_client_lib.h"
  29 #include "gnunet_constants.h"
  30 #include "gnunet_container_lib.h"
  31 #include "gnunet_arm_service.h"
  32 #include "gnunet_hello_lib.h"
  33 #include "gnunet_protocols.h"
  34 #include "gnunet_server_lib.h"
  35 #include "gnunet_time_lib.h"
  36 #include "gnunet_transport_service.h"
  37 #include "transport.h"
  38
  39 /**
  40  * After how long do we give up on transmitting a HELLO
  41  * to the service?
  42  */
  43 #define OFFER_HELLO_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  44
  45 /**
  46  * After how long do we automatically retry an unsuccessful
  47  * CONNECT request?
  48  */
  49 #define CONNECT_RETRY_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, 750)
  50
  51 /**
  52  * How long should ARM wait when starting up the
  53  * transport service before reporting back?
  54  */
  55 #define START_SERVICE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
  56
  57 /**
  58  * How long should ARM wait when stopping the
  59  * transport service before reporting back?
  60  */
  61 #define STOP_SERVICE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
  62
  63 /**
  64  * How large to start with for the hashmap of neighbours.
  65  */
  66 #define STARTING_NEIGHBOURS_SIZE 10
  67
  68
  69 /**
  70  * What stage are we in for transmission processing?
  71  */
  72 enum TransmitStage
  73   {
  74     /**
  75      * No active message.
  76      */
  77     TS_NEW = 0,
  78
  79     /**
  80      * Message in local queue, not given to service.
  81      */
  82     TS_QUEUED = 1,
  83
  84     /**
  85      * Message given to service, not confirmed (no SEND_OK).
  86      */
  87     TS_TRANSMITTED = 2,
  88
  89     /**
  90      * One message was given to service and before it was confirmed,
  91      * another one was already queued (waiting for SEND_OK to pass on
  92      * to service).
  93      */
  94     TS_TRANSMITTED_QUEUED = 3
  95   };
  96
  97
  98 /**
  99  * Handle for a transmission-ready request.
 100  */
 101 struct GNUNET_TRANSPORT_TransmitHandle
 102 {
 103
 104   /**
 105    * Neighbour for this handle, NULL for control-traffic.
 106    */
 107   struct NeighbourList *neighbour;
 108
 109   /**
 110    * Function to call when notify_size bytes are available
 111    * for transmission.
 112    */
 113   GNUNET_CONNECTION_TransmitReadyNotify notify;
 114
 115   /**
 116    * Closure for notify.
 117    */
 118   void *notify_cls;
 119
 120   /**
 121    * transmit_ready task Id.  The task is used to introduce the
 122    * artificial delay that may be required to maintain the bandwidth
 123    * limits.  Later, this will be the ID of the "transmit_timeout"
 124    * task which is used to signal a timeout if the transmission could
 125    * not be done in a timely fashion.
 126    */
 127   GNUNET_SCHEDULER_TaskIdentifier notify_delay_task;
 128
 129   /**
 130    * Timeout for this request.
 131    */
 132   struct GNUNET_TIME_Absolute timeout;
 133
 134   /**
 135    * How many bytes is our notify callback waiting for?
 136    */
 137   size_t notify_size;
 138
 139   /**
 140    * How important is this message?
 141    */
 142   uint32_t priority;
 143
 144 };
 145
 146
 147 /**
 148  * Handle for a control message queue entry.
 149  */
 150 struct ControlMessage
 151 {
 152
 153   /**
 154    * This is a doubly-linked list.
 155    */
 156   struct ControlMessage *next;
 157
 158   /**
 159    * This is a doubly-linked list.
 160    */
 161   struct ControlMessage *prev;
 162
 163   /**
 164    * Overall transport handle.
 165    */
 166   struct GNUNET_TRANSPORT_Handle *h;
 167
 168   /**
 169    * Function to call when notify_size bytes are available
 170    * for transmission.
 171    */
 172   GNUNET_CONNECTION_TransmitReadyNotify notify;
 173
 174   /**
 175    * Closure for notify.
 176    */
 177   void *notify_cls;
 178
 179   /**
 180    * transmit_ready task Id.  The task is used to introduce the
 181    * artificial delay that may be required to maintain the bandwidth
 182    * limits.  Later, this will be the ID of the "transmit_timeout"
 183    * task which is used to signal a timeout if the transmission could
 184    * not be done in a timely fashion.
 185    */
 186   GNUNET_SCHEDULER_TaskIdentifier notify_delay_task;
 187
 188   /**
 189    * How many bytes is our notify callback waiting for?
 190    */
 191   size_t notify_size;
 192
 193 };
 194
 195 /**
 196  * Context for storing information about attempted next transmission.
 197  */
 198 struct TryTransmitContext
 199 {
 200
 201   /**
 202    * Main transport handle.
 203    */
 204   struct GNUNET_TRANSPORT_Handle *h;
 205
 206   /**
 207    * Returned transmission handle.
 208    */
 209   struct GNUNET_TRANSPORT_TransmitHandle *ret;
 210
 211   /**
 212    * Time to retry the send task.
 213    */
 214   struct GNUNET_TIME_Relative retry_time;
 215 };
 216
 217 /**
 218  * Entry in hash table of all of our current neighbours.
 219  */
 220 struct NeighbourList
 221 {
 222   /**
 223    * Overall transport handle.
 224    */
 225   struct GNUNET_TRANSPORT_Handle *h;
 226
 227   /**
 228    * Active transmit handle; available if 'transmit_forbidden'
 229    * is GNUNET_NO.
 230    */
 231   struct GNUNET_TRANSPORT_TransmitHandle transmit_handle;
 232
 233   /**
 234    * Identity of this neighbour.
 235    */
 236   struct GNUNET_PeerIdentity id;
 237
 238   /**
 239    * Outbound bandwidh tracker.
 240    */
 241   struct GNUNET_BANDWIDTH_Tracker out_tracker;
 242
 243   /**
 244    * Set to GNUNET_NO if we are currently allowed to accept a
 245    * message to the transport service for this peer, GNUNET_YES
 246    * if we have one and are waiting for transmission, GNUNET_SYSERR
 247    * if we are waiting for confirmation AND have already accepted
 248    * yet another message.
 249    */
 250   enum TransmitStage transmit_stage;
 251
 252   /**
 253    * Have we received a notification that this peer is connected
 254    * to us right now?
 255    */
 256   int is_connected;
 257
 258   /**
 259    * Are we in the middle of disconnecting the peer already?
 260    */
 261   unsigned int in_disconnect;
 262
 263 };
 264
 265
 266 /**
 267  * Linked list of requests from clients for our HELLO that were
 268  * deferred.
 269  */
 270 struct HelloWaitList
 271 {
 272
 273   /**
 274    * This is a linked list.
 275    */
 276   struct HelloWaitList *next;
 277
 278   /**
 279    * Reference back to our transport handle.
 280    */
 281   struct GNUNET_TRANSPORT_Handle *handle;
 282
 283   /**
 284    * Callback to call once we got our HELLO.
 285    */
 286   GNUNET_TRANSPORT_HelloUpdateCallback rec;
 287
 288   /**
 289    * Closure for rec.
 290    */
 291   void *rec_cls;
 292
 293 };
 294
 295
 296 /**
 297  * Handle for the transport service (includes all of the
 298  * state for the transport service).
 299  */
 300 struct GNUNET_TRANSPORT_Handle
 301 {
 302
 303   /**
 304    * Closure for the callbacks.
 305    */
 306   void *cls;
 307
 308   /**
 309    * Function to call for received data.
 310    */
 311   GNUNET_TRANSPORT_ReceiveCallback rec;
 312
 313   /**
 314    * function to call on connect events
 315    */
 316   GNUNET_TRANSPORT_NotifyConnect nc_cb;
 317
 318   /**
 319    * function to call on disconnect events
 320    */
 321   GNUNET_TRANSPORT_NotifyDisconnect nd_cb;
 322
 323   /**
 324    * Head of DLL of control messages.
 325    */
 326   struct ControlMessage *control_head;
 327
 328   /**
 329    * Tail of DLL of control messages.
 330    */
 331   struct ControlMessage *control_tail;
 332
 333   /**
 334    * The current HELLO message for this peer.  Updated
 335    * whenever transports change their addresses.
 336    */
 337   struct GNUNET_HELLO_Message *my_hello;
 338
 339   /**
 340    * My client connection to the transport service.
 341    */
 342   struct GNUNET_CLIENT_Connection *client;
 343
 344   /**
 345    * Handle to our registration with the client for notification.
 346    */
 347   struct GNUNET_CLIENT_TransmitHandle *network_handle;
 348
 349   /**
 350    * Linked list of pending requests for our HELLO.
 351    */
 352   struct HelloWaitList *hwl_head;
 353
 354   /**
 355    * My configuration.
 356    */
 357   const struct GNUNET_CONFIGURATION_Handle *cfg;
 358
 359   /**
 360    * Linked list of the current neighbours of this peer.
 361    */
 362   struct GNUNET_CONTAINER_MultiHashMap *neighbours;
 363
 364   /**
 365    * Peer identity as assumed by this process, or all zeros.
 366    */
 367   struct GNUNET_PeerIdentity self;
 368
 369   /**
 370    * ID of the task trying to reconnect to the service.
 371    */
 372   GNUNET_SCHEDULER_TaskIdentifier reconnect_task;
 373
 374   /**
 375    * ID of the task trying to trigger transmission for a peer
 376    * while maintaining bandwidth quotas.
 377    */
 378   GNUNET_SCHEDULER_TaskIdentifier quota_task;
 379
 380   /**
 381    * Delay until we try to reconnect.
 382    */
 383   struct GNUNET_TIME_Relative reconnect_delay;
 384
 385   /**
 386    * Set once we are in the process of disconnecting from the
 387    * service.
 388    */
 389   int in_disconnect;
 390
 391   /**
 392    * Should we check that 'self' matches what the service thinks?
 393    * (if GNUNET_NO, then 'self' is all zeros!).
 394    */
 395   int check_self;
 396 };
 397
 398 struct HelloContext
 399 {
 400
 401   /**
 402    * Size of the HELLO copied to end of struct.
 403    */
 404   uint16_t size;
 405
 406   /**
 407    * Continuation to call once HELLO sent.
 408    */
 409   GNUNET_SCHEDULER_Task cont;
 410
 411   /**
 412    * Closure to call with the continuation.
 413    */
 414   void *cont_cls;
 415
 416   /* HELLO */
 417 };
 418
 419
 420 /**
 421  * Get the neighbour list entry for the given peer
 422  *
 423  * @param h our context
 424  * @param peer peer to look up
 425  * @return NULL if no such peer entry exists
 426  */
 427 static struct NeighbourList *
 428 neighbour_find (struct GNUNET_TRANSPORT_Handle *h,
 429                 const struct GNUNET_PeerIdentity *peer)
 430 {
 431   return GNUNET_CONTAINER_multihashmap_get(h->neighbours, &peer->hashPubKey);
 432 }
 433
 434
 435 /**
 436  * Schedule the task to send one message, either from the control
 437  * list or the peer message queues  to the service.
 438  */
 439 static void schedule_transmission (struct GNUNET_TRANSPORT_Handle *h);
 440
 441
 442 /**
 443  * Function called by the scheduler when the timeout for bandwidth
 444  * availablility for the target neighbour is reached.
 445  *
 446  * @param cls the 'struct GNUNET_TRANSPORT_Handle*'
 447  * @param tc scheduler context
 448  */
 449 static void
 450 quota_transmit_ready (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 451 {
 452   struct GNUNET_TRANSPORT_Handle *h = cls;
 453
 454   h->quota_task = GNUNET_SCHEDULER_NO_TASK;
 455   schedule_transmission (h);
 456 }
 457
 458
 459 /**
 460  * Iterator over hash map entries, attempt to schedule
 461  * a transmission to entries in the neighbour hashmap.
 462  *
 463  * @param cls closure a TryTransmitContext
 464  * @param key current key code
 465  * @param value value in the hash map, the neighbour entry to consider
 466  * @return GNUNET_YES if we should continue to
 467  *         iterate,
 468  *         GNUNET_NO if not.
 469  */
 470 static int
 471 try_schedule_transmission (void *cls,
 472                            const GNUNET_HashCode * key,
 473                            void *value)
 474 {
 475   struct NeighbourList *n = value;
 476   struct TryTransmitContext *try_transmit_ctx = cls;
 477   struct GNUNET_TIME_Relative duration;
 478   GNUNET_CONNECTION_TransmitReadyNotify notify;
 479   struct GNUNET_TRANSPORT_TransmitHandle *th;
 480   struct GNUNET_TIME_Absolute duration_abs;
 481
 482   if (n->transmit_stage != TS_QUEUED)
 483     return GNUNET_YES; /* not eligible, keep iterating */
 484   if (n->is_connected != GNUNET_YES)
 485     return GNUNET_YES; /* keep iterating */
 486
 487   th = &n->transmit_handle;
 488   GNUNET_break (n == th->neighbour);
 489   /* check outgoing quota */
 490   duration = GNUNET_BANDWIDTH_tracker_get_delay (&n->out_tracker,
 491                                                  th->notify_size - sizeof (struct OutboundMessage));
 492   duration_abs = GNUNET_TIME_relative_to_absolute (duration);
 493   if (th->timeout.abs_value < duration_abs.abs_value)
 494     {
 495       /* signal timeout! */
 496 #if DEBUG_TRANSPORT
 497       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 498                   "Would need %llu ms before bandwidth is available for delivery to `%4s', that is too long.  Signaling timeout.\n",
 499                   duration.rel_value,
 500                   GNUNET_i2s (&n->id));
 501 #endif
 502       if (th->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
 503         {
 504           GNUNET_SCHEDULER_cancel (th->notify_delay_task);
 505           th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
 506         }
 507       n->transmit_stage = TS_NEW;
 508       if (NULL != (notify = th->notify))
 509         {
 510           th->notify = NULL;
 511           GNUNET_assert (0 == notify (th->notify_cls, 0, NULL));
 512         }
 513       return GNUNET_YES; /* keep iterating */
 514     }
 515   if (duration.rel_value > 0)
 516     {
 517 #if DEBUG_TRANSPORT
 518       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 519                   "Need more bandwidth (%u b/s allowed, %u b needed), delaying delivery to `%4s' by %llu ms\n",
 520                   (unsigned int) n->out_tracker.available_bytes_per_s__,
 521                   (unsigned int) th->notify_size - sizeof (struct OutboundMessage),
 522                   GNUNET_i2s (&n->id),
 523                   (unsigned long long) duration.rel_value);
 524 #endif
 525       try_transmit_ctx->retry_time = GNUNET_TIME_relative_min (try_transmit_ctx->retry_time,
 526                                                                duration);
 527       return GNUNET_YES; /* keep iterating */
 528     }
 529 #if DEBUG_TRANSPORT
 530   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 531               "Have %u bytes of bandwidth available for transmission to `%4s' right now\n",
 532               th->notify_size - sizeof (struct OutboundMessage),
 533               GNUNET_i2s (&n->id));
 534 #endif
 535
 536   if ( (try_transmit_ctx->ret == NULL) ||
 537        (try_transmit_ctx->ret->priority < th->priority) )
 538     try_transmit_ctx->ret = th;
 539   return GNUNET_YES;
 540 }
 541
 542
 543 /**
 544  * Figure out which transmission to a peer can be done right now.
 545  * If none can, schedule a task to call 'schedule_transmission'
 546  * whenever a peer transmission can be done in the future and
 547  * return NULL.  Otherwise return the next transmission to be
 548  * performed.
 549  *
 550  * @param h handle to transport
 551  * @return NULL to wait longer before doing any peer transmissions
 552  */
 553 static struct GNUNET_TRANSPORT_TransmitHandle *
 554 schedule_peer_transmission (struct GNUNET_TRANSPORT_Handle *h)
 555 {
 556   struct TryTransmitContext try_transmit_ctx;
 557
 558   if (h->quota_task != GNUNET_SCHEDULER_NO_TASK)
 559     {
 560       GNUNET_SCHEDULER_cancel (h->quota_task);
 561       h->quota_task = GNUNET_SCHEDULER_NO_TASK;
 562     }
 563   try_transmit_ctx.h = h;
 564   try_transmit_ctx.ret = NULL;
 565   try_transmit_ctx.retry_time = GNUNET_TIME_UNIT_FOREVER_REL;
 566   GNUNET_CONTAINER_multihashmap_iterate(h->neighbours,
 567                                         &try_schedule_transmission,
 568                                         &try_transmit_ctx);
 569   if (try_transmit_ctx.ret == NULL)
 570     h->quota_task = GNUNET_SCHEDULER_add_delayed (try_transmit_ctx.retry_time,
 571                                                   &quota_transmit_ready,
 572                                                   h);
 573   return try_transmit_ctx.ret;
 574 }
 575
 576
 577 /**
 578  * Transmit message(s) to service.
 579  *
 580  * @param cls handle to transport
 581  * @param size number of bytes available in buf
 582  * @param buf where to copy the message
 583  * @return number of bytes copied to buf
 584  */
 585 static size_t
 586 transport_notify_ready (void *cls, size_t size, void *buf)
 587 {
 588   struct GNUNET_TRANSPORT_Handle *h = cls;
 589   struct ControlMessage *cm;
 590   struct GNUNET_TRANSPORT_TransmitHandle *th;
 591   struct NeighbourList *n;
 592   struct OutboundMessage obm;
 593   GNUNET_CONNECTION_TransmitReadyNotify notify;
 594   size_t ret;
 595   size_t mret;
 596   size_t nret;
 597   char *cbuf;
 598
 599   h->network_handle = NULL;
 600   if (buf == NULL)
 601     {
 602       schedule_transmission (h);
 603       return 0;
 604     }
 605 #if DEBUG_TRANSPORT
 606   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 607               "Ready to transmit %u bytes to transport service\n", size);
 608 #endif
 609   cbuf = buf;
 610   ret = 0;
 611   while ( (NULL != (cm = h->control_head)) &&
 612           (cm->notify_size <= size) )
 613     {
 614       if (cm->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
 615         {
 616           GNUNET_SCHEDULER_cancel (cm->notify_delay_task);
 617           cm->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
 618         }
 619       GNUNET_CONTAINER_DLL_remove (h->control_head,
 620                                    h->control_tail,
 621                                    cm);
 622       nret = cm->notify (cm->notify_cls, size, &cbuf[ret]);
 623 #if DEBUG_TRANSPORT
 624       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 625                   "Added %u bytes of control message at %u\n",
 626                   nret,
 627                   ret);
 628 #endif
 629       GNUNET_free (cm);
 630       ret += nret;
 631       size -= nret;
 632     }
 633   while ( (NULL != (th = schedule_peer_transmission (h))) &&
 634           (th->notify_size <= size) )
 635     {
 636       if (th->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
 637         {
 638           GNUNET_SCHEDULER_cancel (th->notify_delay_task);
 639           th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
 640         }
 641       n = th->neighbour;
 642       switch (n->transmit_stage)
 643         {
 644         case TS_NEW:
 645           GNUNET_break (0);
 646           break;
 647         case TS_QUEUED:
 648           n->transmit_stage = TS_TRANSMITTED;
 649           break;
 650         case TS_TRANSMITTED:
 651           GNUNET_break (0);
 652           break;
 653         case TS_TRANSMITTED_QUEUED:
 654           GNUNET_break (0);
 655           break;
 656         default:
 657           GNUNET_break (0);
 658         }
 659       GNUNET_assert (size >= sizeof (struct OutboundMessage));
 660       notify = th->notify;
 661       th->notify = NULL;
 662       mret = notify (th->notify_cls,
 663                      size - sizeof (struct OutboundMessage),
 664                      &cbuf[ret + sizeof (struct OutboundMessage)]);
 665       GNUNET_assert (mret <= size - sizeof (struct OutboundMessage));
 666 #if DEBUG_TRANSPORT
 667       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 668                   "Message of %u bytes with timeout %llums constructed for `%4s'\n",
 669                   (unsigned int) mret,
 670                   (unsigned long long) GNUNET_TIME_absolute_get_remaining (th->timeout).rel_value,
 671                   GNUNET_i2s (&n->id));
 672 #endif
 673       if (mret != 0)
 674         {
 675           GNUNET_assert (mret + sizeof (struct OutboundMessage) < GNUNET_SERVER_MAX_MESSAGE_SIZE);
 676           obm.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_SEND);
 677           obm.header.size = htons (mret + sizeof (struct OutboundMessage));
 678           obm.priority = htonl (th->priority);
 679           obm.timeout = GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining (th->timeout));
 680           obm.peer = n->id;
 681           memcpy (&cbuf[ret], &obm, sizeof (struct OutboundMessage));
 682           ret += (mret + sizeof (struct OutboundMessage));
 683           size -= (mret + sizeof (struct OutboundMessage));
 684           GNUNET_BANDWIDTH_tracker_consume (&n->out_tracker, mret);
 685         }
 686       else
 687         {
 688           switch (n->transmit_stage)
 689             {
 690             case TS_NEW:
 691               GNUNET_break (0);
 692               break;
 693             case TS_QUEUED:
 694               GNUNET_break (0);
 695               break;
 696             case TS_TRANSMITTED:
 697               n->transmit_stage = TS_NEW;
 698               break;
 699             case TS_TRANSMITTED_QUEUED:
 700               n->transmit_stage = TS_QUEUED;
 701               continue;
 702             default:
 703               GNUNET_break (0);
 704             }
 705         }
 706     }
 707   schedule_transmission (h);
 708 #if DEBUG_TRANSPORT
 709   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 710               "Transmitting %u bytes to transport service\n", ret);
 711 #endif
 712   return ret;
 713 }
 714
 715
 716 /**
 717  * Schedule the task to send one message, either from the control
 718  * list or the peer message queues  to the service.
 719  */
 720 static void
 721 schedule_transmission (struct GNUNET_TRANSPORT_Handle *h)
 722 {
 723   size_t size;
 724   struct GNUNET_TIME_Relative timeout;
 725   struct GNUNET_TRANSPORT_TransmitHandle *th;
 726
 727   if (NULL != h->network_handle)
 728     return;
 729   if (h->client == NULL)
 730     {
 731       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 732                   _("Could not yet schedule transmission: we are not yet connected to the transport service!\n"));
 733       return;                   /* not yet connected */
 734     }
 735   if (NULL != h->control_head)
 736     {
 737       size = h->control_head->notify_size;
 738       timeout = GNUNET_TIME_UNIT_FOREVER_REL;
 739     }
 740   else
 741     {
 742       th = schedule_peer_transmission (h);
 743       if (th == NULL)
 744         {
 745           /* no transmission ready right now */
 746 #if DEBUG_TRANSPORT
 747           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 748                       "Could not yet schedule transmission: none ready\n");
 749 #endif
 750           return;
 751         }
 752       size = th->notify_size;
 753       timeout = GNUNET_TIME_absolute_get_remaining (th->timeout);
 754     }
 755 #if DEBUG_TRANSPORT
 756     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 757                 "Calling notify_transmit_ready\n");
 758 #endif
 759   h->network_handle =
 760     GNUNET_CLIENT_notify_transmit_ready (h->client,
 761                                          size,
 762                                          timeout,
 763                                          GNUNET_NO,
 764                                          &transport_notify_ready,
 765                                          h);
 766   GNUNET_assert (NULL != h->network_handle);
 767 }
 768
 769
 770 /**
 771  * Called when our transmit request timed out before any transport
 772  * reported success connecting to the desired peer or before the
 773  * transport was ready to receive.  Signal error and free
 774  * TransmitHandle.
 775  */
 776 static void
 777 control_transmit_timeout (void *cls,
 778                           const struct GNUNET_SCHEDULER_TaskContext *tc)
 779 {
 780   struct ControlMessage *th = cls;
 781
 782   th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
 783   if (NULL != th->notify)
 784     th->notify (th->notify_cls, 0, NULL);
 785   GNUNET_CONTAINER_DLL_remove (th->h->control_head,
 786                                th->h->control_tail,
 787                                th);
 788   GNUNET_free (th);
 789 }
 790
 791
 792 /**
 793  * Queue control request for transmission to the transport
 794  * service.
 795  *
 796  * @param h handle to the transport service
 797  * @param size number of bytes to be transmitted
 798  * @param at_head request must be added to the head of the queue
 799  *        (otherwise request will be appended)
 800  * @param timeout how long this transmission can wait (at most)
 801  * @param notify function to call to get the content
 802  * @param notify_cls closure for notify
 803  */
 804 static void
 805 schedule_control_transmit (struct GNUNET_TRANSPORT_Handle *h,
 806                            size_t size,
 807                            int at_head,
 808                            struct GNUNET_TIME_Relative timeout,
 809                            GNUNET_CONNECTION_TransmitReadyNotify notify,
 810                            void *notify_cls)
 811 {
 812   struct ControlMessage *cm;
 813
 814 #if DEBUG_TRANSPORT
 815   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 816               "Control transmit of %u bytes within %llums requested\n",
 817               size, (unsigned long long) timeout.rel_value);
 818 #endif
 819   cm = GNUNET_malloc (sizeof (struct ControlMessage));
 820   cm->h = h;
 821   cm->notify = notify;
 822   cm->notify_cls = notify_cls;
 823   cm->notify_size = size;
 824   cm->notify_delay_task
 825     = GNUNET_SCHEDULER_add_delayed (timeout, &control_transmit_timeout, cm);
 826   if (at_head)
 827     GNUNET_CONTAINER_DLL_insert (h->control_head,
 828                                  h->control_tail,
 829                                  cm);
 830   else
 831     GNUNET_CONTAINER_DLL_insert_after (h->control_head,
 832                                        h->control_tail,
 833                                        h->control_tail,
 834                                        cm);
 835   schedule_transmission (h);
 836 }
 837
 838
 839 /**
 840  * FIXME: document
 841  */
 842 struct SetQuotaContext
 843 {
 844   /**
 845    * FIXME: document
 846    */
 847   struct GNUNET_TRANSPORT_Handle *handle;
 848
 849   /**
 850    * FIXME: document
 851    */
 852   struct GNUNET_PeerIdentity target;
 853
 854   /**
 855    * FIXME: document
 856    */
 857   GNUNET_SCHEDULER_Task cont;
 858
 859   /**
 860    * Closure for 'cont'.
 861    */
 862   void *cont_cls;
 863
 864   /**
 865    * FIXME: document
 866    */
 867   struct GNUNET_TIME_Absolute timeout;
 868
 869   /**
 870    * FIXME: document
 871    */
 872   struct GNUNET_BANDWIDTH_Value32NBO quota_in;
 873 };
 874
 875
 876 /**
 877  * Send SET_QUOTA message to the service.
 878  *
 879  * @param cls the 'struct SetQuotaContext'
 880  * @param size number of bytes available in buf
 881  * @param buf where to copy the message
 882  * @return number of bytes copied to buf
 883  */
 884 static size_t
 885 send_set_quota (void *cls, size_t size, void *buf)
 886 {
 887   struct SetQuotaContext *sqc = cls;
 888   struct QuotaSetMessage *msg;
 889
 890   if (buf == NULL)
 891     {
 892       if (sqc->cont != NULL)
 893         GNUNET_SCHEDULER_add_continuation (sqc->cont,
 894                                            sqc->cont_cls,
 895                                            GNUNET_SCHEDULER_REASON_TIMEOUT);
 896       GNUNET_free (sqc);
 897       return 0;
 898     }
 899 #if DEBUG_TRANSPORT
 900   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 901               "Transmitting `%s' request with respect to `%4s'.\n",
 902               "SET_QUOTA",
 903               GNUNET_i2s (&sqc->target));
 904 #endif
 905   GNUNET_assert (size >= sizeof (struct QuotaSetMessage));
 906   msg = buf;
 907   msg->header.size = htons (sizeof (struct QuotaSetMessage));
 908   msg->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_SET_QUOTA);
 909   msg->quota = sqc->quota_in;
 910   memcpy (&msg->peer, &sqc->target, sizeof (struct GNUNET_PeerIdentity));
 911   if (sqc->cont != NULL)
 912     GNUNET_SCHEDULER_add_continuation (sqc->cont,
 913                                        sqc->cont_cls,
 914                                        GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 915   GNUNET_free (sqc);
 916   return sizeof (struct QuotaSetMessage);
 917 }
 918
 919
 920 /**
 921  * Set the share of incoming bandwidth for the given
 922  * peer to the specified amount.
 923  *
 924  * @param handle connection to transport service
 925  * @param target who's bandwidth quota is being changed
 926  * @param quota_in incoming bandwidth quota in bytes per ms
 927  * @param quota_out outgoing bandwidth quota in bytes per ms
 928  */
 929 void
 930 GNUNET_TRANSPORT_set_quota (struct GNUNET_TRANSPORT_Handle *handle,
 931                             const struct GNUNET_PeerIdentity *target,
 932                             struct GNUNET_BANDWIDTH_Value32NBO quota_in,
 933                             struct GNUNET_BANDWIDTH_Value32NBO quota_out)
 934 {
 935   struct NeighbourList *n;
 936   struct SetQuotaContext *sqc;
 937
 938   n = neighbour_find (handle, target);
 939   if (n != NULL)
 940     {
 941 #if DEBUG_TRANSPORT
 942       if (ntohl (quota_out.value__) != n->out_tracker.available_bytes_per_s__)
 943         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 944                     "Quota changed from %u to %u for peer `%s'\n",
 945                     (unsigned int) n->out_tracker.available_bytes_per_s__,
 946                     (unsigned int) ntohl (quota_out.value__),
 947                     GNUNET_i2s (target));
 948       else
 949         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 950                     "Quota remains at %u for peer `%s'\n",
 951                     (unsigned int) n->out_tracker.available_bytes_per_s__,
 952                     GNUNET_i2s (target));
 953 #endif
 954       GNUNET_BANDWIDTH_tracker_update_quota (&n->out_tracker,
 955                                              quota_out);
 956     }
 957   else
 958     {
 959 #if DEBUG_TRANSPORT
 960       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 961                   "Quota changed to %u for peer `%s', but I have no such neighbour!\n",
 962                   (unsigned int) ntohl (quota_out.value__),
 963                   GNUNET_i2s (target));
 964 #endif
 965     }
 966   sqc = GNUNET_malloc (sizeof (struct SetQuotaContext));
 967   sqc->handle = handle;
 968   sqc->target = *target;
 969   sqc->cont = NULL;
 970   sqc->cont_cls = NULL;
 971   sqc->timeout = GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_FOREVER_REL);
 972   sqc->quota_in = quota_in;
 973   schedule_control_transmit (handle,
 974                              sizeof (struct QuotaSetMessage),
 975                              GNUNET_NO,
 976                              GNUNET_TIME_UNIT_FOREVER_REL, &send_set_quota, sqc);
 977 }
 978
 979
 980 /**
 981  * Ask the transport service to establish a connection to
 982  * the given peer.
 983  *
 984  * @param handle connection to transport service
 985  * @param target who we should try to connect to
 986  */
 987 void
 988 GNUNET_TRANSPORT_try_connect (struct GNUNET_TRANSPORT_Handle *handle,
 989                               const struct GNUNET_PeerIdentity *target)
 990 {
 991   /* old API does nothing */
 992 }
 993
 994
 995
 996
 997 /**
 998  * Obtain the HELLO message for this peer.
 999  *
1000  * @param handle connection to transport service
1001  * @param rec function to call with the HELLO, sender will be our peer
1002  *            identity; message and sender will be NULL on timeout
1003  *            (handshake with transport service pending/failed).
1004  *             cost estimate will be 0.
1005  * @param rec_cls closure for rec
1006  */
1007 void
1008 GNUNET_TRANSPORT_get_hello (struct GNUNET_TRANSPORT_Handle *handle,
1009                             GNUNET_TRANSPORT_HelloUpdateCallback rec,
1010                             void *rec_cls)
1011 {
1012   struct HelloWaitList *hwl;
1013
1014   hwl = GNUNET_malloc (sizeof (struct HelloWaitList));
1015   hwl->next = handle->hwl_head;
1016   handle->hwl_head = hwl;
1017   hwl->handle = handle;
1018   hwl->rec = rec;
1019   hwl->rec_cls = rec_cls;
1020   if (handle->my_hello == NULL)
1021     {
1022 #if DEBUG_TRANSPORT_HELLO
1023       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1024                   "No HELLO yet, waiting to receive it from transport service\n");
1025 #endif
1026       return;
1027     }
1028   rec (rec_cls, (const struct GNUNET_MessageHeader *) handle->my_hello);
1029 }
1030
1031
1032
1033 /**
1034  * Stop receiving updates about changes to our HELLO message.
1035  *
1036  * @param handle connection to transport service
1037  * @param rec function previously registered to be called with the HELLOs
1038  * @param rec_cls closure for rec
1039  */
1040 void
1041 GNUNET_TRANSPORT_get_hello_cancel (struct GNUNET_TRANSPORT_Handle *handle,
1042                                    GNUNET_TRANSPORT_HelloUpdateCallback rec,
1043                                    void *rec_cls)
1044 {
1045   struct HelloWaitList *pos;
1046   struct HelloWaitList *prev;
1047
1048   prev = NULL;
1049   pos = handle->hwl_head;
1050   while (pos != NULL)
1051     {
1052       if ( (pos->rec == rec) &&
1053            (pos->rec_cls == rec_cls) )
1054         break;
1055       prev = pos;
1056       pos = pos->next;
1057     }
1058   GNUNET_break (pos != NULL);
1059   if (pos == NULL)
1060     return;
1061   if (prev == NULL)
1062     handle->hwl_head = pos->next;
1063   else
1064     prev->next = pos->next;
1065   GNUNET_free (pos);
1066 }
1067
1068
1069 /**
1070  * Send HELLO message to the service.
1071  *
1072  * @param cls the HELLO message to send
1073  * @param size number of bytes available in buf
1074  * @param buf where to copy the message
1075  * @return number of bytes copied to buf
1076  */
1077 static size_t
1078 send_hello (void *cls, size_t size, void *buf)
1079 {
1080   struct HelloContext *hc = cls;
1081   uint16_t ssize;
1082   if (buf == NULL)
1083     {
1084 #if DEBUG_TRANSPORT_TIMEOUT
1085       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1086                   "Timeout while trying to transmit `%s' request.\n",
1087                   "HELLO");
1088 #endif
1089       if (NULL != hc->cont)
1090         GNUNET_SCHEDULER_add_now(hc->cont, hc->cont_cls);
1091       GNUNET_free (hc);
1092       return 0;
1093     }
1094 #if DEBUG_TRANSPORT
1095   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1096               "Transmitting `%s' request.\n", "HELLO");
1097 #endif
1098   GNUNET_assert (size >= hc->size);
1099   memcpy (buf, &hc[1], hc->size);
1100
1101   if (hc->cont != NULL)
1102     {
1103       GNUNET_SCHEDULER_add_continuation(hc->cont, hc->cont_cls, GNUNET_SCHEDULER_REASON_PREREQ_DONE);
1104     }
1105   ssize = hc->size;
1106   GNUNET_free (hc);
1107   return ssize;
1108 }
1109
1110
1111 /**
1112  * Offer the transport service the HELLO of another peer.  Note that
1113  * the transport service may just ignore this message if the HELLO is
1114  * malformed or useless due to our local configuration.
1115  *
1116  * @param handle connection to transport service
1117  * @param hello the hello message
1118  * @param cont continuation to call when HELLO has been sent
1119  * @param cls closure for continuation
1120  *
1121  */
1122 void
1123 GNUNET_TRANSPORT_offer_hello (struct GNUNET_TRANSPORT_Handle *handle,
1124                               const struct GNUNET_MessageHeader *hello,
1125                               GNUNET_SCHEDULER_Task cont,
1126                               void *cls)
1127 {
1128   uint16_t size;
1129   struct GNUNET_PeerIdentity peer;
1130   struct HelloContext *hc;
1131
1132   GNUNET_break (ntohs (hello->type) == GNUNET_MESSAGE_TYPE_HELLO);
1133   size = ntohs (hello->size);
1134   GNUNET_break (size >= sizeof (struct GNUNET_MessageHeader));
1135   if (GNUNET_OK != GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message*) hello,
1136                                         &peer))
1137     {
1138       GNUNET_break (0);
1139       return;
1140     }
1141   hc = GNUNET_malloc(sizeof(struct HelloContext) + size);
1142   hc->size = size;
1143   hc->cont = cont;
1144   hc->cont_cls = cls;
1145   memcpy (&hc[1], hello, size);
1146
1147 #if DEBUG_TRANSPORT
1148   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1149               "Offering `%s' message of `%4s' to transport for validation.\n",
1150               "HELLO",
1151               GNUNET_i2s (&peer));
1152 #endif
1153
1154   schedule_control_transmit (handle,
1155                              size,
1156                              GNUNET_NO, OFFER_HELLO_TIMEOUT, &send_hello, hc);
1157 }
1158
1159
1160 /**
1161  * Transmit START message to service.
1162  *
1163  * @param cls unused
1164  * @param size number of bytes available in buf
1165  * @param buf where to copy the message
1166  * @return number of bytes copied to buf
1167  */
1168 static size_t
1169 send_start (void *cls, size_t size, void *buf)
1170 {
1171   struct GNUNET_TRANSPORT_Handle *h = cls;
1172   struct StartMessage s;
1173
1174   if (buf == NULL)
1175     {
1176       /* Can only be shutdown, just give up */
1177 #if DEBUG_TRANSPORT
1178       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1179                   "Shutdown while trying to transmit `%s' request.\n",
1180                   "START");
1181 #endif
1182       return 0;
1183     }
1184 #if DEBUG_TRANSPORT
1185   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1186               "Transmitting `%s' request.\n", "START");
1187 #endif
1188   GNUNET_assert (size >= sizeof (struct StartMessage));
1189   s.header.size = htons (sizeof (struct StartMessage));
1190   s.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_START);
1191   s.do_check = htonl (h->check_self);
1192   s.self = h->self;
1193   memcpy (buf, &s, sizeof (struct StartMessage));
1194   return sizeof (struct StartMessage);
1195 }
1196
1197
1198 /**
1199  * Free neighbour.
1200  *
1201  * @param n the entry to free
1202  */
1203 static void
1204 neighbour_free (struct NeighbourList *n)
1205 {
1206   struct GNUNET_TRANSPORT_Handle *h;
1207
1208   /* Added so task gets canceled when a disconnect is received! */
1209   /* Method 1
1210   if (n->transmit_handle.notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
1211     {
1212       GNUNET_SCHEDULER_cancel(n->transmit_handle.notify_delay_task);
1213       n->transmit_handle.notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
1214       n->transmit_handle.notify = NULL;
1215     }
1216   */
1217   /* NATE: if the above is not needed, then clearly this assertion
1218      should hold (I've checked the code and I'm pretty sure this is
1219      true. -CG
1220      FIXME: remove above comments once we've seen tests pass with the assert... */
1221   GNUNET_assert (n->transmit_handle.notify_delay_task == GNUNET_SCHEDULER_NO_TASK);
1222   GNUNET_assert (n->transmit_handle.notify == NULL);
1223   h = n->h;
1224 #if DEBUG_TRANSPORT
1225   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1226               "Removing neighbour `%s' from list of connected peers.\n",
1227               GNUNET_i2s (&n->id));
1228 #endif
1229   GNUNET_break (n->is_connected == GNUNET_NO);
1230   GNUNET_break (n->transmit_stage == TS_NEW);
1231
1232   GNUNET_assert(GNUNET_YES ==
1233                 GNUNET_CONTAINER_multihashmap_remove(h->neighbours,
1234                                                      &n->id.hashPubKey,
1235                                                      n));
1236   GNUNET_free (n);
1237 }
1238
1239
1240 /**
1241  * Mark neighbour as disconnected.
1242  *
1243  * @param n the entry to mark as disconnected
1244  */
1245 static void
1246 neighbour_disconnect (struct NeighbourList *n)
1247 {
1248   struct GNUNET_TRANSPORT_Handle *h = n->h;
1249 #if DEBUG_TRANSPORT
1250   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1251               "Removing neighbour `%s' from list of connected peers.\n",
1252               GNUNET_i2s (&n->id));
1253 #endif
1254   GNUNET_break (n->is_connected == GNUNET_YES);
1255   n->is_connected = GNUNET_NO;
1256   /* FIXME: this 'in_disconnect' flag is dubious; we should define
1257      clearly what disconnect means for pending 'notify_transmit_ready'
1258      requests; maybe a good approach is to REQUIRE clients to
1259      call 'notify_transmit_ready_cancel' on pending requests on disconnect
1260      and otherwise FAIL HARD with an assertion failure before
1261      'neighbour_free' right here (transmit_stage would be forced
1262      to 'TS_NEW') */
1263   n->in_disconnect = GNUNET_YES;
1264   if (h->nd_cb != NULL)
1265     h->nd_cb (h->cls, &n->id);
1266   if (n->transmit_stage == TS_NEW)
1267     neighbour_free (n);
1268 }
1269
1270
1271 /**
1272  * Function we use for handling incoming messages.
1273  *
1274  * @param cls closure (struct GNUNET_TRANSPORT_Handle *)
1275  * @param msg message received, NULL on timeout or fatal error
1276  */
1277 static void demultiplexer (void *cls,
1278                            const struct GNUNET_MessageHeader *msg);
1279
1280
1281 /**
1282  * Iterator over hash map entries, for getting rid of a neighbor
1283  * upon a reconnect call.
1284  *
1285  * @param cls closure (NULL)
1286  * @param key current key code
1287  * @param value value in the hash map, the neighbour entry to forget
1288  * @return GNUNET_YES if we should continue to
1289  *         iterate,
1290  *         GNUNET_NO if not.
1291  */
1292 static int
1293 forget_neighbours (void *cls,
1294                    const GNUNET_HashCode * key,
1295                    void *value)
1296 {
1297   struct NeighbourList *n = value;
1298
1299 #if DEBUG_TRANSPORT_DISCONNECT
1300   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1301               "Disconnecting due to reconnect being called\n");
1302 #endif
1303   if (n->is_connected)
1304     neighbour_disconnect (n);
1305
1306   return GNUNET_YES;
1307 }
1308
1309
1310 /**
1311  * Try again to connect to transport service.
1312  *
1313  * @param cls the handle to the transport service
1314  * @param tc scheduler context
1315  */
1316 static void
1317 reconnect (void *cls,
1318            const struct GNUNET_SCHEDULER_TaskContext *tc)
1319 {
1320   struct GNUNET_TRANSPORT_Handle *h = cls;
1321   struct ControlMessage *pos;
1322
1323   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1324   if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
1325     {
1326       /* shutdown, just give up */
1327       return;
1328     }
1329   /* Forget about all neighbours that we used to be connected to */
1330   GNUNET_CONTAINER_multihashmap_iterate(h->neighbours,
1331                                         &forget_neighbours,
1332                                         NULL);
1333
1334 #if DEBUG_TRANSPORT
1335   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1336               "Connecting to transport service.\n");
1337 #endif
1338   GNUNET_assert (h->client == NULL);
1339   h->client = GNUNET_CLIENT_connect ("transport", h->cfg);
1340   GNUNET_assert (h->client != NULL);
1341   /* make sure we don't send "START" twice, remove existing entry from
1342      queue (if present) */
1343   pos = h->control_head;
1344   while (pos != NULL)
1345     {
1346       if (pos->notify == &send_start)
1347         {
1348           GNUNET_CONTAINER_DLL_remove (h->control_head,
1349                                        h->control_tail,
1350                                        pos);
1351           if (GNUNET_SCHEDULER_NO_TASK != pos->notify_delay_task)
1352             {
1353               GNUNET_SCHEDULER_cancel (pos->notify_delay_task);
1354               pos->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
1355             }
1356           GNUNET_free (pos);
1357           break;
1358         }
1359       pos = pos->next;
1360     }
1361   schedule_control_transmit (h,
1362                              sizeof (struct StartMessage),
1363                              GNUNET_YES,
1364                              GNUNET_TIME_UNIT_FOREVER_REL, &send_start, h);
1365   GNUNET_CLIENT_receive (h->client,
1366                          &demultiplexer, h, GNUNET_TIME_UNIT_FOREVER_REL);
1367 }
1368
1369
1370 /**
1371  * Function that will schedule the job that will try
1372  * to connect us again to the client.
1373  *
1374  * @param h transport service to reconnect
1375  */
1376 static void
1377 schedule_reconnect (struct GNUNET_TRANSPORT_Handle *h)
1378 {
1379 #if DEBUG_TRANSPORT
1380   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1381               "Scheduling task to reconnect to transport service in %llu ms.\n",
1382               h->reconnect_delay.rel_value);
1383 #endif
1384   GNUNET_assert (h->client == NULL);
1385   GNUNET_assert (h->reconnect_task == GNUNET_SCHEDULER_NO_TASK);
1386   h->reconnect_task
1387     = GNUNET_SCHEDULER_add_delayed (h->reconnect_delay, &reconnect, h);
1388   if (h->reconnect_delay.rel_value == 0)
1389     {
1390       h->reconnect_delay = GNUNET_TIME_UNIT_MILLISECONDS;
1391     }
1392   else
1393     {
1394       h->reconnect_delay = GNUNET_TIME_relative_multiply (h->reconnect_delay, 2);
1395       h->reconnect_delay = GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_SECONDS,
1396                                                      h->reconnect_delay);
1397     }
1398 }
1399
1400
1401 /**
1402  * Send request connect message to the service.
1403  *
1404  * @param cls the TransportRequestConnectMessage
1405  * @param size number of bytes available in buf
1406  * @param buf where to copy the message
1407  * @return number of bytes copied to buf
1408  */
1409 static size_t
1410 send_transport_request_connect (void *cls, size_t size, void *buf)
1411 {
1412   struct TransportRequestConnectMessage *trcm = cls;
1413
1414   if (buf == NULL)
1415     {
1416 #if DEBUG_TRANSPORT
1417       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1418                   "Buffer null for %s\n",
1419                   "REQUEST_CONNECT");
1420 #endif
1421       GNUNET_free (trcm);
1422       return 0;
1423     }
1424 #if DEBUG_TRANSPORT
1425   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1426               "Transmitting `%s' request for `%4s'.\n",
1427               "REQUEST_CONNECT",
1428               GNUNET_i2s (&trcm->peer));
1429 #endif
1430   GNUNET_assert (size >= sizeof (struct TransportRequestConnectMessage));
1431   memcpy(buf, trcm, sizeof(struct TransportRequestConnectMessage));
1432   GNUNET_free (trcm);
1433   return sizeof(struct TransportRequestConnectMessage);
1434 }
1435
1436 /**
1437  * Create and send a request connect message to
1438  * the transport service for a particular peer.
1439  *
1440  * @param h handle to the transport service
1441  * @param n the neighbor to send the request connect message about
1442  *
1443  */
1444 static void
1445 send_request_connect_message(struct GNUNET_TRANSPORT_Handle *h, struct NeighbourList *n)
1446 {
1447   struct TransportRequestConnectMessage *trcm;
1448
1449   trcm = GNUNET_malloc(sizeof(struct TransportRequestConnectMessage));
1450   trcm->header.type = htons(GNUNET_MESSAGE_TYPE_TRANSPORT_REQUEST_CONNECT);
1451   trcm->header.size = htons(sizeof(struct TransportRequestConnectMessage));
1452   memcpy(&trcm->peer, &n->id, sizeof(struct GNUNET_PeerIdentity));
1453   schedule_control_transmit (h,
1454                              sizeof (struct TransportRequestConnectMessage),
1455                              GNUNET_NO,
1456                              GNUNET_TIME_UNIT_FOREVER_REL, &send_transport_request_connect, trcm);
1457 }
1458
1459
1460 /**
1461  * Add neighbour to our list
1462  *
1463  * @return NULL if this API is currently disconnecting from the service
1464  */
1465 static struct NeighbourList *
1466 neighbour_add (struct GNUNET_TRANSPORT_Handle *h,
1467                const struct GNUNET_PeerIdentity *pid)
1468 {
1469   struct NeighbourList *n;
1470
1471   if (GNUNET_YES == h->in_disconnect)
1472     return NULL;
1473   /* check for duplicates */
1474   if (NULL != (n = neighbour_find (h, pid)))
1475     {
1476       GNUNET_break (0);
1477       return n;
1478     }
1479 #if DEBUG_TRANSPORT
1480   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1481               "Creating entry for neighbour `%4s'.\n",
1482               GNUNET_i2s (pid));
1483 #endif
1484   n = GNUNET_malloc (sizeof (struct NeighbourList));
1485   n->id = *pid;
1486   n->h = h;
1487   GNUNET_BANDWIDTH_tracker_init (&n->out_tracker,
1488                                  GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT,
1489                                  MAX_BANDWIDTH_CARRY_S);
1490   GNUNET_CONTAINER_multihashmap_put (h->neighbours,
1491                                      &pid->hashPubKey,
1492                                      n,
1493                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1494
1495   return n;
1496 }
1497
1498
1499 /**
1500  * Iterator over hash map entries, for deleting state of a neighbor.
1501  *
1502  * @param cls closure (NULL)
1503  * @param key current key code
1504  * @param value value in the hash map, the neighbour entry to delete
1505  * @return GNUNET_YES if we should continue to
1506  *         iterate,
1507  *         GNUNET_NO if not.
1508  */
1509 static int
1510 delete_neighbours (void *cls,
1511                    const GNUNET_HashCode * key,
1512                    void *value)
1513 {
1514   struct NeighbourList *n = value;
1515   struct GNUNET_TRANSPORT_TransmitHandle *th;
1516
1517   switch (n->transmit_stage)
1518     {
1519     case TS_NEW:
1520     case TS_TRANSMITTED:
1521       /* nothing to do */
1522       break;
1523     case TS_QUEUED:
1524     case TS_TRANSMITTED_QUEUED:
1525       th = &n->transmit_handle;
1526       if (th->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
1527         {
1528           GNUNET_SCHEDULER_cancel (th->notify_delay_task);
1529           th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
1530         }
1531       GNUNET_assert (0 == th->notify (th->notify_cls, 0, NULL));
1532       break;
1533     default:
1534       GNUNET_break (0);
1535     }
1536   GNUNET_free (n);
1537   return GNUNET_YES;
1538 }
1539
1540
1541 /**
1542  * Connect to the transport service.  Note that the connection may
1543  * complete (or fail) asynchronously.
1544  *
1545  * @param cfg configuration to use
1546  * @param self our own identity (API should check that it matches
1547  *             the identity found by transport), or NULL (no check)
1548  * @param cls closure for the callbacks
1549  * @param rec receive function to call
1550  * @param nc function to call on connect events
1551  * @param nd function to call on disconnect events
1552  */
1553 struct GNUNET_TRANSPORT_Handle *
1554 GNUNET_TRANSPORT_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1555                           const struct GNUNET_PeerIdentity *self,
1556                           void *cls,
1557                           GNUNET_TRANSPORT_ReceiveCallback rec,
1558                           GNUNET_TRANSPORT_NotifyConnect nc,
1559                           GNUNET_TRANSPORT_NotifyDisconnect nd)
1560 {
1561   struct GNUNET_TRANSPORT_Handle *ret;
1562
1563   ret = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_Handle));
1564   if (self != NULL)
1565     {
1566       ret->self = *self;
1567       ret->check_self = GNUNET_YES;
1568     }
1569   ret->cfg = cfg;
1570   ret->cls = cls;
1571   ret->rec = rec;
1572   ret->nc_cb = nc;
1573   ret->nd_cb = nd;
1574   ret->reconnect_delay = GNUNET_TIME_UNIT_ZERO;
1575   ret->neighbours = GNUNET_CONTAINER_multihashmap_create(STARTING_NEIGHBOURS_SIZE);
1576   schedule_reconnect (ret);
1577   return ret;
1578 }
1579
1580
1581 /**
1582  * Disconnect from the transport service.
1583  */
1584 void
1585 GNUNET_TRANSPORT_disconnect (struct GNUNET_TRANSPORT_Handle *handle)
1586 {
1587   struct HelloWaitList *hwl;
1588   struct GNUNET_CLIENT_Connection *client;
1589   struct ControlMessage *cm;
1590
1591 #if DEBUG_TRANSPORT
1592   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Transport disconnect called!\n");
1593 #endif
1594   /* FIXME: this flag is dubious, we should be able to do this
1595      more cleanly; also, we should probably do 'disconnect'
1596      callbacks for every connected peer here, i.e. by calling
1597      the iterator with 'forget_neighbours' instead of 'delete_neighbours'.
1598   */
1599
1600   handle->in_disconnect = GNUNET_YES;
1601
1602   GNUNET_assert (GNUNET_SYSERR !=
1603                  GNUNET_CONTAINER_multihashmap_iterate(handle->neighbours,
1604                                                        &delete_neighbours,
1605                                                        handle));
1606   GNUNET_CONTAINER_multihashmap_destroy (handle->neighbours);
1607
1608   while (NULL != (hwl = handle->hwl_head))
1609     {
1610       handle->hwl_head = hwl->next;
1611       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1612                   _
1613                   ("Disconnect while notification for `%s' still registered.\n"),
1614                   "HELLO");
1615       if (hwl->rec != NULL)
1616         hwl->rec (hwl->rec_cls, NULL);
1617       GNUNET_free (hwl);
1618     }
1619
1620   /* Check for still scheduled control messages, cancel delay tasks if so */
1621   /* Added because somehow a notify_delay_task is remaining scheduled and is ever so annoying */
1622   while ( (NULL != (cm = handle->control_head)))
1623     {
1624 #if DEBUG_TRANSPORT
1625       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1626                   "Disconnect before control message sent!\n");
1627 #endif
1628       if (cm->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
1629         {
1630           GNUNET_SCHEDULER_cancel (cm->notify_delay_task);
1631           cm->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
1632         }
1633       GNUNET_CONTAINER_DLL_remove (handle->control_head,
1634                                    handle->control_tail,
1635                                    cm);
1636       GNUNET_free (cm);
1637     }
1638   /* end check */
1639
1640   if (handle->reconnect_task != GNUNET_SCHEDULER_NO_TASK)
1641     {
1642       GNUNET_SCHEDULER_cancel (handle->reconnect_task);
1643       handle->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1644     }
1645   if (handle->quota_task != GNUNET_SCHEDULER_NO_TASK)
1646     {
1647       GNUNET_SCHEDULER_cancel (handle->quota_task);
1648       handle->quota_task = GNUNET_SCHEDULER_NO_TASK;
1649     }
1650   GNUNET_free_non_null (handle->my_hello);
1651   handle->my_hello = NULL;
1652
1653   if (NULL != handle->network_handle)
1654     {
1655       GNUNET_CLIENT_notify_transmit_ready_cancel (handle->network_handle);
1656       handle->network_handle = NULL;
1657     }
1658   if (NULL != (client = handle->client))
1659     {
1660 #if DEBUG_TRANSPORT
1661       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1662                   "Disconnecting from transport service for good.\n");
1663 #endif
1664       handle->client = NULL;
1665       GNUNET_CLIENT_disconnect (client, GNUNET_YES);
1666     }
1667   GNUNET_free (handle);
1668 }
1669
1670
1671 /**
1672  * Function we use for handling incoming messages.
1673  *
1674  * @param cls closure (struct GNUNET_TRANSPORT_Handle *)
1675  * @param msg message received, NULL on timeout or fatal error
1676  */
1677 static void
1678 demultiplexer (void *cls, const struct GNUNET_MessageHeader *msg)
1679 {
1680   struct GNUNET_TRANSPORT_Handle *h = cls;
1681   const struct DisconnectInfoMessage *dim;
1682   const struct ConnectInfoMessage *cim;
1683   const struct InboundMessage *im;
1684   const struct GNUNET_MessageHeader *imm;
1685   const struct SendOkMessage *okm;
1686   struct HelloWaitList *hwl;
1687   struct HelloWaitList *next_hwl;
1688   struct NeighbourList *n;
1689   struct GNUNET_PeerIdentity me;
1690   uint16_t size;
1691   uint32_t ats_count;
1692
1693   if (h->client == NULL)
1694     {
1695       /* shutdown initiated from 'GNUNET_TRANSPORT_disconnect',
1696          finish clean up work! */
1697       GNUNET_free (h);
1698       return;
1699     }
1700   if (msg == NULL)
1701     {
1702 #if DEBUG_TRANSPORT
1703       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1704                   "Error receiving from transport service, disconnecting temporarily.\n");
1705 #endif
1706       if (h->network_handle != NULL)
1707         {
1708           GNUNET_CLIENT_notify_transmit_ready_cancel (h->network_handle);
1709           h->network_handle = NULL;
1710         }
1711       GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
1712       h->client = NULL;
1713       schedule_reconnect (h);
1714       return;
1715     }
1716   GNUNET_CLIENT_receive (h->client,
1717                          &demultiplexer, h, GNUNET_TIME_UNIT_FOREVER_REL);
1718   size = ntohs (msg->size);
1719   switch (ntohs (msg->type))
1720     {
1721     case GNUNET_MESSAGE_TYPE_HELLO:
1722       if (GNUNET_OK !=
1723           GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *) msg,
1724                                &me))
1725         {
1726           GNUNET_break (0);
1727           break;
1728         }
1729 #if DEBUG_TRANSPORT
1730       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1731                   "Receiving (my own) `%s' message, I am `%4s'.\n",
1732                   "HELLO", GNUNET_i2s (&me));
1733 #endif
1734       GNUNET_free_non_null (h->my_hello);
1735       h->my_hello = NULL;
1736       if (size < sizeof (struct GNUNET_MessageHeader))
1737         {
1738           GNUNET_break (0);
1739           break;
1740         }
1741       h->my_hello = GNUNET_malloc (size);
1742       memcpy (h->my_hello, msg, size);
1743       hwl = h->hwl_head;
1744       while (NULL != hwl)
1745         {
1746           next_hwl = hwl->next;
1747           hwl->rec (hwl->rec_cls,
1748                     (const struct GNUNET_MessageHeader *) h->my_hello);
1749           hwl = next_hwl;
1750         }
1751       break;
1752     case GNUNET_MESSAGE_TYPE_TRANSPORT_CONNECT:
1753       if (size < sizeof (struct ConnectInfoMessage))
1754         {
1755           GNUNET_break (0);
1756           break;
1757         }
1758       cim = (const struct ConnectInfoMessage *) msg;
1759       ats_count = ntohl (cim->ats_count);
1760       if (size != sizeof (struct ConnectInfoMessage) + ats_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information))
1761         {
1762           GNUNET_break (0);
1763           break;
1764         }
1765       if (0 == memcmp (&cim->id,
1766                        &h->self,
1767                        sizeof (struct GNUNET_PeerIdentity)))
1768         {
1769           /* connect to self!? */
1770           GNUNET_break (0);
1771           break;
1772         }
1773 #if DEBUG_TRANSPORT
1774       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1775                   "Receiving `%s' message for `%4s'.\n",
1776                   "CONNECT", GNUNET_i2s (&cim->id));
1777 #endif
1778       n = neighbour_find (h, &cim->id);
1779       if (n == NULL)
1780           n = neighbour_add (h, &cim->id);
1781       if (n == NULL)
1782                  return;
1783       GNUNET_break (n->is_connected == GNUNET_NO);
1784       n->is_connected = GNUNET_YES;
1785       /* FIXME */
1786       if (h->nc_cb != NULL)
1787           h->nc_cb (h->cls, &n->id,
1788                   &cim->ats,ats_count);
1789       /* FIXEND */
1790       break;
1791     case GNUNET_MESSAGE_TYPE_TRANSPORT_DISCONNECT:
1792       if (size != sizeof (struct DisconnectInfoMessage))
1793         {
1794           GNUNET_break (0);
1795           break;
1796         }
1797       dim = (const struct DisconnectInfoMessage *) msg;
1798       GNUNET_break (ntohl (dim->reserved) == 0);
1799       if (0 == memcmp (&dim->peer,
1800                        &h->self,
1801                        sizeof (struct GNUNET_PeerIdentity)))
1802         {
1803           /* discconnect from self!? */
1804           GNUNET_break (0);
1805           break;
1806         }
1807 #if DEBUG_TRANSPORT_DISCONNECT
1808       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1809                   "Receiving `%s' message for `%4s'.\n",
1810                   "DISCONNECT",
1811                   GNUNET_i2s (&dim->peer));
1812 #endif
1813       n = neighbour_find (h, &dim->peer);
1814       GNUNET_break (n != NULL);
1815       if (n != NULL)
1816         neighbour_disconnect (n);
1817       break;
1818     case GNUNET_MESSAGE_TYPE_TRANSPORT_SEND_OK:
1819       if (size != sizeof (struct SendOkMessage))
1820         {
1821           GNUNET_break (0);
1822           break;
1823         }
1824       okm = (const struct SendOkMessage *) msg;
1825 #if DEBUG_TRANSPORT
1826       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1827                   "Receiving `%s' message, transmission %s.\n", "SEND_OK",
1828                   ntohl (okm->success) == GNUNET_OK ? "succeeded" : "failed");
1829 #endif
1830       n = neighbour_find (h, &okm->peer);
1831       GNUNET_assert (n != NULL);
1832       switch (n->transmit_stage)
1833         {
1834         case TS_NEW:
1835           GNUNET_break (0);
1836           break;
1837         case TS_QUEUED:
1838           GNUNET_break (0);
1839           break;
1840         case TS_TRANSMITTED:
1841           n->transmit_stage = TS_NEW;
1842           break;
1843         case TS_TRANSMITTED_QUEUED:
1844           n->transmit_stage = TS_QUEUED;
1845           schedule_transmission (h);
1846           break;
1847         default:
1848           GNUNET_break (0);
1849         }
1850       break;
1851     case GNUNET_MESSAGE_TYPE_TRANSPORT_RECV:
1852 #if DEBUG_TRANSPORT
1853       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1854                   "Receiving `%s' message.\n", "RECV");
1855 #endif
1856       if (size <
1857           sizeof (struct InboundMessage) +
1858           sizeof (struct GNUNET_MessageHeader))
1859         {
1860           GNUNET_break (0);
1861           break;
1862         }
1863       im = (const struct InboundMessage *) msg;
1864       GNUNET_break (0 == ntohl (im->reserved));
1865       ats_count = ntohl(im->ats_count);
1866       //imm = (const struct GNUNET_MessageHeader *) &im[1];
1867       imm = (const struct GNUNET_MessageHeader *) &((&(im->ats))[ats_count+1]);
1868
1869       if (ntohs (imm->size) + sizeof (struct InboundMessage) + ats_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information) != size)
1870         {
1871           GNUNET_break (0);
1872           break;
1873         }
1874 #if DEBUG_TRANSPORT
1875       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1876                   "Received message of type %u from `%4s'.\n",
1877                   ntohs (imm->type), GNUNET_i2s (&im->peer));
1878 #endif
1879       n = neighbour_find (h, &im->peer);
1880       if (n == NULL)
1881         {
1882           GNUNET_break (0);
1883           break;
1884         }
1885       if (n->is_connected != GNUNET_YES)
1886         {
1887           GNUNET_break (0);
1888           break;
1889         }
1890       /* FIXME: */
1891       if (h->rec != NULL)
1892                 h->rec (h->cls, &im->peer, imm,
1893                         &im->ats, ats_count);
1894       /* ENDFIX */
1895       break;
1896     default:
1897       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1898                   _
1899                   ("Received unexpected message of type %u in %s:%u\n"),
1900                   ntohs (msg->type), __FILE__, __LINE__);
1901       GNUNET_break (0);
1902       break;
1903     }
1904 }
1905
1906
1907 /**
1908  * Called when our transmit request timed out before any transport
1909  * reported success connecting to the desired peer or before the
1910  * transport was ready to receive.  Signal error and free
1911  * TransmitHandle.
1912  *
1913  * @param cls the 'struct GNUNET_TRANSPORT_TransmitHandle*' that is timing out
1914  * @param tc scheduler context
1915  */
1916 static void
1917 peer_transmit_timeout (void *cls,
1918                        const struct GNUNET_SCHEDULER_TaskContext *tc)
1919 {
1920   struct GNUNET_TRANSPORT_TransmitHandle *th = cls;
1921   struct NeighbourList *n;
1922   GNUNET_CONNECTION_TransmitReadyNotify notify;
1923   void *notify_cls;
1924
1925   th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
1926   n = th->neighbour;
1927 #if DEBUG_TRANSPORT
1928   GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1929               "Triggering timeout for request to transmit to `%4s' (%d)\n",
1930               GNUNET_i2s (&n->id),
1931               n->transmit_stage);
1932 #endif
1933   notify = th->notify;
1934   th->notify = NULL;
1935   notify_cls = th->notify_cls;
1936   switch (n->transmit_stage)
1937     {
1938     case TS_NEW:
1939       GNUNET_break (0);
1940       break;
1941     case TS_QUEUED:
1942       n->transmit_stage = TS_NEW;
1943       if (n->is_connected == GNUNET_NO)
1944         neighbour_free (n);
1945       break;
1946     case TS_TRANSMITTED:
1947       GNUNET_break (0);
1948       break;
1949     case TS_TRANSMITTED_QUEUED:
1950       n->transmit_stage = TS_TRANSMITTED;
1951       break;
1952     default:
1953       GNUNET_break (0);
1954     }
1955   if (NULL != notify)
1956     notify (notify_cls, 0, NULL);
1957 }
1958
1959
1960 /**
1961  * Check if we could queue a message of the given size for
1962  * transmission.  The transport service will take both its
1963  * internal buffers and bandwidth limits imposed by the
1964  * other peer into consideration when answering this query.
1965  *
1966  * @param handle connection to transport service
1967  * @param target who should receive the message
1968  * @param size how big is the message we want to transmit?
1969  * @param priority how important is the message?
1970  * @param timeout after how long should we give up (and call
1971  *        notify with buf NULL and size 0)?
1972  * @param notify function to call when we are ready to
1973  *        send such a message
1974  * @param notify_cls closure for notify
1975  * @return NULL if someone else is already waiting to be notified
1976  *         non-NULL if the notify callback was queued (can be used to cancel
1977  *         using GNUNET_TRANSPORT_notify_transmit_ready_cancel)
1978  */
1979 struct GNUNET_TRANSPORT_TransmitHandle *
1980 GNUNET_TRANSPORT_notify_transmit_ready (struct GNUNET_TRANSPORT_Handle
1981                                         *handle,
1982                                         const struct GNUNET_PeerIdentity
1983                                         *target, size_t size,
1984                                         uint32_t priority,
1985                                         struct GNUNET_TIME_Relative timeout,
1986                                         GNUNET_CONNECTION_TransmitReadyNotify
1987                                         notify, void *notify_cls)
1988 {
1989   struct GNUNET_TRANSPORT_TransmitHandle *th;
1990   struct NeighbourList *n;
1991
1992   if (size + sizeof (struct OutboundMessage) >=
1993       GNUNET_SERVER_MAX_MESSAGE_SIZE)
1994     {
1995 #if DEBUG_TRANSPORT
1996       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1997                   "Message size is %d, max allowed is %d.\n",
1998                   size + sizeof (struct OutboundMessage), GNUNET_SERVER_MAX_MESSAGE_SIZE - 1);
1999 #endif
2000       GNUNET_break (0);
2001       return NULL;
2002     }
2003 #if DEBUG_TRANSPORT
2004   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2005               "Asking transport service for transmission of %u bytes to peer `%4s' within %llu ms.\n",
2006               size, GNUNET_i2s (target),
2007               (unsigned long long) timeout.rel_value);
2008 #endif
2009   n = neighbour_find (handle, target);
2010   if (n == NULL)
2011     {
2012       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2013                   "Created neighbour entry for peer `%s'\n",
2014                   GNUNET_i2s (target));
2015       n = neighbour_add (handle, target);
2016     }
2017   if (n == NULL)
2018     {
2019       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2020                   "Could not create neighbour entry for peer `%s'\n",
2021                   GNUNET_i2s (target));
2022       return NULL;
2023     }
2024
2025   /**
2026    *  Send a request connect message if not connected,
2027    *  otherwise we will never send anything to
2028    *  transport service
2029    */
2030   if (n->is_connected == GNUNET_NO)
2031     {
2032       send_request_connect_message(handle, n);
2033     }
2034
2035   switch (n->transmit_stage)
2036     {
2037     case TS_NEW:
2038       n->transmit_stage = TS_QUEUED;
2039       break;
2040     case TS_QUEUED:
2041       GNUNET_break (0);
2042       return NULL;
2043     case TS_TRANSMITTED:
2044       n->transmit_stage = TS_TRANSMITTED_QUEUED;
2045       break;
2046     case TS_TRANSMITTED_QUEUED:
2047       GNUNET_break (0);
2048       return NULL;
2049     default:
2050       GNUNET_break (0);
2051       return NULL;
2052     }
2053   th = &n->transmit_handle;
2054   th->neighbour = n;
2055   th->notify = notify;
2056   th->notify_cls = notify_cls;
2057   th->timeout = GNUNET_TIME_relative_to_absolute (timeout);
2058   th->notify_size = size + sizeof (struct OutboundMessage);
2059   th->priority = priority;
2060   th->notify_delay_task
2061     = GNUNET_SCHEDULER_add_delayed (timeout,
2062                                     &peer_transmit_timeout, th);
2063   schedule_transmission (handle);
2064   return th;
2065 }
2066
2067
2068 /**
2069  * Cancel the specified transmission-ready notification.
2070  */
2071 void
2072 GNUNET_TRANSPORT_notify_transmit_ready_cancel (struct
2073                                                GNUNET_TRANSPORT_TransmitHandle
2074                                                *th)
2075 {
2076   struct NeighbourList *n;
2077
2078   th->notify = NULL;
2079   n = th->neighbour;
2080 #if DEBUG_TRANSPORT
2081   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2082               "Transmission request of %u bytes to `%4s' was canceled.\n",
2083               th->notify_size - sizeof (struct OutboundMessage),
2084               GNUNET_i2s (&n->id));
2085 #endif
2086   if (th->notify_delay_task != GNUNET_SCHEDULER_NO_TASK)
2087     {
2088       GNUNET_SCHEDULER_cancel (th->notify_delay_task);
2089       th->notify_delay_task = GNUNET_SCHEDULER_NO_TASK;
2090     }
2091   switch (n->transmit_stage)
2092     {
2093     case TS_NEW:
2094       GNUNET_assert (0);
2095       break;
2096     case TS_QUEUED:
2097       n->transmit_stage = TS_NEW;
2098       if ( (n->in_disconnect == GNUNET_NO) &&
2099            (n->is_connected == GNUNET_NO) )
2100         neighbour_free (n);
2101       break;
2102     case TS_TRANSMITTED:
2103       GNUNET_break (0);
2104       break;
2105     case TS_TRANSMITTED_QUEUED:
2106       n->transmit_stage = TS_TRANSMITTED;
2107       break;
2108     default:
2109       GNUNET_break (0);
2110     }
2111 }
2112
2113
2114 /* end of transport_api.c */