Subversion Repositories Kolibri OS

Rev

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

  1. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  2. ;;                                                                 ;;
  3. ;;  IP.INC                                                         ;;
  4. ;;                                                                 ;;
  5. ;;  IP Processes for Menuet OS  TCP/IP stack                       ;;
  6. ;;                                                                 ;;
  7. ;;  Version 0.3  29 August 2002                                    ;;
  8. ;;                                                                 ;;
  9. ;;  Copyright 2002 Mike Hibbett, mikeh@oceanfree.net               ;;
  10. ;;                                                                 ;;
  11. ;;  See file COPYING for details                                   ;;
  12. ;;                                                                 ;;
  13. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
  14.  
  15. $Revision: 2971 $
  16.  
  17.  
  18. ; IP underlying protocols numbers
  19. PROTOCOL_ICMP     equ      1
  20. PROTOCOL_TCP     equ      6
  21. PROTOCOL_UDP     equ      17
  22.  
  23. struc IP_PACKET
  24. {  .VersionAndIHL           db   ?  ;+00 - Version[0-3 bits] and IHL(header length)[4-7 bits]
  25.    .TypeOfService           db   ?  ;+01
  26.    .TotalLength             dw   ?  ;+02
  27.    .Identification          dw   ?  ;+04
  28.    .FlagsAndFragmentOffset  dw   ?  ;+06 - Flags[0-2] and FragmentOffset[3-15]
  29.    .TimeToLive              db   ?  ;+08
  30.    .Protocol                db   ?  ;+09
  31.    .HeaderChecksum          dw   ?  ;+10
  32.    .SourceAddress           dd   ?  ;+12
  33.    .DestinationAddress      dd   ?  ;+16
  34.    .DataOrOptional          dd   ?  ;+20
  35. }
  36.  
  37. virtual at 0
  38.   IP_PACKET IP_PACKET
  39. end virtual
  40.  
  41.  
  42. ;*******************************************************************
  43. ;   Interface
  44. ;
  45. ;       ip_rx       processes all packets received by the network layer
  46. ;                   It calls the appropriate protocol handler
  47. ;
  48. ;
  49. ;
  50. ;*******************************************************************
  51.  
  52.  
  53. ;
  54. ;   IP Packet after reception - Normal IP packet format
  55. ;
  56. ;           0               1               2               3
  57. ;    0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7
  58. ;
  59. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  60. ;0  |Version|  IHL  |Type of Service|       Total Length            |
  61. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  62. ;4  |         Identification        |Flags|      Fragment Offset    |
  63. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  64. ;8  |  Time to Live |    Protocol   |         Header Checksum       |
  65. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  66. ;12 |                       Source Address                          |
  67. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  68. ;16 |                    Destination Address                        |
  69. ;   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  70. ;20 |      Data                                                     |
  71. ;   +-+-+-..........                                               -+
  72. ;
  73. ;
  74. ;    [smb] attention! according to RFC 791 IP packet may have 'options' sections,
  75. ; so we can't simply think, that data have offset 20. We must calculate offset from
  76. ; IHL field
  77. ;
  78. macro   GET_IHL reg, header_addr
  79. {
  80.         movzx   reg, byte [header_addr]
  81.        
  82.         ; we need 4-7 bits, so....
  83.         and     reg, 0x0000000F
  84.        
  85.         ; IHL keeps number of octets, so we need to << 2 'reg'
  86.         shl     reg, 2
  87. }
  88.  
  89.  
  90. include "tcp.inc"
  91. include "udp.inc"
  92. include "icmp.inc"
  93.  
  94. ;***************************************************************************
  95. ;   Function
  96. ;      ip_rx
  97. ;
  98. ;   Description
  99. ;       This is a kernel function, called by stack_handler
  100. ;       Processes all IP-packets received by the network layer
  101. ;       It calls the appropriate protocol handler
  102. ;
  103. ;***************************************************************************
  104. proc ip_rx stdcall
  105. local buffer_number dd ?
  106.  
  107.     ; Look for a buffer to tx
  108.     mov     eax, IPIN_QUEUE
  109.     call    dequeue
  110.     cmp     ax, NO_BUFFER
  111.     je      .exit         ; Exit if no buffer available
  112.  
  113.     mov     [buffer_number], eax    ;save buffer number
  114.  
  115.     ; convert buffer pointer eax to the absolute address
  116.     imul    eax, IPBUFFSIZE
  117.     add     eax, IPbuffs
  118.  
  119.     mov     ebx, eax  ; ebx=pointer to IP_PACKET
  120.  
  121. ;        DEBUGF  1, "K : ip_rx - proto: %u\n", [ebx + IP_PACKET.Protocol]:1
  122.  
  123.     ; Validate the IP checksum
  124.     mov     dx, word[ebx + IP_PACKET.HeaderChecksum]
  125.     xchg    dh,dl          ; Get the checksum in intel format
  126.  
  127.     mov     [ebx + IP_PACKET.HeaderChecksum], 0  ; clear checksum field - need to when
  128.                                 ; recalculating checksum
  129.     ;  this needs two data pointers and two size #.
  130.     ;  2nd pointer can be of length 0
  131.  
  132.     GET_IHL ecx, ebx + IP_PACKET.VersionAndIHL ;get packet length in ecx
  133.     stdcall checksum_jb, ebx, ecx   ;buf_ptr, buf_size
  134.     cmp     dx, ax
  135.  
  136. ;        DEBUGF  1, "K : ip_rx - checksums: %x - %x\n", dx, ax
  137.  
  138.     jnz     .dump.1  ;if CHECKSUM isn't valid then dump packet
  139.     mov     edx, ebx ; EDX (IP-BUFFER POINTER) WILL BE USED FOR *_rx HANDLERS BELOW!!!
  140.  
  141. ;        DEBUGF  1, "K : ip_rx - dest: %x - %x\n", [ebx + IP_PACKET.DestinationAddress], [stack_ip]
  142.  
  143.     ; Validate the IP address, if it isn't broadcast
  144.     mov     eax, [stack_ip]
  145.     cmp     dword[ebx + IP_PACKET.DestinationAddress], eax
  146.     je      @f
  147.  
  148.     ; If the IP address is 255.255.255.255, accept it
  149.     ; - it is a broadcast packet, which we need for dhcp
  150.  
  151.         mov     eax, [ebx + IP_PACKET.DestinationAddress]
  152.         cmp     eax, 0xffffffff
  153.         ;je      @f
  154.         ;mov     ecx, [stack_ip]
  155.         ;and     eax, [subnet_mask]
  156.         ;and     ecx, [subnet_mask]
  157.         ;cmp     eax, ecx
  158.         ;jne     .dump.2
  159.         ;mov     eax, [ebx + IP_PACKET.DestinationAddress]
  160.         ;or      eax, [subnet_mask]
  161.         ;cmp     eax, 0xffffffff
  162.         jne     .dump.2
  163.  
  164.   @@:
  165.     mov     al, [ebx + IP_PACKET.VersionAndIHL]
  166.     and     al, 0x0f  ;get IHL(header length)
  167.     cmp     al, 0x05  ;if IHL!= 5*4(20 bytes)
  168. ;        DEBUGF  1, "K : ip_rx - ihl: %x - 05\n", al
  169.     jnz     .dump.3     ;then dump it
  170.  
  171. ;        DEBUGF  1, "K : ip_rx - ttl: %x - 00\n", [ebx + IP_PACKET.TimeToLive]:2
  172.  
  173.     cmp     [ebx + IP_PACKET.TimeToLive], 0
  174.     je      .dump.4     ;if TTL==0 then dump it
  175.  
  176.     mov     ax, [ebx + IP_PACKET.FlagsAndFragmentOffset]
  177.     and     ax, 0xFFBF   ;get flags
  178. ;        DEBUGF  1, "K : ip_rx - flags: %x - 0000\n", ax
  179.     cmp     ax, 0        ;if some flags was set then we dump this packet
  180.     jnz     .dump.5        ;the flags should be used for fragmented packets
  181.  
  182.     ; Check the protocol, and call the appropriate handler
  183.     ; Each handler will re-use or free the queue buffer as appropriate
  184.  
  185.     mov     al, [ebx + IP_PACKET.Protocol]
  186.  
  187.     cmp     al , PROTOCOL_TCP
  188.     jne     .not_tcp
  189. ;    DEBUGF  1,"K : ip_rx - TCP packet\n"
  190.     mov     eax, dword[buffer_number]
  191.     call    tcp_rx
  192.     jmp     .exit
  193.  
  194.   .not_tcp:
  195.     cmp     al, PROTOCOL_UDP
  196.     jne     .not_udp
  197. ;    DEBUGF  1,"K : ip_rx - UDP packet\n"
  198.     mov     eax, dword[buffer_number]
  199.     call    udp_rx
  200.     jmp     .exit
  201.  
  202.   .not_udp:
  203.     cmp     al, PROTOCOL_ICMP
  204.     jne     .dump.6              ;protocol ain't supported
  205.  
  206. ;    DEBUGF  1,"K : ip_rx - ICMP packet\n"
  207.     ;GET_IHL ecx, ebx + IP_PACKET.VersionAndIHL ;get packet length in ecx
  208.     mov     eax, dword[buffer_number]
  209.     stdcall icmp_rx,eax,ebx,ecx  ;buffer_number,IPPacketBase,IPHeaderLength
  210.     jmp     .exit
  211.  
  212.  
  213.   .dump.1:
  214.         DEBUGF  1, "K : ip_rx - dumped (checksum: 0x%x-0x%x)\n", dx, ax
  215.         jmp     .dump.x
  216.  
  217.   .dump.2:
  218.         DEBUGF  1, "K : ip_rx - dumped (ip: %u.%u.%u.%u)\n", [ebx + IP_PACKET.DestinationAddress + 0]:1, [ebx + IP_PACKET.DestinationAddress + 1]:1, [ebx + IP_PACKET.DestinationAddress + 2]:1, [ebx + IP_PACKET.DestinationAddress + 3]:1
  219.         jmp     .dump.x
  220.  
  221.   .dump.3:
  222.         DEBUGF  1, "K : ip_rx - dumped (ihl: %u)\n", al
  223.         jmp     .dump.x
  224.  
  225.   .dump.4:
  226.         DEBUGF  1, "K : ip_rx - dumped (ttl: %u)\n", [ebx + IP_PACKET.TimeToLive]
  227.         jmp     .dump.x
  228.  
  229.   .dump.5:
  230.         DEBUGF  1, "K : ip_rx - dumped (flags: 0x%x)\n", ax
  231.         jmp     .dump.x
  232.  
  233.   .dump.6:
  234.         DEBUGF  1, "K : ip_rx - dumped (proto: %u)\n", [ebx + IP_PACKET.Protocol]:1
  235.  
  236.   .dump.x:
  237.     inc     dword[dumped_rx_count]
  238.     mov     eax, [buffer_number]
  239.     call    freeBuff
  240.  
  241.   .exit:
  242.     ret
  243. endp
  244.  
  245.