Add a couple of tests for new STACK_FLEX_ARRAY_SIZE() helper.

Signed-off-by: Gustavo A. R. Silva <gustavo...@kernel.org>
---
Changes v2:
 - Add 0 test for `empty` instance.

v1:
 - Link: 
https://lore.kernel.org/linux-hardening/8cf48c3f9d8ef9b999d87cc0a822ffe539bf7a64.1745342381.git.gustavo...@kernel.org/

 lib/tests/overflow_kunit.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lib/tests/overflow_kunit.c b/lib/tests/overflow_kunit.c
index 894691b4411a..19cb03b25dc5 100644
--- a/lib/tests/overflow_kunit.c
+++ b/lib/tests/overflow_kunit.c
@@ -1210,6 +1210,10 @@ static void DEFINE_FLEX_test(struct kunit *test)
        KUNIT_EXPECT_EQ(test, __struct_size(empty->array), 0);
        KUNIT_EXPECT_EQ(test, __member_size(empty->array), 0);
 
+       KUNIT_EXPECT_EQ(test, STACK_FLEX_ARRAY_SIZE(two, array), 2);
+       KUNIT_EXPECT_EQ(test, STACK_FLEX_ARRAY_SIZE(eight, array), 8);
+       KUNIT_EXPECT_EQ(test, STACK_FLEX_ARRAY_SIZE(empty, array), 0);
+
        /* If __counted_by is not being used, array size will have the on-stack 
size. */
        if (!IS_ENABLED(CONFIG_CC_HAS_COUNTED_BY))
                array_size_override = 2 * sizeof(s16);
-- 
2.43.0


Reply via email to