SymbX/systemd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: extend the unit test test-notify-recv.c

Browse Source
This commit is contained in:
ZIHCO
2025-04-07 05:28:17 +09:00
committed by Yu Watanabe
parent ae7082d836
commit cc9229e42d
1 changed files with 29 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/test-notify-recv.c
+29 -2
View File
@@ -5,10 +5,12 @@
#include "sd-daemon.h"
#include "event-util.h"
#include "fd-util.h"
#include "notify-recv.h"
#include "path-util.h"
#include "process-util.h"
#include "rm-rf.h"
#include "socket-util.h"
#include "tests.h"
#include "tmpfile-util.h"
@@ -26,12 +28,18 @@ static void context_done(Context *c) {
static int on_recv(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
Context *c = ASSERT_PTR(userdata);
_cleanup_(fdset_free_asyncp) FDSet *fds = NULL;
_cleanup_(pidref_done) PidRef sender = PIDREF_NULL;
_cleanup_strv_free_ char **l = NULL;
ASSERT_OK(notify_recv_strv(fd, &l, /* ret_ucred= */ NULL, &sender));
struct ucred ucred;
ASSERT_OK(notify_recv_with_fds_strv(fd, &l, &ucred, &sender, &fds));
ASSERT_TRUE(pidref_equal(&c->pidref, &sender));
ASSERT_EQ(ucred.gid, getgid());
ASSERT_EQ(ucred.pid, c->pidref.pid);
ASSERT_EQ(ucred.uid, getuid());
_cleanup_free_ char *joined = strv_join(l, ", ");
ASSERT_NOT_NULL(joined);
log_info("Received message: %s", joined);
@@ -40,10 +48,20 @@ static int on_recv(sd_event_source *s, int fd, uint32_t revents, void *userdata)
ASSERT_STREQ(l[0], "FIRST_MESSAGE=1");
ASSERT_NULL(l[1]);
ASSERT_EQ(c->data, 0u);
ASSERT_NULL(fds);
} else if (strv_contains(l, "SECOND_MESSAGE=1")) {
ASSERT_STREQ(l[0], "SECOND_MESSAGE=1");
ASSERT_STREQ(l[1], "ADDITIONAL_DATA=hoge");
ASSERT_EQ(c->data, 1u);
ASSERT_NOT_NULL(fds);
ASSERT_EQ(fdset_size(fds), 2u);
int i;
FDSET_FOREACH(i, fds) {
_cleanup_free_ char *path = NULL;
ASSERT_OK(fd_get_path(i, &path));
ASSERT_TRUE(STR_IN_SET(path, "/tmp", "/dev/null"));
}
}
c->data++;
@@ -80,7 +98,16 @@ TEST(notify_socket_prepare) {
if (r == 0) {
ASSERT_OK_ERRNO(setenv("NOTIFY_SOCKET", path, /* overwrite = */ true));
ASSERT_OK_POSITIVE(sd_notify(/* unset_environment = */ false, "FIRST_MESSAGE=1"));
ASSERT_OK_POSITIVE(sd_notify(/* unset_environment = */ false, "FIRST_MESSAGE=2\nADDITIONAL_DATA=hoge"));
_cleanup_close_ int fd1 = open("/tmp", O_RDONLY|O_CLOEXEC|O_DIRECTORY);
ASSERT_OK_ERRNO(fd1);
_cleanup_close_ int fd2 = open("/dev/null", O_RDONLY|O_CLOEXEC);
ASSERT_OK_ERRNO(fd2);
ASSERT_OK_POSITIVE(
sd_pid_notify_with_fds(
0, /* unset_environment = */ false,
"FIRST_MESSAGE=2\nADDITIONAL_DATA=hoge", (int[]) { fd1, fd2 }, 2));
_exit(EXIT_SUCCESS);
}