Subversion Repositories Kolibri OS

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

;;                                                                 ;;

;; Copyright (C) KolibriOS team 2004-2009. All rights reserved.    ;;

;; Distributed under terms of the GNU General Public License       ;;

;;                                                                 ;;

;;  TCP.INC                                                        ;;

;;                                                                 ;;

;;  Part of the tcp/ip network stack for KolibriOS                 ;;

;;                                                                 ;;

;;    Written by hidnplayr@kolibrios.org                           ;;

;;                                                                 ;;

;;          GNU GENERAL PUBLIC LICENSE                             ;;

;;             Version 2, June 1991                                ;;

;;                                                                 ;;

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

$Revision: 1249 $

TCB_LISTEN		equ 1

TCB_SYN_SENT		equ 2

TCB_SYN_RECEIVED	equ 3

TCB_ESTABLISHED 	equ 4

TCB_FIN_WAIT_1		equ 5

TCB_FIN_WAIT_2		equ 6

TCB_CLOSE_WAIT		equ 7

TCB_CLOSING		equ 8

TCB_LAST_ACK		equ 9

TCB_TIMED_WAIT		equ 10

TCB_CLOSED		equ 11

TH_FIN			equ 1 shl 0

TH_SYN			equ 1 shl 1

TH_RST			equ 1 shl 2

TH_PUSH 		equ 1 shl 3

TH_ACK			equ 1 shl 4

TH_URG			equ 1 shl 5

TWOMSL			equ 10		; # of secs to wait before closing socket

TCP_RETRIES		equ 5		; Number of times to resend a Packet

TCP_TIMEOUT		equ 10		; resend if not replied to in 1/100 s

TCP_QUEUE_SIZE		equ 16

struct	TCP_Packet

	.SourcePort		dw ?

	.DestinationPort	dw ?

	.SequenceNumber 	dd ?

	.AckNumber		dd ?

	.DataOffset		db ?	; DataOffset[0-3 bits] and Reserved[4-7]

	.Flags			db ?	; Reserved[0-1 bits]|URG|ACK|PSH|RST|SYN|FIN

	.Window 		dw ?

	.Checksum		dw ?

	.UrgentPointer		dw ?

	.Options		rb 3

	.Padding		db ?

	.Data:

ends

align 4

uglobal

	TCP_PACKETS_TX		rd  MAX_IP

	TCP_PACKETS_RX		rd  MAX_IP

	TCP_IN_QUEUE		rd  (tcp_in_queue_entry.size*TCP_QUEUE_SIZE+queue.data)/4

	TCP_OUT_QUEUE		dd ?

				rd  (tcp_out_queue_entry.size*TCP_QUEUE_SIZE)/4

endg

align 4

iglobal

TCBStateHandler:

  dd  stateTCB_LISTEN

  dd  stateTCB_SYN_SENT

  dd  stateTCB_SYN_RECEIVED

  dd  stateTCB_ESTABLISHED

  dd  stateTCB_FIN_WAIT_1

  dd  stateTCB_FIN_WAIT_2

  dd  stateTCB_CLOSE_WAIT

  dd  stateTCB_CLOSING

  dd  stateTCB_LAST_ACK

  dd  stateTCB_TIME_WAIT

  dd  stateTCB_CLOSED

endg

macro inc_INET reg {

	inc	byte [reg + 0]

	adc	byte [reg + 1], 0

	adc	byte [reg + 2], 0

	adc	byte [reg + 3], 0

}

macro add_INET reg {

	rol	ecx, 16

	adc	byte [reg + 0], ch

	adc	byte [reg + 1], cl

	rol	ecx, 16

	adc	byte [reg + 2], ch

	adc	byte [reg + 3], cl

}

;-----------------------------------------------------------------

;

; TCP_init

;

;  This function resets all TCP variables

;

;  IN:  /

;  OUT: /

;

;-----------------------------------------------------------------

align 4

TCP_init:

	xor	eax, eax

	mov	edi, TCP_PACKETS_TX

	mov	ecx, 2*MAX_IP

	rep	stosd

	init_queue TCP_IN_QUEUE

	init_queue TCP_OUT_QUEUE

	ret

;-----------------------------------------------------------------

;

;  TCP_decrease_socket_ttls

;

;  IN:  /

;  OUT: /

;

;-----------------------------------------------------------------

align 4

TCP_decrease_socket_ttls:

	; scan through all the sockets, decrementing active timers

	mov	ebx, net_sockets

	cmp	[ebx + SOCKET_head.NextPtr], 0

	je	.exit

  .next_socket:

	mov	ebx, [ebx + SOCKET_head.NextPtr]

	or	ebx, ebx

	jz	.exit

	cmp	[ebx + SOCKET_head.Type], IP_PROTO_TCP

	jne	.next_socket

;        DEBUGF  1, "K :   %x-%x: %x-%x-%x-%u\n", [ebx + SOCKET.PID]:2, [ebx + SOCKET.Number]:2, [ebx + SOCKET.LocalPort]:4, [ebx + SOCKET.RemoteIP], [ebx + SOCKET.RemotePort]:4, [ebx + SOCKET.TCBState]

	cmp	[ebx + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.TCBTimer], 0

	jne	.decrement_tcb

	cmp	[ebx + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.wndsizeTimer], 0

	jne	.decrement_wnd

	jmp	.next_socket

  .decrement_tcb:

	; decrement it, delete socket if TCB timer = 0 & socket in timewait state

	dec	[ebx + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.TCBTimer]

	jnz	.next_socket

	cmp	[ebx + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.TCBState], TCB_TIMED_WAIT

	jne	.next_socket

	push	[ebx + SOCKET_head.PrevPtr]

	stdcall net_socket_free, ebx

	pop	ebx

	jmp	.next_socket

  .decrement_wnd:

	; TODO - prove it works!

	dec	[ebx + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.wndsizeTimer]

	jmp	.next_socket

  .exit:

	ret

;-----------------------------------------------------------------

;

; TCP_send_queued:

;

;  Decreases 'ttl' of tcp packets queued.

;  if 'ttl' reaches 0, resend the packet and decrease 'retries'

;  if 'retries' reaches zero, remove the queued packet

;

;  IN:  /

;  OUT: /

;

;-----------------------------------------------------------------

align 4

TCP_send_queued: