Subversion Repositories Kolibri OS

Rev

Rev 9709 | Rev 9715 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  1. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  2. ;;                                                              ;;
  3. ;; Copyright (C) KolibriOS team 2004-2021. All rights reserved. ;;
  4. ;;  Distributed under terms of the GNU General Public License.  ;;
  5. ;;                                                              ;;
  6. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  7.  
  8. $Revision: 9710 $
  9.  
  10.  
  11. GREEDY_KERNEL  = 0
  12.  
  13. struct  APP_HEADER_00_
  14.         banner          dq ?
  15.         version         dd ?    ;+8
  16.         start           dd ?    ;+12
  17.         i_end           dd ?    ;+16
  18.         mem_size        dd ?    ;+20
  19.         i_param         dd ?    ;+24
  20. ends
  21.  
  22. struct  APP_HEADER_01_
  23.         banner          dq ?
  24.         version         dd ?    ;+8
  25.         start           dd ?    ;+12
  26.         i_end           dd ?    ;+16
  27.         mem_size        dd ?    ;+20
  28.         stack_top       dd ?    ;+24
  29.         i_param         dd ?    ;+28
  30.         i_icon          dd ?    ;+32
  31. ends
  32.  
  33. struct  APP_HDR
  34.         cmdline         rd 1    ;0x00
  35.         path            rd 1    ;0x04
  36.         eip             rd 1    ;0x08
  37.         esp             rd 1    ;0x0C
  38.         _edata          rd 1    ;0x10
  39.         _emem           rd 1    ;0x14
  40.         img_base        rd 1    ;0x18
  41.         img_size        rd 1
  42.         filename_size   rd 1
  43.         cmdline_size    rd 1
  44.         path_string     rd 1
  45. ends
  46.  
  47. macro _clear_ op
  48. {  mov ecx, op/4
  49.         xor     eax, eax
  50.         cld
  51.         rep stosd
  52. }
  53.  
  54. align 4
  55. _strnlen:
  56.         mov     edx, ecx
  57.         xor     eax, eax
  58.         repne scasb
  59.         jne     @F
  60.         inc     ecx
  61. @@:
  62.         mov     eax, edx
  63.         sub     eax, ecx
  64.         retn
  65.  
  66. fs_execute_from_sysdir:
  67.         xor     ebx, ebx
  68. fs_execute_from_sysdir_param:
  69.         stdcall kernel_alloc, maxPathLength
  70.         push    eax ebx
  71.         mov     esi, ebp
  72.         mov     edi, eax
  73.         xor     eax, eax
  74.         call    getFullPath
  75.         pop     ecx ebx
  76.         xor     edx, edx
  77. ; @brief Executes a program
  78. ; @param edx Flags
  79. ; @param ecx Commandline
  80. ; @param ebx Absolute file path
  81. ; @param eax String length
  82. ; @returns Negated error code or new process number
  83. proc fs_execute
  84.     locals
  85.         cmdline         rd  1
  86.         flags           rd  1
  87.         slot            rd  1  ; index of new thread slot
  88.         slot_base       rd  1  ; base address of it
  89. ; app header data
  90.         hdr_cmdline     rd  1
  91.         hdr_path        rd  1
  92.         hdr_eip         rd  1
  93.         hdr_esp         rd  1
  94.         hdr_edata       rd  1
  95.         hdr_emem        rd  1
  96.         file_base       rd  1
  97.         file_size       rd  1
  98.         filename_size   rd  1
  99.         cmdline_size    rd  1
  100.         path_string     rd  1
  101.     endl
  102.  
  103.         and     edx, 1  ;clear flags user. TODO: add api for start apps with user flags
  104.         mov     [flags], edx
  105.         mov     [cmdline], ecx
  106.         mov     [path_string], ebx
  107.         mov     [filename_size], eax
  108.         mov     esi, -ERROR_FILE_NOT_FOUND
  109.         test    eax, eax
  110.         jz      .err_file
  111.         stdcall load_file, ebx
  112.         test    eax, eax
  113.         jz      .err_file
  114.  
  115.         mov     [file_base], eax
  116.         mov     [file_size], ebx
  117.         lea     ebx, [hdr_cmdline]
  118.         call    test_app_header  ; fill our app header data locals with values from header of given program (if its correct)
  119.         mov     esi, -TASKMAN_ERROR_NOT_A_EXECUTABLE
  120.         test    eax, eax
  121.         jz      .err_hdr
  122.  
  123.         call    lock_application_table
  124.         call    alloc_thread_slot   ; create a slot for new thread
  125.         mov     esi, -TASKMAN_ERROR_TOO_MANY_PROCESSES
  126.         test    eax, eax
  127.         jz      .err_0
  128.  
  129.         mov     [slot], eax
  130.         shl     eax, 8
  131.         lea     edi, [SLOT_BASE+eax]
  132.         mov     [slot_base], edi
  133. ; clean extended information about process
  134.         mov     ecx, sizeof.APPDATA/4
  135.         xor     eax, eax
  136.         cld
  137.         rep stosd
  138. ; write application name ( APPDATA.appname )
  139.         stdcall strrchr, [path_string], '/'
  140.         lea     esi, [eax+1]    ; -> name without path
  141.         mov     ecx, 11
  142.         mov     edi, [slot_base]
  143. @@:
  144.         call    utf8to16
  145.         call    uni2ansi_char
  146.         cmp     al, '.'
  147.         jz      @f
  148.         test    al, al
  149.         jz      @f
  150.         stosb
  151.         loop    @b
  152. @@:
  153.         mov     edi, [cmdline]
  154.         xor     eax, eax
  155.         test    edi, edi
  156.         jz      @f
  157.         mov     ecx, 65535
  158.         call    _strnlen
  159.         cmp     eax, 256
  160.         jb      @f
  161. ; if cmdline length >= 256 then increase needed memory size by this length
  162.         lea     ebx, [eax+1]
  163.         add     [hdr_emem], ebx
  164. @@:
  165.         mov     [cmdline_size], eax
  166.         stdcall create_process, [hdr_emem]  ; create a new process
  167.         mov     esi, -TASKMAN_ERROR_OUT_OF_MEMORY
  168.         test    eax, eax
  169.         jz      .err_hdr
  170.  
  171. ; add new process to the list
  172.         mov     ebx, [sys_proc+LHEAD.prev]
  173.         __list_add eax, ebx, sys_proc
  174. ; fill the structure fields:
  175.         mov     ebx, [hdr_emem]
  176.         mov     [eax+PROC.mem_used], ebx
  177.  
  178. ; write that main thread of app belongs to new process
  179.         mov     ebx, [slot_base]
  180.         mov     [ebx+APPDATA.process], eax
  181.  
  182. ; initialize the thread list of process: at this moment it consists only of one main thread
  183.         lea     edx, [ebx+APPDATA.list]
  184.         lea     ecx, [eax+PROC.thr_list]
  185.         list_add_tail edx, ecx
  186.  
  187. ; allocate space and copy app header data locals and cmdline string there, put pointer to exec_params of new thread
  188.         mov     eax, [cmdline_size]
  189.         add     eax, sizeof.APP_HDR
  190.         stdcall kernel_alloc, eax
  191.         mov     [ebx+APPDATA.exec_params], eax
  192.         mov     edi, eax
  193.         lea     esi, [hdr_cmdline]
  194.         mov     ecx, sizeof.APP_HDR/4
  195.         rep movsd
  196.         mov     ecx, [cmdline_size]
  197.         mov     esi, [cmdline]
  198.         rep movsb
  199. ; set other parameters of application
  200.         lea     eax, [hdr_cmdline]
  201.         stdcall set_app_params , [slot], eax, [flags]
  202.         mov     eax, [process_number]   ; return process number
  203.         call    unlock_application_table
  204.         ret
  205.  
  206. .err_0:
  207.         call    unlock_application_table
  208. .err_hdr:
  209.         stdcall kernel_free, [file_base]
  210. .err_file:
  211.         stdcall kernel_free, [path_string]
  212.         mov     eax, esi
  213.         ret
  214. endp
  215.  
  216. align 4
  217. test_app_header:
  218.        virtual at eax
  219.          APP_HEADER_00 APP_HEADER_00_
  220.        end virtual
  221.        virtual at eax
  222.          APP_HEADER_01 APP_HEADER_01_
  223.        end virtual
  224.  
  225.         cmp     dword [eax], 'MENU'
  226.         jne     .fail
  227.         cmp     word [eax+4], 'ET'
  228.         jne     .fail
  229.  
  230.         cmp     [eax+6], word '00'
  231.         jne     .check_01_header
  232.  
  233.         mov     ecx, [APP_HEADER_00.start]
  234.         mov     [ebx+APP_HDR.eip], ecx
  235.         mov     edx, [APP_HEADER_00.mem_size]
  236.         mov     [ebx+APP_HDR._emem], edx
  237.         shr     edx, 1
  238.         sub     edx, 0x10
  239.         mov     [ebx+APP_HDR.esp], edx
  240.         mov     ecx, [APP_HEADER_00.i_param]
  241.         mov     [ebx+APP_HDR.cmdline], ecx
  242.         mov     [ebx+APP_HDR.path], 0
  243.         mov     edx, [APP_HEADER_00.i_end]
  244.         mov     [ebx+APP_HDR._edata], edx
  245.         ret
  246.  
  247.  .check_01_header:
  248.  
  249.         cmp     [eax+6], word '01'
  250.         je      @f
  251.         cmp     [eax+6], word '02'
  252.         jne     .fail
  253. @@:
  254.         mov     ecx, [APP_HEADER_01.start]
  255.         mov     [ebx+0x08], ecx
  256.         mov     edx, [APP_HEADER_01.mem_size]
  257.  
  258. ; \begin{diamond}[20.08.2006]
  259. ; sanity check (functions 19,58 load app_i_end bytes and that must
  260. ; fit in allocated memory to prevent kernel faults)
  261.         cmp     edx, [APP_HEADER_01.i_end]
  262.         jb      .fail
  263. ; \end{diamond}[20.08.2006]
  264.  
  265.         mov     [ebx+APP_HDR._emem], edx
  266.         mov     ecx, [APP_HEADER_01.stack_top]
  267.         mov     [ebx+APP_HDR.esp], ecx
  268.         mov     edx, [APP_HEADER_01.i_param]
  269.         mov     [ebx+APP_HDR.cmdline], edx
  270.         mov     ecx, [APP_HEADER_01.i_icon]
  271.         mov     [ebx+APP_HDR.path], ecx
  272.         mov     edx, [APP_HEADER_01.i_end]
  273.         mov     [ebx+APP_HDR._edata], edx
  274.         ret
  275. .fail:
  276.         xor     eax, eax
  277.         ret
  278.  
  279. align 4
  280. alloc_thread_slot:
  281. ;input:
  282. ;  none
  283. ;result:
  284. ;  eax=[new_thread_slot]<>0 - ok
  285. ;      0 - failed.
  286. ;This function find least empty slot.
  287. ;It doesn't increase [thread_count]!
  288.  
  289.  
  290.         mov     edx, thr_slot_map
  291.         pushfd
  292.         cli
  293. .l1:
  294.         bsf     eax, [edx]
  295.         jnz     .found
  296.         add     edx, 4
  297.         cmp     edx, thr_slot_map+32
  298.         jb      .l1
  299.  
  300.         popfd
  301.         xor     eax, eax
  302.         ret
  303. .found:
  304.         btr     [edx], eax
  305.         sub     edx, thr_slot_map
  306.         lea     eax, [eax+edx*8]
  307.         popfd
  308.         ret
  309.  
  310. align 4
  311. proc create_process stdcall, app_size:dword
  312.        locals
  313.          process     dd ?
  314.          app_tabs    dd ?
  315.        endl
  316.  
  317.         push    ebx
  318.         push    esi
  319.         push    edi
  320.  
  321.         xor     eax, eax
  322.         mov     [process], eax
  323.  
  324.         mov     eax, [app_size]
  325.         add     eax, 0x3FFFFF
  326.         shr     eax, 22
  327.         mov     [app_tabs], eax
  328.  
  329.         stdcall kernel_alloc, 0x2000
  330.         test    eax, eax
  331.         jz      .fail
  332.         mov     [process], eax
  333.  
  334.         lea     edi, [eax+PROC.heap_lock]
  335.         mov     ecx, (PROC.ht_free-PROC.heap_lock)/4
  336.  
  337.         list_init eax
  338.         add     eax, PROC.thr_list
  339.         list_init eax
  340.  
  341.         xor     eax, eax
  342.         cld
  343.         rep stosd
  344.  
  345.         mov     [edi], dword (PROC.pdt_0 - PROC.htab)/4 - 3
  346.         mov     [edi+4], dword 3           ;reserve handles for stdin stdout and stderr
  347.         mov     ecx, (PROC.pdt_0 - PROC.htab)/4
  348.         add     edi, 8
  349.         inc     eax
  350. @@:
  351.         stosd
  352.         inc     eax
  353.         cmp     eax, ecx
  354.         jbe     @B
  355.  
  356.         mov     eax, edi
  357.         call    get_pg_addr
  358.         mov     [edi-4096+PROC.pdt_0_phys], eax
  359.  
  360.         mov     ecx, (OS_BASE shr 20)/4
  361.         xor     eax, eax
  362.         rep stosd
  363.  
  364.         mov     ecx, (OS_BASE shr 20)/4
  365.         mov     esi, sys_proc+PROC.pdt_0+(OS_BASE shr 20)
  366.         rep movsd
  367.  
  368.         mov     eax, [edi-8192+PROC.pdt_0_phys]
  369.         or      eax, PG_SWR
  370.         mov     [edi-4096+(page_tabs shr 20)], eax
  371.  
  372.         lea     edx, [edi-4096]
  373.         mov     esi, [app_tabs]
  374.  
  375. .alloc_page_dir:
  376.         call    alloc_page
  377.         test    eax, eax
  378.         jz      .fail
  379.         or      eax, PG_UWR
  380.         mov     [edx], eax
  381.  
  382.         mov     edi, [tmp_task_ptab]
  383.         stdcall map_page, edi, eax, PG_SWR
  384.         mov     ecx, 1024
  385.         xor     eax, eax
  386.         rep stosd
  387.  
  388.         add     edx, 4
  389.         dec     esi
  390.         jnz     .alloc_page_dir
  391.  
  392.         stdcall map_page, [tmp_task_ptab], 0, PG_UNMAP
  393.         mov     eax, [process]
  394.  
  395.         pop     edi
  396.         pop     esi
  397.         pop     ebx
  398.         ret
  399. .fail:
  400.         mov     ecx, [process]
  401.         jcxz    @F
  402.  
  403.         call    destroy_process
  404. @@:
  405.         xor     eax, eax
  406.         pop     edi
  407.         pop     esi
  408.         pop     ebx
  409.         ret
  410. endp
  411.  
  412. align 4
  413. proc destroy_page_table stdcall, pg_tab:dword
  414.  
  415.         push    esi
  416.  
  417.         mov     esi, [pg_tab]
  418.         mov     ecx, 1024
  419. .free:
  420.         mov     eax, [esi]
  421.         test    eax, 1
  422.         jz      .next
  423.         test    eax, 2
  424.         jz      .next
  425.         test    eax, 1 shl 9
  426.         jnz     .next                     ;skip shared pages
  427.         call    free_page
  428. .next:
  429.         add     esi, 4
  430.         dec     ecx
  431.         jnz     .free
  432.         pop     esi
  433.         ret
  434. endp
  435.  
  436. align 4
  437. destroy_process: ;fastcall ecx= ptr to process
  438.  
  439.         lea     eax, [ecx+PROC.thr_list]
  440.         cmp     eax, [eax+LHEAD.next]
  441.         jne     .exit
  442.  
  443. align 4
  444. .internal:
  445.         push    ecx
  446.  
  447.         mov     esi, ecx
  448.         list_del esi
  449.  
  450.         mov     esi, [esi+PROC.dlls_list_ptr]
  451.         call    destroy_all_hdlls
  452.  
  453.         mov     esi, [esp]
  454.         add     esi, PROC.pdt_0
  455.         mov     edi, (0x80000000 shr 20)/4
  456. .destroy:
  457.         mov     eax, [esi]
  458.         test    eax, 1
  459.         jz      .next
  460.         and     eax, not 0xFFF
  461.         stdcall map_page, [tmp_task_ptab], eax, PG_SWR
  462.         stdcall destroy_page_table, [tmp_task_ptab]
  463.         mov     eax, [esi]
  464.         call    free_page
  465. .next:
  466.         add     esi, 4
  467.         dec     edi
  468.         jnz     .destroy
  469.  
  470.         call    kernel_free     ;ecx still in stack
  471.         stdcall map_page, [tmp_task_ptab], 0, PG_UNMAP
  472. .exit:
  473.         ret
  474.  
  475. align 4
  476. get_pid:
  477.         mov     eax, [current_slot]
  478.         mov     eax, [eax+APPDATA.tid]
  479.         ret
  480.  
  481. pid_to_slot:
  482. ;Input:
  483. ;  eax - pid of process
  484. ;Output:
  485. ;  eax - slot of process or 0 if process don't exists
  486. ;Search process by PID.
  487.         push    ebx
  488.         push    ecx
  489.         mov     ebx, [thread_count]
  490.         shl     ebx, BSF sizeof.APPDATA ; multiply by size
  491.         ; add 2*32 cause:
  492.         ; [TASK_TABLE; TASK_TABLE + 32) isnt a task actually
  493.         ; skip first process in the task table
  494.         ;mov     ecx, 2*32    ;sizeof.TASKDATA
  495.         mov     ecx, sizeof.APPDATA
  496.  
  497. .loop:
  498. ;ecx = offset of current process info entry
  499. ;ebx = maximum permitted offset
  500.         cmp     [SLOT_BASE + ecx + APPDATA.state], TSTATE_FREE
  501.         jz      .endloop ;skip empty slots
  502.         cmp     [ecx + SLOT_BASE + APPDATA.tid], eax;check PID
  503.         jz      .pid_found
  504. .endloop:
  505.         add     ecx, sizeof.APPDATA
  506.         cmp     ecx, ebx
  507.         jle     .loop
  508.  
  509.         pop     ecx
  510.         pop     ebx
  511.         xor     eax, eax
  512.         ret
  513.  
  514. .pid_found:
  515.         shr     ecx, BSF sizeof.APPDATA ; divide by size
  516.         mov     eax, ecx ;convert offset to index of slot
  517.         pop     ecx
  518.         pop     ebx
  519.         ret
  520.  
  521.  
  522. align 4
  523. proc read_process_memory
  524. ;Input:
  525. ;  eax - process slot
  526. ;  ecx - buffer address
  527. ;  edx - buffer size
  528. ;  esi - start address in other process
  529. ;Output:
  530. ;  eax - number of bytes read.
  531.        locals
  532.          slot   dd ?
  533.          buff   dd ?
  534.          r_count    dd ?
  535.          offset dd ?
  536.          tmp_r_cnt  dd ?
  537.        endl
  538.  
  539.         mov     [slot], eax
  540.         mov     [buff], ecx
  541.         and     [r_count], 0
  542.         mov     [tmp_r_cnt], edx
  543.         mov     [offset], esi
  544.  
  545.         pushad
  546. .read_mem:
  547.         mov     edx, [offset]
  548.         mov     ebx, [tmp_r_cnt]
  549.  
  550.         mov     ecx, 0x400000
  551.         and     edx, 0x3FFFFF
  552.         sub     ecx, edx
  553.         cmp     ecx, ebx
  554.         jbe     @f
  555.         mov     ecx, ebx
  556. @@:
  557.         cmp     ecx, 0x8000
  558.         jna     @F
  559.         mov     ecx, 0x8000
  560. @@:
  561.         mov     ebx, [offset]
  562.  
  563.         push    ecx
  564.         stdcall map_memEx, [proc_mem_map], \
  565.                 [slot], ebx, ecx, PG_READ
  566.         pop     ecx
  567.  
  568.         mov     esi, [offset]
  569.         and     esi, 0xfff
  570.         sub     eax, esi
  571.         jbe     .ret
  572.         cmp     ecx, eax
  573.         jbe     @f
  574.         mov     ecx, eax
  575.         mov     [tmp_r_cnt], eax
  576. @@:
  577.         add     esi, [proc_mem_map]
  578.         mov     edi, [buff]
  579.         mov     edx, ecx
  580.         rep movsb
  581.         add     [r_count], edx
  582.  
  583.         add     [offset], edx
  584.         sub     [tmp_r_cnt], edx
  585.         jnz     .read_mem
  586. .ret:
  587.         popad
  588.         mov     eax, [r_count]
  589.         ret
  590. endp
  591.  
  592. align 4
  593. proc write_process_memory
  594. ;Input:
  595. ;  eax - process slot
  596. ;  ecx - buffer address
  597. ;  edx - buffer size
  598. ;  esi - start address in other process
  599. ;Output:
  600. ;  eax - number of bytes written
  601.  
  602.        locals
  603.          slot   dd ?
  604.          buff   dd ?
  605.          w_count    dd ?
  606.          offset dd ?
  607.          tmp_w_cnt  dd ?
  608.        endl
  609.  
  610.         mov     [slot], eax
  611.         mov     [buff], ecx
  612.         and     [w_count], 0
  613.         mov     [tmp_w_cnt], edx
  614.         mov     [offset], esi
  615.  
  616.         pushad
  617. .read_mem:
  618.         mov     edx, [offset]
  619.         mov     ebx, [tmp_w_cnt]
  620.  
  621.         mov     ecx, 0x400000
  622.         and     edx, 0x3FFFFF
  623.         sub     ecx, edx
  624.         cmp     ecx, ebx
  625.         jbe     @f
  626.         mov     ecx, ebx
  627. @@:
  628.         cmp     ecx, 0x8000
  629.         jna     @F
  630.         mov     ecx, 0x8000
  631. @@:
  632.         mov     ebx, [offset]
  633.         push    ecx
  634.         stdcall map_memEx, [proc_mem_map], \
  635.                 [slot], ebx, ecx, PG_SWR
  636.         pop     ecx
  637.  
  638.         mov     edi, [offset]
  639.         and     edi, 0xfff
  640.         sub     eax, edi
  641.         jbe     .ret
  642.         cmp     ecx, eax
  643.         jbe     @f
  644.         mov     ecx, eax
  645.         mov     [tmp_w_cnt], eax
  646. @@:
  647.         add     edi, [proc_mem_map]
  648.         mov     esi, [buff]
  649.         mov     edx, ecx
  650.         rep movsb
  651.  
  652.         add     [w_count], edx
  653.         add     [offset], edx
  654.         sub     [tmp_w_cnt], edx
  655.         jnz     .read_mem
  656. .ret:
  657.         popad
  658.         mov     eax, [w_count]
  659.         ret
  660. endp
  661.  
  662. ;ebx = 1 - kernel thread
  663. ;ecx=thread entry point
  664. ;edx=thread stack pointer
  665. ;creation flags  0x01 - debugged
  666. ;                0x02 - kernel
  667.  
  668. align 4
  669. proc new_sys_threads
  670.        locals
  671.          slot          dd ?
  672.          flags         dd ?
  673.          app_cmdline   dd ? ;0x00
  674.          app_path      dd ? ;0x04
  675.          app_eip       dd ? ;0x08
  676.          app_esp       dd ? ;0x0C
  677.          app_mem       dd ? ;0x10
  678.        endl
  679.  
  680.         shl     ebx, 1
  681.         mov     [flags], ebx
  682.  
  683.         xor     eax, eax
  684.         mov     [app_eip], ecx
  685.         mov     [app_cmdline], eax
  686.         mov     [app_esp], edx
  687.         mov     [app_path], eax
  688.  
  689.         call    lock_application_table
  690.  
  691.         call    alloc_thread_slot
  692.         test    eax, eax
  693.         jz      .failed
  694.  
  695.         mov     [slot], eax
  696.  
  697.         mov     esi, [current_slot]
  698.         mov     ebx, esi      ;ebx=esi - pointer to extended information about current thread
  699.  
  700.         mov     edi, eax
  701.         shl     edi, 8
  702.         add     edi, SLOT_BASE
  703.         mov     edx, edi      ;edx=edi - pointer to extended infomation about new thread
  704.         mov     ecx, sizeof.APPDATA/4
  705.         xor     eax, eax
  706.         cld
  707.         rep stosd             ;clean extended information about new thread
  708.         mov     esi, ebx
  709.         mov     edi, edx
  710.         mov     ecx, 11
  711.         rep movsb             ;copy process name
  712.  
  713.  
  714.         mov     eax, [ebx+APPDATA.tls_base]
  715.         test    eax, eax
  716.         jz      @F
  717.  
  718.         push    edx
  719.         stdcall user_alloc, 4096
  720.         pop     edx
  721.         test    eax, eax
  722.         jz      .failed1;eax=0
  723. @@:
  724.         mov     [edx+APPDATA.tls_base], eax
  725.  
  726.         mov     eax, [ebx+APPDATA.process]
  727.         mov     [edx+APPDATA.process], eax
  728.  
  729.         lea     ebx, [edx+APPDATA.list]
  730.         lea     ecx, [eax+PROC.thr_list]
  731.         list_add_tail ebx, ecx               ;add thread to process child's list
  732.  
  733.         lea     eax, [app_cmdline]
  734.         stdcall set_app_params , [slot], eax, [flags]
  735.  
  736.         mov     eax, [process_number]           ;set result
  737.         call    unlock_application_table
  738.         ret
  739. .failed:
  740.         xor     eax, eax
  741. .failed1:
  742.         call    unlock_application_table
  743.         dec     eax     ;-1
  744.         ret
  745. endp
  746.  
  747. proc map_process_image stdcall, img_size:dword, file_base:dword, file_size:dword
  748.  
  749.         mov     edx, [img_size]
  750.         mov     esi, [file_base]
  751.         mov     ecx, [file_size]
  752.         add     edx, 4095
  753.         add     ecx, 4095
  754.         shr     edx, 12        ; total pages
  755.         shr     ecx, 12        ; image pages
  756.  
  757.         mov     edi, page_tabs
  758.         shr     esi, 10
  759.         add     esi, edi
  760.  
  761. .map_image:
  762.         lodsd
  763.         and     eax, -4096
  764.         or      eax, PG_UWR
  765.         stosd
  766.         dec     edx
  767.         loop    .map_image
  768.  
  769.         test    edx, edx
  770.         jz      .done
  771. .map_bss:
  772.         call    alloc_page
  773.         test    eax, eax
  774.         jz      .fail
  775.  
  776.         or      eax, PG_UWR
  777.         stosd
  778.         dec     edx
  779.         jnz     .map_bss
  780.  
  781.         mov     edi, [file_size]
  782.         mov     ecx, [img_size]
  783.         add     edi, 4095
  784.         and     edi, -4096
  785.         add     ecx, 4095
  786.         and     ecx, -4096
  787.         sub     ecx, edi
  788.         shr     ecx, 2
  789.         xor     eax, eax
  790.         rep stosd
  791. .done:
  792. .fail:
  793.         ret
  794. endp
  795.  
  796. align 4
  797. common_app_entry:
  798.         mov     ebp, [current_slot]
  799.         mov     ebp, [ebp+APPDATA.exec_params]
  800.         test    ebp, ebp
  801.         jz      .exit
  802. ; APPDATA.exec_params have first thread only,
  803. ; so second and next threads don't get here (they jump to .exit)
  804.         stdcall map_process_image, [ebp+APP_HDR._emem],\
  805.                 [ebp+APP_HDR.img_base], [ebp+APP_HDR.img_size]
  806.         mov     esi, [ebp+APP_HDR.path_string]
  807.         mov     edi, [ebp+APP_HDR.path]
  808.         mov     ecx, [ebp+APP_HDR.filename_size]
  809.         cmp     ecx, 1023
  810.         jc      @f
  811.         mov     ecx, 1022
  812. @@:
  813.         push    esi
  814.         test    edi, edi
  815.         jz      @f
  816.         stdcall is_region_userspace, edi, [ebp+APP_HDR.filename_size]
  817.         jnz     @f
  818.         mov     al, '/'
  819.         stosb
  820.         rep movsb
  821.         mov     byte [edi], 0
  822. @@:
  823.         call    kernel_free
  824.         mov     edi, [ebp+APP_HDR.cmdline]
  825.         test    edi, edi
  826.         jz      .check_tls_header
  827.         lea     esi, [ebp+sizeof.APP_HDR]
  828.         mov     ecx, [ebp+APP_HDR.cmdline_size]
  829.         cmp     ecx, 256
  830.         jb      .copy_cmdline
  831.         mov     edi, [ebp+APP_HDR._emem]
  832.         add     edi, 4095
  833.         and     edi, -4096
  834.         sub     edi, ecx
  835.         dec     edi
  836.         cmp     word [6], '00'
  837.         jne     @f
  838.         mov     [APP_HEADER_00_.i_param], edi
  839.         jmp     .copy_cmdline
  840. @@:
  841.         mov     [APP_HEADER_01_.i_param], edi
  842. .copy_cmdline:
  843.         inc     ecx  ; keep in mind about 0 in the end
  844.         stdcall is_region_userspace, edi, ecx
  845.         jnz     .check_tls_header
  846.         dec     ecx
  847.         rep movsb
  848.         mov     byte [edi], 0
  849. .check_tls_header:
  850.         cmp     word [6], '02'
  851.         jne     .try_load_dll ;.cleanup
  852.         call    init_heap
  853.         stdcall user_alloc, 4096
  854.         mov     edx, [current_slot]
  855.         mov     [edx+APPDATA.tls_base], eax
  856.         mov     [tls_data_l+2], ax
  857.         shr     eax, 16
  858.         mov     [tls_data_l+4], al
  859.         mov     [tls_data_l+7], ah
  860.         mov     dx, app_tls
  861.         mov     fs, dx      
  862. ; { Patch by Coldy, For DLL autoload    
  863. .try_load_dll:        
  864. ; Test app header version
  865.         mov     ecx, dword[ebp+APP_HDR.img_base]
  866.         cmp     dword[ecx+8], 2
  867.         jne     .cleanup
  868. ;if APP_HEADER.version = 2 => load lib/dll.obj & change eip to APP_STARTUP_THUNK
  869.         DEBUGF 1, 'K : App header version 2\n'
  870.         stdcall load_library, dll_lib_path, 0
  871.         cmp     eax, 0
  872.         jne     @f
  873. ; Something went wrong (TODO: Next 2 line is code copy after .cleanup)  
  874.         stdcall free_kernel_space, [ebp+APP_HDR.img_base]
  875.         stdcall kernel_free, ebp
  876.         DEBUGF 1, 'K : DLL.OBJ not found! Terminate application!\n'
  877.         mov     ebx, dll_error_msg
  878.         mov     ebp, notifyapp
  879.         call    fs_execute_from_sysdir_param
  880. ; Terminate process (TODO: Need jump to .cleanup after sys_end ?)
  881.         call    sys_end
  882.        
  883. @@:      
  884. ; Find APP_STARTUP_THUNK in DLL.OBJ
  885.         sub     eax, 4
  886.         mov     eax, [eax]
  887.        
  888. ;.change_eip:
  889.         mov     ecx, [current_slot]
  890.         mov     ecx, [ecx+APPDATA.pl0_stack]
  891.         mov     [ecx+REG_EIP], eax
  892.        
  893. ; } End patch by Coldy, For DLL autoload
  894. .cleanup:
  895.         stdcall free_kernel_space, [ebp+APP_HDR.img_base]
  896.         stdcall kernel_free, ebp
  897.         mov     ebx, [current_slot]
  898.         cmp     [ebx+APPDATA.debugger_slot], 0
  899.         je      .exit
  900.         mov     [ebx+APPDATA.state], TSTATE_RUN_SUSPENDED
  901.         mov     [ebx + APPDATA.state], TSTATE_RUN_SUSPENDED
  902.         call    change_task
  903. .exit:
  904.         popad
  905.         iretd
  906.  
  907. EFL_IF      = 0x0200
  908. EFL_IOPL1   = 0x1000
  909. EFL_IOPL2   = 0x2000
  910. EFL_IOPL3   = 0x3000
  911.  
  912. align 4
  913. proc set_app_params stdcall,slot:dword, params:dword, flags:dword
  914.  
  915.        locals
  916.          pl0_stack dd ?
  917.        endl
  918.  
  919.         mov     eax, [xsave_area_size]
  920.         add     eax, RING0_STACK_SIZE
  921.         stdcall kernel_alloc, eax
  922.         mov     [pl0_stack], eax
  923.  
  924.         lea     edi, [eax+RING0_STACK_SIZE]
  925.  
  926.         mov     eax, [slot]
  927.         mov     ebx, eax
  928.  
  929.         shl     eax, BSF sizeof.APPDATA
  930.         mov     [eax+SLOT_BASE+APPDATA.fpu_state], edi
  931.         mov     [eax+SLOT_BASE+APPDATA.exc_handler], 0
  932.         mov     [eax+SLOT_BASE+APPDATA.except_mask], 0
  933.         mov     [eax+SLOT_BASE+APPDATA.terminate_protection], 80000001h
  934.  
  935. ;set default io permission map
  936.         mov     ecx, [SLOT_BASE+sizeof.APPDATA+APPDATA.io_map]
  937.         mov     [eax+SLOT_BASE+APPDATA.io_map], ecx
  938.         mov     ecx, [SLOT_BASE+sizeof.APPDATA+APPDATA.io_map+4]
  939.         mov     [eax+SLOT_BASE+APPDATA.io_map+4], ecx
  940.  
  941.         mov     esi, fpu_data
  942.         mov     ecx, [xsave_area_size]
  943.         add     ecx, 3
  944.         shr     ecx, 2
  945.         rep movsd
  946.  
  947.         cmp     [thread_count], ebx
  948.         adc     [thread_count], 0   ; update number of processes
  949.         shl     ebx, BSF sizeof.APPDATA
  950.         lea     edx, [ebx+SLOT_BASE+APP_EV_OFFSET]
  951.         mov     [SLOT_BASE+APPDATA.fd_ev+ebx], edx
  952.         mov     [SLOT_BASE+APPDATA.bk_ev+ebx], edx
  953.  
  954.         add     edx, APP_OBJ_OFFSET-APP_EV_OFFSET
  955.         mov     [SLOT_BASE+APPDATA.fd_obj+ebx], edx
  956.         mov     [SLOT_BASE+APPDATA.bk_obj+ebx], edx
  957.  
  958.         mov     ecx, [def_cursor]
  959.         mov     [SLOT_BASE+APPDATA.cursor+ebx], ecx
  960.         mov     eax, [pl0_stack]
  961.         mov     [SLOT_BASE+APPDATA.pl0_stack+ebx], eax
  962.         add     eax, RING0_STACK_SIZE
  963.         mov     [SLOT_BASE+APPDATA.saved_esp0+ebx], eax
  964.  
  965.         push    ebx
  966.         stdcall kernel_alloc, maxPathLength
  967.         pop     ebx
  968.         mov     esi, [current_slot]
  969.         mov     esi, [esi+APPDATA.cur_dir]
  970.         mov     ecx, maxPathLength/4
  971.         mov     edi, eax
  972.         mov     [ebx+SLOT_BASE+APPDATA.cur_dir], eax
  973.         rep movsd
  974.  
  975.         mov     [ebx+SLOT_BASE+APPDATA.event_mask], dword 1+2+4;set default event flags (see 40 function)
  976.         inc     dword [process_number]
  977.         mov     eax, [process_number]
  978.         mov     [ebx+SLOT_BASE+APPDATA.tid], eax    ;set TID
  979.  
  980.         mov     eax, [slot]
  981.         mov     [ebx+SLOT_BASE+APPDATA.wnd_number], al
  982.         mov     ebx, eax
  983.         shl     ebx, BSF sizeof.WDATA
  984.         lea     ecx, [draw_data+ebx];ecx - pointer to draw data
  985.  
  986. ; set window state to 'normal' (non-minimized/maximized/rolled-up) state
  987.         mov     [ebx+window_data+WDATA.fl_wstate], WSTATE_NORMAL
  988.         mov     [ebx+window_data+WDATA.fl_redraw], 1
  989.  
  990. ;set draw data to full screen
  991.         xor     eax, eax
  992.         mov     [ecx+0], dword eax
  993.         mov     [ecx+4], dword eax
  994.         mov     eax, [screen_workarea.right]
  995.         mov     [ecx+8], eax
  996.         mov     eax, [screen_workarea.bottom]
  997.         mov     [ecx+12], eax
  998.  
  999.         mov     ebx, [pl0_stack]
  1000.         mov     esi, [params]
  1001.         lea     ecx, [ebx+REG_EIP]
  1002.         xor     eax, eax
  1003.  
  1004.         mov     [ebx+REG_RET], dword common_app_entry
  1005.         mov     [ebx+REG_EDI], eax
  1006.         mov     [ebx+REG_ESI], eax
  1007.         mov     [ebx+REG_EBP], eax
  1008.         mov     [ebx+REG_ESP], ecx;ebx+REG_EIP
  1009.         mov     [ebx+REG_EBX], eax
  1010.         mov     [ebx+REG_EDX], eax
  1011.         mov     [ebx+REG_ECX], eax
  1012.         mov     [ebx+REG_EAX], eax
  1013.  
  1014.         mov     eax, [esi+APP_HDR.eip]
  1015.         mov     [ebx+REG_EIP], eax
  1016.         mov     [ebx+REG_CS], dword app_code
  1017.         mov     ecx, USER_PRIORITY
  1018.  
  1019.         test    byte [flags], 2
  1020.         jz      @F
  1021.  
  1022.         mov     [ebx+REG_CS], dword os_code ; kernel thread
  1023.         mov     ecx, MAX_PRIORITY
  1024. @@:
  1025.         mov     [ebx+REG_EFLAGS], dword EFL_IOPL1+EFL_IF
  1026.  
  1027.         mov     eax, [esi+APP_HDR.esp]
  1028.         mov     [ebx+REG_APP_ESP], eax
  1029.         mov     [ebx+REG_SS], dword app_data
  1030.  
  1031.         lea     edx, [ebx+REG_RET]
  1032.         mov     ebx, [slot]
  1033.         shl     ebx, BSF sizeof.APPDATA
  1034.         mov     [ebx+SLOT_BASE+APPDATA.saved_esp], edx
  1035.  
  1036.         xor     edx, edx; process state - running
  1037. ; set if debuggee
  1038.         test    byte [flags], 1
  1039.         jz      .no_debug
  1040.         mov     eax, [current_slot_idx]
  1041.         mov     [SLOT_BASE+ebx+APPDATA.debugger_slot], eax
  1042. .no_debug:
  1043.         mov     [SLOT_BASE + ebx + APPDATA.state], dl
  1044.         lea     edx, [SLOT_BASE+ebx]
  1045.         call    scheduler_add_thread
  1046.         ret
  1047. endp
  1048.  
  1049. align 4
  1050. get_stack_base:
  1051.         mov     eax, [current_slot]
  1052.         mov     eax, [eax+APPDATA.pl0_stack]
  1053.         ret
  1054.  
  1055.  
  1056. include "debug.inc"
  1057.