xref: /illumos-gate/usr/src/cmd/auditreduce/main.c (revision 46b592853d0f4f11781b6b0a7533f267c6aee132)
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 /*
23  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 /*
28  * The Secure SunOS audit reduction tool - auditreduce.
29  * Document SM0071 is the primary source of information on auditreduce.
30  *
31  * Composed of 4 source modules:
32  * main.c - main driver.
33  * option.c - command line option processing.
34  * process.c - record/file/process functions.
35  * time.c - date/time handling.
36  *
37  * Main(), write_header(), audit_stats(), and a_calloc()
38  * are the only functions visible outside this module.
39  */
40 
41 #include <siginfo.h>
42 #include <locale.h>
43 #include <libintl.h>
44 #include "auditr.h"
45 #include "auditrd.h"
46 
47 #if !defined(TEXT_DOMAIN)
48 #define	TEXT_DOMAIN "SUNW_OST_OSCMD"
49 #endif
50 
51 extern void	derive_str(time_t, char *);
52 extern int	process_options(int, char **);
53 extern int	mproc(audit_pcb_t *);
54 extern void	init_tokens(void);	/* shared with praudit */
55 
56 static int	a_pow(int, int);
57 static void	calc_procs(void);
58 static void	chld_handler(int);
59 static int	close_outfile(void);
60 static void	c_close(audit_pcb_t *, int);
61 static void	delete_infiles(void);
62 static void	gather_pcb(audit_pcb_t *, int, int);
63 static void	init_options(void);
64 static int	init_sig(void);
65 static void	int_handler(int);
66 static int	mfork(audit_pcb_t *, int, int, int);
67 static void	mcount(int, int);
68 static int	open_outfile(void);
69 static void	p_close(audit_pcb_t *);
70 static int	rename_outfile(void);
71 static void	rm_mem(audit_pcb_t *);
72 static void	rm_outfile(void);
73 static void	trim_mem(audit_pcb_t *);
74 static int	write_file_token(time_t);
75 static int	write_trailer(void);
76 
77 /*
78  * File globals.
79  */
80 static int	max_sproc;	/* maximum number of subprocesses per process */
81 static int	total_procs;	/* number of processes in the process tree */
82 static int	total_layers;	/* number of layers in the process tree */
83 
84 /*
85  * .func main - main.
86  * .desc The beginning. Main() calls each of the initialization routines
87  *	and then allocates the root pcb. Then it calls mfork() to get
88  *	the work done.
89  * .call	main(argc, argv).
90  * .arg	argc	- number of arguments.
91  * .arg	argv	- array of pointers to arguments.
92  * .ret	0	- via exit() - no errors detected.
93  * .ret	1	- via exit() - errors detected (messages printed).
94  */
95 int
96 main(int argc, char **argv)
97 {
98 	int	ret;
99 	audit_pcb_t *pcb;
100 
101 	/* Internationalization */
102 	(void) setlocale(LC_ALL, "");
103 	(void) textdomain(TEXT_DOMAIN);
104 
105 	root_pid = getpid();	/* know who is root process for error */
106 	init_options();		/* initialize options */
107 	init_tokens();		/* initialize token processing table */
108 	if (init_sig())		/* initialize signals */
109 		exit(1);
110 	if (process_options(argc, argv))
111 		exit(1);	/* process command line options */
112 	if (open_outfile())	/* setup root process output stream */
113 		exit(1);
114 	calc_procs();		/* see how many subprocesses we need */
115 	/*
116 	 * Allocate the root pcb and set it up.
117 	 */
118 	pcb = (audit_pcb_t *)a_calloc(1, sizeof (audit_pcb_t));
119 	pcb->pcb_procno = root_pid;
120 	pcb->pcb_flags |= PF_ROOT;
121 	pcb->pcb_fpw = stdout;
122 	pcb->pcb_time = -1;
123 	/*
124 	 * Now start the whole thing rolling.
125 	 */
126 	if (mfork(pcb, pcbnum, 0, pcbnum - 1)) {
127 		/*
128 		 * Error in processing somewhere. A message is already printed.
129 		 * Display usage statistics and remove the outfile.
130 		 */
131 		if (getpid() == root_pid) {
132 			audit_stats();
133 			(void) close_outfile();
134 			rm_outfile();
135 		}
136 		exit(1);
137 	}
138 	/*
139 	 * Clean up afterwards.
140 	 * Only do outfile cleanup if we are root process.
141 	 */
142 	if (getpid() == root_pid) {
143 		if ((ret = write_trailer()) == 0) { /* write trailer to file */
144 
145 			ret = close_outfile();	/* close the outfile */
146 		}
147 		/*
148 		 * If there was an error in cleanup then remove outfile.
149 		 */
150 		if (ret) {
151 			rm_outfile();
152 			exit(1);
153 		}
154 		/*
155 		 * And lastly delete the infiles if the user so wishes.
156 		 */
157 		if (f_delete)
158 			delete_infiles();
159 	}
160 	return (0);
161 /*NOTREACHED*/
162 }
163 
164 
165 /*
166  * .func mfork - main fork routine.
167  * .desc Create a (sub-)tree of processses if needed, or just do the work
168  *	if we have few enough groups to process. This is a recursive routine
169  *	which stops recursing when the number of files to process is small
170  *	enough. Each call to mfork() is responsible for a range of pcbs
171  *	from audit_pcbs[]. This range is designated by the lo and hi
172  *	arguments (inclusive). If the number of pcbs is small enough
173  *	then we have hit a leaf of the tree and mproc() is called to
174  *	do the processing. Otherwise we fork some processes and break
175  *	the range of pcbs up amongst them.
176  * .call	ret = mfork(pcb, nsp, lo, hi).
177  * .arg	pcb	- ptr to pcb that is root node of the to-be-created tree.
178  * .arg	nsp	- number of sub-processes this tree must process.
179  * .arg	lo	- lower-limit of process number range. Index into audit_pcbs.
180  * .arg	hi	- higher limit of pcb range. Index into audit_pcbs.
181  * .ret	0	- succesful completion.
182  * .ret	-1	- error encountered in processing - message already printed.
183  */
184 static int
185 mfork(audit_pcb_t *pcb, int nsp, int lo, int hi)
186 {
187 	int	range, procno, i, tofork, nnsp, nrem;
188 	int	fildes[2];
189 	audit_pcb_t *pcbn;
190 
191 #if AUDIT_PROC_TRACE
192 	(void) fprintf(stderr, "mfork: nsp %d %d->%d\n", nsp, lo, hi);
193 #endif
194 
195 	/*
196 	 * The range of pcb's to process is small enough now. Do the work.
197 	 */
198 	if (nsp <= max_sproc) {
199 		pcb->pcb_flags |= PF_LEAF;	/* leaf in process tree */
200 		pcb->pcb_below = audit_pcbs;	/* proc pcbs from audit_pcbs */
201 		gather_pcb(pcb, lo, hi);
202 		trim_mem(pcb);			/* trim allocated memory */
203 		return (mproc(pcb));		/* do the work */
204 	}
205 	/*
206 	 * Too many pcb's for one process - must fork.
207 	 * Try to balance the tree as it grows and make it short and fat.
208 	 * The thing to minimize is the number of times a record passes
209 	 * through a pipe.
210 	 */
211 	else {
212 		/*
213 		 * Fork less than the maximum number of processes.
214 		 */
215 		if (nsp <= max_sproc * (max_sproc - 1)) {
216 			tofork = nsp / max_sproc;
217 			if (nsp % max_sproc)
218 				tofork++;	/* how many to fork */
219 		}
220 		/*
221 		 * Fork the maximum number of processes.
222 		 */
223 		else {
224 			tofork = max_sproc;	/* how many to fork */
225 		}
226 		/*
227 		 * Allocate the nodes below us in the process tree.
228 		 */
229 		pcb->pcb_below = (audit_pcb_t *)
230 			a_calloc(tofork, sizeof (*pcb));
231 		nnsp = nsp / tofork;	/* # of pcbs per forked process */
232 		nrem = nsp % tofork;	/* remainder to spread around */
233 		/*
234 		 * Loop to fork all of the subs. Open a pipe for each.
235 		 * If there are any errors in pipes, forks, or getting streams
236 		 * for the pipes then quit altogether.
237 		 */
238 		for (i = 0; i < tofork; i++) {
239 			pcbn = &pcb->pcb_below[i];
240 			pcbn->pcb_time = -1;
241 			if (pipe(fildes)) {
242 				perror(gettext(
243 					"auditreduce: couldn't get a pipe"));
244 				return (-1);
245 			}
246 			/*
247 			 * Convert descriptors to streams.
248 			 */
249 			if ((pcbn->pcb_fpr = fdopen(fildes[0], "r")) == NULL) {
250 	perror(gettext("auditreduce: couldn't get read stream for pipe"));
251 				return (-1);
252 			}
253 			if ((pcbn->pcb_fpw = fdopen(fildes[1], "w")) == NULL) {
254 	perror(gettext("auditreduce: couldn't get write stream for pipe"));
255 				return (-1);
256 			}
257 			if ((procno = fork()) == -1) {
258 				perror(gettext("auditreduce: fork failed"));
259 				return (-1);
260 			}
261 			/*
262 			 * Calculate the range of pcbs from audit_pcbs [] this
263 			 * branch of the tree will be responsible for.
264 			 */
265 			range = (nrem > 0) ? nnsp + 1 : nnsp;
266 			/*
267 			 * Child route.
268 			 */
269 			if (procno == 0) {
270 				pcbn->pcb_procno = getpid();
271 				c_close(pcb, i); /* close unused streams */
272 				/*
273 				 * Continue resolving this branch.
274 				 */
275 				return (mfork(pcbn, range, lo, lo + range - 1));
276 			}
277 			/* Parent route. */
278 			else {
279 				pcbn->pcb_procno = i;
280 				/* allocate buffer to hold record */
281 				pcbn->pcb_rec = (char *)a_calloc(1,
282 				    AUDITBUFSIZE);
283 				pcbn->pcb_size = AUDITBUFSIZE;
284 				p_close(pcbn);	/* close unused streams */
285 
286 				nrem--;
287 				lo += range;
288 			}
289 		}
290 		/*
291 		 * Done forking all of the subs.
292 		 */
293 		gather_pcb(pcb, 0, tofork - 1);
294 		trim_mem(pcb);			/* free unused memory */
295 		return (mproc(pcb));
296 	}
297 }
298 
299 
300 /*
301  * .func	trim_mem - trim memory usage.
302  * .desc	Free un-needed allocated memory.
303  * .call	trim_mem(pcb).
304  * .arg	pcb	- ptr to pcb for current process.
305  * .ret	void.
306  */
307 static void
308 trim_mem(audit_pcb_t *pcb)
309 {
310 	int	count;
311 	size_t	size;
312 
313 	/*
314 	 * For the root don't free anything. We need to save audit_pcbs[]
315 	 * in case we are deleting the infiles at the end.
316 	 */
317 	if (pcb->pcb_flags & PF_ROOT)
318 		return;
319 	/*
320 	 * For a leaf save its part of audit_pcbs[] and then remove it all.
321 	 */
322 	if (pcb->pcb_flags & PF_LEAF) {
323 		count = pcb->pcb_count;
324 		size = sizeof (audit_pcb_t);
325 		/* allocate a new buffer to hold the pcbs */
326 		pcb->pcb_below = (audit_pcb_t *)a_calloc(count, size);
327 		/* save this pcb's portion */
328 		(void) memcpy((void *) pcb->pcb_below,
329 		    (void *) &audit_pcbs[pcb->pcb_lo], count * size);
330 		rm_mem(pcb);
331 		gather_pcb(pcb, 0, count - 1);
332 	}
333 		/*
334 		 * If this is an intermediate node then just remove it all.
335 		 */
336 	else {
337 		rm_mem(pcb);
338 	}
339 }
340 
341 
342 /*
343  * .func	rm_mem - remove memory.
344  * .desc	Remove unused memory associated with audit_pcbs[]. For each
345  *	pcb in audit_pcbs[] free the record buffer and all of
346  *	the fcbs. Then free audit_pcbs[].
347  * .call	rm_mem(pcbr).
348  * .arg	pcbr	- ptr to pcb of current process.
349  * .ret	void.
350  */
351 static void
352 rm_mem(audit_pcb_t *pcbr)
353 {
354 	int	i;
355 	audit_pcb_t *pcb;
356 	audit_fcb_t *fcb, *fcbn;
357 
358 	for (i = 0; i < pcbsize; i++) {
359 		/*
360 		 * Don't free the record buffer and fcbs for the pcbs this
361 		 * process is using.
362 		 */
363 		if (pcbr->pcb_flags & PF_LEAF) {
364 			if (pcbr->pcb_lo <= i || i <= pcbr->pcb_hi)
365 				continue;
366 		}
367 		pcb = &audit_pcbs[i];
368 		free(pcb->pcb_rec);
369 		for (fcb = pcb->pcb_first; fcb != NULL; /* */) {
370 			fcbn = fcb->fcb_next;
371 			free((char *)fcb);
372 			fcb = fcbn;
373 		}
374 	}
375 	free((char *)audit_pcbs);
376 }
377 
378 
379 /*
380  * .func	c_close - close unused streams.
381  * .desc	This is called for each child process just after being born.
382  *	The child closes the read stream for the pipe to its parent.
383  *	It also closes the read streams for the other children that
384  *	have been born before it. If any closes fail a warning message
385  *	is printed, but processing continues.
386  * .call	ret = c_close(pcb, i).
387  * .arg	pcb	- ptr to the child's parent pcb.
388  * .arg	i	- iteration # of child in forking loop.
389  * .ret	void.
390  */
391 static void
392 c_close(audit_pcb_t *pcb, int	i)
393 {
394 	int	j;
395 	audit_pcb_t *pcbt;
396 
397 	/*
398 	 * Do all pcbs in parent's group up to and including us
399 	 */
400 	for (j = 0; j <= i; j++) {
401 		pcbt = &pcb->pcb_below[j];
402 		if (fclose(pcbt->pcb_fpr) == EOF) {
403 			if (!f_quiet)
404 		perror(gettext("auditreduce: initial close on pipe failed"));
405 		}
406 		/*
407 		 * Free the buffer allocated to hold incoming records.
408 		 */
409 		if (i != j) {
410 			free(pcbt->pcb_rec);
411 		}
412 	}
413 }
414 
415 
416 /*
417  * .func	p_close - close unused streams for parent.
418  * .desc	Called by the parent right after forking a child.
419  *	Closes the write stream on the pipe to the child since
420  *	we will never use it.
421  * .call	p_close(pcbn),
422  * .arg	pcbn	- ptr to pcb.
423  * .ret	void.
424  */
425 static void
426 p_close(audit_pcb_t *pcbn)
427 {
428 	if (fclose(pcbn->pcb_fpw) == EOF) {
429 		if (!f_quiet)
430 		perror(gettext("auditreduce: close for write pipe failed"));
431 	}
432 }
433 
434 
435 /*
436  * .func	audit_stats - print statistics.
437  * .desc	Print usage statistics for the user if the run fails.
438  *	Tells them how many files they had and how many groups this
439  *	totalled. Also tell them how many layers and processes the
440  *	process tree had.
441  * .call	audit_stats().
442  * .arg	none.
443  * .ret	void.
444  */
445 void
446 audit_stats(void)
447 {
448 	struct rlimit rl;
449 
450 	if (getrlimit(RLIMIT_NOFILE, &rl) != -1)
451 		(void) fprintf(stderr,
452 		    gettext("%s The system allows %d files per process.\n"),
453 		    ar, rl.rlim_cur);
454 	(void) fprintf(stderr, gettext(
455 "%s There were %d file(s) %d file group(s) %d process(es) %d layer(s).\n"),
456 		ar, filenum, pcbnum, total_procs, total_layers);
457 }
458 
459 
460 /*
461  * .func gather_pcb - gather pcbs.
462  * .desc Gather together the range of the sub-processes that we are
463  *	responsible for. For a pcb that controls processes this is all
464  *	of the sub-processes that it forks. For a pcb that controls
465  *	files this is the the range of pcbs from audit_pcbs[].
466  * .call gather_pcb(pcb, lo, hi).
467  * .arg	pcb	- ptr to pcb.
468  * .arg	lo	- lo index into pcb_below.
469  * .arg	hi	- hi index into pcb_below.
470  * .ret	void.
471  */
472 static void
473 gather_pcb(audit_pcb_t *pcb, int lo, int hi)
474 {
475 	pcb->pcb_lo = lo;
476 	pcb->pcb_hi = hi;
477 	pcb->pcb_count = hi - lo + 1;
478 }
479 
480 
481 /*
482  * .func calc_procs - calculate process parameters.
483  * .desc Calculate the current run's paramters regarding how many
484  *	processes will have to be forked (maybe none).
485  *	5 is subtracted from maxfiles_proc to allow for stdin, stdout,
486  *	stderr, and the pipe to a parent process. The outfile
487  *	in the root process is assigned to stdout. The unused half of each
488  *	pipe is closed, to allow for more connections, but we still
489  *	have to have the 5th spot because in order to get the pipe
490  *	we need 2 descriptors up front.
491  * .call calc_procs().
492  * .arg	none.
493  * .ret	void.
494  */
495 static void
496 calc_procs(void)
497 {
498 	int	val;
499 	int	maxfiles_proc;
500 	struct rlimit rl;
501 
502 	if (getrlimit(RLIMIT_NOFILE, &rl) == -1) {
503 		perror("auditreduce: getrlimit");
504 		exit(1);
505 	}
506 
507 	maxfiles_proc = rl.rlim_cur;
508 
509 	max_sproc = maxfiles_proc - 5;	/* max subprocesses per process */
510 
511 	/*
512 	 * Calculate how many layers the process tree has.
513 	 */
514 	total_layers = 1;
515 	for (/* */; /* */; /* */) {
516 		val = a_pow(max_sproc, total_layers);
517 		if (val > pcbnum)
518 			break;
519 		total_layers++;
520 	}
521 	/*
522 	 * Count how many processes are in the process tree.
523 	 */
524 	mcount(pcbnum, 0);
525 
526 #if AUDIT_PROC_TRACE
527 	(void) fprintf(stderr,
528 	    "pcbnum %d filenum %d mfp %d msp %d ly %d tot %d\n\n",
529 	    pcbnum, filenum, maxfiles_proc, max_sproc,
530 	    total_layers, total_procs);
531 #endif
532 }
533 
534 
535 static int
536 a_pow(int base, int exp)
537 {
538 	int	i;
539 	int	answer;
540 
541 	if (exp == 0) {
542 		answer = 1;
543 	} else {
544 		answer = base;
545 		for (i = 0; i < (exp - 1); i++)
546 			answer *= base;
547 	}
548 	return (answer);
549 }
550 
551 
552 /*
553  * .func mcount - main count.
554  * .desc Go through the motions of building the process tree just
555  *	to count how many processes there are. Don't really
556  *	build anything. Answer is in global var total_procs.
557  * .call mcount(nsp, lo).
558  * .arg	nsp	- number of subs for this tree branch.
559  * .arg	lo	- lo side of range of subs.
560  * .ret	void.
561  */
562 static void
563 mcount(int nsp, int lo)
564 {
565 	int	range, i, tofork, nnsp, nrem;
566 
567 	total_procs++;		/* count another process created */
568 
569 	if (nsp > max_sproc) {
570 		if (nsp <= max_sproc * (max_sproc - 1)) {
571 			tofork = nsp / max_sproc;
572 			if (nsp % max_sproc)
573 				tofork++;
574 		} else {
575 			tofork = max_sproc;
576 		}
577 		nnsp = nsp / tofork;
578 		nrem = nsp % tofork;
579 		for (i = 0; i < tofork; i++) {
580 			range = (nrem > 0) ? nnsp + 1 : nnsp;
581 			mcount(range, lo);
582 			nrem--;
583 			lo += range;
584 		}
585 	}
586 }
587 
588 
589 /*
590  * .func delete_infiles - delete the input files.
591  * .desc If the user asked us to (via 'D' flag) then unlink the input files.
592  * .call ret = delete_infiles().
593  * .arg none.
594  * .ret void.
595  */
596 static void
597 delete_infiles(void)
598 {
599 	int	i;
600 	audit_pcb_t *pcb;
601 	audit_fcb_t *fcb;
602 
603 	for (i = 0; i < pcbsize; i++) {
604 		pcb = &audit_pcbs[i];
605 		fcb = pcb->pcb_dfirst;
606 		while (fcb != NULL) {
607 			/*
608 			 * Only delete a file if it was succesfully processed.
609 			 * If there were any read errors or bad records
610 			 * then don't delete it.
611 			 * There may still be unprocessed records in it.
612 			 */
613 			if (fcb->fcb_flags & FF_DELETE) {
614 				if (unlink(fcb->fcb_file)) {
615 					if (f_verbose) {
616 						(void) sprintf(errbuf, gettext(
617 						"%s delete on %s failed"),
618 						ar, fcb->fcb_file);
619 					}
620 					perror(errbuf);
621 				}
622 			}
623 			fcb = fcb->fcb_next;
624 		}
625 	}
626 }
627 
628 
629 /*
630  * .func rm_outfile - remove the outfile.
631  * .desc Remove the file we are writing the records to. We do this if
632  *	processing failed and we are quitting before finishing.
633  *	Update - don't actually remove the outfile, but generate
634  *	a warning about its possible heathen nature.
635  * .call ret = rm_outfile().
636  * .arg	none.
637  * .ret	void.
638  */
639 static void
640 rm_outfile(void)
641 {
642 #if 0
643 	if (f_outfile) {
644 		if (unlink(f_outtemp) == -1) {
645 			(void) sprintf(errbuf,
646 				gettext("%s delete on %s failed"),
647 				ar, f_outtemp);
648 			perror(errbuf);
649 		}
650 	}
651 #else
652 	(void) fprintf(stderr,
653 gettext("%s Warning: Incomplete audit file may have been generated - %s\n"),
654 		ar,
655 		(f_outfile == NULL) ? gettext("standard output") : f_outfile);
656 #endif
657 }
658 
659 
660 /*
661  * .func	close_outfile - close the outfile.
662  * .desc	Close the file we are writing records to.
663  * .call	ret = close_outfile().
664  * .arg	none.
665  * .ret	0	- close was succesful.
666  * .ret	-1	- close failed.
667  */
668 static int
669 close_outfile(void)
670 {
671 	if (fclose(stdout) == EOF) {
672 		(void) sprintf(errbuf, gettext("%s close on %s failed"),
673 		    ar, f_outfile ? f_outfile : "standard output");
674 		perror(errbuf);
675 		return (-1);
676 	}
677 	(void) fsync(fileno(stdout));
678 	return (rename_outfile());
679 }
680 
681 
682 /*
683  * .func write_header - write audit file header.
684  * .desc Write an audit file header to the output stream. The time in the
685  *	header is the time of the first record written to the stream. This
686  *	routine is called by the process handling the root node of the
687  *	process tree just before it writes the first record to the output
688  *	stream.
689  * .ret	0 - succesful write.
690  * .ret -1 - failed write - message printed.
691  */
692 int
693 write_header(void)
694 {
695 	return (write_file_token(f_start));
696 }
697 
698 
699 static int
700 write_file_token(time_t when)
701 {
702 	adr_t adr;			/* adr ptr */
703 	struct timeval tv;		/* time now */
704 	char	for_adr[16];		/* plenty of room */
705 #ifdef _LP64
706 	char	token_id = AUT_OTHER_FILE64;
707 #else
708 	char	token_id = AUT_OTHER_FILE32;
709 #endif
710 	short	i = 1;
711 	char	c = '\0';
712 
713 	tv.tv_sec = when;
714 	tv.tv_usec = 0;
715 	adr_start(&adr, for_adr);
716 	adr_char(&adr, &token_id, 1);
717 #ifdef _LP64
718 	adr_int64(&adr, (int64_t *)&tv, 2);
719 #else
720 	adr_int32(&adr, (int32_t *)&tv, 2);
721 #endif
722 	adr_short(&adr, &i, 1);
723 	adr_char(&adr, &c, 1);
724 
725 	if (fwrite(for_adr, sizeof (char), adr_count(&adr), stdout) !=
726 	    adr_count(&adr)) {
727 		if (when == f_start) {
728 			(void) sprintf(errbuf,
729 				gettext("%s error writing header to %s. "),
730 				ar,
731 				f_outfile ? f_outfile :
732 					gettext("standard output"));
733 		} else {
734 			(void) sprintf(errbuf,
735 				gettext("%s error writing trailer to %s. "),
736 				ar,
737 				f_outfile ? f_outfile :
738 					gettext("standard output"));
739 		}
740 		perror(errbuf);
741 		return (-1);
742 	}
743 	return (0);
744 }
745 
746 
747 /*
748  * .func  write_trailer - write audit file trailer.
749  * .desc  Write an audit file trailer to the output stream. The finish
750  *	time for the trailer is the time of the last record written
751  *	to the stream.
752  * .ret	0 - succesful write.
753  * .ret	-1 - failed write - message printed.
754  */
755 static int
756 write_trailer(void)
757 {
758 	return (write_file_token(f_end));
759 }
760 
761 
762 /*
763  * .func rename_outfile - rename the outfile.
764  * .desc If the user used the -O flag they only gave us the suffix name
765  *	for the outfile. We have to add the time stamps to put the filename
766  *	in the proper audit file name format. The start time will be the time
767  *	of the first record in the file and the end time will be the time of
768  *	the last record in the file.
769  * .ret	0 - rename succesful.
770  * .ret	-1 - rename failed - message printed.
771  */
772 static int
773 rename_outfile(void)
774 {
775 	char	f_newfile[MAXFILELEN];
776 	char	buf1[15], buf2[15];
777 	char	*f_file, *f_nfile, *f_time, *f_name;
778 
779 	if (f_outfile != NULL) {
780 		/*
781 		 * Get string representations of start and end times.
782 		 */
783 		derive_str(f_start, buf1);
784 		derive_str(f_end, buf2);
785 
786 		f_nfile = f_time = f_newfile;	/* working copy */
787 		f_file = f_name = f_outfile;	/* their version */
788 		while (*f_file) {
789 			if (*f_file == '/') {	/* look for filename */
790 				f_time = f_nfile + 1;
791 				f_name = f_file + 1;
792 			}
793 			*f_nfile++ = *f_file++;	/* make copy of their version */
794 		}
795 		*f_time = '\0';
796 		/* start time goes first */
797 		(void) strcat(f_newfile, buf1);
798 		(void) strcat(f_newfile, ".");
799 		/* then the finish time */
800 		(void) strcat(f_newfile, buf2);
801 		(void) strcat(f_newfile, ".");
802 		/* and the name they gave us */
803 		(void) strcat(f_newfile, f_name);
804 
805 #if AUDIT_FILE
806 		(void) fprintf(stderr, "rename_outfile: <%s> --> <%s>\n",
807 			f_outfile, f_newfile);
808 #endif
809 
810 #if AUDIT_RENAME
811 		if (rename(f_outtemp, f_newfile) == -1) {
812 			(void) fprintf(stderr,
813 			    "%s rename of %s to %s failed.\n",
814 			    ar, f_outtemp, f_newfile);
815 			return (-1);
816 		}
817 		f_outfile = f_newfile;
818 #else
819 		if (rename(f_outtemp, f_outfile) == -1) {
820 			(void) fprintf(stderr,
821 			    gettext("%s rename of %s to %s failed.\n"),
822 			    ar, f_outtemp, f_outfile);
823 			return (-1);
824 		}
825 #endif
826 	}
827 	return (0);
828 }
829 
830 
831 /*
832  * .func open_outfile - open the outfile.
833  * .desc Open the outfile specified by the -O option. Assign it to the
834  *	the standard output. Get a unique temporary name to use so we
835  *	don't clobber an existing file.
836  * .ret	0 - no errors detected.
837  * .ret	-1 - errors in processing (message already printed).
838  */
839 static int
840 open_outfile(void)
841 {
842 	int	tmpfd = -1;
843 
844 	if (f_outfile != NULL) {
845 		f_outtemp = (char *)a_calloc(1, strlen(f_outfile) + 8);
846 		(void) strcpy(f_outtemp, f_outfile);
847 		(void) strcat(f_outtemp, "XXXXXX");
848 		if ((tmpfd = mkstemp(f_outtemp)) == -1) {
849 			(void) sprintf(errbuf,
850 			    gettext("%s couldn't create temporary file"), ar);
851 			perror(errbuf);
852 			return (-1);
853 		}
854 		(void) fflush(stdout);
855 		if (tmpfd != fileno(stdout)) {
856 			if ((dup2(tmpfd, fileno(stdout))) == -1) {
857 				(void) sprintf(errbuf,
858 				    gettext("%s can't assign %s to the "
859 				    "standard output"), ar, f_outfile);
860 				perror(errbuf);
861 				return (-1);
862 			}
863 			(void) close(tmpfd);
864 		}
865 	}
866 	return (0);
867 }
868 
869 
870 /*
871  * .func init_options - initialize the options.
872  * .desc Give initial and/or default values to some options.
873  * .call init_options();
874  * .arg	none.
875  * .ret	void.
876  */
877 static void
878 init_options(void)
879 {
880 	struct timeval tp;
881 	struct timezone tpz;
882 
883 	/*
884 	 * Get current time for general use.
885 	 */
886 	if (gettimeofday(&tp, &tpz) == -1)
887 		perror(gettext("auditreduce: initial getttimeofday failed"));
888 
889 	time_now = tp.tv_sec;		/* save for general use */
890 	f_start = 0;			/* first record time default */
891 	f_end = time_now;		/* last record time default */
892 	m_after = 0;			/* Jan 1, 1970 00:00:00 */
893 
894 	/*
895 	 * Setup initial size of audit_pcbs[].
896 	 */
897 	pcbsize = PCB_INITSIZE;		/* initial size of file-holding pcb's */
898 
899 	audit_pcbs = (audit_pcb_t *)a_calloc(pcbsize, sizeof (audit_pcb_t));
900 
901 	/* description of 'current' error */
902 	error_str = gettext("initial error");
903 
904 }
905 
906 
907 /*
908  * .func a_calloc - audit calloc.
909  * .desc Calloc with check for failure. This is called by all of the
910  *	places that want memory.
911  * .call ptr = a_calloc(nelem, size).
912  * .arg	nelem - number of elements to allocate.
913  * .arg	size - size of each element.
914  * .ret	ptr - ptr to allocated and zeroed memory.
915  * .ret	never - if calloc fails then we never return.
916  */
917 void	*
918 a_calloc(int nelem, size_t size)
919 {
920 	void	*ptr;
921 
922 	if ((ptr = calloc((unsigned)nelem, size)) == NULL) {
923 		perror(gettext("auditreduce: memory allocation failed"));
924 		exit(1);
925 	}
926 	return (ptr);
927 }
928 
929 
930 /*
931  * .func init_sig - initial signal catching.
932  *
933  * .desc
934  *	Setup the signal catcher to catch the SIGCHLD signal plus
935  *	"environmental" signals -- keyboard plus other externally
936  *	generated signals such as out of file space or cpu time.  If a
937  *	child exits with either a non-zero exit code or was killed by
938  *	a signal to it then we will also exit with a non-zero exit
939  *	code. In this way abnormal conditions can be passed up to the
940  *	root process and the entire run be halted. Also catch the int
941  *	and quit signals. Remove the output file since it is in an
942  *	inconsistent state.
943  * .call ret = init_sig().
944  * .arg none.
945  * .ret 0 - no errors detected.
946  * .ret -1 - signal failed (message printed).
947  */
948 static int
949 init_sig(void)
950 {
951 	if (signal(SIGCHLD, chld_handler) == SIG_ERR) {
952 		perror(gettext("auditreduce: SIGCHLD signal failed"));
953 		return (-1);
954 	}
955 
956 	if (signal(SIGHUP, int_handler) == SIG_ERR) {
957 		perror(gettext("auditreduce: SIGHUP signal failed"));
958 		return (-1);
959 	}
960 	if (signal(SIGINT, int_handler) == SIG_ERR) {
961 		perror(gettext("auditreduce: SIGINT signal failed"));
962 		return (-1);
963 	}
964 	if (signal(SIGQUIT, int_handler) == SIG_ERR) {
965 		perror(gettext("auditreduce: SIGQUIT signal failed"));
966 		return (-1);
967 	}
968 	if (signal(SIGABRT, int_handler) == SIG_ERR) {
969 		perror(gettext("auditreduce: SIGABRT signal failed"));
970 		return (-1);
971 	}
972 	if (signal(SIGTERM, int_handler) == SIG_ERR) {
973 		perror(gettext("auditreduce: SIGTERM signal failed"));
974 		return (-1);
975 	}
976 	if (signal(SIGPWR, int_handler) == SIG_ERR) {
977 		perror(gettext("auditreduce: SIGPWR signal failed"));
978 		return (-1);
979 	}
980 	if (signal(SIGXCPU, int_handler) == SIG_ERR) {
981 		perror(gettext("auditreduce: SIGXCPU signal failed"));
982 		return (-1);
983 	}
984 	if (signal(SIGXFSZ, int_handler) == SIG_ERR) {
985 		perror(gettext("auditreduce: SIGXFSZ signal failed"));
986 		return (-1);
987 	}
988 	if (signal(SIGSEGV, int_handler) == SIG_ERR) {
989 		perror(gettext("auditreduce: SIGSEGV signal failed"));
990 		return (-1);
991 	}
992 
993 	return (0);
994 }
995 
996 
997 /*
998  * .func chld_handler - handle child signals.
999  * .desc Catch the SIGCHLD signals. Remove the root process
1000  *	output file because it is in an inconsistent state.
1001  *	Print a message giving the signal number and/or return code
1002  *	of the child who caused the signal.
1003  * .ret	void.
1004  */
1005 /* ARGSUSED */
1006 void
1007 chld_handler(int sig)
1008 {
1009 	int	pid;
1010 	int	status;
1011 
1012 	/*
1013 	 * Get pid and reasons for cause of event.
1014 	 */
1015 	pid = wait(&status);
1016 
1017 	if (pid > 0) {
1018 		/*
1019 		 * If child received a signal or exited with a non-zero
1020 		 * exit status then print message and exit
1021 		 */
1022 		if ((WHIBYTE(status) == 0 && WLOBYTE(status) != 0) ||
1023 		    (WHIBYTE(status) != 0 && WLOBYTE(status) == 0)) {
1024 			(void) fprintf(stderr,
1025 			    gettext("%s abnormal child termination - "), ar);
1026 
1027 			if (WHIBYTE(status) == 0 && WLOBYTE(status) != 0) {
1028 				psignal(WLOBYTE(status), "signal");
1029 				if (WCOREDUMP(status))
1030 					(void) fprintf(stderr,
1031 					    gettext("core dumped\n"));
1032 			}
1033 
1034 			if (WHIBYTE(status) != 0 && WLOBYTE(status) == 0)
1035 				(void) fprintf(stderr, gettext(
1036 					"return code %d\n"),
1037 					WHIBYTE(status));
1038 
1039 			/*
1040 			 * Get rid of outfile - it is suspect.
1041 			 */
1042 			if (f_outfile != NULL) {
1043 				(void) close_outfile();
1044 				rm_outfile();
1045 			}
1046 			/*
1047 			 * Give statistical info that may be useful.
1048 			 */
1049 			audit_stats();
1050 
1051 			exit(1);
1052 		}
1053 	}
1054 }
1055 
1056 
1057 /*
1058  * .func	int_handler - handle quit/int signals.
1059  * .desc	Catch the keyboard and other environmental signals.
1060  *		Remove the root process output file because it is in
1061  *		an inconsistent state.
1062  * .ret	void.
1063  */
1064 /* ARGSUSED */
1065 void
1066 int_handler(int sig)
1067 {
1068 	if (getpid() == root_pid) {
1069 		(void) close_outfile();
1070 		rm_outfile();
1071 		exit(1);
1072 	}
1073 	/*
1074 	 * For a child process don't give an error exit or the
1075 	 * parent process will catch it with the chld_handler and
1076 	 * try to erase the outfile again.
1077 	 */
1078 	exit(0);
1079 }
1080