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