Subversion Repositories Kolibri OS

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

;;                                                              ;;

;; Copyright (C) KolibriOS team 2004-2009. All rights reserved. ;;

;; Distributed under terms of the GNU General Public License    ;;

;;                                                              ;;

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

$Revision: 2014 $

struc MEM_BLOCK

{  .next_block	dd ?

   .prev_block	dd ? ;+4

   .list_fd	dd ? ;+8

   .list_bk	dd ? ;+12

   .base	dd ? ;+16

   .size	dd ? ;+20

   .flags	dd ? ;+24

   .handle	dd ? ;+28

}

MEM_LIST_OFFSET equ  8

FREE_BLOCK	equ  4

USED_BLOCK	equ  8

DONT_FREE_BLOCK equ  10h

virtual at 0

  MEM_BLOCK MEM_BLOCK

end virtual

MEM_BLOCK_SIZE equ 8*4

block_next   equ MEM_BLOCK.next_block

block_prev   equ MEM_BLOCK.prev_block

list_fd      equ MEM_BLOCK.list_fd

list_bk      equ MEM_BLOCK.list_bk

block_base   equ MEM_BLOCK.base

block_size   equ MEM_BLOCK.size

block_flags  equ MEM_BLOCK.flags

macro calc_index op

{	   shr op, 12

	   dec op

	   cmp op, 63

	   jna @f

	   mov op, 63

@@:

}

macro remove_from_list op

{	   mov edx, [op+list_fd]

	   mov ecx, [op+list_bk]

	   test edx, edx

	   jz @f

	   mov [edx+list_bk], ecx

@@:

	   test ecx, ecx

	   jz @f

	   mov [ecx+list_fd], edx

@@:

	   mov [op+list_fd],0

	   mov [op+list_bk],0

}

macro remove_from_free op

{

	   remove_from_list op

	   mov eax, [op+block_size]

	   calc_index eax

	   cmp [mem_block_list+eax*4], op

	   jne @f

	   mov [mem_block_list+eax*4], edx

@@:

	   cmp [mem_block_list+eax*4], 0

	   jne @f

	   btr [mem_block_mask], eax

@@:

}

macro remove_from_used op

{

	   mov edx, [op+list_fd]

	   mov ecx, [op+list_bk]

	   mov [edx+list_bk], ecx

	   mov [ecx+list_fd], edx

	   mov [op+list_fd], 0

	   mov [op+list_bk], 0

}

align 4

proc init_kernel_heap

	   mov ecx, 64

	   mov edi, mem_block_list

	   xor eax, eax

	   cld

	   rep stosd

	   mov ecx, 512/4

	   mov edi, mem_block_map

	   not eax

	   rep stosd

	   mov [mem_block_start], mem_block_map

	   mov [mem_block_end], mem_block_map+512

	   mov [mem_block_arr], HEAP_BASE

	   mov eax, mem_used.fd-MEM_LIST_OFFSET

	   mov [mem_used.fd], eax

	   mov [mem_used.bk], eax

	   mov [KERNEL_ALLOC_FLAG], dword PG_SW

	   stdcall alloc_pages, dword 32

	   mov ecx, 32

	   mov edx, eax

	   mov edi, HEAP_BASE

.l1:

	   stdcall map_page,edi,edx,PG_SW

	   add edi, 0x1000

	   add edx, 0x1000

	   dec ecx

	   jnz .l1

	   mov edi, HEAP_BASE

	   mov ebx, HEAP_BASE+MEM_BLOCK_SIZE

	   xor eax, eax

	   mov [edi+block_next], ebx

	   mov [edi+block_prev], eax

	   mov [edi+list_fd], eax

	   mov [edi+list_bk], eax

	   mov [edi+block_base], HEAP_BASE

	   mov [edi+block_size], 4096*MEM_BLOCK_SIZE

	   mov [edi+block_flags], USED_BLOCK

	   mov [ebx+block_next], eax

	   mov [ebx+block_prev], eax

	   mov [ebx+list_fd], eax

	   mov [ebx+list_bk], eax

	   mov [ebx+block_base], HEAP_BASE+4096*MEM_BLOCK_SIZE

	   mov ecx, [pg_data.kernel_pages]

	   shl ecx, 12

	   sub ecx, HEAP_BASE-OS_BASE+4096*MEM_BLOCK_SIZE

	   mov [heap_size], ecx

	   mov [heap_free], ecx

	   mov [ebx+block_size], ecx

	   mov [ebx+block_flags], FREE_BLOCK

	   mov [mem_block_mask], eax

	   mov [mem_block_mask+4],0x80000000

	   mov [mem_block_list+63*4], ebx

	   mov byte [mem_block_map], 0xFC

	   and [heap_mutex], 0

	   mov [heap_blocks], 4095

	   mov [free_blocks], 4094

	   ret

endp

; param

;  eax= required size

;

; retval

;  edi= memory block descriptor

;  ebx= descriptor index

align 4

get_small_block:

	   mov ecx, eax

	   shr ecx, 12

	   dec ecx

	   cmp ecx, 63

	   jle .get_index

	   mov ecx, 63

.get_index:

	   lea esi, [mem_block_mask]

	   xor ebx, ebx

	   or edx, -1

	   cmp ecx, 32

	   jb .bit_test

	   sub ecx, 32

	   add ebx, 32

	   add esi, 4

.bit_test:

	   shl edx, cl

	   and edx, [esi]

.find:

	   bsf edi, edx

	   jz .high_mask

	   add ebx, edi

	   mov edi, [mem_block_list+ebx*4]

.check_size:

	   cmp eax, [edi+block_size]

	   ja .next

	   ret

.high_mask:

	   add esi, 4

	   cmp esi, mem_block_mask+8

	   jae .err

	   add ebx, 32

	   mov edx, [esi]

	   jmp .find

.next:

	   mov edi, [edi+list_fd]

	   test edi, edi

	   jnz .check_size

.err: