Skip to content
Snippets Groups Projects
Unverified Commit 036199bb authored by Angelo Capossele's avatar Angelo Capossele Committed by GitHub
Browse files

Merge pull request #340 from iotaledger/fix/drng-check

:bug: adds len check
parents c2242261 e1ac1a5a
No related branches found
No related tags found
No related merge requests found
...@@ -5,6 +5,7 @@ import ( ...@@ -5,6 +5,7 @@ import (
"github.com/iotaledger/goshimmer/packages/binary/drng" "github.com/iotaledger/goshimmer/packages/binary/drng"
"github.com/iotaledger/goshimmer/packages/binary/drng/payload" "github.com/iotaledger/goshimmer/packages/binary/drng/payload"
"github.com/iotaledger/goshimmer/packages/binary/drng/payload/header"
"github.com/iotaledger/goshimmer/packages/binary/drng/state" "github.com/iotaledger/goshimmer/packages/binary/drng/state"
cbPayload "github.com/iotaledger/goshimmer/packages/binary/drng/subtypes/collectiveBeacon/payload" cbPayload "github.com/iotaledger/goshimmer/packages/binary/drng/subtypes/collectiveBeacon/payload"
"github.com/iotaledger/goshimmer/packages/binary/messagelayer/message" "github.com/iotaledger/goshimmer/packages/binary/messagelayer/message"
...@@ -68,6 +69,12 @@ func configureEvents() { ...@@ -68,6 +69,12 @@ func configureEvents() {
cachedMessageMetadata.Release() cachedMessageMetadata.Release()
cachedMessage.Consume(func(msg *message.Message) { cachedMessage.Consume(func(msg *message.Message) {
if msg.Payload().Type() != payload.Type {
return
}
if len(msg.Payload().Bytes()) < header.Length {
return
}
marshalUtil := marshalutil.New(msg.Payload().Bytes()) marshalUtil := marshalutil.New(msg.Payload().Bytes())
parsedPayload, err := payload.Parse(marshalUtil) parsedPayload, err := payload.Parse(marshalUtil)
if err != nil { if err != nil {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment