linux-imx

/*

 * Testsuite for atomic64_t functions

 *

 * Copyright © 2010  Luca Barbieri

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 */

​

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

​

#include <linux/init.h>

#include <linux/bug.h>

#include <linux/kernel.h>

#include <linux/atomic.h>

#include <linux/module.h>

​

#ifdef CONFIG_X86

#include <asm/cpufeature.h> /* for boot_cpu_has below */

#endif

​

#define TEST(bit, op, c_op, val)                \

do {                                \

    atomic##bit##_set(&v, v0);              \

    r = v0;                         \

    atomic##bit##_##op(val, &v);                \

    r c_op val;                     \

    WARN(atomic##bit##_read(&v) != r, "%Lx != %Lx\n",   \

        (unsigned long long)atomic##bit##_read(&v), \

        (unsigned long long)r);             \

} while (0)

​

/*

 * Test for a atomic operation family,

 * @test should be a macro accepting parameters (bit, op, ...)

 */

​

#define FAMILY_TEST(test, bit, op, args...) \

do {                        \

    test(bit, op, ##args);      \

    test(bit, op##_acquire, ##args);    \

    test(bit, op##_release, ##args);    \

    test(bit, op##_relaxed, ##args);    \

} while (0)

​

#define TEST_RETURN(bit, op, c_op, val)             \

do {                                \

    atomic##bit##_set(&v, v0);              \

    r = v0;                         \

    r c_op val;                     \

    BUG_ON(atomic##bit##_##op(val, &v) != r);       \

    BUG_ON(atomic##bit##_read(&v) != r);            \

} while (0)

​

#define TEST_FETCH(bit, op, c_op, val)              \

do {                                \

    atomic##bit##_set(&v, v0);              \

    r = v0;                         \

    r c_op val;                     \

    BUG_ON(atomic##bit##_##op(val, &v) != v0);      \

    BUG_ON(atomic##bit##_read(&v) != r);            \

} while (0)

​

#define RETURN_FAMILY_TEST(bit, op, c_op, val)          \

do {                                \

    FAMILY_TEST(TEST_RETURN, bit, op, c_op, val);       \

} while (0)

​

#define FETCH_FAMILY_TEST(bit, op, c_op, val)           \

do {                                \

    FAMILY_TEST(TEST_FETCH, bit, op, c_op, val);        \

} while (0)

​

#define TEST_ARGS(bit, op, init, ret, expect, args...)      \

do {                                \

    atomic##bit##_set(&v, init);                \

    BUG_ON(atomic##bit##_##op(&v, ##args) != ret);      \

    BUG_ON(atomic##bit##_read(&v) != expect);       \

} while (0)

​

#define XCHG_FAMILY_TEST(bit, init, new)                \

do {                                    \

    FAMILY_TEST(TEST_ARGS, bit, xchg, init, init, new, new);    \

} while (0)

​

#define CMPXCHG_FAMILY_TEST(bit, init, new, wrong)          \

do {                                    \

    FAMILY_TEST(TEST_ARGS, bit, cmpxchg,                \

            init, init, new, init, new);            \

    FAMILY_TEST(TEST_ARGS, bit, cmpxchg,                \