From 2dc4620c7cb7c7c7130fc098e706c583f3c0182a Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Thu, 28 Jul 2016 00:41:51 +0200 Subject: Check len --- peer/messages.go | 3 +++ 1 file changed, 3 insertions(+) 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) -- cgit v1.2.3