urma_node/

source.rs

use crate::{
    Error,
    model::{ChainPoint, CommitSourceStatus},
    recorded::RecordedError,
};
use bitcoin::{BlockHash, Txid, block::Header};
use std::{
    net::SocketAddr,
    path::{Path, PathBuf},
    sync::{
        Arc, Mutex, MutexGuard,
        atomic::{AtomicBool, Ordering},
    },
    time::{Duration, Instant},
};
use urma_chain::{decode, observation::Chain};
use urma_runtime::{
    node::{Node, NodeConfig},
    p2p::{Anchor, HeaderBootstrap, P2pProvider},
};

/// Port implemented by a chain verifier, not by an untrusted remote JSON API.
/// The implementation owns header difficulty, PoW and branch selection. Bodies
/// remain untrusted bytes and are checked again against the selected header.
/// Provider-claimed observations used by autonomous clients are a different
/// contract and do not enter this port by relabelling them as verified.
pub trait ChainSource: Send + Sync {
    fn genesis(&self) -> Result<BlockHash, Error>;
    fn tip(&self) -> Result<ChainPoint, Error>;
    fn header(&self, height: u64) -> Result<Header, Error>;
    fn block(&self, height: u64) -> Result<Vec<u8>, Error>;
    /// Connected chain peers now; 0 for sources without a peer network.
    fn chain_peers(&self) -> Result<u64, Error> {
        Ok(0)
    }
    /// Absolute header-sync target announced by chain peers; `None` while
    /// unknown or for sources without a peer network.
    fn chain_target(&self) -> Option<u64> {
        None
    }
}

/// Untrusted auxiliary parent bytes, not a chain verifier. The chain getter
/// reads local configuration without I/O. Each fetch must bound its own I/O;
/// Engine cancellation is cooperative and cannot interrupt a running fetch.
pub trait CommitSource {
    fn chain(&self) -> Chain;
    fn fetch(&self, txid: Txid, cancelled: &AtomicBool) -> Result<Option<Vec<u8>>, Error>;
    /// A liveness check for status, rate-limited by the source itself. `None`
    /// when no fresh check ran (or the source has none); otherwise its result.
    fn probe(&self) -> Option<Result<(), Error>> {
        None
    }
}

pub(crate) fn check_cancelled(cancelled: &AtomicBool) -> Result<(), Error> {
    if cancelled.load(Ordering::Acquire) {
        return Err(Error::Busy("dependency acquisition cancelled".into()));
    }
    Ok(())
}

/// Decode with the Engine's chain decoder, so Litecoin MWEB/HogEx encodings
/// are accepted, and require the computed TXID to equal the requested one.
pub fn verify_parent(chain: Chain, txid: Txid, raw: &[u8]) -> Result<(), Error> {
    let transaction = decode::transaction(raw, chain)
        .map_err(|error| Error::Invalid(format!("commit parent bytes do not decode: {error}")))?;
    if transaction.compute_txid() != txid {
        return Err(Error::Invalid(
            "commit parent bytes do not match the requested TXID".into(),
        ));
    }
    Ok(())
}

/// Share an explicitly configured, owner-provided runtime Node. Constructing
/// this wrapper or reading its chain creates no Node, connection or defaults.
/// Only fetch calls the existing Node; no header/placement authority is added.
pub struct RuntimeCommitSource {
    pub node: Arc<Node>,
}

impl CommitSource for RuntimeCommitSource {
    fn chain(&self) -> Chain {
        self.node.chain()
    }

    /// Preserve the original returned hex bytes, rather than reserializing a
    /// normalized transaction. The 16M-character/8M-byte admission cap applies
    /// after Node::call: its transport may allocate/decode first and owns its
    /// timeouts. There is no global deadline or transport-allocation promise.
    /// Absence is the runtime's explicit Missing result or a local JSON-RPC
    /// error with code -5 (no such mempool/blockchain transaction, including
    /// lookups a node without -txindex cannot answer), per is_missing_transaction.
    /// Other runtime errors become CommitSource without text matching.
    fn fetch(&self, txid: Txid, cancelled: &AtomicBool) -> Result<Option<Vec<u8>>, Error> {
        check_cancelled(cancelled)?;
        fetch_from_node(&self.node, txid)
    }
}

/// Optional local Litecoin Core: it may start, stop or restart at any time.
/// Holds only the URL and cookie path; every lookup and probe reads the cookie,
/// and a node is rebuilt and its txindex verified whenever the cookie changes.
pub struct LocalRpcCommitSource {
    chain: Chain,
    rpc_url: String,
    cookie_file: PathBuf,
    probe_interval: Duration,
    verified: Mutex<Option<(Vec<u8>, Arc<Node>)>>,
    probed: Mutex<Option<Instant>>,
}

impl LocalRpcCommitSource {
    pub const PROBE_INTERVAL: Duration = Duration::from_secs(10);

    pub fn new(chain: Chain, rpc_url: String, cookie_file: PathBuf) -> Self {
        Self {
            chain,
            rpc_url,
            cookie_file,
            probe_interval: Self::PROBE_INTERVAL,
            verified: Mutex::new(None),
            probed: Mutex::new(None),
        }
    }

    pub fn with_probe_interval(mut self, interval: Duration) -> Self {
        self.probe_interval = interval;
        self
    }

    fn node(&self) -> Result<Arc<Node>, Error> {
        let cookie = std::fs::read(&self.cookie_file).map_err(|_| {
            Error::CommitSource("cookie file missing: litecoind not running?".into())
        })?;
        let mut verified = self
            .verified
            .lock()
            .map_err(|_| Error::Invalid("local RPC commit source poisoned".into()))?;
        if let Some((seen, node)) = verified.as_ref() {
            if *seen == cookie {
                return Ok(Arc::clone(node));
            }
        }
        *verified = None;
        let node = Node::connect(NodeConfig {
            chain: self.chain,
            rpc_url: self.rpc_url.clone(),
            cookie_file: self.cookie_file.clone(),
        })
        .and_then(|node| node.require_txindex().map(|()| Arc::new(node)))
        .map_err(|error| Error::CommitSource(error.to_string()))?;
        *verified = Some((cookie, Arc::clone(&node)));
        Ok(node)
    }

    fn probe_due(&self) -> Result<bool, Error> {
        let mut probed = self
            .probed
            .lock()
            .map_err(|_| Error::Invalid("local RPC probe clock poisoned".into()))?;
        let due = probed.is_none_or(|last| last.elapsed() >= self.probe_interval);
        if due {
            *probed = Some(Instant::now());
        }
        Ok(due)
    }
}

impl CommitSource for LocalRpcCommitSource {
    fn chain(&self) -> Chain {
        self.chain
    }

    fn fetch(&self, txid: Txid, cancelled: &AtomicBool) -> Result<Option<Vec<u8>>, Error> {
        check_cancelled(cancelled)?;
        let node = self.node()?;
        fetch_from_node(&node, txid)
    }

    fn probe(&self) -> Option<Result<(), Error>> {
        match self.probe_due() {
            Ok(false) => None,
            Ok(true) => Some(self.node().and_then(|node| {
                node.call("getblockcount", &[])
                    .map(|_| ())
                    .map_err(|error| Error::CommitSource(error.to_string()))
            })),
            Err(error) => Some(Err(error)),
        }
    }
}

fn fetch_from_node(node: &Node, txid: Txid) -> Result<Option<Vec<u8>>, Error> {
    let result = match node.call(
        "getrawtransaction",
        &[serde_json::json!(txid), serde_json::json!(false)],
    ) {
        Ok(result) => result,
        Err(error) if error.is_missing_transaction() => return Ok(None),
        Err(error) => return Err(Error::CommitSource(error.to_string())),
    };
    let encoded = result
        .as_str()
        .ok_or_else(|| Error::Invalid("commit source must return transaction hex".into()))?;
    if encoded.len() > 16_000_000 || encoded.len() % 2 != 0 {
        return Err(Error::Invalid(
            "commit source hex exceeds capacity or has odd length".into(),
        ));
    }
    let raw = hex::decode(encoded)
        .map_err(|error| Error::Invalid(format!("commit source hex: {error}")))?;
    Ok(Some(raw))
}

/// Default public providers for one chain, without a light client. Creating
/// it does no I/O; the first fetch builds the Node. A failed build is not
/// rebuilt for RETRY and its error is returned meanwhile. Untrusted bytes.
pub struct PublicCommitSource {
    chain: Chain,
    cache_dir: PathBuf,
    state: Mutex<PublicState>,
}

struct PublicState {
    node: Option<Arc<Node>>,
    failure: Option<(Instant, String)>,
}

impl PublicCommitSource {
    const RETRY: Duration = Duration::from_secs(60);

    pub fn new(chain: Chain, cache_dir: PathBuf) -> Self {
        Self {
            chain,
            cache_dir,
            state: Mutex::new(PublicState {
                node: None,
                failure: None,
            }),
        }
    }

    fn node(&self) -> Result<Arc<Node>, Error> {
        let mut state = self
            .state
            .lock()
            .map_err(|_| Error::Invalid("public commit source poisoned".into()))?;
        if let Some(node) = state.node.as_ref() {
            return Ok(Arc::clone(node));
        }
        if let Some((failed, message)) = state.failure.as_ref() {
            if failed.elapsed() < Self::RETRY {
                return Err(Error::CommitSource(message.clone()));
            }
        }
        match Node::public_pinned(self.chain, &self.cache_dir) {
            Ok(node) => {
                let node = Arc::new(node);
                state.node = Some(Arc::clone(&node));
                state.failure = None;
                Ok(node)
            }
            Err(error) => {
                let message = error.to_string();
                state.failure = Some((Instant::now(), message.clone()));
                Err(Error::CommitSource(message))
            }
        }
    }
}

impl CommitSource for PublicCommitSource {
    fn chain(&self) -> Chain {
        self.chain
    }

    fn fetch(&self, txid: Txid, cancelled: &AtomicBool) -> Result<Option<Vec<u8>>, Error> {
        check_cancelled(cancelled)?;
        let node = self.node()?;
        fetch_from_node(&node, txid)
    }
}

/// Ordered labelled sources tried until one returns bytes. A miss or error
/// moves to the next source; the set errors only when every source errored.
/// The error table is locked only to read or write, never across a fetch.
pub struct CommitSources {
    chain: Chain,
    entries: Vec<(String, Box<dyn CommitSource + Send + Sync>)>,
    errors: Mutex<Vec<Option<RecordedError>>>,
}

impl CommitSources {
    pub fn new(chain: Chain) -> Self {
        Self {
            chain,
            entries: Vec::new(),
            errors: Mutex::new(Vec::new()),
        }
    }

    pub fn push(&mut self, label: String, source: Box<dyn CommitSource + Send + Sync>) {
        self.entries.push((label, source));
    }

    pub fn is_empty(&self) -> bool {
        self.entries.is_empty()
    }

    pub fn status(&self) -> Result<Vec<CommitSourceStatus>, Error> {
        for (index, entry) in self.entries.iter().enumerate() {
            if let Some(probed) = entry.1.probe() {
                self.record(index, probed.err().map(|error| RecordedError::now(&error)))?;
            }
        }
        let errors = self.errors()?;
        Ok(self
            .entries
            .iter()
            .zip(errors.iter())
            .map(|(entry, error)| {
                let (last_error, last_error_at) = RecordedError::split(error.clone());
                let mut status = CommitSourceStatus::default();
                status.label = entry.0.clone();
                status.last_error = last_error;
                status.last_error_at = last_error_at;
                status
            })
            .collect())
    }

    pub(crate) fn source_chains(&self) -> Vec<Chain> {
        self.entries.iter().map(|entry| entry.1.chain()).collect()
    }

    fn errors(&self) -> Result<MutexGuard<'_, Vec<Option<RecordedError>>>, Error> {
        let mut errors = self
            .errors
            .lock()
            .map_err(|_| Error::Invalid("commit source status poisoned".into()))?;
        errors.resize(self.entries.len(), None);
        Ok(errors)
    }

    fn verified(&self, txid: Txid, raw: Option<Vec<u8>>) -> Result<Option<Vec<u8>>, Error> {
        let Some(raw) = raw else {
            return Ok(None);
        };
        verify_parent(self.chain, txid, &raw)?;
        Ok(Some(raw))
    }

    fn record(&self, index: usize, error: Option<RecordedError>) -> Result<(), Error> {
        let mut errors = self.errors()?;
        let slot = errors
            .get_mut(index)
            .ok_or_else(|| Error::Invalid("commit source status out of range".into()))?;
        *slot = error;
        Ok(())
    }
}

impl CommitSource for CommitSources {
    fn chain(&self) -> Chain {
        self.chain
    }

    fn fetch(&self, txid: Txid, cancelled: &AtomicBool) -> Result<Option<Vec<u8>>, Error> {
        let mut last = None;
        let mut failures = 0;
        for (index, entry) in self.entries.iter().enumerate() {
            check_cancelled(cancelled)?;
            let fetched = entry
                .1
                .fetch(txid, cancelled)
                .and_then(|raw| self.verified(txid, raw));
            check_cancelled(cancelled)?;
            match fetched {
                Ok(Some(raw)) => {
                    self.record(index, None)?;
                    return Ok(Some(raw));
                }
                Ok(None) => self.record(index, None)?,
                Err(error) => {
                    self.record(index, Some(RecordedError::now(&error)))?;
                    failures += 1;
                    last = Some(error);
                }
            }
        }
        let Some(error) = last else {
            return Ok(None);
        };
        if failures == self.entries.len() {
            return Err(error);
        }
        Ok(None)
    }
}

/// Checkpoint-relative Litecoin light client. Does not validate scripts, UTXOs
/// or MWEB consensus. Opening it does not start network connections; the first
/// sync attempt does. An application start before the configured anchor is an
/// explicit error, never silently replaced with the anchor's height.
pub struct LitecoinSource {
    chain: Chain,
    provider: P2pProvider,
}

impl LitecoinSource {
    pub fn open(
        chain: Chain,
        cache: &Path,
        anchor: Anchor,
        peers: Vec<SocketAddr>,
    ) -> Result<Self, Error> {
        if !matches!(chain, Chain::LitecoinTestnet | Chain::LitecoinMainnet) {
            return Err(Error::Invalid(
                "Litecoin source requires a Litecoin network".into(),
            ));
        }
        Ok(Self {
            chain,
            provider: P2pProvider::with_peers(chain, cache, anchor, peers).map_err(source_error)?,
        })
    }

    /// Open a Litecoin source with authenticated preceding header context.
    /// The operator trusts the bootstrap anchor's chain membership and height;
    /// the context does not prove pre-anchor consensus or earlier scan coverage.
    /// This source does not validate scripts, UTXOs or MWEB consensus. Opening
    /// only loads the local cache; the first sync attempt starts connections.
    pub fn open_bootstrap(
        cache: &Path,
        bootstrap: HeaderBootstrap,
        peers: Vec<SocketAddr>,
    ) -> Result<Self, Error> {
        let chain = bootstrap.chain();
        if !matches!(chain, Chain::LitecoinTestnet | Chain::LitecoinMainnet) {
            return Err(Error::Invalid(
                "Litecoin source requires a Litecoin network".into(),
            ));
        }
        Ok(Self {
            chain,
            provider: P2pProvider::with_bootstrap(cache, bootstrap, peers).map_err(source_error)?,
        })
    }
}

fn source_error(e: urma_runtime::error::Error) -> Error {
    Error::Source(Box::new(e))
}

impl ChainSource for LitecoinSource {
    fn genesis(&self) -> Result<BlockHash, Error> {
        Ok(self
            .chain
            .genesis()
            .map_err(|e| Error::Invalid(e.to_string()))?
            .0)
    }
    fn tip(&self) -> Result<ChainPoint, Error> {
        let (height, header) = self.provider.verified_tip().map_err(source_error)?;
        Ok(ChainPoint {
            height,
            hash: header.block_hash(),
        })
    }
    fn header(&self, height: u64) -> Result<Header, Error> {
        self.provider.verified_header(height).map_err(source_error)
    }
    fn block(&self, height: u64) -> Result<Vec<u8>, Error> {
        Ok(self
            .provider
            .verified_block(height)
            .map_err(source_error)?
            .raw()
            .to_vec())
    }
    fn chain_peers(&self) -> Result<u64, Error> {
        u64::try_from(self.provider.progress().peers_connected)
            .map_err(|error| Error::Invalid(format!("chain peer count: {error}")))
    }
    fn chain_target(&self) -> Option<u64> {
        self.provider.target_height()
    }
}