Send Message2 from Bob to Alice

In order for Alice to complete the handshake she needs to transition to state 3,
for this she needs message 2 from Bob.

Send `bob::Message2` to Alice and transition to `State3` - completing the
handshake.
This commit is contained in:
Tobin C. Harding 2020-10-26 14:27:41 +11:00
parent 90f6c55290
commit cbccc282ef
8 changed files with 239 additions and 16 deletions

View File

@ -8,13 +8,14 @@ use libp2p::{
}; };
use rand::rngs::OsRng; use rand::rngs::OsRng;
use std::thread; use std::thread;
use tracing::debug; use tracing::{debug, info};
mod amounts; mod amounts;
mod message0; mod message0;
mod message1; mod message1;
mod message2;
use self::{amounts::*, message0::*, message1::*}; use self::{amounts::*, message0::*, message1::*, message2::*};
use crate::{ use crate::{
network::{ network::{
peer_tracker::{self, PeerTracker}, peer_tracker::{self, PeerTracker},
@ -42,7 +43,7 @@ pub async fn swap(
loop { loop {
match swarm.next().await { match swarm.next().await {
OutEvent::ConnectionEstablished(id) => { OutEvent::ConnectionEstablished(id) => {
tracing::info!("Connection established with: {}", id); info!("Connection established with: {}", id);
} }
OutEvent::Request(amounts::OutEvent::Btc { btc, channel }) => { OutEvent::Request(amounts::OutEvent::Btc { btc, channel }) => {
debug!("Got request from Bob to swap {}", btc); debug!("Got request from Bob to swap {}", btc);
@ -98,9 +99,13 @@ pub async fn swap(
let msg = state2.next_message(); let msg = state2.next_message();
swarm.send_message1(channel, msg); swarm.send_message1(channel, msg);
tracing::info!("handshake complete, we now have State2 for Alice."); let _state3 = match swarm.next().await {
OutEvent::Message2(msg) => state2.receive(msg)?,
other => panic!("Unexpected event: {:?}", other),
};
info!("Handshake complete, we now have State3 for Alice.");
tracing::warn!("parking thread ...");
thread::park(); thread::park();
Ok(()) Ok(())
} }
@ -124,7 +129,7 @@ fn new_swarm(listen: Multiaddr) -> Result<Swarm> {
Swarm::listen_on(&mut swarm, listen.clone()) Swarm::listen_on(&mut swarm, listen.clone())
.with_context(|| format!("Address is not supported: {:#}", listen))?; .with_context(|| format!("Address is not supported: {:#}", listen))?;
tracing::info!("Initialized swarm: {}", local_peer_id); info!("Initialized swarm: {}", local_peer_id);
Ok(swarm) Ok(swarm)
} }
@ -139,6 +144,7 @@ pub enum OutEvent {
msg: bob::Message1, msg: bob::Message1,
channel: ResponseChannel<AliceToBob>, channel: ResponseChannel<AliceToBob>,
}, },
Message2(bob::Message2),
} }
impl From<peer_tracker::OutEvent> for OutEvent { impl From<peer_tracker::OutEvent> for OutEvent {
@ -173,6 +179,14 @@ impl From<message1::OutEvent> for OutEvent {
} }
} }
impl From<message2::OutEvent> for OutEvent {
fn from(event: message2::OutEvent) -> Self {
match event {
message2::OutEvent::Msg(msg) => OutEvent::Message2(msg),
}
}
}
/// A `NetworkBehaviour` that represents an XMR/BTC swap node as Alice. /// A `NetworkBehaviour` that represents an XMR/BTC swap node as Alice.
#[derive(NetworkBehaviour)] #[derive(NetworkBehaviour)]
#[behaviour(out_event = "OutEvent", event_process = false)] #[behaviour(out_event = "OutEvent", event_process = false)]
@ -182,6 +196,7 @@ pub struct Alice {
amounts: Amounts, amounts: Amounts,
message0: Message0, message0: Message0,
message1: Message1, message1: Message1,
message2: Message2,
#[behaviour(ignore)] #[behaviour(ignore)]
identity: Keypair, identity: Keypair,
} }
@ -225,6 +240,7 @@ impl Default for Alice {
amounts: Amounts::default(), amounts: Amounts::default(),
message0: Message0::default(), message0: Message0::default(),
message1: Message1::default(), message1: Message1::default(),
message2: Message2::default(),
identity, identity,
} }
} }

View File

@ -0,0 +1,94 @@
use libp2p::{
request_response::{
handler::RequestProtocol, ProtocolSupport, RequestResponse, RequestResponseConfig,
RequestResponseEvent, RequestResponseMessage, ResponseChannel,
},
swarm::{NetworkBehaviourAction, NetworkBehaviourEventProcess, PollParameters},
NetworkBehaviour,
};
use std::{
collections::VecDeque,
task::{Context, Poll},
time::Duration,
};
use tracing::{debug, error};
use crate::network::request_response::{AliceToBob, BobToAlice, Codec, Protocol, TIMEOUT};
use xmr_btc::bob;
#[derive(Debug)]
pub enum OutEvent {
Msg(bob::Message2),
}
/// A `NetworkBehaviour` that represents receiving of message 2 from Bob.
#[derive(NetworkBehaviour)]
#[behaviour(out_event = "OutEvent", poll_method = "poll")]
#[allow(missing_debug_implementations)]
pub struct Message2 {
rr: RequestResponse<Codec>,
#[behaviour(ignore)]
events: VecDeque<OutEvent>,
}
impl Message2 {
pub fn send(&mut self, channel: ResponseChannel<AliceToBob>, msg: xmr_btc::alice::Message2) {
let msg = AliceToBob::Message2(msg);
self.rr.send_response(channel, msg);
}
fn poll(
&mut self,
_: &mut Context<'_>,
_: &mut impl PollParameters,
) -> Poll<NetworkBehaviourAction<RequestProtocol<Codec>, OutEvent>> {
if let Some(event) = self.events.pop_front() {
return Poll::Ready(NetworkBehaviourAction::GenerateEvent(event));
}
Poll::Pending
}
}
impl Default for Message2 {
fn default() -> Self {
let timeout = Duration::from_secs(TIMEOUT);
let mut config = RequestResponseConfig::default();
config.set_request_timeout(timeout);
Self {
rr: RequestResponse::new(
Codec::default(),
vec![(Protocol, ProtocolSupport::Full)],
config,
),
events: Default::default(),
}
}
}
impl NetworkBehaviourEventProcess<RequestResponseEvent<BobToAlice, AliceToBob>> for Message2 {
fn inject_event(&mut self, event: RequestResponseEvent<BobToAlice, AliceToBob>) {
match event {
RequestResponseEvent::Message {
message: RequestResponseMessage::Request { request, .. },
..
} => match request {
BobToAlice::Message2(msg) => {
self.events.push_back(OutEvent::Msg(msg));
}
other => debug!("got request: {:?}", other),
},
RequestResponseEvent::Message {
message: RequestResponseMessage::Response { .. },
..
} => panic!("Alice should not get a Response"),
RequestResponseEvent::InboundFailure { error, .. } => {
error!("Inbound failure: {:?}", error);
}
RequestResponseEvent::OutboundFailure { error, .. } => {
error!("Outbound failure: {:?}", error);
}
}
}
}

View File

@ -13,8 +13,9 @@ use tracing::{debug, info};
mod amounts; mod amounts;
mod message0; mod message0;
mod message1; mod message1;
mod message2;
use self::{amounts::*, message0::*, message1::*}; use self::{amounts::*, message0::*, message1::*, message2::*};
use crate::{ use crate::{
network::{ network::{
peer_tracker::{self, PeerTracker}, peer_tracker::{self, PeerTracker},
@ -24,7 +25,7 @@ use crate::{
}; };
use xmr_btc::{ use xmr_btc::{
alice, alice,
bitcoin::BuildTxLockPsbt, bitcoin::{BroadcastSignedTransaction, BuildTxLockPsbt, SignTxLock},
bob::{self, State0}, bob::{self, State0},
}; };
@ -38,7 +39,7 @@ pub async fn swap<W>(
wallet: W, wallet: W,
) -> Result<()> ) -> Result<()>
where where
W: BuildTxLockPsbt + Send + Sync + 'static, W: BuildTxLockPsbt + SignTxLock + BroadcastSignedTransaction + Send + Sync + 'static,
{ {
let mut swarm = new_swarm()?; let mut swarm = new_swarm()?;
@ -93,14 +94,16 @@ where
}; };
swarm.send_message1(alice.clone(), state1.next_message()); swarm.send_message1(alice.clone(), state1.next_message());
let _state2 = match swarm.next().await { let state2 = match swarm.next().await {
OutEvent::Message1(msg) => { OutEvent::Message1(msg) => {
state1.receive(msg) // TODO: More graceful error handling. state1.receive(msg)? // TODO: More graceful error handling.
} }
other => panic!("unexpected event: {:?}", other), other => panic!("unexpected event: {:?}", other),
}; };
info!("handshake complete, we now have State2 for Bob."); swarm.send_message2(alice.clone(), state2.next_message());
info!("Handshake complete, we now have State2 for Bob.");
thread::park(); thread::park();
Ok(()) Ok(())
@ -134,6 +137,7 @@ pub enum OutEvent {
Amounts(SwapAmounts), Amounts(SwapAmounts),
Message0(alice::Message0), Message0(alice::Message0),
Message1(alice::Message1), Message1(alice::Message1),
Message2(alice::Message2),
} }
impl From<peer_tracker::OutEvent> for OutEvent { impl From<peer_tracker::OutEvent> for OutEvent {
@ -170,6 +174,14 @@ impl From<message1::OutEvent> for OutEvent {
} }
} }
impl From<message2::OutEvent> for OutEvent {
fn from(event: message2::OutEvent) -> Self {
match event {
message2::OutEvent::Msg(msg) => OutEvent::Message2(msg),
}
}
}
/// A `NetworkBehaviour` that represents an XMR/BTC swap node as Bob. /// A `NetworkBehaviour` that represents an XMR/BTC swap node as Bob.
#[derive(NetworkBehaviour)] #[derive(NetworkBehaviour)]
#[behaviour(out_event = "OutEvent", event_process = false)] #[behaviour(out_event = "OutEvent", event_process = false)]
@ -179,6 +191,7 @@ pub struct Bob {
amounts: Amounts, amounts: Amounts,
message0: Message0, message0: Message0,
message1: Message1, message1: Message1,
message2: Message2,
#[behaviour(ignore)] #[behaviour(ignore)]
identity: Keypair, identity: Keypair,
} }
@ -209,6 +222,11 @@ impl Bob {
self.message1.send(alice, msg) self.message1.send(alice, msg)
} }
/// Sends Bob's third message to Alice.
pub fn send_message2(&mut self, alice: PeerId, msg: bob::Message2) {
self.message2.send(alice, msg)
}
/// Returns Alice's peer id if we are connected. /// Returns Alice's peer id if we are connected.
pub fn peer_id_of_alice(&self) -> Option<PeerId> { pub fn peer_id_of_alice(&self) -> Option<PeerId> {
self.pt.counterparty_peer_id() self.pt.counterparty_peer_id()
@ -224,6 +242,7 @@ impl Default for Bob {
amounts: Amounts::default(), amounts: Amounts::default(),
message0: Message0::default(), message0: Message0::default(),
message1: Message1::default(), message1: Message1::default(),
message2: Message2::default(),
identity, identity,
} }
} }

92
swap/src/bob/message2.rs Normal file
View File

@ -0,0 +1,92 @@
use libp2p::{
request_response::{
handler::RequestProtocol, ProtocolSupport, RequestResponse, RequestResponseConfig,
RequestResponseEvent, RequestResponseMessage,
},
swarm::{NetworkBehaviourAction, NetworkBehaviourEventProcess, PollParameters},
NetworkBehaviour, PeerId,
};
use std::{
collections::VecDeque,
task::{Context, Poll},
time::Duration,
};
use tracing::{debug, error};
use crate::network::request_response::{AliceToBob, BobToAlice, Codec, Protocol, TIMEOUT};
use xmr_btc::{alice, bob};
#[derive(Debug)]
pub enum OutEvent {
Msg(alice::Message2),
}
/// A `NetworkBehaviour` that represents sending message 2 to Alice.
#[derive(NetworkBehaviour)]
#[behaviour(out_event = "OutEvent", poll_method = "poll")]
#[allow(missing_debug_implementations)]
pub struct Message2 {
rr: RequestResponse<Codec>,
#[behaviour(ignore)]
events: VecDeque<OutEvent>,
}
impl Message2 {
pub fn send(&mut self, alice: PeerId, msg: bob::Message2) {
let msg = BobToAlice::Message2(msg);
let _id = self.rr.send_request(&alice, msg);
}
fn poll(
&mut self,
_: &mut Context<'_>,
_: &mut impl PollParameters,
) -> Poll<NetworkBehaviourAction<RequestProtocol<Codec>, OutEvent>> {
if let Some(event) = self.events.pop_front() {
return Poll::Ready(NetworkBehaviourAction::GenerateEvent(event));
}
Poll::Pending
}
}
impl Default for Message2 {
fn default() -> Self {
let timeout = Duration::from_secs(TIMEOUT);
let mut config = RequestResponseConfig::default();
config.set_request_timeout(timeout);
Self {
rr: RequestResponse::new(
Codec::default(),
vec![(Protocol, ProtocolSupport::Full)],
config,
),
events: Default::default(),
}
}
}
impl NetworkBehaviourEventProcess<RequestResponseEvent<BobToAlice, AliceToBob>> for Message2 {
fn inject_event(&mut self, event: RequestResponseEvent<BobToAlice, AliceToBob>) {
match event {
RequestResponseEvent::Message {
message: RequestResponseMessage::Request { .. },
..
} => panic!("Bob should never get a request from Alice"),
RequestResponseEvent::Message {
message: RequestResponseMessage::Response { response, .. },
..
} => match response {
AliceToBob::Message2(msg) => self.events.push_back(OutEvent::Msg(msg)),
other => debug!("got response: {:?}", other),
},
RequestResponseEvent::InboundFailure { error, .. } => {
error!("Inbound failure: {:?}", error);
}
RequestResponseEvent::OutboundFailure { error, .. } => {
error!("Outbound failure: {:?}", error);
}
}
}
}

View File

@ -26,7 +26,7 @@ mod trace;
use cli::Options; use cli::Options;
use swap::{alice, bitcoin::Wallet, bob, Cmd, Rsp, SwapAmounts}; use swap::{alice, bitcoin::Wallet, bob, Cmd, Rsp, SwapAmounts};
use xmr_btc::bitcoin::BuildTxLockPsbt; use xmr_btc::bitcoin::{BroadcastSignedTransaction, BuildTxLockPsbt, SignTxLock};
// TODO: Add root seed file instead of generating new seed each run. // TODO: Add root seed file instead of generating new seed each run.
// TODO: Remove all instances of the todo! macro // TODO: Remove all instances of the todo! macro
@ -109,7 +109,7 @@ async fn swap_as_bob<W>(
wallet: W, wallet: W,
) -> Result<()> ) -> Result<()>
where where
W: BuildTxLockPsbt + Send + Sync + 'static, W: BuildTxLockPsbt + SignTxLock + BroadcastSignedTransaction + Send + Sync + 'static,
{ {
let (cmd_tx, mut cmd_rx) = mpsc::channel(1); let (cmd_tx, mut cmd_rx) = mpsc::channel(1);
let (mut rsp_tx, rsp_rx) = mpsc::channel(1); let (mut rsp_tx, rsp_rx) = mpsc::channel(1);

View File

@ -25,6 +25,7 @@ pub enum BobToAlice {
AmountsFromXmr(monero::Amount), AmountsFromXmr(monero::Amount),
Message0(bob::Message0), Message0(bob::Message0),
Message1(bob::Message1), Message1(bob::Message1),
Message2(bob::Message2),
} }
/// Messages Alice sends to Bob. /// Messages Alice sends to Bob.
@ -34,6 +35,7 @@ pub enum AliceToBob {
Amounts(SwapAmounts), Amounts(SwapAmounts),
Message0(alice::Message0), Message0(alice::Message0),
Message1(alice::Message1), Message1(alice::Message1),
Message2(alice::Message2),
} }
#[derive(Debug, Clone, Copy, Default)] #[derive(Debug, Clone, Copy, Default)]

View File

@ -29,7 +29,7 @@ pub struct Message1 {
pub(crate) tx_refund_encsig: EncryptedSignature, pub(crate) tx_refund_encsig: EncryptedSignature,
} }
#[derive(Clone, Debug)] #[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Message2 { pub struct Message2 {
pub(crate) tx_lock_proof: monero::TransferProof, pub(crate) tx_lock_proof: monero::TransferProof,
} }

View File

@ -27,7 +27,7 @@ pub struct Message1 {
pub(crate) tx_lock: bitcoin::TxLock, pub(crate) tx_lock: bitcoin::TxLock,
} }
#[derive(Clone, Debug)] #[derive(Clone, Debug, Serialize, Deserialize)]
pub struct Message2 { pub struct Message2 {
pub(crate) tx_punish_sig: Signature, pub(crate) tx_punish_sig: Signature,
pub(crate) tx_cancel_sig: Signature, pub(crate) tx_cancel_sig: Signature,