1// Copyright (c) 2021-present The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5#include <rpc/server_util.h>
6
7#include <chain.h>
8#include <node/context.h>
9#include <node/miner.h>
10#include <pow.h>
11#include <primitives/block.h>
12#include <rpc/protocol.h>
13#include <rpc/request.h>
14#include <uint256.h>
15#include <util/any.h>
16
17#include <any>
18#include <memory>
19#include <string>
20
21namespace Consensus {
22struct Params;
23} // namespace Consensus
24
25using node::NodeContext;
26using node::UpdateTime;
27
28NodeContext& EnsureAnyNodeContext(const std::any& context)
29{
30    auto node_context = util::AnyPtr<NodeContext>(context);
31    if (!node_context) {
32        throw JSONRPCError(RPC_INTERNAL_ERROR, "Node context not found");
33    }
34    return *node_context;
35}
36
37CTxMemPool& EnsureMemPool(const NodeContext& node)
38{
39    if (!node.mempool) {
40        throw JSONRPCError(RPC_CLIENT_MEMPOOL_DISABLED, "Mempool disabled or instance not found");
41    }
42    return *node.mempool;
43}
44
45CTxMemPool& EnsureAnyMemPool(const std::any& context)
46{
47    return EnsureMemPool(EnsureAnyNodeContext(context));
48}
49
50
51BanMan& EnsureBanman(const NodeContext& node)
52{
53    if (!node.banman) {
54        throw JSONRPCError(RPC_DATABASE_ERROR, "Error: Ban database not loaded");
55    }
56    return *node.banman;
57}
58
59BanMan& EnsureAnyBanman(const std::any& context)
60{
61    return EnsureBanman(EnsureAnyNodeContext(context));
62}
63
64ArgsManager& EnsureArgsman(const NodeContext& node)
65{
66    if (!node.args) {
67        throw JSONRPCError(RPC_INTERNAL_ERROR, "Node args not found");
68    }
69    return *node.args;
70}
71
72ArgsManager& EnsureAnyArgsman(const std::any& context)
73{
74    return EnsureArgsman(EnsureAnyNodeContext(context));
75}
76
77ChainstateManager& EnsureChainman(const NodeContext& node)
78{
79    if (!node.chainman) {
80        throw JSONRPCError(RPC_INTERNAL_ERROR, "Node chainman not found");
81    }
82    return *node.chainman;
83}
84
85ChainstateManager& EnsureAnyChainman(const std::any& context)
86{
87    return EnsureChainman(EnsureAnyNodeContext(context));
88}
89
90FeeRateEstimatorManager& EnsureFeeEstimatorMan(const NodeContext& node)
91{
92    if (!node.fee_estimator_man) {
93        throw JSONRPCError(RPC_INTERNAL_ERROR, "Fee estimation disabled");
94    }
95    return *node.fee_estimator_man;
96}
97
98FeeRateEstimatorManager& EnsureAnyFeeEstimatorMan(const std::any& context)
99{
100    return EnsureFeeEstimatorMan(EnsureAnyNodeContext(context));
101}
102
103CConnman& EnsureConnman(const NodeContext& node)
104{
105    if (!node.connman) {
106        throw JSONRPCError(RPC_CLIENT_P2P_DISABLED, "Error: Peer-to-peer functionality missing or disabled");
107    }
108    return *node.connman;
109}
110
111interfaces::Mining& EnsureMining(const NodeContext& node)
112{
113    if (!node.mining) {
114        throw JSONRPCError(RPC_INTERNAL_ERROR, "Node miner not found");
115    }
116    return *node.mining;
117}
118
119PeerManager& EnsurePeerman(const NodeContext& node)
120{
121    if (!node.peerman) {
122        throw JSONRPCError(RPC_CLIENT_P2P_DISABLED, "Error: Peer-to-peer functionality missing or disabled");
123    }
124    return *node.peerman;
125}
126
127AddrMan& EnsureAddrman(const NodeContext& node)
128{
129    if (!node.addrman) {
130        throw JSONRPCError(RPC_CLIENT_P2P_DISABLED, "Error: Address manager functionality missing or disabled");
131    }
132    return *node.addrman;
133}
134
135AddrMan& EnsureAnyAddrman(const std::any& context)
136{
137    return EnsureAddrman(EnsureAnyNodeContext(context));
138}
139
140void NextEmptyBlockIndex(CBlockIndex& tip, const Consensus::Params& consensusParams, CBlockIndex& next_index)
141{
142    CBlockHeader next_header{};
143    next_header.hashPrevBlock  = tip.GetBlockHash();
144    UpdateTime(&next_header, consensusParams, &tip);
145    next_header.nBits = GetNextWorkRequired(&tip, &next_header, consensusParams);
146    next_header.nNonce = 0;
147
148    next_index.pprev = &tip;
149    next_index.nTime = next_header.nTime;
150    next_index.nBits = next_header.nBits;
151    next_index.nNonce = next_header.nNonce;
152    next_index.nHeight = tip.nHeight + 1;
153}
154