usr.sbin/devctl/devctl.c

   1 /*-
   2  * Copyright (c) 2014 John Baldwin <jhb@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include <sys/linker_set.h>
  31 #include <devctl.h>
  32 #include <err.h>
  33 #include <errno.h>
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <string.h>
  37 #include <strings.h>
  38 #include <unistd.h>
  39
  40 struct devctl_command {
  41         const char *name;
  42         int (*handler)(int ac, char **av);
  43 };
  44
  45 #define DEVCTL_DATASET(name)    devctl_ ## name ## _table
  46
  47 #define DEVCTL_COMMAND(set, name, function)                             \
  48         static struct devctl_command function ## _devctl_command =      \
  49         { #name, function };                                            \
  50         DATA_SET(DEVCTL_DATASET(set), function ## _devctl_command)
  51
  52 #define DEVCTL_TABLE(set, name)                                         \
  53         SET_DECLARE(DEVCTL_DATASET(name), struct devctl_command);       \
  54                                                                         \
  55         static int                                                      \
  56         devctl_ ## name ## _table_handler(int ac, char **av)            \
  57         {                                                               \
  58                 return (devctl_table_handler(SET_BEGIN(DEVCTL_DATASET(name)), \
  59                     SET_LIMIT(DEVCTL_DATASET(name)), ac, av));          \
  60         }                                                               \
  61         DEVCTL_COMMAND(set, name, devctl_ ## name ## _table_handler)
  62
  63 static int      devctl_table_handler(struct devctl_command **start,
  64     struct devctl_command **end, int ac, char **av);
  65
  66 SET_DECLARE(DEVCTL_DATASET(top), struct devctl_command);
  67
  68 DEVCTL_TABLE(top, clear);
  69 DEVCTL_TABLE(top, set);
  70
  71 static void
  72 usage(void)
  73 {
  74         fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n",
  75             "usage: devctl attach device",
  76             "       devctl detach [-f] device",
  77             "       devctl disable [-f] device",
  78             "       devctl enable device",
  79             "       devctl set driver [-f] device driver",
  80             "       devctl clear driver [-f] device");
  81         exit(1);
  82 }
  83
  84 static int
  85 devctl_table_handler(struct devctl_command **start,
  86     struct devctl_command **end, int ac, char **av)
  87 {
  88         struct devctl_command **cmd;
  89
  90         if (ac < 2) {
  91                 warnx("The %s command requires a sub-command.", av[0]);
  92                 return (EINVAL);
  93         }
  94         for (cmd = start; cmd < end; cmd++) {
  95                 if (strcmp((*cmd)->name, av[1]) == 0)
  96                         return ((*cmd)->handler(ac - 1, av + 1));
  97         }
  98
  99         warnx("%s is not a valid sub-command of %s.", av[1], av[0]);
 100         return (ENOENT);
 101 }
 102
 103 static int
 104 help(int ac __unused, char **av __unused)
 105 {
 106
 107         usage();
 108         return (0);
 109 }
 110 DEVCTL_COMMAND(top, help, help);
 111
 112 static int
 113 attach(int ac, char **av)
 114 {
 115
 116         if (ac != 2)
 117                 usage();
 118         if (devctl_attach(av[1]) < 0)
 119                 err(1, "Failed to attach %s", av[1]);
 120         return (0);
 121 }
 122 DEVCTL_COMMAND(top, attach, attach);
 123
 124 static void
 125 detach_usage(void)
 126 {
 127
 128         fprintf(stderr, "usage: devctl detach [-f] device\n");
 129         exit(1);
 130 }
 131
 132 static int
 133 detach(int ac, char **av)
 134 {
 135         bool force;
 136         int ch;
 137
 138         force = false;
 139         while ((ch = getopt(ac, av, "f")) != -1)
 140                 switch (ch) {
 141                 case 'f':
 142                         force = true;
 143                         break;
 144                 default:
 145                         detach_usage();
 146                 }
 147         ac -= optind;
 148         av += optind;
 149
 150         if (ac != 1)
 151                 detach_usage();
 152         if (devctl_detach(av[0], force) < 0)
 153                 err(1, "Failed to detach %s", av[0]);
 154         return (0);
 155 }
 156 DEVCTL_COMMAND(top, detach, detach);
 157
 158 static void
 159 disable_usage(void)
 160 {
 161
 162         fprintf(stderr, "usage: devctl disable [-f] device\n");
 163         exit(1);
 164 }
 165
 166 static int
 167 disable(int ac, char **av)
 168 {
 169         bool force;
 170         int ch;
 171
 172         force = false;
 173         while ((ch = getopt(ac, av, "f")) != -1)
 174                 switch (ch) {
 175                 case 'f':
 176                         force = true;
 177                         break;
 178                 default:
 179                         disable_usage();
 180                 }
 181         ac -= optind;
 182         av += optind;
 183
 184         if (ac != 1)
 185                 disable_usage();
 186         if (devctl_disable(av[0], force) < 0)
 187                 err(1, "Failed to disable %s", av[0]);
 188         return (0);
 189 }
 190 DEVCTL_COMMAND(top, disable, disable);
 191
 192 static int
 193 enable(int ac, char **av)
 194 {
 195
 196         if (ac != 2)
 197                 usage();
 198         if (devctl_enable(av[1]) < 0)
 199                 err(1, "Failed to enable %s", av[1]);
 200         return (0);
 201 }
 202 DEVCTL_COMMAND(top, enable, enable);
 203
 204 static void
 205 set_driver_usage(void)
 206 {
 207
 208         fprintf(stderr, "usage: devctl set driver [-f] device driver\n");
 209         exit(1);
 210 }
 211
 212 static int
 213 set_driver(int ac, char **av)
 214 {
 215         bool force;
 216         int ch;
 217
 218         force = false;
 219         while ((ch = getopt(ac, av, "f")) != -1)
 220                 switch (ch) {
 221                 case 'f':
 222                         force = true;
 223                         break;
 224                 default:
 225                         set_driver_usage();
 226                 }
 227         ac -= optind;
 228         av += optind;
 229
 230         if (ac != 2)
 231                 set_driver_usage();
 232         if (devctl_set_driver(av[0], av[1], force) < 0)
 233                 err(1, "Failed to set %s driver to %s", av[0], av[1]);
 234         return (0);
 235 }
 236 DEVCTL_COMMAND(set, driver, set_driver);
 237
 238 static void
 239 clear_driver_usage(void)
 240 {
 241
 242         fprintf(stderr, "usage: devctl clear driver [-f] device\n");
 243         exit(1);
 244 }
 245
 246 static int
 247 clear_driver(int ac, char **av)
 248 {
 249         bool force;
 250         int ch;
 251
 252         force = false;
 253         while ((ch = getopt(ac, av, "f")) != -1)
 254                 switch (ch) {
 255                 case 'f':
 256                         force = true;
 257                         break;
 258                 default:
 259                         clear_driver_usage();
 260                 }
 261         ac -= optind;
 262         av += optind;
 263
 264         if (ac != 1)
 265                 clear_driver_usage();
 266         if (devctl_clear_driver(av[0], force) < 0)
 267                 err(1, "Failed to clear %s driver", av[0]);
 268         return (0);
 269 }
 270 DEVCTL_COMMAND(clear, driver, clear_driver);
 271
 272 int
 273 main(int ac, char *av[])
 274 {
 275         struct devctl_command **cmd;
 276
 277         if (ac == 1)
 278                 usage();
 279         ac--;
 280         av++;
 281
 282         SET_FOREACH(cmd, DEVCTL_DATASET(top)) {
 283                 if (strcmp((*cmd)->name, av[0]) == 0) {
 284                         if ((*cmd)->handler(ac, av) != 0)
 285                                 return (1);
 286                         else
 287                                 return (0);
 288                 }
 289         }
 290         warnx("Unknown command %s.", av[0]);
 291         return (1);
 292 }