Zbigniew Jędrzejewski-Szmek 3a5d9b
From 9d77c48a80e1cc2ad016eba1756a5ca293d51f86 Mon Sep 17 00:00:00 2001
Zbigniew Jędrzejewski-Szmek 3a5d9b
From: =?UTF-8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= <zbyszek@in.waw.pl>
Zbigniew Jędrzejewski-Szmek 3a5d9b
Date: Thu, 29 Sep 2016 16:06:02 +0200
Zbigniew Jędrzejewski-Szmek 3a5d9b
Subject: [PATCH] pid1: process zero-length notification messages again
Zbigniew Jędrzejewski-Szmek 3a5d9b
Zbigniew Jędrzejewski-Szmek 3a5d9b
This undoes 531ac2b234. I acked that patch without looking at the code
Zbigniew Jędrzejewski-Szmek 3a5d9b
carefully enough. There are two problems:
Zbigniew Jędrzejewski-Szmek 3a5d9b
- we want to process the fds anyway
Zbigniew Jędrzejewski-Szmek 3a5d9b
- in principle empty notification messages are valid, and we should
Zbigniew Jędrzejewski-Szmek 3a5d9b
  process them as usual, including logging using log_unit_debug().
Zbigniew Jędrzejewski-Szmek 3a5d9b
Zbigniew Jędrzejewski-Szmek 3a5d9b
(cherry picked from commit 8523bf7dd514a3a2c6114b7b8fb8f308b4f09fc4)
Zbigniew Jędrzejewski-Szmek 3a5d9b
---
Zbigniew Jędrzejewski-Szmek 3a5d9b
 src/core/manager.c | 15 ++++++---------
Zbigniew Jędrzejewski-Szmek 3a5d9b
 1 file changed, 6 insertions(+), 9 deletions(-)
Zbigniew Jędrzejewski-Szmek 3a5d9b
Zbigniew Jędrzejewski-Szmek 3a5d9b
diff --git a/src/core/manager.c b/src/core/manager.c
Zbigniew Jędrzejewski-Szmek 3a5d9b
index 36488b673a..85bf858992 100644
Zbigniew Jędrzejewski-Szmek 3a5d9b
--- a/src/core/manager.c
Zbigniew Jędrzejewski-Szmek 3a5d9b
+++ b/src/core/manager.c
Zbigniew Jędrzejewski-Szmek 3a5d9b
@@ -1584,13 +1584,12 @@ static int manager_dispatch_cgroups_agent_fd(sd_event_source *source, int fd, ui
Zbigniew Jędrzejewski-Szmek 3a5d9b
         return 0;
Zbigniew Jędrzejewski-Szmek 3a5d9b
 }
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
-static void manager_invoke_notify_message(Manager *m, Unit *u, pid_t pid, const char *buf, size_t n, FDSet *fds) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
+static void manager_invoke_notify_message(Manager *m, Unit *u, pid_t pid, const char *buf, FDSet *fds) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
         _cleanup_strv_free_ char **tags = NULL;
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         assert(m);
Zbigniew Jędrzejewski-Szmek 3a5d9b
         assert(u);
Zbigniew Jędrzejewski-Szmek 3a5d9b
         assert(buf);
Zbigniew Jędrzejewski-Szmek 3a5d9b
-        assert(n > 0);
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         tags = strv_split(buf, "\n\r");
Zbigniew Jędrzejewski-Szmek 3a5d9b
         if (!tags) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
@@ -1652,10 +1651,6 @@ static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t
Zbigniew Jędrzejewski-Szmek 3a5d9b
                  * example... */
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 return 0;
Zbigniew Jędrzejewski-Szmek 3a5d9b
         }
Zbigniew Jędrzejewski-Szmek 3a5d9b
-        if (n == 0) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
-                log_debug("Got zero-length notification message. Ignoring.");
Zbigniew Jędrzejewski-Szmek 3a5d9b
-                return 0;
Zbigniew Jędrzejewski-Szmek 3a5d9b
-        }
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         CMSG_FOREACH(cmsg, &msghdr) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 if (cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
@@ -1692,25 +1687,27 @@ static int manager_dispatch_notify_fd(sd_event_source *source, int fd, uint32_t
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 return 0;
Zbigniew Jędrzejewski-Szmek 3a5d9b
         }
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
+        /* The message should be a string. Here we make sure it's NUL-terminated,
Zbigniew Jędrzejewski-Szmek 3a5d9b
+         * but only the part until first NUL will be used anyway. */
Zbigniew Jędrzejewski-Szmek 3a5d9b
         buf[n] = 0;
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         /* Notify every unit that might be interested, but try
Zbigniew Jędrzejewski-Szmek 3a5d9b
          * to avoid notifying the same one multiple times. */
Zbigniew Jędrzejewski-Szmek 3a5d9b
         u1 = manager_get_unit_by_pid_cgroup(m, ucred->pid);
Zbigniew Jędrzejewski-Szmek 3a5d9b
         if (u1) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
-                manager_invoke_notify_message(m, u1, ucred->pid, buf, n, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
+                manager_invoke_notify_message(m, u1, ucred->pid, buf, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 found = true;
Zbigniew Jędrzejewski-Szmek 3a5d9b
         }
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         u2 = hashmap_get(m->watch_pids1, PID_TO_PTR(ucred->pid));
Zbigniew Jędrzejewski-Szmek 3a5d9b
         if (u2 && u2 != u1) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
-                manager_invoke_notify_message(m, u2, ucred->pid, buf, n, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
+                manager_invoke_notify_message(m, u2, ucred->pid, buf, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 found = true;
Zbigniew Jędrzejewski-Szmek 3a5d9b
         }
Zbigniew Jędrzejewski-Szmek 3a5d9b
 
Zbigniew Jędrzejewski-Szmek 3a5d9b
         u3 = hashmap_get(m->watch_pids2, PID_TO_PTR(ucred->pid));
Zbigniew Jędrzejewski-Szmek 3a5d9b
         if (u3 && u3 != u2 && u3 != u1) {
Zbigniew Jędrzejewski-Szmek 3a5d9b
-                manager_invoke_notify_message(m, u3, ucred->pid, buf, n, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
+                manager_invoke_notify_message(m, u3, ucred->pid, buf, fds);
Zbigniew Jędrzejewski-Szmek 3a5d9b
                 found = true;
Zbigniew Jędrzejewski-Szmek 3a5d9b
         }
Zbigniew Jędrzejewski-Szmek 3a5d9b