src/core/gnunet-service-core.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 2, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file core/gnunet-service-core.c
  23  * @brief high-level P2P messaging
  24  * @author Christian Grothoff
  25  *
  26  * Considerations for later:
  27  * - check that hostkey used by transport (for HELLOs) is the
  28  *   same as the hostkey that we are using!
  29  * - add code to send PINGs if we are about to time-out otherwise
  30  * - optimize lookup (many O(n) list traversals
  31  *   could ideally be changed to O(1) hash map lookups)
  32  */
  33 #include "platform.h"
  34 #include "gnunet_constants.h"
  35 #include "gnunet_util_lib.h"
  36 #include "gnunet_hello_lib.h"
  37 #include "gnunet_peerinfo_service.h"
  38 #include "gnunet_protocols.h"
  39 #include "gnunet_signatures.h"
  40 #include "gnunet_statistics_service.h"
  41 #include "gnunet_transport_service.h"
  42 #include "core.h"
  43
  44
  45 #define DEBUG_HANDSHAKE GNUNET_NO
  46
  47 #define DEBUG_CORE_QUOTA GNUNET_YES
  48
  49 /**
  50  * Receive and send buffer windows grow over time.  For
  51  * how long can 'unused' bandwidth accumulate before we
  52  * need to cap it?  (specified in seconds).
  53  */
  54 #define MAX_WINDOW_TIME_S (5 * 60)
  55
  56 /**
  57  * How many messages do we queue up at most for optional
  58  * notifications to a client?  (this can cause notifications
  59  * about outgoing messages to be dropped).
  60  */
  61 #define MAX_NOTIFY_QUEUE 16
  62
  63 /**
  64  * Minimum bandwidth (out) to assign to any connected peer.
  65  * Should be rather low; values larger than DEFAULT_BW_IN_OUT make no
  66  * sense.
  67  */
  68 #define MIN_BANDWIDTH_PER_PEER GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT
  69
  70 /**
  71  * After how much time past the "official" expiration time do
  72  * we discard messages?  Should not be zero since we may
  73  * intentionally defer transmission until close to the deadline
  74  * and then may be slightly past the deadline due to inaccuracy
  75  * in sleep and our own CPU consumption.
  76  */
  77 #define PAST_EXPIRATION_DISCARD_TIME GNUNET_TIME_UNIT_SECONDS
  78
  79 /**
  80  * What is the maximum delay for a SET_KEY message?
  81  */
  82 #define MAX_SET_KEY_DELAY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
  83
  84 /**
  85  * What how long do we wait for SET_KEY confirmation initially?
  86  */
  87 #define INITIAL_SET_KEY_RETRY_FREQUENCY GNUNET_TIME_relative_multiply (MAX_SET_KEY_DELAY, 3)
  88
  89 /**
  90  * What is the maximum delay for a PING message?
  91  */
  92 #define MAX_PING_DELAY GNUNET_TIME_relative_multiply (MAX_SET_KEY_DELAY, 2)
  93
  94 /**
  95  * What is the maximum delay for a PONG message?
  96  */
  97 #define MAX_PONG_DELAY GNUNET_TIME_relative_multiply (MAX_PING_DELAY, 2)
  98
  99 /**
 100  * What is the minimum frequency for a PING message?
 101  */
 102 #define MIN_PING_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
 103
 104 /**
 105  * How often do we recalculate bandwidth quotas?
 106  */
 107 #define QUOTA_UPDATE_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
 108
 109 /**
 110  * What is the priority for a SET_KEY message?
 111  */
 112 #define SET_KEY_PRIORITY 0xFFFFFF
 113
 114 /**
 115  * What is the priority for a PING message?
 116  */
 117 #define PING_PRIORITY 0xFFFFFF
 118
 119 /**
 120  * What is the priority for a PONG message?
 121  */
 122 #define PONG_PRIORITY 0xFFFFFF
 123
 124 /**
 125  * How many messages do we queue per peer at most?  Must be at
 126  * least two.
 127  */
 128 #define MAX_PEER_QUEUE_SIZE 16
 129
 130 /**
 131  * How many non-mandatory messages do we queue per client at most?
 132  */
 133 #define MAX_CLIENT_QUEUE_SIZE 32
 134
 135 /**
 136  * What is the maximum age of a message for us to consider
 137  * processing it?  Note that this looks at the timestamp used
 138  * by the other peer, so clock skew between machines does
 139  * come into play here.  So this should be picked high enough
 140  * so that a little bit of clock skew does not prevent peers
 141  * from connecting to us.
 142  */
 143 #define MAX_MESSAGE_AGE GNUNET_TIME_UNIT_DAYS
 144
 145 /**
 146  * What is the maximum size for encrypted messages?  Note that this
 147  * number imposes a clear limit on the maximum size of any message.
 148  * Set to a value close to 64k but not so close that transports will
 149  * have trouble with their headers.
 150  */
 151 #define MAX_ENCRYPTED_MESSAGE_SIZE (63 * 1024)
 152
 153
 154 /**
 155  * State machine for our P2P encryption handshake.  Everyone starts in
 156  * "DOWN", if we receive the other peer's key (other peer initiated)
 157  * we start in state RECEIVED (since we will immediately send our
 158  * own); otherwise we start in SENT.  If we get back a PONG from
 159  * within either state, we move up to CONFIRMED (the PONG will always
 160  * be sent back encrypted with the key we sent to the other peer).
 161  */
 162 enum PeerStateMachine
 163 {
 164   PEER_STATE_DOWN,
 165   PEER_STATE_KEY_SENT,
 166   PEER_STATE_KEY_RECEIVED,
 167   PEER_STATE_KEY_CONFIRMED
 168 };
 169
 170
 171 /**
 172  * Number of bytes (at the beginning) of "struct EncryptedMessage"
 173  * that are NOT encrypted.
 174  */
 175 #define ENCRYPTED_HEADER_SIZE (sizeof(struct GNUNET_MessageHeader) + sizeof(uint32_t))
 176
 177
 178 /**
 179  * Encapsulation for encrypted messages exchanged between
 180  * peers.  Followed by the actual encrypted data.
 181  */
 182 struct EncryptedMessage
 183 {
 184   /**
 185    * Message type is either CORE_ENCRYPTED_MESSAGE.
 186    */
 187   struct GNUNET_MessageHeader header;
 188
 189   /**
 190    * Random value used for IV generation.  ENCRYPTED_HEADER_SIZE must
 191    * be set to the offset of the *next* field.
 192    */
 193   uint32_t iv_seed GNUNET_PACKED;
 194
 195   /**
 196    * Hash of the plaintext (starting at 'sequence_number'), used to
 197    * verify message integrity.  Everything after this hash (including
 198    * this hash itself) will be encrypted.
 199    */
 200   GNUNET_HashCode plaintext_hash;
 201
 202   /**
 203    * Sequence number, in network byte order.  This field
 204    * must be the first encrypted/decrypted field and the
 205    * first byte that is hashed for the plaintext hash.
 206    */
 207   uint32_t sequence_number GNUNET_PACKED;
 208
 209   /**
 210    * Desired bandwidth (how much we should send to this peer / how
 211    * much is the sender willing to receive)?
 212    */
 213   struct GNUNET_BANDWIDTH_Value32NBO inbound_bw_limit;
 214
 215   /**
 216    * Timestamp.  Used to prevent reply of ancient messages
 217    * (recent messages are caught with the sequence number).
 218    */
 219   struct GNUNET_TIME_AbsoluteNBO timestamp;
 220
 221 };
 222
 223
 224 /**
 225  * We're sending an (encrypted) PING to the other peer to check if he
 226  * can decrypt.  The other peer should respond with a PONG with the
 227  * same content, except this time encrypted with the receiver's key.
 228  */
 229 struct PingMessage
 230 {
 231   /**
 232    * Message type is CORE_PING.
 233    */
 234   struct GNUNET_MessageHeader header;
 235
 236   /**
 237    * Random number chosen to make reply harder.
 238    */
 239   uint32_t challenge GNUNET_PACKED;
 240
 241   /**
 242    * Intended target of the PING, used primarily to check
 243    * that decryption actually worked.
 244    */
 245   struct GNUNET_PeerIdentity target;
 246 };
 247
 248
 249
 250 /**
 251  * Response to a PING.  Includes data from the original PING
 252  * plus initial bandwidth quota information.
 253  */
 254 struct PongMessage
 255 {
 256   /**
 257    * Message type is CORE_PONG.
 258    */
 259   struct GNUNET_MessageHeader header;
 260
 261   /**
 262    * Random number proochosen to make reply harder.  Must be
 263    * first field after header (this is where we start to encrypt!).
 264    */
 265   uint32_t challenge GNUNET_PACKED;
 266
 267   /**
 268    * Must be zero.
 269    */
 270   uint32_t reserved GNUNET_PACKED;
 271
 272   /**
 273    * Desired bandwidth (how much we should send to this
 274    * peer / how much is the sender willing to receive).
 275    */
 276   struct GNUNET_BANDWIDTH_Value32NBO inbound_bw_limit;
 277
 278   /**
 279    * Intended target of the PING, used primarily to check
 280    * that decryption actually worked.
 281    */
 282   struct GNUNET_PeerIdentity target;
 283 };
 284
 285
 286 /**
 287  * Message transmitted to set (or update) a session key.
 288  */
 289 struct SetKeyMessage
 290 {
 291
 292   /**
 293    * Message type is either CORE_SET_KEY.
 294    */
 295   struct GNUNET_MessageHeader header;
 296
 297   /**
 298    * Status of the sender (should be in "enum PeerStateMachine"), nbo.
 299    */
 300   int32_t sender_status GNUNET_PACKED;
 301
 302   /**
 303    * Purpose of the signature, will be
 304    * GNUNET_SIGNATURE_PURPOSE_SET_KEY.
 305    */
 306   struct GNUNET_CRYPTO_RsaSignaturePurpose purpose;
 307
 308   /**
 309    * At what time was this key created?
 310    */
 311   struct GNUNET_TIME_AbsoluteNBO creation_time;
 312
 313   /**
 314    * The encrypted session key.
 315    */
 316   struct GNUNET_CRYPTO_RsaEncryptedData encrypted_key;
 317
 318   /**
 319    * Who is the intended recipient?
 320    */
 321   struct GNUNET_PeerIdentity target;
 322
 323   /**
 324    * Signature of the stuff above (starting at purpose).
 325    */
 326   struct GNUNET_CRYPTO_RsaSignature signature;
 327
 328 };
 329
 330
 331 /**
 332  * Message waiting for transmission. This struct
 333  * is followed by the actual content of the message.
 334  */
 335 struct MessageEntry
 336 {
 337
 338   /**
 339    * We keep messages in a doubly linked list.
 340    */
 341   struct MessageEntry *next;
 342
 343   /**
 344    * We keep messages in a doubly linked list.
 345    */
 346   struct MessageEntry *prev;
 347
 348   /**
 349    * By when are we supposed to transmit this message?
 350    */
 351   struct GNUNET_TIME_Absolute deadline;
 352
 353   /**
 354    * By when are we supposed to transmit this message (after
 355    * giving slack)?
 356    */
 357   struct GNUNET_TIME_Absolute slack_deadline;
 358
 359   /**
 360    * How important is this message to us?
 361    */
 362   unsigned int priority;
 363
 364   /**
 365    * How long is the message? (number of bytes following
 366    * the "struct MessageEntry", but not including the
 367    * size of "struct MessageEntry" itself!)
 368    */
 369   uint16_t size;
 370
 371   /**
 372    * Was this message selected for transmission in the
 373    * current round? GNUNET_YES or GNUNET_NO.
 374    */
 375   int8_t do_transmit;
 376
 377   /**
 378    * Did we give this message some slack (delayed sending) previously
 379    * (and hence should not give it any more slack)? GNUNET_YES or
 380    * GNUNET_NO.
 381    */
 382   int8_t got_slack;
 383
 384 };
 385
 386
 387 struct Neighbour
 388 {
 389   /**
 390    * We keep neighbours in a linked list (for now).
 391    */
 392   struct Neighbour *next;
 393
 394   /**
 395    * Unencrypted messages destined for this peer.
 396    */
 397   struct MessageEntry *messages;
 398
 399   /**
 400    * Head of the batched, encrypted message queue (already ordered,
 401    * transmit starting with the head).
 402    */
 403   struct MessageEntry *encrypted_head;
 404
 405   /**
 406    * Tail of the batched, encrypted message queue (already ordered,
 407    * append new messages to tail)
 408    */
 409   struct MessageEntry *encrypted_tail;
 410
 411   /**
 412    * Handle for pending requests for transmission to this peer
 413    * with the transport service.  NULL if no request is pending.
 414    */
 415   struct GNUNET_TRANSPORT_TransmitHandle *th;
 416
 417   /**
 418    * Public key of the neighbour, NULL if we don't have it yet.
 419    */
 420   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *public_key;
 421
 422   /**
 423    * We received a PING message before we got the "public_key"
 424    * (or the SET_KEY).  We keep it here until we have a key
 425    * to decrypt it.  NULL if no PING is pending.
 426    */
 427   struct PingMessage *pending_ping;
 428
 429   /**
 430    * We received a PONG message before we got the "public_key"
 431    * (or the SET_KEY).  We keep it here until we have a key
 432    * to decrypt it.  NULL if no PONG is pending.
 433    */
 434   struct PongMessage *pending_pong;
 435
 436   /**
 437    * Non-NULL if we are currently looking up HELLOs for this peer.
 438    * for this peer.
 439    */
 440   struct GNUNET_PEERINFO_IteratorContext *pitr;
 441
 442   /**
 443    * SetKeyMessage to transmit, NULL if we are not currently trying
 444    * to send one.
 445    */
 446   struct SetKeyMessage *skm;
 447
 448   /**
 449    * Identity of the neighbour.
 450    */
 451   struct GNUNET_PeerIdentity peer;
 452
 453   /**
 454    * Key we use to encrypt our messages for the other peer
 455    * (initialized by us when we do the handshake).
 456    */
 457   struct GNUNET_CRYPTO_AesSessionKey encrypt_key;
 458
 459   /**
 460    * Key we use to decrypt messages from the other peer
 461    * (given to us by the other peer during the handshake).
 462    */
 463   struct GNUNET_CRYPTO_AesSessionKey decrypt_key;
 464
 465   /**
 466    * ID of task used for re-trying plaintext scheduling.
 467    */
 468   GNUNET_SCHEDULER_TaskIdentifier retry_plaintext_task;
 469
 470   /**
 471    * ID of task used for re-trying SET_KEY and PING message.
 472    */
 473   GNUNET_SCHEDULER_TaskIdentifier retry_set_key_task;
 474
 475   /**
 476    * ID of task used for updating bandwidth quota for this neighbour.
 477    */
 478   GNUNET_SCHEDULER_TaskIdentifier quota_update_task;
 479
 480   /**
 481    * ID of task used for sending keep-alive pings.
 482    */
 483   GNUNET_SCHEDULER_TaskIdentifier keep_alive_task;
 484
 485   /**
 486    * ID of task used for cleaning up dead neighbour entries.
 487    */
 488   GNUNET_SCHEDULER_TaskIdentifier dead_clean_task;
 489
 490   /**
 491    * At what time did we generate our encryption key?
 492    */
 493   struct GNUNET_TIME_Absolute encrypt_key_created;
 494
 495   /**
 496    * At what time did the other peer generate the decryption key?
 497    */
 498   struct GNUNET_TIME_Absolute decrypt_key_created;
 499
 500   /**
 501    * At what time did we initially establish (as in, complete session
 502    * key handshake) this connection?  Should be zero if status != KEY_CONFIRMED.
 503    */
 504   struct GNUNET_TIME_Absolute time_established;
 505
 506   /**
 507    * At what time did we last receive an encrypted message from the
 508    * other peer?  Should be zero if status != KEY_CONFIRMED.
 509    */
 510   struct GNUNET_TIME_Absolute last_activity;
 511
 512   /**
 513    * Last latency observed from this peer.
 514    */
 515   struct GNUNET_TIME_Relative last_latency;
 516
 517   /**
 518    * At what frequency are we currently re-trying SET_KEY messages?
 519    */
 520   struct GNUNET_TIME_Relative set_key_retry_frequency;
 521
 522   /**
 523    * Tracking bandwidth for sending to this peer.
 524    */
 525   struct GNUNET_BANDWIDTH_Tracker available_send_window;
 526
 527   /**
 528    * Tracking bandwidth for receiving from this peer.
 529    */
 530   struct GNUNET_BANDWIDTH_Tracker available_recv_window;
 531
 532   /**
 533    * How valueable were the messages of this peer recently?
 534    */
 535   unsigned long long current_preference;
 536
 537   /**
 538    * Bit map indicating which of the 32 sequence numbers before the last
 539    * were received (good for accepting out-of-order packets and
 540    * estimating reliability of the connection)
 541    */
 542   unsigned int last_packets_bitmap;
 543
 544   /**
 545    * last sequence number received on this connection (highest)
 546    */
 547   uint32_t last_sequence_number_received;
 548
 549   /**
 550    * last sequence number transmitted
 551    */
 552   uint32_t last_sequence_number_sent;
 553
 554   /**
 555    * Available bandwidth in for this peer (current target).
 556    */
 557   struct GNUNET_BANDWIDTH_Value32NBO bw_in;
 558
 559   /**
 560    * Available bandwidth out for this peer (current target).
 561    */
 562   struct GNUNET_BANDWIDTH_Value32NBO bw_out;
 563
 564   /**
 565    * Internal bandwidth limit set for this peer (initially typically
 566    * set to "-1").  Actual "bw_out" is MIN of
 567    * "bpm_out_internal_limit" and "bw_out_external_limit".
 568    */
 569   struct GNUNET_BANDWIDTH_Value32NBO bw_out_internal_limit;
 570
 571   /**
 572    * External bandwidth limit set for this peer by the
 573    * peer that we are communicating with.  "bw_out" is MIN of
 574    * "bw_out_internal_limit" and "bw_out_external_limit".
 575    */
 576   struct GNUNET_BANDWIDTH_Value32NBO bw_out_external_limit;
 577
 578   /**
 579    * What was our PING challenge number (for this peer)?
 580    */
 581   uint32_t ping_challenge;
 582
 583   /**
 584    * What was the last distance to this peer as reported by the transports?
 585    */
 586   uint32_t last_distance;
 587
 588   /**
 589    * What is our connection status?
 590    */
 591   enum PeerStateMachine status;
 592
 593   /**
 594    * Are we currently connected to this neighbour?
 595    */
 596   int is_connected;
 597
 598 };
 599
 600
 601 /**
 602  * Data structure for each client connected to the core service.
 603  */
 604 struct Client
 605 {
 606   /**
 607    * Clients are kept in a linked list.
 608    */
 609   struct Client *next;
 610
 611   /**
 612    * Handle for the client with the server API.
 613    */
 614   struct GNUNET_SERVER_Client *client_handle;
 615
 616   /**
 617    * Array of the types of messages this peer cares
 618    * about (with "tcnt" entries).  Allocated as part
 619    * of this client struct, do not free!
 620    */
 621   const uint16_t *types;
 622
 623   /**
 624    * Options for messages this client cares about,
 625    * see GNUNET_CORE_OPTION_ values.
 626    */
 627   uint32_t options;
 628
 629   /**
 630    * Number of types of incoming messages this client
 631    * specifically cares about.  Size of the "types" array.
 632    */
 633   unsigned int tcnt;
 634
 635 };
 636
 637
 638 /**
 639  * Our public key.
 640  */
 641 static struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded my_public_key;
 642
 643 /**
 644  * Our identity.
 645  */
 646 static struct GNUNET_PeerIdentity my_identity;
 647
 648 /**
 649  * Our private key.
 650  */
 651 static struct GNUNET_CRYPTO_RsaPrivateKey *my_private_key;
 652
 653 /**
 654  * Our scheduler.
 655  */
 656 struct GNUNET_SCHEDULER_Handle *sched;
 657
 658 /**
 659  * Our configuration.
 660  */
 661 const struct GNUNET_CONFIGURATION_Handle *cfg;
 662
 663 /**
 664  * Our server.
 665  */
 666 static struct GNUNET_SERVER_Handle *server;
 667
 668 /**
 669  * Transport service.
 670  */
 671 static struct GNUNET_TRANSPORT_Handle *transport;
 672
 673 /**
 674  * Linked list of our clients.
 675  */
 676 static struct Client *clients;
 677
 678 /**
 679  * Context for notifications we need to send to our clients.
 680  */
 681 static struct GNUNET_SERVER_NotificationContext *notifier;
 682
 683 /**
 684  * We keep neighbours in a linked list (for now).
 685  */
 686 static struct Neighbour *neighbours;
 687
 688 /**
 689  * For creating statistics.
 690  */
 691 static struct GNUNET_STATISTICS_Handle *stats;
 692
 693 /**
 694  * Sum of all preferences among all neighbours.
 695  */
 696 static unsigned long long preference_sum;
 697
 698 /**
 699  * Total number of neighbours we have.
 700  */
 701 static unsigned int neighbour_count;
 702
 703 /**
 704  * How much inbound bandwidth are we supposed to be using per second?
 705  * FIXME: this value is not used!
 706  */
 707 static unsigned long long bandwidth_target_in_bps;
 708
 709 /**
 710  * How much outbound bandwidth are we supposed to be using per second?
 711  */
 712 static unsigned long long bandwidth_target_out_bps;
 713
 714
 715
 716 /**
 717  * A preference value for a neighbour was update.  Update
 718  * the preference sum accordingly.
 719  *
 720  * @param inc how much was a preference value increased?
 721  */
 722 static void
 723 update_preference_sum (unsigned long long inc)
 724 {
 725   struct Neighbour *n;
 726   unsigned long long os;
 727
 728   os = preference_sum;
 729   preference_sum += inc;
 730   if (preference_sum >= os)
 731     return; /* done! */
 732   /* overflow! compensate by cutting all values in half! */
 733   preference_sum = 0;
 734   n = neighbours;
 735   while (n != NULL)
 736     {
 737       n->current_preference /= 2;
 738       preference_sum += n->current_preference;
 739       n = n->next;
 740     }
 741   GNUNET_STATISTICS_set (stats, gettext_noop ("# total peer preference"), preference_sum, GNUNET_NO);
 742 }
 743
 744
 745 /**
 746  * Find the entry for the given neighbour.
 747  *
 748  * @param peer identity of the neighbour
 749  * @return NULL if we are not connected, otherwise the
 750  *         neighbour's entry.
 751  */
 752 static struct Neighbour *
 753 find_neighbour (const struct GNUNET_PeerIdentity *peer)
 754 {
 755   struct Neighbour *ret;
 756
 757   ret = neighbours;
 758   while ((ret != NULL) &&
 759          (0 != memcmp (&ret->peer,
 760                        peer, sizeof (struct GNUNET_PeerIdentity))))
 761     ret = ret->next;
 762   return ret;
 763 }
 764
 765
 766 /**
 767  * Send a message to one of our clients.
 768  *
 769  * @param client target for the message
 770  * @param msg message to transmit
 771  * @param can_drop could this message be dropped if the
 772  *        client's queue is getting too large?
 773  */
 774 static void
 775 send_to_client (struct Client *client,
 776                 const struct GNUNET_MessageHeader *msg,
 777                 int can_drop)
 778 {
 779 #if DEBUG_CORE_CLIENT
 780   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 781               "Preparing to send message of type %u to client.\n",
 782               (unsigned int) ntohs (msg->type));
 783 #endif
 784   GNUNET_SERVER_notification_context_unicast (notifier,
 785                                               client->client_handle,
 786                                               msg,
 787                                               can_drop);
 788 }
 789
 790
 791 /**
 792  * Send a message to all of our current clients that have
 793  * the right options set.
 794  *
 795  * @param msg message to multicast
 796  * @param can_drop can this message be discarded if the queue is too long
 797  * @param options mask to use
 798  */
 799 static void
 800 send_to_all_clients (const struct GNUNET_MessageHeader *msg,
 801                      int can_drop,
 802                      int options)
 803 {
 804   struct Client *c;
 805
 806   c = clients;
 807   while (c != NULL)
 808     {
 809       if (0 != (c->options & options))
 810         {
 811 #if DEBUG_CORE_CLIENT
 812           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 813                       "Sending message of type %u to client.\n",
 814                       (unsigned int) ntohs (msg->type));
 815 #endif
 816           send_to_client (c, msg, can_drop);
 817         }
 818       c = c->next;
 819     }
 820 }
 821
 822
 823 /**
 824  * Handle CORE_INIT request.
 825  */
 826 static void
 827 handle_client_init (void *cls,
 828                     struct GNUNET_SERVER_Client *client,
 829                     const struct GNUNET_MessageHeader *message)
 830 {
 831   const struct InitMessage *im;
 832   struct InitReplyMessage irm;
 833   struct Client *c;
 834   uint16_t msize;
 835   const uint16_t *types;
 836   uint16_t *wtypes;
 837   struct Neighbour *n;
 838   struct ConnectNotifyMessage cnm;
 839   unsigned int i;
 840
 841 #if DEBUG_CORE_CLIENT
 842   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 843               "Client connecting to core service with `%s' message\n",
 844               "INIT");
 845 #endif
 846   /* check that we don't have an entry already */
 847   c = clients;
 848   while (c != NULL)
 849     {
 850       if (client == c->client_handle)
 851         {
 852           GNUNET_break (0);
 853           GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 854           return;
 855         }
 856       c = c->next;
 857     }
 858   msize = ntohs (message->size);
 859   if (msize < sizeof (struct InitMessage))
 860     {
 861       GNUNET_break (0);
 862       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 863       return;
 864     }
 865   GNUNET_SERVER_notification_context_add (notifier, client);
 866   im = (const struct InitMessage *) message;
 867   types = (const uint16_t *) &im[1];
 868   msize -= sizeof (struct InitMessage);
 869   c = GNUNET_malloc (sizeof (struct Client) + msize);
 870   c->client_handle = client;
 871   c->next = clients;
 872   clients = c;
 873   c->tcnt = msize / sizeof (uint16_t);
 874   c->types = (const uint16_t *) &c[1];
 875   wtypes = (uint16_t *) &c[1];
 876   for (i=0;i<c->tcnt;i++)
 877     wtypes[i] = ntohs (types[i]);
 878   c->options = ntohl (im->options);
 879 #if DEBUG_CORE_CLIENT
 880   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 881               "Client %p is interested in %u message types\n",
 882               c,
 883               (unsigned int) c->tcnt);
 884 #endif
 885   /* send init reply message */
 886   irm.header.size = htons (sizeof (struct InitReplyMessage));
 887   irm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_INIT_REPLY);
 888   irm.reserved = htonl (0);
 889   memcpy (&irm.publicKey,
 890           &my_public_key,
 891           sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
 892 #if DEBUG_CORE_CLIENT
 893   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 894               "Sending `%s' message to client.\n", "INIT_REPLY");
 895 #endif
 896   send_to_client (c, &irm.header, GNUNET_NO);
 897   if (0 != (c->options & GNUNET_CORE_OPTION_SEND_CONNECT))
 898     {
 899       /* notify new client about existing neighbours */
 900       cnm.header.size = htons (sizeof (struct ConnectNotifyMessage));
 901       cnm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_CONNECT);
 902       n = neighbours;
 903       while (n != NULL)
 904         {
 905           if (n->status == PEER_STATE_KEY_CONFIRMED)
 906             {
 907 #if DEBUG_CORE_CLIENT
 908               GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 909                           "Sending `%s' message to client.\n", "NOTIFY_CONNECT");
 910 #endif
 911               cnm.distance = htonl (n->last_distance);
 912               cnm.latency = GNUNET_TIME_relative_hton (n->last_latency);
 913               cnm.peer = n->peer;
 914               send_to_client (c, &cnm.header, GNUNET_NO);
 915             }
 916           n = n->next;
 917         }
 918     }
 919   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 920 }
 921
 922
 923 /**
 924  * A client disconnected, clean up.
 925  *
 926  * @param cls closure
 927  * @param client identification of the client
 928  */
 929 static void
 930 handle_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
 931 {
 932   struct Client *pos;
 933   struct Client *prev;
 934
 935   if (client == NULL)
 936     return;
 937 #if DEBUG_CORE_CLIENT
 938   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 939               "Client %p has disconnected from core service.\n",
 940               client);
 941 #endif
 942   prev = NULL;
 943   pos = clients;
 944   while (pos != NULL)
 945     {
 946       if (client == pos->client_handle)
 947         {
 948           if (prev == NULL)
 949             clients = pos->next;
 950           else
 951             prev->next = pos->next;
 952           GNUNET_free (pos);
 953           return;
 954         }
 955       prev = pos;
 956       pos = pos->next;
 957     }
 958   /* client never sent INIT */
 959 }
 960
 961
 962 /**
 963  * Handle REQUEST_INFO request.
 964  */
 965 static void
 966 handle_client_request_info (void *cls,
 967                             struct GNUNET_SERVER_Client *client,
 968                             const struct GNUNET_MessageHeader *message)
 969 {
 970   const struct RequestInfoMessage *rcm;
 971   struct Neighbour *n;
 972   struct ConfigurationInfoMessage cim;
 973   int32_t want_reserv;
 974   int32_t got_reserv;
 975   unsigned long long old_preference;
 976   struct GNUNET_SERVER_TransmitContext *tc;
 977
 978 #if DEBUG_CORE_CLIENT
 979   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 980               "Core service receives `%s' request.\n", "REQUEST_INFO");
 981 #endif
 982   rcm = (const struct RequestInfoMessage *) message;
 983   n = find_neighbour (&rcm->peer);
 984   memset (&cim, 0, sizeof (cim));
 985   if (n != NULL)
 986     {
 987       want_reserv = ntohl (rcm->reserve_inbound);
 988       if (n->bw_out_internal_limit.value__ != rcm->limit_outbound.value__)
 989         {
 990           n->bw_out_internal_limit = rcm->limit_outbound;
 991           n->bw_out = GNUNET_BANDWIDTH_value_min (n->bw_out_internal_limit,
 992                                                   n->bw_out_external_limit);
 993           GNUNET_BANDWIDTH_tracker_update_quota (&n->available_recv_window,
 994                                                  n->bw_out);
 995           GNUNET_TRANSPORT_set_quota (transport,
 996                                       &n->peer,
 997                                       n->bw_in,
 998                                       n->bw_out,
 999                                       GNUNET_TIME_UNIT_FOREVER_REL,
1000                                       NULL, NULL);
1001         }
1002       if (want_reserv < 0)
1003         {
1004           got_reserv = want_reserv;
1005         }
1006       else if (want_reserv > 0)
1007         {
1008           if (GNUNET_BANDWIDTH_tracker_get_delay (&n->available_recv_window,
1009                                                   want_reserv).value == 0)
1010             got_reserv = want_reserv;
1011           else
1012             got_reserv = 0; /* all or nothing */
1013         }
1014       else
1015         got_reserv = 0;
1016       GNUNET_BANDWIDTH_tracker_consume (&n->available_recv_window,
1017                                         got_reserv);
1018       old_preference = n->current_preference;
1019       n->current_preference += GNUNET_ntohll(rcm->preference_change);
1020       if (old_preference > n->current_preference)
1021         {
1022           /* overflow; cap at maximum value */
1023           n->current_preference = (unsigned long long) -1;
1024         }
1025       update_preference_sum (n->current_preference - old_preference);
1026 #if DEBUG_CORE_QUOTA
1027       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1028                   "Received reservation request for %d bytes for peer `%4s', reserved %d bytes\n",
1029                   (int) want_reserv,
1030                   GNUNET_i2s (&rcm->peer),
1031                   (int) got_reserv);
1032 #endif
1033       cim.reserved_amount = htonl (got_reserv);
1034       cim.bw_in = n->bw_in;
1035       cim.bw_out = n->bw_out;
1036       cim.preference = n->current_preference;
1037     }
1038   cim.header.size = htons (sizeof (struct ConfigurationInfoMessage));
1039   cim.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_CONFIGURATION_INFO);
1040   cim.peer = rcm->peer;
1041
1042 #if DEBUG_CORE_CLIENT
1043   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1044               "Sending `%s' message to client.\n", "CONFIGURATION_INFO");
1045 #endif
1046   tc = GNUNET_SERVER_transmit_context_create (client);
1047   GNUNET_SERVER_transmit_context_append_message (tc, &cim.header);
1048   GNUNET_SERVER_transmit_context_run (tc,
1049                                       GNUNET_TIME_UNIT_FOREVER_REL);
1050 }
1051
1052
1053 /**
1054  * Free the given entry for the neighbour (it has
1055  * already been removed from the list at this point).
1056  *
1057  * @param n neighbour to free
1058  */
1059 static void
1060 free_neighbour (struct Neighbour *n)
1061 {
1062   struct MessageEntry *m;
1063
1064   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1065               "Destroying neighbour entry for peer `%4s'\n",
1066               GNUNET_i2s (&n->peer));
1067   if (n->pitr != NULL)
1068     {
1069       GNUNET_PEERINFO_iterate_cancel (n->pitr);
1070       n->pitr = NULL;
1071     }
1072   if (n->skm != NULL)
1073     {
1074       GNUNET_free (n->skm);
1075       n->skm = NULL;
1076     }
1077   while (NULL != (m = n->messages))
1078     {
1079       n->messages = m->next;
1080       GNUNET_free (m);
1081     }
1082   while (NULL != (m = n->encrypted_head))
1083     {
1084       GNUNET_CONTAINER_DLL_remove (n->encrypted_head,
1085                                    n->encrypted_tail,
1086                                    m);
1087       GNUNET_free (m);
1088     }
1089   if (NULL != n->th)
1090     {
1091       GNUNET_TRANSPORT_notify_transmit_ready_cancel (n->th);
1092       n->th = NULL;
1093     }
1094   if (n->retry_plaintext_task != GNUNET_SCHEDULER_NO_TASK)
1095     GNUNET_SCHEDULER_cancel (sched, n->retry_plaintext_task);
1096   if (n->retry_set_key_task != GNUNET_SCHEDULER_NO_TASK)
1097     GNUNET_SCHEDULER_cancel (sched, n->retry_set_key_task);
1098   if (n->quota_update_task != GNUNET_SCHEDULER_NO_TASK)
1099     GNUNET_SCHEDULER_cancel (sched, n->quota_update_task);
1100   if (n->dead_clean_task != GNUNET_SCHEDULER_NO_TASK)
1101     GNUNET_SCHEDULER_cancel (sched, n->dead_clean_task);
1102   if (n->keep_alive_task != GNUNET_SCHEDULER_NO_TASK)
1103       GNUNET_SCHEDULER_cancel (sched, n->keep_alive_task);
1104   if (n->status == PEER_STATE_KEY_CONFIRMED)
1105     GNUNET_STATISTICS_update (stats, gettext_noop ("# established sessions"), -1, GNUNET_NO);
1106   GNUNET_free_non_null (n->public_key);
1107   GNUNET_free_non_null (n->pending_ping);
1108   GNUNET_free_non_null (n->pending_pong);
1109   GNUNET_free (n);
1110 }
1111
1112
1113 /**
1114  * Check if we have encrypted messages for the specified neighbour
1115  * pending, and if so, check with the transport about sending them
1116  * out.
1117  *
1118  * @param n neighbour to check.
1119  */
1120 static void process_encrypted_neighbour_queue (struct Neighbour *n);
1121
1122
1123 /**
1124  * Encrypt size bytes from in and write the result to out.  Use the
1125  * key for outbound traffic of the given neighbour.
1126  *
1127  * @param n neighbour we are sending to
1128  * @param iv initialization vector to use
1129  * @param in ciphertext
1130  * @param out plaintext
1131  * @param size size of in/out
1132  * @return GNUNET_OK on success
1133  */
1134 static int
1135 do_encrypt (struct Neighbour *n,
1136             const GNUNET_HashCode * iv,
1137             const void *in, void *out, size_t size)
1138 {
1139   if (size != (uint16_t) size)
1140     {
1141       GNUNET_break (0);
1142       return GNUNET_NO;
1143     }
1144   GNUNET_assert (size ==
1145                  GNUNET_CRYPTO_aes_encrypt (in,
1146                                             (uint16_t) size,
1147                                             &n->encrypt_key,
1148                                             (const struct
1149                                              GNUNET_CRYPTO_AesInitializationVector
1150                                              *) iv, out));
1151   GNUNET_STATISTICS_update (stats, gettext_noop ("# bytes encrypted"), size, GNUNET_NO);
1152 #if DEBUG_CORE
1153   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1154               "Encrypted %u bytes for `%4s' using key %u\n",
1155               (unsigned int) size,
1156               GNUNET_i2s (&n->peer),
1157               (unsigned int) n->encrypt_key.crc32);
1158 #endif
1159   return GNUNET_OK;
1160 }
1161
1162
1163 /**
1164  * Consider freeing the given neighbour since we may not need
1165  * to keep it around anymore.
1166  *
1167  * @param n neighbour to consider discarding
1168  */
1169 static void
1170 consider_free_neighbour (struct Neighbour *n);
1171
1172
1173 /**
1174  * Task triggered when a neighbour entry is about to time out
1175  * (and we should prevent this by sending a PING).
1176  *
1177  * @param cls the 'struct Neighbour'
1178  * @param tc scheduler context (not used)
1179  */
1180 static void
1181 send_keep_alive (void *cls,
1182                  const struct GNUNET_SCHEDULER_TaskContext *tc)
1183 {
1184   struct Neighbour *n = cls;
1185   struct GNUNET_TIME_Relative retry;
1186   struct GNUNET_TIME_Relative left;
1187   struct MessageEntry *me;
1188   struct PingMessage pp;
1189   struct PingMessage *pm;
1190
1191   n->keep_alive_task = GNUNET_SCHEDULER_NO_TASK;
1192   /* send PING */
1193   me = GNUNET_malloc (sizeof (struct MessageEntry) +
1194                       sizeof (struct PingMessage));
1195   me->deadline = GNUNET_TIME_relative_to_absolute (MAX_PING_DELAY);
1196   me->priority = PING_PRIORITY;
1197   me->size = sizeof (struct PingMessage);
1198   GNUNET_CONTAINER_DLL_insert_after (n->encrypted_head,
1199                                      n->encrypted_tail,
1200                                      n->encrypted_tail,
1201                                      me);
1202   pm = (struct PingMessage *) &me[1];
1203   pm->header.size = htons (sizeof (struct PingMessage));
1204   pm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_PING);
1205   pp.challenge = htonl (n->ping_challenge);
1206   pp.target = n->peer;
1207 #if DEBUG_HANDSHAKE
1208   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1209               "Encrypting `%s' message with challenge %u for `%4s' using key %u.\n",
1210               "PING",
1211               (unsigned int) n->ping_challenge,
1212               GNUNET_i2s (&n->peer),
1213               (unsigned int) n->encrypt_key.crc32);
1214 #endif
1215   do_encrypt (n,
1216               &n->peer.hashPubKey,
1217               &pp.challenge,
1218               &pm->challenge,
1219               sizeof (struct PingMessage) -
1220               sizeof (struct GNUNET_MessageHeader));
1221   process_encrypted_neighbour_queue (n);
1222   /* reschedule PING job */
1223   left = GNUNET_TIME_absolute_get_remaining (GNUNET_TIME_absolute_add (n->last_activity,
1224                                                                        GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT));
1225   retry = GNUNET_TIME_relative_max (GNUNET_TIME_relative_divide (left, 2),
1226                                     MIN_PING_FREQUENCY);
1227   n->keep_alive_task
1228     = GNUNET_SCHEDULER_add_delayed (sched,
1229                                     retry,
1230                                     &send_keep_alive,
1231                                     n);
1232
1233 }
1234
1235
1236 /**
1237  * Task triggered when a neighbour entry might have gotten stale.
1238  *
1239  * @param cls the 'struct Neighbour'
1240  * @param tc scheduler context (not used)
1241  */
1242 static void
1243 consider_free_task (void *cls,
1244                     const struct GNUNET_SCHEDULER_TaskContext *tc)
1245 {
1246   struct Neighbour *n = cls;
1247
1248   n->dead_clean_task = GNUNET_SCHEDULER_NO_TASK;
1249   consider_free_neighbour (n);
1250 }
1251
1252
1253 /**
1254  * Consider freeing the given neighbour since we may not need
1255  * to keep it around anymore.
1256  *
1257  * @param n neighbour to consider discarding
1258  */
1259 static void
1260 consider_free_neighbour (struct Neighbour *n)
1261 {
1262   struct Neighbour *pos;
1263   struct Neighbour *prev;
1264   struct GNUNET_TIME_Relative left;
1265
1266   if ( (n->th != NULL) ||
1267        (n->pitr != NULL) ||
1268        (n->status == PEER_STATE_KEY_CONFIRMED) ||
1269        (GNUNET_YES == n->is_connected) )
1270     return; /* no chance */
1271
1272   left = GNUNET_TIME_absolute_get_remaining (GNUNET_TIME_absolute_add (n->last_activity,
1273                                                                        GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT));
1274   if (left.value > 0)
1275     {
1276       if (n->dead_clean_task != GNUNET_SCHEDULER_NO_TASK)
1277         GNUNET_SCHEDULER_cancel (sched, n->dead_clean_task);
1278       n->dead_clean_task = GNUNET_SCHEDULER_add_delayed (sched,
1279                                                          left,
1280                                                          &consider_free_task,
1281                                                          n);
1282       return;
1283     }
1284   /* actually free the neighbour... */
1285   prev = NULL;
1286   pos = neighbours;
1287   while (pos != n)
1288     {
1289       prev = pos;
1290       pos = pos->next;
1291     }
1292   if (prev == NULL)
1293     neighbours = n->next;
1294   else
1295     prev->next = n->next;
1296   GNUNET_assert (neighbour_count > 0);
1297   neighbour_count--;
1298   GNUNET_STATISTICS_set (stats,
1299                          gettext_noop ("# active neighbours"),
1300                          neighbour_count,
1301                          GNUNET_NO);
1302   free_neighbour (n);
1303 }
1304
1305
1306 /**
1307  * Function called when the transport service is ready to
1308  * receive an encrypted message for the respective peer
1309  *
1310  * @param cls neighbour to use message from
1311  * @param size number of bytes we can transmit
1312  * @param buf where to copy the message
1313  * @return number of bytes transmitted
1314  */
1315 static size_t
1316 notify_encrypted_transmit_ready (void *cls, size_t size, void *buf)
1317 {
1318   struct Neighbour *n = cls;
1319   struct MessageEntry *m;
1320   size_t ret;
1321   char *cbuf;
1322
1323   n->th = NULL;
1324   m = n->encrypted_head;
1325   if (m == NULL)
1326     return 0;
1327   GNUNET_CONTAINER_DLL_remove (n->encrypted_head,
1328                                n->encrypted_tail,
1329                                m);
1330   ret = 0;
1331   cbuf = buf;
1332   if (buf != NULL)
1333     {
1334       GNUNET_assert (size >= m->size);
1335       memcpy (cbuf, &m[1], m->size);
1336       ret = m->size;
1337       GNUNET_BANDWIDTH_tracker_consume (&n->available_send_window,
1338                                         m->size);
1339 #if DEBUG_CORE
1340       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1341                   "Copied message of type %u and size %u into transport buffer for `%4s'\n",
1342                   (unsigned int) ntohs (((struct GNUNET_MessageHeader *) &m[1])->type),
1343                   (unsigned int) ret,
1344                   GNUNET_i2s (&n->peer));
1345 #endif
1346       process_encrypted_neighbour_queue (n);
1347     }
1348   else
1349     {
1350 #if DEBUG_CORE
1351       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1352                   "Transmission of message of type %u and size %u failed\n",
1353                   (unsigned int) ntohs (((struct GNUNET_MessageHeader *) &m[1])->type),
1354                   (unsigned int) m->size);
1355 #endif
1356     }
1357   GNUNET_free (m);
1358   consider_free_neighbour (n);
1359   return ret;
1360 }
1361
1362
1363 /**
1364  * Check if we have plaintext messages for the specified neighbour
1365  * pending, and if so, consider batching and encrypting them (and
1366  * then trigger processing of the encrypted queue if needed).
1367  *
1368  * @param n neighbour to check.
1369  */
1370 static void process_plaintext_neighbour_queue (struct Neighbour *n);
1371
1372
1373 /**
1374  * Check if we have encrypted messages for the specified neighbour
1375  * pending, and if so, check with the transport about sending them
1376  * out.
1377  *
1378  * @param n neighbour to check.
1379  */
1380 static void
1381 process_encrypted_neighbour_queue (struct Neighbour *n)
1382 {
1383   struct MessageEntry *m;
1384
1385   if (n->th != NULL)
1386     return;  /* request already pending */
1387   m = n->encrypted_head;
1388   if (m == NULL)
1389     {
1390       /* encrypted queue empty, try plaintext instead */
1391       process_plaintext_neighbour_queue (n);
1392       return;
1393     }
1394 #if DEBUG_CORE
1395   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1396               "Asking transport for transmission of %u bytes to `%4s' in next %llu ms\n",
1397               (unsigned int) m->size,
1398               GNUNET_i2s (&n->peer),
1399               (unsigned long long) GNUNET_TIME_absolute_get_remaining (m->deadline).
1400               value);
1401 #endif
1402   n->th =
1403     GNUNET_TRANSPORT_notify_transmit_ready (transport, &n->peer,
1404                                             m->size,
1405                                             m->priority,
1406                                             GNUNET_TIME_absolute_get_remaining
1407                                             (m->deadline),
1408                                             &notify_encrypted_transmit_ready,
1409                                             n);
1410   if (n->th == NULL)
1411     {
1412       /* message request too large or duplicate request */
1413       GNUNET_break (0);
1414       /* discard encrypted message */
1415       GNUNET_CONTAINER_DLL_remove (n->encrypted_head,
1416                                    n->encrypted_tail,
1417                                    m);
1418       GNUNET_free (m);
1419       process_encrypted_neighbour_queue (n);
1420     }
1421 }
1422
1423
1424 /**
1425  * Decrypt size bytes from in and write the result to out.  Use the
1426  * key for inbound traffic of the given neighbour.  This function does
1427  * NOT do any integrity-checks on the result.
1428  *
1429  * @param n neighbour we are receiving from
1430  * @param iv initialization vector to use
1431  * @param in ciphertext
1432  * @param out plaintext
1433  * @param size size of in/out
1434  * @return GNUNET_OK on success
1435  */
1436 static int
1437 do_decrypt (struct Neighbour *n,
1438             const GNUNET_HashCode * iv,
1439             const void *in, void *out, size_t size)
1440 {
1441   if (size != (uint16_t) size)
1442     {
1443       GNUNET_break (0);
1444       return GNUNET_NO;
1445     }
1446   if ((n->status != PEER_STATE_KEY_RECEIVED) &&
1447       (n->status != PEER_STATE_KEY_CONFIRMED))
1448     {
1449       GNUNET_break_op (0);
1450       return GNUNET_SYSERR;
1451     }
1452   if (size !=
1453       GNUNET_CRYPTO_aes_decrypt (in,
1454                                  (uint16_t) size,
1455                                  &n->decrypt_key,
1456                                  (const struct
1457                                   GNUNET_CRYPTO_AesInitializationVector *) iv,
1458                                  out))
1459     {
1460       GNUNET_break (0);
1461       return GNUNET_SYSERR;
1462     }
1463   GNUNET_STATISTICS_update (stats, gettext_noop ("# bytes decrypted"), size, GNUNET_NO);
1464 #if DEBUG_CORE
1465   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1466               "Decrypted %u bytes from `%4s' using key %u\n",
1467               (unsigned int) size,
1468               GNUNET_i2s (&n->peer),
1469               (unsigned int) n->decrypt_key.crc32);
1470 #endif
1471   return GNUNET_OK;
1472 }
1473
1474
1475 /**
1476  * Select messages for transmission.  This heuristic uses a combination
1477  * of earliest deadline first (EDF) scheduling (with bounded horizon)
1478  * and priority-based discard (in case no feasible schedule exist) and
1479  * speculative optimization (defer any kind of transmission until
1480  * we either create a batch of significant size, 25% of max, or until
1481  * we are close to a deadline).  Furthermore, when scheduling the
1482  * heuristic also packs as many messages into the batch as possible,
1483  * starting with those with the earliest deadline.  Yes, this is fun.
1484  *
1485  * @param n neighbour to select messages from
1486  * @param size number of bytes to select for transmission
1487  * @param retry_time set to the time when we should try again
1488  *        (only valid if this function returns zero)
1489  * @return number of bytes selected, or 0 if we decided to
1490  *         defer scheduling overall; in that case, retry_time is set.
1491  */
1492 static size_t
1493 select_messages (struct Neighbour *n,
1494                  size_t size, struct GNUNET_TIME_Relative *retry_time)
1495 {
1496   struct MessageEntry *pos;
1497   struct MessageEntry *min;
1498   struct MessageEntry *last;
1499   unsigned int min_prio;
1500   struct GNUNET_TIME_Absolute t;
1501   struct GNUNET_TIME_Absolute now;
1502   struct GNUNET_TIME_Relative delta;
1503   uint64_t avail;
1504   struct GNUNET_TIME_Relative slack;     /* how long could we wait before missing deadlines? */
1505   size_t off;
1506   uint64_t tsize;
1507   unsigned int queue_size;
1508   int discard_low_prio;
1509
1510   GNUNET_assert (NULL != n->messages);
1511   now = GNUNET_TIME_absolute_get ();
1512   /* last entry in linked list of messages processed */
1513   last = NULL;
1514   /* should we remove the entry with the lowest
1515      priority from consideration for scheduling at the
1516      end of the loop? */
1517   queue_size = 0;
1518   tsize = 0;
1519   pos = n->messages;
1520   while (pos != NULL)
1521     {
1522       queue_size++;
1523       tsize += pos->size;
1524       pos = pos->next;
1525     }
1526   discard_low_prio = GNUNET_YES;
1527   while (GNUNET_YES == discard_low_prio)
1528     {
1529       min = NULL;
1530       min_prio = -1;
1531       discard_low_prio = GNUNET_NO;
1532       /* calculate number of bytes available for transmission at time "t" */
1533       avail = GNUNET_BANDWIDTH_tracker_get_available (&n->available_send_window);
1534       t = now;
1535       /* how many bytes have we (hypothetically) scheduled so far */
1536       off = 0;
1537       /* maximum time we can wait before transmitting anything
1538          and still make all of our deadlines */
1539       slack = GNUNET_CONSTANTS_MAX_CORK_DELAY;
1540       pos = n->messages;
1541       /* note that we use "*2" here because we want to look
1542          a bit further into the future; much more makes no
1543          sense since new message might be scheduled in the
1544          meantime... */
1545       while ((pos != NULL) && (off < size * 2))
1546         {
1547           if (pos->do_transmit == GNUNET_YES)
1548             {
1549               /* already removed from consideration */
1550               pos = pos->next;
1551               continue;
1552             }
1553           if (discard_low_prio == GNUNET_NO)
1554             {
1555               delta = GNUNET_TIME_absolute_get_difference (t, pos->deadline);
1556               if (delta.value > 0)
1557                 {
1558                   // FIXME: HUH? Check!
1559                   t = pos->deadline;
1560                   avail += GNUNET_BANDWIDTH_value_get_available_until (n->bw_out,
1561                                                                        delta);
1562                 }
1563               if (avail < pos->size)
1564                 {
1565                   // FIXME: HUH? Check!
1566                   discard_low_prio = GNUNET_YES;        /* we could not schedule this one! */
1567                 }
1568               else
1569                 {
1570                   avail -= pos->size;
1571                   /* update slack, considering both its absolute deadline
1572                      and relative deadlines caused by other messages
1573                      with their respective load */
1574                   slack = GNUNET_TIME_relative_min (slack,
1575                                                     GNUNET_BANDWIDTH_value_get_delay_for (n->bw_out,
1576                                                                                           avail));
1577                   if (pos->deadline.value <= now.value)
1578                     {
1579                       /* now or never */
1580                       slack = GNUNET_TIME_UNIT_ZERO;
1581                     }
1582                   else if (GNUNET_YES == pos->got_slack)
1583                     {
1584                       /* should be soon now! */
1585                       slack = GNUNET_TIME_relative_min (slack,
1586                                                         GNUNET_TIME_absolute_get_remaining (pos->slack_deadline));
1587                     }
1588                   else
1589                     {
1590                       slack =
1591                         GNUNET_TIME_relative_min (slack,
1592                                                   GNUNET_TIME_absolute_get_difference (now, pos->deadline));
1593                       pos->got_slack = GNUNET_YES;
1594                       pos->slack_deadline = GNUNET_TIME_absolute_min (pos->deadline,
1595                                                                       GNUNET_TIME_relative_to_absolute (GNUNET_CONSTANTS_MAX_CORK_DELAY));
1596                     }
1597                 }
1598             }
1599           off += pos->size;
1600           t = GNUNET_TIME_absolute_max (pos->deadline, t); // HUH? Check!
1601           if (pos->priority <= min_prio)
1602             {
1603               /* update min for discard */
1604               min_prio = pos->priority;
1605               min = pos;
1606             }
1607           pos = pos->next;
1608         }
1609       if (discard_low_prio)
1610         {
1611           GNUNET_assert (min != NULL);
1612           /* remove lowest-priority entry from consideration */
1613           min->do_transmit = GNUNET_YES;        /* means: discard (for now) */
1614         }
1615       last = pos;
1616     }
1617   /* guard against sending "tiny" messages with large headers without
1618      urgent deadlines */
1619   if ( (slack.value > 0) &&
1620        (size > 4 * off) &&
1621        (queue_size <= MAX_PEER_QUEUE_SIZE - 2) )
1622     {
1623       /* less than 25% of message would be filled with deadlines still
1624          being met if we delay by one second or more; so just wait for
1625          more data; but do not wait longer than 1s (since we don't want
1626          to delay messages for a really long time either). */
1627       *retry_time = GNUNET_CONSTANTS_MAX_CORK_DELAY;
1628       /* reset do_transmit values for next time */
1629       while (pos != last)
1630         {
1631           pos->do_transmit = GNUNET_NO;
1632           pos = pos->next;
1633         }
1634 #if DEBUG_CORE
1635       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1636                   "Deferring transmission for %llums due to underfull message buffer size (%u/%u)\n",
1637                   (unsigned long long) slack.value,
1638                   (unsigned int) off,
1639                   (unsigned int) size);
1640 #endif
1641       return 0;
1642     }
1643   /* select marked messages (up to size) for transmission */
1644   off = 0;
1645   pos = n->messages;
1646   while (pos != last)
1647     {
1648       if ((pos->size <= size) && (pos->do_transmit == GNUNET_NO))
1649         {
1650           pos->do_transmit = GNUNET_YES;        /* mark for transmission */
1651           off += pos->size;
1652           size -= pos->size;
1653         }
1654       else
1655         pos->do_transmit = GNUNET_NO;   /* mark for not transmitting! */
1656       pos = pos->next;
1657     }
1658 #if DEBUG_CORE
1659   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1660               "Selected %llu/%llu bytes of %u/%u plaintext messages for transmission to `%4s'.\n",
1661               (unsigned long long) off, (unsigned long long) tsize,
1662               queue_size, (unsigned int) MAX_PEER_QUEUE_SIZE,
1663               GNUNET_i2s (&n->peer));
1664 #endif
1665   return off;
1666 }
1667
1668
1669 /**
1670  * Batch multiple messages into a larger buffer.
1671  *
1672  * @param n neighbour to take messages from
1673  * @param buf target buffer
1674  * @param size size of buf
1675  * @param deadline set to transmission deadline for the result
1676  * @param retry_time set to the time when we should try again
1677  *        (only valid if this function returns zero)
1678  * @param priority set to the priority of the batch
1679  * @return number of bytes written to buf (can be zero)
1680  */
1681 static size_t
1682 batch_message (struct Neighbour *n,
1683                char *buf,
1684                size_t size,
1685                struct GNUNET_TIME_Absolute *deadline,
1686                struct GNUNET_TIME_Relative *retry_time,
1687                unsigned int *priority)
1688 {
1689   char ntmb[GNUNET_SERVER_MAX_MESSAGE_SIZE];
1690   struct NotifyTrafficMessage *ntm = (struct NotifyTrafficMessage*) ntmb;
1691   struct MessageEntry *pos;
1692   struct MessageEntry *prev;
1693   struct MessageEntry *next;
1694   size_t ret;
1695
1696   ret = 0;
1697   *priority = 0;
1698   *deadline = GNUNET_TIME_UNIT_FOREVER_ABS;
1699   *retry_time = GNUNET_TIME_UNIT_FOREVER_REL;
1700   if (0 == select_messages (n, size, retry_time))
1701     {
1702 #if DEBUG_CORE
1703       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1704                   "No messages selected, will try again in %llu ms\n",
1705                   retry_time->value);
1706 #endif
1707       return 0;
1708     }
1709   ntm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_OUTBOUND);
1710   ntm->distance = htonl (n->last_distance);
1711   ntm->latency = GNUNET_TIME_relative_hton (n->last_latency);
1712   ntm->peer = n->peer;
1713
1714   pos = n->messages;
1715   prev = NULL;
1716   while ((pos != NULL) && (size >= sizeof (struct GNUNET_MessageHeader)))
1717     {
1718       next = pos->next;
1719       if (GNUNET_YES == pos->do_transmit)
1720         {
1721           GNUNET_assert (pos->size <= size);
1722           /* do notifications */
1723           /* FIXME: track if we have *any* client that wants
1724              full notifications and only do this if that is
1725              actually true */
1726           if (pos->size < GNUNET_SERVER_MAX_MESSAGE_SIZE - sizeof (struct NotifyTrafficMessage))
1727             {
1728               memcpy (&ntm[1], &pos[1], pos->size);
1729               ntm->header.size = htons (sizeof (struct NotifyTrafficMessage) +
1730                                         sizeof (struct GNUNET_MessageHeader));
1731               send_to_all_clients (&ntm->header,
1732                                    GNUNET_YES,
1733                                    GNUNET_CORE_OPTION_SEND_HDR_OUTBOUND);
1734             }
1735           else
1736             {
1737               /* message too large for 'full' notifications, we do at
1738                  least the 'hdr' type */
1739               memcpy (&ntm[1],
1740                       &pos[1],
1741                       sizeof (struct GNUNET_MessageHeader));
1742             }
1743           ntm->header.size = htons (sizeof (struct NotifyTrafficMessage) +
1744                                     pos->size);
1745           send_to_all_clients (&ntm->header,
1746                                GNUNET_YES,
1747                                GNUNET_CORE_OPTION_SEND_FULL_OUTBOUND);
1748 #if DEBUG_HANDSHAKE
1749           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1750                       "Encrypting message of type %u\n",
1751                       (unsigned int) ntohs(((struct GNUNET_MessageHeader*)&pos[1])->type));
1752 #endif
1753           /* copy for encrypted transmission */
1754           memcpy (&buf[ret], &pos[1], pos->size);
1755           ret += pos->size;
1756           size -= pos->size;
1757           *priority += pos->priority;
1758 #if DEBUG_CORE
1759           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1760                       "Adding plaintext message of size %u with deadline %llu ms to batch\n",
1761                       (unsigned int) pos->size,
1762                       (unsigned long long) GNUNET_TIME_absolute_get_remaining (pos->deadline).value);
1763 #endif
1764           deadline->value = GNUNET_MIN (deadline->value, pos->deadline.value);
1765           GNUNET_free (pos);
1766           if (prev == NULL)
1767             n->messages = next;
1768           else
1769             prev->next = next;
1770         }
1771       else
1772         {
1773           prev = pos;
1774         }
1775       pos = next;
1776     }
1777 #if DEBUG_CORE
1778   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1779               "Deadline for message batch is %llu ms\n",
1780               GNUNET_TIME_absolute_get_remaining (*deadline).value);
1781 #endif
1782   return ret;
1783 }
1784
1785
1786 /**
1787  * Remove messages with deadlines that have long expired from
1788  * the queue.
1789  *
1790  * @param n neighbour to inspect
1791  */
1792 static void
1793 discard_expired_messages (struct Neighbour *n)
1794 {
1795   struct MessageEntry *prev;
1796   struct MessageEntry *next;
1797   struct MessageEntry *pos;
1798   struct GNUNET_TIME_Absolute now;
1799   struct GNUNET_TIME_Relative delta;
1800
1801   now = GNUNET_TIME_absolute_get ();
1802   prev = NULL;
1803   pos = n->messages;
1804   while (pos != NULL)
1805     {
1806       next = pos->next;
1807       delta = GNUNET_TIME_absolute_get_difference (pos->deadline, now);
1808       if (delta.value > PAST_EXPIRATION_DISCARD_TIME.value)
1809         {
1810 #if DEBUG_CORE
1811           GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1812                       "Message is %llu ms past due, discarding.\n",
1813                       delta.value);
1814 #endif
1815           if (prev == NULL)
1816             n->messages = next;
1817           else
1818             prev->next = next;
1819           GNUNET_free (pos);
1820         }
1821       else
1822         prev = pos;
1823       pos = next;
1824     }
1825 }
1826
1827
1828 /**
1829  * Signature of the main function of a task.
1830  *
1831  * @param cls closure
1832  * @param tc context information (why was this task triggered now)
1833  */
1834 static void
1835 retry_plaintext_processing (void *cls,
1836                             const struct GNUNET_SCHEDULER_TaskContext *tc)
1837 {
1838   struct Neighbour *n = cls;
1839
1840   n->retry_plaintext_task = GNUNET_SCHEDULER_NO_TASK;
1841   process_plaintext_neighbour_queue (n);
1842 }
1843
1844
1845 /**
1846  * Send our key (and encrypted PING) to the other peer.
1847  *
1848  * @param n the other peer
1849  */
1850 static void send_key (struct Neighbour *n);
1851
1852 /**
1853  * Task that will retry "send_key" if our previous attempt failed
1854  * to yield a PONG.
1855  */
1856 static void
1857 set_key_retry_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1858 {
1859   struct Neighbour *n = cls;
1860
1861 #if DEBUG_CORE
1862   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1863               "Retrying key transmission to `%4s'\n",
1864               GNUNET_i2s (&n->peer));
1865 #endif
1866   n->retry_set_key_task = GNUNET_SCHEDULER_NO_TASK;
1867   n->set_key_retry_frequency =
1868     GNUNET_TIME_relative_multiply (n->set_key_retry_frequency, 2);
1869   send_key (n);
1870 }
1871
1872
1873 /**
1874  * Check if we have plaintext messages for the specified neighbour
1875  * pending, and if so, consider batching and encrypting them (and
1876  * then trigger processing of the encrypted queue if needed).
1877  *
1878  * @param n neighbour to check.
1879  */
1880 static void
1881 process_plaintext_neighbour_queue (struct Neighbour *n)
1882 {
1883   char pbuf[MAX_ENCRYPTED_MESSAGE_SIZE];        /* plaintext */
1884   size_t used;
1885   size_t esize;
1886   struct EncryptedMessage *em;  /* encrypted message */
1887   struct EncryptedMessage *ph;  /* plaintext header */
1888   struct MessageEntry *me;
1889   unsigned int priority;
1890   struct GNUNET_TIME_Absolute deadline;
1891   struct GNUNET_TIME_Relative retry_time;
1892   GNUNET_HashCode iv;
1893
1894   if (n->retry_plaintext_task != GNUNET_SCHEDULER_NO_TASK)
1895     {
1896       GNUNET_SCHEDULER_cancel (sched, n->retry_plaintext_task);
1897       n->retry_plaintext_task = GNUNET_SCHEDULER_NO_TASK;
1898     }
1899   switch (n->status)
1900     {
1901     case PEER_STATE_DOWN:
1902       send_key (n);
1903 #if DEBUG_CORE
1904       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1905                   "Not yet connected to `%4s', deferring processing of plaintext messages.\n",
1906                   GNUNET_i2s(&n->peer));
1907 #endif
1908       return;
1909     case PEER_STATE_KEY_SENT:
1910       if (n->retry_set_key_task == GNUNET_SCHEDULER_NO_TASK)
1911         n->retry_set_key_task
1912           = GNUNET_SCHEDULER_add_delayed (sched,
1913                                           n->set_key_retry_frequency,
1914                                           &set_key_retry_task, n);
1915 #if DEBUG_CORE
1916       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1917                   "Not yet connected to `%4s', deferring processing of plaintext messages.\n",
1918                   GNUNET_i2s(&n->peer));
1919 #endif
1920       return;
1921     case PEER_STATE_KEY_RECEIVED:
1922       if (n->retry_set_key_task == GNUNET_SCHEDULER_NO_TASK)
1923         n->retry_set_key_task
1924           = GNUNET_SCHEDULER_add_delayed (sched,
1925                                           n->set_key_retry_frequency,
1926                                           &set_key_retry_task, n);
1927 #if DEBUG_CORE
1928       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1929                   "Not yet connected to `%4s', deferring processing of plaintext messages.\n",
1930                   GNUNET_i2s(&n->peer));
1931 #endif
1932       return;
1933     case PEER_STATE_KEY_CONFIRMED:
1934       /* ready to continue */
1935       break;
1936     }
1937   discard_expired_messages (n);
1938   if (n->messages == NULL)
1939     {
1940 #if DEBUG_CORE
1941       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1942                   "Plaintext message queue for `%4s' is empty.\n",
1943                   GNUNET_i2s(&n->peer));
1944 #endif
1945       return;                   /* no pending messages */
1946     }
1947   if (n->encrypted_head != NULL)
1948     {
1949 #if DEBUG_CORE
1950       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1951                   "Encrypted message queue for `%4s' is still full, delaying plaintext processing.\n",
1952                   GNUNET_i2s(&n->peer));
1953 #endif
1954       return;                   /* wait for messages already encrypted to be
1955                                    processed first! */
1956     }
1957   ph = (struct EncryptedMessage *) pbuf;
1958   deadline = GNUNET_TIME_UNIT_FOREVER_ABS;
1959   priority = 0;
1960   used = sizeof (struct EncryptedMessage);
1961   used += batch_message (n,
1962                          &pbuf[used],
1963                          MAX_ENCRYPTED_MESSAGE_SIZE - used,
1964                          &deadline, &retry_time, &priority);
1965   if (used == sizeof (struct EncryptedMessage))
1966     {
1967 #if DEBUG_CORE
1968       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1969                   "No messages selected for transmission to `%4s' at this time, will try again later.\n",
1970                   GNUNET_i2s(&n->peer));
1971 #endif
1972       /* no messages selected for sending, try again later... */
1973       n->retry_plaintext_task =
1974         GNUNET_SCHEDULER_add_delayed (sched,
1975                                       retry_time,
1976                                       &retry_plaintext_processing, n);
1977       return;
1978     }
1979 #if DEBUG_CORE_QUOTA
1980   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1981               "Sending %u b/s as new limit to peer `%4s'\n",
1982               (unsigned int) ntohl (n->bw_in.value__),
1983               GNUNET_i2s (&n->peer));
1984 #endif
1985   ph->iv_seed = htonl (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, -1));
1986   ph->sequence_number = htonl (++n->last_sequence_number_sent);
1987   ph->inbound_bw_limit = n->bw_in;
1988   ph->timestamp = GNUNET_TIME_absolute_hton (GNUNET_TIME_absolute_get ());
1989
1990   /* setup encryption message header */
1991   me = GNUNET_malloc (sizeof (struct MessageEntry) + used);
1992   me->deadline = deadline;
1993   me->priority = priority;
1994   me->size = used;
1995   em = (struct EncryptedMessage *) &me[1];
1996   em->header.size = htons (used);
1997   em->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_ENCRYPTED_MESSAGE);
1998   em->iv_seed = ph->iv_seed;
1999   esize = used - ENCRYPTED_HEADER_SIZE;
2000   GNUNET_CRYPTO_hash (&ph->sequence_number,
2001                       esize - sizeof (GNUNET_HashCode),
2002                       &ph->plaintext_hash);
2003   GNUNET_CRYPTO_hash (&ph->iv_seed, sizeof (uint32_t), &iv);
2004 #if DEBUG_HANDSHAKE
2005   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2006               "Hashed %u bytes of plaintext (`%s') using IV `%d'\n",
2007               (unsigned int) (esize - sizeof (GNUNET_HashCode)),
2008               GNUNET_h2s (&ph->plaintext_hash),
2009               (int) ph->iv_seed);
2010 #endif
2011   /* encrypt */
2012 #if DEBUG_HANDSHAKE
2013   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2014               "Encrypting %u bytes of plaintext messages for `%4s' for transmission in %llums.\n",
2015               (unsigned int) esize,
2016               GNUNET_i2s(&n->peer),
2017               (unsigned long long) GNUNET_TIME_absolute_get_remaining (deadline).value);
2018 #endif
2019   GNUNET_assert (GNUNET_OK ==
2020                  do_encrypt (n,
2021                              &iv,
2022                              &ph->plaintext_hash,
2023                              &em->plaintext_hash, esize));
2024   /* append to transmission list */
2025   GNUNET_CONTAINER_DLL_insert_after (n->encrypted_head,
2026                                      n->encrypted_tail,
2027                                      n->encrypted_tail,
2028                                      me);
2029   process_encrypted_neighbour_queue (n);
2030 }
2031
2032
2033 /**
2034  * Function that recalculates the bandwidth quota for the
2035  * given neighbour and transmits it to the transport service.
2036  *
2037  * @param cls neighbour for the quota update
2038  * @param tc context
2039  */
2040 static void
2041 neighbour_quota_update (void *cls,
2042                         const struct GNUNET_SCHEDULER_TaskContext *tc);
2043
2044
2045 /**
2046  * Schedule the task that will recalculate the bandwidth
2047  * quota for this peer (and possibly force a disconnect of
2048  * idle peers by calculating a bandwidth of zero).
2049  */
2050 static void
2051 schedule_quota_update (struct Neighbour *n)
2052 {
2053   GNUNET_assert (n->quota_update_task ==
2054                  GNUNET_SCHEDULER_NO_TASK);
2055   n->quota_update_task
2056     = GNUNET_SCHEDULER_add_delayed (sched,
2057                                     QUOTA_UPDATE_FREQUENCY,
2058                                     &neighbour_quota_update,
2059                                     n);
2060 }
2061
2062
2063 /**
2064  * Initialize a new 'struct Neighbour'.
2065  *
2066  * @param pid ID of the new neighbour
2067  * @return handle for the new neighbour
2068  */
2069 static struct Neighbour *
2070 create_neighbour (const struct GNUNET_PeerIdentity *pid)
2071 {
2072   struct Neighbour *n;
2073   struct GNUNET_TIME_Absolute now;
2074
2075   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2076               "Creating neighbour entry for peer `%4s'\n",
2077               GNUNET_i2s (pid));
2078   n = GNUNET_malloc (sizeof (struct Neighbour));
2079   n->next = neighbours;
2080   neighbours = n;
2081   neighbour_count++;
2082   GNUNET_STATISTICS_set (stats, gettext_noop ("# active neighbours"), neighbour_count, GNUNET_NO);
2083   n->peer = *pid;
2084   GNUNET_CRYPTO_aes_create_session_key (&n->encrypt_key);
2085   now = GNUNET_TIME_absolute_get ();
2086   n->encrypt_key_created = now;
2087   n->last_activity = now;
2088   n->set_key_retry_frequency = INITIAL_SET_KEY_RETRY_FREQUENCY;
2089   n->bw_in = GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT;
2090   n->bw_out = GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT;
2091   n->bw_out_internal_limit = GNUNET_BANDWIDTH_value_init ((uint32_t) - 1);
2092   n->bw_out_external_limit = GNUNET_CONSTANTS_DEFAULT_BW_IN_OUT;
2093   n->ping_challenge = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
2094                                                 (uint32_t) - 1);
2095   neighbour_quota_update (n, NULL);
2096   return n;
2097 }
2098
2099
2100 /**
2101  * Handle CORE_SEND request.
2102  *
2103  * @param cls unused
2104  * @param client the client issuing the request
2105  * @param message the "struct SendMessage"
2106  */
2107 static void
2108 handle_client_send (void *cls,
2109                     struct GNUNET_SERVER_Client *client,
2110                     const struct GNUNET_MessageHeader *message)
2111 {
2112   const struct SendMessage *sm;
2113   struct Neighbour *n;
2114   struct MessageEntry *prev;
2115   struct MessageEntry *pos;
2116   struct MessageEntry *e;
2117   struct MessageEntry *min_prio_entry;
2118   struct MessageEntry *min_prio_prev;
2119   unsigned int min_prio;
2120   unsigned int queue_size;
2121   uint16_t msize;
2122
2123   msize = ntohs (message->size);
2124   if (msize <
2125       sizeof (struct SendMessage) + sizeof (struct GNUNET_MessageHeader))
2126     {
2127       GNUNET_break (0);
2128       if (client != NULL)
2129         GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
2130       return;
2131     }
2132   sm = (const struct SendMessage *) message;
2133   msize -= sizeof (struct SendMessage);
2134   if (0 == memcmp (&sm->peer, &my_identity, sizeof (struct GNUNET_PeerIdentity)))
2135     {
2136       /* FIXME: should we not allow loopback-injection here? */
2137       GNUNET_break (0);
2138       if (client != NULL)
2139         GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
2140       return;
2141     }
2142   n = find_neighbour (&sm->peer);
2143   if (n == NULL)
2144     n = create_neighbour (&sm->peer);
2145 #if DEBUG_CORE
2146   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2147               "Core received `%s' request, queueing %u bytes of plaintext data for transmission to `%4s'.\n",
2148               "SEND",
2149               (unsigned int) msize,
2150               GNUNET_i2s (&sm->peer));
2151 #endif
2152   /* bound queue size */
2153   discard_expired_messages (n);
2154   min_prio = (unsigned int) -1;
2155   min_prio_entry = NULL;
2156   min_prio_prev = NULL;
2157   queue_size = 0;
2158   prev = NULL;
2159   pos = n->messages;
2160   while (pos != NULL)
2161     {
2162       if (pos->priority < min_prio)
2163         {
2164           min_prio_entry = pos;
2165           min_prio_prev = prev;
2166           min_prio = pos->priority;
2167         }
2168       queue_size++;
2169       prev = pos;
2170       pos = pos->next;
2171     }
2172   if (queue_size >= MAX_PEER_QUEUE_SIZE)
2173     {
2174       /* queue full */
2175       if (ntohl(sm->priority) <= min_prio)
2176         {
2177           /* discard new entry */
2178 #if DEBUG_CORE
2179           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2180                       "Queue full (%u/%u), discarding new request (%u bytes of type %u)\n",
2181                       queue_size,
2182                       (unsigned int) MAX_PEER_QUEUE_SIZE,
2183                       (unsigned int) msize,
2184                       (unsigned int) ntohs (message->type));
2185 #endif
2186           if (client != NULL)
2187             GNUNET_SERVER_receive_done (client, GNUNET_OK);
2188           return;
2189         }
2190       /* discard "min_prio_entry" */
2191 #if DEBUG_CORE
2192       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2193                   "Queue full, discarding existing older request\n");
2194 #endif
2195       if (min_prio_prev == NULL)
2196         n->messages = min_prio_entry->next;
2197       else
2198         min_prio_prev->next = min_prio_entry->next;
2199       GNUNET_free (min_prio_entry);
2200     }
2201
2202 #if DEBUG_CORE
2203   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2204               "Adding transmission request for `%4s' of size %u to queue\n",
2205               GNUNET_i2s (&sm->peer),
2206               (unsigned int) msize);
2207 #endif
2208   e = GNUNET_malloc (sizeof (struct MessageEntry) + msize);
2209   e->deadline = GNUNET_TIME_absolute_ntoh (sm->deadline);
2210   e->priority = ntohl (sm->priority);
2211   e->size = msize;
2212   memcpy (&e[1], &sm[1], msize);
2213
2214   /* insert, keep list sorted by deadline */
2215   prev = NULL;
2216   pos = n->messages;
2217   while ((pos != NULL) && (pos->deadline.value < e->deadline.value))
2218     {
2219       prev = pos;
2220       pos = pos->next;
2221     }
2222   if (prev == NULL)
2223     n->messages = e;
2224   else
2225     prev->next = e;
2226   e->next = pos;
2227
2228   /* consider scheduling now */
2229   process_plaintext_neighbour_queue (n);
2230   if (client != NULL)
2231     GNUNET_SERVER_receive_done (client, GNUNET_OK);
2232 }
2233
2234
2235 /**
2236  * Function called when the transport service is ready to
2237  * receive a message.  Only resets 'n->th' to NULL.
2238  *
2239  * @param cls neighbour to use message from
2240  * @param size number of bytes we can transmit
2241  * @param buf where to copy the message
2242  * @return number of bytes transmitted
2243  */
2244 static size_t
2245 notify_transport_connect_done (void *cls, size_t size, void *buf)
2246 {
2247   struct Neighbour *n = cls;
2248
2249   n->th = NULL;
2250   if (buf == NULL)
2251     {
2252       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2253                   _("Failed to connect to `%4s': transport failed to connect\n"),
2254                   GNUNET_i2s (&n->peer));
2255       return 0;
2256     }
2257   send_key (n);
2258   return 0;
2259 }
2260
2261
2262 /**
2263  * Handle CORE_REQUEST_CONNECT request.
2264  *
2265  * @param cls unused
2266  * @param client the client issuing the request
2267  * @param message the "struct ConnectMessage"
2268  */
2269 static void
2270 handle_client_request_connect (void *cls,
2271                                struct GNUNET_SERVER_Client *client,
2272                                const struct GNUNET_MessageHeader *message)
2273 {
2274   const struct ConnectMessage *cm = (const struct ConnectMessage*) message;
2275   struct Neighbour *n;
2276   struct GNUNET_TIME_Relative timeout;
2277
2278   if (0 == memcmp (&cm->peer, &my_identity, sizeof (struct GNUNET_PeerIdentity)))
2279     {
2280       GNUNET_break (0);
2281       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
2282       return;
2283     }
2284   GNUNET_SERVER_receive_done (client, GNUNET_OK);
2285   n = find_neighbour (&cm->peer);
2286   if (n == NULL)
2287     n = create_neighbour (&cm->peer);
2288   if ( (n->is_connected) ||
2289        (n->th != NULL) )
2290     return; /* already connected, or at least trying */
2291   GNUNET_STATISTICS_update (stats, gettext_noop ("# connection requests received"), 1, GNUNET_NO);
2292 #if DEBUG_CORE
2293   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2294               "Core received `%s' request for `%4s', will try to establish connection\n",
2295               "REQUEST_CONNECT",
2296               GNUNET_i2s (&cm->peer));
2297 #endif
2298   timeout = GNUNET_TIME_relative_ntoh (cm->timeout);
2299   /* ask transport to connect to the peer */
2300   n->th = GNUNET_TRANSPORT_notify_transmit_ready (transport,
2301                                                   &cm->peer,
2302                                                   sizeof (struct GNUNET_MessageHeader), 0,
2303                                                   timeout,
2304                                                   &notify_transport_connect_done,
2305                                                   n);
2306   GNUNET_break (NULL != n->th);
2307 }
2308
2309
2310 /**
2311  * List of handlers for the messages understood by this
2312  * service.
2313  */
2314 static struct GNUNET_SERVER_MessageHandler handlers[] = {
2315   {&handle_client_init, NULL,
2316    GNUNET_MESSAGE_TYPE_CORE_INIT, 0},
2317   {&handle_client_request_info, NULL,
2318    GNUNET_MESSAGE_TYPE_CORE_REQUEST_INFO,
2319    sizeof (struct RequestInfoMessage)},
2320   {&handle_client_send, NULL,
2321    GNUNET_MESSAGE_TYPE_CORE_SEND, 0},
2322   {&handle_client_request_connect, NULL,
2323    GNUNET_MESSAGE_TYPE_CORE_REQUEST_CONNECT,
2324    sizeof (struct ConnectMessage)},
2325   {NULL, NULL, 0, 0}
2326 };
2327
2328
2329 /**
2330  * PEERINFO is giving us a HELLO for a peer.  Add the public key to
2331  * the neighbour's struct and retry send_key.  Or, if we did not get a
2332  * HELLO, just do nothing.
2333  *
2334  * @param cls the 'struct Neighbour' to retry sending the key for
2335  * @param peer the peer for which this is the HELLO
2336  * @param hello HELLO message of that peer
2337  * @param trust amount of trust we currently have in that peer
2338  */
2339 static void
2340 process_hello_retry_send_key (void *cls,
2341                               const struct GNUNET_PeerIdentity *peer,
2342                               const struct GNUNET_HELLO_Message *hello,
2343                               uint32_t trust)
2344 {
2345   struct Neighbour *n = cls;
2346
2347   if (peer == NULL)
2348     {
2349 #if DEBUG_CORE
2350       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2351                   "Entered `process_hello_retry_send_key' and `peer' is NULL!\n");
2352 #endif
2353       n->pitr = NULL;
2354       if (n->public_key != NULL)
2355         {
2356           GNUNET_STATISTICS_update (stats,
2357                                     gettext_noop ("# SETKEY messages deferred (need public key)"),
2358                                     -1,
2359                                     GNUNET_NO);
2360           send_key (n);
2361         }
2362       else
2363         {
2364           GNUNET_STATISTICS_update (stats,
2365                                     gettext_noop ("# Delayed connecting due to lack of public key"),
2366                                     1,
2367                                     GNUNET_NO);
2368           if (GNUNET_SCHEDULER_NO_TASK == n->retry_set_key_task)
2369             n->retry_set_key_task
2370               = GNUNET_SCHEDULER_add_delayed (sched,
2371                                               n->set_key_retry_frequency,
2372                                               &set_key_retry_task, n);
2373         }
2374       return;
2375     }
2376
2377 #if DEBUG_CORE
2378   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2379               "Entered `process_hello_retry_send_key' for peer `%4s'\n",
2380               GNUNET_i2s (peer));
2381 #endif
2382   if (n->public_key != NULL)
2383     {
2384 #if DEBUG_CORE
2385       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2386               "already have public key for peer %s!! (so why are we here?)\n",
2387               GNUNET_i2s (peer));
2388 #endif
2389       return;
2390     }
2391
2392 #if DEBUG_CORE
2393   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2394               "Received new `%s' message for `%4s', initiating key exchange.\n",
2395               "HELLO",
2396               GNUNET_i2s (peer));
2397 #endif
2398   n->public_key =
2399     GNUNET_malloc (sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
2400   if (GNUNET_OK != GNUNET_HELLO_get_key (hello, n->public_key))
2401     {
2402       GNUNET_STATISTICS_update (stats,
2403                                 gettext_noop ("# Error extracting public key from HELLO"),
2404                                 1,
2405                                 GNUNET_NO);
2406       GNUNET_free (n->public_key);
2407       n->public_key = NULL;
2408 #if DEBUG_CORE
2409   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2410               "GNUNET_HELLO_get_key returned awfully\n");
2411 #endif
2412       return;
2413     }
2414 }
2415
2416
2417 /**
2418  * Send our key (and encrypted PING) to the other peer.
2419  *
2420  * @param n the other peer
2421  */
2422 static void
2423 send_key (struct Neighbour *n)
2424 {
2425   struct SetKeyMessage *sm;
2426   struct MessageEntry *me;
2427   struct PingMessage pp;
2428   struct PingMessage *pm;
2429
2430   if ( (n->retry_set_key_task != GNUNET_SCHEDULER_NO_TASK) ||
2431        (n->pitr != NULL) )
2432     {
2433 #if DEBUG_CORE
2434       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2435                   "Key exchange in progress with `%4s'.\n",
2436                   GNUNET_i2s (&n->peer));
2437 #endif
2438       return; /* already in progress */
2439     }
2440   if (! n->is_connected)
2441     {
2442       if (NULL == n->th)
2443         {
2444           GNUNET_STATISTICS_update (stats,
2445                                     gettext_noop ("# Asking transport to connect (for SETKEY)"),
2446                                     1,
2447                                     GNUNET_NO);
2448           n->th = GNUNET_TRANSPORT_notify_transmit_ready (transport,
2449                                                           &n->peer,
2450                                                           sizeof (struct SetKeyMessage) + sizeof (struct PingMessage),
2451                                                           0,
2452                                                           GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT,
2453                                                           &notify_encrypted_transmit_ready,
2454                                                           n);
2455         }
2456       return;
2457     }
2458 #if DEBUG_CORE
2459   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2460               "Asked to perform key exchange with `%4s'.\n",
2461               GNUNET_i2s (&n->peer));
2462 #endif
2463   if (n->public_key == NULL)
2464     {
2465       /* lookup n's public key, then try again */
2466 #if DEBUG_CORE
2467       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2468                   "Lacking public key for `%4s', trying to obtain one (send_key).\n",
2469                   GNUNET_i2s (&n->peer));
2470 #endif
2471       GNUNET_assert (n->pitr == NULL);
2472       n->pitr = GNUNET_PEERINFO_iterate (cfg,
2473                                          sched,
2474                                          &n->peer,
2475                                          0,
2476                                          GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 20),
2477                                          &process_hello_retry_send_key, n);
2478       return;
2479     }
2480   /* first, set key message */
2481   me = GNUNET_malloc (sizeof (struct MessageEntry) +
2482                       sizeof (struct SetKeyMessage));
2483   me->deadline = GNUNET_TIME_relative_to_absolute (MAX_SET_KEY_DELAY);
2484   me->priority = SET_KEY_PRIORITY;
2485   me->size = sizeof (struct SetKeyMessage);
2486   GNUNET_CONTAINER_DLL_insert_after (n->encrypted_head,
2487                                      n->encrypted_tail,
2488                                      n->encrypted_tail,
2489                                      me);
2490   sm = (struct SetKeyMessage *) &me[1];
2491   sm->header.size = htons (sizeof (struct SetKeyMessage));
2492   sm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_SET_KEY);
2493   sm->sender_status = htonl ((int32_t) ((n->status == PEER_STATE_DOWN) ?
2494                                         PEER_STATE_KEY_SENT : n->status));
2495   sm->purpose.size =
2496     htonl (sizeof (struct GNUNET_CRYPTO_RsaSignaturePurpose) +
2497            sizeof (struct GNUNET_TIME_AbsoluteNBO) +
2498            sizeof (struct GNUNET_CRYPTO_RsaEncryptedData) +
2499            sizeof (struct GNUNET_PeerIdentity));
2500   sm->purpose.purpose = htonl (GNUNET_SIGNATURE_PURPOSE_SET_KEY);
2501   sm->creation_time = GNUNET_TIME_absolute_hton (n->encrypt_key_created);
2502   sm->target = n->peer;
2503   GNUNET_assert (GNUNET_OK ==
2504                  GNUNET_CRYPTO_rsa_encrypt (&n->encrypt_key,
2505                                             sizeof (struct
2506                                                     GNUNET_CRYPTO_AesSessionKey),
2507                                             n->public_key,
2508                                             &sm->encrypted_key));
2509   GNUNET_assert (GNUNET_OK ==
2510                  GNUNET_CRYPTO_rsa_sign (my_private_key, &sm->purpose,
2511                                          &sm->signature));
2512
2513   /* second, encrypted PING message */
2514   me = GNUNET_malloc (sizeof (struct MessageEntry) +
2515                       sizeof (struct PingMessage));
2516   me->deadline = GNUNET_TIME_relative_to_absolute (MAX_PING_DELAY);
2517   me->priority = PING_PRIORITY;
2518   me->size = sizeof (struct PingMessage);
2519   GNUNET_CONTAINER_DLL_insert_after (n->encrypted_head,
2520                                      n->encrypted_tail,
2521                                      n->encrypted_tail,
2522                                      me);
2523   pm = (struct PingMessage *) &me[1];
2524   pm->header.size = htons (sizeof (struct PingMessage));
2525   pm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_PING);
2526   pp.challenge = htonl (n->ping_challenge);
2527   pp.target = n->peer;
2528 #if DEBUG_HANDSHAKE
2529   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2530               "Encrypting `%s' and `%s' messages with challenge %u for `%4s' using key %u.\n",
2531               "SET_KEY", "PING",
2532               (unsigned int) n->ping_challenge,
2533               GNUNET_i2s (&n->peer),
2534               (unsigned int) n->encrypt_key.crc32);
2535 #endif
2536   do_encrypt (n,
2537               &n->peer.hashPubKey,
2538               &pp.challenge,
2539               &pm->challenge,
2540               sizeof (struct PingMessage) -
2541               sizeof (struct GNUNET_MessageHeader));
2542   /* update status */
2543   switch (n->status)
2544     {
2545     case PEER_STATE_DOWN:
2546       n->status = PEER_STATE_KEY_SENT;
2547       break;
2548     case PEER_STATE_KEY_SENT:
2549       break;
2550     case PEER_STATE_KEY_RECEIVED:
2551       break;
2552     case PEER_STATE_KEY_CONFIRMED:
2553       break;
2554     default:
2555       GNUNET_break (0);
2556       break;
2557     }
2558   GNUNET_STATISTICS_update (stats,
2559                             gettext_noop ("# SETKEY and PING messages created"),
2560                             1,
2561                             GNUNET_NO);
2562 #if DEBUG_CORE
2563   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2564               "Have %llu ms left for `%s' transmission.\n",
2565               (unsigned long long) GNUNET_TIME_absolute_get_remaining (me->deadline).value,
2566               "SET_KEY");
2567 #endif
2568   /* trigger queue processing */
2569   process_encrypted_neighbour_queue (n);
2570   if ( (n->status != PEER_STATE_KEY_CONFIRMED) &&
2571        (GNUNET_SCHEDULER_NO_TASK == n->retry_set_key_task) )
2572     n->retry_set_key_task
2573       = GNUNET_SCHEDULER_add_delayed (sched,
2574                                       n->set_key_retry_frequency,
2575                                       &set_key_retry_task, n);
2576 }
2577
2578
2579 /**
2580  * We received a SET_KEY message.  Validate and update
2581  * our key material and status.
2582  *
2583  * @param n the neighbour from which we received message m
2584  * @param m the set key message we received
2585  */
2586 static void
2587 handle_set_key (struct Neighbour *n, const struct SetKeyMessage *m);
2588
2589
2590 /**
2591  * PEERINFO is giving us a HELLO for a peer.  Add the public key to
2592  * the neighbour's struct and retry handling the set_key message.  Or,
2593  * if we did not get a HELLO, just free the set key message.
2594  *
2595  * @param cls pointer to the set key message
2596  * @param peer the peer for which this is the HELLO
2597  * @param hello HELLO message of that peer
2598  * @param trust amount of trust we currently have in that peer
2599  */
2600 static void
2601 process_hello_retry_handle_set_key (void *cls,
2602                                     const struct GNUNET_PeerIdentity *peer,
2603                                     const struct GNUNET_HELLO_Message *hello,
2604                                     uint32_t trust)
2605 {
2606   struct Neighbour *n = cls;
2607   struct SetKeyMessage *sm = n->skm;
2608
2609   if (peer == NULL)
2610     {
2611       GNUNET_free (sm);
2612       n->skm = NULL;
2613       n->pitr = NULL;
2614       return;
2615     }
2616   if (n->public_key != NULL)
2617     return;                     /* multiple HELLOs match!? */
2618   n->public_key =
2619     GNUNET_malloc (sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
2620   if (GNUNET_OK != GNUNET_HELLO_get_key (hello, n->public_key))
2621     {
2622       GNUNET_break_op (0);
2623       GNUNET_free (n->public_key);
2624       n->public_key = NULL;
2625       return;
2626     }
2627 #if DEBUG_CORE
2628   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2629               "Received `%s' for `%4s', continuing processing of `%s' message.\n",
2630               "HELLO", GNUNET_i2s (peer), "SET_KEY");
2631 #endif
2632   handle_set_key (n, sm);
2633 }
2634
2635
2636 /**
2637  * We received a PING message.  Validate and transmit
2638  * PONG.
2639  *
2640  * @param n sender of the PING
2641  * @param m the encrypted PING message itself
2642  */
2643 static void
2644 handle_ping (struct Neighbour *n, const struct PingMessage *m)
2645 {
2646   struct PingMessage t;
2647   struct PongMessage tx;
2648   struct PongMessage *tp;
2649   struct MessageEntry *me;
2650
2651 #if DEBUG_CORE
2652   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2653               "Core service receives `%s' request from `%4s'.\n",
2654               "PING", GNUNET_i2s (&n->peer));
2655 #endif
2656   if (GNUNET_OK !=
2657       do_decrypt (n,
2658                   &my_identity.hashPubKey,
2659                   &m->challenge,
2660                   &t.challenge,
2661                   sizeof (struct PingMessage) -
2662                   sizeof (struct GNUNET_MessageHeader)))
2663     return;
2664 #if DEBUG_HANDSHAKE
2665   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2666               "Decrypted `%s' to `%4s' with challenge %u decrypted using key %u\n",
2667               "PING",
2668               GNUNET_i2s (&t.target),
2669               (unsigned int) ntohl (t.challenge),
2670               (unsigned int) n->decrypt_key.crc32);
2671 #endif
2672   GNUNET_STATISTICS_update (stats,
2673                             gettext_noop ("# PING messages decrypted"),
2674                             1,
2675                             GNUNET_NO);
2676   if (0 != memcmp (&t.target,
2677                    &my_identity, sizeof (struct GNUNET_PeerIdentity)))
2678     {
2679       GNUNET_break_op (0);
2680       return;
2681     }
2682   me = GNUNET_malloc (sizeof (struct MessageEntry) +
2683                       sizeof (struct PongMessage));
2684   GNUNET_CONTAINER_DLL_insert_after (n->encrypted_head,
2685                                      n->encrypted_tail,
2686                                      n->encrypted_tail,
2687                                      me);
2688   me->deadline = GNUNET_TIME_relative_to_absolute (MAX_PONG_DELAY);
2689   me->priority = PONG_PRIORITY;
2690   me->size = sizeof (struct PongMessage);
2691   tx.reserved = htonl (0);
2692   tx.inbound_bw_limit = n->bw_in;
2693   tx.challenge = t.challenge;
2694   tx.target = t.target;
2695   tp = (struct PongMessage *) &me[1];
2696   tp->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_PONG);
2697   tp->header.size = htons (sizeof (struct PongMessage));
2698   do_encrypt (n,
2699               &my_identity.hashPubKey,
2700               &tx.challenge,
2701               &tp->challenge,
2702               sizeof (struct PongMessage) -
2703               sizeof (struct GNUNET_MessageHeader));
2704   GNUNET_STATISTICS_update (stats,
2705                             gettext_noop ("# PONG messages created"),
2706                             1,
2707                             GNUNET_NO);
2708 #if DEBUG_HANDSHAKE
2709   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2710               "Encrypting `%s' with challenge %u using key %u\n",
2711               "PONG",
2712               (unsigned int) ntohl (t.challenge),
2713               (unsigned int) n->encrypt_key.crc32);
2714 #endif
2715   /* trigger queue processing */
2716   process_encrypted_neighbour_queue (n);
2717 }
2718
2719
2720 /**
2721  * We received a PONG message.  Validate and update our status.
2722  *
2723  * @param n sender of the PONG
2724  * @param m the encrypted PONG message itself
2725  */
2726 static void
2727 handle_pong (struct Neighbour *n,
2728              const struct PongMessage *m)
2729 {
2730   struct PongMessage t;
2731   struct ConnectNotifyMessage cnm;
2732
2733 #if DEBUG_CORE
2734   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2735               "Core service receives `%s' request from `%4s'.\n",
2736               "PONG", GNUNET_i2s (&n->peer));
2737 #endif
2738   if (GNUNET_OK !=
2739       do_decrypt (n,
2740                   &n->peer.hashPubKey,
2741                   &m->challenge,
2742                   &t.challenge,
2743                   sizeof (struct PongMessage) -
2744                   sizeof (struct GNUNET_MessageHeader)))
2745     {
2746       GNUNET_break_op (0);
2747       return;
2748     }
2749   GNUNET_STATISTICS_update (stats,
2750                             gettext_noop ("# PONG messages decrypted"),
2751                             1,
2752                             GNUNET_NO);
2753   if (0 != ntohl (t.reserved))
2754     {
2755       GNUNET_break_op (0);
2756       return;
2757     }
2758 #if DEBUG_HANDSHAKE
2759   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2760               "Decrypted `%s' from `%4s' with challenge %u using key %u\n",
2761               "PONG",
2762               GNUNET_i2s (&t.target),
2763               (unsigned int) ntohl (t.challenge),
2764               (unsigned int) n->decrypt_key.crc32);
2765 #endif
2766   if ((0 != memcmp (&t.target,
2767                     &n->peer,
2768                     sizeof (struct GNUNET_PeerIdentity))) ||
2769       (n->ping_challenge != ntohl (t.challenge)))
2770     {
2771       /* PONG malformed */
2772 #if DEBUG_CORE
2773       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2774                   "Received malformed `%s' wanted sender `%4s' with challenge %u\n",
2775                   "PONG",
2776                   GNUNET_i2s (&n->peer),
2777                   (unsigned int) n->ping_challenge);
2778       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2779                   "Received malformed `%s' received from `%4s' with challenge %u\n",
2780                   "PONG", GNUNET_i2s (&t.target),
2781                   (unsigned int) ntohl (t.challenge));
2782 #endif
2783       GNUNET_break_op (0);
2784       return;
2785     }
2786   switch (n->status)
2787     {
2788     case PEER_STATE_DOWN:
2789       GNUNET_break (0);         /* should be impossible */
2790       return;
2791     case PEER_STATE_KEY_SENT:
2792       GNUNET_break (0);         /* should be impossible, how did we decrypt? */
2793       return;
2794     case PEER_STATE_KEY_RECEIVED:
2795       GNUNET_STATISTICS_update (stats,
2796                                 gettext_noop ("# Session keys confirmed via PONG"),
2797                                 1,
2798                                 GNUNET_NO);
2799       n->status = PEER_STATE_KEY_CONFIRMED;
2800       if (n->bw_out_external_limit.value__ != t.inbound_bw_limit.value__)
2801         {
2802           n->bw_out_external_limit = t.inbound_bw_limit;
2803           n->bw_out = GNUNET_BANDWIDTH_value_min (n->bw_out_external_limit,
2804                                                   n->bw_out_internal_limit);
2805           GNUNET_BANDWIDTH_tracker_update_quota (&n->available_send_window,
2806                                                  n->bw_out);
2807           GNUNET_TRANSPORT_set_quota (transport,
2808                                       &n->peer,
2809                                       n->bw_in,
2810                                       n->bw_out,
2811                                       GNUNET_TIME_UNIT_FOREVER_REL,
2812                                       NULL, NULL);
2813         }
2814 #if DEBUG_CORE
2815       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2816                   "Confirmed key via `%s' message for peer `%4s'\n",
2817                   "PONG", GNUNET_i2s (&n->peer));
2818 #endif
2819       if (n->retry_set_key_task != GNUNET_SCHEDULER_NO_TASK)
2820         {
2821           GNUNET_SCHEDULER_cancel (sched, n->retry_set_key_task);
2822           n->retry_set_key_task = GNUNET_SCHEDULER_NO_TASK;
2823         }
2824       cnm.header.size = htons (sizeof (struct ConnectNotifyMessage));
2825       cnm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_CONNECT);
2826       cnm.distance = htonl (n->last_distance);
2827       cnm.latency = GNUNET_TIME_relative_hton (n->last_latency);
2828       cnm.peer = n->peer;
2829       send_to_all_clients (&cnm.header, GNUNET_YES, GNUNET_CORE_OPTION_SEND_CONNECT);
2830       process_encrypted_neighbour_queue (n);
2831       /* fall-through! */
2832     case PEER_STATE_KEY_CONFIRMED:
2833       n->last_activity = GNUNET_TIME_absolute_get ();
2834       if (n->keep_alive_task != GNUNET_SCHEDULER_NO_TASK)
2835         GNUNET_SCHEDULER_cancel (sched, n->keep_alive_task);
2836       n->keep_alive_task
2837         = GNUNET_SCHEDULER_add_delayed (sched,
2838                                         GNUNET_TIME_relative_divide (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, 2),
2839                                         &send_keep_alive,
2840                                         n);
2841       break;
2842     default:
2843       GNUNET_break (0);
2844       break;
2845     }
2846 }
2847
2848
2849 /**
2850  * We received a SET_KEY message.  Validate and update
2851  * our key material and status.
2852  *
2853  * @param n the neighbour from which we received message m
2854  * @param m the set key message we received
2855  */
2856 static void
2857 handle_set_key (struct Neighbour *n, const struct SetKeyMessage *m)
2858 {
2859   struct SetKeyMessage *m_cpy;
2860   struct GNUNET_TIME_Absolute t;
2861   struct GNUNET_CRYPTO_AesSessionKey k;
2862   struct PingMessage *ping;
2863   struct PongMessage *pong;
2864   enum PeerStateMachine sender_status;
2865
2866 #if DEBUG_CORE
2867   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2868               "Core service receives `%s' request from `%4s'.\n",
2869               "SET_KEY", GNUNET_i2s (&n->peer));
2870 #endif
2871   if (n->public_key == NULL)
2872     {
2873       if (n->pitr != NULL)
2874         {
2875 #if DEBUG_CORE
2876           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2877                       "Ignoring `%s' message due to lack of public key for peer (still trying to obtain one).\n",
2878                       "SET_KEY");
2879 #endif
2880           return;
2881         }
2882 #if DEBUG_CORE
2883       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2884                   "Lacking public key for peer, trying to obtain one (handle_set_key).\n");
2885 #endif
2886       m_cpy = GNUNET_malloc (sizeof (struct SetKeyMessage));
2887       memcpy (m_cpy, m, sizeof (struct SetKeyMessage));
2888       /* lookup n's public key, then try again */
2889       GNUNET_assert (n->skm == NULL);
2890       n->skm = m_cpy;
2891       n->pitr = GNUNET_PEERINFO_iterate (cfg,
2892                                          sched,
2893                                          &n->peer,
2894                                          0,
2895                                          GNUNET_TIME_UNIT_MINUTES,
2896                                          &process_hello_retry_handle_set_key, n);
2897       GNUNET_STATISTICS_update (stats,
2898                                 gettext_noop ("# SETKEY messages deferred (need public key)"),
2899                                 1,
2900                                 GNUNET_NO);
2901       return;
2902     }
2903   if (0 != memcmp (&m->target,
2904                    &my_identity,
2905                    sizeof (struct GNUNET_PeerIdentity)))
2906     {
2907       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
2908                   _("Received `%s' message that was for `%s', not for me.  Ignoring.\n"),
2909                   "SET_KEY",
2910                   GNUNET_i2s (&m->target));
2911       return;
2912     }
2913   if ((ntohl (m->purpose.size) !=
2914        sizeof (struct GNUNET_CRYPTO_RsaSignaturePurpose) +
2915        sizeof (struct GNUNET_TIME_AbsoluteNBO) +
2916        sizeof (struct GNUNET_CRYPTO_RsaEncryptedData) +
2917        sizeof (struct GNUNET_PeerIdentity)) ||
2918       (GNUNET_OK !=
2919        GNUNET_CRYPTO_rsa_verify (GNUNET_SIGNATURE_PURPOSE_SET_KEY,
2920                                  &m->purpose, &m->signature, n->public_key)))
2921     {
2922       /* invalid signature */
2923       GNUNET_break_op (0);
2924       return;
2925     }
2926   t = GNUNET_TIME_absolute_ntoh (m->creation_time);
2927   if (((n->status == PEER_STATE_KEY_RECEIVED) ||
2928        (n->status == PEER_STATE_KEY_CONFIRMED)) &&
2929       (t.value < n->decrypt_key_created.value))
2930     {
2931       /* this could rarely happen due to massive re-ordering of
2932          messages on the network level, but is most likely either
2933          a bug or some adversary messing with us.  Report. */
2934       GNUNET_break_op (0);
2935       return;
2936     }
2937 #if DEBUG_CORE
2938   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2939               "Decrypting key material.\n");
2940 #endif
2941   if ((GNUNET_CRYPTO_rsa_decrypt (my_private_key,
2942                                   &m->encrypted_key,
2943                                   &k,
2944                                   sizeof (struct GNUNET_CRYPTO_AesSessionKey))
2945        != sizeof (struct GNUNET_CRYPTO_AesSessionKey)) ||
2946       (GNUNET_OK != GNUNET_CRYPTO_aes_check_session_key (&k)))
2947     {
2948       /* failed to decrypt !? */
2949       GNUNET_break_op (0);
2950       return;
2951     }
2952   GNUNET_STATISTICS_update (stats,
2953                             gettext_noop ("# SETKEY messages decrypted"),
2954                             1,
2955                             GNUNET_NO);
2956   n->decrypt_key = k;
2957   if (n->decrypt_key_created.value != t.value)
2958     {
2959       /* fresh key, reset sequence numbers */
2960       n->last_sequence_number_received = 0;
2961       n->last_packets_bitmap = 0;
2962       n->decrypt_key_created = t;
2963     }
2964   sender_status = (enum PeerStateMachine) ntohl (m->sender_status);
2965   switch (n->status)
2966     {
2967     case PEER_STATE_DOWN:
2968       n->status = PEER_STATE_KEY_RECEIVED;
2969 #if DEBUG_CORE
2970       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2971                   "Responding to `%s' with my own key.\n", "SET_KEY");
2972 #endif
2973       send_key (n);
2974       break;
2975     case PEER_STATE_KEY_SENT:
2976     case PEER_STATE_KEY_RECEIVED:
2977       n->status = PEER_STATE_KEY_RECEIVED;
2978       if ((sender_status != PEER_STATE_KEY_RECEIVED) &&
2979           (sender_status != PEER_STATE_KEY_CONFIRMED))
2980         {
2981 #if DEBUG_CORE
2982           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2983                       "Responding to `%s' with my own key (other peer has status %u).\n",
2984                       "SET_KEY",
2985                       (unsigned int) sender_status);
2986 #endif
2987           send_key (n);
2988         }
2989       break;
2990     case PEER_STATE_KEY_CONFIRMED:
2991       if ((sender_status != PEER_STATE_KEY_RECEIVED) &&
2992           (sender_status != PEER_STATE_KEY_CONFIRMED))
2993         {
2994 #if DEBUG_CORE
2995           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2996                       "Responding to `%s' with my own key (other peer has status %u), I was already fully up.\n",
2997                       "SET_KEY",
2998                       (unsigned int) sender_status);
2999 #endif
3000           send_key (n);
3001         }
3002       break;
3003     default:
3004       GNUNET_break (0);
3005       break;
3006     }
3007   if (n->pending_ping != NULL)
3008     {
3009       ping = n->pending_ping;
3010       n->pending_ping = NULL;
3011       handle_ping (n, ping);
3012       GNUNET_free (ping);
3013     }
3014   if (n->pending_pong != NULL)
3015     {
3016       pong = n->pending_pong;
3017       n->pending_pong = NULL;
3018       handle_pong (n, pong);
3019       GNUNET_free (pong);
3020     }
3021 }
3022
3023
3024 /**
3025  * Send a P2P message to a client.
3026  *
3027  * @param sender who sent us the message?
3028  * @param client who should we give the message to?
3029  * @param m contains the message to transmit
3030  * @param msize number of bytes in buf to transmit
3031  */
3032 static void
3033 send_p2p_message_to_client (struct Neighbour *sender,
3034                             struct Client *client,
3035                             const void *m, size_t msize)
3036 {
3037   char buf[msize + sizeof (struct NotifyTrafficMessage)];
3038   struct NotifyTrafficMessage *ntm;
3039
3040 #if DEBUG_CORE
3041   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3042               "Core service passes message from `%4s' of type %u to client.\n",
3043               GNUNET_i2s(&sender->peer),
3044               (unsigned int) ntohs (((const struct GNUNET_MessageHeader *) m)->type));
3045 #endif
3046   ntm = (struct NotifyTrafficMessage *) buf;
3047   ntm->header.size = htons (msize + sizeof (struct NotifyTrafficMessage));
3048   ntm->header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_INBOUND);
3049   ntm->distance = htonl (sender->last_distance);
3050   ntm->latency = GNUNET_TIME_relative_hton (sender->last_latency);
3051   ntm->peer = sender->peer;
3052   memcpy (&ntm[1], m, msize);
3053   send_to_client (client, &ntm->header, GNUNET_YES);
3054 }
3055
3056
3057 /**
3058  * Deliver P2P message to interested clients.
3059  *
3060  * @param sender who sent us the message?
3061  * @param m the message
3062  * @param msize size of the message (including header)
3063  */
3064 static void
3065 deliver_message (struct Neighbour *sender,
3066                  const struct GNUNET_MessageHeader *m, size_t msize)
3067 {
3068   char buf[256];
3069   struct Client *cpos;
3070   uint16_t type;
3071   unsigned int tpos;
3072   int deliver_full;
3073   int dropped;
3074
3075   type = ntohs (m->type);
3076 #if DEBUG_CORE
3077   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3078               "Received encapsulated message of type %u from `%4s'\n",
3079               (unsigned int) type,
3080               GNUNET_i2s (&sender->peer));
3081 #endif
3082   GNUNET_snprintf (buf,
3083                    sizeof(buf),
3084                    gettext_noop ("# bytes of messages of type %u received"),
3085                    (unsigned int) type);
3086   GNUNET_STATISTICS_set (stats,
3087                          buf,
3088                          msize,
3089                          GNUNET_NO);
3090   dropped = GNUNET_YES;
3091   cpos = clients;
3092   while (cpos != NULL)
3093     {
3094       deliver_full = GNUNET_NO;
3095       if (0 != (cpos->options & GNUNET_CORE_OPTION_SEND_FULL_INBOUND))
3096         deliver_full = GNUNET_YES;
3097       else
3098         {
3099           for (tpos = 0; tpos < cpos->tcnt; tpos++)
3100             {
3101               if (type != cpos->types[tpos])
3102                 continue;
3103               deliver_full = GNUNET_YES;
3104               break;
3105             }
3106         }
3107       if (GNUNET_YES == deliver_full)
3108         {
3109           send_p2p_message_to_client (sender, cpos, m, msize);
3110           dropped = GNUNET_NO;
3111         }
3112       else if (cpos->options & GNUNET_CORE_OPTION_SEND_HDR_INBOUND)
3113         {
3114           send_p2p_message_to_client (sender, cpos, m,
3115                                       sizeof (struct GNUNET_MessageHeader));
3116         }
3117       cpos = cpos->next;
3118     }
3119   if (dropped == GNUNET_YES)
3120     {
3121 #if DEBUG_CORE
3122       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3123                   "Message of type %u from `%4s' not delivered to any client.\n",
3124                   (unsigned int) type,
3125                   GNUNET_i2s (&sender->peer));
3126 #endif
3127       /* FIXME: stats... */
3128     }
3129 }
3130
3131
3132 /**
3133  * Align P2P message and then deliver to interested clients.
3134  *
3135  * @param sender who sent us the message?
3136  * @param buffer unaligned (!) buffer containing message
3137  * @param msize size of the message (including header)
3138  */
3139 static void
3140 align_and_deliver (struct Neighbour *sender, const char *buffer, size_t msize)
3141 {
3142   char abuf[msize];
3143
3144   /* TODO: call to statistics? */
3145   memcpy (abuf, buffer, msize);
3146   deliver_message (sender, (const struct GNUNET_MessageHeader *) abuf, msize);
3147 }
3148
3149
3150 /**
3151  * Deliver P2P messages to interested clients.
3152  *
3153  * @param sender who sent us the message?
3154  * @param buffer buffer containing messages, can be modified
3155  * @param buffer_size size of the buffer (overall)
3156  * @param offset offset where messages in the buffer start
3157  */
3158 static void
3159 deliver_messages (struct Neighbour *sender,
3160                   const char *buffer, size_t buffer_size, size_t offset)
3161 {
3162   struct GNUNET_MessageHeader *mhp;
3163   struct GNUNET_MessageHeader mh;
3164   uint16_t msize;
3165   int need_align;
3166
3167   while (offset + sizeof (struct GNUNET_MessageHeader) <= buffer_size)
3168     {
3169       if (0 != offset % sizeof (uint16_t))
3170         {
3171           /* outch, need to copy to access header */
3172           memcpy (&mh, &buffer[offset], sizeof (struct GNUNET_MessageHeader));
3173           mhp = &mh;
3174         }
3175       else
3176         {
3177           /* can access header directly */
3178           mhp = (struct GNUNET_MessageHeader *) &buffer[offset];
3179         }
3180       msize = ntohs (mhp->size);
3181       if (msize + offset > buffer_size)
3182         {
3183           /* malformed message, header says it is larger than what
3184              would fit into the overall buffer */
3185           GNUNET_break_op (0);
3186           break;
3187         }
3188 #if HAVE_UNALIGNED_64_ACCESS
3189       need_align = (0 != offset % 4) ? GNUNET_YES : GNUNET_NO;
3190 #else
3191       need_align = (0 != offset % 8) ? GNUNET_YES : GNUNET_NO;
3192 #endif
3193       if (GNUNET_YES == need_align)
3194         align_and_deliver (sender, &buffer[offset], msize);
3195       else
3196         deliver_message (sender,
3197                          (const struct GNUNET_MessageHeader *)
3198                          &buffer[offset], msize);
3199       offset += msize;
3200     }
3201 }
3202
3203
3204 /**
3205  * We received an encrypted message.  Decrypt, validate and
3206  * pass on to the appropriate clients.
3207  */
3208 static void
3209 handle_encrypted_message (struct Neighbour *n,
3210                           const struct EncryptedMessage *m)
3211 {
3212   size_t size = ntohs (m->header.size);
3213   char buf[size];
3214   struct EncryptedMessage *pt;  /* plaintext */
3215   GNUNET_HashCode ph;
3216   uint32_t snum;
3217   struct GNUNET_TIME_Absolute t;
3218   GNUNET_HashCode iv;
3219
3220 #if DEBUG_CORE
3221   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3222               "Core service receives `%s' request from `%4s'.\n",
3223               "ENCRYPTED_MESSAGE", GNUNET_i2s (&n->peer));
3224 #endif
3225   GNUNET_CRYPTO_hash (&m->iv_seed, sizeof (uint32_t), &iv);
3226   /* decrypt */
3227   if (GNUNET_OK !=
3228       do_decrypt (n,
3229                   &iv,
3230                   &m->plaintext_hash,
3231                   &buf[ENCRYPTED_HEADER_SIZE],
3232                   size - ENCRYPTED_HEADER_SIZE))
3233     return;
3234   pt = (struct EncryptedMessage *) buf;
3235   /* validate hash */
3236   GNUNET_CRYPTO_hash (&pt->sequence_number,
3237                       size - ENCRYPTED_HEADER_SIZE - sizeof (GNUNET_HashCode), &ph);
3238 #if DEBUG_HANDSHAKE
3239   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3240               "V-Hashed %u bytes of plaintext (`%s') using IV `%d'\n",
3241               (unsigned int) (size - ENCRYPTED_HEADER_SIZE - sizeof (GNUNET_HashCode)),
3242               GNUNET_h2s (&ph),
3243               (int) m->iv_seed);
3244 #endif
3245   if (0 != memcmp (&ph,
3246                    &pt->plaintext_hash,
3247                    sizeof (GNUNET_HashCode)))
3248     {
3249       /* checksum failed */
3250       GNUNET_break_op (0);
3251       return;
3252     }
3253
3254   /* validate sequence number */
3255   snum = ntohl (pt->sequence_number);
3256   if (n->last_sequence_number_received == snum)
3257     {
3258       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3259                   "Received duplicate message, ignoring.\n");
3260       /* duplicate, ignore */
3261       GNUNET_STATISTICS_set (stats,
3262                              gettext_noop ("# bytes dropped (duplicates)"),
3263                              size,
3264                              GNUNET_NO);
3265       return;
3266     }
3267   if ((n->last_sequence_number_received > snum) &&
3268       (n->last_sequence_number_received - snum > 32))
3269     {
3270       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3271                   "Received ancient out of sequence message, ignoring.\n");
3272       /* ancient out of sequence, ignore */
3273       GNUNET_STATISTICS_set (stats,
3274                              gettext_noop ("# bytes dropped (out of sequence)"),
3275                              size,
3276                              GNUNET_NO);
3277       return;
3278     }
3279   if (n->last_sequence_number_received > snum)
3280     {
3281       unsigned int rotbit =
3282         1 << (n->last_sequence_number_received - snum - 1);
3283       if ((n->last_packets_bitmap & rotbit) != 0)
3284         {
3285           GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3286                       "Received duplicate message, ignoring.\n");
3287           GNUNET_STATISTICS_set (stats,
3288                                  gettext_noop ("# bytes dropped (duplicates)"),
3289                                  size,
3290                                  GNUNET_NO);
3291           /* duplicate, ignore */
3292           return;
3293         }
3294       n->last_packets_bitmap |= rotbit;
3295     }
3296   if (n->last_sequence_number_received < snum)
3297     {
3298       n->last_packets_bitmap <<= (snum - n->last_sequence_number_received);
3299       n->last_sequence_number_received = snum;
3300     }
3301
3302   /* check timestamp */
3303   t = GNUNET_TIME_absolute_ntoh (pt->timestamp);
3304   if (GNUNET_TIME_absolute_get_duration (t).value > MAX_MESSAGE_AGE.value)
3305     {
3306       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3307                   _
3308                   ("Message received far too old (%llu ms). Content ignored.\n"),
3309                   GNUNET_TIME_absolute_get_duration (t).value);
3310       GNUNET_STATISTICS_set (stats,
3311                              gettext_noop ("# bytes dropped (ancient message)"),
3312                              size,
3313                              GNUNET_NO);
3314       return;
3315     }
3316
3317   /* process decrypted message(s) */
3318   if (n->bw_out_external_limit.value__ != pt->inbound_bw_limit.value__)
3319     {
3320 #if DEBUG_CORE_SET_QUOTA
3321       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3322                   "Received %u b/s as new inbound limit for peer `%4s'\n",
3323                   (unsigned int) ntohl (pt->inbound_bw_limit.value__),
3324                   GNUNET_i2s (&n->peer));
3325 #endif
3326       n->bw_out_external_limit = pt->inbound_bw_limit;
3327       n->bw_out = GNUNET_BANDWIDTH_value_min (n->bw_out_external_limit,
3328                                               n->bw_out_internal_limit);
3329       GNUNET_BANDWIDTH_tracker_update_quota (&n->available_send_window,
3330                                              n->bw_out);
3331       GNUNET_TRANSPORT_set_quota (transport,
3332                                   &n->peer,
3333                                   n->bw_in,
3334                                   n->bw_out,
3335                                   GNUNET_TIME_UNIT_FOREVER_REL,
3336                                   NULL, NULL);
3337     }
3338   n->last_activity = GNUNET_TIME_absolute_get ();
3339   if (n->keep_alive_task != GNUNET_SCHEDULER_NO_TASK)
3340     GNUNET_SCHEDULER_cancel (sched, n->keep_alive_task);
3341   n->keep_alive_task
3342     = GNUNET_SCHEDULER_add_delayed (sched,
3343                                     GNUNET_TIME_relative_divide (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, 2),
3344                                     &send_keep_alive,
3345                                     n);
3346   GNUNET_STATISTICS_set (stats,
3347                          gettext_noop ("# bytes of payload decrypted"),
3348                          size - sizeof (struct EncryptedMessage),
3349                          GNUNET_NO);
3350   deliver_messages (n, buf, size, sizeof (struct EncryptedMessage));
3351 }
3352
3353
3354 /**
3355  * Function called by the transport for each received message.
3356  *
3357  * @param cls closure
3358  * @param peer (claimed) identity of the other peer
3359  * @param message the message
3360  * @param latency estimated latency for communicating with the
3361  *             given peer (round-trip)
3362  * @param distance in overlay hops, as given by transport plugin
3363  */
3364 static void
3365 handle_transport_receive (void *cls,
3366                           const struct GNUNET_PeerIdentity *peer,
3367                           const struct GNUNET_MessageHeader *message,
3368                           struct GNUNET_TIME_Relative latency,
3369                           unsigned int distance)
3370 {
3371   struct Neighbour *n;
3372   struct GNUNET_TIME_Absolute now;
3373   int up;
3374   uint16_t type;
3375   uint16_t size;
3376
3377 #if DEBUG_CORE
3378   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3379               "Received message of type %u from `%4s', demultiplexing.\n",
3380               (unsigned int) ntohs (message->type),
3381               GNUNET_i2s (peer));
3382 #endif
3383   if (0 == memcmp (peer, &my_identity, sizeof (struct GNUNET_PeerIdentity)))
3384     {
3385       GNUNET_break (0);
3386       return;
3387     }
3388   n = find_neighbour (peer);
3389   if (n == NULL)
3390     n = create_neighbour (peer);
3391   n->last_latency = latency;
3392   n->last_distance = distance;
3393   up = (n->status == PEER_STATE_KEY_CONFIRMED);
3394   type = ntohs (message->type);
3395   size = ntohs (message->size);
3396 #if DEBUG_HANDSHAKE
3397   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3398               "Received message of type %u from `%4s'\n",
3399               (unsigned int) type,
3400               GNUNET_i2s (peer));
3401 #endif
3402   switch (type)
3403     {
3404     case GNUNET_MESSAGE_TYPE_CORE_SET_KEY:
3405       if (size != sizeof (struct SetKeyMessage))
3406         {
3407           GNUNET_break_op (0);
3408           return;
3409         }
3410       GNUNET_STATISTICS_update (stats, gettext_noop ("# session keys received"), 1, GNUNET_NO);
3411       handle_set_key (n, (const struct SetKeyMessage *) message);
3412       break;
3413     case GNUNET_MESSAGE_TYPE_CORE_ENCRYPTED_MESSAGE:
3414       if (size < sizeof (struct EncryptedMessage) +
3415           sizeof (struct GNUNET_MessageHeader))
3416         {
3417           GNUNET_break_op (0);
3418           return;
3419         }
3420       if ((n->status != PEER_STATE_KEY_RECEIVED) &&
3421           (n->status != PEER_STATE_KEY_CONFIRMED))
3422         {
3423           GNUNET_break_op (0);
3424           /* blacklist briefly (?); might help recover (?) */
3425           GNUNET_TRANSPORT_blacklist (sched, cfg,
3426                                       &n->peer,
3427                                       GNUNET_TIME_UNIT_SECONDS,
3428                                       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS,
3429                                                                      5),
3430                                       NULL, NULL);
3431           return;
3432         }
3433       handle_encrypted_message (n, (const struct EncryptedMessage *) message);
3434       break;
3435     case GNUNET_MESSAGE_TYPE_CORE_PING:
3436       if (size != sizeof (struct PingMessage))
3437         {
3438           GNUNET_break_op (0);
3439           return;
3440         }
3441       GNUNET_STATISTICS_update (stats, gettext_noop ("# PING messages received"), 1, GNUNET_NO);
3442       if ((n->status != PEER_STATE_KEY_RECEIVED) &&
3443           (n->status != PEER_STATE_KEY_CONFIRMED))
3444         {
3445 #if DEBUG_CORE
3446           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3447                       "Core service receives `%s' request from `%4s' but have not processed key; marking as pending.\n",
3448                       "PING", GNUNET_i2s (&n->peer));
3449 #endif
3450           GNUNET_free_non_null (n->pending_ping);
3451           n->pending_ping = GNUNET_malloc (sizeof (struct PingMessage));
3452           memcpy (n->pending_ping, message, sizeof (struct PingMessage));
3453           return;
3454         }
3455       handle_ping (n, (const struct PingMessage *) message);
3456       break;
3457     case GNUNET_MESSAGE_TYPE_CORE_PONG:
3458       if (size != sizeof (struct PongMessage))
3459         {
3460           GNUNET_break_op (0);
3461           return;
3462         }
3463       GNUNET_STATISTICS_update (stats, gettext_noop ("# PONG messages received"), 1, GNUNET_NO);
3464       if ( (n->status != PEER_STATE_KEY_RECEIVED) &&
3465            (n->status != PEER_STATE_KEY_CONFIRMED) )
3466         {
3467 #if DEBUG_CORE
3468           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3469                       "Core service receives `%s' request from `%4s' but have not processed key; marking as pending.\n",
3470                       "PONG", GNUNET_i2s (&n->peer));
3471 #endif
3472           GNUNET_free_non_null (n->pending_pong);
3473           n->pending_pong = GNUNET_malloc (sizeof (struct PongMessage));
3474           memcpy (n->pending_pong, message, sizeof (struct PongMessage));
3475           return;
3476         }
3477       handle_pong (n, (const struct PongMessage *) message);
3478       break;
3479     default:
3480       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3481                   _("Unsupported message of type %u received.\n"),
3482                   (unsigned int) type);
3483       return;
3484     }
3485   if (n->status == PEER_STATE_KEY_CONFIRMED)
3486     {
3487       now = GNUNET_TIME_absolute_get ();
3488       n->last_activity = now;
3489       if (!up)
3490         {
3491           GNUNET_STATISTICS_update (stats, gettext_noop ("# established sessions"), 1, GNUNET_NO);
3492           n->time_established = now;
3493         }
3494       if (n->keep_alive_task != GNUNET_SCHEDULER_NO_TASK)
3495         GNUNET_SCHEDULER_cancel (sched, n->keep_alive_task);
3496       n->keep_alive_task
3497         = GNUNET_SCHEDULER_add_delayed (sched,
3498                                         GNUNET_TIME_relative_divide (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, 2),
3499                                         &send_keep_alive,
3500                                         n);
3501     }
3502 }
3503
3504
3505 /**
3506  * Function that recalculates the bandwidth quota for the
3507  * given neighbour and transmits it to the transport service.
3508  *
3509  * @param cls neighbour for the quota update
3510  * @param tc context
3511  */
3512 static void
3513 neighbour_quota_update (void *cls,
3514                         const struct GNUNET_SCHEDULER_TaskContext *tc)
3515 {
3516   struct Neighbour *n = cls;
3517   struct GNUNET_BANDWIDTH_Value32NBO q_in;
3518   double pref_rel;
3519   double share;
3520   unsigned long long distributable;
3521   uint64_t need_per_peer;
3522   uint64_t need_per_second;
3523
3524   n->quota_update_task = GNUNET_SCHEDULER_NO_TASK;
3525   /* calculate relative preference among all neighbours;
3526      divides by a bit more to avoid division by zero AND to
3527      account for possibility of new neighbours joining any time
3528      AND to convert to double... */
3529   if (preference_sum == 0)
3530     {
3531       pref_rel = 1.0 / (double) neighbour_count;
3532     }
3533   else
3534     {
3535       pref_rel = n->current_preference / preference_sum;
3536     }
3537   need_per_peer = GNUNET_BANDWIDTH_value_get_available_until (MIN_BANDWIDTH_PER_PEER,
3538                                                               GNUNET_TIME_UNIT_SECONDS);
3539   need_per_second = need_per_peer * neighbour_count;
3540   distributable = 0;
3541   if (bandwidth_target_out_bps > need_per_second)
3542     distributable = bandwidth_target_out_bps - need_per_second;
3543   share = distributable * pref_rel;
3544   if (share + need_per_peer > ( (uint32_t)-1))
3545     q_in = GNUNET_BANDWIDTH_value_init ((uint32_t) -1);
3546   else
3547     q_in = GNUNET_BANDWIDTH_value_init (need_per_peer + (uint32_t) share);
3548   /* check if we want to disconnect for good due to inactivity */
3549   if ( (GNUNET_TIME_absolute_get_duration (n->last_activity).value > GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.value) &&
3550        (GNUNET_TIME_absolute_get_duration (n->time_established).value > GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.value) )
3551     {
3552 #if DEBUG_CORE
3553       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3554                   "Forcing disconnect of `%4s' due to inactivity (?).\n",
3555                   GNUNET_i2s (&n->peer));
3556 #endif
3557       q_in = GNUNET_BANDWIDTH_value_init (0); /* force disconnect */
3558     }
3559 #if DEBUG_CORE_QUOTA
3560   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3561               "Current quota for `%4s' is %u/%llu b/s in (old: %u b/s) / %u out (%u internal)\n",
3562               GNUNET_i2s (&n->peer),
3563               (unsigned int) ntohl (q_in.value__),
3564               bandwidth_target_out_bps,
3565               (unsigned int) ntohl (n->bw_in.value__),
3566               (unsigned int) ntohl (n->bw_out.value__),
3567               (unsigned int) ntohl (n->bw_out_internal_limit.value__));
3568 #endif
3569   if (n->bw_in.value__ != q_in.value__)
3570     {
3571       n->bw_in = q_in;
3572       GNUNET_TRANSPORT_set_quota (transport,
3573                                   &n->peer,
3574                                   n->bw_in,
3575                                   n->bw_out,
3576                                   GNUNET_TIME_UNIT_FOREVER_REL,
3577                                   NULL, NULL);
3578     }
3579   schedule_quota_update (n);
3580 }
3581
3582
3583 /**
3584  * Function called by transport to notify us that
3585  * a peer connected to us (on the network level).
3586  *
3587  * @param cls closure
3588  * @param peer the peer that connected
3589  * @param latency current latency of the connection
3590  * @param distance in overlay hops, as given by transport plugin
3591  */
3592 static void
3593 handle_transport_notify_connect (void *cls,
3594                                  const struct GNUNET_PeerIdentity *peer,
3595                                  struct GNUNET_TIME_Relative latency,
3596                                  unsigned int distance)
3597 {
3598   struct Neighbour *n;
3599   struct ConnectNotifyMessage cnm;
3600
3601   if (0 == memcmp (peer, &my_identity, sizeof (struct GNUNET_PeerIdentity)))
3602     {
3603       GNUNET_break (0);
3604       return;
3605     }
3606   n = find_neighbour (peer);
3607   if (n != NULL)
3608     {
3609       if (n->is_connected)
3610         {
3611           /* duplicate connect notification!? */
3612           GNUNET_break (0);
3613           return;
3614         }
3615     }
3616   else
3617     {
3618       n = create_neighbour (peer);
3619     }
3620   GNUNET_STATISTICS_update (stats,
3621                             gettext_noop ("# peers connected"),
3622                             1,
3623                             GNUNET_NO);
3624   n->is_connected = GNUNET_YES;
3625   n->last_latency = latency;
3626   n->last_distance = distance;
3627   GNUNET_BANDWIDTH_tracker_init (&n->available_send_window,
3628                                  n->bw_out,
3629                                  MAX_WINDOW_TIME_S);
3630   GNUNET_BANDWIDTH_tracker_init (&n->available_recv_window,
3631                                  n->bw_in,
3632                                  MAX_WINDOW_TIME_S);
3633 #if DEBUG_CORE
3634   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3635               "Received connection from `%4s'.\n",
3636               GNUNET_i2s (&n->peer));
3637 #endif
3638   cnm.header.size = htons (sizeof (struct ConnectNotifyMessage));
3639   cnm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_PRE_CONNECT);
3640   cnm.distance = htonl (n->last_distance);
3641   cnm.latency = GNUNET_TIME_relative_hton (n->last_latency);
3642   cnm.peer = *peer;
3643   send_to_all_clients (&cnm.header, GNUNET_YES, GNUNET_CORE_OPTION_SEND_PRE_CONNECT);
3644   GNUNET_TRANSPORT_set_quota (transport,
3645                               &n->peer,
3646                               n->bw_in,
3647                               n->bw_out,
3648                               GNUNET_TIME_UNIT_FOREVER_REL,
3649                               NULL, NULL);
3650   send_key (n);
3651 }
3652
3653
3654 /**
3655  * Function called by transport telling us that a peer
3656  * disconnected.
3657  *
3658  * @param cls closure
3659  * @param peer the peer that disconnected
3660  */
3661 static void
3662 handle_transport_notify_disconnect (void *cls,
3663                                     const struct GNUNET_PeerIdentity *peer)
3664 {
3665   struct DisconnectNotifyMessage cnm;
3666   struct Neighbour *n;
3667
3668 #if DEBUG_CORE
3669   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3670               "Peer `%4s' disconnected from us.\n", GNUNET_i2s (peer));
3671 #endif
3672   n = find_neighbour (peer);
3673   if (n == NULL)
3674     {
3675       GNUNET_break (0);
3676       return;
3677     }
3678   GNUNET_break (n->is_connected);
3679   cnm.header.size = htons (sizeof (struct DisconnectNotifyMessage));
3680   cnm.header.type = htons (GNUNET_MESSAGE_TYPE_CORE_NOTIFY_DISCONNECT);
3681   cnm.peer = *peer;
3682   send_to_all_clients (&cnm.header, GNUNET_YES, GNUNET_CORE_OPTION_SEND_DISCONNECT);
3683   n->is_connected = GNUNET_NO;
3684   GNUNET_STATISTICS_update (stats,
3685                             gettext_noop ("# peers connected"),
3686                             -1,
3687                             GNUNET_NO);
3688 }
3689
3690
3691 /**
3692  * Last task run during shutdown.  Disconnects us from
3693  * the transport.
3694  */
3695 static void
3696 cleaning_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
3697 {
3698   struct Neighbour *n;
3699   struct Client *c;
3700
3701 #if DEBUG_CORE
3702   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3703               "Core service shutting down.\n");
3704 #endif
3705   GNUNET_assert (transport != NULL);
3706   GNUNET_TRANSPORT_disconnect (transport);
3707   transport = NULL;
3708   while (NULL != (n = neighbours))
3709     {
3710       neighbours = n->next;
3711       GNUNET_assert (neighbour_count > 0);
3712       neighbour_count--;
3713       free_neighbour (n);
3714     }
3715   GNUNET_STATISTICS_set (stats, gettext_noop ("# active neighbours"), neighbour_count, GNUNET_NO);
3716   GNUNET_SERVER_notification_context_destroy (notifier);
3717   notifier = NULL;
3718   while (NULL != (c = clients))
3719     handle_client_disconnect (NULL, c->client_handle);
3720   if (my_private_key != NULL)
3721     GNUNET_CRYPTO_rsa_key_free (my_private_key);
3722   if (stats != NULL)
3723     GNUNET_STATISTICS_destroy (stats, GNUNET_NO);
3724 }
3725
3726
3727 /**
3728  * Initiate core service.
3729  *
3730  * @param cls closure
3731  * @param s scheduler to use
3732  * @param serv the initialized server
3733  * @param c configuration to use
3734  */
3735 static void
3736 run (void *cls,
3737      struct GNUNET_SCHEDULER_Handle *s,
3738      struct GNUNET_SERVER_Handle *serv,
3739      const struct GNUNET_CONFIGURATION_Handle *c)
3740 {
3741   char *keyfile;
3742
3743   sched = s;
3744   cfg = c;
3745   /* parse configuration */
3746   if (
3747        (GNUNET_OK !=
3748         GNUNET_CONFIGURATION_get_value_number (c,
3749                                                "CORE",
3750                                                "TOTAL_QUOTA_IN",
3751                                                &bandwidth_target_in_bps)) ||
3752        (GNUNET_OK !=
3753         GNUNET_CONFIGURATION_get_value_number (c,
3754                                                "CORE",
3755                                                "TOTAL_QUOTA_OUT",
3756                                                &bandwidth_target_out_bps)) ||
3757        (GNUNET_OK !=
3758         GNUNET_CONFIGURATION_get_value_filename (c,
3759                                                  "GNUNETD",
3760                                                  "HOSTKEY", &keyfile)))
3761     {
3762       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3763                   _
3764                   ("Core service is lacking key configuration settings.  Exiting.\n"));
3765       GNUNET_SCHEDULER_shutdown (s);
3766       return;
3767     }
3768   my_private_key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
3769   GNUNET_free (keyfile);
3770   if (my_private_key == NULL)
3771     {
3772       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3773                   _("Core service could not access hostkey.  Exiting.\n"));
3774       GNUNET_SCHEDULER_shutdown (s);
3775       return;
3776     }
3777   GNUNET_CRYPTO_rsa_key_get_public (my_private_key, &my_public_key);
3778   GNUNET_CRYPTO_hash (&my_public_key,
3779                       sizeof (my_public_key), &my_identity.hashPubKey);
3780   /* setup notification */
3781   server = serv;
3782   notifier = GNUNET_SERVER_notification_context_create (server,
3783                                                         MAX_NOTIFY_QUEUE);
3784   GNUNET_SERVER_disconnect_notify (server, &handle_client_disconnect, NULL);
3785   /* setup transport connection */
3786   transport = GNUNET_TRANSPORT_connect (sched,
3787                                         cfg,
3788                                         NULL,
3789                                         &handle_transport_receive,
3790                                         &handle_transport_notify_connect,
3791                                         &handle_transport_notify_disconnect);
3792   GNUNET_assert (NULL != transport);
3793   stats = GNUNET_STATISTICS_create (sched, "core", cfg);
3794   GNUNET_SCHEDULER_add_delayed (sched,
3795                                 GNUNET_TIME_UNIT_FOREVER_REL,
3796                                 &cleaning_task, NULL);
3797   /* process client requests */
3798   GNUNET_SERVER_add_handlers (server, handlers);
3799   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
3800               _("Core service of `%4s' ready.\n"), GNUNET_i2s (&my_identity));
3801 }
3802
3803
3804
3805 /**
3806  * The main function for the transport service.
3807  *
3808  * @param argc number of arguments from the command line
3809  * @param argv command line arguments
3810  * @return 0 ok, 1 on error
3811  */
3812 int
3813 main (int argc, char *const *argv)
3814 {
3815   return (GNUNET_OK ==
3816           GNUNET_SERVICE_run (argc,
3817                               argv,
3818                               "core",
3819                               GNUNET_SERVICE_OPTION_NONE,
3820                               &run, NULL)) ? 0 : 1;
3821 }
3822
3823 /* end of gnunet-service-core.c */