Subversion Repositories Kolibri OS

           mov [mem_used_list], eax

           mov [mem_used_list], eax

           mov [mem_block_list+63*4], ebx

           mov [mem_block_list+63*4], ebx

           mov byte [mem_block_map], 0xFC

           mov byte [mem_block_map], 0xFC

           mov [heap_blocks], 4095

           mov [heap_blocks], 4095

           mov [free_blocks], 4095

           mov [free_blocks], 4095

           ret

           ret

endp

endp

align 4

align 4

proc alloc_mem_block

proc alloc_mem_block

           cli

           mov ebx, [mem_block_start]

           mov ebx, [mem_block_start]

           mov ecx, [mem_block_end]

           mov ecx, [mem_block_end]

.l1:

.l1:

           bsf eax,[ebx];

           bsf eax,[ebx];

           jnz found

           jnz found

           add ebx,4

           add ebx,4

           cmp ebx, ecx

           cmp ebx, ecx

           popfd

           jb .l1

           xor eax,eax

           xor eax,eax

           ret

           ret

           lea eax,[eax+ebx*8]

           lea eax,[eax+ebx*8]

           shl eax, 5

           shl eax, 5

           add eax, [mem_block_arr]

           add eax, [mem_block_arr]

           dec [free_blocks]

           dec [free_blocks]

           popfd

           ret

endp

endp

proc free_mem_block

proc free_mem_block

           sub eax, [mem_block_arr]

           sub eax, [mem_block_arr]

           shr eax, 5

           shr eax, 5

           mov ebx, mem_block_map

           mov ebx, mem_block_map

           and eax, not 3

           and eax, not 3

           add eax, ebx

           add eax, ebx

           cmp [mem_block_start], eax

           cmp [mem_block_start], eax

           ja @f

           ja @f

           popfd

           ret

@@:

@@:

           mov [mem_block_start], eax

           mov [mem_block_start], eax

           popfd

	   ret

.err:

.err:

           xor eax, eax

           xor eax, eax

           popfd

	   ret

endp

endp

align 4

align 4

proc alloc_kernel_space stdcall, size:dword

proc alloc_kernel_space stdcall, size:dword

           local block_ind:DWORD

           local block_ind:DWORD

           mov eax, [size]

           mov eax, [size]

           add eax, 4095

           add eax, 4095

           and eax, not 4095

           and eax, not 4095

           mov [size], eax

           cmp eax, [heap_free]

           cmp eax, [heap_free]

           ja .error

           ja .error

           shr eax, 12

           shr eax, 12

           sub eax, 1

           sub eax, 1

           mov [block_ind], eax

           mov [block_ind], eax

           stdcall get_block, eax

           stdcall get_block, eax

           mov [mem_used_list], esi

           mov [mem_used_list], esi

           mov eax, [esi+block_base]

           mov eax, [esi+block_base]

           mov ebx, [size]

           mov ebx, [size]

           sub [heap_free], ebx

           sub [heap_free], ebx

           popfd

           and [heap_mutex], 0

.m_eq_size:

           remove_from_list edi

           remove_from_list edi

           mov [mem_block_list+ebx*4], edx

           mov [mem_block_list+ebx*4], edx

           and edx, edx

           and edx, edx

           jnz @f

           jnz @f

           btr [mem_block_mask], ebx

           btr [mem_block_mask], ebx

           mov [edi+block_flags], USED_BLOCK

           mov [edi+block_flags], USED_BLOCK

           mov eax, [edi+block_base]

           mov eax, [edi+block_base]

           mov ebx, [size]

           mov ebx, [size]

           sub [heap_free], ebx

           sub [heap_free], ebx

           popfd

           and [heap_mutex], 0

.error:

.error:

           xor eax, eax

           xor eax, eax

           popfd

           mov [heap_mutex], eax

endp

endp

align 4

align 4

proc free_kernel_space stdcall, base:dword

proc free_kernel_space stdcall, base:dword

           mov eax, [base]

           mov eax, [base]

           mov esi, [mem_used_list]

           mov esi, [mem_used_list]

@@:

@@:

           test esi, esi

           test esi, esi

@@:

@@:

           bts [mem_block_mask], eax

           bts [mem_block_mask], eax

.m_eq:

.m_eq:

           xor eax, eax

           xor eax, eax

           not eax

           mov [heap_mutex], eax

           ret

           ret

.insert:

.insert:

           remove_from_used esi

           remove_from_used esi

@@:

@@:

           bts [mem_block_mask], eax

           bts [mem_block_mask], eax

           mov [esi+block_flags],FREE_BLOCK

           mov [esi+block_flags],FREE_BLOCK

           xor eax, eax

           xor eax, eax

           not eax

           mov [heap_mutex], eax

           ret

           ret

.fail:

.fail:

           xor eax, eax

           xor eax, eax

           ret

           mov [heap_mutex], eax

endp

endp

align 4

align 4

proc kernel_alloc stdcall, size:dword

proc kernel_alloc stdcall, size:dword

           jnz @B

           jnz @B

.end:

.end:

           mov eax, [lin_addr]

           mov eax, [lin_addr]

           ret

           ret

.error:

           xor eax, eax

           xor eax, eax

           ret

           ret

endp

endp

align 4

align 4

proc kernel_free stdcall, base:dword

proc kernel_free stdcall, base:dword

           mov eax, [base]

           mov eax, [base]

           mov esi, [mem_used_list]

           mov esi, [mem_used_list]

@@:

@@:

.found:

.found:

           cmp [esi+block_flags], USED_BLOCK

           cmp [esi+block_flags], USED_BLOCK

           jne .fail

           jne .fail

           mov ecx, [esi+block_size];

           mov ecx, [esi+block_size];

           call release_pages   ;eax, ecx

           jnz .release

           ret

.fail:

           and [heap_mutex], 0

           ret

           ret

endp

endp