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

use32
    org 0x0

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

include '../proc32.inc'
include '../macros.inc'
include '../libio.inc'
include '../dll.inc'
include 'editbox_ex.mac'

include '../network.inc'


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

AF_INET4	equ 2
IP_PROTO_UDP	equ 17

opcode_rrq	equ 1
opcode_wrq	equ 2
opcode_data	equ 3
opcode_ack	equ 4
opcode_error	equ 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, 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,50*65536+137, 0x80000000, str_complete

	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	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, 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
	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, 10000101b

	mov	[last_ack], 0






receive_data_loop:

	mcall	23, TIMEOUT

	dec	eax
	jz	.red

	dec	eax
	jz	.key


	mcall	recv, [socketnum], buffer, buffer_len, 0  ; 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, 0  ; 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 69
IP	db 192,168,1,115
sockaddr_len = $ - sockaddr

align 16
@IMPORT:

library box_lib , 'box_lib.obj'
library io_lib	, 'libio.obj'
library 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,  11,11
edit2 edit_box 300,80,25,0xffffff,0x6a9480,0,0,0,99 ,remote_addr,ed_figure_only, 10,10
edit3 edit_box 300,80,45,0xffffff,0x6a9480,0,0,0,99 ,local_addr,ed_figure_only, 27,27
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_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:
buffer:
rb buffer_len

IM_END: