896 lines
29 KiB
C
896 lines
29 KiB
C
/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
|
|
|
|
/***
|
|
This file is part of systemd.
|
|
|
|
Copyright 2011 Lennart Poettering
|
|
|
|
systemd is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
systemd 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
|
|
General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with systemd; If not, see <http://www.gnu.org/licenses/>.
|
|
***/
|
|
|
|
#include <dbus/dbus.h>
|
|
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "util.h"
|
|
#include "strv.h"
|
|
#include "dbus-common.h"
|
|
#include "polkit.h"
|
|
#include "def.h"
|
|
|
|
#define NULL_ADJTIME_UTC "0.0 0 0\n0\nUTC\n"
|
|
#define NULL_ADJTIME_LOCAL "0.0 0 0\n0\nLOCAL\n"
|
|
|
|
#define INTERFACE \
|
|
" <interface name=\"org.freedesktop.timedate1\">\n" \
|
|
" <property name=\"Timezone\" type=\"s\" access=\"read\"/>\n" \
|
|
" <property name=\"LocalRTC\" type=\"b\" access=\"read\"/>\n" \
|
|
" <property name=\"NTP\" type=\"b\" access=\"read\"/>\n" \
|
|
" <method name=\"SetTime\">\n" \
|
|
" <arg name=\"usec_utc\" type=\"x\" direction=\"in\"/>\n" \
|
|
" <arg name=\"relative\" type=\"b\" direction=\"in\"/>\n" \
|
|
" <arg name=\"user_interaction\" type=\"b\" direction=\"in\"/>\n" \
|
|
" </method>\n" \
|
|
" <method name=\"SetTimezone\">\n" \
|
|
" <arg name=\"timezone\" type=\"s\" direction=\"in\"/>\n" \
|
|
" <arg name=\"user_interaction\" type=\"b\" direction=\"in\"/>\n" \
|
|
" </method>\n" \
|
|
" <method name=\"SetLocalRTC\">\n" \
|
|
" <arg name=\"local_rtc\" type=\"b\" direction=\"in\"/>\n" \
|
|
" <arg name=\"fix_system\" type=\"b\" direction=\"in\"/>\n" \
|
|
" <arg name=\"user_interaction\" type=\"b\" direction=\"in\"/>\n" \
|
|
" </method>\n" \
|
|
" <method name=\"SetNTP\">\n" \
|
|
" <arg name=\"use_ntp\" type=\"b\" direction=\"in\"/>\n" \
|
|
" <arg name=\"user_interaction\" type=\"b\" direction=\"in\"/>\n" \
|
|
" </method>\n" \
|
|
" </interface>\n"
|
|
|
|
#define INTROSPECTION \
|
|
DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE \
|
|
"<node>\n" \
|
|
INTERFACE \
|
|
BUS_PROPERTIES_INTERFACE \
|
|
BUS_INTROSPECTABLE_INTERFACE \
|
|
BUS_PEER_INTERFACE \
|
|
"</node>\n"
|
|
|
|
#define INTERFACES_LIST \
|
|
BUS_GENERIC_INTERFACES_LIST \
|
|
"org.freedesktop.locale1\0"
|
|
|
|
const char timedate_interface[] _introspect_("timedate1") = INTERFACE;
|
|
|
|
static char *zone = NULL;
|
|
static bool local_rtc = false;
|
|
static int use_ntp = -1;
|
|
|
|
static usec_t remain_until = 0;
|
|
|
|
static void free_data(void) {
|
|
free(zone);
|
|
zone = NULL;
|
|
|
|
local_rtc = false;
|
|
}
|
|
|
|
static bool valid_timezone(const char *name) {
|
|
const char *p;
|
|
char *t;
|
|
bool slash = false;
|
|
int r;
|
|
struct stat st;
|
|
|
|
assert(name);
|
|
|
|
if (*name == '/' || *name == 0)
|
|
return false;
|
|
|
|
for (p = name; *p; p++) {
|
|
if (!(*p >= '0' && *p <= '9') &&
|
|
!(*p >= 'a' && *p <= 'z') &&
|
|
!(*p >= 'A' && *p <= 'Z') &&
|
|
!(*p == '-' || *p == '_' || *p == '+' || *p == '/'))
|
|
return false;
|
|
|
|
if (*p == '/') {
|
|
|
|
if (slash)
|
|
return false;
|
|
|
|
slash = true;
|
|
} else
|
|
slash = false;
|
|
}
|
|
|
|
if (slash)
|
|
return false;
|
|
|
|
t = strappend("/usr/share/zoneinfo/", name);
|
|
if (!t)
|
|
return false;
|
|
|
|
r = stat(t, &st);
|
|
free(t);
|
|
|
|
if (r < 0)
|
|
return false;
|
|
|
|
if (!S_ISREG(st.st_mode))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
static void verify_timezone(void) {
|
|
char *p, *a = NULL, *b = NULL;
|
|
size_t l, q;
|
|
int j, k;
|
|
|
|
if (!zone)
|
|
return;
|
|
|
|
p = strappend("/usr/share/zoneinfo/", zone);
|
|
if (!p) {
|
|
log_error("Out of memory");
|
|
return;
|
|
}
|
|
|
|
j = read_full_file("/etc/localtime", &a, &l);
|
|
k = read_full_file(p, &b, &q);
|
|
|
|
free(p);
|
|
|
|
if (j < 0 || k < 0 || l != q || memcmp(a, b, l)) {
|
|
log_warning("/etc/localtime and /etc/timezone out of sync.");
|
|
free(zone);
|
|
zone = NULL;
|
|
}
|
|
|
|
free(a);
|
|
free(b);
|
|
}
|
|
|
|
static int read_data(void) {
|
|
int r;
|
|
|
|
free_data();
|
|
|
|
r = read_one_line_file("/etc/timezone", &zone);
|
|
if (r < 0 && r != -ENOENT)
|
|
return r;
|
|
|
|
verify_timezone();
|
|
|
|
local_rtc = hwclock_is_localtime() > 0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int write_data_timezone(void) {
|
|
int r = 0;
|
|
char *p;
|
|
|
|
if (!zone) {
|
|
if (unlink("/etc/timezone") < 0 && errno != ENOENT)
|
|
r = -errno;
|
|
|
|
if (unlink("/etc/localtime") < 0 && errno != ENOENT)
|
|
r = -errno;
|
|
|
|
return r;
|
|
}
|
|
|
|
p = strappend("/usr/share/zoneinfo/", zone);
|
|
if (!p) {
|
|
log_error("Out of memory");
|
|
return -ENOMEM;
|
|
}
|
|
|
|
r = symlink_or_copy_atomic(p, "/etc/localtime");
|
|
free(p);
|
|
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = write_one_line_file_atomic("/etc/timezone", zone);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int write_data_local_rtc(void) {
|
|
int r;
|
|
char *s, *w;
|
|
|
|
r = read_full_file("/etc/adjtime", &s, NULL);
|
|
if (r < 0) {
|
|
if (r != -ENOENT)
|
|
return r;
|
|
|
|
if (!local_rtc)
|
|
return 0;
|
|
|
|
w = strdup(NULL_ADJTIME_LOCAL);
|
|
if (!w)
|
|
return -ENOMEM;
|
|
} else {
|
|
char *p, *e;
|
|
size_t a, b;
|
|
|
|
p = strchr(s, '\n');
|
|
if (!p) {
|
|
free(s);
|
|
return -EIO;
|
|
}
|
|
|
|
p = strchr(p+1, '\n');
|
|
if (!p) {
|
|
free(s);
|
|
return -EIO;
|
|
}
|
|
|
|
p++;
|
|
e = strchr(p, '\n');
|
|
if (!p) {
|
|
free(s);
|
|
return -EIO;
|
|
}
|
|
|
|
a = p - s;
|
|
b = strlen(e);
|
|
|
|
w = new(char, a + (local_rtc ? 5 : 3) + b + 1);
|
|
if (!w) {
|
|
free(s);
|
|
return -ENOMEM;
|
|
}
|
|
|
|
*(char*) mempcpy(stpcpy(mempcpy(w, s, a), local_rtc ? "LOCAL" : "UTC"), e, b) = 0;
|
|
|
|
if (streq(w, NULL_ADJTIME_UTC)) {
|
|
free(w);
|
|
|
|
if (unlink("/etc/adjtime") < 0) {
|
|
if (errno != ENOENT)
|
|
return -errno;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
r = write_one_line_file_atomic("/etc/adjtime", w);
|
|
free(w);
|
|
|
|
return r;
|
|
}
|
|
|
|
static int read_ntp(DBusConnection *bus) {
|
|
DBusMessage *m = NULL, *reply = NULL;
|
|
const char *name = "ntpd.service", *s;
|
|
DBusError error;
|
|
int r;
|
|
|
|
assert(bus);
|
|
|
|
dbus_error_init(&error);
|
|
|
|
m = dbus_message_new_method_call(
|
|
"org.freedesktop.systemd1",
|
|
"/org/freedesktop/systemd1",
|
|
"org.freedesktop.systemd1.Manager",
|
|
"GetUnitFileState");
|
|
|
|
if (!m) {
|
|
log_error("Out of memory");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
if (!dbus_message_append_args(m,
|
|
DBUS_TYPE_STRING, &name,
|
|
DBUS_TYPE_INVALID)) {
|
|
log_error("Could not append arguments to message.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error);
|
|
if (!reply) {
|
|
log_error("Failed to issue method call: %s", bus_error_message(&error));
|
|
r = -EIO;
|
|
goto finish;
|
|
}
|
|
|
|
if (!dbus_message_get_args(reply, &error,
|
|
DBUS_TYPE_STRING, &s,
|
|
DBUS_TYPE_INVALID)) {
|
|
log_error("Failed to parse reply: %s", bus_error_message(&error));
|
|
r = -EIO;
|
|
goto finish;
|
|
}
|
|
|
|
use_ntp =
|
|
streq(s, "enabled") ||
|
|
streq(s, "enabled-runtime");
|
|
r = 0;
|
|
|
|
finish:
|
|
if (m)
|
|
dbus_message_unref(m);
|
|
|
|
if (reply)
|
|
dbus_message_unref(reply);
|
|
|
|
dbus_error_free(&error);
|
|
|
|
return r;
|
|
}
|
|
|
|
static int start_ntp(DBusConnection *bus, DBusError *error) {
|
|
DBusMessage *m = NULL, *reply = NULL;
|
|
const char *name = "ntpd.service", *mode = "replace";
|
|
int r;
|
|
|
|
assert(bus);
|
|
assert(error);
|
|
|
|
m = dbus_message_new_method_call(
|
|
"org.freedesktop.systemd1",
|
|
"/org/freedesktop/systemd1",
|
|
"org.freedesktop.systemd1.Manager",
|
|
use_ntp ? "StartUnit" : "StopUnit");
|
|
if (!m) {
|
|
log_error("Could not allocate message.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
if (!dbus_message_append_args(m,
|
|
DBUS_TYPE_STRING, &name,
|
|
DBUS_TYPE_STRING, &mode,
|
|
DBUS_TYPE_INVALID)) {
|
|
log_error("Could not append arguments to message.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
reply = dbus_connection_send_with_reply_and_block(bus, m, -1, error);
|
|
if (!reply) {
|
|
log_error("Failed to issue method call: %s", bus_error_message(error));
|
|
r = -EIO;
|
|
goto finish;
|
|
}
|
|
|
|
r = 0;
|
|
|
|
finish:
|
|
if (m)
|
|
dbus_message_unref(m);
|
|
|
|
if (reply)
|
|
dbus_message_unref(reply);
|
|
|
|
return r;
|
|
}
|
|
|
|
static int enable_ntp(DBusConnection *bus, DBusError *error) {
|
|
DBusMessage *m = NULL, *reply = NULL;
|
|
const char * const names[] = { "ntpd.service", NULL };
|
|
int r;
|
|
DBusMessageIter iter;
|
|
dbus_bool_t f = FALSE, t = TRUE;
|
|
|
|
assert(bus);
|
|
assert(error);
|
|
|
|
m = dbus_message_new_method_call(
|
|
"org.freedesktop.systemd1",
|
|
"/org/freedesktop/systemd1",
|
|
"org.freedesktop.systemd1.Manager",
|
|
use_ntp ? "EnableUnitFiles" : "DisableUnitFiles");
|
|
|
|
if (!m) {
|
|
log_error("Could not allocate message.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
dbus_message_iter_init_append(m, &iter);
|
|
|
|
r = bus_append_strv_iter(&iter, (char**) names);
|
|
if (r < 0) {
|
|
log_error("Failed to append unit files.");
|
|
goto finish;
|
|
}
|
|
/* send runtime bool */
|
|
if (!dbus_message_iter_append_basic(&iter, DBUS_TYPE_BOOLEAN, &f)) {
|
|
log_error("Failed to append runtime boolean.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
if (use_ntp) {
|
|
/* send force bool */
|
|
if (!dbus_message_iter_append_basic(&iter, DBUS_TYPE_BOOLEAN, &t)) {
|
|
log_error("Failed to append force boolean.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
}
|
|
|
|
reply = dbus_connection_send_with_reply_and_block(bus, m, -1, error);
|
|
if (!reply) {
|
|
log_error("Failed to issue method call: %s", bus_error_message(error));
|
|
r = -EIO;
|
|
goto finish;
|
|
}
|
|
|
|
dbus_message_unref(m);
|
|
m = dbus_message_new_method_call(
|
|
"org.freedesktop.systemd1",
|
|
"/org/freedesktop/systemd1",
|
|
"org.freedesktop.systemd1.Manager",
|
|
"Reload");
|
|
if (!m) {
|
|
log_error("Could not allocate message.");
|
|
r = -ENOMEM;
|
|
goto finish;
|
|
}
|
|
|
|
dbus_message_unref(reply);
|
|
reply = dbus_connection_send_with_reply_and_block(bus, m, -1, error);
|
|
if (!reply) {
|
|
log_error("Failed to issue method call: %s", bus_error_message(error));
|
|
r = -EIO;
|
|
goto finish;
|
|
}
|
|
|
|
r = 0;
|
|
|
|
finish:
|
|
if (m)
|
|
dbus_message_unref(m);
|
|
|
|
if (reply)
|
|
dbus_message_unref(reply);
|
|
|
|
return r;
|
|
}
|
|
|
|
static int property_append_ntp(DBusMessageIter *i, const char *property, void *data) {
|
|
dbus_bool_t db;
|
|
|
|
assert(i);
|
|
assert(property);
|
|
|
|
db = use_ntp > 0;
|
|
|
|
if (!dbus_message_iter_append_basic(i, DBUS_TYPE_BOOLEAN, &db))
|
|
return -ENOMEM;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static DBusHandlerResult timedate_message_handler(
|
|
DBusConnection *connection,
|
|
DBusMessage *message,
|
|
void *userdata) {
|
|
|
|
const BusProperty properties[] = {
|
|
{ "org.freedesktop.timedate1", "Timezone", bus_property_append_string, "s", zone },
|
|
{ "org.freedesktop.timedate1", "LocalRTC", bus_property_append_bool, "b", &local_rtc },
|
|
{ "org.freedesktop.timedate1", "NTP", property_append_ntp, "b", NULL },
|
|
{ NULL, NULL, NULL, NULL, NULL }
|
|
};
|
|
|
|
DBusMessage *reply = NULL, *changed = NULL;
|
|
DBusError error;
|
|
int r;
|
|
|
|
assert(connection);
|
|
assert(message);
|
|
|
|
dbus_error_init(&error);
|
|
|
|
if (dbus_message_is_method_call(message, "org.freedesktop.timedate1", "SetTimezone")) {
|
|
const char *z;
|
|
dbus_bool_t interactive;
|
|
|
|
if (!dbus_message_get_args(
|
|
message,
|
|
&error,
|
|
DBUS_TYPE_STRING, &z,
|
|
DBUS_TYPE_BOOLEAN, &interactive,
|
|
DBUS_TYPE_INVALID))
|
|
return bus_send_error_reply(connection, message, &error, -EINVAL);
|
|
|
|
if (!valid_timezone(z))
|
|
return bus_send_error_reply(connection, message, NULL, -EINVAL);
|
|
|
|
if (!streq_ptr(z, zone)) {
|
|
char *t;
|
|
|
|
r = verify_polkit(connection, message, "org.freedesktop.timedate1.set-timezone", interactive, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
t = strdup(z);
|
|
if (!t)
|
|
goto oom;
|
|
|
|
free(zone);
|
|
zone = t;
|
|
|
|
/* 1. Write new configuration file */
|
|
r = write_data_timezone();
|
|
if (r < 0) {
|
|
log_error("Failed to set timezone: %s", strerror(-r));
|
|
return bus_send_error_reply(connection, message, NULL, r);
|
|
}
|
|
|
|
if (local_rtc) {
|
|
struct timespec ts;
|
|
struct tm *tm;
|
|
|
|
/* 2. Teach kernel new timezone */
|
|
hwclock_apply_localtime_delta(NULL);
|
|
|
|
/* 3. Sync RTC from system clock, with the new delta */
|
|
assert_se(clock_gettime(CLOCK_REALTIME, &ts) == 0);
|
|
assert_se(tm = localtime(&ts.tv_sec));
|
|
hwclock_set_time(tm);
|
|
}
|
|
|
|
log_info("Changed timezone to '%s'.", zone);
|
|
|
|
changed = bus_properties_changed_new(
|
|
"/org/freedesktop/timedate1",
|
|
"org.freedesktop.timedate1",
|
|
"Timezone\0");
|
|
if (!changed)
|
|
goto oom;
|
|
}
|
|
|
|
} else if (dbus_message_is_method_call(message, "org.freedesktop.timedate1", "SetLocalRTC")) {
|
|
dbus_bool_t lrtc;
|
|
dbus_bool_t fix_system;
|
|
dbus_bool_t interactive;
|
|
|
|
if (!dbus_message_get_args(
|
|
message,
|
|
&error,
|
|
DBUS_TYPE_BOOLEAN, &lrtc,
|
|
DBUS_TYPE_BOOLEAN, &fix_system,
|
|
DBUS_TYPE_BOOLEAN, &interactive,
|
|
DBUS_TYPE_INVALID))
|
|
return bus_send_error_reply(connection, message, &error, -EINVAL);
|
|
|
|
if (lrtc != local_rtc) {
|
|
struct timespec ts;
|
|
|
|
r = verify_polkit(connection, message, "org.freedesktop.timedate1.set-local-rtc", interactive, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
local_rtc = lrtc;
|
|
|
|
/* 1. Write new configuration file */
|
|
r = write_data_local_rtc();
|
|
if (r < 0) {
|
|
log_error("Failed to set RTC to local/UTC: %s", strerror(-r));
|
|
return bus_send_error_reply(connection, message, NULL, r);
|
|
}
|
|
|
|
/* 2. Teach kernel new timezone */
|
|
if (local_rtc)
|
|
hwclock_apply_localtime_delta(NULL);
|
|
else
|
|
hwclock_reset_localtime_delta();
|
|
|
|
/* 3. Synchronize clocks */
|
|
assert_se(clock_gettime(CLOCK_REALTIME, &ts) == 0);
|
|
|
|
if (fix_system) {
|
|
struct tm tm;
|
|
|
|
/* Sync system clock from RTC; first,
|
|
* initialize the timezone fields of
|
|
* struct tm. */
|
|
if (local_rtc)
|
|
tm = *localtime(&ts.tv_sec);
|
|
else
|
|
tm = *gmtime(&ts.tv_sec);
|
|
|
|
/* Override the main fields of
|
|
* struct tm, but not the timezone
|
|
* fields */
|
|
if (hwclock_get_time(&tm) >= 0) {
|
|
|
|
/* And set the system clock
|
|
* with this */
|
|
if (local_rtc)
|
|
ts.tv_sec = mktime(&tm);
|
|
else
|
|
ts.tv_sec = timegm(&tm);
|
|
|
|
clock_settime(CLOCK_REALTIME, &ts);
|
|
}
|
|
|
|
} else {
|
|
struct tm *tm;
|
|
|
|
/* Sync RTC from system clock */
|
|
if (local_rtc)
|
|
tm = localtime(&ts.tv_sec);
|
|
else
|
|
tm = gmtime(&ts.tv_sec);
|
|
|
|
hwclock_set_time(tm);
|
|
}
|
|
|
|
log_info("RTC configured to %s time.", local_rtc ? "local" : "UTC");
|
|
|
|
changed = bus_properties_changed_new(
|
|
"/org/freedesktop/timedate1",
|
|
"org.freedesktop.timedate1",
|
|
"LocalRTC\0");
|
|
if (!changed)
|
|
goto oom;
|
|
}
|
|
|
|
} else if (dbus_message_is_method_call(message, "org.freedesktop.timedate1", "SetTime")) {
|
|
int64_t utc;
|
|
dbus_bool_t relative;
|
|
dbus_bool_t interactive;
|
|
|
|
if (!dbus_message_get_args(
|
|
message,
|
|
&error,
|
|
DBUS_TYPE_INT64, &utc,
|
|
DBUS_TYPE_BOOLEAN, &relative,
|
|
DBUS_TYPE_BOOLEAN, &interactive,
|
|
DBUS_TYPE_INVALID))
|
|
return bus_send_error_reply(connection, message, &error, -EINVAL);
|
|
|
|
if (!relative && utc <= 0)
|
|
return bus_send_error_reply(connection, message, NULL, -EINVAL);
|
|
|
|
if (!relative || utc != 0) {
|
|
struct timespec ts;
|
|
struct tm* tm;
|
|
|
|
r = verify_polkit(connection, message, "org.freedesktop.timedate1.set-time", interactive, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
if (relative)
|
|
timespec_store(&ts, now(CLOCK_REALTIME) + utc);
|
|
else
|
|
timespec_store(&ts, utc);
|
|
|
|
/* Set system clock */
|
|
if (clock_settime(CLOCK_REALTIME, &ts) < 0) {
|
|
log_error("Failed to set local time: %m");
|
|
return bus_send_error_reply(connection, message, NULL, -errno);
|
|
}
|
|
|
|
/* Sync down to RTC */
|
|
if (local_rtc)
|
|
tm = localtime(&ts.tv_sec);
|
|
else
|
|
tm = gmtime(&ts.tv_sec);
|
|
|
|
hwclock_set_time(tm);
|
|
|
|
log_info("Changed local time to %s", ctime(&ts.tv_sec));
|
|
}
|
|
} else if (dbus_message_is_method_call(message, "org.freedesktop.timedate1", "SetNTP")) {
|
|
dbus_bool_t ntp;
|
|
dbus_bool_t interactive;
|
|
|
|
if (!dbus_message_get_args(
|
|
message,
|
|
&error,
|
|
DBUS_TYPE_BOOLEAN, &ntp,
|
|
DBUS_TYPE_BOOLEAN, &interactive,
|
|
DBUS_TYPE_INVALID))
|
|
return bus_send_error_reply(connection, message, &error, -EINVAL);
|
|
|
|
if (ntp != !!use_ntp) {
|
|
|
|
r = verify_polkit(connection, message, "org.freedesktop.timedate1.set-ntp", interactive, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
use_ntp = !!ntp;
|
|
|
|
r = enable_ntp(connection, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
r = start_ntp(connection, &error);
|
|
if (r < 0)
|
|
return bus_send_error_reply(connection, message, &error, r);
|
|
|
|
log_info("Set NTP to %s", use_ntp ? "enabled" : "disabled");
|
|
|
|
changed = bus_properties_changed_new(
|
|
"/org/freedesktop/timedate1",
|
|
"org.freedesktop.timedate1",
|
|
"NTP\0");
|
|
if (!changed)
|
|
goto oom;
|
|
}
|
|
|
|
} else
|
|
return bus_default_message_handler(connection, message, INTROSPECTION, INTERFACES_LIST, properties);
|
|
|
|
if (!(reply = dbus_message_new_method_return(message)))
|
|
goto oom;
|
|
|
|
if (!dbus_connection_send(connection, reply, NULL))
|
|
goto oom;
|
|
|
|
dbus_message_unref(reply);
|
|
reply = NULL;
|
|
|
|
if (changed) {
|
|
|
|
if (!dbus_connection_send(connection, changed, NULL))
|
|
goto oom;
|
|
|
|
dbus_message_unref(changed);
|
|
}
|
|
|
|
return DBUS_HANDLER_RESULT_HANDLED;
|
|
|
|
oom:
|
|
if (reply)
|
|
dbus_message_unref(reply);
|
|
|
|
if (changed)
|
|
dbus_message_unref(changed);
|
|
|
|
dbus_error_free(&error);
|
|
|
|
return DBUS_HANDLER_RESULT_NEED_MEMORY;
|
|
}
|
|
|
|
static int connect_bus(DBusConnection **_bus) {
|
|
static const DBusObjectPathVTable timedate_vtable = {
|
|
.message_function = timedate_message_handler
|
|
};
|
|
DBusError error;
|
|
DBusConnection *bus = NULL;
|
|
int r;
|
|
|
|
assert(_bus);
|
|
|
|
dbus_error_init(&error);
|
|
|
|
bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, &error);
|
|
if (!bus) {
|
|
log_error("Failed to get system D-Bus connection: %s", bus_error_message(&error));
|
|
r = -ECONNREFUSED;
|
|
goto fail;
|
|
}
|
|
|
|
dbus_connection_set_exit_on_disconnect(bus, FALSE);
|
|
|
|
if (!dbus_connection_register_object_path(bus, "/org/freedesktop/timedate1", &timedate_vtable, NULL) ||
|
|
!dbus_connection_add_filter(bus, bus_exit_idle_filter, &remain_until, NULL)) {
|
|
log_error("Not enough memory");
|
|
r = -ENOMEM;
|
|
goto fail;
|
|
}
|
|
|
|
r = dbus_bus_request_name(bus, "org.freedesktop.timedate1", DBUS_NAME_FLAG_DO_NOT_QUEUE, &error);
|
|
if (dbus_error_is_set(&error)) {
|
|
log_error("Failed to register name on bus: %s", bus_error_message(&error));
|
|
r = -EEXIST;
|
|
goto fail;
|
|
}
|
|
|
|
if (r != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER) {
|
|
log_error("Failed to acquire name.");
|
|
r = -EEXIST;
|
|
goto fail;
|
|
}
|
|
|
|
if (_bus)
|
|
*_bus = bus;
|
|
|
|
return 0;
|
|
|
|
fail:
|
|
dbus_connection_close(bus);
|
|
dbus_connection_unref(bus);
|
|
|
|
dbus_error_free(&error);
|
|
|
|
return r;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
int r;
|
|
DBusConnection *bus = NULL;
|
|
bool exiting = false;
|
|
|
|
log_set_target(LOG_TARGET_AUTO);
|
|
log_parse_environment();
|
|
log_open();
|
|
|
|
umask(0022);
|
|
|
|
if (argc == 2 && streq(argv[1], "--introspect")) {
|
|
fputs(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE
|
|
"<node>\n", stdout);
|
|
fputs(timedate_interface, stdout);
|
|
fputs("</node>\n", stdout);
|
|
return 0;
|
|
}
|
|
|
|
if (argc != 1) {
|
|
log_error("This program takes no arguments.");
|
|
r = -EINVAL;
|
|
goto finish;
|
|
}
|
|
|
|
r = read_data();
|
|
if (r < 0) {
|
|
log_error("Failed to read timezone data: %s", strerror(-r));
|
|
goto finish;
|
|
}
|
|
|
|
r = connect_bus(&bus);
|
|
if (r < 0)
|
|
goto finish;
|
|
|
|
r = read_ntp(bus);
|
|
if (r < 0) {
|
|
log_error("Failed to determine whether NTP is enabled: %s", strerror(-r));
|
|
goto finish;
|
|
}
|
|
|
|
remain_until = now(CLOCK_MONOTONIC) + DEFAULT_EXIT_USEC;
|
|
for (;;) {
|
|
|
|
if (!dbus_connection_read_write_dispatch(bus, exiting ? -1 : (int) (DEFAULT_EXIT_USEC/USEC_PER_MSEC)))
|
|
break;
|
|
|
|
if (!exiting && remain_until < now(CLOCK_MONOTONIC)) {
|
|
exiting = true;
|
|
bus_async_unregister_and_exit(bus, "org.freedesktop.hostname1");
|
|
}
|
|
}
|
|
|
|
r = 0;
|
|
|
|
finish:
|
|
free_data();
|
|
|
|
if (bus) {
|
|
dbus_connection_flush(bus);
|
|
dbus_connection_close(bus);
|
|
dbus_connection_unref(bus);
|
|
}
|
|
|
|
return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
|
|
}
|