From 09398989b6fe517d5036443ca3edfcbef5a9bf63 Mon Sep 17 00:00:00 2001 From: Darien Raymond Date: Tue, 18 Apr 2017 21:31:14 +0200 Subject: [PATCH] fix mss calculation --- transport/internet/kcp/connection.go | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/transport/internet/kcp/connection.go b/transport/internet/kcp/connection.go index 933c0dff..aa82d61c 100644 --- a/transport/internet/kcp/connection.go +++ b/transport/internet/kcp/connection.go @@ -334,19 +334,14 @@ func (v *Connection) Write(b []byte) (int, error) { func (c *Connection) WriteMultiBuffer(mb buf.MultiBuffer) (int, error) { defer mb.Release() - if len(mb) == 0 { - return 0, nil - } - if len(mb) == 1 { - return c.Write(mb[0].Bytes()) - } - buffer := buf.New() defer buffer.Release() totalBytes := 0 for !mb.IsEmpty() { - buffer.Reset(buf.ReadFrom(&mb)) + buffer.Reset(func(b []byte) (int, error) { + return mb.Read(b[:c.mss]) + }) nBytes, err := c.Write(buffer.Bytes()) totalBytes += nBytes if err != nil {