Send shadowsocks handshake with payload if available

v2fly · Sep 25, 2021 · 543c799 · 543c799
1 parent c278db5
commit 543c799
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/proxy/shadowsocks/client.go b/proxy/shadowsocks/client.go
@@ -5,6 +5,7 @@ package shadowsocks
 
 import (
 	"context"
+	"time"
 
 	core "github.com/v2fly/v2ray-core/v4"
 	"github.com/v2fly/v2ray-core/v4/common"
@@ -108,6 +109,10 @@ func (c *Client) Process(ctx context.Context, link *transport.Link, dialer inter
 			return newError("failed to write request").Base(err)
 		}
 
+		if err = buf.CopyOnceTimeout(link.Reader, bodyWriter, time.Millisecond*100); err != nil && err != buf.ErrNotTimeoutReader && err != buf.ErrReadTimeout {
+			return newError("failed to write A request payload").Base(err).AtWarning()
+		}
+
 		if err := bufferedWriter.SetBuffered(false); err != nil {
 			return err
 		}