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_map(MAP_FIXED)
9 */
10 #include "common.h"
11 #include "lwp_user_mm.h"
12 #include "utest_assert.h"
13 #include <mm_aspace.h>
14
15 #include <rtthread.h>
16
17 static struct rt_lwp *lwp;
18 static size_t former_vsz;
19 static size_t former_vcount;
20
21 static void *vaddr = (void *)0x100000000;
22 static size_t ex_size = 0x5000;
23 static char *override_start;
24 static size_t override_size = 0x2000;
25 static long fd = -1;
26 static long pgoffset = 0;
27 static size_t ex_prot = PROT_NONE;
28 static size_t ex_flags = MAP_PRIVATE | MAP_ANONYMOUS;
29 static size_t override_prot = PROT_READ | PROT_WRITE;
30 static size_t override_flags = MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED;
31
_count_vsz(rt_varea_t varea,void * arg)32 static int _count_vsz(rt_varea_t varea, void *arg)
33 {
34 rt_base_t *pvsz = arg;
35 *pvsz += 1;
36 return 0;
37 }
38
count_vcount(rt_aspace_t aspace)39 static rt_base_t count_vcount(rt_aspace_t aspace)
40 {
41 rt_base_t vcount = 0;
42 rt_aspace_traversal(aspace, _count_vsz, &vcount);
43 return vcount;
44 }
45
46 static char put_data[] = "hello,world";
47
test_map_fixed(void)48 static void test_map_fixed(void)
49 {
50 void *effect_override;
51
52 former_vsz = rt_aspace_count_vsz(lwp->aspace);
53 former_vcount = count_vcount(lwp->aspace);
54
55 /* create an existed mapping */
56 vaddr = lwp_mmap2(lwp, vaddr, ex_size, ex_prot, ex_flags, fd, pgoffset);
57 uassert_true((long)vaddr > 0);
58 utest_int_equal(former_vsz + ex_size, rt_aspace_count_vsz(lwp->aspace));
59 utest_int_equal(former_vcount + 1, count_vcount(lwp->aspace));
60 former_vsz += ex_size;
61 former_vcount += 1;
62
63 /* fix private in the middle */
64 override_start = (char *)vaddr + 0x1000;
65 effect_override = lwp_mmap2(lwp, override_start, override_size, override_prot, override_flags, fd, pgoffset);
66 uassert_true(effect_override == override_start);
67 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
68 utest_int_equal(former_vcount + 2, count_vcount(lwp->aspace));
69 utest_int_equal(
70 lwp_data_put(lwp, effect_override, put_data, sizeof(put_data)),
71 sizeof(put_data)
72 );
73
74 utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, vaddr, ex_size));
75 }
76
aspace_unmap_tc(void)77 static void aspace_unmap_tc(void)
78 {
79 test_map_fixed();
80 }
81
82 static rt_size_t total, used, max_used;
83 static rt_size_t totala, useda, max_useda;
84 static rt_ubase_t level;
utest_tc_init(void)85 static rt_err_t utest_tc_init(void)
86 {
87 lwp = lwp_create(0);
88 if (lwp)
89 lwp_user_space_init(lwp, 1);
90 else
91 return -RT_ENOMEM;
92
93 /* stats */
94 level = rt_heap_lock();
95 rt_memory_info(&total, &used, &max_used);
96
97 return RT_EOK;
98 }
99
utest_tc_cleanup(void)100 static rt_err_t utest_tc_cleanup(void)
101 {
102 lwp_ref_dec(lwp);
103
104 /* check */
105 rt_memory_info(&totala, &useda, &max_useda);
106 rt_heap_unlock(level);
107 utest_int_equal(total, totala);
108 utest_int_less_equal(useda, used);
109
110 return RT_EOK;
111 }
112
testcase(void)113 static void testcase(void)
114 {
115 UTEST_UNIT_RUN(aspace_unmap_tc);
116 }
117 UTEST_TC_EXPORT(testcase, "testcases.lwp.mman.mmap_anon.fix_private", utest_tc_init, utest_tc_cleanup, 10);
118