rustdesk/hbb_common
1
0
Fork 0
mirror of https://github.com/rustdesk/hbb_common.git synced 2026-04-10 09:56:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] tcp success. Websocket test. (temp ver.)

Browse Source
This commit is contained in:
YinMo19
2025-04-23 08:33:01 +08:00
parent d299e4909f
commit 6798bf3780
2 changed files with 21 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/socket_client.rs
View File

@@ -116,18 +116,15 @@ pub async fn connect_tcp_local<
) -> ResultType<Stream> {
let target_str = target.to_string();
// 根据目标地址协议决定连接方式
if target_str.starts_with("ws://") || target_str.starts_with("wss://") {
// WebSocket 连接逻辑
Ok(Stream::WebSocket(websocket::WsFramedStream::new(
target_str,
local,
None,
ms_timeout,
)
.await?))
// if target_str.starts_with("ws://") || target_str.starts_with("wss://") {
// Ok(Stream::WebSocket(
// websocket::WsFramedStream::new(target_str, local, None, ms_timeout).await?,
// ))
if true {
Ok(Stream::WebSocket(
websocket::WsFramedStream::new(target_str, local, None, ms_timeout).await?,
))
} else {
// TCP 连接逻辑
if let Some(conf) = Config::get_socks() {
return Ok(Stream::Tcp(
FramedStream::connect(target, local, &conf, ms_timeout).await?,
@@ -139,8 +136,7 @@ pub async fn connect_tcp_local<
if local_addr.is_ipv6() && target_addr.is_ipv4() {
let resolved_target = query_nip_io(target_addr).await?;
return Ok(Stream::Tcp(
FramedStream::new(resolved_target, Some(local_addr), ms_timeout)
.await?,
FramedStream::new(resolved_target, Some(local_addr), ms_timeout).await?,
));
}
}