src/transport/gnunet-service-transport_clients.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 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 transport/gnunet-service-transport_clients.c
  23  * @brief plugin management API
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet-service-transport_blacklist.h"
  28 #include "gnunet-service-transport_clients.h"
  29 #include "gnunet-service-transport_hello.h"
  30 #include "gnunet-service-transport_neighbours.h"
  31 #include "gnunet-service-transport_plugins.h"
  32 #include "gnunet-service-transport_validation.h"
  33 #include "gnunet-service-transport_manipulation.h"
  34 #include "gnunet-service-transport.h"
  35 #include "transport.h"
  36
  37
  38 /**
  39  * How many messages can we have pending for a given client process
  40  * before we start to drop incoming messages?  We typically should
  41  * have only one client and so this would be the primary buffer for
  42   * messages, so the number should be chosen rather generously.
  43  *
  44  * The expectation here is that most of the time the queue is large
  45  * enough so that a drop is virtually never required.  Note that
  46  * this value must be about as large as 'TOTAL_MSGS' in the
  47  * 'test_transport_api_reliability.c', otherwise that testcase may
  48  * fail.
  49  */
  50 #define MAX_PENDING (128 * 1024)
  51
  52
  53 /**
  54  * Linked list of messages to be transmitted to the client.  Each
  55  * entry is followed by the actual message.
  56  */
  57 struct ClientMessageQueueEntry
  58 {
  59   /**
  60    * This is a doubly-linked list.
  61    */
  62   struct ClientMessageQueueEntry *next;
  63
  64   /**
  65    * This is a doubly-linked list.
  66    */
  67   struct ClientMessageQueueEntry *prev;
  68 };
  69
  70
  71 /**
  72  * Client connected to the transport service.
  73  */
  74 struct TransportClient
  75 {
  76
  77   /**
  78    * This is a doubly-linked list.
  79    */
  80   struct TransportClient *next;
  81
  82   /**
  83    * This is a doubly-linked list.
  84    */
  85   struct TransportClient *prev;
  86
  87   /**
  88    * Handle to the client.
  89    */
  90   struct GNUNET_SERVER_Client *client;
  91
  92   /**
  93    * Linked list of messages yet to be transmitted to
  94    * the client.
  95    */
  96   struct ClientMessageQueueEntry *message_queue_head;
  97
  98   /**
  99    * Tail of linked list of messages yet to be transmitted to the
 100    * client.
 101    */
 102   struct ClientMessageQueueEntry *message_queue_tail;
 103
 104   /**
 105    * Current transmit request handle.
 106    */
 107   struct GNUNET_SERVER_TransmitHandle *th;
 108
 109   /**
 110    * Length of the list of messages pending for this client.
 111    */
 112   unsigned int message_count;
 113
 114   /**
 115    * Is this client interested in payload messages?
 116    */
 117   int send_payload;
 118 };
 119
 120 /**
 121  * Context for address to string operations
 122  */
 123 struct AddressToStringContext
 124 {
 125   /**
 126    * This is a doubly-linked list.
 127    */
 128   struct AddressToStringContext *next;
 129
 130   /**
 131    * This is a doubly-linked list.
 132    */
 133   struct AddressToStringContext *prev;
 134
 135   /**
 136    * Transmission context
 137    */
 138   struct GNUNET_SERVER_TransmitContext* tc;
 139 };
 140
 141 /**
 142  * Client monitoring changes of active addresses of our neighbours.
 143  */
 144 struct MonitoringClient
 145 {
 146   /**
 147    * This is a doubly-linked list.
 148    */
 149   struct MonitoringClient *next;
 150
 151   /**
 152    * This is a doubly-linked list.
 153    */
 154   struct MonitoringClient *prev;
 155
 156   /**
 157    * Handle to the client.
 158    */
 159   struct GNUNET_SERVER_Client *client;
 160
 161   /**
 162    * Peer identity to monitor the addresses of.
 163    * Zero to monitor all neighrours.
 164    */
 165   struct GNUNET_PeerIdentity peer;
 166
 167 };
 168
 169
 170 /**
 171  * Head of linked list of all clients to this service.
 172  */
 173 static struct TransportClient *clients_head;
 174
 175 /**
 176  * Tail of linked list of all clients to this service.
 177  */
 178 static struct TransportClient *clients_tail;
 179
 180 /**
 181  * Head of linked list of all pending address iterations
 182  */
 183 struct AddressToStringContext *a2s_head;
 184
 185 /**
 186  * Tail of linked list of all pending address iterations
 187  */
 188 struct AddressToStringContext *a2s_tail;
 189
 190 /**
 191  * Head of linked list of monitoring clients.
 192  */
 193 static struct MonitoringClient *monitoring_clients_head;
 194
 195 /**
 196  * Tail of linked list of monitoring clients.
 197  */
 198 static struct MonitoringClient *monitoring_clients_tail;
 199
 200 /**
 201  * Notification context, to send updates on changes to active addresses
 202  * of our neighbours.
 203  */
 204 struct GNUNET_SERVER_NotificationContext *nc = NULL;
 205
 206
 207 /**
 208  * Find the internal handle associated with the given client handle
 209  *
 210  * @param client server's client handle to look up
 211  * @return internal client handle
 212  */
 213 static struct TransportClient *
 214 lookup_client (struct GNUNET_SERVER_Client *client)
 215 {
 216   struct TransportClient *tc;
 217
 218   tc = clients_head;
 219   while (tc != NULL)
 220   {
 221     if (tc->client == client)
 222       return tc;
 223     tc = tc->next;
 224   }
 225   return NULL;
 226 }
 227
 228
 229 /**
 230  * Create the internal handle for the given server client handle
 231  *
 232  * @param client server's client handle to create our internal handle for
 233  * @return fresh internal client handle
 234  */
 235 static struct TransportClient *
 236 setup_client (struct GNUNET_SERVER_Client *client)
 237 {
 238   struct TransportClient *tc;
 239
 240   GNUNET_assert (lookup_client (client) == NULL);
 241   tc = GNUNET_malloc (sizeof (struct TransportClient));
 242   tc->client = client;
 243   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Client %p connected\n", tc);
 244   return tc;
 245 }
 246
 247
 248 /**
 249  * Find the handle to the monitoring client associated with the given
 250  * client handle
 251  *
 252  * @param client server's client handle to look up
 253  * @return handle to the monitoring client
 254  */
 255 static struct MonitoringClient *
 256 lookup_monitoring_client (struct GNUNET_SERVER_Client *client)
 257 {
 258   struct MonitoringClient *mc;
 259
 260   mc = monitoring_clients_head;
 261   while (mc != NULL)
 262   {
 263     if (mc->client == client)
 264       return mc;
 265     mc = mc->next;
 266   }
 267   return NULL;
 268 }
 269
 270
 271 /**
 272  * Setup a new monitoring client using the given server client handle and
 273  * the peer identity.
 274  *
 275  * @param client server's client handle to create our internal handle for
 276  * @param peer identity of the peer to monitor the addresses of,
 277  *             zero to monitor all neighrours.
 278  * @return handle to the new monitoring client
 279  */
 280 static struct MonitoringClient *
 281 setup_monitoring_client (struct GNUNET_SERVER_Client *client,
 282                          struct GNUNET_PeerIdentity *peer)
 283 {
 284   struct MonitoringClient *mc;
 285   static struct GNUNET_PeerIdentity all_zeros;
 286
 287   GNUNET_assert (lookup_monitoring_client (client) == NULL);
 288   mc = GNUNET_malloc (sizeof (struct MonitoringClient));
 289   mc->client = client;
 290   mc->peer = *peer;
 291   GNUNET_CONTAINER_DLL_insert (monitoring_clients_head,
 292                                monitoring_clients_tail,
 293                                mc);
 294   GNUNET_SERVER_notification_context_add (nc, client);
 295
 296   if (0 != memcmp (peer, &all_zeros, sizeof (struct GNUNET_PeerIdentity)))
 297     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 298                 "Client %p started monitoring of the peer `%s'\n",
 299                 mc, GNUNET_i2s (peer));
 300   else
 301     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 302               "Client %p started monitoring all peers\n", mc);
 303   return mc;
 304 }
 305
 306
 307 /**
 308  * Function called to notify a client about the socket being ready to
 309  * queue more data.  "buf" will be NULL and "size" zero if the socket
 310  * was closed for writing in the meantime.
 311  *
 312  * @param cls closure
 313  * @param size number of bytes available in buf
 314  * @param buf where the callee should write the message
 315  * @return number of bytes written to buf
 316  */
 317 static size_t
 318 transmit_to_client_callback (void *cls, size_t size, void *buf)
 319 {
 320   struct TransportClient *tc = cls;
 321   struct ClientMessageQueueEntry *q;
 322   const struct GNUNET_MessageHeader *msg;
 323   char *cbuf;
 324   uint16_t msize;
 325   size_t tsize;
 326
 327   tc->th = NULL;
 328   if (buf == NULL)
 329   {
 330     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 331                 "Transmission to client failed, closing connection.\n");
 332     return 0;
 333   }
 334   cbuf = buf;
 335   tsize = 0;
 336   while (NULL != (q = tc->message_queue_head))
 337   {
 338     msg = (const struct GNUNET_MessageHeader *) &q[1];
 339     msize = ntohs (msg->size);
 340     if (msize + tsize > size)
 341       break;
 342     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 343                 "Transmitting message of type %u to client %p.\n",
 344                 ntohs (msg->type), tc);
 345     GNUNET_CONTAINER_DLL_remove (tc->message_queue_head, tc->message_queue_tail,
 346                                  q);
 347     tc->message_count--;
 348     memcpy (&cbuf[tsize], msg, msize);
 349     GNUNET_free (q);
 350     tsize += msize;
 351   }
 352   if (NULL != q)
 353   {
 354     GNUNET_assert (msize >= sizeof (struct GNUNET_MessageHeader));
 355     tc->th =
 356         GNUNET_SERVER_notify_transmit_ready (tc->client, msize,
 357                                              GNUNET_TIME_UNIT_FOREVER_REL,
 358                                              &transmit_to_client_callback, tc);
 359     GNUNET_assert (tc->th != NULL);
 360   }
 361   return tsize;
 362 }
 363
 364
 365 /**
 366  * Queue the given message for transmission to the given client
 367  *
 368  * @param tc target of the message
 369  * @param msg message to transmit
 370  * @param may_drop GNUNET_YES if the message can be dropped
 371  */
 372 static void
 373 unicast (struct TransportClient *tc, const struct GNUNET_MessageHeader *msg,
 374          int may_drop)
 375 {
 376   struct ClientMessageQueueEntry *q;
 377   uint16_t msize;
 378
 379   if (msg == NULL)
 380   {
 381     GNUNET_break (0);
 382     return;
 383   }
 384
 385   if ((tc->message_count >= MAX_PENDING) && (GNUNET_YES == may_drop))
 386   {
 387     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 388                 _
 389                 ("Dropping message of type %u and size %u, have %u/%u messages pending\n"),
 390                 ntohs (msg->type), ntohs (msg->size), tc->message_count,
 391                 MAX_PENDING);
 392     GNUNET_STATISTICS_update (GST_stats,
 393                               gettext_noop
 394                               ("# messages dropped due to slow client"), 1,
 395                               GNUNET_NO);
 396     return;
 397   }
 398   msize = ntohs (msg->size);
 399   GNUNET_assert (msize >= sizeof (struct GNUNET_MessageHeader));
 400   q = GNUNET_malloc (sizeof (struct ClientMessageQueueEntry) + msize);
 401   memcpy (&q[1], msg, msize);
 402   GNUNET_CONTAINER_DLL_insert_tail (tc->message_queue_head,
 403                                     tc->message_queue_tail, q);
 404   tc->message_count++;
 405   if (tc->th != NULL)
 406     return;
 407   tc->th =
 408       GNUNET_SERVER_notify_transmit_ready (tc->client, msize,
 409                                            GNUNET_TIME_UNIT_FOREVER_REL,
 410                                            &transmit_to_client_callback, tc);
 411   GNUNET_assert (tc->th != NULL);
 412 }
 413
 414
 415 /**
 416  * Called whenever a client is disconnected.  Frees our
 417  * resources associated with that client.
 418  *
 419  * @param cls closure
 420  * @param client identification of the client
 421  */
 422 static void
 423 client_disconnect_notification (void *cls, struct GNUNET_SERVER_Client *client)
 424 {
 425   struct TransportClient *tc;
 426   struct MonitoringClient *mc;
 427   struct ClientMessageQueueEntry *mqe;
 428
 429   if (client == NULL)
 430     return;
 431   mc = lookup_monitoring_client (client);
 432   if (mc != NULL)
 433   {
 434     GNUNET_CONTAINER_DLL_remove (monitoring_clients_head,
 435                                  monitoring_clients_tail,
 436                                  mc);
 437     GNUNET_free (mc);
 438   }
 439   tc = lookup_client (client);
 440   if (tc == NULL)
 441     return;
 442   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
 443               "Client %p disconnected, cleaning up.\n", tc);
 444   while (NULL != (mqe = tc->message_queue_head))
 445   {
 446     GNUNET_CONTAINER_DLL_remove (tc->message_queue_head, tc->message_queue_tail,
 447                                  mqe);
 448     tc->message_count--;
 449     GNUNET_free (mqe);
 450   }
 451   GNUNET_CONTAINER_DLL_remove (clients_head, clients_tail, tc);
 452   if (tc->th != NULL)
 453   {
 454     GNUNET_SERVER_notify_transmit_ready_cancel (tc->th);
 455     tc->th = NULL;
 456   }
 457   GNUNET_break (0 == tc->message_count);
 458   GNUNET_free (tc);
 459 }
 460
 461
 462 /**
 463  * Function called for each of our connected neighbours.  Notify the
 464  * client about the existing neighbour.
 465  *
 466  * @param cls the 'struct TransportClient' to notify
 467  * @param peer identity of the neighbour
 468  * @param ats performance data
 469  * @param ats_count number of entries in ats (excluding 0-termination)
 470  * @param address the address
 471  * @param bandwidth_in inbound bandwidth in NBO
 472  * @param bandwidth_out outbound bandwidth in NBO
 473  */
 474 static void
 475 notify_client_about_neighbour (void *cls,
 476                                const struct GNUNET_PeerIdentity *peer,
 477                                const struct GNUNET_ATS_Information *ats,
 478                                uint32_t ats_count,
 479                                const struct GNUNET_HELLO_Address *address,
 480                                struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
 481                                struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out)
 482 {
 483   struct TransportClient *tc = cls;
 484   struct ConnectInfoMessage *cim;
 485   struct GNUNET_ATS_Information *ap;
 486   size_t size =
 487       sizeof (struct ConnectInfoMessage) +
 488       ats_count * sizeof (struct GNUNET_ATS_Information);
 489   char buf[size] GNUNET_ALIGN;
 490
 491   GNUNET_assert (size < GNUNET_SERVER_MAX_MESSAGE_SIZE);
 492   cim = (struct ConnectInfoMessage *) buf;
 493   cim->header.size = htons (size);
 494   cim->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_CONNECT);
 495   cim->ats_count = htonl (ats_count);
 496   cim->id = *peer;
 497   cim->quota_in = bandwidth_in;
 498   cim->quota_out = bandwidth_out;
 499   ap = (struct GNUNET_ATS_Information *) &cim[1];
 500   memcpy (ap, ats, ats_count * sizeof (struct GNUNET_ATS_Information));
 501   unicast (tc, &cim->header, GNUNET_NO);
 502 }
 503
 504
 505 /**
 506  * Initialize a normal client.  We got a start message from this
 507  * client, add him to the list of clients for broadcasting of inbound
 508  * messages.
 509  *
 510  * @param cls unused
 511  * @param client the client
 512  * @param message the start message that was sent
 513  */
 514 static void
 515 clients_handle_start (void *cls, struct GNUNET_SERVER_Client *client,
 516                       const struct GNUNET_MessageHeader *message)
 517 {
 518   const struct StartMessage *start;
 519   struct TransportClient *tc;
 520   uint32_t options;
 521
 522   tc = lookup_client (client);
 523
 524   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
 525               "Client %p sent START\n", tc);
 526   if (tc != NULL)
 527   {
 528     /* got 'start' twice from the same client, not allowed */
 529     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
 530                 "TransportClient %p ServerClient %p sent multiple START messages\n",
 531                 tc, tc->client);
 532     GNUNET_break (0);
 533     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 534     return;
 535   }
 536   start = (const struct StartMessage *) message;
 537   options = ntohl (start->options);
 538   if ((0 != (1 & options)) &&
 539       (0 !=
 540        memcmp (&start->self, &GST_my_identity,
 541                sizeof (struct GNUNET_PeerIdentity))))
 542   {
 543     /* client thinks this is a different peer, reject */
 544     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 545                 _
 546                 ("Rejecting control connection from peer `%s', which is not me!\n"),
 547                 GNUNET_i2s (&start->self));
 548     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 549     return;
 550   }
 551   tc = setup_client (client);
 552   tc->send_payload = (0 != (2 & options));
 553   unicast (tc, GST_hello_get (), GNUNET_NO);
 554   GST_neighbours_iterate (&notify_client_about_neighbour, tc);
 555   GNUNET_CONTAINER_DLL_insert (clients_head, clients_tail, tc);
 556   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 557 }
 558
 559
 560 /**
 561  * Client sent us a HELLO.  Process the request.
 562  *
 563  * @param cls unused
 564  * @param client the client
 565  * @param message the HELLO message
 566  */
 567 static void
 568 clients_handle_hello (void *cls, struct GNUNET_SERVER_Client *client,
 569                       const struct GNUNET_MessageHeader *message)
 570 {
 571   GST_validation_handle_hello (message);
 572   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 573 }
 574
 575
 576 /**
 577  * Closure for 'handle_send_transmit_continuation'
 578  */
 579 struct SendTransmitContinuationContext
 580 {
 581   /**
 582    * Client that made the request.
 583    */
 584   struct GNUNET_SERVER_Client *client;
 585
 586   /**
 587    * Peer that was the target.
 588    */
 589   struct GNUNET_PeerIdentity target;
 590 };
 591
 592
 593 /**
 594  * Function called after the transmission is done.  Notify the client that it is
 595  * OK to send the next message.
 596  *
 597  * @param cls closure
 598  * @param success GNUNET_OK on success, GNUNET_NO on failure, GNUNET_SYSERR if we're not connected
 599  * @param bytes_payload bytes payload sent
 600  * @param bytes_on_wire bytes sent on wire
 601  */
 602 static void
 603 handle_send_transmit_continuation (void *cls, int success,
 604                                    size_t bytes_payload, size_t bytes_on_wire)
 605 {
 606   struct SendTransmitContinuationContext *stcc = cls;
 607   struct SendOkMessage send_ok_msg;
 608
 609   //GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Payload: %u, On wire %u result: %i\n", bytes_payload, bytes_on_wire, success);
 610   /*
 611   if (GNUNET_OK == success)
 612     GNUNET_assert (bytes_on_wire >= bytes_payload);
 613
 614   else
 615     GNUNET_assert (bytes_on_wire <= bytes_payload);
 616 */
 617   send_ok_msg.header.size = htons (sizeof (send_ok_msg));
 618   send_ok_msg.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_SEND_OK);
 619   send_ok_msg.bytes_msg = htonl (bytes_payload);
 620   send_ok_msg.bytes_physical = htonl (bytes_on_wire);
 621   send_ok_msg.success = htonl (success);
 622   send_ok_msg.latency =
 623       GNUNET_TIME_relative_hton (GNUNET_TIME_UNIT_FOREVER_REL);
 624   send_ok_msg.peer = stcc->target;
 625   GST_clients_unicast (stcc->client, &send_ok_msg.header, GNUNET_NO);
 626   GNUNET_SERVER_client_drop (stcc->client);
 627   GNUNET_free (stcc);
 628 }
 629
 630
 631 /**
 632  * Client asked for transmission to a peer.  Process the request.
 633  *
 634  * @param cls unused
 635  * @param client the client
 636  * @param message the send message that was sent
 637  */
 638 static void
 639 clients_handle_send (void *cls, struct GNUNET_SERVER_Client *client,
 640                      const struct GNUNET_MessageHeader *message)
 641 {
 642   const struct OutboundMessage *obm;
 643   const struct GNUNET_MessageHeader *obmm;
 644   struct SendTransmitContinuationContext *stcc;
 645   uint16_t size;
 646   uint16_t msize;
 647   struct TransportClient *tc;
 648
 649   tc = lookup_client (client);
 650   if (NULL == tc)
 651   {
 652     /* client asked for transmission before 'START' */
 653     GNUNET_break (0);
 654     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 655     return;
 656   }
 657
 658   size = ntohs (message->size);
 659   if (size <
 660       sizeof (struct OutboundMessage) + sizeof (struct GNUNET_MessageHeader))
 661   {
 662     GNUNET_break (0);
 663     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 664     return;
 665   }
 666   obm = (const struct OutboundMessage *) message;
 667   obmm = (const struct GNUNET_MessageHeader *) &obm[1];
 668   msize = size - sizeof (struct OutboundMessage);
 669   if (msize < sizeof (struct GNUNET_MessageHeader))
 670   {
 671     GNUNET_break (0);
 672     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 673     return;
 674   }
 675
 676   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 677               "Received `%s' request from client with target `%4s' and first message of type %u and total size %u\n",
 678               "SEND", GNUNET_i2s (&obm->peer), ntohs (obmm->type), msize);
 679   if (GNUNET_NO == GST_neighbours_test_connected (&obm->peer))
 680   {
 681     /* not connected, not allowed to send; can happen due to asynchronous operations */
 682     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 683                 "Could not send message to peer `%s': not connected\n",
 684                 GNUNET_i2s (&obm->peer));
 685     GNUNET_STATISTICS_update (GST_stats,
 686                               gettext_noop
 687                               ("# bytes payload dropped (other peer was not connected)"),
 688                               msize, GNUNET_NO);
 689     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 690     return;
 691   }
 692   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 693   stcc = GNUNET_malloc (sizeof (struct SendTransmitContinuationContext));
 694   stcc->target = obm->peer;
 695   stcc->client = client;
 696   GNUNET_SERVER_client_keep (client);
 697   GST_manipulation_send (&obm->peer, obmm, msize,
 698                        GNUNET_TIME_relative_ntoh (obm->timeout),
 699                        &handle_send_transmit_continuation, stcc);
 700 }
 701
 702
 703 /**
 704  * Try to initiate a connection to the given peer if the blacklist
 705  * allowed it.
 706  *
 707  * @param cls closure (unused, NULL)
 708  * @param peer identity of peer that was tested
 709  * @param result GNUNET_OK if the connection is allowed,
 710  *               GNUNET_NO if not
 711  */
 712 static void
 713 try_connect_if_allowed (void *cls, const struct GNUNET_PeerIdentity *peer,
 714                         int result)
 715 {
 716   if (GNUNET_OK != result)
 717     return;                     /* not allowed */
 718   GST_neighbours_try_connect (peer);
 719 }
 720
 721
 722 /**
 723  * Handle request connect message
 724  *
 725  * @param cls closure (always NULL)
 726  * @param client identification of the client
 727  * @param message the actual message
 728  */
 729 static void
 730 clients_handle_request_connect (void *cls, struct GNUNET_SERVER_Client *client,
 731                                 const struct GNUNET_MessageHeader *message)
 732 {
 733   const struct TransportRequestConnectMessage *trcm =
 734       (const struct TransportRequestConnectMessage *) message;
 735
 736   GNUNET_STATISTICS_update (GST_stats,
 737                             gettext_noop
 738                             ("# REQUEST CONNECT messages received"), 1,
 739                             GNUNET_NO);
 740   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 741               "Received a request connect message for peer `%s'\n",
 742               GNUNET_i2s (&trcm->peer));
 743   (void) GST_blacklist_test_allowed (&trcm->peer, NULL, &try_connect_if_allowed,
 744                                      NULL);
 745   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 746 }
 747
 748
 749 /**
 750  * Take the given address and append it to the set of results sent back to
 751  * the client.
 752  *
 753  * @param cls the transmission context used ('struct GNUNET_SERVER_TransmitContext*')
 754  * @param buf text to transmit
 755  */
 756 static void
 757 transmit_address_to_client (void *cls, const char *buf)
 758 {
 759   struct AddressToStringContext *actx = cls;
 760   if (NULL == buf)
 761   {
 762     GNUNET_SERVER_transmit_context_append_data (actx->tc, NULL, 0,
 763                                                 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_TO_STRING_REPLY);
 764     GNUNET_SERVER_transmit_context_run (actx->tc, GNUNET_TIME_UNIT_FOREVER_REL);
 765     GNUNET_CONTAINER_DLL_remove (a2s_head, a2s_tail, actx);
 766     GNUNET_free (actx);
 767     return;
 768   }
 769   GNUNET_SERVER_transmit_context_append_data (actx->tc, buf, strlen (buf) + 1,
 770                                               GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_TO_STRING_REPLY);
 771 }
 772
 773
 774 /**
 775  * Client asked to resolve an address.  Process the request.
 776  *
 777  * @param cls unused
 778  * @param client the client
 779  * @param message the resolution request
 780  */
 781 static void
 782 clients_handle_address_to_string (void *cls,
 783                                   struct GNUNET_SERVER_Client *client,
 784                                   const struct GNUNET_MessageHeader *message)
 785 {
 786   const struct AddressLookupMessage *alum;
 787   struct GNUNET_TRANSPORT_PluginFunctions *papi;
 788   const char *plugin_name;
 789   const char *address;
 790   uint32_t address_len;
 791   uint16_t size;
 792   struct GNUNET_SERVER_TransmitContext *tc;
 793   struct AddressToStringContext *actx;
 794   struct GNUNET_TIME_Relative rtimeout;
 795   int32_t numeric;
 796
 797   size = ntohs (message->size);
 798   if (size < sizeof (struct AddressLookupMessage))
 799   {
 800     GNUNET_break (0);
 801     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 802     return;
 803   }
 804   alum = (const struct AddressLookupMessage *) message;
 805   address_len = ntohs (alum->addrlen);
 806   if (size <= sizeof (struct AddressLookupMessage) + address_len)
 807   {
 808     GNUNET_break (0);
 809     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 810     return;
 811   }
 812   address = (const char *) &alum[1];
 813   plugin_name = (const char *) &address[address_len];
 814   if (plugin_name[size - sizeof (struct AddressLookupMessage) - address_len - 1]
 815       != '\0')
 816   {
 817     GNUNET_break (0);
 818     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 819     return;
 820   }
 821   rtimeout = GNUNET_TIME_relative_ntoh (alum->timeout);
 822   numeric = ntohs (alum->numeric_only);
 823   tc = GNUNET_SERVER_transmit_context_create (client);
 824   papi = GST_plugins_printer_find (plugin_name);
 825   if (NULL == papi)
 826   {
 827     GNUNET_SERVER_transmit_context_append_data (tc, NULL, 0,
 828                                                 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_TO_STRING_REPLY);
 829     GNUNET_SERVER_transmit_context_run (tc, rtimeout);
 830     return;
 831   }
 832   actx = GNUNET_malloc (sizeof (struct AddressToStringContext));
 833   actx->tc = tc;
 834   GNUNET_CONTAINER_DLL_insert (a2s_head, a2s_tail, actx);
 835   GNUNET_SERVER_disable_receive_done_warning (client);
 836   papi->address_pretty_printer (papi->cls, plugin_name, address, address_len,
 837                                 numeric, rtimeout, &transmit_address_to_client,
 838                                 actx);
 839 }
 840
 841
 842 /**
 843  * Compose AddressIterateResponseMessage using the given peer and address.
 844  *
 845  * @param peer identity of the peer
 846  * @param address the address, NULL on disconnect
 847  * @return composed message
 848  */
 849 static struct AddressIterateResponseMessage *
 850 compose_address_iterate_response_message (const struct GNUNET_PeerIdentity
 851                                           *peer,
 852                                           const struct GNUNET_HELLO_Address
 853                                           *address)
 854 {
 855   struct AddressIterateResponseMessage *msg;
 856   size_t size;
 857   size_t tlen;
 858   size_t alen;
 859   char *addr;
 860
 861   GNUNET_assert (NULL != peer);
 862   if (NULL != address)
 863   {
 864     tlen = strlen (address->transport_name) + 1;
 865     alen = address->address_length;
 866   }
 867   else
 868     tlen = alen = 0;
 869   size = (sizeof (struct AddressIterateResponseMessage) + alen + tlen);
 870   msg = GNUNET_malloc (size);
 871   msg->header.size = htons (size);
 872   msg->header.type =
 873       htons (GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_ITERATE_RESPONSE);
 874   msg->reserved = htonl (0);
 875   msg->peer = *peer;
 876   msg->addrlen = htonl (alen);
 877   msg->pluginlen = htonl (tlen);
 878   if (NULL != address)
 879   {
 880     addr = (char *) &msg[1];
 881     memcpy (addr, address->address, alen);
 882     memcpy (&addr[alen], address->transport_name, tlen);
 883   }
 884   return msg;
 885 }
 886
 887
 888 /**
 889  * Output the active address of connected neighbours to the given client.
 890  *
 891  * @param cls the 'struct GNUNET_SERVER_TransmitContext' for transmission to the client
 892  * @param peer identity of the neighbour
 893  * @param ats performance data
 894  * @param ats_count number of entries in ats (excluding 0-termination)
 895  * @param address the address
 896  * @param bandwidth_in inbound quota in NBO
 897  * @param bandwidth_out outbound quota in NBO
 898  */
 899 static void
 900 output_address (void *cls, const struct GNUNET_PeerIdentity *peer,
 901                 const struct GNUNET_ATS_Information *ats, uint32_t ats_count,
 902                 const struct GNUNET_HELLO_Address *address,
 903                 struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
 904                 struct GNUNET_BANDWIDTH_Value32NBO bandwidth_out)
 905 {
 906   struct GNUNET_SERVER_TransmitContext *tc = cls;
 907   struct AddressIterateResponseMessage *msg;
 908
 909   msg = compose_address_iterate_response_message (peer, address);
 910   GNUNET_SERVER_transmit_context_append_message (tc, &msg->header);
 911   GNUNET_free (msg);
 912 }
 913
 914
 915 /**
 916  * Client asked to obtain information about all actively used addresses
 917  * of connected peers
 918  * Process the request.
 919  *
 920  * @param cls unused
 921  * @param client the client
 922  * @param message the peer address information request
 923  */
 924 static void
 925 clients_handle_address_iterate (void *cls, struct GNUNET_SERVER_Client *client,
 926                                 const struct GNUNET_MessageHeader *message)
 927 {
 928   static struct GNUNET_PeerIdentity all_zeros;
 929   struct GNUNET_SERVER_TransmitContext *tc;
 930   struct AddressIterateMessage *msg;
 931   struct GNUNET_HELLO_Address *address;
 932
 933   if (ntohs (message->type) != GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_ITERATE)
 934   {
 935     GNUNET_break (0);
 936     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 937     return;
 938   }
 939   if (ntohs (message->size) != sizeof (struct AddressIterateMessage))
 940   {
 941     GNUNET_break (0);
 942     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 943     return;
 944   }
 945   msg = (struct AddressIterateMessage *) message;
 946   if ( (GNUNET_YES != ntohl (msg->one_shot)) &&
 947        (NULL != lookup_monitoring_client (client)) )
 948   {
 949     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
 950                 "ServerClient %p tried to start monitoring twice\n",
 951                 client);
 952     GNUNET_break (0);
 953     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 954     return;
 955   }
 956   GNUNET_SERVER_disable_receive_done_warning (client);
 957   tc = GNUNET_SERVER_transmit_context_create (client);
 958   if (0 == memcmp (&msg->peer, &all_zeros, sizeof (struct GNUNET_PeerIdentity)))
 959   {
 960     /* iterate over all neighbours */
 961     GST_neighbours_iterate (&output_address, tc);
 962   }
 963   else
 964   {
 965     /* just return one neighbour */
 966     address = GST_neighbour_get_current_address (&msg->peer);
 967     if (address != NULL)
 968       output_address (tc, &msg->peer, NULL, 0, address,
 969                       GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT,
 970                       GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT);
 971   }
 972   if (GNUNET_YES != ntohl (msg->one_shot))
 973     setup_monitoring_client (client, &msg->peer);
 974   else
 975     GNUNET_SERVER_transmit_context_append_data (tc, NULL, 0,
 976                                                 GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_ITERATE_RESPONSE);
 977   GNUNET_SERVER_transmit_context_run (tc, GNUNET_TIME_UNIT_FOREVER_REL);
 978 }
 979
 980
 981 /**
 982  * Start handling requests from clients.
 983  *
 984  * @param server server used to accept clients from.
 985  */
 986 void
 987 GST_clients_start (struct GNUNET_SERVER_Handle *server)
 988 {
 989   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
 990     {&clients_handle_start, NULL,
 991      GNUNET_MESSAGE_TYPE_TRANSPORT_START, sizeof (struct StartMessage)},
 992     {&clients_handle_hello, NULL,
 993      GNUNET_MESSAGE_TYPE_HELLO, 0},
 994     {&clients_handle_send, NULL,
 995      GNUNET_MESSAGE_TYPE_TRANSPORT_SEND, 0},
 996     {&clients_handle_request_connect, NULL,
 997      GNUNET_MESSAGE_TYPE_TRANSPORT_REQUEST_CONNECT,
 998      sizeof (struct TransportRequestConnectMessage)},
 999     {&clients_handle_address_to_string, NULL,
1000      GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_TO_STRING, 0},
1001     {&clients_handle_address_iterate, NULL,
1002      GNUNET_MESSAGE_TYPE_TRANSPORT_ADDRESS_ITERATE,
1003      sizeof (struct AddressIterateMessage)},
1004     {&GST_blacklist_handle_init, NULL,
1005      GNUNET_MESSAGE_TYPE_TRANSPORT_BLACKLIST_INIT,
1006      sizeof (struct GNUNET_MessageHeader)},
1007     {&GST_blacklist_handle_reply, NULL,
1008      GNUNET_MESSAGE_TYPE_TRANSPORT_BLACKLIST_REPLY,
1009      sizeof (struct BlacklistMessage)},
1010     {&GST_manipulation_set_metric, NULL,
1011      GNUNET_MESSAGE_TYPE_TRANSPORT_TRAFFIC_METRIC, 0},
1012     {NULL, NULL, 0, 0}
1013   };
1014   nc = GNUNET_SERVER_notification_context_create (server, 0);
1015   GNUNET_SERVER_add_handlers (server, handlers);
1016   GNUNET_SERVER_disconnect_notify (server, &client_disconnect_notification,
1017                                    NULL);
1018 }
1019
1020
1021 /**
1022  * Stop processing clients.
1023  */
1024 void
1025 GST_clients_stop ()
1026 {
1027   struct AddressToStringContext *cur;
1028
1029   while (NULL != (cur = a2s_head))
1030   {
1031     GNUNET_SERVER_transmit_context_destroy (cur->tc, GNUNET_NO);
1032     GNUNET_CONTAINER_DLL_remove (a2s_head, a2s_tail, cur);
1033     GNUNET_free (cur);
1034   }
1035
1036   if (NULL != nc)
1037   {
1038     GNUNET_SERVER_notification_context_destroy (nc);
1039     nc = NULL;
1040   }
1041 }
1042
1043 /**
1044  * Broadcast the given message to all of our clients.
1045  *
1046  * @param msg message to broadcast
1047  * @param may_drop GNUNET_YES if the message can be dropped / is payload
1048  */
1049 void
1050 GST_clients_broadcast (const struct GNUNET_MessageHeader *msg, int may_drop)
1051 {
1052   struct TransportClient *tc;
1053
1054   for (tc = clients_head; tc != NULL; tc = tc->next)
1055   {
1056     if ((GNUNET_YES == may_drop) && (GNUNET_YES != tc->send_payload))
1057       continue;                 /* skip, this client does not care about payload */
1058     unicast (tc, msg, may_drop);
1059   }
1060 }
1061
1062
1063 /**
1064  * Send the given message to a particular client
1065  *
1066  * @param client target of the message
1067  * @param msg message to transmit
1068  * @param may_drop GNUNET_YES if the message can be dropped
1069  */
1070 void
1071 GST_clients_unicast (struct GNUNET_SERVER_Client *client,
1072                      const struct GNUNET_MessageHeader *msg, int may_drop)
1073 {
1074   struct TransportClient *tc;
1075
1076   tc = lookup_client (client);
1077   if (NULL == tc)
1078     return;                     /* client got disconnected in the meantime, drop message */
1079   unicast (tc, msg, may_drop);
1080 }
1081
1082
1083 /**
1084  * Broadcast the new active address to all clients monitoring the peer.
1085  *
1086  * @param peer peer this update is about (never NULL)
1087  * @param address address, NULL on disconnect
1088  */
1089 void
1090 GST_clients_broadcast_address_notification (const struct GNUNET_PeerIdentity
1091                                             *peer,
1092                                             const struct GNUNET_HELLO_Address
1093                                             *address)
1094 {
1095   struct AddressIterateResponseMessage *msg;
1096   struct MonitoringClient *mc;
1097   static struct GNUNET_PeerIdentity all_zeros;
1098   msg = compose_address_iterate_response_message (peer, address);
1099   mc = monitoring_clients_head;
1100   while (mc != NULL)
1101   {
1102     if ((0 == memcmp (&mc->peer, &all_zeros,
1103                       sizeof (struct GNUNET_PeerIdentity))) ||
1104         (0 == memcmp (&mc->peer, peer,
1105                       sizeof (struct GNUNET_PeerIdentity))))
1106     {
1107       GNUNET_SERVER_notification_context_unicast (nc, mc->client,
1108                                                   &msg->header, GNUNET_NO);
1109     }
1110
1111     mc = mc->next;
1112   }
1113   GNUNET_free (msg);
1114 }
1115
1116
1117 /* end of file gnunet-service-transport_clients.c */