/*-*- 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.timedate1\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) { if (r != -ENOENT) log_warning("Failed to read /etc/timezone: %s", strerror(-r)); #ifdef TARGET_FEDORA r = parse_env_file("/etc/sysconfig/clock", NEWLINE, "ZONE", &zone, NULL); if (r < 0 && r != -ENOENT) log_warning("Failed to read /etc/sysconfig/clock: %s", strerror(-r)); #endif } if (isempty(zone)) { free(zone); zone = NULL; } 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 (!e) { 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; }