Subversion Repositories Kolibri OS

Rev

Rev 6474 | Rev 6909 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
3545 hidnplayr 1
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2
;;                                                                 ;;
6474 hidnplayr 3
;; Copyright (C) KolibriOS team 2004-2016. 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: 6476 $
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
 
93
        cmp     [eax + TCP_SOCKET.t_force], 0
94
        je      .no_force
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
 
203
        test    [eax + TCP_SOCKET.t_force], -1  ;;;
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
 
315
; Fixme: returnvalue?
316
 
317
        ret
318
 
319
 
320
;-----------------------------------------------
321
;
6476 hidnplayr 322
; Send a segment
3545 hidnplayr 323
;
324
; eax = socket pointer
325
; esi = data len
326
;  dl = flags
327
;
328
;-----------------------------------------------
4387 hidnplayr 329
  .send:
3545 hidnplayr 330
 
3556 hidnplayr 331
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: socket=%x length=%u flags=%x\n", eax, esi, dl
3545 hidnplayr 332
 
333
        push    eax                     ; save socket ptr
334
        push    esi                     ; and data length too
335
        mov     edi, sizeof.TCP_header  ; edi will contain headersize
336
 
337
;------------------------------------
338
; Send options with first SYN segment
339
 
340
        test    dl, TH_SYN
341
        jz      .options_done
342
 
343
        push    [eax + TCP_SOCKET.ISS]
344
        pop     [eax + TCP_SOCKET.SND_NXT]
345
 
346
        test    [eax + TCP_SOCKET.t_flags], TF_NOOPT
347
        jnz     .options_done
348
 
349
        mov     ecx, 1460                              ;;;; FIXME: use routing blablabla to determine MSS
350
        or      ecx, TCP_OPT_MAXSEG shl 24 + 4 shl 16
351
        bswap   ecx
352
        push    ecx
353
        add     di, 4
354
 
3556 hidnplayr 355
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: added maxseg option\n"
3545 hidnplayr 356
 
357
        test    [eax + TCP_SOCKET.t_flags], TF_REQ_SCALE
358
        jz      .no_scale
359
 
360
        test    dl, TH_ACK
361
        jz      .scale_opt
362
 
363
        test    [eax + TCP_SOCKET.t_flags], TF_RCVD_SCALE
364
        jz      .no_scale
365
 
366
  .scale_opt:
367
        mov     cl, [eax + TCP_SOCKET.request_r_scale]
368
        mov     ch, TCP_OPT_NOP
369
        pushw   cx
370
        pushw   TCP_OPT_WINDOW + 3 shl 8
371
        add     di, 4
372
 
3556 hidnplayr 373
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: added scale option\n"
3545 hidnplayr 374
 
375
  .no_scale:
376
  .no_syn:
377
 
378
;------------------------------------
379
; Make the timestamp option if needed
380
 
381
        test    [eax + TCP_SOCKET.t_flags], TF_REQ_TSTMP
382
        jz      .no_timestamp
383
 
384
        test    dl, TH_RST
385
        jnz     .no_timestamp
386
 
387
        test    dl, TH_ACK
388
        jz      .timestamp
389
 
390
        test    [eax + TCP_SOCKET.t_flags], TF_RCVD_TSTMP
391
        jz      .no_timestamp
392
 
393
  .timestamp:
394
        pushd   0
395
        pushd   [timer_ticks]
396
        pushd   TCP_OPT_NOP + TCP_OPT_NOP shl 8 + TCP_OPT_TIMESTAMP shl 16 + 10 shl 24
397
        add     di, 12
398
 
3556 hidnplayr 399
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: added timestamp\n"
3545 hidnplayr 400
 
401
  .no_timestamp:
402
 
403
        ; 
404
 
405
  .options_done:
406
 
407
; eax = socket ptr
408
; edx = flags
409
; edi = header size
410
; esi = data len
411
 
412
;---------------------------------------------
6476 hidnplayr 413
; check if we dont exceed the max segment size
3545 hidnplayr 414
 
415
        add     esi, edi                        ; total TCP segment size
416
        cmp     esi, [eax + TCP_SOCKET.t_maxseg]
417
        jbe     .no_overflow
418
 
419
        mov     esi, [eax + TCP_SOCKET.t_maxseg]
4347 hidnplayr 420
        or      [temp_bits], TCP_BIT_SENDALOT
3545 hidnplayr 421
  .no_overflow:
422
 
5442 hidnplayr 423
; Update stats
6476 hidnplayr 424
 
5442 hidnplayr 425
        test    esi, esi
426
        jz      .zero_data
427
 
428
        cmp     [eax + TCP_SOCKET.t_force], 1
429
        jne     @f
430
        cmp     esi, 1
431
        jne     @f
432
        inc     [TCPS_sndprobe]
433
        jmp     .eos
434
  @@:
435
 
436
        mov     ebx, [eax + TCP_SOCKET.SND_NXT]
437
        cmp     ebx, [eax + TCP_SOCKET.SND_MAX]
438
        jae     @f
439
        inc     [TCPS_sndrexmitpack]
440
        add     [TCPS_sndrexmitbyte], esi
441
        jmp     .eos
442
  @@:
443
        inc     [TCPS_sndpack]
444
        add     [TCPS_sndbyte], esi
445
        jmp     .eos
446
 
447
  .zero_data:
448
        test    [eax + TCP_SOCKET.t_flags], TF_ACKNOW
449
        jz      @f
450
        inc     [TCPS_sndacks]
451
        jmp     .eos
452
  @@:
453
        test    dl, TH_SYN + TH_FIN + TH_RST
454
        jz      @f
455
        inc     [TCPS_sndctrl]
456
        jmp     .eos
457
  @@:
458
        mov     ebx, [eax + TCP_SOCKET.SND_UP]
459
        cmp     ebx, [eax + TCP_SOCKET.SND_UNA]
460
        jb      @f
461
        inc     [TCPS_sndurg]
462
        jmp     .eos
463
  @@:
464
        inc     [TCPS_sndwinup]
465
 
466
  .eos:
467
 
4347 hidnplayr 468
;----------------------------------------------------
469
; Calculate the receive window.
470
; Dont shrink window, but avoid silly window syndrome
471
 
6413 hidnplayr 472
        mov     ebx, SOCKET_BUFFER_SIZE
4347 hidnplayr 473
        sub     ebx, [eax + STREAM_SOCKET.rcv.size]
474
 
6413 hidnplayr 475
        cmp     ebx, SOCKET_BUFFER_SIZE/4
4347 hidnplayr 476
        jae     @f
477
        cmp     ebx, [eax + TCP_SOCKET.t_maxseg]
478
        jae     @f
479
        xor     ebx, ebx
480
  @@:
481
 
482
        cmp     ebx, TCP_max_win
483
        jbe     @f
484
        mov     ebx, TCP_max_win
485
  @@:
486
 
487
        mov     ecx, [eax + TCP_SOCKET.RCV_ADV]
488
        sub     ecx, [eax + TCP_SOCKET.RCV_NXT]
489
        cmp     ebx, ecx
490
        ja      @f
491
        mov     ebx, ecx
492
  @@:
493
 
6474 hidnplayr 494
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: window=%u\n", ebx
495
        mov     [window], ebx
4347 hidnplayr 496
 
497
        mov     cl, [eax + TCP_SOCKET.RCV_SCALE]
498
        shr     ebx, cl
499
        xchg    bl, bh
500
 
3545 hidnplayr 501
;-----------------------------------------------------------------
502
; Start by pushing all TCP header values in reverse order on stack
503
; (essentially, creating the tcp header on the stack!)
504
 
6476 hidnplayr 505
        pushw   0                               ; UrgentPointer
506
        pushw   0                               ; Checksum
507
        pushw   bx                              ; Window
508
        shl     edi, 2                          ; DataOffset
3545 hidnplayr 509
        shl     dx, 8
6476 hidnplayr 510
        or      dx, di                          ; Flags
3545 hidnplayr 511
        pushw   dx
6476 hidnplayr 512
        shr     edi, 2                          ; DataOffset
3545 hidnplayr 513
 
6476 hidnplayr 514
        push    [eax + TCP_SOCKET.RCV_NXT]      ; AckNumber
3545 hidnplayr 515
        ntohd   [esp]
516
 
6476 hidnplayr 517
        push    [eax + TCP_SOCKET.SND_NXT]      ; SequenceNumber
3545 hidnplayr 518
        ntohd   [esp]
519
 
6476 hidnplayr 520
        push    [eax + TCP_SOCKET.RemotePort]   ; DestinationPort
521
        push    [eax + TCP_SOCKET.LocalPort]    ; SourcePort
3545 hidnplayr 522
 
6476 hidnplayr 523
        push    edi                             ; header size
3545 hidnplayr 524
 
525
;---------------------
526
; Create the IP packet
527
 
528
        mov     ecx, esi
5842 hidnplayr 529
        mov     ebx, [eax + IP_SOCKET.device]
3545 hidnplayr 530
        mov     edx, [eax + IP_SOCKET.LocalIP]  ; source ip
5842 hidnplayr 531
        mov     edi, [eax + IP_SOCKET.RemoteIP] ; dest ip
532
        mov     al, [eax + IP_SOCKET.ttl]
533
        mov     ah, IP_PROTO_TCP
6011 hidnplayr 534
        call    ipv4_output
3545 hidnplayr 535
        jz      .ip_error
536
 
5442 hidnplayr 537
;------------------------------------------
538
; Move TCP header from stack to TCP segment
3545 hidnplayr 539
 
540
        push    ecx
541
        mov     ecx, [esp + 4]
542
        lea     esi, [esp + 8]
6474 hidnplayr 543
        shr     ecx, 2                          ; count is in bytes, we will work with dwords
3711 clevermous 544
        rep movsd
6474 hidnplayr 545
        pop     ecx                             ; full TCP packet size
3545 hidnplayr 546
 
6474 hidnplayr 547
        pop     esi                             ; headersize
548
        add     esp, esi                        ; remove it from stack
3545 hidnplayr 549
 
6474 hidnplayr 550
        push    eax                             ; packet ptr for send proc
3545 hidnplayr 551
 
6474 hidnplayr 552
        mov     edx, edi                        ; begin of data
553
        sub     edx, esi                        ; begin of packet (edi = begin of data)
3545 hidnplayr 554
        push    ecx
6474 hidnplayr 555
        sub     ecx, esi                        ; data size
3545 hidnplayr 556
 
557
;--------------
558
; Copy the data
559
 
560
; eax = ptr to ring struct
561
; ecx = buffer size
562
; edi = ptr to buffer
563
 
5522 hidnplayr 564
        mov     eax, [esp + 12]                 ; get socket ptr
3545 hidnplayr 565
 
566
        push    edx
567
        push    [eax + TCP_SOCKET.SND_NXT]      ; we'll need this for timing the transmission
568
        test    ecx, ecx
569
        jz      .nodata
570
        mov     edx, [eax + TCP_SOCKET.SND_NXT]
5442 hidnplayr 571
        add     [eax + TCP_SOCKET.SND_NXT], ecx ; update sequence number
3545 hidnplayr 572
        sub     edx, [eax + TCP_SOCKET.SND_UNA] ; offset
573
        add     eax, STREAM_SOCKET.snd
6011 hidnplayr 574
        call    socket_ring_read
3545 hidnplayr 575
  .nodata:
576
        pop     edi
577
        pop     esi                             ; begin of data
578
        pop     ecx                             ; full packet size
5522 hidnplayr 579
        mov     eax, [esp + 8]                  ; socket ptr
3545 hidnplayr 580
 
6476 hidnplayr 581
;----------------------------
582
; initialize retransmit timer
3545 hidnplayr 583
 
584
;TODO: check t_force and persist
585
 
586
        test    [esi + TCP_header.Flags], TH_SYN + TH_FIN       ; syn and fin take a sequence number
587
        jz      @f
588
        inc     [eax + TCP_SOCKET.SND_NXT]
589
        test    [esi + TCP_header.Flags], TH_FIN
590
        jz      @f
591
        or      [eax + TCP_SOCKET.t_flags], TF_SENTFIN          ; if we sent a fin, set the sentfin flag
592
       @@:
593
 
594
        mov     edx, [eax + TCP_SOCKET.SND_NXT]
595
        cmp     edx, [eax + TCP_SOCKET.SND_MAX]                 ; is this a retransmission?
596
        jbe     @f
597
        mov     [eax + TCP_SOCKET.SND_MAX], edx                 ; [eax + TCP_SOCKET.SND_NXT] from before we updated it
598
 
599
        cmp     [eax + TCP_SOCKET.t_rtt], 0                     ; are we currently timing anything?
600
        je      @f
601
        mov     [eax + TCP_SOCKET.t_rtt], 1                     ; nope, start transmission timer
602
        mov     [eax + TCP_SOCKET.t_rtseq], edi
5442 hidnplayr 603
        inc     [TCPS_segstimed]
3545 hidnplayr 604
       @@:
605
 
606
; set retransmission timer if not already set, and not doing an ACK or keepalive probe
6476 hidnplayr 607
 
3600 hidnplayr 608
        test    [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission
609
        jnz     .retransmit_set
3545 hidnplayr 610
 
611
        cmp     edx, [eax + TCP_SOCKET.SND_UNA]                 ; edx is still [eax + TCP_SOCKET.SND_NXT]
612
        je      .retransmit_set
613
 
614
        mov     edx, [eax + TCP_SOCKET.t_rxtcur]
615
        mov     [eax + TCP_SOCKET.timer_retransmission], edx
3603 hidnplayr 616
        or      [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission
3545 hidnplayr 617
 
3600 hidnplayr 618
        test    [eax + TCP_SOCKET.timer_flags], timer_flag_persist
619
        jz      .retransmit_set
3603 hidnplayr 620
        and     [eax + TCP_SOCKET.timer_flags], not timer_flag_persist
3545 hidnplayr 621
        mov     [eax + TCP_SOCKET.t_rxtshift], 0
622
  .retransmit_set:
623
 
624
;--------------------
625
; Create the checksum
626
 
4387 hidnplayr 627
        xor     dx, dx
628
        test    [ebx + NET_DEVICE.hwacc], NET_HWACC_TCP_IPv4_OUT
629
        jnz     .checksum_ok
630
 
6011 hidnplayr 631
        tcp_checksum (eax + IP_SOCKET.LocalIP), (eax + IP_SOCKET.RemoteIP)
4387 hidnplayr 632
 
633
  .checksum_ok:
3545 hidnplayr 634
        mov     [esi + TCP_header.Checksum], dx
635
 
636
;----------------
637
; Send the packet
638
 
3556 hidnplayr 639
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: Sending with device %x\n", ebx
3545 hidnplayr 640
        call    [ebx + NET_DEVICE.transmit]
641
        jnz     .send_error
642
 
643
;---------------
644
; Ok, data sent!
645
 
646
        pop     ecx
647
        pop     eax
648
 
6011 hidnplayr 649
        call    net_ptr_to_num4
3644 hidnplayr 650
        inc     [TCP_segments_tx + edi]
5442 hidnplayr 651
        inc     [TCPS_sndtotal]
3545 hidnplayr 652
 
653
; update advertised receive window
6474 hidnplayr 654
 
655
        mov     ecx, [window]
3545 hidnplayr 656
        test    ecx, ecx
657
        jz      @f
658
        add     ecx, [eax + TCP_SOCKET.RCV_NXT]
659
        cmp     ecx, [eax + TCP_SOCKET.RCV_ADV]
660
        jbe     @f
661
        mov     [eax + TCP_SOCKET.RCV_ADV], ecx
662
       @@:
663
 
664
; update last ack sent
6476 hidnplayr 665
 
3545 hidnplayr 666
        push    [eax + TCP_SOCKET.RCV_NXT]
667
        pop     [eax + TCP_SOCKET.last_ack_sent]
668
 
4347 hidnplayr 669
; clear the ACK flags
6476 hidnplayr 670
 
3545 hidnplayr 671
        and     [eax + TCP_SOCKET.t_flags], not (TF_ACKNOW + TF_DELACK)
672
 
673
;--------------
674
; unlock socket
675
 
3652 hidnplayr 676
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: unlocking socket 0x%x\n", eax
677
 
3545 hidnplayr 678
        push    eax
679
        lea     ecx, [eax + SOCKET.mutex]
680
        call    mutex_unlock
681
        pop     eax
682
 
683
;-----------------------------
684
; Check if we need more output
685
 
4347 hidnplayr 686
        test    [temp_bits], TCP_BIT_SENDALOT
6011 hidnplayr 687
        jnz     tcp_output.again
3545 hidnplayr 688
 
3556 hidnplayr 689
        DEBUGF  DEBUG_NETWORK_VERBOSE, "TCP_send: success!\n"
3545 hidnplayr 690
 
691
        xor     eax, eax
692
        ret
693
 
694
 
695
  .ip_error:
696
        pop     ecx
697
        add     esp, ecx
698
        add     esp, 4
699
        pop     eax
700
 
701
        mov     [eax + TCP_SOCKET.timer_retransmission], TCP_time_re_min
3602 hidnplayr 702
        or      [eax + TCP_SOCKET.timer_flags], timer_flag_retransmission
3545 hidnplayr 703
 
704
        lea     ecx, [eax + SOCKET.mutex]
705
        call    mutex_unlock
706
 
3602 hidnplayr 707
        DEBUGF  DEBUG_NETWORK_ERROR, "TCP_send: IP error\n"
3545 hidnplayr 708
 
709
        or      eax, -1
710
        ret
711
 
712
 
713
  .send_error:
3652 hidnplayr 714
        add     esp, 4
715
        pop     eax
3545 hidnplayr 716
 
717
        lea     ecx, [eax + SOCKET.mutex]
718
        call    mutex_unlock
719
 
3602 hidnplayr 720
        DEBUGF  DEBUG_NETWORK_ERROR, "TCP_send: sending failed\n"
3545 hidnplayr 721
 
722
        or      eax, -2
723
        ret
724
 
725
 
4347 hidnplayr 726
endp