Rev 1196 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
1159 | hidnplayr | 1 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
2 | ;; ;; |
||
1196 | hidnplayr | 3 | ;; Copyright (C) KolibriOS team 2004-2009. All rights reserved. ;; |
1159 | hidnplayr | 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 | |||
1206 | hidnplayr | 20 | $Revision: 1206 $ |
1159 | hidnplayr | 21 | |
22 | ; ICMP types & codes |
||
23 | |||
24 | ICMP_ECHOREPLY equ 0 ; echo reply message |
||
25 | |||
26 | ICMP_UNREACH equ 3 |
||
27 | ICMP_UNREACH_NET equ 0 ; bad net |
||
28 | ICMP_UNREACH_HOST equ 1 ; bad host |
||
29 | ICMP_UNREACH_PROTOCOL equ 2 ; bad protocol |
||
30 | ICMP_UNREACH_PORT equ 3 ; bad port |
||
31 | ICMP_UNREACH_NEEDFRAG equ 4 ; IP_DF caused drop |
||
32 | ICMP_UNREACH_SRCFAIL equ 5 ; src route failed |
||
33 | ICMP_UNREACH_NET_UNKNOWN equ 6 ; unknown net |
||
34 | ICMP_UNREACH_HOST_UNKNOWN equ 7 ; unknown host |
||
35 | ICMP_UNREACH_ISOLATED equ 8 ; src host isolated |
||
36 | ICMP_UNREACH_NET_PROHIB equ 9 ; prohibited access |
||
37 | ICMP_UNREACH_HOST_PROHIB equ 10 ; ditto |
||
38 | ICMP_UNREACH_TOSNET equ 11 ; bad tos for net |
||
39 | ICMP_UNREACH_TOSHOST equ 12 ; bad tos for host |
||
40 | ICMP_UNREACH_FILTER_PROHIB equ 13 ; admin prohib |
||
41 | ICMP_UNREACH_HOST_PRECEDENCE equ 14 ; host prec vio. |
||
42 | ICMP_UNREACH_PRECEDENCE_CUTOFF equ 15 ; prec cutoff |
||
43 | |||
44 | ICMP_SOURCEQUENCH equ 4 ; Packet lost, slow down |
||
45 | |||
46 | ICMP_REDIRECT equ 5 ; shorter route, codes: |
||
47 | ICMP_REDIRECT_NET equ 0 ; for network |
||
48 | ICMP_REDIRECT_HOST equ 1 ; for host |
||
49 | ICMP_REDIRECT_TOSNET equ 2 ; for tos and net |
||
50 | ICMP_REDIRECT_TOSHOST equ 3 ; for tos and host |
||
51 | |||
52 | ICMP_ALTHOSTADDR equ 6 ; alternate host address |
||
53 | ICMP_ECHO equ 8 ; echo service |
||
54 | ICMP_ROUTERADVERT equ 9 ; router advertisement |
||
55 | ICMP_ROUTERADVERT_NORMAL equ 0 ; normal advertisement |
||
56 | ICMP_ROUTERADVERT_NOROUTE_COMMON equ 16 ; selective routing |
||
57 | |||
58 | ICMP_ROUTERSOLICIT equ 10 ; router solicitation |
||
59 | ICMP_TIMXCEED equ 11 ; time exceeded, code: |
||
60 | ICMP_TIMXCEED_INTRANS equ 0 ; ttl==0 in transit |
||
61 | ICMP_TIMXCEED_REASS equ 1 ; ttl==0 in reass |
||
62 | |||
63 | ICMP_PARAMPROB equ 12 ; ip header bad |
||
64 | ICMP_PARAMPROB_ERRATPTR equ 0 ; error at param ptr |
||
65 | ICMP_PARAMPROB_OPTABSENT equ 1 ; req. opt. absent |
||
66 | ICMP_PARAMPROB_LENGTH equ 2 ; bad length |
||
67 | |||
68 | ICMP_TSTAMP equ 13 ; timestamp request |
||
69 | ICMP_TSTAMPREPLY equ 14 ; timestamp reply |
||
70 | ICMP_IREQ equ 15 ; information request |
||
71 | ICMP_IREQREPLY equ 16 ; information reply |
||
72 | ICMP_MASKREQ equ 17 ; address mask request |
||
73 | ICMP_MASKREPLY equ 18 ; address mask reply |
||
74 | ICMP_TRACEROUTE equ 30 ; traceroute |
||
75 | ICMP_DATACONVERR equ 31 ; data conversion error |
||
76 | ICMP_MOBILE_REDIRECT equ 32 ; mobile host redirect |
||
77 | ICMP_IPV6_WHEREAREYOU equ 33 ; IPv6 where-are-you |
||
78 | ICMP_IPV6_IAMHERE equ 34 ; IPv6 i-am-here |
||
79 | ICMP_MOBILE_REGREQUEST equ 35 ; mobile registration req |
||
80 | ICMP_MOBILE_REGREPLY equ 36 ; mobile registreation reply |
||
81 | ICMP_SKIP equ 39 ; SKIP |
||
82 | |||
83 | ICMP_PHOTURIS equ 40 ; Photuris |
||
84 | ICMP_PHOTURIS_UNKNOWN_INDEX equ 1 ; unknown sec index |
||
85 | ICMP_PHOTURIS_AUTH_FAILED equ 2 ; auth failed |
||
86 | ICMP_PHOTURIS_DECRYPT_FAILED equ 3 ; decrypt failed |
||
87 | |||
88 | |||
89 | |||
90 | struct ICMP_Packet |
||
91 | .Type db ? |
||
92 | .Code db ? |
||
93 | .Checksum dw ? |
||
94 | .Identifier dw ? |
||
95 | .SequenceNumber dw ? |
||
96 | .Data: |
||
97 | ends |
||
98 | |||
99 | |||
100 | align 4 |
||
101 | uglobal |
||
102 | ICMP_PACKETS_TX rd MAX_IP |
||
103 | ICMP_PACKETS_RX rd MAX_IP |
||
104 | endg |
||
105 | |||
1185 | hidnplayr | 106 | |
107 | |||
1159 | hidnplayr | 108 | ;----------------------------------------------------------------- |
109 | ; |
||
110 | ; ICMP_init |
||
111 | ; |
||
112 | ; This function resets all ICMP variables |
||
113 | ; |
||
114 | ; IN: / |
||
115 | ; OUT: / |
||
116 | ; |
||
117 | ;----------------------------------------------------------------- |
||
118 | |||
119 | align 4 |
||
120 | ICMP_init: |
||
121 | |||
122 | xor eax, eax |
||
123 | mov edi, ICMP_PACKETS_TX |
||
124 | mov ecx, 2*MAX_IP |
||
125 | rep stosd |
||
126 | |||
127 | ret |
||
1171 | hidnplayr | 128 | |
129 | |||
130 | |||
131 | |||
1159 | hidnplayr | 132 | ;-------------------------------- |
133 | ; |
||
134 | ; ICMP_Handler: |
||
135 | ; |
||
136 | ; Called by IP_handler, |
||
137 | ; this procedure will send reply's to ICMP echo's etc ;;; TODO: update this to work with fragmented packets too! |
||
138 | ; |
||
139 | ; IN: Pointer to buffer in [esp] |
||
140 | ; size of buffer in [esp+4] |
||
141 | ; pointer to device struct in ebx |
||
142 | ; ICMP Packet size in ecx |
||
143 | ; pointer to ICMP Packet data in edx |
||
144 | ; OUT: / |
||
145 | ; |
||
146 | ;-------------------------------- |
||
147 | |||
148 | align 4 |
||
1196 | hidnplayr | 149 | ICMP_handler: ;TODO: works only on pure ethernet right now ! |
1159 | hidnplayr | 150 | |
151 | DEBUGF 1,"ICMP_Handler - start\n" |
||
1185 | hidnplayr | 152 | cmp byte [edx + ICMP_Packet.Type], ICMP_ECHO ; Is this an echo request? |
1159 | hidnplayr | 153 | jne .check_sockets |
154 | |||
155 | mov byte [edx + ICMP_Packet.Type], ICMP_ECHOREPLY ; Change Packet type to reply |
||
156 | mov word [edx + ICMP_Packet.Checksum], 0 ; Set checksum to 0, needed to calculate new checksum |
||
157 | |||
1171 | hidnplayr | 158 | call ETH_struc2dev |
159 | cmp edi,-1 |
||
160 | je .dump |
||
161 | inc [ICMP_PACKETS_RX+4*edi] |
||
1185 | hidnplayr | 162 | inc [ICMP_PACKETS_TX+4*edi] |
1171 | hidnplayr | 163 | |
1159 | hidnplayr | 164 | ; exchange dest and source address in IP header |
165 | ; exchange dest and source MAC in ETH header |
||
166 | mov esi, [esp] |
||
167 | |||
168 | mov eax, dword [esi + ETH_FRAME.DstMAC] |
||
169 | mov ecx, dword [esi + ETH_FRAME.SrcMAC] |
||
170 | mov dword [esi + ETH_FRAME.SrcMAC], eax |
||
171 | mov dword [esi + ETH_FRAME.DstMAC], ecx |
||
172 | |||
173 | mov ax, word [esi + ETH_FRAME.DstMAC + 4] |
||
174 | mov cx, word [esi + ETH_FRAME.SrcMAC + 4] |
||
175 | mov word [esi + ETH_FRAME.SrcMAC + 4], ax |
||
176 | mov word [esi + ETH_FRAME.DstMAC + 4], cx |
||
177 | |||
178 | mov eax, dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress] |
||
179 | mov ecx, dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress] |
||
180 | mov dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress], eax |
||
181 | mov dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress], ecx |
||
182 | |||
183 | ; Recalculate ip header checksum |
||
184 | ; mov esi, [esp] |
||
185 | add esi, ETH_FRAME.Data ; Point esi to start of IP Packet |
||
186 | movzx eax, byte [esi + IPv4_Packet.VersionAndIHL] ; Calculate IP Header length by using IHL field |
||
187 | and eax, 0x0000000F ; |
||
188 | shl eax, 2 ; |
||
189 | push ebx edx esi |
||
190 | stdcall checksum_jb, esi, eax ; calculate the checksum |
||
191 | pop esi edx ebx |
||
192 | xchg al, ah ; convert to intel byte order |
||
193 | ; mov esi, [esp] |
||
194 | mov word [esi + IPv4_Packet.HeaderChecksum], ax ; Store it in the IP Packet header |
||
195 | |||
196 | ; Recalculate ICMP CheckSum |
||
197 | ; mov esi, [esp] ; Find length of IP Packet |
||
198 | movzx eax, word[esi + IPv4_Packet.TotalLength] ; |
||
199 | xchg ah , al ; |
||
200 | movzx edi, byte [esi + IPv4_Packet.VersionAndIHL] ; Calculate IP Header length by using IHL field |
||
201 | and edi, 0x0000000F ; |
||
202 | shl edi, 2 ; |
||
203 | sub ax , di ; Now we know the length of ICMP data in eax |
||
204 | push ebx edx |
||
205 | stdcall checksum_jb,edx,eax ; Calculate the checksum of icmp data |
||
206 | pop edx ebx |
||
207 | xchg al, ah ; Convert to intel byte order |
||
208 | mov word [edx + ICMP_Packet.Checksum], ax |
||
209 | |||
1171 | hidnplayr | 210 | jmp ETH_Sender ; Send the reply |
1159 | hidnplayr | 211 | |
1171 | hidnplayr | 212 | |
213 | |||
214 | |||
215 | |||
1159 | hidnplayr | 216 | .check_sockets: |
1206 | hidnplayr | 217 | ; TODO: validate the header & checksum. |
1159 | hidnplayr | 218 | |
219 | ; Look for an open ICMP socket |
||
220 | |||
221 | mov esi, net_sockets |
||
222 | .try_more: |
||
223 | mov ax , [edx + ICMP_Packet.Identifier] |
||
224 | .next_socket: |
||
225 | mov esi, [esi + SOCKET.NextPtr] |
||
226 | or esi, esi |
||
227 | jz .dump |
||
228 | cmp [esi + SOCKET.Type], IP_PROTO_ICMP |
||
229 | jne .next_socket |
||
230 | cmp [esi + SOCKET.LocalPort], ax |
||
231 | jne .next_socket |
||
232 | |||
233 | cmp [esi + SOCKET.rxDataCount],0 ; get # of bytes already in buffer |
||
234 | jnz .dump ; only one packet at a time may be in the buffer! |
||
235 | |||
236 | cmp ecx, SOCKETBUFFSIZE - SOCKETHEADERSIZE; TODO: fix this problem ! |
||
237 | jg .dump |
||
238 | |||
1171 | hidnplayr | 239 | call IPv4_dest_to_dev |
240 | cmp edi,-1 |
||
241 | je .dump |
||
242 | inc [ICMP_PACKETS_RX+4*edi] |
||
243 | |||
1159 | hidnplayr | 244 | DEBUGF 1,"Found valid ICMP packet for socket %x\n", esi |
245 | |||
246 | lea ebx, [esi + SOCKET.lock] |
||
247 | call wait_mutex |
||
248 | |||
249 | ; Now, copy data to socket. We have socket address in esi. |
||
250 | ; We have ICMP Packet in edx |
||
251 | ; number of bytes in ecx |
||
252 | |||
253 | ; note: we do not strip the header! |
||
254 | |||
255 | DEBUGF 1,"bytes: %u\n", ecx |
||
256 | |||
257 | mov [esi + SOCKET.rxDataCount], ecx |
||
258 | |||
259 | lea edi, [esi + SOCKETHEADERSIZE] |
||
260 | push esi |
||
261 | push ecx |
||
262 | mov esi, edx |
||
263 | shr ecx, 2 |
||
264 | rep movsd ; copy the data across |
||
265 | pop ecx |
||
266 | and ecx, 3 |
||
267 | rep movsb |
||
268 | pop esi |
||
269 | |||
270 | DEBUGF 1,"ICMP socket updated\n" |
||
271 | |||
272 | mov [esi + SOCKET.lock], 0 |
||
273 | |||
274 | ; flag an event to the application |
||
275 | mov eax, [esi + SOCKET.PID] ; get socket owner PID |
||
276 | mov ecx, 1 |
||
277 | mov esi, TASK_DATA + TASKDATA.pid |
||
278 | |||
279 | .next_pid: |
||
280 | cmp [esi], eax |
||
281 | je .found_pid |
||
282 | inc ecx |
||
283 | add esi, 0x20 |
||
284 | cmp ecx, [TASK_COUNT] |
||
285 | jbe .next_pid |
||
286 | |||
287 | jmp .dump |
||
288 | |||
289 | .found_pid: |
||
290 | shl ecx, 8 |
||
291 | or [ecx + SLOT_BASE + APPDATA.event_mask], EVENT_NETWORK ; stack event |
||
292 | |||
293 | mov [check_idle_semaphore], 200 |
||
294 | |||
295 | .dump: |
||
296 | DEBUGF 1,"ICMP_Handler - dumping\n" |
||
297 | |||
298 | call kernel_free |
||
1165 | hidnplayr | 299 | add esp, 8 ; pop (balance stack) |
1159 | hidnplayr | 300 | |
301 | ret |
||
302 | |||
303 | |||
304 | ;-------------------------------- |
||
305 | ; |
||
306 | ; ICMP_Handler_fragments: |
||
307 | ; |
||
308 | ; Called by IP_handler, |
||
309 | ; this procedure will send reply's to ICMP echo's etc |
||
310 | ; |
||
311 | ; IN: Pointer to buffer in [esp] |
||
312 | ; size of buffer in [esp+4] |
||
313 | ; pointer to device struct in ebx |
||
314 | ; ICMP Packet size in ecx |
||
315 | ; pointer to ICMP Packet data in edx |
||
316 | ; OUT: / |
||
317 | ; |
||
318 | ;-------------------------------- |
||
319 | |||
320 | align 4 |
||
1196 | hidnplayr | 321 | ICMP_handler_fragments: ; works only on pure ethernet right now ! |
1159 | hidnplayr | 322 | |
323 | DEBUGF 1,"ICMP_Handler_fragments - start\n" |
||
324 | |||
325 | cmp ecx, 65500 |
||
326 | jg .dump |
||
327 | |||
328 | cmp byte [edx + ICMP_Packet.Type], ICMP_ECHO ; Is this an echo request? discard if not |
||
329 | jne .dump |
||
330 | |||
331 | mov esi, [esp] |
||
332 | |||
333 | sub ecx, ICMP_Packet.Data |
||
334 | mov eax, [esi + IPv4_Packet.SourceAddress] |
||
335 | mov ebx, [esi + IPv4_Packet.DestinationAddress] |
||
336 | push word [esi + IPv4_Packet.Identification] |
||
337 | |||
338 | mov di , [edx + ICMP_Packet.Identifier] |
||
339 | shl edi, 16 |
||
340 | mov di , [edx + ICMP_Packet.SequenceNumber] |
||
341 | |||
342 | mov esi, edx |
||
343 | add esi, ICMP_Packet.Data |
||
344 | pop dx |
||
345 | shl edx, 16 |
||
346 | mov dx , ICMP_ECHOREPLY shl 8 + 0 ; Type + Code |
||
347 | |||
1196 | hidnplayr | 348 | call ICMP_create_packet |
1159 | hidnplayr | 349 | |
350 | .dump: |
||
351 | DEBUGF 1,"ICMP_Handler_fragments - end\n" |
||
352 | |||
353 | call kernel_free |
||
1165 | hidnplayr | 354 | add esp, 8 ; pop (balance stack) |
1159 | hidnplayr | 355 | ret |
356 | |||
357 | ;----------------------------------------------------------------- |
||
358 | ; |
||
359 | ; Note: ICMP only works on top of IP protocol :) |
||
360 | ; |
||
361 | ; inputs: |
||
362 | ; |
||
363 | ; eax = dest ip |
||
364 | ; ebx = source ip |
||
365 | ; ecx = data length |
||
366 | ; dh = type |
||
367 | ; dl = code |
||
368 | ; high 16 bits of edx = fragment id (for IP header) |
||
369 | ; esi = data offset |
||
370 | ; edi = identifier shl 16 + sequence number |
||
371 | ; |
||
372 | ;----------------------------------------------------------------- |
||
373 | |||
374 | align 4 |
||
1196 | hidnplayr | 375 | ICMP_create_packet: |
1159 | hidnplayr | 376 | |
377 | DEBUGF 1,"Create ICMP Packet\n" |
||
378 | |||
379 | push esi edi edx |
||
380 | |||
381 | add ecx, ICMP_Packet.Data |
||
382 | mov di , IP_PROTO_ICMP |
||
383 | shr edx, 16 |
||
384 | |||
1196 | hidnplayr | 385 | call IPv4_create_packet |
1159 | hidnplayr | 386 | |
387 | cmp edi, -1 |
||
388 | je .exit |
||
389 | |||
1206 | hidnplayr | 390 | DEBUGF 1,"full icmp packet size: %u\n", edx |
1165 | hidnplayr | 391 | |
1159 | hidnplayr | 392 | pop eax |
393 | mov word [edi + ICMP_Packet.Type], ax ; Write both type and code bytes at once |
||
394 | pop eax |
||
395 | mov [edi + ICMP_Packet.SequenceNumber], ax |
||
396 | shr eax, 16 |
||
397 | mov [edi + ICMP_Packet.Identifier], ax |
||
398 | mov [edi + ICMP_Packet.Checksum], 0 |
||
399 | |||
400 | stdcall checksum_jb, edi , ecx |
||
401 | xchg al, ah |
||
402 | mov [edi + ICMP_Packet.Checksum], ax |
||
403 | |||
404 | pop esi |
||
405 | sub ecx, ICMP_Packet.Data |
||
406 | add edi, ICMP_Packet.Data |
||
407 | push cx |
||
408 | shr cx , 2 |
||
409 | rep movsd |
||
410 | pop cx |
||
411 | and cx , 3 |
||
412 | rep movsb |
||
413 | |||
1206 | hidnplayr | 414 | sub edi, edx ;; TODO: find a better way to remember start of packet |
1159 | hidnplayr | 415 | mov ecx, [ebx + ETH_DEVICE.transmit] |
416 | push edx edi ecx |
||
417 | DEBUGF 1,"Sending ICMP Packet\n" |
||
418 | ret ; Send the packet (create_packet routine outputs pointer to routine to send packet in eax) |
||
419 | |||
420 | .exit: |
||
1165 | hidnplayr | 421 | DEBUGF 1,"Creating ICMP Packet failed\n" |
422 | add esp, 3*4 |
||
1159 | hidnplayr | 423 | ret |
424 | |||
425 | |||
426 | |||
427 | |||
428 | ;--------------------------------------------------------------------------- |
||
429 | ; |
||
430 | ; ICMP_API |
||
431 | ; |
||
432 | ; This function is called by system function 75 |
||
433 | ; |
||
434 | ; IN: subfunction number in bl |
||
435 | ; device number in bh |
||
436 | ; ecx, edx, .. depends on subfunction |
||
437 | ; |
||
438 | ; OUT: |
||
439 | ; |
||
440 | ;--------------------------------------------------------------------------- |
||
441 | |||
442 | align 4 |
||
443 | ICMP_API: |
||
444 | |||
445 | movzx eax, bh |
||
446 | shl eax, 2 |
||
447 | |||
448 | test bl, bl |
||
449 | jz .packets_tx ; 0 |
||
450 | dec bl |
||
451 | jz .packets_rx ; 1 |
||
452 | |||
453 | .error: |
||
454 | mov eax, -1 |
||
455 | ret |
||
456 | |||
457 | .packets_tx: |
||
458 | add eax, ICMP_PACKETS_TX |
||
459 | mov eax, [eax] |
||
460 | ret |
||
461 | |||
462 | .packets_rx: |
||
463 | add eax, ICMP_PACKETS_RX |
||
464 | mov eax, [eax] |
||
465 | ret |