WebSVN – Kolibri OS – Blame – /kernel/branches/kolibri-process/network/icmp.inc

Rev	Author	Line No.	Line
4429	Serge	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		2	;; ;;
		3	;; Copyright (C) KolibriOS team 2004-2013. All rights reserved. ;;
		4	;; Distributed under terms of the GNU General Public License ;;
		5	;; ;;
		6	;; ICMP.INC ;;
		7	;; ;;
		8	;; Part of the tcp/ip network stack for KolibriOS ;;
		9	;; ;;
		10	;; Based on the work of [Johnny_B] and [smb] ;;
		11	;; ;;
		12	;; Written by hidnplayr@kolibrios.org ;;
		13	;; ;;
		14	;; GNU GENERAL PUBLIC LICENSE ;;
		15	;; Version 2, June 1991 ;;
		16	;; ;;
		17	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		18
		19	$Revision: 2924 $
		20
		21	; ICMP types & codes
		22
		23	ICMP_ECHOREPLY = 0 ; echo reply message
		24
		25	ICMP_UNREACH = 3
		26	ICMP_UNREACH_NET = 0 ; bad net
		27	ICMP_UNREACH_HOST = 1 ; bad host
		28	ICMP_UNREACH_PROTOCOL = 2 ; bad protocol
		29	ICMP_UNREACH_PORT = 3 ; bad port
		30	ICMP_UNREACH_NEEDFRAG = 4 ; IP_DF caused drop
		31	ICMP_UNREACH_SRCFAIL = 5 ; src route failed
		32	ICMP_UNREACH_NET_UNKNOWN = 6 ; unknown net
		33	ICMP_UNREACH_HOST_UNKNOWN = 7 ; unknown host
		34	ICMP_UNREACH_ISOLATED = 8 ; src host isolated
		35	ICMP_UNREACH_NET_PROHIB = 9 ; prohibited access
		36	ICMP_UNREACH_HOST_PROHIB = 10 ; ditto
		37	ICMP_UNREACH_TOSNET = 11 ; bad tos for net
		38	ICMP_UNREACH_TOSHOST = 12 ; bad tos for host
		39	ICMP_UNREACH_FILTER_PROHIB = 13 ; admin prohib
		40	ICMP_UNREACH_HOST_PRECEDENCE = 14 ; host prec vio.
		41	ICMP_UNREACH_PRECEDENCE_CUTOFF = 15 ; prec cutoff
		42
		43	ICMP_SOURCEQUENCH = 4 ; Packet lost, slow down
		44
		45	ICMP_REDIRECT = 5 ; shorter route, codes:
		46	ICMP_REDIRECT_NET = 0 ; for network
		47	ICMP_REDIRECT_HOST = 1 ; for host
		48	ICMP_REDIRECT_TOSNET = 2 ; for tos and net
		49	ICMP_REDIRECT_TOSHOST = 3 ; for tos and host
		50
		51	ICMP_ALTHOSTADDR = 6 ; alternate host address
		52	ICMP_ECHO = 8 ; echo service
		53	ICMP_ROUTERADVERT = 9 ; router advertisement
		54	ICMP_ROUTERADVERT_NORMAL = 0 ; normal advertisement
		55	ICMP_ROUTERADVERT_NOROUTE_COMMON= 16 ; selective routing
		56
		57	ICMP_ROUTERSOLICIT = 10 ; router solicitation
		58	ICMP_TIMXCEED = 11 ; time exceeded, code:
		59	ICMP_TIMXCEED_INTRANS = 0 ; ttl==0 in transit
		60	ICMP_TIMXCEED_REASS = 1 ; ttl==0 in reass
		61
		62	ICMP_PARAMPROB = 12 ; ip header bad
		63	ICMP_PARAMPROB_ERRATPTR = 0 ; error at param ptr
		64	ICMP_PARAMPROB_OPTABSENT = 1 ; req. opt. absent
		65	ICMP_PARAMPROB_LENGTH = 2 ; bad length
		66
		67	ICMP_TSTAMP = 13 ; timestamp request
		68	ICMP_TSTAMPREPLY = 14 ; timestamp reply
		69	ICMP_IREQ = 15 ; information request
		70	ICMP_IREQREPLY = 16 ; information reply
		71	ICMP_MASKREQ = 17 ; address mask request
		72	ICMP_MASKREPLY = 18 ; address mask reply
		73	ICMP_TRACEROUTE = 30 ; traceroute
		74	ICMP_DATACONVERR = 31 ; data conversion error
		75	ICMP_MOBILE_REDIRECT = 32 ; mobile host redirect
		76	ICMP_IPV6_WHEREAREYOU = 33 ; IPv6 where-are-you
		77	ICMP_IPV6_IAMHERE = 34 ; IPv6 i-am-here
		78	ICMP_MOBILE_REGREQUEST = 35 ; mobile registration req
		79	ICMP_MOBILE_REGREPLY = 36 ; mobile registreation reply
		80	ICMP_SKIP = 39 ; SKIP
		81
		82	ICMP_PHOTURIS = 40 ; Photuris
		83	ICMP_PHOTURIS_UNKNOWN_INDEX = 1 ; unknown sec index
		84	ICMP_PHOTURIS_AUTH_FAILED = 2 ; auth failed
		85	ICMP_PHOTURIS_DECRYPT_FAILED = 3 ; decrypt failed
		86
		87
		88
		89	struct ICMP_header
		90
		91	Type db ?
		92	Code db ?
		93	Checksum dw ?
		94	Identifier dw ?
		95	SequenceNumber dw ?
		96
		97	ends
		98
		99
		100	uglobal
		101	align 4
		102
		103	ICMP_PACKETS_TX rd NET_DEVICES_MAX
		104	ICMP_PACKETS_RX rd NET_DEVICES_MAX
		105
		106	endg
		107
		108
		109
		110	;-----------------------------------------------------------------
		111	;
		112	; ICMP_init
		113	;
		114	;-----------------------------------------------------------------
		115
		116	macro ICMP_init {
		117
		118	xor eax, eax
		119	mov edi, ICMP_PACKETS_TX
		120	mov ecx, 2*NET_DEVICES_MAX
		121	rep stosd
		122
		123	}
		124
		125
		126	;-----------------------------------------------------------------
		127	;
		128	; ICMP_input:
		129	;
		130	; This procedure will send reply's to ICMP echo's
		131	; and insert packets into sockets when needed
		132	;
		133	; IN: Pointer to buffer in [esp]
		134	; size of buffer in [esp+4]
		135	; ebx = pointer to device struct
		136	; ecx = ICMP Packet size
		137	; esi = ptr to ICMP Packet data
		138	; edi = ptr to ipv4 source and dest address
		139	;
		140	; OUT: /
		141	;
		142	;-----------------------------------------------------------------
		143	align 4
		144	ICMP_input:
		145
		146	DEBUGF DEBUG_NETWORK_VERBOSE, "ICMP_input:\n"
		147
		148	; First, check the checksum (altough some implementations ignore it)
		149
		150	push esi ecx
		151	push [esi + ICMP_header.Checksum]
		152	mov [esi + ICMP_header.Checksum], 0
		153	xor edx, edx
		154	call checksum_1
		155	call checksum_2
		156	pop si
		157	cmp dx, si
		158	pop ecx edx
		159	jne .checksum_mismatch
		160
		161	; Check packet type
		162
		163	cmp [edx + ICMP_header.Type], ICMP_ECHO ; Is this an echo request?
		164	jne .check_sockets
		165
		166	; Update stats (and validate device ptr)
		167	call NET_ptr_to_num4
		168	cmp edi, -1
		169	je .dump
		170	inc [ICMP_PACKETS_RX + edi]
		171
		172	; We well re-use the packet so we can create the response as fast as possible
		173	; Notice: this only works on pure ethernet
		174
		175	DEBUGF DEBUG_NETWORK_VERBOSE, "got echo request\n"
		176	mov [edx + ICMP_header.Type], ICMP_ECHOREPLY ; Change Packet type to reply
		177
		178	mov esi, [esp] ; Start of buffer
		179	cmp ebx, LOOPBACK_DEVICE
		180	je .loopback
		181
		182	; FIXME: dont assume device is an ethernet device!
		183
		184	; exchange dest and source address in IP header
		185	; exchange dest and source MAC in ETH header
		186	push dword [esi + ETH_header.DstMAC]
		187	push dword [esi + ETH_header.SrcMAC]
		188	pop dword [esi + ETH_header.DstMAC]
		189	pop dword [esi + ETH_header.SrcMAC]
		190	push word [esi + ETH_header.DstMAC + 4]
		191	push word [esi + ETH_header.SrcMAC + 4]
		192	pop word [esi + ETH_header.DstMAC + 4]
		193	pop word [esi + ETH_header.SrcMAC + 4]
		194	add esi, sizeof.ETH_header-4
		195
		196	.loopback:
		197	add esi, 4
		198	push [esi + IPv4_header.SourceAddress]
		199	push [esi + IPv4_header.DestinationAddress]
		200	pop [esi + IPv4_header.SourceAddress]
		201	pop [esi + IPv4_header.DestinationAddress]
		202
		203	; Recalculate ip header checksum
		204	movzx ecx, [esi + IPv4_header.VersionAndIHL] ; Calculate IP Header length by using IHL field
		205	and ecx, 0x0f
		206	shl cx, 2
		207	mov edi, ecx ; IP header length
		208	mov eax, edx ; ICMP packet start addr
		209
		210	push esi ; Calculate the IP checksum
		211	xor edx, edx ;
		212	call checksum_1 ;
		213	call checksum_2 ;
		214	pop esi ;
		215	mov [esi + IPv4_header.HeaderChecksum], dx ;
		216
		217	; Recalculate ICMP CheckSum
		218	movzx ecx, [esi + IPv4_header.TotalLength] ; Find length of IP Packet
		219	xchg ch, cl ;
		220	sub ecx, edi ; IP packet length - IP header length = ICMP packet length
		221
		222	mov esi, eax ; Calculate ICMP checksum
		223	xor edx, edx ;
		224	call checksum_1 ;
		225	call checksum_2 ;
		226	mov [eax + ICMP_header.Checksum], dx ;
		227
		228	; Transmit the packet (notice that packet ptr and packet size have been on stack since start of the procedure!)
		229	call [ebx + NET_DEVICE.transmit]
		230	test eax, eax
		231	jnz @f
		232	call NET_ptr_to_num4
		233	inc [ICMP_PACKETS_TX + edi]
		234	@@:
		235	ret
		236
		237
		238
		239
		240	.check_sockets:
		241	; Look for an open ICMP socket
		242
		243	pusha
		244	mov ecx, socket_mutex
		245	call mutex_lock
		246	popa
		247
		248	mov esi, [edi] ; ipv4 source address
		249	mov eax, net_sockets
		250	.try_more:
		251	; mov , [edx + ICMP_header.Identifier]
		252	.next_socket:
		253	mov eax, [eax + SOCKET.NextPtr]
		254	or eax, eax
		255	jz .dump_
		256
		257	cmp [eax + SOCKET.Domain], AF_INET4
		258	jne .next_socket
		259
		260	cmp [eax + SOCKET.Protocol], IP_PROTO_ICMP
		261	jne .next_socket
		262
		263	cmp [eax + IP_SOCKET.RemoteIP], esi
		264	jne .next_socket
		265
		266	; cmp [eax + ICMP_SOCKET.Identifier],
		267	; jne .next_socket
		268
		269	; Update stats (and validate device ptr)
		270	call NET_ptr_to_num4
		271	cmp edi, -1
		272	je .dump_
		273	inc [ICMP_PACKETS_RX + edi]
		274
		275	pusha
		276	mov ecx, socket_mutex
		277	call mutex_unlock
		278	popa
		279
		280	DEBUGF DEBUG_NETWORK_VERBOSE, "socket=%x\n", eax
		281
		282	pusha
		283	lea ecx, [eax + SOCKET.mutex]
		284	call mutex_lock
		285	popa
		286
		287	mov esi, edx
		288	jmp SOCKET_input
		289
		290	.dump_:
		291
		292	pusha
		293	mov ecx, socket_mutex
		294	call mutex_unlock
		295	popa
		296
4923	Serge	297	DEBUGF DEBUG_NETWORK_ERROR, "ICMP_input: no socket found\n"
4429	Serge	298	jmp .dump
		299
		300
		301	.checksum_mismatch:
4923	Serge	302	DEBUGF DEBUG_NETWORK_ERROR, "ICMP_input: checksum mismatch\n"
4429	Serge	303
		304	.dump:
		305	DEBUGF DEBUG_NETWORK_VERBOSE, "ICMP_input: dumping\n"
		306
		307	call NET_packet_free
		308	add esp, 4 ; pop (balance stack)
		309
		310	ret
		311
		312
		313	if 0
		314	;-----------------------------------------------------------------
		315	;
		316	; ICMP_output
		317	;
		318	; IN: eax = dest ip
		319	; bh = type
		320	; bl = code
		321	; ecx = data length
		322	; edx = source ip
		323	; esi = data offset
		324	; edi = identifier shl 16 + sequence number
		325	;
		326	;-----------------------------------------------------------------
		327	align 4
		328	ICMP_output:
		329
		330	DEBUGF DEBUG_NETWORK_VERBOSE, "Creating ICMP Packet\n"
		331
		332	push esi edi bx
		333	add ecx, sizeof.ICMP_header
		334	mov di, IP_PROTO_ICMP SHL 8 + 128 ; TTL
		335	call IPv4_output
		336	jz .exit
		337
		338	DEBUGF DEBUG_NETWORK_VERBOSE, "full icmp packet size: %u\n", edx
		339
		340	pop word [edi + ICMP_header.Type] ; Write both type and code bytes at once
		341	pop dword [edi + ICMP_header.Identifier] ; identifier and sequence number
		342	mov [edi + ICMP_header.Checksum], 0
		343
		344	push ebx ecx edx
		345	mov esi, edi
		346	xor edx, edx
		347	call checksum_1
		348	call checksum_2
		349	mov [edi + ICMP_header.Checksum], dx
		350	pop edx ecx ebx esi
		351
		352	sub ecx, sizeof.ICMP_header
		353	add edi, sizeof.ICMP_header
		354	push cx
		355	shr cx, 2
		356	rep movsd
		357	pop cx
		358	and cx, 3
		359	rep movsb
		360
		361	sub edi, edx ;;; TODO: find a better way to remember start of packet
		362	push edx edi
		363	DEBUGF DEBUG_NETWORK_VERBOSE, "Sending ICMP Packet\n"
		364	call [ebx + NET_DEVICE.transmit]
		365	test eax, eax
		366	jnz @f
		367	call NET_ptr_to_num4
		368	inc [ICMP_PACKETS_TX + edi]
		369	@@:
		370	ret
		371	.exit:
		372	DEBUGF DEBUG_NETWORK_ERROR, "Creating ICMP Packet failed\n"
		373	add esp, 2*4 + 2
		374	ret
		375	end if
		376
		377
		378
		379
		380	;-----------------------------------------------------------------
		381	;
		382	; ICMP_output_raw
		383	;
		384	; IN: eax = socket ptr
		385	; ecx = data length
		386	; esi = data offset
		387	;
		388	;-----------------------------------------------------------------
		389	align 4
		390	ICMP_output_raw:
		391
		392	DEBUGF DEBUG_NETWORK_VERBOSE, "Creating ICMP Packet for socket %x, data ptr=%x\n", eax, edx
		393
		394	push edx
		395
		396	mov di, IP_PROTO_ICMP SHL 8 + 128 ; TTL
		397	mov edx, [eax + IP_SOCKET.LocalIP]
		398	mov eax, [eax + IP_SOCKET.RemoteIP]
		399	call IPv4_output
		400	jz .exit
		401
		402	pop esi
		403	push edx
		404	push eax
		405
		406	push edi ecx
		407	DEBUGF DEBUG_NETWORK_VERBOSE, "copying %u bytes from %x to %x\n", ecx, esi, edi
		408	rep movsb
		409	pop ecx edi
		410
		411	mov [edi + ICMP_header.Checksum], 0
		412
		413	mov esi, edi
		414	xor edx, edx
		415	call checksum_1
		416	call checksum_2
		417	mov [edi + ICMP_header.Checksum], dx
		418
		419	DEBUGF DEBUG_NETWORK_VERBOSE, "Sending ICMP Packet\n"
		420	call [ebx + NET_DEVICE.transmit]
		421	test eax, eax
		422	jnz @f
		423	call NET_ptr_to_num4
		424	inc [ICMP_PACKETS_TX + edi]
		425	@@:
		426	ret
		427	.exit:
		428	DEBUGF DEBUG_NETWORK_ERROR, "Creating ICMP Packet failed\n"
		429	add esp, 4
		430	ret
		431
		432
		433
		434
		435	;-----------------------------------------------------------------
		436	;
		437	; ICMP_API
		438	;
		439	; This function is called by system function 75
		440	;
		441	; IN: subfunction number in bl
		442	; device number in bh
		443	; ecx, edx, .. depends on subfunction
		444	;
		445	; OUT:
		446	;
		447	;-----------------------------------------------------------------
		448	align 4
		449	ICMP_api:
		450
		451	movzx eax, bh
		452	shl eax, 2
		453
		454	test bl, bl
		455	jz .packets_tx ; 0
		456	dec bl
		457	jz .packets_rx ; 1
		458
		459	.error:
		460	mov eax, -1
		461	ret
		462
		463	.packets_tx:
		464	mov eax, [ICMP_PACKETS_TX + eax]
		465	ret
		466
		467	.packets_rx:
		468	mov eax, [ICMP_PACKETS_RX + eax]
		469	ret

Subversion Repositories Kolibri OS

(root)/kernel/branches/kolibri-process/network/icmp.inc – Rev 4923