lnet/klnds/qswlnd/qswlnd_cb.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * Copyright (C) 2002 Cluster File Systems, Inc.
   5  *   Author: Eric Barton <eric@bartonsoftware.com>
   6  *
   7  * Copyright (C) 2002, Lawrence Livermore National Labs (LLNL)
   8  * W. Marcus Miller - Based on ksocknal
   9  *
  10  * This file is part of Portals, http://www.sf.net/projects/sandiaportals/
  11  *
  12  * Portals is free software; you can redistribute it and/or
  13  * modify it under the terms of version 2 of the GNU General Public
  14  * License as published by the Free Software Foundation.
  15  *
  16  * Portals is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with Portals; if not, write to the Free Software
  23  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  24  *
  25  */
  26
  27 #include "qswnal.h"
  28
  29 /*
  30  *  LIB functions follow
  31  *
  32  */
  33 static int
  34 kqswnal_dist(lib_nal_t *nal, ptl_nid_t nid, unsigned long *dist)
  35 {
  36         if (nid == nal->libnal_ni.ni_pid.nid)
  37                 *dist = 0;                      /* it's me */
  38         else if (kqswnal_nid2elanid (nid) >= 0)
  39                 *dist = 1;                      /* it's my peer */
  40         else
  41                 *dist = 2;                      /* via router */
  42         return (0);
  43 }
  44
  45 void
  46 kqswnal_notify_peer_down(kqswnal_tx_t *ktx)
  47 {
  48         struct timeval     now;
  49         time_t             then;
  50
  51         do_gettimeofday (&now);
  52         then = now.tv_sec - (jiffies - ktx->ktx_launchtime)/HZ;
  53
  54         kpr_notify(&kqswnal_data.kqn_router, ktx->ktx_nid, 0, then);
  55 }
  56
  57 void
  58 kqswnal_unmap_tx (kqswnal_tx_t *ktx)
  59 {
  60 #if MULTIRAIL_EKC
  61         int      i;
  62
  63         ktx->ktx_rail = -1;                     /* unset rail */
  64 #endif
  65
  66         if (ktx->ktx_nmappedpages == 0)
  67                 return;
  68
  69 #if MULTIRAIL_EKC
  70         CDEBUG(D_NET, "%p unloading %d frags starting at %d\n",
  71                ktx, ktx->ktx_nfrag, ktx->ktx_firsttmpfrag);
  72
  73         for (i = ktx->ktx_firsttmpfrag; i < ktx->ktx_nfrag; i++)
  74                 ep_dvma_unload(kqswnal_data.kqn_ep,
  75                                kqswnal_data.kqn_ep_tx_nmh,
  76                                &ktx->ktx_frags[i]);
  77 #else
  78         CDEBUG (D_NET, "%p[%d] unloading pages %d for %d\n",
  79                 ktx, ktx->ktx_nfrag, ktx->ktx_basepage, ktx->ktx_nmappedpages);
  80
  81         LASSERT (ktx->ktx_nmappedpages <= ktx->ktx_npages);
  82         LASSERT (ktx->ktx_basepage + ktx->ktx_nmappedpages <=
  83                  kqswnal_data.kqn_eptxdmahandle->NumDvmaPages);
  84
  85         elan3_dvma_unload(kqswnal_data.kqn_ep->DmaState,
  86                           kqswnal_data.kqn_eptxdmahandle,
  87                           ktx->ktx_basepage, ktx->ktx_nmappedpages);
  88 #endif
  89         ktx->ktx_nmappedpages = 0;
  90 }
  91
  92 int
  93 kqswnal_map_tx_kiov (kqswnal_tx_t *ktx, int offset, int nob, int niov, ptl_kiov_t *kiov)
  94 {
  95         int       nfrags    = ktx->ktx_nfrag;
  96         int       nmapped   = ktx->ktx_nmappedpages;
  97         int       maxmapped = ktx->ktx_npages;
  98         uint32_t  basepage  = ktx->ktx_basepage + nmapped;
  99         char     *ptr;
 100 #if MULTIRAIL_EKC
 101         EP_RAILMASK railmask;
 102         int         rail;
 103
 104         if (ktx->ktx_rail < 0)
 105                 ktx->ktx_rail = ep_xmtr_prefrail(kqswnal_data.kqn_eptx,
 106                                                  EP_RAILMASK_ALL,
 107                                                  kqswnal_nid2elanid(ktx->ktx_nid));
 108         rail = ktx->ktx_rail;
 109         if (rail < 0) {
 110                 CERROR("No rails available for "LPX64"\n", ktx->ktx_nid);
 111                 return (-ENETDOWN);
 112         }
 113         railmask = 1 << rail;
 114 #endif
 115         LASSERT (nmapped <= maxmapped);
 116         LASSERT (nfrags >= ktx->ktx_firsttmpfrag);
 117         LASSERT (nfrags <= EP_MAXFRAG);
 118         LASSERT (niov > 0);
 119         LASSERT (nob > 0);
 120
 121         /* skip complete frags before 'offset' */
 122         while (offset >= kiov->kiov_len) {
 123                 offset -= kiov->kiov_len;
 124                 kiov++;
 125                 niov--;
 126                 LASSERT (niov > 0);
 127         }
 128
 129         do {
 130                 int  fraglen = kiov->kiov_len - offset;
 131
 132                 /* each page frag is contained in one page */
 133                 LASSERT (kiov->kiov_offset + kiov->kiov_len <= PAGE_SIZE);
 134
 135                 if (fraglen > nob)
 136                         fraglen = nob;
 137
 138                 nmapped++;
 139                 if (nmapped > maxmapped) {
 140                         CERROR("Can't map message in %d pages (max %d)\n",
 141                                nmapped, maxmapped);
 142                         return (-EMSGSIZE);
 143                 }
 144
 145                 if (nfrags == EP_MAXFRAG) {
 146                         CERROR("Message too fragmented in Elan VM (max %d frags)\n",
 147                                EP_MAXFRAG);
 148                         return (-EMSGSIZE);
 149                 }
 150
 151                 /* XXX this is really crap, but we'll have to kmap until
 152                  * EKC has a page (rather than vaddr) mapping interface */
 153
 154                 ptr = ((char *)kmap (kiov->kiov_page)) + kiov->kiov_offset + offset;
 155
 156                 CDEBUG(D_NET,
 157                        "%p[%d] loading %p for %d, page %d, %d total\n",
 158                         ktx, nfrags, ptr, fraglen, basepage, nmapped);
 159
 160 #if MULTIRAIL_EKC
 161                 ep_dvma_load(kqswnal_data.kqn_ep, NULL,
 162                              ptr, fraglen,
 163                              kqswnal_data.kqn_ep_tx_nmh, basepage,
 164                              &railmask, &ktx->ktx_frags[nfrags]);
 165
 166                 if (nfrags == ktx->ktx_firsttmpfrag ||
 167                     !ep_nmd_merge(&ktx->ktx_frags[nfrags - 1],
 168                                   &ktx->ktx_frags[nfrags - 1],
 169                                   &ktx->ktx_frags[nfrags])) {
 170                         /* new frag if this is the first or can't merge */
 171                         nfrags++;
 172                 }
 173 #else
 174                 elan3_dvma_kaddr_load (kqswnal_data.kqn_ep->DmaState,
 175                                        kqswnal_data.kqn_eptxdmahandle,
 176                                        ptr, fraglen,
 177                                        basepage, &ktx->ktx_frags[nfrags].Base);
 178
 179                 if (nfrags > 0 &&                /* previous frag mapped */
 180                     ktx->ktx_frags[nfrags].Base == /* contiguous with this one */
 181                     (ktx->ktx_frags[nfrags-1].Base + ktx->ktx_frags[nfrags-1].Len))
 182                         /* just extend previous */
 183                         ktx->ktx_frags[nfrags - 1].Len += fraglen;
 184                 else {
 185                         ktx->ktx_frags[nfrags].Len = fraglen;
 186                         nfrags++;                /* new frag */
 187                 }
 188 #endif
 189
 190                 kunmap (kiov->kiov_page);
 191
 192                 /* keep in loop for failure case */
 193                 ktx->ktx_nmappedpages = nmapped;
 194
 195                 basepage++;
 196                 kiov++;
 197                 niov--;
 198                 nob -= fraglen;
 199                 offset = 0;
 200
 201                 /* iov must not run out before end of data */
 202                 LASSERT (nob == 0 || niov > 0);
 203
 204         } while (nob > 0);
 205
 206         ktx->ktx_nfrag = nfrags;
 207         CDEBUG (D_NET, "%p got %d frags over %d pages\n",
 208                 ktx, ktx->ktx_nfrag, ktx->ktx_nmappedpages);
 209
 210         return (0);
 211 }
 212
 213 int
 214 kqswnal_map_tx_iov (kqswnal_tx_t *ktx, int offset, int nob,
 215                     int niov, struct iovec *iov)
 216 {
 217         int       nfrags    = ktx->ktx_nfrag;
 218         int       nmapped   = ktx->ktx_nmappedpages;
 219         int       maxmapped = ktx->ktx_npages;
 220         uint32_t  basepage  = ktx->ktx_basepage + nmapped;
 221 #if MULTIRAIL_EKC
 222         EP_RAILMASK railmask;
 223         int         rail;
 224
 225         if (ktx->ktx_rail < 0)
 226                 ktx->ktx_rail = ep_xmtr_prefrail(kqswnal_data.kqn_eptx,
 227                                                  EP_RAILMASK_ALL,
 228                                                  kqswnal_nid2elanid(ktx->ktx_nid));
 229         rail = ktx->ktx_rail;
 230         if (rail < 0) {
 231                 CERROR("No rails available for "LPX64"\n", ktx->ktx_nid);
 232                 return (-ENETDOWN);
 233         }
 234         railmask = 1 << rail;
 235 #endif
 236         LASSERT (nmapped <= maxmapped);
 237         LASSERT (nfrags >= ktx->ktx_firsttmpfrag);
 238         LASSERT (nfrags <= EP_MAXFRAG);
 239         LASSERT (niov > 0);
 240         LASSERT (nob > 0);
 241
 242         /* skip complete frags before offset */
 243         while (offset >= iov->iov_len) {
 244                 offset -= iov->iov_len;
 245                 iov++;
 246                 niov--;
 247                 LASSERT (niov > 0);
 248         }
 249
 250         do {
 251                 int  fraglen = iov->iov_len - offset;
 252                 long npages;
 253
 254                 if (fraglen > nob)
 255                         fraglen = nob;
 256                 npages = kqswnal_pages_spanned (iov->iov_base, fraglen);
 257
 258                 nmapped += npages;
 259                 if (nmapped > maxmapped) {
 260                         CERROR("Can't map message in %d pages (max %d)\n",
 261                                nmapped, maxmapped);
 262                         return (-EMSGSIZE);
 263                 }
 264
 265                 if (nfrags == EP_MAXFRAG) {
 266                         CERROR("Message too fragmented in Elan VM (max %d frags)\n",
 267                                EP_MAXFRAG);
 268                         return (-EMSGSIZE);
 269                 }
 270
 271                 CDEBUG(D_NET,
 272                        "%p[%d] loading %p for %d, pages %d for %ld, %d total\n",
 273                        ktx, nfrags, iov->iov_base + offset, fraglen,
 274                        basepage, npages, nmapped);
 275
 276 #if MULTIRAIL_EKC
 277                 ep_dvma_load(kqswnal_data.kqn_ep, NULL,
 278                              iov->iov_base + offset, fraglen,
 279                              kqswnal_data.kqn_ep_tx_nmh, basepage,
 280                              &railmask, &ktx->ktx_frags[nfrags]);
 281
 282                 if (nfrags == ktx->ktx_firsttmpfrag ||
 283                     !ep_nmd_merge(&ktx->ktx_frags[nfrags - 1],
 284                                   &ktx->ktx_frags[nfrags - 1],
 285                                   &ktx->ktx_frags[nfrags])) {
 286                         /* new frag if this is the first or can't merge */
 287                         nfrags++;
 288                 }
 289 #else
 290                 elan3_dvma_kaddr_load (kqswnal_data.kqn_ep->DmaState,
 291                                        kqswnal_data.kqn_eptxdmahandle,
 292                                        iov->iov_base + offset, fraglen,
 293                                        basepage, &ktx->ktx_frags[nfrags].Base);
 294
 295                 if (nfrags > 0 &&                /* previous frag mapped */
 296                     ktx->ktx_frags[nfrags].Base == /* contiguous with this one */
 297                     (ktx->ktx_frags[nfrags-1].Base + ktx->ktx_frags[nfrags-1].Len))
 298                         /* just extend previous */
 299                         ktx->ktx_frags[nfrags - 1].Len += fraglen;
 300                 else {
 301                         ktx->ktx_frags[nfrags].Len = fraglen;
 302                         nfrags++;                /* new frag */
 303                 }
 304 #endif
 305
 306                 /* keep in loop for failure case */
 307                 ktx->ktx_nmappedpages = nmapped;
 308
 309                 basepage += npages;
 310                 iov++;
 311                 niov--;
 312                 nob -= fraglen;
 313                 offset = 0;
 314
 315                 /* iov must not run out before end of data */
 316                 LASSERT (nob == 0 || niov > 0);
 317
 318         } while (nob > 0);
 319
 320         ktx->ktx_nfrag = nfrags;
 321         CDEBUG (D_NET, "%p got %d frags over %d pages\n",
 322                 ktx, ktx->ktx_nfrag, ktx->ktx_nmappedpages);
 323
 324         return (0);
 325 }
 326
 327
 328 void
 329 kqswnal_put_idle_tx (kqswnal_tx_t *ktx)
 330 {
 331         kpr_fwd_desc_t   *fwd = NULL;
 332         unsigned long     flags;
 333
 334         kqswnal_unmap_tx (ktx);                 /* release temporary mappings */
 335         ktx->ktx_state = KTX_IDLE;
 336
 337         spin_lock_irqsave (&kqswnal_data.kqn_idletxd_lock, flags);
 338
 339         list_del (&ktx->ktx_list);              /* take off active list */
 340
 341         if (ktx->ktx_isnblk) {
 342                 /* reserved for non-blocking tx */
 343                 list_add (&ktx->ktx_list, &kqswnal_data.kqn_nblk_idletxds);
 344                 spin_unlock_irqrestore (&kqswnal_data.kqn_idletxd_lock, flags);
 345                 return;
 346         }
 347
 348         list_add (&ktx->ktx_list, &kqswnal_data.kqn_idletxds);
 349
 350         /* anything blocking for a tx descriptor? */
 351         if (!kqswnal_data.kqn_shuttingdown &&
 352             !list_empty(&kqswnal_data.kqn_idletxd_fwdq)) /* forwarded packet? */
 353         {
 354                 CDEBUG(D_NET,"wakeup fwd\n");
 355
 356                 fwd = list_entry (kqswnal_data.kqn_idletxd_fwdq.next,
 357                                   kpr_fwd_desc_t, kprfd_list);
 358                 list_del (&fwd->kprfd_list);
 359         }
 360
 361         wake_up (&kqswnal_data.kqn_idletxd_waitq);
 362
 363         spin_unlock_irqrestore (&kqswnal_data.kqn_idletxd_lock, flags);
 364
 365         if (fwd == NULL)
 366                 return;
 367
 368         /* schedule packet for forwarding again */
 369         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
 370
 371         list_add_tail (&fwd->kprfd_list, &kqswnal_data.kqn_delayedfwds);
 372         wake_up (&kqswnal_data.kqn_sched_waitq);
 373
 374         spin_unlock_irqrestore (&kqswnal_data.kqn_sched_lock, flags);
 375 }
 376
 377 kqswnal_tx_t *
 378 kqswnal_get_idle_tx (kpr_fwd_desc_t *fwd, int may_block)
 379 {
 380         unsigned long  flags;
 381         kqswnal_tx_t  *ktx = NULL;
 382
 383         for (;;) {
 384                 spin_lock_irqsave (&kqswnal_data.kqn_idletxd_lock, flags);
 385
 386                 if (kqswnal_data.kqn_shuttingdown)
 387                         break;
 388
 389                 /* "normal" descriptor is free */
 390                 if (!list_empty (&kqswnal_data.kqn_idletxds)) {
 391                         ktx = list_entry (kqswnal_data.kqn_idletxds.next,
 392                                           kqswnal_tx_t, ktx_list);
 393                         break;
 394                 }
 395
 396                 if (fwd != NULL)                /* forwarded packet? */
 397                         break;
 398
 399                 /* doing a local transmit */
 400                 if (!may_block) {
 401                         if (list_empty (&kqswnal_data.kqn_nblk_idletxds)) {
 402                                 CERROR ("intr tx desc pool exhausted\n");
 403                                 break;
 404                         }
 405
 406                         ktx = list_entry (kqswnal_data.kqn_nblk_idletxds.next,
 407                                           kqswnal_tx_t, ktx_list);
 408                         break;
 409                 }
 410
 411                 /* block for idle tx */
 412
 413                 spin_unlock_irqrestore (&kqswnal_data.kqn_idletxd_lock, flags);
 414
 415                 CDEBUG (D_NET, "blocking for tx desc\n");
 416                 wait_event (kqswnal_data.kqn_idletxd_waitq,
 417                             !list_empty (&kqswnal_data.kqn_idletxds) ||
 418                             kqswnal_data.kqn_shuttingdown);
 419         }
 420
 421         if (ktx != NULL) {
 422                 list_del (&ktx->ktx_list);
 423                 list_add (&ktx->ktx_list, &kqswnal_data.kqn_activetxds);
 424                 ktx->ktx_launcher = current->pid;
 425                 atomic_inc(&kqswnal_data.kqn_pending_txs);
 426         } else if (fwd != NULL) {
 427                 /* queue forwarded packet until idle txd available */
 428                 CDEBUG (D_NET, "blocked fwd [%p]\n", fwd);
 429                 list_add_tail (&fwd->kprfd_list,
 430                                &kqswnal_data.kqn_idletxd_fwdq);
 431         }
 432
 433         spin_unlock_irqrestore (&kqswnal_data.kqn_idletxd_lock, flags);
 434
 435         /* Idle descs can't have any mapped (as opposed to pre-mapped) pages */
 436         LASSERT (ktx == NULL || ktx->ktx_nmappedpages == 0);
 437
 438         return (ktx);
 439 }
 440
 441 void
 442 kqswnal_tx_done (kqswnal_tx_t *ktx, int error)
 443 {
 444         switch (ktx->ktx_state) {
 445         case KTX_FORWARDING:       /* router asked me to forward this packet */
 446                 kpr_fwd_done (&kqswnal_data.kqn_router,
 447                               (kpr_fwd_desc_t *)ktx->ktx_args[0], error);
 448                 break;
 449
 450         case KTX_RDMAING:          /* optimized GET/PUT handled */
 451         case KTX_PUTTING:          /* optimized PUT sent */
 452         case KTX_SENDING:          /* normal send */
 453                 lib_finalize (&kqswnal_lib, NULL,
 454                               (lib_msg_t *)ktx->ktx_args[1],
 455                               (error == 0) ? PTL_OK : PTL_FAIL);
 456                 break;
 457
 458         case KTX_GETTING:          /* optimized GET sent & REPLY received */
 459                 /* Complete the GET with success since we can't avoid
 460                  * delivering a REPLY event; we committed to it when we
 461                  * launched the GET */
 462                 lib_finalize (&kqswnal_lib, NULL,
 463                               (lib_msg_t *)ktx->ktx_args[1], PTL_OK);
 464                 lib_finalize (&kqswnal_lib, NULL,
 465                               (lib_msg_t *)ktx->ktx_args[2],
 466                               (error == 0) ? PTL_OK : PTL_FAIL);
 467                 break;
 468
 469         default:
 470                 LASSERT (0);
 471         }
 472
 473         kqswnal_put_idle_tx (ktx);
 474 }
 475
 476 static void
 477 kqswnal_txhandler(EP_TXD *txd, void *arg, int status)
 478 {
 479         kqswnal_tx_t      *ktx = (kqswnal_tx_t *)arg;
 480
 481         LASSERT (txd != NULL);
 482         LASSERT (ktx != NULL);
 483
 484         CDEBUG(D_NET, "txd %p, arg %p status %d\n", txd, arg, status);
 485
 486         if (status != EP_SUCCESS) {
 487
 488                 CERROR ("Tx completion to "LPX64" failed: %d\n",
 489                         ktx->ktx_nid, status);
 490
 491                 kqswnal_notify_peer_down(ktx);
 492                 status = -EHOSTDOWN;
 493
 494         } else switch (ktx->ktx_state) {
 495
 496         case KTX_GETTING:
 497         case KTX_PUTTING:
 498                 /* RPC completed OK; but what did our peer put in the status
 499                  * block? */
 500 #if MULTIRAIL_EKC
 501                 status = ep_txd_statusblk(txd)->Data[0];
 502 #else
 503                 status = ep_txd_statusblk(txd)->Status;
 504 #endif
 505                 break;
 506
 507         case KTX_FORWARDING:
 508         case KTX_SENDING:
 509                 status = 0;
 510                 break;
 511
 512         default:
 513                 LBUG();
 514                 break;
 515         }
 516
 517         kqswnal_tx_done (ktx, status);
 518 }
 519
 520 int
 521 kqswnal_launch (kqswnal_tx_t *ktx)
 522 {
 523         /* Don't block for transmit descriptor if we're in interrupt context */
 524         int   attr = in_interrupt() ? (EP_NO_SLEEP | EP_NO_ALLOC) : 0;
 525         int   dest = kqswnal_nid2elanid (ktx->ktx_nid);
 526         unsigned long flags;
 527         int   rc;
 528
 529         ktx->ktx_launchtime = jiffies;
 530
 531         if (kqswnal_data.kqn_shuttingdown)
 532                 return (-ESHUTDOWN);
 533
 534         LASSERT (dest >= 0);                    /* must be a peer */
 535
 536 #if MULTIRAIL_EKC
 537         if (ktx->ktx_nmappedpages != 0)
 538                 attr = EP_SET_PREFRAIL(attr, ktx->ktx_rail);
 539 #endif
 540
 541         switch (ktx->ktx_state) {
 542         case KTX_GETTING:
 543         case KTX_PUTTING:
 544                 /* NB ktx_frag[0] is the GET/PUT hdr + kqswnal_remotemd_t.
 545                  * The other frags are the payload, awaiting RDMA */
 546                 rc = ep_transmit_rpc(kqswnal_data.kqn_eptx, dest,
 547                                      ktx->ktx_port, attr,
 548                                      kqswnal_txhandler, ktx,
 549                                      NULL, ktx->ktx_frags, 1);
 550                 break;
 551
 552         case KTX_FORWARDING:
 553         case KTX_SENDING:
 554 #if MULTIRAIL_EKC
 555                 rc = ep_transmit_message(kqswnal_data.kqn_eptx, dest,
 556                                          ktx->ktx_port, attr,
 557                                          kqswnal_txhandler, ktx,
 558                                          NULL, ktx->ktx_frags, ktx->ktx_nfrag);
 559 #else
 560                 rc = ep_transmit_large(kqswnal_data.kqn_eptx, dest,
 561                                        ktx->ktx_port, attr,
 562                                        kqswnal_txhandler, ktx,
 563                                        ktx->ktx_frags, ktx->ktx_nfrag);
 564 #endif
 565                 break;
 566
 567         default:
 568                 LBUG();
 569                 rc = -EINVAL;                   /* no compiler warning please */
 570                 break;
 571         }
 572
 573         switch (rc) {
 574         case EP_SUCCESS: /* success */
 575                 return (0);
 576
 577         case EP_ENOMEM: /* can't allocate ep txd => queue for later */
 578                 spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
 579
 580                 list_add_tail (&ktx->ktx_delayed_list, &kqswnal_data.kqn_delayedtxds);
 581                 wake_up (&kqswnal_data.kqn_sched_waitq);
 582
 583                 spin_unlock_irqrestore (&kqswnal_data.kqn_sched_lock, flags);
 584                 return (0);
 585
 586         default: /* fatal error */
 587                 CERROR ("Tx to "LPX64" failed: %d\n", ktx->ktx_nid, rc);
 588                 kqswnal_notify_peer_down(ktx);
 589                 return (-EHOSTUNREACH);
 590         }
 591 }
 592
 593 #if 0
 594 static char *
 595 hdr_type_string (ptl_hdr_t *hdr)
 596 {
 597         switch (hdr->type) {
 598         case PTL_MSG_ACK:
 599                 return ("ACK");
 600         case PTL_MSG_PUT:
 601                 return ("PUT");
 602         case PTL_MSG_GET:
 603                 return ("GET");
 604         case PTL_MSG_REPLY:
 605                 return ("REPLY");
 606         default:
 607                 return ("<UNKNOWN>");
 608         }
 609 }
 610
 611 static void
 612 kqswnal_cerror_hdr(ptl_hdr_t * hdr)
 613 {
 614         char *type_str = hdr_type_string (hdr);
 615
 616         CERROR("P3 Header at %p of type %s length %d\n", hdr, type_str,
 617                le32_to_cpu(hdr->payload_length));
 618         CERROR("    From nid/pid "LPU64"/%u\n", le64_to_cpu(hdr->src_nid),
 619                le32_to_cpu(hdr->src_pid));
 620         CERROR("    To nid/pid "LPU64"/%u\n", le64_to_cpu(hdr->dest_nid),
 621                le32_to_cpu(hdr->dest_pid));
 622
 623         switch (le32_to_cpu(hdr->type)) {
 624         case PTL_MSG_PUT:
 625                 CERROR("    Ptl index %d, ack md "LPX64"."LPX64", "
 626                        "match bits "LPX64"\n",
 627                        le32_to_cpu(hdr->msg.put.ptl_index),
 628                        hdr->msg.put.ack_wmd.wh_interface_cookie,
 629                        hdr->msg.put.ack_wmd.wh_object_cookie,
 630                        le64_to_cpu(hdr->msg.put.match_bits));
 631                 CERROR("    offset %d, hdr data "LPX64"\n",
 632                        le32_to_cpu(hdr->msg.put.offset),
 633                        hdr->msg.put.hdr_data);
 634                 break;
 635
 636         case PTL_MSG_GET:
 637                 CERROR("    Ptl index %d, return md "LPX64"."LPX64", "
 638                        "match bits "LPX64"\n",
 639                        le32_to_cpu(hdr->msg.get.ptl_index),
 640                        hdr->msg.get.return_wmd.wh_interface_cookie,
 641                        hdr->msg.get.return_wmd.wh_object_cookie,
 642                        hdr->msg.get.match_bits);
 643                 CERROR("    Length %d, src offset %d\n",
 644                        le32_to_cpu(hdr->msg.get.sink_length),
 645                        le32_to_cpu(hdr->msg.get.src_offset));
 646                 break;
 647
 648         case PTL_MSG_ACK:
 649                 CERROR("    dst md "LPX64"."LPX64", manipulated length %d\n",
 650                        hdr->msg.ack.dst_wmd.wh_interface_cookie,
 651                        hdr->msg.ack.dst_wmd.wh_object_cookie,
 652                        le32_to_cpu(hdr->msg.ack.mlength));
 653                 break;
 654
 655         case PTL_MSG_REPLY:
 656                 CERROR("    dst md "LPX64"."LPX64"\n",
 657                        hdr->msg.reply.dst_wmd.wh_interface_cookie,
 658                        hdr->msg.reply.dst_wmd.wh_object_cookie);
 659         }
 660
 661 }                               /* end of print_hdr() */
 662 #endif
 663
 664 #if !MULTIRAIL_EKC
 665 void
 666 kqswnal_print_eiov (int how, char *str, int n, EP_IOVEC *iov)
 667 {
 668         int          i;
 669
 670         CDEBUG (how, "%s: %d\n", str, n);
 671         for (i = 0; i < n; i++) {
 672                 CDEBUG (how, "   %08x for %d\n", iov[i].Base, iov[i].Len);
 673         }
 674 }
 675
 676 int
 677 kqswnal_eiovs2datav (int ndv, EP_DATAVEC *dv,
 678                      int nsrc, EP_IOVEC *src,
 679                      int ndst, EP_IOVEC *dst)
 680 {
 681         int        count;
 682         int        nob;
 683
 684         LASSERT (ndv > 0);
 685         LASSERT (nsrc > 0);
 686         LASSERT (ndst > 0);
 687
 688         for (count = 0; count < ndv; count++, dv++) {
 689
 690                 if (nsrc == 0 || ndst == 0) {
 691                         if (nsrc != ndst) {
 692                                 /* For now I'll barf on any left over entries */
 693                                 CERROR ("mismatched src and dst iovs\n");
 694                                 return (-EINVAL);
 695                         }
 696                         return (count);
 697                 }
 698
 699                 nob = (src->Len < dst->Len) ? src->Len : dst->Len;
 700                 dv->Len    = nob;
 701                 dv->Source = src->Base;
 702                 dv->Dest   = dst->Base;
 703
 704                 if (nob >= src->Len) {
 705                         src++;
 706                         nsrc--;
 707                 } else {
 708                         src->Len -= nob;
 709                         src->Base += nob;
 710                 }
 711
 712                 if (nob >= dst->Len) {
 713                         dst++;
 714                         ndst--;
 715                 } else {
 716                         src->Len -= nob;
 717                         src->Base += nob;
 718                 }
 719         }
 720
 721         CERROR ("DATAVEC too small\n");
 722         return (-E2BIG);
 723 }
 724 #else
 725 int
 726 kqswnal_check_rdma (int nlfrag, EP_NMD *lfrag,
 727                     int nrfrag, EP_NMD *rfrag)
 728 {
 729         int  i;
 730
 731         if (nlfrag != nrfrag) {
 732                 CERROR("Can't cope with unequal # frags: %d local %d remote\n",
 733                        nlfrag, nrfrag);
 734                 return (-EINVAL);
 735         }
 736
 737         for (i = 0; i < nlfrag; i++)
 738                 if (lfrag[i].nmd_len != rfrag[i].nmd_len) {
 739                         CERROR("Can't cope with unequal frags %d(%d):"
 740                                " %d local %d remote\n",
 741                                i, nlfrag, lfrag[i].nmd_len, rfrag[i].nmd_len);
 742                         return (-EINVAL);
 743                 }
 744
 745         return (0);
 746 }
 747 #endif
 748
 749 kqswnal_remotemd_t *
 750 kqswnal_parse_rmd (kqswnal_rx_t *krx, int type, ptl_nid_t expected_nid)
 751 {
 752         char               *buffer = (char *)page_address(krx->krx_kiov[0].kiov_page);
 753         ptl_hdr_t          *hdr = (ptl_hdr_t *)buffer;
 754         kqswnal_remotemd_t *rmd = (kqswnal_remotemd_t *)(buffer + KQSW_HDR_SIZE);
 755         ptl_nid_t           nid = kqswnal_rx_nid(krx);
 756
 757         /* Note (1) lib_parse has already flipped hdr.
 758          *      (2) RDMA addresses are sent in native endian-ness.  When
 759          *      EKC copes with different endian nodes, I'll fix this (and
 760          *      eat my hat :) */
 761
 762         LASSERT (krx->krx_nob >= sizeof(*hdr));
 763
 764         if (hdr->type != type) {
 765                 CERROR ("Unexpected optimized get/put type %d (%d expected)"
 766                         "from "LPX64"\n", hdr->type, type, nid);
 767                 return (NULL);
 768         }
 769
 770         if (hdr->src_nid != nid) {
 771                 CERROR ("Unexpected optimized get/put source NID "
 772                         LPX64" from "LPX64"\n", hdr->src_nid, nid);
 773                 return (NULL);
 774         }
 775
 776         LASSERT (nid == expected_nid);
 777
 778         if (buffer + krx->krx_nob < (char *)(rmd + 1)) {
 779                 /* msg too small to discover rmd size */
 780                 CERROR ("Incoming message [%d] too small for RMD (%d needed)\n",
 781                         krx->krx_nob, (int)(((char *)(rmd + 1)) - buffer));
 782                 return (NULL);
 783         }
 784
 785         if (buffer + krx->krx_nob < (char *)&rmd->kqrmd_frag[rmd->kqrmd_nfrag]) {
 786                 /* rmd doesn't fit in the incoming message */
 787                 CERROR ("Incoming message [%d] too small for RMD[%d] (%d needed)\n",
 788                         krx->krx_nob, rmd->kqrmd_nfrag,
 789                         (int)(((char *)&rmd->kqrmd_frag[rmd->kqrmd_nfrag]) - buffer));
 790                 return (NULL);
 791         }
 792
 793         return (rmd);
 794 }
 795
 796 void
 797 kqswnal_rdma_store_complete (EP_RXD *rxd)
 798 {
 799         int           status = ep_rxd_status(rxd);
 800         kqswnal_tx_t *ktx = (kqswnal_tx_t *)ep_rxd_arg(rxd);
 801         kqswnal_rx_t *krx = (kqswnal_rx_t *)ktx->ktx_args[0];
 802
 803         CDEBUG((status == EP_SUCCESS) ? D_NET : D_ERROR,
 804                "rxd %p, ktx %p, status %d\n", rxd, ktx, status);
 805
 806         LASSERT (ktx->ktx_state == KTX_RDMAING);
 807         LASSERT (krx->krx_rxd == rxd);
 808         LASSERT (krx->krx_rpc_reply_needed);
 809
 810         krx->krx_rpc_reply_needed = 0;
 811         kqswnal_rx_decref (krx);
 812
 813         /* free ktx & finalize() its lib_msg_t */
 814         kqswnal_tx_done(ktx, (status == EP_SUCCESS) ? 0 : -ECONNABORTED);
 815 }
 816
 817 void
 818 kqswnal_rdma_fetch_complete (EP_RXD *rxd)
 819 {
 820         /* Completed fetching the PUT data */
 821         int           status = ep_rxd_status(rxd);
 822         kqswnal_tx_t *ktx = (kqswnal_tx_t *)ep_rxd_arg(rxd);
 823         kqswnal_rx_t *krx = (kqswnal_rx_t *)ktx->ktx_args[0];
 824         unsigned long flags;
 825
 826         CDEBUG((status == EP_SUCCESS) ? D_NET : D_ERROR,
 827                "rxd %p, ktx %p, status %d\n", rxd, ktx, status);
 828
 829         LASSERT (ktx->ktx_state == KTX_RDMAING);
 830         LASSERT (krx->krx_rxd == rxd);
 831         LASSERT (krx->krx_rpc_reply_needed);
 832
 833         /* Set the RPC completion status */
 834         status = (status == EP_SUCCESS) ? 0 : -ECONNABORTED;
 835         krx->krx_rpc_reply_status = status;
 836
 837         /* free ktx & finalize() its lib_msg_t */
 838         kqswnal_tx_done(ktx, status);
 839
 840         if (!in_interrupt()) {
 841                 /* OK to complete the RPC now (iff I had the last ref) */
 842                 kqswnal_rx_decref (krx);
 843                 return;
 844         }
 845
 846         LASSERT (krx->krx_state == KRX_PARSE);
 847         krx->krx_state = KRX_COMPLETING;
 848
 849         /* Complete the RPC in thread context */
 850         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
 851
 852         list_add_tail (&krx->krx_list, &kqswnal_data.kqn_readyrxds);
 853         wake_up (&kqswnal_data.kqn_sched_waitq);
 854
 855         spin_unlock_irqrestore (&kqswnal_data.kqn_sched_lock, flags);
 856 }
 857
 858 int
 859 kqswnal_rdma (kqswnal_rx_t *krx, lib_msg_t *libmsg, int type,
 860               int niov, struct iovec *iov, ptl_kiov_t *kiov,
 861               size_t offset, size_t len)
 862 {
 863         kqswnal_remotemd_t *rmd;
 864         kqswnal_tx_t       *ktx;
 865         int                 eprc;
 866         int                 rc;
 867 #if !MULTIRAIL_EKC
 868         EP_DATAVEC          datav[EP_MAXFRAG];
 869         int                 ndatav;
 870 #endif
 871
 872         LASSERT (type == PTL_MSG_GET || type == PTL_MSG_PUT);
 873         /* Not both mapped and paged payload */
 874         LASSERT (iov == NULL || kiov == NULL);
 875         /* RPC completes with failure by default */
 876         LASSERT (krx->krx_rpc_reply_needed);
 877         LASSERT (krx->krx_rpc_reply_status != 0);
 878
 879         rmd = kqswnal_parse_rmd(krx, type, libmsg->ev.initiator.nid);
 880         if (rmd == NULL)
 881                 return (-EPROTO);
 882
 883         if (len == 0) {
 884                 /* data got truncated to nothing. */
 885                 lib_finalize(&kqswnal_lib, krx, libmsg, PTL_OK);
 886                 /* Let kqswnal_rx_done() complete the RPC with success */
 887                 krx->krx_rpc_reply_status = 0;
 888                 return (0);
 889         }
 890
 891         /* NB I'm using 'ktx' just to map the local RDMA buffers; I'm not
 892            actually sending a portals message with it */
 893         ktx = kqswnal_get_idle_tx(NULL, 0);
 894         if (ktx == NULL) {
 895                 CERROR ("Can't get txd for RDMA with "LPX64"\n",
 896                         libmsg->ev.initiator.nid);
 897                 return (-ENOMEM);
 898         }
 899
 900         ktx->ktx_state   = KTX_RDMAING;
 901         ktx->ktx_nid     = libmsg->ev.initiator.nid;
 902         ktx->ktx_args[0] = krx;
 903         ktx->ktx_args[1] = libmsg;
 904
 905 #if MULTIRAIL_EKC
 906         /* Map on the rail the RPC prefers */
 907         ktx->ktx_rail = ep_rcvr_prefrail(krx->krx_eprx,
 908                                          ep_rxd_railmask(krx->krx_rxd));
 909 #endif
 910
 911         /* Start mapping at offset 0 (we're not mapping any headers) */
 912         ktx->ktx_nfrag = ktx->ktx_firsttmpfrag = 0;
 913
 914         if (kiov != NULL)
 915                 rc = kqswnal_map_tx_kiov(ktx, offset, len, niov, kiov);
 916         else
 917                 rc = kqswnal_map_tx_iov(ktx, offset, len, niov, iov);
 918
 919         if (rc != 0) {
 920                 CERROR ("Can't map local RDMA data: %d\n", rc);
 921                 goto out;
 922         }
 923
 924 #if MULTIRAIL_EKC
 925         rc = kqswnal_check_rdma (ktx->ktx_nfrag, ktx->ktx_frags,
 926                                  rmd->kqrmd_nfrag, rmd->kqrmd_frag);
 927         if (rc != 0) {
 928                 CERROR ("Incompatible RDMA descriptors\n");
 929                 goto out;
 930         }
 931 #else
 932         switch (type) {
 933         default:
 934                 LBUG();
 935
 936         case PTL_MSG_GET:
 937                 ndatav = kqswnal_eiovs2datav(EP_MAXFRAG, datav,
 938                                              ktx->ktx_nfrag, ktx->ktx_frags,
 939                                              rmd->kqrmd_nfrag, rmd->kqrmd_frag);
 940                 break;
 941
 942         case PTL_MSG_PUT:
 943                 ndatav = kqswnal_eiovs2datav(EP_MAXFRAG, datav,
 944                                              rmd->kqrmd_nfrag, rmd->kqrmd_frag,
 945                                              ktx->ktx_nfrag, ktx->ktx_frags);
 946                 break;
 947         }
 948
 949         if (ndatav < 0) {
 950                 CERROR ("Can't create datavec: %d\n", ndatav);
 951                 rc = ndatav;
 952                 goto out;
 953         }
 954 #endif
 955
 956         LASSERT (atomic_read(&krx->krx_refcount) > 0);
 957         /* Take an extra ref for the completion callback */
 958         atomic_inc(&krx->krx_refcount);
 959
 960         switch (type) {
 961         default:
 962                 LBUG();
 963
 964         case PTL_MSG_GET:
 965 #if MULTIRAIL_EKC
 966                 eprc = ep_complete_rpc(krx->krx_rxd,
 967                                        kqswnal_rdma_store_complete, ktx,
 968                                        &kqswnal_data.kqn_rpc_success,
 969                                        ktx->ktx_frags, rmd->kqrmd_frag, rmd->kqrmd_nfrag);
 970 #else
 971                 eprc = ep_complete_rpc (krx->krx_rxd,
 972                                         kqswnal_rdma_store_complete, ktx,
 973                                         &kqswnal_data.kqn_rpc_success,
 974                                         datav, ndatav);
 975                 if (eprc != EP_SUCCESS) /* "old" EKC destroys rxd on failed completion */
 976                         krx->krx_rxd = NULL;
 977 #endif
 978                 if (eprc != EP_SUCCESS) {
 979                         CERROR("can't complete RPC: %d\n", eprc);
 980                         /* don't re-attempt RPC completion */
 981                         krx->krx_rpc_reply_needed = 0;
 982                         rc = -ECONNABORTED;
 983                 }
 984                 break;
 985
 986         case PTL_MSG_PUT:
 987 #if MULTIRAIL_EKC
 988                 eprc = ep_rpc_get (krx->krx_rxd,
 989                                    kqswnal_rdma_fetch_complete, ktx,
 990                                    rmd->kqrmd_frag, ktx->ktx_frags, ktx->ktx_nfrag);
 991 #else
 992                 eprc = ep_rpc_get (krx->krx_rxd,
 993                                    kqswnal_rdma_fetch_complete, ktx,
 994                                    datav, ndatav);
 995 #endif
 996                 if (eprc != EP_SUCCESS) {
 997                         CERROR("ep_rpc_get failed: %d\n", eprc);
 998                         rc = -ECONNABORTED;
 999                 }
1000                 break;
1001         }
1002
1003  out:
1004         if (rc != 0) {
1005                 kqswnal_rx_decref(krx);                 /* drop callback's ref */
1006                 kqswnal_put_idle_tx (ktx);
1007         }
1008
1009         atomic_dec(&kqswnal_data.kqn_pending_txs);
1010         return (rc);
1011 }
1012
1013 static ptl_err_t
1014 kqswnal_sendmsg (lib_nal_t    *nal,
1015                  void         *private,
1016                  lib_msg_t    *libmsg,
1017                  ptl_hdr_t    *hdr,
1018                  int           type,
1019                  ptl_nid_t     nid,
1020                  ptl_pid_t     pid,
1021                  unsigned int  payload_niov,
1022                  struct iovec *payload_iov,
1023                  ptl_kiov_t   *payload_kiov,
1024                  size_t        payload_offset,
1025                  size_t        payload_nob)
1026 {
1027         kqswnal_tx_t      *ktx;
1028         int                rc;
1029         ptl_nid_t          targetnid;
1030 #if KQSW_CHECKSUM
1031         int                i;
1032         kqsw_csum_t        csum;
1033         int                sumoff;
1034         int                sumnob;
1035 #endif
1036         /* NB 1. hdr is in network byte order */
1037         /*    2. 'private' depends on the message type */
1038
1039         CDEBUG(D_NET, "sending "LPSZ" bytes in %d frags to nid: "LPX64
1040                " pid %u\n", payload_nob, payload_niov, nid, pid);
1041
1042         LASSERT (payload_nob == 0 || payload_niov > 0);
1043         LASSERT (payload_niov <= PTL_MD_MAX_IOV);
1044
1045         /* It must be OK to kmap() if required */
1046         LASSERT (payload_kiov == NULL || !in_interrupt ());
1047         /* payload is either all vaddrs or all pages */
1048         LASSERT (!(payload_kiov != NULL && payload_iov != NULL));
1049
1050         if (payload_nob > KQSW_MAXPAYLOAD) {
1051                 CERROR ("request exceeds MTU size "LPSZ" (max %u).\n",
1052                         payload_nob, KQSW_MAXPAYLOAD);
1053                 return (PTL_FAIL);
1054         }
1055
1056         if (type == PTL_MSG_REPLY &&            /* can I look in 'private' */
1057             ((kqswnal_rx_t *)private)->krx_rpc_reply_needed) { /* is it an RPC */
1058                 /* Must be a REPLY for an optimized GET */
1059                 rc = kqswnal_rdma ((kqswnal_rx_t *)private, libmsg, PTL_MSG_GET,
1060                                    payload_niov, payload_iov, payload_kiov,
1061                                    payload_offset, payload_nob);
1062                 return ((rc == 0) ? PTL_OK : PTL_FAIL);
1063         }
1064
1065         targetnid = nid;
1066         if (kqswnal_nid2elanid (nid) < 0) {     /* Can't send direct: find gateway? */
1067                 rc = kpr_lookup (&kqswnal_data.kqn_router, nid,
1068                                  sizeof (ptl_hdr_t) + payload_nob, &targetnid);
1069                 if (rc != 0) {
1070                         CERROR("Can't route to "LPX64": router error %d\n",
1071                                nid, rc);
1072                         return (PTL_FAIL);
1073                 }
1074                 if (kqswnal_nid2elanid (targetnid) < 0) {
1075                         CERROR("Bad gateway "LPX64" for "LPX64"\n",
1076                                targetnid, nid);
1077                         return (PTL_FAIL);
1078                 }
1079         }
1080
1081         /* I may not block for a transmit descriptor if I might block the
1082          * receiver, or an interrupt handler. */
1083         ktx = kqswnal_get_idle_tx(NULL, !(type == PTL_MSG_ACK ||
1084                                           type == PTL_MSG_REPLY ||
1085                                           in_interrupt()));
1086         if (ktx == NULL) {
1087                 CERROR ("Can't get txd for msg type %d for "LPX64"\n",
1088                         type, libmsg->ev.initiator.nid);
1089                 return (PTL_NO_SPACE);
1090         }
1091
1092         ktx->ktx_state   = KTX_SENDING;
1093         ktx->ktx_nid     = targetnid;
1094         ktx->ktx_args[0] = private;
1095         ktx->ktx_args[1] = libmsg;
1096         ktx->ktx_args[2] = NULL;    /* set when a GET commits to REPLY */
1097
1098         memcpy (ktx->ktx_buffer, hdr, sizeof (*hdr)); /* copy hdr from caller's stack */
1099
1100 #if KQSW_CHECKSUM
1101         csum = kqsw_csum (0, (char *)hdr, sizeof (*hdr));
1102         memcpy (ktx->ktx_buffer + sizeof (*hdr), &csum, sizeof (csum));
1103         for (csum = 0, i = 0, sumoff = payload_offset, sumnob = payload_nob; sumnob > 0; i++) {
1104                 LASSERT(i < niov);
1105                 if (payload_kiov != NULL) {
1106                         ptl_kiov_t *kiov = &payload_kiov[i];
1107
1108                         if (sumoff >= kiov->kiov_len) {
1109                                 sumoff -= kiov->kiov_len;
1110                         } else {
1111                                 char *addr = ((char *)kmap (kiov->kiov_page)) +
1112                                              kiov->kiov_offset + sumoff;
1113                                 int   fragnob = kiov->kiov_len - sumoff;
1114
1115                                 csum = kqsw_csum(csum, addr, MIN(sumnob, fragnob));
1116                                 sumnob -= fragnob;
1117                                 sumoff = 0;
1118                                 kunmap(kiov->kiov_page);
1119                         }
1120                 } else {
1121                         struct iovec *iov = &payload_iov[i];
1122
1123                         if (sumoff > iov->iov_len) {
1124                                 sumoff -= iov->iov_len;
1125                         } else {
1126                                 char *addr = iov->iov_base + sumoff;
1127                                 int   fragnob = iov->iov_len - sumoff;
1128
1129                                 csum = kqsw_csum(csum, addr, MIN(sumnob, fragnob));
1130                                 sumnob -= fragnob;
1131                                 sumoff = 0;
1132                         }
1133                 }
1134         }
1135         memcpy(ktx->ktx_buffer + sizeof(*hdr) + sizeof(csum), &csum, sizeof(csum));
1136 #endif
1137
1138         /* The first frag will be the pre-mapped buffer for (at least) the
1139          * portals header. */
1140         ktx->ktx_nfrag = ktx->ktx_firsttmpfrag = 1;
1141
1142         if (nid == targetnid &&                 /* not forwarding */
1143             ((type == PTL_MSG_GET &&            /* optimize GET? */
1144               kqswnal_tunables.kqn_optimized_gets != 0 &&
1145               le32_to_cpu(hdr->msg.get.sink_length) >= kqswnal_tunables.kqn_optimized_gets) ||
1146              (type == PTL_MSG_PUT &&            /* optimize PUT? */
1147               kqswnal_tunables.kqn_optimized_puts != 0 &&
1148               payload_nob >= kqswnal_tunables.kqn_optimized_puts))) {
1149                 lib_md_t           *md = libmsg->md;
1150                 kqswnal_remotemd_t *rmd = (kqswnal_remotemd_t *)(ktx->ktx_buffer + KQSW_HDR_SIZE);
1151
1152                 /* Optimised path: I send over the Elan vaddrs of the local
1153                  * buffers, and my peer DMAs directly to/from them.
1154                  *
1155                  * First I set up ktx as if it was going to send this
1156                  * payload, (it needs to map it anyway).  This fills
1157                  * ktx_frags[1] and onward with the network addresses
1158                  * of the GET sink frags.  I copy these into ktx_buffer,
1159                  * immediately after the header, and send that as my
1160                  * message. */
1161
1162                 ktx->ktx_state = (type == PTL_MSG_PUT) ? KTX_PUTTING : KTX_GETTING;
1163
1164                 if ((libmsg->md->options & PTL_MD_KIOV) != 0)
1165                         rc = kqswnal_map_tx_kiov (ktx, 0, md->length,
1166                                                   md->md_niov, md->md_iov.kiov);
1167                 else
1168                         rc = kqswnal_map_tx_iov (ktx, 0, md->length,
1169                                                  md->md_niov, md->md_iov.iov);
1170                 if (rc != 0)
1171                         goto out;
1172
1173                 rmd->kqrmd_nfrag = ktx->ktx_nfrag - 1;
1174
1175                 payload_nob = offsetof(kqswnal_remotemd_t,
1176                                        kqrmd_frag[rmd->kqrmd_nfrag]);
1177                 LASSERT (KQSW_HDR_SIZE + payload_nob <= KQSW_TX_BUFFER_SIZE);
1178
1179 #if MULTIRAIL_EKC
1180                 memcpy(&rmd->kqrmd_frag[0], &ktx->ktx_frags[1],
1181                        rmd->kqrmd_nfrag * sizeof(EP_NMD));
1182
1183                 ep_nmd_subset(&ktx->ktx_frags[0], &ktx->ktx_ebuffer,
1184                               0, KQSW_HDR_SIZE + payload_nob);
1185 #else
1186                 memcpy(&rmd->kqrmd_frag[0], &ktx->ktx_frags[1],
1187                        rmd->kqrmd_nfrag * sizeof(EP_IOVEC));
1188
1189                 ktx->ktx_frags[0].Base = ktx->ktx_ebuffer;
1190                 ktx->ktx_frags[0].Len = KQSW_HDR_SIZE + payload_nob;
1191 #endif
1192                 if (type == PTL_MSG_GET) {
1193                         /* Allocate reply message now while I'm in thread context */
1194                         ktx->ktx_args[2] = lib_create_reply_msg (&kqswnal_lib,
1195                                                                  nid, libmsg);
1196                         if (ktx->ktx_args[2] == NULL)
1197                                 goto out;
1198
1199                         /* NB finalizing the REPLY message is my
1200                          * responsibility now, whatever happens. */
1201                 }
1202
1203         } else if (payload_nob <= KQSW_TX_MAXCONTIG) {
1204
1205                 /* small message: single frag copied into the pre-mapped buffer */
1206
1207 #if MULTIRAIL_EKC
1208                 ep_nmd_subset(&ktx->ktx_frags[0], &ktx->ktx_ebuffer,
1209                               0, KQSW_HDR_SIZE + payload_nob);
1210 #else
1211                 ktx->ktx_frags[0].Base = ktx->ktx_ebuffer;
1212                 ktx->ktx_frags[0].Len = KQSW_HDR_SIZE + payload_nob;
1213 #endif
1214                 if (payload_nob > 0) {
1215                         if (payload_kiov != NULL)
1216                                 lib_copy_kiov2buf (ktx->ktx_buffer + KQSW_HDR_SIZE,
1217                                                    payload_niov, payload_kiov,
1218                                                    payload_offset, payload_nob);
1219                         else
1220                                 lib_copy_iov2buf (ktx->ktx_buffer + KQSW_HDR_SIZE,
1221                                                   payload_niov, payload_iov,
1222                                                   payload_offset, payload_nob);
1223                 }
1224         } else {
1225
1226                 /* large message: multiple frags: first is hdr in pre-mapped buffer */
1227
1228 #if MULTIRAIL_EKC
1229                 ep_nmd_subset(&ktx->ktx_frags[0], &ktx->ktx_ebuffer,
1230                               0, KQSW_HDR_SIZE);
1231 #else
1232                 ktx->ktx_frags[0].Base = ktx->ktx_ebuffer;
1233                 ktx->ktx_frags[0].Len = KQSW_HDR_SIZE;
1234 #endif
1235                 if (payload_kiov != NULL)
1236                         rc = kqswnal_map_tx_kiov (ktx, payload_offset, payload_nob,
1237                                                   payload_niov, payload_kiov);
1238                 else
1239                         rc = kqswnal_map_tx_iov (ktx, payload_offset, payload_nob,
1240                                                  payload_niov, payload_iov);
1241                 if (rc != 0)
1242                         goto out;
1243         }
1244
1245         ktx->ktx_port = (payload_nob <= KQSW_SMALLPAYLOAD) ?
1246                         EP_MSG_SVC_PORTALS_SMALL : EP_MSG_SVC_PORTALS_LARGE;
1247
1248         rc = kqswnal_launch (ktx);
1249
1250  out:
1251         CDEBUG(rc == 0 ? D_NET : D_ERROR,
1252                "%s "LPSZ" bytes to "LPX64" via "LPX64": rc %d\n",
1253                rc == 0 ? "Sent" : "Failed to send",
1254                payload_nob, nid, targetnid, rc);
1255
1256         if (rc != 0) {
1257                 if (ktx->ktx_state == KTX_GETTING &&
1258                     ktx->ktx_args[2] != NULL) {
1259                         /* We committed to reply, but there was a problem
1260                          * launching the GET.  We can't avoid delivering a
1261                          * REPLY event since we committed above, so we
1262                          * pretend the GET succeeded but the REPLY
1263                          * failed. */
1264                         rc = 0;
1265                         lib_finalize (&kqswnal_lib, private, libmsg, PTL_OK);
1266                         lib_finalize (&kqswnal_lib, private,
1267                                       (lib_msg_t *)ktx->ktx_args[2], PTL_FAIL);
1268                 }
1269
1270                 kqswnal_put_idle_tx (ktx);
1271         }
1272
1273         atomic_dec(&kqswnal_data.kqn_pending_txs);
1274         return (rc == 0 ? PTL_OK : PTL_FAIL);
1275 }
1276
1277 static ptl_err_t
1278 kqswnal_send (lib_nal_t    *nal,
1279               void         *private,
1280               lib_msg_t    *libmsg,
1281               ptl_hdr_t    *hdr,
1282               int           type,
1283               ptl_nid_t     nid,
1284               ptl_pid_t     pid,
1285               unsigned int  payload_niov,
1286               struct iovec *payload_iov,
1287               size_t        payload_offset,
1288               size_t        payload_nob)
1289 {
1290         return (kqswnal_sendmsg (nal, private, libmsg, hdr, type, nid, pid,
1291                                  payload_niov, payload_iov, NULL,
1292                                  payload_offset, payload_nob));
1293 }
1294
1295 static ptl_err_t
1296 kqswnal_send_pages (lib_nal_t    *nal,
1297                     void         *private,
1298                     lib_msg_t    *libmsg,
1299                     ptl_hdr_t    *hdr,
1300                     int           type,
1301                     ptl_nid_t     nid,
1302                     ptl_pid_t     pid,
1303                     unsigned int  payload_niov,
1304                     ptl_kiov_t   *payload_kiov,
1305                     size_t        payload_offset,
1306                     size_t        payload_nob)
1307 {
1308         return (kqswnal_sendmsg (nal, private, libmsg, hdr, type, nid, pid,
1309                                  payload_niov, NULL, payload_kiov,
1310                                  payload_offset, payload_nob));
1311 }
1312
1313 void
1314 kqswnal_fwd_packet (void *arg, kpr_fwd_desc_t *fwd)
1315 {
1316         int             rc;
1317         kqswnal_tx_t   *ktx;
1318         ptl_kiov_t     *kiov = fwd->kprfd_kiov;
1319         int             niov = fwd->kprfd_niov;
1320         int             nob = fwd->kprfd_nob;
1321         ptl_nid_t       nid = fwd->kprfd_gateway_nid;
1322
1323 #if KQSW_CHECKSUM
1324         CERROR ("checksums for forwarded packets not implemented\n");
1325         LBUG ();
1326 #endif
1327         /* The router wants this NAL to forward a packet */
1328         CDEBUG (D_NET, "forwarding [%p] to "LPX64", payload: %d frags %d bytes\n",
1329                 fwd, nid, niov, nob);
1330
1331         ktx = kqswnal_get_idle_tx (fwd, 0);
1332         if (ktx == NULL)        /* can't get txd right now */
1333                 return;         /* fwd will be scheduled when tx desc freed */
1334
1335         if (nid == kqswnal_lib.libnal_ni.ni_pid.nid) /* gateway is me */
1336                 nid = fwd->kprfd_target_nid;    /* target is final dest */
1337
1338         if (kqswnal_nid2elanid (nid) < 0) {
1339                 CERROR("Can't forward [%p] to "LPX64": not a peer\n", fwd, nid);
1340                 rc = -EHOSTUNREACH;
1341                 goto out;
1342         }
1343
1344         /* copy hdr into pre-mapped buffer */
1345         memcpy(ktx->ktx_buffer, fwd->kprfd_hdr, sizeof(ptl_hdr_t));
1346
1347         ktx->ktx_port    = (nob <= KQSW_SMALLPAYLOAD) ?
1348                            EP_MSG_SVC_PORTALS_SMALL : EP_MSG_SVC_PORTALS_LARGE;
1349         ktx->ktx_nid     = nid;
1350         ktx->ktx_state   = KTX_FORWARDING;
1351         ktx->ktx_args[0] = fwd;
1352         ktx->ktx_nfrag   = ktx->ktx_firsttmpfrag = 1;
1353
1354         if (nob <= KQSW_TX_MAXCONTIG)
1355         {
1356                 /* send payload from ktx's pre-mapped contiguous buffer */
1357 #if MULTIRAIL_EKC
1358                 ep_nmd_subset(&ktx->ktx_frags[0], &ktx->ktx_ebuffer,
1359                               0, KQSW_HDR_SIZE + nob);
1360 #else
1361                 ktx->ktx_frags[0].Base = ktx->ktx_ebuffer;
1362                 ktx->ktx_frags[0].Len = KQSW_HDR_SIZE + nob;
1363 #endif
1364                 if (nob > 0)
1365                         lib_copy_kiov2buf(ktx->ktx_buffer + KQSW_HDR_SIZE,
1366                                           niov, kiov, 0, nob);
1367         }
1368         else
1369         {
1370                 /* zero copy payload */
1371 #if MULTIRAIL_EKC
1372                 ep_nmd_subset(&ktx->ktx_frags[0], &ktx->ktx_ebuffer,
1373                               0, KQSW_HDR_SIZE);
1374 #else
1375                 ktx->ktx_frags[0].Base = ktx->ktx_ebuffer;
1376                 ktx->ktx_frags[0].Len = KQSW_HDR_SIZE;
1377 #endif
1378                 rc = kqswnal_map_tx_kiov (ktx, 0, nob, niov, kiov);
1379                 if (rc != 0)
1380                         goto out;
1381         }
1382
1383         rc = kqswnal_launch (ktx);
1384  out:
1385         if (rc != 0) {
1386                 CERROR ("Failed to forward [%p] to "LPX64": %d\n", fwd, nid, rc);
1387
1388                 /* complete now (with failure) */
1389                 kqswnal_tx_done (ktx, rc);
1390         }
1391
1392         atomic_dec(&kqswnal_data.kqn_pending_txs);
1393 }
1394
1395 void
1396 kqswnal_fwd_callback (void *arg, int error)
1397 {
1398         kqswnal_rx_t *krx = (kqswnal_rx_t *)arg;
1399
1400         /* The router has finished forwarding this packet */
1401
1402         if (error != 0)
1403         {
1404                 ptl_hdr_t *hdr = (ptl_hdr_t *)page_address (krx->krx_kiov[0].kiov_page);
1405
1406                 CERROR("Failed to route packet from "LPX64" to "LPX64": %d\n",
1407                        le64_to_cpu(hdr->src_nid), le64_to_cpu(hdr->dest_nid),error);
1408         }
1409
1410         LASSERT (atomic_read(&krx->krx_refcount) == 1);
1411         kqswnal_rx_decref (krx);
1412 }
1413
1414 void
1415 kqswnal_requeue_rx (kqswnal_rx_t *krx)
1416 {
1417         LASSERT (atomic_read(&krx->krx_refcount) == 0);
1418         LASSERT (!krx->krx_rpc_reply_needed);
1419
1420         krx->krx_state = KRX_POSTED;
1421
1422 #if MULTIRAIL_EKC
1423         if (kqswnal_data.kqn_shuttingdown) {
1424                 /* free EKC rxd on shutdown */
1425                 ep_complete_receive(krx->krx_rxd);
1426         } else {
1427                 /* repost receive */
1428                 ep_requeue_receive(krx->krx_rxd,
1429                                    kqswnal_rxhandler, krx,
1430                                    &krx->krx_elanbuffer, 0);
1431         }
1432 #else
1433         if (kqswnal_data.kqn_shuttingdown)
1434                 return;
1435
1436         if (krx->krx_rxd == NULL) {
1437                 /* We had a failed ep_complete_rpc() which nukes the
1438                  * descriptor in "old" EKC */
1439                 int eprc = ep_queue_receive(krx->krx_eprx,
1440                                             kqswnal_rxhandler, krx,
1441                                             krx->krx_elanbuffer,
1442                                             krx->krx_npages * PAGE_SIZE, 0);
1443                 LASSERT (eprc == EP_SUCCESS);
1444                 /* We don't handle failure here; it's incredibly rare
1445                  * (never reported?) and only happens with "old" EKC */
1446         } else {
1447                 ep_requeue_receive(krx->krx_rxd, kqswnal_rxhandler, krx,
1448                                    krx->krx_elanbuffer,
1449                                    krx->krx_npages * PAGE_SIZE);
1450         }
1451 #endif
1452 }
1453
1454 void
1455 kqswnal_rpc_complete (EP_RXD *rxd)
1456 {
1457         int           status = ep_rxd_status(rxd);
1458         kqswnal_rx_t *krx    = (kqswnal_rx_t *)ep_rxd_arg(rxd);
1459
1460         CDEBUG((status == EP_SUCCESS) ? D_NET : D_ERROR,
1461                "rxd %p, krx %p, status %d\n", rxd, krx, status);
1462
1463         LASSERT (krx->krx_rxd == rxd);
1464         LASSERT (krx->krx_rpc_reply_needed);
1465
1466         krx->krx_rpc_reply_needed = 0;
1467         kqswnal_requeue_rx (krx);
1468 }
1469
1470 void
1471 kqswnal_rx_done (kqswnal_rx_t *krx)
1472 {
1473         int           rc;
1474         EP_STATUSBLK *sblk;
1475
1476         LASSERT (atomic_read(&krx->krx_refcount) == 0);
1477
1478         if (krx->krx_rpc_reply_needed) {
1479                 /* We've not completed the peer's RPC yet... */
1480                 sblk = (krx->krx_rpc_reply_status == 0) ?
1481                        &kqswnal_data.kqn_rpc_success :
1482                        &kqswnal_data.kqn_rpc_failed;
1483
1484                 LASSERT (!in_interrupt());
1485 #if MULTIRAIL_EKC
1486                 rc = ep_complete_rpc(krx->krx_rxd,
1487                                      kqswnal_rpc_complete, krx,
1488                                      sblk, NULL, NULL, 0);
1489                 if (rc == EP_SUCCESS)
1490                         return;
1491 #else
1492                 rc = ep_complete_rpc(krx->krx_rxd,
1493                                      kqswnal_rpc_complete, krx,
1494                                      sblk, NULL, 0);
1495                 if (rc == EP_SUCCESS)
1496                         return;
1497
1498                 /* "old" EKC destroys rxd on failed completion */
1499                 krx->krx_rxd = NULL;
1500 #endif
1501                 CERROR("can't complete RPC: %d\n", rc);
1502                 krx->krx_rpc_reply_needed = 0;
1503         }
1504
1505         kqswnal_requeue_rx(krx);
1506 }
1507
1508 void
1509 kqswnal_parse (kqswnal_rx_t *krx)
1510 {
1511         ptl_hdr_t      *hdr = (ptl_hdr_t *) page_address(krx->krx_kiov[0].kiov_page);
1512         ptl_nid_t       dest_nid = le64_to_cpu(hdr->dest_nid);
1513         int             payload_nob;
1514         int             nob;
1515         int             niov;
1516
1517         LASSERT (atomic_read(&krx->krx_refcount) == 1);
1518
1519         if (dest_nid == kqswnal_lib.libnal_ni.ni_pid.nid) { /* It's for me :) */
1520                 /* I ignore parse errors since I'm not consuming a byte
1521                  * stream */
1522                 (void)lib_parse (&kqswnal_lib, hdr, krx);
1523
1524                 /* Drop my ref; any RDMA activity takes an additional ref */
1525                 kqswnal_rx_decref(krx);
1526                 return;
1527         }
1528
1529 #if KQSW_CHECKSUM
1530         LASSERTF (0, "checksums for forwarded packets not implemented\n");
1531 #endif
1532
1533         if (kqswnal_nid2elanid (dest_nid) >= 0)  /* should have gone direct to peer */
1534         {
1535                 CERROR("dropping packet from "LPX64" for "LPX64
1536                        ": target is peer\n", le64_to_cpu(hdr->src_nid), dest_nid);
1537
1538                 kqswnal_rx_decref (krx);
1539                 return;
1540         }
1541
1542         nob = payload_nob = krx->krx_nob - KQSW_HDR_SIZE;
1543         niov = 0;
1544         if (nob > 0) {
1545                 krx->krx_kiov[0].kiov_offset = KQSW_HDR_SIZE;
1546                 krx->krx_kiov[0].kiov_len = MIN(PAGE_SIZE - KQSW_HDR_SIZE, nob);
1547                 niov = 1;
1548                 nob -= PAGE_SIZE - KQSW_HDR_SIZE;
1549
1550                 while (nob > 0) {
1551                         LASSERT (niov < krx->krx_npages);
1552
1553                         krx->krx_kiov[niov].kiov_offset = 0;
1554                         krx->krx_kiov[niov].kiov_len = MIN(PAGE_SIZE, nob);
1555                         niov++;
1556                         nob -= PAGE_SIZE;
1557                 }
1558         }
1559
1560         kpr_fwd_init (&krx->krx_fwd, dest_nid,
1561                       hdr, payload_nob, niov, krx->krx_kiov,
1562                       kqswnal_fwd_callback, krx);
1563
1564         kpr_fwd_start (&kqswnal_data.kqn_router, &krx->krx_fwd);
1565 }
1566
1567 /* Receive Interrupt Handler: posts to schedulers */
1568 void
1569 kqswnal_rxhandler(EP_RXD *rxd)
1570 {
1571         unsigned long flags;
1572         int           nob    = ep_rxd_len (rxd);
1573         int           status = ep_rxd_status (rxd);
1574         kqswnal_rx_t *krx    = (kqswnal_rx_t *)ep_rxd_arg (rxd);
1575
1576         CDEBUG(D_NET, "kqswnal_rxhandler: rxd %p, krx %p, nob %d, status %d\n",
1577                rxd, krx, nob, status);
1578
1579         LASSERT (krx != NULL);
1580         LASSERT (krx->krx_state = KRX_POSTED);
1581
1582         krx->krx_state = KRX_PARSE;
1583         krx->krx_rxd = rxd;
1584         krx->krx_nob = nob;
1585 #if MULTIRAIL_EKC
1586         krx->krx_rpc_reply_needed = (status != EP_SHUTDOWN) && ep_rxd_isrpc(rxd);
1587 #else
1588         krx->krx_rpc_reply_needed = ep_rxd_isrpc(rxd);
1589 #endif
1590         /* Default to failure if an RPC reply is requested but not handled */
1591         krx->krx_rpc_reply_status = -EPROTO;
1592         atomic_set (&krx->krx_refcount, 1);
1593
1594         /* must receive a whole header to be able to parse */
1595         if (status != EP_SUCCESS || nob < sizeof (ptl_hdr_t))
1596         {
1597                 /* receives complete with failure when receiver is removed */
1598 #if MULTIRAIL_EKC
1599                 if (status == EP_SHUTDOWN)
1600                         LASSERT (kqswnal_data.kqn_shuttingdown);
1601                 else
1602                         CERROR("receive status failed with status %d nob %d\n",
1603                                ep_rxd_status(rxd), nob);
1604 #else
1605                 if (!kqswnal_data.kqn_shuttingdown)
1606                         CERROR("receive status failed with status %d nob %d\n",
1607                                ep_rxd_status(rxd), nob);
1608 #endif
1609                 kqswnal_rx_decref(krx);
1610                 return;
1611         }
1612
1613         if (!in_interrupt()) {
1614                 kqswnal_parse(krx);
1615                 return;
1616         }
1617
1618         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
1619
1620         list_add_tail (&krx->krx_list, &kqswnal_data.kqn_readyrxds);
1621         wake_up (&kqswnal_data.kqn_sched_waitq);
1622
1623         spin_unlock_irqrestore (&kqswnal_data.kqn_sched_lock, flags);
1624 }
1625
1626 #if KQSW_CHECKSUM
1627 void
1628 kqswnal_csum_error (kqswnal_rx_t *krx, int ishdr)
1629 {
1630         ptl_hdr_t *hdr = (ptl_hdr_t *)page_address (krx->krx_kiov[0].kiov_page);
1631
1632         CERROR ("%s checksum mismatch %p: dnid "LPX64", snid "LPX64
1633                 ", dpid %d, spid %d, type %d\n",
1634                 ishdr ? "Header" : "Payload", krx,
1635                 le64_to_cpu(hdr->dest_nid), le64_to_cpu(hdr->src_nid)
1636                 le32_to_cpu(hdr->dest_pid), le32_to_cpu(hdr->src_pid),
1637                 le32_to_cpu(hdr->type));
1638
1639         switch (le32_to_cpu(hdr->type))
1640         {
1641         case PTL_MSG_ACK:
1642                 CERROR("ACK: mlen %d dmd "LPX64"."LPX64" match "LPX64
1643                        " len %u\n",
1644                        le32_to_cpu(hdr->msg.ack.mlength),
1645                        hdr->msg.ack.dst_wmd.handle_cookie,
1646                        hdr->msg.ack.dst_wmd.handle_idx,
1647                        le64_to_cpu(hdr->msg.ack.match_bits),
1648                        le32_to_cpu(hdr->msg.ack.length));
1649                 break;
1650         case PTL_MSG_PUT:
1651                 CERROR("PUT: ptl %d amd "LPX64"."LPX64" match "LPX64
1652                        " len %u off %u data "LPX64"\n",
1653                        le32_to_cpu(hdr->msg.put.ptl_index),
1654                        hdr->msg.put.ack_wmd.handle_cookie,
1655                        hdr->msg.put.ack_wmd.handle_idx,
1656                        le64_to_cpu(hdr->msg.put.match_bits),
1657                        le32_to_cpu(hdr->msg.put.length),
1658                        le32_to_cpu(hdr->msg.put.offset),
1659                        hdr->msg.put.hdr_data);
1660                 break;
1661         case PTL_MSG_GET:
1662                 CERROR ("GET: <>\n");
1663                 break;
1664         case PTL_MSG_REPLY:
1665                 CERROR ("REPLY: <>\n");
1666                 break;
1667         default:
1668                 CERROR ("TYPE?: <>\n");
1669         }
1670 }
1671 #endif
1672
1673 static ptl_err_t
1674 kqswnal_recvmsg (lib_nal_t    *nal,
1675                  void         *private,
1676                  lib_msg_t    *libmsg,
1677                  unsigned int  niov,
1678                  struct iovec *iov,
1679                  ptl_kiov_t   *kiov,
1680                  size_t        offset,
1681                  size_t        mlen,
1682                  size_t        rlen)
1683 {
1684         kqswnal_rx_t *krx = (kqswnal_rx_t *)private;
1685         char         *buffer = page_address(krx->krx_kiov[0].kiov_page);
1686         ptl_hdr_t    *hdr = (ptl_hdr_t *)buffer;
1687         int           page;
1688         char         *page_ptr;
1689         int           page_nob;
1690         char         *iov_ptr;
1691         int           iov_nob;
1692         int           frag;
1693         int           rc;
1694 #if KQSW_CHECKSUM
1695         kqsw_csum_t   senders_csum;
1696         kqsw_csum_t   payload_csum = 0;
1697         kqsw_csum_t   hdr_csum = kqsw_csum(0, hdr, sizeof(*hdr));
1698         size_t        csum_len = mlen;
1699         int           csum_frags = 0;
1700         int           csum_nob = 0;
1701         static atomic_t csum_counter;
1702         int           csum_verbose = (atomic_read(&csum_counter)%1000001) == 0;
1703
1704         atomic_inc (&csum_counter);
1705
1706         memcpy (&senders_csum, buffer + sizeof (ptl_hdr_t), sizeof (kqsw_csum_t));
1707         if (senders_csum != hdr_csum)
1708                 kqswnal_csum_error (krx, 1);
1709 #endif
1710         /* NB lib_parse() has already flipped *hdr */
1711
1712         CDEBUG(D_NET,"kqswnal_recv, mlen="LPSZ", rlen="LPSZ"\n", mlen, rlen);
1713
1714         if (krx->krx_rpc_reply_needed &&
1715             hdr->type == PTL_MSG_PUT) {
1716                 /* This must be an optimized PUT */
1717                 rc = kqswnal_rdma (krx, libmsg, PTL_MSG_PUT,
1718                                    niov, iov, kiov, offset, mlen);
1719                 return (rc == 0 ? PTL_OK : PTL_FAIL);
1720         }
1721
1722         /* What was actually received must be >= payload. */
1723         LASSERT (mlen <= rlen);
1724         if (krx->krx_nob < KQSW_HDR_SIZE + mlen) {
1725                 CERROR("Bad message size: have %d, need %d + %d\n",
1726                        krx->krx_nob, (int)KQSW_HDR_SIZE, (int)mlen);
1727                 return (PTL_FAIL);
1728         }
1729
1730         /* It must be OK to kmap() if required */
1731         LASSERT (kiov == NULL || !in_interrupt ());
1732         /* Either all pages or all vaddrs */
1733         LASSERT (!(kiov != NULL && iov != NULL));
1734
1735         if (mlen != 0) {
1736                 page     = 0;
1737                 page_ptr = buffer + KQSW_HDR_SIZE;
1738                 page_nob = PAGE_SIZE - KQSW_HDR_SIZE;
1739
1740                 LASSERT (niov > 0);
1741
1742                 if (kiov != NULL) {
1743                         /* skip complete frags */
1744                         while (offset >= kiov->kiov_len) {
1745                                 offset -= kiov->kiov_len;
1746                                 kiov++;
1747                                 niov--;
1748                                 LASSERT (niov > 0);
1749                         }
1750                         iov_ptr = ((char *)kmap (kiov->kiov_page)) +
1751                                 kiov->kiov_offset + offset;
1752                         iov_nob = kiov->kiov_len - offset;
1753                 } else {
1754                         /* skip complete frags */
1755                         while (offset >= iov->iov_len) {
1756                                 offset -= iov->iov_len;
1757                                 iov++;
1758                                 niov--;
1759                                 LASSERT (niov > 0);
1760                         }
1761                         iov_ptr = iov->iov_base + offset;
1762                         iov_nob = iov->iov_len - offset;
1763                 }
1764
1765                 for (;;)
1766                 {
1767                         frag = mlen;
1768                         if (frag > page_nob)
1769                                 frag = page_nob;
1770                         if (frag > iov_nob)
1771                                 frag = iov_nob;
1772
1773                         memcpy (iov_ptr, page_ptr, frag);
1774 #if KQSW_CHECKSUM
1775                         payload_csum = kqsw_csum (payload_csum, iov_ptr, frag);
1776                         csum_nob += frag;
1777                         csum_frags++;
1778 #endif
1779                         mlen -= frag;
1780                         if (mlen == 0)
1781                                 break;
1782
1783                         page_nob -= frag;
1784                         if (page_nob != 0)
1785                                 page_ptr += frag;
1786                         else
1787                         {
1788                                 page++;
1789                                 LASSERT (page < krx->krx_npages);
1790                                 page_ptr = page_address(krx->krx_kiov[page].kiov_page);
1791                                 page_nob = PAGE_SIZE;
1792                         }
1793
1794                         iov_nob -= frag;
1795                         if (iov_nob != 0)
1796                                 iov_ptr += frag;
1797                         else if (kiov != NULL) {
1798                                 kunmap (kiov->kiov_page);
1799                                 kiov++;
1800                                 niov--;
1801                                 LASSERT (niov > 0);
1802                                 iov_ptr = ((char *)kmap (kiov->kiov_page)) + kiov->kiov_offset;
1803                                 iov_nob = kiov->kiov_len;
1804                         } else {
1805                                 iov++;
1806                                 niov--;
1807                                 LASSERT (niov > 0);
1808                                 iov_ptr = iov->iov_base;
1809                                 iov_nob = iov->iov_len;
1810                         }
1811                 }
1812
1813                 if (kiov != NULL)
1814                         kunmap (kiov->kiov_page);
1815         }
1816
1817 #if KQSW_CHECKSUM
1818         memcpy (&senders_csum, buffer + sizeof(ptl_hdr_t) + sizeof(kqsw_csum_t),
1819                 sizeof(kqsw_csum_t));
1820
1821         if (csum_len != rlen)
1822                 CERROR("Unable to checksum data in user's buffer\n");
1823         else if (senders_csum != payload_csum)
1824                 kqswnal_csum_error (krx, 0);
1825
1826         if (csum_verbose)
1827                 CERROR("hdr csum %lx, payload_csum %lx, csum_frags %d, "
1828                        "csum_nob %d\n",
1829                         hdr_csum, payload_csum, csum_frags, csum_nob);
1830 #endif
1831         lib_finalize(nal, private, libmsg, PTL_OK);
1832
1833         return (PTL_OK);
1834 }
1835
1836 static ptl_err_t
1837 kqswnal_recv(lib_nal_t    *nal,
1838              void         *private,
1839              lib_msg_t    *libmsg,
1840              unsigned int  niov,
1841              struct iovec *iov,
1842              size_t        offset,
1843              size_t        mlen,
1844              size_t        rlen)
1845 {
1846         return (kqswnal_recvmsg(nal, private, libmsg,
1847                                 niov, iov, NULL,
1848                                 offset, mlen, rlen));
1849 }
1850
1851 static ptl_err_t
1852 kqswnal_recv_pages (lib_nal_t    *nal,
1853                     void         *private,
1854                     lib_msg_t    *libmsg,
1855                     unsigned int  niov,
1856                     ptl_kiov_t   *kiov,
1857                     size_t        offset,
1858                     size_t        mlen,
1859                     size_t        rlen)
1860 {
1861         return (kqswnal_recvmsg(nal, private, libmsg,
1862                                 niov, NULL, kiov,
1863                                 offset, mlen, rlen));
1864 }
1865
1866 int
1867 kqswnal_thread_start (int (*fn)(void *arg), void *arg)
1868 {
1869         long    pid = kernel_thread (fn, arg, 0);
1870
1871         if (pid < 0)
1872                 return ((int)pid);
1873
1874         atomic_inc (&kqswnal_data.kqn_nthreads);
1875         return (0);
1876 }
1877
1878 void
1879 kqswnal_thread_fini (void)
1880 {
1881         atomic_dec (&kqswnal_data.kqn_nthreads);
1882 }
1883
1884 int
1885 kqswnal_scheduler (void *arg)
1886 {
1887         kqswnal_rx_t    *krx;
1888         kqswnal_tx_t    *ktx;
1889         kpr_fwd_desc_t  *fwd;
1890         unsigned long    flags;
1891         int              rc;
1892         int              counter = 0;
1893         int              did_something;
1894
1895         kportal_daemonize ("kqswnal_sched");
1896         kportal_blockallsigs ();
1897
1898         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
1899
1900         for (;;)
1901         {
1902                 did_something = 0;
1903
1904                 if (!list_empty (&kqswnal_data.kqn_readyrxds))
1905                 {
1906                         krx = list_entry(kqswnal_data.kqn_readyrxds.next,
1907                                          kqswnal_rx_t, krx_list);
1908                         list_del (&krx->krx_list);
1909                         spin_unlock_irqrestore(&kqswnal_data.kqn_sched_lock,
1910                                                flags);
1911
1912                         switch (krx->krx_state) {
1913                         case KRX_PARSE:
1914                                 kqswnal_parse (krx);
1915                                 break;
1916                         case KRX_COMPLETING:
1917                                 /* Drop last ref to reply to RPC and requeue */
1918                                 LASSERT (krx->krx_rpc_reply_needed);
1919                                 kqswnal_rx_decref (krx);
1920                                 break;
1921                         default:
1922                                 LBUG();
1923                         }
1924
1925                         did_something = 1;
1926                         spin_lock_irqsave(&kqswnal_data.kqn_sched_lock, flags);
1927                 }
1928
1929                 if (!list_empty (&kqswnal_data.kqn_delayedtxds))
1930                 {
1931                         ktx = list_entry(kqswnal_data.kqn_delayedtxds.next,
1932                                          kqswnal_tx_t, ktx_list);
1933                         list_del_init (&ktx->ktx_delayed_list);
1934                         spin_unlock_irqrestore(&kqswnal_data.kqn_sched_lock,
1935                                                flags);
1936
1937                         rc = kqswnal_launch (ktx);
1938                         if (rc != 0) {
1939                                 CERROR("Failed delayed transmit to "LPX64
1940                                        ": %d\n", ktx->ktx_nid, rc);
1941                                 kqswnal_tx_done (ktx, rc);
1942                         }
1943                         atomic_dec (&kqswnal_data.kqn_pending_txs);
1944
1945                         did_something = 1;
1946                         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
1947                 }
1948
1949                 if (!list_empty (&kqswnal_data.kqn_delayedfwds))
1950                 {
1951                         fwd = list_entry (kqswnal_data.kqn_delayedfwds.next, kpr_fwd_desc_t, kprfd_list);
1952                         list_del (&fwd->kprfd_list);
1953                         spin_unlock_irqrestore (&kqswnal_data.kqn_sched_lock, flags);
1954
1955                         /* If we're shutting down, this will just requeue fwd on kqn_idletxd_fwdq */
1956                         kqswnal_fwd_packet (NULL, fwd);
1957
1958                         did_something = 1;
1959                         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
1960                 }
1961
1962                 /* nothing to do or hogging CPU */
1963                 if (!did_something || counter++ == KQSW_RESCHED) {
1964                         spin_unlock_irqrestore(&kqswnal_data.kqn_sched_lock,
1965                                                flags);
1966
1967                         counter = 0;
1968
1969                         if (!did_something) {
1970                                 if (kqswnal_data.kqn_shuttingdown == 2) {
1971                                         /* We only exit in stage 2 of shutdown when
1972                                          * there's nothing left to do */
1973                                         break;
1974                                 }
1975                                 rc = wait_event_interruptible (kqswnal_data.kqn_sched_waitq,
1976                                                                kqswnal_data.kqn_shuttingdown == 2 ||
1977                                                                !list_empty(&kqswnal_data.kqn_readyrxds) ||
1978                                                                !list_empty(&kqswnal_data.kqn_delayedtxds) ||
1979                                                                !list_empty(&kqswnal_data.kqn_delayedfwds));
1980                                 LASSERT (rc == 0);
1981                         } else if (need_resched())
1982                                 schedule ();
1983
1984                         spin_lock_irqsave (&kqswnal_data.kqn_sched_lock, flags);
1985                 }
1986         }
1987
1988         kqswnal_thread_fini ();
1989         return (0);
1990 }
1991
1992 lib_nal_t kqswnal_lib =
1993 {
1994         libnal_data:       &kqswnal_data,         /* NAL private data */
1995         libnal_send:        kqswnal_send,
1996         libnal_send_pages:  kqswnal_send_pages,
1997         libnal_recv:        kqswnal_recv,
1998         libnal_recv_pages:  kqswnal_recv_pages,
1999         libnal_dist:        kqswnal_dist
2000 };