lnet/lnet/peer.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.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2012, 2014, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  * Lustre is a trademark of Sun Microsystems, Inc.
  31  *
  32  * lnet/lnet/peer.c
  33  */
  34
  35 #define DEBUG_SUBSYSTEM S_LNET
  36
  37 #include <lnet/lib-lnet.h>
  38 #include <lnet/lib-dlc.h>
  39
  40 static void
  41 lnet_peer_remove_from_remote_list(struct lnet_peer_ni *lpni)
  42 {
  43         if (!list_empty(&lpni->lpni_on_remote_peer_ni_list)) {
  44                 list_del_init(&lpni->lpni_on_remote_peer_ni_list);
  45                 lnet_peer_ni_decref_locked(lpni);
  46         }
  47 }
  48
  49 void
  50 lnet_peer_net_added(struct lnet_net *net)
  51 {
  52         struct lnet_peer_ni *lpni, *tmp;
  53
  54         list_for_each_entry_safe(lpni, tmp, &the_lnet.ln_remote_peer_ni_list,
  55                                  lpni_on_remote_peer_ni_list) {
  56
  57                 if (LNET_NIDNET(lpni->lpni_nid) == net->net_id) {
  58                         lpni->lpni_net = net;
  59
  60                         spin_lock(&lpni->lpni_lock);
  61                         lpni->lpni_txcredits =
  62                                 lpni->lpni_net->net_tunables.lct_peer_tx_credits;
  63                         lpni->lpni_mintxcredits = lpni->lpni_txcredits;
  64                         lpni->lpni_rtrcredits =
  65                                 lnet_peer_buffer_credits(lpni->lpni_net);
  66                         lpni->lpni_minrtrcredits = lpni->lpni_rtrcredits;
  67                         spin_unlock(&lpni->lpni_lock);
  68
  69                         lnet_peer_remove_from_remote_list(lpni);
  70                 }
  71         }
  72 }
  73
  74 static void
  75 lnet_peer_tables_destroy(void)
  76 {
  77         struct lnet_peer_table  *ptable;
  78         struct list_head        *hash;
  79         int                     i;
  80         int                     j;
  81
  82         if (!the_lnet.ln_peer_tables)
  83                 return;
  84
  85         cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
  86                 hash = ptable->pt_hash;
  87                 if (!hash) /* not intialized */
  88                         break;
  89
  90                 LASSERT(list_empty(&ptable->pt_zombie_list));
  91
  92                 ptable->pt_hash = NULL;
  93                 for (j = 0; j < LNET_PEER_HASH_SIZE; j++)
  94                         LASSERT(list_empty(&hash[j]));
  95
  96                 LIBCFS_FREE(hash, LNET_PEER_HASH_SIZE * sizeof(*hash));
  97         }
  98
  99         cfs_percpt_free(the_lnet.ln_peer_tables);
 100         the_lnet.ln_peer_tables = NULL;
 101 }
 102
 103 int
 104 lnet_peer_tables_create(void)
 105 {
 106         struct lnet_peer_table  *ptable;
 107         struct list_head        *hash;
 108         int                     i;
 109         int                     j;
 110
 111         the_lnet.ln_peer_tables = cfs_percpt_alloc(lnet_cpt_table(),
 112                                                    sizeof(*ptable));
 113         if (the_lnet.ln_peer_tables == NULL) {
 114                 CERROR("Failed to allocate cpu-partition peer tables\n");
 115                 return -ENOMEM;
 116         }
 117
 118         cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
 119                 LIBCFS_CPT_ALLOC(hash, lnet_cpt_table(), i,
 120                                  LNET_PEER_HASH_SIZE * sizeof(*hash));
 121                 if (hash == NULL) {
 122                         CERROR("Failed to create peer hash table\n");
 123                         lnet_peer_tables_destroy();
 124                         return -ENOMEM;
 125                 }
 126
 127                 spin_lock_init(&ptable->pt_zombie_lock);
 128                 INIT_LIST_HEAD(&ptable->pt_zombie_list);
 129
 130                 for (j = 0; j < LNET_PEER_HASH_SIZE; j++)
 131                         INIT_LIST_HEAD(&hash[j]);
 132                 ptable->pt_hash = hash; /* sign of initialization */
 133         }
 134
 135         return 0;
 136 }
 137
 138 static struct lnet_peer_ni *
 139 lnet_peer_ni_alloc(lnet_nid_t nid)
 140 {
 141         struct lnet_peer_ni *lpni;
 142         struct lnet_net *net;
 143         int cpt;
 144
 145         cpt = lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
 146
 147         LIBCFS_CPT_ALLOC(lpni, lnet_cpt_table(), cpt, sizeof(*lpni));
 148         if (!lpni)
 149                 return NULL;
 150
 151         INIT_LIST_HEAD(&lpni->lpni_txq);
 152         INIT_LIST_HEAD(&lpni->lpni_rtrq);
 153         INIT_LIST_HEAD(&lpni->lpni_routes);
 154         INIT_LIST_HEAD(&lpni->lpni_hashlist);
 155         INIT_LIST_HEAD(&lpni->lpni_on_peer_net_list);
 156         INIT_LIST_HEAD(&lpni->lpni_on_remote_peer_ni_list);
 157
 158         spin_lock_init(&lpni->lpni_lock);
 159
 160         lpni->lpni_alive = !lnet_peers_start_down(); /* 1 bit!! */
 161         lpni->lpni_last_alive = cfs_time_current(); /* assumes alive */
 162         lpni->lpni_ping_feats = LNET_PING_FEAT_INVAL;
 163         lpni->lpni_nid = nid;
 164         lpni->lpni_cpt = cpt;
 165         lnet_set_peer_ni_health_locked(lpni, true);
 166
 167         net = lnet_get_net_locked(LNET_NIDNET(nid));
 168         lpni->lpni_net = net;
 169         if (net) {
 170                 lpni->lpni_txcredits = net->net_tunables.lct_peer_tx_credits;
 171                 lpni->lpni_mintxcredits = lpni->lpni_txcredits;
 172                 lpni->lpni_rtrcredits = lnet_peer_buffer_credits(net);
 173                 lpni->lpni_minrtrcredits = lpni->lpni_rtrcredits;
 174         } else {
 175                 /*
 176                  * This peer_ni is not on a local network, so we
 177                  * cannot add the credits here. In case the net is
 178                  * added later, add the peer_ni to the remote peer ni
 179                  * list so it can be easily found and revisited.
 180                  */
 181                 /* FIXME: per-net implementation instead? */
 182                 atomic_inc(&lpni->lpni_refcount);
 183                 list_add_tail(&lpni->lpni_on_remote_peer_ni_list,
 184                               &the_lnet.ln_remote_peer_ni_list);
 185         }
 186
 187         /* TODO: update flags */
 188
 189         return lpni;
 190 }
 191
 192 static struct lnet_peer_net *
 193 lnet_peer_net_alloc(__u32 net_id)
 194 {
 195         struct lnet_peer_net *lpn;
 196
 197         LIBCFS_CPT_ALLOC(lpn, lnet_cpt_table(), CFS_CPT_ANY, sizeof(*lpn));
 198         if (!lpn)
 199                 return NULL;
 200
 201         INIT_LIST_HEAD(&lpn->lpn_on_peer_list);
 202         INIT_LIST_HEAD(&lpn->lpn_peer_nis);
 203         lpn->lpn_net_id = net_id;
 204
 205         return lpn;
 206 }
 207
 208 static struct lnet_peer *
 209 lnet_peer_alloc(lnet_nid_t nid)
 210 {
 211         struct lnet_peer *lp;
 212
 213         LIBCFS_CPT_ALLOC(lp, lnet_cpt_table(), CFS_CPT_ANY, sizeof(*lp));
 214         if (!lp)
 215                 return NULL;
 216
 217         INIT_LIST_HEAD(&lp->lp_on_lnet_peer_list);
 218         INIT_LIST_HEAD(&lp->lp_peer_nets);
 219         lp->lp_primary_nid = nid;
 220
 221         /* TODO: update flags */
 222
 223         return lp;
 224 }
 225
 226
 227 static void
 228 lnet_try_destroy_peer_hierarchy_locked(struct lnet_peer_ni *lpni)
 229 {
 230         struct lnet_peer_net *peer_net;
 231         struct lnet_peer *peer;
 232
 233         /* TODO: could the below situation happen? accessing an already
 234          * destroyed peer? */
 235         if (lpni->lpni_peer_net == NULL ||
 236             lpni->lpni_peer_net->lpn_peer == NULL)
 237                 return;
 238
 239         peer_net = lpni->lpni_peer_net;
 240         peer = lpni->lpni_peer_net->lpn_peer;
 241
 242         list_del_init(&lpni->lpni_on_peer_net_list);
 243         lpni->lpni_peer_net = NULL;
 244
 245         /* if peer_net is empty, then remove it from the peer */
 246         if (list_empty(&peer_net->lpn_peer_nis)) {
 247                 list_del_init(&peer_net->lpn_on_peer_list);
 248                 peer_net->lpn_peer = NULL;
 249                 LIBCFS_FREE(peer_net, sizeof(*peer_net));
 250
 251                 /* if the peer is empty then remove it from the
 252                  * the_lnet.ln_peers */
 253                 if (list_empty(&peer->lp_peer_nets)) {
 254                         list_del_init(&peer->lp_on_lnet_peer_list);
 255                         LIBCFS_FREE(peer, sizeof(*peer));
 256                 }
 257         }
 258 }
 259
 260 /* called with lnet_net_lock LNET_LOCK_EX held */
 261 static int
 262 lnet_peer_ni_del_locked(struct lnet_peer_ni *lpni)
 263 {
 264         struct lnet_peer_table *ptable = NULL;
 265
 266         /* don't remove a peer_ni if it's also a gateway */
 267         if (lpni->lpni_rtr_refcount > 0) {
 268                 CERROR("Peer NI %s is a gateway. Can not delete it\n",
 269                        libcfs_nid2str(lpni->lpni_nid));
 270                 return -EBUSY;
 271         }
 272
 273         lnet_peer_remove_from_remote_list(lpni);
 274
 275         /* remove peer ni from the hash list. */
 276         list_del_init(&lpni->lpni_hashlist);
 277
 278         /* decrement the ref count on the peer table */
 279         ptable = the_lnet.ln_peer_tables[lpni->lpni_cpt];
 280         LASSERT(atomic_read(&ptable->pt_number) > 0);
 281         atomic_dec(&ptable->pt_number);
 282
 283         /*
 284          * The peer_ni can no longer be found with a lookup. But there
 285          * can be current users, so keep track of it on the zombie
 286          * list until the reference count has gone to zero.
 287          *
 288          * The last reference may be lost in a place where the
 289          * lnet_net_lock locks only a single cpt, and that cpt may not
 290          * be lpni->lpni_cpt. So the zombie list of lnet_peer_table
 291          * has its own lock.
 292          */
 293         spin_lock(&ptable->pt_zombie_lock);
 294         list_add(&lpni->lpni_hashlist, &ptable->pt_zombie_list);
 295         ptable->pt_zombies++;
 296         spin_unlock(&ptable->pt_zombie_lock);
 297
 298         /* no need to keep this peer on the hierarchy anymore */
 299         lnet_try_destroy_peer_hierarchy_locked(lpni);
 300
 301         /* decrement reference on peer */
 302         lnet_peer_ni_decref_locked(lpni);
 303
 304         return 0;
 305 }
 306
 307 void lnet_peer_uninit(void)
 308 {
 309         struct lnet_peer_ni *lpni, *tmp;
 310
 311         lnet_net_lock(LNET_LOCK_EX);
 312
 313         /* remove all peer_nis from the remote peer and the hash list */
 314         list_for_each_entry_safe(lpni, tmp, &the_lnet.ln_remote_peer_ni_list,
 315                                  lpni_on_remote_peer_ni_list)
 316                 lnet_peer_ni_del_locked(lpni);
 317
 318         lnet_peer_tables_destroy();
 319
 320         lnet_net_unlock(LNET_LOCK_EX);
 321 }
 322
 323 static int
 324 lnet_peer_del_locked(struct lnet_peer *peer)
 325 {
 326         struct lnet_peer_ni *lpni = NULL, *lpni2;
 327         int rc = 0, rc2 = 0;
 328
 329         lpni = lnet_get_next_peer_ni_locked(peer, NULL, lpni);
 330         while (lpni != NULL) {
 331                 lpni2 = lnet_get_next_peer_ni_locked(peer, NULL, lpni);
 332                 rc = lnet_peer_ni_del_locked(lpni);
 333                 if (rc != 0)
 334                         rc2 = rc;
 335                 lpni = lpni2;
 336         }
 337
 338         return rc2;
 339 }
 340
 341 static int
 342 lnet_peer_del(struct lnet_peer *peer)
 343 {
 344         lnet_net_lock(LNET_LOCK_EX);
 345         lnet_peer_del_locked(peer);
 346         lnet_net_unlock(LNET_LOCK_EX);
 347
 348         return 0;
 349 }
 350
 351 /*
 352  * Delete a NID from a peer.
 353  * Implements a few sanity checks.
 354  * Call with ln_api_mutex held.
 355  */
 356 static int
 357 lnet_peer_del_nid(struct lnet_peer *lp, lnet_nid_t nid)
 358 {
 359         struct lnet_peer *lp2;
 360         struct lnet_peer_ni *lpni;
 361
 362         lpni = lnet_find_peer_ni_locked(nid);
 363         if (!lpni) {
 364                 CERROR("Cannot remove unknown nid %s from peer %s\n",
 365                        libcfs_nid2str(nid),
 366                        libcfs_nid2str(lp->lp_primary_nid));
 367                 return -ENOENT;
 368         }
 369         lnet_peer_ni_decref_locked(lpni);
 370         lp2 = lpni->lpni_peer_net->lpn_peer;
 371         if (lp2 != lp) {
 372                 CERROR("Nid %s is attached to peer %s, not peer %s\n",
 373                        libcfs_nid2str(nid),
 374                        libcfs_nid2str(lp2->lp_primary_nid),
 375                        libcfs_nid2str(lp->lp_primary_nid));
 376                 return -EINVAL;
 377         }
 378
 379         /*
 380          * This function only allows deletion of the primary NID if it
 381          * is the only NID.
 382          */
 383         if (nid == lp->lp_primary_nid && lnet_get_num_peer_nis(lp) != 1) {
 384                 CERROR("Cannot delete primary NID %s from multi-NID peer\n",
 385                        libcfs_nid2str(nid));
 386                 return -EINVAL;
 387         }
 388
 389         lnet_net_lock(LNET_LOCK_EX);
 390         lnet_peer_ni_del_locked(lpni);
 391         lnet_net_unlock(LNET_LOCK_EX);
 392
 393         return 0;
 394 }
 395
 396 static void
 397 lnet_peer_table_cleanup_locked(struct lnet_net *net,
 398                                struct lnet_peer_table *ptable)
 399 {
 400         int                      i;
 401         struct lnet_peer_ni     *next;
 402         struct lnet_peer_ni     *lpni;
 403         struct lnet_peer        *peer;
 404
 405         for (i = 0; i < LNET_PEER_HASH_SIZE; i++) {
 406                 list_for_each_entry_safe(lpni, next, &ptable->pt_hash[i],
 407                                          lpni_hashlist) {
 408                         if (net != NULL && net != lpni->lpni_net)
 409                                 continue;
 410
 411                         peer = lpni->lpni_peer_net->lpn_peer;
 412                         if (peer->lp_primary_nid != lpni->lpni_nid) {
 413                                 lnet_peer_ni_del_locked(lpni);
 414                                 continue;
 415                         }
 416                         /*
 417                          * Removing the primary NID implies removing
 418                          * the entire peer. Advance next beyond any
 419                          * peer_ni that belongs to the same peer.
 420                          */
 421                         list_for_each_entry_from(next, &ptable->pt_hash[i],
 422                                                  lpni_hashlist) {
 423                                 if (next->lpni_peer_net->lpn_peer != peer)
 424                                         break;
 425                         }
 426                         lnet_peer_del_locked(peer);
 427                 }
 428         }
 429 }
 430
 431 static void
 432 lnet_peer_ni_finalize_wait(struct lnet_peer_table *ptable)
 433 {
 434         int     i = 3;
 435
 436         spin_lock(&ptable->pt_zombie_lock);
 437         while (ptable->pt_zombies) {
 438                 spin_unlock(&ptable->pt_zombie_lock);
 439
 440                 if (IS_PO2(i)) {
 441                         CDEBUG(D_WARNING,
 442                                "Waiting for %d zombies on peer table\n",
 443                                ptable->pt_zombies);
 444                 }
 445                 set_current_state(TASK_UNINTERRUPTIBLE);
 446                 schedule_timeout(cfs_time_seconds(1) >> 1);
 447                 spin_lock(&ptable->pt_zombie_lock);
 448         }
 449         spin_unlock(&ptable->pt_zombie_lock);
 450 }
 451
 452 static void
 453 lnet_peer_table_del_rtrs_locked(struct lnet_net *net,
 454                                 struct lnet_peer_table *ptable)
 455 {
 456         struct lnet_peer_ni     *lp;
 457         struct lnet_peer_ni     *tmp;
 458         lnet_nid_t              lpni_nid;
 459         int                     i;
 460
 461         for (i = 0; i < LNET_PEER_HASH_SIZE; i++) {
 462                 list_for_each_entry_safe(lp, tmp, &ptable->pt_hash[i],
 463                                          lpni_hashlist) {
 464                         if (net != lp->lpni_net)
 465                                 continue;
 466
 467                         if (lp->lpni_rtr_refcount == 0)
 468                                 continue;
 469
 470                         lpni_nid = lp->lpni_nid;
 471
 472                         lnet_net_unlock(LNET_LOCK_EX);
 473                         lnet_del_route(LNET_NIDNET(LNET_NID_ANY), lpni_nid);
 474                         lnet_net_lock(LNET_LOCK_EX);
 475                 }
 476         }
 477 }
 478
 479 void
 480 lnet_peer_tables_cleanup(struct lnet_net *net)
 481 {
 482         int                             i;
 483         struct lnet_peer_table          *ptable;
 484
 485         LASSERT(the_lnet.ln_state != LNET_STATE_SHUTDOWN || net != NULL);
 486         /* If just deleting the peers for a NI, get rid of any routes these
 487          * peers are gateways for. */
 488         cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
 489                 lnet_net_lock(LNET_LOCK_EX);
 490                 lnet_peer_table_del_rtrs_locked(net, ptable);
 491                 lnet_net_unlock(LNET_LOCK_EX);
 492         }
 493
 494         /* Start the cleanup process */
 495         cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
 496                 lnet_net_lock(LNET_LOCK_EX);
 497                 lnet_peer_table_cleanup_locked(net, ptable);
 498                 lnet_net_unlock(LNET_LOCK_EX);
 499         }
 500
 501         cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables)
 502                 lnet_peer_ni_finalize_wait(ptable);
 503 }
 504
 505 static struct lnet_peer_ni *
 506 lnet_get_peer_ni_locked(struct lnet_peer_table *ptable, lnet_nid_t nid)
 507 {
 508         struct list_head        *peers;
 509         struct lnet_peer_ni     *lp;
 510
 511         LASSERT(the_lnet.ln_state == LNET_STATE_RUNNING);
 512
 513         peers = &ptable->pt_hash[lnet_nid2peerhash(nid)];
 514         list_for_each_entry(lp, peers, lpni_hashlist) {
 515                 if (lp->lpni_nid == nid) {
 516                         lnet_peer_ni_addref_locked(lp);
 517                         return lp;
 518                 }
 519         }
 520
 521         return NULL;
 522 }
 523
 524 struct lnet_peer_ni *
 525 lnet_find_peer_ni_locked(lnet_nid_t nid)
 526 {
 527         struct lnet_peer_ni *lpni;
 528         struct lnet_peer_table *ptable;
 529         int cpt;
 530
 531         cpt = lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
 532
 533         ptable = the_lnet.ln_peer_tables[cpt];
 534         lpni = lnet_get_peer_ni_locked(ptable, nid);
 535
 536         return lpni;
 537 }
 538
 539 struct lnet_peer_ni *
 540 lnet_get_peer_ni_idx_locked(int idx, struct lnet_peer_net **lpn,
 541                             struct lnet_peer **lp)
 542 {
 543         struct lnet_peer_ni     *lpni;
 544
 545         list_for_each_entry((*lp), &the_lnet.ln_peers, lp_on_lnet_peer_list) {
 546                 list_for_each_entry((*lpn), &((*lp)->lp_peer_nets), lpn_on_peer_list) {
 547                         list_for_each_entry(lpni, &((*lpn)->lpn_peer_nis),
 548                                             lpni_on_peer_net_list)
 549                                 if (idx-- == 0)
 550                                         return lpni;
 551                 }
 552         }
 553
 554         return NULL;
 555 }
 556
 557 struct lnet_peer_ni *
 558 lnet_get_next_peer_ni_locked(struct lnet_peer *peer,
 559                              struct lnet_peer_net *peer_net,
 560                              struct lnet_peer_ni *prev)
 561 {
 562         struct lnet_peer_ni *lpni;
 563         struct lnet_peer_net *net = peer_net;
 564
 565         if (!prev) {
 566                 if (!net)
 567                         net = list_entry(peer->lp_peer_nets.next,
 568                                          struct lnet_peer_net,
 569                                          lpn_on_peer_list);
 570                 lpni = list_entry(net->lpn_peer_nis.next, struct lnet_peer_ni,
 571                                   lpni_on_peer_net_list);
 572
 573                 return lpni;
 574         }
 575
 576         if (prev->lpni_on_peer_net_list.next ==
 577             &prev->lpni_peer_net->lpn_peer_nis) {
 578                 /*
 579                  * if you reached the end of the peer ni list and the peer
 580                  * net is specified then there are no more peer nis in that
 581                  * net.
 582                  */
 583                 if (net)
 584                         return NULL;
 585
 586                 /*
 587                  * we reached the end of this net ni list. move to the
 588                  * next net
 589                  */
 590                 if (prev->lpni_peer_net->lpn_on_peer_list.next ==
 591                     &peer->lp_peer_nets)
 592                         /* no more nets and no more NIs. */
 593                         return NULL;
 594
 595                 /* get the next net */
 596                 net = list_entry(prev->lpni_peer_net->lpn_on_peer_list.next,
 597                                  struct lnet_peer_net,
 598                                  lpn_on_peer_list);
 599                 /* get the ni on it */
 600                 lpni = list_entry(net->lpn_peer_nis.next, struct lnet_peer_ni,
 601                                   lpni_on_peer_net_list);
 602
 603                 return lpni;
 604         }
 605
 606         /* there are more nis left */
 607         lpni = list_entry(prev->lpni_on_peer_net_list.next,
 608                           struct lnet_peer_ni, lpni_on_peer_net_list);
 609
 610         return lpni;
 611 }
 612
 613 bool
 614 lnet_peer_is_ni_pref_locked(struct lnet_peer_ni *lpni, struct lnet_ni *ni)
 615 {
 616         int i;
 617
 618         for (i = 0; i < lpni->lpni_pref_nnids; i++) {
 619                 if (lpni->lpni_pref_nids[i] == ni->ni_nid)
 620                         return true;
 621         }
 622         return false;
 623 }
 624
 625 lnet_nid_t
 626 lnet_peer_primary_nid(lnet_nid_t nid)
 627 {
 628         struct lnet_peer_ni *lpni;
 629         lnet_nid_t primary_nid = nid;
 630         int cpt;
 631
 632         cpt = lnet_net_lock_current();
 633         lpni = lnet_find_peer_ni_locked(nid);
 634         if (lpni) {
 635                 primary_nid = lpni->lpni_peer_net->lpn_peer->lp_primary_nid;
 636                 lnet_peer_ni_decref_locked(lpni);
 637         }
 638         lnet_net_unlock(cpt);
 639
 640         return primary_nid;
 641 }
 642
 643 lnet_nid_t
 644 LNetPrimaryNID(lnet_nid_t nid)
 645 {
 646         struct lnet_peer_ni *lpni;
 647         lnet_nid_t primary_nid = nid;
 648         int rc = 0;
 649         int cpt;
 650
 651         cpt = lnet_net_lock_current();
 652         lpni = lnet_nid2peerni_locked(nid, cpt);
 653         if (IS_ERR(lpni)) {
 654                 rc = PTR_ERR(lpni);
 655                 goto out_unlock;
 656         }
 657         primary_nid = lpni->lpni_peer_net->lpn_peer->lp_primary_nid;
 658         lnet_peer_ni_decref_locked(lpni);
 659 out_unlock:
 660         lnet_net_unlock(cpt);
 661
 662         CDEBUG(D_NET, "NID %s primary NID %s rc %d\n", libcfs_nid2str(nid),
 663                libcfs_nid2str(primary_nid), rc);
 664         return primary_nid;
 665 }
 666 EXPORT_SYMBOL(LNetPrimaryNID);
 667
 668 struct lnet_peer_net *
 669 lnet_peer_get_net_locked(struct lnet_peer *peer, __u32 net_id)
 670 {
 671         struct lnet_peer_net *peer_net;
 672         list_for_each_entry(peer_net, &peer->lp_peer_nets, lpn_on_peer_list) {
 673                 if (peer_net->lpn_net_id == net_id)
 674                         return peer_net;
 675         }
 676         return NULL;
 677 }
 678
 679 static int
 680 lnet_peer_setup_hierarchy(struct lnet_peer *lp, struct lnet_peer_ni *lpni,
 681                           lnet_nid_t nid)
 682 {
 683         struct lnet_peer_net *lpn = NULL;
 684         struct lnet_peer_table *ptable;
 685         __u32 net_id = LNET_NIDNET(nid);
 686
 687         /*
 688          * Create the peer_ni, peer_net, and peer if they don't exist
 689          * yet.
 690          */
 691         if (lp) {
 692                 lpn = lnet_peer_get_net_locked(lp, net_id);
 693         } else {
 694                 lp = lnet_peer_alloc(nid);
 695                 if (!lp)
 696                         goto out_enomem;
 697         }
 698
 699         if (!lpn) {
 700                 lpn = lnet_peer_net_alloc(net_id);
 701                 if (!lpn)
 702                         goto out_maybe_free_lp;
 703         }
 704
 705         if (!lpni) {
 706                 lpni = lnet_peer_ni_alloc(nid);
 707                 if (!lpni)
 708                         goto out_maybe_free_lpn;
 709         }
 710
 711         /* Install the new peer_ni */
 712         lnet_net_lock(LNET_LOCK_EX);
 713         /* Add peer_ni to global peer table hash, if necessary. */
 714         if (list_empty(&lpni->lpni_hashlist)) {
 715                 ptable = the_lnet.ln_peer_tables[lpni->lpni_cpt];
 716                 list_add_tail(&lpni->lpni_hashlist,
 717                               &ptable->pt_hash[lnet_nid2peerhash(nid)]);
 718                 ptable->pt_version++;
 719                 atomic_inc(&ptable->pt_number);
 720                 atomic_inc(&lpni->lpni_refcount);
 721         }
 722
 723         /* Detach the peer_ni from an existing peer, if necessary. */
 724         if (lpni->lpni_peer_net && lpni->lpni_peer_net->lpn_peer != lp)
 725                 lnet_try_destroy_peer_hierarchy_locked(lpni);
 726
 727         /* Add peer_ni to peer_net */
 728         lpni->lpni_peer_net = lpn;
 729         list_add_tail(&lpni->lpni_on_peer_net_list, &lpn->lpn_peer_nis);
 730
 731         /* Add peer_net to peer */
 732         if (!lpn->lpn_peer) {
 733                 lpn->lpn_peer = lp;
 734                 list_add_tail(&lpn->lpn_on_peer_list, &lp->lp_peer_nets);
 735         }
 736
 737         /* Add peer to global peer list */
 738         if (list_empty(&lp->lp_on_lnet_peer_list))
 739                 list_add_tail(&lp->lp_on_lnet_peer_list, &the_lnet.ln_peers);
 740         lnet_net_unlock(LNET_LOCK_EX);
 741
 742         return 0;
 743
 744 out_maybe_free_lpn:
 745         if (list_empty(&lpn->lpn_on_peer_list))
 746                 LIBCFS_FREE(lpn, sizeof(*lpn));
 747 out_maybe_free_lp:
 748         if (list_empty(&lp->lp_on_lnet_peer_list))
 749                 LIBCFS_FREE(lp, sizeof(*lp));
 750 out_enomem:
 751         return -ENOMEM;
 752 }
 753
 754 /*
 755  * Create a new peer, with nid as its primary nid.
 756  *
 757  * It is not an error if the peer already exists, provided that the
 758  * given nid is the primary NID.
 759  *
 760  * Call with the lnet_api_mutex held.
 761  */
 762 static int
 763 lnet_peer_add(lnet_nid_t nid, bool mr)
 764 {
 765         struct lnet_peer *lp;
 766         struct lnet_peer_ni *lpni;
 767         int rc;
 768
 769         LASSERT(nid != LNET_NID_ANY);
 770
 771         /*
 772          * No need for the lnet_net_lock here, because the
 773          * lnet_api_mutex is held.
 774          */
 775         lpni = lnet_find_peer_ni_locked(nid);
 776         if (!lpni) {
 777                 rc = lnet_peer_setup_hierarchy(NULL, NULL, nid);
 778                 if (rc != 0)
 779                         return rc;
 780                 lpni = lnet_find_peer_ni_locked(nid);
 781                 LASSERT(lpni);
 782         }
 783         lp = lpni->lpni_peer_net->lpn_peer;
 784         lnet_peer_ni_decref_locked(lpni);
 785
 786         /* A found peer must have this primary NID */
 787         if (lp->lp_primary_nid != nid)
 788                 return -EEXIST;
 789
 790         /*
 791          * If we found an lpni that is not a multi-rail, which could occur
 792          * if lpni is already created as a non-mr lpni or we just created
 793          * it, then make sure you indicate that this lpni is a primary mr
 794          * capable peer.
 795          *
 796          * TODO: update flags if necessary
 797          */
 798         if (mr && !lp->lp_multi_rail) {
 799                 lp->lp_multi_rail = true;
 800         } else if (!mr && lp->lp_multi_rail) {
 801                 /* The mr state is sticky. */
 802                 CDEBUG(D_NET, "Cannot clear multi-flag from peer %s\n",
 803                        libcfs_nid2str(nid));
 804         }
 805
 806         return 0;
 807 }
 808
 809 static int
 810 lnet_peer_add_nid(struct lnet_peer *lp, lnet_nid_t nid, bool mr)
 811 {
 812         struct lnet_peer_ni *lpni;
 813
 814         LASSERT(lp);
 815         LASSERT(nid != LNET_NID_ANY);
 816
 817         if (!mr && !lp->lp_multi_rail) {
 818                 CERROR("Cannot add nid %s to non-multi-rail peer %s\n",
 819                        libcfs_nid2str(nid),
 820                        libcfs_nid2str(lp->lp_primary_nid));
 821                 return -EPERM;
 822         }
 823
 824         if (!lp->lp_multi_rail)
 825                 lp->lp_multi_rail = true;
 826
 827         lpni = lnet_find_peer_ni_locked(nid);
 828         if (!lpni)
 829                 return lnet_peer_setup_hierarchy(lp, NULL, nid);
 830
 831         if (lpni->lpni_peer_net->lpn_peer != lp) {
 832                 struct lnet_peer *lp2 = lpni->lpni_peer_net->lpn_peer;
 833                 CERROR("Cannot add NID %s owned by peer %s to peer %s\n",
 834                        libcfs_nid2str(lpni->lpni_nid),
 835                        libcfs_nid2str(lp2->lp_primary_nid),
 836                        libcfs_nid2str(lp->lp_primary_nid));
 837                 return -EEXIST;
 838         }
 839
 840         CDEBUG(D_NET, "NID %s is already owned by peer %s\n",
 841                libcfs_nid2str(lpni->lpni_nid),
 842                libcfs_nid2str(lp->lp_primary_nid));
 843         return 0;
 844 }
 845
 846 /*
 847  * lpni creation initiated due to traffic either sending or receiving.
 848  */
 849 static int
 850 lnet_peer_ni_traffic_add(lnet_nid_t nid)
 851 {
 852         struct lnet_peer_ni *lpni;
 853         int rc = 0;
 854
 855         if (nid == LNET_NID_ANY)
 856                 return -EINVAL;
 857
 858         /* lnet_net_lock is not needed here because ln_api_lock is held */
 859         lpni = lnet_find_peer_ni_locked(nid);
 860         if (lpni) {
 861                 /*
 862                  * TODO: lnet_update_primary_nid() but not all of it
 863                  * only indicate if we're converting this to MR capable
 864                  * Can happen due to DD
 865                  */
 866                 lnet_peer_ni_decref_locked(lpni);
 867         } else {
 868                 rc = lnet_peer_setup_hierarchy(NULL, NULL, nid);
 869         }
 870
 871         return rc;
 872
 873 }
 874
 875 /*
 876  * Implementation of IOC_LIBCFS_ADD_PEER_NI.
 877  *
 878  * This API handles the following combinations:
 879  *   Create a peer with its primary NI if only the prim_nid is provided
 880  *   Add a NID to a peer identified by the prim_nid. The peer identified
 881  *   by the prim_nid must already exist.
 882  *   The peer being created may be non-MR.
 883  *
 884  * The caller must hold ln_api_mutex. This prevents the peer from
 885  * being created/modified/deleted by a different thread.
 886  */
 887 int
 888 lnet_add_peer_ni(lnet_nid_t prim_nid, lnet_nid_t nid, bool mr)
 889 {
 890         struct lnet_peer *lp = NULL;
 891         struct lnet_peer_ni *lpni;
 892
 893         /* The prim_nid must always be specified */
 894         if (prim_nid == LNET_NID_ANY)
 895                 return -EINVAL;
 896
 897         /*
 898          * If nid isn't specified, we must create a new peer with
 899          * prim_nid as its primary nid.
 900          */
 901         if (nid == LNET_NID_ANY)
 902                 return lnet_peer_add(prim_nid, mr);
 903
 904         /* Look up the prim_nid, which must exist. */
 905         lpni = lnet_find_peer_ni_locked(prim_nid);
 906         if (!lpni)
 907                 return -ENOENT;
 908         lnet_peer_ni_decref_locked(lpni);
 909         lp = lpni->lpni_peer_net->lpn_peer;
 910
 911         if (lp->lp_primary_nid != prim_nid) {
 912                 CDEBUG(D_NET, "prim_nid %s is not primary for peer %s\n",
 913                        libcfs_nid2str(prim_nid),
 914                        libcfs_nid2str(lp->lp_primary_nid));
 915                 return -ENODEV;
 916         }
 917
 918         return lnet_peer_add_nid(lp, nid, mr);
 919 }
 920
 921 /*
 922  * Implementation of IOC_LIBCFS_DEL_PEER_NI.
 923  *
 924  * This API handles the following combinations:
 925  *   Delete a NI from a peer if both prim_nid and nid are provided.
 926  *   Delete a peer if only prim_nid is provided.
 927  *   Delete a peer if its primary nid is provided.
 928  *
 929  * The caller must hold ln_api_mutex. This prevents the peer from
 930  * being modified/deleted by a different thread.
 931  */
 932 int
 933 lnet_del_peer_ni(lnet_nid_t prim_nid, lnet_nid_t nid)
 934 {
 935         struct lnet_peer *lp;
 936         struct lnet_peer_ni *lpni;
 937
 938         if (prim_nid == LNET_NID_ANY)
 939                 return -EINVAL;
 940
 941         lpni = lnet_find_peer_ni_locked(prim_nid);
 942         if (!lpni)
 943                 return -ENOENT;
 944         lnet_peer_ni_decref_locked(lpni);
 945         lp = lpni->lpni_peer_net->lpn_peer;
 946
 947         if (prim_nid != lp->lp_primary_nid) {
 948                 CDEBUG(D_NET, "prim_nid %s is not primary for peer %s\n",
 949                        libcfs_nid2str(prim_nid),
 950                        libcfs_nid2str(lp->lp_primary_nid));
 951                 return -ENODEV;
 952         }
 953
 954         if (nid == LNET_NID_ANY || nid == lp->lp_primary_nid)
 955                 return lnet_peer_del(lp);
 956
 957         return lnet_peer_del_nid(lp, nid);
 958 }
 959
 960 void
 961 lnet_destroy_peer_ni_locked(struct lnet_peer_ni *lpni)
 962 {
 963         struct lnet_peer_table *ptable;
 964
 965         LASSERT(atomic_read(&lpni->lpni_refcount) == 0);
 966         LASSERT(lpni->lpni_rtr_refcount == 0);
 967         LASSERT(list_empty(&lpni->lpni_txq));
 968         LASSERT(lpni->lpni_txqnob == 0);
 969
 970         lpni->lpni_net = NULL;
 971
 972         /* remove the peer ni from the zombie list */
 973         ptable = the_lnet.ln_peer_tables[lpni->lpni_cpt];
 974         spin_lock(&ptable->pt_zombie_lock);
 975         list_del_init(&lpni->lpni_hashlist);
 976         ptable->pt_zombies--;
 977         spin_unlock(&ptable->pt_zombie_lock);
 978
 979         LIBCFS_FREE(lpni, sizeof(*lpni));
 980 }
 981
 982 struct lnet_peer_ni *
 983 lnet_nid2peerni_ex(lnet_nid_t nid, int cpt)
 984 {
 985         struct lnet_peer_ni *lpni = NULL;
 986         int rc;
 987
 988         if (the_lnet.ln_state != LNET_STATE_RUNNING)
 989                 return ERR_PTR(-ESHUTDOWN);
 990
 991         /*
 992          * find if a peer_ni already exists.
 993          * If so then just return that.
 994          */
 995         lpni = lnet_find_peer_ni_locked(nid);
 996         if (lpni)
 997                 return lpni;
 998
 999         lnet_net_unlock(cpt);
1000
1001         rc = lnet_peer_ni_traffic_add(nid);
1002         if (rc) {
1003                 lpni = ERR_PTR(rc);
1004                 goto out_net_relock;
1005         }
1006
1007         lpni = lnet_find_peer_ni_locked(nid);
1008         LASSERT(lpni);
1009
1010 out_net_relock:
1011         lnet_net_lock(cpt);
1012
1013         return lpni;
1014 }
1015
1016 struct lnet_peer_ni *
1017 lnet_nid2peerni_locked(lnet_nid_t nid, int cpt)
1018 {
1019         struct lnet_peer_ni *lpni = NULL;
1020         int rc;
1021
1022         if (the_lnet.ln_state != LNET_STATE_RUNNING)
1023                 return ERR_PTR(-ESHUTDOWN);
1024
1025         /*
1026          * find if a peer_ni already exists.
1027          * If so then just return that.
1028          */
1029         lpni = lnet_find_peer_ni_locked(nid);
1030         if (lpni)
1031                 return lpni;
1032
1033         /*
1034          * Slow path:
1035          * use the lnet_api_mutex to serialize the creation of the peer_ni
1036          * and the creation/deletion of the local ni/net. When a local ni is
1037          * created, if there exists a set of peer_nis on that network,
1038          * they need to be traversed and updated. When a local NI is
1039          * deleted, which could result in a network being deleted, then
1040          * all peer nis on that network need to be removed as well.
1041          *
1042          * Creation through traffic should also be serialized with
1043          * creation through DLC.
1044          */
1045         lnet_net_unlock(cpt);
1046         mutex_lock(&the_lnet.ln_api_mutex);
1047         /*
1048          * Shutdown is only set under the ln_api_lock, so a single
1049          * check here is sufficent.
1050          */
1051         if (the_lnet.ln_state != LNET_STATE_RUNNING) {
1052                 lpni = ERR_PTR(-ESHUTDOWN);
1053                 goto out_mutex_unlock;
1054         }
1055
1056         rc = lnet_peer_ni_traffic_add(nid);
1057         if (rc) {
1058                 lpni = ERR_PTR(rc);
1059                 goto out_mutex_unlock;
1060         }
1061
1062         lpni = lnet_find_peer_ni_locked(nid);
1063         LASSERT(lpni);
1064
1065 out_mutex_unlock:
1066         mutex_unlock(&the_lnet.ln_api_mutex);
1067         lnet_net_lock(cpt);
1068
1069         return lpni;
1070 }
1071
1072 void
1073 lnet_debug_peer(lnet_nid_t nid)
1074 {
1075         char                    *aliveness = "NA";
1076         struct lnet_peer_ni     *lp;
1077         int                     cpt;
1078
1079         cpt = lnet_cpt_of_nid(nid, NULL);
1080         lnet_net_lock(cpt);
1081
1082         lp = lnet_nid2peerni_locked(nid, cpt);
1083         if (IS_ERR(lp)) {
1084                 lnet_net_unlock(cpt);
1085                 CDEBUG(D_WARNING, "No peer %s\n", libcfs_nid2str(nid));
1086                 return;
1087         }
1088
1089         if (lnet_isrouter(lp) || lnet_peer_aliveness_enabled(lp))
1090                 aliveness = lp->lpni_alive ? "up" : "down";
1091
1092         CDEBUG(D_WARNING, "%-24s %4d %5s %5d %5d %5d %5d %5d %ld\n",
1093                libcfs_nid2str(lp->lpni_nid), atomic_read(&lp->lpni_refcount),
1094                aliveness, lp->lpni_net->net_tunables.lct_peer_tx_credits,
1095                lp->lpni_rtrcredits, lp->lpni_minrtrcredits,
1096                lp->lpni_txcredits, lp->lpni_mintxcredits, lp->lpni_txqnob);
1097
1098         lnet_peer_ni_decref_locked(lp);
1099
1100         lnet_net_unlock(cpt);
1101 }
1102
1103 int lnet_get_peer_ni_info(__u32 peer_index, __u64 *nid,
1104                           char aliveness[LNET_MAX_STR_LEN],
1105                           __u32 *cpt_iter, __u32 *refcount,
1106                           __u32 *ni_peer_tx_credits, __u32 *peer_tx_credits,
1107                           __u32 *peer_rtr_credits, __u32 *peer_min_rtr_credits,
1108                           __u32 *peer_tx_qnob)
1109 {
1110         struct lnet_peer_table          *peer_table;
1111         struct lnet_peer_ni             *lp;
1112         int                             j;
1113         int                             lncpt;
1114         bool                            found = false;
1115
1116         /* get the number of CPTs */
1117         lncpt = cfs_percpt_number(the_lnet.ln_peer_tables);
1118
1119         /* if the cpt number to be examined is >= the number of cpts in
1120          * the system then indicate that there are no more cpts to examin
1121          */
1122         if (*cpt_iter >= lncpt)
1123                 return -ENOENT;
1124
1125         /* get the current table */
1126         peer_table = the_lnet.ln_peer_tables[*cpt_iter];
1127         /* if the ptable is NULL then there are no more cpts to examine */
1128         if (peer_table == NULL)
1129                 return -ENOENT;
1130
1131         lnet_net_lock(*cpt_iter);
1132
1133         for (j = 0; j < LNET_PEER_HASH_SIZE && !found; j++) {
1134                 struct list_head *peers = &peer_table->pt_hash[j];
1135
1136                 list_for_each_entry(lp, peers, lpni_hashlist) {
1137                         if (peer_index-- > 0)
1138                                 continue;
1139
1140                         snprintf(aliveness, LNET_MAX_STR_LEN, "NA");
1141                         if (lnet_isrouter(lp) ||
1142                                 lnet_peer_aliveness_enabled(lp))
1143                                 snprintf(aliveness, LNET_MAX_STR_LEN,
1144                                          lp->lpni_alive ? "up" : "down");
1145
1146                         *nid = lp->lpni_nid;
1147                         *refcount = atomic_read(&lp->lpni_refcount);
1148                         *ni_peer_tx_credits =
1149                                 lp->lpni_net->net_tunables.lct_peer_tx_credits;
1150                         *peer_tx_credits = lp->lpni_txcredits;
1151                         *peer_rtr_credits = lp->lpni_rtrcredits;
1152                         *peer_min_rtr_credits = lp->lpni_mintxcredits;
1153                         *peer_tx_qnob = lp->lpni_txqnob;
1154
1155                         found = true;
1156                 }
1157
1158         }
1159         lnet_net_unlock(*cpt_iter);
1160
1161         *cpt_iter = lncpt;
1162
1163         return found ? 0 : -ENOENT;
1164 }
1165
1166 int lnet_get_peer_info(__u32 idx, lnet_nid_t *primary_nid, lnet_nid_t *nid,
1167                        bool *mr, struct lnet_peer_ni_credit_info *peer_ni_info,
1168                        struct lnet_ioctl_element_stats *peer_ni_stats)
1169 {
1170         struct lnet_peer_ni *lpni = NULL;
1171         struct lnet_peer_net *lpn = NULL;
1172         struct lnet_peer *lp = NULL;
1173
1174         lpni = lnet_get_peer_ni_idx_locked(idx, &lpn, &lp);
1175
1176         if (!lpni)
1177                 return -ENOENT;
1178
1179         *primary_nid = lp->lp_primary_nid;
1180         *mr = lp->lp_multi_rail;
1181         *nid = lpni->lpni_nid;
1182         snprintf(peer_ni_info->cr_aliveness, LNET_MAX_STR_LEN, "NA");
1183         if (lnet_isrouter(lpni) ||
1184                 lnet_peer_aliveness_enabled(lpni))
1185                 snprintf(peer_ni_info->cr_aliveness, LNET_MAX_STR_LEN,
1186                          lpni->lpni_alive ? "up" : "down");
1187
1188         peer_ni_info->cr_refcount = atomic_read(&lpni->lpni_refcount);
1189         peer_ni_info->cr_ni_peer_tx_credits = (lpni->lpni_net != NULL) ?
1190                 lpni->lpni_net->net_tunables.lct_peer_tx_credits : 0;
1191         peer_ni_info->cr_peer_tx_credits = lpni->lpni_txcredits;
1192         peer_ni_info->cr_peer_rtr_credits = lpni->lpni_rtrcredits;
1193         peer_ni_info->cr_peer_min_rtr_credits = lpni->lpni_minrtrcredits;
1194         peer_ni_info->cr_peer_min_tx_credits = lpni->lpni_mintxcredits;
1195         peer_ni_info->cr_peer_tx_qnob = lpni->lpni_txqnob;
1196
1197         peer_ni_stats->send_count = atomic_read(&lpni->lpni_stats.send_count);
1198         peer_ni_stats->recv_count = atomic_read(&lpni->lpni_stats.recv_count);
1199         peer_ni_stats->drop_count = atomic_read(&lpni->lpni_stats.drop_count);
1200
1201         return 0;
1202 }