xref: /freebsd/usr.bin/uname/uname.c (revision f05cddf9)
1 /*-
2  * Copyright (c) 2002 Juli Mallett.
3  * Copyright (c) 1993
4  *	The Regents of the University of California.  All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. All advertising materials mentioning features or use of this software
15  *    must display the following acknowledgement:
16  *	This product includes software developed by the University of
17  *	California, Berkeley and its contributors.
18  * 4. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34 
35 #include <sys/cdefs.h>
36 
37 __FBSDID("$FreeBSD$");
38 
39 #ifndef lint
40 static const char copyright[] =
41 "@(#) Copyright (c) 1993\n\
42 	The Regents of the University of California.  All rights reserved.\n";
43 #endif
44 
45 #ifndef lint
46 static const char sccsid[] = "@(#)uname.c	8.2 (Berkeley) 5/4/95";
47 #endif
48 
49 #include <sys/param.h>
50 #include <sys/sysctl.h>
51 
52 #include <err.h>
53 #include <stdio.h>
54 #include <stdlib.h>
55 #include <unistd.h>
56 
57 #define	MFLAG	0x01
58 #define	NFLAG	0x02
59 #define	PFLAG	0x04
60 #define	RFLAG	0x08
61 #define	SFLAG	0x10
62 #define	VFLAG	0x20
63 #define	IFLAG	0x40
64 
65 typedef void (*get_t)(void);
66 static get_t get_ident, get_platform, get_hostname, get_arch,
67     get_release, get_sysname, get_version;
68 
69 static void native_ident(void);
70 static void native_platform(void);
71 static void native_hostname(void);
72 static void native_arch(void);
73 static void native_release(void);
74 static void native_sysname(void);
75 static void native_version(void);
76 static void print_uname(u_int);
77 static void setup_get(void);
78 static void usage(void);
79 
80 static char *ident, *platform, *hostname, *arch, *release, *sysname, *version;
81 static int space;
82 
83 int
84 main(int argc, char *argv[])
85 {
86 	u_int flags;
87 	int ch;
88 
89 	setup_get();
90 	flags = 0;
91 
92 	while ((ch = getopt(argc, argv, "aimnoprsv")) != -1)
93 		switch(ch) {
94 		case 'a':
95 			flags |= (MFLAG | NFLAG | RFLAG | SFLAG | VFLAG);
96 			break;
97 		case 'i':
98 			flags |= IFLAG;
99 			break;
100 		case 'm':
101 			flags |= MFLAG;
102 			break;
103 		case 'n':
104 			flags |= NFLAG;
105 			break;
106 		case 'p':
107 			flags |= PFLAG;
108 			break;
109 		case 'r':
110 			flags |= RFLAG;
111 			break;
112 		case 's':
113 		case 'o':
114 			flags |= SFLAG;
115 			break;
116 		case 'v':
117 			flags |= VFLAG;
118 			break;
119 		case '?':
120 		default:
121 			usage();
122 		}
123 
124 	argc -= optind;
125 	argv += optind;
126 
127 	if (argc)
128 		usage();
129 
130 	if (!flags)
131 		flags |= SFLAG;
132 
133 	print_uname(flags);
134 	exit(0);
135 }
136 
137 #define	CHECK_ENV(opt,var)				\
138 do {							\
139 	if ((var = getenv("UNAME_" opt)) == NULL) {	\
140 		get_##var = native_##var;		\
141 	} else {					\
142 		get_##var = (get_t)NULL;		\
143 	}						\
144 } while (0)
145 
146 static void
147 setup_get(void)
148 {
149 	CHECK_ENV("s", sysname);
150 	CHECK_ENV("n", hostname);
151 	CHECK_ENV("r", release);
152 	CHECK_ENV("v", version);
153 	CHECK_ENV("m", platform);
154 	CHECK_ENV("p", arch);
155 	CHECK_ENV("i", ident);
156 }
157 
158 #define	PRINT_FLAG(flags,flag,var)		\
159 	if ((flags & flag) == flag) {		\
160 		if (space)			\
161 			printf(" ");		\
162 		else				\
163 			space++;		\
164 		if (get_##var != NULL)		\
165 			(*get_##var)();		\
166 		printf("%s", var);		\
167 	}
168 
169 static void
170 print_uname(u_int flags)
171 {
172 	PRINT_FLAG(flags, SFLAG, sysname);
173 	PRINT_FLAG(flags, NFLAG, hostname);
174 	PRINT_FLAG(flags, RFLAG, release);
175 	PRINT_FLAG(flags, VFLAG, version);
176 	PRINT_FLAG(flags, MFLAG, platform);
177 	PRINT_FLAG(flags, PFLAG, arch);
178 	PRINT_FLAG(flags, IFLAG, ident);
179 	printf("\n");
180 }
181 
182 #define	NATIVE_SYSCTL2_GET(var,mib0,mib1)	\
183 static void					\
184 native_##var(void)				\
185 {						\
186 	int mib[] = { (mib0), (mib1) };		\
187 	size_t len;				\
188 	static char buf[1024];			\
189 	char **varp = &(var);			\
190 						\
191 	len = sizeof buf;			\
192 	if (sysctl(mib, sizeof mib / sizeof mib[0],	\
193 	   &buf, &len, NULL, 0) == -1)		\
194 		err(1, "sysctl");
195 
196 #define	NATIVE_SYSCTLNAME_GET(var,name)		\
197 static void					\
198 native_##var(void)				\
199 {						\
200 	size_t len;				\
201 	static char buf[1024];			\
202 	char **varp = &(var);			\
203 						\
204 	len = sizeof buf;			\
205 	if (sysctlbyname(name, &buf, &len, NULL,\
206 	    0) == -1)				\
207 		err(1, "sysctlbyname");
208 
209 #define	NATIVE_SET				\
210 	*varp = buf;				\
211 	return;					\
212 }	struct __hack
213 
214 #define	NATIVE_BUFFER	(buf)
215 #define	NATIVE_LENGTH	(len)
216 
217 NATIVE_SYSCTL2_GET(sysname, CTL_KERN, KERN_OSTYPE) {
218 } NATIVE_SET;
219 
220 NATIVE_SYSCTL2_GET(hostname, CTL_KERN, KERN_HOSTNAME) {
221 } NATIVE_SET;
222 
223 NATIVE_SYSCTL2_GET(release, CTL_KERN, KERN_OSRELEASE) {
224 } NATIVE_SET;
225 
226 NATIVE_SYSCTL2_GET(version, CTL_KERN, KERN_VERSION) {
227 	size_t n;
228 	char *p;
229 
230 	p = NATIVE_BUFFER;
231 	n = NATIVE_LENGTH;
232 	for (; n--; ++p)
233 		if (*p == '\n' || *p == '\t')
234 			*p = ' ';
235 } NATIVE_SET;
236 
237 NATIVE_SYSCTL2_GET(platform, CTL_HW, HW_MACHINE) {
238 } NATIVE_SET;
239 
240 NATIVE_SYSCTL2_GET(arch, CTL_HW, HW_MACHINE_ARCH) {
241 } NATIVE_SET;
242 
243 NATIVE_SYSCTLNAME_GET(ident, "kern.ident") {
244 } NATIVE_SET;
245 
246 static void
247 usage(void)
248 {
249 	fprintf(stderr, "usage: uname [-aimnoprsv]\n");
250 	exit(1);
251 }
252