Subversion Repositories Kolibri OS

MEM_WC     equ 1               ;write combined memory

MEM_UC     equ 0               ;uncached memory

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, 0x100000

           xchg ebx, dword [edi]

           cmp dword [edi], 'TEST'

           xchg ebx, dword [edi]

           je @b

           mov [MEM_AMOUNT-OS_BASE], edi

           mov cr0, eax

           mov eax, edi

           ret

endp

proc init_mem

           mov eax, [MEM_AMOUNT-OS_BASE]

           mov [pg_data.mem_amount-OS_BASE], eax

           mov edx, eax

           mov [pg_data.pages_count-OS_BASE], eax

           shr eax, 3

           mov [pg_data.pagemap_size-OS_BASE], eax

           jbe @F

           mov edx, (OS_BASE/4096)

           jmp .set

@@:

           cmp edx, (HEAP_MIN_SIZE/4096)

           jae .set

           mov edx, (HEAP_MIN_SIZE/4096)

.set:

           mov [pg_data.kernel_pages-OS_BASE], edx

           shr edx, 10

           mov [pg_data.kernel_tables-OS_BASE], edx

           mov edi, sys_pgdir-OS_BASE

           mov ecx, 4096/4

           cld

           rep stosd

           bt [cpu_caps-OS_BASE], CAPS_PSE

           jnc .no_PSE

           or ebx, CR4_PSE

           mov eax, PG_LARGE+PG_SW

           jnc @F

           or ebx, CR4_PGE

@@:

           mov cr4, ebx

           sub [pg_data.kernel_tables-OS_BASE], 1

           add eax, 0x00400000

;           mov [edx+4], eax

           add edx, 4

           mov ecx, (tmp_page_tab-0x400000)/4096

           jmp .map_low

.no_PSE:

           mov eax, PG_SW

           mov ecx, tmp_page_tab/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

           lea esi, [edi+(OS_BASE shr 20)]

           movsd

           movsd

           movsd

           ret

endp

proc init_page_map

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

           mov ebx, ecx

           xor eax,eax

           cld

           rep stosd

           mov ecx, [pg_data.pagemap_size-OS_BASE]

           sub ecx, ebx

           shr ecx, 2

           rep stosd

           mov edx, [pg_data.pages_count-OS_BASE]

           mov ecx, [pg_data.kernel_tables-OS_BASE]

           add ecx, ((HEAP_BASE-OS_BASE)/4096) and 31

           sub edx, (HEAP_BASE-OS_BASE)/4096

           sub edx, ecx

           mov [pg_data.pages_free-OS_BASE], edx

           mov ebx, ecx

           shr ecx, 5

           rep stosd

           mov ecx, ebx

           and ecx, 31

           shl eax, cl

           mov [edi], eax

           add edi, OS_BASE

           mov [page_start-OS_BASE], edi;

           add ebx, [pg_data.pagemap_size-OS_BASE]

           mov [page_end-OS_BASE], ebx

endp

proc test_cpu

           locals

              cpu_type   dd ?

              cpu_id     dd ?

              cpu_Intel  dd ?

              cpu_AMD    dd ?

           endl

           xor eax, eax

           mov [cpu_caps-OS_BASE], eax

           mov [cpu_caps+4-OS_BASE], eax

           pop eax

           mov ecx, eax

           xor eax, 0x40000

           push eax

           popfd

           pushfd

           pop eax

           xor eax, ecx

           mov [cpu_type], CPU_386

           jz .end_cpuid

           push ecx

           popfd

           mov eax, ecx

           xor eax, 0x200000

           push eax

           popfd

           pushfd

           pop eax

           xor eax, ecx

           je .end_cpuid

           mov [cpu_id], 1

           cpuid

           mov [cpu_vendor+4-OS_BASE], edx

           mov [cpu_vendor+8-OS_BASE], ecx

           cmp ebx, dword [intel_str-OS_BASE]

           jne .check_AMD

           cmp edx, dword [intel_str+4-OS_BASE]

           jne .check_AMD

           cmp ecx, dword [intel_str+8-OS_BASE]

           jne .check_AMD

           mov [cpu_Intel], 1

           cmp eax, 1

           jl .end_cpuid

           mov eax, 1

           cpuid

           mov [cpu_sign-OS_BASE], eax

           mov [cpu_info-OS_BASE],  ebx

           mov [cpu_caps-OS_BASE],  edx

           mov [cpu_caps+4-OS_BASE],ecx

           and eax, 0x0f

           ret

.end_cpuid:

           mov eax, [cpu_type]

           ret

           cmp ebx, dword [AMD_str-OS_BASE]

           jne .unknown

           cmp edx, dword [AMD_str+4-OS_BASE]

           jne .unknown

           cmp ecx, dword [AMD_str+8-OS_BASE]

           jne .unknown

           mov [cpu_AMD], 1

           cmp eax, 1

           jl .unknown

           mov eax, 1

           cpuid

           mov [cpu_sign-OS_BASE], eax

           mov [cpu_info-OS_BASE],  ebx

           mov [cpu_caps-OS_BASE],  edx

           mov [cpu_caps+4-OS_BASE],ecx

           shr eax, 8

           and eax, 0x0f

           ret

.unknown:

           mov eax, 1

           cpuid

           mov [cpu_sign-OS_BASE], eax

           mov [cpu_info-OS_BASE],  ebx

           mov [cpu_caps-OS_BASE],  edx

           mov [cpu_caps+4-OS_BASE],ecx

           shr eax, 8

           and eax, 0x0f

           ret

endp