xref: /freebsd/sbin/geom/core/geom.c (revision 35efcc8b69e2e7665a9f8bfcd2564ea94bfef9f8)
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