1 use crate::convert::{BlockchainInfo, FeeResponse, FundedTx, NewAddress, RawTx, SignedTx};
3 use bitcoin::blockdata::block::Block;
4 use bitcoin::blockdata::transaction::Transaction;
5 use bitcoin::consensus::encode;
6 use bitcoin::hash_types::BlockHash;
7 use bitcoin::util::address::Address;
8 use lightning::chain::chaininterface::{BroadcasterInterface, ConfirmationTarget, FeeEstimator};
9 use lightning_block_sync::http::HttpEndpoint;
10 use lightning_block_sync::rpc::RpcClient;
11 use lightning_block_sync::{AsyncBlockSourceResult, BlockHeaderData, BlockSource};
13 use std::collections::HashMap;
14 use std::str::FromStr;
15 use std::sync::atomic::{AtomicU32, Ordering};
17 use std::time::Duration;
18 use tokio::sync::Mutex;
20 pub struct BitcoindClient {
21 bitcoind_rpc_client: Arc<Mutex<RpcClient>>,
26 fees: Arc<HashMap<Target, AtomicU32>>,
29 #[derive(Clone, Eq, Hash, PartialEq)]
36 impl BlockSource for &BitcoindClient {
38 &'a mut self, header_hash: &'a BlockHash, height_hint: Option<u32>,
39 ) -> AsyncBlockSourceResult<'a, BlockHeaderData> {
41 let mut rpc = self.bitcoind_rpc_client.lock().await;
42 rpc.get_header(header_hash, height_hint).await
47 &'a mut self, header_hash: &'a BlockHash,
48 ) -> AsyncBlockSourceResult<'a, Block> {
50 let mut rpc = self.bitcoind_rpc_client.lock().await;
51 rpc.get_block(header_hash).await
55 fn get_best_block<'a>(&'a mut self) -> AsyncBlockSourceResult<(BlockHash, Option<u32>)> {
57 let mut rpc = self.bitcoind_rpc_client.lock().await;
58 rpc.get_best_block().await
63 /// The minimum feerate we are allowed to send, as specify by LDK.
64 const MIN_FEERATE: u32 = 253;
68 host: String, port: u16, rpc_user: String, rpc_password: String,
69 ) -> std::io::Result<Self> {
70 let http_endpoint = HttpEndpoint::for_host(host.clone()).with_port(port);
72 base64::encode(format!("{}:{}", rpc_user.clone(), rpc_password.clone()));
73 let mut bitcoind_rpc_client = RpcClient::new(&rpc_credentials, http_endpoint)?;
74 let _dummy = bitcoind_rpc_client
75 .call_method::<BlockchainInfo>("getblockchaininfo", &vec![])
78 std::io::Error::new(std::io::ErrorKind::PermissionDenied,
79 "Failed to make initial call to bitcoind - please check your RPC user/password and access settings")
81 let mut fees: HashMap<Target, AtomicU32> = HashMap::new();
82 fees.insert(Target::Background, AtomicU32::new(MIN_FEERATE));
83 fees.insert(Target::Normal, AtomicU32::new(2000));
84 fees.insert(Target::HighPriority, AtomicU32::new(5000));
86 bitcoind_rpc_client: Arc::new(Mutex::new(bitcoind_rpc_client)),
93 BitcoindClient::poll_for_fee_estimates(
95 client.bitcoind_rpc_client.clone(),
101 async fn poll_for_fee_estimates(
102 fees: Arc<HashMap<Target, AtomicU32>>, rpc_client: Arc<Mutex<RpcClient>>,
104 tokio::spawn(async move {
106 let background_estimate = {
107 let mut rpc = rpc_client.lock().await;
108 let background_conf_target = serde_json::json!(144);
109 let background_estimate_mode = serde_json::json!("ECONOMICAL");
111 .call_method::<FeeResponse>(
113 &vec![background_conf_target, background_estimate_mode],
117 match resp.feerate_sat_per_kw {
118 Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
123 let normal_estimate = {
124 let mut rpc = rpc_client.lock().await;
125 let normal_conf_target = serde_json::json!(18);
126 let normal_estimate_mode = serde_json::json!("ECONOMICAL");
128 .call_method::<FeeResponse>(
130 &vec![normal_conf_target, normal_estimate_mode],
134 match resp.feerate_sat_per_kw {
135 Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
140 let high_prio_estimate = {
141 let mut rpc = rpc_client.lock().await;
142 let high_prio_conf_target = serde_json::json!(6);
143 let high_prio_estimate_mode = serde_json::json!("CONSERVATIVE");
145 .call_method::<FeeResponse>(
147 &vec![high_prio_conf_target, high_prio_estimate_mode],
152 match resp.feerate_sat_per_kw {
153 Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
158 fees.get(&Target::Background)
160 .store(background_estimate, Ordering::Release);
161 fees.get(&Target::Normal).unwrap().store(normal_estimate, Ordering::Release);
162 fees.get(&Target::HighPriority)
164 .store(high_prio_estimate, Ordering::Release);
165 tokio::time::sleep(Duration::from_secs(60)).await;
170 pub fn get_new_rpc_client(&self) -> std::io::Result<RpcClient> {
171 let http_endpoint = HttpEndpoint::for_host(self.host.clone()).with_port(self.port);
172 let rpc_credentials =
173 base64::encode(format!("{}:{}", self.rpc_user.clone(), self.rpc_password.clone()));
174 RpcClient::new(&rpc_credentials, http_endpoint)
177 pub async fn create_raw_transaction(&self, outputs: Vec<HashMap<String, f64>>) -> RawTx {
178 let mut rpc = self.bitcoind_rpc_client.lock().await;
180 let outputs_json = serde_json::json!(outputs);
181 rpc.call_method::<RawTx>("createrawtransaction", &vec![serde_json::json!([]), outputs_json])
186 pub async fn fund_raw_transaction(&self, raw_tx: RawTx) -> FundedTx {
187 let mut rpc = self.bitcoind_rpc_client.lock().await;
189 let raw_tx_json = serde_json::json!(raw_tx.0);
190 rpc.call_method("fundrawtransaction", &[raw_tx_json]).await.unwrap()
193 pub async fn send_raw_transaction(&self, raw_tx: RawTx) {
194 let mut rpc = self.bitcoind_rpc_client.lock().await;
196 let raw_tx_json = serde_json::json!(raw_tx.0);
197 rpc.call_method::<RawTx>("sendrawtransaction", &[raw_tx_json]).await.unwrap();
200 pub async fn sign_raw_transaction_with_wallet(&self, tx_hex: String) -> SignedTx {
201 let mut rpc = self.bitcoind_rpc_client.lock().await;
203 let tx_hex_json = serde_json::json!(tx_hex);
204 rpc.call_method("signrawtransactionwithwallet", &vec![tx_hex_json]).await.unwrap()
207 pub async fn get_new_address(&self) -> Address {
208 let mut rpc = self.bitcoind_rpc_client.lock().await;
210 let addr_args = vec![serde_json::json!("LDK output address")];
211 let addr = rpc.call_method::<NewAddress>("getnewaddress", &addr_args).await.unwrap();
212 Address::from_str(addr.0.as_str()).unwrap()
215 pub async fn get_blockchain_info(&self) -> BlockchainInfo {
216 let mut rpc = self.bitcoind_rpc_client.lock().await;
217 rpc.call_method::<BlockchainInfo>("getblockchaininfo", &vec![]).await.unwrap()
221 impl FeeEstimator for BitcoindClient {
222 fn get_est_sat_per_1000_weight(&self, confirmation_target: ConfirmationTarget) -> u32 {
223 match confirmation_target {
224 ConfirmationTarget::Background => {
225 self.fees.get(&Target::Background).unwrap().load(Ordering::Acquire)
227 ConfirmationTarget::Normal => {
228 self.fees.get(&Target::Normal).unwrap().load(Ordering::Acquire)
230 ConfirmationTarget::HighPriority => {
231 self.fees.get(&Target::HighPriority).unwrap().load(Ordering::Acquire)
237 impl BroadcasterInterface for BitcoindClient {
238 fn broadcast_transaction(&self, tx: &Transaction) {
239 let bitcoind_rpc_client = self.bitcoind_rpc_client.clone();
240 let tx_serialized = serde_json::json!(encode::serialize_hex(tx));
241 tokio::spawn(async move {
242 let mut rpc = bitcoind_rpc_client.lock().await;
243 // This may error due to RL calling `broadcast_transaction` with the same transaction
244 // multiple times, but the error is safe to ignore.
245 match rpc.call_method::<RawTx>("sendrawtransaction", &vec![tx_serialized]).await {
248 let err_str = e.get_ref().unwrap().to_string();
249 if !err_str.contains("Transaction already in block chain")
250 && !err_str.contains("Inputs missing or spent")
251 && !err_str.contains("bad-txns-inputs-missingorspent")
252 && !err_str.contains("non-BIP68-final")