src/transport/plugin_transport_udp.c

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