Subversion Repositories Kolibri OS

proc mem_test

           mov eax, cr0

           or eax, (CR0_CD+CR0_NW);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 cr0, eax

           mov eax, edi

           ret

endp

proc init_memEx

           xor eax, eax

           mov edi, sys_pgdir

           mov ecx, 2048

           rep stosd

           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

           add eax, 0x00400000

           mov dword [sys_pgdir+4], eax

           add eax, 0x00400000

           mov dword [sys_pgdir+8], eax

           add eax, 0x00400000

           mov dword [sys_pgdir+12], eax

           mov dword [sys_master_tab+0x600], sys_master_tab+PG_SW

           sub ecx, 4

           mov eax, tmp_page_tab+PG_SW

           mov edi, sys_pgdir+16

           mov esi, sys_master_tab+16

.no_PSE:

           mov eax, PG_SW

           mov esi, tmp_page_tab

           mov ecx, 4096/4      ;0x0 - 0x00FFFFFF

.map_low:

           mov [esi], eax

           add eax, 0x1000

           mov [esi+4], eax

           add eax, 0x1000

           mov [esi+8], eax

           add eax, 0x1000

           mov [esi+12], eax

           add eax, 0x1000

           add esi, 16

           dec ecx

           jnz .map_low            ;ÿäðî

           mov eax, tmp_page_tab+PG_SW

           mov edi, sys_pgdir

           mov esi, sys_master_tab

           mov [esi], eax

           add eax, 0x1000

           add edi, 4

           add esi, 4

           dec ecx

           jnz .map_kernel_tabs

           bt [cpu_caps], CAPS_PSE

           jc @F

           add edi, 4096*4      ;skip low kernel memory

@@:

           mov ecx, [pg_data.kernel_tables]

           sub ecx, 4

           shl ecx, 10

           xor eax, eax

           cld

           rep stosd

           mov dword [sys_master_tab+0x600], sys_master_tab+PG_SW

           ret

endp

proc init_page_map

           mov edi, sys_pgmap

           mov ecx, 512/4

           xor eax,eax

           cld

           rep stosd

           mov ecx, [pg_data.pagemap_size]

           sub ecx, 512

           shr ecx, 2

           rep stosd

           mov edx, [pg_data.pages_count]

           mov ecx, [pg_data.kernel_tables]

           bt [cpu_caps], CAPS_PSE

           jnc @f

           sub ecx, 4

@@:

           sub edx, 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

           stosd

           mov ebx, sys_pgmap

           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

           shl ebx, 3

           add eax,ebx

           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

           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

proc map_page stdcall,lin_addr:dword,phis_addr:dword,flags:dword

           mov eax, [phis_addr]

           and eax, not 0xFFF

           or eax, [flags]

           mov ebx, [lin_addr]

           shr ebx, 12

           mov [pages_tab+ebx*4], eax

           mov eax, [lin_addr]

           invlpg [eax]

           ret

endp

proc free_page

;arg:  eax  page address

           pushfd

           cli

           inc [pg_data.pages_free]

           shr eax, 12              ;page index

           mov ebx, sys_pgmap

           bts [ebx], eax           ;that's all!

           shr eax, 3

           and eax, not 3           ;dword offset from page_map

           add eax, ebx

           cmp [page_start], eax

           ja @f

           popfd

           ret

@@:

           mov [page_start], eax

           popfd

           ret

endp