correctly release segments

2016-07-13 23:15:40 +02:00 · 2016-07-13 23:15:40 +02:00 · 05ed05b9a0
parent 73afe9a001
commit 05ed05b9a0
2 changed files with 3 additions and 0 deletions
--- a/transport/internet/kcp/connection.go
+++ b/transport/internet/kcp/connection.go
@ -414,6 +414,7 @@ func (this *Connection) Input(data []byte) int {
 			}
 			this.sendingWorker.ProcessReceivingNext(seg.ReceivinNext)
 			this.receivingWorker.ProcessSendingNext(seg.SendingNext)
+			seg.Release()
 		default:
 		}
 	}
--- a/transport/internet/kcp/sending.go
+++ b/transport/internet/kcp/sending.go
@ -292,6 +292,8 @@ func (this *SendingWorker) ProcessAck(number uint32) {
 }

 func (this *SendingWorker) ProcessSegment(current uint32, seg *AckSegment) {
+	defer seg.Release()
+
 	this.Lock()
 	defer this.Unlock()