lnet/lnet/lib-move.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lnet/lnet/lib-move.c
  37  *
  38  * Data movement routines
  39  */
  40
  41 #define DEBUG_SUBSYSTEM S_LNET
  42
  43 #include <lnet/lib-lnet.h>
  44
  45 static int local_nid_dist_zero = 1;
  46 CFS_MODULE_PARM(local_nid_dist_zero, "i", int, 0444,
  47                 "Reserved");
  48
  49 /* forward ref */
  50 static void lnet_commit_md (lnet_libmd_t *md, lnet_msg_t *msg);
  51
  52 #define LNET_MATCHMD_NONE     0   /* Didn't match */
  53 #define LNET_MATCHMD_OK       1   /* Matched OK */
  54 #define LNET_MATCHMD_DROP     2   /* Must be discarded */
  55
  56 static int
  57 lnet_try_match_md (int index, int op_mask, lnet_process_id_t src,
  58                    unsigned int rlength, unsigned int roffset,
  59                    __u64 match_bits, lnet_libmd_t *md, lnet_msg_t *msg,
  60                    unsigned int *mlength_out, unsigned int *offset_out)
  61 {
  62         /* ALWAYS called holding the LNET_LOCK, and can't LNET_UNLOCK;
  63          * lnet_match_blocked_msg() relies on this to avoid races */
  64         unsigned int  offset;
  65         unsigned int  mlength;
  66         lnet_me_t    *me = md->md_me;
  67
  68         /* mismatched MD op */
  69         if ((md->md_options & op_mask) == 0)
  70                 return LNET_MATCHMD_NONE;
  71
  72         /* MD exhausted */
  73         if (lnet_md_exhausted(md))
  74                 return LNET_MATCHMD_NONE;
  75
  76         /* mismatched ME nid/pid? */
  77         if (me->me_match_id.nid != LNET_NID_ANY &&
  78             me->me_match_id.nid != src.nid)
  79                 return LNET_MATCHMD_NONE;
  80
  81         if (me->me_match_id.pid != LNET_PID_ANY &&
  82             me->me_match_id.pid != src.pid)
  83                 return LNET_MATCHMD_NONE;
  84
  85         /* mismatched ME matchbits? */
  86         if (((me->me_match_bits ^ match_bits) & ~me->me_ignore_bits) != 0)
  87                 return LNET_MATCHMD_NONE;
  88
  89         /* Hurrah! This _is_ a match; check it out... */
  90
  91         if ((md->md_options & LNET_MD_MANAGE_REMOTE) == 0)
  92                 offset = md->md_offset;
  93         else
  94                 offset = roffset;
  95
  96         if ((md->md_options & LNET_MD_MAX_SIZE) != 0) {
  97                 mlength = md->md_max_size;
  98                 LASSERT (md->md_offset + mlength <= md->md_length);
  99         } else {
 100                 mlength = md->md_length - offset;
 101         }
 102
 103         if (rlength <= mlength) {        /* fits in allowed space */
 104                 mlength = rlength;
 105         } else if ((md->md_options & LNET_MD_TRUNCATE) == 0) {
 106                 /* this packet _really_ is too big */
 107                 CERROR("Matching packet from %s, match "LPU64
 108                        " length %d too big: %d left, %d allowed\n",
 109                        libcfs_id2str(src), match_bits, rlength,
 110                        md->md_length - offset, mlength);
 111
 112                 return LNET_MATCHMD_DROP;
 113         }
 114
 115         /* Commit to this ME/MD */
 116         CDEBUG(D_NET, "Incoming %s index %x from %s of "
 117                "length %d/%d into md "LPX64" [%d] + %d\n",
 118                (op_mask == LNET_MD_OP_PUT) ? "put" : "get",
 119                index, libcfs_id2str(src), mlength, rlength,
 120                md->md_lh.lh_cookie, md->md_niov, offset);
 121
 122         lnet_commit_md(md, msg);
 123         md->md_offset = offset + mlength;
 124
 125         /* NB Caller will set ev.type and ev.hdr_data */
 126         msg->msg_ev.initiator = src;
 127         msg->msg_ev.pt_index = index;
 128         msg->msg_ev.match_bits = match_bits;
 129         msg->msg_ev.rlength = rlength;
 130         msg->msg_ev.mlength = mlength;
 131         msg->msg_ev.offset = offset;
 132
 133         lnet_md_deconstruct(md, &msg->msg_ev.md);
 134         lnet_md2handle(&msg->msg_ev.md_handle, md);
 135
 136         *offset_out = offset;
 137         *mlength_out = mlength;
 138
 139         /* Auto-unlink NOW, so the ME gets unlinked if required.
 140          * We bumped md->md_refcount above so the MD just gets flagged
 141          * for unlink when it is finalized. */
 142         if ((md->md_flags & LNET_MD_FLAG_AUTO_UNLINK) != 0 &&
 143             lnet_md_exhausted(md)) {
 144                 lnet_md_unlink(md);
 145         }
 146
 147         return LNET_MATCHMD_OK;
 148 }
 149
 150 static int
 151 lnet_match_md(int index, int op_mask, lnet_process_id_t src,
 152               unsigned int rlength, unsigned int roffset,
 153               __u64 match_bits, lnet_msg_t *msg,
 154               unsigned int *mlength_out, unsigned int *offset_out,
 155               lnet_libmd_t **md_out)
 156 {
 157         lnet_portal_t    *ptl = &the_lnet.ln_portals[index];
 158         cfs_list_t       *head;
 159         lnet_me_t        *me;
 160         lnet_me_t        *tmp;
 161         lnet_libmd_t     *md;
 162         int               rc;
 163
 164         CDEBUG (D_NET, "Request from %s of length %d into portal %d "
 165                 "MB="LPX64"\n", libcfs_id2str(src), rlength, index, match_bits);
 166
 167         if (index < 0 || index >= the_lnet.ln_nportals) {
 168                 CERROR("Invalid portal %d not in [0-%d]\n",
 169                        index, the_lnet.ln_nportals);
 170                 return LNET_MATCHMD_DROP;
 171         }
 172
 173         head = lnet_portal_me_head(index, src, match_bits);
 174         if (head == NULL) /* nobody posted anything on this portal */
 175                 goto out;
 176
 177         cfs_list_for_each_entry_safe_typed (me, tmp, head,
 178                                             lnet_me_t, me_list) {
 179                 md = me->me_md;
 180
 181                 /* ME attached but MD not attached yet */
 182                 if (md == NULL)
 183                         continue;
 184
 185                 LASSERT (me == md->md_me);
 186
 187                 rc = lnet_try_match_md(index, op_mask, src, rlength,
 188                                        roffset, match_bits, md, msg,
 189                                        mlength_out, offset_out);
 190                 switch (rc) {
 191                 default:
 192                         LBUG();
 193
 194                 case LNET_MATCHMD_NONE:
 195                         continue;
 196
 197                 case LNET_MATCHMD_OK:
 198                         *md_out = md;
 199                         return LNET_MATCHMD_OK;
 200
 201                 case LNET_MATCHMD_DROP:
 202                         return LNET_MATCHMD_DROP;
 203                 }
 204                 /* not reached */
 205         }
 206
 207  out:
 208         if (op_mask == LNET_MD_OP_GET ||
 209             !lnet_portal_is_lazy(ptl))
 210                 return LNET_MATCHMD_DROP;
 211
 212         return LNET_MATCHMD_NONE;
 213 }
 214
 215 int
 216 lnet_fail_nid (lnet_nid_t nid, unsigned int threshold)
 217 {
 218         lnet_test_peer_t  *tp;
 219         cfs_list_t        *el;
 220         cfs_list_t        *next;
 221         cfs_list_t         cull;
 222
 223         LASSERT (the_lnet.ln_init);
 224
 225         if (threshold != 0) {
 226                 /* Adding a new entry */
 227                 LIBCFS_ALLOC(tp, sizeof(*tp));
 228                 if (tp == NULL)
 229                         return -ENOMEM;
 230
 231                 tp->tp_nid = nid;
 232                 tp->tp_threshold = threshold;
 233
 234                 LNET_LOCK();
 235                 cfs_list_add_tail (&tp->tp_list, &the_lnet.ln_test_peers);
 236                 LNET_UNLOCK();
 237                 return 0;
 238         }
 239
 240         /* removing entries */
 241         CFS_INIT_LIST_HEAD (&cull);
 242
 243         LNET_LOCK();
 244
 245         cfs_list_for_each_safe (el, next, &the_lnet.ln_test_peers) {
 246                 tp = cfs_list_entry (el, lnet_test_peer_t, tp_list);
 247
 248                 if (tp->tp_threshold == 0 ||    /* needs culling anyway */
 249                     nid == LNET_NID_ANY ||       /* removing all entries */
 250                     tp->tp_nid == nid)          /* matched this one */
 251                 {
 252                         cfs_list_del (&tp->tp_list);
 253                         cfs_list_add (&tp->tp_list, &cull);
 254                 }
 255         }
 256
 257         LNET_UNLOCK();
 258
 259         while (!cfs_list_empty (&cull)) {
 260                 tp = cfs_list_entry (cull.next, lnet_test_peer_t, tp_list);
 261
 262                 cfs_list_del (&tp->tp_list);
 263                 LIBCFS_FREE(tp, sizeof (*tp));
 264         }
 265         return 0;
 266 }
 267
 268 static int
 269 fail_peer (lnet_nid_t nid, int outgoing)
 270 {
 271         lnet_test_peer_t *tp;
 272         cfs_list_t       *el;
 273         cfs_list_t       *next;
 274         cfs_list_t        cull;
 275         int               fail = 0;
 276
 277         CFS_INIT_LIST_HEAD (&cull);
 278
 279         LNET_LOCK();
 280
 281         cfs_list_for_each_safe (el, next, &the_lnet.ln_test_peers) {
 282                 tp = cfs_list_entry (el, lnet_test_peer_t, tp_list);
 283
 284                 if (tp->tp_threshold == 0) {
 285                         /* zombie entry */
 286                         if (outgoing) {
 287                                 /* only cull zombies on outgoing tests,
 288                                  * since we may be at interrupt priority on
 289                                  * incoming messages. */
 290                                 cfs_list_del (&tp->tp_list);
 291                                 cfs_list_add (&tp->tp_list, &cull);
 292                         }
 293                         continue;
 294                 }
 295
 296                 if (tp->tp_nid == LNET_NID_ANY || /* fail every peer */
 297                     nid == tp->tp_nid) {        /* fail this peer */
 298                         fail = 1;
 299
 300                         if (tp->tp_threshold != LNET_MD_THRESH_INF) {
 301                                 tp->tp_threshold--;
 302                                 if (outgoing &&
 303                                     tp->tp_threshold == 0) {
 304                                         /* see above */
 305                                         cfs_list_del (&tp->tp_list);
 306                                         cfs_list_add (&tp->tp_list, &cull);
 307                                 }
 308                         }
 309                         break;
 310                 }
 311         }
 312
 313         LNET_UNLOCK ();
 314
 315         while (!cfs_list_empty (&cull)) {
 316                 tp = cfs_list_entry (cull.next, lnet_test_peer_t, tp_list);
 317                 cfs_list_del (&tp->tp_list);
 318
 319                 LIBCFS_FREE(tp, sizeof (*tp));
 320         }
 321
 322         return (fail);
 323 }
 324
 325 unsigned int
 326 lnet_iov_nob (unsigned int niov, struct iovec *iov)
 327 {
 328         unsigned int nob = 0;
 329
 330         while (niov-- > 0)
 331                 nob += (iov++)->iov_len;
 332
 333         return (nob);
 334 }
 335
 336 void
 337 lnet_copy_iov2iov (unsigned int ndiov, struct iovec *diov, unsigned int doffset,
 338                    unsigned int nsiov, struct iovec *siov, unsigned int soffset,
 339                    unsigned int nob)
 340 {
 341         /* NB diov, siov are READ-ONLY */
 342         unsigned int  this_nob;
 343
 344         if (nob == 0)
 345                 return;
 346
 347         /* skip complete frags before 'doffset' */
 348         LASSERT (ndiov > 0);
 349         while (doffset >= diov->iov_len) {
 350                 doffset -= diov->iov_len;
 351                 diov++;
 352                 ndiov--;
 353                 LASSERT (ndiov > 0);
 354         }
 355
 356         /* skip complete frags before 'soffset' */
 357         LASSERT (nsiov > 0);
 358         while (soffset >= siov->iov_len) {
 359                 soffset -= siov->iov_len;
 360                 siov++;
 361                 nsiov--;
 362                 LASSERT (nsiov > 0);
 363         }
 364
 365         do {
 366                 LASSERT (ndiov > 0);
 367                 LASSERT (nsiov > 0);
 368                 this_nob = MIN(diov->iov_len - doffset,
 369                                siov->iov_len - soffset);
 370                 this_nob = MIN(this_nob, nob);
 371
 372                 memcpy ((char *)diov->iov_base + doffset,
 373                         (char *)siov->iov_base + soffset, this_nob);
 374                 nob -= this_nob;
 375
 376                 if (diov->iov_len > doffset + this_nob) {
 377                         doffset += this_nob;
 378                 } else {
 379                         diov++;
 380                         ndiov--;
 381                         doffset = 0;
 382                 }
 383
 384                 if (siov->iov_len > soffset + this_nob) {
 385                         soffset += this_nob;
 386                 } else {
 387                         siov++;
 388                         nsiov--;
 389                         soffset = 0;
 390                 }
 391         } while (nob > 0);
 392 }
 393
 394 int
 395 lnet_extract_iov (int dst_niov, struct iovec *dst,
 396                   int src_niov, struct iovec *src,
 397                   unsigned int offset, unsigned int len)
 398 {
 399         /* Initialise 'dst' to the subset of 'src' starting at 'offset',
 400          * for exactly 'len' bytes, and return the number of entries.
 401          * NB not destructive to 'src' */
 402         unsigned int    frag_len;
 403         unsigned int    niov;
 404
 405         if (len == 0)                           /* no data => */
 406                 return (0);                     /* no frags */
 407
 408         LASSERT (src_niov > 0);
 409         while (offset >= src->iov_len) {      /* skip initial frags */
 410                 offset -= src->iov_len;
 411                 src_niov--;
 412                 src++;
 413                 LASSERT (src_niov > 0);
 414         }
 415
 416         niov = 1;
 417         for (;;) {
 418                 LASSERT (src_niov > 0);
 419                 LASSERT ((int)niov <= dst_niov);
 420
 421                 frag_len = src->iov_len - offset;
 422                 dst->iov_base = ((char *)src->iov_base) + offset;
 423
 424                 if (len <= frag_len) {
 425                         dst->iov_len = len;
 426                         return (niov);
 427                 }
 428
 429                 dst->iov_len = frag_len;
 430
 431                 len -= frag_len;
 432                 dst++;
 433                 src++;
 434                 niov++;
 435                 src_niov--;
 436                 offset = 0;
 437         }
 438 }
 439
 440 #ifndef __KERNEL__
 441 unsigned int
 442 lnet_kiov_nob (unsigned int niov, lnet_kiov_t *kiov)
 443 {
 444         LASSERT (0);
 445         return (0);
 446 }
 447
 448 void
 449 lnet_copy_kiov2kiov (unsigned int ndkiov, lnet_kiov_t *dkiov, unsigned int doffset,
 450                      unsigned int nskiov, lnet_kiov_t *skiov, unsigned int soffset,
 451                      unsigned int nob)
 452 {
 453         LASSERT (0);
 454 }
 455
 456 void
 457 lnet_copy_kiov2iov (unsigned int niov, struct iovec *iov, unsigned int iovoffset,
 458                     unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
 459                     unsigned int nob)
 460 {
 461         LASSERT (0);
 462 }
 463
 464 void
 465 lnet_copy_iov2kiov (unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
 466                     unsigned int niov, struct iovec *iov, unsigned int iovoffset,
 467                     unsigned int nob)
 468 {
 469         LASSERT (0);
 470 }
 471
 472 int
 473 lnet_extract_kiov (int dst_niov, lnet_kiov_t *dst,
 474                    int src_niov, lnet_kiov_t *src,
 475                    unsigned int offset, unsigned int len)
 476 {
 477         LASSERT (0);
 478 }
 479
 480 #else /* __KERNEL__ */
 481
 482 unsigned int
 483 lnet_kiov_nob (unsigned int niov, lnet_kiov_t *kiov)
 484 {
 485         unsigned int  nob = 0;
 486
 487         while (niov-- > 0)
 488                 nob += (kiov++)->kiov_len;
 489
 490         return (nob);
 491 }
 492
 493 void
 494 lnet_copy_kiov2kiov (unsigned int ndiov, lnet_kiov_t *diov, unsigned int doffset,
 495                      unsigned int nsiov, lnet_kiov_t *siov, unsigned int soffset,
 496                      unsigned int nob)
 497 {
 498         /* NB diov, siov are READ-ONLY */
 499         unsigned int    this_nob;
 500         char           *daddr = NULL;
 501         char           *saddr = NULL;
 502
 503         if (nob == 0)
 504                 return;
 505
 506         LASSERT (!cfs_in_interrupt ());
 507
 508         LASSERT (ndiov > 0);
 509         while (doffset >= diov->kiov_len) {
 510                 doffset -= diov->kiov_len;
 511                 diov++;
 512                 ndiov--;
 513                 LASSERT (ndiov > 0);
 514         }
 515
 516         LASSERT (nsiov > 0);
 517         while (soffset >= siov->kiov_len) {
 518                 soffset -= siov->kiov_len;
 519                 siov++;
 520                 nsiov--;
 521                 LASSERT (nsiov > 0);
 522         }
 523
 524         do {
 525                 LASSERT (ndiov > 0);
 526                 LASSERT (nsiov > 0);
 527                 this_nob = MIN(diov->kiov_len - doffset,
 528                                siov->kiov_len - soffset);
 529                 this_nob = MIN(this_nob, nob);
 530
 531                 if (daddr == NULL)
 532                         daddr = ((char *)cfs_kmap(diov->kiov_page)) +
 533                                 diov->kiov_offset + doffset;
 534                 if (saddr == NULL)
 535                         saddr = ((char *)cfs_kmap(siov->kiov_page)) +
 536                                 siov->kiov_offset + soffset;
 537
 538                 /* Vanishing risk of kmap deadlock when mapping 2 pages.
 539                  * However in practice at least one of the kiovs will be mapped
 540                  * kernel pages and the map/unmap will be NOOPs */
 541
 542                 memcpy (daddr, saddr, this_nob);
 543                 nob -= this_nob;
 544
 545                 if (diov->kiov_len > doffset + this_nob) {
 546                         daddr += this_nob;
 547                         doffset += this_nob;
 548                 } else {
 549                         cfs_kunmap(diov->kiov_page);
 550                         daddr = NULL;
 551                         diov++;
 552                         ndiov--;
 553                         doffset = 0;
 554                 }
 555
 556                 if (siov->kiov_len > soffset + this_nob) {
 557                         saddr += this_nob;
 558                         soffset += this_nob;
 559                 } else {
 560                         cfs_kunmap(siov->kiov_page);
 561                         saddr = NULL;
 562                         siov++;
 563                         nsiov--;
 564                         soffset = 0;
 565                 }
 566         } while (nob > 0);
 567
 568         if (daddr != NULL)
 569                 cfs_kunmap(diov->kiov_page);
 570         if (saddr != NULL)
 571                 cfs_kunmap(siov->kiov_page);
 572 }
 573
 574 void
 575 lnet_copy_kiov2iov (unsigned int niov, struct iovec *iov, unsigned int iovoffset,
 576                     unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
 577                     unsigned int nob)
 578 {
 579         /* NB iov, kiov are READ-ONLY */
 580         unsigned int    this_nob;
 581         char           *addr = NULL;
 582
 583         if (nob == 0)
 584                 return;
 585
 586         LASSERT (!cfs_in_interrupt ());
 587
 588         LASSERT (niov > 0);
 589         while (iovoffset >= iov->iov_len) {
 590                 iovoffset -= iov->iov_len;
 591                 iov++;
 592                 niov--;
 593                 LASSERT (niov > 0);
 594         }
 595
 596         LASSERT (nkiov > 0);
 597         while (kiovoffset >= kiov->kiov_len) {
 598                 kiovoffset -= kiov->kiov_len;
 599                 kiov++;
 600                 nkiov--;
 601                 LASSERT (nkiov > 0);
 602         }
 603
 604         do {
 605                 LASSERT (niov > 0);
 606                 LASSERT (nkiov > 0);
 607                 this_nob = MIN(iov->iov_len - iovoffset,
 608                                kiov->kiov_len - kiovoffset);
 609                 this_nob = MIN(this_nob, nob);
 610
 611                 if (addr == NULL)
 612                         addr = ((char *)cfs_kmap(kiov->kiov_page)) +
 613                                 kiov->kiov_offset + kiovoffset;
 614
 615                 memcpy ((char *)iov->iov_base + iovoffset, addr, this_nob);
 616                 nob -= this_nob;
 617
 618                 if (iov->iov_len > iovoffset + this_nob) {
 619                         iovoffset += this_nob;
 620                 } else {
 621                         iov++;
 622                         niov--;
 623                         iovoffset = 0;
 624                 }
 625
 626                 if (kiov->kiov_len > kiovoffset + this_nob) {
 627                         addr += this_nob;
 628                         kiovoffset += this_nob;
 629                 } else {
 630                         cfs_kunmap(kiov->kiov_page);
 631                         addr = NULL;
 632                         kiov++;
 633                         nkiov--;
 634                         kiovoffset = 0;
 635                 }
 636
 637         } while (nob > 0);
 638
 639         if (addr != NULL)
 640                 cfs_kunmap(kiov->kiov_page);
 641 }
 642
 643 void
 644 lnet_copy_iov2kiov (unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
 645                     unsigned int niov, struct iovec *iov, unsigned int iovoffset,
 646                     unsigned int nob)
 647 {
 648         /* NB kiov, iov are READ-ONLY */
 649         unsigned int    this_nob;
 650         char           *addr = NULL;
 651
 652         if (nob == 0)
 653                 return;
 654
 655         LASSERT (!cfs_in_interrupt ());
 656
 657         LASSERT (nkiov > 0);
 658         while (kiovoffset >= kiov->kiov_len) {
 659                 kiovoffset -= kiov->kiov_len;
 660                 kiov++;
 661                 nkiov--;
 662                 LASSERT (nkiov > 0);
 663         }
 664
 665         LASSERT (niov > 0);
 666         while (iovoffset >= iov->iov_len) {
 667                 iovoffset -= iov->iov_len;
 668                 iov++;
 669                 niov--;
 670                 LASSERT (niov > 0);
 671         }
 672
 673         do {
 674                 LASSERT (nkiov > 0);
 675                 LASSERT (niov > 0);
 676                 this_nob = MIN(kiov->kiov_len - kiovoffset,
 677                                iov->iov_len - iovoffset);
 678                 this_nob = MIN(this_nob, nob);
 679
 680                 if (addr == NULL)
 681                         addr = ((char *)cfs_kmap(kiov->kiov_page)) +
 682                                 kiov->kiov_offset + kiovoffset;
 683
 684                 memcpy (addr, (char *)iov->iov_base + iovoffset, this_nob);
 685                 nob -= this_nob;
 686
 687                 if (kiov->kiov_len > kiovoffset + this_nob) {
 688                         addr += this_nob;
 689                         kiovoffset += this_nob;
 690                 } else {
 691                         cfs_kunmap(kiov->kiov_page);
 692                         addr = NULL;
 693                         kiov++;
 694                         nkiov--;
 695                         kiovoffset = 0;
 696                 }
 697
 698                 if (iov->iov_len > iovoffset + this_nob) {
 699                         iovoffset += this_nob;
 700                 } else {
 701                         iov++;
 702                         niov--;
 703                         iovoffset = 0;
 704                 }
 705         } while (nob > 0);
 706
 707         if (addr != NULL)
 708                 cfs_kunmap(kiov->kiov_page);
 709 }
 710
 711 int
 712 lnet_extract_kiov (int dst_niov, lnet_kiov_t *dst,
 713                    int src_niov, lnet_kiov_t *src,
 714                    unsigned int offset, unsigned int len)
 715 {
 716         /* Initialise 'dst' to the subset of 'src' starting at 'offset',
 717          * for exactly 'len' bytes, and return the number of entries.
 718          * NB not destructive to 'src' */
 719         unsigned int    frag_len;
 720         unsigned int    niov;
 721
 722         if (len == 0)                           /* no data => */
 723                 return (0);                     /* no frags */
 724
 725         LASSERT (src_niov > 0);
 726         while (offset >= src->kiov_len) {      /* skip initial frags */
 727                 offset -= src->kiov_len;
 728                 src_niov--;
 729                 src++;
 730                 LASSERT (src_niov > 0);
 731         }
 732
 733         niov = 1;
 734         for (;;) {
 735                 LASSERT (src_niov > 0);
 736                 LASSERT ((int)niov <= dst_niov);
 737
 738                 frag_len = src->kiov_len - offset;
 739                 dst->kiov_page = src->kiov_page;
 740                 dst->kiov_offset = src->kiov_offset + offset;
 741
 742                 if (len <= frag_len) {
 743                         dst->kiov_len = len;
 744                         LASSERT (dst->kiov_offset + dst->kiov_len <= CFS_PAGE_SIZE);
 745                         return (niov);
 746                 }
 747
 748                 dst->kiov_len = frag_len;
 749                 LASSERT (dst->kiov_offset + dst->kiov_len <= CFS_PAGE_SIZE);
 750
 751                 len -= frag_len;
 752                 dst++;
 753                 src++;
 754                 niov++;
 755                 src_niov--;
 756                 offset = 0;
 757         }
 758 }
 759 #endif
 760
 761 void
 762 lnet_ni_recv(lnet_ni_t *ni, void *private, lnet_msg_t *msg, int delayed,
 763              unsigned int offset, unsigned int mlen, unsigned int rlen)
 764 {
 765         unsigned int  niov = 0;
 766         struct iovec *iov = NULL;
 767         lnet_kiov_t  *kiov = NULL;
 768         int           rc;
 769
 770         LASSERT (!cfs_in_interrupt ());
 771         LASSERT (mlen == 0 || msg != NULL);
 772
 773         if (msg != NULL) {
 774                 LASSERT(msg->msg_receiving);
 775                 LASSERT(!msg->msg_sending);
 776                 LASSERT(rlen == msg->msg_len);
 777                 LASSERT(mlen <= msg->msg_len);
 778
 779                 msg->msg_wanted = mlen;
 780                 msg->msg_offset = offset;
 781                 msg->msg_receiving = 0;
 782
 783                 if (mlen != 0) {
 784                         niov = msg->msg_niov;
 785                         iov  = msg->msg_iov;
 786                         kiov = msg->msg_kiov;
 787
 788                         LASSERT (niov > 0);
 789                         LASSERT ((iov == NULL) != (kiov == NULL));
 790                 }
 791         }
 792
 793         rc = (ni->ni_lnd->lnd_recv)(ni, private, msg, delayed,
 794                                     niov, iov, kiov, offset, mlen, rlen);
 795         if (rc < 0)
 796                 lnet_finalize(ni, msg, rc);
 797 }
 798
 799 int
 800 lnet_compare_routes(lnet_route_t *r1, lnet_route_t *r2)
 801 {
 802         lnet_peer_t *p1 = r1->lr_gateway;
 803         lnet_peer_t *p2 = r2->lr_gateway;
 804
 805         if (r1->lr_hops < r2->lr_hops)
 806                 return 1;
 807
 808         if (r1->lr_hops > r2->lr_hops)
 809                 return -1;
 810
 811         if (p1->lp_txqnob < p2->lp_txqnob)
 812                 return 1;
 813
 814         if (p1->lp_txqnob > p2->lp_txqnob)
 815                 return -1;
 816
 817         if (p1->lp_txcredits > p2->lp_txcredits)
 818                 return 1;
 819
 820         if (p1->lp_txcredits < p2->lp_txcredits)
 821                 return -1;
 822
 823         return 0;
 824 }
 825
 826
 827 void
 828 lnet_setpayloadbuffer(lnet_msg_t *msg)
 829 {
 830         lnet_libmd_t *md = msg->msg_md;
 831
 832         LASSERT (msg->msg_len > 0);
 833         LASSERT (!msg->msg_routing);
 834         LASSERT (md != NULL);
 835         LASSERT (msg->msg_niov == 0);
 836         LASSERT (msg->msg_iov == NULL);
 837         LASSERT (msg->msg_kiov == NULL);
 838
 839         msg->msg_niov = md->md_niov;
 840         if ((md->md_options & LNET_MD_KIOV) != 0)
 841                 msg->msg_kiov = md->md_iov.kiov;
 842         else
 843                 msg->msg_iov = md->md_iov.iov;
 844 }
 845
 846 void
 847 lnet_prep_send(lnet_msg_t *msg, int type, lnet_process_id_t target,
 848                unsigned int offset, unsigned int len)
 849 {
 850         msg->msg_type = type;
 851         msg->msg_target = target;
 852         msg->msg_len = len;
 853         msg->msg_offset = offset;
 854
 855         if (len != 0)
 856                 lnet_setpayloadbuffer(msg);
 857
 858         memset (&msg->msg_hdr, 0, sizeof (msg->msg_hdr));
 859         msg->msg_hdr.type           = cpu_to_le32(type);
 860         msg->msg_hdr.dest_nid       = cpu_to_le64(target.nid);
 861         msg->msg_hdr.dest_pid       = cpu_to_le32(target.pid);
 862         /* src_nid will be set later */
 863         msg->msg_hdr.src_pid        = cpu_to_le32(the_lnet.ln_pid);
 864         msg->msg_hdr.payload_length = cpu_to_le32(len);
 865 }
 866
 867 void
 868 lnet_ni_send(lnet_ni_t *ni, lnet_msg_t *msg)
 869 {
 870         void   *priv = msg->msg_private;
 871         int     rc;
 872
 873         LASSERT (!cfs_in_interrupt ());
 874         LASSERT (LNET_NETTYP(LNET_NIDNET(ni->ni_nid)) == LOLND ||
 875                  (msg->msg_txcredit && msg->msg_peertxcredit));
 876
 877         rc = (ni->ni_lnd->lnd_send)(ni, priv, msg);
 878         if (rc < 0)
 879                 lnet_finalize(ni, msg, rc);
 880 }
 881
 882 int
 883 lnet_eager_recv_locked(lnet_msg_t *msg)
 884 {
 885         lnet_peer_t *peer;
 886         lnet_ni_t   *ni;
 887         int          rc = 0;
 888
 889         LASSERT (!msg->msg_delayed);
 890         msg->msg_delayed = 1;
 891
 892         LASSERT (msg->msg_receiving);
 893         LASSERT (!msg->msg_sending);
 894
 895         peer = msg->msg_rxpeer;
 896         ni   = peer->lp_ni;
 897
 898         if (ni->ni_lnd->lnd_eager_recv != NULL) {
 899                 LNET_UNLOCK();
 900
 901                 rc = (ni->ni_lnd->lnd_eager_recv)(ni, msg->msg_private, msg,
 902                                                   &msg->msg_private);
 903                 if (rc != 0) {
 904                         CERROR("recv from %s / send to %s aborted: "
 905                                "eager_recv failed %d\n",
 906                                libcfs_nid2str(peer->lp_nid),
 907                                libcfs_id2str(msg->msg_target), rc);
 908                         LASSERT (rc < 0); /* required by my callers */
 909                 }
 910
 911                 LNET_LOCK();
 912         }
 913
 914         return rc;
 915 }
 916
 917 /* NB: caller shall hold a ref on 'lp' as I'd drop LNET_LOCK */
 918 void
 919 lnet_ni_peer_alive(lnet_peer_t *lp)
 920 {
 921         cfs_time_t  last_alive = 0;
 922         lnet_ni_t  *ni = lp->lp_ni;
 923
 924         LASSERT (lnet_peer_aliveness_enabled(lp));
 925         LASSERT (ni->ni_lnd->lnd_query != NULL);
 926
 927         LNET_UNLOCK();
 928         (ni->ni_lnd->lnd_query)(ni, lp->lp_nid, &last_alive);
 929         LNET_LOCK();
 930
 931         lp->lp_last_query = cfs_time_current();
 932
 933         if (last_alive != 0) /* NI has updated timestamp */
 934                 lp->lp_last_alive = last_alive;
 935         return;
 936 }
 937
 938 /* NB: always called with LNET_LOCK held */
 939 static inline int
 940 lnet_peer_is_alive (lnet_peer_t *lp, cfs_time_t now)
 941 {
 942         int        alive;
 943         cfs_time_t deadline;
 944
 945         LASSERT (lnet_peer_aliveness_enabled(lp));
 946
 947         /* Trust lnet_notify() if it has more recent aliveness news, but
 948          * ignore the initial assumed death (see lnet_peers_start_down()).
 949          */
 950         if (!lp->lp_alive && lp->lp_alive_count > 0 &&
 951             cfs_time_aftereq(lp->lp_timestamp, lp->lp_last_alive))
 952                 return 0;
 953
 954         deadline = cfs_time_add(lp->lp_last_alive,
 955                                 cfs_time_seconds(lp->lp_ni->ni_peertimeout));
 956         alive = cfs_time_after(deadline, now);
 957
 958         /* Update obsolete lp_alive except for routers assumed to be dead
 959          * initially, because router checker would update aliveness in this
 960          * case, and moreover lp_last_alive at peer creation is assumed.
 961          */
 962         if (alive && !lp->lp_alive &&
 963             !(lnet_isrouter(lp) && lp->lp_alive_count == 0))
 964                 lnet_notify_locked(lp, 0, 1, lp->lp_last_alive);
 965
 966         return alive;
 967 }
 968
 969
 970 /* NB: returns 1 when alive, 0 when dead, negative when error;
 971  *     may drop the LNET_LOCK */
 972 int
 973 lnet_peer_alive_locked (lnet_peer_t *lp)
 974 {
 975         cfs_time_t now = cfs_time_current();
 976
 977         if (!lnet_peer_aliveness_enabled(lp))
 978                 return -ENODEV;
 979
 980         if (lnet_peer_is_alive(lp, now))
 981                 return 1;
 982
 983         /* Peer appears dead, but we should avoid frequent NI queries (at
 984          * most once per lnet_queryinterval seconds). */
 985         if (lp->lp_last_query != 0) {
 986                 static const int lnet_queryinterval = 1;
 987
 988                 cfs_time_t next_query =
 989                            cfs_time_add(lp->lp_last_query,
 990                                         cfs_time_seconds(lnet_queryinterval));
 991
 992                 if (cfs_time_before(now, next_query)) {
 993                         if (lp->lp_alive)
 994                                 CWARN("Unexpected aliveness of peer %s: "
 995                                       "%d < %d (%d/%d)\n",
 996                                       libcfs_nid2str(lp->lp_nid),
 997                                       (int)now, (int)next_query,
 998                                       lnet_queryinterval,
 999                                       lp->lp_ni->ni_peertimeout);
1000                         return 0;
1001                 }
1002         }
1003
1004         /* query NI for latest aliveness news */
1005         lnet_ni_peer_alive(lp);
1006
1007         if (lnet_peer_is_alive(lp, now))
1008                 return 1;
1009
1010         lnet_notify_locked(lp, 0, 0, lp->lp_last_alive);
1011         return 0;
1012 }
1013
1014 int
1015 lnet_post_send_locked (lnet_msg_t *msg, int do_send)
1016 {
1017         /* lnet_send is going to LNET_UNLOCK immediately after this, so it sets
1018          * do_send FALSE and I don't do the unlock/send/lock bit.  I return
1019          * EAGAIN if msg blocked, EHOSTUNREACH if msg_txpeer appears dead, and
1020          * 0 if sent or OK to send */
1021         lnet_peer_t *lp = msg->msg_txpeer;
1022         lnet_ni_t   *ni = lp->lp_ni;
1023
1024         /* non-lnet_send() callers have checked before */
1025         LASSERT (!do_send || msg->msg_delayed);
1026         LASSERT (!msg->msg_receiving);
1027
1028         /* NB 'lp' is always the next hop */
1029         if ((msg->msg_target.pid & LNET_PID_USERFLAG) == 0 &&
1030             lnet_peer_alive_locked(lp) == 0) {
1031                 the_lnet.ln_counters.drop_count++;
1032                 the_lnet.ln_counters.drop_length += msg->msg_len;
1033                 LNET_UNLOCK();
1034
1035                 CNETERR("Dropping message for %s: peer not alive\n",
1036                         libcfs_id2str(msg->msg_target));
1037                 if (do_send)
1038                         lnet_finalize(ni, msg, -EHOSTUNREACH);
1039
1040                 LNET_LOCK();
1041                 return EHOSTUNREACH;
1042         }
1043
1044         if (!msg->msg_peertxcredit) {
1045                 LASSERT ((lp->lp_txcredits < 0) ==
1046                          !cfs_list_empty(&lp->lp_txq));
1047
1048                 msg->msg_peertxcredit = 1;
1049                 lp->lp_txqnob += msg->msg_len + sizeof(lnet_hdr_t);
1050                 lp->lp_txcredits--;
1051
1052                 if (lp->lp_txcredits < lp->lp_mintxcredits)
1053                         lp->lp_mintxcredits = lp->lp_txcredits;
1054
1055                 if (lp->lp_txcredits < 0) {
1056                         msg->msg_delayed = 1;
1057                         cfs_list_add_tail(&msg->msg_list, &lp->lp_txq);
1058                         return EAGAIN;
1059                 }
1060         }
1061
1062         if (!msg->msg_txcredit) {
1063                 LASSERT ((ni->ni_txcredits < 0) ==
1064                          !cfs_list_empty(&ni->ni_txq));
1065
1066                 msg->msg_txcredit = 1;
1067                 ni->ni_txcredits--;
1068
1069                 if (ni->ni_txcredits < ni->ni_mintxcredits)
1070                         ni->ni_mintxcredits = ni->ni_txcredits;
1071
1072                 if (ni->ni_txcredits < 0) {
1073                         msg->msg_delayed = 1;
1074                         cfs_list_add_tail(&msg->msg_list, &ni->ni_txq);
1075                         return EAGAIN;
1076                 }
1077         }
1078
1079         if (do_send) {
1080                 LNET_UNLOCK();
1081                 lnet_ni_send(ni, msg);
1082                 LNET_LOCK();
1083         }
1084         return 0;
1085 }
1086
1087 #ifdef __KERNEL__
1088 static void
1089 lnet_commit_routedmsg (lnet_msg_t *msg)
1090 {
1091         /* ALWAYS called holding the LNET_LOCK */
1092         LASSERT (msg->msg_routing);
1093
1094         the_lnet.ln_counters.msgs_alloc++;
1095         if (the_lnet.ln_counters.msgs_alloc >
1096             the_lnet.ln_counters.msgs_max)
1097                 the_lnet.ln_counters.msgs_max =
1098                         the_lnet.ln_counters.msgs_alloc;
1099
1100         the_lnet.ln_counters.route_count++;
1101         the_lnet.ln_counters.route_length += msg->msg_len;
1102
1103         LASSERT (!msg->msg_onactivelist);
1104         msg->msg_onactivelist = 1;
1105         cfs_list_add (&msg->msg_activelist, &the_lnet.ln_active_msgs);
1106 }
1107
1108 lnet_rtrbufpool_t *
1109 lnet_msg2bufpool(lnet_msg_t *msg)
1110 {
1111         lnet_rtrbufpool_t *rbp = &the_lnet.ln_rtrpools[0];
1112
1113         LASSERT (msg->msg_len <= LNET_MTU);
1114         while (msg->msg_len > (unsigned int)rbp->rbp_npages * CFS_PAGE_SIZE) {
1115                 rbp++;
1116                 LASSERT (rbp < &the_lnet.ln_rtrpools[LNET_NRBPOOLS]);
1117         }
1118
1119         return rbp;
1120 }
1121
1122 int
1123 lnet_post_routed_recv_locked (lnet_msg_t *msg, int do_recv)
1124 {
1125         /* lnet_parse is going to LNET_UNLOCK immediately after this, so it
1126          * sets do_recv FALSE and I don't do the unlock/send/lock bit.  I
1127          * return EAGAIN if msg blocked and 0 if received or OK to receive */
1128         lnet_peer_t         *lp = msg->msg_rxpeer;
1129         lnet_rtrbufpool_t   *rbp;
1130         lnet_rtrbuf_t       *rb;
1131
1132         LASSERT (msg->msg_iov == NULL);
1133         LASSERT (msg->msg_kiov == NULL);
1134         LASSERT (msg->msg_niov == 0);
1135         LASSERT (msg->msg_routing);
1136         LASSERT (msg->msg_receiving);
1137         LASSERT (!msg->msg_sending);
1138
1139         /* non-lnet_parse callers only send delayed messages */
1140         LASSERT (!do_recv || msg->msg_delayed);
1141
1142         if (!msg->msg_peerrtrcredit) {
1143                 LASSERT ((lp->lp_rtrcredits < 0) ==
1144                          !cfs_list_empty(&lp->lp_rtrq));
1145
1146                 msg->msg_peerrtrcredit = 1;
1147                 lp->lp_rtrcredits--;
1148                 if (lp->lp_rtrcredits < lp->lp_minrtrcredits)
1149                         lp->lp_minrtrcredits = lp->lp_rtrcredits;
1150
1151                 if (lp->lp_rtrcredits < 0) {
1152                         /* must have checked eager_recv before here */
1153                         LASSERT (msg->msg_delayed);
1154                         cfs_list_add_tail(&msg->msg_list, &lp->lp_rtrq);
1155                         return EAGAIN;
1156                 }
1157         }
1158
1159         rbp = lnet_msg2bufpool(msg);
1160
1161         if (!msg->msg_rtrcredit) {
1162                 LASSERT ((rbp->rbp_credits < 0) ==
1163                          !cfs_list_empty(&rbp->rbp_msgs));
1164
1165                 msg->msg_rtrcredit = 1;
1166                 rbp->rbp_credits--;
1167                 if (rbp->rbp_credits < rbp->rbp_mincredits)
1168                         rbp->rbp_mincredits = rbp->rbp_credits;
1169
1170                 if (rbp->rbp_credits < 0) {
1171                         /* must have checked eager_recv before here */
1172                         LASSERT (msg->msg_delayed);
1173                         cfs_list_add_tail(&msg->msg_list, &rbp->rbp_msgs);
1174                         return EAGAIN;
1175                 }
1176         }
1177
1178         LASSERT (!cfs_list_empty(&rbp->rbp_bufs));
1179         rb = cfs_list_entry(rbp->rbp_bufs.next, lnet_rtrbuf_t, rb_list);
1180         cfs_list_del(&rb->rb_list);
1181
1182         msg->msg_niov = rbp->rbp_npages;
1183         msg->msg_kiov = &rb->rb_kiov[0];
1184
1185         if (do_recv) {
1186                 LNET_UNLOCK();
1187                 lnet_ni_recv(lp->lp_ni, msg->msg_private, msg, 1,
1188                              0, msg->msg_len, msg->msg_len);
1189                 LNET_LOCK();
1190         }
1191         return 0;
1192 }
1193 #endif
1194
1195 void
1196 lnet_return_credits_locked (lnet_msg_t *msg)
1197 {
1198         lnet_peer_t       *txpeer = msg->msg_txpeer;
1199         lnet_peer_t       *rxpeer = msg->msg_rxpeer;
1200         lnet_msg_t        *msg2;
1201         lnet_ni_t         *ni;
1202
1203         if (msg->msg_txcredit) {
1204                 /* give back NI txcredits */
1205                 msg->msg_txcredit = 0;
1206                 ni = txpeer->lp_ni;
1207
1208                 LASSERT((ni->ni_txcredits < 0) == !cfs_list_empty(&ni->ni_txq));
1209
1210                 ni->ni_txcredits++;
1211                 if (ni->ni_txcredits <= 0) {
1212                         msg2 = cfs_list_entry(ni->ni_txq.next, lnet_msg_t,
1213                                               msg_list);
1214                         cfs_list_del(&msg2->msg_list);
1215
1216                         LASSERT(msg2->msg_txpeer->lp_ni == ni);
1217                         LASSERT(msg2->msg_delayed);
1218
1219                         (void) lnet_post_send_locked(msg2, 1);
1220                 }
1221         }
1222
1223         if (msg->msg_peertxcredit) {
1224                 /* give back peer txcredits */
1225                 msg->msg_peertxcredit = 0;
1226
1227                 LASSERT((txpeer->lp_txcredits < 0) ==
1228                         !cfs_list_empty(&txpeer->lp_txq));
1229
1230                 txpeer->lp_txqnob -= msg->msg_len + sizeof(lnet_hdr_t);
1231                 LASSERT (txpeer->lp_txqnob >= 0);
1232
1233                 txpeer->lp_txcredits++;
1234                 if (txpeer->lp_txcredits <= 0) {
1235                         msg2 = cfs_list_entry(txpeer->lp_txq.next,
1236                                               lnet_msg_t, msg_list);
1237                         cfs_list_del(&msg2->msg_list);
1238
1239                         LASSERT (msg2->msg_txpeer == txpeer);
1240                         LASSERT (msg2->msg_delayed);
1241
1242                         (void) lnet_post_send_locked(msg2, 1);
1243                 }
1244         }
1245
1246         if (txpeer != NULL) {
1247                 msg->msg_txpeer = NULL;
1248                 lnet_peer_decref_locked(txpeer);
1249         }
1250
1251 #ifdef __KERNEL__
1252         if (msg->msg_rtrcredit) {
1253                 /* give back global router credits */
1254                 lnet_rtrbuf_t     *rb;
1255                 lnet_rtrbufpool_t *rbp;
1256
1257                 /* NB If a msg ever blocks for a buffer in rbp_msgs, it stays
1258                  * there until it gets one allocated, or aborts the wait
1259                  * itself */
1260                 LASSERT (msg->msg_kiov != NULL);
1261
1262                 rb = cfs_list_entry(msg->msg_kiov, lnet_rtrbuf_t, rb_kiov[0]);
1263                 rbp = rb->rb_pool;
1264                 LASSERT (rbp == lnet_msg2bufpool(msg));
1265
1266                 msg->msg_kiov = NULL;
1267                 msg->msg_rtrcredit = 0;
1268
1269                 LASSERT((rbp->rbp_credits < 0) ==
1270                         !cfs_list_empty(&rbp->rbp_msgs));
1271                 LASSERT((rbp->rbp_credits > 0) ==
1272                         !cfs_list_empty(&rbp->rbp_bufs));
1273
1274                 cfs_list_add(&rb->rb_list, &rbp->rbp_bufs);
1275                 rbp->rbp_credits++;
1276                 if (rbp->rbp_credits <= 0) {
1277                         msg2 = cfs_list_entry(rbp->rbp_msgs.next,
1278                                               lnet_msg_t, msg_list);
1279                         cfs_list_del(&msg2->msg_list);
1280
1281                         (void) lnet_post_routed_recv_locked(msg2, 1);
1282                 }
1283         }
1284
1285         if (msg->msg_peerrtrcredit) {
1286                 /* give back peer router credits */
1287                 msg->msg_peerrtrcredit = 0;
1288
1289                 LASSERT((rxpeer->lp_rtrcredits < 0) ==
1290                         !cfs_list_empty(&rxpeer->lp_rtrq));
1291
1292                 rxpeer->lp_rtrcredits++;
1293                 if (rxpeer->lp_rtrcredits <= 0) {
1294                         msg2 = cfs_list_entry(rxpeer->lp_rtrq.next,
1295                                               lnet_msg_t, msg_list);
1296                         cfs_list_del(&msg2->msg_list);
1297
1298                         (void) lnet_post_routed_recv_locked(msg2, 1);
1299                 }
1300         }
1301 #else
1302         LASSERT (!msg->msg_rtrcredit);
1303         LASSERT (!msg->msg_peerrtrcredit);
1304 #endif
1305         if (rxpeer != NULL) {
1306                 msg->msg_rxpeer = NULL;
1307                 lnet_peer_decref_locked(rxpeer);
1308         }
1309 }
1310
1311 int
1312 lnet_send(lnet_nid_t src_nid, lnet_msg_t *msg)
1313 {
1314         lnet_nid_t        dst_nid = msg->msg_target.nid;
1315         lnet_ni_t        *src_ni;
1316         lnet_ni_t        *local_ni;
1317         lnet_remotenet_t *rnet;
1318         lnet_route_t     *route;
1319         lnet_route_t     *best_route;
1320         cfs_list_t       *tmp;
1321         lnet_peer_t      *lp;
1322         lnet_peer_t      *lp2;
1323         int               rc;
1324
1325         LASSERT (msg->msg_txpeer == NULL);
1326         LASSERT (!msg->msg_sending);
1327         LASSERT (!msg->msg_target_is_router);
1328         LASSERT (!msg->msg_receiving);
1329
1330         msg->msg_sending = 1;
1331
1332         /* NB! ni != NULL == interface pre-determined (ACK/REPLY) */
1333
1334         LNET_LOCK();
1335
1336         if (the_lnet.ln_shutdown) {
1337                 LNET_UNLOCK();
1338                 return -ESHUTDOWN;
1339         }
1340
1341         if (src_nid == LNET_NID_ANY) {
1342                 src_ni = NULL;
1343         } else {
1344                 src_ni = lnet_nid2ni_locked(src_nid);
1345                 if (src_ni == NULL) {
1346                         LNET_UNLOCK();
1347                         LCONSOLE_WARN("Can't send to %s: src %s is not a "
1348                                       "local nid\n", libcfs_nid2str(dst_nid),
1349                                       libcfs_nid2str(src_nid));
1350                         return -EINVAL;
1351                 }
1352                 LASSERT (!msg->msg_routing);
1353         }
1354
1355         /* Is this for someone on a local network? */
1356         local_ni = lnet_net2ni_locked(LNET_NIDNET(dst_nid));
1357
1358         if (local_ni != NULL) {
1359                 if (src_ni == NULL) {
1360                         src_ni = local_ni;
1361                         src_nid = src_ni->ni_nid;
1362                 } else if (src_ni == local_ni) {
1363                         lnet_ni_decref_locked(local_ni);
1364                 } else {
1365                         lnet_ni_decref_locked(local_ni);
1366                         lnet_ni_decref_locked(src_ni);
1367                         LNET_UNLOCK();
1368                         LCONSOLE_WARN("No route to %s via from %s\n",
1369                                       libcfs_nid2str(dst_nid),
1370                                       libcfs_nid2str(src_nid));
1371                         return -EINVAL;
1372                 }
1373
1374                 LASSERT (src_nid != LNET_NID_ANY);
1375
1376                 if (!msg->msg_routing)
1377                         msg->msg_hdr.src_nid = cpu_to_le64(src_nid);
1378
1379                 if (src_ni == the_lnet.ln_loni) {
1380                         /* No send credit hassles with LOLND */
1381                         LNET_UNLOCK();
1382                         lnet_ni_send(src_ni, msg);
1383                         lnet_ni_decref(src_ni);
1384                         return 0;
1385                 }
1386
1387                 rc = lnet_nid2peer_locked(&lp, dst_nid);
1388                 lnet_ni_decref_locked(src_ni);  /* lp has ref on src_ni; lose mine */
1389                 if (rc != 0) {
1390                         LNET_UNLOCK();
1391                         LCONSOLE_WARN("Error %d finding peer %s\n", rc,
1392                                       libcfs_nid2str(dst_nid));
1393                         /* ENOMEM or shutting down */
1394                         return rc;
1395                 }
1396                 LASSERT (lp->lp_ni == src_ni);
1397         } else {
1398 #ifndef __KERNEL__
1399                 LNET_UNLOCK();
1400
1401                 /* NB
1402                  * - once application finishes computation, check here to update
1403                  *   router states before it waits for pending IO in LNetEQPoll
1404                  * - recursion breaker: router checker sends no message
1405                  *   to remote networks */
1406                 if (the_lnet.ln_rc_state == LNET_RC_STATE_RUNNING)
1407                         lnet_router_checker();
1408
1409                 LNET_LOCK();
1410 #endif
1411                 /* sending to a remote network */
1412                 rnet = lnet_find_net_locked(LNET_NIDNET(dst_nid));
1413                 if (rnet == NULL) {
1414                         if (src_ni != NULL)
1415                                 lnet_ni_decref_locked(src_ni);
1416                         LNET_UNLOCK();
1417                         LCONSOLE_WARN("No route to %s\n",
1418                                       libcfs_id2str(msg->msg_target));
1419                         return -EHOSTUNREACH;
1420                 }
1421
1422                 /* Find the best gateway I can use */
1423                 lp = NULL;
1424                 best_route = NULL;
1425                 cfs_list_for_each(tmp, &rnet->lrn_routes) {
1426                         route = cfs_list_entry(tmp, lnet_route_t, lr_list);
1427                         lp2 = route->lr_gateway;
1428
1429                         if (lp2->lp_alive &&
1430                             lnet_router_down_ni(lp2, rnet->lrn_net) <= 0 &&
1431                             (src_ni == NULL || lp2->lp_ni == src_ni) &&
1432                             (lp == NULL ||
1433                              lnet_compare_routes(route, best_route) > 0)) {
1434                                 best_route = route;
1435                                 lp = lp2;
1436                         }
1437                 }
1438
1439                 if (lp == NULL) {
1440                         if (src_ni != NULL)
1441                                 lnet_ni_decref_locked(src_ni);
1442                         LNET_UNLOCK();
1443
1444                         LCONSOLE_WARN("No route to %s via %s "
1445                                       "(all routers down)\n",
1446                                       libcfs_id2str(msg->msg_target),
1447                                       libcfs_nid2str(src_nid));
1448                         return -EHOSTUNREACH;
1449                 }
1450
1451                 /* Place selected route at the end of the route list to ensure
1452                  * fairness; everything else being equal... */
1453                 cfs_list_del(&best_route->lr_list);
1454                 cfs_list_add_tail(&best_route->lr_list, &rnet->lrn_routes);
1455                 CDEBUG(D_NET, "Best route to %s via %s for %s %d\n",
1456                        libcfs_nid2str(dst_nid), libcfs_nid2str(lp->lp_nid),
1457                        lnet_msgtyp2str(msg->msg_type), msg->msg_len);
1458
1459                 if (src_ni == NULL) {
1460                         src_ni = lp->lp_ni;
1461                         src_nid = src_ni->ni_nid;
1462                 } else {
1463                         LASSERT (src_ni == lp->lp_ni);
1464                         lnet_ni_decref_locked(src_ni);
1465                 }
1466
1467                 lnet_peer_addref_locked(lp);
1468
1469                 LASSERT (src_nid != LNET_NID_ANY);
1470
1471                 if (!msg->msg_routing) {
1472                         /* I'm the source and now I know which NI to send on */
1473                         msg->msg_hdr.src_nid = cpu_to_le64(src_nid);
1474                 }
1475
1476                 msg->msg_target_is_router = 1;
1477                 msg->msg_target.nid = lp->lp_nid;
1478                 msg->msg_target.pid = LUSTRE_SRV_LNET_PID;
1479         }
1480
1481         /* 'lp' is our best choice of peer */
1482
1483         LASSERT (!msg->msg_peertxcredit);
1484         LASSERT (!msg->msg_txcredit);
1485         LASSERT (msg->msg_txpeer == NULL);
1486
1487         msg->msg_txpeer = lp;                   /* msg takes my ref on lp */
1488
1489         rc = lnet_post_send_locked(msg, 0);
1490         LNET_UNLOCK();
1491
1492         if (rc == EHOSTUNREACH)
1493                 return -EHOSTUNREACH;
1494
1495         if (rc == 0)
1496                 lnet_ni_send(src_ni, msg);
1497
1498         return 0;
1499 }
1500
1501 static void
1502 lnet_commit_md (lnet_libmd_t *md, lnet_msg_t *msg)
1503 {
1504         /* ALWAYS called holding the LNET_LOCK */
1505         /* Here, we commit the MD to a network OP by marking it busy and
1506          * decrementing its threshold.  Come what may, the network "owns"
1507          * the MD until a call to lnet_finalize() signals completion. */
1508         LASSERT (!msg->msg_routing);
1509
1510         msg->msg_md = md;
1511
1512         md->md_refcount++;
1513         if (md->md_threshold != LNET_MD_THRESH_INF) {
1514                 LASSERT (md->md_threshold > 0);
1515                 md->md_threshold--;
1516         }
1517
1518         the_lnet.ln_counters.msgs_alloc++;
1519         if (the_lnet.ln_counters.msgs_alloc >
1520             the_lnet.ln_counters.msgs_max)
1521                 the_lnet.ln_counters.msgs_max =
1522                         the_lnet.ln_counters.msgs_alloc;
1523
1524         LASSERT (!msg->msg_onactivelist);
1525         msg->msg_onactivelist = 1;
1526         cfs_list_add (&msg->msg_activelist, &the_lnet.ln_active_msgs);
1527 }
1528
1529 static void
1530 lnet_drop_message (lnet_ni_t *ni, void *private, unsigned int nob)
1531 {
1532         LNET_LOCK();
1533         the_lnet.ln_counters.drop_count++;
1534         the_lnet.ln_counters.drop_length += nob;
1535         LNET_UNLOCK();
1536
1537         lnet_ni_recv(ni, private, NULL, 0, 0, 0, nob);
1538 }
1539
1540 static void
1541 lnet_drop_delayed_put(lnet_msg_t *msg, char *reason)
1542 {
1543         lnet_process_id_t id = {0};
1544
1545         id.nid = msg->msg_hdr.src_nid;
1546         id.pid = msg->msg_hdr.src_pid;
1547
1548         LASSERT (msg->msg_md == NULL);
1549         LASSERT (msg->msg_delayed);
1550         LASSERT (msg->msg_rxpeer != NULL);
1551         LASSERT (msg->msg_hdr.type == LNET_MSG_PUT);
1552
1553         CWARN("Dropping delayed PUT from %s portal %d match "LPU64
1554               " offset %d length %d: %s\n",
1555               libcfs_id2str(id),
1556               msg->msg_hdr.msg.put.ptl_index,
1557               msg->msg_hdr.msg.put.match_bits,
1558               msg->msg_hdr.msg.put.offset,
1559               msg->msg_hdr.payload_length,
1560               reason);
1561
1562         /* NB I can't drop msg's ref on msg_rxpeer until after I've
1563          * called lnet_drop_message(), so I just hang onto msg as well
1564          * until that's done */
1565
1566         lnet_drop_message(msg->msg_rxpeer->lp_ni,
1567                           msg->msg_private, msg->msg_len);
1568
1569         LNET_LOCK();
1570
1571         lnet_peer_decref_locked(msg->msg_rxpeer);
1572         msg->msg_rxpeer = NULL;
1573
1574         lnet_msg_free(msg);
1575
1576         LNET_UNLOCK();
1577 }
1578
1579 /**
1580  * Turn on the lazy portal attribute. Use with caution!
1581  *
1582  * This portal attribute only affects incoming PUT requests to the portal,
1583  * and is off by default. By default, if there's no matching MD for an
1584  * incoming PUT request, it is simply dropped. With the lazy attribute on,
1585  * such requests are queued indefinitely until either a matching MD is
1586  * posted to the portal or the lazy attribute is turned off.
1587  *
1588  * It would prevent dropped requests, however it should be regarded as the
1589  * last line of defense - i.e. users must keep a close watch on active
1590  * buffers on a lazy portal and once it becomes too low post more buffers as
1591  * soon as possible. This is because delayed requests usually have detrimental
1592  * effects on underlying network connections. A few delayed requests often
1593  * suffice to bring an underlying connection to a complete halt, due to flow
1594  * control mechanisms.
1595  *
1596  * There's also a DOS attack risk. If users don't post match-all MDs on a
1597  * lazy portal, a malicious peer can easily stop a service by sending some
1598  * PUT requests with match bits that won't match any MD. A routed server is
1599  * especially vulnerable since the connections to its neighbor routers are
1600  * shared among all clients.
1601  *
1602  * \param portal Index of the portal to enable the lazy attribute on.
1603  *
1604  * \retval 0       On success.
1605  * \retval -EINVAL If \a portal is not a valid index.
1606  */
1607 int
1608 LNetSetLazyPortal(int portal)
1609 {
1610         lnet_portal_t *ptl = &the_lnet.ln_portals[portal];
1611
1612         if (portal < 0 || portal >= the_lnet.ln_nportals)
1613                 return -EINVAL;
1614
1615         CDEBUG(D_NET, "Setting portal %d lazy\n", portal);
1616
1617         LNET_LOCK();
1618         lnet_portal_setopt(ptl, LNET_PTL_LAZY);
1619         LNET_UNLOCK();
1620
1621         return 0;
1622 }
1623
1624 /**
1625  * Turn off the lazy portal attribute. Delayed requests on the portal,
1626  * if any, will be all dropped when this function returns.
1627  *
1628  * \param portal Index of the portal to disable the lazy attribute on.
1629  *
1630  * \retval 0       On success.
1631  * \retval -EINVAL If \a portal is not a valid index.
1632  */
1633 int
1634 LNetClearLazyPortal(int portal)
1635 {
1636         cfs_list_t        zombies;
1637         lnet_portal_t    *ptl = &the_lnet.ln_portals[portal];
1638         lnet_msg_t       *msg;
1639
1640         if (portal < 0 || portal >= the_lnet.ln_nportals)
1641                 return -EINVAL;
1642
1643         LNET_LOCK();
1644
1645         if (!lnet_portal_is_lazy(ptl)) {
1646                 LNET_UNLOCK();
1647                 return 0;
1648         }
1649
1650         if (the_lnet.ln_shutdown)
1651                 CWARN ("Active lazy portal %d on exit\n", portal);
1652         else
1653                 CDEBUG (D_NET, "clearing portal %d lazy\n", portal);
1654
1655         /* grab all the blocked messages atomically */
1656         cfs_list_add(&zombies, &ptl->ptl_msgq);
1657         cfs_list_del_init(&ptl->ptl_msgq);
1658
1659         ptl->ptl_msgq_version++;
1660         lnet_portal_unsetopt(ptl, LNET_PTL_LAZY);
1661
1662         LNET_UNLOCK();
1663
1664         while (!cfs_list_empty(&zombies)) {
1665                 msg = cfs_list_entry(zombies.next, lnet_msg_t, msg_list);
1666                 cfs_list_del(&msg->msg_list);
1667
1668                 lnet_drop_delayed_put(msg, "Clearing lazy portal attr");
1669         }
1670
1671         return 0;
1672 }
1673
1674 static void
1675 lnet_recv_put(lnet_libmd_t *md, lnet_msg_t *msg, int delayed,
1676               unsigned int offset, unsigned int mlength)
1677 {
1678         lnet_hdr_t       *hdr = &msg->msg_hdr;
1679
1680         LNET_LOCK();
1681
1682         the_lnet.ln_counters.recv_count++;
1683         the_lnet.ln_counters.recv_length += mlength;
1684
1685         LNET_UNLOCK();
1686
1687         if (mlength != 0)
1688                 lnet_setpayloadbuffer(msg);
1689
1690         msg->msg_ev.type       = LNET_EVENT_PUT;
1691         msg->msg_ev.target.pid = hdr->dest_pid;
1692         msg->msg_ev.target.nid = hdr->dest_nid;
1693         msg->msg_ev.hdr_data   = hdr->msg.put.hdr_data;
1694
1695         /* Must I ACK?  If so I'll grab the ack_wmd out of the header and put
1696          * it back into the ACK during lnet_finalize() */
1697         msg->msg_ack = (!lnet_is_wire_handle_none(&hdr->msg.put.ack_wmd) &&
1698                         (md->md_options & LNET_MD_ACK_DISABLE) == 0);
1699
1700         lnet_ni_recv(msg->msg_rxpeer->lp_ni,
1701                      msg->msg_private,
1702                      msg, delayed, offset, mlength,
1703                      hdr->payload_length);
1704 }
1705
1706 /* called with LNET_LOCK held */
1707 void
1708 lnet_match_blocked_msg(lnet_libmd_t *md)
1709 {
1710         CFS_LIST_HEAD    (drops);
1711         CFS_LIST_HEAD    (matches);
1712         cfs_list_t       *tmp;
1713         cfs_list_t       *entry;
1714         lnet_msg_t       *msg;
1715         lnet_portal_t    *ptl;
1716         lnet_me_t        *me  = md->md_me;
1717
1718         LASSERT (me->me_portal < (unsigned int)the_lnet.ln_nportals);
1719
1720         ptl = &the_lnet.ln_portals[me->me_portal];
1721         if (!lnet_portal_is_lazy(ptl)) {
1722                 LASSERT (cfs_list_empty(&ptl->ptl_msgq));
1723                 return;
1724         }
1725
1726         LASSERT (md->md_refcount == 0); /* a brand new MD */
1727
1728         cfs_list_for_each_safe (entry, tmp, &ptl->ptl_msgq) {
1729                 int               rc;
1730                 int               index;
1731                 unsigned int      mlength;
1732                 unsigned int      offset;
1733                 lnet_hdr_t       *hdr;
1734                 lnet_process_id_t src;
1735
1736                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1737
1738                 LASSERT (msg->msg_delayed);
1739
1740                 hdr   = &msg->msg_hdr;
1741                 index = hdr->msg.put.ptl_index;
1742
1743                 src.nid = hdr->src_nid;
1744                 src.pid = hdr->src_pid;
1745
1746                 rc = lnet_try_match_md(index, LNET_MD_OP_PUT, src,
1747                                        hdr->payload_length,
1748                                        hdr->msg.put.offset,
1749                                        hdr->msg.put.match_bits,
1750                                        md, msg, &mlength, &offset);
1751
1752                 if (rc == LNET_MATCHMD_NONE)
1753                         continue;
1754
1755                 /* Hurrah! This _is_ a match */
1756                 cfs_list_del(&msg->msg_list);
1757                 ptl->ptl_msgq_version++;
1758
1759                 if (rc == LNET_MATCHMD_OK) {
1760                         cfs_list_add_tail(&msg->msg_list, &matches);
1761
1762                         CDEBUG(D_NET, "Resuming delayed PUT from %s portal %d "
1763                                "match "LPU64" offset %d length %d.\n",
1764                                libcfs_id2str(src),
1765                                hdr->msg.put.ptl_index,
1766                                hdr->msg.put.match_bits,
1767                                hdr->msg.put.offset,
1768                                hdr->payload_length);
1769                 } else {
1770                         LASSERT (rc == LNET_MATCHMD_DROP);
1771
1772                         cfs_list_add_tail(&msg->msg_list, &drops);
1773                 }
1774
1775                 if (lnet_md_exhausted(md))
1776                         break;
1777         }
1778
1779         LNET_UNLOCK();
1780
1781         cfs_list_for_each_safe (entry, tmp, &drops) {
1782                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1783
1784                 cfs_list_del(&msg->msg_list);
1785
1786                 lnet_drop_delayed_put(msg, "Bad match");
1787         }
1788
1789         cfs_list_for_each_safe (entry, tmp, &matches) {
1790                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1791
1792                 cfs_list_del(&msg->msg_list);
1793
1794                 /* md won't disappear under me, since each msg
1795                  * holds a ref on it */
1796                 lnet_recv_put(md, msg, 1,
1797                               msg->msg_ev.offset,
1798                               msg->msg_ev.mlength);
1799         }
1800
1801         LNET_LOCK();
1802 }
1803
1804 static int
1805 lnet_parse_put(lnet_ni_t *ni, lnet_msg_t *msg)
1806 {
1807         int               rc;
1808         int               index;
1809         __u64             version;
1810         lnet_hdr_t       *hdr = &msg->msg_hdr;
1811         unsigned int      rlength = hdr->payload_length;
1812         unsigned int      mlength = 0;
1813         unsigned int      offset = 0;
1814         lnet_process_id_t src= {0};
1815         lnet_libmd_t     *md;
1816         lnet_portal_t    *ptl;
1817
1818         src.nid = hdr->src_nid;
1819         src.pid = hdr->src_pid;
1820
1821         /* Convert put fields to host byte order */
1822         hdr->msg.put.match_bits = le64_to_cpu(hdr->msg.put.match_bits);
1823         hdr->msg.put.ptl_index = le32_to_cpu(hdr->msg.put.ptl_index);
1824         hdr->msg.put.offset = le32_to_cpu(hdr->msg.put.offset);
1825
1826         index = hdr->msg.put.ptl_index;
1827
1828         LNET_LOCK();
1829
1830  again:
1831         rc = lnet_match_md(index, LNET_MD_OP_PUT, src,
1832                            rlength, hdr->msg.put.offset,
1833                            hdr->msg.put.match_bits, msg,
1834                            &mlength, &offset, &md);
1835         switch (rc) {
1836         default:
1837                 LBUG();
1838
1839         case LNET_MATCHMD_OK:
1840                 LNET_UNLOCK();
1841                 lnet_recv_put(md, msg, msg->msg_delayed, offset, mlength);
1842                 return 0;
1843
1844         case LNET_MATCHMD_NONE:
1845                 ptl = &the_lnet.ln_portals[index];
1846                 version = ptl->ptl_ml_version;
1847
1848                 rc = 0;
1849                 if (!msg->msg_delayed)
1850                         rc = lnet_eager_recv_locked(msg);
1851
1852                 if (rc == 0 &&
1853                     !the_lnet.ln_shutdown &&
1854                     lnet_portal_is_lazy(ptl)) {
1855                         if (version != ptl->ptl_ml_version)
1856                                 goto again;
1857
1858                         cfs_list_add_tail(&msg->msg_list, &ptl->ptl_msgq);
1859                         ptl->ptl_msgq_version++;
1860                         LNET_UNLOCK();
1861
1862                         CDEBUG(D_NET, "Delaying PUT from %s portal %d match "
1863                                LPU64" offset %d length %d: no match \n",
1864                                libcfs_id2str(src), index,
1865                                hdr->msg.put.match_bits,
1866                                hdr->msg.put.offset, rlength);
1867                         return 0;
1868                 }
1869                 /* fall through */
1870
1871         case LNET_MATCHMD_DROP:
1872                 CNETERR("Dropping PUT from %s portal %d match "LPU64
1873                         " offset %d length %d: %d\n",
1874                         libcfs_id2str(src), index,
1875                         hdr->msg.put.match_bits,
1876                         hdr->msg.put.offset, rlength, rc);
1877                 LNET_UNLOCK();
1878
1879                 return ENOENT;          /* +ve: OK but no match */
1880         }
1881 }
1882
1883 static int
1884 lnet_parse_get(lnet_ni_t *ni, lnet_msg_t *msg, int rdma_get)
1885 {
1886         lnet_hdr_t        *hdr = &msg->msg_hdr;
1887         unsigned int       mlength = 0;
1888         unsigned int       offset = 0;
1889         lnet_process_id_t  src = {0};
1890         lnet_handle_wire_t reply_wmd;
1891         lnet_libmd_t      *md;
1892         int                rc;
1893
1894         src.nid = hdr->src_nid;
1895         src.pid = hdr->src_pid;
1896
1897         /* Convert get fields to host byte order */
1898         hdr->msg.get.match_bits = le64_to_cpu(hdr->msg.get.match_bits);
1899         hdr->msg.get.ptl_index = le32_to_cpu(hdr->msg.get.ptl_index);
1900         hdr->msg.get.sink_length = le32_to_cpu(hdr->msg.get.sink_length);
1901         hdr->msg.get.src_offset = le32_to_cpu(hdr->msg.get.src_offset);
1902
1903         LNET_LOCK();
1904
1905         rc = lnet_match_md(hdr->msg.get.ptl_index, LNET_MD_OP_GET, src,
1906                            hdr->msg.get.sink_length, hdr->msg.get.src_offset,
1907                            hdr->msg.get.match_bits, msg,
1908                            &mlength, &offset, &md);
1909         if (rc == LNET_MATCHMD_DROP) {
1910                 CNETERR("Dropping GET from %s portal %d match "LPU64
1911                         " offset %d length %d\n",
1912                         libcfs_id2str(src),
1913                         hdr->msg.get.ptl_index,
1914                         hdr->msg.get.match_bits,
1915                         hdr->msg.get.src_offset,
1916                         hdr->msg.get.sink_length);
1917                 LNET_UNLOCK();
1918                 return ENOENT;                  /* +ve: OK but no match */
1919         }
1920
1921         LASSERT (rc == LNET_MATCHMD_OK);
1922
1923         the_lnet.ln_counters.send_count++;
1924         the_lnet.ln_counters.send_length += mlength;
1925
1926         LNET_UNLOCK();
1927
1928         msg->msg_ev.type = LNET_EVENT_GET;
1929         msg->msg_ev.target.pid = hdr->dest_pid;
1930         msg->msg_ev.target.nid = hdr->dest_nid;
1931         msg->msg_ev.hdr_data = 0;
1932
1933         reply_wmd = hdr->msg.get.return_wmd;
1934
1935         lnet_prep_send(msg, LNET_MSG_REPLY, src, offset, mlength);
1936
1937         msg->msg_hdr.msg.reply.dst_wmd = reply_wmd;
1938
1939         if (rdma_get) {
1940                 /* The LND completes the REPLY from her recv procedure */
1941                 lnet_ni_recv(ni, msg->msg_private, msg, 0,
1942                              msg->msg_offset, msg->msg_len, msg->msg_len);
1943                 return 0;
1944         }
1945
1946         lnet_ni_recv(ni, msg->msg_private, NULL, 0, 0, 0, 0);
1947         msg->msg_receiving = 0;
1948
1949         rc = lnet_send(ni->ni_nid, msg);
1950         if (rc < 0) {
1951                 /* didn't get as far as lnet_ni_send() */
1952                 CERROR("%s: Unable to send REPLY for GET from %s: %d\n",
1953                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src), rc);
1954
1955                 lnet_finalize(ni, msg, rc);
1956         }
1957
1958         return 0;
1959 }
1960
1961 static int
1962 lnet_parse_reply(lnet_ni_t *ni, lnet_msg_t *msg)
1963 {
1964         void             *private = msg->msg_private;
1965         lnet_hdr_t       *hdr = &msg->msg_hdr;
1966         lnet_process_id_t src = {0};
1967         lnet_libmd_t     *md;
1968         int               rlength;
1969         int               mlength;
1970
1971         LNET_LOCK();
1972
1973         src.nid = hdr->src_nid;
1974         src.pid = hdr->src_pid;
1975
1976         /* NB handles only looked up by creator (no flips) */
1977         md = lnet_wire_handle2md(&hdr->msg.reply.dst_wmd);
1978         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
1979                 CNETERR("%s: Dropping REPLY from %s for %s "
1980                         "MD "LPX64"."LPX64"\n",
1981                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
1982                         (md == NULL) ? "invalid" : "inactive",
1983                         hdr->msg.reply.dst_wmd.wh_interface_cookie,
1984                         hdr->msg.reply.dst_wmd.wh_object_cookie);
1985                 if (md != NULL && md->md_me != NULL)
1986                         CERROR("REPLY MD also attached to portal %d\n",
1987                                md->md_me->me_portal);
1988
1989                 LNET_UNLOCK();
1990                 return ENOENT;                  /* +ve: OK but no match */
1991         }
1992
1993         LASSERT (md->md_offset == 0);
1994
1995         rlength = hdr->payload_length;
1996         mlength = MIN(rlength, (int)md->md_length);
1997
1998         if (mlength < rlength &&
1999             (md->md_options & LNET_MD_TRUNCATE) == 0) {
2000                 CNETERR("%s: Dropping REPLY from %s length %d "
2001                         "for MD "LPX64" would overflow (%d)\n",
2002                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2003                         rlength, hdr->msg.reply.dst_wmd.wh_object_cookie,
2004                         mlength);
2005                 LNET_UNLOCK();
2006                 return ENOENT;          /* +ve: OK but no match */
2007         }
2008
2009         CDEBUG(D_NET, "%s: Reply from %s of length %d/%d into md "LPX64"\n",
2010                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2011                mlength, rlength, hdr->msg.reply.dst_wmd.wh_object_cookie);
2012
2013         lnet_commit_md(md, msg);
2014
2015         if (mlength != 0)
2016                 lnet_setpayloadbuffer(msg);
2017
2018         msg->msg_ev.type = LNET_EVENT_REPLY;
2019         msg->msg_ev.target.pid = hdr->dest_pid;
2020         msg->msg_ev.target.nid = hdr->dest_nid;
2021         msg->msg_ev.initiator = src;
2022         msg->msg_ev.rlength = rlength;
2023         msg->msg_ev.mlength = mlength;
2024         msg->msg_ev.offset = 0;
2025
2026         lnet_md_deconstruct(md, &msg->msg_ev.md);
2027         lnet_md2handle(&msg->msg_ev.md_handle, md);
2028
2029         the_lnet.ln_counters.recv_count++;
2030         the_lnet.ln_counters.recv_length += mlength;
2031
2032         LNET_UNLOCK();
2033
2034         lnet_ni_recv(ni, private, msg, 0, 0, mlength, rlength);
2035         return 0;
2036 }
2037
2038 static int
2039 lnet_parse_ack(lnet_ni_t *ni, lnet_msg_t *msg)
2040 {
2041         lnet_hdr_t       *hdr = &msg->msg_hdr;
2042         lnet_process_id_t src = {0};
2043         lnet_libmd_t     *md;
2044
2045         src.nid = hdr->src_nid;
2046         src.pid = hdr->src_pid;
2047
2048         /* Convert ack fields to host byte order */
2049         hdr->msg.ack.match_bits = le64_to_cpu(hdr->msg.ack.match_bits);
2050         hdr->msg.ack.mlength = le32_to_cpu(hdr->msg.ack.mlength);
2051
2052         LNET_LOCK();
2053
2054         /* NB handles only looked up by creator (no flips) */
2055         md = lnet_wire_handle2md(&hdr->msg.ack.dst_wmd);
2056         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2057                 /* Don't moan; this is expected */
2058                 CDEBUG(D_NET,
2059                        "%s: Dropping ACK from %s to %s MD "LPX64"."LPX64"\n",
2060                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2061                        (md == NULL) ? "invalid" : "inactive",
2062                        hdr->msg.ack.dst_wmd.wh_interface_cookie,
2063                        hdr->msg.ack.dst_wmd.wh_object_cookie);
2064                 if (md != NULL && md->md_me != NULL)
2065                         CERROR("Source MD also attached to portal %d\n",
2066                                md->md_me->me_portal);
2067
2068                 LNET_UNLOCK();
2069                 return ENOENT;                  /* +ve! */
2070         }
2071
2072         CDEBUG(D_NET, "%s: ACK from %s into md "LPX64"\n",
2073                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2074                hdr->msg.ack.dst_wmd.wh_object_cookie);
2075
2076         lnet_commit_md(md, msg);
2077
2078         msg->msg_ev.type = LNET_EVENT_ACK;
2079         msg->msg_ev.target.pid = hdr->dest_pid;
2080         msg->msg_ev.target.nid = hdr->dest_nid;
2081         msg->msg_ev.initiator = src;
2082         msg->msg_ev.mlength = hdr->msg.ack.mlength;
2083         msg->msg_ev.match_bits = hdr->msg.ack.match_bits;
2084
2085         lnet_md_deconstruct(md, &msg->msg_ev.md);
2086         lnet_md2handle(&msg->msg_ev.md_handle, md);
2087
2088         the_lnet.ln_counters.recv_count++;
2089
2090         LNET_UNLOCK();
2091
2092         lnet_ni_recv(ni, msg->msg_private, msg, 0, 0, 0, msg->msg_len);
2093         return 0;
2094 }
2095
2096 char *
2097 lnet_msgtyp2str (int type)
2098 {
2099         switch (type) {
2100         case LNET_MSG_ACK:
2101                 return ("ACK");
2102         case LNET_MSG_PUT:
2103                 return ("PUT");
2104         case LNET_MSG_GET:
2105                 return ("GET");
2106         case LNET_MSG_REPLY:
2107                 return ("REPLY");
2108         case LNET_MSG_HELLO:
2109                 return ("HELLO");
2110         default:
2111                 return ("<UNKNOWN>");
2112         }
2113 }
2114
2115 void
2116 lnet_print_hdr(lnet_hdr_t * hdr)
2117 {
2118         lnet_process_id_t src = {0};
2119         lnet_process_id_t dst = {0};
2120         char *type_str = lnet_msgtyp2str (hdr->type);
2121
2122         src.nid = hdr->src_nid;
2123         src.pid = hdr->src_pid;
2124
2125         dst.nid = hdr->dest_nid;
2126         dst.pid = hdr->dest_pid;
2127
2128         CWARN("P3 Header at %p of type %s\n", hdr, type_str);
2129         CWARN("    From %s\n", libcfs_id2str(src));
2130         CWARN("    To   %s\n", libcfs_id2str(dst));
2131
2132         switch (hdr->type) {
2133         default:
2134                 break;
2135
2136         case LNET_MSG_PUT:
2137                 CWARN("    Ptl index %d, ack md "LPX64"."LPX64", "
2138                       "match bits "LPU64"\n",
2139                       hdr->msg.put.ptl_index,
2140                       hdr->msg.put.ack_wmd.wh_interface_cookie,
2141                       hdr->msg.put.ack_wmd.wh_object_cookie,
2142                       hdr->msg.put.match_bits);
2143                 CWARN("    Length %d, offset %d, hdr data "LPX64"\n",
2144                       hdr->payload_length, hdr->msg.put.offset,
2145                       hdr->msg.put.hdr_data);
2146                 break;
2147
2148         case LNET_MSG_GET:
2149                 CWARN("    Ptl index %d, return md "LPX64"."LPX64", "
2150                       "match bits "LPU64"\n", hdr->msg.get.ptl_index,
2151                       hdr->msg.get.return_wmd.wh_interface_cookie,
2152                       hdr->msg.get.return_wmd.wh_object_cookie,
2153                       hdr->msg.get.match_bits);
2154                 CWARN("    Length %d, src offset %d\n",
2155                       hdr->msg.get.sink_length,
2156                       hdr->msg.get.src_offset);
2157                 break;
2158
2159         case LNET_MSG_ACK:
2160                 CWARN("    dst md "LPX64"."LPX64", "
2161                       "manipulated length %d\n",
2162                       hdr->msg.ack.dst_wmd.wh_interface_cookie,
2163                       hdr->msg.ack.dst_wmd.wh_object_cookie,
2164                       hdr->msg.ack.mlength);
2165                 break;
2166
2167         case LNET_MSG_REPLY:
2168                 CWARN("    dst md "LPX64"."LPX64", "
2169                       "length %d\n",
2170                       hdr->msg.reply.dst_wmd.wh_interface_cookie,
2171                       hdr->msg.reply.dst_wmd.wh_object_cookie,
2172                       hdr->payload_length);
2173         }
2174
2175 }
2176
2177 int
2178 lnet_parse(lnet_ni_t *ni, lnet_hdr_t *hdr, lnet_nid_t from_nid,
2179            void *private, int rdma_req)
2180 {
2181         int            rc = 0;
2182         int            for_me;
2183         lnet_msg_t    *msg;
2184         lnet_pid_t     dest_pid;
2185         lnet_nid_t     dest_nid;
2186         lnet_nid_t     src_nid;
2187         __u32          payload_length;
2188         __u32          type;
2189
2190         LASSERT (!cfs_in_interrupt ());
2191
2192         type = le32_to_cpu(hdr->type);
2193         src_nid = le64_to_cpu(hdr->src_nid);
2194         dest_nid = le64_to_cpu(hdr->dest_nid);
2195         dest_pid = le32_to_cpu(hdr->dest_pid);
2196         payload_length = le32_to_cpu(hdr->payload_length);
2197
2198         for_me = (ni->ni_nid == dest_nid);
2199
2200         switch (type) {
2201         case LNET_MSG_ACK:
2202         case LNET_MSG_GET:
2203                 if (payload_length > 0) {
2204                         CERROR("%s, src %s: bad %s payload %d (0 expected)\n",
2205                                libcfs_nid2str(from_nid),
2206                                libcfs_nid2str(src_nid),
2207                                lnet_msgtyp2str(type), payload_length);
2208                         return -EPROTO;
2209                 }
2210                 break;
2211
2212         case LNET_MSG_PUT:
2213         case LNET_MSG_REPLY:
2214                 if (payload_length > (__u32)(for_me ? LNET_MAX_PAYLOAD : LNET_MTU)) {
2215                         CERROR("%s, src %s: bad %s payload %d "
2216                                "(%d max expected)\n",
2217                                libcfs_nid2str(from_nid),
2218                                libcfs_nid2str(src_nid),
2219                                lnet_msgtyp2str(type),
2220                                payload_length,
2221                                for_me ? LNET_MAX_PAYLOAD : LNET_MTU);
2222                         return -EPROTO;
2223                 }
2224                 break;
2225
2226         default:
2227                 CERROR("%s, src %s: Bad message type 0x%x\n",
2228                        libcfs_nid2str(from_nid),
2229                        libcfs_nid2str(src_nid), type);
2230                 return -EPROTO;
2231         }
2232
2233         if (the_lnet.ln_routing) {
2234                 cfs_time_t now = cfs_time_current();
2235
2236                 LNET_LOCK();
2237
2238                 ni->ni_last_alive = now;
2239                 if (ni->ni_status != NULL &&
2240                     ni->ni_status->ns_status == LNET_NI_STATUS_DOWN)
2241                         ni->ni_status->ns_status = LNET_NI_STATUS_UP;
2242
2243                 LNET_UNLOCK();
2244         }
2245
2246         /* Regard a bad destination NID as a protocol error.  Senders should
2247          * know what they're doing; if they don't they're misconfigured, buggy
2248          * or malicious so we chop them off at the knees :) */
2249
2250         if (!for_me) {
2251                 if (LNET_NIDNET(dest_nid) == LNET_NIDNET(ni->ni_nid)) {
2252                         /* should have gone direct */
2253                         CERROR ("%s, src %s: Bad dest nid %s "
2254                                 "(should have been sent direct)\n",
2255                                 libcfs_nid2str(from_nid),
2256                                 libcfs_nid2str(src_nid),
2257                                 libcfs_nid2str(dest_nid));
2258                         return -EPROTO;
2259                 }
2260
2261                 if (lnet_islocalnid(dest_nid)) {
2262                         /* dest is another local NI; sender should have used
2263                          * this node's NID on its own network */
2264                         CERROR ("%s, src %s: Bad dest nid %s "
2265                                 "(it's my nid but on a different network)\n",
2266                                 libcfs_nid2str(from_nid),
2267                                 libcfs_nid2str(src_nid),
2268                                 libcfs_nid2str(dest_nid));
2269                         return -EPROTO;
2270                 }
2271
2272                 if (rdma_req && type == LNET_MSG_GET) {
2273                         CERROR ("%s, src %s: Bad optimized GET for %s "
2274                                 "(final destination must be me)\n",
2275                                 libcfs_nid2str(from_nid),
2276                                 libcfs_nid2str(src_nid),
2277                                 libcfs_nid2str(dest_nid));
2278                         return -EPROTO;
2279                 }
2280
2281                 if (!the_lnet.ln_routing) {
2282                         CERROR ("%s, src %s: Dropping message for %s "
2283                                 "(routing not enabled)\n",
2284                                 libcfs_nid2str(from_nid),
2285                                 libcfs_nid2str(src_nid),
2286                                 libcfs_nid2str(dest_nid));
2287                         goto drop;
2288                 }
2289         }
2290
2291         /* Message looks OK; we're not going to return an error, so we MUST
2292          * call back lnd_recv() come what may... */
2293
2294         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2295             fail_peer (src_nid, 0))             /* shall we now? */
2296         {
2297                 CERROR("%s, src %s: Dropping %s to simulate failure\n",
2298                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2299                        lnet_msgtyp2str(type));
2300                 goto drop;
2301         }
2302
2303         msg = lnet_msg_alloc();
2304         if (msg == NULL) {
2305                 CERROR("%s, src %s: Dropping %s (out of memory)\n",
2306                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2307                        lnet_msgtyp2str(type));
2308                 goto drop;
2309         }
2310
2311         /* msg zeroed in lnet_msg_alloc; i.e. flags all clear, pointers NULL etc */
2312
2313         msg->msg_type = type;
2314         msg->msg_private = private;
2315         msg->msg_receiving = 1;
2316         msg->msg_len = msg->msg_wanted = payload_length;
2317         msg->msg_offset = 0;
2318         msg->msg_hdr = *hdr;
2319
2320         LNET_LOCK();
2321         rc = lnet_nid2peer_locked(&msg->msg_rxpeer, from_nid);
2322         if (rc != 0) {
2323                 LNET_UNLOCK();
2324                 CERROR("%s, src %s: Dropping %s "
2325                        "(error %d looking up sender)\n",
2326                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2327                        lnet_msgtyp2str(type), rc);
2328                 goto free_drop;
2329         }
2330         LNET_UNLOCK();
2331
2332 #ifndef __KERNEL__
2333         LASSERT (for_me);
2334 #else
2335         if (!for_me) {
2336                 msg->msg_target.pid = dest_pid;
2337                 msg->msg_target.nid = dest_nid;
2338                 msg->msg_routing = 1;
2339                 msg->msg_offset = 0;
2340
2341                 LNET_LOCK();
2342                 if (msg->msg_rxpeer->lp_rtrcredits <= 0 ||
2343                     lnet_msg2bufpool(msg)->rbp_credits <= 0) {
2344                         rc = lnet_eager_recv_locked(msg);
2345                         if (rc != 0) {
2346                                 LNET_UNLOCK();
2347                                 goto free_drop;
2348                         }
2349                 }
2350                 lnet_commit_routedmsg(msg);
2351                 rc = lnet_post_routed_recv_locked(msg, 0);
2352                 LNET_UNLOCK();
2353
2354                 if (rc == 0)
2355                         lnet_ni_recv(ni, msg->msg_private, msg, 0,
2356                                      0, payload_length, payload_length);
2357                 return 0;
2358         }
2359 #endif
2360         /* convert common msg->hdr fields to host byteorder */
2361         msg->msg_hdr.type = type;
2362         msg->msg_hdr.src_nid = src_nid;
2363         msg->msg_hdr.src_pid = le32_to_cpu(msg->msg_hdr.src_pid);
2364         msg->msg_hdr.dest_nid = dest_nid;
2365         msg->msg_hdr.dest_pid = dest_pid;
2366         msg->msg_hdr.payload_length = payload_length;
2367
2368         msg->msg_ev.sender = from_nid;
2369
2370         switch (type) {
2371         case LNET_MSG_ACK:
2372                 rc = lnet_parse_ack(ni, msg);
2373                 break;
2374         case LNET_MSG_PUT:
2375                 rc = lnet_parse_put(ni, msg);
2376                 break;
2377         case LNET_MSG_GET:
2378                 rc = lnet_parse_get(ni, msg, rdma_req);
2379                 break;
2380         case LNET_MSG_REPLY:
2381                 rc = lnet_parse_reply(ni, msg);
2382                 break;
2383         default:
2384                 LASSERT(0);
2385                 goto free_drop;  /* prevent an unused label if !kernel */
2386         }
2387
2388         if (rc == 0)
2389                 return 0;
2390
2391         LASSERT (rc == ENOENT);
2392
2393  free_drop:
2394         LASSERT (msg->msg_md == NULL);
2395         LNET_LOCK();
2396         if (msg->msg_rxpeer != NULL) {
2397                 lnet_peer_decref_locked(msg->msg_rxpeer);
2398                 msg->msg_rxpeer = NULL;
2399         }
2400         lnet_msg_free(msg);                     /* expects LNET_LOCK held */
2401         LNET_UNLOCK();
2402
2403  drop:
2404         lnet_drop_message(ni, private, payload_length);
2405         return 0;
2406 }
2407
2408 /**
2409  * Initiate an asynchronous PUT operation.
2410  *
2411  * There are several events associated with a PUT: completion of the send on
2412  * the initiator node (LNET_EVENT_SEND), and when the send completes
2413  * successfully, the receipt of an acknowledgment (LNET_EVENT_ACK) indicating
2414  * that the operation was accepted by the target. The event LNET_EVENT_PUT is
2415  * used at the target node to indicate the completion of incoming data
2416  * delivery.
2417  *
2418  * The local events will be logged in the EQ associated with the MD pointed to
2419  * by \a mdh handle. Using a MD without an associated EQ results in these
2420  * events being discarded. In this case, the caller must have another
2421  * mechanism (e.g., a higher level protocol) for determining when it is safe
2422  * to modify the memory region associated with the MD.
2423  *
2424  * Note that LNet does not guarantee the order of LNET_EVENT_SEND and
2425  * LNET_EVENT_ACK, though intuitively ACK should happen after SEND.
2426  *
2427  * \param self Indicates the NID of a local interface through which to send
2428  * the PUT request. Use LNET_NID_ANY to let LNet choose one by itself.
2429  * \param mdh A handle for the MD that describes the memory to be sent. The MD
2430  * must be "free floating" (See LNetMDBind()).
2431  * \param ack Controls whether an acknowledgment is requested.
2432  * Acknowledgments are only sent when they are requested by the initiating
2433  * process and the target MD enables them.
2434  * \param target A process identifier for the target process.
2435  * \param portal The index in the \a target's portal table.
2436  * \param match_bits The match bits to use for MD selection at the target
2437  * process.
2438  * \param offset The offset into the target MD (only used when the target
2439  * MD has the LNET_MD_MANAGE_REMOTE option set).
2440  * \param hdr_data 64 bits of user data that can be included in the message
2441  * header. This data is written to an event queue entry at the target if an
2442  * EQ is present on the matching MD.
2443  *
2444  * \retval  0      Success, and only in this case events will be generated
2445  * and logged to EQ (if it exists).
2446  * \retval -EIO    Simulated failure.
2447  * \retval -ENOMEM Memory allocation failure.
2448  * \retval -ENOENT Invalid MD object.
2449  *
2450  * \see lnet_event_t::hdr_data and lnet_event_kind_t.
2451  */
2452 int
2453 LNetPut(lnet_nid_t self, lnet_handle_md_t mdh, lnet_ack_req_t ack,
2454         lnet_process_id_t target, unsigned int portal,
2455         __u64 match_bits, unsigned int offset,
2456         __u64 hdr_data)
2457 {
2458         lnet_msg_t       *msg;
2459         lnet_libmd_t     *md;
2460         int               rc;
2461
2462         LASSERT (the_lnet.ln_init);
2463         LASSERT (the_lnet.ln_refcount > 0);
2464
2465         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2466             fail_peer (target.nid, 1))          /* shall we now? */
2467         {
2468                 CERROR("Dropping PUT to %s: simulated failure\n",
2469                        libcfs_id2str(target));
2470                 return -EIO;
2471         }
2472
2473         msg = lnet_msg_alloc();
2474         if (msg == NULL) {
2475                 CERROR("Dropping PUT to %s: ENOMEM on lnet_msg_t\n",
2476                        libcfs_id2str(target));
2477                 return -ENOMEM;
2478         }
2479         msg->msg_vmflush = !!cfs_memory_pressure_get();
2480
2481         LNET_LOCK();
2482
2483         md = lnet_handle2md(&mdh);
2484         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2485                 lnet_msg_free(msg);
2486
2487                 CERROR("Dropping PUT ("LPU64":%d:%s): MD (%d) invalid\n",
2488                        match_bits, portal, libcfs_id2str(target),
2489                        md == NULL ? -1 : md->md_threshold);
2490                 if (md != NULL && md->md_me != NULL)
2491                         CERROR("Source MD also attached to portal %d\n",
2492                                md->md_me->me_portal);
2493
2494                 LNET_UNLOCK();
2495                 return -ENOENT;
2496         }
2497
2498         CDEBUG(D_NET, "LNetPut -> %s\n", libcfs_id2str(target));
2499
2500         lnet_commit_md(md, msg);
2501
2502         lnet_prep_send(msg, LNET_MSG_PUT, target, 0, md->md_length);
2503
2504         msg->msg_hdr.msg.put.match_bits = cpu_to_le64(match_bits);
2505         msg->msg_hdr.msg.put.ptl_index = cpu_to_le32(portal);
2506         msg->msg_hdr.msg.put.offset = cpu_to_le32(offset);
2507         msg->msg_hdr.msg.put.hdr_data = hdr_data;
2508
2509         /* NB handles only looked up by creator (no flips) */
2510         if (ack == LNET_ACK_REQ) {
2511                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2512                         the_lnet.ln_interface_cookie;
2513                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2514                         md->md_lh.lh_cookie;
2515         } else {
2516                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2517                         LNET_WIRE_HANDLE_COOKIE_NONE;
2518                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2519                         LNET_WIRE_HANDLE_COOKIE_NONE;
2520         }
2521
2522         msg->msg_ev.type = LNET_EVENT_SEND;
2523         msg->msg_ev.initiator.nid = LNET_NID_ANY;
2524         msg->msg_ev.initiator.pid = the_lnet.ln_pid;
2525         msg->msg_ev.target = target;
2526         msg->msg_ev.sender = LNET_NID_ANY;
2527         msg->msg_ev.pt_index = portal;
2528         msg->msg_ev.match_bits = match_bits;
2529         msg->msg_ev.rlength = md->md_length;
2530         msg->msg_ev.mlength = md->md_length;
2531         msg->msg_ev.offset = offset;
2532         msg->msg_ev.hdr_data = hdr_data;
2533
2534         lnet_md_deconstruct(md, &msg->msg_ev.md);
2535         lnet_md2handle(&msg->msg_ev.md_handle, md);
2536
2537         the_lnet.ln_counters.send_count++;
2538         the_lnet.ln_counters.send_length += md->md_length;
2539
2540         LNET_UNLOCK();
2541
2542         rc = lnet_send(self, msg);
2543         if (rc != 0) {
2544                 CNETERR( "Error sending PUT to %s: %d\n",
2545                        libcfs_id2str(target), rc);
2546                 lnet_finalize (NULL, msg, rc);
2547         }
2548
2549         /* completion will be signalled by an event */
2550         return 0;
2551 }
2552
2553 lnet_msg_t *
2554 lnet_create_reply_msg (lnet_ni_t *ni, lnet_msg_t *getmsg)
2555 {
2556         /* The LND can DMA direct to the GET md (i.e. no REPLY msg).  This
2557          * returns a msg for the LND to pass to lnet_finalize() when the sink
2558          * data has been received.
2559          *
2560          * CAVEAT EMPTOR: 'getmsg' is the original GET, which is freed when
2561          * lnet_finalize() is called on it, so the LND must call this first */
2562
2563         lnet_msg_t        *msg = lnet_msg_alloc();
2564         lnet_libmd_t      *getmd = getmsg->msg_md;
2565         lnet_process_id_t  peer_id = getmsg->msg_target;
2566
2567         LASSERT (!getmsg->msg_target_is_router);
2568         LASSERT (!getmsg->msg_routing);
2569
2570         LNET_LOCK();
2571
2572         LASSERT (getmd->md_refcount > 0);
2573
2574         if (msg == NULL) {
2575                 CERROR ("%s: Dropping REPLY from %s: can't allocate msg\n",
2576                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id));
2577                 goto drop;
2578         }
2579
2580         if (getmd->md_threshold == 0) {
2581                 CERROR ("%s: Dropping REPLY from %s for inactive MD %p\n",
2582                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id),
2583                         getmd);
2584                 goto drop_msg;
2585         }
2586
2587         LASSERT (getmd->md_offset == 0);
2588
2589         CDEBUG(D_NET, "%s: Reply from %s md %p\n",
2590                libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id), getmd);
2591
2592         lnet_commit_md (getmd, msg);
2593
2594         msg->msg_type = LNET_MSG_GET; /* flag this msg as an "optimized" GET */
2595
2596         msg->msg_ev.type = LNET_EVENT_REPLY;
2597         msg->msg_ev.initiator = peer_id;
2598         msg->msg_ev.sender = peer_id.nid;  /* optimized GETs can't be routed */
2599         msg->msg_ev.rlength = msg->msg_ev.mlength = getmd->md_length;
2600         msg->msg_ev.offset = 0;
2601
2602         lnet_md_deconstruct(getmd, &msg->msg_ev.md);
2603         lnet_md2handle(&msg->msg_ev.md_handle, getmd);
2604
2605         the_lnet.ln_counters.recv_count++;
2606         the_lnet.ln_counters.recv_length += getmd->md_length;
2607
2608         LNET_UNLOCK();
2609
2610         return msg;
2611
2612  drop_msg:
2613         lnet_msg_free(msg);
2614  drop:
2615         the_lnet.ln_counters.drop_count++;
2616         the_lnet.ln_counters.drop_length += getmd->md_length;
2617
2618         LNET_UNLOCK ();
2619
2620         return NULL;
2621 }
2622
2623 void
2624 lnet_set_reply_msg_len(lnet_ni_t *ni, lnet_msg_t *reply, unsigned int len)
2625 {
2626         /* Set the REPLY length, now the RDMA that elides the REPLY message has
2627          * completed and I know it. */
2628         LASSERT (reply != NULL);
2629         LASSERT (reply->msg_type == LNET_MSG_GET);
2630         LASSERT (reply->msg_ev.type == LNET_EVENT_REPLY);
2631
2632         /* NB I trusted my peer to RDMA.  If she tells me she's written beyond
2633          * the end of my buffer, I might as well be dead. */
2634         LASSERT (len <= reply->msg_ev.mlength);
2635
2636         reply->msg_ev.mlength = len;
2637 }
2638
2639 /**
2640  * Initiate an asynchronous GET operation.
2641  *
2642  * On the initiator node, an LNET_EVENT_SEND is logged when the GET request
2643  * is sent, and an LNET_EVENT_REPLY is logged when the data returned from
2644  * the target node in the REPLY has been written to local MD.
2645  *
2646  * On the target node, an LNET_EVENT_GET is logged when the GET request
2647  * arrives and is accepted into a MD.
2648  *
2649  * \param self,target,portal,match_bits,offset See the discussion in LNetPut().
2650  * \param mdh A handle for the MD that describes the memory into which the
2651  * requested data will be received. The MD must be "free floating" (See LNetMDBind()).
2652  *
2653  * \retval  0      Success, and only in this case events will be generated
2654  * and logged to EQ (if it exists) of the MD.
2655  * \retval -EIO    Simulated failure.
2656  * \retval -ENOMEM Memory allocation failure.
2657  * \retval -ENOENT Invalid MD object.
2658  */
2659 int
2660 LNetGet(lnet_nid_t self, lnet_handle_md_t mdh,
2661         lnet_process_id_t target, unsigned int portal,
2662         __u64 match_bits, unsigned int offset)
2663 {
2664         lnet_msg_t       *msg;
2665         lnet_libmd_t     *md;
2666         int               rc;
2667
2668         LASSERT (the_lnet.ln_init);
2669         LASSERT (the_lnet.ln_refcount > 0);
2670
2671         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2672             fail_peer (target.nid, 1))          /* shall we now? */
2673         {
2674                 CERROR("Dropping GET to %s: simulated failure\n",
2675                        libcfs_id2str(target));
2676                 return -EIO;
2677         }
2678
2679         msg = lnet_msg_alloc();
2680         if (msg == NULL) {
2681                 CERROR("Dropping GET to %s: ENOMEM on lnet_msg_t\n",
2682                        libcfs_id2str(target));
2683                 return -ENOMEM;
2684         }
2685
2686         LNET_LOCK();
2687
2688         md = lnet_handle2md(&mdh);
2689         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2690                 lnet_msg_free(msg);
2691
2692                 CERROR("Dropping GET ("LPU64":%d:%s): MD (%d) invalid\n",
2693                        match_bits, portal, libcfs_id2str(target),
2694                        md == NULL ? -1 : md->md_threshold);
2695                 if (md != NULL && md->md_me != NULL)
2696                         CERROR("REPLY MD also attached to portal %d\n",
2697                                md->md_me->me_portal);
2698
2699                 LNET_UNLOCK();
2700                 return -ENOENT;
2701         }
2702
2703         CDEBUG(D_NET, "LNetGet -> %s\n", libcfs_id2str(target));
2704
2705         lnet_commit_md(md, msg);
2706
2707         lnet_prep_send(msg, LNET_MSG_GET, target, 0, 0);
2708
2709         msg->msg_hdr.msg.get.match_bits = cpu_to_le64(match_bits);
2710         msg->msg_hdr.msg.get.ptl_index = cpu_to_le32(portal);
2711         msg->msg_hdr.msg.get.src_offset = cpu_to_le32(offset);
2712         msg->msg_hdr.msg.get.sink_length = cpu_to_le32(md->md_length);
2713
2714         /* NB handles only looked up by creator (no flips) */
2715         msg->msg_hdr.msg.get.return_wmd.wh_interface_cookie =
2716                 the_lnet.ln_interface_cookie;
2717         msg->msg_hdr.msg.get.return_wmd.wh_object_cookie =
2718                 md->md_lh.lh_cookie;
2719
2720         msg->msg_ev.type = LNET_EVENT_SEND;
2721         msg->msg_ev.initiator.nid = LNET_NID_ANY;
2722         msg->msg_ev.initiator.pid = the_lnet.ln_pid;
2723         msg->msg_ev.target = target;
2724         msg->msg_ev.sender = LNET_NID_ANY;
2725         msg->msg_ev.pt_index = portal;
2726         msg->msg_ev.match_bits = match_bits;
2727         msg->msg_ev.rlength = md->md_length;
2728         msg->msg_ev.mlength = md->md_length;
2729         msg->msg_ev.offset = offset;
2730         msg->msg_ev.hdr_data = 0;
2731
2732         lnet_md_deconstruct(md, &msg->msg_ev.md);
2733         lnet_md2handle(&msg->msg_ev.md_handle, md);
2734
2735         the_lnet.ln_counters.send_count++;
2736
2737         LNET_UNLOCK();
2738
2739         rc = lnet_send(self, msg);
2740         if (rc < 0) {
2741                 CNETERR( "Error sending GET to %s: %d\n",
2742                        libcfs_id2str(target), rc);
2743                 lnet_finalize (NULL, msg, rc);
2744         }
2745
2746         /* completion will be signalled by an event */
2747         return 0;
2748 }
2749
2750 /**
2751  * Calculate distance to node at \a dstnid.
2752  *
2753  * \param dstnid Target NID.
2754  * \param srcnidp If not NULL, NID of the local interface to reach \a dstnid
2755  * is saved here.
2756  * \param orderp If not NULL, order of the route to reach \a dstnid is saved
2757  * here.
2758  *
2759  * \retval 0 If \a dstnid belongs to a local interface, and reserved option
2760  * local_nid_dist_zero is set, which is the default.
2761  * \retval positives Distance to target NID, i.e. number of hops plus one.
2762  * \retval -EHOSTUNREACH If \a dstnid is not reachable.
2763  */
2764 int
2765 LNetDist (lnet_nid_t dstnid, lnet_nid_t *srcnidp, __u32 *orderp)
2766 {
2767         cfs_list_t       *e;
2768         lnet_ni_t        *ni;
2769         lnet_remotenet_t *rnet;
2770         __u32             dstnet = LNET_NIDNET(dstnid);
2771         int               hops;
2772         __u32             order = 2;
2773
2774         /* if !local_nid_dist_zero, I don't return a distance of 0 ever
2775          * (when lustre sees a distance of 0, it substitutes 0@lo), so I
2776          * keep order 0 free for 0@lo and order 1 free for a local NID
2777          * match */
2778
2779         LASSERT (the_lnet.ln_init);
2780         LASSERT (the_lnet.ln_refcount > 0);
2781
2782         LNET_LOCK();
2783
2784         cfs_list_for_each (e, &the_lnet.ln_nis) {
2785                 ni = cfs_list_entry(e, lnet_ni_t, ni_list);
2786
2787                 if (ni->ni_nid == dstnid) {
2788                         if (srcnidp != NULL)
2789                                 *srcnidp = dstnid;
2790                         if (orderp != NULL) {
2791                                 if (LNET_NETTYP(LNET_NIDNET(dstnid)) == LOLND)
2792                                         *orderp = 0;
2793                                 else
2794                                         *orderp = 1;
2795                         }
2796                         LNET_UNLOCK();
2797
2798                         return local_nid_dist_zero ? 0 : 1;
2799                 }
2800
2801                 if (LNET_NIDNET(ni->ni_nid) == dstnet) {
2802                         if (srcnidp != NULL)
2803                                 *srcnidp = ni->ni_nid;
2804                         if (orderp != NULL)
2805                                 *orderp = order;
2806                         LNET_UNLOCK();
2807                         return 1;
2808                 }
2809
2810                 order++;
2811         }
2812
2813         cfs_list_for_each (e, &the_lnet.ln_remote_nets) {
2814                 rnet = cfs_list_entry(e, lnet_remotenet_t, lrn_list);
2815
2816                 if (rnet->lrn_net == dstnet) {
2817                         lnet_route_t *route;
2818                         lnet_route_t *shortest = NULL;
2819
2820                         LASSERT (!cfs_list_empty(&rnet->lrn_routes));
2821
2822                         cfs_list_for_each_entry(route, &rnet->lrn_routes,
2823                                                 lr_list) {
2824                                 if (shortest == NULL ||
2825                                     route->lr_hops < shortest->lr_hops)
2826                                         shortest = route;
2827                         }
2828
2829                         LASSERT (shortest != NULL);
2830                         hops = shortest->lr_hops;
2831                         if (srcnidp != NULL)
2832                                 *srcnidp = shortest->lr_gateway->lp_ni->ni_nid;
2833                         if (orderp != NULL)
2834                                 *orderp = order;
2835                         LNET_UNLOCK();
2836                         return hops + 1;
2837                 }
2838                 order++;
2839         }
2840
2841         LNET_UNLOCK();
2842         return -EHOSTUNREACH;
2843 }
2844
2845 /**
2846  * Set the number of asynchronous messages expected from a target process.
2847  *
2848  * This function is only meaningful for userspace callers. It's a no-op when
2849  * called from kernel.
2850  *
2851  * Asynchronous messages are those that can come from a target when the
2852  * userspace process is not waiting for IO to complete; e.g., AST callbacks
2853  * from Lustre servers. Specifying the expected number of such messages
2854  * allows them to be eagerly received when user process is not running in
2855  * LNet; otherwise network errors may occur.
2856  *
2857  * \param id Process ID of the target process.
2858  * \param nasync Number of asynchronous messages expected from the target.
2859  *
2860  * \return 0 on success, and an error code otherwise.
2861  */
2862 int
2863 LNetSetAsync(lnet_process_id_t id, int nasync)
2864 {
2865 #ifdef __KERNEL__
2866         return 0;
2867 #else
2868         lnet_ni_t        *ni;
2869         lnet_remotenet_t *rnet;
2870         cfs_list_t       *tmp;
2871         lnet_route_t     *route;
2872         lnet_nid_t       *nids;
2873         int               nnids;
2874         int               maxnids = 256;
2875         int               rc = 0;
2876         int               rc2;
2877
2878         /* Target on a local network? */
2879         ni = lnet_net2ni(LNET_NIDNET(id.nid));
2880         if (ni != NULL) {
2881                 if (ni->ni_lnd->lnd_setasync != NULL)
2882                         rc = (ni->ni_lnd->lnd_setasync)(ni, id, nasync);
2883                 lnet_ni_decref(ni);
2884                 return rc;
2885         }
2886
2887         /* Target on a remote network: apply to routers */
2888  again:
2889         LIBCFS_ALLOC(nids, maxnids * sizeof(*nids));
2890         if (nids == NULL)
2891                 return -ENOMEM;
2892         nnids = 0;
2893
2894         /* Snapshot all the router NIDs */
2895         LNET_LOCK();
2896         rnet = lnet_find_net_locked(LNET_NIDNET(id.nid));
2897         if (rnet != NULL) {
2898                 cfs_list_for_each(tmp, &rnet->lrn_routes) {
2899                         if (nnids == maxnids) {
2900                                 LNET_UNLOCK();
2901                                 LIBCFS_FREE(nids, maxnids * sizeof(*nids));
2902                                 maxnids *= 2;
2903                                 goto again;
2904                         }
2905
2906                         route = cfs_list_entry(tmp, lnet_route_t, lr_list);
2907                         nids[nnids++] = route->lr_gateway->lp_nid;
2908                 }
2909         }
2910         LNET_UNLOCK();
2911
2912         /* set async on all the routers */
2913         while (nnids-- > 0) {
2914                 id.pid = LUSTRE_SRV_LNET_PID;
2915                 id.nid = nids[nnids];
2916
2917                 ni = lnet_net2ni(LNET_NIDNET(id.nid));
2918                 if (ni == NULL)
2919                         continue;
2920
2921                 if (ni->ni_lnd->lnd_setasync != NULL) {
2922                         rc2 = (ni->ni_lnd->lnd_setasync)(ni, id, nasync);
2923                         if (rc2 != 0)
2924                                 rc = rc2;
2925                 }
2926                 lnet_ni_decref(ni);
2927         }
2928
2929         LIBCFS_FREE(nids, maxnids * sizeof(*nids));
2930         return rc;
2931 #endif
2932 }