add getextrapoolinfo rpc

Author: 1440000bytes

src/net_processing.cpp

@@ -522,6 +522,9 @@ class PeerManagerImpl final : public PeerManager
     ServiceFlags GetDesirableServiceFlags(ServiceFlags services) const override;
     int GetNumberOfPeersWithValidatedDownloads() const override EXCLUSIVE_LOCKS_REQUIRED(::cs_main);
 
+    size_t ExtraTxnForCompactCount() const override EXCLUSIVE_LOCKS_REQUIRED(::cs_main);
+    size_t ExtraTxnForCompactBytes() const override EXCLUSIVE_LOCKS_REQUIRED(::cs_main);
+
 private:
     /** Consider evicting an outbound peer based on the amount of time they've been behind our tip */
     void ConsiderEviction(CNode& pto, Peer& peer, std::chrono::seconds time_in_seconds) EXCLUSIVE_LOCKS_REQUIRED(cs_main, g_msgproc_mutex);
@@ -1063,6 +1066,19 @@ CNodeState* PeerManagerImpl::State(NodeId pnode)
     return const_cast<CNodeState*>(std::as_const(*this).State(pnode));
 }
 
+size_t PeerManagerImpl::ExtraTxnForCompactCount() const {
+    AssertLockHeld(::cs_main);
+    size_t count = 0;
+    for (const auto& tx : vExtraTxnForCompact) {
+        if (tx) count++;
+    }
+    return count;
+}
+size_t PeerManagerImpl::ExtraTxnForCompactBytes() const {
+    AssertLockHeld(::cs_main);
+    return blockreconstructionextratxn_memusage;
+}
+
 /**
  * Whether the peer supports the address. For example, a peer that does not
  * implement BIP155 cannot receive Tor v3 addresses because it requires

src/net_processing.h

@@ -113,6 +113,9 @@ class PeerManager : public CValidationInterface, public NetEventsInterface
     /** Get peer manager info. */
     virtual PeerManagerInfo GetInfo() const = 0;
 
+    virtual size_t ExtraTxnForCompactCount() const = 0;
+    virtual size_t ExtraTxnForCompactBytes() const = 0;
+
     /** Relay transaction to all peers. */
     virtual void RelayTransaction(const uint256& txid, const uint256& wtxid) = 0;
 

src/rpc/mempool.cpp

@@ -1064,6 +1064,36 @@ static RPCHelpMan getmempoolinfo()
     };
 }
 
+static RPCHelpMan getextrapoolinfo()
+{
+    return RPCHelpMan{"getextrapoolinfo",
+        "Returns details about the extra tx pool used for compact block reconstruction.\n",
+        {},
+        RPCResult{
+            RPCResult::Type::OBJ, "", "",
+            {
+                {RPCResult::Type::NUM, "count", "Number of transactions in the extra pool"},
+                {RPCResult::Type::NUM, "bytes", "Sum of virtual transaction size for all transactions in the extra pool"},
+            }},
+        RPCExamples{
+            HelpExampleCli("getextrapoolinfo", "") +
+            HelpExampleRpc("getextrapoolinfo", "")
+        },
+        [&](const RPCHelpMan& self, const JSONRPCRequest& request) -> UniValue
+        {
+            const NodeContext& node = EnsureAnyNodeContext(request.context);
+            PeerManager& peerman = EnsurePeerman(node);
+
+            LOCK(::cs_main);
+
+            UniValue ret(UniValue::VOBJ);
+            ret.pushKV("count", (uint64_t)peerman.ExtraTxnForCompactCount());
+            ret.pushKV("bytes", (uint64_t)peerman.ExtraTxnForCompactBytes());
+            return ret;
+        },
+    };
+}
+
 static RPCHelpMan importmempool()
 {
     return RPCHelpMan{
@@ -1480,6 +1510,7 @@ void RegisterMempoolRPCCommands(CRPCTable& t)
         {"blockchain", &getmempoolentry},
         {"blockchain", &gettxspendingprevout},
         {"blockchain", &getmempoolinfo},
+        {"blockchain", &getextrapoolinfo},
         {"blockchain", &getrawmempool},
         {"blockchain", &importmempool},
         {"blockchain", &savemempool},