Commit 78c74aee authored by Alexander Potapenko's avatar Alexander Potapenko Committed by Andrew Morton
Browse files

kmsan: add memsetXX tests 

Add tests ensuring that memset16()/memset32()/memset64() are instrumented
by KMSAN and correctly initialize the memory.

Link: https://lkml.kernel.org/r/20230303141433.3422671-4-glider@google.com


Signed-off-by: default avatarAlexander Potapenko <glider@google.com>
Reviewed-by: default avatarMarco Elver <elver@google.com>
Cc: Daniel Vetter <daniel@ffwll.ch>
Cc: Geert Uytterhoeven <geert@linux-m68k.org>
Cc: Helge Deller <deller@gmx.de>
Cc: Kees Cook <keescook@chromium.org>
Cc: Tetsuo Handa <penguin-kernel@i-love.sakura.ne.jp>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
parent 27f644dc

mm/kmsan/kmsan_test.c

+22 −0
Original line number Diff line number Diff line
@@ -504,6 +504,25 @@ static void test_memcpy_aligned_to_unaligned2(struct kunit *test) 
	KUNIT_EXPECT_TRUE(test, report_matches(&expect));

}



/* Generate test cases for memset16(), memset32(), memset64(). */

#define DEFINE_TEST_MEMSETXX(size)                                          \

	static void test_memset##size(struct kunit *test)                   \

	{                                                                   \

		EXPECTATION_NO_REPORT(expect);                              \

		volatile uint##size##_t uninit;                             \

                                                                            \

		kunit_info(test,                                            \

			   "memset" #size "() should initialize memory\n"); \

		DO_NOT_OPTIMIZE(uninit);                                    \

		memset##size((uint##size##_t *)&uninit, 0, 1);              \

		kmsan_check_memory((void *)&uninit, sizeof(uninit));        \

		KUNIT_EXPECT_TRUE(test, report_matches(&expect));           \

	}



DEFINE_TEST_MEMSETXX(16)

DEFINE_TEST_MEMSETXX(32)

DEFINE_TEST_MEMSETXX(64)



static noinline void fibonacci(int *array, int size, int start)

{

	if (start < 2 || (start == size))
@@ -550,6 +569,9 @@ static struct kunit_case kmsan_test_cases[] = { 
	KUNIT_CASE(test_memcpy_aligned_to_aligned),

	KUNIT_CASE(test_memcpy_aligned_to_unaligned),

	KUNIT_CASE(test_memcpy_aligned_to_unaligned2),

	KUNIT_CASE(test_memset16),

	KUNIT_CASE(test_memset32),

	KUNIT_CASE(test_memset64),

	KUNIT_CASE(test_long_origin_chain),

	{},

};