1 /* $NetBSD: ccdconfig.c,v 1.2.2.1 1995/11/11 02:43:35 thorpej Exp $ */ 2 3 /* 4 * Copyright (c) 1995 Jason R. Thorpe. 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. All advertising materials mentioning features or use of this software 16 * must display the following acknowledgement: 17 * This product includes software developed for the NetBSD Project 18 * by Jason R. Thorpe. 19 * 4. The name of the author may not be used to endorse or promote products 20 * derived from this software without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 23 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 24 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 25 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 26 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 27 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 28 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 29 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 30 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 */ 34 35 #ifndef lint 36 static const char rcsid[] = 37 "$FreeBSD$"; 38 #endif /* not lint */ 39 40 #include <sys/param.h> 41 #include <sys/linker.h> 42 #include <sys/disklabel.h> 43 #include <sys/stat.h> 44 #include <sys/module.h> 45 #include <ctype.h> 46 #include <err.h> 47 #include <errno.h> 48 #include <fcntl.h> 49 #include <kvm.h> 50 #include <limits.h> 51 #include <stdio.h> 52 #include <stdlib.h> 53 #include <string.h> 54 #include <unistd.h> 55 56 #include <sys/devicestat.h> 57 #include <sys/ccdvar.h> 58 59 #include "pathnames.h" 60 61 static int lineno = 0; 62 static int verbose = 0; 63 static char *ccdconf = _PATH_CCDCONF; 64 65 static char *core = NULL; 66 static char *kernel = NULL; 67 68 struct flagval { 69 char *fv_flag; 70 int fv_val; 71 } flagvaltab[] = { 72 { "CCDF_SWAP", CCDF_SWAP }, 73 { "CCDF_UNIFORM", CCDF_UNIFORM }, 74 { "CCDF_MIRROR", CCDF_MIRROR }, 75 { "CCDF_PARITY", CCDF_PARITY }, 76 { NULL, 0 }, 77 }; 78 79 static struct nlist nl[] = { 80 { "_ccd_softc" }, 81 #define SYM_CCDSOFTC 0 82 { "_numccd" }, 83 #define SYM_NUMCCD 1 84 { NULL }, 85 }; 86 87 #define CCD_CONFIG 0 /* configure a device */ 88 #define CCD_CONFIGALL 1 /* configure all devices */ 89 #define CCD_UNCONFIG 2 /* unconfigure a device */ 90 #define CCD_UNCONFIGALL 3 /* unconfigure all devices */ 91 #define CCD_DUMP 4 /* dump a ccd's configuration */ 92 93 static int checkdev __P((char *)); 94 static int do_io __P((char *, u_long, struct ccd_ioctl *)); 95 static int do_single __P((int, char **, int)); 96 static int do_all __P((int)); 97 static int dump_ccd __P((int, char **)); 98 static int getmaxpartitions __P((void)); 99 static int getrawpartition __P((void)); 100 static int flags_to_val __P((char *)); 101 static void print_ccd_info __P((struct ccd_softc *, kvm_t *)); 102 static char *resolve_ccdname __P((char *)); 103 static void usage __P((void)); 104 105 int 106 main(argc, argv) 107 int argc; 108 char **argv; 109 { 110 int ch, options = 0, action = CCD_CONFIG; 111 112 while ((ch = getopt(argc, argv, "cCf:gM:N:uUv")) != -1) { 113 switch (ch) { 114 case 'c': 115 action = CCD_CONFIG; 116 ++options; 117 break; 118 119 case 'C': 120 action = CCD_CONFIGALL; 121 ++options; 122 break; 123 124 case 'f': 125 ccdconf = optarg; 126 break; 127 128 case 'g': 129 action = CCD_DUMP; 130 break; 131 132 case 'M': 133 core = optarg; 134 break; 135 136 case 'N': 137 kernel = optarg; 138 break; 139 140 case 'u': 141 action = CCD_UNCONFIG; 142 ++options; 143 break; 144 145 case 'U': 146 action = CCD_UNCONFIGALL; 147 ++options; 148 break; 149 150 case 'v': 151 verbose = 1; 152 break; 153 154 default: 155 usage(); 156 } 157 } 158 argc -= optind; 159 argv += optind; 160 161 if (options > 1) 162 usage(); 163 164 /* 165 * Discard setgid privileges if not the running kernel so that bad 166 * guys can't print interesting stuff from kernel memory. 167 */ 168 if (core != NULL || kernel != NULL || action != CCD_DUMP) { 169 setegid(getgid()); 170 setgid(getgid()); 171 } 172 173 if (modfind("ccd") < 0) { 174 /* Not present in kernel, try loading it */ 175 if (kldload("ccd") < 0 || modfind("ccd") < 0) 176 warn("ccd module not available!"); 177 } 178 179 switch (action) { 180 case CCD_CONFIG: 181 case CCD_UNCONFIG: 182 exit(do_single(argc, argv, action)); 183 /* NOTREACHED */ 184 185 case CCD_CONFIGALL: 186 case CCD_UNCONFIGALL: 187 exit(do_all(action)); 188 /* NOTREACHED */ 189 190 case CCD_DUMP: 191 exit(dump_ccd(argc, argv)); 192 /* NOTREACHED */ 193 } 194 /* NOTREACHED */ 195 return (0); 196 } 197 198 static int 199 do_single(argc, argv, action) 200 int argc; 201 char **argv; 202 int action; 203 { 204 struct ccd_ioctl ccio; 205 char *ccd, *cp, *cp2, **disks; 206 int noflags = 0, i, ileave, flags = 0, j; 207 208 bzero(&ccio, sizeof(ccio)); 209 210 /* 211 * If unconfiguring, all arguments are treated as ccds. 212 */ 213 if (action == CCD_UNCONFIG || action == CCD_UNCONFIGALL) { 214 for (i = 0; argc != 0; ) { 215 cp = *argv++; --argc; 216 if ((ccd = resolve_ccdname(cp)) == NULL) { 217 warnx("invalid ccd name: %s", cp); 218 i = 1; 219 continue; 220 } 221 if (do_io(ccd, CCDIOCCLR, &ccio)) 222 i = 1; 223 else 224 if (verbose) 225 printf("%s unconfigured\n", cp); 226 } 227 return (i); 228 } 229 230 /* Make sure there are enough arguments. */ 231 if (argc < 4) { 232 if (argc == 3) { 233 /* Assume that no flags are specified. */ 234 noflags = 1; 235 } else { 236 if (action == CCD_CONFIGALL) { 237 warnx("%s: bad line: %d", ccdconf, lineno); 238 return (1); 239 } else 240 usage(); 241 } 242 } 243 244 /* First argument is the ccd to configure. */ 245 cp = *argv++; --argc; 246 if ((ccd = resolve_ccdname(cp)) == NULL) { 247 warnx("invalid ccd name: %s", cp); 248 return (1); 249 } 250 251 /* Next argument is the interleave factor. */ 252 cp = *argv++; --argc; 253 errno = 0; /* to check for ERANGE */ 254 ileave = (int)strtol(cp, &cp2, 10); 255 if ((errno == ERANGE) || (ileave < 0) || (*cp2 != '\0')) { 256 warnx("invalid interleave factor: %s", cp); 257 return (1); 258 } 259 260 if (noflags == 0) { 261 /* Next argument is the ccd configuration flags. */ 262 cp = *argv++; --argc; 263 if ((flags = flags_to_val(cp)) < 0) { 264 warnx("invalid flags argument: %s", cp); 265 return (1); 266 } 267 } 268 269 /* Next is the list of disks to make the ccd from. */ 270 disks = malloc(argc * sizeof(char *)); 271 if (disks == NULL) { 272 warnx("no memory to configure ccd"); 273 return (1); 274 } 275 for (i = 0; argc != 0; ) { 276 cp = *argv++; --argc; 277 if ((j = checkdev(cp)) == 0) 278 disks[i++] = cp; 279 else { 280 warnx("%s: %s", cp, strerror(j)); 281 return (1); 282 } 283 } 284 285 /* Fill in the ccio. */ 286 ccio.ccio_disks = disks; 287 ccio.ccio_ndisks = i; 288 ccio.ccio_ileave = ileave; 289 ccio.ccio_flags = flags; 290 291 if (do_io(ccd, CCDIOCSET, &ccio)) { 292 free(disks); 293 return (1); 294 } 295 296 if (verbose) { 297 printf("ccd%d: %d components ", ccio.ccio_unit, 298 ccio.ccio_ndisks); 299 for (i = 0; i < ccio.ccio_ndisks; ++i) { 300 if ((cp2 = strrchr(disks[i], '/')) != NULL) 301 ++cp2; 302 else 303 cp2 = disks[i]; 304 printf("%c%s%c", 305 i == 0 ? '(' : ' ', cp2, 306 i == ccio.ccio_ndisks - 1 ? ')' : ','); 307 } 308 printf(", %lu blocks ", (u_long)ccio.ccio_size); 309 if (ccio.ccio_ileave != 0) 310 printf("interleaved at %d blocks\n", ccio.ccio_ileave); 311 else 312 printf("concatenated\n"); 313 } 314 315 free(disks); 316 return (0); 317 } 318 319 static int 320 do_all(action) 321 int action; 322 { 323 FILE *f; 324 char line[_POSIX2_LINE_MAX]; 325 char *cp, **argv; 326 int argc, rval; 327 gid_t egid; 328 329 rval = 0; 330 egid = getegid(); 331 setegid(getgid()); 332 if ((f = fopen(ccdconf, "r")) == NULL) { 333 setegid(egid); 334 warn("fopen: %s", ccdconf); 335 return (1); 336 } 337 setegid(egid); 338 339 while (fgets(line, sizeof(line), f) != NULL) { 340 argc = 0; 341 argv = NULL; 342 ++lineno; 343 if ((cp = strrchr(line, '\n')) != NULL) 344 *cp = '\0'; 345 346 /* Break up the line and pass it's contents to do_single(). */ 347 if (line[0] == '\0') 348 goto end_of_line; 349 for (cp = line; (cp = strtok(cp, " \t")) != NULL; cp = NULL) { 350 if (*cp == '#') 351 break; 352 if ((argv = realloc(argv, 353 sizeof(char *) * ++argc)) == NULL) { 354 warnx("no memory to configure ccds"); 355 return (1); 356 } 357 argv[argc - 1] = cp; 358 /* 359 * If our action is to unconfigure all, then pass 360 * just the first token to do_single() and ignore 361 * the rest. Since this will be encountered on 362 * our first pass through the line, the Right 363 * Thing will happen. 364 */ 365 if (action == CCD_UNCONFIGALL) { 366 if (do_single(argc, argv, action)) 367 rval = 1; 368 goto end_of_line; 369 } 370 } 371 if (argc != 0) 372 if (do_single(argc, argv, action)) 373 rval = 1; 374 375 end_of_line: 376 if (argv != NULL) 377 free(argv); 378 } 379 380 (void)fclose(f); 381 return (rval); 382 } 383 384 static int 385 checkdev(path) 386 char *path; 387 { 388 struct stat st; 389 390 if (stat(path, &st) != 0) 391 return (errno); 392 393 if (!S_ISBLK(st.st_mode) && !S_ISCHR(st.st_mode)) 394 return (EINVAL); 395 396 return (0); 397 } 398 399 static int 400 pathtounit(path, unitp) 401 char *path; 402 int *unitp; 403 { 404 struct stat st; 405 int maxpartitions; 406 407 if (stat(path, &st) != 0) 408 return (errno); 409 410 if (!S_ISBLK(st.st_mode) && !S_ISCHR(st.st_mode)) 411 return (EINVAL); 412 413 if ((maxpartitions = getmaxpartitions()) < 0) 414 return (errno); 415 416 *unitp = minor(st.st_rdev) / maxpartitions; 417 418 return (0); 419 } 420 421 static char * 422 resolve_ccdname(name) 423 char *name; 424 { 425 char c, *path; 426 size_t len, newlen; 427 int rawpart; 428 429 if (name[0] == '/' || name[0] == '.') { 430 /* Assume they gave the correct pathname. */ 431 return (strdup(name)); 432 } 433 434 len = strlen(name); 435 c = name[len - 1]; 436 437 newlen = len + 8; 438 if ((path = malloc(newlen)) == NULL) 439 return (NULL); 440 bzero(path, newlen); 441 442 if (isdigit(c)) { 443 if ((rawpart = getrawpartition()) < 0) { 444 free(path); 445 return (NULL); 446 } 447 (void)sprintf(path, "/dev/%s%c", name, 'a' + rawpart); 448 } else 449 (void)sprintf(path, "/dev/%s", name); 450 451 return (path); 452 } 453 454 static int 455 do_io(path, cmd, cciop) 456 char *path; 457 u_long cmd; 458 struct ccd_ioctl *cciop; 459 { 460 int fd; 461 char *cp; 462 463 if ((fd = open(path, O_RDWR, 0640)) < 0) { 464 warn("open: %s", path); 465 return (1); 466 } 467 468 if (ioctl(fd, cmd, cciop) < 0) { 469 switch (cmd) { 470 case CCDIOCSET: 471 cp = "CCDIOCSET"; 472 break; 473 474 case CCDIOCCLR: 475 cp = "CCDIOCCLR"; 476 break; 477 478 default: 479 cp = "unknown"; 480 } 481 warn("ioctl (%s): %s", cp, path); 482 return (1); 483 } 484 485 return (0); 486 } 487 488 #define KVM_ABORT(kd, str) { \ 489 (void)kvm_close((kd)); \ 490 warnx("%s", (str)); \ 491 warnx("%s", kvm_geterr((kd))); \ 492 return (1); \ 493 } 494 495 static int 496 dump_ccd(argc, argv) 497 int argc; 498 char **argv; 499 { 500 char errbuf[_POSIX2_LINE_MAX], *ccd, *cp; 501 struct ccd_softc *cs, *kcs; 502 size_t readsize; 503 int i, error, numccd, numconfiged = 0; 504 kvm_t *kd; 505 506 bzero(errbuf, sizeof(errbuf)); 507 508 if ((kd = kvm_openfiles(kernel, core, NULL, O_RDONLY, 509 errbuf)) == NULL) { 510 warnx("can't open kvm: %s", errbuf); 511 return (1); 512 } 513 514 if (kvm_nlist(kd, nl)) 515 KVM_ABORT(kd, "ccd-related symbols not available"); 516 517 /* Check to see how many ccds are currently configured. */ 518 if (kvm_read(kd, nl[SYM_NUMCCD].n_value, (char *)&numccd, 519 sizeof(numccd)) != sizeof(numccd)) 520 KVM_ABORT(kd, "can't determine number of configured ccds"); 521 522 if (numccd == 0) { 523 printf("ccd driver in kernel, but is uninitialized\n"); 524 goto done; 525 } 526 527 /* Allocate space for the configuration data. */ 528 readsize = numccd * sizeof(struct ccd_softc); 529 if ((cs = malloc(readsize)) == NULL) { 530 warnx("no memory for configuration data"); 531 goto bad; 532 } 533 bzero(cs, readsize); 534 535 /* 536 * Read the ccd configuration data from the kernel and dump 537 * it to stdout. 538 */ 539 if (kvm_read(kd, nl[SYM_CCDSOFTC].n_value, (char *)&kcs, 540 sizeof(kcs)) != sizeof(kcs)) { 541 free(cs); 542 KVM_ABORT(kd, "can't find pointer to configuration data"); 543 } 544 if (kvm_read(kd, (u_long)kcs, (char *)cs, readsize) != readsize) { 545 free(cs); 546 KVM_ABORT(kd, "can't read configuration data"); 547 } 548 549 if (argc == 0) { 550 for (i = 0; i < numccd; ++i) 551 if (cs[i].sc_flags & CCDF_INITED) { 552 ++numconfiged; 553 print_ccd_info(&cs[i], kd); 554 } 555 556 if (numconfiged == 0) 557 printf("no concatenated disks configured\n"); 558 } else { 559 while (argc) { 560 cp = *argv++; --argc; 561 if ((ccd = resolve_ccdname(cp)) == NULL) { 562 warnx("invalid ccd name: %s", cp); 563 continue; 564 } 565 if ((error = pathtounit(ccd, &i)) != 0) { 566 warnx("%s: %s", ccd, strerror(error)); 567 continue; 568 } 569 if (i >= numccd) { 570 warnx("ccd%d not configured", i); 571 continue; 572 } 573 if (cs[i].sc_flags & CCDF_INITED) 574 print_ccd_info(&cs[i], kd); 575 else 576 printf("ccd%d not configured\n", i); 577 } 578 } 579 580 free(cs); 581 582 done: 583 (void)kvm_close(kd); 584 return (0); 585 586 bad: 587 (void)kvm_close(kd); 588 return (1); 589 } 590 591 static void 592 print_ccd_info(cs, kd) 593 struct ccd_softc *cs; 594 kvm_t *kd; 595 { 596 static int header_printed = 0; 597 struct ccdcinfo *cip; 598 size_t readsize; 599 char path[MAXPATHLEN]; 600 int i; 601 602 if (header_printed == 0 && verbose) { 603 printf("# ccd\t\tileave\tflags\tcompnent devices\n"); 604 header_printed = 1; 605 } 606 607 readsize = cs->sc_nccdisks * sizeof(struct ccdcinfo); 608 if ((cip = malloc(readsize)) == NULL) { 609 warn("ccd%d: can't allocate memory for component info", 610 cs->sc_unit); 611 return; 612 } 613 bzero(cip, readsize); 614 615 /* Dump out softc information. */ 616 printf("ccd%d\t\t%d\t%d\t", cs->sc_unit, cs->sc_ileave, 617 cs->sc_cflags & CCDF_USERMASK); 618 fflush(stdout); 619 620 /* Read in the component info. */ 621 if (kvm_read(kd, (u_long)cs->sc_cinfo, (char *)cip, 622 readsize) != readsize) { 623 printf("\n"); 624 warnx("can't read component info"); 625 warnx("%s", kvm_geterr(kd)); 626 goto done; 627 } 628 629 /* Read component pathname and display component info. */ 630 for (i = 0; i < cs->sc_nccdisks; ++i) { 631 if (kvm_read(kd, (u_long)cip[i].ci_path, (char *)path, 632 cip[i].ci_pathlen) != cip[i].ci_pathlen) { 633 printf("\n"); 634 warnx("can't read component pathname"); 635 warnx("%s", kvm_geterr(kd)); 636 goto done; 637 } 638 printf((i + 1 < cs->sc_nccdisks) ? "%s " : "%s\n", path); 639 fflush(stdout); 640 } 641 642 done: 643 free(cip); 644 } 645 646 static int 647 getmaxpartitions() 648 { 649 return (MAXPARTITIONS); 650 } 651 652 static int 653 getrawpartition() 654 { 655 return (RAW_PART); 656 } 657 658 static int 659 flags_to_val(flags) 660 char *flags; 661 { 662 char *cp, *tok; 663 int i, tmp, val = ~CCDF_USERMASK; 664 size_t flagslen; 665 666 /* 667 * The most common case is that of NIL flags, so check for 668 * those first. 669 */ 670 if (strcmp("none", flags) == 0 || strcmp("0x0", flags) == 0 || 671 strcmp("0", flags) == 0) 672 return (0); 673 674 flagslen = strlen(flags); 675 676 /* Check for values represented by strings. */ 677 if ((cp = strdup(flags)) == NULL) 678 err(1, "no memory to parse flags"); 679 tmp = 0; 680 for (tok = cp; (tok = strtok(tok, ",")) != NULL; tok = NULL) { 681 for (i = 0; flagvaltab[i].fv_flag != NULL; ++i) 682 if (strcmp(tok, flagvaltab[i].fv_flag) == 0) 683 break; 684 if (flagvaltab[i].fv_flag == NULL) { 685 free(cp); 686 goto bad_string; 687 } 688 tmp |= flagvaltab[i].fv_val; 689 } 690 691 /* If we get here, the string was ok. */ 692 free(cp); 693 val = tmp; 694 goto out; 695 696 bad_string: 697 698 /* Check for values represented in hex. */ 699 if (flagslen > 2 && flags[0] == '0' && flags[1] == 'x') { 700 errno = 0; /* to check for ERANGE */ 701 val = (int)strtol(&flags[2], &cp, 16); 702 if ((errno == ERANGE) || (*cp != '\0')) 703 return (-1); 704 goto out; 705 } 706 707 /* Check for values represented in decimal. */ 708 errno = 0; /* to check for ERANGE */ 709 val = (int)strtol(flags, &cp, 10); 710 if ((errno == ERANGE) || (*cp != '\0')) 711 return (-1); 712 713 out: 714 return (((val & ~CCDF_USERMASK) == 0) ? val : -1); 715 } 716 717 static void 718 usage() 719 { 720 fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n", 721 "usage: ccdconfig [-cv] ccd ileave [flags] dev [...]", 722 " ccdconfig -C [-v] [-f config_file]", 723 " ccdconfig -u [-v] ccd [...]", 724 " ccdconfig -U [-v] [-f config_file]", 725 " ccdconfig -g [-M core] [-N system] [ccd [...]]"); 726 exit(1); 727 } 728 729 /* Local Variables: */ 730 /* c-argdecl-indent: 8 */ 731 /* c-indent-level: 8 */ 732 /* End: */ 733