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