lnet/klnds/socklnd/socklnd.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) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2011, 2017, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  *
  31  * lnet/klnds/socklnd/socklnd.c
  32  *
  33  * Author: Zach Brown <zab@zabbo.net>
  34  * Author: Peter J. Braam <braam@clusterfs.com>
  35  * Author: Phil Schwan <phil@clusterfs.com>
  36  * Author: Eric Barton <eric@bartonsoftware.com>
  37  */
  38
  39 #include <linux/ethtool.h>
  40 #include <linux/inetdevice.h>
  41 #include <linux/kernel.h>
  42 #include <linux/sunrpc/addr.h>
  43 #include <net/addrconf.h>
  44 #include "socklnd.h"
  45
  46 static const struct lnet_lnd the_ksocklnd;
  47 struct ksock_nal_data ksocknal_data;
  48
  49 static struct ksock_interface *
  50 ksocknal_index2iface(struct lnet_ni *ni, int index)
  51 {
  52         struct ksock_net *net = ni->ni_data;
  53         struct ksock_interface *iface;
  54
  55         iface = &net->ksnn_interface;
  56
  57         if (iface->ksni_index == index)
  58                 return iface;
  59
  60         return NULL;
  61 }
  62
  63 static int ksocknal_ip2index(struct sockaddr *addr, struct lnet_ni *ni)
  64 {
  65         struct net_device *dev;
  66         int ret = -1;
  67         DECLARE_CONST_IN_IFADDR(ifa);
  68
  69         if (addr->sa_family != AF_INET && addr->sa_family != AF_INET6)
  70                 return ret;
  71
  72         rcu_read_lock();
  73         for_each_netdev(ni->ni_net_ns, dev) {
  74                 int flags = dev_get_flags(dev);
  75                 struct in_device *in_dev;
  76
  77                 if (flags & IFF_LOOPBACK) /* skip the loopback IF */
  78                         continue;
  79
  80                 if (!(flags & IFF_UP))
  81                         continue;
  82
  83                 switch (addr->sa_family) {
  84                 case AF_INET:
  85                         in_dev = __in_dev_get_rcu(dev);
  86                         if (!in_dev)
  87                                 continue;
  88
  89                         in_dev_for_each_ifa_rcu(ifa, in_dev) {
  90                                 if (ifa->ifa_local ==
  91                                     ((struct sockaddr_in *)addr)->sin_addr.s_addr)
  92                                         ret = dev->ifindex;
  93                         }
  94                         endfor_ifa(in_dev);
  95                         break;
  96 #if IS_ENABLED(CONFIG_IPV6)
  97                 case AF_INET6: {
  98                         struct inet6_dev *in6_dev;
  99                         const struct inet6_ifaddr *ifa6;
 100                         struct sockaddr_in6 *addr6 = (struct sockaddr_in6*)addr;
 101
 102                         in6_dev = __in6_dev_get(dev);
 103                         if (!in6_dev)
 104                                 continue;
 105
 106                         list_for_each_entry_rcu(ifa6, &in6_dev->addr_list, if_list) {
 107                                 if (ipv6_addr_cmp(&ifa6->addr,
 108                                                  &addr6->sin6_addr) == 0)
 109                                         ret = dev->ifindex;
 110                         }
 111                         break;
 112                         }
 113 #endif /* IS_ENABLED(CONFIG_IPV6) */
 114                 }
 115                 if (ret >= 0)
 116                         break;
 117         }
 118         rcu_read_unlock();
 119
 120         return ret;
 121 }
 122
 123 static struct ksock_conn_cb *
 124 ksocknal_create_conn_cb(struct sockaddr *addr)
 125 {
 126         struct ksock_conn_cb *conn_cb;
 127
 128         LIBCFS_ALLOC(conn_cb, sizeof(*conn_cb));
 129         if (!conn_cb)
 130                 return NULL;
 131
 132         refcount_set(&conn_cb->ksnr_refcount, 1);
 133         conn_cb->ksnr_peer = NULL;
 134         conn_cb->ksnr_retry_interval = 0;         /* OK to connect at any time */
 135         rpc_copy_addr((struct sockaddr *)&conn_cb->ksnr_addr, addr);
 136         rpc_set_port((struct sockaddr *)&conn_cb->ksnr_addr,
 137                      rpc_get_port(addr));
 138         conn_cb->ksnr_myiface = -1;
 139         conn_cb->ksnr_scheduled = 0;
 140         conn_cb->ksnr_connecting = 0;
 141         conn_cb->ksnr_connected = 0;
 142         conn_cb->ksnr_deleted = 0;
 143         conn_cb->ksnr_conn_count = 0;
 144         conn_cb->ksnr_ctrl_conn_count = 0;
 145         conn_cb->ksnr_blki_conn_count = 0;
 146         conn_cb->ksnr_blko_conn_count = 0;
 147         conn_cb->ksnr_max_conns = 0;
 148         conn_cb->ksnr_busy_retry_count = 0;
 149
 150         return conn_cb;
 151 }
 152
 153 void
 154 ksocknal_destroy_conn_cb(struct ksock_conn_cb *conn_cb)
 155 {
 156         LASSERT(refcount_read(&conn_cb->ksnr_refcount) == 0);
 157
 158         if (conn_cb->ksnr_peer)
 159                 ksocknal_peer_decref(conn_cb->ksnr_peer);
 160
 161         LIBCFS_FREE(conn_cb, sizeof(*conn_cb));
 162 }
 163
 164 static struct ksock_peer_ni *
 165 ksocknal_create_peer(struct lnet_ni *ni, struct lnet_processid *id)
 166 {
 167         int cpt = lnet_nid2cpt(&id->nid, ni);
 168         struct ksock_net *net = ni->ni_data;
 169         struct ksock_peer_ni *peer_ni;
 170
 171         LASSERT(!LNET_NID_IS_ANY(&id->nid));
 172         LASSERT(id->pid != LNET_PID_ANY);
 173         LASSERT(!in_interrupt());
 174
 175         if (!atomic_inc_unless_negative(&net->ksnn_npeers)) {
 176                 CERROR("Can't create peer_ni: network shutdown\n");
 177                 return ERR_PTR(-ESHUTDOWN);
 178         }
 179
 180         LIBCFS_CPT_ALLOC(peer_ni, lnet_cpt_table(), cpt, sizeof(*peer_ni));
 181         if (!peer_ni) {
 182                 atomic_dec(&net->ksnn_npeers);
 183                 return ERR_PTR(-ENOMEM);
 184         }
 185
 186         peer_ni->ksnp_ni = ni;
 187         peer_ni->ksnp_id = *id;
 188         refcount_set(&peer_ni->ksnp_refcount, 1); /* 1 ref for caller */
 189         peer_ni->ksnp_closing = 0;
 190         peer_ni->ksnp_accepting = 0;
 191         peer_ni->ksnp_proto = NULL;
 192         peer_ni->ksnp_last_alive = 0;
 193         peer_ni->ksnp_zc_next_cookie = SOCKNAL_KEEPALIVE_PING + 1;
 194         peer_ni->ksnp_conn_cb = NULL;
 195
 196         INIT_LIST_HEAD(&peer_ni->ksnp_conns);
 197         INIT_LIST_HEAD(&peer_ni->ksnp_tx_queue);
 198         INIT_LIST_HEAD(&peer_ni->ksnp_zc_req_list);
 199         spin_lock_init(&peer_ni->ksnp_lock);
 200
 201         return peer_ni;
 202 }
 203
 204 void
 205 ksocknal_destroy_peer(struct ksock_peer_ni *peer_ni)
 206 {
 207         struct ksock_net *net = peer_ni->ksnp_ni->ni_data;
 208
 209         CDEBUG (D_NET, "peer_ni %s %p deleted\n",
 210                 libcfs_idstr(&peer_ni->ksnp_id), peer_ni);
 211
 212         LASSERT(refcount_read(&peer_ni->ksnp_refcount) == 0);
 213         LASSERT(peer_ni->ksnp_accepting == 0);
 214         LASSERT(list_empty(&peer_ni->ksnp_conns));
 215         LASSERT(peer_ni->ksnp_conn_cb == NULL);
 216         LASSERT(list_empty(&peer_ni->ksnp_tx_queue));
 217         LASSERT(list_empty(&peer_ni->ksnp_zc_req_list));
 218
 219         LIBCFS_FREE(peer_ni, sizeof(*peer_ni));
 220
 221         /* NB a peer_ni's connections and conn_cb keep a reference on their
 222          * peer_ni until they are destroyed, so we can be assured that _all_
 223          * state to do with this peer_ni has been cleaned up when its refcount
 224          * drops to zero.
 225          */
 226         if (atomic_dec_and_test(&net->ksnn_npeers))
 227                 wake_up_var(&net->ksnn_npeers);
 228 }
 229
 230 struct ksock_peer_ni *
 231 ksocknal_find_peer_locked(struct lnet_ni *ni, struct lnet_processid *id)
 232 {
 233         struct ksock_peer_ni *peer_ni;
 234         unsigned long hash = nidhash(&id->nid);
 235
 236         hash_for_each_possible(ksocknal_data.ksnd_peers, peer_ni,
 237                                ksnp_list, hash) {
 238                 LASSERT(!peer_ni->ksnp_closing);
 239
 240                 if (peer_ni->ksnp_ni != ni)
 241                         continue;
 242
 243                 if (!nid_same(&peer_ni->ksnp_id.nid, &id->nid) ||
 244                     peer_ni->ksnp_id.pid != id->pid)
 245                         continue;
 246
 247                 CDEBUG(D_NET, "got peer_ni [%p] -> %s (%d)\n",
 248                        peer_ni, libcfs_idstr(id),
 249                        refcount_read(&peer_ni->ksnp_refcount));
 250                 return peer_ni;
 251         }
 252         return NULL;
 253 }
 254
 255 struct ksock_peer_ni *
 256 ksocknal_find_peer(struct lnet_ni *ni, struct lnet_processid *id)
 257 {
 258         struct ksock_peer_ni *peer_ni;
 259
 260         read_lock(&ksocknal_data.ksnd_global_lock);
 261         peer_ni = ksocknal_find_peer_locked(ni, id);
 262         if (peer_ni != NULL)                    /* +1 ref for caller? */
 263                 ksocknal_peer_addref(peer_ni);
 264         read_unlock(&ksocknal_data.ksnd_global_lock);
 265
 266         return peer_ni;
 267 }
 268
 269 static void
 270 ksocknal_unlink_peer_locked(struct ksock_peer_ni *peer_ni)
 271 {
 272         LASSERT(list_empty(&peer_ni->ksnp_conns));
 273         LASSERT(peer_ni->ksnp_conn_cb == NULL);
 274         LASSERT(!peer_ni->ksnp_closing);
 275         peer_ni->ksnp_closing = 1;
 276         hlist_del(&peer_ni->ksnp_list);
 277         /* lose peerlist's ref */
 278         ksocknal_peer_decref(peer_ni);
 279 }
 280
 281
 282 static void
 283 ksocknal_dump_peer_debug_info(struct ksock_peer_ni *peer_ni)
 284 {
 285         struct ksock_conn *conn;
 286         struct list_head *ctmp;
 287         struct list_head *txtmp;
 288         int ccount = 0;
 289         int txcount = 0;
 290
 291         list_for_each(ctmp, &peer_ni->ksnp_conns) {
 292                 conn = list_entry(ctmp, struct ksock_conn, ksnc_list);
 293
 294                 if (!list_empty(&conn->ksnc_tx_queue))
 295                         list_for_each(txtmp, &conn->ksnc_tx_queue) txcount++;
 296
 297                 CDEBUG(D_CONSOLE, "Conn %d [type, closing, crefcnt, srefcnt]: %d, %d, %d, %d\n",
 298                        ccount,
 299                        conn->ksnc_type,
 300                        conn->ksnc_closing,
 301                        refcount_read(&conn->ksnc_conn_refcount),
 302                        refcount_read(&conn->ksnc_sock_refcount));
 303                 CDEBUG(D_CONSOLE, "Conn %d rx [scheduled, ready, state]: %d, %d, %d\n",
 304                        ccount,
 305                        conn->ksnc_rx_scheduled,
 306                        conn->ksnc_rx_ready,
 307                        conn->ksnc_rx_state);
 308                 CDEBUG(D_CONSOLE, "Conn %d tx [txqcnt, scheduled, last_post, ready, deadline]: %d, %d, %lld, %d, %lld\n",
 309                        ccount,
 310                        txcount,
 311                        conn->ksnc_tx_scheduled,
 312                        conn->ksnc_tx_last_post,
 313                        conn->ksnc_rx_ready,
 314                        conn->ksnc_rx_deadline);
 315
 316                 if (conn->ksnc_scheduler)
 317                         CDEBUG(D_CONSOLE, "Conn %d sched [nconns, cpt]: %d, %d\n",
 318                                ccount,
 319                                conn->ksnc_scheduler->kss_nconns,
 320                                conn->ksnc_scheduler->kss_cpt);
 321
 322                 txcount = 0;
 323                 ccount++;
 324         }
 325 }
 326
 327 static int
 328 ksocknal_get_peer_info(struct lnet_ni *ni, int index,
 329                        struct lnet_processid *id, __u32 *myip, __u32 *peer_ip,
 330                        int *port, int *conn_count, int *share_count)
 331 {
 332         struct ksock_peer_ni *peer_ni;
 333         struct ksock_conn_cb *conn_cb;
 334         int i;
 335         int rc = -ENOENT;
 336
 337         read_lock(&ksocknal_data.ksnd_global_lock);
 338
 339         hash_for_each(ksocknal_data.ksnd_peers, i, peer_ni, ksnp_list) {
 340
 341                 if (peer_ni->ksnp_ni != ni)
 342                         continue;
 343                 if (index-- > 0)
 344                         continue;
 345
 346                 *id = peer_ni->ksnp_id;
 347                 conn_cb = peer_ni->ksnp_conn_cb;
 348                 if (conn_cb == NULL) {
 349                         *myip = 0;
 350                         *peer_ip = 0;
 351                         *port = 0;
 352                         *conn_count = 0;
 353                         *share_count = 0;
 354                         rc = 0;
 355                 } else {
 356                         ksocknal_dump_peer_debug_info(peer_ni);
 357
 358                         if (conn_cb->ksnr_addr.ss_family == AF_INET) {
 359                                 struct sockaddr_in *sa =
 360                                         (void *)&conn_cb->ksnr_addr;
 361
 362                                 rc = choose_ipv4_src(myip,
 363                                                      conn_cb->ksnr_myiface,
 364                                                      ntohl(sa->sin_addr.s_addr),
 365                                                      ni->ni_net_ns);
 366                                 *peer_ip = ntohl(sa->sin_addr.s_addr);
 367                                 *port = ntohs(sa->sin_port);
 368
 369                         } else {
 370                                 *myip = 0xFFFFFFFF;
 371                                 *peer_ip = 0xFFFFFFFF;
 372                                 *port = 0;
 373                                 rc = -ENOTSUPP;
 374                         }
 375                         *conn_count = conn_cb->ksnr_conn_count;
 376                         *share_count = 1;
 377                 }
 378                 break;
 379         }
 380         read_unlock(&ksocknal_data.ksnd_global_lock);
 381         return rc;
 382 }
 383
 384 static unsigned int
 385 ksocknal_get_conn_count_by_type(struct ksock_conn_cb *conn_cb,
 386                                 int type)
 387 {
 388         unsigned int count = 0;
 389
 390         switch (type) {
 391         case SOCKLND_CONN_CONTROL:
 392                 count = conn_cb->ksnr_ctrl_conn_count;
 393                 break;
 394         case SOCKLND_CONN_BULK_IN:
 395                 count = conn_cb->ksnr_blki_conn_count;
 396                 break;
 397         case SOCKLND_CONN_BULK_OUT:
 398                 count = conn_cb->ksnr_blko_conn_count;
 399                 break;
 400         case SOCKLND_CONN_ANY:
 401                 count = conn_cb->ksnr_conn_count;
 402                 break;
 403         default:
 404                 LBUG();
 405                 break;
 406         }
 407
 408         return count;
 409 }
 410
 411 static unsigned int
 412 ksocknal_get_conns_per_peer(struct ksock_peer_ni *peer_ni)
 413 {
 414         struct lnet_ni *ni = peer_ni->ksnp_ni;
 415         struct lnet_ioctl_config_socklnd_tunables *tunables;
 416
 417         LASSERT(ni);
 418
 419         tunables = &ni->ni_lnd_tunables.lnd_tun_u.lnd_sock;
 420
 421         return tunables->lnd_conns_per_peer;
 422 }
 423
 424 static void
 425 ksocknal_incr_conn_count(struct ksock_conn_cb *conn_cb,
 426                          int type)
 427 {
 428         conn_cb->ksnr_conn_count++;
 429
 430         /* check if all connections of the given type got created */
 431         switch (type) {
 432         case SOCKLND_CONN_CONTROL:
 433                 conn_cb->ksnr_ctrl_conn_count++;
 434                 /* there's a single control connection per peer,
 435                  * two in case of loopback
 436                  */
 437                 conn_cb->ksnr_connected |= BIT(type);
 438                 break;
 439         case SOCKLND_CONN_BULK_IN:
 440                 conn_cb->ksnr_blki_conn_count++;
 441                 if (conn_cb->ksnr_blki_conn_count >= conn_cb->ksnr_max_conns)
 442                         conn_cb->ksnr_connected |= BIT(type);
 443                 break;
 444         case SOCKLND_CONN_BULK_OUT:
 445                 conn_cb->ksnr_blko_conn_count++;
 446                 if (conn_cb->ksnr_blko_conn_count >= conn_cb->ksnr_max_conns)
 447                         conn_cb->ksnr_connected |= BIT(type);
 448                 break;
 449         case SOCKLND_CONN_ANY:
 450                 if (conn_cb->ksnr_conn_count >= conn_cb->ksnr_max_conns)
 451                         conn_cb->ksnr_connected |= BIT(type);
 452                 break;
 453         default:
 454                 LBUG();
 455                 break;
 456         }
 457
 458         CDEBUG(D_NET, "Add conn type %d, ksnr_connected %x ksnr_max_conns %d\n",
 459                type, conn_cb->ksnr_connected, conn_cb->ksnr_max_conns);
 460 }
 461
 462
 463 static void
 464 ksocknal_decr_conn_count(struct ksock_conn_cb *conn_cb,
 465                          int type)
 466 {
 467         conn_cb->ksnr_conn_count--;
 468
 469         /* check if all connections of the given type got created */
 470         switch (type) {
 471         case SOCKLND_CONN_CONTROL:
 472                 conn_cb->ksnr_ctrl_conn_count--;
 473                 /* there's a single control connection per peer,
 474                  * two in case of loopback
 475                  */
 476                 if (conn_cb->ksnr_ctrl_conn_count == 0)
 477                         conn_cb->ksnr_connected &= ~BIT(type);
 478                 break;
 479         case SOCKLND_CONN_BULK_IN:
 480                 conn_cb->ksnr_blki_conn_count--;
 481                 if (conn_cb->ksnr_blki_conn_count < conn_cb->ksnr_max_conns)
 482                         conn_cb->ksnr_connected &= ~BIT(type);
 483                 break;
 484         case SOCKLND_CONN_BULK_OUT:
 485                 conn_cb->ksnr_blko_conn_count--;
 486                 if (conn_cb->ksnr_blko_conn_count < conn_cb->ksnr_max_conns)
 487                         conn_cb->ksnr_connected &= ~BIT(type);
 488                 break;
 489         case SOCKLND_CONN_ANY:
 490                 if (conn_cb->ksnr_conn_count < conn_cb->ksnr_max_conns)
 491                         conn_cb->ksnr_connected &= ~BIT(type);
 492                 break;
 493         default:
 494                 LBUG();
 495                 break;
 496         }
 497
 498         CDEBUG(D_NET, "Del conn type %d, ksnr_connected %x ksnr_max_conns %d\n",
 499                type, conn_cb->ksnr_connected, conn_cb->ksnr_max_conns);
 500 }
 501
 502 static void
 503 ksocknal_associate_cb_conn_locked(struct ksock_conn_cb *conn_cb,
 504                                   struct ksock_conn *conn)
 505 {
 506         struct ksock_peer_ni *peer_ni = conn_cb->ksnr_peer;
 507         int type = conn->ksnc_type;
 508         struct ksock_interface *iface;
 509         int conn_iface;
 510
 511         conn_iface = ksocknal_ip2index((struct sockaddr *)&conn->ksnc_myaddr,
 512                                        peer_ni->ksnp_ni);
 513         conn->ksnc_conn_cb = conn_cb;
 514         ksocknal_conn_cb_addref(conn_cb);
 515
 516         if (conn_cb->ksnr_myiface != conn_iface) {
 517                 if (conn_cb->ksnr_myiface < 0) {
 518                         /* route wasn't bound locally yet (the initial route) */
 519                         CDEBUG(D_NET, "Binding %s %pISc to interface %d\n",
 520                                libcfs_idstr(&peer_ni->ksnp_id),
 521                                &conn_cb->ksnr_addr,
 522                                conn_iface);
 523                 } else {
 524                         CDEBUG(D_NET,
 525                                "Rebinding %s %pISc from interface %d to %d\n",
 526                                libcfs_idstr(&peer_ni->ksnp_id),
 527                                &conn_cb->ksnr_addr,
 528                                conn_cb->ksnr_myiface,
 529                                conn_iface);
 530
 531                         iface = ksocknal_index2iface(peer_ni->ksnp_ni,
 532                                                      conn_cb->ksnr_myiface);
 533                         if (iface)
 534                                 iface->ksni_nroutes--;
 535                 }
 536                 conn_cb->ksnr_myiface = conn_iface;
 537                 iface = ksocknal_index2iface(peer_ni->ksnp_ni,
 538                                              conn_cb->ksnr_myiface);
 539                 if (iface)
 540                         iface->ksni_nroutes++;
 541         }
 542
 543         ksocknal_incr_conn_count(conn_cb, type);
 544
 545         /* Successful connection => further attempts can
 546          * proceed immediately
 547          */
 548         conn_cb->ksnr_retry_interval = 0;
 549 }
 550
 551 static void
 552 ksocknal_add_conn_cb_locked(struct ksock_peer_ni *peer_ni,
 553                             struct ksock_conn_cb *conn_cb)
 554 {
 555         struct ksock_conn *conn;
 556         struct ksock_net *net = peer_ni->ksnp_ni->ni_data;
 557
 558         LASSERT(!peer_ni->ksnp_closing);
 559         LASSERT(!conn_cb->ksnr_peer);
 560         LASSERT(!conn_cb->ksnr_scheduled);
 561         LASSERT(!conn_cb->ksnr_connecting);
 562         LASSERT(conn_cb->ksnr_connected == 0);
 563
 564         conn_cb->ksnr_peer = peer_ni;
 565         ksocknal_peer_addref(peer_ni);
 566
 567         /* set the conn_cb's interface to the current net's interface */
 568         conn_cb->ksnr_myiface = net->ksnn_interface.ksni_index;
 569         net->ksnn_interface.ksni_nroutes++;
 570
 571         /* peer_ni's route list takes over my ref on 'route' */
 572         peer_ni->ksnp_conn_cb = conn_cb;
 573
 574         list_for_each_entry(conn, &peer_ni->ksnp_conns, ksnc_list) {
 575                 if (!rpc_cmp_addr((struct sockaddr *)&conn->ksnc_peeraddr,
 576                                   (struct sockaddr *)&conn_cb->ksnr_addr))
 577                         continue;
 578
 579                 ksocknal_associate_cb_conn_locked(conn_cb, conn);
 580                 /* keep going (typed conns) */
 581         }
 582 }
 583
 584 static void
 585 ksocknal_del_conn_cb_locked(struct ksock_conn_cb *conn_cb)
 586 {
 587         struct ksock_peer_ni *peer_ni = conn_cb->ksnr_peer;
 588         struct ksock_interface *iface;
 589         struct ksock_conn *conn;
 590         struct ksock_conn *cnxt;
 591
 592         LASSERT(!conn_cb->ksnr_deleted);
 593
 594         /* Close associated conns */
 595         list_for_each_entry_safe(conn, cnxt, &peer_ni->ksnp_conns, ksnc_list) {
 596                 if (conn->ksnc_conn_cb != conn_cb)
 597                         continue;
 598
 599                 ksocknal_close_conn_locked(conn, 0);
 600         }
 601
 602         if (conn_cb->ksnr_myiface >= 0) {
 603                 iface = ksocknal_index2iface(peer_ni->ksnp_ni,
 604                                              conn_cb->ksnr_myiface);
 605                 if (iface)
 606                         iface->ksni_nroutes--;
 607         }
 608
 609         conn_cb->ksnr_deleted = 1;
 610         ksocknal_conn_cb_decref(conn_cb);               /* drop peer_ni's ref */
 611         peer_ni->ksnp_conn_cb = NULL;
 612
 613         if (list_empty(&peer_ni->ksnp_conns)) {
 614                 /* I've just removed the last route to a peer_ni with no active
 615                  * connections
 616                  */
 617                 ksocknal_unlink_peer_locked(peer_ni);
 618         }
 619 }
 620
 621 int
 622 ksocknal_add_peer(struct lnet_ni *ni, struct lnet_processid *id,
 623                   struct sockaddr *addr)
 624 {
 625         struct ksock_peer_ni *peer_ni;
 626         struct ksock_peer_ni *peer2;
 627         struct ksock_conn_cb *conn_cb;
 628
 629         if (LNET_NID_IS_ANY(&id->nid) ||
 630             id->pid == LNET_PID_ANY)
 631                 return (-EINVAL);
 632
 633         /* Have a brand new peer_ni ready... */
 634         peer_ni = ksocknal_create_peer(ni, id);
 635         if (IS_ERR(peer_ni))
 636                 return PTR_ERR(peer_ni);
 637
 638         conn_cb = ksocknal_create_conn_cb(addr);
 639         if (!conn_cb) {
 640                 ksocknal_peer_decref(peer_ni);
 641                 return -ENOMEM;
 642         }
 643
 644         write_lock_bh(&ksocknal_data.ksnd_global_lock);
 645
 646         /* always called with a ref on ni, so shutdown can't have started */
 647         LASSERT(atomic_read(&((struct ksock_net *)ni->ni_data)->ksnn_npeers)
 648                 >= 0);
 649
 650         peer2 = ksocknal_find_peer_locked(ni, id);
 651         if (peer2 != NULL) {
 652                 ksocknal_peer_decref(peer_ni);
 653                 peer_ni = peer2;
 654         } else {
 655                 /* peer_ni table takes my ref on peer_ni */
 656                 hash_add(ksocknal_data.ksnd_peers, &peer_ni->ksnp_list,
 657                          nidhash(&id->nid));
 658         }
 659
 660         if (peer_ni->ksnp_conn_cb) {
 661                 ksocknal_conn_cb_decref(conn_cb);
 662         } else {
 663                 ksocknal_add_conn_cb_locked(peer_ni, conn_cb);
 664                 /* Remember conns_per_peer setting at the time
 665                  * of connection initiation. It will define the
 666                  * max number of conns per type for this conn_cb
 667                  * while it's in use.
 668                  */
 669                 conn_cb->ksnr_max_conns = ksocknal_get_conns_per_peer(peer_ni);
 670         }
 671
 672         write_unlock_bh(&ksocknal_data.ksnd_global_lock);
 673
 674         return 0;
 675 }
 676
 677 static void
 678 ksocknal_del_peer_locked(struct ksock_peer_ni *peer_ni)
 679 {
 680         struct ksock_conn *conn;
 681         struct ksock_conn *cnxt;
 682         struct ksock_conn_cb *conn_cb;
 683
 684         LASSERT(!peer_ni->ksnp_closing);
 685
 686         /* Extra ref prevents peer_ni disappearing until I'm done with it */
 687         ksocknal_peer_addref(peer_ni);
 688         conn_cb = peer_ni->ksnp_conn_cb;
 689         if (conn_cb)
 690                 ksocknal_del_conn_cb_locked(conn_cb);
 691
 692         list_for_each_entry_safe(conn, cnxt, &peer_ni->ksnp_conns,
 693                                  ksnc_list)
 694                 ksocknal_close_conn_locked(conn, 0);
 695
 696         ksocknal_peer_decref(peer_ni);
 697         /* NB peer_ni unlinks itself when last conn/conn_cb is removed */
 698 }
 699
 700 static int
 701 ksocknal_del_peer(struct lnet_ni *ni, struct lnet_processid *id)
 702 {
 703         LIST_HEAD(zombies);
 704         struct hlist_node *pnxt;
 705         struct ksock_peer_ni *peer_ni;
 706         int lo;
 707         int hi;
 708         int i;
 709         int rc = -ENOENT;
 710
 711         write_lock_bh(&ksocknal_data.ksnd_global_lock);
 712
 713         if (id && !LNET_NID_IS_ANY(&id->nid)) {
 714                 lo = hash_min(nidhash(&id->nid),
 715                               HASH_BITS(ksocknal_data.ksnd_peers));
 716                 hi = lo;
 717         } else {
 718                 lo = 0;
 719                 hi = HASH_SIZE(ksocknal_data.ksnd_peers) - 1;
 720         }
 721
 722         for (i = lo; i <= hi; i++) {
 723                 hlist_for_each_entry_safe(peer_ni, pnxt,
 724                                           &ksocknal_data.ksnd_peers[i],
 725                                           ksnp_list) {
 726                         if (peer_ni->ksnp_ni != ni)
 727                                 continue;
 728
 729                         if (!((!id || LNET_NID_IS_ANY(&id->nid) ||
 730                                nid_same(&peer_ni->ksnp_id.nid, &id->nid)) &&
 731                               (!id || id->pid == LNET_PID_ANY ||
 732                                peer_ni->ksnp_id.pid == id->pid)))
 733                                 continue;
 734
 735                         ksocknal_peer_addref(peer_ni);  /* a ref for me... */
 736
 737                         ksocknal_del_peer_locked(peer_ni);
 738
 739                         if (peer_ni->ksnp_closing &&
 740                             !list_empty(&peer_ni->ksnp_tx_queue)) {
 741                                 LASSERT(list_empty(&peer_ni->ksnp_conns));
 742                                 LASSERT(peer_ni->ksnp_conn_cb == NULL);
 743
 744                                 list_splice_init(&peer_ni->ksnp_tx_queue,
 745                                                  &zombies);
 746                         }
 747
 748                         ksocknal_peer_decref(peer_ni);  /* ...till here */
 749
 750                         rc = 0;                         /* matched! */
 751                 }
 752         }
 753
 754         write_unlock_bh(&ksocknal_data.ksnd_global_lock);
 755
 756         ksocknal_txlist_done(ni, &zombies, -ENETDOWN);
 757
 758         return rc;
 759 }
 760
 761 static struct ksock_conn *
 762 ksocknal_get_conn_by_idx(struct lnet_ni *ni, int index)
 763 {
 764         struct ksock_peer_ni *peer_ni;
 765         struct ksock_conn *conn;
 766         int i;
 767
 768         read_lock(&ksocknal_data.ksnd_global_lock);
 769
 770         hash_for_each(ksocknal_data.ksnd_peers, i, peer_ni, ksnp_list) {
 771                 LASSERT(!peer_ni->ksnp_closing);
 772
 773                 if (peer_ni->ksnp_ni != ni)
 774                         continue;
 775
 776                 list_for_each_entry(conn, &peer_ni->ksnp_conns,
 777                                     ksnc_list) {
 778                         if (index-- > 0)
 779                                 continue;
 780
 781                         ksocknal_conn_addref(conn);
 782                         read_unlock(&ksocknal_data.ksnd_global_lock);
 783                         return conn;
 784                 }
 785         }
 786
 787         read_unlock(&ksocknal_data.ksnd_global_lock);
 788         return NULL;
 789 }
 790
 791 static struct ksock_sched *
 792 ksocknal_choose_scheduler_locked(unsigned int cpt)
 793 {
 794         struct ksock_sched *sched = ksocknal_data.ksnd_schedulers[cpt];
 795         int i;
 796
 797         if (sched->kss_nthreads == 0) {
 798                 cfs_percpt_for_each(sched, i, ksocknal_data.ksnd_schedulers) {
 799                         if (sched->kss_nthreads > 0) {
 800                                 CDEBUG(D_NET, "scheduler[%d] has no threads. selected scheduler[%d]\n",
 801                                        cpt, sched->kss_cpt);
 802                                 return sched;
 803                         }
 804                 }
 805                 return NULL;
 806         }
 807
 808         return sched;
 809 }
 810
 811 int
 812 ksocknal_accept(struct lnet_ni *ni, struct socket *sock)
 813 {
 814         struct ksock_connreq *cr;
 815         int rc;
 816         struct sockaddr_storage peer;
 817
 818         rc = lnet_sock_getaddr(sock, true, &peer);
 819         if (rc != 0) {
 820                 CERROR("Can't determine new connection's address\n");
 821                 return rc;
 822         }
 823
 824         LIBCFS_ALLOC(cr, sizeof(*cr));
 825         if (cr == NULL) {
 826                 LCONSOLE_ERROR_MSG(0x12f,
 827                                    "Dropping connection request from %pISc: memory exhausted\n",
 828                                    &peer);
 829                 return -ENOMEM;
 830         }
 831
 832         lnet_ni_addref(ni);
 833         cr->ksncr_ni   = ni;
 834         cr->ksncr_sock = sock;
 835
 836         spin_lock_bh(&ksocknal_data.ksnd_connd_lock);
 837
 838         list_add_tail(&cr->ksncr_list, &ksocknal_data.ksnd_connd_connreqs);
 839         wake_up(&ksocknal_data.ksnd_connd_waitq);
 840
 841         spin_unlock_bh(&ksocknal_data.ksnd_connd_lock);
 842         return 0;
 843 }
 844
 845 static const struct ln_key_list ksocknal_tunables_keys = {
 846         .lkl_maxattr                    = LNET_NET_SOCKLND_TUNABLES_ATTR_MAX,
 847         .lkl_list                       = {
 848                 [LNET_NET_SOCKLND_TUNABLES_ATTR_CONNS_PER_PEER]  = {
 849                         .lkp_value      = "conns_per_peer",
 850                         .lkp_data_type  = NLA_S32
 851                 },
 852         },
 853 };
 854
 855 static int
 856 ksocknal_nl_set(int cmd, struct nlattr *attr, int type, void *data)
 857 {
 858         struct lnet_lnd_tunables *tunables = data;
 859         s64 num;
 860
 861         if (cmd != LNET_CMD_NETS)
 862                 return -EOPNOTSUPP;
 863
 864         if (type != LNET_NET_SOCKLND_TUNABLES_ATTR_CONNS_PER_PEER ||
 865             nla_type(attr) != LN_SCALAR_ATTR_INT_VALUE)
 866                 return -EINVAL;
 867
 868         num = nla_get_s64(attr);
 869         clamp_t(s64, num, 1, 127);
 870         tunables->lnd_tun_u.lnd_sock.lnd_conns_per_peer = num;
 871
 872         return 0;
 873 }
 874
 875 static int
 876 ksocknal_connecting(struct ksock_conn_cb *conn_cb, struct sockaddr *sa)
 877 {
 878         if (conn_cb &&
 879             rpc_cmp_addr((struct sockaddr *)&conn_cb->ksnr_addr, sa))
 880                 return conn_cb->ksnr_connecting;
 881         return 0;
 882 }
 883
 884 int
 885 ksocknal_create_conn(struct lnet_ni *ni, struct ksock_conn_cb *conn_cb,
 886                      struct socket *sock, int type)
 887 {
 888         rwlock_t *global_lock = &ksocknal_data.ksnd_global_lock;
 889         LIST_HEAD(zombies);
 890         struct lnet_processid peerid;
 891         u64 incarnation;
 892         struct ksock_conn *conn;
 893         struct ksock_conn *conn2;
 894         struct ksock_peer_ni *peer_ni = NULL;
 895         struct ksock_peer_ni *peer2;
 896         struct ksock_sched *sched;
 897         struct ksock_hello_msg *hello;
 898         int cpt;
 899         struct ksock_tx *tx;
 900         struct ksock_tx *txtmp;
 901         int rc;
 902         int rc2;
 903         int active;
 904         int num_dup = 0;
 905         char *warn = NULL;
 906
 907         active = (conn_cb != NULL);
 908
 909         LASSERT(active == (type != SOCKLND_CONN_NONE));
 910
 911         LIBCFS_ALLOC(conn, sizeof(*conn));
 912         if (conn == NULL) {
 913                 rc = -ENOMEM;
 914                 goto failed_0;
 915         }
 916
 917         conn->ksnc_peer = NULL;
 918         conn->ksnc_conn_cb = NULL;
 919         conn->ksnc_sock = sock;
 920         /* 2 ref, 1 for conn, another extra ref prevents socket
 921          * being closed before establishment of connection */
 922         refcount_set(&conn->ksnc_sock_refcount, 2);
 923         conn->ksnc_type = type;
 924         ksocknal_lib_save_callback(sock, conn);
 925         refcount_set(&conn->ksnc_conn_refcount, 1); /* 1 ref for me */
 926
 927         conn->ksnc_rx_ready = 0;
 928         conn->ksnc_rx_scheduled = 0;
 929
 930         INIT_LIST_HEAD(&conn->ksnc_tx_queue);
 931         conn->ksnc_tx_ready = 0;
 932         conn->ksnc_tx_scheduled = 0;
 933         conn->ksnc_tx_carrier = NULL;
 934         atomic_set (&conn->ksnc_tx_nob, 0);
 935
 936         LIBCFS_ALLOC(hello, offsetof(struct ksock_hello_msg,
 937                                      kshm_ips[LNET_INTERFACES_NUM]));
 938         if (hello == NULL) {
 939                 rc = -ENOMEM;
 940                 goto failed_1;
 941         }
 942
 943         /* stash conn's local and remote addrs */
 944         rc = ksocknal_lib_get_conn_addrs(conn);
 945         if (rc != 0)
 946                 goto failed_1;
 947
 948         /* Find out/confirm peer_ni's NID and connection type and get the
 949          * vector of interfaces she's willing to let me connect to.
 950          * Passive connections use the listener timeout since the peer_ni sends
 951          * eagerly
 952          */
 953
 954         if (active) {
 955                 peer_ni = conn_cb->ksnr_peer;
 956                 LASSERT(ni == peer_ni->ksnp_ni);
 957
 958                 /* Active connection sends HELLO eagerly */
 959                 hello->kshm_nips = 0;
 960                 peerid = peer_ni->ksnp_id;
 961
 962                 write_lock_bh(global_lock);
 963                 conn->ksnc_proto = peer_ni->ksnp_proto;
 964                 write_unlock_bh(global_lock);
 965
 966                 if (conn->ksnc_proto == NULL) {
 967                         conn->ksnc_proto = &ksocknal_protocol_v3x;
 968 #if SOCKNAL_VERSION_DEBUG
 969                         if (*ksocknal_tunables.ksnd_protocol == 2)
 970                                 conn->ksnc_proto = &ksocknal_protocol_v2x;
 971                         else if (*ksocknal_tunables.ksnd_protocol == 1)
 972                                 conn->ksnc_proto = &ksocknal_protocol_v1x;
 973 #endif
 974                 }
 975
 976                 rc = ksocknal_send_hello(ni, conn, &peerid.nid, hello);
 977                 if (rc != 0)
 978                         goto failed_1;
 979         } else {
 980                 peerid.nid = LNET_ANY_NID;
 981                 peerid.pid = LNET_PID_ANY;
 982
 983                 /* Passive, get protocol from peer_ni */
 984                 conn->ksnc_proto = NULL;
 985         }
 986
 987         rc = ksocknal_recv_hello(ni, conn, hello, &peerid, &incarnation);
 988         if (rc < 0)
 989                 goto failed_1;
 990
 991         LASSERT(rc == 0 || active);
 992         LASSERT(conn->ksnc_proto != NULL);
 993         LASSERT(!LNET_NID_IS_ANY(&peerid.nid));
 994
 995         cpt = lnet_nid2cpt(&peerid.nid, ni);
 996
 997         if (active) {
 998                 ksocknal_peer_addref(peer_ni);
 999                 write_lock_bh(global_lock);
1000         } else {
1001                 peer_ni = ksocknal_create_peer(ni, &peerid);
1002                 if (IS_ERR(peer_ni)) {
1003                         rc = PTR_ERR(peer_ni);
1004                         goto failed_1;
1005                 }
1006
1007                 write_lock_bh(global_lock);
1008
1009                 /* called with a ref on ni, so shutdown can't have started */
1010                 LASSERT(atomic_read(&((struct ksock_net *)ni->ni_data)->ksnn_npeers) >= 0);
1011
1012                 peer2 = ksocknal_find_peer_locked(ni, &peerid);
1013                 if (peer2 == NULL) {
1014                         /* NB this puts an "empty" peer_ni in the peer_ni
1015                          * table (which takes my ref) */
1016                         hash_add(ksocknal_data.ksnd_peers,
1017                                  &peer_ni->ksnp_list, nidhash(&peerid.nid));
1018                 } else {
1019                         ksocknal_peer_decref(peer_ni);
1020                         peer_ni = peer2;
1021                 }
1022
1023                 /* +1 ref for me */
1024                 ksocknal_peer_addref(peer_ni);
1025                 peer_ni->ksnp_accepting++;
1026
1027                 /* Am I already connecting to this guy?  Resolve in
1028                  * favour of higher NID...
1029                  */
1030                 if (memcmp(&peerid.nid, &ni->ni_nid, sizeof(peerid.nid)) < 0 &&
1031                     ksocknal_connecting(peer_ni->ksnp_conn_cb,
1032                                         ((struct sockaddr *) &conn->ksnc_peeraddr))) {
1033                         rc = EALREADY;
1034                         warn = "connection race resolution";
1035                         goto failed_2;
1036                 }
1037         }
1038
1039         if (peer_ni->ksnp_closing ||
1040             (active && conn_cb->ksnr_deleted)) {
1041                 /* peer_ni/conn_cb got closed under me */
1042                 rc = -ESTALE;
1043                 warn = "peer_ni/conn_cb removed";
1044                 goto failed_2;
1045         }
1046
1047         if (peer_ni->ksnp_proto == NULL) {
1048                 /* Never connected before.
1049                  * NB recv_hello may have returned EPROTO to signal my peer_ni
1050                  * wants a different protocol than the one I asked for.
1051                  */
1052                 LASSERT(list_empty(&peer_ni->ksnp_conns));
1053
1054                 peer_ni->ksnp_proto = conn->ksnc_proto;
1055                 peer_ni->ksnp_incarnation = incarnation;
1056         }
1057
1058         if (peer_ni->ksnp_proto != conn->ksnc_proto ||
1059             peer_ni->ksnp_incarnation != incarnation) {
1060                 /* peer_ni rebooted or I've got the wrong protocol version */
1061                 ksocknal_close_peer_conns_locked(peer_ni, NULL, 0);
1062
1063                 peer_ni->ksnp_proto = NULL;
1064                 rc = ESTALE;
1065                 warn = peer_ni->ksnp_incarnation != incarnation ?
1066                         "peer_ni rebooted" :
1067                         "wrong proto version";
1068                 goto failed_2;
1069         }
1070
1071         switch (rc) {
1072         default:
1073                 LBUG();
1074         case 0:
1075                 break;
1076         case EALREADY:
1077                 warn = "lost conn race";
1078                 goto failed_2;
1079         case EPROTO:
1080                 warn = "retry with different protocol version";
1081                 goto failed_2;
1082         }
1083
1084         /* Refuse to duplicate an existing connection, unless this is a
1085          * loopback connection */
1086         if (!rpc_cmp_addr((struct sockaddr *)&conn->ksnc_peeraddr,
1087                           (struct sockaddr *)&conn->ksnc_myaddr)) {
1088                 list_for_each_entry(conn2, &peer_ni->ksnp_conns, ksnc_list) {
1089                         if (!rpc_cmp_addr(
1090                                     (struct sockaddr *)&conn2->ksnc_peeraddr,
1091                                     (struct sockaddr *)&conn->ksnc_peeraddr) ||
1092                             !rpc_cmp_addr(
1093                                     (struct sockaddr *)&conn2->ksnc_myaddr,
1094                                     (struct sockaddr *)&conn->ksnc_myaddr) ||
1095                             conn2->ksnc_type != conn->ksnc_type)
1096                                 continue;
1097
1098                         num_dup++;
1099                         /* If max conns per type is not registered in conn_cb
1100                          * as ksnr_max_conns, use ni's conns_per_peer
1101                          */
1102                         if ((peer_ni->ksnp_conn_cb &&
1103                             num_dup < peer_ni->ksnp_conn_cb->ksnr_max_conns) ||
1104                             (!peer_ni->ksnp_conn_cb &&
1105                             num_dup < ksocknal_get_conns_per_peer(peer_ni)))
1106                                 continue;
1107
1108                         /* Reply on a passive connection attempt so the peer_ni
1109                          * realises we're connected.
1110                          */
1111                         LASSERT(rc == 0);
1112                         if (!active)
1113                                 rc = EALREADY;
1114
1115                         warn = "duplicate";
1116                         goto failed_2;
1117                 }
1118         }
1119         /* If the connection created by this route didn't bind to the IP
1120          * address the route connected to, the connection/route matching
1121          * code below probably isn't going to work.
1122          */
1123         if (active &&
1124             !rpc_cmp_addr((struct sockaddr *)&conn_cb->ksnr_addr,
1125                           (struct sockaddr *)&conn->ksnc_peeraddr)) {
1126                 CERROR("Route %s %pISc connected to %pISc\n",
1127                        libcfs_idstr(&peer_ni->ksnp_id),
1128                        &conn_cb->ksnr_addr,
1129                        &conn->ksnc_peeraddr);
1130         }
1131
1132         /* Search for a conn_cb corresponding to the new connection and
1133          * create an association.  This allows incoming connections created
1134          * by conn_cbs in my peer_ni to match my own conn_cb entries so I don't
1135          * continually create duplicate conn_cbs.
1136          */
1137         conn_cb = peer_ni->ksnp_conn_cb;
1138
1139         if (conn_cb && rpc_cmp_addr((struct sockaddr *)&conn->ksnc_peeraddr,
1140                                     (struct sockaddr *)&conn_cb->ksnr_addr))
1141                 ksocknal_associate_cb_conn_locked(conn_cb, conn);
1142
1143         conn->ksnc_peer = peer_ni;                 /* conn takes my ref on peer_ni */
1144         peer_ni->ksnp_last_alive = ktime_get_seconds();
1145         peer_ni->ksnp_send_keepalive = 0;
1146         peer_ni->ksnp_error = 0;
1147
1148         sched = ksocknal_choose_scheduler_locked(cpt);
1149         if (!sched) {
1150                 CERROR("no schedulers available. node is unhealthy\n");
1151                 goto failed_2;
1152         }
1153         /*
1154          * The cpt might have changed if we ended up selecting a non cpt
1155          * native scheduler. So use the scheduler's cpt instead.
1156          */
1157         cpt = sched->kss_cpt;
1158         sched->kss_nconns++;
1159         conn->ksnc_scheduler = sched;
1160
1161         conn->ksnc_tx_last_post = ktime_get_seconds();
1162         /* Set the deadline for the outgoing HELLO to drain */
1163         conn->ksnc_tx_bufnob = sock->sk->sk_wmem_queued;
1164         conn->ksnc_tx_deadline = ktime_get_seconds() +
1165                                  ksocknal_timeout();
1166         smp_mb();   /* order with adding to peer_ni's conn list */
1167
1168         list_add(&conn->ksnc_list, &peer_ni->ksnp_conns);
1169         ksocknal_conn_addref(conn);
1170
1171         ksocknal_new_packet(conn, 0);
1172
1173         conn->ksnc_zc_capable = ksocknal_lib_zc_capable(conn);
1174
1175         /* Take packets blocking for this connection. */
1176         list_for_each_entry_safe(tx, txtmp, &peer_ni->ksnp_tx_queue, tx_list) {
1177                 if (conn->ksnc_proto->pro_match_tx(conn, tx, tx->tx_nonblk) ==
1178                     SOCKNAL_MATCH_NO)
1179                         continue;
1180
1181                 list_del(&tx->tx_list);
1182                 ksocknal_queue_tx_locked(tx, conn);
1183         }
1184
1185         write_unlock_bh(global_lock);
1186         /* We've now got a new connection.  Any errors from here on are just
1187          * like "normal" comms errors and we close the connection normally.
1188          * NB (a) we still have to send the reply HELLO for passive
1189          *        connections,
1190          *    (b) normal I/O on the conn is blocked until I setup and call the
1191          *        socket callbacks.
1192          */
1193
1194         CDEBUG(D_NET, "New conn %s p %d.x %pISc -> %pIScp"
1195                " incarnation:%lld sched[%d]\n",
1196                libcfs_idstr(&peerid), conn->ksnc_proto->pro_version,
1197                &conn->ksnc_myaddr, &conn->ksnc_peeraddr,
1198                incarnation, cpt);
1199
1200         if (!active) {
1201                 hello->kshm_nips = 0;
1202                 rc = ksocknal_send_hello(ni, conn, &peerid.nid, hello);
1203         }
1204
1205         LIBCFS_FREE(hello, offsetof(struct ksock_hello_msg,
1206                                     kshm_ips[LNET_INTERFACES_NUM]));
1207
1208         /* setup the socket AFTER I've received hello (it disables
1209          * SO_LINGER).  I might call back to the acceptor who may want
1210          * to send a protocol version response and then close the
1211          * socket; this ensures the socket only tears down after the
1212          * response has been sent.
1213          */
1214         if (rc == 0)
1215                 rc = ksocknal_lib_setup_sock(sock);
1216
1217         write_lock_bh(global_lock);
1218
1219         /* NB my callbacks block while I hold ksnd_global_lock */
1220         ksocknal_lib_set_callback(sock, conn);
1221
1222         if (!active)
1223                 peer_ni->ksnp_accepting--;
1224
1225         write_unlock_bh(global_lock);
1226
1227         if (rc != 0) {
1228                 write_lock_bh(global_lock);
1229                 if (!conn->ksnc_closing) {
1230                         /* could be closed by another thread */
1231                         ksocknal_close_conn_locked(conn, rc);
1232                 }
1233                 write_unlock_bh(global_lock);
1234         } else if (ksocknal_connsock_addref(conn) == 0) {
1235                 /* Allow I/O to proceed. */
1236                 ksocknal_read_callback(conn);
1237                 ksocknal_write_callback(conn);
1238                 ksocknal_connsock_decref(conn);
1239         }
1240
1241         ksocknal_connsock_decref(conn);
1242         ksocknal_conn_decref(conn);
1243         return rc;
1244
1245 failed_2:
1246
1247         if (!peer_ni->ksnp_closing &&
1248             list_empty(&peer_ni->ksnp_conns) &&
1249             peer_ni->ksnp_conn_cb == NULL) {
1250                 list_splice_init(&peer_ni->ksnp_tx_queue, &zombies);
1251                 ksocknal_unlink_peer_locked(peer_ni);
1252         }
1253
1254         write_unlock_bh(global_lock);
1255
1256         if (warn != NULL) {
1257                 if (rc < 0)
1258                         CERROR("Not creating conn %s type %d: %s\n",
1259                                libcfs_idstr(&peerid), conn->ksnc_type, warn);
1260                 else
1261                         CDEBUG(D_NET, "Not creating conn %s type %d: %s\n",
1262                                libcfs_idstr(&peerid), conn->ksnc_type, warn);
1263         }
1264
1265         if (!active) {
1266                 if (rc > 0) {
1267                         /* Request retry by replying with CONN_NONE
1268                          * ksnc_proto has been set already
1269                          */
1270                         conn->ksnc_type = SOCKLND_CONN_NONE;
1271                         hello->kshm_nips = 0;
1272                         ksocknal_send_hello(ni, conn, &peerid.nid, hello);
1273                 }
1274
1275                 write_lock_bh(global_lock);
1276                 peer_ni->ksnp_accepting--;
1277                 write_unlock_bh(global_lock);
1278         }
1279
1280         /*
1281          * If we get here without an error code, just use -EALREADY.
1282          * Depending on how we got here, the error may be positive
1283          * or negative. Normalize the value for ksocknal_txlist_done().
1284          */
1285         rc2 = (rc == 0 ? -EALREADY : (rc > 0 ? -rc : rc));
1286         ksocknal_txlist_done(ni, &zombies, rc2);
1287         ksocknal_peer_decref(peer_ni);
1288
1289 failed_1:
1290         if (hello != NULL)
1291                 LIBCFS_FREE(hello, offsetof(struct ksock_hello_msg,
1292                                             kshm_ips[LNET_INTERFACES_NUM]));
1293
1294         LIBCFS_FREE(conn, sizeof(*conn));
1295
1296 failed_0:
1297         sock_release(sock);
1298
1299         return rc;
1300 }
1301
1302 void
1303 ksocknal_close_conn_locked(struct ksock_conn *conn, int error)
1304 {
1305         /* This just does the immmediate housekeeping, and queues the
1306          * connection for the reaper to terminate.
1307          * Caller holds ksnd_global_lock exclusively in irq context */
1308         struct ksock_peer_ni *peer_ni = conn->ksnc_peer;
1309         struct ksock_conn_cb *conn_cb;
1310         struct ksock_conn *conn2;
1311         int conn_count;
1312         int duplicate_count = 0;
1313
1314         LASSERT(peer_ni->ksnp_error == 0);
1315         LASSERT(!conn->ksnc_closing);
1316         conn->ksnc_closing = 1;
1317
1318         /* ksnd_deathrow_conns takes over peer_ni's ref */
1319         list_del(&conn->ksnc_list);
1320
1321         conn_cb = conn->ksnc_conn_cb;
1322         if (conn_cb != NULL) {
1323                 /* dissociate conn from cb... */
1324                 LASSERT(!conn_cb->ksnr_deleted);
1325
1326                 conn_count = ksocknal_get_conn_count_by_type(conn_cb,
1327                                                              conn->ksnc_type);
1328                 /* connected bit is set only if all connections
1329                  * of the given type got created
1330                  */
1331                 if (conn_count == conn_cb->ksnr_max_conns)
1332                         LASSERT((conn_cb->ksnr_connected &
1333                                 BIT(conn->ksnc_type)) != 0);
1334
1335                 if (conn_count == 1) {
1336                         list_for_each_entry(conn2, &peer_ni->ksnp_conns,
1337                                             ksnc_list) {
1338                                 if (conn2->ksnc_conn_cb == conn_cb &&
1339                                     conn2->ksnc_type == conn->ksnc_type)
1340                                         duplicate_count += 1;
1341                         }
1342                         if (duplicate_count > 0)
1343                                 CERROR("Found %d duplicate conns type %d\n",
1344                                        duplicate_count,
1345                                        conn->ksnc_type);
1346                 }
1347                 ksocknal_decr_conn_count(conn_cb, conn->ksnc_type);
1348
1349                 conn->ksnc_conn_cb = NULL;
1350
1351                 /* drop conn's ref on conn_cb */
1352                 ksocknal_conn_cb_decref(conn_cb);
1353         }
1354
1355         if (list_empty(&peer_ni->ksnp_conns)) {
1356                 /* No more connections to this peer_ni */
1357
1358                 if (!list_empty(&peer_ni->ksnp_tx_queue)) {
1359                         struct ksock_tx *tx;
1360
1361                         LASSERT(conn->ksnc_proto == &ksocknal_protocol_v3x);
1362
1363                         /* throw them to the last connection...,
1364                          * these TXs will be send to /dev/null by scheduler */
1365                         list_for_each_entry(tx, &peer_ni->ksnp_tx_queue,
1366                                             tx_list)
1367                                 ksocknal_tx_prep(conn, tx);
1368
1369                         spin_lock_bh(&conn->ksnc_scheduler->kss_lock);
1370                         list_splice_init(&peer_ni->ksnp_tx_queue,
1371                                          &conn->ksnc_tx_queue);
1372                         spin_unlock_bh(&conn->ksnc_scheduler->kss_lock);
1373                 }
1374
1375                 /* renegotiate protocol version */
1376                 peer_ni->ksnp_proto = NULL;
1377                 /* stash last conn close reason */
1378                 peer_ni->ksnp_error = error;
1379
1380                 if (peer_ni->ksnp_conn_cb == NULL) {
1381                         /* I've just closed last conn belonging to a
1382                          * peer_ni with no connections to it
1383                          */
1384                         ksocknal_unlink_peer_locked(peer_ni);
1385                 }
1386         }
1387
1388         spin_lock_bh(&ksocknal_data.ksnd_reaper_lock);
1389
1390         list_add_tail(&conn->ksnc_list, &ksocknal_data.ksnd_deathrow_conns);
1391         wake_up(&ksocknal_data.ksnd_reaper_waitq);
1392
1393         spin_unlock_bh(&ksocknal_data.ksnd_reaper_lock);
1394 }
1395
1396 void
1397 ksocknal_peer_failed(struct ksock_peer_ni *peer_ni)
1398 {
1399         bool notify = false;
1400         time64_t last_alive = 0;
1401
1402         /* There has been a connection failure or comms error; but I'll only
1403          * tell LNET I think the peer_ni is dead if it's to another kernel and
1404          * there are no connections or connection attempts in existence. */
1405
1406         read_lock(&ksocknal_data.ksnd_global_lock);
1407
1408         if ((peer_ni->ksnp_id.pid & LNET_PID_USERFLAG) == 0 &&
1409              list_empty(&peer_ni->ksnp_conns) &&
1410              peer_ni->ksnp_accepting == 0 &&
1411              !ksocknal_find_connecting_conn_cb_locked(peer_ni)) {
1412                 notify = true;
1413                 last_alive = peer_ni->ksnp_last_alive;
1414         }
1415
1416         read_unlock(&ksocknal_data.ksnd_global_lock);
1417
1418         if (notify)
1419                 lnet_notify(peer_ni->ksnp_ni,
1420                             &peer_ni->ksnp_id.nid,
1421                             false, false, last_alive);
1422 }
1423
1424 void
1425 ksocknal_finalize_zcreq(struct ksock_conn *conn)
1426 {
1427         struct ksock_peer_ni *peer_ni = conn->ksnc_peer;
1428         struct ksock_tx *tx;
1429         struct ksock_tx *tmp;
1430         LIST_HEAD(zlist);
1431
1432         /* NB safe to finalize TXs because closing of socket will
1433          * abort all buffered data */
1434         LASSERT(conn->ksnc_sock == NULL);
1435
1436         spin_lock(&peer_ni->ksnp_lock);
1437
1438         list_for_each_entry_safe(tx, tmp, &peer_ni->ksnp_zc_req_list,
1439                                  tx_zc_list) {
1440                 if (tx->tx_conn != conn)
1441                         continue;
1442
1443                 LASSERT(tx->tx_msg.ksm_zc_cookies[0] != 0);
1444
1445                 tx->tx_msg.ksm_zc_cookies[0] = 0;
1446                 tx->tx_zc_aborted = 1;  /* mark it as not-acked */
1447                 list_move(&tx->tx_zc_list, &zlist);
1448         }
1449
1450         spin_unlock(&peer_ni->ksnp_lock);
1451
1452         while ((tx = list_first_entry_or_null(&zlist, struct ksock_tx,
1453                                               tx_zc_list)) != NULL) {
1454                 list_del(&tx->tx_zc_list);
1455                 ksocknal_tx_decref(tx);
1456         }
1457 }
1458
1459 void
1460 ksocknal_terminate_conn(struct ksock_conn *conn)
1461 {
1462         /* This gets called by the reaper (guaranteed thread context) to
1463          * disengage the socket from its callbacks and close it.
1464          * ksnc_refcount will eventually hit zero, and then the reaper will
1465          * destroy it.
1466          */
1467         struct ksock_peer_ni *peer_ni = conn->ksnc_peer;
1468         struct ksock_sched *sched = conn->ksnc_scheduler;
1469         bool failed = false;
1470
1471         LASSERT(conn->ksnc_closing);
1472
1473         /* wake up the scheduler to "send" all remaining packets to /dev/null */
1474         spin_lock_bh(&sched->kss_lock);
1475
1476         /* a closing conn is always ready to tx */
1477         conn->ksnc_tx_ready = 1;
1478
1479         if (!conn->ksnc_tx_scheduled &&
1480             !list_empty(&conn->ksnc_tx_queue)) {
1481                 list_add_tail(&conn->ksnc_tx_list,
1482                               &sched->kss_tx_conns);
1483                 conn->ksnc_tx_scheduled = 1;
1484                 /* extra ref for scheduler */
1485                 ksocknal_conn_addref(conn);
1486
1487                 wake_up(&sched->kss_waitq);
1488         }
1489
1490         spin_unlock_bh(&sched->kss_lock);
1491
1492         /* serialise with callbacks */
1493         write_lock_bh(&ksocknal_data.ksnd_global_lock);
1494
1495         ksocknal_lib_reset_callback(conn->ksnc_sock, conn);
1496
1497         /* OK, so this conn may not be completely disengaged from its
1498          * scheduler yet, but it _has_ committed to terminate...
1499          */
1500         conn->ksnc_scheduler->kss_nconns--;
1501
1502         if (peer_ni->ksnp_error != 0) {
1503                 /* peer_ni's last conn closed in error */
1504                 LASSERT(list_empty(&peer_ni->ksnp_conns));
1505                 failed = true;
1506                 peer_ni->ksnp_error = 0;     /* avoid multiple notifications */
1507         }
1508
1509         write_unlock_bh(&ksocknal_data.ksnd_global_lock);
1510
1511         if (failed)
1512                 ksocknal_peer_failed(peer_ni);
1513
1514         /* The socket is closed on the final put; either here, or in
1515          * ksocknal_{send,recv}msg().  Since we set up the linger2 option
1516          * when the connection was established, this will close the socket
1517          * immediately, aborting anything buffered in it. Any hung
1518          * zero-copy transmits will therefore complete in finite time.
1519          */
1520         ksocknal_connsock_decref(conn);
1521 }
1522
1523 void
1524 ksocknal_queue_zombie_conn(struct ksock_conn *conn)
1525 {
1526         /* Queue the conn for the reaper to destroy */
1527         LASSERT(refcount_read(&conn->ksnc_conn_refcount) == 0);
1528         spin_lock_bh(&ksocknal_data.ksnd_reaper_lock);
1529
1530         list_add_tail(&conn->ksnc_list, &ksocknal_data.ksnd_zombie_conns);
1531         wake_up(&ksocknal_data.ksnd_reaper_waitq);
1532
1533         spin_unlock_bh(&ksocknal_data.ksnd_reaper_lock);
1534 }
1535
1536 void
1537 ksocknal_destroy_conn(struct ksock_conn *conn)
1538 {
1539         time64_t last_rcv;
1540
1541         /* Final coup-de-grace of the reaper */
1542         CDEBUG(D_NET, "connection %p\n", conn);
1543
1544         LASSERT(refcount_read(&conn->ksnc_conn_refcount) == 0);
1545         LASSERT(refcount_read(&conn->ksnc_sock_refcount) == 0);
1546         LASSERT(conn->ksnc_sock == NULL);
1547         LASSERT(conn->ksnc_conn_cb == NULL);
1548         LASSERT(!conn->ksnc_tx_scheduled);
1549         LASSERT(!conn->ksnc_rx_scheduled);
1550         LASSERT(list_empty(&conn->ksnc_tx_queue));
1551
1552         /* complete current receive if any */
1553         switch (conn->ksnc_rx_state) {
1554         case SOCKNAL_RX_LNET_PAYLOAD:
1555                 last_rcv = conn->ksnc_rx_deadline -
1556                            ksocknal_timeout();
1557                 CERROR("Completing partial receive from %s[%d], ip %pIScp, with error, wanted: %d, left: %d, last alive is %lld secs ago\n",
1558                        libcfs_idstr(&conn->ksnc_peer->ksnp_id),
1559                        conn->ksnc_type,
1560                        &conn->ksnc_peeraddr,
1561                        conn->ksnc_rx_nob_wanted, conn->ksnc_rx_nob_left,
1562                        ktime_get_seconds() - last_rcv);
1563                 if (conn->ksnc_lnet_msg)
1564                         conn->ksnc_lnet_msg->msg_health_status =
1565                                 LNET_MSG_STATUS_REMOTE_ERROR;
1566                 lnet_finalize(conn->ksnc_lnet_msg, -EIO);
1567                 break;
1568         case SOCKNAL_RX_LNET_HEADER:
1569                 if (conn->ksnc_rx_started)
1570                         CERROR("Incomplete receive of lnet header from %s, ip %pIScp, with error, protocol: %d.x.\n",
1571                                libcfs_idstr(&conn->ksnc_peer->ksnp_id),
1572                                &conn->ksnc_peeraddr,
1573                                conn->ksnc_proto->pro_version);
1574                 break;
1575         case SOCKNAL_RX_KSM_HEADER:
1576                 if (conn->ksnc_rx_started)
1577                         CERROR("Incomplete receive of ksock message from %s, ip %pIScp, with error, protocol: %d.x.\n",
1578                                libcfs_idstr(&conn->ksnc_peer->ksnp_id),
1579                                &conn->ksnc_peeraddr,
1580                                conn->ksnc_proto->pro_version);
1581                 break;
1582         case SOCKNAL_RX_SLOP:
1583                 if (conn->ksnc_rx_started)
1584                         CERROR("Incomplete receive of slops from %s, ip %pIScp, with error\n",
1585                                libcfs_idstr(&conn->ksnc_peer->ksnp_id),
1586                                &conn->ksnc_peeraddr);
1587                 break;
1588         default:
1589                 LBUG();
1590                 break;
1591         }
1592
1593         ksocknal_peer_decref(conn->ksnc_peer);
1594
1595         LIBCFS_FREE(conn, sizeof(*conn));
1596 }
1597
1598 int
1599 ksocknal_close_peer_conns_locked(struct ksock_peer_ni *peer_ni,
1600                                  struct sockaddr *addr, int why)
1601 {
1602         struct ksock_conn *conn;
1603         struct ksock_conn *cnxt;
1604         int count = 0;
1605
1606         list_for_each_entry_safe(conn, cnxt, &peer_ni->ksnp_conns, ksnc_list) {
1607                 if (!addr ||
1608                     rpc_cmp_addr(addr,
1609                                  (struct sockaddr *)&conn->ksnc_peeraddr)) {
1610                         count++;
1611                         ksocknal_close_conn_locked(conn, why);
1612                 }
1613         }
1614
1615         return count;
1616 }
1617
1618 int
1619 ksocknal_close_conn_and_siblings(struct ksock_conn *conn, int why)
1620 {
1621         struct ksock_peer_ni *peer_ni = conn->ksnc_peer;
1622         int count;
1623
1624         write_lock_bh(&ksocknal_data.ksnd_global_lock);
1625
1626         count = ksocknal_close_peer_conns_locked(
1627                 peer_ni, (struct sockaddr *)&conn->ksnc_peeraddr, why);
1628
1629         write_unlock_bh(&ksocknal_data.ksnd_global_lock);
1630
1631         return count;
1632 }
1633
1634 int
1635 ksocknal_close_matching_conns(struct lnet_processid *id, __u32 ipaddr)
1636 {
1637         struct ksock_peer_ni *peer_ni;
1638         struct hlist_node *pnxt;
1639         int lo;
1640         int hi;
1641         int i;
1642         int count = 0;
1643         struct sockaddr_in sa = {.sin_family = AF_INET};
1644
1645         write_lock_bh(&ksocknal_data.ksnd_global_lock);
1646
1647         if (!LNET_NID_IS_ANY(&id->nid)) {
1648                 lo = hash_min(nidhash(&id->nid),
1649                               HASH_BITS(ksocknal_data.ksnd_peers));
1650                 hi = lo;
1651         } else {
1652                 lo = 0;
1653                 hi = HASH_SIZE(ksocknal_data.ksnd_peers) - 1;
1654         }
1655
1656         sa.sin_addr.s_addr = htonl(ipaddr);
1657         for (i = lo; i <= hi; i++) {
1658                 hlist_for_each_entry_safe(peer_ni, pnxt,
1659                                           &ksocknal_data.ksnd_peers[i],
1660                                           ksnp_list) {
1661
1662                         if (!((LNET_NID_IS_ANY(&id->nid) ||
1663                                nid_same(&id->nid, &peer_ni->ksnp_id.nid)) &&
1664                               (id->pid == LNET_PID_ANY ||
1665                                id->pid == peer_ni->ksnp_id.pid)))
1666                                 continue;
1667
1668                         count += ksocknal_close_peer_conns_locked(
1669                                 peer_ni,
1670                                 ipaddr ? (struct sockaddr *)&sa : NULL, 0);
1671                 }
1672         }
1673
1674         write_unlock_bh(&ksocknal_data.ksnd_global_lock);
1675
1676         /* wildcards always succeed */
1677         if (LNET_NID_IS_ANY(&id->nid) || id->pid == LNET_PID_ANY ||
1678             ipaddr == 0)
1679                 return 0;
1680
1681         return (count == 0 ? -ENOENT : 0);
1682 }
1683
1684 void
1685 ksocknal_notify_gw_down(struct lnet_nid *gw_nid)
1686 {
1687         /* The router is telling me she's been notified of a change in
1688          * gateway state....
1689          */
1690         struct lnet_processid id = {
1691                 .pid    = LNET_PID_ANY,
1692                 .nid    = *gw_nid,
1693         };
1694
1695         CDEBUG(D_NET, "gw %s down\n", libcfs_nidstr(gw_nid));
1696
1697         /* If the gateway crashed, close all open connections... */
1698         ksocknal_close_matching_conns(&id, 0);
1699         return;
1700
1701         /* We can only establish new connections
1702          * if we have autroutes, and these connect on demand.
1703          */
1704 }
1705
1706 static void
1707 ksocknal_push_peer(struct ksock_peer_ni *peer_ni)
1708 {
1709         int index;
1710         int i;
1711         struct ksock_conn *conn;
1712
1713         for (index = 0; ; index++) {
1714                 read_lock(&ksocknal_data.ksnd_global_lock);
1715
1716                 i = 0;
1717                 conn = NULL;
1718
1719                 list_for_each_entry(conn, &peer_ni->ksnp_conns, ksnc_list) {
1720                         if (i++ == index) {
1721                                 ksocknal_conn_addref(conn);
1722                                 break;
1723                         }
1724                 }
1725
1726                 read_unlock(&ksocknal_data.ksnd_global_lock);
1727
1728                 if (i <= index)
1729                         break;
1730
1731                 ksocknal_lib_push_conn (conn);
1732                 ksocknal_conn_decref(conn);
1733         }
1734 }
1735
1736 static int
1737 ksocknal_push(struct lnet_ni *ni, struct lnet_processid *id)
1738 {
1739         int lo;
1740         int hi;
1741         int bkt;
1742         int rc = -ENOENT;
1743
1744         if (!LNET_NID_IS_ANY(&id->nid)) {
1745                 lo = hash_min(nidhash(&id->nid),
1746                               HASH_BITS(ksocknal_data.ksnd_peers));
1747                 hi = lo;
1748         } else {
1749                 lo = 0;
1750                 hi = HASH_SIZE(ksocknal_data.ksnd_peers) - 1;
1751         }
1752
1753         for (bkt = lo; bkt <= hi; bkt++) {
1754                 int peer_off; /* searching offset in peer_ni hash table */
1755
1756                 for (peer_off = 0; ; peer_off++) {
1757                         struct ksock_peer_ni *peer_ni;
1758                         int           i = 0;
1759
1760                         read_lock(&ksocknal_data.ksnd_global_lock);
1761                         hlist_for_each_entry(peer_ni,
1762                                              &ksocknal_data.ksnd_peers[bkt],
1763                                              ksnp_list) {
1764                                 if (!((LNET_NID_IS_ANY(&id->nid) ||
1765                                        nid_same(&id->nid,
1766                                                  &peer_ni->ksnp_id.nid)) &&
1767                                       (id->pid == LNET_PID_ANY ||
1768                                        id->pid == peer_ni->ksnp_id.pid)))
1769                                         continue;
1770
1771                                 if (i++ == peer_off) {
1772                                         ksocknal_peer_addref(peer_ni);
1773                                         break;
1774                                 }
1775                         }
1776                         read_unlock(&ksocknal_data.ksnd_global_lock);
1777
1778                         if (i <= peer_off) /* no match */
1779                                 break;
1780
1781                         rc = 0;
1782                         ksocknal_push_peer(peer_ni);
1783                         ksocknal_peer_decref(peer_ni);
1784                 }
1785         }
1786         return rc;
1787 }
1788
1789 int
1790 ksocknal_ctl(struct lnet_ni *ni, unsigned int cmd, void *arg)
1791 {
1792         struct lnet_processid id = {};
1793         struct libcfs_ioctl_data *data = arg;
1794         int rc;
1795
1796         switch(cmd) {
1797         case IOC_LIBCFS_GET_INTERFACE: {
1798                 struct ksock_net *net = ni->ni_data;
1799                 struct ksock_interface *iface;
1800                 struct sockaddr_in *sa;
1801
1802                 read_lock(&ksocknal_data.ksnd_global_lock);
1803
1804                 if (data->ioc_count >= 1) {
1805                         rc = -ENOENT;
1806                 } else {
1807                         rc = 0;
1808                         iface = &net->ksnn_interface;
1809
1810                         sa = (void *)&iface->ksni_addr;
1811                         if (sa->sin_family == AF_INET) {
1812                                 data->ioc_u32[0] = ntohl(sa->sin_addr.s_addr);
1813                                 data->ioc_u32[1] = iface->ksni_netmask;
1814                         } else {
1815                                 data->ioc_u32[0] = 0xFFFFFFFF;
1816                                 data->ioc_u32[1] = 0;
1817                         }
1818                         data->ioc_u32[2] = iface->ksni_npeers;
1819                         data->ioc_u32[3] = iface->ksni_nroutes;
1820                 }
1821
1822                 read_unlock(&ksocknal_data.ksnd_global_lock);
1823                 return rc;
1824         }
1825
1826         case IOC_LIBCFS_GET_PEER: {
1827                 __u32            myip = 0;
1828                 __u32            ip = 0;
1829                 int              port = 0;
1830                 int              conn_count = 0;
1831                 int              share_count = 0;
1832
1833                 rc = ksocknal_get_peer_info(ni, data->ioc_count,
1834                                             &id, &myip, &ip, &port,
1835                                             &conn_count,  &share_count);
1836                 if (rc != 0)
1837                         return rc;
1838
1839                 if (!nid_is_nid4(&id.nid))
1840                         return -EINVAL;
1841                 data->ioc_nid    = lnet_nid_to_nid4(&id.nid);
1842                 data->ioc_count  = share_count;
1843                 data->ioc_u32[0] = ip;
1844                 data->ioc_u32[1] = port;
1845                 data->ioc_u32[2] = myip;
1846                 data->ioc_u32[3] = conn_count;
1847                 data->ioc_u32[4] = id.pid;
1848                 return 0;
1849         }
1850
1851         case IOC_LIBCFS_ADD_PEER: {
1852                 struct sockaddr_in sa = {.sin_family = AF_INET};
1853
1854                 id.pid = LNET_PID_LUSTRE;
1855                 lnet_nid4_to_nid(data->ioc_nid, &id.nid);
1856                 sa.sin_addr.s_addr = htonl(data->ioc_u32[0]);
1857                 sa.sin_port = htons(data->ioc_u32[1]);
1858                 return ksocknal_add_peer(ni, &id, (struct sockaddr *)&sa);
1859         }
1860         case IOC_LIBCFS_DEL_PEER:
1861                 lnet_nid4_to_nid(data->ioc_nid, &id.nid);
1862                 id.pid = LNET_PID_ANY;
1863                 return ksocknal_del_peer(ni, &id);
1864
1865         case IOC_LIBCFS_GET_CONN: {
1866                 int           txmem;
1867                 int           rxmem;
1868                 int           nagle;
1869                 struct ksock_conn *conn = ksocknal_get_conn_by_idx(ni, data->ioc_count);
1870                 struct sockaddr_in *psa = (void *)&conn->ksnc_peeraddr;
1871                 struct sockaddr_in *mysa = (void *)&conn->ksnc_myaddr;
1872
1873                 if (conn == NULL)
1874                         return -ENOENT;
1875
1876                 ksocknal_lib_get_conn_tunables(conn, &txmem, &rxmem, &nagle);
1877
1878                 data->ioc_count = txmem;
1879                 data->ioc_nid = lnet_nid_to_nid4(&conn->ksnc_peer->ksnp_id.nid);
1880                 data->ioc_flags = nagle;
1881                 if (psa->sin_family == AF_INET)
1882                         data->ioc_u32[0] = ntohl(psa->sin_addr.s_addr);
1883                 else
1884                         data->ioc_u32[0] = 0xFFFFFFFF;
1885                 data->ioc_u32[1] = rpc_get_port((struct sockaddr *)
1886                                                 &conn->ksnc_peeraddr);
1887                 if (mysa->sin_family == AF_INET)
1888                         data->ioc_u32[2] = ntohl(mysa->sin_addr.s_addr);
1889                 else
1890                         data->ioc_u32[2] = 0xFFFFFFFF;
1891                 data->ioc_u32[3] = conn->ksnc_type;
1892                 data->ioc_u32[4] = conn->ksnc_scheduler->kss_cpt;
1893                 data->ioc_u32[5] = rxmem;
1894                 data->ioc_u32[6] = conn->ksnc_peer->ksnp_id.pid;
1895                 ksocknal_conn_decref(conn);
1896                 return 0;
1897         }
1898
1899         case IOC_LIBCFS_CLOSE_CONNECTION:
1900                 lnet_nid4_to_nid(data->ioc_nid, &id.nid);
1901                 id.pid = LNET_PID_ANY;
1902                 return ksocknal_close_matching_conns(&id,
1903                                                      data->ioc_u32[0]);
1904
1905         case IOC_LIBCFS_REGISTER_MYNID:
1906                 /* Ignore if this is a noop */
1907                 if (nid_is_nid4(&ni->ni_nid) &&
1908                     data->ioc_nid == lnet_nid_to_nid4(&ni->ni_nid))
1909                         return 0;
1910
1911                 CERROR("obsolete IOC_LIBCFS_REGISTER_MYNID: %s(%s)\n",
1912                        libcfs_nid2str(data->ioc_nid),
1913                        libcfs_nidstr(&ni->ni_nid));
1914                 return -EINVAL;
1915
1916         case IOC_LIBCFS_PUSH_CONNECTION:
1917                 lnet_nid4_to_nid(data->ioc_nid, &id.nid);
1918                 id.pid = LNET_PID_ANY;
1919                 return ksocknal_push(ni, &id);
1920
1921         default:
1922                 return -EINVAL;
1923         }
1924         /* not reached */
1925 }
1926
1927 static void
1928 ksocknal_free_buffers (void)
1929 {
1930         LASSERT (atomic_read(&ksocknal_data.ksnd_nactive_txs) == 0);
1931
1932         if (ksocknal_data.ksnd_schedulers != NULL)
1933                 cfs_percpt_free(ksocknal_data.ksnd_schedulers);
1934
1935         spin_lock(&ksocknal_data.ksnd_tx_lock);
1936
1937         if (!list_empty(&ksocknal_data.ksnd_idle_noop_txs)) {
1938                 LIST_HEAD(zlist);
1939                 struct ksock_tx *tx;
1940
1941                 list_splice_init(&ksocknal_data.ksnd_idle_noop_txs, &zlist);
1942                 spin_unlock(&ksocknal_data.ksnd_tx_lock);
1943
1944                 while ((tx = list_first_entry_or_null(&zlist, struct ksock_tx,
1945                                                       tx_list)) != NULL) {
1946                         list_del(&tx->tx_list);
1947                         LIBCFS_FREE(tx, tx->tx_desc_size);
1948                 }
1949         } else {
1950                 spin_unlock(&ksocknal_data.ksnd_tx_lock);
1951         }
1952 }
1953
1954 static int ksocknal_get_link_status(struct net_device *dev)
1955 {
1956         int ret = -1;
1957
1958         LASSERT(dev);
1959
1960         if (!netif_running(dev)) {
1961                 ret = 0;
1962                 CDEBUG(D_NET, "device not running\n");
1963         }
1964         /* Some devices may not be providing link settings */
1965         else if (dev->ethtool_ops->get_link) {
1966                 ret = dev->ethtool_ops->get_link(dev);
1967                 CDEBUG(D_NET, "get_link returns %u\n", ret);
1968         }
1969
1970         return ret;
1971 }
1972
1973 static int
1974 ksocknal_handle_link_state_change(struct net_device *dev,
1975                                   unsigned char operstate)
1976 {
1977         struct lnet_ni *ni = NULL;
1978         struct ksock_net *net;
1979         struct ksock_net *cnxt;
1980         int ifindex;
1981         unsigned char link_down = !(operstate == IF_OPER_UP);
1982         struct in_device *in_dev;
1983         bool found_ip = false;
1984         struct ksock_interface *ksi = NULL;
1985         struct sockaddr_in *sa;
1986         DECLARE_CONST_IN_IFADDR(ifa);
1987
1988         ifindex = dev->ifindex;
1989
1990         if (!ksocknal_data.ksnd_nnets)
1991                 goto out;
1992
1993         list_for_each_entry_safe(net, cnxt, &ksocknal_data.ksnd_nets,
1994                                  ksnn_list) {
1995
1996                 ksi = &net->ksnn_interface;
1997                 sa = (void *)&ksi->ksni_addr;
1998                 found_ip = false;
1999
2000                 if (strcmp(ksi->ksni_name, dev->name))
2001                         continue;
2002
2003                 if (ksi->ksni_index == -1) {
2004                         if (dev->reg_state != NETREG_REGISTERED)
2005                                 continue;
2006                         /* A registration just happened: save the new index for
2007                          * the device */
2008                         ksi->ksni_index = ifindex;
2009                         goto out;
2010                 }
2011
2012                 if (ksi->ksni_index != ifindex)
2013                         continue;
2014
2015                 if (dev->reg_state == NETREG_UNREGISTERING) {
2016                         /* Device is being unregitering, we need to clear the
2017                          * index, it can change when device will be back */
2018                         ksi->ksni_index = -1;
2019                         goto out;
2020                 }
2021
2022                 ni = net->ksnn_ni;
2023
2024                 in_dev = __in_dev_get_rtnl(dev);
2025                 if (!in_dev) {
2026                         CDEBUG(D_NET, "Interface %s has no IPv4 status.\n",
2027                                dev->name);
2028                         CDEBUG(D_NET, "set link fatal state to 1\n");
2029                         atomic_set(&ni->ni_fatal_error_on, 1);
2030                         continue;
2031                 }
2032                 in_dev_for_each_ifa_rtnl(ifa, in_dev) {
2033                         if (sa->sin_addr.s_addr == ifa->ifa_local)
2034                                 found_ip = true;
2035                 }
2036                 endfor_ifa(in_dev);
2037
2038                 if (!found_ip) {
2039                         CDEBUG(D_NET, "Interface %s has no matching ip\n",
2040                                dev->name);
2041                         CDEBUG(D_NET, "set link fatal state to 1\n");
2042                         atomic_set(&ni->ni_fatal_error_on, 1);
2043                         continue;
2044                 }
2045
2046                 if (link_down) {
2047                         CDEBUG(D_NET, "set link fatal state to 1\n");
2048                         atomic_set(&ni->ni_fatal_error_on, link_down);
2049                 } else {
2050                         CDEBUG(D_NET, "set link fatal state to %u\n",
2051                                (ksocknal_get_link_status(dev) == 0));
2052                         atomic_set(&ni->ni_fatal_error_on,
2053                                    (ksocknal_get_link_status(dev) == 0));
2054                 }
2055         }
2056 out:
2057         return 0;
2058 }
2059
2060
2061 static int
2062 ksocknal_handle_inetaddr_change(struct in_ifaddr *ifa, unsigned long event)
2063 {
2064         struct lnet_ni *ni;
2065         struct ksock_net *net;
2066         struct ksock_net *cnxt;
2067         struct net_device *event_netdev = ifa->ifa_dev->dev;
2068         int ifindex;
2069         struct ksock_interface *ksi = NULL;
2070         struct sockaddr_in *sa;
2071
2072         if (!ksocknal_data.ksnd_nnets)
2073                 goto out;
2074
2075         ifindex = event_netdev->ifindex;
2076
2077         list_for_each_entry_safe(net, cnxt, &ksocknal_data.ksnd_nets,
2078                                  ksnn_list) {
2079
2080                 ksi = &net->ksnn_interface;
2081                 sa = (void *)&ksi->ksni_addr;
2082
2083                 if (ksi->ksni_index != ifindex ||
2084                     strcmp(ksi->ksni_name, event_netdev->name))
2085                         continue;
2086
2087                 if (sa->sin_addr.s_addr == ifa->ifa_local) {
2088                         CDEBUG(D_NET, "set link fatal state to %u\n",
2089                                (event == NETDEV_DOWN));
2090                         ni = net->ksnn_ni;
2091                         atomic_set(&ni->ni_fatal_error_on,
2092                                    (event == NETDEV_DOWN));
2093                 }
2094         }
2095 out:
2096         return 0;
2097 }
2098
2099 /************************************
2100  * Net device notifier event handler
2101  ************************************/
2102 static int ksocknal_device_event(struct notifier_block *unused,
2103                                  unsigned long event, void *ptr)
2104 {
2105         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
2106         unsigned char operstate;
2107
2108         operstate = dev->operstate;
2109
2110         CDEBUG(D_NET, "devevent: status=%ld, iface=%s ifindex %d state %u\n",
2111                event, dev->name, dev->ifindex, operstate);
2112
2113         switch (event) {
2114         case NETDEV_UP:
2115         case NETDEV_DOWN:
2116         case NETDEV_CHANGE:
2117         case NETDEV_REGISTER:
2118         case NETDEV_UNREGISTER:
2119                 ksocknal_handle_link_state_change(dev, operstate);
2120                 break;
2121         }
2122
2123         return NOTIFY_OK;
2124 }
2125
2126 /************************************
2127  * Inetaddr notifier event handler
2128  ************************************/
2129 static int ksocknal_inetaddr_event(struct notifier_block *unused,
2130                                    unsigned long event, void *ptr)
2131 {
2132         struct in_ifaddr *ifa = ptr;
2133
2134         CDEBUG(D_NET, "addrevent: status %ld ip addr %pI4, netmask %pI4.\n",
2135                event, &ifa->ifa_address, &ifa->ifa_mask);
2136
2137         switch (event) {
2138         case NETDEV_UP:
2139         case NETDEV_DOWN:
2140         case NETDEV_CHANGE:
2141                 ksocknal_handle_inetaddr_change(ifa, event);
2142                 break;
2143
2144         }
2145         return NOTIFY_OK;
2146 }
2147
2148 static struct notifier_block ksocknal_dev_notifier_block = {
2149         .notifier_call = ksocknal_device_event,
2150 };
2151
2152 static struct notifier_block ksocknal_inetaddr_notifier_block = {
2153         .notifier_call = ksocknal_inetaddr_event,
2154 };
2155
2156 static void
2157 ksocknal_base_shutdown(void)
2158 {
2159         struct ksock_sched *sched;
2160         struct ksock_peer_ni *peer_ni;
2161         int i;
2162
2163         CDEBUG(D_MALLOC, "before NAL cleanup: kmem %lld\n",
2164                libcfs_kmem_read());
2165         LASSERT (ksocknal_data.ksnd_nnets == 0);
2166
2167         if (ksocknal_data.ksnd_init == SOCKNAL_INIT_ALL) {
2168                 unregister_netdevice_notifier(&ksocknal_dev_notifier_block);
2169                 unregister_inetaddr_notifier(&ksocknal_inetaddr_notifier_block);
2170         }
2171
2172         switch (ksocknal_data.ksnd_init) {
2173         default:
2174                 LASSERT(0);
2175                 fallthrough;
2176
2177         case SOCKNAL_INIT_ALL:
2178         case SOCKNAL_INIT_DATA:
2179                 hash_for_each(ksocknal_data.ksnd_peers, i, peer_ni, ksnp_list)
2180                         LASSERT(0);
2181
2182                 LASSERT(list_empty(&ksocknal_data.ksnd_nets));
2183                 LASSERT(list_empty(&ksocknal_data.ksnd_enomem_conns));
2184                 LASSERT(list_empty(&ksocknal_data.ksnd_zombie_conns));
2185                 LASSERT(list_empty(&ksocknal_data.ksnd_connd_connreqs));
2186                 LASSERT(list_empty(&ksocknal_data.ksnd_connd_routes));
2187
2188                 if (ksocknal_data.ksnd_schedulers != NULL) {
2189                         cfs_percpt_for_each(sched, i,
2190                                             ksocknal_data.ksnd_schedulers) {
2191
2192                                 LASSERT(list_empty(&sched->kss_tx_conns));
2193                                 LASSERT(list_empty(&sched->kss_rx_conns));
2194                                 LASSERT(list_empty(&sched->kss_zombie_noop_txs));
2195                                 LASSERT(sched->kss_nconns == 0);
2196                         }
2197                 }
2198
2199                 /* flag threads to terminate; wake and wait for them to die */
2200                 ksocknal_data.ksnd_shuttingdown = 1;
2201                 wake_up_all(&ksocknal_data.ksnd_connd_waitq);
2202                 wake_up(&ksocknal_data.ksnd_reaper_waitq);
2203
2204                 if (ksocknal_data.ksnd_schedulers != NULL) {
2205                         cfs_percpt_for_each(sched, i,
2206                                             ksocknal_data.ksnd_schedulers)
2207                                         wake_up_all(&sched->kss_waitq);
2208                 }
2209
2210                 wait_var_event_warning(&ksocknal_data.ksnd_nthreads,
2211                                        atomic_read(&ksocknal_data.ksnd_nthreads) == 0,
2212                                        "waiting for %d threads to terminate\n",
2213                                        atomic_read(&ksocknal_data.ksnd_nthreads));
2214
2215                 ksocknal_free_buffers();
2216
2217                 ksocknal_data.ksnd_init = SOCKNAL_INIT_NOTHING;
2218                 break;
2219         }
2220
2221         CDEBUG(D_MALLOC, "after NAL cleanup: kmem %lld\n",
2222                libcfs_kmem_read());
2223
2224         module_put(THIS_MODULE);
2225 }
2226
2227 static int
2228 ksocknal_base_startup(void)
2229 {
2230         struct ksock_sched *sched;
2231         int rc;
2232         int i;
2233
2234         LASSERT(ksocknal_data.ksnd_init == SOCKNAL_INIT_NOTHING);
2235         LASSERT(ksocknal_data.ksnd_nnets == 0);
2236
2237         memset(&ksocknal_data, 0, sizeof(ksocknal_data)); /* zero pointers */
2238
2239         hash_init(ksocknal_data.ksnd_peers);
2240
2241         rwlock_init(&ksocknal_data.ksnd_global_lock);
2242         INIT_LIST_HEAD(&ksocknal_data.ksnd_nets);
2243
2244         spin_lock_init(&ksocknal_data.ksnd_reaper_lock);
2245         INIT_LIST_HEAD(&ksocknal_data.ksnd_enomem_conns);
2246         INIT_LIST_HEAD(&ksocknal_data.ksnd_zombie_conns);
2247         INIT_LIST_HEAD(&ksocknal_data.ksnd_deathrow_conns);
2248         init_waitqueue_head(&ksocknal_data.ksnd_reaper_waitq);
2249
2250         spin_lock_init(&ksocknal_data.ksnd_connd_lock);
2251         INIT_LIST_HEAD(&ksocknal_data.ksnd_connd_connreqs);
2252         INIT_LIST_HEAD(&ksocknal_data.ksnd_connd_routes);
2253         init_waitqueue_head(&ksocknal_data.ksnd_connd_waitq);
2254
2255         spin_lock_init(&ksocknal_data.ksnd_tx_lock);
2256         INIT_LIST_HEAD(&ksocknal_data.ksnd_idle_noop_txs);
2257
2258         /* NB memset above zeros whole of ksocknal_data */
2259
2260         /* flag lists/ptrs/locks initialised */
2261         ksocknal_data.ksnd_init = SOCKNAL_INIT_DATA;
2262         if (!try_module_get(THIS_MODULE))
2263                 goto failed;
2264
2265         /* Create a scheduler block per available CPT */
2266         ksocknal_data.ksnd_schedulers = cfs_percpt_alloc(lnet_cpt_table(),
2267                                                          sizeof(*sched));
2268         if (ksocknal_data.ksnd_schedulers == NULL)
2269                 goto failed;
2270
2271         cfs_percpt_for_each(sched, i, ksocknal_data.ksnd_schedulers) {
2272                 int nthrs;
2273
2274                 /*
2275                  * make sure not to allocate more threads than there are
2276                  * cores/CPUs in teh CPT
2277                  */
2278                 nthrs = cfs_cpt_weight(lnet_cpt_table(), i);
2279                 if (*ksocknal_tunables.ksnd_nscheds > 0) {
2280                         nthrs = min(nthrs, *ksocknal_tunables.ksnd_nscheds);
2281                 } else {
2282                         /*
2283                          * max to half of CPUs, assume another half should be
2284                          * reserved for upper layer modules
2285                          */
2286                         nthrs = min(max(SOCKNAL_NSCHEDS, nthrs >> 1), nthrs);
2287                 }
2288
2289                 sched->kss_nthreads_max = nthrs;
2290                 sched->kss_cpt = i;
2291
2292                 spin_lock_init(&sched->kss_lock);
2293                 INIT_LIST_HEAD(&sched->kss_rx_conns);
2294                 INIT_LIST_HEAD(&sched->kss_tx_conns);
2295                 INIT_LIST_HEAD(&sched->kss_zombie_noop_txs);
2296                 init_waitqueue_head(&sched->kss_waitq);
2297         }
2298
2299         ksocknal_data.ksnd_connd_starting         = 0;
2300         ksocknal_data.ksnd_connd_failed_stamp     = 0;
2301         ksocknal_data.ksnd_connd_starting_stamp   = ktime_get_real_seconds();
2302         /* must have at least 2 connds to remain responsive to accepts while
2303          * connecting */
2304         if (*ksocknal_tunables.ksnd_nconnds < SOCKNAL_CONND_RESV + 1)
2305                 *ksocknal_tunables.ksnd_nconnds = SOCKNAL_CONND_RESV + 1;
2306
2307         if (*ksocknal_tunables.ksnd_nconnds_max <
2308             *ksocknal_tunables.ksnd_nconnds) {
2309                 ksocknal_tunables.ksnd_nconnds_max =
2310                         ksocknal_tunables.ksnd_nconnds;
2311         }
2312
2313         for (i = 0; i < *ksocknal_tunables.ksnd_nconnds; i++) {
2314                 spin_lock_bh(&ksocknal_data.ksnd_connd_lock);
2315                 ksocknal_data.ksnd_connd_starting++;
2316                 spin_unlock_bh(&ksocknal_data.ksnd_connd_lock);
2317
2318                 rc = ksocknal_thread_start(ksocknal_connd,
2319                                            (void *)((uintptr_t)i),
2320                                            "socknal_cd%02d", i);
2321                 if (rc != 0) {
2322                         spin_lock_bh(&ksocknal_data.ksnd_connd_lock);
2323                         ksocknal_data.ksnd_connd_starting--;
2324                         spin_unlock_bh(&ksocknal_data.ksnd_connd_lock);
2325                         CERROR("Can't spawn socknal connd: %d\n", rc);
2326                         goto failed;
2327                 }
2328         }
2329
2330         rc = ksocknal_thread_start(ksocknal_reaper, NULL, "socknal_reaper");
2331         if (rc != 0) {
2332                 CERROR ("Can't spawn socknal reaper: %d\n", rc);
2333                 goto failed;
2334         }
2335
2336         register_netdevice_notifier(&ksocknal_dev_notifier_block);
2337         register_inetaddr_notifier(&ksocknal_inetaddr_notifier_block);
2338
2339         /* flag everything initialised */
2340         ksocknal_data.ksnd_init = SOCKNAL_INIT_ALL;
2341
2342         return 0;
2343
2344  failed:
2345         ksocknal_base_shutdown();
2346         return -ENETDOWN;
2347 }
2348
2349 static int
2350 ksocknal_debug_peerhash(struct lnet_ni *ni)
2351 {
2352         struct ksock_peer_ni *peer_ni;
2353         int i;
2354
2355         read_lock(&ksocknal_data.ksnd_global_lock);
2356
2357         hash_for_each(ksocknal_data.ksnd_peers, i, peer_ni, ksnp_list) {
2358                 struct ksock_conn_cb *conn_cb;
2359                 struct ksock_conn *conn;
2360
2361                 if (peer_ni->ksnp_ni != ni)
2362                         continue;
2363
2364                 CWARN("Active peer_ni on shutdown: %s, ref %d, closing %d, accepting %d, err %d, zcookie %llu, txq %d, zc_req %d\n",
2365                       libcfs_idstr(&peer_ni->ksnp_id),
2366                       refcount_read(&peer_ni->ksnp_refcount),
2367                       peer_ni->ksnp_closing,
2368                       peer_ni->ksnp_accepting, peer_ni->ksnp_error,
2369                       peer_ni->ksnp_zc_next_cookie,
2370                       !list_empty(&peer_ni->ksnp_tx_queue),
2371                       !list_empty(&peer_ni->ksnp_zc_req_list));
2372
2373                 conn_cb = peer_ni->ksnp_conn_cb;
2374                 if (conn_cb) {
2375                         CWARN("ConnCB: ref %d, schd %d, conn %d, cnted %d, del %d\n",
2376                               refcount_read(&conn_cb->ksnr_refcount),
2377                               conn_cb->ksnr_scheduled, conn_cb->ksnr_connecting,
2378                               conn_cb->ksnr_connected, conn_cb->ksnr_deleted);
2379                 }
2380
2381                 list_for_each_entry(conn, &peer_ni->ksnp_conns, ksnc_list) {
2382                         CWARN("Conn: ref %d, sref %d, t %d, c %d\n",
2383                               refcount_read(&conn->ksnc_conn_refcount),
2384                               refcount_read(&conn->ksnc_sock_refcount),
2385                               conn->ksnc_type, conn->ksnc_closing);
2386                 }
2387                 break;
2388         }
2389
2390         read_unlock(&ksocknal_data.ksnd_global_lock);
2391         return 0;
2392 }
2393
2394 void
2395 ksocknal_shutdown(struct lnet_ni *ni)
2396 {
2397         struct ksock_net *net = ni->ni_data;
2398
2399         LASSERT(ksocknal_data.ksnd_init == SOCKNAL_INIT_ALL);
2400         LASSERT(ksocknal_data.ksnd_nnets > 0);
2401
2402         /* prevent new peers */
2403         atomic_add(SOCKNAL_SHUTDOWN_BIAS, &net->ksnn_npeers);
2404
2405         /* Delete all peers */
2406         ksocknal_del_peer(ni, NULL);
2407
2408         /* Wait for all peer_ni state to clean up */
2409         wait_var_event_warning(&net->ksnn_npeers,
2410                                atomic_read(&net->ksnn_npeers) ==
2411                                SOCKNAL_SHUTDOWN_BIAS,
2412                                "waiting for %d peers to disconnect\n",
2413                                ksocknal_debug_peerhash(ni) +
2414                                atomic_read(&net->ksnn_npeers) -
2415                                SOCKNAL_SHUTDOWN_BIAS);
2416
2417         LASSERT(net->ksnn_interface.ksni_npeers == 0);
2418         LASSERT(net->ksnn_interface.ksni_nroutes == 0);
2419
2420         list_del(&net->ksnn_list);
2421         LIBCFS_FREE(net, sizeof(*net));
2422
2423         ksocknal_data.ksnd_nnets--;
2424         if (ksocknal_data.ksnd_nnets == 0)
2425                 ksocknal_base_shutdown();
2426 }
2427
2428 static int
2429 ksocknal_search_new_ipif(struct ksock_net *net)
2430 {
2431         int new_ipif = 0;
2432         char *ifnam = &net->ksnn_interface.ksni_name[0];
2433         char *colon = strchr(ifnam, ':');
2434         bool found = false;
2435         struct ksock_net *tmp;
2436
2437         if (colon != NULL)
2438                 *colon = 0;
2439
2440         list_for_each_entry(tmp, &ksocknal_data.ksnd_nets, ksnn_list) {
2441                 char *ifnam2 = &tmp->ksnn_interface.ksni_name[0];
2442                 char *colon2 = strchr(ifnam2, ':');
2443
2444                 if (colon2 != NULL)
2445                         *colon2 = 0;
2446
2447                 found = strcmp(ifnam, ifnam2) == 0;
2448                 if (colon2 != NULL)
2449                         *colon2 = ':';
2450         }
2451
2452         new_ipif += !found;
2453         if (colon != NULL)
2454                 *colon = ':';
2455
2456         return new_ipif;
2457 }
2458
2459 static int
2460 ksocknal_start_schedulers(struct ksock_sched *sched)
2461 {
2462         int     nthrs;
2463         int     rc = 0;
2464         int     i;
2465
2466         if (sched->kss_nthreads == 0) {
2467                 if (*ksocknal_tunables.ksnd_nscheds > 0) {
2468                         nthrs = sched->kss_nthreads_max;
2469                 } else {
2470                         nthrs = cfs_cpt_weight(lnet_cpt_table(),
2471                                                sched->kss_cpt);
2472                         nthrs = min(max(SOCKNAL_NSCHEDS, nthrs >> 1), nthrs);
2473                         nthrs = min(SOCKNAL_NSCHEDS_HIGH, nthrs);
2474                 }
2475                 nthrs = min(nthrs, sched->kss_nthreads_max);
2476         } else {
2477                 LASSERT(sched->kss_nthreads <= sched->kss_nthreads_max);
2478                 /* increase two threads if there is new interface */
2479                 nthrs = min(2, sched->kss_nthreads_max - sched->kss_nthreads);
2480         }
2481
2482         for (i = 0; i < nthrs; i++) {
2483                 long id;
2484
2485                 id = KSOCK_THREAD_ID(sched->kss_cpt, sched->kss_nthreads + i);
2486                 rc = ksocknal_thread_start(ksocknal_scheduler, (void *)id,
2487                                            "socknal_sd%02d_%02d",
2488                                            sched->kss_cpt,
2489                                            (int)KSOCK_THREAD_SID(id));
2490                 if (rc == 0)
2491                         continue;
2492
2493                 CERROR("Can't spawn thread %d for scheduler[%d]: %d\n",
2494                        sched->kss_cpt, (int) KSOCK_THREAD_SID(id), rc);
2495                 break;
2496         }
2497
2498         sched->kss_nthreads += i;
2499         return rc;
2500 }
2501
2502 static int
2503 ksocknal_net_start_threads(struct ksock_net *net, __u32 *cpts, int ncpts)
2504 {
2505         int newif = ksocknal_search_new_ipif(net);
2506         int rc;
2507         int i;
2508
2509         if (ncpts > 0 && ncpts > cfs_cpt_number(lnet_cpt_table()))
2510                 return -EINVAL;
2511
2512         for (i = 0; i < ncpts; i++) {
2513                 struct ksock_sched *sched;
2514                 int cpt = (cpts == NULL) ? i : cpts[i];
2515
2516                 LASSERT(cpt < cfs_cpt_number(lnet_cpt_table()));
2517                 sched = ksocknal_data.ksnd_schedulers[cpt];
2518
2519                 if (!newif && sched->kss_nthreads > 0)
2520                         continue;
2521
2522                 rc = ksocknal_start_schedulers(sched);
2523                 if (rc != 0)
2524                         return rc;
2525         }
2526         return 0;
2527 }
2528
2529 int
2530 ksocknal_startup(struct lnet_ni *ni)
2531 {
2532         struct ksock_net *net;
2533         struct ksock_interface *ksi = NULL;
2534         struct lnet_inetdev *ifaces = NULL;
2535         int rc, if_idx;
2536
2537         LASSERT (ni->ni_net->net_lnd == &the_ksocklnd);
2538         if (ksocknal_data.ksnd_init == SOCKNAL_INIT_NOTHING) {
2539                 rc = ksocknal_base_startup();
2540                 if (rc != 0)
2541                         return rc;
2542         }
2543         LIBCFS_ALLOC(net, sizeof(*net));
2544         if (net == NULL)
2545                 goto out_base;
2546
2547         net->ksnn_incarnation = ktime_get_real_ns();
2548         ni->ni_data = net;
2549
2550         ksocknal_tunables_setup(ni);
2551
2552         rc = lnet_inet_enumerate(&ifaces, ni->ni_net_ns, true);
2553         if (rc < 0)
2554                 goto out_net;
2555
2556         ksi = &net->ksnn_interface;
2557
2558         /* Interface and/or IP address is specified otherwise default to
2559          * the first Interface
2560          */
2561         if_idx = lnet_inet_select(ni, ifaces, rc);
2562         if (if_idx < 0)
2563                 goto out_net;
2564
2565         if (!ni->ni_interface) {
2566                 rc = lnet_ni_add_interface(ni, ifaces[if_idx].li_name);
2567                 if (rc < 0)
2568                         CWARN("ksocklnd failed to allocate ni_interface\n");
2569         }
2570
2571         ni->ni_dev_cpt = ifaces[if_idx].li_cpt;
2572         ksi->ksni_index = ifaces[if_idx].li_index;
2573         if (ifaces[if_idx].li_ipv6) {
2574                 struct sockaddr_in6 *sa;
2575                 sa = (void *)&ksi->ksni_addr;
2576                 memset(sa, 0, sizeof(*sa));
2577                 sa->sin6_family = AF_INET6;
2578                 memcpy(&sa->sin6_addr, ifaces[if_idx].li_ipv6addr,
2579                        sizeof(struct in6_addr));
2580                 ni->ni_nid.nid_size = sizeof(struct in6_addr) - 4;
2581                 memcpy(&ni->ni_nid.nid_addr, ifaces[if_idx].li_ipv6addr,
2582                        sizeof(struct in6_addr));
2583         } else {
2584                 struct sockaddr_in *sa;
2585                 sa = (void *)&ksi->ksni_addr;
2586                 memset(sa, 0, sizeof(*sa));
2587                 sa->sin_family = AF_INET;
2588                 sa->sin_addr.s_addr = htonl(ifaces[if_idx].li_ipaddr);
2589                 ksi->ksni_netmask = ifaces[if_idx].li_netmask;
2590                 ni->ni_nid.nid_size = 4 - 4;
2591                 ni->ni_nid.nid_addr[0] = sa->sin_addr.s_addr;
2592         }
2593         strlcpy(ksi->ksni_name, ifaces[if_idx].li_name, sizeof(ksi->ksni_name));
2594
2595         /* call it before add it to ksocknal_data.ksnd_nets */
2596         rc = ksocknal_net_start_threads(net, ni->ni_cpts, ni->ni_ncpts);
2597         if (rc != 0)
2598                 goto out_net;
2599
2600         list_add(&net->ksnn_list, &ksocknal_data.ksnd_nets);
2601         net->ksnn_ni = ni;
2602         ksocknal_data.ksnd_nnets++;
2603
2604         return 0;
2605
2606 out_net:
2607         LIBCFS_FREE(net, sizeof(*net));
2608 out_base:
2609         if (ksocknal_data.ksnd_nnets == 0)
2610                 ksocknal_base_shutdown();
2611
2612         return -ENETDOWN;
2613 }
2614
2615 static void __exit ksocklnd_exit(void)
2616 {
2617         lnet_unregister_lnd(&the_ksocklnd);
2618 }
2619
2620 static const struct lnet_lnd the_ksocklnd = {
2621         .lnd_type               = SOCKLND,
2622         .lnd_startup            = ksocknal_startup,
2623         .lnd_shutdown           = ksocknal_shutdown,
2624         .lnd_ctl                = ksocknal_ctl,
2625         .lnd_send               = ksocknal_send,
2626         .lnd_recv               = ksocknal_recv,
2627         .lnd_notify_peer_down   = ksocknal_notify_gw_down,
2628         .lnd_accept             = ksocknal_accept,
2629         .lnd_nl_set             = ksocknal_nl_set,
2630         .lnd_keys               = &ksocknal_tunables_keys,
2631 };
2632
2633 static int __init ksocklnd_init(void)
2634 {
2635         int rc;
2636
2637         /* check ksnr_connected/connecting field large enough */
2638         BUILD_BUG_ON(SOCKLND_CONN_NTYPES > 4);
2639         BUILD_BUG_ON(SOCKLND_CONN_ACK != SOCKLND_CONN_BULK_IN);
2640
2641         rc = ksocknal_tunables_init();
2642         if (rc != 0)
2643                 return rc;
2644
2645         lnet_register_lnd(&the_ksocklnd);
2646
2647         return 0;
2648 }
2649
2650 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
2651 MODULE_DESCRIPTION("TCP Socket LNet Network Driver");
2652 MODULE_VERSION("2.8.0");
2653 MODULE_LICENSE("GPL");
2654
2655 module_init(ksocklnd_init);
2656 module_exit(ksocklnd_exit);