From 680b86afbb8ba466176ad6866409496225660009 Mon Sep 17 00:00:00 2001 From: Denis Vlasenko Date: Thu, 24 Jan 2008 02:28:00 +0000 Subject: halt: write wtmp entry if wtmp support is enabled --- init/halt.c | 43 ++++++++++++++++++++++++++++++++++++++----- 1 file changed, 38 insertions(+), 5 deletions(-) (limited to 'init/halt.c') diff --git a/init/halt.c b/init/halt.c index d9f8b1afc..c50d8af63 100644 --- a/init/halt.c +++ b/init/halt.c @@ -10,6 +10,11 @@ #include "libbb.h" #include +#if ENABLE_FEATURE_WTMP +#include +#include +#endif + int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int halt_main(int argc, char **argv) { @@ -26,18 +31,46 @@ RB_POWERDOWN, #endif RB_AUTOBOOT }; - static const int signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; + static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; char *delay; int which, flags, rc = 1; +#if ENABLE_FEATURE_WTMP + struct utmp utmp; + struct timeval tv; + struct utsname uts; +#endif /* Figure out which applet we're running */ - for (which = 0; "hpr"[which] != *applet_name; which++); + for (which = 0; "hpr"[which] != *applet_name; which++) + continue; /* Parse and handle arguments */ - flags = getopt32(argv, "d:nf", &delay); - if (flags & 1) sleep(xatou(delay)); - if (!(flags & 2)) sync(); + flags = getopt32(argv, "d:nfw", &delay); + if (flags & 1) + sleep(xatou(delay)); + +#if ENABLE_FEATURE_WTMP + if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) { + close(creat(bb_path_wtmp_file, 0664)); + } + memset(&utmp, 0, sizeof(utmp)); + gettimeofday(&tv, NULL); + utmp.ut_tv.tv_sec = tv.tv_sec; + utmp.ut_tv.tv_usec = tv.tv_usec; + safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE); + utmp.ut_type = RUN_LVL; + safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id)); + safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE); + if (uname(&uts) == 0) + safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host)); + updwtmp(bb_path_wtmp_file, &utmp); +#endif /* !ENABLE_FEATURE_WTMP */ + + if (flags & 8) /* -w */ + return 0; + if (!(flags & 2)) /* no -n */ + sync(); /* Perform action. */ if (ENABLE_INIT && !(flags & 4)) { -- cgit v1.2.3