Subversion Repositories Kolibri OS

Rev

Rev 1519 | Blame | Last modification | View Log | Download | RSS feed

  1. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  2. ;;                                                                 ;;
  3. ;; Copyright (C) KolibriOS team 2004-2010. 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.  
  20. $Revision: 1529 $
  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.  
  106.  
  107.  
  108. ;-----------------------------------------------------------------
  109. ;
  110. ; ICMP_init
  111. ;
  112. ;-----------------------------------------------------------------
  113.  
  114. macro ICMP_init {
  115.  
  116.         xor     eax, eax
  117.         mov     edi, ICMP_PACKETS_TX
  118.         mov     ecx, 2*MAX_IP
  119.         rep     stosd
  120.  
  121. }
  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. ;       pointer to device struct in ebx
  136. ;       ICMP Packet size in ecx
  137. ;       pointer to ICMP Packet data in edx
  138. ;  OUT: /
  139. ;
  140. ;-----------------------------------------------------------------
  141. align 4
  142. ICMP_input:
  143.  
  144. ;;; TODO: works only on pure ethernet right now !
  145.  
  146.         DEBUGF  1,"ICMP_Handler - start\n"
  147.         cmp     byte [edx + ICMP_Packet.Type], ICMP_ECHO                    ; Is this an echo request?
  148.         jne     .check_sockets
  149.  
  150. ;;; TODO: check checksum!
  151.  
  152.         DEBUGF  1,"ICMP_Handler - echo request\n"
  153.  
  154.         mov     byte [edx + ICMP_Packet.Type], ICMP_ECHOREPLY               ; Change Packet type to reply
  155.         mov     word [edx + ICMP_Packet.Checksum], 0                        ; Set checksum to 0, needed to calculate new checksum
  156.  
  157.         call    NET_ptr_to_num
  158.         cmp     edi,-1
  159.         je      .dump
  160.         inc     [ICMP_PACKETS_RX+4*edi]
  161.         inc     [ICMP_PACKETS_TX+4*edi]
  162.  
  163. ; exchange dest and source address in IP header
  164. ; exchange dest and source MAC in ETH header
  165.         mov     esi, [esp]
  166.  
  167.         mov     eax, dword [esi + ETH_FRAME.DstMAC]
  168.         mov     ecx, dword [esi + ETH_FRAME.SrcMAC]
  169.         mov     dword [esi + ETH_FRAME.SrcMAC], eax
  170.         mov     dword [esi + ETH_FRAME.DstMAC], ecx
  171.  
  172.         mov     ax, word [esi + ETH_FRAME.DstMAC + 4]
  173.         mov     cx, word [esi + ETH_FRAME.SrcMAC + 4]
  174.         mov     word [esi + ETH_FRAME.SrcMAC + 4], ax
  175.         mov     word [esi + ETH_FRAME.DstMAC + 4], cx
  176.  
  177.         mov     eax, dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress]
  178.         mov     ecx, dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress]
  179.         mov     dword [esi + ETH_FRAME.Data + IPv4_Packet.DestinationAddress], eax
  180.         mov     dword [esi + ETH_FRAME.Data + IPv4_Packet.SourceAddress], ecx
  181.  
  182. ; Recalculate ip header checksum
  183.         add     esi, ETH_FRAME.Data                                         ; Point esi to start of IP Packet
  184.         movzx   ecx, byte [esi + IPv4_Packet.VersionAndIHL]                 ; Calculate IP Header length by using IHL field
  185.         and     ecx, 0x0000000F                                             ;
  186.         shl     cx , 2
  187.         push    ebx edx ecx esi
  188.         xor     edx, edx
  189.         call    checksum_1
  190.         call    checksum_2
  191.         pop     esi
  192.         mov     word [esi + IPv4_Packet.HeaderChecksum], dx                 ; Store it in the IP Packet header
  193.  
  194. ; Recalculate ICMP CheckSum
  195.         movzx   eax, word[esi + IPv4_Packet.TotalLength]                    ; Find length of IP Packet
  196.         xchg    ah , al                                                     ;
  197.         sub     eax, [esp]                                                   ; Now we know the length of ICMP data in eax
  198.         mov     ecx, eax
  199.         mov     esi, [esp + 4]
  200.         xor     edx, edx
  201.         call    checksum_1
  202.         call    checksum_2
  203.         mov     ax , dx
  204.         pop     ecx edx ebx
  205.         mov     word [edx + ICMP_Packet.Checksum], ax
  206.  
  207.         call    [ebx + NET_DEVICE.transmit]
  208.         ret
  209.  
  210.  
  211.  
  212.  
  213.        .check_sockets:
  214.         ; TODO: validate the header & checksum.
  215.  
  216.         ; Look for an open ICMP socket
  217.  
  218.         mov     esi, net_sockets
  219.   .try_more:
  220.         mov     ax , [edx + ICMP_Packet.Identifier]
  221.   .next_socket:
  222.         mov     esi, [esi + SOCKET.NextPtr]
  223.         or      esi, esi
  224.         jz      .dump
  225.         cmp     [esi + SOCKET.Type], IP_PROTO_ICMP
  226.         jne     .next_socket
  227.         cmp     [esi + ICMP_SOCKET.Identifier], ax
  228.         jne     .next_socket
  229.  
  230.         call    IPv4_dest_to_dev
  231.         cmp     edi,-1
  232.         je      .dump
  233.         inc     [ICMP_PACKETS_RX+edi]
  234.  
  235.         DEBUGF 1,"Found valid ICMP packet for socket %x\n", esi
  236.  
  237.         lea     ebx, [esi + SOCKET.lock]
  238.         call    wait_mutex
  239.  
  240.         ; Now, assign data to socket. We have socket address in esi.
  241.         ; We have ICMP Packet in edx
  242.         ; number of bytes in ecx
  243.  
  244.         mov     eax, esi
  245.         pop     esi
  246.         add     esp, 4
  247.         sub     edx, esi
  248.         mov     edi, edx
  249.         jmp     SOCKET_input
  250.  
  251.        .dump:
  252.         DEBUGF  1,"ICMP_Handler - dumping\n"
  253.  
  254.         call    kernel_free
  255.         add     esp, 4 ; pop (balance stack)
  256.  
  257.         ret
  258.  
  259.  
  260. ;-----------------------------------------------------------------
  261. ;
  262. ; ICMP_output
  263. ;
  264. ; IN:  eax = dest ip
  265. ;      ebx = source ip
  266. ;      ecx = data length
  267. ;      dh  = type
  268. ;      dl  = code
  269. ;      high 16 bits of edx = fragment id (for IP header)
  270. ;      esi = data offset
  271. ;      edi = identifier shl 16 + sequence number
  272. ;
  273. ;-----------------------------------------------------------------
  274. align 4
  275. ICMP_output:
  276.  
  277.         DEBUGF  1,"Creating ICMP Packet\n"
  278.  
  279.         push    esi edi edx
  280.  
  281.         add     ecx, ICMP_Packet.Data
  282.         mov     di , IP_PROTO_ICMP SHL 8 + 128  ; TTL
  283.         shr     edx, 16
  284.  
  285.         call    IPv4_output
  286.         jz      .exit
  287.  
  288.         DEBUGF  1,"full icmp packet size: %u\n", edx
  289.  
  290.         pop     eax
  291.         mov     word [edi + ICMP_Packet.Type], ax       ; Write both type and code bytes at once
  292.         pop     eax
  293.         mov     [edi + ICMP_Packet.SequenceNumber], ax
  294.         shr     eax, 16
  295.         mov     [edi + ICMP_Packet.Identifier], ax
  296.         mov     [edi + ICMP_Packet.Checksum], 0
  297.  
  298.         push    eax ebx ecx edx
  299.         mov     esi, edi
  300.         xor     edx, edx
  301.         call    checksum_1
  302.         call    checksum_2
  303.         mov     [edi + ICMP_Packet.Checksum], dx
  304.         pop     edx ecx ebx eax esi
  305.  
  306.         sub     ecx, ICMP_Packet.Data
  307.         add     edi, ICMP_Packet.Data
  308.         push    cx
  309.         shr     cx , 2
  310.         rep     movsd
  311.         pop     cx
  312.         and     cx , 3
  313.         rep     movsb
  314.  
  315.         sub     edi, edx                                ;;; TODO: find a better way to remember start of packet
  316.         push    edx edi
  317.         DEBUGF  1,"Sending ICMP Packet\n"
  318.         call    [ebx + NET_DEVICE.transmit]
  319.         ret
  320.   .exit:
  321.         DEBUGF  1,"Creating ICMP Packet failed\n"
  322.         add     esp, 3*4
  323.         ret
  324.  
  325.  
  326.  
  327.  
  328. ;-----------------------------------------------------------------
  329. ;
  330. ; ICMP_API
  331. ;
  332. ; This function is called by system function 75
  333. ;
  334. ; IN:  subfunction number in bl
  335. ;      device number in bh
  336. ;      ecx, edx, .. depends on subfunction
  337. ;
  338. ; OUT:
  339. ;
  340. ;-----------------------------------------------------------------
  341. align 4
  342. ICMP_API:
  343.  
  344.         movzx   eax, bh
  345.         shl     eax, 2
  346.  
  347.         test    bl, bl
  348.         jz      .packets_tx     ; 0
  349.         dec     bl
  350.         jz      .packets_rx     ; 1
  351.  
  352. .error:
  353.         mov     eax, -1
  354.         ret
  355.  
  356. .packets_tx:
  357.         add     eax, ICMP_PACKETS_TX
  358.         mov     eax, [eax]
  359.         ret
  360.  
  361. .packets_rx:
  362.         add     eax, ICMP_PACKETS_RX
  363.         mov     eax, [eax]
  364.         ret
  365.