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-03-27 WangXiaoyao testcase for lwp
9 */
10
11 #include "common.h"
12 #include <lwp.h>
13 #include "lwp_arch.h"
14 #include "lwp_user_mm.h"
15 #include "mm_aspace.h"
16 #include "mm_flag.h"
17 #include "mmu.h"
18
19 /**
20 * @brief user map API
21 * rt_varea_t lwp_map_user_varea(struct rt_lwp *lwp, void *map_va, size_t map_size);
22 * rt_varea_t lwp_map_user_varea_ext(struct rt_lwp *lwp, void *map_va, size_t map_size, size_t flags);
23 */
24 #if 1 /* make it clear to identify the block :) */
25 /* for testing on _aspace_traverse */
26 static void *_prev_end;
27 static size_t _count;
_test_increase(rt_varea_t varea,void * param)28 static int _test_increase(rt_varea_t varea, void *param)
29 {
30 uassert_true(varea->start >= _prev_end);
31 _prev_end = varea->start + varea->size;
32 _count += 1;
33 return 0;
34 }
35 #define TEST_VAREA_INSERT(statement, aspace) do {\
36 size_t _prev_count; \
37 _count = 0; \
38 _prev_end = 0; \
39 rt_aspace_traversal((aspace), _test_increase, NULL);\
40 _prev_count = _count; \
41 statement; \
42 _count = 0; \
43 _prev_end = 0; \
44 rt_aspace_traversal((aspace), _test_increase, NULL);\
45 uassert_true(_prev_count + 1 == _count); \
46 } while (0)
47 #endif
48
test_user_map_varea(void)49 static void test_user_map_varea(void)
50 {
51 const size_t buf_sz = ARCH_PAGE_SIZE * 4;
52 struct rt_lwp *lwp;
53 rt_varea_t varea;
54 lwp = lwp_create(LWP_CREATE_FLAG_NONE);
55
56 /* prepare environment */
57 uassert_true(!!lwp);
58 uassert_true(!lwp_user_space_init(lwp, 1));
59 TEST_VAREA_INSERT(
60 varea = lwp_map_user_varea(lwp, 0, buf_sz),
61 lwp->aspace);
62 uassert_true(!!varea);
63 uassert_true(varea->attr == (MMU_MAP_U_RWCB));
64 uassert_true(varea->size == buf_sz);
65 uassert_true(varea->aspace == lwp->aspace);
66 uassert_true(varea->flag == MMF_MAP_PRIVATE);
67 uassert_true(varea->start != 0);
68 uassert_true(varea->start >= (void *)USER_VADDR_START && varea->start < (void *)USER_VADDR_TOP);
69
70 uassert_true(!(lwp_ref_dec(lwp) - 1));
71 }
72
test_user_map_varea_ext(void)73 static void test_user_map_varea_ext(void)
74 {
75 const size_t buf_sz = ARCH_PAGE_SIZE * 4;
76 struct rt_lwp *lwp;
77 rt_varea_t varea;
78 lwp = lwp_create(LWP_CREATE_FLAG_NONE);
79
80 uassert_true(!!lwp);
81 uassert_true(!lwp_user_space_init(lwp, 1));
82
83 TEST_VAREA_INSERT(
84 varea = lwp_map_user_varea_ext(lwp, 0, buf_sz, LWP_MAP_FLAG_NOCACHE),
85 lwp->aspace);
86 uassert_true(!!varea);
87 uassert_true(varea->attr == (MMU_MAP_U_RW));
88 uassert_true(varea->size == buf_sz);
89 uassert_true(varea->aspace == lwp->aspace);
90 uassert_true(varea->flag == MMF_MAP_PRIVATE);
91 uassert_true(varea->start != 0);
92 uassert_true(varea->start >= (void *)USER_VADDR_START && varea->start < (void *)USER_VADDR_TOP);
93
94 uassert_true(!(lwp_ref_dec(lwp) - 1));
95 }
96
user_map_varea_tc(void)97 static void user_map_varea_tc(void)
98 {
99 CONSIST_HEAP(test_user_map_varea());
100 CONSIST_HEAP(test_user_map_varea_ext());
101 }
102
test_user_accessible(void)103 static void test_user_accessible(void)
104 {
105 /* Prepare Environment */
106 char *test_address = (char *)(USER_STACK_VEND);
107 struct rt_lwp *lwp;
108 lwp = lwp_create(LWP_CREATE_FLAG_NONE);
109 uassert_true(!!lwp);
110 uassert_true(!lwp_user_space_init(lwp, 0));
111
112 /* test if user accessible can operate */
113 uassert_true(!lwp_user_accessible_ext(lwp, test_address + 0x1, 0x1));
114 /* test if mapping exist, accessible can fill the page and return True */
115 uassert_true(lwp_user_accessible_ext(lwp, test_address - 0x10, 0x10));
116
117 /* Cleanup */
118 lwp_ref_dec(lwp);
119 }
120
accessible_tc(void)121 static void accessible_tc(void)
122 {
123 CONSIST_HEAP(test_user_accessible());
124 }
125
utest_tc_init(void)126 static rt_err_t utest_tc_init(void)
127 {
128 return RT_EOK;
129 }
130
utest_tc_cleanup(void)131 static rt_err_t utest_tc_cleanup(void)
132 {
133 return RT_EOK;
134 }
135
testcase(void)136 static void testcase(void)
137 {
138 UTEST_UNIT_RUN(user_map_varea_tc);
139 UTEST_UNIT_RUN(accessible_tc);
140 }
141 UTEST_TC_EXPORT(testcase, "testcases.lwp.mm_tc", utest_tc_init, utest_tc_cleanup, 20);
142