forked from KolibriOS/kolibrios
db3d073d65
More fixes in UDP added TFTP client git-svn-id: svn://kolibrios.org@1337 a494cfbc-eb01-0410-851d-a64ba20cac60
636 lines
11 KiB
NASM
636 lines
11 KiB
NASM
use32
|
|
|
|
org 0x0
|
|
|
|
db 'MENUET01'
|
|
dd 1, START, I_END, IM_END+0x1000, IM_END+0x1000, 0, 0
|
|
|
|
include '../proc32.inc'
|
|
include '../macros.inc'
|
|
include '../dll.inc'
|
|
include '../libio.inc'
|
|
include 'editbox_ex.mac'
|
|
|
|
include '../network.inc'
|
|
|
|
|
|
filebuffer_size equ 4*4096 ; 16kb (dont try to change it yet..)
|
|
TIMEOUT equ 500
|
|
buffer_len equ 1500
|
|
|
|
AF_INET4 equ 2
|
|
IP_PROTO_UDP equ 17
|
|
|
|
opcode_rrq equ 1 shl 8
|
|
opcode_wrq equ 2 shl 8
|
|
opcode_data equ 3 shl 8
|
|
opcode_ack equ 4 shl 8
|
|
opcode_error equ 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
|
|
|
|
stop_transfer:
|
|
mcall 40, 00100111b
|
|
|
|
red_win:
|
|
call draw_window
|
|
|
|
align 4
|
|
still:
|
|
mcall 10
|
|
|
|
dec eax
|
|
jz red_win
|
|
|
|
dec eax
|
|
jz key
|
|
|
|
dec eax
|
|
jz button
|
|
|
|
push dword edit1
|
|
call [edit_box_mouse]
|
|
|
|
push dword edit2
|
|
call [edit_box_mouse]
|
|
|
|
push dword edit3
|
|
call [edit_box_mouse]
|
|
|
|
push dword edit4
|
|
call [edit_box_mouse]
|
|
|
|
push dword Option_boxs1
|
|
call [option_box_mouse]
|
|
|
|
push dword Option_boxs2
|
|
call [option_box_mouse]
|
|
|
|
jmp still
|
|
|
|
button:
|
|
mcall 17
|
|
|
|
cmp ah,0x10
|
|
je start_transfer
|
|
|
|
|
|
test ah , ah
|
|
jz still
|
|
|
|
exit: mcall -1
|
|
key:
|
|
mcall 2
|
|
|
|
push dword edit1
|
|
call [edit_box_key]
|
|
|
|
push dword edit2
|
|
call [edit_box_key]
|
|
|
|
push dword edit3
|
|
call [edit_box_key]
|
|
|
|
push dword edit4
|
|
call [edit_box_key]
|
|
|
|
jmp still
|
|
|
|
|
|
align 4
|
|
draw_window:
|
|
mcall 12,1
|
|
mcall 0,(50*65536+400),(30*65536+180),0x34AABBCC,0x085080DD,str_title
|
|
|
|
mcall 4,35*65536+10,0x80000000, str_server
|
|
|
|
mov ebx,5*65536+ 30
|
|
mov edx, str_source
|
|
mcall
|
|
|
|
mov ebx,11*65536+ 50
|
|
mov edx, str_destination
|
|
mcall
|
|
|
|
mov ebx,47*65536+72
|
|
mov edx, str_mode
|
|
mcall
|
|
|
|
mov ebx,160*65536+72
|
|
mov edx, str_method
|
|
mcall
|
|
|
|
mov ebx,270*65536+72
|
|
mov edx, str_blocksize
|
|
mcall
|
|
|
|
push dword edit1
|
|
call [edit_box_draw]
|
|
|
|
push dword edit2
|
|
call [edit_box_draw]
|
|
|
|
push dword edit3
|
|
call [edit_box_draw]
|
|
|
|
push dword edit4
|
|
call [edit_box_draw]
|
|
|
|
push dword Option_boxs1
|
|
call [option_box_draw]
|
|
|
|
push dword Option_boxs2
|
|
call [option_box_draw]
|
|
|
|
mcall 8,210*65536+170, 105*65536+16,0x00000010,0x085080DD
|
|
|
|
mcall 4,260*65536+110, 0x80000000, str_transfer
|
|
|
|
mcall 38,10*65536+380, 130*65536+130,0x00000000
|
|
|
|
mcall 4,350*65536+137, 0x80000000, str_kb_s
|
|
|
|
; mcall 47,1 shl 31 + 7 shl 16 + 1,kbps,305*65536+137,0x00000000
|
|
|
|
mcall 4,20*65536+137, 0x80000000, [status]
|
|
|
|
; mcall 47,1 shl 31 + 3 shl 16 + 1,done,25*65536+137,0x00000000
|
|
|
|
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 SRV ; first parameter
|
|
call [getaddrinfo]
|
|
|
|
pop esi ; now we will have pointer to result in esi
|
|
|
|
; test for error
|
|
test eax, eax
|
|
jnz still
|
|
|
|
mov esi, [esi + addrinfo.ai_addr]
|
|
mov esi, [esi + sockaddr_in.sin_addr]
|
|
mov dword [IP], esi
|
|
|
|
stdcall mem.Alloc, buffer_len
|
|
test eax, eax
|
|
jz stop_transfer
|
|
mov [packetbuff], eax
|
|
|
|
invoke file_open, local_addr, O_READ + O_WRITE + O_CREATE
|
|
cmp eax, 32
|
|
jb stop_transfer
|
|
|
|
mov [fh], eax
|
|
|
|
mcall socket, AF_INET4, IP_PROTO_UDP, 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 ; method = get?
|
|
jz @f
|
|
mov word [I_END], opcode_wrq
|
|
@@:
|
|
|
|
xor al , al
|
|
mov edi, remote_addr
|
|
mov ecx, 250
|
|
repnz scasb
|
|
sub edi, remote_addr
|
|
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
|
|
|
|
mov [last_ack], 0
|
|
|
|
; mcall 26, 9
|
|
; mov [last_time], eax
|
|
|
|
mov [status], str_transfering
|
|
call draw_window
|
|
|
|
mcall 40, 10000101b
|
|
|
|
cmp [option_group2],op3 ; method = get?
|
|
jnz send_data_loop
|
|
|
|
invoke file_truncate, [fh]
|
|
|
|
receive_data_loop:
|
|
|
|
mcall 23, TIMEOUT
|
|
|
|
dec eax
|
|
jz .redraw
|
|
|
|
dec eax
|
|
dec eax
|
|
jz .btn
|
|
|
|
mcall recv, [socketnum], [packetbuff], buffer_len ; receive data
|
|
|
|
mov esi, [packetbuff]
|
|
cmp word[esi], opcode_data
|
|
jne .error
|
|
|
|
mov bx, [last_ack]
|
|
inc bx
|
|
rol bx, 8
|
|
|
|
cmp word [esi + 2], bx
|
|
jne .packet_got_lost
|
|
|
|
inc [last_ack]
|
|
|
|
|
|
; now, we need to store the data
|
|
|
|
add esi, 4
|
|
sub eax, 4
|
|
mov ecx, eax
|
|
invoke file_write, [fh], esi ,ecx
|
|
|
|
cmp ecx, 512 ; full data packet?
|
|
jge .continue
|
|
|
|
; last packet, or something else
|
|
|
|
mov [status], str_success
|
|
|
|
.kill_xfer:
|
|
|
|
invoke file_close, [fh]
|
|
mcall close, [socketnum]
|
|
jmp stop_transfer
|
|
|
|
.error:
|
|
|
|
cmp word[esi], opcode_error
|
|
je .decode_error
|
|
|
|
jmp .continue
|
|
|
|
.packet_got_lost:
|
|
|
|
|
|
.continue:
|
|
|
|
; mcall 26, 9
|
|
; mov ebx, [last_time]
|
|
; mov [last_time], eax
|
|
; xor edx, edx
|
|
; sub eax, ebx
|
|
; xchg eax, ecx
|
|
; div ecx
|
|
; mov [kbps], eax
|
|
; mcall 47,1 shl 31 + 7 shl 16 + 1,kbps,305*65536+137,0x40000000, 0x00ffffff
|
|
|
|
mov word [buffer], opcode_ack ; send ack
|
|
mov ax, [last_ack]
|
|
rol ax, 8
|
|
mov word [buffer+2], ax
|
|
mcall send, [socketnum], buffer, 4, 0
|
|
|
|
jmp receive_data_loop
|
|
|
|
|
|
.btn:
|
|
mcall 17
|
|
|
|
jmp .kill_xfer
|
|
|
|
.redraw:
|
|
call draw_window
|
|
jmp receive_data_loop
|
|
|
|
|
|
.decode_error:
|
|
movzx esi, word[esi + 2]
|
|
cmp esi, 7
|
|
cmovg esi, [zero]
|
|
|
|
mov esi, dword [4*esi + error_crosslist]
|
|
mov [status], esi
|
|
|
|
jmp .kill_xfer
|
|
|
|
|
|
|
|
;--------------------------------
|
|
|
|
|
|
send_data_loop:
|
|
|
|
mov word[buffer], opcode_data
|
|
|
|
.read_chunk:
|
|
|
|
inc [last_ack]
|
|
|
|
mov ax, [last_ack]
|
|
xchg al, ah
|
|
mov word[buffer+2], ax
|
|
|
|
invoke file_read, [fh], buffer + 4, 512
|
|
cmp eax, -1
|
|
je .kill_xfer
|
|
|
|
add eax, 4
|
|
mov [packetsize], eax
|
|
|
|
.send_packet:
|
|
mcall send, [socketnum], buffer, [packetsize], 0 ; send data
|
|
|
|
.loop:
|
|
mcall 23, TIMEOUT
|
|
|
|
dec eax
|
|
jz .red
|
|
|
|
dec eax
|
|
dec eax
|
|
jz .btn
|
|
|
|
mcall recv, [socketnum], [packetbuff], buffer_len ; receive ack
|
|
cmp eax, -1
|
|
je .kill_xfer
|
|
|
|
mov esi, [packetbuff]
|
|
|
|
cmp word[esi], opcode_error
|
|
je .decode_error
|
|
|
|
cmp word[esi], opcode_ack
|
|
jne .send_packet
|
|
|
|
mov ax, [last_ack]
|
|
xchg al, ah
|
|
cmp word[esi+2], ax
|
|
jne .send_packet
|
|
|
|
cmp [packetsize], 512+4
|
|
jne .xfer_ok ; transfer is done
|
|
|
|
|
|
|
|
jmp .read_chunk
|
|
|
|
.red:
|
|
call draw_window
|
|
jmp .loop
|
|
|
|
.btn:
|
|
mcall 17
|
|
|
|
|
|
.kill_xfer:
|
|
mov [status], str_fail
|
|
|
|
.xfer_done:
|
|
invoke file_close, [fh]
|
|
mcall close, [socketnum]
|
|
jmp stop_transfer
|
|
|
|
|
|
.xfer_ok:
|
|
mov [status], str_success
|
|
jmp .xfer_done
|
|
|
|
|
|
.decode_error:
|
|
movzx esi, word[esi + 2]
|
|
cmp esi, 7
|
|
cmovg esi, [zero]
|
|
|
|
mov esi, dword [4*esi + error_crosslist]
|
|
mov [status], esi
|
|
|
|
jmp .send_packet
|
|
|
|
|
|
|
|
;-------------------------
|
|
; DATA
|
|
|
|
socketnum dd 0
|
|
kbps dd 0
|
|
done dd 0
|
|
|
|
sockaddr:
|
|
dw AF_INET4
|
|
dw 69 shl 8
|
|
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' ,\
|
|
check_box_draw ,'check_box_draw' ,\
|
|
check_box_mouse ,'check_box_mouse' ,\
|
|
version_ch ,'version_ch' ,\
|
|
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,ed_focus, 13,13
|
|
edit2 edit_box 300,80,25,0xffffff,0x6a9480,0,0,0,99 ,remote_addr,ed_figure_only, 5,5
|
|
edit3 edit_box 300,80,45,0xffffff,0x6a9480,0,0,0,99 ,local_addr,ed_figure_only, 13,13
|
|
edit4 edit_box 40,340,68,0xffffff,0x6a9480,0,0,0,5 ,BLK,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 for KolibriOS',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_waiting db 'Welcome!',0
|
|
str_transfering db 'Transfering...',0
|
|
|
|
str_success db 'Tranfser completed sucessfully',0
|
|
str_fail db 'Transfer failed!',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
|
|
|
|
|
|
error_crosslist:
|
|
dd str_error._0
|
|
dd str_error._1
|
|
dd str_error._2
|
|
dd str_error._3
|
|
dd str_error._4
|
|
dd str_error._5
|
|
dd str_error._6
|
|
dd str_error._7
|
|
|
|
|
|
|
|
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
|
|
|
|
last_time dd ?
|
|
|
|
packetbuff dd ?
|
|
packetsize dd ?
|
|
status dd str_waiting
|
|
zero dd 0
|
|
|
|
SRV db "192.168.1.115",0
|
|
times (SRV + 256 - $) db 0
|
|
|
|
remote_addr db "3.png",0
|
|
times (remote_addr + 256 - $) db 0
|
|
|
|
local_addr db "/sys/test.png",0
|
|
times (local_addr + 256 - $) db 0
|
|
|
|
I_END:
|
|
|
|
buffer:
|
|
rb buffer_len
|
|
|
|
IM_END: |