Changed dmsg.Transport default buffer sizes.

skycoin · Jun 10, 2019 · 785775a · 785775a
1 parent 6933890
commit 785775a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/pkg/dmsg/frame.go b/pkg/dmsg/frame.go
@@ -17,11 +17,12 @@ const (
 	// Type returns the transport type string.
 	Type = "dmsg"
 
-	hsTimeout    = time.Second * 10
-	tpBufCap     = math.MaxUint16
-	tpAckCap     = math.MaxUint8
-	acceptChSize = 20
-	headerLen    = 5 // fType(1 byte), chID(2 byte), payLen(2 byte)
+	hsTimeout     = time.Second * 10
+	tpBufCap      = math.MaxUint16
+	tpBufFrameCap = math.MaxUint8
+	tpAckCap      = math.MaxUint8
+	acceptChSize  = 20
+	headerLen     = 5 // fType(1 byte), chID(2 byte), payLen(2 byte)
 )
 
 func isInitiatorID(tpID uint16) bool { return tpID%2 == 0 }

diff --git a/pkg/dmsg/transport.go b/pkg/dmsg/transport.go
@@ -57,6 +57,7 @@ func NewTransport(conn net.Conn, log *logging.Logger, local, remote cipher.PubKe
 		remote:   remote,
 		inCh:     make(chan Frame),
 		ackBuf:   make([]byte, 0, tpAckCap),
+		buf:      make(net.Buffers, 0, tpBufFrameCap),
 		bufCh:    make(chan struct{}, 1),
 		done:     make(chan struct{}),
 		doneFunc: doneFunc,