From 7d235a454d9be0568265e7fd28314b07ba9e2062 Mon Sep 17 00:00:00 2001
From: Jakob Borg <jakob@nym.se>
Date: Sat, 21 Dec 2013 23:52:20 +0100
Subject: [PATCH] Refactor length check

---
 protocol/marshal.go | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/protocol/marshal.go b/protocol/marshal.go
index f363fb3c7..bc97287f0 100644
--- a/protocol/marshal.go
+++ b/protocol/marshal.go
@@ -1,6 +1,7 @@
 package protocol
 
 import (
+	"errors"
 	"io"
 
 	"github.com/calmh/syncthing/buffers"
@@ -22,6 +23,13 @@ type marshalWriter struct {
 	err error
 }
 
+// We will never encode nor expect to decode blobs larger than 10 MB. Check
+// inserted to protect against attempting to allocate arbitrary amounts of
+// memory when reading a corrupt message.
+const maxBytesFieldLength = 10 * 1 << 20
+
+var ErrFieldLengthExceeded = errors.New("Raw bytes field size exceeds limit")
+
 func (w *marshalWriter) writeString(s string) {
 	w.writeBytes([]byte(s))
 }
@@ -30,6 +38,10 @@ func (w *marshalWriter) writeBytes(bs []byte) {
 	if w.err != nil {
 		return
 	}
+	if len(bs) > maxBytesFieldLength {
+		w.err = ErrFieldLengthExceeded
+		return
+	}
 	w.writeUint32(uint32(len(bs)))
 	if w.err != nil {
 		return
@@ -91,10 +103,9 @@ func (r *marshalReader) readBytes() []byte {
 	if r.err != nil {
 		return nil
 	}
-	if l > 10*1<<20 {
-		// Individual blobs in BEP are not significantly larger than BlockSize.
-		// BlockSize is not larger than 1MB.
-		panic("too large read - protocol error or out of sync")
+	if l > maxBytesFieldLength {
+		r.err = ErrFieldLengthExceeded
+		return nil
 	}
 	b := buffers.Get(l + pad(l))
 	_, r.err = io.ReadFull(r.r, b)