forked from rrcarlosr/Jetpack
359 lines
9.0 KiB
C
359 lines
9.0 KiB
C
/*
|
|
* Test cases for printf facility.
|
|
*/
|
|
|
|
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
|
|
|
|
#include <linux/bitmap.h>
|
|
#include <linux/init.h>
|
|
#include <linux/kernel.h>
|
|
#include <linux/module.h>
|
|
#include <linux/printk.h>
|
|
#include <linux/slab.h>
|
|
#include <linux/string.h>
|
|
|
|
static unsigned total_tests __initdata;
|
|
static unsigned failed_tests __initdata;
|
|
|
|
static char pbl_buffer[PAGE_SIZE] __initdata;
|
|
|
|
|
|
static bool __init
|
|
__check_eq_uint(const char *srcfile, unsigned int line,
|
|
const unsigned int exp_uint, unsigned int x)
|
|
{
|
|
if (exp_uint != x) {
|
|
pr_warn("[%s:%u] expected %u, got %u\n",
|
|
srcfile, line, exp_uint, x);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
static bool __init
|
|
__check_eq_bitmap(const char *srcfile, unsigned int line,
|
|
const unsigned long *exp_bmap, unsigned int exp_nbits,
|
|
const unsigned long *bmap, unsigned int nbits)
|
|
{
|
|
if (exp_nbits != nbits) {
|
|
pr_warn("[%s:%u] bitmap length mismatch: expected %u, got %u\n",
|
|
srcfile, line, exp_nbits, nbits);
|
|
return false;
|
|
}
|
|
|
|
if (!bitmap_equal(exp_bmap, bmap, nbits)) {
|
|
pr_warn("[%s:%u] bitmaps contents differ: expected \"%*pbl\", got \"%*pbl\"\n",
|
|
srcfile, line,
|
|
exp_nbits, exp_bmap, nbits, bmap);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool __init
|
|
__check_eq_pbl(const char *srcfile, unsigned int line,
|
|
const char *expected_pbl,
|
|
const unsigned long *bitmap, unsigned int nbits)
|
|
{
|
|
snprintf(pbl_buffer, sizeof(pbl_buffer), "%*pbl", nbits, bitmap);
|
|
if (strcmp(expected_pbl, pbl_buffer)) {
|
|
pr_warn("[%s:%u] expected \"%s\", got \"%s\"\n",
|
|
srcfile, line,
|
|
expected_pbl, pbl_buffer);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool __init
|
|
__check_eq_u32_array(const char *srcfile, unsigned int line,
|
|
const u32 *exp_arr, unsigned int exp_len,
|
|
const u32 *arr, unsigned int len)
|
|
{
|
|
if (exp_len != len) {
|
|
pr_warn("[%s:%u] array length differ: expected %u, got %u\n",
|
|
srcfile, line,
|
|
exp_len, len);
|
|
return false;
|
|
}
|
|
|
|
if (memcmp(exp_arr, arr, len*sizeof(*arr))) {
|
|
pr_warn("[%s:%u] array contents differ\n", srcfile, line);
|
|
print_hex_dump(KERN_WARNING, " exp: ", DUMP_PREFIX_OFFSET,
|
|
32, 4, exp_arr, exp_len*sizeof(*exp_arr), false);
|
|
print_hex_dump(KERN_WARNING, " got: ", DUMP_PREFIX_OFFSET,
|
|
32, 4, arr, len*sizeof(*arr), false);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
#define __expect_eq(suffix, ...) \
|
|
({ \
|
|
int result = 0; \
|
|
total_tests++; \
|
|
if (!__check_eq_ ## suffix(__FILE__, __LINE__, \
|
|
##__VA_ARGS__)) { \
|
|
failed_tests++; \
|
|
result = 1; \
|
|
} \
|
|
result; \
|
|
})
|
|
|
|
#define expect_eq_uint(...) __expect_eq(uint, ##__VA_ARGS__)
|
|
#define expect_eq_bitmap(...) __expect_eq(bitmap, ##__VA_ARGS__)
|
|
#define expect_eq_pbl(...) __expect_eq(pbl, ##__VA_ARGS__)
|
|
#define expect_eq_u32_array(...) __expect_eq(u32_array, ##__VA_ARGS__)
|
|
|
|
static void __init test_zero_fill_copy(void)
|
|
{
|
|
DECLARE_BITMAP(bmap1, 1024);
|
|
DECLARE_BITMAP(bmap2, 1024);
|
|
|
|
bitmap_zero(bmap1, 1024);
|
|
bitmap_zero(bmap2, 1024);
|
|
|
|
/* single-word bitmaps */
|
|
expect_eq_pbl("", bmap1, 23);
|
|
|
|
bitmap_fill(bmap1, 19);
|
|
expect_eq_pbl("0-18", bmap1, 1024);
|
|
|
|
bitmap_copy(bmap2, bmap1, 23);
|
|
expect_eq_pbl("0-18", bmap2, 1024);
|
|
|
|
bitmap_fill(bmap2, 23);
|
|
expect_eq_pbl("0-22", bmap2, 1024);
|
|
|
|
bitmap_copy(bmap2, bmap1, 23);
|
|
expect_eq_pbl("0-18", bmap2, 1024);
|
|
|
|
bitmap_zero(bmap1, 23);
|
|
expect_eq_pbl("", bmap1, 1024);
|
|
|
|
/* multi-word bitmaps */
|
|
bitmap_zero(bmap1, 1024);
|
|
expect_eq_pbl("", bmap1, 1024);
|
|
|
|
bitmap_fill(bmap1, 109);
|
|
expect_eq_pbl("0-108", bmap1, 1024);
|
|
|
|
bitmap_copy(bmap2, bmap1, 1024);
|
|
expect_eq_pbl("0-108", bmap2, 1024);
|
|
|
|
bitmap_fill(bmap2, 1024);
|
|
expect_eq_pbl("0-1023", bmap2, 1024);
|
|
|
|
bitmap_copy(bmap2, bmap1, 1024);
|
|
expect_eq_pbl("0-108", bmap2, 1024);
|
|
|
|
/* the following tests assume a 32- or 64-bit arch (even 128b
|
|
* if we care)
|
|
*/
|
|
|
|
bitmap_fill(bmap2, 1024);
|
|
bitmap_copy(bmap2, bmap1, 109); /* ... but 0-padded til word length */
|
|
expect_eq_pbl("0-108,128-1023", bmap2, 1024);
|
|
|
|
bitmap_fill(bmap2, 1024);
|
|
bitmap_copy(bmap2, bmap1, 97); /* ... but aligned on word length */
|
|
expect_eq_pbl("0-108,128-1023", bmap2, 1024);
|
|
|
|
bitmap_zero(bmap2, 97); /* ... but 0-padded til word length */
|
|
expect_eq_pbl("128-1023", bmap2, 1024);
|
|
}
|
|
|
|
static void __init test_bitmap_u32_array_conversions(void)
|
|
{
|
|
DECLARE_BITMAP(bmap1, 1024);
|
|
DECLARE_BITMAP(bmap2, 1024);
|
|
u32 exp_arr[32], arr[32];
|
|
unsigned nbits;
|
|
|
|
for (nbits = 0 ; nbits < 257 ; ++nbits) {
|
|
const unsigned int used_u32s = DIV_ROUND_UP(nbits, 32);
|
|
unsigned int i, rv;
|
|
|
|
bitmap_zero(bmap1, nbits);
|
|
bitmap_set(bmap1, nbits, 1024 - nbits); /* garbage */
|
|
|
|
memset(arr, 0xff, sizeof(arr));
|
|
rv = bitmap_to_u32array(arr, used_u32s, bmap1, nbits);
|
|
expect_eq_uint(nbits, rv);
|
|
|
|
memset(exp_arr, 0xff, sizeof(exp_arr));
|
|
memset(exp_arr, 0, used_u32s*sizeof(*exp_arr));
|
|
expect_eq_u32_array(exp_arr, 32, arr, 32);
|
|
|
|
bitmap_fill(bmap2, 1024);
|
|
rv = bitmap_from_u32array(bmap2, nbits, arr, used_u32s);
|
|
expect_eq_uint(nbits, rv);
|
|
expect_eq_bitmap(bmap1, 1024, bmap2, 1024);
|
|
|
|
for (i = 0 ; i < nbits ; ++i) {
|
|
/*
|
|
* test conversion bitmap -> u32[]
|
|
*/
|
|
|
|
bitmap_zero(bmap1, 1024);
|
|
__set_bit(i, bmap1);
|
|
bitmap_set(bmap1, nbits, 1024 - nbits); /* garbage */
|
|
|
|
memset(arr, 0xff, sizeof(arr));
|
|
rv = bitmap_to_u32array(arr, used_u32s, bmap1, nbits);
|
|
expect_eq_uint(nbits, rv);
|
|
|
|
/* 1st used u32 words contain expected bit set, the
|
|
* remaining words are left unchanged (0xff)
|
|
*/
|
|
memset(exp_arr, 0xff, sizeof(exp_arr));
|
|
memset(exp_arr, 0, used_u32s*sizeof(*exp_arr));
|
|
exp_arr[i/32] = (1U<<(i%32));
|
|
expect_eq_u32_array(exp_arr, 32, arr, 32);
|
|
|
|
|
|
/* same, with longer array to fill
|
|
*/
|
|
memset(arr, 0xff, sizeof(arr));
|
|
rv = bitmap_to_u32array(arr, 32, bmap1, nbits);
|
|
expect_eq_uint(nbits, rv);
|
|
|
|
/* 1st used u32 words contain expected bit set, the
|
|
* remaining words are all 0s
|
|
*/
|
|
memset(exp_arr, 0, sizeof(exp_arr));
|
|
exp_arr[i/32] = (1U<<(i%32));
|
|
expect_eq_u32_array(exp_arr, 32, arr, 32);
|
|
|
|
/*
|
|
* test conversion u32[] -> bitmap
|
|
*/
|
|
|
|
/* the 1st nbits of bmap2 are identical to
|
|
* bmap1, the remaining bits of bmap2 are left
|
|
* unchanged (all 1s)
|
|
*/
|
|
bitmap_fill(bmap2, 1024);
|
|
rv = bitmap_from_u32array(bmap2, nbits,
|
|
exp_arr, used_u32s);
|
|
expect_eq_uint(nbits, rv);
|
|
|
|
expect_eq_bitmap(bmap1, 1024, bmap2, 1024);
|
|
|
|
/* same, with more bits to fill
|
|
*/
|
|
memset(arr, 0xff, sizeof(arr)); /* garbage */
|
|
memset(arr, 0, used_u32s*sizeof(u32));
|
|
arr[i/32] = (1U<<(i%32));
|
|
|
|
bitmap_fill(bmap2, 1024);
|
|
rv = bitmap_from_u32array(bmap2, 1024, arr, used_u32s);
|
|
expect_eq_uint(used_u32s*32, rv);
|
|
|
|
/* the 1st nbits of bmap2 are identical to
|
|
* bmap1, the remaining bits of bmap2 are cleared
|
|
*/
|
|
bitmap_zero(bmap1, 1024);
|
|
__set_bit(i, bmap1);
|
|
expect_eq_bitmap(bmap1, 1024, bmap2, 1024);
|
|
|
|
|
|
/*
|
|
* test short conversion bitmap -> u32[] (1
|
|
* word too short)
|
|
*/
|
|
if (used_u32s > 1) {
|
|
bitmap_zero(bmap1, 1024);
|
|
__set_bit(i, bmap1);
|
|
bitmap_set(bmap1, nbits,
|
|
1024 - nbits); /* garbage */
|
|
memset(arr, 0xff, sizeof(arr));
|
|
|
|
rv = bitmap_to_u32array(arr, used_u32s - 1,
|
|
bmap1, nbits);
|
|
expect_eq_uint((used_u32s - 1)*32, rv);
|
|
|
|
/* 1st used u32 words contain expected
|
|
* bit set, the remaining words are
|
|
* left unchanged (0xff)
|
|
*/
|
|
memset(exp_arr, 0xff, sizeof(exp_arr));
|
|
memset(exp_arr, 0,
|
|
(used_u32s-1)*sizeof(*exp_arr));
|
|
if ((i/32) < (used_u32s - 1))
|
|
exp_arr[i/32] = (1U<<(i%32));
|
|
expect_eq_u32_array(exp_arr, 32, arr, 32);
|
|
}
|
|
|
|
/*
|
|
* test short conversion u32[] -> bitmap (3
|
|
* bits too short)
|
|
*/
|
|
if (nbits > 3) {
|
|
memset(arr, 0xff, sizeof(arr)); /* garbage */
|
|
memset(arr, 0, used_u32s*sizeof(*arr));
|
|
arr[i/32] = (1U<<(i%32));
|
|
|
|
bitmap_zero(bmap1, 1024);
|
|
rv = bitmap_from_u32array(bmap1, nbits - 3,
|
|
arr, used_u32s);
|
|
expect_eq_uint(nbits - 3, rv);
|
|
|
|
/* we are expecting the bit < nbits -
|
|
* 3 (none otherwise), and the rest of
|
|
* bmap1 unchanged (0-filled)
|
|
*/
|
|
bitmap_zero(bmap2, 1024);
|
|
if (i < nbits - 3)
|
|
__set_bit(i, bmap2);
|
|
expect_eq_bitmap(bmap2, 1024, bmap1, 1024);
|
|
|
|
/* do the same with bmap1 initially
|
|
* 1-filled
|
|
*/
|
|
|
|
bitmap_fill(bmap1, 1024);
|
|
rv = bitmap_from_u32array(bmap1, nbits - 3,
|
|
arr, used_u32s);
|
|
expect_eq_uint(nbits - 3, rv);
|
|
|
|
/* we are expecting the bit < nbits -
|
|
* 3 (none otherwise), and the rest of
|
|
* bmap1 unchanged (1-filled)
|
|
*/
|
|
bitmap_zero(bmap2, 1024);
|
|
if (i < nbits - 3)
|
|
__set_bit(i, bmap2);
|
|
bitmap_set(bmap2, nbits-3, 1024 - nbits + 3);
|
|
expect_eq_bitmap(bmap2, 1024, bmap1, 1024);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static int __init test_bitmap_init(void)
|
|
{
|
|
test_zero_fill_copy();
|
|
test_bitmap_u32_array_conversions();
|
|
|
|
if (failed_tests == 0)
|
|
pr_info("all %u tests passed\n", total_tests);
|
|
else
|
|
pr_warn("failed %u out of %u tests\n",
|
|
failed_tests, total_tests);
|
|
|
|
return failed_tests ? -EINVAL : 0;
|
|
}
|
|
|
|
static void __exit test_bitmap_cleanup(void)
|
|
{
|
|
}
|
|
|
|
module_init(test_bitmap_init);
|
|
module_exit(test_bitmap_cleanup);
|
|
|
|
MODULE_AUTHOR("david decotigny <david.decotigny@googlers.com>");
|
|
MODULE_LICENSE("GPL");
|