aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDimitri Sokolyuk <demon@dim13.org>2016-07-28 00:41:51 +0200
committerDimitri Sokolyuk <demon@dim13.org>2016-07-28 00:41:51 +0200
commit2dc4620c7cb7c7c7130fc098e706c583f3c0182a (patch)
tree7e09dbceeea6a5db47e6f3c91e515eb8d2cebd86
parent475e544bbeffc70a9796766b6e28a6a4b531a510 (diff)
Check len
-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)