Only set vote extension signatures when signing is successful

Signed-off-by: Thane Thomson <connect@thanethomson.com>
This commit is contained in:
Thane Thomson
2022-04-12 13:53:32 -04:00
parent b054c42b03
commit 35aa879c7e
2 changed files with 8 additions and 3 deletions

View File

@@ -405,10 +405,11 @@ func (pv *FilePV) signVote(chainID string, vote *tmproto.Vote) error {
vote.Signature = sig
// Sign the vote extension, regardless of whether it's present or not
vote.ExtensionSignature, err = pv.Key.PrivKey.Sign(extSignBytes)
extSig, err := pv.Key.PrivKey.Sign(extSignBytes)
if err != nil {
return err
}
vote.ExtensionSignature = extSig
return nil
}

View File

@@ -96,8 +96,12 @@ func (pv MockPV) SignVote(ctx context.Context, chainID string, vote *tmproto.Vot
return err
}
vote.Signature = sig
vote.ExtensionSignature, err = pv.PrivKey.Sign(extSignBytes)
return err
extSig, err := pv.PrivKey.Sign(extSignBytes)
if err != nil {
return err
}
vote.ExtensionSignature = extSig
return nil
}
// Implements PrivValidator.