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