xref: /titanic_51/usr/src/cmd/auditstat/auditstat.c (revision 67dbe2be0c0f1e2eb428b89088bb5667e8f0b9f6)
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 2009 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 #include "statcommon.h"
27 
28 #include <sys/types.h>
29 #include <stdlib.h>
30 #include <ctype.h>
31 #include <stdio.h>
32 #include <bsm/audit.h>
33 #include <bsm/libbsm.h>
34 #include <unistd.h>
35 #include <locale.h>
36 
37 #if !defined(TEXT_DOMAIN)		/* Should be defined by cc -D */
38 #define	TEXT_DOMAIN "SYS_TEST"		/* Use this only if it isn't */
39 #endif
40 
41 
42 /*
43  * Display header every HEADER_MOD lines printed
44  */
45 #define		DFLT_HEADER_MOD (20)
46 #define		ONEK (1024)
47 
48 #define		CFLG (0x01)
49 #define		HFLG (0x02)
50 #define		IFLG (0x04)
51 #define		NFLG (0x08)
52 #define		VFLG (0x10)
53 
54 extern char	*optarg;
55 
56 static int	count;
57 static int	flags;
58 static int	header_mod = DFLT_HEADER_MOD;
59 static int	interval;
60 static uint_t timestamp_fmt = NODATE;
61 
62 static void	display_stats();
63 static void	eauditon();
64 static void	parse_args();
65 static void	usage_exit();
66 static int	strisdigit();
67 
68 int
69 main(argc, argv)
70 int	argc;
71 char	**argv;
72 {
73 	register int	i;
74 	au_stat_t s;
75 
76 	(void) setlocale(LC_ALL, "");
77 	(void) textdomain(TEXT_DOMAIN);
78 
79 	(void) setbuf(stdout, (char *)0);
80 	(void) setbuf(stderr, (char *)0);
81 
82 	parse_args(argc, argv);
83 
84 	if (!flags) {
85 		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
86 		if (timestamp_fmt != NODATE)
87 			print_timestamp(timestamp_fmt);
88 		display_stats(&s, 0);
89 		exit(0);
90 	}
91 
92 	if (flags & VFLG || flags & NFLG)
93 		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
94 
95 	if (flags & VFLG)
96 		(void) printf("version = %d\n", s.as_version);
97 
98 	if (flags & NFLG)
99 		(void) printf("number of kernel events = %d\n", s.as_numevent);
100 
101 	if (!(flags & IFLG))
102 		exit(0);
103 
104 	/* CSTYLED */
105 	for (i = 0;; i++) {
106 		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
107 		if (timestamp_fmt != NODATE)
108 			print_timestamp(timestamp_fmt);
109 		display_stats(&s, i);
110 		if ((flags & CFLG) && count)
111 			if (i == count - 1)
112 				break;
113 		(void) sleep(interval);
114 	}
115 
116 	return (0);
117 }
118 
119 
120 static void
121 display_stats(s, cnt)
122 au_stat_t *s;
123 {
124 	int	offset[12];   /* used to line the header up correctly */
125 	char	buf[512];
126 
127 	(void) sprintf(buf,
128 "%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u%n",
129 		s->as_generated, 	&(offset[0]),
130 		s->as_nonattrib, 	&(offset[1]),
131 		s->as_kernel, 		&(offset[2]),
132 		s->as_audit, 		&(offset[3]),
133 		s->as_auditctl, 	&(offset[4]),
134 		s->as_enqueue, 		&(offset[5]),
135 		s->as_written, 		&(offset[6]),
136 		s->as_wblocked, 	&(offset[7]),
137 		s->as_rblocked, 	&(offset[8]),
138 		s->as_dropped, 		&(offset[9]),
139 		s->as_totalsize / ONEK,	&(offset[10]),
140 		s->as_memused / ONEK, 	&(offset[11]));
141 
142 	/* print a properly aligned header every HEADER_MOD lines */
143 	if (header_mod && (!cnt || ((timestamp_fmt != NODATE) ?
144 	    !(cnt % (header_mod / 2)) : !(cnt % header_mod)))) {
145 		(void) printf(
146 			"%*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s\n",
147 			offset[0] - 1,			"gen",
148 			offset[1] - offset[0] - 1,	"nona",
149 			offset[2] - offset[1] - 1,	"kern",
150 			offset[3] - offset[2] - 1,	"aud",
151 			offset[4] - offset[3] - 1,	"ctl",
152 			offset[5] - offset[4] - 1,	"enq",
153 			offset[6] - offset[5] - 1,	"wrtn",
154 			offset[7] - offset[6] - 1,	"wblk",
155 			offset[8] - offset[7] - 1,	"rblk",
156 			offset[9] - offset[8] - 1,	"drop",
157 			offset[10] - offset[9] - 1,	"tot",
158 			offset[11] - offset[10],	"mem");
159 	}
160 
161 	(void) puts(buf);
162 }
163 
164 
165 static void
166 eauditon(cmd, data, length)
167 int	cmd;
168 caddr_t data;
169 int	length;
170 {
171 	if (auditon(cmd, data, length) == -1) {
172 		perror("auditstat: auditon");
173 		exit(1);
174 	}
175 }
176 
177 
178 static void
179 parse_args(argc, argv)
180 int	argc;
181 char	**argv;
182 {
183 	int	c;
184 
185 	while ((c = getopt(argc, argv, "c:h:i:vnT:")) != -1) {
186 		switch (c) {
187 		case 'c':
188 			if (flags & CFLG)
189 				usage_exit();
190 			flags |= CFLG;
191 			if (strisdigit(optarg)) {
192 				(void) fprintf(stderr,
193 				"auditstat: invalid count specified.\n");
194 				exit(1);
195 			}
196 			count = atoi(optarg);
197 			break;
198 		case 'h':
199 			if (flags & HFLG)
200 				usage_exit();
201 			flags |= HFLG;
202 			if (strisdigit(optarg)) {
203 				(void) fprintf(stderr,
204 				"auditstat: invalid header arg specified.\n");
205 				exit(1);
206 			}
207 			header_mod = atoi(optarg);
208 			break;
209 		case 'i':
210 			if (flags & IFLG)
211 				usage_exit();
212 			flags |= IFLG;
213 			if (strisdigit(optarg)) {
214 				(void) fprintf(stderr,
215 				"auditstat: invalid interval specified.\n");
216 				exit(1);
217 			}
218 			interval = atoi(optarg);
219 			break;
220 		case 'n':
221 			if (flags & NFLG)
222 				usage_exit();
223 			flags |= NFLG;
224 			break;
225 		case 'v':
226 			if (flags & VFLG)
227 				usage_exit();
228 			flags |= VFLG;
229 			break;
230 		case 'T':
231 			if (optarg) {
232 				if (*optarg == 'u')
233 					timestamp_fmt = UDATE;
234 				else if (*optarg == 'd')
235 					timestamp_fmt = DDATE;
236 				else
237 					usage_exit();
238 			} else {
239 				usage_exit();
240 			}
241 			break;
242 		case '?':
243 		default:
244 			usage_exit();
245 			break;
246 		}
247 	}
248 }
249 
250 
251 static void
252 usage_exit()
253 {
254 	(void) fprintf(stderr,
255 	    "auditstat: usage: auditstat [-c count] [-h lines] "
256 	    "[-T d|u] [-i interval] [-n] [-v]\n");
257 	exit(1);
258 }
259 
260 
261 static int
262 strisdigit(s)
263 char	*s;
264 {
265 	for (; *s; s++)
266 		if (!isdigit(*s))
267 			return (1);
268 
269 	return (0);
270 }
271