From 45cfa65b479f5479e121e9e1ef1493e12e0122f2 Mon Sep 17 00:00:00 2001 From: Conner Fromknecht Date: Tue, 15 Jan 2019 19:46:04 -0800 Subject: [PATCH] watchtower/wtserver/server: handle missed err case from read msg --- watchtower/wtserver/server.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/watchtower/wtserver/server.go b/watchtower/wtserver/server.go index 496c21f3d..3f09e07aa 100644 --- a/watchtower/wtserver/server.go +++ b/watchtower/wtserver/server.go @@ -191,6 +191,12 @@ func (s *Server) handleClient(peer Peer) { defer s.removePeer(&id, peer.RemoteAddr()) msg, err := s.readMessage(peer) + if err != nil { + log.Errorf("Unable to read message from client %s@%s: %v", + id, peer.RemoteAddr(), err) + return + } + remoteInit, ok := msg.(*wtwire.Init) if !ok { log.Errorf("Client %s@%s did not send Init msg as first "+