mirror of
https://github.com/SagerNet/sing-quic.git
synced 2025-04-03 03:47:39 +03:00
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package tuic
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/sagernet/quic-go"
|
|
"github.com/sagernet/quic-go/congestion"
|
|
congestion_meta1 "github.com/sagernet/sing-quic/congestion_meta1"
|
|
congestion_meta2 "github.com/sagernet/sing-quic/congestion_meta2"
|
|
"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.ByteCount(connection.Config().InitialPacketSize),
|
|
false,
|
|
nil,
|
|
),
|
|
)
|
|
case "new_reno":
|
|
connection.SetCongestionControl(
|
|
congestion_meta1.NewCubicSender(
|
|
congestion_meta1.DefaultClock{TimeFunc: timeFunc},
|
|
congestion.ByteCount(connection.Config().InitialPacketSize),
|
|
true,
|
|
nil,
|
|
),
|
|
)
|
|
case "bbr_meta_v1":
|
|
connection.SetCongestionControl(congestion_meta1.NewBBRSender(
|
|
congestion_meta1.DefaultClock{TimeFunc: timeFunc},
|
|
congestion.ByteCount(connection.Config().InitialPacketSize),
|
|
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.ByteCount(connection.Config().InitialPacketSize),
|
|
congestion.ByteCount(congestion_meta1.InitialCongestionWindow),
|
|
))
|
|
}
|
|
}
|