src/core/core_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 core/core_api.c
  23  * @brief core service; this is the main API for encrypted P2P
  24  *        communications
  25  * @author Christian Grothoff
  26  */
  27 #include "platform.h"
  28 #include "gnunet_constants.h"
  29 #include "gnunet_core_service.h"
  30 #include "core.h"
  31
  32 #define LOG(kind,...) GNUNET_log_from (kind, "core-api",__VA_ARGS__)
  33
  34 /**
  35  * Information we track for each peer.
  36  */
  37 struct PeerRecord
  38 {
  39
  40   /**
  41    * We generally do NOT keep peer records in a DLL; this
  42    * DLL is only used IF this peer's 'pending_head' message
  43    * is ready for transmission.
  44    */
  45   struct PeerRecord *prev;
  46
  47   /**
  48    * We generally do NOT keep peer records in a DLL; this
  49    * DLL is only used IF this peer's 'pending_head' message
  50    * is ready for transmission.
  51    */
  52   struct PeerRecord *next;
  53
  54   /**
  55    * Peer the record is about.
  56    */
  57   struct GNUNET_PeerIdentity peer;
  58
  59   /**
  60    * Corresponding core handle.
  61    */
  62   struct GNUNET_CORE_Handle *ch;
  63
  64   /**
  65    * Head of doubly-linked list of pending requests.
  66    * Requests are sorted by deadline *except* for HEAD,
  67    * which is only modified upon transmission to core.
  68    */
  69   struct GNUNET_CORE_TransmitHandle *pending_head;
  70
  71   /**
  72    * Tail of doubly-linked list of pending requests.
  73    */
  74   struct GNUNET_CORE_TransmitHandle *pending_tail;
  75
  76   /**
  77    * ID of timeout task for the 'pending_head' handle
  78    * which is the one with the smallest timeout.
  79    */
  80   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
  81
  82   /**
  83    * ID of task to run 'next_request_transmission'.
  84    */
  85   GNUNET_SCHEDULER_TaskIdentifier ntr_task;
  86
  87   /**
  88    * Current size of the queue of pending requests.
  89    */
  90   unsigned int queue_size;
  91
  92   /**
  93    * SendMessageRequest ID generator for this peer.
  94    */
  95   uint16_t smr_id_gen;
  96
  97 };
  98
  99
 100 /**
 101  * Type of function called upon completion.
 102  *
 103  * @param cls closure
 104  * @param success GNUNET_OK on success (which for request_connect
 105  *        ONLY means that we transmitted the connect request to CORE,
 106  *        it does not mean that we are actually now connected!);
 107  *        GNUNET_NO on timeout,
 108  *        GNUNET_SYSERR if core was shut down
 109  */
 110 typedef void (*GNUNET_CORE_ControlContinuation) (void *cls, int success);
 111
 112
 113 /**
 114  * Entry in a doubly-linked list of control messages to be transmitted
 115  * to the core service.  Control messages include traffic allocation,
 116  * connection requests and of course our initial 'init' request.
 117  *
 118  * The actual message is allocated at the end of this struct.
 119  */
 120 struct ControlMessage
 121 {
 122   /**
 123    * This is a doubly-linked list.
 124    */
 125   struct ControlMessage *next;
 126
 127   /**
 128    * This is a doubly-linked list.
 129    */
 130   struct ControlMessage *prev;
 131
 132   /**
 133    * Function to run after transmission failed/succeeded.
 134    */
 135   GNUNET_CORE_ControlContinuation cont;
 136
 137   /**
 138    * Closure for 'cont'.
 139    */
 140   void *cont_cls;
 141
 142   /**
 143    * Transmit handle (if one is associated with this ControlMessage), or NULL.
 144    */
 145   struct GNUNET_CORE_TransmitHandle *th;
 146 };
 147
 148
 149
 150 /**
 151  * Context for the core service connection.
 152  */
 153 struct GNUNET_CORE_Handle
 154 {
 155
 156   /**
 157    * Configuration we're using.
 158    */
 159   const struct GNUNET_CONFIGURATION_Handle *cfg;
 160
 161   /**
 162    * Closure for the various callbacks.
 163    */
 164   void *cls;
 165
 166   /**
 167    * Function to call once we've handshaked with the core service.
 168    */
 169   GNUNET_CORE_StartupCallback init;
 170
 171   /**
 172    * Function to call whenever we're notified about a peer connecting.
 173    */
 174   GNUNET_CORE_ConnectEventHandler connects;
 175
 176   /**
 177    * Function to call whenever we're notified about a peer disconnecting.
 178    */
 179   GNUNET_CORE_DisconnectEventHandler disconnects;
 180
 181   /**
 182    * Function to call whenever we receive an inbound message.
 183    */
 184   GNUNET_CORE_MessageCallback inbound_notify;
 185
 186   /**
 187    * Function to call whenever we receive an outbound message.
 188    */
 189   GNUNET_CORE_MessageCallback outbound_notify;
 190
 191   /**
 192    * Function handlers for messages of particular type.
 193    */
 194   const struct GNUNET_CORE_MessageHandler *handlers;
 195
 196   /**
 197    * Our connection to the service.
 198    */
 199   struct GNUNET_CLIENT_Connection *client;
 200
 201   /**
 202    * Handle for our current transmission request.
 203    */
 204   struct GNUNET_CLIENT_TransmitHandle *cth;
 205
 206   /**
 207    * Head of doubly-linked list of pending requests.
 208    */
 209   struct ControlMessage *control_pending_head;
 210
 211   /**
 212    * Tail of doubly-linked list of pending requests.
 213    */
 214   struct ControlMessage *control_pending_tail;
 215
 216   /**
 217    * Head of doubly-linked list of peers that are core-approved
 218    * to send their next message.
 219    */
 220   struct PeerRecord *ready_peer_head;
 221
 222   /**
 223    * Tail of doubly-linked list of peers that are core-approved
 224    * to send their next message.
 225    */
 226   struct PeerRecord *ready_peer_tail;
 227
 228   /**
 229    * Hash map listing all of the peers that we are currently
 230    * connected to.
 231    */
 232   struct GNUNET_CONTAINER_MultiHashMap *peers;
 233
 234   /**
 235    * Identity of this peer.
 236    */
 237   struct GNUNET_PeerIdentity me;
 238
 239   /**
 240    * ID of reconnect task (if any).
 241    */
 242   GNUNET_SCHEDULER_TaskIdentifier reconnect_task;
 243
 244   /**
 245    * Current delay we use for re-trying to connect to core.
 246    */
 247   struct GNUNET_TIME_Relative retry_backoff;
 248
 249   /**
 250    * Number of messages we are allowed to queue per target.
 251    */
 252   unsigned int queue_size;
 253
 254   /**
 255    * Number of entries in the handlers array.
 256    */
 257   unsigned int hcnt;
 258
 259   /**
 260    * For inbound notifications without a specific handler, do
 261    * we expect to only receive headers?
 262    */
 263   int inbound_hdr_only;
 264
 265   /**
 266    * For outbound notifications without a specific handler, do
 267    * we expect to only receive headers?
 268    */
 269   int outbound_hdr_only;
 270
 271   /**
 272    * Are we currently disconnected and hence unable to forward
 273    * requests?
 274    */
 275   int currently_down;
 276
 277 };
 278
 279
 280 /**
 281  * Handle for a transmission request.
 282  */
 283 struct GNUNET_CORE_TransmitHandle
 284 {
 285
 286   /**
 287    * We keep active transmit handles in a doubly-linked list.
 288    */
 289   struct GNUNET_CORE_TransmitHandle *next;
 290
 291   /**
 292    * We keep active transmit handles in a doubly-linked list.
 293    */
 294   struct GNUNET_CORE_TransmitHandle *prev;
 295
 296   /**
 297    * Corresponding peer record.
 298    */
 299   struct PeerRecord *peer;
 300
 301   /**
 302    * Corresponding SEND_REQUEST message.  Only non-NULL
 303    * while SEND_REQUEST message is pending.
 304    */
 305   struct ControlMessage *cm;
 306
 307   /**
 308    * Function that will be called to get the actual request
 309    * (once we are ready to transmit this request to the core).
 310    * The function will be called with a NULL buffer to signal
 311    * timeout.
 312    */
 313   GNUNET_CONNECTION_TransmitReadyNotify get_message;
 314
 315   /**
 316    * Closure for get_message.
 317    */
 318   void *get_message_cls;
 319
 320   /**
 321    * Timeout for this handle.
 322    */
 323   struct GNUNET_TIME_Absolute timeout;
 324
 325   /**
 326    * How important is this message?
 327    */
 328   uint32_t priority;
 329
 330   /**
 331    * Size of this request.
 332    */
 333   uint16_t msize;
 334
 335   /**
 336    * Send message request ID for this request.
 337    */
 338   uint16_t smr_id;
 339
 340   /**
 341    * Is corking allowed?
 342    */
 343   int cork;
 344
 345 };
 346
 347
 348 /**
 349  * Our current client connection went down.  Clean it up
 350  * and try to reconnect!
 351  *
 352  * @param h our handle to the core service
 353  */
 354 static void
 355 reconnect (struct GNUNET_CORE_Handle *h);
 356
 357
 358 /**
 359  * Task schedule to try to re-connect to core.
 360  *
 361  * @param cls the 'struct GNUNET_CORE_Handle'
 362  * @param tc task context
 363  */
 364 static void
 365 reconnect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 366 {
 367   struct GNUNET_CORE_Handle *h = cls;
 368
 369   h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
 370   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connecting to CORE service after delay\n");
 371   reconnect (h);
 372 }
 373
 374
 375 /**
 376  * Notify clients about disconnect and free
 377  * the entry for connected peer.
 378  *
 379  * @param cls the 'struct GNUNET_CORE_Handle*'
 380  * @param key the peer identity (not used)
 381  * @param value the 'struct PeerRecord' to free.
 382  * @return GNUNET_YES (continue)
 383  */
 384 static int
 385 disconnect_and_free_peer_entry (void *cls, const GNUNET_HashCode * key,
 386                                 void *value)
 387 {
 388   struct GNUNET_CORE_Handle *h = cls;
 389   struct GNUNET_CORE_TransmitHandle *th;
 390   struct PeerRecord *pr = value;
 391
 392   if (pr->timeout_task != GNUNET_SCHEDULER_NO_TASK)
 393   {
 394     GNUNET_SCHEDULER_cancel (pr->timeout_task);
 395     pr->timeout_task = GNUNET_SCHEDULER_NO_TASK;
 396   }
 397   if (pr->ntr_task != GNUNET_SCHEDULER_NO_TASK)
 398   {
 399     GNUNET_SCHEDULER_cancel (pr->ntr_task);
 400     pr->ntr_task = GNUNET_SCHEDULER_NO_TASK;
 401   }
 402   if ((pr->prev != NULL) || (pr->next != NULL) || (h->ready_peer_head == pr))
 403     GNUNET_CONTAINER_DLL_remove (h->ready_peer_head, h->ready_peer_tail, pr);
 404   if (h->disconnects != NULL)
 405     h->disconnects (h->cls, &pr->peer);
 406   /* all requests should have been cancelled, clean up anyway, just in case */
 407   GNUNET_break (pr->queue_size == 0);
 408   while (NULL != (th = pr->pending_head))
 409   {
 410     GNUNET_break (0);
 411     GNUNET_CONTAINER_DLL_remove (pr->pending_head, pr->pending_tail, th);
 412     pr->queue_size--;
 413     if (th->cm != NULL)
 414       th->cm->th = NULL;
 415     GNUNET_free (th);
 416   }
 417   /* done with 'voluntary' cleanups, now on to normal freeing */
 418   GNUNET_assert (GNUNET_YES ==
 419                  GNUNET_CONTAINER_multihashmap_remove (h->peers, key, pr));
 420   GNUNET_assert (pr->pending_head == NULL);
 421   GNUNET_assert (pr->pending_tail == NULL);
 422   GNUNET_assert (pr->ch == h);
 423   GNUNET_assert (pr->queue_size == 0);
 424   GNUNET_assert (pr->timeout_task == GNUNET_SCHEDULER_NO_TASK);
 425   GNUNET_assert (pr->ntr_task == GNUNET_SCHEDULER_NO_TASK);
 426   GNUNET_free (pr);
 427   return GNUNET_YES;
 428 }
 429
 430
 431 /**
 432  * Close down any existing connection to the CORE service and
 433  * try re-establishing it later.
 434  *
 435  * @param h our handle
 436  */
 437 static void
 438 reconnect_later (struct GNUNET_CORE_Handle *h)
 439 {
 440   struct ControlMessage *cm;
 441   struct PeerRecord *pr;
 442
 443   GNUNET_assert (h->reconnect_task == GNUNET_SCHEDULER_NO_TASK);
 444   if (NULL != h->cth)
 445   {
 446     GNUNET_CLIENT_notify_transmit_ready_cancel (h->cth);
 447     h->cth = NULL;
 448   }
 449   if (h->client != NULL)
 450   {
 451     GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
 452     h->client = NULL;
 453   }
 454   h->currently_down = GNUNET_YES;
 455   GNUNET_assert (h->reconnect_task == GNUNET_SCHEDULER_NO_TASK);
 456   h->reconnect_task =
 457       GNUNET_SCHEDULER_add_delayed (h->retry_backoff, &reconnect_task, h);
 458   while (NULL != (cm = h->control_pending_head))
 459   {
 460     GNUNET_CONTAINER_DLL_remove (h->control_pending_head,
 461                                  h->control_pending_tail, cm);
 462     if (cm->th != NULL)
 463       cm->th->cm = NULL;
 464     if (cm->cont != NULL)
 465       cm->cont (cm->cont_cls, GNUNET_NO);
 466     GNUNET_free (cm);
 467   }
 468   GNUNET_CONTAINER_multihashmap_iterate (h->peers,
 469                                          &disconnect_and_free_peer_entry, h);
 470   while (NULL != (pr = h->ready_peer_head))
 471     GNUNET_CONTAINER_DLL_remove (h->ready_peer_head, h->ready_peer_tail, pr);
 472   GNUNET_assert (h->control_pending_head == NULL);
 473   h->retry_backoff =
 474       GNUNET_TIME_relative_min (GNUNET_TIME_UNIT_SECONDS, h->retry_backoff);
 475   h->retry_backoff = GNUNET_TIME_relative_multiply (h->retry_backoff, 2);
 476 }
 477
 478
 479 /**
 480  * Check the list of pending requests, send the next
 481  * one to the core.
 482  *
 483  * @param h core handle
 484  * @param ignore_currently_down transmit message even if not initialized?
 485  */
 486 static void
 487 trigger_next_request (struct GNUNET_CORE_Handle *h, int ignore_currently_down);
 488
 489
 490 /**
 491  * The given request hit its timeout.  Remove from the
 492  * doubly-linked list and call the respective continuation.
 493  *
 494  * @param cls the transmit handle of the request that timed out
 495  * @param tc context, can be NULL (!)
 496  */
 497 static void
 498 transmission_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
 499
 500
 501 /**
 502  * Send a control message to the peer asking for transmission
 503  * of the message in the given peer record.
 504  *
 505  * @param pr peer to request transmission to
 506  */
 507 static void
 508 request_next_transmission (struct PeerRecord *pr)
 509 {
 510   struct GNUNET_CORE_Handle *h = pr->ch;
 511   struct ControlMessage *cm;
 512   struct SendMessageRequest *smr;
 513   struct GNUNET_CORE_TransmitHandle *th;
 514
 515   if (pr->timeout_task != GNUNET_SCHEDULER_NO_TASK)
 516   {
 517     GNUNET_SCHEDULER_cancel (pr->timeout_task);
 518     pr->timeout_task = GNUNET_SCHEDULER_NO_TASK;
 519   }
 520   if (NULL == (th = pr->pending_head))
 521   {
 522     trigger_next_request (h, GNUNET_NO);
 523     return;
 524   }
 525   if (th->cm != NULL)
 526     return;                     /* already done */
 527   GNUNET_assert (pr->prev == NULL);
 528   GNUNET_assert (pr->next == NULL);
 529   pr->timeout_task =
 530       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
 531                                     (th->timeout), &transmission_timeout, pr);
 532   cm = GNUNET_malloc (sizeof (struct ControlMessage) +
 533                       sizeof (struct SendMessageRequest));
 534   th->cm = cm;
 535   cm->th = th;
 536   smr = (struct SendMessageRequest *) &cm[1];
 537   smr->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_SEND_REQUEST);
 538   smr->header.size = htons (sizeof (struct SendMessageRequest));
 539   smr->priority = htonl (th->priority);
 540   smr->deadline = GNUNET_TIME_absolute_hton (th->timeout);
 541   smr->peer = pr->peer;
 542   smr->queue_size = htonl (pr->queue_size);
 543   smr->size = htons (th->msize);
 544   smr->smr_id = htons (th->smr_id = pr->smr_id_gen++);
 545   GNUNET_CONTAINER_DLL_insert_tail (h->control_pending_head,
 546                                     h->control_pending_tail, cm);
 547   LOG (GNUNET_ERROR_TYPE_DEBUG,
 548        "Adding SEND REQUEST for peer `%s' to message queue\n",
 549        GNUNET_i2s (&pr->peer));
 550   trigger_next_request (h, GNUNET_NO);
 551 }
 552
 553
 554 /**
 555  * The given request hit its timeout.  Remove from the
 556  * doubly-linked list and call the respective continuation.
 557  *
 558  * @param cls the transmit handle of the request that timed out
 559  * @param tc context, can be NULL (!)
 560  */
 561 static void
 562 transmission_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 563 {
 564   struct PeerRecord *pr = cls;
 565   struct GNUNET_CORE_Handle *h = pr->ch;
 566   struct GNUNET_CORE_TransmitHandle *th;
 567
 568   pr->timeout_task = GNUNET_SCHEDULER_NO_TASK;
 569   th = pr->pending_head;
 570   GNUNET_CONTAINER_DLL_remove (pr->pending_head, pr->pending_tail, th);
 571   pr->queue_size--;
 572   if ((pr->prev != NULL) || (pr->next != NULL) || (pr == h->ready_peer_head))
 573   {
 574     /* the request that was 'approved' by core was
 575      * canceled before it could be transmitted; remove
 576      * us from the 'ready' list */
 577     GNUNET_CONTAINER_DLL_remove (h->ready_peer_head, h->ready_peer_tail, pr);
 578   }
 579   LOG (GNUNET_ERROR_TYPE_DEBUG,
 580        "Signalling timeout of request for transmission to CORE service\n");
 581   request_next_transmission (pr);
 582   GNUNET_assert (0 == th->get_message (th->get_message_cls, 0, NULL));
 583   GNUNET_free (th);
 584 }
 585
 586
 587 /**
 588  * Transmit the next message to the core service.
 589  */
 590 static size_t
 591 transmit_message (void *cls, size_t size, void *buf)
 592 {
 593   struct GNUNET_CORE_Handle *h = cls;
 594   struct ControlMessage *cm;
 595   struct GNUNET_CORE_TransmitHandle *th;
 596   struct PeerRecord *pr;
 597   struct SendMessage *sm;
 598   const struct GNUNET_MessageHeader *hdr;
 599   uint16_t msize;
 600   size_t ret;
 601
 602   GNUNET_assert (h->reconnect_task == GNUNET_SCHEDULER_NO_TASK);
 603   h->cth = NULL;
 604   if (buf == NULL)
 605   {
 606     LOG (GNUNET_ERROR_TYPE_DEBUG,
 607          "Transmission failed, initiating reconnect\n");
 608     reconnect_later (h);
 609     return 0;
 610   }
 611   /* first check for control messages */
 612   if (NULL != (cm = h->control_pending_head))
 613   {
 614     hdr = (const struct GNUNET_MessageHeader *) &cm[1];
 615     msize = ntohs (hdr->size);
 616     if (size < msize)
 617     {
 618       trigger_next_request (h, GNUNET_NO);
 619       return 0;
 620     }
 621     LOG (GNUNET_ERROR_TYPE_DEBUG,
 622          "Transmitting control message with %u bytes of type %u to core.\n",
 623          (unsigned int) msize, (unsigned int) ntohs (hdr->type));
 624     memcpy (buf, hdr, msize);
 625     GNUNET_CONTAINER_DLL_remove (h->control_pending_head,
 626                                  h->control_pending_tail, cm);
 627     if (cm->th != NULL)
 628       cm->th->cm = NULL;
 629     if (NULL != cm->cont)
 630       cm->cont (cm->cont_cls, GNUNET_OK);
 631     GNUNET_free (cm);
 632     trigger_next_request (h, GNUNET_NO);
 633     return msize;
 634   }
 635   /* now check for 'ready' P2P messages */
 636   if (NULL != (pr = h->ready_peer_head))
 637   {
 638     GNUNET_assert (pr->pending_head != NULL);
 639     th = pr->pending_head;
 640     if (size < th->msize + sizeof (struct SendMessage))
 641     {
 642       trigger_next_request (h, GNUNET_NO);
 643       return 0;
 644     }
 645     GNUNET_CONTAINER_DLL_remove (h->ready_peer_head, h->ready_peer_tail, pr);
 646     GNUNET_CONTAINER_DLL_remove (pr->pending_head, pr->pending_tail, th);
 647     pr->queue_size--;
 648     if (pr->timeout_task != GNUNET_SCHEDULER_NO_TASK)
 649     {
 650       GNUNET_SCHEDULER_cancel (pr->timeout_task);
 651       pr->timeout_task = GNUNET_SCHEDULER_NO_TASK;
 652     }
 653     LOG (GNUNET_ERROR_TYPE_DEBUG,
 654          "Transmitting SEND request to `%s' with %u bytes.\n",
 655          GNUNET_i2s (&pr->peer), (unsigned int) th->msize);
 656     sm = (struct SendMessage *) buf;
 657     sm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_SEND);
 658     sm->priority = htonl (th->priority);
 659     sm->deadline = GNUNET_TIME_absolute_hton (th->timeout);
 660     sm->peer = pr->peer;
 661     sm->cork = htonl ((uint32_t) th->cork);
 662     sm->reserved = htonl (0);
 663     ret =
 664         th->get_message (th->get_message_cls,
 665                          size - sizeof (struct SendMessage), &sm[1]);
 666
 667     LOG (GNUNET_ERROR_TYPE_DEBUG,
 668          "Transmitting SEND request to `%s' yielded %u bytes.\n",
 669          GNUNET_i2s (&pr->peer), ret);
 670     GNUNET_free (th);
 671     if (0 == ret)
 672     {
 673       LOG (GNUNET_ERROR_TYPE_DEBUG,
 674            "Size of clients message to peer %s is 0!\n",
 675            GNUNET_i2s (&pr->peer));
 676       /* client decided to send nothing! */
 677       request_next_transmission (pr);
 678       return 0;
 679     }
 680     LOG (GNUNET_ERROR_TYPE_DEBUG,
 681          "Produced SEND message to core with %u bytes payload\n",
 682          (unsigned int) ret);
 683     GNUNET_assert (ret >= sizeof (struct GNUNET_MessageHeader));
 684     if (ret + sizeof (struct SendMessage) >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
 685     {
 686       GNUNET_break (0);
 687       request_next_transmission (pr);
 688       return 0;
 689     }
 690     ret += sizeof (struct SendMessage);
 691     sm->header.size = htons (ret);
 692     GNUNET_assert (ret <= size);
 693     request_next_transmission (pr);
 694     return ret;
 695   }
 696   return 0;
 697 }
 698
 699
 700 /**
 701  * Check the list of pending requests, send the next
 702  * one to the core.
 703  *
 704  * @param h core handle
 705  * @param ignore_currently_down transmit message even if not initialized?
 706  */
 707 static void
 708 trigger_next_request (struct GNUNET_CORE_Handle *h, int ignore_currently_down)
 709 {
 710   uint16_t msize;
 711
 712   if ((GNUNET_YES == h->currently_down) && (ignore_currently_down == GNUNET_NO))
 713   {
 714     LOG (GNUNET_ERROR_TYPE_DEBUG,
 715          "Core connection down, not processing queue\n");
 716     return;
 717   }
 718   if (NULL != h->cth)
 719   {
 720     LOG (GNUNET_ERROR_TYPE_DEBUG, "Request pending, not processing queue\n");
 721     return;
 722   }
 723   if (h->control_pending_head != NULL)
 724     msize =
 725         ntohs (((struct GNUNET_MessageHeader *) &h->
 726                 control_pending_head[1])->size);
 727   else if (h->ready_peer_head != NULL)
 728     msize =
 729         h->ready_peer_head->pending_head->msize + sizeof (struct SendMessage);
 730   else
 731   {
 732     LOG (GNUNET_ERROR_TYPE_DEBUG,
 733          "Request queue empty, not processing queue\n");
 734     return;                     /* no pending message */
 735   }
 736   h->cth =
 737       GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
 738                                            GNUNET_TIME_UNIT_FOREVER_REL,
 739                                            GNUNET_NO, &transmit_message, h);
 740 }
 741
 742
 743 /**
 744  * Handler for notification messages received from the core.
 745  *
 746  * @param cls our "struct GNUNET_CORE_Handle"
 747  * @param msg the message received from the core service
 748  */
 749 static void
 750 main_notify_handler (void *cls, const struct GNUNET_MessageHeader *msg)
 751 {
 752   struct GNUNET_CORE_Handle *h = cls;
 753   const struct InitReplyMessage *m;
 754   const struct ConnectNotifyMessage *cnm;
 755   const struct DisconnectNotifyMessage *dnm;
 756   const struct NotifyTrafficMessage *ntm;
 757   const struct GNUNET_MessageHeader *em;
 758   const struct SendMessageReady *smr;
 759   const struct GNUNET_CORE_MessageHandler *mh;
 760   const struct GNUNET_ATS_Information *ats;
 761   GNUNET_CORE_StartupCallback init;
 762   struct PeerRecord *pr;
 763   struct GNUNET_CORE_TransmitHandle *th;
 764   unsigned int hpos;
 765   int trigger;
 766   uint16_t msize;
 767   uint16_t et;
 768   uint32_t ats_count;
 769
 770   if (msg == NULL)
 771   {
 772     LOG (GNUNET_ERROR_TYPE_INFO,
 773          _
 774          ("Client was disconnected from core service, trying to reconnect.\n"));
 775     reconnect_later (h);
 776     return;
 777   }
 778   msize = ntohs (msg->size);
 779   LOG (GNUNET_ERROR_TYPE_DEBUG,
 780        "Processing message of type %u and size %u from core service\n",
 781        ntohs (msg->type), msize);
 782   switch (ntohs (msg->type))
 783   {
 784   case GNUNET_MESSAGE_TYPE_CORE_INIT_REPLY:
 785     if (ntohs (msg->size) != sizeof (struct InitReplyMessage))
 786     {
 787       GNUNET_break (0);
 788       reconnect_later (h);
 789       return;
 790     }
 791     m = (const struct InitReplyMessage *) msg;
 792     GNUNET_break (0 == ntohl (m->reserved));
 793     /* start our message processing loop */
 794     if (GNUNET_YES == h->currently_down)
 795     {
 796       h->currently_down = GNUNET_NO;
 797       trigger_next_request (h, GNUNET_NO);
 798     }
 799     h->retry_backoff = GNUNET_TIME_UNIT_MILLISECONDS;
 800     h->me = m->my_identity;
 801     if (NULL != (init = h->init))
 802     {
 803       /* mark so we don't call init on reconnect */
 804       h->init = NULL;
 805       LOG (GNUNET_ERROR_TYPE_DEBUG, "Connected to core service of peer `%s'.\n",
 806            GNUNET_i2s (&h->me));
 807       init (h->cls, h, &h->me);
 808     }
 809     else
 810     {
 811       LOG (GNUNET_ERROR_TYPE_DEBUG,
 812            "Successfully reconnected to core service.\n");
 813     }
 814     /* fake 'connect to self' */
 815     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &h->me.hashPubKey);
 816     GNUNET_assert (pr == NULL);
 817     pr = GNUNET_malloc (sizeof (struct PeerRecord));
 818     pr->peer = h->me;
 819     pr->ch = h;
 820     GNUNET_assert (GNUNET_YES ==
 821                    GNUNET_CONTAINER_multihashmap_put (h->peers,
 822                                                       &h->me.hashPubKey, pr,
 823                                                       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST));
 824     if (NULL != h->connects)
 825       h->connects (h->cls, &h->me, NULL, 0);
 826     break;
 827   case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_CONNECT:
 828     if (msize < sizeof (struct ConnectNotifyMessage))
 829     {
 830       GNUNET_break (0);
 831       reconnect_later (h);
 832       return;
 833     }
 834     cnm = (const struct ConnectNotifyMessage *) msg;
 835     ats_count = ntohl (cnm->ats_count);
 836     if (msize !=
 837         sizeof (struct ConnectNotifyMessage) +
 838         ats_count * sizeof (struct GNUNET_ATS_Information))
 839     {
 840       GNUNET_break (0);
 841       reconnect_later (h);
 842       return;
 843     }
 844     LOG (GNUNET_ERROR_TYPE_DEBUG,
 845          "Received notification about connection from `%s'.\n",
 846          GNUNET_i2s (&cnm->peer));
 847     if (0 == memcmp (&h->me, &cnm->peer, sizeof (struct GNUNET_PeerIdentity)))
 848     {
 849       /* connect to self!? */
 850       GNUNET_break (0);
 851       return;
 852     }
 853     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &cnm->peer.hashPubKey);
 854     if (pr != NULL)
 855     {
 856       GNUNET_break (0);
 857       reconnect_later (h);
 858       return;
 859     }
 860     pr = GNUNET_malloc (sizeof (struct PeerRecord));
 861     pr->peer = cnm->peer;
 862     pr->ch = h;
 863     GNUNET_assert (GNUNET_YES ==
 864                    GNUNET_CONTAINER_multihashmap_put (h->peers,
 865                                                       &cnm->peer.hashPubKey, pr,
 866                                                       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST));
 867     ats = (const struct GNUNET_ATS_Information *) &cnm[1];
 868     if (NULL != h->connects)
 869       h->connects (h->cls, &cnm->peer, ats, ats_count);
 870     break;
 871   case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_DISCONNECT:
 872     if (msize != sizeof (struct DisconnectNotifyMessage))
 873     {
 874       GNUNET_break (0);
 875       reconnect_later (h);
 876       return;
 877     }
 878     dnm = (const struct DisconnectNotifyMessage *) msg;
 879     if (0 == memcmp (&h->me, &dnm->peer, sizeof (struct GNUNET_PeerIdentity)))
 880     {
 881       /* connection to self!? */
 882       GNUNET_break (0);
 883       return;
 884     }
 885     GNUNET_break (0 == ntohl (dnm->reserved));
 886     LOG (GNUNET_ERROR_TYPE_DEBUG,
 887          "Received notification about disconnect from `%s'.\n",
 888          GNUNET_i2s (&dnm->peer));
 889     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &dnm->peer.hashPubKey);
 890     if (pr == NULL)
 891     {
 892       GNUNET_break (0);
 893       reconnect_later (h);
 894       return;
 895     }
 896     trigger = ((pr->prev != NULL) || (pr->next != NULL) ||
 897                (h->ready_peer_head == pr));
 898     disconnect_and_free_peer_entry (h, &dnm->peer.hashPubKey, pr);
 899     if (trigger)
 900       trigger_next_request (h, GNUNET_NO);
 901     break;
 902   case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_INBOUND:
 903     if (msize < sizeof (struct NotifyTrafficMessage))
 904     {
 905       GNUNET_break (0);
 906       reconnect_later (h);
 907       return;
 908     }
 909     ntm = (const struct NotifyTrafficMessage *) msg;
 910
 911     ats_count = ntohl (ntm->ats_count);
 912     if ((msize <
 913          sizeof (struct NotifyTrafficMessage) +
 914          ats_count * sizeof (struct GNUNET_ATS_Information) +
 915          sizeof (struct GNUNET_MessageHeader)) ||
 916         (GNUNET_ATS_ARRAY_TERMINATOR != ntohl ((&ntm->ats)[ats_count].type)))
 917     {
 918       GNUNET_break (0);
 919       reconnect_later (h);
 920       return;
 921     }
 922     em = (const struct GNUNET_MessageHeader *) &(&ntm->ats)[ats_count + 1];
 923     LOG (GNUNET_ERROR_TYPE_DEBUG,
 924          "Received message of type %u and size %u from peer `%4s'\n",
 925          ntohs (em->type), ntohs (em->size), GNUNET_i2s (&ntm->peer));
 926     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &ntm->peer.hashPubKey);
 927     if (pr == NULL)
 928     {
 929       GNUNET_break (0);
 930       reconnect_later (h);
 931       return;
 932     }
 933     if ((GNUNET_NO == h->inbound_hdr_only) &&
 934         (msize !=
 935          ntohs (em->size) + sizeof (struct NotifyTrafficMessage) +
 936          +ats_count * sizeof (struct GNUNET_ATS_Information)))
 937     {
 938       GNUNET_break (0);
 939       reconnect_later (h);
 940       return;
 941     }
 942     et = ntohs (em->type);
 943     for (hpos = 0; hpos < h->hcnt; hpos++)
 944     {
 945       mh = &h->handlers[hpos];
 946       if (mh->type != et)
 947         continue;
 948       if ((mh->expected_size != ntohs (em->size)) && (mh->expected_size != 0))
 949       {
 950         GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 951                     "Unexpected message size %u for message of type %u from peer `%4s'\n",
 952                     htons (em->size), mh->type, GNUNET_i2s (&ntm->peer));
 953         GNUNET_break_op (0);
 954         continue;
 955       }
 956       if (GNUNET_OK !=
 957           h->handlers[hpos].callback (h->cls, &ntm->peer, em, &ntm->ats,
 958                                       ats_count))
 959       {
 960         /* error in processing, do not process other messages! */
 961         break;
 962       }
 963     }
 964     if (NULL != h->inbound_notify)
 965       h->inbound_notify (h->cls, &ntm->peer, em, &ntm->ats, ats_count);
 966     break;
 967   case GNUNET_MESSAGE_TYPE_CORE_NOTIFY_OUTBOUND:
 968     if (msize < sizeof (struct NotifyTrafficMessage))
 969     {
 970       GNUNET_break (0);
 971       reconnect_later (h);
 972       return;
 973     }
 974     ntm = (const struct NotifyTrafficMessage *) msg;
 975     if (0 == memcmp (&h->me, &ntm->peer, sizeof (struct GNUNET_PeerIdentity)))
 976     {
 977       /* self-change!? */
 978       GNUNET_break (0);
 979       return;
 980     }
 981     ats_count = ntohl (ntm->ats_count);
 982     if ((msize <
 983          sizeof (struct NotifyTrafficMessage) +
 984          ats_count * sizeof (struct GNUNET_ATS_Information) +
 985          sizeof (struct GNUNET_MessageHeader)) ||
 986         (GNUNET_ATS_ARRAY_TERMINATOR != ntohl ((&ntm->ats)[ats_count].type)))
 987     {
 988       GNUNET_break (0);
 989       reconnect_later (h);
 990       return;
 991     }
 992     em = (const struct GNUNET_MessageHeader *) &(&ntm->ats)[ats_count + 1];
 993     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &ntm->peer.hashPubKey);
 994     if (pr == NULL)
 995     {
 996       GNUNET_break (0);
 997       reconnect_later (h);
 998       return;
 999     }
1000     LOG (GNUNET_ERROR_TYPE_DEBUG,
1001          "Received notification about transmission to `%s'.\n",
1002          GNUNET_i2s (&ntm->peer));
1003     if ((GNUNET_NO == h->outbound_hdr_only) &&
1004         (msize !=
1005          ntohs (em->size) + sizeof (struct NotifyTrafficMessage) +
1006          ats_count * sizeof (struct GNUNET_ATS_Information)))
1007     {
1008       GNUNET_break (0);
1009       reconnect_later (h);
1010       return;
1011     }
1012     if (NULL == h->outbound_notify)
1013     {
1014       GNUNET_break (0);
1015       break;
1016     }
1017     h->outbound_notify (h->cls, &ntm->peer, em, &ntm->ats, ats_count);
1018     break;
1019   case GNUNET_MESSAGE_TYPE_CORE_SEND_READY:
1020     if (msize != sizeof (struct SendMessageReady))
1021     {
1022       GNUNET_break (0);
1023       reconnect_later (h);
1024       return;
1025     }
1026     smr = (const struct SendMessageReady *) msg;
1027     pr = GNUNET_CONTAINER_multihashmap_get (h->peers, &smr->peer.hashPubKey);
1028     if (pr == NULL)
1029     {
1030       GNUNET_break (0);
1031       reconnect_later (h);
1032       return;
1033     }
1034     LOG (GNUNET_ERROR_TYPE_DEBUG,
1035          "Received notification about transmission readiness to `%s'.\n",
1036          GNUNET_i2s (&smr->peer));
1037     if (pr->pending_head == NULL)
1038     {
1039       /* request must have been cancelled between the original request
1040        * and the response from core, ignore core's readiness */
1041       break;
1042     }
1043
1044     th = pr->pending_head;
1045     if (ntohs (smr->smr_id) != th->smr_id)
1046     {
1047       /* READY message is for expired or cancelled message,
1048        * ignore! (we should have already sent another request) */
1049       break;
1050     }
1051     if ((pr->prev != NULL) || (pr->next != NULL) || (h->ready_peer_head == pr))
1052     {
1053       /* we should not already be on the ready list... */
1054       GNUNET_break (0);
1055       reconnect_later (h);
1056       return;
1057     }
1058     GNUNET_CONTAINER_DLL_insert (h->ready_peer_head, h->ready_peer_tail, pr);
1059     trigger_next_request (h, GNUNET_NO);
1060     break;
1061   default:
1062     reconnect_later (h);
1063     return;
1064   }
1065   GNUNET_CLIENT_receive (h->client, &main_notify_handler, h,
1066                          GNUNET_TIME_UNIT_FOREVER_REL);
1067 }
1068
1069
1070 /**
1071  * Task executed once we are done transmitting the INIT message.
1072  * Starts our 'receive' loop.
1073  *
1074  * @param cls the 'struct GNUNET_CORE_Handle'
1075  * @param success were we successful
1076  */
1077 static void
1078 init_done_task (void *cls, int success)
1079 {
1080   struct GNUNET_CORE_Handle *h = cls;
1081
1082   if (success == GNUNET_SYSERR)
1083     return;                     /* shutdown */
1084   if (success == GNUNET_NO)
1085   {
1086     LOG (GNUNET_ERROR_TYPE_DEBUG,
1087          "Failed to exchange INIT with core, retrying\n");
1088     if (h->reconnect_task == GNUNET_SCHEDULER_NO_TASK)
1089       reconnect_later (h);
1090     return;
1091   }
1092   GNUNET_CLIENT_receive (h->client, &main_notify_handler, h,
1093                          GNUNET_TIME_UNIT_FOREVER_REL);
1094 }
1095
1096
1097 /**
1098  * Our current client connection went down.  Clean it up
1099  * and try to reconnect!
1100  *
1101  * @param h our handle to the core service
1102  */
1103 static void
1104 reconnect (struct GNUNET_CORE_Handle *h)
1105 {
1106   struct ControlMessage *cm;
1107   struct InitMessage *init;
1108   uint32_t opt;
1109   uint16_t msize;
1110   uint16_t *ts;
1111   unsigned int hpos;
1112
1113   GNUNET_assert (h->client == NULL);
1114   GNUNET_assert (h->currently_down == GNUNET_YES);
1115   h->client = GNUNET_CLIENT_connect ("core", h->cfg);
1116   if (h->client == NULL)
1117   {
1118     reconnect_later (h);
1119     return;
1120   }
1121   msize = h->hcnt * sizeof (uint16_t) + sizeof (struct InitMessage);
1122   cm = GNUNET_malloc (sizeof (struct ControlMessage) + msize);
1123   cm->cont = &init_done_task;
1124   cm->cont_cls = h;
1125   init = (struct InitMessage *) &cm[1];
1126   init->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_INIT);
1127   init->header.size = htons (msize);
1128   opt = 0;
1129   if (h->inbound_notify != NULL)
1130   {
1131     if (h->inbound_hdr_only)
1132       opt |= GNUNET_CORE_OPTION_SEND_HDR_INBOUND;
1133     else
1134       opt |= GNUNET_CORE_OPTION_SEND_FULL_INBOUND;
1135   }
1136   if (h->outbound_notify != NULL)
1137   {
1138     if (h->outbound_hdr_only)
1139       opt |= GNUNET_CORE_OPTION_SEND_HDR_OUTBOUND;
1140     else
1141       opt |= GNUNET_CORE_OPTION_SEND_FULL_OUTBOUND;
1142   }
1143   LOG (GNUNET_ERROR_TYPE_INFO,
1144        "(Re)connecting to CORE service, monitoring messages of type %u\n",
1145        opt);
1146
1147   init->options = htonl (opt);
1148   ts = (uint16_t *) & init[1];
1149   for (hpos = 0; hpos < h->hcnt; hpos++)
1150     ts[hpos] = htons (h->handlers[hpos].type);
1151   GNUNET_CONTAINER_DLL_insert (h->control_pending_head, h->control_pending_tail,
1152                                cm);
1153   trigger_next_request (h, GNUNET_YES);
1154 }
1155
1156
1157
1158 /**
1159  * Connect to the core service.  Note that the connection may
1160  * complete (or fail) asynchronously.
1161  *
1162  * @param cfg configuration to use
1163  * @param queue_size size of the per-peer message queue
1164  * @param cls closure for the various callbacks that follow (including handlers in the handlers array)
1165  * @param init callback to call once we have successfully
1166  *        connected to the core service
1167  * @param connects function to call on peer connect, can be NULL
1168  * @param disconnects function to call on peer disconnect / timeout, can be NULL
1169  * @param inbound_notify function to call for all inbound messages, can be NULL
1170  * @param inbound_hdr_only set to GNUNET_YES if inbound_notify will only read the
1171  *                GNUNET_MessageHeader and hence we do not need to give it the full message;
1172  *                can be used to improve efficiency, ignored if inbound_notify is NULLL
1173  * @param outbound_notify function to call for all outbound messages, can be NULL
1174  * @param outbound_hdr_only set to GNUNET_YES if outbound_notify will only read the
1175  *                GNUNET_MessageHeader and hence we do not need to give it the full message
1176  *                can be used to improve efficiency, ignored if outbound_notify is NULLL
1177  * @param handlers callbacks for messages we care about, NULL-terminated
1178  * @return handle to the core service (only useful for disconnect until 'init' is called);
1179  *                NULL on error (in this case, init is never called)
1180  */
1181 struct GNUNET_CORE_Handle *
1182 GNUNET_CORE_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
1183                      unsigned int queue_size, void *cls,
1184                      GNUNET_CORE_StartupCallback init,
1185                      GNUNET_CORE_ConnectEventHandler connects,
1186                      GNUNET_CORE_DisconnectEventHandler disconnects,
1187                      GNUNET_CORE_MessageCallback inbound_notify,
1188                      int inbound_hdr_only,
1189                      GNUNET_CORE_MessageCallback outbound_notify,
1190                      int outbound_hdr_only,
1191                      const struct GNUNET_CORE_MessageHandler *handlers)
1192 {
1193   struct GNUNET_CORE_Handle *h;
1194
1195   h = GNUNET_malloc (sizeof (struct GNUNET_CORE_Handle));
1196   h->cfg = cfg;
1197   h->queue_size = queue_size;
1198   h->cls = cls;
1199   h->init = init;
1200   h->connects = connects;
1201   h->disconnects = disconnects;
1202   h->inbound_notify = inbound_notify;
1203   h->outbound_notify = outbound_notify;
1204   h->inbound_hdr_only = inbound_hdr_only;
1205   h->outbound_hdr_only = outbound_hdr_only;
1206   h->handlers = handlers;
1207   h->hcnt = 0;
1208   h->currently_down = GNUNET_YES;
1209   h->peers = GNUNET_CONTAINER_multihashmap_create (128);
1210   h->retry_backoff = GNUNET_TIME_UNIT_MILLISECONDS;
1211   if (NULL != handlers)
1212     while (handlers[h->hcnt].callback != NULL)
1213       h->hcnt++;
1214   GNUNET_assert (h->hcnt <
1215                  (GNUNET_SERVER_MAX_MESSAGE_SIZE -
1216                   sizeof (struct InitMessage)) / sizeof (uint16_t));
1217   LOG (GNUNET_ERROR_TYPE_DEBUG, "Connecting to CORE service\n");
1218   reconnect (h);
1219   return h;
1220 }
1221
1222
1223 /**
1224  * Disconnect from the core service.  This function can only
1225  * be called *after* all pending 'GNUNET_CORE_notify_transmit_ready'
1226  * requests have been explicitly canceled.
1227  *
1228  * @param handle connection to core to disconnect
1229  */
1230 void
1231 GNUNET_CORE_disconnect (struct GNUNET_CORE_Handle *handle)
1232 {
1233   struct ControlMessage *cm;
1234
1235   LOG (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from CORE service\n");
1236   if (handle->cth != NULL)
1237   {
1238     GNUNET_CLIENT_notify_transmit_ready_cancel (handle->cth);
1239     handle->cth = NULL;
1240   }
1241   while (NULL != (cm = handle->control_pending_head))
1242   {
1243     GNUNET_CONTAINER_DLL_remove (handle->control_pending_head,
1244                                  handle->control_pending_tail, cm);
1245     if (cm->th != NULL)
1246       cm->th->cm = NULL;
1247     if (cm->cont != NULL)
1248       cm->cont (cm->cont_cls, GNUNET_SYSERR);
1249     GNUNET_free (cm);
1250   }
1251   if (handle->client != NULL)
1252   {
1253     GNUNET_CLIENT_disconnect (handle->client, GNUNET_NO);
1254     handle->client = NULL;
1255   }
1256   GNUNET_CONTAINER_multihashmap_iterate (handle->peers,
1257                                          &disconnect_and_free_peer_entry,
1258                                          handle);
1259   if (handle->reconnect_task != GNUNET_SCHEDULER_NO_TASK)
1260   {
1261     GNUNET_SCHEDULER_cancel (handle->reconnect_task);
1262     handle->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
1263   }
1264   GNUNET_CONTAINER_multihashmap_destroy (handle->peers);
1265   handle->peers = NULL;
1266   GNUNET_break (handle->ready_peer_head == NULL);
1267   GNUNET_free (handle);
1268 }
1269
1270
1271 /**
1272  * Task that calls 'request_next_transmission'.
1273  *
1274  * @param cls the 'struct PeerRecord*'
1275  * @param tc scheduler context
1276  */
1277 static void
1278 run_request_next_transmission (void *cls,
1279                                const struct GNUNET_SCHEDULER_TaskContext *tc)
1280 {
1281   struct PeerRecord *pr = cls;
1282
1283   pr->ntr_task = GNUNET_SCHEDULER_NO_TASK;
1284   request_next_transmission (pr);
1285 }
1286
1287
1288 /**
1289  * Ask the core to call "notify" once it is ready to transmit the
1290  * given number of bytes to the specified "target".    Must only be
1291  * called after a connection to the respective peer has been
1292  * established (and the client has been informed about this).
1293  *
1294  * @param handle connection to core service
1295  * @param cork is corking allowed for this transmission?
1296  * @param priority how important is the message?
1297  * @param maxdelay how long can the message wait?
1298  * @param target who should receive the message, never NULL (can be this peer's identity for loopback)
1299  * @param notify_size how many bytes of buffer space does notify want?
1300  * @param notify function to call when buffer space is available
1301  * @param notify_cls closure for notify
1302  * @return non-NULL if the notify callback was queued,
1303  *         NULL if we can not even queue the request (insufficient
1304  *         memory); if NULL is returned, "notify" will NOT be called.
1305  */
1306 struct GNUNET_CORE_TransmitHandle *
1307 GNUNET_CORE_notify_transmit_ready (struct GNUNET_CORE_Handle *handle, int cork,
1308                                    uint32_t priority,
1309                                    struct GNUNET_TIME_Relative maxdelay,
1310                                    const struct GNUNET_PeerIdentity *target,
1311                                    size_t notify_size,
1312                                    GNUNET_CONNECTION_TransmitReadyNotify notify,
1313                                    void *notify_cls)
1314 {
1315   struct PeerRecord *pr;
1316   struct GNUNET_CORE_TransmitHandle *th;
1317   struct GNUNET_CORE_TransmitHandle *pos;
1318   struct GNUNET_CORE_TransmitHandle *prev;
1319   struct GNUNET_CORE_TransmitHandle *minp;
1320
1321   pr = GNUNET_CONTAINER_multihashmap_get (handle->peers, &target->hashPubKey);
1322   if (NULL == pr)
1323   {
1324     /* attempt to send to peer that is not connected */
1325     LOG (GNUNET_ERROR_TYPE_WARNING,
1326          "Attempting to send to peer `%s' from peer `%s', but not connected!\n",
1327          GNUNET_i2s (target), GNUNET_h2s (&handle->me.hashPubKey));
1328     GNUNET_break (0);
1329     return NULL;
1330   }
1331   GNUNET_assert (notify_size + sizeof (struct SendMessage) <
1332                  GNUNET_SERVER_MAX_MESSAGE_SIZE);
1333   th = GNUNET_malloc (sizeof (struct GNUNET_CORE_TransmitHandle));
1334   th->peer = pr;
1335   GNUNET_assert (NULL != notify);
1336   th->get_message = notify;
1337   th->get_message_cls = notify_cls;
1338   th->timeout = GNUNET_TIME_relative_to_absolute (maxdelay);
1339   th->priority = priority;
1340   th->msize = notify_size;
1341   th->cork = cork;
1342   /* bound queue size */
1343   if (pr->queue_size == handle->queue_size)
1344   {
1345     /* find lowest-priority entry, but skip the head of the list */
1346     minp = pr->pending_head->next;
1347     prev = minp;
1348     while (prev != NULL)
1349     {
1350       if (prev->priority < minp->priority)
1351         minp = prev;
1352       prev = prev->next;
1353     }
1354     if (minp == NULL)
1355     {
1356       GNUNET_break (handle->queue_size != 0);
1357       GNUNET_break (pr->queue_size == 1);
1358       GNUNET_free (th);
1359       LOG (GNUNET_ERROR_TYPE_DEBUG,
1360            "Dropping transmission request: cannot drop queue head and limit is one\n");
1361       return NULL;
1362     }
1363     if (priority <= minp->priority)
1364     {
1365       LOG (GNUNET_ERROR_TYPE_DEBUG,
1366            "Dropping transmission request: priority too low\n");
1367       GNUNET_free (th);
1368       return NULL;              /* priority too low */
1369     }
1370     GNUNET_CONTAINER_DLL_remove (pr->pending_head, pr->pending_tail, minp);
1371     pr->queue_size--;
1372     GNUNET_assert (0 == minp->get_message (minp->get_message_cls, 0, NULL));
1373     GNUNET_free (minp);
1374   }
1375
1376   /* Order entries by deadline, but SKIP 'HEAD' (as we may have transmitted
1377    * that request already or might even already be approved to transmit that
1378    * message to core) */
1379   pos = pr->pending_head;
1380   if (pos != NULL)
1381     pos = pos->next;            /* skip head */
1382
1383   /* insertion sort */
1384   prev = pos;
1385   while ((pos != NULL) && (pos->timeout.abs_value < th->timeout.abs_value))
1386   {
1387     prev = pos;
1388     pos = pos->next;
1389   }
1390   GNUNET_CONTAINER_DLL_insert_after (pr->pending_head, pr->pending_tail, prev,
1391                                      th);
1392   pr->queue_size++;
1393   /* was the request queue previously empty? */
1394   LOG (GNUNET_ERROR_TYPE_DEBUG, "Transmission request added to queue\n");
1395   if ((pr->pending_head == th) && (pr->ntr_task == GNUNET_SCHEDULER_NO_TASK) &&
1396       (pr->next == NULL) && (pr->prev == NULL) &&
1397       (handle->ready_peer_head != pr))
1398     pr->ntr_task =
1399         GNUNET_SCHEDULER_add_now (&run_request_next_transmission, pr);
1400   return th;
1401 }
1402
1403
1404 /**
1405  * Cancel the specified transmission-ready notification.
1406  *
1407  * @param th handle that was returned by "notify_transmit_ready".
1408  */
1409 void
1410 GNUNET_CORE_notify_transmit_ready_cancel (struct GNUNET_CORE_TransmitHandle *th)
1411 {
1412   struct PeerRecord *pr = th->peer;
1413   struct GNUNET_CORE_Handle *h = pr->ch;
1414   int was_head;
1415
1416   was_head = (pr->pending_head == th);
1417   GNUNET_CONTAINER_DLL_remove (pr->pending_head, pr->pending_tail, th);
1418   pr->queue_size--;
1419   if (th->cm != NULL)
1420   {
1421     /* we're currently in the control queue, remove */
1422     GNUNET_CONTAINER_DLL_remove (h->control_pending_head,
1423                                  h->control_pending_tail, th->cm);
1424     GNUNET_free (th->cm);
1425   }
1426   GNUNET_free (th);
1427   if (was_head)
1428   {
1429     if ((pr->prev != NULL) || (pr->next != NULL) || (pr == h->ready_peer_head))
1430     {
1431       /* the request that was 'approved' by core was
1432        * canceled before it could be transmitted; remove
1433        * us from the 'ready' list */
1434       GNUNET_CONTAINER_DLL_remove (h->ready_peer_head, h->ready_peer_tail, pr);
1435     }
1436     request_next_transmission (pr);
1437   }
1438 }
1439
1440
1441 /* end of core_api.c */