105c91076SPawel Jakub Dawidek /*- 287070efbSPawel Jakub Dawidek * Copyright (c) 2004-2009 Pawel Jakub Dawidek <pjd@FreeBSD.org> 305c91076SPawel Jakub Dawidek * All rights reserved. 405c91076SPawel Jakub Dawidek * 505c91076SPawel Jakub Dawidek * Redistribution and use in source and binary forms, with or without 605c91076SPawel Jakub Dawidek * modification, are permitted provided that the following conditions 705c91076SPawel Jakub Dawidek * are met: 805c91076SPawel Jakub Dawidek * 1. Redistributions of source code must retain the above copyright 905c91076SPawel Jakub Dawidek * notice, this list of conditions and the following disclaimer. 1005c91076SPawel Jakub Dawidek * 2. Redistributions in binary form must reproduce the above copyright 1105c91076SPawel Jakub Dawidek * notice, this list of conditions and the following disclaimer in the 1205c91076SPawel Jakub Dawidek * documentation and/or other materials provided with the distribution. 1305c91076SPawel Jakub Dawidek * 1405c91076SPawel Jakub Dawidek * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND 1505c91076SPawel Jakub Dawidek * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 1605c91076SPawel Jakub Dawidek * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 1705c91076SPawel Jakub Dawidek * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE 1805c91076SPawel Jakub Dawidek * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 1905c91076SPawel Jakub Dawidek * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2005c91076SPawel Jakub Dawidek * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2105c91076SPawel Jakub Dawidek * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2205c91076SPawel Jakub Dawidek * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 2305c91076SPawel Jakub Dawidek * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 2405c91076SPawel Jakub Dawidek * SUCH DAMAGE. 2505c91076SPawel Jakub Dawidek */ 2605c91076SPawel Jakub Dawidek 2705c91076SPawel Jakub Dawidek #include <sys/cdefs.h> 2805c91076SPawel Jakub Dawidek __FBSDID("$FreeBSD$"); 2905c91076SPawel Jakub Dawidek 3005c91076SPawel Jakub Dawidek #include <sys/param.h> 3105c91076SPawel Jakub Dawidek #include <sys/linker.h> 3205c91076SPawel Jakub Dawidek #include <sys/module.h> 3305c91076SPawel Jakub Dawidek #include <sys/stat.h> 3405c91076SPawel Jakub Dawidek #include <sys/sysctl.h> 3505c91076SPawel Jakub Dawidek #include <ctype.h> 3605c91076SPawel Jakub Dawidek #include <err.h> 3705c91076SPawel Jakub Dawidek #include <errno.h> 3805c91076SPawel Jakub Dawidek #include <stdio.h> 3905c91076SPawel Jakub Dawidek #include <stdlib.h> 4005c91076SPawel Jakub Dawidek #include <stdarg.h> 4105c91076SPawel Jakub Dawidek #include <stdint.h> 4205c91076SPawel Jakub Dawidek #include <string.h> 4305c91076SPawel Jakub Dawidek #include <unistd.h> 4405c91076SPawel Jakub Dawidek #include <libgen.h> 45af565b58SPawel Jakub Dawidek #include <libutil.h> 4605c91076SPawel Jakub Dawidek #include <inttypes.h> 4705c91076SPawel Jakub Dawidek #include <dlfcn.h> 4805c91076SPawel Jakub Dawidek #include <assert.h> 4905c91076SPawel Jakub Dawidek #include <libgeom.h> 5005c91076SPawel Jakub Dawidek #include <geom.h> 5105c91076SPawel Jakub Dawidek 5205c91076SPawel Jakub Dawidek #include "misc/subr.h" 5305c91076SPawel Jakub Dawidek 5414bf405bSMarcel Moolenaar #ifdef STATIC_GEOM_CLASSES 55a16f9b36SMarcel Moolenaar extern uint32_t gpart_version; 56a16f9b36SMarcel Moolenaar extern struct g_command gpart_class_commands[]; 57bc69d66fSXin LI extern uint32_t glabel_version; 58bc69d66fSXin LI extern struct g_command glabel_class_commands[]; 59a16f9b36SMarcel Moolenaar #endif 6005c91076SPawel Jakub Dawidek 6105c91076SPawel Jakub Dawidek static char comm[MAXPATHLEN], *class_name = NULL, *gclass_name = NULL; 6205c91076SPawel Jakub Dawidek static uint32_t *version = NULL; 6305c91076SPawel Jakub Dawidek static int verbose = 0; 6405c91076SPawel Jakub Dawidek static struct g_command *class_commands = NULL; 6505c91076SPawel Jakub Dawidek 66138cedfdSPawel Jakub Dawidek #define GEOM_CLASS_CMDS 0x01 67138cedfdSPawel Jakub Dawidek #define GEOM_STD_CMDS 0x02 68138cedfdSPawel Jakub Dawidek static struct g_command *find_command(const char *cmdstr, int flags); 6905c91076SPawel Jakub Dawidek static int std_available(const char *name); 7005c91076SPawel Jakub Dawidek 71b70eccf3SPawel Jakub Dawidek static void std_help(struct gctl_req *req, unsigned flags); 7205c91076SPawel Jakub Dawidek static void std_list(struct gctl_req *req, unsigned flags); 7318ee8840SPawel Jakub Dawidek static void std_status(struct gctl_req *req, unsigned flags); 7405c91076SPawel Jakub Dawidek static void std_load(struct gctl_req *req, unsigned flags); 7505c91076SPawel Jakub Dawidek static void std_unload(struct gctl_req *req, unsigned flags); 7605c91076SPawel Jakub Dawidek 7705c91076SPawel Jakub Dawidek struct g_command std_commands[] = { 783cf55d3aSMarcel Moolenaar { "help", 0, std_help, G_NULL_OPTS, NULL, NULL }, 793cf55d3aSMarcel Moolenaar { "list", 0, std_list, G_NULL_OPTS, NULL, 80c979e206SPawel Jakub Dawidek "[name ...]" 81c979e206SPawel Jakub Dawidek }, 82ba6821f0SPawel Jakub Dawidek { "status", 0, std_status, 83ba6821f0SPawel Jakub Dawidek { 846fc60008SPawel Jakub Dawidek { 's', "script", NULL, G_TYPE_BOOL }, 85ba6821f0SPawel Jakub Dawidek G_OPT_SENTINEL 86ba6821f0SPawel Jakub Dawidek }, 873cf55d3aSMarcel Moolenaar NULL, "[-s] [name ...]" 88c979e206SPawel Jakub Dawidek }, 893cf55d3aSMarcel Moolenaar { "load", G_FLAG_VERBOSE | G_FLAG_LOADKLD, std_load, G_NULL_OPTS, 903cf55d3aSMarcel Moolenaar NULL, NULL }, 913cf55d3aSMarcel Moolenaar { "unload", G_FLAG_VERBOSE, std_unload, G_NULL_OPTS, NULL, NULL }, 9205c91076SPawel Jakub Dawidek G_CMD_SENTINEL 9305c91076SPawel Jakub Dawidek }; 9405c91076SPawel Jakub Dawidek 9505c91076SPawel Jakub Dawidek static void 96c979e206SPawel Jakub Dawidek usage_command(struct g_command *cmd, const char *prefix) 9705c91076SPawel Jakub Dawidek { 9805c91076SPawel Jakub Dawidek struct g_option *opt; 99c979e206SPawel Jakub Dawidek unsigned i; 10005c91076SPawel Jakub Dawidek 101c979e206SPawel Jakub Dawidek if (cmd->gc_usage != NULL) { 102963feac4SPawel Jakub Dawidek char *pos, *ptr, *sptr; 103963feac4SPawel Jakub Dawidek 104963feac4SPawel Jakub Dawidek sptr = ptr = strdup(cmd->gc_usage); 105963feac4SPawel Jakub Dawidek while ((pos = strsep(&ptr, "\n")) != NULL) { 106963feac4SPawel Jakub Dawidek if (*pos == '\0') 107963feac4SPawel Jakub Dawidek continue; 108963feac4SPawel Jakub Dawidek fprintf(stderr, "%s %s %s %s\n", prefix, comm, 109963feac4SPawel Jakub Dawidek cmd->gc_name, pos); 110963feac4SPawel Jakub Dawidek } 111963feac4SPawel Jakub Dawidek free(sptr); 112c979e206SPawel Jakub Dawidek return; 113c979e206SPawel Jakub Dawidek } 114963feac4SPawel Jakub Dawidek 115963feac4SPawel Jakub Dawidek fprintf(stderr, "%s %s %s", prefix, comm, cmd->gc_name); 11605c91076SPawel Jakub Dawidek if ((cmd->gc_flags & G_FLAG_VERBOSE) != 0) 11705c91076SPawel Jakub Dawidek fprintf(stderr, " [-v]"); 118c979e206SPawel Jakub Dawidek for (i = 0; ; i++) { 119c979e206SPawel Jakub Dawidek opt = &cmd->gc_options[i]; 12005c91076SPawel Jakub Dawidek if (opt->go_name == NULL) 12105c91076SPawel Jakub Dawidek break; 1226fc60008SPawel Jakub Dawidek if (opt->go_val != NULL || G_OPT_TYPE(opt) == G_TYPE_BOOL) 12305c91076SPawel Jakub Dawidek fprintf(stderr, " ["); 12405c91076SPawel Jakub Dawidek else 12505c91076SPawel Jakub Dawidek fprintf(stderr, " "); 12605c91076SPawel Jakub Dawidek fprintf(stderr, "-%c", opt->go_char); 1276fc60008SPawel Jakub Dawidek if (G_OPT_TYPE(opt) != G_TYPE_BOOL) 12805c91076SPawel Jakub Dawidek fprintf(stderr, " %s", opt->go_name); 1296fc60008SPawel Jakub Dawidek if (opt->go_val != NULL || G_OPT_TYPE(opt) == G_TYPE_BOOL) 13005c91076SPawel Jakub Dawidek fprintf(stderr, "]"); 13105c91076SPawel Jakub Dawidek } 1323cf55d3aSMarcel Moolenaar if (cmd->gc_argname) 1333cf55d3aSMarcel Moolenaar fprintf(stderr, " %s", cmd->gc_argname); 134c979e206SPawel Jakub Dawidek fprintf(stderr, "\n"); 13505c91076SPawel Jakub Dawidek } 13605c91076SPawel Jakub Dawidek 13705c91076SPawel Jakub Dawidek static void 138c979e206SPawel Jakub Dawidek usage(void) 13905c91076SPawel Jakub Dawidek { 14005c91076SPawel Jakub Dawidek 14105c91076SPawel Jakub Dawidek if (class_name == NULL) { 142b70eccf3SPawel Jakub Dawidek errx(EXIT_FAILURE, "usage: %s <class> <command> [options]", 143b70eccf3SPawel Jakub Dawidek "geom"); 14405c91076SPawel Jakub Dawidek } else { 145c979e206SPawel Jakub Dawidek struct g_command *cmd; 14605c91076SPawel Jakub Dawidek const char *prefix; 14705c91076SPawel Jakub Dawidek unsigned i; 14805c91076SPawel Jakub Dawidek 14905c91076SPawel Jakub Dawidek prefix = "usage:"; 150c979e206SPawel Jakub Dawidek if (class_commands != NULL) { 151c979e206SPawel Jakub Dawidek for (i = 0; ; i++) { 152c979e206SPawel Jakub Dawidek cmd = &class_commands[i]; 153c979e206SPawel Jakub Dawidek if (cmd->gc_name == NULL) 154c979e206SPawel Jakub Dawidek break; 155c979e206SPawel Jakub Dawidek usage_command(cmd, prefix); 15605c91076SPawel Jakub Dawidek prefix = " "; 15705c91076SPawel Jakub Dawidek } 158c979e206SPawel Jakub Dawidek } 15905c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 16005c91076SPawel Jakub Dawidek cmd = &std_commands[i]; 16105c91076SPawel Jakub Dawidek if (cmd->gc_name == NULL) 16205c91076SPawel Jakub Dawidek break; 163138cedfdSPawel Jakub Dawidek /* 164138cedfdSPawel Jakub Dawidek * If class defines command, which has the same name as 165138cedfdSPawel Jakub Dawidek * standard command, skip it, because it was already 166138cedfdSPawel Jakub Dawidek * shown on usage(). 167138cedfdSPawel Jakub Dawidek */ 168138cedfdSPawel Jakub Dawidek if (find_command(cmd->gc_name, GEOM_CLASS_CMDS) != NULL) 16905c91076SPawel Jakub Dawidek continue; 170c979e206SPawel Jakub Dawidek usage_command(cmd, prefix); 17105c91076SPawel Jakub Dawidek prefix = " "; 17205c91076SPawel Jakub Dawidek } 17305c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 17405c91076SPawel Jakub Dawidek } 17505c91076SPawel Jakub Dawidek } 17605c91076SPawel Jakub Dawidek 17705c91076SPawel Jakub Dawidek static void 17805c91076SPawel Jakub Dawidek load_module(void) 17905c91076SPawel Jakub Dawidek { 18005c91076SPawel Jakub Dawidek char name1[64], name2[64]; 18105c91076SPawel Jakub Dawidek 18205c91076SPawel Jakub Dawidek snprintf(name1, sizeof(name1), "g_%s", class_name); 18305c91076SPawel Jakub Dawidek snprintf(name2, sizeof(name2), "geom_%s", class_name); 18405c91076SPawel Jakub Dawidek if (modfind(name1) < 0) { 18505c91076SPawel Jakub Dawidek /* Not present in kernel, try loading it. */ 18605c91076SPawel Jakub Dawidek if (kldload(name2) < 0 || modfind(name1) < 0) { 18705c91076SPawel Jakub Dawidek if (errno != EEXIST) { 18805c91076SPawel Jakub Dawidek errx(EXIT_FAILURE, 18905c91076SPawel Jakub Dawidek "%s module not available!", name2); 19005c91076SPawel Jakub Dawidek } 19105c91076SPawel Jakub Dawidek } 19205c91076SPawel Jakub Dawidek } 19305c91076SPawel Jakub Dawidek } 19405c91076SPawel Jakub Dawidek 19505c91076SPawel Jakub Dawidek static int 19605c91076SPawel Jakub Dawidek strlcatf(char *str, size_t size, const char *format, ...) 19705c91076SPawel Jakub Dawidek { 19805c91076SPawel Jakub Dawidek size_t len; 19905c91076SPawel Jakub Dawidek va_list ap; 20005c91076SPawel Jakub Dawidek int ret; 20105c91076SPawel Jakub Dawidek 20205c91076SPawel Jakub Dawidek len = strlen(str); 20305c91076SPawel Jakub Dawidek str += len; 20405c91076SPawel Jakub Dawidek size -= len; 20505c91076SPawel Jakub Dawidek 20605c91076SPawel Jakub Dawidek va_start(ap, format); 20705c91076SPawel Jakub Dawidek ret = vsnprintf(str, size, format, ap); 20805c91076SPawel Jakub Dawidek va_end(ap); 20905c91076SPawel Jakub Dawidek 21005c91076SPawel Jakub Dawidek return (ret); 21105c91076SPawel Jakub Dawidek } 21205c91076SPawel Jakub Dawidek 21305c91076SPawel Jakub Dawidek /* 21405c91076SPawel Jakub Dawidek * Find given option in options available for given command. 21505c91076SPawel Jakub Dawidek */ 21605c91076SPawel Jakub Dawidek static struct g_option * 21705c91076SPawel Jakub Dawidek find_option(struct g_command *cmd, char ch) 21805c91076SPawel Jakub Dawidek { 21905c91076SPawel Jakub Dawidek struct g_option *opt; 22005c91076SPawel Jakub Dawidek unsigned i; 22105c91076SPawel Jakub Dawidek 22205c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 22305c91076SPawel Jakub Dawidek opt = &cmd->gc_options[i]; 22405c91076SPawel Jakub Dawidek if (opt->go_name == NULL) 22505c91076SPawel Jakub Dawidek return (NULL); 22605c91076SPawel Jakub Dawidek if (opt->go_char == ch) 22705c91076SPawel Jakub Dawidek return (opt); 22805c91076SPawel Jakub Dawidek } 22905c91076SPawel Jakub Dawidek /* NOTREACHED */ 23005c91076SPawel Jakub Dawidek return (NULL); 23105c91076SPawel Jakub Dawidek } 23205c91076SPawel Jakub Dawidek 23305c91076SPawel Jakub Dawidek /* 23405c91076SPawel Jakub Dawidek * Add given option to gctl_req. 23505c91076SPawel Jakub Dawidek */ 23605c91076SPawel Jakub Dawidek static void 23705c91076SPawel Jakub Dawidek set_option(struct gctl_req *req, struct g_option *opt, const char *val) 23805c91076SPawel Jakub Dawidek { 239dc344ca5SMarcel Moolenaar char *s; 240*35efcc8bSDag-Erling Smørgrav uint64_t number; 24105c91076SPawel Jakub Dawidek 242dc344ca5SMarcel Moolenaar if (G_OPT_TYPE(opt) == G_TYPE_NUMBER || 243dc344ca5SMarcel Moolenaar G_OPT_TYPE(opt) == G_TYPE_ASCNUM) { 244e84091cbSPawel Jakub Dawidek if (expand_number(val, &number) == -1) { 24505c91076SPawel Jakub Dawidek err(EXIT_FAILURE, "Invalid value for '%c' argument.", 24605c91076SPawel Jakub Dawidek opt->go_char); 24705c91076SPawel Jakub Dawidek } 248dc344ca5SMarcel Moolenaar if (G_OPT_TYPE(opt) == G_TYPE_NUMBER) 24905c91076SPawel Jakub Dawidek opt->go_val = malloc(sizeof(intmax_t)); 250dc344ca5SMarcel Moolenaar else { 251dc344ca5SMarcel Moolenaar asprintf(&s, "%jd", number); 252dc344ca5SMarcel Moolenaar opt->go_val = s; 253dc344ca5SMarcel Moolenaar } 25405c91076SPawel Jakub Dawidek if (opt->go_val == NULL) 25505c91076SPawel Jakub Dawidek errx(EXIT_FAILURE, "No memory."); 256dc344ca5SMarcel Moolenaar if (G_OPT_TYPE(opt) == G_TYPE_NUMBER) { 25705c91076SPawel Jakub Dawidek *(intmax_t *)opt->go_val = number; 258dc344ca5SMarcel Moolenaar gctl_ro_param(req, opt->go_name, sizeof(intmax_t), 259dc344ca5SMarcel Moolenaar opt->go_val); 260dc344ca5SMarcel Moolenaar } else 261dc344ca5SMarcel Moolenaar gctl_ro_param(req, opt->go_name, -1, opt->go_val); 2626fc60008SPawel Jakub Dawidek } else if (G_OPT_TYPE(opt) == G_TYPE_STRING) { 26376490718SPawel Jakub Dawidek gctl_ro_param(req, opt->go_name, -1, val); 2646fc60008SPawel Jakub Dawidek } else if (G_OPT_TYPE(opt) == G_TYPE_BOOL) { 26505c91076SPawel Jakub Dawidek opt->go_val = malloc(sizeof(int)); 26605c91076SPawel Jakub Dawidek if (opt->go_val == NULL) 26705c91076SPawel Jakub Dawidek errx(EXIT_FAILURE, "No memory."); 26805c91076SPawel Jakub Dawidek *(int *)opt->go_val = *val - '0'; 269dc344ca5SMarcel Moolenaar gctl_ro_param(req, opt->go_name, sizeof(int), opt->go_val); 2706fc60008SPawel Jakub Dawidek } else { 2716fc60008SPawel Jakub Dawidek assert(!"Invalid type"); 27205c91076SPawel Jakub Dawidek } 27305c91076SPawel Jakub Dawidek } 27405c91076SPawel Jakub Dawidek 27505c91076SPawel Jakub Dawidek /* 27605c91076SPawel Jakub Dawidek * 1. Add given argument by caller. 27705c91076SPawel Jakub Dawidek * 2. Add default values of not given arguments. 27805c91076SPawel Jakub Dawidek * 3. Add the rest of arguments. 27905c91076SPawel Jakub Dawidek */ 28005c91076SPawel Jakub Dawidek static void 28105c91076SPawel Jakub Dawidek parse_arguments(struct g_command *cmd, struct gctl_req *req, int *argc, 28205c91076SPawel Jakub Dawidek char ***argv) 28305c91076SPawel Jakub Dawidek { 28405c91076SPawel Jakub Dawidek struct g_option *opt; 28505c91076SPawel Jakub Dawidek char opts[64]; 28605c91076SPawel Jakub Dawidek unsigned i; 28705c91076SPawel Jakub Dawidek int ch; 28805c91076SPawel Jakub Dawidek 28905c91076SPawel Jakub Dawidek *opts = '\0'; 29005c91076SPawel Jakub Dawidek if ((cmd->gc_flags & G_FLAG_VERBOSE) != 0) 29105c91076SPawel Jakub Dawidek strlcat(opts, "v", sizeof(opts)); 29205c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 29305c91076SPawel Jakub Dawidek opt = &cmd->gc_options[i]; 29405c91076SPawel Jakub Dawidek if (opt->go_name == NULL) 29505c91076SPawel Jakub Dawidek break; 2966fc60008SPawel Jakub Dawidek assert(G_OPT_TYPE(opt) != 0); 2976fc60008SPawel Jakub Dawidek assert((opt->go_type & ~G_TYPE_MASK) == 0); 29805c91076SPawel Jakub Dawidek strlcatf(opts, sizeof(opts), "%c", opt->go_char); 2996fc60008SPawel Jakub Dawidek if (G_OPT_TYPE(opt) != G_TYPE_BOOL) 30005c91076SPawel Jakub Dawidek strlcat(opts, ":", sizeof(opts)); 30105c91076SPawel Jakub Dawidek } 30205c91076SPawel Jakub Dawidek 30305c91076SPawel Jakub Dawidek /* 30405c91076SPawel Jakub Dawidek * Add specified arguments. 30505c91076SPawel Jakub Dawidek */ 30605c91076SPawel Jakub Dawidek while ((ch = getopt(*argc, *argv, opts)) != -1) { 30705c91076SPawel Jakub Dawidek /* Standard (not passed to kernel) options. */ 30805c91076SPawel Jakub Dawidek switch (ch) { 30905c91076SPawel Jakub Dawidek case 'v': 31005c91076SPawel Jakub Dawidek verbose = 1; 31105c91076SPawel Jakub Dawidek continue; 31205c91076SPawel Jakub Dawidek } 31305c91076SPawel Jakub Dawidek /* Options passed to kernel. */ 31405c91076SPawel Jakub Dawidek opt = find_option(cmd, ch); 31505c91076SPawel Jakub Dawidek if (opt == NULL) 316c979e206SPawel Jakub Dawidek usage(); 31705c91076SPawel Jakub Dawidek if (G_OPT_ISDONE(opt)) { 3186fc60008SPawel Jakub Dawidek warnx("Option '%c' specified twice.", opt->go_char); 319c979e206SPawel Jakub Dawidek usage(); 32005c91076SPawel Jakub Dawidek } 32105c91076SPawel Jakub Dawidek G_OPT_DONE(opt); 32205c91076SPawel Jakub Dawidek 3236fc60008SPawel Jakub Dawidek if (G_OPT_TYPE(opt) == G_TYPE_BOOL) 32405c91076SPawel Jakub Dawidek set_option(req, opt, "1"); 32505c91076SPawel Jakub Dawidek else 32605c91076SPawel Jakub Dawidek set_option(req, opt, optarg); 32705c91076SPawel Jakub Dawidek } 32805c91076SPawel Jakub Dawidek *argc -= optind; 32905c91076SPawel Jakub Dawidek *argv += optind; 33005c91076SPawel Jakub Dawidek 33105c91076SPawel Jakub Dawidek /* 33205c91076SPawel Jakub Dawidek * Add not specified arguments, but with default values. 33305c91076SPawel Jakub Dawidek */ 33405c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 33505c91076SPawel Jakub Dawidek opt = &cmd->gc_options[i]; 33605c91076SPawel Jakub Dawidek if (opt->go_name == NULL) 33705c91076SPawel Jakub Dawidek break; 33805c91076SPawel Jakub Dawidek if (G_OPT_ISDONE(opt)) 33905c91076SPawel Jakub Dawidek continue; 34005c91076SPawel Jakub Dawidek 3416fc60008SPawel Jakub Dawidek if (G_OPT_TYPE(opt) == G_TYPE_BOOL) { 34205c91076SPawel Jakub Dawidek assert(opt->go_val == NULL); 34305c91076SPawel Jakub Dawidek set_option(req, opt, "0"); 34405c91076SPawel Jakub Dawidek } else { 34505c91076SPawel Jakub Dawidek if (opt->go_val == NULL) { 3466fc60008SPawel Jakub Dawidek warnx("Option '%c' not specified.", 34705c91076SPawel Jakub Dawidek opt->go_char); 348c979e206SPawel Jakub Dawidek usage(); 34905c91076SPawel Jakub Dawidek } else { 3506fc60008SPawel Jakub Dawidek if (G_OPT_TYPE(opt) == G_TYPE_NUMBER) { 35105c91076SPawel Jakub Dawidek gctl_ro_param(req, opt->go_name, 35205c91076SPawel Jakub Dawidek sizeof(intmax_t), opt->go_val); 353dc344ca5SMarcel Moolenaar } else if (G_OPT_TYPE(opt) == G_TYPE_STRING || 35479d89bb0SAndrey V. Elsukov G_OPT_TYPE(opt) == G_TYPE_ASCNUM) { 355cf805fabSMarcel Moolenaar if (cmd->gc_argname == NULL || 356cf805fabSMarcel Moolenaar opt->go_val == NULL || 357cf805fabSMarcel Moolenaar *(char *)opt->go_val != '\0') 3583cf55d3aSMarcel Moolenaar gctl_ro_param(req, opt->go_name, 3593cf55d3aSMarcel Moolenaar -1, opt->go_val); 3606fc60008SPawel Jakub Dawidek } else { 3616fc60008SPawel Jakub Dawidek assert(!"Invalid type"); 36205c91076SPawel Jakub Dawidek } 36305c91076SPawel Jakub Dawidek } 36405c91076SPawel Jakub Dawidek } 36505c91076SPawel Jakub Dawidek } 3663cf55d3aSMarcel Moolenaar 3673cf55d3aSMarcel Moolenaar if (cmd->gc_argname == NULL) { 36805c91076SPawel Jakub Dawidek /* 36905c91076SPawel Jakub Dawidek * Add rest of given arguments. 37005c91076SPawel Jakub Dawidek */ 37105c91076SPawel Jakub Dawidek gctl_ro_param(req, "nargs", sizeof(int), argc); 37205c91076SPawel Jakub Dawidek for (i = 0; i < (unsigned)*argc; i++) { 37305c91076SPawel Jakub Dawidek char argname[16]; 37405c91076SPawel Jakub Dawidek 37505c91076SPawel Jakub Dawidek snprintf(argname, sizeof(argname), "arg%u", i); 37605c91076SPawel Jakub Dawidek gctl_ro_param(req, argname, -1, (*argv)[i]); 37705c91076SPawel Jakub Dawidek } 3783cf55d3aSMarcel Moolenaar } else { 3793cf55d3aSMarcel Moolenaar if (*argc != 1) 3803cf55d3aSMarcel Moolenaar usage(); 3813cf55d3aSMarcel Moolenaar gctl_ro_param(req, cmd->gc_argname, -1, (*argv)[0]); 3823cf55d3aSMarcel Moolenaar } 38305c91076SPawel Jakub Dawidek } 38405c91076SPawel Jakub Dawidek 38505c91076SPawel Jakub Dawidek /* 38605c91076SPawel Jakub Dawidek * Find given command in commands available for given class. 38705c91076SPawel Jakub Dawidek */ 38805c91076SPawel Jakub Dawidek static struct g_command * 389138cedfdSPawel Jakub Dawidek find_command(const char *cmdstr, int flags) 39005c91076SPawel Jakub Dawidek { 39105c91076SPawel Jakub Dawidek struct g_command *cmd; 39205c91076SPawel Jakub Dawidek unsigned i; 39305c91076SPawel Jakub Dawidek 39405c91076SPawel Jakub Dawidek /* 39505c91076SPawel Jakub Dawidek * First try to find command defined by loaded library. 39605c91076SPawel Jakub Dawidek */ 397138cedfdSPawel Jakub Dawidek if ((flags & GEOM_CLASS_CMDS) != 0 && class_commands != NULL) { 39805c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 39905c91076SPawel Jakub Dawidek cmd = &class_commands[i]; 40005c91076SPawel Jakub Dawidek if (cmd->gc_name == NULL) 40105c91076SPawel Jakub Dawidek break; 40205c91076SPawel Jakub Dawidek if (strcmp(cmd->gc_name, cmdstr) == 0) 40305c91076SPawel Jakub Dawidek return (cmd); 40405c91076SPawel Jakub Dawidek } 40505c91076SPawel Jakub Dawidek } 40605c91076SPawel Jakub Dawidek /* 40705c91076SPawel Jakub Dawidek * Now try to find in standard commands. 40805c91076SPawel Jakub Dawidek */ 409138cedfdSPawel Jakub Dawidek if ((flags & GEOM_STD_CMDS) != 0) { 41005c91076SPawel Jakub Dawidek for (i = 0; ; i++) { 41105c91076SPawel Jakub Dawidek cmd = &std_commands[i]; 41205c91076SPawel Jakub Dawidek if (cmd->gc_name == NULL) 41305c91076SPawel Jakub Dawidek break; 41405c91076SPawel Jakub Dawidek if (strcmp(cmd->gc_name, cmdstr) == 0) 41505c91076SPawel Jakub Dawidek return (cmd); 41605c91076SPawel Jakub Dawidek } 417138cedfdSPawel Jakub Dawidek } 41805c91076SPawel Jakub Dawidek return (NULL); 41905c91076SPawel Jakub Dawidek } 42005c91076SPawel Jakub Dawidek 42105c91076SPawel Jakub Dawidek static unsigned 42205c91076SPawel Jakub Dawidek set_flags(struct g_command *cmd) 42305c91076SPawel Jakub Dawidek { 42405c91076SPawel Jakub Dawidek unsigned flags = 0; 42505c91076SPawel Jakub Dawidek 42605c91076SPawel Jakub Dawidek if ((cmd->gc_flags & G_FLAG_VERBOSE) != 0 && verbose) 42705c91076SPawel Jakub Dawidek flags |= G_FLAG_VERBOSE; 42805c91076SPawel Jakub Dawidek 42905c91076SPawel Jakub Dawidek return (flags); 43005c91076SPawel Jakub Dawidek } 43105c91076SPawel Jakub Dawidek 43205c91076SPawel Jakub Dawidek /* 43305c91076SPawel Jakub Dawidek * Run command. 43405c91076SPawel Jakub Dawidek */ 43505c91076SPawel Jakub Dawidek static void 43605c91076SPawel Jakub Dawidek run_command(int argc, char *argv[]) 43705c91076SPawel Jakub Dawidek { 43805c91076SPawel Jakub Dawidek struct g_command *cmd; 43905c91076SPawel Jakub Dawidek struct gctl_req *req; 44005c91076SPawel Jakub Dawidek const char *errstr; 44105c91076SPawel Jakub Dawidek char buf[4096]; 44205c91076SPawel Jakub Dawidek 443138cedfdSPawel Jakub Dawidek /* First try to find a command defined by a class. */ 444138cedfdSPawel Jakub Dawidek cmd = find_command(argv[0], GEOM_CLASS_CMDS); 445138cedfdSPawel Jakub Dawidek if (cmd == NULL) { 446138cedfdSPawel Jakub Dawidek /* Now, try to find a standard command. */ 447138cedfdSPawel Jakub Dawidek cmd = find_command(argv[0], GEOM_STD_CMDS); 44805c91076SPawel Jakub Dawidek if (cmd == NULL) { 4496fc60008SPawel Jakub Dawidek warnx("Unknown command: %s.", argv[0]); 450c979e206SPawel Jakub Dawidek usage(); 45105c91076SPawel Jakub Dawidek } 452138cedfdSPawel Jakub Dawidek if (!std_available(cmd->gc_name)) { 4536fc60008SPawel Jakub Dawidek warnx("Command '%s' not available.", argv[0]); 454138cedfdSPawel Jakub Dawidek exit(EXIT_FAILURE); 455138cedfdSPawel Jakub Dawidek } 456138cedfdSPawel Jakub Dawidek } 45705c91076SPawel Jakub Dawidek if ((cmd->gc_flags & G_FLAG_LOADKLD) != 0) 45805c91076SPawel Jakub Dawidek load_module(); 45905c91076SPawel Jakub Dawidek 46005c91076SPawel Jakub Dawidek req = gctl_get_handle(); 46105c91076SPawel Jakub Dawidek gctl_ro_param(req, "class", -1, gclass_name); 46205c91076SPawel Jakub Dawidek gctl_ro_param(req, "verb", -1, argv[0]); 46305c91076SPawel Jakub Dawidek if (version != NULL) 46405c91076SPawel Jakub Dawidek gctl_ro_param(req, "version", sizeof(*version), version); 46505c91076SPawel Jakub Dawidek parse_arguments(cmd, req, &argc, &argv); 46605c91076SPawel Jakub Dawidek 4671bcfab7fSPawel Jakub Dawidek bzero(buf, sizeof(buf)); 46805c91076SPawel Jakub Dawidek if (cmd->gc_func != NULL) { 46905c91076SPawel Jakub Dawidek unsigned flags; 47005c91076SPawel Jakub Dawidek 47105c91076SPawel Jakub Dawidek flags = set_flags(cmd); 47205c91076SPawel Jakub Dawidek cmd->gc_func(req, flags); 47305c91076SPawel Jakub Dawidek errstr = req->error; 47405c91076SPawel Jakub Dawidek } else { 47505c91076SPawel Jakub Dawidek gctl_rw_param(req, "output", sizeof(buf), buf); 47605c91076SPawel Jakub Dawidek errstr = gctl_issue(req); 47705c91076SPawel Jakub Dawidek } 47853767efdSPawel Jakub Dawidek if (errstr != NULL && errstr[0] != '\0') { 4796fc60008SPawel Jakub Dawidek warnx("%s", errstr); 480f792f1d8SPawel Jakub Dawidek if (strncmp(errstr, "warning: ", strlen("warning: ")) != 0) { 48105c91076SPawel Jakub Dawidek gctl_free(req); 48205c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 48305c91076SPawel Jakub Dawidek } 484f792f1d8SPawel Jakub Dawidek } 4851bcfab7fSPawel Jakub Dawidek if (buf[0] != '\0') 48605c91076SPawel Jakub Dawidek printf("%s", buf); 48705c91076SPawel Jakub Dawidek gctl_free(req); 48805c91076SPawel Jakub Dawidek if (verbose) 48905c91076SPawel Jakub Dawidek printf("Done.\n"); 49005c91076SPawel Jakub Dawidek exit(EXIT_SUCCESS); 49105c91076SPawel Jakub Dawidek } 49205c91076SPawel Jakub Dawidek 49314bf405bSMarcel Moolenaar #ifndef STATIC_GEOM_CLASSES 4943bad06e9SPawel Jakub Dawidek static const char * 4953bad06e9SPawel Jakub Dawidek library_path(void) 4963bad06e9SPawel Jakub Dawidek { 4973bad06e9SPawel Jakub Dawidek const char *path; 4983bad06e9SPawel Jakub Dawidek 4993bad06e9SPawel Jakub Dawidek path = getenv("GEOM_LIBRARY_PATH"); 5003bad06e9SPawel Jakub Dawidek if (path == NULL) 5013bad06e9SPawel Jakub Dawidek path = CLASS_DIR; 5023bad06e9SPawel Jakub Dawidek return (path); 5033bad06e9SPawel Jakub Dawidek } 5043bad06e9SPawel Jakub Dawidek 50505c91076SPawel Jakub Dawidek static void 50605c91076SPawel Jakub Dawidek load_library(void) 50705c91076SPawel Jakub Dawidek { 5085d824386SUlf Lilleengen char *curpath, path[MAXPATHLEN], *tofree, *totalpath; 50905c91076SPawel Jakub Dawidek uint32_t *lib_version; 51005c91076SPawel Jakub Dawidek void *dlh; 5112591e96cSUlf Lilleengen int ret; 51205c91076SPawel Jakub Dawidek 5132591e96cSUlf Lilleengen ret = 0; 5145d824386SUlf Lilleengen tofree = totalpath = strdup(library_path()); 5152591e96cSUlf Lilleengen if (totalpath == NULL) 5162591e96cSUlf Lilleengen err(EXIT_FAILURE, "Not enough memory for library path"); 5172591e96cSUlf Lilleengen 5182591e96cSUlf Lilleengen if (strchr(totalpath, ':') != NULL) 5192591e96cSUlf Lilleengen curpath = strsep(&totalpath, ":"); 5202591e96cSUlf Lilleengen else 5212591e96cSUlf Lilleengen curpath = totalpath; 5222591e96cSUlf Lilleengen /* Traverse the paths to find one that contains the library we want. */ 5232591e96cSUlf Lilleengen while (curpath != NULL) { 5242591e96cSUlf Lilleengen snprintf(path, sizeof(path), "%s/geom_%s.so", curpath, 5253bad06e9SPawel Jakub Dawidek class_name); 5262591e96cSUlf Lilleengen ret = access(path, F_OK); 5272591e96cSUlf Lilleengen if (ret == -1) { 528a73148d2SPawel Jakub Dawidek if (errno == ENOENT) { 52905c91076SPawel Jakub Dawidek /* 5302591e96cSUlf Lilleengen * If we cannot find library, try the next 5312591e96cSUlf Lilleengen * path. 53205c91076SPawel Jakub Dawidek */ 5332591e96cSUlf Lilleengen curpath = strsep(&totalpath, ":"); 5342591e96cSUlf Lilleengen continue; 53505c91076SPawel Jakub Dawidek } 536a73148d2SPawel Jakub Dawidek err(EXIT_FAILURE, "Cannot access library"); 537a73148d2SPawel Jakub Dawidek } 5382591e96cSUlf Lilleengen break; 5392591e96cSUlf Lilleengen } 5405d824386SUlf Lilleengen free(tofree); 5412591e96cSUlf Lilleengen /* No library was found, but standard commands can still be used */ 5422591e96cSUlf Lilleengen if (ret == -1) 5432591e96cSUlf Lilleengen return; 544a73148d2SPawel Jakub Dawidek dlh = dlopen(path, RTLD_NOW); 545a73148d2SPawel Jakub Dawidek if (dlh == NULL) 546a73148d2SPawel Jakub Dawidek errx(EXIT_FAILURE, "Cannot open library: %s.", dlerror()); 54705c91076SPawel Jakub Dawidek lib_version = dlsym(dlh, "lib_version"); 54805c91076SPawel Jakub Dawidek if (lib_version == NULL) { 5496fc60008SPawel Jakub Dawidek warnx("Cannot find symbol %s: %s.", "lib_version", dlerror()); 55005c91076SPawel Jakub Dawidek dlclose(dlh); 55105c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 55205c91076SPawel Jakub Dawidek } 55305c91076SPawel Jakub Dawidek if (*lib_version != G_LIB_VERSION) { 55405c91076SPawel Jakub Dawidek dlclose(dlh); 555f792f1d8SPawel Jakub Dawidek errx(EXIT_FAILURE, "%s and %s are not synchronized.", 556f792f1d8SPawel Jakub Dawidek getprogname(), path); 55705c91076SPawel Jakub Dawidek } 55805c91076SPawel Jakub Dawidek version = dlsym(dlh, "version"); 55905c91076SPawel Jakub Dawidek if (version == NULL) { 5606fc60008SPawel Jakub Dawidek warnx("Cannot find symbol %s: %s.", "version", dlerror()); 56105c91076SPawel Jakub Dawidek dlclose(dlh); 56205c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 56305c91076SPawel Jakub Dawidek } 56405c91076SPawel Jakub Dawidek class_commands = dlsym(dlh, "class_commands"); 56505c91076SPawel Jakub Dawidek if (class_commands == NULL) { 5666fc60008SPawel Jakub Dawidek warnx("Cannot find symbol %s: %s.", "class_commands", 5676fc60008SPawel Jakub Dawidek dlerror()); 56805c91076SPawel Jakub Dawidek dlclose(dlh); 56905c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 57005c91076SPawel Jakub Dawidek } 57105c91076SPawel Jakub Dawidek } 57214bf405bSMarcel Moolenaar #endif /* !STATIC_GEOM_CLASSES */ 57305c91076SPawel Jakub Dawidek 57405c91076SPawel Jakub Dawidek /* 57505c91076SPawel Jakub Dawidek * Class name should be all capital letters. 57605c91076SPawel Jakub Dawidek */ 57705c91076SPawel Jakub Dawidek static void 57805c91076SPawel Jakub Dawidek set_class_name(void) 57905c91076SPawel Jakub Dawidek { 58005c91076SPawel Jakub Dawidek char *s1, *s2; 58105c91076SPawel Jakub Dawidek 582ee602fbbSPawel Jakub Dawidek s1 = class_name; 583ee602fbbSPawel Jakub Dawidek for (; *s1 != '\0'; s1++) 584ee602fbbSPawel Jakub Dawidek *s1 = tolower(*s1); 585cb94ab30SPawel Jakub Dawidek gclass_name = malloc(strlen(class_name) + 1); 58605c91076SPawel Jakub Dawidek if (gclass_name == NULL) 58705c91076SPawel Jakub Dawidek errx(EXIT_FAILURE, "No memory"); 58805c91076SPawel Jakub Dawidek s1 = gclass_name; 58905c91076SPawel Jakub Dawidek s2 = class_name; 59005c91076SPawel Jakub Dawidek for (; *s2 != '\0'; s2++) 59105c91076SPawel Jakub Dawidek *s1++ = toupper(*s2); 59205c91076SPawel Jakub Dawidek *s1 = '\0'; 59305c91076SPawel Jakub Dawidek } 59405c91076SPawel Jakub Dawidek 59505c91076SPawel Jakub Dawidek static void 59605c91076SPawel Jakub Dawidek get_class(int *argc, char ***argv) 59705c91076SPawel Jakub Dawidek { 59805c91076SPawel Jakub Dawidek 59905c91076SPawel Jakub Dawidek snprintf(comm, sizeof(comm), "%s", basename((*argv)[0])); 60005c91076SPawel Jakub Dawidek if (strcmp(comm, "geom") == 0) { 60105c91076SPawel Jakub Dawidek if (*argc < 2) 602c979e206SPawel Jakub Dawidek usage(); 603687b0015SPawel Jakub Dawidek else if (*argc == 2) { 604687b0015SPawel Jakub Dawidek if (strcmp((*argv)[1], "-h") == 0 || 605687b0015SPawel Jakub Dawidek strcmp((*argv)[1], "help") == 0) { 606c979e206SPawel Jakub Dawidek usage(); 607687b0015SPawel Jakub Dawidek } 608687b0015SPawel Jakub Dawidek } 60905c91076SPawel Jakub Dawidek strlcatf(comm, sizeof(comm), " %s", (*argv)[1]); 61005c91076SPawel Jakub Dawidek class_name = (*argv)[1]; 61105c91076SPawel Jakub Dawidek *argc -= 2; 61205c91076SPawel Jakub Dawidek *argv += 2; 61305c91076SPawel Jakub Dawidek } else if (*comm == 'g') { 61405c91076SPawel Jakub Dawidek class_name = comm + 1; 61505c91076SPawel Jakub Dawidek *argc -= 1; 61605c91076SPawel Jakub Dawidek *argv += 1; 61705c91076SPawel Jakub Dawidek } else { 61805c91076SPawel Jakub Dawidek errx(EXIT_FAILURE, "Invalid utility name."); 61905c91076SPawel Jakub Dawidek } 620a16f9b36SMarcel Moolenaar 62114bf405bSMarcel Moolenaar #ifndef STATIC_GEOM_CLASSES 62205c91076SPawel Jakub Dawidek load_library(); 623a16f9b36SMarcel Moolenaar #else 624a16f9b36SMarcel Moolenaar if (!strcasecmp(class_name, "part")) { 625a16f9b36SMarcel Moolenaar version = &gpart_version; 626a16f9b36SMarcel Moolenaar class_commands = gpart_class_commands; 627bc69d66fSXin LI } else if (!strcasecmp(class_name, "label")) { 628bc69d66fSXin LI version = &glabel_version; 629bc69d66fSXin LI class_commands = glabel_class_commands; 630a16f9b36SMarcel Moolenaar } else 631a16f9b36SMarcel Moolenaar errx(EXIT_FAILURE, "Invalid class name."); 63214bf405bSMarcel Moolenaar #endif /* !STATIC_GEOM_CLASSES */ 633a16f9b36SMarcel Moolenaar 634a16f9b36SMarcel Moolenaar set_class_name(); 63505c91076SPawel Jakub Dawidek if (*argc < 1) 636c979e206SPawel Jakub Dawidek usage(); 63705c91076SPawel Jakub Dawidek } 63805c91076SPawel Jakub Dawidek 63905c91076SPawel Jakub Dawidek int 64005c91076SPawel Jakub Dawidek main(int argc, char *argv[]) 64105c91076SPawel Jakub Dawidek { 64205c91076SPawel Jakub Dawidek 64305c91076SPawel Jakub Dawidek get_class(&argc, &argv); 64405c91076SPawel Jakub Dawidek run_command(argc, argv); 64505c91076SPawel Jakub Dawidek /* NOTREACHED */ 64605c91076SPawel Jakub Dawidek 64705c91076SPawel Jakub Dawidek exit(EXIT_FAILURE); 64805c91076SPawel Jakub Dawidek } 64905c91076SPawel Jakub Dawidek 65005c91076SPawel Jakub Dawidek static struct gclass * 65105c91076SPawel Jakub Dawidek find_class(struct gmesh *mesh, const char *name) 65205c91076SPawel Jakub Dawidek { 65305c91076SPawel Jakub Dawidek struct gclass *classp; 65405c91076SPawel Jakub Dawidek 65505c91076SPawel Jakub Dawidek LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 65605c91076SPawel Jakub Dawidek if (strcmp(classp->lg_name, name) == 0) 65705c91076SPawel Jakub Dawidek return (classp); 65805c91076SPawel Jakub Dawidek } 65905c91076SPawel Jakub Dawidek return (NULL); 66005c91076SPawel Jakub Dawidek } 66105c91076SPawel Jakub Dawidek 6621d723f1dSPawel Jakub Dawidek static struct ggeom * 6631d723f1dSPawel Jakub Dawidek find_geom(struct gclass *classp, const char *name) 66405c91076SPawel Jakub Dawidek { 66505c91076SPawel Jakub Dawidek struct ggeom *gp; 66605c91076SPawel Jakub Dawidek 66705c91076SPawel Jakub Dawidek LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 6681d723f1dSPawel Jakub Dawidek if (strcmp(gp->lg_name, name) == 0) 6691d723f1dSPawel Jakub Dawidek return (gp); 67005c91076SPawel Jakub Dawidek } 67105c91076SPawel Jakub Dawidek return (NULL); 67205c91076SPawel Jakub Dawidek } 67305c91076SPawel Jakub Dawidek 67405c91076SPawel Jakub Dawidek static void 6752d76e2dbSPawel Jakub Dawidek list_one_provider(struct gprovider *pp, const char *prefix) 67605c91076SPawel Jakub Dawidek { 67705c91076SPawel Jakub Dawidek struct gconfig *conf; 678af565b58SPawel Jakub Dawidek char buf[5]; 67905c91076SPawel Jakub Dawidek 6801d723f1dSPawel Jakub Dawidek printf("Name: %s\n", pp->lg_name); 681af565b58SPawel Jakub Dawidek humanize_number(buf, sizeof(buf), (int64_t)pp->lg_mediasize, "", 682af565b58SPawel Jakub Dawidek HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL); 6831d723f1dSPawel Jakub Dawidek printf("%sMediasize: %jd (%s)\n", prefix, (intmax_t)pp->lg_mediasize, 6841d723f1dSPawel Jakub Dawidek buf); 6851d723f1dSPawel Jakub Dawidek printf("%sSectorsize: %u\n", prefix, pp->lg_sectorsize); 686e192c6e8SXin LI if (pp->lg_stripesize > 0 || pp->lg_stripeoffset > 0) { 68735daa28fSXin LI printf("%sStripesize: %ju\n", prefix, pp->lg_stripesize); 68835daa28fSXin LI printf("%sStripeoffset: %ju\n", prefix, pp->lg_stripeoffset); 68935daa28fSXin LI } 6901d723f1dSPawel Jakub Dawidek printf("%sMode: %s\n", prefix, pp->lg_mode); 69105c91076SPawel Jakub Dawidek LIST_FOREACH(conf, &pp->lg_config, lg_config) { 6921d723f1dSPawel Jakub Dawidek printf("%s%s: %s\n", prefix, conf->lg_name, conf->lg_val); 6931d723f1dSPawel Jakub Dawidek } 6941d723f1dSPawel Jakub Dawidek } 6951d723f1dSPawel Jakub Dawidek 6961d723f1dSPawel Jakub Dawidek static void 6972d76e2dbSPawel Jakub Dawidek list_one_consumer(struct gconsumer *cp, const char *prefix) 6981d723f1dSPawel Jakub Dawidek { 6991d723f1dSPawel Jakub Dawidek struct gprovider *pp; 7001d723f1dSPawel Jakub Dawidek struct gconfig *conf; 7011d723f1dSPawel Jakub Dawidek 7021d723f1dSPawel Jakub Dawidek pp = cp->lg_provider; 7031d723f1dSPawel Jakub Dawidek if (pp == NULL) 7041d723f1dSPawel Jakub Dawidek printf("[no provider]\n"); 7051d723f1dSPawel Jakub Dawidek else { 7061d723f1dSPawel Jakub Dawidek char buf[5]; 7071d723f1dSPawel Jakub Dawidek 7081d723f1dSPawel Jakub Dawidek printf("Name: %s\n", pp->lg_name); 7091d723f1dSPawel Jakub Dawidek humanize_number(buf, sizeof(buf), (int64_t)pp->lg_mediasize, "", 7101d723f1dSPawel Jakub Dawidek HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL); 7111d723f1dSPawel Jakub Dawidek printf("%sMediasize: %jd (%s)\n", prefix, 7121d723f1dSPawel Jakub Dawidek (intmax_t)pp->lg_mediasize, buf); 7131d723f1dSPawel Jakub Dawidek printf("%sSectorsize: %u\n", prefix, pp->lg_sectorsize); 71400b236aaSXin LI if (pp->lg_stripesize > 0 || pp->lg_stripeoffset > 0) { 71535daa28fSXin LI printf("%sStripesize: %ju\n", prefix, pp->lg_stripesize); 71635daa28fSXin LI printf("%sStripeoffset: %ju\n", prefix, pp->lg_stripeoffset); 71735daa28fSXin LI } 7181d723f1dSPawel Jakub Dawidek printf("%sMode: %s\n", prefix, cp->lg_mode); 7191d723f1dSPawel Jakub Dawidek } 7201d723f1dSPawel Jakub Dawidek LIST_FOREACH(conf, &cp->lg_config, lg_config) { 7211d723f1dSPawel Jakub Dawidek printf("%s%s: %s\n", prefix, conf->lg_name, conf->lg_val); 7221d723f1dSPawel Jakub Dawidek } 7231d723f1dSPawel Jakub Dawidek } 7241d723f1dSPawel Jakub Dawidek 7251d723f1dSPawel Jakub Dawidek static void 7262d76e2dbSPawel Jakub Dawidek list_one_geom(struct ggeom *gp) 7271d723f1dSPawel Jakub Dawidek { 7281d723f1dSPawel Jakub Dawidek struct gprovider *pp; 7291d723f1dSPawel Jakub Dawidek struct gconsumer *cp; 7301d723f1dSPawel Jakub Dawidek struct gconfig *conf; 7311d723f1dSPawel Jakub Dawidek unsigned n; 7321d723f1dSPawel Jakub Dawidek 7331d723f1dSPawel Jakub Dawidek printf("Geom name: %s\n", gp->lg_name); 7341d723f1dSPawel Jakub Dawidek LIST_FOREACH(conf, &gp->lg_config, lg_config) { 7351d723f1dSPawel Jakub Dawidek printf("%s: %s\n", conf->lg_name, conf->lg_val); 7361d723f1dSPawel Jakub Dawidek } 7371d723f1dSPawel Jakub Dawidek if (!LIST_EMPTY(&gp->lg_provider)) { 7381d723f1dSPawel Jakub Dawidek printf("Providers:\n"); 7391d723f1dSPawel Jakub Dawidek n = 1; 7401d723f1dSPawel Jakub Dawidek LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 7411d723f1dSPawel Jakub Dawidek printf("%u. ", n++); 7422d76e2dbSPawel Jakub Dawidek list_one_provider(pp, " "); 7431d723f1dSPawel Jakub Dawidek } 7441d723f1dSPawel Jakub Dawidek } 7451d723f1dSPawel Jakub Dawidek if (!LIST_EMPTY(&gp->lg_consumer)) { 7461d723f1dSPawel Jakub Dawidek printf("Consumers:\n"); 7471d723f1dSPawel Jakub Dawidek n = 1; 7481d723f1dSPawel Jakub Dawidek LIST_FOREACH(cp, &gp->lg_consumer, lg_consumer) { 7491d723f1dSPawel Jakub Dawidek printf("%u. ", n++); 7502d76e2dbSPawel Jakub Dawidek list_one_consumer(cp, " "); 7511d723f1dSPawel Jakub Dawidek } 75205c91076SPawel Jakub Dawidek } 75305c91076SPawel Jakub Dawidek printf("\n"); 75405c91076SPawel Jakub Dawidek } 75505c91076SPawel Jakub Dawidek 756b70eccf3SPawel Jakub Dawidek static void 757b70eccf3SPawel Jakub Dawidek std_help(struct gctl_req *req __unused, unsigned flags __unused) 758b70eccf3SPawel Jakub Dawidek { 759b70eccf3SPawel Jakub Dawidek 760c979e206SPawel Jakub Dawidek usage(); 761b70eccf3SPawel Jakub Dawidek } 762b70eccf3SPawel Jakub Dawidek 76305c91076SPawel Jakub Dawidek static int 76405c91076SPawel Jakub Dawidek std_list_available(void) 76505c91076SPawel Jakub Dawidek { 76605c91076SPawel Jakub Dawidek struct gmesh mesh; 76705c91076SPawel Jakub Dawidek struct gclass *classp; 76805c91076SPawel Jakub Dawidek int error; 76905c91076SPawel Jakub Dawidek 77005c91076SPawel Jakub Dawidek error = geom_gettree(&mesh); 7716fc60008SPawel Jakub Dawidek if (error != 0) 7726fc60008SPawel Jakub Dawidek errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 77305c91076SPawel Jakub Dawidek classp = find_class(&mesh, gclass_name); 77405c91076SPawel Jakub Dawidek geom_deletetree(&mesh); 77505c91076SPawel Jakub Dawidek if (classp != NULL) 77605c91076SPawel Jakub Dawidek return (1); 77705c91076SPawel Jakub Dawidek return (0); 77805c91076SPawel Jakub Dawidek } 77905c91076SPawel Jakub Dawidek 78005c91076SPawel Jakub Dawidek static void 78105c91076SPawel Jakub Dawidek std_list(struct gctl_req *req, unsigned flags __unused) 78205c91076SPawel Jakub Dawidek { 78305c91076SPawel Jakub Dawidek struct gmesh mesh; 78405c91076SPawel Jakub Dawidek struct gclass *classp; 7851d723f1dSPawel Jakub Dawidek struct ggeom *gp; 786f13942a7SPawel Jakub Dawidek const char *name; 787f13942a7SPawel Jakub Dawidek int error, i, nargs; 78805c91076SPawel Jakub Dawidek 78905c91076SPawel Jakub Dawidek error = geom_gettree(&mesh); 7906fc60008SPawel Jakub Dawidek if (error != 0) 7916fc60008SPawel Jakub Dawidek errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 79205c91076SPawel Jakub Dawidek classp = find_class(&mesh, gclass_name); 79305c91076SPawel Jakub Dawidek if (classp == NULL) { 79405c91076SPawel Jakub Dawidek geom_deletetree(&mesh); 7955d47cd0fSPawel Jakub Dawidek errx(EXIT_FAILURE, "Class %s not found.", gclass_name); 79605c91076SPawel Jakub Dawidek } 797f13942a7SPawel Jakub Dawidek nargs = gctl_get_int(req, "nargs"); 798f13942a7SPawel Jakub Dawidek if (nargs > 0) { 799f13942a7SPawel Jakub Dawidek for (i = 0; i < nargs; i++) { 800f13942a7SPawel Jakub Dawidek name = gctl_get_ascii(req, "arg%d", i); 8011d723f1dSPawel Jakub Dawidek gp = find_geom(classp, name); 8021d723f1dSPawel Jakub Dawidek if (gp != NULL) 8032d76e2dbSPawel Jakub Dawidek list_one_geom(gp); 8041d723f1dSPawel Jakub Dawidek else 8055d47cd0fSPawel Jakub Dawidek errx(EXIT_FAILURE, "No such geom: %s.", name); 80605c91076SPawel Jakub Dawidek } 80705c91076SPawel Jakub Dawidek } else { 80805c91076SPawel Jakub Dawidek LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 8095fd66a44SPawel Jakub Dawidek if (LIST_EMPTY(&gp->lg_provider)) 8105fd66a44SPawel Jakub Dawidek continue; 8112d76e2dbSPawel Jakub Dawidek list_one_geom(gp); 81205c91076SPawel Jakub Dawidek } 81305c91076SPawel Jakub Dawidek } 81405c91076SPawel Jakub Dawidek geom_deletetree(&mesh); 81505c91076SPawel Jakub Dawidek } 81605c91076SPawel Jakub Dawidek 81705c91076SPawel Jakub Dawidek static int 81818ee8840SPawel Jakub Dawidek std_status_available(void) 81918ee8840SPawel Jakub Dawidek { 82018ee8840SPawel Jakub Dawidek 82118ee8840SPawel Jakub Dawidek /* 'status' command is available when 'list' command is. */ 82218ee8840SPawel Jakub Dawidek return (std_list_available()); 82318ee8840SPawel Jakub Dawidek } 82418ee8840SPawel Jakub Dawidek 82518ee8840SPawel Jakub Dawidek static void 826da80913dSPawel Jakub Dawidek status_update_len(struct ggeom *gp, int *name_len, int *status_len) 82718ee8840SPawel Jakub Dawidek { 82818ee8840SPawel Jakub Dawidek struct gprovider *pp; 82918ee8840SPawel Jakub Dawidek struct gconfig *conf; 830da80913dSPawel Jakub Dawidek int len; 83118ee8840SPawel Jakub Dawidek 83218ee8840SPawel Jakub Dawidek assert(gp != NULL); 83318ee8840SPawel Jakub Dawidek assert(name_len != NULL); 83418ee8840SPawel Jakub Dawidek assert(status_len != NULL); 83518ee8840SPawel Jakub Dawidek 83618ee8840SPawel Jakub Dawidek pp = LIST_FIRST(&gp->lg_provider); 83718ee8840SPawel Jakub Dawidek if (pp != NULL) 83818ee8840SPawel Jakub Dawidek len = strlen(pp->lg_name); 83918ee8840SPawel Jakub Dawidek else 84018ee8840SPawel Jakub Dawidek len = strlen(gp->lg_name); 84118ee8840SPawel Jakub Dawidek if (*name_len < len) 84218ee8840SPawel Jakub Dawidek *name_len = len; 84318ee8840SPawel Jakub Dawidek LIST_FOREACH(conf, &gp->lg_config, lg_config) { 84418ee8840SPawel Jakub Dawidek if (strcasecmp(conf->lg_name, "state") == 0) { 84518ee8840SPawel Jakub Dawidek len = strlen(conf->lg_val); 84618ee8840SPawel Jakub Dawidek if (*status_len < len) 84718ee8840SPawel Jakub Dawidek *status_len = len; 84818ee8840SPawel Jakub Dawidek } 84918ee8840SPawel Jakub Dawidek } 85018ee8840SPawel Jakub Dawidek } 85118ee8840SPawel Jakub Dawidek 852ba6821f0SPawel Jakub Dawidek static char * 85318ee8840SPawel Jakub Dawidek status_one_consumer(struct gconsumer *cp) 85418ee8840SPawel Jakub Dawidek { 855ba6821f0SPawel Jakub Dawidek static char buf[256]; 85618ee8840SPawel Jakub Dawidek struct gprovider *pp; 85718ee8840SPawel Jakub Dawidek struct gconfig *conf; 85818ee8840SPawel Jakub Dawidek 85918ee8840SPawel Jakub Dawidek pp = cp->lg_provider; 86018ee8840SPawel Jakub Dawidek if (pp == NULL) 861ba6821f0SPawel Jakub Dawidek return (NULL); 86218ee8840SPawel Jakub Dawidek LIST_FOREACH(conf, &cp->lg_config, lg_config) { 86318ee8840SPawel Jakub Dawidek if (strcasecmp(conf->lg_name, "synchronized") == 0) 864ba6821f0SPawel Jakub Dawidek break; 86518ee8840SPawel Jakub Dawidek } 866ba6821f0SPawel Jakub Dawidek if (conf == NULL) 867ba6821f0SPawel Jakub Dawidek snprintf(buf, sizeof(buf), "%s", pp->lg_name); 868ba6821f0SPawel Jakub Dawidek else { 869ba6821f0SPawel Jakub Dawidek snprintf(buf, sizeof(buf), "%s (%s)", pp->lg_name, 870ba6821f0SPawel Jakub Dawidek conf->lg_val); 871ba6821f0SPawel Jakub Dawidek } 872ba6821f0SPawel Jakub Dawidek return (buf); 87318ee8840SPawel Jakub Dawidek } 87418ee8840SPawel Jakub Dawidek 87518ee8840SPawel Jakub Dawidek static void 876ba6821f0SPawel Jakub Dawidek status_one_geom(struct ggeom *gp, int script, int name_len, int status_len) 87718ee8840SPawel Jakub Dawidek { 87818ee8840SPawel Jakub Dawidek struct gprovider *pp; 87918ee8840SPawel Jakub Dawidek struct gconsumer *cp; 88018ee8840SPawel Jakub Dawidek struct gconfig *conf; 881ba6821f0SPawel Jakub Dawidek const char *name, *status, *component; 882ba6821f0SPawel Jakub Dawidek int gotone; 88318ee8840SPawel Jakub Dawidek 88418ee8840SPawel Jakub Dawidek pp = LIST_FIRST(&gp->lg_provider); 88518ee8840SPawel Jakub Dawidek if (pp != NULL) 88618ee8840SPawel Jakub Dawidek name = pp->lg_name; 88718ee8840SPawel Jakub Dawidek else 88818ee8840SPawel Jakub Dawidek name = gp->lg_name; 88918ee8840SPawel Jakub Dawidek LIST_FOREACH(conf, &gp->lg_config, lg_config) { 890ba6821f0SPawel Jakub Dawidek if (strcasecmp(conf->lg_name, "state") == 0) 89118ee8840SPawel Jakub Dawidek break; 89218ee8840SPawel Jakub Dawidek } 89318ee8840SPawel Jakub Dawidek if (conf == NULL) 894ba6821f0SPawel Jakub Dawidek status = "N/A"; 895ba6821f0SPawel Jakub Dawidek else 896ba6821f0SPawel Jakub Dawidek status = conf->lg_val; 897ba6821f0SPawel Jakub Dawidek gotone = 0; 89818ee8840SPawel Jakub Dawidek LIST_FOREACH(cp, &gp->lg_consumer, lg_consumer) { 899ba6821f0SPawel Jakub Dawidek component = status_one_consumer(cp); 900ba6821f0SPawel Jakub Dawidek if (component == NULL) 901ba6821f0SPawel Jakub Dawidek continue; 902ba6821f0SPawel Jakub Dawidek gotone = 1; 903ba6821f0SPawel Jakub Dawidek printf("%*s %*s %s\n", name_len, name, status_len, status, 904ba6821f0SPawel Jakub Dawidek component); 905ba6821f0SPawel Jakub Dawidek if (!script) 906ba6821f0SPawel Jakub Dawidek name = status = ""; 90718ee8840SPawel Jakub Dawidek } 908ba6821f0SPawel Jakub Dawidek if (!gotone) { 909ba6821f0SPawel Jakub Dawidek printf("%*s %*s %s\n", name_len, name, status_len, status, 910ba6821f0SPawel Jakub Dawidek "N/A"); 911ba6821f0SPawel Jakub Dawidek } 91218ee8840SPawel Jakub Dawidek } 91318ee8840SPawel Jakub Dawidek 91418ee8840SPawel Jakub Dawidek static void 91518ee8840SPawel Jakub Dawidek std_status(struct gctl_req *req, unsigned flags __unused) 91618ee8840SPawel Jakub Dawidek { 91718ee8840SPawel Jakub Dawidek struct gmesh mesh; 91818ee8840SPawel Jakub Dawidek struct gclass *classp; 91918ee8840SPawel Jakub Dawidek struct ggeom *gp; 920f13942a7SPawel Jakub Dawidek const char *name; 921da80913dSPawel Jakub Dawidek int name_len, status_len; 922f13942a7SPawel Jakub Dawidek int error, i, n, nargs, script; 92318ee8840SPawel Jakub Dawidek 92418ee8840SPawel Jakub Dawidek error = geom_gettree(&mesh); 9256fc60008SPawel Jakub Dawidek if (error != 0) 9266fc60008SPawel Jakub Dawidek errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 92718ee8840SPawel Jakub Dawidek classp = find_class(&mesh, gclass_name); 9285d47cd0fSPawel Jakub Dawidek if (classp == NULL) 9295d47cd0fSPawel Jakub Dawidek errx(EXIT_FAILURE, "Class %s not found.", gclass_name); 930f13942a7SPawel Jakub Dawidek nargs = gctl_get_int(req, "nargs"); 931f13942a7SPawel Jakub Dawidek script = gctl_get_int(req, "script"); 93218ee8840SPawel Jakub Dawidek name_len = strlen("Name"); 93318ee8840SPawel Jakub Dawidek status_len = strlen("Status"); 934f13942a7SPawel Jakub Dawidek if (nargs > 0) { 935f13942a7SPawel Jakub Dawidek for (i = 0, n = 0; i < nargs; i++) { 936f13942a7SPawel Jakub Dawidek name = gctl_get_ascii(req, "arg%d", i); 93718ee8840SPawel Jakub Dawidek gp = find_geom(classp, name); 93818ee8840SPawel Jakub Dawidek if (gp == NULL) 9395d47cd0fSPawel Jakub Dawidek errx(EXIT_FAILURE, "No such geom: %s.", name); 94018ee8840SPawel Jakub Dawidek else { 94118ee8840SPawel Jakub Dawidek status_update_len(gp, &name_len, &status_len); 94218ee8840SPawel Jakub Dawidek n++; 94318ee8840SPawel Jakub Dawidek } 94418ee8840SPawel Jakub Dawidek } 94518ee8840SPawel Jakub Dawidek if (n == 0) 94618ee8840SPawel Jakub Dawidek goto end; 94718ee8840SPawel Jakub Dawidek } else { 948f13942a7SPawel Jakub Dawidek n = 0; 94918ee8840SPawel Jakub Dawidek LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 95018ee8840SPawel Jakub Dawidek if (LIST_EMPTY(&gp->lg_provider)) 95118ee8840SPawel Jakub Dawidek continue; 95218ee8840SPawel Jakub Dawidek status_update_len(gp, &name_len, &status_len); 95318ee8840SPawel Jakub Dawidek n++; 95418ee8840SPawel Jakub Dawidek } 95518ee8840SPawel Jakub Dawidek if (n == 0) 95618ee8840SPawel Jakub Dawidek goto end; 95718ee8840SPawel Jakub Dawidek } 958f13942a7SPawel Jakub Dawidek if (!script) { 959da80913dSPawel Jakub Dawidek printf("%*s %*s %s\n", name_len, "Name", status_len, "Status", 96018ee8840SPawel Jakub Dawidek "Components"); 961ba6821f0SPawel Jakub Dawidek } 962f13942a7SPawel Jakub Dawidek if (nargs > 0) { 963f13942a7SPawel Jakub Dawidek for (i = 0; i < nargs; i++) { 964f13942a7SPawel Jakub Dawidek name = gctl_get_ascii(req, "arg%d", i); 96518ee8840SPawel Jakub Dawidek gp = find_geom(classp, name); 96618ee8840SPawel Jakub Dawidek if (gp != NULL) { 967f13942a7SPawel Jakub Dawidek status_one_geom(gp, script, name_len, 968ba6821f0SPawel Jakub Dawidek status_len); 96918ee8840SPawel Jakub Dawidek } 97018ee8840SPawel Jakub Dawidek } 97118ee8840SPawel Jakub Dawidek } else { 97218ee8840SPawel Jakub Dawidek LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 97318ee8840SPawel Jakub Dawidek if (LIST_EMPTY(&gp->lg_provider)) 97418ee8840SPawel Jakub Dawidek continue; 975f13942a7SPawel Jakub Dawidek status_one_geom(gp, script, name_len, status_len); 97618ee8840SPawel Jakub Dawidek } 97718ee8840SPawel Jakub Dawidek } 97818ee8840SPawel Jakub Dawidek end: 97918ee8840SPawel Jakub Dawidek geom_deletetree(&mesh); 98018ee8840SPawel Jakub Dawidek } 98118ee8840SPawel Jakub Dawidek 98218ee8840SPawel Jakub Dawidek static int 98305c91076SPawel Jakub Dawidek std_load_available(void) 98405c91076SPawel Jakub Dawidek { 98505c91076SPawel Jakub Dawidek char name[MAXPATHLEN], paths[MAXPATHLEN * 8], *p; 98605c91076SPawel Jakub Dawidek struct stat sb; 98705c91076SPawel Jakub Dawidek size_t len; 98805c91076SPawel Jakub Dawidek 98905c91076SPawel Jakub Dawidek snprintf(name, sizeof(name), "g_%s", class_name); 99005c91076SPawel Jakub Dawidek /* 99105c91076SPawel Jakub Dawidek * If already in kernel, "load" command is not available. 99205c91076SPawel Jakub Dawidek */ 99305c91076SPawel Jakub Dawidek if (modfind(name) >= 0) 99405c91076SPawel Jakub Dawidek return (0); 99505c91076SPawel Jakub Dawidek bzero(paths, sizeof(paths)); 99605c91076SPawel Jakub Dawidek len = sizeof(paths); 99705c91076SPawel Jakub Dawidek if (sysctlbyname("kern.module_path", paths, &len, NULL, 0) < 0) 99805c91076SPawel Jakub Dawidek err(EXIT_FAILURE, "sysctl(kern.module_path)"); 99905c91076SPawel Jakub Dawidek for (p = strtok(paths, ";"); p != NULL; p = strtok(NULL, ";")) { 100005c91076SPawel Jakub Dawidek snprintf(name, sizeof(name), "%s/geom_%s.ko", p, class_name); 100105c91076SPawel Jakub Dawidek /* 100205c91076SPawel Jakub Dawidek * If geom_<name>.ko file exists, "load" command is available. 100305c91076SPawel Jakub Dawidek */ 100405c91076SPawel Jakub Dawidek if (stat(name, &sb) == 0) 100505c91076SPawel Jakub Dawidek return (1); 100605c91076SPawel Jakub Dawidek } 100705c91076SPawel Jakub Dawidek return (0); 100805c91076SPawel Jakub Dawidek } 100905c91076SPawel Jakub Dawidek 101005c91076SPawel Jakub Dawidek static void 101105c91076SPawel Jakub Dawidek std_load(struct gctl_req *req __unused, unsigned flags) 101205c91076SPawel Jakub Dawidek { 101305c91076SPawel Jakub Dawidek 101405c91076SPawel Jakub Dawidek /* 101505c91076SPawel Jakub Dawidek * Do nothing special here, because of G_FLAG_LOADKLD flag, 101605c91076SPawel Jakub Dawidek * module is already loaded. 101705c91076SPawel Jakub Dawidek */ 101805c91076SPawel Jakub Dawidek if ((flags & G_FLAG_VERBOSE) != 0) 101905c91076SPawel Jakub Dawidek printf("Module available.\n"); 102005c91076SPawel Jakub Dawidek } 102105c91076SPawel Jakub Dawidek 102205c91076SPawel Jakub Dawidek static int 102305c91076SPawel Jakub Dawidek std_unload_available(void) 102405c91076SPawel Jakub Dawidek { 102505c91076SPawel Jakub Dawidek char name[64]; 102605c91076SPawel Jakub Dawidek int id; 102705c91076SPawel Jakub Dawidek 102805c91076SPawel Jakub Dawidek snprintf(name, sizeof(name), "geom_%s", class_name); 102905c91076SPawel Jakub Dawidek id = kldfind(name); 103005c91076SPawel Jakub Dawidek if (id >= 0) 103105c91076SPawel Jakub Dawidek return (1); 103205c91076SPawel Jakub Dawidek return (0); 103305c91076SPawel Jakub Dawidek } 103405c91076SPawel Jakub Dawidek 103505c91076SPawel Jakub Dawidek static void 103605c91076SPawel Jakub Dawidek std_unload(struct gctl_req *req, unsigned flags __unused) 103705c91076SPawel Jakub Dawidek { 103805c91076SPawel Jakub Dawidek char name[64]; 103905c91076SPawel Jakub Dawidek int id; 104005c91076SPawel Jakub Dawidek 104105c91076SPawel Jakub Dawidek snprintf(name, sizeof(name), "geom_%s", class_name); 104205c91076SPawel Jakub Dawidek id = kldfind(name); 104305c91076SPawel Jakub Dawidek if (id < 0) { 104405c91076SPawel Jakub Dawidek gctl_error(req, "Could not find module: %s.", strerror(errno)); 104505c91076SPawel Jakub Dawidek return; 104605c91076SPawel Jakub Dawidek } 104705c91076SPawel Jakub Dawidek if (kldunload(id) < 0) { 104805c91076SPawel Jakub Dawidek gctl_error(req, "Could not unload module: %s.", 104905c91076SPawel Jakub Dawidek strerror(errno)); 105005c91076SPawel Jakub Dawidek return; 105105c91076SPawel Jakub Dawidek } 105205c91076SPawel Jakub Dawidek } 105305c91076SPawel Jakub Dawidek 105405c91076SPawel Jakub Dawidek static int 105505c91076SPawel Jakub Dawidek std_available(const char *name) 105605c91076SPawel Jakub Dawidek { 105705c91076SPawel Jakub Dawidek 1058b70eccf3SPawel Jakub Dawidek if (strcmp(name, "help") == 0) 1059b70eccf3SPawel Jakub Dawidek return (1); 1060b70eccf3SPawel Jakub Dawidek else if (strcmp(name, "list") == 0) 106105c91076SPawel Jakub Dawidek return (std_list_available()); 106218ee8840SPawel Jakub Dawidek else if (strcmp(name, "status") == 0) 106318ee8840SPawel Jakub Dawidek return (std_status_available()); 106405c91076SPawel Jakub Dawidek else if (strcmp(name, "load") == 0) 106505c91076SPawel Jakub Dawidek return (std_load_available()); 106605c91076SPawel Jakub Dawidek else if (strcmp(name, "unload") == 0) 106705c91076SPawel Jakub Dawidek return (std_unload_available()); 106805c91076SPawel Jakub Dawidek else 106905c91076SPawel Jakub Dawidek assert(!"Unknown standard command."); 107005c91076SPawel Jakub Dawidek return (0); 107105c91076SPawel Jakub Dawidek } 1072