diff --git a/node/crypto/wesolowski_frame_prover.go b/node/crypto/wesolowski_frame_prover.go index 5f72318..dcb36a9 100644 --- a/node/crypto/wesolowski_frame_prover.go +++ b/node/crypto/wesolowski_frame_prover.go @@ -103,6 +103,13 @@ func (w *WesolowskiFrameProver) VerifyMasterClockFrame( ) } + if len(frame.Output) != 516 { + return errors.Wrap( + errors.New("invalid output"), + "verify clock frame", + ) + } + b := sha3.Sum256(input) v := vdf.New(frame.Difficulty, b) proof := [516]byte{} @@ -421,6 +428,13 @@ func (w *WesolowskiFrameProver) VerifyDataClockFrame( ) } + if len(frame.Output) != 516 { + return errors.Wrap( + errors.New("invalid output"), + "verify clock frame", + ) + } + b := sha3.Sum256(input) v := vdf.New(frame.Difficulty, b) proof := [516]byte{}