lnet/lnet/api-ni.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2011, Whamcloud, Inc.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #define DEBUG_SUBSYSTEM S_LNET
  38 #include <lnet/lib-lnet.h>
  39
  40 #ifdef __KERNEL__
  41 #define D_LNI D_CONSOLE
  42 #else
  43 #define D_LNI D_CONFIG
  44 #endif
  45
  46 lnet_t      the_lnet;                           /* THE state of the network */
  47 EXPORT_SYMBOL(the_lnet);
  48
  49 #ifdef __KERNEL__
  50
  51 static char *ip2nets = "";
  52 CFS_MODULE_PARM(ip2nets, "s", charp, 0444,
  53                 "LNET network <- IP table");
  54
  55 static char *networks = "";
  56 CFS_MODULE_PARM(networks, "s", charp, 0444,
  57                 "local networks");
  58
  59 static char *routes = "";
  60 CFS_MODULE_PARM(routes, "s", charp, 0444,
  61                 "routes to non-local networks");
  62
  63 char *
  64 lnet_get_routes(void)
  65 {
  66         return routes;
  67 }
  68
  69 char *
  70 lnet_get_networks(void)
  71 {
  72         char   *nets;
  73         int     rc;
  74
  75         if (*networks != 0 && *ip2nets != 0) {
  76                 LCONSOLE_ERROR_MSG(0x101, "Please specify EITHER 'networks' or "
  77                                    "'ip2nets' but not both at once\n");
  78                 return NULL;
  79         }
  80
  81         if (*ip2nets != 0) {
  82                 rc = lnet_parse_ip2nets(&nets, ip2nets);
  83                 return (rc == 0) ? nets : NULL;
  84         }
  85
  86         if (*networks != 0)
  87                 return networks;
  88
  89         return "tcp";
  90 }
  91
  92 void
  93 lnet_init_locks(void)
  94 {
  95         cfs_spin_lock_init(&the_lnet.ln_eq_wait_lock);
  96         cfs_waitq_init(&the_lnet.ln_eq_waitq);
  97         cfs_mutex_init(&the_lnet.ln_lnd_mutex);
  98         cfs_mutex_init(&the_lnet.ln_api_mutex);
  99 }
 100
 101 void
 102 lnet_fini_locks(void)
 103 {
 104 }
 105
 106 #else
 107
 108 char *
 109 lnet_get_routes(void)
 110 {
 111         char *str = getenv("LNET_ROUTES");
 112
 113         return (str == NULL) ? "" : str;
 114 }
 115
 116 char *
 117 lnet_get_networks (void)
 118 {
 119         static char       default_networks[256];
 120         char             *networks = getenv ("LNET_NETWORKS");
 121         char             *ip2nets  = getenv ("LNET_IP2NETS");
 122         char             *str;
 123         char             *sep;
 124         int               len;
 125         int               nob;
 126         int               rc;
 127         cfs_list_t       *tmp;
 128
 129 #ifdef NOT_YET
 130         if (networks != NULL && ip2nets != NULL) {
 131                 LCONSOLE_ERROR_MSG(0x103, "Please set EITHER 'LNET_NETWORKS' or"
 132                                    " 'LNET_IP2NETS' but not both at once\n");
 133                 return NULL;
 134         }
 135
 136         if (ip2nets != NULL) {
 137                 rc = lnet_parse_ip2nets(&networks, ip2nets);
 138                 return (rc == 0) ? networks : NULL;
 139         }
 140 #else
 141         SET_BUT_UNUSED(ip2nets);
 142         SET_BUT_UNUSED(rc);
 143 #endif
 144         if (networks != NULL)
 145                 return networks;
 146
 147         /* In userland, the default 'networks=' is the list of known net types */
 148
 149         len = sizeof(default_networks);
 150         str = default_networks;
 151         *str = 0;
 152         sep = "";
 153
 154         cfs_list_for_each (tmp, &the_lnet.ln_lnds) {
 155                 lnd_t *lnd = cfs_list_entry(tmp, lnd_t, lnd_list);
 156
 157                 nob = snprintf(str, len, "%s%s", sep,
 158                                libcfs_lnd2str(lnd->lnd_type));
 159                 len -= nob;
 160                 if (len < 0) {
 161                         /* overflowed the string; leave it where it was */
 162                         *str = 0;
 163                         break;
 164                 }
 165
 166                 str += nob;
 167                 sep = ",";
 168         }
 169
 170         return default_networks;
 171 }
 172
 173 # ifndef HAVE_LIBPTHREAD
 174
 175 void lnet_init_locks(void)
 176 {
 177         the_lnet.ln_eq_wait_lock = 0;
 178         the_lnet.ln_lnd_mutex = 0;
 179         the_lnet.ln_api_mutex = 0;
 180 }
 181
 182 void lnet_fini_locks(void)
 183 {
 184         LASSERT(the_lnet.ln_api_mutex == 0);
 185         LASSERT(the_lnet.ln_lnd_mutex == 0);
 186         LASSERT(the_lnet.ln_eq_wait_lock == 0);
 187 }
 188
 189 # else
 190
 191 void lnet_init_locks(void)
 192 {
 193         pthread_cond_init(&the_lnet.ln_eq_cond, NULL);
 194         pthread_mutex_init(&the_lnet.ln_eq_wait_lock, NULL);
 195         pthread_mutex_init(&the_lnet.ln_lnd_mutex, NULL);
 196         pthread_mutex_init(&the_lnet.ln_api_mutex, NULL);
 197 }
 198
 199 void lnet_fini_locks(void)
 200 {
 201         pthread_mutex_destroy(&the_lnet.ln_api_mutex);
 202         pthread_mutex_destroy(&the_lnet.ln_lnd_mutex);
 203         pthread_mutex_destroy(&the_lnet.ln_eq_wait_lock);
 204         pthread_cond_destroy(&the_lnet.ln_eq_cond);
 205 }
 206
 207 # endif
 208 #endif
 209
 210 static int
 211 lnet_create_locks(void)
 212 {
 213         lnet_init_locks();
 214
 215         the_lnet.ln_res_lock = cfs_percpt_lock_alloc(lnet_cpt_table());
 216         if (the_lnet.ln_res_lock == NULL)
 217                 goto failed;
 218
 219         the_lnet.ln_net_lock = cfs_percpt_lock_alloc(lnet_cpt_table());
 220         if (the_lnet.ln_net_lock == NULL)
 221                 goto failed;
 222
 223         return 0;
 224
 225  failed:
 226         lnet_fini_locks();
 227         return -ENOMEM;
 228 }
 229
 230 static void
 231 lnet_destroy_locks(void)
 232 {
 233         if (the_lnet.ln_res_lock != NULL) {
 234                 cfs_percpt_lock_free(the_lnet.ln_res_lock);
 235                 the_lnet.ln_res_lock = NULL;
 236         }
 237
 238         if (the_lnet.ln_net_lock != NULL) {
 239                 cfs_percpt_lock_free(the_lnet.ln_net_lock);
 240                 the_lnet.ln_net_lock = NULL;
 241         }
 242
 243         lnet_fini_locks();
 244 }
 245
 246 void lnet_assert_wire_constants (void)
 247 {
 248         /* Wire protocol assertions generated by 'wirecheck'
 249          * running on Linux robert.bartonsoftware.com 2.6.8-1.521
 250          * #1 Mon Aug 16 09:01:18 EDT 2004 i686 athlon i386 GNU/Linux
 251          * with gcc version 3.3.3 20040412 (Red Hat Linux 3.3.3-7) */
 252
 253         /* Constants... */
 254         CLASSERT (LNET_PROTO_TCP_MAGIC == 0xeebc0ded);
 255         CLASSERT (LNET_PROTO_TCP_VERSION_MAJOR == 1);
 256         CLASSERT (LNET_PROTO_TCP_VERSION_MINOR == 0);
 257         CLASSERT (LNET_MSG_ACK == 0);
 258         CLASSERT (LNET_MSG_PUT == 1);
 259         CLASSERT (LNET_MSG_GET == 2);
 260         CLASSERT (LNET_MSG_REPLY == 3);
 261         CLASSERT (LNET_MSG_HELLO == 4);
 262
 263         /* Checks for struct ptl_handle_wire_t */
 264         CLASSERT ((int)sizeof(lnet_handle_wire_t) == 16);
 265         CLASSERT ((int)offsetof(lnet_handle_wire_t, wh_interface_cookie) == 0);
 266         CLASSERT ((int)sizeof(((lnet_handle_wire_t *)0)->wh_interface_cookie) == 8);
 267         CLASSERT ((int)offsetof(lnet_handle_wire_t, wh_object_cookie) == 8);
 268         CLASSERT ((int)sizeof(((lnet_handle_wire_t *)0)->wh_object_cookie) == 8);
 269
 270         /* Checks for struct lnet_magicversion_t */
 271         CLASSERT ((int)sizeof(lnet_magicversion_t) == 8);
 272         CLASSERT ((int)offsetof(lnet_magicversion_t, magic) == 0);
 273         CLASSERT ((int)sizeof(((lnet_magicversion_t *)0)->magic) == 4);
 274         CLASSERT ((int)offsetof(lnet_magicversion_t, version_major) == 4);
 275         CLASSERT ((int)sizeof(((lnet_magicversion_t *)0)->version_major) == 2);
 276         CLASSERT ((int)offsetof(lnet_magicversion_t, version_minor) == 6);
 277         CLASSERT ((int)sizeof(((lnet_magicversion_t *)0)->version_minor) == 2);
 278
 279         /* Checks for struct lnet_hdr_t */
 280         CLASSERT ((int)sizeof(lnet_hdr_t) == 72);
 281         CLASSERT ((int)offsetof(lnet_hdr_t, dest_nid) == 0);
 282         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->dest_nid) == 8);
 283         CLASSERT ((int)offsetof(lnet_hdr_t, src_nid) == 8);
 284         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->src_nid) == 8);
 285         CLASSERT ((int)offsetof(lnet_hdr_t, dest_pid) == 16);
 286         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->dest_pid) == 4);
 287         CLASSERT ((int)offsetof(lnet_hdr_t, src_pid) == 20);
 288         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->src_pid) == 4);
 289         CLASSERT ((int)offsetof(lnet_hdr_t, type) == 24);
 290         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->type) == 4);
 291         CLASSERT ((int)offsetof(lnet_hdr_t, payload_length) == 28);
 292         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->payload_length) == 4);
 293         CLASSERT ((int)offsetof(lnet_hdr_t, msg) == 32);
 294         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg) == 40);
 295
 296         /* Ack */
 297         CLASSERT ((int)offsetof(lnet_hdr_t, msg.ack.dst_wmd) == 32);
 298         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.ack.dst_wmd) == 16);
 299         CLASSERT ((int)offsetof(lnet_hdr_t, msg.ack.match_bits) == 48);
 300         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.ack.match_bits) == 8);
 301         CLASSERT ((int)offsetof(lnet_hdr_t, msg.ack.mlength) == 56);
 302         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.ack.mlength) == 4);
 303
 304         /* Put */
 305         CLASSERT ((int)offsetof(lnet_hdr_t, msg.put.ack_wmd) == 32);
 306         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.put.ack_wmd) == 16);
 307         CLASSERT ((int)offsetof(lnet_hdr_t, msg.put.match_bits) == 48);
 308         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.put.match_bits) == 8);
 309         CLASSERT ((int)offsetof(lnet_hdr_t, msg.put.hdr_data) == 56);
 310         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.put.hdr_data) == 8);
 311         CLASSERT ((int)offsetof(lnet_hdr_t, msg.put.ptl_index) == 64);
 312         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.put.ptl_index) == 4);
 313         CLASSERT ((int)offsetof(lnet_hdr_t, msg.put.offset) == 68);
 314         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.put.offset) == 4);
 315
 316         /* Get */
 317         CLASSERT ((int)offsetof(lnet_hdr_t, msg.get.return_wmd) == 32);
 318         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.get.return_wmd) == 16);
 319         CLASSERT ((int)offsetof(lnet_hdr_t, msg.get.match_bits) == 48);
 320         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.get.match_bits) == 8);
 321         CLASSERT ((int)offsetof(lnet_hdr_t, msg.get.ptl_index) == 56);
 322         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.get.ptl_index) == 4);
 323         CLASSERT ((int)offsetof(lnet_hdr_t, msg.get.src_offset) == 60);
 324         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.get.src_offset) == 4);
 325         CLASSERT ((int)offsetof(lnet_hdr_t, msg.get.sink_length) == 64);
 326         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.get.sink_length) == 4);
 327
 328         /* Reply */
 329         CLASSERT ((int)offsetof(lnet_hdr_t, msg.reply.dst_wmd) == 32);
 330         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.reply.dst_wmd) == 16);
 331
 332         /* Hello */
 333         CLASSERT ((int)offsetof(lnet_hdr_t, msg.hello.incarnation) == 32);
 334         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.hello.incarnation) == 8);
 335         CLASSERT ((int)offsetof(lnet_hdr_t, msg.hello.type) == 40);
 336         CLASSERT ((int)sizeof(((lnet_hdr_t *)0)->msg.hello.type) == 4);
 337 }
 338
 339 lnd_t *
 340 lnet_find_lnd_by_type (int type)
 341 {
 342         lnd_t              *lnd;
 343         cfs_list_t         *tmp;
 344
 345         /* holding lnd mutex */
 346         cfs_list_for_each (tmp, &the_lnet.ln_lnds) {
 347                 lnd = cfs_list_entry(tmp, lnd_t, lnd_list);
 348
 349                 if ((int)lnd->lnd_type == type)
 350                         return lnd;
 351         }
 352
 353         return NULL;
 354 }
 355
 356 void
 357 lnet_register_lnd (lnd_t *lnd)
 358 {
 359         LNET_MUTEX_LOCK(&the_lnet.ln_lnd_mutex);
 360
 361         LASSERT (the_lnet.ln_init);
 362         LASSERT (libcfs_isknown_lnd(lnd->lnd_type));
 363         LASSERT (lnet_find_lnd_by_type(lnd->lnd_type) == NULL);
 364
 365         cfs_list_add_tail (&lnd->lnd_list, &the_lnet.ln_lnds);
 366         lnd->lnd_refcount = 0;
 367
 368         CDEBUG(D_NET, "%s LND registered\n", libcfs_lnd2str(lnd->lnd_type));
 369
 370         LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
 371 }
 372 EXPORT_SYMBOL(lnet_register_lnd);
 373
 374 void
 375 lnet_unregister_lnd (lnd_t *lnd)
 376 {
 377         LNET_MUTEX_LOCK(&the_lnet.ln_lnd_mutex);
 378
 379         LASSERT (the_lnet.ln_init);
 380         LASSERT (lnet_find_lnd_by_type(lnd->lnd_type) == lnd);
 381         LASSERT (lnd->lnd_refcount == 0);
 382
 383         cfs_list_del (&lnd->lnd_list);
 384         CDEBUG(D_NET, "%s LND unregistered\n", libcfs_lnd2str(lnd->lnd_type));
 385
 386         LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
 387 }
 388 EXPORT_SYMBOL(lnet_unregister_lnd);
 389
 390 void
 391 lnet_counters_get(lnet_counters_t *counters)
 392 {
 393         lnet_counters_t *ctr;
 394         int             i;
 395
 396         memset(counters, 0, sizeof(*counters));
 397
 398         lnet_net_lock(LNET_LOCK_EX);
 399
 400         cfs_percpt_for_each(ctr, i, the_lnet.ln_counters) {
 401                 counters->msgs_max     += ctr->msgs_max;
 402                 counters->msgs_alloc   += ctr->msgs_alloc;
 403                 counters->errors       += ctr->errors;
 404                 counters->send_count   += ctr->send_count;
 405                 counters->recv_count   += ctr->recv_count;
 406                 counters->route_count  += ctr->route_count;
 407                 counters->drop_length  += ctr->drop_length;
 408                 counters->send_length  += ctr->send_length;
 409                 counters->recv_length  += ctr->recv_length;
 410                 counters->route_length += ctr->route_length;
 411                 counters->drop_length  += ctr->drop_length;
 412
 413         }
 414         lnet_net_unlock(LNET_LOCK_EX);
 415 }
 416 EXPORT_SYMBOL(lnet_counters_get);
 417
 418 void
 419 lnet_counters_reset(void)
 420 {
 421         lnet_counters_t *counters;
 422         int             i;
 423
 424         lnet_net_lock(LNET_LOCK_EX);
 425
 426         cfs_percpt_for_each(counters, i, the_lnet.ln_counters)
 427                 memset(counters, 0, sizeof(lnet_counters_t));
 428
 429         lnet_net_unlock(LNET_LOCK_EX);
 430 }
 431 EXPORT_SYMBOL(lnet_counters_reset);
 432
 433 #ifdef LNET_USE_LIB_FREELIST
 434
 435 int
 436 lnet_freelist_init (lnet_freelist_t *fl, int n, int size)
 437 {
 438         char *space;
 439
 440         LASSERT (n > 0);
 441
 442         size += offsetof (lnet_freeobj_t, fo_contents);
 443
 444         LIBCFS_ALLOC(space, n * size);
 445         if (space == NULL)
 446                 return (-ENOMEM);
 447
 448         CFS_INIT_LIST_HEAD (&fl->fl_list);
 449         fl->fl_objs = space;
 450         fl->fl_nobjs = n;
 451         fl->fl_objsize = size;
 452
 453         do
 454         {
 455                 memset (space, 0, size);
 456                 cfs_list_add ((cfs_list_t *)space, &fl->fl_list);
 457                 space += size;
 458         } while (--n != 0);
 459
 460         return (0);
 461 }
 462
 463 void
 464 lnet_freelist_fini (lnet_freelist_t *fl)
 465 {
 466         cfs_list_t       *el;
 467         int               count;
 468
 469         if (fl->fl_nobjs == 0)
 470                 return;
 471
 472         count = 0;
 473         for (el = fl->fl_list.next; el != &fl->fl_list; el = el->next)
 474                 count++;
 475
 476         LASSERT (count == fl->fl_nobjs);
 477
 478         LIBCFS_FREE(fl->fl_objs, fl->fl_nobjs * fl->fl_objsize);
 479         memset (fl, 0, sizeof (*fl));
 480 }
 481
 482 #endif /* LNET_USE_LIB_FREELIST */
 483
 484 __u64
 485 lnet_create_interface_cookie (void)
 486 {
 487         /* NB the interface cookie in wire handles guards against delayed
 488          * replies and ACKs appearing valid after reboot. Initialisation time,
 489          * even if it's only implemented to millisecond resolution is probably
 490          * easily good enough. */
 491         struct timeval tv;
 492         __u64          cookie;
 493 #ifndef __KERNEL__
 494         int            rc = gettimeofday (&tv, NULL);
 495         LASSERT (rc == 0);
 496 #else
 497         cfs_gettimeofday(&tv);
 498 #endif
 499         cookie = tv.tv_sec;
 500         cookie *= 1000000;
 501         cookie += tv.tv_usec;
 502         return cookie;
 503 }
 504
 505 static char *
 506 lnet_res_type2str(int type)
 507 {
 508         switch (type) {
 509         default:
 510                 LBUG();
 511         case LNET_COOKIE_TYPE_MD:
 512                 return "MD";
 513         case LNET_COOKIE_TYPE_ME:
 514                 return "ME";
 515         case LNET_COOKIE_TYPE_EQ:
 516                 return "EQ";
 517         }
 518 }
 519
 520 void
 521 lnet_res_container_cleanup(struct lnet_res_container *rec)
 522 {
 523         int     count = 0;
 524
 525         if (rec->rec_type == 0) /* not set yet, it's uninitialized */
 526                 return;
 527
 528         while (!cfs_list_empty(&rec->rec_active)) {
 529                 cfs_list_t *e = rec->rec_active.next;
 530
 531                 cfs_list_del_init(e);
 532                 if (rec->rec_type == LNET_COOKIE_TYPE_EQ) {
 533                         lnet_eq_free(cfs_list_entry(e, lnet_eq_t, eq_list));
 534
 535                 } else if (rec->rec_type == LNET_COOKIE_TYPE_MD) {
 536                         lnet_md_free(cfs_list_entry(e, lnet_libmd_t, md_list));
 537
 538                 } else { /* NB: Active MEs should be attached on portals */
 539                         LBUG();
 540                 }
 541                 count++;
 542         }
 543
 544         if (count > 0) {
 545                 /* Found alive MD/ME/EQ, user really should unlink/free
 546                  * all of them before finalize LNet, but if someone didn't,
 547                  * we have to recycle garbage for him */
 548                 CERROR("%d active elements on exit of %s container\n",
 549                        count, lnet_res_type2str(rec->rec_type));
 550         }
 551
 552 #ifdef LNET_USE_LIB_FREELIST
 553         lnet_freelist_fini(&rec->rec_freelist);
 554 #endif
 555         if (rec->rec_lh_hash != NULL) {
 556                 LIBCFS_FREE(rec->rec_lh_hash,
 557                             LNET_LH_HASH_SIZE * sizeof(rec->rec_lh_hash[0]));
 558                 rec->rec_lh_hash = NULL;
 559         }
 560
 561         rec->rec_type = 0; /* mark it as finalized */
 562 }
 563
 564 int
 565 lnet_res_container_setup(struct lnet_res_container *rec,
 566                          int cpt, int type, int objnum, int objsz)
 567 {
 568         int     rc = 0;
 569         int     i;
 570
 571         LASSERT(rec->rec_type == 0);
 572
 573         rec->rec_type = type;
 574         CFS_INIT_LIST_HEAD(&rec->rec_active);
 575
 576 #ifdef LNET_USE_LIB_FREELIST
 577         memset(&rec->rec_freelist, 0, sizeof(rec->rec_freelist));
 578         rc = lnet_freelist_init(&rec->rec_freelist, objnum, objsz);
 579         if (rc != 0)
 580                 goto out;
 581 #endif
 582         rec->rec_lh_cookie = (cpt << LNET_COOKIE_TYPE_BITS) | type;
 583
 584         /* Arbitrary choice of hash table size */
 585         LIBCFS_CPT_ALLOC(rec->rec_lh_hash, lnet_cpt_table(), cpt,
 586                          LNET_LH_HASH_SIZE * sizeof(rec->rec_lh_hash[0]));
 587         if (rec->rec_lh_hash == NULL) {
 588                 rc = -ENOMEM;
 589                 goto out;
 590         }
 591
 592         for (i = 0; i < LNET_LH_HASH_SIZE; i++)
 593                 CFS_INIT_LIST_HEAD(&rec->rec_lh_hash[i]);
 594
 595         return 0;
 596
 597 out:
 598         CERROR("Failed to setup %s resource container\n",
 599                lnet_res_type2str(type));
 600         lnet_res_container_cleanup(rec);
 601         return rc;
 602 }
 603
 604 static void
 605 lnet_res_containers_destroy(struct lnet_res_container **recs)
 606 {
 607         struct lnet_res_container       *rec;
 608         int                             i;
 609
 610         cfs_percpt_for_each(rec, i, recs)
 611                 lnet_res_container_cleanup(rec);
 612
 613         cfs_percpt_free(recs);
 614 }
 615
 616 static struct lnet_res_container **
 617 lnet_res_containers_create(int type, int objnum, int objsz)
 618 {
 619         struct lnet_res_container       **recs;
 620         struct lnet_res_container       *rec;
 621         int                             rc;
 622         int                             i;
 623
 624         recs = cfs_percpt_alloc(lnet_cpt_table(), sizeof(*rec));
 625         if (recs == NULL) {
 626                 CERROR("Failed to allocate %s resource containers\n",
 627                        lnet_res_type2str(type));
 628                 return NULL;
 629         }
 630
 631         cfs_percpt_for_each(rec, i, recs) {
 632                 rc = lnet_res_container_setup(rec, i, type, objnum, objsz);
 633                 if (rc != 0) {
 634                         lnet_res_containers_destroy(recs);
 635                         return NULL;
 636                 }
 637         }
 638
 639         return recs;
 640 }
 641
 642 lnet_libhandle_t *
 643 lnet_res_lh_lookup(struct lnet_res_container *rec, __u64 cookie)
 644 {
 645         /* ALWAYS called with lnet_res_lock held */
 646         cfs_list_t              *head;
 647         lnet_libhandle_t        *lh;
 648         unsigned int            hash;
 649
 650         if ((cookie & LNET_COOKIE_MASK) != rec->rec_type)
 651                 return NULL;
 652
 653         hash = cookie >> (LNET_COOKIE_TYPE_BITS + LNET_CPT_BITS);
 654         head = &rec->rec_lh_hash[hash & LNET_LH_HASH_MASK];
 655
 656         cfs_list_for_each_entry(lh, head, lh_hash_chain) {
 657                 if (lh->lh_cookie == cookie)
 658                         return lh;
 659         }
 660
 661         return NULL;
 662 }
 663
 664 void
 665 lnet_res_lh_initialize(struct lnet_res_container *rec, lnet_libhandle_t *lh)
 666 {
 667         /* ALWAYS called with lnet_res_lock held */
 668         unsigned int    ibits = LNET_COOKIE_TYPE_BITS + LNET_CPT_BITS;
 669         unsigned int    hash;
 670
 671         lh->lh_cookie = rec->rec_lh_cookie;
 672         rec->rec_lh_cookie += 1 << ibits;
 673
 674         hash = (lh->lh_cookie >> ibits) & LNET_LH_HASH_MASK;
 675
 676         cfs_list_add(&lh->lh_hash_chain, &rec->rec_lh_hash[hash]);
 677 }
 678
 679 #ifndef __KERNEL__
 680 /**
 681  * Reserved API - do not use.
 682  * Temporary workaround to allow uOSS and test programs force server
 683  * mode in userspace. See comments near ln_server_mode_flag in
 684  * lnet/lib-types.h */
 685
 686 void
 687 lnet_server_mode() {
 688         the_lnet.ln_server_mode_flag = 1;
 689 }
 690 #endif
 691
 692 int lnet_unprepare(void);
 693
 694 int
 695 lnet_prepare(lnet_pid_t requested_pid)
 696 {
 697         /* Prepare to bring up the network */
 698         struct lnet_res_container **recs;
 699         int                       rc = 0;
 700
 701         LASSERT (the_lnet.ln_refcount == 0);
 702
 703         the_lnet.ln_routing = 0;
 704
 705 #ifdef __KERNEL__
 706         LASSERT ((requested_pid & LNET_PID_USERFLAG) == 0);
 707         the_lnet.ln_pid = requested_pid;
 708 #else
 709         if (the_lnet.ln_server_mode_flag) {/* server case (uOSS) */
 710                 LASSERT ((requested_pid & LNET_PID_USERFLAG) == 0);
 711
 712                 if (cfs_curproc_uid())/* Only root can run user-space server */
 713                         return -EPERM;
 714                 the_lnet.ln_pid = requested_pid;
 715
 716         } else {/* client case (liblustre) */
 717
 718                 /* My PID must be unique on this node and flag I'm userspace */
 719                 the_lnet.ln_pid = getpid() | LNET_PID_USERFLAG;
 720         }
 721 #endif
 722
 723         CFS_INIT_LIST_HEAD(&the_lnet.ln_test_peers);
 724         CFS_INIT_LIST_HEAD(&the_lnet.ln_nis);
 725         CFS_INIT_LIST_HEAD(&the_lnet.ln_nis_cpt);
 726         CFS_INIT_LIST_HEAD(&the_lnet.ln_nis_zombie);
 727         CFS_INIT_LIST_HEAD(&the_lnet.ln_remote_nets);
 728         CFS_INIT_LIST_HEAD(&the_lnet.ln_routers);
 729
 730         the_lnet.ln_interface_cookie = lnet_create_interface_cookie();
 731
 732         the_lnet.ln_counters = cfs_percpt_alloc(lnet_cpt_table(),
 733                                                 sizeof(lnet_counters_t));
 734         if (the_lnet.ln_counters == NULL) {
 735                 CERROR("Failed to allocate counters for LNet\n");
 736                 rc = -ENOMEM;
 737                 goto failed;
 738         }
 739
 740         rc = lnet_peer_tables_create();
 741         if (rc != 0)
 742                 goto failed;
 743
 744         rc = lnet_msg_containers_create();
 745         if (rc != 0)
 746                 goto failed;
 747
 748         rc = lnet_res_container_setup(&the_lnet.ln_eq_container, 0,
 749                                       LNET_COOKIE_TYPE_EQ, LNET_FL_MAX_EQS,
 750                                       sizeof(lnet_eq_t));
 751         if (rc != 0)
 752                 goto failed;
 753
 754         recs = lnet_res_containers_create(LNET_COOKIE_TYPE_ME, LNET_FL_MAX_MES,
 755                                           sizeof(lnet_me_t));
 756         if (recs == NULL)
 757                 goto failed;
 758
 759         the_lnet.ln_me_containers = recs;
 760
 761         recs = lnet_res_containers_create(LNET_COOKIE_TYPE_MD, LNET_FL_MAX_MDS,
 762                                           sizeof(lnet_libmd_t));
 763         if (recs == NULL)
 764                 goto failed;
 765
 766         the_lnet.ln_md_containers = recs;
 767
 768         rc = lnet_portals_create();
 769         if (rc != 0) {
 770                 CERROR("Failed to create portals for LNet: %d\n", rc);
 771                 goto failed;
 772         }
 773
 774         return 0;
 775
 776  failed:
 777         lnet_unprepare();
 778         return rc;
 779 }
 780
 781 int
 782 lnet_unprepare (void)
 783 {
 784         /* NB no LNET_LOCK since this is the last reference.  All LND instances
 785          * have shut down already, so it is safe to unlink and free all
 786          * descriptors, even those that appear committed to a network op (eg MD
 787          * with non-zero pending count) */
 788
 789         lnet_fail_nid(LNET_NID_ANY, 0);
 790
 791         LASSERT(the_lnet.ln_refcount == 0);
 792         LASSERT(cfs_list_empty(&the_lnet.ln_test_peers));
 793         LASSERT(cfs_list_empty(&the_lnet.ln_nis));
 794         LASSERT(cfs_list_empty(&the_lnet.ln_nis_cpt));
 795         LASSERT(cfs_list_empty(&the_lnet.ln_nis_zombie));
 796
 797         lnet_portals_destroy();
 798
 799         if (the_lnet.ln_md_containers != NULL) {
 800                 lnet_res_containers_destroy(the_lnet.ln_md_containers);
 801                 the_lnet.ln_md_containers = NULL;
 802         }
 803
 804         if (the_lnet.ln_me_containers != NULL) {
 805                 lnet_res_containers_destroy(the_lnet.ln_me_containers);
 806                 the_lnet.ln_me_containers = NULL;
 807         }
 808
 809         lnet_res_container_cleanup(&the_lnet.ln_eq_container);
 810
 811         lnet_msg_containers_destroy();
 812         lnet_peer_tables_destroy();
 813         lnet_rtrpools_free();
 814
 815         if (the_lnet.ln_counters != NULL) {
 816                 cfs_percpt_free(the_lnet.ln_counters);
 817                 the_lnet.ln_counters = NULL;
 818         }
 819
 820         return 0;
 821 }
 822
 823 lnet_ni_t  *
 824 lnet_net2ni_locked(__u32 net, int cpt)
 825 {
 826         cfs_list_t      *tmp;
 827         lnet_ni_t       *ni;
 828
 829         LASSERT(cpt != LNET_LOCK_EX);
 830
 831         cfs_list_for_each(tmp, &the_lnet.ln_nis) {
 832                 ni = cfs_list_entry(tmp, lnet_ni_t, ni_list);
 833
 834                 if (LNET_NIDNET(ni->ni_nid) == net) {
 835                         lnet_ni_addref_locked(ni, cpt);
 836                         return ni;
 837                 }
 838         }
 839
 840         return NULL;
 841 }
 842
 843 lnet_ni_t *
 844 lnet_net2ni(__u32 net)
 845 {
 846         lnet_ni_t *ni;
 847
 848         lnet_net_lock(0);
 849         ni = lnet_net2ni_locked(net, 0);
 850         lnet_net_unlock(0);
 851
 852         return ni;
 853 }
 854 EXPORT_SYMBOL(lnet_net2ni);
 855
 856 static unsigned int
 857 lnet_nid_cpt_hash(lnet_nid_t nid, unsigned int number)
 858 {
 859         __u64           key = nid;
 860         unsigned int    val;
 861
 862         LASSERT(number >= 1 && number <= LNET_CPT_NUMBER);
 863
 864         if (number == 1)
 865                 return 0;
 866
 867         val = cfs_hash_long(key, LNET_CPT_BITS);
 868         /* NB: LNET_CP_NUMBER doesn't have to be PO2 */
 869         if (val < number)
 870                 return val;
 871
 872         return (unsigned int)(key + val + (val >> 1)) % number;
 873 }
 874
 875 int
 876 lnet_cpt_of_nid_locked(lnet_nid_t nid)
 877 {
 878         struct lnet_ni *ni;
 879
 880         /* must called with hold of lnet_net_lock */
 881         if (LNET_CPT_NUMBER == 1)
 882                 return 0; /* the only one */
 883
 884         /* take lnet_net_lock(any) would be OK */
 885         if (!cfs_list_empty(&the_lnet.ln_nis_cpt)) {
 886                 cfs_list_for_each_entry(ni, &the_lnet.ln_nis_cpt, ni_cptlist) {
 887                         if (LNET_NIDNET(ni->ni_nid) != LNET_NIDNET(nid))
 888                                 continue;
 889
 890                         LASSERT(ni->ni_cpts != NULL);
 891                         return ni->ni_cpts[lnet_nid_cpt_hash
 892                                            (nid, ni->ni_ncpts)];
 893                 }
 894         }
 895
 896         return lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
 897 }
 898
 899 int
 900 lnet_cpt_of_nid(lnet_nid_t nid)
 901 {
 902         int     cpt;
 903         int     cpt2;
 904
 905         if (LNET_CPT_NUMBER == 1)
 906                 return 0; /* the only one */
 907
 908         if (cfs_list_empty(&the_lnet.ln_nis_cpt))
 909                 return lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
 910
 911         cpt = lnet_net_lock_current();
 912         cpt2 = lnet_cpt_of_nid_locked(nid);
 913         lnet_net_unlock(cpt);
 914
 915         return cpt2;
 916 }
 917 EXPORT_SYMBOL(lnet_cpt_of_nid);
 918
 919 int
 920 lnet_islocalnet(__u32 net)
 921 {
 922         struct lnet_ni  *ni;
 923         int             cpt;
 924
 925         cpt = lnet_net_lock_current();
 926
 927         ni = lnet_net2ni_locked(net, cpt);
 928         if (ni != NULL)
 929                 lnet_ni_decref_locked(ni, cpt);
 930
 931         lnet_net_unlock(cpt);
 932
 933         return ni != NULL;
 934 }
 935
 936 lnet_ni_t  *
 937 lnet_nid2ni_locked(lnet_nid_t nid, int cpt)
 938 {
 939         struct lnet_ni  *ni;
 940         cfs_list_t      *tmp;
 941
 942         LASSERT(cpt != LNET_LOCK_EX);
 943
 944         cfs_list_for_each(tmp, &the_lnet.ln_nis) {
 945                 ni = cfs_list_entry(tmp, lnet_ni_t, ni_list);
 946
 947                 if (ni->ni_nid == nid) {
 948                         lnet_ni_addref_locked(ni, cpt);
 949                         return ni;
 950                 }
 951         }
 952
 953         return NULL;
 954 }
 955
 956 int
 957 lnet_islocalnid(lnet_nid_t nid)
 958 {
 959         struct lnet_ni  *ni;
 960         int             cpt;
 961
 962         cpt = lnet_net_lock_current();
 963         ni = lnet_nid2ni_locked(nid, cpt);
 964         if (ni != NULL)
 965                 lnet_ni_decref_locked(ni, cpt);
 966         lnet_net_unlock(cpt);
 967
 968         return ni != NULL;
 969 }
 970
 971 int
 972 lnet_count_acceptor_nis (void)
 973 {
 974         /* Return the # of NIs that need the acceptor. */
 975         int             count = 0;
 976 #if defined(__KERNEL__) || defined(HAVE_LIBPTHREAD)
 977         cfs_list_t      *tmp;
 978         struct lnet_ni  *ni;
 979         int             cpt;
 980
 981         cpt = lnet_net_lock_current();
 982         cfs_list_for_each(tmp, &the_lnet.ln_nis) {
 983                 ni = cfs_list_entry(tmp, lnet_ni_t, ni_list);
 984
 985                 if (ni->ni_lnd->lnd_accept != NULL)
 986                         count++;
 987         }
 988
 989         lnet_net_unlock(cpt);
 990
 991 #endif /* defined(__KERNEL__) || defined(HAVE_LIBPTHREAD) */
 992         return count;
 993 }
 994
 995 static int
 996 lnet_ni_tq_credits(lnet_ni_t *ni)
 997 {
 998         int     credits;
 999
1000         LASSERT(ni->ni_ncpts >= 1);
1001
1002         if (ni->ni_ncpts == 1)
1003                 return ni->ni_maxtxcredits;
1004
1005         credits = ni->ni_maxtxcredits / ni->ni_ncpts;
1006         credits = max(credits, 8 * ni->ni_peertxcredits);
1007         credits = min(credits, ni->ni_maxtxcredits);
1008
1009         return credits;
1010 }
1011
1012 void
1013 lnet_shutdown_lndnis (void)
1014 {
1015         int                i;
1016         int                islo;
1017         lnet_ni_t         *ni;
1018
1019         /* NB called holding the global mutex */
1020
1021         /* All quiet on the API front */
1022         LASSERT(!the_lnet.ln_shutdown);
1023         LASSERT(the_lnet.ln_refcount == 0);
1024         LASSERT(cfs_list_empty(&the_lnet.ln_nis_zombie));
1025         LASSERT(cfs_list_empty(&the_lnet.ln_remote_nets));
1026
1027         lnet_net_lock(LNET_LOCK_EX);
1028         the_lnet.ln_shutdown = 1;       /* flag shutdown */
1029
1030         /* Unlink NIs from the global table */
1031         while (!cfs_list_empty(&the_lnet.ln_nis)) {
1032                 ni = cfs_list_entry(the_lnet.ln_nis.next,
1033                                     lnet_ni_t, ni_list);
1034                 /* move it to zombie list and nobody can find it anymore */
1035                 cfs_list_move(&ni->ni_list, &the_lnet.ln_nis_zombie);
1036                 lnet_ni_decref_locked(ni, 0);   /* drop ln_nis' ref */
1037
1038                 if (!cfs_list_empty(&ni->ni_cptlist)) {
1039                         cfs_list_del_init(&ni->ni_cptlist);
1040                         lnet_ni_decref_locked(ni, 0);
1041                 }
1042         }
1043
1044         /* Drop the cached eqwait NI. */
1045         if (the_lnet.ln_eq_waitni != NULL) {
1046                 lnet_ni_decref_locked(the_lnet.ln_eq_waitni, 0);
1047                 the_lnet.ln_eq_waitni = NULL;
1048         }
1049
1050         /* Drop the cached loopback NI. */
1051         if (the_lnet.ln_loni != NULL) {
1052                 lnet_ni_decref_locked(the_lnet.ln_loni, 0);
1053                 the_lnet.ln_loni = NULL;
1054         }
1055
1056         lnet_net_unlock(LNET_LOCK_EX);
1057
1058         /* Clear lazy portals and drop delayed messages which hold refs
1059          * on their lnet_msg_t::msg_rxpeer */
1060         for (i = 0; i < the_lnet.ln_nportals; i++)
1061                 LNetClearLazyPortal(i);
1062
1063         /* Clear the peer table and wait for all peers to go (they hold refs on
1064          * their NIs) */
1065         lnet_peer_tables_cleanup();
1066
1067         lnet_net_lock(LNET_LOCK_EX);
1068         /* Now wait for the NI's I just nuked to show up on ln_zombie_nis
1069          * and shut them down in guaranteed thread context */
1070         i = 2;
1071         while (!cfs_list_empty(&the_lnet.ln_nis_zombie)) {
1072                 int     *ref;
1073                 int     j;
1074
1075                 ni = cfs_list_entry(the_lnet.ln_nis_zombie.next,
1076                                     lnet_ni_t, ni_list);
1077                 cfs_list_del_init(&ni->ni_list);
1078                 cfs_percpt_for_each(ref, j, ni->ni_refs) {
1079                         if (*ref == 0)
1080                                 continue;
1081                         /* still busy, add it back to zombie list */
1082                         cfs_list_add(&ni->ni_list, &the_lnet.ln_nis_zombie);
1083                         break;
1084                 }
1085
1086                 while (!cfs_list_empty(&ni->ni_list)) {
1087                         lnet_net_unlock(LNET_LOCK_EX);
1088                         ++i;
1089                         if ((i & (-i)) == i) {
1090                                 CDEBUG(D_WARNING,
1091                                        "Waiting for zombie LNI %s\n",
1092                                        libcfs_nid2str(ni->ni_nid));
1093                         }
1094                         cfs_pause(cfs_time_seconds(1));
1095                         lnet_net_lock(LNET_LOCK_EX);
1096                         continue;
1097                 }
1098
1099                 ni->ni_lnd->lnd_refcount--;
1100                 lnet_net_unlock(LNET_LOCK_EX);
1101
1102                 islo = ni->ni_lnd->lnd_type == LOLND;
1103
1104                 LASSERT (!cfs_in_interrupt ());
1105                 (ni->ni_lnd->lnd_shutdown)(ni);
1106
1107                 /* can't deref lnd anymore now; it might have unregistered
1108                  * itself...  */
1109
1110                 if (!islo)
1111                         CDEBUG(D_LNI, "Removed LNI %s\n",
1112                                libcfs_nid2str(ni->ni_nid));
1113
1114                 lnet_ni_free(ni);
1115                 lnet_net_lock(LNET_LOCK_EX);
1116         }
1117
1118         the_lnet.ln_shutdown = 0;
1119         lnet_net_unlock(LNET_LOCK_EX);
1120
1121         if (the_lnet.ln_network_tokens != NULL) {
1122                 LIBCFS_FREE(the_lnet.ln_network_tokens,
1123                             the_lnet.ln_network_tokens_nob);
1124                 the_lnet.ln_network_tokens = NULL;
1125         }
1126 }
1127
1128 int
1129 lnet_startup_lndnis (void)
1130 {
1131         lnd_t                   *lnd;
1132         struct lnet_ni          *ni;
1133         struct lnet_tx_queue    *tq;
1134         cfs_list_t              nilist;
1135         int                     i;
1136         int                rc = 0;
1137         int                lnd_type;
1138         int                nicount = 0;
1139         char              *nets = lnet_get_networks();
1140
1141         CFS_INIT_LIST_HEAD(&nilist);
1142
1143         if (nets == NULL)
1144                 goto failed;
1145
1146         rc = lnet_parse_networks(&nilist, nets);
1147         if (rc != 0)
1148                 goto failed;
1149
1150         while (!cfs_list_empty(&nilist)) {
1151                 ni = cfs_list_entry(nilist.next, lnet_ni_t, ni_list);
1152                 lnd_type = LNET_NETTYP(LNET_NIDNET(ni->ni_nid));
1153
1154                 LASSERT (libcfs_isknown_lnd(lnd_type));
1155
1156                 if (lnd_type == CIBLND    ||
1157                     lnd_type == OPENIBLND ||
1158                     lnd_type == IIBLND    ||
1159                     lnd_type == VIBLND) {
1160                         CERROR("LND %s obsoleted\n",
1161                                libcfs_lnd2str(lnd_type));
1162                         goto failed;
1163                 }
1164
1165                 LNET_MUTEX_LOCK(&the_lnet.ln_lnd_mutex);
1166                 lnd = lnet_find_lnd_by_type(lnd_type);
1167
1168 #ifdef __KERNEL__
1169                 if (lnd == NULL) {
1170                         LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
1171                         rc = cfs_request_module("%s",
1172                                                 libcfs_lnd2modname(lnd_type));
1173                         LNET_MUTEX_LOCK(&the_lnet.ln_lnd_mutex);
1174
1175                         lnd = lnet_find_lnd_by_type(lnd_type);
1176                         if (lnd == NULL) {
1177                                 LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
1178                                 CERROR("Can't load LND %s, module %s, rc=%d\n",
1179                                        libcfs_lnd2str(lnd_type),
1180                                        libcfs_lnd2modname(lnd_type), rc);
1181 #ifndef HAVE_MODULE_LOADING_SUPPORT
1182                                 LCONSOLE_ERROR_MSG(0x104, "Your kernel must be "
1183                                          "compiled with kernel module "
1184                                          "loading support.");
1185 #endif
1186                                 goto failed;
1187                         }
1188                 }
1189 #else
1190                 if (lnd == NULL) {
1191                         LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
1192                         CERROR("LND %s not supported\n",
1193                                libcfs_lnd2str(lnd_type));
1194                         goto failed;
1195                 }
1196 #endif
1197
1198                 lnet_net_lock(LNET_LOCK_EX);
1199                 lnd->lnd_refcount++;
1200                 lnet_net_unlock(LNET_LOCK_EX);
1201
1202                 ni->ni_lnd = lnd;
1203
1204                 rc = (lnd->lnd_startup)(ni);
1205
1206                 LNET_MUTEX_UNLOCK(&the_lnet.ln_lnd_mutex);
1207
1208                 if (rc != 0) {
1209                         LCONSOLE_ERROR_MSG(0x105, "Error %d starting up LNI %s"
1210                                            "\n",
1211                                            rc, libcfs_lnd2str(lnd->lnd_type));
1212                         lnet_net_lock(LNET_LOCK_EX);
1213                         lnd->lnd_refcount--;
1214                         lnet_net_unlock(LNET_LOCK_EX);
1215                         goto failed;
1216                 }
1217
1218                 LASSERT (ni->ni_peertimeout <= 0 || lnd->lnd_query != NULL);
1219
1220                 cfs_list_del(&ni->ni_list);
1221
1222                 lnet_net_lock(LNET_LOCK_EX);
1223                 /* refcount for ln_nis */
1224                 lnet_ni_addref_locked(ni, 0);
1225                 cfs_list_add_tail(&ni->ni_list, &the_lnet.ln_nis);
1226                 if (ni->ni_cpts != NULL) {
1227                         cfs_list_add_tail(&ni->ni_cptlist,
1228                                           &the_lnet.ln_nis_cpt);
1229                         lnet_ni_addref_locked(ni, 0);
1230                 }
1231
1232                 lnet_net_unlock(LNET_LOCK_EX);
1233
1234                 if (lnd->lnd_type == LOLND) {
1235                         lnet_ni_addref(ni);
1236                         LASSERT (the_lnet.ln_loni == NULL);
1237                         the_lnet.ln_loni = ni;
1238                         continue;
1239                 }
1240
1241 #ifndef __KERNEL__
1242                 if (lnd->lnd_wait != NULL) {
1243                         if (the_lnet.ln_eq_waitni == NULL) {
1244                                 lnet_ni_addref(ni);
1245                                 the_lnet.ln_eq_waitni = ni;
1246                         }
1247                 } else {
1248 # ifndef HAVE_LIBPTHREAD
1249                         LCONSOLE_ERROR_MSG(0x106, "LND %s not supported in a "
1250                                            "single-threaded runtime\n",
1251                                            libcfs_lnd2str(lnd_type));
1252                         goto failed;
1253 # endif
1254                 }
1255 #endif
1256                 if (ni->ni_peertxcredits == 0 ||
1257                     ni->ni_maxtxcredits == 0) {
1258                         LCONSOLE_ERROR_MSG(0x107, "LNI %s has no %scredits\n",
1259                                            libcfs_lnd2str(lnd->lnd_type),
1260                                            ni->ni_peertxcredits == 0 ?
1261                                            "" : "per-peer ");
1262                         goto failed;
1263                 }
1264
1265                 cfs_percpt_for_each(tq, i, ni->ni_tx_queues) {
1266                         tq->tq_credits_min =
1267                         tq->tq_credits_max =
1268                         tq->tq_credits = lnet_ni_tq_credits(ni);
1269                 }
1270
1271                 CDEBUG(D_LNI, "Added LNI %s [%d/%d/%d/%d]\n",
1272                        libcfs_nid2str(ni->ni_nid), ni->ni_peertxcredits,
1273                        lnet_ni_tq_credits(ni) * LNET_CPT_NUMBER,
1274                        ni->ni_peerrtrcredits, ni->ni_peertimeout);
1275
1276                 nicount++;
1277         }
1278
1279         if (the_lnet.ln_eq_waitni != NULL && nicount > 1) {
1280                 lnd_type = the_lnet.ln_eq_waitni->ni_lnd->lnd_type;
1281                 LCONSOLE_ERROR_MSG(0x109, "LND %s can only run single-network"
1282                                    "\n",
1283                                    libcfs_lnd2str(lnd_type));
1284                 goto failed;
1285         }
1286
1287         return 0;
1288
1289  failed:
1290         lnet_shutdown_lndnis();
1291
1292         while (!cfs_list_empty(&nilist)) {
1293                 ni = cfs_list_entry(nilist.next, lnet_ni_t, ni_list);
1294                 cfs_list_del(&ni->ni_list);
1295                 lnet_ni_free(ni);
1296         }
1297
1298         return -ENETDOWN;
1299 }
1300
1301 /**
1302  * Initialize LNet library.
1303  *
1304  * Only userspace program needs to call this function - it's automatically
1305  * called in the kernel at module loading time. Caller has to call LNetFini()
1306  * after a call to LNetInit(), if and only if the latter returned 0. It must
1307  * be called exactly once.
1308  *
1309  * \return 0 on success, and -ve on failures.
1310  */
1311 int
1312 LNetInit(void)
1313 {
1314         int     rc;
1315
1316         lnet_assert_wire_constants();
1317         LASSERT(!the_lnet.ln_init);
1318
1319         memset(&the_lnet, 0, sizeof(the_lnet));
1320
1321         /* refer to global cfs_cpt_table for now */
1322         the_lnet.ln_cpt_table   = cfs_cpt_table;
1323         the_lnet.ln_cpt_number  = cfs_cpt_number(cfs_cpt_table);
1324
1325         LASSERT(the_lnet.ln_cpt_number > 0);
1326         if (the_lnet.ln_cpt_number > LNET_CPT_MAX) {
1327                 /* we are under risk of consuming all lh_cookie */
1328                 CERROR("Can't have %d CPTs for LNet (max allowed is %d), "
1329                        "please change setting of CPT-table and retry\n",
1330                        the_lnet.ln_cpt_number, LNET_CPT_MAX);
1331                 return -1;
1332         }
1333
1334         while ((1 << the_lnet.ln_cpt_bits) < the_lnet.ln_cpt_number)
1335                 the_lnet.ln_cpt_bits++;
1336
1337         rc = lnet_create_locks();
1338         if (rc != 0) {
1339                 CERROR("Can't create LNet global locks: %d\n", rc);
1340                 return -1;
1341         }
1342
1343         the_lnet.ln_refcount = 0;
1344         the_lnet.ln_init = 1;
1345         LNetInvalidateHandle(&the_lnet.ln_rc_eqh);
1346         CFS_INIT_LIST_HEAD(&the_lnet.ln_lnds);
1347         CFS_INIT_LIST_HEAD(&the_lnet.ln_rcd_zombie);
1348         CFS_INIT_LIST_HEAD(&the_lnet.ln_rcd_deathrow);
1349
1350 #ifdef __KERNEL__
1351         /* All LNDs apart from the LOLND are in separate modules.  They
1352          * register themselves when their module loads, and unregister
1353          * themselves when their module is unloaded. */
1354 #else
1355         /* Register LNDs
1356          * NB the order here determines default 'networks=' order */
1357 # ifdef HAVE_LIBPTHREAD
1358         LNET_REGISTER_ULND(the_tcplnd);
1359 # endif
1360 #endif
1361         lnet_register_lnd(&the_lolnd);
1362         return 0;
1363 }
1364 EXPORT_SYMBOL(LNetInit);
1365
1366 /**
1367  * Finalize LNet library.
1368  *
1369  * Only userspace program needs to call this function. It can be called
1370  * at most once.
1371  *
1372  * \pre LNetInit() called with success.
1373  * \pre All LNet users called LNetNIFini() for matching LNetNIInit() calls.
1374  */
1375 void
1376 LNetFini(void)
1377 {
1378         LASSERT(the_lnet.ln_init);
1379         LASSERT(the_lnet.ln_refcount == 0);
1380
1381         while (!cfs_list_empty(&the_lnet.ln_lnds))
1382                 lnet_unregister_lnd(cfs_list_entry(the_lnet.ln_lnds.next,
1383                                                    lnd_t, lnd_list));
1384         lnet_destroy_locks();
1385
1386         the_lnet.ln_init = 0;
1387 }
1388 EXPORT_SYMBOL(LNetFini);
1389
1390 /**
1391  * Set LNet PID and start LNet interfaces, routing, and forwarding.
1392  *
1393  * Userspace program should call this after a successful call to LNetInit().
1394  * Users must call this function at least once before any other functions.
1395  * For each successful call there must be a corresponding call to
1396  * LNetNIFini(). For subsequent calls to LNetNIInit(), \a requested_pid is
1397  * ignored.
1398  *
1399  * The PID used by LNet may be different from the one requested.
1400  * See LNetGetId().
1401  *
1402  * \param requested_pid PID requested by the caller.
1403  *
1404  * \return >= 0 on success, and < 0 error code on failures.
1405  */
1406 int
1407 LNetNIInit(lnet_pid_t requested_pid)
1408 {
1409         int         im_a_router = 0;
1410         int         rc;
1411
1412         LNET_MUTEX_LOCK(&the_lnet.ln_api_mutex);
1413
1414         LASSERT (the_lnet.ln_init);
1415         CDEBUG(D_OTHER, "refs %d\n", the_lnet.ln_refcount);
1416
1417         if (the_lnet.ln_refcount > 0) {
1418                 rc = the_lnet.ln_refcount++;
1419                 goto out;
1420         }
1421
1422         lnet_get_tunables();
1423
1424         if (requested_pid == LNET_PID_ANY) {
1425                 /* Don't instantiate LNET just for me */
1426                 rc = -ENETDOWN;
1427                 goto failed0;
1428         }
1429
1430         rc = lnet_prepare(requested_pid);
1431         if (rc != 0)
1432                 goto failed0;
1433
1434         rc = lnet_startup_lndnis();
1435         if (rc != 0)
1436                 goto failed1;
1437
1438         rc = lnet_parse_routes(lnet_get_routes(), &im_a_router);
1439         if (rc != 0)
1440                 goto failed2;
1441
1442         rc = lnet_check_routes();
1443         if (rc != 0)
1444                 goto failed2;
1445
1446         rc = lnet_rtrpools_alloc(im_a_router);
1447         if (rc != 0)
1448                 goto failed2;
1449
1450         rc = lnet_acceptor_start();
1451         if (rc != 0)
1452                 goto failed2;
1453
1454         the_lnet.ln_refcount = 1;
1455         /* Now I may use my own API functions... */
1456
1457         /* NB router checker needs the_lnet.ln_ping_info in
1458          * lnet_router_checker -> lnet_update_ni_status_locked */
1459         rc = lnet_ping_target_init();
1460         if (rc != 0)
1461                 goto failed3;
1462
1463         rc = lnet_router_checker_start();
1464         if (rc != 0)
1465                 goto failed4;
1466
1467         lnet_proc_init();
1468         goto out;
1469
1470  failed4:
1471         lnet_ping_target_fini();
1472  failed3:
1473         the_lnet.ln_refcount = 0;
1474         lnet_acceptor_stop();
1475  failed2:
1476         lnet_destroy_routes();
1477         lnet_shutdown_lndnis();
1478  failed1:
1479         lnet_unprepare();
1480  failed0:
1481         LASSERT (rc < 0);
1482  out:
1483         LNET_MUTEX_UNLOCK(&the_lnet.ln_api_mutex);
1484         return rc;
1485 }
1486 EXPORT_SYMBOL(LNetNIInit);
1487
1488 /**
1489  * Stop LNet interfaces, routing, and forwarding.
1490  *
1491  * Users must call this function once for each successful call to LNetNIInit().
1492  * Once the LNetNIFini() operation has been started, the results of pending
1493  * API operations are undefined.
1494  *
1495  * \return always 0 for current implementation.
1496  */
1497 int
1498 LNetNIFini()
1499 {
1500         LNET_MUTEX_LOCK(&the_lnet.ln_api_mutex);
1501
1502         LASSERT (the_lnet.ln_init);
1503         LASSERT (the_lnet.ln_refcount > 0);
1504
1505         if (the_lnet.ln_refcount != 1) {
1506                 the_lnet.ln_refcount--;
1507         } else {
1508                 LASSERT (!the_lnet.ln_niinit_self);
1509
1510                 lnet_proc_fini();
1511                 lnet_router_checker_stop();
1512                 lnet_ping_target_fini();
1513
1514                 /* Teardown fns that use my own API functions BEFORE here */
1515                 the_lnet.ln_refcount = 0;
1516
1517                 lnet_acceptor_stop();
1518                 lnet_destroy_routes();
1519                 lnet_shutdown_lndnis();
1520                 lnet_unprepare();
1521         }
1522
1523         LNET_MUTEX_UNLOCK(&the_lnet.ln_api_mutex);
1524         return 0;
1525 }
1526 EXPORT_SYMBOL(LNetNIFini);
1527
1528 /**
1529  * This is an ugly hack to export IOC_LIBCFS_DEBUG_PEER and
1530  * IOC_LIBCFS_PORTALS_COMPATIBILITY commands to users, by tweaking the LNet
1531  * internal ioctl handler.
1532  *
1533  * IOC_LIBCFS_PORTALS_COMPATIBILITY is now deprecated, don't use it.
1534  *
1535  * \param cmd IOC_LIBCFS_DEBUG_PEER to print debugging data about a peer.
1536  * The data will be printed to system console. Don't use it excessively.
1537  * \param arg A pointer to lnet_process_id_t, process ID of the peer.
1538  *
1539  * \return Always return 0 when called by users directly (i.e., not via ioctl).
1540  */
1541 int
1542 LNetCtl(unsigned int cmd, void *arg)
1543 {
1544         struct libcfs_ioctl_data *data = arg;
1545         lnet_process_id_t         id = {0};
1546         lnet_ni_t                *ni;
1547         int                       rc;
1548
1549         LASSERT (the_lnet.ln_init);
1550         LASSERT (the_lnet.ln_refcount > 0);
1551
1552         switch (cmd) {
1553         case IOC_LIBCFS_GET_NI:
1554                 rc = LNetGetId(data->ioc_count, &id);
1555                 data->ioc_nid = id.nid;
1556                 return rc;
1557
1558         case IOC_LIBCFS_FAIL_NID:
1559                 return lnet_fail_nid(data->ioc_nid, data->ioc_count);
1560
1561         case IOC_LIBCFS_ADD_ROUTE:
1562                 rc = lnet_add_route(data->ioc_net, data->ioc_count,
1563                                     data->ioc_nid);
1564                 return (rc != 0) ? rc : lnet_check_routes();
1565
1566         case IOC_LIBCFS_DEL_ROUTE:
1567                 return lnet_del_route(data->ioc_net, data->ioc_nid);
1568
1569         case IOC_LIBCFS_GET_ROUTE:
1570                 return lnet_get_route(data->ioc_count,
1571                                       &data->ioc_net, &data->ioc_count,
1572                                       &data->ioc_nid, &data->ioc_flags);
1573         case IOC_LIBCFS_NOTIFY_ROUTER:
1574                 return lnet_notify(NULL, data->ioc_nid, data->ioc_flags,
1575                                    cfs_time_current() -
1576                                    cfs_time_seconds(cfs_time_current_sec() -
1577                                                     (time_t)data->ioc_u64[0]));
1578
1579         case IOC_LIBCFS_PORTALS_COMPATIBILITY:
1580                 /* This can be removed once lustre stops calling it */
1581                 return 0;
1582
1583         case IOC_LIBCFS_LNET_DIST:
1584                 rc = LNetDist(data->ioc_nid, &data->ioc_nid, &data->ioc_u32[1]);
1585                 if (rc < 0 && rc != -EHOSTUNREACH)
1586                         return rc;
1587
1588                 data->ioc_u32[0] = rc;
1589                 return 0;
1590
1591         case IOC_LIBCFS_TESTPROTOCOMPAT:
1592                 lnet_net_lock(LNET_LOCK_EX);
1593                 the_lnet.ln_testprotocompat = data->ioc_flags;
1594                 lnet_net_unlock(LNET_LOCK_EX);
1595                 return 0;
1596
1597         case IOC_LIBCFS_PING:
1598                 id.nid = data->ioc_nid;
1599                 id.pid = data->ioc_u32[0];
1600                 rc = lnet_ping(id, data->ioc_u32[1], /* timeout */
1601                                (lnet_process_id_t *)data->ioc_pbuf1,
1602                                data->ioc_plen1/sizeof(lnet_process_id_t));
1603                 if (rc < 0)
1604                         return rc;
1605                 data->ioc_count = rc;
1606                 return 0;
1607
1608         case IOC_LIBCFS_DEBUG_PEER: {
1609                 /* CAVEAT EMPTOR: this one designed for calling directly; not
1610                  * via an ioctl */
1611                 id = *((lnet_process_id_t *) arg);
1612
1613                 lnet_debug_peer(id.nid);
1614
1615                 ni = lnet_net2ni(LNET_NIDNET(id.nid));
1616                 if (ni == NULL) {
1617                         CDEBUG(D_WARNING, "No NI for %s\n", libcfs_id2str(id));
1618                 } else {
1619                         if (ni->ni_lnd->lnd_ctl == NULL) {
1620                                 CDEBUG(D_WARNING, "No ctl for %s\n",
1621                                        libcfs_id2str(id));
1622                         } else {
1623                                 (void)ni->ni_lnd->lnd_ctl(ni, cmd, arg);
1624                         }
1625
1626                         lnet_ni_decref(ni);
1627                 }
1628                 return 0;
1629         }
1630
1631         default:
1632                 ni = lnet_net2ni(data->ioc_net);
1633                 if (ni == NULL)
1634                         return -EINVAL;
1635
1636                 if (ni->ni_lnd->lnd_ctl == NULL)
1637                         rc = -EINVAL;
1638                 else
1639                         rc = ni->ni_lnd->lnd_ctl(ni, cmd, arg);
1640
1641                 lnet_ni_decref(ni);
1642                 return rc;
1643         }
1644         /* not reached */
1645 }
1646 EXPORT_SYMBOL(LNetCtl);
1647
1648 /**
1649  * Retrieve the lnet_process_id_t ID of LNet interface at \a index. Note that
1650  * all interfaces share a same PID, as requested by LNetNIInit().
1651  *
1652  * \param index Index of the interface to look up.
1653  * \param id On successful return, this location will hold the
1654  * lnet_process_id_t ID of the interface.
1655  *
1656  * \retval 0 If an interface exists at \a index.
1657  * \retval -ENOENT If no interface has been found.
1658  */
1659 int
1660 LNetGetId(unsigned int index, lnet_process_id_t *id)
1661 {
1662         struct lnet_ni  *ni;
1663         cfs_list_t      *tmp;
1664         int             cpt;
1665         int             rc = -ENOENT;
1666
1667         LASSERT(the_lnet.ln_init);
1668         LASSERT(the_lnet.ln_refcount > 0);
1669
1670         cpt = lnet_net_lock_current();
1671
1672         cfs_list_for_each(tmp, &the_lnet.ln_nis) {
1673                 if (index-- != 0)
1674                         continue;
1675
1676                 ni = cfs_list_entry(tmp, lnet_ni_t, ni_list);
1677
1678                 id->nid = ni->ni_nid;
1679                 id->pid = the_lnet.ln_pid;
1680                 rc = 0;
1681                 break;
1682         }
1683
1684         lnet_net_unlock(cpt);
1685         return rc;
1686 }
1687 EXPORT_SYMBOL(LNetGetId);
1688
1689 /**
1690  * Print a string representation of handle \a h into buffer \a str of
1691  * \a len bytes.
1692  */
1693 void
1694 LNetSnprintHandle(char *str, int len, lnet_handle_any_t h)
1695 {
1696         snprintf(str, len, LPX64, h.cookie);
1697 }
1698 EXPORT_SYMBOL(LNetSnprintHandle);
1699
1700 static int
1701 lnet_create_ping_info(void)
1702 {
1703         int               i;
1704         int               n;
1705         int               rc;
1706         unsigned int      infosz;
1707         lnet_ni_t        *ni;
1708         lnet_process_id_t id;
1709         lnet_ping_info_t *pinfo;
1710
1711         for (n = 0; ; n++) {
1712                 rc = LNetGetId(n, &id);
1713                 if (rc == -ENOENT)
1714                         break;
1715
1716                 LASSERT (rc == 0);
1717         }
1718
1719         infosz = offsetof(lnet_ping_info_t, pi_ni[n]);
1720         LIBCFS_ALLOC(pinfo, infosz);
1721         if (pinfo == NULL) {
1722                 CERROR("Can't allocate ping info[%d]\n", n);
1723                 return -ENOMEM;
1724         }
1725
1726         pinfo->pi_nnis    = n;
1727         pinfo->pi_pid     = the_lnet.ln_pid;
1728         pinfo->pi_magic   = LNET_PROTO_PING_MAGIC;
1729         pinfo->pi_features = LNET_PING_FEAT_NI_STATUS;
1730
1731         for (i = 0; i < n; i++) {
1732                 lnet_ni_status_t *ns = &pinfo->pi_ni[i];
1733
1734                 rc = LNetGetId(i, &id);
1735                 LASSERT (rc == 0);
1736
1737                 ns->ns_nid    = id.nid;
1738                 ns->ns_status = LNET_NI_STATUS_UP;
1739
1740                 lnet_net_lock(0);
1741
1742                 ni = lnet_nid2ni_locked(id.nid, 0);
1743                 LASSERT(ni != NULL);
1744
1745                 lnet_ni_lock(ni);
1746                 LASSERT(ni->ni_status == NULL);
1747                 ni->ni_status = ns;
1748                 lnet_ni_unlock(ni);
1749
1750                 lnet_ni_decref_locked(ni, 0);
1751                 lnet_net_unlock(0);
1752         }
1753
1754         the_lnet.ln_ping_info = pinfo;
1755         return 0;
1756 }
1757
1758 static void
1759 lnet_destroy_ping_info(void)
1760 {
1761         struct lnet_ni  *ni;
1762
1763         lnet_net_lock(0);
1764
1765         cfs_list_for_each_entry(ni, &the_lnet.ln_nis, ni_list) {
1766                 lnet_ni_lock(ni);
1767                 ni->ni_status = NULL;
1768                 lnet_ni_unlock(ni);
1769         }
1770
1771         lnet_net_unlock(0);
1772
1773         LIBCFS_FREE(the_lnet.ln_ping_info,
1774                     offsetof(lnet_ping_info_t,
1775                              pi_ni[the_lnet.ln_ping_info->pi_nnis]));
1776         the_lnet.ln_ping_info = NULL;
1777         return;
1778 }
1779
1780 int
1781 lnet_ping_target_init(void)
1782 {
1783         lnet_md_t         md = {0};
1784         lnet_handle_me_t  meh;
1785         lnet_process_id_t id;
1786         int               rc;
1787         int               rc2;
1788         int               infosz;
1789
1790         rc = lnet_create_ping_info();
1791         if (rc != 0)
1792                 return rc;
1793
1794         /* We can have a tiny EQ since we only need to see the unlink event on
1795          * teardown, which by definition is the last one! */
1796         rc = LNetEQAlloc(2, LNET_EQ_HANDLER_NONE, &the_lnet.ln_ping_target_eq);
1797         if (rc != 0) {
1798                 CERROR("Can't allocate ping EQ: %d\n", rc);
1799                 goto failed_0;
1800         }
1801
1802         memset(&id, 0, sizeof(lnet_process_id_t));
1803         id.nid = LNET_NID_ANY;
1804         id.pid = LNET_PID_ANY;
1805
1806         rc = LNetMEAttach(LNET_RESERVED_PORTAL, id,
1807                           LNET_PROTO_PING_MATCHBITS, 0,
1808                           LNET_UNLINK, LNET_INS_AFTER,
1809                           &meh);
1810         if (rc != 0) {
1811                 CERROR("Can't create ping ME: %d\n", rc);
1812                 goto failed_1;
1813         }
1814
1815         /* initialize md content */
1816         infosz = offsetof(lnet_ping_info_t,
1817                           pi_ni[the_lnet.ln_ping_info->pi_nnis]);
1818         md.start     = the_lnet.ln_ping_info;
1819         md.length    = infosz;
1820         md.threshold = LNET_MD_THRESH_INF;
1821         md.max_size  = 0;
1822         md.options   = LNET_MD_OP_GET | LNET_MD_TRUNCATE |
1823                        LNET_MD_MANAGE_REMOTE;
1824         md.user_ptr  = NULL;
1825         md.eq_handle = the_lnet.ln_ping_target_eq;
1826
1827         rc = LNetMDAttach(meh, md,
1828                           LNET_RETAIN,
1829                           &the_lnet.ln_ping_target_md);
1830         if (rc != 0) {
1831                 CERROR("Can't attach ping MD: %d\n", rc);
1832                 goto failed_2;
1833         }
1834
1835         return 0;
1836
1837  failed_2:
1838         rc2 = LNetMEUnlink(meh);
1839         LASSERT (rc2 == 0);
1840  failed_1:
1841         rc2 = LNetEQFree(the_lnet.ln_ping_target_eq);
1842         LASSERT (rc2 == 0);
1843  failed_0:
1844         lnet_destroy_ping_info();
1845         return rc;
1846 }
1847
1848 void
1849 lnet_ping_target_fini(void)
1850 {
1851         lnet_event_t    event;
1852         int             rc;
1853         int             which;
1854         int             timeout_ms = 1000;
1855         cfs_sigset_t    blocked = cfs_block_allsigs();
1856
1857         LNetMDUnlink(the_lnet.ln_ping_target_md);
1858         /* NB md could be busy; this just starts the unlink */
1859
1860         for (;;) {
1861                 rc = LNetEQPoll(&the_lnet.ln_ping_target_eq, 1,
1862                                 timeout_ms, &event, &which);
1863
1864                 /* I expect overflow... */
1865                 LASSERT (rc >= 0 || rc == -EOVERFLOW);
1866
1867                 if (rc == 0) {
1868                         /* timed out: provide a diagnostic */
1869                         CWARN("Still waiting for ping MD to unlink\n");
1870                         timeout_ms *= 2;
1871                         continue;
1872                 }
1873
1874                 /* Got a valid event */
1875                 if (event.unlinked)
1876                         break;
1877         }
1878
1879         rc = LNetEQFree(the_lnet.ln_ping_target_eq);
1880         LASSERT (rc == 0);
1881         lnet_destroy_ping_info();
1882         cfs_restore_sigs(blocked);
1883 }
1884
1885 int
1886 lnet_ping (lnet_process_id_t id, int timeout_ms, lnet_process_id_t *ids, int n_ids)
1887 {
1888         lnet_handle_eq_t     eqh;
1889         lnet_handle_md_t     mdh;
1890         lnet_event_t         event;
1891         lnet_md_t            md = {0};
1892         int                  which;
1893         int                  unlinked = 0;
1894         int                  replied = 0;
1895         const int            a_long_time = 60000; /* mS */
1896         int                  infosz = offsetof(lnet_ping_info_t, pi_ni[n_ids]);
1897         lnet_ping_info_t    *info;
1898         lnet_process_id_t    tmpid;
1899         int                  i;
1900         int                  nob;
1901         int                  rc;
1902         int                  rc2;
1903         cfs_sigset_t         blocked;
1904
1905         if (n_ids <= 0 ||
1906             id.nid == LNET_NID_ANY ||
1907             timeout_ms > 500000 ||              /* arbitrary limit! */
1908             n_ids > 20)                         /* arbitrary limit! */
1909                 return -EINVAL;
1910
1911         if (id.pid == LNET_PID_ANY)
1912                 id.pid = LUSTRE_SRV_LNET_PID;
1913
1914         LIBCFS_ALLOC(info, infosz);
1915         if (info == NULL)
1916                 return -ENOMEM;
1917
1918         /* NB 2 events max (including any unlink event) */
1919         rc = LNetEQAlloc(2, LNET_EQ_HANDLER_NONE, &eqh);
1920         if (rc != 0) {
1921                 CERROR("Can't allocate EQ: %d\n", rc);
1922                 goto out_0;
1923         }
1924
1925         /* initialize md content */
1926         md.start     = info;
1927         md.length    = infosz;
1928         md.threshold = 2; /*GET/REPLY*/
1929         md.max_size  = 0;
1930         md.options   = LNET_MD_TRUNCATE;
1931         md.user_ptr  = NULL;
1932         md.eq_handle = eqh;
1933
1934         rc = LNetMDBind(md, LNET_UNLINK, &mdh);
1935         if (rc != 0) {
1936                 CERROR("Can't bind MD: %d\n", rc);
1937                 goto out_1;
1938         }
1939
1940         rc = LNetGet(LNET_NID_ANY, mdh, id,
1941                      LNET_RESERVED_PORTAL,
1942                      LNET_PROTO_PING_MATCHBITS, 0);
1943
1944         if (rc != 0) {
1945                 /* Don't CERROR; this could be deliberate! */
1946
1947                 rc2 = LNetMDUnlink(mdh);
1948                 LASSERT (rc2 == 0);
1949
1950                 /* NB must wait for the UNLINK event below... */
1951                 unlinked = 1;
1952                 timeout_ms = a_long_time;
1953         }
1954
1955         do {
1956                 /* MUST block for unlink to complete */
1957                 if (unlinked)
1958                         blocked = cfs_block_allsigs();
1959
1960                 rc2 = LNetEQPoll(&eqh, 1, timeout_ms, &event, &which);
1961
1962                 if (unlinked)
1963                         cfs_restore_sigs(blocked);
1964
1965                 CDEBUG(D_NET, "poll %d(%d %d)%s\n", rc2,
1966                        (rc2 <= 0) ? -1 : event.type,
1967                        (rc2 <= 0) ? -1 : event.status,
1968                        (rc2 > 0 && event.unlinked) ? " unlinked" : "");
1969
1970                 LASSERT (rc2 != -EOVERFLOW);     /* can't miss anything */
1971
1972                 if (rc2 <= 0 || event.status != 0) {
1973                         /* timeout or error */
1974                         if (!replied && rc == 0)
1975                                 rc = (rc2 < 0) ? rc2 :
1976                                      (rc2 == 0) ? -ETIMEDOUT :
1977                                      event.status;
1978
1979                         if (!unlinked) {
1980                                 /* Ensure completion in finite time... */
1981                                 LNetMDUnlink(mdh);
1982                                 /* No assertion (racing with network) */
1983                                 unlinked = 1;
1984                                 timeout_ms = a_long_time;
1985                         } else if (rc2 == 0) {
1986                                 /* timed out waiting for unlink */
1987                                 CWARN("ping %s: late network completion\n",
1988                                       libcfs_id2str(id));
1989                         }
1990                 } else if (event.type == LNET_EVENT_REPLY) {
1991                         replied = 1;
1992                         rc = event.mlength;
1993                 }
1994
1995         } while (rc2 <= 0 || !event.unlinked);
1996
1997         if (!replied) {
1998                 if (rc >= 0)
1999                         CWARN("%s: Unexpected rc >= 0 but no reply!\n",
2000                               libcfs_id2str(id));
2001                 rc = -EIO;
2002                 goto out_1;
2003         }
2004
2005         nob = rc;
2006         LASSERT (nob >= 0 && nob <= infosz);
2007
2008         rc = -EPROTO;                           /* if I can't parse... */
2009
2010         if (nob < 8) {
2011                 /* can't check magic/version */
2012                 CERROR("%s: ping info too short %d\n",
2013                        libcfs_id2str(id), nob);
2014                 goto out_1;
2015         }
2016
2017         if (info->pi_magic == __swab32(LNET_PROTO_PING_MAGIC)) {
2018                 lnet_swap_pinginfo(info);
2019         } else if (info->pi_magic != LNET_PROTO_PING_MAGIC) {
2020                 CERROR("%s: Unexpected magic %08x\n",
2021                        libcfs_id2str(id), info->pi_magic);
2022                 goto out_1;
2023         }
2024
2025         if ((info->pi_features & LNET_PING_FEAT_NI_STATUS) == 0) {
2026                 CERROR("%s: ping w/o NI status: 0x%x\n",
2027                        libcfs_id2str(id), info->pi_features);
2028                 goto out_1;
2029         }
2030
2031         if (nob < offsetof(lnet_ping_info_t, pi_ni[0])) {
2032                 CERROR("%s: Short reply %d(%d min)\n", libcfs_id2str(id),
2033                        nob, (int)offsetof(lnet_ping_info_t, pi_ni[0]));
2034                 goto out_1;
2035         }
2036
2037         if (info->pi_nnis < n_ids)
2038                 n_ids = info->pi_nnis;
2039
2040         if (nob < offsetof(lnet_ping_info_t, pi_ni[n_ids])) {
2041                 CERROR("%s: Short reply %d(%d expected)\n", libcfs_id2str(id),
2042                        nob, (int)offsetof(lnet_ping_info_t, pi_ni[n_ids]));
2043                 goto out_1;
2044         }
2045
2046         rc = -EFAULT;                           /* If I SEGV... */
2047
2048         for (i = 0; i < n_ids; i++) {
2049                 tmpid.pid = info->pi_pid;
2050                 tmpid.nid = info->pi_ni[i].ns_nid;
2051 #ifdef __KERNEL__
2052                 if (cfs_copy_to_user(&ids[i], &tmpid, sizeof(tmpid)))
2053                         goto out_1;
2054 #else
2055                 ids[i] = tmpid;
2056 #endif
2057         }
2058         rc = info->pi_nnis;
2059
2060  out_1:
2061         rc2 = LNetEQFree(eqh);
2062         if (rc2 != 0)
2063                 CERROR("rc2 %d\n", rc2);
2064         LASSERT (rc2 == 0);
2065
2066  out_0:
2067         LIBCFS_FREE(info, infosz);
2068         return rc;
2069 }