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]
           test eax, eax
           jz .fail

           mov [image], eax

           mov edx, [eax+60]

           stdcall kernel_alloc, [eax+80+edx]
           test eax, eax
           jz .cleanup

           mov [base], eax

           stdcall map_PE, eax, [image]

           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
           cld
        push    ebp
	push	edi
	push	esi
	push	ebx
	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]

	shr ecx, 2
	rep movsd

	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	L4
	mov	esi, ebx
	mov	edi, ebp
	add	esi, DWORD PTR [edx+268]
	mov	ecx, eax
	add	edi, DWORD PTR [edx+260]

	shr ecx, 2
	rep movsd

L4:
	mov	ecx, DWORD PTR [edx+256]
	add	ecx, 4095
	and	ecx, -4096
	cmp	ecx, eax
	jbe	L6
	sub	ecx, eax
	add	eax, DWORD PTR [edx+260]
	lea	edi, [eax+ebp]

	xor eax, eax
	rep stosb

L6:
	inc	DWORD PTR [esp+36]
	add	edx, 40
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, DWORD PTR [ecx]
	cmp	ax, 2
	je	L17
	cmp	ax, 3
	je	L18
	dec	ax
	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	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	L27
	test	eax, eax
	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	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	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
          malloc,             'Kmalloc',            \
          free,               'Kfree',              \
          map_io_mem,         'MapIoMem',           \      ; stdcall
\
          pci_api,            'PciApi',             \
          pci_read8,          'PciRead8',           \      ; stdcall
          pci_read16,         'PciRead16',          \      ; stdcall
          pci_read32,         'PciRead32',          \      ; stdcall
          pci_write8,         'PciWrite8',          \      ; stdcall
          pci_write16,        'PciWrite16',         \      ; stdcall
          pci_write32,        'PciWrite32',         \      ; stdcall
\
          reg_service,        'RegService',         \
          user_alloc,         'UserAlloc',          \      ; stdcall
          user_free,          'UserFree',           \      ; stdcall
          sys_msg_board_str,  'SysMsgBoardStr'