Merge pull request #238 from nguyenbatam/fix_dv

fix error double validation

consensus/errors.go

@@ -34,4 +34,6 @@ var (
 	// ErrInvalidNumber is returned if a block's number doesn't equal it's parent's
 	// plus one.
 	ErrInvalidNumber = errors.New("invalid block number")
+
+	ErrMissingValidatorSignature = errors.New("missing validator in header")
 )

consensus/posv/posv.go

@@ -216,7 +216,7 @@ type Posv struct {
 
 	recents             *lru.ARCCache           // Snapshots for recent block to speed up reorgs
 	signatures          *lru.ARCCache           // Signatures of recent blocks to speed up mining
-
+	validatorSignatures *lru.ARCCache           // Signatures of recent blocks to speed up mining
 	proposals           map[common.Address]bool // Current list of proposals we are pushing
 
 	signer common.Address  // Ethereum address of the signing key
@@ -240,12 +240,13 @@ func New(config *params.PosvConfig, db ethdb.Database) *Posv {
 	// Allocate the snapshot caches and create the engine
 	recents, _ := lru.NewARC(inmemorySnapshots)
 	signatures, _ := lru.NewARC(inmemorySignatures)
-
+	validatorSignatures, _ := lru.NewARC(inmemorySignatures)
 	return &Posv{
 		config:              &conf,
 		db:                  db,
 		recents:             recents,
 		signatures:          signatures,
+		validatorSignatures: validatorSignatures,
 		proposals:           make(map[common.Address]bool),
 	}
 }
@@ -956,12 +957,12 @@ func (c *Posv) RecoverSigner(header *types.Header) (common.Address, error) {
 func (c *Posv) RecoverValidator(header *types.Header) (common.Address, error) {
 	// If the signature's already cached, return that
 	hash := header.Hash()
-	if address, known := c.signatures.Get(hash); known {
+	if address, known := c.validatorSignatures.Get(hash); known {
 		return address.(common.Address), nil
 	}
 	// Retrieve the signature from the header extra-data
 	if len(header.Validator) < extraSeal {
-		return common.Address{}, errMissingSignature
+		return common.Address{}, consensus.ErrMissingValidatorSignature
 	}
 	signature := header.Validator[len(header.Validator)-extraSeal:]
 
@@ -973,7 +974,7 @@ func (c *Posv) RecoverValidator(header *types.Header) (common.Address, error) {
 	var signer common.Address
 	copy(signer[:], crypto.Keccak256(pubkey[1:])[12:])
 
-	c.signatures.Add(hash, signer)
+	c.validatorSignatures.Add(hash, signer)
 	return signer, nil
 }
 

eth/fetcher/fetcher.go

@@ -142,7 +142,6 @@ type Fetcher struct {
 	queueChangeHook    func(common.Hash, bool) // Method to call upon adding or deleting a block from the import queue
 	fetchingHook       func([]common.Hash)     // Method to call upon starting a block (eth/61) or header (eth/62) fetch
 	completingHook     func([]common.Hash)     // Method to call upon starting a block body fetch (eth/62)
-	doubleValidateHook func(*types.Block) error
 	signHook           func(*types.Block) error
 	appendM2HeaderHook func(*types.Block) (*types.Block, error)
 }
@@ -654,33 +653,30 @@ func (f *Fetcher) insert(peer string, block *types.Block) {
 		// Quickly validate the header and propagate the block if it passes
 		switch err := f.verifyHeader(block.Header()); err {
 		case nil:
+			// All ok, quickly propagate to our peers
+			propBroadcastOutTimer.UpdateSince(block.ReceivedAt)
+			go f.broadcastBlock(block, true)
+		case consensus.ErrFutureBlock:
+		case consensus.ErrMissingValidatorSignature:
+			newBlock := block
 			if f.appendM2HeaderHook != nil {
-				if block, err = f.appendM2HeaderHook(block); err != nil {
+				if newBlock, err = f.appendM2HeaderHook(block); err != nil {
 					log.Error("Append m2 to block header fail", "err", err)
 					return
 				}
 			}
-
-			// All ok, quickly propagate to our peers
-			propBroadcastOutTimer.UpdateSince(block.ReceivedAt)
 			go f.broadcastBlock(block, true)
-
+			if newBlock.Hash() == block.Hash() {
-		case consensus.ErrFutureBlock:
+				return
-			// Weird future block, don't fail, but neither propagate
+			}
-
+			block = newBlock
+			propBroadcastOutTimer.UpdateSince(block.ReceivedAt)
 		default:
 			// Something went very wrong, drop the peer
 			log.Debug("Propagated block verification failed", "peer", peer, "number", block.Number(), "hash", hash, "err", err)
 			f.dropPeer(peer)
 			return
 		}
-		// Invoke the dv hook to run double validation layer
-		if f.doubleValidateHook != nil {
-			if err := f.doubleValidateHook(block); err != nil {
-				log.Error("Double validation failed", "err", err, "Discard this block!")
-				return
-			}
-		}
 
 		// Run the actual import and log any issues
 		if _, err := f.insertChain(types.Blocks{block}); err != nil {
@@ -756,11 +752,6 @@ func (f *Fetcher) forgetBlock(hash common.Hash) {
 	}
 }
 
-// Bind double validate hook before block imported into chain.
-func (f *Fetcher) SetDoubleValidateHook(doubleValidateHook func(*types.Block) error) {
-	f.doubleValidateHook = doubleValidateHook
-}
-
 // Bind double validate hook before block imported into chain.
 func (f *Fetcher) SetSignHook(signHook func(*types.Block) error) {
 	f.signHook = signHook