[utest] implement more test case for rt_memcpy

2 files changed, 114 insertions, 37 deletions

diff --git a/src/klibc/utest/TC_kstdlib.c b/src/klibc/utest/TC_kstdlib.c
deleted file mode 100644
index f22ddc108f..0000000000
--- a/src/klibc/utest/TC_kstdlib.c
+++ /dev/null
@@ -1,37 +0,0 @@
-/*
- * Copyright (c) 2006-2024, RT-Thread Development Team
- *
- * SPDX-License-Identifier: Apache-2.0
- *
- * Change Logs:
- * Date           Author       Notes
- * 2024-12-21     Meco Man     the first version
- */
-
-#include <rtklibc.h>
-#include <utest.h>
-
-static rt_err_t utest_tc_init(void)
-{
-    return RT_EOK;
-}
-
-static rt_err_t utest_tc_cleanup(void)
-{
-    return RT_EOK;
-}
-
-static void TC_rt_memcpy_1(void)
-{
-    const char src[] = "Hello, memcpy!";
-    char dest[20] = {0};
-    rt_memcpy(dest, src, sizeof(src));
-    uassert_true(rt_strcmp(src, dest) == 0);
-}
-
-static void utest_do_tc(void)
-{
-    UTEST_UNIT_RUN(TC_rt_memcpy_1);
-}
-
-UTEST_TC_EXPORT(utest_do_tc, "klibc.kstdlibc", utest_tc_init, utest_tc_cleanup, 1000);
diff --git a/src/klibc/utest/TC_rt_memcpy.c b/src/klibc/utest/TC_rt_memcpy.c
new file mode 100644
index 0000000000..f5cbdaa847
--- /dev/null
+++ b/src/klibc/utest/TC_rt_memcpy.c
@@ -0,0 +1,114 @@
+/*
+ * Copyright (c) 2006-2024, RT-Thread Development Team
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Change Logs:
+ * Date           Author             Notes
+ * 2020-05-06     Phillip Johnston   the first version
+ * 2024-12-24     Meco Man           port to utest
+ */
+
+#include <rtklibc.h>
+#include <utest.h>
+
+static rt_err_t utest_tc_init(void)
+{
+    return RT_EOK;
+}
+
+static rt_err_t utest_tc_cleanup(void)
+{
+    return RT_EOK;
+}
+
+#define N 80 /**< Define the constant N for buffer size as 80 */
+static char buf[512] = {0}; /**< Define a static buffer of 512 bytes, initialized to 0 */
+
+/**
+ * Align a given pointer to a 64-byte boundary.
+ * @param p The pointer to align.
+ * @return The aligned pointer.
+ */
+static void* aligned(void* p)
+{
+    return (void*)(((intptr_t)p + 63) & -64);
+}
+
+/**
+ * Test memory copy with alignment.
+ * @param dalign The alignment offset for the destination buffer.
+ * @param salign The alignment offset for the source buffer.
+ * @param len The length of data to copy.
+ */
+static void test_align(unsigned dalign, unsigned salign, size_t len)
+{
+    char* src = aligned(buf); /**< Source buffer starting address, 64-byte aligned */
+    char* dst = aligned(buf + 128); /**< Destination buffer starting address, 64-byte aligned from buf+128 */
+    char* want = aligned(buf + 256); /**< Expected result buffer starting address, 64-byte aligned from buf+256 */
+    char* p; /**< Pointer to receive the return value of rt_memcpy */
+    unsigned i;
+
+    /** Assert that the source alignment offset plus length does not exceed N */
+    uassert_false(salign + len > N);
+    /** Assert that the destination alignment offset plus length does not exceed N */
+    uassert_false(dalign + len > N);
+
+    /** Initialize all buffers with '#' or ' ' */
+    for(i = 0; i < N; i++)
+    {
+        src[i] = '#';
+        dst[i] = want[i] = ' ';
+    }
+
+    /** Set data in the specified alignment offsets of the source and expected result buffers */
+    for(i = 0; i < len; i++)
+    {
+        src[salign + i] = want[dalign + i] = (char)('0' + i);
+    }
+
+    /** Call rt_memcpy to copy data */
+    p = rt_memcpy(dst + dalign, src + salign, len);
+
+    /** Assert that the return value of rt_memcpy is the pointer to the start of the copied data in the destination buffer */
+    uassert_ptr_equal(p, dst + dalign);
+
+    /** Assert that the content of the destination buffer matches the expected result buffer */
+    for(i = 0; i < N; i++)
+    {
+        uassert_int_equal(dst[i], want[i]);
+    }
+}
+
+/**
+ * Test case to iterate over all possible alignment offsets and length combinations.
+ */
+static void TC_rt_memcpy_align(void)
+{
+    for(unsigned i = 0; i < 16; i++) /**< Iterate over source alignment offsets from 0 to 15 */
+    {
+        for(unsigned j = 0; j < 16; j++) /**< Iterate over destination alignment offsets from 0 to 15 */
+        {
+            for(size_t k = 0; k < 64; k++) /**< Iterate over data lengths from 0 to 63 */
+            {
+                test_align(i, j, k); /**< Call the test_align function */
+            }
+        }
+    }
+}
+
+static void TC_rt_memcpy_str(void)
+{
+    const char src[] = "Hello, memcpy!";
+    char dest[20] = {0};
+    rt_memcpy(dest, src, sizeof(src));
+    uassert_true(rt_strcmp(src, dest) == 0);
+}
+
+static void utest_do_tc(void)
+{
+    UTEST_UNIT_RUN(TC_rt_memcpy_str);
+    UTEST_UNIT_RUN(TC_rt_memcpy_align);
+}
+
+UTEST_TC_EXPORT(utest_do_tc, "klibc.rt_memcpy", utest_tc_init, utest_tc_cleanup, 1000);