Skip to content

Commit

Permalink
Port dns transport
Browse files Browse the repository at this point in the history
  • Loading branch information
umgefahren committed Dec 27, 2023
1 parent efdbe35 commit c7350f9
Showing 1 changed file with 6 additions and 13 deletions.
19 changes: 6 additions & 13 deletions transports/dns/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -148,9 +148,8 @@ pub mod tokio {
use async_trait::async_trait;
use futures::{future::BoxFuture, prelude::*};
use libp2p_core::{
connection::Endpoint,
multiaddr::{Multiaddr, Protocol},
transport::{ListenerId, TransportError, TransportEvent},
transport::{DialOpts, ListenerId, TransportError, TransportEvent},
};
use parking_lot::Mutex;
use smallvec::SmallVec;
Expand Down Expand Up @@ -231,15 +230,12 @@ where
self.inner.lock().remove_listener(id)
}

fn dial(&mut self, addr: Multiaddr) -> Result<Self::Dial, TransportError<Self::Error>> {
self.do_dial(addr, Endpoint::Dialer)
}

fn dial_as_listener(
fn dial(
&mut self,
addr: Multiaddr,
dial_opts: DialOpts,
) -> Result<Self::Dial, TransportError<Self::Error>> {
self.do_dial(addr, Endpoint::Listener)
self.do_dial(addr, dial_opts)
}

fn address_translation(&self, server: &Multiaddr, observed: &Multiaddr) -> Option<Multiaddr> {
Expand Down Expand Up @@ -269,7 +265,7 @@ where
fn do_dial(
&mut self,
addr: Multiaddr,
role_override: Endpoint,
dial_opts: DialOpts,
) -> Result<
<Self as libp2p_core::Transport>::Dial,
TransportError<<Self as libp2p_core::Transport>::Error>,
Expand Down Expand Up @@ -358,10 +354,7 @@ where
tracing::debug!(address=%addr, "Dialing address");

let transport = inner.clone();
let dial = match role_override {
Endpoint::Dialer => transport.lock().dial(addr),
Endpoint::Listener => transport.lock().dial_as_listener(addr),
};
let dial = transport.lock().dial(addr, dial_opts);
let result = match dial {
Ok(out) => {
// We only count attempts that the inner transport
Expand Down

0 comments on commit c7350f9

Please sign in to comment.