WebSVN – Kolibri OS – Blame – /kernel/branches/Kolibri-acpi/network/udp.inc

Rev	Author	Line No.	Line
3555	Serge	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		2	;; ;;
3908	Serge	3	;; Copyright (C) KolibriOS team 2004-2013. All rights reserved. ;;
3555	Serge	4	;; Distributed under terms of the GNU General Public License ;;
		5	;; ;;
		6	;; UDP.INC ;;
		7	;; ;;
		8	;; Part of the tcp/ip network stack for KolibriOS ;;
		9	;; ;;
		10	;; Written by hidnplayr@kolibrios.org ;;
		11	;; ;;
		12	;; GNU GENERAL PUBLIC LICENSE ;;
		13	;; Version 2, June 1991 ;;
		14	;; ;;
		15	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
379	serge	16
593	mikedld	17	$Revision: 4265 $
379	serge	18
593	mikedld	19
3555	Serge	20	struct UDP_header
		21
		22	SourcePort dw ?
		23	DestinationPort dw ?
		24	Length dw ? ; Length of (UDP Header + Data)
		25	Checksum dw ?
		26
		27	ends
		28
		29
3725	Serge	30	uglobal
3555	Serge	31	align 4
3725	Serge	32
3626	Serge	33	UDP_PACKETS_TX rd NET_DEVICES_MAX
		34	UDP_PACKETS_RX rd NET_DEVICES_MAX
3725	Serge	35
3555	Serge	36	endg
		37
		38
		39	;-----------------------------------------------------------------
1	ha	40	;
3555	Serge	41	; UDP_init
379	serge	42	;
3555	Serge	43	; This function resets all UDP variables
		44	;
		45	;-----------------------------------------------------------------
		46	macro UDP_init {
261	hidnplayr	47
3555	Serge	48	xor eax, eax
		49	mov edi, UDP_PACKETS_TX
3626	Serge	50	mov ecx, 2*NET_DEVICES_MAX
3725	Serge	51	rep stosd
3555	Serge	52	}
261	hidnplayr	53
3555	Serge	54
		55	macro UDP_checksum IP1, IP2 { ; esi = ptr to udp packet, ecx = packet size, destroys: ecx, edx
		56
		57	; Pseudoheader
		58	mov edx, IP_PROTO_UDP
		59
		60	add dl, [IP1+1]
		61	adc dh, [IP1+0]
		62	adc dl, [IP1+3]
		63	adc dh, [IP1+2]
		64
		65	adc dl, [IP2+1]
		66	adc dh, [IP2+0]
		67	adc dl, [IP2+3]
		68	adc dh, [IP2+2]
		69
		70	adc dl, cl ; byte[esi+UDP_header.Length+1]
		71	adc dh, ch ; byte[esi+UDP_header.Length+0]
		72
		73	; Done with pseudoheader, now do real header
		74	adc dl, byte[esi+UDP_header.SourcePort+1]
		75	adc dh, byte[esi+UDP_header.SourcePort+0]
		76
		77	adc dl, byte[esi+UDP_header.DestinationPort+1]
		78	adc dh, byte[esi+UDP_header.DestinationPort+0]
		79
		80	adc dl, byte[esi+UDP_header.Length+1]
		81	adc dh, byte[esi+UDP_header.Length+0]
		82
		83	adc edx, 0
		84
		85	; Done with header, now do data
		86	push esi
		87	movzx ecx, [esi+UDP_header.Length]
		88	rol cx , 8
		89	sub cx , sizeof.UDP_header
		90	add esi, sizeof.UDP_header
		91
		92	call checksum_1
		93	call checksum_2
		94	pop esi
		95
		96	add [esi+UDP_header.Checksum], dx ; this final instruction will set or clear ZF :)
		97
		98	}
		99
		100
		101	;-----------------------------------------------------------------
261	hidnplayr	102	;
3555	Serge	103	; UDP_input:
261	hidnplayr	104	;
3555	Serge	105	; Called by IPv4_input,
		106	; this procedure will inject the udp data diagrams in the application sockets.
261	hidnplayr	107	;
3555	Serge	108	; IN: [esp] = Pointer to buffer
		109	; [esp+4] = size of buffer
		110	; ebx = ptr to device struct
		111	; ecx = UDP Packet size
		112	; esi = ptr to UDP header
		113	; edi = ptr to ipv4 source and dest address
261	hidnplayr	114	;
3555	Serge	115	; OUT: /
		116	;
		117	;-----------------------------------------------------------------
		118	align 4
		119	UDP_input:
261	hidnplayr	120
3589	Serge	121	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: size=%u\n", ecx
261	hidnplayr	122
3555	Serge	123	; First validate, checksum
261	hidnplayr	124
3555	Serge	125	neg [esi + UDP_header.Checksum] ; substract checksum from 0
		126	jz .no_checksum ; if checksum is zero, it is considered valid
261	hidnplayr	127
3555	Serge	128	; otherwise, we will re-calculate the checksum and add it to this value, thus creating 0 when it is correct
		129
		130	UDP_checksum (edi), (edi+4)
		131	jnz .checksum_mismatch
		132
		133	.no_checksum:
3589	Serge	134	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: checksum ok\n"
3555	Serge	135
		136	; Convert length to little endian
		137
		138	rol [esi + UDP_header.Length], 8
		139
		140	; Look for a socket where
		141	; IP Packet UDP Destination Port = local Port
		142	; IP Packet SA = Remote IP
		143
3725	Serge	144	pusha
		145	mov ecx, socket_mutex
		146	call mutex_lock
		147	popa
		148
3555	Serge	149	mov cx, [esi + UDP_header.SourcePort]
		150	mov dx, [esi + UDP_header.DestinationPort]
		151	mov edi, [edi + 4] ; ipv4 source address
		152	mov eax, net_sockets
		153
		154	.next_socket:
		155	mov eax, [eax + SOCKET.NextPtr]
		156	or eax, eax
3725	Serge	157	jz .dump_
3555	Serge	158
		159	cmp [eax + SOCKET.Domain], AF_INET4
		160	jne .next_socket
		161
		162	cmp [eax + SOCKET.Protocol], IP_PROTO_UDP
		163	jne .next_socket
		164
		165	cmp [eax + UDP_SOCKET.LocalPort], dx
		166	jne .next_socket
		167
3589	Serge	168	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: socket=%x\n", eax
3555	Serge	169
3725	Serge	170	pusha
		171	mov ecx, socket_mutex
		172	call mutex_unlock
		173	popa
		174
3555	Serge	175	;;; TODO: when packet is processed, check more sockets!
		176
		177	; cmp [eax + IP_SOCKET.RemoteIP], 0xffffffff
		178	; je @f
		179	; cmp [eax + IP_SOCKET.RemoteIP], edi
		180	; jne .next_socket
		181	; @@:
1	ha	182	;
3555	Serge	183	; FIXME: UDP should check remote IP, but not under all circumstances!
		184
4265	Serge	185	cmp [eax + UDP_SOCKET.RemotePort], 0
3555	Serge	186	je .updateport
		187
		188	cmp [eax + UDP_SOCKET.RemotePort], cx
		189	jne .dump
		190
		191	pusha
		192	lea ecx, [eax + SOCKET.mutex]
		193	call mutex_lock
		194	popa
		195
		196	.updatesock:
3725	Serge	197	call NET_ptr_to_num4
		198	inc [UDP_PACKETS_RX + edi]
3555	Serge	199
		200	movzx ecx, [esi + UDP_header.Length]
		201	sub ecx, sizeof.UDP_header
		202	add esi, sizeof.UDP_header
		203
		204	jmp SOCKET_input
		205
		206	.updateport:
		207	pusha
		208	lea ecx, [eax + SOCKET.mutex]
		209	call mutex_lock
		210	popa
		211
3589	Serge	212	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: new remote port=%x\n", cx ; FIXME: find a way to print big endian values with debugf
3555	Serge	213	mov [eax + UDP_SOCKET.RemotePort], cx
		214	jmp .updatesock
		215
3725	Serge	216	.dump_:
3555	Serge	217
3725	Serge	218	pusha
		219	mov ecx, socket_mutex
		220	call mutex_unlock
		221	popa
		222
		223	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: no socket found\n"
		224
		225	jmp .dump
		226
3555	Serge	227	.checksum_mismatch:
3589	Serge	228	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_input: checksum mismatch\n"
3555	Serge	229
		230	.dump:
3908	Serge	231	call NET_packet_free
3555	Serge	232	add esp, 4 ; pop (balance stack)
3589	Serge	233	DEBUGF DEBUG_NETWORK_VERBOSE,"UDP_input: dumping\n"
3555	Serge	234
		235	ret
		236
		237
		238
		239
		240	;-----------------------------------------------------------------
1	ha	241	;
3555	Serge	242	; UDP_output
		243	;
		244	; IN: eax = socket pointer
		245	; ecx = number of bytes to send
		246	; esi = pointer to data
		247	;
		248	;-----------------------------------------------------------------
379	serge	249
3555	Serge	250	align 4
		251	UDP_output:
379	serge	252
3589	Serge	253	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_output: socket=%x bytes=%u data_ptr=%x\n", eax, ecx, esi
379	serge	254
3555	Serge	255	mov dx, [eax + UDP_SOCKET.RemotePort]
3589	Serge	256	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_output: remote port=%x, ", dx ; FIXME: find a way to print big endian values with debugf
3555	Serge	257	rol edx, 16
		258	mov dx, [eax + UDP_SOCKET.LocalPort]
3589	Serge	259	DEBUGF DEBUG_NETWORK_VERBOSE, "local port=%x\n", dx
379	serge	260
3555	Serge	261	sub esp, 8 ; Data ptr and data size will be placed here
		262	push edx esi
		263	mov edx, [eax + IP_SOCKET.LocalIP]
		264	mov eax, [eax + IP_SOCKET.RemoteIP]
		265	mov di, IP_PROTO_UDP shl 8 + 128
		266	add ecx, sizeof.UDP_header
		267	call IPv4_output
		268	jz .fail
		269	mov [esp + 8], eax ; pointer to buffer start
		270	mov [esp + 8 + 4], edx ; buffer size
379	serge	271
3555	Serge	272	mov [edi + UDP_header.Length], cx
		273	rol [edi + UDP_header.Length], 8
379	serge	274
3555	Serge	275	pop esi
		276	push edi ecx
		277	sub ecx, sizeof.UDP_header
		278	add edi, sizeof.UDP_header
		279	shr ecx, 2
3725	Serge	280	rep movsd
3555	Serge	281	mov ecx, [esp]
		282	and ecx, 3
3725	Serge	283	rep movsb
3555	Serge	284	pop ecx edi
379	serge	285
3555	Serge	286	pop dword [edi + UDP_header.SourcePort]
1	ha	287
3555	Serge	288	; Checksum
		289	mov esi, edi
		290	mov [edi + UDP_header.Checksum], 0
		291	UDP_checksum (edi-4), (edi-8) ; FIXME: IPv4 packet could have options..
379	serge	292
3589	Serge	293	DEBUGF DEBUG_NETWORK_VERBOSE, "UDP_output: sending with device %x\n", ebx
3555	Serge	294	call [ebx + NET_DEVICE.transmit]
		295	test eax, eax
		296	jnz @f
3725	Serge	297	call NET_ptr_to_num4
		298	inc [UDP_PACKETS_TX + edi]
3555	Serge	299	@@:
379	serge	300
3555	Serge	301	ret
379	serge	302
3555	Serge	303	.fail:
3589	Serge	304	DEBUGF DEBUG_NETWORK_ERROR, "UDP_output: failed\n"
3555	Serge	305	add esp, 4+4+8
		306	or eax, -1
		307	ret
379	serge	308
		309
		310
4265	Serge	311
		312	;-----------------------------------------------------------------
		313	;
		314	; UDP_connect
		315	;
		316	; IN: eax = socket pointer
		317	; OUT: eax = 0 ok / -1 error
		318	; ebx = error code
		319	;
		320	;-------------------------
		321	align 4
		322	UDP_connect:
		323
		324	test [eax + SOCKET.state], SS_ISCONNECTED
		325	jz @f
		326	call UDP_disconnect
		327	@@:
		328
		329	push eax edx
		330	lea ecx, [eax + SOCKET.mutex]
		331	call mutex_lock
		332	pop edx eax
		333
		334	; Fill in local IP
		335	cmp [eax + IP_SOCKET.LocalIP], 0
		336	jne @f
		337	push [IP_LIST + 4] ; FIXME: use correct local IP
		338	pop [eax + IP_SOCKET.LocalIP]
		339
		340	; Fill in remote port and IP, overwriting eventually previous values
		341	pushw [edx + 2]
		342	pop [eax + UDP_SOCKET.RemotePort]
		343
		344	pushd [edx + 4]
		345	pop [eax + IP_SOCKET.RemoteIP]
		346
		347	; Find a local port, if user didnt define one
		348	cmp [eax + UDP_SOCKET.LocalPort], 0
		349	jne @f
		350	call SOCKET_find_port
		351	@@:
		352
		353	push eax
		354	init_queue (eax + SOCKET_QUEUE_LOCATION) ; Set up data receiving queue
		355	pop eax
		356
		357	push eax
		358	lea ecx, [eax + SOCKET.mutex]
		359	call mutex_unlock
		360	pop eax
		361
		362	call SOCKET_is_connected
		363
		364	xor eax, eax
		365	ret
		366
		367
		368	;-----------------------------------------------------------------
		369	;
		370	; UDP_disconnect
		371	;
		372	; IN: eax = socket pointer
		373	; OUT: eax = socket pointer
		374	;
		375	;-------------------------
		376	align 4
		377	UDP_disconnect:
		378
		379	; TODO: remove the pending received data
		380
		381	call SOCKET_is_disconnected
		382
		383	ret
		384
		385
		386
		387
		388
3555	Serge	389	;---------------------------------------------------------------------------
		390	;
		391	; UDP_API
		392	;
		393	; This function is called by system function 75
		394	;
		395	; IN: subfunction number in bl
		396	; device number in bh
		397	; ecx, edx, .. depends on subfunction
		398	;
		399	; OUT:
		400	;
		401	;---------------------------------------------------------------------------
1	ha	402
3555	Serge	403	align 4
		404	UDP_api:
379	serge	405
3555	Serge	406	movzx eax, bh
		407	shl eax, 2
379	serge	408
3555	Serge	409	test bl, bl
		410	jz .packets_tx ; 0
		411	dec bl
		412	jz .packets_rx ; 1
379	serge	413
3555	Serge	414	.error:
		415	mov eax, -1
		416	ret
379	serge	417
3555	Serge	418	.packets_tx:
		419	mov eax, [UDP_PACKETS_TX + eax]
		420	ret
379	serge	421
3555	Serge	422	.packets_rx:
		423	mov eax, [UDP_PACKETS_RX + eax]
		424	ret

Subversion Repositories Kolibri OS

(root)/kernel/branches/Kolibri-acpi/network/udp.inc – Rev 4265