xref: /illumos-gate/usr/src/cmd/isainfo/isainfo.c (revision 8d94f651a44d41a7147253bb5dad1a53941e8f50)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*
27  * Copyright (c) 2012, Joyent, Inc. All rights reserved.
28  */
29 
30 #include <sys/types.h>
31 #include <sys/systeminfo.h>
32 #include <sys/utsname.h>
33 #include <sys/stat.h>
34 
35 #include <sys/auxv.h>
36 #include <sys/cpuid_drv.h>
37 #include <sys/elf.h>
38 
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <strings.h>
42 #include <unistd.h>
43 #include <errno.h>
44 #include <libintl.h>
45 #include <locale.h>
46 #include <fcntl.h>
47 
48 #include <elfcap.h>
49 
50 static const char dev_cpu_self_cpuid[] = "/dev/" CPUID_SELF_NAME;
51 static char *pgmname;
52 static int mode = 0;
53 
54 #define	BITS_MODE	0x1
55 #define	NATIVE_MODE	0x2
56 #define	KERN_MODE	0x4
57 #define	VERBOSE_MODE	0x8
58 #define	EXTN_MODE	0x10
59 
60 static char *
61 getsysinfo(int cmd)
62 {
63 	char *buf;
64 	size_t bufsize = 20;	/* wild guess */
65 	long ret;
66 
67 	if ((buf = malloc(bufsize)) == NULL)
68 		return (NULL);
69 	do {
70 		ret = sysinfo(cmd, buf, bufsize);
71 		if (ret == -1)
72 			return (NULL);
73 		if (ret > bufsize) {
74 			bufsize = ret;
75 			buf = realloc(buf, bufsize);
76 		} else
77 			break;
78 	} while (buf != NULL);
79 
80 	return (buf);
81 }
82 
83 /*
84  * Classify isa's as to bitness of the corresponding ABIs.
85  * isa's which have no "official" Solaris ABI are returned
86  * unrecognised i.e. "zero bit".
87  */
88 static uint_t
89 bitness(const char *isaname)
90 {
91 	if (strcmp(isaname, "sparc") == 0 ||
92 	    strcmp(isaname, "i386") == 0)
93 		return (32);
94 
95 	if (strcmp(isaname, "sparcv9") == 0 ||
96 	    strcmp(isaname, "amd64") == 0)
97 		return (64);
98 
99 	return (0);
100 }
101 
102 static char *
103 report_abi(int cmd, const char *vfmt)
104 {
105 	uint_t bits;
106 	char *isa;
107 
108 	if ((isa = getsysinfo(cmd)) == NULL)
109 		return (0);
110 	if ((bits = bitness(isa)) == 0) {
111 		(void) fprintf(stderr,
112 		    gettext("%s: unable to identify isa '%s'!\n"),
113 		    pgmname, isa);
114 		exit(3);
115 	}
116 
117 	if (mode & VERBOSE_MODE)
118 		(void) printf(vfmt, bits, isa);
119 	else if (mode & BITS_MODE)
120 		(void) printf("%d\n", bits);
121 	else if (mode & (NATIVE_MODE|KERN_MODE))
122 		(void) printf("%s\n", isa);
123 	else
124 		(void) printf("%s", isa);
125 	return (isa);
126 }
127 
128 /*
129  * Classify isas as their machine type.
130  */
131 static ushort_t
132 machtype(const char *isaname)
133 {
134 	if (strcmp(isaname, "sparc") == 0)
135 		return (EM_SPARC);
136 	if (strcmp(isaname, "sparcv9") == 0)
137 		return (EM_SPARCV9);
138 	if (strcmp(isaname, "i386") == 0)
139 		return (EM_386);
140 	if (strcmp(isaname, "amd64") == 0)
141 		return (EM_AMD64);
142 
143 	return (0);
144 }
145 
146 static void
147 report_hwcap(int d, const char *isa)
148 {
149 	struct cpuid_get_hwcap __cgh, *cgh = &__cgh;
150 	char buffer[1024], cap2[1024];
151 
152 	cgh->cgh_archname = (char *)isa;
153 	if (ioctl(d, CPUID_GET_HWCAP, cgh) != 0)
154 		return;
155 
156 	(void) elfcap_hw1_to_str(ELFCAP_STYLE_LC, cgh->cgh_hwcap[0],
157 	    buffer, sizeof (buffer), ELFCAP_FMT_SNGSPACE, machtype(isa));
158 
159 	if (cgh->cgh_hwcap[1] != 0)
160 		(void) elfcap_hw2_to_str(ELFCAP_STYLE_LC, cgh->cgh_hwcap[1],
161 		    cap2, sizeof (cap2), ELFCAP_FMT_SNGSPACE, machtype(isa));
162 	else
163 		cap2[0] = '\0';
164 
165 	if (mode & EXTN_MODE) {
166 		(void) printf(":");
167 		if (cgh->cgh_hwcap[1] != 0)
168 			(void) printf(" %s", cap2);
169 		(void) printf(" %s", buffer);
170 		(void) printf("\n");
171 	} else {
172 		char *p;
173 		int linecnt = 0;
174 
175 		for (p = strtok(cap2, " "); p; p = strtok(NULL, " ")) {
176 			if (linecnt + strlen(p) > 68) {
177 				(void) printf("\n");
178 				linecnt = 0;
179 			}
180 			if (linecnt == 0)
181 				linecnt = printf("\t");
182 			linecnt += printf("%s ", p);
183 		}
184 
185 		for (p = strtok(buffer, " "); p; p = strtok(NULL, " ")) {
186 			if (linecnt + strlen(p) > 68) {
187 				(void) printf("\n");
188 				linecnt = 0;
189 			}
190 			if (linecnt == 0)
191 				linecnt = printf("\t");
192 			linecnt += printf("%s ", p);
193 		}
194 
195 		if (linecnt != 0)
196 			(void) printf("\n");
197 	}
198 }
199 
200 #if !defined(TEXT_DOMAIN)
201 #define	TEXT_DOMAIN	"SYS_TEST"
202 #endif
203 
204 int
205 main(int argc, char *argv[])
206 {
207 	int errflg = 0;
208 	int c;
209 	char *vfmt;
210 	char *isa, *isa32;
211 	int d = -1;
212 	const int excl_modes =	/* exclusive mode settings */
213 	    NATIVE_MODE | BITS_MODE | KERN_MODE | EXTN_MODE;
214 
215 	(void) setlocale(LC_ALL, "");
216 	(void) textdomain(TEXT_DOMAIN);
217 
218 	if ((pgmname = strrchr(*argv, '/')) == 0)
219 		pgmname = argv[0];
220 	else
221 		pgmname++;
222 
223 	while ((c = getopt(argc, argv, "nbkvx")) != EOF)
224 		switch (c) {
225 		case 'n':
226 			if (mode & excl_modes)
227 				errflg++;
228 			mode |= NATIVE_MODE;
229 			break;
230 		case 'b':
231 			if (mode & excl_modes)
232 				errflg++;
233 			mode |= BITS_MODE;
234 			break;
235 		case 'k':
236 			if (mode & excl_modes)
237 				errflg++;
238 			mode |= KERN_MODE;
239 			break;
240 		case 'x':
241 			if (mode & excl_modes || mode & VERBOSE_MODE)
242 				errflg++;
243 			mode |= EXTN_MODE;
244 			break;
245 		case 'v':
246 			if (mode & EXTN_MODE)
247 				errflg++;
248 			mode |= VERBOSE_MODE;
249 			break;
250 		case '?':
251 		default:
252 			errflg++;
253 			break;
254 		}
255 
256 	if (errflg || optind != argc) {
257 		(void) fprintf(stderr,
258 		    gettext("usage: %s [ [-v] [-b | -n | -k] | [-x] ]\n"),
259 		    pgmname);
260 		return (1);
261 	}
262 
263 	/*
264 	 * We use dev_cpu_self_cpuid for discovering hardware capabilities;
265 	 * but we only complain if we can't open it if we've been
266 	 * asked to report on those capabilities.
267 	 */
268 	if ((mode & (VERBOSE_MODE|EXTN_MODE)) != 0 &&
269 	    (d = open(dev_cpu_self_cpuid, O_RDONLY)) == -1)
270 		perror(dev_cpu_self_cpuid), exit(1);
271 
272 	if (mode & KERN_MODE) {
273 		vfmt = gettext("%d-bit %s kernel modules\n");
274 		(void) report_abi(SI_ARCHITECTURE_K, vfmt);
275 		return (0);
276 	}
277 
278 	vfmt = gettext("%d-bit %s applications\n");
279 
280 	if (mode & (BITS_MODE | NATIVE_MODE)) {
281 		if ((isa = report_abi(SI_ARCHITECTURE_64, vfmt)) == NULL)
282 			isa = report_abi(SI_ARCHITECTURE_32, vfmt);
283 		if (isa != NULL && (mode & VERBOSE_MODE) != 0)
284 			report_hwcap(d, isa);
285 	} else {
286 		if ((isa = report_abi(SI_ARCHITECTURE_64, vfmt)) != NULL) {
287 			if (mode & (EXTN_MODE|VERBOSE_MODE))
288 				report_hwcap(d, isa);
289 			else
290 				(void) putchar(' ');
291 		}
292 
293 		if ((isa32 = report_abi(SI_ARCHITECTURE_32, vfmt)) != NULL) {
294 			if (mode & (EXTN_MODE|VERBOSE_MODE))
295 				report_hwcap(d, isa32);
296 		}
297 
298 		if ((isa32 != NULL || isa != NULL) &&
299 		    (mode & (EXTN_MODE|VERBOSE_MODE)) == 0)
300 			(void) putchar('\n');
301 	}
302 
303 	return (0);
304 }
305