From dacf8817be3943b8911d290fb9cbaaa5d659ad95 Mon Sep 17 00:00:00 2001 From: Laurent Bercot Date: Wed, 21 Apr 2021 13:50:10 +0000 Subject: The final wtmp entry is also os-specific --- src/include-local/os.h | 2 ++ src/os/deps-lib/os-freebsd | 1 + src/os/deps-lib/os-linux | 1 + src/os/freebsd-os_final_wtmp.c | 10 ++++++ src/os/freebsd-os_kbspecials.c | 5 +-- src/os/linux-os_final_wtmp.c | 68 ++++++++++++++++++++++++++++++++++++++++ src/shutdown/s6-linux-init-hpr.c | 60 ++--------------------------------- 7 files changed, 87 insertions(+), 60 deletions(-) create mode 100644 src/os/freebsd-os_final_wtmp.c create mode 100644 src/os/linux-os_final_wtmp.c (limited to 'src') diff --git a/src/include-local/os.h b/src/include-local/os.h index 72062e9..4a1d7c2 100644 --- a/src/include-local/os.h +++ b/src/include-local/os.h @@ -3,7 +3,9 @@ #ifndef S6_LINUX_INIT_OS_H #define S6_LINUX_INIT_OS_H +extern void os_final_wtmp (int) ; extern void os_reboot (int) ; + extern void os_kbspecials (int) ; extern void os_mount_tmpfs (char const *, unsigned int) ; extern int os_mount_devtmpfs (char const *) ; diff --git a/src/os/deps-lib/os-freebsd b/src/os/deps-lib/os-freebsd index 54edc05..201369f 100644 --- a/src/os/deps-lib/os-freebsd +++ b/src/os/deps-lib/os-freebsd @@ -1,3 +1,4 @@ +freebsd-os_final_wtmp.o freebsd-os_kbspecials.o freebsd-os_mount_devtmpfs.o freebsd-os_mount_tmpfs.o diff --git a/src/os/deps-lib/os-linux b/src/os/deps-lib/os-linux index f58c20e..f257471 100644 --- a/src/os/deps-lib/os-linux +++ b/src/os/deps-lib/os-linux @@ -1,3 +1,4 @@ +linux-os_final_wtmp.o linux-os_kbspecials.o linux-os_mount_devtmpfs.o linux-os_mount_tmpfs.o diff --git a/src/os/freebsd-os_final_wtmp.c b/src/os/freebsd-os_final_wtmp.c new file mode 100644 index 0000000..868e0c3 --- /dev/null +++ b/src/os/freebsd-os_final_wtmp.c @@ -0,0 +1,10 @@ +/* ISC license. */ + +#include "os.h" + + /* FreeBSD only does utx.log (wtmp) operations at the rc level. */ + +void os_final_wtmp (int what) +{ + (void)what ; +} diff --git a/src/os/freebsd-os_kbspecials.c b/src/os/freebsd-os_kbspecials.c index 71a0480..1d7092d 100644 --- a/src/os/freebsd-os_kbspecials.c +++ b/src/os/freebsd-os_kbspecials.c @@ -2,8 +2,9 @@ #include "os.h" + /* FreeBSD hardcodes cad and doesn't have kbrequest */ + void os_kbspecials (int inns) { - if (inns) return ; - return ; + (void)inns ; } diff --git a/src/os/linux-os_final_wtmp.c b/src/os/linux-os_final_wtmp.c new file mode 100644 index 0000000..b3ce7c5 --- /dev/null +++ b/src/os/linux-os_final_wtmp.c @@ -0,0 +1,68 @@ +/* ISC license. */ + +#include + +#include +#include +#include +#include + +#include +#include + +#include "os.h" + +#ifndef WTMPX_FILE +# include +# ifdef WTMP_FILE +# define WTMPX_FILE WTMP_FILE +# else +# ifdef _PATH_WTMP +# define WTMPX_FILE _PATH_WTMP +# else +# define WTMPX_FILE "/var/log/wtmp" +# endif +# endif +#endif + +#ifndef UT_NAMESIZE +# define UT_NAMESIZE 32 +#endif + +#ifndef UT_HOSTSIZE +# define UT_HOSTSIZE 256 +#endif + +void os_final_wtmp (int what) +{ + struct utmpx utx = + { + .ut_type = RUN_LVL, + .ut_pid = getpid(), + .ut_line = "~", + .ut_id = "", + .ut_session = getsid(0) + } ; + strncpy(utx.ut_user, what == 3 ? "reboot" : "shutdown", UT_NAMESIZE) ; + if (gethostname(utx.ut_host, UT_HOSTSIZE) < 0) + { + utx.ut_host[0] = 0 ; + strerr_warnwu1sys("gethostname") ; + } + else utx.ut_host[UT_HOSTSIZE - 1] = 0 ; + + /* glibc multilib can go fuck itself */ +#ifdef __WORDSIZE_TIME64_COMPAT32 + { + struct timeval tv ; + if (!timeval_from_tain(&tv, &STAMP)) + strerr_warnwu1sys("timeval_from_tain") ; + utx.ut_tv.tv_sec = tv.tv_sec ; + utx.ut_tv.tv_usec = tv.tv_usec ; + } +#else + if (!timeval_from_tain(&utx.ut_tv, &STAMP)) + strerr_warnwu1sys("timeval_from_tain") ; +#endif + updwtmpx(WTMPX_FILE, &utx) ; +} diff --git a/src/shutdown/s6-linux-init-hpr.c b/src/shutdown/s6-linux-init-hpr.c index c92717c..14e1844 100644 --- a/src/shutdown/s6-linux-init-hpr.c +++ b/src/shutdown/s6-linux-init-hpr.c @@ -1,37 +1,14 @@ /* ISC license. */ -#include - #include -#include #include -#include #include #include -#include #include -#include -#include "os.h" -#include "defaults.h" #include "hpr.h" - -#ifndef UT_NAMESIZE -#define UT_NAMESIZE 32 -#endif - -#ifndef UT_HOSTSIZE -#define UT_HOSTSIZE 256 -#endif - -#ifndef _PATH_WTMP -# ifdef WTMPX_FILE -# define _PATH_WTMP WTMPX_FILE -# else -# define _PATH_WTMP "/var/log/wtmp" -# endif -#endif +#include "os.h" #define USAGE "s6-linux-init-hpr [ -h | -p | -r ] [ -n ] [ -d | -w ] [ -W ] [ -f ] [ -i ]" @@ -87,40 +64,7 @@ int main (int argc, char const *const *argv) } if (!tain_now_g()) strerr_warnw1sys("get current time") ; - if (dowtmp) - { - struct utmpx utx = - { - .ut_type = RUN_LVL, - .ut_pid = getpid(), - .ut_line = "~", - .ut_id = "", - .ut_session = getsid(0) - } ; - strncpy(utx.ut_user, what == 3 ? "reboot" : "shutdown", UT_NAMESIZE) ; - if (gethostname(utx.ut_host, UT_HOSTSIZE) < 0) - { - utx.ut_host[0] = 0 ; - strerr_warnwu1sys("gethostname") ; - } - else utx.ut_host[UT_HOSTSIZE - 1] = 0 ; - - /* glibc multilib can go fuck itself */ -#ifdef __WORDSIZE_TIME64_COMPAT32 - { - struct timeval tv ; - if (!timeval_from_tain(&tv, &STAMP)) - strerr_warnwu1sys("timeval_from_tain") ; - utx.ut_tv.tv_sec = tv.tv_sec ; - utx.ut_tv.tv_usec = tv.tv_usec ; - } -#else - if (!timeval_from_tain(&utx.ut_tv, &STAMP)) - strerr_warnwu1sys("timeval_from_tain") ; -#endif - - updwtmpx(_PATH_WTMP, &utx) ; - } + if (dowtmp) os_final_wtmp(what) ; if (dowall) hpr_wall(HPR_WALL_BANNER) ; if (dowtmp < 2) { -- cgit v1.2.3