* Copyright (c) 1987, 1988, 1993

/* $NetBSD: time.c,v 1.24 2020/04/23 07:54:53 simonb Exp $ */

/*
* Copyright (c) 1987, 1988, 1993
* The Regents of the University of California. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/

#include <sys/cdefs.h>
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1987, 1988, 1993\
The Regents of the University of California. All rights reserved.");
#endif /* not lint */

#ifndef lint
#if 0
static char sccsid[] = "@(#)time.c 8.1 (Berkeley) 6/6/93";
#endif
__RCSID("$NetBSD: time.c,v 1.24 2020/04/23 07:54:53 simonb Exp $");
#endif /* not lint */

#include <sys/types.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/wait.h>
#include <errno.h>
#include <err.h>
#include <locale.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "ext.h"

__dead static void usage(void);
static void prl(long, const char *);
static void prts(const char *, const char *, const struct timespec *,
const char *);
static void prtv(const char *, const char *, const struct timeval *,
const char *);

int
main(int argc, char ** volatile argv)
{
int pid;
int ch, status, prec;
int volatile portableflag;
int volatile lflag;
const char *decpt;
const char *fmt;
const struct lconv *lconv;
struct timespec before, after;
struct rusage ru;

(void)setlocale(LC_ALL, "");

lflag = portableflag = 0;
prec = 1;
fmt = NULL;
while ((ch = getopt(argc, argv, "cf:lpt")) != -1) {
switch (ch) {
case 'f':
fmt = optarg;
portableflag = 0;
lflag = 0;
break;
case 'c': /* csh format */
fmt = "%Uu %Ss %E %P %X+%Dk %I+%Oio %Fpf+%Ww";
portableflag = 0;
lflag = 0;
break;
case 'p': /* POSIX.2 format */
portableflag = 1;
fmt = NULL;
lflag = 0;
break;
case 'l':
lflag = 1;
portableflag = 0;
fmt = NULL;
break;
case 't': /* tcsh format */
fmt = "%Uu %Ss %E %P\t%X+%Dk %I+%Oio %Fpf+%Ww";
prec = 3;
portableflag = 0;
lflag = 0;
break;
case '?':
default:
usage();
}
}
argc -= optind;
argv += optind;

if (argc < 1)
usage();

(void)clock_gettime(CLOCK_MONOTONIC, &before);
switch(pid = vfork()) {
case -1: /* error */
err(EXIT_FAILURE, "Vfork failed");
/* NOTREACHED */
case 0: /* child */
/* LINTED will return only on failure */
execvp(*argv, argv);
err((errno == ENOENT) ? 127 : 126, "Can't exec `%s'", *argv);
/* NOTREACHED */
}

/* parent */
(void)signal(SIGINT, SIG_IGN);
(void)signal(SIGQUIT, SIG_IGN);
if ((pid = wait4(pid, &status, 0, &ru)) == -1)
err(EXIT_FAILURE, "wait4 %d failed", pid);
(void)clock_gettime(CLOCK_MONOTONIC, &after);
if (!WIFEXITED(status))
warnx("Command terminated abnormally.");
timespecsub(&after, &before, &after);

if ((lconv = localeconv()) == NULL ||
(decpt = lconv->decimal_point) == NULL)
decpt = ".";

if (fmt) {
static struct rusage null_ru;
before.tv_sec = 0;
before.tv_nsec = 0;
prusage1(stderr, fmt, prec, &null_ru, &ru, &after, &before);
} else if (portableflag) {
prts("real ", decpt, &after, "\n");
prtv("user ", decpt, &ru.ru_utime, "\n");
prtv("sys ", decpt, &ru.ru_stime, "\n");
} else {
prts("", decpt, &after, " real ");
prtv("", decpt, &ru.ru_utime, " user ");
prtv("", decpt, &ru.ru_stime, " sys\n");
}

if (lflag) {
int hz = (int)sysconf(_SC_CLK_TCK);
unsigned long long ticks;
#define SCALE(x) (long)(ticks ? x / ticks : 0)

ticks = hz * (ru.ru_utime.tv_sec + ru.ru_stime.tv_sec) +
hz * (ru.ru_utime.tv_usec + ru.ru_stime.tv_usec) / 1000000;
prl(ru.ru_maxrss, "maximum resident set size");
prl(SCALE(ru.ru_ixrss), "average shared memory size");
prl(SCALE(ru.ru_idrss), "average unshared data size");
prl(SCALE(ru.ru_isrss), "average unshared stack size");
prl(ru.ru_minflt, "page reclaims");
prl(ru.ru_majflt, "page faults");
prl(ru.ru_nswap, "swaps");
prl(ru.ru_inblock, "block input operations");
prl(ru.ru_oublock, "block output operations");
prl(ru.ru_msgsnd, "messages sent");
prl(ru.ru_msgrcv, "messages received");
prl(ru.ru_nsignals, "signals received");
prl(ru.ru_nvcsw, "voluntary context switches");
prl(ru.ru_nivcsw, "involuntary context switches");
}

return (WIFEXITED(status) ? WEXITSTATUS(status) : EXIT_FAILURE);
}

static void
usage(void)
{

(void)fprintf(stderr,
"Usage: %s [-clpt] [-f <fmt>] utility [argument ...]\n",
getprogname());
exit(EXIT_FAILURE);
}

static void
prl(long val, const char *expn)
{

(void)fprintf(stderr, "%10ld %s\n", val, expn);
}

static void
prts(const char *pre, const char *decpt, const struct timespec *ts,
const char *post)
{

(void)fprintf(stderr, "%s%9lld%s%02ld%s", pre, (long long)ts->tv_sec,
decpt, (long)ts->tv_nsec / 10000000, post);
}

static void
prtv(const char *pre, const char *decpt, const struct timeval *tv,
const char *post)
{

(void)fprintf(stderr, "%s%9lld%s%02ld%s", pre, (long long)tv->tv_sec,
decpt, (long)tv->tv_usec / 10000, post);
}