complete state reset functionality
[oweals/gnunet.git] / src / transport / plugin_transport_udp_new.c
1 /*
2      This file is part of GNUnet
3      (C) 2010, 2011 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file transport/plugin_transport_udp.c
23  * @brief Implementation of the UDP NAT punching
24  *        transport service
25  * @author Christian Grothoff
26  * @author Nathan Evans
27  */
28 #include "platform.h"
29 #include "gnunet_hello_lib.h"
30 #include "gnunet_util_lib.h"
31 #include "gnunet_fragmentation_lib.h"
32 #include "gnunet_nat_lib.h"
33 #include "gnunet_protocols.h"
34 #include "gnunet_resolver_service.h"
35 #include "gnunet_signatures.h"
36 #include "gnunet_constants.h"
37 #include "gnunet_statistics_service.h"
38 #include "gnunet_transport_service.h"
39 #include "gnunet_transport_plugin.h"
40 #include "transport.h"
41
42 #define LOG(kind,...) GNUNET_log_from (kind, "transport-udp", __VA_ARGS__)
43
44
45 #define DEBUG_UDP GNUNET_EXTRA_LOGGING
46
47 /**
48  * MTU for fragmentation subsystem.  Should be conservative since
49  * all communicating peers MUST work with this MTU.
50  */
51 #define UDP_MTU 1400
52
53 /**
54  * Number of messages we can defragment in parallel.  We only really
55  * defragment 1 message at a time, but if messages get re-ordered, we
56  * may want to keep knowledge about the previous message to avoid
57  * discarding the current message in favor of a single fragment of a
58  * previous message.  3 should be good since we don't expect massive
59  * message reorderings with UDP.
60  */
61 #define UDP_MAX_MESSAGES_IN_DEFRAG 3
62
63 /**
64  * We keep a defragmentation queue per sender address.  How many
65  * sender addresses do we support at the same time? Memory consumption
66  * is roughly a factor of 32k * UDP_MAX_MESSAGES_IN_DEFRAG times this
67  * value. (So 128 corresponds to 12 MB and should suffice for
68  * connecting to roughly 128 peers via UDP).
69  */
70 #define UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG 128
71
72
73 /**
74  * UDP Message-Packet header (after defragmentation).
75  */
76 struct UDPMessage
77 {
78   /**
79    * Message header.
80    */
81   struct GNUNET_MessageHeader header;
82
83   /**
84    * Always zero for now.
85    */
86   uint32_t reserved;
87
88   /**
89    * What is the identity of the sender
90    */
91   struct GNUNET_PeerIdentity sender;
92
93 };
94
95
96 /**
97  * UDP ACK Message-Packet header (after defragmentation).
98  */
99 struct UDP_ACK_Message
100 {
101   /**
102    * Message header.
103    */
104   struct GNUNET_MessageHeader header;
105
106   /**
107    * Desired delay for flow control
108    */
109   uint32_t delay;
110
111   /**
112    * What is the identity of the sender
113    */
114   struct GNUNET_PeerIdentity sender;
115 };
116
117
118 /**
119  * Network format for IPv4 addresses.
120  */
121 struct IPv4UdpAddress
122 {
123   /**
124    * IPv4 address, in network byte order.
125    */
126   uint32_t ipv4_addr GNUNET_PACKED;
127
128   /**
129    * Port number, in network byte order.
130    */
131   uint16_t u4_port GNUNET_PACKED;
132 };
133
134
135 /**
136  * Network format for IPv6 addresses.
137  */
138 struct IPv6UdpAddress
139 {
140
141   /**
142    * IPv6 address.
143    */
144   struct in6_addr ipv6_addr GNUNET_PACKED;
145
146   /**
147    * Port number, in network byte order.
148    */
149   uint16_t u6_port GNUNET_PACKED;
150 };
151
152
153 /* Forward definition */
154 struct Plugin;
155
156
157 /**
158  * Session with another peer.  FIXME: why not make this into
159  * a regular 'struct Session' and pass it around!?
160  */
161 struct Session
162 {
163
164   /**
165    * Which peer is this session for or from?
166    */
167   struct GNUNET_PeerIdentity target;
168
169   /**
170    * Pointer to the global plugin struct.
171    */
172   struct Plugin *plugin;
173
174   /**
175    * Address of the other peer
176    */
177   const struct sockaddr *sock_addr;
178
179   size_t addrlen;
180
181   /**
182    * Function to call upon completion of the transmission.
183    */
184   GNUNET_TRANSPORT_TransmitContinuation cont;
185
186   /**
187    * Closure for 'cont'.
188    */
189   void *cont_cls;
190
191   /**
192    * Current outgoing message to this peer.
193    */
194   struct GNUNET_FRAGMENT_Context *frag;
195
196   /*
197    * Is this a reliable, bidirectional connection?
198    * YES if we already received a message with this session
199    */
200   int bidirectional;
201
202   /*
203    * Task invalidating this session if idle
204    */
205   GNUNET_SCHEDULER_TaskIdentifier invalidation_task;
206
207   /*
208    * Desired delay for next sending we send to other peer
209    */
210   struct GNUNET_TIME_Relative flow_delay_for_other_peer;
211
212   /*
213    * Desired delay for next sending we received from other peer
214    */
215   struct GNUNET_TIME_Absolute flow_delay_from_other_peer;
216 };
217
218 /**
219  * Encapsulation of all of the state of the plugin.
220  */
221 struct Plugin
222 {
223
224   /**
225    * Our environment.
226    */
227   struct GNUNET_TRANSPORT_PluginEnvironment *env;
228
229   /**
230    * Session of peers with whom we are currently connected,
231    * map of peer identity to 'struct PeerSession'.
232    */
233   struct GNUNET_CONTAINER_MultiHashMap *sessions;
234
235   /**
236    * Heap with all of our defragmentation activities.
237    */
238   struct GNUNET_CONTAINER_Heap *defrags;
239
240   /**
241    * ID of select task
242    */
243   GNUNET_SCHEDULER_TaskIdentifier select_task;
244
245   /**
246    * Tokenizer for inbound messages.
247    */
248   struct GNUNET_SERVER_MessageStreamTokenizer *mst;
249
250   /**
251    * Bandwidth tracker to limit global UDP traffic.
252    */
253   struct GNUNET_BANDWIDTH_Tracker tracker;
254
255   /**
256    * Address we were told to bind to exclusively (IPv4).
257    */
258   char *bind4_address;
259
260   /**
261    * Address we were told to bind to exclusively (IPv6).
262    */
263   char *bind6_address;
264
265   /**
266    * Handle to NAT traversal support.
267    */
268   struct GNUNET_NAT_Handle *nat;
269
270   /**
271    * FD Read set
272    */
273   struct GNUNET_NETWORK_FDSet *rs;
274
275   /**
276    * The read socket for IPv4
277    */
278   struct GNUNET_NETWORK_Handle *sockv4;
279
280   /**
281    * The read socket for IPv6
282    */
283   struct GNUNET_NETWORK_Handle *sockv6;
284
285   /**
286    * expected delay for ACKs
287    */
288   struct GNUNET_TIME_Relative last_expected_delay;
289
290   /**
291    * Port we listen on.
292    */
293   uint16_t port;
294
295   /**
296    * Port we advertise on.
297    */
298   uint16_t aport;
299
300 };
301
302 int delete_session_iterator (void *cls,
303                              const GNUNET_HashCode * key,
304                              void *value)
305 {
306   struct Session *s = (struct Session *) value;
307   if (s->invalidation_task != GNUNET_SCHEDULER_NO_TASK)
308   {
309     GNUNET_SCHEDULER_cancel(s->invalidation_task);
310     s->invalidation_task = GNUNET_SCHEDULER_NO_TASK;
311   }
312   GNUNET_CONTAINER_multihashmap_remove(s->plugin->sessions, key, s);
313
314   GNUNET_free (s);
315   return GNUNET_YES;
316 }
317 /**
318  * Disconnect from a remote node.  Clean up session if we have one for this peer
319  *
320  * @param cls closure for this call (should be handle to Plugin)
321  * @param target the peeridentity of the peer to disconnect
322  * @return GNUNET_OK on success, GNUNET_SYSERR if the operation failed
323  */
324 static void
325 udp_disconnect (void *cls, const struct GNUNET_PeerIdentity *target)
326 {
327   struct Plugin *plugin = cls;
328
329   GNUNET_CONTAINER_multihashmap_get_multiple(plugin, &target->hashPubKey, &delete_session_iterator, NULL);
330
331 }
332
333
334 /**
335  * Actually send out the message.
336  *
337  * @param plugin the plugin
338  * @param sa the address to send the message to
339  * @param msg message to transmit
340  * @return the number of bytes written
341  */
342 static ssize_t
343 udp_send (struct Plugin *plugin, const struct sockaddr *sa,
344           const struct GNUNET_MessageHeader *msg)
345 {
346   ssize_t sent;
347   size_t slen;
348
349   switch (sa->sa_family)
350   {
351   case AF_INET:
352     if (NULL == plugin->sockv4)
353       return 0;
354     sent =
355         GNUNET_NETWORK_socket_sendto (plugin->sockv4, msg, ntohs (msg->size),
356                                       sa, slen = sizeof (struct sockaddr_in));
357     break;
358   case AF_INET6:
359     if (NULL == plugin->sockv6)
360       return 0;
361     sent =
362         GNUNET_NETWORK_socket_sendto (plugin->sockv6, msg, ntohs (msg->size),
363                                       sa, slen = sizeof (struct sockaddr_in6));
364     break;
365   default:
366     GNUNET_break (0);
367     return 0;
368   }
369   if (GNUNET_SYSERR == sent)
370   {
371     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "sendto");
372     LOG (GNUNET_ERROR_TYPE_ERROR,
373                 "UDP transmited %u-byte message to %s (%d: %s)\n",
374                 (unsigned int) ntohs (msg->size), GNUNET_a2s (sa, slen),
375                 (int) sent, (sent < 0) ? STRERROR (errno) : "ok");
376
377   }
378   LOG (GNUNET_ERROR_TYPE_DEBUG,
379               "UDP transmited %u-byte message to %s (%d: %s)\n",
380               (unsigned int) ntohs (msg->size), GNUNET_a2s (sa, slen),
381               (int) sent, (sent < 0) ? STRERROR (errno) : "ok");
382   return sent;
383 }
384
385
386 static struct Session *
387 create_session (struct Plugin *plugin, const struct GNUNET_PeerIdentity *target,
388     const void *addr, size_t addrlen,
389     GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls)
390 {
391   struct Session * s;
392   const struct IPv4UdpAddress *t4;
393   const struct IPv6UdpAddress *t6;
394   struct sockaddr_in *v4;
395   struct sockaddr_in6 *v6;
396   size_t len;
397
398   switch (addrlen)
399   {
400   case sizeof (struct IPv4UdpAddress):
401     if (NULL == plugin->sockv4)
402     {
403       return NULL;
404     }
405     t4 = addr;
406     s =
407         GNUNET_malloc (sizeof (struct Session) +
408                        sizeof (struct sockaddr_in));
409     len = sizeof (struct sockaddr_in);
410     v4 = (struct sockaddr_in *) &s[1];
411     v4->sin_family = AF_INET;
412 #if HAVE_SOCKADDR_IN_SIN_LEN
413     v4->sin_len = sizeof (struct sockaddr_in);
414 #endif
415     v4->sin_port = t4->u4_port;
416     v4->sin_addr.s_addr = t4->ipv4_addr;
417     break;
418   case sizeof (struct IPv6UdpAddress):
419     if (NULL == plugin->sockv6)
420     {
421       return NULL;
422     }
423     t6 = addr;
424     s =
425         GNUNET_malloc (sizeof (struct Session) +
426                        sizeof (struct sockaddr_in6));
427     len = sizeof (struct sockaddr_in6);
428     v6 = (struct sockaddr_in6 *) &s[1];
429     v6->sin6_family = AF_INET6;
430 #if HAVE_SOCKADDR_IN_SIN_LEN
431     v6->sin6_len = sizeof (struct sockaddr_in6);
432 #endif
433     v6->sin6_port = t6->u6_port;
434     v6->sin6_addr = t6->ipv6_addr;
435     break;
436   default:
437     /* Must have a valid address to send to */
438     GNUNET_break_op (0);
439     return NULL;
440   }
441
442   s->valid_until = GNUNET_TIME_absolute_get_zero ();
443   s->flow_delay_for_other_peer = GNUNET_TIME_relative_get_zero ();
444   s->flow_delay_from_other_peer = GNUNET_TIME_absolute_get_zero ();
445   s->bidirectional = GNUNET_NO;
446   s->invalidation_task = GNUNET_SCHEDULER_NO_TASK;
447   s->addrlen = len;
448   s->target = *target;
449   s->plugin = plugin;
450   s->sock_addr = (const struct sockaddr *) &s[1];
451   s->cont = cont;
452   s->cont_cls = cont_cls;
453
454   return s;
455 }
456
457 static void
458 invalidation_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
459 {
460   struct Session * s = cls;
461   s->invalidation_task = GNUNET_SCHEDULER_NO_TASK;
462   LOG (GNUNET_ERROR_TYPE_ERROR,
463                    "Session %X (`%s') is now invalid\n", s, GNUNET_a2s (s->sock_addr,s->addrlen));
464
465   s->plugin->env->session_end(s->plugin->env->cls, &s->target, s);
466   GNUNET_assert (GNUNET_YES == GNUNET_CONTAINER_multihashmap_remove(s->plugin->sessions, &s->target.hashPubKey, s));
467   GNUNET_free (s);
468 }
469
470 static const char *
471 udp_address_to_string (void *cls, const void *addr, size_t addrlen);
472
473 /**
474  * Function that is called with messages created by the fragmentation
475  * module.  In the case of the 'proc' callback of the
476  * GNUNET_FRAGMENT_context_create function, this function must
477  * eventually call 'GNUNET_FRAGMENT_context_transmission_done'.
478  *
479  * @param cls closure, the 'struct PeerSession'
480  * @param msg the message that was created
481  */
482 static void
483 send_fragment (void *cls, const struct GNUNET_MessageHeader *msg)
484 {
485   struct Session *s = cls;
486
487   udp_send (session->plugin, session->sock_addr, msg);
488   GNUNET_FRAGMENT_context_transmission_done (session->frag);
489 }
490
491 /**
492  * Function that can be used by the transport service to transmit
493  * a message using the plugin.
494  *
495  * @param cls closure
496  * @param target who should receive this message (ignored by UDP)
497  * @param msgbuf one or more GNUNET_MessageHeader(s) strung together
498  * @param msgbuf_size the size of the msgbuf to send
499  * @param priority how important is the message (ignored by UDP)
500  * @param timeout when should we time out (give up) if we can not transmit?
501  * @param session identifier used for this session (NULL for UDP)
502  * @param addr the addr to send the message to
503  * @param addrlen the len of addr
504  * @param force_address not used, we had better have an address to send to
505  *        because we are stateless!!
506  * @param cont continuation to call once the message has
507  *        been transmitted (or if the transport is ready
508  *        for the next transmission call; or if the
509  *        peer disconnected...)
510  * @param cont_cls closure for cont
511  *
512  * @return the number of bytes written (may return 0 and the message can
513  *         still be transmitted later!)
514  */
515 static ssize_t
516 udp_plugin_send (void *cls, const struct GNUNET_PeerIdentity *target,
517                  const char *msgbuf, size_t msgbuf_size, unsigned int priority,
518                  struct GNUNET_TIME_Relative timeout, struct Session *session,
519                  const void *addr, size_t addrlen, int force_address,
520                  GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls)
521 {
522   struct Plugin *plugin = cls;
523   struct Session *s;
524   const struct IPv4UdpAddress *t4;
525   const struct IPv6UdpAddress *t6;
526   size_t mlen = msgbuf_size + sizeof (struct UDPMessage);
527   char mbuf[mlen];
528   struct UDPMessage *udp;
529
530   if (mlen >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
531   {
532     GNUNET_break (0);
533     return GNUNET_SYSERR;
534   }
535
536   LOG (GNUNET_ERROR_TYPE_DEBUG,
537       "UDP transmits %u-byte message to `%s' using address `%s' session 0x%X mode %i\n",
538       msgbuf_size, GNUNET_i2s (target),
539       udp_address_to_string (NULL, addr, addrlen),
540       session, force_address);
541
542   /* no valid address given */
543   if ((addr = NULL) || (addrlen == 0))
544   {
545     GNUNET_break (0);
546     return GNUNET_SYSERR;
547   }
548
549   if (session != NULL)
550   {
551     s = find_session (target, addr, addrlen);
552     if (s != session)
553     {
554       /* found a conflicting session for this peer */
555       GNUNET_break (0);
556       return GNUNET_SYSERR;
557     }
558   }
559   else
560   {
561     /* use a reliable transmission but no session given */
562     if (force_address == GNUNET_SYSERR)
563       return GNUNET_SYSERR;
564     /* create new session */
565     s = create_session(plugin, target, addr, addrlen, cont, cont_cls);
566     GNUNET_CONTAINER_multihashmap_put(plugin->sessions, &target->hashPubKey, s, GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
567     /* schedule invalidation task */
568     s->invalidation_task = GNUNET_SCHEDULER_add_delayed(GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, &invalidation_task, NULL);
569   }
570
571   /* Message */
572   udp = (struct UDPMessage *) mbuf;
573   udp->header.size = htons (mlen);
574   udp->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE);
575   udp->reserved = htonl (0);
576   udp->sender = *plugin->env->my_identity;
577   memcpy (&udp[1], msgbuf, msgbuf_size);
578
579   struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get();
580   if (s->flow_delay_from_other_peer.abs_value > now.abs_value)
581   {
582     struct GNUNET_TIME_Relative delta = GNUNET_TIME_absolute_get_difference(now, s->flow_delay_from_other_peer);
583     LOG (GNUNET_ERROR_TYPE_DEBUG,
584                   "We try to send to early! Should in %llu!\n", delta.rel_value);
585   }
586   else
587     LOG (GNUNET_ERROR_TYPE_DEBUG, "We can send!\n");
588
589
590   /* send without fragmentation */
591   if (mlen <= UDP_MTU)
592   {
593     mlen = udp_send (plugin, peer_session->sock_addr, &udp->header);
594     if (cont != NULL)
595       cont (cont_cls, target, (mlen > 0) ? GNUNET_OK : GNUNET_SYSERR);
596     GNUNET_free_non_null (peer_session);
597   }
598   /* send with fragmentation */
599   else
600   {
601     s->frag = GNUNET_FRAGMENT_context_create (plugin->env->stats, UDP_MTU,
602                                         &plugin->tracker,
603                                         plugin->last_expected_delay,
604                                         &udp->header, &send_fragment,
605                                         s);
606   }
607   return mlen;
608 }
609
610
611 /**
612  * Message tokenizer has broken up an incomming message. Pass it on
613  * to the service.
614  *
615  * @param cls the 'struct Plugin'
616  * @param client the Session
617  * @param hdr the actual message
618  */
619 static void
620 process_inbound_tokenized_messages (void *cls, void *client,
621                                     const struct GNUNET_MessageHeader *hdr)
622 {
623   struct Plugin *plugin = cls;
624   struct Session * s = client;
625   struct GNUNET_ATS_Information distance;
626   struct GNUNET_TIME_Relative delay;
627
628   /* setup ATS */
629   distance.type = htonl (GNUNET_ATS_QUALITY_NET_DISTANCE);
630   distance.value = htonl (1);
631
632   LOG (GNUNET_ERROR_TYPE_DEBUG,
633                    "Giving Session %X %s  to transport\n", s->session, GNUNET_i2s(&s->target));
634   delay = plugin->env->receive (plugin->env->cls, &s->target, hdr, &distance, 1, s,
635       // FIXME: USE UDP ADDRESSES!!!!
636                         s->sock_addr, s->addrlen);
637   s->flow_delay_for_other_peer = delay;
638 }
639
640
641 /**
642  * We've received a UDP Message.  Process it (pass contents to main service).
643  *
644  * @param plugin plugin context
645  * @param msg the message
646  * @param sender_addr sender address
647  * @param sender_addr_len number of bytes in sender_addr
648  */
649 static void
650 process_udp_message (struct Plugin *plugin, const struct UDPMessage *msg,
651                      const struct sockaddr *sender_addr,
652                      socklen_t sender_addr_len)
653 {
654   struct IPv4UdpAddress u4;
655   struct IPv6UdpAddress u6;
656   const void *arg;
657   size_t args;
658
659   if (0 != ntohl (msg->reserved))
660   {
661     GNUNET_break_op (0);
662     return;
663   }
664   if (ntohs (msg->header.size) <
665       sizeof (struct GNUNET_MessageHeader) + sizeof (struct UDPMessage))
666   {
667     GNUNET_break_op (0);
668     return;
669   }
670
671   /* convert address */
672   switch (sender_addr->sa_family)
673   {
674   case AF_INET:
675     GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in));
676     u4.ipv4_addr = ((struct sockaddr_in *) sender_addr)->sin_addr.s_addr;
677     u4.u4_port = ((struct sockaddr_in *) sender_addr)->sin_port;
678     arg = &u4;
679     args = sizeof (u4);
680     break;
681   case AF_INET6:
682     GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in6));
683     u6.ipv6_addr = ((struct sockaddr_in6 *) sender_addr)->sin6_addr;
684     u6.u6_port = ((struct sockaddr_in6 *) sender_addr)->sin6_port;
685     arg = &u6;
686     args = sizeof (u6);
687     break;
688   default:
689     GNUNET_break (0);
690     return;
691   }
692 #if DEBUG_UDP
693   LOG (GNUNET_ERROR_TYPE_DEBUG,
694                    "Received message with %u bytes from peer `%s' at `%s'\n",
695                    (unsigned int) ntohs (msg->header.size),
696                    GNUNET_i2s (&msg->sender), GNUNET_a2s (sender_addr,
697                                                           sender_addr_len));
698 #endif
699
700   const struct UDPMessage * udp_msg = (const struct UDPMessage *) msg;
701   LOG (GNUNET_ERROR_TYPE_DEBUG,
702               "Lookup inbound UDP sessions for peer `%s' address `%s'\n",
703               GNUNET_i2s (&udp_msg->sender),
704               udp_address_to_string(NULL, arg, args));
705
706   /* create a session for inbound connections */
707   struct Session * s = NULL;
708   s = find_inbound_session (plugin, &udp_msg->sender, sender_addr, sender_addr_len);
709
710   if (s != NULL)
711   {
712     LOG (GNUNET_ERROR_TYPE_DEBUG,
713                 "Found existing inbound UDP sessions 0x%X for peer `%s' address `%s'\n",
714                 s,
715                 GNUNET_i2s (&s->target),
716                 udp_address_to_string(NULL, arg, args));
717   }
718   else
719   {
720     s = create_session (plugin, &udp_msg->sender, arg, args, NULL, NULL);
721     LOG (GNUNET_ERROR_TYPE_DEBUG,
722                 "Creating inbound UDP sessions 0x%X for peer `%s' address `%s'\n",
723                 s,
724                 GNUNET_i2s (&s->target),
725                 udp_address_to_string(NULL, arg, args));
726
727     GNUNET_assert (GNUNET_OK == GNUNET_CONTAINER_multihashmap_put (plugin->sessions,
728                                                       &s->target.hashPubKey,
729                                                       s,
730                                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
731   }
732   s->valid_until = GNUNET_TIME_absolute_add(GNUNET_TIME_absolute_get(), GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT);
733   if (s->invalidation_task != GNUNET_SCHEDULER_NO_TASK)
734   {
735     GNUNET_SCHEDULER_cancel(s->invalidation_task);
736     s->invalidation_task = GNUNET_SCHEDULER_NO_TASK;
737     LOG (GNUNET_ERROR_TYPE_DEBUG,
738                 "Rescheduling %X' `%s'\n",
739                 s, udp_address_to_string(NULL, arg, args));
740   }
741   s->invalidation_task = GNUNET_SCHEDULER_add_delayed(GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, &invalidation_task, s);
742   /* we received from peer, so session is bidirectional and reliable */
743   s->bidirectional = GNUNET_YES;
744
745   GNUNET_SERVER_mst_receive (plugin->mst, &s, (const char *) &msg[1],
746                              ntohs (msg->header.size) -
747                              sizeof (struct UDPMessage), GNUNET_YES, GNUNET_NO);
748 }
749
750
751 /**
752  * Read and process a message from the given socket.
753  *
754  * @param plugin the overall plugin
755  * @param rsock socket to read from
756  */
757 static void
758 udp_read (struct Plugin *plugin, struct GNUNET_NETWORK_Handle *rsock)
759 {
760   socklen_t fromlen;
761   char addr[32];
762   char buf[65536];
763   ssize_t ret;
764   const struct GNUNET_MessageHeader *msg;
765   const struct GNUNET_MessageHeader *ack;
766   struct Session *peer_session;
767   const struct UDP_ACK_Message *udp_ack;
768   struct ReceiveContext *rc;
769   struct GNUNET_TIME_Absolute now;
770   struct FindReceiveContext frc;
771   struct Session * s = NULL;
772   struct GNUNET_TIME_Relative flow_delay;
773
774   fromlen = sizeof (addr);
775   memset (&addr, 0, sizeof (addr));
776   ret =
777       GNUNET_NETWORK_socket_recvfrom (rsock, buf, sizeof (buf),
778                                       (struct sockaddr *) &addr, &fromlen);
779   if (ret < sizeof (struct GNUNET_MessageHeader))
780   {
781     GNUNET_break_op (0);
782     return;
783   }
784   msg = (const struct GNUNET_MessageHeader *) buf;
785
786   LOG (GNUNET_ERROR_TYPE_DEBUG,
787               "UDP received %u-byte message from `%s' type %i\n", (unsigned int) ret,
788               GNUNET_a2s ((const struct sockaddr *) addr, fromlen), ntohs(msg->type));
789
790   if (ret != ntohs (msg->size))
791   {
792     GNUNET_break_op (0);
793     return;
794   }
795
796   switch (ntohs (msg->type))
797   {
798   case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE:
799     if (ntohs (msg->size) < sizeof (struct UDPMessage))
800     {
801       GNUNET_break_op (0);
802       return;
803     }
804     process_udp_message (plugin, (const struct UDPMessage *) msg,
805                          (const struct sockaddr *) addr, fromlen);
806     return;
807
808   case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK:
809     if (ntohs (msg->size) <
810         sizeof (struct UDP_ACK_Message) + sizeof (struct GNUNET_MessageHeader))
811     {
812       GNUNET_break_op (0);
813       return;
814     }
815     udp_ack = (const struct UDP_ACK_Message *) msg;
816     LOG (GNUNET_ERROR_TYPE_DEBUG,
817                 "UDP processes %u-byte acknowledgement from `%s' at `%s'\n",
818                 (unsigned int) ntohs (msg->size), GNUNET_i2s (&udp->sender),
819                 GNUNET_a2s ((const struct sockaddr *) addr, fromlen));
820
821     // TODO Guess a session from address, give it defragmenter and check in process_udp if guess was right
822     s = guess_session (plugin->sessions, addr, addrlen);
823     if (GNUNET_OK != GNUNET_FRAGMENT_process_ack (peer_session->frag, ack))
824     // ... other stuff
825     return;
826
827   case GNUNET_MESSAGE_TYPE_FRAGMENT:
828     LOG (GNUNET_ERROR_TYPE_DEBUG,
829                 "UDP processes %u-byte fragment from `%s'\n",
830                 (unsigned int) ntohs (msg->size),
831                 GNUNET_a2s ((const struct sockaddr *) addr, fromlen));
832     // TODO Guess a session from address, give it defragmenter and check in process_udp if guess was right
833     s = guess_session (plugin->sessions, addr, addrlen);
834     if (GNUNET_OK == GNUNET_DEFRAGMENT_process_fragment (rc->defrag, msg))
835     // other stuff
836     return;
837   default:
838     GNUNET_break_op (0);
839     return;
840   }
841 }
842
843
844 /**
845  * We have been notified that our writeset has something to read.  We don't
846  * know which socket needs to be read, so we have to check each one
847  * Then reschedule this function to be called again once more is available.
848  *
849  * @param cls the plugin handle
850  * @param tc the scheduling context (for rescheduling this function again)
851  */
852 static void
853 udp_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
854 {
855   struct Plugin *plugin = cls;
856
857   plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
858   if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
859     return;
860   if ((NULL != plugin->sockv4) &&
861       (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv4)))
862     udp_read (plugin, plugin->sockv4);
863   if ((NULL != plugin->sockv6) &&
864       (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv6)))
865     udp_read (plugin, plugin->sockv6);
866   plugin->select_task =
867       GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
868                                    GNUNET_SCHEDULER_NO_TASK,
869                                    GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
870                                    NULL, &udp_plugin_select, plugin);
871
872 }
873
874
875 /**
876  * Check if the given port is plausible (must be either our listen
877  * port or our advertised port).  If it is neither, we return
878  * GNUNET_SYSERR.
879  *
880  * @param plugin global variables
881  * @param in_port port number to check
882  * @return GNUNET_OK if port is either open_port or adv_port
883  */
884 static int
885 check_port (struct Plugin *plugin, uint16_t in_port)
886 {
887   if ((in_port == plugin->port) || (in_port == plugin->aport))
888     return GNUNET_OK;
889   return GNUNET_SYSERR;
890 }
891
892
893 /**
894  * Function that will be called to check if a binary address for this
895  * plugin is well-formed and corresponds to an address for THIS peer
896  * (as per our configuration).  Naturally, if absolutely necessary,
897  * plugins can be a bit conservative in their answer, but in general
898  * plugins should make sure that the address does not redirect
899  * traffic to a 3rd party that might try to man-in-the-middle our
900  * traffic.
901  *
902  * @param cls closure, should be our handle to the Plugin
903  * @param addr pointer to the address
904  * @param addrlen length of addr
905  * @return GNUNET_OK if this is a plausible address for this peer
906  *         and transport, GNUNET_SYSERR if not
907  *
908  */
909 static int
910 udp_plugin_check_address (void *cls, const void *addr, size_t addrlen)
911 {
912   struct Plugin *plugin = cls;
913   struct IPv4UdpAddress *v4;
914   struct IPv6UdpAddress *v6;
915
916   if ((addrlen != sizeof (struct IPv4UdpAddress)) &&
917       (addrlen != sizeof (struct IPv6UdpAddress)))
918   {
919     GNUNET_break_op (0);
920     return GNUNET_SYSERR;
921   }
922   if (addrlen == sizeof (struct IPv4UdpAddress))
923   {
924     v4 = (struct IPv4UdpAddress *) addr;
925     if (GNUNET_OK != check_port (plugin, ntohs (v4->u4_port)))
926       return GNUNET_SYSERR;
927     if (GNUNET_OK !=
928         GNUNET_NAT_test_address (plugin->nat, &v4->ipv4_addr,
929                                  sizeof (struct in_addr)))
930       return GNUNET_SYSERR;
931   }
932   else
933   {
934     v6 = (struct IPv6UdpAddress *) addr;
935     if (IN6_IS_ADDR_LINKLOCAL (&v6->ipv6_addr))
936     {
937       GNUNET_break_op (0);
938       return GNUNET_SYSERR;
939     }
940     if (GNUNET_OK != check_port (plugin, ntohs (v6->u6_port)))
941       return GNUNET_SYSERR;
942     if (GNUNET_OK !=
943         GNUNET_NAT_test_address (plugin->nat, &v6->ipv6_addr,
944                                  sizeof (struct in6_addr)))
945       return GNUNET_SYSERR;
946   }
947   return GNUNET_OK;
948 }
949
950
951 /**
952  * Function called for a quick conversion of the binary address to
953  * a numeric address.  Note that the caller must not free the
954  * address and that the next call to this function is allowed
955  * to override the address again.
956  *
957  * @param cls closure
958  * @param addr binary address
959  * @param addrlen length of the address
960  * @return string representing the same address
961  */
962 static const char *
963 udp_address_to_string (void *cls, const void *addr, size_t addrlen)
964 {
965   static char rbuf[INET6_ADDRSTRLEN + 10];
966   char buf[INET6_ADDRSTRLEN];
967   const void *sb;
968   struct in_addr a4;
969   struct in6_addr a6;
970   const struct IPv4UdpAddress *t4;
971   const struct IPv6UdpAddress *t6;
972   int af;
973   uint16_t port;
974
975   if (addrlen == sizeof (struct IPv6UdpAddress))
976   {
977     t6 = addr;
978     af = AF_INET6;
979     port = ntohs (t6->u6_port);
980     memcpy (&a6, &t6->ipv6_addr, sizeof (a6));
981     sb = &a6;
982   }
983   else if (addrlen == sizeof (struct IPv4UdpAddress))
984   {
985     t4 = addr;
986     af = AF_INET;
987     port = ntohs (t4->u4_port);
988     memcpy (&a4, &t4->ipv4_addr, sizeof (a4));
989     sb = &a4;
990   }
991   else
992   {
993     GNUNET_break_op (0);
994     return NULL;
995   }
996   inet_ntop (af, sb, buf, INET6_ADDRSTRLEN);
997   GNUNET_snprintf (rbuf, sizeof (rbuf), (af == AF_INET6) ? "[%s]:%u" : "%s:%u",
998                    buf, port);
999   return rbuf;
1000 }
1001
1002
1003 /**
1004  * Closure for 'append_port'.
1005  */
1006 struct PrettyPrinterContext
1007 {
1008   /**
1009    * Function to call with the result.
1010    */
1011   GNUNET_TRANSPORT_AddressStringCallback asc;
1012
1013   /**
1014    * Clsoure for 'asc'.
1015    */
1016   void *asc_cls;
1017
1018   /**
1019    * Port to add after the IP address.
1020    */
1021   uint16_t port;
1022 };
1023
1024
1025 /**
1026  * Append our port and forward the result.
1027  *
1028  * @param cls a 'struct PrettyPrinterContext'
1029  * @param hostname result from DNS resolver
1030  */
1031 static void
1032 append_port (void *cls, const char *hostname)
1033 {
1034   struct PrettyPrinterContext *ppc = cls;
1035   char *ret;
1036
1037   if (hostname == NULL)
1038   {
1039     ppc->asc (ppc->asc_cls, NULL);
1040     GNUNET_free (ppc);
1041     return;
1042   }
1043   GNUNET_asprintf (&ret, "%s:%d", hostname, ppc->port);
1044   ppc->asc (ppc->asc_cls, ret);
1045   GNUNET_free (ret);
1046 }
1047
1048
1049 /**
1050  * Convert the transports address to a nice, human-readable
1051  * format.
1052  *
1053  * @param cls closure
1054  * @param type name of the transport that generated the address
1055  * @param addr one of the addresses of the host, NULL for the last address
1056  *        the specific address format depends on the transport
1057  * @param addrlen length of the address
1058  * @param numeric should (IP) addresses be displayed in numeric form?
1059  * @param timeout after how long should we give up?
1060  * @param asc function to call on each string
1061  * @param asc_cls closure for asc
1062  */
1063 static void
1064 udp_plugin_address_pretty_printer (void *cls, const char *type,
1065                                    const void *addr, size_t addrlen,
1066                                    int numeric,
1067                                    struct GNUNET_TIME_Relative timeout,
1068                                    GNUNET_TRANSPORT_AddressStringCallback asc,
1069                                    void *asc_cls)
1070 {
1071   struct PrettyPrinterContext *ppc;
1072   const void *sb;
1073   size_t sbs;
1074   struct sockaddr_in a4;
1075   struct sockaddr_in6 a6;
1076   const struct IPv4UdpAddress *u4;
1077   const struct IPv6UdpAddress *u6;
1078   uint16_t port;
1079
1080   if (addrlen == sizeof (struct IPv6UdpAddress))
1081   {
1082     u6 = addr;
1083     memset (&a6, 0, sizeof (a6));
1084     a6.sin6_family = AF_INET6;
1085 #if HAVE_SOCKADDR_IN_SIN_LEN
1086     a6.sin6_len = sizeof (a6);
1087 #endif
1088     a6.sin6_port = u6->u6_port;
1089     memcpy (&a6.sin6_addr, &u6->ipv6_addr, sizeof (struct in6_addr));
1090     port = ntohs (u6->u6_port);
1091     sb = &a6;
1092     sbs = sizeof (a6);
1093   }
1094   else if (addrlen == sizeof (struct IPv4UdpAddress))
1095   {
1096     u4 = addr;
1097     memset (&a4, 0, sizeof (a4));
1098     a4.sin_family = AF_INET;
1099 #if HAVE_SOCKADDR_IN_SIN_LEN
1100     a4.sin_len = sizeof (a4);
1101 #endif
1102     a4.sin_port = u4->u4_port;
1103     a4.sin_addr.s_addr = u4->ipv4_addr;
1104     port = ntohs (u4->u4_port);
1105     sb = &a4;
1106     sbs = sizeof (a4);
1107   }
1108   else
1109   {
1110     /* invalid address */
1111     GNUNET_break_op (0);
1112     asc (asc_cls, NULL);
1113     return;
1114   }
1115   ppc = GNUNET_malloc (sizeof (struct PrettyPrinterContext));
1116   ppc->asc = asc;
1117   ppc->asc_cls = asc_cls;
1118   ppc->port = port;
1119   GNUNET_RESOLVER_hostname_get (sb, sbs, !numeric, timeout, &append_port, ppc);
1120 }
1121
1122
1123 /**
1124  * Our external IP address/port mapping has changed.
1125  *
1126  * @param cls closure, the 'struct LocalAddrList'
1127  * @param add_remove GNUNET_YES to mean the new public IP address, GNUNET_NO to mean
1128  *     the previous (now invalid) one
1129  * @param addr either the previous or the new public IP address
1130  * @param addrlen actual lenght of the address
1131  */
1132 static void
1133 udp_nat_port_map_callback (void *cls, int add_remove,
1134                            const struct sockaddr *addr, socklen_t addrlen)
1135 {
1136   struct Plugin *plugin = cls;
1137   struct IPv4UdpAddress u4;
1138   struct IPv6UdpAddress u6;
1139   void *arg;
1140   size_t args;
1141
1142   /* convert 'addr' to our internal format */
1143   switch (addr->sa_family)
1144   {
1145   case AF_INET:
1146     GNUNET_assert (addrlen == sizeof (struct sockaddr_in));
1147     u4.ipv4_addr = ((struct sockaddr_in *) addr)->sin_addr.s_addr;
1148     u4.u4_port = ((struct sockaddr_in *) addr)->sin_port;
1149     arg = &u4;
1150     args = sizeof (u4);
1151     break;
1152   case AF_INET6:
1153     GNUNET_assert (addrlen == sizeof (struct sockaddr_in6));
1154     memcpy (&u6.ipv6_addr, &((struct sockaddr_in6 *) addr)->sin6_addr,
1155             sizeof (struct in6_addr));
1156     u6.u6_port = ((struct sockaddr_in6 *) addr)->sin6_port;
1157     arg = &u6;
1158     args = sizeof (u6);
1159     break;
1160   default:
1161     GNUNET_break (0);
1162     return;
1163   }
1164   /* modify our published address list */
1165   plugin->env->notify_address (plugin->env->cls, add_remove, arg, args);
1166 }
1167
1168
1169 /**
1170  * The exported method. Makes the core api available via a global and
1171  * returns the udp transport API.
1172  *
1173  * @param cls our 'struct GNUNET_TRANSPORT_PluginEnvironment'
1174  * @return our 'struct GNUNET_TRANSPORT_PluginFunctions'
1175  */
1176 void *
1177 libgnunet_plugin_transport_udp_init (void *cls)
1178 {
1179   struct GNUNET_TRANSPORT_PluginEnvironment *env = cls;
1180   unsigned long long port;
1181   unsigned long long aport;
1182   struct GNUNET_TRANSPORT_PluginFunctions *api;
1183   struct Plugin *plugin;
1184   int sockets_created;
1185   struct sockaddr_in serverAddrv4;
1186   struct sockaddr_in6 serverAddrv6;
1187   struct sockaddr *serverAddr;
1188   struct sockaddr *addrs[2];
1189   socklen_t addrlens[2];
1190   socklen_t addrlen;
1191   unsigned int tries;
1192   unsigned long long udp_max_bps;
1193
1194   if (GNUNET_OK !=
1195       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", "PORT",
1196                                              &port))
1197     port = 2086;
1198   if (GNUNET_OK !=
1199       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp",
1200                                              "MAX_BPS", &udp_max_bps))
1201     udp_max_bps = 1024 * 1024 * 50;     /* 50 MB/s == infinity for practical purposes */
1202   if (GNUNET_OK !=
1203       GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp",
1204                                              "ADVERTISED_PORT", &aport))
1205     aport = port;
1206   if (port > 65535)
1207   {
1208     LOG (GNUNET_ERROR_TYPE_WARNING,
1209                 _("Given `%s' option is out of range: %llu > %u\n"), "PORT",
1210                 port, 65535);
1211     return NULL;
1212   }
1213   memset (&serverAddrv6, 0, sizeof (serverAddrv6));
1214   memset (&serverAddrv4, 0, sizeof (serverAddrv4));
1215
1216   plugin = GNUNET_malloc (sizeof (struct Plugin));
1217   GNUNET_BANDWIDTH_tracker_init (&plugin->tracker,
1218                                  GNUNET_BANDWIDTH_value_init ((uint32_t)
1219                                                               udp_max_bps), 30);
1220   plugin->last_expected_delay = GNUNET_TIME_UNIT_SECONDS;
1221   plugin->port = port;
1222   plugin->aport = aport;
1223   plugin->env = env;
1224   api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions));
1225   api->cls = plugin;
1226
1227   api->send = &udp_plugin_send;
1228   api->disconnect = &udp_disconnect;
1229   api->address_pretty_printer = &udp_plugin_address_pretty_printer;
1230   api->address_to_string = &udp_address_to_string;
1231   api->check_address = &udp_plugin_check_address;
1232
1233   if (GNUNET_YES ==
1234       GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp",
1235                                              "BINDTO", &plugin->bind4_address))
1236   {
1237     LOG (GNUNET_ERROR_TYPE_DEBUG,
1238                 "Binding udp plugin to specific address: `%s'\n",
1239                 plugin->bind4_address);
1240     if (1 != inet_pton (AF_INET, plugin->bind4_address, &serverAddrv4.sin_addr))
1241     {
1242       GNUNET_free (plugin->bind4_address);
1243       GNUNET_free (plugin);
1244       GNUNET_free (api);
1245       return NULL;
1246     }
1247   }
1248
1249   if (GNUNET_YES ==
1250       GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp",
1251                                              "BINDTO6", &plugin->bind6_address))
1252   {
1253     LOG (GNUNET_ERROR_TYPE_DEBUG,
1254                 "Binding udp plugin to specific address: `%s'\n",
1255                 plugin->bind6_address);
1256     if (1 !=
1257         inet_pton (AF_INET6, plugin->bind6_address, &serverAddrv6.sin6_addr))
1258     {
1259       LOG (GNUNET_ERROR_TYPE_ERROR, _("Invalid IPv6 address: `%s'\n"),
1260                   plugin->bind6_address);
1261       GNUNET_free_non_null (plugin->bind4_address);
1262       GNUNET_free (plugin->bind6_address);
1263       GNUNET_free (plugin);
1264       GNUNET_free (api);
1265       return NULL;
1266     }
1267   }
1268
1269   plugin->defrags =
1270       GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
1271   plugin->sessions =
1272       GNUNET_CONTAINER_multihashmap_create (UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG
1273                                             * 2);
1274   plugin->inbound_sessions =
1275       GNUNET_CONTAINER_multihashmap_create (UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG
1276                                             * 2);
1277   sockets_created = 0;
1278   if ((GNUNET_YES !=
1279        GNUNET_CONFIGURATION_get_value_yesno (plugin->env->cfg, "nat",
1280                                              "DISABLEV6")))
1281   {
1282     plugin->sockv6 = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_DGRAM, 0);
1283     if (NULL == plugin->sockv6)
1284     {
1285       GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "socket");
1286     }
1287     else
1288     {
1289 #if HAVE_SOCKADDR_IN_SIN_LEN
1290       serverAddrv6.sin6_len = sizeof (serverAddrv6);
1291 #endif
1292       serverAddrv6.sin6_family = AF_INET6;
1293       serverAddrv6.sin6_addr = in6addr_any;
1294       serverAddrv6.sin6_port = htons (plugin->port);
1295       addrlen = sizeof (serverAddrv6);
1296       serverAddr = (struct sockaddr *) &serverAddrv6;
1297 #if DEBUG_UDP
1298       LOG (GNUNET_ERROR_TYPE_DEBUG, "Binding to IPv6 port %d\n",
1299                   ntohs (serverAddrv6.sin6_port));
1300 #endif
1301       tries = 0;
1302       while (GNUNET_NETWORK_socket_bind (plugin->sockv6, serverAddr, addrlen) !=
1303              GNUNET_OK)
1304       {
1305         serverAddrv6.sin6_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000);        /* Find a good, non-root port */
1306 #if DEBUG_UDP
1307         LOG (GNUNET_ERROR_TYPE_DEBUG,
1308                     "IPv6 Binding failed, trying new port %d\n",
1309                     ntohs (serverAddrv6.sin6_port));
1310 #endif
1311         tries++;
1312         if (tries > 10)
1313         {
1314           GNUNET_NETWORK_socket_close (plugin->sockv6);
1315           plugin->sockv6 = NULL;
1316           break;
1317         }
1318       }
1319       if (plugin->sockv6 != NULL)
1320       {
1321         addrs[sockets_created] = (struct sockaddr *) &serverAddrv6;
1322         addrlens[sockets_created] = sizeof (serverAddrv6);
1323         sockets_created++;
1324       }
1325     }
1326   }
1327
1328   plugin->mst =
1329       GNUNET_SERVER_mst_create (&process_inbound_tokenized_messages, plugin);
1330   plugin->sockv4 = GNUNET_NETWORK_socket_create (PF_INET, SOCK_DGRAM, 0);
1331   if (NULL == plugin->sockv4)
1332   {
1333     GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "socket");
1334   }
1335   else
1336   {
1337 #if HAVE_SOCKADDR_IN_SIN_LEN
1338     serverAddrv4.sin_len = sizeof (serverAddrv4);
1339 #endif
1340     serverAddrv4.sin_family = AF_INET;
1341     serverAddrv4.sin_addr.s_addr = INADDR_ANY;
1342     serverAddrv4.sin_port = htons (plugin->port);
1343     addrlen = sizeof (serverAddrv4);
1344     serverAddr = (struct sockaddr *) &serverAddrv4;
1345 #if DEBUG_UDP
1346     LOG (GNUNET_ERROR_TYPE_DEBUG, "Binding to IPv4 port %d\n",
1347                 ntohs (serverAddrv4.sin_port));
1348 #endif
1349     tries = 0;
1350     while (GNUNET_NETWORK_socket_bind (plugin->sockv4, serverAddr, addrlen) !=
1351            GNUNET_OK)
1352     {
1353       serverAddrv4.sin_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000);   /* Find a good, non-root port */
1354 #if DEBUG_UDP
1355       LOG (GNUNET_ERROR_TYPE_DEBUG,
1356                   "IPv4 Binding failed, trying new port %d\n",
1357                   ntohs (serverAddrv4.sin_port));
1358 #endif
1359       tries++;
1360       if (tries > 10)
1361       {
1362         GNUNET_NETWORK_socket_close (plugin->sockv4);
1363         plugin->sockv4 = NULL;
1364         break;
1365       }
1366     }
1367     if (plugin->sockv4 != NULL)
1368     {
1369       addrs[sockets_created] = (struct sockaddr *) &serverAddrv4;
1370       addrlens[sockets_created] = sizeof (serverAddrv4);
1371       sockets_created++;
1372     }
1373   }
1374
1375   plugin->rs = GNUNET_NETWORK_fdset_create ();
1376   GNUNET_NETWORK_fdset_zero (plugin->rs);
1377   if (NULL != plugin->sockv4)
1378     GNUNET_NETWORK_fdset_set (plugin->rs, plugin->sockv4);
1379   if (NULL != plugin->sockv6)
1380     GNUNET_NETWORK_fdset_set (plugin->rs, plugin->sockv6);
1381
1382   plugin->select_task =
1383       GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1384                                    GNUNET_SCHEDULER_NO_TASK,
1385                                    GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
1386                                    NULL, &udp_plugin_select, plugin);
1387   if (sockets_created == 0)
1388     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, _("Failed to open UDP sockets\n"));
1389   plugin->nat =
1390       GNUNET_NAT_register (env->cfg, GNUNET_NO, port, sockets_created,
1391                            (const struct sockaddr **) addrs, addrlens,
1392                            &udp_nat_port_map_callback, NULL, plugin);
1393   return api;
1394 }
1395
1396 /**
1397  * Shutdown the plugin.
1398  *
1399  * @param cls our 'struct GNUNET_TRANSPORT_PluginFunctions'
1400  * @return NULL
1401  */
1402 void *
1403 libgnunet_plugin_transport_udp_done (void *cls)
1404 {
1405   struct GNUNET_TRANSPORT_PluginFunctions *api = cls;
1406   struct Plugin *plugin = api->cls;
1407   struct ReceiveContext *rc;
1408
1409   /* FIXME: clean up heap and hashmap */
1410   GNUNET_CONTAINER_multihashmap_iterate (plugin->sessions, &destroy_session,
1411                                          NULL);
1412   GNUNET_CONTAINER_multihashmap_destroy (plugin->sessions);
1413   plugin->sessions = NULL;
1414   GNUNET_CONTAINER_multihashmap_iterate (plugin->inbound_sessions, &destroy_inbound_session,
1415                                          NULL);
1416   GNUNET_CONTAINER_multihashmap_destroy (plugin->inbound_sessions);
1417   plugin->inbound_sessions = NULL;
1418   while (NULL != (rc = GNUNET_CONTAINER_heap_remove_root (plugin->defrags)))
1419   {
1420     GNUNET_DEFRAGMENT_context_destroy (rc->defrag);
1421     GNUNET_free (rc);
1422   }
1423   GNUNET_CONTAINER_heap_destroy (plugin->defrags);
1424
1425   if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK)
1426   {
1427     GNUNET_SCHEDULER_cancel (plugin->select_task);
1428     plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
1429   }
1430   if (plugin->sockv4 != NULL)
1431   {
1432     GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->sockv4));
1433     plugin->sockv4 = NULL;
1434   }
1435   if (plugin->sockv6 != NULL)
1436   {
1437     GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->sockv6));
1438     plugin->sockv6 = NULL;
1439   }
1440   GNUNET_SERVER_mst_destroy (plugin->mst);
1441   GNUNET_NETWORK_fdset_destroy (plugin->rs);
1442   GNUNET_NAT_unregister (plugin->nat);
1443   plugin->nat = NULL;
1444   GNUNET_free (plugin);
1445   GNUNET_free (api);
1446   return NULL;
1447 }
1448
1449 /* end of plugin_transport_udp.c */