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_private.h>
14
15 #include <rtthread.h>
16
17 #define PAGE_SZ (1 << MM_PAGE_SHIFT)
18 #define PAGE_COUNT ('z' - 'a' + 1)
19 #define FILE_PATH "/test_mmap"
20 #define FILE_SZ (PAGE_COUNT * PAGE_SZ)
21
22 static struct rt_lwp *lwp;
23 static size_t former_vsz;
24 static size_t former_vcount;
25 static char page_sz_buf[PAGE_SZ];
26
27 static void *vaddr = (void *)0x100000000;
28 static long pgoffset = 0;
29 static size_t ex_prot = PROT_NONE;
30 static size_t ex_flags = MAP_PRIVATE | MAP_ANONYMOUS;
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
_lwp_get_user(struct rt_lwp * lwp,char * vaddr,char * buffer)46 static rt_err_t _lwp_get_user(struct rt_lwp *lwp, char *vaddr, char *buffer)
47 {
48 rt_varea_t varea = _aspace_bst_search(lwp->aspace, vaddr);
49 if (varea && varea->mem_obj && varea->mem_obj->page_read)
50 {
51 struct rt_aspace_io_msg io_msg;
52 rt_mm_io_msg_init(&io_msg, MM_PA_TO_OFF(vaddr), vaddr, buffer);
53 varea->mem_obj->page_read(varea, &io_msg);
54 }
55 return RT_EOK;
56 }
57
_verify_file_content(const char * mmap_buf)58 static void _verify_file_content(const char *mmap_buf)
59 {
60 char ch = 'a';
61 for (char *read_va = (char *)mmap_buf; read_va < mmap_buf + FILE_SZ; read_va += PAGE_SZ, ch++)
62 {
63 _lwp_get_user(lwp, read_va, page_sz_buf);
64 utest_int_equal(RT_EOK, memtest(page_sz_buf, ch, PAGE_SZ));
65 }
66 }
67
test_mmap_fd(void)68 static void test_mmap_fd(void)
69 {
70 former_vsz = rt_aspace_count_vsz(lwp->aspace);
71 former_vcount = count_vcount(lwp->aspace);
72
73 /* create an existed mapping */
74 long temp_fd;
75 temp_fd = open(FILE_PATH, O_RDONLY);
76 LOG_D("New fd=%ld path=%s", temp_fd, FILE_PATH);
77 uassert_true(temp_fd >= 0);
78 utest_int_equal(
79 lwp_mmap2(lwp, vaddr, FILE_SZ, ex_prot, ex_flags, temp_fd, pgoffset),
80 vaddr);
81 close(temp_fd);
82
83 utest_int_equal(former_vsz + FILE_SZ, rt_aspace_count_vsz(lwp->aspace));
84 utest_int_equal(former_vcount + 1, count_vcount(lwp->aspace));
85
86 _verify_file_content(vaddr);
87
88 /* create an override mapping */
89
90 /* close */
91
92 utest_int_equal(RT_EOK, rt_aspace_unmap_range(lwp->aspace, vaddr, FILE_SZ));
93 }
94
testcase_main(void)95 static void testcase_main(void)
96 {
97 test_mmap_fd();
98 }
99
_setup_file_content(long fd)100 static void _setup_file_content(long fd)
101 {
102 char ch = 'a';
103
104 for (size_t i = 0; i < PAGE_COUNT; i++, ch++)
105 {
106 memset(page_sz_buf, ch, PAGE_SZ);
107 write(fd, page_sz_buf, PAGE_SZ);
108 }
109 }
110
utest_tc_init(void)111 static rt_err_t utest_tc_init(void)
112 {
113 /* setup file */
114 long temp_file_des;
115 temp_file_des = open(FILE_PATH, O_RDWR | O_CREAT, 0777);
116 LOG_D("New fd=%ld path=%s", temp_file_des, FILE_PATH);
117 if (temp_file_des < 0)
118 return -RT_ERROR;
119 _setup_file_content(temp_file_des);
120 close(temp_file_des);
121
122 lwp = lwp_create(0);
123 if (lwp)
124 lwp_user_space_init(lwp, 1);
125 else
126 return -RT_ENOMEM;
127 return RT_EOK;
128 }
129
utest_tc_cleanup(void)130 static rt_err_t utest_tc_cleanup(void)
131 {
132 lwp_ref_dec(lwp);
133 return RT_EOK;
134 }
135
testcase(void)136 static void testcase(void)
137 {
138 UTEST_UNIT_RUN(testcase_main);
139 }
140 UTEST_TC_EXPORT(testcase, "testcases.lwp.mman.mmap_fd.basic", utest_tc_init, utest_tc_cleanup, 10);
141