text: /*
   * CDDL HEADER START
   *
   * The contents of this file are subject to the terms of the
   * Common Development and Distribution License (the "License").
   * You may not use this file except in compliance with the License.
   *
   * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   * or http://www.opensolaris.org/os/licensing.
   * See the License for the specific language governing permissions
   * and limitations under the License.
   *
   * When distributing Covered Code, include this CDDL HEADER in each
   * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
   * If applicable, add the following below this CDDL HEADER, with the
   * fields enclosed by brackets "[]" replaced with your own identifying
   * information: Portions Copyright [yyyy] [name of copyright owner]
   *
   * CDDL HEADER END
   */
  /*
   * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
   * Use is subject to license terms.
   */
  
  /*
   * Copyright (c) 2018, Joyent, Inc.
   */
  
  #include "statcommon.h"
  
  #include <sys/types.h>
  #include <stdlib.h>
  #include <ctype.h>
  #include <stdio.h>
  #include <bsm/audit.h>
  #include <bsm/libbsm.h>
  #include <unistd.h>
  #include <locale.h>
  
  #if !defined(TEXT_DOMAIN)		/* Should be defined by cc -D */
  #define	TEXT_DOMAIN "SYS_TEST"		/* Use this only if it isn't */
  #endif
  
  
  /*
   * Display header every HEADER_MOD lines printed
   */
  #define		DFLT_HEADER_MOD (20)
  #define		ONEK (1024)
  
  #define		CFLG (0x01)
  #define		HFLG (0x02)
  #define		IFLG (0x04)
  #define		NFLG (0x08)
  #define		VFLG (0x10)
  
  extern char	*optarg;
  
  static int	count;
  static int	flags;
  static int	header_mod = DFLT_HEADER_MOD;
  static int	interval;
  static uint_t timestamp_fmt = NODATE;
  
  static void	display_stats();
  static void	eauditon();
  static void	parse_args();
  static void	usage_exit();
  static int	strisdigit();
  
  int
  main(argc, argv)
  int	argc;
  char	**argv;
  {
  	register int	i;
  	au_stat_t s;
  
  	(void) setlocale(LC_ALL, "");
  	(void) textdomain(TEXT_DOMAIN);
  
  	(void) setbuf(stdout, (char *)0);
  	(void) setbuf(stderr, (char *)0);
  
  	parse_args(argc, argv);
  
  	if (!flags) {
  		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
  		if (timestamp_fmt != NODATE)
  			print_timestamp(timestamp_fmt);
  		display_stats(&s, 0);
  		exit(0);
  	}
  
  	if (flags & VFLG || flags & NFLG)
  		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
  
  	if (flags & VFLG)
  		(void) printf("version = %d\n", s.as_version);
  
  	if (flags & NFLG)
  		(void) printf("number of kernel events = %d\n", s.as_numevent);
  
  	if (!(flags & IFLG))
  		exit(0);
  
  	/* CSTYLED */
  	for (i = 0;; i++) {
  		eauditon(A_GETSTAT, (caddr_t)&s, NULL);
  		if (timestamp_fmt != NODATE)
  			print_timestamp(timestamp_fmt);
  		display_stats(&s, i);
  		if ((flags & CFLG) && count)
  			if (i == count - 1)
  				break;
  		(void) sleep(interval);
  	}
  
  	return (0);
  }
  
  
  static void
  display_stats(au_stat_t *s, int cnt)
  {
  	int	offset[12];   /* used to line the header up correctly */
  	char	buf[512];
  
  	(void) sprintf(buf,
  "%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u %n%4u%n",
  		s->as_generated, 	&(offset[0]),
  		s->as_nonattrib, 	&(offset[1]),
  		s->as_kernel, 		&(offset[2]),
  		s->as_audit, 		&(offset[3]),
  		s->as_auditctl, 	&(offset[4]),
  		s->as_enqueue, 		&(offset[5]),
  		s->as_written, 		&(offset[6]),
  		s->as_wblocked, 	&(offset[7]),
  		s->as_rblocked, 	&(offset[8]),
  		s->as_dropped, 		&(offset[9]),
  		s->as_totalsize / ONEK,	&(offset[10]),
  		s->as_memused / ONEK, 	&(offset[11]));
  
  	/* print a properly aligned header every HEADER_MOD lines */
  	if (header_mod && (!cnt || ((timestamp_fmt != NODATE) ?
  	    !(cnt % (header_mod / 2)) : !(cnt % header_mod)))) {
  		(void) printf(
  			"%*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s %*s\n",
  			offset[0] - 1,			"gen",
  			offset[1] - offset[0] - 1,	"nona",
  			offset[2] - offset[1] - 1,	"kern",
  			offset[3] - offset[2] - 1,	"aud",
  			offset[4] - offset[3] - 1,	"ctl",
  			offset[5] - offset[4] - 1,	"enq",
  			offset[6] - offset[5] - 1,	"wrtn",
  			offset[7] - offset[6] - 1,	"wblk",
  			offset[8] - offset[7] - 1,	"rblk",
  			offset[9] - offset[8] - 1,	"drop",
  			offset[10] - offset[9] - 1,	"tot",
  			offset[11] - offset[10],	"mem");
  	}
  
  	(void) puts(buf);
  }
  
  
  static void
  eauditon(cmd, data, length)
  int	cmd;
  caddr_t data;
  int	length;
  {
  	if (auditon(cmd, data, length) == -1) {
  		perror("auditstat: auditon");
  		exit(1);
  	}
  }
  
  
  static void
  parse_args(argc, argv)
  int	argc;
  char	**argv;
  {
  	int	c;
  
  	while ((c = getopt(argc, argv, "c:h:i:vnT:")) != -1) {
  		switch (c) {
  		case 'c':
  			if (flags & CFLG)
  				usage_exit();
  			flags |= CFLG;
  			if (strisdigit(optarg)) {
  				(void) fprintf(stderr,
  				"auditstat: invalid count specified.\n");
  				exit(1);
  			}
  			count = atoi(optarg);
  			break;
  		case 'h':
  			if (flags & HFLG)
  				usage_exit();
  			flags |= HFLG;
  			if (strisdigit(optarg)) {
  				(void) fprintf(stderr,
  				"auditstat: invalid header arg specified.\n");
  				exit(1);
  			}
  			header_mod = atoi(optarg);
  			break;
  		case 'i':
  			if (flags & IFLG)
  				usage_exit();
  			flags |= IFLG;
  			if (strisdigit(optarg)) {
  				(void) fprintf(stderr,
  				"auditstat: invalid interval specified.\n");
  				exit(1);
  			}
  			interval = atoi(optarg);
  			break;
  		case 'n':
  			if (flags & NFLG)
  				usage_exit();
  			flags |= NFLG;
  			break;
  		case 'v':
  			if (flags & VFLG)
  				usage_exit();
  			flags |= VFLG;
  			break;
  		case 'T':
  			if (optarg) {
  				if (*optarg == 'u')
  					timestamp_fmt = UDATE;
  				else if (*optarg == 'd')
  					timestamp_fmt = DDATE;
  				else
  					usage_exit();
  			} else {
  				usage_exit();
  			}
  			break;
  		case '?':
  		default:
  			usage_exit();
  			break;
  		}
  	}
  }
  
  
  static void
  usage_exit()
  {
  	(void) fprintf(stderr,
  	    "auditstat: usage: auditstat [-c count] [-h lines] "
  	    "[-T d|u] [-i interval] [-n] [-v]\n");
  	exit(1);
  }
  
  
  static int
  strisdigit(s)
  char	*s;
  {
  	for (; *s; s++)
  		if (!isdigit(*s))
  			return (1);
  
  	return (0);
  }
  