eight333.go

package spvwallet

import (
	"encoding/hex"
	"errors"
	"fmt"
	"github.com/btcsuite/btcd/chaincfg"
	"github.com/btcsuite/btcd/chaincfg/chainhash"
	peerpkg "github.com/btcsuite/btcd/peer"
	"github.com/btcsuite/btcd/txscript"
	"github.com/btcsuite/btcd/wire"
	btc "github.com/btcsuite/btcutil"
	"net"
	"strings"
	"time"
)

const (
	maxRequestedTxns  = wire.MaxInvPerMsg
	maxFalsePositives = 7
)

var lastTimeRequestBlock = time.Now()
var alivePeer *peerpkg.Peer

// newPeerMsg signifies a newly connected peer to the block handler.
type newPeerMsg struct {
	peer *peerpkg.Peer
}

// donePeerMsg signifies a newly disconnected peer to the block handler.
type donePeerMsg struct {
	peer *peerpkg.Peer
}

// headersMsg packages a bitcoin headers message and the peer it came from
// together so the handler has access to that information.
type headersMsg struct {
	headers *wire.MsgHeaders
	peer    *peerpkg.Peer
}

type blockMsg struct {
	block *wire.MsgBlock
	peer  *peerpkg.Peer
}

// merkleBlockMsg packages a merkle block message and the peer it came from
// together so the handler has access to that information.
type merkleBlockMsg struct {
	merkleBlock *wire.MsgMerkleBlock
	peer        *peerpkg.Peer
}

// invMsg packages a bitcoin inv message and the peer it came from together
// so the handler has access to that information.
type invMsg struct {
	inv  *wire.MsgInv
	peer *peerpkg.Peer
}

type heightAndTime struct {
	height    uint32
	timestamp time.Time
}

// txMsg packages a bitcoin tx message and the peer it came from together
// so the handler has access to that information.
type txMsg struct {
	tx    *wire.MsgTx
	peer  *peerpkg.Peer
	reply chan struct{}
}

type updateFiltersMsg struct{}

type WireServiceConfig struct {
	params             *chaincfg.Params
	chain              *Blockchain
	txStore            *TxStore
	walletCreationDate time.Time
	minPeersForSync    int
}

// peerSyncState stores additional information that the WireService tracks
// about a peer.
type peerSyncState struct {
	syncCandidate   bool
	requestQueue    []*wire.InvVect
	requestedTxns   map[chainhash.Hash]heightAndTime
	requestedBlocks map[chainhash.Hash]struct{}
	falsePositives  uint32
	blockScore      int32
}

type WireService struct {
	params             *chaincfg.Params
	chain              *Blockchain
	txStore            *TxStore
	walletCreationDate time.Time
	syncPeer           *peerpkg.Peer
	peerStates         map[*peerpkg.Peer]*peerSyncState
	requestedTxns      map[chainhash.Hash]heightAndTime
	requestedBlocks    map[chainhash.Hash]struct{}
	mempool            map[chainhash.Hash]struct{}
	msgChan            chan interface{}
	quit               chan struct{}
	minPeersForSync    int
	zeroHash           chainhash.Hash
}

func NewWireService(config *WireServiceConfig) *WireService {
	return &WireService{
		params:             config.params,
		chain:              config.chain,
		walletCreationDate: config.walletCreationDate,
		minPeersForSync:    config.minPeersForSync,
		txStore:            config.txStore,
		peerStates:         make(map[*peerpkg.Peer]*peerSyncState),
		requestedTxns:      make(map[chainhash.Hash]heightAndTime),
		requestedBlocks:    make(map[chainhash.Hash]struct{}),
		mempool:            make(map[chainhash.Hash]struct{}),
		msgChan:            make(chan interface{}),
	}
}

func (ws *WireService) MsgChan() chan interface{} {
	return ws.msgChan
}

// The start function must be run in its own goroutine. The entire WireService is single
// threaded which means all messages are processed sequentially removing the need for complex
// locking.
func (ws *WireService) Start() {
	ws.quit = make(chan struct{})
	best, err := ws.chain.BestBlock()
	if err != nil {
		log.Error(err.Error())
	}

	go func() {
		// 加入定时监听 ,发起请求区块blockHash 请求，查找数据库中记录的，还未收到的blockhash
		var intervalTimer = time.NewTicker(IntervalToScanBlock)
		defer intervalTimer.Stop()
		var requestBlockFunc = func() {
			if time.Now().After(lastTimeRequestBlock.Add(IntervalToScanBlock)) {
				if alivePeer == nil {
					log.Info("alivePeer is nil,time.Now().String()===>", time.Now().String())
					return
				}
				var blockHashDb, err = ws.txStore.ScanBlocks().GetLatestUnScanBlockHash()
				if err != nil {
					log.Info("ws.txStore.ScanBlocks().GetLatestUnScanBlockHash error is ===>", err)
					return
				}
				var blockHash, err1 = chainhash.NewHashFromStr(blockHashDb)
				if err1 != nil {
					log.Info("chainhash.NewHashFromSt error is ===>", err)
					return
				}
				var invet *wire.InvVect = wire.NewInvVect(wire.InvTypeBlock, blockHash)
				gdmsg2 := wire.NewMsgGetData()
				gdmsg2.AddInvVect(invet)
				alivePeer.QueueMessage(gdmsg2, nil)
				log.Info("requestBlockFunc(), send wire.InvTypeBlock request blockHash is  %s ===>", blockHash.String())
			} else {
				log.Info("time not enougth  something===>", lastTimeRequestBlock.Local().String()+"||time.Now()--->"+time.Now().String())
			}
		}
		for {
			select {
			case <-intervalTimer.C:
				requestBlockFunc()
			}
		}
	}()

	log.Infof("Starting wire service at height %d", int(best.height))
out:
	for {
		select {
		case m := <-ws.msgChan:
			switch msg := m.(type) {
			case newPeerMsg:
				fmt.Printf("newPeerMsg is %+v \n", msg.peer)
				ws.handleNewPeerMsg(msg.peer)
			case donePeerMsg:
				ws.handleDonePeerMsg(msg.peer)
			case headersMsg:
				_, _, height, err := ws.chain.CommitHeader(*msg.headers.Headers[0])
				if err != nil {
					fmt.Printf("headersMsg.headers err %+v \n", err)
				}
				fmt.Printf("height %+v; msg.headers is %+v \n", height, msg.headers.Headers[0].BlockHash().String())
				ws.handleHeadersMsg(&msg)
			/*case merkleBlockMsg:
			ws.handleMerkleBlockMsg(&msg)*/
			case blockMsg:
				_, _, height, err := ws.chain.CommitHeader((&msg).block.Header)
				if err != nil {
					fmt.Printf("blockMsg.headers err %+v \n", err)
				}
				fmt.Printf("height is %v, blockMsg BlockHash is---> %+v \n", height, msg.block.Header.BlockHash())

				ws.handleBlockMsg(&msg)
			case invMsg:
				ws.handleInvMsg(&msg)
			/*case txMsg:
			ws.handleTxMsg(&msg)*/
			case updateFiltersMsg:
				ws.handleUpdateFiltersMsg()
			default:
				log.Warningf("Unknown message type sent to WireService message chan: %T", msg)
			}
		case <-ws.quit:
			break out
		}
	}
}

func (ws *WireService) Stop() {
	ws.syncPeer = nil
	close(ws.quit)
}

func (ws *WireService) Resync() {
	ws.startSync(ws.syncPeer)
}

func (ws *WireService) handleNewPeerMsg(peer *peerpkg.Peer) {
	// Initialize the peer state
	ws.peerStates[peer] = &peerSyncState{
		syncCandidate:   ws.isSyncCandidate(peer),
		requestedTxns:   make(map[chainhash.Hash]heightAndTime),
		requestedBlocks: make(map[chainhash.Hash]struct{}),
	}

	ws.updateFilterAndSend(peer)

	// If we don't have a sync peer and we are not current we should start a sync
	if ws.syncPeer == nil && !ws.Current() {
		ws.startSync(nil)
	}
}

// isSyncCandidate returns whether or not the peer is a candidate to consider
// syncing from.
func (ws *WireService) isSyncCandidate(peer *peerpkg.Peer) bool {
	// Typically a peer is not a candidate for sync if it's not a full node,
	// however regression test is special in that the regression tool is
	// not a full node and still needs to be considered a sync candidate.
	if ws.params.Name == chaincfg.RegressionNetParams.Name {
		// The peer is not a candidate if it's not coming from localhost
		// or the hostname can't be determined for some reason.
		host, _, err := net.SplitHostPort(peer.Addr())
		if err != nil {
			return false
		}

		if host != "127.0.0.1" && host != "localhost" {
			return false
		}
	} else {
		// The peer is not a candidate for sync if it's not a full node
		nodeServices := peer.Services()
		if nodeServices&wire.SFNodeNetwork != wire.SFNodeNetwork {
			return false
		}
	}

	// Candidate if all checks passed.
	return true
}

func (ws *WireService) startSync(syncPeer *peerpkg.Peer) {
	// Wait for a minimum number of peers to connect. This makes sure we have a good
	// selection to choose from before starting the sync.
	if len(ws.peerStates) < ws.minPeersForSync {
		return
	}
	ws.Rebroadcast()
	bestBlock, err := ws.chain.BestBlock()
	if err != nil {
		log.Error(err.Error())
		return
	}
	var bestPeer *peerpkg.Peer
	if syncPeer == nil {
		var bestPeerHeight int32
		for peer, state := range ws.peerStates {
			if !state.syncCandidate {
				continue
			}

			// Remove sync candidate peers that are no longer candidates due
			// to passing their latest known block.  NOTE: The < is
			// intentional as opposed to <=.  While technically the peer
			// doesn't have a later block when it's equal, it will likely
			// have one soon so it is a reasonable choice.  It also allows
			// the case where both are at 0 such as during regression test.
			if peer.LastBlock() < int32(bestBlock.height) {
				state.syncCandidate = false
				continue
			}

			// Select peer which is reporting the greatest height
			if peer.LastBlock() > bestPeerHeight {
				bestPeer = peer
				bestPeerHeight = peer.LastBlock()
			}
		}
	} else {
		bestPeer = syncPeer
	}

	// Start syncing this bitch
	if bestPeer != nil {
		// TODO: use checkpoints here
		ws.syncPeer = bestPeer

		// Clear the requestedBlocks if the sync peer changes, otherwise
		// we may ignore blocks we need that the last sync peer failed
		// to send.
		ws.requestedBlocks = make(map[chainhash.Hash]struct{})

		locator := ws.chain.GetBlockLocator()

		// If the best header we have was created before this wallet then we can sync just headers
		// up to the wallet creation date since we know there wont be any transactions in those
		// blocks we're interested in. However, if we're past the wallet creation date we need to
		// start downloading merkle blocks so we learn of the wallet's transactions. We'll use a
		// buffer of one week to make sure we don't miss anything.
		log.Infof("Starting chain download from %s", bestPeer)
		if bestBlock.header.Timestamp.Before(ws.walletCreationDate.Add(-time.Hour * 24 * 7)) {
			bestPeer.PushGetHeadersMsg(locator, &ws.zeroHash)
		} else {
			bestPeer.PushGetBlocksMsg(locator, &ws.zeroHash)
		}
	} else {
		log.Warning("No sync candidates available")
	}
}

func (ws *WireService) Current() bool {
	best, err := ws.chain.BestBlock()
	if err != nil {
		return false
	}

	// If our best header's timestamp was more than 24 hours ago, we're probably not current
	if best.header.Timestamp.Before(time.Now().Add(-24 * time.Hour)) {
		return false
	}

	// Check our other peers to see if any are reporting a greater height than we have
	for peer := range ws.peerStates {
		if int32(best.height) < peer.LastBlock() {
			return false
		}
	}
	return true
}

func (ws *WireService) handleDonePeerMsg(peer *peerpkg.Peer) {
	state, exists := ws.peerStates[peer]
	if !exists {
		return
	}

	// Remove the peer from the list of candidate peers.
	delete(ws.peerStates, peer)

	// Remove requested transactions from the global map so that they will
	// be fetched from elsewhere next time we get an inv.
	for txHash := range state.requestedTxns {
		delete(ws.requestedTxns, txHash)
	}

	// Remove requested blocks from the global map so that they will be
	// fetched from elsewhere next time we get an inv.
	// TODO: we could possibly here check which peers have these blocks
	// and request them now to speed things up a little.
	for blockHash := range state.requestedBlocks {
		delete(ws.requestedBlocks, blockHash)
	}

	// Attempt to find a new peer to sync from if the quitting peer is the
	// sync peer.
	if ws.syncPeer == peer && !ws.Current() {
		log.Info("Sync peer disconnected")
		ws.syncPeer = nil
		ws.startSync(nil)
	}
}

var scanBlockHeadsCallBackFunc func(ScanBlockStruct, error) bool
var scanTxsCallbackFunc func(ScanTxStruct, error) bool

func (ws *WireService) AddScanBlockHeadsCallBack(callback func(ScanBlockStruct, error) bool) {
	scanBlockHeadsCallBackFunc = callback
}

func (ws *WireService) AddScanTxsCallBack(callback func(ScanTxStruct, error) bool) {
	scanTxsCallbackFunc = callback
}

// handleHeadersMsg handles block header messages from all peers.  Headers are
// requested when performing a headers-first sync.
func (ws *WireService) handleHeadersMsg(hmsg *headersMsg) {
	peer := hmsg.peer
	if peer != ws.syncPeer {
		log.Warning("Received header message from a peer that isn't our sync peer")
		peer.Disconnect()
		return
	}
	_, exists := ws.peerStates[peer]
	if !exists {
		log.Warningf("Received headers message from unknown peer %s", peer)
		peer.Disconnect()
		return
	}

	//update the alivePeer
	alivePeer = peer

	msg := hmsg.headers
	numHeaders := len(msg.Headers)

	// Nothing to do for an empty headers message
	if numHeaders == 0 {
		return
	}

	// Process each header we received. Make sure when check that each one is before our
	// wallet creation date (minus the buffer). If we pass the creation date we will exit
	// request merkle blocks from this point forward and exit the function.
	badHeaders := 0
	for _, blockHeader := range msg.Headers {
		//if blockHeader.Timestamp.Before(ws.walletCreationDate.Add(-time.Hour * 24 * 7)) {
		_, _, height, err := ws.chain.CommitHeader(*blockHeader)
		if err != nil {
			badHeaders++
			log.Errorf("Commit header error: %s", err.Error())
		}
		log.Infof("Received header %s at height %d", blockHeader.BlockHash().String(), height)
		{
			if int(height) > StartBlockNum {
				// test verify not notify wire.InvTypeBlock
				{
					var invet = wire.InvVect{
						Type: wire.InvTypeBlock,
						Hash: blockHeader.BlockHash(),
					}
					gdmsg2 := wire.NewMsgGetData()
					gdmsg2.AddInvVect(&invet)
					peer.QueueMessage(gdmsg2, nil)
					log.Infof("send wire.InvTypeBlock request %s || height is %d", blockHeader.BlockHash().String(), height)
				}
				{
					err = ws.txStore.ScanBlocks().Put(blockHeader.BlockHash().String(), int(height), int(0)) // isFixScan 0:failure  1:successful
					if err != nil {
						log.Infof("ws.txStore.ScanBlocks().Put err is ", err)
					} else {
						//回调通知callback
						var scanBlock = ScanBlockStruct{
							BlockHash:   blockHeader.BlockHash().String(),
							BlockHeight: int(height),
							IsScan:      0,
						}
						if scanBlockHeadsCallBackFunc != nil {
							var callbackStatus = scanBlockHeadsCallBackFunc(scanBlock, nil)
							if !callbackStatus {
								log.Infof("sync BlockHead err is failure", callbackStatus)
							}
						}
					}
				}
			}
		}
		/*} else {
			log.Info("Switching to downloading merkle blocks")
			locator := ws.chain.GetBlockLocator()
			peer.PushGetBlocksMsg(locator, &ws.zeroHash)
			return
		}*/
	}
	// Usually the peer will send the header at the tip of the chain in each batch. This will trigger
	// one commit error so we'll consider that acceptable, but anything more than that suggests misbehavior
	// so we'll dump this peer.
	if badHeaders > 1 {
		log.Warningf("Disconnecting from peer %s because he sent us too many bad headers", peer)
		peer.Disconnect()
		return
	}

	// Request the next batch of headers
	locator := ws.chain.GetBlockLocator()
	err := peer.PushGetHeadersMsg(locator, &ws.zeroHash)
	if err != nil {
		log.Warningf("Failed to send getheaders message to peer %s: %v", peer.Addr(), err)
		return
	}
}

func (ws *WireService) handleBlockMsg(bmsg *blockMsg) {
	peer := bmsg.peer
	_, exists := ws.peerStates[peer]
	if !exists {
		log.Warningf(" parker Received merkle block message from unknown peer %s", peer)
		return
	}

	//update lastTimeRequestBlock
	lastTimeRequestBlock = time.Now()
	//update the alivePeer
	alivePeer = peer

	go func() {
		block := bmsg.block
		header := block.Header
		blockHash := header.BlockHash()
		log.Warningf("handleBlockMsg(): header.BlockHash() is ==> %s ,|| block timestamp is %v ,|| previous hash is %s", blockHash, header.Timestamp, header.PrevBlock)
		txs := block.Transactions
		var isSuccessAnalyseAllBlock = true
		{
		AnalyseTxsLabel:
			for _, value := range txs {
				var txeg = value
				var txOut = txeg.TxOut
				var tempPkScript []byte
				var tempValue int64
				var isExistTargetAddress = false
				var targetAddress = ""
				for _, value := range txOut {
					config := NewDefaultConfig()
					config.Params = &chaincfg.MainNetParams
					//log.Warningf("parker txOut.index is %v, || value is ==> %v ", index, value)
					var btcAddr, err = scriptToAddress(value.PkScript, &chaincfg.MainNetParams)
					if err != nil {
						if len(value.PkScript) > 2 {
							tempPkScript = value.PkScript[2:] //前2位是标识位，取后面内容
						}
					} else {
						//	log.Warningf("parker  btcAddr is =====> %v ,txHash is ", btcAddr, txeg.TxHash().String())
						if (TargetScanAddresses != nil) && len(TargetScanAddresses) > 0 {
							for _, address := range TargetScanAddresses {
								if strings.ToLower(strings.Trim(btcAddr.String(), "")) == strings.ToLower(strings.Trim(address, "")) {
									//	log.Infof("appear equal txeg.TxHash()===> ", txeg.TxHash().String())
									tempValue = value.Value
									isExistTargetAddress = true
									targetAddress = strings.ToLower(strings.Trim(address, ""))
								}
							}
						} else {
							//fmt.Println("err(), error is===> no watch TargetScanAddresses")
						}
					}
				}
				if isExistTargetAddress { // 写入这笔交易到数据库，标识通知到小程序的状态为 未通知
					var txStruct = ScanTxStruct{
						TxHash:        txeg.TxHash().String(),
						Value:         int(tempValue),
						WechatTxId:    hex.EncodeToString(tempPkScript[:]),
						TargetAddress: targetAddress,
						IsNotice:      0,
						NoticedCount:  0,
					}
					if scanTxsCallbackFunc != nil {
						var callbackStatus = scanTxsCallbackFunc(txStruct, nil) //保证kim那边 成功接收
						if !callbackStatus {
							isSuccessAnalyseAllBlock = false
							break AnalyseTxsLabel
						}
					}
					log.Warningf("parker  txHash is %v, tempPkScrip =====> %v ", txeg.TxHash().String(), hex.EncodeToString(tempPkScript[:]))
					var err = ws.txStore.NoticeTxs().Put(txeg.TxHash().String(), int(tempValue), hex.EncodeToString(tempPkScript[:]), targetAddress, 0, 0) // isNotice 0:failure , 1:successful
					if err != nil {
						log.Infof("ws.txStore.ScanBlocks().Put err is =>", err)
						isSuccessAnalyseAllBlock = false
						break AnalyseTxsLabel
					}
					isExistTargetAddress = false
				}
			}
		}
		if isSuccessAnalyseAllBlock {
			var err = ws.txStore.ScanBlocks().UpdateBlock(blockHash.String(), int(1)) // isFixScan 0:failure  1:successful
			if err != nil {
				log.Infof("ws.txStore.ScanBlocks().Put err is ", err)
			}
		}
	}()
}

func scriptToAddress(script []byte, params *chaincfg.Params) (btc.Address, error) {
	_, addrs, _, err := txscript.ExtractPkScriptAddrs(script, params)
	if err != nil {
		return &btc.AddressPubKeyHash{}, err
	}
	if len(addrs) == 0 {
		return &btc.AddressPubKeyHash{}, errors.New("unknown script")
	}
	return addrs[0], nil
}

// handleMerkleBlockMsg handles merkle block messages from all peers.  Merkle blocks are
// requested in response to inv packets both during initial sync and after.
func (ws *WireService) handleMerkleBlockMsg(bmsg *merkleBlockMsg) {
	peer := bmsg.peer

	// We don't need to process blocks when we're syncing. They wont connect anyway
	if peer != ws.syncPeer && !ws.Current() {
		log.Warningf("Received block from %s when we aren't current", peer)
		return
	}
	state, exists := ws.peerStates[peer]
	if !exists {
		log.Warningf("Received merkle block message from unknown peer %s", peer)
		peer.Disconnect()
		return
	}

	// If we didn't ask for this block then the peer is misbehaving.
	merkleBlock := bmsg.merkleBlock
	header := merkleBlock.Header
	blockHash := header.BlockHash()
	if _, exists = state.requestedBlocks[blockHash]; !exists {
		// The regression test intentionally sends some blocks twice
		// to test duplicate block insertion fails.  Don't disconnect
		// the peer or ignore the block when we're in regression test
		// mode in this case so the chain code is actually fed the
		// duplicate blocks.
		if ws.params.Name != chaincfg.RegressionNetParams.Name {
			log.Warningf("Got unrequested block %v from %s -- "+
				"disconnecting", blockHash, peer.Addr())
			peer.Disconnect()
			return
		}
	}

	// Remove block from request maps. Either chain will know about it and
	// so we shouldn't have any more instances of trying to fetch it, or we
	// will fail the insert and thus we'll retry next time we get an inv.
	delete(state.requestedBlocks, blockHash)
	delete(ws.requestedBlocks, blockHash)

	txids, err := checkMBlock(merkleBlock)
	if err != nil {
		log.Warningf("Peer %s sent an invalid MerkleBlock", peer)
		peer.Disconnect()
		return
	}

	newBlock, reorg, newHeight, err := ws.chain.CommitHeader(header)
	// If this is an orphan block which doesn't connect to the chain, it's possible
	// that we might be synced on the longest chain, but not the most-work chain like
	// we should be. To make sure this isn't the case, let's sync from the peer who
	// sent us this orphan block.
	if err == OrphanHeaderError && ws.Current() {
		log.Debug("Received orphan header, checking peer for more blocks")
		state.requestQueue = []*wire.InvVect{}
		state.requestedBlocks = make(map[chainhash.Hash]struct{})
		ws.requestedBlocks = make(map[chainhash.Hash]struct{})
		ws.startSync(peer)
		return
	} else if err == OrphanHeaderError && !ws.Current() {
		// The sync peer sent us an orphan header in the middle of a sync. This could
		// just be the last block in the batch which represents the tip of the chain.
		// In either case let's adjust the score for this peer downwards. If it goes
		// negative it means he's slamming us with blocks that don't fit in our chain
		// so disconnect.
		state.blockScore--
		if state.blockScore < 0 {
			log.Warningf("Disconnecting from peer %s because he sent us too many bad blocks", peer)
			peer.Disconnect()
			return
		}
		log.Warningf("Received unrequested block from peer %s", peer)
		return
	} else if err != nil {
		log.Error(err.Error())
		return
	}
	state.blockScore++

	if ws.Current() {
		peer.UpdateLastBlockHeight(int32(newHeight))
	}

	// Request the transactions in this block
	for _, txid := range txids {
		ws.requestedTxns[*txid] = heightAndTime{newHeight, header.Timestamp}
		limitMap(ws.requestedTxns, maxRequestedTxns)
		state.requestedTxns[*txid] = heightAndTime{newHeight, header.Timestamp}
	}

	// We can exit here if the block is already known
	if !newBlock {
		log.Debugf("Received duplicate block %s", blockHash.String())
		return
	}

	log.Infof("Received merkle block %s at height %d", blockHash.String(), newHeight)

	{
		err = ws.txStore.ScanBlocks().Put(header.BlockHash().String(), int(newHeight), int(0)) // isFixScan 0:failure  1:successful
		if err != nil {
			log.Infof("ws.txStore.ScanBlocks().Put err is ", err)
		}
	}

	// Check reorg
	if reorg != nil && ws.Current() {
		// Rollback the appropriate transactions in our database
		err := ws.txStore.processReorg(reorg.height)
		if err != nil {
			log.Error(err.Error())
		}
		// Set the reorg block as current best block in the header db
		// This will cause a new chain sync from the reorg point
		err = ws.chain.db.Put(*reorg, true)
		if err != nil {
			log.Error(err.Error())
		}

		// Clear request state for new sync
		state.requestQueue = []*wire.InvVect{}
		state.requestedBlocks = make(map[chainhash.Hash]struct{})
		ws.requestedBlocks = make(map[chainhash.Hash]struct{})
	}

	// Clear mempool
	ws.mempool = make(map[chainhash.Hash]struct{})

	// If we're not current and we've downloaded everything we've requested send another getblocks message.
	// Otherwise we'll request the next block in the queue.
	if !ws.Current() && len(state.requestQueue) == 0 {
		locator := ws.chain.GetBlockLocator()
		peer.PushGetBlocksMsg(locator, &ws.zeroHash)
		log.Debug("Request queue at zero. Pushing new locator.")
	} else if !ws.Current() && len(state.requestQueue) > 0 {
		iv := state.requestQueue[0]
		iv.Type = wire.InvTypeFilteredBlock
		state.requestQueue = state.requestQueue[1:]
		state.requestedBlocks[iv.Hash] = struct{}{}
		gdmsg2 := wire.NewMsgGetData()
		gdmsg2.AddInvVect(iv)
		peer.QueueMessage(gdmsg2, nil)
		log.Debugf("Requesting block %s, len request queue: %d", iv.Hash.String(), len(state.requestQueue))
	}
}

// handleInvMsg handles inv messages from all peers.
// We examine the inventory advertised by the remote peer and act accordingly.
func (ws *WireService) handleInvMsg(imsg *invMsg) {
	peer := imsg.peer
	state, exists := ws.peerStates[peer]
	if !exists {
		log.Warningf("Received inv message from unknown peer %s", peer)
		return
	}

	//update the alivePeer
	alivePeer = peer

	// Attempt to find the final block in the inventory list.  There may
	// not be one.
	lastBlock := -1
	invVects := imsg.inv.InvList
	for i := len(invVects) - 1; i >= 0; i-- {
		if invVects[i].Type == wire.InvTypeBlock {
			lastBlock = i
			break
		}
	}

	// If this inv contains a block announcement, and this isn't coming from
	// our current sync peer or we're current, then update the last
	// announced block for this peer. We'll use this information later to
	// update the heights of peers based on blocks we've accepted that they
	// previously announced.
	if lastBlock != -1 && (peer != ws.syncPeer || ws.Current()) {
		peer.UpdateLastAnnouncedBlock(&invVects[lastBlock].Hash)
	}

	// Ignore invs from peers that aren't the sync if we are not current.
	// Helps prevent fetching a mass of orphans.
	if peer != ws.syncPeer && !ws.Current() {
		return
	}

	// If our chain is current and a peer announces a block we already
	// know of, then update their current block height.
	if lastBlock != -1 && ws.Current() {
		sh, err := ws.chain.GetHeader(&invVects[lastBlock].Hash)
		if err == nil {
			peer.UpdateLastBlockHeight(int32(sh.height))
		}
	}

	// Request the advertised inventory if we don't already have it
	gdmsg := wire.NewMsgGetData()
	numRequested := 0
	shouldSendGetData := false
	if len(state.requestQueue) == 0 {
		shouldSendGetData = true
	}
	for _, iv := range invVects {

		// Add the inventory to the cache of known inventory
		// for the peer.
		peer.AddKnownInventory(iv)

		// Request the inventory if we don't already have it.
		haveInv, err := ws.haveInventory(iv)
		if err != nil {
			log.Warningf("Unexpected failure when checking for "+
				"existing inventory during inv message "+
				"processing: %v", err)
			continue
		}

		switch iv.Type {
		case wire.InvTypeFilteredBlock:
			fallthrough
		case wire.InvTypeBlock:
			// Block inventory goes into a request queue to be downloaded
			// one at a time. Sadly we can't batch these because the remote
			// peer  will not update the bloom filter until he's done processing
			// the batch which means we will have a super high false positive rate.
			if _, exists := ws.requestedBlocks[iv.Hash]; (!ws.Current() && !exists && !haveInv && shouldSendGetData) || ws.Current() {
				iv.Type = wire.InvTypeFilteredBlock
				state.requestQueue = append(state.requestQueue, iv)
			}
		case wire.InvTypeTx:
			// Transaction inventory can be requested in batches
			if _, exists := ws.requestedTxns[iv.Hash]; !exists && numRequested < wire.MaxInvPerMsg && !haveInv {
				ws.requestedTxns[iv.Hash] = heightAndTime{0, time.Now()} // unconfirmed tx
				limitMap(ws.requestedTxns, maxRequestedTxns)
				state.requestedTxns[iv.Hash] = heightAndTime{0, time.Now()}

				gdmsg.AddInvVect(iv)
				numRequested++
			}
		default:
			continue
		}
	}

	// Pop the first block off the queue and request it
	if len(state.requestQueue) > 0 && (shouldSendGetData || ws.Current()) {
		iv := state.requestQueue[0]
		gdmsg.AddInvVect(iv)
		if len(state.requestQueue) > 1 {
			state.requestQueue = state.requestQueue[1:]
		} else {
			state.requestQueue = []*wire.InvVect{}
		}
		log.Debugf("Requesting block %s, len request queue: %d", iv.Hash.String(), len(state.requestQueue))
		state.requestedBlocks[iv.Hash] = struct{}{}
	}
	if len(gdmsg.InvList) > 0 {
		peer.QueueMessage(gdmsg, nil)
	}
}

func (ws *WireService) handleTxMsg(tmsg *txMsg) {
	tx := tmsg.tx
	peer := tmsg.peer
	txHash := tx.TxHash()

	state, exists := ws.peerStates[peer]
	if !exists {
		log.Warningf("Received tx message from unknown peer %s", peer)
		return
	}
	ht, ok := state.requestedTxns[tx.TxHash()]
	if !ok {
		log.Warningf("Peer %s is sending us transactions we didn't request", peer)
		peer.Disconnect()
		return
	}
	ws.mempool[txHash] = struct{}{}
	hits, err := ws.txStore.Ingest(tx, int32(ht.height), ht.timestamp)
	if err != nil {
		log.Errorf("Error ingesting tx: %s\n", err.Error())
	}

	// Remove transaction from request maps. Either the mempool/chain
	// already knows about it and as such we shouldn't have any more
	// instances of trying to fetch it, or we failed to insert and thus
	// we'll retry next time we get an inv.
	delete(state.requestedTxns, txHash)
	delete(ws.requestedTxns, txHash)

	// If this transaction had no hits, update the peer's false positive counter
	if hits == 0 {
		log.Debugf("Tx %s from Peer%d had no hits, filter false positive.", txHash.String(), peer.ID())
		state.falsePositives++
	} else {
		log.Debugf("Ingested new tx %s at height %d", txHash.String(), ht.height)
	}

	// Check to see if false positives exceeds the maximum allowed. If so, reset and resend the filter.
	if state.falsePositives > maxFalsePositives {
		state.falsePositives = 0
		ws.updateFilterAndSend(peer)
	}
}

func (ws *WireService) Rebroadcast() {
	// get all unconfirmed txs
	invMsg, err := ws.txStore.GetPendingInv()
	if err != nil {
		log.Errorf("Rebroadcast error: %s", err.Error())
	}
	// Nothing to broadcast, so don't
	if len(invMsg.InvList) == 0 {
		return
	}
	for peer := range ws.peerStates {
		peer.QueueMessage(invMsg, nil)
	}
}

func (ws *WireService) updateFilterAndSend(peer *peerpkg.Peer) {
	if ws.txStore != nil {
		filter, err := ws.txStore.GimmeFilter()
		if err == nil {
			peer.QueueMessage(filter.MsgFilterLoad(), nil)
		} else {
			log.Errorf("Error loading bloom filter: %s", err.Error())
		}
	}
}

// handleUpdateFiltersMsg sends a filter update message to all peers
func (ws *WireService) handleUpdateFiltersMsg() {
	for peer := range ws.peerStates {
		ws.updateFilterAndSend(peer)
	}
}

// haveInventory returns whether or not the inventory represented by the passed
// inventory vector is known.  This includes checking all of the various places
// inventory can be when it is in different states such as blocks that are part
// of the main chain, on a side chain, in the orphan pool, and transactions that
// are in the memory pool (either the main pool or orphan pool).
func (ws *WireService) haveInventory(invVect *wire.InvVect) (bool, error) {
	switch invVect.Type {
	case wire.InvTypeWitnessBlock:
		fallthrough
	case wire.InvTypeBlock:
		// Ask chain if the block is known to it in any form (main
		// chain, side chain, or orphan).
		_, err := ws.chain.GetHeader(&invVect.Hash)
		if err != nil {
			return false, nil
		}
		return true, nil
	case wire.InvTypeTx:
		// Is transaction already in mempool
		if _, ok := ws.mempool[invVect.Hash]; ok {
			return true, nil
		}
		return false, nil
	}
	// The requested inventory is is an unsupported type, so just claim
	// it is known to avoid requesting it.
	return true, nil
}

// limitMap is a helper function for maps that require a maximum limit by
// evicting a random transaction if adding a new value would cause it to
// overflow the maximum allowed.
func limitMap(i interface{}, limit int) {
	m, ok := i.(map[chainhash.Hash]struct{})
	if ok {
		if len(m)+1 > limit {
			// Remove a random entry from the map.  For most compilers, Go's
			// range statement iterates starting at a random item although
			// that is not 100% guaranteed by the spec.  The iteration order
			// is not important here because an adversary would have to be
			// able to pull off preimage attacks on the hashing function in
			// order to target eviction of specific entries anyways.
			for txHash := range m {
				delete(m, txHash)
				return
			}
		}
		return
	}
	n, ok := i.(map[chainhash.Hash]uint32)
	if ok {
		if len(n)+1 > limit {
			for txHash := range n {
				delete(n, txHash)
				return
			}
		}
	}
}