kolibrios/programs/network/tftpc/tftpc.asm

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                                 ;;
;; Copyright (C) KolibriOS team 2010-2014. All rights reserved.    ;;
;; Distributed under terms of the GNU General Public License       ;;
;;                                                                 ;;
;;  tftpc.asm - TFTP client for KolibriOS                          ;;
;;                                                                 ;;
;;  Written by hidnplayr@kolibrios.org                             ;;
;;                                                                 ;;
;;          GNU GENERAL PUBLIC LICENSE                             ;;
;;             Version 2, June 1991                                ;;
;;                                                                 ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

format binary as ""

use32
        org     0x0

        db      'MENUET01'
        dd      0x1
        dd      START
        dd      I_END
        dd      IM_END+0x1000
        dd      IM_END+0x1000
        dd      0, 0

include '../../proc32.inc'
include '../../macros.inc'
include '../../libio.inc'
include '../../dll.inc'
include '../../develop/libraries/box_lib/trunk/box_lib.mac'

include '../../network.inc'


filebuffer_size = 4*4096        ; 16kb   (dont try to change it yet..)
TIMEOUT         = 100
buffer_len      = 1500

opcode_rrq      = 1
opcode_wrq      = 2
opcode_data     = 3
opcode_ack      = 4
opcode_error    = 5

; read/write request packet
;
;  2 bytes     string    1 byte     string   1 byte
;  ------------------------------------------------
; | Opcode |  Filename  |   0  |    Mode    |   0  |
;  ------------------------------------------------

; data packet
;
;  2 bytes     2 bytes      n bytes
;  ----------------------------------
; | Opcode |   Block #  |   Data     |
;  ----------------------------------

; acknowledgement packet
;
;  2 bytes     2 bytes
;  ---------------------
; | Opcode |   Block #  |
;  ---------------------

; error packet
;
;  2 bytes  2 bytes        string    1 byte
;  ----------------------------------------
; | Opcode |  ErrorCode |   ErrMsg   |   0  |
;  ----------------------------------------


START:

        mcall   68, 11

        stdcall dll.Load, @IMPORT
        or      eax, eax
        jnz     exit

stop_transfer:
        mcall   40, EVM_MOUSE + EVM_MOUSE_FILTER + EVM_REDRAW + EVM_BUTTON + EVM_KEY + EVM_STACK

red_win:
        call    draw_window

align 4
still:
        mcall   10

        dec     eax
        jz      red_win

        dec     eax
        jz      key

        dec     eax
        jz      button

        invoke  edit_box_mouse, edit1
        invoke  edit_box_mouse, edit2
        invoke  edit_box_mouse, edit3
        invoke  edit_box_mouse, edit4

        invoke  option_box_mouse, Option_boxs1
        invoke  option_box_mouse, Option_boxs2

        jmp     still

button:
        mcall   17

        cmp     ah, 0x10        ; connect button
        je      start_transfer

        test    ah , ah
        jz      still

exit:
        mcall   -1
key:
        mcall   2

        invoke  edit_box_key, edit1
        invoke  edit_box_key, edit2
        invoke  edit_box_key, edit3
        invoke  edit_box_key, edit4

        jmp still


draw_window:
; get system colors
        mcall   48, 3, sc, 40

        mcall   12, 1

        mov     edx, [sc.work]
        or      edx, 0x34000000
        xor     esi, esi
        mov     edi, str_title
        mcall   0, 50 shl 16 + 400, 30 shl 16 + 180

        mov     ebx, 35 shl 16 + 10
        mov     ecx, 0x80000000
        or      ecx, [sc.work_text]
        mov     edx, str_server
        mcall   4
        mov     ebx, 5 shl 16 + 30
        mov     edx, str_source
        mcall
        mov     ebx, 11 shl 16 + 50
        mov     edx, str_destination
        mcall
        mov     ebx, 47 shl 16 + 72
        mov     edx, str_mode
        mcall
        mov     ebx, 160 shl 16 + 72
        mov     edx, str_method
        mcall
        mov     ebx, 270 shl 16 + 72
        mov     edx, str_blocksize
        mcall

        invoke  edit_box_draw, edit1
        invoke  edit_box_draw, edit2
        invoke  edit_box_draw, edit3
        invoke  edit_box_draw, edit4

        invoke  option_box_draw, Option_boxs1
        invoke  option_box_draw, Option_boxs2

        mov     esi, [sc.work_button]
        mcall   8, 210 shl 16 + 170, 105 shl 16 + 16, 0x10

        mov     ecx, 0x80000000
        or      ecx, [sc.work_button_text]
        mcall   4, 260 shl 16 + 110, , str_transfer

        mov     ecx, 0x80000000
        or      ecx, [sc.work_text]
        mcall   4, 350 shl 16 + 137, , str_kb_s
        mcall   4, 50 shl 16 + 137, , str_complete

        mcall   38, 10 shl 16 + 380, 130 shl 16 + 130, [sc.work_graph]

        mcall   47, 1 shl 31 + 7 shl 16 + 1, kbps, 305 shl 16 + 137, [sc.work_text]
        mcall   47, 1 shl 31 + 3 shl 16 + 1, done, 25 shl 16 + 137, ;[sc.work_text]

        mcall   12, 2

        ret





start_transfer:

        ; first, resolve the hostname

        push    esp     ; reserve stack place

        push    esp     ; fourth parameter
        push    0       ; third parameter
        push    0       ; second parameter
        push    dword SRV     ; first parameter
        call    [getaddrinfo]

        pop     esi

; test for error
        test    eax, eax
        jnz     still

        mov     esi, [esi]
        mov     esi, [esi + sockaddr_in.sin_addr]
        mov     dword [IP], esi

        mcall   socket, AF_INET4, SOCK_DGRAM, 0                ; socket_open
        cmp     eax, -1
        je      still

        mov     [socketnum], eax

        mcall   connect, [socketnum], sockaddr, sockaddr_len         ; socket_connect
        cmp     eax, -1
        je      still

        mov     word [I_END], opcode_rrq
        cmp     [option_group2],op3
        je      @f
        mov     word [I_END], opcode_wrq
      @@:

        xor     al , al
        mov     edi, remote_addr
        mov     ecx, 250
        repnz   scasb
        sub     edi, remote_addr-1
        mov     ecx, edi
        mov     edi, I_END+2
        mov     esi, remote_addr
        rep     movsb

        cmp     [option_group1], op1
        je      .ascii

        mov     esi, octet
        movsd
        movsb

        jmp     .send_request

      .ascii:

        mov     esi, netascii
        movsd
        movsd

      .send_request:

        xor     al, al
        stosb

        sub     edi, I_END
        mov     esi, edi
        mcall   send, [socketnum], I_END

        mcall   40, EVM_REDRAW + EVM_BUTTON + EVM_STACK

        mov     [last_ack], 0




receive_data_loop:

        mcall   23, TIMEOUT

        dec     eax
        jz      .red

        dec     eax
        jz      .key


        mcall   recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT     ; receive data

        cmp     word[buffer], opcode_data
        jne     .error

        mov     bx, [last_ack]
        cmp     word [buffer + 2], bx
        jne     .packet_got_lost
        inc     [last_ack]

        cmp     eax, 4+512
        je      .continue

; last packet, or something else
.error:

.packet_got_lost:



.continue:

        mov     word[buffer], opcode_ack                ; send ack
        mcall   send, [socketnum], buffer, 4, 0

        jmp     receive_data_loop

.red:

        call    draw_window

        jmp     receive_data_loop


.key:
        mcall   2
        cmp     ah, 2
        jz      exit

        ; close socket ?

        jmp     receive_data_loop






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


send_:

        invoke  file_open, local_addr, O_READ
        or      eax, eax
        jz      .exit
        mov     [fh], eax

        stdcall mem.Alloc, filebuffer_size
        or      eax, eax
        jz      .exit
        mov     [fb], eax

        mov     [last_ack], 0
        mov     [fo], 0

.read_chunk:

        invoke  file_seek, [fh], [fo], SEEK_END
        cmp     eax, -1
        je      .exit
        invoke  file_read, [fh], [fb], filebuffer_size
        cmp     eax, -1
        je      .exit
        add     [fo], filebuffer_size
        cmp     eax, filebuffer_size
        je      .packet

        ; ijhidfhfdsndsfqk

.packet:

        movzx   esi, [last_ack]
        and     esi, 0x000000001f   ; last five bits    BUFFER SIZE MUST BE 16 kb for this to work !!!
        shl     esi, 9              ; = * 512
        add     esi, [fb]
        mov     edi, buffer
        mov     ax, opcode_data
        stosw
        mov     ax, [last_ack]
        stosw
        mov     ecx, 512/4
        rep     movsd

        mcall   send, [socketnum], buffer, 4+512, 0       ; send data


.loop:

        mcall   23, TIMEOUT

        dec     eax
        jz      .red

        dec     eax
        jz      .key

        mcall   recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT  ; receive ack

        cmp     word[buffer], opcode_ack
        jne     .exit

        mov     ax, [last_ack]
        cmp     word[buffer+2], ax
        jne     .packet
        inc     [last_ack]
        test    [last_ack],0x001f
        jz      .read_chunk
        jmp     .packet


.red:

        call    draw_window

        jmp     .loop


.key:
        mcall   2
        cmp     ah, 2
        jz      exit

        ; close socket ?

        jmp     .loop

.exit:
        invoke  file_close, [fh]
        jmp     still





;-------------------------
; DATA

socketnum       dd 0
kbps            dd 0
done            dd 0

sockaddr:
                dw AF_INET4
                dw 0x4500       ; 69
IP              db 192, 168, 1, 115

sockaddr_len = $ - sockaddr

align 16
@IMPORT:

library box_lib , 'box_lib.obj', \
        io_lib  , 'libio.obj', \
        network , 'network.obj'

import  box_lib                                 ,\
        edit_box_draw    ,'edit_box'            ,\
        edit_box_key     ,'edit_box_key'        ,\
        edit_box_mouse   ,'edit_box_mouse'      ,\
        version_ed       ,'version_ed'          ,\
        init_checkbox    ,'init_checkbox2'      ,\
        check_box_draw   ,'check_box_draw2'     ,\
        check_box_mouse  ,'check_box_mouse2'    ,\
        version_ch       ,'version_ch2'         ,\
        option_box_draw  ,'option_box_draw'     ,\
        option_box_mouse ,'option_box_mouse'    ,\
        version_op       ,'version_op'

import  io_lib                                  ,\
        file_find_first , 'file_find_first'     ,\
        file_find_next  , 'file_find_next'      ,\
        file_find_close , 'file_find_close'     ,\
        file_size       , 'file_size'           ,\
        file_open       , 'file_open'           ,\
        file_read       , 'file_read'           ,\
        file_write      , 'file_write'          ,\
        file_seek       , 'file_seek'           ,\
        file_tell       , 'file_tell'           ,\
        file_eof?       , 'file_iseof'          ,\
        file_seteof     , 'file_seteof'         ,\
        file_truncate   , 'file_truncate'       ,\
        file_close      , 'file_close'

import  network                                 ,\
        inet_ntoa       , 'inet_ntoa'           ,\
        getaddrinfo     , 'getaddrinfo'         ,\
        freeaddrinfo    , 'freeaddrinfo'


edit1 edit_box 300, 80, 5, 0xffffff, 0x6f9480, 0, 0, 0, 99, SRV, mouse_dd, ed_focus, 13, 13
edit2 edit_box 300, 80, 25, 0xffffff, 0x6a9480, 0, 0, 0, 99, remote_addr, mouse_dd, ed_figure_only, 5, 5
edit3 edit_box 300, 80, 45, 0xffffff, 0x6a9480, 0, 0, 0, 99, local_addr, mouse_dd, ed_figure_only, 27, 27
edit4 edit_box 40, 340, 68, 0xffffff, 0x6a9480, 0, 0, 0, 5, BLK, mouse_dd, ed_figure_only, 3, 3

op1 option_box option_group1, 80, 68, 6, 12, 0xffffff, 0, 0, netascii, octet-netascii
op2 option_box option_group1, 80, 85, 6, 12, 0xFFFFFF, 0, 0, octet, get-octet

op3 option_box option_group2, 210, 68, 6, 12, 0xffffff, 0, 0, get, put-get
op4 option_box option_group2, 210, 85, 6, 12, 0xFFFFFF, 0, 0, put, BLK-put

option_group1   dd op1
option_group2   dd op3
Option_boxs1    dd op1, op2, 0
Option_boxs2    dd op3, op4, 0

str_title       db 'TFTP client', 0
str_server      db 'Server:', 0
str_source      db 'Remote file:', 0
str_destination db 'Local file:', 0
str_mode        db 'Mode:', 0
str_method      db 'Method:', 0
str_blocksize   db 'Blocksize:', 0
str_kb_s        db 'kB/s', 0
str_complete    db '% complete', 0
str_transfer    db 'Transfer', 0

str_error:
._0 db 'Not defined, see error message (if any).', 0
._1 db 'File not found.', 0
._2 db 'Access violation.', 0
._3 db 'Disk full or allocation exceeded.', 0
._4 db 'Illegal TFTP operation.', 0
._5 db 'Unknown transfer ID.', 0
._6 db 'File already exists.', 0
._7 db 'No such user.', 0


netascii db 'NetASCII'
octet    db 'Octet'
get      db 'GET'
put      db 'PUT'

BLK      db "512", 0, 0, 0

last_ack dw ?

fh       dd ?   ; file handle
fo       dd ?   ; file offset
fb       dd ?   ; file buffer

SRV             db "192.168.1.115", 0
                rb (SRV + 256 - $)

remote_addr     db "IMG00", 0
                rb (remote_addr + 256 - $)

local_addr      db "/hd0/1/KolibriOS/kernel.mnt", 0
                rb (local_addr + 256 - $)

I_END:

sc              system_colors

mouse_dd        dd ?
buffer          rb buffer_len

IM_END: