src/core/gnunet-service-core_clients.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009, 2010, 2011 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/gnunet-service-core_clients.c
  23  * @brief code for managing interactions with clients of core service
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_statistics_service.h"
  29 #include "gnunet_transport_service.h"
  30 #include "gnunet-service-core.h"
  31 #include "gnunet-service-core_clients.h"
  32 #include "gnunet-service-core_sessions.h"
  33 #include "gnunet-service-core_typemap.h"
  34 #include "core.h"
  35
  36
  37 /**
  38  * How many messages do we queue up at most for optional
  39  * notifications to a client?  (this can cause notifications
  40  * about outgoing messages to be dropped).
  41  */
  42 #define MAX_NOTIFY_QUEUE 1024
  43
  44
  45 /**
  46  * Data structure for each client connected to the core service.
  47  */
  48 struct GSC_Client
  49 {
  50   /**
  51    * Clients are kept in a linked list.
  52    */
  53   struct GSC_Client *next;
  54
  55   /**
  56    * Clients are kept in a linked list.
  57    */
  58   struct GSC_Client *prev;
  59
  60   /**
  61    * Handle for the client with the server API.
  62    */
  63   struct GNUNET_SERVER_Client *client_handle;
  64
  65   /**
  66    * Array of the types of messages this peer cares
  67    * about (with "tcnt" entries).  Allocated as part
  68    * of this client struct, do not free!
  69    */
  70   const uint16_t *types;
  71
  72   /**
  73    * Map of peer identities to active transmission requests of this
  74    * client to the peer (of type 'struct GSC_ClientActiveRequest').
  75    */
  76   struct GNUNET_CONTAINER_MultiHashMap *requests;
  77
  78   /**
  79    * Map containing all peers that this client knows we're connected to.
  80    */
  81   struct GNUNET_CONTAINER_MultiHashMap *connectmap;
  82
  83   /**
  84    * Options for messages this client cares about,
  85    * see GNUNET_CORE_OPTION_ values.
  86    */
  87   uint32_t options;
  88
  89   /**
  90    * Number of types of incoming messages this client
  91    * specifically cares about.  Size of the "types" array.
  92    */
  93   unsigned int tcnt;
  94
  95 };
  96
  97
  98 /**
  99  * Head of linked list of our clients.
 100  */
 101 static struct GSC_Client *client_head;
 102
 103 /**
 104  * Tail of linked list of our clients.
 105  */
 106 static struct GSC_Client *client_tail;
 107
 108 /**
 109  * Context for notifications we need to send to our clients.
 110  */
 111 static struct GNUNET_SERVER_NotificationContext *notifier;
 112
 113 /**
 114  * Tokenizer for messages received from clients.
 115  */
 116 static struct GNUNET_SERVER_MessageStreamTokenizer *client_mst;
 117
 118
 119 /**
 120  * Lookup our client struct given the server's client handle.
 121  *
 122  * @param client server client handle to look up
 123  * @return our client handle for the client
 124  */
 125 static struct GSC_Client *
 126 find_client (struct GNUNET_SERVER_Client *client)
 127 {
 128   struct GSC_Client *c;
 129
 130   c = client_head;
 131   while ((c != NULL) && (c->client_handle != client))
 132     c = c->next;
 133   return c;
 134 }
 135
 136
 137 /**
 138  * Send a message to one of our clients.
 139  *
 140  * @param client target for the message
 141  * @param msg message to transmit
 142  * @param can_drop could this message be dropped if the
 143  *        client's queue is getting too large?
 144  */
 145 static void
 146 send_to_client (struct GSC_Client *client,
 147                 const struct GNUNET_MessageHeader *msg, int can_drop)
 148 {
 149 #if DEBUG_CORE
 150   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 151               "Preparing to send %u bytes of message of type %u to client.\n",
 152               (unsigned int) ntohs (msg->size),
 153               (unsigned int) ntohs (msg->type));
 154 #endif
 155   GNUNET_SERVER_notification_context_unicast (notifier, client->client_handle,
 156                                               msg, can_drop);
 157 }
 158
 159
 160 /**
 161  * Send a message to one of our clients.
 162  *
 163  * @param client target for the message
 164  * @param msg message to transmit
 165  * @param can_drop could this message be dropped if the
 166  *        client's queue is getting too large?
 167  */
 168 void
 169 GSC_CLIENTS_send_to_client (struct GNUNET_SERVER_Client *client,
 170                             const struct GNUNET_MessageHeader *msg,
 171                             int can_drop)
 172 {
 173   struct GSC_Client *c;
 174
 175   c = find_client (client);
 176   if (NULL == c)
 177   {
 178     GNUNET_break (0);
 179     return;
 180   }
 181   send_to_client (c, msg, can_drop);
 182 }
 183
 184
 185 /**
 186  * Test if the client is interested in messages of the given type.
 187  *
 188  * @param type message type
 189  * @param c client to test
 190  * @return GNUNET_YES if 'c' is interested, GNUNET_NO if not.
 191  */
 192 static int
 193 type_match (uint16_t type, struct GSC_Client *c)
 194 {
 195   unsigned int i;
 196
 197   if (c->tcnt == 0)
 198     return GNUNET_YES;          /* peer without handlers matches ALL */
 199   for (i = 0; i < c->tcnt; i++)
 200     if (type == c->types[i])
 201       return GNUNET_YES;
 202   return GNUNET_NO;
 203 }
 204
 205
 206 /**
 207  * Send a message to all of our current clients that have the right
 208  * options set.
 209  *
 210  * @param msg message to multicast
 211  * @param can_drop can this message be discarded if the queue is too long
 212  * @param options mask to use
 213  * @param type type of the embedded message, 0 for none
 214  */
 215 static void
 216 send_to_all_clients (const struct GNUNET_PeerIdentity *sender,
 217                      const struct GNUNET_MessageHeader *msg, int can_drop,
 218                      int options, uint16_t type)
 219 {
 220   struct GSC_Client *c;
 221
 222   for (c = client_head; c != NULL; c = c->next)
 223   {
 224     if ((0 == (options & GNUNET_CORE_OPTION_SEND_FULL_INBOUND)) &&
 225         (GNUNET_YES == type_match (type, c)))
 226       continue;                 /* not the full message, but we'd like the full one! */
 227     if ((0 == (c->options & options)) && (GNUNET_YES != type_match (type, c)))
 228       continue;                 /* neither options nor type match permit the message */
 229 #if DEBUG_CORE
 230     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 231                 "Sending message to client interested in messages of type %u.\n",
 232                 (unsigned int) type);
 233 #endif
 234     GNUNET_assert (GNUNET_YES ==
 235                    GNUNET_CONTAINER_multihashmap_contains (c->connectmap,
 236                                                            &sender->hashPubKey));
 237     send_to_client (c, msg, can_drop);
 238   }
 239 }
 240
 241
 242 /**
 243  * Handle CORE_INIT request.
 244  *
 245  * @param cls unused
 246  * @param client new client that sent INIT
 247  * @param message the 'struct InitMessage' (presumably)
 248  */
 249 static void
 250 handle_client_init (void *cls, struct GNUNET_SERVER_Client *client,
 251                     const struct GNUNET_MessageHeader *message)
 252 {
 253   const struct InitMessage *im;
 254   struct InitReplyMessage irm;
 255   struct GSC_Client *c;
 256   uint16_t msize;
 257   const uint16_t *types;
 258   uint16_t *wtypes;
 259   unsigned int i;
 260
 261   /* check that we don't have an entry already */
 262   c = find_client (client);
 263   if (NULL != c)
 264   {
 265     GNUNET_break (0);
 266     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 267     return;
 268   }
 269   msize = ntohs (message->size);
 270   if (msize < sizeof (struct InitMessage))
 271   {
 272     GNUNET_break (0);
 273     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 274     return;
 275   }
 276   GNUNET_SERVER_notification_context_add (notifier, client);
 277   im = (const struct InitMessage *) message;
 278   types = (const uint16_t *) &im[1];
 279   msize -= sizeof (struct InitMessage);
 280   c = GNUNET_malloc (sizeof (struct GSC_Client) + msize);
 281   c->client_handle = client;
 282   c->tcnt = msize / sizeof (uint16_t);
 283   c->options = ntohl (im->options);
 284   c->types = (const uint16_t *) &c[1];
 285   c->connectmap = GNUNET_CONTAINER_multihashmap_create (16);
 286   GNUNET_assert (GNUNET_YES ==
 287                  GNUNET_CONTAINER_multihashmap_put (c->connectmap,
 288                                                     &GSC_my_identity.hashPubKey,
 289                                                     NULL,
 290                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 291   wtypes = (uint16_t *) & c[1];
 292   for (i = 0; i < c->tcnt; i++)
 293     wtypes[i] = ntohs (types[i]);
 294   GSC_TYPEMAP_add (wtypes, c->tcnt);
 295   GNUNET_CONTAINER_DLL_insert (client_head, client_tail, c);
 296 #if DEBUG_CORE
 297   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 298               "Client connecting to core service is interested in %u message types\n",
 299               (unsigned int) c->tcnt);
 300 #endif
 301   /* send init reply message */
 302   irm.header.size = htons (sizeof (struct InitReplyMessage));
 303   irm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_INIT_REPLY);
 304   irm.reserved = htonl (0);
 305   irm.my_identity = GSC_my_identity;
 306   send_to_client (c, &irm.header, GNUNET_NO);
 307   GSC_SESSIONS_notify_client_about_sessions (c);
 308   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 309 }
 310
 311
 312 /**
 313  * Handle CORE_SEND_REQUEST message.
 314  *
 315  * @param cls unused
 316  * @param client new client that sent CORE_SEND_REQUEST
 317  * @param message the 'struct SendMessageRequest' (presumably)
 318  */
 319 static void
 320 handle_client_send_request (void *cls, struct GNUNET_SERVER_Client *client,
 321                             const struct GNUNET_MessageHeader *message)
 322 {
 323   const struct SendMessageRequest *req;
 324   struct GSC_Client *c;
 325   struct GSC_ClientActiveRequest *car;
 326   int is_loopback;
 327
 328   req = (const struct SendMessageRequest *) message;
 329   c = find_client (client);
 330   if (c == NULL)
 331   {
 332     /* client did not send INIT first! */
 333     GNUNET_break (0);
 334     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 335     return;
 336   }
 337   if (c->requests == NULL)
 338     c->requests = GNUNET_CONTAINER_multihashmap_create (16);
 339 #if DEBUG_CORE
 340   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 341               "Client asked for transmission to `%s'\n",
 342               GNUNET_i2s (&req->peer));
 343 #endif
 344   is_loopback = (0 ==
 345                  memcmp (&req->peer, &GSC_my_identity,
 346                          sizeof (struct GNUNET_PeerIdentity)));
 347   if ( (! is_loopback) &&
 348        (GNUNET_YES !=
 349         GNUNET_CONTAINER_multihashmap_contains (c->connectmap,
 350                                                 &req->peer.hashPubKey)) )
 351   {
 352     /* neighbour must have disconnected since request was issued,
 353      * ignore (client will realize it once it processes the
 354      * disconnect notification) */
 355     GNUNET_STATISTICS_update (GSC_stats,
 356                               gettext_noop
 357                               ("# send requests dropped (disconnected)"), 1,
 358                               GNUNET_NO);
 359     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 360     return;
 361   }
 362
 363   car = GNUNET_CONTAINER_multihashmap_get (c->requests, &req->peer.hashPubKey);
 364   if (car == NULL)
 365   {
 366     /* create new entry */
 367     car = GNUNET_malloc (sizeof (struct GSC_ClientActiveRequest));
 368     GNUNET_assert (GNUNET_OK ==
 369                    GNUNET_CONTAINER_multihashmap_put (c->requests,
 370                                                       &req->peer.hashPubKey,
 371                                                       car,
 372                                                       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST));
 373     car->client_handle = c;
 374   }
 375   else
 376   {
 377     GSC_SESSIONS_dequeue_request (car);
 378   }
 379   car->target = req->peer;
 380   car->deadline = GNUNET_TIME_absolute_ntoh (req->deadline);
 381   car->priority = ntohl (req->priority);
 382   car->msize = ntohs (req->size);
 383   car->smr_id = req->smr_id;
 384   car->was_solicited = GNUNET_NO;
 385   if (is_loopback)
 386   {
 387     /* loopback, satisfy immediately */
 388     GSC_CLIENTS_solicit_request (car);
 389     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 390     return;
 391   }
 392   GSC_SESSIONS_queue_request (car);
 393   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 394 }
 395
 396
 397 /**
 398  * Closure for the 'client_tokenizer_callback'.
 399  */
 400 struct TokenizerContext
 401 {
 402
 403   /**
 404    * Active request handle for the message.
 405    */
 406   struct GSC_ClientActiveRequest *car;
 407
 408   /**
 409    * Is corking allowed (set only once we have the real message).
 410    */
 411   int cork;
 412
 413 };
 414
 415
 416 /**
 417  * Handle CORE_SEND request.
 418  *
 419  * @param cls unused
 420  * @param client the client issuing the request
 421  * @param message the "struct SendMessage"
 422  */
 423 static void
 424 handle_client_send (void *cls, struct GNUNET_SERVER_Client *client,
 425                     const struct GNUNET_MessageHeader *message)
 426 {
 427   const struct SendMessage *sm;
 428   struct GSC_Client *c;
 429   struct TokenizerContext tc;
 430   uint16_t msize;
 431
 432   msize = ntohs (message->size);
 433   if (msize <
 434       sizeof (struct SendMessage) + sizeof (struct GNUNET_MessageHeader))
 435   {
 436     GNUNET_break (0);
 437     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 438     return;
 439   }
 440   sm = (const struct SendMessage *) message;
 441   msize -= sizeof (struct SendMessage);
 442   GNUNET_break (0 == ntohl (sm->reserved));
 443   c = find_client (client);
 444   if (c == NULL)
 445   {
 446     /* client did not send INIT first! */
 447     GNUNET_break (0);
 448     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 449     return;
 450   }
 451   tc.car =
 452       GNUNET_CONTAINER_multihashmap_get (c->requests, &sm->peer.hashPubKey);
 453   if (NULL == tc.car)
 454   {
 455     /* Must have been that we first approved the request, then got disconnected
 456      * (which triggered removal of the 'car') and now the client gives us a message
 457      * just *before* the client learns about the disconnect.  Theoretically, we
 458      * might also now be *again* connected.  So this can happen (but should be
 459      * rare).  If it does happen, the message is discarded. */
 460     GNUNET_STATISTICS_update (GSC_stats,
 461                               gettext_noop
 462                               ("# messages discarded (session disconnected)"),
 463                               1, GNUNET_NO);
 464     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 465     return;
 466   }
 467   GNUNET_assert (GNUNET_YES ==
 468                  GNUNET_CONTAINER_multihashmap_remove (c->requests,
 469                                                        &sm->peer.hashPubKey,
 470                                                        tc.car));
 471   tc.cork = ntohl (sm->cork);
 472 #if DEBUG_CORE
 473   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 474               "Client asked for transmission of %u bytes to `%s' %s\n", msize,
 475               GNUNET_i2s (&sm->peer), tc.cork ? "now" : "");
 476 #endif
 477   GNUNET_SERVER_mst_receive (client_mst, &tc, (const char *) &sm[1], msize,
 478                              GNUNET_YES, GNUNET_NO);
 479   if (0 !=
 480       memcmp (&tc.car->target, &GSC_my_identity,
 481               sizeof (struct GNUNET_PeerIdentity)))
 482     GSC_SESSIONS_dequeue_request (tc.car);
 483   GNUNET_free (tc.car);
 484   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 485 }
 486
 487
 488 /**
 489  * Functions with this signature are called whenever a complete
 490  * message is received by the tokenizer.  Used by the 'client_mst' for
 491  * dispatching messages from clients to either the SESSION subsystem
 492  * or other CLIENT (for loopback).
 493  *
 494  * @param cls closure
 495  * @param client reservation request ('struct GSC_ClientActiveRequest')
 496  * @param message the actual message
 497  */
 498 static void
 499 client_tokenizer_callback (void *cls, void *client,
 500                            const struct GNUNET_MessageHeader *message)
 501 {
 502   struct TokenizerContext *tc = client;
 503   struct GSC_ClientActiveRequest *car = tc->car;
 504
 505   if (0 ==
 506       memcmp (&car->target, &GSC_my_identity,
 507               sizeof (struct GNUNET_PeerIdentity)))
 508   {
 509 #if DEBUG_CORE
 510     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 511                 "Delivering message of type %u to myself\n",
 512                 ntohs (message->type));
 513 #endif
 514     GSC_CLIENTS_deliver_message (&GSC_my_identity, NULL, 0, message,
 515                                  ntohs (message->size),
 516                                  GNUNET_CORE_OPTION_SEND_FULL_INBOUND |
 517                                  GNUNET_CORE_OPTION_SEND_FULL_OUTBOUND);
 518     GSC_CLIENTS_deliver_message (&GSC_my_identity, NULL, 0, message,
 519                                  sizeof (struct GNUNET_MessageHeader),
 520                                  GNUNET_CORE_OPTION_SEND_HDR_INBOUND |
 521                                  GNUNET_CORE_OPTION_SEND_HDR_OUTBOUND);
 522   }
 523   else
 524   {
 525 #if DEBUG_CORE
 526     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 527                 "Delivering message of type %u to %s\n", ntohs (message->type),
 528                 GNUNET_i2s (&car->target));
 529 #endif
 530     GSC_SESSIONS_transmit (car, message, tc->cork);
 531   }
 532 }
 533
 534
 535 /**
 536  * Free client request records.
 537  *
 538  * @param cls NULL
 539  * @param key identity of peer for which this is an active request
 540  * @param value the 'struct GSC_ClientActiveRequest' to free
 541  * @return GNUNET_YES (continue iteration)
 542  */
 543 static int
 544 destroy_active_client_request (void *cls, const GNUNET_HashCode * key,
 545                                void *value)
 546 {
 547   struct GSC_ClientActiveRequest *car = value;
 548
 549   GNUNET_assert (GNUNET_YES ==
 550                  GNUNET_CONTAINER_multihashmap_remove (car->
 551                                                        client_handle->requests,
 552                                                        &car->target.hashPubKey,
 553                                                        car));
 554   GSC_SESSIONS_dequeue_request (car);
 555   GNUNET_free (car);
 556   return GNUNET_YES;
 557 }
 558
 559
 560 /**
 561  * A client disconnected, clean up.
 562  *
 563  * @param cls closure
 564  * @param client identification of the client
 565  */
 566 static void
 567 handle_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
 568 {
 569   struct GSC_Client *c;
 570
 571   if (client == NULL)
 572     return;
 573 #if DEBUG_CORE
 574   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 575               "Client %p has disconnected from core service.\n", client);
 576 #endif
 577   c = find_client (client);
 578   if (c == NULL)
 579     return;                     /* client never sent INIT */
 580   GNUNET_CONTAINER_DLL_remove (client_head, client_tail, c);
 581   if (c->requests != NULL)
 582   {
 583     GNUNET_CONTAINER_multihashmap_iterate (c->requests,
 584                                            &destroy_active_client_request,
 585                                            NULL);
 586     GNUNET_CONTAINER_multihashmap_destroy (c->requests);
 587   }
 588   GNUNET_CONTAINER_multihashmap_destroy (c->connectmap);
 589   c->connectmap = NULL;
 590   GSC_TYPEMAP_remove (c->types, c->tcnt);
 591   GNUNET_free (c);
 592 }
 593
 594
 595 /**
 596  * Tell a client that we are ready to receive the message.
 597  *
 598  * @param car request that is now ready; the responsibility
 599  *        for the handle remains shared between CLIENTS
 600  *        and SESSIONS after this call.
 601  */
 602 void
 603 GSC_CLIENTS_solicit_request (struct GSC_ClientActiveRequest *car)
 604 {
 605   struct GSC_Client *c;
 606   struct SendMessageReady smr;
 607
 608   c = car->client_handle;
 609   smr.header.size = htons (sizeof (struct SendMessageReady));
 610   smr.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_SEND_READY);
 611   smr.size = htons (car->msize);
 612   smr.smr_id = car->smr_id;
 613   smr.peer = car->target;
 614   GNUNET_assert (GNUNET_YES ==
 615                  GNUNET_CONTAINER_multihashmap_contains (c->connectmap,
 616                                                          &car->
 617                                                          target.hashPubKey));
 618   send_to_client (c, &smr.header, GNUNET_NO);
 619 }
 620
 621
 622 /**
 623  * Tell a client that we will never be ready to receive the
 624  * given message in time (disconnect or timeout).
 625  *
 626  * @param car request that now permanently failed; the
 627  *        responsibility for the handle is now returned
 628  *        to CLIENTS (SESSIONS is done with it).
 629  */
 630 void
 631 GSC_CLIENTS_reject_request (struct GSC_ClientActiveRequest *car)
 632 {
 633   GNUNET_assert (GNUNET_YES ==
 634                  GNUNET_CONTAINER_multihashmap_remove (car->
 635                                                        client_handle->requests,
 636                                                        &car->target.hashPubKey,
 637                                                        car));
 638   GNUNET_free (car);
 639 }
 640
 641
 642 /**
 643  * Notify a particular client about a change to existing connection to
 644  * one of our neighbours (check if the client is interested).  Called
 645  * from 'GSC_SESSIONS_notify_client_about_sessions'.
 646  *
 647  * @param client client to notify
 648  * @param neighbour identity of the neighbour that changed status
 649  * @param atsi performance information about neighbour
 650  * @param atsi_count number of entries in 'ats' array
 651  * @param tmap_old previous type map for the neighbour, NULL for disconnect
 652  * @param tmap_new updated type map for the neighbour, NULL for disconnect
 653  * @param is_new GNUNET_YES if this is a completely new neighbour
 654  */
 655 void
 656 GSC_CLIENTS_notify_client_about_neighbour (struct GSC_Client *client,
 657                                            const struct GNUNET_PeerIdentity
 658                                            *neighbour,
 659                                            const struct GNUNET_ATS_Information
 660                                            *atsi, unsigned int atsi_count,
 661                                            const struct GSC_TypeMap *tmap_old,
 662                                            const struct GSC_TypeMap *tmap_new)
 663 {
 664   struct ConnectNotifyMessage *cnm;
 665   size_t size;
 666   char buf[GNUNET_SERVER_MAX_MESSAGE_SIZE - 1];
 667   struct GNUNET_ATS_Information *a;
 668   struct DisconnectNotifyMessage dcm;
 669   int old_match;
 670   int new_match;
 671
 672   old_match = GSC_TYPEMAP_test_match (tmap_old, client->types, client->tcnt);
 673   new_match = GSC_TYPEMAP_test_match (tmap_new, client->types, client->tcnt);
 674   if (old_match == new_match)
 675   {
 676     GNUNET_assert (old_match ==
 677                    GNUNET_CONTAINER_multihashmap_contains (client->connectmap,
 678                                                            &neighbour->hashPubKey));
 679     return;                     /* no change */
 680   }
 681   if (old_match == GNUNET_NO)
 682   {
 683     /* send connect */
 684     GNUNET_assert (GNUNET_NO ==
 685                    GNUNET_CONTAINER_multihashmap_contains (client->connectmap,
 686                                                            &neighbour->hashPubKey));
 687     GNUNET_assert (GNUNET_YES ==
 688                    GNUNET_CONTAINER_multihashmap_put (client->connectmap,
 689                                                       &neighbour->hashPubKey,
 690                                                       NULL,
 691                                                       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 692     size =
 693         sizeof (struct ConnectNotifyMessage) +
 694         (atsi_count) * sizeof (struct GNUNET_ATS_Information);
 695     if (size >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
 696     {
 697       GNUNET_break (0);
 698       /* recovery strategy: throw away performance data */
 699       atsi_count = 0;
 700       size = sizeof (struct ConnectNotifyMessage);
 701     }
 702     cnm = (struct ConnectNotifyMessage *) buf;
 703     cnm->header.size = htons (size);
 704     cnm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_CONNECT);
 705     cnm->ats_count = htonl (atsi_count);
 706     a = (struct GNUNET_ATS_Information *) &cnm[1];
 707     memcpy (a, atsi, sizeof (struct GNUNET_ATS_Information) * atsi_count);
 708 #if DEBUG_CORE
 709     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message to client.\n",
 710                 "NOTIFY_CONNECT");
 711 #endif
 712     cnm->peer = *neighbour;
 713     send_to_client (client, &cnm->header, GNUNET_NO);
 714   }
 715   else
 716   {
 717     /* send disconnect */
 718     GNUNET_assert (GNUNET_YES ==
 719                    GNUNET_CONTAINER_multihashmap_contains (client->connectmap,
 720                                                            &neighbour->hashPubKey));
 721     GNUNET_assert (GNUNET_YES ==
 722                    GNUNET_CONTAINER_multihashmap_remove (client->connectmap,
 723                                                          &neighbour->hashPubKey,
 724                                                          NULL));
 725     dcm.header.size = htons (sizeof (struct DisconnectNotifyMessage));
 726     dcm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_DISCONNECT);
 727     dcm.reserved = htonl (0);
 728     dcm.peer = *neighbour;
 729     send_to_client (client, &dcm.header, GNUNET_NO);
 730   }
 731 }
 732
 733
 734 /**
 735  * Notify all clients about a change to existing session.
 736  * Called from SESSIONS whenever there is a change in sessions
 737  * or types processed by the respective peer.
 738  *
 739  * @param neighbour identity of the neighbour that changed status
 740  * @param atsi performance information about neighbour
 741  * @param atsi_count number of entries in 'ats' array
 742  * @param tmap_old previous type map for the neighbour, NULL for disconnect
 743  * @param tmap_new updated type map for the neighbour, NULL for disconnect
 744  */
 745 void
 746 GSC_CLIENTS_notify_clients_about_neighbour (const struct GNUNET_PeerIdentity
 747                                             *neighbour,
 748                                             const struct GNUNET_ATS_Information
 749                                             *atsi, unsigned int atsi_count,
 750                                             const struct GSC_TypeMap *tmap_old,
 751                                             const struct GSC_TypeMap *tmap_new)
 752 {
 753   struct GSC_Client *c;
 754
 755   for (c = client_head; c != NULL; c = c->next)
 756     GSC_CLIENTS_notify_client_about_neighbour (c, neighbour, atsi, atsi_count,
 757                                                tmap_old, tmap_new);
 758 }
 759
 760
 761 /**
 762  * Deliver P2P message to interested clients.  Caller must have checked
 763  * that the sending peer actually lists the given message type as one
 764  * of its types.
 765  *
 766  * @param sender peer who sent us the message
 767  * @param atsi performance information about neighbour
 768  * @param atsi_count number of entries in 'ats' array
 769  * @param msg the message
 770  * @param msize number of bytes to transmit
 771  * @param options options for checking which clients should
 772  *        receive the message
 773  */
 774 void
 775 GSC_CLIENTS_deliver_message (const struct GNUNET_PeerIdentity *sender,
 776                              const struct GNUNET_ATS_Information *atsi,
 777                              unsigned int atsi_count,
 778                              const struct GNUNET_MessageHeader *msg,
 779                              uint16_t msize, int options)
 780 {
 781   size_t size =
 782       msize + sizeof (struct NotifyTrafficMessage) +
 783       atsi_count * sizeof (struct GNUNET_ATS_Information);
 784   char buf[size];
 785   struct NotifyTrafficMessage *ntm;
 786   struct GNUNET_ATS_Information *a;
 787
 788   if (0 == options)
 789   {
 790     GNUNET_snprintf (buf, sizeof (buf),
 791                      gettext_noop ("# bytes of messages of type %u received"),
 792                      (unsigned int) ntohs (msg->type));
 793     GNUNET_STATISTICS_update (GSC_stats, buf, msize, GNUNET_NO);
 794   }
 795   if (size >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
 796   {
 797     GNUNET_break (0);
 798     /* recovery strategy: throw performance data away... */
 799     atsi_count = 0;
 800     size = msize + sizeof (struct NotifyTrafficMessage);
 801   }
 802 #if DEBUG_CORE
 803   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 804               "Core service passes message from `%4s' of type %u to client.\n",
 805               GNUNET_i2s (sender), (unsigned int) ntohs (msg->type));
 806 #endif
 807   GSC_SESSIONS_add_to_typemap (sender, ntohs (msg->type));
 808   ntm = (struct NotifyTrafficMessage *) buf;
 809   ntm->header.size = htons (size);
 810   ntm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_INBOUND);
 811   ntm->ats_count = htonl (atsi_count);
 812   ntm->peer = *sender;
 813   a = &ntm->ats;
 814   memcpy (a, atsi, sizeof (struct GNUNET_ATS_Information) * atsi_count);
 815   a[atsi_count].type = htonl (GNUNET_ATS_ARRAY_TERMINATOR);
 816   a[atsi_count].value = htonl (0);
 817   memcpy (&a[atsi_count + 1], msg, msize);
 818   send_to_all_clients (sender, &ntm->header, GNUNET_YES, options,
 819                        ntohs (msg->type));
 820 }
 821
 822
 823 /**
 824  * Initialize clients subsystem.
 825  *
 826  * @param server handle to server clients connect to
 827  */
 828 void
 829 GSC_CLIENTS_init (struct GNUNET_SERVER_Handle *server)
 830 {
 831   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
 832     {&handle_client_init, NULL,
 833      GNUNET_MESSAGE_TYPE_CORE_INIT, 0},
 834     {&GSC_SESSIONS_handle_client_iterate_peers, NULL,
 835      GNUNET_MESSAGE_TYPE_CORE_ITERATE_PEERS,
 836      sizeof (struct GNUNET_MessageHeader)},
 837     {&GSC_SESSIONS_handle_client_have_peer, NULL,
 838      GNUNET_MESSAGE_TYPE_CORE_PEER_CONNECTED,
 839      sizeof (struct GNUNET_MessageHeader) +
 840      sizeof (struct GNUNET_PeerIdentity)},
 841     {&handle_client_send_request, NULL,
 842      GNUNET_MESSAGE_TYPE_CORE_SEND_REQUEST,
 843      sizeof (struct SendMessageRequest)},
 844     {&handle_client_send, NULL,
 845      GNUNET_MESSAGE_TYPE_CORE_SEND, 0},
 846     {NULL, NULL, 0, 0}
 847   };
 848
 849   /* setup notification */
 850   client_mst = GNUNET_SERVER_mst_create (&client_tokenizer_callback, NULL);
 851   notifier =
 852       GNUNET_SERVER_notification_context_create (server, MAX_NOTIFY_QUEUE);
 853   GNUNET_SERVER_disconnect_notify (server, &handle_client_disconnect, NULL);
 854   GNUNET_SERVER_add_handlers (server, handlers);
 855 }
 856
 857
 858 /**
 859  * Shutdown clients subsystem.
 860  */
 861 void
 862 GSC_CLIENTS_done ()
 863 {
 864   struct GSC_Client *c;
 865
 866   while (NULL != (c = client_head))
 867     handle_client_disconnect (NULL, c->client_handle);
 868   if (NULL != notifier)
 869   {
 870     GNUNET_SERVER_notification_context_destroy (notifier);
 871     notifier = NULL;
 872   }
 873   GNUNET_SERVER_mst_destroy (client_mst);
 874   client_mst = NULL;
 875 }
 876
 877 /* end of gnunet-service-core_clients.c */