ask-password: refer to right binary name in wall message

[elogind.git] / src / tty-ask-password-agent.c

diff --git a/src/tty-ask-password-agent.c b/src/tty-ask-password-agent.c
index 8a9330d4acbbc09d7929df14a09270d913328ffb..2e8a92fe53fad1c1f61ab42593b8e153f3f5d97b 100644 (file)
--- a/src/tty-ask-password-agent.c
+++ b/src/tty-ask-password-agent.c
@@ -29,11 +29,13 @@
 #include <sys/inotify.h>
 #include <unistd.h>
 #include <getopt.h>
 #include <sys/inotify.h>
 #include <unistd.h>
 #include <getopt.h>

+#include <sys/signalfd.h>

 
 #include "util.h"
 #include "conf-parser.h"
 #include "utmp-wtmp.h"
 #include "socket-util.h"
 
 #include "util.h"
 #include "conf-parser.h"
 #include "utmp-wtmp.h"
 #include "socket-util.h"

+#include "ask-password-api.h"

 
 static enum {
         ACTION_LIST,
 
 static enum {
         ACTION_LIST,


@@ -200,7 +202,7 @@ finish:
         return r;
 }
 
         return r;
 }
 

-static int parse_password(const char *filename) {
+static int parse_password(const char *filename, char **wall) {

         char *socket_name = NULL, *message = NULL, *packet = NULL;
         uint64_t not_after = 0;
         unsigned pid = 0;
         char *socket_name = NULL, *message = NULL, *packet = NULL;
         uint64_t not_after = 0;
         unsigned pid = 0;


@@ -211,6 +213,7 @@ static int parse_password(const char *filename) {
                 { "NotAfter", config_parse_uint64,   &not_after,   "Ask" },
                 { "Message",  config_parse_string,   &message,     "Ask" },
                 { "PID",      config_parse_unsigned, &pid,         "Ask" },
                 { "NotAfter", config_parse_uint64,   &not_after,   "Ask" },
                 { "Message",  config_parse_string,   &message,     "Ask" },
                 { "PID",      config_parse_unsigned, &pid,         "Ask" },

+                { NULL, NULL, NULL, NULL }

         };
 
         FILE *f;
         };
 
         FILE *f;


@@ -228,7 +231,7 @@ static int parse_password(const char *filename) {
                 return -errno;
         }
 
                 return -errno;
         }
 

-        if ((r = config_parse(filename, f, NULL, items, false, NULL)) < 0) {
+        if ((r = config_parse(filename, f, NULL, items, true, NULL)) < 0) {

                 log_error("Failed to parse password file %s: %s", filename, strerror(-r));
                 goto finish;
         }
                 log_error("Failed to parse password file %s: %s", filename, strerror(-r));
                 goto finish;
         }


@@ -248,11 +251,13 @@ static int parse_password(const char *filename) {
         if (arg_action == ACTION_LIST)
                 printf("'%s' (PID %u)\n", message, pid);
         else if (arg_action == ACTION_WALL) {
         if (arg_action == ACTION_LIST)
                 printf("'%s' (PID %u)\n", message, pid);
         else if (arg_action == ACTION_WALL) {

-                char *wall;
+                char *_wall;

 
 

-                if (asprintf(&wall,
-                             "Password entry required for \'%s\' (PID %u).\r\n"
-                             "Please enter password with the systemd-tty-password-agent tool!",
+                if (asprintf(&_wall,
+                             "%s%sPassword entry required for \'%s\' (PID %u).\r\n"
+                             "Please enter password with the systemd-tty-ask-password-agent tool!",
+                             *wall ? *wall : "",
+                             *wall ? "\r\n\r\n" : "",

                              message,
                              pid) < 0) {
                         log_error("Out of memory");
                              message,
                              pid) < 0) {
                         log_error("Out of memory");


@@ -260,8 +265,8 @@ static int parse_password(const char *filename) {
                         goto finish;
                 }
 
                         goto finish;
                 }
 

-                r = utmp_wall(wall);
-                free(wall);
+                free(*wall);
+                *wall = _wall;

         } else {
                 union {
                         struct sockaddr sa;
         } else {
                 union {
                         struct sockaddr sa;


@@ -346,6 +351,7 @@ static int show_passwords(void) {
         while ((de = readdir(d))) {
                 char *p;
                 int q;
         while ((de = readdir(d))) {
                 char *p;
                 int q;

+                char *wall;

 
                 if (de->d_type != DT_REG)
                         continue;
 
                 if (de->d_type != DT_REG)
                         continue;


@@ -362,10 +368,16 @@ static int show_passwords(void) {
                         goto finish;
                 }
 
                         goto finish;
                 }
 

-                if ((q = parse_password(p)) < 0)
+                wall = NULL;
+                if ((q = parse_password(p, &wall)) < 0)

                         r = q;
 
                 free(p);
                         r = q;
 
                 free(p);

+
+                if (wall) {
+                        utmp_wall(wall);
+                        free(wall);
+                }

         }
 
 finish:
         }
 
 finish:


@@ -376,8 +388,15 @@ finish:
 }
 
 static int watch_passwords(void) {
 }
 
 static int watch_passwords(void) {

-        int notify;
-        struct pollfd pollfd;
+        enum {
+                FD_INOTIFY,
+                FD_SIGNAL,
+                _FD_MAX
+        };
+
+        int notify = -1, signal_fd = -1;
+        struct pollfd pollfd[_FD_MAX];
+        sigset_t mask;

         int r;
 
         mkdir_p("/dev/.systemd/ask-password", 0755);
         int r;
 
         mkdir_p("/dev/.systemd/ask-password", 0755);


@@ -392,15 +411,27 @@ static int watch_passwords(void) {
                 goto finish;
         }
 
                 goto finish;
         }
 

+        assert_se(sigemptyset(&mask) == 0);
+        sigset_add_many(&mask, SIGINT, SIGTERM, -1);
+        assert_se(sigprocmask(SIG_SETMASK, &mask, NULL) == 0);
+
+        if ((signal_fd = signalfd(-1, &mask, SFD_NONBLOCK|SFD_CLOEXEC)) < 0) {
+                log_error("signalfd(): %m");
+                r = -errno;
+                goto finish;
+        }
+

         zero(pollfd);
         zero(pollfd);

-        pollfd.fd = notify;
-        pollfd.events = POLLIN;
+        pollfd[FD_INOTIFY].fd = notify;
+        pollfd[FD_INOTIFY].events = POLLIN;
+        pollfd[FD_SIGNAL].fd = signal_fd;
+        pollfd[FD_SIGNAL].events = POLLIN;

 
         for (;;) {
                 if ((r = show_passwords()) < 0)
                         break;
 
 
         for (;;) {
                 if ((r = show_passwords()) < 0)
                         break;
 

-                if (poll(&pollfd, 1, -1) < 0) {
+                if (poll(pollfd, _FD_MAX, -1) < 0) {

 
                         if (errno == EINTR)
                                 continue;
 
                         if (errno == EINTR)
                                 continue;


@@ -409,8 +440,11 @@ static int watch_passwords(void) {
                         goto finish;
                 }
 
                         goto finish;
                 }
 

-                if (pollfd.revents != 0)
+                if (pollfd[FD_INOTIFY].revents != 0)

                         flush_fd(notify);
                         flush_fd(notify);

+
+                if (pollfd[FD_SIGNAL].revents != 0)
+                        break;

         }
 
         r = 0;
         }
 
         r = 0;


@@ -419,6 +453,9 @@ finish:
         if (notify >= 0)
                 close_nointr_nofail(notify);
 
         if (notify >= 0)
                 close_nointr_nofail(notify);
 

+        if (signal_fd >= 0)
+                close_nointr_nofail(signal_fd);
+

         return r;
 }
 
         return r;
 }