From 9abe7fa133cc34090ac610c2954d8c3d36a98b3f Mon Sep 17 00:00:00 2001
From: yukang <moorekang@gmail.com>
Date: Fri, 9 Feb 2024 10:15:48 +0800
Subject: [PATCH] A peer send manlformed tx with large cycles will also be
 banned

---
 test/src/main.rs                          |  1 +
 test/src/specs/relay/transaction_relay.rs | 92 +++++++++++++----------
 tx-pool/src/process.rs                    |  2 +-
 3 files changed, 56 insertions(+), 39 deletions(-)

diff --git a/test/src/main.rs b/test/src/main.rs
index d9b87c9ec3..97317f1fc5 100644
--- a/test/src/main.rs
+++ b/test/src/main.rs
@@ -444,6 +444,7 @@ fn all_specs() -> Vec<Box<dyn Spec>> {
         // TODO failed on poor CI server
         // Box::new(TransactionRelayMultiple),
         Box::new(RelayInvalidTransaction),
+        Box::new(RelayInvalidTransactionResumable),
         Box::new(TransactionRelayTimeout),
         Box::new(TransactionRelayEmptyPeers),
         Box::new(TransactionRelayConflict),
diff --git a/test/src/specs/relay/transaction_relay.rs b/test/src/specs/relay/transaction_relay.rs
index 75a34e0df6..1a9512314c 100644
--- a/test/src/specs/relay/transaction_relay.rs
+++ b/test/src/specs/relay/transaction_relay.rs
@@ -116,49 +116,65 @@ impl Spec for TransactionRelayTimeout {
     }
 }
 
-pub struct RelayInvalidTransaction;
+fn run_relay_malformed_tx_with_cycle(spec: &dyn Spec, nodes: &mut Vec<Node>, cycle: u64) {
+    let node = &nodes.pop().unwrap();
+    node.mine(4);
+    let mut net = Net::new(
+        spec.name(),
+        node.consensus(),
+        vec![SupportProtocols::Sync, SupportProtocols::RelayV3],
+    );
+    net.connect(node);
+    let dummy_tx = TransactionBuilder::default().build();
+    info!("Sending RelayTransactionHashes to node");
+    net.send(
+        node,
+        SupportProtocols::RelayV3,
+        build_relay_tx_hashes(&[dummy_tx.hash()]),
+    );
+    info!("Receiving GetRelayTransactions message from node");
+    assert!(
+        wait_get_relay_txs(&net, node),
+        "timeout to wait GetRelayTransactions"
+    );
+
+    assert!(
+        node.rpc_client().get_banned_addresses().is_empty(),
+        "Banned addresses list should empty"
+    );
+    info!("Sending RelayTransactions to node");
+    net.send(
+        node,
+        SupportProtocols::RelayV3,
+        build_relay_txs(&[(dummy_tx, cycle)]),
+    );
+
+    wait_until(20, || node.rpc_client().get_banned_addresses().len() == 1);
+    let banned_addrs = node.rpc_client().get_banned_addresses();
+    assert_eq!(
+        banned_addrs.len(),
+        1,
+        "Net should be banned: {banned_addrs:?}"
+    );
+}
 
+pub struct RelayInvalidTransaction;
 impl Spec for RelayInvalidTransaction {
     fn run(&self, nodes: &mut Vec<Node>) {
-        let node = &nodes.pop().unwrap();
-        node.mine(4);
-        let mut net = Net::new(
-            self.name(),
-            node.consensus(),
-            vec![SupportProtocols::Sync, SupportProtocols::RelayV3],
-        );
-        net.connect(node);
-        let dummy_tx = TransactionBuilder::default().build();
-        info!("Sending RelayTransactionHashes to node");
-        net.send(
-            node,
-            SupportProtocols::RelayV3,
-            build_relay_tx_hashes(&[dummy_tx.hash()]),
-        );
-        info!("Receiving GetRelayTransactions message from node");
-        assert!(
-            wait_get_relay_txs(&net, node),
-            "timeout to wait GetRelayTransactions"
-        );
+        run_relay_malformed_tx_with_cycle(self, nodes, 444);
+    }
+}
 
-        assert!(
-            node.rpc_client().get_banned_addresses().is_empty(),
-            "Banned addresses list should empty"
-        );
-        info!("Sending RelayTransactions to node");
-        net.send(
-            node,
-            SupportProtocols::RelayV3,
-            build_relay_txs(&[(dummy_tx, 333)]),
-        );
+pub struct RelayInvalidTransactionResumable;
+impl Spec for RelayInvalidTransactionResumable {
+    fn run(&self, nodes: &mut Vec<Node>) {
+        // when a tx with large cycle is processed by resumeble_process_tx
+        // a peer send manlformed tx will also be banned
+        run_relay_malformed_tx_with_cycle(self, nodes, 102);
+    }
 
-        wait_until(20, || node.rpc_client().get_banned_addresses().len() == 1);
-        let banned_addrs = node.rpc_client().get_banned_addresses();
-        assert_eq!(
-            banned_addrs.len(),
-            1,
-            "Net should be banned: {banned_addrs:?}"
-        );
+    fn modify_app_config(&self, config: &mut ckb_app_config::CKBAppConfig) {
+        config.tx_pool.max_tx_verify_cycles = 100;
     }
 }
 
diff --git a/tx-pool/src/process.rs b/tx-pool/src/process.rs
index 41b2d86e8b..85b6d523fc 100644
--- a/tx-pool/src/process.rs
+++ b/tx-pool/src/process.rs
@@ -291,7 +291,7 @@ impl TxPoolService {
         remote: Option<(Cycle, PeerIndex)>,
     ) -> Result<(), Reject> {
         // non contextual verify first
-        self.non_contextual_verify(&tx, None)?;
+        self.non_contextual_verify(&tx, remote)?;
 
         if self.chunk_contains(&tx).await {
             return Err(Reject::Duplicated(tx.hash()));