use io.ReadFull to read data from socket and pipe

roadrunner-server · Dec 10, 2020 · df6c7d1 · df6c7d1
1 parent 1cf8475
commit df6c7d1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pipe.go b/pipe.go
@@ -58,7 +58,7 @@ func (rl *PipeRelay) Receive(frame *Frame) error {
 
 	// read the read payload
 	pb := make([]byte, header.ReadPayloadLen())
-	_, err = rl.in.Read(pb)
+	_, err = io.ReadFull(rl.in, pb)
 	if err != nil {
 		return errors.E(op, err)
 	}

diff --git a/socket.go b/socket.go
@@ -57,7 +57,7 @@ func (rl *SocketRelay) Receive(frame *Frame) error {
 
 	// read the read payload
 	pb := make([]byte, header.ReadPayloadLen())
-	_, err = rl.rwc.Read(pb)
+	_, err = io.ReadFull(rl.rwc, pb)
 	if err != nil {
 		return errors.E(op, err)
 	}