From 0caf2e6d30c18b0b61c2a54a16568b96be274646 Mon Sep 17 00:00:00 2001
From: v2ray <admin@v2ray.com>
Date: Sun, 7 Aug 2016 01:01:44 +0200
Subject: [PATCH] compact kcp stream

---
 transport/internet/kcp/sending.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/transport/internet/kcp/sending.go b/transport/internet/kcp/sending.go
index 789f29b3..e1136af5 100644
--- a/transport/internet/kcp/sending.go
+++ b/transport/internet/kcp/sending.go
@@ -212,6 +212,13 @@ func (this *SendingQueue) Pop() *alloc.Buffer {
 	return seg
 }
 
+func (this *SendingQueue) Last() *alloc.Buffer {
+	if this.IsEmpty() {
+		return nil
+	}
+	return this.list[(this.start+this.len-1+this.cap)%this.cap]
+}
+
 func (this *SendingQueue) Push(seg *alloc.Buffer) {
 	if this.IsFull() {
 		return
@@ -339,6 +346,18 @@ func (this *SendingWorker) Push(b []byte) int {
 	nBytes := 0
 	this.Lock()
 	defer this.Unlock()
+	if !this.queue.IsEmpty() {
+		lastSeg := this.queue.Last()
+		if lastSeg.Len() < int(this.conn.mss) {
+			delta := int(this.conn.mss) - lastSeg.Len()
+			if delta > len(b) {
+				delta = len(b)
+			}
+			lastSeg.Append(b[:delta])
+			b = b[delta:]
+			nBytes += delta
+		}
+	}
 	for len(b) > 0 && !this.queue.IsFull() {
 		var size int
 		if len(b) > int(this.conn.mss) {