The alloc/copy code pattern is better consolidated to single kstrdup (and
kstrndup) calls instead. This gets rid of deprecated[1] strncpy() uses as
well. Replace one other strncpy() use with the more idiomatic strscpy().

Link: https://github.com/KSPP/linux/issues/90 [1]
Reviewed-by: David Gow <david...@google.com>
Signed-off-by: Kees Cook <k...@kernel.org>
---
Cc: Brendan Higgins <brendan.higg...@linux.dev>
Cc: David Gow <david...@google.com>
Cc: Rae Moar <rm...@google.com>
Cc: linux-kselftest@vger.kernel.org
Cc: kunit-...@googlegroups.com
---
 lib/kunit/executor.c      | 12 +++---------
 lib/kunit/executor_test.c |  2 +-
 2 files changed, 4 insertions(+), 10 deletions(-)

diff --git a/lib/kunit/executor.c b/lib/kunit/executor.c
index 70b9a43cd257..34b7b6833df3 100644
--- a/lib/kunit/executor.c
+++ b/lib/kunit/executor.c
@@ -70,32 +70,26 @@ struct kunit_glob_filter {
 static int kunit_parse_glob_filter(struct kunit_glob_filter *parsed,
                                    const char *filter_glob)
 {
-       const int len = strlen(filter_glob);
        const char *period = strchr(filter_glob, '.');
 
        if (!period) {
-               parsed->suite_glob = kzalloc(len + 1, GFP_KERNEL);
+               parsed->suite_glob = kstrdup(filter_glob, GFP_KERNEL);
                if (!parsed->suite_glob)
                        return -ENOMEM;
-
                parsed->test_glob = NULL;
-               strcpy(parsed->suite_glob, filter_glob);
                return 0;
        }
 
-       parsed->suite_glob = kzalloc(period - filter_glob + 1, GFP_KERNEL);
+       parsed->suite_glob = kstrndup(filter_glob, period - filter_glob, 
GFP_KERNEL);
        if (!parsed->suite_glob)
                return -ENOMEM;
 
-       parsed->test_glob = kzalloc(len - (period - filter_glob) + 1, 
GFP_KERNEL);
+       parsed->test_glob = kstrdup(period + 1, GFP_KERNEL);
        if (!parsed->test_glob) {
                kfree(parsed->suite_glob);
                return -ENOMEM;
        }
 
-       strncpy(parsed->suite_glob, filter_glob, period - filter_glob);
-       strncpy(parsed->test_glob, period + 1, len - (period - filter_glob));
-
        return 0;
 }
 
diff --git a/lib/kunit/executor_test.c b/lib/kunit/executor_test.c
index 3f7f967e3688..f0090c2729cd 100644
--- a/lib/kunit/executor_test.c
+++ b/lib/kunit/executor_test.c
@@ -286,7 +286,7 @@ static struct kunit_suite *alloc_fake_suite(struct kunit 
*test,
 
        /* We normally never expect to allocate suites, hence the non-const 
cast. */
        suite = kunit_kzalloc(test, sizeof(*suite), GFP_KERNEL);
-       strncpy((char *)suite->name, suite_name, sizeof(suite->name) - 1);
+       strscpy((char *)suite->name, suite_name, sizeof(suite->name));
        suite->test_cases = test_cases;
 
        return suite;
-- 
2.34.1


Reply via email to