Subversion Repositories Kolibri OS

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

;;                                                              ;;

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

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

;;                                                              ;;

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

$Revision: 2434 $

struct  MEM_BLOCK

        list            LHEAD

        next_block      dd ? ;+8

        prev_block      dd ? ;+4

        base            dd ? ;+16

        size            dd ? ;+20

        flags           dd ? ;+24

        handle          dd ? ;+28

ends

FREE_BLOCK      equ  4

USED_BLOCK      equ  8

DONT_FREE_BLOCK equ  10h

block_next   equ MEM_BLOCK.next_block

block_prev   equ MEM_BLOCK.prev_block

list_fd      equ MEM_BLOCK.list.next

list_bk      equ MEM_BLOCK.list.prev

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

@@:

}

align 4

md:

.add_to_used:

           mov eax, [esi+block_base]

           mov ebx, [esi+block_base]

           shr ebx, 6

           add eax, ebx

           shr ebx, 6

           add eax, ebx

           shr eax, 12

           and eax, 63

           inc [mem_hash_cnt+eax*4]

           lea ecx, [mem_used_list+eax*8]

           list_add esi, ecx

           mov [esi+block_flags], USED_BLOCK

           mov eax, [esi+block_size]

           sub [heap_free], eax

           ret

align 4

.find_used:

           mov ecx, eax

           mov ebx, eax

           shr ebx, 6

           add ecx, ebx

           shr ebx, 6

           add ecx, ebx

           shr ecx, 12

           and ecx, 63

           lea ebx, [mem_used_list+ecx*8]

           mov esi, ebx

.next:

           mov esi, [esi+list_fd]

           cmp esi, ebx

           je .fail

           cmp eax, [esi+block_base]

           jne .next

           ret

.fail:

           xor esi, esi

           ret

align 4

.del_from_used:

           call .find_used

           test esi, esi

           jz .done

           cmp [esi+block_flags], USED_BLOCK

           jne .fatal

           dec [mem_hash_cnt+ecx*4]

           list_del esi

.done:

           ret

.fatal:                            ;FIXME panic here

           xor esi, esi

           ret

;Initial heap state

;

;+heap_size               terminator        USED_BLOCK

;+4096*MEM_BLOCK.sizeof   free space        FREE_BLOCK

;HEAP_BASE                heap_descriptors  USED_BLOCK

;

align 4

proc init_kernel_heap

            mov ecx, 64

            mov edi, mem_block_list

@@:

            mov eax, edi

            stosd

            stosd

            loop @B

            mov ecx, 64

            mov edi, mem_used_list

@@:

            mov eax, edi

            stosd

            stosd

            loop @B

            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                     ;descriptors

        mov     ebx, HEAP_BASE+sizeof.MEM_BLOCK      ;free space

        mov     ecx, HEAP_BASE+sizeof.MEM_BLOCK*2    ;terminator

            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*sizeof.MEM_BLOCK

            mov [edi+block_flags], USED_BLOCK

            mov [ecx+block_next], eax

            mov [ecx+block_prev], ebx

            mov [edi+list_fd], eax

            mov [edi+list_bk], eax

            mov [edi+block_base], eax

            mov [edi+block_size], eax

            mov [edi+block_flags], USED_BLOCK

            mov [ebx+block_next], ecx

            mov [ebx+block_prev], edi

        mov     [ebx+block_base], HEAP_BASE+4096*sizeof.MEM_BLOCK

            mov ecx, [pg_data.kernel_pages]

            shl ecx, 12

        sub     ecx, HEAP_BASE-OS_BASE+4096*sizeof.MEM_BLOCK

            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 ecx, mem_block_list+63*8

            list_add ebx, ecx

            mov ecx, 4096-3-1

        mov     eax, HEAP_BASE+sizeof.MEM_BLOCK*4

        mov     [next_memblock], HEAP_BASE+sizeof.MEM_BLOCK *3

@@:

        mov     [eax-sizeof.MEM_BLOCK], eax

        add     eax, sizeof.MEM_BLOCK

            loop @B

        mov     [eax-sizeof.MEM_BLOCK], dword 0

            mov ecx, heap_mutex

            call mutex_init

            mov [heap_blocks], 4094

            mov [free_blocks], 4093

            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: