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:
Tzung-Bi Shih 2026-01-16 08:02:35 +00:00 committed by Greg Kroah-Hartman
parent cd7693419b
commit 9d4502fef0
7 changed files with 389 additions and 0 deletions

View File

@ -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>

View File

@ -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

View 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

View 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

View 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}"

View File

@ -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

View File

@ -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");