Subversion Repositories Kolibri OS

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

;;                                                                 ;;

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

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

;;                                                                 ;;

;;  SOCKET.INC                                                     ;;

;;                                                                 ;;

;;    Written by hidnplayr@kolibrios.org                           ;;

;;    based on code by mike.dld                                    ;;

;;                                                                 ;;

;;          GNU GENERAL PUBLIC LICENSE                             ;;

;;             Version 2, June 1991                                ;;

;;                                                                 ;;

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

$Revision: 1318 $

struct	SOCKET_head

	 .PrevPtr		dd ? ; pointer to previous socket in list

	 .NextPtr		dd ? ; pointer to next socket in list

	 .Number		dd ? ; socket number (unique within single process)

	 .PID			dd ? ; application process id

	 .Domain		dd ? ; INET/UNIX/..

	 .Type			dd ? ; RAW/UDP/TCP/...

	 .Protocol		dd ? ; ICMP/IPv4/ARP/

	 .lock			dd ? ; lock mutex

	 .errorcode		dd ?

	 .end:

ends

struct	IPv4_SOCKET

	 .LocalIP		dd ?

	 .RemoteIP		dd ?

	 .SequenceNumber	dd ?

	; todo: add options (for func 8 and 9)

	 .end:

ends

struct	TCP_SOCKET

	 .LocalPort		dw ? ; In INET byte order

	 .RemotePort		dw ? ; In INET byte order

	 .backlog		dw ? ; Backlog

	 .backlog_cur		dw ? ; current size of queue for un-accept-ed connections

	 .last_ack_number	dd ? ; used only to let application know that ACK has been received

					; todo: may be use SND_UNA instead

					; todo: may be use events which allow additional information instead

					; todo: may be count acknowledged bytes (at least it has obvious sense)

	 .OrigRemoteIP		dd ? ; original remote IP address (used to reset to LISTEN state)

	 .OrigRemotePort	dw ? ; original remote port (used to reset to LISTEN state)

	 .wndsizeTimer		dd ? ; window size timer

	; Transmission control block

	 .state 		dd ? ; TCB state

	 .timer 		dd ? ; TCB timer (seconds)

	 .ISS			dd ? ; initial send sequence number

	 .IRS			dd ? ; initial receive sequence number

	 .SND_UNA		dd ? ; sequence number of unack'ed sent Packets

	 .SND_NXT		dd ? ; next send sequence number to use

	 .SND_WND		dd ? ; send window

	 .RCV_NXT		dd ? ; next receive sequence number to use

	 .RCV_WND		dd ? ; receive window

	 .SEG_LEN		dd ? ; segment length

	 .SEG_WND		dd ? ; segment window

	 .flags 		db ? ; packet flags

	 .end:

ends

struct	UDP_SOCKET

	 .LocalPort		dw ? ; In INET byte order

	 .RemotePort		dw ? ; In INET byte order

	 .end:

ends

struct	ICMP_SOCKET

	.Identifier		dw ? ;

	.end:

ends

struct	IPC_SOCKET

	.ConnectedTo		dd ? ; Socket number of other socket this one is connected to

	.end:

ends

struct	socket_queue_entry

	.data_ptr	dd ?

	.data_size	dd ?

	.offset 	dd ?

	.size:

ends

MAX_backlog		equ 20	     ; backlog for stream sockets

SOCKETBUFFSIZE		equ 4096     ; in bytes

SOCKET_QUEUE_SIZE	equ 10	     ; maximum number ofincoming packets queued for 1 socket

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

uglobal

	net_sockets	rd 2

	last_UDP_port	dw ? ; These values give the number of the last used ephemeral port

	last_TCP_port	dw ? ;

endg

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

;

; SOCKET_init

;

;  -

;

;  IN:  /

;  OUT: /

;

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

align 4

socket_init:

	mov	[net_sockets], 0

	mov	[net_sockets + 4], 0

	mov	[last_UDP_port], MIN_EPHEMERAL_PORT

	mov	[last_TCP_port], MIN_EPHEMERAL_PORT

	ret

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

;

; Socket API (function 74)

;

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

align 4

sys_socket:

	and	ebx, 0x000000FF ; should i remove this line ?

	cmp	bl , 8		; highest possible number

	jg	s_error

	lea	ebx, [.table + 4*ebx]

	jmp	dword [ebx]

.table:

	dd	socket_open	; 0

	dd	socket_close	; 1

	dd	socket_bind	; 2

	dd	socket_listen	; 3

	dd	socket_connect	; 4

	dd	socket_accept	; 5

	dd	socket_send	; 6

	dd	socket_recv	; 7

	dd	socket_get_opt	; 8

;        dd      socket_set_opt  ; 9

s_error:

	mov	dword [esp+32],-1

	ret

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

;

; SOCKET_open

;

;

;  IN:  domain in ecx

;       type in edx

;       protocol in esi

;  OUT: eax is socket num, -1 on error

;

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

align 4

socket_open:

	DEBUGF	1,"socket_open: domain: %u, type: %u",ecx, edx

	call	net_socket_alloc

	or	eax, eax

	jz	s_error

	mov	[eax + SOCKET_head.Domain], ecx

	mov	[eax + SOCKET_head.Type], edx

	mov	[eax + SOCKET_head.Protocol], esi

	cmp	ecx, AF_INET4

	je	.af_inet4

	jmp	.done

  .af_inet4:

	cmp	edx, IP_PROTO_TCP

	je	.tcp

	jmp	.done

  .tcp:

	mov	[eax + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.state], TCB_CLOSED

	pseudo_random ebx

	mov	[eax + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.ISS], ebx

	mov	[eax + SOCKET_head.end + IPv4_SOCKET.end + TCP_SOCKET.SND_NXT], ebx

  .done:

	stdcall net_socket_addr_to_num, eax