lnet/selftest/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) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2012, 2015, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  */
  31
  32 #define DEBUG_SUBSYSTEM S_LNET
  33
  34 #include "selftest.h"
  35 #include "console.h"
  36
  37 enum {
  38         LST_INIT_NONE           = 0,
  39         LST_INIT_WI_SERIAL,
  40         LST_INIT_WI_TEST,
  41         LST_INIT_RPC,
  42         LST_INIT_FW,
  43         LST_INIT_CONSOLE
  44 };
  45
  46 static int lst_init_step = LST_INIT_NONE;
  47
  48 struct workqueue_struct *lst_serial_wq;
  49 struct workqueue_struct **lst_test_wq;
  50
  51 static void
  52 lnet_selftest_exit(void)
  53 {
  54         int i;
  55
  56         switch (lst_init_step) {
  57         case LST_INIT_CONSOLE:
  58                 lstcon_console_fini();
  59                 fallthrough;
  60         case LST_INIT_FW:
  61                 sfw_shutdown();
  62                 fallthrough;
  63         case LST_INIT_RPC:
  64                 srpc_shutdown();
  65                 fallthrough;
  66         case LST_INIT_WI_TEST:
  67                 for (i = 0;
  68                      i < cfs_cpt_number(lnet_cpt_table()); i++) {
  69                         if (!lst_test_wq[i])
  70                                 continue;
  71                         destroy_workqueue(lst_test_wq[i]);
  72                 }
  73                 CFS_FREE_PTR_ARRAY(lst_test_wq,
  74                                    cfs_cpt_number(lnet_cpt_table()));
  75                 lst_test_wq = NULL;
  76                 fallthrough;
  77         case LST_INIT_WI_SERIAL:
  78                 destroy_workqueue(lst_serial_wq);
  79                 lst_serial_wq = NULL;
  80                 fallthrough;
  81         case LST_INIT_NONE:
  82                 break;
  83         default:
  84                 LBUG();
  85         }
  86 }
  87
  88 void
  89 lnet_selftest_structure_assertion(void)
  90 {
  91         BUILD_BUG_ON(sizeof(struct srpc_msg) != 160);
  92         BUILD_BUG_ON(sizeof(struct srpc_test_reqst) != 70);
  93         BUILD_BUG_ON(offsetof(struct srpc_msg, msg_body.tes_reqst.tsr_concur) !=
  94                      72);
  95         BUILD_BUG_ON(offsetof(struct srpc_msg, msg_body.tes_reqst.tsr_ndest) !=
  96                               78);
  97         BUILD_BUG_ON(sizeof(struct srpc_stat_reply) != 136);
  98         BUILD_BUG_ON(sizeof(struct srpc_stat_reqst) != 28);
  99 }
 100
 101 static int __init
 102 lnet_selftest_init(void)
 103 {
 104         int nscheds;
 105         int rc = -ENOMEM;
 106         int i;
 107
 108         lst_serial_wq = alloc_ordered_workqueue("lst_s", 0);
 109         if (!lst_serial_wq) {
 110                 CERROR("Failed to create serial WI scheduler for LST\n");
 111                 return rc;
 112         }
 113         lst_init_step = LST_INIT_WI_SERIAL;
 114
 115         nscheds = cfs_cpt_number(lnet_cpt_table());
 116         CFS_ALLOC_PTR_ARRAY(lst_test_wq, nscheds);
 117         if (!lst_test_wq) {
 118                 rc = -ENOMEM;
 119                 goto error;
 120         }
 121
 122         lst_init_step = LST_INIT_WI_TEST;
 123         for (i = 0; i < nscheds; i++) {
 124                 int nthrs = cfs_cpt_weight(lnet_cpt_table(), i);
 125
 126                 /* reserve at least one CPU for LND */
 127                 nthrs = max(nthrs - 1, 1);
 128                 lst_test_wq[i] = cfs_cpt_bind_workqueue("lst_t",
 129                                                         lnet_cpt_table(), 0,
 130                                                         i, nthrs);
 131                 if (IS_ERR(lst_test_wq[i])) {
 132                         rc = PTR_ERR(lst_test_wq[i]);
 133                         CERROR("Failed to create CPU partition affinity WI scheduler %d for LST: rc = %d\n",
 134                                i, rc);
 135                         lst_test_wq[i] = NULL;
 136                         goto error;
 137                 }
 138         }
 139
 140         rc = srpc_startup();
 141         if (rc != 0) {
 142                 CERROR("LST can't startup rpc\n");
 143                 goto error;
 144         }
 145         lst_init_step = LST_INIT_RPC;
 146
 147         rc = sfw_startup();
 148         if (rc != 0) {
 149                 CERROR("LST can't startup framework\n");
 150                 goto error;
 151         }
 152         lst_init_step = LST_INIT_FW;
 153
 154         rc = lstcon_console_init();
 155         if (rc != 0) {
 156                 CERROR("LST can't startup console\n");
 157                 goto error;
 158         }
 159         lst_init_step = LST_INIT_CONSOLE;
 160         return 0;
 161 error:
 162         lnet_selftest_exit();
 163         return rc;
 164 }
 165
 166 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
 167 MODULE_DESCRIPTION("LNet Selftest");
 168 MODULE_VERSION("2.8.0");
 169 MODULE_LICENSE("GPL");
 170
 171 module_init(lnet_selftest_init);
 172 module_exit(lnet_selftest_exit);