fixing coverity 10465: Resource leak
[oweals/gnunet.git] / src / transport / plugin_transport_unix.c
index 2a7dc90650a1720654e3f50e87b837bf801ab5c8..064a91663e4341dba4e6831686ffa27fbe9757cb 100644 (file)
@@ -1,6 +1,6 @@
 /*
      This file is part of GNUnet
-     (C) 2010 Christian Grothoff (and other contributing authors)
+     (C) 2010, 2013 Christian Grothoff (and other contributing authors)
 
      GNUnet is free software; you can redistribute it and/or modify
      it under the terms of the GNU General Public License as published
  * @author Christian Grothoff
  * @author Nathan Evans
  */
-
 #include "platform.h"
 #include "gnunet_hello_lib.h"
-#include "gnunet_connection_lib.h"
-#include "gnunet_container_lib.h"
-#include "gnunet_os_lib.h"
-#include "gnunet_peerinfo_service.h"
+#include "gnunet_util_lib.h"
 #include "gnunet_protocols.h"
-#include "gnunet_resolver_service.h"
-#include "gnunet_server_lib.h"
-#include "gnunet_signatures.h"
 #include "gnunet_statistics_service.h"
 #include "gnunet_transport_service.h"
 #include "gnunet_transport_plugin.h"
 #include "transport.h"
 
-#define DEBUG_UNIX GNUNET_NO
-#define DETAILS GNUNET_NO
-
-#define MAX_PROBES 20
 
-/*
- * Transport cost to peer, always 1 for UNIX (direct connection)
+/**
+ * Return code we give on 'send' if we failed to send right now
+ * but it makes sense to retry later. (Note: we might want to
+ * move this to the plugin API!?).
  */
-#define UNIX_DIRECT_DISTANCE 1
-
-#define DEFAULT_NAT_PORT 0
+#define RETRY 0
 
 /**
  * How long until we give up on transmitting the welcome message?
 #define HOSTNAME_RESOLVE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
 
 /**
- * Starting port for listening and sending, eventually a config value
+ * Default "port" to use, if configuration does not specify.
+ * Essentially just a number appended to the UNIX path.
  */
 #define UNIX_NAT_DEFAULT_PORT 22086
 
+
+#define LOG(kind,...) GNUNET_log_from (kind, "transport-unix",__VA_ARGS__)
+
+
+GNUNET_NETWORK_STRUCT_BEGIN
+
 /**
  * UNIX Message-Packet header.
  */
@@ -81,145 +77,89 @@ struct UNIXMessage
 
 };
 
-struct RetryList
-{
-  /**
-   * Pointer to next element.
-   */
-  struct RetryList *next;
-
-  /**
-   * Pointer to previous element.
-   */
-  struct RetryList *prev;
-
-  /**
-   * The actual retry context.
-   */
-  struct RetrySendContext *retry_ctx;
-};
 
 /**
- * Network format for IPv4 addresses.
+ * Handle for a session.
  */
-struct IPv4UdpAddress
+struct Session
 {
-  /**
-   * IPv4 address, in network byte order.
-   */
-  uint32_t ipv4_addr GNUNET_PACKED;
+  struct GNUNET_PeerIdentity target;
 
-  /**
-   * Port number, in network byte order.
-   */
-  uint16_t u_port GNUNET_PACKED;
-};
+  struct Plugin * plugin;
 
+  void *addr;
 
-/**
- * Network format for IPv6 addresses.
- */
-struct IPv6UdpAddress
-{
-  /**
-   * IPv6 address.
-   */
-  struct in6_addr ipv6_addr GNUNET_PACKED;
+  size_t addrlen;
 
   /**
-   * Port number, in network byte order.
+   * Session timeout task
    */
-  uint16_t u6_port GNUNET_PACKED;
+  GNUNET_SCHEDULER_TaskIdentifier timeout_task;
 };
 
-/* Forward definition */
-struct Plugin;
 
-struct PrettyPrinterContext
+struct UNIXMessageWrapper
 {
-  GNUNET_TRANSPORT_AddressStringCallback asc;
-  void *asc_cls;
-  uint16_t port;
-};
-
-struct RetrySendContext
-{
-
   /**
-   * Main plugin handle.
+   * We keep messages in a doubly linked list.
    */
-  struct Plugin *plugin;
+  struct UNIXMessageWrapper *next;
 
   /**
-   * Address of recipient.
+   * We keep messages in a doubly linked list.
    */
-  char *addr;
+  struct UNIXMessageWrapper *prev;
 
   /**
-   * Length of address.
+   * The actual payload (allocated separately right now).
    */
-  ssize_t addrlen;
+  struct UNIXMessage * msg;
 
   /**
-   * Message to send.
+   * Session this message belongs to.
    */
-  char *msg;
+  struct Session *session;
 
   /**
-   * Size of the message.
-   */
-  int msg_size;
-
-  /**
-   * Handle to send message out on.
-   */
-  struct GNUNET_NETWORK_Handle *send_handle;
-
-  /**
-   * Continuation to call on success or
-   * timeout.
+   * Function to call upon transmission.
    */
   GNUNET_TRANSPORT_TransmitContinuation cont;
 
   /**
-   * Closure for continuation.
+   * Closure for 'cont'.
    */
   void *cont_cls;
 
   /**
-   * The peer the message is destined for.
-   */
-  struct GNUNET_PeerIdentity target;
-
-  /**
-   * How long before not retrying any longer.
+   * Timeout for this message.
    */
   struct GNUNET_TIME_Absolute timeout;
 
   /**
-   * How long the last message was delayed.
+   * Number of bytes in 'msg'.
    */
-  struct GNUNET_TIME_Relative delay;
+  size_t msgsize;
 
   /**
-   * The actual retry task.
+   * Number of bytes of payload encapsulated in 'msg'.
    */
-  GNUNET_SCHEDULER_TaskIdentifier retry_task;
+  size_t payload;
 
   /**
-   * The priority of the message.
+   * Priority of the message (ignored, just dragged along in UNIX).
    */
   unsigned int priority;
-
-  /**
-   * Entry in the DLL of retry items.
-   */
-  struct RetryList *retry_list_entry;
 };
 
 
 /**
- * UNIX NAT "Session"
+ * Encapsulation of all of the state of the plugin.
+ */
+struct Plugin;
+
+
+/**
+ * UNIX "Session"
  */
 struct PeerSession
 {
@@ -269,6 +209,7 @@ struct PeerSession
 
 };
 
+
 /**
  * Information we keep for each of our listen sockets.
  */
@@ -280,7 +221,9 @@ struct UNIX_Sock_Info
   struct GNUNET_NETWORK_Handle *desc;
 
   /**
-   * The port we bound to
+   * The port we bound to (not an actual PORT, as UNIX domain sockets
+   * don't have ports, but rather a number in the path name to make this
+   * one unique).
    */
   uint16_t port;
 };
@@ -291,15 +234,6 @@ struct UNIX_Sock_Info
  */
 struct Plugin
 {
-  /**
-   * Our environment.
-   */
-  struct GNUNET_TRANSPORT_PluginEnvironment *env;
-
-  /*
-   * Session of peers with whom we are currently connected
-   */
-  struct PeerSession *sessions;
 
   /**
    * ID of task used to update our addresses when one expires.
@@ -312,9 +246,19 @@ struct Plugin
   GNUNET_SCHEDULER_TaskIdentifier select_task;
 
   /**
-   * Integer to append to unix domain socket.
+   * Number of bytes we currently have in our write queue.
    */
-  uint16_t port;
+  unsigned long long bytes_in_queue;
+
+  /**
+   * Our environment.
+   */
+  struct GNUNET_TRANSPORT_PluginEnvironment *env;
+
+  /**
+   * Sessions
+   */
+  struct GNUNET_CONTAINER_MultiHashMap *session_map;
 
   /**
    * FD Read set
@@ -322,157 +266,237 @@ struct Plugin
   struct GNUNET_NETWORK_FDSet *rs;
 
   /**
-   * socket that we transmit all data with
+   * FD Write set
    */
-  struct UNIX_Sock_Info unix_sock;
+  struct GNUNET_NETWORK_FDSet *ws;
 
   /**
    * Path of our unix domain socket (/tmp/unix-plugin-PORT)
    */
   char *unix_socket_path;
 
+  /**
+   * Head of queue of messages to transmit.
+   */
+  struct UNIXMessageWrapper *msg_head;
+
+  /**
+   * Tail of queue of messages to transmit.
+   */
+  struct UNIXMessageWrapper *msg_tail;
+
+  /**
+   * socket that we transmit all data with
+   */
+  struct UNIX_Sock_Info unix_sock;
+
+  /**
+   * ATS network
+   */
+  struct GNUNET_ATS_Information ats_network;
+
+  /**
+   * Is the write set in the current 'select' task?  GNUNET_NO if the
+   * write queue was empty when the main task was scheduled,
+   * GNUNET_YES if we're already waiting for being allowed to write.
+   */
+  int with_ws;
+
+  /**
+   * Integer to append to unix domain socket.
+   */
+  uint16_t port;
+
 };
 
+
 /**
- * Head of retry DLL.
+ * Increment session timeout due to activity
+ *
+ * @param s session for which the timeout should be moved
  */
-static struct RetryList *retry_list_head;
+static void
+reschedule_session_timeout (struct Session *s);
+
 
 /**
- * Tail of retry DLL.
+ * 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.
+ *
+ * @param cls the plugin handle
+ * @param tc the scheduling context (for rescheduling this function again)
  */
-static struct RetryList *retry_list_tail;
+static void
+unix_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
 
 
 /**
- * Disconnect from a remote node.  Clean up session if we have one for this peer
+ * Re-schedule the main 'select' callback (unix_plugin_select)
+ * for this plugin.
  *
- * @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 plugin the plugin context
  */
-void
-unix_disconnect (void *cls, const struct GNUNET_PeerIdentity *target)
+static void
+reschedule_select (struct Plugin * plugin)
 {
-  /** TODO: Implement! */
-  return;
+  if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK)
+  {
+    GNUNET_SCHEDULER_cancel (plugin->select_task);
+    plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
+  }
+  if (NULL != plugin->msg_head)
+  {
+    plugin->select_task =
+      GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
+                                   GNUNET_TIME_UNIT_FOREVER_REL,
+                                   plugin->rs,
+                                   plugin->ws,
+                                   &unix_plugin_select, plugin);
+    plugin->with_ws = GNUNET_YES;
+  }
+  else
+  {
+    plugin->select_task =
+      GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
+                                   GNUNET_TIME_UNIT_FOREVER_REL,
+                                   plugin->rs,
+                                   NULL,
+                                   &unix_plugin_select, plugin);
+    plugin->with_ws = GNUNET_NO;
+  }
 }
 
+
 /**
- * Shutdown the server process (stop receiving inbound traffic). Maybe
- * restarted later!
- *
- * @param cls Handle to the plugin for this transport
- *
- * @return returns the number of sockets successfully closed,
- *         should equal the number of sockets successfully opened
+ * Closure to 'lookup_session_it'.
  */
-static int
-unix_transport_server_stop (void *cls)
+struct LookupCtx
 {
-  struct Plugin *plugin = cls;
-  struct RetryList *pos;
+  /**
+   * Location to store the session, if found.
+   */
+  struct Session *s;
 
-  pos = retry_list_head;
+  /**
+   * Address we are looking for.
+   */
+  const struct sockaddr_un *addr;
+};
 
-  while(NULL != (pos = retry_list_head))
-    {
-      GNUNET_CONTAINER_DLL_remove(retry_list_head, retry_list_tail, pos);
-      if (GNUNET_SCHEDULER_NO_TASK != pos->retry_ctx->retry_task)
-        {
-          GNUNET_SCHEDULER_cancel(pos->retry_ctx->retry_task);
-        }
-      GNUNET_free(pos->retry_ctx->msg);
-      GNUNET_free(pos->retry_ctx->addr);
-      GNUNET_free(pos->retry_ctx);
-      GNUNET_free(pos);
-    }
 
-  if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK)
-    {
-      GNUNET_SCHEDULER_cancel (plugin->select_task);
-      plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
-    }
-
-  GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (plugin->unix_sock.desc));
-  plugin->unix_sock.desc = NULL;
+/**
+ * Function called to find a session by address.
+ *
+ * @param cls the 'struct LookupCtx'
+ * @param key peer we are looking for (unused)
+ * @param value a session
+ * @return GNUNET_YES if not found (continue looking), GNUNET_NO on success
+ */
+static int 
+lookup_session_it (void *cls,
+                  const struct GNUNET_HashCode * key,
+                  void *value)
+{
+  struct LookupCtx *lctx = cls;
+  struct Session *t = value;
 
-  return GNUNET_OK;
+  if (0 == strcmp (t->addr, lctx->addr->sun_path))
+  {
+    lctx->s = t;
+    return GNUNET_NO;
+  }
+  return GNUNET_YES;
 }
 
 
-struct PeerSession *
-find_session (struct Plugin *plugin,
-             const struct GNUNET_PeerIdentity *peer)
+/**
+ * Find an existing session by address.
+ *
+ * @param plugin the plugin
+ * @param sender for which peer should the session be?
+ * @param addr address to look for
+ * @return NULL if session was not found
+ */
+static struct Session *
+lookup_session (struct Plugin *plugin, 
+               const struct GNUNET_PeerIdentity *sender, 
+               const struct sockaddr_un *addr)
 {
-  struct PeerSession *pos;
-
-  pos = plugin->sessions;
-  while (pos != NULL)
-    {
-      if (memcmp(&pos->target, peer, sizeof(struct GNUNET_PeerIdentity)) == 0)
-        return pos;
-      pos = pos->next;
-    }
-
-  return pos;
+  struct LookupCtx lctx;
+
+  GNUNET_assert (NULL != plugin);
+  GNUNET_assert (NULL != sender);
+  GNUNET_assert (NULL != addr);
+  lctx.s = NULL;
+  lctx.addr = addr;
+  GNUNET_CONTAINER_multihashmap_get_multiple (plugin->session_map, 
+                                             &sender->hashPubKey,
+                                             &lookup_session_it, &lctx);
+  return lctx.s;
 }
 
-/* Forward Declaration */
-static ssize_t
-unix_real_send (void *cls,
-                struct RetrySendContext *incoming_retry_context,
-                struct GNUNET_NETWORK_Handle *send_handle,
-                const struct GNUNET_PeerIdentity *target,
-                const char *msgbuf,
-                size_t msgbuf_size,
-                unsigned int priority,
-                struct GNUNET_TIME_Relative timeout,
-                const void *addr,
-                size_t addrlen,
-                GNUNET_TRANSPORT_TransmitContinuation cont,
-                void *cont_cls);
 
 /**
- * Retry sending a message.
+ * Functions with this signature are called whenever we need
+ * to close a session due to a disconnect or failure to
+ * establish a connection.
  *
- * @param cls closure a struct RetrySendContext
- * @param tc context information
+ * @param s session to close down
  */
-void retry_send_message (void *cls,
-                         const struct GNUNET_SCHEDULER_TaskContext * tc)
+static void
+disconnect_session (struct Session *s)
 {
-  struct RetrySendContext *retry_ctx = cls;
-
-  if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
-    {
-      GNUNET_free(retry_ctx->msg);
-      GNUNET_free(retry_ctx->addr);
-      GNUNET_free(retry_ctx);
-      return;
-    }
-
-  unix_real_send (retry_ctx->plugin,
-                  retry_ctx,
-                  retry_ctx->send_handle,
-                  &retry_ctx->target,
-                  retry_ctx->msg,
-                  retry_ctx->msg_size,
-                  retry_ctx->priority,
-                  GNUNET_TIME_absolute_get_remaining (retry_ctx->timeout),
-                  retry_ctx->addr,
-                  retry_ctx->addrlen,
-                  retry_ctx->cont,
-                  retry_ctx->cont_cls);
-  return;
+  struct Plugin *plugin = s->plugin;
+  struct UNIXMessageWrapper *msgw;
+  struct UNIXMessageWrapper *next;
+  int removed;
+
+  LOG (GNUNET_ERROR_TYPE_DEBUG, 
+       "Disconnecting session for peer `%s' `%s'\n",
+       GNUNET_i2s (&s->target), 
+       s->addr);
+  plugin->env->session_end (plugin->env->cls, &s->target, s);
+  removed = GNUNET_NO;
+  next = plugin->msg_head;
+  while (NULL != next)
+  {
+    msgw = next;
+    next = msgw->next;
+    if (msgw->session != s)
+      continue;
+    GNUNET_CONTAINER_DLL_remove (plugin->msg_head, plugin->msg_tail, msgw);
+    if (NULL != msgw->cont)
+      msgw->cont (msgw->cont_cls,  &msgw->session->target, GNUNET_SYSERR,
+                  msgw->payload, 0);
+    GNUNET_free (msgw->msg);
+    GNUNET_free (msgw);
+    removed = GNUNET_YES;    
+  }
+  GNUNET_assert (GNUNET_YES ==
+                 GNUNET_CONTAINER_multihashmap_remove (plugin->session_map, 
+                                                      &s->target.hashPubKey, 
+                                                      s));
+  GNUNET_STATISTICS_set (plugin->env->stats,
+                        "# UNIX sessions active",
+                        GNUNET_CONTAINER_multihashmap_size (plugin->session_map),
+                        GNUNET_NO);
+  if ((GNUNET_YES == removed) && (NULL == plugin->msg_head))
+    reschedule_select (plugin);
+  if (GNUNET_SCHEDULER_NO_TASK != s->timeout_task)
+  {
+    GNUNET_SCHEDULER_cancel (s->timeout_task);
+    s->timeout_task = GNUNET_SCHEDULER_NO_TASK;
+  }
+  GNUNET_free (s);
 }
 
+
 /**
  * Actually send out the message, assume we've got the address and
  * send_handle squared away!
  *
  * @param cls closure
- * @param incoming_retry_context the retry context to use
  * @param send_handle which handle to send message on
  * @param target who should receive this message (ignored by UNIX)
  * @param msgbuf one or more GNUNET_MessageHeader(s) strung together
@@ -481,79 +505,52 @@ void retry_send_message (void *cls,
  * @param timeout when should we time out (give up) if we can not transmit?
  * @param addr the addr to send the message to, needs to be a sockaddr for us
  * @param addrlen the len of addr
+ * @param payload bytes payload to send
  * @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
- *
- * @return the number of bytes written, -1 on errors
+ * @return on success the number of bytes written, RETRY for retry, -1 on errors
  */
 static ssize_t
 unix_real_send (void *cls,
-                struct RetrySendContext *incoming_retry_context,
                 struct GNUNET_NETWORK_Handle *send_handle,
-               const struct GNUNET_PeerIdentity *target,
-               const char *msgbuf,
-               size_t msgbuf_size,
-               unsigned int priority,
-               struct GNUNET_TIME_Relative timeout,
-               const void *addr,
-               size_t addrlen,
-               GNUNET_TRANSPORT_TransmitContinuation cont,
-               void *cont_cls)
+                const struct GNUNET_PeerIdentity *target, const char *msgbuf,
+                size_t msgbuf_size, unsigned int priority,
+                struct GNUNET_TIME_Absolute timeout,
+                const void *addr,
+                size_t addrlen,
+                size_t payload,
+                GNUNET_TRANSPORT_TransmitContinuation cont,
+                void *cont_cls)
 {
   struct Plugin *plugin = cls;
-  struct UNIXMessage *message;
-  struct RetrySendContext *retry_ctx;
-  int ssize;
   ssize_t sent;
   const void *sb;
   size_t sbs;
   struct sockaddr_un un;
   size_t slen;
-  struct RetryList *retry_list_entry;
-  int retry;
-
-  if (send_handle == NULL)
-    {
-#if DEBUG_UNIX
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                       "unix_real_send with send_handle NULL!\n");
-#endif
-      /* failed to open send socket for AF */
-      if (cont != NULL)
-        cont (cont_cls, target, GNUNET_SYSERR);
-      return 0;
-    }
-  if ((addr == NULL) || (addrlen == 0))
-    {
-#if DEBUG_UNIX
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                      "unix_real_send called without address, returning!\n");
-#endif
-      if (cont != NULL)
-        cont (cont_cls, target, GNUNET_SYSERR);
-      return 0; /* Can never send if we don't have an address!! */
-    }
 
-  /* Build the message to be sent */
-  message = GNUNET_malloc (sizeof (struct UNIXMessage) + msgbuf_size);
-  ssize = sizeof (struct UNIXMessage) + msgbuf_size;
-
-  message->header.size = htons (ssize);
-  message->header.type = htons (0);
-  memcpy (&message->sender, plugin->env->my_identity,
-          sizeof (struct GNUNET_PeerIdentity));
-  memcpy (&message[1], msgbuf, msgbuf_size);
+  GNUNET_assert (NULL != plugin);
+  if (NULL == send_handle)
+  {
+    GNUNET_break (0); /* We do not have a send handle */
+    return GNUNET_SYSERR;
+  }
+  if ((NULL == addr) || (0 == addrlen))
+  {
+    GNUNET_break (0); /* Can never send if we don't have an address */
+    return GNUNET_SYSERR;
+  }
 
-  memset(&un, 0, sizeof(un));
+  /* Prepare address */
+  memset (&un, 0, sizeof (un));
   un.sun_family = AF_UNIX;
   slen = strlen (addr) + 1;
   if (slen >= sizeof (un.sun_path))
     slen = sizeof (un.sun_path) - 1;
-  sent = 0;
-  GNUNET_assert(slen < sizeof(un.sun_path));
+  GNUNET_assert (slen < sizeof (un.sun_path));
   memcpy (un.sun_path, addr, slen);
   un.sun_path[slen] = '\0';
   slen = sizeof (struct sockaddr_un);
@@ -563,170 +560,283 @@ unix_real_send (void *cls,
 #if HAVE_SOCKADDR_IN_SIN_LEN
   un.sun_len = (u_char) slen;
 #endif
-  sb = (struct sockaddr*) &un;
+  sb = (struct sockaddr *) &un;
   sbs = slen;
-  retry = GNUNET_NO;
 
-  sent = GNUNET_NETWORK_socket_sendto(send_handle, message, ssize, sb, sbs);
-
-  if ((GNUNET_SYSERR == sent) && (errno == EAGAIN))
-         retry = GNUNET_YES;
+resend:
+  /* Send the data */
+  sent = 0;
+  sent = GNUNET_NETWORK_socket_sendto (send_handle, msgbuf, msgbuf_size, sb, sbs);
 
-  if ((GNUNET_SYSERR == sent) && (errno == EMSGSIZE))
+  if (GNUNET_SYSERR == sent)
   {
-         socklen_t size = 0;
-         socklen_t len = sizeof (size);
-         GNUNET_NETWORK_socket_getsockopt ((struct GNUNET_NETWORK_Handle * ) send_handle,
-                         SOL_SOCKET,
-                         SO_SNDBUF,
-                         &size, &len);
-
-         if (size < ssize)
-         {
-#if DEBUG_UNIX
-          GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-             "Trying to increase socket buffer size from %i to %i for message size %i\n",
-             size,
-              ((ssize / 1000) + 2) * 1000, ssize);
-#endif
-                 size = ((ssize / 1000) + 2) * 1000;
-                 if (GNUNET_NETWORK_socket_setsockopt ((struct GNUNET_NETWORK_Handle * ) send_handle,
-                                 SOL_SOCKET,
-                                 SO_SNDBUF,
-                                 &size, sizeof(size)) == GNUNET_OK)
-                        retry = GNUNET_YES;
-                 else
-                        GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "setsockopt");
-         }
-  }
-
-  if (retry == GNUNET_YES)
+    if ( (EAGAIN == errno) ||
+        (ENOBUFS == errno) )
+      return RETRY; /* We have to retry later  */
+    if (EMSGSIZE == errno)
     {
-      if (incoming_retry_context == NULL)
+      socklen_t size = 0;
+      socklen_t len = sizeof (size);
+
+      GNUNET_NETWORK_socket_getsockopt ((struct GNUNET_NETWORK_Handle *)
+                                        send_handle, SOL_SOCKET, SO_SNDBUF, &size,
+                                        &len);
+      if (size < msgbuf_size)
+      {
+        LOG (GNUNET_ERROR_TYPE_DEBUG,
+                    "Trying to increase socket buffer size from %i to %i for message size %i\n",
+                    size, ((msgbuf_size / 1000) + 2) * 1000, msgbuf_size);
+        size = ((msgbuf_size / 1000) + 2) * 1000;
+        if (GNUNET_OK == GNUNET_NETWORK_socket_setsockopt
+            ((struct GNUNET_NETWORK_Handle *) send_handle, SOL_SOCKET, SO_SNDBUF,
+             &size, sizeof (size)))
+          goto resend; /* Increased buffer size, retry sending */
+        else
         {
-          retry_list_entry = GNUNET_malloc(sizeof(struct RetryList));
-          retry_ctx = GNUNET_malloc(sizeof(struct RetrySendContext));
-          retry_ctx->addr = GNUNET_malloc(addrlen);
-          retry_ctx->msg = GNUNET_malloc(msgbuf_size);
-          retry_ctx->plugin = plugin;
-          memcpy(retry_ctx->addr, addr, addrlen);
-          memcpy(retry_ctx->msg, msgbuf, msgbuf_size);
-          retry_ctx->msg_size = msgbuf_size;
-          retry_ctx->addrlen = addrlen;
-          retry_ctx->send_handle = send_handle;
-          retry_ctx->cont = cont;
-          retry_ctx->cont_cls = cont_cls;
-          retry_ctx->priority = priority;
-          retry_ctx->timeout = GNUNET_TIME_relative_to_absolute(timeout);
-          memcpy(&retry_ctx->target, target, sizeof(struct GNUNET_PeerIdentity));
-          retry_ctx->delay = GNUNET_TIME_UNIT_MILLISECONDS;
-          retry_ctx->retry_list_entry = retry_list_entry;
-          retry_list_entry->retry_ctx = retry_ctx;
-          GNUNET_CONTAINER_DLL_insert(retry_list_head, retry_list_tail, retry_list_entry);
+          /* Could not increase buffer size: error, no retry */
+          GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "setsockopt");
+          return GNUNET_SYSERR;
         }
+      }
       else
-        {
-          retry_ctx = incoming_retry_context;
-          retry_ctx->delay = GNUNET_TIME_relative_multiply(retry_ctx->delay, 2);
-        }
-      retry_ctx->retry_task = GNUNET_SCHEDULER_add_delayed(retry_ctx->delay, &retry_send_message, retry_ctx);
-
-      //GNUNET_log_strerror (GNUNET_ERROR_TYPE_DEBUG, "send");
-      GNUNET_free(message);
-      return ssize;
-    }
-#if DEBUG_UNIX
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-             "UNIX transmit %u-byte message to %s (%d: %s)\n",
-             (unsigned int) ssize,
-             GNUNET_a2s (sb, sbs),
-             (int) sent,
-             (sent < 0) ? STRERROR (errno) : "ok");
-#endif
-  if (cont != NULL)
-    {
-      if (sent == GNUNET_SYSERR)
-        cont (cont_cls, target, GNUNET_SYSERR);
-      else
-        {
-          cont (cont_cls, target, GNUNET_OK);
-        }
-    }
-
-  if (incoming_retry_context != NULL)
-    {
-      GNUNET_CONTAINER_DLL_remove(retry_list_head, retry_list_tail, incoming_retry_context->retry_list_entry);
-      GNUNET_free(incoming_retry_context->retry_list_entry);
-      GNUNET_free(incoming_retry_context->msg);
-      GNUNET_free(incoming_retry_context->addr);
-      GNUNET_free(incoming_retry_context);
+      {
+        /* Buffer is bigger than message:  error, no retry
+         * This should never happen!*/
+        GNUNET_break (0);
+        return GNUNET_SYSERR;
+      }
     }
+  }
 
-  GNUNET_free (message);
+  LOG (GNUNET_ERROR_TYPE_DEBUG,
+       "UNIX transmit %u-byte message to %s (%d: %s)\n",
+       (unsigned int) msgbuf_size, 
+       GNUNET_a2s (sb, sbs), 
+       (int) sent,
+       (sent < 0) ? STRERROR (errno) : "ok");
   return sent;
 }
 
 
+/**
+ * Closure for 'get_session_it'.
+ */
+struct GetSessionIteratorContext
+{ 
+  /**
+   * Location to store the session, if found.
+   */
+  struct Session *res;
+
+  /**
+   * Address information.
+   */
+  const char *address;
+
+  /**
+   * Number of bytes in 'address'
+   */
+  size_t addrlen;
+};
+
+
+/**
+ * Function called to find a session by address.
+ *
+ * @param cls the 'struct LookupCtx'
+ * @param key peer we are looking for (unused)
+ * @param value a session
+ * @return GNUNET_YES if not found (continue looking), GNUNET_NO on success
+ */
+static int
+get_session_it (void *cls, 
+               const struct GNUNET_HashCode *key, 
+               void *value)
+{
+  struct GetSessionIteratorContext *gsi = cls;
+  struct Session *s = value;
+
+  if ( (gsi->addrlen == s->addrlen) &&
+       (0 == memcmp (gsi->address, s->addr, s->addrlen)) )
+  {
+    gsi->res = s;
+    return GNUNET_NO;
+  }
+  return GNUNET_YES;
+}
+
+
+/**
+ * Session was idle for too long, so disconnect it
+ *
+ * @param cls the 'struct Session' to disconnect
+ * @param tc scheduler context
+ */
+static void
+session_timeout (void *cls, 
+                const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct Session *s = cls;
+  
+  s->timeout_task = GNUNET_SCHEDULER_NO_TASK;
+  LOG (GNUNET_ERROR_TYPE_DEBUG,
+       "Session %p was idle for %s, disconnecting\n",
+       s,
+       GNUNET_STRINGS_relative_time_to_string (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT,
+                                              GNUNET_YES));
+  disconnect_session (s);
+}
+
+
+/**
+ * Creates a new outbound session the transport service will use to send data to the
+ * peer
+ *
+ * @param cls the plugin
+ * @param address the address
+ * @return the session or NULL of max connections exceeded
+ */
+static struct Session *
+unix_plugin_get_session (void *cls,
+                        const struct GNUNET_HELLO_Address *address)
+{
+  struct Plugin *plugin = cls;
+  struct Session *s;
+  struct GetSessionIteratorContext gsi;
+
+  GNUNET_assert (NULL != plugin);
+  GNUNET_assert (NULL != address);
+
+  /* Check if already existing */
+  gsi.address = (const char *) address->address;
+  gsi.addrlen = address->address_length;
+  gsi.res = NULL;
+  GNUNET_CONTAINER_multihashmap_get_multiple (plugin->session_map, 
+                                             &address->peer.hashPubKey, 
+                                             &get_session_it, &gsi);
+  if (NULL != gsi.res)
+  {
+    LOG (GNUNET_ERROR_TYPE_DEBUG, 
+        "Found existing session\n");
+    return gsi.res;
+  }
+
+  /* create a new session */
+  s = GNUNET_malloc (sizeof (struct Session) + address->address_length);
+  s->addr = &s[1];
+  s->addrlen = address->address_length;
+  s->plugin = plugin;
+  memcpy (s->addr, address->address, s->addrlen);
+  memcpy (&s->target, &address->peer, sizeof (struct GNUNET_PeerIdentity));
+
+  GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == s->timeout_task);
+  s->timeout_task = GNUNET_SCHEDULER_add_delayed (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT,
+                                                 &session_timeout,
+                                                 s);
+  LOG (GNUNET_ERROR_TYPE_DEBUG,
+       "Creating a new session %p with timeout set to %s\n",
+       s,  
+       GNUNET_STRINGS_relative_time_to_string (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT,
+                                              GNUNET_YES));
+  (void) GNUNET_CONTAINER_multihashmap_put (plugin->session_map,
+                                           &address->peer.hashPubKey, s,
+                                           GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
+  GNUNET_STATISTICS_set (plugin->env->stats,
+                        "# UNIX sessions active",
+                        GNUNET_CONTAINER_multihashmap_size (plugin->session_map),
+                        GNUNET_NO);
+  return s;
+}
+
+
 /**
  * Function that can be used by the transport service to transmit
- * a message using the plugin.
+ * 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 target who should receive this message (ignored by UNIX)
- * @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 UNIX)
- * @param timeout when should we time out (give up) if we can not transmit?
- * @param session identifier used for this session (can be NULL)
- * @param addr the addr to send the message to, needs to be a sockaddr for us
- * @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 session 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...)
+ *        peer disconnected...); can be NULL
  * @param cont_cls closure for cont
- *
- * @return the number of bytes written (may return 0 and the message can
- *         still be transmitted later!)
+ * @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 ssize_t
 unix_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)
+                  struct Session *session,
+                  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;
-  ssize_t sent;
+  struct UNIXMessageWrapper *wrapper;
+  struct UNIXMessage *message;
+  int ssize;
+  
+  GNUNET_assert (NULL != plugin);
+  GNUNET_assert (NULL != session);
 
-  if (force_address == GNUNET_SYSERR)
+  if (GNUNET_OK != 
+      GNUNET_CONTAINER_multihashmap_contains_value (plugin->session_map,
+                                                   &session->target.hashPubKey,
+                                                   session))
+  {
+    LOG (GNUNET_ERROR_TYPE_ERROR, 
+        "Invalid session for peer `%s' `%s'\n",
+        GNUNET_i2s (&session->target),
+        (const char *) session->addr);
+    GNUNET_break (0);
     return GNUNET_SYSERR;
-  GNUNET_assert (NULL == session);
-
-#if DEBUG_UNIX
-  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Asked to send message to `%s'\n", (char *)addr);
-#endif
-  sent = unix_real_send(cls,
-                        NULL,
-                        plugin->unix_sock.desc,
-                        target,
-                        msgbuf, msgbuf_size,
-                        priority, timeout, addr, addrlen,
-                        cont, cont_cls);
-#if DEBUG_UNIX
-  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Sent %d bytes to `%s'\n", sent, (char *)addr);
-#endif
-  if (sent == GNUNET_SYSERR)
-    return 0;
-  return sent;
+  }
+  LOG (GNUNET_ERROR_TYPE_DEBUG, 
+       "Sending %u bytes with session for peer `%s' `%s'\n",
+       msgbuf_size,
+       GNUNET_i2s (&session->target),
+       (const char *) session->addr);
+  ssize = sizeof (struct UNIXMessage) + msgbuf_size;
+  message = GNUNET_malloc (sizeof (struct UNIXMessage) + msgbuf_size);
+  message->header.size = htons (ssize);
+  message->header.type = htons (0);
+  memcpy (&message->sender, plugin->env->my_identity,
+          sizeof (struct GNUNET_PeerIdentity));
+  memcpy (&message[1], msgbuf, msgbuf_size);
+  reschedule_session_timeout (session);
+  wrapper = GNUNET_malloc (sizeof (struct UNIXMessageWrapper));
+  wrapper->msg = message;
+  wrapper->msgsize = ssize;
+  wrapper->payload = msgbuf_size;
+  wrapper->priority = priority;
+  wrapper->timeout = GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get(), to);
+  wrapper->cont = cont;
+  wrapper->cont_cls = cont_cls;
+  wrapper->session = session;
+  GNUNET_CONTAINER_DLL_insert (plugin->msg_head, 
+                              plugin->msg_tail,
+                              wrapper);
+  plugin->bytes_in_queue += ssize;
+  GNUNET_STATISTICS_set (plugin->env->stats,
+                        "# bytes currently in UNIX buffers",
+                        plugin->bytes_in_queue, 
+                        GNUNET_NO);
+  if (GNUNET_NO == plugin->with_ws)
+    reschedule_select (plugin);
+  return ssize;
 }
 
 
@@ -740,45 +850,57 @@ unix_plugin_send (void *cls,
  * @param fromlen the length of the address
  */
 static void
-unix_demultiplexer(struct Plugin *plugin,
-                  struct GNUNET_PeerIdentity *sender,
-                   const struct GNUNET_MessageHeader *currhdr,
-                   const struct sockaddr_un *un,
-                   size_t fromlen)
+unix_demultiplexer (struct Plugin *plugin, struct GNUNET_PeerIdentity *sender,
+                    const struct GNUNET_MessageHeader *currhdr,
+                    const struct sockaddr_un *un, size_t fromlen)
 {
-  struct GNUNET_TRANSPORT_ATS_Information distance[2];
+  struct Session *s = NULL;
+  struct GNUNET_HELLO_Address * addr;
+
+  GNUNET_break (ntohl(plugin->ats_network.value) != GNUNET_ATS_NET_UNSPECIFIED);
+
+  GNUNET_assert (fromlen >= sizeof (struct sockaddr_un));
+
+  LOG (GNUNET_ERROR_TYPE_DEBUG, 
+       "Received message from %s\n",
+       un->sun_path);
+  GNUNET_STATISTICS_update (plugin->env->stats,
+                           "# bytes received via UNIX",
+                           ntohs (currhdr->size),
+                           GNUNET_NO);
+
+  addr = GNUNET_HELLO_address_allocate (sender,
+                                       "unix",
+                                       un->sun_path, 
+                                       strlen (un->sun_path) + 1);
+  s = lookup_session (plugin, sender, un);
+  if (NULL == s)
+    s = unix_plugin_get_session (plugin, addr);
+  reschedule_session_timeout (s);
 
-  distance[0].type = htonl (GNUNET_TRANSPORT_ATS_QUALITY_NET_DISTANCE);
-  distance[0].value = htonl (UNIX_DIRECT_DISTANCE);
-  distance[1].type = htonl (GNUNET_TRANSPORT_ATS_ARRAY_TERMINATOR);
-  distance[1].value = htonl (0);
+  plugin->env->receive (plugin->env->cls, sender, currhdr,
+                        s, un->sun_path, strlen (un->sun_path) + 1);
 
-  GNUNET_assert(fromlen >= sizeof(struct sockaddr_un));
+  plugin->env->update_address_metrics (plugin->env->cls,
+                                      sender,
+                                      un->sun_path,
+                                      strlen (un->sun_path) + 1,
+                                      s,
+                                      &plugin->ats_network, 1);
 
-#if DEBUG_UNIX
-  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received message from %s\n", un->sun_path);
-#endif
-  plugin->env->receive (plugin->env->cls, sender, currhdr,
-                       (const struct GNUNET_TRANSPORT_ATS_Information *) &distance, 2,
-                              NULL, un->sun_path, strlen(un->sun_path) + 1);
+  GNUNET_free (addr);
 }
 
 
-/*
- * @param cls the plugin handle
- * @param tc the scheduling context (for rescheduling this function again)
- *
- * 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.
+/**
+ * Read from UNIX domain socket (it is ready).
  *
+ * @param plugin the plugin
  */
 static void
-unix_plugin_select (void *cls,
-                   const struct GNUNET_SCHEDULER_TaskContext *tc)
+unix_plugin_select_read (struct Plugin *plugin)
 {
-  struct Plugin *plugin = cls;
-  char buf[65536];
+  char buf[65536] GNUNET_ALIGN;
   struct UNIXMessage *msg;
   struct GNUNET_PeerIdentity sender;
   struct sockaddr_un un;
@@ -790,83 +912,209 @@ unix_plugin_select (void *cls,
   const struct GNUNET_MessageHeader *currhdr;
   uint16_t csize;
 
-  plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
-  if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
-    return;
+  addrlen = sizeof (un);
+  memset (&un, 0, sizeof (un));
 
-  addrlen = sizeof(un);
-  memset(&un, 0, sizeof(un));
-  GNUNET_assert (GNUNET_NETWORK_fdset_isset (tc->read_ready, plugin->unix_sock.desc));
   ret =
-    GNUNET_NETWORK_socket_recvfrom (plugin->unix_sock.desc, buf, sizeof (buf),
-                                    (struct sockaddr *)&un, &addrlen);
+      GNUNET_NETWORK_socket_recvfrom (plugin->unix_sock.desc, buf, sizeof (buf),
+                                      (struct sockaddr *) &un, &addrlen);
+
+  if ((GNUNET_SYSERR == ret) && ((errno == EAGAIN) || (errno == ENOBUFS)))
+    return;
 
   if (ret == GNUNET_SYSERR)
-    {
-      GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "recvfrom");
-      plugin->select_task =
-        GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
-                                     GNUNET_SCHEDULER_NO_TASK,
-                                     GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
-                                     NULL, &unix_plugin_select, plugin);
-      return;
-    }
+  {
+    GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "recvfrom");
+    return;
+  }
   else
   {
 #if LINUX
     un.sun_path[0] = '/';
 #endif
-#if DEBUG_UNIX
-    GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Read %d bytes from socket %s\n", ret, &un.sun_path[0]);
-#endif
+    LOG (GNUNET_ERROR_TYPE_DEBUG, "Read %d bytes from socket %s\n", ret,
+                &un.sun_path[0]);
   }
 
   GNUNET_assert (AF_UNIX == (un.sun_family));
 
   msg = (struct UNIXMessage *) buf;
   csize = ntohs (msg->header.size);
-  if ( (csize < sizeof (struct UNIXMessage)) ||
-       (csize > ret) )
-    {
-      GNUNET_break_op (0);
-      plugin->select_task =
-       GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
-                                    GNUNET_SCHEDULER_NO_TASK,
-                                    GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
-                                    NULL, &unix_plugin_select, plugin);
-      return;
-    }
-  msgbuf = (char *)&msg[1];
+  if ((csize < sizeof (struct UNIXMessage)) || (csize > ret))
+  {
+    GNUNET_break_op (0);
+    return;
+  }
+  msgbuf = (char *) &msg[1];
   memcpy (&sender, &msg->sender, sizeof (struct GNUNET_PeerIdentity));
   offset = 0;
   tsize = csize - sizeof (struct UNIXMessage);
   while (offset + sizeof (struct GNUNET_MessageHeader) <= tsize)
+  {
+    currhdr = (struct GNUNET_MessageHeader *) &msgbuf[offset];
+    csize = ntohs (currhdr->size);
+    if ((csize < sizeof (struct GNUNET_MessageHeader)) ||
+        (csize > tsize - offset))
     {
-      currhdr = (struct GNUNET_MessageHeader *)&msgbuf[offset];
-      csize = ntohs (currhdr->size);
-      if ( (csize < sizeof (struct GNUNET_MessageHeader)) ||
-          (csize > tsize - offset) )
-       {
-         GNUNET_break_op (0);
-         break;
-       }
-      unix_demultiplexer(plugin, &sender, currhdr,
-                        &un, sizeof(un));
-      offset += csize;
+      GNUNET_break_op (0);
+      break;
     }
-  plugin->select_task =
-    GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
-                                 GNUNET_SCHEDULER_NO_TASK,
-                                 GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
-                                 NULL, &unix_plugin_select, plugin);
+    unix_demultiplexer (plugin, &sender, currhdr, &un, sizeof (un));
+    offset += csize;
+  }
 }
 
+
+/**
+ * Write to UNIX domain socket (it is ready).
+ *
+ * @param plugin the plugin
+ */
+static void
+unix_plugin_select_write (struct Plugin *plugin)
+{
+  int sent = 0;
+  struct UNIXMessageWrapper * msgw;
+
+  while (NULL != (msgw = plugin->msg_tail))
+  {
+    if (GNUNET_TIME_absolute_get_remaining (msgw->timeout).rel_value > 0)
+      break; /* Message is ready for sending */
+    /* Message has a timeout */
+    LOG (GNUNET_ERROR_TYPE_DEBUG,
+        "Timeout for message with %u bytes \n", 
+        (unsigned int) msgw->msgsize);
+    GNUNET_CONTAINER_DLL_remove (plugin->msg_head, plugin->msg_tail, msgw);
+    plugin->bytes_in_queue -= msgw->msgsize;
+    GNUNET_STATISTICS_set (plugin->env->stats, 
+                          "# bytes currently in UNIX buffers",
+                          plugin->bytes_in_queue, GNUNET_NO);    
+    GNUNET_STATISTICS_update (plugin->env->stats,
+                             "# UNIX bytes discarded",
+                             msgw->msgsize,
+                             GNUNET_NO);
+    if (NULL != msgw->cont)
+      msgw->cont (msgw->cont_cls,
+                 &msgw->session->target, 
+                 GNUNET_SYSERR, 
+                 msgw->payload, 
+                 0);    
+    GNUNET_free (msgw->msg);
+    GNUNET_free (msgw);  
+  }
+  if (NULL == msgw)
+    return; /* Nothing to send at the moment */
+
+  sent = unix_real_send (plugin,
+                         plugin->unix_sock.desc,
+                         &msgw->session->target,
+                         (const char *) msgw->msg,
+                         msgw->msgsize,
+                         msgw->priority,
+                         msgw->timeout,
+                         msgw->session->addr,
+                         msgw->session->addrlen,
+                         msgw->payload,
+                         msgw->cont, msgw->cont_cls);
+
+  if (RETRY == sent)
+  {
+    GNUNET_STATISTICS_update (plugin->env->stats,
+                             "# UNIX retry attempts",
+                             1, GNUNET_NO);
+    return;
+  }
+  if (GNUNET_SYSERR == sent)
+  {
+    /* failed and no retry */
+    if (NULL != msgw->cont)
+      msgw->cont (msgw->cont_cls, &msgw->session->target, GNUNET_SYSERR, msgw->payload, 0);
+
+    GNUNET_CONTAINER_DLL_remove(plugin->msg_head, plugin->msg_tail, msgw);
+
+    GNUNET_assert (plugin->bytes_in_queue >= msgw->msgsize);
+    plugin->bytes_in_queue -= msgw->msgsize;
+    GNUNET_STATISTICS_set (plugin->env->stats, 
+                          "# bytes currently in UNIX buffers",
+                          plugin->bytes_in_queue, GNUNET_NO);
+    GNUNET_STATISTICS_update (plugin->env->stats,
+                             "# UNIX bytes discarded",
+                             msgw->msgsize,
+                             GNUNET_NO);
+
+    GNUNET_free (msgw->msg);
+    GNUNET_free (msgw);
+    return;
+  }
+  /* successfully sent bytes */
+  GNUNET_break (sent > 0);
+  GNUNET_CONTAINER_DLL_remove (plugin->msg_head, 
+                              plugin->msg_tail, 
+                              msgw); 
+  GNUNET_assert (plugin->bytes_in_queue >= msgw->msgsize);
+  plugin->bytes_in_queue -= msgw->msgsize;
+  GNUNET_STATISTICS_set (plugin->env->stats,
+                        "# bytes currently in UNIX buffers",
+                        plugin->bytes_in_queue,
+                        GNUNET_NO);
+  GNUNET_STATISTICS_update (plugin->env->stats,
+                           "# bytes transmitted via UNIX",
+                           msgw->msgsize,
+                           GNUNET_NO);  
+  if (NULL != msgw->cont)
+    msgw->cont (msgw->cont_cls, &msgw->session->target, 
+               GNUNET_OK,
+               msgw->payload, 
+               msgw->msgsize);  
+  GNUNET_free (msgw->msg);
+  GNUNET_free (msgw);
+}
+
+
+/**
+ * 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.
+ *
+ * @param cls the plugin handle
+ * @param tc the scheduling context (for rescheduling this function again)
+ */
+static void
+unix_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 ((tc->reason & GNUNET_SCHEDULER_REASON_WRITE_READY) != 0)
+  {
+    /* Ready to send data */
+    GNUNET_assert (GNUNET_NETWORK_fdset_isset
+                   (tc->write_ready, plugin->unix_sock.desc));
+    if (NULL != plugin->msg_head)
+      unix_plugin_select_write (plugin);
+  }
+
+  if ((tc->reason & GNUNET_SCHEDULER_REASON_READ_READY) != 0)
+  {
+    /* Ready to receive data */
+    GNUNET_assert (GNUNET_NETWORK_fdset_isset
+                   (tc->read_ready, plugin->unix_sock.desc));
+    unix_plugin_select_read (plugin);
+  }
+  reschedule_select (plugin);
+}
+
+
 /**
  * Create a slew of UNIX sockets.  If possible, use IPv6 and IPv4.
  *
  * @param cls closure for server start, should be a struct Plugin *
  * @return number of sockets created or GNUNET_SYSERR on error
-*/
+ */
 static int
 unix_transport_server_start (void *cls)
 {
@@ -876,7 +1124,7 @@ unix_transport_server_start (void *cls)
   struct sockaddr_un un;
   size_t slen;
 
-  memset(&un, 0, sizeof(un));
+  memset (&un, 0, sizeof (un));
   un.sun_family = AF_UNIX;
   slen = strlen (plugin->unix_socket_path) + 1;
   if (slen >= sizeof (un.sun_path))
@@ -889,42 +1137,37 @@ unix_transport_server_start (void *cls)
   un.sun_len = (u_char) slen;
 #endif
 
-  serverAddr = (struct sockaddr*) &un;
+  serverAddr = (struct sockaddr *) &un;
   addrlen = slen;
 #if LINUX
   un.sun_path[0] = '\0';
 #endif
-
-  plugin->unix_sock.desc = GNUNET_NETWORK_socket_create (AF_UNIX, SOCK_DGRAM, 0);
+  plugin->ats_network = plugin->env->get_address_type (plugin->env->cls, serverAddr, addrlen);
+  plugin->unix_sock.desc =
+      GNUNET_NETWORK_socket_create (AF_UNIX, SOCK_DGRAM, 0);
   if (NULL == plugin->unix_sock.desc)
-    {
-      GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "socket");
-      return GNUNET_SYSERR;
-    }
-  if (GNUNET_NETWORK_socket_bind (plugin->unix_sock.desc, serverAddr, addrlen) !=
-      GNUNET_OK)
-    {
-      GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "bind");
-      GNUNET_NETWORK_socket_close (plugin->unix_sock.desc);
-      plugin->unix_sock.desc = NULL;
-      return GNUNET_SYSERR;
-    }
-#if DEBUG_UNIX
-  GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, 
-                  "unix",
-                  "Bound to `%s'\n",
-                  &un.sun_path[0]);
-#endif
+  {
+    GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "socket");
+    return GNUNET_SYSERR;
+  }
+  if (GNUNET_NETWORK_socket_bind (plugin->unix_sock.desc, serverAddr, addrlen, 0)
+      != GNUNET_OK)
+  {
+    GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "bind");
+    GNUNET_NETWORK_socket_close (plugin->unix_sock.desc);
+    plugin->unix_sock.desc = NULL;
+    return GNUNET_SYSERR;
+  }
+  LOG (GNUNET_ERROR_TYPE_DEBUG, "Bound to `%s'\n", plugin->unix_socket_path);
   plugin->rs = GNUNET_NETWORK_fdset_create ();
+  plugin->ws = GNUNET_NETWORK_fdset_create ();
   GNUNET_NETWORK_fdset_zero (plugin->rs);
-  GNUNET_NETWORK_fdset_set (plugin->rs,
-                            plugin->unix_sock.desc);
-
-  plugin->select_task =
-    GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
-                                 GNUNET_SCHEDULER_NO_TASK,
-                                 GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
-                                 NULL, &unix_plugin_select, plugin);
+  GNUNET_NETWORK_fdset_zero (plugin->ws);
+  GNUNET_NETWORK_fdset_set (plugin->rs, plugin->unix_sock.desc);
+  GNUNET_NETWORK_fdset_set (plugin->ws, plugin->unix_sock.desc);
+
+  reschedule_select (plugin);
+
   return 1;
 }
 
@@ -946,41 +1189,15 @@ unix_transport_server_start (void *cls)
  *
  */
 static int
-unix_check_address (void *cls,
-                  const void *addr,
-                  size_t addrlen)
+unix_check_address (void *cls, const void *addr, size_t addrlen)
 {
-
-#if DEBUG_UNIX
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                   "Informing transport service about my address `%s'\n",
-                   (char *)addr);
-#endif
+  LOG (GNUNET_ERROR_TYPE_DEBUG,
+              "Informing transport service about my address `%s'\n",
+              (char *) addr);
   return GNUNET_OK;
 }
 
 
-/**
- * Append our port and forward the result.
- */
-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;
-    }
-  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.
@@ -996,64 +1213,66 @@ append_port (void *cls, const char *hostname)
  * @param asc_cls closure for asc
  */
 static void
-unix_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)
+unix_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;
-      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;
-      a4.sin_port = u4->u_port;
-      a4.sin_addr.s_addr = u4->ipv4_addr;
-      port = ntohs (u4->u_port);
-      sb = &a4;
-      sbs = sizeof (a4);
-    }
+  if ((NULL != addr) && (addrlen > 0))
+  {
+    asc (asc_cls, (const char *) addr);
+  }
   else
-    {
-      /* invalid address */
-      GNUNET_break_op (0);
-      asc (asc_cls, NULL);
-      return;
-    }
-  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);
+  {
+    GNUNET_break (0);
+    asc (asc_cls, "<invalid UNIX address>");
+  }
+  asc (asc_cls, NULL);
 }
 
+
+/**
+ * Function called to convert a string address to
+ * a binary address.
+ *
+ * @param cls closure ('struct Plugin*')
+ * @param addr string address
+ * @param addrlen length of the address (strlen(addr) + '\0')
+ * @param buf location to store the buffer
+ *        If the function returns GNUNET_SYSERR, its contents are undefined.
+ * @param added length of created address
+ * @return GNUNET_OK on success, GNUNET_SYSERR on failure
+ */
+static int
+unix_string_to_address (void *cls, const char *addr, uint16_t addrlen,
+    void **buf, size_t *added)
+{
+  if ((NULL == addr) || (0 == addrlen))
+  {
+    GNUNET_break (0);
+    return GNUNET_SYSERR;
+  }
+
+  if ('\0' != addr[addrlen - 1])
+  {
+    GNUNET_break (0);
+    return GNUNET_SYSERR;
+  }
+
+  if (strlen (addr) != addrlen - 1)
+  {
+    GNUNET_break (0);
+    return GNUNET_SYSERR;
+  }
+
+  (*buf) = strdup (addr);
+  (*added) = strlen (addr) + 1;
+  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
@@ -1065,17 +1284,15 @@ unix_plugin_address_pretty_printer (void *cls,
  * @param addrlen length of the address
  * @return string representing the same address
  */
-static const char*
-unix_address_to_string (void *cls,
-                       const void *addr,
-                       size_t addrlen)
+static const char *
+unix_address_to_string (void *cls, const void *addr, size_t addrlen)
 {
   if ((addr != NULL) && (addrlen > 0))
     return (const char *) addr;
-  else
-    return NULL;
+  return NULL;
 }
 
+
 /**
  * Notify transport service about address
  *
@@ -1084,18 +1301,84 @@ unix_address_to_string (void *cls,
  */
 static void
 address_notification (void *cls,
-                    const struct GNUNET_SCHEDULER_TaskContext *tc)
+                     const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct Plugin *plugin = cls;
+
+  plugin->address_update_task = GNUNET_SCHEDULER_NO_TASK;
+  plugin->env->notify_address (plugin->env->cls, GNUNET_YES,
+                               plugin->unix_socket_path,
+                               strlen (plugin->unix_socket_path) + 1,
+                               "unix");
+}
+
+
+/**
+ * Increment session timeout due to activity
+ *
+ * @param s session for which the timeout should be moved
+ */
+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);
+  LOG (GNUNET_ERROR_TYPE_DEBUG,
+       "Timeout rescheduled for session %p set to %s\n",
+       s,
+       GNUNET_STRINGS_relative_time_to_string (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT,
+                                              GNUNET_YES));
+}
+
+
+/**
+ * Function called on sessions to disconnect
+ *
+ * @param cls the plugin (unused)
+ * @param key peer identity (unused)
+ * @param value the 'struct Session' to disconnect
+ * @return GNUNET_YES (always, continue to iterate)
+ */
+static int
+get_session_delete_it (void *cls, const struct GNUNET_HashCode * key, void *value)
+{
+  struct Session *s = value;
+
+  disconnect_session (s);
+  return GNUNET_YES;
+}
+
+
+/**
+ * Disconnect from a remote node.  Clean up session if we have one for this peer
+ *
+ * @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
+unix_disconnect (void *cls, 
+                const struct GNUNET_PeerIdentity *target)
 {
   struct Plugin *plugin = cls;
-  plugin->env->notify_address(plugin->env->cls,
-                              GNUNET_YES,
-                              plugin->unix_socket_path,
-                              strlen(plugin->unix_socket_path) + 1);
+
+  GNUNET_assert (plugin != NULL);
+  GNUNET_CONTAINER_multihashmap_get_multiple (plugin->session_map,
+                                             &target->hashPubKey, 
+                                             &get_session_delete_it, plugin);
 }
 
+
 /**
- * The exported method. Makes the core api available via a global and
- * returns the unix transport API.
+ * The exported method.  Initializes the plugin and returns a
+ * struct with the callbacks.
+ *
+ * @param cls the plugin's execution environment
+ * @return NULL on error, plugin functions otherwise
  */
 void *
 libgnunet_plugin_transport_unix_init (void *cls)
@@ -1106,45 +1389,104 @@ libgnunet_plugin_transport_unix_init (void *cls)
   struct Plugin *plugin;
   int sockets_created;
 
+  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 = &unix_plugin_address_pretty_printer;
+    api->address_to_string = &unix_address_to_string;
+    api->string_to_address = &unix_string_to_address;
+    return api;
+  }
   if (GNUNET_OK !=
-      GNUNET_CONFIGURATION_get_value_number (env->cfg,
-                                            "transport-unix",
-                                            "PORT",
-                                            &port))
+      GNUNET_CONFIGURATION_get_value_number (env->cfg, "transport-unix", "PORT",
+                                             &port))
     port = UNIX_NAT_DEFAULT_PORT;
   plugin = GNUNET_malloc (sizeof (struct Plugin));
   plugin->port = port;
   plugin->env = env;
   GNUNET_asprintf (&plugin->unix_socket_path, 
-                  "/tmp/unix-plugin-sock.%d", 
-                  plugin->port);
-  
+                  "/tmp/unix-plugin-sock.%d",
+                   plugin->port);
+
   api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions));
   api->cls = plugin;
 
+  api->get_session = &unix_plugin_get_session;
   api->send = &unix_plugin_send;
   api->disconnect = &unix_disconnect;
   api->address_pretty_printer = &unix_plugin_address_pretty_printer;
   api->address_to_string = &unix_address_to_string;
   api->check_address = &unix_check_address;
+  api->string_to_address = &unix_string_to_address;
   sockets_created = unix_transport_server_start (plugin);
-  if (sockets_created == 0)
-    GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
-               _("Failed to open UNIX sockets\n"));
-
-  GNUNET_SCHEDULER_add_now(address_notification, plugin);
+  if (0 == sockets_created)
+    LOG (GNUNET_ERROR_TYPE_WARNING,
+        _("Failed to open UNIX listen socket\n"));
+  plugin->session_map = GNUNET_CONTAINER_multihashmap_create (10, GNUNET_NO);
+  plugin->address_update_task = GNUNET_SCHEDULER_add_now (&address_notification, plugin);
   return api;
 }
 
+
+/**
+ * Shutdown the plugin.
+ *
+ * @param cls the plugin API returned from the initialization function
+ * @return NULL (always)
+ */
 void *
 libgnunet_plugin_transport_unix_done (void *cls)
 {
   struct GNUNET_TRANSPORT_PluginFunctions *api = cls;
   struct Plugin *plugin = api->cls;
+  struct UNIXMessageWrapper * msgw;
 
-  unix_transport_server_stop (plugin);
+  if (NULL == plugin)
+  {
+    GNUNET_free (api);
+    return NULL;
+  }
+  plugin->env->notify_address (plugin->env->cls, GNUNET_NO,
+                               plugin->unix_socket_path,
+                               strlen (plugin->unix_socket_path) + 1,
+                               "unix");
+  while (NULL != (msgw = plugin->msg_head))
+  {
+    GNUNET_CONTAINER_DLL_remove (plugin->msg_head, plugin->msg_tail, msgw);
+    if (msgw->cont != NULL)
+      msgw->cont (msgw->cont_cls,  &msgw->session->target, GNUNET_SYSERR,
+                  msgw->payload, 0);
+    GNUNET_free (msgw->msg);
+    GNUNET_free (msgw);
+  }
 
-  GNUNET_NETWORK_fdset_destroy (plugin->rs);
+  if (GNUNET_SCHEDULER_NO_TASK != plugin->select_task)
+  {
+    GNUNET_SCHEDULER_cancel (plugin->select_task);
+    plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
+  }
+  if (GNUNET_SCHEDULER_NO_TASK != plugin->address_update_task)
+  {
+    GNUNET_SCHEDULER_cancel (plugin->address_update_task);
+    plugin->address_update_task = GNUNET_SCHEDULER_NO_TASK;
+  }
+  if (NULL != plugin->unix_sock.desc)
+  {
+    GNUNET_break (GNUNET_OK ==
+                  GNUNET_NETWORK_socket_close (plugin->unix_sock.desc));
+    plugin->unix_sock.desc = NULL;
+    plugin->with_ws = GNUNET_NO;
+  }
+  GNUNET_CONTAINER_multihashmap_iterate (plugin->session_map,
+                                        &get_session_delete_it, plugin);
+  GNUNET_CONTAINER_multihashmap_destroy (plugin->session_map);
+  if (NULL != plugin->rs)
+    GNUNET_NETWORK_fdset_destroy (plugin->rs);
+  if (NULL != plugin->ws)
+    GNUNET_NETWORK_fdset_destroy (plugin->ws);
   GNUNET_free (plugin->unix_socket_path);
   GNUNET_free (plugin);
   GNUNET_free (api);