lnet/lnet/module.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) 2004, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2012, 2016, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  * Lustre is a trademark of Sun Microsystems, Inc.
  31  */
  32
  33 #define DEBUG_SUBSYSTEM S_LNET
  34 #include <lnet/lib-lnet.h>
  35 #include <lnet/lib-dlc.h>
  36
  37 static int config_on_load = 0;
  38 module_param(config_on_load, int, 0444);
  39 MODULE_PARM_DESC(config_on_load, "configure network at module load");
  40
  41 static struct mutex lnet_config_mutex;
  42
  43 static int
  44 lnet_configure(void *arg)
  45 {
  46         /* 'arg' only there so I can be passed to cfs_create_thread() */
  47         int    rc = 0;
  48
  49         mutex_lock(&lnet_config_mutex);
  50
  51         if (!the_lnet.ln_niinit_self) {
  52                 rc = try_module_get(THIS_MODULE);
  53
  54                 if (rc != 1)
  55                         goto out;
  56
  57                 rc = LNetNIInit(LNET_PID_LUSTRE);
  58                 if (rc >= 0) {
  59                         the_lnet.ln_niinit_self = 1;
  60                         rc = 0;
  61                 } else {
  62                         module_put(THIS_MODULE);
  63                 }
  64         }
  65
  66 out:
  67         mutex_unlock(&lnet_config_mutex);
  68         return rc;
  69 }
  70
  71 static int
  72 lnet_unconfigure (void)
  73 {
  74         int   refcount;
  75
  76         mutex_lock(&lnet_config_mutex);
  77
  78         if (the_lnet.ln_niinit_self) {
  79                 the_lnet.ln_niinit_self = 0;
  80                 LNetNIFini();
  81                 module_put(THIS_MODULE);
  82         }
  83
  84         mutex_lock(&the_lnet.ln_api_mutex);
  85         refcount = the_lnet.ln_refcount;
  86         mutex_unlock(&the_lnet.ln_api_mutex);
  87
  88         mutex_unlock(&lnet_config_mutex);
  89
  90         return (refcount == 0) ? 0 : -EBUSY;
  91 }
  92
  93 static int
  94 lnet_dyn_configure_net(struct libcfs_ioctl_hdr *hdr)
  95 {
  96         struct lnet_ioctl_config_data *conf =
  97           (struct lnet_ioctl_config_data *)hdr;
  98         int                           rc;
  99
 100         if (conf->cfg_hdr.ioc_len < sizeof(*conf))
 101                 return -EINVAL;
 102
 103         mutex_lock(&lnet_config_mutex);
 104         if (the_lnet.ln_niinit_self)
 105                 rc = lnet_dyn_add_net(conf);
 106         else
 107                 rc = -EINVAL;
 108         mutex_unlock(&lnet_config_mutex);
 109
 110         return rc;
 111 }
 112
 113 static int
 114 lnet_dyn_unconfigure_net(struct libcfs_ioctl_hdr *hdr)
 115 {
 116         struct lnet_ioctl_config_data *conf =
 117           (struct lnet_ioctl_config_data *) hdr;
 118         int                           rc;
 119
 120         if (conf->cfg_hdr.ioc_len < sizeof(*conf))
 121                 return -EINVAL;
 122
 123         mutex_lock(&lnet_config_mutex);
 124         if (the_lnet.ln_niinit_self)
 125                 rc = lnet_dyn_del_net(conf->cfg_net);
 126         else
 127                 rc = -EINVAL;
 128         mutex_unlock(&lnet_config_mutex);
 129
 130         return rc;
 131 }
 132
 133 static int
 134 lnet_dyn_configure_ni(struct libcfs_ioctl_hdr *hdr)
 135 {
 136         struct lnet_ioctl_config_ni *conf =
 137           (struct lnet_ioctl_config_ni *)hdr;
 138         int                           rc;
 139
 140         if (conf->lic_cfg_hdr.ioc_len < sizeof(*conf))
 141                 return -EINVAL;
 142
 143         mutex_lock(&lnet_config_mutex);
 144         if (the_lnet.ln_niinit_self)
 145                 rc = lnet_dyn_add_ni(conf);
 146         else
 147                 rc = -EINVAL;
 148         mutex_unlock(&lnet_config_mutex);
 149
 150         return rc;
 151 }
 152
 153 static int
 154 lnet_dyn_unconfigure_ni(struct libcfs_ioctl_hdr *hdr)
 155 {
 156         struct lnet_ioctl_config_ni *conf =
 157           (struct lnet_ioctl_config_ni *) hdr;
 158         int                           rc;
 159
 160         if (conf->lic_cfg_hdr.ioc_len < sizeof(*conf))
 161                 return -EINVAL;
 162
 163         mutex_lock(&lnet_config_mutex);
 164         if (the_lnet.ln_niinit_self)
 165                 rc = lnet_dyn_del_ni(conf);
 166         else
 167                 rc = -EINVAL;
 168         mutex_unlock(&lnet_config_mutex);
 169
 170         return rc;
 171 }
 172
 173 static int
 174 lnet_ioctl(unsigned int cmd, struct libcfs_ioctl_hdr *hdr)
 175 {
 176         int   rc;
 177
 178         switch (cmd) {
 179         case IOC_LIBCFS_CONFIGURE: {
 180                 struct libcfs_ioctl_data *data =
 181                   (struct libcfs_ioctl_data *)hdr;
 182
 183                 if (data->ioc_hdr.ioc_len < sizeof(*data))
 184                         return -EINVAL;
 185
 186                 the_lnet.ln_nis_from_mod_params = data->ioc_flags;
 187                 return lnet_configure(NULL);
 188         }
 189
 190         case IOC_LIBCFS_UNCONFIGURE:
 191                 return lnet_unconfigure();
 192
 193         case IOC_LIBCFS_ADD_NET:
 194                 return lnet_dyn_configure_net(hdr);
 195
 196         case IOC_LIBCFS_DEL_NET:
 197                 return lnet_dyn_unconfigure_net(hdr);
 198
 199         case IOC_LIBCFS_ADD_LOCAL_NI:
 200                 return lnet_dyn_configure_ni(hdr);
 201
 202         case IOC_LIBCFS_DEL_LOCAL_NI:
 203                 return lnet_dyn_unconfigure_ni(hdr);
 204
 205         default:
 206                 /* Passing LNET_PID_ANY only gives me a ref if the net is up
 207                  * already; I'll need it to ensure the net can't go down while
 208                  * I'm called into it */
 209                 rc = LNetNIInit(LNET_PID_ANY);
 210                 if (rc >= 0) {
 211                         rc = LNetCtl(cmd, hdr);
 212                         LNetNIFini();
 213                 }
 214                 return rc;
 215         }
 216 }
 217
 218 DECLARE_IOCTL_HANDLER(lnet_ioctl_handler, lnet_ioctl);
 219
 220 static int __init lnet_init(void)
 221 {
 222         int rc;
 223         ENTRY;
 224
 225         mutex_init(&lnet_config_mutex);
 226
 227         rc = lnet_lib_init();
 228         if (rc != 0) {
 229                 CERROR("lnet_lib_init: error %d\n", rc);
 230                 RETURN(rc);
 231         }
 232
 233         rc = libcfs_register_ioctl(&lnet_ioctl_handler);
 234         LASSERT(rc == 0);
 235
 236         if (config_on_load) {
 237                 /* Have to schedule a separate thread to avoid deadlocking
 238                  * in modload */
 239                 (void)kthread_run(lnet_configure, NULL, "lnet_initd");
 240         }
 241
 242         RETURN(0);
 243 }
 244
 245 static void __exit lnet_exit(void)
 246 {
 247         int rc;
 248
 249         rc = libcfs_deregister_ioctl(&lnet_ioctl_handler);
 250         LASSERT(rc == 0);
 251
 252         lnet_lib_exit();
 253 }
 254
 255 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
 256 MODULE_DESCRIPTION("Lustre Networking layer");
 257 MODULE_VERSION(LNET_VERSION);
 258 MODULE_LICENSE("GPL");
 259
 260 module_init(lnet_init);
 261 module_exit(lnet_exit);