lnet/klnds/ptllnd/ptllnd.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lnet/klnds/ptllnd/ptllnd.c
  37  *
  38  * Author: PJ Kirner <pjkirner@clusterfs.com>
  39  */
  40
  41 #include "ptllnd.h"
  42
  43 lnd_t kptllnd_lnd = {
  44         .lnd_type       = PTLLND,
  45         .lnd_startup    = kptllnd_startup,
  46         .lnd_shutdown   = kptllnd_shutdown,
  47         .lnd_ctl        = kptllnd_ctl,
  48         .lnd_query      = kptllnd_query,
  49         .lnd_send       = kptllnd_send,
  50         .lnd_recv       = kptllnd_recv,
  51         .lnd_eager_recv = kptllnd_eager_recv,
  52 };
  53
  54 kptl_data_t kptllnd_data;
  55
  56 char *
  57 kptllnd_ptlid2str(ptl_process_id_t id)
  58 {
  59         static char    strs[64][32];
  60         static int     idx = 0;
  61
  62         unsigned long  flags;
  63         char          *str;
  64
  65         spin_lock_irqsave(&kptllnd_data.kptl_ptlid2str_lock, flags);
  66         str = strs[idx++];
  67         if (idx >= sizeof(strs)/sizeof(strs[0]))
  68                 idx = 0;
  69         spin_unlock_irqrestore(&kptllnd_data.kptl_ptlid2str_lock, flags);
  70
  71         snprintf(str, sizeof(strs[0]), FMT_PTLID, id.pid, id.nid);
  72         return str;
  73 }
  74
  75 void
  76 kptllnd_assert_wire_constants (void)
  77 {
  78         /* Wire protocol assertions generated by 'wirecheck'
  79          * running on Linux fedora 2.6.11-co-0.6.4 #1 Mon Jun 19 05:36:13 UTC 2006 i686 i686 i386 GNU
  80          * with gcc version 4.1.1 20060525 (Red Hat 4.1.1-1) */
  81
  82
  83         /* Constants... */
  84         CLASSERT (PTL_RESERVED_MATCHBITS == 0x100);
  85         CLASSERT (LNET_MSG_MATCHBITS == 0);
  86         CLASSERT (PTLLND_MSG_MAGIC == 0x50746C4E);
  87         CLASSERT (PTLLND_MSG_VERSION == 0x04);
  88         CLASSERT (PTLLND_RDMA_OK == 0x00);
  89         CLASSERT (PTLLND_RDMA_FAIL == 0x01);
  90         CLASSERT (PTLLND_MSG_TYPE_INVALID == 0x00);
  91         CLASSERT (PTLLND_MSG_TYPE_PUT == 0x01);
  92         CLASSERT (PTLLND_MSG_TYPE_GET == 0x02);
  93         CLASSERT (PTLLND_MSG_TYPE_IMMEDIATE == 0x03);
  94         CLASSERT (PTLLND_MSG_TYPE_NOOP == 0x04);
  95         CLASSERT (PTLLND_MSG_TYPE_HELLO == 0x05);
  96         CLASSERT (PTLLND_MSG_TYPE_NAK == 0x06);
  97
  98         /* Checks for struct kptl_msg_t */
  99         CLASSERT ((int)sizeof(kptl_msg_t) == 136);
 100         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_magic) == 0);
 101         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_magic) == 4);
 102         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_version) == 4);
 103         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_version) == 2);
 104         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_type) == 6);
 105         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_type) == 1);
 106         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_credits) == 7);
 107         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_credits) == 1);
 108         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_nob) == 8);
 109         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_nob) == 4);
 110         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_cksum) == 12);
 111         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_cksum) == 4);
 112         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_srcnid) == 16);
 113         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_srcnid) == 8);
 114         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_srcstamp) == 24);
 115         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_srcstamp) == 8);
 116         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_dstnid) == 32);
 117         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_dstnid) == 8);
 118         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_dststamp) == 40);
 119         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_dststamp) == 8);
 120         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_srcpid) == 48);
 121         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_srcpid) == 4);
 122         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_dstpid) == 52);
 123         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_dstpid) == 4);
 124         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_u.immediate) == 56);
 125         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_u.immediate) == 72);
 126         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_u.rdma) == 56);
 127         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_u.rdma) == 80);
 128         CLASSERT ((int)offsetof(kptl_msg_t, ptlm_u.hello) == 56);
 129         CLASSERT ((int)sizeof(((kptl_msg_t *)0)->ptlm_u.hello) == 12);
 130
 131         /* Checks for struct kptl_immediate_msg_t */
 132         CLASSERT ((int)sizeof(kptl_immediate_msg_t) == 72);
 133         CLASSERT ((int)offsetof(kptl_immediate_msg_t, kptlim_hdr) == 0);
 134         CLASSERT ((int)sizeof(((kptl_immediate_msg_t *)0)->kptlim_hdr) == 72);
 135         CLASSERT ((int)offsetof(kptl_immediate_msg_t, kptlim_payload[13]) == 85);
 136         CLASSERT ((int)sizeof(((kptl_immediate_msg_t *)0)->kptlim_payload[13]) == 1);
 137
 138         /* Checks for struct kptl_rdma_msg_t */
 139         CLASSERT ((int)sizeof(kptl_rdma_msg_t) == 80);
 140         CLASSERT ((int)offsetof(kptl_rdma_msg_t, kptlrm_hdr) == 0);
 141         CLASSERT ((int)sizeof(((kptl_rdma_msg_t *)0)->kptlrm_hdr) == 72);
 142         CLASSERT ((int)offsetof(kptl_rdma_msg_t, kptlrm_matchbits) == 72);
 143         CLASSERT ((int)sizeof(((kptl_rdma_msg_t *)0)->kptlrm_matchbits) == 8);
 144
 145         /* Checks for struct kptl_hello_msg_t */
 146         CLASSERT ((int)sizeof(kptl_hello_msg_t) == 12);
 147         CLASSERT ((int)offsetof(kptl_hello_msg_t, kptlhm_matchbits) == 0);
 148         CLASSERT ((int)sizeof(((kptl_hello_msg_t *)0)->kptlhm_matchbits) == 8);
 149         CLASSERT ((int)offsetof(kptl_hello_msg_t, kptlhm_max_msg_size) == 8);
 150         CLASSERT ((int)sizeof(((kptl_hello_msg_t *)0)->kptlhm_max_msg_size) == 4);
 151 }
 152
 153 const char *kptllnd_evtype2str(int type)
 154 {
 155 #define DO_TYPE(x) case x: return #x;
 156         switch(type)
 157         {
 158                 DO_TYPE(PTL_EVENT_GET_START);
 159                 DO_TYPE(PTL_EVENT_GET_END);
 160                 DO_TYPE(PTL_EVENT_PUT_START);
 161                 DO_TYPE(PTL_EVENT_PUT_END);
 162                 DO_TYPE(PTL_EVENT_REPLY_START);
 163                 DO_TYPE(PTL_EVENT_REPLY_END);
 164                 DO_TYPE(PTL_EVENT_ACK);
 165                 DO_TYPE(PTL_EVENT_SEND_START);
 166                 DO_TYPE(PTL_EVENT_SEND_END);
 167                 DO_TYPE(PTL_EVENT_UNLINK);
 168         default:
 169                 return "<unknown event type>";
 170         }
 171 #undef DO_TYPE
 172 }
 173
 174 const char *kptllnd_msgtype2str(int type)
 175 {
 176 #define DO_TYPE(x) case x: return #x;
 177         switch(type)
 178         {
 179                 DO_TYPE(PTLLND_MSG_TYPE_INVALID);
 180                 DO_TYPE(PTLLND_MSG_TYPE_PUT);
 181                 DO_TYPE(PTLLND_MSG_TYPE_GET);
 182                 DO_TYPE(PTLLND_MSG_TYPE_IMMEDIATE);
 183                 DO_TYPE(PTLLND_MSG_TYPE_HELLO);
 184                 DO_TYPE(PTLLND_MSG_TYPE_NOOP);
 185                 DO_TYPE(PTLLND_MSG_TYPE_NAK);
 186         default:
 187                 return "<unknown msg type>";
 188         }
 189 #undef DO_TYPE
 190 }
 191
 192 const char *kptllnd_errtype2str(int type)
 193 {
 194 #define DO_TYPE(x) case x: return #x;
 195         switch(type)
 196         {
 197                 DO_TYPE(PTL_OK);
 198                 DO_TYPE(PTL_SEGV);
 199                 DO_TYPE(PTL_NO_SPACE);
 200                 DO_TYPE(PTL_ME_IN_USE);
 201                 DO_TYPE(PTL_NAL_FAILED);
 202                 DO_TYPE(PTL_NO_INIT);
 203                 DO_TYPE(PTL_IFACE_DUP);
 204                 DO_TYPE(PTL_IFACE_INVALID);
 205                 DO_TYPE(PTL_HANDLE_INVALID);
 206                 DO_TYPE(PTL_MD_INVALID);
 207                 DO_TYPE(PTL_ME_INVALID);
 208                 DO_TYPE(PTL_PROCESS_INVALID);
 209                 DO_TYPE(PTL_PT_INDEX_INVALID);
 210                 DO_TYPE(PTL_SR_INDEX_INVALID);
 211                 DO_TYPE(PTL_EQ_INVALID);
 212                 DO_TYPE(PTL_EQ_DROPPED);
 213                 DO_TYPE(PTL_EQ_EMPTY);
 214                 DO_TYPE(PTL_MD_NO_UPDATE);
 215                 DO_TYPE(PTL_FAIL);
 216                 DO_TYPE(PTL_AC_INDEX_INVALID);
 217                 DO_TYPE(PTL_MD_ILLEGAL);
 218                 DO_TYPE(PTL_ME_LIST_TOO_LONG);
 219                 DO_TYPE(PTL_MD_IN_USE);
 220                 DO_TYPE(PTL_NI_INVALID);
 221                 DO_TYPE(PTL_PID_INVALID);
 222                 DO_TYPE(PTL_PT_FULL);
 223                 DO_TYPE(PTL_VAL_FAILED);
 224                 DO_TYPE(PTL_NOT_IMPLEMENTED);
 225                 DO_TYPE(PTL_NO_ACK);
 226                 DO_TYPE(PTL_EQ_IN_USE);
 227                 DO_TYPE(PTL_PID_IN_USE);
 228                 DO_TYPE(PTL_INV_EQ_SIZE);
 229                 DO_TYPE(PTL_AGAIN);
 230         default:
 231                 return "<unknown event type>";
 232         }
 233 #undef DO_TYPE
 234 }
 235
 236 __u32
 237 kptllnd_cksum (void *ptr, int nob)
 238 {
 239         char  *c  = ptr;
 240         __u32  sum = 0;
 241
 242         while (nob-- > 0)
 243                 sum = ((sum << 1) | (sum >> 31)) + *c++;
 244
 245         /* ensure I don't return 0 (== no checksum) */
 246         return (sum == 0) ? 1 : sum;
 247 }
 248
 249 void
 250 kptllnd_init_msg(kptl_msg_t *msg, int type,
 251                  lnet_process_id_t target, int body_nob)
 252 {
 253         msg->ptlm_type = type;
 254         msg->ptlm_nob  = (offsetof(kptl_msg_t, ptlm_u) + body_nob + 7) & ~7;
 255         msg->ptlm_dstpid = target.pid;
 256         msg->ptlm_dstnid = target.nid;
 257         msg->ptlm_srcpid = the_lnet.ln_pid;
 258         msg->ptlm_srcnid = kptllnd_ptl2lnetnid(target.nid,
 259                                                kptllnd_data.kptl_portals_id.nid);
 260
 261         LASSERT(msg->ptlm_nob <= *kptllnd_tunables.kptl_max_msg_size);
 262 }
 263
 264 void
 265 kptllnd_msg_pack(kptl_msg_t *msg, kptl_peer_t *peer)
 266 {
 267         msg->ptlm_magic    = PTLLND_MSG_MAGIC;
 268         msg->ptlm_version  = PTLLND_MSG_VERSION;
 269         /* msg->ptlm_type  Filled in kptllnd_init_msg()  */
 270         msg->ptlm_credits  = peer->peer_outstanding_credits;
 271         /* msg->ptlm_nob   Filled in kptllnd_init_msg()  */
 272         msg->ptlm_cksum    = 0;
 273         /* msg->ptlm_{src|dst}[pn]id Filled in kptllnd_init_msg */
 274         msg->ptlm_srcstamp = peer->peer_myincarnation;
 275         msg->ptlm_dststamp = peer->peer_incarnation;
 276
 277         if (*kptllnd_tunables.kptl_checksum) {
 278                 /* NB ptlm_cksum zero while computing cksum */
 279                 msg->ptlm_cksum = kptllnd_cksum(msg,
 280                                                 offsetof(kptl_msg_t, ptlm_u));
 281         }
 282 }
 283
 284 int
 285 kptllnd_msg_unpack(kptl_msg_t *msg, int nob)
 286 {
 287         const int hdr_size = offsetof(kptl_msg_t, ptlm_u);
 288         __u32     msg_cksum;
 289         __u16     msg_version;
 290         int       flip;
 291
 292         /* 6 bytes are enough to have received magic + version */
 293         if (nob < 6) {
 294                 CERROR("Very Short message: %d\n", nob);
 295                 return -EPROTO;
 296         }
 297
 298         /*
 299          * Determine if we need to flip
 300          */
 301         if (msg->ptlm_magic == PTLLND_MSG_MAGIC) {
 302                 flip = 0;
 303         } else if (msg->ptlm_magic == __swab32(PTLLND_MSG_MAGIC)) {
 304                 flip = 1;
 305         } else {
 306                 CERROR("Bad magic: %08x\n", msg->ptlm_magic);
 307                 return -EPROTO;
 308         }
 309
 310         msg_version = flip ? __swab16(msg->ptlm_version) : msg->ptlm_version;
 311
 312         if (msg_version != PTLLND_MSG_VERSION) {
 313                 CERROR("Bad version: got %04x expected %04x\n",
 314                         (__u32)msg_version, PTLLND_MSG_VERSION);
 315                 return -EPROTO;
 316         }
 317
 318         if (nob < hdr_size) {
 319                 CERROR("Short message: got %d, wanted at least %d\n",
 320                        nob, hdr_size);
 321                 return -EPROTO;
 322         }
 323
 324         /* checksum must be computed with
 325          * 1) ptlm_cksum zero and
 326          * 2) BEFORE anything gets modified/flipped
 327          */
 328         msg_cksum = flip ? __swab32(msg->ptlm_cksum) : msg->ptlm_cksum;
 329         msg->ptlm_cksum = 0;
 330         if (msg_cksum != 0 &&
 331             msg_cksum != kptllnd_cksum(msg, hdr_size)) {
 332                 CERROR("Bad checksum\n");
 333                 return -EPROTO;
 334         }
 335
 336         msg->ptlm_version = msg_version;
 337         msg->ptlm_cksum = msg_cksum;
 338
 339         if (flip) {
 340                 /* These two are 1 byte long so we don't swap them
 341                    But check this assumtion*/
 342                 CLASSERT (sizeof(msg->ptlm_type) == 1);
 343                 CLASSERT (sizeof(msg->ptlm_credits) == 1);
 344                 /* src & dst stamps are opaque cookies */
 345                 __swab32s(&msg->ptlm_nob);
 346                 __swab64s(&msg->ptlm_srcnid);
 347                 __swab64s(&msg->ptlm_dstnid);
 348                 __swab32s(&msg->ptlm_srcpid);
 349                 __swab32s(&msg->ptlm_dstpid);
 350         }
 351
 352         if (msg->ptlm_nob != nob) {
 353                 CERROR("msg_nob corrupt: got 0x%08x, wanted %08x\n",
 354                        msg->ptlm_nob, nob);
 355                 return -EPROTO;
 356         }
 357
 358         switch(msg->ptlm_type)
 359         {
 360         case PTLLND_MSG_TYPE_PUT:
 361         case PTLLND_MSG_TYPE_GET:
 362                 if (nob < hdr_size + sizeof(kptl_rdma_msg_t)) {
 363                         CERROR("Short rdma request: got %d, want %d\n",
 364                                nob, hdr_size + (int)sizeof(kptl_rdma_msg_t));
 365                         return -EPROTO;
 366                 }
 367
 368                 if (flip)
 369                         __swab64s(&msg->ptlm_u.rdma.kptlrm_matchbits);
 370
 371                 if (msg->ptlm_u.rdma.kptlrm_matchbits < PTL_RESERVED_MATCHBITS) {
 372                         CERROR("Bad matchbits "LPX64"\n",
 373                                msg->ptlm_u.rdma.kptlrm_matchbits);
 374                         return -EPROTO;
 375                 }
 376                 break;
 377
 378         case PTLLND_MSG_TYPE_IMMEDIATE:
 379                 if (nob < offsetof(kptl_msg_t,
 380                                    ptlm_u.immediate.kptlim_payload)) {
 381                         CERROR("Short immediate: got %d, want %d\n", nob,
 382                                (int)offsetof(kptl_msg_t,
 383                                              ptlm_u.immediate.kptlim_payload));
 384                         return -EPROTO;
 385                 }
 386                 /* Do nothing */
 387                 break;
 388
 389         case PTLLND_MSG_TYPE_NOOP:
 390         case PTLLND_MSG_TYPE_NAK:
 391                 /* Do nothing */
 392                 break;
 393
 394         case PTLLND_MSG_TYPE_HELLO:
 395                 if (nob < hdr_size + sizeof(kptl_hello_msg_t)) {
 396                         CERROR("Short hello: got %d want %d\n",
 397                                nob, hdr_size + (int)sizeof(kptl_hello_msg_t));
 398                         return -EPROTO;
 399                 }
 400                 if (flip) {
 401                         __swab64s(&msg->ptlm_u.hello.kptlhm_matchbits);
 402                         __swab32s(&msg->ptlm_u.hello.kptlhm_max_msg_size);
 403                 }
 404                 break;
 405
 406         default:
 407                 CERROR("Bad message type: 0x%02x\n", (__u32)msg->ptlm_type);
 408                 return -EPROTO;
 409         }
 410
 411         return 0;
 412 }
 413
 414 int
 415 kptllnd_ctl(lnet_ni_t *ni, unsigned int cmd, void *arg)
 416 {
 417         kptl_net_t  *net = ni->ni_data;
 418         struct libcfs_ioctl_data *data = arg;
 419         int          rc = -EINVAL;
 420
 421         CDEBUG(D_NET, ">>> kptllnd_ctl cmd=%u arg=%p\n", cmd, arg);
 422
 423         /*
 424          * Validate that the context block is actually
 425          * pointing to this interface
 426          */
 427         LASSERT (ni == net->net_ni);
 428
 429         switch(cmd) {
 430         case IOC_LIBCFS_DEL_PEER: {
 431                 lnet_process_id_t id;
 432
 433                 id.nid = data->ioc_nid;
 434                 id.pid = data->ioc_u32[1];
 435
 436                 rc = kptllnd_peer_del(id);
 437                 break;
 438         }
 439
 440         case IOC_LIBCFS_GET_PEER: {
 441                 lnet_process_id_t   id = {.nid = LNET_NID_ANY,
 442                                           .pid = LNET_PID_ANY};
 443                 __u64               incarnation = 0;
 444                 __u64               next_matchbits = 0;
 445                 __u64               last_matchbits_seen = 0;
 446                 int                 state = 0;
 447                 int                 sent_hello = 0;
 448                 int                 refcount = 0;
 449                 int                 nsendq = 0;
 450                 int                 nactiveq = 0;
 451                 int                 credits = 0;
 452                 int                 outstanding_credits = 0;
 453
 454                 rc = kptllnd_get_peer_info(data->ioc_count, &id,
 455                                            &state, &sent_hello,
 456                                            &refcount, &incarnation,
 457                                            &next_matchbits, &last_matchbits_seen,
 458                                            &nsendq, &nactiveq,
 459                                            &credits, &outstanding_credits);
 460                 /* wince... */
 461                 data->ioc_nid = id.nid;
 462                 data->ioc_net = state;
 463                 data->ioc_flags  = sent_hello;
 464                 data->ioc_count = refcount;
 465                 data->ioc_u64[0] = incarnation;
 466                 data->ioc_u32[0] = (__u32)next_matchbits;
 467                 data->ioc_u32[1] = (__u32)(next_matchbits >> 32);
 468                 data->ioc_u32[2] = (__u32)last_matchbits_seen;
 469                 data->ioc_u32[3] = (__u32)(last_matchbits_seen >> 32);
 470                 data->ioc_u32[4] = id.pid;
 471                 data->ioc_u32[5] = (nsendq << 16) | nactiveq;
 472                 data->ioc_u32[6] = (credits << 16) | outstanding_credits;
 473                 break;
 474         }
 475
 476         default:
 477                 rc=-EINVAL;
 478                 break;
 479         }
 480         CDEBUG(D_NET, "<<< kptllnd_ctl rc=%d\n", rc);
 481         return rc;
 482 }
 483
 484 void
 485 kptllnd_query (lnet_ni_t *ni, lnet_nid_t nid, cfs_time_t *when)
 486 {
 487         kptl_net_t        *net = ni->ni_data;
 488         kptl_peer_t       *peer = NULL;
 489         lnet_process_id_t  id = {.nid = nid, .pid = LUSTRE_SRV_LNET_PID};
 490         unsigned long      flags;
 491
 492         /* NB: kptllnd_find_target connects to peer if necessary */
 493         if (kptllnd_find_target(net, id, &peer) != 0)
 494                 return;
 495
 496         spin_lock_irqsave(&peer->peer_lock, flags);
 497         if (peer->peer_last_alive != 0)
 498                 *when = peer->peer_last_alive;
 499         spin_unlock_irqrestore(&peer->peer_lock, flags);
 500         kptllnd_peer_decref(peer);
 501         return;
 502 }
 503
 504 void
 505 kptllnd_base_shutdown (void)
 506 {
 507         int               i;
 508         ptl_err_t         prc;
 509         unsigned long     flags;
 510         lnet_process_id_t process_id;
 511
 512         read_lock(&kptllnd_data.kptl_net_rw_lock);
 513         LASSERT (list_empty(&kptllnd_data.kptl_nets));
 514         read_unlock(&kptllnd_data.kptl_net_rw_lock);
 515
 516         switch (kptllnd_data.kptl_init) {
 517         default:
 518                 LBUG();
 519
 520         case PTLLND_INIT_ALL:
 521         case PTLLND_INIT_DATA:
 522                 /* stop receiving */
 523                 kptllnd_rx_buffer_pool_fini(&kptllnd_data.kptl_rx_buffer_pool);
 524                 LASSERT (list_empty(&kptllnd_data.kptl_sched_rxq));
 525                 LASSERT (list_empty(&kptllnd_data.kptl_sched_rxbq));
 526
 527                 /* lock to interleave cleanly with peer birth/death */
 528                 write_lock_irqsave(&kptllnd_data.kptl_peer_rw_lock, flags);
 529                 LASSERT (kptllnd_data.kptl_shutdown == 0);
 530                 kptllnd_data.kptl_shutdown = 1; /* phase 1 == destroy peers */
 531                 /* no new peers possible now */
 532                 write_unlock_irqrestore(&kptllnd_data.kptl_peer_rw_lock, flags);
 533
 534                 /* nuke all existing peers */
 535                 process_id.nid = LNET_NID_ANY;
 536                 process_id.pid = LNET_PID_ANY;
 537                 kptllnd_peer_del(process_id);
 538
 539                 read_lock_irqsave(&kptllnd_data.kptl_peer_rw_lock, flags);
 540
 541                 LASSERT (kptllnd_data.kptl_n_active_peers == 0);
 542
 543                 i = 2;
 544                 while (kptllnd_data.kptl_npeers != 0) {
 545                         i++;
 546                         CDEBUG(((i & (-i)) == i) ? D_WARNING : D_NET,
 547                                "Waiting for %d peers to terminate\n",
 548                                kptllnd_data.kptl_npeers);
 549
 550                         read_unlock_irqrestore(&kptllnd_data.kptl_peer_rw_lock,
 551                                                flags);
 552
 553                         cfs_pause(cfs_time_seconds(1));
 554
 555                         read_lock_irqsave(&kptllnd_data.kptl_peer_rw_lock,
 556                                           flags);
 557                 }
 558
 559                 LASSERT (list_empty(&kptllnd_data.kptl_closing_peers));
 560                 LASSERT (list_empty(&kptllnd_data.kptl_zombie_peers));
 561                 LASSERT (kptllnd_data.kptl_peers != NULL);
 562                 for (i = 0; i < kptllnd_data.kptl_peer_hash_size; i++)
 563                         LASSERT (list_empty (&kptllnd_data.kptl_peers[i]));
 564
 565                 read_unlock_irqrestore(&kptllnd_data.kptl_peer_rw_lock, flags);
 566                 CDEBUG(D_NET, "All peers deleted\n");
 567
 568                 /* Shutdown phase 2: kill the daemons... */
 569                 kptllnd_data.kptl_shutdown = 2;
 570                 mb();
 571
 572                 i = 2;
 573                 while (atomic_read (&kptllnd_data.kptl_nthreads) != 0) {
 574                         /* Wake up all threads*/
 575                         wake_up_all(&kptllnd_data.kptl_sched_waitq);
 576                         wake_up_all(&kptllnd_data.kptl_watchdog_waitq);
 577
 578                         i++;
 579                         CDEBUG(((i & (-i)) == i) ? D_WARNING : D_NET, /* power of 2? */
 580                                "Waiting for %d threads to terminate\n",
 581                                atomic_read(&kptllnd_data.kptl_nthreads));
 582                         cfs_pause(cfs_time_seconds(1));
 583                 }
 584
 585                 CDEBUG(D_NET, "All Threads stopped\n");
 586                 LASSERT(list_empty(&kptllnd_data.kptl_sched_txq));
 587
 588                 kptllnd_cleanup_tx_descs();
 589
 590                 /* Nothing here now, but libcfs might soon require
 591                  * us to explicitly destroy wait queues and semaphores
 592                  * that would be done here */
 593
 594                 /* fall through */
 595
 596         case PTLLND_INIT_NOTHING:
 597                 CDEBUG(D_NET, "PTLLND_INIT_NOTHING\n");
 598                 break;
 599         }
 600
 601         if (!PtlHandleIsEqual(kptllnd_data.kptl_eqh, PTL_INVALID_HANDLE)) {
 602                 prc = PtlEQFree(kptllnd_data.kptl_eqh);
 603                 if (prc != PTL_OK)
 604                         CERROR("Error %s(%d) freeing portals EQ\n",
 605                                kptllnd_errtype2str(prc), prc);
 606         }
 607
 608         if (!PtlHandleIsEqual(kptllnd_data.kptl_nih, PTL_INVALID_HANDLE)) {
 609                 prc = PtlNIFini(kptllnd_data.kptl_nih);
 610                 if (prc != PTL_OK)
 611                         CERROR("Error %s(%d) finalizing portals NI\n",
 612                                kptllnd_errtype2str(prc), prc);
 613         }
 614
 615         LASSERT (atomic_read(&kptllnd_data.kptl_ntx) == 0);
 616         LASSERT (list_empty(&kptllnd_data.kptl_idle_txs));
 617
 618         if (kptllnd_data.kptl_rx_cache != NULL)
 619                 cfs_mem_cache_destroy(kptllnd_data.kptl_rx_cache);
 620
 621         if (kptllnd_data.kptl_peers != NULL)
 622                 LIBCFS_FREE(kptllnd_data.kptl_peers,
 623                             sizeof (struct list_head) *
 624                             kptllnd_data.kptl_peer_hash_size);
 625
 626         if (kptllnd_data.kptl_nak_msg != NULL)
 627                 LIBCFS_FREE(kptllnd_data.kptl_nak_msg,
 628                             offsetof(kptl_msg_t, ptlm_u));
 629
 630         memset(&kptllnd_data, 0, sizeof(kptllnd_data));
 631         PORTAL_MODULE_UNUSE;
 632         return;
 633 }
 634
 635 int
 636 kptllnd_base_startup (void)
 637 {
 638         int             i;
 639         int                rc;
 640         int             spares;
 641         struct timeval  tv;
 642         lnet_process_id_t  target;
 643         ptl_err_t       ptl_rc;
 644
 645         if (*kptllnd_tunables.kptl_max_procs_per_node < 1) {
 646                 CERROR("max_procs_per_node must be >= 1\n");
 647                 return -EINVAL;
 648         }
 649
 650         if (*kptllnd_tunables.kptl_peertxcredits > PTLLND_MSG_MAX_CREDITS) {
 651                 CERROR("peercredits must be <= %d\n", PTLLND_MSG_MAX_CREDITS);
 652                 return -EINVAL;
 653         }
 654
 655         *kptllnd_tunables.kptl_max_msg_size &= ~7;
 656         if (*kptllnd_tunables.kptl_max_msg_size < PTLLND_MIN_BUFFER_SIZE)
 657                 *kptllnd_tunables.kptl_max_msg_size = PTLLND_MIN_BUFFER_SIZE;
 658
 659         CLASSERT ((PTLLND_MIN_BUFFER_SIZE & 7) == 0);
 660         CLASSERT (sizeof(kptl_msg_t) <= PTLLND_MIN_BUFFER_SIZE);
 661
 662         /* Zero pointers, flags etc; put everything into a known state. */
 663         memset (&kptllnd_data, 0, sizeof (kptllnd_data));
 664
 665         LIBCFS_ALLOC(kptllnd_data.kptl_nak_msg, offsetof(kptl_msg_t, ptlm_u));
 666         if (kptllnd_data.kptl_nak_msg == NULL) {
 667                 CERROR("Can't allocate NAK msg\n");
 668                 return -ENOMEM;
 669         }
 670         memset(kptllnd_data.kptl_nak_msg, 0, offsetof(kptl_msg_t, ptlm_u));
 671
 672         kptllnd_data.kptl_eqh = PTL_INVALID_HANDLE;
 673         kptllnd_data.kptl_nih = PTL_INVALID_HANDLE;
 674
 675         rwlock_init(&kptllnd_data.kptl_net_rw_lock);
 676         INIT_LIST_HEAD(&kptllnd_data.kptl_nets);
 677
 678         /* Setup the sched locks/lists/waitq */
 679         spin_lock_init(&kptllnd_data.kptl_sched_lock);
 680         init_waitqueue_head(&kptllnd_data.kptl_sched_waitq);
 681         INIT_LIST_HEAD(&kptllnd_data.kptl_sched_txq);
 682         INIT_LIST_HEAD(&kptllnd_data.kptl_sched_rxq);
 683         INIT_LIST_HEAD(&kptllnd_data.kptl_sched_rxbq);
 684
 685         /* Init kptl_ptlid2str_lock before any call to kptllnd_ptlid2str */
 686         spin_lock_init(&kptllnd_data.kptl_ptlid2str_lock);
 687
 688         /* Setup the tx locks/lists */
 689         spin_lock_init(&kptllnd_data.kptl_tx_lock);
 690         INIT_LIST_HEAD(&kptllnd_data.kptl_idle_txs);
 691         atomic_set(&kptllnd_data.kptl_ntx, 0);
 692
 693         /* Uptick the module reference count */
 694         PORTAL_MODULE_USE;
 695
 696         kptllnd_data.kptl_expected_peers =
 697                 *kptllnd_tunables.kptl_max_nodes *
 698                 *kptllnd_tunables.kptl_max_procs_per_node;
 699
 700         /*
 701          * Initialize the Network interface instance
 702          * We use the default because we don't have any
 703          * way to choose a better interface.
 704          * Requested and actual limits are ignored.
 705          */
 706         ptl_rc = PtlNIInit(
 707 #ifdef _USING_LUSTRE_PORTALS_
 708                 PTL_IFACE_DEFAULT,
 709 #else
 710                 CRAY_KERN_NAL,
 711 #endif
 712                 *kptllnd_tunables.kptl_pid, NULL, NULL,
 713                 &kptllnd_data.kptl_nih);
 714
 715         /*
 716          * Note: PTL_IFACE_DUP simply means that the requested
 717          * interface was already inited and that we're sharing it.
 718          * Which is ok.
 719          */
 720         if (ptl_rc != PTL_OK && ptl_rc != PTL_IFACE_DUP) {
 721                 CERROR ("PtlNIInit: error %s(%d)\n",
 722                         kptllnd_errtype2str(ptl_rc), ptl_rc);
 723                 rc = -EINVAL;
 724                 goto failed;
 725         }
 726
 727         /* NB eq size irrelevant if using a callback */
 728         ptl_rc = PtlEQAlloc(kptllnd_data.kptl_nih,
 729                             8,                       /* size */
 730                             kptllnd_eq_callback,     /* handler callback */
 731                             &kptllnd_data.kptl_eqh); /* output handle */
 732         if (ptl_rc != PTL_OK) {
 733                 CERROR("PtlEQAlloc failed %s(%d)\n",
 734                        kptllnd_errtype2str(ptl_rc), ptl_rc);
 735                 rc = -ENOMEM;
 736                 goto failed;
 737         }
 738
 739         /* Fetch the lower NID */
 740         ptl_rc = PtlGetId(kptllnd_data.kptl_nih,
 741                           &kptllnd_data.kptl_portals_id);
 742         if (ptl_rc != PTL_OK) {
 743                 CERROR ("PtlGetID: error %s(%d)\n",
 744                         kptllnd_errtype2str(ptl_rc), ptl_rc);
 745                 rc = -EINVAL;
 746                 goto failed;
 747         }
 748
 749         if (kptllnd_data.kptl_portals_id.pid != *kptllnd_tunables.kptl_pid) {
 750                 /* The kernel ptllnd must have the expected PID */
 751                 CERROR("Unexpected PID: %u (%u expected)\n",
 752                        kptllnd_data.kptl_portals_id.pid,
 753                        *kptllnd_tunables.kptl_pid);
 754                 rc = -EINVAL;
 755                 goto failed;
 756         }
 757
 758         /* Initialized the incarnation - it must be for-all-time unique, even
 759          * accounting for the fact that we increment it when we disconnect a
 760          * peer that's using it */
 761         do_gettimeofday(&tv);
 762         kptllnd_data.kptl_incarnation = (((__u64)tv.tv_sec) * 1000000) +
 763                                         tv.tv_usec;
 764         CDEBUG(D_NET, "Incarnation="LPX64"\n", kptllnd_data.kptl_incarnation);
 765
 766         target.nid = LNET_NID_ANY;
 767         target.pid = LNET_PID_ANY; /* NB target for NAK doesn't matter */
 768         kptllnd_init_msg(kptllnd_data.kptl_nak_msg, PTLLND_MSG_TYPE_NAK, target, 0);
 769         kptllnd_data.kptl_nak_msg->ptlm_magic    = PTLLND_MSG_MAGIC;
 770         kptllnd_data.kptl_nak_msg->ptlm_version  = PTLLND_MSG_VERSION;
 771         kptllnd_data.kptl_nak_msg->ptlm_srcpid   = the_lnet.ln_pid;
 772         kptllnd_data.kptl_nak_msg->ptlm_srcstamp = kptllnd_data.kptl_incarnation;
 773
 774         rwlock_init(&kptllnd_data.kptl_peer_rw_lock);
 775         init_waitqueue_head(&kptllnd_data.kptl_watchdog_waitq);
 776         atomic_set(&kptllnd_data.kptl_needs_ptltrace, 0);
 777         INIT_LIST_HEAD(&kptllnd_data.kptl_closing_peers);
 778         INIT_LIST_HEAD(&kptllnd_data.kptl_zombie_peers);
 779
 780         /* Allocate and setup the peer hash table */
 781         kptllnd_data.kptl_peer_hash_size =
 782                 *kptllnd_tunables.kptl_peer_hash_table_size;
 783         LIBCFS_ALLOC(kptllnd_data.kptl_peers,
 784                      sizeof(struct list_head) *
 785                      kptllnd_data.kptl_peer_hash_size);
 786         if (kptllnd_data.kptl_peers == NULL) {
 787                 CERROR("Failed to allocate space for peer hash table size=%d\n",
 788                         kptllnd_data.kptl_peer_hash_size);
 789                 rc = -ENOMEM;
 790                 goto failed;
 791         }
 792         for (i = 0; i < kptllnd_data.kptl_peer_hash_size; i++)
 793                 INIT_LIST_HEAD(&kptllnd_data.kptl_peers[i]);
 794
 795         kptllnd_rx_buffer_pool_init(&kptllnd_data.kptl_rx_buffer_pool);
 796
 797         kptllnd_data.kptl_rx_cache =
 798                 cfs_mem_cache_create("ptllnd_rx",
 799                                      sizeof(kptl_rx_t) +
 800                                      *kptllnd_tunables.kptl_max_msg_size,
 801                                      0,    /* offset */
 802                                      0);   /* flags */
 803         if (kptllnd_data.kptl_rx_cache == NULL) {
 804                 CERROR("Can't create slab for RX descriptors\n");
 805                 rc = -ENOMEM;
 806                 goto failed;
 807         }
 808
 809         /* lists/ptrs/locks initialised */
 810         kptllnd_data.kptl_init = PTLLND_INIT_DATA;
 811
 812         /*****************************************************/
 813
 814         rc = kptllnd_setup_tx_descs();
 815         if (rc != 0) {
 816                 CERROR("Can't pre-allocate %d TX descriptors: %d\n",
 817                        *kptllnd_tunables.kptl_ntx, rc);
 818                 goto failed;
 819         }
 820
 821         /* Start the scheduler threads for handling incoming requests.  No need
 822          * to advance the state because this will be automatically cleaned up
 823          * now that PTLLND_INIT_DATA state has been entered */
 824         CDEBUG(D_NET, "starting %d scheduler threads\n", PTLLND_N_SCHED);
 825         for (i = 0; i < PTLLND_N_SCHED; i++) {
 826                 rc = kptllnd_thread_start(kptllnd_scheduler, (void *)((long)i));
 827                 if (rc != 0) {
 828                         CERROR("Can't spawn scheduler[%d]: %d\n", i, rc);
 829                         goto failed;
 830                 }
 831         }
 832
 833         rc = kptllnd_thread_start(kptllnd_watchdog, NULL);
 834         if (rc != 0) {
 835                 CERROR("Can't spawn watchdog: %d\n", rc);
 836                 goto failed;
 837         }
 838
 839         /* Ensure that 'rxb_nspare' buffers can be off the net (being emptied)
 840          * and we will still have enough buffers posted for all our peers */
 841         spares = *kptllnd_tunables.kptl_rxb_nspare *
 842                  ((*kptllnd_tunables.kptl_rxb_npages * PAGE_SIZE)/
 843                   *kptllnd_tunables.kptl_max_msg_size);
 844
 845         /* reserve and post the buffers */
 846         rc = kptllnd_rx_buffer_pool_reserve(&kptllnd_data.kptl_rx_buffer_pool,
 847                                             kptllnd_data.kptl_expected_peers +
 848                                             spares);
 849         if (rc != 0) {
 850                 CERROR("Can't reserve RX Buffer pool: %d\n", rc);
 851                 goto failed;
 852         }
 853
 854         /* flag everything initialised */
 855         kptllnd_data.kptl_init = PTLLND_INIT_ALL;
 856
 857         /*****************************************************/
 858
 859         if (*kptllnd_tunables.kptl_checksum)
 860                 CWARN("Checksumming enabled\n");
 861
 862         CDEBUG(D_NET, "<<< kptllnd_base_startup SUCCESS\n");
 863         return 0;
 864
 865  failed:
 866         CERROR("kptllnd_base_startup failed: %d\n", rc);
 867         kptllnd_base_shutdown();
 868         return rc;
 869 }
 870
 871 int
 872 kptllnd_startup (lnet_ni_t *ni)
 873 {
 874         int         rc;
 875         kptl_net_t *net;
 876
 877         LASSERT (ni->ni_lnd == &kptllnd_lnd);
 878
 879         if (kptllnd_data.kptl_init == PTLLND_INIT_NOTHING) {
 880                 rc = kptllnd_base_startup();
 881                 if (rc != 0)
 882                         return rc;
 883         }
 884
 885         LIBCFS_ALLOC(net, sizeof(*net));
 886         ni->ni_data = net;
 887         if (net == NULL) {
 888                 CERROR("Can't allocate kptl_net_t\n");
 889                 rc = -ENOMEM;
 890                 goto failed;
 891         }
 892         memset(net, 0, sizeof(*net));
 893         net->net_ni = ni;
 894
 895         ni->ni_maxtxcredits   = *kptllnd_tunables.kptl_credits;
 896         ni->ni_peertxcredits  = *kptllnd_tunables.kptl_peertxcredits;
 897         ni->ni_peerrtrcredits = *kptllnd_tunables.kptl_peerrtrcredits;
 898         ni->ni_nid = kptllnd_ptl2lnetnid(ni->ni_nid,
 899                                          kptllnd_data.kptl_portals_id.nid);
 900         CDEBUG(D_NET, "ptl id=%s, lnet id=%s\n",
 901                kptllnd_ptlid2str(kptllnd_data.kptl_portals_id),
 902                libcfs_nid2str(ni->ni_nid));
 903
 904         /* NB LNET_NIDNET(ptlm_srcnid) of NAK doesn't matter in case of
 905          * multiple NIs */
 906         kptllnd_data.kptl_nak_msg->ptlm_srcnid = ni->ni_nid;
 907
 908         atomic_set(&net->net_refcount, 1);
 909         write_lock(&kptllnd_data.kptl_net_rw_lock);
 910         list_add_tail(&net->net_list, &kptllnd_data.kptl_nets);
 911         write_unlock(&kptllnd_data.kptl_net_rw_lock);
 912         return 0;
 913
 914  failed:
 915         kptllnd_shutdown(ni);
 916         return rc;
 917 }
 918
 919 void
 920 kptllnd_shutdown (lnet_ni_t *ni)
 921 {
 922         kptl_net_t    *net = ni->ni_data;
 923         int               i;
 924         unsigned long     flags;
 925
 926         LASSERT (kptllnd_data.kptl_init == PTLLND_INIT_ALL);
 927
 928         CDEBUG(D_MALLOC, "before LND cleanup: kmem %d\n",
 929                atomic_read (&libcfs_kmemory));
 930
 931         if (net == NULL)
 932                 goto out;
 933
 934         LASSERT (ni == net->net_ni);
 935         LASSERT (!net->net_shutdown);
 936         LASSERT (!list_empty(&net->net_list));
 937         LASSERT (atomic_read(&net->net_refcount) != 0);
 938         ni->ni_data = NULL;
 939         net->net_ni = NULL;
 940
 941         write_lock(&kptllnd_data.kptl_net_rw_lock);
 942         kptllnd_net_decref(net);
 943         list_del_init(&net->net_list);
 944         write_unlock(&kptllnd_data.kptl_net_rw_lock);
 945
 946         /* Can't nuke peers here - they are shared among all NIs */
 947                 write_lock_irqsave(&kptllnd_data.kptl_peer_rw_lock, flags);
 948         net->net_shutdown = 1;   /* Order with peer creation */
 949         write_unlock_irqrestore(&kptllnd_data.kptl_peer_rw_lock, flags);
 950
 951                 i = 2;
 952         while (atomic_read(&net->net_refcount) != 0) {
 953                         i++;
 954                         CDEBUG(((i & (-i)) == i) ? D_WARNING : D_NET,
 955                        "Waiting for %d references to drop\n",
 956                        atomic_read(&net->net_refcount));
 957
 958                         cfs_pause(cfs_time_seconds(1));
 959                 }
 960
 961         LIBCFS_FREE(net, sizeof(*net));
 962 out:
 963         /* NB no locking since I don't race with writers */
 964         if (list_empty(&kptllnd_data.kptl_nets))
 965                 kptllnd_base_shutdown();
 966         CDEBUG(D_MALLOC, "after LND cleanup: kmem %d\n",
 967                atomic_read (&libcfs_kmemory));
 968         return;
 969 }
 970
 971 int __init
 972 kptllnd_module_init (void)
 973 {
 974         int    rc;
 975
 976         kptllnd_assert_wire_constants();
 977
 978         rc = kptllnd_tunables_init();
 979         if (rc != 0)
 980                 return rc;
 981
 982         kptllnd_init_ptltrace();
 983
 984         lnet_register_lnd(&kptllnd_lnd);
 985
 986         return 0;
 987 }
 988
 989 void __exit
 990 kptllnd_module_fini (void)
 991 {
 992         lnet_unregister_lnd(&kptllnd_lnd);
 993         kptllnd_tunables_fini();
 994 }
 995
 996 MODULE_AUTHOR("Sun Microsystems, Inc. <http://www.lustre.org/>");
 997 MODULE_DESCRIPTION("Kernel Portals LND v1.00");
 998 MODULE_LICENSE("GPL");
 999
1000 module_init(kptllnd_module_init);
1001 module_exit(kptllnd_module_fini);