Subversion Repositories Kolibri OS

proc mem_test

           and eax, not (CR0_CD+CR0_NW)

           or eax, CR0_CD         ;disable caching

           mov cr0, eax

           wbinvd                 ;invalidate cache

           mov ebx, 'TEST'

@@:

           add edi, 0x400000

           xchg ebx, dword [edi]

           cmp dword [edi], 'TEST'

           xchg ebx, dword [edi]

           je @b

           mov [MEM_AMOUNT], edi

           mov cr0, eax

           mov eax, edi

           mov [LFBSize], 0x00800000

           ret

endp

proc init_mem

           mov [pg_data.kernel_max], eax

           mov edx, eax

           mov [pg_data.pages_count], eax

           mov [pg_data.kernel_pages], eax

           mov [pg_data.pagemap_size], eax

           cmp edx, 3

           ja @f

           inc edx       ;at least 4Mb for kernel heap

@@:

           mov [pg_data.kernel_tables], edx

           mov edi, sys_pgdir

           mov ecx, 2048

           cld

           rep stosd

           bt [cpu_caps], CAPS_PSE

           jnc .no_PSE

           or ebx, CR4_PSE

           mov eax, PG_LARGE+PG_SW

           bt [cpu_caps], CAPS_PGE

           jnc @F

           or eax, PG_GLOBAL

           or ebx, CR4_PGE

@@:

           mov cr4, ebx

           sub [pg_data.kernel_tables], 2

           add eax, 0x00400000

           mov [edx+4], eax

           add edx, 8

           mov ecx, (HEAP_BASE-0x800000)/4096

           jmp .map_low

.no_PSE:

           mov eax, PG_SW

           mov ecx, HEAP_BASE/4096

.map_low:

           mov edi, tmp_page_tab

@@:                                   ;

           stosd

           add eax, 0x1000

           dec ecx

           jnz @B

           shl ecx, 10

           xor eax, eax

           rep stosd

           mov eax, tmp_page_tab+PG_SW

           mov edi, edx

           add eax, 0x1000

           dec ecx

           jnz .map_kernel_tabs

           ret

endp

proc init_page_map

           mov ecx, (HEAP_BASE/4096)/32      ;384/4

           mov ebx, ecx

           xor eax,eax

           cld

           rep stosd

           mov ecx, [pg_data.pagemap_size]

           sub ecx, ebx

           shr ecx, 2

           rep stosd

           mov edx, [pg_data.pages_count]

           mov ecx, [pg_data.kernel_tables]

           add ecx, (HEAP_BASE/4096) and 31

           sub edx, HEAP_BASE/4096

           sub edx, ecx

           mov [pg_data.pages_free], edx

           mov ebx, ecx

           shr ecx, 5

           rep stosd

           mov ecx, ebx

           and ecx, 31

           shl eax, cl

           mov [page_start], edi;           sys_pgmap+384

           stosd

           add ebx, [pg_data.pagemap_size]

           mov [page_end], ebx

endp

proc alloc_page

           cli

           mov ebx, [page_start]

           mov ecx, [page_end]

.l1:

           bsf eax,[ebx];

           jnz .found

           add ebx,4

           cmp ebx, ecx

           jb .l1

           popfd

           xor eax,eax

           ret

.found:

           btr [ebx], eax

           mov [page_start],ebx

           sub ebx, sys_pgmap

           lea eax, [eax+ebx*8]

           shl eax, 12

           dec [pg_data.pages_free]

           popfd

           ret

endp

proc alloc_pages stdcall, count:dword

           pushfd

           cli

           mov eax, [count]

           add eax, 7

           shr eax, 3

           mov [count], eax

           cmp eax, [pg_data.pages_free]

           ja .fail

           mov ebx, [page_end]

.find:

           mov edx, [count]

           mov edi, ecx

.match:

           cmp byte [ecx], 0xFF

           jne .next

           dec edx

           jz .ok

           inc ecx

           cmp ecx,ebx

           jb .match

.fail:     xor eax, eax

           popfd

           ret

.next:

           inc ecx

           cmp ecx, ebx

           jb .find

           popfd

           xor eax, eax

           ret

.ok:

           sub ecx, edi

           inc ecx

           mov esi, edi

           xor eax, eax

           rep stosb

           sub esi, sys_pgmap

           shl esi, 3+12

           mov eax, esi

           mov ebx, [count]

           shl ebx, 3

           sub [pg_data.pages_free], ebx

           popfd

           ret

endp