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