Subversion Repositories Kolibri OS

        dec     ecx

        dec     ecx

        jnz     .loop

        jnz     .loop

        jmp     .done

        jmp     .done

        DEBUGF  2,"IPv4 Fragment slot timed-out!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4 Fragment slot timed-out!\n"

;;; TODO: clear all entry's of timed-out slot

;;; TODO: clear all entry's of timed-out slot

        jmp     .next

        jmp     .next

;

;

;-----------------------------------------------------------------

;-----------------------------------------------------------------

align 4

align 4

IPv4_input:                                                     ; TODO: add IPv4 raw sockets support

IPv4_input:                                                     ; TODO: add IPv4 raw sockets support

        DEBUGF  2,"IPv4_input, packet from: %u.%u.%u.%u ",\

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input, packet from: %u.%u.%u.%u ",\

        [edx + IPv4_header.SourceAddress + 0]:1,[edx + IPv4_header.SourceAddress + 1]:1,\

        [edx + IPv4_header.SourceAddress + 0]:1,[edx + IPv4_header.SourceAddress + 1]:1,\

        DEBUGF  2,"to: %u.%u.%u.%u\n",\

        DEBUGF  DEBUG_NETWORK_VERBOSE, "to: %u.%u.%u.%u\n",\

        [edx + IPv4_header.DestinationAddress + 0]:1,[edx + IPv4_header.DestinationAddress + 1]:1,\

        [edx + IPv4_header.DestinationAddress + 0]:1,[edx + IPv4_header.DestinationAddress + 1]:1,\

        [edx + IPv4_header.DestinationAddress + 2]:1,[edx + IPv4_header.DestinationAddress + 3]:1

        [edx + IPv4_header.DestinationAddress + 2]:1,[edx + IPv4_header.DestinationAddress + 3]:1

;-------------------------------

;-------------------------------

; re-calculate the checksum

; re-calculate the checksum

        IPv4_checksum edx

        IPv4_checksum edx

        jnz     .dump                                           ; if checksum isn't valid then dump packet

        jnz     .dump                                           ; if checksum isn't valid then dump packet

        cmp     eax, 127

        cmp     eax, 127

        je      .ip_ok

        je      .ip_ok

        DEBUGF  2,"IPv4_input: Destination address does not match!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Destination address does not match!\n"

        jmp     .dump

        jmp     .dump

        je      UDP_input

        je      UDP_input

        cmp     al, IP_PROTO_ICMP

        cmp     al, IP_PROTO_ICMP

  .dump:

  .dump:

        DEBUGF  1,"IPv4_input: dumping\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: dumping\n"

        inc     [IP_packets_dumped]                             ; FIXME: use correct interface

        inc     [IP_packets_dumped]                             ; FIXME: use correct interface

  .has_fragments:

  .has_fragments:

        movzx   eax, [edx + IPv4_header.FlagsAndFragmentOffset]

        movzx   eax, [edx + IPv4_header.FlagsAndFragmentOffset]

        xchg    al, ah

        xchg    al, ah

        shl     ax, 3

        shl     ax, 3

        DEBUGF  1,"IPv4_input: fragmented packet offset=%u id=%x\n", ax, [edx + IPv4_header.Identification]:4

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: fragmented packet offset=%u id=%x\n", ax, [edx + IPv4_header.Identification]:4

        test    ax, ax                                          ; Is this the first packet of the fragment?

        test    ax, ax                                          ; Is this the first packet of the fragment?

        jz      .is_first_fragment

        jz      .is_first_fragment

;-------------------------------------------------------

;-------------------------------------------------------

; We have a fragmented IP packet, but it's not the first

; We have a fragmented IP packet, but it's not the first

;------------------------------------

;------------------------------------

; We have received the first fragment

; We have received the first fragment

  .is_first_fragment:

  .is_first_fragment:

        DEBUGF  1,"IPv4_input: First fragment packet received!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: First fragment packet received!\n"

                                                                ; try to locate a free slot..

                                                                ; try to locate a free slot..

        mov     ecx, MAX_FRAGMENTS

        mov     ecx, MAX_FRAGMENTS

        mov     esi, FRAGMENT_LIST

        mov     esi, FRAGMENT_LIST

;-----------------------------------

;-----------------------------------

; We have received the last fragment

; We have received the last fragment

  .is_last_fragment:

  .is_last_fragment:

        DEBUGF  1,"IPv4_input: Last fragment packet received!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Last fragment packet received!\n"

  .count_bytes:

  .count_bytes:

        cmp     [esi + FRAGMENT_entry.PrevPtr], edi

        cmp     [esi + FRAGMENT_entry.PrevPtr], edi

        jne     .destroy_slot_pop                                               ; Damn, something screwed up, remove the whole slot (and free buffers too if possible!)

        jne     .destroy_slot_pop                                               ; Damn, something screwed up, remove the whole slot (and free buffers too if possible!)

        mov     cx, [esi + sizeof.FRAGMENT_entry + IPv4_header.TotalLength]     ; Add total length

        mov     cx, [esi + sizeof.FRAGMENT_entry + IPv4_header.TotalLength]     ; Add total length

        xchg    cl, ch

        xchg    cl, ch

        add     ax, cx

        add     ax, cx

        movzx   cx, [esi + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]   ; Sub Header length

        movzx   cx, [esi + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]   ; Sub Header length

        and     cx, 0x000F

        and     cx, 0x000F

        shl     cx, 2

        shl     cx, 2

        sub     ax, cx

        sub     ax, cx

        mov     edi, esi

        mov     edi, esi

        mov     esi, [esi + FRAGMENT_entry.NextPtr]

        mov     esi, [esi + FRAGMENT_entry.NextPtr]

        cmp     esi, -1

        cmp     esi, -1

        jne     .count_bytes

        jne     .count_bytes

        mov     [esi + FRAGMENT_entry.PrevPtr], edi

        mov     [esi + FRAGMENT_entry.PrevPtr], edi

        mov     [esi + FRAGMENT_entry.Owner], ebx

        mov     [esi + FRAGMENT_entry.Owner], ebx

        mov     cx, [edx + IPv4_header.TotalLength]                             ; Note: This time we dont substract Header length

        mov     cx, [edx + IPv4_header.TotalLength]                             ; Note: This time we dont substract Header length

        DEBUGF  1,"IPv4_input: Packet size=%u\n", cx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Packet size=%u\n", cx

        DEBUGF  1,"IPv4_input: Total Received data size=%u\n", eax

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Total Received data size=%u\n", eax

        push    eax

        push    eax

        mov     ax, [edx + IPv4_header.FlagsAndFragmentOffset]

        mov     ax, [edx + IPv4_header.FlagsAndFragmentOffset]

        xchg    al, ah

        xchg    al, ah

        shl     ax, 3

        shl     ax, 3

        pop     eax

        pop     eax

        DEBUGF  1,"IPv4_input: Total Fragment size=%u\n", ecx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Total Fragment size=%u\n", ecx

  .rebuild_packet_loop:

  .rebuild_packet_loop:

        movzx   ecx, [edx + sizeof.FRAGMENT_entry + IPv4_header.FlagsAndFragmentOffset] ; Calculate the fragment offset

        movzx   ecx, [edx + sizeof.FRAGMENT_entry + IPv4_header.FlagsAndFragmentOffset] ; Calculate the fragment offset

        xchg    cl, ch                                                                  ;  intel byte order

        xchg    cl, ch                                                                  ;  intel byte order

        DEBUGF  1,"IPv4_input: Fragment offset=%u\n", cx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Fragment offset=%u\n", cx

        lea     edi, [eax + ecx]                                                        ; Notice that edi will be equal to eax for first fragment

        lea     edi, [eax + ecx]                                                        ; Notice that edi will be equal to eax for first fragment

        movzx   ebx, [edx + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]          ; Find header size (in ebx) of fragment

        movzx   ebx, [edx + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]          ; Find header size (in ebx) of fragment

        jmp     .handle_it          ; edx = buf ptr, ecx = size, [esp] buf ptr, [esp+4], total size, ebx=device ptr

        jmp     .handle_it          ; edx = buf ptr, ecx = size, [esp] buf ptr, [esp+4], total size, ebx=device ptr

  .destroy_slot_pop:

  .destroy_slot_pop:

        add     esp, 4

        add     esp, 4

        DEBUGF  1,"IPv4_input: Destroy fragment slot!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Destroy fragment slot!\n"

        ; TODO!

        ; TODO!

;

;

;------------------------------------------------------------------

;------------------------------------------------------------------

align 4

align 4

IPv4_output:

IPv4_output:

        DEBUGF  1,"IPv4_output: size=%u\n", ecx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output: size=%u\n", ecx

        cmp     ecx, 65500              ; Max IPv4 packet size

        cmp     ecx, 65500              ; Max IPv4 packet size

        pop     ecx

        pop     ecx

        add     edi, sizeof.IPv4_header

        add     edi, sizeof.IPv4_header

        DEBUGF  2,"IPv4_output: success!\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output: success!\n"

  .eth_error:

  .eth_error:

        DEBUGF  2,"IPv4_output: ethernet error\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output: ethernet error\n"

        add     esp, 3*4+2+6

        add     esp, 3*4+2+6

        ret

        ret

  .arp_error:

  .arp_error:

        DEBUGF  2,"IPv4_output: ARP error=%x\n", eax

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output: ARP error=%x\n", eax

        xor     edi, edi

        xor     edi, edi

        ret

        ret

  .too_large:

  .too_large:

;        [edi + IPv4_header.SourceAddress]

;        [edi + IPv4_header.SourceAddress]

;        [edi + IPv4_header.DestinationAddress]

;        [edi + IPv4_header.DestinationAddress]

        IPv4_checksum edi                       ;;;; todo: checksum for IP packet with options!

        IPv4_checksum edi                       ;;;; todo: checksum for IP packet with options!

        DEBUGF  2,"IPv4_output_raw: device=%x\n", ebx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output_raw: device=%x\n", ebx

        call    [ebx + NET_DEVICE.transmit]

        call    [ebx + NET_DEVICE.transmit]

        ret

        ret

  .error:

  .error:

        add     esp, 6

        add     esp, 6

  .arp_error:

  .arp_error:

  .too_large:

  .too_large:

        DEBUGF  2,"IPv4_output_raw: Failed\n"

        DEBUGF  DEBUG_NETWORK_ERROR, "IPv4_output_raw: Failed\n"

        sub     edi, edi

        sub     edi, edi

;--------------------------------------------------------

;--------------------------------------------------------

align 4

align 4

        DEBUGF 1,"IPv4_fragment\n"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_fragment\n"

        sub     ecx, sizeof.IPv4_header         ; substract header size

        sub     ecx, sizeof.IPv4_header         ; substract header size

        push    ecx                             ; max data size

        push    ecx                             ; max data size

        push    dword 0                         ; offset

        push    dword 0                         ; offset

        DEBUGF 1,"Ipv4_fragment: new fragment"

        DEBUGF  DEBUG_NETWORK_VERBOSE, "Ipv4_fragment: new fragment"

        mov     esi, [esp + 2*4]

        mov     esi, [esp + 2*4]

        add     esi, sizeof.IPv4_header

        add     esi, sizeof.IPv4_header

        add     esi, [esp]      ; offset

        add     esi, [esp]      ; offset

        DEBUGF 1,"IPv4_fragment: copying %u bytes\n", ecx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_fragment: copying %u bytes\n", ecx

        rep     movsb

        rep     movsb

; now, correct header

; now, correct header

        mov     ecx, [esp+3*4+6+4]      ; ptr to begin of buff

        mov     ecx, [esp+3*4+6+4]      ; ptr to begin of buff

        add     ecx, [esp+3*4+6+4+4]    ; buff size

        add     ecx, [esp+3*4+6+4+4]    ; buff size

        sub     ecx, [esp+2*4]          ; ptr to ip header

        sub     ecx, [esp+2*4]          ; ptr to ip header

        add     ecx, [esp]              ; offset

        add     ecx, [esp]              ; offset

        DEBUGF 1,"Ipv4_fragment: %u bytes remaining\n", ecx

        DEBUGF  DEBUG_NETWORK_VERBOSE, "Ipv4_fragment: %u bytes remaining\n", ecx

        cmp     ecx, [esp+1*4]

        cmp     ecx, [esp+1*4]

        jae     .new_fragment

        jae     .new_fragment

        jmp     .new_fragment

        jmp     .new_fragment

      .err:

      .err:

      .done:

      .done:

        add     esp, 12 + 4 + 6

        add     esp, 12 + 4 + 6

      .err2:

      .err2:

  .invalid:

  .invalid:

        xor     edi, edi                        ; if none found, use device 0 as default

        xor     edi, edi                        ; if none found, use device 0 as default

        mov     eax, [GATEWAY_LIST]

        mov     eax, [GATEWAY_LIST]

        DEBUGF  1,"IPv4_dest_to_dev: %u\n", edi

        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_route: %u\n", edi

        ret

        ret

  .broadcast:

  .broadcast: