sbin/sysctl/sysctl.c

   1 /*
   2  * Copyright (c) 1993
   3  *      The Regents of the University of California.  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 by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  * @(#) Copyright (c) 1993 The Regents of the University of California.  All rights reserved.
  34  * @(#)from: sysctl.c   8.1 (Berkeley) 6/6/93
  35  * $FreeBSD: src/sbin/sysctl/sysctl.c,v 1.25.2.11 2003/05/01 22:48:08 trhodes Exp $
  36  * $DragonFly: src/sbin/sysctl/sysctl.c,v 1.14 2007/01/06 00:00:10 corecode Exp $
  37  */
  38
  39 #ifdef __i386__
  40 #include <sys/diskslice.h>      /* used for bootdev parsing */
  41 #include <sys/reboot.h>         /* used for bootdev parsing */
  42 #endif
  43 #include <sys/types.h>
  44 #include <sys/stat.h>
  45 #include <sys/sysctl.h>
  46 #include <sys/resource.h>
  47 #include <sys/param.h>
  48
  49 #include <machine/inttypes.h>
  50
  51 #include <ctype.h>
  52 #include <err.h>
  53 #include <errno.h>
  54 #include <stdio.h>
  55 #include <stdlib.h>
  56 #include <string.h>
  57 #include <unistd.h>
  58
  59 static int      aflag, bflag, dflag, eflag, Nflag, nflag, oflag, xflag;
  60
  61 static int      oidfmt(int *, size_t, char *, u_int *);
  62 static void     parse(const char *);
  63 static int      show_var(int *, size_t);
  64 static int      sysctl_all(int *, size_t);
  65 static void     set_T_dev_t(const char *, void **, int *);
  66
  67 static void
  68 usage(void)
  69 {
  70
  71         fprintf(stderr, "%s\n%s\n",
  72             "usage: sysctl [-bdeNnox] variable[=value] ...",
  73             "       sysctl [-bdeNnox] -a");
  74         exit(1);
  75 }
  76
  77 int
  78 main(int argc, char **argv)
  79 {
  80         int ch;
  81         setbuf(stdout,0);
  82         setbuf(stderr,0);
  83
  84         while ((ch = getopt(argc, argv, "AabdeNnowxX")) != -1) {
  85                 switch (ch) {
  86                 case 'A':
  87                         /* compatibility */
  88                         aflag = oflag = 1;
  89                         break;
  90                 case 'a':
  91                         aflag = 1;
  92                         break;
  93                 case 'b':
  94                         bflag = 1;
  95                         break;
  96                 case 'd':
  97                         dflag = 1;
  98                         break;
  99                 case 'e':
 100                         eflag = 1;
 101                         break;
 102                 case 'N':
 103                         Nflag = 1;
 104                         break;
 105                 case 'n':
 106                         nflag = 1;
 107                         break;
 108                 case 'o':
 109                         oflag = 1;
 110                         break;
 111                 case 'w':
 112                         /* compatibility */
 113                         /* ignored */
 114                         break;
 115                 case 'X':
 116                         /* compatibility */
 117                         aflag = xflag = 1;
 118                         break;
 119                 case 'x':
 120                         xflag = 1;
 121                         break;
 122                 default:
 123                         usage();
 124                 }
 125         }
 126         argc -= optind;
 127         argv += optind;
 128
 129         if (Nflag && nflag)
 130                 usage();
 131         if (aflag && argc == 0)
 132                 exit(sysctl_all(0, 0));
 133         if (argc == 0)
 134                 usage();
 135         while (argc-- > 0)
 136                 parse(*argv++);
 137         exit(0);
 138 }
 139
 140 /*
 141  * Parse a name into a MIB entry.
 142  * Lookup and print out the MIB entry if it exists.
 143  * Set a new value if requested.
 144  */
 145 static void
 146 parse(const char *string)
 147 {
 148         size_t len;
 149         int i, j;
 150         void *newval = NULL;
 151         int intval;
 152         unsigned int uintval;
 153         long longval;
 154         unsigned long ulongval;
 155         size_t newsize = 0;
 156         quad_t quadval;
 157         u_quad_t uquadval;
 158         int mib[CTL_MAXNAME];
 159         char *cp, fmt[BUFSIZ];
 160         const char *name;
 161         char *name_allocated = NULL;
 162         u_int kind;
 163
 164         if ((cp = strchr(string, '=')) != NULL) {
 165                 if ((name_allocated = malloc(cp - string + 1)) == NULL)
 166                         err(1, "malloc failed");
 167                 strlcpy(name_allocated, string, cp - string + 1);
 168                 name = name_allocated;
 169
 170                 while (isspace(*++cp))
 171                         ;
 172
 173                 newval = cp;
 174                 newsize = strlen(cp);
 175         } else {
 176                 name = string;
 177         }
 178
 179         len = CTL_MAXNAME;
 180         if (sysctlnametomib(name, mib, &len) < 0) {
 181                 if (errno == ENOENT) {
 182                         errx(1, "unknown oid '%s'", name);
 183                 } else {
 184                         err(1, "sysctlnametomib(\"%s\")", name);
 185                 }
 186         }
 187
 188         if (oidfmt(mib, len, fmt, &kind))
 189                 err(1, "couldn't find format of oid '%s'", name);
 190
 191         if (newval == NULL) {
 192                 if ((kind & CTLTYPE) == CTLTYPE_NODE) {
 193                         sysctl_all(mib, len);
 194                 } else {
 195                         i = show_var(mib, len);
 196                         if (!i && !bflag)
 197                                 putchar('\n');
 198                 }
 199         } else {
 200                 if ((kind & CTLTYPE) == CTLTYPE_NODE)
 201                         errx(1, "oid '%s' isn't a leaf node", name);
 202
 203                 if (!(kind&CTLFLAG_WR))
 204                         errx(1, "oid '%s' is read only", name);
 205
 206                 switch (kind & CTLTYPE) {
 207                         case CTLTYPE_INT:
 208                                 intval = (int) strtol(newval, NULL, 0);
 209                                 newval = &intval;
 210                                 newsize = sizeof(intval);
 211                                 break;
 212                         case CTLTYPE_UINT:
 213                                 uintval = (int) strtoul(newval, NULL, 0);
 214                                 newval = &uintval;
 215                                 newsize = sizeof uintval;
 216                                 break;
 217                         case CTLTYPE_LONG:
 218                                 longval = strtol(newval, NULL, 0);
 219                                 newval = &longval;
 220                                 newsize = sizeof longval;
 221                                 break;
 222                         case CTLTYPE_ULONG:
 223                                 ulongval = strtoul(newval, NULL, 0);
 224                                 newval = &ulongval;
 225                                 newsize = sizeof ulongval;
 226                                 break;
 227                         case CTLTYPE_STRING:
 228                                 break;
 229                         case CTLTYPE_QUAD:
 230                                 quadval = strtoq(newval, NULL, 0);
 231                                 newval = &quadval;
 232                                 newsize = sizeof(quadval);
 233                                 break;
 234                         case CTLTYPE_UQUAD:
 235                                 uquadval = strtouq(newval, NULL, 0);
 236                                 newval = &quadval;
 237                                 newsize = sizeof(quadval);
 238                                 break;
 239                         case CTLTYPE_OPAQUE:
 240                                 if (strcmp(fmt, "T,dev_t") == 0 ||
 241                                     strcmp(fmt, "T,udev_t") == 0
 242                                 ) {
 243                                         set_T_dev_t((char*)newval, &newval,
 244                                                     &newsize);
 245                                         break;
 246                                 }
 247                                 /* FALLTHROUGH */
 248                         default:
 249                                 errx(1, "oid '%s' is type %d,"
 250                                         " cannot set that", name,
 251                                         kind & CTLTYPE);
 252                 }
 253
 254                 i = show_var(mib, len);
 255                 if (sysctl(mib, len, 0, 0, newval, newsize) == -1) {
 256                         if (!i && !bflag)
 257                                 putchar('\n');
 258                         switch (errno) {
 259                         case EOPNOTSUPP:
 260                                 errx(1, "%s: value is not available",
 261                                         string);
 262                         case ENOTDIR:
 263                                 errx(1, "%s: specification is incomplete",
 264                                         string);
 265                         case ENOMEM:
 266                                 errx(1, "%s: type is unknown to this program",
 267                                         string);
 268                         default:
 269                                 warn("%s", string);
 270                                 return;
 271                         }
 272                 }
 273                 if (!bflag)
 274                         printf(" -> ");
 275                 i = nflag;
 276                 nflag = 1;
 277                 j = show_var(mib, len);
 278                 if (!j && !bflag)
 279                         putchar('\n');
 280                 nflag = i;
 281         }
 282
 283         if (name_allocated != NULL)
 284                 free(name_allocated);
 285 }
 286
 287 /* These functions will dump out various interesting structures. */
 288
 289 static int
 290 S_clockinfo(int l2, void *p)
 291 {
 292         struct clockinfo *ci = (struct clockinfo*)p;
 293         if (l2 != sizeof(*ci))
 294                 err(1, "S_clockinfo %d != %d", l2, sizeof(*ci));
 295         printf("{ hz = %d, tick = %d, tickadj = %d, profhz = %d, stathz = %d }",
 296                 ci->hz, ci->tick, ci->tickadj, ci->profhz, ci->stathz);
 297         return (0);
 298 }
 299
 300 static int
 301 S_loadavg(int l2, void *p)
 302 {
 303         struct loadavg *tv = (struct loadavg*)p;
 304
 305         if (l2 != sizeof(*tv))
 306                 err(1, "S_loadavg %d != %d", l2, sizeof(*tv));
 307
 308         printf("{ %.2f %.2f %.2f }",
 309                 (double)tv->ldavg[0]/(double)tv->fscale,
 310                 (double)tv->ldavg[1]/(double)tv->fscale,
 311                 (double)tv->ldavg[2]/(double)tv->fscale);
 312         return (0);
 313 }
 314
 315 static int
 316 S_timespec(int l2, void *p)
 317 {
 318         struct timespec *ts = (struct timespec*)p;
 319         time_t tv_sec;
 320         char *p1, *p2;
 321
 322         if (l2 != sizeof(*ts))
 323                 err(1, "S_timespec %d != %d", l2, sizeof(*ts));
 324         printf("{ sec = %ld, nsec = %ld } ",
 325                 ts->tv_sec, ts->tv_nsec);
 326         tv_sec = ts->tv_sec;
 327         p1 = strdup(ctime(&tv_sec));
 328         for (p2=p1; *p2 ; p2++)
 329                 if (*p2 == '\n')
 330                         *p2 = '\0';
 331         fputs(p1, stdout);
 332         return (0);
 333 }
 334
 335 static int
 336 S_timeval(int l2, void *p)
 337 {
 338         struct timeval *tv = (struct timeval*)p;
 339         time_t tv_sec;
 340         char *p1, *p2;
 341
 342         if (l2 != sizeof(*tv))
 343                 err(1, "S_timeval %d != %d", l2, sizeof(*tv));
 344         printf("{ sec = %ld, usec = %ld } ",
 345                 tv->tv_sec, tv->tv_usec);
 346         tv_sec = tv->tv_sec;
 347         p1 = strdup(ctime(&tv_sec));
 348         for (p2=p1; *p2 ; p2++)
 349                 if (*p2 == '\n')
 350                         *p2 = '\0';
 351         fputs(p1, stdout);
 352         return (0);
 353 }
 354
 355 static int
 356 T_dev_t(int l2, void *p)
 357 {
 358         dev_t *d = (dev_t *)p;
 359         if (l2 != sizeof(*d))
 360                 err(1, "T_dev_T %d != %d", l2, sizeof(*d));
 361         if ((int)(*d) != -1) {
 362                 if (minor(*d) > 255 || minor(*d) < 0)
 363                         printf("{ major = %d, minor = 0x%x }",
 364                                 major(*d), minor(*d));
 365                 else
 366                         printf("{ major = %d, minor = %d }",
 367                                 major(*d), minor(*d));
 368         }
 369         return (0);
 370 }
 371
 372 static void
 373 set_T_dev_t(const char *path, void **val, int *size)
 374 {
 375         static struct stat statb;
 376
 377         if (strcmp(path, "none") && strcmp(path, "off")) {
 378                 int rc = stat (path, &statb);
 379                 if (rc) {
 380                         err(1, "cannot stat %s", path);
 381                 }
 382
 383                 if (!S_ISCHR(statb.st_mode)) {
 384                         errx(1, "must specify a device special file.");
 385                 }
 386         } else {
 387                 statb.st_rdev = NODEV;
 388         }
 389         *val = (char*) &statb.st_rdev;
 390         *size = sizeof statb.st_rdev;
 391 }
 392
 393 /*
 394  * These functions uses a presently undocumented interface to the kernel
 395  * to walk the tree and get the type so it can print the value.
 396  * This interface is under work and consideration, and should probably
 397  * be killed with a big axe by the first person who can find the time.
 398  * (be aware though, that the proper interface isn't as obvious as it
 399  * may seem, there are various conflicting requirements.
 400  */
 401
 402 static int
 403 oidfmt(int *oid, size_t len, char *fmt, u_int *kind)
 404 {
 405         int qoid[CTL_MAXNAME+2];
 406         u_char buf[BUFSIZ];
 407         int i;
 408         size_t j;
 409
 410         qoid[0] = 0;
 411         qoid[1] = 4;
 412         memcpy(qoid + 2, oid, len * sizeof(int));
 413
 414         j = sizeof(buf);
 415         i = sysctl(qoid, len + 2, buf, &j, 0, 0);
 416         if (i)
 417                 err(1, "sysctl fmt %d %d %d", i, j, errno);
 418
 419         if (kind)
 420                 *kind = *(u_int *)buf;
 421
 422         if (fmt)
 423                 strcpy(fmt, (char *)(buf + sizeof(u_int)));
 424         return 0;
 425 }
 426
 427 #ifdef __i386__
 428 /*
 429  * Code to map a bootdev major number into a suitable device name.
 430  * Major numbers are mapped into names as in boot2.c
 431  */
 432 struct _foo {
 433         int majdev;
 434         const char *name;
 435 } maj2name[] = {
 436         { 30,   "ad" },
 437         { 0,    "wd" },
 438         { 1,    "wfd" },
 439         { 2,    "fd" },
 440         { 4,    "da" },
 441         { -1,   NULL }  /* terminator */
 442 };
 443
 444 static int
 445 machdep_bootdev(u_long value)
 446 {
 447         int majdev, unit, slice, part;
 448         struct _foo *p;
 449
 450         if ((value & B_MAGICMASK) != B_DEVMAGIC) {
 451                 printf("invalid (0x%08lx)", value);
 452                 return 0;
 453         }
 454         majdev = B_TYPE(value);
 455         unit = B_UNIT(value);
 456         slice = B_SLICE(value);
 457         part = B_PARTITION(value);
 458         if (majdev == 2) {      /* floppy, as known to the boot block... */
 459                 printf("/dev/fd%d", unit);
 460                 return 0;
 461         }
 462         for (p = maj2name; p->name != NULL && p->majdev != majdev ; p++) ;
 463         if (p->name != NULL) {  /* found */
 464                 if (slice == WHOLE_DISK_SLICE)
 465                         printf("/dev/%s%d%c", p->name, unit, part);
 466                 else
 467                         printf("/dev/%s%ds%d%c",
 468                             p->name, unit, slice - BASE_SLICE + 1, part + 'a');
 469         } else
 470                 printf("unknown (major %d unit %d slice %d part %d)",
 471                         majdev, unit, slice, part);
 472         return 0;
 473 }
 474 #endif
 475
 476 /*
 477  * This formats and outputs the value of one variable
 478  *
 479  * Returns zero if anything was actually output.
 480  * Returns one if didn't know what to do with this.
 481  * Return minus one if we had errors.
 482  */
 483
 484 static int
 485 show_var(int *oid, size_t nlen)
 486 {
 487         u_char buf[BUFSIZ], *val, *p, *nul;
 488         char name[BUFSIZ], *fmt;
 489         const char *sep, *spacer;
 490         int qoid[CTL_MAXNAME+2];
 491         int i;
 492         size_t j, len;
 493         u_int kind;
 494         int (*func)(int, void *);
 495
 496         qoid[0] = 0;
 497         memcpy(qoid + 2, oid, nlen * sizeof(int));
 498
 499         qoid[1] = 1;
 500         j = sizeof(name);
 501         i = sysctl(qoid, nlen + 2, name, &j, 0, 0);
 502         if (i || !j)
 503                 err(1, "sysctl name %d %d %d", i, j, errno);
 504
 505         if (Nflag) {
 506                 printf("%s", name);
 507                 return (0);
 508         }
 509
 510         if (eflag)
 511                 sep = "=";
 512         else
 513                 sep = ": ";
 514
 515         if (dflag) {    /* just print description */
 516                 qoid[1] = 5;
 517                 j = sizeof(buf);
 518                 i = sysctl(qoid, nlen + 2, buf, &j, 0, 0);
 519                 if (!nflag)
 520                         printf("%s%s", name, sep);
 521                 printf("%s", buf);
 522                 return(0);
 523         }
 524         /* find an estimate of how much we need for this var */
 525         j = 0;
 526         i = sysctl(oid, nlen, 0, &j, 0, 0);
 527         j += j; /* we want to be sure :-) */
 528
 529         val = alloca(j + 1);
 530         len = j;
 531         i = sysctl(oid, nlen, val, &len, 0, 0);
 532         if (i || !len)
 533                 return (1);
 534
 535         if (bflag) {
 536                 fwrite(val, 1, len, stdout);
 537                 return (0);
 538         }
 539
 540         val[len] = '\0';
 541         fmt = buf;
 542         oidfmt(oid, nlen, fmt, &kind);
 543         p = val;
 544         switch (*fmt) {
 545         case 'A':
 546                 if (!nflag)
 547                         printf("%s%s", name, sep);
 548                 nul = memchr(p, '\0', len);
 549                 fwrite(p, nul == NULL ? len : nul - p, 1, stdout);
 550                 return (0);
 551
 552         case 'I':
 553                 if (!nflag)
 554                         printf("%s%s", name, sep);
 555                 fmt++;
 556                 spacer = "";
 557                 while (len >= sizeof(int)) {
 558                         if(*fmt == 'U')
 559                                 printf("%s%u", spacer, *(unsigned int *)p);
 560                         else
 561                                 printf("%s%d", spacer, *(int *)p);
 562                         spacer = " ";
 563                         len -= sizeof(int);
 564                         p += sizeof(int);
 565                 }
 566                 return (0);
 567
 568         case 'L':
 569                 if (!nflag)
 570                         printf("%s%s", name, sep);
 571                 fmt++;
 572 #ifdef __i386__
 573                 if (!strcmp(name, "machdep.guessed_bootdev"))
 574                         return machdep_bootdev(*(unsigned long *)p);
 575 #endif
 576                 spacer = "";
 577                 while (len >= sizeof(long)) {
 578                         if(*fmt == 'U')
 579                                 printf("%s%lu", spacer, *(unsigned long *)p);
 580                         else
 581                                 printf("%s%ld", spacer, *(long *)p);
 582                         spacer = " ";
 583                         len -= sizeof(long);
 584                         p += sizeof(long);
 585                 }
 586                 return (0);
 587
 588         case 'P':
 589                 if (!nflag)
 590                         printf("%s%s", name, sep);
 591                 printf("%p", *(void **)p);
 592                 return (0);
 593
 594         case 'Q':
 595                 if (!nflag)
 596                         printf("%s%s", name, sep);
 597                 fmt++;
 598                 spacer = "";
 599                 while (len >= sizeof(quad_t)) {
 600                         if(*fmt == 'U')
 601                                 printf("%s%qu", spacer, *(u_quad_t *)p);
 602                         else
 603                                 printf("%s%qd", spacer, *(quad_t *)p);
 604                         spacer = " ";
 605                         len -= sizeof(int64_t);
 606                         p += sizeof(int64_t);
 607                 }
 608                 return (0);
 609
 610         case 'T':
 611         case 'S':
 612                 if (!oflag && !xflag) {
 613                         i = 0;
 614                         if (strcmp(fmt, "S,clockinfo") == 0)
 615                                 func = S_clockinfo;
 616                         else if (strcmp(fmt, "S,timespec") == 0)
 617                                 func = S_timespec;
 618                         else if (strcmp(fmt, "S,timeval") == 0)
 619                                 func = S_timeval;
 620                         else if (strcmp(fmt, "S,loadavg") == 0)
 621                                 func = S_loadavg;
 622                         else if (strcmp(fmt, "T,dev_t") == 0)
 623                                 func = T_dev_t;
 624                         else if (strcmp(fmt, "T,udev_t") == 0)
 625                                 func = T_dev_t;
 626                         else
 627                                 func = NULL;
 628                         if (func) {
 629                                 if (!nflag)
 630                                         printf("%s%s", name, sep);
 631                                 return ((*func)(len, p));
 632                         }
 633                 }
 634                 /* FALL THROUGH */
 635         default:
 636                 if (!oflag && !xflag)
 637                         return (1);
 638                 if (!nflag)
 639                         printf("%s%s", name, sep);
 640                 printf("Format:%s Length:%d Dump:0x", fmt, len);
 641                 while (len-- && (xflag || p < val + 16))
 642                         printf("%02x", *p++);
 643                 if (!xflag && len > 16)
 644                         printf("...");
 645                 return (0);
 646         }
 647         return (1);
 648 }
 649
 650 static int
 651 sysctl_all(int *oid, size_t len)
 652 {
 653         int name1[22], name2[22];
 654         int retval;
 655         size_t i, l1, l2;
 656
 657         name1[0] = 0;
 658         name1[1] = 2;
 659         l1 = 2;
 660         if (len) {
 661                 memcpy(name1+2, oid, len * sizeof(int));
 662                 l1 += len;
 663         } else {
 664                 name1[2] = 1;
 665                 l1++;
 666         }
 667         for (;;) {
 668                 l2 = sizeof(name2);
 669                 retval = sysctl(name1, l1, name2, &l2, 0, 0);
 670                 if (retval < 0) {
 671                         if (errno == ENOENT)
 672                                 return 0;
 673                         else
 674                                 err(1, "sysctl(getnext) %d %d", retval, l2);
 675                 }
 676
 677                 l2 /= sizeof(int);
 678
 679                 if (l2 < len)
 680                         return 0;
 681
 682                 for (i = 0; i < len; i++)
 683                         if (name2[i] != oid[i])
 684                                 return 0;
 685
 686                 retval = show_var(name2, l2);
 687                 if (retval == 0 && !bflag)
 688                         putchar('\n');
 689
 690                 memcpy(name1+2, name2, l2 * sizeof(int));
 691                 l1 = 2 + l2;
 692         }
 693 }