add --connect to cli

This commit is contained in:
rustdesk
2022-12-29 11:18:03 +08:00
parent 7dc0c07578
commit 6bf1710477
2 changed files with 28 additions and 2 deletions

View File

@@ -4,9 +4,9 @@ use hbb_common::{
log,
message_proto::*,
protobuf::Message as _,
rendezvous_proto::ConnType,
tokio::{self, sync::mpsc},
Stream,
rendezvous_proto::ConnType,
};
use std::sync::{Arc, RwLock};
@@ -44,7 +44,7 @@ impl Interface for Session {
fn get_login_config_handler(&self) -> Arc<RwLock<LoginConfigHandler>> {
return self.lc.clone();
}
fn msgbox(&self, msgtype: &str, title: &str, text: &str, link: &str) {
if msgtype == "input-password" {
self.sender
@@ -86,6 +86,25 @@ impl Interface for Session {
}
}
#[tokio::main(flavor = "current_thread")]
pub async fn connect_test(
id: &str,
key: String,
token: String,
) {
let (sender, mut receiver) = mpsc::unbounded_channel::<Data>();
let handler = Session::new(&id, sender);
if let Err(err) = crate::client::Client::start(
id,
&key,
&token,
ConnType::PORT_FORWARD,
handler,
).await {
log::error!("Failed to connect {}: {}", &id, err);
}
}
#[tokio::main(flavor = "current_thread")]
pub async fn start_one_port_forward(
id: String,