Clear sending queue when peer closed

v2ray · Jun 23, 2016 · 0613d0c · 0613d0c
1 parent ab710f2
commit 0613d0c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/transport/internet/kcp/connection.go b/transport/internet/kcp/connection.go
@@ -358,6 +358,9 @@ func (this *Connection) MarkPeerClose() {
 	if this.state == ConnStateActive {
 		this.state = ConnStatePeerClosed
 	}
+	this.kcpAccess.Lock()
+	this.kcp.ClearSendQueue()
+	this.kcpAccess.Unlock()
 }
 
 func (this *Connection) kcpInput(data []byte) {

diff --git a/transport/internet/kcp/kcp.go b/transport/internet/kcp/kcp.go
@@ -842,3 +842,16 @@ func (kcp *KCP) WndSize(sndwnd, rcvwnd int) int {
 func (kcp *KCP) WaitSnd() int {
 	return len(kcp.snd_buf) + len(kcp.snd_queue)
 }
+
+func (this *KCP) ClearSendQueue() {
+	for _, seg := range this.snd_queue {
+		seg.Release()
+	}
+	this.snd_queue = nil
+
+	for _, seg := range this.snd_buf {
+		seg.Release()
+	}
+
+	this.snd_buf = nil
+}