mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2025-08-24 11:58:18 +00:00
e0f60ba041
Let's add a simple test to check the basic functionality of SO_INQ. The test does the following: 1. Create socketpair in self->fd[] 2. Enable SO_INQ 3. Send data via self->fd[0] 4. Receive data from self->fd[1] 5. Compare the SCM_INQ cmsg with ioctl(SIOCINQ) Signed-off-by: Kuniyuki Iwashima <kuniyu@google.com> Reviewed-by: Willem de Bruijn <willemb@google.com> Link: https://patch.msgid.link/20250702223606.1054680-8-kuniyu@google.com Signed-off-by: Jakub Kicinski <kuba@kernel.org>
126 lines
2.2 KiB
C
126 lines
2.2 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright 2025 Google LLC */
|
|
|
|
#include <linux/sockios.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "../../kselftest_harness.h"
|
|
|
|
#define NR_CHUNKS 100
|
|
#define MSG_LEN 256
|
|
|
|
struct scm_inq {
|
|
struct cmsghdr cmsghdr;
|
|
int inq;
|
|
};
|
|
|
|
FIXTURE(scm_inq)
|
|
{
|
|
int fd[2];
|
|
};
|
|
|
|
FIXTURE_VARIANT(scm_inq)
|
|
{
|
|
int type;
|
|
};
|
|
|
|
FIXTURE_VARIANT_ADD(scm_inq, stream)
|
|
{
|
|
.type = SOCK_STREAM,
|
|
};
|
|
|
|
FIXTURE_VARIANT_ADD(scm_inq, dgram)
|
|
{
|
|
.type = SOCK_DGRAM,
|
|
};
|
|
|
|
FIXTURE_VARIANT_ADD(scm_inq, seqpacket)
|
|
{
|
|
.type = SOCK_SEQPACKET,
|
|
};
|
|
|
|
FIXTURE_SETUP(scm_inq)
|
|
{
|
|
int err;
|
|
|
|
err = socketpair(AF_UNIX, variant->type | SOCK_NONBLOCK, 0, self->fd);
|
|
ASSERT_EQ(0, err);
|
|
}
|
|
|
|
FIXTURE_TEARDOWN(scm_inq)
|
|
{
|
|
close(self->fd[0]);
|
|
close(self->fd[1]);
|
|
}
|
|
|
|
static void send_chunks(struct __test_metadata *_metadata,
|
|
FIXTURE_DATA(scm_inq) *self)
|
|
{
|
|
char buf[MSG_LEN] = {};
|
|
int i, ret;
|
|
|
|
for (i = 0; i < NR_CHUNKS; i++) {
|
|
ret = send(self->fd[0], buf, sizeof(buf), 0);
|
|
ASSERT_EQ(sizeof(buf), ret);
|
|
}
|
|
}
|
|
|
|
static void recv_chunks(struct __test_metadata *_metadata,
|
|
FIXTURE_DATA(scm_inq) *self)
|
|
{
|
|
struct msghdr msg = {};
|
|
struct iovec iov = {};
|
|
struct scm_inq cmsg;
|
|
char buf[MSG_LEN];
|
|
int i, ret;
|
|
int inq;
|
|
|
|
msg.msg_iov = &iov;
|
|
msg.msg_iovlen = 1;
|
|
msg.msg_control = &cmsg;
|
|
msg.msg_controllen = CMSG_SPACE(sizeof(cmsg.inq));
|
|
|
|
iov.iov_base = buf;
|
|
iov.iov_len = sizeof(buf);
|
|
|
|
for (i = 0; i < NR_CHUNKS; i++) {
|
|
memset(buf, 0, sizeof(buf));
|
|
memset(&cmsg, 0, sizeof(cmsg));
|
|
|
|
ret = recvmsg(self->fd[1], &msg, 0);
|
|
ASSERT_EQ(MSG_LEN, ret);
|
|
ASSERT_NE(NULL, CMSG_FIRSTHDR(&msg));
|
|
ASSERT_EQ(CMSG_LEN(sizeof(cmsg.inq)), cmsg.cmsghdr.cmsg_len);
|
|
ASSERT_EQ(SOL_SOCKET, cmsg.cmsghdr.cmsg_level);
|
|
ASSERT_EQ(SCM_INQ, cmsg.cmsghdr.cmsg_type);
|
|
|
|
ret = ioctl(self->fd[1], SIOCINQ, &inq);
|
|
ASSERT_EQ(0, ret);
|
|
ASSERT_EQ(cmsg.inq, inq);
|
|
}
|
|
}
|
|
|
|
TEST_F(scm_inq, basic)
|
|
{
|
|
int err, inq;
|
|
|
|
err = setsockopt(self->fd[1], SOL_SOCKET, SO_INQ, &(int){1}, sizeof(int));
|
|
if (variant->type != SOCK_STREAM) {
|
|
ASSERT_EQ(-ENOPROTOOPT, -errno);
|
|
return;
|
|
}
|
|
|
|
ASSERT_EQ(0, err);
|
|
|
|
err = ioctl(self->fd[1], SIOCINQ, &inq);
|
|
ASSERT_EQ(0, err);
|
|
ASSERT_EQ(0, inq);
|
|
|
|
send_chunks(_metadata, self);
|
|
recv_chunks(_metadata, self);
|
|
}
|
|
|
|
TEST_HARNESS_MAIN
|