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 char *unmap_end;
20 static size_t former_vsz;
21 static size_t unmap_size = 0x2000;
22 static struct rt_lwp *lwp;
23
test_unmap_range_shrink(void)24 static void test_unmap_range_shrink(void)
25 {
26 /* create an existed mapping */
27 former_vsz = rt_aspace_count_vsz(lwp->aspace);
28 uassert_true(!rt_aspace_map(lwp->aspace, &vaddr, existed_size, MMU_MAP_K_RWCB, MMF_PREFETCH, &rt_mm_dummy_mapper, 0));
29 utest_int_equal(former_vsz + existed_size, rt_aspace_count_vsz(lwp->aspace));
30 former_vsz += existed_size;
31
32 /* test the shrink mode of unmap from LEFT side */
33 unmap_start = (char *)vaddr - unmap_size/2;
34 uassert_true(!rt_aspace_unmap_range(lwp->aspace, unmap_start, unmap_size));
35 unmap_end = unmap_start + unmap_size;
36 uassert_true(rt_hw_mmu_v2p(lwp->aspace, unmap_end) != ARCH_MAP_FAILED);
37 utest_int_equal(former_vsz - (unmap_end - (char *)vaddr), rt_aspace_count_vsz(lwp->aspace));
38 former_vsz -= unmap_end - (char *)vaddr;
39
40 /* test the shrink mode of unmap from RIGHT side */
41 unmap_start = (char *)vaddr + existed_size - unmap_size / 2;
42 uassert_true(!rt_aspace_unmap_range(lwp->aspace, unmap_start, unmap_size));
43 uassert_true(rt_hw_mmu_v2p(lwp->aspace, unmap_start - 1) != ARCH_MAP_FAILED);
44 utest_int_equal(former_vsz - (unmap_end - (char *)vaddr), rt_aspace_count_vsz(lwp->aspace));
45 former_vsz -= unmap_end - (char *)vaddr;
46
47 utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, vaddr, existed_size));
48 }
49
aspace_unmap_tc(void)50 static void aspace_unmap_tc(void)
51 {
52 CONSIST_HEAP(test_unmap_range_shrink());
53 }
54
utest_tc_init(void)55 static rt_err_t utest_tc_init(void)
56 {
57 lwp = lwp_create(0);
58 if (lwp)
59 lwp_user_space_init(lwp, 1);
60 else
61 return -RT_ENOMEM;
62 return RT_EOK;
63 }
64
utest_tc_cleanup(void)65 static rt_err_t utest_tc_cleanup(void)
66 {
67 lwp_ref_dec(lwp);
68 return RT_EOK;
69 }
70
testcase(void)71 static void testcase(void)
72 {
73 UTEST_UNIT_RUN(aspace_unmap_tc);
74 }
75 UTEST_TC_EXPORT(testcase, "testcases.mm.aspace_unmap_range.shrink", utest_tc_init, utest_tc_cleanup, 10);
76