WebSVN – Kolibri OS – Diff – /kernel/branches/net/network/stack.inc


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                                 ;;
;; Copyright (C) KolibriOS team 2004-2011. All rights reserved.    ;;
;; Distributed under terms of the GNU General Public License       ;;
;;                                                                 ;;
;;  STACK.INC                                                      ;;
;;                                                                 ;;
;;  TCP/IP stack for KolibriOS                                     ;;
;;                                                                 ;;
;;    Written by hidnplayr@kolibrios.org                           ;;
;;                                                                 ;;
;;     Some parts of code are based on the work of:                ;;
;;      Mike Hibbett (menuetos network stack)                      ;;
;;      Eugen Brasoveanu (solar os network stack and drivers)      ;;
;;      mike.dld (kolibrios socket code)                           ;;
;;                                                                 ;;
;;     TCP part is based on 4.4BSD                                 ;;
;;                                                                 ;;
;;          GNU GENERAL PUBLIC LICENSE                             ;;
;;             Version 2, June 1991                                ;;
;;                                                                 ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
$Revision: 2731 $
 
__DEBUG_LEVEL_OLD__     equ __DEBUG_LEVEL__     ; use seperate debug level for network part of kernel
__DEBUG_LEVEL__         equ 1
 
uglobal
        net_10ms        dd ?
        net_tmr_count   dw ?
endg
 
MAX_NET_DEVICES         = 16
 
MIN_EPHEMERAL_PORT      = 49152
MAX_EPHEMERAL_PORT      = 61000
 
; Ethernet protocol numbers
ETHER_ARP               = 0x0608
ETHER_IPv4              = 0x0008
ETHER_IPv6              = 0xDD86
ETHER_PPP_DISCOVERY     = 0x6388
ETHER_PPP_SESSION       = 0x6488
 
;Protocol family
AF_UNSPEC               = 0
AF_UNIX                 = 1
AF_INET4                = 2
AF_INET6                = 10
 
; Internet protocol numbers
IP_PROTO_IP             = 0
IP_PROTO_ICMP           = 1
IP_PROTO_TCP            = 6
IP_PROTO_UDP            = 17
 
; Socket types
SOCK_STREAM             = 1
SOCK_DGRAM              = 2
SOCK_RAW                = 3
 
; Socket options
SO_ACCEPTCON            = 1 shl 0
SO_BROADCAST            = 1 shl 1
SO_DEBUG                = 1 shl 2
SO_DONTROUTE            = 1 shl 3
SO_KEEPALIVE            = 1 shl 4
SO_OOBINLINE            = 1 shl 5
SO_REUSEADDR            = 1 shl 6
SO_REUSEPORT            = 1 shl 7
SO_USELOOPBACK          = 1 shl 8
 
 
; Socket States
SS_NOFDREF              = 0x001       ; no file table ref any more
SS_ISCONNECTED          = 0x002       ; socket connected to a peer
SS_ISCONNECTING         = 0x004       ; in process of connecting to peer
SS_ISDISCONNECTING      = 0x008       ; in process of disconnecting
SS_CANTSENDMORE         = 0x010       ; can't send more data to peer
SS_CANTRCVMORE          = 0x020       ; can't receive more data from peer
SS_RCVATMARK            = 0x040       ; at mark on input
SS_ISABORTING           = 0x080       ; aborting fd references - close()
SS_RESTARTSYS           = 0x100       ; restart blocked system calls
SS_ISDISCONNECTED       = 0x800       ; socket disconnected from peer
 
SS_ASYNC                = 0x100       ; async i/o notify
SS_ISCONFIRMING         = 0x200       ; deciding to accept connection req
SS_MORETOCOME           = 0x400
 
 
SOCKET_MAXDATA          = 4096*32     ; must be 4096*(power of 2) where 'power of 2' is at least 8
 
; Network driver types
NET_TYPE_ETH            = 1
NET_TYPE_SLIP           = 2
 
MAX_backlog             = 20          ; maximum backlog for stream sockets
 
; Error Codes
ENOBUFS                 = 55
ECONNREFUSED            = 61
ECONNRESET              = 52
ETIMEDOUT               = 60
ECONNABORTED            = 53
 
; Api protocol numbers
API_ETH                 = 0
API_IPv4                = 1
API_ICMP                = 2
API_UDP                 = 3
 
API_TCP                 = 4
API_ARP                 = 5
API_PPPOE               = 6
 
struct  NET_DEVICE
 
        type            dd ?    ; Type field
        mtu             dd ?    ; Maximal Transmission Unit
        name            dd ?    ; Ptr to 0 terminated string
 
        unload          dd ?    ; Ptrs to driver functions
        reset           dd ?    ;
        transmit        dd ?    ;
 
        bytes_tx        dq ?    ; Statistics, updated by the driver
        bytes_rx        dq ?    ;
        packets_tx      dd ?    ;
        packets_rx      dd ?    ;
 
;       hwacc           dd ?    ; bitmask stating available hardware accelerations (offload engines)
 
ends
 
 
; Exactly as it says..
macro pseudo_random reg {
        add     reg, [esp]
        rol     reg, 5
        xor     reg, [timer_ticks]
        add     reg, [CPU_FREQ]
        imul    reg, 214013
        xor     reg, 0xdeadbeef
        rol     reg, 9
}
 
macro ntohd reg {
 
        rol     word reg, 8
        rol     dword reg, 16
        rol     word reg , 8
 
}
 
macro ntohw reg {
 
        rol     word reg, 8
 
}
 
 
include "queue.inc"
 
include "ethernet.inc"
 
;include "PPPoE.inc"
 
include "ARP.inc"
include "IPv4.inc"
;include "IPv6.inc"
 
include "icmp.inc"
include "udp.inc"
include "tcp.inc"
 
include "socket.inc"
 
 
 
align 4
uglobal
 
        NET_RUNNING     dd  ?
        NET_DEFAULT     dd  ?
        NET_DRV_LIST    rd  MAX_NET_DEVICES
 
endg
 
 
;-----------------------------------------------------------------
;
; stack_init
;
;  This function calls all network init procedures
;
;  IN:  /
;  OUT: /
;
;-----------------------------------------------------------------
align 4
stack_init:
 
; Init the network drivers list
        xor     eax, eax
        mov     edi, NET_RUNNING
        mov     ecx, (MAX_NET_DEVICES + 2)
        rep     stosd
 
 
;        PPPOE_init
 
        IPv4_init
;        IPv6_init
        ICMP_init
 
        ARP_init
        UDP_init
        TCP_init
 
        SOCKET_init
 
        mov     [net_tmr_count], 0
 
        ret
 
 
;-----------------------------------------------------------------
;
; stack_handler
;
;  This function is called in kernel loop
;
;  IN:  /
;  OUT: /
;
;-----------------------------------------------------------------
align 4
stack_handler:
 
        cmp     [NET_RUNNING], 0
        je      .exit
 
        ; Test for 10ms tick
        mov     eax, [timer_ticks]
        cmp     eax, [net_10ms]
        je      .exit
        mov     [net_10ms], eax
 
        test    [net_10ms], 0x0f        ; 160ms
        jnz     .exit
 
        TCP_timer_160ms
 
        test    [net_10ms], 0x3f        ; 640ms
        jnz     .exit
 
        TCP_timer_640ms
        ARP_decrease_entry_ttls
        IPv4_decrease_fragment_ttls
 
  .exit:
        ret
 
 
 
;-----------------------------------------------------------------
;
; NET_add_device:
;
;  This function is called by the network drivers,
;  to register each running NIC to the kernel
;
;  IN:  Pointer to device structure in ebx
;  OUT: Device num in eax, -1 on error
;
;-----------------------------------------------------------------
align 4
NET_add_device:
 
        DEBUGF  1,"NET_Add_Device: %x\n", ebx   ;;; TODO: use mutex to lock net device list
 
        mov     eax, [NET_RUNNING]
        cmp     eax, MAX_NET_DEVICES
        jae     .error
 
;----------------------------------
; Check if device is already listed
        mov     eax, ebx
        mov     ecx, MAX_NET_DEVICES    ; We need to check whole list because a device may be removed without re-organizing list
        mov     edi, NET_DRV_LIST
 
        repne   scasd                   ; See if device is already in the list
        jz      .error
 
;----------------------------
; Find empty slot in the list
        xor     eax, eax
        mov     ecx, MAX_NET_DEVICES
        mov     edi, NET_DRV_LIST
 
        repne   scasd
        jnz     .error
 
        sub     edi, 4
 
;-----------------------------
; Add device to the found slot
        mov     [edi], ebx              ; add device to list
 
        mov     eax, edi                ; Calculate device number in eax
        sub     eax, NET_DRV_LIST
        shr     eax, 2
 
        inc     [NET_RUNNING]           ; Indicate that one more network device is up and running
 
        cmp     eax, 1                  ; If it's the first network device, try to set it as default
        jne     @f
        push    eax
        call    NET_set_default
        pop     eax
       @@:
 
        DEBUGF  1,"Device number: %u\n", eax
        ret
 
  .error:
        or      eax, -1
        DEBUGF  2,"Adding network device failed\n"
        ret
 
 
 
;-----------------------------------------------------------------
;
; NET_set_default
;
;  API to set the default interface
;
;  IN:  Device num in eax
;  OUT: Device num in eax, -1 on error
;
;-----------------------------------------------------------------
align 4
NET_set_default:
 
        DEBUGF  1,"NET_set_default %x\n", eax
 
        cmp     eax, MAX_NET_DEVICES
        jae     .error
 
        cmp     [NET_DRV_LIST+eax*4], 0
        je      .error
 
        mov     [NET_DEFAULT], eax
 
        DEBUGF  1,"Device number %u is now default!\n", eax
        ret
 
  .error:
        or      eax, -1
        DEBUGF  2,"Setting default network device failed\n"
        ret
 
 
;-----------------------------------------------------------------
;
; NET_Remove_Device:
;
;  This function is called by etwork drivers,
;  to unregister network devices from the kernel
;                                                                                                                                d
;  IN:  Pointer to device structure in ebx
;  OUT: eax: -1 on error
;
;-----------------------------------------------------------------
align 4
NET_remove_device:
 
        cmp     [NET_RUNNING], 0
        je      .error
 
        cmp     [NET_DRV_LIST], ebx
        jne     @f
        mov     [NET_DRV_LIST], 0
        cmp     [NET_RUNNING], 1
        je      @f
        ; there are still active devices, find one and make it default
        xor     eax, eax
        mov     ecx, MAX_NET_DEVICES
        mov     edi, NET_DRV_LIST
        repe    scasd
        je      @f
        shr     edi, 2
        dec     edi
        mov     [NET_DEFAULT], edi
       @@:
 
;----------------------------
; Find the driver in the list
 
        mov     eax, ebx
        mov     ecx, MAX_NET_DEVICES
        mov     edi, NET_DRV_LIST+4
 
        repne   scasd
        jnz     .error
 
;------------------------
; Remove it from the list
 
        xor     eax, eax
        mov     dword [edi-4], eax
 
        dec     [NET_RUNNING]
        ret
 
  .error:
        or      eax, -1
        ret
 
 
 
;-----------------------------------------------------------------
;
; NET_ptr_to_num
;
; IN:  ebx = ptr to device struct
; OUT: edi = -1 on error, device number otherwise
;
;-----------------------------------------------------------------
align 4
NET_ptr_to_num:
        push    ecx
 
        mov     ecx, MAX_NET_DEVICES
        mov     edi, NET_DRV_LIST
 
  .loop:
        cmp     ebx, [edi]
        jz      .found
        add     edi, 4
        dec     ecx
        jnz     .loop
 
        ; repnz  scasd could work too if eax is used instead of ebx!
 
        or      edi, -1
 
        pop     ecx
        ret
 
  .found:
        sub     edi, NET_DRV_LIST
        shr     edi, 2
 
        pop     ecx
        ret
 
;-----------------------------------------------------------------
;
; checksum_1
;
;  This is the first of two functions needed to calculate a checksum.
;
;  IN:  edx = start offset for semi-checksum
;       esi = pointer to data
;       ecx = data size
;  OUT: edx = semi-checksum
;
;
; Code was optimized by diamond
;
;-----------------------------------------------------------------
align 4
checksum_1:
 
        shr     ecx, 1
        pushf
        jz      .no_2
 
        shr     ecx, 1
        pushf
        jz      .no_4
 
        shr     ecx, 1
        pushf
        jz      .no_8
 
  .loop:
        add     dl, [esi+1]
        adc     dh, [esi+0]
 
        adc     dl, [esi+3]
        adc     dh, [esi+2]
 
        adc     dl, [esi+5]
        adc     dh, [esi+4]
 
        adc     dl, [esi+7]
        adc     dh, [esi+6]
 
        adc     edx, 0
        add     esi, 8
 
        dec     ecx
        jnz     .loop
 
        adc     edx, 0
 
  .no_8:
        popf
        jnc     .no_4
 
        add     dl, [esi+1]
        adc     dh, [esi+0]
 
        adc     dl, [esi+3]
        adc     dh, [esi+2]
 
        adc     edx, 0
        add     esi, 4
 
  .no_4:
        popf
        jnc     .no_2
 
        add     dl, [esi+1]
        adc     dh, [esi+0]
 
        adc     edx, 0
        inc     esi
        inc     esi
 
  .no_2:
        popf
        jnc     .end
 
        add     dh, [esi+0]
        adc     edx, 0
  .end:
        ret
 
;-----------------------------------------------------------------
;
; checksum_2
;
;  This function calculates the final ip/tcp/udp checksum for you
;
;  IN:  edx = semi-checksum
;  OUT: dx = checksum (in INET byte order)
;
;-----------------------------------------------------------------
align 4
checksum_2:
 
        mov     ecx, edx
        shr     ecx, 16
        and     edx, 0xffff
        add     edx, ecx
 
        mov     ecx, edx
        shr     ecx, 16
        add     dx, cx
        test    dx, dx          ; it seems that ZF is not set when CF is set :(
        not     dx
        jnz     .not_zero
        dec     dx
  .not_zero:
        xchg    dl, dh
 
        DEBUGF 1,"Checksum: %x\n", dx
 
        ret
 
 
 
;----------------------------------------------------------------
;
;  System function to work with network devices (73)
;
;----------------------------------------------------------------
align 4
sys_network:                    ; FIXME: make default device easily accessible
 
        cmp     ebx, -1
        jne     @f
 
        mov     eax, [NET_RUNNING]
        jmp     .return
 
   @@:
        cmp     bh, MAX_NET_DEVICES             ; Check if device number exists
        jae     .doesnt_exist
 
        mov     esi, ebx
        and     esi, 0x0000ff00
        shr     esi, 6
 
        cmp     dword [esi + NET_DRV_LIST], 0   ; check if driver is running
        je      .doesnt_exist
 
        mov     eax, [esi + NET_DRV_LIST]
 
        and     ebx, 0x000000ff
        cmp     ebx, .number
        ja      .doesnt_exist
        jmp     dword [.table + 4*ebx]
 
  .table:
        dd      .get_type               ; 0
        dd      .get_dev_name           ; 1
        dd      .reset                  ; 2
        dd      .stop                   ; 3
        dd      .get_ptr                ; 4
        dd      .get_drv_name           ; 5
        dd      .set_default            ; 6
  .number = ($ - .table) / 4 - 1
 
  .get_type:                            ; 0 = Get device type (ethernet/token ring/...)
 
        mov     eax, [eax + NET_DEVICE.type]
        jmp     .return
 
 
  .get_dev_name:                        ; 1 = Get device name
 
        mov     esi, [eax + NET_DEVICE.name]
        mov     edi, ecx
 
        mov     ecx, 64/4 ; max length
        rep     movsd
 
        xor     eax, eax
        jmp     .return
 
  .reset:                               ; 2 = Reset the device
 
        call    [eax + NET_DEVICE.reset]
        jmp     .return
 
  .stop:                                ; 3 = Stop driver for this device
 
        call    [eax + NET_DEVICE.unload]
        jmp     .return
 
 
  .get_ptr:                             ; 4 = Get driver pointer
 
        jmp     .return
 
 
  .get_drv_name:                        ; 5 = Get driver name
 
        xor     eax, eax
        jmp     .return
 
 
  .set_default:                         ; 6 = Set default device
 
        call    NET_set_default
        jmp     .return
 
  .doesnt_exist:
        DEBUGF  1,"sys_network: invalid device/function specified!\n"
        mov     eax, -1
 
  .return:
        mov     [esp+32], eax
        ret
 
 
;----------------------------------------------------------------
;
;  System function to work with protocols  (76)
;
;----------------------------------------------------------------
align 4
sys_protocols:
        cmp     bh, MAX_NET_DEVICES             ; Check if device number exists
        jae     .doesnt_exist
 
        mov     esi, ebx
        and     esi, 0x0000ff00
        shr     esi, 6                          ; now we have the device num * 4 in esi
        cmp     [esi + NET_DRV_LIST], 0         ; check if driver is running
        je      .doesnt_exist
 
        push    .return                         ; return address (we will be using jumps instead of calls)
 
        mov     eax, ebx                        ; set ax to protocol number
        shr     eax, 16                         ;
 
        cmp     ax, API_ETH
        je      ETH_api
 
        cmp     ax, API_IPv4
        je      IPv4_api
 
        cmp     ax, API_ICMP
        je      ICMP_api
 
        cmp     ax, API_UDP
        je      UDP_api
 
 
 
 
        cmp     ax, API_TCP
        je      TCP_api
 
        cmp     ax, API_ARP
        je      ARP_api
 
;        cmp     ax, API_PPPOE
;        je      PPPoE_api
 
        add     esp, 4                           ; if we reached here, no function was called, so we need to balance stack
 
  .doesnt_exist:
        DEBUGF  1,"sys_protocols: protocol %u doesnt exist on device %u!\n", ax, bh
        mov     eax, -1
 
  .return:
        mov     [esp+28+4], eax                 ; return eax value to the program
        ret
 
 
__DEBUG_LEVEL__ equ __DEBUG_LEVEL_OLD__

Rev 2621	Rev 2731
1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2	;; ;;	2	;; ;;
3	;; Copyright (C) KolibriOS team 2004-2011. All rights reserved. ;;	3	;; Copyright (C) KolibriOS team 2004-2011. All rights reserved. ;;
4	;; Distributed under terms of the GNU General Public License ;;	4	;; Distributed under terms of the GNU General Public License ;;
5	;; ;;	5	;; ;;
6	;; STACK.INC ;;	6	;; STACK.INC ;;
7	;; ;;	7	;; ;;
8	;; TCP/IP stack for KolibriOS ;;	8	;; TCP/IP stack for KolibriOS ;;
9	;; ;;	9	;; ;;
10	;; Written by hidnplayr@kolibrios.org ;;	10	;; Written by hidnplayr@kolibrios.org ;;
11	;; ;;	11	;; ;;
12	;; Some parts of code are based on the work of: ;;	12	;; Some parts of code are based on the work of: ;;
13	;; Mike Hibbett (menuetos network stack) ;;	13	;; Mike Hibbett (menuetos network stack) ;;
14	;; Eugen Brasoveanu (solar os network stack and drivers) ;;	14	;; Eugen Brasoveanu (solar os network stack and drivers) ;;
15	;; mike.dld (kolibrios socket code) ;;	15	;; mike.dld (kolibrios socket code) ;;
16	;; ;;	16	;; ;;
17	;; TCP part is based on 4.4BSD ;;	17	;; TCP part is based on 4.4BSD ;;
18	;; ;;	18	;; ;;
19	;; GNU GENERAL PUBLIC LICENSE ;;	19	;; GNU GENERAL PUBLIC LICENSE ;;
20	;; Version 2, June 1991 ;;	20	;; Version 2, June 1991 ;;
21	;; ;;	21	;; ;;
22	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	22	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
23		23
24	$Revision: 2621 $	24	$Revision: 2731 $
25		25
26	__DEBUG_LEVEL_OLD__ equ __DEBUG_LEVEL__ ; use seperate debug level for network part of kernel	26	__DEBUG_LEVEL_OLD__ equ __DEBUG_LEVEL__ ; use seperate debug level for network part of kernel
27	__DEBUG_LEVEL__ equ 1	27	__DEBUG_LEVEL__ equ 1
28		28
29	uglobal	29	uglobal
30	net_10ms dd ?	30	net_10ms dd ?
31	net_tmr_count dw ?	31	net_tmr_count dw ?
32	endg	32	endg
33		33
34	MAX_NET_DEVICES = 16	34	MAX_NET_DEVICES = 16
35		35
36	MIN_EPHEMERAL_PORT = 49152	36	MIN_EPHEMERAL_PORT = 49152
37	MAX_EPHEMERAL_PORT = 61000	37	MAX_EPHEMERAL_PORT = 61000
38		38
39	; Ethernet protocol numbers	39	; Ethernet protocol numbers
40	ETHER_ARP = 0x0608	40	ETHER_ARP = 0x0608
41	ETHER_IPv4 = 0x0008	41	ETHER_IPv4 = 0x0008
-		42	ETHER_IPv6 = 0xDD86
42	ETHER_PPP_DISCOVERY = 0x6388	43	ETHER_PPP_DISCOVERY = 0x6388
43	ETHER_PPP_SESSION = 0x6488	44	ETHER_PPP_SESSION = 0x6488
44		45
45	;Protocol family	46	;Protocol family
46	AF_UNSPEC = 0	47	AF_UNSPEC = 0
47	AF_UNIX = 1	48	AF_UNIX = 1
48	AF_INET4 = 2	49	AF_INET4 = 2
49	AF_INET6 = 10	50	AF_INET6 = 10
50		51
51	; Internet protocol numbers	52	; Internet protocol numbers
52	IP_PROTO_IP = 0	53	IP_PROTO_IP = 0
53	IP_PROTO_ICMP = 1	54	IP_PROTO_ICMP = 1
54	IP_PROTO_TCP = 6	55	IP_PROTO_TCP = 6
55	IP_PROTO_UDP = 17	56	IP_PROTO_UDP = 17
56		57
57	; Socket types	58	; Socket types
58	SOCK_STREAM = 1	59	SOCK_STREAM = 1
59	SOCK_DGRAM = 2	60	SOCK_DGRAM = 2
60	SOCK_RAW = 3	61	SOCK_RAW = 3
61		62
62	; Socket options	63	; Socket options
63	SO_ACCEPTCON = 1 shl 0	64	SO_ACCEPTCON = 1 shl 0
64	SO_BROADCAST = 1 shl 1	65	SO_BROADCAST = 1 shl 1
65	SO_DEBUG = 1 shl 2	66	SO_DEBUG = 1 shl 2
66	SO_DONTROUTE = 1 shl 3	67	SO_DONTROUTE = 1 shl 3
67	SO_KEEPALIVE = 1 shl 4	68	SO_KEEPALIVE = 1 shl 4
68	SO_OOBINLINE = 1 shl 5	69	SO_OOBINLINE = 1 shl 5
69	SO_REUSEADDR = 1 shl 6	70	SO_REUSEADDR = 1 shl 6
70	SO_REUSEPORT = 1 shl 7	71	SO_REUSEPORT = 1 shl 7
71	SO_USELOOPBACK = 1 shl 8	72	SO_USELOOPBACK = 1 shl 8
72		73
73		74
74	; Socket States	75	; Socket States
75	SS_NOFDREF = 0x001 ; no file table ref any more	76	SS_NOFDREF = 0x001 ; no file table ref any more
76	SS_ISCONNECTED = 0x002 ; socket connected to a peer	77	SS_ISCONNECTED = 0x002 ; socket connected to a peer
77	SS_ISCONNECTING = 0x004 ; in process of connecting to peer	78	SS_ISCONNECTING = 0x004 ; in process of connecting to peer
78	SS_ISDISCONNECTING = 0x008 ; in process of disconnecting	79	SS_ISDISCONNECTING = 0x008 ; in process of disconnecting
79	SS_CANTSENDMORE = 0x010 ; can't send more data to peer	80	SS_CANTSENDMORE = 0x010 ; can't send more data to peer
80	SS_CANTRCVMORE = 0x020 ; can't receive more data from peer	81	SS_CANTRCVMORE = 0x020 ; can't receive more data from peer
81	SS_RCVATMARK = 0x040 ; at mark on input	82	SS_RCVATMARK = 0x040 ; at mark on input
82	SS_ISABORTING = 0x080 ; aborting fd references - close()	83	SS_ISABORTING = 0x080 ; aborting fd references - close()
83	SS_RESTARTSYS = 0x100 ; restart blocked system calls	84	SS_RESTARTSYS = 0x100 ; restart blocked system calls
84	SS_ISDISCONNECTED = 0x800 ; socket disconnected from peer	85	SS_ISDISCONNECTED = 0x800 ; socket disconnected from peer
85		86
86	SS_ASYNC = 0x100 ; async i/o notify	87	SS_ASYNC = 0x100 ; async i/o notify
87	SS_ISCONFIRMING = 0x200 ; deciding to accept connection req	88	SS_ISCONFIRMING = 0x200 ; deciding to accept connection req
88	SS_MORETOCOME = 0x400	89	SS_MORETOCOME = 0x400
89		90
90		91
91	SOCKET_MAXDATA = 409632 ; must be 4096(power of 2) where 'power of 2' is at least 8	92	SOCKET_MAXDATA = 409632 ; must be 4096(power of 2) where 'power of 2' is at least 8
92		93
93	; Network driver types	94	; Network driver types
94	NET_TYPE_ETH = 1	95	NET_TYPE_ETH = 1
95	NET_TYPE_SLIP = 2	96	NET_TYPE_SLIP = 2
96		97
97	MAX_backlog = 20 ; maximum backlog for stream sockets	98	MAX_backlog = 20 ; maximum backlog for stream sockets
98		99
99	; Error Codes	100	; Error Codes
100	ENOBUFS = 55	101	ENOBUFS = 55
101	ECONNREFUSED = 61	102	ECONNREFUSED = 61
102	ECONNRESET = 52	103	ECONNRESET = 52
103	ETIMEDOUT = 60	104	ETIMEDOUT = 60
104	ECONNABORTED = 53	105	ECONNABORTED = 53
-		106
-		107	; Api protocol numbers
-		108	API_ETH = 0
-		109	API_IPv4 = 1
-		110	API_ICMP = 2
-		111	API_UDP = 3
105		-
-		112	API_TCP = 4
-		113	API_ARP = 5
106		114	API_PPPOE = 6
107		115
108	struct NET_DEVICE	116	struct NET_DEVICE
109		117
110	type dd ? ; Type field	118	type dd ? ; Type field
111	mtu dd ? ; Maximal Transmission Unit	119	mtu dd ? ; Maximal Transmission Unit
112	name dd ? ; Ptr to 0 terminated string	120	name dd ? ; Ptr to 0 terminated string
113		121
114	unload dd ? ; Ptrs to driver functions	122	unload dd ? ; Ptrs to driver functions
115	reset dd ? ;	123	reset dd ? ;
116	transmit dd ? ;	124	transmit dd ? ;
117		125
118	bytes_tx dq ? ; Statistics, updated by the driver	126	bytes_tx dq ? ; Statistics, updated by the driver
119	bytes_rx dq ? ;	127	bytes_rx dq ? ;
120	packets_tx dd ? ;	128	packets_tx dd ? ;
121	packets_rx dd ? ;	129	packets_rx dd ? ;
122		130
123	; hwacc dd ? ; bitmask stating available hardware accelerations (offload engines)	131	; hwacc dd ? ; bitmask stating available hardware accelerations (offload engines)
124		132
125	ends	133	ends
126		134
127		135
128	; Exactly as it says..	136	; Exactly as it says..
129	macro pseudo_random reg {	137	macro pseudo_random reg {
130	add reg, [esp]	138	add reg, [esp]
131	rol reg, 5	139	rol reg, 5
132	xor reg, [timer_ticks]	140	xor reg, [timer_ticks]
133	add reg, [CPU_FREQ]	141	add reg, [CPU_FREQ]
134	imul reg, 214013	142	imul reg, 214013
135	xor reg, 0xdeadbeef	143	xor reg, 0xdeadbeef
136	rol reg, 9	144	rol reg, 9
137	}	145	}
138		146
139	macro ntohd reg {	147	macro ntohd reg {
140		148
141	rol word reg, 8	149	rol word reg, 8
142	rol dword reg, 16	150	rol dword reg, 16
143	rol word reg , 8	151	rol word reg , 8
144		152
145	}	153	}
146		154
147	macro ntohw reg {	155	macro ntohw reg {
148		156
149	rol word reg, 8	157	rol word reg, 8
150		158
151	}	159	}
152		160
153		161
154	include "queue.inc"	162	include "queue.inc"
155		163
156	include "ethernet.inc"	164	include "ethernet.inc"
157		165
158	;include "PPPoE.inc"	166	;include "PPPoE.inc"
159		167
160	include "ARP.inc"	168	include "ARP.inc"
161	include "IPv4.inc"	169	include "IPv4.inc"
-		170	;include "IPv6.inc"
162		171
163	include "icmp.inc"	172	include "icmp.inc"
164	include "udp.inc"	173	include "udp.inc"
165	include "tcp.inc"	174	include "tcp.inc"
166		175
167	include "socket.inc"	176	include "socket.inc"
168		177
169		178
170		179
171	align 4	180	align 4
172	uglobal	181	uglobal
173		182
174	NET_RUNNING dd ?	183	NET_RUNNING dd ?
175	NET_DEFAULT dd ?	184	NET_DEFAULT dd ?
176	NET_DRV_LIST rd MAX_NET_DEVICES	185	NET_DRV_LIST rd MAX_NET_DEVICES
177		186
178	endg	187	endg
179		188
180		189
181	;-----------------------------------------------------------------	190	;-----------------------------------------------------------------
182	;	191	;
183	; stack_init	192	; stack_init
184	;	193	;
185	; This function calls all network init procedures	194	; This function calls all network init procedures
186	;	195	;
187	; IN: /	196	; IN: /
188	; OUT: /	197	; OUT: /
189	;	198	;
190	;-----------------------------------------------------------------	199	;-----------------------------------------------------------------
191	align 4	200	align 4
192	stack_init:	201	stack_init:
193		202
194	; Init the network drivers list	203	; Init the network drivers list
195	xor eax, eax	204	xor eax, eax
196	mov edi, NET_RUNNING	205	mov edi, NET_RUNNING
197	mov ecx, (MAX_NET_DEVICES + 2)	206	mov ecx, (MAX_NET_DEVICES + 2)
198	rep stosd	207	rep stosd
199		-
200	; SLIP_init	208
201	; PPPOE_init	209	; PPPOE_init
202		210
203	IPv4_init	211	IPv4_init
-		212	; IPv6_init
204	ICMP_init	213	ICMP_init
205		214
206	ARP_init	215	ARP_init
207	UDP_init	216	UDP_init
208	TCP_init	217	TCP_init
209		218
210	SOCKET_init	219	SOCKET_init
211		220
212	mov [net_tmr_count], 0	221	mov [net_tmr_count], 0
213		222
214	ret	223	ret
215		224
216		225
217	;-----------------------------------------------------------------	226	;-----------------------------------------------------------------
218	;	227	;
219	; stack_handler	228	; stack_handler
220	;	229	;
221	; This function is called in kernel loop	230	; This function is called in kernel loop
222	;	231	;
223	; IN: /	232	; IN: /
224	; OUT: /	233	; OUT: /
225	;	234	;
226	;-----------------------------------------------------------------	235	;-----------------------------------------------------------------
227	align 4	236	align 4
228	stack_handler:	237	stack_handler:
229		238
230	cmp [NET_RUNNING], 0	239	cmp [NET_RUNNING], 0
231	je .exit	240	je .exit
232		241
233	; Test for 10ms tick	242	; Test for 10ms tick
234	mov eax, [timer_ticks]	243	mov eax, [timer_ticks]
235	cmp eax, [net_10ms]	244	cmp eax, [net_10ms]
236	je .exit	245	je .exit
237	mov [net_10ms], eax	246	mov [net_10ms], eax
238		247
239	test [net_10ms], 0x0f ; 160ms	248	test [net_10ms], 0x0f ; 160ms
240	jnz .exit	249	jnz .exit
241		250
242	TCP_timer_160ms	251	TCP_timer_160ms
243		252
244	test [net_10ms], 0x3f ; 640ms	253	test [net_10ms], 0x3f ; 640ms
245	jnz .exit	254	jnz .exit
246		255
247	TCP_timer_640ms	256	TCP_timer_640ms
248	ARP_decrease_entry_ttls	257	ARP_decrease_entry_ttls
249	IPv4_decrease_fragment_ttls	258	IPv4_decrease_fragment_ttls
250		259
251	.exit:	260	.exit:
252	ret	261	ret
253		262
254		263
255		264
256	;-----------------------------------------------------------------	265	;-----------------------------------------------------------------
257	;	266	;
258	; NET_add_device:	267	; NET_add_device:
259	;	268	;
260	; This function is called by the network drivers,	269	; This function is called by the network drivers,
261	; to register each running NIC to the kernel	270	; to register each running NIC to the kernel
262	;	271	;
263	; IN: Pointer to device structure in ebx	272	; IN: Pointer to device structure in ebx
264	; OUT: Device num in eax, -1 on error	273	; OUT: Device num in eax, -1 on error
265	;	274	;
266	;-----------------------------------------------------------------	275	;-----------------------------------------------------------------
267	align 4	276	align 4
268	NET_add_device:	277	NET_add_device:
269		278
270	DEBUGF 1,"NET_Add_Device: %x\n", ebx ;;; TODO: use mutex to lock net device list	279	DEBUGF 1,"NET_Add_Device: %x\n", ebx ;;; TODO: use mutex to lock net device list
271		280
272	mov eax, [NET_RUNNING]	281	mov eax, [NET_RUNNING]
273	cmp eax, MAX_NET_DEVICES	282	cmp eax, MAX_NET_DEVICES
274	jae .error	283	jae .error
275		284
276	;----------------------------------	285	;----------------------------------
277	; Check if device is already listed	286	; Check if device is already listed
278	mov eax, ebx	287	mov eax, ebx
279	mov ecx, MAX_NET_DEVICES ; We need to check whole list because a device may be removed without re-organizing list	288	mov ecx, MAX_NET_DEVICES ; We need to check whole list because a device may be removed without re-organizing list
280	mov edi, NET_DRV_LIST	289	mov edi, NET_DRV_LIST
281		290
282	repne scasd ; See if device is already in the list	291	repne scasd ; See if device is already in the list
283	jz .error	292	jz .error
284		293
285	;----------------------------	294	;----------------------------
286	; Find empty slot in the list	295	; Find empty slot in the list
287	xor eax, eax	296	xor eax, eax
288	mov ecx, MAX_NET_DEVICES	297	mov ecx, MAX_NET_DEVICES
289	mov edi, NET_DRV_LIST	298	mov edi, NET_DRV_LIST
290		299
291	repne scasd	300	repne scasd
292	jnz .error	301	jnz .error
293		302
294	sub edi, 4	303	sub edi, 4
295		304
296	;-----------------------------	305	;-----------------------------
297	; Add device to the found slot	306	; Add device to the found slot
298	mov [edi], ebx ; add device to list	307	mov [edi], ebx ; add device to list
299		308
300	mov eax, edi ; Calculate device number in eax	309	mov eax, edi ; Calculate device number in eax
301	sub eax, NET_DRV_LIST	310	sub eax, NET_DRV_LIST
302	shr eax, 2	311	shr eax, 2
303		312
304	inc [NET_RUNNING] ; Indicate that one more network device is up and running	313	inc [NET_RUNNING] ; Indicate that one more network device is up and running
305		314
306	cmp eax, 1 ; If it's the first network device, try to set it as default	315	cmp eax, 1 ; If it's the first network device, try to set it as default
307	jne @f	316	jne @f
308	push eax	317	push eax
309	call NET_set_default	318	call NET_set_default
310	pop eax	319	pop eax
311	@@:	320	@@:
312		321
313	DEBUGF 1,"Device number: %u\n", eax	322	DEBUGF 1,"Device number: %u\n", eax
314	ret	323	ret
315		324
316	.error:	325	.error:
317	or eax, -1	326	or eax, -1
318	DEBUGF 2,"Adding network device failed\n"	327	DEBUGF 2,"Adding network device failed\n"
319	ret	328	ret
320		329
321		330
322		331
323	;-----------------------------------------------------------------	332	;-----------------------------------------------------------------
324	;	333	;
325	; NET_set_default	334	; NET_set_default
326	;	335	;
327	; API to set the default interface	336	; API to set the default interface
328	;	337	;
329	; IN: Device num in eax	338	; IN: Device num in eax
330	; OUT: Device num in eax, -1 on error	339	; OUT: Device num in eax, -1 on error
331	;	340	;
332	;-----------------------------------------------------------------	341	;-----------------------------------------------------------------
333	align 4	342	align 4
334	NET_set_default:	343	NET_set_default:
335		344
336	DEBUGF 1,"NET_set_default %x\n", eax	345	DEBUGF 1,"NET_set_default %x\n", eax
337		346
338	cmp eax, MAX_NET_DEVICES	347	cmp eax, MAX_NET_DEVICES
339	jae .error	348	jae .error
340		349
341	cmp [NET_DRV_LIST+eax*4], 0	350	cmp [NET_DRV_LIST+eax*4], 0
342	je .error	351	je .error
343		352
344	mov [NET_DEFAULT], eax	353	mov [NET_DEFAULT], eax
345		354
346	DEBUGF 1,"Device number %u is now default!\n", eax	355	DEBUGF 1,"Device number %u is now default!\n", eax
347	ret	356	ret
348		357
349	.error:	358	.error:
350	or eax, -1	359	or eax, -1
351	DEBUGF 2,"Setting default network device failed\n"	360	DEBUGF 2,"Setting default network device failed\n"
352	ret	361	ret
353		362
354		363
355	;-----------------------------------------------------------------	364	;-----------------------------------------------------------------
356	;	365	;
357	; NET_Remove_Device:	366	; NET_Remove_Device:
358	;	367	;
359	; This function is called by etwork drivers,	368	; This function is called by etwork drivers,
360	; to unregister network devices from the kernel	369	; to unregister network devices from the kernel
361	; d	370	; d
362	; IN: Pointer to device structure in ebx	371	; IN: Pointer to device structure in ebx
363	; OUT: eax: -1 on error	372	; OUT: eax: -1 on error
364	;	373	;
365	;-----------------------------------------------------------------	374	;-----------------------------------------------------------------
366	align 4	375	align 4
367	NET_remove_device:	376	NET_remove_device:
368		377
369	cmp [NET_RUNNING], 0	378	cmp [NET_RUNNING], 0
370	je .error	379	je .error
371		380
372	cmp [NET_DRV_LIST], ebx	381	cmp [NET_DRV_LIST], ebx
373	jne @f	382	jne @f
374	mov [NET_DRV_LIST], 0	383	mov [NET_DRV_LIST], 0
375	cmp [NET_RUNNING], 1	384	cmp [NET_RUNNING], 1
376	je @f	385	je @f
377	; there are still active devices, find one and make it default	386	; there are still active devices, find one and make it default
378	xor eax, eax	387	xor eax, eax
379	mov ecx, MAX_NET_DEVICES	388	mov ecx, MAX_NET_DEVICES
380	mov edi, NET_DRV_LIST	389	mov edi, NET_DRV_LIST
381	repe scasd	390	repe scasd
382	je @f	391	je @f
383	shr edi, 2	392	shr edi, 2
384	dec edi	393	dec edi
385	mov [NET_DEFAULT], edi	394	mov [NET_DEFAULT], edi
386	@@:	395	@@:
387		396
388	;----------------------------	397	;----------------------------
389	; Find the driver in the list	398	; Find the driver in the list
390		399
391	mov eax, ebx	400	mov eax, ebx
392	mov ecx, MAX_NET_DEVICES	401	mov ecx, MAX_NET_DEVICES
393	mov edi, NET_DRV_LIST+4	402	mov edi, NET_DRV_LIST+4
394		403
395	repne scasd	404	repne scasd
396	jnz .error	405	jnz .error
397		406
398	;------------------------	407	;------------------------
399	; Remove it from the list	408	; Remove it from the list
400		409
401	xor eax, eax	410	xor eax, eax
402	mov dword [edi-4], eax	411	mov dword [edi-4], eax
403		412
404	dec [NET_RUNNING]	413	dec [NET_RUNNING]
405	ret	414	ret
406		415
407	.error:	416	.error:
408	or eax, -1	417	or eax, -1
409	ret	418	ret
410		419
411		420
412		421
413	;-----------------------------------------------------------------	422	;-----------------------------------------------------------------
414	;	423	;
415	; NET_ptr_to_num	424	; NET_ptr_to_num
416	;	425	;
417	; IN: ebx = ptr to device struct	426	; IN: ebx = ptr to device struct
418	; OUT: edi = -1 on error, device number otherwise	427	; OUT: edi = -1 on error, device number otherwise
419	;	428	;
420	;-----------------------------------------------------------------	429	;-----------------------------------------------------------------
421	align 4	430	align 4
422	NET_ptr_to_num:	431	NET_ptr_to_num:
423	push ecx	432	push ecx
424		433
425	mov ecx, MAX_NET_DEVICES	434	mov ecx, MAX_NET_DEVICES
426	mov edi, NET_DRV_LIST	435	mov edi, NET_DRV_LIST
427		436
428	.loop:	437	.loop:
429	cmp ebx, [edi]	438	cmp ebx, [edi]
430	jz .found	439	jz .found
431	add edi, 4	440	add edi, 4
432	dec ecx	441	dec ecx
433	jnz .loop	442	jnz .loop
434		443
435	; repnz scasd could work too if eax is used instead of ebx!	444	; repnz scasd could work too if eax is used instead of ebx!
436		445
437	or edi, -1	446	or edi, -1
438		447
439	pop ecx	448	pop ecx
440	ret	449	ret
441		450
442	.found:	451	.found:
443	sub edi, NET_DRV_LIST	452	sub edi, NET_DRV_LIST
444	shr edi, 2	453	shr edi, 2
445		454
446	pop ecx	455	pop ecx
447	ret	456	ret
448		457
449	;-----------------------------------------------------------------	458	;-----------------------------------------------------------------
450	;	459	;
451	; checksum_1	460	; checksum_1
452	;	461	;
453	; This is the first of two functions needed to calculate a checksum.	462	; This is the first of two functions needed to calculate a checksum.
454	;	463	;
455	; IN: edx = start offset for semi-checksum	464	; IN: edx = start offset for semi-checksum
456	; esi = pointer to data	465	; esi = pointer to data
457	; ecx = data size	466	; ecx = data size
458	; OUT: edx = semi-checksum	467	; OUT: edx = semi-checksum
459	;	468	;
460	;	469	;
461	; Code was optimized by diamond	470	; Code was optimized by diamond
462	;	471	;
463	;-----------------------------------------------------------------	472	;-----------------------------------------------------------------
464	align 4	473	align 4
465	checksum_1:	474	checksum_1:
466		475
467	shr ecx, 1	476	shr ecx, 1
468	pushf	477	pushf
469	jz .no_2	478	jz .no_2
470		479
471	shr ecx, 1	480	shr ecx, 1
472	pushf	481	pushf
473	jz .no_4	482	jz .no_4
474		483
475	shr ecx, 1	484	shr ecx, 1
476	pushf	485	pushf
477	jz .no_8	486	jz .no_8
478		487
479	.loop:	488	.loop:
480	add dl, [esi+1]	489	add dl, [esi+1]
481	adc dh, [esi+0]	490	adc dh, [esi+0]
482		491
483	adc dl, [esi+3]	492	adc dl, [esi+3]
484	adc dh, [esi+2]	493	adc dh, [esi+2]
485		494
486	adc dl, [esi+5]	495	adc dl, [esi+5]
487	adc dh, [esi+4]	496	adc dh, [esi+4]
488		497
489	adc dl, [esi+7]	498	adc dl, [esi+7]

Subversion Repositories Kolibri OS

(root)/kernel/branches/net/network/stack.inc – Rev 2621 → 2731