X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fshared%2Futmp-wtmp.c;h=32996fad36cd711fcf03a12286742bbd35de05cd;hp=342e0a9cf1aceefb7dd7313e9da6a5307a2b8781;hb=b6e2f3293d0f9a821819d96038eec2a19062151f;hpb=7fd1b19bc9e9f5574f2877936b8ac267c7706947 diff --git a/src/shared/utmp-wtmp.c b/src/shared/utmp-wtmp.c index 342e0a9cf..32996fad3 100644 --- a/src/shared/utmp-wtmp.c +++ b/src/shared/utmp-wtmp.c @@ -46,13 +46,15 @@ int utmp_get_runlevel(int *runlevel, int *previous) { * very new and does not apply to the current script being * executed. */ - if ((e = getenv("RUNLEVEL")) && e[0] > 0) { + e = getenv("RUNLEVEL"); + if (e && e[0] > 0) { *runlevel = e[0]; if (previous) { /* $PREVLEVEL seems to be an Upstart thing */ - if ((e = getenv("PREVLEVEL")) && e[0] > 0) + e = getenv("PREVLEVEL"); + if (e && e[0] > 0) *previous = e[0]; else *previous = 0; @@ -66,7 +68,8 @@ int utmp_get_runlevel(int *runlevel, int *previous) { setutxent(); - if (!(found = getutxid(&lookup))) + found = getutxid(&lookup); + if (!found) r = -errno; else { int a, b; @@ -190,7 +193,7 @@ int utmp_put_reboot(usec_t t) { return write_entry_both(&store); } -static const char *sanitize_id(const char *id) { +_pure_ static const char *sanitize_id(const char *id) { size_t l; assert(id); @@ -216,7 +219,7 @@ int utmp_put_init_process(const char *id, pid_t pid, pid_t sid, const char *line strncpy(store.ut_id, sanitize_id(id), sizeof(store.ut_id)); if (line) - strncpy(store.ut_line, path_get_file_name(line), sizeof(store.ut_line)); + strncpy(store.ut_line, basename(line), sizeof(store.ut_line)); return write_entry_both(&store); } @@ -232,7 +235,8 @@ int utmp_put_dead_process(const char *id, pid_t pid, int code, int status) { lookup.ut_type = INIT_PROCESS; /* looks for DEAD_PROCESS, LOGIN_PROCESS, USER_PROCESS, too */ strncpy(lookup.ut_id, sanitize_id(id), sizeof(lookup.ut_id)); - if (!(found = getutxid(&lookup))) + found = getutxid(&lookup); + if (!found) return 0; if (found->ut_pid != pid) @@ -264,7 +268,8 @@ int utmp_put_runlevel(int runlevel, int previous) { if (previous <= 0) { /* Find the old runlevel automatically */ - if ((r = utmp_get_runlevel(&previous, NULL)) < 0) { + r = utmp_get_runlevel(&previous, NULL); + if (r < 0) { if (r != -ESRCH) return r; @@ -343,16 +348,15 @@ static int write_to_terminal(const char *tty, const char *message) { } int utmp_wall(const char *message, bool (*match_tty)(const char *tty)) { - struct utmpx *u; + _cleanup_free_ char *text = NULL, *hn = NULL, *un = NULL, *tty = NULL; char date[FORMAT_TIMESTAMP_MAX]; - char *text = NULL, *hn = NULL, *un = NULL, *tty = NULL; + struct utmpx *u; int r; - if (!(hn = gethostname_malloc()) || - !(un = getlogname_malloc())) { - r = -ENOMEM; - goto finish; - } + hn = gethostname_malloc(); + un = getlogname_malloc(); + if (!hn || !un) + return -ENOMEM; getttyname_harder(STDIN_FILENO, &tty); @@ -363,19 +367,17 @@ int utmp_wall(const char *message, bool (*match_tty)(const char *tty)) { un, hn, tty ? " on " : "", strempty(tty), format_timestamp(date, sizeof(date), now(CLOCK_REALTIME)), - message) < 0) { - r = -ENOMEM; - goto finish; - } + message) < 0) + return -ENOMEM; setutxent(); r = 0; while ((u = getutxent())) { - int q; + _cleanup_free_ char *buf = NULL; const char *path; - char *buf = NULL; + int q; if (u->ut_type != USER_PROCESS || u->ut_user[0] == 0) continue; @@ -384,27 +386,18 @@ int utmp_wall(const char *message, bool (*match_tty)(const char *tty)) { if (path_startswith(u->ut_line, "/dev/")) path = u->ut_line; else { - if (asprintf(&buf, "/dev/%.*s", - (int) sizeof(u->ut_line), u->ut_line) < 0) { - r = -ENOMEM; - goto finish; - } + if (asprintf(&buf, "/dev/%.*s", (int) sizeof(u->ut_line), u->ut_line) < 0) + return -ENOMEM; path = buf; } - if (!match_tty || match_tty(path)) - if ((q = write_to_terminal(path, text)) < 0) + if (!match_tty || match_tty(path)) { + q = write_to_terminal(path, text); + if (q < 0) r = q; - - free(buf); + } } -finish: - free(hn); - free(un); - free(tty); - free(text); - return r; }