mirror of
https://github.com/tbsdtv/linux_media.git
synced 2025-07-23 20:51:03 +02:00
9b89a03551
XArray marks are like the radix tree tags, only slightly more strongly typed. They are renamed in order to distinguish them from tagged pointers. This commit adds the basic get/set/clear operations. Signed-off-by: Matthew Wilcox <willy@infradead.org>
122 lines
2.9 KiB
C
122 lines
2.9 KiB
C
// SPDX-License-Identifier: GPL-2.0+
|
|
/*
|
|
* test_xarray.c: Test the XArray API
|
|
* Copyright (c) 2017-2018 Microsoft Corporation
|
|
* Author: Matthew Wilcox <willy@infradead.org>
|
|
*/
|
|
|
|
#include <linux/xarray.h>
|
|
#include <linux/module.h>
|
|
|
|
static unsigned int tests_run;
|
|
static unsigned int tests_passed;
|
|
|
|
#ifndef XA_DEBUG
|
|
# ifdef __KERNEL__
|
|
void xa_dump(const struct xarray *xa) { }
|
|
# endif
|
|
#undef XA_BUG_ON
|
|
#define XA_BUG_ON(xa, x) do { \
|
|
tests_run++; \
|
|
if (x) { \
|
|
printk("BUG at %s:%d\n", __func__, __LINE__); \
|
|
xa_dump(xa); \
|
|
dump_stack(); \
|
|
} else { \
|
|
tests_passed++; \
|
|
} \
|
|
} while (0)
|
|
#endif
|
|
|
|
static void *xa_store_index(struct xarray *xa, unsigned long index, gfp_t gfp)
|
|
{
|
|
radix_tree_insert(xa, index, xa_mk_value(index));
|
|
return NULL;
|
|
}
|
|
|
|
static void xa_erase_index(struct xarray *xa, unsigned long index)
|
|
{
|
|
radix_tree_delete(xa, index);
|
|
}
|
|
|
|
static noinline void check_xa_load(struct xarray *xa)
|
|
{
|
|
unsigned long i, j;
|
|
|
|
for (i = 0; i < 1024; i++) {
|
|
for (j = 0; j < 1024; j++) {
|
|
void *entry = xa_load(xa, j);
|
|
if (j < i)
|
|
XA_BUG_ON(xa, xa_to_value(entry) != j);
|
|
else
|
|
XA_BUG_ON(xa, entry);
|
|
}
|
|
XA_BUG_ON(xa, xa_store_index(xa, i, GFP_KERNEL) != NULL);
|
|
}
|
|
|
|
for (i = 0; i < 1024; i++) {
|
|
for (j = 0; j < 1024; j++) {
|
|
void *entry = xa_load(xa, j);
|
|
if (j >= i)
|
|
XA_BUG_ON(xa, xa_to_value(entry) != j);
|
|
else
|
|
XA_BUG_ON(xa, entry);
|
|
}
|
|
xa_erase_index(xa, i);
|
|
}
|
|
XA_BUG_ON(xa, !xa_empty(xa));
|
|
}
|
|
|
|
static noinline void check_xa_mark_1(struct xarray *xa, unsigned long index)
|
|
{
|
|
/* NULL elements have no marks set */
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_0));
|
|
xa_set_mark(xa, index, XA_MARK_0);
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_0));
|
|
|
|
/* Storing a pointer will not make a mark appear */
|
|
XA_BUG_ON(xa, xa_store_index(xa, index, GFP_KERNEL) != NULL);
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_0));
|
|
xa_set_mark(xa, index, XA_MARK_0);
|
|
XA_BUG_ON(xa, !xa_get_mark(xa, index, XA_MARK_0));
|
|
|
|
/* Setting one mark will not set another mark */
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index + 1, XA_MARK_0));
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_1));
|
|
|
|
/* Storing NULL clears marks, and they can't be set again */
|
|
xa_erase_index(xa, index);
|
|
XA_BUG_ON(xa, !xa_empty(xa));
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_0));
|
|
xa_set_mark(xa, index, XA_MARK_0);
|
|
XA_BUG_ON(xa, xa_get_mark(xa, index, XA_MARK_0));
|
|
}
|
|
|
|
static noinline void check_xa_mark(struct xarray *xa)
|
|
{
|
|
unsigned long index;
|
|
|
|
for (index = 0; index < 16384; index += 4)
|
|
check_xa_mark_1(xa, index);
|
|
}
|
|
|
|
static RADIX_TREE(array, GFP_KERNEL);
|
|
|
|
static int xarray_checks(void)
|
|
{
|
|
check_xa_load(&array);
|
|
check_xa_mark(&array);
|
|
|
|
printk("XArray: %u of %u tests passed\n", tests_passed, tests_run);
|
|
return (tests_run == tests_passed) ? 0 : -EINVAL;
|
|
}
|
|
|
|
static void xarray_exit(void)
|
|
{
|
|
}
|
|
|
|
module_init(xarray_checks);
|
|
module_exit(xarray_exit);
|
|
MODULE_AUTHOR("Matthew Wilcox <willy@infradead.org>");
|
|
MODULE_LICENSE("GPL");
|