netsync: add txs msg (#78)

[bytom/vapor.git] / netsync / handle.go

diff --git a/netsync/handle.go b/netsync/handle.go
index ce10e1a..7dc35b9 100644 (file)
--- a/netsync/handle.go
+++ b/netsync/handle.go
@@ -323,6 +323,28 @@ func (sm *SyncManager) handleTransactionMsg(peer *peer, msg *TransactionMessage)
        if isOrphan, err := sm.chain.ValidateTx(tx); err != nil && err != core.ErrDustTx && !isOrphan {
                sm.peers.addBanScore(peer.ID(), 10, 0, "fail on validate tx transaction")
        }
+       sm.peers.markTx(peer.ID(), tx.ID)
+}
+
+func (sm *SyncManager) handleTransactionsMsg(peer *peer, msg *TransactionsMessage) {
+       txs, err := msg.GetTransactions()
+       if err != nil {
+               sm.peers.addBanScore(peer.ID(), 0, 20, "fail on get txs from message")
+               return
+       }
+
+       if len(txs) > txsMsgMaxTxNum {
+               sm.peers.addBanScore(peer.ID(), 20, 0, "exceeded the maximum tx number limit")
+               return
+       }
+
+       for _, tx := range txs {
+               if isOrphan, err := sm.chain.ValidateTx(tx); err != nil && !isOrphan {
+                       sm.peers.addBanScore(peer.ID(), 10, 0, "fail on validate tx transaction")
+                       return
+               }
+               sm.peers.markTx(peer.ID(), tx.ID)
+       }
 }
 
 func (sm *SyncManager) IsListening() bool {
@@ -365,6 +387,9 @@ func (sm *SyncManager) processMsg(basePeer BasePeer, msgType byte, msg Blockchai
        case *TransactionMessage:
                sm.handleTransactionMsg(peer, msg)
 
+       case *TransactionsMessage:
+               sm.handleTransactionsMsg(peer, msg)
+
        case *MineBlockMessage:
                sm.handleMineBlockMsg(peer, msg)