mirror of
https://github.com/torvalds/linux.git
synced 2026-05-30 18:13:41 +02:00
selftests: revocable: Add kselftest cases
Add kselftest cases for the revocable API. The test consists of three parts: - A kernel module (revocable_test.ko) that creates a debugfs interface with `/provider` and `/consumer` files. - A user-space C program (revocable_test) that uses the kselftest harness to interact with the debugfs files. - An orchestrating shell script (test-revocable.sh) that loads the module, runs the C program, and unloads the module. The test cases cover the following scenarios: - Basic: Verifies that a consumer can successfully access the resource provided via the provider. - Revocation: Verifies that after the provider revokes the resource, the consumer correctly receives a NULL pointer on a subsequent access. - Try Access Macro: Same as "Revocation" but uses the REVOCABLE_TRY_ACCESS_WITH() and REVOCABLE_TRY_ACCESS_SCOPED(). Signed-off-by: Tzung-Bi Shih <tzungbi@kernel.org> Link: https://patch.msgid.link/20260116080235.350305-4-tzungbi@kernel.org Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
This commit is contained in:
parent
cd7693419b
commit
9d4502fef0
|
|
@ -22375,6 +22375,7 @@ S: Maintained
|
|||
F: drivers/base/revocable.c
|
||||
F: drivers/base/revocable_test.c
|
||||
F: include/linux/revocable.h
|
||||
F: tools/testing/selftests/drivers/base/revocable/
|
||||
|
||||
RFKILL
|
||||
M: Johannes Berg <johannes@sipsolutions.net>
|
||||
|
|
|
|||
|
|
@ -17,6 +17,7 @@ TARGETS += damon
|
|||
TARGETS += devices/error_logs
|
||||
TARGETS += devices/probe
|
||||
TARGETS += dmabuf-heaps
|
||||
TARGETS += drivers/base/revocable
|
||||
TARGETS += drivers/dma-buf
|
||||
TARGETS += drivers/ntsync
|
||||
TARGETS += drivers/s390x/uvdevice
|
||||
|
|
|
|||
7
tools/testing/selftests/drivers/base/revocable/Makefile
Normal file
7
tools/testing/selftests/drivers/base/revocable/Makefile
Normal file
|
|
@ -0,0 +1,7 @@
|
|||
# SPDX-License-Identifier: GPL-2.0
|
||||
|
||||
TEST_GEN_MODS_DIR := test_modules
|
||||
TEST_GEN_PROGS_EXTENDED := revocable_test
|
||||
TEST_PROGS := test-revocable.sh
|
||||
|
||||
include ../../../lib.mk
|
||||
136
tools/testing/selftests/drivers/base/revocable/revocable_test.c
Normal file
136
tools/testing/selftests/drivers/base/revocable/revocable_test.c
Normal file
|
|
@ -0,0 +1,136 @@
|
|||
// SPDX-License-Identifier: GPL-2.0
|
||||
/*
|
||||
* Copyright 2026 Google LLC
|
||||
*
|
||||
* A selftest for the revocable API.
|
||||
*
|
||||
* The test cases cover the following scenarios:
|
||||
*
|
||||
* - Basic: Verifies that a consumer can successfully access the resource
|
||||
* provided via the provider.
|
||||
*
|
||||
* - Revocation: Verifies that after the provider revokes the resource,
|
||||
* the consumer correctly receives a NULL pointer on a subsequent access.
|
||||
*
|
||||
* - Try Access Macro: Same as "Revocation" but uses the
|
||||
* REVOCABLE_TRY_ACCESS_WITH() and REVOCABLE_TRY_ACCESS_SCOPED().
|
||||
*/
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "../../../kselftest_harness.h"
|
||||
|
||||
#define DEBUGFS_PATH "/sys/kernel/debug/revocable_test"
|
||||
#define TEST_CMD_RESOURCE_GONE "resource_gone"
|
||||
#define TEST_DATA "12345678"
|
||||
#define TEST_MAGIC_OFFSET 0x1234
|
||||
#define TEST_MAGIC_OFFSET2 0x5678
|
||||
|
||||
FIXTURE(revocable_fixture) {
|
||||
int pfd;
|
||||
int cfd;
|
||||
};
|
||||
|
||||
FIXTURE_SETUP(revocable_fixture) {
|
||||
int ret;
|
||||
|
||||
self->pfd = open(DEBUGFS_PATH "/provider", O_WRONLY);
|
||||
ASSERT_NE(-1, self->pfd)
|
||||
TH_LOG("failed to open provider fd");
|
||||
|
||||
ret = write(self->pfd, TEST_DATA, strlen(TEST_DATA));
|
||||
ASSERT_NE(-1, ret) {
|
||||
close(self->pfd);
|
||||
TH_LOG("failed to write test data");
|
||||
}
|
||||
|
||||
self->cfd = open(DEBUGFS_PATH "/consumer", O_RDONLY);
|
||||
ASSERT_NE(-1, self->cfd)
|
||||
TH_LOG("failed to open consumer fd");
|
||||
}
|
||||
|
||||
FIXTURE_TEARDOWN(revocable_fixture) {
|
||||
close(self->cfd);
|
||||
close(self->pfd);
|
||||
}
|
||||
|
||||
/*
|
||||
* ASSERT_* is only available in TEST or TEST_F block. Use
|
||||
* macro for the helper.
|
||||
*/
|
||||
#define READ_TEST_DATA(_fd, _offset, _data, _msg) \
|
||||
do { \
|
||||
int ret; \
|
||||
\
|
||||
ret = lseek(_fd, _offset, SEEK_SET); \
|
||||
ASSERT_NE(-1, ret) \
|
||||
TH_LOG("failed to lseek"); \
|
||||
\
|
||||
ret = read(_fd, _data, sizeof(_data) - 1); \
|
||||
ASSERT_NE(-1, ret) \
|
||||
TH_LOG(_msg); \
|
||||
data[ret] = '\0'; \
|
||||
} while (0)
|
||||
|
||||
TEST_F(revocable_fixture, basic) {
|
||||
char data[16];
|
||||
|
||||
READ_TEST_DATA(self->cfd, 0, data, "failed to read test data");
|
||||
EXPECT_STREQ(TEST_DATA, data);
|
||||
}
|
||||
|
||||
TEST_F(revocable_fixture, revocation) {
|
||||
char data[16];
|
||||
int ret;
|
||||
|
||||
READ_TEST_DATA(self->cfd, 0, data, "failed to read test data");
|
||||
EXPECT_STREQ(TEST_DATA, data);
|
||||
|
||||
ret = write(self->pfd, TEST_CMD_RESOURCE_GONE,
|
||||
strlen(TEST_CMD_RESOURCE_GONE));
|
||||
ASSERT_NE(-1, ret)
|
||||
TH_LOG("failed to write resource gone cmd");
|
||||
|
||||
READ_TEST_DATA(self->cfd, 0, data,
|
||||
"failed to read test data after resource gone");
|
||||
EXPECT_STREQ("(null)", data);
|
||||
}
|
||||
|
||||
TEST_F(revocable_fixture, try_access_macro) {
|
||||
char data[16];
|
||||
int ret;
|
||||
|
||||
READ_TEST_DATA(self->cfd, TEST_MAGIC_OFFSET, data,
|
||||
"failed to read test data");
|
||||
EXPECT_STREQ(TEST_DATA, data);
|
||||
|
||||
ret = write(self->pfd, TEST_CMD_RESOURCE_GONE,
|
||||
strlen(TEST_CMD_RESOURCE_GONE));
|
||||
ASSERT_NE(-1, ret)
|
||||
TH_LOG("failed to write resource gone cmd");
|
||||
|
||||
READ_TEST_DATA(self->cfd, TEST_MAGIC_OFFSET, data,
|
||||
"failed to read test data after resource gone");
|
||||
EXPECT_STREQ("(null)", data);
|
||||
}
|
||||
|
||||
TEST_F(revocable_fixture, try_access_macro2) {
|
||||
char data[16];
|
||||
int ret;
|
||||
|
||||
READ_TEST_DATA(self->cfd, TEST_MAGIC_OFFSET2, data,
|
||||
"failed to read test data");
|
||||
EXPECT_STREQ(TEST_DATA, data);
|
||||
|
||||
ret = write(self->pfd, TEST_CMD_RESOURCE_GONE,
|
||||
strlen(TEST_CMD_RESOURCE_GONE));
|
||||
ASSERT_NE(-1, ret)
|
||||
TH_LOG("failed to write resource gone cmd");
|
||||
|
||||
READ_TEST_DATA(self->cfd, TEST_MAGIC_OFFSET2, data,
|
||||
"failed to read test data after resource gone");
|
||||
EXPECT_STREQ("(null)", data);
|
||||
}
|
||||
|
||||
TEST_HARNESS_MAIN
|
||||
39
tools/testing/selftests/drivers/base/revocable/test-revocable.sh
Executable file
39
tools/testing/selftests/drivers/base/revocable/test-revocable.sh
Executable file
|
|
@ -0,0 +1,39 @@
|
|||
#!/bin/bash
|
||||
# SPDX-License-Identifier: GPL-2.0
|
||||
|
||||
mod_name="revocable_test"
|
||||
ksft_fail=1
|
||||
ksft_skip=4
|
||||
|
||||
if [ "$(id -u)" -ne 0 ]; then
|
||||
echo "$0: Must be run as root"
|
||||
exit "$ksft_skip"
|
||||
fi
|
||||
|
||||
if ! which insmod > /dev/null 2>&1; then
|
||||
echo "$0: Need insmod"
|
||||
exit "$ksft_skip"
|
||||
fi
|
||||
|
||||
if ! which rmmod > /dev/null 2>&1; then
|
||||
echo "$0: Need rmmod"
|
||||
exit "$ksft_skip"
|
||||
fi
|
||||
|
||||
insmod test_modules/"${mod_name}".ko
|
||||
|
||||
if [ ! -d /sys/kernel/debug/revocable_test/ ]; then
|
||||
mount -t debugfs none /sys/kernel/debug/
|
||||
|
||||
if [ ! -d /sys/kernel/debug/revocable_test/ ]; then
|
||||
echo "$0: Error mounting debugfs"
|
||||
exit "$ksft_fail"
|
||||
fi
|
||||
fi
|
||||
|
||||
./revocable_test
|
||||
ret=$?
|
||||
|
||||
rmmod "${mod_name}"
|
||||
|
||||
exit "${ret}"
|
||||
|
|
@ -0,0 +1,10 @@
|
|||
TESTMODS_DIR := $(realpath $(dir $(abspath $(lastword $(MAKEFILE_LIST)))))
|
||||
KDIR ?= /lib/modules/$(shell uname -r)/build
|
||||
|
||||
obj-m += revocable_test.o
|
||||
|
||||
all:
|
||||
$(Q)$(MAKE) -C $(KDIR) M=$(TESTMODS_DIR)
|
||||
|
||||
clean:
|
||||
$(Q)$(MAKE) -C $(KDIR) M=$(TESTMODS_DIR) clean
|
||||
|
|
@ -0,0 +1,195 @@
|
|||
// SPDX-License-Identifier: GPL-2.0
|
||||
/*
|
||||
* Copyright 2026 Google LLC
|
||||
*
|
||||
* A kernel module for testing the revocable API.
|
||||
*/
|
||||
|
||||
#include <linux/debugfs.h>
|
||||
#include <linux/module.h>
|
||||
#include <linux/revocable.h>
|
||||
#include <linux/slab.h>
|
||||
|
||||
#define TEST_CMD_RESOURCE_GONE "resource_gone"
|
||||
#define TEST_MAGIC_OFFSET 0x1234
|
||||
#define TEST_MAGIC_OFFSET2 0x5678
|
||||
|
||||
static struct dentry *debugfs_dir;
|
||||
|
||||
struct revocable_test_provider_priv {
|
||||
struct revocable_provider *rp;
|
||||
struct dentry *dentry;
|
||||
char res[16];
|
||||
};
|
||||
|
||||
static int revocable_test_consumer_open(struct inode *inode, struct file *filp)
|
||||
{
|
||||
struct revocable *rev;
|
||||
struct revocable_provider *rp = inode->i_private;
|
||||
|
||||
rev = revocable_alloc(rp);
|
||||
if (!rev)
|
||||
return -ENOMEM;
|
||||
filp->private_data = rev;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int revocable_test_consumer_release(struct inode *inode,
|
||||
struct file *filp)
|
||||
{
|
||||
struct revocable *rev = filp->private_data;
|
||||
|
||||
revocable_free(rev);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static ssize_t revocable_test_consumer_read(struct file *filp,
|
||||
char __user *buf,
|
||||
size_t count, loff_t *offset)
|
||||
{
|
||||
char *res;
|
||||
char data[16];
|
||||
size_t len;
|
||||
struct revocable *rev = filp->private_data;
|
||||
|
||||
switch (*offset) {
|
||||
case 0:
|
||||
res = revocable_try_access(rev);
|
||||
snprintf(data, sizeof(data), "%s", res ?: "(null)");
|
||||
revocable_withdraw_access(rev);
|
||||
break;
|
||||
case TEST_MAGIC_OFFSET:
|
||||
{
|
||||
REVOCABLE_TRY_ACCESS_WITH(rev, res);
|
||||
snprintf(data, sizeof(data), "%s", res ?: "(null)");
|
||||
}
|
||||
break;
|
||||
case TEST_MAGIC_OFFSET2:
|
||||
REVOCABLE_TRY_ACCESS_SCOPED(rev, res)
|
||||
snprintf(data, sizeof(data), "%s", res ?: "(null)");
|
||||
break;
|
||||
default:
|
||||
return 0;
|
||||
}
|
||||
|
||||
len = min_t(size_t, strlen(data), count);
|
||||
if (copy_to_user(buf, data, len))
|
||||
return -EFAULT;
|
||||
|
||||
*offset = len;
|
||||
return len;
|
||||
}
|
||||
|
||||
static const struct file_operations revocable_test_consumer_fops = {
|
||||
.open = revocable_test_consumer_open,
|
||||
.release = revocable_test_consumer_release,
|
||||
.read = revocable_test_consumer_read,
|
||||
.llseek = default_llseek,
|
||||
};
|
||||
|
||||
static int revocable_test_provider_open(struct inode *inode, struct file *filp)
|
||||
{
|
||||
struct revocable_test_provider_priv *priv;
|
||||
|
||||
priv = kzalloc(sizeof(*priv), GFP_KERNEL);
|
||||
if (!priv)
|
||||
return -ENOMEM;
|
||||
filp->private_data = priv;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int revocable_test_provider_release(struct inode *inode,
|
||||
struct file *filp)
|
||||
{
|
||||
struct revocable_test_provider_priv *priv = filp->private_data;
|
||||
|
||||
debugfs_remove(priv->dentry);
|
||||
if (priv->rp)
|
||||
revocable_provider_revoke(priv->rp);
|
||||
kfree(priv);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static ssize_t revocable_test_provider_write(struct file *filp,
|
||||
const char __user *buf,
|
||||
size_t count, loff_t *offset)
|
||||
{
|
||||
size_t copied;
|
||||
char data[64];
|
||||
struct revocable_test_provider_priv *priv = filp->private_data;
|
||||
|
||||
copied = strncpy_from_user(data, buf, sizeof(data));
|
||||
if (copied < 0)
|
||||
return copied;
|
||||
if (copied == sizeof(data))
|
||||
data[sizeof(data) - 1] = '\0';
|
||||
|
||||
/*
|
||||
* Note: The test can't just close the FD for signaling the
|
||||
* resource gone. Subsequent file operations on the opening
|
||||
* FD of debugfs return -EIO after calling debugfs_remove().
|
||||
* See also debugfs_file_get().
|
||||
*
|
||||
* Here is a side command channel for signaling the resource
|
||||
* gone.
|
||||
*/
|
||||
if (!strcmp(data, TEST_CMD_RESOURCE_GONE)) {
|
||||
revocable_provider_revoke(priv->rp);
|
||||
priv->rp = NULL;
|
||||
} else {
|
||||
if (priv->res[0] != '\0')
|
||||
return 0;
|
||||
|
||||
strscpy(priv->res, data);
|
||||
|
||||
priv->rp = revocable_provider_alloc(&priv->res);
|
||||
if (!priv->rp)
|
||||
return -ENOMEM;
|
||||
|
||||
priv->dentry = debugfs_create_file("consumer", 0400,
|
||||
debugfs_dir, priv->rp,
|
||||
&revocable_test_consumer_fops);
|
||||
if (!priv->dentry) {
|
||||
revocable_provider_revoke(priv->rp);
|
||||
return -ENOMEM;
|
||||
}
|
||||
}
|
||||
|
||||
return copied;
|
||||
}
|
||||
|
||||
static const struct file_operations revocable_test_provider_fops = {
|
||||
.open = revocable_test_provider_open,
|
||||
.release = revocable_test_provider_release,
|
||||
.write = revocable_test_provider_write,
|
||||
};
|
||||
|
||||
static int __init revocable_test_init(void)
|
||||
{
|
||||
debugfs_dir = debugfs_create_dir("revocable_test", NULL);
|
||||
if (!debugfs_dir)
|
||||
return -ENOMEM;
|
||||
|
||||
if (!debugfs_create_file("provider", 0200, debugfs_dir, NULL,
|
||||
&revocable_test_provider_fops)) {
|
||||
debugfs_remove_recursive(debugfs_dir);
|
||||
return -ENOMEM;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void __exit revocable_test_exit(void)
|
||||
{
|
||||
debugfs_remove_recursive(debugfs_dir);
|
||||
}
|
||||
|
||||
module_init(revocable_test_init);
|
||||
module_exit(revocable_test_exit);
|
||||
|
||||
MODULE_LICENSE("GPL");
|
||||
MODULE_AUTHOR("Tzung-Bi Shih <tzungbi@kernel.org>");
|
||||
MODULE_DESCRIPTION("Revocable Kselftest");
|
||||
Loading…
Reference in New Issue
Block a user