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


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

Subversion Repositories Kolibri OS

(root)/kernel/trunk/core/taskman.inc – Rev 437 → 465