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(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 = -EINVAL;
 106                 goto out_unlock;
 107         }
 108         rc = lnet_dyn_add_ni(LNET_PID_LUSTRE, conf);
 109 out_unlock:
 110         mutex_unlock(&lnet_config_mutex);
 111
 112         return rc;
 113 }
 114
 115 static int
 116 lnet_dyn_unconfigure(struct libcfs_ioctl_hdr *hdr)
 117 {
 118         struct lnet_ioctl_config_data *conf =
 119           (struct lnet_ioctl_config_data *) hdr;
 120         int                           rc;
 121
 122         if (conf->cfg_hdr.ioc_len < sizeof(*conf))
 123                 return -EINVAL;
 124
 125         mutex_lock(&lnet_config_mutex);
 126         if (!the_lnet.ln_niinit_self) {
 127                 rc = -EINVAL;
 128                 goto out_unlock;
 129         }
 130         rc = lnet_dyn_del_ni(conf->cfg_net);
 131 out_unlock:
 132         mutex_unlock(&lnet_config_mutex);
 133
 134         return rc;
 135 }
 136
 137 static int
 138 lnet_ioctl(unsigned int cmd, struct libcfs_ioctl_hdr *hdr)
 139 {
 140         int   rc;
 141
 142         switch (cmd) {
 143         case IOC_LIBCFS_CONFIGURE: {
 144                 struct libcfs_ioctl_data *data =
 145                   (struct libcfs_ioctl_data *)hdr;
 146
 147                 if (data->ioc_hdr.ioc_len < sizeof(*data))
 148                         return -EINVAL;
 149
 150                 the_lnet.ln_nis_from_mod_params = data->ioc_flags;
 151                 return lnet_configure(NULL);
 152         }
 153
 154         case IOC_LIBCFS_UNCONFIGURE:
 155                 return lnet_unconfigure();
 156
 157         case IOC_LIBCFS_ADD_NET:
 158                 return lnet_dyn_configure(hdr);
 159
 160         case IOC_LIBCFS_DEL_NET:
 161                 return lnet_dyn_unconfigure(hdr);
 162
 163         default:
 164                 /* Passing LNET_PID_ANY only gives me a ref if the net is up
 165                  * already; I'll need it to ensure the net can't go down while
 166                  * I'm called into it */
 167                 rc = LNetNIInit(LNET_PID_ANY);
 168                 if (rc >= 0) {
 169                         rc = LNetCtl(cmd, hdr);
 170                         LNetNIFini();
 171                 }
 172                 return rc;
 173         }
 174 }
 175
 176 DECLARE_IOCTL_HANDLER(lnet_ioctl_handler, lnet_ioctl);
 177
 178 static int __init lnet_init(void)
 179 {
 180         int rc;
 181         ENTRY;
 182
 183         mutex_init(&lnet_config_mutex);
 184
 185         rc = lnet_lib_init();
 186         if (rc != 0) {
 187                 CERROR("lnet_lib_init: error %d\n", rc);
 188                 RETURN(rc);
 189         }
 190
 191         rc = libcfs_register_ioctl(&lnet_ioctl_handler);
 192         LASSERT(rc == 0);
 193
 194         if (config_on_load) {
 195                 /* Have to schedule a separate thread to avoid deadlocking
 196                  * in modload */
 197                 (void)kthread_run(lnet_configure, NULL, "lnet_initd");
 198         }
 199
 200         RETURN(0);
 201 }
 202
 203 static void __exit lnet_exit(void)
 204 {
 205         int rc;
 206
 207         rc = libcfs_deregister_ioctl(&lnet_ioctl_handler);
 208         LASSERT(rc == 0);
 209
 210         lnet_lib_exit();
 211 }
 212
 213 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
 214 MODULE_DESCRIPTION("Lustre Networking layer");
 215 MODULE_VERSION(LNET_VERSION);
 216 MODULE_LICENSE("GPL");
 217
 218 module_init(lnet_init);
 219 module_exit(lnet_exit);