Rev 6916 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
3545 | hidnplayr | 1 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
2 | ;; ;; |
||
6909 | hidnplayr | 3 | ;; Copyright (C) KolibriOS team 2004-2017. All rights reserved. ;; |
3545 | hidnplayr | 4 | ;; Distributed under terms of the GNU General Public License ;; |
5 | ;; ;; |
||
6 | ;; Part of the TCP/IP network stack for KolibriOS ;; |
||
7 | ;; ;; |
||
8 | ;; Written by hidnplayr@kolibrios.org ;; |
||
9 | ;; ;; |
||
10 | ;; Based on the code of 4.4BSD ;; |
||
11 | ;; ;; |
||
12 | ;; GNU GENERAL PUBLIC LICENSE ;; |
||
13 | ;; Version 2, June 1991 ;; |
||
14 | ;; ;; |
||
15 | ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
||
16 | |||
4850 | mario79 | 17 | $Revision: 7680 $ |
3545 | hidnplayr | 18 | |
6476 | hidnplayr | 19 | TCP_BIT_SENDALOT = 1 shl 0 |
20 | |||
5976 | hidnplayr | 21 | ;-----------------------------------------------------------------; |
22 | ; ; |
||
6011 | hidnplayr | 23 | ; tcp_output ; |
5976 | hidnplayr | 24 | ; ; |
25 | ; IN: eax = socket pointer ; |
||
26 | ; ; |
||
27 | ; OUT: eax = 0 on success/errorcode ; |
||
28 | ; ; |
||
29 | ;-----------------------------------------------------------------; |
||
3545 | hidnplayr | 30 | align 4 |
6011 | hidnplayr | 31 | proc tcp_output |
3545 | hidnplayr | 32 | |
4347 | hidnplayr | 33 | locals |
34 | temp_bits db ? |
||
6474 | hidnplayr | 35 | window dd ? |
4347 | hidnplayr | 36 | endl |
37 | |||
4366 | hidnplayr | 38 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: socket=%x state=%u\n", eax, [eax + TCP_SOCKET.t_state] |
3545 | hidnplayr | 39 | |
40 | push eax |
||
41 | lea ecx, [eax + SOCKET.mutex] |
||
42 | call mutex_lock |
||
43 | pop eax |
||
44 | |||
3652 | hidnplayr | 45 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: socket locked\n" |
46 | |||
3545 | hidnplayr | 47 | ; We'll detect the length of the data to be transmitted, and flags to be used |
48 | ; If there is some data, or any critical controls to send (SYN / RST), then transmit |
||
49 | ; Otherwise, investigate further |
||
50 | |||
51 | mov ebx, [eax + TCP_SOCKET.SND_MAX] |
||
52 | cmp ebx, [eax + TCP_SOCKET.SND_UNA] |
||
53 | jbe .not_idle |
||
54 | |||
55 | mov ebx, [eax + TCP_SOCKET.t_idle] |
||
56 | cmp ebx, [eax + TCP_SOCKET.t_rxtcur] |
||
57 | jbe .not_idle |
||
58 | |||
59 | ; We have been idle for a while and no ACKS are expected to clock out any data we send.. |
||
60 | ; Slow start to get ack "clock" running again. |
||
61 | |||
62 | mov ebx, [eax + TCP_SOCKET.t_maxseg] |
||
63 | mov [eax + TCP_SOCKET.SND_CWND], ebx |
||
64 | |||
65 | .not_idle: |
||
66 | .again: |
||
4347 | hidnplayr | 67 | mov [temp_bits], 0 |
3545 | hidnplayr | 68 | |
6476 | hidnplayr | 69 | ; Calculate offset |
3545 | hidnplayr | 70 | |
6476 | hidnplayr | 71 | mov ebx, [eax + TCP_SOCKET.SND_NXT] |
72 | sub ebx, [eax + TCP_SOCKET.SND_UNA] |
||
3545 | hidnplayr | 73 | |
6476 | hidnplayr | 74 | ; Determine window |
3545 | hidnplayr | 75 | |
6476 | hidnplayr | 76 | mov ecx, [eax + TCP_SOCKET.SND_WND] |
77 | cmp ecx, [eax + TCP_SOCKET.SND_CWND] |
||
78 | jb @f |
||
79 | mov ecx, [eax + TCP_SOCKET.SND_CWND] |
||
80 | @@: |
||
81 | |||
82 | ; get flags in dl |
||
83 | |||
84 | call tcp_outflags |
||
85 | |||
3545 | hidnplayr | 86 | ;------------------------ |
87 | ; data being forced out ? |
||
88 | |||
89 | ; If in persist timeout with window of 0, send 1 byte. |
||
90 | ; Otherwise, if window is small but nonzero, and timer expired, |
||
91 | ; we will send what we can and go to transmit state |
||
92 | |||
6916 | hidnplayr | 93 | test [eax + TCP_SOCKET.t_flags], TF_FORCE |
94 | jz .no_force |
||
3545 | hidnplayr | 95 | |
3556 | hidnplayr | 96 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: forcing data out\n" |
3545 | hidnplayr | 97 | |
98 | test ecx, ecx |
||
99 | jnz .no_zero_window |
||
100 | |||
101 | cmp ebx, [eax + STREAM_SOCKET.snd.size] |
||
102 | jae @f |
||
103 | and dl, not (TH_FIN) |
||
5442 | hidnplayr | 104 | @@: |
3545 | hidnplayr | 105 | |
106 | inc ecx |
||
107 | jmp .no_force |
||
108 | |||
109 | .no_zero_window: |
||
3603 | hidnplayr | 110 | and [eax + TCP_SOCKET.timer_flags], not timer_flag_persist |
3545 | hidnplayr | 111 | mov [eax + TCP_SOCKET.t_rxtshift], 0 |
112 | |||
113 | .no_force: |
||
114 | |||
115 | ;-------------------------------- |
||
6476 | hidnplayr | 116 | ; Calculate how much data to send |
3545 | hidnplayr | 117 | |
118 | mov esi, [eax + STREAM_SOCKET.snd.size] |
||
119 | cmp esi, ecx |
||
120 | jb @f |
||
121 | mov esi, ecx |
||
122 | @@: |
||
123 | sub esi, ebx |
||
124 | |||
125 | ;------------------------ |
||
6476 | hidnplayr | 126 | ; check for window shrink |
3545 | hidnplayr | 127 | |
6476 | hidnplayr | 128 | ; If FIN has been sent, but not ACKed, but we havent been called to retransmit, esi will be -1 |
3545 | hidnplayr | 129 | ; Otherwise, window shrank after we sent into it. |
130 | |||
131 | jae .not_persist |
||
132 | |||
133 | ; enter persist state |
||
6476 | hidnplayr | 134 | |
3545 | hidnplayr | 135 | xor esi, esi |
136 | |||
137 | ; If window shrank to 0 |
||
6476 | hidnplayr | 138 | |
3545 | hidnplayr | 139 | test ecx, ecx |
140 | jnz @f |
||
141 | |||
142 | ; cancel pending retransmit |
||
6476 | hidnplayr | 143 | |
3603 | hidnplayr | 144 | and [eax + TCP_SOCKET.timer_flags], not timer_flag_retransmission |
3545 | hidnplayr | 145 | |
146 | ; pull SND_NXT back to (closed) window, We will enter persist state below. |
||
6476 | hidnplayr | 147 | |
3545 | hidnplayr | 148 | push [eax + TCP_SOCKET.SND_UNA] |
149 | pop [eax + TCP_SOCKET.SND_NXT] |
||
150 | @@: |
||
151 | |||
152 | ; If window didn't close completely, just wait for an ACK |
||
153 | |||
154 | .not_persist: |
||
155 | |||
156 | ;--------------------------- |
||
6476 | hidnplayr | 157 | ; Send one segment at a time |
3545 | hidnplayr | 158 | |
159 | cmp esi, [eax + TCP_SOCKET.t_maxseg] |
||
160 | jbe @f |
||
161 | mov esi, [eax + TCP_SOCKET.t_maxseg] |
||
4347 | hidnplayr | 162 | or [temp_bits], TCP_BIT_SENDALOT |
3545 | hidnplayr | 163 | @@: |
164 | |||
165 | ;-------------------------------------------- |
||
6476 | hidnplayr | 166 | ; Turn of FIN flag if send buffer not emptied |
3545 | hidnplayr | 167 | |
168 | mov edi, [eax + TCP_SOCKET.SND_NXT] |
||
169 | add edi, esi |
||
170 | sub edi, [eax + TCP_SOCKET.SND_UNA] |
||
171 | cmp edi, [eax + STREAM_SOCKET.snd.size] |
||
172 | jae @f |
||
173 | and dl, not (TH_FIN) |
||
174 | @@: |
||
175 | |||
176 | ;------------------------------- |
||
6476 | hidnplayr | 177 | ; calculate window advertisement |
3545 | hidnplayr | 178 | |
6413 | hidnplayr | 179 | mov ecx, SOCKET_BUFFER_SIZE |
3545 | hidnplayr | 180 | sub ecx, [eax + STREAM_SOCKET.rcv.size] |
181 | |||
182 | ;------------------------------ |
||
6476 | hidnplayr | 183 | ; Sender silly window avoidance |
3545 | hidnplayr | 184 | |
185 | test esi, esi |
||
186 | jz .len_zero |
||
187 | |||
188 | cmp esi, [eax + TCP_SOCKET.t_maxseg] |
||
4387 | hidnplayr | 189 | je .send |
3545 | hidnplayr | 190 | |
191 | add ebx, esi ; offset + length |
||
192 | cmp ebx, [eax + STREAM_SOCKET.snd.size] |
||
193 | jb @f |
||
194 | |||
195 | test [eax + TCP_SOCKET.t_flags], TF_NODELAY |
||
4387 | hidnplayr | 196 | jnz .send |
3545 | hidnplayr | 197 | |
198 | mov ebx, [eax + TCP_SOCKET.SND_MAX] |
||
199 | cmp ebx, [eax + TCP_SOCKET.SND_UNA] |
||
4387 | hidnplayr | 200 | je .send |
3545 | hidnplayr | 201 | @@: |
202 | |||
6916 | hidnplayr | 203 | test [eax + TCP_SOCKET.t_flags], TF_FORCE |
4387 | hidnplayr | 204 | jnz .send |
3545 | hidnplayr | 205 | |
206 | mov ebx, [eax + TCP_SOCKET.max_sndwnd] |
||
207 | shr ebx, 1 |
||
208 | cmp esi, ebx |
||
4387 | hidnplayr | 209 | jae .send |
3545 | hidnplayr | 210 | |
211 | mov ebx, [eax + TCP_SOCKET.SND_NXT] |
||
212 | cmp ebx, [eax + TCP_SOCKET.SND_MAX] |
||
4387 | hidnplayr | 213 | jb .send |
3545 | hidnplayr | 214 | |
215 | .len_zero: |
||
216 | |||
217 | ;---------------------------------------- |
||
6476 | hidnplayr | 218 | ; Check if a window update should be sent |
3545 | hidnplayr | 219 | |
3556 | hidnplayr | 220 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: window=%d\n", ecx |
3545 | hidnplayr | 221 | |
222 | ; Compare available window to amount of window known to peer (as advertised window less next expected input) |
||
223 | ; If the difference is at least two max size segments, or at least 50% of the maximum possible window, |
||
224 | ; Then we want to send a window update to the peer. |
||
225 | |||
226 | test ecx, ecx |
||
227 | jz .no_window |
||
228 | |||
229 | push ecx |
||
230 | mov cl, [eax + TCP_SOCKET.RCV_SCALE] |
||
231 | mov ebx, TCP_max_win |
||
232 | shl ebx, cl |
||
233 | pop ecx |
||
3652 | hidnplayr | 234 | |
3545 | hidnplayr | 235 | cmp ebx, ecx |
236 | jb @f |
||
237 | mov ebx, ecx |
||
238 | @@: |
||
239 | sub ebx, [eax + TCP_SOCKET.RCV_ADV] |
||
240 | add ebx, [eax + TCP_SOCKET.RCV_NXT] |
||
241 | |||
6476 | hidnplayr | 242 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: we can increase window by %d bytes\n", ebx |
243 | |||
3545 | hidnplayr | 244 | mov edi, [eax + TCP_SOCKET.t_maxseg] |
245 | shl edi, 1 |
||
4387 | hidnplayr | 246 | cmp ebx, edi |
247 | jae .send |
||
3545 | hidnplayr | 248 | |
4387 | hidnplayr | 249 | shl ebx, 1 |
3545 | hidnplayr | 250 | ; cmp ebx, [eax + TCP_SOCKET.] ;;; TODO: check with receive buffer high water mark |
251 | ; jae TCP_send |
||
252 | |||
253 | .no_window: |
||
254 | |||
255 | ;-------------------------- |
||
6476 | hidnplayr | 256 | ; Should a segment be sent? |
3545 | hidnplayr | 257 | |
6476 | hidnplayr | 258 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: Should a segment be sent?\n" |
259 | |||
3545 | hidnplayr | 260 | test [eax + TCP_SOCKET.t_flags], TF_ACKNOW ; we need to ACK |
4387 | hidnplayr | 261 | jnz .send |
3545 | hidnplayr | 262 | |
263 | test dl, TH_SYN + TH_RST ; we need to send a SYN or RST |
||
4387 | hidnplayr | 264 | jnz .send |
3545 | hidnplayr | 265 | |
266 | mov ebx, [eax + TCP_SOCKET.SND_UP] ; when urgent pointer is beyond start of send bufer |
||
267 | cmp ebx, [eax + TCP_SOCKET.SND_UNA] |
||
4387 | hidnplayr | 268 | ja .send |
3545 | hidnplayr | 269 | |
6476 | hidnplayr | 270 | ; Do we need to send a FIN according to our state? |
271 | |||
3545 | hidnplayr | 272 | test dl, TH_FIN |
6476 | hidnplayr | 273 | jz .enter_persist ; no reason to send, enter persist state |
3545 | hidnplayr | 274 | |
6476 | hidnplayr | 275 | ; Do so if we didnt do it already |
3545 | hidnplayr | 276 | |
277 | test [eax + TCP_SOCKET.t_flags], TF_SENTFIN |
||
4387 | hidnplayr | 278 | jz .send |
3545 | hidnplayr | 279 | |
6476 | hidnplayr | 280 | ; Or when we need to retransmit the FIN |
281 | |||
3545 | hidnplayr | 282 | mov ebx, [eax + TCP_SOCKET.SND_NXT] |
283 | cmp ebx, [eax + TCP_SOCKET.SND_UNA] |
||
4387 | hidnplayr | 284 | je .send |
3545 | hidnplayr | 285 | |
286 | ;-------------------- |
||
6476 | hidnplayr | 287 | ; Enter persist state |
3545 | hidnplayr | 288 | |
289 | .enter_persist: |
||
290 | |||
3600 | hidnplayr | 291 | cmp [eax + STREAM_SOCKET.snd.size], 0 ; Data ready to send? |
3545 | hidnplayr | 292 | jne @f |
3603 | hidnplayr | 293 | and [eax + TCP_SOCKET.timer_flags], not timer_flag_retransmission |
3545 | hidnplayr | 294 | jne @f |
295 | |||
3603 | hidnplayr | 296 | test [eax + TCP_SOCKET.timer_flags], timer_flag_persist ; Persist timer already expired? |
3600 | hidnplayr | 297 | jnz @f |
298 | |||
3556 | hidnplayr | 299 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: Entering persist state\n" |
3545 | hidnplayr | 300 | |
301 | mov [eax + TCP_SOCKET.t_rxtshift], 0 |
||
6011 | hidnplayr | 302 | call tcp_set_persist |
3545 | hidnplayr | 303 | @@: |
304 | |||
305 | ;---------------------------- |
||
6476 | hidnplayr | 306 | ; No reason to send a segment |
3545 | hidnplayr | 307 | |
3556 | hidnplayr | 308 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_output: No reason to send a segment\n" |
3545 | hidnplayr | 309 | |
310 | pusha |
||
311 | lea ecx, [eax + SOCKET.mutex] |
||
312 | call mutex_unlock |
||
313 | popa |
||
314 | |||
6914 | hidnplayr | 315 | xor eax, eax |
3545 | hidnplayr | 316 | ret |
317 | |||
318 | |||
319 | ;----------------------------------------------- |
||
320 | ; |
||
6476 | hidnplayr | 321 | ; Send a segment |
3545 | hidnplayr | 322 | ; |
323 | ; eax = socket pointer |
||
324 | ; esi = data len |
||
325 | ; dl = flags |
||
326 | ; |
||
327 | ;----------------------------------------------- |
||
4387 | hidnplayr | 328 | .send: |
3545 | hidnplayr | 329 | |
3556 | hidnplayr | 330 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: socket=%x length=%u flags=%x\n", eax, esi, dl |
3545 | hidnplayr | 331 | |
332 | push eax ; save socket ptr |
||
333 | push esi ; and data length too |
||
334 | mov edi, sizeof.TCP_header ; edi will contain headersize |
||
335 | |||
336 | ;------------------------------------ |
||
337 | ; Send options with first SYN segment |
||
338 | |||
339 | test dl, TH_SYN |
||
340 | jz .options_done |
||
341 | |||
342 | push [eax + TCP_SOCKET.ISS] |
||
343 | pop [eax + TCP_SOCKET.SND_NXT] |
||
344 | |||
345 | test [eax + TCP_SOCKET.t_flags], TF_NOOPT |
||
346 | jnz .options_done |
||
347 | |||
348 | mov ecx, 1460 ;;;; FIXME: use routing blablabla to determine MSS |
||
349 | or ecx, TCP_OPT_MAXSEG shl 24 + 4 shl 16 |
||
350 | bswap ecx |
||
351 | push ecx |
||
352 | add di, 4 |
||
353 | |||
3556 | hidnplayr | 354 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: added maxseg option\n" |
3545 | hidnplayr | 355 | |
356 | test [eax + TCP_SOCKET.t_flags], TF_REQ_SCALE |
||
357 | jz .no_scale |
||
358 | |||
359 | test dl, TH_ACK |
||
360 | jz .scale_opt |
||
361 | |||
362 | test [eax + TCP_SOCKET.t_flags], TF_RCVD_SCALE |
||
363 | jz .no_scale |
||
364 | |||
365 | .scale_opt: |
||
366 | mov cl, [eax + TCP_SOCKET.request_r_scale] |
||
367 | mov ch, TCP_OPT_NOP |
||
368 | pushw cx |
||
369 | pushw TCP_OPT_WINDOW + 3 shl 8 |
||
370 | add di, 4 |
||
371 | |||
3556 | hidnplayr | 372 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: added scale option\n" |
3545 | hidnplayr | 373 | |
374 | .no_scale: |
||
375 | .no_syn: |
||
376 | |||
377 | ;------------------------------------ |
||
378 | ; Make the timestamp option if needed |
||
379 | |||
380 | test [eax + TCP_SOCKET.t_flags], TF_REQ_TSTMP |
||
381 | jz .no_timestamp |
||
382 | |||
383 | test dl, TH_RST |
||
384 | jnz .no_timestamp |
||
385 | |||
386 | test dl, TH_ACK |
||
387 | jz .timestamp |
||
388 | |||
389 | test [eax + TCP_SOCKET.t_flags], TF_RCVD_TSTMP |
||
390 | jz .no_timestamp |
||
391 | |||
392 | .timestamp: |
||
393 | pushd 0 |
||
394 | pushd [timer_ticks] |
||
395 | pushd TCP_OPT_NOP + TCP_OPT_NOP shl 8 + TCP_OPT_TIMESTAMP shl 16 + 10 shl 24 |
||
396 | add di, 12 |
||
397 | |||
3556 | hidnplayr | 398 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: added timestamp\n" |
3545 | hidnplayr | 399 | |
400 | .no_timestamp: |
||
401 | |||
402 | ; |
||
403 | |||
404 | .options_done: |
||
405 | |||
406 | ; eax = socket ptr |
||
407 | ; edx = flags |
||
408 | ; edi = header size |
||
409 | ; esi = data len |
||
410 | |||
411 | ;--------------------------------------------- |
||
6476 | hidnplayr | 412 | ; check if we dont exceed the max segment size |
3545 | hidnplayr | 413 | |
414 | add esi, edi ; total TCP segment size |
||
415 | cmp esi, [eax + TCP_SOCKET.t_maxseg] |
||
416 | jbe .no_overflow |
||
417 | |||
418 | mov esi, [eax + TCP_SOCKET.t_maxseg] |
||
4347 | hidnplayr | 419 | or [temp_bits], TCP_BIT_SENDALOT |
3545 | hidnplayr | 420 | .no_overflow: |
421 | |||
5442 | hidnplayr | 422 | ; Update stats |
6476 | hidnplayr | 423 | |
5442 | hidnplayr | 424 | test esi, esi |
425 | jz .zero_data |
||
426 | |||
6916 | hidnplayr | 427 | test [eax + TCP_SOCKET.t_flags], TF_FORCE |
428 | jz @f |
||
5442 | hidnplayr | 429 | cmp esi, 1 |
430 | jne @f |
||
431 | inc [TCPS_sndprobe] |
||
432 | jmp .eos |
||
433 | @@: |
||
434 | |||
435 | mov ebx, [eax + TCP_SOCKET.SND_NXT] |
||
436 | cmp ebx, [eax + TCP_SOCKET.SND_MAX] |
||
437 | jae @f |
||
438 | inc [TCPS_sndrexmitpack] |
||
439 | add [TCPS_sndrexmitbyte], esi |
||
440 | jmp .eos |
||
441 | @@: |
||
442 | inc [TCPS_sndpack] |
||
443 | add [TCPS_sndbyte], esi |
||
444 | jmp .eos |
||
445 | |||
446 | .zero_data: |
||
447 | test [eax + TCP_SOCKET.t_flags], TF_ACKNOW |
||
448 | jz @f |
||
449 | inc [TCPS_sndacks] |
||
450 | jmp .eos |
||
451 | @@: |
||
452 | test dl, TH_SYN + TH_FIN + TH_RST |
||
453 | jz @f |
||
454 | inc [TCPS_sndctrl] |
||
455 | jmp .eos |
||
456 | @@: |
||
457 | mov ebx, [eax + TCP_SOCKET.SND_UP] |
||
458 | cmp ebx, [eax + TCP_SOCKET.SND_UNA] |
||
459 | jb @f |
||
460 | inc [TCPS_sndurg] |
||
461 | jmp .eos |
||
462 | @@: |
||
463 | inc [TCPS_sndwinup] |
||
464 | |||
465 | .eos: |
||
466 | |||
6914 | hidnplayr | 467 | ;--------------------------------------------------- |
468 | ; Dont increase sequence number when resending a FIN |
||
469 | |||
470 | test dl, TH_FIN |
||
471 | jz .no_fin_retransmit |
||
472 | |||
473 | test [eax + TCP_SOCKET.t_flags], TF_SENTFIN |
||
474 | jz .no_fin_retransmit |
||
475 | |||
476 | mov ebx, [eax + TCP_SOCKET.SND_NXT] |
||
477 | cmp ebx, [eax + TCP_SOCKET.SND_MAX] |
||
478 | jne .no_fin_retransmit |
||
479 | |||
480 | dec [eax + TCP_SOCKET.SND_NXT] |
||
481 | |||
482 | .no_fin_retransmit: |
||
483 | |||
4347 | hidnplayr | 484 | ;---------------------------------------------------- |
485 | ; Calculate the receive window. |
||
486 | ; Dont shrink window, but avoid silly window syndrome |
||
487 | |||
6413 | hidnplayr | 488 | mov ebx, SOCKET_BUFFER_SIZE |
4347 | hidnplayr | 489 | sub ebx, [eax + STREAM_SOCKET.rcv.size] |
490 | |||
6413 | hidnplayr | 491 | cmp ebx, SOCKET_BUFFER_SIZE/4 |
4347 | hidnplayr | 492 | jae @f |
493 | cmp ebx, [eax + TCP_SOCKET.t_maxseg] |
||
494 | jae @f |
||
495 | xor ebx, ebx |
||
496 | @@: |
||
497 | |||
7680 | hidnplayr | 498 | cmp ebx, TCP_max_win ;;;; shl rcv_scale |
4347 | hidnplayr | 499 | jbe @f |
7680 | hidnplayr | 500 | mov ebx, TCP_max_win ;;;; shl rcv_scale |
4347 | hidnplayr | 501 | @@: |
502 | |||
503 | mov ecx, [eax + TCP_SOCKET.RCV_ADV] |
||
504 | sub ecx, [eax + TCP_SOCKET.RCV_NXT] |
||
505 | cmp ebx, ecx |
||
6909 | hidnplayr | 506 | jg @f |
4347 | hidnplayr | 507 | mov ebx, ecx |
508 | @@: |
||
509 | |||
6474 | hidnplayr | 510 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: window=%u\n", ebx |
511 | mov [window], ebx |
||
4347 | hidnplayr | 512 | |
513 | mov cl, [eax + TCP_SOCKET.RCV_SCALE] |
||
514 | shr ebx, cl |
||
515 | xchg bl, bh |
||
516 | |||
3545 | hidnplayr | 517 | ;----------------------------------------------------------------- |
518 | ; Start by pushing all TCP header values in reverse order on stack |
||
519 | ; (essentially, creating the tcp header on the stack!) |
||
520 | |||
6476 | hidnplayr | 521 | pushw 0 ; UrgentPointer |
522 | pushw 0 ; Checksum |
||
523 | pushw bx ; Window |
||
524 | shl edi, 2 ; DataOffset |
||
3545 | hidnplayr | 525 | shl dx, 8 |
6476 | hidnplayr | 526 | or dx, di ; Flags |
3545 | hidnplayr | 527 | pushw dx |
6476 | hidnplayr | 528 | shr edi, 2 ; DataOffset |
3545 | hidnplayr | 529 | |
6476 | hidnplayr | 530 | push [eax + TCP_SOCKET.RCV_NXT] ; AckNumber |
3545 | hidnplayr | 531 | ntohd [esp] |
532 | |||
6476 | hidnplayr | 533 | push [eax + TCP_SOCKET.SND_NXT] ; SequenceNumber |
3545 | hidnplayr | 534 | ntohd [esp] |
535 | |||
6476 | hidnplayr | 536 | push [eax + TCP_SOCKET.RemotePort] ; DestinationPort |
537 | push [eax + TCP_SOCKET.LocalPort] ; SourcePort |
||
3545 | hidnplayr | 538 | |
6476 | hidnplayr | 539 | push edi ; header size |
3545 | hidnplayr | 540 | |
541 | ;--------------------- |
||
542 | ; Create the IP packet |
||
543 | |||
544 | mov ecx, esi |
||
5842 | hidnplayr | 545 | mov ebx, [eax + IP_SOCKET.device] |
3545 | hidnplayr | 546 | mov edx, [eax + IP_SOCKET.LocalIP] ; source ip |
5842 | hidnplayr | 547 | mov edi, [eax + IP_SOCKET.RemoteIP] ; dest ip |
548 | mov al, [eax + IP_SOCKET.ttl] |
||
549 | mov ah, IP_PROTO_TCP |
||
6011 | hidnplayr | 550 | call ipv4_output |
3545 | hidnplayr | 551 | jz .ip_error |
552 | |||
5442 | hidnplayr | 553 | ;------------------------------------------ |
554 | ; Move TCP header from stack to TCP segment |
||
3545 | hidnplayr | 555 | |
556 | push ecx |
||
557 | mov ecx, [esp + 4] |
||
558 | lea esi, [esp + 8] |
||
6474 | hidnplayr | 559 | shr ecx, 2 ; count is in bytes, we will work with dwords |
3711 | clevermous | 560 | rep movsd |
6474 | hidnplayr | 561 | pop ecx ; full TCP packet size |
3545 | hidnplayr | 562 | |
6474 | hidnplayr | 563 | pop esi ; headersize |
564 | add esp, esi ; remove it from stack |
||
3545 | hidnplayr | 565 | |
6474 | hidnplayr | 566 | push eax ; packet ptr for send proc |
3545 | hidnplayr | 567 | |
6474 | hidnplayr | 568 | mov edx, edi ; begin of data |
569 | sub edx, esi ; begin of packet (edi = begin of data) |
||
3545 | hidnplayr | 570 | push ecx |
6474 | hidnplayr | 571 | sub ecx, esi ; data size |
3545 | hidnplayr | 572 | |
573 | ;-------------- |
||
574 | ; Copy the data |
||
575 | |||
576 | ; eax = ptr to ring struct |
||
577 | ; ecx = buffer size |
||
578 | ; edi = ptr to buffer |
||
579 | |||
5522 | hidnplayr | 580 | mov eax, [esp + 12] ; get socket ptr |
3545 | hidnplayr | 581 | |
582 | push edx |
||
583 | push [eax + TCP_SOCKET.SND_NXT] ; we'll need this for timing the transmission |
||
584 | test ecx, ecx |
||
585 | jz .nodata |
||
586 | mov edx, [eax + TCP_SOCKET.SND_NXT] |
||
5442 | hidnplayr | 587 | add [eax + TCP_SOCKET.SND_NXT], ecx ; update sequence number |
3545 | hidnplayr | 588 | sub edx, [eax + TCP_SOCKET.SND_UNA] ; offset |
589 | add eax, STREAM_SOCKET.snd |
||
6011 | hidnplayr | 590 | call socket_ring_read |
3545 | hidnplayr | 591 | .nodata: |
592 | pop edi |
||
593 | pop esi ; begin of data |
||
594 | pop ecx ; full packet size |
||
5522 | hidnplayr | 595 | mov eax, [esp + 8] ; socket ptr |
3545 | hidnplayr | 596 | |
6476 | hidnplayr | 597 | ;---------------------------- |
598 | ; initialize retransmit timer |
||
3545 | hidnplayr | 599 | |
600 | ;TODO: check t_force and persist |
||
601 | |||
602 | test [esi + TCP_header.Flags], TH_SYN + TH_FIN ; syn and fin take a sequence number |
||
603 | jz @f |
||
604 | inc [eax + TCP_SOCKET.SND_NXT] |
||
605 | test [esi + TCP_header.Flags], TH_FIN |
||
606 | jz @f |
||
607 | or [eax + TCP_SOCKET.t_flags], TF_SENTFIN ; if we sent a fin, set the sentfin flag |
||
608 | @@: |
||
609 | |||
610 | mov edx, [eax + TCP_SOCKET.SND_NXT] |
||
611 | cmp edx, [eax + TCP_SOCKET.SND_MAX] ; is this a retransmission? |
||
612 | jbe @f |
||
613 | mov [eax + TCP_SOCKET.SND_MAX], edx ; [eax + TCP_SOCKET.SND_NXT] from before we updated it |
||
614 | |||
615 | cmp [eax + TCP_SOCKET.t_rtt], 0 ; are we currently timing anything? |
||
616 | je @f |
||
617 | mov [eax + TCP_SOCKET.t_rtt], 1 ; nope, start transmission timer |
||
618 | mov [eax + TCP_SOCKET.t_rtseq], edi |
||
5442 | hidnplayr | 619 | inc [TCPS_segstimed] |
3545 | hidnplayr | 620 | @@: |
621 | |||
622 | ; set retransmission timer if not already set, and not doing an ACK or keepalive probe |
||
6476 | hidnplayr | 623 | |
3600 | hidnplayr | 624 | test [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission |
625 | jnz .retransmit_set |
||
3545 | hidnplayr | 626 | |
627 | cmp edx, [eax + TCP_SOCKET.SND_UNA] ; edx is still [eax + TCP_SOCKET.SND_NXT] |
||
628 | je .retransmit_set |
||
629 | |||
630 | mov edx, [eax + TCP_SOCKET.t_rxtcur] |
||
631 | mov [eax + TCP_SOCKET.timer_retransmission], edx |
||
3603 | hidnplayr | 632 | or [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission |
3545 | hidnplayr | 633 | |
3600 | hidnplayr | 634 | test [eax + TCP_SOCKET.timer_flags], timer_flag_persist |
635 | jz .retransmit_set |
||
3603 | hidnplayr | 636 | and [eax + TCP_SOCKET.timer_flags], not timer_flag_persist |
3545 | hidnplayr | 637 | mov [eax + TCP_SOCKET.t_rxtshift], 0 |
638 | .retransmit_set: |
||
639 | |||
640 | ;-------------------- |
||
641 | ; Create the checksum |
||
642 | |||
4387 | hidnplayr | 643 | xor dx, dx |
644 | test [ebx + NET_DEVICE.hwacc], NET_HWACC_TCP_IPv4_OUT |
||
645 | jnz .checksum_ok |
||
646 | |||
6011 | hidnplayr | 647 | tcp_checksum (eax + IP_SOCKET.LocalIP), (eax + IP_SOCKET.RemoteIP) |
4387 | hidnplayr | 648 | |
649 | .checksum_ok: |
||
3545 | hidnplayr | 650 | mov [esi + TCP_header.Checksum], dx |
651 | |||
652 | ;---------------- |
||
653 | ; Send the packet |
||
654 | |||
3556 | hidnplayr | 655 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: Sending with device %x\n", ebx |
3545 | hidnplayr | 656 | call [ebx + NET_DEVICE.transmit] |
657 | jnz .send_error |
||
658 | |||
659 | ;--------------- |
||
660 | ; Ok, data sent! |
||
661 | |||
662 | pop ecx |
||
663 | pop eax |
||
664 | |||
6011 | hidnplayr | 665 | call net_ptr_to_num4 |
3644 | hidnplayr | 666 | inc [TCP_segments_tx + edi] |
5442 | hidnplayr | 667 | inc [TCPS_sndtotal] |
3545 | hidnplayr | 668 | |
669 | ; update advertised receive window |
||
6474 | hidnplayr | 670 | |
671 | mov ecx, [window] |
||
3545 | hidnplayr | 672 | test ecx, ecx |
673 | jz @f |
||
674 | add ecx, [eax + TCP_SOCKET.RCV_NXT] |
||
675 | cmp ecx, [eax + TCP_SOCKET.RCV_ADV] |
||
676 | jbe @f |
||
677 | mov [eax + TCP_SOCKET.RCV_ADV], ecx |
||
678 | @@: |
||
679 | |||
680 | ; update last ack sent |
||
6476 | hidnplayr | 681 | |
3545 | hidnplayr | 682 | push [eax + TCP_SOCKET.RCV_NXT] |
683 | pop [eax + TCP_SOCKET.last_ack_sent] |
||
684 | |||
4347 | hidnplayr | 685 | ; clear the ACK flags |
6476 | hidnplayr | 686 | |
3545 | hidnplayr | 687 | and [eax + TCP_SOCKET.t_flags], not (TF_ACKNOW + TF_DELACK) |
688 | |||
689 | ;-------------- |
||
690 | ; unlock socket |
||
691 | |||
3652 | hidnplayr | 692 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: unlocking socket 0x%x\n", eax |
693 | |||
3545 | hidnplayr | 694 | push eax |
695 | lea ecx, [eax + SOCKET.mutex] |
||
696 | call mutex_unlock |
||
697 | pop eax |
||
698 | |||
699 | ;----------------------------- |
||
700 | ; Check if we need more output |
||
701 | |||
4347 | hidnplayr | 702 | test [temp_bits], TCP_BIT_SENDALOT |
6011 | hidnplayr | 703 | jnz tcp_output.again |
3545 | hidnplayr | 704 | |
3556 | hidnplayr | 705 | DEBUGF DEBUG_NETWORK_VERBOSE, "TCP_send: success!\n" |
3545 | hidnplayr | 706 | |
707 | xor eax, eax |
||
708 | ret |
||
709 | |||
710 | |||
711 | .ip_error: |
||
712 | pop ecx |
||
713 | add esp, ecx |
||
714 | add esp, 4 |
||
715 | pop eax |
||
716 | |||
717 | mov [eax + TCP_SOCKET.timer_retransmission], TCP_time_re_min |
||
3602 | hidnplayr | 718 | or [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission |
3545 | hidnplayr | 719 | |
720 | lea ecx, [eax + SOCKET.mutex] |
||
721 | call mutex_unlock |
||
722 | |||
3602 | hidnplayr | 723 | DEBUGF DEBUG_NETWORK_ERROR, "TCP_send: IP error\n" |
3545 | hidnplayr | 724 | |
725 | or eax, -1 |
||
726 | ret |
||
727 | |||
728 | |||
729 | .send_error: |
||
3652 | hidnplayr | 730 | add esp, 4 |
731 | pop eax |
||
3545 | hidnplayr | 732 | |
733 | lea ecx, [eax + SOCKET.mutex] |
||
734 | call mutex_unlock |
||
735 | |||
3602 | hidnplayr | 736 | DEBUGF DEBUG_NETWORK_ERROR, "TCP_send: sending failed\n" |
3545 | hidnplayr | 737 | |
738 | or eax, -2 |
||
739 | ret |
||
740 | |||
741 | |||
4347 | hidnplayr | 742 | endp |