Whamcloud - gitweb
LU-7734 lnet: handle non-MR peers
[fs/lustre-release.git] / lnet / lnet / peer.c
index d4b58fa..c9e93c5 100644 (file)
-/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
- * vim:expandtab:shiftwidth=8:tabstop=8:
+/*
+ * GPL HEADER START
  *
- * lib/lib-move.c
- * Data movement routines
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
- *  Copyright (c) 2001-2003 Cluster File Systems, Inc.
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 only,
+ * as published by the Free Software Foundation.
  *
- *   This file is part of Lustre, http://www.lustre.org
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License version 2 for more details (a copy is included
+ * in the LICENSE file that accompanied this code).
  *
- *   Lustre is free software; you can redistribute it and/or
- *   modify it under the terms of version 2 of the GNU General Public
- *   License as published by the Free Software Foundation.
+ * You should have received a copy of the GNU General Public License
+ * version 2 along with this program; If not, see
+ * http://www.gnu.org/licenses/gpl-2.0.html
  *
- *   Lustre is distributed in the hope that it will be useful,
- *   but WITHOUT ANY WARRANTY; without even the implied warranty of
- *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- *   GNU General Public License for more details.
+ * GPL HEADER END
+ */
+/*
+ * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
+ * Use is subject to license terms.
+ *
+ * Copyright (c) 2012, 2014, Intel Corporation.
+ */
+/*
+ * This file is part of Lustre, http://www.lustre.org/
+ * Lustre is a trademark of Sun Microsystems, Inc.
  *
- *   You should have received a copy of the GNU General Public License
- *   along with Lustre; if not, write to the Free Software
- *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * lnet/lnet/peer.c
  */
 
 #define DEBUG_SUBSYSTEM S_LNET
 
 #include <lnet/lib-lnet.h>
+#include <lnet/lib-dlc.h>
 
-int
-lnet_create_peer_table(void)
+static void
+lnet_peer_remove_from_remote_list(struct lnet_peer_ni *lpni)
 {
-       struct list_head *hash;
-       int               i;
+       if (!list_empty(&lpni->lpni_on_remote_peer_ni_list)) {
+               list_del_init(&lpni->lpni_on_remote_peer_ni_list);
+               lnet_peer_ni_decref_locked(lpni);
+       }
+}
 
-       LASSERT (the_lnet.ln_peer_hash == NULL);
-       LIBCFS_ALLOC(hash, LNET_PEER_HASHSIZE * sizeof(struct list_head));
-       
-       if (hash == NULL) {
-               CERROR("Can't allocate peer hash table\n");
+void
+lnet_peer_net_added(struct lnet_net *net)
+{
+       struct lnet_peer_ni *lpni, *tmp;
+
+       list_for_each_entry_safe(lpni, tmp, &the_lnet.ln_remote_peer_ni_list,
+                                lpni_on_remote_peer_ni_list) {
+
+               if (LNET_NIDNET(lpni->lpni_nid) == net->net_id) {
+                       lpni->lpni_net = net;
+                       lpni->lpni_txcredits =
+                       lpni->lpni_mintxcredits =
+                               lpni->lpni_net->net_tunables.lct_peer_tx_credits;
+                       lpni->lpni_rtrcredits =
+                       lpni->lpni_minrtrcredits =
+                               lnet_peer_buffer_credits(lpni->lpni_net);
+
+                       lnet_peer_remove_from_remote_list(lpni);
+               }
+       }
+}
+
+static void
+lnet_peer_tables_destroy(void)
+{
+       struct lnet_peer_table  *ptable;
+       struct list_head        *hash;
+       int                     i;
+       int                     j;
+
+       if (!the_lnet.ln_peer_tables)
+               return;
+
+       cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
+               hash = ptable->pt_hash;
+               if (!hash) /* not intialized */
+                       break;
+
+               ptable->pt_hash = NULL;
+               for (j = 0; j < LNET_PEER_HASH_SIZE; j++)
+                       LASSERT(list_empty(&hash[j]));
+
+               LIBCFS_FREE(hash, LNET_PEER_HASH_SIZE * sizeof(*hash));
+       }
+
+       cfs_percpt_free(the_lnet.ln_peer_tables);
+       the_lnet.ln_peer_tables = NULL;
+}
+
+int
+lnet_peer_tables_create(void)
+{
+       struct lnet_peer_table  *ptable;
+       struct list_head        *hash;
+       int                     i;
+       int                     j;
+
+       the_lnet.ln_peer_tables = cfs_percpt_alloc(lnet_cpt_table(),
+                                                  sizeof(*ptable));
+       if (the_lnet.ln_peer_tables == NULL) {
+               CERROR("Failed to allocate cpu-partition peer tables\n");
                return -ENOMEM;
        }
 
-       for (i = 0; i < LNET_PEER_HASHSIZE; i++)
-               CFS_INIT_LIST_HEAD(&hash[i]);
+       cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
+               LIBCFS_CPT_ALLOC(hash, lnet_cpt_table(), i,
+                                LNET_PEER_HASH_SIZE * sizeof(*hash));
+               if (hash == NULL) {
+                       CERROR("Failed to create peer hash table\n");
+                       lnet_peer_tables_destroy();
+                       return -ENOMEM;
+               }
+
+               for (j = 0; j < LNET_PEER_HASH_SIZE; j++)
+                       INIT_LIST_HEAD(&hash[j]);
+               ptable->pt_hash = hash; /* sign of initialization */
+       }
 
-       the_lnet.ln_peer_hash = hash;
        return 0;
 }
 
-void
-lnet_destroy_peer_table(void)
+void lnet_peer_uninit()
 {
-       int         i;
+       int cpt;
+       struct lnet_peer_ni *lpni, *tmp;
+       struct lnet_peer_table *ptable = NULL;
+
+       /* remove all peer_nis from the remote peer and he hash list */
+       list_for_each_entry_safe(lpni, tmp, &the_lnet.ln_remote_peer_ni_list,
+                                lpni_on_remote_peer_ni_list) {
+               list_del_init(&lpni->lpni_on_remote_peer_ni_list);
+               lnet_peer_ni_decref_locked(lpni);
+
+               cpt = lnet_cpt_of_nid_locked(lpni->lpni_nid, NULL);
+               ptable = the_lnet.ln_peer_tables[cpt];
+               ptable->pt_zombies++;
+
+               list_del_init(&lpni->lpni_hashlist);
+               lnet_peer_ni_decref_locked(lpni);
+       }
 
-        if (the_lnet.ln_peer_hash == NULL)
-                return;
+       lnet_peer_tables_destroy();
+}
 
-       for (i = 0; i < LNET_PEER_HASHSIZE; i++)
-               LASSERT (list_empty(&the_lnet.ln_peer_hash[i]));
-       
-       LIBCFS_FREE(the_lnet.ln_peer_hash,
-                   LNET_PEER_HASHSIZE * sizeof (struct list_head));
-        the_lnet.ln_peer_hash = NULL;
+static void
+lnet_peer_table_cleanup_locked(lnet_ni_t *ni, struct lnet_peer_table *ptable)
+{
+       int                      i;
+       struct lnet_peer_ni     *lp;
+       struct lnet_peer_ni     *tmp;
+
+       for (i = 0; i < LNET_PEER_HASH_SIZE; i++) {
+               list_for_each_entry_safe(lp, tmp, &ptable->pt_hash[i],
+                                        lpni_hashlist) {
+                       if (ni != NULL && ni->ni_net != lp->lpni_net)
+                               continue;
+                       list_del_init(&lp->lpni_hashlist);
+                       /* Lose hash table's ref */
+                       ptable->pt_zombies++;
+                       lnet_peer_ni_decref_locked(lp);
+               }
+       }
 }
 
-void
-lnet_clear_peer_table(void)
-{
-       int         i;
-
-        LASSERT (the_lnet.ln_shutdown);         /* i.e. no new peers */
-       
-       for (i = 0; i < LNET_PEER_HASHSIZE; i++) {
-               struct list_head *peers = &the_lnet.ln_peer_hash[i];
-
-               LNET_LOCK();
-               while (!list_empty(peers)) {
-                       lnet_peer_t *lp = list_entry(peers->next,
-                                                    lnet_peer_t, lp_hashlist);
-                       
-                       list_del(&lp->lp_hashlist);
-                        lnet_peer_decref_locked(lp);   /* lose hash table's ref */
+static void
+lnet_peer_table_finalize_wait_locked(struct lnet_peer_table *ptable,
+                                    int cpt_locked)
+{
+       int     i;
+
+       for (i = 3; ptable->pt_zombies != 0; i++) {
+               lnet_net_unlock(cpt_locked);
+
+               if (IS_PO2(i)) {
+                       CDEBUG(D_WARNING,
+                              "Waiting for %d zombies on peer table\n",
+                              ptable->pt_zombies);
                }
-               LNET_UNLOCK();
+               set_current_state(TASK_UNINTERRUPTIBLE);
+               schedule_timeout(cfs_time_seconds(1) >> 1);
+               lnet_net_lock(cpt_locked);
        }
+}
+
+static void
+lnet_peer_table_del_rtrs_locked(lnet_ni_t *ni, struct lnet_peer_table *ptable,
+                               int cpt_locked)
+{
+       struct lnet_peer_ni     *lp;
+       struct lnet_peer_ni     *tmp;
+       lnet_nid_t              lpni_nid;
+       int                     i;
+
+       for (i = 0; i < LNET_PEER_HASH_SIZE; i++) {
+               list_for_each_entry_safe(lp, tmp, &ptable->pt_hash[i],
+                                        lpni_hashlist) {
+                       if (ni->ni_net != lp->lpni_net)
+                               continue;
 
-        LNET_LOCK();
-        for (i = 3; the_lnet.ln_npeers != 0;i++) {
-                LNET_UNLOCK();
+                       if (lp->lpni_rtr_refcount == 0)
+                               continue;
 
-                if ((i & (i-1)) == 0)
-                        CDEBUG(D_WARNING,"Waiting for %d peers\n", 
-                               the_lnet.ln_npeers);
-                cfs_pause(cfs_time_seconds(1));
+                       lpni_nid = lp->lpni_nid;
 
-                LNET_LOCK();
-        }
-        LNET_UNLOCK();
+                       lnet_net_unlock(cpt_locked);
+                       lnet_del_route(LNET_NIDNET(LNET_NID_ANY), lpni_nid);
+                       lnet_net_lock(cpt_locked);
+               }
+       }
 }
 
 void
-lnet_destroy_peer_locked (lnet_peer_t *lp) 
+lnet_peer_tables_cleanup(lnet_ni_t *ni)
 {
-        lnet_ni_decref_locked(lp->lp_ni);
-        LNET_UNLOCK();
+       int                             i;
+       struct lnet_peer_table          *ptable;
+
+       LASSERT(the_lnet.ln_shutdown || ni != NULL);
+       /* If just deleting the peers for a NI, get rid of any routes these
+        * peers are gateways for. */
+       cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
+               lnet_net_lock(LNET_LOCK_EX);
+               lnet_peer_table_del_rtrs_locked(ni, ptable, i);
+               lnet_net_unlock(LNET_LOCK_EX);
+       }
 
-        LASSERT (lp->lp_refcount == 0);
-        LASSERT (lp->lp_rtr_refcount == 0);
-       LASSERT (list_empty(&lp->lp_txq));
-        LASSERT (lp->lp_txqnob == 0);
+       /* Start the cleanup process */
+       cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
+               lnet_net_lock(LNET_LOCK_EX);
+               lnet_peer_table_cleanup_locked(ni, ptable);
+               lnet_net_unlock(LNET_LOCK_EX);
+       }
 
-       LIBCFS_FREE(lp, sizeof(*lp));
+       /* Wait until all peers have been destroyed. */
+       cfs_percpt_for_each(ptable, i, the_lnet.ln_peer_tables) {
+               lnet_net_lock(LNET_LOCK_EX);
+               lnet_peer_table_finalize_wait_locked(ptable, i);
+               lnet_net_unlock(LNET_LOCK_EX);
+       }
+}
 
-        LNET_LOCK();
+static struct lnet_peer_ni *
+lnet_get_peer_ni_locked(struct lnet_peer_table *ptable, lnet_nid_t nid)
+{
+       struct list_head        *peers;
+       struct lnet_peer_ni     *lp;
+
+       LASSERT(!the_lnet.ln_shutdown);
+
+       peers = &ptable->pt_hash[lnet_nid2peerhash(nid)];
+       list_for_each_entry(lp, peers, lpni_hashlist) {
+               if (lp->lpni_nid == nid) {
+                       lnet_peer_ni_addref_locked(lp);
+                       return lp;
+               }
+       }
 
-        LASSERT(the_lnet.ln_npeers > 0);
-        the_lnet.ln_npeers--;
+       return NULL;
 }
 
-lnet_peer_t *
-lnet_find_peer_locked (lnet_nid_t nid)
+struct lnet_peer_ni *
+lnet_find_peer_ni_locked(lnet_nid_t nid)
 {
-       unsigned int      idx = LNET_NIDADDR(nid) % LNET_PEER_HASHSIZE;
-       struct list_head *peers = &the_lnet.ln_peer_hash[idx];
-       struct list_head *tmp;
-        lnet_peer_t      *lp;
+       struct lnet_peer_ni *lpni;
+       struct lnet_peer_table *ptable;
+       int cpt;
 
-       if (the_lnet.ln_shutdown)
-                return NULL;
+       cpt = lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
 
-       list_for_each (tmp, peers) {
-               lp = list_entry(tmp, lnet_peer_t, lp_hashlist);
-               
-               if (lp->lp_nid == nid) {
-                        lnet_peer_addref_locked(lp);
-                       return lp;
-                }
+       ptable = the_lnet.ln_peer_tables[cpt];
+       lpni = lnet_get_peer_ni_locked(ptable, nid);
+
+       return lpni;
+}
+
+int
+lnet_find_or_create_peer_locked(lnet_nid_t dst_nid, int cpt, struct lnet_peer **peer)
+{
+       struct lnet_peer_ni *lpni;
+
+       lpni = lnet_find_peer_ni_locked(dst_nid);
+       if (!lpni) {
+               int rc;
+               rc = lnet_nid2peerni_locked(&lpni, dst_nid, cpt);
+               if (rc != 0)
+                       return rc;
+       }
+
+       *peer = lpni->lpni_peer_net->lpn_peer;
+       lnet_peer_ni_decref_locked(lpni);
+
+       return 0;
+}
+
+struct lnet_peer_ni *
+lnet_get_peer_ni_idx_locked(int idx, struct lnet_peer_net **lpn,
+                           struct lnet_peer **lp)
+{
+       struct lnet_peer_ni     *lpni;
+
+       list_for_each_entry((*lp), &the_lnet.ln_peers, lp_on_lnet_peer_list) {
+               list_for_each_entry((*lpn), &((*lp)->lp_peer_nets), lpn_on_peer_list) {
+                       list_for_each_entry(lpni, &((*lpn)->lpn_peer_nis),
+                                           lpni_on_peer_net_list)
+                               if (idx-- == 0)
+                                       return lpni;
+               }
+       }
+
+       return NULL;
+}
+
+struct lnet_peer_ni *
+lnet_get_next_peer_ni_locked(struct lnet_peer *peer,
+                            struct lnet_peer_net *peer_net,
+                            struct lnet_peer_ni *prev)
+{
+       struct lnet_peer_ni *lpni;
+       struct lnet_peer_net *net = peer_net;
+
+       if (!prev) {
+               if (!net)
+                       net = list_entry(peer->lp_peer_nets.next,
+                                        struct lnet_peer_net,
+                                        lpn_on_peer_list);
+               lpni = list_entry(net->lpn_peer_nis.next, struct lnet_peer_ni,
+                                 lpni_on_peer_net_list);
+
+               return lpni;
+       }
+
+       if (prev->lpni_on_peer_net_list.next ==
+           &prev->lpni_peer_net->lpn_peer_nis) {
+               /*
+                * if you reached the end of the peer ni list and the peer
+                * net is specified then there are no more peer nis in that
+                * net.
+                */
+               if (net)
+                       return NULL;
+
+               /*
+                * we reached the end of this net ni list. move to the
+                * next net
+                */
+               if (prev->lpni_peer_net->lpn_on_peer_list.next ==
+                   &peer->lp_peer_nets)
+                       /* no more nets and no more NIs. */
+                       return NULL;
+
+               /* get the next net */
+               net = list_entry(prev->lpni_peer_net->lpn_on_peer_list.next,
+                                struct lnet_peer_net,
+                                lpn_on_peer_list);
+               /* get the ni on it */
+               lpni = list_entry(net->lpn_peer_nis.next, struct lnet_peer_ni,
+                                 lpni_on_peer_net_list);
+
+               return lpni;
+       }
+
+       /* there are more nis left */
+       lpni = list_entry(prev->lpni_on_peer_net_list.next,
+                         struct lnet_peer_ni, lpni_on_peer_net_list);
+
+       return lpni;
+}
+
+bool
+lnet_peer_is_ni_pref_locked(struct lnet_peer_ni *lpni, struct lnet_ni *ni)
+{
+       int i;
+
+       for (i = 0; i < lpni->lpni_pref_nnids; i++) {
+               if (lpni->lpni_pref_nids[i] == ni->ni_nid)
+                       return true;
+       }
+       return false;
+}
+
+lnet_nid_t
+lnet_peer_primary_nid(lnet_nid_t nid)
+{
+       struct lnet_peer_ni *lpni;
+       lnet_nid_t primary_nid = nid;
+       int cpt;
+
+       cpt = lnet_net_lock_current();
+       lpni = lnet_find_peer_ni_locked(nid);
+       if (lpni) {
+               primary_nid = lpni->lpni_peer_net->lpn_peer->lp_primary_nid;
+               lnet_peer_ni_decref_locked(lpni);
+       }
+       lnet_net_unlock(cpt);
+
+       return primary_nid;
+}
+
+static void
+lnet_try_destroy_peer_hierarchy_locked(struct lnet_peer_ni *lpni)
+{
+       struct lnet_peer_net *peer_net;
+       struct lnet_peer *peer;
+
+       /* TODO: could the below situation happen? accessing an already
+        * destroyed peer? */
+       if (lpni->lpni_peer_net == NULL ||
+           lpni->lpni_peer_net->lpn_peer == NULL)
+               return;
+
+       peer_net = lpni->lpni_peer_net;
+       peer = lpni->lpni_peer_net->lpn_peer;
+
+       list_del_init(&lpni->lpni_on_peer_net_list);
+       lpni->lpni_peer_net = NULL;
+
+       /* if peer_net is empty, then remove it from the peer */
+       if (list_empty(&peer_net->lpn_peer_nis)) {
+               list_del_init(&peer_net->lpn_on_peer_list);
+               peer_net->lpn_peer = NULL;
+               LIBCFS_FREE(peer_net, sizeof(*peer_net));
+
+               /* if the peer is empty then remove it from the
+                * the_lnet.ln_peers */
+               if (list_empty(&peer->lp_peer_nets)) {
+                       list_del_init(&peer->lp_on_lnet_peer_list);
+                       LIBCFS_FREE(peer, sizeof(*peer));
+               }
+       }
+}
+
+static int
+lnet_build_peer_hierarchy(struct lnet_peer_ni *lpni)
+{
+       struct lnet_peer *peer;
+       struct lnet_peer_net *peer_net;
+       __u32 lpni_net = LNET_NIDNET(lpni->lpni_nid);
+
+       peer = NULL;
+       peer_net = NULL;
+
+       LIBCFS_ALLOC(peer, sizeof(*peer));
+       if (peer == NULL)
+               return -ENOMEM;
+
+       LIBCFS_ALLOC(peer_net, sizeof(*peer_net));
+       if (peer_net == NULL) {
+               LIBCFS_FREE(peer, sizeof(*peer));
+               return -ENOMEM;
+       }
+
+       INIT_LIST_HEAD(&peer->lp_on_lnet_peer_list);
+       INIT_LIST_HEAD(&peer->lp_peer_nets);
+       INIT_LIST_HEAD(&peer_net->lpn_on_peer_list);
+       INIT_LIST_HEAD(&peer_net->lpn_peer_nis);
+
+       /* build the hierarchy */
+       peer_net->lpn_net_id = lpni_net;
+       peer_net->lpn_peer = peer;
+       lpni->lpni_peer_net = peer_net;
+       peer->lp_primary_nid = lpni->lpni_nid;
+       peer->lp_multi_rail = false;
+       list_add_tail(&peer_net->lpn_on_peer_list, &peer->lp_peer_nets);
+       list_add_tail(&lpni->lpni_on_peer_net_list, &peer_net->lpn_peer_nis);
+       list_add_tail(&peer->lp_on_lnet_peer_list, &the_lnet.ln_peers);
+
+       return 0;
+}
+
+struct lnet_peer_net *
+lnet_peer_get_net_locked(struct lnet_peer *peer, __u32 net_id)
+{
+       struct lnet_peer_net *peer_net;
+       list_for_each_entry(peer_net, &peer->lp_peer_nets, lpn_on_peer_list) {
+               if (peer_net->lpn_net_id == net_id)
+                       return peer_net;
        }
-        
        return NULL;
 }
 
+/*
+ * given the key nid find the peer to add the new peer NID to. If the key
+ * nid is NULL, then create a new peer, but first make sure that the NID
+ * is unique
+ */
 int
-lnet_nid2peer_locked(lnet_peer_t **lpp, lnet_nid_t nid)
-{
-       lnet_peer_t    *lp;
-       lnet_peer_t    *lp2;
-
-        lp = lnet_find_peer_locked(nid);
-        if (lp != NULL) {
-                *lpp = lp;
-                return 0;
-        }
-        
-        LNET_UNLOCK();
-       
-       LIBCFS_ALLOC(lp, sizeof(*lp));
-       if (lp == NULL) {
-                *lpp = NULL;
-                LNET_LOCK();
-                return -ENOMEM;
-        }
-
-        memset(lp, 0, sizeof(*lp));             /* zero counters etc */
-        
-       CFS_INIT_LIST_HEAD(&lp->lp_txq);
-        CFS_INIT_LIST_HEAD(&lp->lp_rtrq);
-       
-       lp->lp_alive = !lnet_peers_start_down(); /* 1 bit!! */
-        lp->lp_notify = 0;
-        lp->lp_notifylnd = 0;
-        lp->lp_notifying = 0;
-        lp->lp_alive_count = 0;
-       lp->lp_timestamp = 0;
-        lp->lp_ping_timestamp = 0;
-       lp->lp_nid = nid;
-        lp->lp_refcount = 2;                    /* 1 for caller; 1 for hash */
-        lp->lp_rtr_refcount = 0;
-
-        LNET_LOCK();
-
-        lp2 = lnet_find_peer_locked(nid);
-        if (lp2 != NULL) {
-                LNET_UNLOCK();
-                LIBCFS_FREE(lp, sizeof(*lp));
-                LNET_LOCK();
-
-                if (the_lnet.ln_shutdown) {
-                        lnet_peer_decref_locked(lp2);
-                        *lpp = NULL;
-                        return -ESHUTDOWN;
-                }
-
-                *lpp = lp2;
-                return 0;
-        }
-                
-        lp->lp_ni = lnet_net2ni_locked(LNET_NIDNET(nid));
-        if (lp->lp_ni == NULL) {
-                LNET_UNLOCK();
-                LIBCFS_FREE(lp, sizeof(*lp));
-                LNET_LOCK();
-
-                *lpp = NULL;
-                return the_lnet.ln_shutdown ? -ESHUTDOWN : -EHOSTUNREACH;
-        }
-
-       lp->lp_txcredits = 
-                lp->lp_mintxcredits = lp->lp_ni->ni_peertxcredits;
-
-        /* As a first approximation; allow this peer the same number of router
-         * buffers as it is allowed outstanding sends */
-        lp->lp_rtrcredits = lp->lp_minrtrcredits = lp->lp_txcredits;
-
-        LASSERT (!the_lnet.ln_shutdown);
-        /* can't add peers after shutdown starts */
-
-        list_add_tail(&lp->lp_hashlist, lnet_nid2peerhash(nid));
-        the_lnet.ln_npeers++;
-        the_lnet.ln_peertable_version++;
-        *lpp = lp;
-        return 0;
+lnet_add_peer_ni_to_peer(lnet_nid_t key_nid, lnet_nid_t nid, bool mr)
+{
+       struct lnet_peer_ni *lpni, *lpni2;
+       struct lnet_peer *peer;
+       struct lnet_peer_net *peer_net, *pn;
+       int cpt, cpt2, rc;
+       struct lnet_peer_table *ptable = NULL;
+       __u32 net_id = LNET_NIDNET(nid);
+
+       if (nid == LNET_NID_ANY)
+               return -EINVAL;
+
+       /* check that nid is unique */
+       cpt = lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
+       lnet_net_lock(cpt);
+       lpni = lnet_find_peer_ni_locked(nid);
+       if (lpni != NULL) {
+               lnet_peer_ni_decref_locked(lpni);
+               lnet_net_unlock(cpt);
+               return -EEXIST;
+       }
+       lnet_net_unlock(cpt);
+
+       if (key_nid != LNET_NID_ANY) {
+               cpt2 = lnet_nid_cpt_hash(key_nid, LNET_CPT_NUMBER);
+               lnet_net_lock(cpt2);
+               lpni = lnet_find_peer_ni_locked(key_nid);
+               if (lpni == NULL) {
+                       lnet_net_unlock(cpt2);
+                       /* key_nid refers to a non-existant peer_ni.*/
+                       return -EINVAL;
+               }
+               peer = lpni->lpni_peer_net->lpn_peer;
+               peer->lp_multi_rail = mr;
+               lnet_peer_ni_decref_locked(lpni);
+               lnet_net_unlock(cpt2);
+       } else {
+               lnet_net_lock(LNET_LOCK_EX);
+               rc = lnet_nid2peerni_locked(&lpni, nid, LNET_LOCK_EX);
+               if (rc == 0) {
+                       lpni->lpni_peer_net->lpn_peer->lp_multi_rail = mr;
+                       lnet_peer_ni_decref_locked(lpni);
+               }
+               lnet_net_unlock(LNET_LOCK_EX);
+               return rc;
+       }
+
+       lpni = NULL;
+
+       LIBCFS_CPT_ALLOC(lpni, lnet_cpt_table(), cpt, sizeof(*lpni));
+       if (lpni == NULL)
+               return -ENOMEM;
+
+       INIT_LIST_HEAD(&lpni->lpni_txq);
+       INIT_LIST_HEAD(&lpni->lpni_rtrq);
+       INIT_LIST_HEAD(&lpni->lpni_routes);
+       INIT_LIST_HEAD(&lpni->lpni_hashlist);
+       INIT_LIST_HEAD(&lpni->lpni_on_peer_net_list);
+       INIT_LIST_HEAD(&lpni->lpni_on_remote_peer_ni_list);
+
+       lpni->lpni_alive = !lnet_peers_start_down(); /* 1 bit!! */
+       lpni->lpni_last_alive = cfs_time_current(); /* assumes alive */
+       lpni->lpni_ping_feats = LNET_PING_FEAT_INVAL;
+       lpni->lpni_nid = nid;
+       lpni->lpni_cpt = cpt;
+       lnet_set_peer_ni_health_locked(lpni, true);
+
+       /* allocate here in case we need to add a new peer_net */
+       peer_net = NULL;
+       LIBCFS_ALLOC(peer_net, sizeof(*peer_net));
+       if (peer_net == NULL) {
+               rc = -ENOMEM;
+               if (lpni != NULL)
+                       LIBCFS_FREE(lpni, sizeof(*lpni));
+               return rc;
+       }
+
+       lnet_net_lock(LNET_LOCK_EX);
+
+       ptable = the_lnet.ln_peer_tables[cpt];
+       ptable->pt_number++;
+
+       lpni2 = lnet_find_peer_ni_locked(nid);
+       if (lpni2 != NULL) {
+               lnet_peer_ni_decref_locked(lpni2);
+               /* sanity check that lpni2's peer is what we expect */
+               if (lpni2->lpni_peer_net->lpn_peer != peer)
+                       rc = -EEXIST;
+               else
+                       rc = -EINVAL;
+
+               ptable->pt_number--;
+               /* another thread has already added it */
+               lnet_net_unlock(LNET_LOCK_EX);
+               LIBCFS_FREE(peer_net, sizeof(*peer_net));
+               return rc;
+       }
+
+       lpni->lpni_net = lnet_get_net_locked(LNET_NIDNET(lpni->lpni_nid));
+       if (lpni->lpni_net != NULL) {
+               lpni->lpni_txcredits    =
+               lpni->lpni_mintxcredits =
+                       lpni->lpni_net->net_tunables.lct_peer_tx_credits;
+               lpni->lpni_rtrcredits =
+               lpni->lpni_minrtrcredits = lnet_peer_buffer_credits(lpni->lpni_net);
+       } else {
+               /*
+                * if you're adding a peer which is not on a local network
+                * then we can't assign any of the credits. It won't be
+                * picked for sending anyway. Eventually a network can be
+                * added, in this case we need to revisit this peer and
+                * update its credits.
+                */
+
+               /* increment refcount for remote peer list */
+               atomic_inc(&lpni->lpni_refcount);
+               list_add_tail(&lpni->lpni_on_remote_peer_ni_list,
+                             &the_lnet.ln_remote_peer_ni_list);
+       }
+
+       /* increment refcount for peer on hash list */
+       atomic_inc(&lpni->lpni_refcount);
+
+       list_add_tail(&lpni->lpni_hashlist,
+                     &ptable->pt_hash[lnet_nid2peerhash(nid)]);
+       ptable->pt_version++;
+
+       /* add the lpni to a net */
+       list_for_each_entry(pn, &peer->lp_peer_nets, lpn_on_peer_list) {
+               if (pn->lpn_net_id == net_id) {
+                       list_add_tail(&lpni->lpni_on_peer_net_list,
+                                     &pn->lpn_peer_nis);
+                       lpni->lpni_peer_net = pn;
+                       lnet_net_unlock(LNET_LOCK_EX);
+                       LIBCFS_FREE(peer_net, sizeof(*peer_net));
+                       return 0;
+               }
+       }
+
+       INIT_LIST_HEAD(&peer_net->lpn_on_peer_list);
+       INIT_LIST_HEAD(&peer_net->lpn_peer_nis);
+
+       /* build the hierarchy */
+       peer_net->lpn_net_id = net_id;
+       peer_net->lpn_peer = peer;
+       lpni->lpni_peer_net = peer_net;
+       list_add_tail(&lpni->lpni_on_peer_net_list, &peer_net->lpn_peer_nis);
+       list_add_tail(&peer_net->lpn_on_peer_list, &peer->lp_peer_nets);
+
+       lnet_net_unlock(LNET_LOCK_EX);
+       return 0;
+}
+
+int
+lnet_del_peer_ni_from_peer(lnet_nid_t key_nid, lnet_nid_t nid)
+{
+       int cpt;
+       lnet_nid_t local_nid;
+       struct lnet_peer *peer;
+       struct lnet_peer_ni *lpni, *lpni2;
+       struct lnet_peer_table *ptable = NULL;
+
+       if (key_nid == LNET_NID_ANY)
+               return -EINVAL;
+
+       local_nid = (nid != LNET_NID_ANY) ? nid : key_nid;
+       cpt = lnet_nid_cpt_hash(local_nid, LNET_CPT_NUMBER);
+       lnet_net_lock(LNET_LOCK_EX);
+
+       lpni = lnet_find_peer_ni_locked(local_nid);
+       if (lpni == NULL) {
+               lnet_net_unlock(cpt);
+               return -EINVAL;
+       }
+       lnet_peer_ni_decref_locked(lpni);
+
+       peer = lpni->lpni_peer_net->lpn_peer;
+       LASSERT(peer != NULL);
+
+       if (peer->lp_primary_nid == lpni->lpni_nid) {
+               /*
+                * deleting the primary ni is equivalent to deleting the
+                * entire peer
+                */
+               lpni = NULL;
+               lpni = lnet_get_next_peer_ni_locked(peer, NULL, lpni);
+               while (lpni != NULL) {
+                       lpni2 = lnet_get_next_peer_ni_locked(peer, NULL, lpni);
+                       cpt = lnet_nid_cpt_hash(lpni->lpni_nid,
+                                               LNET_CPT_NUMBER);
+                       lnet_peer_remove_from_remote_list(lpni);
+                       ptable = the_lnet.ln_peer_tables[cpt];
+                       ptable->pt_zombies++;
+                       list_del_init(&lpni->lpni_hashlist);
+                       lnet_peer_ni_decref_locked(lpni);
+                       lpni = lpni2;
+               }
+               lnet_net_unlock(LNET_LOCK_EX);
+
+               return 0;
+       }
+
+       lnet_peer_remove_from_remote_list(lpni);
+       cpt = lnet_nid_cpt_hash(lpni->lpni_nid, LNET_CPT_NUMBER);
+       ptable = the_lnet.ln_peer_tables[cpt];
+       ptable->pt_zombies++;
+       list_del_init(&lpni->lpni_hashlist);
+       lnet_peer_ni_decref_locked(lpni);
+       lnet_net_unlock(LNET_LOCK_EX);
+
+       return 0;
+}
+
+void
+lnet_destroy_peer_ni_locked(struct lnet_peer_ni *lpni)
+{
+       struct lnet_peer_table *ptable;
+
+       LASSERT(atomic_read(&lpni->lpni_refcount) == 0);
+       LASSERT(lpni->lpni_rtr_refcount == 0);
+       LASSERT(list_empty(&lpni->lpni_txq));
+       LASSERT(list_empty(&lpni->lpni_hashlist));
+       LASSERT(lpni->lpni_txqnob == 0);
+       LASSERT(lpni->lpni_peer_net != NULL);
+       LASSERT(lpni->lpni_peer_net->lpn_peer != NULL);
+
+       ptable = the_lnet.ln_peer_tables[lpni->lpni_cpt];
+       LASSERT(ptable->pt_number > 0);
+       ptable->pt_number--;
+
+       lpni->lpni_net = NULL;
+
+       lnet_try_destroy_peer_hierarchy_locked(lpni);
+
+       LIBCFS_FREE(lpni, sizeof(*lpni));
+
+       LASSERT(ptable->pt_zombies > 0);
+       ptable->pt_zombies--;
+}
+
+int
+lnet_nid2peerni_locked(struct lnet_peer_ni **lpnip, lnet_nid_t nid, int cpt)
+{
+       struct lnet_peer_table  *ptable;
+       struct lnet_peer_ni     *lpni = NULL;
+       struct lnet_peer_ni     *lpni2;
+       int                     cpt2;
+       int                     rc = 0;
+
+       *lpnip = NULL;
+       if (the_lnet.ln_shutdown) /* it's shutting down */
+               return -ESHUTDOWN;
+
+       /*
+        * calculate cpt2 with the standard hash function
+        * This cpt2 becomes the slot where we'll find or create the peer.
+        */
+       cpt2 = lnet_nid_cpt_hash(nid, LNET_CPT_NUMBER);
+
+       /*
+        * Any changes to the peer tables happen under exclusive write
+        * lock. Any reads to the peer tables can be done via a standard
+        * CPT read lock.
+        */
+       if (cpt != LNET_LOCK_EX) {
+               lnet_net_unlock(cpt);
+               lnet_net_lock(LNET_LOCK_EX);
+       }
+
+       ptable = the_lnet.ln_peer_tables[cpt2];
+       lpni = lnet_get_peer_ni_locked(ptable, nid);
+       if (lpni != NULL) {
+               *lpnip = lpni;
+               if (cpt != LNET_LOCK_EX) {
+                       lnet_net_unlock(LNET_LOCK_EX);
+                       lnet_net_lock(cpt);
+               }
+               return 0;
+       }
+
+       /*
+        * take extra refcount in case another thread has shutdown LNet
+        * and destroyed locks and peer-table before I finish the allocation
+        */
+       ptable->pt_number++;
+       lnet_net_unlock(LNET_LOCK_EX);
+
+       LIBCFS_CPT_ALLOC(lpni, lnet_cpt_table(), cpt2, sizeof(*lpni));
+
+       if (lpni == NULL) {
+               rc = -ENOMEM;
+               lnet_net_lock(cpt);
+               goto out;
+       }
+
+       INIT_LIST_HEAD(&lpni->lpni_txq);
+       INIT_LIST_HEAD(&lpni->lpni_rtrq);
+       INIT_LIST_HEAD(&lpni->lpni_routes);
+       INIT_LIST_HEAD(&lpni->lpni_hashlist);
+       INIT_LIST_HEAD(&lpni->lpni_on_peer_net_list);
+       INIT_LIST_HEAD(&lpni->lpni_on_remote_peer_ni_list);
+
+       lpni->lpni_alive = !lnet_peers_start_down(); /* 1 bit!! */
+       lpni->lpni_last_alive = cfs_time_current(); /* assumes alive */
+       lpni->lpni_ping_feats = LNET_PING_FEAT_INVAL;
+       lpni->lpni_nid = nid;
+       lpni->lpni_cpt = cpt2;
+       atomic_set(&lpni->lpni_refcount, 2);    /* 1 for caller; 1 for hash */
+
+       rc = lnet_build_peer_hierarchy(lpni);
+       if (rc != 0)
+               goto out;
+
+       lnet_net_lock(LNET_LOCK_EX);
+
+       if (the_lnet.ln_shutdown) {
+               rc = -ESHUTDOWN;
+               goto out;
+       }
+
+       lpni2 = lnet_get_peer_ni_locked(ptable, nid);
+       if (lpni2 != NULL) {
+               *lpnip = lpni2;
+               goto out;
+       }
+
+       lpni->lpni_net = lnet_get_net_locked(LNET_NIDNET(lpni->lpni_nid));
+       if (lpni->lpni_net) {
+               lpni->lpni_txcredits    =
+               lpni->lpni_mintxcredits =
+                       lpni->lpni_net->net_tunables.lct_peer_tx_credits;
+               lpni->lpni_rtrcredits    =
+               lpni->lpni_minrtrcredits =
+                       lnet_peer_buffer_credits(lpni->lpni_net);
+       } else {
+               /*
+                * if you're adding a peer which is not on a local network
+                * then we can't assign any of the credits. It won't be
+                * picked for sending anyway. Eventually a network can be
+                * added, in this case we need to revisit this peer and
+                * update its credits.
+                */
+
+               CDEBUG(D_NET, "peer_ni %s is not directly connected\n",
+                      libcfs_nid2str(nid));
+               /* increment refcount for remote peer list */
+               atomic_inc(&lpni->lpni_refcount);
+               list_add_tail(&lpni->lpni_on_remote_peer_ni_list,
+                             &the_lnet.ln_remote_peer_ni_list);
+       }
+
+       lnet_set_peer_ni_health_locked(lpni, true);
+
+       list_add_tail(&lpni->lpni_hashlist,
+                       &ptable->pt_hash[lnet_nid2peerhash(nid)]);
+       ptable->pt_version++;
+       *lpnip = lpni;
+
+       if (cpt != LNET_LOCK_EX) {
+               lnet_net_unlock(LNET_LOCK_EX);
+               lnet_net_lock(cpt);
+       }
+
+       return 0;
+out:
+       if (lpni != NULL) {
+               lnet_try_destroy_peer_hierarchy_locked(lpni);
+               LIBCFS_FREE(lpni, sizeof(*lpni));
+       }
+       ptable->pt_number--;
+       if (cpt != LNET_LOCK_EX) {
+               lnet_net_unlock(LNET_LOCK_EX);
+               lnet_net_lock(cpt);
+       }
+       return rc;
 }
 
 void
 lnet_debug_peer(lnet_nid_t nid)
 {
-        int          rc;
-        lnet_peer_t *lp;
+       char                    *aliveness = "NA";
+       struct lnet_peer_ni     *lp;
+       int                     rc;
+       int                     cpt;
+
+       cpt = lnet_cpt_of_nid(nid, NULL);
+       lnet_net_lock(cpt);
+
+       rc = lnet_nid2peerni_locked(&lp, nid, cpt);
+       if (rc != 0) {
+               lnet_net_unlock(cpt);
+               CDEBUG(D_WARNING, "No peer %s\n", libcfs_nid2str(nid));
+               return;
+       }
+
+       if (lnet_isrouter(lp) || lnet_peer_aliveness_enabled(lp))
+               aliveness = lp->lpni_alive ? "up" : "down";
+
+       CDEBUG(D_WARNING, "%-24s %4d %5s %5d %5d %5d %5d %5d %ld\n",
+              libcfs_nid2str(lp->lpni_nid), atomic_read(&lp->lpni_refcount),
+              aliveness, lp->lpni_net->net_tunables.lct_peer_tx_credits,
+              lp->lpni_rtrcredits, lp->lpni_minrtrcredits,
+              lp->lpni_txcredits, lp->lpni_mintxcredits, lp->lpni_txqnob);
+
+       lnet_peer_ni_decref_locked(lp);
+
+       lnet_net_unlock(cpt);
+}
+
+int lnet_get_peer_ni_info(__u32 peer_index, __u64 *nid,
+                         char aliveness[LNET_MAX_STR_LEN],
+                         __u32 *cpt_iter, __u32 *refcount,
+                         __u32 *ni_peer_tx_credits, __u32 *peer_tx_credits,
+                         __u32 *peer_rtr_credits, __u32 *peer_min_rtr_credits,
+                         __u32 *peer_tx_qnob)
+{
+       struct lnet_peer_table          *peer_table;
+       struct lnet_peer_ni             *lp;
+       int                             j;
+       int                             lncpt;
+       bool                            found = false;
+
+       /* get the number of CPTs */
+       lncpt = cfs_percpt_number(the_lnet.ln_peer_tables);
+
+       /* if the cpt number to be examined is >= the number of cpts in
+        * the system then indicate that there are no more cpts to examin
+        */
+       if (*cpt_iter >= lncpt)
+               return -ENOENT;
+
+       /* get the current table */
+       peer_table = the_lnet.ln_peer_tables[*cpt_iter];
+       /* if the ptable is NULL then there are no more cpts to examine */
+       if (peer_table == NULL)
+               return -ENOENT;
+
+       lnet_net_lock(*cpt_iter);
+
+       for (j = 0; j < LNET_PEER_HASH_SIZE && !found; j++) {
+               struct list_head *peers = &peer_table->pt_hash[j];
+
+               list_for_each_entry(lp, peers, lpni_hashlist) {
+                       if (peer_index-- > 0)
+                               continue;
+
+                       snprintf(aliveness, LNET_MAX_STR_LEN, "NA");
+                       if (lnet_isrouter(lp) ||
+                               lnet_peer_aliveness_enabled(lp))
+                               snprintf(aliveness, LNET_MAX_STR_LEN,
+                                        lp->lpni_alive ? "up" : "down");
+
+                       *nid = lp->lpni_nid;
+                       *refcount = atomic_read(&lp->lpni_refcount);
+                       *ni_peer_tx_credits =
+                               lp->lpni_net->net_tunables.lct_peer_tx_credits;
+                       *peer_tx_credits = lp->lpni_txcredits;
+                       *peer_rtr_credits = lp->lpni_rtrcredits;
+                       *peer_min_rtr_credits = lp->lpni_mintxcredits;
+                       *peer_tx_qnob = lp->lpni_txqnob;
+
+                       found = true;
+               }
+
+       }
+       lnet_net_unlock(*cpt_iter);
 
-        LNET_LOCK();
-        
-        rc = lnet_nid2peer_locked(&lp, nid);
-        if (rc != 0) {
-                LNET_UNLOCK();
-                CDEBUG(D_WARNING, "No peer %s\n", libcfs_nid2str(nid));
-                return;
-        }
+       *cpt_iter = lncpt;
 
-        CDEBUG(D_WARNING, "%-24s %4d %5s %5d %5d %5d %5d %5d %ld\n",
-               libcfs_nid2str(lp->lp_nid), lp->lp_refcount, 
-               !lnet_isrouter(lp) ? "~rtr" : (lp->lp_alive ? "up" : "down"),
-               lp->lp_ni->ni_peertxcredits, 
-               lp->lp_rtrcredits, lp->lp_minrtrcredits, 
-               lp->lp_txcredits, lp->lp_mintxcredits, lp->lp_txqnob);
+       return found ? 0 : -ENOENT;
+}
 
-        lnet_peer_decref_locked(lp);
+int lnet_get_peer_info(__u32 idx, lnet_nid_t *primary_nid, lnet_nid_t *nid,
+                      struct lnet_peer_ni_credit_info *peer_ni_info)
+{
+       struct lnet_peer_ni *lpni = NULL;
+       struct lnet_peer_net *lpn = NULL;
+       struct lnet_peer *lp = NULL;
+
+       lpni = lnet_get_peer_ni_idx_locked(idx, &lpn, &lp);
+
+       if (!lpni)
+               return -ENOENT;
+
+       *primary_nid = lp->lp_primary_nid;
+       *nid = lpni->lpni_nid;
+       snprintf(peer_ni_info->cr_aliveness, LNET_MAX_STR_LEN, "NA");
+       if (lnet_isrouter(lpni) ||
+               lnet_peer_aliveness_enabled(lpni))
+               snprintf(peer_ni_info->cr_aliveness, LNET_MAX_STR_LEN,
+                        lpni->lpni_alive ? "up" : "down");
+
+       peer_ni_info->cr_refcount = atomic_read(&lpni->lpni_refcount);
+       peer_ni_info->cr_ni_peer_tx_credits = (lpni->lpni_net != NULL) ?
+               lpni->lpni_net->net_tunables.lct_peer_tx_credits : 0;
+       peer_ni_info->cr_peer_tx_credits = lpni->lpni_txcredits;
+       peer_ni_info->cr_peer_rtr_credits = lpni->lpni_rtrcredits;
+       peer_ni_info->cr_peer_min_rtr_credits = lpni->lpni_mintxcredits;
+       peer_ni_info->cr_peer_tx_qnob = lpni->lpni_txqnob;
 
-        LNET_UNLOCK();
+       return 0;
 }