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 37 static const char *g_pname; 38 static fmd_adm_t *g_adm; 39 static int g_quiet; 40 41 /*PRINTFLIKE1*/ 42 void 43 note(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 56 static void 57 vwarn(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*/ 75 void 76 warn(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*/ 86 void 87 die(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 99 static 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 124 static int 125 usage(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 150 static uint32_t 151 getu32(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 168 int 169 main(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