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