WebSVN – Kolibri OS – Diff – /programs/network/tftpc/tftpc.asm


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                                 ;;
;; Copyright (C) KolibriOS team 2010-2017. 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 ""
 
__DEBUG__       = 0
__DEBUG_LEVEL__ = 1
 
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 '../../develop/libraries/libs-dev/libio/libio.inc'
include '../../dll.inc'
include '../../develop/libraries/box_lib/trunk/box_lib.mac'
 
include '../../network.inc'
include '../../debug-fdo.inc'
 
TIMEOUT         = 100
buffer_len      = 4096
 
opcode_rrq      = 1 shl 8
opcode_wrq      = 2 shl 8
opcode_data     = 3 shl 8
opcode_ack      = 4 shl 8
opcode_error    = 5 shl 8
 
; 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
 
home:
        mcall   40, EVM_MOUSE + EVM_MOUSE_FILTER + EVM_REDRAW + EVM_BUTTON + EVM_KEY + EVM_STACK
 
redraw:
        call    draw_window
 
mainloop:
        mcall   10
        dec     eax
        jz      redraw
        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     mainloop
 
button:
        mcall   17
 
        cmp     ah, 0x10        ; connect button
        je      start_transfer
 
        test    ah , ah
        jz      mainloop
 
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     mainloop
 
 
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
 
        mcall   38, 10 shl 16 + 380, 130 shl 16 + 130, [sc.work_graph]
 
        cmp     [errormsg], 0
        je      .no_error
 
        mov     ecx, 0x80000000
        or      ecx, [sc.work_text]
        mcall   4, 20 shl 16 + 137, , [errormsg]
        mcall   12, 2
        jmp     .draw_btn
 
  .no_error:
        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   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
 
  .draw_btn:
        cmp     [socketnum], 0
        je      .no_transfer
 
        mov     ecx, 0x80000000
        or      ecx, [sc.work_button_text]
        mcall   4, 270 shl 16 + 110, , str_stop
 
        mcall   12, 2
        ret
 
  .no_transfer:
        mov     ecx, 0x80000000
        or      ecx, [sc.work_button_text]
        mcall   4, 260 shl 16 + 110, , str_transfer
 
        mcall   12, 2
        ret
 
 
 
start_transfer:
 
; resolve the hostname
        mov     [errormsg], str_err_resolve
 
        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     home
 
        mov     eax, [esi + addrinfo.ai_addr]
        mov     eax, [eax + sockaddr_in.sin_addr]
        mov     [sockaddr.ip], eax
 
        ; free allocated memory
        push    esi
        call    [freeaddrinfo]
 
; Open a socket & connect to server
        mov     [errormsg], str_err_socket
 
        mcall   socket, AF_INET4, SOCK_DGRAM, 0
        cmp     eax, -1
        je      home
        mov     [socketnum], eax
 
        mcall   connect, [socketnum], sockaddr, sockaddr_len
        cmp     eax, -1
        je      home
 
; Create the read/write request packet
        mov     word[buffer], opcode_rrq
        cmp     [option_group2], op3
        je      @f
        mov     word[buffer], opcode_wrq
      @@:
 
; Copy in the remote filename (asciiz)
        xor     al, al
        mov     edi, remote_addr
        mov     ecx, 255
        repnz   scasb
        lea     ecx, [edi - remote_addr - 1]
        mov     esi, remote_addr
        mov     edi, buffer+2
        rep     movsb
        stosb
 
; Append the data type (asciiz)
        cmp     [option_group1], op1
        je      .ascii
        mov     esi, octet
        movsd
        movsb
        jmp     .send_request
 
      .ascii:
        mov     esi, netascii
        movsd
        movsd
 
; Send the request to the server
      .send_request:
        xor     al, al
        stosb
        lea     esi, [edi - buffer]
        xor     edi, edi
        mcall   send, [socketnum], buffer
        cmp     eax, -1
        je      home
 
; Jump to send/receive code
        cmp     word[buffer], opcode_wrq
        je      tftp_send
 
 
tftp_receive:
 
        mcall   40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
        mov     [last_ack], 0
        mov     [errormsg], 0
 
        call    draw_window
 
; Open/create local file
        mov     [file_struct.subfn], 2
        mov     [file_struct.offset], 0
        mov     [file_struct.size], 0
        mcall   70, file_struct
 
; Truncate it to 0 bytes
        mov     [file_struct.subfn], 4
        mcall   70, file_struct
 
; Set parameters for writing to file
        mov     [file_struct.subfn], 3
        mov     [file_struct.data], buffer + 4
 
  .loop:
        mcall   23, TIMEOUT
        dec     eax
        jz      .red
        dec     eax
        dec     eax
        jz      .button
 
        mcall   recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT     ; receive data
        cmp     eax, -1
        je      .loop
 
        DEBUGF  1, "Got %u bytes\n", eax
        cmp     word[buffer], opcode_error
        je      tftp_error
        cmp     word[buffer], opcode_data
        jne     .error
 
; Verify ACK number
        mov     bx, word[buffer + 2]
        rol     bx, 8
        cmp     [last_ack], 0
        je      @f
        cmp     [last_ack], bx
        jne     .packet_got_lost
        inc     bx
      @@:
        mov     [last_ack], bx
 
; Write to the file
        lea     ecx, [eax - 4]
        mov     [file_struct.size], ecx
        mcall   70, file_struct
        add     [file_struct.offset], ecx
 
; Send ACK
        mov     word[buffer], opcode_ack
        mcall   send, [socketnum], buffer, 4, 0
        jmp     .loop
 
  .packet_got_lost:
        ;TODO
        jmp     .loop
 
  .red:
        call    draw_window
        jmp     .loop
 
  .button:
        mcall   17
        cmp     ah, 1
        jne     .abort
 
        mcall   close, [socketnum]
        mcall   -1
 
  .abort:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_abort
        jmp     home
 
  .error:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_err_unexp
        jmp     home
 
  .done:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_success
        jmp     home
 
 
 
tftp_send:
 
        mcall   40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
        mov     [last_ack], 0
        mov     [errormsg], 0
 
        call    draw_window
 
        mov     [file_struct.subfn], 0
        mov     [file_struct.offset], 0
        mov     [file_struct.size], buffer_len
        mov     [file_struct.data], buffer + 4
 
  .next:
        mov     edi, buffer
        mov     ax, opcode_data
        stosw
        mov     ax, [last_ack]
        stosw
 
        mcall   70, file_struct
        test    eax, eax
;        jnz     .done
        mov     [size], ebx
 
  .resend:
        mov     ebx, [size]
        lea     esi, [ebx + 4]
        xor     edi, edi
        mcall   send, [socketnum], buffer
 
  .loop:
        mcall   23, TIMEOUT
        dec     eax
        jz      .red
        dec     eax
        dec     eax
        jz      .button
 
        mcall   recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT
        cmp     eax, -1
        je      .loop
 
        cmp     word[buffer], opcode_error
        je      tftp_error
        cmp     word[buffer], opcode_ack
        jne     .error
 
        mov     ax, [last_ack]
        cmp     word[buffer+2], ax
        jne     .resend
 
        mov     eax, [size]
        cmp     eax, buffer_len
        jb      .done
        add     [file_struct.offset], eax
 
        inc     [last_ack]
        jmp     .next
 
  .red:
        call    draw_window
        jmp     .loop
 
  .button:
        mcall   17
        cmp     ah, 1
        jne     .abort
 
        mcall   close, [socketnum]
        mcall   -1
 
  .abort:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_abort
        jmp     home
 
  .error:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_err_unexp
        jmp     home
 
  .done:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
        mov     [errormsg], str_success
        jmp     home
 
 
 
tftp_error:
        mcall   close, [socketnum]
        xor     eax, eax
        mov     [socketnum], eax
 
        mov     ax, word[buffer+2]
        xchg    al, ah
 
        test    ax, ax
        jz      .0
        dec     ax
        jz      .1
        dec     ax
        jz      .2
        dec     ax
        jz      .3
        dec     ax
        jz      .4
        dec     ax
        jz      .5
        dec     ax
        jz      .6
        dec     ax
        jz      .7
 
  .0:
        mov     [errormsg], str_error.0
        jmp     home
  .1:
        mov     [errormsg], str_error.1
        jmp     redraw
  .2:
        mov     [errormsg], str_error.2
        jmp     home
  .3:
        mov     [errormsg], str_error.3
        jmp     home
  .4:
        mov     [errormsg], str_error.4
        jmp     home
  .5:
        mov     [errormsg], str_error.5
        jmp     home
  .6:
        mov     [errormsg], str_error.6
        jmp     home
  .7:
        mov     [errormsg], str_error.7
        jmp     home
 
;-------------------------
; DATA
 
socketnum       dd 0
kbps            dd 0
done            dd 0
errormsg        dd str_welcome
 
sockaddr:
                dw AF_INET4
                dw 0x4500       ; 69
  .ip           dd ?
sockaddr_len = $ - sockaddr
 
file_struct:
  .subfn        dd ?
  .offset       dd ?
                dd 0
  .size         dd ?
  .data         dd ?
                db 0
  .filename     dd local_addr
 
align 16
@IMPORT:
 
library box_lib         , 'box_lib.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  network                                 ,\
        inet_ntoa       , 'inet_ntoa'           ,\
        getaddrinfo     , 'getaddrinfo'         ,\
        freeaddrinfo    , 'freeaddrinfo'
 
 
edit1 edit_box 300, 80, 5, 0xffffff, 0x6f9480, 0, 0, 0, 255, SRV, mouse_dd, ed_focus, 13, 13
edit2 edit_box 300, 80, 25, 0xffffff, 0x6f9480, 0, 0, 0, 255, remote_addr, mouse_dd, 0, 4, 4
edit3 edit_box 300, 80, 45, 0xffffff, 0x6f9480, 0, 0, 0, 255, local_addr, mouse_dd, 0, 12, 12
edit4 edit_box 40, 340, 68, 0xffffff, 0x6f9480, 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_stop        db 'Stop', 0
 
str_error:
.0              db 'Error.', 0                      ; not further defined error
.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
 
str_welcome     db 'Welcome.', 0
str_err_resolve db 'Unable to resolve server address.', 0
str_err_socket  db 'Socket error.', 0
str_err_unexp   db 'Unexpected command from server.', 0
str_success     db 'Operation completed successfully.', 0
str_abort       db 'Operation aborted by user.', 0
 
netascii        db 'NetASCII'
octet           db 'Octet'
get             db 'GET'
put             db 'PUT'
 
BLK             db "512", 0, 0, 0
 
SRV             db "192.168.1.115", 0
                rb (SRV + 256 - $)
 
remote_addr     db "file", 0
                rb (remote_addr + 256 - $)
 
local_addr      db "/tmp0/1/file", 0
                rb (local_addr + 256 - $)
 
include_debug_strings
 
I_END:
 
last_ack        dw ?
size            dd ?
mouse_dd        dd ?
 
sc              system_colors
 
buffer          rb buffer_len
 
IM_END:

Rev 4655	Rev 6836
1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2	;; ;;	2	;; ;;
3	;; Copyright (C) KolibriOS team 2010-2014. All rights reserved. ;;	3	;; Copyright (C) KolibriOS team 2010-2017. All rights reserved. ;;
4	;; Distributed under terms of the GNU General Public License ;;	4	;; Distributed under terms of the GNU General Public License ;;
5	;; ;;	5	;; ;;
6	;; tftpc.asm - TFTP client for KolibriOS ;;	6	;; tftpc.asm - TFTP client for KolibriOS ;;
7	;; ;;	7	;; ;;
8	;; Written by hidnplayr@kolibrios.org ;;	8	;; Written by hidnplayr@kolibrios.org ;;
9	;; ;;	9	;; ;;
10	;; GNU GENERAL PUBLIC LICENSE ;;	10	;; GNU GENERAL PUBLIC LICENSE ;;
11	;; Version 2, June 1991 ;;	11	;; Version 2, June 1991 ;;
12	;; ;;	12	;; ;;
13	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	13	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
14		14
15	format binary as ""	15	format binary as ""
16		16
17	__DEBUG__ = 0	17	__DEBUG__ = 0
18	__DEBUG_LEVEL__ = 1	18	__DEBUG_LEVEL__ = 1
19		19
20	use32	20	use32
21	org 0x0	21	org 0x0
22		22
23	db 'MENUET01'	23	db 'MENUET01'
24	dd 0x1	24	dd 0x1
25	dd START	25	dd START
26	dd I_END	26	dd I_END
27	dd IM_END+0x1000	27	dd IM_END+0x1000
28	dd IM_END+0x1000	28	dd IM_END+0x1000
29	dd 0, 0	29	dd 0, 0
30		30
31	include '../../proc32.inc'	31	include '../../proc32.inc'
32	include '../../macros.inc'	32	include '../../macros.inc'
33	include '../../libio.inc'	33	include '../../develop/libraries/libs-dev/libio/libio.inc'
34	include '../../dll.inc'	34	include '../../dll.inc'
35	include '../../develop/libraries/box_lib/trunk/box_lib.mac'	35	include '../../develop/libraries/box_lib/trunk/box_lib.mac'
36		36
37	include '../../network.inc'	37	include '../../network.inc'
38	include '../../debug-fdo.inc'	38	include '../../debug-fdo.inc'
39		39
40	TIMEOUT = 100	40	TIMEOUT = 100
41	buffer_len = 4096	41	buffer_len = 4096
42		42
43	opcode_rrq = 1 shl 8	43	opcode_rrq = 1 shl 8
44	opcode_wrq = 2 shl 8	44	opcode_wrq = 2 shl 8
45	opcode_data = 3 shl 8	45	opcode_data = 3 shl 8
46	opcode_ack = 4 shl 8	46	opcode_ack = 4 shl 8
47	opcode_error = 5 shl 8	47	opcode_error = 5 shl 8
48		48
49	; read/write request packet	49	; read/write request packet
50	;	50	;
51	; 2 bytes string 1 byte string 1 byte	51	; 2 bytes string 1 byte string 1 byte
52	; ------------------------------------------------	52	; ------------------------------------------------
53	; \| Opcode \| Filename \| 0 \| Mode \| 0 \|	53	; \| Opcode \| Filename \| 0 \| Mode \| 0 \|
54	; ------------------------------------------------	54	; ------------------------------------------------
55		55
56	; data packet	56	; data packet
57	;	57	;
58	; 2 bytes 2 bytes n bytes	58	; 2 bytes 2 bytes n bytes
59	; ----------------------------------	59	; ----------------------------------
60	; \| Opcode \| Block # \| Data \|	60	; \| Opcode \| Block # \| Data \|
61	; ----------------------------------	61	; ----------------------------------
62		62
63	; acknowledgement packet	63	; acknowledgement packet
64	;	64	;
65	; 2 bytes 2 bytes	65	; 2 bytes 2 bytes
66	; ---------------------	66	; ---------------------
67	; \| Opcode \| Block # \|	67	; \| Opcode \| Block # \|
68	; ---------------------	68	; ---------------------
69		69
70	; error packet	70	; error packet
71	;	71	;
72	; 2 bytes 2 bytes string 1 byte	72	; 2 bytes 2 bytes string 1 byte
73	; ----------------------------------------	73	; ----------------------------------------
74	; \| Opcode \| ErrorCode \| ErrMsg \| 0 \|	74	; \| Opcode \| ErrorCode \| ErrMsg \| 0 \|
75	; ----------------------------------------	75	; ----------------------------------------
76		76
77		77
78	START:	78	START:
79		79
80	mcall 68, 11	80	mcall 68, 11
81		81
82	stdcall dll.Load, @IMPORT	82	stdcall dll.Load, @IMPORT
83	or eax, eax	83	or eax, eax
84	jnz exit	84	jnz exit
85		85
86	home:	86	home:
87	mcall 40, EVM_MOUSE + EVM_MOUSE_FILTER + EVM_REDRAW + EVM_BUTTON + EVM_KEY + EVM_STACK	87	mcall 40, EVM_MOUSE + EVM_MOUSE_FILTER + EVM_REDRAW + EVM_BUTTON + EVM_KEY + EVM_STACK
88		88
89	redraw:	89	redraw:
90	call draw_window	90	call draw_window
91		91
92	mainloop:	92	mainloop:
93	mcall 10	93	mcall 10
94	dec eax	94	dec eax
95	jz redraw	95	jz redraw
96	dec eax	96	dec eax
97	jz key	97	jz key
98	dec eax	98	dec eax
99	jz button	99	jz button
100		100
101	invoke edit_box_mouse, edit1	101	invoke edit_box_mouse, edit1
102	invoke edit_box_mouse, edit2	102	invoke edit_box_mouse, edit2
103	invoke edit_box_mouse, edit3	103	invoke edit_box_mouse, edit3
104	invoke edit_box_mouse, edit4	104	invoke edit_box_mouse, edit4
105		105
106	invoke option_box_mouse, Option_boxs1	106	invoke option_box_mouse, Option_boxs1
107	invoke option_box_mouse, Option_boxs2	107	invoke option_box_mouse, Option_boxs2
108		108
109	jmp mainloop	109	jmp mainloop
110		110
111	button:	111	button:
112	mcall 17	112	mcall 17
113		113
114	cmp ah, 0x10 ; connect button	114	cmp ah, 0x10 ; connect button
115	je start_transfer	115	je start_transfer
116		116
117	test ah , ah	117	test ah , ah
118	jz mainloop	118	jz mainloop
119		119
120	exit:	120	exit:
121	mcall -1	121	mcall -1
122	key:	122	key:
123	mcall 2	123	mcall 2
124		124
125	invoke edit_box_key, edit1	125	invoke edit_box_key, edit1
126	invoke edit_box_key, edit2	126	invoke edit_box_key, edit2
127	invoke edit_box_key, edit3	127	invoke edit_box_key, edit3
128	invoke edit_box_key, edit4	128	invoke edit_box_key, edit4
129		129
130	jmp mainloop	130	jmp mainloop
131		131
132		132
133	draw_window:	133	draw_window:
134	; get system colors	134	; get system colors
135	mcall 48, 3, sc, 40	135	mcall 48, 3, sc, 40
136		136
137	mcall 12, 1	137	mcall 12, 1
138		138
139	mov edx, [sc.work]	139	mov edx, [sc.work]
140	or edx, 0x34000000	140	or edx, 0x34000000
141	xor esi, esi	141	xor esi, esi
142	mov edi, str_title	142	mov edi, str_title
143	mcall 0, 50 shl 16 + 400, 30 shl 16 + 180	143	mcall 0, 50 shl 16 + 400, 30 shl 16 + 180
144		144
145	mov ebx, 35 shl 16 + 10	145	mov ebx, 35 shl 16 + 10
146	mov ecx, 0x80000000	146	mov ecx, 0x80000000
147	or ecx, [sc.work_text]	147	or ecx, [sc.work_text]
148	mov edx, str_server	148	mov edx, str_server
149	mcall 4	149	mcall 4
150	mov ebx, 5 shl 16 + 30	150	mov ebx, 5 shl 16 + 30
151	mov edx, str_source	151	mov edx, str_source
152	mcall	152	mcall
153	mov ebx, 11 shl 16 + 50	153	mov ebx, 11 shl 16 + 50
154	mov edx, str_destination	154	mov edx, str_destination
155	mcall	155	mcall
156	mov ebx, 47 shl 16 + 72	156	mov ebx, 47 shl 16 + 72
157	mov edx, str_mode	157	mov edx, str_mode
158	mcall	158	mcall
159	mov ebx, 160 shl 16 + 72	159	mov ebx, 160 shl 16 + 72
160	mov edx, str_method	160	mov edx, str_method
161	mcall	161	mcall
162	mov ebx, 270 shl 16 + 72	162	mov ebx, 270 shl 16 + 72
163	mov edx, str_blocksize	163	mov edx, str_blocksize
164	mcall	164	mcall
165		165
166	invoke edit_box_draw, edit1	166	invoke edit_box_draw, edit1
167	invoke edit_box_draw, edit2	167	invoke edit_box_draw, edit2
168	invoke edit_box_draw, edit3	168	invoke edit_box_draw, edit3
169	invoke edit_box_draw, edit4	169	invoke edit_box_draw, edit4
170		170
171	invoke option_box_draw, Option_boxs1	171	invoke option_box_draw, Option_boxs1
172	invoke option_box_draw, Option_boxs2	172	invoke option_box_draw, Option_boxs2
173		173
174	mov esi, [sc.work_button]	174	mov esi, [sc.work_button]
175	mcall 8, 210 shl 16 + 170, 105 shl 16 + 16, 0x10	175	mcall 8, 210 shl 16 + 170, 105 shl 16 + 16, 0x10
176		176
177	mcall 38, 10 shl 16 + 380, 130 shl 16 + 130, [sc.work_graph]	177	mcall 38, 10 shl 16 + 380, 130 shl 16 + 130, [sc.work_graph]
178		178
179	cmp [errormsg], 0	179	cmp [errormsg], 0
180	je .no_error	180	je .no_error
181		181
182	mov ecx, 0x80000000	182	mov ecx, 0x80000000
183	or ecx, [sc.work_text]	183	or ecx, [sc.work_text]
184	mcall 4, 20 shl 16 + 137, , [errormsg]	184	mcall 4, 20 shl 16 + 137, , [errormsg]
185	mcall 12, 2	185	mcall 12, 2
186	jmp .draw_btn	186	jmp .draw_btn
187		187
188	.no_error:	188	.no_error:
189	mov ecx, 0x80000000	189	mov ecx, 0x80000000
190	or ecx, [sc.work_text]	190	or ecx, [sc.work_text]
191	mcall 4, 350 shl 16 + 137, , str_kb_s	191	mcall 4, 350 shl 16 + 137, , str_kb_s
192	mcall 4, 50 shl 16 + 137, , str_complete	192	mcall 4, 50 shl 16 + 137, , str_complete
193	mcall 47, 1 shl 31 + 7 shl 16 + 1, kbps, 305 shl 16 + 137, [sc.work_text]	193	mcall 47, 1 shl 31 + 7 shl 16 + 1, kbps, 305 shl 16 + 137, [sc.work_text]
194	mcall 47, 1 shl 31 + 3 shl 16 + 1, done, 25 shl 16 + 137	194	mcall 47, 1 shl 31 + 3 shl 16 + 1, done, 25 shl 16 + 137
195		195
196	.draw_btn:	196	.draw_btn:
197	cmp [socketnum], 0	197	cmp [socketnum], 0
198	je .no_transfer	198	je .no_transfer
199		199
200	mov ecx, 0x80000000	200	mov ecx, 0x80000000
201	or ecx, [sc.work_button_text]	201	or ecx, [sc.work_button_text]
202	mcall 4, 270 shl 16 + 110, , str_stop	202	mcall 4, 270 shl 16 + 110, , str_stop
203		203
204	mcall 12, 2	204	mcall 12, 2
205	ret	205	ret
206		206
207	.no_transfer:	207	.no_transfer:
208	mov ecx, 0x80000000	208	mov ecx, 0x80000000
209	or ecx, [sc.work_button_text]	209	or ecx, [sc.work_button_text]
210	mcall 4, 260 shl 16 + 110, , str_transfer	210	mcall 4, 260 shl 16 + 110, , str_transfer
211		211
212	mcall 12, 2	212	mcall 12, 2
213	ret	213	ret
214		214
215		215
216		216
217	start_transfer:	217	start_transfer:
218		218
219	; resolve the hostname	219	; resolve the hostname
220	mov [errormsg], str_err_resolve	220	mov [errormsg], str_err_resolve
221		221
222	push esp ; reserve stack place	222	push esp ; reserve stack place
223		223
224	push esp ; fourth parameter	224	push esp ; fourth parameter
225	push 0 ; third parameter	225	push 0 ; third parameter
226	push 0 ; second parameter	226	push 0 ; second parameter
227	push dword SRV ; first parameter	227	push dword SRV ; first parameter
228	call [getaddrinfo]	228	call [getaddrinfo]
229		229
230	pop esi	230	pop esi
231		231
232	; test for error	232	; test for error
233	test eax, eax	233	test eax, eax
234	jnz home	234	jnz home
235		235
236	mov eax, [esi + addrinfo.ai_addr]	236	mov eax, [esi + addrinfo.ai_addr]
237	mov eax, [eax + sockaddr_in.sin_addr]	237	mov eax, [eax + sockaddr_in.sin_addr]
238	mov [sockaddr.ip], eax	238	mov [sockaddr.ip], eax
239		239
240	; free allocated memory	240	; free allocated memory
241	push esi	241	push esi
242	call [freeaddrinfo]	242	call [freeaddrinfo]
243		243
244	; Open a socket & connect to server	244	; Open a socket & connect to server
245	mov [errormsg], str_err_socket	245	mov [errormsg], str_err_socket
246		246
247	mcall socket, AF_INET4, SOCK_DGRAM, 0	247	mcall socket, AF_INET4, SOCK_DGRAM, 0
248	cmp eax, -1	248	cmp eax, -1
249	je home	249	je home
250	mov [socketnum], eax	250	mov [socketnum], eax
251		251
252	mcall connect, [socketnum], sockaddr, sockaddr_len	252	mcall connect, [socketnum], sockaddr, sockaddr_len
253	cmp eax, -1	253	cmp eax, -1
254	je home	254	je home
255		255
256	; Create the read/write request packet	256	; Create the read/write request packet
257	mov word[buffer], opcode_rrq	257	mov word[buffer], opcode_rrq
258	cmp [option_group2], op3	258	cmp [option_group2], op3
259	je @f	259	je @f
260	mov word[buffer], opcode_wrq	260	mov word[buffer], opcode_wrq
261	@@:	261	@@:
262		262
263	; Copy in the remote filename (asciiz)	263	; Copy in the remote filename (asciiz)
264	xor al, al	264	xor al, al
265	mov edi, remote_addr	265	mov edi, remote_addr
266	mov ecx, 255	266	mov ecx, 255
267	repnz scasb	267	repnz scasb
268	lea ecx, [edi - remote_addr - 1]	268	lea ecx, [edi - remote_addr - 1]
269	mov esi, remote_addr	269	mov esi, remote_addr
270	mov edi, buffer+2	270	mov edi, buffer+2
271	rep movsb	271	rep movsb
272	stosb	272	stosb
273		273
274	; Append the data type (asciiz)	274	; Append the data type (asciiz)
275	cmp [option_group1], op1	275	cmp [option_group1], op1
276	je .ascii	276	je .ascii
277	mov esi, octet	277	mov esi, octet
278	movsd	278	movsd
279	movsb	279	movsb
280	jmp .send_request	280	jmp .send_request
281		281
282	.ascii:	282	.ascii:
283	mov esi, netascii	283	mov esi, netascii
284	movsd	284	movsd
285	movsd	285	movsd
286		286
287	; Send the request to the server	287	; Send the request to the server
288	.send_request:	288	.send_request:
289	xor al, al	289	xor al, al
290	stosb	290	stosb
291	lea esi, [edi - buffer]	291	lea esi, [edi - buffer]
292	xor edi, edi	292	xor edi, edi
293	mcall send, [socketnum], buffer	293	mcall send, [socketnum], buffer
294	cmp eax, -1	294	cmp eax, -1
295	je home	295	je home
296		296
297	; Jump to send/receive code	297	; Jump to send/receive code
298	cmp word[buffer], opcode_wrq	298	cmp word[buffer], opcode_wrq
299	je tftp_send	299	je tftp_send
300		300
301		301
302	tftp_receive:	302	tftp_receive:
303		303
304	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK	304	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
305	mov [last_ack], 0	305	mov [last_ack], 0
306	mov [errormsg], 0	306	mov [errormsg], 0
307		307
308	call draw_window	308	call draw_window
309		309
310	; Open/create local file	310	; Open/create local file
311	mov [file_struct.subfn], 2	311	mov [file_struct.subfn], 2
312	mov [file_struct.offset], 0	312	mov [file_struct.offset], 0
313	mov [file_struct.size], 0	313	mov [file_struct.size], 0
314	mcall 70, file_struct	314	mcall 70, file_struct
315		315
316	; Truncate it to 0 bytes	316	; Truncate it to 0 bytes
317	mov [file_struct.subfn], 4	317	mov [file_struct.subfn], 4
318	mcall 70, file_struct	318	mcall 70, file_struct
319		319
320	; Set parameters for writing to file	320	; Set parameters for writing to file
321	mov [file_struct.subfn], 3	321	mov [file_struct.subfn], 3
322	mov [file_struct.data], buffer + 4	322	mov [file_struct.data], buffer + 4
323		323
324	.loop:	324	.loop:
325	mcall 23, TIMEOUT	325	mcall 23, TIMEOUT
326	dec eax	326	dec eax
327	jz .red	327	jz .red
328	dec eax	328	dec eax
329	dec eax	329	dec eax
330	jz .button	330	jz .button
331		331
332	mcall recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT ; receive data	332	mcall recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT ; receive data
333	cmp eax, -1	333	cmp eax, -1
334	je .loop	334	je .loop
335		335
336	DEBUGF 1, "Got %u bytes\n", eax	336	DEBUGF 1, "Got %u bytes\n", eax
337	cmp word[buffer], opcode_error	337	cmp word[buffer], opcode_error
338	je tftp_error	338	je tftp_error
339	cmp word[buffer], opcode_data	339	cmp word[buffer], opcode_data
340	jne .error	340	jne .error
341		341
342	; Verify ACK number	342	; Verify ACK number
343	mov bx, word[buffer + 2]	343	mov bx, word[buffer + 2]
344	rol bx, 8	344	rol bx, 8
345	cmp [last_ack], 0	345	cmp [last_ack], 0
346	je @f	346	je @f
347	cmp [last_ack], bx	347	cmp [last_ack], bx
348	jne .packet_got_lost	348	jne .packet_got_lost
349	inc bx	349	inc bx
350	@@:	350	@@:
351	mov [last_ack], bx	351	mov [last_ack], bx
352		352
353	; Write to the file	353	; Write to the file
354	lea ecx, [eax - 4]	354	lea ecx, [eax - 4]
355	mov [file_struct.size], ecx	355	mov [file_struct.size], ecx
356	mcall 70, file_struct	356	mcall 70, file_struct
357	add [file_struct.offset], ecx	357	add [file_struct.offset], ecx
358		358
359	; Send ACK	359	; Send ACK
360	mov word[buffer], opcode_ack	360	mov word[buffer], opcode_ack
361	mcall send, [socketnum], buffer, 4, 0	361	mcall send, [socketnum], buffer, 4, 0
362	jmp .loop	362	jmp .loop
363		363
364	.packet_got_lost:	364	.packet_got_lost:
365	;TODO	365	;TODO
366	jmp .loop	366	jmp .loop
367		367
368	.red:	368	.red:
369	call draw_window	369	call draw_window
370	jmp .loop	370	jmp .loop
371		371
372	.button:	372	.button:
373	mcall 17	373	mcall 17
374	cmp ah, 1	374	cmp ah, 1
375	jne .abort	375	jne .abort
376		376
377	mcall close, [socketnum]	377	mcall close, [socketnum]
378	mcall -1	378	mcall -1
379		379
380	.abort:	380	.abort:
381	mcall close, [socketnum]	381	mcall close, [socketnum]
382	xor eax, eax	382	xor eax, eax
383	mov [socketnum], eax	383	mov [socketnum], eax
384	mov [errormsg], str_abort	384	mov [errormsg], str_abort
385	jmp home	385	jmp home
386		386
387	.error:	387	.error:
388	mcall close, [socketnum]	388	mcall close, [socketnum]
389	xor eax, eax	389	xor eax, eax
390	mov [socketnum], eax	390	mov [socketnum], eax
391	mov [errormsg], str_err_unexp	391	mov [errormsg], str_err_unexp
392	jmp home	392	jmp home
393		393
394	.done:	394	.done:
395	mcall close, [socketnum]	395	mcall close, [socketnum]
396	xor eax, eax	396	xor eax, eax
397	mov [socketnum], eax	397	mov [socketnum], eax
398	mov [errormsg], str_success	398	mov [errormsg], str_success
399	jmp home	399	jmp home
400		400
401		401
402		402
403	tftp_send:	403	tftp_send:
404		404
405	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK	405	mcall 40, EVM_REDRAW + EVM_BUTTON + EVM_STACK
406	mov [last_ack], 0	406	mov [last_ack], 0
407	mov [errormsg], 0	407	mov [errormsg], 0
408		408
409	call draw_window	409	call draw_window
410		410
411	mov [file_struct.subfn], 0	411	mov [file_struct.subfn], 0
412	mov [file_struct.offset], 0	412	mov [file_struct.offset], 0
413	mov [file_struct.size], buffer_len	413	mov [file_struct.size], buffer_len
414	mov [file_struct.data], buffer + 4	414	mov [file_struct.data], buffer + 4
415		415
416	.next:	416	.next:
417	mov edi, buffer	417	mov edi, buffer
418	mov ax, opcode_data	418	mov ax, opcode_data
419	stosw	419	stosw
420	mov ax, [last_ack]	420	mov ax, [last_ack]
421	stosw	421	stosw
422		422
423	mcall 70, file_struct	423	mcall 70, file_struct
424	test eax, eax	424	test eax, eax
425	; jnz .done	425	; jnz .done
426	mov [size], ebx	426	mov [size], ebx
427		427
428	.resend:	428	.resend:
429	mov ebx, [size]	429	mov ebx, [size]
430	lea esi, [ebx + 4]	430	lea esi, [ebx + 4]
431	xor edi, edi	431	xor edi, edi
432	mcall send, [socketnum], buffer	432	mcall send, [socketnum], buffer
433		433
434	.loop:	434	.loop:
435	mcall 23, TIMEOUT	435	mcall 23, TIMEOUT
436	dec eax	436	dec eax
437	jz .red	437	jz .red
438	dec eax	438	dec eax
439	dec eax	439	dec eax
440	jz .button	440	jz .button
441		441
442	mcall recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT	442	mcall recv, [socketnum], buffer, buffer_len, MSG_DONTWAIT
443	cmp eax, -1	443	cmp eax, -1
444	je .loop	444	je .loop
445		445
446	cmp word[buffer], opcode_error	446	cmp word[buffer], opcode_error
447	je tftp_error	447	je tftp_error
448	cmp word[buffer], opcode_ack	448	cmp word[buffer], opcode_ack
449	jne .error	449	jne .error
450		450
451	mov ax, [last_ack]	451	mov ax, [last_ack]
452	cmp word[buffer+2], ax	452	cmp word[buffer+2], ax
453	jne .resend	453	jne .resend
454		454
455	mov eax, [size]	455	mov eax, [size]
456	cmp eax, buffer_len	456	cmp eax, buffer_len
457	jb .done	457	jb .done
458	add [file_struct.offset], eax	458	add [file_struct.offset], eax
459		459
460	inc [last_ack]	460	inc [last_ack]
461	jmp .next	461	jmp .next
462		462
463	.red:	463	.red:
464	call draw_window	464	call draw_window
465	jmp .loop	465	jmp .loop
466		466
467	.button:	467	.button:
468	mcall 17	468	mcall 17
469	cmp ah, 1	469	cmp ah, 1
470	jne .abort	470	jne .abort
471		471
472	mcall close, [socketnum]	472	mcall close, [socketnum]
473	mcall -1	473	mcall -1
474		474
475	.abort:	475	.abort:
476	mcall close, [socketnum]	476	mcall close, [socketnum]
477	xor eax, eax	477	xor eax, eax
478	mov [socketnum], eax	478	mov [socketnum], eax
479	mov [errormsg], str_abort	479	mov [errormsg], str_abort
480	jmp home	480	jmp home
481		481
482	.error:	482	.error:
483	mcall close, [socketnum]	483	mcall close, [socketnum]
484	xor eax, eax	484	xor eax, eax
485	mov [socketnum], eax	485	mov [socketnum], eax
486	mov [errormsg], str_err_unexp	486	mov [errormsg], str_err_unexp
487	jmp home	487	jmp home
488		488
489	.done:	489	.done:
490	mcall close, [socketnum]	490	mcall close, [socketnum]
491	xor eax, eax	491	xor eax, eax
492	mov [socketnum], eax	492	mov [socketnum], eax
493	mov [errormsg], str_success	493	mov [errormsg], str_success
494	jmp home	494	jmp home
495		495
496		496
497		497
498	tftp_error:	498	tftp_error:
499	mcall close, [socketnum]	499	mcall close, [socketnum]
500	xor eax, eax	500	xor eax, eax

Subversion Repositories Kolibri OS

(root)/programs/network/tftpc/tftpc.asm – Rev 4655 → 6836