purge outgoing segments when closing

LK26 · Nov 1, 2019 · 9313fe2 · 9313fe2
1 parent 62a262c
commit 9313fe2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/kcp.go b/kcp.go
@@ -1051,3 +1051,15 @@ func (kcp *KCP) remove_front(q []segment, n int) []segment {
 	}
 	return q[n:]
 }
+
+// Release all cached outgoing segments
+func (kcp *KCP) ReleaseTX() {
+	for k := range kcp.snd_queue {
+		xmitBuf.Put(kcp.snd_queue[k].data)
+	}
+	for k := range kcp.snd_buf {
+		xmitBuf.Put(kcp.snd_buf[k].data)
+	}
+	kcp.snd_queue = nil
+	kcp.snd_buf = nil
+}
diff --git a/sess.go b/sess.go
@@ -353,6 +353,8 @@ func (s *UDPSession) Close() error {
 		s.mu.Lock()
 		s.kcp.flush(false)
 		s.uncork()
+		// release pending segments
+		s.kcp.ReleaseTX()
 		s.mu.Unlock()
 
 		if s.l != nil { // belongs to listener