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                 LNET_UNLOCK();
1032
1033                 CDEBUG(D_NETERROR, "Dropping message for %s: peer not alive\n",
1034                        libcfs_id2str(msg->msg_target));
1035                 if (do_send)
1036                         lnet_finalize(ni, msg, -EHOSTUNREACH);
1037
1038                 LNET_LOCK();
1039                 return EHOSTUNREACH;
1040         }
1041
1042         if (!msg->msg_peertxcredit) {
1043                 LASSERT ((lp->lp_txcredits < 0) ==
1044                          !cfs_list_empty(&lp->lp_txq));
1045
1046                 msg->msg_peertxcredit = 1;
1047                 lp->lp_txqnob += msg->msg_len + sizeof(lnet_hdr_t);
1048                 lp->lp_txcredits--;
1049
1050                 if (lp->lp_txcredits < lp->lp_mintxcredits)
1051                         lp->lp_mintxcredits = lp->lp_txcredits;
1052
1053                 if (lp->lp_txcredits < 0) {
1054                         msg->msg_delayed = 1;
1055                         cfs_list_add_tail(&msg->msg_list, &lp->lp_txq);
1056                         return EAGAIN;
1057                 }
1058         }
1059
1060         if (!msg->msg_txcredit) {
1061                 LASSERT ((ni->ni_txcredits < 0) ==
1062                          !cfs_list_empty(&ni->ni_txq));
1063
1064                 msg->msg_txcredit = 1;
1065                 ni->ni_txcredits--;
1066
1067                 if (ni->ni_txcredits < ni->ni_mintxcredits)
1068                         ni->ni_mintxcredits = ni->ni_txcredits;
1069
1070                 if (ni->ni_txcredits < 0) {
1071                         msg->msg_delayed = 1;
1072                         cfs_list_add_tail(&msg->msg_list, &ni->ni_txq);
1073                         return EAGAIN;
1074                 }
1075         }
1076
1077         if (do_send) {
1078                 LNET_UNLOCK();
1079                 lnet_ni_send(ni, msg);
1080                 LNET_LOCK();
1081         }
1082         return 0;
1083 }
1084
1085 #ifdef __KERNEL__
1086 static void
1087 lnet_commit_routedmsg (lnet_msg_t *msg)
1088 {
1089         /* ALWAYS called holding the LNET_LOCK */
1090         LASSERT (msg->msg_routing);
1091
1092         the_lnet.ln_counters.msgs_alloc++;
1093         if (the_lnet.ln_counters.msgs_alloc >
1094             the_lnet.ln_counters.msgs_max)
1095                 the_lnet.ln_counters.msgs_max =
1096                         the_lnet.ln_counters.msgs_alloc;
1097
1098         the_lnet.ln_counters.route_count++;
1099         the_lnet.ln_counters.route_length += msg->msg_len;
1100
1101         LASSERT (!msg->msg_onactivelist);
1102         msg->msg_onactivelist = 1;
1103         cfs_list_add (&msg->msg_activelist, &the_lnet.ln_active_msgs);
1104 }
1105
1106 lnet_rtrbufpool_t *
1107 lnet_msg2bufpool(lnet_msg_t *msg)
1108 {
1109         lnet_rtrbufpool_t *rbp = &the_lnet.ln_rtrpools[0];
1110
1111         LASSERT (msg->msg_len <= LNET_MTU);
1112         while (msg->msg_len > (unsigned int)rbp->rbp_npages * CFS_PAGE_SIZE) {
1113                 rbp++;
1114                 LASSERT (rbp < &the_lnet.ln_rtrpools[LNET_NRBPOOLS]);
1115         }
1116
1117         return rbp;
1118 }
1119
1120 int
1121 lnet_post_routed_recv_locked (lnet_msg_t *msg, int do_recv)
1122 {
1123         /* lnet_parse is going to LNET_UNLOCK immediately after this, so it
1124          * sets do_recv FALSE and I don't do the unlock/send/lock bit.  I
1125          * return EAGAIN if msg blocked and 0 if received or OK to receive */
1126         lnet_peer_t         *lp = msg->msg_rxpeer;
1127         lnet_rtrbufpool_t   *rbp;
1128         lnet_rtrbuf_t       *rb;
1129
1130         LASSERT (msg->msg_iov == NULL);
1131         LASSERT (msg->msg_kiov == NULL);
1132         LASSERT (msg->msg_niov == 0);
1133         LASSERT (msg->msg_routing);
1134         LASSERT (msg->msg_receiving);
1135         LASSERT (!msg->msg_sending);
1136
1137         /* non-lnet_parse callers only send delayed messages */
1138         LASSERT (!do_recv || msg->msg_delayed);
1139
1140         if (!msg->msg_peerrtrcredit) {
1141                 LASSERT ((lp->lp_rtrcredits < 0) ==
1142                          !cfs_list_empty(&lp->lp_rtrq));
1143
1144                 msg->msg_peerrtrcredit = 1;
1145                 lp->lp_rtrcredits--;
1146                 if (lp->lp_rtrcredits < lp->lp_minrtrcredits)
1147                         lp->lp_minrtrcredits = lp->lp_rtrcredits;
1148
1149                 if (lp->lp_rtrcredits < 0) {
1150                         /* must have checked eager_recv before here */
1151                         LASSERT (msg->msg_delayed);
1152                         cfs_list_add_tail(&msg->msg_list, &lp->lp_rtrq);
1153                         return EAGAIN;
1154                 }
1155         }
1156
1157         rbp = lnet_msg2bufpool(msg);
1158
1159         if (!msg->msg_rtrcredit) {
1160                 LASSERT ((rbp->rbp_credits < 0) ==
1161                          !cfs_list_empty(&rbp->rbp_msgs));
1162
1163                 msg->msg_rtrcredit = 1;
1164                 rbp->rbp_credits--;
1165                 if (rbp->rbp_credits < rbp->rbp_mincredits)
1166                         rbp->rbp_mincredits = rbp->rbp_credits;
1167
1168                 if (rbp->rbp_credits < 0) {
1169                         /* must have checked eager_recv before here */
1170                         LASSERT (msg->msg_delayed);
1171                         cfs_list_add_tail(&msg->msg_list, &rbp->rbp_msgs);
1172                         return EAGAIN;
1173                 }
1174         }
1175
1176         LASSERT (!cfs_list_empty(&rbp->rbp_bufs));
1177         rb = cfs_list_entry(rbp->rbp_bufs.next, lnet_rtrbuf_t, rb_list);
1178         cfs_list_del(&rb->rb_list);
1179
1180         msg->msg_niov = rbp->rbp_npages;
1181         msg->msg_kiov = &rb->rb_kiov[0];
1182
1183         if (do_recv) {
1184                 LNET_UNLOCK();
1185                 lnet_ni_recv(lp->lp_ni, msg->msg_private, msg, 1,
1186                              0, msg->msg_len, msg->msg_len);
1187                 LNET_LOCK();
1188         }
1189         return 0;
1190 }
1191 #endif
1192
1193 void
1194 lnet_return_credits_locked (lnet_msg_t *msg)
1195 {
1196         lnet_peer_t       *txpeer = msg->msg_txpeer;
1197         lnet_peer_t       *rxpeer = msg->msg_rxpeer;
1198         lnet_msg_t        *msg2;
1199         lnet_ni_t         *ni;
1200
1201         if (msg->msg_txcredit) {
1202                 /* give back NI txcredits */
1203                 msg->msg_txcredit = 0;
1204                 ni = txpeer->lp_ni;
1205
1206                 LASSERT((ni->ni_txcredits < 0) == !cfs_list_empty(&ni->ni_txq));
1207
1208                 ni->ni_txcredits++;
1209                 if (ni->ni_txcredits <= 0) {
1210                         msg2 = cfs_list_entry(ni->ni_txq.next, lnet_msg_t,
1211                                               msg_list);
1212                         cfs_list_del(&msg2->msg_list);
1213
1214                         LASSERT(msg2->msg_txpeer->lp_ni == ni);
1215                         LASSERT(msg2->msg_delayed);
1216
1217                         (void) lnet_post_send_locked(msg2, 1);
1218                 }
1219         }
1220
1221         if (msg->msg_peertxcredit) {
1222                 /* give back peer txcredits */
1223                 msg->msg_peertxcredit = 0;
1224
1225                 LASSERT((txpeer->lp_txcredits < 0) ==
1226                         !cfs_list_empty(&txpeer->lp_txq));
1227
1228                 txpeer->lp_txqnob -= msg->msg_len + sizeof(lnet_hdr_t);
1229                 LASSERT (txpeer->lp_txqnob >= 0);
1230
1231                 txpeer->lp_txcredits++;
1232                 if (txpeer->lp_txcredits <= 0) {
1233                         msg2 = cfs_list_entry(txpeer->lp_txq.next,
1234                                               lnet_msg_t, msg_list);
1235                         cfs_list_del(&msg2->msg_list);
1236
1237                         LASSERT (msg2->msg_txpeer == txpeer);
1238                         LASSERT (msg2->msg_delayed);
1239
1240                         (void) lnet_post_send_locked(msg2, 1);
1241                 }
1242         }
1243
1244         if (txpeer != NULL) {
1245                 msg->msg_txpeer = NULL;
1246                 lnet_peer_decref_locked(txpeer);
1247         }
1248
1249 #ifdef __KERNEL__
1250         if (msg->msg_rtrcredit) {
1251                 /* give back global router credits */
1252                 lnet_rtrbuf_t     *rb;
1253                 lnet_rtrbufpool_t *rbp;
1254
1255                 /* NB If a msg ever blocks for a buffer in rbp_msgs, it stays
1256                  * there until it gets one allocated, or aborts the wait
1257                  * itself */
1258                 LASSERT (msg->msg_kiov != NULL);
1259
1260                 rb = cfs_list_entry(msg->msg_kiov, lnet_rtrbuf_t, rb_kiov[0]);
1261                 rbp = rb->rb_pool;
1262                 LASSERT (rbp == lnet_msg2bufpool(msg));
1263
1264                 msg->msg_kiov = NULL;
1265                 msg->msg_rtrcredit = 0;
1266
1267                 LASSERT((rbp->rbp_credits < 0) ==
1268                         !cfs_list_empty(&rbp->rbp_msgs));
1269                 LASSERT((rbp->rbp_credits > 0) ==
1270                         !cfs_list_empty(&rbp->rbp_bufs));
1271
1272                 cfs_list_add(&rb->rb_list, &rbp->rbp_bufs);
1273                 rbp->rbp_credits++;
1274                 if (rbp->rbp_credits <= 0) {
1275                         msg2 = cfs_list_entry(rbp->rbp_msgs.next,
1276                                               lnet_msg_t, msg_list);
1277                         cfs_list_del(&msg2->msg_list);
1278
1279                         (void) lnet_post_routed_recv_locked(msg2, 1);
1280                 }
1281         }
1282
1283         if (msg->msg_peerrtrcredit) {
1284                 /* give back peer router credits */
1285                 msg->msg_peerrtrcredit = 0;
1286
1287                 LASSERT((rxpeer->lp_rtrcredits < 0) ==
1288                         !cfs_list_empty(&rxpeer->lp_rtrq));
1289
1290                 rxpeer->lp_rtrcredits++;
1291                 if (rxpeer->lp_rtrcredits <= 0) {
1292                         msg2 = cfs_list_entry(rxpeer->lp_rtrq.next,
1293                                               lnet_msg_t, msg_list);
1294                         cfs_list_del(&msg2->msg_list);
1295
1296                         (void) lnet_post_routed_recv_locked(msg2, 1);
1297                 }
1298         }
1299 #else
1300         LASSERT (!msg->msg_rtrcredit);
1301         LASSERT (!msg->msg_peerrtrcredit);
1302 #endif
1303         if (rxpeer != NULL) {
1304                 msg->msg_rxpeer = NULL;
1305                 lnet_peer_decref_locked(rxpeer);
1306         }
1307 }
1308
1309 int
1310 lnet_send(lnet_nid_t src_nid, lnet_msg_t *msg)
1311 {
1312         lnet_nid_t        dst_nid = msg->msg_target.nid;
1313         lnet_ni_t        *src_ni;
1314         lnet_ni_t        *local_ni;
1315         lnet_remotenet_t *rnet;
1316         lnet_route_t     *route;
1317         lnet_route_t     *best_route;
1318         cfs_list_t       *tmp;
1319         lnet_peer_t      *lp;
1320         lnet_peer_t      *lp2;
1321         int               rc;
1322
1323         LASSERT (msg->msg_txpeer == NULL);
1324         LASSERT (!msg->msg_sending);
1325         LASSERT (!msg->msg_target_is_router);
1326         LASSERT (!msg->msg_receiving);
1327
1328         msg->msg_sending = 1;
1329
1330         /* NB! ni != NULL == interface pre-determined (ACK/REPLY) */
1331
1332         LNET_LOCK();
1333
1334         if (the_lnet.ln_shutdown) {
1335                 LNET_UNLOCK();
1336                 return -ESHUTDOWN;
1337         }
1338
1339         if (src_nid == LNET_NID_ANY) {
1340                 src_ni = NULL;
1341         } else {
1342                 src_ni = lnet_nid2ni_locked(src_nid);
1343                 if (src_ni == NULL) {
1344                         LNET_UNLOCK();
1345                         CERROR("Can't send to %s: src %s is not a local nid\n",
1346                                libcfs_nid2str(dst_nid), libcfs_nid2str(src_nid));
1347                         return -EINVAL;
1348                 }
1349                 LASSERT (!msg->msg_routing);
1350         }
1351
1352         /* Is this for someone on a local network? */
1353         local_ni = lnet_net2ni_locked(LNET_NIDNET(dst_nid));
1354
1355         if (local_ni != NULL) {
1356                 if (src_ni == NULL) {
1357                         src_ni = local_ni;
1358                         src_nid = src_ni->ni_nid;
1359                 } else if (src_ni == local_ni) {
1360                         lnet_ni_decref_locked(local_ni);
1361                 } else {
1362                         lnet_ni_decref_locked(local_ni);
1363                         lnet_ni_decref_locked(src_ni);
1364                         LNET_UNLOCK();
1365                         CERROR("No route to %s via from %s\n",
1366                                libcfs_nid2str(dst_nid), libcfs_nid2str(src_nid));
1367                         return -EINVAL;
1368                 }
1369
1370                 LASSERT (src_nid != LNET_NID_ANY);
1371
1372                 if (!msg->msg_routing)
1373                         msg->msg_hdr.src_nid = cpu_to_le64(src_nid);
1374
1375                 if (src_ni == the_lnet.ln_loni) {
1376                         /* No send credit hassles with LOLND */
1377                         LNET_UNLOCK();
1378                         lnet_ni_send(src_ni, msg);
1379                         lnet_ni_decref(src_ni);
1380                         return 0;
1381                 }
1382
1383                 rc = lnet_nid2peer_locked(&lp, dst_nid);
1384                 lnet_ni_decref_locked(src_ni);  /* lp has ref on src_ni; lose mine */
1385                 if (rc != 0) {
1386                         LNET_UNLOCK();
1387                         CERROR("Error %d finding peer %s\n", rc,
1388                                libcfs_nid2str(dst_nid));
1389                         /* ENOMEM or shutting down */
1390                         return rc;
1391                 }
1392                 LASSERT (lp->lp_ni == src_ni);
1393         } else {
1394 #ifndef __KERNEL__
1395                 LNET_UNLOCK();
1396
1397                 /* NB
1398                  * - once application finishes computation, check here to update
1399                  *   router states before it waits for pending IO in LNetEQPoll
1400                  * - recursion breaker: router checker sends no message
1401                  *   to remote networks */
1402                 if (the_lnet.ln_rc_state == LNET_RC_STATE_RUNNING)
1403                         lnet_router_checker();
1404
1405                 LNET_LOCK();
1406 #endif
1407                 /* sending to a remote network */
1408                 rnet = lnet_find_net_locked(LNET_NIDNET(dst_nid));
1409                 if (rnet == NULL) {
1410                         if (src_ni != NULL)
1411                                 lnet_ni_decref_locked(src_ni);
1412                         LNET_UNLOCK();
1413                         CERROR("No route to %s\n", libcfs_id2str(msg->msg_target));
1414                         return -EHOSTUNREACH;
1415                 }
1416
1417                 /* Find the best gateway I can use */
1418                 lp = NULL;
1419                 best_route = NULL;
1420                 cfs_list_for_each(tmp, &rnet->lrn_routes) {
1421                         route = cfs_list_entry(tmp, lnet_route_t, lr_list);
1422                         lp2 = route->lr_gateway;
1423
1424                         if (lp2->lp_alive &&
1425                             lnet_router_down_ni(lp2, rnet->lrn_net) <= 0 &&
1426                             (src_ni == NULL || lp2->lp_ni == src_ni) &&
1427                             (lp == NULL ||
1428                              lnet_compare_routes(route, best_route) > 0)) {
1429                                 best_route = route;
1430                                 lp = lp2;
1431                         }
1432                 }
1433
1434                 if (lp == NULL) {
1435                         if (src_ni != NULL)
1436                                 lnet_ni_decref_locked(src_ni);
1437                         LNET_UNLOCK();
1438
1439                         CERROR("No route to %s via %s (all routers down)\n",
1440                                libcfs_id2str(msg->msg_target),
1441                                libcfs_nid2str(src_nid));
1442                         return -EHOSTUNREACH;
1443                 }
1444
1445                 /* Place selected route at the end of the route list to ensure
1446                  * fairness; everything else being equal... */
1447                 cfs_list_del(&best_route->lr_list);
1448                 cfs_list_add_tail(&best_route->lr_list, &rnet->lrn_routes);
1449
1450                 if (src_ni == NULL) {
1451                         src_ni = lp->lp_ni;
1452                         src_nid = src_ni->ni_nid;
1453                 } else {
1454                         LASSERT (src_ni == lp->lp_ni);
1455                         lnet_ni_decref_locked(src_ni);
1456                 }
1457
1458                 lnet_peer_addref_locked(lp);
1459
1460                 LASSERT (src_nid != LNET_NID_ANY);
1461
1462                 if (!msg->msg_routing) {
1463                         /* I'm the source and now I know which NI to send on */
1464                         msg->msg_hdr.src_nid = cpu_to_le64(src_nid);
1465                 }
1466
1467                 msg->msg_target_is_router = 1;
1468                 msg->msg_target.nid = lp->lp_nid;
1469                 msg->msg_target.pid = LUSTRE_SRV_LNET_PID;
1470         }
1471
1472         /* 'lp' is our best choice of peer */
1473
1474         LASSERT (!msg->msg_peertxcredit);
1475         LASSERT (!msg->msg_txcredit);
1476         LASSERT (msg->msg_txpeer == NULL);
1477
1478         msg->msg_txpeer = lp;                   /* msg takes my ref on lp */
1479
1480         rc = lnet_post_send_locked(msg, 0);
1481         LNET_UNLOCK();
1482
1483         if (rc == EHOSTUNREACH)
1484                 return -EHOSTUNREACH;
1485
1486         if (rc == 0)
1487                 lnet_ni_send(src_ni, msg);
1488
1489         return 0;
1490 }
1491
1492 static void
1493 lnet_commit_md (lnet_libmd_t *md, lnet_msg_t *msg)
1494 {
1495         /* ALWAYS called holding the LNET_LOCK */
1496         /* Here, we commit the MD to a network OP by marking it busy and
1497          * decrementing its threshold.  Come what may, the network "owns"
1498          * the MD until a call to lnet_finalize() signals completion. */
1499         LASSERT (!msg->msg_routing);
1500
1501         msg->msg_md = md;
1502
1503         md->md_refcount++;
1504         if (md->md_threshold != LNET_MD_THRESH_INF) {
1505                 LASSERT (md->md_threshold > 0);
1506                 md->md_threshold--;
1507         }
1508
1509         the_lnet.ln_counters.msgs_alloc++;
1510         if (the_lnet.ln_counters.msgs_alloc >
1511             the_lnet.ln_counters.msgs_max)
1512                 the_lnet.ln_counters.msgs_max =
1513                         the_lnet.ln_counters.msgs_alloc;
1514
1515         LASSERT (!msg->msg_onactivelist);
1516         msg->msg_onactivelist = 1;
1517         cfs_list_add (&msg->msg_activelist, &the_lnet.ln_active_msgs);
1518 }
1519
1520 static void
1521 lnet_drop_message (lnet_ni_t *ni, void *private, unsigned int nob)
1522 {
1523         LNET_LOCK();
1524         the_lnet.ln_counters.drop_count++;
1525         the_lnet.ln_counters.drop_length += nob;
1526         LNET_UNLOCK();
1527
1528         lnet_ni_recv(ni, private, NULL, 0, 0, 0, nob);
1529 }
1530
1531 static void
1532 lnet_drop_delayed_put(lnet_msg_t *msg, char *reason)
1533 {
1534         lnet_process_id_t id = {0};
1535
1536         id.nid = msg->msg_hdr.src_nid;
1537         id.pid = msg->msg_hdr.src_pid;
1538
1539         LASSERT (msg->msg_md == NULL);
1540         LASSERT (msg->msg_delayed);
1541         LASSERT (msg->msg_rxpeer != NULL);
1542         LASSERT (msg->msg_hdr.type == LNET_MSG_PUT);
1543
1544         CWARN("Dropping delayed PUT from %s portal %d match "LPU64
1545               " offset %d length %d: %s\n",
1546               libcfs_id2str(id),
1547               msg->msg_hdr.msg.put.ptl_index,
1548               msg->msg_hdr.msg.put.match_bits,
1549               msg->msg_hdr.msg.put.offset,
1550               msg->msg_hdr.payload_length,
1551               reason);
1552
1553         /* NB I can't drop msg's ref on msg_rxpeer until after I've
1554          * called lnet_drop_message(), so I just hang onto msg as well
1555          * until that's done */
1556
1557         lnet_drop_message(msg->msg_rxpeer->lp_ni,
1558                           msg->msg_private, msg->msg_len);
1559
1560         LNET_LOCK();
1561
1562         lnet_peer_decref_locked(msg->msg_rxpeer);
1563         msg->msg_rxpeer = NULL;
1564
1565         lnet_msg_free(msg);
1566
1567         LNET_UNLOCK();
1568 }
1569
1570 /**
1571  * Turn on the lazy portal attribute. Use with caution!
1572  *
1573  * This portal attribute only affects incoming PUT requests to the portal,
1574  * and is off by default. By default, if there's no matching MD for an
1575  * incoming PUT request, it is simply dropped. With the lazy attribute on,
1576  * such requests are queued indefinitely until either a matching MD is
1577  * posted to the portal or the lazy attribute is turned off.
1578  *
1579  * It would prevent dropped requests, however it should be regarded as the
1580  * last line of defense - i.e. users must keep a close watch on active
1581  * buffers on a lazy portal and once it becomes too low post more buffers as
1582  * soon as possible. This is because delayed requests usually have detrimental
1583  * effects on underlying network connections. A few delayed requests often
1584  * suffice to bring an underlying connection to a complete halt, due to flow
1585  * control mechanisms.
1586  *
1587  * There's also a DOS attack risk. If users don't post match-all MDs on a
1588  * lazy portal, a malicious peer can easily stop a service by sending some
1589  * PUT requests with match bits that won't match any MD. A routed server is
1590  * especially vulnerable since the connections to its neighbor routers are
1591  * shared among all clients.
1592  *
1593  * \param portal Index of the portal to enable the lazy attribute on.
1594  *
1595  * \retval 0       On success.
1596  * \retval -EINVAL If \a portal is not a valid index.
1597  */
1598 int
1599 LNetSetLazyPortal(int portal)
1600 {
1601         lnet_portal_t *ptl = &the_lnet.ln_portals[portal];
1602
1603         if (portal < 0 || portal >= the_lnet.ln_nportals)
1604                 return -EINVAL;
1605
1606         CDEBUG(D_NET, "Setting portal %d lazy\n", portal);
1607
1608         LNET_LOCK();
1609         lnet_portal_setopt(ptl, LNET_PTL_LAZY);
1610         LNET_UNLOCK();
1611
1612         return 0;
1613 }
1614
1615 /**
1616  * Turn off the lazy portal attribute. Delayed requests on the portal,
1617  * if any, will be all dropped when this function returns.
1618  *
1619  * \param portal Index of the portal to disable the lazy attribute on.
1620  *
1621  * \retval 0       On success.
1622  * \retval -EINVAL If \a portal is not a valid index.
1623  */
1624 int
1625 LNetClearLazyPortal(int portal)
1626 {
1627         cfs_list_t        zombies;
1628         lnet_portal_t    *ptl = &the_lnet.ln_portals[portal];
1629         lnet_msg_t       *msg;
1630
1631         if (portal < 0 || portal >= the_lnet.ln_nportals)
1632                 return -EINVAL;
1633
1634         LNET_LOCK();
1635
1636         if (!lnet_portal_is_lazy(ptl)) {
1637                 LNET_UNLOCK();
1638                 return 0;
1639         }
1640
1641         if (the_lnet.ln_shutdown)
1642                 CWARN ("Active lazy portal %d on exit\n", portal);
1643         else
1644                 CDEBUG (D_NET, "clearing portal %d lazy\n", portal);
1645
1646         /* grab all the blocked messages atomically */
1647         cfs_list_add(&zombies, &ptl->ptl_msgq);
1648         cfs_list_del_init(&ptl->ptl_msgq);
1649
1650         ptl->ptl_msgq_version++;
1651         lnet_portal_unsetopt(ptl, LNET_PTL_LAZY);
1652
1653         LNET_UNLOCK();
1654
1655         while (!cfs_list_empty(&zombies)) {
1656                 msg = cfs_list_entry(zombies.next, lnet_msg_t, msg_list);
1657                 cfs_list_del(&msg->msg_list);
1658
1659                 lnet_drop_delayed_put(msg, "Clearing lazy portal attr");
1660         }
1661
1662         return 0;
1663 }
1664
1665 static void
1666 lnet_recv_put(lnet_libmd_t *md, lnet_msg_t *msg, int delayed,
1667               unsigned int offset, unsigned int mlength)
1668 {
1669         lnet_hdr_t       *hdr = &msg->msg_hdr;
1670
1671         LNET_LOCK();
1672
1673         the_lnet.ln_counters.recv_count++;
1674         the_lnet.ln_counters.recv_length += mlength;
1675
1676         LNET_UNLOCK();
1677
1678         if (mlength != 0)
1679                 lnet_setpayloadbuffer(msg);
1680
1681         msg->msg_ev.type       = LNET_EVENT_PUT;
1682         msg->msg_ev.target.pid = hdr->dest_pid;
1683         msg->msg_ev.target.nid = hdr->dest_nid;
1684         msg->msg_ev.hdr_data   = hdr->msg.put.hdr_data;
1685
1686         /* Must I ACK?  If so I'll grab the ack_wmd out of the header and put
1687          * it back into the ACK during lnet_finalize() */
1688         msg->msg_ack = (!lnet_is_wire_handle_none(&hdr->msg.put.ack_wmd) &&
1689                         (md->md_options & LNET_MD_ACK_DISABLE) == 0);
1690
1691         lnet_ni_recv(msg->msg_rxpeer->lp_ni,
1692                      msg->msg_private,
1693                      msg, delayed, offset, mlength,
1694                      hdr->payload_length);
1695 }
1696
1697 /* called with LNET_LOCK held */
1698 void
1699 lnet_match_blocked_msg(lnet_libmd_t *md)
1700 {
1701         CFS_LIST_HEAD    (drops);
1702         CFS_LIST_HEAD    (matches);
1703         cfs_list_t       *tmp;
1704         cfs_list_t       *entry;
1705         lnet_msg_t       *msg;
1706         lnet_portal_t    *ptl;
1707         lnet_me_t        *me  = md->md_me;
1708
1709         LASSERT (me->me_portal < (unsigned int)the_lnet.ln_nportals);
1710
1711         ptl = &the_lnet.ln_portals[me->me_portal];
1712         if (!lnet_portal_is_lazy(ptl)) {
1713                 LASSERT (cfs_list_empty(&ptl->ptl_msgq));
1714                 return;
1715         }
1716
1717         LASSERT (md->md_refcount == 0); /* a brand new MD */
1718
1719         cfs_list_for_each_safe (entry, tmp, &ptl->ptl_msgq) {
1720                 int               rc;
1721                 int               index;
1722                 unsigned int      mlength;
1723                 unsigned int      offset;
1724                 lnet_hdr_t       *hdr;
1725                 lnet_process_id_t src;
1726
1727                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1728
1729                 LASSERT (msg->msg_delayed);
1730
1731                 hdr   = &msg->msg_hdr;
1732                 index = hdr->msg.put.ptl_index;
1733
1734                 src.nid = hdr->src_nid;
1735                 src.pid = hdr->src_pid;
1736
1737                 rc = lnet_try_match_md(index, LNET_MD_OP_PUT, src,
1738                                        hdr->payload_length,
1739                                        hdr->msg.put.offset,
1740                                        hdr->msg.put.match_bits,
1741                                        md, msg, &mlength, &offset);
1742
1743                 if (rc == LNET_MATCHMD_NONE)
1744                         continue;
1745
1746                 /* Hurrah! This _is_ a match */
1747                 cfs_list_del(&msg->msg_list);
1748                 ptl->ptl_msgq_version++;
1749
1750                 if (rc == LNET_MATCHMD_OK) {
1751                         cfs_list_add_tail(&msg->msg_list, &matches);
1752
1753                         CDEBUG(D_NET, "Resuming delayed PUT from %s portal %d "
1754                                "match "LPU64" offset %d length %d.\n",
1755                                libcfs_id2str(src),
1756                                hdr->msg.put.ptl_index,
1757                                hdr->msg.put.match_bits,
1758                                hdr->msg.put.offset,
1759                                hdr->payload_length);
1760                 } else {
1761                         LASSERT (rc == LNET_MATCHMD_DROP);
1762
1763                         cfs_list_add_tail(&msg->msg_list, &drops);
1764                 }
1765
1766                 if (lnet_md_exhausted(md))
1767                         break;
1768         }
1769
1770         LNET_UNLOCK();
1771
1772         cfs_list_for_each_safe (entry, tmp, &drops) {
1773                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1774
1775                 cfs_list_del(&msg->msg_list);
1776
1777                 lnet_drop_delayed_put(msg, "Bad match");
1778         }
1779
1780         cfs_list_for_each_safe (entry, tmp, &matches) {
1781                 msg = cfs_list_entry(entry, lnet_msg_t, msg_list);
1782
1783                 cfs_list_del(&msg->msg_list);
1784
1785                 /* md won't disappear under me, since each msg
1786                  * holds a ref on it */
1787                 lnet_recv_put(md, msg, 1,
1788                               msg->msg_ev.offset,
1789                               msg->msg_ev.mlength);
1790         }
1791
1792         LNET_LOCK();
1793 }
1794
1795 static int
1796 lnet_parse_put(lnet_ni_t *ni, lnet_msg_t *msg)
1797 {
1798         int               rc;
1799         int               index;
1800         __u64             version;
1801         lnet_hdr_t       *hdr = &msg->msg_hdr;
1802         unsigned int      rlength = hdr->payload_length;
1803         unsigned int      mlength = 0;
1804         unsigned int      offset = 0;
1805         lnet_process_id_t src= {0};
1806         lnet_libmd_t     *md;
1807         lnet_portal_t    *ptl;
1808
1809         src.nid = hdr->src_nid;
1810         src.pid = hdr->src_pid;
1811
1812         /* Convert put fields to host byte order */
1813         hdr->msg.put.match_bits = le64_to_cpu(hdr->msg.put.match_bits);
1814         hdr->msg.put.ptl_index = le32_to_cpu(hdr->msg.put.ptl_index);
1815         hdr->msg.put.offset = le32_to_cpu(hdr->msg.put.offset);
1816
1817         index = hdr->msg.put.ptl_index;
1818
1819         LNET_LOCK();
1820
1821  again:
1822         rc = lnet_match_md(index, LNET_MD_OP_PUT, src,
1823                            rlength, hdr->msg.put.offset,
1824                            hdr->msg.put.match_bits, msg,
1825                            &mlength, &offset, &md);
1826         switch (rc) {
1827         default:
1828                 LBUG();
1829
1830         case LNET_MATCHMD_OK:
1831                 LNET_UNLOCK();
1832                 lnet_recv_put(md, msg, msg->msg_delayed, offset, mlength);
1833                 return 0;
1834
1835         case LNET_MATCHMD_NONE:
1836                 ptl = &the_lnet.ln_portals[index];
1837                 version = ptl->ptl_ml_version;
1838
1839                 rc = 0;
1840                 if (!msg->msg_delayed)
1841                         rc = lnet_eager_recv_locked(msg);
1842
1843                 if (rc == 0 &&
1844                     !the_lnet.ln_shutdown &&
1845                     lnet_portal_is_lazy(ptl)) {
1846                         if (version != ptl->ptl_ml_version)
1847                                 goto again;
1848
1849                         cfs_list_add_tail(&msg->msg_list, &ptl->ptl_msgq);
1850                         ptl->ptl_msgq_version++;
1851                         LNET_UNLOCK();
1852
1853                         CDEBUG(D_NET, "Delaying PUT from %s portal %d match "
1854                                LPU64" offset %d length %d: no match \n",
1855                                libcfs_id2str(src), index,
1856                                hdr->msg.put.match_bits,
1857                                hdr->msg.put.offset, rlength);
1858                         return 0;
1859                 }
1860                 /* fall through */
1861
1862         case LNET_MATCHMD_DROP:
1863                 CDEBUG(D_NETERROR,
1864                        "Dropping PUT from %s portal %d match "LPU64
1865                        " offset %d length %d: %d\n",
1866                        libcfs_id2str(src), index,
1867                        hdr->msg.put.match_bits,
1868                        hdr->msg.put.offset, rlength, rc);
1869                 LNET_UNLOCK();
1870
1871                 return ENOENT;          /* +ve: OK but no match */
1872         }
1873 }
1874
1875 static int
1876 lnet_parse_get(lnet_ni_t *ni, lnet_msg_t *msg, int rdma_get)
1877 {
1878         lnet_hdr_t        *hdr = &msg->msg_hdr;
1879         unsigned int       mlength = 0;
1880         unsigned int       offset = 0;
1881         lnet_process_id_t  src = {0};
1882         lnet_handle_wire_t reply_wmd;
1883         lnet_libmd_t      *md;
1884         int                rc;
1885
1886         src.nid = hdr->src_nid;
1887         src.pid = hdr->src_pid;
1888
1889         /* Convert get fields to host byte order */
1890         hdr->msg.get.match_bits = le64_to_cpu(hdr->msg.get.match_bits);
1891         hdr->msg.get.ptl_index = le32_to_cpu(hdr->msg.get.ptl_index);
1892         hdr->msg.get.sink_length = le32_to_cpu(hdr->msg.get.sink_length);
1893         hdr->msg.get.src_offset = le32_to_cpu(hdr->msg.get.src_offset);
1894
1895         LNET_LOCK();
1896
1897         rc = lnet_match_md(hdr->msg.get.ptl_index, LNET_MD_OP_GET, src,
1898                            hdr->msg.get.sink_length, hdr->msg.get.src_offset,
1899                            hdr->msg.get.match_bits, msg,
1900                            &mlength, &offset, &md);
1901         if (rc == LNET_MATCHMD_DROP) {
1902                 CDEBUG(D_NETERROR,
1903                        "Dropping GET from %s portal %d match "LPU64
1904                        " offset %d length %d\n",
1905                        libcfs_id2str(src),
1906                        hdr->msg.get.ptl_index,
1907                        hdr->msg.get.match_bits,
1908                        hdr->msg.get.src_offset,
1909                        hdr->msg.get.sink_length);
1910                 LNET_UNLOCK();
1911                 return ENOENT;                  /* +ve: OK but no match */
1912         }
1913
1914         LASSERT (rc == LNET_MATCHMD_OK);
1915
1916         the_lnet.ln_counters.send_count++;
1917         the_lnet.ln_counters.send_length += mlength;
1918
1919         LNET_UNLOCK();
1920
1921         msg->msg_ev.type = LNET_EVENT_GET;
1922         msg->msg_ev.target.pid = hdr->dest_pid;
1923         msg->msg_ev.target.nid = hdr->dest_nid;
1924         msg->msg_ev.hdr_data = 0;
1925
1926         reply_wmd = hdr->msg.get.return_wmd;
1927
1928         lnet_prep_send(msg, LNET_MSG_REPLY, src, offset, mlength);
1929
1930         msg->msg_hdr.msg.reply.dst_wmd = reply_wmd;
1931
1932         if (rdma_get) {
1933                 /* The LND completes the REPLY from her recv procedure */
1934                 lnet_ni_recv(ni, msg->msg_private, msg, 0,
1935                              msg->msg_offset, msg->msg_len, msg->msg_len);
1936                 return 0;
1937         }
1938
1939         lnet_ni_recv(ni, msg->msg_private, NULL, 0, 0, 0, 0);
1940         msg->msg_receiving = 0;
1941
1942         rc = lnet_send(ni->ni_nid, msg);
1943         if (rc < 0) {
1944                 /* didn't get as far as lnet_ni_send() */
1945                 CERROR("%s: Unable to send REPLY for GET from %s: %d\n",
1946                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src), rc);
1947
1948                 lnet_finalize(ni, msg, rc);
1949         }
1950
1951         return 0;
1952 }
1953
1954 static int
1955 lnet_parse_reply(lnet_ni_t *ni, lnet_msg_t *msg)
1956 {
1957         void             *private = msg->msg_private;
1958         lnet_hdr_t       *hdr = &msg->msg_hdr;
1959         lnet_process_id_t src = {0};
1960         lnet_libmd_t     *md;
1961         int               rlength;
1962         int               mlength;
1963
1964         LNET_LOCK();
1965
1966         src.nid = hdr->src_nid;
1967         src.pid = hdr->src_pid;
1968
1969         /* NB handles only looked up by creator (no flips) */
1970         md = lnet_wire_handle2md(&hdr->msg.reply.dst_wmd);
1971         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
1972                 CDEBUG(D_NETERROR, "%s: Dropping REPLY from %s for %s "
1973                        "MD "LPX64"."LPX64"\n",
1974                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
1975                        (md == NULL) ? "invalid" : "inactive",
1976                        hdr->msg.reply.dst_wmd.wh_interface_cookie,
1977                        hdr->msg.reply.dst_wmd.wh_object_cookie);
1978                 if (md != NULL && md->md_me != NULL)
1979                         CERROR("REPLY MD also attached to portal %d\n",
1980                                md->md_me->me_portal);
1981
1982                 LNET_UNLOCK();
1983                 return ENOENT;                  /* +ve: OK but no match */
1984         }
1985
1986         LASSERT (md->md_offset == 0);
1987
1988         rlength = hdr->payload_length;
1989         mlength = MIN(rlength, (int)md->md_length);
1990
1991         if (mlength < rlength &&
1992             (md->md_options & LNET_MD_TRUNCATE) == 0) {
1993                 CDEBUG(D_NETERROR, "%s: Dropping REPLY from %s length %d "
1994                        "for MD "LPX64" would overflow (%d)\n",
1995                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
1996                         rlength, hdr->msg.reply.dst_wmd.wh_object_cookie,
1997                         mlength);
1998                 LNET_UNLOCK();
1999                 return ENOENT;          /* +ve: OK but no match */
2000         }
2001
2002         CDEBUG(D_NET, "%s: Reply from %s of length %d/%d into md "LPX64"\n",
2003                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2004                mlength, rlength, hdr->msg.reply.dst_wmd.wh_object_cookie);
2005
2006         lnet_commit_md(md, msg);
2007
2008         if (mlength != 0)
2009                 lnet_setpayloadbuffer(msg);
2010
2011         msg->msg_ev.type = LNET_EVENT_REPLY;
2012         msg->msg_ev.target.pid = hdr->dest_pid;
2013         msg->msg_ev.target.nid = hdr->dest_nid;
2014         msg->msg_ev.initiator = src;
2015         msg->msg_ev.rlength = rlength;
2016         msg->msg_ev.mlength = mlength;
2017         msg->msg_ev.offset = 0;
2018
2019         lnet_md_deconstruct(md, &msg->msg_ev.md);
2020         lnet_md2handle(&msg->msg_ev.md_handle, md);
2021
2022         the_lnet.ln_counters.recv_count++;
2023         the_lnet.ln_counters.recv_length += mlength;
2024
2025         LNET_UNLOCK();
2026
2027         lnet_ni_recv(ni, private, msg, 0, 0, mlength, rlength);
2028         return 0;
2029 }
2030
2031 static int
2032 lnet_parse_ack(lnet_ni_t *ni, lnet_msg_t *msg)
2033 {
2034         lnet_hdr_t       *hdr = &msg->msg_hdr;
2035         lnet_process_id_t src = {0};
2036         lnet_libmd_t     *md;
2037
2038         src.nid = hdr->src_nid;
2039         src.pid = hdr->src_pid;
2040
2041         /* Convert ack fields to host byte order */
2042         hdr->msg.ack.match_bits = le64_to_cpu(hdr->msg.ack.match_bits);
2043         hdr->msg.ack.mlength = le32_to_cpu(hdr->msg.ack.mlength);
2044
2045         LNET_LOCK();
2046
2047         /* NB handles only looked up by creator (no flips) */
2048         md = lnet_wire_handle2md(&hdr->msg.ack.dst_wmd);
2049         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2050                 /* Don't moan; this is expected */
2051                 CDEBUG(D_NET,
2052                        "%s: Dropping ACK from %s to %s MD "LPX64"."LPX64"\n",
2053                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2054                        (md == NULL) ? "invalid" : "inactive",
2055                        hdr->msg.ack.dst_wmd.wh_interface_cookie,
2056                        hdr->msg.ack.dst_wmd.wh_object_cookie);
2057                 if (md != NULL && md->md_me != NULL)
2058                         CERROR("Source MD also attached to portal %d\n",
2059                                md->md_me->me_portal);
2060
2061                 LNET_UNLOCK();
2062                 return ENOENT;                  /* +ve! */
2063         }
2064
2065         CDEBUG(D_NET, "%s: ACK from %s into md "LPX64"\n",
2066                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2067                hdr->msg.ack.dst_wmd.wh_object_cookie);
2068
2069         lnet_commit_md(md, msg);
2070
2071         msg->msg_ev.type = LNET_EVENT_ACK;
2072         msg->msg_ev.target.pid = hdr->dest_pid;
2073         msg->msg_ev.target.nid = hdr->dest_nid;
2074         msg->msg_ev.initiator = src;
2075         msg->msg_ev.mlength = hdr->msg.ack.mlength;
2076         msg->msg_ev.match_bits = hdr->msg.ack.match_bits;
2077
2078         lnet_md_deconstruct(md, &msg->msg_ev.md);
2079         lnet_md2handle(&msg->msg_ev.md_handle, md);
2080
2081         the_lnet.ln_counters.recv_count++;
2082
2083         LNET_UNLOCK();
2084
2085         lnet_ni_recv(ni, msg->msg_private, msg, 0, 0, 0, msg->msg_len);
2086         return 0;
2087 }
2088
2089 char *
2090 lnet_msgtyp2str (int type)
2091 {
2092         switch (type) {
2093         case LNET_MSG_ACK:
2094                 return ("ACK");
2095         case LNET_MSG_PUT:
2096                 return ("PUT");
2097         case LNET_MSG_GET:
2098                 return ("GET");
2099         case LNET_MSG_REPLY:
2100                 return ("REPLY");
2101         case LNET_MSG_HELLO:
2102                 return ("HELLO");
2103         default:
2104                 return ("<UNKNOWN>");
2105         }
2106 }
2107
2108 void
2109 lnet_print_hdr(lnet_hdr_t * hdr)
2110 {
2111         lnet_process_id_t src = {0};
2112         lnet_process_id_t dst = {0};
2113         char *type_str = lnet_msgtyp2str (hdr->type);
2114
2115         src.nid = hdr->src_nid;
2116         src.pid = hdr->src_pid;
2117
2118         dst.nid = hdr->dest_nid;
2119         dst.pid = hdr->dest_pid;
2120
2121         CWARN("P3 Header at %p of type %s\n", hdr, type_str);
2122         CWARN("    From %s\n", libcfs_id2str(src));
2123         CWARN("    To   %s\n", libcfs_id2str(dst));
2124
2125         switch (hdr->type) {
2126         default:
2127                 break;
2128
2129         case LNET_MSG_PUT:
2130                 CWARN("    Ptl index %d, ack md "LPX64"."LPX64", "
2131                       "match bits "LPU64"\n",
2132                       hdr->msg.put.ptl_index,
2133                       hdr->msg.put.ack_wmd.wh_interface_cookie,
2134                       hdr->msg.put.ack_wmd.wh_object_cookie,
2135                       hdr->msg.put.match_bits);
2136                 CWARN("    Length %d, offset %d, hdr data "LPX64"\n",
2137                       hdr->payload_length, hdr->msg.put.offset,
2138                       hdr->msg.put.hdr_data);
2139                 break;
2140
2141         case LNET_MSG_GET:
2142                 CWARN("    Ptl index %d, return md "LPX64"."LPX64", "
2143                       "match bits "LPU64"\n", hdr->msg.get.ptl_index,
2144                       hdr->msg.get.return_wmd.wh_interface_cookie,
2145                       hdr->msg.get.return_wmd.wh_object_cookie,
2146                       hdr->msg.get.match_bits);
2147                 CWARN("    Length %d, src offset %d\n",
2148                       hdr->msg.get.sink_length,
2149                       hdr->msg.get.src_offset);
2150                 break;
2151
2152         case LNET_MSG_ACK:
2153                 CWARN("    dst md "LPX64"."LPX64", "
2154                       "manipulated length %d\n",
2155                       hdr->msg.ack.dst_wmd.wh_interface_cookie,
2156                       hdr->msg.ack.dst_wmd.wh_object_cookie,
2157                       hdr->msg.ack.mlength);
2158                 break;
2159
2160         case LNET_MSG_REPLY:
2161                 CWARN("    dst md "LPX64"."LPX64", "
2162                       "length %d\n",
2163                       hdr->msg.reply.dst_wmd.wh_interface_cookie,
2164                       hdr->msg.reply.dst_wmd.wh_object_cookie,
2165                       hdr->payload_length);
2166         }
2167
2168 }
2169
2170 int
2171 lnet_parse(lnet_ni_t *ni, lnet_hdr_t *hdr, lnet_nid_t from_nid,
2172            void *private, int rdma_req)
2173 {
2174         int            rc = 0;
2175         int            for_me;
2176         lnet_msg_t    *msg;
2177         lnet_pid_t     dest_pid;
2178         lnet_nid_t     dest_nid;
2179         lnet_nid_t     src_nid;
2180         __u32          payload_length;
2181         __u32          type;
2182
2183         LASSERT (!cfs_in_interrupt ());
2184
2185         type = le32_to_cpu(hdr->type);
2186         src_nid = le64_to_cpu(hdr->src_nid);
2187         dest_nid = le64_to_cpu(hdr->dest_nid);
2188         dest_pid = le32_to_cpu(hdr->dest_pid);
2189         payload_length = le32_to_cpu(hdr->payload_length);
2190
2191         for_me = (ni->ni_nid == dest_nid);
2192
2193         switch (type) {
2194         case LNET_MSG_ACK:
2195         case LNET_MSG_GET:
2196                 if (payload_length > 0) {
2197                         CERROR("%s, src %s: bad %s payload %d (0 expected)\n",
2198                                libcfs_nid2str(from_nid),
2199                                libcfs_nid2str(src_nid),
2200                                lnet_msgtyp2str(type), payload_length);
2201                         return -EPROTO;
2202                 }
2203                 break;
2204
2205         case LNET_MSG_PUT:
2206         case LNET_MSG_REPLY:
2207                 if (payload_length > (__u32)(for_me ? LNET_MAX_PAYLOAD : LNET_MTU)) {
2208                         CERROR("%s, src %s: bad %s payload %d "
2209                                "(%d max expected)\n",
2210                                libcfs_nid2str(from_nid),
2211                                libcfs_nid2str(src_nid),
2212                                lnet_msgtyp2str(type),
2213                                payload_length,
2214                                for_me ? LNET_MAX_PAYLOAD : LNET_MTU);
2215                         return -EPROTO;
2216                 }
2217                 break;
2218
2219         default:
2220                 CERROR("%s, src %s: Bad message type 0x%x\n",
2221                        libcfs_nid2str(from_nid),
2222                        libcfs_nid2str(src_nid), type);
2223                 return -EPROTO;
2224         }
2225
2226         if (the_lnet.ln_routing) {
2227                 cfs_time_t now = cfs_time_current();
2228
2229                 LNET_LOCK();
2230
2231                 ni->ni_last_alive = now;
2232                 if (ni->ni_status != NULL &&
2233                     ni->ni_status->ns_status == LNET_NI_STATUS_DOWN)
2234                         ni->ni_status->ns_status = LNET_NI_STATUS_UP;
2235
2236                 LNET_UNLOCK();
2237         }
2238
2239         /* Regard a bad destination NID as a protocol error.  Senders should
2240          * know what they're doing; if they don't they're misconfigured, buggy
2241          * or malicious so we chop them off at the knees :) */
2242
2243         if (!for_me) {
2244                 if (LNET_NIDNET(dest_nid) == LNET_NIDNET(ni->ni_nid)) {
2245                         /* should have gone direct */
2246                         CERROR ("%s, src %s: Bad dest nid %s "
2247                                 "(should have been sent direct)\n",
2248                                 libcfs_nid2str(from_nid),
2249                                 libcfs_nid2str(src_nid),
2250                                 libcfs_nid2str(dest_nid));
2251                         return -EPROTO;
2252                 }
2253
2254                 if (lnet_islocalnid(dest_nid)) {
2255                         /* dest is another local NI; sender should have used
2256                          * this node's NID on its own network */
2257                         CERROR ("%s, src %s: Bad dest nid %s "
2258                                 "(it's my nid but on a different network)\n",
2259                                 libcfs_nid2str(from_nid),
2260                                 libcfs_nid2str(src_nid),
2261                                 libcfs_nid2str(dest_nid));
2262                         return -EPROTO;
2263                 }
2264
2265                 if (rdma_req && type == LNET_MSG_GET) {
2266                         CERROR ("%s, src %s: Bad optimized GET for %s "
2267                                 "(final destination must be me)\n",
2268                                 libcfs_nid2str(from_nid),
2269                                 libcfs_nid2str(src_nid),
2270                                 libcfs_nid2str(dest_nid));
2271                         return -EPROTO;
2272                 }
2273
2274                 if (!the_lnet.ln_routing) {
2275                         CERROR ("%s, src %s: Dropping message for %s "
2276                                 "(routing not enabled)\n",
2277                                 libcfs_nid2str(from_nid),
2278                                 libcfs_nid2str(src_nid),
2279                                 libcfs_nid2str(dest_nid));
2280                         goto drop;
2281                 }
2282         }
2283
2284         /* Message looks OK; we're not going to return an error, so we MUST
2285          * call back lnd_recv() come what may... */
2286
2287         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2288             fail_peer (src_nid, 0))             /* shall we now? */
2289         {
2290                 CERROR("%s, src %s: Dropping %s to simulate failure\n",
2291                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2292                        lnet_msgtyp2str(type));
2293                 goto drop;
2294         }
2295
2296         msg = lnet_msg_alloc();
2297         if (msg == NULL) {
2298                 CERROR("%s, src %s: Dropping %s (out of memory)\n",
2299                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2300                        lnet_msgtyp2str(type));
2301                 goto drop;
2302         }
2303
2304         /* msg zeroed in lnet_msg_alloc; i.e. flags all clear, pointers NULL etc */
2305
2306         msg->msg_type = type;
2307         msg->msg_private = private;
2308         msg->msg_receiving = 1;
2309         msg->msg_len = msg->msg_wanted = payload_length;
2310         msg->msg_offset = 0;
2311         msg->msg_hdr = *hdr;
2312
2313         LNET_LOCK();
2314         rc = lnet_nid2peer_locked(&msg->msg_rxpeer, from_nid);
2315         if (rc != 0) {
2316                 LNET_UNLOCK();
2317                 CERROR("%s, src %s: Dropping %s "
2318                        "(error %d looking up sender)\n",
2319                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2320                        lnet_msgtyp2str(type), rc);
2321                 goto free_drop;
2322         }
2323         LNET_UNLOCK();
2324
2325 #ifndef __KERNEL__
2326         LASSERT (for_me);
2327 #else
2328         if (!for_me) {
2329                 msg->msg_target.pid = dest_pid;
2330                 msg->msg_target.nid = dest_nid;
2331                 msg->msg_routing = 1;
2332                 msg->msg_offset = 0;
2333
2334                 LNET_LOCK();
2335                 if (msg->msg_rxpeer->lp_rtrcredits <= 0 ||
2336                     lnet_msg2bufpool(msg)->rbp_credits <= 0) {
2337                         rc = lnet_eager_recv_locked(msg);
2338                         if (rc != 0) {
2339                                 LNET_UNLOCK();
2340                                 goto free_drop;
2341                         }
2342                 }
2343                 lnet_commit_routedmsg(msg);
2344                 rc = lnet_post_routed_recv_locked(msg, 0);
2345                 LNET_UNLOCK();
2346
2347                 if (rc == 0)
2348                         lnet_ni_recv(ni, msg->msg_private, msg, 0,
2349                                      0, payload_length, payload_length);
2350                 return 0;
2351         }
2352 #endif
2353         /* convert common msg->hdr fields to host byteorder */
2354         msg->msg_hdr.type = type;
2355         msg->msg_hdr.src_nid = src_nid;
2356         msg->msg_hdr.src_pid = le32_to_cpu(msg->msg_hdr.src_pid);
2357         msg->msg_hdr.dest_nid = dest_nid;
2358         msg->msg_hdr.dest_pid = dest_pid;
2359         msg->msg_hdr.payload_length = payload_length;
2360
2361         msg->msg_ev.sender = from_nid;
2362
2363         switch (type) {
2364         case LNET_MSG_ACK:
2365                 rc = lnet_parse_ack(ni, msg);
2366                 break;
2367         case LNET_MSG_PUT:
2368                 rc = lnet_parse_put(ni, msg);
2369                 break;
2370         case LNET_MSG_GET:
2371                 rc = lnet_parse_get(ni, msg, rdma_req);
2372                 break;
2373         case LNET_MSG_REPLY:
2374                 rc = lnet_parse_reply(ni, msg);
2375                 break;
2376         default:
2377                 LASSERT(0);
2378                 goto free_drop;  /* prevent an unused label if !kernel */
2379         }
2380
2381         if (rc == 0)
2382                 return 0;
2383
2384         LASSERT (rc == ENOENT);
2385
2386  free_drop:
2387         LASSERT (msg->msg_md == NULL);
2388         LNET_LOCK();
2389         if (msg->msg_rxpeer != NULL) {
2390                 lnet_peer_decref_locked(msg->msg_rxpeer);
2391                 msg->msg_rxpeer = NULL;
2392         }
2393         lnet_msg_free(msg);                     /* expects LNET_LOCK held */
2394         LNET_UNLOCK();
2395
2396  drop:
2397         lnet_drop_message(ni, private, payload_length);
2398         return 0;
2399 }
2400
2401 /**
2402  * Initiate an asynchronous PUT operation.
2403  *
2404  * There are several events associated with a PUT: completion of the send on
2405  * the initiator node (LNET_EVENT_SEND), and when the send completes
2406  * successfully, the receipt of an acknowledgment (LNET_EVENT_ACK) indicating
2407  * that the operation was accepted by the target. The event LNET_EVENT_PUT is
2408  * used at the target node to indicate the completion of incoming data
2409  * delivery.
2410  *
2411  * The local events will be logged in the EQ associated with the MD pointed to
2412  * by \a mdh handle. Using a MD without an associated EQ results in these
2413  * events being discarded. In this case, the caller must have another
2414  * mechanism (e.g., a higher level protocol) for determining when it is safe
2415  * to modify the memory region associated with the MD.
2416  *
2417  * Note that LNet does not guarantee the order of LNET_EVENT_SEND and
2418  * LNET_EVENT_ACK, though intuitively ACK should happen after SEND.
2419  *
2420  * \param self Indicates the NID of a local interface through which to send
2421  * the PUT request. Use LNET_NID_ANY to let LNet choose one by itself.
2422  * \param mdh A handle for the MD that describes the memory to be sent. The MD
2423  * must be "free floating" (See LNetMDBind()).
2424  * \param ack Controls whether an acknowledgment is requested.
2425  * Acknowledgments are only sent when they are requested by the initiating
2426  * process and the target MD enables them.
2427  * \param target A process identifier for the target process.
2428  * \param portal The index in the \a target's portal table.
2429  * \param match_bits The match bits to use for MD selection at the target
2430  * process.
2431  * \param offset The offset into the target MD (only used when the target
2432  * MD has the LNET_MD_MANAGE_REMOTE option set).
2433  * \param hdr_data 64 bits of user data that can be included in the message
2434  * header. This data is written to an event queue entry at the target if an
2435  * EQ is present on the matching MD.
2436  *
2437  * \retval  0      Success, and only in this case events will be generated
2438  * and logged to EQ (if it exists).
2439  * \retval -EIO    Simulated failure.
2440  * \retval -ENOMEM Memory allocation failure.
2441  * \retval -ENOENT Invalid MD object.
2442  *
2443  * \see lnet_event_t::hdr_data and lnet_event_kind_t.
2444  */
2445 int
2446 LNetPut(lnet_nid_t self, lnet_handle_md_t mdh, lnet_ack_req_t ack,
2447         lnet_process_id_t target, unsigned int portal,
2448         __u64 match_bits, unsigned int offset,
2449         __u64 hdr_data)
2450 {
2451         lnet_msg_t       *msg;
2452         lnet_libmd_t     *md;
2453         int               rc;
2454
2455         LASSERT (the_lnet.ln_init);
2456         LASSERT (the_lnet.ln_refcount > 0);
2457
2458         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2459             fail_peer (target.nid, 1))          /* shall we now? */
2460         {
2461                 CERROR("Dropping PUT to %s: simulated failure\n",
2462                        libcfs_id2str(target));
2463                 return -EIO;
2464         }
2465
2466         msg = lnet_msg_alloc();
2467         if (msg == NULL) {
2468                 CERROR("Dropping PUT to %s: ENOMEM on lnet_msg_t\n",
2469                        libcfs_id2str(target));
2470                 return -ENOMEM;
2471         }
2472         msg->msg_vmflush = !!cfs_memory_pressure_get();
2473
2474         LNET_LOCK();
2475
2476         md = lnet_handle2md(&mdh);
2477         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2478                 lnet_msg_free(msg);
2479
2480                 CERROR("Dropping PUT ("LPU64":%d:%s): MD (%d) invalid\n",
2481                        match_bits, portal, libcfs_id2str(target),
2482                        md == NULL ? -1 : md->md_threshold);
2483                 if (md != NULL && md->md_me != NULL)
2484                         CERROR("Source MD also attached to portal %d\n",
2485                                md->md_me->me_portal);
2486
2487                 LNET_UNLOCK();
2488                 return -ENOENT;
2489         }
2490
2491         CDEBUG(D_NET, "LNetPut -> %s\n", libcfs_id2str(target));
2492
2493         lnet_commit_md(md, msg);
2494
2495         lnet_prep_send(msg, LNET_MSG_PUT, target, 0, md->md_length);
2496
2497         msg->msg_hdr.msg.put.match_bits = cpu_to_le64(match_bits);
2498         msg->msg_hdr.msg.put.ptl_index = cpu_to_le32(portal);
2499         msg->msg_hdr.msg.put.offset = cpu_to_le32(offset);
2500         msg->msg_hdr.msg.put.hdr_data = hdr_data;
2501
2502         /* NB handles only looked up by creator (no flips) */
2503         if (ack == LNET_ACK_REQ) {
2504                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2505                         the_lnet.ln_interface_cookie;
2506                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2507                         md->md_lh.lh_cookie;
2508         } else {
2509                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2510                         LNET_WIRE_HANDLE_COOKIE_NONE;
2511                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2512                         LNET_WIRE_HANDLE_COOKIE_NONE;
2513         }
2514
2515         msg->msg_ev.type = LNET_EVENT_SEND;
2516         msg->msg_ev.initiator.nid = LNET_NID_ANY;
2517         msg->msg_ev.initiator.pid = the_lnet.ln_pid;
2518         msg->msg_ev.target = target;
2519         msg->msg_ev.sender = LNET_NID_ANY;
2520         msg->msg_ev.pt_index = portal;
2521         msg->msg_ev.match_bits = match_bits;
2522         msg->msg_ev.rlength = md->md_length;
2523         msg->msg_ev.mlength = md->md_length;
2524         msg->msg_ev.offset = offset;
2525         msg->msg_ev.hdr_data = hdr_data;
2526
2527         lnet_md_deconstruct(md, &msg->msg_ev.md);
2528         lnet_md2handle(&msg->msg_ev.md_handle, md);
2529
2530         the_lnet.ln_counters.send_count++;
2531         the_lnet.ln_counters.send_length += md->md_length;
2532
2533         LNET_UNLOCK();
2534
2535         rc = lnet_send(self, msg);
2536         if (rc != 0) {
2537                 CERROR("Error sending PUT to %s: %d\n",
2538                        libcfs_id2str(target), rc);
2539                 lnet_finalize (NULL, msg, rc);
2540         }
2541
2542         /* completion will be signalled by an event */
2543         return 0;
2544 }
2545
2546 lnet_msg_t *
2547 lnet_create_reply_msg (lnet_ni_t *ni, lnet_msg_t *getmsg)
2548 {
2549         /* The LND can DMA direct to the GET md (i.e. no REPLY msg).  This
2550          * returns a msg for the LND to pass to lnet_finalize() when the sink
2551          * data has been received.
2552          *
2553          * CAVEAT EMPTOR: 'getmsg' is the original GET, which is freed when
2554          * lnet_finalize() is called on it, so the LND must call this first */
2555
2556         lnet_msg_t        *msg = lnet_msg_alloc();
2557         lnet_libmd_t      *getmd = getmsg->msg_md;
2558         lnet_process_id_t  peer_id = getmsg->msg_target;
2559
2560         LASSERT (!getmsg->msg_target_is_router);
2561         LASSERT (!getmsg->msg_routing);
2562
2563         LNET_LOCK();
2564
2565         LASSERT (getmd->md_refcount > 0);
2566
2567         if (msg == NULL) {
2568                 CERROR ("%s: Dropping REPLY from %s: can't allocate msg\n",
2569                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id));
2570                 goto drop;
2571         }
2572
2573         if (getmd->md_threshold == 0) {
2574                 CERROR ("%s: Dropping REPLY from %s for inactive MD %p\n",
2575                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id),
2576                         getmd);
2577                 goto drop_msg;
2578         }
2579
2580         LASSERT (getmd->md_offset == 0);
2581
2582         CDEBUG(D_NET, "%s: Reply from %s md %p\n",
2583                libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id), getmd);
2584
2585         lnet_commit_md (getmd, msg);
2586
2587         msg->msg_type = LNET_MSG_GET; /* flag this msg as an "optimized" GET */
2588
2589         msg->msg_ev.type = LNET_EVENT_REPLY;
2590         msg->msg_ev.initiator = peer_id;
2591         msg->msg_ev.sender = peer_id.nid;  /* optimized GETs can't be routed */
2592         msg->msg_ev.rlength = msg->msg_ev.mlength = getmd->md_length;
2593         msg->msg_ev.offset = 0;
2594
2595         lnet_md_deconstruct(getmd, &msg->msg_ev.md);
2596         lnet_md2handle(&msg->msg_ev.md_handle, getmd);
2597
2598         the_lnet.ln_counters.recv_count++;
2599         the_lnet.ln_counters.recv_length += getmd->md_length;
2600
2601         LNET_UNLOCK();
2602
2603         return msg;
2604
2605  drop_msg:
2606         lnet_msg_free(msg);
2607  drop:
2608         the_lnet.ln_counters.drop_count++;
2609         the_lnet.ln_counters.drop_length += getmd->md_length;
2610
2611         LNET_UNLOCK ();
2612
2613         return NULL;
2614 }
2615
2616 void
2617 lnet_set_reply_msg_len(lnet_ni_t *ni, lnet_msg_t *reply, unsigned int len)
2618 {
2619         /* Set the REPLY length, now the RDMA that elides the REPLY message has
2620          * completed and I know it. */
2621         LASSERT (reply != NULL);
2622         LASSERT (reply->msg_type == LNET_MSG_GET);
2623         LASSERT (reply->msg_ev.type == LNET_EVENT_REPLY);
2624
2625         /* NB I trusted my peer to RDMA.  If she tells me she's written beyond
2626          * the end of my buffer, I might as well be dead. */
2627         LASSERT (len <= reply->msg_ev.mlength);
2628
2629         reply->msg_ev.mlength = len;
2630 }
2631
2632 /**
2633  * Initiate an asynchronous GET operation.
2634  *
2635  * On the initiator node, an LNET_EVENT_SEND is logged when the GET request
2636  * is sent, and an LNET_EVENT_REPLY is logged when the data returned from
2637  * the target node in the REPLY has been written to local MD.
2638  *
2639  * On the target node, an LNET_EVENT_GET is logged when the GET request
2640  * arrives and is accepted into a MD.
2641  *
2642  * \param self,target,portal,match_bits,offset See the discussion in LNetPut().
2643  * \param mdh A handle for the MD that describes the memory into which the
2644  * requested data will be received. The MD must be "free floating" (See LNetMDBind()).
2645  *
2646  * \retval  0      Success, and only in this case events will be generated
2647  * and logged to EQ (if it exists) of the MD.
2648  * \retval -EIO    Simulated failure.
2649  * \retval -ENOMEM Memory allocation failure.
2650  * \retval -ENOENT Invalid MD object.
2651  */
2652 int
2653 LNetGet(lnet_nid_t self, lnet_handle_md_t mdh,
2654         lnet_process_id_t target, unsigned int portal,
2655         __u64 match_bits, unsigned int offset)
2656 {
2657         lnet_msg_t       *msg;
2658         lnet_libmd_t     *md;
2659         int               rc;
2660
2661         LASSERT (the_lnet.ln_init);
2662         LASSERT (the_lnet.ln_refcount > 0);
2663
2664         if (!cfs_list_empty (&the_lnet.ln_test_peers) && /* normally we don't */
2665             fail_peer (target.nid, 1))          /* shall we now? */
2666         {
2667                 CERROR("Dropping GET to %s: simulated failure\n",
2668                        libcfs_id2str(target));
2669                 return -EIO;
2670         }
2671
2672         msg = lnet_msg_alloc();
2673         if (msg == NULL) {
2674                 CERROR("Dropping GET to %s: ENOMEM on lnet_msg_t\n",
2675                        libcfs_id2str(target));
2676                 return -ENOMEM;
2677         }
2678
2679         LNET_LOCK();
2680
2681         md = lnet_handle2md(&mdh);
2682         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2683                 lnet_msg_free(msg);
2684
2685                 CERROR("Dropping GET ("LPU64":%d:%s): MD (%d) invalid\n",
2686                        match_bits, portal, libcfs_id2str(target),
2687                        md == NULL ? -1 : md->md_threshold);
2688                 if (md != NULL && md->md_me != NULL)
2689                         CERROR("REPLY MD also attached to portal %d\n",
2690                                md->md_me->me_portal);
2691
2692                 LNET_UNLOCK();
2693                 return -ENOENT;
2694         }
2695
2696         CDEBUG(D_NET, "LNetGet -> %s\n", libcfs_id2str(target));
2697
2698         lnet_commit_md(md, msg);
2699
2700         lnet_prep_send(msg, LNET_MSG_GET, target, 0, 0);
2701
2702         msg->msg_hdr.msg.get.match_bits = cpu_to_le64(match_bits);
2703         msg->msg_hdr.msg.get.ptl_index = cpu_to_le32(portal);
2704         msg->msg_hdr.msg.get.src_offset = cpu_to_le32(offset);
2705         msg->msg_hdr.msg.get.sink_length = cpu_to_le32(md->md_length);
2706
2707         /* NB handles only looked up by creator (no flips) */
2708         msg->msg_hdr.msg.get.return_wmd.wh_interface_cookie =
2709                 the_lnet.ln_interface_cookie;
2710         msg->msg_hdr.msg.get.return_wmd.wh_object_cookie =
2711                 md->md_lh.lh_cookie;
2712
2713         msg->msg_ev.type = LNET_EVENT_SEND;
2714         msg->msg_ev.initiator.nid = LNET_NID_ANY;
2715         msg->msg_ev.initiator.pid = the_lnet.ln_pid;
2716         msg->msg_ev.target = target;
2717         msg->msg_ev.sender = LNET_NID_ANY;
2718         msg->msg_ev.pt_index = portal;
2719         msg->msg_ev.match_bits = match_bits;
2720         msg->msg_ev.rlength = md->md_length;
2721         msg->msg_ev.mlength = md->md_length;
2722         msg->msg_ev.offset = offset;
2723         msg->msg_ev.hdr_data = 0;
2724
2725         lnet_md_deconstruct(md, &msg->msg_ev.md);
2726         lnet_md2handle(&msg->msg_ev.md_handle, md);
2727
2728         the_lnet.ln_counters.send_count++;
2729
2730         LNET_UNLOCK();
2731
2732         rc = lnet_send(self, msg);
2733         if (rc < 0) {
2734                 CERROR("error sending GET to %s: %d\n",
2735                        libcfs_id2str(target), rc);
2736                 lnet_finalize (NULL, msg, rc);
2737         }
2738
2739         /* completion will be signalled by an event */
2740         return 0;
2741 }
2742
2743 /**
2744  * Calculate distance to node at \a dstnid.
2745  *
2746  * \param dstnid Target NID.
2747  * \param srcnidp If not NULL, NID of the local interface to reach \a dstnid
2748  * is saved here.
2749  * \param orderp If not NULL, order of the route to reach \a dstnid is saved
2750  * here.
2751  *
2752  * \retval 0 If \a dstnid belongs to a local interface, and reserved option
2753  * local_nid_dist_zero is set, which is the default.
2754  * \retval positives Distance to target NID, i.e. number of hops plus one.
2755  * \retval -EHOSTUNREACH If \a dstnid is not reachable.
2756  */
2757 int
2758 LNetDist (lnet_nid_t dstnid, lnet_nid_t *srcnidp, __u32 *orderp)
2759 {
2760         cfs_list_t       *e;
2761         lnet_ni_t        *ni;
2762         lnet_remotenet_t *rnet;
2763         __u32             dstnet = LNET_NIDNET(dstnid);
2764         int               hops;
2765         __u32             order = 2;
2766
2767         /* if !local_nid_dist_zero, I don't return a distance of 0 ever
2768          * (when lustre sees a distance of 0, it substitutes 0@lo), so I
2769          * keep order 0 free for 0@lo and order 1 free for a local NID
2770          * match */
2771
2772         LASSERT (the_lnet.ln_init);
2773         LASSERT (the_lnet.ln_refcount > 0);
2774
2775         LNET_LOCK();
2776
2777         cfs_list_for_each (e, &the_lnet.ln_nis) {
2778                 ni = cfs_list_entry(e, lnet_ni_t, ni_list);
2779
2780                 if (ni->ni_nid == dstnid) {
2781                         if (srcnidp != NULL)
2782                                 *srcnidp = dstnid;
2783                         if (orderp != NULL) {
2784                                 if (LNET_NETTYP(LNET_NIDNET(dstnid)) == LOLND)
2785                                         *orderp = 0;
2786                                 else
2787                                         *orderp = 1;
2788                         }
2789                         LNET_UNLOCK();
2790
2791                         return local_nid_dist_zero ? 0 : 1;
2792                 }
2793
2794                 if (LNET_NIDNET(ni->ni_nid) == dstnet) {
2795                         if (srcnidp != NULL)
2796                                 *srcnidp = ni->ni_nid;
2797                         if (orderp != NULL)
2798                                 *orderp = order;
2799                         LNET_UNLOCK();
2800                         return 1;
2801                 }
2802
2803                 order++;
2804         }
2805
2806         cfs_list_for_each (e, &the_lnet.ln_remote_nets) {
2807                 rnet = cfs_list_entry(e, lnet_remotenet_t, lrn_list);
2808
2809                 if (rnet->lrn_net == dstnet) {
2810                         lnet_route_t *route;
2811                         lnet_route_t *shortest = NULL;
2812
2813                         LASSERT (!cfs_list_empty(&rnet->lrn_routes));
2814
2815                         cfs_list_for_each_entry(route, &rnet->lrn_routes,
2816                                                 lr_list) {
2817                                 if (shortest == NULL ||
2818                                     route->lr_hops < shortest->lr_hops)
2819                                         shortest = route;
2820                         }
2821
2822                         LASSERT (shortest != NULL);
2823                         hops = shortest->lr_hops;
2824                         if (srcnidp != NULL)
2825                                 *srcnidp = shortest->lr_gateway->lp_ni->ni_nid;
2826                         if (orderp != NULL)
2827                                 *orderp = order;
2828                         LNET_UNLOCK();
2829                         return hops + 1;
2830                 }
2831                 order++;
2832         }
2833
2834         LNET_UNLOCK();
2835         return -EHOSTUNREACH;
2836 }
2837
2838 /**
2839  * Set the number of asynchronous messages expected from a target process.
2840  *
2841  * This function is only meaningful for userspace callers. It's a no-op when
2842  * called from kernel.
2843  *
2844  * Asynchronous messages are those that can come from a target when the
2845  * userspace process is not waiting for IO to complete; e.g., AST callbacks
2846  * from Lustre servers. Specifying the expected number of such messages
2847  * allows them to be eagerly received when user process is not running in
2848  * LNet; otherwise network errors may occur.
2849  *
2850  * \param id Process ID of the target process.
2851  * \param nasync Number of asynchronous messages expected from the target.
2852  *
2853  * \return 0 on success, and an error code otherwise.
2854  */
2855 int
2856 LNetSetAsync(lnet_process_id_t id, int nasync)
2857 {
2858 #ifdef __KERNEL__
2859         return 0;
2860 #else
2861         lnet_ni_t        *ni;
2862         lnet_remotenet_t *rnet;
2863         cfs_list_t       *tmp;
2864         lnet_route_t     *route;
2865         lnet_nid_t       *nids;
2866         int               nnids;
2867         int               maxnids = 256;
2868         int               rc = 0;
2869         int               rc2;
2870
2871         /* Target on a local network? */
2872         ni = lnet_net2ni(LNET_NIDNET(id.nid));
2873         if (ni != NULL) {
2874                 if (ni->ni_lnd->lnd_setasync != NULL)
2875                         rc = (ni->ni_lnd->lnd_setasync)(ni, id, nasync);
2876                 lnet_ni_decref(ni);
2877                 return rc;
2878         }
2879
2880         /* Target on a remote network: apply to routers */
2881  again:
2882         LIBCFS_ALLOC(nids, maxnids * sizeof(*nids));
2883         if (nids == NULL)
2884                 return -ENOMEM;
2885         nnids = 0;
2886
2887         /* Snapshot all the router NIDs */
2888         LNET_LOCK();
2889         rnet = lnet_find_net_locked(LNET_NIDNET(id.nid));
2890         if (rnet != NULL) {
2891                 cfs_list_for_each(tmp, &rnet->lrn_routes) {
2892                         if (nnids == maxnids) {
2893                                 LNET_UNLOCK();
2894                                 LIBCFS_FREE(nids, maxnids * sizeof(*nids));
2895                                 maxnids *= 2;
2896                                 goto again;
2897                         }
2898
2899                         route = cfs_list_entry(tmp, lnet_route_t, lr_list);
2900                         nids[nnids++] = route->lr_gateway->lp_nid;
2901                 }
2902         }
2903         LNET_UNLOCK();
2904
2905         /* set async on all the routers */
2906         while (nnids-- > 0) {
2907                 id.pid = LUSTRE_SRV_LNET_PID;
2908                 id.nid = nids[nnids];
2909
2910                 ni = lnet_net2ni(LNET_NIDNET(id.nid));
2911                 if (ni == NULL)
2912                         continue;
2913
2914                 if (ni->ni_lnd->lnd_setasync != NULL) {
2915                         rc2 = (ni->ni_lnd->lnd_setasync)(ni, id, nasync);
2916                         if (rc2 != 0)
2917                                 rc = rc2;
2918                 }
2919                 lnet_ni_decref(ni);
2920         }
2921
2922         LIBCFS_FREE(nids, maxnids * sizeof(*nids));
2923         return rc;
2924 #endif
2925 }