kolibrios/kernel/branches/net/applications/tftpc/TFTP.asm

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: