internal/inproc: detect short-buffer during reads

go-zeromq · Jun 25, 2018 · 7bebbd3 · 7bebbd3
1 parent d10332e
commit 7bebbd3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/inproc/conn.go b/internal/inproc/conn.go
@@ -78,6 +78,9 @@ func (c *conn) read(data []byte) (int, error) {
 	select {
 	case bw := <-c.r:
 		nr := copy(data, bw)
+		if len(data) < len(bw) {
+			return nr, io.ErrShortBuffer
+		}
 		return nr, nil
 	case <-c.rdeadline.wait():
 		return 0, timeoutError{}