#include <getopt.h>

#include <locale.h>

#include <stdbool.h>

#include <stdlib.h>

#include "sd-bus.h"

#include "bus-error.h"

#include "bus-util.h"

#include "pager.h"

#include "spawn-polkit-agent.h"

#include "strv.h"

#include "terminal-util.h"

#include "util.h"

static bool arg_no_pager = false;

static bool arg_ask_password = true;

static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;

static char *arg_host = NULL;

static bool arg_adjust_system_clock = false;

static void pager_open_if_enabled(void) {

if (arg_no_pager)

return;

pager_open(false);

}

static void polkit_agent_open_if_enabled(void) {

/* Open the polkit agent as a child process if necessary */

if (!arg_ask_password)

return;

if (arg_transport != BUS_TRANSPORT_LOCAL)

return;

polkit_agent_open();

}

typedef struct StatusInfo {

usec_t time;

char *timezone;

usec_t rtc_time;

bool rtc_local;

bool ntp_enabled;

bool ntp_capable;

bool ntp_synced;

} StatusInfo;

static void status_info_clear(StatusInfo *info) {

if (info) {

free(info->timezone);

zero(*info);

}

}

static void print_status_info(const StatusInfo *i) {

char a[FORMAT_TIMESTAMP_MAX];

struct tm tm;

time_t sec;

bool have_time = false;

const char *old_tz = NULL, *tz;

int r;

assert(i);

/* Save the old $TZ */

tz = getenv("TZ");

if (tz)

old_tz = strdupa(tz);

/* Set the new $TZ */

if (setenv("TZ", isempty(i->timezone) ? "UTC" : i->timezone, true) < 0)

log_warning_errno(errno, "Failed to set TZ environment variable, ignoring: %m");

else

tzset();

if (i->time != 0) {

sec = (time_t) (i->time / USEC_PER_SEC);

have_time = true;

} else if (IN_SET(arg_transport, BUS_TRANSPORT_LOCAL, BUS_TRANSPORT_MACHINE)) {

sec = time(NULL);

have_time = true;

} else

log_warning("Could not get time from timedated and not operating locally, ignoring.");

if (have_time) {

xstrftime(a, "%a %Y-%m-%d %H:%M:%S %Z", localtime_r(&sec, &tm));

printf(" Local time: %.*s\n", (int) sizeof(a), a);

xstrftime(a, "%a %Y-%m-%d %H:%M:%S UTC", gmtime_r(&sec, &tm));

printf(" Universal time: %.*s\n", (int) sizeof(a), a);

} else {

printf(" Local time: %s\n", "n/a");

printf(" Universal time: %s\n", "n/a");

}

if (i->rtc_time > 0) {

time_t rtc_sec;

rtc_sec = (time_t) (i->rtc_time / USEC_PER_SEC);

xstrftime(a, "%a %Y-%m-%d %H:%M:%S", gmtime_r(&rtc_sec, &tm));

printf(" RTC time: %.*s\n", (int) sizeof(a), a);

} else

printf(" RTC time: %s\n", "n/a");

if (have_time)

xstrftime(a, "%Z, %z", localtime_r(&sec, &tm));

/* Restore the $TZ */

if (old_tz)

r = setenv("TZ", old_tz, true);

else

r = unsetenv("TZ");

if (r < 0)

log_warning_errno(errno, "Failed to set TZ environment variable, ignoring: %m");

else

tzset();

printf(" Time zone: %s (%.*s)\n"

" Network time on: %s\n"

"NTP synchronized: %s\n"

" RTC in local TZ: %s\n",

strna(i->timezone), (int) sizeof(a), have_time ? a : "n/a",

i->ntp_capable ? yes_no(i->ntp_enabled) : "n/a",

yes_no(i->ntp_synced),

yes_no(i->rtc_local));

if (i->rtc_local)

fputs("\n" ANSI_HIGHLIGHT

"Warning: The system is configured to read the RTC time in the local time zone.\n"

" This mode can not be fully supported. It will create various problems\n"

" with time zone changes and daylight saving time adjustments. The RTC\n"

" time is never updated, it relies on external facilities to maintain it.\n"

" If at all possible, use RTC in UTC by calling\n"

" 'timedatectl set-local-rtc 0'." ANSI_NORMAL "\n", stdout);

}

static int show_status(sd_bus *bus, char **args, unsigned n) {

_cleanup_(status_info_clear) StatusInfo info = {};

static const struct bus_properties_map map[] = {

{ "Timezone", "s", NULL, offsetof(StatusInfo, timezone) },

{ "LocalRTC", "b", NULL, offsetof(StatusInfo, rtc_local) },

{ "NTP", "b", NULL, offsetof(StatusInfo, ntp_enabled) },

{ "CanNTP", "b", NULL, offsetof(StatusInfo, ntp_capable) },

{ "NTPSynchronized", "b", NULL, offsetof(StatusInfo, ntp_synced) },

{ "TimeUSec", "t", NULL, offsetof(StatusInfo, time) },

{ "RTCTimeUSec", "t", NULL, offsetof(StatusInfo, rtc_time) },

{}

};

int r;

assert(bus);

r = bus_map_all_properties(bus,

"org.freedesktop.timedate1",

"/org/freedesktop/timedate1",

map,

&info);

if (r < 0)

return log_error_errno(r, "Failed to query server: %m");

print_status_info(&info);

return r;

}

static int set_time(sd_bus *bus, char **args, unsigned n) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

bool relative = false, interactive = arg_ask_password;

usec_t t;

int r;

assert(args);

assert(n == 2);

polkit_agent_open_if_enabled();

r = parse_timestamp(args[1], &t);

if (r < 0) {

log_error("Failed to parse time specification: %s", args[1]);

return r;

}

r = sd_bus_call_method(bus,

"org.freedesktop.timedate1",

"/org/freedesktop/timedate1",

"org.freedesktop.timedate1",

"SetTime",

&error,

NULL,

"xbb", (int64_t)t, relative, interactive);

if (r < 0)

log_error("Failed to set time: %s", bus_error_message(&error, -r));

return r;

}

static int set_timezone(sd_bus *bus, char **args, unsigned n) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

int r;

assert(args);

assert(n == 2);

polkit_agent_open_if_enabled();

r = sd_bus_call_method(bus,

"org.freedesktop.timedate1",

"/org/freedesktop/timedate1",

"org.freedesktop.timedate1",

"SetTimezone",

&error,

NULL,

"sb", args[1], arg_ask_password);

if (r < 0)

log_error("Failed to set time zone: %s", bus_error_message(&error, -r));

return r;

}

static int set_local_rtc(sd_bus *bus, char **args, unsigned n) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

int r, b;

assert(args);

assert(n == 2);

polkit_agent_open_if_enabled();

b = parse_boolean(args[1]);

if (b < 0) {

log_error("Failed to parse local RTC setting: %s", args[1]);

return b;

}

r = sd_bus_call_method(bus,

"org.freedesktop.timedate1",

"/org/freedesktop/timedate1",

"org.freedesktop.timedate1",

"SetLocalRTC",

&error,

NULL,

"bbb", b, arg_adjust_system_clock, arg_ask_password);

if (r < 0)

log_error("Failed to set local RTC: %s", bus_error_message(&error, -r));

return r;

}

static int set_ntp(sd_bus *bus, char **args, unsigned n) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

int b, r;

assert(args);

assert(n == 2);

polkit_agent_open_if_enabled();

b = parse_boolean(args[1]);

if (b < 0) {

log_error("Failed to parse NTP setting: %s", args[1]);

return b;

}

r = sd_bus_call_method(bus,

"org.freedesktop.timedate1",

"/org/freedesktop/timedate1",

"org.freedesktop.timedate1",

"SetNTP",

&error,

NULL,

"bb", b, arg_ask_password);

if (r < 0)

log_error("Failed to set ntp: %s", bus_error_message(&error, -r));

return r;

}

static int list_timezones(sd_bus *bus, char **args, unsigned n) {

_cleanup_strv_free_ char **zones = NULL;

int r;

assert(args);

assert(n == 1);

r = get_timezones(&zones);

if (r < 0)

return log_error_errno(r, "Failed to read list of time zones: %m");

pager_open_if_enabled();

strv_print(zones);

return 0;

}

static void help(void) {

printf("%s [OPTIONS...] COMMAND ...\n\n"

"Query or change system time and date settings.\n\n"

" -h --help Show this help message\n"

" --version Show package version\n"

" --no-pager Do not pipe output into a pager\n"

" --no-ask-password Do not prompt for password\n"

" -H --host=[USER@]HOST Operate on remote host\n"

" -M --machine=CONTAINER Operate on local container\n"

" --adjust-system-clock Adjust system clock when changing local RTC mode\n\n"

"Commands:\n"

" status Show current time settings\n"

" set-time TIME Set system time\n"

" set-timezone ZONE Set system time zone\n"

" list-timezones Show known time zones\n"

" set-local-rtc BOOL Control whether RTC is in local time\n"

" set-ntp BOOL Enable or disable network time synchronization\n",

program_invocation_short_name);

}

static int parse_argv(int argc, char *argv[]) {

enum {

ARG_VERSION = 0x100,

ARG_NO_PAGER,

ARG_ADJUST_SYSTEM_CLOCK,

ARG_NO_ASK_PASSWORD

};

static const struct option options[] = {

{ "help", no_argument, NULL, 'h' },

{ "version", no_argument, NULL, ARG_VERSION },

{ "no-pager", no_argument, NULL, ARG_NO_PAGER },

{ "host", required_argument, NULL, 'H' },

{ "machine", required_argument, NULL, 'M' },

{ "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD },

{ "adjust-system-clock", no_argument, NULL, ARG_ADJUST_SYSTEM_CLOCK },

{}

};

int c;

assert(argc >= 0);

assert(argv);

while ((c = getopt_long(argc, argv, "hH:M:", options, NULL)) >= 0)

switch (c) {

case 'h':

help();

return 0;

case ARG_VERSION:

return version();

case 'H':

arg_transport = BUS_TRANSPORT_REMOTE;

arg_host = optarg;

break;

case 'M':

arg_transport = BUS_TRANSPORT_MACHINE;

arg_host = optarg;

break;

case ARG_NO_ASK_PASSWORD:

arg_ask_password = false;

break;

case ARG_ADJUST_SYSTEM_CLOCK:

arg_adjust_system_clock = true;

break;

case ARG_NO_PAGER:

arg_no_pager = true;

break;

case '?':

return -EINVAL;

default:

assert_not_reached("Unhandled option");

}

return 1;

}

static int timedatectl_main(sd_bus *bus, int argc, char *argv[]) {

static const struct {

const char* verb;

const enum {

MORE,

LESS,

EQUAL

} argc_cmp;

const int argc;

int (* const dispatch)(sd_bus *bus, char **args, unsigned n);

} verbs[] = {

{ "status", LESS, 1, show_status },

{ "set-time", EQUAL, 2, set_time },

{ "set-timezone", EQUAL, 2, set_timezone },

{ "list-timezones", EQUAL, 1, list_timezones },

{ "set-local-rtc", EQUAL, 2, set_local_rtc },

{ "set-ntp", EQUAL, 2, set_ntp, },

};

int left;

unsigned i;

assert(argc >= 0);

assert(argv);

left = argc - optind;

if (left <= 0)

/* Special rule: no arguments means "status" */

i = 0;

else {

if (streq(argv[optind], "help")) {

help();

return 0;

}

for (i = 0; i < ELEMENTSOF(verbs); i++)

if (streq(argv[optind], verbs[i].verb))

break;

if (i >= ELEMENTSOF(verbs)) {

log_error("Unknown operation %s", argv[optind]);

return -EINVAL;

}

}

switch (verbs[i].argc_cmp) {

case EQUAL:

if (left != verbs[i].argc) {

log_error("Invalid number of arguments.");

return -EINVAL;

}

break;

case MORE:

if (left < verbs[i].argc) {

log_error("Too few arguments.");

return -EINVAL;

}

break;

case LESS:

if (left > verbs[i].argc) {

log_error("Too many arguments.");

return -EINVAL;

}

break;

default:

assert_not_reached("Unknown comparison operator.");

}

return verbs[i].dispatch(bus, argv + optind, left);

}

int main(int argc, char *argv[]) {

_cleanup_bus_flush_close_unref_ sd_bus *bus = NULL;

int r;

setlocale(LC_ALL, "");

log_parse_environment();

log_open();

r = parse_argv(argc, argv);

if (r <= 0)

goto finish;

r = bus_open_transport(arg_transport, arg_host, false, &bus);

if (r < 0) {

log_error_errno(r, "Failed to create bus connection: %m");

goto finish;

}

r = timedatectl_main(bus, argc, argv);

finish:

pager_close();

return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;

}