WebSVN – Kolibri OS – Blame – /programs/network/dhcp/trunk/dhcp.asm

Rev	Author	Line No.	Line
31	halyavin	1	;
		2	; DHCP Client
		3	;
		4	; Compile with FASM for Menuet
		5	;
		6
		7	include 'lang.inc'
485	heavyiron	8	include '..\..\..\macros.inc'
31	halyavin	9
		10	use32
		11
485	heavyiron	12	org 0x0
31	halyavin	13
485	heavyiron	14	db 'MENUET01' ; 8 byte id
		15	dd 0x01 ; header version
		16	dd START ; start of code
		17	dd I_END ; size of image
		18	dd I_END+0x8000 ; memory for app
		19	dd I_END+0x8000 ; esp
		20	dd 0x0 , 0x0 ; I_Param , I_Icon
31	halyavin	21
		22
		23	START: ; start of execution
		24	mov eax,40 ; Report events
		25	mov ebx,10000111b ; Stack 8 + defaults
		26	int 0x40
		27
485	heavyiron	28	red: ; redraw
31	halyavin	29	call draw_window ; at first, draw the window
		30
		31	still:
		32	mov eax,10 ; wait here for event
485	heavyiron	33	mcall
31	halyavin	34
		35	cmp eax,1 ; redraw request ?
		36	jz red
		37	cmp eax,2 ; key in buffer ?
		38	jz key
		39	cmp eax,3 ; button in buffer ?
		40	jz button
		41
		42	jmp still
		43	key: ; Keys are not valid at this part of the
		44	mov eax,2 ; loop. Just read it and ignore
485	heavyiron	45	mcall
31	halyavin	46	jmp still
		47
		48	button: ; button
		49	mov eax,17 ; get id
485	heavyiron	50	mcall
31	halyavin	51
		52	cmp ah,1 ; button id=1 ?
		53	jnz noclose
		54
		55	; close socket before exiting
		56	mov eax, 53
		57	mov ebx, 1
		58	mov ecx, [socketNum]
485	heavyiron	59	mcall
31	halyavin	60
		61	mov eax,0xffffffff ; close this program
485	heavyiron	62	mcall
31	halyavin	63
		64	noclose:
		65	cmp ah,3 ; Resolve address?
		66	jnz still
		67
		68	call draw_window
		69
		70	call contactDHCPServer
		71
		72	jmp still
		73
		74
		75	;***************************************************************************
		76	; Function
		77	; parseResponse
		78	;
		79	; Description
		80	; extracts the fields ( client IP address and options ) from
		81	; a DHCP response
		82	; The values go into
		83	; dhcpMsgType,dhcpLease,dhcpClientIP,dhcpServerIP,
		84	; dhcpDNSIP, dhcpSubnet
		85	; The message is stored in dhcpMsg
		86	;
		87	;***************************************************************************
		88	parseResponse:
		89	mov edx, dhcpMsg
		90
		91	mov eax, [edx+16]
		92	mov [dhcpClientIP], eax
		93
		94	; Scan options
		95
		96	add edx, 240 ; Point to first option
		97
		98	pr001:
		99	; Get option id
		100	mov al, [edx]
		101	cmp al, 0xff ; End of options?
		102	je pr_exit
		103
		104	cmp al, 53 ; Msg type is a single byte option
		105	jne pr002
		106
		107	mov al, [edx+2]
		108	mov [dhcpMsgType], al
		109	add edx, 3
		110	jmp pr001 ; Get next option
		111
		112	pr002:
		113	; All other (accepted) options are 4 bytes in length
		114	inc edx
		115	movzx ecx, byte [edx]
		116	inc edx ; point to data
		117
		118	cmp al, 54 ; server id
		119	jne pr0021
		120	mov eax, [edx] ; All options are 4 bytes, so get it
		121	mov [dhcpServerIP], eax
		122	jmp pr003
		123
		124	pr0021:
		125	cmp al, 51 ; lease
		126	jne pr0022
		127	mov eax, [edx] ; All options are 4 bytes, so get it
		128	mov [dhcpLease], eax
		129	jmp pr003
		130
		131	pr0022:
		132	cmp al, 1 ; subnet mask
		133	jne pr0023
		134	mov eax, [edx] ; All options are 4 bytes, so get it
		135	mov [dhcpSubnet], eax
		136	jmp pr003
		137
		138	pr0023:
		139	cmp al, 6 ; dns ip
		140	jne pr0024
		141	mov eax, [edx] ; All options are 4 bytes, so get it
		142	mov [dhcpDNSIP], eax
		143
		144	pr0024:
		145	cmp al, 3 ; gateway ip
		146	jne pr003
		147	mov eax, [edx] ; All options are 4 bytes, so get it
		148	mov [dhcpGateway], eax
		149
		150	pr003:
		151	add edx, ecx
		152	jmp pr001
		153
		154	pr_exit:
		155	ret
		156
		157
		158	;***************************************************************************
		159	; Function
		160	; buildRequest
		161	;
		162	; Description
		163	; Creates a DHCP request packet.
		164	;
		165	;***************************************************************************
		166	buildRequest:
		167	; Clear dhcpMsg to all zeros
		168	xor eax,eax
		169	mov edi,dhcpMsg
		170	mov ecx,512
		171	cld
		172	rep stosb
		173
		174	mov edx, dhcpMsg
		175
		176	mov [edx], byte 0x01 ; Boot request
		177	mov [edx+1], byte 0x01 ; Ethernet
		178	mov [edx+2], byte 0x06 ; Ethernet h/w len
		179	mov [edx+4], dword 0x11223344 ; xid
		180	mov [edx+10], byte 0x80 ; broadcast flag set
		181	mov [edx+236], dword 0x63538263 ; magic number
		182
		183	; option DHCP msg type
		184	mov [edx+240], word 0x0135
		185	mov al, [dhcpMsgType]
		186	mov [edx+240+2], al
		187
		188	; option Lease time = infinity
		189	mov [edx+240+3], word 0x0433
		190	mov eax, [dhcpLease]
		191	mov [edx+240+5], eax
		192
		193	; option requested IP address
		194	mov [edx+240+9], word 0x0432
		195	mov eax, [dhcpClientIP]
		196	mov [edx+240+11], eax
		197
		198	; option request list
		199	mov [edx+240+15], word 0x0437
		200	mov [edx+240+17], dword 0x0f060301
		201
		202	; Check which msg we are sending
		203	cmp [dhcpMsgType], byte 0x01
		204	jne br001
		205
		206	; "Discover" options
		207	; end of options marker
		208	mov [edx+240+21], byte 0xff
		209
		210	mov [dhcpMsgLen], dword 262
		211	jmp br_exit
		212
		213	br001:
		214	; "Request" options
		215
		216	; server IP
		217	mov [edx+240+21], word 0x0436
		218	mov eax, [dhcpServerIP]
		219	mov [edx+240+23], eax
		220
		221	; end of options marker
		222	mov [edx+240+27], byte 0xff
		223
		224	mov [dhcpMsgLen], dword 268
		225
		226	br_exit:
		227	ret
		228
		229
		230
		231	;***************************************************************************
		232	; Function
		233	; contactDHCPServer
		234	;
		235	; Description
		236	; negotiates settings with a DHCP server
		237	;
		238	;***************************************************************************
		239	contactDHCPServer:
		240	; First, open socket
		241	mov eax, 53
		242	mov ebx, 0
		243	mov ecx, 68 ; local port dhcp client
		244	mov edx, 67 ; remote port - dhcp server
		245	mov esi, 0xffffffff ; broadcast
485	heavyiron	246	mcall
31	halyavin	247
		248	mov [socketNum], eax
		249
		250	; Setup the first msg we will send
		251	mov [dhcpMsgType], byte 0x01 ; DHCP discover
		252	mov [dhcpLease], dword 0xffffffff
		253	mov [dhcpClientIP], dword 0
		254	mov [dhcpServerIP], dword 0
		255
		256	call buildRequest
		257
		258	ctr000:
		259	; write to socket ( send broadcast request )
		260	mov eax, 53
		261	mov ebx, 4
		262	mov ecx, [socketNum]
		263	mov edx, [dhcpMsgLen]
		264	mov esi, dhcpMsg
485	heavyiron	265	mcall
31	halyavin	266
		267	; Setup the DHCP buffer to receive response
		268
		269	mov eax, dhcpMsg
		270	mov [dhcpMsgLen], eax ; Used as a pointer to the data
		271
		272	; now, we wait for
		273	; UI redraw
		274	; UI close
		275	; or data from remote
		276
		277	ctr001:
		278	mov eax,10 ; wait here for event
485	heavyiron	279	mcall
31	halyavin	280
		281	cmp eax,1 ; redraw request ?
		282	je ctr003
		283	cmp eax,2 ; key in buffer ?
		284	je ctr004
		285	cmp eax,3 ; button in buffer ?
		286	je ctr005
		287
		288
		289	; Any data in the UDP receive buffer?
		290	mov eax, 53
		291	mov ebx, 2
		292	mov ecx, [socketNum]
485	heavyiron	293	mcall
31	halyavin	294
		295	cmp eax, 0
		296	je ctr001
		297
		298	; we have data - this will be the response
		299	ctr002:
		300	mov eax, 53
		301	mov ebx, 3
		302	mov ecx, [socketNum]
485	heavyiron	303	mcall ; read byte - block (high byte)
31	halyavin	304
		305	; Store the data in the response buffer
		306	mov eax, [dhcpMsgLen]
		307	mov [eax], bl
		308	inc dword [dhcpMsgLen]
		309
		310	mov eax, 53
		311	mov ebx, 2
		312	mov ecx, [socketNum]
485	heavyiron	313	mcall ; any more data?
31	halyavin	314
		315	cmp eax, 0
		316	jne ctr002 ; yes, so get it
		317
		318	; depending on which msg we sent, handle the response
		319	; accordingly.
		320	; If the response is to a dhcp discover, then:
		321	; 1) If response is DHCP OFFER then
		322	; 1.1) record server IP, lease time & IP address.
		323	; 1.2) send a request packet
		324	; 2) else exit ( display error )
		325	; If the response is to a dhcp request, then:
		326	; 1) If the response is DHCP ACK then
		327	; 1.1) extract the DNS & subnet fields. Set them in the stack
		328	; 2) else exit ( display error )
		329
		330
		331	cmp [dhcpMsgType], byte 0x01 ; did we send a discover?
		332	je ctr007
		333	cmp [dhcpMsgType], byte 0x03 ; did we send a request?
		334	je ctr008
		335
		336	; should never get here - we only send discover or request
		337	jmp ctr006
		338
		339	ctr007:
		340	call parseResponse
		341
		342	; Was the response an offer? It should be
		343	cmp [dhcpMsgType], byte 0x02
		344	jne ctr006 ; NO - so quit
		345
		346	; send request
		347	mov [dhcpMsgType], byte 0x03 ; DHCP request
		348	call buildRequest
		349	jmp ctr000
		350
		351	ctr008:
		352	call parseResponse
		353
		354	; Was the response an ACK? It should be
		355	cmp [dhcpMsgType], byte 0x05
		356	jne ctr006 ; NO - so quit
		357
		358	; Set or display addresses here...
		359
		360	ctr006:
		361	; close socket
		362	mov eax, 53
		363	mov ebx, 1
		364	mov ecx, [socketNum]
485	heavyiron	365	mcall
31	halyavin	366
		367	mov [socketNum], dword 0xFFFF
		368
		369	call draw_window
		370
		371	jmp ctr001
		372
		373	ctr003: ; redraw
		374	call draw_window
		375	jmp ctr001
		376
		377	ctr004: ; key
		378	mov eax,2 ; just read it and ignore
485	heavyiron	379	mcall
31	halyavin	380	jmp ctr001
		381
		382	ctr005: ; button
		383	mov eax,17 ; get id
485	heavyiron	384	mcall
31	halyavin	385
		386	; close socket
		387	mov eax, 53
		388	mov ebx, 1
		389	mov ecx, [socketNum]
485	heavyiron	390	mcall
31	halyavin	391
		392	mov [socketNum], dword 0xFFFF
		393
		394	call draw_window ; at first, draw the window
		395
		396	ret
		397
		398
		399
		400
		401	; *********************************************
		402	; ***** WINDOW DEFINITIONS AND DRAW ******
		403	; *********************************************
		404
		405
		406	; Pass in the IP address in edi
		407	; row to display in [ya]
		408	drawIP:
		409	; mov edi,hostIP
		410	mov ecx, edi
		411	add ecx, 4
		412	mov edx,[ya]
		413	add edx, 97*65536
		414	mov esi,0x00ffffff
		415	mov ebx,3*65536
		416
		417	ipdisplay:
		418	mov eax,47
		419	push ecx
		420	movzx ecx,byte [edi]
485	heavyiron	421	mcall
31	halyavin	422	pop ecx
		423	add edx,6465536
		424	inc edi
		425	cmp edi,ecx
		426	jb ipdisplay
		427	ret
		428
		429
		430	drawDHMS:
		431
		432	mov eax,[edi]
		433	bswap eax
		434
		435	mov esi,dhms
		436	mov ecx,16
		437	mov edi,text+40*4+12
		438	cmp eax,0xffffffff
		439	jne nforever
		440	mov esi,forever
		441	cld
		442	rep movsb
		443	ret
		444	nforever:
		445	cld
		446	rep movsb
		447
		448	mov ecx,28
		449	xor edx,edx
		450	mov ebx,60
		451	div ebx
		452	call displayDHMS
		453	xor edx,edx
		454	div ebx
		455	call displayDHMS
		456	xor edx,edx
		457	mov ebx,24
		458	div ebx
		459	call displayDHMS
		460	mov edx,eax
		461	call displayDHMS
		462
		463	ret
		464
		465
		466	displayDHMS:
		467
		468	pusha
		469	mov eax,47
		470	mov ebx,3*65536
		471	mov edx,ecx
		472	imul edx,6
		473	shl edx,16
		474	add edx,1*65536+99
		475	mov ecx,[esp+20]
		476	mov esi,0xffffff
485	heavyiron	477	mcall
31	halyavin	478	popa
		479	sub ecx,4
		480	ret
		481
		482
		483	draw_window:
		484
		485	mov eax,12 ; function 12:tell os about windowdraw
		486	mov ebx,1 ; 1, start of draw
485	heavyiron	487	mcall
31	halyavin	488	; DRAW WINDOW
		489	mov eax,0 ; function 0 : define and draw window
		490	mov ebx,10065536+300 ; [x start] 65536 + [x size]
		491	mov ecx,10065536+156 ; [y start] 65536 + [y size]
485	heavyiron	492	mov edx,0x13224466 ; color of work area RRGGBB
		493	mov edi,title ; WINDOW LABEL
		494	mcall
		495
31	halyavin	496	mov eax,8 ; Resolve
		497	mov ebx,20*65536+90
		498	mov ecx,127*65536+15
		499	mov edx,3
		500	mov esi,0x557799
485	heavyiron	501	mcall
31	halyavin	502
		503	; Pass in the IP address in edi
		504	; row to display in [ya]
		505	mov edi, dhcpClientIP
		506	mov eax, 35
		507	mov [ya], eax
		508	call drawIP
		509	mov edi, dhcpGateway
		510	mov eax, 35 + 16
		511	mov [ya], eax
		512	call drawIP
		513	mov edi, dhcpSubnet
		514	mov eax, 35 + 32
		515	mov [ya], eax
		516	call drawIP
		517	mov edi, dhcpDNSIP
		518	mov eax, 35 + 48
		519	mov [ya], eax
		520	call drawIP
		521	mov edi, dhcpLease
		522	call drawDHMS
		523
		524	; Re-draw the screen text
		525	cld
485	heavyiron	526	mov eax,4
31	halyavin	527	mov ebx,25*65536+35 ; draw info text with function 4
		528	mov ecx,0xffffff
		529	mov edx,text
		530	mov esi,40
		531	newline:
485	heavyiron	532	mcall
31	halyavin	533	add ebx,16
		534	add edx,40
		535	cmp [edx],byte 'x'
		536	jnz newline
		537
		538
		539	mov eax,12 ; function 12:tell os about windowdraw
		540	mov ebx,2 ; 2, end of draw
485	heavyiron	541	mcall
31	halyavin	542
		543	ret
		544
		545
		546
		547	; DATA AREA
		548
		549	ya dd 0x0
		550
		551	text:
		552	db 'Client IP : . . . '
		553	db 'Gateway IP: . . . '
		554	db 'Subnet : . . . '
		555	db 'DNS IP : . . . '
		556	db 'Lease Time: d h m s '
		557	db ' '
		558	db ' SEND REQUEST '
		559	db 'x <- END MARKER, DONT DELETE '
		560
		561
		562	dhms db ' d h m s'
		563	forever db 'Forever '
		564
485	heavyiron	565	title db 'DHCP Client Test',0
31	halyavin	566
		567	dhcpMsgType: db 0
		568	dhcpLease: dd 0
		569	dhcpClientIP: dd 0
		570	dhcpServerIP: dd 0
		571	dhcpDNSIP: dd 0
		572	dhcpSubnet: dd 0
		573	dhcpGateway: dd 0
		574
		575	dhcpMsgLen: dd 0
		576	socketNum: dd 0xFFFF
		577	dhcpMsg:
		578	I_END:
		579

Subversion Repositories Kolibri OS

(root)/programs/network/dhcp/trunk/dhcp.asm – Rev 485