use crate::error::{Context, Error, ensure};
use crate::{config, transaction::decode};
use crate::{
endpoints::{self, PublicEndpoint},
transport::Pool,
};
use bitcoin::{Transaction, Txid, consensus::deserialize};
use bitcoincore_rpc::{Auth, Client, RpcApi};
use serde::{Deserialize, Serialize};
use serde_json::{Value, json};
use std::path::PathBuf;
use urma_chain::{
observation::Chain,
validation::{BlockValidationError, validate_block_integrity},
};
use urma_identity::identity::IdentitySigner;
use urma_wallet::funding::Funding;
#[derive(Clone)]
pub struct NodeConfig {
pub chain: Chain,
pub rpc_url: String,
pub cookie_file: PathBuf,
}
pub struct Node {
chain: Chain,
backend: Backend,
}
enum Backend {
Local(Client),
Public(Pool),
}
#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Utxo {
pub txid: Txid,
pub vout: u32,
pub value: u64,
pub height: u64,
}
#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[serde(tag = "status", rename_all = "snake_case")]
pub enum Presence {
Missing,
Mempool,
Confirmed { height: u64, block_hash: String },
}
impl Node {
pub fn connect(config: NodeConfig) -> Result<Self, Error> {
let url = url::Url::parse(&config.rpc_url)?;
ensure!(
url.scheme() == "http"
&& url.username().is_empty()
&& url.password().iter().count() == 0
&& url.query().iter().count() == 0
&& url.fragment().iter().count() == 0
&& url.path() == "/",
"RPC requires a local HTTP origin and cookie authentication"
);
let host = url
.host_str()
.context("missing RPC host")?
.trim_matches(['[', ']']);
ensure!(
host.parse::<std::net::IpAddr>()?.is_loopback(),
"RPC must use a loopback IP"
);
let client = Client::new(url.as_str(), Auth::CookieFile(config.cookie_file.clone()))?;
let node = Self {
chain: config.chain,
backend: Backend::Local(client),
};
node.verify_network()?;
Ok(node)
}
pub fn public(chain: Chain) -> Result<Self, Error> {
Self::with_public_sources(chain, endpoints::defaults(chain))
}
pub fn with_public_sources(
chain: Chain,
endpoints: Vec<PublicEndpoint>,
) -> Result<Self, Error> {
let sources = Pool::new(endpoints)?;
let node = Self {
chain,
backend: Backend::Public(sources),
};
node.verify_network()?;
Ok(node)
}
pub fn is_public(&self) -> bool {
matches!(self.backend, Backend::Public(_))
}
pub fn inclusion_evidence(&self) -> &'static str {
match &self.backend {
Backend::Local(_) => "local_validating_node",
Backend::Public(_) => "public_provider_observation",
}
}
pub fn chain(&self) -> Chain {
self.chain
}
pub fn tip_height(&self) -> Result<u64, Error> {
Ok(self.tip()?.0)
}
pub fn block_hash(&self, height: u64) -> Result<bitcoin::BlockHash, Error> {
let value = self.call("getblockhash", &[json!(height)])?;
Ok(value.as_str().context("missing block hash")?.parse()?)
}
pub fn block(&self, height: u64) -> Result<bitcoin::Block, Error> {
let hash = self.block_hash(height)?;
let value = self.call("getblock", &[json!(hash), json!(0)])?;
let block: bitcoin::Block = deserialize(&hex::decode(
value.as_str().context("missing block bytes")?,
)?)?;
validate_block_integrity(&block, hash).map_err(|cause| match cause {
BlockValidationError::HashMismatch => Error::Invalid("RPC block hash mismatch".into()),
BlockValidationError::MerkleRootMismatch => {
Error::Invalid("RPC block merkle root mismatch".into())
}
cause => Error::from(cause),
})?;
Ok(block)
}
pub fn confirmations(&self, txid: Txid) -> Result<u32, Error> {
match self.presence(txid)? {
Presence::Confirmed { height, .. } => Ok(u32::try_from(
self.tip_height()?
.checked_sub(height)
.context("chain changed during confirmation lookup")?
.checked_add(1)
.context("confirmation count overflow")?,
)?),
Presence::Missing | Presence::Mempool => Ok(0),
}
}
pub fn call(&self, method: &str, args: &[Value]) -> Result<Value, Error> {
match &self.backend {
Backend::Local(client) => Ok(client.call(method, args)?),
Backend::Public(sources) => sources.call(self.chain, method, args),
}
}
pub fn require_txindex(&self) -> Result<(), Error> {
if matches!(self.backend, Backend::Public(_)) {
return self.verify_network();
}
let indexes = self.call("getindexinfo", &[json!("txindex")])?;
ensure!(
indexes["txindex"]["synced"] == true,
"publication and recovery require synchronized txindex=1 on the local node"
);
Ok(())
}
pub fn verify_network(&self) -> Result<(), Error> {
let genesis = self.call("getblockhash", &[json!(0)])?;
ensure!(
genesis.as_str() == Some(self.chain().genesis()?.0.to_string().as_str()),
"RPC genesis does not match selected chain"
);
Ok(())
}
pub fn tip(&self) -> Result<(u64, String), Error> {
let info = self.call("getblockchaininfo", &[])?;
Ok((
info["blocks"].as_u64().context("missing height")?,
info["bestblockhash"]
.as_str()
.context("missing tip hash")?
.to_owned(),
))
}
pub fn transaction(&self, txid: Txid) -> Result<Transaction, Error> {
let raw = self.call("getrawtransaction", &[json!(txid), json!(false)])?;
let raw = raw.as_str().context("missing raw transaction")?;
let transaction = decode(raw, raw.len())?;
ensure!(
transaction.compute_txid() == txid,
"RPC transaction hash mismatch"
);
Ok(transaction)
}
pub fn presence(&self, txid: Txid) -> Result<Presence, Error> {
self.presence_cached(txid, &mut std::collections::HashMap::new())
}
pub(crate) fn presence_cached(
&self,
txid: Txid,
blocks: &mut std::collections::HashMap<String, Presence>,
) -> Result<Presence, Error> {
let result = self.call("getrawtransaction", &[json!(txid), json!(true)]);
let value = match result {
Ok(value) => value,
Err(Error::Rpc(bitcoincore_rpc::Error::JsonRpc(
bitcoincore_rpc::jsonrpc::Error::Rpc(error),
))) if error.code == -5 => {
tracing::warn!(code = error.code, "transaction absent from node");
return Ok(Presence::Missing);
}
Err(Error::Missing(message)) => {
tracing::warn!(%message, "transaction absent from public sources");
return Ok(Presence::Missing);
}
Err(error) => return Err(error),
};
if config::confirmations(&value)? <= 0 {
let mempool = self.call("getrawmempool", &[])?;
return Ok(
if mempool
.as_array()
.context("invalid mempool")?
.contains(&json!(txid))
{
Presence::Mempool
} else {
Presence::Missing
},
);
}
let hash = value["blockhash"]
.as_str()
.context("missing inclusion block")?;
let cached = blocks.get(hash);
for presence in cached.iter() {
return Ok((*presence).clone());
}
let header = self.call("getblockheader", &[json!(hash)])?;
let height = header["height"]
.as_u64()
.context("missing inclusion height")?;
let active = self.call("getblockhash", &[json!(height)])?;
if active != json!(hash) {
return Ok(Presence::Missing);
}
let presence = Presence::Confirmed {
height,
block_hash: hash.to_owned(),
};
blocks.insert(hash.to_owned(), presence.clone());
Ok(presence)
}
pub fn utxos(&self, signer: &impl IdentitySigner) -> Result<Vec<Utxo>, Error> {
self.verify_network()?;
if matches!(self.backend, Backend::Public(_)) {
return self.public_utxos(signer);
}
let script = urma_wallet::signing::script(signer)?;
let scan = self.call(
"scantxoutset",
&[
json!("start"),
json!([format!("raw({})", hex::encode(script.as_bytes()))]),
],
)?;
ensure!(scan["success"] == true, "UTXO scan did not complete");
let rows = scan["unspents"].as_array().context("missing UTXO scan")?;
ensure!(rows.len() <= 1000, "UTXO client capacity exceeded");
let mut outputs = Vec::new();
for row in rows {
let amount = bitcoin::Amount::from_str_in(
&row["amount"].to_string(),
bitcoin::Denomination::Bitcoin,
)?;
outputs.push(Utxo {
txid: row["txid"].as_str().context("missing UTXO txid")?.parse()?,
vout: u32::try_from(row["vout"].as_u64().context("missing UTXO vout")?)?,
value: amount.to_sat(),
height: row["height"].as_u64().context("missing UTXO height")?,
});
}
outputs.sort_by_key(|output| (output.value, output.txid, output.vout));
Ok(outputs)
}
fn public_utxos(&self, signer: &impl IdentitySigner) -> Result<Vec<Utxo>, Error> {
let address = urma_wallet::address::receive_address(signer, self.chain)?;
let rows = self.call("addressutxos", &[json!(address)])?;
let rows = rows.as_array().context("missing public UTXOs")?;
ensure!(rows.len() <= 1000, "UTXO client capacity exceeded");
let mut outputs = Vec::new();
for row in rows {
if row["status"]["confirmed"] != true {
continue;
}
outputs.push(Utxo {
txid: row["txid"].as_str().context("missing UTXO txid")?.parse()?,
vout: u32::try_from(row["vout"].as_u64().context("missing UTXO index")?)?,
value: row["value"].as_u64().context("missing UTXO value")?,
height: row["status"]["block_height"]
.as_u64()
.context("missing UTXO height")?,
});
}
outputs.sort_by_key(|output| (output.value, output.txid, output.vout));
Ok(outputs)
}
pub fn available_utxos(&self, signer: &impl IdentitySigner) -> Result<Vec<Utxo>, Error> {
let expected = hex::encode(urma_wallet::signing::script(signer)?.as_bytes());
let mut available = Vec::new();
for output in self.utxos(signer)? {
let unspent = self.call(
"gettxout",
&[json!(output.txid), json!(output.vout), json!(true)],
)?;
if unspent.is_null() {
continue;
}
let confirmations = config::confirmations(&unspent)?;
if confirmations < 1 || (unspent["coinbase"] == true && confirmations < 100) {
continue;
}
ensure!(
unspent["scriptPubKey"]["hex"] == expected,
"live funding script disagrees with identity"
);
let amount = bitcoin::Amount::from_str_in(
&unspent["value"].to_string(),
bitcoin::Denomination::Bitcoin,
)?;
ensure!(
amount.to_sat() == output.value,
"live funding value disagrees with UTXO scan"
);
available.push(output);
}
Ok(available)
}
pub fn select_funding(
&self,
signer: &impl IdentitySigner,
minimum: u64,
) -> Result<Funding, Error> {
for output in self.available_utxos(signer)? {
if output.value < minimum {
continue;
}
let block = self.call("getblockhash", &[json!(output.height)])?;
let raw = self.call(
"getrawtransaction",
&[json!(output.txid), json!(false), block],
)?;
let funding = Funding {
raw_transaction: raw.as_str().context("missing funding bytes")?.to_owned(),
vout: output.vout,
};
let (outpoint, previous) = funding.prevout()?;
ensure!(
outpoint.txid == output.txid && previous.value.to_sat() == output.value,
"funding transaction disagrees with scanned outpoint"
);
ensure!(
previous.script_pubkey == urma_wallet::signing::script(signer)?,
"funding identity mismatch"
);
return Ok(funding);
}
Err(Error::Missing(
"no confirmed identity UTXO large enough; fund its address and confirm first".into(),
))
}
}