2009-09-22 20:42:54 +04:00
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
;; ;;
|
2011-01-25 01:01:54 +03:00
|
|
|
;; Copyright (C) KolibriOS team 2004-2011. All rights reserved. ;;
|
2009-09-22 20:42:54 +04:00
|
|
|
;; Distributed under terms of the GNU General Public License ;;
|
|
|
|
;; ;;
|
|
|
|
;; STACK.INC ;;
|
|
|
|
;; ;;
|
2011-01-25 01:01:54 +03:00
|
|
|
;; TCP/IP stack for KolibriOS ;;
|
2009-09-22 20:42:54 +04:00
|
|
|
;; ;;
|
|
|
|
;; Written by hidnplayr@kolibrios.org ;;
|
|
|
|
;; ;;
|
2011-01-25 01:01:54 +03:00
|
|
|
;; 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 ;;
|
2009-09-22 20:42:54 +04:00
|
|
|
;; ;;
|
|
|
|
;; GNU GENERAL PUBLIC LICENSE ;;
|
|
|
|
;; Version 2, June 1991 ;;
|
|
|
|
;; ;;
|
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2009-10-12 18:39:59 +04:00
|
|
|
$Revision$
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
__DEBUG_LEVEL_OLD__ equ __DEBUG_LEVEL__ ; use seperate debug level for network part of kernel
|
|
|
|
__DEBUG_LEVEL__ equ 1
|
2010-05-29 00:47:32 +04:00
|
|
|
|
2009-09-17 15:55:38 +04:00
|
|
|
uglobal
|
2011-01-25 01:01:54 +03:00
|
|
|
net_10ms dd ?
|
|
|
|
net_tmr_count dw ?
|
2009-09-17 15:55:38 +04:00
|
|
|
endg
|
|
|
|
|
2009-11-09 20:59:50 +03:00
|
|
|
MAX_NET_DEVICES equ 16
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
MIN_EPHEMERAL_PORT equ 49152
|
|
|
|
MAX_EPHEMERAL_PORT equ 61000
|
2009-09-28 23:48:32 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
; Ethernet protocol numbers
|
2010-07-15 22:54:19 +04:00
|
|
|
ETHER_ARP equ 0x0608
|
2011-01-08 17:59:21 +03:00
|
|
|
ETHER_IPv4 equ 0x0008
|
2010-07-15 22:54:19 +04:00
|
|
|
ETHER_PPP_DISCOVERY equ 0x6388
|
|
|
|
ETHER_PPP_SESSION equ 0x6488
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
;Protocol family
|
2011-01-29 02:07:25 +03:00
|
|
|
AF_UNSPEC equ 0
|
|
|
|
AF_UNIX equ 1
|
|
|
|
AF_INET4 equ 2
|
|
|
|
AF_INET6 equ 10
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
; Internet protocol numbers
|
2011-01-29 02:07:25 +03:00
|
|
|
IP_PROTO_IP equ 0
|
|
|
|
IP_PROTO_ICMP equ 1
|
|
|
|
IP_PROTO_TCP equ 6
|
|
|
|
IP_PROTO_UDP equ 17
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2009-10-08 00:36:58 +04:00
|
|
|
; Socket types
|
2011-01-29 02:07:25 +03:00
|
|
|
SOCK_STREAM equ 1
|
|
|
|
SOCK_DGRAM equ 2
|
|
|
|
SOCK_RAW equ 3
|
2009-11-08 21:00:01 +03:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
; Socket options
|
2011-02-24 17:16:02 +03:00
|
|
|
SO_ACCEPTCON equ 1 shl 0
|
|
|
|
SO_BROADCAST equ 1 shl 1
|
|
|
|
SO_DEBUG equ 1 shl 2
|
|
|
|
SO_DONTROUTE equ 1 shl 3
|
|
|
|
SO_KEEPALIVE equ 1 shl 4
|
|
|
|
SO_OOBINLINE equ 1 shl 5
|
|
|
|
SO_REUSEADDR equ 1 shl 6
|
|
|
|
SO_REUSEPORT equ 1 shl 7
|
|
|
|
SO_USELOOPBACK equ 1 shl 8
|
|
|
|
|
2009-11-08 21:00:01 +03:00
|
|
|
|
2011-01-28 19:49:08 +03:00
|
|
|
; Socket States
|
2011-01-29 02:07:25 +03:00
|
|
|
SS_NOFDREF equ 0x001 ; no file table ref any more
|
|
|
|
SS_ISCONNECTED equ 0x002 ; socket connected to a peer
|
|
|
|
SS_ISCONNECTING equ 0x004 ; in process of connecting to peer
|
|
|
|
SS_ISDISCONNECTING equ 0x008 ; in process of disconnecting
|
|
|
|
SS_CANTSENDMORE equ 0x010 ; can't send more data to peer
|
|
|
|
SS_CANTRCVMORE equ 0x020 ; can't receive more data from peer
|
|
|
|
SS_RCVATMARK equ 0x040 ; at mark on input
|
|
|
|
SS_ISABORTING equ 0x080 ; aborting fd references - close()
|
|
|
|
SS_RESTARTSYS equ 0x100 ; restart blocked system calls
|
|
|
|
SS_ISDISCONNECTED equ 0x800 ; socket disconnected from peer
|
2011-01-28 19:49:08 +03:00
|
|
|
|
2011-01-29 02:07:25 +03:00
|
|
|
SS_ASYNC equ 0x100 ; async i/o notify
|
|
|
|
SS_ISCONFIRMING equ 0x200 ; deciding to accept connection req
|
|
|
|
SS_MORETOCOME equ 0x400
|
2011-01-28 19:49:08 +03:00
|
|
|
|
|
|
|
|
2011-01-29 02:07:25 +03:00
|
|
|
SOCKET_MAXDATA equ 4096*32 ; must be 4096*(power of 2) where 'power of 2' is at least 8
|
2009-11-08 21:00:01 +03:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
; Network driver types
|
2011-01-29 02:07:25 +03:00
|
|
|
NET_TYPE_ETH equ 1
|
|
|
|
NET_TYPE_SLIP equ 2
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2011-01-29 02:07:25 +03:00
|
|
|
MAX_backlog equ 20 ; maximum backlog for stream sockets
|
2010-07-27 22:53:38 +04:00
|
|
|
|
2010-12-10 21:37:02 +03:00
|
|
|
; Error Codes
|
2011-01-29 02:07:25 +03:00
|
|
|
ENOBUFS equ 55
|
|
|
|
ECONNREFUSED equ 61
|
|
|
|
ECONNRESET equ 52
|
|
|
|
ETIMEDOUT equ 60
|
|
|
|
ECONNABORTED equ 53
|
2010-12-10 21:37:02 +03:00
|
|
|
|
|
|
|
|
2009-12-13 17:55:42 +03:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
virtual at 0
|
|
|
|
|
|
|
|
NET_DEVICE:
|
2010-07-15 22:54:19 +04:00
|
|
|
|
|
|
|
.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 ? ;
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
; .hwacc dd ? ; bitmask stating available hardware accelerations (offload engines)
|
2010-07-27 22:53:38 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
.end:
|
2009-12-13 17:55:42 +03:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
end virtual
|
|
|
|
|
|
|
|
|
|
|
|
; Exactly as it says..
|
|
|
|
macro pseudo_random reg {
|
2009-12-13 17:55:42 +03:00
|
|
|
add reg, [esp]
|
|
|
|
rol reg, 5
|
|
|
|
xor reg, [timer_ticks]
|
2010-07-27 22:53:38 +04:00
|
|
|
add reg, [CPU_FREQ]
|
2009-12-13 17:55:42 +03:00
|
|
|
imul reg, 214013
|
|
|
|
xor reg, 0xdeadbeef
|
|
|
|
rol reg, 9
|
2010-07-12 03:13:12 +04:00
|
|
|
}
|
2009-12-13 17:55:42 +03:00
|
|
|
|
2010-07-31 23:44:17 +04:00
|
|
|
macro ntohd reg {
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
rol word reg, 8
|
|
|
|
rol dword reg, 16
|
2010-07-27 22:53:38 +04:00
|
|
|
rol word reg , 8
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2010-07-31 23:44:17 +04:00
|
|
|
macro ntohw reg {
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
rol word reg, 8
|
2009-12-13 17:55:42 +03:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2009-09-17 15:55:38 +04:00
|
|
|
include "queue.inc"
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
include "ethernet.inc"
|
2010-12-18 02:04:18 +03:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
;include "slip.inc"
|
2010-12-18 02:04:18 +03:00
|
|
|
;include "pppoe.inc"
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2009-09-29 00:16:23 +04:00
|
|
|
include "ARP.inc"
|
|
|
|
include "IPv4.inc"
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2009-09-28 23:48:32 +04:00
|
|
|
include "icmp.inc"
|
2010-07-12 03:13:12 +04:00
|
|
|
include "udp.inc"
|
|
|
|
include "tcp.inc"
|
|
|
|
|
|
|
|
include "socket.inc"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
align 4
|
|
|
|
uglobal
|
|
|
|
|
|
|
|
NET_RUNNING dd ?
|
2011-09-22 08:08:56 +04:00
|
|
|
NET_DRV_LIST rd (MAX_NET_DEVICES + 1) ; device 0 is a link to the default device
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
endg
|
|
|
|
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2009-11-09 16:59:46 +03:00
|
|
|
;-----------------------------------------------------------------
|
2009-09-17 15:55:38 +04:00
|
|
|
;
|
|
|
|
; stack_init
|
|
|
|
;
|
|
|
|
; This function calls all network init procedures
|
|
|
|
;
|
|
|
|
; IN: /
|
|
|
|
; OUT: /
|
|
|
|
;
|
2009-11-09 16:59:46 +03:00
|
|
|
;-----------------------------------------------------------------
|
2009-09-17 15:55:38 +04:00
|
|
|
align 4
|
|
|
|
stack_init:
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
; Init the network drivers list
|
|
|
|
xor eax, eax
|
|
|
|
mov edi, NET_RUNNING
|
2011-09-22 08:08:56 +04:00
|
|
|
mov ecx, (MAX_NET_DEVICES + 2)
|
2010-07-12 03:13:12 +04:00
|
|
|
rep stosd
|
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
; SLIP_init
|
|
|
|
; PPPOE_init
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
IPv4_init
|
|
|
|
ICMP_init
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
ARP_init
|
|
|
|
UDP_init
|
|
|
|
TCP_init
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
SOCKET_init
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
mov [net_tmr_count], 0
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
ret
|
|
|
|
|
|
|
|
|
2009-11-09 16:59:46 +03:00
|
|
|
;-----------------------------------------------------------------
|
2009-09-17 15:55:38 +04:00
|
|
|
;
|
|
|
|
; stack_handler
|
|
|
|
;
|
2010-07-12 03:13:12 +04:00
|
|
|
; This function is called in kernel loop
|
2009-09-17 15:55:38 +04:00
|
|
|
;
|
|
|
|
; IN: /
|
|
|
|
; OUT: /
|
|
|
|
;
|
2009-11-09 16:59:46 +03:00
|
|
|
;-----------------------------------------------------------------
|
2009-09-17 15:55:38 +04:00
|
|
|
align 4
|
|
|
|
stack_handler:
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
cmp [NET_RUNNING], 0
|
2009-11-09 16:59:46 +03:00
|
|
|
je .exit
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2009-12-13 17:55:42 +03:00
|
|
|
; Test for 10ms tick
|
2009-11-09 16:59:46 +03:00
|
|
|
mov eax, [timer_ticks]
|
2010-07-12 03:13:12 +04:00
|
|
|
cmp eax, [net_10ms]
|
2009-11-09 16:59:46 +03:00
|
|
|
je .exit
|
2010-07-12 03:13:12 +04:00
|
|
|
mov [net_10ms], eax
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-15 22:54:19 +04:00
|
|
|
test [net_10ms], 0x0f ; 160ms
|
|
|
|
jnz .exit
|
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
TCP_timer_160ms
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-15 22:54:19 +04:00
|
|
|
test [net_10ms], 0x3f ; 640ms
|
|
|
|
jnz .exit
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
TCP_timer_640ms
|
|
|
|
ARP_decrease_entry_ttls
|
|
|
|
IPv4_decrease_fragment_ttls
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
.exit:
|
2009-11-09 16:59:46 +03:00
|
|
|
ret
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
2011-09-22 08:08:56 +04:00
|
|
|
; NET_add_device:
|
2010-07-12 03:13:12 +04:00
|
|
|
;
|
|
|
|
; 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:
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
DEBUGF 1,"NET_Add_Device: %x\n", ebx ;;; TODO: use mutex to lock net device list
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
mov eax, [NET_RUNNING]
|
|
|
|
cmp eax, MAX_NET_DEVICES
|
2011-11-08 00:25:47 +04:00
|
|
|
jae .error
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
;----------------------------------
|
|
|
|
; Check if device is already listed
|
|
|
|
mov eax, ebx
|
2011-09-22 08:08:56 +04:00
|
|
|
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+4
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
repne scasd ; See if device is already in the list
|
2010-07-12 03:13:12 +04:00
|
|
|
jz .error
|
|
|
|
|
|
|
|
;----------------------------
|
|
|
|
; Find empty slot in the list
|
|
|
|
xor eax, eax
|
|
|
|
mov ecx, MAX_NET_DEVICES
|
2011-09-22 08:08:56 +04:00
|
|
|
mov edi, NET_DRV_LIST+4
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
repne scasd
|
|
|
|
jnz .error
|
|
|
|
|
|
|
|
sub edi, 4
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
;-----------------------------
|
|
|
|
; Add device to the found slot
|
|
|
|
mov [edi], ebx ; add device to list
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
mov eax, edi ; Calculate device number in eax
|
|
|
|
sub eax, NET_DRV_LIST
|
|
|
|
shr eax, 2
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
inc [NET_RUNNING] ; Indicate that one more network device is up and running
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
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
|
|
|
|
@@:
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
DEBUGF 1,"Device number: %u\n", eax
|
|
|
|
ret
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
.error:
|
|
|
|
or eax, -1
|
|
|
|
DEBUGF 2,"Adding network device failed\n"
|
|
|
|
ret
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
|
|
|
; 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:
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
DEBUGF 1,"NET_set_default %x\n", eax
|
|
|
|
|
|
|
|
cmp eax, MAX_NET_DEVICES
|
2011-11-08 00:25:47 +04:00
|
|
|
jae .error
|
2011-09-22 08:08:56 +04:00
|
|
|
|
|
|
|
cmp [NET_DRV_LIST+eax*4], 0
|
|
|
|
je .error
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
push [NET_DRV_LIST+eax*4]
|
|
|
|
pop [NET_DRV_LIST]
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
DEBUGF 1,"Device number %u is now default!\n", eax
|
2010-07-12 03:13:12 +04:00
|
|
|
ret
|
|
|
|
|
|
|
|
.error:
|
|
|
|
or eax, -1
|
2011-09-22 08:08:56 +04:00
|
|
|
DEBUGF 2,"Setting default network device failed\n"
|
2010-07-12 03:13:12 +04:00
|
|
|
ret
|
|
|
|
|
|
|
|
|
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
|
|
|
; NET_Remove_Device:
|
|
|
|
;
|
|
|
|
; This function is called by etwork drivers,
|
|
|
|
; to unregister network devices from the kernel
|
2011-09-22 08:08:56 +04:00
|
|
|
; d
|
2010-07-12 03:13:12 +04:00
|
|
|
; IN: Pointer to device structure in ebx
|
|
|
|
; OUT: eax: -1 on error
|
|
|
|
;
|
|
|
|
;-----------------------------------------------------------------
|
|
|
|
align 4
|
|
|
|
NET_remove_device:
|
|
|
|
|
|
|
|
cmp [NET_RUNNING], 0
|
|
|
|
je .error
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
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+4
|
|
|
|
repe scasd
|
|
|
|
jz @f
|
|
|
|
push dword [edi-4]
|
|
|
|
pop [NET_DRV_LIST]
|
|
|
|
@@:
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
;----------------------------
|
|
|
|
; Find the driver in the list
|
|
|
|
|
|
|
|
mov eax, ebx
|
|
|
|
mov ecx, MAX_NET_DEVICES
|
2011-09-22 08:08:56 +04:00
|
|
|
mov edi, NET_DRV_LIST+4
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
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
|
2011-09-22 08:08:56 +04:00
|
|
|
mov edi, NET_DRV_LIST+4
|
2010-07-12 03:13:12 +04:00
|
|
|
|
|
|
|
.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
|
|
|
|
|
2009-11-06 23:45:08 +03:00
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
|
|
|
; checksum_1
|
|
|
|
;
|
2010-06-07 13:37:13 +04:00
|
|
|
; This is the first of two functions needed to calculate a checksum.
|
2009-11-06 23:45:08 +03:00
|
|
|
;
|
2010-05-29 00:47:32 +04:00
|
|
|
; IN: edx = start offset for semi-checksum
|
2009-11-06 23:45:08 +03:00
|
|
|
; esi = pointer to data
|
|
|
|
; ecx = data size
|
|
|
|
; OUT: edx = semi-checksum
|
|
|
|
;
|
2010-05-29 00:47:32 +04:00
|
|
|
;
|
|
|
|
; Code was optimized by diamond
|
|
|
|
;
|
2009-11-06 23:45:08 +03:00
|
|
|
;-----------------------------------------------------------------
|
|
|
|
align 4
|
|
|
|
checksum_1:
|
|
|
|
|
|
|
|
shr ecx, 1
|
|
|
|
pushf
|
2010-05-29 00:47:32 +04:00
|
|
|
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
|
2009-11-06 23:45:08 +03:00
|
|
|
|
2010-05-29 00:47:32 +04:00
|
|
|
.no_4:
|
|
|
|
popf
|
|
|
|
jnc .no_2
|
|
|
|
|
|
|
|
add dl, [esi+1]
|
|
|
|
adc dh, [esi+0]
|
|
|
|
|
|
|
|
adc edx, 0
|
2010-06-07 16:57:52 +04:00
|
|
|
inc esi
|
|
|
|
inc esi
|
2010-05-29 00:47:32 +04:00
|
|
|
|
|
|
|
.no_2:
|
2009-11-06 23:45:08 +03:00
|
|
|
popf
|
|
|
|
jnc .end
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-05-29 00:47:32 +04:00
|
|
|
add dh, [esi+0]
|
|
|
|
adc edx, 0
|
|
|
|
.end:
|
|
|
|
ret
|
|
|
|
|
2009-11-06 23:45:08 +03:00
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
|
|
|
; 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
|
2010-06-07 13:37:13 +04:00
|
|
|
|
|
|
|
mov ecx, edx
|
|
|
|
shr ecx, 16
|
2010-07-27 22:53:38 +04:00
|
|
|
add dx, cx
|
|
|
|
test dx, dx ; it seems that ZF is not set when CF is set :(
|
2009-11-06 23:45:08 +03:00
|
|
|
not dx
|
|
|
|
jnz .not_zero
|
|
|
|
dec dx
|
|
|
|
.not_zero:
|
|
|
|
xchg dl, dh
|
|
|
|
|
2010-07-27 22:53:38 +04:00
|
|
|
DEBUGF 1,"Checksum: %x\n", dx
|
2009-11-06 23:45:08 +03:00
|
|
|
|
|
|
|
ret
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;----------------------------------------------------------------
|
|
|
|
;
|
2009-09-22 20:42:54 +04:00
|
|
|
; System function to work with network devices (73)
|
2009-09-17 15:55:38 +04:00
|
|
|
;
|
|
|
|
;----------------------------------------------------------------
|
|
|
|
align 4
|
|
|
|
sys_network:
|
|
|
|
|
2009-10-06 00:47:27 +04:00
|
|
|
cmp ebx, -1
|
|
|
|
jne @f
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
mov eax, [NET_RUNNING]
|
2009-10-06 00:47:27 +04:00
|
|
|
jmp .return
|
|
|
|
|
|
|
|
@@:
|
2011-09-22 08:08:56 +04:00
|
|
|
cmp bh, MAX_NET_DEVICES ; Check if device number exists
|
2011-11-08 00:25:47 +04:00
|
|
|
jae .doesnt_exist
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
mov esi, ebx
|
|
|
|
and esi, 0x0000ff00
|
|
|
|
shr esi, 6
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
cmp dword [esi + NET_DRV_LIST], 0 ; check if driver is running
|
2009-09-17 15:55:38 +04:00
|
|
|
je .doesnt_exist
|
|
|
|
|
2011-09-22 08:08:56 +04:00
|
|
|
|
|
|
|
|
|
|
|
test bl, bl ; 0 = Get device type (ethernet/token ring/...)
|
2009-09-17 15:55:38 +04:00
|
|
|
jnz @f
|
2010-07-12 03:13:12 +04:00
|
|
|
|
2009-10-02 23:49:42 +04:00
|
|
|
xor eax, eax
|
|
|
|
jmp .return
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
|
|
|
|
@@:
|
|
|
|
dec bl ; 1 = Get device name
|
|
|
|
jnz @f
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
mov esi, [esi + NET_DRV_LIST]
|
2010-07-15 22:54:19 +04:00
|
|
|
mov esi, [esi + NET_DEVICE.name]
|
2009-09-17 15:55:38 +04:00
|
|
|
mov edi, ecx
|
|
|
|
|
|
|
|
mov ecx, 64 ; max length
|
|
|
|
repnz movsb
|
|
|
|
|
2009-10-02 23:49:42 +04:00
|
|
|
xor eax, eax
|
|
|
|
jmp .return
|
|
|
|
|
|
|
|
@@:
|
|
|
|
|
|
|
|
dec bl ; 2 = Reset the device
|
|
|
|
jnz @f
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
mov esi, [esi + NET_DRV_LIST]
|
2010-07-15 22:54:19 +04:00
|
|
|
call [esi + NET_DEVICE.reset]
|
2009-10-02 23:49:42 +04:00
|
|
|
jmp .return
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
@@:
|
2009-10-02 23:49:42 +04:00
|
|
|
|
|
|
|
dec bl ; 3 = Stop driver for this device
|
|
|
|
jnz @f
|
|
|
|
|
2010-07-12 03:13:12 +04:00
|
|
|
mov esi, [esi + NET_DRV_LIST]
|
2010-07-15 22:54:19 +04:00
|
|
|
call [esi + NET_DEVICE.unload]
|
2009-10-02 23:49:42 +04:00
|
|
|
jmp .return
|
|
|
|
|
|
|
|
@@:
|
2009-11-06 23:45:08 +03:00
|
|
|
dec bl ; 4 = Get driver pointer
|
|
|
|
jnz @f
|
|
|
|
|
|
|
|
; ..;
|
2011-09-22 08:08:56 +04:00
|
|
|
xor eax, eax
|
|
|
|
jmp .return
|
|
|
|
|
|
|
|
|
|
|
|
@@:
|
|
|
|
dec bl ; 5 = Get driver name
|
|
|
|
jnz @f
|
|
|
|
|
|
|
|
; ..;
|
|
|
|
xor eax, eax
|
|
|
|
jmp .return
|
|
|
|
|
|
|
|
|
|
|
|
@@:
|
|
|
|
dec bl ; 6 = Set default device
|
|
|
|
jnz @f
|
|
|
|
|
|
|
|
mov eax, esi
|
|
|
|
call NET_set_default
|
|
|
|
jmp .return
|
2009-11-06 23:45:08 +03:00
|
|
|
|
|
|
|
|
|
|
|
@@:
|
2009-10-02 23:49:42 +04:00
|
|
|
|
2009-09-17 15:55:38 +04:00
|
|
|
.doesnt_exist:
|
|
|
|
DEBUGF 1,"sys_network: invalid device/function specified!\n"
|
|
|
|
mov eax, -1
|
|
|
|
|
2009-10-02 23:49:42 +04:00
|
|
|
.return:
|
|
|
|
mov [esp+28+4], eax
|
2009-09-17 15:55:38 +04:00
|
|
|
ret
|
|
|
|
|
|
|
|
|
|
|
|
;----------------------------------------------------------------
|
|
|
|
;
|
2010-07-12 03:13:12 +04:00
|
|
|
; System function to work with protocols (75)
|
2009-09-17 15:55:38 +04:00
|
|
|
;
|
|
|
|
;----------------------------------------------------------------
|
|
|
|
align 4
|
|
|
|
sys_protocols:
|
|
|
|
cmp bh, MAX_NET_DEVICES ; Check if device number exists
|
2011-11-08 00:25:47 +04:00
|
|
|
jae .doesnt_exist
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
mov esi, ebx
|
|
|
|
and esi, 0x0000ff00
|
2010-07-12 03:13:12 +04:00
|
|
|
shr esi, 6 ; now we have the device num * 4 in esi
|
|
|
|
cmp dword [esi + NET_DRV_LIST], 0 ; check if driver is running
|
2009-09-17 15:55:38 +04:00
|
|
|
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 , IP_PROTO_IP
|
|
|
|
je IPv4_API
|
|
|
|
|
|
|
|
cmp ax , IP_PROTO_ICMP
|
|
|
|
je ICMP_API
|
|
|
|
|
|
|
|
cmp ax , IP_PROTO_UDP
|
|
|
|
je UDP_API
|
|
|
|
|
2009-09-22 20:42:54 +04:00
|
|
|
cmp ax , IP_PROTO_TCP
|
2009-11-08 21:00:01 +03:00
|
|
|
je TCP_API
|
2009-09-17 15:55:38 +04:00
|
|
|
|
2009-09-22 20:42:54 +04:00
|
|
|
cmp ax , ETHER_ARP
|
2009-09-17 15:55:38 +04:00
|
|
|
je ARP_API
|
|
|
|
|
2010-07-15 22:54:19 +04:00
|
|
|
cmp ax , 1337 ;;;;;
|
2009-09-17 15:55:38 +04:00
|
|
|
je ETH_API
|
|
|
|
|
2009-09-22 20:42:54 +04:00
|
|
|
add esp, 4 ; if we reached here, no function was called, so we need to balance stack
|
2009-09-17 15:55:38 +04:00
|
|
|
|
|
|
|
.doesnt_exist:
|
2010-07-15 22:54:19 +04:00
|
|
|
DEBUGF 1,"sys_protocols: protocol %u doesnt exist on device %u!\n", ax, bh
|
2009-09-17 15:55:38 +04:00
|
|
|
mov eax, -1
|
|
|
|
|
|
|
|
.return:
|
2009-09-22 20:42:54 +04:00
|
|
|
mov [esp+28+4], eax
|
2009-11-09 16:59:46 +03:00
|
|
|
ret
|
2010-05-29 00:47:32 +04:00
|
|
|
|
|
|
|
|
|
|
|
__DEBUG_LEVEL__ equ __DEBUG_LEVEL_OLD__
|