WebSVN – Kolibri OS – Diff – /kernel/branches/Kolibri-acpi/network/IPv4.inc

 ;;                                                                 ;;
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 $Revision: 3515 $
-MAX_FRAGMENTS                   = 64
+IPv4_MAX_FRAGMENTS              = 64
         SourceAddress           dd ?
         DestinationAddress      dd ?
 ends
-struct  FRAGMENT_slot
+struct  IPv4_FRAGMENT_slot
         ttl                     dw ?    ; Time to live for this entry, 0 for empty slot's
         id                      dw ?    ; Identification field from IP header
         SrcIP                   dd ?    ; .. from IP header
         DstIP                   dd ?    ; .. from IP header
         ptr                     dd ?    ; Pointer to first packet
 ends
         SUBNET_LIST             rd NET_DEVICES_MAX
         DNS_LIST                rd NET_DEVICES_MAX
         GATEWAY_LIST            rd NET_DEVICES_MAX
         BROADCAST_LIST          rd NET_DEVICES_MAX
-        IP_packets_tx           rd NET_DEVICES_MAX
+        IPv4_packets_tx         rd NET_DEVICES_MAX
-        IP_packets_rx           rd NET_DEVICES_MAX
+        IPv4_packets_rx         rd NET_DEVICES_MAX
-        IP_packets_dumped       rd NET_DEVICES_MAX
+        IPv4_packets_dumped     rd NET_DEVICES_MAX
-        FRAGMENT_LIST           rb MAX_FRAGMENTS * sizeof.FRAGMENT_slot
+        IPv4_FRAGMENT_LIST      rb IPv4_MAX_FRAGMENTS * sizeof.IPv4_FRAGMENT_slot
 ;-----------------------------------------------------------------
 macro   IPv4_init {
         xor     eax, eax
         mov     edi, IP_LIST
-        mov     ecx, 7*NET_DEVICES_MAX + (sizeof.FRAGMENT_slot*MAX_FRAGMENTS)/4
+        mov     ecx, 7*NET_DEVICES_MAX + (sizeof.IPv4_FRAGMENT_slot*IPv4_MAX_FRAGMENTS)/4
         rep stosd
 ;-----------------------------------------------------------------
 macro IPv4_decrease_fragment_ttls {
 local   .loop, .next
-        mov     esi, FRAGMENT_LIST
+        mov     esi, IPv4_FRAGMENT_LIST
-        mov     ecx, MAX_FRAGMENTS
+        mov     ecx, IPv4_MAX_FRAGMENTS
   .loop:
-        cmp     [esi + FRAGMENT_slot.ttl], 0
+        cmp     [esi + IPv4_FRAGMENT_slot.ttl], 0
         je      .next
-        dec     [esi + FRAGMENT_slot.ttl]
+        dec     [esi + IPv4_FRAGMENT_slot.ttl]
         jz      .died
   .next:
-        add     esi, sizeof.FRAGMENT_slot
+        add     esi, sizeof.IPv4_FRAGMENT_slot
         dec     ecx
         jnz     .loop
 ;------------------------
 ; Now we can update stats
   .ip_ok:
-        inc     [IP_packets_rx + edi]
+        inc     [IPv4_packets_rx + edi]
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: unknown protocol %u\n", al
   .dump:
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: dumping\n"
-        inc     [IP_packets_dumped]                             ; FIXME: use correct interface
+        inc     [IPv4_packets_dumped]                           ; FIXME: use correct interface
         call    NET_packet_free
   .has_fragments:
         movzx   eax, [edx + IPv4_header.FlagsAndFragmentOffset]
         xchg    al, ah
         shl     ax, 3
-        DEBUGF  DEBUG_NETWORK_VERBOSE, "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 ptr=0x%x\n", ax, [edx + IPv4_header.Identification]:4, edx
         call    IPv4_find_fragment_slot
         cmp     esi, -1
         je      .dump
-        mov     [esi + FRAGMENT_slot.ttl], 15                   ; Reset the ttl
+        mov     [esi + IPv4_FRAGMENT_slot.ttl], 15              ; Reset the ttl
-        mov     esi, [esi + FRAGMENT_slot.ptr]
+        mov     esi, [esi + IPv4_FRAGMENT_slot.ptr]
         or      edi, -1
   .find_last_entry:                                             ; The following routine will try to find the last entry
-        cmp     edi, [esi + FRAGMENT_entry.PrevPtr]
+        cmp     edi, [esi + IPv4_FRAGMENT_entry.PrevPtr]
         jne     .destroy_slot                                   ; Damn, something screwed up, remove the whole slot (and free buffers too if possible!)
         mov     edi, esi
-        mov     esi, [esi + FRAGMENT_entry.NextPtr]
+        mov     esi, [esi + IPv4_FRAGMENT_entry.NextPtr]
         cmp     esi, -1
         jne     .find_last_entry
                                                                 ; We found the last entry (pointer is now in edi)
                                                                 ; We are going to overwrite the ethernet header in received packet with a FRAGMENT_entry structure
         pop     eax                                             ; pointer to packet
-        mov     [edi + FRAGMENT_entry.NextPtr], eax             ; update pointer of previous entry to the new entry
+        mov     [edi + IPv4_FRAGMENT_entry.NextPtr], eax        ; update pointer of previous entry to the new entry
-        mov     [eax + FRAGMENT_entry.NextPtr], -1
+        mov     [eax + IPv4_FRAGMENT_entry.NextPtr], -1
-        mov     [eax + FRAGMENT_entry.PrevPtr], edi
+        mov     [eax + IPv4_FRAGMENT_entry.PrevPtr], edi
 ; We have received the first fragment
   .is_first_fragment:
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: First fragment packet received!\n"
                                                                 ; try to locate a free slot..
-        mov     ecx, MAX_FRAGMENTS
+        mov     ecx, IPv4_MAX_FRAGMENTS
-        mov     esi, FRAGMENT_LIST
+        mov     esi, IPv4_FRAGMENT_LIST
   .find_free_slot:
-        cmp     word [esi + FRAGMENT_slot.ttl], 0
+        cmp     word [esi + IPv4_FRAGMENT_slot.ttl], 0
         je      .found_free_slot
-        add     esi, sizeof.FRAGMENT_slot
+        add     esi, sizeof.IPv4_FRAGMENT_slot
         loop    .find_free_slot
         jmp     .dump                                           ; If no free slot was found, dump the packet
   .found_free_slot:                                             ; We found a free slot, let's fill in the FRAGMENT_slot structure
-        mov     [esi + FRAGMENT_slot.ttl], 15                   ; RFC recommends 15 secs as ttl
+        mov     [esi + IPv4_FRAGMENT_slot.ttl], 15              ; RFC recommends 15 secs as ttl
         mov     ax, [edx + IPv4_header.Identification]
-        mov     [esi + FRAGMENT_slot.id], ax
+        mov     [esi + IPv4_FRAGMENT_slot.id], ax
         mov     eax, [edx + IPv4_header.SourceAddress]
-        mov     [esi + FRAGMENT_slot.SrcIP], eax
+        mov     [esi + IPv4_FRAGMENT_slot.SrcIP], eax
         mov     eax, [edx + IPv4_header.DestinationAddress]
-        mov     [esi + FRAGMENT_slot.DstIP], eax
+        mov     [esi + IPv4_FRAGMENT_slot.DstIP], eax
         pop     eax
-        mov     [esi + FRAGMENT_slot.ptr], eax
+        mov     [esi + IPv4_FRAGMENT_slot.ptr], eax
                                                                 ; Now, replace ethernet header in original buffer with a FRAGMENT_entry structure
-        mov     [eax + FRAGMENT_entry.NextPtr], -1
+        mov     [eax + IPv4_FRAGMENT_entry.NextPtr], -1
-        mov     [eax + FRAGMENT_entry.PrevPtr], -1
+        mov     [eax + IPv4_FRAGMENT_entry.PrevPtr], -1
-        mov     [eax + FRAGMENT_entry.Owner], ebx
+        mov     [eax + IPv4_FRAGMENT_entry.Owner], ebx
         call    IPv4_find_fragment_slot
         cmp     esi, -1
         je      .dump
-        mov     esi, [esi + FRAGMENT_slot.ptr]                  ; We found the first entry, let's calculate total size of the packet in eax, so we can allocate a buffer
+        mov     esi, [esi + IPv4_FRAGMENT_slot.ptr]                     ; We found the first entry, let's calculate total size of the packet in eax, so we can allocate a buffer
         push    esi
         xor     eax, eax
         or      edi, -1
   .count_bytes:
-        cmp     [esi + FRAGMENT_entry.PrevPtr], edi
+        cmp     [esi + IPv4_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.IPv4_FRAGMENT_entry + IPv4_header.TotalLength]        ; Add total length
         xchg    cl, ch
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Packet size=%u\n", cx
         add     ax, cx
-        movzx   cx, [esi + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]   ; Sub Header length
+        movzx   cx, [esi + sizeof.IPv4_FRAGMENT_entry + IPv4_header.VersionAndIHL]      ; Sub Header length
         and     cx, 0x000F
         shl     cx, 2
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Header size=%u\n", cx
         sub     ax, cx
         mov     edi, esi
-        mov     esi, [esi + FRAGMENT_entry.NextPtr]
+        mov     esi, [esi + IPv4_FRAGMENT_entry.NextPtr]
         cmp     esi, -1
         jne     .count_bytes
         mov     esi, [esp+4]
-        mov     [edi + FRAGMENT_entry.NextPtr], esi                             ; Add this packet to the chain, this simplifies the following code
+        mov     [edi + IPv4_FRAGMENT_entry.NextPtr], esi                                ; Add this packet to the chain, this simplifies the following code
-        mov     [esi + FRAGMENT_entry.NextPtr], -1
+        mov     [esi + IPv4_FRAGMENT_entry.NextPtr], -1
-        mov     [esi + FRAGMENT_entry.PrevPtr], edi
+        mov     [esi + IPv4_FRAGMENT_entry.PrevPtr], edi
-        mov     [esi + FRAGMENT_entry.Owner], ebx
+        mov     [esi + IPv4_FRAGMENT_entry.Owner], ebx
         test    eax, eax
         je      .destroy_slot_pop                                                       ; If we dont have enough space to allocate the buffer, discard all packets in slot
         mov     edx, [esp+4]                                                            ; Get pointer to first fragment entry back in edx
   .rebuild_packet_loop:
-        movzx   ecx, [edx + sizeof.FRAGMENT_entry + IPv4_header.FlagsAndFragmentOffset] ; Calculate the fragment offset
+        movzx   ecx, [edx + sizeof.IPv4_FRAGMENT_entry + IPv4_header.FlagsAndFragmentOffset] ; Calculate the fragment offset
         xchg    cl, ch                                                                  ;  intel byte order
         shl     cx, 3                                                                   ;   multiply by 8 and clear first 3 bits
         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
-        movzx   ebx, [edx + sizeof.FRAGMENT_entry + IPv4_header.VersionAndIHL]          ; Find header size (in ebx) of fragment
+        movzx   ebx, [edx + sizeof.IPv4_FRAGMENT_entry + IPv4_header.VersionAndIHL]     ; Find header size (in ebx) of fragment
         and     bx, 0x000F                                                              ;
         shl     bx, 2                                                                   ;
-        lea     esi, [edx + sizeof.FRAGMENT_entry]                                      ; Set esi to the correct begin of fragment
+        lea     esi, [edx + sizeof.IPv4_FRAGMENT_entry]                                 ; Set esi to the correct begin of fragment
-        movzx   ecx, [edx + sizeof.FRAGMENT_entry + IPv4_header.TotalLength]            ; Calculate total length of fragment
+        movzx   ecx, [edx + sizeof.IPv4_FRAGMENT_entry + IPv4_header.TotalLength]       ; Calculate total length of fragment
         xchg    cl, ch                                                                  ;  intel byte order
         cmp     edi, eax                                                                ; Is this packet the first fragment ?
+        je      .first_fragment
+        sub     cx, bx                                                                  ; If not, dont copy the header
-        je      .first_fragment
+        add     esi, ebx                                                                ;
-        sub     cx, bx                                                                  ; If not, dont copy the header
+  .first_fragment:
         add     esi, ebx                                                                ;
   .first_fragment:
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Copying %u bytes from 0x%x to 0x%x\n", ecx, esi, edi
         push    cx                                                                      ; First copy dword-wise, then byte-wise
         shr     cx, 2                                                                   ;
+        rep movsd                                                                       ;
+        pop     cx                                                                      ;
-        rep movsd                                                                       ;
+        and     cx, 3                                                                   ;
-        pop     cx                                                                      ;
-        and     cx, 3                                                                   ;
+        rep movsb                                                                       ;
         rep movsb                                                                       ;
+        push    eax
-        push    eax
+        push    [edx + IPv4_FRAGMENT_entry.Owner]                                       ; we need to remeber the owner, in case this is the last packet
-        push    edx                                                                     ; Push pointer to fragment onto stack
+        push    [edx + IPv4_FRAGMENT_entry.NextPtr]                                     ; Set edx to the next pointer
-        mov     ebx, [edx + FRAGMENT_entry.Owner]                                       ; we need to remeber the owner, in case this is the last packet
+        push    edx                                                                     ; Push pointer to fragment onto stack
-        mov     edx, [edx + FRAGMENT_entry.NextPtr]                                     ; Set edx to the next pointer
+        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_input: Next Fragment: 0x%x\n", edx
         call    NET_packet_free                                                         ; free the previous fragment buffer (this uses the value from stack)
-        pop     eax
+        pop     edx ebx eax
         cmp     edx, -1                                                                 ; Check if it is last fragment in chain
         jne     .rebuild_packet_loop
-        pop     ecx
-        xchg    cl, ch
-        mov     edx, eax
+        pop     ecx
-        mov     [edx + IPv4_header.TotalLength], cx
+        xchg    cl, ch
-        add     esp, 8
+        mov     edx, eax
-        xchg    cl, ch
+        mov     [edx + IPv4_header.TotalLength], cx
 ;;; TODO: the RFC says we should check protocol number too
         push    eax ebx ecx edx
         mov     ax, [edx + IPv4_header.Identification]
-        mov     ecx, MAX_FRAGMENTS
+        mov     ecx, IPv4_MAX_FRAGMENTS
-        mov     esi, FRAGMENT_LIST
+        mov     esi, IPv4_FRAGMENT_LIST
         mov     ebx, [edx + IPv4_header.SourceAddress]
         mov     edx, [edx + IPv4_header.DestinationAddress]
   .find_slot:
-        cmp     [esi + FRAGMENT_slot.id], ax
+        cmp     [esi + IPv4_FRAGMENT_slot.id], ax
         jne     .try_next
-        cmp     [esi + FRAGMENT_slot.SrcIP], ebx
+        cmp     [esi + IPv4_FRAGMENT_slot.SrcIP], ebx
         jne     .try_next
-        cmp     [esi + FRAGMENT_slot.DstIP], edx
+        cmp     [esi + IPv4_FRAGMENT_slot.DstIP], edx
         je      .found_slot
   .try_next:
-        add     esi, sizeof.FRAGMENT_slot
+        add     esi, sizeof.IPv4_FRAGMENT_slot
         loop    .find_slot
 ;------------------------------------------------------------------
+;
 ; IPv4_output
 ;
-; IN: eax = dest ip
-;     ebx = output device ptr/0 for automatic choice
+; IN:   eax = Destination IP
-;     ecx = data length
+;       ecx = data length
-;     edx = source ip
+;       edx = Source IP
-;     di  = TTL shl 8 + protocol
+;       di  = TTL shl 8 + protocol
+;
-; OUT: eax = pointer to buffer start
+; OUT:  eax = pointer to buffer start
-;      ebx = pointer to device struct (needed for sending procedure)
+;       ebx = pointer to device struct (needed for sending procedure)
         DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_output: size=%u ip=0x%x\n", ecx, eax
         cmp     ecx, 65500              ; Max IPv4 packet size
         ja      .too_large
-        push    ecx eax edx di
+        push    ecx di eax
         call    IPv4_route              ; outputs device number in edi, dest ip in eax, source IP in edx
-        call    IPv4_route              ; outputs device number in edi, dest ip in eax
+        push    edx
         test    edi, edi
         jz      .loopback
         call    ARP_IP_to_MAC
         test    eax, 0xffff0000         ; error bits
         jnz     .arp_error
         push    ebx                     ; push the mac onto the stack
         push    ax
-        inc     [IP_packets_tx + edi]   ; update stats
+        inc     [IPv4_packets_tx + edi]   ; update stats
         mov     ebx, [NET_DRV_LIST + edi]
         lea     eax, [ebx + ETH_DEVICE.mac]
         mov     edx, esp
         mov     [edi + IPv4_header.VersionAndIHL], 0x45 ; IPv4, normal length (no Optional header)
         mov     [edi + IPv4_header.TypeOfService], 0    ; nothing special, just plain ip packet
         mov     [edi + IPv4_header.TotalLength], cx
         mov     [edi + IPv4_header.Identification], 0   ; fragment id: FIXME
         mov     [edi + IPv4_header.FlagsAndFragmentOffset], 0
-        pop     word [edi + IPv4_header.TimeToLive]     ; ttl shl 8 + protocol
-;               [edi + IPv4_header.Protocol]
         mov     [edi + IPv4_header.HeaderChecksum], 0
         popd    [edi + IPv4_header.SourceAddress]
         popd    [edi + IPv4_header.DestinationAddress]
+        pop     word[edi + IPv4_header.TimeToLive]      ; ttl shl 8 + protocol
+;               [edi + IPv4_header.Protocol]
         pop     ecx
         IPv4_checksum edi
         jnz     .arp_error
         push    ebx                     ; push the mac
         push    ax
-        inc     [IP_packets_tx + 4*edi]
+        inc     [IPv4_packets_tx + 4*edi]
         mov     ebx, [NET_DRV_LIST + 4*edi]
         lea     eax, [ebx + ETH_DEVICE.mac]
         mov     edx, esp
 ;---------------------------------------------------------------------------
+;
 ; IPv4_route
+;
 ; IN:   eax = Destination IP
-; OUT:  edi = device number*4
+;       edx = Source IP
-;       eax = ip of gateway if nescessary, unchanged otherwise
+; OUT:  eax = Destination IP (or gateway IP)
+;       edx = Source IP
+;       edi = device number*4
+; DESTROYED:
+;       ecx
+;
 ;---------------------------------------------------------------------------
 align 4
-IPv4_route:
+IPv4_route:     ; TODO: return error if no valid route found
         cmp     eax, 0xffffffff
         je      .broadcast
         xor     edi, edi
-        mov     ecx, NET_DEVICES_MAX
+        xor     edi, edi
   .loop:
-        mov     ebx, [IP_LIST+edi]
+        mov     ebx, [IP_LIST + edi]
-        and     ebx, [SUBNET_LIST+edi]
+        and     ebx, [SUBNET_LIST + edi]
         jz      .next
-        mov     edx, eax
-        and     edx, [SUBNET_LIST+edi]
+        mov     ecx, eax
+        and     ecx, [SUBNET_LIST + edi]
-        cmp     ebx, edx
-        jne     .next
-        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_route: %u\n", edi
-        ret
+        cmp     ebx, ecx
         je      .got_it
   .next:
         add     edi, 4
-        dec     ecx
-        jnz     .loop
+        cmp     edi, 4*NET_DEVICES_MAX
         jb      .loop
-  .invalid:
+        mov     eax, [GATEWAY_LIST + 4]         ; TODO: let user (or a user space daemon) configure default route
+  .broadcast:
+        mov     edi, 4                          ; TODO: same as above
+  .got_it:
+        DEBUGF  DEBUG_NETWORK_VERBOSE, "IPv4_route: %u\n", edi
+        test    edx, edx
+        jnz     @f
-        mov     eax, [GATEWAY_LIST+4]       ;;; FIXME
+        mov     edx, [IP_LIST + edi]
-  .broadcast:
+  @@:
         xor     ax, ax  ;;; TODO: replace this with real code
+        ret
+;-----------------------------------------------------------------
+;
+; IPv4_connect
+;
+;   IN: eax = socket pointer
+;  OUT: eax = 0 ok / -1 error
+;       ebx = error code
+;
+;-------------------------
+align 4
+IPv4_connect:
+        push    eax edx
+        lea     ecx, [eax + SOCKET.mutex]
+        call    mutex_lock
+        pop     edx eax
+; Fill in local IP
+        cmp     [eax + IP_SOCKET.LocalIP], 0
+        jne     @f
+        push    [IP_LIST + 4]                                   ; FIXME: use correct local IP
+        pop     [eax + IP_SOCKET.LocalIP]
+; Fill in remote IP
+        pushd   [edx + 4]
+        pop     [eax + IP_SOCKET.RemoteIP]
+; Set up data receiving queue
+        push    eax
+        init_queue (eax + SOCKET_QUEUE_LOCATION)
+        pop     eax
+        lea     ecx, [eax + SOCKET.mutex]
+        call    mutex_unlock
+        xor     eax, eax
         ret
 ;---------------------------------------------------------------------------
+;
   .error:
         mov     eax, -1
         ret
   .packets_tx:
-        mov     eax, [IP_packets_tx + eax]
+        mov     eax, [IPv4_packets_tx + eax]
         ret
   .packets_rx:
-        mov     eax, [IP_packets_rx + eax]
+        mov     eax, [IPv4_packets_rx + eax]
         ret

Subversion Repositories Kolibri OS

(root)/kernel/branches/Kolibri-acpi/network/IPv4.inc – Rev 3908 → 4265