kolibrios-fun/kernel/branches/Kolibri-acpi/core/peload.inc

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                              ;;
;; Copyright (C) KolibriOS team 2004-2012. All rights reserved. ;;
;; Distributed under terms of the GNU General Public License    ;;
;;                                                              ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

$Revision$

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
        add     edi, OS_BASE
        add     esi, OS_BASE
        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]
        add     eax, OS_BASE
        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+OS_BASE+ebx*4]
        add     eax, OS_BASE
        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]
        add     eax, OS_BASE
        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]
        add     eax, OS_BASE
        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