1/*	$OpenBSD: time.c,v 1.26 2023/10/15 18:20:25 cheloha Exp $	*/
2/*	$NetBSD: time.c,v 1.7 1995/06/27 00:34:00 jtc Exp $	*/
3
4/*
5 * Copyright (c) 1987, 1988, 1993
6 *	The Regents of the University of California.  All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of the University nor the names of its contributors
17 *    may be used to endorse or promote products derived from this software
18 *    without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33#include <sys/time.h>
34#include <sys/resource.h>
35#include <sys/wait.h>
36#include <sys/sysctl.h>
37
38#include <err.h>
39#include <errno.h>
40#include <signal.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <unistd.h>
44
45__dead void usage(void);
46
47int
48main(int argc, char *argv[])
49{
50	struct rusage ru;
51	struct timespec before, after, during;
52	int ch, exitonsig, lflag, portableflag, status;
53	pid_t pid;
54
55	if (pledge("stdio proc exec", NULL) == -1)
56		err(1, "pledge");
57
58	exitonsig = lflag = portableflag = 0;
59
60	while ((ch = getopt(argc, argv, "lp")) != -1) {
61		switch(ch) {
62		case 'l':
63			lflag = 1;
64			break;
65		case 'p':
66			portableflag = 1;
67			break;
68		default:
69			usage();
70		}
71	}
72	argc -= optind;
73	argv += optind;
74
75	if (argc < 1)
76		usage();
77
78	clock_gettime(CLOCK_MONOTONIC, &before);
79	switch (pid = vfork()) {
80	case -1:			/* error */
81		warn("fork");
82		return 1;
83	case 0:				/* child */
84		execvp(*argv, argv);
85		warn("%s", *argv);
86		_exit(errno == ENOENT ? 127 : 126);
87	}
88
89	/* parent */
90	signal(SIGINT, SIG_IGN);
91	signal(SIGQUIT, SIG_IGN);
92	while (wait3(&status, 0, &ru) != pid)
93		continue;
94	clock_gettime(CLOCK_MONOTONIC, &after);
95	if (WIFSIGNALED(status))
96		exitonsig = WTERMSIG(status);
97	if (!WIFEXITED(status))
98		fprintf(stderr, "Command terminated abnormally.\n");
99	timespecsub(&after, &before, &during);
100
101	if (portableflag) {
102		fprintf(stderr, "real %9lld.%02ld\n",
103		    (long long)during.tv_sec, during.tv_nsec / 10000000);
104		fprintf(stderr, "user %9lld.%02ld\n",
105		    (long long)ru.ru_utime.tv_sec, ru.ru_utime.tv_usec / 10000);
106		fprintf(stderr, "sys  %9lld.%02ld\n",
107		    (long long)ru.ru_stime.tv_sec, ru.ru_stime.tv_usec / 10000);
108	} else {
109		fprintf(stderr, "%9lld.%02ld real ",
110		    (long long)during.tv_sec, during.tv_nsec / 10000000);
111		fprintf(stderr, "%9lld.%02ld user ",
112		    (long long)ru.ru_utime.tv_sec, ru.ru_utime.tv_usec / 10000);
113		fprintf(stderr, "%9lld.%02ld sys\n",
114		    (long long)ru.ru_stime.tv_sec, ru.ru_stime.tv_usec / 10000);
115	}
116
117	if (lflag) {
118		int hz;
119		long ticks;
120		int mib[2];
121		struct clockinfo clkinfo;
122		size_t size;
123
124		mib[0] = CTL_KERN;
125		mib[1] = KERN_CLOCKRATE;
126		size = sizeof(clkinfo);
127		if (sysctl(mib, 2, &clkinfo, &size, NULL, 0) < 0)
128			err(1, "sysctl");
129
130		hz = clkinfo.hz;
131		ticks = hz * (ru.ru_utime.tv_sec + ru.ru_stime.tv_sec) +
132		     hz * (ru.ru_utime.tv_usec + ru.ru_stime.tv_usec) / 1000000;
133
134		fprintf(stderr, "%10ld  %s\n",
135		    ru.ru_maxrss, "maximum resident set size");
136		fprintf(stderr, "%10ld  %s\n", ticks ? ru.ru_ixrss / ticks : 0,
137		    "average shared memory size");
138		fprintf(stderr, "%10ld  %s\n", ticks ? ru.ru_idrss / ticks : 0,
139		    "average unshared data size");
140		fprintf(stderr, "%10ld  %s\n", ticks ? ru.ru_isrss / ticks : 0,
141		    "average unshared stack size");
142		fprintf(stderr, "%10ld  %s\n",
143		    ru.ru_minflt, "minor page faults");
144		fprintf(stderr, "%10ld  %s\n",
145		    ru.ru_majflt, "major page faults");
146		fprintf(stderr, "%10ld  %s\n",
147		    ru.ru_nswap, "swaps");
148		fprintf(stderr, "%10ld  %s\n",
149		    ru.ru_inblock, "block input operations");
150		fprintf(stderr, "%10ld  %s\n",
151		    ru.ru_oublock, "block output operations");
152		fprintf(stderr, "%10ld  %s\n",
153		    ru.ru_msgsnd, "messages sent");
154		fprintf(stderr, "%10ld  %s\n",
155		    ru.ru_msgrcv, "messages received");
156		fprintf(stderr, "%10ld  %s\n",
157		    ru.ru_nsignals, "signals received");
158		fprintf(stderr, "%10ld  %s\n",
159		    ru.ru_nvcsw, "voluntary context switches");
160		fprintf(stderr, "%10ld  %s\n",
161		    ru.ru_nivcsw, "involuntary context switches");
162	}
163
164	if (exitonsig) {
165		if (signal(exitonsig, SIG_DFL) == SIG_ERR)
166			return 128 + exitonsig;
167		else
168			raise(exitonsig);
169	}
170	return WIFEXITED(status) ? WEXITSTATUS(status) : EXIT_FAILURE;
171}
172
173__dead void
174usage(void)
175{
176	fprintf(stderr, "usage: %s [-lp] utility [argument ...]\n",
177	    getprogname());
178	exit(1);
179}
180