1 /*
2 * Copyright (c) 1998 Kenneth D. Merry.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 *
28 * $FreeBSD: src/usr.bin/systat/devs.c,v 1.4 1999/08/28 01:05:59 peter Exp $
29 * $DragonFly: src/usr.bin/systat/devs.c,v 1.8 2008/11/10 04:59:45 swildner Exp $
30 */
31 /*
32 * Some code and ideas taken from the old disks.c.
33 * static char sccsid[] = "@(#)disks.c 8.1 (Berkeley) 6/6/93";
34 */
35 /*-
36 * Copyright (c) 1980, 1992, 1993
37 * The Regents of the University of California. All rights reserved.
38 *
39 * Redistribution and use in source and binary forms, with or without
40 * modification, are permitted provided that the following conditions
41 * are met:
42 * 1. Redistributions of source code must retain the above copyright
43 * notice, this list of conditions and the following disclaimer.
44 * 2. Redistributions in binary form must reproduce the above copyright
45 * notice, this list of conditions and the following disclaimer in the
46 * documentation and/or other materials provided with the distribution.
47 * 3. Neither the name of the University nor the names of its contributors
48 * may be used to endorse or promote products derived from this software
49 * without specific prior written permission.
50 *
51 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
52 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
53 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
54 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
55 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
56 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
57 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
58 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
59 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
60 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
61 * SUCH DAMAGE.
62 */
63
64 #include <sys/types.h>
65 #include <sys/devicestat.h>
66
67 #include <string.h>
68 #include <devstat.h>
69 #include <stdlib.h>
70 #include <unistd.h>
71 #include <ctype.h>
72 #include <err.h>
73 #include "systat.h"
74 #include "extern.h"
75 #include "devs.h"
76
77 typedef enum {
78 DS_MATCHTYPE_NONE,
79 DS_MATCHTYPE_SPEC,
80 DS_MATCHTYPE_PATTERN
81 } last_match_type;
82
83 last_match_type last_type;
84 struct device_selection *dev_select;
85 long generation;
86 int num_devices, num_selected;
87 int num_selections;
88 long select_generation;
89 struct devstat_match *matches = NULL;
90 int num_matches = 0;
91 char **specified_devices;
92 int num_devices_specified = 0;
93
94 static void dsinit_ignores(int maxshowdevs, struct statinfo *s1);
95 static int dsmatchselect(char *args, devstat_select_mode select_mode,
96 int maxshowdevs, struct statinfo *s1);
97 static int dsselect(char *args, devstat_select_mode select_mode,
98 int maxshowdevs, struct statinfo *s1);
99
100 int
dsinit(int maxshowdevs,struct statinfo * s1,struct statinfo * s2 __unused,struct statinfo * s3 __unused)101 dsinit(int maxshowdevs, struct statinfo *s1, struct statinfo *s2 __unused,
102 struct statinfo *s3 __unused)
103 {
104
105 /*
106 * Make sure that the userland devstat version matches the kernel
107 * devstat version. If not, exit and print a message informing
108 * the user of his mistake.
109 */
110 if (checkversion() < 0)
111 errx(1, "%s", devstat_errbuf);
112
113 generation = 0;
114 num_devices = 0;
115 num_selected = 0;
116 num_selections = 0;
117 select_generation = 0;
118 last_type = DS_MATCHTYPE_NONE;
119
120 if (getdevs(s1) == -1)
121 errx(1, "%s", devstat_errbuf);
122
123 num_devices = s1->dinfo->numdevs;
124 generation = s1->dinfo->generation;
125
126 dev_select = NULL;
127
128 /*
129 * At this point, selectdevs will almost surely indicate that the
130 * device list has changed, so we don't look for return values of 0
131 * or 1. If we get back -1, though, there is an error.
132 */
133 if (selectdevs(&dev_select, &num_selected, &num_selections,
134 &select_generation, generation, s1->dinfo->devices,
135 num_devices, NULL, 0, NULL, 0, DS_SELECT_ADD,
136 maxshowdevs, 0) == -1) {
137 errx(1, "%s", devstat_errbuf);
138 }
139 dsinit_ignores(maxshowdevs, s1);
140
141 return(1);
142 }
143
144 /*
145 * Clean up the device list a bit by removing devices people usually
146 * aren't interested in.
147 */
148 static
149 void
dsinit_ignores(int maxshowdevs,struct statinfo * s1)150 dsinit_ignores(int maxshowdevs, struct statinfo *s1)
151 {
152 char tmp[64];
153
154 snprintf(tmp, sizeof(tmp), "md*");
155 dscmd("ignore", tmp, maxshowdevs, s1);
156 snprintf(tmp, sizeof(tmp), "pass*");
157 dscmd("ignore", tmp, maxshowdevs, s1);
158 snprintf(tmp, sizeof(tmp), "sg*");
159 dscmd("ignore", tmp, maxshowdevs, s1);
160 }
161
162 int
dscmd(const char * cmd,char * args,int maxshowdevs,struct statinfo * s1)163 dscmd(const char *cmd, char *args, int maxshowdevs, struct statinfo *s1)
164 {
165 int retval;
166
167 if (prefix(cmd, "display") || prefix(cmd, "add"))
168 return(dsselect(args, DS_SELECT_ADDONLY, maxshowdevs, s1));
169 if (prefix(cmd, "ignore") || prefix(cmd, "delete"))
170 return(dsselect(args, DS_SELECT_REMOVE, maxshowdevs, s1));
171 if (prefix(cmd, "show") || prefix(cmd, "only"))
172 return(dsselect(args, DS_SELECT_ONLY, maxshowdevs, s1));
173 if (prefix(cmd, "type") || prefix(cmd, "match"))
174 return(dsmatchselect(args, DS_SELECT_ONLY, maxshowdevs, s1));
175 if (prefix(cmd, "refresh")) {
176 retval = selectdevs(&dev_select, &num_selected, &num_selections,
177 &select_generation, generation,
178 s1->dinfo->devices, num_devices,
179 (last_type == DS_MATCHTYPE_PATTERN) ?
180 matches : NULL,
181 (last_type == DS_MATCHTYPE_PATTERN) ?
182 num_matches : 0,
183 (last_type == DS_MATCHTYPE_SPEC) ?
184 specified_devices : NULL,
185 (last_type == DS_MATCHTYPE_SPEC) ?
186 num_devices_specified : 0,
187 (last_type == DS_MATCHTYPE_NONE) ?
188 DS_SELECT_ADD : DS_SELECT_ADDONLY,
189 maxshowdevs, 0);
190
191 /*
192 * refresh resets the list, ignore some non-useful devices
193 */
194 dsinit_ignores(maxshowdevs, s1);
195
196 if (retval == -1) {
197 warnx("%s", devstat_errbuf);
198 return(0);
199 } else if (retval == 1) {
200 return(2);
201 }
202 }
203 if (prefix(cmd, "drives")) {
204 int i;
205 move(CMDLINE, 0);
206 clrtoeol();
207 for (i = 0; i < num_devices; i++) {
208 printw("%s%d ", s1->dinfo->devices[i].device_name,
209 s1->dinfo->devices[i].unit_number);
210 }
211 return(1);
212 }
213 return(0);
214 }
215
216 static int
dsmatchselect(char * args,devstat_select_mode select_mode,int maxshowdevs,struct statinfo * s1)217 dsmatchselect(char *args, devstat_select_mode select_mode, int maxshowdevs,
218 struct statinfo *s1)
219 {
220 char **tempstr;
221 char *tstr[100];
222 int num_args = 0;
223 int i;
224 int retval = 0;
225
226 /*
227 * Break the (pipe delimited) input string out into separate
228 * strings.
229 */
230 for (tempstr = tstr, num_args = 0;
231 (*tempstr = strsep(&args, "|")) != NULL && (num_args < 100);
232 num_args++)
233 if (**tempstr != '\0')
234 if (++tempstr >= &tstr[100])
235 break;
236
237 if (num_args > 99) {
238 warnx("dsmatchselect: too many match arguments");
239 return(0);
240 }
241
242 /*
243 * If we've gone through the matching code before, clean out
244 * previously used memory.
245 */
246 if (num_matches > 0) {
247 free(matches);
248 matches = NULL;
249 num_matches = 0;
250 }
251
252 for (i = 0; i < num_args; i++) {
253 if (buildmatch(tstr[i], &matches, &num_matches) != 0) {
254 warnx("%s", devstat_errbuf);
255 return(0);
256 }
257 }
258 if (num_args > 0) {
259
260 last_type = DS_MATCHTYPE_PATTERN;
261
262 retval = selectdevs(&dev_select, &num_selected, &num_selections,
263 &select_generation, generation,
264 s1->dinfo->devices, num_devices, matches,
265 num_matches, NULL, 0, select_mode,
266 maxshowdevs, 0);
267 if (retval == -1)
268 err(1, "device selection error");
269 else if (retval == 1)
270 return(2);
271 }
272 return(1);
273 }
274
275 static int
dsselect(char * args,devstat_select_mode select_mode,int maxshowdevs,struct statinfo * s1)276 dsselect(char *args, devstat_select_mode select_mode, int maxshowdevs,
277 struct statinfo *s1)
278 {
279 char *cp;
280 int i;
281 int retval = 0;
282 int iswild;
283
284 /*
285 * If we've gone through this code before, free previously
286 * allocated resources.
287 */
288 if (num_devices_specified > 0) {
289 for (i = 0; i < num_devices_specified; i++)
290 free(specified_devices[i]);
291 free(specified_devices);
292 specified_devices = NULL;
293 num_devices_specified = 0;
294 }
295
296 /* do an initial malloc */
297 specified_devices = (char **)malloc(sizeof(char *));
298
299 cp = strchr(args, '\n');
300 if (cp)
301 *cp = '\0';
302 for (;;) {
303 for (cp = args; *cp && isspace(*cp); cp++)
304 ;
305 args = cp;
306 for (; *cp && !isspace(*cp); cp++)
307 ;
308 if (*cp)
309 *cp++ = '\0';
310 if (cp - args == 0)
311 break;
312 if (cp[-1] == '*') {
313 cp[-1] = 0;
314 iswild = 1;
315 } else {
316 iswild = 0;
317 }
318
319 for (i = 0; i < num_devices; i++) {
320 char tmpstr[80];
321
322 sprintf(tmpstr, "%s%d", dev_select[i].device_name,
323 dev_select[i].unit_number);
324 if (strcmp(args, tmpstr) == 0 ||
325 (iswild &&
326 strcmp(args, dev_select[i].device_name) == 0)) {
327 num_devices_specified++;
328
329 specified_devices =(char **)realloc(
330 specified_devices,
331 sizeof(char *) *
332 num_devices_specified);
333 specified_devices[num_devices_specified - 1] =
334 strdup(tmpstr);
335 }
336 }
337 #if 0
338 /* don't complain if device not known */
339 if (i >= num_devices)
340 error("%s: unknown drive", args);
341 #endif
342 args = cp;
343 }
344
345 if (num_devices_specified > 0) {
346 last_type = DS_MATCHTYPE_SPEC;
347
348 retval = selectdevs(&dev_select, &num_selected, &num_selections,
349 &select_generation, generation,
350 s1->dinfo->devices, num_devices, NULL, 0,
351 specified_devices, num_devices_specified,
352 select_mode, maxshowdevs, 0);
353 if (retval == -1)
354 err(1, "%s", devstat_errbuf);
355 else if (retval == 1)
356 return(2);
357 }
358 return(1);
359 }
360