WebSVN – Kolibri OS – Diff – /kernel/trunk/core/memory.inc


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                              ;;
;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;;
;; Distributed under terms of the GNU General Public License    ;;
;;                                                              ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
$Revision: 750 $
 
 
align 4
proc alloc_page
 
           pushfd
           cli
           push ebx
           mov ebx, [page_start]
           mov ecx, [page_end]
.l1:
           bsf eax,[ebx];
           jnz .found
           add ebx,4
           cmp ebx, ecx
           jb .l1
           pop ebx
           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]
           pop ebx
           popfd
           ret
endp
 
align 4
proc alloc_pages stdcall, count:dword
           pushfd
           push ebx
           push edi
           cli
           mov eax, [count]
           add eax, 7
           shr eax, 3
           mov [count], eax
           cmp eax, [pg_data.pages_free]
           ja .fail
 
           mov ecx, [page_start]
           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
           pop edi
           pop ebx
           popfd
           ret
.next:
           inc ecx
           cmp ecx, ebx
           jb .find
           pop edi
           pop ebx
           popfd
           xor eax, eax
           ret
.ok:
           sub ecx, edi
           inc ecx
           push esi
           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
           pop esi
           pop edi
           pop ebx
           popfd
           ret
endp
 
align 4
proc map_page stdcall,lin_addr:dword,phis_addr:dword,flags:dword
           push ebx
           mov eax, [phis_addr]
           and eax, not 0xFFF
           or eax, [flags]
           mov ebx, [lin_addr]
           shr ebx, 12
           mov [page_tabs+ebx*4], eax
           mov eax, [lin_addr]
           invlpg [eax]
           pop ebx
           ret
endp
 
align 4
map_space:    ;not implemented
 
 
           ret
 
 
align 4
proc free_page
;arg:  eax  page address
           pushfd
           cli
           shr eax, 12              ;page index
           bts dword [sys_pgmap], eax           ;that's all!
           cmc
           adc [pg_data.pages_free], 0
           shr eax, 3
           and eax, not 3           ;dword offset from page_map
           add eax, sys_pgmap
           cmp [page_start], eax
           ja @f
           popfd
           ret
@@:
           mov [page_start], eax
           popfd
           ret
endp
 
proc map_io_mem stdcall, base:dword, size:dword, flags:dword
 
           push ebx
           push edi
           mov eax, [size]
           add eax, 4095
           and eax, -4096
           mov [size], eax
           stdcall alloc_kernel_space, eax
           test eax, eax
           jz .fail
           push eax
 
           mov edi, 0x1000
           mov ebx, eax
           mov ecx,[size]
           mov edx, [base]
           shr eax, 12
           shr ecx, 12
           and edx, -4096
           or edx, [flags]
@@:
           mov [page_tabs+eax*4], edx
           invlpg [ebx]
           inc eax
           add ebx, edi
           add edx, edi
           loop @B
 
           pop eax
           mov edx, [base]
           and edx, 4095
           add eax, edx
.fail:
           pop edi
           pop ebx
           ret
endp
 
; param
;  eax= page base + page flags
;  ebx= liear address
;  ecx= count
 
align 4
commit_pages:
           push edi
           test ecx, ecx
           jz .fail
 
           mov edi, ebx
           mov ebx, pg_data.pg_mutex
           call wait_mutex      ;ebx
 
           mov edx, 0x1000
           mov ebx, edi
           shr ebx, 12
@@:
           mov [page_tabs+ebx*4], eax
           invlpg [edi]
           add edi, edx
           add eax, edx
           inc ebx
           dec ecx
           jnz @B
           mov [pg_data.pg_mutex],ecx
.fail:
           pop edi
           ret
 
 
; param
;  eax= base
;  ecx= count
 
align 4
release_pages:
 
           pushad
           mov ebx, pg_data.pg_mutex
           call wait_mutex      ;ebx
 
           mov esi, eax
           mov edi, eax
 
           shr esi, 10
           add esi, page_tabs
 
           mov ebp, [pg_data.pages_free]
           mov ebx, [page_start]
           mov edx, sys_pgmap
@@:
           xor eax, eax
           xchg eax, [esi]
           invlpg [edi]
 
           test eax, 1
           jz .next
 
           shr eax, 12
           bts [edx], eax
           cmc
           adc ebp, 0
           shr eax, 3
           and eax, -4
           add eax, edx
           cmp eax, ebx
           jae .next
 
           mov ebx, eax
.next:
           add edi, 0x1000
           add esi, 4
           dec ecx
           jnz @B
           mov [pg_data.pages_free], ebp
           and [pg_data.pg_mutex],0
           popad
           ret
 
align 4
proc map_page_table stdcall, lin_addr:dword, phis_addr:dword
           push ebx
           mov ebx, [lin_addr]
           shr ebx, 22
           mov eax, [phis_addr]
           and eax, not 0xFFF
           or eax, PG_UW          ;+PG_NOCACHE
           mov dword [master_tab+ebx*4], eax
           mov eax, [lin_addr]
           shr eax, 10
           add eax, page_tabs
           invlpg [eax]
           pop ebx
           ret
endp
 
align 4
proc init_LFB
           locals
             pg_count dd ?
           endl
 
           cmp dword [LFBAddress], -1
           jne @f
           mov [BOOT_VAR+0x901c],byte 2
           stdcall kernel_alloc, 0x280000
           mov [LFBAddress], eax
           ret
@@:
           test [SCR_MODE],word 0100000000000000b
           jnz @f
           mov [BOOT_VAR+0x901c],byte 2
           ret
@@:
           call init_mtrr
 
           mov edx, LFB_BASE
           mov esi, [LFBAddress]
           mov edi, 0x00800000
           mov dword [exp_lfb+4], edx
 
           shr edi, 12
           mov [pg_count], edi
           shr edi, 10
 
           bt [cpu_caps], CAPS_PSE
           jnc .map_page_tables
           or esi, PG_LARGE+PG_UW
           mov edx, sys_pgdir+(LFB_BASE shr 20)
@@:
           mov [edx], esi
           add edx, 4
           add esi, 0x00400000
           dec edi
           jnz @B
 
           bt [cpu_caps], CAPS_PGE
           jnc @F
           or dword [sys_pgdir+(LFB_BASE shr 20)], PG_GLOBAL
@@:
           mov dword [LFBAddress], LFB_BASE
           mov eax, cr3       ;flush TLB
           mov cr3, eax
           ret
 
.map_page_tables:
 
@@:
           call alloc_page
           stdcall map_page_table, edx, eax
           add edx, 0x00400000
           dec edi
           jnz @B
 
           mov eax, [LFBAddress]
           mov edi, page_tabs + (LFB_BASE shr 10)
           or eax, PG_UW
           mov ecx, [pg_count]
           cld
@@:
           stosd
           add eax, 0x1000
           dec ecx
           jnz @B
 
           mov dword [LFBAddress], LFB_BASE
           mov eax, cr3       ;flush TLB
           mov cr3, eax
 
           ret
endp
 
align 4
proc new_mem_resize stdcall, new_size:dword
 
           mov ebx, pg_data.pg_mutex
           call wait_mutex    ;ebx
 
           mov edi, [new_size]
           add edi,4095
           and edi,not 4095
           mov [new_size], edi
 
           mov edx,[current_slot]
           cmp [edx+APPDATA.heap_base],0
           jne .exit
 
           mov esi, [edx+APPDATA.mem_size]
           add esi, 4095
           and esi, not 4095
 
           cmp edi, esi
           jae .expand
 
           shr edi, 12
           shr esi, 12
@@:
           mov eax, [app_page_tabs+edi*4]
           test eax, 1
           jz .next
           mov dword [app_page_tabs+edi*4], 2
           mov ebx, edi
           shl ebx, 12
           invlpg [ebx]
           call free_page
 
.next:     add edi, 1
           cmp edi, esi
           jb @B
 
.update_size:
           mov     ebx, [new_size]
           call    update_mem_size
 
           xor eax, eax
           dec [pg_data.pg_mutex]
           ret
.expand:
 
           push esi
           push edi
 
           add edi, 0x3FFFFF
           and edi, not(0x3FFFFF)
           add esi, 0x3FFFFF
           and esi, not(0x3FFFFF)
 
           cmp esi, edi
           jae .grow
 
           xchg esi, edi
 
@@:
           call alloc_page
           test eax, eax
           jz .exit
 
           stdcall map_page_table, edi, eax
 
           push edi
           shr edi, 10
           add edi, page_tabs
           mov ecx, 1024
           xor eax, eax
           cld
           rep stosd
           pop edi
 
           add edi, 0x00400000
           cmp edi, esi
           jb @B
.grow:
           pop edi
           pop esi
@@:
           call alloc_page
           test eax, eax
           jz .exit
           stdcall map_page,esi,eax,dword PG_UW
 
           push edi
           mov edi, esi
           xor eax, eax
           mov ecx, 1024
           cld
           rep stosd
           pop edi
 
           add esi, 0x1000
           cmp esi, edi
           jb  @B
 
           jmp .update_size
.exit:
           xor eax, eax
           inc eax
           dec [pg_data.pg_mutex]
           ret
endp
 
update_mem_size:
; in: edx = slot base
;     ebx = new memory size
; destroys eax,ecx,edx
 
           mov    [APPDATA.mem_size+edx],ebx
;search threads and update
;application memory size infomation
           mov    ecx,[APPDATA.dir_table+edx]
           mov    eax,2
 
.search_threads:
;eax = current slot
;ebx = new memory size
;ecx = page directory
           cmp    eax,[TASK_COUNT]
           jg     .search_threads_end
           mov    edx,eax
           shl    edx,5
           cmp    word [CURRENT_TASK+edx+TASKDATA.state],9 ;if slot empty?
           jz     .search_threads_next
           shl    edx,3
           cmp    [SLOT_BASE+edx+APPDATA.dir_table],ecx     ;if it is our thread?
           jnz    .search_threads_next
           mov    [SLOT_BASE+edx+APPDATA.mem_size],ebx     ;update memory size
.search_threads_next:
           inc    eax
           jmp    .search_threads
.search_threads_end:
           ret
 
; param
;  eax= linear address
;
; retval
;  eax= phisical page address
 
align 4
get_pg_addr:
           shr eax, 12
           mov eax, [page_tabs+eax*4]
           and eax, 0xFFFFF000
           ret
 
 
align 4
proc page_fault_handler
 
        test    byte [esp+12+2], 2
        jnz     v86_page_fault
 
        .err_code equ ebp+32
        .err_addr equ ebp-4
 
           pushad
           mov ebp, esp
           mov eax, cr2
           push eax
 
           mov ax, app_data
           mov ds, ax
           mov es, ax
 
           inc [pg_data.pages_faults]
 
;     push eax
;     push edx
;     mov edx, 0x400   ;bochs
;     mov al,0xff      ;bochs
;     out dx, al       ;bochs
;     pop edx
;     pop eax
 
           mov ebx, [.err_addr]
           mov eax, [.err_code]
 
           cmp ebx, OS_BASE
           jb .user_space      ;ñòðàíèöà â ïàìÿòè ïðèëîæåíèÿ ;
 
           cmp ebx, page_tabs
           jb .kernel_space    ;ñòðàíèöà â ïàìÿòè ÿäðà
 
           cmp ebx, kernel_tabs
           jb .alloc;.app_tabs ;òàáëèöû ñòðàíèö ïðèëîæåíèÿ ;
                               ;ïðîñòî ñîçäàäèì îäíó
 
           cmp ebx, LFB_BASE
           jb .core_tabs       ;òàáëèöû ñòðàíèö ÿäðà
                               ;Îøèáêà
.lfb:
                               ;îáëàñòü LFB
                               ;Îøèáêà
           jmp .fail
 
align 4
.user_space:
           test eax, PG_MAP
           jnz .err_access     ;Ñòðàíèöà ïðèñóòñòâóåò
                               ;Îøèáêà äîñòóïà ?
 
           shr ebx, 12
           mov ecx, ebx
           shr ecx, 10
           mov edx, [master_tab+ecx*4]
           test edx, PG_MAP
           jz .fail            ;òàáëèöà ñòðàíèö íå ñîçäàíà
                               ;íåâåðíûé àäðåñ â ïðîãðàììå
 
           mov eax, [page_tabs+ebx*4]
           test eax, 2
           jz .fail            ;àäðåñ íå çàðåçåðâèðîâàí äëÿ ;
                               ;èñïîëüçîâàíèÿ. Îøèáêà
.alloc:
           call alloc_page
           test eax, eax
           jz .fail
 
           stdcall map_page,[ebp-4],eax,dword PG_UW
 
           mov edi, [ebp-4]
           and edi, 0xFFFFF000
           mov ecx, 1024
           xor eax, eax
           cld
           rep stosd
.exit:
           mov esp, ebp
           popad
           add esp, 4
           iretd
 
.err_access:
;íèêîãäà íå ïðîèñõîäèò
           jmp .fail
 
.kernel_space:
           test eax, PG_MAP
           jz .fail        ;ñòðàíèöà íå ïðèñóòñòâóåò
 
           test eax, 4     ;U/S
           jnz .fail       ;ïðèëîæåíèå îáðàòèëîñü ê ïàìÿòè
                           ;ÿäðà
           test eax, 8
           jnz .fail       ;óñòàíîâëåí çàðåçåðâèðîâàííûé áèò
                           ;â òàáëèöàõ ñòðàíèö. äîáàâëåíî â P4/Xeon
 
;ïîïûòêà çàïèñè â çàùèù¸ííóþ ñòðàíèöó ÿäðà
 
           cmp ebx, tss._io_map_0
           jb .fail
 
           cmp ebx, tss._io_map_0+8192
           jae .fail
 
; io permission map
; copy-on-write protection
 
           call alloc_page
           test eax, eax
           jz .fail
 
           push eax
           stdcall map_page,[ebp-4],eax,dword PG_SW
           pop eax
           mov edi, [.err_addr]
           and edi, -4096
           lea esi, [edi+(not tss._io_map_0)+1]; -tss._io_map_0
 
           mov ebx, esi
           shr ebx, 12
           mov edx, [current_slot]
           or eax, PG_SW
           mov [edx+APPDATA.io_map+ebx*4], eax
 
           add esi, [default_io_map]
           mov ecx, 4096/4
           cld
           rep movsd
           jmp .exit
 
 
;íå îáðàáàòûâàåì. Îøèáêà
 
.core_tabs:
.fail:
           mov esp, ebp
           popad
           add esp, 4
 
;           iretd
 
           save_ring3_context     ;debugger support
 
           mov bl, 14
           jmp exc_c
           iretd
endp
 
align 4
proc map_mem stdcall, lin_addr:dword,pdir:dword,\
                      ofs:dword,buf_size:dword
           mov eax, [buf_size]
           test eax, eax
           jz .exit
 
           mov eax, [pdir]
           and eax, 0xFFFFF000
 
           stdcall map_page,[ipc_pdir],eax,PG_UW
           mov ebx, [ofs]
           shr ebx, 22
           mov esi, [ipc_pdir]
           mov edi, [ipc_ptab]
           mov eax, [esi+ebx*4]
           and eax, 0xFFFFF000
           jz .exit
           stdcall map_page,edi,eax,PG_UW
;           inc ebx
;           add edi, 0x1000
;           mov eax, [esi+ebx*4]
;           test eax, eax
;           jz @f
;          and eax, 0xFFFFF000
;           stdcall map_page, edi, eax
 
@@:        mov edi, [lin_addr]
           and edi, 0xFFFFF000
           mov ecx, [buf_size]
           add ecx, 4095
           shr ecx, 12
           inc ecx
 
           mov edx, [ofs]
           shr edx, 12
           and edx, 0x3FF
           mov esi, [ipc_ptab]
 
.map:      mov eax, [esi+edx*4]
           and eax, 0xFFFFF000
           jz  .exit
           stdcall map_page,edi,eax,PG_UW
           dec ecx
           jz  .exit
           add edi, 0x1000
           inc edx
           cmp edx, 0x400
           jnz .map
           inc ebx
           mov eax, [ipc_pdir]
           mov eax, [eax+ebx*4]
           and eax, 0xFFFFF000
           jz  .exit
           stdcall map_page,esi,eax,PG_UW
           xor edx, edx
           jmp .map
 
.exit:
           ret
endp
 
align 4
proc map_memEx stdcall, lin_addr:dword,pdir:dword,\
                        ofs:dword,buf_size:dword
           mov eax, [buf_size]
           test eax, eax
           jz .exit
 
           mov eax, [pdir]
           and eax, 0xFFFFF000
 
           stdcall map_page,[proc_mem_pdir],eax,dword PG_UW
           mov ebx, [ofs]
           shr ebx, 22
           mov esi, [proc_mem_pdir]
           mov edi, [proc_mem_tab]
           mov eax, [esi+ebx*4]
           and eax, 0xFFFFF000
           test eax, eax
           jz .exit
           stdcall map_page,edi,eax,dword PG_UW
 
@@:        mov edi, [lin_addr]
           and edi, 0xFFFFF000
           mov ecx, [buf_size]
           add ecx, 4095
           shr ecx, 12
           inc ecx
 
           mov edx, [ofs]
           shr edx, 12
           and edx, 0x3FF
           mov esi, [proc_mem_tab]
 
.map:      mov eax, [esi+edx*4]
;           and eax, 0xFFFFF000
;           test eax, eax
;           jz .exit
           stdcall map_page,edi,eax,dword PG_UW
           add edi, 0x1000
           inc edx
           dec ecx
           jnz .map
.exit:
           ret
endp
 
 
 
 
sys_IPC:
;input:
;  eax=1 - set ipc buffer area
;    ebx=address of buffer
;    ecx=size of buffer
;  eax=2 - send message
;    ebx=PID
;    ecx=address of message
;    edx=size of message
 
           cmp  eax,1
           jne @f
           call set_ipc_buff
           mov [esp+36], eax
           ret
@@:
           cmp eax, 2
           jne @f
           stdcall sys_ipc_send, ebx, ecx, edx
           mov [esp+36], eax
           ret
@@:
           xor eax, eax
           not eax
           mov [esp+36], eax
           ret
 
align 4
proc set_ipc_buff
 
           mov  eax,[current_slot]
           pushf
           cli
           mov  [eax+APPDATA.ipc_start],ebx     ;set fields in extended information area
           mov  [eax+APPDATA.ipc_size],ecx
 
           add ecx, ebx
           add ecx, 4095
           and ecx, not 4095
 
.touch:    mov eax, [ebx]
           add ebx, 0x1000
           cmp ebx, ecx
           jb  .touch
 
           popf
           xor eax, eax
           ret
endp
 
proc sys_ipc_send stdcall, PID:dword, msg_addr:dword, msg_size:dword
           locals
             dst_slot   dd ?
             dst_offset dd ?
             buf_size   dd ?
             used_buf   dd ?
           endl
 
           pushf
           cli
 
           mov  eax, [PID]
           call pid_to_slot
           test eax,eax
           jz   .no_pid
 
           mov [dst_slot], eax
           shl  eax,8
           mov  edi,[eax+SLOT_BASE+0xa0]  ;is ipc area defined?
           test edi,edi
           jz   .no_ipc_area
 
           mov ebx, edi
           and ebx, 0xFFF
           mov [dst_offset], ebx
 
           mov esi, [eax+SLOT_BASE+0xa4]
           mov [buf_size], esi
 
           mov ecx, [ipc_tmp]
           cmp esi, 0x40000-0x1000 ; size of [ipc_tmp] minus one page
           jbe @f
           push eax esi edi
           add esi,0x1000
           stdcall alloc_kernel_space,esi
           mov ecx, eax
           pop edi esi eax
@@:
           mov [used_buf], ecx
           stdcall map_mem, ecx, [SLOT_BASE+eax+0xB8],\
                             edi, esi
 
           mov edi, [dst_offset]
           add edi, [used_buf]
           cmp dword [edi], 0
           jnz  .ipc_blocked          ;if dword [buffer]<>0 - ipc blocked now
 
           mov edx, dword [edi+4]
           lea ebx, [edx+8]
           add ebx, [msg_size]
           cmp ebx, [buf_size]
           ja .buffer_overflow         ;esi<0 - not enough memory in buffer
 
           mov dword [edi+4], ebx
           mov eax,[TASK_BASE]
           mov eax, [eax+0x04]         ;eax - our PID
           add edi, edx
           mov [edi], eax
           mov ecx, [msg_size]
 
           mov [edi+4], ecx
           add edi, 8
           mov esi, [msg_addr]
       ;    add esi, new_app_base
           cld
           rep movsb
 
           mov ebx, [ipc_tmp]
           mov edx, ebx
           shr ebx, 12
           xor eax, eax
           mov [page_tabs+ebx*4], eax
           invlpg [edx]
 
           mov ebx, [ipc_pdir]
           mov edx, ebx
           shr ebx, 12
           xor eax, eax
           mov [page_tabs+ebx*4], eax
           invlpg [edx]
 
           mov ebx, [ipc_ptab]
           mov edx, ebx
           shr ebx, 12
           xor eax, eax
           mov [page_tabs+ebx*4], eax
           invlpg [edx]
 
           mov  eax, [dst_slot]
           shl eax, 8
           or   [eax+SLOT_BASE+0xA8],dword 0x40
           cmp  dword [check_idle_semaphore],20
           jge  .ipc_no_cis
 
           mov  dword [check_idle_semaphore],5
.ipc_no_cis:
           push 0
           jmp .ret
.no_pid:
           popf
           mov eax, 4
           ret
.no_ipc_area:
           popf
           xor eax, eax
           inc eax
           ret
.ipc_blocked:
           push 2
           jmp .ret
.buffer_overflow:
           push 3
.ret:
           mov eax, [used_buf]
           cmp eax, [ipc_tmp]
           jz @f
           stdcall free_kernel_space,eax
@@:
           pop eax
           popf
           ret
endp
 
align 4
sysfn_meminfo:
 
        ;   add ebx, new_app_base
           cmp ebx, OS_BASE
           jae .fail
 
           mov eax, [pg_data.pages_count]
           mov [ebx], eax
           shl eax, 12
           mov [esp+36], eax
           mov ecx, [pg_data.pages_free]
           mov [ebx+4], ecx
           mov edx, [pg_data.pages_faults]
           mov [ebx+8], edx
           mov esi, [heap_size]
           mov [ebx+12], esi
           mov edi, [heap_free]
           mov [ebx+16], edi
           mov eax, [heap_blocks]
           mov [ebx+20], eax
           mov ecx, [free_blocks]
           mov [ebx+24], ecx
           ret
.fail:
           mov dword [esp+36], -1
           ret
 
align 4
new_services:
 
           cmp  eax,4
           jle  sys_sheduler
 
           cmp eax, 11
           jb .fail
           ja @f
 
           call init_heap
           mov [esp+36], eax
           ret
@@:
           cmp eax, 12
           ja @f
 
           stdcall user_alloc, ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 13
           ja @f
           stdcall user_free, ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 14
           ja @f
           cmp ebx, OS_BASE
           jae .fail
           stdcall get_event_ex, ebx, ecx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 15
           ja @f
           mov ecx, [current_slot]
           mov eax, [ecx+APPDATA.fpu_handler]
           mov [ecx+APPDATA.fpu_handler], ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 16
           ja @f
 
           test ebx, ebx
           jz .fail
           cmp ebx, OS_BASE
           jae .fail
           stdcall get_service, ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 17
           ja @f
           call srv_handlerEx   ;ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 18
           ja @f
           mov ecx, [current_slot]
           mov eax, [ecx+APPDATA.sse_handler]
           mov [ecx+APPDATA.sse_handler], ebx
           mov [esp+36], eax
           ret
@@:
           cmp eax, 19
           ja @f
           cmp ebx, OS_BASE
           jae .fail
           stdcall load_library, ebx
           mov [esp+36], eax
           ret
@@:
           cmp     eax, 20
           ja      @F
           mov     eax, ecx
           call    user_realloc
           mov     [esp+36], eax
           ret
@@:
           cmp eax, 21                     ;for test purposes only
           ja @f                           ;will be removed soon
           cmp ebx, OS_BASE
           jae .fail
 
           stdcall load_PE, ebx
 
           test eax, eax
           jz @F
 
           mov esi, eax
           stdcall eax, DRV_ENTRY
 
           test eax, eax
           jz @F
 
           mov [eax+SRV.entry], esi
 
@@:
           mov [esp+36], eax
           ret
 
 
.fail:
           xor eax, eax
           mov [esp+36], eax
           ret
 
align 4
proc init_mtrr
 
           cmp [BOOT_VAR+0x901c],byte 2
           je  .exit
 
           bt [cpu_caps], CAPS_MTRR
           jnc .exit
 
           mov eax, cr0
           or eax, 0x60000000   ;disable caching
           mov cr0, eax
           wbinvd               ;invalidate cache
 
           mov ecx, 0x2FF
           rdmsr                ;
           push eax
 
           xor edx, edx
           xor eax, eax
           mov ecx, 0x2FF
           wrmsr                ;disable all MTRR
 
           stdcall set_mtrr, dword 0,dword 0,[MEM_AMOUNT],MEM_WB
           stdcall set_mtrr, dword 1,[LFBAddress],[LFBSize],MEM_WC
           xor edx, edx
           xor eax, eax
           mov ecx, 0x204
           mov ebx, 6
@@:
           wrmsr                ;disable unused MTRR
           inc ecx
           wrmsr
           inc ecx
           dec ebx
           jnz @b
 
           wbinvd               ;again invalidate
 
           pop eax
           or eax, 0x800        ;set default memtype to UC
           and al, 0xF0
           mov ecx, 0x2FF
           wrmsr                ;and enable MTRR
 
           mov eax, cr0
           and eax, not 0x60000000
           mov cr0, eax         ; enable caching
.exit:
           ret
endp
 
align 4
proc set_mtrr stdcall, reg:dword,base:dword,size:dword,mem_type:dword
 
           xor edx, edx
           mov eax, [base]
           or eax, [mem_type]
           mov ecx, [reg]
           lea ecx, [0x200+ecx*2]
           wrmsr
 
           mov ebx, [size]
           dec ebx
           mov eax, 0xFFFFFFFF
           mov edx, 0x0000000F
           sub eax, ebx
           sbb edx, 0
           or eax, 0x800
           inc ecx
           wrmsr
           ret
endp
 
 
 
align 4
proc stall stdcall, delay:dword
           push ecx
           push edx
           push ebx
           push eax
 
           mov eax, [delay]
           mul [stall_mcs]
           mov ebx, eax       ;low
           mov ecx, edx       ;high
           rdtsc
           add ebx, eax
           adc ecx,edx
@@:
           rdtsc
           sub eax, ebx
           sbb edx, ecx
           jb @B
 
           pop eax
           pop ebx
           pop edx
           pop ecx
           ret
endp
 
align 4
proc create_ring_buffer stdcall, size:dword, flags:dword
           locals
             buf_ptr  dd ?
           endl
 
           mov eax, [size]
           test eax, eax
           jz .fail
 
           add eax, eax
           stdcall alloc_kernel_space, eax
           test eax, eax
           jz .fail
 
           push ebx
 
           mov [buf_ptr], eax
 
           mov ebx, [size]
           shr ebx, 12
           push ebx
 
           stdcall alloc_pages, ebx
           pop ecx
 
           test eax, eax
           jz .mm_fail
 
           push edi
 
           or eax, [flags]
           mov edi, [buf_ptr]
           mov ebx, [buf_ptr]
           mov edx, ecx
           shl edx, 2
           shr edi, 10
@@:
           mov [page_tabs+edi], eax
           mov [page_tabs+edi+edx], eax
           invlpg [ebx]
           invlpg [ebx+esi]
           add eax, 0x1000
           add ebx, 0x1000
           add edi, 4
           dec ecx
           jnz @B
 
           mov eax, [buf_ptr]
           pop edi
           pop ebx
           ret
.mm_fail:
           stdcall free_kernel_space, [buf_ptr]
           xor eax, eax
           pop ebx
.fail:
           ret
endp
 
if 0
     push eax
     push edx
     mov edx, 0x400   ;bochs
     mov al,0xff      ;bochs
     out dx, al       ;bochs
     pop edx
     pop eax
end if

Rev 747	Rev 750
1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2	;; ;;	2	;; ;;
3	;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;;	3	;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;;
4	;; Distributed under terms of the GNU General Public License ;;	4	;; Distributed under terms of the GNU General Public License ;;
5	;; ;;	5	;; ;;
6	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	6	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
7		7
8	$Revision: 747 $	8	$Revision: 750 $
9		9
10		10
11	align 4	11	align 4
12	proc alloc_page	12	proc alloc_page
13		13
14	pushfd	14	pushfd
15	cli	15	cli
16	push ebx	16	push ebx
17	mov ebx, [page_start]	17	mov ebx, [page_start]
18	mov ecx, [page_end]	18	mov ecx, [page_end]
19	.l1:	19	.l1:
20	bsf eax,[ebx];	20	bsf eax,[ebx];
21	jnz .found	21	jnz .found
22	add ebx,4	22	add ebx,4
23	cmp ebx, ecx	23	cmp ebx, ecx
24	jb .l1	24	jb .l1
25	pop ebx	25	pop ebx
26	popfd	26	popfd
27	xor eax,eax	27	xor eax,eax
28	ret	28	ret
29	.found:	29	.found:
30	btr [ebx], eax	30	btr [ebx], eax
31	mov [page_start],ebx	31	mov [page_start],ebx
32	sub ebx, sys_pgmap	32	sub ebx, sys_pgmap
33	lea eax, [eax+ebx*8]	33	lea eax, [eax+ebx*8]
34	shl eax, 12	34	shl eax, 12
35	dec [pg_data.pages_free]	35	dec [pg_data.pages_free]
36	pop ebx	36	pop ebx
37	popfd	37	popfd
38	ret	38	ret
39	endp	39	endp
40		40
41	align 4	41	align 4
42	proc alloc_pages stdcall, count:dword	42	proc alloc_pages stdcall, count:dword
43	pushfd	43	pushfd
44	push ebx	44	push ebx
45	push edi	45	push edi
46	cli	46	cli
47	mov eax, [count]	47	mov eax, [count]
48	add eax, 7	48	add eax, 7
49	shr eax, 3	49	shr eax, 3
50	mov [count], eax	50	mov [count], eax
51	cmp eax, [pg_data.pages_free]	51	cmp eax, [pg_data.pages_free]
52	ja .fail	52	ja .fail
53		53
54	mov ecx, [page_start]	54	mov ecx, [page_start]
55	mov ebx, [page_end]	55	mov ebx, [page_end]
56	.find:	56	.find:
57	mov edx, [count]	57	mov edx, [count]
58	mov edi, ecx	58	mov edi, ecx
59	.match:	59	.match:
60	cmp byte [ecx], 0xFF	60	cmp byte [ecx], 0xFF
61	jne .next	61	jne .next
62	dec edx	62	dec edx
63	jz .ok	63	jz .ok
64	inc ecx	64	inc ecx
65	cmp ecx,ebx	65	cmp ecx,ebx
66	jb .match	66	jb .match
67	.fail:	67	.fail:
68	xor eax, eax	68	xor eax, eax
69	pop edi	69	pop edi
70	pop ebx	70	pop ebx
71	popfd	71	popfd
72	ret	72	ret
73	.next:	73	.next:
74	inc ecx	74	inc ecx
75	cmp ecx, ebx	75	cmp ecx, ebx
76	jb .find	76	jb .find
77	pop edi	77	pop edi
78	pop ebx	78	pop ebx
79	popfd	79	popfd
80	xor eax, eax	80	xor eax, eax
81	ret	81	ret
82	.ok:	82	.ok:
83	sub ecx, edi	83	sub ecx, edi
84	inc ecx	84	inc ecx
85	push esi	85	push esi
86	mov esi, edi	86	mov esi, edi
87	xor eax, eax	87	xor eax, eax
88	rep stosb	88	rep stosb
89	sub esi, sys_pgmap	89	sub esi, sys_pgmap
90	shl esi, 3+12	90	shl esi, 3+12
91	mov eax, esi	91	mov eax, esi
92	mov ebx, [count]	92	mov ebx, [count]
93	shl ebx, 3	93	shl ebx, 3
94	sub [pg_data.pages_free], ebx	94	sub [pg_data.pages_free], ebx
95	pop esi	95	pop esi
96	pop edi	96	pop edi
97	pop ebx	97	pop ebx
98	popfd	98	popfd
99	ret	99	ret
100	endp	100	endp
101		101
102	align 4	102	align 4
103	proc map_page stdcall,lin_addr:dword,phis_addr:dword,flags:dword	103	proc map_page stdcall,lin_addr:dword,phis_addr:dword,flags:dword
104	push ebx	104	push ebx
105	mov eax, [phis_addr]	105	mov eax, [phis_addr]
106	and eax, not 0xFFF	106	and eax, not 0xFFF
107	or eax, [flags]	107	or eax, [flags]
108	mov ebx, [lin_addr]	108	mov ebx, [lin_addr]
109	shr ebx, 12	109	shr ebx, 12
110	mov [page_tabs+ebx*4], eax	110	mov [page_tabs+ebx*4], eax
111	mov eax, [lin_addr]	111	mov eax, [lin_addr]
112	invlpg [eax]	112	invlpg [eax]
113	pop ebx	113	pop ebx
114	ret	114	ret
115	endp	115	endp
116		116
117	align 4	117	align 4
118	map_space: ;not implemented	118	map_space: ;not implemented
119		119
120		120
121	ret	121	ret
122		122
123		123
124	align 4	124	align 4
125	proc free_page	125	proc free_page
126	;arg: eax page address	126	;arg: eax page address
127	pushfd	127	pushfd
128	cli	128	cli
129	shr eax, 12 ;page index	129	shr eax, 12 ;page index
130	bts dword [sys_pgmap], eax ;that's all!	130	bts dword [sys_pgmap], eax ;that's all!
131	cmc	131	cmc
132	adc [pg_data.pages_free], 0	132	adc [pg_data.pages_free], 0
133	shr eax, 3	133	shr eax, 3
134	and eax, not 3 ;dword offset from page_map	134	and eax, not 3 ;dword offset from page_map
135	add eax, sys_pgmap	135	add eax, sys_pgmap
136	cmp [page_start], eax	136	cmp [page_start], eax
137	ja @f	137	ja @f
138	popfd	138	popfd
139	ret	139	ret
140	@@:	140	@@:
141	mov [page_start], eax	141	mov [page_start], eax
142	popfd	142	popfd
143	ret	143	ret
144	endp	144	endp
145		145
146	proc map_io_mem stdcall, base:dword, size:dword, flags:dword	146	proc map_io_mem stdcall, base:dword, size:dword, flags:dword
147		147
148	push ebx	148	push ebx
149	push edi	149	push edi
150	mov eax, [size]	150	mov eax, [size]
151	add eax, 4095	151	add eax, 4095
152	and eax, -4096	152	and eax, -4096
153	mov [size], eax	153	mov [size], eax
154	stdcall alloc_kernel_space, eax	154	stdcall alloc_kernel_space, eax
155	test eax, eax	155	test eax, eax
156	jz .fail	156	jz .fail
157	push eax	157	push eax
158		158
159	mov edi, 0x1000	159	mov edi, 0x1000
160	mov ebx, eax	160	mov ebx, eax
161	mov ecx,[size]	161	mov ecx,[size]
162	mov edx, [base]	162	mov edx, [base]
163	shr eax, 12	163	shr eax, 12
164	shr ecx, 12	164	shr ecx, 12
165	and edx, -4096	165	and edx, -4096
166	or edx, [flags]	166	or edx, [flags]
167	@@:	167	@@:
168	mov [page_tabs+eax*4], edx	168	mov [page_tabs+eax*4], edx
169	invlpg [ebx]	169	invlpg [ebx]
170	inc eax	170	inc eax
171	add ebx, edi	171	add ebx, edi
172	add edx, edi	172	add edx, edi
173	loop @B	173	loop @B
174		174
175	pop eax	175	pop eax
176	mov edx, [base]	176	mov edx, [base]
177	and edx, 4095	177	and edx, 4095
178	add eax, edx	178	add eax, edx
179	.fail:	179	.fail:
180	pop edi	180	pop edi
181	pop ebx	181	pop ebx
182	ret	182	ret
183	endp	183	endp
184		184
185	; param	185	; param
186	; eax= page base + page flags	186	; eax= page base + page flags
187	; ebx= liear address	187	; ebx= liear address
188	; ecx= count	188	; ecx= count
189		189
190	align 4	190	align 4
191	commit_pages:	191	commit_pages:
192	push edi	192	push edi
193	test ecx, ecx	193	test ecx, ecx
194	jz .fail	194	jz .fail
195		195
196	mov edi, ebx	196	mov edi, ebx
197	mov ebx, pg_data.pg_mutex	197	mov ebx, pg_data.pg_mutex
198	call wait_mutex ;ebx	198	call wait_mutex ;ebx
199		199
200	mov edx, 0x1000	200	mov edx, 0x1000
201	mov ebx, edi	201	mov ebx, edi
202	shr ebx, 12	202	shr ebx, 12
203	@@:	203	@@:
204	mov [page_tabs+ebx*4], eax	204	mov [page_tabs+ebx*4], eax
205	invlpg [edi]	205	invlpg [edi]
206	add edi, edx	206	add edi, edx
207	add eax, edx	207	add eax, edx
208	inc ebx	208	inc ebx
209	dec ecx	209	dec ecx
210	jnz @B	210	jnz @B
211	mov [pg_data.pg_mutex],ecx	211	mov [pg_data.pg_mutex],ecx
212	.fail:	212	.fail:
213	pop edi	213	pop edi
214	ret	214	ret
215		215
216		216
217	; param	217	; param
218	; eax= base	218	; eax= base
219	; ecx= count	219	; ecx= count
220		220
221	align 4	221	align 4
222	release_pages:	222	release_pages:
223		223
224	pushad	224	pushad
225	mov ebx, pg_data.pg_mutex	225	mov ebx, pg_data.pg_mutex
226	call wait_mutex ;ebx	226	call wait_mutex ;ebx
227		227
228	mov esi, eax	228	mov esi, eax
229	mov edi, eax	229	mov edi, eax
230		230
231	shr esi, 10	231	shr esi, 10
232	add esi, page_tabs	232	add esi, page_tabs
233		233
234	mov ebp, [pg_data.pages_free]	234	mov ebp, [pg_data.pages_free]
235	mov ebx, [page_start]	235	mov ebx, [page_start]
236	mov edx, sys_pgmap	236	mov edx, sys_pgmap
237	@@:	237	@@:
238	xor eax, eax	238	xor eax, eax
239	xchg eax, [esi]	239	xchg eax, [esi]
240	invlpg [edi]	240	invlpg [edi]
241		241
242	test eax, 1	242	test eax, 1
243	jz .next	243	jz .next
244		244
245	shr eax, 12	245	shr eax, 12
246	bts [edx], eax	246	bts [edx], eax
247	cmc	247	cmc
248	adc ebp, 0	248	adc ebp, 0
249	shr eax, 3	249	shr eax, 3
250	and eax, -4	250	and eax, -4
251	add eax, edx	251	add eax, edx
252	cmp eax, ebx	252	cmp eax, ebx
253	jae .next	253	jae .next
254		254
255	mov ebx, eax	255	mov ebx, eax
256	.next:	256	.next:
257	add edi, 0x1000	257	add edi, 0x1000
258	add esi, 4	258	add esi, 4
259	dec ecx	259	dec ecx
260	jnz @B	260	jnz @B
261	mov [pg_data.pages_free], ebp	261	mov [pg_data.pages_free], ebp
262	and [pg_data.pg_mutex],0	262	and [pg_data.pg_mutex],0
263	popad	263	popad
264	ret	264	ret
265		265
266	align 4	266	align 4
267	proc map_page_table stdcall, lin_addr:dword, phis_addr:dword	267	proc map_page_table stdcall, lin_addr:dword, phis_addr:dword
268	push ebx	268	push ebx
269	mov ebx, [lin_addr]	269	mov ebx, [lin_addr]
270	shr ebx, 22	270	shr ebx, 22
271	mov eax, [phis_addr]	271	mov eax, [phis_addr]
272	and eax, not 0xFFF	272	and eax, not 0xFFF
273	or eax, PG_UW ;+PG_NOCACHE	273	or eax, PG_UW ;+PG_NOCACHE
274	mov dword [master_tab+ebx*4], eax	274	mov dword [master_tab+ebx*4], eax
275	mov eax, [lin_addr]	275	mov eax, [lin_addr]
276	shr eax, 10	276	shr eax, 10
277	add eax, page_tabs	277	add eax, page_tabs
278	invlpg [eax]	278	invlpg [eax]
279	pop ebx	279	pop ebx
280	ret	280	ret
281	endp	281	endp
282		282
283	align 4	283	align 4
284	proc init_LFB	284	proc init_LFB
285	locals	285	locals
286	pg_count dd ?	286	pg_count dd ?
287	endl	287	endl
288		288
289	cmp dword [LFBAddress], -1	289	cmp dword [LFBAddress], -1
290	jne @f	290	jne @f
291	mov [BOOT_VAR+0x901c],byte 2	291	mov [BOOT_VAR+0x901c],byte 2
292	stdcall kernel_alloc, 0x280000	292	stdcall kernel_alloc, 0x280000
293	mov [LFBAddress], eax	293	mov [LFBAddress], eax
294	ret	294	ret
295	@@:	295	@@:
296	test [SCR_MODE],word 0100000000000000b	296	test [SCR_MODE],word 0100000000000000b
297	jnz @f	297	jnz @f
298	mov [BOOT_VAR+0x901c],byte 2	298	mov [BOOT_VAR+0x901c],byte 2
299	ret	299	ret
300	@@:	300	@@:
301	call init_mtrr	301	call init_mtrr
302		302
303	mov edx, LFB_BASE	303	mov edx, LFB_BASE
304	mov esi, [LFBAddress]	304	mov esi, [LFBAddress]
305	mov edi, 0x00800000	305	mov edi, 0x00800000
306	mov dword [exp_lfb+4], edx	306	mov dword [exp_lfb+4], edx
307		307
308	shr edi, 12	308	shr edi, 12
309	mov [pg_count], edi	309	mov [pg_count], edi
310	shr edi, 10	310	shr edi, 10
311		311
312	bt [cpu_caps], CAPS_PSE	312	bt [cpu_caps], CAPS_PSE
313	jnc .map_page_tables	313	jnc .map_page_tables
314	or esi, PG_LARGE+PG_UW	314	or esi, PG_LARGE+PG_UW
315	mov edx, sys_pgdir+(LFB_BASE shr 20)	315	mov edx, sys_pgdir+(LFB_BASE shr 20)
316	@@:	316	@@:
317	mov [edx], esi	317	mov [edx], esi
318	add edx, 4	318	add edx, 4
319	add esi, 0x00400000	319	add esi, 0x00400000
320	dec edi	320	dec edi
321	jnz @B	321	jnz @B
322		322
323	bt [cpu_caps], CAPS_PGE	323	bt [cpu_caps], CAPS_PGE
324	jnc @F	324	jnc @F
325	or dword [sys_pgdir+(LFB_BASE shr 20)], PG_GLOBAL	325	or dword [sys_pgdir+(LFB_BASE shr 20)], PG_GLOBAL
326	@@:	326	@@:
327	mov dword [LFBAddress], LFB_BASE	327	mov dword [LFBAddress], LFB_BASE
328	mov eax, cr3 ;flush TLB	328	mov eax, cr3 ;flush TLB
329	mov cr3, eax	329	mov cr3, eax
330	ret	330	ret
331		331
332	.map_page_tables:	332	.map_page_tables:
333		333
334	@@:	334	@@:
335	call alloc_page	335	call alloc_page
336	stdcall map_page_table, edx, eax	336	stdcall map_page_table, edx, eax
337	add edx, 0x00400000	337	add edx, 0x00400000
338	dec edi	338	dec edi
339	jnz @B	339	jnz @B
340		340
341	mov eax, [LFBAddress]	341	mov eax, [LFBAddress]
342	mov edi, page_tabs + (LFB_BASE shr 10)	342	mov edi, page_tabs + (LFB_BASE shr 10)
343	or eax, PG_UW	343	or eax, PG_UW
344	mov ecx, [pg_count]	344	mov ecx, [pg_count]
345	cld	345	cld
346	@@:	346	@@:
347	stosd	347	stosd
348	add eax, 0x1000	348	add eax, 0x1000
349	dec ecx	349	dec ecx
350	jnz @B	350	jnz @B
351		351
352	mov dword [LFBAddress], LFB_BASE	352	mov dword [LFBAddress], LFB_BASE
353	mov eax, cr3 ;flush TLB	353	mov eax, cr3 ;flush TLB
354	mov cr3, eax	354	mov cr3, eax
355		355
356	ret	356	ret
357	endp	357	endp
358		358
359	align 4	359	align 4
360	proc new_mem_resize stdcall, new_size:dword	360	proc new_mem_resize stdcall, new_size:dword
361		361
362	mov ebx, pg_data.pg_mutex	362	mov ebx, pg_data.pg_mutex
363	call wait_mutex ;ebx	363	call wait_mutex ;ebx
364		364
365	mov edi, [new_size]	365	mov edi, [new_size]
366	add edi,4095	366	add edi,4095
367	and edi,not 4095	367	and edi,not 4095
368	mov [new_size], edi	368	mov [new_size], edi
369		369
370	mov edx,[current_slot]	370	mov edx,[current_slot]
371	cmp [edx+APPDATA.heap_base],0	371	cmp [edx+APPDATA.heap_base],0
372	jne .exit	372	jne .exit
373		373
374	mov esi, [edx+APPDATA.mem_size]	374	mov esi, [edx+APPDATA.mem_size]
375	add esi, 4095	375	add esi, 4095
376	and esi, not 4095	376	and esi, not 4095
377		377
378	cmp edi, esi	378	cmp edi, esi
379	jae .expand	379	jae .expand
380		380
381	shr edi, 12	381	shr edi, 12
382	shr esi, 12	382	shr esi, 12
383	@@:	383	@@:
384	mov eax, [app_page_tabs+edi*4]	384	mov eax, [app_page_tabs+edi*4]
385	test eax, 1	385	test eax, 1
386	jz .next	386	jz .next
387	mov dword [app_page_tabs+edi*4], 2	387	mov dword [app_page_tabs+edi*4], 2
388	mov ebx, edi	388	mov ebx, edi
389	shl ebx, 12	389	shl ebx, 12
390	invlpg [ebx]	390	invlpg [ebx]
391	call free_page	391	call free_page
392		392
393	.next: add edi, 1	393	.next: add edi, 1
394	cmp edi, esi	394	cmp edi, esi
395	jb @B	395	jb @B
396		396
397	.update_size:	397	.update_size:
398	mov ebx, [new_size]	398	mov ebx, [new_size]
399	call update_mem_size	399	call update_mem_size
400		400
401	xor eax, eax	401	xor eax, eax
402	dec [pg_data.pg_mutex]	402	dec [pg_data.pg_mutex]
403	ret	403	ret
404	.expand:	404	.expand:
405		405
406	push esi	406	push esi
407	push edi	407	push edi
408		408
409	add edi, 0x3FFFFF	409	add edi, 0x3FFFFF
410	and edi, not(0x3FFFFF)	410	and edi, not(0x3FFFFF)
411	add esi, 0x3FFFFF	411	add esi, 0x3FFFFF
412	and esi, not(0x3FFFFF)	412	and esi, not(0x3FFFFF)
413		413
414	cmp esi, edi	414	cmp esi, edi
415	jae .grow	415	jae .grow
416		416
417	xchg esi, edi	417	xchg esi, edi
418		418
419	@@:	419	@@:
420	call alloc_page	420	call alloc_page
421	test eax, eax	421	test eax, eax
422	jz .exit	422	jz .exit
423		423
424	stdcall map_page_table, edi, eax	424	stdcall map_page_table, edi, eax
425		425
426	push edi	426	push edi
427	shr edi, 10	427	shr edi, 10
428	add edi, page_tabs	428	add edi, page_tabs
429	mov ecx, 1024	429	mov ecx, 1024
430	xor eax, eax	430	xor eax, eax
431	cld	431	cld
432	rep stosd	432	rep stosd
433	pop edi	433	pop edi
434		434
435	add edi, 0x00400000	435	add edi, 0x00400000
436	cmp edi, esi	436	cmp edi, esi
437	jb @B	437	jb @B
438	.grow:	438	.grow:
439	pop edi	439	pop edi
440	pop esi	440	pop esi
441	@@:	441	@@:
442	call alloc_page	442	call alloc_page
443	test eax, eax	443	test eax, eax
444	jz .exit	444	jz .exit
445	stdcall map_page,esi,eax,dword PG_UW	445	stdcall map_page,esi,eax,dword PG_UW
446		446
447	push edi	447	push edi
448	mov edi, esi	448	mov edi, esi
449	xor eax, eax	449	xor eax, eax
450	mov ecx, 1024	450	mov ecx, 1024
451	cld	451	cld
452	rep stosd	452	rep stosd
453	pop edi	453	pop edi
454		454
455	add esi, 0x1000	455	add esi, 0x1000
456	cmp esi, edi	456	cmp esi, edi
457	jb @B	457	jb @B
458		458
459	jmp .update_size	459	jmp .update_size
460	.exit:	460	.exit:
461	xor eax, eax	461	xor eax, eax
462	inc eax	462	inc eax
463	dec [pg_data.pg_mutex]	463	dec [pg_data.pg_mutex]
464	ret	464	ret
465	endp	465	endp
466		466
467	update_mem_size:	467	update_mem_size:
468	; in: edx = slot base	468	; in: edx = slot base
469	; ebx = new memory size	469	; ebx = new memory size
470	; destroys eax,ecx,edx	470	; destroys eax,ecx,edx
471		471
472	mov [APPDATA.mem_size+edx],ebx	472	mov [APPDATA.mem_size+edx],ebx
473	;search threads and update	473	;search threads and update
474	;application memory size infomation	474	;application memory size infomation
475	mov ecx,[APPDATA.dir_table+edx]	475	mov ecx,[APPDATA.dir_table+edx]
476	mov eax,2	476	mov eax,2
477		477
478	.search_threads:	478	.search_threads:
479	;eax = current slot	479	;eax = current slot
480	;ebx = new memory size	480	;ebx = new memory size
481	;ecx = page directory	481	;ecx = page directory
482	cmp eax,[TASK_COUNT]	482	cmp eax,[TASK_COUNT]
483	jg .search_threads_end	483	jg .search_threads_end
484	mov edx,eax	484	mov edx,eax
485	shl edx,5	485	shl edx,5
486	cmp word [CURRENT_TASK+edx+TASKDATA.state],9 ;if slot empty?	486	cmp word [CURRENT_TASK+edx+TASKDATA.state],9 ;if slot empty?
487	jz .search_threads_next	487	jz .search_threads_next
488	shl edx,3	488	shl edx,3
489	cmp [SLOT_BASE+edx+APPDATA.dir_table],ecx ;if it is our thread?	489	cmp [SLOT_BASE+edx+APPDATA.dir_table],ecx ;if it is our thread?
490	jnz .search_threads_next	490	jnz .search_threads_next
491	mov [SLOT_BASE+edx+APPDATA.mem_size],ebx ;update memory size	491	mov [SLOT_BASE+edx+APPDATA.mem_size],ebx ;update memory size
492	.search_threads_next:	492	.search_threads_next:
493	inc eax	493	inc eax
494	jmp .search_threads	494	jmp .search_threads
495	.search_threads_end:	495	.search_threads_end:
496	ret	496	ret
497		497
498	; param	498	; param
499	; eax= linear address	499	; eax= linear address
500	;	500	;

Subversion Repositories Kolibri OS

(root)/kernel/trunk/core/memory.inc – Rev 747 → 750