Add the 'varsym' utility which may be used to manipulate system-wide and
[dragonfly.git] / bin / varsym / varsym.c
1 /*
2  * Copyright (c) 2003 Matthew Dillon <dillon@backplane.com>
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  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $DragonFly: src/bin/varsym/varsym.c,v 1.1 2003/11/05 23:29:37 dillon Exp $
27  */
28
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <string.h>
32 #include <errno.h>
33 #include <unistd.h>
34 #include <sys/varsym.h>
35
36 int
37 main(int ac, char **av)
38 {
39     int i;
40     int mask =  VARSYM_ALL_MASK;
41     int level = VARSYM_USER;
42     int nary = 0;
43     int mary = 8;
44     int deleteOpt = 0;
45     int verboseOpt = 1;
46     char **ary;
47
48     ary = malloc(sizeof(char *) * mary);
49     for (i = 1; i < ac; ++i) {
50         char *ptr = av[i];
51         if (*ptr != '-') {
52             ary[nary++] = ptr;
53             if (nary == mary) {
54                 mary *= 2;
55                 ary = realloc(ary, sizeof(char *) * mary);
56             }
57             continue;
58         }
59         ptr += 2;
60         switch(ptr[-1]) {
61         case 'q':
62             verboseOpt = 0;
63             break;
64         case 'd':
65             deleteOpt = 1;
66             break;
67         case 's':
68             mask = VARSYM_SYS_MASK;
69             level = VARSYM_SYS;
70             break;
71         case 'u':
72             mask = VARSYM_USER_MASK;
73             level = VARSYM_USER;
74             break;
75         case 'p':
76             mask = VARSYM_PROC_MASK;
77             level = VARSYM_PROC;
78             break;
79         }
80     }
81     for (i = 0; i < nary; ++i) {
82         char *name = ary[i];
83         char *data = strchr(name, '=');
84         int error;
85         char buf[MAXVARSYM_DATA];
86
87         if (data)
88             *data++ = 0;
89
90         if (deleteOpt) {
91             error = varsym_set(level, name, NULL);
92         } else if (data) {
93             error = varsym_set(level, name, data);
94         } else {
95             error = varsym_get(mask, name, buf, sizeof(buf));
96             if (error >= 0 && error <= (int)sizeof(buf)) {
97                 if (verboseOpt)
98                     printf("%s=", name);
99                 printf("%s\n", buf);
100             }
101         }
102         if (error < 0)
103             fprintf(stderr, "%s: %s\n", name, strerror(errno));
104     }
105     return(0);
106 }
107