-
Notifications
You must be signed in to change notification settings - Fork 10
/
send_buffer.go
109 lines (89 loc) · 1.97 KB
/
send_buffer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// Copyright 2017 Tim Oster. All rights reserved.
// Use of this source code is governed by the MIT license.
// More information can be found in the LICENSE file.
package rmnp
import "sync"
type sendBufferOP byte
const (
sendBufferDelete sendBufferOP = iota
sendBufferCancel
sendBufferContinue
)
type sendPacket struct {
packet *packet
sendTime int64
noRTT bool
}
type sendBuffer struct {
head *sendBufferElement
tail *sendBufferElement
mutex sync.Mutex
}
type sendBufferElement struct {
next *sendBufferElement
prev *sendBufferElement
data sendPacket
}
func newSendBuffer() *sendBuffer {
return new(sendBuffer)
}
func (buffer *sendBuffer) reset() {
buffer.mutex.Lock()
defer buffer.mutex.Unlock()
buffer.head = nil
buffer.tail = nil
}
func (buffer *sendBuffer) add(packet *packet, noRTT bool) {
buffer.mutex.Lock()
defer buffer.mutex.Unlock()
e := &sendBufferElement{data: sendPacket{
packet: packet,
sendTime: currentTime(),
noRTT: noRTT,
}}
if buffer.head == nil {
buffer.head = e
buffer.tail = e
} else {
e.prev = buffer.tail
buffer.tail.next = e
buffer.tail = e
}
}
func (buffer *sendBuffer) remove(e *sendBufferElement) {
if e.prev == nil {
buffer.head = e.next
} else {
e.prev.next = e.next
}
if e.next == nil {
buffer.tail = e.prev
} else {
e.next.prev = e.prev
}
}
func (buffer *sendBuffer) retrieve(sequence sequenceNumber) (sendPacket, bool) {
buffer.mutex.Lock()
defer buffer.mutex.Unlock()
for e := buffer.head; e != nil; e = e.next {
if e.data.packet.sequence == sequence {
buffer.remove(e)
return e.data, true
}
}
var packet sendPacket
return packet, false
}
func (buffer *sendBuffer) iterate(iterator func(int, *sendPacket) sendBufferOP) {
buffer.mutex.Lock()
defer buffer.mutex.Unlock()
for i, e := 0, buffer.head; e != nil; i, e = i+1, e.next {
switch iterator(i, &e.data) {
case sendBufferDelete:
buffer.remove(e)
case sendBufferCancel:
return
case sendBufferContinue:
}
}
}