2 * Copyright (c) 1998 Kenneth D. Merry.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
29 * Copyright (c) 1980, 1992, 1993
30 * The Regents of the University of California. All rights reserved.
32 * Redistribution and use in source and binary forms, with or without
33 * modification, are permitted provided that the following conditions
35 * 1. Redistributions of source code must retain the above copyright
36 * notice, this list of conditions and the following disclaimer.
37 * 2. Redistributions in binary form must reproduce the above copyright
38 * notice, this list of conditions and the following disclaimer in the
39 * documentation and/or other materials provided with the distribution.
40 * 4. Neither the name of the University nor the names of its contributors
41 * may be used to endorse or promote products derived from this software
42 * without specific prior written permission.
44 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
45 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
46 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
47 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
48 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
49 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
50 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
51 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
52 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
53 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
57 #include <sys/cdefs.h>
59 __FBSDID("$FreeBSD$");
62 static const char sccsid[] = "@(#)disks.c 8.1 (Berkeley) 6/6/93";
65 #include <sys/types.h>
66 #include <sys/devicestat.h>
67 #include <sys/resource.h>
85 last_match_type last_type;
86 struct device_selection *dev_select;
88 int num_devices, num_selected;
90 long select_generation;
91 struct devstat_match *matches = NULL;
93 char **specified_devices;
94 int num_devices_specified = 0;
96 static int dsmatchselect(const char *args, devstat_select_mode select_mode,
97 int maxshowdevs, struct statinfo *s1);
98 static int dsselect(const char *args, devstat_select_mode select_mode,
99 int maxshowdevs, struct statinfo *s1);
102 dsinit(int maxshowdevs, struct statinfo *s1, struct statinfo *s2 __unused,
103 struct statinfo *s3 __unused)
107 * Make sure that the userland devstat version matches the kernel
108 * devstat version. If not, exit and print a message informing
109 * the user of his mistake.
111 if (devstat_checkversion(NULL) < 0)
112 errx(1, "%s", devstat_errbuf);
118 select_generation = 0;
119 last_type = DS_MATCHTYPE_NONE;
121 if (devstat_getdevs(NULL, s1) == -1)
122 errx(1, "%s", devstat_errbuf);
124 num_devices = s1->dinfo->numdevs;
125 generation = s1->dinfo->generation;
130 * At this point, selectdevs will almost surely indicate that the
131 * device list has changed, so we don't look for return values of 0
132 * or 1. If we get back -1, though, there is an error.
134 if (devstat_selectdevs(&dev_select, &num_selected, &num_selections,
135 &select_generation, generation, s1->dinfo->devices, num_devices,
136 NULL, 0, NULL, 0, DS_SELECT_ADD, maxshowdevs, 0) == -1)
137 errx(1, "%d %s", __LINE__, devstat_errbuf);
143 dscmd(const char *cmd, const char *args, int maxshowdevs, struct statinfo *s1)
147 if (prefix(cmd, "display") || prefix(cmd, "add"))
148 return(dsselect(args, DS_SELECT_ADDONLY, maxshowdevs, s1));
149 if (prefix(cmd, "ignore") || prefix(cmd, "delete"))
150 return(dsselect(args, DS_SELECT_REMOVE, maxshowdevs, s1));
151 if (prefix(cmd, "show") || prefix(cmd, "only"))
152 return(dsselect(args, DS_SELECT_ONLY, maxshowdevs, s1));
153 if (prefix(cmd, "type") || prefix(cmd, "match"))
154 return(dsmatchselect(args, DS_SELECT_ONLY, maxshowdevs, s1));
155 if (prefix(cmd, "refresh")) {
156 retval = devstat_selectdevs(&dev_select, &num_selected,
157 &num_selections, &select_generation, generation,
158 s1->dinfo->devices, num_devices,
159 (last_type ==DS_MATCHTYPE_PATTERN) ? matches : NULL,
160 (last_type ==DS_MATCHTYPE_PATTERN) ? num_matches : 0,
161 (last_type == DS_MATCHTYPE_SPEC) ?specified_devices : NULL,
162 (last_type == DS_MATCHTYPE_SPEC) ?num_devices_specified : 0,
163 (last_type == DS_MATCHTYPE_NONE) ? DS_SELECT_ADD :
164 DS_SELECT_ADDONLY, maxshowdevs, 0);
166 warnx("%s", devstat_errbuf);
168 } else if (retval == 1)
171 if (prefix(cmd, "drives")) {
175 for (i = 0; i < num_devices; i++) {
176 printw("%s%d ", s1->dinfo->devices[i].device_name,
177 s1->dinfo->devices[i].unit_number);
185 dsmatchselect(const char *args, devstat_select_mode select_mode, int maxshowdevs,
188 char **tempstr, *tmpstr, *tmpstr1;
195 * Break the (pipe delimited) input string out into separate
198 tmpstr = tmpstr1 = strdup(args);
199 for (tempstr = tstr, num_args = 0;
200 (*tempstr = strsep(&tmpstr1, "|")) != NULL && (num_args < 100);
202 if (**tempstr != '\0')
203 if (++tempstr >= &tstr[100])
208 warnx("dsmatchselect: too many match arguments");
213 * If we've gone through the matching code before, clean out
214 * previously used memory.
216 if (num_matches > 0) {
222 for (i = 0; i < num_args; i++) {
223 if (devstat_buildmatch(tstr[i], &matches, &num_matches) != 0) {
224 warnx("%s", devstat_errbuf);
230 last_type = DS_MATCHTYPE_PATTERN;
232 retval = devstat_selectdevs(&dev_select, &num_selected,
233 &num_selections, &select_generation, generation,
234 s1->dinfo->devices, num_devices, matches, num_matches,
235 NULL, 0, select_mode, maxshowdevs, 0);
237 err(1, "device selection error");
238 else if (retval == 1)
245 dsselect(const char *args, devstat_select_mode select_mode, int maxshowdevs,
248 char *cp, *tmpstr, *tmpstr1, *buffer;
253 * If we've gone through this code before, free previously
254 * allocated resources.
256 if (num_devices_specified > 0) {
257 for (i = 0; i < num_devices_specified; i++)
258 free(specified_devices[i]);
259 free(specified_devices);
260 specified_devices = NULL;
261 num_devices_specified = 0;
264 /* do an initial malloc */
265 specified_devices = (char **)malloc(sizeof(char *));
267 tmpstr = tmpstr1 = strdup(args);
268 cp = index(tmpstr1, '\n');
272 for (cp = tmpstr1; *cp && isspace(*cp); cp++)
275 for (; *cp && !isspace(*cp); cp++)
281 for (i = 0; i < num_devices; i++) {
282 asprintf(&buffer, "%s%d", dev_select[i].device_name,
283 dev_select[i].unit_number);
284 if (strcmp(buffer, tmpstr1) == 0) {
286 num_devices_specified++;
288 specified_devices =(char **)realloc(
291 num_devices_specified);
292 specified_devices[num_devices_specified -1]=
301 if (i >= num_devices)
302 error("%s: unknown drive", args);
307 if (num_devices_specified > 0) {
308 last_type = DS_MATCHTYPE_SPEC;
310 retval = devstat_selectdevs(&dev_select, &num_selected,
311 &num_selections, &select_generation, generation,
312 s1->dinfo->devices, num_devices, NULL, 0,
313 specified_devices, num_devices_specified,
314 select_mode, maxshowdevs, 0);
316 err(1, "%s", devstat_errbuf);
317 else if (retval == 1)