WebSVN – Kolibri OS – Blame – /programs/network/telnet/telnet.asm

Rev	Author	Line No.	Line
3545	hidnplayr	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		2	;; ;;
3618	hidnplayr	3	;; Copyright (C) KolibriOS team 2010-2013. All rights reserved. ;;
3545	hidnplayr	4	;; Distributed under terms of the GNU General Public License ;;
		5	;; ;;
		6	;; telnet.asm - Telnet client for KolibriOS ;;
		7	;; ;;
		8	;; Written by hidnplayr@kolibrios.org ;;
		9	;; ;;
		10	;; GNU GENERAL PUBLIC LICENSE ;;
		11	;; Version 2, June 1991 ;;
		12	;; ;;
		13	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		14
		15	format binary as ""
		16
3618	hidnplayr	17	__DEBUG__ = 0
		18	__DEBUG_LEVEL__ = 1
		19	BUFFERSIZE = 4096
3545	hidnplayr	20
		21	use32
		22	; standard header
		23	db 'MENUET01' ; signature
		24	dd 1 ; header version
		25	dd start ; entry point
		26	dd i_end ; initialized size
3574	hidnplayr	27	dd mem+4096 ; required memory
		28	dd mem+4096 ; stack pointer
		29	dd hostname ; parameters
3545	hidnplayr	30	dd 0 ; path
		31
3618	hidnplayr	32	include '../../macros.inc'
3545	hidnplayr	33	purge mov,add,sub
3618	hidnplayr	34	include '../../proc32.inc'
		35	include '../../dll.inc'
		36	include '../../debug-fdo.inc'
		37	include '../../network.inc'
3545	hidnplayr	38
		39	; entry point
		40	start:
		41	; load libraries
		42	stdcall dll.Load, @IMPORT
		43	test eax, eax
		44	jnz exit
		45	; initialize console
		46	push 1
		47	call [con_start]
		48	push title
		49	push 25
		50	push 80
		51	push 25
		52	push 80
		53	call [con_init]
		54
		55	; Check for parameters
3574	hidnplayr	56	cmp byte [hostname], 0
3545	hidnplayr	57	jne resolve
		58
		59	main:
		60	call [con_cls]
		61	; Welcome user
		62	push str1
		63	call [con_write_asciiz]
		64
3574	hidnplayr	65	prompt:
3545	hidnplayr	66	; write prompt
		67	push str2
		68	call [con_write_asciiz]
		69	; read string
3574	hidnplayr	70	mov esi, hostname
3545	hidnplayr	71	push 256
		72	push esi
		73	call [con_gets]
		74	; check for exit
		75	test eax, eax
		76	jz done
		77	cmp byte [esi], 10
		78	jz done
		79
		80	resolve:
		81
3574	hidnplayr	82	mov [sockaddr1.port], 23 shl 8
		83
3545	hidnplayr	84	; delete terminating '\n'
3574	hidnplayr	85	mov esi, hostname
3545	hidnplayr	86	@@:
		87	lodsb
3574	hidnplayr	88	cmp al, ':'
		89	je .do_port
3545	hidnplayr	90	cmp al, 0x20
		91	ja @r
		92	mov byte [esi-1], 0
3574	hidnplayr	93	jmp .done
3545	hidnplayr	94
3574	hidnplayr	95	.do_port:
		96	xor eax, eax
		97	xor ebx, ebx
		98	mov byte [esi-1], 0
		99	.portloop:
		100	lodsb
		101	cmp al, 0x20
		102	jbe .port_done
		103	sub al, '0'
		104	jb hostname_error
		105	cmp al, 9
		106	ja hostname_error
		107	lea ebx, [ebx*4 + ebx]
		108	shl ebx, 1
		109	add ebx, eax
		110	jmp .portloop
3545	hidnplayr	111
3574	hidnplayr	112	.port_done:
		113	xchg bl, bh
		114	mov [sockaddr1.port], bx
		115
		116	.done:
		117
3545	hidnplayr	118	; resolve name
		119	push esp ; reserve stack place
3574	hidnplayr	120	push esp ; ptr to result
		121	push 0 ; addrinfo hints
		122	push 0 ; servname
		123	push hostname; hostname
3545	hidnplayr	124	call [getaddrinfo]
		125	pop esi
		126	; test for error
		127	test eax, eax
3687	hidnplayr	128	jnz dns_error
3545	hidnplayr	129
3574	hidnplayr	130	call [con_cls]
		131	push str3
		132	call [con_write_asciiz]
		133	push hostname
		134	call [con_write_asciiz]
		135
3545	hidnplayr	136	; write results
		137	push str8
		138	call [con_write_asciiz]
		139	; mov edi, esi
		140
		141	; convert IP address to decimal notation
		142	mov eax, [esi+addrinfo.ai_addr]
		143	mov eax, [eax+sockaddr_in.sin_addr]
		144	mov [sockaddr1.ip], eax
		145	push eax
		146	call [inet_ntoa]
		147	; write result
		148	push eax
		149	call [con_write_asciiz]
		150	; free allocated memory
		151	push esi
		152	call [freeaddrinfo]
		153
		154	push str9
		155	call [con_write_asciiz]
		156
		157	mcall socket, AF_INET4, SOCK_STREAM, 0
		158	cmp eax, -1
3687	hidnplayr	159	jz socket_err
3545	hidnplayr	160	mov [socketnum], eax
		161
		162	mcall connect, [socketnum], sockaddr1, 18
		163
		164	mcall 40, 1 shl 7 ; + 7
		165	call [con_cls]
		166
		167	mcall 18, 7
		168	push eax
		169	mcall 51, 1, thread, mem - 2048
		170	pop ecx
		171	mcall 18, 3
		172
		173	mainloop:
		174	call [con_get_flags]
		175	test eax, 0x200 ; con window closed?
		176	jnz exit
		177
		178	mcall recv, [socketnum], buffer_ptr, BUFFERSIZE, 0
		179	cmp eax, -1
3704	hidnplayr	180	je closed
3545	hidnplayr	181
3703	hidnplayr	182
3545	hidnplayr	183	DEBUGF 1, 'TELNET: got %u bytes of data !\n', eax
		184
		185	mov esi, buffer_ptr
		186	lea edi, [esi + eax]
		187	mov byte [edi], 0
		188
		189	.scan_cmd:
		190	cmp byte [esi], 0xff ; Interpret As Command
		191	jne .no_cmd
		192	; TODO: parse options, for now, we will reply with 'WONT' to everything
		193	mov byte [esi + 1], 252 ; WONT
		194	add esi, 3 ; a command is always 3 bytes
		195	jmp .scan_cmd
		196	.no_cmd:
		197
		198	cmp esi, buffer_ptr
3687	hidnplayr	199	je .print
3545	hidnplayr	200
		201	DEBUGF 1, 'TELNET: sending data\n'
		202
		203	push esi edi
		204	sub esi, buffer_ptr
		205	mcall send, [socketnum], buffer_ptr, , 0
		206	pop edi esi
		207
3687	hidnplayr	208	.print:
3545	hidnplayr	209	cmp esi, edi
3704	hidnplayr	210	jae mainloop
3545	hidnplayr	211
3687	hidnplayr	212	push esi
		213	call [con_write_asciiz]
3545	hidnplayr	214
3687	hidnplayr	215	.loop:
		216	lodsb
		217	test al, al
		218	jz .print
		219	jmp .loop
3545	hidnplayr	220
		221
3687	hidnplayr	222	socket_err:
		223	DEBUGF 1, "TELNET: socket error %d", ebx
3545	hidnplayr	224	push str6
		225	call [con_write_asciiz]
		226
3574	hidnplayr	227	jmp prompt
3545	hidnplayr	228
3687	hidnplayr	229	dns_error:
		230	DEBUGF 1, "TELNET: DNS error %d", eax
3545	hidnplayr	231	push str5
		232	call [con_write_asciiz]
3574	hidnplayr	233
		234	jmp prompt
		235
		236	hostname_error:
		237	push str11
3545	hidnplayr	238	call [con_write_asciiz]
3574	hidnplayr	239	jmp prompt
3545	hidnplayr	240
3703	hidnplayr	241	closed:
		242	push str12
		243	call [con_write_asciiz]
		244	jmp prompt
		245
3545	hidnplayr	246	done:
		247	push 1
		248	call [con_exit]
		249	exit:
		250
		251	mcall close, [socketnum]
		252	mcall -1
		253
		254
		255
		256	thread:
		257	mcall 40, 0
		258	.loop:
		259	call [con_getch2]
		260	mov byte [send_data], al
		261	mcall send, [socketnum], send_data, 1
		262
		263	call [con_get_flags]
		264	test eax, 0x200 ; con window closed?
		265	jz .loop
		266	mcall -1
		267
		268	; data
		269	title db 'Telnet',0
3574	hidnplayr	270	str1 db 'Telnet for KolibriOS',10,10,\
		271	'Please enter URL of telnet server (for example: towel.blinkenlights.nl:23)',10,10,0
3545	hidnplayr	272	str2 db '> ',0
3574	hidnplayr	273	str3 db 'Connecting to ',0
3545	hidnplayr	274	str4 db 10,0
		275	str8 db ' (',0
		276	str9 db ')',10,0
		277
3574	hidnplayr	278	str5 db 'Name resolution failed.',10,10,0
		279	str6 db 'Could not open socket.',10,10,0
		280	str11 db 'Invalid hostname.',10,10,0
3703	hidnplayr	281	str12 db 10,'Remote host closed the connection.',10,10,0
3574	hidnplayr	282
3545	hidnplayr	283	sockaddr1:
		284	dw AF_INET4
3574	hidnplayr	285	.port dw 0
3545	hidnplayr	286	.ip dd 0
		287	rb 10
		288
		289	include_debug_strings ; ALWAYS present in data section
		290
		291
		292
		293	; import
		294	align 4
		295	@IMPORT:
		296
		297	library network, 'network.obj', console, 'console.obj'
		298	import network, \
		299	getaddrinfo, 'getaddrinfo', \
		300	freeaddrinfo, 'freeaddrinfo', \
		301	inet_ntoa, 'inet_ntoa'
		302	import console, \
		303	con_start, 'START', \
		304	con_init, 'con_init', \
		305	con_write_asciiz, 'con_write_asciiz', \
		306	con_exit, 'con_exit', \
		307	con_gets, 'con_gets',\
		308	con_cls, 'con_cls',\
		309	con_getch2, 'con_getch2',\
		310	con_set_cursor_pos, 'con_set_cursor_pos',\
		311	con_write_string, 'con_write_string',\
		312	con_get_flags, 'con_get_flags'
		313
		314
		315	i_end:
		316
		317	socketnum dd ?
		318	buffer_ptr rb BUFFERSIZE+1
3574	hidnplayr	319	send_data rb 1
3545	hidnplayr	320
3574	hidnplayr	321	hostname rb 1024
		322
3545	hidnplayr	323	mem:

Subversion Repositories Kolibri OS

(root)/programs/network/telnet/telnet.asm – Rev 3704