Whamcloud - gitweb
Land b_smallfix onto HEAD (20040414_1359)
[fs/lustre-release.git] / lnet / klnds / socklnd / socklnd_cb.c
index 656a0c5..21e0abe 100644 (file)
  */
 
 #include "socknal.h"
-
-int        ksocknal_io_timeout = SOCKNAL_IO_TIMEOUT;
-#if SOCKNAL_ZC
-int        ksocknal_do_zc = 1;
-int        ksocknal_zc_min_frag = SOCKNAL_ZC_MIN_FRAG;
+#if (LINUX_VERSION_CODE > KERNEL_VERSION(2,5,0))
+# include <linux/syscalls.h>
 #endif
 
 /*
  *  LIB functions follow
  *
  */
-int
+ptl_err_t
 ksocknal_read(nal_cb_t *nal, void *private, void *dst_addr,
               user_ptr src_addr, size_t len)
 {
@@ -43,10 +40,10 @@ ksocknal_read(nal_cb_t *nal, void *private, void *dst_addr,
                nal->ni.nid, (long)len, src_addr, dst_addr);
 
         memcpy( dst_addr, src_addr, len );
-        return 0;
+        return PTL_OK;
 }
 
-int
+ptl_err_t
 ksocknal_write(nal_cb_t *nal, void *private, user_ptr dst_addr,
                void *src_addr, size_t len)
 {
@@ -54,20 +51,7 @@ ksocknal_write(nal_cb_t *nal, void *private, user_ptr dst_addr,
                nal->ni.nid, (long)len, src_addr, dst_addr);
 
         memcpy( dst_addr, src_addr, len );
-        return 0;
-}
-
-int
-ksocknal_callback (nal_cb_t * nal, void *private, lib_eq_t *eq,
-                         ptl_event_t *ev)
-{
-        CDEBUG(D_NET, LPX64": callback eq %p ev %p\n",
-               nal->ni.nid, eq, ev);
-
-        if (eq->event_callback != NULL)
-                eq->event_callback(ev);
-
-        return 0;
+        return PTL_OK;
 }
 
 void *
@@ -109,6 +93,8 @@ ksocknal_cli(nal_cb_t *nal, unsigned long *flags)
 {
         ksock_nal_data_t *data = nal->nal_data;
 
+        /* OK to ignore 'flags'; we're only ever serialise threads and
+         * never need to lock out interrupts */
         spin_lock(&data->ksnd_nal_cb_lock);
 }
 
@@ -118,9 +104,23 @@ ksocknal_sti(nal_cb_t *nal, unsigned long *flags)
         ksock_nal_data_t *data;
         data = nal->nal_data;
 
+        /* OK to ignore 'flags'; we're only ever serialise threads and
+         * never need to lock out interrupts */
         spin_unlock(&data->ksnd_nal_cb_lock);
 }
 
+void
+ksocknal_callback(nal_cb_t *nal, void *private, lib_eq_t *eq, ptl_event_t *ev)
+{
+        /* holding ksnd_nal_cb_lock */
+
+        if (eq->event_callback != NULL)
+                eq->event_callback(ev);
+        
+        if (waitqueue_active(&ksocknal_data.ksnd_yield_waitq))
+                wake_up_all(&ksocknal_data.ksnd_yield_waitq);
+}
+
 int
 ksocknal_dist(nal_cb_t *nal, ptl_nid_t nid, unsigned long *dist)
 {
@@ -135,64 +135,14 @@ ksocknal_dist(nal_cb_t *nal, ptl_nid_t nid, unsigned long *dist)
         return 0;
 }
 
-ksock_ltx_t *
-ksocknal_get_ltx (int may_block)
-{
-        unsigned long flags;
-        ksock_ltx_t *ltx = NULL;
-
-        for (;;) {
-                spin_lock_irqsave (&ksocknal_data.ksnd_idle_ltx_lock, flags);
-
-                if (!list_empty (&ksocknal_data.ksnd_idle_ltx_list)) {
-                        ltx = list_entry(ksocknal_data.ksnd_idle_ltx_list.next,
-                                         ksock_ltx_t, ltx_tx.tx_list);
-                        list_del (&ltx->ltx_tx.tx_list);
-                        ksocknal_data.ksnd_active_ltxs++;
-                        break;
-                }
-
-                if (!may_block) {
-                        if (!list_empty(&ksocknal_data.ksnd_idle_nblk_ltx_list)) {
-                                ltx = list_entry(ksocknal_data.ksnd_idle_nblk_ltx_list.next,
-                                                 ksock_ltx_t, ltx_tx.tx_list);
-                                list_del (&ltx->ltx_tx.tx_list);
-                                ksocknal_data.ksnd_active_ltxs++;
-                        }
-                        break;
-                }
-
-                spin_unlock_irqrestore(&ksocknal_data.ksnd_idle_ltx_lock,
-                                       flags);
-
-                wait_event (ksocknal_data.ksnd_idle_ltx_waitq,
-                            !list_empty (&ksocknal_data.ksnd_idle_ltx_list));
-        }
-
-        spin_unlock_irqrestore (&ksocknal_data.ksnd_idle_ltx_lock, flags);
-
-        return (ltx);
-}
-
 void
-ksocknal_put_ltx (ksock_ltx_t *ltx)
+ksocknal_free_ltx (ksock_ltx_t *ltx)
 {
-        unsigned long   flags;
-        
-        spin_lock_irqsave (&ksocknal_data.ksnd_idle_ltx_lock, flags);
-
-        ksocknal_data.ksnd_active_ltxs--;
-        list_add_tail (&ltx->ltx_tx.tx_list, ltx->ltx_idle);
-
-        /* normal tx desc => wakeup anyone blocking for one */
-        if (ltx->ltx_idle == &ksocknal_data.ksnd_idle_ltx_list &&
-            waitqueue_active (&ksocknal_data.ksnd_idle_ltx_waitq))
-                wake_up (&ksocknal_data.ksnd_idle_ltx_waitq);
-
-        spin_unlock_irqrestore (&ksocknal_data.ksnd_idle_ltx_lock, flags);
+        atomic_dec(&ksocknal_data.ksnd_nactive_ltxs);
+        PORTAL_FREE(ltx, ltx->ltx_desc_size);
 }
 
-#if SOCKNAL_ZC
+#if (SOCKNAL_ZC && SOCKNAL_VADDR_ZC)
 struct page *
 ksocknal_kvaddr_to_page (unsigned long vaddr)
 {
@@ -225,10 +175,10 @@ ksocknal_send_iov (ksock_conn_t *conn, ksock_tx_t *tx)
         struct iovec  *iov = tx->tx_iov;
         int            fragsize = iov->iov_len;
         unsigned long  vaddr = (unsigned long)iov->iov_base;
-        int            more = !list_empty (&conn->ksnc_tx_queue) |
-                              (tx->tx_niov > 1) |
-                              (tx->tx_nkiov > 1);
-#if SOCKNAL_ZC
+        int            more = (tx->tx_niov > 1) || 
+                              (tx->tx_nkiov > 0) ||
+                              (!list_empty (&conn->ksnc_tx_queue));
+#if (SOCKNAL_ZC && SOCKNAL_VADDR_ZC)
         int            offset = vaddr & (PAGE_SIZE - 1);
         int            zcsize = MIN (fragsize, PAGE_SIZE - offset);
         struct page   *page;
@@ -240,11 +190,10 @@ ksocknal_send_iov (ksock_conn_t *conn, ksock_tx_t *tx)
         LASSERT (fragsize <= tx->tx_resid);
         LASSERT (tx->tx_niov > 0);
         
-#if SOCKNAL_ZC
-        if (ksocknal_do_zc &&
+#if (SOCKNAL_ZC && SOCKNAL_VADDR_ZC)
+        if (zcsize >= ksocknal_data.ksnd_zc_min_frag &&
             (sock->sk->route_caps & NETIF_F_SG) &&
             (sock->sk->route_caps & (NETIF_F_IP_CSUM | NETIF_F_NO_CSUM | NETIF_F_HW_CSUM)) &&
-            zcsize >= ksocknal_zc_min_frag &&
             (page = ksocknal_kvaddr_to_page (vaddr)) != NULL) {
                 
                 CDEBUG(D_NET, "vaddr %p, page %p->%p + offset %x for %d\n",
@@ -274,28 +223,26 @@ ksocknal_send_iov (ksock_conn_t *conn, ksock_tx_t *tx)
                         .msg_flags      = more ? (MSG_DONTWAIT | MSG_MORE) : MSG_DONTWAIT
                 };
                 mm_segment_t oldmm = get_fs();
-                
+
                 set_fs (KERNEL_DS);
                 rc = sock_sendmsg(sock, &msg, fragsize);
                 set_fs (oldmm);
         } 
 
-        if (rc <= 0)
-                return (rc);
+        if (rc > 0) {
+                tx->tx_resid -= rc;
 
-        tx->tx_resid -= rc;
-
-        if (rc < iov->iov_len) {
-                /* didn't send whole iov entry... */
-                iov->iov_base = (void *)(vaddr + rc);
-                iov->iov_len -= rc;
-                /* ...but did we send everything we tried to send? */
-                return ((rc == fragsize) ? 1 : -EAGAIN);
+                if (rc < iov->iov_len) {
+                        /* didn't send whole iov entry... */
+                        iov->iov_base = (void *)(vaddr + rc);
+                        iov->iov_len -= rc;
+                } else {
+                        tx->tx_iov++;
+                        tx->tx_niov--;
+                }
         }
-
-        tx->tx_iov++;
-        tx->tx_niov--;
-        return (1);
+        
+        return (rc);
 }
 
 int
@@ -306,8 +253,8 @@ ksocknal_send_kiov (ksock_conn_t *conn, ksock_tx_t *tx)
         int            fragsize = kiov->kiov_len;
         struct page   *page = kiov->kiov_page;
         int            offset = kiov->kiov_offset;
-        int            more = !list_empty (&conn->ksnc_tx_queue) |
-                              (tx->tx_nkiov > 1);
+        int            more = (tx->tx_nkiov > 1) ||
+                              (!list_empty (&conn->ksnc_tx_queue));
         int            rc;
 
         /* NB we can't trust socket ops to either consume our iovs
@@ -318,10 +265,9 @@ ksocknal_send_kiov (ksock_conn_t *conn, ksock_tx_t *tx)
         LASSERT (tx->tx_nkiov > 0);
 
 #if SOCKNAL_ZC
-        if (ksocknal_do_zc &&
+        if (fragsize >= ksocknal_tunables.ksnd_zc_min_frag &&
             (sock->sk->route_caps & NETIF_F_SG) &&
-            (sock->sk->route_caps & (NETIF_F_IP_CSUM | NETIF_F_NO_CSUM | NETIF_F_HW_CSUM)) &&
-            fragsize >= ksocknal_zc_min_frag) {
+            (sock->sk->route_caps & (NETIF_F_IP_CSUM | NETIF_F_NO_CSUM | NETIF_F_HW_CSUM))) {
 
                 CDEBUG(D_NET, "page %p + offset %x for %d\n",
                                page, offset, fragsize);
@@ -349,75 +295,120 @@ ksocknal_send_kiov (ksock_conn_t *conn, ksock_tx_t *tx)
                 set_fs (KERNEL_DS);
                 rc = sock_sendmsg(sock, &msg, fragsize);
                 set_fs (oldmm);
+
                 kunmap (page);
         }
 
-        if (rc <= 0)
-                return (rc);
-
-        tx->tx_resid -= rc;
+        if (rc > 0) {
+                tx->tx_resid -= rc;
  
-        if (rc < fragsize) {
-                /* didn't send whole frag */
-                kiov->kiov_offset = offset + rc;
-                kiov->kiov_len    = fragsize - rc;
-                return (-EAGAIN);
+                if (rc < fragsize) {
+                        kiov->kiov_offset = offset + rc;
+                        kiov->kiov_len    = fragsize - rc;
+                } else {
+                        tx->tx_kiov++;
+                        tx->tx_nkiov--;
+                }
         }
 
-        /* everything went */
-        LASSERT (rc == fragsize);
-        tx->tx_kiov++;
-        tx->tx_nkiov--;
-        return (1);
+        return (rc);
 }
 
 int
-ksocknal_sendmsg (ksock_conn_t *conn, ksock_tx_t *tx)
+ksocknal_transmit (ksock_conn_t *conn, ksock_tx_t *tx)
 {
-        /* Return 0 on success, < 0 on error.
-         * caller checks tx_resid to determine progress/completion */
         int      rc;
-        ENTRY;
         
         if (ksocknal_data.ksnd_stall_tx != 0) {
                 set_current_state (TASK_UNINTERRUPTIBLE);
                 schedule_timeout (ksocknal_data.ksnd_stall_tx * HZ);
         }
 
-        rc = ksocknal_getconnsock (conn);
-        if (rc != 0)
-                return (rc);
-
-        for (;;) {
-                LASSERT (tx->tx_resid != 0);
+        LASSERT (tx->tx_resid != 0);
 
-                if (conn->ksnc_closing) {
-                        rc = -ESHUTDOWN;
-                        break;
-                }
+        rc = ksocknal_getconnsock (conn);
+        if (rc != 0) {
+                LASSERT (conn->ksnc_closing);
+                return (-ESHUTDOWN);
+        }
 
-                if (tx->tx_niov != 0)
+        do {
+                if (ksocknal_data.ksnd_enomem_tx > 0) {
+                        /* testing... */
+                        ksocknal_data.ksnd_enomem_tx--;
+                        rc = -EAGAIN;
+                } else if (tx->tx_niov != 0) {
                         rc = ksocknal_send_iov (conn, tx);
-                else
+                } else {
                         rc = ksocknal_send_kiov (conn, tx);
+                }
 
-                if (rc <= 0) {                  /* error or socket full? */
-                        /* NB: rc == 0 and rc == -EAGAIN both mean try
+                if (rc <= 0) {
+                        /* Didn't write anything.
+                         *
+                         * NB: rc == 0 and rc == -EAGAIN both mean try
                          * again later (linux stack returns -EAGAIN for
-                         * this, but Adaptech TOE returns 0) */
-                        if (rc == -EAGAIN)
-                                rc = 0;
+                         * this, but Adaptech TOE returns 0).
+                         *
+                         * Also, sends never fail with -ENOMEM, just
+                         * -EAGAIN, but with the added bonus that we can't
+                         * expect write_space() to call us back to tell us
+                         * when to try sending again.  We use the
+                         * SOCK_NOSPACE flag to diagnose...  */
+
+                        LASSERT(rc != -ENOMEM);
+
+                        if (rc == 0 || rc == -EAGAIN) {
+                                if (test_bit(SOCK_NOSPACE, 
+                                             &conn->ksnc_sock->flags)) {
+                                        rc = -EAGAIN;
+                                } else {
+                                        static int counter;
+                         
+                                        counter++;
+                                        if ((counter & (-counter)) == counter)
+                                                CWARN("%d ENOMEM tx %p\n", 
+                                                      counter, conn);
+                                        rc = -ENOMEM;
+                                }
+                        }
                         break;
                 }
 
-                if (tx->tx_resid == 0) {        /* sent everything */
-                        rc = 0;
-                        break;
-                }
-        }
+                rc = 0;
+
+                /* Consider the connection alive since we managed to chuck
+                 * more data into it.  Really, we'd like to consider it
+                 * alive only when the peer ACKs something, but
+                 * write_space() only gets called back while SOCK_NOSPACE
+                 * is set.  Instead, we presume peer death has occurred if
+                 * the socket doesn't drain within a timout */
+                conn->ksnc_tx_deadline = jiffies + 
+                                         ksocknal_tunables.ksnd_io_timeout * HZ;
+                conn->ksnc_peer->ksnp_last_alive = jiffies;
+
+        } while (tx->tx_resid != 0);
 
         ksocknal_putconnsock (conn);
-        RETURN (rc);
+        return (rc);
+}
+
+void
+ksocknal_eager_ack (ksock_conn_t *conn)
+{
+        int            opt = 1;
+        mm_segment_t   oldmm = get_fs();
+        struct socket *sock = conn->ksnc_sock;
+        
+        /* Remind the socket to ACK eagerly.  If I don't, the socket might
+         * think I'm about to send something it could piggy-back the ACK
+         * on, introducing delay in completing zero-copy sends in my
+         * peer. */
+
+        set_fs(KERNEL_DS);
+        sock->ops->setsockopt (sock, SOL_TCP, TCP_QUICKACK,
+                               (char *)&opt, sizeof (opt));
+        set_fs(oldmm);
 }
 
 int
@@ -444,7 +435,7 @@ ksocknal_recv_iov (ksock_conn_t *conn)
          * or leave them alone, so we only receive 1 frag at a time. */
         LASSERT (conn->ksnc_rx_niov > 0);
         LASSERT (fragsize <= conn->ksnc_rx_nob_wanted);
-        
+
         set_fs (KERNEL_DS);
         rc = sock_recvmsg (conn->ksnc_sock, &msg, fragsize, MSG_DONTWAIT);
         /* NB this is just a boolean............................^ */
@@ -453,6 +444,13 @@ ksocknal_recv_iov (ksock_conn_t *conn)
         if (rc <= 0)
                 return (rc);
 
+        /* received something... */
+        conn->ksnc_peer->ksnp_last_alive = jiffies;
+        conn->ksnc_rx_deadline = jiffies + 
+                                 ksocknal_tunables.ksnd_io_timeout * HZ;
+        mb();                           /* order with setting rx_started */
+        conn->ksnc_rx_started = 1;
+
         conn->ksnc_rx_nob_wanted -= rc;
         conn->ksnc_rx_nob_left -= rc;
                 
@@ -494,16 +492,24 @@ ksocknal_recv_kiov (ksock_conn_t *conn)
         LASSERT (fragsize <= conn->ksnc_rx_nob_wanted);
         LASSERT (conn->ksnc_rx_nkiov > 0);
         LASSERT (offset + fragsize <= PAGE_SIZE);
-        
+
         set_fs (KERNEL_DS);
         rc = sock_recvmsg (conn->ksnc_sock, &msg, fragsize, MSG_DONTWAIT);
         /* NB this is just a boolean............................^ */
         set_fs (oldmm);
+
         kunmap (page);
         
         if (rc <= 0)
                 return (rc);
         
+        /* received something... */
+        conn->ksnc_peer->ksnp_last_alive = jiffies;
+        conn->ksnc_rx_deadline = jiffies + 
+                                 ksocknal_tunables.ksnd_io_timeout * HZ;
+        mb();                           /* order with setting rx_started */
+        conn->ksnc_rx_started = 1;
+
         conn->ksnc_rx_nob_wanted -= rc;
         conn->ksnc_rx_nob_left -= rc;
                 
@@ -519,7 +525,7 @@ ksocknal_recv_kiov (ksock_conn_t *conn)
 }
 
 int
-ksocknal_recvmsg (ksock_conn_t *conn) 
+ksocknal_receive (ksock_conn_t *conn) 
 {
         /* Return 1 on success, 0 on EOF, < 0 on error.
          * Caller checks ksnc_rx_nob_wanted to determine
@@ -533,28 +539,38 @@ ksocknal_recvmsg (ksock_conn_t *conn)
         }
 
         rc = ksocknal_getconnsock (conn);
-        if (rc != 0)
-                return (rc);
+        if (rc != 0) {
+                LASSERT (conn->ksnc_closing);
+                return (-ESHUTDOWN);
+        }
 
         for (;;) {
-                if (conn->ksnc_closing) {
-                        rc = -ESHUTDOWN;
-                        break;
-                }
-                
                 if (conn->ksnc_rx_niov != 0)
                         rc = ksocknal_recv_iov (conn);
                 else
                         rc = ksocknal_recv_kiov (conn);
-                
+
                 if (rc <= 0) {
                         /* error/EOF or partial receive */
-                        if (rc == -EAGAIN)
+                        if (rc == -EAGAIN) {
                                 rc = 1;
+                        } else if (rc == 0 && conn->ksnc_rx_started) {
+                                /* EOF in the middle of a message */
+                                rc = -EPROTO;
+                        }
                         break;
                 }
 
+                /* Completed a fragment */
+
                 if (conn->ksnc_rx_nob_wanted == 0) {
+                        /* Completed a message segment (header or payload) */
+                        if ((ksocknal_tunables.ksnd_eager_ack & conn->ksnc_type) != 0 &&
+                            (conn->ksnc_rx_state ==  SOCKNAL_RX_BODY ||
+                             conn->ksnc_rx_state == SOCKNAL_RX_BODY_FWD)) {
+                                /* Remind the socket to ack eagerly... */
+                                ksocknal_eager_ack(conn);
+                        }
                         rc = 1;
                         break;
                 }
@@ -577,10 +593,8 @@ ksocknal_zc_callback (zccd_t *zcd)
 
         spin_lock_irqsave (&sched->kss_lock, flags);
 
-        list_del (&tx->tx_list); /* remove from kss_zctxpending_list */
         list_add_tail (&tx->tx_list, &sched->kss_zctxdone_list);
-        if (waitqueue_active (&sched->kss_waitq))
-                wake_up (&sched->kss_waitq);
+        wake_up (&sched->kss_waitq);
 
         spin_unlock_irqrestore (&sched->kss_lock, flags);
         EXIT;
@@ -609,7 +623,8 @@ ksocknal_tx_done (ksock_tx_t *tx, int asynch)
 
         if (tx->tx_isfwd) {             /* was a forwarded packet? */
                 kpr_fwd_done (&ksocknal_data.ksnd_router,
-                              KSOCK_TX_2_KPR_FWD_DESC (tx), 0);
+                              KSOCK_TX_2_KPR_FWD_DESC (tx), 
+                              (tx->tx_resid == 0) ? 0 : -ECONNABORTED);
                 EXIT;
                 return;
         }
@@ -617,9 +632,10 @@ ksocknal_tx_done (ksock_tx_t *tx, int asynch)
         /* local send */
         ltx = KSOCK_TX_2_KSOCK_LTX (tx);
 
-        lib_finalize (&ksocknal_lib, ltx->ltx_private, ltx->ltx_cookie);
+        lib_finalize (&ksocknal_lib, ltx->ltx_private, ltx->ltx_cookie,
+                      (tx->tx_resid == 0) ? PTL_OK : PTL_FAIL);
 
-        ksocknal_put_ltx (ltx);
+        ksocknal_free_ltx (ltx);
         EXIT;
 }
 
@@ -628,23 +644,12 @@ ksocknal_tx_launched (ksock_tx_t *tx)
 {
 #if SOCKNAL_ZC
         if (atomic_read (&tx->tx_zccd.zccd_count) != 1) {
-                unsigned long  flags;
                 ksock_conn_t  *conn = tx->tx_conn;
-                ksock_sched_t *sched = conn->ksnc_scheduler;
                 
                 /* zccd skbufs are still in-flight.  First take a ref on
                  * conn, so it hangs about for ksocknal_tx_done... */
                 atomic_inc (&conn->ksnc_refcount);
 
-                /* Stash it for timeout...
-                 * NB We have to hold a lock to stash the tx, and we have
-                 * stash it before we zcc_put(), but we have to _not_ hold
-                 * this lock when we zcc_put(), otherwise we could deadlock
-                 * if it turns out to be the last put. Aaaaarrrrggghhh! */
-                spin_lock_irqsave (&sched->kss_lock, flags);
-                list_add_tail (&tx->tx_list, &conn->ksnc_tx_pending);
-                spin_unlock_irqrestore (&sched->kss_lock, flags);
-                
                 /* ...then drop the initial ref on zccd, so the zero copy
                  * callback can occur */
                 zccd_put (&tx->tx_zccd);
@@ -656,71 +661,57 @@ ksocknal_tx_launched (ksock_tx_t *tx)
         ksocknal_tx_done (tx, 0);
 }
 
-void
-ksocknal_process_transmit (ksock_sched_t *sched, unsigned long *irq_flags)
+int
+ksocknal_process_transmit (ksock_conn_t *conn, ksock_tx_t *tx)
 {
-        ksock_conn_t *conn;
-        ksock_tx_t *tx;
-        int         rc;
-
-        LASSERT (!list_empty (&sched->kss_tx_conns));
-        conn = list_entry(sched->kss_tx_conns.next, ksock_conn_t, ksnc_tx_list);
-        list_del (&conn->ksnc_tx_list);
-
-        LASSERT (conn->ksnc_tx_scheduled);
-        LASSERT (conn->ksnc_tx_ready);
-        LASSERT (!list_empty (&conn->ksnc_tx_queue));
-        tx = list_entry (conn->ksnc_tx_queue.next, ksock_tx_t, tx_list);
-        /* assume transmit will complete now, so dequeue while I've got lock */
-        list_del (&tx->tx_list);
-
-        spin_unlock_irqrestore (&sched->kss_lock, *irq_flags);
-
-        LASSERT (tx->tx_resid > 0);
-
-        conn->ksnc_tx_ready = 0;/* write_space may race with me and set ready */
-        mb();                   /* => clear BEFORE trying to write */
-
-        rc = ksocknal_sendmsg (conn, tx);
+        unsigned long  flags;
+        int            rc;
+       
+        rc = ksocknal_transmit (conn, tx);
 
         CDEBUG (D_NET, "send(%d) %d\n", tx->tx_resid, rc);
 
-        if (rc != 0) {
-                if (ksocknal_close_conn_unlocked (conn)) {
-                        /* I'm the first to close */
-                        CERROR ("[%p] Error %d on write to "LPX64" ip %08x:%d\n",
-                                conn, rc, conn->ksnc_peer->ksnp_nid,
-                                conn->ksnc_ipaddr, conn->ksnc_port);
-                }
+        if (tx->tx_resid == 0) {
+                /* Sent everything OK */
+                LASSERT (rc == 0);
+
                 ksocknal_tx_launched (tx);
-                spin_lock_irqsave (&sched->kss_lock, *irq_flags);
+                return (0);
+        }
 
-        } else if (tx->tx_resid == 0) {  
+        if (rc == -EAGAIN)
+                return (rc);
 
-                /* everything went; assume more can go, and avoid
-                 * write_space locking */
-                conn->ksnc_tx_ready = 1;
+        if (rc == -ENOMEM) {
+                /* Queue on ksnd_enomem_conns for retry after a timeout */
+                spin_lock_irqsave(&ksocknal_data.ksnd_reaper_lock, flags);
+
+                /* enomem list takes over scheduler's ref... */
+                LASSERT (conn->ksnc_tx_scheduled);
+                list_add_tail(&conn->ksnc_tx_list,
+                              &ksocknal_data.ksnd_enomem_conns);
+                if (!time_after_eq(jiffies + SOCKNAL_ENOMEM_RETRY,
+                                   ksocknal_data.ksnd_reaper_waketime))
+                        wake_up (&ksocknal_data.ksnd_reaper_waitq);
+                
+                spin_unlock_irqrestore(&ksocknal_data.ksnd_reaper_lock, flags);
+                return (rc);
+        }
 
-                ksocknal_tx_launched (tx);
-                spin_lock_irqsave (&sched->kss_lock, *irq_flags);
-        } else {
-                spin_lock_irqsave (&sched->kss_lock, *irq_flags);
+        /* Actual error */
+        LASSERT (rc < 0);
 
-                /* back onto HEAD of tx_queue */
-                list_add (&tx->tx_list, &conn->ksnc_tx_queue);
-        }
+        if (!conn->ksnc_closing)
+                CERROR("[%p] Error %d on write to "LPX64
+                       " ip %d.%d.%d.%d:%d\n", conn, rc,
+                       conn->ksnc_peer->ksnp_nid,
+                       HIPQUAD(conn->ksnc_ipaddr),
+                       conn->ksnc_port);
 
-        /* no space to write, or nothing to write? */
-        if (!conn->ksnc_tx_ready ||
-            list_empty (&conn->ksnc_tx_queue)) {
-                /* mark not scheduled */
-                conn->ksnc_tx_scheduled = 0;
-                /* drop scheduler's ref */
-                ksocknal_put_conn (conn);
-        } else {
-                /* stay scheduled */
-                list_add_tail (&conn->ksnc_tx_list, &sched->kss_tx_conns);
-        }
+        ksocknal_close_conn_and_siblings (conn, rc);
+        ksocknal_tx_launched (tx);
+
+        return (rc);
 }
 
 void
@@ -730,19 +721,24 @@ ksocknal_launch_autoconnect_locked (ksock_route_t *route)
 
         /* called holding write lock on ksnd_global_lock */
 
-        LASSERT (route->ksnr_conn == NULL);
-        LASSERT (!route->ksnr_deleted && !route->ksnr_connecting);
+        LASSERT (!route->ksnr_deleted);
+        LASSERT ((route->ksnr_connected & (1 << SOCKNAL_CONN_ANY)) == 0);
+        LASSERT ((route->ksnr_connected & KSNR_TYPED_ROUTES) != KSNR_TYPED_ROUTES);
+        LASSERT (!route->ksnr_connecting);
         
-        route->ksnr_connecting = 1;
+        if (ksocknal_tunables.ksnd_typed_conns)
+                route->ksnr_connecting = 
+                        KSNR_TYPED_ROUTES & ~route->ksnr_connected;
+        else
+                route->ksnr_connecting = (1 << SOCKNAL_CONN_ANY);
+
         atomic_inc (&route->ksnr_refcount);     /* extra ref for asynchd */
         
         spin_lock_irqsave (&ksocknal_data.ksnd_autoconnectd_lock, flags);
         
         list_add_tail (&route->ksnr_connect_list,
                        &ksocknal_data.ksnd_autoconnectd_routes);
-
-        if (waitqueue_active (&ksocknal_data.ksnd_autoconnectd_waitq))
-                wake_up (&ksocknal_data.ksnd_autoconnectd_waitq);
+        wake_up (&ksocknal_data.ksnd_autoconnectd_waitq);
         
         spin_unlock_irqrestore (&ksocknal_data.ksnd_autoconnectd_lock, flags);
 }
@@ -750,22 +746,26 @@ ksocknal_launch_autoconnect_locked (ksock_route_t *route)
 ksock_peer_t *
 ksocknal_find_target_peer_locked (ksock_tx_t *tx, ptl_nid_t nid)
 {
+        char          ipbuf[PTL_NALFMT_SIZE];
         ptl_nid_t     target_nid;
         int           rc;
         ksock_peer_t *peer = ksocknal_find_peer_locked (nid);
-        
+
         if (peer != NULL)
                 return (peer);
-        
+
         if (tx->tx_isfwd) {
                 CERROR ("Can't send packet to "LPX64
-                        ": routed target is not a peer\n", nid);
+                       " %s: routed target is not a peer\n",
+                        nid, portals_nid2str(SOCKNAL, nid, ipbuf));
                 return (NULL);
         }
-        
-        rc = kpr_lookup (&ksocknal_data.ksnd_router, nid, &target_nid);
+
+        rc = kpr_lookup (&ksocknal_data.ksnd_router, nid, tx->tx_nob,
+                         &target_nid);
         if (rc != 0) {
-                CERROR ("Can't route to "LPX64": router error %d\n", nid, rc);
+                CERROR ("Can't route to "LPX64" %s: router error %d\n",
+                        nid, portals_nid2str(SOCKNAL, nid, ipbuf), rc);
                 return (NULL);
         }
 
@@ -773,29 +773,61 @@ ksocknal_find_target_peer_locked (ksock_tx_t *tx, ptl_nid_t nid)
         if (peer != NULL)
                 return (peer);
 
-        CERROR ("Can't send packet to "LPX64": no peer entry\n", target_nid);
+        CERROR ("Can't send packet to "LPX64" %s: no peer entry\n",
+                target_nid, portals_nid2str(SOCKNAL, target_nid, ipbuf));
         return (NULL);
 }
 
 ksock_conn_t *
-ksocknal_find_conn_locked (ksock_tx_t *tx, ksock_peer_t *peer) 
+ksocknal_find_conn_locked (ksock_tx_t *tx, ksock_peer_t *peer)
 {
         struct list_head *tmp;
-        ksock_conn_t     *conn = NULL;
+        ksock_conn_t     *typed = NULL;
+        int               tnob  = 0;
+        ksock_conn_t     *fallback = NULL;
+        int               fnob     = 0;
 
         /* Find the conn with the shortest tx queue */
         list_for_each (tmp, &peer->ksnp_conns) {
-                ksock_conn_t *c = list_entry (tmp, ksock_conn_t, ksnc_list);
+                ksock_conn_t *c = list_entry(tmp, ksock_conn_t, ksnc_list);
+                int           nob = atomic_read(&c->ksnc_tx_nob) +
+                                        c->ksnc_sock->sk->sk_wmem_queued;
 
                 LASSERT (!c->ksnc_closing);
-                
-                if (conn == NULL ||
-                    atomic_read (&conn->ksnc_tx_nob) >
-                    atomic_read (&c->ksnc_tx_nob))
-                        conn = c;
+
+                if (fallback == NULL || nob < fnob) {
+                        fallback = c;
+                        fnob     = nob;
+                }
+
+                if (!ksocknal_tunables.ksnd_typed_conns)
+                        continue;
+
+                switch (c->ksnc_type) {
+                default:
+                        LBUG();
+                case SOCKNAL_CONN_ANY:
+                        break;
+                case SOCKNAL_CONN_BULK_IN:
+                        continue;
+                case SOCKNAL_CONN_BULK_OUT:
+                        if (tx->tx_nob < ksocknal_tunables.ksnd_min_bulk)
+                                continue;
+                        break;
+                case SOCKNAL_CONN_CONTROL:
+                        if (tx->tx_nob >= ksocknal_tunables.ksnd_min_bulk)
+                                continue;
+                        break;
+                }
+
+                if (typed == NULL || nob < tnob) {
+                        typed = c;
+                        tnob  = nob;
+                }
         }
 
-        return (conn);
+        /* prefer the typed selection */
+        return ((typed != NULL) ? typed : fallback);
 }
 
 void
@@ -804,13 +836,19 @@ ksocknal_queue_tx_locked (ksock_tx_t *tx, ksock_conn_t *conn)
         unsigned long  flags;
         ksock_sched_t *sched = conn->ksnc_scheduler;
 
-        /* called holding global lock (read or irq-write) */
-
-        CDEBUG (D_NET, "Sending to "LPX64" on port %d\n", 
-                conn->ksnc_peer->ksnp_nid, conn->ksnc_port);
+        /* called holding global lock (read or irq-write) and caller may
+         * not have dropped this lock between finding conn and calling me,
+         * so we don't need the {get,put}connsock dance to deref
+         * ksnc_sock... */
+        LASSERT(!conn->ksnc_closing);
+        LASSERT(tx->tx_resid == tx->tx_nob);
+        
+        CDEBUG (D_NET, "Sending to "LPX64" ip %d.%d.%d.%d:%d\n", 
+                conn->ksnc_peer->ksnp_nid,
+                HIPQUAD(conn->ksnc_ipaddr),
+                conn->ksnc_port);
 
         atomic_add (tx->tx_nob, &conn->ksnc_tx_nob);
-        tx->tx_resid = tx->tx_nob;
         tx->tx_conn = conn;
 
 #if SOCKNAL_ZC
@@ -818,10 +856,12 @@ ksocknal_queue_tx_locked (ksock_tx_t *tx, ksock_conn_t *conn)
         /* NB this sets 1 ref on zccd, so the callback can only occur after
          * I've released this ref. */
 #endif
-
         spin_lock_irqsave (&sched->kss_lock, flags);
-                
-        tx->tx_deadline = jiffies_64 + ksocknal_io_timeout;
+
+        conn->ksnc_tx_deadline = jiffies + 
+                                 ksocknal_tunables.ksnd_io_timeout * HZ;
+        mb();                                   /* order with list_add_tail */
+
         list_add_tail (&tx->tx_list, &conn->ksnc_tx_queue);
                 
         if (conn->ksnc_tx_ready &&      /* able to send */
@@ -831,8 +871,7 @@ ksocknal_queue_tx_locked (ksock_tx_t *tx, ksock_conn_t *conn)
                 list_add_tail (&conn->ksnc_tx_list, 
                                &sched->kss_tx_conns);
                 conn->ksnc_tx_scheduled = 1;
-                if (waitqueue_active (&sched->kss_waitq))
-                        wake_up (&sched->kss_waitq);
+                wake_up (&sched->kss_waitq);
         }
 
         spin_unlock_irqrestore (&sched->kss_lock, flags);
@@ -843,17 +882,42 @@ ksocknal_find_connectable_route_locked (ksock_peer_t *peer)
 {
         struct list_head  *tmp;
         ksock_route_t     *route;
+        ksock_route_t     *candidate = NULL;
+        int                found = 0;
+        int                bits;
         
         list_for_each (tmp, &peer->ksnp_routes) {
                 route = list_entry (tmp, ksock_route_t, ksnr_list);
+                bits  = route->ksnr_connected;
+                
+                if ((bits & KSNR_TYPED_ROUTES) == KSNR_TYPED_ROUTES ||
+                    (bits & (1 << SOCKNAL_CONN_ANY)) != 0 ||
+                    route->ksnr_connecting != 0) {
+                        /* All typed connections have been established, or
+                         * an untyped connection has been established, or
+                         * connections are currently being established */
+                        found = 1;
+                        continue;
+                }
+
+                /* too soon to retry this guy? */
+                if (!time_after_eq (jiffies, route->ksnr_timeout))
+                        continue;
                 
-                if (route->ksnr_conn == NULL && /* not connected */
-                    !route->ksnr_connecting &&  /* not connecting */
-                    route->ksnr_timeout <= jiffies_64) /* OK to retry */
+                /* always do eager routes */
+                if (route->ksnr_eager)
                         return (route);
+
+                if (candidate == NULL) {
+                        /* If we don't find any other route that is fully
+                         * connected or connecting, the first connectable
+                         * route is returned.  If it fails to connect, it
+                         * will get placed at the end of the list */
+                        candidate = route;
+                }
         }
-        
-        return (NULL);
+        return (found ? NULL : candidate);
 }
 
 ksock_route_t *
@@ -865,7 +929,7 @@ ksocknal_find_connecting_route_locked (ksock_peer_t *peer)
         list_for_each (tmp, &peer->ksnp_routes) {
                 route = list_entry (tmp, ksock_route_t, ksnr_list);
                 
-                if (route->ksnr_connecting)
+                if (route->ksnr_connecting != 0)
                         return (route);
         }
         
@@ -897,6 +961,8 @@ ksocknal_launch_packet (ksock_tx_t *tx, ptl_nid_t nid)
                 tx->tx_nob, tx->tx_niov, tx->tx_nkiov);
 
         tx->tx_conn = NULL;                     /* only set when assigned a conn */
+        tx->tx_resid = tx->tx_nob;
+        tx->tx_hdr = (ptl_hdr_t *)tx->tx_iov[0].iov_base;
 
         g_lock = &ksocknal_data.ksnd_global_lock;
         read_lock (g_lock);
@@ -904,20 +970,21 @@ ksocknal_launch_packet (ksock_tx_t *tx, ptl_nid_t nid)
         peer = ksocknal_find_target_peer_locked (tx, nid);
         if (peer == NULL) {
                 read_unlock (g_lock);
-                return (PTL_FAIL);
+                return (-EHOSTUNREACH);
         }
 
-        /* Any routes need to be connected? (need write lock if so) */
-        if (ksocknal_find_connectable_route_locked (peer) == NULL) {
+        if (ksocknal_find_connectable_route_locked(peer) == NULL) {
                 conn = ksocknal_find_conn_locked (tx, peer);
                 if (conn != NULL) {
+                        /* I've got no autoconnect routes that need to be
+                         * connecting and I do have an actual connection... */
                         ksocknal_queue_tx_locked (tx, conn);
                         read_unlock (g_lock);
-                        return (PTL_OK);
+                        return (0);
                 }
         }
         
-        /* need a write lock now to change peer state... */
+        /* Making one or more connections; I'll need a write lock... */
 
         atomic_inc (&peer->ksnp_refcount);      /* +1 ref for me while I unlock */
         read_unlock (g_lock);
@@ -926,147 +993,166 @@ ksocknal_launch_packet (ksock_tx_t *tx, ptl_nid_t nid)
         if (peer->ksnp_closing) {               /* peer deleted as I blocked! */
                 write_unlock_irqrestore (g_lock, flags);
                 ksocknal_put_peer (peer);
-                return (PTL_FAIL);
+                return (-EHOSTUNREACH);
         }
         ksocknal_put_peer (peer);               /* drop ref I got above */
 
-        /* I may launch autoconnects, now we're write locked... */
-        while ((route = ksocknal_find_connectable_route_locked (peer)) != NULL)
+        for (;;) {
+                /* launch any/all autoconnections that need it */
+                route = ksocknal_find_connectable_route_locked (peer);
+                if (route == NULL)
+                        break;
+
                 ksocknal_launch_autoconnect_locked (route);
+        }
 
         conn = ksocknal_find_conn_locked (tx, peer);
         if (conn != NULL) {
+                /* Connection exists; queue message on it */
                 ksocknal_queue_tx_locked (tx, conn);
                 write_unlock_irqrestore (g_lock, flags);
-                return (PTL_OK);
+                return (0);
         }
-                
-        if (ksocknal_find_connecting_route_locked (peer) == NULL) {
-                /* no routes actually connecting now */
+
+        route = ksocknal_find_connecting_route_locked (peer);
+        if (route != NULL) {
+                /* At least 1 connection is being established; queue the
+                 * message... */
+                list_add_tail (&tx->tx_list, &peer->ksnp_tx_queue);
                 write_unlock_irqrestore (g_lock, flags);
-                return (PTL_FAIL);
+                return (0);
         }
-
-        list_add_tail (&tx->tx_list, &peer->ksnp_tx_queue);
-
+        
         write_unlock_irqrestore (g_lock, flags);
-        return (PTL_OK);
+        return (-EHOSTUNREACH);
 }
 
-ksock_ltx_t *
-ksocknal_setup_hdr (nal_cb_t *nal, void *private, lib_msg_t *cookie, 
-                    ptl_hdr_t *hdr, int type)
+ptl_err_t
+ksocknal_sendmsg(nal_cb_t     *nal, 
+                 void         *private, 
+                 lib_msg_t    *cookie,
+                 ptl_hdr_t    *hdr, 
+                 int           type, 
+                 ptl_nid_t     nid, 
+                 ptl_pid_t     pid,
+                 unsigned int  payload_niov, 
+                 struct iovec *payload_iov, 
+                 ptl_kiov_t   *payload_kiov,
+                 size_t        payload_offset,
+                 size_t        payload_nob)
 {
         ksock_ltx_t  *ltx;
+        int           desc_size;
+        int           rc;
 
-        /* I may not block for a transmit descriptor if I might block the
-         * receiver, or an interrupt handler. */
-        ltx = ksocknal_get_ltx (!(type == PTL_MSG_ACK ||
-                                  type == PTL_MSG_REPLY ||
-                                  in_interrupt ()));
+        /* NB 'private' is different depending on what we're sending.
+         * Just ignore it... */
+
+        CDEBUG(D_NET, "sending "LPSZ" bytes in %d frags to nid:"LPX64
+               " pid %d\n", payload_nob, payload_niov, nid , pid);
+
+        LASSERT (payload_nob == 0 || payload_niov > 0);
+        LASSERT (payload_niov <= PTL_MD_MAX_IOV);
+
+        /* It must be OK to kmap() if required */
+        LASSERT (payload_kiov == NULL || !in_interrupt ());
+        /* payload is either all vaddrs or all pages */
+        LASSERT (!(payload_kiov != NULL && payload_iov != NULL));
+        
+        if (payload_iov != NULL)
+                desc_size = offsetof(ksock_ltx_t, ltx_iov[1 + payload_niov]);
+        else
+                desc_size = offsetof(ksock_ltx_t, ltx_kiov[payload_niov]);
+        
+        if (in_interrupt() ||
+            type == PTL_MSG_ACK ||
+            type == PTL_MSG_REPLY) {
+                /* Can't block if in interrupt or responding to an incoming
+                 * message */
+                PORTAL_ALLOC_ATOMIC(ltx, desc_size);
+        } else {
+                PORTAL_ALLOC(ltx, desc_size);
+        }
+        
         if (ltx == NULL) {
-                CERROR ("Can't allocate tx desc\n");
-                return (NULL);
+                CERROR("Can't allocate tx desc type %d size %d %s\n",
+                       type, desc_size, in_interrupt() ? "(intr)" : "");
+                return (PTL_NO_SPACE);
         }
 
-        /* Init local send packet (storage for hdr, finalize() args) */
+        atomic_inc(&ksocknal_data.ksnd_nactive_ltxs);
+
+        ltx->ltx_desc_size = desc_size;
+        
+        /* We always have 1 mapped frag for the header */
+        ltx->ltx_tx.tx_iov = ltx->ltx_iov;
+        ltx->ltx_iov[0].iov_base = &ltx->ltx_hdr;
+        ltx->ltx_iov[0].iov_len = sizeof(*hdr);
         ltx->ltx_hdr = *hdr;
+        
         ltx->ltx_private = private;
         ltx->ltx_cookie = cookie;
         
-        /* Init common ltx_tx */
         ltx->ltx_tx.tx_isfwd = 0;
-        ltx->ltx_tx.tx_nob = sizeof (*hdr);
+        ltx->ltx_tx.tx_nob = sizeof (*hdr) + payload_nob;
 
-        /* We always have 1 mapped frag for the header */
-        ltx->ltx_tx.tx_niov = 1;
-        ltx->ltx_tx.tx_iov = &ltx->ltx_iov_space.hdr;
-        ltx->ltx_tx.tx_iov[0].iov_base = &ltx->ltx_hdr;
-        ltx->ltx_tx.tx_iov[0].iov_len = sizeof (ltx->ltx_hdr);
+        if (payload_iov != NULL) {
+                /* payload is all mapped */
+                ltx->ltx_tx.tx_kiov  = NULL;
+                ltx->ltx_tx.tx_nkiov = 0;
+
+                ltx->ltx_tx.tx_niov = 
+                        1 + lib_extract_iov(payload_niov, &ltx->ltx_iov[1],
+                                            payload_niov, payload_iov,
+                                            payload_offset, payload_nob);
+        } else {
+                /* payload is all pages */
+                ltx->ltx_tx.tx_niov = 1;
 
-        ltx->ltx_tx.tx_kiov  = NULL;
-        ltx->ltx_tx.tx_nkiov = 0;
+                ltx->ltx_tx.tx_kiov = ltx->ltx_kiov;
+                ltx->ltx_tx.tx_nkiov =
+                        lib_extract_kiov(payload_niov, ltx->ltx_kiov,
+                                         payload_niov, payload_kiov,
+                                         payload_offset, payload_nob);
+        }
 
-        return (ltx);
+        rc = ksocknal_launch_packet(&ltx->ltx_tx, nid);
+        if (rc == 0)
+                return (PTL_OK);
+        
+        ksocknal_free_ltx(ltx);
+        return (PTL_FAIL);
 }
 
-int
+ptl_err_t
 ksocknal_send (nal_cb_t *nal, void *private, lib_msg_t *cookie,
                ptl_hdr_t *hdr, int type, ptl_nid_t nid, ptl_pid_t pid,
                unsigned int payload_niov, struct iovec *payload_iov,
-               size_t payload_len)
+               size_t payload_offset, size_t payload_len)
 {
-        ksock_ltx_t  *ltx;
-        int           rc;
-
-        /* NB 'private' is different depending on what we're sending.
-         * Just ignore it until we can rely on it
-         */
-
-        CDEBUG(D_NET,
-               "sending "LPSZ" bytes in %d mapped frags to nid: "LPX64
-               " pid %d\n", payload_len, payload_niov, nid, pid);
-
-        ltx = ksocknal_setup_hdr (nal, private, cookie, hdr, type);
-        if (ltx == NULL)
-                return (PTL_FAIL);
-
-        /* append the payload_iovs to the one pointing at the header */
-        LASSERT (ltx->ltx_tx.tx_niov == 1 && ltx->ltx_tx.tx_nkiov == 0);
-        LASSERT (payload_niov <= PTL_MD_MAX_IOV);
-
-        memcpy (ltx->ltx_tx.tx_iov + 1, payload_iov,
-                payload_niov * sizeof (*payload_iov));
-        ltx->ltx_tx.tx_niov = 1 + payload_niov;
-        ltx->ltx_tx.tx_nob = sizeof (*hdr) + payload_len;
-
-        rc = ksocknal_launch_packet (&ltx->ltx_tx, nid);
-        if (rc != PTL_OK)
-                ksocknal_put_ltx (ltx);
-        
-        return (rc);
+        return (ksocknal_sendmsg(nal, private, cookie,
+                                 hdr, type, nid, pid,
+                                 payload_niov, payload_iov, NULL,
+                                 payload_offset, payload_len));
 }
 
-int
+ptl_err_t
 ksocknal_send_pages (nal_cb_t *nal, void *private, lib_msg_t *cookie, 
                      ptl_hdr_t *hdr, int type, ptl_nid_t nid, ptl_pid_t pid,
-                     unsigned int payload_niov, ptl_kiov_t *payload_iov, size_t payload_len)
+                     unsigned int payload_niov, ptl_kiov_t *payload_kiov, 
+                     size_t payload_offset, size_t payload_len)
 {
-        ksock_ltx_t *ltx;
-        int          rc;
-
-        /* NB 'private' is different depending on what we're sending.
-         * Just ignore it until we can rely on it */
-
-        CDEBUG(D_NET,
-               "sending "LPSZ" bytes in %d mapped frags to nid: "LPX64" pid %d\n",
-               payload_len, payload_niov, nid, pid);
-
-        ltx = ksocknal_setup_hdr (nal, private, cookie, hdr, type);
-        if (ltx == NULL)
-                return (PTL_FAIL);
-
-        LASSERT (ltx->ltx_tx.tx_niov == 1 && ltx->ltx_tx.tx_nkiov == 0);
-        LASSERT (payload_niov <= PTL_MD_MAX_IOV);
-        
-        ltx->ltx_tx.tx_kiov = ltx->ltx_iov_space.payload.kiov;
-        memcpy (ltx->ltx_tx.tx_kiov, payload_iov, 
-                payload_niov * sizeof (*payload_iov));
-        ltx->ltx_tx.tx_nkiov = payload_niov;
-        ltx->ltx_tx.tx_nob = sizeof (*hdr) + payload_len;
-
-        rc = ksocknal_launch_packet (&ltx->ltx_tx, nid);
-        if (rc != PTL_OK) 
-                ksocknal_put_ltx (ltx);
-                
-        return (rc);
+        return (ksocknal_sendmsg(nal, private, cookie,
+                                 hdr, type, nid, pid,
+                                 payload_niov, NULL, payload_kiov,
+                                 payload_offset, payload_len));
 }
 
 void
 ksocknal_fwd_packet (void *arg, kpr_fwd_desc_t *fwd)
 {
         ptl_nid_t     nid = fwd->kprfd_gateway_nid;
-        ksock_tx_t   *tx  = (ksock_tx_t *)&fwd->kprfd_scratch;
+        ksock_ftx_t  *ftx = (ksock_ftx_t *)&fwd->kprfd_scratch;
         int           rc;
         
         CDEBUG (D_NET, "Forwarding [%p] -> "LPX64" ("LPX64"))\n", fwd,
@@ -1076,19 +1162,20 @@ ksocknal_fwd_packet (void *arg, kpr_fwd_desc_t *fwd)
         if (nid == ksocknal_lib.ni.nid)
                 nid = fwd->kprfd_target_nid;
 
-        tx->tx_isfwd = 1;                   /* This is a forwarding packet */
-        tx->tx_nob   = fwd->kprfd_nob;
-        tx->tx_niov  = fwd->kprfd_niov;
-        tx->tx_iov   = fwd->kprfd_iov;
-        tx->tx_nkiov = 0;
-        tx->tx_kiov  = NULL;
-        tx->tx_hdr   = (ptl_hdr_t *)fwd->kprfd_iov[0].iov_base;
+        /* setup iov for hdr */
+        ftx->ftx_iov.iov_base = fwd->kprfd_hdr;
+        ftx->ftx_iov.iov_len = sizeof(ptl_hdr_t);
 
-        rc = ksocknal_launch_packet (tx, nid);
-        if (rc != 0) {
-                /* FIXME, could pass a better completion error */
-                kpr_fwd_done (&ksocknal_data.ksnd_router, fwd, -EHOSTUNREACH);
-        }
+        ftx->ftx_tx.tx_isfwd = 1;                  /* This is a forwarding packet */
+        ftx->ftx_tx.tx_nob   = sizeof(ptl_hdr_t) + fwd->kprfd_nob;
+        ftx->ftx_tx.tx_niov  = 1;
+        ftx->ftx_tx.tx_iov   = &ftx->ftx_iov;
+        ftx->ftx_tx.tx_nkiov = fwd->kprfd_niov;
+        ftx->ftx_tx.tx_kiov  = fwd->kprfd_kiov;
+
+        rc = ksocknal_launch_packet (&ftx->ftx_tx, nid);
+        if (rc != 0)
+                kpr_fwd_done (&ksocknal_data.ksnd_router, fwd, rc);
 }
 
 int
@@ -1114,22 +1201,32 @@ ksocknal_fmb_callback (void *arg, int error)
 {
         ksock_fmb_t       *fmb = (ksock_fmb_t *)arg;
         ksock_fmb_pool_t  *fmp = fmb->fmb_pool;
-        ptl_hdr_t         *hdr = (ptl_hdr_t *) page_address(fmb->fmb_pages[0]);
+        ptl_hdr_t         *hdr = (ptl_hdr_t *)page_address(fmb->fmb_kiov[0].kiov_page);
         ksock_conn_t      *conn = NULL;
         ksock_sched_t     *sched;
         unsigned long      flags;
+        char               ipbuf[PTL_NALFMT_SIZE];
+        char               ipbuf2[PTL_NALFMT_SIZE];
 
         if (error != 0)
-                CERROR("Failed to route packet from "LPX64" to "LPX64": %d\n",
-                       NTOH__u64(hdr->src_nid), NTOH__u64(hdr->dest_nid),
+                CERROR("Failed to route packet from "
+                       LPX64" %s to "LPX64" %s: %d\n",
+                       NTOH__u64(hdr->src_nid),
+                       portals_nid2str(SOCKNAL, NTOH__u64(hdr->src_nid), ipbuf),
+                       NTOH__u64(hdr->dest_nid),
+                       portals_nid2str(SOCKNAL, NTOH__u64(hdr->dest_nid), ipbuf2),
                        error);
         else
                 CDEBUG (D_NET, "routed packet from "LPX64" to "LPX64": OK\n",
                         NTOH__u64 (hdr->src_nid), NTOH__u64 (hdr->dest_nid));
 
+        /* drop peer ref taken on init */
+        ksocknal_put_peer (fmb->fmb_peer);
+
         spin_lock_irqsave (&fmp->fmp_lock, flags);
 
         list_add (&fmb->fmb_list, &fmp->fmp_idle_fmbs);
+        fmp->fmp_nactive_fmbs--;
 
         if (!list_empty (&fmp->fmp_blocked_conns)) {
                 conn = list_entry (fmb->fmb_pool->fmp_blocked_conns.next,
@@ -1139,9 +1236,6 @@ ksocknal_fmb_callback (void *arg, int error)
 
         spin_unlock_irqrestore (&fmp->fmp_lock, flags);
 
-        /* drop peer ref taken on init */
-        ksocknal_put_peer (fmb->fmb_peer);
-        
         if (conn == NULL)
                 return;
 
@@ -1156,9 +1250,7 @@ ksocknal_fmb_callback (void *arg, int error)
         spin_lock_irqsave (&sched->kss_lock, flags);
 
         list_add_tail (&conn->ksnc_rx_list, &sched->kss_rx_conns);
-
-        if (waitqueue_active (&sched->kss_waitq))
-                wake_up (&sched->kss_waitq);
+        wake_up (&sched->kss_waitq);
 
         spin_unlock_irqrestore (&sched->kss_lock, flags);
 }
@@ -1167,15 +1259,14 @@ ksock_fmb_t *
 ksocknal_get_idle_fmb (ksock_conn_t *conn)
 {
         int               payload_nob = conn->ksnc_rx_nob_left;
-        int               packet_nob = sizeof (ptl_hdr_t) + payload_nob;
         unsigned long     flags;
         ksock_fmb_pool_t *pool;
         ksock_fmb_t      *fmb;
 
         LASSERT (conn->ksnc_rx_state == SOCKNAL_RX_GET_FMB);
-        LASSERT (ksocknal_data.ksnd_fmbs != NULL);
+        LASSERT (kpr_routing(&ksocknal_data.ksnd_router));
 
-        if (packet_nob <= SOCKNAL_SMALL_FWD_PAGES * PAGE_SIZE)
+        if (payload_nob <= SOCKNAL_SMALL_FWD_PAGES * PAGE_SIZE)
                 pool = &ksocknal_data.ksnd_small_fmp;
         else
                 pool = &ksocknal_data.ksnd_large_fmp;
@@ -1186,6 +1277,7 @@ ksocknal_get_idle_fmb (ksock_conn_t *conn)
                 fmb = list_entry(pool->fmp_idle_fmbs.next,
                                  ksock_fmb_t, fmb_list);
                 list_del (&fmb->fmb_list);
+                pool->fmp_nactive_fmbs++;
                 spin_unlock_irqrestore (&pool->fmp_lock, flags);
 
                 return (fmb);
@@ -1205,99 +1297,64 @@ ksocknal_get_idle_fmb (ksock_conn_t *conn)
 int
 ksocknal_init_fmb (ksock_conn_t *conn, ksock_fmb_t *fmb)
 {
-        int payload_nob = conn->ksnc_rx_nob_left;
-        int packet_nob = sizeof (ptl_hdr_t) + payload_nob;
+        int       payload_nob = conn->ksnc_rx_nob_left;
         ptl_nid_t dest_nid = NTOH__u64 (conn->ksnc_hdr.dest_nid);
-        int niov;                               /* at least the header */
-        int nob;
+        int       niov = 0;
+        int       nob = payload_nob;
 
         LASSERT (conn->ksnc_rx_scheduled);
         LASSERT (conn->ksnc_rx_state == SOCKNAL_RX_GET_FMB);
         LASSERT (conn->ksnc_rx_nob_wanted == conn->ksnc_rx_nob_left);
         LASSERT (payload_nob >= 0);
-        LASSERT (packet_nob <= fmb->fmb_npages * PAGE_SIZE);
+        LASSERT (payload_nob <= fmb->fmb_pool->fmp_buff_pages * PAGE_SIZE);
         LASSERT (sizeof (ptl_hdr_t) < PAGE_SIZE);
-
-        /* Got a forwarding buffer; copy the header we just read into the
-         * forwarding buffer.  If there's payload, start reading reading it
-         * into the buffer, otherwise the forwarding buffer can be kicked
-         * off immediately.
-         *
-         * NB fmb->fmb_iov spans the WHOLE packet.
-         *    conn->ksnc_rx_iov spans just the payload.
-         */
-        fmb->fmb_iov[0].iov_base = page_address (fmb->fmb_pages[0]);
-
-        /* copy header */
-        memcpy (fmb->fmb_iov[0].iov_base, &conn->ksnc_hdr, sizeof (ptl_hdr_t));
+        LASSERT (fmb->fmb_kiov[0].kiov_offset == 0);
 
         /* Take a ref on the conn's peer to prevent module unload before
-         * forwarding completes.  NB we ref peer and not conn since because
-         * all refs on conn after it has been closed must remove themselves
-         * in finite time */
+         * forwarding completes. */
         fmb->fmb_peer = conn->ksnc_peer;
         atomic_inc (&conn->ksnc_peer->ksnp_refcount);
 
-        if (payload_nob == 0) {         /* got complete packet already */
-                CDEBUG (D_NET, "%p "LPX64"->"LPX64" %d fwd_start (immediate)\n",
-                        conn, NTOH__u64 (conn->ksnc_hdr.src_nid),
-                        dest_nid, packet_nob);
+        /* Copy the header we just read into the forwarding buffer.  If
+         * there's payload, start reading reading it into the buffer,
+         * otherwise the forwarding buffer can be kicked off
+         * immediately. */
+        fmb->fmb_hdr = conn->ksnc_hdr;
 
-                fmb->fmb_iov[0].iov_len = sizeof (ptl_hdr_t);
+        while (nob > 0) {
+                LASSERT (niov < fmb->fmb_pool->fmp_buff_pages);
+                LASSERT (fmb->fmb_kiov[niov].kiov_offset == 0);
+                fmb->fmb_kiov[niov].kiov_len = MIN (PAGE_SIZE, nob);
+                nob -= PAGE_SIZE;
+                niov++;
+        }
+
+        kpr_fwd_init(&fmb->fmb_fwd, dest_nid, &fmb->fmb_hdr,
+                     payload_nob, niov, fmb->fmb_kiov,
+                     ksocknal_fmb_callback, fmb);
 
-                kpr_fwd_init (&fmb->fmb_fwd, dest_nid,
-                              packet_nob, 1, fmb->fmb_iov,
-                              ksocknal_fmb_callback, fmb);
+        if (payload_nob == 0) {         /* got complete packet already */
+                CDEBUG (D_NET, "%p "LPX64"->"LPX64" fwd_start (immediate)\n",
+                        conn, NTOH__u64 (conn->ksnc_hdr.src_nid), dest_nid);
 
-                /* forward it now */
                 kpr_fwd_start (&ksocknal_data.ksnd_router, &fmb->fmb_fwd);
 
                 ksocknal_new_packet (conn, 0);  /* on to next packet */
                 return (1);
         }
 
-        niov = 1;
-        if (packet_nob <= PAGE_SIZE) {  /* whole packet fits in first page */
-                fmb->fmb_iov[0].iov_len = packet_nob;
-        } else {
-                fmb->fmb_iov[0].iov_len = PAGE_SIZE;
-                nob = packet_nob - PAGE_SIZE;
-
-                do {
-                        LASSERT (niov < fmb->fmb_npages);
-                        fmb->fmb_iov[niov].iov_base =
-                                page_address (fmb->fmb_pages[niov]);
-                        fmb->fmb_iov[niov].iov_len = MIN (PAGE_SIZE, nob);
-                        nob -= PAGE_SIZE;
-                        niov++;
-                } while (nob > 0);
-        }
-
-        kpr_fwd_init (&fmb->fmb_fwd, dest_nid,
-                      packet_nob, niov, fmb->fmb_iov,
-                      ksocknal_fmb_callback, fmb);
-
         conn->ksnc_cookie = fmb;                /* stash fmb for later */
         conn->ksnc_rx_state = SOCKNAL_RX_BODY_FWD; /* read in the payload */
-        conn->ksnc_rx_deadline = jiffies_64 + ksocknal_io_timeout; /* start timeout */
         
-        /* payload is desc's iov-ed buffer, but skipping the hdr */
-        LASSERT (niov <= sizeof (conn->ksnc_rx_iov_space) /
-                 sizeof (struct iovec));
-
-        conn->ksnc_rx_iov = (struct iovec *)&conn->ksnc_rx_iov_space;
-        conn->ksnc_rx_iov[0].iov_base =
-                (void *)(((unsigned long)fmb->fmb_iov[0].iov_base) +
-                         sizeof (ptl_hdr_t));
-        conn->ksnc_rx_iov[0].iov_len =
-                fmb->fmb_iov[0].iov_len - sizeof (ptl_hdr_t);
-
-        if (niov > 1)
-                memcpy(&conn->ksnc_rx_iov[1], &fmb->fmb_iov[1],
-                       (niov - 1) * sizeof (struct iovec));
-
-        conn->ksnc_rx_niov = niov;
+        /* Set up conn->ksnc_rx_kiov to read the payload into fmb's kiov-ed
+         * buffer */
+        LASSERT (niov <= sizeof(conn->ksnc_rx_iov_space)/sizeof(ptl_kiov_t));
 
+        conn->ksnc_rx_niov = 0;
+        conn->ksnc_rx_nkiov = niov;
+        conn->ksnc_rx_kiov = conn->ksnc_rx_iov_space.kiov;
+        memcpy(conn->ksnc_rx_kiov, fmb->fmb_kiov, niov * sizeof(ptl_kiov_t));
+        
         CDEBUG (D_NET, "%p "LPX64"->"LPX64" %d reading body\n", conn,
                 NTOH__u64 (conn->ksnc_hdr.src_nid), dest_nid, payload_nob);
         return (0);
@@ -1308,38 +1365,44 @@ ksocknal_fwd_parse (ksock_conn_t *conn)
 {
         ksock_peer_t *peer;
         ptl_nid_t     dest_nid = NTOH__u64 (conn->ksnc_hdr.dest_nid);
-        int           body_len = NTOH__u32 (PTL_HDR_LENGTH(&conn->ksnc_hdr));
+        ptl_nid_t     src_nid = NTOH__u64 (conn->ksnc_hdr.src_nid);
+        int           body_len = NTOH__u32 (conn->ksnc_hdr.payload_length);
+        char str[PTL_NALFMT_SIZE];
+        char str2[PTL_NALFMT_SIZE];
 
         CDEBUG (D_NET, "%p "LPX64"->"LPX64" %d parsing header\n", conn,
-                NTOH__u64 (conn->ksnc_hdr.src_nid),
-                dest_nid, conn->ksnc_rx_nob_left);
+                src_nid, dest_nid, conn->ksnc_rx_nob_left);
 
         LASSERT (conn->ksnc_rx_state == SOCKNAL_RX_HEADER);
         LASSERT (conn->ksnc_rx_scheduled);
 
         if (body_len < 0) {                 /* length corrupt (overflow) */
-                CERROR("dropping packet from "LPX64" for "LPX64": packet "
-                       "size %d illegal\n", NTOH__u64 (conn->ksnc_hdr.src_nid),
-                       dest_nid, body_len);
+                CERROR("dropping packet from "LPX64" (%s) for "LPX64" (%s): "
+                       "packet size %d illegal\n",
+                       src_nid, portals_nid2str(TCPNAL, src_nid, str),
+                       dest_nid, portals_nid2str(TCPNAL, dest_nid, str2),
+                       body_len);
 
                 ksocknal_new_packet (conn, 0);  /* on to new packet */
-                ksocknal_close_conn_unlocked (conn); /* give up on conn */
                 return;
         }
 
-        if (ksocknal_data.ksnd_fmbs == NULL) {        /* not forwarding */
-                CERROR("dropping packet from "LPX64" for "LPX64": not "
-                       "forwarding\n", conn->ksnc_hdr.src_nid,
-                       conn->ksnc_hdr.dest_nid);
+        if (!kpr_routing(&ksocknal_data.ksnd_router)) {    /* not forwarding */
+                CERROR("dropping packet from "LPX64" (%s) for "LPX64
+                       " (%s): not forwarding\n",
+                       src_nid, portals_nid2str(TCPNAL, src_nid, str),
+                       dest_nid, portals_nid2str(TCPNAL, dest_nid, str2));
                 /* on to new packet (skip this one's body) */
                 ksocknal_new_packet (conn, body_len);
                 return;
         }
 
-        if (body_len > SOCKNAL_MAX_FWD_PAYLOAD) {      /* too big to forward */
-                CERROR ("dropping packet from "LPX64" for "LPX64
-                        ": packet size %d too big\n", conn->ksnc_hdr.src_nid,
-                        conn->ksnc_hdr.dest_nid, body_len);
+        if (body_len > PTL_MTU) {      /* too big to forward */
+                CERROR ("dropping packet from "LPX64" (%s) for "LPX64
+                        "(%s): packet size %d too big\n",
+                        src_nid, portals_nid2str(TCPNAL, src_nid, str),
+                        dest_nid, portals_nid2str(TCPNAL, dest_nid, str2),
+                        body_len);
                 /* on to new packet (skip this one's body) */
                 ksocknal_new_packet (conn, body_len);
                 return;
@@ -1348,9 +1411,10 @@ ksocknal_fwd_parse (ksock_conn_t *conn)
         /* should have gone direct */
         peer = ksocknal_get_peer (conn->ksnc_hdr.dest_nid);
         if (peer != NULL) {
-                CERROR ("dropping packet from "LPX64" for "LPX64
-                        ": target is a peer\n", conn->ksnc_hdr.src_nid,
-                        conn->ksnc_hdr.dest_nid);
+                CERROR ("dropping packet from "LPX64" (%s) for "LPX64
+                        "(%s): target is a peer\n",
+                        src_nid, portals_nid2str(TCPNAL, src_nid, str),
+                        dest_nid, portals_nid2str(TCPNAL, dest_nid, str2));
                 ksocknal_put_peer (peer);  /* drop ref from get above */
 
                 /* on to next packet (skip this one's body) */
@@ -1373,6 +1437,9 @@ ksocknal_new_packet (ksock_conn_t *conn, int nob_to_skip)
         int   skipped;
 
         if (nob_to_skip == 0) {         /* right at next packet boundary now */
+                conn->ksnc_rx_started = 0;
+                mb ();                          /* racing with timeout thread */
+                
                 conn->ksnc_rx_state = SOCKNAL_RX_HEADER;
                 conn->ksnc_rx_nob_wanted = sizeof (ptl_hdr_t);
                 conn->ksnc_rx_nob_left = sizeof (ptl_hdr_t);
@@ -1415,25 +1482,13 @@ ksocknal_new_packet (ksock_conn_t *conn, int nob_to_skip)
         return (0);
 }
 
-void
-ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
+int
+ksocknal_process_receive (ksock_conn_t *conn)
 {
-        ksock_conn_t *conn;
         ksock_fmb_t  *fmb;
         int           rc;
-
-        /* NB: sched->ksnc_lock lock held */
-
-        LASSERT (!list_empty (&sched->kss_rx_conns));
-        conn = list_entry(sched->kss_rx_conns.next, ksock_conn_t, ksnc_rx_list);
-        list_del (&conn->ksnc_rx_list);
-
-        spin_unlock_irqrestore (&sched->kss_lock, *irq_flags);
-
-        CDEBUG(D_NET, "sched %p conn %p\n", sched, conn);
+        
         LASSERT (atomic_read (&conn->ksnc_refcount) > 0);
-        LASSERT (conn->ksnc_rx_scheduled);
-        LASSERT (conn->ksnc_rx_ready);
 
         /* doesn't need a forwarding buffer */
         if (conn->ksnc_rx_state != SOCKNAL_RX_GET_FMB)
@@ -1441,13 +1496,15 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
 
  get_fmb:
         fmb = ksocknal_get_idle_fmb (conn);
-        if (fmb == NULL) {      /* conn descheduled waiting for idle fmb */
-                spin_lock_irqsave (&sched->kss_lock, *irq_flags);
-                return;
+        if (fmb == NULL) {
+                /* conn descheduled waiting for idle fmb */
+                return (0);
         }
 
-        if (ksocknal_init_fmb (conn, fmb)) /* packet forwarded ? */
-                goto out;               /* come back later for next packet */
+        if (ksocknal_init_fmb (conn, fmb)) {
+                /* packet forwarded */
+                return (0);
+        }
 
  try_read:
         /* NB: sched lock NOT held */
@@ -1458,32 +1515,32 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
 
         LASSERT (conn->ksnc_rx_nob_wanted > 0);
 
-        conn->ksnc_rx_ready = 0;/* data ready may race with me and set ready */
-        mb();                   /* => clear BEFORE trying to read */
-
-        rc = ksocknal_recvmsg(conn);
+        rc = ksocknal_receive(conn);
 
         if (rc <= 0) {
-                if (ksocknal_close_conn_unlocked (conn)) {
-                        /* I'm the first to close */
-                        if (rc < 0)
-                                CERROR ("[%p] Error %d on read from "LPX64" ip %08x:%d\n",
-                                        conn, rc, conn->ksnc_peer->ksnp_nid,
-                                        conn->ksnc_ipaddr, conn->ksnc_port);
-                        else
-                                CERROR ("[%p] EOF from "LPX64" ip %08x:%d\n",
-                                        conn, conn->ksnc_peer->ksnp_nid,
-                                        conn->ksnc_ipaddr, conn->ksnc_port);
-                }
-                goto out;
-        }
+                LASSERT (rc != -EAGAIN);
 
-        if (conn->ksnc_rx_nob_wanted != 0)      /* short read */
-                goto out;                       /* try again later */
+                if (rc == 0)
+                        CWARN ("[%p] EOF from "LPX64" ip %d.%d.%d.%d:%d\n",
+                               conn, conn->ksnc_peer->ksnp_nid,
+                               HIPQUAD(conn->ksnc_ipaddr),
+                               conn->ksnc_port);
+                else if (!conn->ksnc_closing)
+                        CERROR ("[%p] Error %d on read from "LPX64
+                                " ip %d.%d.%d.%d:%d\n",
+                                conn, rc, conn->ksnc_peer->ksnp_nid,
+                                HIPQUAD(conn->ksnc_ipaddr),
+                                conn->ksnc_port);
 
-        /* got all I wanted, assume there's more - prevent data_ready locking */
-        conn->ksnc_rx_ready = 1;
+                ksocknal_close_conn_and_siblings (conn, rc);
+                return (rc == 0 ? -ESHUTDOWN : rc);
+        }
 
+        if (conn->ksnc_rx_nob_wanted != 0) {
+                /* short read */
+                return (-EAGAIN);
+        }
+        
         switch (conn->ksnc_rx_state) {
         case SOCKNAL_RX_HEADER:
                 if (conn->ksnc_hdr.type != HTON__u32(PTL_MSG_HELLO) &&
@@ -1492,7 +1549,7 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
                         ksocknal_fwd_parse (conn);
                         switch (conn->ksnc_rx_state) {
                         case SOCKNAL_RX_HEADER: /* skipped (zero payload) */
-                                goto out;       /* => come back later */
+                                return (0);     /* => come back later */
                         case SOCKNAL_RX_SLOP:   /* skipping packet's body */
                                 goto try_read;  /* => go read it */
                         case SOCKNAL_RX_GET_FMB: /* forwarding */
@@ -1506,9 +1563,6 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
                 /* sets wanted_len, iovs etc */
                 lib_parse(&ksocknal_lib, &conn->ksnc_hdr, conn);
 
-                /* start timeout (lib is waiting for finalize) */
-                conn->ksnc_rx_deadline = jiffies_64 + ksocknal_io_timeout;
-
                 if (conn->ksnc_rx_nob_wanted != 0) { /* need to get payload? */
                         conn->ksnc_rx_state = SOCKNAL_RX_BODY;
                         goto try_read;          /* go read the payload */
@@ -1517,14 +1571,13 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
 
         case SOCKNAL_RX_BODY:
                 /* payload all received */
-                conn->ksnc_rx_deadline = 0;     /* cancel timeout */
-                lib_finalize(&ksocknal_lib, NULL, conn->ksnc_cookie);
+                lib_finalize(&ksocknal_lib, NULL, conn->ksnc_cookie, PTL_OK);
                 /* Fall through */
 
         case SOCKNAL_RX_SLOP:
                 /* starting new packet? */
                 if (ksocknal_new_packet (conn, conn->ksnc_rx_nob_left))
-                        goto out;       /* come back later */
+                        return (0);     /* come back later */
                 goto try_read;          /* try to finish reading slop now */
 
         case SOCKNAL_RX_BODY_FWD:
@@ -1534,9 +1587,6 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
                         NTOH__u64 (conn->ksnc_hdr.dest_nid),
                         conn->ksnc_rx_nob_left);
 
-                /* cancel timeout (only needed it while fmb allocated) */
-                conn->ksnc_rx_deadline = 0;
-
                 /* forward the packet. NB ksocknal_init_fmb() put fmb into
                  * conn->ksnc_cookie */
                 fmb = (ksock_fmb_t *)conn->ksnc_cookie;
@@ -1546,7 +1596,7 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
                 LASSERT (conn->ksnc_rx_nob_left == 0);
 
                 ksocknal_new_packet (conn, 0);  /* on to next packet */
-                goto out;                       /* (later) */
+                return (0);                     /* (later) */
 
         default:
                 break;
@@ -1554,25 +1604,13 @@ ksocknal_process_receive (ksock_sched_t *sched, unsigned long *irq_flags)
 
         /* Not Reached */
         LBUG ();
-
- out:
-        spin_lock_irqsave (&sched->kss_lock, *irq_flags);
-
-        /* no data there to read? */
-        if (!conn->ksnc_rx_ready) {
-                /* let socket callback schedule again */
-                conn->ksnc_rx_scheduled = 0;
-                /* drop scheduler's ref */
-                ksocknal_put_conn (conn);   
-        } else {
-                /* stay scheduled */
-                list_add_tail (&conn->ksnc_rx_list, &sched->kss_rx_conns);
-        }
+        return (-EINVAL);                       /* keep gcc happy */
 }
 
-int
+ptl_err_t
 ksocknal_recv (nal_cb_t *nal, void *private, lib_msg_t *msg,
-               unsigned int niov, struct iovec *iov, size_t mlen, size_t rlen)
+               unsigned int niov, struct iovec *iov, 
+               size_t offset, size_t mlen, size_t rlen)
 {
         ksock_conn_t *conn = (ksock_conn_t *)private;
 
@@ -1585,20 +1623,22 @@ ksocknal_recv (nal_cb_t *nal, void *private, lib_msg_t *msg,
 
         conn->ksnc_rx_nkiov = 0;
         conn->ksnc_rx_kiov = NULL;
-        conn->ksnc_rx_niov = niov;
         conn->ksnc_rx_iov = conn->ksnc_rx_iov_space.iov;
-        memcpy (conn->ksnc_rx_iov, iov, niov * sizeof (*iov));
+        conn->ksnc_rx_niov =
+                lib_extract_iov(PTL_MD_MAX_IOV, conn->ksnc_rx_iov,
+                                niov, iov, offset, mlen);
 
         LASSERT (mlen == 
                  lib_iov_nob (conn->ksnc_rx_niov, conn->ksnc_rx_iov) +
                  lib_kiov_nob (conn->ksnc_rx_nkiov, conn->ksnc_rx_kiov));
 
-        return (rlen);
+        return (PTL_OK);
 }
 
-int
+ptl_err_t
 ksocknal_recv_pages (nal_cb_t *nal, void *private, lib_msg_t *msg,
-                     unsigned int niov, ptl_kiov_t *kiov, size_t mlen, size_t rlen)
+                     unsigned int niov, ptl_kiov_t *kiov, 
+                     size_t offset, size_t mlen, size_t rlen)
 {
         ksock_conn_t *conn = (ksock_conn_t *)private;
 
@@ -1611,35 +1651,43 @@ ksocknal_recv_pages (nal_cb_t *nal, void *private, lib_msg_t *msg,
 
         conn->ksnc_rx_niov = 0;
         conn->ksnc_rx_iov  = NULL;
-        conn->ksnc_rx_nkiov = niov;
         conn->ksnc_rx_kiov = conn->ksnc_rx_iov_space.kiov;
-        memcpy (conn->ksnc_rx_kiov, kiov, niov * sizeof (*kiov));
+        conn->ksnc_rx_nkiov = 
+                lib_extract_kiov(PTL_MD_MAX_IOV, conn->ksnc_rx_kiov,
+                                 niov, kiov, offset, mlen);
 
         LASSERT (mlen == 
                  lib_iov_nob (conn->ksnc_rx_niov, conn->ksnc_rx_iov) +
                  lib_kiov_nob (conn->ksnc_rx_nkiov, conn->ksnc_rx_kiov));
 
-        return (rlen);
+        return (PTL_OK);
 }
 
 int ksocknal_scheduler (void *arg)
 {
         ksock_sched_t     *sched = (ksock_sched_t *)arg;
+        ksock_conn_t      *conn;
+        ksock_tx_t        *tx;
         unsigned long      flags;
         int                rc;
         int                nloops = 0;
         int                id = sched - ksocknal_data.ksnd_schedulers;
         char               name[16];
 
-        snprintf (name, sizeof (name),"ksocknald[%d]", id);
+        snprintf (name, sizeof (name),"ksocknald_%02d", id);
         kportal_daemonize (name);
         kportal_blockallsigs ();
 
 #if (CONFIG_SMP && CPU_AFFINITY)
-        if ((cpu_online_map & (1 << id)) != 0)
+        if ((cpu_online_map & (1 << id)) != 0) {
+#if 1
                 current->cpus_allowed = (1 << id);
-        else
+#else
+                set_cpus_allowed (current, 1<<id);
+#endif
+        } else {
                 CERROR ("Can't set CPU affinity for %s\n", name);
+        }
 #endif /* CONFIG_SMP && CPU_AFFINITY */
         
         spin_lock_irqsave (&sched->kss_lock, flags);
@@ -1650,15 +1698,99 @@ int ksocknal_scheduler (void *arg)
                 /* Ensure I progress everything semi-fairly */
 
                 if (!list_empty (&sched->kss_rx_conns)) {
+                        conn = list_entry(sched->kss_rx_conns.next,
+                                          ksock_conn_t, ksnc_rx_list);
+                        list_del(&conn->ksnc_rx_list);
+
+                        LASSERT(conn->ksnc_rx_scheduled);
+                        LASSERT(conn->ksnc_rx_ready);
+
+                        /* clear rx_ready in case receive isn't complete.
+                         * Do it BEFORE we call process_recv, since
+                         * data_ready can set it any time after we release
+                         * kss_lock. */
+                        conn->ksnc_rx_ready = 0;
+                        spin_unlock_irqrestore(&sched->kss_lock, flags);
+                        
+                        rc = ksocknal_process_receive(conn);
+                        
+                        spin_lock_irqsave(&sched->kss_lock, flags);
+
+                        /* I'm the only one that can clear this flag */
+                        LASSERT(conn->ksnc_rx_scheduled);
+
+                        /* Did process_receive get everything it wanted? */
+                        if (rc == 0)
+                                conn->ksnc_rx_ready = 1;
+                        
+                        if (conn->ksnc_rx_state == SOCKNAL_RX_FMB_SLEEP ||
+                            conn->ksnc_rx_state == SOCKNAL_RX_GET_FMB) {
+                                /* Conn blocked for a forwarding buffer.
+                                 * It will get queued for my attention when
+                                 * one becomes available (and it might just
+                                 * already have been!).  Meanwhile my ref
+                                 * on it stays put. */
+                        } else if (conn->ksnc_rx_ready) {
+                                /* reschedule for rx */
+                                list_add_tail (&conn->ksnc_rx_list,
+                                               &sched->kss_rx_conns);
+                        } else {
+                                conn->ksnc_rx_scheduled = 0;
+                                /* drop my ref */
+                                ksocknal_put_conn(conn);
+                        }
+
                         did_something = 1;
-                        /* drops & regains kss_lock */
-                        ksocknal_process_receive (sched, &flags);
                 }
 
                 if (!list_empty (&sched->kss_tx_conns)) {
+                        conn = list_entry(sched->kss_tx_conns.next,
+                                          ksock_conn_t, ksnc_tx_list);
+                        list_del (&conn->ksnc_tx_list);
+                        
+                        LASSERT(conn->ksnc_tx_scheduled);
+                        LASSERT(conn->ksnc_tx_ready);
+                        LASSERT(!list_empty(&conn->ksnc_tx_queue));
+                        
+                        tx = list_entry(conn->ksnc_tx_queue.next,
+                                        ksock_tx_t, tx_list);
+                        /* dequeue now so empty list => more to send */
+                        list_del(&tx->tx_list);
+                        
+                        /* Clear tx_ready in case send isn't complete.  Do
+                         * it BEFORE we call process_transmit, since
+                         * write_space can set it any time after we release
+                         * kss_lock. */
+                        conn->ksnc_tx_ready = 0;
+                        spin_unlock_irqrestore (&sched->kss_lock, flags);
+
+                        rc = ksocknal_process_transmit(conn, tx);
+
+                        spin_lock_irqsave (&sched->kss_lock, flags);
+
+                        if (rc == -ENOMEM || rc == -EAGAIN) {
+                                /* Incomplete send: replace tx on HEAD of tx_queue */
+                                list_add (&tx->tx_list, &conn->ksnc_tx_queue);
+                        } else {
+                                /* Complete send; assume space for more */
+                                conn->ksnc_tx_ready = 1;
+                        }
+
+                        if (rc == -ENOMEM) {
+                                /* Do nothing; after a short timeout, this
+                                 * conn will be reposted on kss_tx_conns. */
+                        } else if (conn->ksnc_tx_ready &&
+                                   !list_empty (&conn->ksnc_tx_queue)) {
+                                /* reschedule for tx */
+                                list_add_tail (&conn->ksnc_tx_list, 
+                                               &sched->kss_tx_conns);
+                        } else {
+                                conn->ksnc_tx_scheduled = 0;
+                                /* drop my ref */
+                                ksocknal_put_conn (conn);
+                        }
+                                
                         did_something = 1;
-                        /* drops and regains kss_lock */
-                        ksocknal_process_transmit (sched, &flags);
                 }
 #if SOCKNAL_ZC
                 if (!list_empty (&sched->kss_zctxdone_list)) {
@@ -1719,18 +1851,14 @@ ksocknal_data_ready (struct sock *sk, int n)
         read_lock (&ksocknal_data.ksnd_global_lock);
 
         conn = sk->sk_user_data;
-        if (conn == NULL) {             /* raced with ksocknal_close_sock */
+        if (conn == NULL) {             /* raced with ksocknal_terminate_conn */
                 LASSERT (sk->sk_data_ready != &ksocknal_data_ready);
                 sk->sk_data_ready (sk, n);
-        } else if (!conn->ksnc_rx_ready) {        /* new news */
-                /* Set ASAP in case of concurrent calls to me */
-                conn->ksnc_rx_ready = 1;
-
+        } else {
                 sched = conn->ksnc_scheduler;
 
                 spin_lock_irqsave (&sched->kss_lock, flags);
 
-                /* Set again (process_receive may have cleared while I blocked for the lock) */
                 conn->ksnc_rx_ready = 1;
 
                 if (!conn->ksnc_rx_scheduled) {  /* not being progressed */
@@ -1740,8 +1868,7 @@ ksocknal_data_ready (struct sock *sk, int n)
                         /* extra ref for scheduler */
                         atomic_inc (&conn->ksnc_refcount);
 
-                        if (waitqueue_active (&sched->kss_waitq))
-                                wake_up (&sched->kss_waitq);
+                        wake_up (&sched->kss_waitq);
                 }
 
                 spin_unlock_irqrestore (&sched->kss_lock, flags);
@@ -1773,38 +1900,35 @@ ksocknal_write_space (struct sock *sk)
                (conn == NULL) ? "" : (list_empty (&conn->ksnc_tx_queue) ?
                                       " empty" : " queued"));
 
-        if (conn == NULL) {             /* raced with ksocknal_close_sock */
+        if (conn == NULL) {             /* raced with ksocknal_terminate_conn */
                 LASSERT (sk->sk_write_space != &ksocknal_write_space);
                 sk->sk_write_space (sk);
-        } else if (tcp_wspace(sk) >= SOCKNAL_TX_LOW_WATER(sk)) { /* got enough space */
-                clear_bit (SOCK_NOSPACE, &sk->sk_socket->flags);
 
-                if (!conn->ksnc_tx_ready) {      /* new news */
-                        /* Set ASAP in case of concurrent calls to me */
-                        conn->ksnc_tx_ready = 1;
+                read_unlock (&ksocknal_data.ksnd_global_lock);
+                return;
+        }
 
-                        sched = conn->ksnc_scheduler;
+        if (tcp_wspace(sk) >= SOCKNAL_TX_LOW_WATER(sk)) { /* got enough space */
+                clear_bit (SOCK_NOSPACE, &sk->sk_socket->flags);
 
-                        spin_lock_irqsave (&sched->kss_lock, flags);
+                sched = conn->ksnc_scheduler;
 
-                        /* Set again (process_transmit may have
-                           cleared while I blocked for the lock) */
-                        conn->ksnc_tx_ready = 1;
+                spin_lock_irqsave (&sched->kss_lock, flags);
 
-                        if (!conn->ksnc_tx_scheduled && // not being progressed
-                            !list_empty(&conn->ksnc_tx_queue)){//packets to send
-                                list_add_tail (&conn->ksnc_tx_list,
-                                               &sched->kss_tx_conns);
-                                conn->ksnc_tx_scheduled = 1;
-                                /* extra ref for scheduler */
-                                atomic_inc (&conn->ksnc_refcount);
+                conn->ksnc_tx_ready = 1;
 
-                                if (waitqueue_active (&sched->kss_waitq))
-                                        wake_up (&sched->kss_waitq);
-                        }
+                if (!conn->ksnc_tx_scheduled && // not being progressed
+                    !list_empty(&conn->ksnc_tx_queue)){//packets to send
+                        list_add_tail (&conn->ksnc_tx_list,
+                                       &sched->kss_tx_conns);
+                        conn->ksnc_tx_scheduled = 1;
+                        /* extra ref for scheduler */
+                        atomic_inc (&conn->ksnc_refcount);
 
-                        spin_unlock_irqrestore (&sched->kss_lock, flags);
+                        wake_up (&sched->kss_waitq);
                 }
+
+                spin_unlock_irqrestore (&sched->kss_lock, flags);
         }
 
         read_unlock (&ksocknal_data.ksnd_global_lock);
@@ -1889,106 +2013,164 @@ ksocknal_sock_read (struct socket *sock, void *buffer, int nob)
 }
 
 int
-ksocknal_exchange_nids (struct socket *sock, ptl_nid_t nid)
+ksocknal_hello (struct socket *sock, ptl_nid_t *nid, int *type,
+                __u64 *incarnation)
 {
         int                 rc;
         ptl_hdr_t           hdr;
         ptl_magicversion_t *hmv = (ptl_magicversion_t *)&hdr.dest_nid;
+        char                ipbuf[PTL_NALFMT_SIZE];
+        char                ipbuf2[PTL_NALFMT_SIZE];
 
         LASSERT (sizeof (*hmv) == sizeof (hdr.dest_nid));
 
         memset (&hdr, 0, sizeof (hdr));
         hmv->magic         = __cpu_to_le32 (PORTALS_PROTO_MAGIC);
-        hmv->version_major = __cpu_to_le32 (PORTALS_PROTO_VERSION_MAJOR);
-        hmv->version_minor = __cpu_to_le32 (PORTALS_PROTO_VERSION_MINOR);
-        
+        hmv->version_major = __cpu_to_le16 (PORTALS_PROTO_VERSION_MAJOR);
+        hmv->version_minor = __cpu_to_le16 (PORTALS_PROTO_VERSION_MINOR);
+
         hdr.src_nid = __cpu_to_le64 (ksocknal_lib.ni.nid);
         hdr.type    = __cpu_to_le32 (PTL_MSG_HELLO);
-        
+
+        hdr.msg.hello.type = __cpu_to_le32 (*type);
+        hdr.msg.hello.incarnation =
+                __cpu_to_le64 (ksocknal_data.ksnd_incarnation);
+
         /* Assume sufficient socket buffering for this message */
         rc = ksocknal_sock_write (sock, &hdr, sizeof (hdr));
         if (rc != 0) {
-                CERROR ("Error %d sending HELLO to "LPX64"\n", rc, nid);
+                CERROR ("Error %d sending HELLO to "LPX64" %s\n",
+                        rc, *nid, portals_nid2str(SOCKNAL, *nid, ipbuf));
                 return (rc);
         }
 
         rc = ksocknal_sock_read (sock, hmv, sizeof (*hmv));
         if (rc != 0) {
-                CERROR ("Error %d reading HELLO from "LPX64"\n", rc, nid);
+                CERROR ("Error %d reading HELLO from "LPX64" %s\n",
+                        rc, *nid, portals_nid2str(SOCKNAL, *nid, ipbuf));
                 return (rc);
         }
-        
+
         if (hmv->magic != __le32_to_cpu (PORTALS_PROTO_MAGIC)) {
-                CERROR ("Bad magic %#08x (%#08x expected) from "LPX64"\n",
-                        __cpu_to_le32 (hmv->magic), PORTALS_PROTO_MAGIC, nid);
-                return (-EINVAL);
+                CERROR ("Bad magic %#08x (%#08x expected) from "LPX64" %s\n",
+                        __cpu_to_le32 (hmv->magic), PORTALS_PROTO_MAGIC, *nid,
+                        portals_nid2str(SOCKNAL, *nid, ipbuf));
+                return (-EPROTO);
         }
 
         if (hmv->version_major != __cpu_to_le16 (PORTALS_PROTO_VERSION_MAJOR) ||
             hmv->version_minor != __cpu_to_le16 (PORTALS_PROTO_VERSION_MINOR)) {
                 CERROR ("Incompatible protocol version %d.%d (%d.%d expected)"
-                        " from "LPX64"\n",
+                        " from "LPX64" %s\n",
                         __le16_to_cpu (hmv->version_major),
                         __le16_to_cpu (hmv->version_minor),
                         PORTALS_PROTO_VERSION_MAJOR,
                         PORTALS_PROTO_VERSION_MINOR,
-                        nid);
-                return (-EINVAL);
+                        *nid, portals_nid2str(SOCKNAL, *nid, ipbuf));
+                return (-EPROTO);
         }
 
-        LASSERT (PORTALS_PROTO_VERSION_MAJOR == 0);
+#if (PORTALS_PROTO_VERSION_MAJOR != 0)
+# error "This code only understands protocol version 0.x"
+#endif
         /* version 0 sends magic/version as the dest_nid of a 'hello' header,
          * so read the rest of it in now... */
 
         rc = ksocknal_sock_read (sock, hmv + 1, sizeof (hdr) - sizeof (*hmv));
         if (rc != 0) {
-                CERROR ("Error %d reading rest of HELLO hdr from "LPX64"\n",
-                        rc, nid);
+                CERROR ("Error %d reading rest of HELLO hdr from "LPX64" %s\n",
+                        rc, *nid, portals_nid2str(SOCKNAL, *nid, ipbuf));
                 return (rc);
         }
 
         /* ...and check we got what we expected */
         if (hdr.type != __cpu_to_le32 (PTL_MSG_HELLO) ||
-            PTL_HDR_LENGTH (&hdr) != __cpu_to_le32 (0)) {
+            hdr.payload_length != __cpu_to_le32 (0)) {
                 CERROR ("Expecting a HELLO hdr with 0 payload,"
-                        " but got type %d with %d payload from "LPX64"\n",
+                        " but got type %d with %d payload from "LPX64" %s\n",
                         __le32_to_cpu (hdr.type),
-                        __le32_to_cpu (PTL_HDR_LENGTH (&hdr)), nid);
-                return (-EINVAL);
-        }
-        
-        if (__le64_to_cpu (hdr.src_nid) != nid) {
-                CERROR ("Connected to nid "LPX64", but expecting "LPX64"\n",
-                        __le64_to_cpu (hdr.src_nid), nid);
-                return (-EINVAL);
+                        __le32_to_cpu (hdr.payload_length), *nid,
+                        portals_nid2str(SOCKNAL, *nid, ipbuf));
+                return (-EPROTO);
+        }
+
+        if (__le64_to_cpu(hdr.src_nid) == PTL_NID_ANY) {
+                CERROR("Expecting a HELLO hdr with a NID, but got PTL_NID_ANY\n");
+                return (-EPROTO);
+        }
+
+        if (*nid == PTL_NID_ANY) {              /* don't know peer's nid yet */
+                *nid = __le64_to_cpu(hdr.src_nid);
+        } else if (*nid != __le64_to_cpu (hdr.src_nid)) {
+                CERROR ("Connected to nid "LPX64" %s, but expecting "LPX64" %s\n",
+                        __le64_to_cpu (hdr.src_nid),
+                        portals_nid2str(SOCKNAL,
+                                        __le64_to_cpu(hdr.src_nid),
+                                        ipbuf),
+                        *nid, portals_nid2str(SOCKNAL, *nid, ipbuf2));
+                return (-EPROTO);
+        }
+
+        if (*type == SOCKNAL_CONN_NONE) {
+                /* I've accepted this connection; peer determines type */
+                *type = __le32_to_cpu(hdr.msg.hello.type);
+                switch (*type) {
+                case SOCKNAL_CONN_ANY:
+                case SOCKNAL_CONN_CONTROL:
+                        break;
+                case SOCKNAL_CONN_BULK_IN:
+                        *type = SOCKNAL_CONN_BULK_OUT;
+                        break;
+                case SOCKNAL_CONN_BULK_OUT:
+                        *type = SOCKNAL_CONN_BULK_IN;
+                        break;
+                default:
+                        CERROR ("Unexpected type %d from "LPX64" %s\n",
+                                *type, *nid,
+                                portals_nid2str(SOCKNAL, *nid, ipbuf));
+                        return (-EPROTO);
+                }
+        } else if (__le32_to_cpu(hdr.msg.hello.type) != SOCKNAL_CONN_NONE) {
+                CERROR ("Mismatched types: me %d "LPX64" %s %d\n",
+                        *type, *nid, portals_nid2str(SOCKNAL, *nid, ipbuf),
+                        __le32_to_cpu(hdr.msg.hello.type));
+                return (-EPROTO);
         }
 
+        *incarnation = __le64_to_cpu(hdr.msg.hello.incarnation);
+
         return (0);
 }
 
 int
-ksocknal_set_linger (struct socket *sock) 
+ksocknal_setup_sock (struct socket *sock)
 {
         mm_segment_t    oldmm = get_fs ();
         int             rc;
         int             option;
         struct linger   linger;
 
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(2,5,0))
+        sock->sk->sk_allocation = GFP_NOFS;
+#else
+        sock->sk->allocation = GFP_NOFS;
+#endif
+
         /* Ensure this socket aborts active sends immediately when we close
          * it. */
-        
+
         linger.l_onoff = 0;
         linger.l_linger = 0;
 
         set_fs (KERNEL_DS);
-        rc = sock_setsockopt (sock, SOL_SOCKET, SO_LINGER, 
+        rc = sock_setsockopt (sock, SOL_SOCKET, SO_LINGER,
                               (char *)&linger, sizeof (linger));
         set_fs (oldmm);
         if (rc != 0) {
                 CERROR ("Can't set SO_LINGER: %d\n", rc);
                 return (rc);
         }
-        
+
         option = -1;
         set_fs (KERNEL_DS);
         rc = sock->ops->setsockopt (sock, SOL_TCP, TCP_LINGER2,
@@ -1998,20 +2180,64 @@ ksocknal_set_linger (struct socket *sock)
                 CERROR ("Can't set SO_LINGER2: %d\n", rc);
                 return (rc);
         }
+
+#if SOCKNAL_USE_KEEPALIVES
+        /* Keepalives: If 3/4 of the timeout elapses, start probing every
+         * second until the timeout elapses. */
+
+        option = (ksocknal_tunables.ksnd_io_timeout * 3) / 4;
+        set_fs (KERNEL_DS);
+        rc = sock->ops->setsockopt (sock, SOL_TCP, TCP_KEEPIDLE,
+                                    (char *)&option, sizeof (option));
+        set_fs (oldmm);
+        if (rc != 0) {
+                CERROR ("Can't set TCP_KEEPIDLE: %d\n", rc);
+                return (rc);
+        }
         
+        option = 1;
+        set_fs (KERNEL_DS);
+        rc = sock->ops->setsockopt (sock, SOL_TCP, TCP_KEEPINTVL,
+                                    (char *)&option, sizeof (option));
+        set_fs (oldmm);
+        if (rc != 0) {
+                CERROR ("Can't set TCP_KEEPINTVL: %d\n", rc);
+                return (rc);
+        }
+        
+        option = ksocknal_tunables.ksnd_io_timeout / 4;
+        set_fs (KERNEL_DS);
+        rc = sock->ops->setsockopt (sock, SOL_TCP, TCP_KEEPCNT,
+                                    (char *)&option, sizeof (option));
+        set_fs (oldmm);
+        if (rc != 0) {
+                CERROR ("Can't set TCP_KEEPINTVL: %d\n", rc);
+                return (rc);
+        }
+
+        option = 1;
+        set_fs (KERNEL_DS);
+        rc = sock_setsockopt (sock, SOL_SOCKET, SO_KEEPALIVE, 
+                              (char *)&option, sizeof (option));
+        set_fs (oldmm);
+        if (rc != 0) {
+                CERROR ("Can't set SO_KEEPALIVE: %d\n", rc);
+                return (rc);
+        }
+#endif
         return (0);
 }
 
 int
-ksocknal_connect_peer (ksock_route_t *route)
+ksocknal_connect_peer (ksock_route_t *route, int type)
 {
         struct sockaddr_in  peer_addr;
         mm_segment_t        oldmm = get_fs();
-        __u64               n;
         struct timeval      tv;
         int                 fd;
         struct socket      *sock;
         int                 rc;
+        char                ipbuf[PTL_NALFMT_SIZE];
 
         rc = sock_create (PF_INET, SOCK_STREAM, 0, &sock);
         if (rc != 0) {
@@ -2020,8 +2246,8 @@ ksocknal_connect_peer (ksock_route_t *route)
         }
 
         /* Ugh; have to map_fd for compatibility with sockets passed in
-         * from userspace.  And we actually need the refcounting that
-         * this gives you :) */
+         * from userspace.  And we actually need the sock->file refcounting
+         * that this gives you :) */
 
         fd = sock_map_fd (sock);
         if (fd < 0) {
@@ -2033,22 +2259,19 @@ ksocknal_connect_peer (ksock_route_t *route)
         /* NB the fd now owns the ref on sock->file */
         LASSERT (sock->file != NULL);
         LASSERT (file_count(sock->file) == 1);
-        
+
         /* Set the socket timeouts, so our connection attempt completes in
          * finite time */
-        tv.tv_sec = ksocknal_io_timeout / HZ;
-        n = ksocknal_io_timeout % HZ;
-        n = n * 1000000 + HZ - 1;
-        do_div (n, HZ);
-        tv.tv_usec = n;
+        tv.tv_sec = ksocknal_tunables.ksnd_io_timeout;
+        tv.tv_usec = 0;
 
         set_fs (KERNEL_DS);
         rc = sock_setsockopt (sock, SOL_SOCKET, SO_SNDTIMEO,
                               (char *)&tv, sizeof (tv));
         set_fs (oldmm);
         if (rc != 0) {
-                CERROR ("Can't set send timeout %d (in HZ): %d\n", 
-                        ksocknal_io_timeout, rc);
+                CERROR ("Can't set send timeout %d: %d\n", 
+                        ksocknal_tunables.ksnd_io_timeout, rc);
                 goto out;
         }
         
@@ -2057,12 +2280,12 @@ ksocknal_connect_peer (ksock_route_t *route)
                               (char *)&tv, sizeof (tv));
         set_fs (oldmm);
         if (rc != 0) {
-                CERROR ("Can't set receive timeout %d (in HZ): %d\n",
-                        ksocknal_io_timeout, rc);
+                CERROR ("Can't set receive timeout %d: %d\n",
+                        ksocknal_tunables.ksnd_io_timeout, rc);
                 goto out;
         }
 
-        if (route->ksnr_nonagel) {
+        {
                 int  option = 1;
                 
                 set_fs (KERNEL_DS);
@@ -2070,7 +2293,7 @@ ksocknal_connect_peer (ksock_route_t *route)
                                             (char *)&option, sizeof (option));
                 set_fs (oldmm);
                 if (rc != 0) {
-                        CERROR ("Can't disable nagel: %d\n", rc);
+                        CERROR ("Can't disable nagle: %d\n", rc);
                         goto out;
                 }
         }
@@ -2107,19 +2330,15 @@ ksocknal_connect_peer (ksock_route_t *route)
         rc = sock->ops->connect (sock, (struct sockaddr *)&peer_addr, 
                                  sizeof (peer_addr), sock->file->f_flags);
         if (rc != 0) {
-                CERROR ("Error %d connecting to "LPX64"\n", rc,
-                        route->ksnr_peer->ksnp_nid);
+                CERROR ("Error %d connecting to "LPX64" %s\n", rc,
+                        route->ksnr_peer->ksnp_nid,
+                        portals_nid2str(SOCKNAL,
+                                        route->ksnr_peer->ksnp_nid,
+                                        ipbuf));
                 goto out;
         }
         
-        if (route->ksnr_xchange_nids) {
-                rc = ksocknal_exchange_nids (sock, route->ksnr_peer->ksnp_nid);
-                if (rc != 0)
-                        goto out;
-        }
-
-        rc = ksocknal_create_conn (route->ksnr_peer->ksnp_nid,
-                                   route, sock, route->ksnr_irq_affinity);
+        rc = ksocknal_create_conn (route, sock, route->ksnr_irq_affinity, type);
         if (rc == 0) {
                 /* Take an extra ref on sock->file to compensate for the
                  * upcoming close which will lose fd's ref on it. */
@@ -2139,22 +2358,38 @@ ksocknal_autoconnect (ksock_route_t *route)
         ksock_peer_t     *peer;
         unsigned long     flags;
         int               rc;
+        int               type;
         
-        rc = ksocknal_connect_peer (route);
-        if (rc == 0) {
+        for (;;) {
+                for (type = 0; type < SOCKNAL_CONN_NTYPES; type++)
+                        if ((route->ksnr_connecting & (1 << type)) != 0)
+                                break;
+                LASSERT (type < SOCKNAL_CONN_NTYPES);
+
+                rc = ksocknal_connect_peer (route, type);
+
+                if (rc != 0)
+                        break;
+                
                 /* successfully autoconnected: create_conn did the
-                 * route/conn binding and scheduled any blocked packets, 
-                 * so there's nothing left to do now. */
-                return;
+                 * route/conn binding and scheduled any blocked packets */
+
+                if (route->ksnr_connecting == 0) {
+                        /* No more connections required */
+                        return;
+                }
         }
 
+        /* Connection attempt failed */
+
         write_lock_irqsave (&ksocknal_data.ksnd_global_lock, flags);
 
         peer = route->ksnr_peer;
         route->ksnr_connecting = 0;
 
+        /* This is a retry rather than a new connection */
         LASSERT (route->ksnr_retry_interval != 0);
-        route->ksnr_timeout = jiffies_64 + route->ksnr_retry_interval;
+        route->ksnr_timeout = jiffies + route->ksnr_retry_interval;
         route->ksnr_retry_interval = MIN (route->ksnr_retry_interval * 2,
                                           SOCKNAL_MAX_RECONNECT_INTERVAL);
 
@@ -2172,16 +2407,30 @@ ksocknal_autoconnect (ksock_route_t *route)
                 } while (!list_empty (&peer->ksnp_tx_queue));
         }
 
+        /* make this route least-favourite for re-selection */
+        if (!route->ksnr_deleted) {
+                list_del(&route->ksnr_list);
+                list_add_tail(&route->ksnr_list, &peer->ksnp_routes);
+        }
+        
         write_unlock_irqrestore (&ksocknal_data.ksnd_global_lock, flags);
 
         while (!list_empty (&zombies)) {
+                char ipbuf[PTL_NALFMT_SIZE];
+                char ipbuf2[PTL_NALFMT_SIZE];
                 tx = list_entry (zombies.next, ksock_tx_t, tx_list);
-                
-                CERROR ("Deleting packet type %d len %d ("LPX64"->"LPX64")\n",
+
+                CERROR ("Deleting packet type %d len %d ("LPX64" %s->"LPX64" %s)\n",
                         NTOH__u32 (tx->tx_hdr->type),
-                        NTOH__u32 (PTL_HDR_LENGTH(tx->tx_hdr)),
+                        NTOH__u32 (tx->tx_hdr->payload_length),
                         NTOH__u64 (tx->tx_hdr->src_nid),
-                        NTOH__u64 (tx->tx_hdr->dest_nid));
+                        portals_nid2str(SOCKNAL,
+                                        NTOH__u64(tx->tx_hdr->src_nid),
+                                        ipbuf),
+                        NTOH__u64 (tx->tx_hdr->dest_nid),
+                        portals_nid2str(SOCKNAL,
+                                        NTOH__u64(tx->tx_hdr->src_nid),
+                                        ipbuf2));
 
                 list_del (&tx->tx_list);
                 /* complete now */
@@ -2198,7 +2447,7 @@ ksocknal_autoconnectd (void *arg)
         ksock_route_t     *route;
         int                rc;
 
-        snprintf (name, sizeof (name), "ksocknal_ad[%ld]", id);
+        snprintf (name, sizeof (name), "ksocknal_ad%02ld", id);
         kportal_daemonize (name);
         kportal_blockallsigs ();
 
@@ -2239,55 +2488,48 @@ ksock_conn_t *
 ksocknal_find_timed_out_conn (ksock_peer_t *peer) 
 {
         /* We're called with a shared lock on ksnd_global_lock */
-        unsigned long      flags;
         ksock_conn_t      *conn;
         struct list_head  *ctmp;
-        ksock_tx_t        *tx;
-        struct list_head  *ttmp;
         ksock_sched_t     *sched;
 
         list_for_each (ctmp, &peer->ksnp_conns) {
                 conn = list_entry (ctmp, ksock_conn_t, ksnc_list);
                 sched = conn->ksnc_scheduler;
-                
-                if (conn->ksnc_rx_deadline != 0 &&
-                    conn->ksnc_rx_deadline <= jiffies_64)
-                        goto timed_out;
 
-                spin_lock_irqsave (&sched->kss_lock, flags);
+                /* Don't need the {get,put}connsock dance to deref ksnc_sock... */
+                LASSERT (!conn->ksnc_closing);
                 
-                list_for_each (ttmp, &conn->ksnc_tx_queue) {
-                        tx = list_entry (ttmp, ksock_tx_t, tx_list);
-                        LASSERT (tx->tx_deadline != 0);
-                        
-                        if (tx->tx_deadline <= jiffies_64)
-                                goto timed_out_locked;
+                if (conn->ksnc_rx_started &&
+                    time_after_eq (jiffies, conn->ksnc_rx_deadline)) {
+                        /* Timed out incomplete incoming message */
+                        atomic_inc (&conn->ksnc_refcount);
+                        CERROR ("Timed out RX from "LPX64" %p %d.%d.%d.%d\n",
+                                peer->ksnp_nid, conn, HIPQUAD(conn->ksnc_ipaddr));
+                        return (conn);
                 }
-#if SOCKNAL_ZC
-                list_for_each (ttmp, &conn->ksnc_tx_pending) {
-                        tx = list_entry (ttmp, ksock_tx_t, tx_list);
-                        LASSERT (tx->tx_deadline != 0);
-
-                        if (tx->tx_deadline <= jiffies_64)
-                                goto timed_out_locked;
+                
+                if ((!list_empty (&conn->ksnc_tx_queue) ||
+                     conn->ksnc_sock->sk->sk_wmem_queued != 0) &&
+                    time_after_eq (jiffies, conn->ksnc_tx_deadline)) {
+                        /* Timed out messages queued for sending, or
+                         * messages buffered in the socket's send buffer */
+                        atomic_inc (&conn->ksnc_refcount);
+                        CERROR ("Timed out TX to "LPX64" %s%d %p %d.%d.%d.%d\n", 
+                                peer->ksnp_nid, 
+                                list_empty (&conn->ksnc_tx_queue) ? "" : "Q ",
+                                conn->ksnc_sock->sk->sk_wmem_queued, conn,
+                                HIPQUAD(conn->ksnc_ipaddr));
+                        return (conn);
                 }
-#endif                
-                spin_unlock_irqrestore (&sched->kss_lock, flags);
-                continue;
-
-        timed_out_locked:
-                spin_unlock_irqrestore (&sched->kss_lock, flags);
-        timed_out:
-                atomic_inc (&conn->ksnc_refcount);
-                return (conn);
         }
 
         return (NULL);
 }
 
 void
-ksocknal_check_peer_timeouts (struct list_head *peers)
+ksocknal_check_peer_timeouts (int idx)
 {
+        struct list_head *peers = &ksocknal_data.ksnd_peers[idx];
         struct list_head *ptmp;
         ksock_peer_t     *peer;
         ksock_conn_t     *conn;
@@ -2305,13 +2547,12 @@ ksocknal_check_peer_timeouts (struct list_head *peers)
                 if (conn != NULL) {
                         read_unlock (&ksocknal_data.ksnd_global_lock);
 
-                        if (ksocknal_close_conn_unlocked (conn)) {
-                                /* I actually closed... */
-                                CERROR ("Timeout out conn->"LPX64" ip %x:%d\n",
-                                        peer->ksnp_nid, conn->ksnc_ipaddr,
-                                        conn->ksnc_port);
-                        }
-                
+                        CERROR ("Timeout out conn->"LPX64" ip %d.%d.%d.%d:%d\n",
+                                peer->ksnp_nid,
+                                HIPQUAD(conn->ksnc_ipaddr),
+                                conn->ksnc_port);
+                        ksocknal_close_conn_and_siblings (conn, -ETIMEDOUT);
+                        
                         /* NB we won't find this one again, but we can't
                          * just proceed with the next peer, since we dropped
                          * ksnd_global_lock and it might be dead already! */
@@ -2329,13 +2570,18 @@ ksocknal_reaper (void *arg)
         wait_queue_t       wait;
         unsigned long      flags;
         ksock_conn_t      *conn;
+        ksock_sched_t     *sched;
+        struct list_head   enomem_conns;
+        int                nenomem_conns;
         int                timeout;
+        int                i;
         int                peer_index = 0;
-        __u64              deadline = jiffies_64;
+        unsigned long      deadline = jiffies;
         
         kportal_daemonize ("ksocknal_reaper");
         kportal_blockallsigs ();
 
+        INIT_LIST_HEAD(&enomem_conns);
         init_waitqueue_entry (&wait, current);
 
         spin_lock_irqsave (&ksocknal_data.ksnd_reaper_lock, flags);
@@ -2368,19 +2614,72 @@ ksocknal_reaper (void *arg)
                         spin_lock_irqsave (&ksocknal_data.ksnd_reaper_lock, flags);
                         continue;
                 }
-                
+
+                if (!list_empty (&ksocknal_data.ksnd_enomem_conns)) {
+                        list_add(&enomem_conns, &ksocknal_data.ksnd_enomem_conns);
+                        list_del_init(&ksocknal_data.ksnd_enomem_conns);
+                }
+
                 spin_unlock_irqrestore (&ksocknal_data.ksnd_reaper_lock, flags);
 
-                while ((timeout = deadline - jiffies_64) <= 0) {
-                        /* Time to check for timeouts on a few more peers */
-                        ksocknal_check_peer_timeouts (&ksocknal_data.ksnd_peers[peer_index]);
+                /* reschedule all the connections that stalled with ENOMEM... */
+                nenomem_conns = 0;
+                while (!list_empty (&enomem_conns)) {
+                        conn = list_entry (enomem_conns.next,
+                                           ksock_conn_t, ksnc_tx_list);
+                        list_del (&conn->ksnc_tx_list);
+
+                        sched = conn->ksnc_scheduler;
+
+                        spin_lock_irqsave (&sched->kss_lock, flags);
+
+                        LASSERT (conn->ksnc_tx_scheduled);
+                        conn->ksnc_tx_ready = 1;
+                        list_add_tail (&conn->ksnc_tx_list, &sched->kss_tx_conns);
+                        wake_up (&sched->kss_waitq);
 
-                        peer_index = (peer_index + 1) % SOCKNAL_PEER_HASH_SIZE;
-                        deadline += HZ;
+                        spin_unlock_irqrestore (&sched->kss_lock, flags);
+                        nenomem_conns++;
                 }
+                
+                /* careful with the jiffy wrap... */
+                while ((timeout = (int)(deadline - jiffies)) <= 0) {
+                        const int n = 4;
+                        const int p = 1;
+                        int       chunk = ksocknal_data.ksnd_peer_hash_size;
+                        
+                        /* Time to check for timeouts on a few more peers: I do
+                         * checks every 'p' seconds on a proportion of the peer
+                         * table and I need to check every connection 'n' times
+                         * within a timeout interval, to ensure I detect a
+                         * timeout on any connection within (n+1)/n times the
+                         * timeout interval. */
+
+                        if (ksocknal_tunables.ksnd_io_timeout > n * p)
+                                chunk = (chunk * n * p) / 
+                                        ksocknal_tunables.ksnd_io_timeout;
+                        if (chunk == 0)
+                                chunk = 1;
+
+                        for (i = 0; i < chunk; i++) {
+                                ksocknal_check_peer_timeouts (peer_index);
+                                peer_index = (peer_index + 1) % 
+                                             ksocknal_data.ksnd_peer_hash_size;
+                        }
+
+                        deadline += p * HZ;
+                }
+
+                if (nenomem_conns != 0) {
+                        /* Reduce my timeout if I rescheduled ENOMEM conns.
+                         * This also prevents me getting woken immediately
+                         * if any go back on my enomem list. */
+                        timeout = SOCKNAL_ENOMEM_RETRY;
+                }
+                ksocknal_data.ksnd_reaper_waketime = jiffies + timeout;
 
-                add_wait_queue (&ksocknal_data.ksnd_reaper_waitq, &wait);
                 set_current_state (TASK_INTERRUPTIBLE);
+                add_wait_queue (&ksocknal_data.ksnd_reaper_waitq, &wait);
 
                 if (!ksocknal_data.ksnd_shuttingdown &&
                     list_empty (&ksocknal_data.ksnd_deathrow_conns) &&
@@ -2407,11 +2706,11 @@ nal_cb_t ksocknal_lib = {
         cb_recv_pages:   ksocknal_recv_pages,
         cb_read:         ksocknal_read,
         cb_write:        ksocknal_write,
-        cb_callback:     ksocknal_callback,
         cb_malloc:       ksocknal_malloc,
         cb_free:         ksocknal_free,
         cb_printf:       ksocknal_printf,
         cb_cli:          ksocknal_cli,
         cb_sti:          ksocknal_sti,
+        cb_callback:     ksocknal_callback,
         cb_dist:         ksocknal_dist
 };