WebSVN – Kolibri OS – Diff – /kernel/branches/flat_kernel/core/taskman.inc


$Revision: 437 $
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                              ;;
;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;;
;; Distributed under terms of the GNU General Public License    ;;
;;                                                              ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
GREEDY_KERNEL  equ 0
 
struc APP_HEADER_00
{ .banner      dq ?
  .version     dd ?    ;+8
  .start       dd ?    ;+12
  .i_end       dd ?    ;+16
  .mem_size    dd ?    ;+20
  .i_param     dd ?    ;+24
}
 
struc APP_HEADER_01
{ .banner      dq ?
  .version     dd ?    ;+8
  .start       dd ?    ;+12
  .i_end       dd ?    ;+16
  .mem_size    dd ?    ;+20
  .stack_top   dd ?    ;+24
  .i_param     dd ?    ;+28
  .i_icon      dd ?    ;+32
}
 
 
struc APP_PARAMS
{ .app_cmdline   ;0x00
  .app_path      ;0x04
  .app_eip       ;0x08
  .app_esp       ;0x0C
  .app_mem       ;0x10
}
 
macro _clear_ op
{  mov ecx, op/4
   xor eax, eax
   cld
   rep stosd
}
 
 
align 4
proc fs_execute
 
;fn_read:dword, file_size:dword, cluster:dword
 
; ebx - cmdline
; edx - flags
; ebp - full filename
; [esp+4] = procedure DoRead, [esp+8] = filesize & [esp+12]... - arguments for it
 
           locals
             cmdline       rd 64        ;256/4
             filename      rd 256       ;1024/4
             flags         dd ?
 
             save_cr3      dd ?
             slot          dd ?
             slot_base     dd ?
             file_base     dd ?
             file_size     dd ?
                                          ;app header data
             hdr_cmdline   dd ? ;0x00
             hdr_path      dd ? ;0x04
             hdr_eip       dd ? ;0x08
             hdr_esp       dd ? ;0x0C
             hdr_mem       dd ? ;0x10
             hdr_i_end     dd ? ;0x14
           endl
 
           pushad
 
           mov [cmdline], ebx
           mov [flags], edx
 
; [ebp]  pointer to filename
 
           lea eax, [filename]
           mov dword [eax+1020],0              ;force terminate
                                               ;string
           stdcall k_strncpy, eax, [ebp], 1023
 
           lea eax, [cmdline]
           mov dword [eax+252], 0
           stdcall k_strncpy, eax, [cmdline], 255
 
           lea eax, [filename]
           stdcall load_file, eax
           mov  ecx, -ERROR_FILE_NOT_FOUND
           test eax, eax
           jz .err_file
 
           mov [file_base], eax
           mov [file_size], ebx
 
           lea ebx, [hdr_cmdline]
           call test_app_header
           mov ecx, -0x1F
           test eax, eax
           jz .err_hdr
 
           mov esi, new_process_loading
           call sys_msg_board_str       ; write message to message board
 
.wait_lock:
           cmp [application_table_status],0
           je .get_lock
           call   change_task
           jmp .wait_lock
 
.get_lock:
           mov eax, 1
           xchg eax, [application_table_status]
           cmp eax, 0
           jne .wait_lock
 
           call set_application_table_status
 
           call get_new_process_place
           test eax, eax
           mov ecx, -0x20      ; too many processes
           jz .err
 
           mov [slot], eax
           shl eax, 8
           add eax, SLOT_BASE
           mov [slot_base], eax
           mov edi, eax
           _clear_ 256     ;clean extended information about process
 
; write application name
           lea edi, [filename]
           mov al, '/'
           call k_strrchr  ; now eax points to name without path
 
           lea esi, [eax+1]
           test eax, eax
           jnz @F
           lea esi, [filename]
@@:
           mov ecx, 8  ; 8 chars for name
           mov edi, [slot_base]
.copy_process_name_loop:
           lodsb
           cmp al, '.'
           jz .copy_process_name_done
           test al, al
           jz .copy_process_name_done
           stosb
           loop .copy_process_name_loop
.copy_process_name_done:
 
           mov ebx, cr3
           mov [save_cr3], ebx
 
           stdcall create_app_space,[hdr_mem],[file_base],[file_size]
           mov ecx, -30  ; no memory
           test eax, eax
           jz .failed
 
           mov   ebx,[slot_base]
           mov   [ebx+APPDATA.dir_table],eax
           mov   eax,[hdr_mem]
           mov   [ebx+APPDATA.mem_size],eax
 
if GREEDY_KERNEL
else
           mov ecx, [hdr_mem]
           mov edi, [file_size]
           add edi, 4095
           and edi, not 4095
           sub ecx, edi
           jna @F
 
           xor eax, eax
      ;     add edi, new_app_base
           cld
           rep stosb
@@:
end if
 
; release only virtual space, not phisical memory
 
           stdcall free_kernel_space, [file_base]
           lea eax, [hdr_cmdline]
           lea ebx, [cmdline]
           lea ecx, [filename]
           stdcall set_app_params ,[slot],eax,ebx,ecx,[flags]
 
           mov eax, [save_cr3]
           call set_cr3
 
           xor ebx, ebx
           mov [application_table_status],ebx ;unlock application_table_status mutex
           mov eax,[process_number]  ;set result
           ret
.failed:
           mov eax, [save_cr3]
           call set_cr3
.err:
.err_hdr:
           stdcall kernel_free,[file_base]
.err_file:
           xor eax, eax
           mov [application_table_status],eax
           mov eax, ecx
           ret
endp
 
align 4
test_app_header:
           virtual at eax
             APP_HEADER_00 APP_HEADER_00
           end virtual
           virtual at eax
             APP_HEADER_01 APP_HEADER_01
           end virtual
 
           cmp dword [eax], 'MENU'
           jne .fail
           cmp word [eax+4],'ET'
           jne .fail
 
           cmp [eax+6], word '00'
           jne  .check_01_header
 
           mov  ecx,[APP_HEADER_00.start]
           mov  [ebx+0x08], ecx                ;app_eip
           mov  edx,[APP_HEADER_00.mem_size]
           mov  [ebx+0x10], edx                ;app_mem
           shr  edx,1
           sub  edx,0x10
           mov  [ebx+0x0C], edx                ;app_esp
           mov  ecx,[APP_HEADER_00.i_param]
           mov  [ebx], ecx                     ;app_cmdline
           mov  [ebx+4], dword 0               ;app_path
           mov  edx, [APP_HEADER_00.i_end]
           mov  [ebx+0x14], edx
           ret
 
 .check_01_header:
 
           cmp  [eax+6],word '01'
           jne  .fail
 
           mov  ecx,[APP_HEADER_01.start]
           mov  [ebx+0x08], ecx                ;app_eip
           mov  edx,[APP_HEADER_01.mem_size]
 
; \begin{diamond}[20.08.2006]
; sanity check (functions 19,58 load app_i_end bytes and that must
; fit in allocated memory to prevent kernel faults)
           cmp  edx,[APP_HEADER_01.i_end]
           jb   .fail
; \end{diamond}[20.08.2006]
 
           mov  [ebx+0x10], edx                ;app_mem
           mov  ecx,[APP_HEADER_01.stack_top]
           mov  [ebx+0x0C], ecx                ;app_esp
           mov  edx,[APP_HEADER_01.i_param]
           mov  [ebx], edx                     ;app_cmdline
           mov  ecx,[APP_HEADER_01.i_icon]
           mov  [ebx+4], ecx                   ;app_path
           mov  edx, [APP_HEADER_01.i_end]
           mov  [ebx+0x14], edx
           ret
.fail:
           xor eax, eax
           ret
 
align 4
proc get_new_process_place
;input:
;  none
;result:
;  eax=[new_process_place]<>0 - ok
;      0 - failed.
;This function find least empty slot.
;It doesn't increase [TASK_COUNT]!
           mov    eax,CURRENT_TASK
           mov    ebx,[TASK_COUNT]
           inc    ebx
           shl    ebx,5
           add    ebx,eax               ;ebx - address of process information for (last+1) slot
.newprocessplace:
;eax = address of process information for current slot
           cmp    eax,ebx
           jz     .endnewprocessplace   ;empty slot after high boundary
           add    eax,0x20
           cmp    word [eax+0xa],9      ;check process state, 9 means that process slot is empty
           jnz    .newprocessplace
.endnewprocessplace:
           mov    ebx,eax
           sub    eax,CURRENT_TASK
           shr    eax,5                 ;calculate slot index
           cmp    eax,256
           jge    .failed               ;it should be <256
           mov    word [ebx+0xa],9      ;set process state to 9 (for slot after hight boundary)
           ret
.failed:
           xor    eax,eax
           ret
endp
 
align 4
proc create_app_space stdcall, app_size:dword,img_base:dword,img_size:dword
           locals
             app_pages   dd ?
             img_pages   dd ?
             dir_addr    dd ?
             app_tabs    dd ?
           endl
 
           mov ebx, pg_data.pg_mutex
           call wait_mutex   ;ebx
 
           xor eax, eax
           mov [dir_addr], eax
 
           mov eax, [app_size]
           add eax, 4095
           and eax, NOT(4095)
           mov [app_size], eax
           mov ebx, eax
           shr eax, 12
           mov [app_pages], eax
 
           add ebx, 0x3FFFFF
           and ebx, NOT(0x3FFFFF)
           shr ebx, 22
           mov [app_tabs], ebx
 
           mov ecx, [img_size]
           add ecx, 4095
           and ecx, NOT(4095)
 
           mov [img_size], ecx
           shr ecx, 12
           mov [img_pages], ecx
 
     if GREEDY_KERNEL
           lea eax, [ecx+ebx+2]    ;only image size
     else
           lea eax, [eax+ebx+2]    ;all requested memory
     end if
           cmp eax, [pg_data.pages_free]
           ja .fail
 
           call alloc_page
           test eax, eax
           jz .fail
           mov [dir_addr], eax
           stdcall map_page,[tmp_task_pdir],eax,dword PG_SW
 
           mov edi, [tmp_task_pdir]
           mov ecx, (OS_BASE shr 20)/4
           xor eax, eax
           cld
           rep stosd
 
           mov ecx, (OS_BASE shr 20)/4
           mov esi, sys_pgdir+(OS_BASE shr 20)
           rep movsd
 
           mov eax, [dir_addr]
           or eax, PG_SW
           mov [edi-4096+(page_tabs shr 20)], eax
 
           and eax, -4096
           call set_cr3
 
           mov edx, [app_tabs]
           mov edi, new_app_base
@@:
           call alloc_page
           test eax, eax
           jz .fail
 
           stdcall map_page_table, edi, eax
           add edi, 0x00400000
           dec edx
           jnz @B
 
           mov edi, new_app_base
           shr edi, 10
           add edi, page_tabs
 
           mov ecx, [app_tabs]
           shl ecx, 10
           xor eax, eax
           rep stosd
 
           mov ecx, [img_pages]
           mov ebx, PG_UW
           mov edx, new_app_base
           mov esi, [img_base]
           mov edi, new_app_base
           shr esi, 10
           shr edi, 10
           add esi, page_tabs
           add edi, page_tabs
.remap:
           lodsd
           or eax, ebx      ; force user level r/w access
           stosd
           add edx, 0x1000
           dec [app_pages]
           dec ecx
           jnz .remap
 
           mov ecx, [app_pages]
           test ecx, ecx
           jz .done
 
if GREEDY_KERNEL
           mov eax, 0x02
.reserve:
           stosd
           invlpg [edx]
           add edx, 4096
           dec ecx
           jnz .reserve
else
 
.alloc:
           call alloc_page
           test eax, eax
           jz .fail
 
           stdcall map_page,edx,eax,dword PG_UW
           add edx, 0x1000
           dec [app_pages]
           jnz .alloc
end if
 
.done:
           stdcall map_page,[tmp_task_pdir],dword 0,dword PG_UNMAP
 
           dec [pg_data.pg_mutex]
           mov eax, [dir_addr]
           ret
.fail:
           dec [pg_data.pg_mutex]
           cmp [dir_addr], 0
           je @f
           stdcall destroy_app_space, [dir_addr]
@@:
           xor eax, eax
           ret
endp
 
align 4
set_cr3:
 
           mov ebx, [current_slot]
           mov [ebx+APPDATA.dir_table], eax
           mov cr3, eax
           ret
 
align 4
proc destroy_page_table stdcall, pg_tab:dword
 
           push esi
 
           mov esi, [pg_tab]
           mov ecx, 1024
.free:
           mov eax, [esi]
           test eax, 1
           jz .next
           call free_page
.next:
           add esi, 4
           dec ecx
           jnz .free
           pop esi
           ret
endp
 
align 4
proc destroy_app_space stdcall, pg_dir:dword
 
           mov ebx, pg_data.pg_mutex
           call wait_mutex   ;ebx
 
           xor   edx,edx
           mov   eax,0x2
           mov ebx, [pg_dir]
.loop:
;eax = current slot of process
           mov   ecx,eax
           shl   ecx,5
           cmp   byte [CURRENT_TASK+ecx+0xa],9  ;if process running?
           jz    @f                              ;skip empty slots
           shl   ecx,3
           cmp   [SLOT_BASE+ecx+0xB8],ebx       ;compare page directory addresses
           jnz   @f
           inc   edx                            ;thread found
@@:
           inc   eax
           cmp   eax,[TASK_COUNT]               ;exit loop if we look through all processes
           jle   .loop
 
;edx = number of threads
;our process is zombi so it isn't counted
           cmp   edx,1
           jg    .exit
;if there isn't threads then clear memory.
 
           mov eax, [pg_dir]
           and eax, not 0xFFF
           stdcall map_page,[tmp_task_pdir],eax,dword PG_SW
           mov esi, [tmp_task_pdir]
           mov edi, (OS_BASE shr 20)/4
.destroy:
           mov eax, [esi]
           test eax, 1
           jz .next
           and eax, not 0xFFF
           stdcall map_page,[tmp_task_ptab],eax,dword PG_SW
           stdcall destroy_page_table, [tmp_task_ptab]
           mov eax, [esi]
           call free_page
.next:
           add esi, 4
           dec edi
           jnz .destroy
 
           mov eax, [pg_dir]
           call free_page
.exit:
           stdcall map_page,[tmp_task_ptab],dword 0,dword PG_UNMAP
           stdcall map_page,[tmp_task_pdir],dword 0,dword PG_UNMAP
           dec [pg_data.pg_mutex]
           ret
endp
 
pid_to_slot:
;Input:
;  eax - pid of process
;Output:
;  eax - slot of process or 0 if process don't exists
;Search process by PID.
    push   ebx
    push   ecx
    mov    ebx,[TASK_COUNT]
    shl    ebx,5
    mov    ecx,2*32
 
.loop:
;ecx=offset of current process info entry
;ebx=maximum permitted offset
    cmp    byte [CURRENT_TASK+ecx+0xa],9
    jz     .endloop              ;skip empty slots
    cmp    [CURRENT_TASK+ecx+0x4],eax ;check PID
    jz     .pid_found
.endloop:
    add    ecx,32
    cmp    ecx,ebx
    jle    .loop
 
    pop    ecx
    pop    ebx
    xor    eax,eax
    ret
 
.pid_found:
    shr    ecx,5
    mov    eax,ecx               ;convert offset to index of slot
    pop    ecx
    pop    ebx
    ret
 
check_region:
;input:
;  ebx - start of buffer
;  ecx - size of buffer
;result:
;  eax = 1 region lays in app memory
;  eax = 0 region don't lays in app memory
     mov  eax,[CURRENT_TASK]
     jmp  check_process_region
;-----------------------------------------------------------------------------
check_process_region:
;input:
;  eax - slot
;  ebx - start of buffer
;  ecx - size of buffer
;result:
;  eax = 1 region lays in app memory
;  eax = 0 region don't lays in app memory
 
     test ecx,ecx
     jle  .ok
     shl  eax,5
     cmp  word [CURRENT_TASK+eax+0xa],0
     jnz  .failed
     shl  eax,3
     mov  eax,[SLOT_BASE+eax+0xb8]
     test eax,eax
     jz   .failed
 
     mov  eax,1
     ret
 
 
;    call MEM_Get_Linear_Address
;    push ebx
;    push ecx
;    push edx
;    mov  edx,ebx
;    and  edx,not (4096-1)
;    sub  ebx,edx
;    add  ecx,ebx
;    mov  ebx,edx
;    add  ecx,(4096-1)
;    and  ecx,not (4096-1)
;.loop:
;;eax - linear address of page directory
;;ebx - current page
;;ecx - current size
;    mov  edx,ebx
;    shr  edx,22
;    mov  edx,[eax+4*edx]
;    and  edx,not (4096-1)
;    test edx,edx
;    jz   .failed1
;    push eax
;    mov  eax,edx
;    call MEM_Get_Linear_Address
;    mov  edx,ebx
;    shr  edx,12
;    and  edx,(1024-1)
;    mov  eax,[eax+4*edx]
;    and  eax,not (4096-1)
;    test eax,eax
;    pop  eax
;    jz   .failed1
;    add  ebx,4096
;    sub  ecx,4096
;    jg   .loop
;    pop  edx
;    pop  ecx
;    pop  ebx
.ok:
    mov  eax,1
    ret
;
;.failed1:
;    pop  edx
;    pop  ecx
;    pop  ebx
.failed:
    xor  eax,eax
    ret
 
align 4
proc read_process_memory
;Input:
;  eax - process slot
;  ebx - buffer address
;  ecx - buffer size
;  edx - start address in other process
;Output:
;  eax - number of bytes read.
           locals
             slot       dd ?
             buff       dd ?
             r_count    dd ?
             offset     dd ?
             tmp_r_cnt  dd ?
           endl
 
           mov [slot], eax
           mov [buff], ebx
           mov [r_count], ecx
           mov [tmp_r_cnt], ecx
           mov [offset], edx
 
           pushad
.read_mem:
           mov edx, [offset]
           mov ebx, [tmp_r_cnt]
 
           mov ecx, 0x400000
           and edx, 0x3FFFFF
           sub ecx, edx
           cmp ecx, ebx
           jbe @f
           mov ecx, ebx
@@:
           cmp ecx, 0x8000
           jna @F
           mov ecx, 0x8000
@@:
           mov eax, [slot]
           shl  eax,8
           mov ebx, [offset]
     ;      add ebx, new_app_base
           push ecx
           stdcall map_memEx, [proc_mem_map],\
                              [SLOT_BASE+eax+0xB8],\
                              ebx, ecx
           pop ecx
 
           mov esi, [offset]
           and esi, 0xfff
           add esi, [proc_mem_map]
           mov edi, [buff]
           mov edx, ecx
           rep movsb
 
           add [offset], edx
           sub [tmp_r_cnt], edx
           jnz .read_mem
 
           popad
           mov eax, [r_count]
           ret
endp
 
align 4
proc write_process_memory
;Input:
;  eax - process slot
;  ebx - buffer address
;  ecx - buffer size
;  edx - start address in other process
;Output:
;  eax - number of bytes written
 
           locals
             slot       dd ?
             buff       dd ?
             w_count    dd ?
             offset     dd ?
             tmp_w_cnt  dd ?
           endl
 
           mov [slot], eax
           mov [buff], ebx
           mov [w_count], ecx
           mov [tmp_w_cnt], ecx
           mov [offset], edx
 
           pushad
.read_mem:
           mov edx, [offset]
           mov ebx, [tmp_w_cnt]
 
           mov ecx, 0x400000
           and edx, 0x3FFFFF
           sub ecx, edx
           cmp ecx, ebx
           jbe @f
           mov ecx, ebx
@@:
           cmp ecx, 0x8000
           jna @F
           mov ecx, 0x8000
@@:
           mov eax, [slot]
           shl  eax,8
           mov ebx, [offset]
      ;     add ebx, new_app_base
           push ecx
           stdcall map_memEx, [proc_mem_map],\
                              [SLOT_BASE+eax+0xB8],\
                              ebx, ecx
           pop ecx
 
           mov edi, [offset]
           and edi, 0xfff
           add edi, [proc_mem_map]
           mov esi, [buff]
           mov edx, ecx
           rep movsb
 
           add [offset], edx
           sub [tmp_w_cnt], edx
           jnz .read_mem
 
           popad
           mov eax, [w_count]
           ret
endp
 
align 4
proc new_sys_threads
           locals
             slot          dd ?
             app_cmdline   dd ? ;0x00
             app_path      dd ? ;0x04
             app_eip       dd ? ;0x08
             app_esp       dd ? ;0x0C
             app_mem       dd ? ;0x10
           endl
 
           cmp eax,1
           jne .failed                  ;other subfunctions
 
           xor  eax,eax
           mov [app_cmdline], eax
           mov [app_path], eax
           mov [app_eip], ebx
           mov [app_esp], ecx
 
           mov    esi,new_process_loading
           call   sys_msg_board_str
.wait_lock:
           cmp [application_table_status],0
           je .get_lock
           call   change_task
           jmp .wait_lock
 
.get_lock:
           mov eax, 1
           xchg eax, [application_table_status]
           cmp eax, 0
           jne .wait_lock
 
           call   set_application_table_status
 
           call get_new_process_place
           test eax, eax
           jz .failed
 
           mov [slot], eax
 
           mov    esi,[current_slot]
           mov    ebx,esi             ;ebx=esi - pointer to extended information about current thread
 
           mov    edi, eax
           shl    edi,8
           add    edi,SLOT_BASE
           mov    edx,edi             ;edx=edi - pointer to extended infomation about new thread
           mov    ecx,256/4
           xor eax, eax
           cld
           rep    stosd               ;clean extended information about new thread
           mov    esi,ebx
           mov    edi,edx
           mov    ecx,11
           rep    movsb               ;copy process name
 
           mov eax,[ebx+APPDATA.heap_base]
           mov [edx+APPDATA.heap_base], eax
 
           mov ecx,[ebx+APPDATA.heap_top]
           mov [edx+APPDATA.heap_top], ecx
 
           mov eax,[ebx+APPDATA.mem_size]
           mov [edx+APPDATA.mem_size], eax
 
           mov ecx,[ebx+APPDATA.dir_table]
           mov [edx+APPDATA.dir_table],ecx      ;copy page directory
 
           lea eax, [app_cmdline]
           stdcall set_app_params ,[slot],eax,dword 0,\
                                         dword 0,dword 0
 
           mov    esi,new_process_running
           call   sys_msg_board_str     ;output information about succefull startup
 
           mov    [application_table_status],0 ;unlock application_table_status mutex
           mov    eax,[process_number]  ;set result
           ret
.failed:
           mov    [application_table_status],0
           mov    eax,-1
           ret
endp
 
; param
;  ebx=mutex
 
align 4
wait_mutex:
           push eax
           push ebx
.do_wait:
           cmp dword [ebx],0
           je .get_lock
           call change_task
           jmp .do_wait
.get_lock:
           mov eax, 1
           xchg eax, [ebx]
           test eax, eax
           jnz .do_wait
           pop ebx
           pop eax
           ret
 
align 4
proc set_app_params stdcall,slot:dword, params:dword,\
                        cmd_line:dword, app_path:dword, flags:dword
 
           locals
             pl0_stack dd ?
           endl
 
           stdcall kernel_alloc, RING0_STACK_SIZE+512
           mov [pl0_stack], eax
 
           lea edi, [eax+RING0_STACK_SIZE]
 
           mov eax, [slot]
           mov ebx, eax
 
           shl eax, 8
           mov [eax+SLOT_BASE+APPDATA.fpu_state], edi
           mov [eax+SLOT_BASE+APPDATA.fpu_handler], 0
           mov [eax+SLOT_BASE+APPDATA.sse_handler], 0
 
;set default io permission map
           mov [eax+SLOT_BASE+APPDATA.io_map],\
                  (tss._io_map_0-OS_BASE+PG_MAP)
           mov [eax+SLOT_BASE+APPDATA.io_map+4],\
                  (tss._io_map_1-OS_BASE+PG_MAP)
 
           mov esi, fpu_data
           mov ecx, 512/4
           rep movsd
 
           cmp    ebx,[TASK_COUNT]
           jle    .noinc
           inc    dword [TASK_COUNT]       ;update number of processes
.noinc:
           shl ebx,8
           lea edx, [ebx+SLOT_BASE+APP_EV_OFFSET]
           mov [SLOT_BASE+APPDATA.fd_ev+ebx],edx
           mov [SLOT_BASE+APPDATA.bk_ev+ebx],edx
 
           add edx, APP_OBJ_OFFSET-APP_EV_OFFSET
           mov [SLOT_BASE+APPDATA.fd_obj+ebx],edx
           mov [SLOT_BASE+APPDATA.bk_obj+ebx],edx
 
           mov ecx, [def_cursor]
           mov [SLOT_BASE+APPDATA.cursor+ebx],ecx
           mov eax, [pl0_stack]
           mov [SLOT_BASE+APPDATA.pl0_stack+ebx],eax
 
           shr ebx,3
           mov eax, new_app_base
           mov dword [CURRENT_TASK+ebx+0x10],eax
 
.add_command_line:
           mov edx,[params]
           mov edx,[edx]           ;app_cmdline
           test edx,edx
           jz @F                   ;application don't need parameters
 
           mov     eax, edx
           add     eax, 256
           jc      @f
 
           cmp     eax, [SLOT_BASE+APPDATA.mem_size+ebx*8]
           ja      @f
 
           stdcall k_strncpy, edx, [cmd_line], 256
@@:
           mov edx,[params]
           mov edx, [edx+4]        ;app_path
           test edx,edx
           jz @F                   ;application don't need path of file
           mov     eax, edx
           add     eax, 1024
           jc      @f
           cmp     eax, [SLOT_BASE+APPDATA.mem_size+ebx*8]
           ja      @f
           stdcall k_strncpy, edx, [app_path], 1024
@@:
           mov    ebx,[slot]
           mov    eax,ebx
           shl    ebx,5
           lea    ecx,[draw_data+ebx]  ;ecx - pointer to draw data
 
; set window state to 'normal' (non-minimized/maximized/rolled-up) state
           mov     [ebx+window_data+WDATA.fl_wstate], WSTATE_NORMAL
           mov     [ebx+window_data+WDATA.fl_redraw], 1
           add    ebx,CURRENT_TASK            ;ebx - pointer to information about process
           mov    [ebx+TASKDATA.wnd_number],al;set window number on screen = process slot
 
           mov    [ebx+TASKDATA.event_mask],dword 1+2+4 ;set default event flags (see 40 function)
 
           inc    dword [process_number]
           mov    eax,[process_number]
           mov    [ebx+4],eax           ;set PID
 
;set draw data to full screen
 
           mov    [ecx+0],dword 0
           mov    [ecx+4],dword 0
           mov    eax,[ScreenWidth]
           mov    [ecx+8],eax
           mov    eax,[ScreenHeight]
           mov    [ecx+12],eax
 
           mov ebx, [pl0_stack]
           mov esi,[params]
           lea ecx, [ebx+REG_EIP]
           xor eax, eax
 
           mov [ebx+REG_RET], dword irq0.return
           mov [ebx+REG_EDI], eax
           mov [ebx+REG_ESI], eax
           mov [ebx+REG_EBP], eax
           mov [ebx+REG_ESP], ecx   ;ebx+REG_EIP
           mov [ebx+REG_EBX], eax
           mov [ebx+REG_EDX], eax
           mov [ebx+REG_ECX], eax
           mov [ebx+REG_EAX], eax
 
           mov [ebx+REG_EFL_2], dword 0x1002
 
           mov eax, [esi+0x08]       ;app_eip
           mov [ebx+REG_EIP],  eax   ;app_entry
           mov [ebx+REG_CS], dword app_code
           mov [ebx+REG_EFLAGS], dword 0x3202
 
           mov eax, [esi+0x0C]       ;app_esp
           mov [ebx+REG_APP_ESP], eax    ;app_stack
           mov [ebx+REG_SS], dword app_data
 
           lea ecx, [ebx+REG_RET]
           mov ebx, [slot]
           shl ebx, 8
           mov [ebx+SLOT_BASE+APPDATA.saved_esp], ecx
 
;flush keyboard and buttons queue
           mov    [KEY_COUNT],byte 0
           mov    [BTN_COUNT],byte 0
 
           mov    edi,[slot]
           shl    edi,5
           add    edi,window_data
           mov    ebx,[slot]
           movzx  esi,word [WIN_STACK+ebx*2]
           lea    esi,[WIN_POS+esi*2]
           call   windowactivate        ;gui initialization
 
           mov    ebx,[slot]
           shl    ebx,5
           mov    [CURRENT_TASK+ebx+0xa],byte 0 ;set process state - running
; set if debuggee
           mov eax, [flags]
           test byte [flags], 1
           jz   .no_debug
           mov  [CURRENT_TASK+ebx+0xa],byte 1 ;set process state - suspended
           mov  eax,[CURRENT_TASK]
           mov  [SLOT_BASE+ebx*8+0xac],eax ;set debugger PID - current
.no_debug:
           mov    esi,new_process_running
           call   sys_msg_board_str     ;output information about succefull startup
           ret
endp
 
include "debug.inc"
>

Rev 438	Rev 444
1	GREEDY_KERNEL equ 0	1	$Revision: 437 $
-		2	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-		3	;; ;;
-		4	;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;;
-		5	;; Distributed under terms of the GNU General Public License ;;
-		6	;; ;;
-		7	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-		8
2		9	GREEDY_KERNEL equ 0
3		10
4	struc APP_HEADER_00	11	struc APP_HEADER_00
5	{ .banner dq ?	12	{ .banner dq ?
6	.version dd ? ;+8	13	.version dd ? ;+8
7	.start dd ? ;+12	14	.start dd ? ;+12
8	.i_end dd ? ;+16	15	.i_end dd ? ;+16
9	.mem_size dd ? ;+20	16	.mem_size dd ? ;+20
10	.i_param dd ? ;+24	17	.i_param dd ? ;+24
11	}	18	}
12		19
13	struc APP_HEADER_01	20	struc APP_HEADER_01
14	{ .banner dq ?	21	{ .banner dq ?
15	.version dd ? ;+8	22	.version dd ? ;+8
16	.start dd ? ;+12	23	.start dd ? ;+12
17	.i_end dd ? ;+16	24	.i_end dd ? ;+16
18	.mem_size dd ? ;+20	25	.mem_size dd ? ;+20
19	.stack_top dd ? ;+24	26	.stack_top dd ? ;+24
20	.i_param dd ? ;+28	27	.i_param dd ? ;+28
21	.i_icon dd ? ;+32	28	.i_icon dd ? ;+32
22	}	29	}
23		30
24		31
25	struc APP_PARAMS	32	struc APP_PARAMS
26	{ .app_cmdline ;0x00	33	{ .app_cmdline ;0x00
27	.app_path ;0x04	34	.app_path ;0x04
28	.app_eip ;0x08	35	.app_eip ;0x08
29	.app_esp ;0x0C	36	.app_esp ;0x0C
30	.app_mem ;0x10	37	.app_mem ;0x10
31	}	38	}
32		39
33	macro _clear_ op	40	macro _clear_ op
34	{ mov ecx, op/4	41	{ mov ecx, op/4
35	xor eax, eax	42	xor eax, eax
36	cld	43	cld
37	rep stosd	44	rep stosd
38	}	45	}
39		46
40		47
41	align 4	48	align 4
42	proc fs_execute	49	proc fs_execute
43		50
44	;fn_read:dword, file_size:dword, cluster:dword	51	;fn_read:dword, file_size:dword, cluster:dword
45		52
46	; ebx - cmdline	53	; ebx - cmdline
47	; edx - flags	54	; edx - flags
48	; ebp - full filename	55	; ebp - full filename
49	; [esp+4] = procedure DoRead, [esp+8] = filesize & [esp+12]... - arguments for it	56	; [esp+4] = procedure DoRead, [esp+8] = filesize & [esp+12]... - arguments for it
50		57
51	locals	58	locals
52	cmdline rd 64 ;256/4	59	cmdline rd 64 ;256/4
53	filename rd 256 ;1024/4	60	filename rd 256 ;1024/4
54	flags dd ?	61	flags dd ?
55		62
56	save_cr3 dd ?	63	save_cr3 dd ?
57	slot dd ?	64	slot dd ?
58	slot_base dd ?	65	slot_base dd ?
59	file_base dd ?	66	file_base dd ?
60	file_size dd ?	67	file_size dd ?
61	;app header data	68	;app header data
62	hdr_cmdline dd ? ;0x00	69	hdr_cmdline dd ? ;0x00
63	hdr_path dd ? ;0x04	70	hdr_path dd ? ;0x04
64	hdr_eip dd ? ;0x08	71	hdr_eip dd ? ;0x08
65	hdr_esp dd ? ;0x0C	72	hdr_esp dd ? ;0x0C
66	hdr_mem dd ? ;0x10	73	hdr_mem dd ? ;0x10
67	hdr_i_end dd ? ;0x14	74	hdr_i_end dd ? ;0x14
68	endl	75	endl
69		76
70	pushad	77	pushad
71		78
72	mov [cmdline], ebx	79	mov [cmdline], ebx
73	mov [flags], edx	80	mov [flags], edx
74		81
75	; [ebp] pointer to filename	82	; [ebp] pointer to filename
76		83
77	lea eax, [filename]	84	lea eax, [filename]
78	mov dword [eax+1020],0 ;force terminate	85	mov dword [eax+1020],0 ;force terminate
79	;string	86	;string
80	stdcall k_strncpy, eax, [ebp], 1023	87	stdcall k_strncpy, eax, [ebp], 1023
81		88
82	lea eax, [cmdline]	89	lea eax, [cmdline]
83	mov dword [eax+252], 0	90	mov dword [eax+252], 0
84	stdcall k_strncpy, eax, [cmdline], 255	91	stdcall k_strncpy, eax, [cmdline], 255
85		92
86	lea eax, [filename]	93	lea eax, [filename]
87	stdcall load_file, eax	94	stdcall load_file, eax
88	mov ecx, -ERROR_FILE_NOT_FOUND	95	mov ecx, -ERROR_FILE_NOT_FOUND
89	test eax, eax	96	test eax, eax
90	jz .err_file	97	jz .err_file
91		98
92	mov [file_base], eax	99	mov [file_base], eax
93	mov [file_size], ebx	100	mov [file_size], ebx
94		101
95	lea ebx, [hdr_cmdline]	102	lea ebx, [hdr_cmdline]
96	call test_app_header	103	call test_app_header
97	mov ecx, -0x1F	104	mov ecx, -0x1F
98	test eax, eax	105	test eax, eax
99	jz .err_hdr	106	jz .err_hdr
100		107
101	mov esi, new_process_loading	108	mov esi, new_process_loading
102	call sys_msg_board_str ; write message to message board	109	call sys_msg_board_str ; write message to message board
103		110
104	.wait_lock:	111	.wait_lock:
105	cmp [application_table_status],0	112	cmp [application_table_status],0
106	je .get_lock	113	je .get_lock
107	call change_task	114	call change_task
108	jmp .wait_lock	115	jmp .wait_lock
109		116
110	.get_lock:	117	.get_lock:
111	mov eax, 1	118	mov eax, 1
112	xchg eax, [application_table_status]	119	xchg eax, [application_table_status]
113	cmp eax, 0	120	cmp eax, 0
114	jne .wait_lock	121	jne .wait_lock
115		122
116	call set_application_table_status	123	call set_application_table_status
117		124
118	call get_new_process_place	125	call get_new_process_place
119	test eax, eax	126	test eax, eax
120	mov ecx, -0x20 ; too many processes	127	mov ecx, -0x20 ; too many processes
121	jz .err	128	jz .err
122		129
123	mov [slot], eax	130	mov [slot], eax
124	shl eax, 8	131	shl eax, 8
125	add eax, SLOT_BASE	132	add eax, SLOT_BASE
126	mov [slot_base], eax	133	mov [slot_base], eax
127	mov edi, eax	134	mov edi, eax
128	_clear_ 256 ;clean extended information about process	135	_clear_ 256 ;clean extended information about process
129		136
130	; write application name	137	; write application name
131	lea edi, [filename]	138	lea edi, [filename]
132	mov al, '/'	139	mov al, '/'
133	call k_strrchr ; now eax points to name without path	140	call k_strrchr ; now eax points to name without path
134		141
135	lea esi, [eax+1]	142	lea esi, [eax+1]
136	test eax, eax	143	test eax, eax
137	jnz @F	144	jnz @F
138	lea esi, [filename]	145	lea esi, [filename]
139	@@:	146	@@:
140	mov ecx, 8 ; 8 chars for name	147	mov ecx, 8 ; 8 chars for name
141	mov edi, [slot_base]	148	mov edi, [slot_base]
142	.copy_process_name_loop:	149	.copy_process_name_loop:
143	lodsb	150	lodsb
144	cmp al, '.'	151	cmp al, '.'
145	jz .copy_process_name_done	152	jz .copy_process_name_done
146	test al, al	153	test al, al
147	jz .copy_process_name_done	154	jz .copy_process_name_done
148	stosb	155	stosb
149	loop .copy_process_name_loop	156	loop .copy_process_name_loop
150	.copy_process_name_done:	157	.copy_process_name_done:
151		158
152	mov ebx, cr3	159	mov ebx, cr3
153	mov [save_cr3], ebx	160	mov [save_cr3], ebx
154		161
155	stdcall create_app_space,[hdr_mem],[file_base],[file_size]	162	stdcall create_app_space,[hdr_mem],[file_base],[file_size]
156	mov ecx, -30 ; no memory	163	mov ecx, -30 ; no memory
157	test eax, eax	164	test eax, eax
158	jz .failed	165	jz .failed
159		166
160	mov ebx,[slot_base]	167	mov ebx,[slot_base]
161	mov [ebx+APPDATA.dir_table],eax	168	mov [ebx+APPDATA.dir_table],eax
162	mov eax,[hdr_mem]	169	mov eax,[hdr_mem]
163	mov [ebx+APPDATA.mem_size],eax	170	mov [ebx+APPDATA.mem_size],eax
164		171
165	if GREEDY_KERNEL	172	if GREEDY_KERNEL
166	else	173	else
167	mov ecx, [hdr_mem]	174	mov ecx, [hdr_mem]
168	mov edi, [file_size]	175	mov edi, [file_size]
169	add edi, 4095	176	add edi, 4095
170	and edi, not 4095	177	and edi, not 4095
171	sub ecx, edi	178	sub ecx, edi
172	jna @F	179	jna @F
173		180
174	xor eax, eax	181	xor eax, eax
175	; add edi, new_app_base	182	; add edi, new_app_base
176	cld	183	cld
177	rep stosb	184	rep stosb
178	@@:	185	@@:
179	end if	186	end if
180		187
181	; release only virtual space, not phisical memory	188	; release only virtual space, not phisical memory
182		189
183	stdcall free_kernel_space, [file_base]	190	stdcall free_kernel_space, [file_base]
184	lea eax, [hdr_cmdline]	191	lea eax, [hdr_cmdline]
185	lea ebx, [cmdline]	192	lea ebx, [cmdline]
186	lea ecx, [filename]	193	lea ecx, [filename]
187	stdcall set_app_params ,[slot],eax,ebx,ecx,[flags]	194	stdcall set_app_params ,[slot],eax,ebx,ecx,[flags]
188		195
189	mov eax, [save_cr3]	196	mov eax, [save_cr3]
190	call set_cr3	197	call set_cr3
191		198
192	xor ebx, ebx	199	xor ebx, ebx
193	mov [application_table_status],ebx ;unlock application_table_status mutex	200	mov [application_table_status],ebx ;unlock application_table_status mutex
194	mov eax,[process_number] ;set result	201	mov eax,[process_number] ;set result
195	ret	202	ret
196	.failed:	203	.failed:
197	mov eax, [save_cr3]	204	mov eax, [save_cr3]
198	call set_cr3	205	call set_cr3
199	.err:	206	.err:
200	.err_hdr:	207	.err_hdr:
201	stdcall kernel_free,[file_base]	208	stdcall kernel_free,[file_base]
202	.err_file:	209	.err_file:
203	xor eax, eax	210	xor eax, eax
204	mov [application_table_status],eax	211	mov [application_table_status],eax
205	mov eax, ecx	212	mov eax, ecx
206	ret	213	ret
207	endp	214	endp
208		215
209	align 4	216	align 4
210	test_app_header:	217	test_app_header:
211	virtual at eax	218	virtual at eax
212	APP_HEADER_00 APP_HEADER_00	219	APP_HEADER_00 APP_HEADER_00
213	end virtual	220	end virtual
214	virtual at eax	221	virtual at eax
215	APP_HEADER_01 APP_HEADER_01	222	APP_HEADER_01 APP_HEADER_01
216	end virtual	223	end virtual
217		224
218	cmp dword [eax], 'MENU'	225	cmp dword [eax], 'MENU'
219	jne .fail	226	jne .fail
220	cmp word [eax+4],'ET'	227	cmp word [eax+4],'ET'
221	jne .fail	228	jne .fail
222		229
223	cmp [eax+6], word '00'	230	cmp [eax+6], word '00'
224	jne .check_01_header	231	jne .check_01_header
225		232
226	mov ecx,[APP_HEADER_00.start]	233	mov ecx,[APP_HEADER_00.start]
227	mov [ebx+0x08], ecx ;app_eip	234	mov [ebx+0x08], ecx ;app_eip
228	mov edx,[APP_HEADER_00.mem_size]	235	mov edx,[APP_HEADER_00.mem_size]
229	mov [ebx+0x10], edx ;app_mem	236	mov [ebx+0x10], edx ;app_mem
230	shr edx,1	237	shr edx,1
231	sub edx,0x10	238	sub edx,0x10
232	mov [ebx+0x0C], edx ;app_esp	239	mov [ebx+0x0C], edx ;app_esp
233	mov ecx,[APP_HEADER_00.i_param]	240	mov ecx,[APP_HEADER_00.i_param]
234	mov [ebx], ecx ;app_cmdline	241	mov [ebx], ecx ;app_cmdline
235	mov [ebx+4], dword 0 ;app_path	242	mov [ebx+4], dword 0 ;app_path
236	mov edx, [APP_HEADER_00.i_end]	243	mov edx, [APP_HEADER_00.i_end]
237	mov [ebx+0x14], edx	244	mov [ebx+0x14], edx
238	ret	245	ret
239		246
240	.check_01_header:	247	.check_01_header:
241		248
242	cmp [eax+6],word '01'	249	cmp [eax+6],word '01'
243	jne .fail	250	jne .fail
244		251
245	mov ecx,[APP_HEADER_01.start]	252	mov ecx,[APP_HEADER_01.start]
246	mov [ebx+0x08], ecx ;app_eip	253	mov [ebx+0x08], ecx ;app_eip
247	mov edx,[APP_HEADER_01.mem_size]	254	mov edx,[APP_HEADER_01.mem_size]
248		255
249	; \begin{diamond}[20.08.2006]	256	; \begin{diamond}[20.08.2006]
250	; sanity check (functions 19,58 load app_i_end bytes and that must	257	; sanity check (functions 19,58 load app_i_end bytes and that must
251	; fit in allocated memory to prevent kernel faults)	258	; fit in allocated memory to prevent kernel faults)
252	cmp edx,[APP_HEADER_01.i_end]	259	cmp edx,[APP_HEADER_01.i_end]
253	jb .fail	260	jb .fail
254	; \end{diamond}[20.08.2006]	261	; \end{diamond}[20.08.2006]
255		262
256	mov [ebx+0x10], edx ;app_mem	263	mov [ebx+0x10], edx ;app_mem
257	mov ecx,[APP_HEADER_01.stack_top]	264	mov ecx,[APP_HEADER_01.stack_top]
258	mov [ebx+0x0C], ecx ;app_esp	265	mov [ebx+0x0C], ecx ;app_esp
259	mov edx,[APP_HEADER_01.i_param]	266	mov edx,[APP_HEADER_01.i_param]
260	mov [ebx], edx ;app_cmdline	267	mov [ebx], edx ;app_cmdline
261	mov ecx,[APP_HEADER_01.i_icon]	268	mov ecx,[APP_HEADER_01.i_icon]
262	mov [ebx+4], ecx ;app_path	269	mov [ebx+4], ecx ;app_path
263	mov edx, [APP_HEADER_01.i_end]	270	mov edx, [APP_HEADER_01.i_end]
264	mov [ebx+0x14], edx	271	mov [ebx+0x14], edx
265	ret	272	ret
266	.fail:	273	.fail:
267	xor eax, eax	274	xor eax, eax
268	ret	275	ret
269		276
270	align 4	277	align 4
271	proc get_new_process_place	278	proc get_new_process_place
272	;input:	279	;input:
273	; none	280	; none
274	;result:	281	;result:
275	; eax=[new_process_place]<>0 - ok	282	; eax=[new_process_place]<>0 - ok
276	; 0 - failed.	283	; 0 - failed.
277	;This function find least empty slot.	284	;This function find least empty slot.
278	;It doesn't increase [TASK_COUNT]!	285	;It doesn't increase [TASK_COUNT]!
279	mov eax,CURRENT_TASK	286	mov eax,CURRENT_TASK
280	mov ebx,[TASK_COUNT]	287	mov ebx,[TASK_COUNT]
281	inc ebx	288	inc ebx
282	shl ebx,5	289	shl ebx,5
283	add ebx,eax ;ebx - address of process information for (last+1) slot	290	add ebx,eax ;ebx - address of process information for (last+1) slot
284	.newprocessplace:	291	.newprocessplace:
285	;eax = address of process information for current slot	292	;eax = address of process information for current slot
286	cmp eax,ebx	293	cmp eax,ebx
287	jz .endnewprocessplace ;empty slot after high boundary	294	jz .endnewprocessplace ;empty slot after high boundary
288	add eax,0x20	295	add eax,0x20
289	cmp word [eax+0xa],9 ;check process state, 9 means that process slot is empty	296	cmp word [eax+0xa],9 ;check process state, 9 means that process slot is empty
290	jnz .newprocessplace	297	jnz .newprocessplace
291	.endnewprocessplace:	298	.endnewprocessplace:
292	mov ebx,eax	299	mov ebx,eax
293	sub eax,CURRENT_TASK	300	sub eax,CURRENT_TASK
294	shr eax,5 ;calculate slot index	301	shr eax,5 ;calculate slot index
295	cmp eax,256	302	cmp eax,256
296	jge .failed ;it should be <256	303	jge .failed ;it should be <256
297	mov word [ebx+0xa],9 ;set process state to 9 (for slot after hight boundary)	304	mov word [ebx+0xa],9 ;set process state to 9 (for slot after hight boundary)
298	ret	305	ret
299	.failed:	306	.failed:
300	xor eax,eax	307	xor eax,eax
301	ret	308	ret
302	endp	309	endp
303		310
304	align 4	311	align 4
305	proc create_app_space stdcall, app_size:dword,img_base:dword,img_size:dword	312	proc create_app_space stdcall, app_size:dword,img_base:dword,img_size:dword
306	locals	313	locals
307	app_pages dd ?	314	app_pages dd ?
308	img_pages dd ?	315	img_pages dd ?
309	dir_addr dd ?	316	dir_addr dd ?
310	app_tabs dd ?	317	app_tabs dd ?
311	endl	318	endl
312		319
313	mov ebx, pg_data.pg_mutex	320	mov ebx, pg_data.pg_mutex
314	call wait_mutex ;ebx	321	call wait_mutex ;ebx
315		322
316	xor eax, eax	323	xor eax, eax
317	mov [dir_addr], eax	324	mov [dir_addr], eax
318		325
319	mov eax, [app_size]	326	mov eax, [app_size]
320	add eax, 4095	327	add eax, 4095
321	and eax, NOT(4095)	328	and eax, NOT(4095)
322	mov [app_size], eax	329	mov [app_size], eax
323	mov ebx, eax	330	mov ebx, eax
324	shr eax, 12	331	shr eax, 12
325	mov [app_pages], eax	332	mov [app_pages], eax
326		333
327	add ebx, 0x3FFFFF	334	add ebx, 0x3FFFFF
328	and ebx, NOT(0x3FFFFF)	335	and ebx, NOT(0x3FFFFF)
329	shr ebx, 22	336	shr ebx, 22
330	mov [app_tabs], ebx	337	mov [app_tabs], ebx
331		338
332	mov ecx, [img_size]	339	mov ecx, [img_size]
333	add ecx, 4095	340	add ecx, 4095
334	and ecx, NOT(4095)	341	and ecx, NOT(4095)
335		342
336	mov [img_size], ecx	343	mov [img_size], ecx
337	shr ecx, 12	344	shr ecx, 12
338	mov [img_pages], ecx	345	mov [img_pages], ecx
339		346
340	if GREEDY_KERNEL	347	if GREEDY_KERNEL
341	lea eax, [ecx+ebx+2] ;only image size	348	lea eax, [ecx+ebx+2] ;only image size
342	else	349	else
343	lea eax, [eax+ebx+2] ;all requested memory	350	lea eax, [eax+ebx+2] ;all requested memory
344	end if	351	end if
345	cmp eax, [pg_data.pages_free]	352	cmp eax, [pg_data.pages_free]
346	ja .fail	353	ja .fail
347		354
348	call alloc_page	355	call alloc_page
349	test eax, eax	356	test eax, eax
350	jz .fail	357	jz .fail
351	mov [dir_addr], eax	358	mov [dir_addr], eax
352	stdcall map_page,[tmp_task_pdir],eax,dword PG_SW	359	stdcall map_page,[tmp_task_pdir],eax,dword PG_SW
353		360
354	mov edi, [tmp_task_pdir]	361	mov edi, [tmp_task_pdir]
355	mov ecx, (OS_BASE shr 20)/4	362	mov ecx, (OS_BASE shr 20)/4
356	xor eax, eax	363	xor eax, eax
357	cld	364	cld
358	rep stosd	365	rep stosd
359		366
360	mov ecx, (OS_BASE shr 20)/4	367	mov ecx, (OS_BASE shr 20)/4
361	mov esi, sys_pgdir+(OS_BASE shr 20)	368	mov esi, sys_pgdir+(OS_BASE shr 20)
362	rep movsd	369	rep movsd
363		370
364	mov eax, [dir_addr]	371	mov eax, [dir_addr]
365	or eax, PG_SW	372	or eax, PG_SW
366	mov [edi-4096+(page_tabs shr 20)], eax	373	mov [edi-4096+(page_tabs shr 20)], eax
367		374
368	and eax, -4096	375	and eax, -4096
369	call set_cr3	376	call set_cr3
370		377
371	mov edx, [app_tabs]	378	mov edx, [app_tabs]
372	mov edi, new_app_base	379	mov edi, new_app_base
373	@@:	380	@@:
374	call alloc_page	381	call alloc_page
375	test eax, eax	382	test eax, eax
376	jz .fail	383	jz .fail
377		384
378	stdcall map_page_table, edi, eax	385	stdcall map_page_table, edi, eax
379	add edi, 0x00400000	386	add edi, 0x00400000
380	dec edx	387	dec edx
381	jnz @B	388	jnz @B
382		389
383	mov edi, new_app_base	390	mov edi, new_app_base
384	shr edi, 10	391	shr edi, 10
385	add edi, page_tabs	392	add edi, page_tabs
386		393
387	mov ecx, [app_tabs]	394	mov ecx, [app_tabs]
388	shl ecx, 10	395	shl ecx, 10
389	xor eax, eax	396	xor eax, eax
390	rep stosd	397	rep stosd
391		398
392	mov ecx, [img_pages]	399	mov ecx, [img_pages]
393	mov ebx, PG_UW	400	mov ebx, PG_UW
394	mov edx, new_app_base	401	mov edx, new_app_base
395	mov esi, [img_base]	402	mov esi, [img_base]
396	mov edi, new_app_base	403	mov edi, new_app_base
397	shr esi, 10	404	shr esi, 10
398	shr edi, 10	405	shr edi, 10
399	add esi, page_tabs	406	add esi, page_tabs
400	add edi, page_tabs	407	add edi, page_tabs
401	.remap:	408	.remap:
402	lodsd	409	lodsd
403	or eax, ebx ; force user level r/w access	410	or eax, ebx ; force user level r/w access
404	stosd	411	stosd
405	add edx, 0x1000	412	add edx, 0x1000
406	dec [app_pages]	413	dec [app_pages]
407	dec ecx	414	dec ecx
408	jnz .remap	415	jnz .remap
409		416
410	mov ecx, [app_pages]	417	mov ecx, [app_pages]
411	test ecx, ecx	418	test ecx, ecx
412	jz .done	419	jz .done
413		420
414	if GREEDY_KERNEL	421	if GREEDY_KERNEL
415	mov eax, 0x02	422	mov eax, 0x02
416	.reserve:	423	.reserve:
417	stosd	424	stosd
418	invlpg [edx]	425	invlpg [edx]
419	add edx, 4096	426	add edx, 4096
420	dec ecx	427	dec ecx
421	jnz .reserve	428	jnz .reserve
422	else	429	else
423		430
424	.alloc:	431	.alloc:
425	call alloc_page	432	call alloc_page
426	test eax, eax	433	test eax, eax
427	jz .fail	434	jz .fail
428		435
429	stdcall map_page,edx,eax,dword PG_UW	436	stdcall map_page,edx,eax,dword PG_UW
430	add edx, 0x1000	437	add edx, 0x1000
431	dec [app_pages]	438	dec [app_pages]
432	jnz .alloc	439	jnz .alloc
433	end if	440	end if
434		441
435	.done:	442	.done:
436	stdcall map_page,[tmp_task_pdir],dword 0,dword PG_UNMAP	443	stdcall map_page,[tmp_task_pdir],dword 0,dword PG_UNMAP
437		444
438	dec [pg_data.pg_mutex]	445	dec [pg_data.pg_mutex]
439	mov eax, [dir_addr]	446	mov eax, [dir_addr]
440	ret	447	ret
441	.fail:	448	.fail:
442	dec [pg_data.pg_mutex]	449	dec [pg_data.pg_mutex]
443	cmp [dir_addr], 0	450	cmp [dir_addr], 0
444	je @f	451	je @f
445	stdcall destroy_app_space, [dir_addr]	452	stdcall destroy_app_space, [dir_addr]
446	@@:	453	@@:
447	xor eax, eax	454	xor eax, eax
448	ret	455	ret
449	endp	456	endp
450		457
451	align 4	458	align 4
452	set_cr3:	459	set_cr3:
453		460
454	mov ebx, [current_slot]	461	mov ebx, [current_slot]
455	mov [ebx+APPDATA.dir_table], eax	462	mov [ebx+APPDATA.dir_table], eax
456	mov cr3, eax	463	mov cr3, eax
457	ret	464	ret
458		465
459	align 4	466	align 4
460	proc destroy_page_table stdcall, pg_tab:dword	467	proc destroy_page_table stdcall, pg_tab:dword
461		468
462	push esi	469	push esi
463		470
464	mov esi, [pg_tab]	471	mov esi, [pg_tab]
465	mov ecx, 1024	472	mov ecx, 1024
466	.free:	473	.free:
467	mov eax, [esi]	474	mov eax, [esi]
468	test eax, 1	475	test eax, 1
469	jz .next	476	jz .next
470	call free_page	477	call free_page
471	.next:	478	.next:
472	add esi, 4	479	add esi, 4
473	dec ecx	480	dec ecx
474	jnz .free	481	jnz .free
475	pop esi	482	pop esi
476	ret	483	ret
477	endp	484	endp
478		485
479	align 4	486	align 4
480	proc destroy_app_space stdcall, pg_dir:dword	487	proc destroy_app_space stdcall, pg_dir:dword
481		488
482	mov ebx, pg_data.pg_mutex	489	mov ebx, pg_data.pg_mutex
483	call wait_mutex ;ebx	490	call wait_mutex ;ebx
484		491
485	xor edx,edx	492	xor edx,edx
486	mov eax,0x2	493	mov eax,0x2
487	mov ebx, [pg_dir]	494	mov ebx, [pg_dir]
488	.loop:	495	.loop:
489	;eax = current slot of process	496	;eax = current slot of process
490	mov ecx,eax	497	mov ecx,eax
491	shl ecx,5	498	shl ecx,5
492	cmp byte [CURRENT_TASK+ecx+0xa],9 ;if process running?	499	cmp byte [CURRENT_TASK+ecx+0xa],9 ;if process running?
493	jz @f ;skip empty slots	500	jz @f ;skip empty slots

Subversion Repositories Kolibri OS

(root)/kernel/branches/flat_kernel/core/taskman.inc – Rev 438 → 444