hesoyam/Xray-core
1
0
Fork 0
mirror of https://github.com/XTLS/Xray-core.git synced 2026-10-22 17:24:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

SS2022: Correct interrupt behavior and UDP conn change

Browse Source 
Co-authored-by: ChownYourLife <260039878+ChownYourLife@users.noreply.github.com>
This commit is contained in:
Fangliding
2026-10-14 02:12:09 +08:00
co-authored by ChownYourLife
parent 701af60772
commit 2f24328df4
3 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/shadowsocks_2022/stream.go
+6 -2
View File
@@ -11,6 +11,7 @@ import (
"sync"
"time"
"github.com/xtls/xray-core/common"
"github.com/xtls/xray-core/common/antireplay"
"github.com/xtls/xray-core/common/buf"
"github.com/xtls/xray-core/common/errors"
@@ -629,6 +630,9 @@ func InitServerStream(conn net.Conn, method *CipherMethod, psk, saltSlice []byte
func TransportTCP(ctx context.Context, sessionPolicy policy.Session, reader buf.Reader, writer buf.Writer, link *transport.Link) error {
ctx, cancel := context.WithCancel(ctx)
timer := signal.CancelAfterInactivity(ctx, cancel, sessionPolicy.Timeouts.ConnectionIdle)
defer timer.SetTimeout(0)
defer common.Interrupt(link.Reader)
defer common.Interrupt(link.Writer)
ctx = policy.ContextWithBufferPolicy(ctx, sessionPolicy.Buffer)
requestDone := func() error {
@@ -644,6 +648,6 @@ func TransportTCP(ctx context.Context, sessionPolicy policy.Session, reader buf.
return buf.Copy(link.Reader, writer, buf.UpdateActivity(timer))
}
responseDoneAndCloseWriter := task.OnSuccess(responseDone, task.Close(link.Writer))
return task.Run(ctx, requestDone, responseDoneAndCloseWriter)
requestDonePost := task.OnSuccess(requestDone, task.Close(link.Writer))
return task.Run(ctx, requestDonePost, responseDone)
}