1 /*
2  * Copyright (c) 2006-2023, RT-Thread Development Team
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  *
6  * Change Logs:
7  * Date           Author       Notes
8  * 2023-08-17     Shell        test case for aspace_unmap_range
9  */
10 #include "common.h"
11 #include "lwp_user_mm.h"
12 #include <mm_aspace.h>
13 
14 #include <rtthread.h>
15 
16 static void *vaddr = (void *)0x100000000;
17 static size_t existed_size = 0x5000;
18 static char *unmap_start;
19 static size_t unmap_size = 0x2000;
20 static size_t former_vsz;
21 static struct rt_lwp *lwp;
22 static size_t flags = MMF_PREFETCH | MMF_MAP_FIXED;
23 
test_unmap_range_invalid_param(void)24 static void test_unmap_range_invalid_param(void)
25 {
26     rt_mem_obj_t notsupp_object;
27 
28     /* create an existed mapping */
29     former_vsz = rt_aspace_count_vsz(lwp->aspace);
30     uassert_true(!rt_aspace_map(lwp->aspace, &vaddr, existed_size, MMU_MAP_K_RWCB, flags, &rt_mm_dummy_mapper, 0));
31     utest_int_equal(former_vsz + existed_size, rt_aspace_count_vsz(lwp->aspace));
32     former_vsz += existed_size;
33 
34     /* test unaligned vaddr start */
35     unmap_start = (char *)vaddr - 0x1234;
36     utest_int_equal(-RT_EINVAL, rt_aspace_unmap_range(lwp->aspace, unmap_start, unmap_size));
37     utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
38 
39     /* test unaligned size */
40     unmap_size = 0x2000;
41     unmap_start = (char *)vaddr + existed_size - unmap_size;
42     utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, unmap_start, unmap_size - 0x123));
43     utest_int_equal(former_vsz - unmap_size, rt_aspace_count_vsz(lwp->aspace));
44 
45     /* create another mapping binding to mem_obj without proper handler */
46     notsupp_object = rt_mem_obj_create(&rt_mm_dummy_mapper);
47     notsupp_object->on_varea_shrink = RT_NULL;
48 
49     utest_int_equal(
50         RT_EOK,
51         rt_aspace_map(lwp->aspace, (void *)&unmap_start, unmap_size, MMU_MAP_K_RWCB, flags, notsupp_object, 0)
52     );
53 
54     utest_int_equal(-RT_EPERM, rt_aspace_unmap_range(lwp->aspace, unmap_start, 0x1000));
55     utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, vaddr, existed_size));
56     rt_free(notsupp_object);
57 }
58 
aspace_unmap_tc(void)59 static void aspace_unmap_tc(void)
60 {
61     CONSIST_HEAP(test_unmap_range_invalid_param());
62 }
63 
utest_tc_init(void)64 static rt_err_t utest_tc_init(void)
65 {
66     lwp = lwp_create(0);
67     if (lwp)
68         lwp_user_space_init(lwp, 1);
69     else
70         return -RT_ENOMEM;
71     return RT_EOK;
72 }
73 
utest_tc_cleanup(void)74 static rt_err_t utest_tc_cleanup(void)
75 {
76     lwp_ref_dec(lwp);
77     return RT_EOK;
78 }
79 
testcase(void)80 static void testcase(void)
81 {
82     UTEST_UNIT_RUN(aspace_unmap_tc);
83 }
84 UTEST_TC_EXPORT(testcase, "testcases.mm.aspace_unmap_range.invalid_param", utest_tc_init, utest_tc_cleanup, 10);
85