Subversion Repositories Kolibri OS

;;         GNU GENERAL PUBLIC LICENSE                              ;;

;;         GNU GENERAL PUBLIC LICENSE                              ;;

;;          Version 2, June 1991                                   ;;

;;          Version 2, June 1991                                   ;;

;;                                                                 ;;

;;                                                                 ;;

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

$Revision: 6011 $

$Revision: 6413 $

struct  SOCKET

struct  SOCKET

        optlen                  dd ?

        optlen                  dd ?

        optval                  dd ?

        optval                  dd ?

SOCKETBUFFSIZE          = 4096          ; in bytes

SOCKET_STRUCT_SIZE      = 4096          ; in bytes

SOCKET_QUEUE_SIZE       = 10            ; maximum number of incoming packets queued for 1 socket

SOCKET_QUEUE_SIZE       = 10            ; maximum number of incoming packets queued for 1 socket

; the incoming packet queue for sockets is placed in the socket struct itself, at this location from start

; the incoming packet queue for sockets is placed in the socket struct itself, at this location from start

SOCKET_QUEUE_LOCATION   = (SOCKETBUFFSIZE - SOCKET_QUEUE_SIZE*sizeof.socket_queue_entry - sizeof.queue)

SOCKET_QUEUE_LOCATION   = (SOCKET_STRUCT_SIZE - SOCKET_QUEUE_SIZE*sizeof.socket_queue_entry - sizeof.queue)

        call    socket_num_to_ptr

        call    socket_num_to_ptr

        test    eax, eax

        test    eax, eax

        jz      .invalid

        jz      .invalid

        mov     ecx, SOCKETBUFFSIZE/4

        mov     ecx, SOCKET_STRUCT_SIZE/4

        rep movsd

        rep movsd

        push    esi

        push    esi

        mov     esi, eax

        mov     esi, eax

        push    edx

        push    edx

        stdcall create_ring_buffer, SOCKET_MAXDATA, PG_SWR

        stdcall create_ring_buffer, SOCKET_BUFFER_SIZE, PG_SWR

        pop     edx

        pop     edx

        test    eax, eax

        test    eax, eax

        mov     [esi + RING_BUFFER.start_ptr], eax

        mov     [esi + RING_BUFFER.start_ptr], eax

        mov     [esi + RING_BUFFER.write_ptr], eax

        mov     [esi + RING_BUFFER.write_ptr], eax

        mov     [esi + RING_BUFFER.read_ptr], eax

        mov     [esi + RING_BUFFER.read_ptr], eax

        add     eax, SOCKET_MAXDATA

        add     eax, SOCKET_BUFFER_SIZE

        mov     [esi + RING_BUFFER.end_ptr], eax

        mov     [esi + RING_BUFFER.end_ptr], eax

        mov     eax, esi

        mov     eax, esi

        lea     ecx, [eax + RING_BUFFER.mutex]

        lea     ecx, [eax + RING_BUFFER.mutex]

        call    mutex_lock                                      ; TODO: check what registers this function actually destroys

        call    mutex_lock                                      ; TODO: check what registers this function actually destroys

        popa

        popa

        mov     edi, SOCKET_MAXDATA

        mov     edi, SOCKET_BUFFER_SIZE

        sub     edi, [eax + RING_BUFFER.size]                   ; available buffer size in edi

        sub     edi, [eax + RING_BUFFER.size]                   ; available buffer size in edi

        cmp     ecx, edi

        cmp     ecx, edi

        jbe     .copy

        jbe     .copy

        mov     ecx, edi

        mov     ecx, edi

; update write ptr

; update write ptr

        push    edi

        push    edi

        add     edi, ecx

        add     edi, ecx

        cmp     edi, [eax + RING_BUFFER.end_ptr]

        cmp     edi, [eax + RING_BUFFER.end_ptr]

        jb      @f

        jb      @f

  @@:

  @@:

        mov     [eax + RING_BUFFER.write_ptr], edi

        mov     [eax + RING_BUFFER.write_ptr], edi

        add     [eax + RING_BUFFER.read_ptr], ecx

        add     [eax + RING_BUFFER.read_ptr], ecx

        mov     edx, [eax + RING_BUFFER.end_ptr]

        mov     edx, [eax + RING_BUFFER.end_ptr]

        cmp     [eax + RING_BUFFER.read_ptr], edx

        cmp     [eax + RING_BUFFER.read_ptr], edx

        sub     [eax + RING_BUFFER.read_ptr], SOCKET_MAXDATA

        sub     [eax + RING_BUFFER.read_ptr], SOCKET_BUFFER_SIZE

       @@:

       @@:

        push    eax ecx

        push    eax ecx

        jnz     .un_block

        jnz     .un_block

; Socket and thread exists and socket is of non blocking type.

; Socket and thread exists and socket is of non blocking type.

; We'll try to flag an event to the thread.

; We'll try to flag an event to the thread.

        DEBUGF  DEBUG_NETWORK_VERBOSE, "SOCKET_notify: poking thread %u!\n", eax

        DEBUGF  DEBUG_NETWORK_VERBOSE, "SOCKET_notify: poking thread %u!\n", ebx

        pop     esi ecx ebx

        pop     esi ecx ebx

align 4

align 4

socket_alloc:

socket_alloc:

        stdcall kernel_alloc, SOCKETBUFFSIZE

        stdcall kernel_alloc, SOCKET_STRUCT_SIZE

        or      eax, eax

        or      eax, eax

        jz      .nomem

        jz      .nomem

        DEBUGF  DEBUG_NETWORK_VERBOSE, "SOCKET_alloc: ptr=%x\n", eax

        DEBUGF  DEBUG_NETWORK_VERBOSE, "SOCKET_alloc: ptr=%x\n", eax

        push    eax

        push    eax

        mov     edi, eax

        mov     edi, eax

        mov     ecx, SOCKETBUFFSIZE / 4

        mov     ecx, SOCKET_STRUCT_SIZE / 4

        xor     eax, eax

        xor     eax, eax