src/transport/transport_api.c

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