githubmirror/procps

src/uptime.c
180 строк5 KB
Craig Small
local: Standardise error functions
17 мар 2025, 10:41
17 мар 2025, 10:415e694f5
/*
 * uptime.c - display system uptime
 *
 * Copyright © 2002-2024 Craig Small <csmall@dropbear.xyz>
 * Copyright © 2020-2024 Jim Warner <james.warner@comcast.net>
 * Copyright © 2011-2012 Sami Kerola <kerolasa@iki.fi>
 *
 * Based on old public domain uptime/whattime by:
 * Larry Greenfield <greenfie@gauss.rutgers.edu>
 * Michael K. Johnson <johnsonm@sunsite.unc.edu>
 * J. Cowley
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include <errno.h>
#include <getopt.h>
#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <sys/time.h>

#include "c.h"
#include "fileutils.h"
#include "nls.h"

#include "misc.h"

#define UPTIME_LEN 100

static double get_uptime_secs(const int container_mode)
{
    double uptime_secs=0;

    if (container_mode) {
	if (procps_container_uptime(&uptime_secs) < 0)
		err(EXIT_FAILURE, _("Cannot get container uptime"));
    } else {
	if (procps_uptime(&uptime_secs, NULL) < 0)
		err(EXIT_FAILURE, _("Cannot get system uptime"));
    }
    return uptime_secs;
}

static void print_uptime_since(const int container_mode)
{
    double now, uptime_secs;
    time_t up_since_secs;
    struct tm *up_since;
    struct timeval tim;

    /* Get the current time and convert it to a double */
	if (gettimeofday(&tim, NULL) != 0)
        err(EXIT_FAILURE, "gettimeofday");
    now = (tim.tv_sec * 1000000.0) + tim.tv_usec;

    /* Get the uptime and calculate when that was */
    uptime_secs = get_uptime_secs(container_mode);
    up_since_secs = (time_t) ((now/1000000.0) - uptime_secs);

    /* Show this */
	if ((up_since = localtime(&up_since_secs)) == NULL)
		errx(EXIT_FAILURE, "localtime");
    printf("%04d-%02d-%02d %02d:%02d:%02d\n",
        up_since->tm_year + 1900, up_since->tm_mon + 1, up_since->tm_mday,
        up_since->tm_hour, up_since->tm_min, up_since->tm_sec);
}

/*
 * Print the standard fields but in raw format in a single line
 */
static void print_uptime_raw()
{
    time_t realseconds;
    double uptime_secs;
    double av1, av5, av15;
    int users=0;

    if ((realseconds = time(NULL)) < 0)
        errx(EXIT_FAILURE, "time");
    if (procps_uptime(&uptime_secs, NULL) < 0)
        errx(EXIT_FAILURE, "procps_uptime_secs");
    if ((users = procps_users()) < 0)
        errx(EXIT_FAILURE, "procps_users");
    if (procps_loadavg(&av1, &av5, &av15) < 0)
        errx(EXIT_FAILURE, "procps_loadavg");

    printf("%lld %f %d %.2f %.2f %.2f\n",
            (long long)realseconds, uptime_secs, users, av1, av5, av15);
}
static void __attribute__ ((__noreturn__)) usage(FILE * out)
{
    fputs(USAGE_HEADER, out);
    fprintf(out, _(" %s [options]\n"), program_invocation_short_name);
    fputs(USAGE_OPTIONS, out);
    fputs(_(" -c, --container show container uptime\n"),out);
    fputs(_(" -p, --pretty   show uptime in pretty format\n"), out);
    fputs(_(" -r, --raw      show uptime values in raw format\n"), out);
    fputs(_(" -s, --since    system up since\n"), out);
    fputs(USAGE_SEPARATOR, out);
    fputs(USAGE_HELP, out);
    fputs(USAGE_VERSION, out);
    fprintf(out, USAGE_MAN_TAIL("uptime(1)"));

    exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
}

int main(int argc, char **argv)
{
    int c, len, p = 0;
    int container_mode = 0;
    char uptime_str[UPTIME_LEN];
    double uptime_secs;

    static const struct option longopts[] = {
        {"container", no_argument, NULL, 'c'},
        {"pretty", no_argument, NULL, 'p'},
        {"help", no_argument, NULL, 'h'},
        {"raw", no_argument, NULL, 'r'},
        {"since", no_argument, NULL, 's'},
        {"version", no_argument, NULL, 'V'},
        {NULL, 0, NULL, 0}
    };

#ifdef HAVE_PROGRAM_INVOCATION_NAME
    program_invocation_name = program_invocation_short_name;
#endif
    setlocale (LC_ALL, "");
    bindtextdomain(PACKAGE, LOCALEDIR);
    textdomain(PACKAGE);
    atexit(close_stdout);

    if ( getenv("PROCPS_CONTAINER") != NULL)
        container_mode = 1;
    while ((c = getopt_long(argc, argv, "cphrsV", longopts, NULL)) != -1)
        switch (c) {
        case 'c':
            container_mode = 1;
            break;
        case 'p':
            p = 1;
            break;
        case 'h':
            usage(stdout);
        case 'r':
            print_uptime_raw();
            return EXIT_SUCCESS;
        case 's':
            print_uptime_since(container_mode);
            return EXIT_SUCCESS;
        case 'V':
            printf(PROCPS_NG_VERSION);
            return EXIT_SUCCESS;
        default:
            usage(stderr);
        }

    if (optind != argc)
        usage(stderr);

    uptime_secs = get_uptime_secs(container_mode);
    len = procps_uptime_snprint( uptime_str, UPTIME_LEN, uptime_secs, p);
    if (len <= 0 || len == UPTIME_LEN)
       err(EXIT_FAILURE, _("Cannot get system uptime"));

    printf("%s\n", uptime_str);
    return EXIT_SUCCESS;
}