mirrors/XTLS_REALITY
0
0
Fork 0
mirror of https://github.com/XTLS/REALITY.git synced 2025-08-22 14:38:35 +00:00
Code

Update tls.go

Browse Source
This commit is contained in:
RPRX 2025-06-08 12:02:51 +00:00 committed by GitHub
parent 931974b633
commit b3dfe09a07
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tls.go
View File

@ -120,22 +120,19 @@ func (c *RatelimitedConn) Read(b []byte) (int, error) {
} }
func NewRatelimitedConn(conn net.Conn, limit *LimitFallback) net.Conn { func NewRatelimitedConn(conn net.Conn, limit *LimitFallback) net.Conn {
bytesPerSec := limit.BytesPerSec if limit.BytesPerSec == 0 {
burstBytesPerSec := limit.BurstBytesPerSec
afterBytes := limit.AfterBytes
if bytesPerSec == 0 {
return conn return conn
} }
if burstBytesPerSec < bytesPerSec { burstBytesPerSec := limit.BurstBytesPerSec
burstBytesPerSec = bytesPerSec if burstBytesPerSec < limit.BytesPerSec {
burstBytesPerSec = limit.BytesPerSec
} }
return &RatelimitedConn{ return &RatelimitedConn{
Conn: conn, Conn: conn,
Bucket: ratelimit.NewBucketWithRate(float64(bytesPerSec), int64(burstBytesPerSec)), Bucket: ratelimit.NewBucketWithRate(float64(limit.BytesPerSec), int64(burstBytesPerSec)),
LimitAfter: int64(afterBytes), LimitAfter: int64(limit.AfterBytes),
} }
} }
@ -267,7 +264,6 @@ func Server(ctx context.Context, conn net.Conn, config *Config) (*Conn, error) {
if config.Show && hs.clientHello != nil { if config.Show && hs.clientHello != nil {
fmt.Printf("REALITY remoteAddr: %v\tforwarded SNI: %v\n", remoteAddr, hs.clientHello.serverName) fmt.Printf("REALITY remoteAddr: %v\tforwarded SNI: %v\n", remoteAddr, hs.clientHello.serverName)
} }
// Limit upload speed for fallback connection
io.Copy(target, NewRatelimitedConn(underlying, &config.LimitFallbackUpload)) io.Copy(target, NewRatelimitedConn(underlying, &config.LimitFallbackUpload))
} }
waitGroup.Done() waitGroup.Done()
@ -399,13 +395,11 @@ func Server(ctx context.Context, conn net.Conn, config *Config) (*Conn, error) {
if hs.c.conn == conn { // if we processed the Client Hello successfully but the target did not if hs.c.conn == conn { // if we processed the Client Hello successfully but the target did not
waitGroup.Add(1) waitGroup.Add(1)
go func() { go func() {
// Limit upload speed for fallback connection (handshake ok but hello failed)
io.Copy(target, NewRatelimitedConn(underlying, &config.LimitFallbackUpload)) io.Copy(target, NewRatelimitedConn(underlying, &config.LimitFallbackUpload))
waitGroup.Done() waitGroup.Done()
}() }()
} }
conn.Write(s2cSaved) conn.Write(s2cSaved)
// Limit download speed for fallback connection
io.Copy(underlying, NewRatelimitedConn(target, &config.LimitFallbackDownload)) io.Copy(underlying, NewRatelimitedConn(target, &config.LimitFallbackDownload))
// Here is bidirectional direct forwarding: // Here is bidirectional direct forwarding:
// client ---underlying--- server ---target--- dest // client ---underlying--- server ---target--- dest