Subversion Repositories Kolibri OS

Rev

Rev 4429 | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  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.  
  297.         DEBUGF  DEBUG_NETWORK_ERROR, "ICMP_input: no socket found\n"
  298.         jmp     .dump
  299.  
  300.  
  301.   .checksum_mismatch:
  302.         DEBUGF  DEBUG_NETWORK_ERROR, "ICMP_input: checksum mismatch\n"
  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
  470.