2011-01-25 01:01:54 +03:00
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
;; ;;
|
2012-02-21 12:44:51 +04:00
|
|
|
;; Copyright (C) KolibriOS team 2004-2012. All rights reserved. ;;
|
2011-01-25 01:01:54 +03:00
|
|
|
;; Distributed under terms of the GNU General Public License ;;
|
|
|
|
;; ;;
|
|
|
|
;; Part of the tcp/ip network stack for KolibriOS ;;
|
|
|
|
;; ;;
|
|
|
|
;; Written by hidnplayr@kolibrios.org ;;
|
|
|
|
;; ;;
|
|
|
|
;; Based on the code of 4.4BSD ;;
|
|
|
|
;; ;;
|
|
|
|
;; GNU GENERAL PUBLIC LICENSE ;;
|
|
|
|
;; Version 2, June 1991 ;;
|
|
|
|
;; ;;
|
|
|
|
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
|
|
|
|
|
|
$Revision$
|
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
;-----------------------------------------------------------------
|
|
|
|
;
|
|
|
|
; TCP_input:
|
|
|
|
;
|
|
|
|
; IN: [esp] = ptr to buffer
|
|
|
|
; [esp+4] = buffer size
|
|
|
|
; ebx = ptr to device struct
|
|
|
|
; ecx = segment size
|
2011-11-13 00:05:07 +04:00
|
|
|
; esi = ptr to TCP segment
|
2011-11-12 19:39:15 +04:00
|
|
|
; edi = ptr to ipv4 source address, followed by ipv4 dest address
|
2011-01-08 17:59:21 +03:00
|
|
|
;
|
|
|
|
; OUT: /
|
|
|
|
;
|
|
|
|
;-----------------------------------------------------------------
|
2011-02-01 19:14:18 +03:00
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
align 4
|
|
|
|
TCP_input:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP_input size=%u\n", ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
; First, re-calculate the checksum
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ecx esi
|
|
|
|
pushw [esi + TCP_header.Checksum]
|
|
|
|
mov [esi + TCP_header.Checksum], 0
|
|
|
|
TCP_checksum (edi), (edi+4)
|
|
|
|
pop cx ; previous checksum
|
|
|
|
cmp cx, dx
|
|
|
|
pop edx ecx
|
|
|
|
jne .drop_not_locked
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Checksum ok\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
and [edx + TCP_header.DataOffset], 0xf0 ; Calculate TCP segment header size (throwing away unused reserved bits in TCP header)
|
|
|
|
shr [edx + TCP_header.DataOffset], 2
|
|
|
|
cmp [edx + TCP_header.DataOffset], sizeof.TCP_header ; Now see if it's at least the size of a standard TCP header
|
|
|
|
jb .drop_not_locked ; If not, drop the packet
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx eax, [edx + TCP_header.DataOffset]
|
|
|
|
sub ecx, eax ; substract TCP header size from total segment size
|
|
|
|
jb .drop_not_locked ; If total segment size is less then the advertised header size, drop packet
|
|
|
|
DEBUGF 1,"we got %u bytes of data\n", ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;-----------------------------------------------------------------------------------------
|
|
|
|
; Check if this packet has a timestamp option (We do it here so we can process it quickly)
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp eax, sizeof.TCP_header + 12 ; Timestamp option is 12 bytes
|
|
|
|
jb .no_timestamp
|
|
|
|
je .is_ok
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [edx + sizeof.TCP_header + 12], TCP_OPT_EOL ; end of option list
|
|
|
|
jne .no_timestamp
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.is_ok:
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN ; SYN flag must not be set
|
|
|
|
jnz .no_timestamp
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp dword [edx + sizeof.TCP_header], 0x0101080a ; Timestamp header
|
|
|
|
jne .no_timestamp
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"timestamp ok\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
; TODO: Parse the option
|
|
|
|
; TODO: Set a Bit in the TCP to tell all options are parsed
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_timestamp:
|
|
|
|
|
|
|
|
;-------------------------------------------
|
|
|
|
; Convert Big-endian values to little endian
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
ntohd [edx + TCP_header.SequenceNumber]
|
|
|
|
ntohd [edx + TCP_header.AckNumber]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
ntohw [edx + TCP_header.Window]
|
|
|
|
ntohw [edx + TCP_header.UrgentPointer]
|
|
|
|
ntohw [edx + TCP_header.SourcePort]
|
|
|
|
ntohw [edx + TCP_header.DestinationPort]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;------------------------------------------------------------
|
2011-02-01 20:00:04 +03:00
|
|
|
; Next thing to do is find the TCPS (thus, the socket pointer)
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; IP Packet TCP Destination Port = local Port
|
|
|
|
; (IP Packet SenderAddress = Remote IP) OR (Remote IP = 0)
|
2011-11-12 23:20:33 +04:00
|
|
|
; (IP Packet TCP Source Port = remote Port) OR (remote Port = 0)
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov ebx, net_sockets
|
|
|
|
mov si, [edx + TCP_header.DestinationPort]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.socket_loop:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov ebx, [ebx + SOCKET.NextPtr]
|
|
|
|
or ebx, ebx
|
|
|
|
jz .drop_with_reset_not_locked
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + SOCKET.Domain], AF_INET4
|
|
|
|
jne .socket_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + SOCKET.Protocol], IP_PROTO_TCP
|
|
|
|
jne .socket_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.LocalPort], si
|
|
|
|
jne .socket_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + IP_SOCKET.RemoteIP]
|
|
|
|
cmp eax, [edi] ; Ipv4 source addres
|
|
|
|
je @f
|
|
|
|
test eax, eax
|
|
|
|
jnz .socket_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov ax, [ebx + TCP_SOCKET.RemotePort]
|
|
|
|
cmp [edx + TCP_header.SourcePort], ax
|
|
|
|
je .found_socket
|
|
|
|
test ax, ax
|
|
|
|
jnz .socket_loop
|
|
|
|
.found_socket: ; ebx now contains the socketpointer
|
|
|
|
DEBUGF 1,"Socket ptr: %x\n", ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
;----------------------------
|
|
|
|
; Check if socket isnt closed
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_CLOSED
|
|
|
|
je .drop_not_locked
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;----------------
|
|
|
|
; Lock the socket
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + SOCKET.lock], 0
|
|
|
|
jne .drop_not_locked ;;; HACK ! HACK ! dirty fucking HACK ! !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
2011-02-01 19:14:18 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add ebx, SOCKET.lock
|
|
|
|
DEBUGF 1,"lock: %x\n", [ebx]
|
|
|
|
call wait_mutex
|
|
|
|
sub ebx, SOCKET.lock
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Socket locked\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;---------------------------------------
|
|
|
|
; unscale the window into a 32 bit value
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx eax, [edx + TCP_header.Window]
|
|
|
|
push ecx
|
|
|
|
mov cl, [ebx + TCP_SOCKET.SND_SCALE]
|
|
|
|
shl eax, cl
|
|
|
|
mov dword [edx + TCP_header.Window], eax ; word after window is checksum, we dont need checksum anymore
|
|
|
|
pop ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;-----------------------------------
|
|
|
|
; Is this socket a listening socket?
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [ebx + SOCKET.options], SO_ACCEPTCON
|
|
|
|
jz .no_listening_socket
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Accepting new connection\n"
|
2011-02-04 00:35:16 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
|
|
|
push ecx edx esi edi ;;;
|
|
|
|
call SOCKET_fork
|
|
|
|
pop edi esi edx ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test eax, eax
|
|
|
|
jz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push dword [edi + 4] ; Ipv4 destination addres
|
|
|
|
pop [eax + IP_SOCKET.LocalIP]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.DestinationPort]
|
|
|
|
pop [eax + TCP_SOCKET.LocalPort]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [eax + TCP_SOCKET.t_state], TCPS_LISTEN
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-02-04 00:35:16 +03:00
|
|
|
; mov [ebx + SOCKET.lock], 0
|
2012-02-24 02:21:40 +04:00
|
|
|
mov ebx, eax
|
2011-02-04 00:35:16 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .LISTEN
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_listening_socket:
|
|
|
|
|
|
|
|
;-------------------------------------
|
|
|
|
; Reset idle timer and keepalive timer
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_idle], 0
|
|
|
|
mov [ebx + TCP_SOCKET.timer_keepalive], TCP_time_keep_interval
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;--------------------
|
|
|
|
; Process TCP options
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx eax, [edx + TCP_header.DataOffset]
|
|
|
|
cmp eax, sizeof.TCP_header ; Does header contain any options?
|
|
|
|
je .no_options
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Segment has options\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_LISTEN ; no options when in listen state
|
|
|
|
jz .not_uni_xfer ; also no header prediction
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add eax, edx
|
|
|
|
lea esi, [edx + sizeof.TCP_header]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.opt_loop:
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp esi, eax ; are we scanning outside of header?
|
|
|
|
jae .no_options
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi], TCP_OPT_EOL ; end of option list?
|
|
|
|
jz .no_options
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi], TCP_OPT_NOP ; nop ?
|
|
|
|
jz .opt_nop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi], TCP_OPT_MAXSEG
|
|
|
|
je .opt_maxseg
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi], TCP_OPT_WINDOW
|
|
|
|
je .opt_window
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi], TCP_OPT_TIMESTAMP
|
|
|
|
je .opt_timestamp
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .no_options ; If we reach here, some unknown options were received, skip them all!
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.opt_nop:
|
2012-02-24 02:21:40 +04:00
|
|
|
inc edi
|
|
|
|
jmp .opt_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.opt_maxseg:
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi+1], 4
|
|
|
|
jne .no_options ; error occured, ignore all options!
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx eax, word[esi+2]
|
|
|
|
rol ax, 8
|
|
|
|
DEBUGF 1,"Maxseg: %u\n", ax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_maxseg], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
add edi, 4
|
|
|
|
jmp .opt_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
.opt_window:
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi+1], 3
|
|
|
|
jne .no_options
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Got window option\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;;;;
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
add edi, 3
|
|
|
|
jmp .opt_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
.opt_timestamp:
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp byte [esi+1], 10
|
|
|
|
jne .no_options
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Got timestamp option\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;;;;
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add esi, 10
|
|
|
|
jmp .opt_loop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_options:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-----------------------------------------------------------------------
|
|
|
|
; Time to do some header prediction (Original Principle by Van Jacobson)
|
|
|
|
|
|
|
|
; There are two common cases for an uni-directional data transfer.
|
|
|
|
;
|
|
|
|
; General rule: the packets has no control flags, is in-sequence,
|
|
|
|
; window width didnt change and we're not retransmitting.
|
|
|
|
;
|
|
|
|
; Second rules:
|
|
|
|
; - If the length is 0 and the ACK moved forward, we're the sender side of the transfer.
|
|
|
|
; In this case we'll free the ACK'ed data and notify higher levels that we have free space in buffer
|
|
|
|
;
|
|
|
|
; - If the length is not 0 and the ACK didn't move, we're the receiver side of the transfer.
|
|
|
|
; If the packets are in order (data queue is empty), add the data to the socket buffer and request a delayed ACK
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_ESTABLISHED
|
|
|
|
jnz .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN + TH_FIN + TH_RST + TH_URG
|
|
|
|
jnz .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.RCV_NXT]
|
|
|
|
jne .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, dword [edx + TCP_header.Window]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_WND]
|
|
|
|
jne .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_NXT]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
jne .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;---------------------------------------
|
|
|
|
; check if we are sender in the uni-xfer
|
|
|
|
|
|
|
|
; If the following 4 conditions are all true, this segment is a pure ACK.
|
|
|
|
;
|
|
|
|
; - The segment contains no data.
|
2012-02-24 02:21:40 +04:00
|
|
|
test ecx, ecx
|
|
|
|
jnz .not_sender
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; - The congestion window is greater than or equal to the current send window.
|
|
|
|
; This test is true only if the window is fully open, that is, the connection is not in the middle of slow start or congestion avoidance.
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_CWND]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_WND]
|
|
|
|
jb .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; - The acknowledgment field in the segment is less than or equal to the maximum sequence number sent.
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
ja .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; - The acknowledgment field in the segment is greater than the largest unacknowledged sequence number.
|
2012-02-24 02:21:40 +04:00
|
|
|
sub eax, [ebx + TCP_SOCKET.SND_UNA]
|
|
|
|
jbe .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Header prediction: we are sender\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;---------------------------------
|
|
|
|
; Packet is a pure ACK, process it
|
|
|
|
|
|
|
|
; Update RTT estimators
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
;;; TODO
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
; Delete acknowledged bytes from send buffer
|
2012-02-24 02:21:40 +04:00
|
|
|
pusha
|
|
|
|
mov ecx, eax
|
|
|
|
lea eax, [ebx + STREAM_SOCKET.snd]
|
|
|
|
call SOCKET_ring_free
|
|
|
|
popa
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; update window pointers
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_UNA], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Stop retransmit timer
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.timer_retransmission], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Awaken waiting processes
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_notify_owner
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
; Generate more output
|
2012-02-24 02:21:40 +04:00
|
|
|
call TCP_output
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop_not_locked
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;-------------------------------------------------
|
|
|
|
; maybe we are the receiver in the uni-xfer then..
|
|
|
|
|
|
|
|
.not_sender:
|
|
|
|
; - The amount of data in the segment is greater than 0 (data count is in ecx)
|
|
|
|
|
|
|
|
; - The acknowledgment field equals the largest unacknowledged sequence number. This means no data is acknowledged by this segment.
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_UNA]
|
|
|
|
jne .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; - The reassembly list of out-of-order segments for the connection is empty (seg_next equals tp).
|
|
|
|
|
|
|
|
;;; TODO
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jnz .not_uni_xfer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Complete processing of received data
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"header prediction: we are receiver\nreceiving %u bytes of data\n", ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add [ebx + TCP_SOCKET.RCV_NXT], ecx ; Update sequence number with number of bytes we have copied
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx esi, [edx + TCP_header.DataOffset]
|
|
|
|
add esi, edx
|
|
|
|
lea eax, [ebx + STREAM_SOCKET.rcv]
|
|
|
|
call SOCKET_ring_write ; Add the data to the socket buffer
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_notify_owner
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
or [ebx + TCP_SOCKET.t_flags], TF_DELACK ; Set delayed ack flag
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;--------------------------------------------------
|
|
|
|
; Header prediction failed, do it the slow way
|
|
|
|
|
|
|
|
.not_uni_xfer:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Header prediction failed\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Calculate receive window size
|
|
|
|
|
2012-02-21 12:44:51 +04:00
|
|
|
; mov eax, [ebx + STREAM_SOCKET.rcv.size]
|
2011-11-09 03:06:26 +04:00
|
|
|
; neg eax
|
|
|
|
; add eax, SOCKETBUFFSIZE
|
|
|
|
; mov edx, [ebx + TCP_SOCKET.RCV_ADV]
|
|
|
|
; sub edx, [ebx + TCP_SOCKET.RCV_NXT]
|
|
|
|
; cmp eax, edx
|
|
|
|
; jae @f
|
|
|
|
; mov eax, edx
|
|
|
|
; @@:
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_LISTEN
|
|
|
|
je .LISTEN
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_SYN_SENT
|
|
|
|
je .SYN_SENT
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .NOT_LISTEN_OR_SYN_SENT
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-------------
|
|
|
|
; Passive Open
|
|
|
|
|
|
|
|
align 4
|
|
|
|
.LISTEN:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP state: listen\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_RST ;;; TODO: kill new socket on error
|
|
|
|
jnz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jnz .drop_with_reset
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO: check if it's a broadcast or multicast, and drop if so
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push dword [edi + 4] ; Ipv4 destination addres
|
|
|
|
pop [ebx + IP_SOCKET.RemoteIP]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.SourcePort]
|
|
|
|
pop [ebx + TCP_SOCKET.RemotePort]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.SequenceNumber]
|
|
|
|
pop [ebx + TCP_SOCKET.IRS]
|
2011-02-04 00:35:16 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [TCP_sequence_num] ;;;;;
|
|
|
|
add [TCP_sequence_num], 64000 / 2
|
|
|
|
pop [ebx + TCP_SOCKET.ISS]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [ebx + TCP_SOCKET.ISS]
|
|
|
|
pop [ebx + TCP_SOCKET.SND_NXT]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
TCP_sendseqinit ebx
|
|
|
|
TCP_rcvseqinit ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_SYN_RECEIVED
|
|
|
|
mov [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
|
|
|
mov [ebx + TCP_SOCKET.timer_keepalive], TCP_time_keep_interval ;;;; macro
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
lea eax, [ebx + STREAM_SOCKET.snd]
|
|
|
|
call SOCKET_ring_create
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
lea eax, [ebx + STREAM_SOCKET.rcv]
|
|
|
|
call SOCKET_ring_create
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-02-04 00:35:16 +03:00
|
|
|
;;; call SOCKET_notify_owner
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .trim_then_step6
|
|
|
|
|
2011-02-04 00:35:16 +03:00
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;------------
|
|
|
|
; Active Open
|
|
|
|
|
|
|
|
align 4
|
|
|
|
.SYN_SENT:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP state: syn_sent\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.ISS]
|
|
|
|
jbe .drop_with_reset
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
ja .drop_with_reset
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_RST
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
mov ebx, ECONNREFUSED
|
|
|
|
call TCP_drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; at this point, segment seems to be valid
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .no_syn_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; now, process received SYN in response to an active open
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_UNA], eax
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_NXT]
|
|
|
|
jbe @f
|
|
|
|
mov [ebx + TCP_SOCKET.SND_NXT], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
.no_syn_ack:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.timer_retransmission], 0 ; disable retransmission
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.SequenceNumber]
|
|
|
|
pop [ebx + TCP_SOCKET.IRS]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
TCP_rcvseqinit ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
or [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_UNA]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.ISS]
|
|
|
|
jbe .simultaneous_open
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .simultaneous_open
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP: active open\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO: update stats
|
|
|
|
|
|
|
|
; set socket state to connected
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.state],1 ;;;; FIXME
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_ESTABLISHED
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO: check if we should scale the connection (567-572)
|
|
|
|
;;; TODO: update RTT estimators
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .trim_then_step6
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.simultaneous_open:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP: simultaneous open\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
; We have received a syn but no ACK, so we are having a simultaneous open..
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_SYN_RECEIVED
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-------------------------------------
|
|
|
|
; Common processing for receipt of SYN
|
|
|
|
|
|
|
|
.trim_then_step6:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
inc [edx + TCP_header.SequenceNumber]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO: Drop any received data that follows receive window (590)
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
mov [ebx + TCP_SOCKET.RCV_UP], eax
|
|
|
|
dec eax
|
|
|
|
mov [ebx + TCP_SOCKET.SND_WL1], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .ack_processed
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.NOT_LISTEN_OR_SYN_SENT:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Slow TCP input: not listen or syn_sent state\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;--------------------------------------------
|
|
|
|
; Protection Against Wrapped Sequence Numbers
|
|
|
|
|
|
|
|
; First, check if timestamp is present
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
;;;; TODO 602
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Then, check if at least some bytes of data are within window
|
|
|
|
|
|
|
|
;;;; TODO
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;----------------------------
|
|
|
|
; trim any data not in window
|
|
|
|
|
|
|
|
; check for duplicate data at beginning of segment
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.RCV_NXT]
|
|
|
|
sub eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
jbe .no_duplicate
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Uh oh.. %u bytes of duplicate data!\n", eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz .no_dup_syn
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; remove duplicate syn
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
and [edx + TCP_header.Flags], not (TH_SYN)
|
|
|
|
inc [edx + TCP_header.SequenceNumber]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [edx + TCP_header.UrgentPointer], 1
|
|
|
|
jbe @f
|
|
|
|
dec [edx + TCP_header.UrgentPointer]
|
|
|
|
jmp .dup_syn
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
and [edx + TCP_header.Flags], not (TH_URG)
|
2011-02-04 00:35:16 +03:00
|
|
|
.dup_syn:
|
2012-02-24 02:21:40 +04:00
|
|
|
dec eax
|
2011-01-25 01:01:54 +03:00
|
|
|
.no_dup_syn:
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; eax holds number of bytes to drop
|
|
|
|
|
|
|
|
; Check for entire duplicate packet
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp eax, ecx
|
|
|
|
jae .duplicate
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Going to drop %u out of %u bytes\n", eax, ecx
|
2011-02-01 19:14:18 +03:00
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
;;; TODO: apply figure 28.30
|
|
|
|
|
|
|
|
; Check for duplicate FIN
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_FIN
|
|
|
|
jz @f
|
|
|
|
inc ecx
|
|
|
|
cmp eax, ecx
|
|
|
|
dec ecx
|
|
|
|
jne @f
|
|
|
|
|
|
|
|
mov eax, ecx
|
|
|
|
and [edx + TCP_header.Flags], not TH_FIN
|
|
|
|
or [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
|
|
|
jmp .no_duplicate
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
; Handle the case when a bound socket connects to itself
|
|
|
|
; Allow packets with a SYN and an ACKto continue with the processing
|
|
|
|
|
|
|
|
;-------------------------------------
|
|
|
|
; Generate duplicate ACK if nescessary
|
|
|
|
|
|
|
|
; This code also handles simultaneous half-open or self-connects
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test eax, eax
|
|
|
|
jnz .drop_after_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .drop_after_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.duplicate:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Duplicate received\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;----------------------------------------
|
|
|
|
; Update statistics for duplicate packets
|
|
|
|
|
|
|
|
;;; TODO
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop_after_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_duplicate:
|
|
|
|
|
|
|
|
;-----------------------------------------------
|
|
|
|
; Remove duplicate data and update urgent offset
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add [edx + TCP_header.SequenceNumber], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
sub [edx + TCP_header.UrgentPointer], ax
|
|
|
|
ja @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
and [edx + TCP_header.Flags], not (TH_URG)
|
|
|
|
mov [edx + TCP_header.UrgentPointer], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
;--------------------------------------------------
|
|
|
|
; Handle data that arrives after process terminates
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + SOCKET.PID], 0
|
|
|
|
ja @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_CLOSE_WAIT
|
|
|
|
jbe @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test ecx, ecx
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; Close the socket
|
|
|
|
;;; update stats
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop_with_reset
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
;----------------------------------------
|
|
|
|
; Remove data beyond right edge of window
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
add eax, ecx
|
|
|
|
sub eax, [ebx + TCP_SOCKET.RCV_NXT]
|
|
|
|
sub ax, [ebx + TCP_SOCKET.RCV_WND]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; eax now holds the number of bytes to drop
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jbe .no_excess_data
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO: update stats
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp eax, ecx
|
|
|
|
jb .dont_drop_all
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;;; TODO 700-736
|
|
|
|
|
|
|
|
.dont_drop_all:
|
|
|
|
|
|
|
|
.no_excess_data:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-----------------
|
|
|
|
; Record timestamp
|
|
|
|
|
|
|
|
;;; TODO 737-746
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;------------------
|
|
|
|
; Process RST flags
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_RST
|
|
|
|
jz .rst_skip
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Got an RST flag"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.t_state]
|
|
|
|
shl eax, 2
|
|
|
|
jmp dword [eax + .rst_sw_list]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.rst_sw_list:
|
2012-02-24 02:21:40 +04:00
|
|
|
dd .rst_skip ;TCPS_CLOSED
|
|
|
|
dd .rst_skip ;TCPS_LISTEN
|
|
|
|
dd .rst_skip ;TCPS_SYN_SENT
|
|
|
|
dd .econnrefused ;TCPS_SYN_RECEIVED
|
|
|
|
dd .econnreset ;TCPS_ESTABLISHED
|
|
|
|
dd .econnreset ;TCPS_CLOSE_WAIT
|
|
|
|
dd .econnreset ;TCPS_FIN_WAIT_1
|
|
|
|
dd .rst_close ;TCPS_CLOSING
|
|
|
|
dd .rst_close ;TCPS_LAST_ACK
|
|
|
|
dd .econnreset ;TCPS_FIN_WAIT_2
|
|
|
|
dd .rst_close ;TCPS_TIMED_WAIT
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.econnrefused:
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Connection refused"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.errorcode], ECONNREFUSED
|
|
|
|
jmp .close
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.econnreset:
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Connection reset"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.errorcode], ECONNRESET
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-02-01 20:00:04 +03:00
|
|
|
.close:
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Closing connection"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_CLOSED
|
|
|
|
;;; TODO: update stats
|
|
|
|
mov eax, ebx
|
|
|
|
call TCP_close
|
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.rst_close:
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Closing with reset\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call TCP_close
|
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
.rst_skip:
|
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;--------------------------------------
|
|
|
|
; handle SYN-full and ACK-less segments
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jz @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
mov ebx, ECONNRESET
|
|
|
|
call TCP_drop
|
|
|
|
jmp .drop_with_reset
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;---------------
|
|
|
|
; ACK processing
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_SYN_RECEIVED
|
|
|
|
jnz .no_syn_rcv
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP state = syn received\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp [ebx + TCP_SOCKET.SND_UNA], eax
|
|
|
|
ja .drop_with_reset
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
ja .drop_with_reset
|
2011-02-04 00:35:16 +03:00
|
|
|
|
2011-11-12 23:20:33 +04:00
|
|
|
;;; TODO: update stats
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_is_connected
|
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_ESTABLISHED
|
2011-02-01 20:00:04 +03:00
|
|
|
|
|
|
|
; Do window scaling?
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [ebx + TCP_SOCKET.t_flags], TF_RCVD_SCALE
|
|
|
|
jz @f
|
|
|
|
test [ebx + TCP_SOCKET.t_flags], TF_REQ_SCALE
|
|
|
|
jz @f
|
2011-02-01 20:00:04 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push word [ebx + TCP_SOCKET.requested_s_scale] ; Set send and receive scale factors to the received values
|
|
|
|
pop word [ebx + TCP_SOCKET.SND_SCALE]
|
2011-02-01 20:00:04 +03:00
|
|
|
@@:
|
|
|
|
|
2011-11-12 23:20:33 +04:00
|
|
|
;;; TODO: copy the data (if any) into the socket
|
2011-02-01 20:00:04 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
dec eax
|
|
|
|
mov [ebx + TCP_SOCKET.SND_WL1], eax
|
|
|
|
jmp .not_dup_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_syn_rcv:
|
|
|
|
|
|
|
|
; check for duplicate ACK
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_UNA]
|
|
|
|
ja .not_dup_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test ecx, ecx
|
|
|
|
jnz .reset_dupacks
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, dword [edx + TCP_header.Window]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_WND]
|
|
|
|
jne .reset_dupacks
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Processing a duplicate ACK..\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.timer_retransmission], 10000 ;;;; FIXME
|
|
|
|
ja @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_UNA]
|
|
|
|
je .dup_ack
|
2011-01-25 01:01:54 +03:00
|
|
|
|
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_dupacks], 0
|
|
|
|
jmp .not_dup_ack
|
2011-01-25 01:01:54 +03:00
|
|
|
|
|
|
|
.dup_ack:
|
2012-02-24 02:21:40 +04:00
|
|
|
inc [ebx + TCP_SOCKET.t_dupacks]
|
|
|
|
cmp [ebx + TCP_SOCKET.t_dupacks], TCP_re_xmit_thresh
|
|
|
|
jne .no_re_xmit
|
|
|
|
|
|
|
|
push [ebx + TCP_SOCKET.SND_NXT] ; >>>>
|
|
|
|
|
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_WND]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_CWND]
|
|
|
|
cmovg eax, [ebx + TCP_SOCKET.SND_CWND]
|
|
|
|
shr eax, 1
|
|
|
|
push edx
|
|
|
|
xor edx, edx
|
|
|
|
div [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
cmp eax, 2
|
|
|
|
jae @f
|
|
|
|
mov ax, 2
|
2011-01-25 01:01:54 +03:00
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
mul [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
pop edx
|
|
|
|
mov [ebx + TCP_SOCKET.SND_SSTHRESH], eax
|
|
|
|
|
|
|
|
mov [ebx + TCP_SOCKET.timer_retransmission], 0 ; turn off retransmission timer
|
|
|
|
mov [ebx + TCP_SOCKET.t_rtt], 0
|
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_NXT], eax
|
|
|
|
mov eax, [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_CWND], eax
|
|
|
|
|
|
|
|
mov eax, ebx
|
|
|
|
call TCP_output ; retransmit missing segment
|
|
|
|
|
|
|
|
push edx
|
|
|
|
xor edx, edx
|
|
|
|
mov eax, [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
mul [ebx + TCP_SOCKET.t_dupacks]
|
|
|
|
pop edx
|
|
|
|
add eax, [ebx + TCP_SOCKET.SND_SSTHRESH]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_CWND], eax
|
|
|
|
|
|
|
|
pop eax ; <<<<
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_NXT]
|
|
|
|
jb @f
|
|
|
|
mov [ebx + TCP_SOCKET.SND_NXT], eax
|
2011-01-25 01:01:54 +03:00
|
|
|
@@:
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
.no_re_xmit:
|
2012-02-24 02:21:40 +04:00
|
|
|
jbe .not_dup_ack
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Increasing congestion window\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
add [ebx + TCP_SOCKET.SND_CWND], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call TCP_output
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
|
|
|
|
.not_dup_ack:
|
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
;-------------------------------------------------
|
|
|
|
; If the congestion window was inflated to account
|
|
|
|
; for the other side's cached packets, retract it
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_SSTHRESH]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_CWND]
|
|
|
|
ja @f
|
|
|
|
cmp [ebx + TCP_SOCKET.t_dupacks], TCP_re_xmit_thresh
|
|
|
|
jbe @f
|
|
|
|
mov [ebx + TCP_SOCKET.SND_CWND], eax
|
2011-01-25 01:01:54 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_dupacks], 0
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
jbe @f
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; TODO: update stats
|
|
|
|
jmp .drop_after_ack
|
2011-01-25 01:01:54 +03:00
|
|
|
|
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov edi, [edx + TCP_header.AckNumber]
|
|
|
|
sub edi, [ebx + TCP_SOCKET.SND_UNA] ; now we got the number of acked bytes in edi
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; TODO: update stats
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"We have an acceptable ACK of %x bytes\n", esi
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;------------------------------------------
|
|
|
|
; RTT measurements and retransmission timer
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;;;; 912 - 926
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.timer_retransmission], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_MAX]
|
|
|
|
cmp eax, [edx + TCP_header.AckNumber]
|
|
|
|
je .all_outstanding
|
|
|
|
mov [ebx + TCP_SOCKET.timer_retransmission], 120 ;;;; TODO: correct this value (use a macro for it)
|
2011-01-08 17:59:21 +03:00
|
|
|
.all_outstanding:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-------------------------------------------
|
|
|
|
; Open congestion window in response to ACKs
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov esi, [ebx + TCP_SOCKET.SND_CWND]
|
|
|
|
mov eax, [ebx + TCP_SOCKET.t_maxseg]
|
|
|
|
|
|
|
|
cmp esi, [ebx + TCP_SOCKET.SND_SSTHRESH]
|
|
|
|
jbe @f
|
|
|
|
push edx
|
|
|
|
push eax
|
|
|
|
mul eax
|
|
|
|
div esi
|
|
|
|
pop edx
|
|
|
|
shr edx, 3
|
|
|
|
add eax, edx
|
|
|
|
pop edx
|
2011-01-25 01:01:54 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
add esi, eax
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ecx
|
|
|
|
mov cl, [ebx + TCP_SOCKET.SND_SCALE]
|
|
|
|
mov eax, TCP_max_win
|
|
|
|
shl eax, cl
|
|
|
|
pop ecx
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp esi, eax
|
|
|
|
cmovg esi, eax
|
|
|
|
mov [ebx + TCP_SOCKET.SND_CWND], esi
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;------------------------------------------
|
|
|
|
; Remove acknowledged data from send buffer
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ecx edx ebx
|
|
|
|
mov ecx, edi
|
|
|
|
lea eax, [ebx + STREAM_SOCKET.snd]
|
|
|
|
call SOCKET_ring_free
|
|
|
|
pop ebx
|
|
|
|
sub [ebx + TCP_SOCKET.SND_WND], ecx
|
|
|
|
pop edx ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Wake up process waiting on send buffer
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_notify_owner
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2011-02-01 20:00:04 +03:00
|
|
|
; Update TCPS
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [edx + TCP_header.AckNumber]
|
|
|
|
mov [ebx + TCP_SOCKET.SND_UNA], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_NXT]
|
|
|
|
jb @f
|
|
|
|
mov [ebx + TCP_SOCKET.SND_NXT], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;; TODO: use zero flag as 'ourfinisacked'
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
2011-01-25 01:01:54 +03:00
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
; General ACK handling complete
|
|
|
|
; Now do the state-specific ones
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.t_state]
|
|
|
|
jmp dword [eax*4 + .ACK_sw_list]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.ACK_sw_list:
|
2012-02-24 02:21:40 +04:00
|
|
|
dd .ack_processed ;TCPS_CLOSED
|
|
|
|
dd .ack_processed ;TCPS_LISTEN
|
|
|
|
dd .ack_processed ;TCPS_SYN_SENT
|
|
|
|
dd .ack_processed ;TCPS_SYN_RECEIVED
|
|
|
|
dd .ack_processed ;TCPS_ESTABLISHED
|
|
|
|
dd .ack_processed ;TCPS_CLOSE_WAIT
|
|
|
|
dd .ack_fw1 ;TCPS_FIN_WAIT_1
|
|
|
|
dd .ack_c ;TCPS_CLOSING
|
|
|
|
dd .ack_la ;TCPS_LAST_ACK
|
|
|
|
dd .ack_processed ;TCPS_FIN_WAIT_2
|
|
|
|
dd .ack_tw ;TCPS_TIMED_WAIT
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
2011-01-28 19:16:31 +03:00
|
|
|
.ack_fw1:
|
2012-02-24 02:21:40 +04:00
|
|
|
jz .ack_processed
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [ebx + SOCKET.state], SS_CANTRCVMORE
|
|
|
|
jnz @f
|
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_is_disconnected
|
2011-01-28 19:49:08 +03:00
|
|
|
;;; mov [ebx + TCP_SOCKET.timer_timed_wait], TCP_time_max_idle
|
2011-01-28 19:16:31 +03:00
|
|
|
@@:
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_FIN_WAIT_2
|
|
|
|
jmp .ack_processed
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
2011-01-28 19:16:31 +03:00
|
|
|
.ack_c:
|
2012-02-24 02:21:40 +04:00
|
|
|
jz .ack_processed
|
2011-01-28 19:16:31 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_TIMED_WAIT
|
|
|
|
mov eax, ebx
|
|
|
|
call TCP_cancel_timers
|
|
|
|
mov [ebx + TCP_SOCKET.timer_timed_wait], 2 * TCP_time_MSL
|
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_is_disconnected
|
|
|
|
jmp .ack_processed
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
2011-01-28 19:16:31 +03:00
|
|
|
.ack_la:
|
2012-02-24 02:21:40 +04:00
|
|
|
jz .ack_processed
|
2011-01-28 19:16:31 +03:00
|
|
|
|
2011-01-28 19:49:08 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call TCP_close
|
|
|
|
jmp .drop
|
2011-01-28 19:16:31 +03:00
|
|
|
|
|
|
|
|
|
|
|
.ack_tw:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.timer_timed_wait], 2 * TCP_time_MSL
|
|
|
|
jmp .drop_after_ack
|
2011-01-28 19:16:31 +03:00
|
|
|
|
|
|
|
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
.reset_dupacks: ; We got a new ACK, reset duplicate ACK counter
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_dupacks], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
.ack_processed: ; (step 6)
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"ACK processed\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;----------------------------------------------
|
|
|
|
; check if we need to update window information
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jz .no_window_update
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_WL1]
|
|
|
|
cmp eax, [edx + TCP_header.SequenceNumber]
|
|
|
|
jb .update_window
|
|
|
|
ja @f
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_WL2]
|
|
|
|
cmp eax, [edx + TCP_header.AckNumber]
|
|
|
|
jb .update_window
|
|
|
|
ja .no_window_update
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.SND_WL2]
|
|
|
|
cmp eax, [edx + TCP_header.AckNumber]
|
|
|
|
jne .no_window_update
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, dword [edx + TCP_header.Window]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.SND_WND]
|
|
|
|
jbe .no_window_update
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.update_window:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Updating window\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Keep track of pure window updates
|
|
|
|
|
|
|
|
; test ecx, ecx
|
|
|
|
; jz @f
|
|
|
|
;
|
|
|
|
; mov eax, [ebx + TCP_SOCKET.SND_WL2]
|
2011-11-09 03:06:26 +04:00
|
|
|
; cmp eax, [edx + TCP_header.AckNumber]
|
2011-01-08 17:59:21 +03:00
|
|
|
; jne @f
|
|
|
|
;
|
|
|
|
; ;; mov eax, tiwin
|
|
|
|
; cmp eax, [ebx + TCP_SOCKET.SND_WND]
|
2011-11-08 00:25:47 +04:00
|
|
|
; jbe @f
|
2011-01-08 17:59:21 +03:00
|
|
|
;
|
|
|
|
; ;;; update stats
|
|
|
|
;
|
|
|
|
; @@:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, dword [edx + TCP_header.Window]
|
|
|
|
cmp eax, [ebx + TCP_SOCKET.max_sndwnd]
|
|
|
|
jbe @f
|
|
|
|
mov [ebx + TCP_SOCKET.max_sndwnd], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
@@:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.SND_WND], eax
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.SequenceNumber]
|
|
|
|
pop [ebx + TCP_SOCKET.SND_WL1]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push [edx + TCP_header.AckNumber]
|
|
|
|
pop [ebx + TCP_SOCKET.SND_WL2]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; needoutput = 1
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_window_update:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-----------------
|
|
|
|
; process URG flag
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_URG
|
|
|
|
jz .not_urgent
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [edx + TCP_header.UrgentPointer], 0
|
|
|
|
jz .not_urgent
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_TIMED_WAIT
|
|
|
|
je .not_urgent
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
; Ignore bogus urgent offsets
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; 1040-1050
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx eax, [edx + TCP_header.UrgentPointer]
|
|
|
|
add eax, [ebx + STREAM_SOCKET.rcv.size]
|
|
|
|
cmp eax, SOCKET_MAXDATA
|
|
|
|
jbe .not_urgent
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [edx + TCP_header.UrgentPointer], 0
|
|
|
|
and [edx + TCP_header.Flags], not (TH_URG)
|
|
|
|
jmp .do_data
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.not_urgent:
|
|
|
|
|
|
|
|
; processing of received urgent pointer
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; TODO (1051-1093)
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;--------------------------------
|
|
|
|
; process the data in the segment
|
|
|
|
|
|
|
|
.do_data:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"TCP: do data (%u)\n", ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_FIN
|
|
|
|
jnz .process_fin
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_FIN_WAIT_1
|
|
|
|
jae .dont_do_data
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test ecx, ecx
|
|
|
|
jz .final_processing
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Processing data in segment\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
;; TODO: check if data is in sequence !
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
movzx esi, [edx + TCP_header.DataOffset]
|
|
|
|
add esi, edx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
or [ebx + TCP_SOCKET.t_flags], TF_DELACK
|
|
|
|
add [ebx + TCP_SOCKET.RCV_NXT], ecx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
lea eax, [ebx + STREAM_SOCKET.rcv]
|
|
|
|
call SOCKET_ring_write
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_notify_owner
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .final_processing
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
.dont_do_data:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;---------------
|
|
|
|
; FIN processing
|
|
|
|
|
|
|
|
.process_fin:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Processing FIN\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_CLOSE_WAIT
|
|
|
|
je .not_first_fin
|
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_CLOSING
|
|
|
|
je .not_first_fin
|
|
|
|
cmp [ebx + TCP_SOCKET.t_state], TCPS_FIN_WAIT_2
|
|
|
|
je .not_first_fin
|
2011-01-29 02:07:25 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"First FIN for this connection\n"
|
2011-01-29 02:07:25 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call SOCKET_cant_recv_more
|
2011-01-29 02:07:25 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
|
|
|
inc [ebx + TCP_SOCKET.RCV_NXT]
|
2011-01-29 02:07:25 +03:00
|
|
|
|
|
|
|
.not_first_fin:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, [ebx + TCP_SOCKET.t_state]
|
|
|
|
shl eax, 2
|
|
|
|
jmp dword [eax + .FIN_sw_list]
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.FIN_sw_list:
|
2012-02-24 02:21:40 +04:00
|
|
|
dd .no_fin ;TCPS_CLOSED
|
|
|
|
dd .no_fin ;TCPS_LISTEN
|
|
|
|
dd .no_fin ;TCPS_SYN_SENT
|
|
|
|
dd .fin_syn_est ;TCPS_SYN_RECEIVED
|
|
|
|
dd .fin_syn_est ;TCPS_ESTABLISHED
|
|
|
|
dd .no_fin ;TCPS_CLOSE_WAIT
|
|
|
|
dd .fin_wait1 ;TCPS_FIN_WAIT_1
|
|
|
|
dd .no_fin ;TCPS_CLOSING
|
|
|
|
dd .no_fin ;TCPS_LAST_ACK
|
|
|
|
dd .fin_wait2 ;TCPS_FIN_WAIT_2
|
|
|
|
dd .fin_timed ;TCPS_TIMED_WAIT
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.fin_syn_est:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_CLOSE_WAIT
|
|
|
|
jmp .no_fin
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.fin_wait1:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_CLOSING
|
|
|
|
jmp .no_fin
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.fin_wait2:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.t_state], TCPS_TIMED_WAIT
|
|
|
|
mov eax, ebx
|
|
|
|
call TCP_cancel_timers
|
|
|
|
mov [ebx + TCP_SOCKET.timer_timed_wait], 2 * TCP_time_MSL
|
|
|
|
call SOCKET_is_disconnected
|
|
|
|
jmp .no_fin
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.fin_timed:
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + TCP_SOCKET.timer_timed_wait], 2 * TCP_time_MSL
|
|
|
|
jmp .no_fin
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.no_fin:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-----------------
|
|
|
|
; Final processing
|
|
|
|
|
|
|
|
.final_processing:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Final processing\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; if debug enabled, output packet
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;test needoutput, needoutput
|
|
|
|
;jz .dumpit
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
|
|
|
jz .dumpit
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"ACK now!\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ebx
|
|
|
|
mov eax, ebx
|
|
|
|
call TCP_output
|
|
|
|
pop ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.dumpit:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
call kernel_free
|
|
|
|
add esp, 4
|
|
|
|
ret
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;------------------------------------------
|
|
|
|
; Generate an ACK, droping incoming segment
|
|
|
|
|
|
|
|
align 4
|
|
|
|
.drop_after_ack:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Drop after ACK\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_RST
|
|
|
|
jnz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
and [ebx + TCP_SOCKET.t_flags], TF_ACKNOW
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ebx
|
2011-02-04 00:35:16 +03:00
|
|
|
; mov cl, TH_ACK
|
|
|
|
; call TCP_respond_socket
|
2012-02-24 02:21:40 +04:00
|
|
|
mov eax, ebx
|
|
|
|
call TCP_output
|
|
|
|
pop ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
call kernel_free
|
|
|
|
add esp, 4
|
|
|
|
ret
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-------------------------------------------
|
|
|
|
; Generate an RST, dropping incoming segment
|
|
|
|
|
|
|
|
align 4
|
|
|
|
.drop_with_reset:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.drop_with_reset_not_locked:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Drop with reset\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_RST
|
|
|
|
jnz .drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;; if its a multicast/broadcast, also drop
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_ACK
|
|
|
|
jnz .respond_ack
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
test [edx + TCP_header.Flags], TH_SYN
|
|
|
|
jnz .respond_syn
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
call kernel_free
|
|
|
|
add esp, 4
|
|
|
|
ret
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.respond_ack:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ebx
|
|
|
|
mov cl, TH_RST
|
|
|
|
call TCP_respond_socket
|
|
|
|
pop ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .destroy_new_socket
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
.respond_syn:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
push ebx
|
|
|
|
mov cl, TH_RST + TH_ACK
|
|
|
|
call TCP_respond_socket
|
|
|
|
pop ebx
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
jmp .destroy_new_socket
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;-----
|
|
|
|
; Drop
|
|
|
|
|
|
|
|
align 4
|
|
|
|
.drop:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
mov [ebx + SOCKET.lock], 0
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.drop_not_locked:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
DEBUGF 1,"Dropping packet\n"
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;;; If debugging options are enabled, output the packet somwhere
|
2011-01-08 17:59:21 +03:00
|
|
|
|
|
|
|
.destroy_new_socket:
|
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
;;;; kill the newly created socket
|
2011-01-08 17:59:21 +03:00
|
|
|
|
2012-02-24 02:21:40 +04:00
|
|
|
call kernel_free
|
|
|
|
add esp, 4
|
|
|
|
ret
|