forked from snapview/tokio-tungstenite
/
connect.rs
91 lines (79 loc) · 2.82 KB
/
connect.rs
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
//! Connection helper.
use tokio::net::TcpStream;
use tungstenite::{
error::{Error, UrlError},
handshake::client::Response,
protocol::WebSocketConfig,
};
use crate::{domain, stream::MaybeTlsStream, IntoClientRequest, WebSocketStream};
#[cfg(any(feature = "native-tls", feature = "rustls-tls"))]
use crate::Connector;
/// Connect to a given URL.
pub async fn connect_async<R>(
request: R,
) -> Result<(WebSocketStream<MaybeTlsStream<TcpStream>>, Response), Error>
where
R: IntoClientRequest + Unpin,
{
connect_async_with_config(request, None).await
}
/// The same as `connect_async()` but the one can specify a websocket configuration.
/// Please refer to `connect_async()` for more details.
pub async fn connect_async_with_config<R>(
request: R,
config: Option<WebSocketConfig>,
) -> Result<(WebSocketStream<MaybeTlsStream<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(UrlError::UnsupportedUrlScheme))?;
let addr = format!("{}:{}", domain, port);
let try_socket = TcpStream::connect(addr).await;
let socket = try_socket.map_err(Error::Io)?;
#[cfg(not(any(feature = "native-tls", feature = "__rustls-tls")))]
{
crate::client_async_with_config(request, MaybeTlsStream::Plain(socket), config).await
}
#[cfg(any(feature = "native-tls", feature = "__rustls-tls"))]
{
crate::tls::client_async_tls_with_config(request, socket, config, None).await
}
}
/// The same as `connect_async()` but the one can specify a websocket configuration,
/// and a TLS connector to use.
/// Please refer to `connect_async()` for more details.
#[cfg(any(feature = "native-tls", feature = "rustls-tls"))]
pub async fn connect_async_tls_with_config<R>(
request: R,
config: Option<WebSocketConfig>,
connector: Option<Connector>,
) -> Result<(WebSocketStream<MaybeTlsStream<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(UrlError::UnsupportedUrlScheme))?;
let addr = format!("{}:{}", domain, port);
let try_socket = TcpStream::connect(addr).await;
let socket = try_socket.map_err(Error::Io)?;
crate::tls::client_async_tls_with_config(request, socket, config, connector).await
}