You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.3 KiB
Rust
93 lines
2.3 KiB
Rust
use crate::harness::{transport::Transport, wallet};
|
|
use anyhow::Result;
|
|
use rand::{CryptoRng, RngCore};
|
|
use xmr_btc::{alice, bob};
|
|
|
|
// TODO: merge this with bob node
|
|
// This struct is responsible for I/O
|
|
pub struct AliceNode<'a> {
|
|
transport: Transport<alice::Message, bob::Message>,
|
|
pub bitcoin_wallet: wallet::bitcoin::Wallet,
|
|
pub monero_wallet: wallet::monero::AliceWallet<'a>,
|
|
}
|
|
|
|
impl<'a> AliceNode<'a> {
|
|
pub fn new(
|
|
transport: Transport<alice::Message, bob::Message>,
|
|
bitcoin_wallet: wallet::bitcoin::Wallet,
|
|
monero_wallet: wallet::monero::AliceWallet<'a>,
|
|
) -> AliceNode<'a> {
|
|
Self {
|
|
transport,
|
|
bitcoin_wallet,
|
|
monero_wallet,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub async fn run_alice_until<'a, R: RngCore + CryptoRng>(
|
|
alice: &mut AliceNode<'a>,
|
|
initial_state: alice::State,
|
|
is_state: fn(&alice::State) -> bool,
|
|
rng: &mut R,
|
|
) -> Result<alice::State> {
|
|
let mut result = initial_state;
|
|
loop {
|
|
result = alice::next_state(
|
|
&alice.bitcoin_wallet,
|
|
&alice.monero_wallet,
|
|
&mut alice.transport,
|
|
result,
|
|
rng,
|
|
)
|
|
.await?;
|
|
if is_state(&result) {
|
|
return Ok(result);
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO: merge this with alice node
|
|
// This struct is responsible for I/O
|
|
pub struct BobNode<'a> {
|
|
transport: Transport<bob::Message, alice::Message>,
|
|
pub bitcoin_wallet: wallet::bitcoin::Wallet,
|
|
pub monero_wallet: wallet::monero::BobWallet<'a>,
|
|
}
|
|
|
|
impl<'a> BobNode<'a> {
|
|
pub fn new(
|
|
transport: Transport<bob::Message, alice::Message>,
|
|
bitcoin_wallet: wallet::bitcoin::Wallet,
|
|
monero_wallet: wallet::monero::BobWallet<'a>,
|
|
) -> BobNode<'a> {
|
|
Self {
|
|
transport,
|
|
bitcoin_wallet,
|
|
monero_wallet,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub async fn run_bob_until<'a, R: RngCore + CryptoRng>(
|
|
bob: &mut BobNode<'a>,
|
|
initial_state: bob::State,
|
|
is_state: fn(&bob::State) -> bool,
|
|
rng: &mut R,
|
|
) -> Result<bob::State> {
|
|
let mut result = initial_state;
|
|
loop {
|
|
result = bob::next_state(
|
|
&bob.bitcoin_wallet,
|
|
&bob.monero_wallet,
|
|
&mut bob.transport,
|
|
result,
|
|
rng,
|
|
)
|
|
.await?;
|
|
if is_state(&result) {
|
|
return Ok(result);
|
|
}
|
|
}
|
|
}
|