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-22 Shell test case for aspace_map with varea_expand
9 */
10 #include "common.h"
11 #include "lwp_user_mm.h"
12 #include "mm_fault.h"
13 #include <mm_aspace.h>
14
15 #include <rtthread.h>
16
17 static long fd = -1;
18 static long pgoffset = 0;
19 static size_t flags = MAP_FIXED | MAP_ANONYMOUS;
20 static size_t prot = PROT_READ | PROT_WRITE;
21
22 static char *ex_vaddr = (char *)0x100000000;
23 static size_t ex_size = 0x5000;
24 static char *private0 = (char *)0x100000000;
25 static char *private1 = (char *)0x100000000 + 0x1000;
26 static char *private2 = (char *)0x100000000 + 0x2000;
27 static char *private3 = (char *)0x100000000 + 0x3000;
28 static char *private4 = (char *)0x100000000 + 0x4000;
29
30 /**
31 * todo: suppoprt prefetch pages, so more than 1 page can install to private at a time
32 * static size_t priv_size = 0x1000;
33 */
34
35 static size_t former_vsz;
36 static size_t former_vcount;
37
38 static struct rt_lwp *lwp;
39
_count_vsz(rt_varea_t varea,void * arg)40 static int _count_vsz(rt_varea_t varea, void *arg)
41 {
42 rt_base_t *pvsz = arg;
43 *pvsz += 1;
44 return 0;
45 }
46
count_vcount(rt_aspace_t aspace)47 static rt_base_t count_vcount(rt_aspace_t aspace)
48 {
49 rt_base_t vcount = 0;
50 rt_aspace_traversal(aspace, _count_vsz, &vcount);
51 return vcount;
52 }
53
test_mmap_fix_private(void)54 static void test_mmap_fix_private(void)
55 {
56 char *next_va;
57 struct rt_aspace_fault_msg msg;
58 msg.fault_op = MM_FAULT_OP_WRITE;
59 msg.fault_type = MM_FAULT_TYPE_RWX_PERM;
60
61 /* map new pages at ex_vaddr to anonymous */
62 next_va = ex_vaddr;
63 former_vsz = rt_aspace_count_vsz(lwp->aspace);
64 former_vcount = count_vcount(lwp->aspace);
65 next_va = lwp_mmap2(lwp, next_va, ex_size, prot, flags, fd, pgoffset);
66 uassert_true(next_va == ex_vaddr);
67 utest_int_equal(former_vsz + ex_size, rt_aspace_count_vsz(lwp->aspace));
68 utest_int_equal(former_vcount + 1, count_vcount(lwp->aspace));
69 former_vsz += ex_size;
70 former_vcount += 1;
71
72 /* fix private in the middle */
73 msg.fault_vaddr = private2;
74 utest_int_equal(MM_FAULT_FIXABLE_TRUE, rt_aspace_fault_try_fix(lwp->aspace, &msg));
75 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
76 utest_int_equal(former_vcount + 2, count_vcount(lwp->aspace));
77 former_vcount += 2;
78
79 /* fix private from left most */
80 msg.fault_vaddr = private0;
81 utest_int_equal(MM_FAULT_FIXABLE_TRUE, rt_aspace_fault_try_fix(lwp->aspace, &msg));
82 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
83 utest_int_equal(former_vcount + 1, count_vcount(lwp->aspace));
84 former_vcount += 1;
85
86 /* fix private from right most */
87 msg.fault_vaddr = private4;
88 utest_int_equal(MM_FAULT_FIXABLE_TRUE, rt_aspace_fault_try_fix(lwp->aspace, &msg));
89 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
90 utest_int_equal(former_vcount + 1, count_vcount(lwp->aspace));
91 former_vcount += 1;
92
93 /* fix private from left-middle */
94 msg.fault_vaddr = private1;
95 utest_int_equal(MM_FAULT_FIXABLE_TRUE, rt_aspace_fault_try_fix(lwp->aspace, &msg));
96 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
97 utest_int_equal(former_vcount - 1, count_vcount(lwp->aspace));
98 former_vcount -= 1;
99
100 /* fix private from right-middle */
101 msg.fault_vaddr = private3;
102 utest_int_equal(MM_FAULT_FIXABLE_TRUE, rt_aspace_fault_try_fix(lwp->aspace, &msg));
103 utest_int_equal(former_vsz, rt_aspace_count_vsz(lwp->aspace));
104 utest_int_equal(former_vcount - 1, count_vcount(lwp->aspace));
105 former_vcount -= 1;
106
107 /* clear mapping */
108 utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, ex_vaddr, ex_size));
109 rt_free(lwp->aspace->private_object);
110 lwp->aspace->private_object = RT_NULL;
111 }
112
testcase_main(void)113 static void testcase_main(void)
114 {
115 CONSIST_HEAP(test_mmap_fix_private());
116 }
117
utest_tc_init(void)118 static rt_err_t utest_tc_init(void)
119 {
120 lwp = lwp_create(0);
121 if (lwp)
122 lwp_user_space_init(lwp, 1);
123 else
124 return -RT_ENOMEM;
125 return RT_EOK;
126 }
127
utest_tc_cleanup(void)128 static rt_err_t utest_tc_cleanup(void)
129 {
130 lwp_ref_dec(lwp);
131 return RT_EOK;
132 }
133
testcase(void)134 static void testcase(void)
135 {
136 UTEST_UNIT_RUN(testcase_main);
137 }
138 UTEST_TC_EXPORT(testcase, "testcases.lwp.mman.mmap_anon.fix_private", utest_tc_init, utest_tc_cleanup, 10);
139