Subversion Repositories Kolibri OS

include '../proc32.inc'

format MS COFF

public _load_libconsole

public _con_init@20

public _con_exit@4

public _con_get_flags

public _con_set_flags@4

public _con_cls

public _con_write_string@8

section '.text' align 16

;void* __fastcall getprocaddr(export, name)

align 4

getprocaddress:

        push    esi

        push    edi

        xor     eax, eax

        test    ecx, ecx        ; If hlib = 0 then goto .end

        jz      .end

.next:

        cmp     [ecx], dword 0  ; If end of export table then goto .end

        jz      .end

        xor     eax, eax

        mov     esi, [ecx]

        mov     edi, edx        ; name

.next_:

        lodsb

        scasb

        jne     .fail

        or      al, al

        jnz     .next_

        jmp     .ok

.fail:

        add     ecx, 8

        jmp     .next

.ok:

        mov eax, [ecx + 4]      ; return address

.end:

        pop     edi

        pop     esi

        ret

;void fastcall dll_link(export, import)

align 4

dll_link:

        push    esi

        push    ecx

        mov     esi, edx

        test    esi, esi

        jz      .done

.next:

        mov     edx, [esi]

        test    edx, edx

        jz      .done

        mov     ecx, [esp]

        call    getprocaddress

        test    eax, eax

        jz      .done

        mov     [esi], eax

        add     esi, 4

        jmp     .next

.done:

        pop     ecx

        pop     esi

        ret

align 4

dll_load:

        push    ebp

        push    ebx

        push    esi

        push    edi

        mov     ebp, [esp+20]

.next_lib:

        mov     edx, [ebp]

        test    edx, edx

        jz      .exit

        mov     esi, [ebp+4]

        mov     edi, s_libdir.fname

@@:

        lodsb

        stosb

        test  al, al

        jnz @b

        mov     eax, 68

        mov     ebx, 19

        mov     ecx, s_libdir

        int     0x40

        test    eax, eax

        jz      .fail

        mov     ecx, eax

        call    dll_link

        mov     eax, [ecx]

        cmp     dword[eax], 'lib_'

        jnz     @f

        mov     esi, [ecx+4]

        pushad

        mov     eax, mem.Alloc

        mov     ebx, mem.Free

        mov     ecx, mem.ReAlloc

        mov     edx, dll_load

        call    esi

        popad

@@:

        add     ebp, 8

        jmp     .next_lib

.exit:

        pop     edi

        pop     esi

        pop     ebx

        pop     ebp

        xor     eax, eax

        ret 4

.fail:

        pop     edi

        pop     esi

        pop     ebx

        pop     ebp

        inc     eax

        ret 4

align 4

_load_libconsole:

        push    ebx

        mov     eax, 40

        mov     ebx, 1 shl 8

        int     0x40

        pop     ebx

        push    @IMPORT

        call    dll_load

        test    eax, eax

        jnz     .fail

        push    1

        call    [con_start]

        xor     eax, eax

.fail:

        ret

align 4

_con_init@20:

        jmp     [con_init]

align 4

_con_exit@4:

        jmp     [con_exit]

align 4

_con_write_string@8:

        jmp     [con_write_string]

_con_get_flags:

_con_set_flags@4:

_con_cls:

        ret

proc mem.Alloc, size

        push    ebx ecx

        mov     ecx, [size]

        mov     eax, 68

        mov     ebx, 12

        int     0x40

        pop     ecx ebx

        ret

endp

;-----------------------------------------------------------------------------

proc mem.ReAlloc, mptr, size

        push    ebx ecx edx

        mov     ecx, [size]

        test    ecx, ecx

        jz      @f

@@:

        mov     edx, [mptr]

        test    edx, edx

        jz      @f

@@:

        mov     eax, 68

        mov     ebx, 20

        int     0x40

        test    eax, eax

        jz  @f

@@:

        pop edx ecx ebx

        ret

endp

;-----------------------------------------------------------------------------

proc mem.Free, mptr

        push    ebx ecx

        mov     ecx,[mptr]

        test    ecx,ecx

        jz  @f

@@:

        mov     eax, 68

        mov     ebx, 13

        int     0x40

        pop ecx ebx

        ret

endp

;section '.ctors' align 4

;align 4

;dd _load_libconsole

section '.data' align 16

; -------------------------

macro library [lname,fname]

{

  forward

    dd __#lname#_library_table__,__#lname#_library_name__

  common

    dd 0

  forward

    align 4

    __#lname#_library_name__ db fname,0

}

macro import lname,[name,sname]

{

  common

    align 4

    __#lname#_library_table__:

  forward

    if used name

      name dd __#name#_import_name__

    end if

  common

    dd 0

  forward

    if used name

      align 4

      __#name#_import_name__ db sname,0

    end if

}

align   4

@IMPORT:

library console,        'console.obj'

import  console,                            \

        con_start,      'START',            \

        con_init,       'con_init',         \

        con_exit,       'con_exit',         \

        con_gets,       'con_gets',         \

        con_cls,        'con_cls',          \

        con_getch2,     'con_getch2',       \

        con_set_cursor_pos, 'con_set_cursor_pos',\

        con_write_string, 'con_write_string',\

        con_get_flags,  'con_get_flags',    \

        con_set_flags,  'con_set_flags'

s_libdir:

  db '/sys/lib/'

  .fname rb 32