forked from SagerNet/sing-quic
/
congestion.go
52 lines (49 loc) · 1.61 KB
/
congestion.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
package tuic
import (
"context"
"time"
congestion_meta1 "github.com/Alex-Joe-Lee/sing-quic/congestion_meta1"
congestion_meta2 "github.com/Alex-Joe-Lee/sing-quic/congestion_meta2"
"github.com/sagernet/quic-go"
"github.com/sagernet/quic-go/congestion"
"github.com/sagernet/sing/common/ntp"
)
func setCongestion(ctx context.Context, connection quic.Connection, congestionName string) {
timeFunc := ntp.TimeFuncFromContext(ctx)
if timeFunc == nil {
timeFunc = time.Now
}
switch congestionName {
case "cubic":
connection.SetCongestionControl(
congestion_meta1.NewCubicSender(
congestion_meta1.DefaultClock{TimeFunc: timeFunc},
congestion_meta1.GetInitialPacketSize(connection.RemoteAddr()),
false,
nil,
),
)
case "new_reno":
connection.SetCongestionControl(
congestion_meta1.NewCubicSender(
congestion_meta1.DefaultClock{TimeFunc: timeFunc},
congestion_meta1.GetInitialPacketSize(connection.RemoteAddr()),
true,
nil,
),
)
case "bbr_meta_v1":
connection.SetCongestionControl(congestion_meta1.NewBBRSender(
congestion_meta1.DefaultClock{TimeFunc: timeFunc},
congestion_meta1.GetInitialPacketSize(connection.RemoteAddr()),
congestion_meta1.InitialCongestionWindow*congestion_meta1.InitialMaxDatagramSize,
congestion_meta1.DefaultBBRMaxCongestionWindow*congestion_meta1.InitialMaxDatagramSize,
))
case "bbr":
connection.SetCongestionControl(congestion_meta2.NewBbrSender(
congestion_meta2.DefaultClock{TimeFunc: timeFunc},
congestion_meta2.GetInitialPacketSize(connection.RemoteAddr()),
congestion.ByteCount(congestion_meta1.InitialCongestionWindow),
))
}
}