varsym: Update man page & sync usage()
[dragonfly.git] / bin / varsym / varsym.c
CommitLineData
65b3153c
MD
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 *
5c0b1e71 26 * $DragonFly: src/bin/varsym/varsym.c,v 1.4 2003/12/11 20:33:49 dillon Exp $
65b3153c
MD
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
0a004aa2
MD
36static void usage(void);
37static void dumpvars(char *buf, int bytes);
aac72047 38
65b3153c
MD
39int
40main(int ac, char **av)
41{
aac72047
EN
42 int i;
43 int mask = VARSYM_ALL_MASK;
44 int level = VARSYM_USER;
45 int deleteOpt = 0;
46 int verboseOpt = 1;
0a004aa2 47 int allOpt = 0;
65b3153c 48
0a004aa2 49 while ((i = getopt(ac, av, "adhpqsu")) != -1) {
aac72047 50 switch (i) {
0a004aa2
MD
51 case 'a':
52 allOpt = 1;
53 break;
aac72047
EN
54 case 'd':
55 deleteOpt = 1;
56 break;
57 case 'p':
58 mask = VARSYM_PROC_MASK;
59 level = VARSYM_PROC;
60 break;
61 case 'q':
62 verboseOpt = 0;
63 break;
64 case 's':
65 mask = VARSYM_SYS_MASK;
66 level = VARSYM_SYS;
67 break;
68 case 'u':
69 mask = VARSYM_USER_MASK;
70 level = VARSYM_USER;
71 break;
72 case 'h':
73 default:
74 usage();
75 return(-1);
76 }
65b3153c 77 }
65b3153c 78
0a004aa2
MD
79 if (allOpt) {
80 char buf[1024];
81 int marker = 0;
82 int bytes;
83
84 for (;;) {
85 bytes = varsym_list(level, buf, sizeof(buf), &marker);
86 if (bytes < 0) /* error occured */
87 break;
88 dumpvars(buf, bytes);
89 if (marker < 0) /* no more vars */
90 break;
91 }
92 if (bytes < 0)
93 fprintf(stderr, "varsym_list(): %s\n", strerror(errno));
94 }
95
aac72047
EN
96 for ( ; optind < ac; optind++) {
97 char *name = av[optind];
98 char *data = strchr(name, '=');
99 int error;
100 char buf[MAXVARSYM_DATA];
65b3153c 101
aac72047
EN
102 if (data)
103 *data++ = 0;
104
105 if (deleteOpt) {
106 error = varsym_set(level, name, NULL);
107 }
108 else if (data) {
109 error = varsym_set(level, name, data);
110 }
111 else {
112 error = varsym_get(mask, name, buf, sizeof(buf));
113 if (error >= 0 && error <= (int)sizeof(buf)) {
114 if (verboseOpt)
115 printf("%s=", name);
116 printf("%s\n", buf);
117 }
118 }
5c0b1e71 119 if (error < 0 && verboseOpt)
aac72047 120 fprintf(stderr, "%s: %s\n", name, strerror(errno));
65b3153c 121 }
aac72047
EN
122
123 return(0);
124}
125
0a004aa2
MD
126static void
127dumpvars(char *buf, int bytes)
128{
129 int b;
130 int i;
131 char *vname = NULL;
132 char *vdata = NULL;
133
134 for (b = i = 0; i < bytes; ++i) {
135 if (buf[i] == 0) {
136 if (vname == NULL) {
137 vname = buf + b;
138 } else {
139 vdata = buf + b;
140 printf("%s=%s\n", vname, vdata);
141 vname = vdata = NULL;
142 }
143 b = i + 1;
144 }
145 }
146}
147
148static void
aac72047
EN
149usage(void)
150{
1072b009 151 fprintf(stderr, "usage: varsym [-adpqsu] [var[=data] ...]\n");
65b3153c
MD
152}
153