|
|
@@ -85,7 +85,6 @@ impl RendezvousServer {
|
|
85
|
85
|
port: i32,
|
|
86
|
86
|
serial: i32,
|
|
87
|
87
|
key: &str,
|
|
88
|
|
- id_change_support: bool,
|
|
89
|
88
|
rmem: usize,
|
|
90
|
89
|
) -> ResultType<()> {
|
|
91
|
90
|
let addr = format!("0.0.0.0:{}", port);
|
|
|
@@ -97,7 +96,6 @@ impl RendezvousServer {
|
|
97
|
96
|
log::info!("Listening on tcp/udp {}", addr);
|
|
98
|
97
|
log::info!("Listening on tcp {}, extra port for NAT test", addr2);
|
|
99
|
98
|
log::info!("Listening on websocket {}", addr3);
|
|
100
|
|
- log::info!("change-id={:?}", id_change_support);
|
|
101
|
99
|
let mut socket = FramedSocket::new_with_buf_size(&addr, rmem).await?;
|
|
102
|
100
|
let (tx, mut rx) = mpsc::unbounded_channel::<Data>();
|
|
103
|
101
|
let software_url = get_arg("software-url");
|
|
|
@@ -163,7 +161,6 @@ impl RendezvousServer {
|
|
163
|
161
|
&mut listener3,
|
|
164
|
162
|
&mut socket,
|
|
165
|
163
|
&key,
|
|
166
|
|
- id_change_support,
|
|
167
|
164
|
)
|
|
168
|
165
|
.await
|
|
169
|
166
|
{
|
|
|
@@ -195,7 +192,6 @@ impl RendezvousServer {
|
|
195
|
192
|
listener3: &mut TcpListener,
|
|
196
|
193
|
socket: &mut FramedSocket,
|
|
197
|
194
|
key: &str,
|
|
198
|
|
- id_change_support: bool,
|
|
199
|
195
|
) -> LoopFailure {
|
|
200
|
196
|
let mut timer_check_relay = interval(Duration::from_millis(CHECK_RELAY_TIMEOUT));
|
|
201
|
197
|
loop {
|
|
|
@@ -249,7 +245,7 @@ impl RendezvousServer {
|
|
249
|
245
|
match res {
|
|
250
|
246
|
Ok((stream, addr)) => {
|
|
251
|
247
|
stream.set_nodelay(true).ok();
|
|
252
|
|
- self.handle_listener(stream, addr, key, id_change_support, true).await;
|
|
|
248
|
+ self.handle_listener(stream, addr, key, true).await;
|
|
253
|
249
|
}
|
|
254
|
250
|
Err(err) => {
|
|
255
|
251
|
log::error!("listener3.accept failed: {}", err);
|
|
|
@@ -261,7 +257,7 @@ impl RendezvousServer {
|
|
261
|
257
|
match res {
|
|
262
|
258
|
Ok((stream, addr)) => {
|
|
263
|
259
|
stream.set_nodelay(true).ok();
|
|
264
|
|
- self.handle_listener(stream, addr, key, id_change_support, false).await;
|
|
|
260
|
+ self.handle_listener(stream, addr, key, false).await;
|
|
265
|
261
|
}
|
|
266
|
262
|
Err(err) => {
|
|
267
|
263
|
log::error!("listener.accept failed: {}", err);
|
|
|
@@ -446,7 +442,6 @@ impl RendezvousServer {
|
|
446
|
442
|
sink: &mut Option<Sink>,
|
|
447
|
443
|
addr: SocketAddr,
|
|
448
|
444
|
key: &str,
|
|
449
|
|
- id_change_support: bool,
|
|
450
|
445
|
ws: bool,
|
|
451
|
446
|
) -> bool {
|
|
452
|
447
|
if let Ok(msg_in) = RendezvousMessage::parse_from_bytes(&bytes) {
|
|
|
@@ -1022,7 +1017,6 @@ impl RendezvousServer {
|
|
1022
|
1017
|
stream: TcpStream,
|
|
1023
|
1018
|
addr: SocketAddr,
|
|
1024
|
1019
|
key: &str,
|
|
1025
|
|
- id_change_support: bool,
|
|
1026
|
1020
|
ws: bool,
|
|
1027
|
1021
|
) {
|
|
1028
|
1022
|
log::debug!("Tcp connection from {:?}, ws: {}", addr, ws);
|
|
|
@@ -1030,7 +1024,7 @@ impl RendezvousServer {
|
|
1030
|
1024
|
let key = key.to_owned();
|
|
1031
|
1025
|
tokio::spawn(async move {
|
|
1032
|
1026
|
allow_err!(
|
|
1033
|
|
- rs.handle_listener_inner(stream, addr, &key, id_change_support, ws)
|
|
|
1027
|
+ rs.handle_listener_inner(stream, addr, &key, ws)
|
|
1034
|
1028
|
.await
|
|
1035
|
1029
|
);
|
|
1036
|
1030
|
});
|
|
|
@@ -1042,7 +1036,6 @@ impl RendezvousServer {
|
|
1042
|
1036
|
stream: TcpStream,
|
|
1043
|
1037
|
addr: SocketAddr,
|
|
1044
|
1038
|
key: &str,
|
|
1045
|
|
- id_change_support: bool,
|
|
1046
|
1039
|
ws: bool,
|
|
1047
|
1040
|
) -> ResultType<()> {
|
|
1048
|
1041
|
let mut sink;
|
|
|
@@ -1054,7 +1047,7 @@ impl RendezvousServer {
|
|
1054
|
1047
|
match msg {
|
|
1055
|
1048
|
tungstenite::Message::Binary(bytes) => {
|
|
1056
|
1049
|
if !self
|
|
1057
|
|
- .handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws)
|
|
|
1050
|
+ .handle_tcp(&bytes, &mut sink, addr, key, ws)
|
|
1058
|
1051
|
.await
|
|
1059
|
1052
|
{
|
|
1060
|
1053
|
break;
|
|
|
@@ -1068,7 +1061,7 @@ impl RendezvousServer {
|
|
1068
|
1061
|
sink = Some(Sink::TcpStream(a));
|
|
1069
|
1062
|
while let Ok(Some(Ok(bytes))) = timeout(30_000, b.next()).await {
|
|
1070
|
1063
|
if !self
|
|
1071
|
|
- .handle_tcp(&bytes, &mut sink, addr, key, id_change_support, ws)
|
|
|
1064
|
+ .handle_tcp(&bytes, &mut sink, addr, key, ws)
|
|
1072
|
1065
|
.await
|
|
1073
|
1066
|
{
|
|
1074
|
1067
|
break;
|