Subversion Repositories Kolibri OS

format binary as ""

use32

; standard header

        db      'MENUET01'      ; signature

        dd      1               ; header version

        dd      start           ; entry point

        dd      i_end           ; initialized size

        dd      mem             ; required memory

        dd      mem             ; stack pointer

        dd      0               ; parameters

        dd      0               ; path

__DEBUG__ equ 0

__DEBUG_LEVEL__ equ 1

BUFFERSIZE      equ 4096

; useful includes

include '../macros.inc'

purge mov,add,sub

include '../proc32.inc'

include '../dll.inc'

include '../debug-fdo.inc'

include '../network.inc'

; entry point

start:

; load libraries

        stdcall dll.Load, @IMPORT

        test    eax, eax

        jnz     exit

; initialize console

        push    1

        call    [con_start]

        push    title

        push    25

        push    80

        push    25

        push    80

        call    [con_init]

; main loop

        push    str1

        call    [con_write_asciiz]

main:

; write prompt

        push    str2

        call    [con_write_asciiz]

; read string

        mov     esi, s

        push    256

        push    esi

        call    [con_gets]

; check for exit

        test    eax, eax

        jz      done

        cmp     byte [esi], 10

        jz      done

; delete terminating '\n'

        push    esi

@@:

        lodsb

        test    al, al

        jnz     @b

        mov     byte [esi-2], al

        pop     esi

; resolve name

        push    esp     ; reserve stack place

        push    esp     ; fourth parameter

        push    0       ; third parameter

        push    0       ; second parameter

        push    esi     ; first parameter

        call    [getaddrinfo]

        pop     esi

; test for error

        test    eax, eax

        jnz     fail

; write results

        push    str3

        call    [con_write_asciiz]

;        mov     edi, esi

; convert IP address to decimal notation

        mov     eax, [esi+addrinfo.ai_addr]

        mov     eax, [eax+sockaddr_in.sin_addr]

        mov     [sockaddr1.ip], eax

        push    eax

        call    [inet_ntoa]

; write result

        push    eax

        call    [con_write_asciiz]

; free allocated memory

        push    esi

        call    [freeaddrinfo]

        push    str4

        call    [con_write_asciiz]

        mcall   socket, AF_INET4, SOCK_STREAM, 0

        cmp     eax, -1

        jz      fail2

        mov     [socketnum], eax

        mcall   connect, [socketnum], sockaddr1, 18

        mcall   40, 1 shl 7 ; + 7

        call    [con_cls]

        mcall   18, 7

        push    eax

        mcall   51, 1, thread, mem - 2048

        pop     ecx

        mcall   18, 3

mainloop:

    DEBUGF  1, 'TELNET: Waiting for events\n'

        mcall   10

    DEBUGF  1, 'TELNET: EVENT %x !\n', eax

        mcall   recv, [socketnum], buffer_ptr, BUFFERSIZE, 0

        cmp     eax, -1

        je      mainloop

    DEBUGF  1, 'TELNET: got %u bytes of data !\n', eax

        mov     esi, buffer_ptr

        lea     edi, [esi + eax]

        mov     byte [edi], 0

  .scan_cmd:

        cmp     byte [esi], 0xff        ; Interpret As Command

        jne     .no_cmd

        ; TODO: parse options, for now, we will reply with 'WONT' to everything

        mov     byte [esi + 1], 252     ; WONT

        add     esi, 3                  ; a command is always 3 bytes

        jmp     .scan_cmd

  .no_cmd:

        cmp     esi, buffer_ptr

        je      .print_loop

    DEBUGF  1, 'TELNET: sending data\n'

        push    esi edi

        sub     esi, buffer_ptr

        mcall   send, [socketnum], buffer_ptr, , 0

        pop     edi esi

  .print_loop:

    DEBUGF  1, 'TELNET: printloop\n'

        cmp     esi, edi

        jae     mainloop

        cmp     byte [esi], 0x1b        ; escape character

        jne     .print_byte

        inc     esi

        cmp     word [esi], 0x485b      ; move cursor to beginning

        jne     @f

        inc     esi

        inc     esi

    DEBUGF  1, 'TELNET: resetting cursor \n'

        push    0

        push    0

        call    [con_set_cursor_pos]

        jmp     .print_loop

  @@:

        inc     esi

        inc     esi

        jmp     .print_loop

  .print_byte:

        push    dword 1

        push    esi                     ; next string to print

        inc     esi

        call    [con_write_string]

        jmp     .print_loop

fail:

        push    str5

        jmp     main

fail2:

        push    str6

        jmp     main

done:

        push    1

        call    [con_exit]

exit:

        mcall   -1

thread:

        mcall   40, 0

  .loop:

        call    [con_getch2]

        mov     byte [send_data], al

        mcall   send, [socketnum], send_data, 1

        jmp     .loop

; data

title   db      'Telnet',0

str1    db      'Telnet v0.1',10,' for KolibriOS # 1281 or later. ',10,10,'If you dont know where to connect to, try towel.blinkenlights.nl',10,10,0

str2    db      '> ',0

str3    db      'Connecting to: ',0

str4    db      10,0

str5    db      'Name resolution failed.',10,10,0

str6    db      'Could not open socket',10,10,0

str7    db      'Got data!',10,10,0

sockaddr1:

        dw AF_INET4

.port   dw 23

.ip     dd 0

        rb 10

include_debug_strings    ; ALWAYS present in data section

; import

align 4

@IMPORT:

library network, 'network.obj', console, 'console.obj'

import  network,        \

        getaddrinfo,    'getaddrinfo',  \

        freeaddrinfo,   'freeaddrinfo', \

        inet_ntoa,      'inet_ntoa'

import  console,        \

        con_start,      'START',        \

        con_init,       'con_init',     \

        con_write_asciiz,       'con_write_asciiz',     \

        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'

i_end:

socketnum       dd ?

buffer_ptr      rb BUFFERSIZE+1

send_data       rb 100

s       rb      256

align   4

rb      4096    ; stack

mem: