src/transport/plugin_transport_udp.c

   1 /*
   2  This file is part of GNUnet
   3  (C) 2010-2014 Christian Grothoff (and other contributing authors)
   4
   5  GNUnet is free software; you can redistribute it and/or modify
   6  it under the terms of the GNU General Public License as published
   7  by the Free Software Foundation; either version 3, or (at your
   8  option) any later version.
   9
  10  GNUnet is distributed in the hope that it will be useful, but
  11  WITHOUT ANY WARRANTY; without even the implied warranty of
  12  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  General Public License for more details.
  14
  15  You should have received a copy of the GNU General Public License
  16  along with GNUnet; see the file COPYING.  If not, write to the
  17  Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  Boston, MA 02111-1307, USA.
  19  */
  20
  21 /**
  22  * @file transport/plugin_transport_udp.c
  23  * @brief Implementation of the UDP transport protocol
  24  * @author Christian Grothoff
  25  * @author Nathan Evans
  26  * @author Matthias Wachs
  27  */
  28 #include "platform.h"
  29 #include "plugin_transport_udp.h"
  30 #include "gnunet_hello_lib.h"
  31 #include "gnunet_util_lib.h"
  32 #include "gnunet_fragmentation_lib.h"
  33 #include "gnunet_nat_lib.h"
  34 #include "gnunet_protocols.h"
  35 #include "gnunet_resolver_service.h"
  36 #include "gnunet_signatures.h"
  37 #include "gnunet_constants.h"
  38 #include "gnunet_statistics_service.h"
  39 #include "gnunet_transport_service.h"
  40 #include "gnunet_transport_plugin.h"
  41 #include "transport.h"
  42
  43 #define LOG(kind,...) GNUNET_log_from (kind, "transport-udp", __VA_ARGS__)
  44
  45 #define UDP_SESSION_TIME_OUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60)
  46
  47 /**
  48  * Number of messages we can defragment in parallel.  We only really
  49  * defragment 1 message at a time, but if messages get re-ordered, we
  50  * may want to keep knowledge about the previous message to avoid
  51  * discarding the current message in favor of a single fragment of a
  52  * previous message.  3 should be good since we don't expect massive
  53  * message reorderings with UDP.
  54  */
  55 #define UDP_MAX_MESSAGES_IN_DEFRAG 3
  56
  57 /**
  58  * We keep a defragmentation queue per sender address.  How many
  59  * sender addresses do we support at the same time? Memory consumption
  60  * is roughly a factor of 32k * UDP_MAX_MESSAGES_IN_DEFRAG times this
  61  * value. (So 128 corresponds to 12 MB and should suffice for
  62  * connecting to roughly 128 peers via UDP).
  63  */
  64 #define UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG 128
  65
  66
  67 /**
  68  * Closure for #append_port().
  69  */
  70 struct PrettyPrinterContext
  71 {
  72   /**
  73    * DLL
  74    */
  75   struct PrettyPrinterContext *next;
  76
  77   /**
  78    * DLL
  79    */
  80   struct PrettyPrinterContext *prev;
  81
  82   /**
  83    * Our plugin.
  84    */
  85   struct Plugin *plugin;
  86
  87   /**
  88    * Resolver handle
  89    */
  90   struct GNUNET_RESOLVER_RequestHandle *resolver_handle;
  91
  92   /**
  93    * Function to call with the result.
  94    */
  95   GNUNET_TRANSPORT_AddressStringCallback asc;
  96
  97   /**
  98    * Clsoure for @e asc.
  99    */
 100   void *asc_cls;
 101
 102   /**
 103    * Timeout task
 104    */
 105   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
 106
 107   /**
 108    * IPv6 address
 109    */
 110   int ipv6;
 111
 112   /**
 113    * Options
 114    */
 115   uint32_t options;
 116
 117   /**
 118    * Port to add after the IP address.
 119    */
 120   uint16_t port;
 121
 122 };
 123
 124
 125 /**
 126  * Session with another peer.
 127  */
 128 struct Session
 129 {
 130   /**
 131    * Which peer is this session for?
 132    */
 133   struct GNUNET_PeerIdentity target;
 134
 135   /**
 136    * Plugin this session belongs to.
 137    */
 138   struct Plugin *plugin;
 139
 140   /**
 141    * Context for dealing with fragments.
 142    */
 143   struct UDP_FragmentationContext *frag_ctx;
 144
 145   /**
 146    * Desired delay for next sending we send to other peer
 147    */
 148   struct GNUNET_TIME_Relative flow_delay_for_other_peer;
 149
 150   /**
 151    * Desired delay for next sending we received from other peer
 152    */
 153   struct GNUNET_TIME_Absolute flow_delay_from_other_peer;
 154
 155   /**
 156    * Session timeout task
 157    */
 158   GNUNET_SCHEDULER_TaskIdentifier timeout_task;
 159
 160   /**
 161    * When does this session time out?
 162    */
 163   struct GNUNET_TIME_Absolute timeout;
 164
 165   /**
 166    * expected delay for ACKs
 167    */
 168   struct GNUNET_TIME_Relative last_expected_ack_delay;
 169
 170   /**
 171    * desired delay between UDP messages
 172    */
 173   struct GNUNET_TIME_Relative last_expected_msg_delay;
 174
 175   /**
 176    * Address metrics (as set by the "update_address_metrics" by
 177    * the environment).
 178    */
 179   struct GNUNET_ATS_Information ats;
 180
 181   /**
 182    * Our own address.
 183    */
 184   struct GNUNET_HELLO_Address *address;
 185
 186   /**
 187    * Number of bytes waiting for transmission to this peer.
 188    */
 189   unsigned long long bytes_in_queue;
 190
 191   /**
 192    * Number of messages waiting for transmission to this peer.
 193    */
 194   unsigned int msgs_in_queue;
 195
 196   /**
 197    * Reference counter to indicate that this session is
 198    * currently being used and must not be destroyed;
 199    * setting @e in_destroy will destroy it as soon as
 200    * possible.
 201    */
 202   unsigned int rc;
 203
 204   /**
 205    * Is this session about to be destroyed (sometimes we cannot
 206    * destroy a session immediately as below us on the stack
 207    * there might be code that still uses it; in this case,
 208    * @e rc is non-zero).
 209    */
 210   int in_destroy;
 211 };
 212
 213
 214 /**
 215  * Closure for #process_inbound_tokenized_messages().
 216  */
 217 struct SourceInformation
 218 {
 219   /**
 220    * Sender identity.
 221    */
 222   struct GNUNET_PeerIdentity sender;
 223
 224   /**
 225    * Source address.
 226    */
 227   const void *arg;
 228
 229   /**
 230    * Associated session.
 231    */
 232   struct Session *session;
 233
 234   /**
 235    * Number of bytes in source address.
 236    */
 237   size_t args;
 238
 239 };
 240
 241 /**
 242  * Closure for #find_receive_context().
 243  */
 244 struct FindReceiveContext
 245 {
 246   /**
 247    * Where to store the result.
 248    */
 249   struct DefragContext *rc;
 250
 251   /**
 252    * Address to find.
 253    */
 254   const struct sockaddr *addr;
 255
 256   /**
 257    *
 258    */
 259   struct Session *session;
 260
 261   /**
 262    * Number of bytes in @e addr.
 263    */
 264   socklen_t addr_len;
 265
 266 };
 267
 268 /**
 269  * Data structure to track defragmentation contexts based
 270  * on the source of the UDP traffic.
 271  */
 272 struct DefragContext
 273 {
 274
 275   /**
 276    * Defragmentation context.
 277    */
 278   struct GNUNET_DEFRAGMENT_Context *defrag;
 279
 280   /**
 281    * Source address this receive context is for (allocated at the
 282    * end of the struct).
 283    */
 284   const struct sockaddr *src_addr;
 285
 286   /**
 287    * Reference to master plugin struct.
 288    */
 289   struct Plugin *plugin;
 290
 291   /**
 292    * Node in the defrag heap.
 293    */
 294   struct GNUNET_CONTAINER_HeapNode *hnode;
 295
 296   /**
 297    * Length of @e src_addr.
 298    */
 299   size_t addr_len;
 300 };
 301
 302
 303 /**
 304  * Context to send fragmented messages
 305  */
 306 struct UDP_FragmentationContext
 307 {
 308   /**
 309    * Next in linked list
 310    */
 311   struct UDP_FragmentationContext *next;
 312
 313   /**
 314    * Previous in linked list
 315    */
 316   struct UDP_FragmentationContext *prev;
 317
 318   /**
 319    * The plugin
 320    */
 321   struct Plugin *plugin;
 322
 323   /**
 324    * Handle for GNUNET_FRAGMENT context
 325    */
 326   struct GNUNET_FRAGMENT_Context *frag;
 327
 328   /**
 329    * The session this fragmentation context belongs to
 330    */
 331   struct Session *session;
 332
 333   /**
 334    * Function to call upon completion of the transmission.
 335    */
 336   GNUNET_TRANSPORT_TransmitContinuation cont;
 337
 338   /**
 339    * Closure for @e cont.
 340    */
 341   void *cont_cls;
 342
 343   /**
 344    * Message timeout
 345    */
 346   struct GNUNET_TIME_Absolute timeout;
 347
 348   /**
 349    * Payload size of original unfragmented message
 350    */
 351   size_t payload_size;
 352
 353   /**
 354    * Bytes used to send all fragments on wire including UDP overhead
 355    */
 356   size_t on_wire_size;
 357
 358   /**
 359    * FIXME.
 360    */
 361   unsigned int fragments_used;
 362
 363 };
 364
 365
 366 /**
 367  * Message types included in a `struct UDP_MessageWrapper`
 368  */
 369 enum UDP_MessageType
 370 {
 371   /**
 372    * Uninitialized (error)
 373    */
 374   UMT_UNDEFINED = 0,
 375
 376   /**
 377    * Fragment of a message.
 378    */
 379   UMT_MSG_FRAGMENTED = 1,
 380
 381   /**
 382    *
 383    */
 384   UMT_MSG_FRAGMENTED_COMPLETE = 2,
 385
 386   /**
 387    * Unfragmented message.
 388    */
 389   UMT_MSG_UNFRAGMENTED = 3,
 390
 391   /**
 392    * Receipt confirmation.
 393    */
 394   UMT_MSG_ACK = 4
 395
 396 };
 397
 398
 399 /**
 400  * Information we track for each message in the queue.
 401  */
 402 struct UDP_MessageWrapper
 403 {
 404   /**
 405    * Session this message belongs to
 406    */
 407   struct Session *session;
 408
 409   /**
 410    * DLL of messages
 411    * previous element
 412    */
 413   struct UDP_MessageWrapper *prev;
 414
 415   /**
 416    * DLL of messages
 417    * previous element
 418    */
 419   struct UDP_MessageWrapper *next;
 420
 421   /**
 422    * Message with size msg_size including UDP specific overhead
 423    */
 424   char *msg_buf;
 425
 426   /**
 427    * Function to call upon completion of the transmission.
 428    */
 429   GNUNET_TRANSPORT_TransmitContinuation cont;
 430
 431   /**
 432    * Closure for @e cont.
 433    */
 434   void *cont_cls;
 435
 436   /**
 437    * Fragmentation context
 438    * frag_ctx == NULL if transport <= MTU
 439    * frag_ctx != NULL if transport > MTU
 440    */
 441   struct UDP_FragmentationContext *frag_ctx;
 442
 443   /**
 444    * Message timeout
 445    */
 446   struct GNUNET_TIME_Absolute timeout;
 447
 448   /**
 449    * Size of UDP message to send including UDP specific overhead
 450    */
 451   size_t msg_size;
 452
 453   /**
 454    * Payload size of original message
 455    */
 456   size_t payload_size;
 457
 458   /**
 459    * Message type
 460    */
 461   enum UDP_MessageType msg_type;
 462
 463 };
 464
 465
 466 /**
 467  * UDP ACK Message-Packet header (after defragmentation).
 468  */
 469 struct UDP_ACK_Message
 470 {
 471   /**
 472    * Message header.
 473    */
 474   struct GNUNET_MessageHeader header;
 475
 476   /**
 477    * Desired delay for flow control
 478    */
 479   uint32_t delay;
 480
 481   /**
 482    * What is the identity of the sender
 483    */
 484   struct GNUNET_PeerIdentity sender;
 485
 486 };
 487
 488
 489 /**
 490  * If a session monitor is attached, notify it about the new
 491  * session state.
 492  *
 493  * @param plugin our plugin
 494  * @param session session that changed state
 495  * @param state new state of the session
 496  */
 497 static void
 498 notify_session_monitor (struct Plugin *plugin,
 499                         struct Session *session,
 500                         enum GNUNET_TRANSPORT_SessionState state)
 501 {
 502   struct GNUNET_TRANSPORT_SessionInfo info;
 503
 504   if (NULL == plugin->sic)
 505     return;
 506   memset (&info, 0, sizeof (info));
 507   info.state = state;
 508   info.is_inbound = GNUNET_SYSERR; /* hard to say */
 509   info.num_msg_pending = session->msgs_in_queue;
 510   info.num_bytes_pending = session->bytes_in_queue;
 511   /* info.receive_delay remains zero as this is not supported by UDP
 512      (cannot selectively not receive from 'some' peer while continuing
 513      to receive from others) */
 514   info.session_timeout = session->timeout;
 515   info.address = session->address;
 516   plugin->sic (plugin->sic_cls,
 517                session,
 518                &info);
 519 }
 520
 521
 522 /**
 523  * We have been notified that our readset has something to read.  We don't
 524  * know which socket needs to be read, so we have to check each one
 525  * Then reschedule this function to be called again once more is available.
 526  *
 527  * @param cls the plugin handle
 528  * @param tc the scheduling context (for rescheduling this function again)
 529  */
 530 static void
 531 udp_plugin_select (void *cls,
 532                    const struct GNUNET_SCHEDULER_TaskContext *tc);
 533
 534
 535 /**
 536  * We have been notified that our readset has something to read.  We don't
 537  * know which socket needs to be read, so we have to check each one
 538  * Then reschedule this function to be called again once more is available.
 539  *
 540  * @param cls the plugin handle
 541  * @param tc the scheduling context (for rescheduling this function again)
 542  */
 543 static void
 544 udp_plugin_select_v6 (void *cls,
 545                       const struct GNUNET_SCHEDULER_TaskContext *tc);
 546
 547
 548 /**
 549  * (re)schedule select tasks for this plugin.
 550  *
 551  * @param plugin plugin to reschedule
 552  */
 553 static void
 554 schedule_select (struct Plugin *plugin)
 555 {
 556   struct GNUNET_TIME_Relative min_delay;
 557   struct UDP_MessageWrapper *udpw;
 558
 559   if ((GNUNET_YES == plugin->enable_ipv4) && (NULL != plugin->sockv4))
 560   {
 561     /* Find a message ready to send:
 562      * Flow delay from other peer is expired or not set (0) */
 563     min_delay = GNUNET_TIME_UNIT_FOREVER_REL;
 564     for (udpw = plugin->ipv4_queue_head; NULL != udpw; udpw = udpw->next)
 565       min_delay = GNUNET_TIME_relative_min (min_delay,
 566           GNUNET_TIME_absolute_get_remaining (
 567               udpw->session->flow_delay_from_other_peer));
 568
 569     if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK )
 570       GNUNET_SCHEDULER_cancel (plugin->select_task);
 571
 572     /* Schedule with:
 573      * - write active set if message is ready
 574      * - timeout minimum delay */
 575     plugin->select_task = GNUNET_SCHEDULER_add_select (
 576         GNUNET_SCHEDULER_PRIORITY_DEFAULT,
 577         (0 == min_delay.rel_value_us) ?
 578             GNUNET_TIME_UNIT_FOREVER_REL : min_delay, plugin->rs_v4,
 579         (0 == min_delay.rel_value_us) ? plugin->ws_v4 : NULL,
 580         &udp_plugin_select, plugin);
 581   }
 582   if ((GNUNET_YES == plugin->enable_ipv6) && (NULL != plugin->sockv6))
 583   {
 584     min_delay = GNUNET_TIME_UNIT_FOREVER_REL;
 585     for (udpw = plugin->ipv6_queue_head; NULL != udpw; udpw = udpw->next)
 586       min_delay = GNUNET_TIME_relative_min (min_delay,
 587           GNUNET_TIME_absolute_get_remaining (
 588               udpw->session->flow_delay_from_other_peer));
 589
 590     if (GNUNET_SCHEDULER_NO_TASK != plugin->select_task_v6)
 591       GNUNET_SCHEDULER_cancel (plugin->select_task_v6);
 592     plugin->select_task_v6 = GNUNET_SCHEDULER_add_select (
 593         GNUNET_SCHEDULER_PRIORITY_DEFAULT,
 594         (0 == min_delay.rel_value_us) ?
 595             GNUNET_TIME_UNIT_FOREVER_REL : min_delay, plugin->rs_v6,
 596         (0 == min_delay.rel_value_us) ? plugin->ws_v6 : NULL,
 597         &udp_plugin_select_v6, plugin);
 598   }
 599 }
 600
 601
 602 /**
 603  * Function called for a quick conversion of the binary address to
 604  * a numeric address.  Note that the caller must not free the
 605  * address and that the next call to this function is allowed
 606  * to override the address again.
 607  *
 608  * @param cls closure
 609  * @param addr binary address
 610  * @param addrlen length of the address
 611  * @return string representing the same address
 612  */
 613 const char *
 614 udp_address_to_string (void *cls,
 615                        const void *addr,
 616                        size_t addrlen)
 617 {
 618   static char rbuf[INET6_ADDRSTRLEN + 10];
 619   char buf[INET6_ADDRSTRLEN];
 620   const void *sb;
 621   struct in_addr a4;
 622   struct in6_addr a6;
 623   const struct IPv4UdpAddress *t4;
 624   const struct IPv6UdpAddress *t6;
 625   int af;
 626   uint16_t port;
 627   uint32_t options;
 628
 629   if ((NULL != addr) && (addrlen == sizeof(struct IPv6UdpAddress)))
 630   {
 631     t6 = addr;
 632     af = AF_INET6;
 633     options = ntohl (t6->options);
 634     port = ntohs (t6->u6_port);
 635     memcpy (&a6, &t6->ipv6_addr, sizeof(a6));
 636     sb = &a6;
 637   }
 638   else if ((NULL != addr) && (addrlen == sizeof(struct IPv4UdpAddress)))
 639   {
 640     t4 = addr;
 641     af = AF_INET;
 642     options = ntohl (t4->options);
 643     port = ntohs (t4->u4_port);
 644     memcpy (&a4, &t4->ipv4_addr, sizeof(a4));
 645     sb = &a4;
 646   }
 647   else
 648   {
 649     return NULL;
 650   }
 651   inet_ntop (af, sb, buf, INET6_ADDRSTRLEN);
 652
 653   GNUNET_snprintf (rbuf, sizeof(rbuf),
 654       (af == AF_INET6) ? "%s.%u.[%s]:%u" : "%s.%u.%s:%u", PLUGIN_NAME, options,
 655       buf, port);
 656   return rbuf;
 657 }
 658
 659
 660 /**
 661  * Function called to convert a string address to
 662  * a binary address.
 663  *
 664  * @param cls closure ('struct Plugin*')
 665  * @param addr string address
 666  * @param addrlen length of the address
 667  * @param buf location to store the buffer
 668  * @param added location to store the number of bytes in the buffer.
 669  *        If the function returns #GNUNET_SYSERR, its contents are undefined.
 670  * @return #GNUNET_OK on success, #GNUNET_SYSERR on failure
 671  */
 672 static int
 673 udp_string_to_address (void *cls,
 674                        const char *addr,
 675                        uint16_t addrlen,
 676                        void **buf,
 677                        size_t *added)
 678 {
 679   struct sockaddr_storage socket_address;
 680   char *address;
 681   char *plugin;
 682   char *optionstr;
 683   uint32_t options;
 684
 685   /* Format tcp.options.address:port */
 686   address = NULL;
 687   plugin = NULL;
 688   optionstr = NULL;
 689
 690   if ((NULL == addr) || (addrlen == 0))
 691   {
 692     GNUNET_break(0);
 693     return GNUNET_SYSERR;
 694   }
 695   if ('\0' != addr[addrlen - 1])
 696   {
 697     GNUNET_break(0);
 698     return GNUNET_SYSERR;
 699   }
 700   if (strlen (addr) != addrlen - 1)
 701   {
 702     GNUNET_break(0);
 703     return GNUNET_SYSERR;
 704   }
 705   plugin = GNUNET_strdup (addr);
 706   optionstr = strchr (plugin, '.');
 707   if (NULL == optionstr)
 708   {
 709     GNUNET_break(0);
 710     GNUNET_free(plugin);
 711     return GNUNET_SYSERR;
 712   }
 713   optionstr[0] = '\0';
 714   optionstr++;
 715   options = atol (optionstr);
 716   address = strchr (optionstr, '.');
 717   if (NULL == address)
 718   {
 719     GNUNET_break(0);
 720     GNUNET_free(plugin);
 721     return GNUNET_SYSERR;
 722   }
 723   address[0] = '\0';
 724   address++;
 725
 726   if (GNUNET_OK !=
 727       GNUNET_STRINGS_to_address_ip (address, strlen (address),
 728                                     &socket_address))
 729   {
 730     GNUNET_break(0);
 731     GNUNET_free(plugin);
 732     return GNUNET_SYSERR;
 733   }
 734
 735   GNUNET_free(plugin);
 736
 737   switch (socket_address.ss_family)
 738   {
 739   case AF_INET:
 740   {
 741     struct IPv4UdpAddress *u4;
 742     struct sockaddr_in *in4 = (struct sockaddr_in *) &socket_address;
 743     u4 = GNUNET_new (struct IPv4UdpAddress);
 744     u4->options = htonl (options);
 745     u4->ipv4_addr = in4->sin_addr.s_addr;
 746     u4->u4_port = in4->sin_port;
 747     *buf = u4;
 748     *added = sizeof(struct IPv4UdpAddress);
 749     return GNUNET_OK;
 750   }
 751   case AF_INET6:
 752   {
 753     struct IPv6UdpAddress *u6;
 754     struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &socket_address;
 755     u6 = GNUNET_new (struct IPv6UdpAddress);
 756     u6->options = htonl (options);
 757     u6->ipv6_addr = in6->sin6_addr;
 758     u6->u6_port = in6->sin6_port;
 759     *buf = u6;
 760     *added = sizeof(struct IPv6UdpAddress);
 761     return GNUNET_OK;
 762   }
 763   default:
 764     GNUNET_break(0);
 765     return GNUNET_SYSERR;
 766   }
 767 }
 768
 769
 770 /**
 771  * Append our port and forward the result.
 772  *
 773  * @param cls a `struct PrettyPrinterContext *`
 774  * @param hostname result from DNS resolver
 775  */
 776 static void
 777 append_port (void *cls,
 778              const char *hostname)
 779 {
 780   struct PrettyPrinterContext *ppc = cls;
 781   struct Plugin *plugin = ppc->plugin;
 782   char *ret;
 783
 784   if (NULL == hostname)
 785   {
 786     /* Final call, done */
 787     ppc->asc (ppc->asc_cls,
 788               NULL,
 789               GNUNET_OK);
 790     GNUNET_CONTAINER_DLL_remove (plugin->ppc_dll_head,
 791                                  plugin->ppc_dll_tail,
 792                                  ppc);
 793     ppc->resolver_handle = NULL;
 794     GNUNET_free (ppc);
 795     return;
 796   }
 797   if (GNUNET_YES == ppc->ipv6)
 798     GNUNET_asprintf (&ret,
 799                      "%s.%u.[%s]:%d",
 800                      PLUGIN_NAME,
 801                      ppc->options,
 802                      hostname,
 803                      ppc->port);
 804   else
 805     GNUNET_asprintf (&ret,
 806                      "%s.%u.%s:%d",
 807                      PLUGIN_NAME,
 808                      ppc->options,
 809                      hostname,
 810                      ppc->port);
 811   ppc->asc (ppc->asc_cls,
 812             ret,
 813             GNUNET_OK);
 814   GNUNET_free (ret);
 815 }
 816
 817
 818 /**
 819  * Convert the transports address to a nice, human-readable
 820  * format.
 821  *
 822  * @param cls closure with the `struct Plugin *`
 823  * @param type name of the transport that generated the address
 824  * @param addr one of the addresses of the host, NULL for the last address
 825  *        the specific address format depends on the transport
 826  * @param addrlen length of the address
 827  * @param numeric should (IP) addresses be displayed in numeric form?
 828  * @param timeout after how long should we give up?
 829  * @param asc function to call on each string
 830  * @param asc_cls closure for @a asc
 831  */
 832 static void
 833 udp_plugin_address_pretty_printer (void *cls,
 834                                    const char *type,
 835                                    const void *addr,
 836                                    size_t addrlen,
 837                                    int numeric,
 838                                    struct GNUNET_TIME_Relative timeout,
 839                                    GNUNET_TRANSPORT_AddressStringCallback asc,
 840                                    void *asc_cls)
 841 {
 842   struct Plugin *plugin = cls;
 843   struct PrettyPrinterContext *ppc;
 844   const void *sb;
 845   size_t sbs;
 846   struct sockaddr_in a4;
 847   struct sockaddr_in6 a6;
 848   const struct IPv4UdpAddress *u4;
 849   const struct IPv6UdpAddress *u6;
 850   uint16_t port;
 851   uint32_t options;
 852
 853   if (addrlen == sizeof(struct IPv6UdpAddress))
 854   {
 855     u6 = addr;
 856     memset (&a6, 0, sizeof(a6));
 857     a6.sin6_family = AF_INET6;
 858 #if HAVE_SOCKADDR_IN_SIN_LEN
 859     a6.sin6_len = sizeof (a6);
 860 #endif
 861     a6.sin6_port = u6->u6_port;
 862     memcpy (&a6.sin6_addr, &u6->ipv6_addr, sizeof(struct in6_addr));
 863     port = ntohs (u6->u6_port);
 864     options = ntohl (u6->options);
 865     sb = &a6;
 866     sbs = sizeof(a6);
 867   }
 868   else if (addrlen == sizeof(struct IPv4UdpAddress))
 869   {
 870     u4 = addr;
 871     memset (&a4, 0, sizeof(a4));
 872     a4.sin_family = AF_INET;
 873 #if HAVE_SOCKADDR_IN_SIN_LEN
 874     a4.sin_len = sizeof (a4);
 875 #endif
 876     a4.sin_port = u4->u4_port;
 877     a4.sin_addr.s_addr = u4->ipv4_addr;
 878     port = ntohs (u4->u4_port);
 879     options = ntohl (u4->options);
 880     sb = &a4;
 881     sbs = sizeof(a4);
 882   }
 883   else
 884   {
 885     /* invalid address */
 886     GNUNET_break_op (0);
 887     asc (asc_cls, NULL , GNUNET_SYSERR);
 888     asc (asc_cls, NULL, GNUNET_OK);
 889     return;
 890   }
 891   ppc = GNUNET_new (struct PrettyPrinterContext);
 892   ppc->plugin = plugin;
 893   ppc->asc = asc;
 894   ppc->asc_cls = asc_cls;
 895   ppc->port = port;
 896   ppc->options = options;
 897   if (addrlen == sizeof(struct IPv6UdpAddress))
 898     ppc->ipv6 = GNUNET_YES;
 899   else
 900     ppc->ipv6 = GNUNET_NO;
 901   GNUNET_CONTAINER_DLL_insert (plugin->ppc_dll_head,
 902                                plugin->ppc_dll_tail,
 903                                ppc);
 904   ppc->resolver_handle
 905     = GNUNET_RESOLVER_hostname_get (sb,
 906                                     sbs,
 907                                     ! numeric,
 908                                     timeout,
 909                                     &append_port, ppc);
 910 }
 911
 912
 913 /**
 914  * FIXME.
 915  */
 916 static void
 917 call_continuation (struct UDP_MessageWrapper *udpw,
 918                    int result)
 919 {
 920   struct Session *session = udpw->session;
 921   struct Plugin *plugin = session->plugin;
 922   size_t overhead;
 923
 924   LOG (GNUNET_ERROR_TYPE_DEBUG,
 925        "Calling continuation for %u byte message to `%s' with result %s\n",
 926        udpw->payload_size, GNUNET_i2s (&udpw->session->target),
 927        (GNUNET_OK == result) ? "OK" : "SYSERR");
 928
 929   if (udpw->msg_size >= udpw->payload_size)
 930     overhead = udpw->msg_size - udpw->payload_size;
 931   else
 932     overhead = udpw->msg_size;
 933
 934   switch (result)
 935   {
 936   case GNUNET_OK:
 937     switch (udpw->msg_type)
 938     {
 939     case UMT_MSG_UNFRAGMENTED:
 940       if (NULL != udpw->cont)
 941       {
 942         /* Transport continuation */
 943         udpw->cont (udpw->cont_cls, &udpw->session->target, result,
 944             udpw->payload_size, udpw->msg_size);
 945       }
 946       GNUNET_STATISTICS_update (plugin->env->stats,
 947           "# UDP, unfragmented msgs, messages, sent, success", 1, GNUNET_NO);
 948       GNUNET_STATISTICS_update (plugin->env->stats,
 949           "# UDP, unfragmented msgs, bytes payload, sent, success",
 950           udpw->payload_size, GNUNET_NO);
 951       GNUNET_STATISTICS_update (plugin->env->stats,
 952           "# UDP, unfragmented msgs, bytes overhead, sent, success", overhead,
 953           GNUNET_NO);
 954       GNUNET_STATISTICS_update (plugin->env->stats,
 955           "# UDP, total, bytes overhead, sent", overhead, GNUNET_NO);
 956       GNUNET_STATISTICS_update (plugin->env->stats,
 957           "# UDP, total, bytes payload, sent", udpw->payload_size, GNUNET_NO);
 958       break;
 959     case UMT_MSG_FRAGMENTED_COMPLETE:
 960       GNUNET_assert(NULL != udpw->frag_ctx);
 961       if (udpw->frag_ctx->cont != NULL )
 962         udpw->frag_ctx->cont (udpw->frag_ctx->cont_cls, &udpw->session->target,
 963             GNUNET_OK, udpw->frag_ctx->payload_size,
 964             udpw->frag_ctx->on_wire_size);
 965       GNUNET_STATISTICS_update (plugin->env->stats,
 966           "# UDP, fragmented msgs, messages, sent, success", 1, GNUNET_NO);
 967       GNUNET_STATISTICS_update (plugin->env->stats,
 968           "# UDP, fragmented msgs, bytes payload, sent, success",
 969           udpw->payload_size, GNUNET_NO);
 970       GNUNET_STATISTICS_update (plugin->env->stats,
 971           "# UDP, fragmented msgs, bytes overhead, sent, success", overhead,
 972           GNUNET_NO);
 973       GNUNET_STATISTICS_update (plugin->env->stats,
 974           "# UDP, total, bytes overhead, sent", overhead, GNUNET_NO);
 975       GNUNET_STATISTICS_update (plugin->env->stats,
 976           "# UDP, total, bytes payload, sent", udpw->payload_size, GNUNET_NO);
 977       GNUNET_STATISTICS_update (plugin->env->stats,
 978           "# UDP, fragmented msgs, messages, pending", -1, GNUNET_NO);
 979       break;
 980     case UMT_MSG_FRAGMENTED:
 981       /* Fragmented message: enqueue next fragment */
 982       if (NULL != udpw->cont)
 983         udpw->cont (udpw->cont_cls, &udpw->session->target, result,
 984             udpw->payload_size, udpw->msg_size);
 985       GNUNET_STATISTICS_update (plugin->env->stats,
 986           "# UDP, fragmented msgs, fragments, sent, success", 1, GNUNET_NO);
 987       GNUNET_STATISTICS_update (plugin->env->stats,
 988           "# UDP, fragmented msgs, fragments bytes, sent, success",
 989           udpw->msg_size, GNUNET_NO);
 990       break;
 991     case UMT_MSG_ACK:
 992       /* No continuation */
 993       GNUNET_STATISTICS_update (plugin->env->stats,
 994           "# UDP, ACK msgs, messages, sent, success", 1, GNUNET_NO);
 995       GNUNET_STATISTICS_update (plugin->env->stats,
 996           "# UDP, ACK msgs, bytes overhead, sent, success", overhead,
 997           GNUNET_NO);
 998       GNUNET_STATISTICS_update (plugin->env->stats,
 999           "# UDP, total, bytes overhead, sent", overhead, GNUNET_NO);
1000       break;
1001     default:
1002       GNUNET_break(0);
1003       break;
1004     }
1005     break;
1006   case GNUNET_SYSERR:
1007     switch (udpw->msg_type)
1008     {
1009     case UMT_MSG_UNFRAGMENTED:
1010       /* Unfragmented message: failed to send */
1011       if (NULL != udpw->cont)
1012         udpw->cont (udpw->cont_cls, &udpw->session->target, result,
1013             udpw->payload_size, overhead);
1014       GNUNET_STATISTICS_update (plugin->env->stats,
1015           "# UDP, unfragmented msgs, messages, sent, failure", 1, GNUNET_NO);
1016       GNUNET_STATISTICS_update (plugin->env->stats,
1017           "# UDP, unfragmented msgs, bytes payload, sent, failure",
1018           udpw->payload_size, GNUNET_NO);
1019       GNUNET_STATISTICS_update (plugin->env->stats,
1020           "# UDP, unfragmented msgs, bytes overhead, sent, failure", overhead,
1021           GNUNET_NO);
1022       break;
1023     case UMT_MSG_FRAGMENTED_COMPLETE:
1024       GNUNET_assert(NULL != udpw->frag_ctx);
1025       if (udpw->frag_ctx->cont != NULL )
1026         udpw->frag_ctx->cont (udpw->frag_ctx->cont_cls, &udpw->session->target,
1027             GNUNET_SYSERR, udpw->frag_ctx->payload_size,
1028             udpw->frag_ctx->on_wire_size);
1029       GNUNET_STATISTICS_update (plugin->env->stats,
1030           "# UDP, fragmented msgs, messages, sent, failure", 1, GNUNET_NO);
1031       GNUNET_STATISTICS_update (plugin->env->stats,
1032           "# UDP, fragmented msgs, bytes payload, sent, failure",
1033           udpw->payload_size, GNUNET_NO);
1034       GNUNET_STATISTICS_update (plugin->env->stats,
1035           "# UDP, fragmented msgs, bytes payload, sent, failure", overhead,
1036           GNUNET_NO);
1037       GNUNET_STATISTICS_update (plugin->env->stats,
1038           "# UDP, fragmented msgs, bytes payload, sent, failure", overhead,
1039           GNUNET_NO);
1040       GNUNET_STATISTICS_update (plugin->env->stats,
1041           "# UDP, fragmented msgs, messages, pending", -1, GNUNET_NO);
1042       break;
1043     case UMT_MSG_FRAGMENTED:
1044       GNUNET_assert(NULL != udpw->frag_ctx);
1045       /* Fragmented message: failed to send */
1046       GNUNET_STATISTICS_update (plugin->env->stats,
1047           "# UDP, fragmented msgs, fragments, sent, failure", 1, GNUNET_NO);
1048       GNUNET_STATISTICS_update (plugin->env->stats,
1049           "# UDP, fragmented msgs, fragments bytes, sent, failure",
1050           udpw->msg_size, GNUNET_NO);
1051       break;
1052     case UMT_MSG_ACK:
1053       /* ACK message: failed to send */
1054       GNUNET_STATISTICS_update (plugin->env->stats,
1055           "# UDP, ACK msgs, messages, sent, failure", 1, GNUNET_NO);
1056       break;
1057     default:
1058       GNUNET_break(0);
1059       break;
1060     }
1061     break;
1062   default:
1063     GNUNET_break(0);
1064     break;
1065   }
1066 }
1067
1068
1069 /**
1070  * Check if the given port is plausible (must be either our listen
1071  * port or our advertised port).  If it is neither, we return
1072  * #GNUNET_SYSERR.
1073  *
1074  * @param plugin global variables
1075  * @param in_port port number to check
1076  * @return #GNUNET_OK if port is either open_port or adv_port
1077  */
1078 static int
1079 check_port (struct Plugin *plugin,
1080             uint16_t in_port)
1081 {
1082   if ((in_port == plugin->port) || (in_port == plugin->aport))
1083     return GNUNET_OK;
1084   return GNUNET_SYSERR;
1085 }
1086
1087
1088 /**
1089  * Function that will be called to check if a binary address for this
1090  * plugin is well-formed and corresponds to an address for THIS peer
1091  * (as per our configuration).  Naturally, if absolutely necessary,
1092  * plugins can be a bit conservative in their answer, but in general
1093  * plugins should make sure that the address does not redirect
1094  * traffic to a 3rd party that might try to man-in-the-middle our
1095  * traffic.
1096  *
1097  * @param cls closure, should be our handle to the Plugin
1098  * @param addr pointer to the address
1099  * @param addrlen length of @a addr
1100  * @return #GNUNET_OK if this is a plausible address for this peer
1101  *         and transport, #GNUNET_SYSERR if not
1102  */
1103 static int
1104 udp_plugin_check_address (void *cls,
1105                           const void *addr,
1106                           size_t addrlen)
1107 {
1108   struct Plugin *plugin = cls;
1109   struct IPv4UdpAddress *v4;
1110   struct IPv6UdpAddress *v6;
1111
1112   if ((addrlen != sizeof(struct IPv4UdpAddress))
1113       && (addrlen != sizeof(struct IPv6UdpAddress)))
1114   {
1115     GNUNET_break_op(0);
1116     return GNUNET_SYSERR;
1117   }
1118   if (addrlen == sizeof(struct IPv4UdpAddress))
1119   {
1120     v4 = (struct IPv4UdpAddress *) addr;
1121     if (GNUNET_OK != check_port (plugin, ntohs (v4->u4_port)))
1122       return GNUNET_SYSERR;
1123     if (GNUNET_OK
1124         != GNUNET_NAT_test_address (plugin->nat, &v4->ipv4_addr,
1125             sizeof(struct in_addr)))
1126       return GNUNET_SYSERR;
1127   }
1128   else
1129   {
1130     v6 = (struct IPv6UdpAddress *) addr;
1131     if (IN6_IS_ADDR_LINKLOCAL (&v6->ipv6_addr))
1132     {
1133       GNUNET_break_op(0);
1134       return GNUNET_SYSERR;
1135     }
1136     if (GNUNET_OK != check_port (plugin, ntohs (v6->u6_port)))
1137       return GNUNET_SYSERR;
1138     if (GNUNET_OK !=
1139         GNUNET_NAT_test_address (plugin->nat,
1140                                  &v6->ipv6_addr,
1141                                  sizeof(struct in6_addr)))
1142       return GNUNET_SYSERR;
1143   }
1144   return GNUNET_OK;
1145 }
1146
1147
1148 /**
1149  * Function to free last resources associated with a session.
1150  *
1151  * @param s session to free
1152  */
1153 static void
1154 free_session (struct Session *s)
1155 {
1156   if (NULL != s->frag_ctx)
1157   {
1158     GNUNET_FRAGMENT_context_destroy (s->frag_ctx->frag, NULL, NULL );
1159     GNUNET_free(s->frag_ctx);
1160     s->frag_ctx = NULL;
1161   }
1162   GNUNET_free(s);
1163 }
1164
1165
1166 /**
1167  * Remove a message from the transmission queue.
1168  *
1169  * @param plugin the UDP plugin
1170  * @param udpw message wrapper to queue
1171  */
1172 static void
1173 dequeue (struct Plugin *plugin,
1174          struct UDP_MessageWrapper *udpw)
1175 {
1176   struct Session *session = udpw->session;
1177
1178   if (plugin->bytes_in_buffer < udpw->msg_size)
1179   {
1180     GNUNET_break (0);
1181   }
1182   else
1183   {
1184     GNUNET_STATISTICS_update (plugin->env->stats,
1185                               "# UDP, total, bytes in buffers",
1186                               -(long long) udpw->msg_size,
1187                               GNUNET_NO);
1188     plugin->bytes_in_buffer -= udpw->msg_size;
1189   }
1190   GNUNET_STATISTICS_update (plugin->env->stats,
1191                             "# UDP, total, msgs in buffers",
1192                             -1, GNUNET_NO);
1193   if (udpw->session->address->address_length == sizeof(struct IPv4UdpAddress))
1194     GNUNET_CONTAINER_DLL_remove (plugin->ipv4_queue_head,
1195                                  plugin->ipv4_queue_tail,
1196                                  udpw);
1197   else if (udpw->session->address->address_length == sizeof(struct IPv6UdpAddress))
1198     GNUNET_CONTAINER_DLL_remove (plugin->ipv6_queue_head,
1199                                  plugin->ipv6_queue_tail,
1200                                  udpw);
1201   else
1202   {
1203     GNUNET_break (0);
1204     return;
1205   }
1206   GNUNET_assert (session->msgs_in_queue > 0);
1207   session->msgs_in_queue--;
1208   GNUNET_assert (session->bytes_in_queue >= udpw->msg_size);
1209   session->bytes_in_queue -= udpw->msg_size;
1210 }
1211
1212
1213 /**
1214  * FIXME.
1215  */
1216 static void
1217 fragmented_message_done (struct UDP_FragmentationContext *fc,
1218                          int result)
1219 {
1220   struct Plugin *plugin = fc->plugin;
1221   struct Session *s = fc->session;
1222   struct UDP_MessageWrapper *udpw;
1223   struct UDP_MessageWrapper *tmp;
1224   struct UDP_MessageWrapper dummy;
1225
1226   LOG (GNUNET_ERROR_TYPE_DEBUG,
1227        "%p : Fragmented message removed with result %s\n",
1228        fc,
1229        (result == GNUNET_SYSERR) ? "FAIL" : "SUCCESS");
1230
1231   /* Call continuation for fragmented message */
1232   memset (&dummy, 0, sizeof(dummy));
1233   dummy.msg_type = UMT_MSG_FRAGMENTED_COMPLETE;
1234   dummy.msg_size = s->frag_ctx->on_wire_size;
1235   dummy.payload_size = s->frag_ctx->payload_size;
1236   dummy.frag_ctx = s->frag_ctx;
1237   dummy.cont = NULL;
1238   dummy.cont_cls = NULL;
1239   dummy.session = s;
1240   call_continuation (&dummy, result);
1241   /* Remove leftover fragments from queue */
1242   if (s->address->address_length == sizeof(struct IPv6UdpAddress))
1243   {
1244     udpw = plugin->ipv6_queue_head;
1245     while (NULL != udpw)
1246     {
1247       tmp = udpw->next;
1248       if ((udpw->frag_ctx != NULL )&& (udpw->frag_ctx == s->frag_ctx)){
1249       dequeue (plugin, udpw);
1250       call_continuation (udpw, GNUNET_SYSERR);
1251       GNUNET_free (udpw);
1252     }
1253       udpw = tmp;
1254     }
1255   }
1256   if (s->address->address_length == sizeof(struct IPv4UdpAddress))
1257   {
1258     udpw = plugin->ipv4_queue_head;
1259     while (udpw != NULL )
1260     {
1261       tmp = udpw->next;
1262       if ((NULL != udpw->frag_ctx) && (udpw->frag_ctx == s->frag_ctx))
1263       {
1264         dequeue (plugin, udpw);
1265         call_continuation (udpw, GNUNET_SYSERR);
1266         GNUNET_free(udpw);
1267       }
1268       udpw = tmp;
1269     }
1270   }
1271   notify_session_monitor (s->plugin,
1272                           s,
1273                           GNUNET_TRANSPORT_SS_UP);
1274   /* Destroy fragmentation context */
1275   GNUNET_FRAGMENT_context_destroy (fc->frag,
1276                                    &s->last_expected_msg_delay,
1277                                    &s->last_expected_ack_delay);
1278   s->frag_ctx = NULL;
1279   GNUNET_free (fc);
1280 }
1281
1282
1283 /**
1284  * Scan the heap for a receive context with the given address.
1285  *
1286  * @param cls the `struct FindReceiveContext`
1287  * @param node internal node of the heap
1288  * @param element value stored at the node (a 'struct ReceiveContext')
1289  * @param cost cost associated with the node
1290  * @return #GNUNET_YES if we should continue to iterate,
1291  *         #GNUNET_NO if not.
1292  */
1293 static int
1294 find_receive_context (void *cls,
1295                       struct GNUNET_CONTAINER_HeapNode *node,
1296                       void *element,
1297                       GNUNET_CONTAINER_HeapCostType cost)
1298 {
1299   struct FindReceiveContext *frc = cls;
1300   struct DefragContext *e = element;
1301
1302   if ((frc->addr_len == e->addr_len)
1303       && (0 == memcmp (frc->addr, e->src_addr, frc->addr_len)))
1304   {
1305     frc->rc = e;
1306     return GNUNET_NO;
1307   }
1308   return GNUNET_YES;
1309 }
1310
1311
1312 /**
1313  * Functions with this signature are called whenever we need
1314  * to close a session due to a disconnect or failure to
1315  * establish a connection.
1316  *
1317  * @param cls closure with the `struct Plugin`
1318  * @param s session to close down
1319  * @return #GNUNET_OK on success
1320  */
1321 static int
1322 udp_disconnect_session (void *cls,
1323                         struct Session *s)
1324 {
1325   struct Plugin *plugin = cls;
1326   struct UDP_MessageWrapper *udpw;
1327   struct UDP_MessageWrapper *next;
1328   struct FindReceiveContext frc;
1329
1330   GNUNET_assert(GNUNET_YES != s->in_destroy);
1331   LOG(GNUNET_ERROR_TYPE_DEBUG, "Session %p to peer `%s' address ended\n", s,
1332       GNUNET_i2s (&s->target),
1333       udp_address_to_string (NULL, s->address->address, s->address->address_length));
1334   /* stop timeout task */
1335   if (GNUNET_SCHEDULER_NO_TASK != s->timeout_task)
1336   {
1337     GNUNET_SCHEDULER_cancel (s->timeout_task);
1338     s->timeout_task = GNUNET_SCHEDULER_NO_TASK;
1339   }
1340   if (NULL != s->frag_ctx)
1341   {
1342     /* Remove fragmented message due to disconnect */
1343     fragmented_message_done (s->frag_ctx, GNUNET_SYSERR);
1344   }
1345
1346   frc.rc = NULL;
1347   frc.addr = s->address->address;
1348   frc.addr_len = s->address->address_length;
1349   /* Lookup existing receive context for this address */
1350   if (NULL != plugin->defrag_ctxs)
1351   {
1352     GNUNET_CONTAINER_heap_iterate (plugin->defrag_ctxs,
1353                                    &find_receive_context,
1354                                    &frc);
1355     if (NULL != frc.rc)
1356     {
1357       struct DefragContext *d_ctx = frc.rc;
1358
1359       GNUNET_CONTAINER_heap_remove_node (d_ctx->hnode);
1360       GNUNET_DEFRAGMENT_context_destroy (d_ctx->defrag);
1361       GNUNET_free (d_ctx);
1362     }
1363   }
1364   next = plugin->ipv4_queue_head;
1365   while (NULL != (udpw = next))
1366   {
1367     next = udpw->next;
1368     if (udpw->session == s)
1369     {
1370       dequeue (plugin, udpw);
1371       call_continuation (udpw, GNUNET_SYSERR);
1372       GNUNET_free(udpw);
1373     }
1374   }
1375   next = plugin->ipv6_queue_head;
1376   while (NULL != (udpw = next))
1377   {
1378     next = udpw->next;
1379     if (udpw->session == s)
1380     {
1381       dequeue (plugin, udpw);
1382       call_continuation (udpw, GNUNET_SYSERR);
1383       GNUNET_free(udpw);
1384     }
1385   }
1386   notify_session_monitor (s->plugin,
1387                           s,
1388                           GNUNET_TRANSPORT_SS_DOWN);
1389   plugin->env->session_end (plugin->env->cls,
1390                             s->address,
1391                             s);
1392
1393   if (NULL != s->frag_ctx)
1394   {
1395     if (NULL != s->frag_ctx->cont)
1396     {
1397       s->frag_ctx->cont (s->frag_ctx->cont_cls,
1398                          &s->target,
1399                          GNUNET_SYSERR,
1400                          s->frag_ctx->payload_size,
1401                          s->frag_ctx->on_wire_size);
1402       LOG (GNUNET_ERROR_TYPE_DEBUG,
1403            "Calling continuation for fragemented message to `%s' with result SYSERR\n",
1404            GNUNET_i2s (&s->target));
1405     }
1406   }
1407
1408   GNUNET_assert(GNUNET_YES ==
1409                 GNUNET_CONTAINER_multipeermap_remove (plugin->sessions,
1410                                                       &s->target,
1411                                                       s));
1412   GNUNET_STATISTICS_set (plugin->env->stats,
1413                          "# UDP sessions active",
1414                          GNUNET_CONTAINER_multipeermap_size (plugin->sessions),
1415                          GNUNET_NO);
1416   if (s->rc > 0)
1417   {
1418     s->in_destroy = GNUNET_YES;
1419   }
1420   else
1421   {
1422     GNUNET_HELLO_address_free (s->address);
1423     free_session (s);
1424   }
1425   return GNUNET_OK;
1426 }
1427
1428
1429 /**
1430  * Function that is called to get the keepalive factor.
1431  * #GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT is divided by this number to
1432  * calculate the interval between keepalive packets.
1433  *
1434  * @param cls closure with the `struct Plugin`
1435  * @return keepalive factor
1436  */
1437 static unsigned int
1438 udp_query_keepalive_factor (void *cls)
1439 {
1440   return 15;
1441 }
1442
1443
1444 /**
1445  * Destroy a session, plugin is being unloaded.
1446  *
1447  * @param cls the `struct Plugin`
1448  * @param key hash of public key of target peer
1449  * @param value a `struct PeerSession *` to clean up
1450  * @return #GNUNET_OK (continue to iterate)
1451  */
1452 static int
1453 disconnect_and_free_it (void *cls,
1454                         const struct GNUNET_PeerIdentity *key,
1455                         void *value)
1456 {
1457   struct Plugin *plugin = cls;
1458
1459   udp_disconnect_session (plugin, value);
1460   return GNUNET_OK;
1461 }
1462
1463
1464 /**
1465  * Disconnect from a remote node.  Clean up session if we have one for
1466  * this peer.
1467  *
1468  * @param cls closure for this call (should be handle to Plugin)
1469  * @param target the peeridentity of the peer to disconnect
1470  * @return #GNUNET_OK on success, #GNUNET_SYSERR if the operation failed
1471  */
1472 static void
1473 udp_disconnect (void *cls,
1474                 const struct GNUNET_PeerIdentity *target)
1475 {
1476   struct Plugin *plugin = cls;
1477
1478   LOG (GNUNET_ERROR_TYPE_DEBUG,
1479        "Disconnecting from peer `%s'\n",
1480        GNUNET_i2s (target));
1481   /* Clean up sessions */
1482   GNUNET_CONTAINER_multipeermap_get_multiple (plugin->sessions,
1483                                               target,
1484                                               &disconnect_and_free_it,
1485                                               plugin);
1486 }
1487
1488
1489 /**
1490  * Session was idle, so disconnect it
1491  *
1492  * @param cls the `struct Session` to time out
1493  * @param tc scheduler context
1494  */
1495 static void
1496 session_timeout (void *cls,
1497                  const struct GNUNET_SCHEDULER_TaskContext *tc)
1498 {
1499   struct Session *s = cls;
1500   struct Plugin *plugin = s->plugin;
1501   struct GNUNET_TIME_Relative left;
1502
1503   s->timeout_task = GNUNET_SCHEDULER_NO_TASK;
1504   left = GNUNET_TIME_absolute_get_remaining (s->timeout);
1505   if (left.rel_value_us > 0)
1506   {
1507     /* not actually our turn yet, but let's at least update
1508        the monitor, it may think we're about to die ... */
1509     notify_session_monitor (s->plugin,
1510                             s,
1511                             GNUNET_TRANSPORT_SS_UP);
1512     s->timeout_task = GNUNET_SCHEDULER_add_delayed (left,
1513                                                     &session_timeout,
1514                                                     s);
1515     return;
1516   }
1517   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1518               "Session %p was idle for %s, disconnecting\n",
1519               s,
1520               GNUNET_STRINGS_relative_time_to_string (UDP_SESSION_TIME_OUT,
1521                                                       GNUNET_YES));
1522   /* call session destroy function */
1523   udp_disconnect_session (plugin, s);
1524 }
1525
1526
1527 /**
1528  * Increment session timeout due to activity
1529  *
1530  * @param s session to reschedule timeout activity for
1531  */
1532 static void
1533 reschedule_session_timeout (struct Session *s)
1534 {
1535   if (GNUNET_YES == s->in_destroy)
1536     return;
1537   GNUNET_assert(GNUNET_SCHEDULER_NO_TASK != s->timeout_task);
1538   s->timeout = GNUNET_TIME_relative_to_absolute (UDP_SESSION_TIME_OUT);
1539 }
1540
1541
1542 /**
1543  * FIXME.
1544  */
1545 static struct Session *
1546 create_session (struct Plugin *plugin,
1547                 const struct GNUNET_HELLO_Address *address)
1548 {
1549   struct Session *s;
1550
1551   s = GNUNET_new (struct Session);
1552   s->plugin = plugin;
1553   s->address = GNUNET_HELLO_address_copy (address);
1554   s->target = address->peer;
1555   s->last_expected_ack_delay = GNUNET_TIME_relative_multiply (
1556       GNUNET_TIME_UNIT_MILLISECONDS, 250);
1557   s->last_expected_msg_delay = GNUNET_TIME_UNIT_MILLISECONDS;
1558   s->flow_delay_from_other_peer = GNUNET_TIME_UNIT_ZERO_ABS;
1559   s->flow_delay_for_other_peer = GNUNET_TIME_UNIT_ZERO;
1560   s->timeout = GNUNET_TIME_relative_to_absolute (UDP_SESSION_TIME_OUT);
1561   s->timeout_task = GNUNET_SCHEDULER_add_delayed (UDP_SESSION_TIME_OUT,
1562                                                   &session_timeout, s);
1563   return s;
1564 }
1565
1566
1567 /**
1568  * Function obtain the network type for a session
1569  *
1570  * @param cls closure ('struct Plugin*')
1571  * @param session the session
1572  * @return the network type
1573  */
1574 static enum GNUNET_ATS_Network_Type
1575 udp_get_network (void *cls,
1576                  struct Session *session)
1577 {
1578   return ntohl (session->ats.value);
1579 }
1580
1581
1582 /**
1583  * Closure for #session_cmp_it().
1584  */
1585 struct SessionCompareContext
1586 {
1587   /**
1588    * Set to session matching the address.
1589    */
1590   struct Session *res;
1591
1592   /**
1593    * Address we are looking for.
1594    */
1595   const struct GNUNET_HELLO_Address *address;
1596 };
1597
1598
1599 /**
1600  * Find a session with a matching address.
1601  *
1602  * @param cls the `struct SessionCompareContext *`
1603  * @param key peer identity (unused)
1604  * @param value the `struct Session *`
1605  * @return #GNUNET_NO if we found the session, #GNUNET_OK if not
1606  */
1607 static int
1608 session_cmp_it (void *cls,
1609                 const struct GNUNET_PeerIdentity *key,
1610                 void *value)
1611 {
1612   struct SessionCompareContext *cctx = cls;
1613   const struct GNUNET_HELLO_Address *address = cctx->address;
1614   struct Session *s = value;
1615
1616   LOG (GNUNET_ERROR_TYPE_DEBUG,
1617        "Comparing address %s <-> %s\n",
1618        udp_address_to_string (NULL,
1619                               address->address,
1620                               address->address_length),
1621        udp_address_to_string (NULL,
1622                               s->address->address,
1623                               s->address->address_length));
1624   if (0 == GNUNET_HELLO_address_cmp(s->address, cctx->address))
1625   {
1626     cctx->res = s;
1627     return GNUNET_NO;
1628   }
1629   return GNUNET_YES;
1630 }
1631
1632
1633 /**
1634  * Creates a new outbound session the transport service will use to
1635  * send data to the peer
1636  *
1637  * @param cls the plugin
1638  * @param address the address
1639  * @return the session or NULL of max connections exceeded
1640  */
1641 static struct Session *
1642 udp_plugin_lookup_session (void *cls,
1643                            const struct GNUNET_HELLO_Address *address)
1644 {
1645   struct Plugin * plugin = cls;
1646   struct IPv6UdpAddress *udp_a6;
1647   struct IPv4UdpAddress *udp_a4;
1648   struct SessionCompareContext cctx;
1649
1650   if ( (NULL == address->address) ||
1651        ((address->address_length != sizeof (struct IPv4UdpAddress)) &&
1652         (address->address_length != sizeof (struct IPv6UdpAddress))))
1653   {
1654     LOG (GNUNET_ERROR_TYPE_WARNING,
1655          _("Trying to create session for address of unexpected length %u (should be %u or %u)\n"),
1656          address->address_length,
1657          sizeof (struct IPv4UdpAddress),
1658          sizeof (struct IPv6UdpAddress));
1659     return NULL;
1660   }
1661
1662   if (address->address_length == sizeof(struct IPv4UdpAddress))
1663   {
1664     if (plugin->sockv4 == NULL)
1665       return NULL;
1666     udp_a4 = (struct IPv4UdpAddress *) address->address;
1667     if (udp_a4->u4_port == 0)
1668       return NULL;
1669   }
1670
1671   if (address->address_length == sizeof(struct IPv6UdpAddress))
1672   {
1673     if (plugin->sockv6 == NULL)
1674       return NULL;
1675     udp_a6 = (struct IPv6UdpAddress *) address->address;
1676     if (udp_a6->u6_port == 0)
1677       return NULL;
1678   }
1679
1680   /* check if session already exists */
1681   cctx.address = address;
1682   cctx.res = NULL;
1683   LOG (GNUNET_ERROR_TYPE_DEBUG,
1684        "Looking for existing session for peer `%s' `%s' \n",
1685        GNUNET_i2s (&address->peer),
1686        udp_address_to_string(NULL, address->address, address->address_length));
1687   GNUNET_CONTAINER_multipeermap_get_multiple (plugin->sessions, &address->peer,
1688       session_cmp_it, &cctx);
1689   if (cctx.res != NULL )
1690   {
1691     LOG (GNUNET_ERROR_TYPE_DEBUG,
1692          "Found existing session %p\n",
1693          cctx.res);
1694     return cctx.res;
1695   }
1696   return NULL;
1697 }
1698
1699
1700 /**
1701  * Context to lookup a session based on a IP address
1702  */
1703 struct LookupContext
1704 {
1705   /**
1706    * The result
1707    */
1708   struct Session *res;
1709
1710   /**
1711    * The socket address
1712    */
1713   const struct sockaddr *address;
1714
1715   /**
1716    * The socket address length
1717    */
1718   size_t addr_len;
1719
1720   /**
1721    * Is a fragmentation context required for the session
1722    */
1723   int must_have_frag_ctx;
1724 };
1725
1726
1727 /**
1728  * Find a session with a matching address.
1729  * FIXME: very similar code to #udp_plugin_lookup_session() above.
1730  * Unify?
1731  *
1732  * @param cls the `struct LookupContext *`
1733  * @param key peer identity (unused)
1734  * @param value the `struct Session *`
1735  * @return #GNUNET_NO if we found the session, #GNUNET_OK if not
1736  */
1737 static int
1738 lookup_session_by_sockaddr_it (void *cls,
1739                                const struct GNUNET_PeerIdentity *key,
1740                                void *value)
1741 {
1742   struct LookupContext *l_ctx = cls;
1743   struct Session *s = value;
1744   struct IPv4UdpAddress u4;
1745   struct IPv6UdpAddress u6;
1746   void *arg;
1747   size_t args;
1748
1749   /* convert address */
1750   switch (l_ctx->address->sa_family)
1751   {
1752   case AF_INET:
1753     GNUNET_assert(l_ctx->addr_len == sizeof(struct sockaddr_in));
1754     memset (&u4, 0, sizeof(u4));
1755     u6.options = htonl (0);
1756     u4.ipv4_addr = ((struct sockaddr_in *) l_ctx->address)->sin_addr.s_addr;
1757     u4.u4_port = ((struct sockaddr_in *) l_ctx->address)->sin_port;
1758     arg = &u4;
1759     args = sizeof(u4);
1760     break;
1761   case AF_INET6:
1762     GNUNET_assert(l_ctx->addr_len == sizeof(struct sockaddr_in6));
1763     memset (&u6, 0, sizeof(u6));
1764     u6.options = htonl (0);
1765     u6.ipv6_addr = ((struct sockaddr_in6 *) l_ctx->address)->sin6_addr;
1766     u6.u6_port = ((struct sockaddr_in6 *) l_ctx->address)->sin6_port;
1767     arg = &u6;
1768     args = sizeof(u6);
1769     break;
1770   default:
1771     GNUNET_break(0);
1772     return GNUNET_YES;
1773   }
1774   if ( (GNUNET_YES == l_ctx->must_have_frag_ctx) &&
1775        (NULL == s->frag_ctx))
1776     return GNUNET_YES;
1777
1778   /* Does not compare peer identities but addresses */
1779   if ((args == s->address->address_length) &&
1780       (0 == memcmp (arg, s->address->address, args)))
1781   {
1782     l_ctx->res = s;
1783     return GNUNET_NO;
1784   }
1785   return GNUNET_YES;
1786 }
1787
1788
1789 static struct Session *
1790 udp_plugin_create_session (void *cls,
1791                            const struct GNUNET_HELLO_Address *address)
1792 {
1793   struct Plugin *plugin = cls;
1794   struct Session *s;
1795   struct IPv4UdpAddress *udp_v4;
1796   struct IPv6UdpAddress *udp_v6;
1797
1798   s = create_session (plugin, address);
1799   if (sizeof (struct IPv4UdpAddress) == address->address_length)
1800   {
1801     struct sockaddr_in v4;
1802     udp_v4 = (struct IPv4UdpAddress *) address->address;
1803     memset (&v4, '\0', sizeof (v4));
1804     v4.sin_family = AF_INET;
1805 #if HAVE_SOCKADDR_IN_SIN_LEN
1806     v4.sin_len = sizeof (struct sockaddr_in);
1807 #endif
1808     v4.sin_port = udp_v4->u4_port;
1809     v4.sin_addr.s_addr = udp_v4->ipv4_addr;
1810     s->ats = plugin->env->get_address_type (plugin->env->cls,
1811                                             (const struct sockaddr *) &v4,
1812                                             sizeof (v4));
1813   }
1814   else if (sizeof (struct IPv6UdpAddress) == address->address_length)
1815   {
1816     struct sockaddr_in6 v6;
1817     udp_v6 = (struct IPv6UdpAddress *) address->address;
1818     memset (&v6, '\0', sizeof (v6));
1819     v6.sin6_family = AF_INET6;
1820 #if HAVE_SOCKADDR_IN_SIN_LEN
1821     v6.sin6_len = sizeof (struct sockaddr_in6);
1822 #endif
1823     v6.sin6_port = udp_v6->u6_port;
1824     v6.sin6_addr = udp_v6->ipv6_addr;
1825     s->ats = plugin->env->get_address_type (plugin->env->cls,
1826         (const struct sockaddr *) &v6, sizeof (v6));
1827   }
1828
1829   if (NULL == s)
1830     return NULL; /* protocol not supported or address invalid */
1831   LOG(GNUNET_ERROR_TYPE_DEBUG,
1832       "Creating new %s session %p for peer `%s' address `%s'\n",
1833       GNUNET_HELLO_address_check_option (address, GNUNET_HELLO_ADDRESS_INFO_INBOUND) ? "inbound" : "outbound",
1834       s, GNUNET_i2s (&address->peer),
1835       udp_address_to_string( NULL,address->address,address->address_length));
1836   GNUNET_assert(
1837       GNUNET_OK == GNUNET_CONTAINER_multipeermap_put (plugin->sessions, &s->target, s, GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
1838   GNUNET_STATISTICS_set (plugin->env->stats, "# UDP sessions active",
1839       GNUNET_CONTAINER_multipeermap_size (plugin->sessions), GNUNET_NO);
1840   return s;
1841 }
1842
1843
1844 /**
1845  * Function that will be called whenever the transport service wants to
1846  * notify the plugin that a session is still active and in use and
1847  * therefore the session timeout for this session has to be updated
1848  *
1849  * @param cls closure
1850  * @param peer which peer was the session for
1851  * @param session which session is being updated
1852  */
1853 static void
1854 udp_plugin_update_session_timeout (void *cls,
1855                                    const struct GNUNET_PeerIdentity *peer,
1856                                    struct Session *session)
1857 {
1858   struct Plugin *plugin = cls;
1859
1860   if (GNUNET_YES !=
1861       GNUNET_CONTAINER_multipeermap_contains_value (plugin->sessions,
1862                                                     peer,
1863                                                     session))
1864   {
1865     GNUNET_break(0);
1866     return;
1867   }
1868   /* Reschedule session timeout */
1869   reschedule_session_timeout (session);
1870 }
1871
1872
1873 /**
1874  * Creates a new outbound session the transport service will use to send data to the
1875  * peer
1876  *
1877  * @param cls the plugin
1878  * @param address the address
1879  * @return the session or NULL of max connections exceeded
1880  */
1881 static struct Session *
1882 udp_plugin_get_session (void *cls,
1883                         const struct GNUNET_HELLO_Address *address)
1884 {
1885   struct Session *s;
1886
1887   if (NULL == address)
1888   {
1889     GNUNET_break(0);
1890     return NULL;
1891   }
1892   if ( (address->address_length != sizeof(struct IPv4UdpAddress)) &&
1893        (address->address_length != sizeof(struct IPv6UdpAddress)) )
1894     return NULL;
1895
1896   /* otherwise create new */
1897   if (NULL != (s = udp_plugin_lookup_session (cls, address)))
1898     return s;
1899   return udp_plugin_create_session (cls, address);
1900 }
1901
1902
1903 /**
1904  * Enqueue a message for transmission.
1905  *
1906  * @param plugin the UDP plugin
1907  * @param udpw message wrapper to queue
1908  */
1909 static void
1910 enqueue (struct Plugin *plugin,
1911          struct UDP_MessageWrapper *udpw)
1912 {
1913   struct Session *session = udpw->session;
1914
1915   if (plugin->bytes_in_buffer + udpw->msg_size > INT64_MAX)
1916   {
1917     GNUNET_break (0);
1918   }
1919   else
1920   {
1921     GNUNET_STATISTICS_update (plugin->env->stats,
1922         "# UDP, total, bytes in buffers", udpw->msg_size, GNUNET_NO);
1923     plugin->bytes_in_buffer += udpw->msg_size;
1924   }
1925   GNUNET_STATISTICS_update (plugin->env->stats,
1926                             "# UDP, total, msgs in buffers",
1927                             1, GNUNET_NO);
1928   if (udpw->session->address->address_length == sizeof (struct IPv4UdpAddress))
1929     GNUNET_CONTAINER_DLL_insert(plugin->ipv4_queue_head,
1930                                 plugin->ipv4_queue_tail,
1931                                 udpw);
1932   else if (udpw->session->address->address_length == sizeof (struct IPv6UdpAddress))
1933     GNUNET_CONTAINER_DLL_insert (plugin->ipv6_queue_head,
1934                                  plugin->ipv6_queue_tail,
1935                                  udpw);
1936   else
1937   {
1938     GNUNET_break (0);
1939     return;
1940   }
1941   session->msgs_in_queue++;
1942   session->bytes_in_queue += udpw->msg_size;
1943 }
1944
1945
1946 /**
1947  * Fragment message was transmitted via UDP, let fragmentation know
1948  * to send the next fragment now.
1949  *
1950  * @param cls the `struct UDPMessageWrapper *` of the fragment
1951  * @param target destination peer (ignored)
1952  * @param result #GNUNET_OK on success (ignored)
1953  * @param payload bytes payload sent
1954  * @param physical bytes physical sent
1955  */
1956 static void
1957 send_next_fragment (void *cls,
1958                     const struct GNUNET_PeerIdentity *target,
1959                     int result,
1960                     size_t payload,
1961                     size_t physical)
1962 {
1963   struct UDP_MessageWrapper *udpw = cls;
1964
1965   GNUNET_FRAGMENT_context_transmission_done (udpw->frag_ctx->frag);
1966 }
1967
1968
1969 /**
1970  * Function that is called with messages created by the fragmentation
1971  * module.  In the case of the 'proc' callback of the
1972  * #GNUNET_FRAGMENT_context_create() function, this function must
1973  * eventually call #GNUNET_FRAGMENT_context_transmission_done().
1974  *
1975  * @param cls closure, the 'struct FragmentationContext'
1976  * @param msg the message that was created
1977  */
1978 static void
1979 enqueue_fragment (void *cls,
1980                   const struct GNUNET_MessageHeader *msg)
1981 {
1982   struct UDP_FragmentationContext *frag_ctx = cls;
1983   struct Plugin *plugin = frag_ctx->plugin;
1984   struct UDP_MessageWrapper * udpw;
1985   size_t msg_len = ntohs (msg->size);
1986
1987   LOG (GNUNET_ERROR_TYPE_DEBUG,
1988        "Enqueuing fragment with %u bytes\n",
1989        msg_len);
1990   frag_ctx->fragments_used++;
1991   udpw = GNUNET_malloc (sizeof (struct UDP_MessageWrapper) + msg_len);
1992   udpw->session = frag_ctx->session;
1993   udpw->msg_buf = (char *) &udpw[1];
1994   udpw->msg_size = msg_len;
1995   udpw->payload_size = msg_len; /*FIXME: minus fragment overhead */
1996   udpw->cont = &send_next_fragment;
1997   udpw->cont_cls = udpw;
1998   udpw->timeout = frag_ctx->timeout;
1999   udpw->frag_ctx = frag_ctx;
2000   udpw->msg_type = UMT_MSG_FRAGMENTED;
2001   memcpy (udpw->msg_buf, msg, msg_len);
2002   enqueue (plugin, udpw);
2003   schedule_select (plugin);
2004 }
2005
2006
2007 /**
2008  * Function that can be used by the transport service to transmit
2009  * a message using the plugin.   Note that in the case of a
2010  * peer disconnecting, the continuation MUST be called
2011  * prior to the disconnect notification itself.  This function
2012  * will be called with this peer's HELLO message to initiate
2013  * a fresh connection to another peer.
2014  *
2015  * @param cls closure
2016  * @param s which session must be used
2017  * @param msgbuf the message to transmit
2018  * @param msgbuf_size number of bytes in 'msgbuf'
2019  * @param priority how important is the message (most plugins will
2020  *                 ignore message priority and just FIFO)
2021  * @param to how long to wait at most for the transmission (does not
2022  *                require plugins to discard the message after the timeout,
2023  *                just advisory for the desired delay; most plugins will ignore
2024  *                this as well)
2025  * @param cont continuation to call once the message has
2026  *        been transmitted (or if the transport is ready
2027  *        for the next transmission call; or if the
2028  *        peer disconnected...); can be NULL
2029  * @param cont_cls closure for cont
2030  * @return number of bytes used (on the physical network, with overheads);
2031  *         -1 on hard errors (i.e. address invalid); 0 is a legal value
2032  *         and does NOT mean that the message was not transmitted (DV)
2033  */
2034 static ssize_t
2035 udp_plugin_send (void *cls,
2036                  struct Session *s,
2037                  const char *msgbuf,
2038                  size_t msgbuf_size,
2039                  unsigned int priority,
2040                  struct GNUNET_TIME_Relative to,
2041                  GNUNET_TRANSPORT_TransmitContinuation cont,
2042                  void *cont_cls)
2043 {
2044   struct Plugin *plugin = cls;
2045   size_t udpmlen = msgbuf_size + sizeof(struct UDPMessage);
2046   struct UDP_FragmentationContext * frag_ctx;
2047   struct UDP_MessageWrapper * udpw;
2048   struct UDPMessage *udp;
2049   char mbuf[udpmlen];
2050   GNUNET_assert(plugin != NULL);
2051   GNUNET_assert(s != NULL);
2052
2053   if ( (s->address->address_length == sizeof(struct IPv6UdpAddress)) &&
2054        (plugin->sockv6 == NULL) )
2055     return GNUNET_SYSERR;
2056   if ( (s->address->address_length == sizeof(struct IPv4UdpAddress)) &&
2057        (plugin->sockv4 == NULL) )
2058     return GNUNET_SYSERR;
2059   if (udpmlen >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
2060   {
2061     GNUNET_break(0);
2062     return GNUNET_SYSERR;
2063   }
2064   if (GNUNET_YES !=
2065       GNUNET_CONTAINER_multipeermap_contains_value (plugin->sessions,
2066                                                     &s->target,
2067                                                     s))
2068   {
2069     GNUNET_break(0);
2070     return GNUNET_SYSERR;
2071   }
2072   LOG (GNUNET_ERROR_TYPE_DEBUG,
2073        "UDP transmits %u-byte message to `%s' using address `%s'\n",
2074        udpmlen,
2075        GNUNET_i2s (&s->target),
2076        udp_address_to_string (NULL,
2077                               s->address->address,
2078                               s->address->address_length));
2079
2080   /* Message */
2081   udp = (struct UDPMessage *) mbuf;
2082   udp->header.size = htons (udpmlen);
2083   udp->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE);
2084   udp->reserved = htonl (0);
2085   udp->sender = *plugin->env->my_identity;
2086
2087   /* We do not update the session time out here!
2088    * Otherwise this session will not timeout since we send keep alive before
2089    * session can timeout
2090    *
2091    * For UDP we update session timeout only on receive, this will cover keep
2092    * alives, since remote peer will reply with keep alive response!
2093    */
2094   if (udpmlen <= UDP_MTU)
2095   {
2096     /* unfragmented message */
2097     udpw = GNUNET_malloc (sizeof (struct UDP_MessageWrapper) + udpmlen);
2098     udpw->session = s;
2099     udpw->msg_buf = (char *) &udpw[1];
2100     udpw->msg_size = udpmlen; /* message size with UDP overhead */
2101     udpw->payload_size = msgbuf_size; /* message size without UDP overhead */
2102     udpw->timeout = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get (), to);
2103     udpw->cont = cont;
2104     udpw->cont_cls = cont_cls;
2105     udpw->frag_ctx = NULL;
2106     udpw->msg_type = UMT_MSG_UNFRAGMENTED;
2107     memcpy (udpw->msg_buf, udp, sizeof(struct UDPMessage));
2108     memcpy (&udpw->msg_buf[sizeof(struct UDPMessage)], msgbuf, msgbuf_size);
2109     enqueue (plugin, udpw);
2110
2111     GNUNET_STATISTICS_update (plugin->env->stats,
2112         "# UDP, unfragmented msgs, messages, attempt", 1, GNUNET_NO);
2113     GNUNET_STATISTICS_update (plugin->env->stats,
2114                               "# UDP, unfragmented msgs, bytes payload, attempt",
2115                               udpw->payload_size,
2116                               GNUNET_NO);
2117   }
2118   else
2119   {
2120     /* fragmented message */
2121     if (s->frag_ctx != NULL)
2122       return GNUNET_SYSERR;
2123     memcpy (&udp[1], msgbuf, msgbuf_size);
2124     frag_ctx = GNUNET_new (struct UDP_FragmentationContext);
2125     frag_ctx->plugin = plugin;
2126     frag_ctx->session = s;
2127     frag_ctx->cont = cont;
2128     frag_ctx->cont_cls = cont_cls;
2129     frag_ctx->timeout = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get (),
2130         to);
2131     frag_ctx->payload_size = msgbuf_size; /* unfragmented message size without UDP overhead */
2132     frag_ctx->on_wire_size = 0; /* bytes with UDP and fragmentation overhead */
2133     frag_ctx->frag = GNUNET_FRAGMENT_context_create (plugin->env->stats,
2134                                                      UDP_MTU,
2135                                                      &plugin->tracker,
2136                                                      s->last_expected_msg_delay,
2137                                                      s->last_expected_ack_delay,
2138                                                      &udp->header,
2139                                                      &enqueue_fragment,
2140                                                      frag_ctx);
2141     s->frag_ctx = frag_ctx;
2142     GNUNET_STATISTICS_update (plugin->env->stats,
2143                               "# UDP, fragmented msgs, messages, pending",
2144                               1,
2145                               GNUNET_NO);
2146     GNUNET_STATISTICS_update (plugin->env->stats,
2147                               "# UDP, fragmented msgs, messages, attempt",
2148                               1,
2149                               GNUNET_NO);
2150     GNUNET_STATISTICS_update (plugin->env->stats,
2151                               "# UDP, fragmented msgs, bytes payload, attempt",
2152                               frag_ctx->payload_size,
2153                               GNUNET_NO);
2154   }
2155   notify_session_monitor (s->plugin,
2156                           s,
2157                           GNUNET_TRANSPORT_SS_UP);
2158   schedule_select (plugin);
2159   return udpmlen;
2160 }
2161
2162
2163 /**
2164  * Our external IP address/port mapping has changed.
2165  *
2166  * @param cls closure, the `struct LocalAddrList`
2167  * @param add_remove #GNUNET_YES to mean the new public IP address, #GNUNET_NO to mean
2168  *     the previous (now invalid) one
2169  * @param addr either the previous or the new public IP address
2170  * @param addrlen actual lenght of the address
2171  */
2172 static void
2173 udp_nat_port_map_callback (void *cls,
2174                            int add_remove,
2175                            const struct sockaddr *addr,
2176                            socklen_t addrlen)
2177 {
2178   struct Plugin *plugin = cls;
2179   struct GNUNET_HELLO_Address *address;
2180   struct IPv4UdpAddress u4;
2181   struct IPv6UdpAddress u6;
2182   void *arg;
2183   size_t args;
2184
2185   LOG (GNUNET_ERROR_TYPE_INFO,
2186        "NAT notification to %s address `%s'\n",
2187        (GNUNET_YES == add_remove) ? "add" : "remove",
2188        GNUNET_a2s (addr, addrlen));
2189
2190   /* convert 'address' to our internal format */
2191   switch (addr->sa_family)
2192   {
2193   case AF_INET:
2194     GNUNET_assert(addrlen == sizeof(struct sockaddr_in));
2195     memset (&u4, 0, sizeof(u4));
2196     u4.options = htonl (plugin->myoptions);
2197     u4.ipv4_addr = ((struct sockaddr_in *) addr)->sin_addr.s_addr;
2198     u4.u4_port = ((struct sockaddr_in *) addr)->sin_port;
2199     if (0 == ((struct sockaddr_in *) addr)->sin_port)
2200       return;
2201     arg = &u4;
2202     args = sizeof(struct IPv4UdpAddress);
2203     break;
2204   case AF_INET6:
2205     GNUNET_assert(addrlen == sizeof(struct sockaddr_in6));
2206     memset (&u6, 0, sizeof(u6));
2207     u6.options = htonl (plugin->myoptions);
2208     if (0 == ((struct sockaddr_in6 *) addr)->sin6_port)
2209       return;
2210     memcpy (&u6.ipv6_addr, &((struct sockaddr_in6 *) addr)->sin6_addr,
2211         sizeof(struct in6_addr));
2212     u6.u6_port = ((struct sockaddr_in6 *) addr)->sin6_port;
2213     arg = &u6;
2214     args = sizeof(struct IPv6UdpAddress);
2215     break;
2216   default:
2217     GNUNET_break(0);
2218     return;
2219   }
2220   /* modify our published address list */
2221   address = GNUNET_HELLO_address_allocate (plugin->env->my_identity,
2222                                            PLUGIN_NAME,
2223                                            arg, args,
2224                                            GNUNET_HELLO_ADDRESS_INFO_NONE);
2225   plugin->env->notify_address (plugin->env->cls, add_remove, address);
2226   GNUNET_HELLO_address_free (address);
2227 }
2228
2229
2230 /**
2231  * Message tokenizer has broken up an incomming message. Pass it on
2232  * to the service.
2233  *
2234  * @param cls the `struct Plugin *`
2235  * @param client the `struct SourceInformation *`
2236  * @param hdr the actual message
2237  * @return #GNUNET_OK (always)
2238  */
2239 static int
2240 process_inbound_tokenized_messages (void *cls,
2241                                     void *client,
2242                                     const struct GNUNET_MessageHeader *hdr)
2243 {
2244   struct Plugin *plugin = cls;
2245   struct SourceInformation *si = client;
2246   struct GNUNET_TIME_Relative delay;
2247
2248   GNUNET_assert(si->session != NULL);
2249   if (GNUNET_YES == si->session->in_destroy)
2250     return GNUNET_OK;
2251   /* setup ATS */
2252   GNUNET_break (ntohl (si->session->ats.value) != GNUNET_ATS_NET_UNSPECIFIED);
2253   reschedule_session_timeout (si->session);
2254   delay = plugin->env->receive (plugin->env->cls,
2255                                 si->session->address,
2256                                 si->session,
2257                                 hdr);
2258   plugin->env->update_address_metrics (plugin->env->cls,
2259                                        si->session->address,
2260                                        si->session,
2261                                        &si->session->ats, 1);
2262   si->session->flow_delay_for_other_peer = delay;
2263   return GNUNET_OK;
2264 }
2265
2266
2267 /**
2268  * We've received a UDP Message.  Process it (pass contents to main service).
2269  *
2270  * @param plugin plugin context
2271  * @param msg the message
2272  * @param sender_addr sender address
2273  * @param sender_addr_len number of bytes in @a sender_addr
2274  */
2275 static void
2276 process_udp_message (struct Plugin *plugin,
2277                      const struct UDPMessage *msg,
2278                      const struct sockaddr *sender_addr,
2279                      socklen_t sender_addr_len)
2280 {
2281   struct SourceInformation si;
2282   struct Session * s;
2283   struct GNUNET_HELLO_Address *address;
2284   struct IPv4UdpAddress u4;
2285   struct IPv6UdpAddress u6;
2286   const void *arg;
2287   size_t args;
2288
2289   if (0 != ntohl (msg->reserved))
2290   {
2291     GNUNET_break_op(0);
2292     return;
2293   }
2294   if (ntohs (msg->header.size)
2295       < sizeof(struct GNUNET_MessageHeader) + sizeof(struct UDPMessage))
2296   {
2297     GNUNET_break_op(0);
2298     return;
2299   }
2300
2301   /* convert address */
2302   switch (sender_addr->sa_family)
2303   {
2304   case AF_INET:
2305     GNUNET_assert(sender_addr_len == sizeof(struct sockaddr_in));
2306     memset (&u4, 0, sizeof(u4));
2307     u6.options = htonl (0);
2308     u4.ipv4_addr = ((struct sockaddr_in *) sender_addr)->sin_addr.s_addr;
2309     u4.u4_port = ((struct sockaddr_in *) sender_addr)->sin_port;
2310     arg = &u4;
2311     args = sizeof(u4);
2312     break;
2313   case AF_INET6:
2314     GNUNET_assert(sender_addr_len == sizeof(struct sockaddr_in6));
2315     memset (&u6, 0, sizeof(u6));
2316     u6.options = htonl (0);
2317     u6.ipv6_addr = ((struct sockaddr_in6 *) sender_addr)->sin6_addr;
2318     u6.u6_port = ((struct sockaddr_in6 *) sender_addr)->sin6_port;
2319     arg = &u6;
2320     args = sizeof(u6);
2321     break;
2322   default:
2323     GNUNET_break(0);
2324     return;
2325   }
2326   LOG(GNUNET_ERROR_TYPE_DEBUG,
2327       "Received message with %u bytes from peer `%s' at `%s'\n",
2328       (unsigned int ) ntohs (msg->header.size), GNUNET_i2s (&msg->sender),
2329       GNUNET_a2s (sender_addr, sender_addr_len));
2330
2331   address = GNUNET_HELLO_address_allocate ( &msg->sender, PLUGIN_NAME,
2332                                             arg, args,
2333                                             GNUNET_HELLO_ADDRESS_INFO_INBOUND);
2334   if (NULL == (s = udp_plugin_lookup_session (plugin, address)))
2335   {
2336     s = udp_plugin_create_session (plugin, address);
2337     plugin->env->session_start (NULL, address, s, NULL, 0);
2338   }
2339   GNUNET_free(address);
2340
2341   /* iterate over all embedded messages */
2342   si.session = s;
2343   si.sender = msg->sender;
2344   si.arg = arg;
2345   si.args = args;
2346   s->rc++;
2347   GNUNET_SERVER_mst_receive (plugin->mst, &si, (const char *) &msg[1],
2348       ntohs (msg->header.size) - sizeof(struct UDPMessage), GNUNET_YES,
2349       GNUNET_NO);
2350   s->rc--;
2351   if ((0 == s->rc) && (GNUNET_YES == s->in_destroy))
2352     free_session (s);
2353 }
2354
2355
2356 /**
2357  * Process a defragmented message.
2358  *
2359  * @param cls the `struct DefragContext *`
2360  * @param msg the message
2361  */
2362 static void
2363 fragment_msg_proc (void *cls,
2364                    const struct GNUNET_MessageHeader *msg)
2365 {
2366   struct DefragContext *rc = cls;
2367
2368   if (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE)
2369   {
2370     GNUNET_break(0);
2371     return;
2372   }
2373   if (ntohs (msg->size) < sizeof(struct UDPMessage))
2374   {
2375     GNUNET_break(0);
2376     return;
2377   }
2378   process_udp_message (rc->plugin,
2379                        (const struct UDPMessage *) msg,
2380                        rc->src_addr,
2381                        rc->addr_len);
2382 }
2383
2384
2385 /**
2386  * Transmit an acknowledgement.
2387  *
2388  * @param cls the `struct DefragContext *`
2389  * @param id message ID (unused)
2390  * @param msg ack to transmit
2391  */
2392 static void
2393 ack_proc (void *cls,
2394           uint32_t id,
2395           const struct GNUNET_MessageHeader *msg)
2396 {
2397   struct DefragContext *rc = cls;
2398   size_t msize = sizeof(struct UDP_ACK_Message) + ntohs (msg->size);
2399   struct UDP_ACK_Message *udp_ack;
2400   uint32_t delay = 0;
2401   struct UDP_MessageWrapper *udpw;
2402   struct Session *s;
2403   struct LookupContext l_ctx;
2404
2405   l_ctx.address = rc->src_addr;
2406   l_ctx.addr_len = rc->addr_len;
2407   l_ctx.must_have_frag_ctx = GNUNET_NO;
2408   l_ctx.res = NULL;
2409   GNUNET_CONTAINER_multipeermap_iterate (rc->plugin->sessions,
2410                                          &lookup_session_by_sockaddr_it,
2411                                          &l_ctx);
2412   s = l_ctx.res;
2413   if (NULL == s)
2414   {
2415     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2416         "Trying to transmit ACK to peer `%s' but not session found!\n",
2417         GNUNET_a2s(rc->src_addr, rc->addr_len));
2418
2419     GNUNET_CONTAINER_heap_remove_node (rc->hnode);
2420     GNUNET_DEFRAGMENT_context_destroy (rc->defrag);
2421     GNUNET_free (rc);
2422
2423     return;
2424   }
2425   if (s->flow_delay_for_other_peer.rel_value_us <= UINT32_MAX)
2426     delay = s->flow_delay_for_other_peer.rel_value_us;
2427
2428   LOG (GNUNET_ERROR_TYPE_DEBUG,
2429        "Sending ACK to `%s' including delay of %s\n",
2430        GNUNET_a2s (rc->src_addr, (rc->src_addr->sa_family == AF_INET) ? sizeof (struct sockaddr_in) : sizeof (struct sockaddr_in6)),
2431        GNUNET_STRINGS_relative_time_to_string (s->flow_delay_for_other_peer,
2432                                                GNUNET_YES));
2433   udpw = GNUNET_malloc (sizeof (struct UDP_MessageWrapper) + msize);
2434   udpw->msg_size = msize;
2435   udpw->payload_size = 0;
2436   udpw->session = s;
2437   udpw->timeout = GNUNET_TIME_UNIT_FOREVER_ABS;
2438   udpw->msg_buf = (char *) &udpw[1];
2439   udpw->msg_type = UMT_MSG_ACK;
2440   udp_ack = (struct UDP_ACK_Message *) udpw->msg_buf;
2441   udp_ack->header.size = htons ((uint16_t) msize);
2442   udp_ack->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK);
2443   udp_ack->delay = htonl (delay);
2444   udp_ack->sender = *rc->plugin->env->my_identity;
2445   memcpy (&udp_ack[1], msg, ntohs (msg->size));
2446   enqueue (rc->plugin, udpw);
2447   notify_session_monitor (s->plugin,
2448                           s,
2449                           GNUNET_TRANSPORT_SS_UP);
2450   schedule_select (rc->plugin);
2451 }
2452
2453
2454 /**
2455  * FIXME.
2456  */
2457 static void
2458 read_process_msg (struct Plugin *plugin,
2459                   const struct GNUNET_MessageHeader *msg,
2460                   const struct sockaddr *addr,
2461                   socklen_t fromlen)
2462 {
2463   if (ntohs (msg->size) < sizeof(struct UDPMessage))
2464   {
2465     GNUNET_break_op(0);
2466     return;
2467   }
2468   process_udp_message (plugin,
2469                        (const struct UDPMessage *) msg,
2470                        addr,
2471                        fromlen);
2472 }
2473
2474
2475 /**
2476  * FIXME.
2477  */
2478 static void
2479 read_process_ack (struct Plugin *plugin,
2480                   const struct GNUNET_MessageHeader *msg,
2481                   const struct sockaddr *addr,
2482                   socklen_t fromlen)
2483 {
2484   const struct GNUNET_MessageHeader *ack;
2485   const struct UDP_ACK_Message *udp_ack;
2486   struct LookupContext l_ctx;
2487   struct Session *s;
2488   struct GNUNET_TIME_Relative flow_delay;
2489
2490   if (ntohs (msg->size)
2491       < sizeof(struct UDP_ACK_Message) + sizeof(struct GNUNET_MessageHeader))
2492   {
2493     GNUNET_break_op(0);
2494     return;
2495   }
2496   udp_ack = (const struct UDP_ACK_Message *) msg;
2497
2498   /* Lookup session based on sockaddr */
2499   l_ctx.address = addr;
2500   l_ctx.addr_len = fromlen;
2501   l_ctx.res = NULL;
2502   l_ctx.must_have_frag_ctx = GNUNET_YES;
2503   GNUNET_CONTAINER_multipeermap_iterate (plugin->sessions,
2504       &lookup_session_by_sockaddr_it, &l_ctx);
2505   s = l_ctx.res;
2506   if ((NULL == s) || (NULL == s->frag_ctx))
2507   {
2508     return;
2509   }
2510
2511   flow_delay.rel_value_us = (uint64_t) ntohl (udp_ack->delay);
2512   LOG(GNUNET_ERROR_TYPE_DEBUG, "We received a sending delay of %s\n",
2513       GNUNET_STRINGS_relative_time_to_string (flow_delay, GNUNET_YES));
2514   s->flow_delay_from_other_peer = GNUNET_TIME_relative_to_absolute (flow_delay);
2515
2516   ack = (const struct GNUNET_MessageHeader *) &udp_ack[1];
2517   if (ntohs (ack->size) != ntohs (msg->size) - sizeof(struct UDP_ACK_Message))
2518   {
2519     GNUNET_break_op(0);
2520     return;
2521   }
2522
2523   if (0
2524       != memcmp (&l_ctx.res->target, &udp_ack->sender,
2525           sizeof(struct GNUNET_PeerIdentity)))
2526     GNUNET_break(0);
2527   if (GNUNET_OK != GNUNET_FRAGMENT_process_ack (s->frag_ctx->frag, ack))
2528   {
2529     LOG(GNUNET_ERROR_TYPE_DEBUG,
2530         "UDP processes %u-byte acknowledgement from `%s' at `%s'\n",
2531         (unsigned int ) ntohs (msg->size), GNUNET_i2s (&udp_ack->sender),
2532         GNUNET_a2s (addr, fromlen));
2533     /* Expect more ACKs to arrive */
2534     return;
2535   }
2536
2537   LOG(GNUNET_ERROR_TYPE_DEBUG, "Message full ACK'ed\n",
2538       (unsigned int ) ntohs (msg->size), GNUNET_i2s (&udp_ack->sender),
2539       GNUNET_a2s (addr, fromlen));
2540
2541   /* Remove fragmented message after successful sending */
2542   fragmented_message_done (s->frag_ctx, GNUNET_OK);
2543 }
2544
2545
2546 /**
2547  * FIXME.
2548  */
2549 static void
2550 read_process_fragment (struct Plugin *plugin,
2551                        const struct GNUNET_MessageHeader *msg,
2552                        const struct sockaddr *addr,
2553                        socklen_t fromlen)
2554 {
2555   struct DefragContext *d_ctx;
2556   struct GNUNET_TIME_Absolute now;
2557   struct FindReceiveContext frc;
2558
2559   frc.rc = NULL;
2560   frc.addr = addr;
2561   frc.addr_len = fromlen;
2562
2563   LOG(GNUNET_ERROR_TYPE_DEBUG, "UDP processes %u-byte fragment from `%s'\n",
2564       (unsigned int ) ntohs (msg->size), GNUNET_a2s (addr, fromlen));
2565   /* Lookup existing receive context for this address */
2566   GNUNET_CONTAINER_heap_iterate (plugin->defrag_ctxs,
2567       &find_receive_context, &frc);
2568   now = GNUNET_TIME_absolute_get ();
2569   d_ctx = frc.rc;
2570
2571   if (d_ctx == NULL )
2572   {
2573     /* Create a new defragmentation context */
2574     d_ctx = GNUNET_malloc (sizeof (struct DefragContext) + fromlen);
2575     memcpy (&d_ctx[1], addr, fromlen);
2576     d_ctx->src_addr = (const struct sockaddr *) &d_ctx[1];
2577     d_ctx->addr_len = fromlen;
2578     d_ctx->plugin = plugin;
2579     d_ctx->defrag = GNUNET_DEFRAGMENT_context_create (plugin->env->stats,
2580         UDP_MTU, UDP_MAX_MESSAGES_IN_DEFRAG, d_ctx, &fragment_msg_proc,
2581         &ack_proc);
2582     d_ctx->hnode = GNUNET_CONTAINER_heap_insert (plugin->defrag_ctxs, d_ctx,
2583         (GNUNET_CONTAINER_HeapCostType) now.abs_value_us);
2584     LOG(GNUNET_ERROR_TYPE_DEBUG,
2585         "Created new defragmentation context for %u-byte fragment from `%s'\n",
2586         (unsigned int ) ntohs (msg->size), GNUNET_a2s (addr, fromlen));
2587   }
2588   else
2589   {
2590     LOG(GNUNET_ERROR_TYPE_DEBUG,
2591         "Found existing defragmentation context for %u-byte fragment from `%s'\n",
2592         (unsigned int ) ntohs (msg->size), GNUNET_a2s (addr, fromlen));
2593   }
2594
2595   if (GNUNET_OK == GNUNET_DEFRAGMENT_process_fragment (d_ctx->defrag, msg))
2596   {
2597     /* keep this 'rc' from expiring */
2598     GNUNET_CONTAINER_heap_update_cost (plugin->defrag_ctxs, d_ctx->hnode,
2599         (GNUNET_CONTAINER_HeapCostType) now.abs_value_us);
2600   }
2601   if (GNUNET_CONTAINER_heap_get_size (plugin->defrag_ctxs) >
2602   UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG)
2603   {
2604     /* remove 'rc' that was inactive the longest */
2605     d_ctx = GNUNET_CONTAINER_heap_remove_root (plugin->defrag_ctxs);
2606     GNUNET_assert(NULL != d_ctx);
2607     GNUNET_DEFRAGMENT_context_destroy (d_ctx->defrag);
2608     GNUNET_free(d_ctx);
2609   }
2610 }
2611
2612
2613 /**
2614  * Read and process a message from the given socket.
2615  *
2616  * @param plugin the overall plugin
2617  * @param rsock socket to read from
2618  */
2619 static void
2620 udp_select_read (struct Plugin *plugin,
2621                  struct GNUNET_NETWORK_Handle *rsock)
2622 {
2623   socklen_t fromlen;
2624   struct sockaddr_storage addr;
2625   char buf[65536] GNUNET_ALIGN;
2626   ssize_t size;
2627   const struct GNUNET_MessageHeader *msg;
2628
2629   fromlen = sizeof(addr);
2630   memset (&addr, 0, sizeof(addr));
2631   size = GNUNET_NETWORK_socket_recvfrom (rsock, buf, sizeof(buf),
2632       (struct sockaddr *) &addr, &fromlen);
2633 #if MINGW
2634   /* On SOCK_DGRAM UDP sockets recvfrom might fail with a
2635    * WSAECONNRESET error to indicate that previous sendto() (yes, sendto!)
2636    * on this socket has failed.
2637    * Quote from MSDN:
2638    *   WSAECONNRESET - The virtual circuit was reset by the remote side
2639    *   executing a hard or abortive close. The application should close
2640    *   the socket; it is no longer usable. On a UDP-datagram socket this
2641    *   error indicates a previous send operation resulted in an ICMP Port
2642    *   Unreachable message.
2643    */
2644   if ( (-1 == size) && (ECONNRESET == errno) )
2645   return;
2646 #endif
2647   if (-1 == size)
2648   {
2649     LOG(GNUNET_ERROR_TYPE_DEBUG, "UDP failed to receive data: %s\n",
2650         STRERROR (errno));
2651     /* Connection failure or something. Not a protocol violation. */
2652     return;
2653   }
2654   if (size < sizeof(struct GNUNET_MessageHeader))
2655   {
2656     LOG(GNUNET_ERROR_TYPE_WARNING,
2657         "UDP got %u bytes, which is not enough for a GNUnet message header\n",
2658         (unsigned int ) size);
2659     /* _MAY_ be a connection failure (got partial message) */
2660     /* But it _MAY_ also be that the other side uses non-GNUnet protocol. */
2661     GNUNET_break_op(0);
2662     return;
2663   }
2664   msg = (const struct GNUNET_MessageHeader *) buf;
2665
2666   LOG(GNUNET_ERROR_TYPE_DEBUG,
2667       "UDP received %u-byte message from `%s' type %u\n", (unsigned int ) size,
2668       GNUNET_a2s ((const struct sockaddr * ) &addr, fromlen),
2669       ntohs (msg->type));
2670
2671   if (size != ntohs (msg->size))
2672   {
2673     GNUNET_break_op(0);
2674     return;
2675   }
2676   GNUNET_STATISTICS_update (plugin->env->stats,
2677                             "# UDP, total, bytes, received",
2678                             size,
2679                             GNUNET_NO);
2680
2681   switch (ntohs (msg->type))
2682   {
2683   case GNUNET_MESSAGE_TYPE_TRANSPORT_BROADCAST_BEACON:
2684     if (GNUNET_YES == plugin->enable_broadcasting_receiving)
2685       udp_broadcast_receive (plugin, buf, size, (const struct sockaddr *) &addr,
2686           fromlen);
2687     return;
2688   case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE:
2689     read_process_msg (plugin, msg, (const struct sockaddr *) &addr, fromlen);
2690     return;
2691   case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK:
2692     read_process_ack (plugin, msg, (const struct sockaddr *) &addr, fromlen);
2693     return;
2694   case GNUNET_MESSAGE_TYPE_FRAGMENT:
2695     read_process_fragment (plugin, msg, (const struct sockaddr *) &addr,
2696         fromlen);
2697     return;
2698   default:
2699     GNUNET_break_op(0);
2700     return;
2701   }
2702 }
2703
2704
2705 /**
2706  * FIXME.
2707  */
2708 static struct UDP_MessageWrapper *
2709 remove_timeout_messages_and_select (struct UDP_MessageWrapper *head,
2710                                     struct GNUNET_NETWORK_Handle *sock)
2711 {
2712   struct UDP_MessageWrapper *udpw = NULL;
2713   struct GNUNET_TIME_Relative remaining;
2714   struct Session *session;
2715   struct Plugin *plugin;
2716   int removed;
2717
2718   removed = GNUNET_NO;
2719   udpw = head;
2720   while (NULL != udpw)
2721   {
2722     session = udpw->session;
2723     plugin = session->plugin;
2724     /* Find messages with timeout */
2725     remaining = GNUNET_TIME_absolute_get_remaining (udpw->timeout);
2726     if (GNUNET_TIME_UNIT_ZERO.rel_value_us == remaining.rel_value_us)
2727     {
2728       /* Message timed out */
2729       switch (udpw->msg_type)
2730       {
2731       case UMT_MSG_UNFRAGMENTED:
2732         GNUNET_STATISTICS_update (plugin->env->stats,
2733                                   "# UDP, total, bytes, sent, timeout",
2734                                   udpw->msg_size,
2735                                   GNUNET_NO);
2736         GNUNET_STATISTICS_update (plugin->env->stats,
2737                                   "# UDP, total, messages, sent, timeout",
2738                                   1,
2739                                   GNUNET_NO);
2740         GNUNET_STATISTICS_update (plugin->env->stats,
2741                                   "# UDP, unfragmented msgs, messages, sent, timeout",
2742                                   1,
2743                                   GNUNET_NO);
2744         GNUNET_STATISTICS_update (plugin->env->stats,
2745                                   "# UDP, unfragmented msgs, bytes, sent, timeout",
2746                                   udpw->payload_size,
2747                                   GNUNET_NO);
2748         /* Not fragmented message */
2749         LOG (GNUNET_ERROR_TYPE_DEBUG,
2750              "Message for peer `%s' with size %u timed out\n",
2751              GNUNET_i2s (&udpw->session->target),
2752              udpw->payload_size);
2753         call_continuation (udpw, GNUNET_SYSERR);
2754         /* Remove message */
2755         removed = GNUNET_YES;
2756         dequeue (plugin, udpw);
2757         GNUNET_free(udpw);
2758         break;
2759       case UMT_MSG_FRAGMENTED:
2760         /* Fragmented message */
2761         GNUNET_STATISTICS_update (plugin->env->stats,
2762                                   "# UDP, total, bytes, sent, timeout",
2763                                   udpw->frag_ctx->on_wire_size,
2764                                   GNUNET_NO);
2765         GNUNET_STATISTICS_update (plugin->env->stats,
2766                                   "# UDP, total, messages, sent, timeout",
2767                                   1,
2768                                   GNUNET_NO);
2769         call_continuation (udpw, GNUNET_SYSERR);
2770         LOG (GNUNET_ERROR_TYPE_DEBUG,
2771              "Fragment for message for peer `%s' with size %u timed out\n",
2772              GNUNET_i2s (&udpw->session->target),
2773             udpw->frag_ctx->payload_size);
2774
2775         GNUNET_STATISTICS_update (plugin->env->stats,
2776                                   "# UDP, fragmented msgs, messages, sent, timeout",
2777                                   1,
2778                                   GNUNET_NO);
2779         GNUNET_STATISTICS_update (plugin->env->stats,
2780                                   "# UDP, fragmented msgs, bytes, sent, timeout",
2781                                   udpw->frag_ctx->payload_size,
2782                                   GNUNET_NO);
2783         /* Remove fragmented message due to timeout */
2784         fragmented_message_done (udpw->frag_ctx, GNUNET_SYSERR);
2785         break;
2786       case UMT_MSG_ACK:
2787         GNUNET_STATISTICS_update (plugin->env->stats,
2788                                   "# UDP, total, bytes, sent, timeout",
2789                                   udpw->msg_size,
2790                                   GNUNET_NO);
2791         GNUNET_STATISTICS_update (plugin->env->stats,
2792                                   "# UDP, total, messages, sent, timeout",
2793                                   1,
2794                                   GNUNET_NO);
2795         LOG (GNUNET_ERROR_TYPE_DEBUG,
2796              "ACK Message for peer `%s' with size %u timed out\n",
2797              GNUNET_i2s (&udpw->session->target),
2798              udpw->payload_size);
2799         call_continuation (udpw, GNUNET_SYSERR);
2800         removed = GNUNET_YES;
2801         dequeue (plugin, udpw);
2802         GNUNET_free(udpw);
2803         break;
2804       default:
2805         break;
2806       }
2807       if (sock == plugin->sockv4)
2808         udpw = plugin->ipv4_queue_head;
2809       else if (sock == plugin->sockv6)
2810         udpw = plugin->ipv6_queue_head;
2811       else
2812       {
2813         GNUNET_break(0); /* should never happen */
2814         udpw = NULL;
2815       }
2816       GNUNET_STATISTICS_update (plugin->env->stats,
2817                                 "# messages discarded due to timeout",
2818                                 1,
2819                                 GNUNET_NO);
2820     }
2821     else
2822     {
2823       /* Message did not time out, check flow delay */
2824       remaining = GNUNET_TIME_absolute_get_remaining (udpw->session->flow_delay_from_other_peer);
2825       if (GNUNET_TIME_UNIT_ZERO.rel_value_us == remaining.rel_value_us)
2826       {
2827         /* this message is not delayed */
2828         LOG (GNUNET_ERROR_TYPE_DEBUG,
2829              "Message for peer `%s' (%u bytes) is not delayed \n",
2830              GNUNET_i2s (&udpw->session->target),
2831              udpw->payload_size);
2832         break; /* Found message to send, break */
2833       }
2834       else
2835       {
2836         /* Message is delayed, try next */
2837         LOG (GNUNET_ERROR_TYPE_DEBUG,
2838              "Message for peer `%s' (%u bytes) is delayed for %s\n",
2839              GNUNET_i2s (&udpw->session->target), udpw->payload_size,
2840              GNUNET_STRINGS_relative_time_to_string (remaining, GNUNET_YES));
2841         udpw = udpw->next;
2842       }
2843     }
2844   }
2845   if (GNUNET_YES == removed)
2846     notify_session_monitor (session->plugin,
2847                             session,
2848                             GNUNET_TRANSPORT_SS_UP);
2849   return udpw;
2850 }
2851
2852
2853 /**
2854  * FIXME.
2855  */
2856 static void
2857 analyze_send_error (struct Plugin *plugin,
2858                     const struct sockaddr *sa,
2859                     socklen_t slen, int error)
2860 {
2861   struct GNUNET_ATS_Information type;
2862
2863   type = plugin->env->get_address_type (plugin->env->cls, sa, slen);
2864   if (((GNUNET_ATS_NET_LAN == ntohl (type.value))
2865       || (GNUNET_ATS_NET_WAN == ntohl (type.value)))
2866       && ((ENETUNREACH == errno)|| (ENETDOWN == errno)))
2867   {
2868     if (slen == sizeof (struct sockaddr_in))
2869     {
2870       /* IPv4: "Network unreachable" or "Network down"
2871        *
2872        * This indicates we do not have connectivity
2873        */
2874       LOG (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK,
2875            _("UDP could not transmit message to `%s': "
2876              "Network seems down, please check your network configuration\n"),
2877            GNUNET_a2s (sa, slen));
2878     }
2879     if (slen == sizeof (struct sockaddr_in6))
2880     {
2881       /* IPv6: "Network unreachable" or "Network down"
2882        *
2883        * This indicates that this system is IPv6 enabled, but does not
2884        * have a valid global IPv6 address assigned or we do not have
2885        * connectivity
2886        */
2887       LOG (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK,
2888            _("UDP could not transmit IPv6 message! "
2889              "Please check your network configuration and disable IPv6 if your "
2890              "connection does not have a global IPv6 address\n"));
2891     }
2892   }
2893   else
2894   {
2895     LOG (GNUNET_ERROR_TYPE_WARNING,
2896          "UDP could not transmit message to `%s': `%s'\n",
2897          GNUNET_a2s (sa, slen), STRERROR (error));
2898   }
2899 }
2900
2901
2902 /**
2903  * FIXME.
2904  */
2905 static size_t
2906 udp_select_send (struct Plugin *plugin,
2907                  struct GNUNET_NETWORK_Handle *sock)
2908 {
2909   ssize_t sent;
2910   socklen_t slen;
2911   struct sockaddr *a;
2912   const struct IPv4UdpAddress *u4;
2913   struct sockaddr_in a4;
2914   const struct IPv6UdpAddress *u6;
2915   struct sockaddr_in6 a6;
2916   struct UDP_MessageWrapper *udpw;
2917
2918   /* Find message to send */
2919   udpw = remove_timeout_messages_and_select ((sock == plugin->sockv4)
2920                                              ? plugin->ipv4_queue_head
2921                                              : plugin->ipv6_queue_head,
2922                                              sock);
2923   if (NULL == udpw)
2924     return 0; /* No message to send */
2925
2926   if (sizeof (struct IPv4UdpAddress) == udpw->session->address->address_length)
2927   {
2928     u4 = udpw->session->address->address;
2929     memset (&a4, 0, sizeof(a4));
2930     a4.sin_family = AF_INET;
2931 #if HAVE_SOCKADDR_IN_SIN_LEN
2932     a4.sin_len = sizeof (a4);
2933 #endif
2934     a4.sin_port = u4->u4_port;
2935     memcpy (&a4.sin_addr, &u4->ipv4_addr, sizeof(struct in_addr));
2936     a = (struct sockaddr *) &a4;
2937     slen = sizeof (a4);
2938   }
2939   else if (sizeof (struct IPv6UdpAddress) == udpw->session->address->address_length)
2940   {
2941     u6 = udpw->session->address->address;
2942     memset (&a6, 0, sizeof(a6));
2943     a6.sin6_family = AF_INET6;
2944 #if HAVE_SOCKADDR_IN_SIN_LEN
2945     a6.sin6_len = sizeof (a6);
2946 #endif
2947     a6.sin6_port = u6->u6_port;
2948     memcpy (&a6.sin6_addr, &u6->ipv6_addr, sizeof(struct in6_addr));
2949     a = (struct sockaddr *) &a6;
2950     slen = sizeof (a6);
2951   }
2952   else
2953   {
2954     call_continuation (udpw, GNUNET_OK);
2955     dequeue (plugin, udpw);
2956     notify_session_monitor (plugin,
2957                             udpw->session,
2958                             GNUNET_TRANSPORT_SS_UP);
2959     GNUNET_free (udpw);
2960     return GNUNET_SYSERR;
2961   }
2962
2963   sent = GNUNET_NETWORK_socket_sendto (sock,
2964                                        udpw->msg_buf,
2965                                        udpw->msg_size,
2966                                        a,
2967                                        slen);
2968   if (GNUNET_SYSERR == sent)
2969   {
2970     /* Failure */
2971     analyze_send_error (plugin, a, slen, errno);
2972     call_continuation (udpw, GNUNET_SYSERR);
2973     GNUNET_STATISTICS_update (plugin->env->stats,
2974         "# UDP, total, bytes, sent, failure", sent, GNUNET_NO);
2975     GNUNET_STATISTICS_update (plugin->env->stats,
2976         "# UDP, total, messages, sent, failure", 1, GNUNET_NO);
2977   }
2978   else
2979   {
2980     /* Success */
2981     LOG(GNUNET_ERROR_TYPE_DEBUG,
2982         "UDP transmitted %u-byte message to  `%s' `%s' (%d: %s)\n",
2983         (unsigned int ) (udpw->msg_size), GNUNET_i2s (&udpw->session->target),
2984         GNUNET_a2s (a, slen), (int ) sent,
2985         (sent < 0) ? STRERROR (errno) : "ok");
2986     GNUNET_STATISTICS_update (plugin->env->stats,
2987         "# UDP, total, bytes, sent, success", sent, GNUNET_NO);
2988     GNUNET_STATISTICS_update (plugin->env->stats,
2989         "# UDP, total, messages, sent, success", 1, GNUNET_NO);
2990     if (NULL != udpw->frag_ctx)
2991       udpw->frag_ctx->on_wire_size += udpw->msg_size;
2992     call_continuation (udpw, GNUNET_OK);
2993   }
2994   dequeue (plugin, udpw);
2995   notify_session_monitor (plugin,
2996                           udpw->session,
2997                           GNUNET_TRANSPORT_SS_UP);
2998   GNUNET_free(udpw);
2999   return sent;
3000 }
3001
3002
3003 /**
3004  * We have been notified that our readset has something to read.  We don't
3005  * know which socket needs to be read, so we have to check each one
3006  * Then reschedule this function to be called again once more is available.
3007  *
3008  * @param cls the plugin handle
3009  * @param tc the scheduling context (for rescheduling this function again)
3010  */
3011 static void
3012 udp_plugin_select (void *cls,
3013                    const struct GNUNET_SCHEDULER_TaskContext *tc)
3014 {
3015   struct Plugin *plugin = cls;
3016
3017   plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
3018   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
3019     return;
3020   if ((0 != (tc->reason & GNUNET_SCHEDULER_REASON_READ_READY))
3021       && (NULL != plugin->sockv4)
3022       && (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv4)))
3023     udp_select_read (plugin, plugin->sockv4);
3024   if ((0 != (tc->reason & GNUNET_SCHEDULER_REASON_WRITE_READY))
3025       && (NULL != plugin->sockv4) && (NULL != plugin->ipv4_queue_head)
3026       && (GNUNET_NETWORK_fdset_isset (tc->write_ready, plugin->sockv4)))
3027     udp_select_send (plugin, plugin->sockv4);
3028   schedule_select (plugin);
3029 }
3030
3031
3032 /**
3033  * We have been notified that our readset has something to read.  We don't
3034  * know which socket needs to be read, so we have to check each one
3035  * Then reschedule this function to be called again once more is available.
3036  *
3037  * @param cls the plugin handle
3038  * @param tc the scheduling context (for rescheduling this function again)
3039  */
3040 static void
3041 udp_plugin_select_v6 (void *cls,
3042                       const struct GNUNET_SCHEDULER_TaskContext *tc)
3043 {
3044   struct Plugin *plugin = cls;
3045
3046   plugin->select_task_v6 = GNUNET_SCHEDULER_NO_TASK;
3047   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
3048     return;
3049   if (((tc->reason & GNUNET_SCHEDULER_REASON_READ_READY) != 0)
3050       && (NULL != plugin->sockv6)
3051       && (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv6)))
3052     udp_select_read (plugin, plugin->sockv6);
3053   if ((0 != (tc->reason & GNUNET_SCHEDULER_REASON_WRITE_READY))
3054       && (NULL != plugin->sockv6) && (plugin->ipv6_queue_head != NULL )&&
3055       (GNUNET_NETWORK_fdset_isset (tc->write_ready, plugin->sockv6)) )udp_select_send (plugin, plugin->sockv6);
3056   schedule_select (plugin);
3057 }
3058
3059
3060 /**
3061  * Setup the UDP sockets (for IPv4 and IPv6) for the plugin.
3062  *
3063  * @param plugin the plugin to initialize
3064  * @param bind_v6 IPv6 address to bind to (can be NULL, for 'any')
3065  * @param bind_v4 IPv4 address to bind to (can be NULL, for 'any')
3066  * @return number of sockets that were successfully bound
3067  */
3068 static int
3069 setup_sockets (struct Plugin *plugin,
3070                const struct sockaddr_in6 *bind_v6,
3071                const struct sockaddr_in *bind_v4)
3072 {
3073   int tries;
3074   int sockets_created = 0;
3075   struct sockaddr_in6 server_addrv6;
3076   struct sockaddr_in server_addrv4;
3077   struct sockaddr *server_addr;
3078   struct sockaddr *addrs[2];
3079   socklen_t addrlens[2];
3080   socklen_t addrlen;
3081   int eno;
3082
3083   /* Create IPv6 socket */
3084   eno = EINVAL;
3085   if (GNUNET_YES == plugin->enable_ipv6)
3086   {
3087     plugin->sockv6 = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_DGRAM, 0);
3088     if (NULL == plugin->sockv6)
3089     {
3090       LOG(GNUNET_ERROR_TYPE_WARNING,
3091           "Disabling IPv6 since it is not supported on this system!\n");
3092       plugin->enable_ipv6 = GNUNET_NO;
3093     }
3094     else
3095     {
3096       memset (&server_addrv6, '\0', sizeof(struct sockaddr_in6));
3097 #if HAVE_SOCKADDR_IN_SIN_LEN
3098       server_addrv6.sin6_len = sizeof (struct sockaddr_in6);
3099 #endif
3100       server_addrv6.sin6_family = AF_INET6;
3101       if (NULL != bind_v6)
3102         server_addrv6.sin6_addr = bind_v6->sin6_addr;
3103       else
3104         server_addrv6.sin6_addr = in6addr_any;
3105
3106       if (0 == plugin->port) /* autodetect */
3107         server_addrv6.sin6_port = htons (
3108             GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537)
3109                 + 32000);
3110       else
3111         server_addrv6.sin6_port = htons (plugin->port);
3112       addrlen = sizeof(struct sockaddr_in6);
3113       server_addr = (struct sockaddr *) &server_addrv6;
3114
3115       tries = 0;
3116       while (tries < 10)
3117       {
3118         LOG(GNUNET_ERROR_TYPE_DEBUG,
3119             "Binding to IPv6 `%s'\n",
3120             GNUNET_a2s (server_addr, addrlen));
3121         /* binding */
3122         if (GNUNET_OK
3123             == GNUNET_NETWORK_socket_bind (plugin->sockv6, server_addr,
3124                 addrlen))
3125           break;
3126         eno = errno;
3127         if (0 != plugin->port)
3128         {
3129           tries = 10; /* fail */
3130           break; /* bind failed on specific port */
3131         }
3132         /* autodetect */
3133         server_addrv6.sin6_port = htons (
3134             GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537)
3135                 + 32000);
3136         tries++;
3137       }
3138       if (tries >= 10)
3139       {
3140         GNUNET_NETWORK_socket_close (plugin->sockv6);
3141         plugin->enable_ipv6 = GNUNET_NO;
3142         plugin->sockv6 = NULL;
3143       }
3144
3145       if (plugin->sockv6 != NULL )
3146       {
3147         LOG (GNUNET_ERROR_TYPE_DEBUG,
3148              "IPv6 socket created on port %s\n",
3149              GNUNET_a2s (server_addr, addrlen));
3150         addrs[sockets_created] = (struct sockaddr *) &server_addrv6;
3151         addrlens[sockets_created] = sizeof(struct sockaddr_in6);
3152         sockets_created++;
3153       }
3154       else
3155       {
3156         LOG (GNUNET_ERROR_TYPE_ERROR,
3157              "Failed to bind UDP socket to %s: %s\n",
3158              GNUNET_a2s (server_addr, addrlen),
3159              STRERROR (eno));
3160       }
3161     }
3162   }
3163
3164   /* Create IPv4 socket */
3165   eno = EINVAL;
3166   plugin->sockv4 = GNUNET_NETWORK_socket_create (PF_INET, SOCK_DGRAM, 0);
3167   if (NULL == plugin->sockv4)
3168   {
3169     GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING,
3170                          "socket");
3171     LOG(GNUNET_ERROR_TYPE_WARNING,
3172         "Disabling IPv4 since it is not supported on this system!\n");
3173     plugin->enable_ipv4 = GNUNET_NO;
3174   }
3175   else
3176   {
3177     memset (&server_addrv4, '\0', sizeof(struct sockaddr_in));
3178 #if HAVE_SOCKADDR_IN_SIN_LEN
3179     server_addrv4.sin_len = sizeof (struct sockaddr_in);
3180 #endif
3181     server_addrv4.sin_family = AF_INET;
3182     if (NULL != bind_v4)
3183       server_addrv4.sin_addr = bind_v4->sin_addr;
3184     else
3185       server_addrv4.sin_addr.s_addr = INADDR_ANY;
3186
3187     if (0 == plugin->port)
3188       /* autodetect */
3189       server_addrv4.sin_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG,
3190                                                                 33537)
3191                                       + 32000);
3192     else
3193       server_addrv4.sin_port = htons (plugin->port);
3194
3195     addrlen = sizeof(struct sockaddr_in);
3196     server_addr = (struct sockaddr *) &server_addrv4;
3197
3198     tries = 0;
3199     while (tries < 10)
3200     {
3201       LOG (GNUNET_ERROR_TYPE_DEBUG,
3202            "Binding to IPv4 `%s'\n",
3203            GNUNET_a2s (server_addr, addrlen));
3204
3205       /* binding */
3206       if (GNUNET_OK
3207           == GNUNET_NETWORK_socket_bind (plugin->sockv4, server_addr, addrlen))
3208         break;
3209       eno = errno;
3210       if (0 != plugin->port)
3211       {
3212         tries = 10; /* fail */
3213         break; /* bind failed on specific port */
3214       }
3215
3216       /* autodetect */
3217       server_addrv4.sin_port = htons (
3218           GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537)
3219               + 32000);
3220       tries++;
3221     }
3222
3223     if (tries >= 10)
3224     {
3225       GNUNET_NETWORK_socket_close (plugin->sockv4);
3226       plugin->enable_ipv4 = GNUNET_NO;
3227       plugin->sockv4 = NULL;
3228     }
3229
3230     if (NULL != plugin->sockv4)
3231     {
3232       LOG(GNUNET_ERROR_TYPE_DEBUG, "IPv4 socket created on port %s\n",
3233           GNUNET_a2s (server_addr, addrlen));
3234       addrs[sockets_created] = (struct sockaddr *) &server_addrv4;
3235       addrlens[sockets_created] = sizeof(struct sockaddr_in);
3236       sockets_created++;
3237     }
3238     else
3239     {
3240       LOG (GNUNET_ERROR_TYPE_ERROR,
3241            _("Failed to bind UDP socket to %s: %s\n"),
3242            GNUNET_a2s (server_addr, addrlen),
3243            STRERROR (eno));
3244     }
3245   }
3246
3247   if (0 == sockets_created)
3248   {
3249     LOG(GNUNET_ERROR_TYPE_WARNING, _("Failed to open UDP sockets\n"));
3250     return 0; /* No sockets created, return */
3251   }
3252
3253   /* Create file descriptors */
3254   if (plugin->enable_ipv4 == GNUNET_YES)
3255   {
3256     plugin->rs_v4 = GNUNET_NETWORK_fdset_create ();
3257     plugin->ws_v4 = GNUNET_NETWORK_fdset_create ();
3258     GNUNET_NETWORK_fdset_zero (plugin->rs_v4);
3259     GNUNET_NETWORK_fdset_zero (plugin->ws_v4);
3260     if (NULL != plugin->sockv4)
3261     {
3262       GNUNET_NETWORK_fdset_set (plugin->rs_v4, plugin->sockv4);
3263       GNUNET_NETWORK_fdset_set (plugin->ws_v4, plugin->sockv4);
3264     }
3265   }
3266
3267   if (plugin->enable_ipv6 == GNUNET_YES)
3268   {
3269     plugin->rs_v6 = GNUNET_NETWORK_fdset_create ();
3270     plugin->ws_v6 = GNUNET_NETWORK_fdset_create ();
3271     GNUNET_NETWORK_fdset_zero (plugin->rs_v6);
3272     GNUNET_NETWORK_fdset_zero (plugin->ws_v6);
3273     if (NULL != plugin->sockv6)
3274     {
3275       GNUNET_NETWORK_fdset_set (plugin->rs_v6, plugin->sockv6);
3276       GNUNET_NETWORK_fdset_set (plugin->ws_v6, plugin->sockv6);
3277     }
3278   }
3279
3280   schedule_select (plugin);
3281   plugin->nat = GNUNET_NAT_register (plugin->env->cfg,
3282                                      GNUNET_NO,
3283                                      plugin->port,
3284                                      sockets_created,
3285                                      (const struct sockaddr **) addrs,
3286                                      addrlens,
3287                                      &udp_nat_port_map_callback,
3288                                      NULL,
3289                                      plugin);
3290
3291   return sockets_created;
3292 }
3293
3294
3295 /**
3296  * Return information about the given session to the
3297  * monitor callback.
3298  *
3299  * @param cls the `struct Plugin` with the monitor callback (`sic`)
3300  * @param peer peer we send information about
3301  * @param value our `struct Session` to send information about
3302  * @return #GNUNET_OK (continue to iterate)
3303  */
3304 static int
3305 send_session_info_iter (void *cls,
3306                         const struct GNUNET_PeerIdentity *peer,
3307                         void *value)
3308 {
3309   struct Plugin *plugin = cls;
3310   struct Session *session = value;
3311
3312   notify_session_monitor (plugin,
3313                           session,
3314                           GNUNET_TRANSPORT_SS_UP);
3315   return GNUNET_OK;
3316 }
3317
3318
3319 /**
3320  * Begin monitoring sessions of a plugin.  There can only
3321  * be one active monitor per plugin (i.e. if there are
3322  * multiple monitors, the transport service needs to
3323  * multiplex the generated events over all of them).
3324  *
3325  * @param cls closure of the plugin
3326  * @param sic callback to invoke, NULL to disable monitor;
3327  *            plugin will being by iterating over all active
3328  *            sessions immediately and then enter monitor mode
3329  * @param sic_cls closure for @a sic
3330  */
3331 static void
3332 udp_plugin_setup_monitor (void *cls,
3333                           GNUNET_TRANSPORT_SessionInfoCallback sic,
3334                           void *sic_cls)
3335 {
3336   struct Plugin *plugin = cls;
3337
3338   plugin->sic = sic;
3339   plugin->sic_cls = sic_cls;
3340   if (NULL != sic)
3341   {
3342     GNUNET_CONTAINER_multipeermap_iterate (plugin->sessions,
3343                                            &send_session_info_iter,
3344                                            plugin);
3345     /* signal end of first iteration */
3346     sic (sic_cls, NULL, NULL);
3347   }
3348 }
3349
3350
3351 /**
3352  * The exported method. Makes the core api available via a global and
3353  * returns the udp transport API.
3354  *
3355  * @param cls our `struct GNUNET_TRANSPORT_PluginEnvironment`
3356  * @return our `struct GNUNET_TRANSPORT_PluginFunctions`
3357  */
3358 void *
3359 libgnunet_plugin_transport_udp_init (void *cls)
3360 {
3361   struct GNUNET_TRANSPORT_PluginEnvironment *env = cls;
3362   struct GNUNET_TRANSPORT_PluginFunctions *api;
3363   struct Plugin *p;
3364   unsigned long long port;
3365   unsigned long long aport;
3366   unsigned long long udp_max_bps;
3367   unsigned long long enable_v6;
3368   unsigned long long enable_broadcasting;
3369   unsigned long long enable_broadcasting_recv;
3370   char *bind4_address;
3371   char *bind6_address;
3372   char *fancy_interval;
3373   struct GNUNET_TIME_Relative interval;
3374   struct sockaddr_in server_addrv4;
3375   struct sockaddr_in6 server_addrv6;
3376   int res;
3377   int have_bind4;
3378   int have_bind6;
3379
3380   if (NULL == env->receive)
3381   {
3382     /* run in 'stub' mode (i.e. as part of gnunet-peerinfo), don't fully
3383      initialze the plugin or the API */
3384     api = GNUNET_new (struct GNUNET_TRANSPORT_PluginFunctions);
3385     api->cls = NULL;
3386     api->address_pretty_printer = &udp_plugin_address_pretty_printer;
3387     api->address_to_string = &udp_address_to_string;
3388     api->string_to_address = &udp_string_to_address;
3389     return api;
3390   }
3391
3392   /* Get port number: port == 0 : autodetect a port,
3393    * > 0 : use this port, not given : 2086 default */
3394   if (GNUNET_OK !=
3395       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp",
3396                                              "PORT", &port))
3397     port = 2086;
3398   if (GNUNET_OK !=
3399       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp",
3400                                              "ADVERTISED_PORT", &aport))
3401     aport = port;
3402   if (port > 65535)
3403   {
3404     LOG (GNUNET_ERROR_TYPE_WARNING,
3405          _("Given `%s' option is out of range: %llu > %u\n"),
3406          "PORT", port,
3407          65535);
3408     return NULL;
3409   }
3410
3411   /* Protocols */
3412   if (GNUNET_YES ==
3413       GNUNET_CONFIGURATION_get_value_yesno (env->cfg, "nat", "DISABLEV6"))
3414     enable_v6 = GNUNET_NO;
3415   else
3416     enable_v6 = GNUNET_YES;
3417
3418   /* Addresses */
3419   have_bind4 = GNUNET_NO;
3420   memset (&server_addrv4, 0, sizeof(server_addrv4));
3421   if (GNUNET_YES ==
3422       GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp",
3423                                              "BINDTO", &bind4_address))
3424   {
3425     LOG (GNUNET_ERROR_TYPE_DEBUG,
3426          "Binding udp plugin to specific address: `%s'\n",
3427          bind4_address);
3428     if (1 != inet_pton (AF_INET,
3429                         bind4_address,
3430                         &server_addrv4.sin_addr))
3431     {
3432       GNUNET_free (bind4_address);
3433       return NULL;
3434     }
3435     have_bind4 = GNUNET_YES;
3436   }
3437   GNUNET_free_non_null(bind4_address);
3438   have_bind6 = GNUNET_NO;
3439   memset (&server_addrv6, 0, sizeof(server_addrv6));
3440   if (GNUNET_YES ==
3441       GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp",
3442                                              "BINDTO6", &bind6_address))
3443   {
3444     LOG (GNUNET_ERROR_TYPE_DEBUG,
3445          "Binding udp plugin to specific address: `%s'\n",
3446          bind6_address);
3447     if (1 != inet_pton (AF_INET6,
3448                         bind6_address,
3449                         &server_addrv6.sin6_addr))
3450     {
3451       LOG (GNUNET_ERROR_TYPE_ERROR,
3452            _("Invalid IPv6 address: `%s'\n"),
3453            bind6_address);
3454       GNUNET_free (bind6_address);
3455       return NULL;
3456     }
3457     have_bind6 = GNUNET_YES;
3458   }
3459   GNUNET_free_non_null (bind6_address);
3460
3461   /* Enable neighbour discovery */
3462   enable_broadcasting = GNUNET_CONFIGURATION_get_value_yesno (env->cfg,
3463       "transport-udp", "BROADCAST");
3464   if (enable_broadcasting == GNUNET_SYSERR)
3465     enable_broadcasting = GNUNET_NO;
3466
3467   enable_broadcasting_recv = GNUNET_CONFIGURATION_get_value_yesno (env->cfg,
3468       "transport-udp", "BROADCAST_RECEIVE");
3469   if (enable_broadcasting_recv == GNUNET_SYSERR)
3470     enable_broadcasting_recv = GNUNET_YES;
3471
3472   if (GNUNET_SYSERR ==
3473       GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp",
3474                                              "BROADCAST_INTERVAL",
3475                                              &fancy_interval))
3476   {
3477     interval = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10);
3478   }
3479   else
3480   {
3481     if (GNUNET_SYSERR ==
3482         GNUNET_STRINGS_fancy_time_to_relative (fancy_interval, &interval))
3483     {
3484       interval = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30);
3485     }
3486     GNUNET_free(fancy_interval);
3487   }
3488
3489   /* Maximum datarate */
3490   if (GNUNET_OK !=
3491       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp",
3492                                              "MAX_BPS", &udp_max_bps))
3493   {
3494     udp_max_bps = 1024 * 1024 * 50; /* 50 MB/s == infinity for practical purposes */
3495   }
3496
3497   p = GNUNET_new (struct Plugin);
3498   p->port = port;
3499   p->aport = aport;
3500   p->broadcast_interval = interval;
3501   p->enable_ipv6 = enable_v6;
3502   p->enable_ipv4 = GNUNET_YES; /* default */
3503   p->enable_broadcasting = enable_broadcasting;
3504   p->enable_broadcasting_receiving = enable_broadcasting_recv;
3505   p->env = env;
3506   p->sessions = GNUNET_CONTAINER_multipeermap_create (10, GNUNET_NO);
3507   p->defrag_ctxs = GNUNET_CONTAINER_heap_create (
3508       GNUNET_CONTAINER_HEAP_ORDER_MIN);
3509   p->mst = GNUNET_SERVER_mst_create (&process_inbound_tokenized_messages, p);
3510   GNUNET_BANDWIDTH_tracker_init (&p->tracker, NULL, NULL,
3511       GNUNET_BANDWIDTH_value_init ((uint32_t) udp_max_bps), 30);
3512   LOG(GNUNET_ERROR_TYPE_DEBUG,
3513       "Setting up sockets\n");
3514   res = setup_sockets (p,
3515                        (GNUNET_YES == have_bind6) ? &server_addrv6 : NULL,
3516                        (GNUNET_YES == have_bind4) ? &server_addrv4 : NULL);
3517   if ((res == 0) || ((p->sockv4 == NULL )&& (p->sockv6 == NULL)))
3518   {
3519     LOG (GNUNET_ERROR_TYPE_ERROR,
3520         _("Failed to create network sockets, plugin failed\n"));
3521     GNUNET_CONTAINER_multipeermap_destroy (p->sessions);
3522     GNUNET_CONTAINER_heap_destroy (p->defrag_ctxs);
3523     GNUNET_SERVER_mst_destroy (p->mst);
3524     GNUNET_free (p);
3525     return NULL;
3526   }
3527
3528   /* Setup broadcasting and receiving beacons */
3529   setup_broadcast (p, &server_addrv6, &server_addrv4);
3530
3531   api = GNUNET_new (struct GNUNET_TRANSPORT_PluginFunctions);
3532   api->cls = p;
3533   api->send = NULL;
3534   api->disconnect_session = &udp_disconnect_session;
3535   api->query_keepalive_factor = &udp_query_keepalive_factor;
3536   api->disconnect_peer = &udp_disconnect;
3537   api->address_pretty_printer = &udp_plugin_address_pretty_printer;
3538   api->address_to_string = &udp_address_to_string;
3539   api->string_to_address = &udp_string_to_address;
3540   api->check_address = &udp_plugin_check_address;
3541   api->get_session = &udp_plugin_get_session;
3542   api->send = &udp_plugin_send;
3543   api->get_network = &udp_get_network;
3544   api->update_session_timeout = &udp_plugin_update_session_timeout;
3545   api->setup_monitor = &udp_plugin_setup_monitor;
3546   return api;
3547 }
3548
3549
3550 /**
3551  * Function called on each entry in the defragmentation heap to
3552  * clean it up.
3553  *
3554  * @param cls NULL
3555  * @param node node in the heap (to be removed)
3556  * @param element a `struct DefragContext` to be cleaned up
3557  * @param cost unused
3558  * @return #GNUNET_YES
3559  */
3560 static int
3561 heap_cleanup_iterator (void *cls,
3562                        struct GNUNET_CONTAINER_HeapNode *node,
3563                        void *element,
3564                        GNUNET_CONTAINER_HeapCostType cost)
3565 {
3566   struct DefragContext *d_ctx = element;
3567
3568   GNUNET_CONTAINER_heap_remove_node (node);
3569   GNUNET_DEFRAGMENT_context_destroy (d_ctx->defrag);
3570   GNUNET_free (d_ctx);
3571   return GNUNET_YES;
3572 }
3573
3574
3575 /**
3576  * The exported method. Makes the core api available via a global and
3577  * returns the udp transport API.
3578  *
3579  * @param cls our `struct GNUNET_TRANSPORT_PluginEnvironment`
3580  * @return NULL
3581  */
3582 void *
3583 libgnunet_plugin_transport_udp_done (void *cls)
3584 {
3585   struct GNUNET_TRANSPORT_PluginFunctions *api = cls;
3586   struct Plugin *plugin = api->cls;
3587   struct PrettyPrinterContext *cur;
3588   struct PrettyPrinterContext *next;
3589   struct UDP_MessageWrapper *udpw;
3590
3591   if (NULL == plugin)
3592   {
3593     GNUNET_free(api);
3594     return NULL;
3595   }
3596   stop_broadcast (plugin);
3597   if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK )
3598   {
3599     GNUNET_SCHEDULER_cancel (plugin->select_task);
3600     plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
3601   }
3602   if (plugin->select_task_v6 != GNUNET_SCHEDULER_NO_TASK )
3603   {
3604     GNUNET_SCHEDULER_cancel (plugin->select_task_v6);
3605     plugin->select_task_v6 = GNUNET_SCHEDULER_NO_TASK;
3606   }
3607
3608   /* Closing sockets */
3609   if (GNUNET_YES == plugin->enable_ipv4)
3610   {
3611     if (NULL != plugin->sockv4)
3612     {
3613       GNUNET_break (GNUNET_OK ==
3614                     GNUNET_NETWORK_socket_close (plugin->sockv4));
3615       plugin->sockv4 = NULL;
3616     }
3617     GNUNET_NETWORK_fdset_destroy (plugin->rs_v4);
3618     GNUNET_NETWORK_fdset_destroy (plugin->ws_v4);
3619   }
3620   if (GNUNET_YES == plugin->enable_ipv6)
3621   {
3622     if (NULL != plugin->sockv6)
3623     {
3624       GNUNET_break (GNUNET_OK ==
3625                     GNUNET_NETWORK_socket_close (plugin->sockv6));
3626       plugin->sockv6 = NULL;
3627
3628       GNUNET_NETWORK_fdset_destroy (plugin->rs_v6);
3629       GNUNET_NETWORK_fdset_destroy (plugin->ws_v6);
3630     }
3631   }
3632   if (NULL != plugin->nat)
3633   {
3634     GNUNET_NAT_unregister (plugin->nat);
3635     plugin->nat = NULL;
3636   }
3637   if (NULL != plugin->defrag_ctxs)
3638   {
3639     GNUNET_CONTAINER_heap_iterate (plugin->defrag_ctxs,
3640                                    &heap_cleanup_iterator, NULL);
3641     GNUNET_CONTAINER_heap_destroy (plugin->defrag_ctxs);
3642     plugin->defrag_ctxs = NULL;
3643   }
3644   if (NULL != plugin->mst)
3645   {
3646     GNUNET_SERVER_mst_destroy (plugin->mst);
3647     plugin->mst = NULL;
3648   }
3649
3650   /* Clean up leftover messages */
3651   udpw = plugin->ipv4_queue_head;
3652   while (NULL != udpw)
3653   {
3654     struct UDP_MessageWrapper *tmp = udpw->next;
3655     dequeue (plugin, udpw);
3656     call_continuation (udpw, GNUNET_SYSERR);
3657     GNUNET_free(udpw);
3658     udpw = tmp;
3659   }
3660   udpw = plugin->ipv6_queue_head;
3661   while (NULL != udpw)
3662   {
3663     struct UDP_MessageWrapper *tmp = udpw->next;
3664     dequeue (plugin, udpw);
3665     call_continuation (udpw, GNUNET_SYSERR);
3666     GNUNET_free(udpw);
3667     udpw = tmp;
3668   }
3669
3670   /* Clean up sessions */
3671   LOG (GNUNET_ERROR_TYPE_DEBUG,
3672        "Cleaning up sessions\n");
3673   GNUNET_CONTAINER_multipeermap_iterate (plugin->sessions,
3674                                          &disconnect_and_free_it, plugin);
3675   GNUNET_CONTAINER_multipeermap_destroy (plugin->sessions);
3676
3677   next = plugin->ppc_dll_head;
3678   for (cur = next; NULL != cur; cur = next)
3679   {
3680     GNUNET_break(0);
3681     next = cur->next;
3682     GNUNET_CONTAINER_DLL_remove (plugin->ppc_dll_head,
3683                                  plugin->ppc_dll_tail,
3684                                  cur);
3685     GNUNET_RESOLVER_request_cancel (cur->resolver_handle);
3686     GNUNET_free (cur);
3687   }
3688   GNUNET_free (plugin);
3689   GNUNET_free (api);
3690   return NULL;
3691 }
3692
3693 /* end of plugin_transport_udp.c */