Update to LDK 0.0.107
[ldk-sample] / src / bitcoind_client.rs
index 48d54f822e783bbb9dabfc0888d581c593f960df..70928a13576732e0b398589ecbe539f3461d4137 100644 (file)
@@ -3,7 +3,7 @@ use base64;
 use bitcoin::blockdata::block::Block;
 use bitcoin::blockdata::transaction::Transaction;
 use bitcoin::consensus::encode;
-use bitcoin::hash_types::BlockHash;
+use bitcoin::hash_types::{BlockHash, Txid};
 use bitcoin::util::address::Address;
 use lightning::chain::chaininterface::{BroadcasterInterface, ConfirmationTarget, FeeEstimator};
 use lightning_block_sync::http::HttpEndpoint;
@@ -15,15 +15,15 @@ use std::str::FromStr;
 use std::sync::atomic::{AtomicU32, Ordering};
 use std::sync::Arc;
 use std::time::Duration;
-use tokio::sync::Mutex;
 
 pub struct BitcoindClient {
-       bitcoind_rpc_client: Arc<Mutex<RpcClient>>,
+       bitcoind_rpc_client: Arc<RpcClient>,
        host: String,
        port: u16,
        rpc_user: String,
        rpc_password: String,
        fees: Arc<HashMap<Target, AtomicU32>>,
+       handle: tokio::runtime::Handle,
 }
 
 #[derive(Clone, Eq, Hash, PartialEq)]
@@ -35,104 +35,102 @@ pub enum Target {
 
 impl BlockSource for &BitcoindClient {
        fn get_header<'a>(
-               &'a mut self, header_hash: &'a BlockHash, height_hint: Option<u32>,
+               &'a self, header_hash: &'a BlockHash, height_hint: Option<u32>,
        ) -> AsyncBlockSourceResult<'a, BlockHeaderData> {
-               Box::pin(async move {
-                       let mut rpc = self.bitcoind_rpc_client.lock().await;
-                       rpc.get_header(header_hash, height_hint).await
-               })
+               Box::pin(async move { self.bitcoind_rpc_client.get_header(header_hash, height_hint).await })
        }
 
-       fn get_block<'a>(
-               &'a mut self, header_hash: &'a BlockHash,
-       ) -> AsyncBlockSourceResult<'a, Block> {
-               Box::pin(async move {
-                       let mut rpc = self.bitcoind_rpc_client.lock().await;
-                       rpc.get_block(header_hash).await
-               })
+       fn get_block<'a>(&'a self, header_hash: &'a BlockHash) -> AsyncBlockSourceResult<'a, Block> {
+               Box::pin(async move { self.bitcoind_rpc_client.get_block(header_hash).await })
        }
 
-       fn get_best_block<'a>(&'a mut self) -> AsyncBlockSourceResult<(BlockHash, Option<u32>)> {
-               Box::pin(async move {
-                       let mut rpc = self.bitcoind_rpc_client.lock().await;
-                       rpc.get_best_block().await
-               })
+       fn get_best_block<'a>(&'a self) -> AsyncBlockSourceResult<(BlockHash, Option<u32>)> {
+               Box::pin(async move { self.bitcoind_rpc_client.get_best_block().await })
        }
 }
 
+/// The minimum feerate we are allowed to send, as specify by LDK.
+const MIN_FEERATE: u32 = 253;
+
 impl BitcoindClient {
        pub async fn new(
                host: String, port: u16, rpc_user: String, rpc_password: String,
+               handle: tokio::runtime::Handle,
        ) -> std::io::Result<Self> {
                let http_endpoint = HttpEndpoint::for_host(host.clone()).with_port(port);
                let rpc_credentials =
                        base64::encode(format!("{}:{}", rpc_user.clone(), rpc_password.clone()));
                let bitcoind_rpc_client = RpcClient::new(&rpc_credentials, http_endpoint)?;
+               let _dummy = bitcoind_rpc_client
+                       .call_method::<BlockchainInfo>("getblockchaininfo", &vec![])
+                       .await
+                       .map_err(|_| {
+                               std::io::Error::new(std::io::ErrorKind::PermissionDenied,
+                               "Failed to make initial call to bitcoind - please check your RPC user/password and access settings")
+                       })?;
                let mut fees: HashMap<Target, AtomicU32> = HashMap::new();
-               fees.insert(Target::Background, AtomicU32::new(253));
+               fees.insert(Target::Background, AtomicU32::new(MIN_FEERATE));
                fees.insert(Target::Normal, AtomicU32::new(2000));
                fees.insert(Target::HighPriority, AtomicU32::new(5000));
                let client = Self {
-                       bitcoind_rpc_client: Arc::new(Mutex::new(bitcoind_rpc_client)),
+                       bitcoind_rpc_client: Arc::new(bitcoind_rpc_client),
                        host,
                        port,
                        rpc_user,
                        rpc_password,
                        fees: Arc::new(fees),
+                       handle: handle.clone(),
                };
                BitcoindClient::poll_for_fee_estimates(
                        client.fees.clone(),
                        client.bitcoind_rpc_client.clone(),
-               )
-               .await;
+                       handle,
+               );
                Ok(client)
        }
 
-       async fn poll_for_fee_estimates(
-               fees: Arc<HashMap<Target, AtomicU32>>, rpc_client: Arc<Mutex<RpcClient>>,
+       fn poll_for_fee_estimates(
+               fees: Arc<HashMap<Target, AtomicU32>>, rpc_client: Arc<RpcClient>,
+               handle: tokio::runtime::Handle,
        ) {
-               tokio::spawn(async move {
+               handle.spawn(async move {
                        loop {
                                let background_estimate = {
-                                       let mut rpc = rpc_client.lock().await;
                                        let background_conf_target = serde_json::json!(144);
                                        let background_estimate_mode = serde_json::json!("ECONOMICAL");
-                                       let resp = rpc
+                                       let resp = rpc_client
                                                .call_method::<FeeResponse>(
                                                        "estimatesmartfee",
                                                        &vec![background_conf_target, background_estimate_mode],
                                                )
                                                .await
                                                .unwrap();
-                                       match resp.feerate {
-                                               Some(fee) => fee,
-                                               None => 253,
+                                       match resp.feerate_sat_per_kw {
+                                               Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
+                                               None => MIN_FEERATE,
                                        }
                                };
-                               // if background_estimate.
 
                                let normal_estimate = {
-                                       let mut rpc = rpc_client.lock().await;
                                        let normal_conf_target = serde_json::json!(18);
                                        let normal_estimate_mode = serde_json::json!("ECONOMICAL");
-                                       let resp = rpc
+                                       let resp = rpc_client
                                                .call_method::<FeeResponse>(
                                                        "estimatesmartfee",
                                                        &vec![normal_conf_target, normal_estimate_mode],
                                                )
                                                .await
                                                .unwrap();
-                                       match resp.feerate {
-                                               Some(fee) => fee,
+                                       match resp.feerate_sat_per_kw {
+                                               Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
                                                None => 2000,
                                        }
                                };
 
                                let high_prio_estimate = {
-                                       let mut rpc = rpc_client.lock().await;
                                        let high_prio_conf_target = serde_json::json!(6);
                                        let high_prio_estimate_mode = serde_json::json!("CONSERVATIVE");
-                                       let resp = rpc
+                                       let resp = rpc_client
                                                .call_method::<FeeResponse>(
                                                        "estimatesmartfee",
                                                        &vec![high_prio_conf_target, high_prio_estimate_mode],
@@ -140,8 +138,8 @@ impl BitcoindClient {
                                                .await
                                                .unwrap();
 
-                                       match resp.feerate {
-                                               Some(fee) => fee,
+                                       match resp.feerate_sat_per_kw {
+                                               Some(feerate) => std::cmp::max(feerate, MIN_FEERATE),
                                                None => 5000,
                                        }
                                };
@@ -153,19 +151,6 @@ impl BitcoindClient {
                                fees.get(&Target::HighPriority)
                                        .unwrap()
                                        .store(high_prio_estimate, Ordering::Release);
-                               // match fees.get(Target::Background) {
-                               //     Some(fee) => fee.store(background_estimate, Ordering::Release),
-                               //     None =>
-                               // }
-                               // if let Some(fee) = background_estimate.feerate {
-                               //     fees.get("background").unwrap().store(fee, Ordering::Release);
-                               // }
-                               // if let Some(fee) = normal_estimate.feerate {
-                               //     fees.get("normal").unwrap().store(fee, Ordering::Release);
-                               // }
-                               // if let Some(fee) = high_prio_estimate.feerate {
-                               //     fees.get("high_prio").unwrap().store(fee, Ordering::Release);
-                               // }
                                tokio::time::sleep(Duration::from_secs(60)).await;
                        }
                });
@@ -179,46 +164,67 @@ impl BitcoindClient {
        }
 
        pub async fn create_raw_transaction(&self, outputs: Vec<HashMap<String, f64>>) -> RawTx {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-
                let outputs_json = serde_json::json!(outputs);
-               rpc.call_method::<RawTx>("createrawtransaction", &vec![serde_json::json!([]), outputs_json])
+               self.bitcoind_rpc_client
+                       .call_method::<RawTx>(
+                               "createrawtransaction",
+                               &vec![serde_json::json!([]), outputs_json],
+                       )
                        .await
                        .unwrap()
        }
 
        pub async fn fund_raw_transaction(&self, raw_tx: RawTx) -> FundedTx {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-
                let raw_tx_json = serde_json::json!(raw_tx.0);
-               rpc.call_method("fundrawtransaction", &[raw_tx_json]).await.unwrap()
+               let options = serde_json::json!({
+                       // LDK gives us feerates in satoshis per KW but Bitcoin Core here expects fees
+                       // denominated in satoshis per vB. First we need to multiply by 4 to convert weight
+                       // units to virtual bytes, then divide by 1000 to convert KvB to vB.
+                       "fee_rate": self.get_est_sat_per_1000_weight(ConfirmationTarget::Normal) as f64 / 250.0,
+                       // While users could "cancel" a channel open by RBF-bumping and paying back to
+                       // themselves, we don't allow it here as its easy to have users accidentally RBF bump
+                       // and pay to the channel funding address, which results in loss of funds. Real
+                       // LDK-based applications should enable RBF bumping and RBF bump either to a local
+                       // change address or to a new channel output negotiated with the same node.
+                       "replaceable": false,
+               });
+               self.bitcoind_rpc_client
+                       .call_method("fundrawtransaction", &[raw_tx_json, options])
+                       .await
+                       .unwrap()
        }
 
        pub async fn send_raw_transaction(&self, raw_tx: RawTx) {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-
                let raw_tx_json = serde_json::json!(raw_tx.0);
-               rpc.call_method::<RawTx>("sendrawtransaction", &[raw_tx_json]).await.unwrap();
+               self.bitcoind_rpc_client
+                       .call_method::<Txid>("sendrawtransaction", &[raw_tx_json])
+                       .await
+                       .unwrap();
        }
 
        pub async fn sign_raw_transaction_with_wallet(&self, tx_hex: String) -> SignedTx {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-
                let tx_hex_json = serde_json::json!(tx_hex);
-               rpc.call_method("signrawtransactionwithwallet", &vec![tx_hex_json]).await.unwrap()
+               self.bitcoind_rpc_client
+                       .call_method("signrawtransactionwithwallet", &vec![tx_hex_json])
+                       .await
+                       .unwrap()
        }
 
        pub async fn get_new_address(&self) -> Address {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-
                let addr_args = vec![serde_json::json!("LDK output address")];
-               let addr = rpc.call_method::<NewAddress>("getnewaddress", &addr_args).await.unwrap();
+               let addr = self
+                       .bitcoind_rpc_client
+                       .call_method::<NewAddress>("getnewaddress", &addr_args)
+                       .await
+                       .unwrap();
                Address::from_str(addr.0.as_str()).unwrap()
        }
 
        pub async fn get_blockchain_info(&self) -> BlockchainInfo {
-               let mut rpc = self.bitcoind_rpc_client.lock().await;
-               rpc.call_method::<BlockchainInfo>("getblockchaininfo", &vec![]).await.unwrap()
+               self.bitcoind_rpc_client
+                       .call_method::<BlockchainInfo>("getblockchaininfo", &vec![])
+                       .await
+                       .unwrap()
        }
 }
 
@@ -235,43 +241,6 @@ impl FeeEstimator for BitcoindClient {
                                self.fees.get(&Target::HighPriority).unwrap().load(Ordering::Acquire)
                        }
                }
-               // self.fees.g
-               // 253
-               // match confirmation_target {
-               //     ConfirmationTarget::Background =>
-               // }
-               // let mut rpc = self.bitcoind_rpc_client.lock().unwrap();
-
-               // let (conf_target, estimate_mode, default) = match confirmation_target {
-               //      ConfirmationTarget::Background => (144, "ECONOMICAL", 253),
-               //      ConfirmationTarget::Normal => (18, "ECONOMICAL", 20000),
-               //      ConfirmationTarget::HighPriority => (6, "CONSERVATIVE", 50000),
-               // };
-
-               // // This function may be called from a tokio runtime, or not. So we need to check before
-               // // making the call to avoid the error "cannot run a tokio runtime from within a tokio runtime".
-               // let conf_target_json = serde_json::json!(conf_target);
-               // let estimate_mode_json = serde_json::json!(estimate_mode);
-               // let resp = match Handle::try_current() {
-               //      Ok(_) => tokio::task::block_in_place(|| {
-               //              runtime
-               //                      .block_on(rpc.call_method::<FeeResponse>(
-               //                              "estimatesmartfee",
-               //                              &vec![conf_target_json, estimate_mode_json],
-               //                      ))
-               //                      .unwrap()
-               //      }),
-               //      _ => runtime
-               //              .block_on(rpc.call_method::<FeeResponse>(
-               //                      "estimatesmartfee",
-               //                      &vec![conf_target_json, estimate_mode_json],
-               //              ))
-               //              .unwrap(),
-               // };
-               // if resp.errored {
-               //      return default;
-               // }
-               // resp.feerate.unwrap()
        }
 }
 
@@ -279,36 +248,27 @@ impl BroadcasterInterface for BitcoindClient {
        fn broadcast_transaction(&self, tx: &Transaction) {
                let bitcoind_rpc_client = self.bitcoind_rpc_client.clone();
                let tx_serialized = serde_json::json!(encode::serialize_hex(tx));
-               tokio::spawn(async move {
-                       let mut rpc = bitcoind_rpc_client.lock().await;
-                       rpc.call_method::<RawTx>("sendrawtransaction", &vec![tx_serialized]).await.unwrap();
+               self.handle.spawn(async move {
+                       // This may error due to RL calling `broadcast_transaction` with the same transaction
+                       // multiple times, but the error is safe to ignore.
+                       match bitcoind_rpc_client
+                               .call_method::<Txid>("sendrawtransaction", &vec![tx_serialized])
+                               .await
+                       {
+                               Ok(_) => {}
+                               Err(e) => {
+                                       let err_str = e.get_ref().unwrap().to_string();
+                                       if !err_str.contains("Transaction already in block chain")
+                                               && !err_str.contains("Inputs missing or spent")
+                                               && !err_str.contains("bad-txns-inputs-missingorspent")
+                                               && !err_str.contains("txn-mempool-conflict")
+                                               && !err_str.contains("non-BIP68-final")
+                                               && !err_str.contains("insufficient fee, rejecting replacement ")
+                                       {
+                                               panic!("{}", e);
+                                       }
+                               }
+                       }
                });
-               // let bitcoind_rpc_client = self.bitcoind_rpc_client.clone();
-               // tokio::spawn(async move {
-               //     let rpc = bitcoind_rpc_client.lock().await;
-               //     rpc.call_method::<R>
-               // });
-               // let mut rpc = self.bitcoind_rpc_client.lock().unwrap();
-               // let runtime = self.runtime.lock().unwrap();
-
-               // let tx_serialized = serde_json::json!(encode::serialize_hex(tx));
-               // // This function may be called from a tokio runtime, or not. So we need to check before
-               // // making the call to avoid the error "cannot run a tokio runtime from within a tokio runtime".
-               // match Handle::try_current() {
-               //      Ok(_) => {
-               //              tokio::task::block_in_place(|| {
-               //                      runtime
-               //                              .block_on(
-               //                                      rpc.call_method::<RawTx>("sendrawtransaction", &vec![tx_serialized]),
-               //                              )
-               //                              .unwrap();
-               //              });
-               //      }
-               //      _ => {
-               //              runtime
-               //                      .block_on(rpc.call_method::<RawTx>("sendrawtransaction", &vec![tx_serialized]))
-               //                      .unwrap();
-               //      }
-               // }
        }
 }