WebSVN – Kolibri OS – Diff – /kernel/trunk/network/tcp_usreq.inc


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;;                                                                 ;;
;; Copyright (C) KolibriOS team 2004-2015. All rights reserved.    ;;
;; 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: 6011 $
 
 
;-----------------------------------------------------------------;
;                                                                 ;
; tcp_usrclosed                                                   ;
;                                                                 ;
;  IN:  eax = socket ptr                                          ;
;                                                                 ;
;  OUT: /                                                         ;
;                                                                 ;
;-----------------------------------------------------------------;
align 4
tcp_usrclosed:
 
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_usrclosed: %x\n", eax
 
        push    ebx
        mov     ebx, [eax + TCP_SOCKET.t_state]
        mov     ebx, dword [.switch + ebx*4]
        jmp     ebx
 
  .switch:
        dd      .close                  ; TCPS_CLOSED
        dd      .close                  ; TCPS_LISTEN
        dd      .close                  ; TCPS_SYN_SENT
        dd      .wait1                  ; TCPS_SYN_RECEIVED
        dd      .wait1                  ; TCPS_ESTABLISHED
        dd      .last_ack               ; TCPS_CLOSE_WAIT
        dd      .ret                    ; TCPS_FIN_WAIT_1
        dd      .ret                    ; TCPS_CLOSING
        dd      .ret                    ; TCPS_LAST_ACK
        dd      .disc                   ; TCPS_FIN_WAIT_2
        dd      .disc                   ; TCPS_TIMED_WAIT
 
  .close:
        mov     [eax + TCP_SOCKET.t_state], TCPS_CLOSED
        call    tcp_close
        pop     ebx
        ret
 
  .wait1:
        mov     [eax + TCP_SOCKET.t_state], TCPS_FIN_WAIT_1
        pop     ebx
        ret
 
  .last_ack:
        mov     [eax + TCP_SOCKET.t_state], TCPS_LAST_ACK
        pop     ebx
        ret
 
  .disc:
        call    socket_is_disconnected
  .ret:
        pop     ebx
        ret
 
 
;-----------------------------------------------------------------;
;                                                                 ;
; tcp_connect                                                     ;
;                                                                 ;
;  IN:  eax = socket ptr                                          ;
;                                                                 ;
;  OUT: eax = 0 on success                                        ;
;       eax = -1 on error                                         ;
;       ebx = error code on error                                 ;
;                                                                 ;
;-----------------------------------------------------------------;
align 4
tcp_connect:
 
        test    [eax + SOCKET.state], SS_ISCONNECTED
        jnz     .eisconn
 
        inc     [TCPS_connattempt]                      ; update stats
 
        push    eax edx
        lea     ecx, [eax + SOCKET.mutex]
        call    mutex_lock
        pop     edx eax
 
; Fill in local IP
        cmp     [eax + IP_SOCKET.LocalIP], 0
        jne     @f
        push    [IP_LIST + 4]                                   ; FIXME: use correct local IP
        pop     [eax + IP_SOCKET.LocalIP]
 
; Fill in remote port and IP
        pushw   [edx + 2]
        pop     [eax + TCP_SOCKET.RemotePort]
 
        pushd   [edx + 4]
        pop     [eax + IP_SOCKET.RemoteIP]
 
; Find a local port, if user didnt define one
        cmp     [eax + TCP_SOCKET.LocalPort], 0
        jne     @f
        call    socket_find_port
       @@:
 
; Start the TCP sequence
        mov     [eax + TCP_SOCKET.timer_persist], 0
        mov     [eax + TCP_SOCKET.t_state], TCPS_SYN_SENT
 
        push    [TCP_sequence_num]
        add     [TCP_sequence_num], 6400
        pop     [eax + TCP_SOCKET.ISS]
        mov     [eax + TCP_SOCKET.timer_keepalive], TCP_time_keep_init
 
        tcp_sendseqinit eax
 
        mov     ebx, eax
        lea     eax, [ebx + STREAM_SOCKET.snd]
        call    socket_ring_create
        test    eax, eax
        jz      .nomem
 
        lea     eax, [ebx + STREAM_SOCKET.rcv]
        call    socket_ring_create
        test    eax, eax
        jz      .nomem
 
        push    ebx
        lea     ecx, [ebx + SOCKET.mutex]
        call    mutex_unlock
        pop     eax
 
        call    socket_is_connecting
 
; Now send the SYN packet to remote end
        push    eax
        call    tcp_output
        pop     eax
 
  .block:
        test    [eax + SOCKET.options], SO_NONBLOCK
        jz      .waitforit
 
        xor     eax, eax
        dec     eax
        mov     ebx, EINPROGRESS
        ret
 
  .nomem:
        xor     eax, eax
        dec     eax
        mov     ebx, ENOMEM
        ret
 
  .eisconn:
        xor     eax, eax
        dec     eax
        mov     ebx, EISCONN
        ret
 
  .waitforit:
        push    eax
        stdcall timer_hs, TCP_time_connect, 0, .timeout, eax
        pop     ebx
        mov     [ebx + TCP_SOCKET.timer_connect], eax
        mov     eax, ebx
 
  .loop:
        cmp     [eax + SOCKET.errorcode], 0
        jne     .fail
        cmp     [eax + TCP_SOCKET.t_state], TCPS_ESTABLISHED
        je      .established
 
        call    socket_block
        jmp     .loop
 
  .timeout:
        mov     eax, [esp+4]
        mov     [eax + SOCKET.errorcode], ETIMEDOUT
        and     [eax + SOCKET.state], not SS_ISCONNECTING
        call    socket_notify
        ret     4
 
  .fail:
        mov     ebx, [eax + SOCKET.errorcode]
        mov     [eax + SOCKET.errorcode], 0                     ; Clear the error, we only need to send it to the caller once
        xor     eax, eax
        dec     eax
        ret
 
  .established:
        stdcall cancel_timer_hs, [eax + TCP_SOCKET.timer_connect]
 
        xor     eax, eax
        ret

Rev 5976	Rev 6011
1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2	;; ;;	2	;; ;;
3	;; Copyright (C) KolibriOS team 2004-2015. All rights reserved. ;;	3	;; Copyright (C) KolibriOS team 2004-2015. All rights reserved. ;;
4	;; Distributed under terms of the GNU General Public License ;;	4	;; Distributed under terms of the GNU General Public License ;;
5	;; ;;	5	;; ;;
6	;; Part of the TCP/IP network stack for KolibriOS ;;	6	;; Part of the TCP/IP network stack for KolibriOS ;;
7	;; ;;	7	;; ;;
8	;; Written by hidnplayr@kolibrios.org ;;	8	;; Written by hidnplayr@kolibrios.org ;;
9	;; ;;	9	;; ;;
10	;; Based on the code of 4.4BSD ;;	10	;; Based on the code of 4.4BSD ;;
11	;; ;;	11	;; ;;
12	;; GNU GENERAL PUBLIC LICENSE ;;	12	;; GNU GENERAL PUBLIC LICENSE ;;
13	;; Version 2, June 1991 ;;	13	;; Version 2, June 1991 ;;
14	;; ;;	14	;; ;;
15	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;	15	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
16		16
17	$Revision: 5976 $	17	$Revision: 6011 $
18		18
19		19
20	;-----------------------------------------------------------------;	20	;-----------------------------------------------------------------;
21	; ;	21	; ;
22	; TCP_usrclosed ;	22	; tcp_usrclosed ;
23	; ;	23	; ;
24	; IN: eax = socket ptr ;	24	; IN: eax = socket ptr ;
25	; ;	25	; ;
26	; OUT: / ;	26	; OUT: / ;
27	; ;	27	; ;
28	;-----------------------------------------------------------------;	28	;-----------------------------------------------------------------;
29	align 4	29	align 4
30	TCP_usrclosed:	30	tcp_usrclosed:
31		31
32	DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_usrclosed: %x\n", eax	32	DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_usrclosed: %x\n", eax
33		33
34	push ebx	34	push ebx
35	mov ebx, [eax + TCP_SOCKET.t_state]	35	mov ebx, [eax + TCP_SOCKET.t_state]
36	mov ebx, dword [.switch + ebx*4]	36	mov ebx, dword [.switch + ebx*4]
37	jmp ebx	37	jmp ebx
38		38
39	.switch:	39	.switch:
40	dd .close ; TCPS_CLOSED	40	dd .close ; TCPS_CLOSED
41	dd .close ; TCPS_LISTEN	41	dd .close ; TCPS_LISTEN
42	dd .close ; TCPS_SYN_SENT	42	dd .close ; TCPS_SYN_SENT
43	dd .wait1 ; TCPS_SYN_RECEIVED	43	dd .wait1 ; TCPS_SYN_RECEIVED
44	dd .wait1 ; TCPS_ESTABLISHED	44	dd .wait1 ; TCPS_ESTABLISHED
45	dd .last_ack ; TCPS_CLOSE_WAIT	45	dd .last_ack ; TCPS_CLOSE_WAIT
46	dd .ret ; TCPS_FIN_WAIT_1	46	dd .ret ; TCPS_FIN_WAIT_1
47	dd .ret ; TCPS_CLOSING	47	dd .ret ; TCPS_CLOSING
48	dd .ret ; TCPS_LAST_ACK	48	dd .ret ; TCPS_LAST_ACK
49	dd .disc ; TCPS_FIN_WAIT_2	49	dd .disc ; TCPS_FIN_WAIT_2
50	dd .disc ; TCPS_TIMED_WAIT	50	dd .disc ; TCPS_TIMED_WAIT
51		51
52	.close:	52	.close:
53	mov [eax + TCP_SOCKET.t_state], TCPS_CLOSED	53	mov [eax + TCP_SOCKET.t_state], TCPS_CLOSED
54	call TCP_close	54	call tcp_close
55	pop ebx	55	pop ebx
56	ret	56	ret
57		57
58	.wait1:	58	.wait1:
59	mov [eax + TCP_SOCKET.t_state], TCPS_FIN_WAIT_1	59	mov [eax + TCP_SOCKET.t_state], TCPS_FIN_WAIT_1
60	pop ebx	60	pop ebx
61	ret	61	ret
62		62
63	.last_ack:	63	.last_ack:
64	mov [eax + TCP_SOCKET.t_state], TCPS_LAST_ACK	64	mov [eax + TCP_SOCKET.t_state], TCPS_LAST_ACK
65	pop ebx	65	pop ebx
66	ret	66	ret
67		67
68	.disc:	68	.disc:
69	call SOCKET_is_disconnected	69	call socket_is_disconnected
70	.ret:	70	.ret:
71	pop ebx	71	pop ebx
72	ret	72	ret
73		73
74		74
75	;-----------------------------------------------------------------;	75	;-----------------------------------------------------------------;
76	; ;	76	; ;
77	; TCP_connect ;	77	; tcp_connect ;
78	; ;	78	; ;
79	; IN: eax = socket ptr ;	79	; IN: eax = socket ptr ;
80	; ;	80	; ;
81	; OUT: eax = 0 on success ;	81	; OUT: eax = 0 on success ;
82	; eax = -1 on error ;	82	; eax = -1 on error ;
83	; ebx = error code on error ;	83	; ebx = error code on error ;
84	; ;	84	; ;
85	;-----------------------------------------------------------------;	85	;-----------------------------------------------------------------;
86	align 4	86	align 4
87	TCP_connect:	87	tcp_connect:
88		88
89	test [eax + SOCKET.state], SS_ISCONNECTED	89	test [eax + SOCKET.state], SS_ISCONNECTED
90	jnz .eisconn	90	jnz .eisconn
91		91
92	inc [TCPS_connattempt] ; update stats	92	inc [TCPS_connattempt] ; update stats
93		93
94	push eax edx	94	push eax edx
95	lea ecx, [eax + SOCKET.mutex]	95	lea ecx, [eax + SOCKET.mutex]
96	call mutex_lock	96	call mutex_lock
97	pop edx eax	97	pop edx eax
98		98
99	; Fill in local IP	99	; Fill in local IP
100	cmp [eax + IP_SOCKET.LocalIP], 0	100	cmp [eax + IP_SOCKET.LocalIP], 0
101	jne @f	101	jne @f
102	push [IP_LIST + 4] ; FIXME: use correct local IP	102	push [IP_LIST + 4] ; FIXME: use correct local IP
103	pop [eax + IP_SOCKET.LocalIP]	103	pop [eax + IP_SOCKET.LocalIP]
104		104
105	; Fill in remote port and IP	105	; Fill in remote port and IP
106	pushw [edx + 2]	106	pushw [edx + 2]
107	pop [eax + TCP_SOCKET.RemotePort]	107	pop [eax + TCP_SOCKET.RemotePort]
108		108
109	pushd [edx + 4]	109	pushd [edx + 4]
110	pop [eax + IP_SOCKET.RemoteIP]	110	pop [eax + IP_SOCKET.RemoteIP]
111		111
112	; Find a local port, if user didnt define one	112	; Find a local port, if user didnt define one
113	cmp [eax + TCP_SOCKET.LocalPort], 0	113	cmp [eax + TCP_SOCKET.LocalPort], 0
114	jne @f	114	jne @f
115	call SOCKET_find_port	115	call socket_find_port
116	@@:	116	@@:
117		117
118	; Start the TCP sequence	118	; Start the TCP sequence
119	mov [eax + TCP_SOCKET.timer_persist], 0	119	mov [eax + TCP_SOCKET.timer_persist], 0
120	mov [eax + TCP_SOCKET.t_state], TCPS_SYN_SENT	120	mov [eax + TCP_SOCKET.t_state], TCPS_SYN_SENT
121		121
122	push [TCP_sequence_num]	122	push [TCP_sequence_num]
123	add [TCP_sequence_num], 6400	123	add [TCP_sequence_num], 6400
124	pop [eax + TCP_SOCKET.ISS]	124	pop [eax + TCP_SOCKET.ISS]
125	mov [eax + TCP_SOCKET.timer_keepalive], TCP_time_keep_init	125	mov [eax + TCP_SOCKET.timer_keepalive], TCP_time_keep_init
126		126
127	TCP_sendseqinit eax	127	tcp_sendseqinit eax
128		128
129	mov ebx, eax	129	mov ebx, eax
130	lea eax, [ebx + STREAM_SOCKET.snd]	130	lea eax, [ebx + STREAM_SOCKET.snd]
131	call SOCKET_ring_create	131	call socket_ring_create
132	test eax, eax	132	test eax, eax
133	jz .nomem	133	jz .nomem
134		134
135	lea eax, [ebx + STREAM_SOCKET.rcv]	135	lea eax, [ebx + STREAM_SOCKET.rcv]
136	call SOCKET_ring_create	136	call socket_ring_create
137	test eax, eax	137	test eax, eax
138	jz .nomem	138	jz .nomem
139		139
140	push ebx	140	push ebx
141	lea ecx, [ebx + SOCKET.mutex]	141	lea ecx, [ebx + SOCKET.mutex]
142	call mutex_unlock	142	call mutex_unlock
143	pop eax	143	pop eax
144		144
145	call SOCKET_is_connecting	145	call socket_is_connecting
146		146
147	; Now send the SYN packet to remote end	147	; Now send the SYN packet to remote end
148	push eax	148	push eax
149	call TCP_output	149	call tcp_output
150	pop eax	150	pop eax
151		151
152	.block:	152	.block:
153	test [eax + SOCKET.options], SO_NONBLOCK	153	test [eax + SOCKET.options], SO_NONBLOCK
154	jz .waitforit	154	jz .waitforit
155		155
156	xor eax, eax	156	xor eax, eax
157	dec eax	157	dec eax
158	mov ebx, EINPROGRESS	158	mov ebx, EINPROGRESS
159	ret	159	ret
160		160
161	.nomem:	161	.nomem:
162	xor eax, eax	162	xor eax, eax
163	dec eax	163	dec eax
164	mov ebx, ENOMEM	164	mov ebx, ENOMEM
165	ret	165	ret
166		166
167	.eisconn:	167	.eisconn:
168	xor eax, eax	168	xor eax, eax
169	dec eax	169	dec eax
170	mov ebx, EISCONN	170	mov ebx, EISCONN
171	ret	171	ret
172		172
173	.waitforit:	173	.waitforit:
174	push eax	174	push eax
175	stdcall timer_hs, TCP_time_connect, 0, .timeout, eax	175	stdcall timer_hs, TCP_time_connect, 0, .timeout, eax
176	pop ebx	176	pop ebx
177	mov [ebx + TCP_SOCKET.timer_connect], eax	177	mov [ebx + TCP_SOCKET.timer_connect], eax
178	mov eax, ebx	178	mov eax, ebx
179		179
180	.loop:	180	.loop:
181	cmp [eax + SOCKET.errorcode], 0	181	cmp [eax + SOCKET.errorcode], 0
182	jne .fail	182	jne .fail
183	cmp [eax + TCP_SOCKET.t_state], TCPS_ESTABLISHED	183	cmp [eax + TCP_SOCKET.t_state], TCPS_ESTABLISHED
184	je .established	184	je .established
185		185
186	call SOCKET_block	186	call socket_block
187	jmp .loop	187	jmp .loop
188		188
189	.timeout:	189	.timeout:
190	mov eax, [esp+4]	190	mov eax, [esp+4]
191	mov [eax + SOCKET.errorcode], ETIMEDOUT	191	mov [eax + SOCKET.errorcode], ETIMEDOUT
192	and [eax + SOCKET.state], not SS_ISCONNECTING	192	and [eax + SOCKET.state], not SS_ISCONNECTING
193	call SOCKET_notify	193	call socket_notify
194	ret 4	194	ret 4
195		195
196	.fail:	196	.fail:
197	mov ebx, [eax + SOCKET.errorcode]	197	mov ebx, [eax + SOCKET.errorcode]
198	mov [eax + SOCKET.errorcode], 0 ; Clear the error, we only need to send it to the caller once	198	mov [eax + SOCKET.errorcode], 0 ; Clear the error, we only need to send it to the caller once
199	xor eax, eax	199	xor eax, eax
200	dec eax	200	dec eax
201	ret	201	ret
202		202
203	.established:	203	.established:
204	stdcall cancel_timer_hs, [eax + TCP_SOCKET.timer_connect]	204	stdcall cancel_timer_hs, [eax + TCP_SOCKET.timer_connect]
205		205
206	xor eax, eax	206	xor eax, eax
207	ret	207	ret

Subversion Repositories Kolibri OS

(root)/kernel/trunk/network/tcp_usreq.inc – Rev 5976 → 6011