1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
22 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
23 * Use is subject to license terms.
24 */
25
26#pragma ident	"%Z%%M%	%I%	%E% SMI"
27
28#include <strings.h>
29#include <limits.h>
30#include <stdlib.h>
31#include <stdarg.h>
32#include <stdio.h>
33#include <errno.h>
34
35#include <fmadm.h>
36
37static const char *g_pname;
38static fmd_adm_t *g_adm;
39static int g_quiet;
40
41/*PRINTFLIKE1*/
42void
43note(const char *format, ...)
44{
45	va_list ap;
46
47	if (g_quiet)
48		return; /* suppress notices if -q specified */
49
50	(void) fprintf(stdout, "%s: ", g_pname);
51	va_start(ap, format);
52	(void) vfprintf(stdout, format, ap);
53	va_end(ap);
54}
55
56static void
57vwarn(const char *format, va_list ap)
58{
59	int err = errno;
60
61	(void) fprintf(stderr, "%s: ", g_pname);
62
63	if (format != NULL)
64		(void) vfprintf(stderr, format, ap);
65
66	errno = err; /* restore errno for fmd_adm_errmsg() */
67
68	if (format == NULL)
69		(void) fprintf(stderr, "%s\n", fmd_adm_errmsg(g_adm));
70	else if (strchr(format, '\n') == NULL)
71		(void) fprintf(stderr, ": %s\n", fmd_adm_errmsg(g_adm));
72}
73
74/*PRINTFLIKE1*/
75void
76warn(const char *format, ...)
77{
78	va_list ap;
79
80	va_start(ap, format);
81	vwarn(format, ap);
82	va_end(ap);
83}
84
85/*PRINTFLIKE1*/
86void
87die(const char *format, ...)
88{
89	va_list ap;
90
91	va_start(ap, format);
92	vwarn(format, ap);
93	va_end(ap);
94
95	fmd_adm_close(g_adm);
96	exit(FMADM_EXIT_ERROR);
97}
98
99static const struct cmd {
100	int (*cmd_func)(fmd_adm_t *, int, char *[]);
101	const char *cmd_name;
102	const char *cmd_usage;
103	const char *cmd_desc;
104} cmds[] = {
105{ cmd_config, "config", NULL, "display fault manager configuration" },
106{ cmd_faulty, "faulty", "[-afgiprsv] [-u <uuid>] [-n <max_fault>]",
107	"display list of faulty resources" },
108{ cmd_flush, "flush", "<fmri> ...", "flush cached state for resource" },
109{ cmd_gc, "gc", "<module>", NULL },
110{ cmd_load, "load", "<path>", "load specified fault manager module" },
111{ cmd_repair, "repair", "<fmri>|label|<uuid>", NULL },
112{ cmd_repaired, "repaired", "<fmri>|label>",
113	"notify fault manager that resource has been repaired" },
114{ cmd_acquit, "acquit", "<fmri> [<uuid>] | label [<uuid>] | <uuid>",
115	"acquit resource or acquit case" },
116{ cmd_replaced, "replaced", "<fmri>|label",
117	"notify fault manager that resource has been replaced" },
118{ cmd_reset, "reset", "[-s serd] <module>", "reset module or sub-component" },
119{ cmd_rotate, "rotate", "<logname>", "rotate log file" },
120{ cmd_unload, "unload", "<module>", "unload specified fault manager module" },
121{ NULL, NULL, NULL }
122};
123
124static int
125usage(FILE *fp)
126{
127	const struct cmd *cp;
128	char buf[256];
129
130	(void) fprintf(fp,
131	    "Usage: %s [-P prog] [-q] [cmd [args ... ]]\n\n", g_pname);
132
133	for (cp = cmds; cp->cmd_name != NULL; cp++) {
134		if (cp->cmd_desc == NULL)
135			continue;
136
137		if (cp->cmd_usage != NULL) {
138			(void) snprintf(buf, sizeof (buf), "%s %s %s",
139			    g_pname, cp->cmd_name, cp->cmd_usage);
140		} else {
141			(void) snprintf(buf, sizeof (buf), "%s %s",
142			    g_pname, cp->cmd_name);
143		}
144		(void) fprintf(fp, "\t%-30s - %s\n", buf, cp->cmd_desc);
145	}
146
147	return (FMADM_EXIT_USAGE);
148}
149
150static uint32_t
151getu32(const char *name, const char *s)
152{
153	u_longlong_t val;
154	char *p;
155
156	errno = 0;
157	val = strtoull(s, &p, 0);
158
159	if (errno != 0 || p == s || *p != '\0' || val > UINT32_MAX) {
160		(void) fprintf(stderr, "%s: invalid %s argument -- %s\n",
161		    g_pname, name, s);
162		exit(FMADM_EXIT_USAGE);
163	}
164
165	return ((uint32_t)val);
166}
167
168int
169main(int argc, char *argv[])
170{
171	const struct cmd *cp;
172	uint32_t program;
173	const char *p;
174	int c, err;
175
176	if ((p = strrchr(argv[0], '/')) == NULL)
177		g_pname = argv[0];
178	else
179		g_pname = p + 1;
180
181	if ((p = getenv("FMD_PROGRAM")) != NULL)
182		program = getu32("$FMD_PROGRAM", p);
183	else
184		program = FMD_ADM_PROGRAM;
185
186	while ((c = getopt(argc, argv, "P:q")) != EOF) {
187		switch (c) {
188		case 'P':
189			program = getu32("program", optarg);
190			break;
191		case 'q':
192			g_quiet++;
193			break;
194		default:
195			return (usage(stderr));
196		}
197	}
198
199	if (optind >= argc)
200		return (usage(stdout));
201
202	for (cp = cmds; cp->cmd_name != NULL; cp++) {
203		if (strcmp(cp->cmd_name, argv[optind]) == 0)
204			break;
205	}
206
207	if (cp->cmd_name == NULL) {
208		(void) fprintf(stderr, "%s: illegal subcommand -- %s\n",
209		    g_pname, argv[optind]);
210		return (usage(stderr));
211	}
212
213	if ((g_adm = fmd_adm_open(NULL, program, FMD_ADM_VERSION)) == NULL)
214		die(NULL); /* fmd_adm_errmsg() has enough info */
215
216	argc -= optind;
217	argv += optind;
218
219	optind = 1; /* reset optind so subcommands can getopt() */
220
221	err = cp->cmd_func(g_adm, argc, argv);
222	fmd_adm_close(g_adm);
223	return (err == FMADM_EXIT_USAGE ? usage(stderr) : err);
224}
225