Rev 3232 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
2465 | Serge | 1 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
2 | ;; ;; |
||
3 | ;; Copyright (C) KolibriOS team 2009-2011. All rights reserved. ;; |
||
4 | ;; Distributed under terms of the GNU General Public License ;; |
||
5 | ;; ;; |
||
6 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
||
7 | |||
1333 | diamond | 8 | ; Tests of malloc()/free() from the kernel heap. |
9 | ; This file is not included in the kernel, it is just test application. |
||
10 | use32 |
||
3908 | Serge | 11 | db 'MENUET01' |
12 | dd 1, start, i_end, mem, mem, 0, 0 |
||
1333 | diamond | 13 | |
14 | start: |
||
15 | ; Zero-initialize uglobals (as in kernel at boot) |
||
3908 | Serge | 16 | mov ecx, (zeroend - zerostart + 3) / 4 |
17 | xor eax, eax |
||
18 | mov edi, zerostart |
||
19 | rep stosd |
||
1333 | diamond | 20 | ; Initialize small heap (as in kernel at boot) |
3908 | Serge | 21 | call init_malloc |
1333 | diamond | 22 | ; Run tests |
3908 | Serge | 23 | call run_test1 |
24 | call run_test2 |
||
25 | call run_test3 |
||
1333 | diamond | 26 | ; All is OK, return |
3908 | Serge | 27 | or eax, -1 |
28 | int 0x40 |
||
1333 | diamond | 29 | |
30 | run_test1: |
||
31 | ; basic test |
||
3908 | Serge | 32 | mov eax, 1 |
33 | call malloc_with_test |
||
34 | mov byte [eax], 0xDD |
||
35 | mov esi, eax |
||
36 | mov eax, 1 |
||
37 | call malloc_with_test |
||
38 | cmp byte [esi], 0xDD |
||
39 | jnz memory_destroyed |
||
40 | mov byte [eax], 0xEE |
||
41 | xchg eax, esi |
||
42 | call free |
||
43 | cmp byte [esi], 0xEE |
||
44 | jnz memory_destroyed |
||
45 | xchg eax, esi |
||
46 | call free |
||
47 | ret |
||
1333 | diamond | 48 | |
49 | run_test2: |
||
3908 | Serge | 50 | ret |
1333 | diamond | 51 | |
52 | run_test3: |
||
3232 | Serge | 53 | ; 1024 times run random operation. |
1333 | diamond | 54 | ; Randomly select malloc(random size from 1 to 1023) |
55 | ; or free(random of previously allocated areas) |
||
3908 | Serge | 56 | mov edi, 0x12345678 |
57 | xor esi, esi ; 0 areas allocated |
||
3232 | Serge | 58 | mov ebx, 1024 |
1333 | diamond | 59 | .loop: |
3908 | Serge | 60 | imul edi, 1103515245 |
61 | add edi, 12345 |
||
62 | mov eax, edi |
||
63 | shr eax, 16 |
||
64 | test ebx, 64 |
||
65 | jz .prefer_free |
||
1333 | diamond | 66 | .prefer_malloc: |
3908 | Serge | 67 | test eax, 3 |
68 | jz .free |
||
69 | jmp @f |
||
1333 | diamond | 70 | .prefer_free: |
3908 | Serge | 71 | test eax, 3 |
72 | jnz .free |
||
1333 | diamond | 73 | @@: |
3908 | Serge | 74 | shr eax, 2 |
75 | and eax, 1023 |
||
76 | jz .loop |
||
77 | push ebx |
||
78 | push eax |
||
79 | ; mov ecx, [saved_state_num] |
||
80 | ; mov [saved_state+ecx*8], eax |
||
3232 | Serge | 81 | push edi |
3908 | Serge | 82 | call malloc_with_test |
3232 | Serge | 83 | pop ecx |
84 | cmp ecx, edi |
||
85 | jnz edi_destroyed |
||
3908 | Serge | 86 | ; mov ecx, [saved_state_num] |
87 | ; mov [saved_state+ecx*8+4], eax |
||
88 | ; inc [saved_state_num] |
||
89 | pop ecx |
||
90 | pop ebx |
||
91 | inc esi |
||
92 | push ecx eax |
||
93 | push edi |
||
94 | mov edi, eax |
||
95 | mov eax, esi |
||
96 | rep stosb |
||
97 | pop edi |
||
98 | jmp .common |
||
1333 | diamond | 99 | .free: |
3908 | Serge | 100 | test esi, esi |
101 | jz .loop |
||
102 | xor edx, edx |
||
103 | div esi |
||
104 | sub edx, esi |
||
105 | neg edx |
||
106 | dec edx |
||
107 | mov eax, [esp+edx*8] |
||
108 | ; mov ecx, [saved_state_num] |
||
109 | ; mov [saved_state+ecx*8], -1 |
||
110 | ; mov [saved_state+ecx*8+4], eax |
||
111 | ; inc [saved_state_num] |
||
112 | mov ecx, [esp+edx*8+4] |
||
113 | push edi eax |
||
114 | mov edi, eax |
||
115 | mov al, [edi] |
||
116 | repz scasb |
||
117 | jnz memory_destroyed |
||
118 | pop eax edi |
||
119 | push ebx edx |
||
3232 | Serge | 120 | push edi |
3908 | Serge | 121 | call free |
3232 | Serge | 122 | pop ecx |
123 | cmp ecx, edi |
||
124 | jnz edi_destroyed |
||
3908 | Serge | 125 | pop edx ebx |
126 | dec esi |
||
127 | pop eax ecx |
||
128 | push edi |
||
129 | lea edi, [esp+4] |
||
1333 | diamond | 130 | @@: |
3908 | Serge | 131 | dec edx |
132 | js @f |
||
133 | xchg eax, [edi] |
||
134 | xchg ecx, [edi+4] |
||
135 | add edi, 8 |
||
136 | jmp @b |
||
1333 | diamond | 137 | @@: |
3908 | Serge | 138 | pop edi |
1333 | diamond | 139 | .common: |
3908 | Serge | 140 | dec ebx |
141 | jnz .loop |
||
1333 | diamond | 142 | @@: |
3908 | Serge | 143 | dec esi |
144 | js @f |
||
145 | pop eax ecx |
||
146 | call free |
||
147 | jmp @b |
||
1333 | diamond | 148 | @@: |
3908 | Serge | 149 | ret |
1333 | diamond | 150 | |
151 | malloc_with_test: |
||
152 | ; calls malloc() and checks returned value |
||
3908 | Serge | 153 | call malloc |
154 | test eax, eax |
||
155 | jz generic_malloc_fail |
||
156 | call check_mutex |
||
157 | call check_range |
||
158 | ret |
||
1333 | diamond | 159 | |
160 | ; Stubs for kernel procedures used by heap code |
||
3232 | Serge | 161 | mutex_init: |
162 | and dword [ecx], 0 |
||
163 | ret |
||
164 | mutex_lock: |
||
165 | inc dword [ecx] |
||
3908 | Serge | 166 | ret |
3232 | Serge | 167 | mutex_unlock: |
168 | dec dword [ecx] |
||
169 | ret |
||
1333 | diamond | 170 | |
171 | kernel_alloc: |
||
3908 | Serge | 172 | cmp dword [esp+4], bufsize |
173 | jnz error1 |
||
174 | mov eax, buffer |
||
175 | ret 4 |
||
1333 | diamond | 176 | |
177 | macro $Revision [args] |
||
178 | { |
||
179 | } |
||
180 | |||
181 | ; Error handlers |
||
182 | error1: |
||
3908 | Serge | 183 | mov eax, 1 |
184 | jmp error_with_code |
||
1333 | diamond | 185 | |
186 | generic_malloc_fail: |
||
3908 | Serge | 187 | mov eax, 2 |
188 | jmp error_with_code |
||
1333 | diamond | 189 | |
190 | check_mutex: |
||
3232 | Serge | 191 | cmp dword [mst.mutex], 0 |
3908 | Serge | 192 | jnz @f |
193 | ret |
||
1333 | diamond | 194 | @@: |
3908 | Serge | 195 | mov eax, 3 |
196 | jmp error_with_code |
||
1333 | diamond | 197 | |
198 | check_range: |
||
3908 | Serge | 199 | cmp eax, buffer |
200 | jb @f |
||
201 | cmp eax, buffer+bufsize |
||
202 | jae @f |
||
203 | ret |
||
1333 | diamond | 204 | @@: |
3908 | Serge | 205 | mov eax, 4 |
206 | jmp error_with_code |
||
1333 | diamond | 207 | |
208 | memory_destroyed: |
||
3908 | Serge | 209 | mov eax, 5 |
210 | jmp error_with_code |
||
1333 | diamond | 211 | |
3232 | Serge | 212 | edi_destroyed: |
213 | mov eax, 6 |
||
214 | jmp error_with_code |
||
215 | |||
1333 | diamond | 216 | error_with_code: |
3908 | Serge | 217 | mov edx, saved_state_num |
1333 | diamond | 218 | ; eax = error code |
219 | ; 1 signals error in testing code (wrong bufsize) |
||
220 | ; 2 = malloc() returned NULL |
||
221 | ; 3 = mutex not released |
||
222 | ; 4 = weird returned value from malloc() |
||
223 | ; 5 = memory destroyed by malloc() or free() |
||
3908 | Serge | 224 | int3 ; simplest way to report error |
225 | jmp $-1 ; just in case |
||
1333 | diamond | 226 | |
227 | ; Include main heap code |
||
228 | include '../proc32.inc' |
||
3232 | Serge | 229 | include '../struct.inc' |
1333 | diamond | 230 | include '../const.inc' |
231 | include 'malloc.inc' |
||
232 | |||
233 | i_end: |
||
234 | |||
235 | align 4 |
||
236 | zerostart: |
||
3908 | Serge | 237 | mst MEM_STATE |
1333 | diamond | 238 | |
239 | align 16 |
||
3908 | Serge | 240 | bufsize = 0x40000 ; change if malloc.inc changes |
241 | buffer rb bufsize |
||
1333 | diamond | 242 | zeroend: |
243 | |||
3908 | Serge | 244 | saved_state_num dd ? |
245 | saved_state rd 0x10000 |
||
1333 | diamond | 246 | |
247 | align 4 |
||
3908 | Serge | 248 | rb 0x10000 ; for stack |
1333 | diamond | 249 | mem: |