drivers/staging/lustre/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, 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
  35 #include "selftest.h"
  36 #include "console.h"
  37
  38 enum {
  39         LST_INIT_NONE           = 0,
  40         LST_INIT_WI_SERIAL,
  41         LST_INIT_WI_TEST,
  42         LST_INIT_RPC,
  43         LST_INIT_FW,
  44         LST_INIT_CONSOLE
  45 };
  46
  47 static int lst_init_step = LST_INIT_NONE;
  48
  49 struct cfs_wi_sched *lst_sched_serial;
  50 struct cfs_wi_sched **lst_sched_test;
  51
  52 static void
  53 lnet_selftest_exit(void)
  54 {
  55         int i;
  56
  57         switch (lst_init_step) {
  58         case LST_INIT_CONSOLE:
  59                 lstcon_console_fini();
  60         case LST_INIT_FW:
  61                 sfw_shutdown();
  62         case LST_INIT_RPC:
  63                 srpc_shutdown();
  64         case LST_INIT_WI_TEST:
  65                 for (i = 0;
  66                      i < cfs_cpt_number(lnet_cpt_table()); i++) {
  67                         if (!lst_sched_test[i])
  68                                 continue;
  69                         cfs_wi_sched_destroy(lst_sched_test[i]);
  70                 }
  71                 LIBCFS_FREE(lst_sched_test,
  72                             sizeof(lst_sched_test[0]) *
  73                             cfs_cpt_number(lnet_cpt_table()));
  74                 lst_sched_test = NULL;
  75
  76         case LST_INIT_WI_SERIAL:
  77                 cfs_wi_sched_destroy(lst_sched_serial);
  78                 lst_sched_serial = NULL;
  79         case LST_INIT_NONE:
  80                 break;
  81         default:
  82                 LBUG();
  83         }
  84 }
  85
  86 static int
  87 lnet_selftest_init(void)
  88 {
  89         int nscheds;
  90         int rc;
  91         int i;
  92
  93         rc = cfs_wi_sched_create("lst_s", lnet_cpt_table(), CFS_CPT_ANY,
  94                                  1, &lst_sched_serial);
  95         if (rc) {
  96                 CERROR("Failed to create serial WI scheduler for LST\n");
  97                 return rc;
  98         }
  99         lst_init_step = LST_INIT_WI_SERIAL;
 100
 101         nscheds = cfs_cpt_number(lnet_cpt_table());
 102         LIBCFS_ALLOC(lst_sched_test, sizeof(lst_sched_test[0]) * nscheds);
 103         if (!lst_sched_test)
 104                 goto error;
 105
 106         lst_init_step = LST_INIT_WI_TEST;
 107         for (i = 0; i < nscheds; i++) {
 108                 int nthrs = cfs_cpt_weight(lnet_cpt_table(), i);
 109
 110                 /* reserve at least one CPU for LND */
 111                 nthrs = max(nthrs - 1, 1);
 112                 rc = cfs_wi_sched_create("lst_t", lnet_cpt_table(), i,
 113                                          nthrs, &lst_sched_test[i]);
 114                 if (rc) {
 115                         CERROR("Failed to create CPT affinity WI scheduler %d for LST\n", i);
 116                         goto error;
 117                 }
 118         }
 119
 120         rc = srpc_startup();
 121         if (rc) {
 122                 CERROR("LST can't startup rpc\n");
 123                 goto error;
 124         }
 125         lst_init_step = LST_INIT_RPC;
 126
 127         rc = sfw_startup();
 128         if (rc) {
 129                 CERROR("LST can't startup framework\n");
 130                 goto error;
 131         }
 132         lst_init_step = LST_INIT_FW;
 133
 134         rc = lstcon_console_init();
 135         if (rc) {
 136                 CERROR("LST can't startup console\n");
 137                 goto error;
 138         }
 139         lst_init_step = LST_INIT_CONSOLE;
 140         return 0;
 141 error:
 142         lnet_selftest_exit();
 143         return rc;
 144 }
 145
 146 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
 147 MODULE_DESCRIPTION("LNet Selftest");
 148 MODULE_VERSION("2.7.0");
 149 MODULE_LICENSE("GPL");
 150
 151 module_init(lnet_selftest_init);
 152 module_exit(lnet_selftest_exit);