support milli timestamp

Signed-off-by: Naohiro Yoshida <[email protected]>

Author: Naohiro Yoshida

module/header.go

@@ -2,6 +2,7 @@ package module
 
 import (
 	"fmt"
+	"github.com/holiman/uint256"
 	"log"
 
 	clienttypes "github.com/cosmos/ibc-go/v8/modules/core/02-client/types"
@@ -82,3 +83,11 @@ func (h *Header) Account(path common.Address) (*types.StateAccount, error) {
 	}
 	return verifyAccount(target, h.AccountProof, path)
 }
+
+func MilliTimestamp(h *types.Header) uint64 {
+	milliseconds := uint64(0)
+	if h.MixDigest != (common.Hash{}) {
+		milliseconds = uint256.NewInt(0).SetBytes32(h.MixDigest[:]).Uint64()
+	}
+	return h.Time*1000 + milliseconds
+}

module/prover.go

@@ -257,7 +257,7 @@ func (pr *Prover) buildInitialState(dstHeader core.Header) (exported.ClientState
 		ForkSpecs:          GetForkParameters(Network(pr.config.Network)),
 	}
 	consensusState := ConsensusState{
-		Timestamp:              header.Time,
+		Timestamp:              MilliTimestamp(header),
 		PreviousValidatorsHash: makeEpochHash(previousValidators, previousTurnLength),
 		CurrentValidatorsHash:  makeEpochHash(currentValidators, currentTurnLength),
 		StateRoot:              stateRoot.Bytes(),

tests/prover_network_test.go

@@ -99,7 +99,7 @@ func (ts *ProverNetworkTestSuite) TestSuccessCreateInitialLightClientState() {
 	consState := s2.(*module.ConsensusState)
 	ts.Require().Equal(consState.CurrentValidatorsHash, module.MakeEpochHash(cVal, cTurn))
 	ts.Require().Equal(consState.PreviousValidatorsHash, module.MakeEpochHash(pVal, pTurn))
-	ts.Require().Equal(consState.Timestamp, header.Time)
+	ts.Require().Equal(consState.Timestamp, module.MilliTimestamp(header))
 	storageRoot, err := ts.prover.GetStorageRoot(header)
 	ts.Require().NoError(err)
 	ts.Require().Equal(common.BytesToHash(consState.StateRoot), storageRoot)