mirror of
https://github.com/KolibriOS/kolibrios.git
synced 2024-12-16 20:02:35 +03:00
fab920c5c8
Refactoring of ARP code. Used universal names for protocol handlers. Moved some pieces of code from procs to macros. Added IPv4_fragment (currently disabled) Added ring-buffer functions for sockets Updates in TCP code git-svn-id: svn://kolibrios.org@1529 a494cfbc-eb01-0410-851d-a64ba20cac60
365 lines
10 KiB
PHP
365 lines
10 KiB
PHP
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;; ;;
|
|
;; Copyright (C) KolibriOS team 2004-2010. All rights reserved. ;;
|
|
;; Distributed under terms of the GNU General Public License ;;
|
|
;; ;;
|
|
;; ICMP.INC ;;
|
|
;; ;;
|
|
;; Part of the tcp/ip network stack for KolibriOS ;;
|
|
;; ;;
|
|
;; Based on the work of [Johnny_B] and [smb] ;;
|
|
;; ;;
|
|
;; Written by hidnplayr@kolibrios.org ;;
|
|
;; ;;
|
|
;; GNU GENERAL PUBLIC LICENSE ;;
|
|
;; Version 2, June 1991 ;;
|
|
;; ;;
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
|
|
$Revision$
|
|
|
|
; ICMP types & codes
|
|
|
|
ICMP_ECHOREPLY equ 0 ; echo reply message
|
|
|
|
ICMP_UNREACH equ 3
|
|
ICMP_UNREACH_NET equ 0 ; bad net
|
|
ICMP_UNREACH_HOST equ 1 ; bad host
|
|
ICMP_UNREACH_PROTOCOL equ 2 ; bad protocol
|
|
ICMP_UNREACH_PORT equ 3 ; bad port
|
|
ICMP_UNREACH_NEEDFRAG equ 4 ; IP_DF caused drop
|
|
ICMP_UNREACH_SRCFAIL equ 5 ; src route failed
|
|
ICMP_UNREACH_NET_UNKNOWN equ 6 ; unknown net
|
|
ICMP_UNREACH_HOST_UNKNOWN equ 7 ; unknown host
|
|
ICMP_UNREACH_ISOLATED equ 8 ; src host isolated
|
|
ICMP_UNREACH_NET_PROHIB equ 9 ; prohibited access
|
|
ICMP_UNREACH_HOST_PROHIB equ 10 ; ditto
|
|
ICMP_UNREACH_TOSNET equ 11 ; bad tos for net
|
|
ICMP_UNREACH_TOSHOST equ 12 ; bad tos for host
|
|
ICMP_UNREACH_FILTER_PROHIB equ 13 ; admin prohib
|
|
ICMP_UNREACH_HOST_PRECEDENCE equ 14 ; host prec vio.
|
|
ICMP_UNREACH_PRECEDENCE_CUTOFF equ 15 ; prec cutoff
|
|
|
|
ICMP_SOURCEQUENCH equ 4 ; Packet lost, slow down
|
|
|
|
ICMP_REDIRECT equ 5 ; shorter route, codes:
|
|
ICMP_REDIRECT_NET equ 0 ; for network
|
|
ICMP_REDIRECT_HOST equ 1 ; for host
|
|
ICMP_REDIRECT_TOSNET equ 2 ; for tos and net
|
|
ICMP_REDIRECT_TOSHOST equ 3 ; for tos and host
|
|
|
|
ICMP_ALTHOSTADDR equ 6 ; alternate host address
|
|
ICMP_ECHO equ 8 ; echo service
|
|
ICMP_ROUTERADVERT equ 9 ; router advertisement
|
|
ICMP_ROUTERADVERT_NORMAL equ 0 ; normal advertisement
|
|
ICMP_ROUTERADVERT_NOROUTE_COMMON equ 16 ; selective routing
|
|
|
|
ICMP_ROUTERSOLICIT equ 10 ; router solicitation
|
|
ICMP_TIMXCEED equ 11 ; time exceeded, code:
|
|
ICMP_TIMXCEED_INTRANS equ 0 ; ttl==0 in transit
|
|
ICMP_TIMXCEED_REASS equ 1 ; ttl==0 in reass
|
|
|
|
ICMP_PARAMPROB equ 12 ; ip header bad
|
|
ICMP_PARAMPROB_ERRATPTR equ 0 ; error at param ptr
|
|
ICMP_PARAMPROB_OPTABSENT equ 1 ; req. opt. absent
|
|
ICMP_PARAMPROB_LENGTH equ 2 ; bad length
|
|
|
|
ICMP_TSTAMP equ 13 ; timestamp request
|
|
ICMP_TSTAMPREPLY equ 14 ; timestamp reply
|
|
ICMP_IREQ equ 15 ; information request
|
|
ICMP_IREQREPLY equ 16 ; information reply
|
|
ICMP_MASKREQ equ 17 ; address mask request
|
|
ICMP_MASKREPLY equ 18 ; address mask reply
|
|
ICMP_TRACEROUTE equ 30 ; traceroute
|
|
ICMP_DATACONVERR equ 31 ; data conversion error
|
|
ICMP_MOBILE_REDIRECT equ 32 ; mobile host redirect
|
|
ICMP_IPV6_WHEREAREYOU equ 33 ; IPv6 where-are-you
|
|
ICMP_IPV6_IAMHERE equ 34 ; IPv6 i-am-here
|
|
ICMP_MOBILE_REGREQUEST equ 35 ; mobile registration req
|
|
ICMP_MOBILE_REGREPLY equ 36 ; mobile registreation reply
|
|
ICMP_SKIP equ 39 ; SKIP
|
|
|
|
ICMP_PHOTURIS equ 40 ; Photuris
|
|
ICMP_PHOTURIS_UNKNOWN_INDEX equ 1 ; unknown sec index
|
|
ICMP_PHOTURIS_AUTH_FAILED equ 2 ; auth failed
|
|
ICMP_PHOTURIS_DECRYPT_FAILED equ 3 ; decrypt failed
|
|
|
|
|
|
|
|
struct ICMP_Packet
|
|
.Type db ?
|
|
.Code db ?
|
|
.Checksum dw ?
|
|
.Identifier dw ?
|
|
.SequenceNumber dw ?
|
|
.Data:
|
|
ends
|
|
|
|
|
|
align 4
|
|
uglobal
|
|
ICMP_PACKETS_TX rd MAX_IP
|
|
ICMP_PACKETS_RX rd MAX_IP
|
|
endg
|
|
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
;
|
|
; ICMP_init
|
|
;
|
|
;-----------------------------------------------------------------
|
|
|
|
macro ICMP_init {
|
|
|
|
xor eax, eax
|
|
mov edi, ICMP_PACKETS_TX
|
|
mov ecx, 2*MAX_IP
|
|
rep stosd
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
;
|
|
; ICMP_input:
|
|
;
|
|
; This procedure will send reply's to ICMP echo's
|
|
; and insert packets into sockets when needed
|
|
;
|
|
; IN: Pointer to buffer in [esp]
|
|
; size of buffer in [esp+4]
|
|
; pointer to device struct in ebx
|
|
; ICMP Packet size in ecx
|
|
; pointer to ICMP Packet data in edx
|
|
; OUT: /
|
|
;
|
|
;-----------------------------------------------------------------
|
|
align 4
|
|
ICMP_input:
|
|
|
|
;;; TODO: works only on pure ethernet right now !
|
|
|
|
DEBUGF 1,"ICMP_Handler - start\n"
|
|
cmp byte [edx + ICMP_Packet.Type], ICMP_ECHO ; Is this an echo request?
|
|
jne .check_sockets
|
|
|
|
;;; TODO: check checksum!
|
|
|
|
DEBUGF 1,"ICMP_Handler - echo request\n"
|
|
|
|
mov byte [edx + ICMP_Packet.Type], ICMP_ECHOREPLY ; Change Packet type to reply
|
|
mov word [edx + ICMP_Packet.Checksum], 0 ; Set checksum to 0, needed to calculate new checksum
|
|
|
|
call NET_ptr_to_num
|
|
cmp edi,-1
|
|
je .dump
|
|
inc [ICMP_PACKETS_RX+4*edi]
|
|
inc [ICMP_PACKETS_TX+4*edi]
|
|
|
|
; exchange dest and source address in IP header
|
|
; exchange dest and source MAC in ETH header
|
|
mov esi, [esp]
|
|
|
|
mov eax, dword [esi + ETH_FRAME.DstMAC]
|
|
mov ecx, dword [esi + ETH_FRAME.SrcMAC]
|
|
mov dword [esi + ETH_FRAME.SrcMAC], eax
|
|
mov dword [esi + ETH_FRAME.DstMAC], ecx
|
|
|
|
mov ax, word [esi + ETH_FRAME.DstMAC + 4]
|
|
mov cx, word [esi + ETH_FRAME.SrcMAC + 4]
|
|
mov word [esi + ETH_FRAME.SrcMAC + 4], ax
|
|
mov word [esi + ETH_FRAME.DstMAC + 4], cx
|
|
|
|
mov eax, dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress]
|
|
mov ecx, dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress]
|
|
mov dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress], eax
|
|
mov dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress], ecx
|
|
|
|
; Recalculate ip header checksum
|
|
add esi, ETH_FRAME.Data ; Point esi to start of IP Packet
|
|
movzx ecx, byte [esi + IPv4_Packet.VersionAndIHL] ; Calculate IP Header length by using IHL field
|
|
and ecx, 0x0000000F ;
|
|
shl cx , 2
|
|
push ebx edx ecx esi
|
|
xor edx, edx
|
|
call checksum_1
|
|
call checksum_2
|
|
pop esi
|
|
mov word [esi + IPv4_Packet.HeaderChecksum], dx ; Store it in the IP Packet header
|
|
|
|
; Recalculate ICMP CheckSum
|
|
movzx eax, word[esi + IPv4_Packet.TotalLength] ; Find length of IP Packet
|
|
xchg ah , al ;
|
|
sub eax, [esp] ; Now we know the length of ICMP data in eax
|
|
mov ecx, eax
|
|
mov esi, [esp + 4]
|
|
xor edx, edx
|
|
call checksum_1
|
|
call checksum_2
|
|
mov ax , dx
|
|
pop ecx edx ebx
|
|
mov word [edx + ICMP_Packet.Checksum], ax
|
|
|
|
call [ebx + NET_DEVICE.transmit]
|
|
ret
|
|
|
|
|
|
|
|
|
|
.check_sockets:
|
|
; TODO: validate the header & checksum.
|
|
|
|
; Look for an open ICMP socket
|
|
|
|
mov esi, net_sockets
|
|
.try_more:
|
|
mov ax , [edx + ICMP_Packet.Identifier]
|
|
.next_socket:
|
|
mov esi, [esi + SOCKET.NextPtr]
|
|
or esi, esi
|
|
jz .dump
|
|
cmp [esi + SOCKET.Type], IP_PROTO_ICMP
|
|
jne .next_socket
|
|
cmp [esi + ICMP_SOCKET.Identifier], ax
|
|
jne .next_socket
|
|
|
|
call IPv4_dest_to_dev
|
|
cmp edi,-1
|
|
je .dump
|
|
inc [ICMP_PACKETS_RX+edi]
|
|
|
|
DEBUGF 1,"Found valid ICMP packet for socket %x\n", esi
|
|
|
|
lea ebx, [esi + SOCKET.lock]
|
|
call wait_mutex
|
|
|
|
; Now, assign data to socket. We have socket address in esi.
|
|
; We have ICMP Packet in edx
|
|
; number of bytes in ecx
|
|
|
|
mov eax, esi
|
|
pop esi
|
|
add esp, 4
|
|
sub edx, esi
|
|
mov edi, edx
|
|
jmp SOCKET_input
|
|
|
|
.dump:
|
|
DEBUGF 1,"ICMP_Handler - dumping\n"
|
|
|
|
call kernel_free
|
|
add esp, 4 ; pop (balance stack)
|
|
|
|
ret
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
;
|
|
; ICMP_output
|
|
;
|
|
; IN: eax = dest ip
|
|
; ebx = source ip
|
|
; ecx = data length
|
|
; dh = type
|
|
; dl = code
|
|
; high 16 bits of edx = fragment id (for IP header)
|
|
; esi = data offset
|
|
; edi = identifier shl 16 + sequence number
|
|
;
|
|
;-----------------------------------------------------------------
|
|
align 4
|
|
ICMP_output:
|
|
|
|
DEBUGF 1,"Creating ICMP Packet\n"
|
|
|
|
push esi edi edx
|
|
|
|
add ecx, ICMP_Packet.Data
|
|
mov di , IP_PROTO_ICMP SHL 8 + 128 ; TTL
|
|
shr edx, 16
|
|
|
|
call IPv4_output
|
|
jz .exit
|
|
|
|
DEBUGF 1,"full icmp packet size: %u\n", edx
|
|
|
|
pop eax
|
|
mov word [edi + ICMP_Packet.Type], ax ; Write both type and code bytes at once
|
|
pop eax
|
|
mov [edi + ICMP_Packet.SequenceNumber], ax
|
|
shr eax, 16
|
|
mov [edi + ICMP_Packet.Identifier], ax
|
|
mov [edi + ICMP_Packet.Checksum], 0
|
|
|
|
push eax ebx ecx edx
|
|
mov esi, edi
|
|
xor edx, edx
|
|
call checksum_1
|
|
call checksum_2
|
|
mov [edi + ICMP_Packet.Checksum], dx
|
|
pop edx ecx ebx eax esi
|
|
|
|
sub ecx, ICMP_Packet.Data
|
|
add edi, ICMP_Packet.Data
|
|
push cx
|
|
shr cx , 2
|
|
rep movsd
|
|
pop cx
|
|
and cx , 3
|
|
rep movsb
|
|
|
|
sub edi, edx ;;; TODO: find a better way to remember start of packet
|
|
push edx edi
|
|
DEBUGF 1,"Sending ICMP Packet\n"
|
|
call [ebx + NET_DEVICE.transmit]
|
|
ret
|
|
.exit:
|
|
DEBUGF 1,"Creating ICMP Packet failed\n"
|
|
add esp, 3*4
|
|
ret
|
|
|
|
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
;
|
|
; ICMP_API
|
|
;
|
|
; This function is called by system function 75
|
|
;
|
|
; IN: subfunction number in bl
|
|
; device number in bh
|
|
; ecx, edx, .. depends on subfunction
|
|
;
|
|
; OUT:
|
|
;
|
|
;-----------------------------------------------------------------
|
|
align 4
|
|
ICMP_API:
|
|
|
|
movzx eax, bh
|
|
shl eax, 2
|
|
|
|
test bl, bl
|
|
jz .packets_tx ; 0
|
|
dec bl
|
|
jz .packets_rx ; 1
|
|
|
|
.error:
|
|
mov eax, -1
|
|
ret
|
|
|
|
.packets_tx:
|
|
add eax, ICMP_PACKETS_TX
|
|
mov eax, [eax]
|
|
ret
|
|
|
|
.packets_rx:
|
|
add eax, ICMP_PACKETS_RX
|
|
mov eax, [eax]
|
|
ret
|