X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Ftransport%2Fplugin_transport_udp.c;h=940bed254f220887257959352ac401bf2afa595d;hb=fc293a0b11c9819e95045dde70e46c160d1cfd60;hp=076fb6b3b91490f3231adb2c275738f0e598c077;hpb=1949a1e7a652f83b117e821d5c965892cf4a4abf;p=oweals%2Fgnunet.git diff --git a/src/transport/plugin_transport_udp.c b/src/transport/plugin_transport_udp.c index 076fb6b3b..940bed254 100644 --- a/src/transport/plugin_transport_udp.c +++ b/src/transport/plugin_transport_udp.c @@ -20,12 +20,13 @@ /** * @file transport/plugin_transport_udp.c - * @brief Implementation of the UDP NAT punching - * transport service + * @brief Implementation of the UDP transport protocol * @author Christian Grothoff * @author Nathan Evans + * @author Matthias Wachs */ #include "platform.h" +#include "plugin_transport_udp.h" #include "gnunet_hello_lib.h" #include "gnunet_util_lib.h" #include "gnunet_fragmentation_lib.h" @@ -42,14 +43,6 @@ #define LOG(kind,...) GNUNET_log_from (kind, "transport-udp", __VA_ARGS__) -#define DEBUG_UDP GNUNET_EXTRA_LOGGING - -/** - * MTU for fragmentation subsystem. Should be conservative since - * all communicating peers MUST work with this MTU. - */ -#define UDP_MTU 1400 - /** * Number of messages we can defragment in parallel. We only really * defragment 1 message at a time, but if messages get re-ordered, we @@ -70,166 +63,136 @@ #define UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG 128 + /** - * UDP Message-Packet header (after defragmentation). + * Closure for 'append_port'. */ -struct UDPMessage +struct PrettyPrinterContext { /** - * Message header. + * Function to call with the result. */ - struct GNUNET_MessageHeader header; + GNUNET_TRANSPORT_AddressStringCallback asc; /** - * Always zero for now. + * Clsoure for 'asc'. */ - uint32_t reserved; + void *asc_cls; /** - * What is the identity of the sender + * Port to add after the IP address. */ - struct GNUNET_PeerIdentity sender; - + uint16_t port; }; -/** - * UDP ACK Message-Packet header (after defragmentation). - */ -struct UDP_ACK_Message +struct Session { /** - * Message header. + * Which peer is this session for? */ - struct GNUNET_MessageHeader header; + struct GNUNET_PeerIdentity target; + + struct FragmentationContext * frag_ctx; /** - * Desired delay for flow control + * Address of the other peer */ - uint32_t delay; + const struct sockaddr *sock_addr; /** - * What is the identity of the sender + * Desired delay for next sending we send to other peer */ - struct GNUNET_PeerIdentity sender; -}; - - -struct UDP_Beacon_Message -{ - /** - * Message header. - */ - struct GNUNET_MessageHeader header; - - /** - * What is the identity of the sender - */ - struct GNUNET_PeerIdentity sender; -}; + struct GNUNET_TIME_Relative flow_delay_for_other_peer; + /** + * Desired delay for next sending we received from other peer + */ + struct GNUNET_TIME_Absolute flow_delay_from_other_peer; -/** - * Network format for IPv4 addresses. - */ -struct IPv4UdpAddress -{ /** - * IPv4 address, in network byte order. + * Session timeout task */ - uint32_t ipv4_addr GNUNET_PACKED; + GNUNET_SCHEDULER_TaskIdentifier timeout_task; /** - * Port number, in network byte order. + * expected delay for ACKs */ - uint16_t u4_port GNUNET_PACKED; -}; + struct GNUNET_TIME_Relative last_expected_delay; + struct GNUNET_ATS_Information ats; -/** - * Network format for IPv6 addresses. - */ -struct IPv6UdpAddress -{ + size_t addrlen; - /** - * IPv6 address. - */ - struct in6_addr ipv6_addr GNUNET_PACKED; - /** - * Port number, in network byte order. - */ - uint16_t u6_port GNUNET_PACKED; + unsigned int rc; + + int in_destroy; }; -/* Forward definition */ -struct Plugin; +struct SessionCompareContext +{ + struct Session *res; + const struct GNUNET_HELLO_Address *addr; +}; /** - * Session with another peer. FIXME: why not make this into - * a regular 'struct Session' and pass it around!? + * Closure for 'process_inbound_tokenized_messages' */ -struct Session +struct SourceInformation { - /** - * Which peer is this session for? + * Sender identity. */ - struct GNUNET_PeerIdentity target; + struct GNUNET_PeerIdentity sender; /** - * Pointer to the global plugin struct. + * Source address. */ - struct Plugin *plugin; + const void *arg; + struct Session *session; /** - * Address of the other peer + * Number of bytes in source address. */ - const struct sockaddr *sock_addr; + size_t args; - size_t addrlen; +}; - /** - * Function to call upon completion of the transmission. - */ - GNUNET_TRANSPORT_TransmitContinuation cont; +/** + * Closure for 'find_receive_context'. + */ +struct FindReceiveContext +{ /** - * Closure for 'cont'. + * Where to store the result. */ - void *cont_cls; + struct DefragContext *rc; /** - * Current outgoing message to this peer. + * Address to find. */ - struct GNUNET_FRAGMENT_Context *frag; - - struct GNUNET_TIME_Absolute valid_until; - - GNUNET_SCHEDULER_TaskIdentifier invalidation_task; + const struct sockaddr *addr; - GNUNET_SCHEDULER_TaskIdentifier delayed_cont_task; + struct Session *session; /** - * Desired delay for next sending we send to other peer + * Number of bytes in 'addr'. */ - struct GNUNET_TIME_Relative flow_delay_for_other_peer; + socklen_t addr_len; - /** - * Desired delay for next sending we received from other peer - */ - struct GNUNET_TIME_Absolute flow_delay_from_other_peer; }; + /** * Data structure to track defragmentation contexts based * on the source of the UDP traffic. */ -struct ReceiveContext +struct DefragContext { /** @@ -257,1540 +220,1018 @@ struct ReceiveContext * Length of 'src_addr' */ size_t addr_len; - - struct GNUNET_PeerIdentity id; - }; + /** - * Encapsulation of all of the state of the plugin. + * Closure for 'process_inbound_tokenized_messages' */ -struct Plugin +struct FragmentationContext { + struct FragmentationContext * next; + struct FragmentationContext * prev; - /** - * Our environment. - */ - struct GNUNET_TRANSPORT_PluginEnvironment *env; - - /** - * Session of peers with whom we are currently connected, - * map of peer identity to 'struct PeerSession'. - */ - struct GNUNET_CONTAINER_MultiHashMap *sessions; - - /** - * Session of peers with whom we are currently connected, - * map of peer identity to 'struct PeerSession'. - */ - struct GNUNET_CONTAINER_MultiHashMap *inbound_sessions; - - /** - * Heap with all of our defragmentation activities. - */ - struct GNUNET_CONTAINER_Heap *defrags; - - /** - * ID of select task - */ - GNUNET_SCHEDULER_TaskIdentifier select_task; - - /** - * Tokenizer for inbound messages. - */ - struct GNUNET_SERVER_MessageStreamTokenizer *mst; - - /** - * Bandwidth tracker to limit global UDP traffic. - */ - struct GNUNET_BANDWIDTH_Tracker tracker; - - /** - * Address we were told to bind to exclusively (IPv4). - */ - char *bind4_address; - - /** - * Address we were told to bind to exclusively (IPv6). - */ - char *bind6_address; - - /** - * Handle to NAT traversal support. - */ - struct GNUNET_NAT_Handle *nat; + struct Plugin * plugin; + struct GNUNET_FRAGMENT_Context * frag; + struct Session * session; /** - * FD Read set + * Function to call upon completion of the transmission. */ - struct GNUNET_NETWORK_FDSet *rs; + GNUNET_TRANSPORT_TransmitContinuation cont; /** - * The read socket for IPv4 + * Closure for 'cont'. */ - struct GNUNET_NETWORK_Handle *sockv4; + void *cont_cls; - /** - * The read socket for IPv6 - */ - struct GNUNET_NETWORK_Handle *sockv6; + struct GNUNET_TIME_Absolute timeout; - /** - * Broadcast? - */ - int broadcast; + size_t bytes_to_send; +}; - /** - * Tokenizer for inbound messages. - */ - struct GNUNET_SERVER_MessageStreamTokenizer *broadcast_mst; +struct UDPMessageWrapper +{ + struct Session *session; + struct UDPMessageWrapper *prev; + struct UDPMessageWrapper *next; + char *udp; /** - * The read socket for IPv4 + * Function to call upon completion of the transmission. */ - struct GNUNET_NETWORK_Handle *sockv4_broadcast; + GNUNET_TRANSPORT_TransmitContinuation cont; /** - * ID of select broadcast task + * Closure for 'cont'. */ - GNUNET_SCHEDULER_TaskIdentifier select_broadcast_task; + void *cont_cls; - /** - * ID of select broadcast task - */ - GNUNET_SCHEDULER_TaskIdentifier send_broadcast_task; + struct FragmentationContext *frag_ctx; - /** - * FD Read set - */ - struct GNUNET_NETWORK_FDSet *broadcast_rs; + size_t msg_size; - /** - * Broadcast interval - */ - struct GNUNET_TIME_Relative broadcast_interval; + struct GNUNET_TIME_Absolute timeout; +}; - /** - * expected delay for ACKs - */ - struct GNUNET_TIME_Relative last_expected_delay; +/** + * UDP ACK Message-Packet header (after defragmentation). + */ +struct UDP_ACK_Message +{ /** - * Port we broadcasting on. + * Message header. */ - uint16_t broadcast_port; + struct GNUNET_MessageHeader header; /** - * Port we listen on. + * Desired delay for flow control */ - uint16_t port; + uint32_t delay; /** - * Port we advertise on. + * What is the identity of the sender */ - uint16_t aport; + struct GNUNET_PeerIdentity sender; }; -struct PeerSessionIteratorContext -{ - struct Session * result; - const void * addr; - size_t addrlen; -}; +/** + * Encapsulation of all of the state of the plugin. + */ +struct Plugin * plugin; /** - * Lookup the session for the given peer. + * We have been notified that our readset has something to read. We don't + * know which socket needs to be read, so we have to check each one + * Then reschedule this function to be called again once more is available. * - * @param plugin the plugin - * @param peer peer's identity - * @return NULL if we have no session + * @param cls the plugin handle + * @param tc the scheduling context (for rescheduling this function again) */ -static struct Session * -find_session (struct Plugin *plugin, const struct GNUNET_PeerIdentity *peer) -{ - return GNUNET_CONTAINER_multihashmap_get (plugin->sessions, - &peer->hashPubKey); -} - - -static int -inbound_session_iterator (void *cls, - const GNUNET_HashCode * key, - void *value) -{ - struct PeerSessionIteratorContext *psc = cls; - struct Session *s = value; - if (s->addrlen == psc->addrlen) - { - if (0 == memcmp (&s[1], psc->addr, s->addrlen)) - psc->result = s; - } - if (psc->result != NULL) - return GNUNET_NO; - return GNUNET_YES; -} +static void +udp_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); /** - * Lookup the session for the given peer. + * We have been notified that our readset has something to read. We don't + * know which socket needs to be read, so we have to check each one + * Then reschedule this function to be called again once more is available. * - * @param plugin the plugin - * @param peer peer's identity - * @return NULL if we have no session + * @param cls the plugin handle + * @param tc the scheduling context (for rescheduling this function again) */ -static struct Session * -find_inbound_session (struct Plugin *plugin, - const struct GNUNET_PeerIdentity *peer, - const void * addr, size_t addrlen) -{ - struct PeerSessionIteratorContext psc; - psc.result = NULL; - psc.addrlen = addrlen; - psc.addr = addr; +static void +udp_plugin_select_v6 (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); - GNUNET_CONTAINER_multihashmap_get_multiple(plugin->inbound_sessions, &peer->hashPubKey, &inbound_session_iterator, &psc); - return psc.result; -} +/** + * Start session timeout + */ +static void +start_session_timeout (struct Session *s); -static int -inbound_session_by_addr_iterator (void *cls, - const GNUNET_HashCode * key, - void *value) -{ - struct PeerSessionIteratorContext *psc = cls; - struct Session *s = value; - if (s->addrlen == psc->addrlen) - { - if (0 == memcmp (&s[1], psc->addr, s->addrlen)) - psc->result = s; - } - if (psc->result != NULL) - return GNUNET_NO; - else - return GNUNET_YES; -}; +/** + * Increment session timeout due to activity + */ +static void +reschedule_session_timeout (struct Session *s); /** - * Lookup the session for the given peer just by address. - * - * @param plugin the plugin - * @param addr address - * @param addrlen address length - * @return NULL if we have no session + * Cancel timeout */ -static struct Session * -find_inbound_session_by_addr (struct Plugin *plugin, const void * addr, size_t addrlen) -{ - struct PeerSessionIteratorContext psc; - psc.result = NULL; - psc.addrlen = addrlen; - psc.addr = addr; +static void +stop_session_timeout (struct Session *s); - GNUNET_CONTAINER_multihashmap_iterate (plugin->inbound_sessions, &inbound_session_by_addr_iterator, &psc); - return psc.result; -} /** - * Destroy a session, plugin is being unloaded. + * Function called for a quick conversion of the binary address to + * a numeric address. Note that the caller must not free the + * address and that the next call to this function is allowed + * to override the address again. * - * @param cls unused - * @param key hash of public key of target peer - * @param value a 'struct PeerSession*' to clean up - * @return GNUNET_OK (continue to iterate) + * @param cls closure + * @param addr binary address + * @param addrlen length of the address + * @return string representing the same address */ -static int -destroy_session (void *cls, const GNUNET_HashCode * key, void *value) +const char * +udp_address_to_string (void *cls, const void *addr, size_t addrlen) { - struct Session *peer_session = value; + static char rbuf[INET6_ADDRSTRLEN + 10]; + char buf[INET6_ADDRSTRLEN]; + const void *sb; + struct in_addr a4; + struct in6_addr a6; + const struct IPv4UdpAddress *t4; + const struct IPv6UdpAddress *t6; + int af; + uint16_t port; - if (peer_session->frag != NULL) - GNUNET_FRAGMENT_context_destroy (peer_session->frag); - if (GNUNET_SCHEDULER_NO_TASK != peer_session->delayed_cont_task) - GNUNET_SCHEDULER_cancel (peer_session->delayed_cont_task); - GNUNET_free (peer_session); - return GNUNET_OK; + if (addrlen == sizeof (struct IPv6UdpAddress)) + { + t6 = addr; + af = AF_INET6; + port = ntohs (t6->u6_port); + memcpy (&a6, &t6->ipv6_addr, sizeof (a6)); + sb = &a6; + } + else if (addrlen == sizeof (struct IPv4UdpAddress)) + { + t4 = addr; + af = AF_INET; + port = ntohs (t4->u4_port); + memcpy (&a4, &t4->ipv4_addr, sizeof (a4)); + sb = &a4; + } + else + { + GNUNET_break_op (0); + return NULL; + } + inet_ntop (af, sb, buf, INET6_ADDRSTRLEN); + GNUNET_snprintf (rbuf, sizeof (rbuf), (af == AF_INET6) ? "[%s]:%u" : "%s:%u", + buf, port); + return rbuf; } -/** - * Destroy a session, plugin is being unloaded. - * - * @param cls unused - * @param key hash of public key of target peer - * @param value a 'struct PeerSession*' to clean up - * @return GNUNET_OK (continue to iterate) - */ -static int -destroy_inbound_session (void *cls, const GNUNET_HashCode * key, void *value) -{ - struct Session *s = value; - - if (s->invalidation_task != GNUNET_SCHEDULER_NO_TASK) - GNUNET_SCHEDULER_cancel(s->invalidation_task); - if (GNUNET_SCHEDULER_NO_TASK != s->delayed_cont_task) - GNUNET_SCHEDULER_cancel (s->delayed_cont_task); - GNUNET_free (s); - return GNUNET_OK; -} /** - * Disconnect from a remote node. Clean up session if we have one for this peer + * Function called to convert a string address to + * a binary address. * - * @param cls closure for this call (should be handle to Plugin) - * @param target the peeridentity of the peer to disconnect - * @return GNUNET_OK on success, GNUNET_SYSERR if the operation failed + * @param cls closure ('struct Plugin*') + * @param addr string address + * @param addrlen length of the address + * @param buf location to store the buffer + * @param added location to store the number of bytes in the buffer. + * If the function returns GNUNET_SYSERR, its contents are undefined. + * @return GNUNET_OK on success, GNUNET_SYSERR on failure */ -static void -udp_disconnect (void *cls, const struct GNUNET_PeerIdentity *target) +static int +udp_string_to_address (void *cls, const char *addr, uint16_t addrlen, + void **buf, size_t *added) { - struct Plugin *plugin = cls; - struct Session *session; + struct sockaddr_storage socket_address; + + if ((NULL == addr) || (0 == addrlen)) + { + GNUNET_break (0); + return GNUNET_SYSERR; + } - session = find_session (plugin, target); - if (NULL == session) - return; - GNUNET_assert (GNUNET_OK == - GNUNET_CONTAINER_multihashmap_remove (plugin->sessions, - &target->hashPubKey, - session)); - - GNUNET_CONTAINER_multihashmap_get_multiple (plugin->inbound_sessions, - &target->hashPubKey, - &destroy_inbound_session, NULL); - GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "sendto"); - LOG (GNUNET_ERROR_TYPE_ERROR, - "UDP DISCONNECT\n"); - - plugin->last_expected_delay = GNUNET_FRAGMENT_context_destroy (session->frag); - if (GNUNET_SCHEDULER_NO_TASK != session->delayed_cont_task) - GNUNET_SCHEDULER_cancel (session->delayed_cont_task); - if (session->cont != NULL) - session->cont (session->cont_cls, target, GNUNET_SYSERR); - GNUNET_free (session); -} + if ('\0' != addr[addrlen - 1]) + { + return GNUNET_SYSERR; + } + if (strlen (addr) != addrlen - 1) + { + return GNUNET_SYSERR; + } -/** - * Actually send out the message. - * - * @param plugin the plugin - * @param sa the address to send the message to - * @param msg message to transmit - * @return the number of bytes written - */ -static ssize_t -udp_send (struct Plugin *plugin, const struct sockaddr *sa, - const struct GNUNET_MessageHeader *msg) -{ - ssize_t sent; - size_t slen; + if (GNUNET_OK != GNUNET_STRINGS_to_address_ip (addr, strlen (addr), + &socket_address)) + { + return GNUNET_SYSERR; + } - switch (sa->sa_family) + switch (socket_address.ss_family) { case AF_INET: - if (NULL == plugin->sockv4) - return 0; - sent = - GNUNET_NETWORK_socket_sendto (plugin->sockv4, msg, ntohs (msg->size), - sa, slen = sizeof (struct sockaddr_in)); - break; + { + struct IPv4UdpAddress *u4; + struct sockaddr_in *in4 = (struct sockaddr_in *) &socket_address; + u4 = GNUNET_malloc (sizeof (struct IPv4UdpAddress)); + u4->ipv4_addr = in4->sin_addr.s_addr; + u4->u4_port = in4->sin_port; + *buf = u4; + *added = sizeof (struct IPv4UdpAddress); + return GNUNET_OK; + } case AF_INET6: - if (NULL == plugin->sockv6) - return 0; - sent = - GNUNET_NETWORK_socket_sendto (plugin->sockv6, msg, ntohs (msg->size), - sa, slen = sizeof (struct sockaddr_in6)); - break; + { + struct IPv6UdpAddress *u6; + struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &socket_address; + u6 = GNUNET_malloc (sizeof (struct IPv6UdpAddress)); + u6->ipv6_addr = in6->sin6_addr; + u6->u6_port = in6->sin6_port; + *buf = u6; + *added = sizeof (struct IPv6UdpAddress); + return GNUNET_OK; + } default: GNUNET_break (0); - return 0; - } - if (GNUNET_SYSERR == sent) - { - GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "sendto"); - LOG (GNUNET_ERROR_TYPE_ERROR, - "UDP transmited %u-byte message to %s (%d: %s)\n", - (unsigned int) ntohs (msg->size), GNUNET_a2s (sa, slen), - (int) sent, (sent < 0) ? STRERROR (errno) : "ok"); - + return GNUNET_SYSERR; } - LOG (GNUNET_ERROR_TYPE_DEBUG, - "UDP transmited %u-byte message to %s (%d: %s)\n", - (unsigned int) ntohs (msg->size), GNUNET_a2s (sa, slen), - (int) sent, (sent < 0) ? STRERROR (errno) : "ok"); - return sent; } /** - * Function that is called with messages created by the fragmentation - * module. In the case of the 'proc' callback of the - * GNUNET_FRAGMENT_context_create function, this function must - * eventually call 'GNUNET_FRAGMENT_context_transmission_done'. + * Append our port and forward the result. * - * @param cls closure, the 'struct PeerSession' - * @param msg the message that was created + * @param cls a 'struct PrettyPrinterContext' + * @param hostname result from DNS resolver */ static void -send_fragment (void *cls, const struct GNUNET_MessageHeader *msg) +append_port (void *cls, const char *hostname) { - struct Session *session = cls; + struct PrettyPrinterContext *ppc = cls; + char *ret; - udp_send (session->plugin, session->sock_addr, msg); - GNUNET_FRAGMENT_context_transmission_done (session->frag); + if (hostname == NULL) + { + ppc->asc (ppc->asc_cls, NULL); + GNUNET_free (ppc); + return; + } + GNUNET_asprintf (&ret, "%s:%d", hostname, ppc->port); + ppc->asc (ppc->asc_cls, ret); + GNUNET_free (ret); } -static struct Session * -create_session (struct Plugin *plugin, const struct GNUNET_PeerIdentity *target, - const void *addr, size_t addrlen, - GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls) +/** + * Convert the transports address to a nice, human-readable + * format. + * + * @param cls closure + * @param type name of the transport that generated the address + * @param addr one of the addresses of the host, NULL for the last address + * the specific address format depends on the transport + * @param addrlen length of the address + * @param numeric should (IP) addresses be displayed in numeric form? + * @param timeout after how long should we give up? + * @param asc function to call on each string + * @param asc_cls closure for asc + */ +static void +udp_plugin_address_pretty_printer (void *cls, const char *type, + const void *addr, size_t addrlen, + int numeric, + struct GNUNET_TIME_Relative timeout, + GNUNET_TRANSPORT_AddressStringCallback asc, + void *asc_cls) { - struct Session * peer_session; - const struct IPv4UdpAddress *t4; - const struct IPv6UdpAddress *t6; - struct sockaddr_in *v4; - struct sockaddr_in6 *v6; - size_t len; + struct PrettyPrinterContext *ppc; + const void *sb; + size_t sbs; + struct sockaddr_in a4; + struct sockaddr_in6 a6; + const struct IPv4UdpAddress *u4; + const struct IPv6UdpAddress *u6; + uint16_t port; - switch (addrlen) + if (addrlen == sizeof (struct IPv6UdpAddress)) { - case sizeof (struct IPv4UdpAddress): - if (NULL == plugin->sockv4) - { - return NULL; - } - t4 = addr; - peer_session = - GNUNET_malloc (sizeof (struct Session) + - sizeof (struct sockaddr_in)); - len = sizeof (struct sockaddr_in); - v4 = (struct sockaddr_in *) &peer_session[1]; - v4->sin_family = AF_INET; + u6 = addr; + memset (&a6, 0, sizeof (a6)); + a6.sin6_family = AF_INET6; #if HAVE_SOCKADDR_IN_SIN_LEN - v4->sin_len = sizeof (struct sockaddr_in); + a6.sin6_len = sizeof (a6); #endif - v4->sin_port = t4->u4_port; - v4->sin_addr.s_addr = t4->ipv4_addr; - break; - case sizeof (struct IPv6UdpAddress): - if (NULL == plugin->sockv6) - { - return NULL; - } - t6 = addr; - peer_session = - GNUNET_malloc (sizeof (struct Session) + - sizeof (struct sockaddr_in6)); - len = sizeof (struct sockaddr_in6); - v6 = (struct sockaddr_in6 *) &peer_session[1]; - v6->sin6_family = AF_INET6; + a6.sin6_port = u6->u6_port; + memcpy (&a6.sin6_addr, &u6->ipv6_addr, sizeof (struct in6_addr)); + port = ntohs (u6->u6_port); + sb = &a6; + sbs = sizeof (a6); + } + else if (addrlen == sizeof (struct IPv4UdpAddress)) + { + u4 = addr; + memset (&a4, 0, sizeof (a4)); + a4.sin_family = AF_INET; #if HAVE_SOCKADDR_IN_SIN_LEN - v6->sin6_len = sizeof (struct sockaddr_in6); + a4.sin_len = sizeof (a4); #endif - v6->sin6_port = t6->u6_port; - v6->sin6_addr = t6->ipv6_addr; - break; - default: - /* Must have a valid address to send to */ + a4.sin_port = u4->u4_port; + a4.sin_addr.s_addr = u4->ipv4_addr; + port = ntohs (u4->u4_port); + sb = &a4; + sbs = sizeof (a4); + } + else if (0 == addrlen) + { + asc (asc_cls, ""); + asc (asc_cls, NULL); + return; + } + else + { + /* invalid address */ GNUNET_break_op (0); - return NULL; + asc (asc_cls, NULL); + return; } - - peer_session->valid_until = GNUNET_TIME_absolute_get_zero (); - peer_session->invalidation_task = GNUNET_SCHEDULER_NO_TASK; - peer_session->addrlen = len; - peer_session->target = *target; - peer_session->plugin = plugin; - peer_session->sock_addr = (const struct sockaddr *) &peer_session[1]; - peer_session->cont = cont; - peer_session->cont_cls = cont_cls; - - return peer_session; + ppc = GNUNET_malloc (sizeof (struct PrettyPrinterContext)); + ppc->asc = asc; + ppc->asc_cls = asc_cls; + ppc->port = port; + GNUNET_RESOLVER_hostname_get (sb, sbs, !numeric, timeout, &append_port, ppc); } -static const char * -udp_address_to_string (void *cls, const void *addr, size_t addrlen); - static void -udp_call_continuation (void *cls, - const struct GNUNET_SCHEDULER_TaskContext *tc) +call_continuation (struct UDPMessageWrapper *udpw, int result) { - struct Session *s = cls; - GNUNET_TRANSPORT_TransmitContinuation cont = s->cont; + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Calling continuation for %u byte message to `%s' with result %s\n", + udpw->msg_size, GNUNET_i2s (&udpw->session->target), + (GNUNET_OK == result) ? "OK" : "SYSERR"); + if (NULL != udpw->cont) + { + udpw->cont (udpw->cont_cls, &udpw->session->target,result); + } - s->delayed_cont_task = GNUNET_SCHEDULER_NO_TASK; - s->cont = NULL; - cont (s->cont_cls, &s->target, GNUNET_OK); } /** - * Function that can be used by the transport service to transmit - * a message using the plugin. - * - * @param cls closure - * @param target who should receive this message (ignored by UDP) - * @param msgbuf one or more GNUNET_MessageHeader(s) strung together - * @param msgbuf_size the size of the msgbuf to send - * @param priority how important is the message (ignored by UDP) - * @param timeout when should we time out (give up) if we can not transmit? - * @param session identifier used for this session (NULL for UDP) - * @param addr the addr to send the message to - * @param addrlen the len of addr - * @param force_address not used, we had better have an address to send to - * because we are stateless!! - * @param cont continuation to call once the message has - * been transmitted (or if the transport is ready - * for the next transmission call; or if the - * peer disconnected...) - * @param cont_cls closure for cont + * Check if the given port is plausible (must be either our listen + * port or our advertised port). If it is neither, we return + * GNUNET_SYSERR. * - * @return the number of bytes written (may return 0 and the message can - * still be transmitted later!) + * @param plugin global variables + * @param in_port port number to check + * @return GNUNET_OK if port is either open_port or adv_port */ -static ssize_t -udp_plugin_send (void *cls, const struct GNUNET_PeerIdentity *target, - const char *msgbuf, size_t msgbuf_size, unsigned int priority, - struct GNUNET_TIME_Relative timeout, struct Session *session, - const void *addr, size_t addrlen, int force_address, - GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls) +static int +check_port (struct Plugin *plugin, uint16_t in_port) { - struct Plugin *plugin = cls; - struct Session *peer_session; - struct Session *s; - const struct IPv4UdpAddress *t4; - const struct IPv6UdpAddress *t6; - size_t mlen = msgbuf_size + sizeof (struct UDPMessage); - char mbuf[mlen]; - struct UDPMessage *udp; - struct GNUNET_TIME_Relative delta; - - if (mlen >= GNUNET_SERVER_MAX_MESSAGE_SIZE) - { - GNUNET_break (0); - return GNUNET_SYSERR; - } - - LOG (GNUNET_ERROR_TYPE_DEBUG, - "UDP transmits %u-byte message to `%s' using address `%s' session 0x%X mode %i\n", - msgbuf_size, GNUNET_i2s (target), - udp_address_to_string (NULL, addr, addrlen), - session, force_address); - - if ((force_address == GNUNET_SYSERR) && (session == NULL)) - return GNUNET_SYSERR; + if ((in_port == plugin->port) || (in_port == plugin->aport)) + return GNUNET_OK; + return GNUNET_SYSERR; +} - s = NULL; - /* safety check: comparing address to address stored in session */ - if ((session != NULL) && (addr != NULL) && (addrlen != 0)) - { - s = session; - GNUNET_assert (GNUNET_YES == GNUNET_CONTAINER_multihashmap_contains_value ( - plugin->inbound_sessions, &target->hashPubKey, s)); - if (0 != memcmp (&s->target, target, sizeof (struct GNUNET_PeerIdentity))) - return GNUNET_SYSERR; - switch (addrlen) - { - case sizeof (struct IPv4UdpAddress): - if (NULL == plugin->sockv4) - { - if (cont != NULL) - cont (cont_cls, target, GNUNET_SYSERR); - return GNUNET_SYSERR; - } - t4 = addr; - if (s->addrlen != (sizeof (struct sockaddr_in))) - return GNUNET_SYSERR; - struct sockaddr_in *a4 = (struct sockaddr_in *) s->sock_addr; - GNUNET_assert (a4->sin_port == t4->u4_port); - GNUNET_assert (0 == memcmp(&a4->sin_addr, &t4->ipv4_addr, sizeof (struct in_addr))); - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Session 0x%X successfully checked!\n", session); - break; - case sizeof (struct IPv6UdpAddress): - if (NULL == plugin->sockv6) - { - if (cont != NULL) - cont (cont_cls, target, GNUNET_SYSERR); - return GNUNET_SYSERR; - } - t6 = addr; - GNUNET_assert (s->addrlen == sizeof (struct sockaddr_in6)); - struct sockaddr_in6 *a6 = (struct sockaddr_in6 *) s->sock_addr; - GNUNET_assert (a6->sin6_port == t6->u6_port); - GNUNET_assert (0 == memcmp(&a6->sin6_addr, &t6->ipv6_addr, sizeof (struct in6_addr))); - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Session 0x%X successfully checked!\n", session); - break; - default: - /* Must have a valid address to send to */ - GNUNET_break_op (0); - } - } -//session_invalid: - if ((addr == NULL) || (addrlen == 0)) +/** + * Function that will be called to check if a binary address for this + * plugin is well-formed and corresponds to an address for THIS peer + * (as per our configuration). Naturally, if absolutely necessary, + * plugins can be a bit conservative in their answer, but in general + * plugins should make sure that the address does not redirect + * traffic to a 3rd party that might try to man-in-the-middle our + * traffic. + * + * @param cls closure, should be our handle to the Plugin + * @param addr pointer to the address + * @param addrlen length of addr + * @return GNUNET_OK if this is a plausible address for this peer + * and transport, GNUNET_SYSERR if not + * + */ +static int +udp_plugin_check_address (void *cls, const void *addr, size_t addrlen) +{ + struct Plugin *plugin = cls; + struct IPv4UdpAddress *v4; + struct IPv6UdpAddress *v6; + + if ((addrlen != sizeof (struct IPv4UdpAddress)) && + (addrlen != sizeof (struct IPv6UdpAddress))) + { + GNUNET_break_op (0); return GNUNET_SYSERR; - peer_session = create_session (plugin, target, addr, addrlen, cont, cont_cls); - if (peer_session == NULL) + } + if (addrlen == sizeof (struct IPv4UdpAddress)) { - if (cont != NULL) - cont (cont_cls, target, GNUNET_SYSERR); - return GNUNET_SYSERR;; + v4 = (struct IPv4UdpAddress *) addr; + if (GNUNET_OK != check_port (plugin, ntohs (v4->u4_port))) + return GNUNET_SYSERR; + if (GNUNET_OK != + GNUNET_NAT_test_address (plugin->nat, &v4->ipv4_addr, + sizeof (struct in_addr))) + return GNUNET_SYSERR; } - - /* Message */ - udp = (struct UDPMessage *) mbuf; - udp->header.size = htons (mlen); - udp->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE); - udp->reserved = htonl (0); - udp->sender = *plugin->env->my_identity; - memcpy (&udp[1], msgbuf, msgbuf_size); - - if (s != NULL) - delta = GNUNET_TIME_absolute_get_remaining (s->flow_delay_from_other_peer); else - delta = GNUNET_TIME_UNIT_ZERO; - if (mlen <= UDP_MTU) { - mlen = udp_send (plugin, peer_session->sock_addr, &udp->header); - if (cont != NULL) + v6 = (struct IPv6UdpAddress *) addr; + if (IN6_IS_ADDR_LINKLOCAL (&v6->ipv6_addr)) { - if ( (delta.rel_value > 0) && - (mlen > 0) ) - { - s->cont = cont; - s->cont_cls = cont_cls; - s->delayed_cont_task = GNUNET_SCHEDULER_add_delayed (delta, - &udp_call_continuation, - s); - } - else - cont (cont_cls, target, (mlen > 0) ? GNUNET_OK : GNUNET_SYSERR); + GNUNET_break_op (0); + return GNUNET_SYSERR; } - GNUNET_free_non_null (peer_session); - } - else - { - GNUNET_assert (GNUNET_OK == - GNUNET_CONTAINER_multihashmap_put (plugin->sessions, - &target->hashPubKey, - peer_session, - GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY)); - peer_session->frag = - GNUNET_FRAGMENT_context_create (plugin->env->stats, UDP_MTU, - &plugin->tracker, - plugin->last_expected_delay, - &udp->header, &send_fragment, - peer_session); + if (GNUNET_OK != check_port (plugin, ntohs (v6->u6_port))) + return GNUNET_SYSERR; + if (GNUNET_OK != + GNUNET_NAT_test_address (plugin->nat, &v6->ipv6_addr, + sizeof (struct in6_addr))) + return GNUNET_SYSERR; } - return mlen; + return GNUNET_OK; } /** - * Closure for 'process_inbound_tokenized_messages' - */ -struct SourceInformation -{ - /** - * Sender identity. - */ - struct GNUNET_PeerIdentity sender; - - /** - * Source address. - */ - const void *arg; - - /** - * Number of bytes in source address. - */ - size_t args; - - struct Session * session; -}; - - -/** - * Message tokenizer has broken up an incomming message. Pass it on - * to the service. + * Task to free resources associated with a session. * - * @param cls the 'struct Plugin' - * @param client the 'struct SourceInformation' - * @param hdr the actual message + * @param s session to free */ static void -process_inbound_tokenized_messages (void *cls, void *client, - const struct GNUNET_MessageHeader *hdr) -{ - struct Plugin *plugin = cls; - struct SourceInformation *si = client; - struct GNUNET_ATS_Information distance; - struct GNUNET_TIME_Relative delay; - - /* setup ATS */ - distance.type = htonl (GNUNET_ATS_QUALITY_NET_DISTANCE); - distance.value = htonl (1); - - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Giving Session %X %s to transport\n", si->session, GNUNET_i2s(&si->session->target)); - delay = plugin->env->receive (plugin->env->cls, &si->sender, hdr, &distance, 1, si->session, - si->arg, si->args); - si->session->flow_delay_for_other_peer = delay; -} - -static void -invalidation_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +free_session (struct Session *s) { - struct Session * s = cls; - s->invalidation_task = GNUNET_SCHEDULER_NO_TASK; - LOG (GNUNET_ERROR_TYPE_ERROR, - "Session %X (`%s') is now invalid\n", s, GNUNET_a2s (s->sock_addr,s->addrlen)); - - s->plugin->env->session_end(s->plugin->env->cls, &s->target, s); - GNUNET_assert (GNUNET_YES == GNUNET_CONTAINER_multihashmap_remove(s->plugin->inbound_sessions, &s->target.hashPubKey, s)); + if (s->frag_ctx != NULL) + { + GNUNET_FRAGMENT_context_destroy(s->frag_ctx->frag); + GNUNET_free (s->frag_ctx); + s->frag_ctx = NULL; + } GNUNET_free (s); } /** - * We've received a UDP Message. Process it (pass contents to main service). + * Functions with this signature are called whenever we need + * to close a session due to a disconnect or failure to + * establish a connection. * - * @param plugin plugin context - * @param msg the message - * @param sender_addr sender address - * @param sender_addr_len number of bytes in sender_addr + * @param s session to close down */ static void -process_udp_message (struct Plugin *plugin, const struct UDPMessage *msg, - const struct sockaddr *sender_addr, - socklen_t sender_addr_len) +disconnect_session (struct Session *s) { - struct SourceInformation si; - struct IPv4UdpAddress u4; - struct IPv6UdpAddress u6; - const void *arg; - size_t args; + struct UDPMessageWrapper *udpw; + struct UDPMessageWrapper *next; - if (0 != ntohl (msg->reserved)) + GNUNET_assert (GNUNET_YES != s->in_destroy); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Session %p to peer `%s' address ended \n", + s, + GNUNET_i2s (&s->target), + GNUNET_a2s (s->sock_addr, s->addrlen)); + stop_session_timeout (s); + next = plugin->ipv4_queue_head; + while (NULL != (udpw = next)) { - GNUNET_break_op (0); - return; + next = udpw->next; + if (udpw->session == s) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + call_continuation(udpw, GNUNET_SYSERR); + GNUNET_free (udpw); + } } - if (ntohs (msg->header.size) < - sizeof (struct GNUNET_MessageHeader) + sizeof (struct UDPMessage)) + next = plugin->ipv6_queue_head; + while (NULL != (udpw = next)) { - GNUNET_break_op (0); - return; + next = udpw->next; + if (udpw->session == s) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); + call_continuation(udpw, GNUNET_SYSERR); + GNUNET_free (udpw); + } + udpw = next; } + plugin->env->session_end (plugin->env->cls, &s->target, s); - /* convert address */ - switch (sender_addr->sa_family) + if (NULL != s->frag_ctx) { - case AF_INET: - GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in)); - u4.ipv4_addr = ((struct sockaddr_in *) sender_addr)->sin_addr.s_addr; - u4.u4_port = ((struct sockaddr_in *) sender_addr)->sin_port; - arg = &u4; - args = sizeof (u4); - break; - case AF_INET6: - GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in6)); - u6.ipv6_addr = ((struct sockaddr_in6 *) sender_addr)->sin6_addr; - u6.u6_port = ((struct sockaddr_in6 *) sender_addr)->sin6_port; - arg = &u6; - args = sizeof (u6); - break; - default: - GNUNET_break (0); - return; + if (NULL != s->frag_ctx->cont) + { + s->frag_ctx->cont (s->frag_ctx->cont_cls, &s->target, GNUNET_SYSERR); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Calling continuation for fragemented message to `%s' with result SYSERR\n", + GNUNET_i2s (&s->target)); + } } -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Received message with %u bytes from peer `%s' at `%s'\n", - (unsigned int) ntohs (msg->header.size), - GNUNET_i2s (&msg->sender), GNUNET_a2s (sender_addr, - sender_addr_len)); -#endif - /* create a session for inbound connections */ - const struct UDPMessage * udp_msg = (const struct UDPMessage *) msg; - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Lookup inbound UDP sessions for peer `%s' address `%s'\n", - GNUNET_i2s (&udp_msg->sender), - udp_address_to_string(NULL, arg, args)); - - struct Session * s = NULL; - s = find_inbound_session (plugin, &udp_msg->sender, sender_addr, sender_addr_len); - - if (s != NULL) - { - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Found existing inbound UDP sessions 0x%X for peer `%s' address `%s'\n", - s, - GNUNET_i2s (&s->target), - udp_address_to_string(NULL, arg, args)); - } + GNUNET_assert (GNUNET_YES == + GNUNET_CONTAINER_multihashmap_remove (plugin->sessions, + &s->target.hashPubKey, + s)); + GNUNET_STATISTICS_set(plugin->env->stats, + "# UDP sessions active", + GNUNET_CONTAINER_multihashmap_size(plugin->sessions), + GNUNET_NO); + if (s->rc > 0) + s->in_destroy = GNUNET_YES; else - { - s = create_session (plugin, &udp_msg->sender, arg, args, NULL, NULL); - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Creating inbound UDP sessions 0x%X for peer `%s' address `%s'\n", - s, - GNUNET_i2s (&s->target), - udp_address_to_string(NULL, arg, args)); - - GNUNET_assert (GNUNET_OK == GNUNET_CONTAINER_multihashmap_put (plugin->inbound_sessions, - &s->target.hashPubKey, - s, - GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE)); - } - s->valid_until = GNUNET_TIME_relative_to_absolute (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT); - if (s->invalidation_task != GNUNET_SCHEDULER_NO_TASK) - { - GNUNET_SCHEDULER_cancel(s->invalidation_task); - s->invalidation_task = GNUNET_SCHEDULER_NO_TASK; - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Rescheduling %X' `%s'\n", - s, udp_address_to_string(NULL, arg, args)); - } - s->invalidation_task = GNUNET_SCHEDULER_add_delayed(GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, &invalidation_task, s); - /* iterate over all embedded messages */ - si.sender = msg->sender; - si.arg = arg; - si.args = args; - si.session = s; - GNUNET_SERVER_mst_receive (plugin->mst, &si, (const char *) &msg[1], - ntohs (msg->header.size) - - sizeof (struct UDPMessage), GNUNET_YES, GNUNET_NO); + free_session (s); } - /** - * Process a defragmented message. + * Destroy a session, plugin is being unloaded. * - * @param cls the 'struct ReceiveContext' - * @param msg the message + * @param cls unused + * @param key hash of public key of target peer + * @param value a 'struct PeerSession*' to clean up + * @return GNUNET_OK (continue to iterate) */ -static void -fragment_msg_proc (void *cls, const struct GNUNET_MessageHeader *msg) +static int +disconnect_and_free_it (void *cls, const struct GNUNET_HashCode * key, void *value) { - struct ReceiveContext *rc = cls; - - if (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE) - { - GNUNET_break (0); - return; - } - if (ntohs (msg->size) < sizeof (struct UDPMessage)) - { - GNUNET_break (0); - return; - } - process_udp_message (rc->plugin, (const struct UDPMessage *) msg, - rc->src_addr, rc->addr_len); + disconnect_session(value); + return GNUNET_OK; } /** - * Transmit an acknowledgement. + * Disconnect from a remote node. Clean up session if we have one for this peer * - * @param cls the 'struct ReceiveContext' - * @param id message ID (unused) - * @param msg ack to transmit + * @param cls closure for this call (should be handle to Plugin) + * @param target the peeridentity of the peer to disconnect + * @return GNUNET_OK on success, GNUNET_SYSERR if the operation failed */ static void -ack_proc (void *cls, uint32_t id, const struct GNUNET_MessageHeader *msg) +udp_disconnect (void *cls, const struct GNUNET_PeerIdentity *target) { - struct ReceiveContext *rc = cls; - - size_t msize = sizeof (struct UDP_ACK_Message) + ntohs (msg->size); - char buf[msize]; - struct UDP_ACK_Message *udp_ack; - uint32_t delay = 0; - - struct Session *s; - s = find_inbound_session_by_addr (rc->plugin, rc->src_addr, rc->addr_len); - if (s != NULL) - { - if (s->flow_delay_for_other_peer.rel_value <= UINT32_MAX) - delay = s->flow_delay_for_other_peer.rel_value; - else - delay = UINT32_MAX; - } - + struct Plugin *plugin = cls; + GNUNET_assert (plugin != NULL); -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, "Sending ACK to `%s' including delay of %u ms\n", - GNUNET_a2s (rc->src_addr, - (rc->src_addr->sa_family == - AF_INET) ? sizeof (struct sockaddr_in) : - sizeof (struct sockaddr_in6)), - delay); -#endif - udp_ack = (struct UDP_ACK_Message *) buf; - udp_ack->header.size = htons ((uint16_t) msize); - udp_ack->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK); - udp_ack->delay = htonl (delay); - udp_ack->sender = *rc->plugin->env->my_identity; - memcpy (&udp_ack[1], msg, ntohs (msg->size)); - (void) udp_send (rc->plugin, rc->src_addr, &udp_ack->header); + GNUNET_assert (target != NULL); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Disconnecting from peer `%s'\n", GNUNET_i2s (target)); + /* Clean up sessions */ + GNUNET_CONTAINER_multihashmap_get_multiple (plugin->sessions, &target->hashPubKey, &disconnect_and_free_it, plugin); } -/** - * Closure for 'find_receive_context'. - */ -struct FindReceiveContext +static struct Session * +create_session (struct Plugin *plugin, const struct GNUNET_PeerIdentity *target, + const void *addr, size_t addrlen, + GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls) { - /** - * Where to store the result. - */ - struct ReceiveContext *rc; - - /** - * Address to find. - */ - const struct sockaddr *addr; - - /** - * Number of bytes in 'addr'. - */ - socklen_t addr_len; - - struct Session * session; -}; - - -/** - * Scan the heap for a receive context with the given address. - * - * @param cls the 'struct FindReceiveContext' - * @param node internal node of the heap - * @param element value stored at the node (a 'struct ReceiveContext') - * @param cost cost associated with the node - * @return GNUNET_YES if we should continue to iterate, - * GNUNET_NO if not. - */ -static int -find_receive_context (void *cls, struct GNUNET_CONTAINER_HeapNode *node, - void *element, GNUNET_CONTAINER_HeapCostType cost) -{ - struct FindReceiveContext *frc = cls; - struct ReceiveContext *e = element; + struct Session *s; + const struct IPv4UdpAddress *t4; + const struct IPv6UdpAddress *t6; + struct sockaddr_in *v4; + struct sockaddr_in6 *v6; + size_t len; - if ((frc->addr_len == e->addr_len) && - (0 == memcmp (frc->addr, e->src_addr, frc->addr_len))) + switch (addrlen) { - frc->rc = e; - return GNUNET_NO; + case sizeof (struct IPv4UdpAddress): + if (NULL == plugin->sockv4) + { + return NULL; + } + t4 = addr; + s = GNUNET_malloc (sizeof (struct Session) + sizeof (struct sockaddr_in)); + len = sizeof (struct sockaddr_in); + v4 = (struct sockaddr_in *) &s[1]; + v4->sin_family = AF_INET; +#if HAVE_SOCKADDR_IN_SIN_LEN + v4->sin_len = sizeof (struct sockaddr_in); +#endif + v4->sin_port = t4->u4_port; + v4->sin_addr.s_addr = t4->ipv4_addr; + s->ats = plugin->env->get_address_type (plugin->env->cls, (const struct sockaddr *) v4, sizeof (struct sockaddr_in)); + break; + case sizeof (struct IPv6UdpAddress): + if (NULL == plugin->sockv6) + { + return NULL; + } + t6 = addr; + s = + GNUNET_malloc (sizeof (struct Session) + sizeof (struct sockaddr_in6)); + len = sizeof (struct sockaddr_in6); + v6 = (struct sockaddr_in6 *) &s[1]; + v6->sin6_family = AF_INET6; +#if HAVE_SOCKADDR_IN_SIN_LEN + v6->sin6_len = sizeof (struct sockaddr_in6); +#endif + v6->sin6_port = t6->u6_port; + v6->sin6_addr = t6->ipv6_addr; + s->ats = plugin->env->get_address_type (plugin->env->cls, (const struct sockaddr *) v6, sizeof (struct sockaddr_in6)); + break; + default: + /* Must have a valid address to send to */ + GNUNET_break_op (0); + return NULL; } - return GNUNET_YES; + s->addrlen = len; + s->target = *target; + s->sock_addr = (const struct sockaddr *) &s[1]; + s->last_expected_delay = GNUNET_TIME_UNIT_SECONDS; + start_session_timeout(s); + return s; } -/** - * Read and process a message from the given socket. - * - * @param plugin the overall plugin - * @param rsock socket to read from - */ -static void -udp_read (struct Plugin *plugin, struct GNUNET_NETWORK_Handle *rsock) +static int +session_cmp_it (void *cls, + const struct GNUNET_HashCode * key, + void *value) { - socklen_t fromlen; - char addr[32]; - char buf[65536]; - ssize_t ret; - const struct GNUNET_MessageHeader *msg; - const struct GNUNET_MessageHeader *ack; - struct Session *peer_session; - const struct UDP_ACK_Message *udp_ack; - struct ReceiveContext *rc; - struct GNUNET_TIME_Absolute now; - struct FindReceiveContext frc; - struct Session * s = NULL; - struct GNUNET_TIME_Relative flow_delay; - - fromlen = sizeof (addr); - memset (&addr, 0, sizeof (addr)); - ret = - GNUNET_NETWORK_socket_recvfrom (rsock, buf, sizeof (buf), - (struct sockaddr *) &addr, &fromlen); - if (ret < sizeof (struct GNUNET_MessageHeader)) - { - GNUNET_break_op (0); - return; - } - msg = (const struct GNUNET_MessageHeader *) buf; + struct SessionCompareContext * cctx = cls; + const struct GNUNET_HELLO_Address *address = cctx->addr; + struct Session *s = value; - LOG (GNUNET_ERROR_TYPE_DEBUG, - "UDP received %u-byte message from `%s' type %i\n", (unsigned int) ret, - GNUNET_a2s ((const struct sockaddr *) addr, fromlen), ntohs(msg->type)); + socklen_t s_addrlen = s->addrlen; - if (ret != ntohs (msg->size)) - { - GNUNET_break_op (0); - return; - } - switch (ntohs (msg->type)) + LOG (GNUNET_ERROR_TYPE_DEBUG, "Comparing address %s <-> %s\n", + udp_address_to_string (NULL, (void *) address->address, address->address_length), + GNUNET_a2s (s->sock_addr, s->addrlen)); + if ((address->address_length == sizeof (struct IPv4UdpAddress)) && + (s_addrlen == sizeof (struct sockaddr_in))) { - case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE: - if (ntohs (msg->size) < sizeof (struct UDPMessage)) - { - GNUNET_break_op (0); - return; - } - process_udp_message (plugin, (const struct UDPMessage *) msg, - (const struct sockaddr *) addr, fromlen); - return; - case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK: - - if (ntohs (msg->size) < - sizeof (struct UDP_ACK_Message) + sizeof (struct GNUNET_MessageHeader)) - { - GNUNET_break_op (0); - return; - } - udp_ack = (const struct UDP_ACK_Message *) msg; - s = find_inbound_session(plugin, &udp_ack->sender, addr, fromlen); - if (s != NULL) - { - flow_delay.rel_value = (uint64_t) ntohl(udp_ack->delay); - - LOG (GNUNET_ERROR_TYPE_DEBUG, - "We received a sending delay of %llu\n", flow_delay.rel_value); - - s->flow_delay_from_other_peer = GNUNET_TIME_relative_to_absolute (flow_delay); - } - ack = (const struct GNUNET_MessageHeader *) &udp_ack[1]; - if (ntohs (ack->size) != ntohs (msg->size) - sizeof (struct UDP_ACK_Message)) - { - GNUNET_break_op (0); - return; - } -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, - "UDP processes %u-byte acknowledgement from `%s' at `%s'\n", - (unsigned int) ntohs (msg->size), GNUNET_i2s (&udp_ack->sender), - GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); -#endif - - peer_session = find_session (plugin, &udp_ack->sender); - if (NULL == peer_session) - { -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Session for ACK not found, dropping ACK!\n"); -#endif - return; - } - if (GNUNET_OK != GNUNET_FRAGMENT_process_ack (peer_session->frag, ack)) - return; - GNUNET_assert (GNUNET_OK == - GNUNET_CONTAINER_multihashmap_remove (plugin->sessions, - &udp_ack-> - sender.hashPubKey, - peer_session)); - plugin->last_expected_delay = - GNUNET_FRAGMENT_context_destroy (peer_session->frag); - if (peer_session->cont != NULL) - peer_session->cont (peer_session->cont_cls, &udp_ack->sender, GNUNET_OK); - GNUNET_free (peer_session); - return; - case GNUNET_MESSAGE_TYPE_FRAGMENT: - frc.rc = NULL; - frc.addr = (const struct sockaddr *) addr; - frc.addr_len = fromlen; - GNUNET_CONTAINER_heap_iterate (plugin->defrags, &find_receive_context, - &frc); - now = GNUNET_TIME_absolute_get (); - rc = frc.rc; - if (rc == NULL) + struct IPv4UdpAddress * u4 = NULL; + u4 = (struct IPv4UdpAddress *) address->address; + const struct sockaddr_in *s4 = (const struct sockaddr_in *) s->sock_addr; + if ((0 == memcmp ((const void *) &u4->ipv4_addr,(const void *) &s4->sin_addr, sizeof (struct in_addr))) && + (u4->u4_port == s4->sin_port)) { - /* need to create a new RC */ - rc = GNUNET_malloc (sizeof (struct ReceiveContext) + fromlen); - memcpy (&rc[1], addr, fromlen); - rc->src_addr = (const struct sockaddr *) &rc[1]; - rc->addr_len = fromlen; - rc->plugin = plugin; - rc->defrag = - GNUNET_DEFRAGMENT_context_create (plugin->env->stats, UDP_MTU, - UDP_MAX_MESSAGES_IN_DEFRAG, rc, - &fragment_msg_proc, &ack_proc); - rc->hnode = - GNUNET_CONTAINER_heap_insert (plugin->defrags, rc, - (GNUNET_CONTAINER_HeapCostType) - now.abs_value); + cctx->res = s; + return GNUNET_NO; } -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, - "UDP processes %u-byte fragment from `%s'\n", - (unsigned int) ntohs (msg->size), - GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); -#endif - if (GNUNET_OK == GNUNET_DEFRAGMENT_process_fragment (rc->defrag, msg)) - { - /* keep this 'rc' from expiring */ - GNUNET_CONTAINER_heap_update_cost (plugin->defrags, rc->hnode, - (GNUNET_CONTAINER_HeapCostType) - now.abs_value); - } - if (GNUNET_CONTAINER_heap_get_size (plugin->defrags) > - UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG) + } + if ((address->address_length == sizeof (struct IPv6UdpAddress)) && + (s_addrlen == sizeof (struct sockaddr_in6))) + { + struct IPv6UdpAddress * u6 = NULL; + u6 = (struct IPv6UdpAddress *) address->address; + const struct sockaddr_in6 *s6 = (const struct sockaddr_in6 *) s->sock_addr; + if ((0 == memcmp (&u6->ipv6_addr, &s6->sin6_addr, sizeof (struct in6_addr))) && + (u6->u6_port == s6->sin6_port)) { - /* remove 'rc' that was inactive the longest */ - rc = GNUNET_CONTAINER_heap_remove_root (plugin->defrags); - GNUNET_assert (NULL != rc); - GNUNET_DEFRAGMENT_context_destroy (rc->defrag); - GNUNET_free (rc); + cctx->res = s; + return GNUNET_NO; } - return; - default: - GNUNET_break_op (0); - return; } + return GNUNET_YES; } /** - * We have been notified that our writeset has something to read. We don't - * know which socket needs to be read, so we have to check each one - * Then reschedule this function to be called again once more is available. + * Creates a new outbound session the transport service will use to send data to the + * peer * - * @param cls the plugin handle - * @param tc the scheduling context (for rescheduling this function again) + * @param cls the plugin + * @param address the address + * @return the session or NULL of max connections exceeded */ -static void -udp_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct Plugin *plugin = cls; - - plugin->select_task = GNUNET_SCHEDULER_NO_TASK; - if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) - return; - if ((NULL != plugin->sockv4) && - (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv4))) - udp_read (plugin, plugin->sockv4); - if ((NULL != plugin->sockv6) && - (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv6))) - udp_read (plugin, plugin->sockv6); - plugin->select_task = - GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, - GNUNET_SCHEDULER_NO_TASK, - GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs, - NULL, &udp_plugin_select, plugin); - -} - - -struct MstContext -{ - struct Plugin * plugin; - - struct IPv4UdpAddress addr; -}; - -void udp_broadcast_mst_cb (void *cls, void *client, - const struct - GNUNET_MessageHeader * - message) +static struct Session * +udp_plugin_get_session (void *cls, + const struct GNUNET_HELLO_Address *address) { + struct Session * s = NULL; struct Plugin * plugin = cls; - struct MstContext * mc = client; - const struct GNUNET_MessageHeader* hello; - struct UDP_Beacon_Message * msg; - msg = (struct UDP_Beacon_Message *) message; - - if (GNUNET_MESSAGE_TYPE_TRANSPORT_BROADCAST_BEACON != ntohs(msg->header.type)) - return; - - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Received beacon with %u bytes from peer `%s' via address `%s'\n", - ntohs(msg->header.size), - GNUNET_i2s (&msg->sender), - udp_address_to_string(NULL, &mc->addr, sizeof (mc->addr))); - - - hello = &message[1]; - plugin->env->receive (plugin->env->cls, &msg->sender, hello, NULL, 0, NULL, (const char *) &mc->addr, sizeof (mc->addr)); + struct IPv6UdpAddress * udp_a6; + struct IPv4UdpAddress * udp_a4; - GNUNET_STATISTICS_update(plugin->env->cfg, _("# HELLO beacons received via udp"), 1, GNUNET_NO); - GNUNET_free (mc); -} + GNUNET_assert (plugin != NULL); + GNUNET_assert (address != NULL); -/** - * Read and process a message from the given socket. - * - * @param plugin the overall plugin - * @param rsock socket to read from - */ -static void -udp_broadcast_read (struct Plugin *plugin, struct GNUNET_NETWORK_Handle *rsock) -{ - socklen_t fromlen; - char addr[32]; - char buf[65536]; - ssize_t ret; - struct MstContext * mc; + if ((address->address == NULL) || + ((address->address_length != sizeof (struct IPv4UdpAddress)) && + (address->address_length != sizeof (struct IPv6UdpAddress)))) + { + GNUNET_break (0); + return NULL; + } - fromlen = sizeof (addr); - memset (&addr, 0, sizeof (addr)); - ret = GNUNET_NETWORK_socket_recvfrom (rsock, buf, sizeof (buf), (struct sockaddr *) &addr, &fromlen); - if (ret < sizeof (struct GNUNET_MessageHeader)) + if (address->address_length == sizeof (struct IPv4UdpAddress)) { - /* malformed beacon, just throw it away */ - return; + if (plugin->sockv4 == NULL) + return NULL; + udp_a4 = (struct IPv4UdpAddress *) address->address; + if (udp_a4->u4_port == 0) + return NULL; } - if (GNUNET_YES == GNUNET_NAT_test_address (plugin->nat, - &((struct sockaddr_in *) addr)->sin_addr, - sizeof (struct in_addr))) + if (address->address_length == sizeof (struct IPv6UdpAddress)) { - /* received my own beacon */ - return; + if (plugin->sockv6 == NULL) + return NULL; + udp_a6 = (struct IPv6UdpAddress *) address->address; + if (udp_a6->u6_port == 0) + return NULL; } + /* check if session already exists */ + struct SessionCompareContext cctx; + cctx.addr = address; + cctx.res = NULL; + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Looking for existing session for peer `%s' `%s' \n", + GNUNET_i2s (&address->peer), + udp_address_to_string(NULL, address->address, address->address_length)); + GNUNET_CONTAINER_multihashmap_get_multiple(plugin->sessions, &address->peer.hashPubKey, session_cmp_it, &cctx); + if (cctx.res != NULL) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "Found existing session %p\n", cctx.res); + return cctx.res; + } - mc = GNUNET_malloc(sizeof (struct MstContext)); + /* otherwise create new */ + s = create_session (plugin, + &address->peer, + address->address, + address->address_length, + NULL, NULL); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Creating new session %p for peer `%s' address `%s'\n", + s, + GNUNET_i2s(&address->peer), + udp_address_to_string(NULL,address->address,address->address_length)); + GNUNET_assert (GNUNET_OK == + GNUNET_CONTAINER_multihashmap_put (plugin->sessions, + &s->target.hashPubKey, + s, + GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE)); - struct sockaddr_in * av4 = (struct sockaddr_in *) &addr; - mc->addr.ipv4_addr = av4->sin_addr.s_addr; - mc->addr.u4_port = av4->sin_port; + GNUNET_STATISTICS_set(plugin->env->stats, + "# UDP sessions active", + GNUNET_CONTAINER_multihashmap_size(plugin->sessions), + GNUNET_NO); - if (GNUNET_OK != GNUNET_SERVER_mst_receive(plugin->broadcast_mst, mc, buf, ret, GNUNET_NO, GNUNET_NO)) - GNUNET_free (mc); + return s; } -static void -udp_broadcast_send (void *cls, - const struct GNUNET_SCHEDULER_TaskContext *tc) +static void +enqueue (struct Plugin *plugin, struct UDPMessageWrapper * udpw) { - struct Plugin * plugin = cls; - int sent; - uint16_t msg_size; - uint16_t hello_size; - char buf[65536]; -// /ssize_t ret; - const struct GNUNET_MessageHeader *hello; - struct UDP_Beacon_Message * msg; - - plugin->send_broadcast_task = GNUNET_SCHEDULER_NO_TASK; - - struct sockaddr_in baddr; - baddr.sin_family = AF_INET; - baddr.sin_port = htons (plugin->broadcast_port); - baddr.sin_addr.s_addr=htonl(-1); /* send message to 255.255.255.255 */ - - hello = plugin->env->get_our_hello (); - hello_size = GNUNET_HELLO_size ((struct GNUNET_HELLO_Message *) hello); - msg_size = hello_size + sizeof (struct UDP_Beacon_Message); - - if (hello_size < (sizeof (struct GNUNET_MessageHeader)) ||(msg_size > (UDP_MTU))) - return; - - msg = (struct UDP_Beacon_Message *) buf; - msg->sender = *(plugin->env->my_identity); - msg->header.size = ntohs (msg_size); - msg->header.type = ntohs (GNUNET_MESSAGE_TYPE_TRANSPORT_BROADCAST_BEACON); - memcpy (&msg[1], hello, hello_size); - - sent = GNUNET_NETWORK_socket_sendto (plugin->sockv4_broadcast, msg, msg_size, - (const struct sockaddr *) &baddr, sizeof (struct sockaddr_in)); - - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Sent HELLO beacon broadcast with %i bytes\n", sent); - - plugin->send_broadcast_task = GNUNET_SCHEDULER_add_delayed(plugin->broadcast_interval, &udp_broadcast_send, plugin); + if (udpw->session->addrlen == sizeof (struct sockaddr_in)) + GNUNET_CONTAINER_DLL_insert(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + if (udpw->session->addrlen == sizeof (struct sockaddr_in6)) + GNUNET_CONTAINER_DLL_insert(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); } + /** - * We have been notified that our writeset has something to read. We don't - * know which socket needs to be read, so we have to check each one - * Then reschedule this function to be called again once more is available. + * Fragment message was transmitted via UDP, let fragmentation know + * to send the next fragment now. * - * @param cls the plugin handle - * @param tc the scheduling context (for rescheduling this function again) + * @param cls the 'struct UDPMessageWrapper' of the fragment + * @param target destination peer (ignored) + * @param result GNUNET_OK on success (ignored) */ static void -udp_plugin_broadcast_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +send_next_fragment (void *cls, + const struct GNUNET_PeerIdentity *target, + int result) { - struct Plugin *plugin = cls; - - plugin->select_broadcast_task = GNUNET_SCHEDULER_NO_TASK; - if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) - return; + struct UDPMessageWrapper *udpw = cls; - if ((NULL != plugin->sockv4_broadcast) && - (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv4_broadcast))) - udp_broadcast_read (plugin, plugin->sockv4_broadcast); - - plugin->select_broadcast_task = - GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, - GNUNET_SCHEDULER_NO_TASK, - GNUNET_TIME_UNIT_FOREVER_REL, plugin->broadcast_rs, - NULL, &udp_plugin_broadcast_select, plugin); + GNUNET_FRAGMENT_context_transmission_done (udpw->frag_ctx->frag); } - /** - * Check if the given port is plausible (must be either our listen - * port or our advertised port). If it is neither, we return - * GNUNET_SYSERR. + * Function that is called with messages created by the fragmentation + * module. In the case of the 'proc' callback of the + * GNUNET_FRAGMENT_context_create function, this function must + * eventually call 'GNUNET_FRAGMENT_context_transmission_done'. * - * @param plugin global variables - * @param in_port port number to check - * @return GNUNET_OK if port is either open_port or adv_port + * @param cls closure, the 'struct FragmentationContext' + * @param msg the message that was created */ -static int -check_port (struct Plugin *plugin, uint16_t in_port) +static void +enqueue_fragment (void *cls, const struct GNUNET_MessageHeader *msg) { - if ((in_port == plugin->port) || (in_port == plugin->aport)) - return GNUNET_OK; - return GNUNET_SYSERR; + struct FragmentationContext *frag_ctx = cls; + struct Plugin *plugin = frag_ctx->plugin; + struct UDPMessageWrapper * udpw; + struct Session *s; + size_t msg_len = ntohs (msg->size); + + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Enqueuing fragment with %u bytes %u\n", msg_len , sizeof (struct UDPMessageWrapper)); + udpw = GNUNET_malloc (sizeof (struct UDPMessageWrapper) + msg_len); + udpw->session = frag_ctx->session; + s = udpw->session; + udpw->udp = (char *) &udpw[1]; + + udpw->msg_size = msg_len; + udpw->cont = &send_next_fragment; + udpw->cont_cls = udpw; + udpw->timeout = frag_ctx->timeout; + udpw->frag_ctx = frag_ctx; + memcpy (udpw->udp, msg, msg_len); + enqueue (plugin, udpw); + + if (s->addrlen == sizeof (struct sockaddr_in)) + { + if (plugin->with_v4_ws == GNUNET_NO) + { + if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel(plugin->select_task); + + plugin->select_task = + GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v4, + plugin->ws_v4, + &udp_plugin_select, plugin); + plugin->with_v4_ws = GNUNET_YES; + } + } + else if (s->addrlen == sizeof (struct sockaddr_in6)) + { + if (plugin->with_v6_ws == GNUNET_NO) + { + if (plugin->select_task_v6 != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel(plugin->select_task_v6); + + plugin->select_task_v6 = + GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v6, + plugin->ws_v6, + &udp_plugin_select_v6, plugin); + plugin->with_v6_ws = GNUNET_YES; + } + } } /** - * Function that will be called to check if a binary address for this - * plugin is well-formed and corresponds to an address for THIS peer - * (as per our configuration). Naturally, if absolutely necessary, - * plugins can be a bit conservative in their answer, but in general - * plugins should make sure that the address does not redirect - * traffic to a 3rd party that might try to man-in-the-middle our - * traffic. - * - * @param cls closure, should be our handle to the Plugin - * @param addr pointer to the address - * @param addrlen length of addr - * @return GNUNET_OK if this is a plausible address for this peer - * and transport, GNUNET_SYSERR if not + * Function that can be used by the transport service to transmit + * a message using the plugin. Note that in the case of a + * peer disconnecting, the continuation MUST be called + * prior to the disconnect notification itself. This function + * will be called with this peer's HELLO message to initiate + * a fresh connection to another peer. * + * @param cls closure + * @param s which session must be used + * @param msgbuf the message to transmit + * @param msgbuf_size number of bytes in 'msgbuf' + * @param priority how important is the message (most plugins will + * ignore message priority and just FIFO) + * @param to how long to wait at most for the transmission (does not + * require plugins to discard the message after the timeout, + * just advisory for the desired delay; most plugins will ignore + * this as well) + * @param cont continuation to call once the message has + * been transmitted (or if the transport is ready + * for the next transmission call; or if the + * peer disconnected...); can be NULL + * @param cont_cls closure for cont + * @return number of bytes used (on the physical network, with overheads); + * -1 on hard errors (i.e. address invalid); 0 is a legal value + * and does NOT mean that the message was not transmitted (DV) */ -static int -udp_plugin_check_address (void *cls, const void *addr, size_t addrlen) +static ssize_t +udp_plugin_send (void *cls, + struct Session *s, + const char *msgbuf, size_t msgbuf_size, + unsigned int priority, + struct GNUNET_TIME_Relative to, + GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls) { struct Plugin *plugin = cls; - struct IPv4UdpAddress *v4; - struct IPv6UdpAddress *v6; + size_t mlen = msgbuf_size + sizeof (struct UDPMessage); + struct UDPMessageWrapper * udpw; + struct UDPMessage *udp; + char mbuf[mlen]; + GNUNET_assert (plugin != NULL); + GNUNET_assert (s != NULL); - if ((addrlen != sizeof (struct IPv4UdpAddress)) && - (addrlen != sizeof (struct IPv6UdpAddress))) - { - GNUNET_break_op (0); + if ((s->addrlen == sizeof (struct sockaddr_in6)) && (plugin->sockv6 == NULL)) return GNUNET_SYSERR; - } - if (addrlen == sizeof (struct IPv4UdpAddress)) + if ((s->addrlen == sizeof (struct sockaddr_in)) && (plugin->sockv4 == NULL)) + return GNUNET_SYSERR; + if (mlen >= GNUNET_SERVER_MAX_MESSAGE_SIZE) { - v4 = (struct IPv4UdpAddress *) addr; - if (GNUNET_OK != check_port (plugin, ntohs (v4->u4_port))) - return GNUNET_SYSERR; - if (GNUNET_OK != - GNUNET_NAT_test_address (plugin->nat, &v4->ipv4_addr, - sizeof (struct in_addr))) - return GNUNET_SYSERR; + GNUNET_break (0); + return GNUNET_SYSERR; } - else + if (GNUNET_YES != GNUNET_CONTAINER_multihashmap_contains_value(plugin->sessions, &s->target.hashPubKey, s)) { - v6 = (struct IPv6UdpAddress *) addr; - if (IN6_IS_ADDR_LINKLOCAL (&v6->ipv6_addr)) - { - GNUNET_break_op (0); - return GNUNET_SYSERR; - } - if (GNUNET_OK != check_port (plugin, ntohs (v6->u6_port))) - return GNUNET_SYSERR; - if (GNUNET_OK != - GNUNET_NAT_test_address (plugin->nat, &v6->ipv6_addr, - sizeof (struct in6_addr))) - return GNUNET_SYSERR; + GNUNET_break (0); + return GNUNET_SYSERR; } - return GNUNET_OK; -} - - -/** - * Function called for a quick conversion of the binary address to - * a numeric address. Note that the caller must not free the - * address and that the next call to this function is allowed - * to override the address again. - * - * @param cls closure - * @param addr binary address - * @param addrlen length of the address - * @return string representing the same address - */ -static const char * -udp_address_to_string (void *cls, const void *addr, size_t addrlen) -{ - static char rbuf[INET6_ADDRSTRLEN + 10]; - char buf[INET6_ADDRSTRLEN]; - const void *sb; - struct in_addr a4; - struct in6_addr a6; - const struct IPv4UdpAddress *t4; - const struct IPv6UdpAddress *t6; - int af; - uint16_t port; + LOG (GNUNET_ERROR_TYPE_DEBUG, + "UDP transmits %u-byte message to `%s' using address `%s'\n", + mlen, + GNUNET_i2s (&s->target), + GNUNET_a2s(s->sock_addr, s->addrlen)); + + /* Message */ + udp = (struct UDPMessage *) mbuf; + udp->header.size = htons (mlen); + udp->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE); + udp->reserved = htonl (0); + udp->sender = *plugin->env->my_identity; - if (addrlen == sizeof (struct IPv6UdpAddress)) - { - t6 = addr; - af = AF_INET6; - port = ntohs (t6->u6_port); - memcpy (&a6, &t6->ipv6_addr, sizeof (a6)); - sb = &a6; - } - else if (addrlen == sizeof (struct IPv4UdpAddress)) + reschedule_session_timeout(s); + if (mlen <= UDP_MTU) { - t4 = addr; - af = AF_INET; - port = ntohs (t4->u4_port); - memcpy (&a4, &t4->ipv4_addr, sizeof (a4)); - sb = &a4; + udpw = GNUNET_malloc (sizeof (struct UDPMessageWrapper) + mlen); + udpw->session = s; + udpw->udp = (char *) &udpw[1]; + udpw->msg_size = mlen; + udpw->timeout = GNUNET_TIME_absolute_add(GNUNET_TIME_absolute_get(), to); + udpw->cont = cont; + udpw->cont_cls = cont_cls; + udpw->frag_ctx = NULL; + memcpy (udpw->udp, udp, sizeof (struct UDPMessage)); + memcpy (&udpw->udp[sizeof (struct UDPMessage)], msgbuf, msgbuf_size); + + enqueue (plugin, udpw); } else { - GNUNET_break_op (0); - return NULL; - } - inet_ntop (af, sb, buf, INET6_ADDRSTRLEN); - GNUNET_snprintf (rbuf, sizeof (rbuf), (af == AF_INET6) ? "[%s]:%u" : "%s:%u", - buf, port); - return rbuf; -} - - -/** - * Closure for 'append_port'. - */ -struct PrettyPrinterContext -{ - /** - * Function to call with the result. - */ - GNUNET_TRANSPORT_AddressStringCallback asc; - - /** - * Clsoure for 'asc'. - */ - void *asc_cls; - - /** - * Port to add after the IP address. - */ - uint16_t port; -}; - - -/** - * Append our port and forward the result. - * - * @param cls a 'struct PrettyPrinterContext' - * @param hostname result from DNS resolver - */ -static void -append_port (void *cls, const char *hostname) -{ - struct PrettyPrinterContext *ppc = cls; - char *ret; - - if (hostname == NULL) - { - ppc->asc (ppc->asc_cls, NULL); - GNUNET_free (ppc); - return; + LOG (GNUNET_ERROR_TYPE_DEBUG, + "UDP has to fragment message \n"); + if (s->frag_ctx != NULL) + return GNUNET_SYSERR; + memcpy (&udp[1], msgbuf, msgbuf_size); + struct FragmentationContext * frag_ctx = GNUNET_malloc(sizeof (struct FragmentationContext)); + + frag_ctx->plugin = plugin; + frag_ctx->session = s; + frag_ctx->cont = cont; + frag_ctx->cont_cls = cont_cls; + frag_ctx->timeout = GNUNET_TIME_absolute_add(GNUNET_TIME_absolute_get(), to); + frag_ctx->bytes_to_send = mlen; + frag_ctx->frag = GNUNET_FRAGMENT_context_create (plugin->env->stats, + UDP_MTU, + &plugin->tracker, + s->last_expected_delay, + &udp->header, + &enqueue_fragment, + frag_ctx); + + s->frag_ctx = frag_ctx; } - GNUNET_asprintf (&ret, "%s:%d", hostname, ppc->port); - ppc->asc (ppc->asc_cls, ret); - GNUNET_free (ret); -} - - -/** - * Convert the transports address to a nice, human-readable - * format. - * - * @param cls closure - * @param type name of the transport that generated the address - * @param addr one of the addresses of the host, NULL for the last address - * the specific address format depends on the transport - * @param addrlen length of the address - * @param numeric should (IP) addresses be displayed in numeric form? - * @param timeout after how long should we give up? - * @param asc function to call on each string - * @param asc_cls closure for asc - */ -static void -udp_plugin_address_pretty_printer (void *cls, const char *type, - const void *addr, size_t addrlen, - int numeric, - struct GNUNET_TIME_Relative timeout, - GNUNET_TRANSPORT_AddressStringCallback asc, - void *asc_cls) -{ - struct PrettyPrinterContext *ppc; - const void *sb; - size_t sbs; - struct sockaddr_in a4; - struct sockaddr_in6 a6; - const struct IPv4UdpAddress *u4; - const struct IPv6UdpAddress *u6; - uint16_t port; - if (addrlen == sizeof (struct IPv6UdpAddress)) - { - u6 = addr; - memset (&a6, 0, sizeof (a6)); - a6.sin6_family = AF_INET6; -#if HAVE_SOCKADDR_IN_SIN_LEN - a6.sin6_len = sizeof (a6); -#endif - a6.sin6_port = u6->u6_port; - memcpy (&a6.sin6_addr, &u6->ipv6_addr, sizeof (struct in6_addr)); - port = ntohs (u6->u6_port); - sb = &a6; - sbs = sizeof (a6); - } - else if (addrlen == sizeof (struct IPv4UdpAddress)) + if (s->addrlen == sizeof (struct sockaddr_in)) { - u4 = addr; - memset (&a4, 0, sizeof (a4)); - a4.sin_family = AF_INET; -#if HAVE_SOCKADDR_IN_SIN_LEN - a4.sin_len = sizeof (a4); -#endif - a4.sin_port = u4->u4_port; - a4.sin_addr.s_addr = u4->ipv4_addr; - port = ntohs (u4->u4_port); - sb = &a4; - sbs = sizeof (a4); + if (plugin->with_v4_ws == GNUNET_NO) + { + if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel(plugin->select_task); + + plugin->select_task = + GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v4, + plugin->ws_v4, + &udp_plugin_select, plugin); + plugin->with_v4_ws = GNUNET_YES; + } } - else + else if (s->addrlen == sizeof (struct sockaddr_in6)) { - /* invalid address */ - GNUNET_break_op (0); - asc (asc_cls, NULL); - return; + if (plugin->with_v6_ws == GNUNET_NO) + { + if (plugin->select_task_v6 != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel(plugin->select_task_v6); + + plugin->select_task_v6 = + GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v6, + plugin->ws_v6, + &udp_plugin_select_v6, plugin); + plugin->with_v6_ws = GNUNET_YES; + } } - ppc = GNUNET_malloc (sizeof (struct PrettyPrinterContext)); - ppc->asc = asc; - ppc->asc_cls = asc_cls; - ppc->port = port; - GNUNET_RESOLVER_hostname_get (sb, sbs, !numeric, timeout, &append_port, ppc); + + return mlen; } @@ -1840,167 +1281,810 @@ udp_nat_port_map_callback (void *cls, int add_remove, } + /** - * The exported method. Makes the core api available via a global and - * returns the udp transport API. + * Message tokenizer has broken up an incomming message. Pass it on + * to the service. * - * @param cls our 'struct GNUNET_TRANSPORT_PluginEnvironment' - * @return our 'struct GNUNET_TRANSPORT_PluginFunctions' + * @param cls the 'struct Plugin' + * @param client the 'struct SourceInformation' + * @param hdr the actual message */ -void * -libgnunet_plugin_transport_udp_init (void *cls) +static int +process_inbound_tokenized_messages (void *cls, void *client, + const struct GNUNET_MessageHeader *hdr) { - struct GNUNET_TRANSPORT_PluginEnvironment *env = cls; - unsigned long long port; - unsigned long long aport; - unsigned long long bport; - struct GNUNET_TRANSPORT_PluginFunctions *api; - struct Plugin *plugin; - int sockets_created; - int broadcast; - struct GNUNET_TIME_Relative interval; - struct sockaddr_in serverAddrv4; - struct sockaddr_in6 serverAddrv6; - struct sockaddr *serverAddr; - struct sockaddr *addrs[2]; - socklen_t addrlens[2]; - socklen_t addrlen; - unsigned int tries; - unsigned long long udp_max_bps; + struct Plugin *plugin = cls; + struct SourceInformation *si = client; + struct GNUNET_ATS_Information ats[2]; + struct GNUNET_TIME_Relative delay; - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", "PORT", - &port)) - port = 2086; + GNUNET_assert (si->session != NULL); + if (GNUNET_YES == si->session->in_destroy) + return GNUNET_OK; + /* setup ATS */ + ats[0].type = htonl (GNUNET_ATS_QUALITY_NET_DISTANCE); + ats[0].value = htonl (1); + ats[1] = si->session->ats; + GNUNET_break (ntohl(ats[1].value) != GNUNET_ATS_NET_UNSPECIFIED); + delay = plugin->env->receive (plugin->env->cls, + &si->sender, + hdr, + (const struct GNUNET_ATS_Information *) &ats, 2, + si->session, + si->arg, + si->args); + si->session->flow_delay_for_other_peer = delay; + reschedule_session_timeout(si->session); + return GNUNET_OK; +} - broadcast = GNUNET_CONFIGURATION_get_value_yesno (env->cfg, "transport-udp", "BROADCAST"); - if (broadcast == GNUNET_SYSERR) - broadcast = GNUNET_NO; - if (GNUNET_SYSERR == GNUNET_CONFIGURATION_get_value_time (env->cfg, "transport-udp", "BROADCAST_INTERVAL", &interval)) - interval = GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 10); +/** + * We've received a UDP Message. Process it (pass contents to main service). + * + * @param plugin plugin context + * @param msg the message + * @param sender_addr sender address + * @param sender_addr_len number of bytes in sender_addr + */ +static void +process_udp_message (struct Plugin *plugin, const struct UDPMessage *msg, + const struct sockaddr *sender_addr, + socklen_t sender_addr_len) +{ + struct SourceInformation si; + struct Session * s; + struct IPv4UdpAddress u4; + struct IPv6UdpAddress u6; + const void *arg; + size_t args; - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", "BROADCAST_PORT", - &bport)) - bport = 20086; + if (0 != ntohl (msg->reserved)) + { + GNUNET_break_op (0); + return; + } + if (ntohs (msg->header.size) < + sizeof (struct GNUNET_MessageHeader) + sizeof (struct UDPMessage)) + { + GNUNET_break_op (0); + return; + } - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", - "MAX_BPS", &udp_max_bps)) - udp_max_bps = 1024 * 1024 * 50; /* 50 MB/s == infinity for practical purposes */ - if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", - "ADVERTISED_PORT", &aport)) - aport = port; - if (port > 65535) + /* convert address */ + switch (sender_addr->sa_family) + { + case AF_INET: + GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in)); + u4.ipv4_addr = ((struct sockaddr_in *) sender_addr)->sin_addr.s_addr; + u4.u4_port = ((struct sockaddr_in *) sender_addr)->sin_port; + arg = &u4; + args = sizeof (u4); + break; + case AF_INET6: + GNUNET_assert (sender_addr_len == sizeof (struct sockaddr_in6)); + u6.ipv6_addr = ((struct sockaddr_in6 *) sender_addr)->sin6_addr; + u6.u6_port = ((struct sockaddr_in6 *) sender_addr)->sin6_port; + arg = &u6; + args = sizeof (u6); + break; + default: + GNUNET_break (0); + return; + } + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Received message with %u bytes from peer `%s' at `%s'\n", + (unsigned int) ntohs (msg->header.size), GNUNET_i2s (&msg->sender), + GNUNET_a2s (sender_addr, sender_addr_len)); + + struct GNUNET_HELLO_Address * address = GNUNET_HELLO_address_allocate(&msg->sender, "udp", arg, args); + s = udp_plugin_get_session(plugin, address); + GNUNET_free (address); + + /* iterate over all embedded messages */ + si.session = s; + si.sender = msg->sender; + si.arg = arg; + si.args = args; + s->rc++; + GNUNET_SERVER_mst_receive (plugin->mst, &si, (const char *) &msg[1], + ntohs (msg->header.size) - + sizeof (struct UDPMessage), GNUNET_YES, GNUNET_NO); + s->rc--; + if ( (0 == s->rc) && (GNUNET_YES == s->in_destroy)) + free_session (s); +} + + +/** + * Scan the heap for a receive context with the given address. + * + * @param cls the 'struct FindReceiveContext' + * @param node internal node of the heap + * @param element value stored at the node (a 'struct ReceiveContext') + * @param cost cost associated with the node + * @return GNUNET_YES if we should continue to iterate, + * GNUNET_NO if not. + */ +static int +find_receive_context (void *cls, struct GNUNET_CONTAINER_HeapNode *node, + void *element, GNUNET_CONTAINER_HeapCostType cost) +{ + struct FindReceiveContext *frc = cls; + struct DefragContext *e = element; + + if ((frc->addr_len == e->addr_len) && + (0 == memcmp (frc->addr, e->src_addr, frc->addr_len))) + { + frc->rc = e; + return GNUNET_NO; + } + return GNUNET_YES; +} + + +/** + * Process a defragmented message. + * + * @param cls the 'struct ReceiveContext' + * @param msg the message + */ +static void +fragment_msg_proc (void *cls, const struct GNUNET_MessageHeader *msg) +{ + struct DefragContext *rc = cls; + + if (ntohs (msg->type) != GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE) + { + GNUNET_break (0); + return; + } + if (ntohs (msg->size) < sizeof (struct UDPMessage)) + { + GNUNET_break (0); + return; + } + process_udp_message (rc->plugin, (const struct UDPMessage *) msg, + rc->src_addr, rc->addr_len); +} + + +struct LookupContext +{ + const struct sockaddr * addr; + + struct Session *res; + + size_t addrlen; +}; + + +static int +lookup_session_by_addr_it (void *cls, const struct GNUNET_HashCode * key, void *value) +{ + struct LookupContext *l_ctx = cls; + struct Session * s = value; + + if ((s->addrlen == l_ctx->addrlen) && + (0 == memcmp (s->sock_addr, l_ctx->addr, s->addrlen))) + { + l_ctx->res = s; + return GNUNET_NO; + } + return GNUNET_YES; +} + + +/** + * Transmit an acknowledgement. + * + * @param cls the 'struct ReceiveContext' + * @param id message ID (unused) + * @param msg ack to transmit + */ +static void +ack_proc (void *cls, uint32_t id, const struct GNUNET_MessageHeader *msg) +{ + struct DefragContext *rc = cls; + size_t msize = sizeof (struct UDP_ACK_Message) + ntohs (msg->size); + struct UDP_ACK_Message *udp_ack; + uint32_t delay = 0; + struct UDPMessageWrapper *udpw; + struct Session *s; + + struct LookupContext l_ctx; + l_ctx.addr = rc->src_addr; + l_ctx.addrlen = rc->addr_len; + l_ctx.res = NULL; + GNUNET_CONTAINER_multihashmap_iterate (rc->plugin->sessions, + &lookup_session_by_addr_it, + &l_ctx); + s = l_ctx.res; + + if (NULL == s) + return; + + if (s->flow_delay_for_other_peer.rel_value <= UINT32_MAX) + delay = s->flow_delay_for_other_peer.rel_value; + + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Sending ACK to `%s' including delay of %u ms\n", + GNUNET_a2s (rc->src_addr, + (rc->src_addr->sa_family == + AF_INET) ? sizeof (struct sockaddr_in) : sizeof (struct + sockaddr_in6)), + delay); + udpw = GNUNET_malloc (sizeof (struct UDPMessageWrapper) + msize); + udpw->msg_size = msize; + udpw->session = s; + udpw->timeout = GNUNET_TIME_UNIT_FOREVER_ABS; + udpw->udp = (char *)&udpw[1]; + udp_ack = (struct UDP_ACK_Message *) udpw->udp; + udp_ack->header.size = htons ((uint16_t) msize); + udp_ack->header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK); + udp_ack->delay = htonl (delay); + udp_ack->sender = *rc->plugin->env->my_identity; + memcpy (&udp_ack[1], msg, ntohs (msg->size)); + + enqueue (rc->plugin, udpw); +} + + +static void +read_process_msg (struct Plugin *plugin, + const struct GNUNET_MessageHeader *msg, + const char *addr, + socklen_t fromlen) +{ + if (ntohs (msg->size) < sizeof (struct UDPMessage)) + { + GNUNET_break_op (0); + return; + } + process_udp_message (plugin, (const struct UDPMessage *) msg, + (const struct sockaddr *) addr, fromlen); +} + + +static void +read_process_ack (struct Plugin *plugin, + const struct GNUNET_MessageHeader *msg, + char *addr, + socklen_t fromlen) +{ + const struct GNUNET_MessageHeader *ack; + const struct UDP_ACK_Message *udp_ack; + struct LookupContext l_ctx; + struct Session *s; + struct GNUNET_TIME_Relative flow_delay; + + if (ntohs (msg->size) < + sizeof (struct UDP_ACK_Message) + sizeof (struct GNUNET_MessageHeader)) + { + GNUNET_break_op (0); + return; + } + udp_ack = (const struct UDP_ACK_Message *) msg; + l_ctx.addr = (const struct sockaddr *) addr; + l_ctx.addrlen = fromlen; + l_ctx.res = NULL; + GNUNET_CONTAINER_multihashmap_iterate (plugin->sessions, + &lookup_session_by_addr_it, + &l_ctx); + s = l_ctx.res; + + if ((s == NULL) || (s->frag_ctx == NULL)) + return; + + flow_delay.rel_value = (uint64_t) ntohl (udp_ack->delay); + LOG (GNUNET_ERROR_TYPE_DEBUG, "We received a sending delay of %llu\n", + flow_delay.rel_value); + s->flow_delay_from_other_peer = + GNUNET_TIME_relative_to_absolute (flow_delay); + + ack = (const struct GNUNET_MessageHeader *) &udp_ack[1]; + if (ntohs (ack->size) != + ntohs (msg->size) - sizeof (struct UDP_ACK_Message)) + { + GNUNET_break_op (0); + return; + } + + if (GNUNET_OK != GNUNET_FRAGMENT_process_ack (s->frag_ctx->frag, ack)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "UDP processes %u-byte acknowledgement from `%s' at `%s'\n", + (unsigned int) ntohs (msg->size), GNUNET_i2s (&udp_ack->sender), + GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); + return; + } + + LOG (GNUNET_ERROR_TYPE_DEBUG, + "FULL MESSAGE ACKed\n", + (unsigned int) ntohs (msg->size), GNUNET_i2s (&udp_ack->sender), + GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); + s->last_expected_delay = GNUNET_FRAGMENT_context_destroy (s->frag_ctx->frag); + + struct UDPMessageWrapper * udpw; + struct UDPMessageWrapper * tmp; + if (s->addrlen == sizeof (struct sockaddr_in6)) + { + udpw = plugin->ipv6_queue_head; + while (NULL != udpw) + { + tmp = udpw->next; + if ((udpw->frag_ctx != NULL) && (udpw->frag_ctx == s->frag_ctx)) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); + GNUNET_free (udpw); + } + udpw = tmp; + } + } + if (s->addrlen == sizeof (struct sockaddr_in)) + { + udpw = plugin->ipv4_queue_head; + while (udpw!= NULL) + { + tmp = udpw->next; + if ((udpw->frag_ctx != NULL) && (udpw->frag_ctx == s->frag_ctx)) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + GNUNET_free (udpw); + } + udpw = tmp; + } + } + + if (s->frag_ctx->cont != NULL) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Calling continuation for fragmented message to `%s' with result %s\n", + GNUNET_i2s (&s->target), "OK"); + s->frag_ctx->cont (s->frag_ctx->cont_cls, &udp_ack->sender, GNUNET_OK); + } + + GNUNET_free (s->frag_ctx); + s->frag_ctx = NULL; +} + + +static void +read_process_fragment (struct Plugin *plugin, + const struct GNUNET_MessageHeader *msg, + char *addr, + socklen_t fromlen) +{ + struct DefragContext *d_ctx; + struct GNUNET_TIME_Absolute now; + struct FindReceiveContext frc; + + frc.rc = NULL; + frc.addr = (const struct sockaddr *) addr; + frc.addr_len = fromlen; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "UDP processes %u-byte fragment from `%s'\n", + (unsigned int) ntohs (msg->size), + GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); + /* Lookup existing receive context for this address */ + GNUNET_CONTAINER_heap_iterate (plugin->defrag_ctxs, + &find_receive_context, + &frc); + now = GNUNET_TIME_absolute_get (); + d_ctx = frc.rc; + + if (d_ctx == NULL) + { + /* Create a new defragmentation context */ + d_ctx = GNUNET_malloc (sizeof (struct DefragContext) + fromlen); + memcpy (&d_ctx[1], addr, fromlen); + d_ctx->src_addr = (const struct sockaddr *) &d_ctx[1]; + d_ctx->addr_len = fromlen; + d_ctx->plugin = plugin; + d_ctx->defrag = + GNUNET_DEFRAGMENT_context_create (plugin->env->stats, UDP_MTU, + UDP_MAX_MESSAGES_IN_DEFRAG, d_ctx, + &fragment_msg_proc, &ack_proc); + d_ctx->hnode = + GNUNET_CONTAINER_heap_insert (plugin->defrag_ctxs, d_ctx, + (GNUNET_CONTAINER_HeapCostType) + now.abs_value); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Created new defragmentation context for %u-byte fragment from `%s'\n", + (unsigned int) ntohs (msg->size), + GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Found existing defragmentation context for %u-byte fragment from `%s'\n", + (unsigned int) ntohs (msg->size), + GNUNET_a2s ((const struct sockaddr *) addr, fromlen)); + } + + if (GNUNET_OK == GNUNET_DEFRAGMENT_process_fragment (d_ctx->defrag, msg)) + { + /* keep this 'rc' from expiring */ + GNUNET_CONTAINER_heap_update_cost (plugin->defrag_ctxs, d_ctx->hnode, + (GNUNET_CONTAINER_HeapCostType) + now.abs_value); + } + if (GNUNET_CONTAINER_heap_get_size (plugin->defrag_ctxs) > + UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG) + { + /* remove 'rc' that was inactive the longest */ + d_ctx = GNUNET_CONTAINER_heap_remove_root (plugin->defrag_ctxs); + GNUNET_assert (NULL != d_ctx); + GNUNET_DEFRAGMENT_context_destroy (d_ctx->defrag); + GNUNET_free (d_ctx); + } +} + + +/** + * Read and process a message from the given socket. + * + * @param plugin the overall plugin + * @param rsock socket to read from + */ +static void +udp_select_read (struct Plugin *plugin, struct GNUNET_NETWORK_Handle *rsock) +{ + socklen_t fromlen; + char addr[32]; + char buf[65536] GNUNET_ALIGN; + ssize_t size; + const struct GNUNET_MessageHeader *msg; + + fromlen = sizeof (addr); + memset (&addr, 0, sizeof (addr)); + size = GNUNET_NETWORK_socket_recvfrom (rsock, buf, sizeof (buf), + (struct sockaddr *) &addr, &fromlen); +#if MINGW + /* On SOCK_DGRAM UDP sockets recvfrom might fail with a + * WSAECONNRESET error to indicate that previous sendto() (???) + * on this socket has failed. + */ + if ( (-1 == size) && (ECONNRESET == errno) ) + return; +#endif + if ( (-1 == size) || (size < sizeof (struct GNUNET_MessageHeader))) + { + GNUNET_break_op (0); + return; + } + msg = (const struct GNUNET_MessageHeader *) buf; + + LOG (GNUNET_ERROR_TYPE_DEBUG, + "UDP received %u-byte message from `%s' type %i\n", (unsigned int) size, + GNUNET_a2s ((const struct sockaddr *) addr, fromlen), ntohs (msg->type)); + + if (size != ntohs (msg->size)) + { + GNUNET_break_op (0); + return; + } + + switch (ntohs (msg->type)) + { + case GNUNET_MESSAGE_TYPE_TRANSPORT_BROADCAST_BEACON: + udp_broadcast_receive (plugin, &buf, size, addr, fromlen); + return; + + case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_MESSAGE: + read_process_msg (plugin, msg, addr, fromlen); + return; + + case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_ACK: + read_process_ack (plugin, msg, addr, fromlen); + return; + + case GNUNET_MESSAGE_TYPE_FRAGMENT: + read_process_fragment (plugin, msg, addr, fromlen); + return; + + default: + GNUNET_break_op (0); + return; + } +} + + +static size_t +udp_select_send (struct Plugin *plugin, struct GNUNET_NETWORK_Handle *sock) +{ + ssize_t sent; + size_t slen; + struct GNUNET_TIME_Absolute max; + struct UDPMessageWrapper *udpw = NULL; + static int network_down_error; + + if (sock == plugin->sockv4) + { + udpw = plugin->ipv4_queue_head; + } + else if (sock == plugin->sockv6) + { + udpw = plugin->ipv6_queue_head; + } + else + { + GNUNET_break (0); + return 0; + } + + const struct sockaddr * sa = udpw->session->sock_addr; + slen = udpw->session->addrlen; + + max = GNUNET_TIME_absolute_max(udpw->timeout, GNUNET_TIME_absolute_get()); + + while (udpw != NULL) + { + if (max.abs_value != udpw->timeout.abs_value) + { + /* Message timed out */ + call_continuation(udpw, GNUNET_SYSERR); + if (udpw->frag_ctx != NULL) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Fragmented message for peer `%s' with size %u timed out\n", + GNUNET_i2s(&udpw->session->target), udpw->frag_ctx->bytes_to_send); + udpw->session->last_expected_delay = GNUNET_FRAGMENT_context_destroy(udpw->frag_ctx->frag); + GNUNET_free (udpw->frag_ctx); + udpw->session->frag_ctx = NULL; + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Message for peer `%s' with size %u timed out\n", + GNUNET_i2s(&udpw->session->target), udpw->msg_size); + } + + if (sock == plugin->sockv4) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + GNUNET_free (udpw); + udpw = plugin->ipv4_queue_head; + } + else if (sock == plugin->sockv6) + { + GNUNET_CONTAINER_DLL_remove(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); + GNUNET_free (udpw); + udpw = plugin->ipv6_queue_head; + } + } + else + { + struct GNUNET_TIME_Relative delta = GNUNET_TIME_absolute_get_remaining (udpw->session->flow_delay_from_other_peer); + if (delta.rel_value == 0) + { + /* this message is not delayed */ + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Message for peer `%s' (%u bytes) is not delayed \n", + GNUNET_i2s(&udpw->session->target), udpw->msg_size); + break; + } + else + { + /* this message is delayed, try next */ + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Message for peer `%s' (%u bytes) is delayed for %llu \n", + GNUNET_i2s(&udpw->session->target), udpw->msg_size, + delta); + udpw = udpw->next; + } + } + } + + if (udpw == NULL) + { + /* No message left */ + return 0; + } + + sent = GNUNET_NETWORK_socket_sendto (sock, udpw->udp, udpw->msg_size, sa, slen); + + if (GNUNET_SYSERR == sent) + { + const struct GNUNET_ATS_Information type = plugin->env->get_address_type + (plugin->env->cls,sa, slen); + + if (((GNUNET_ATS_NET_LAN == ntohl(type.value)) || (GNUNET_ATS_NET_WAN == ntohl(type.value))) && + ((ENETUNREACH == errno) || (ENETDOWN == errno))) + { + if ((network_down_error == GNUNET_NO) && (slen == sizeof (struct sockaddr_in))) + { + /* IPv4: "Network unreachable" or "Network down" + * + * This indicates we do not have connectivity + */ + LOG (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK, + _("UDP could not transmit message to `%s': " + "Network seems down, please check your network configuration\n"), + GNUNET_a2s (sa, slen)); + } + if ((network_down_error == GNUNET_NO) && (slen == sizeof (struct sockaddr_in6))) + { + /* IPv6: "Network unreachable" or "Network down" + * + * This indicates that this system is IPv6 enabled, but does not + * have a valid global IPv6 address assigned or we do not have + * connectivity + */ + + LOG (GNUNET_ERROR_TYPE_WARNING | GNUNET_ERROR_TYPE_BULK, + _("UDP could not transmit message to `%s': " + "Please check your network configuration and disable IPv6 if your " + "connection does not have a global IPv6 address\n"), + GNUNET_a2s (sa, slen)); + } + } + else + { + LOG (GNUNET_ERROR_TYPE_WARNING, + "UDP could not transmit %u-byte message to `%s': `%s'\n", + (unsigned int) (udpw->msg_size), GNUNET_a2s (sa, slen), + STRERROR (errno)); + } + call_continuation(udpw, GNUNET_SYSERR); + } + else + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "UDP transmitted %u-byte message to `%s' (%d: %s)\n", + (unsigned int) (udpw->msg_size), GNUNET_a2s (sa, slen), (int) sent, + (sent < 0) ? STRERROR (errno) : "ok"); + call_continuation(udpw, GNUNET_OK); + network_down_error = GNUNET_NO; + } + + if (sock == plugin->sockv4) + GNUNET_CONTAINER_DLL_remove(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + else if (sock == plugin->sockv6) + GNUNET_CONTAINER_DLL_remove(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); + GNUNET_free (udpw); + udpw = NULL; + + return sent; +} + + +/** + * We have been notified that our readset has something to read. We don't + * know which socket needs to be read, so we have to check each one + * Then reschedule this function to be called again once more is available. + * + * @param cls the plugin handle + * @param tc the scheduling context (for rescheduling this function again) + */ +static void +udp_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + struct Plugin *plugin = cls; + + plugin->select_task = GNUNET_SCHEDULER_NO_TASK; + if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) + return; + plugin->with_v4_ws = GNUNET_NO; + + if ((tc->reason & GNUNET_SCHEDULER_REASON_READ_READY) != 0) + { + if ((NULL != plugin->sockv4) && + (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv4))) + udp_select_read (plugin, plugin->sockv4); + + } + + if ((tc->reason & GNUNET_SCHEDULER_REASON_WRITE_READY) != 0) { - LOG (GNUNET_ERROR_TYPE_WARNING, - _("Given `%s' option is out of range: %llu > %u\n"), "PORT", - port, 65535); - return NULL; + if ((NULL != plugin->sockv4) && (plugin->ipv4_queue_head != NULL) && + (GNUNET_NETWORK_fdset_isset (tc->write_ready, plugin->sockv4))) + { + udp_select_send (plugin, plugin->sockv4); + } } - memset (&serverAddrv6, 0, sizeof (serverAddrv6)); - memset (&serverAddrv4, 0, sizeof (serverAddrv4)); - plugin = GNUNET_malloc (sizeof (struct Plugin)); - GNUNET_BANDWIDTH_tracker_init (&plugin->tracker, - GNUNET_BANDWIDTH_value_init ((uint32_t) - udp_max_bps), 30); - plugin->last_expected_delay = GNUNET_TIME_UNIT_SECONDS; - plugin->port = port; - plugin->aport = aport; - plugin->broadcast_port = bport; - plugin->broadcast = broadcast; - plugin->env = env; - plugin->broadcast_interval = interval; - api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions)); - api->cls = plugin; + if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel (plugin->select_task); + plugin->select_task = GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v4, + (plugin->ipv4_queue_head != NULL) ? plugin->ws_v4 : NULL, + &udp_plugin_select, plugin); + if (plugin->ipv4_queue_head != NULL) + plugin->with_v4_ws = GNUNET_YES; + else + plugin->with_v4_ws = GNUNET_NO; +} - api->send = &udp_plugin_send; - api->disconnect = &udp_disconnect; - api->address_pretty_printer = &udp_plugin_address_pretty_printer; - api->address_to_string = &udp_address_to_string; - api->check_address = &udp_plugin_check_address; - if (GNUNET_YES == - GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp", - "BINDTO", &plugin->bind4_address)) +/** + * We have been notified that our readset has something to read. We don't + * know which socket needs to be read, so we have to check each one + * Then reschedule this function to be called again once more is available. + * + * @param cls the plugin handle + * @param tc the scheduling context (for rescheduling this function again) + */ +static void +udp_plugin_select_v6 (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + struct Plugin *plugin = cls; + + plugin->select_task_v6 = GNUNET_SCHEDULER_NO_TASK; + if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) + return; + + plugin->with_v6_ws = GNUNET_NO; + if ((tc->reason & GNUNET_SCHEDULER_REASON_READ_READY) != 0) { - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Binding udp plugin to specific address: `%s'\n", - plugin->bind4_address); - if (1 != inet_pton (AF_INET, plugin->bind4_address, &serverAddrv4.sin_addr)) - { - GNUNET_free (plugin->bind4_address); - GNUNET_free (plugin); - GNUNET_free (api); - return NULL; - } + if ((NULL != plugin->sockv6) && + (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->sockv6))) + udp_select_read (plugin, plugin->sockv6); } - if (GNUNET_YES == - GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp", - "BINDTO6", &plugin->bind6_address)) + if ((tc->reason & GNUNET_SCHEDULER_REASON_WRITE_READY) != 0) { - LOG (GNUNET_ERROR_TYPE_DEBUG, - "Binding udp plugin to specific address: `%s'\n", - plugin->bind6_address); - if (1 != - inet_pton (AF_INET6, plugin->bind6_address, &serverAddrv6.sin6_addr)) - { - LOG (GNUNET_ERROR_TYPE_ERROR, _("Invalid IPv6 address: `%s'\n"), - plugin->bind6_address); - GNUNET_free_non_null (plugin->bind4_address); - GNUNET_free (plugin->bind6_address); - GNUNET_free (plugin); - GNUNET_free (api); - return NULL; - } + if ((NULL != plugin->sockv6) && (plugin->ipv6_queue_head != NULL) && + (GNUNET_NETWORK_fdset_isset (tc->write_ready, plugin->sockv6))) + { + udp_select_send (plugin, plugin->sockv6); + } } + if (plugin->select_task_v6 != GNUNET_SCHEDULER_NO_TASK) + GNUNET_SCHEDULER_cancel (plugin->select_task_v6); + plugin->select_task_v6 = GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v6, + (plugin->ipv6_queue_head != NULL) ? plugin->ws_v6 : NULL, + &udp_plugin_select_v6, plugin); + if (plugin->ipv6_queue_head != NULL) + plugin->with_v6_ws = GNUNET_YES; + else + plugin->with_v6_ws = GNUNET_NO; +} - plugin->defrags = - GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN); - plugin->sessions = - GNUNET_CONTAINER_multihashmap_create (UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG - * 2); - plugin->inbound_sessions = - GNUNET_CONTAINER_multihashmap_create (UDP_MAX_SENDER_ADDRESSES_WITH_DEFRAG - * 2); - sockets_created = 0; - if ((GNUNET_YES != - GNUNET_CONFIGURATION_get_value_yesno (plugin->env->cfg, "nat", - "DISABLEV6"))) + +static int +setup_sockets (struct Plugin *plugin, struct sockaddr_in6 *serverAddrv6, struct sockaddr_in *serverAddrv4) +{ + int tries; + int sockets_created = 0; + struct sockaddr *serverAddr; + struct sockaddr *addrs[2]; + socklen_t addrlens[2]; + socklen_t addrlen; + + /* Create IPv6 socket */ + if (plugin->enable_ipv6 == GNUNET_YES) { plugin->sockv6 = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_DGRAM, 0); if (NULL == plugin->sockv6) { - GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "socket"); + LOG (GNUNET_ERROR_TYPE_WARNING, "Disabling IPv6 since it is not supported on this system!\n"); + plugin->enable_ipv6 = GNUNET_NO; } else { #if HAVE_SOCKADDR_IN_SIN_LEN - serverAddrv6.sin6_len = sizeof (serverAddrv6); + serverAddrv6->sin6_len = sizeof (serverAddrv6); #endif - serverAddrv6.sin6_family = AF_INET6; - serverAddrv6.sin6_addr = in6addr_any; - serverAddrv6.sin6_port = htons (plugin->port); - addrlen = sizeof (serverAddrv6); - serverAddr = (struct sockaddr *) &serverAddrv6; -#if DEBUG_UDP + serverAddrv6->sin6_family = AF_INET6; + serverAddrv6->sin6_addr = in6addr_any; + serverAddrv6->sin6_port = htons (plugin->port); + addrlen = sizeof (struct sockaddr_in6); + serverAddr = (struct sockaddr *) serverAddrv6; LOG (GNUNET_ERROR_TYPE_DEBUG, "Binding to IPv6 port %d\n", - ntohs (serverAddrv6.sin6_port)); -#endif + ntohs (serverAddrv6->sin6_port)); tries = 0; while (GNUNET_NETWORK_socket_bind (plugin->sockv6, serverAddr, addrlen) != GNUNET_OK) { - serverAddrv6.sin6_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000); /* Find a good, non-root port */ -#if DEBUG_UDP + serverAddrv6->sin6_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000); /* Find a good, non-root port */ LOG (GNUNET_ERROR_TYPE_DEBUG, - "IPv6 Binding failed, trying new port %d\n", - ntohs (serverAddrv6.sin6_port)); -#endif + "IPv6 Binding failed, trying new port %d\n", + ntohs (serverAddrv6->sin6_port)); tries++; if (tries > 10) { @@ -2011,15 +2095,17 @@ libgnunet_plugin_transport_udp_init (void *cls) } if (plugin->sockv6 != NULL) { - addrs[sockets_created] = (struct sockaddr *) &serverAddrv6; - addrlens[sockets_created] = sizeof (serverAddrv6); + LOG (GNUNET_ERROR_TYPE_DEBUG, + "IPv6 socket created on port %d\n", + ntohs (serverAddrv6->sin6_port)); + addrs[sockets_created] = (struct sockaddr *) serverAddrv6; + addrlens[sockets_created] = sizeof (struct sockaddr_in6); sockets_created++; } } } - plugin->mst = - GNUNET_SERVER_mst_create (&process_inbound_tokenized_messages, plugin); + /* Create IPv4 socket */ plugin->sockv4 = GNUNET_NETWORK_socket_create (PF_INET, SOCK_DGRAM, 0); if (NULL == plugin->sockv4) { @@ -2028,27 +2114,23 @@ libgnunet_plugin_transport_udp_init (void *cls) else { #if HAVE_SOCKADDR_IN_SIN_LEN - serverAddrv4.sin_len = sizeof (serverAddrv4); + serverAddrv4->sin_len = sizeof (serverAddrv4); #endif - serverAddrv4.sin_family = AF_INET; - serverAddrv4.sin_addr.s_addr = INADDR_ANY; - serverAddrv4.sin_port = htons (plugin->port); - addrlen = sizeof (serverAddrv4); - serverAddr = (struct sockaddr *) &serverAddrv4; -#if DEBUG_UDP + serverAddrv4->sin_family = AF_INET; + serverAddrv4->sin_addr.s_addr = INADDR_ANY; + serverAddrv4->sin_port = htons (plugin->port); + addrlen = sizeof (struct sockaddr_in); + serverAddr = (struct sockaddr *) serverAddrv4; + LOG (GNUNET_ERROR_TYPE_DEBUG, "Binding to IPv4 port %d\n", - ntohs (serverAddrv4.sin_port)); -#endif + ntohs (serverAddrv4->sin_port)); tries = 0; while (GNUNET_NETWORK_socket_bind (plugin->sockv4, serverAddr, addrlen) != GNUNET_OK) { - serverAddrv4.sin_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000); /* Find a good, non-root port */ -#if DEBUG_UDP - LOG (GNUNET_ERROR_TYPE_DEBUG, - "IPv4 Binding failed, trying new port %d\n", - ntohs (serverAddrv4.sin_port)); -#endif + serverAddrv4->sin_port = htons (GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000); /* Find a good, non-root port */ + LOG (GNUNET_ERROR_TYPE_DEBUG, "IPv4 Binding failed, trying new port %d\n", + ntohs (serverAddrv4->sin_port)); tries++; if (tries > 10) { @@ -2059,100 +2141,338 @@ libgnunet_plugin_transport_udp_init (void *cls) } if (plugin->sockv4 != NULL) { - addrs[sockets_created] = (struct sockaddr *) &serverAddrv4; - addrlens[sockets_created] = sizeof (serverAddrv4); + addrs[sockets_created] = (struct sockaddr *) serverAddrv4; + addrlens[sockets_created] = sizeof (struct sockaddr_in); sockets_created++; } } - plugin->rs = GNUNET_NETWORK_fdset_create (); - GNUNET_NETWORK_fdset_zero (plugin->rs); + /* Create file descriptors */ + plugin->rs_v4 = GNUNET_NETWORK_fdset_create (); + plugin->ws_v4 = GNUNET_NETWORK_fdset_create (); + GNUNET_NETWORK_fdset_zero (plugin->rs_v4); + GNUNET_NETWORK_fdset_zero (plugin->ws_v4); if (NULL != plugin->sockv4) - GNUNET_NETWORK_fdset_set (plugin->rs, plugin->sockv4); - if (NULL != plugin->sockv6) - GNUNET_NETWORK_fdset_set (plugin->rs, plugin->sockv6); + { + GNUNET_NETWORK_fdset_set (plugin->rs_v4, plugin->sockv4); + GNUNET_NETWORK_fdset_set (plugin->ws_v4, plugin->sockv4); + } + + if (sockets_created == 0) + LOG (GNUNET_ERROR_TYPE_WARNING, _("Failed to open UDP sockets\n")); plugin->select_task = GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, - GNUNET_SCHEDULER_NO_TASK, - GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs, - NULL, &udp_plugin_select, plugin); - + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v4, + NULL, + &udp_plugin_select, plugin); + plugin->with_v4_ws = GNUNET_NO; - /* create broadcast socket */ - if (broadcast) + if (plugin->enable_ipv6 == GNUNET_YES) { - plugin->sockv4_broadcast = GNUNET_NETWORK_socket_create (PF_INET, SOCK_DGRAM, 0); - if (NULL == plugin->sockv4_broadcast) + plugin->rs_v6 = GNUNET_NETWORK_fdset_create (); + plugin->ws_v6 = GNUNET_NETWORK_fdset_create (); + GNUNET_NETWORK_fdset_zero (plugin->rs_v6); + GNUNET_NETWORK_fdset_zero (plugin->ws_v6); + if (NULL != plugin->sockv6) { - GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "socket"); + GNUNET_NETWORK_fdset_set (plugin->rs_v6, plugin->sockv6); + GNUNET_NETWORK_fdset_set (plugin->ws_v6, plugin->sockv6); } - else + + plugin->select_task_v6 = + GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, + GNUNET_TIME_UNIT_FOREVER_REL, + plugin->rs_v6, + NULL, + &udp_plugin_select_v6, plugin); + plugin->with_v6_ws = GNUNET_NO; + } + + plugin->nat = GNUNET_NAT_register (plugin->env->cfg, + GNUNET_NO, plugin->port, + sockets_created, + (const struct sockaddr **) addrs, addrlens, + &udp_nat_port_map_callback, NULL, plugin); + + return sockets_created; +} + +/** + * Session was idle, so disconnect it + */ +static void +session_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + GNUNET_assert (NULL != cls); + struct Session *s = cls; + + s->timeout_task = GNUNET_SCHEDULER_NO_TASK; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Session %p was idle for %llu ms, disconnecting\n", + s, (unsigned long long) GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value); + /* call session destroy function */ + disconnect_session(s); +} + + +/** + * Start session timeout + */ +static void +start_session_timeout (struct Session *s) +{ + GNUNET_assert (NULL != s); + GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == s->timeout_task); + s->timeout_task = GNUNET_SCHEDULER_add_delayed (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, + &session_timeout, + s); + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Timeout for session %p set to %llu ms\n", + s, (unsigned long long) GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value); +} + + +/** + * Increment session timeout due to activity + */ +static void +reschedule_session_timeout (struct Session *s) +{ + GNUNET_assert (NULL != s); + GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != s->timeout_task); + + GNUNET_SCHEDULER_cancel (s->timeout_task); + s->timeout_task = GNUNET_SCHEDULER_add_delayed (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT, + &session_timeout, + s); + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Timeout rescheduled for session %p set to %llu ms\n", + s, (unsigned long long) GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value); +} + + +/** + * Cancel timeout + */ +static void +stop_session_timeout (struct Session *s) +{ + GNUNET_assert (NULL != s); + + if (GNUNET_SCHEDULER_NO_TASK != s->timeout_task) + { + GNUNET_SCHEDULER_cancel (s->timeout_task); + s->timeout_task = GNUNET_SCHEDULER_NO_TASK; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Timeout stopped for session %p canceled\n", + s, (unsigned long long) GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value); + } +} + + + +/** + * The exported method. Makes the core api available via a global and + * returns the udp transport API. + * + * @param cls our 'struct GNUNET_TRANSPORT_PluginEnvironment' + * @return our 'struct GNUNET_TRANSPORT_PluginFunctions' + */ +void * +libgnunet_plugin_transport_udp_init (void *cls) +{ + struct GNUNET_TRANSPORT_PluginEnvironment *env = cls; + struct GNUNET_TRANSPORT_PluginFunctions *api; + struct Plugin *p; + unsigned long long port; + unsigned long long aport; + unsigned long long broadcast; + unsigned long long udp_max_bps; + unsigned long long enable_v6; + char * bind4_address; + char * bind6_address; + char * fancy_interval; + struct GNUNET_TIME_Relative interval; + struct sockaddr_in serverAddrv4; + struct sockaddr_in6 serverAddrv6; + int res; + + if (NULL == env->receive) + { + /* run in 'stub' mode (i.e. as part of gnunet-peerinfo), don't fully + initialze the plugin or the API */ + api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions)); + api->cls = NULL; + api->address_pretty_printer = &udp_plugin_address_pretty_printer; + api->address_to_string = &udp_address_to_string; + api->string_to_address = &udp_string_to_address; + return api; + } + + GNUNET_assert( NULL != env->stats); + + /* Get port number */ + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", "PORT", + &port)) + port = 2086; + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", + "ADVERTISED_PORT", &aport)) + aport = port; + if (port > 65535) + { + LOG (GNUNET_ERROR_TYPE_WARNING, + _("Given `%s' option is out of range: %llu > %u\n"), "PORT", port, + 65535); + return NULL; + } + + /* Protocols */ + if ((GNUNET_YES == + GNUNET_CONFIGURATION_get_value_yesno (env->cfg, "nat", + "DISABLEV6"))) + { + enable_v6 = GNUNET_NO; + } + else + enable_v6 = GNUNET_YES; + + /* Addresses */ + memset (&serverAddrv6, 0, sizeof (serverAddrv6)); + memset (&serverAddrv4, 0, sizeof (serverAddrv4)); + + if (GNUNET_YES == + GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp", + "BINDTO", &bind4_address)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Binding udp plugin to specific address: `%s'\n", + bind4_address); + if (1 != inet_pton (AF_INET, bind4_address, &serverAddrv4.sin_addr)) { - #if HAVE_SOCKADDR_IN_SIN_LEN - serverAddrv4.sin_len = sizeof (serverAddrv4); - #endif - serverAddrv4.sin_family = AF_INET; - serverAddrv4.sin_addr.s_addr = INADDR_ANY; - serverAddrv4.sin_port = htons (plugin->broadcast_port); - addrlen = sizeof (serverAddrv4); - serverAddr = (struct sockaddr *) &serverAddrv4; - #if DEBUG_UDP - #endif - LOG (GNUNET_ERROR_TYPE_DEBUG, "Binding Broadcast to IPv4 port %d\n", - ntohs (serverAddrv4.sin_port)); - - if (GNUNET_NETWORK_socket_bind (plugin->sockv4_broadcast, serverAddr, addrlen) != GNUNET_OK) - { - GNUNET_NETWORK_socket_close (plugin->sockv4_broadcast); - plugin->sockv4_broadcast = NULL; - } - if (plugin->sockv4_broadcast != NULL) - { - int yes = 1; - if (GNUNET_NETWORK_socket_setsockopt (plugin->sockv4_broadcast, SOL_SOCKET, SO_BROADCAST, &yes, sizeof(int)) != GNUNET_OK) - { - GNUNET_NETWORK_socket_close(plugin->sockv4_broadcast); - plugin->sockv4_broadcast = NULL; - } - else - { - plugin->broadcast_rs = GNUNET_NETWORK_fdset_create (); - GNUNET_NETWORK_fdset_set (plugin->broadcast_rs, plugin->sockv4_broadcast); - } - } + GNUNET_free (bind4_address); + return NULL; } + } - if (plugin->sockv4_broadcast != NULL) + if (GNUNET_YES == + GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp", + "BINDTO6", &bind6_address)) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Binding udp plugin to specific address: `%s'\n", + bind6_address); + if (1 != + inet_pton (AF_INET6, bind6_address, &serverAddrv6.sin6_addr)) { - plugin->broadcast = GNUNET_YES; - plugin->broadcast_mst = GNUNET_SERVER_mst_create (udp_broadcast_mst_cb, plugin); - GNUNET_STATISTICS_update(plugin->env->cfg, _("# HELLO beacons received via udp"), 1, GNUNET_NO); - plugin->select_broadcast_task = - GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT, - GNUNET_SCHEDULER_NO_TASK, - GNUNET_TIME_UNIT_FOREVER_REL, plugin->broadcast_rs, - NULL, &udp_plugin_broadcast_select, plugin); - - plugin->send_broadcast_task = GNUNET_SCHEDULER_add_now (&udp_broadcast_send, plugin); + LOG (GNUNET_ERROR_TYPE_ERROR, _("Invalid IPv6 address: `%s'\n"), + bind6_address); + GNUNET_free_non_null (bind4_address); + GNUNET_free (bind6_address); + return NULL; } - else - plugin->broadcast = GNUNET_NO; } - if (sockets_created == 0) - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, _("Failed to open UDP sockets\n")); - plugin->nat = - GNUNET_NAT_register (env->cfg, GNUNET_NO, port, sockets_created, - (const struct sockaddr **) addrs, addrlens, - &udp_nat_port_map_callback, NULL, plugin); + /* Enable neighbour discovery */ + broadcast = GNUNET_CONFIGURATION_get_value_yesno (env->cfg, "transport-udp", + "BROADCAST"); + if (broadcast == GNUNET_SYSERR) + broadcast = GNUNET_NO; + + if (GNUNET_SYSERR == GNUNET_CONFIGURATION_get_value_string (env->cfg, "transport-udp", + "BROADCAST_INTERVAL", &fancy_interval)) + { + interval = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10); + } + else + { + if (GNUNET_SYSERR == GNUNET_STRINGS_fancy_time_to_relative(fancy_interval, &interval)) + { + interval = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30); + } + GNUNET_free (fancy_interval); + } + + /* Maximum datarate */ + if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-udp", + "MAX_BPS", &udp_max_bps)) + { + udp_max_bps = 1024 * 1024 * 50; /* 50 MB/s == infinity for practical purposes */ + } + + p = GNUNET_malloc (sizeof (struct Plugin)); + api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions)); + + GNUNET_BANDWIDTH_tracker_init (&p->tracker, + GNUNET_BANDWIDTH_value_init ((uint32_t)udp_max_bps), 30); + p->sessions = GNUNET_CONTAINER_multihashmap_create (10); + p->defrag_ctxs = GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN); + p->mst = GNUNET_SERVER_mst_create (&process_inbound_tokenized_messages, p); + p->port = port; + p->aport = aport; + p->broadcast_interval = interval; + p->enable_ipv6 = enable_v6; + p->env = env; + + plugin = p; + + api->cls = p; + api->send = NULL; + api->disconnect = &udp_disconnect; + api->address_pretty_printer = &udp_plugin_address_pretty_printer; + api->address_to_string = &udp_address_to_string; + api->string_to_address = &udp_string_to_address; + api->check_address = &udp_plugin_check_address; + api->get_session = &udp_plugin_get_session; + api->send = &udp_plugin_send; + + LOG (GNUNET_ERROR_TYPE_DEBUG, "Setting up sockets\n"); + res = setup_sockets (p, &serverAddrv6, &serverAddrv4); + if ((res == 0) || ((p->sockv4 == NULL) && (p->sockv6 == NULL))) + { + LOG (GNUNET_ERROR_TYPE_ERROR, "Failed to create network sockets, plugin failed\n"); + GNUNET_free (p); + GNUNET_free (api); + return NULL; + } + + if (broadcast == GNUNET_YES) + { + LOG (GNUNET_ERROR_TYPE_DEBUG, "Starting broadcasting\n"); + setup_broadcast (p, &serverAddrv6, &serverAddrv4); + } + + GNUNET_free_non_null (bind4_address); + GNUNET_free_non_null (bind6_address); return api; } + +static int +heap_cleanup_iterator (void *cls, + struct GNUNET_CONTAINER_HeapNode * + node, void *element, + GNUNET_CONTAINER_HeapCostType + cost) +{ + struct DefragContext * d_ctx = element; + + GNUNET_CONTAINER_heap_remove_node (node); + GNUNET_DEFRAGMENT_context_destroy(d_ctx->defrag); + GNUNET_free (d_ctx); + + return GNUNET_YES; +} + + /** - * Shutdown the plugin. + * The exported method. Makes the core api available via a global and + * returns the udp transport API. * - * @param cls our 'struct GNUNET_TRANSPORT_PluginFunctions' + * @param cls our 'struct GNUNET_TRANSPORT_PluginEnvironment' * @return NULL */ void * @@ -2160,70 +2480,91 @@ libgnunet_plugin_transport_udp_done (void *cls) { struct GNUNET_TRANSPORT_PluginFunctions *api = cls; struct Plugin *plugin = api->cls; - struct ReceiveContext *rc; - /* FIXME: clean up heap and hashmap */ - GNUNET_CONTAINER_multihashmap_iterate (plugin->sessions, &destroy_session, - NULL); - GNUNET_CONTAINER_multihashmap_destroy (plugin->sessions); - plugin->sessions = NULL; - GNUNET_CONTAINER_multihashmap_iterate (plugin->inbound_sessions, &destroy_inbound_session, - NULL); - GNUNET_CONTAINER_multihashmap_destroy (plugin->inbound_sessions); - plugin->inbound_sessions = NULL; - while (NULL != (rc = GNUNET_CONTAINER_heap_remove_root (plugin->defrags))) + if (NULL == plugin) { - GNUNET_DEFRAGMENT_context_destroy (rc->defrag); - GNUNET_free (rc); + GNUNET_free (api); + return NULL; } - GNUNET_CONTAINER_heap_destroy (plugin->defrags); + + stop_broadcast (plugin); if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK) { GNUNET_SCHEDULER_cancel (plugin->select_task); plugin->select_task = GNUNET_SCHEDULER_NO_TASK; } + if (plugin->select_task_v6 != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (plugin->select_task_v6); + plugin->select_task_v6 = GNUNET_SCHEDULER_NO_TASK; + } + + /* Closing sockets */ if (plugin->sockv4 != NULL) { GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->sockv4)); plugin->sockv4 = NULL; } + GNUNET_NETWORK_fdset_destroy (plugin->rs_v4); + GNUNET_NETWORK_fdset_destroy (plugin->ws_v4); + if (plugin->sockv6 != NULL) { GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->sockv6)); plugin->sockv6 = NULL; + + GNUNET_NETWORK_fdset_destroy (plugin->rs_v6); + GNUNET_NETWORK_fdset_destroy (plugin->ws_v6); } - if (plugin->broadcast) + GNUNET_NAT_unregister (plugin->nat); + + if (plugin->defrag_ctxs != NULL) { - if (plugin->select_broadcast_task != GNUNET_SCHEDULER_NO_TASK) - { - GNUNET_SCHEDULER_cancel (plugin->select_broadcast_task); - plugin->select_broadcast_task = GNUNET_SCHEDULER_NO_TASK; - } - if (plugin->send_broadcast_task != GNUNET_SCHEDULER_NO_TASK) - { - GNUNET_SCHEDULER_cancel (plugin->send_broadcast_task ); - plugin->send_broadcast_task = GNUNET_SCHEDULER_NO_TASK; - } - if (plugin->broadcast_mst != NULL) - GNUNET_SERVER_mst_destroy(plugin->broadcast_mst); - if (plugin->sockv4_broadcast != NULL) - { - GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->sockv4_broadcast)); - plugin->sockv4_broadcast = NULL; - } - GNUNET_NETWORK_fdset_destroy (plugin->broadcast_rs); + GNUNET_CONTAINER_heap_iterate(plugin->defrag_ctxs, + heap_cleanup_iterator, NULL); + GNUNET_CONTAINER_heap_destroy(plugin->defrag_ctxs); + plugin->defrag_ctxs = NULL; + } + if (plugin->mst != NULL) + { + GNUNET_SERVER_mst_destroy(plugin->mst); + plugin->mst = NULL; + } + + /* Clean up leftover messages */ + struct UDPMessageWrapper * udpw; + udpw = plugin->ipv4_queue_head; + while (udpw != NULL) + { + struct UDPMessageWrapper *tmp = udpw->next; + GNUNET_CONTAINER_DLL_remove(plugin->ipv4_queue_head, plugin->ipv4_queue_tail, udpw); + call_continuation(udpw, GNUNET_SYSERR); + GNUNET_free (udpw); + udpw = tmp; + } + udpw = plugin->ipv6_queue_head; + while (udpw != NULL) + { + struct UDPMessageWrapper *tmp = udpw->next; + GNUNET_CONTAINER_DLL_remove(plugin->ipv6_queue_head, plugin->ipv6_queue_tail, udpw); + call_continuation(udpw, GNUNET_SYSERR); + GNUNET_free (udpw); + udpw = tmp; } - GNUNET_SERVER_mst_destroy (plugin->mst); - GNUNET_NETWORK_fdset_destroy (plugin->rs); + /* Clean up sessions */ + LOG (GNUNET_ERROR_TYPE_DEBUG, + "Cleaning up sessions\n"); + GNUNET_CONTAINER_multihashmap_iterate (plugin->sessions, &disconnect_and_free_it, plugin); + GNUNET_CONTAINER_multihashmap_destroy (plugin->sessions); - GNUNET_NAT_unregister (plugin->nat); plugin->nat = NULL; GNUNET_free (plugin); GNUNET_free (api); return NULL; } + /* end of plugin_transport_udp.c */