From patchwork Fri Mar 1 00:26:47 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Kees Cook X-Patchwork-Id: 208579 Return-Path: Delivered-To: ouuuleilei@gmail.com Received: by 2002:a05:7301:2097:b0:108:e6aa:91d0 with SMTP id gs23csp768419dyb; Thu, 29 Feb 2024 16:27:39 -0800 (PST) X-Forwarded-Encrypted: i=3; AJvYcCVJKL7B2hNSRx3x+BT7nN+x/trzZ5kEDN+VN/j52KGMCmzJ+thInlYAPaWEXseiM1kUPpc0N9dj/Gkmn60g1yVy2yJeSg== X-Google-Smtp-Source: AGHT+IH5q/kDywvTD9xPixM8w5CCkAGxxhYKtlaCcrJ1uRlGdCU+/doVZfNr+leFljrimkUPqwi4 X-Received: by 2002:a05:6402:2152:b0:566:ef8:93f6 with SMTP id bq18-20020a056402215200b005660ef893f6mr170988edb.0.1709252859746; Thu, 29 Feb 2024 16:27:39 -0800 (PST) ARC-Seal: i=2; a=rsa-sha256; t=1709252859; cv=pass; d=google.com; s=arc-20160816; b=xPBNtud5IBdo9pFIpckgal7a7aIikChwyPOlgAJx6dE5kLsyDt0zP8uBer0DtvIeRc 9F5EsU7SIE+B8H9K/mKrIcgnP3WId+vgdTv6XsAAaTDUsFC0ZbGVJy4qRnxn8yq/T1AF 6Ls4yonPpfnTtrHUFmFwcX5QLvJM8ds+747pkmPCeASqUrvmDP2bwMZCNEEj+Juk5I3l rErbWbceLjh+d6Jh3BXtMyAT7o9YM31TMDnRFz2L8kkCxOpZZj38p5AocdWlT+36ctPt QW0DXK7h414CdW+ok6PGnuOt91r0HwglrBlYDhlXt2fUxUTdFeLTYfiDFSxr3tdiyFNO 3DOg== ARC-Message-Signature: i=2; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=content-transfer-encoding:mime-version:list-unsubscribe :list-subscribe:list-id:precedence:references:in-reply-to:message-id :date:subject:cc:to:from:dkim-signature; bh=hhQN8lEo47C06uFL3UK+82WnUqWmwbabHP0lvyxYyyI=; fh=OwXBXeVxZY9OOASq0untr+zqhZdP2YPuSNowrwwPrXM=; b=IAhpHgopBVw/CcRX4x9wnQzM2iHEsNkPOZfgxoe9iLEKtQ3Aj/FqlhcSWF/zz933hU lhDn8DqUHuu7hjr5GlkM6Dh8yntQ0tjhpoVGGy3SY2w6yD6tKmTs52FZBHd3T9cXVj3u ftR9zCdIlGzqIPWDJEwDXZ1xB+eXJ7aDX1acAWcT+1f7hY3In8+GwWc5UsXyvXUFPajU YjtPr5PozKjntiRGSqyU1Xo1sT6SUUJauva2vfoAX4PXC5Uk4mmD1LwVD9yyK0enAc1p O3PFzatLDnu/GIlJVX9/geAhlr+e8nJQ4ra0o/R7tFHFyzy5EdAP+PxcfBdpQunUPZgD Fouw==; dara=google.com ARC-Authentication-Results: i=2; mx.google.com; dkim=pass header.i=@chromium.org header.s=google header.b=I966CMgw; arc=pass (i=1 spf=pass spfdomain=chromium.org dkim=pass dkdomain=chromium.org dmarc=pass fromdomain=chromium.org); spf=pass (google.com: domain of linux-kernel+bounces-87733-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.80.249 as permitted sender) smtp.mailfrom="linux-kernel+bounces-87733-ouuuleilei=gmail.com@vger.kernel.org"; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=chromium.org Received: from am.mirrors.kernel.org (am.mirrors.kernel.org. [147.75.80.249]) by mx.google.com with ESMTPS id o7-20020a056402038700b0056442886f79si955267edv.259.2024.02.29.16.27.39 for (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 29 Feb 2024 16:27:39 -0800 (PST) Received-SPF: pass (google.com: domain of linux-kernel+bounces-87733-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.80.249 as permitted sender) client-ip=147.75.80.249; Authentication-Results: mx.google.com; dkim=pass header.i=@chromium.org header.s=google header.b=I966CMgw; arc=pass (i=1 spf=pass spfdomain=chromium.org dkim=pass dkdomain=chromium.org dmarc=pass fromdomain=chromium.org); spf=pass (google.com: domain of linux-kernel+bounces-87733-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.80.249 as permitted sender) smtp.mailfrom="linux-kernel+bounces-87733-ouuuleilei=gmail.com@vger.kernel.org"; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=chromium.org Received: from smtp.subspace.kernel.org (wormhole.subspace.kernel.org [52.25.139.140]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by am.mirrors.kernel.org (Postfix) with ESMTPS id 372311F2313C for ; Fri, 1 Mar 2024 00:27:39 +0000 (UTC) Received: from localhost.localdomain (localhost.localdomain [127.0.0.1]) by smtp.subspace.kernel.org (Postfix) with ESMTP id 0E60E29A1; Fri, 1 Mar 2024 00:27:03 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; dkim=pass (1024-bit key) header.d=chromium.org header.i=@chromium.org header.b="I966CMgw" Received: from mail-il1-f170.google.com (mail-il1-f170.google.com [209.85.166.170]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 52F72386 for ; Fri, 1 Mar 2024 00:26:53 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.166.170 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1709252818; cv=none; b=IYNCEcUdeGshLhA0VUkJDVU9+AMWuF9c0Cmd2bx8HoGG7mFJ25XLB2ULhcJDDDWYyQUX1KzGGbdBmzzkqI+1VLHvEbPR/Lvd8T8ngIrNvM1t+9SSZpk8sFbFcCGGreXHaY0Vc0cxJXVVClwT5dGeZR0oORXBF9SyTUKE89HV8yE= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1709252818; c=relaxed/simple; bh=55xyn8zKO4nNH9Rvcx8r5OPu2ble/KrvXCx28GmYr8o=; h=From:To:Cc:Subject:Date:Message-Id:In-Reply-To:References: MIME-Version; b=tDnec9YKUgW598LJ5GozBF9aTB6pL9AfXFQSZh8Tek+n0pybFePrw9ICg6t6xUaHV4DmUPbbXxobOysKHqCvGTkbW2tUyEECNXbg+WR6iJ4ZI0BO8qtS9XGqyujOTVQeONTq05anElCgvedk8RWhzF8qF6Zg2IDLCfZKcRvFgRQ= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=chromium.org; spf=pass smtp.mailfrom=chromium.org; dkim=pass (1024-bit key) header.d=chromium.org header.i=@chromium.org header.b=I966CMgw; arc=none smtp.client-ip=209.85.166.170 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=chromium.org Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=chromium.org Received: by mail-il1-f170.google.com with SMTP id e9e14a558f8ab-3652731b2b7so6153005ab.2 for ; Thu, 29 Feb 2024 16:26:53 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=chromium.org; s=google; t=1709252812; x=1709857612; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=hhQN8lEo47C06uFL3UK+82WnUqWmwbabHP0lvyxYyyI=; b=I966CMgwFsUOlvnt2dycxXw0qMhxkqVDPfcjooz6KPKJ3iDvGwwHMlImO1OjD2ZSGn 1RyJeGX371u8IOcUv0DdX9xvnBI076cePxf1rI/jRNfET6WER+lpavqwNShOjZ0VmXOz 6lqxmMmuIOqZ6qCNKU+CrqkJuaolvseY4QfMM= X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1709252812; x=1709857612; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=hhQN8lEo47C06uFL3UK+82WnUqWmwbabHP0lvyxYyyI=; b=do2Ln3BjtiJ+HnaBu4v1Di+yIFGRQ03jlgY0H90rovknDFOelxxmWIqozpScowBtYO P+FUb5v+5Gwt7uJZD3dECITtMAYWdILMF/0uanDFua24YkVnjEmDnmi+F8Gwk5d7qV1F dZBIWQxsHC1XjWkvGMSzSOSM2BPbqYu8YSkSGtx8yAze5gxqmS8CNZD2p/yKgwEtB1Es RiWxOyRjy+bbfPVu11Z/ONu8nNsJy+Yc3DuZLVNHNs3OEStTA/L1QxakNcf512qXHwVs fpnMUgjiTsX6PY5bxNIoP/YrWb9HO/Wl8vLewVIWXn/gKsO3b/wuW8zpwKlxFM5kc1M/ 1caQ== X-Forwarded-Encrypted: i=1; AJvYcCVtHdJVgI6tj1N7W07tySD5HCDi1s/s82J3R2R1+6gN0RYGXI+b3rQxm+v51fJLpDKJdQlfDK3PMiQcEP0jt+V+NOx/JDNm5MdNd0gS X-Gm-Message-State: AOJu0Ywda3eBVhMVmQGjliQ3Y5QLggJdzqkwGkakr/rHF45LEw9O4Egt JO1T3Lq+zNs8+ApxiVb8CAArYyxjlctpNEYdtSTa8BrUtUrrE8diDMxw6n/yyGg5SEME+2qel3M = X-Received: by 2002:a05:6e02:1a8c:b0:365:3e12:8bfb with SMTP id k12-20020a056e021a8c00b003653e128bfbmr299302ilv.21.1709252812357; Thu, 29 Feb 2024 16:26:52 -0800 (PST) Received: from www.outflux.net ([198.0.35.241]) by smtp.gmail.com with ESMTPSA id a27-20020a62d41b000000b006e48e64ef54sm1971821pfh.173.2024.02.29.16.26.50 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 29 Feb 2024 16:26:50 -0800 (PST) From: Kees Cook To: Andy Shevchenko Cc: Kees Cook , linux-hardening@vger.kernel.org, linux-kernel@vger.kernel.org Subject: [PATCH v2 1/2] string: Convert selftest to KUnit Date: Thu, 29 Feb 2024 16:26:47 -0800 Message-Id: <20240301002650.1544829-1-keescook@chromium.org> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20240301002416.it.092-kees@kernel.org> References: <20240301002416.it.092-kees@kernel.org> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 X-Developer-Signature: v=1; a=openpgp-sha256; l=9530; i=keescook@chromium.org; h=from:subject; bh=55xyn8zKO4nNH9Rvcx8r5OPu2ble/KrvXCx28GmYr8o=; b=owEBbQKS/ZANAwAKAYly9N/cbcAmAcsmYgBl4SDIk3eOcod2ISKCbrONdo52S/cd/eKMgfWDU 72HZYnkRyCJAjMEAAEKAB0WIQSlw/aPIp3WD3I+bhOJcvTf3G3AJgUCZeEgyAAKCRCJcvTf3G3A JsOIEACv8/xAArv+Wft8iBQU4FUyttrBGIVPhLYyQu+RfXvWSjbQ3X3T5Ofl1pXImdEP/PAKrN2 yZKCEEIV4hLX9pcraulqkrM29te373it+eEbLDfygXuhjRlLkMSY0auegc375nTkBNAvf4rql9X cQx2jgS2u6LKVYYKu7LezIgYo2oEstoA6Hl9G27huYreIcZ6Iw81amesuWa8bWerpL3VTrTP0+S QPgKdz05ceBkVhWi2MA3akGtytHyAPmKY/lakVJXK5pNGWDcHC0I8pYtq1L5pFkJ9A7rzXYJ9TQ Y5s3zaIt0xbCOQRR0TOxQYo4dpsQLHL1vqfF3fxYkMbbPcIbNuE4AeVTjbNuCgb7JXubVSj2CIL gfhtUuYlBKB3rDOIBzQ6tKHYFxgeX7fvCO+DVxw9iukDC06Z3ECzgOO18CG0uvq0UIL73CwADke EEKbvZ6sumFwxXDM0HKr3TGT499yeU0CzzKyy9udfnj6JVRyBhEZfJhuFeY3kV33ken1404kXjN ZaGlk9Dr2QIDCDdDZEhm41z4oQ+tA6jXdELtxCKgE9pYNAgfALm5YbB84Ft1CACEP7rgdL2uiEp NpWSOnDBwn3PMZf9+4t/+FzpKWt4OdL4VgmRT+agW6kFzGMCXnqAUEwLXBZBqibVmn7PbO46weO vBCiKckEkbunZzA== X-Developer-Key: i=keescook@chromium.org; a=openpgp; fpr=A5C3F68F229DD60F723E6E138972F4DFDC6DC026 X-getmail-retrieved-from-mailbox: INBOX X-GMAIL-THRID: 1792281526557432795 X-GMAIL-MSGID: 1792281526557432795 Convert test_string.c to KUnit so it can be easily run with everything else. Signed-off-by: Kees Cook --- Cc: Andy Shevchenko Cc: linux-hardening@vger.kernel.org --- MAINTAINERS | 2 +- lib/Kconfig.debug | 6 +- lib/Makefile | 2 +- lib/{test_string.c => string_kunit.c} | 166 +++++++++----------------- 4 files changed, 61 insertions(+), 115 deletions(-) rename lib/{test_string.c => string_kunit.c} (54%) diff --git a/MAINTAINERS b/MAINTAINERS index cd651c4df019..9f1f68cccd6a 100644 --- a/MAINTAINERS +++ b/MAINTAINERS @@ -8976,9 +8976,9 @@ F: include/linux/string.h F: include/linux/string_choices.h F: include/linux/string_helpers.h F: lib/string.c +F: lib/string_kunit.c F: lib/string_helpers.c F: lib/test-string_helpers.c -F: lib/test_string.c F: scripts/coccinelle/api/string_choices.cocci GENERIC UIO DRIVER FOR PCI DEVICES diff --git a/lib/Kconfig.debug b/lib/Kconfig.debug index 4e2febe3b568..406cdf353488 100644 --- a/lib/Kconfig.debug +++ b/lib/Kconfig.debug @@ -2352,8 +2352,10 @@ config ASYNC_RAID6_TEST config TEST_HEXDUMP tristate "Test functions located in the hexdump module at runtime" -config STRING_SELFTEST - tristate "Test string functions at runtime" +config STRING_KUNIT_TEST + tristate "KUnit test string functions at runtime" if !KUNIT_ALL_TESTS + depends on KUNIT + default KUNIT_ALL_TESTS config TEST_STRING_HELPERS tristate "Test functions located in the string_helpers module at runtime" diff --git a/lib/Makefile b/lib/Makefile index eae87c41b22b..946277c37831 100644 --- a/lib/Makefile +++ b/lib/Makefile @@ -49,7 +49,7 @@ obj-y += bcd.o sort.o parser.o debug_locks.o random32.o \ percpu-refcount.o rhashtable.o base64.o \ once.o refcount.o rcuref.o usercopy.o errseq.o bucket_locks.o \ generic-radix-tree.o bitmap-str.o -obj-$(CONFIG_STRING_SELFTEST) += test_string.o +obj-$(CONFIG_STRING_KUNIT_TEST) += string_kunit.o obj-y += string_helpers.o obj-$(CONFIG_TEST_STRING_HELPERS) += test-string_helpers.o obj-y += hexdump.o diff --git a/lib/test_string.c b/lib/string_kunit.c similarity index 54% rename from lib/test_string.c rename to lib/string_kunit.c index c5cb92fb710e..bbb54ac11f7b 100644 --- a/lib/test_string.c +++ b/lib/string_kunit.c @@ -1,17 +1,23 @@ // SPDX-License-Identifier: GPL-2.0-only +/* + * Test cases for string functions. + */ + +#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt + +#include #include #include #include #include -static __init int memset16_selftest(void) +static void test_memset16(struct kunit *test) { unsigned i, j, k; u16 v, *p; p = kmalloc(256 * 2 * 2, GFP_KERNEL); - if (!p) - return -1; + KUNIT_ASSERT_NOT_NULL(test, p); for (i = 0; i < 256; i++) { for (j = 0; j < 256; j++) { @@ -20,34 +26,28 @@ static __init int memset16_selftest(void) for (k = 0; k < 512; k++) { v = p[k]; if (k < i) { - if (v != 0xa1a1) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1); } else if (k < i + j) { - if (v != 0xb1b2) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xb1b2); } else { - if (v != 0xa1a1) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1); } } } } -fail: kfree(p); if (i < 256) - return (i << 24) | (j << 16) | k | 0x8000; - return 0; + KUNIT_EXPECT_EQ(test, 0, (i << 24) | (j << 16) | k | 0x8000); } -static __init int memset32_selftest(void) +static void test_memset32(struct kunit *test) { unsigned i, j, k; u32 v, *p; p = kmalloc(256 * 2 * 4, GFP_KERNEL); - if (!p) - return -1; + KUNIT_ASSERT_NOT_NULL(test, p); for (i = 0; i < 256; i++) { for (j = 0; j < 256; j++) { @@ -56,34 +56,28 @@ static __init int memset32_selftest(void) for (k = 0; k < 512; k++) { v = p[k]; if (k < i) { - if (v != 0xa1a1a1a1) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1a1a1); } else if (k < i + j) { - if (v != 0xb1b2b3b4) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xb1b2b3b4); } else { - if (v != 0xa1a1a1a1) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1a1a1); } } } } -fail: kfree(p); if (i < 256) - return (i << 24) | (j << 16) | k | 0x8000; - return 0; + KUNIT_EXPECT_EQ(test, 0, (i << 24) | (j << 16) | k | 0x8000); } -static __init int memset64_selftest(void) +static void test_memset64(struct kunit *test) { unsigned i, j, k; u64 v, *p; p = kmalloc(256 * 2 * 8, GFP_KERNEL); - if (!p) - return -1; + KUNIT_ASSERT_NOT_NULL(test, p); for (i = 0; i < 256; i++) { for (j = 0; j < 256; j++) { @@ -92,27 +86,22 @@ static __init int memset64_selftest(void) for (k = 0; k < 512; k++) { v = p[k]; if (k < i) { - if (v != 0xa1a1a1a1a1a1a1a1ULL) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1a1a1a1a1a1a1ULL); } else if (k < i + j) { - if (v != 0xb1b2b3b4b5b6b7b8ULL) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xb1b2b3b4b5b6b7b8ULL); } else { - if (v != 0xa1a1a1a1a1a1a1a1ULL) - goto fail; + KUNIT_EXPECT_EQ(test, v, 0xa1a1a1a1a1a1a1a1ULL); } } } } -fail: kfree(p); if (i < 256) - return (i << 24) | (j << 16) | k | 0x8000; - return 0; + KUNIT_EXPECT_EQ(test, 0, (i << 24) | (j << 16) | k | 0x8000); } -static __init int strchr_selftest(void) +static void test_strchr(struct kunit *test) { const char *test_string = "abcdefghijkl"; const char *empty_string = ""; @@ -121,26 +110,20 @@ static __init int strchr_selftest(void) for (i = 0; i < strlen(test_string) + 1; i++) { result = strchr(test_string, test_string[i]); - if (result - test_string != i) - return i + 'a'; + KUNIT_ASSERT_EQ(test, result - test_string, i); } result = strchr(empty_string, '\0'); - if (result != empty_string) - return 0x101; + KUNIT_ASSERT_PTR_EQ(test, result, empty_string); result = strchr(empty_string, 'a'); - if (result) - return 0x102; + KUNIT_ASSERT_NULL(test, result); result = strchr(test_string, 'z'); - if (result) - return 0x103; - - return 0; + KUNIT_ASSERT_NULL(test, result); } -static __init int strnchr_selftest(void) +static void test_strnchr(struct kunit *test) { const char *test_string = "abcdefghijkl"; const char *empty_string = ""; @@ -153,33 +136,27 @@ static __init int strnchr_selftest(void) if (j <= i) { if (!result) continue; - return ((i + 'a') << 8) | j; + KUNIT_ASSERT_EQ(test, 0, 1); } if (result - test_string != i) - return ((i + 'a') << 8) | j; + KUNIT_ASSERT_EQ(test, 0, 1); } } result = strnchr(empty_string, 0, '\0'); - if (result) - return 0x10001; + KUNIT_ASSERT_NULL(test, result); result = strnchr(empty_string, 1, '\0'); - if (result != empty_string) - return 0x10002; + KUNIT_ASSERT_PTR_EQ(test, result, empty_string); result = strnchr(empty_string, 1, 'a'); - if (result) - return 0x10003; + KUNIT_ASSERT_NULL(test, result); result = strnchr(NULL, 0, '\0'); - if (result) - return 0x10004; - - return 0; + KUNIT_ASSERT_NULL(test, result); } -static __init int strspn_selftest(void) +static void test_strspn(struct kunit *test) { static const struct strspn_test { const char str[16]; @@ -187,7 +164,7 @@ static __init int strspn_selftest(void) const char reject[16]; unsigned a; unsigned r; - } tests[] __initconst = { + } tests[] = { { "foobar", "", "", 0, 6 }, { "abba", "abc", "ABBA", 4, 4 }, { "abba", "a", "b", 1, 1 }, @@ -198,60 +175,27 @@ static __init int strspn_selftest(void) for (i = 0; i < ARRAY_SIZE(tests); ++i, ++s) { res = strspn(s->str, s->accept); - if (res != s->a) - return 0x100 + 2*i; + KUNIT_ASSERT_EQ(test, res, s->a); res = strcspn(s->str, s->reject); - if (res != s->r) - return 0x100 + 2*i + 1; + KUNIT_ASSERT_EQ(test, res, s->r); } - return 0; -} - -static __exit void string_selftest_remove(void) -{ } -static __init int string_selftest_init(void) -{ - int test, subtest; - - test = 1; - subtest = memset16_selftest(); - if (subtest) - goto fail; - - test = 2; - subtest = memset32_selftest(); - if (subtest) - goto fail; - - test = 3; - subtest = memset64_selftest(); - if (subtest) - goto fail; +static struct kunit_case string_test_cases[] = { + KUNIT_CASE(test_memset16), + KUNIT_CASE(test_memset32), + KUNIT_CASE(test_memset64), + KUNIT_CASE(test_strchr), + KUNIT_CASE(test_strnchr), + KUNIT_CASE(test_strspn), + {} +}; - test = 4; - subtest = strchr_selftest(); - if (subtest) - goto fail; +static struct kunit_suite string_test_suite = { + .name = "string", + .test_cases = string_test_cases, +}; - test = 5; - subtest = strnchr_selftest(); - if (subtest) - goto fail; - - test = 6; - subtest = strspn_selftest(); - if (subtest) - goto fail; - - pr_info("String selftests succeeded\n"); - return 0; -fail: - pr_crit("String selftest failure %d.%08x\n", test, subtest); - return 0; -} +kunit_test_suites(&string_test_suite); -module_init(string_selftest_init); -module_exit(string_selftest_remove); MODULE_LICENSE("GPL v2"); From patchwork Fri Mar 1 00:26:48 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Kees Cook X-Patchwork-Id: 208578 Return-Path: Delivered-To: ouuuleilei@gmail.com Received: by 2002:a05:7301:2097:b0:108:e6aa:91d0 with SMTP id gs23csp768401dyb; Thu, 29 Feb 2024 16:27:37 -0800 (PST) X-Forwarded-Encrypted: i=3; AJvYcCUNoS90z1pjQxHg2Ib0ZUyWwQW6lmZYXG9nSnRNJF2Da0/xlqNsxMYqSoBt0Yh5x0A+hX1GEPSvQysEaqDYJYZMn48H3g== X-Google-Smtp-Source: AGHT+IGlwIDAAbkjyPHVYgK7KxTq2fAWlC/etr8jvj2XluRp/iQ605iupY/DntVcBYHt5oFPqmkz X-Received: by 2002:a05:6a00:2d99:b0:6e5:3b95:de38 with SMTP id fb25-20020a056a002d9900b006e53b95de38mr375913pfb.12.1709252857610; Thu, 29 Feb 2024 16:27:37 -0800 (PST) ARC-Seal: i=2; a=rsa-sha256; t=1709252857; cv=pass; d=google.com; s=arc-20160816; b=DikVqC600KRcpEH99nm5Debx7fT4sy5k/FGMEaQpQeIy2aHY9rww92v3VsiYfBmBtl 2Bg4uBmQdDhSX6yZ9kZ10HW8mDbbhlmrXbuWLwTIFO16WXq8ElMJrz6wVqa4ksjfLuYx UNk7GkEZ4UkvrQF013zYm2xxGHtADKQ1+LC4qQmWGY3mtXcadBBYC1z+8om5a7gUZ+jE ORX4We0PzSgoPKi7b0Z4WBmDPl+fAbgAoYZHfIOHTdsiSOgSMlZbQHPZk7mD9VFeo0WZ j9KjernBGWkI5Bts9/uqu/LJVjggVggsNchTHUf8TLKty5vEKdH+yIGbb+0wNVkhsMG8 LZ8w== ARC-Message-Signature: i=2; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=content-transfer-encoding:mime-version:list-unsubscribe :list-subscribe:list-id:precedence:references:in-reply-to:message-id :date:subject:cc:to:from:dkim-signature; bh=N7bRmosJbyZC7ndFguV6EJboVLhr8lLHSQFdlQ3jzWE=; fh=8iuk/zSUBmIYDTYoAE6E/yzM0V3Hbe9zz4FlLAoB3LA=; b=OGHZTegD/RTdlt3LZEF4RKv2wpMdY45fBgWz/DHO52l7BClvSv0rmog4qD3clCAejC 9nzVyTrHr3CbTcBfqy+7yFf/Y9+lly8H6BnulNkZdxGIi1hy6VL6BRC01p2cfkxIiu5d U2be75PRn8cIAyoWyrwT3o7AztcitmRL/oZs9+DL27rd5LW9Rr8AwwVTzdZ+ppbNaH0g QsxbGZOpfIHpnp4zAGyNrxo+4Sf94PL6VWRRDBZHFrpSteHMU7cbZdXkpUSwXfBZkosw F+breZT3NfCPxzMZUp0b0tJY/V5WR/nQEOqmaoFwMMCq+6WUXMg6cgNSCe23IrnMKfSh 0h+Q==; dara=google.com ARC-Authentication-Results: i=2; mx.google.com; dkim=pass header.i=@chromium.org header.s=google header.b="g9Wd+5g/"; arc=pass (i=1 spf=pass spfdomain=chromium.org dkim=pass dkdomain=chromium.org dmarc=pass fromdomain=chromium.org); spf=pass (google.com: domain of linux-kernel+bounces-87732-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.48.161 as permitted sender) smtp.mailfrom="linux-kernel+bounces-87732-ouuuleilei=gmail.com@vger.kernel.org"; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=chromium.org Received: from sy.mirrors.kernel.org (sy.mirrors.kernel.org. [147.75.48.161]) by mx.google.com with ESMTPS id e9-20020a056a0000c900b006e50e7bb75fsi2336959pfj.0.2024.02.29.16.27.37 for (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 29 Feb 2024 16:27:37 -0800 (PST) Received-SPF: pass (google.com: domain of linux-kernel+bounces-87732-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.48.161 as permitted sender) client-ip=147.75.48.161; Authentication-Results: mx.google.com; dkim=pass header.i=@chromium.org header.s=google header.b="g9Wd+5g/"; arc=pass (i=1 spf=pass spfdomain=chromium.org dkim=pass dkdomain=chromium.org dmarc=pass fromdomain=chromium.org); spf=pass (google.com: domain of linux-kernel+bounces-87732-ouuuleilei=gmail.com@vger.kernel.org designates 147.75.48.161 as permitted sender) smtp.mailfrom="linux-kernel+bounces-87732-ouuuleilei=gmail.com@vger.kernel.org"; dmarc=pass (p=NONE sp=NONE dis=NONE) header.from=chromium.org Received: from smtp.subspace.kernel.org (wormhole.subspace.kernel.org [52.25.139.140]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by sy.mirrors.kernel.org (Postfix) with ESMTPS id 3AA7EB2232E for ; Fri, 1 Mar 2024 00:27:34 +0000 (UTC) Received: from localhost.localdomain (localhost.localdomain [127.0.0.1]) by smtp.subspace.kernel.org (Postfix) with ESMTP id E2EC92570; Fri, 1 Mar 2024 00:27:01 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; dkim=pass (1024-bit key) header.d=chromium.org header.i=@chromium.org header.b="g9Wd+5g/" Received: from mail-pf1-f182.google.com (mail-pf1-f182.google.com [209.85.210.182]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 7585E631 for ; Fri, 1 Mar 2024 00:26:54 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.210.182 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1709252816; cv=none; b=s5cG+vRn9tk5eImUu5OC4Olxn2kRq+fXNXVIFW1udLmrYZSk/OENWYRxq+KBcMKhSWCsKe8ORSpQLdSXArEg174vqT5VYp4cGKINXjxulfC7RnBF3iCnh/NDTDuESzo4F+QTzqvlXtfU+mxuz4u4V0uiOLbZM4AWONIcy7BbDEE= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1709252816; c=relaxed/simple; bh=KDDjoM6A5PAucF/jTCWxQ0PKblp1gjUY4Bj03CQrEXo=; h=From:To:Cc:Subject:Date:Message-Id:In-Reply-To:References: MIME-Version; b=mKdPU+B8LYsMn/Vf5YUFSAF8M9Nfcsw4D8L/XUO9sHhptzG2NOMSJktxJYIk0BAGhpGFH9hrzqblD6hSs+2BB0yUax8X4U+wsRnWIzJNzfWV43GMKueV/gsOXsU//i80iax+EVWnQlNQbIG+5pyzzjE6TdOfuYLZDSJOz7TgEoE= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=chromium.org; spf=pass smtp.mailfrom=chromium.org; dkim=pass (1024-bit key) header.d=chromium.org header.i=@chromium.org header.b=g9Wd+5g/; arc=none smtp.client-ip=209.85.210.182 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=chromium.org Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=chromium.org Received: by mail-pf1-f182.google.com with SMTP id d2e1a72fcca58-6da9c834646so1284215b3a.3 for ; Thu, 29 Feb 2024 16:26:54 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=chromium.org; s=google; t=1709252814; x=1709857614; darn=vger.kernel.org; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:from:to:cc:subject:date :message-id:reply-to; bh=N7bRmosJbyZC7ndFguV6EJboVLhr8lLHSQFdlQ3jzWE=; b=g9Wd+5g/iVb+bPitYJXzCGKfdixrdch1eNoIHkYTQNOrP1TqIVB5Mh4wunV6Y4xLgD u7OTZiY+PIfcUO6SgCG+kqteGWWFDsnmSm+o7DcBujX+ngOAW1JoPtvUkFBTlmVVXcUd LtgCc3ZkP0BghJC+E92eQ3ePDyRv/WxCb9sf4= X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1709252814; x=1709857614; h=content-transfer-encoding:mime-version:references:in-reply-to :message-id:date:subject:cc:to:from:x-gm-message-state:from:to:cc :subject:date:message-id:reply-to; bh=N7bRmosJbyZC7ndFguV6EJboVLhr8lLHSQFdlQ3jzWE=; b=QpiOGTbnuqsKm+S+QDk43mv2z9fNH2bXeQQLVhwQq0M3SS1t9+hLDHhnPRtfM8c4yR NzAJ1RI+aS1EHm2UszlsnuVRLTWCTQMQ/wgfcLR+AZjam4vEMERaG4LWfrOWuTjShusv xdhYxC1xznZU3rYYcO94guBrmQ6k+7WLLX7//tT1f7uEg/ngOLYXd6Sxbi1QHJXY/26e 6h3BOiPQIaleSqptyYFt8IhUFRUqpTAttMWTRHKLNUbcY8MOnp10zpyALZgQuKzI3hYn j/TYo0mBrylfNF+AwVtKqW5a2V8q3ABVaVMnwYXVvLq1hJ2gkIDNAIFuRk6LE79lSj5j RSFA== X-Forwarded-Encrypted: i=1; AJvYcCVNFJusOgzUWnoRo7uC1Y4EKfo47P7lNviiut4pVZpoht1XWNWI+Ogp1/seyLKfH67D0bbN5vbAu7KaL3An8Jvcy+QU/LpDUYKl+ktz X-Gm-Message-State: AOJu0YxT83besWeMWp1UJZtE9vZMu4LBc8eeABGTlTL47Foxc0sqQ+bb mPs3MyQcdNXOQltoKp795MGQbKU3pm6h93GQy7eZvCq5UsZwaT0VnAia/aGhVQ== X-Received: by 2002:a05:6a21:3288:b0:1a0:9ab5:1e83 with SMTP id yt8-20020a056a21328800b001a09ab51e83mr36196pzb.24.1709252813623; Thu, 29 Feb 2024 16:26:53 -0800 (PST) Received: from www.outflux.net ([198.0.35.241]) by smtp.gmail.com with ESMTPSA id oe1-20020a17090b394100b0029929dca278sm4249147pjb.16.2024.02.29.16.26.50 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 29 Feb 2024 16:26:50 -0800 (PST) From: Kees Cook To: Andy Shevchenko Cc: Kees Cook , Kent Overstreet , linux-hardening@vger.kernel.org, linux-kernel@vger.kernel.org Subject: [PATCH v2 2/2] string: Convert helpers selftest to KUnit Date: Thu, 29 Feb 2024 16:26:48 -0800 Message-Id: <20240301002650.1544829-2-keescook@chromium.org> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20240301002416.it.092-kees@kernel.org> References: <20240301002416.it.092-kees@kernel.org> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 X-Developer-Signature: v=1; a=openpgp-sha256; l=15224; i=keescook@chromium.org; h=from:subject; bh=KDDjoM6A5PAucF/jTCWxQ0PKblp1gjUY4Bj03CQrEXo=; b=owEBbQKS/ZANAwAKAYly9N/cbcAmAcsmYgBl4SDI5Kz4zXaH8+qTFLSxONpT1sbN6feRXpVsc QeLAK+W6G6JAjMEAAEKAB0WIQSlw/aPIp3WD3I+bhOJcvTf3G3AJgUCZeEgyAAKCRCJcvTf3G3A JnxUD/9a02Wuryk8/8zvgKX5QeBB7lKP1KY2QINTt0kSaafqjtcBy901aaD4Vt5YeVcvKuGvlLs CU33MIrTYtqqKH6EPSyo1URO02/iUDzm29eHP7jEPhgIw3YNCY/cZ8BBBeMcP8cjBNpRweW0ElE Ed61l9noS4bFeATRvEsEBctXJxeeWRL9Mpueal0p0dt5lwI+xnqLk5aFLbgmQibfpH+zEy4Ct/M NPx9nuPB7MeLPr7o+RGKrZLTQplCZXZKr1GAD6YSWzNcC0u5APIsmng8i/GF3Q3P8nq65hV8FOU +dtX0VGnJ7qg2QwbC5lrHcclW9YLgWIHR/xfT2EdyG48HoQNpbHwkGEhW4e7uhqeO8FV40bGQmJ mj8937jKtzdFIwqyyoW5qKWQ9kWyMqiTq5H6MX0YVrahxWbscuCSSl8ZGRZoY/pqe8enwLjfa5q G2nWQbTSc0rvgkm3tKxm+/W9mGpH3pjmGkiHV20mgdlQ28YoGm1nXXX1nwHErA9CTfa56+YW/vH vXFAh9B8bOnywK7BlNKhWLvtG1dLe8TzWjsxvufhqsmeezGk5nGhZCDaUIIqF6XLyq3dvgrEbyp d2HQtxrbravC5gHD1im6sHAMb4czItbvKyyyM0TIfPeFhy3nPHZMczaqqjobvKVqhg5fz+zY0B8 Q4vnclsJpa3/eCg== X-Developer-Key: i=keescook@chromium.org; a=openpgp; fpr=A5C3F68F229DD60F723E6E138972F4DFDC6DC026 X-getmail-retrieved-from-mailbox: INBOX X-GMAIL-THRID: 1792281524618318532 X-GMAIL-MSGID: 1792281524618318532 Convert test-string_helpers.c to KUnit so it can be easily run with everything else. Signed-off-by: Kees Cook --- Cc: Andy Shevchenko Cc: Kent Overstreet Cc: linux-hardening@vger.kernel.org --- MAINTAINERS | 2 +- lib/Kconfig.debug | 6 +- lib/Makefile | 2 +- ...tring_helpers.c => string_helpers_kunit.c} | 177 ++++++++++-------- 4 files changed, 103 insertions(+), 84 deletions(-) rename lib/{test-string_helpers.c => string_helpers_kunit.c} (78%) diff --git a/MAINTAINERS b/MAINTAINERS index 9f1f68cccd6a..f3f26d2d4ffb 100644 --- a/MAINTAINERS +++ b/MAINTAINERS @@ -8978,7 +8978,7 @@ F: include/linux/string_helpers.h F: lib/string.c F: lib/string_kunit.c F: lib/string_helpers.c -F: lib/test-string_helpers.c +F: lib/string_helpers_kunit.c F: scripts/coccinelle/api/string_choices.cocci GENERIC UIO DRIVER FOR PCI DEVICES diff --git a/lib/Kconfig.debug b/lib/Kconfig.debug index 406cdf353488..5429e6f170f3 100644 --- a/lib/Kconfig.debug +++ b/lib/Kconfig.debug @@ -2357,8 +2357,10 @@ config STRING_KUNIT_TEST depends on KUNIT default KUNIT_ALL_TESTS -config TEST_STRING_HELPERS - tristate "Test functions located in the string_helpers module at runtime" +config STRING_HELPERS_KUNIT_TEST + tristate "KUnit test string helpers at runtime" if !KUNIT_ALL_TESTS + depends on KUNIT + default KUNIT_ALL_TESTS config TEST_KSTRTOX tristate "Test kstrto*() family of functions at runtime" diff --git a/lib/Makefile b/lib/Makefile index 946277c37831..97c42e38046f 100644 --- a/lib/Makefile +++ b/lib/Makefile @@ -51,7 +51,7 @@ obj-y += bcd.o sort.o parser.o debug_locks.o random32.o \ generic-radix-tree.o bitmap-str.o obj-$(CONFIG_STRING_KUNIT_TEST) += string_kunit.o obj-y += string_helpers.o -obj-$(CONFIG_TEST_STRING_HELPERS) += test-string_helpers.o +obj-$(CONFIG_STRING_HELPERS_KUNIT_TEST) += string_helpers_kunit.o obj-y += hexdump.o obj-$(CONFIG_TEST_HEXDUMP) += test_hexdump.o obj-y += kstrtox.o diff --git a/lib/test-string_helpers.c b/lib/string_helpers_kunit.c similarity index 78% rename from lib/test-string_helpers.c rename to lib/string_helpers_kunit.c index dce67698297b..1fed4fa38f97 100644 --- a/lib/test-string_helpers.c +++ b/lib/string_helpers_kunit.c @@ -3,8 +3,9 @@ */ #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt -#include #include +#include +#include #include #include #include @@ -12,13 +13,21 @@ #include #include -static __init bool test_string_check_buf(const char *name, unsigned int flags, - char *in, size_t p, - char *out_real, size_t q_real, - char *out_test, size_t q_test) +static void test_string_check_buf(struct kunit *test, + const char *name, unsigned int flags, + char *in, size_t p, + char *out_real, size_t q_real, + char *out_test, size_t q_test) { - if (q_real == q_test && !memcmp(out_test, out_real, q_test)) - return true; + int result; + + KUNIT_EXPECT_EQ(test, q_real, q_test); + + result = memcmp(out_test, out_real, q_test); + KUNIT_EXPECT_EQ(test, 0, result); + + if (q_real == q_test && result == 0) + return; pr_warn("Test '%s' failed: flags = %#x\n", name, flags); @@ -28,8 +37,6 @@ static __init bool test_string_check_buf(const char *name, unsigned int flags, out_test, q_test, true); print_hex_dump(KERN_WARNING, "Got: ", DUMP_PREFIX_NONE, 16, 1, out_real, q_real, true); - - return false; } struct test_string { @@ -38,7 +45,7 @@ struct test_string { unsigned int flags; }; -static const struct test_string strings[] __initconst = { +static const struct test_string strings[] = { { .in = "\\f\\ \\n\\r\\t\\v", .out = "\f\\ \n\r\t\v", @@ -61,8 +68,9 @@ static const struct test_string strings[] __initconst = { }, }; -static void __init test_string_unescape(const char *name, unsigned int flags, - bool inplace) +static void test_string_unescape(struct kunit *test, + const char *name, unsigned int flags, + bool inplace) { int q_real = 256; char *in = kmalloc(q_real, GFP_KERNEL); @@ -104,7 +112,7 @@ static void __init test_string_unescape(const char *name, unsigned int flags, q_real = string_unescape(in, out_real, q_real, flags); } - test_string_check_buf(name, flags, in, p - 1, out_real, q_real, + test_string_check_buf(test, name, flags, in, p - 1, out_real, q_real, out_test, q_test); out: kfree(out_real); @@ -124,7 +132,7 @@ struct test_string_2 { }; #define TEST_STRING_2_DICT_0 NULL -static const struct test_string_2 escape0[] __initconst = {{ +static const struct test_string_2 escape0[] = {{ .in = "\f\\ \n\r\t\v", .s1 = {{ .out = "\\f\\ \\n\\r\\t\\v", @@ -222,7 +230,7 @@ static const struct test_string_2 escape0[] __initconst = {{ }}; #define TEST_STRING_2_DICT_1 "b\\ \t\r\xCF" -static const struct test_string_2 escape1[] __initconst = {{ +static const struct test_string_2 escape1[] = {{ .in = "\f\\ \n\r\t\v", .s1 = {{ .out = "\f\\134\\040\n\\015\\011\v", @@ -359,7 +367,7 @@ static const struct test_string_2 escape1[] __initconst = {{ /* terminator */ }}; -static const struct test_string strings_upper[] __initconst = { +static const struct test_string strings_upper[] = { { .in = "abcdefgh1234567890test", .out = "ABCDEFGH1234567890TEST", @@ -370,7 +378,7 @@ static const struct test_string strings_upper[] __initconst = { }, }; -static const struct test_string strings_lower[] __initconst = { +static const struct test_string strings_lower[] = { { .in = "ABCDEFGH1234567890TEST", .out = "abcdefgh1234567890test", @@ -381,8 +389,8 @@ static const struct test_string strings_lower[] __initconst = { }, }; -static __init const char *test_string_find_match(const struct test_string_2 *s2, - unsigned int flags) +static const char *test_string_find_match(const struct test_string_2 *s2, + unsigned int flags) { const struct test_string_1 *s1 = s2->s1; unsigned int i; @@ -403,21 +411,20 @@ static __init const char *test_string_find_match(const struct test_string_2 *s2, return NULL; } -static __init void -test_string_escape_overflow(const char *in, int p, unsigned int flags, const char *esc, +static void +test_string_escape_overflow(struct kunit *test, + const char *in, int p, unsigned int flags, const char *esc, int q_test, const char *name) { int q_real; q_real = string_escape_mem(in, p, NULL, 0, flags, esc); - if (q_real != q_test) - pr_warn("Test '%s' failed: flags = %#x, osz = 0, expected %d, got %d\n", - name, flags, q_test, q_real); + KUNIT_EXPECT_EQ(test, q_real, q_test); } -static __init void test_string_escape(const char *name, - const struct test_string_2 *s2, - unsigned int flags, const char *esc) +static void test_string_escape(struct kunit *test, const char *name, + const struct test_string_2 *s2, + unsigned int flags, const char *esc) { size_t out_size = 512; char *out_test = kmalloc(out_size, GFP_KERNEL); @@ -463,10 +470,10 @@ static __init void test_string_escape(const char *name, q_real = string_escape_mem(in, p, out_real, out_size, flags, esc); - test_string_check_buf(name, flags, in, p, out_real, q_real, out_test, + test_string_check_buf(test, name, flags, in, p, out_real, q_real, out_test, q_test); - test_string_escape_overflow(in, p, flags, esc, q_test, name); + test_string_escape_overflow(test, in, p, flags, esc, q_test, name); out: kfree(in); @@ -475,22 +482,26 @@ static __init void test_string_escape(const char *name, } #define string_get_size_maxbuf 16 -#define test_string_get_size_one(size, blk_size, exp_result10, exp_result2) \ - do { \ - BUILD_BUG_ON(sizeof(exp_result10) >= string_get_size_maxbuf); \ - BUILD_BUG_ON(sizeof(exp_result2) >= string_get_size_maxbuf); \ - __test_string_get_size((size), (blk_size), (exp_result10), \ - (exp_result2)); \ +#define test_string_get_size_one(size, blk_size, exp_result10, exp_result2) \ + do { \ + BUILD_BUG_ON(sizeof(exp_result10) >= string_get_size_maxbuf); \ + BUILD_BUG_ON(sizeof(exp_result2) >= string_get_size_maxbuf); \ + __test_string_get_size(test, (size), (blk_size), (exp_result10), \ + (exp_result2)); \ } while (0) -static __init void test_string_get_size_check(const char *units, - const char *exp, - char *res, - const u64 size, - const u64 blk_size) +static void test_string_get_size_check(struct kunit *test, + const char *units, + const char *exp, + char *res, + const u64 size, + const u64 blk_size) { - if (!memcmp(res, exp, strlen(exp) + 1)) + int result = memcmp(res, exp, strlen(exp) + 1); + + KUNIT_EXPECT_EQ(test, 0, result); + if (!result) return; res[string_get_size_maxbuf - 1] = '\0'; @@ -501,7 +512,7 @@ static __init void test_string_get_size_check(const char *units, pr_warn("expected: '%s', got '%s'\n", exp, res); } -static __init void __strchrcut(char *dst, const char *src, const char *cut) +static void __strchrcut(char *dst, const char *src, const char *cut) { const char *from = src; size_t len; @@ -515,11 +526,12 @@ static __init void __strchrcut(char *dst, const char *src, const char *cut) *dst = '\0'; } -static __init void __test_string_get_size_one(const u64 size, const u64 blk_size, - const char *exp_result10, - const char *exp_result2, - enum string_size_units units, - const char *cut) +static void __test_string_get_size_one(struct kunit *test, + const u64 size, const u64 blk_size, + const char *exp_result10, + const char *exp_result2, + enum string_size_units units, + const char *cut) { char buf10[string_get_size_maxbuf]; char buf2[string_get_size_maxbuf]; @@ -537,13 +549,14 @@ static __init void __test_string_get_size_one(const u64 size, const u64 blk_size string_get_size(size, blk_size, STRING_UNITS_10 | units, buf10, sizeof(buf10)); string_get_size(size, blk_size, STRING_UNITS_2 | units, buf2, sizeof(buf2)); - test_string_get_size_check(prefix10, exp10, buf10, size, blk_size); - test_string_get_size_check(prefix2, exp2, buf2, size, blk_size); + test_string_get_size_check(test, prefix10, exp10, buf10, size, blk_size); + test_string_get_size_check(test, prefix2, exp2, buf2, size, blk_size); } -static __init void __test_string_get_size(const u64 size, const u64 blk_size, - const char *exp_result10, - const char *exp_result2) +static void __test_string_get_size(struct kunit *test, + const u64 size, const u64 blk_size, + const char *exp_result10, + const char *exp_result2) { struct { enum string_size_units units; @@ -557,12 +570,13 @@ static __init void __test_string_get_size(const u64 size, const u64 blk_size, int i; for (i = 0; i < ARRAY_SIZE(get_size_test_cases); i++) - __test_string_get_size_one(size, blk_size, exp_result10, exp_result2, + __test_string_get_size_one(test, size, blk_size, + exp_result10, exp_result2, get_size_test_cases[i].units, get_size_test_cases[i].cut); } -static __init void test_string_get_size(void) +static void test_get_size(struct kunit *test) { /* small values */ test_string_get_size_one(0, 512, "0 B", "0 B"); @@ -582,7 +596,7 @@ static __init void test_string_get_size(void) test_string_get_size_one(4096, U64_MAX, "75.6 ZB", "64.0 ZiB"); } -static void __init test_string_upper_lower(void) +static void test_upper_lower(struct kunit *test) { char *dst; int i; @@ -590,65 +604,68 @@ static void __init test_string_upper_lower(void) for (i = 0; i < ARRAY_SIZE(strings_upper); i++) { const char *s = strings_upper[i].in; int len = strlen(strings_upper[i].in) + 1; + int result; dst = kmalloc(len, GFP_KERNEL); - if (!dst) - return; + KUNIT_ASSERT_NOT_NULL(test, dst); string_upper(dst, s); - if (memcmp(dst, strings_upper[i].out, len)) { + result = memcmp(dst, strings_upper[i].out, len); + KUNIT_EXPECT_EQ(test, 0, result); + if (result) pr_warn("Test 'string_upper' failed : expected %s, got %s!\n", strings_upper[i].out, dst); - kfree(dst); - return; - } kfree(dst); } for (i = 0; i < ARRAY_SIZE(strings_lower); i++) { const char *s = strings_lower[i].in; int len = strlen(strings_lower[i].in) + 1; + int result; dst = kmalloc(len, GFP_KERNEL); - if (!dst) - return; + KUNIT_ASSERT_NOT_NULL(test, dst); string_lower(dst, s); - if (memcmp(dst, strings_lower[i].out, len)) { + result = memcmp(dst, strings_lower[i].out, len); + KUNIT_EXPECT_EQ(test, 0, result); + if (result) pr_warn("Test 'string_lower failed : : expected %s, got %s!\n", strings_lower[i].out, dst); - kfree(dst); - return; - } kfree(dst); } } -static int __init test_string_helpers_init(void) +static void test_unescape(struct kunit *test) { unsigned int i; - pr_info("Running tests...\n"); for (i = 0; i < UNESCAPE_ALL_MASK + 1; i++) - test_string_unescape("unescape", i, false); - test_string_unescape("unescape inplace", + test_string_unescape(test, "unescape", i, false); + test_string_unescape(test, "unescape inplace", get_random_u32_below(UNESCAPE_ALL_MASK + 1), true); /* Without dictionary */ for (i = 0; i < ESCAPE_ALL_MASK + 1; i++) - test_string_escape("escape 0", escape0, i, TEST_STRING_2_DICT_0); + test_string_escape(test, "escape 0", escape0, i, TEST_STRING_2_DICT_0); /* With dictionary */ for (i = 0; i < ESCAPE_ALL_MASK + 1; i++) - test_string_escape("escape 1", escape1, i, TEST_STRING_2_DICT_1); + test_string_escape(test, "escape 1", escape1, i, TEST_STRING_2_DICT_1); +} - /* Test string_get_size() */ - test_string_get_size(); +static struct kunit_case string_helpers_test_cases[] = { + KUNIT_CASE(test_get_size), + KUNIT_CASE(test_upper_lower), + KUNIT_CASE(test_unescape), + {} +}; - /* Test string upper(), string_lower() */ - test_string_upper_lower(); +static struct kunit_suite string_helpers_test_suite = { + .name = "string_helpers", + .test_cases = string_helpers_test_cases, +}; + +kunit_test_suites(&string_helpers_test_suite); - return -EINVAL; -} -module_init(test_string_helpers_init); MODULE_LICENSE("Dual BSD/GPL");