LU-7734 lnet: set primary NID in ptlrpc_connection_get()

[fs/lustre-release.git] / lnet / lnet / peer.c

diff --git a/lnet/lnet/peer.c b/lnet/lnet/peer.c
index 6b19d2d..c73668c 100644 (file)
--- a/lnet/lnet/peer.c
+++ b/lnet/lnet/peer.c
@@ -604,6 +604,31 @@ lnet_peer_primary_nid(lnet_nid_t nid)
        return primary_nid;
 }
 
+lnet_nid_t
+LNetPrimaryNID(lnet_nid_t nid)
+{
+       struct lnet_peer_ni *lpni;
+       lnet_nid_t primary_nid = nid;
+       int rc = 0;
+       int cpt;
+
+       cpt = lnet_net_lock_current();
+       lpni = lnet_nid2peerni_locked(nid, cpt);
+       if (IS_ERR(lpni)) {
+               rc = PTR_ERR(lpni);
+               goto out_unlock;
+       }
+       primary_nid = lpni->lpni_peer_net->lpn_peer->lp_primary_nid;
+       lnet_peer_ni_decref_locked(lpni);
+out_unlock:
+       lnet_net_unlock(cpt);
+
+       CDEBUG(D_NET, "NID %s primary NID %s rc %d\n", libcfs_nid2str(nid),
+              libcfs_nid2str(primary_nid), rc);
+       return primary_nid;
+}
+EXPORT_SYMBOL(LNetPrimaryNID);
+
 struct lnet_peer_net *
 lnet_peer_get_net_locked(struct lnet_peer *peer, __u32 net_id)
 {