1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
use super::{DataStream, SecretKey, Transport};
use std::{net::SocketAddr, sync::Arc};
use tokio::{
io,
net::{
tcp::{OwnedReadHalf, OwnedWriteHalf},
TcpStream, ToSocketAddrs,
},
};
impl DataStream for TcpStream {
type Read = OwnedReadHalf;
type Write = OwnedWriteHalf;
fn to_connection_tag(&self) -> String {
self.peer_addr()
.map(|addr| format!("{}", addr))
.unwrap_or_else(|_| String::from("--"))
}
fn into_split(self) -> (Self::Read, Self::Write) {
TcpStream::into_split(self)
}
}
impl Transport<TcpStream> {
pub async fn connect(
addrs: impl ToSocketAddrs,
auth_key: Option<Arc<SecretKey>>,
) -> io::Result<Self> {
let stream = TcpStream::connect(addrs).await?;
Self::from_handshake(stream, auth_key).await
}
pub fn peer_addr(&self) -> io::Result<SocketAddr> {
self.conn.get_ref().peer_addr()
}
}