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