Files
rustdesk/src/flutter.rs

2048 lines
84 KiB
Rust
Raw Normal View History

use std::{
2022-08-17 17:23:55 +08:00
collections::HashMap,
2022-08-04 17:24:02 +08:00
sync::{
atomic::{AtomicBool, AtomicUsize, Ordering},
2022-08-04 17:24:02 +08:00
Arc, Mutex, RwLock,
},
};
2022-05-19 23:45:44 +08:00
use flutter_rust_bridge::{StreamSink, ZeroCopyBuffer};
2022-05-12 17:35:25 +08:00
use hbb_common::{
allow_err, bail,
2022-05-12 17:35:25 +08:00
compress::decompress,
config::{Config, LocalConfig, PeerConfig, TransferSerde},
2022-05-28 03:56:42 +08:00
fs::{
self, can_enable_overwrite_detection, get_job, get_string, new_send_confirm,
transform_windows_path, DigestCheckResult,
2022-05-28 03:56:42 +08:00
},
log,
2022-05-12 17:35:25 +08:00
message_proto::*,
protobuf::Message as _,
rendezvous_proto::ConnType,
tokio::{
self,
sync::mpsc,
time::{self, Duration, Instant, Interval},
},
ResultType, Stream,
2022-05-12 17:35:25 +08:00
};
use crate::{
common::{self, make_fd_to_json, CLIPBOARD_INTERVAL},
ui_session_interface::{io_loop, InvokeUi, Session},
};
#[cfg(not(any(target_os = "android", target_os = "ios")))]
use crate::common::{check_clipboard, update_clipboard, ClipboardContext};
2022-07-11 10:30:45 +08:00
use crate::{client::*, flutter_ffi::EventToUI, make_fd_flutter};
2022-05-12 17:35:25 +08:00
pub(super) const APP_TYPE_MAIN: &str = "main";
pub(super) const APP_TYPE_DESKTOP_REMOTE: &str = "remote";
pub(super) const APP_TYPE_DESKTOP_FILE_TRANSFER: &str = "file transfer";
2022-08-31 16:31:31 +08:00
const MILLI1: Duration = Duration::from_millis(1);
2022-05-12 17:35:25 +08:00
lazy_static::lazy_static! {
pub static ref SESSIONS: RwLock<HashMap<String,Session<FlutterHandler>>> = Default::default();
pub static ref GLOBAL_EVENT_STREAM: RwLock<HashMap<String, StreamSink<String>>> = Default::default(); // rust to dart event channel
2022-05-12 17:35:25 +08:00
}
static SERVER_CLIPBOARD_ENABLED: AtomicBool = AtomicBool::new(true);
static SERVER_KEYBOARD_ENABLED: AtomicBool = AtomicBool::new(true);
#[derive(Default, Clone)]
pub struct FlutterHandler {
pub event_stream: Arc<RwLock<Option<StreamSink<EventToUI>>>>,
2022-05-12 17:35:25 +08:00
}
impl FlutterHandler {
2022-05-28 03:56:42 +08:00
/// Push an event to the event queue.
/// An event is stored as json in the event queue.
///
/// # Arguments
///
/// * `name` - The name of the event.
/// * `event` - Fields of the event content.
2022-05-12 17:35:25 +08:00
fn push_event(&self, name: &str, event: Vec<(&str, &str)>) {
let mut h: HashMap<&str, &str> = event.iter().cloned().collect();
assert!(h.get("name").is_none());
h.insert("name", name);
2022-05-31 22:09:36 +08:00
let out = serde_json::ser::to_string(&h).unwrap_or("".to_owned());
if let Some(stream) = &*self.event_stream.read().unwrap() {
stream.add(EventToUI::Event(out));
}
2022-05-12 17:35:25 +08:00
}
}
2022-05-12 17:35:25 +08:00
impl InvokeUi for FlutterHandler {
fn set_cursor_data(&self, cd: CursorData) {
let colors = hbb_common::compress::decompress(&cd.colors);
self.push_event(
"cursor_data",
vec![
("id", &cd.id.to_string()),
("hotx", &cd.hotx.to_string()),
("hoty", &cd.hoty.to_string()),
("width", &cd.width.to_string()),
("height", &cd.height.to_string()),
(
"colors",
&serde_json::ser::to_string(&colors).unwrap_or("".to_owned()),
),
],
);
2022-05-12 17:35:25 +08:00
}
fn set_cursor_id(&self, id: String) {
self.push_event("cursor_id", vec![("id", &id.to_string())]);
2022-05-12 17:35:25 +08:00
}
fn set_cursor_position(&self, cp: CursorPosition) {
self.push_event(
"cursor_position",
vec![("x", &cp.x.to_string()), ("y", &cp.y.to_string())],
);
2022-05-12 17:35:25 +08:00
}
fn set_display(&self, x: i32, y: i32, w: i32, h: i32) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn update_privacy_mode(&self) {
self.push_event("update_privacy_mode", [].into());
2022-05-12 17:35:25 +08:00
}
fn set_permission(&self, name: &str, value: bool) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn update_pi(&self, pi: PeerInfo) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn close_success(&self) {
// todo!()
}
2022-08-04 17:24:02 +08:00
fn update_quality_status(&self, status: QualityStatus) {
const NULL: String = String::new();
self.push_event(
"update_quality_status",
vec![
("speed", &status.speed.map_or(NULL, |it| it)),
("fps", &status.fps.map_or(NULL, |it| it.to_string())),
("delay", &status.delay.map_or(NULL, |it| it.to_string())),
(
"target_bitrate",
&status.target_bitrate.map_or(NULL, |it| it.to_string()),
),
(
"codec_format",
&status.codec_format.map_or(NULL, |it| it.to_string()),
),
],
);
}
2022-08-29 13:08:42 +08:00
fn set_connection_type(&self, is_secured: bool, direct: bool) {
2022-05-12 17:35:25 +08:00
self.push_event(
"connection_ready",
2022-05-12 17:35:25 +08:00
vec![
("secure", &is_secured.to_string()),
("direct", &direct.to_string()),
2022-05-12 17:35:25 +08:00
],
);
}
fn job_error(&self, id: i32, err: String, file_num: i32) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn job_done(&self, id: i32, file_num: i32) {
// todo!()
}
fn clear_all_jobs(&self) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn add_job(
&self,
id: i32,
path: String,
to: String,
file_num: i32,
show_hidden: bool,
is_remote: bool,
) {
// todo!()
2022-05-12 17:35:25 +08:00
}
fn update_transfer_list(&self) {
// todo!()
}
fn confirm_delete_files(&self, id: i32, i: i32, name: String) {
// todo!()
}
fn override_file_confirm(&self, id: i32, file_num: i32, to: String, is_upload: bool) {
// todo!()
}
fn job_progress(&self, id: i32, file_num: i32, speed: f64, finished_size: f64) {
// todo!()
}
fn adapt_size(&self) {
// todo!()
}
fn on_rgba(&self, data: &[u8]) {
if let Some(stream) = &*self.event_stream.read().unwrap() {
stream.add(EventToUI::Rgba(ZeroCopyBuffer(data.to_owned())));
2022-05-12 17:35:25 +08:00
}
}
fn set_peer_info(
&self,
username: &str,
hostname: &str,
platform: &str,
sas_enabled: bool,
displays: &Vec<HashMap<&str, i32>>,
version: &str,
current_display: usize,
is_file_transfer: bool,
2022-05-12 17:35:25 +08:00
) {
let displays = serde_json::ser::to_string(displays).unwrap_or("".to_owned());
self.push_event(
"peer_info",
2022-05-12 17:35:25 +08:00
vec![
("username", username),
("hostname", hostname),
("platform", platform),
("sas_enabled", &sas_enabled.to_string()),
("displays", &displays),
("version", &version),
("current_display", &current_display.to_string()),
("is_file_transfer", &is_file_transfer.to_string()),
2022-05-12 17:35:25 +08:00
],
);
}
fn msgbox(&self, msgtype: &str, title: &str, text: &str, retry: bool) {
let has_retry = if retry { "true" } else { "" };
self.push_event(
"msgbox",
2022-05-19 21:45:25 +08:00
vec![
("type", msgtype),
("title", title),
("text", text),
("hasRetry", has_retry),
2022-05-19 21:45:25 +08:00
],
2022-05-17 20:56:36 +08:00
);
}
}
2022-07-11 18:23:58 +08:00
/// Create a new remote session with the given id.
///
/// # Arguments
///
/// * `id` - The identifier of the remote session with prefix. Regex: [\w]*[\_]*[\d]+
/// * `is_file_transfer` - If the session is used for file transfer.
/// * `is_port_forward` - If the session is used for port forward.
pub fn session_add(id: &str, is_file_transfer: bool, is_port_forward: bool) -> ResultType<()> {
// TODO check same id
let session_id = get_session_id(id.to_owned());
LocalConfig::set_remote_id(&session_id);
// TODO close
// Self::close();
// TODO cmd passwd args
let session: Session<FlutterHandler> = Session {
id: session_id.clone(),
..Default::default()
};
session
.lc
.write()
.unwrap()
.initialize(session_id.clone(), is_file_transfer, is_port_forward);
SESSIONS
.write()
.unwrap()
.insert(id.to_owned(), session.clone());
Ok(())
}
/// start a session with the given id.
///
/// # Arguments
///
/// * `id` - The identifier of the remote session with prefix. Regex: [\w]*[\_]*[\d]+
/// * `events2ui` - The events channel to ui.
pub fn session_start_(id: &str, event_stream: StreamSink<EventToUI>) -> ResultType<()> {
if let Some(session) = SESSIONS.write().unwrap().get_mut(id) {
*session.event_stream.write().unwrap() = Some(event_stream);
let session = session.clone();
std::thread::spawn(move || {
// let is_file_transfer = session.lc.read().unwrap().is_file_transfer;
// let is_port_forward = session.lc.read().unwrap().is_port_forward;
// Connection::start(session, is_file_transfer, is_port_forward);
io_loop(session);
});
Ok(())
} else {
bail!("No session with peer id {}", id)
2022-07-11 18:23:58 +08:00
}
2022-05-12 17:35:25 +08:00
}
// struct Connection {
// video_handler: VideoHandler,
// audio_handler: AudioHandler,
// session: Session,
// first_frame: bool,
// read_jobs: Vec<fs::TransferJob>,
// write_jobs: Vec<fs::TransferJob>,
// timer: Interval,
// last_update_jobs_status: (Instant, HashMap<i32, u64>),
// data_count: Arc<AtomicUsize>,
// frame_count: Arc<AtomicUsize>,
// video_format: CodecFormat,
// }
// impl Connection {
// // TODO: Similar to remote::start_clipboard
// // merge the code
// fn start_clipboard(
// tx_protobuf: mpsc::UnboundedSender<Data>,
// lc: Arc<RwLock<LoginConfigHandler>>,
// ) -> Option<std::sync::mpsc::Sender<()>> {
// let (tx, rx) = std::sync::mpsc::channel();
// #[cfg(not(any(target_os = "android", target_os = "ios")))]
// match ClipboardContext::new() {
// Ok(mut ctx) => {
// let old_clipboard: Arc<Mutex<String>> = Default::default();
// // ignore clipboard update before service start
// check_clipboard(&mut ctx, Some(&old_clipboard));
// std::thread::spawn(move || loop {
// std::thread::sleep(Duration::from_millis(CLIPBOARD_INTERVAL));
// match rx.try_recv() {
// Ok(_) | Err(std::sync::mpsc::TryRecvError::Disconnected) => {
// log::debug!("Exit clipboard service of client");
// break;
// }
// _ => {}
// }
// if !SERVER_CLIPBOARD_ENABLED.load(Ordering::SeqCst)
// || !SERVER_KEYBOARD_ENABLED.load(Ordering::SeqCst)
// || lc.read().unwrap().disable_clipboard
// {
// continue;
// }
// if let Some(msg) = check_clipboard(&mut ctx, Some(&old_clipboard)) {
// tx_protobuf.send(Data::Message(msg)).ok();
// }
// });
// }
// Err(err) => {
// log::error!("Failed to start clipboard service of client: {}", err);
// }
// }
// Some(tx)
// }
// /// Create a new connection.
// ///
// /// # Arguments
// ///
// /// * `session` - The session to create a new connection for.
// /// * `is_file_transfer` - Whether the connection is for file transfer.
// /// * `is_port_forward` - Whether the connection is for port forward.
// #[tokio::main(flavor = "current_thread")]
// async fn start(session: Session, is_file_transfer: bool, is_port_forward: bool) {
// let mut last_recv_time = Instant::now();
// let (sender, mut receiver) = mpsc::unbounded_channel::<Data>();
// let mut stop_clipboard = None;
// if !is_file_transfer && !is_port_forward {
// stop_clipboard = Self::start_clipboard(sender.clone(), session.lc.clone());
// }
// *session.sender.write().unwrap() = Some(sender.clone());
// let conn_type = if is_file_transfer {
// session.lc.write().unwrap().is_file_transfer = true;
// ConnType::FILE_TRANSFER
// } else if is_port_forward {
// ConnType::PORT_FORWARD // TODO: RDP
// } else {
// ConnType::DEFAULT_CONN
// };
// let key = Config::get_option("key");
// let token = Config::get_option("access_token");
// // TODO rdp & cli args
// let is_rdp = false;
// let args: Vec<i32> = Vec::new();
// if is_port_forward {
// if is_rdp {
// // let port = handler
// // .get_option("rdp_port".to_owned())
// // .parse::<i32>()
// // .unwrap_or(3389);
// // std::env::set_var(
// // "rdp_username",
// // handler.get_option("rdp_username".to_owned()),
// // );
// // std::env::set_var(
// // "rdp_password",
// // handler.get_option("rdp_password".to_owned()),
// // );
// // log::info!("Remote rdp port: {}", port);
// // start_one_port_forward(handler, 0, "".to_owned(), port, receiver, &key, &token).await;
// } else if args.len() == 0 {
// let pfs = session.lc.read().unwrap().port_forwards.clone();
// let mut queues = HashMap::<i32, mpsc::UnboundedSender<Data>>::new();
// for d in pfs {
// sender.send(Data::AddPortForward(d)).ok();
// }
// loop {
// match receiver.recv().await {
// Some(Data::AddPortForward((port, remote_host, remote_port))) => {
// if port <= 0 || remote_port <= 0 {
// continue;
// }
// let (sender, receiver) = mpsc::unbounded_channel::<Data>();
// queues.insert(port, sender);
// let handler = session.clone();
// let key = key.clone();
// let token = token.clone();
// tokio::spawn(async move {
// start_one_port_forward(
// handler,
// port,
// remote_host,
// remote_port,
// receiver,
// &key,
// &token,
// )
// .await;
// });
// }
// Some(Data::RemovePortForward(port)) => {
// if let Some(s) = queues.remove(&port) {
// s.send(Data::Close).ok();
// }
// }
// Some(Data::Close) => {
// break;
// }
// Some(d) => {
// for (_, s) in queues.iter() {
// s.send(d.clone()).ok();
// }
// }
// _ => {}
// }
// }
// } else {
// // let port = handler.args[0].parse::<i32>().unwrap_or(0);
// // if handler.args.len() != 3
// // || handler.args[2].parse::<i32>().unwrap_or(0) <= 0
// // || port <= 0
// // {
// // handler.on_error("Invalid arguments, usage:<br><br> rustdesk --port-forward remote-id listen-port remote-host remote-port");
// // }
// // let remote_host = handler.args[1].clone();
// // let remote_port = handler.args[2].parse::<i32>().unwrap_or(0);
// // start_one_port_forward(
// // handler,
// // port,
// // remote_host,
// // remote_port,
// // receiver,
// // &key,
// // &token,
// // )
// // .await;
// }
// return;
// }
// let latency_controller = LatencyController::new();
// let latency_controller_cl = latency_controller.clone();
// let mut conn = Connection {
// video_handler: VideoHandler::new(latency_controller),
// audio_handler: AudioHandler::new(latency_controller_cl),
// session: session.clone(),
// first_frame: false,
// read_jobs: Vec::new(),
// write_jobs: Vec::new(),
// timer: time::interval(SEC30),
// last_update_jobs_status: (Instant::now(), Default::default()),
// data_count: Arc::new(AtomicUsize::new(0)),
// frame_count: Arc::new(AtomicUsize::new(0)),
// video_format: CodecFormat::Unknown,
// };
// match Client::start(&session.id, &key, &token, conn_type, session.clone()).await {
// Ok((mut peer, direct)) => {
// SERVER_KEYBOARD_ENABLED.store(true, Ordering::SeqCst);
// SERVER_CLIPBOARD_ENABLED.store(true, Ordering::SeqCst);
// session.push_event(
// "connection_ready",
// vec![
// ("secure", &peer.is_secured().to_string()),
// ("direct", &direct.to_string()),
// ],
// );
// let mut status_timer = time::interval(Duration::new(1, 0));
// loop {
// tokio::select! {
// res = peer.next() => {
// if let Some(res) = res {
// match res {
// Err(err) => {
// log::error!("Connection closed: {}", err);
// session.msgbox("error", "Connection Error", &err.to_string());
// break;
// }
// Ok(ref bytes) => {
// last_recv_time = Instant::now();
// conn.data_count.fetch_add(bytes.len(), Ordering::Relaxed);
// if !conn.handle_msg_from_peer(bytes, &mut peer).await {
// break
// }
// }
// }
// } else {
// if session.lc.read().unwrap().restarting_remote_device {
// log::info!("Restart remote device");
// session.msgbox("restarting", "Restarting Remote Device", "remote_restarting_tip");
// } else {
// log::info!("Reset by the peer");
// session.msgbox("error", "Connection Error", "Reset by the peer");
// }
// break;
// }
// }
// d = receiver.recv() => {
// if let Some(d) = d {
// if !conn.handle_msg_from_ui(d, &mut peer).await {
// break;
// }
// }
// }
// _ = conn.timer.tick() => {
// if last_recv_time.elapsed() >= SEC30 {
// session.msgbox("error", "Connection Error", "Timeout");
// break;
// }
// if !conn.read_jobs.is_empty() {
// if let Err(err) = fs::handle_read_jobs(&mut conn.read_jobs, &mut peer).await {
// log::debug!("Connection Error: {}", err);
// break;
// }
// conn.update_jobs_status();
// } else {
// conn.timer = time::interval_at(Instant::now() + SEC30, SEC30);
// }
// }
// _ = status_timer.tick() => {
// let speed = conn.data_count.swap(0, Ordering::Relaxed);
// let speed = format!("{:.2}kB/s", speed as f32 / 1024 as f32);
// let fps = conn.frame_count.swap(0, Ordering::Relaxed) as _;
// conn.session.update_quality_status(QualityStatus {
// speed:Some(speed),
// fps:Some(fps),
// ..Default::default()
// });
// }
// }
// }
// log::debug!("Exit io_loop of id={}", session.id);
// }
// Err(err) => {
// session.msgbox("error", "Connection Error", &err.to_string());
// }
// }
// if let Some(stop) = stop_clipboard {
// stop.send(()).ok();
// }
// SERVER_KEYBOARD_ENABLED.store(false, Ordering::SeqCst);
// SERVER_CLIPBOARD_ENABLED.store(false, Ordering::SeqCst);
// }
// /// Handle message from peer.
// /// Return false if the connection should be closed.
// ///
// /// The message is handled by [`Message`], see [`message::Union`] for possible types.
// async fn handle_msg_from_peer(&mut self, data: &[u8], peer: &mut Stream) -> bool {
// if let Ok(msg_in) = Message::parse_from_bytes(&data) {
// match msg_in.union {
// Some(message::Union::VideoFrame(vf)) => {
// if !self.first_frame {
// self.first_frame = true;
// common::send_opts_after_login(&self.session.lc.read().unwrap(), peer).await;
// }
// let incomming_format = CodecFormat::from(&vf);
// if self.video_format != incomming_format {
// self.video_format = incomming_format.clone();
// self.session.update_quality_status(QualityStatus {
// codec_format: Some(incomming_format),
// ..Default::default()
// })
// };
// if let Ok(true) = self.video_handler.handle_frame(vf) {
// if let Some(stream) = &*self.session.events2ui.read().unwrap() {
// self.frame_count.fetch_add(1, Ordering::Relaxed);
// stream.add(EventToUI::Rgba(ZeroCopyBuffer(
// self.video_handler.rgb.clone(),
// )));
// }
// }
// }
// Some(message::Union::Hash(hash)) => {
// self.session.handle_hash("", hash, peer).await;
// }
// Some(message::Union::LoginResponse(lr)) => match lr.union {
// Some(login_response::Union::Error(err)) => {
// if !self.session.handle_login_error(&err) {
// return false;
// }
// }
// Some(login_response::Union::PeerInfo(pi)) => {
// self.session.handle_peer_info(pi);
// }
// _ => {}
// },
// Some(message::Union::Clipboard(cb)) => {
// if !self.session.lc.read().unwrap().disable_clipboard {
// let content = if cb.compress {
// decompress(&cb.content)
// } else {
// cb.content.into()
// };
// if let Ok(content) = String::from_utf8(content) {
// self.session
// .push_event("clipboard", vec![("content", &content)]);
// }
// }
// }
// Some(message::Union::CursorData(cd)) => {
// let colors = hbb_common::compress::decompress(&cd.colors);
// self.session.push_event(
// "cursor_data",
// vec![
// ("id", &cd.id.to_string()),
// ("hotx", &cd.hotx.to_string()),
// ("hoty", &cd.hoty.to_string()),
// ("width", &cd.width.to_string()),
// ("height", &cd.height.to_string()),
// (
// "colors",
// &serde_json::ser::to_string(&colors).unwrap_or("".to_owned()),
// ),
// ],
// );
// }
// Some(message::Union::CursorId(id)) => {
// self.session
// .push_event("cursor_id", vec![("id", &id.to_string())]);
// }
// Some(message::Union::CursorPosition(cp)) => {
// self.session.push_event(
// "cursor_position",
// vec![("x", &cp.x.to_string()), ("y", &cp.y.to_string())],
// );
// }
// Some(message::Union::FileResponse(fr)) => {
// match fr.union {
// Some(file_response::Union::Dir(fd)) => {
// let mut entries = fd.entries.to_vec();
// if self.session.peer_platform() == "Windows" {
// transform_windows_path(&mut entries);
// }
// let id = fd.id;
// self.session.push_event(
// "file_dir",
// vec![("value", &make_fd_to_json(fd)), ("is_local", "false")],
// );
// if let Some(job) = fs::get_job(id, &mut self.write_jobs) {
// job.set_files(entries);
// }
// }
// Some(file_response::Union::Block(block)) => {
// if let Some(job) = fs::get_job(block.id, &mut self.write_jobs) {
// if let Err(_err) = job.write(block, None).await {
// // to-do: add "skip" for writing job
// }
// self.update_jobs_status();
// }
// }
// Some(file_response::Union::Done(d)) => {
// if let Some(job) = fs::get_job(d.id, &mut self.write_jobs) {
// job.modify_time();
// fs::remove_job(d.id, &mut self.write_jobs);
// }
// self.handle_job_status(d.id, d.file_num, None);
// }
// Some(file_response::Union::Error(e)) => {
// self.handle_job_status(e.id, e.file_num, Some(e.error));
// }
// Some(file_response::Union::Digest(digest)) => {
// if digest.is_upload {
// if let Some(job) = fs::get_job(digest.id, &mut self.read_jobs) {
// if let Some(file) = job.files().get(digest.file_num as usize) {
// let read_path = get_string(&job.join(&file.name));
// let overwrite_strategy = job.default_overwrite_strategy();
// if let Some(overwrite) = overwrite_strategy {
// let req = FileTransferSendConfirmRequest {
// id: digest.id,
// file_num: digest.file_num,
// union: Some(if overwrite {
// file_transfer_send_confirm_request::Union::OffsetBlk(0)
// } else {
// file_transfer_send_confirm_request::Union::Skip(
// true,
// )
// }),
// ..Default::default()
// };
// job.confirm(&req);
// let msg = new_send_confirm(req);
// allow_err!(peer.send(&msg).await);
// } else {
// self.handle_override_file_confirm(
// digest.id,
// digest.file_num,
// read_path,
// true,
// );
// }
// }
// }
// } else {
// if let Some(job) = fs::get_job(digest.id, &mut self.write_jobs) {
// if let Some(file) = job.files().get(digest.file_num as usize) {
// let write_path = get_string(&job.join(&file.name));
// let overwrite_strategy = job.default_overwrite_strategy();
// match fs::is_write_need_confirmation(&write_path, &digest) {
// Ok(res) => match res {
// DigestCheckResult::IsSame => {
// let msg= new_send_confirm(FileTransferSendConfirmRequest {
// id: digest.id,
// file_num: digest.file_num,
// union: Some(file_transfer_send_confirm_request::Union::Skip(true)),
// ..Default::default()
// });
// self.session.send_msg(msg);
// }
// DigestCheckResult::NeedConfirm(digest) => {
// if let Some(overwrite) = overwrite_strategy {
// let msg = new_send_confirm(
// FileTransferSendConfirmRequest {
// id: digest.id,
// file_num: digest.file_num,
// union: Some(if overwrite {
// file_transfer_send_confirm_request::Union::OffsetBlk(0)
// } else {
// file_transfer_send_confirm_request::Union::Skip(true)
// }),
// ..Default::default()
// },
// );
// self.session.send_msg(msg);
// } else {
// self.handle_override_file_confirm(
// digest.id,
// digest.file_num,
// write_path.to_string(),
// false,
// );
// }
// }
// DigestCheckResult::NoSuchFile => {
// let msg = new_send_confirm(
// FileTransferSendConfirmRequest {
// id: digest.id,
// file_num: digest.file_num,
// union: Some(file_transfer_send_confirm_request::Union::OffsetBlk(0)),
// ..Default::default()
// },
// );
// self.session.send_msg(msg);
// }
// },
// Err(err) => {
// println!("error recving digest: {}", err);
// }
// }
// }
// }
// }
// }
// _ => {}
// }
// }
// Some(message::Union::Misc(misc)) => match misc.union {
// Some(misc::Union::AudioFormat(f)) => {
// self.audio_handler.handle_format(f); //
// }
// Some(misc::Union::ChatMessage(c)) => {
// self.session
// .push_event("chat_client_mode", vec![("text", &c.text)]);
// }
// Some(misc::Union::PermissionInfo(p)) => {
// log::info!("Change permission {:?} -> {}", p.permission, p.enabled);
// use permission_info::Permission;
// self.session.push_event(
// "permission",
// vec![(
// match p.permission.enum_value_or_default() {
// Permission::Keyboard => "keyboard",
// Permission::Clipboard => "clipboard",
// Permission::Audio => "audio",
// Permission::Restart => "restart",
// _ => "",
// },
// &p.enabled.to_string(),
// )],
// );
// }
// Some(misc::Union::SwitchDisplay(s)) => {
// self.video_handler.reset();
// self.session.push_event(
// "switch_display",
// vec![
// ("display", &s.display.to_string()),
// ("x", &s.x.to_string()),
// ("y", &s.y.to_string()),
// ("width", &s.width.to_string()),
// ("height", &s.height.to_string()),
// ],
// );
// }
// Some(misc::Union::CloseReason(c)) => {
// self.session.msgbox("error", "Connection Error", &c);
// return false;
// }
// Some(misc::Union::BackNotification(notification)) => {
// if !self.handle_back_notification(notification).await {
// return false;
// }
// }
// _ => {}
// },
// Some(message::Union::TestDelay(t)) => {
// self.session.handle_test_delay(t, peer).await;
// }
// Some(message::Union::AudioFrame(frame)) => {
// if !self.session.lc.read().unwrap().disable_audio {
// self.audio_handler.handle_frame(frame);
// }
// }
// Some(message::Union::FileAction(action)) => match action.union {
// Some(file_action::Union::SendConfirm(c)) => {
// if let Some(job) = fs::get_job(c.id, &mut self.read_jobs) {
// job.confirm(&c);
// }
// }
// _ => {}
// },
// _ => {}
// }
// }
// true
// }
// async fn handle_back_notification(&mut self, notification: BackNotification) -> bool {
// match notification.union {
// Some(back_notification::Union::BlockInputState(state)) => {
// self.handle_back_msg_block_input(
// state.enum_value_or(back_notification::BlockInputState::BlkStateUnknown),
// )
// .await;
// }
// Some(back_notification::Union::PrivacyModeState(state)) => {
// if !self
// .handle_back_msg_privacy_mode(
// state.enum_value_or(back_notification::PrivacyModeState::PrvStateUnknown),
// )
// .await
// {
// return false;
// }
// }
// _ => {}
// }
// true
// }
// #[inline(always)]
// fn update_block_input_state(&mut self, on: bool) {
// self.session.push_event(
// "update_block_input_state",
// [("input_state", if on { "on" } else { "off" })].into(),
// );
// }
// async fn handle_back_msg_block_input(&mut self, state: back_notification::BlockInputState) {
// match state {
// back_notification::BlockInputState::BlkOnSucceeded => {
// self.update_block_input_state(true);
// }
// back_notification::BlockInputState::BlkOnFailed => {
// self.session
// .msgbox("custom-error", "Block user input", "Failed");
// self.update_block_input_state(false);
// }
// back_notification::BlockInputState::BlkOffSucceeded => {
// self.update_block_input_state(false);
// }
// back_notification::BlockInputState::BlkOffFailed => {
// self.session
// .msgbox("custom-error", "Unblock user input", "Failed");
// }
// _ => {}
// }
// }
// #[inline(always)]
// fn update_privacy_mode(&mut self, on: bool) {
// let mut config = self.session.load_config();
// config.privacy_mode = on;
// self.session.save_config(&config);
// self.session.lc.write().unwrap().get_config().privacy_mode = on;
// self.session.push_event("update_privacy_mode", [].into());
// }
// async fn handle_back_msg_privacy_mode(
// &mut self,
// state: back_notification::PrivacyModeState,
// ) -> bool {
// match state {
// back_notification::PrivacyModeState::PrvOnByOther => {
// self.session.msgbox(
// "error",
// "Connecting...",
// "Someone turns on privacy mode, exit",
// );
// return false;
// }
// back_notification::PrivacyModeState::PrvNotSupported => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Unsupported");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOnSucceeded => {
// self.session
// .msgbox("custom-nocancel", "Privacy mode", "In privacy mode");
// self.update_privacy_mode(true);
// }
// back_notification::PrivacyModeState::PrvOnFailedDenied => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Peer denied");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOnFailedPlugin => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Please install plugins");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOnFailed => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Failed");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOffSucceeded => {
// self.session
// .msgbox("custom-nocancel", "Privacy mode", "Out privacy mode");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOffByPeer => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Peer exit");
// self.update_privacy_mode(false);
// }
// back_notification::PrivacyModeState::PrvOffFailed => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Failed to turn off");
// }
// back_notification::PrivacyModeState::PrvOffUnknown => {
// self.session
// .msgbox("custom-error", "Privacy mode", "Turned off");
// // log::error!("Privacy mode is turned off with unknown reason");
// self.update_privacy_mode(false);
// }
// _ => {}
// }
// true
// }
// async fn handle_msg_from_ui(&mut self, data: Data, peer: &mut Stream) -> bool {
// match data {
// Data::Close => {
// self.sync_jobs_status_to_local().await;
// return false;
// }
// Data::Login((password, remember)) => {
// self.session
// .handle_login_from_ui(password, remember, peer)
// .await;
// }
// Data::Message(msg) => {
// allow_err!(peer.send(&msg).await);
// }
// Data::SendFiles((id, path, to, file_num, include_hidden, is_remote)) => {
// let od = can_enable_overwrite_detection(self.session.lc.read().unwrap().version);
// if is_remote {
// log::debug!("New job {}, write to {} from remote {}", id, to, path);
// self.write_jobs.push(fs::TransferJob::new_write(
// id,
// path.clone(),
// to,
// file_num,
// include_hidden,
// is_remote,
// Vec::new(),
// od,
// ));
// allow_err!(
// peer.send(&fs::new_send(id, path, file_num, include_hidden))
// .await
// );
// } else {
// match fs::TransferJob::new_read(
// id,
// to.clone(),
// path.clone(),
// file_num,
// include_hidden,
// is_remote,
// od,
// ) {
// Err(err) => {
// self.handle_job_status(id, -1, Some(err.to_string()));
// }
// Ok(job) => {
// log::debug!(
// "New job {}, read {} to remote {}, {} files",
// id,
// path,
// to,
// job.files().len()
// );
// let m = make_fd_flutter(id, job.files(), true);
// self.session
// .push_event("update_folder_files", vec![("info", &m)]);
// let files = job.files().clone();
// self.read_jobs.push(job);
// self.timer = time::interval(MILLI1);
// allow_err!(peer.send(&fs::new_receive(id, to, file_num, files)).await);
// }
// }
// }
// }
// Data::RemoveDirAll((id, path, is_remote, include_hidden)) => {
// if is_remote {
// let mut msg_out = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_all_files(ReadAllFiles {
// id,
// path: path.clone(),
// include_hidden,
// ..Default::default()
// });
// msg_out.set_file_action(file_action);
// allow_err!(peer.send(&msg_out).await);
// } else {
// match fs::get_recursive_files(&path, include_hidden) {
// Ok(entries) => {
// let mut fd = FileDirectory::new();
// fd.id = id;
// fd.path = path;
// fd.entries = entries;
// self.session.push_event(
// "file_dir",
// vec![("value", &make_fd_to_json(fd)), ("is_local", "true")],
// );
// }
// Err(err) => {
// self.handle_job_status(id, -1, Some(err.to_string()));
// }
// }
// }
// }
// Data::CancelJob(id) => {
// let mut msg_out = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_cancel(FileTransferCancel {
// id: id,
// ..Default::default()
// });
// msg_out.set_file_action(file_action);
// allow_err!(peer.send(&msg_out).await);
// if let Some(job) = fs::get_job(id, &mut self.write_jobs) {
// job.remove_download_file();
// fs::remove_job(id, &mut self.write_jobs);
// }
// fs::remove_job(id, &mut self.read_jobs);
// }
// Data::RemoveDir((id, path)) => {
// let mut msg_out = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_remove_dir(FileRemoveDir {
// id,
// path,
// recursive: true,
// ..Default::default()
// });
// msg_out.set_file_action(file_action);
// allow_err!(peer.send(&msg_out).await);
// }
// Data::RemoveFile((id, path, file_num, is_remote)) => {
// if is_remote {
// let mut msg_out = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_remove_file(FileRemoveFile {
// id,
// path,
// file_num,
// ..Default::default()
// });
// msg_out.set_file_action(file_action);
// allow_err!(peer.send(&msg_out).await);
// } else {
// match fs::remove_file(&path) {
// Err(err) => {
// self.handle_job_status(id, file_num, Some(err.to_string()));
// }
// Ok(()) => {
// self.handle_job_status(id, file_num, None);
// }
// }
// }
// }
// Data::CreateDir((id, path, is_remote)) => {
// if is_remote {
// let mut msg_out = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_create(FileDirCreate {
// id,
// path,
// ..Default::default()
// });
// msg_out.set_file_action(file_action);
// allow_err!(peer.send(&msg_out).await);
// } else {
// match fs::create_dir(&path) {
// Err(err) => {
// self.handle_job_status(id, -1, Some(err.to_string()));
// }
// Ok(()) => {
// self.handle_job_status(id, -1, None);
// }
// }
// }
// }
// Data::SetConfirmOverrideFile((id, file_num, need_override, remember, is_upload)) => {
// if is_upload {
// if let Some(job) = fs::get_job(id, &mut self.read_jobs) {
// if remember {
// job.set_overwrite_strategy(Some(need_override));
// }
// job.confirm(&FileTransferSendConfirmRequest {
// id,
// file_num,
// union: if need_override {
// Some(file_transfer_send_confirm_request::Union::OffsetBlk(0))
// } else {
// Some(file_transfer_send_confirm_request::Union::Skip(true))
// },
// ..Default::default()
// });
// }
// } else {
// if let Some(job) = fs::get_job(id, &mut self.write_jobs) {
// if remember {
// job.set_overwrite_strategy(Some(need_override));
// }
// let mut msg = Message::new();
// let mut file_action = FileAction::new();
// file_action.set_send_confirm(FileTransferSendConfirmRequest {
// id,
// file_num,
// union: if need_override {
// Some(file_transfer_send_confirm_request::Union::OffsetBlk(0))
// } else {
// Some(file_transfer_send_confirm_request::Union::Skip(true))
// },
// ..Default::default()
// });
// msg.set_file_action(file_action);
// self.session.send_msg(msg);
// }
// }
// }
// Data::AddJob((id, path, to, file_num, include_hidden, is_remote)) => {
// let od = can_enable_overwrite_detection(self.session.lc.read().unwrap().version);
// if is_remote {
// log::debug!(
// "new write waiting job {}, write to {} from remote {}",
// id,
// to,
// path
// );
// let mut job = fs::TransferJob::new_write(
// id,
// path.clone(),
// to,
// file_num,
// include_hidden,
// is_remote,
// Vec::new(),
// od,
// );
// job.is_last_job = true;
// self.write_jobs.push(job);
// } else {
// match fs::TransferJob::new_read(
// id,
// to.clone(),
// path.clone(),
// file_num,
// include_hidden,
// is_remote,
// od,
// ) {
// Err(err) => {
// self.handle_job_status(id, -1, Some(err.to_string()));
// }
// Ok(mut job) => {
// log::debug!(
// "new read waiting job {}, read {} to remote {}, {} files",
// id,
// path,
// to,
// job.files().len()
// );
// let m = make_fd_flutter(job.id(), job.files(), true);
// self.session
// .push_event("update_folder_files", vec![("info", &m)]);
// job.is_last_job = true;
// self.read_jobs.push(job);
// self.timer = time::interval(MILLI1);
// }
// }
// }
// }
// Data::ResumeJob((id, is_remote)) => {
// if is_remote {
// if let Some(job) = get_job(id, &mut self.write_jobs) {
// job.is_last_job = false;
// allow_err!(
// peer.send(&fs::new_send(
// id,
// job.remote.clone(),
// job.file_num,
// job.show_hidden
// ))
// .await
// );
// }
// } else {
// if let Some(job) = get_job(id, &mut self.read_jobs) {
// job.is_last_job = false;
// allow_err!(
// peer.send(&fs::new_receive(
// id,
// job.path.to_string_lossy().to_string(),
// job.file_num,
// job.files.clone()
// ))
// .await
// );
// }
// }
// }
// _ => {}
// }
// true
// }
// #[inline]
// fn update_job_status(
// job: &fs::TransferJob,
// elapsed: i32,
// last_update_jobs_status: &mut (Instant, HashMap<i32, u64>),
// session: &Session,
// ) {
// if elapsed <= 0 {
// return;
// }
// let transferred = job.transferred();
// let last_transferred = {
// if let Some(v) = last_update_jobs_status.1.get(&job.id()) {
// v.to_owned()
// } else {
// 0
// }
// };
// last_update_jobs_status.1.insert(job.id(), transferred);
// let speed = (transferred - last_transferred) as f64 / (elapsed as f64 / 1000.);
// let file_num = job.file_num() - 1;
// session.push_event(
// "job_progress",
// vec![
// ("id", &job.id().to_string()),
// ("file_num", &file_num.to_string()),
// ("speed", &speed.to_string()),
// ("finished_size", &job.finished_size().to_string()),
// ],
// );
// }
// fn update_jobs_status(&mut self) {
// let elapsed = self.last_update_jobs_status.0.elapsed().as_millis() as i32;
// if elapsed >= 1000 {
// for job in self.read_jobs.iter() {
// Self::update_job_status(
// job,
// elapsed,
// &mut self.last_update_jobs_status,
// &self.session,
// );
// }
// for job in self.write_jobs.iter() {
// Self::update_job_status(
// job,
// elapsed,
// &mut self.last_update_jobs_status,
// &self.session,
// );
// }
// self.last_update_jobs_status.0 = Instant::now();
// }
// }
// fn handle_job_status(&mut self, id: i32, file_num: i32, err: Option<String>) {
// if let Some(err) = err {
// self.session
// .push_event("job_error", vec![("id", &id.to_string()), ("err", &err)]);
// } else {
// self.session.push_event(
// "job_done",
// vec![("id", &id.to_string()), ("file_num", &file_num.to_string())],
// );
// }
// }
// fn handle_override_file_confirm(
// &mut self,
// id: i32,
// file_num: i32,
// read_path: String,
// is_upload: bool,
// ) {
// self.session.push_event(
// "override_file_confirm",
// vec![
// ("id", &id.to_string()),
// ("file_num", &file_num.to_string()),
// ("read_path", &read_path),
// ("is_upload", &is_upload.to_string()),
// ],
// );
// }
// async fn sync_jobs_status_to_local(&mut self) -> bool {
// log::info!("sync transfer job status");
// let mut config: PeerConfig = self.session.load_config();
// let mut transfer_metas = TransferSerde::default();
// for job in self.read_jobs.iter() {
// let json_str = serde_json::to_string(&job.gen_meta()).unwrap();
// transfer_metas.read_jobs.push(json_str);
// }
// for job in self.write_jobs.iter() {
// let json_str = serde_json::to_string(&job.gen_meta()).unwrap();
// transfer_metas.write_jobs.push(json_str);
// }
// log::info!("meta: {:?}", transfer_metas);
// config.transfer = transfer_metas;
// self.session.save_config(&config);
// true
// }
// }
2022-05-12 17:35:25 +08:00
// Server Side
// TODO connection_manager need use struct and trait,impl default method
#[cfg(not(any(target_os = "ios")))]
2022-05-12 17:35:25 +08:00
pub mod connection_manager {
use std::{
collections::HashMap,
2022-05-12 17:35:25 +08:00
iter::FromIterator,
2022-08-17 17:23:55 +08:00
sync::{
atomic::{AtomicI64, Ordering},
RwLock,
},
2022-05-12 17:35:25 +08:00
};
use serde_derive::Serialize;
2022-05-12 17:35:25 +08:00
use hbb_common::{
allow_err,
config::Config,
2022-05-19 21:45:25 +08:00
fs::is_write_need_confirmation,
fs::{self, get_string, new_send_confirm, DigestCheckResult},
log,
message_proto::*,
protobuf::Message as _,
2022-05-12 17:35:25 +08:00
tokio::{
self,
2022-08-17 17:23:55 +08:00
sync::mpsc::{self, UnboundedReceiver, UnboundedSender},
2022-05-12 17:35:25 +08:00
task::spawn_blocking,
2022-05-19 21:45:25 +08:00
},
2022-05-12 17:35:25 +08:00
};
#[cfg(any(target_os = "android"))]
2022-05-12 17:35:25 +08:00
use scrap::android::call_main_service_set_by_name;
use crate::ipc::Data;
2022-08-17 17:23:55 +08:00
use crate::ipc::{self, new_listener, Connection};
2022-05-12 17:35:25 +08:00
2022-05-31 22:09:36 +08:00
use super::GLOBAL_EVENT_STREAM;
2022-05-12 17:35:25 +08:00
#[derive(Debug, Serialize, Clone)]
struct Client {
id: i32,
pub authorized: bool,
is_file_transfer: bool,
name: String,
peer_id: String,
keyboard: bool,
clipboard: bool,
audio: bool,
file: bool,
restart: bool,
2022-05-12 17:35:25 +08:00
#[serde(skip)]
tx: UnboundedSender<Data>,
}
lazy_static::lazy_static! {
static ref CLIENTS: RwLock<HashMap<i32,Client>> = Default::default();
}
2022-08-17 17:23:55 +08:00
static CLICK_TIME: AtomicI64 = AtomicI64::new(0);
// // TODO clipboard_file
// enum ClipboardFileData {
// #[cfg(windows)]
// Clip((i32, ipc::ClipbaordFile)),
// Enable((i32, bool)),
// }
2022-08-17 17:23:55 +08:00
#[cfg(not(any(target_os = "android", target_os = "ios")))]
pub fn start_listen_ipc_thread() {
std::thread::spawn(move || start_ipc());
}
#[cfg(not(any(target_os = "android", target_os = "ios")))]
#[tokio::main(flavor = "current_thread")]
async fn start_ipc() {
// TODO clipboard_file
// let (tx_file, _rx_file) = mpsc::unbounded_channel::<ClipboardFileData>();
// #[cfg(windows)]
// let cm_clip = cm.clone();
// #[cfg(windows)]
// std::thread::spawn(move || start_clipboard_file(cm_clip, _rx_file));
2022-08-17 17:23:55 +08:00
#[cfg(windows)]
std::thread::spawn(move || {
log::info!("try create privacy mode window");
#[cfg(windows)]
{
if let Err(e) = crate::platform::windows::check_update_broker_process() {
log::warn!(
"Failed to check update broker process. Privacy mode may not work properly. {}",
e
);
}
}
allow_err!(crate::ui::win_privacy::start());
});
match new_listener("_cm").await {
Ok(mut incoming) => {
while let Some(result) = incoming.next().await {
match result {
Ok(stream) => {
log::debug!("Got new connection");
let mut stream = Connection::new(stream);
// let tx_file = tx_file.clone();
2022-08-17 17:23:55 +08:00
tokio::spawn(async move {
// for tmp use, without real conn id
let conn_id_tmp = -1;
let mut conn_id: i32 = 0;
let (tx, mut rx) = mpsc::unbounded_channel::<Data>();
let mut write_jobs: Vec<fs::TransferJob> = Vec::new();
loop {
tokio::select! {
res = stream.next() => {
match res {
Err(err) => {
log::info!("cm ipc connection closed: {}", err);
break;
}
Ok(Some(data)) => {
match data {
Data::Login{id, is_file_transfer, port_forward, peer_id, name, authorized, keyboard, clipboard, audio, file, file_transfer_enabled, restart} => {
log::debug!("conn_id: {}", id);
conn_id = id;
// tx_file.send(ClipboardFileData::Enable((id, file_transfer_enabled))).ok();
2022-08-17 17:23:55 +08:00
on_login(id, is_file_transfer, port_forward, peer_id, name, authorized, keyboard, clipboard, audio, file, restart, tx.clone());
}
Data::Close => {
// tx_file.send(ClipboardFileData::Enable((conn_id, false))).ok();
2022-08-17 17:23:55 +08:00
log::info!("cm ipc connection closed from connection request");
break;
}
Data::PrivacyModeState((_, _)) => {
conn_id = conn_id_tmp;
allow_err!(tx.send(data));
}
Data::ClickTime(ms) => {
CLICK_TIME.store(ms, Ordering::SeqCst);
}
Data::ChatMessage { text } => {
handle_chat(conn_id, text);
}
Data::FS(fs) => {
handle_fs(fs, &mut write_jobs, &tx).await;
}
// TODO ClipbaordFile
// #[cfg(windows)]
// Data::ClipbaordFile(_clip) => {
// tx_file
// .send(ClipboardFileData::Clip((id, _clip)))
// .ok();
// }
// #[cfg(windows)]
// Data::ClipboardFileEnabled(enabled) => {
// tx_file
// .send(ClipboardFileData::Enable((id, enabled)))
// .ok();
// }
2022-08-17 17:23:55 +08:00
_ => {}
}
}
_ => {}
}
}
Some(data) = rx.recv() => {
if stream.send(&data).await.is_err() {
break;
}
}
}
}
if conn_id != conn_id_tmp {
remove_connection(conn_id);
}
});
}
Err(err) => {
log::error!("Couldn't get cm client: {:?}", err);
}
}
}
}
Err(err) => {
log::error!("Failed to start cm ipc server: {}", err);
}
}
// crate::platform::quit_gui();
// TODO flutter quit_gui
}
#[cfg(target_os = "android")]
2022-05-12 17:35:25 +08:00
pub fn start_channel(rx: UnboundedReceiver<Data>, tx: UnboundedSender<Data>) {
std::thread::spawn(move || start_listen(rx, tx));
}
2022-08-17 17:23:55 +08:00
#[cfg(target_os = "android")]
2022-05-12 17:35:25 +08:00
#[tokio::main(flavor = "current_thread")]
async fn start_listen(mut rx: UnboundedReceiver<Data>, tx: UnboundedSender<Data>) {
let mut current_id = 0;
2022-08-17 17:23:55 +08:00
let mut write_jobs: Vec<fs::TransferJob> = Vec::new();
2022-05-12 17:35:25 +08:00
loop {
match rx.recv().await {
Some(Data::Login {
id,
is_file_transfer,
2022-08-17 17:23:55 +08:00
port_forward,
2022-05-12 17:35:25 +08:00
peer_id,
name,
authorized,
keyboard,
clipboard,
audio,
2022-08-17 17:23:55 +08:00
file,
restart,
2022-07-28 22:06:59 +08:00
..
2022-05-12 17:35:25 +08:00
}) => {
current_id = id;
2022-08-17 17:23:55 +08:00
on_login(
2022-05-12 17:35:25 +08:00
id,
is_file_transfer,
2022-08-17 17:23:55 +08:00
port_forward,
peer_id,
name,
authorized,
2022-05-12 17:35:25 +08:00
keyboard,
clipboard,
audio,
2022-08-17 17:23:55 +08:00
file,
restart,
tx.clone(),
);
2022-05-12 17:35:25 +08:00
}
Some(Data::ChatMessage { text }) => {
handle_chat(current_id, text);
}
Some(Data::FS(fs)) => {
2022-08-17 17:23:55 +08:00
handle_fs(fs, &mut write_jobs, &tx).await;
2022-05-12 17:35:25 +08:00
}
Some(Data::Close) => {
break;
}
None => {
break;
}
_ => {}
}
}
remove_connection(current_id);
}
2022-08-17 17:23:55 +08:00
fn on_login(
id: i32,
is_file_transfer: bool,
_port_forward: String,
peer_id: String,
name: String,
authorized: bool,
keyboard: bool,
clipboard: bool,
audio: bool,
file: bool,
restart: bool,
2022-08-17 17:23:55 +08:00
tx: mpsc::UnboundedSender<Data>,
) {
let mut client = Client {
id,
authorized,
is_file_transfer,
name: name.clone(),
peer_id: peer_id.clone(),
keyboard,
clipboard,
audio,
file,
restart,
2022-08-17 17:23:55 +08:00
tx,
};
if authorized {
client.authorized = true;
let client_json = serde_json::to_string(&client).unwrap_or("".into());
// send to Android service, active notification no matter UI is shown or not.
#[cfg(any(target_os = "android"))]
if let Err(e) =
call_main_service_set_by_name("on_client_authorized", Some(&client_json), None)
{
log::debug!("call_service_set_by_name fail,{}", e);
}
// send to UI, refresh widget
push_event("on_client_authorized", vec![("client", &client_json)]);
} else {
let client_json = serde_json::to_string(&client).unwrap_or("".into());
// send to Android service, active notification no matter UI is shown or not.
#[cfg(any(target_os = "android"))]
if let Err(e) =
call_main_service_set_by_name("try_start_without_auth", Some(&client_json), None)
{
log::debug!("call_service_set_by_name fail,{}", e);
}
// send to UI, refresh widget
push_event("try_start_without_auth", vec![("client", &client_json)]);
}
CLIENTS.write().unwrap().insert(id, client);
}
fn push_event(name: &str, event: Vec<(&str, &str)>) {
let mut h: HashMap<&str, &str> = event.iter().cloned().collect();
assert!(h.get("name").is_none());
h.insert("name", name);
if let Some(s) = GLOBAL_EVENT_STREAM
.read()
.unwrap()
.get(super::APP_TYPE_MAIN)
{
s.add(serde_json::ser::to_string(&h).unwrap_or("".to_owned()));
};
}
2022-08-17 17:23:55 +08:00
pub fn get_click_time() -> i64 {
CLICK_TIME.load(Ordering::SeqCst)
}
pub fn check_click_time(id: i32) {
if let Some(client) = CLIENTS.read().unwrap().get(&id) {
allow_err!(client.tx.send(Data::ClickTime(0)));
};
}
pub fn switch_permission(id: i32, name: String, enabled: bool) {
if let Some(client) = CLIENTS.read().unwrap().get(&id) {
allow_err!(client.tx.send(Data::SwitchPermission { name, enabled }));
};
}
2022-05-12 17:35:25 +08:00
pub fn get_clients_state() -> String {
let clients = CLIENTS.read().unwrap();
let res = Vec::from_iter(clients.values().cloned());
serde_json::to_string(&res).unwrap_or("".into())
}
pub fn get_clients_length() -> usize {
let clients = CLIENTS.read().unwrap();
clients.len()
}
pub fn close_conn(id: i32) {
2022-08-17 17:23:55 +08:00
if let Some(client) = CLIENTS.read().unwrap().get(&id) {
2022-05-12 17:35:25 +08:00
allow_err!(client.tx.send(Data::Close));
};
}
pub fn on_login_res(id: i32, res: bool) {
if let Some(client) = CLIENTS.write().unwrap().get_mut(&id) {
if res {
allow_err!(client.tx.send(Data::Authorize));
client.authorized = true;
} else {
allow_err!(client.tx.send(Data::Close));
}
};
}
fn remove_connection(id: i32) {
let mut clients = CLIENTS.write().unwrap();
clients.remove(&id);
if clients
.iter()
2022-08-17 17:23:55 +08:00
.filter(|(_k, v)| !v.is_file_transfer)
2022-05-12 17:35:25 +08:00
.next()
.is_none()
{
#[cfg(any(target_os = "android"))]
2022-05-12 17:35:25 +08:00
if let Err(e) = call_main_service_set_by_name("stop_capture", None, None) {
log::debug!("stop_capture err:{}", e);
}
}
push_event("on_client_remove", vec![("id", &id.to_string())]);
2022-05-12 17:35:25 +08:00
}
// server mode handle chat from other peers
fn handle_chat(id: i32, text: String) {
push_event(
"chat_server_mode",
vec![("id", &id.to_string()), ("text", &text)],
);
2022-05-12 17:35:25 +08:00
}
// server mode send chat to peer
pub fn send_chat(id: i32, text: String) {
2022-08-17 17:23:55 +08:00
let clients = CLIENTS.read().unwrap();
2022-05-12 17:35:25 +08:00
if let Some(client) = clients.get(&id) {
allow_err!(client.tx.send(Data::ChatMessage { text }));
}
}
// handle FS server
2022-08-17 17:23:55 +08:00
async fn handle_fs(
fs: ipc::FS,
write_jobs: &mut Vec<fs::TransferJob>,
tx: &UnboundedSender<Data>,
) {
2022-05-12 17:35:25 +08:00
match fs {
ipc::FS::ReadDir {
dir,
include_hidden,
} => {
read_dir(&dir, include_hidden, tx).await;
}
ipc::FS::RemoveDir {
path,
id,
recursive,
} => {
remove_dir(path, id, recursive, tx).await;
}
ipc::FS::RemoveFile { path, id, file_num } => {
remove_file(path, id, file_num, tx).await;
}
ipc::FS::CreateDir { path, id } => {
create_dir(path, id, tx).await;
}
ipc::FS::NewWrite {
path,
id,
2022-05-17 09:10:21 +08:00
file_num,
2022-05-12 17:35:25 +08:00
mut files,
2022-08-04 17:24:02 +08:00
overwrite_detection,
2022-05-12 17:35:25 +08:00
} => {
2022-08-17 17:23:55 +08:00
write_jobs.push(fs::TransferJob::new_write(
2022-05-12 17:35:25 +08:00
id,
2022-05-17 09:10:21 +08:00
"".to_string(),
2022-05-12 17:35:25 +08:00
path,
2022-05-17 09:10:21 +08:00
file_num,
false,
false,
2022-05-12 17:35:25 +08:00
files
.drain(..)
.map(|f| FileEntry {
name: f.0,
modified_time: f.1,
..Default::default()
})
.collect(),
2022-08-04 17:24:02 +08:00
overwrite_detection,
2022-05-12 17:35:25 +08:00
));
}
ipc::FS::CancelWrite { id } => {
if let Some(job) = fs::get_job(id, write_jobs) {
job.remove_download_file();
fs::remove_job(id, write_jobs);
}
}
ipc::FS::WriteDone { id, file_num } => {
if let Some(job) = fs::get_job(id, write_jobs) {
job.modify_time();
send_raw(fs::new_done(id, file_num), tx);
fs::remove_job(id, write_jobs);
}
}
ipc::FS::WriteBlock {
id,
file_num,
data,
compressed,
} => {
2022-08-17 17:23:55 +08:00
if let Some(job) = fs::get_job(id, write_jobs) {
2022-05-12 17:35:25 +08:00
if let Err(err) = job
.write(
FileTransferBlock {
id,
file_num,
data,
compressed,
..Default::default()
},
None,
)
.await
{
send_raw(fs::new_error(id, err, file_num), &tx);
}
}
}
2022-05-17 20:56:36 +08:00
ipc::FS::CheckDigest {
id,
file_num,
file_size,
last_modified,
is_upload,
} => {
2022-08-17 17:23:55 +08:00
if let Some(job) = fs::get_job(id, write_jobs) {
2022-05-17 20:56:36 +08:00
let mut req = FileTransferSendConfirmRequest {
id,
file_num,
2022-07-17 16:47:19 +08:00
union: Some(file_transfer_send_confirm_request::Union::OffsetBlk(0)),
2022-05-17 20:56:36 +08:00
..Default::default()
};
let digest = FileTransferDigest {
id,
file_num,
last_modified,
file_size,
..Default::default()
};
if let Some(file) = job.files().get(file_num as usize) {
let path = get_string(&job.join(&file.name));
match is_write_need_confirmation(&path, &digest) {
Ok(digest_result) => {
match digest_result {
DigestCheckResult::IsSame => {
req.set_skip(true);
let msg_out = new_send_confirm(req);
send_raw(msg_out, &tx);
}
DigestCheckResult::NeedConfirm(mut digest) => {
// upload to server, but server has the same file, request
digest.is_upload = is_upload;
let mut msg_out = Message::new();
let mut fr = FileResponse::new();
fr.set_digest(digest);
msg_out.set_file_response(fr);
send_raw(msg_out, &tx);
}
DigestCheckResult::NoSuchFile => {
let msg_out = new_send_confirm(req);
send_raw(msg_out, &tx);
}
}
}
Err(err) => {
send_raw(fs::new_error(id, err, file_num), &tx);
}
}
}
}
}
2022-05-12 17:35:25 +08:00
_ => {}
}
}
async fn read_dir(dir: &str, include_hidden: bool, tx: &UnboundedSender<Data>) {
let path = {
if dir.is_empty() {
Config::get_home()
} else {
fs::get_path(dir)
}
};
if let Ok(Ok(fd)) = spawn_blocking(move || fs::read_dir(&path, include_hidden)).await {
let mut msg_out = Message::new();
let mut file_response = FileResponse::new();
file_response.set_dir(fd);
msg_out.set_file_response(file_response);
send_raw(msg_out, tx);
}
}
async fn handle_result<F: std::fmt::Display, S: std::fmt::Display>(
res: std::result::Result<std::result::Result<(), F>, S>,
id: i32,
file_num: i32,
tx: &UnboundedSender<Data>,
) {
match res {
Err(err) => {
send_raw(fs::new_error(id, err, file_num), tx);
}
Ok(Err(err)) => {
send_raw(fs::new_error(id, err, file_num), tx);
}
Ok(Ok(())) => {
send_raw(fs::new_done(id, file_num), tx);
}
}
}
async fn remove_file(path: String, id: i32, file_num: i32, tx: &UnboundedSender<Data>) {
handle_result(
spawn_blocking(move || fs::remove_file(&path)).await,
id,
file_num,
tx,
)
.await;
}
async fn create_dir(path: String, id: i32, tx: &UnboundedSender<Data>) {
handle_result(
spawn_blocking(move || fs::create_dir(&path)).await,
id,
0,
tx,
)
.await;
}
async fn remove_dir(path: String, id: i32, recursive: bool, tx: &UnboundedSender<Data>) {
let path = fs::get_path(&path);
handle_result(
spawn_blocking(move || {
if recursive {
fs::remove_all_empty_dir(&path)
} else {
std::fs::remove_dir(&path).map_err(|err| err.into())
}
})
.await,
id,
0,
tx,
)
.await;
}
fn send_raw(msg: Message, tx: &UnboundedSender<Data>) {
match msg.write_to_bytes() {
Ok(bytes) => {
allow_err!(tx.send(Data::RawMessage(bytes)));
}
err => allow_err!(err),
}
}
}
#[inline]
pub fn get_session_id(id: String) -> String {
return if let Some(index) = id.find('_') {
id[index + 1..].to_string()
} else {
id
};
}
2022-08-29 13:08:42 +08:00
// async fn start_one_port_forward(
// handler: Session,
// port: i32,
// remote_host: String,
// remote_port: i32,
// receiver: mpsc::UnboundedReceiver<Data>,
// key: &str,
// token: &str,
// ) {
// if let Err(err) = crate::port_forward::listen(
// handler.id.clone(),
// String::new(), // TODO
// port,
// handler.clone(),
// receiver,
// key,
// token,
// handler.lc.clone(),
// remote_host,
// remote_port,
// )
// .await
// {
// handler.on_error(&format!("Failed to listen on {}: {}", port, err));
// }
// log::info!("port forward (:{}) exit", port);
// }