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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
use tokio::io::{AsyncRead, AsyncWrite};
use tokio::net::TcpStream;
use tungstenite::client::uri_mode;
use tungstenite::handshake::client::Response;
use tungstenite::Error;
use super::{client_async, Request, IntoClientRequest, WebSocketStream};
#[cfg(feature = "tls")]
pub(crate) mod encryption {
use native_tls::TlsConnector;
use tokio_tls::{TlsConnector as TokioTlsConnector, TlsStream};
use tokio::io::{AsyncRead, AsyncWrite};
use tungstenite::stream::Mode;
use tungstenite::Error;
use crate::stream::Stream as StreamSwitcher;
pub type MaybeTlsStream<S> = StreamSwitcher<S, TlsStream<S>>;
pub type AutoStream<S> = MaybeTlsStream<S>;
pub async fn wrap_stream<S>(
socket: S,
domain: String,
mode: Mode,
) -> Result<AutoStream<S>, Error>
where
S: 'static + AsyncRead + AsyncWrite + Send + Unpin,
{
match mode {
Mode::Plain => Ok(StreamSwitcher::Plain(socket)),
Mode::Tls => {
let try_connector = TlsConnector::new();
let connector = try_connector.map_err(Error::Tls)?;
let stream = TokioTlsConnector::from(connector);
let connected = stream.connect(&domain, socket).await;
match connected {
Err(e) => Err(Error::Tls(e)),
Ok(s) => Ok(StreamSwitcher::Tls(s)),
}
}
}
}
}
#[cfg(feature = "tls")]
pub use self::encryption::MaybeTlsStream;
#[cfg(not(feature = "tls"))]
pub(crate) mod encryption {
use tokio::io::{AsyncRead, AsyncWrite};
use tungstenite::stream::Mode;
use tungstenite::Error;
pub type AutoStream<S> = S;
pub async fn wrap_stream<S>(
socket: S,
_domain: String,
mode: Mode,
) -> Result<AutoStream<S>, Error>
where
S: 'static + AsyncRead + AsyncWrite + Send + Unpin,
{
match mode {
Mode::Plain => Ok(socket),
Mode::Tls => Err(Error::Url("TLS support not compiled in.".into())),
}
}
}
use self::encryption::{wrap_stream, AutoStream};
#[inline]
fn domain(request: &Request) -> Result<String, Error> {
match request.uri().host() {
Some(d) => Ok(d.to_string()),
None => Err(Error::Url("no host name in the url".into())),
}
}
pub async fn client_async_tls<R, S>(
request: R,
stream: S,
) -> Result<(WebSocketStream<AutoStream<S>>, Response), Error>
where
R: IntoClientRequest + Unpin,
S: 'static + AsyncRead + AsyncWrite + Send + Unpin,
AutoStream<S>: Unpin,
{
let request = request.into_client_request()?;
let domain = domain(&request)?;
let mode = uri_mode(&request.uri())?;
let stream = wrap_stream(stream, domain, mode).await?;
client_async(request, stream).await
}
pub async fn connect_async<R>(
request: R,
) -> Result<(WebSocketStream<AutoStream<TcpStream>>, Response), Error>
where
R: IntoClientRequest + Unpin,
{
let request = request.into_client_request()?;
let domain = domain(&request)?;
let port = request
.uri()
.port_u16()
.or_else(|| {
match request.uri().scheme_str() {
Some("wss") => Some(443),
Some("ws") => Some(80),
_ => None
}
})
.ok_or(Error::Url("Url scheme not supported".into()))?;
let addr = format!("{}:{}", domain, port);
let try_socket = TcpStream::connect(addr).await;
let socket = try_socket.map_err(Error::Io)?;
client_async_tls(request, socket).await
}