lnet/lnet/lib-move.c

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