sbin/ccdconfig/ccdconfig.c

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