Browse Source

Fix: don't skip fragmentor notice in early exit success case

Rod Hynes 6 years ago
parent
commit
bf809b97a3
1 changed files with 5 additions and 1 deletions
  1. 5 1
      psiphon/common/fragmentor/fragmentor.go

+ 5 - 1
psiphon/common/fragmentor/fragmentor.go

@@ -337,7 +337,11 @@ func (c *Conn) Write(buffer []byte) (int, error) {
 		if c.bytesFragmented >= c.bytesToFragment {
 		if c.bytesFragmented >= c.bytesToFragment {
 			bytesWritten, err := c.Conn.Write(buffer)
 			bytesWritten, err := c.Conn.Write(buffer)
 			totalBytesWritten += bytesWritten
 			totalBytesWritten += bytesWritten
-			return totalBytesWritten, err
+			if err != nil {
+				return totalBytesWritten, err
+			} else {
+				buffer = nil
+			}
 		}
 		}
 	}
 	}