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

 ;;          GNU GENERAL PUBLIC LICENSE                             ;;
 ;;             Version 2, June 1991                                ;;
 ;;                                                                 ;;
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-$Revision: 5594 $
+$Revision: 5984 $
         UDP_PACKETS_RX          rd  NET_DEVICES_MAX
 endg
-;-----------------------------------------------------------------
-;
-; UDP_init
+;-----------------------------------------------------------------;
-;
+;                                                                 ;
-;  This function resets all UDP variables
+; UDP_init: This function resets all UDP variables                ;
 ;                                                                 ;
-;-----------------------------------------------------------------
+;-----------------------------------------------------------------;
 macro   UDP_init {
 macro   UDP_checksum    IP1, IP2  { ; esi = ptr to udp packet, ecx = packet size, destroys: ecx, edx
 ; Pseudoheader
         mov     edx, IP_PROTO_UDP
-        add     dl, [IP1+1]
+        add     dl, byte[IP1+1]
-        adc     dh, [IP1+0]
+        adc     dh, byte[IP1+0]
-        adc     dl, [IP1+3]
+        adc     dl, byte[IP1+3]
-        adc     dh, [IP1+2]
+        adc     dh, byte[IP1+2]
-        adc     dl, [IP2+1]
+        adc     dl, byte[IP2+1]
-        adc     dh, [IP2+0]
+        adc     dh, byte[IP2+0]
-        adc     dl, [IP2+3]
+        adc     dl, byte[IP2+3]
-        adc     dh, [IP2+2]
+        adc     dh, byte[IP2+2]
         add     [esi+UDP_header.Checksum], dx   ; this final instruction will set or clear ZF :)
 }
-;-----------------------------------------------------------------
-;
-; UDP_input:
-;
-;  Called by IPv4_input,
+;-----------------------------------------------------------------;
-;  this procedure will inject the udp data diagrams in the application sockets.
+;                                                                 ;
-;
+; UDP_input: Inject the UDP data in the application sockets.      ;
-;  IN:   [esp]  = Pointer to buffer
+;                                                                 ;
-;       [esp+4] = size of buffer
+;   IN: [esp] = ptr to buffer                                     ;
-;       ebx = ptr to device struct
+;       ebx = ptr to device struct                                ;
-;       ecx = UDP Packet size
+;       ecx = UDP packet size                                     ;
+;       edx = ptr to IPv4 header                                  ;
-;       esi = ptr to UDP header
+;       esi = ptr to UDP packet data                              ;
-;       edi = ptr to ipv4 source and dest address
+;       edi = interface number*4                                  ;
-;
+;                                                                 ;
-;  OUT: /
+;  OUT: /                                                         ;
 ;                                                                 ;
-;-----------------------------------------------------------------
+;-----------------------------------------------------------------;
         neg     [esi + UDP_header.Checksum]     ; substract checksum from 0
         jz      .no_checksum                    ; if checksum is zero, it is considered valid
         ; otherwise, we will re-calculate the checksum and add it to this value, thus creating 0 when it is correct
         mov     eax, edx
-        UDP_checksum (edi), (edi+4)
+        UDP_checksum (eax+IPv4_header.SourceAddress), (eax+IPv4_header.DestinationAddress)
         jnz     .checksum_mismatch
         call    mutex_lock
         popa
         mov     cx, [esi + UDP_header.SourcePort]
-        mov     dx, [esi + UDP_header.DestinationPort]
-        mov     edi, [edi + 4]                          ; ipv4 source address
+        mov     dx, [esi + UDP_header.DestinationPort]
-        mov     eax, net_sockets
         mov     eax, net_sockets
   .next_socket:
         mov     eax, [eax + SOCKET.NextPtr]
         or      eax, eax
         pusha
         mov     ecx, socket_mutex
         call    mutex_unlock
         popa
+        ;;; TODO: when packet is processed, check more sockets?!
-        ;;; TODO: when packet is processed, check more sockets!
+; FIXME: check remote IP if possible
+;
 ;        cmp     [eax + IP_SOCKET.RemoteIP], 0xffffffff
 ;        je      @f
-;        cmp     [eax + IP_SOCKET.RemoteIP], edi
+;        cmp     [eax + IP_SOCKET.RemoteIP],
-;        jne     .next_socket
-;       @@:
 ;        jne     .next_socket
-; FIXME: UDP should check remote IP, but not under all circumstances!
+;       @@:
         lea     ecx, [eax + SOCKET.mutex]
         call    mutex_lock
         popa
-  .updatesock:
-        call    NET_ptr_to_num4
+  .updatesock:
         inc     [UDP_PACKETS_RX + edi]
         movzx   ecx, [esi + UDP_header.Length]
         call    NET_BUFF_free
         ret
-;-----------------------------------------------------------------
+;-----------------------------------------------------------------;
-;
+;                                                                 ;
-; UDP_output
+; UDP_output: Create an UDP packet.                               ;
-;
+;                                                                 ;
-; IN: eax = socket pointer
+;  IN:  eax = socket pointer                                      ;
-;     ecx = number of bytes to send
+;       ecx = number of bytes to send                             ;
-;     esi = pointer to data
+;       esi = pointer to data                                     ;
-;
+;                                                                 ;
-; OUT: eax = -1 on error
+; OUT:  eax = -1 on error                                         ;
 ;                                                                 ;
-;-----------------------------------------------------------------
+;-----------------------------------------------------------------;
         mov     dx, [eax + UDP_SOCKET.LocalPort]
         DEBUGF  DEBUG_NETWORK_VERBOSE, "local port=%x\n", dx
         sub     esp, 4                                          ; Data ptr will be placed here
-        push    edx esi
-        mov     edx, [eax + IP_SOCKET.LocalIP]
+        push    edx esi
+        mov     ebx, [eax + IP_SOCKET.device]
-        mov     ebx, [eax + IP_SOCKET.device]
+        mov     edx, [eax + IP_SOCKET.LocalIP]
+        mov     edi, [eax + IP_SOCKET.RemoteIP]
-        mov     eax, [eax + IP_SOCKET.RemoteIP]
+        mov     al, [eax + IP_SOCKET.ttl]
-        mov     di, IP_PROTO_UDP shl 8 + 128
+        mov     ah, IP_PROTO_UDP
         add     ecx, sizeof.UDP_header
         call    IPv4_output
         jz      .fail
         ret
-;-----------------------------------------------------------------
+;-----------------------------------------------------------------;
+;                                                                 ;
-;
+; UDP_connect                                                     ;
+;                                                                 ;
-; UDP_connect
+;   IN: eax = socket pointer                                      ;
-;
+;                                                                 ;
-;   IN: eax = socket pointer
+;  OUT: eax = 0 on success                                        ;
-;  OUT: eax = 0 ok / -1 error
+;       eax = -1 on error                                         ;
-;       ebx = error code
+;       ebx = error code on error                                 ;
 ;                                                                 ;
-;-------------------------
+;-----------------------------------------------------------------;
         jne     @f
         call    SOCKET_find_port
        @@:
-        push    eax
-        init_queue (eax + SOCKET_QUEUE_LOCATION)                ; Set up data receiving queue
-        pop     eax
         push    eax
         lea     ecx, [eax + SOCKET.mutex]
         call    mutex_unlock
         pop     eax
         call    SOCKET_is_connected
-        xor     eax, eax
+        xor     eax, eax
         ret
+;-----------------------------------------------------------------;
-;-----------------------------------------------------------------
+;                                                                 ;
-;
+; UDP_disconnect                                                  ;
-; UDP_disconnect
+;                                                                 ;
 ;   IN: eax = socket pointer                                      ;
-;   IN: eax = socket pointer
+;                                                                 ;
-;  OUT: eax = socket pointer
+;  OUT: eax = socket pointer                                      ;
-;---------------------------------------------------------------------------
-;
-; UDP_API
+;-----------------------------------------------------------------;
-;
+;                                                                 ;
-; This function is called by system function 75
+; UDP_api: This function is called by system function 76          ;
-;
+;                                                                 ;
-; IN:  subfunction number in bl
+;  IN: bl = subfunction number in bl                              ;
-;      device number in bh
+;      bh = device number in bh                                   ;
-;      ecx, edx, .. depends on subfunction
 ;      ecx, edx, .. depends on subfunction                        ;
-; OUT:
+;                                                                 ;
 ; OUT: depends on subfunction                                     ;
-;---------------------------------------------------------------------------
+;                                                                 ;

Subversion Repositories Kolibri OS

(root)/kernel/branches/Kolibri-acpi/network/udp.inc – Rev 5594 → 5984