package client_test import ( "bytes" "context" "testing" "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" abci "github.com/tendermint/tendermint/abci/types" "github.com/tendermint/tendermint/crypto/ed25519" "github.com/tendermint/tendermint/crypto/encoding" "github.com/tendermint/tendermint/crypto/tmhash" tmrand "github.com/tendermint/tendermint/libs/rand" "github.com/tendermint/tendermint/privval" tmproto "github.com/tendermint/tendermint/proto/tendermint/types" "github.com/tendermint/tendermint/rpc/client" "github.com/tendermint/tendermint/types" ) // For some reason the empty node used in tests has a time of // 2018-10-10 08:20:13.695936996 +0000 UTC // this is because the test genesis time is set here // so in order to validate evidence we need evidence to be the same time var defaultTestTime = time.Date(2018, 10, 10, 8, 20, 13, 695936996, time.UTC) func newEvidence(t *testing.T, val *privval.FilePV, vote *types.Vote, vote2 *types.Vote, chainID string) *types.DuplicateVoteEvidence { t.Helper() var err error v := vote.ToProto() v2 := vote2.ToProto() vote.Signature, err = val.Key.PrivKey.Sign(types.VoteSignBytes(chainID, v)) require.NoError(t, err) vote2.Signature, err = val.Key.PrivKey.Sign(types.VoteSignBytes(chainID, v2)) require.NoError(t, err) validator := types.NewValidator(val.Key.PubKey, 10) valSet := types.NewValidatorSet([]*types.Validator{validator}) ev, err := types.NewDuplicateVoteEvidence(vote, vote2, defaultTestTime, valSet) require.NoError(t, err) return ev } func makeEvidences( t *testing.T, val *privval.FilePV, chainID string, ) (correct *types.DuplicateVoteEvidence, fakes []*types.DuplicateVoteEvidence) { vote := types.Vote{ ValidatorAddress: val.Key.Address, ValidatorIndex: 0, Height: 1, Round: 0, Type: tmproto.PrevoteType, Timestamp: defaultTestTime, BlockID: types.BlockID{ Hash: tmhash.Sum(tmrand.Bytes(tmhash.Size)), PartSetHeader: types.PartSetHeader{ Total: 1000, Hash: tmhash.Sum([]byte("partset")), }, }, } vote2 := vote vote2.BlockID.Hash = tmhash.Sum([]byte("blockhash2")) correct = newEvidence(t, val, &vote, &vote2, chainID) fakes = make([]*types.DuplicateVoteEvidence, 0) // different address { v := vote2 v.ValidatorAddress = []byte("some_address") fakes = append(fakes, newEvidence(t, val, &vote, &v, chainID)) } // different height { v := vote2 v.Height = vote.Height + 1 fakes = append(fakes, newEvidence(t, val, &vote, &v, chainID)) } // different round { v := vote2 v.Round = vote.Round + 1 fakes = append(fakes, newEvidence(t, val, &vote, &v, chainID)) } // different type { v := vote2 v.Type = tmproto.PrecommitType fakes = append(fakes, newEvidence(t, val, &vote, &v, chainID)) } // exactly same vote { v := vote fakes = append(fakes, newEvidence(t, val, &vote, &v, chainID)) } return correct, fakes } func TestBroadcastEvidence_DuplicateVoteEvidence(t *testing.T) { ctx, cancel := context.WithCancel(context.Background()) defer cancel() n, config := NodeSuite(t) chainID := config.ChainID() pv, err := privval.LoadOrGenFilePV(config.PrivValidator.KeyFile(), config.PrivValidator.StateFile()) require.NoError(t, err) for i, c := range GetClients(t, n, config) { correct, fakes := makeEvidences(t, pv, chainID) t.Logf("client %d", i) // make sure that the node has produced enough blocks waitForBlock(ctx, t, c, 2) result, err := c.BroadcastEvidence(ctx, correct) require.NoError(t, err, "BroadcastEvidence(%s) failed", correct) assert.Equal(t, correct.Hash(), result.Hash, "expected result hash to match evidence hash") status, err := c.Status(ctx) require.NoError(t, err) err = client.WaitForHeight(c, status.SyncInfo.LatestBlockHeight+2, nil) require.NoError(t, err) ed25519pub := pv.Key.PubKey.(ed25519.PubKey) rawpub := ed25519pub.Bytes() result2, err := c.ABCIQuery(ctx, "/val", rawpub) require.NoError(t, err) qres := result2.Response require.True(t, qres.IsOK()) var v abci.ValidatorUpdate err = abci.ReadMessage(bytes.NewReader(qres.Value), &v) require.NoError(t, err, "Error reading query result, value %v", qres.Value) pk, err := encoding.PubKeyFromProto(v.PubKey) require.NoError(t, err) require.EqualValues(t, rawpub, pk, "Stored PubKey not equal with expected, value %v", string(qres.Value)) require.Equal(t, int64(9), v.Power, "Stored Power not equal with expected, value %v", string(qres.Value)) for _, fake := range fakes { _, err := c.BroadcastEvidence(ctx, fake) require.Error(t, err, "BroadcastEvidence(%s) succeeded, but the evidence was fake", fake) } } } func TestBroadcastEmptyEvidence(t *testing.T) { n, conf := NodeSuite(t) for _, c := range GetClients(t, n, conf) { _, err := c.BroadcastEvidence(context.Background(), nil) assert.Error(t, err) } } func waitForBlock(ctx context.Context, t *testing.T, c client.Client, height int64) { timer := time.NewTimer(0 * time.Millisecond) defer timer.Stop() for { select { case <-ctx.Done(): return case <-timer.C: status, err := c.Status(ctx) require.NoError(t, err) if status.SyncInfo.LatestBlockHeight >= height { return } timer.Reset(200 * time.Millisecond) } } }