diff --git a/outbound/default.go b/outbound/default.go index 9569d6ab..249ab8c2 100644 --- a/outbound/default.go +++ b/outbound/default.go @@ -162,6 +162,7 @@ func CopyEarlyConn(ctx context.Context, conn net.Conn, serverConn net.Conn) erro payload := cachedReader.ReadCached() if payload != nil && !payload.IsEmpty() { _, err := serverConn.Write(payload.Bytes()) + payload.Release() if err != nil { return err } @@ -173,10 +174,12 @@ func CopyEarlyConn(ctx context.Context, conn net.Conn, serverConn net.Conn) erro err := conn.SetReadDeadline(time.Now().Add(C.ReadPayloadTimeout)) if err != os.ErrInvalid { if err != nil { + payload.Release() return err } _, err = payload.ReadOnceFrom(conn) if err != nil && !E.IsTimeout(err) { + payload.Release() return E.Cause(err, "read payload") } err = conn.SetReadDeadline(time.Time{}) @@ -186,10 +189,10 @@ func CopyEarlyConn(ctx context.Context, conn net.Conn, serverConn net.Conn) erro } } _, err = serverConn.Write(payload.Bytes()) + payload.Release() if err != nil { return N.HandshakeFailure(conn, err) } - payload.Release() } return bufio.CopyConn(ctx, conn, serverConn) }