mirror of
https://github.com/torvalds/linux.git
synced 2026-10-23 23:51:02 +02:00
selftests/net: SO_PEERPIDFD_THREAD
Add tests for SO_PEERPIDFD_THREAD. Reviewed-by: Alexander Mikhalitsyn <aleksandr.mikhalitsyn@futurfusion.io> Signed-off-by: Andrew Halaney <andrew@amutable.com> Link: https://patch.msgid.link/20261005-work-unix-passpidfd-v4-8-350183b6e02a@amutable.com Reviewed-by: Kuniyuki Iwashima <kuniyu@google.com> Signed-off-by: Christian Brauner (Amutable) <brauner@kernel.org>
This commit is contained in:
parent
e10899b2a0
commit
02e658dffe
|
|
@ -36,6 +36,10 @@
|
|||
#define SO_PASSPIDFD_THREAD 86
|
||||
#endif
|
||||
|
||||
#ifndef SO_PEERPIDFD_THREAD
|
||||
#define SO_PEERPIDFD_THREAD 87
|
||||
#endif
|
||||
|
||||
#define CHILD_EXIT_CODE_OK 123
|
||||
|
||||
static void child_die()
|
||||
|
|
@ -1058,4 +1062,97 @@ TEST(scm_pidfd_thread_creds)
|
|||
EXPECT_EQ(msg.ids.pid, msg.thread_info.tgid);
|
||||
}
|
||||
|
||||
static void *peer_connect_thread(void *arg)
|
||||
{
|
||||
struct sock_addr *sa = arg;
|
||||
struct thread_ids ids = {
|
||||
.pid = getpid(),
|
||||
.tid = gettid(),
|
||||
};
|
||||
int fd;
|
||||
char sync;
|
||||
|
||||
fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
if (fd < 0)
|
||||
return (void *)1;
|
||||
|
||||
if (connect(fd, (struct sockaddr *)&sa->listen_addr, sa->addrlen))
|
||||
return (void *)1;
|
||||
|
||||
if (send(fd, &ids, sizeof(ids), 0) != sizeof(ids))
|
||||
return (void *)1;
|
||||
|
||||
/* stay alive until the server has looked at our pidfd */
|
||||
if (read(fd, &sync, 1) != 1)
|
||||
return (void *)1;
|
||||
|
||||
close(fd);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int peer_pidfd_info(int fd, int optname, struct pidfd_info *info)
|
||||
{
|
||||
int pidfd;
|
||||
socklen_t len = sizeof(pidfd);
|
||||
|
||||
if (getsockopt(fd, SOL_SOCKET, optname, &pidfd, &len)) {
|
||||
log_err("getsockopt(SO_PEERPIDFD*)");
|
||||
return -1;
|
||||
}
|
||||
|
||||
info->mask = PIDFD_INFO_PID;
|
||||
if (ioctl(pidfd, PIDFD_GET_INFO, info)) {
|
||||
log_err("ioctl(PIDFD_GET_INFO)");
|
||||
close(pidfd);
|
||||
return -1;
|
||||
}
|
||||
|
||||
close(pidfd);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* SO_PEERPIDFD_THREAD returns a pidfd for the peer's connecting thread. */
|
||||
TEST(so_peerpidfd_thread)
|
||||
{
|
||||
struct sock_addr sa;
|
||||
struct thread_ids ids;
|
||||
struct pidfd_info info;
|
||||
pthread_t thread;
|
||||
void *tret;
|
||||
int server, cfd;
|
||||
|
||||
server = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
ASSERT_LE(0, server);
|
||||
|
||||
fill_sockaddr(&sa, true);
|
||||
ASSERT_EQ(0, bind(server, (struct sockaddr *)&sa.listen_addr, sa.addrlen));
|
||||
ASSERT_EQ(0, listen(server, 1));
|
||||
|
||||
ASSERT_EQ(0, pthread_create(&thread, NULL, peer_connect_thread, &sa));
|
||||
|
||||
cfd = accept(server, NULL, NULL);
|
||||
ASSERT_LE(0, cfd);
|
||||
|
||||
ASSERT_EQ(sizeof(ids), recv(cfd, &ids, sizeof(ids), MSG_WAITALL));
|
||||
ASSERT_NE(ids.pid, ids.tid);
|
||||
|
||||
/* SO_PEERPIDFD refers to the peer's thread-group. */
|
||||
ASSERT_EQ(0, peer_pidfd_info(cfd, SO_PEERPIDFD, &info));
|
||||
EXPECT_EQ(ids.pid, info.pid);
|
||||
EXPECT_EQ(ids.pid, info.tgid);
|
||||
|
||||
/* SO_PEERPIDFD_THREAD refers to the connecting thread. */
|
||||
ASSERT_EQ(0, peer_pidfd_info(cfd, SO_PEERPIDFD_THREAD, &info));
|
||||
EXPECT_EQ(ids.tid, info.pid);
|
||||
EXPECT_EQ(ids.pid, info.tgid);
|
||||
|
||||
/* release the connecting thread */
|
||||
ASSERT_EQ(1, write(cfd, "x", 1));
|
||||
ASSERT_EQ(0, pthread_join(thread, &tret));
|
||||
ASSERT_EQ(NULL, tret);
|
||||
|
||||
close(cfd);
|
||||
close(server);
|
||||
}
|
||||
|
||||
TEST_HARNESS_MAIN
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user