mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-28 14:55:37 +08:00
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package tuic
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/sagernet/quic-go"
|
|
"github.com/sagernet/sing-box/transport/tuic/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.NewCubicSender(
|
|
congestion.DefaultClock{TimeFunc: timeFunc},
|
|
congestion.GetInitialPacketSize(connection.RemoteAddr()),
|
|
false,
|
|
nil,
|
|
),
|
|
)
|
|
case "new_reno":
|
|
connection.SetCongestionControl(
|
|
congestion.NewCubicSender(
|
|
congestion.DefaultClock{TimeFunc: timeFunc},
|
|
congestion.GetInitialPacketSize(connection.RemoteAddr()),
|
|
true,
|
|
nil,
|
|
),
|
|
)
|
|
case "bbr":
|
|
connection.SetCongestionControl(
|
|
congestion.NewBBRSender(
|
|
congestion.DefaultClock{},
|
|
congestion.GetInitialPacketSize(connection.RemoteAddr()),
|
|
congestion.InitialCongestionWindow*congestion.InitialMaxDatagramSize,
|
|
congestion.DefaultBBRMaxCongestionWindow*congestion.InitialMaxDatagramSize,
|
|
),
|
|
)
|
|
}
|
|
}
|