aboutsummaryrefslogtreecommitdiff
path: root/peer/messages.go
diff options
context:
space:
mode:
Diffstat (limited to 'peer/messages.go')
-rw-r--r--peer/messages.go3
1 files changed, 3 insertions, 0 deletions
diff --git a/peer/messages.go b/peer/messages.go
index 5958380..a76248c 100644
--- a/peer/messages.go
+++ b/peer/messages.go
@@ -53,6 +53,9 @@ func DecodeHandshake(r io.Reader) (Handshake, bool) {
var h Handshake
var l int8
binary.Read(r, binary.BigEndian, &l)
+ if int(l) != len(Proto) {
+ return h, false
+ }
proto := make([]byte, int(l))
r.Read(proto)
h.Proto = string(proto)