only limit if bytelimit > 0

2019-09-05 02:08:31 -04:00
parent 7167ba3a1c
commit 3c1a44e1d2
1 changed files with 5 additions and 3 deletions
--- a/tcp/forwarder.go
+++ b/tcp/forwarder.go
@ -290,9 +290,11 @@ func (f *SAMForwarder) forward(conn *sam3.SAMConn) { //(conn net.Conn) {
 		} else {
 			defer client.Close()
 			defer conn.Close()
-			if val, ok := f.Bytes[f.ClientBase64(conn)]; ok == true {
-				if val > f.ByteLimit {
-					return
+			if f.ByteLimit > 0 {
+				if val, ok := f.Bytes[f.ClientBase64(conn)]; ok == true {
+					if val > f.ByteLimit {
+						return
+					}
 				}
 			}
 			if count, err := io.Copy(client, conn); err == nil {