Merge pull request #506 from nspcc-dev/fix/writemsg

network: write messages atomically

pkg/network/tcp_peer.go

@@ -63,8 +63,14 @@ func (p *TCPPeer) writeMsg(msg *Message) error {
 	case err := <-p.done:
 		return err
 	default:
-		w := io.NewBinWriterFromIO(p.conn)
+		w := io.NewBufBinWriter()
-		return msg.Encode(w)
+		if err := msg.Encode(w.BinWriter); err != nil {
+			return err
+		}
+
+		_, err := p.conn.Write(w.Bytes())
+
+		return err
 	}
 }