X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Fupdate-utmp%2Fupdate-utmp.c;h=7162084062fe95f9861322266291b5517ad6406e;hb=9eaebafed6dec1c3b5d1307b37f1b03a19530527;hp=31cae709509daf42edda515966654e034ee9db95;hpb=de0671ee7fe465e108f62dcbbbe9366f81dd9e9a;p=elogind.git diff --git a/src/update-utmp/update-utmp.c b/src/update-utmp/update-utmp.c index 31cae7095..716208406 100644 --- a/src/update-utmp/update-utmp.c +++ b/src/update-utmp/update-utmp.c @@ -1,5 +1,3 @@ -/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ - /*** This file is part of systemd. @@ -19,10 +17,8 @@ along with systemd; If not, see . ***/ -#include #include #include -#include #include #ifdef HAVE_AUDIT @@ -31,14 +27,22 @@ #include "sd-bus.h" -#include "log.h" -#include "macro.h" -#include "util.h" -#include "special.h" -#include "utmp-wtmp.h" -#include "bus-util.h" +//#include "alloc-util.h" #include "bus-error.h" -#include "unit-name.h" +//#include "bus-util.h" +//#include "formats-util.h" +//#include "log.h" +//#include "macro.h" +//#include "special.h" +//#include "unit-name.h" +//#include "util.h" +#include "utmp-wtmp.h" + +/// includes needed by elogind: +#include "string-util.h" +#include "time-util.h" +#include "update-utmp.h" + typedef struct Context { sd_bus *bus; @@ -48,7 +52,7 @@ typedef struct Context { } Context; static usec_t get_startup_time(Context *c) { - _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; usec_t t = 0; int r; @@ -63,13 +67,14 @@ static usec_t get_startup_time(Context *c) { &error, 't', &t); if (r < 0) { - log_error("Failed to get timestamp: %s", bus_error_message(&error, -r)); + log_error_errno(r, "Failed to get timestamp: %s", bus_error_message(&error, r)); return 0; } return t; } +#if 0 /// UNNEEDED by elogind static int get_current_runlevel(Context *c) { static const struct { const int runlevel; @@ -80,14 +85,12 @@ static int get_current_runlevel(Context *c) { * here over the others, since these are the main * runlevels used on Fedora. It might make sense to * change the order on some distributions. */ - { '5', SPECIAL_RUNLEVEL5_TARGET }, - { '3', SPECIAL_RUNLEVEL3_TARGET }, - { '4', SPECIAL_RUNLEVEL4_TARGET }, - { '2', SPECIAL_RUNLEVEL2_TARGET }, - { '1', SPECIAL_RESCUE_TARGET }, + { '5', SPECIAL_GRAPHICAL_TARGET }, + { '3', SPECIAL_MULTI_USER_TARGET }, + { '1', SPECIAL_RESCUE_TARGET }, }; - _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; int r; unsigned i; @@ -108,10 +111,8 @@ static int get_current_runlevel(Context *c) { "ActiveState", &error, &state); - if (r < 0) { - log_warning("Failed to get state: %s", bus_error_message(&error, -r)); - return r; - } + if (r < 0) + return log_warning_errno(r, "Failed to get state: %s", bus_error_message(&error, r)); if (streq(state, "active") || streq(state, "reloading")) return table[i].runlevel; @@ -119,6 +120,7 @@ static int get_current_runlevel(Context *c) { return 0; } +#endif // 0 static int on_reboot(Context *c) { int r = 0, q; @@ -131,10 +133,9 @@ static int on_reboot(Context *c) { #ifdef HAVE_AUDIT if (c->audit_fd >= 0) - if (audit_log_user_message(c->audit_fd, AUDIT_SYSTEM_BOOT, "init", NULL, NULL, NULL, 1) < 0 && + if (audit_log_user_comm_message(c->audit_fd, AUDIT_SYSTEM_BOOT, "", "systemd-update-utmp", NULL, NULL, NULL, 1) < 0 && errno != EPERM) { - log_error("Failed to send audit message: %m"); - r = -errno; + r = log_error_errno(errno, "Failed to send audit message: %m"); } #endif @@ -144,7 +145,7 @@ static int on_reboot(Context *c) { q = utmp_put_reboot(t); if (q < 0) { - log_error("Failed to write utmp record: %s", strerror(-q)); + log_error_errno(q, "Failed to write utmp record: %m"); r = q; } @@ -161,22 +162,22 @@ static int on_shutdown(Context *c) { #ifdef HAVE_AUDIT if (c->audit_fd >= 0) - if (audit_log_user_message(c->audit_fd, AUDIT_SYSTEM_SHUTDOWN, "init", NULL, NULL, NULL, 1) < 0 && + if (audit_log_user_comm_message(c->audit_fd, AUDIT_SYSTEM_SHUTDOWN, "", "systemd-update-utmp", NULL, NULL, NULL, 1) < 0 && errno != EPERM) { - log_error("Failed to send audit message: %m"); - r = -errno; + r = log_error_errno(errno, "Failed to send audit message: %m"); } #endif q = utmp_put_shutdown(); if (q < 0) { - log_error("Failed to write utmp record: %s", strerror(-q)); + log_error_errno(q, "Failed to write utmp record: %m"); r = q; } return r; } +#if 0 /// UNNEEDED by elogind static int on_runlevel(Context *c) { int r = 0, q, previous, runlevel; @@ -189,10 +190,8 @@ static int on_runlevel(Context *c) { q = utmp_get_runlevel(&previous, NULL); if (q < 0) { - if (q != -ESRCH && q != -ENOENT) { - log_error("Failed to get current runlevel: %s", strerror(-q)); - return q; - } + if (q != -ESRCH && q != -ENOENT) + return log_error_errno(q, "Failed to get current runlevel: %m"); previous = 0; } @@ -215,29 +214,32 @@ static int on_runlevel(Context *c) { runlevel > 0 ? runlevel : 'N') < 0) return log_oom(); - if (audit_log_user_message(c->audit_fd, AUDIT_SYSTEM_RUNLEVEL, s, NULL, NULL, NULL, 1) < 0 && - errno != EPERM) { - log_error("Failed to send audit message: %m"); - r = -errno; - } + if (audit_log_user_comm_message(c->audit_fd, AUDIT_SYSTEM_RUNLEVEL, s, "systemd-update-utmp", NULL, NULL, NULL, 1) < 0 && errno != EPERM) + r = log_error_errno(errno, "Failed to send audit message: %m"); } #endif q = utmp_put_runlevel(runlevel, previous); if (q < 0 && q != -ESRCH && q != -ENOENT) { - log_error("Failed to write utmp record: %s", strerror(-q)); + log_error_errno(q, "Failed to write utmp record: %m"); r = q; } return r; } +#endif // 0 +#if 0 /// elogind needs this to be a callable function int main(int argc, char *argv[]) { +#else +void update_utmp(int argc, char* argv[], sd_bus *bus) { +#endif // 0 Context c = { #ifdef HAVE_AUDIT .audit_fd = -1 #endif }; +#if 0 /// UNNEEDED by elogind int r; if (getppid() != 1) { @@ -255,17 +257,24 @@ int main(int argc, char *argv[]) { log_open(); umask(0022); +#else + assert(2 == argc); + assert(argv[1]); + assert(bus); +#endif // 0 #ifdef HAVE_AUDIT /* If the kernel lacks netlink or audit support, * don't worry about it. */ c.audit_fd = audit_open(); if (c.audit_fd < 0 && errno != EAFNOSUPPORT && errno != EPROTONOSUPPORT) - log_error("Failed to connect to audit log: %m"); + log_error_errno(errno, "Failed to connect to audit log: %m"); #endif - r = bus_open_system_systemd(&c.bus); + +#if 0 /// UNNEEDED by elogind + r = bus_connect_system_systemd(&c.bus); if (r < 0) { - log_error("Failed to get D-Bus connection: %s", strerror(-r)); + log_error_errno(r, "Failed to get D-Bus connection: %m"); r = -EIO; goto finish; } @@ -286,13 +295,22 @@ int main(int argc, char *argv[]) { log_debug("systemd-update-utmp stopped as pid "PID_FMT, getpid()); finish: +#else + c.bus = bus; + if (streq(argv[1], "reboot")) + (void)on_reboot(&c); + else if (streq(argv[1], "shutdown")) + (void)on_shutdown(&c); +#endif // 0 + #ifdef HAVE_AUDIT if (c.audit_fd >= 0) audit_close(c.audit_fd); #endif - if (c.bus) - sd_bus_unref(c.bus); + sd_bus_flush_close_unref(c.bus); +#if 0 /// UNNEEDED by elogind return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; +#endif // 0 }