-
Notifications
You must be signed in to change notification settings - Fork 81
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implement direct-send main subroutine
Can be used to spend certain coins, in the manner of coin control. Does not yet support having the wallet choose the coins to spend. So coin control is non-optional for now. Does not yet support spending timelocked utxos.
- Loading branch information
1 parent
99b2111
commit 9b256b1
Showing
4 changed files
with
404 additions
and
74 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,213 @@ | ||
| use std::num::ParseIntError; | ||
| use std::str::FromStr; | ||
|
|
||
| use bitcoin::{Address, Amount, OutPoint, Script, Transaction, TxIn, TxOut}; | ||
|
|
||
| use bitcoincore_rpc::json::ListUnspentResultEntry; | ||
| use bitcoincore_rpc::Client; | ||
|
|
||
| use crate::error::Error; | ||
| use crate::wallet_sync::{UTXOSpendInfo, Wallet, NETWORK, SignTransactionInputInfo}; | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum SendAmount { | ||
| Sweep, | ||
| Amount(Amount), | ||
| } | ||
|
|
||
| impl FromStr for SendAmount { | ||
| type Err = ParseIntError; | ||
|
|
||
| fn from_str(s: &str) -> Result<Self, Self::Err> { | ||
| Ok(if s == "sweep" { | ||
| SendAmount::Sweep | ||
| } else { | ||
| SendAmount::Amount(Amount::from_sat(String::from(s).parse::<u64>()?)) | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum Destination { | ||
| Wallet, | ||
| Address(Address), | ||
| } | ||
|
|
||
| impl FromStr for Destination { | ||
| type Err = bitcoin::util::address::Error; | ||
|
|
||
| fn from_str(s: &str) -> Result<Self, Self::Err> { | ||
| Ok(if s == "wallet" { | ||
| Destination::Wallet | ||
| } else { | ||
| Destination::Address(Address::from_str(s)?) | ||
| }) | ||
| } | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum CoinToSpend { | ||
| LongForm(OutPoint), | ||
| ShortForm { | ||
| prefix: String, | ||
| suffix: String, | ||
| vout: u32, | ||
| }, | ||
| } | ||
|
|
||
| fn parse_short_form_coin(s: &str) -> Option<CoinToSpend> { | ||
| //example short form: 568a4e..83a2e8:0 | ||
| if s.len() < 15 { | ||
| return None; | ||
| } | ||
| let dots = &s[6..8]; | ||
| if dots != ".." { | ||
| return None; | ||
| } | ||
| let colon = s.chars().nth(14).unwrap(); | ||
| if colon != ':' { | ||
| return None; | ||
| } | ||
| let prefix = String::from(&s[0..6]); | ||
| let suffix = String::from(&s[8..14]); | ||
| let vout = *(&s[15..].parse::<u32>().ok()?); | ||
| Some(CoinToSpend::ShortForm { | ||
| prefix, | ||
| suffix, | ||
| vout, | ||
| }) | ||
| } | ||
|
|
||
| impl FromStr for CoinToSpend { | ||
| type Err = bitcoin::blockdata::transaction::ParseOutPointError; | ||
|
|
||
| fn from_str(s: &str) -> Result<Self, Self::Err> { | ||
| let parsed_outpoint = OutPoint::from_str(s); | ||
| if parsed_outpoint.is_ok() { | ||
| Ok(CoinToSpend::LongForm(parsed_outpoint.unwrap())) | ||
| } else { | ||
| let short_form = parse_short_form_coin(s); | ||
| if short_form.is_some() { | ||
| Ok(short_form.unwrap()) | ||
| } else { | ||
| Err(parsed_outpoint.err().unwrap()) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| impl Wallet { | ||
| pub fn create_direct_send( | ||
| &mut self, | ||
| rpc: &Client, | ||
| send_amount: SendAmount, | ||
| destination: Destination, | ||
| coins_to_spend: &[CoinToSpend], | ||
| ) -> Result<Transaction, Error> { | ||
| let mut tx_inputs = Vec::<TxIn>::new(); | ||
| let mut unspent_inputs = Vec::<(ListUnspentResultEntry, UTXOSpendInfo)>::new(); | ||
| //TODO this search within a search could get very slow | ||
| let list_unspent_result = self.list_unspent_from_wallet(rpc)?; | ||
| for (list_unspent_entry, spend_info) in list_unspent_result { | ||
| for cts in coins_to_spend { | ||
| let previous_output = match cts { | ||
| CoinToSpend::LongForm(outpoint) => { | ||
| if list_unspent_entry.txid == outpoint.txid | ||
| && list_unspent_entry.vout == outpoint.vout | ||
| { | ||
| *outpoint | ||
| } else { | ||
| continue; | ||
| } | ||
| } | ||
| CoinToSpend::ShortForm { | ||
| prefix, | ||
| suffix, | ||
| vout, | ||
| } => { | ||
| let txid_hex = list_unspent_entry.txid.to_string(); | ||
| if txid_hex.starts_with(prefix) | ||
| && txid_hex.ends_with(suffix) | ||
| && list_unspent_entry.vout == *vout | ||
| { | ||
| OutPoint { | ||
| txid: list_unspent_entry.txid, | ||
| vout: list_unspent_entry.vout, | ||
| } | ||
| } else { | ||
| continue; | ||
| } | ||
| } | ||
| }; | ||
| tx_inputs.push(TxIn { | ||
| previous_output, | ||
| sequence: 0, | ||
| witness: Vec::new(), | ||
| script_sig: Script::new(), | ||
| }); | ||
| unspent_inputs.push((list_unspent_entry.clone(), spend_info.clone())); | ||
| } | ||
| } | ||
| if tx_inputs.len() != coins_to_spend.len() { | ||
| panic!("unable to find all given inputs, only found = {:?}", tx_inputs); | ||
| } | ||
| let dest_addr = match destination { | ||
| Destination::Wallet => self.get_next_external_address(rpc)?, | ||
| Destination::Address(a) => a, | ||
| }; | ||
| if dest_addr.network != NETWORK { | ||
| panic!("wrong address network type (e.g. testnet, regtest)"); | ||
| } | ||
| let miner_fee = 500; //TODO do this calculation properly | ||
|
|
||
| let mut output = Vec::<TxOut>::new(); | ||
| let total_input_value = unspent_inputs | ||
| .iter() | ||
| .fold(Amount::ZERO, |acc, u| acc + u.0.amount) | ||
| .as_sat(); | ||
| output.push(TxOut { | ||
| script_pubkey: dest_addr.script_pubkey(), | ||
| value: match send_amount { | ||
| SendAmount::Sweep => total_input_value - miner_fee, | ||
| SendAmount::Amount(a) => a.as_sat(), | ||
| }, | ||
| }); | ||
| if let SendAmount::Amount(amount) = send_amount { | ||
| output.push(TxOut { | ||
| script_pubkey: self.get_next_internal_addresses(rpc, 1)?[0].script_pubkey(), | ||
| value: total_input_value - amount.as_sat() - miner_fee, | ||
| }); | ||
| } | ||
|
|
||
| let mut tx = Transaction { | ||
| input: tx_inputs, | ||
| output, | ||
| lock_time: 0, | ||
| version: 2, | ||
| }; | ||
| self.sign_transaction(&mut tx, &mut unspent_inputs.iter() | ||
| .map(|(lure, usi)| | ||
| match usi { | ||
| UTXOSpendInfo::SeedUTXO { path } => { | ||
| SignTransactionInputInfo::SeedCoin { | ||
| path: path.clone(), | ||
| input_value: lure.amount.as_sat() | ||
| } | ||
| } | ||
| UTXOSpendInfo::SwapCoin { multisig_redeemscript } => { | ||
| SignTransactionInputInfo::SwapCoin { | ||
| multisig_redeemscript: multisig_redeemscript.clone() | ||
| } | ||
| } | ||
| UTXOSpendInfo::TimelockContract { swapcoin_multisig_redeemscript } => { | ||
| panic!("not implemented yet"); | ||
| } | ||
| } | ||
| ) | ||
| ); | ||
|
|
||
| println!("tx = {:?}", tx); | ||
|
|
||
| Ok(tx) | ||
| } | ||
| } |
Oops, something went wrong.