178 lines · c
1/* SPDX-License-Identifier: GPL-2.0-or-later */2#ifndef _MEMBLOCK_TEST_H3#define _MEMBLOCK_TEST_H4 5#include <stdlib.h>6#include <assert.h>7#include <linux/types.h>8#include <linux/seq_file.h>9#include <linux/memblock.h>10#include <linux/sizes.h>11#include <linux/printk.h>12#include <../selftests/kselftest.h>13 14#define MEM_SIZE SZ_32K15#define PHYS_MEM_SIZE SZ_16M16#define NUMA_NODES 817 18#define INIT_MEMBLOCK_REGIONS 12819#define INIT_MEMBLOCK_RESERVED_REGIONS INIT_MEMBLOCK_REGIONS20 21enum test_flags {22 /* No special request. */23 TEST_F_NONE = 0x0,24 /* Perform raw allocations (no zeroing of memory). */25 TEST_F_RAW = 0x1,26 /* Perform allocations on the exact node specified. */27 TEST_F_EXACT = 0x228};29 30/**31 * ASSERT_EQ():32 * Check the condition33 * @_expected == @_seen34 * If false, print failed test message (if running with --verbose) and then35 * assert.36 */37#define ASSERT_EQ(_expected, _seen) do { \38 if ((_expected) != (_seen)) \39 test_fail(); \40 assert((_expected) == (_seen)); \41} while (0)42 43#define ASSERT_TRUE(_seen) ASSERT_EQ(true, _seen)44#define ASSERT_FALSE(_seen) ASSERT_EQ(false, _seen)45 46/**47 * ASSERT_NE():48 * Check the condition49 * @_expected != @_seen50 * If false, print failed test message (if running with --verbose) and then51 * assert.52 */53#define ASSERT_NE(_expected, _seen) do { \54 if ((_expected) == (_seen)) \55 test_fail(); \56 assert((_expected) != (_seen)); \57} while (0)58 59/**60 * ASSERT_LT():61 * Check the condition62 * @_expected < @_seen63 * If false, print failed test message (if running with --verbose) and then64 * assert.65 */66#define ASSERT_LT(_expected, _seen) do { \67 if ((_expected) >= (_seen)) \68 test_fail(); \69 assert((_expected) < (_seen)); \70} while (0)71 72/**73 * ASSERT_LE():74 * Check the condition75 * @_expected <= @_seen76 * If false, print failed test message (if running with --verbose) and then77 * assert.78 */79#define ASSERT_LE(_expected, _seen) do { \80 if ((_expected) > (_seen)) \81 test_fail(); \82 assert((_expected) <= (_seen)); \83} while (0)84 85/**86 * ASSERT_MEM_EQ():87 * Check that the first @_size bytes of @_seen are all equal to @_expected.88 * If false, print failed test message (if running with --verbose) and then89 * assert.90 */91#define ASSERT_MEM_EQ(_seen, _expected, _size) do { \92 for (int _i = 0; _i < (_size); _i++) { \93 ASSERT_EQ(((char *)_seen)[_i], (_expected)); \94 } \95} while (0)96 97/**98 * ASSERT_MEM_NE():99 * Check that none of the first @_size bytes of @_seen are equal to @_expected.100 * If false, print failed test message (if running with --verbose) and then101 * assert.102 */103#define ASSERT_MEM_NE(_seen, _expected, _size) do { \104 for (int _i = 0; _i < (_size); _i++) { \105 ASSERT_NE(((char *)_seen)[_i], (_expected)); \106 } \107} while (0)108 109#define PREFIX_PUSH() prefix_push(__func__)110 111/*112 * Available memory registered with memblock needs to be valid for allocs113 * test to run. This is a convenience wrapper for memory allocated in114 * dummy_physical_memory_init() that is later registered with memblock115 * in setup_memblock().116 */117struct test_memory {118 void *base;119};120 121struct region {122 phys_addr_t base;123 phys_addr_t size;124};125 126static inline phys_addr_t __maybe_unused region_end(struct memblock_region *rgn)127{128 return rgn->base + rgn->size;129}130 131void reset_memblock_regions(void);132void reset_memblock_attributes(void);133void setup_memblock(void);134void setup_numa_memblock(const unsigned int node_fracs[]);135void dummy_physical_memory_init(void);136void dummy_physical_memory_cleanup(void);137phys_addr_t dummy_physical_memory_base(void);138void parse_args(int argc, char **argv);139 140void test_fail(void);141void test_pass(void);142void test_print(const char *fmt, ...);143void prefix_reset(void);144void prefix_push(const char *prefix);145void prefix_pop(void);146 147static inline void test_pass_pop(void)148{149 test_pass();150 prefix_pop();151}152 153static inline void run_top_down(int (*func)())154{155 memblock_set_bottom_up(false);156 prefix_push("top-down");157 func();158 prefix_pop();159}160 161static inline void run_bottom_up(int (*func)())162{163 memblock_set_bottom_up(true);164 prefix_push("bottom-up");165 func();166 prefix_pop();167}168 169static inline void assert_mem_content(void *mem, int size, int flags)170{171 if (flags & TEST_F_RAW)172 ASSERT_MEM_NE(mem, 0, size);173 else174 ASSERT_MEM_EQ(mem, 0, size);175}176 177#endif178