forked from KolibriOS/kolibrios
PE loader: error checking
git-svn-id: svn://kolibrios.org@740 a494cfbc-eb01-0410-851d-a64ba20cac60
This commit is contained in:
Sergey Semyonov (Serge)
parent
ff590c086c
commit
2c7785b277
@ -183,6 +183,7 @@ endp
|
||||
|
||||
align 4
|
||||
proc pci_read32 stdcall, bus:dword, devfn:dword, reg:dword
|
||||
push ebx
|
||||
xor eax, eax
|
||||
xor ebx, ebx
|
||||
mov ah, byte [bus]
|
||||
@ -190,6 +191,7 @@ proc pci_read32 stdcall, bus:dword, devfn:dword, reg:dword
|
||||
mov bh, byte [devfn]
|
||||
mov bl, byte [reg]
|
||||
call pci_read_reg
|
||||
pop ebx
|
||||
ret
|
||||
endp
|
||||
|
||||
@ -349,26 +351,32 @@ proc get_service stdcall, sz_name:dword
|
||||
endp
|
||||
|
||||
align 4
|
||||
reg_service:
|
||||
.sz_name equ esp+4
|
||||
.handler equ esp+8
|
||||
mov eax, [.sz_name]
|
||||
test eax, eax
|
||||
jz .fail
|
||||
proc reg_service stdcall, name:dword, handler:dword
|
||||
|
||||
mov ebx, [.handler]
|
||||
test ebx, ebx
|
||||
jz .fail
|
||||
xor eax, eax
|
||||
|
||||
cmp [name], eax
|
||||
je .fail
|
||||
|
||||
cmp [handler], eax
|
||||
je .fail
|
||||
|
||||
push ebx
|
||||
mov eax, SRV_SIZE
|
||||
call malloc ;call alloc_service
|
||||
pop ebx
|
||||
test eax, eax
|
||||
jz .fail
|
||||
|
||||
push ebx
|
||||
push esi
|
||||
push edi
|
||||
mov edi, eax
|
||||
mov esi, [.sz_name]
|
||||
mov esi, [name]
|
||||
mov ecx, 16/4
|
||||
rep movsd
|
||||
pop edi
|
||||
pop esi
|
||||
|
||||
mov [eax+SRV.magic], ' SRV'
|
||||
mov [eax+SRV.size], SRV_SIZE
|
||||
@ -380,12 +388,14 @@ reg_service:
|
||||
mov [ebx+SRV.fd], eax
|
||||
mov [edx+SRV.bk], eax
|
||||
|
||||
mov ecx, [.handler]
|
||||
mov ecx, [handler]
|
||||
mov [eax+SRV.srv_proc], ecx
|
||||
ret 8
|
||||
pop ebx
|
||||
ret
|
||||
.fail:
|
||||
xor eax, eax
|
||||
ret 8
|
||||
ret
|
||||
endp
|
||||
|
||||
align 4
|
||||
proc get_proc stdcall, exp:dword, sz_name:dword
|
||||
|
||||
@ -1011,11 +1011,35 @@ new_services:
|
||||
ret
|
||||
@@:
|
||||
cmp eax, 20
|
||||
ja .fail
|
||||
ja @F
|
||||
mov eax, ecx
|
||||
call user_realloc
|
||||
mov [esp+36], eax
|
||||
ret
|
||||
@@:
|
||||
cmp eax, 21
|
||||
ja @f
|
||||
cmp ebx, OS_BASE
|
||||
jae .fail
|
||||
|
||||
stdcall load_PE, ebx
|
||||
|
||||
test eax, eax
|
||||
jz @F
|
||||
|
||||
mov esi, eax
|
||||
stdcall eax, DRV_ENTRY
|
||||
|
||||
test eax, eax
|
||||
jz @F
|
||||
|
||||
mov [eax+SRV.entry], esi
|
||||
|
||||
@@:
|
||||
mov [esp+36], eax
|
||||
ret
|
||||
|
||||
|
||||
.fail:
|
||||
xor eax, eax
|
||||
mov [esp+36], eax
|
||||
|
||||
@ -1,9 +1,13 @@
|
||||
|
||||
include 'export.inc'
|
||||
|
||||
align 4
|
||||
|
||||
proc load_PE stdcall, file_name:dword
|
||||
locals
|
||||
image dd ?
|
||||
entry dd ?
|
||||
base dd ?
|
||||
endl
|
||||
|
||||
stdcall load_file, [file_name]
|
||||
@ -18,15 +22,26 @@ proc load_PE stdcall, file_name:dword
|
||||
test eax, eax
|
||||
jz .cleanup
|
||||
|
||||
mov [base], eax
|
||||
|
||||
stdcall map_PE, eax, [image]
|
||||
ret
|
||||
|
||||
mov [entry], eax
|
||||
test eax, eax
|
||||
jnz .cleanup
|
||||
|
||||
stdcall kernel_free, [base]
|
||||
.cleanup:
|
||||
stdcall kernel_free, [image]
|
||||
mov eax, [entry]
|
||||
ret
|
||||
.fail:
|
||||
xor eax, eax
|
||||
ret
|
||||
endp
|
||||
|
||||
DWORD equ dword
|
||||
PTR equ
|
||||
|
||||
align 4
|
||||
map_PE: ;stdcall base:dword, image:dword
|
||||
@ -35,173 +50,237 @@ map_PE: ;stdcall base:dword, image:dword
|
||||
push edi
|
||||
push esi
|
||||
push ebx
|
||||
sub esp, 44
|
||||
sub esp, 60
|
||||
mov ebx, DWORD PTR [esp+84]
|
||||
mov ebp, DWORD PTR [esp+80]
|
||||
mov edx, ebx
|
||||
mov esi, ebx
|
||||
add edx, DWORD PTR [ebx+60]
|
||||
mov edi, ebp
|
||||
mov DWORD PTR [esp+32], edx
|
||||
mov ecx, DWORD PTR [edx+84]
|
||||
|
||||
mov ebp, [esp+68]
|
||||
mov ebx, [esp+64]
|
||||
mov edx, ebp
|
||||
mov esi, ebp
|
||||
add edx, [ebp+60]
|
||||
mov edi, ebx
|
||||
mov [esp+32], edx
|
||||
|
||||
mov ecx, [edx+84]
|
||||
shr ecx, 2
|
||||
rep movsd
|
||||
|
||||
movzx eax, word [edx+6]
|
||||
mov dword [esp+36], 0
|
||||
mov [esp+28], eax
|
||||
jmp .L6
|
||||
.L7:
|
||||
mov eax, [edx+264]
|
||||
movzx eax, WORD PTR [edx+6]
|
||||
mov DWORD PTR [esp+36], 0
|
||||
mov DWORD PTR [esp+16], eax
|
||||
jmp L2
|
||||
L3:
|
||||
mov eax, DWORD PTR [edx+264]
|
||||
test eax, eax
|
||||
je .L8
|
||||
|
||||
mov esi, ebp
|
||||
mov edi, ebx
|
||||
add esi, [edx+268]
|
||||
je L4
|
||||
mov esi, ebx
|
||||
mov edi, ebp
|
||||
add esi, DWORD PTR [edx+268]
|
||||
mov ecx, eax
|
||||
add edi, [edx+260]
|
||||
add edi, DWORD PTR [edx+260]
|
||||
|
||||
shr ecx, 2
|
||||
rep movsd
|
||||
.L8:
|
||||
mov ecx, [edx+256]
|
||||
|
||||
L4:
|
||||
mov ecx, DWORD PTR [edx+256]
|
||||
add ecx, 4095
|
||||
and ecx, -4096
|
||||
cmp ecx, eax
|
||||
jbe .L10
|
||||
|
||||
jbe L6
|
||||
sub ecx, eax
|
||||
add eax, [edx+260]
|
||||
lea edi, [eax+ebx]
|
||||
add eax, DWORD PTR [edx+260]
|
||||
lea edi, [eax+ebp]
|
||||
|
||||
xor eax, eax
|
||||
rep stosb
|
||||
.L10:
|
||||
inc dword [esp+36]
|
||||
|
||||
L6:
|
||||
inc DWORD PTR [esp+36]
|
||||
add edx, 40
|
||||
.L6:
|
||||
mov esi, [esp+28]
|
||||
cmp [esp+36], esi
|
||||
jne .L7
|
||||
|
||||
mov edi, [esp+32]
|
||||
cmp dword [edi+164], 0
|
||||
je .L13
|
||||
|
||||
mov eax, [esp+32]
|
||||
mov edi, ebx
|
||||
mov ecx, ebx
|
||||
sub edi, [eax+52]
|
||||
add ecx, [eax+160]
|
||||
|
||||
mov edx, edi
|
||||
shr edx, 16
|
||||
mov [esp+20], edx
|
||||
jmp .L15
|
||||
.L16:
|
||||
lea esi, [eax-8]
|
||||
xor ebp, ebp
|
||||
shr esi, 1
|
||||
jmp .L17
|
||||
.L18:
|
||||
movzx eax, word [ecx+8+ebp*2]
|
||||
L2:
|
||||
mov esi, DWORD PTR [esp+16]
|
||||
cmp DWORD PTR [esp+36], esi
|
||||
jne L3
|
||||
mov edi, DWORD PTR [esp+32]
|
||||
cmp DWORD PTR [edi+164], 0
|
||||
je L9
|
||||
mov esi, ebp
|
||||
mov ecx, ebp
|
||||
sub esi, DWORD PTR [edi+52]
|
||||
add ecx, DWORD PTR [edi+160]
|
||||
mov eax, esi
|
||||
shr eax, 16
|
||||
mov DWORD PTR [esp+12], eax
|
||||
jmp L11
|
||||
L12:
|
||||
lea ebx, [eax-8]
|
||||
xor edi, edi
|
||||
shr ebx,1
|
||||
jmp L13
|
||||
L14:
|
||||
movzx eax, WORD PTR [ecx+8+edi*2]
|
||||
mov edx, eax
|
||||
shr eax, 12
|
||||
and edx, 4095
|
||||
add edx, [ecx]
|
||||
add edx, DWORD PTR [ecx]
|
||||
cmp ax, 2
|
||||
je .L21
|
||||
|
||||
je L17
|
||||
cmp ax, 3
|
||||
je .L22
|
||||
|
||||
je L18
|
||||
dec ax
|
||||
jne .L19
|
||||
|
||||
mov eax, [esp+20]
|
||||
add [edx+ebx], ax
|
||||
.L21:
|
||||
add [edx+ebx], di
|
||||
.L22:
|
||||
add [edx+ebx], edi
|
||||
.L19:
|
||||
inc ebp
|
||||
.L17:
|
||||
cmp ebp, esi
|
||||
jne .L18
|
||||
|
||||
add ecx, [ecx+4]
|
||||
.L15:
|
||||
mov eax, [ecx+4]
|
||||
jne L15
|
||||
mov eax, DWORD PTR [esp+12]
|
||||
add WORD PTR [edx+ebp], ax
|
||||
L17:
|
||||
add WORD PTR [edx+ebp], si
|
||||
L18:
|
||||
add DWORD PTR [edx+ebp], esi
|
||||
L15:
|
||||
inc edi
|
||||
L13:
|
||||
cmp edi, ebx
|
||||
jne L14
|
||||
add ecx, DWORD PTR [ecx+4]
|
||||
L11:
|
||||
mov eax, DWORD PTR [ecx+4]
|
||||
test eax, eax
|
||||
jne .L16
|
||||
.L13:
|
||||
mov edx, [esp+32]
|
||||
cmp dword [edx+132], 0
|
||||
je .L24
|
||||
|
||||
mov eax, ebx
|
||||
add eax, [edx+128]
|
||||
lea esi, [eax+20]
|
||||
.L26:
|
||||
cmp dword [esi-16], 0
|
||||
jne .L27
|
||||
|
||||
cmp dword [esi-8], 0
|
||||
je .L24
|
||||
.L27:
|
||||
mov ecx, [esi-20]
|
||||
mov ebp, ebx
|
||||
add ebp, [esi-4]
|
||||
add ecx, ebx
|
||||
mov [esp+40], ecx
|
||||
.L29:
|
||||
mov edi, [esp+40]
|
||||
mov eax, [edi]
|
||||
jne L12
|
||||
L9:
|
||||
mov edx, DWORD PTR [esp+32]
|
||||
cmp DWORD PTR [edx+132], 0
|
||||
je L20
|
||||
mov eax, ebp
|
||||
add eax, DWORD PTR [edx+128]
|
||||
mov DWORD PTR [esp+40], 0
|
||||
add eax, 20
|
||||
mov DWORD PTR [esp+56], eax
|
||||
L22:
|
||||
mov ecx, DWORD PTR [esp+56]
|
||||
cmp DWORD PTR [ecx-16], 0
|
||||
jne L23
|
||||
cmp DWORD PTR [ecx-8], 0
|
||||
je L25
|
||||
L23:
|
||||
mov edi, DWORD PTR [__exports+32]
|
||||
mov esi, DWORD PTR [__exports+28]
|
||||
mov eax, DWORD PTR [esp+56]
|
||||
mov DWORD PTR [esp+20], edi
|
||||
sub edi, -2147483648
|
||||
sub esi, -2147483648
|
||||
mov DWORD PTR [esp+44], esi
|
||||
mov ecx, DWORD PTR [eax-4]
|
||||
mov DWORD PTR [esp+48], edi
|
||||
mov edx, DWORD PTR [eax-20]
|
||||
mov DWORD PTR [esp+52], 0
|
||||
add ecx, ebp
|
||||
add edx, ebp
|
||||
mov DWORD PTR [esp+24], edx
|
||||
mov DWORD PTR [esp+28], ecx
|
||||
L26:
|
||||
mov esi, DWORD PTR [esp+52]
|
||||
mov edi, DWORD PTR [esp+24]
|
||||
mov eax, DWORD PTR [edi+esi*4]
|
||||
test eax, eax
|
||||
je .L30
|
||||
|
||||
je L27
|
||||
test eax, eax
|
||||
js .L30
|
||||
|
||||
lea eax, [eax+2+ebx]
|
||||
mov edi, kernel_export
|
||||
mov [ebp], dword -1
|
||||
mov [esp+24], eax
|
||||
.L33:
|
||||
js L27
|
||||
lea edi, [ebp+eax]
|
||||
mov eax, DWORD PTR [esp+28]
|
||||
mov DWORD PTR [eax+esi*4], 0
|
||||
lea esi, [edi+2]
|
||||
push eax
|
||||
push 32
|
||||
movzx eax, WORD PTR [edi]
|
||||
mov edx, DWORD PTR [esp+56]
|
||||
mov eax, DWORD PTR [edx+eax*4]
|
||||
sub eax, -2147483648
|
||||
push eax
|
||||
push esi
|
||||
call strncmp
|
||||
pop ebx
|
||||
xor ebx, ebx
|
||||
test eax, eax
|
||||
jne L32
|
||||
jmp L30
|
||||
L33:
|
||||
push ecx
|
||||
push 16
|
||||
push dword [edi]
|
||||
push dword [esp+36]
|
||||
push 32
|
||||
mov ecx, DWORD PTR [esp+28]
|
||||
mov eax, DWORD PTR [ecx-2147483648+ebx*4]
|
||||
sub eax, -2147483648
|
||||
push eax
|
||||
push esi
|
||||
call strncmp
|
||||
pop edx
|
||||
test eax, eax
|
||||
jne .L34
|
||||
mov eax, [edi+4]
|
||||
mov [ebp], eax
|
||||
jmp .L36
|
||||
.L34:
|
||||
add edi, 8
|
||||
cmp dword [edi], 0
|
||||
jne .L33
|
||||
.L36:
|
||||
add dword [esp+40], 4
|
||||
add ebp, 4
|
||||
jmp .L29
|
||||
.L30:
|
||||
add esi, 20
|
||||
jmp .L26
|
||||
.L24:
|
||||
mov eax, [esp+32]
|
||||
add ebx, [eax+40]
|
||||
add esp, 44
|
||||
mov eax, ebx
|
||||
jne L34
|
||||
mov esi, DWORD PTR [esp+44]
|
||||
mov edx, DWORD PTR [esp+52]
|
||||
mov ecx, DWORD PTR [esp+28]
|
||||
mov eax, DWORD PTR [esi+ebx*4]
|
||||
sub eax, -2147483648
|
||||
mov DWORD PTR [ecx+edx*4], eax
|
||||
jmp L36
|
||||
L34:
|
||||
inc ebx
|
||||
L32:
|
||||
cmp ebx, DWORD PTR [__exports+24]
|
||||
jb L33
|
||||
L36:
|
||||
cmp ebx, DWORD PTR [__exports+24]
|
||||
jne L37
|
||||
|
||||
mov esi, msg_unresolved
|
||||
call sys_msg_board_str
|
||||
lea esi, [edi+2]
|
||||
call sys_msg_board_str
|
||||
mov esi, msg_CR
|
||||
call sys_msg_board_str
|
||||
|
||||
mov DWORD PTR [esp+40], 1
|
||||
jmp L37
|
||||
L30:
|
||||
movzx eax, WORD PTR [edi]
|
||||
mov esi, DWORD PTR [esp+44]
|
||||
mov edi, DWORD PTR [esp+52]
|
||||
mov edx, DWORD PTR [esp+28]
|
||||
mov eax, DWORD PTR [esi+eax*4]
|
||||
sub eax, -2147483648
|
||||
mov DWORD PTR [edx+edi*4], eax
|
||||
L37:
|
||||
inc DWORD PTR [esp+52]
|
||||
jmp L26
|
||||
L27:
|
||||
add DWORD PTR [esp+56], 20
|
||||
jmp L22
|
||||
L25:
|
||||
xor eax, eax
|
||||
cmp DWORD PTR [esp+40], 0
|
||||
jne L40
|
||||
L20:
|
||||
mov ecx, DWORD PTR [esp+32]
|
||||
mov eax, ebp
|
||||
add eax, DWORD PTR [ecx+40]
|
||||
L40:
|
||||
add esp, 60
|
||||
pop ebx
|
||||
pop esi
|
||||
pop edi
|
||||
pop ebp
|
||||
ret 8
|
||||
|
||||
align 16
|
||||
__exports:
|
||||
export 'KERNEL', \
|
||||
alloc_kernel_space, 'AllocKernelSpace', \ ; stdcall
|
||||
free_kernel_space, 'FreeKernelSpace', \ ; stdcall
|
||||
kernel_alloc, 'KernelAlloc', \ ; stdcall
|
||||
kernel_free, 'KernelFree', \ ; stdcall
|
||||
pci_api, 'PciApi', \
|
||||
pci_read32, 'PciRead32', \ ; stdcall
|
||||
reg_service, 'RegService', \
|
||||
user_alloc, 'UserAlloc', \ ; stdcall
|
||||
user_free, 'UserFree', \ ; stdcall
|
||||
sys_msg_board_str, 'SysMsgBoardStr'
|
||||
|
||||
|
||||
|
||||
|
||||
@ -500,8 +500,6 @@ term9:
|
||||
|
||||
push esi
|
||||
call [eax+APPOBJ.destroy]
|
||||
;mov esi, msg_obj_destroy
|
||||
;call sys_msg_board_str
|
||||
DEBUGF 1,"%s",msg_obj_destroy
|
||||
pop esi
|
||||
jmp @B
|
||||
|
||||
@ -606,16 +606,6 @@ no_lib_load:
|
||||
|
||||
; LOAD FONTS I and II
|
||||
|
||||
|
||||
; pushad
|
||||
; push eax
|
||||
; mov eax,char
|
||||
; call file_system_lfn
|
||||
; mov eax,char2
|
||||
; call file_system_lfn
|
||||
; pop eax
|
||||
; popad
|
||||
|
||||
stdcall read_file, char, FONT_I, 0, 2304
|
||||
stdcall read_file, char2, FONT_II, 0, 2560
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user